adding support for interrupt signals

2026-09-09 11:32:07 +08:00 · 2019-06-14 09:24:51 -04:00
parent d98e799404
commit 69274d304d
1 changed files with 4 additions and 0 deletions
@@ -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]