diff --git a/fluid/ocr_recognition/crnn_ctc_model.py b/fluid/ocr_recognition/crnn_ctc_model.py index 8db4fab55dc3492d43d8d9aaa4e30d0619108c28..df33100e36e25d871db25dd304e87053dfb77145 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)