313 save final model

casanovo/denovo/model_runner.py

@@ -111,6 +111,15 @@ def train(
             self.loaders.val_dataloader(),
         )
 
+        # Always save final model weights at the end of training
+        if self.config.model_save_folder_path is not None:
+            self.trainer.save_checkpoint(
+                os.path.join(
+                    self.config.model_save_folder_path,
+                    f"train-run-final-{self.trainer.current_epoch}.ckpt",
+                )
+            )
+
     def evaluate(self, peak_path: Iterable[str]) -> None:
         """Evaluate peptide sequence preditions from a trained Casanovo model.