added clean slurm save load test

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)