wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-29 13:30:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

added clean slurm save load test

Browse Source
This commit is contained in:
William Falcon committed 2019-07-26 22:32:27 -04:00
1 parent 57edb08bd8
commit f1de62671d
1 file changed
-1
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/model_saving.py
-1
View File
@@ -182,7 +182,6 @@ class TrainerIO(object):
# call model hook
model.on_hpc_load(checkpoint)
def max_ckpt_in_folder(self, path):
files = os.listdir(path)
files = [x for x in files if 'ckpt_' in x]