From 8cca02d65217fa6387ca41bda18cf79109462c0a Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 14 Jun 2019 09:25:46 -0400 Subject: [PATCH] adding support for interrupt signals --- pytorch_lightning/root_module/model_saving.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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