diff --git a/demo/qa_classification/classifier.py b/demo/qa_classification/classifier.py index 1c7427b257042e635fec2803c8870bf1f29d31c6..6e7ff013fdbf5bcf42738e7c3977a1e3cf4960b0 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)