mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-27 13:12:20 +08:00
updated test
This commit is contained in:
1 parent
0b92fe6cea
commit
df5e10d0aa
1 file changed
+4
@@ -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))
|
||||
|
||||
Reference in new issue
Block a user