mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-04 13:00:36 +08:00
added safeguards for callbacks in loading saving
This commit is contained in:
1 parent
8fd7a6001b
commit
8a3abec83a
1 file changed
+2
@@ -57,6 +57,8 @@ class TrainerIO(object):
|
||||
'global_step': self.global_step
|
||||
}
|
||||
|
||||
from pytorch_lightning.utils.debugging import ForkedPdb
|
||||
ForkedPdb().set_trace()
|
||||
if self.checkpoint_callback is not None:
|
||||
checkpoint['checkpoint_callback_best'] = self.checkpoint_callback_best.best
|
||||
|
||||
|
||||
Reference in new issue
Block a user