added safeguards for callbacks in loading saving

This commit is contained in:
William Falcon 2019-07-24 11:30:14 -04:00
parent 8fd7a6001b
commit 8a3abec83a
1 changed files with 2 additions and 0 deletions

View File

@ -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