From 2baa80d62664d0f506f6b49bfa82ddfba4a38932 Mon Sep 17 00:00:00 2001 From: Nic Eggert Date: Sat, 7 Dec 2019 22:22:03 -0600 Subject: [PATCH] Make sure train doesn't crash when called at max_epoch (#608) --- pytorch_lightning/trainer/training_loop.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py index 541a99f1..2431f88c 100644 --- a/pytorch_lightning/trainer/training_loop.py +++ b/pytorch_lightning/trainer/training_loop.py @@ -258,6 +258,7 @@ class TrainerTrainLoopMixin(ABC): pass def train(self): + model = self.get_model() # run all epochs for epoch in range(self.current_epoch, self.max_epochs): # set seed for distributed sampler (enables shuffling for each epoch)