From 57edb08bd8a73c01e326e4bc259f2ece41acc892 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 26 Jul 2019 22:28:09 -0400 Subject: [PATCH] added clean slurm save load test --- pytorch_lightning/root_module/model_saving.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)