From 82d63a9677d58595b080a7ff9a5acd48aff2287e Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 7 Aug 2019 07:25:20 -0400 Subject: [PATCH] fixed none name --- pytorch_lightning/models/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index fc1fb8f5..4d72618b 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -257,7 +257,7 @@ class Trainer(TrainerIO): for name in checkpoints: if '.ckpt' in name: epoch = name.split('epoch_')[1] - epoch = re.sub('[^0-9]', '' ,epoch) + epoch = int(re.sub('[^0-9]', '' ,epoch)) if epoch > last_epoch: last_epoch = epoch