Unverified Commit 1486fb81 authored by Buddhi Chathuranga Senarathna's avatar Buddhi Chathuranga Senarathna Committed by GitHub
Browse files

Fix typo in the example of MobileBertForPreTraining (#12919)

parent f3d0866e
......@@ -964,7 +964,7 @@ class MobileBertForPreTraining(MobileBertPreTrainedModel):
>>> input_ids = torch.tensor(tokenizer.encode("Hello, my dog is cute", add_special_tokens=True)).unsqueeze(0) # Batch size 1
>>> outputs = model(input_ids)
>>> prediction_logits = outptus.prediction_logits
>>> prediction_logits = outputs.prediction_logits
>>> seq_relationship_logits = outputs.seq_relationship_logits
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment