From f2454f95b66e74ed9b1e2f5711c2f476476ee3eb Mon Sep 17 00:00:00 2001 From: ananthsub Date: Fri, 9 Apr 2021 01:58:25 -0700 Subject: [PATCH] Update training_loop.py --- pytorch_lightning/trainer/training_loop.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py index d2749733812b3..08cec5030cec5 100644 --- a/pytorch_lightning/trainer/training_loop.py +++ b/pytorch_lightning/trainer/training_loop.py @@ -491,15 +491,15 @@ def run_training_epoch(self): if (val_loop_called and not should_check_val) or should_train_only: self.trainer.optimizer_connector.update_learning_rates(interval='epoch') - if should_train_only: - self.check_checkpoint_callback(True) - self.check_early_stopping_callback(True) - if should_check_val: self.trainer.validating = True self.trainer.run_evaluation(on_epoch=True) self.trainer.training = True + if should_train_only: + self.check_checkpoint_callback(True) + self.check_early_stopping_callback(True) + # increment the global step once # progress global step according to grads progress self.increment_accumulated_grad_global_step()