提交 9f3d62cf 编写于 作者: X xuezhong

charge default optimizer

上级 b54c993b
......@@ -98,7 +98,7 @@ def parse_args():
"(default: %(default)d)")
parser.add_argument(
'--optim',
default='rprop',
default='adam',
help='optimizer type')
parser.add_argument(
'--trainset',
......
......@@ -75,6 +75,7 @@ def encoder(input_name, para_name, shape, hidden_size, args):
input=input_ids,
size=shape,
dtype='float32',
is_sparse=True,
param_attr=fluid.ParamAttr(name='embedding_para'))
encoder_out = bi_lstm_encoder(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册