From 147d542ec00ba2b945d18cf00a4a4a4c854420a4 Mon Sep 17 00:00:00 2001
From: zhangxuefei <zhangxuefei@baidu.com>
Date: Wed, 18 Sep 2019 15:50:38 +0800
Subject: [PATCH] fix bug that loads module

---
 demo/multi-label-classification/multi_label_classifier.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demo/multi-label-classification/multi_label_classifier.py b/demo/multi-label-classification/multi_label_classifier.py
index b1e7086f..a3e38cb9 100644
--- a/demo/multi-label-classification/multi_label_classifier.py
+++ b/demo/multi-label-classification/multi_label_classifier.py
@@ -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)
-- 
GitLab