Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

added clean slurm save load test

This commit is contained in:
William Falcon 2019-07-26 22:26:55 -04:00
parent b5419fcd8b
commit ffa7a0dbab
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pytorch_lightning/root_module/model_saving.py
View File

@ -181,6 +181,7 @@ class TrainerIO(object):
# call model hook # call model hook
model.on_hpc_load(checkpoint) model.on_hpc_load(checkpoint)
self.model = model
def max_ckpt_in_folder(self, path): def max_ckpt_in_folder(self, path):
files = os.listdir(path) files = os.listdir(path)