From 87512849952dfd1659c6922633c88290ea135de7 Mon Sep 17 00:00:00 2001 From: wanghaoshuang Date: Tue, 24 Apr 2018 21:13:49 +0800 Subject: [PATCH] Fix argument of ModelAverage in crnn_ctc_model.py --- fluid/ocr_recognition/crnn_ctc_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fluid/ocr_recognition/crnn_ctc_model.py b/fluid/ocr_recognition/crnn_ctc_model.py index dd5aaa3f..53763fa3 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) -- GitLab