From f5a01edfb8112e4c10e4aa6d2a8683e6a0837ea7 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 26 Jul 2019 22:32:34 -0400 Subject: [PATCH] added clean slurm save load test --- pytorch_lightning/root_module/model_saving.py | 1 - 1 file changed, 1 deletion(-) diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index 0638f29421..c5831317e5 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -176,7 +176,6 @@ 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