From 77caafffd50e4b9cd605086e67fed9f63147aeba Mon Sep 17 00:00:00 2001 From: kinghuin Date: Thu, 21 Nov 2019 10:17:49 +0800 Subject: [PATCH] modify ernie tiny name --- demo/sequence-labeling/sequence_label.py | 2 +- demo/text-classification/text_classifier.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/demo/sequence-labeling/sequence_label.py b/demo/sequence-labeling/sequence_label.py index 2fe72b5b..673f2ccf 100644 --- a/demo/sequence-labeling/sequence_label.py +++ b/demo/sequence-labeling/sequence_label.py @@ -37,7 +37,7 @@ args = parser.parse_args() if __name__ == '__main__': # Load Paddlehub ERNIE pretrained model - module = hub.Module(name="ernie_v2_chinese_tiny") + module = hub.Module(name="ernie_tiny") inputs, outputs, program = module.context( trainable=True, max_seq_len=args.max_seq_len) if module.name.startswith("ernie_v2"): diff --git a/demo/text-classification/text_classifier.py b/demo/text-classification/text_classifier.py index abeabcdf..9dee887e 100644 --- a/demo/text-classification/text_classifier.py +++ b/demo/text-classification/text_classifier.py @@ -42,7 +42,7 @@ if __name__ == '__main__': # Download dataset and use ClassifyReader to read dataset if args.dataset.lower() == "chnsenticorp": dataset = hub.dataset.ChnSentiCorp() - module = hub.Module(name="ernie_v2_chinese_tiny") + module = hub.Module(name="ernie_tiny") metrics_choices = ["acc"] elif args.dataset.lower() == "tnews": dataset = hub.dataset.TNews() -- GitLab