diff --git a/demo/ernie-classification/finetune_with_hub.py b/demo/ernie-classification/finetune_with_hub.py index efc1551310f3be248ef6f0a4eacf7bc029b1c5ae..8d088ba352d5e4d7a17ba1b1c5820a4ed7d814d4 100644 --- a/demo/ernie-classification/finetune_with_hub.py +++ b/demo/ernie-classification/finetune_with_hub.py @@ -54,7 +54,7 @@ if __name__ == '__main__': batch_size=args.batch_size, strategy=strategy) - # loading Paddlehub BERT + # loading Paddlehub ERNIE pretrained model module = hub.Module(name="ernie") # Sentence classification dataset reader @@ -76,7 +76,7 @@ if __name__ == '__main__': pooled_output = output_dict["pooled_output"] # Setup feed list for data feeder - # Must feed all the tensor of bert's module need + # Must feed all the tensor of ERNIE's module need feed_list = [ input_dict["input_ids"].name, input_dict["position_ids"].name, input_dict["segment_ids"].name, input_dict["input_mask"].name, diff --git a/demo/ernie-seq-labeling/finetune_with_hub.py b/demo/ernie-seq-labeling/finetune_with_hub.py index ed57a06bd351bb89125f86b72b26657e22ed8d17..8708bdebdb9c9f6b6edf0c553fd05fc0ff4b1143 100644 --- a/demo/ernie-seq-labeling/finetune_with_hub.py +++ b/demo/ernie-seq-labeling/finetune_with_hub.py @@ -55,7 +55,7 @@ if __name__ == '__main__': batch_size=args.batch_size, strategy=strategy) - # loading Paddlehub ERNIE + # loading Paddlehub ERNIE pretrained model module = hub.Module(name="ernie") # Sequence Label dataset reader @@ -79,7 +79,7 @@ if __name__ == '__main__': sequence_output = output_dict["sequence_output"] # Setup feed list for data feeder - # Must feed all the tensor of bert's module need + # Must feed all the tensor of ERNIE's module need feed_list = [ input_dict["input_ids"].name, input_dict["position_ids"].name, input_dict["segment_ids"].name, input_dict["input_mask"].name,