lab-cosmo / metatrain

Training and evaluating machine learning models for atomistic systems.
https://lab-cosmo.github.io/metatrain/
BSD 3-Clause "New" or "Revised" License
13 stars 3 forks source link

Checkpoints are always exported in `float32` and `cpu`, no matter the data type of the checkpoint #272

Closed frostedoyster closed 5 days ago