ACEsuit / mace

MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing.
Other
413 stars 157 forks source link

How do you make mutli-GPU training? #407

Closed stargolike closed 1 month ago

stargolike commented 2 months ago

I find mutli-GPU training has developed in mace version 0.3.5.issue 363

However, I see that the official tutorials do not demonstrate, have some documentation for practice?

ilyes319 commented 1 month ago

There is a section in the readme, see "Multi-GPU training", plus example scripts. We will add documentation soon.

stargolike commented 1 month ago

There is a section in the readme, see "Multi-GPU training", plus example scripts. We will add documentation soon.

tks, i find it. Hope you can improve the documentation.