diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index 4455b945..f9cd7f7f 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -130,6 +130,10 @@ class TrainerIO(object): def max_ckpt_in_folder(self, path): files = os.listdir(path) + files = [x for x if 'ckpt_' in x] + if len(files) == 0: + return 0 + ckpt_vs = [] for name in files: name = name.split('ckpt_')[-1]