mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
added clean slurm save load test
This commit is contained in:
@@ -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]
|
||||
|
||||
Reference in New Issue
Block a user