未验证 提交 143087b8 编写于 作者: Y yukavio 提交者: GitHub

fix lr (#593)

......@@ -81,7 +81,7 @@ def piecewise_decay(args):
def cosine_decay(args):
step = int(math.ceil(float(args.total_images) / args.batch_size))
learning_rate = paddle.optimizer.lr.CosineAnnealingDecay(
learning_rate=args.lr, T_max=args.num_epochs)
learning_rate=args.lr, T_max=args.num_epochs * step)
optimizer = paddle.optimizer.Momentum(
learning_rate=learning_rate,
momentum=args.momentum_rate,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册
反馈
建议
客服 返回
顶部