From 8a3abec83a8ee78ee4e667c7d295ab339cf37d71 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 11:30:14 -0400 Subject: [PATCH] added safeguards for callbacks in loading saving --- pytorch_lightning/root_module/model_saving.py | 2 ++ 1 file changed, 2 insertions(+) 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