diff --git a/dygraph/paddleseg/utils/config.py b/dygraph/paddleseg/utils/config.py index 4823aac135936e072b33a7463da26f37e8a247b4..49ac23b493a7f6d3f81d3d3dbb730e83211e0d8c 100644 --- a/dygraph/paddleseg/utils/config.py +++ b/dygraph/paddleseg/utils/config.py @@ -130,6 +130,7 @@ class Config(object): lr = self._learning_rate args = self.decay_args args.setdefault('decay_steps', self.iters) + args.setdefault('end_lr', 0) return paddle.optimizer.PolynomialLR(lr, **args) else: raise RuntimeError('Only poly decay support.')