From be55065462b4dd505dfb92c72ee455f797aa2d86 Mon Sep 17 00:00:00 2001 From: Steffy-zxf Date: Tue, 12 May 2020 17:23:26 +0800 Subject: [PATCH] fix typo --- demo/text_classification/predict.py | 2 +- demo/text_classification/text_classifier.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/demo/text_classification/predict.py b/demo/text_classification/predict.py index 732df892..e53cf2b8 100644 --- a/demo/text_classification/predict.py +++ b/demo/text_classification/predict.py @@ -40,7 +40,7 @@ args = parser.parse_args() if __name__ == '__main__': # Load Paddlehub ERNIE Tiny pretrained model - module = hub.Module(name="ernie_v2_eng_base") + module = hub.Module(name="ernie_tiny") inputs, outputs, program = module.context( trainable=True, max_seq_len=args.max_seq_len) diff --git a/demo/text_classification/text_classifier.py b/demo/text_classification/text_classifier.py index f987953a..23746c03 100644 --- a/demo/text_classification/text_classifier.py +++ b/demo/text_classification/text_classifier.py @@ -36,7 +36,7 @@ args = parser.parse_args() if __name__ == '__main__': # Load Paddlehub ERNIE Tiny pretrained model - module = hub.Module(name="ernie_v2_eng_base") + module = hub.Module(name="ernie_tiny") inputs, outputs, program = module.context( trainable=True, max_seq_len=args.max_seq_len) -- GitLab