Fix lr scheduler step error in hapi when use static mode (#28297)

* fix lr scheduler * fix code style
4 years ago · 0351168967
parent 8953038400
commit 0351168967
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/hapi/model.py
+++ b/python/paddle/hapi/model.py
@ -454,7 +454,8 @@ class StaticGraphAdapter(object):
                rets.insert(i, feed[name])

        # step learning rate scheduler on each batch end
-        if self.model._optimizer and \
+        if self.model._optimizer and self.mode == 'train' and \
+                hasattr(self.model._optimizer, '_learning_rate') and \
                isinstance(self.model._optimizer._learning_rate,
                           paddle.optimizer.lr.LRScheduler):
            self.model._optimizer._learning_rate.step()