Simplify replacement of pretrained models (#1788)

Author: linjieccc

examples/text_classification/pretrained_models/README.md

@@ -95,13 +95,13 @@ $ python -m paddle.distributed.launch --gpus "0" train.py --device gpu --save_di
 ```python
 # 使用ernie预训练模型
 # ernie-1.0
-model = ppnlp.transformers.ErnieForSequenceClassification.from_pretrained('ernie-1.0',num_classes=2))
-tokenizer = ppnlp.transformers.ErnieTokenizer.from_pretrained('ernie-1.0')
+model = AutoModelForSequenceClassification.from_pretrained('ernie-1.0',num_classes=2))
+tokenizer = AutoTokenizer.from_pretrained('ernie-1.0')
 
 # 使用bert预训练模型
 # bert-base-chinese
-model = ppnlp.transformers.BertForSequenceClassification.from_pretrained('bert-base-chinese', num_class=2)
-tokenizer = ppnlp.transformers.BertTokenizer.from_pretrained('bert-base-chinese')
+model = AutoModelForSequenceClassification.from_pretrained('bert-base-chinese', num_class=2)
+tokenizer = AutoTokenizer.from_pretrained('bert-base-chinese')
 ```
 更多预训练模型，参考[transformers](../../../docs/model_zoo/transformers.rst)
 

examples/text_classification/pretrained_models/train.py

@@ -22,9 +22,9 @@
 import numpy as np
 import paddle
 import paddle.nn.functional as F
-import paddlenlp as ppnlp
 from paddlenlp.data import Stack, Tuple, Pad
 from paddlenlp.datasets import load_dataset
+from paddlenlp.transformers import AutoModelForSequenceClassification, AutoTokenizer
 from paddlenlp.transformers import LinearDecayWithWarmup
 
 from utils import convert_example
@@ -120,9 +120,9 @@ def do_train():
     train_ds, dev_ds, test_ds = load_dataset(
         args.dataset, splits=["train", "dev", "test"])
 
-    model = ppnlp.transformers.ErnieForSequenceClassification.from_pretrained(
+    model = AutoModelForSequenceClassification.from_pretrained(
         'ernie-1.0', num_classes=len(train_ds.label_list))
-    tokenizer = ppnlp.transformers.ErnieTokenizer.from_pretrained('ernie-1.0')
+    tokenizer = AutoTokenizer.from_pretrained('ernie-1.0')
 
     trans_func = partial(
         convert_example,