adding support for interrupt signals

2026-09-11 12:31:23 +08:00 · 2019-06-14 09:25:46 -04:00
parent 69274d304d
commit 8cca02d652
1 changed files with 1 additions and 1 deletions
@@ -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