updated test
This commit is contained in:
parent
0b92fe6cea
commit
df5e10d0aa
|
@ -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))
|
||||
|
|
Loading…
Reference in New Issue