Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

added safeguards for callbacks in loading saving

This commit is contained in:
William Falcon 2019-07-24 11:31:13 -04:00
parent 8a3abec83a
commit 98c112598e
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch_lightning/root_module/model_saving.py
View File

@ -57,10 +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
checkpoint['checkpoint_callback_best'] = self.checkpoint_callback.best
if self.early_stop_callback is not None:
checkpoint['early_stop_callback_wait'] = self.early_stop_callback.wait