search

ACEsuit / mace

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

What is `compiled` model in new version 0.3.5? #401

Closed jungsdao closed 1 month ago

jungsdao commented 2 months ago

After updating to newer version 0.3.5, when training is over, I'm getting total four models in case I've activated swa. 

*.model
*_compiled.model
*_swa.model
*_swa_compiled.model

What's the difference between normal model and compiled model? I think this has not been documented somewhere yet.

ilyes319 commented 2 months ago

The compiled model is just jit compiled. I created that mainly to save metadata, such as the git commit of the model and the hyper parameters dict. It will be documented soon.