updated test

This commit is contained in:
William Falcon 2019-08-07 08:12:54 -04:00
parent 0b92fe6cea
commit df5e10d0aa
1 changed files with 4 additions and 0 deletions

View File

@ -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))