Logic/DDT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

delete callbacks

Browse Source
This commit is contained in:
wangshuai6
2025-04-09 11:15:09 +08:00
parent 4f60297dfe
commit d805c7cb71
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/callbacks/model_checkpoint.py
View File

@@ -18,8 +18,4 @@ class CheckpointHook(ModelCheckpoint):
pl_module: "pl.LightningModule", pl_module: "pl.LightningModule",
checkpoint: Dict[str, Any] checkpoint: Dict[str, Any]
) -> None: ) -> None:
del checkpoint["callbacks"] 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)