提交 334a4b30 编写于 作者: C chenguowei01

update config.py

上级 455cf5d3
......@@ -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.')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册