fix: calling update_lr_schedulers after on_train_batch_end hook

fix issue: #7637

Author: Lucklyric

pytorch_lightning/trainer/training_loop.py

@@ -494,6 +494,9 @@ def run_training_epoch(self):
             if batch_output.signal == -1:
                 break
 
+            # update LR schedulers
+            self.update_lr_schedulers('step')
+
             # hook
             # TODO: add outputs to batches
             self.on_train_batch_end(
@@ -523,8 +526,6 @@ def run_training_epoch(self):
             # -----------------------------------------
             self.save_loggers_on_train_batch_end()
 
-            # update LR schedulers
-            self.update_lr_schedulers('step')
             self.trainer.checkpoint_connector.has_trained = True
 
             self.total_batch_idx += 1