diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index 361a0854..438d67ea 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -176,12 +176,12 @@ class TrainerIO(object): model = self.__get_model() # load the state_dict on the model automatically + pdb.set_trace() model.load_state_dict(checkpoint['state_dict']) # call model hook model.on_hpc_load(checkpoint) - self.model = model def max_ckpt_in_folder(self, path): files = os.listdir(path)