added clean slurm save load test

2019-07-26 22:26:55 -04:00 · 2019-07-26 22:26:55 -04:00 · ffa7a0dbab
parent b5419fcd8b
commit ffa7a0dbab
1 changed files with 1 additions and 0 deletions
--- a/pytorch_lightning/root_module/model_saving.py
+++ b/pytorch_lightning/root_module/model_saving.py
@ -181,6 +181,7 @@ class TrainerIO(object):
        # call model hook
        model.on_hpc_load(checkpoint)

+        self.model = model

    def max_ckpt_in_folder(self, path):
        files = os.listdir(path)