提交 87512849 编写于 作者: W wanghaoshuang

Fix argument of ModelAverage in crnn_ctc_model.py

上级 97cfb9de
...@@ -193,8 +193,8 @@ def ctc_train_net(images, label, args, num_classes): ...@@ -193,8 +193,8 @@ def ctc_train_net(images, label, args, num_classes):
learning_rate=args.learning_rate, momentum=args.momentum) learning_rate=args.learning_rate, momentum=args.momentum)
_, params_grads = optimizer.minimize(sum_cost) _, params_grads = optimizer.minimize(sum_cost)
model_average = fluid.optimizer.ModelAverage( model_average = fluid.optimizer.ModelAverage(
params_grads,
args.average_window, args.average_window,
params_grads,
min_average_window=args.min_average_window, min_average_window=args.min_average_window,
max_average_window=args.max_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.
先完成此消息的编辑!
想要评论请 注册