Skip to content

Commit ccd3557

Browse files
authored
Simplify transformer.rst code block
1 parent af678fb commit ccd3557

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

docs/model_zoo/transformers.rst

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -378,7 +378,7 @@ PaddleNLP Transformer API在提丰富预训练模型的同时,也降低了用
378378
from paddlenlp.datasets import load_dataset
379379
from paddlenlp.transformers import BertForSequenceClassification, BertTokenizer
380380
381-
train_ds, dev_ds, test_ds = load_dataset("chnsenticorp", splits=["train", "dev", "test"])
381+
train_ds = load_dataset("chnsenticorp", splits=["train"])
382382
383383
model = BertForSequenceClassification.from_pretrained("bert-wwm-chinese", num_classes=len(train_ds.label_list))
384384
@@ -400,7 +400,6 @@ PaddleNLP Transformer API在提丰富预训练模型的同时,也降低了用
400400
for input_ids, token_type_ids, labels in train_data_loader():
401401
logits = model(input_ids, token_type_ids)
402402
loss = criterion(logits, labels)
403-
probs = paddle.nn.functional.softmax(logits, axis=1)
404403
loss.backward()
405404
optimizer.step()
406405
optimizer.clear_grad()

0 commit comments

Comments
 (0)