未验证 提交 b9344cd8 编写于 作者: W whs 提交者: GitHub

Merge pull request #876 from wanghaoshuang/fix_params_grad

Fix params grad
......@@ -193,8 +193,8 @@ def ctc_train_net(images, label, args, num_classes):
learning_rate=args.learning_rate, momentum=args.momentum)
_, params_grads = optimizer.minimize(sum_cost)
model_average = fluid.optimizer.ModelAverage(
params_grads,
args.average_window,
params_grads,
min_average_window=args.min_average_window,
max_average_window=args.max_average_window)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册