diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index f9cd7f7f..9838ae55 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -130,7 +130,7 @@ class TrainerIO(object): def max_ckpt_in_folder(self, path): files = os.listdir(path) - files = [x for x if 'ckpt_' in x] + files = [x for x in files if 'ckpt_' in x] if len(files) == 0: return 0