From fb8ae4014286dc1225097a914937a5da23ea23e5 Mon Sep 17 00:00:00 2001 From: wanghaoshuang Date: Thu, 25 Jan 2018 17:43:14 +0800 Subject: [PATCH] Minimized "avg_cost" instead of "cost". --- fluid/ocr/ctc_train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fluid/ocr/ctc_train.py b/fluid/ocr/ctc_train.py index 66949d52..2e7cf5d5 100644 --- a/fluid/ocr/ctc_train.py +++ b/fluid/ocr/ctc_train.py @@ -24,7 +24,7 @@ from utility import add_arguments, print_arguments parser = argparse.ArgumentParser(description=__doc__) add_arg = functools.partial(add_arguments, argparser=parser) # yapf: disable -add_arg('batch_size', int, 2, "Minibatch size.") +add_arg('batch_size', int, 16, "Minibatch size.") add_arg('pass_num', int, 16, "# of training epochs.") add_arg('learning_rate', float, 1.0e-3, "Learning rate.") add_arg('l2', float, 0.0005, "L2 regularizer.") @@ -121,7 +121,7 @@ def train(args, data_reader=dummy_reader): avg_cost = fluid.layers.mean(x=cost) optimizer = fluid.optimizer.Momentum( learning_rate=args.learning_rate, momentum=args.momentum) - opts = optimizer.minimize(cost) + optimizer.minimize(avg_cost) # decoder and evaluator decoded_out = fluid.layers.ctc_greedy_decoder( input=fc_out, blank=num_classes) -- GitLab