Arslan's suggestion

metatensor · Nov 21, 2024 · be98f83 · be98f83
1 parent f3525ed
commit be98f83
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
@@ -719,7 +719,7 @@ def save_model(model_name, model_keeper):
             os.remove(Path(checkpoint_dir) / "checkpoint.temp")
 
         if do_forces:
-            load_path = self.pet_dir / "best_val_mae_forces_model_state_dict"
+            load_path = self.pet_dir / "best_val_mae_both_model_state_dict"
         else:
             load_path = self.pet_dir / "best_val_mae_energies_model_state_dict"