diff --git a/demo/multi-label-classification/multi_label_classifier.py b/demo/multi-label-classification/multi_label_classifier.py index f76facb00ff9b01e93857d38d23ff858709845ad..e11cc0fcd5d874cd03878ae4a50293b78f12143f 100644 --- a/demo/multi-label-classification/multi_label_classifier.py +++ b/demo/multi-label-classification/multi_label_classifier.py @@ -34,7 +34,7 @@ args = parser.parse_args() # yapf: enable. if __name__ == '__main__': - # Step1: load Paddlehub ERNIE pretrained model + # Step1: load Paddlehub BERT pretrained model module = hub.Module(name="bert_uncased_L-12_H-768_A-12") inputs, outputs, program = module.context(