step lr_scheduler on epoch end in hapi/model.py fit. test=develop

PaddlePaddle · Sep 29, 2020 · fc402e0 · fc402e0
1 parent 719309a
commit fc402e0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/paddle/hapi/model.py b/python/paddle/hapi/model.py
@@ -1436,6 +1436,9 @@ def fit(
 
                 cbks.on_end('eval', eval_logs)
 
+            # step learning rate scheduler on each epcoh end
+            self._optimizer._learning_rate.step()
+
         cbks.on_end('train', logs)
         self._test_dataloader = None