From 69274d304dac98c1277aa0b893a9a4740123427f Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Fri, 14 Jun 2019 09:24:51 -0400
Subject: [PATCH] adding support for interrupt signals

---
 pytorch_lightning/root_module/model_saving.py | 4 ++++
 1 file changed, 4 insertions(+)

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]