diff --git a/examples/ase/run_ase.py b/examples/ase/run_ase.py index 1c7550a38..cf31dc59a 100644 --- a/examples/ase/run_ase.py +++ b/examples/ase/run_ase.py @@ -15,7 +15,7 @@ .. literalinclude:: options.yaml :language: yaml -You can use the pretrained and exported :download:`model ` +You can use the pretrained and exported :download:`model ` or train the model yourself with .. literalinclude:: train.sh @@ -82,7 +82,7 @@ # We now register our exported model as the energy calculator to obtain energies and # forces. -atoms.calc = MetatensorCalculator("exported-model.pt") +atoms.calc = MetatensorCalculator("model.pt") # %% #