mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
added safeguards for callbacks in loading saving
This commit is contained in:
@@ -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