From 5c398d7a4ec9d2d1b6795f82d7e7355fc55230aa Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 7 Aug 2019 07:39:41 -0400 Subject: [PATCH] removed bad hook call --- pytorch_lightning/root_module/model_saving.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index a21f8a10..00b95460 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -155,8 +155,6 @@ 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)