added clean slurm save load test

2019-07-26 22:32:34 -04:00 · 2019-07-26 22:32:34 -04:00 · f5a01edfb8
parent f1de62671d
commit f5a01edfb8
1 changed files with 0 additions and 1 deletions
--- 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