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)