From 4a0d8b139b748f93ff886f98e3ee973e60d1cf68 Mon Sep 17 00:00:00 2001 From: gongweibao Date: Fri, 28 Sep 2018 02:44:31 +0000 Subject: [PATCH] fix --- fluid/neural_machine_translation/transformer/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/fluid/neural_machine_translation/transformer/train.py b/fluid/neural_machine_translation/transformer/train.py index d34d4cc1..08a0488f 100644 --- a/fluid/neural_machine_translation/transformer/train.py +++ b/fluid/neural_machine_translation/transformer/train.py @@ -116,7 +116,7 @@ def parse_args(): parser.add_argument( "--use_mem_opt", type=ast.literal_eval, - default=True, + default=False, help="The flag indicating whether to use memory optimization.") parser.add_argument( "--use_py_reader", @@ -622,6 +622,7 @@ def train(args): use_py_reader=args.use_py_reader, is_test=False) + optimizer=None if args.local: lr_decay = fluid.layers.learning_rate_scheduler.noam_decay( ModelHyperParams.d_model, TrainTaskConfig.warmup_steps) -- GitLab