added safeguards for callbacks in loading saving
This commit is contained in:
parent
8fd7a6001b
commit
8a3abec83a
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in New Issue