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

5ee30a94 · Steffy-zxf · wuzewu · 81ef11f8 · 5ee30a94
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

demo/elmo/elmo_finetune.py demo/elmo/elmo_finetune.py +1 -1

未找到文件。
--- a/demo/elmo/elmo_finetune.py
+++ b/demo/elmo/elmo_finetune.py
@@ -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