diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index ab2ceb07f2..b06225dbaf 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -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