From d805c7cb71847bb66bd840909760aa6f19b024c2 Mon Sep 17 00:00:00 2001 From: wangshuai6 Date: Wed, 9 Apr 2025 11:15:09 +0800 Subject: [PATCH] delete callbacks --- src/callbacks/model_checkpoint.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/src/callbacks/model_checkpoint.py b/src/callbacks/model_checkpoint.py index 019454e..1e9b8e6 100644 --- a/src/callbacks/model_checkpoint.py +++ b/src/callbacks/model_checkpoint.py @@ -18,8 +18,4 @@ class CheckpointHook(ModelCheckpoint): pl_module: "pl.LightningModule", 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) \ No newline at end of file + del checkpoint["callbacks"] \ No newline at end of file