delete callbacks

2025-04-09 11:15:09 +08:00
parent 4f60297dfe
commit d805c7cb71
1 changed files with 1 additions and 5 deletions
--- a/src/callbacks/model_checkpoint.py
+++ b/src/callbacks/model_checkpoint.py
@@ -19,7 +19,3 @@ class CheckpointHook(ModelCheckpoint):
            checkpoint: Dict[str, Any]
    ) -> None:
        del checkpoint["callbacks"]
-
-    # def on_exception(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule", exception: BaseException) -> None:
-    #     if not "debug" in self.exception_ckpt_path:
-    #         trainer.save_checkpoint(self.exception_ckpt_path)