From a895bf1b712861ffc806001ab56c9bbfbacbbc8f Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 7 Aug 2019 07:15:23 -0400 Subject: [PATCH] fixed none name --- pytorch_lightning/models/trainer.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index c9c4d850..fc1fb8f5 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -264,9 +264,10 @@ class Trainer(TrainerIO): last_ckpt_name = name # restore last checkpoint - last_ckpt_path = os.path.join(self.checkpoint_callback.filepath, last_ckpt_name) - self.restore(last_ckpt_path, self.on_gpu) - print(f'model and trainer restored from checkpoint: {last_ckpt_path}') + if last_ckpt_name is not None: + last_ckpt_path = os.path.join(self.checkpoint_callback.filepath, last_ckpt_name) + self.restore(last_ckpt_path, self.on_gpu) + print(f'model and trainer restored from checkpoint: {last_ckpt_path}') @property def data_parallel(self):