search

metatensor / metatrain

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

Remove architecture from `mtt export` #383

Closed frostedoyster closed 8 hours ago

frostedoyster commented 2 weeks ago

As discussed in #352, we should probably remove the architecture from the export syntax, e.g. mtt export model.ckpt instead of mtt export experimental.soap_bpnn model.ckpt It would be more user-friendly

PicoCentauri commented 2 weeks ago

Yes, it would be very cool. This means we have to store the architecture name in some way accessible in the checkpoint. I don't know if there is a way to do this without opening the model. But we can't open the model because this requires knowing its name.