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