add .pt

Signed-off-by: Matteo Bettini <[email protected]>
facebookresearch · Sep 14, 2023 · 60051e6 · 60051e6
1 parent c7335e2
commit 60051e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarl/experiment/experiment.py b/benchmarl/experiment/experiment.py
@@ -513,7 +513,7 @@ def load_state_dict(self, state_dict: Dict) -> None:
     def save_trainer(self) -> None:
         checkpoint_folder = self.folder_name / "checkpoints"
         checkpoint_folder.mkdir(parents=False, exist_ok=True)
-        checkpoint_file = checkpoint_folder / f"checkpoint_{self.n_iters_performed}"
+        checkpoint_file = checkpoint_folder / f"checkpoint_{self.n_iters_performed}.pt"
         torch.save(self.state_dict(), checkpoint_file)
 
     def load_trainer(self) -> Experiment: