SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update modeling_bert.py (#6897)

Browse Source 
outptus -> outputs in example of BertForPreTraining
This commit is contained in:
Parthe Pandit
2020-09-02 03:39:01 -07:00
committed by GitHub
parent e3c55ceb8d
commit 7cb0572c64
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_bert.py
View File

@@ -909,7 +909,7 @@ class BertForPreTraining(BertPreTrainedModel):
>>> inputs = tokenizer("Hello, my dog is cute", return_tensors="pt") >>> inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
>>> outputs = model(**inputs) >>> outputs = model(**inputs)
>>> prediction_logits = outptus.prediction_logits >>> prediction_logits = outputs.prediction_logits
>>> seq_relationship_logits = outputs.seq_relationship_logits >>> seq_relationship_logits = outputs.seq_relationship_logits
""" """
if "masked_lm_labels" in kwargs: if "masked_lm_labels" in kwargs: