From df5e10d0aa924881c07a77e461a2cee562d51bf2 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 7 Aug 2019 08:12:54 -0400 Subject: [PATCH] updated test --- pytorch_lightning/models/trainer.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 07dedd53..3f95d045 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -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))