提交 5ee30a94 编写于 作者: S Steffy-zxf 提交者: wuzewu

Update the elmo_module name (elmo.hub_module->elmo) (#40)

上级 81ef11f8
......@@ -98,7 +98,7 @@ def lstm_net(program, input_feature, hid_dim=128, hid_dim2=96):
if __name__ == '__main__':
# Step1: load Paddlehub elmo pretrained model
module = hub.Module(name="elmo.hub_module")
module = hub.Module(name="elmo")
inputs, outputs, program = module.context(trainable=True)
# Step2: Download dataset and use TextClassificationReader to read dataset
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册