multi gpu experiments - Githubissues

molecularsets / moses

Molecular Sets (MOSES): A Benchmarking Platform for Molecular Generation Models

https://arxiv.org/abs/1811.12823

MIT License

821 stars 241 forks source link

multi gpu experiments #83

Closed trinayan closed 4 years ago

trinayan commented 4 years ago

Hi,

Have there been any effort in scaling the models to multi-GPU systems for training?. I am not a molecular domain expert. I am curious if there are situations where using multi-GPU would help for the models used in the benchmark suite.

Best, Trinayan

danpol commented 4 years ago

Hi, @trinayan!

We did not implement multi-GPU training for the baselines, but it could be profitable for reducing the training time for some models.

trinayan commented 4 years ago

Got it. Thank you for the info