added clean slurm save load test

This commit is contained in:
William Falcon
2019-07-26 22:32:27 -04:00
parent 57edb08bd8
commit f1de62671d
@@ -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]