提交 4a0d8b13 编写于 作者: G gongweibao

fix

上级 f6c71cbf
...@@ -116,7 +116,7 @@ def parse_args(): ...@@ -116,7 +116,7 @@ def parse_args():
parser.add_argument( parser.add_argument(
"--use_mem_opt", "--use_mem_opt",
type=ast.literal_eval, type=ast.literal_eval,
default=True, default=False,
help="The flag indicating whether to use memory optimization.") help="The flag indicating whether to use memory optimization.")
parser.add_argument( parser.add_argument(
"--use_py_reader", "--use_py_reader",
...@@ -622,6 +622,7 @@ def train(args): ...@@ -622,6 +622,7 @@ def train(args):
use_py_reader=args.use_py_reader, use_py_reader=args.use_py_reader,
is_test=False) is_test=False)
optimizer=None
if args.local: if args.local:
lr_decay = fluid.layers.learning_rate_scheduler.noam_decay( lr_decay = fluid.layers.learning_rate_scheduler.noam_decay(
ModelHyperParams.d_model, TrainTaskConfig.warmup_steps) ModelHyperParams.d_model, TrainTaskConfig.warmup_steps)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册