From a931ded310a20f3acde9a66dc4c9dd3b1f8cd48f Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 7 Aug 2019 07:35:02 -0400 Subject: [PATCH] removed bad hook call --- 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 00b95460..a21f8a10 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -155,6 +155,8 @@ class TrainerIO(object): self.current_epoch = checkpoint['epoch'] # restore the optimizers + import pdb + pdb.set_trace() optimizer_states = checkpoint['optimizer_states'] for optimizer, opt_state in zip(self.optimizers, optimizer_states): optimizer.load_state_dict(opt_state)