Arslan's suggestion

metatensor · Nov 20, 2024 · 1e26b3c · 1e26b3c
1 parent b0a4b99
commit 1e26b3c
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
@@ -705,7 +705,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"