best_val_mae_both_model is not written if do_forces = MLIP_SETTINGS.U…

…SE_FORCES = False
metatensor · Nov 21, 2024 · f3525ed · f3525ed
1 parent 4493189
commit f3525ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/metatrain/experimental/pet/trainer.py b/src/metatrain/experimental/pet/trainer.py
@@ -721,7 +721,7 @@ def save_model(model_name, model_keeper):
         if do_forces:
             load_path = self.pet_dir / "best_val_mae_forces_model_state_dict"
         else:
-            load_path = self.pet_dir / "best_val_mae_both_model_state_dict"
+            load_path = self.pet_dir / "best_val_mae_energies_model_state_dict"
 
         state_dict = torch.load(load_path, weights_only=True)
         ARCHITECTURAL_HYPERS = Hypers(model.hypers)