From 98c112598e077596e80b121cc1f1d2569300a888 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 11:31:13 -0400 Subject: [PATCH] added safeguards for callbacks in loading saving --- pytorch_lightning/root_module/model_saving.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index b06225dbaf..818c194748 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