added final epoch number to final checkpoint name

Noble-Lab · Jun 20, 2024 · 8e292b1 · 8e292b1
1 parent 3f033b2
commit 8e292b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/casanovo/denovo/model_runner.py b/casanovo/denovo/model_runner.py
@@ -116,7 +116,7 @@ def train(
             self.trainer.save_checkpoint(
                 os.path.join(
                     self.config.model_save_folder_path,
-                    "train-run-final.ckpt"
+                    f"train-run-final-{self.trainer.current_epoch}.ckpt"
                 )
             )