提交 147d542e 编写于 作者: Z zhangxuefei

fix bug that loads module

上级 4ff057d8
......@@ -37,7 +37,7 @@ args = parser.parse_args()
if __name__ == '__main__':
# Load Paddlehub BERT pretrained model
if args.use_taskid:
module = hub.Module(name="ernie_eng_base.hub_module")
module = hub.Module(name="ernie_v2_eng_base")
inputs, outputs, program = module.context(
trainable=True, max_seq_len=args.max_seq_len)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册