updated test

2026-09-11 12:31:23 +08:00 · 2019-08-07 08:12:54 -04:00
parent 0b92fe6cea
commit df5e10d0aa
1 changed files with 4 additions and 0 deletions
@@ -252,6 +252,10 @@ class Trainer(TrainerIO):
        # find last epoch
        checkpoints = os.listdir(self.checkpoint_callback.filepath)
        for name in checkpoints:
+            # ignore hpc ckpts
+            if 'hpc_' in name:
+                continue
+
            if '.ckpt' in name:
                epoch = name.split('epoch_')[1]
                epoch = int(re.sub('[^0-9]', '' ,epoch))