diff --git a/fluid/ocr_recognition/crnn_ctc_model.py b/fluid/ocr_recognition/crnn_ctc_model.py index dd5aaa3f94c1e2668ec75d30735640d14ee8ef0e..53763fa34dc05a6c972683ba396becec5294b887 100644 --- a/fluid/ocr_recognition/crnn_ctc_model.py +++ b/fluid/ocr_recognition/crnn_ctc_model.py @@ -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)