From 390d95f6d55abdbc152f2af61d5b9e48ea535b48 Mon Sep 17 00:00:00 2001 From: Steffy-zxf <48793257+Steffy-zxf@users.noreply.github.com> Date: Thu, 4 Jul 2019 16:41:06 +0800 Subject: [PATCH] Update classifier.py --- demo/qa_classification/classifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/qa_classification/classifier.py b/demo/qa_classification/classifier.py index 1c7427b2..6e7ff013 100644 --- a/demo/qa_classification/classifier.py +++ b/demo/qa_classification/classifier.py @@ -38,7 +38,7 @@ args = parser.parse_args() if __name__ == '__main__': # Load Paddlehub ERNIE pretrained model module = hub.Module(name="ernie") - # module = hub.Module(name="bert_multi_cased_L-12_H-768_A-12") + # module = hub.Module(name="bert_chinese_L-12_H-768_A-12") inputs, outputs, program = module.context( trainable=True, max_seq_len=args.max_seq_len) -- GitLab