Skip to content

GitLab

"...models/efficientnet/test_image_processing_efficientnet.py" did not exist on "29c10a41d04f855c433a6cde7797b325651417d2"
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
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/mobilebert/modeling_mobilebert.py
View file @ 1486fb81
...@@ -964,7 +964,7 @@ class MobileBertForPreTraining(MobileBertPreTrainedModel): ...@@ -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 >>> input_ids = torch.tensor(tokenizer.encode("Hello, my dog is cute", add_special_tokens=True)).unsqueeze(0) # Batch size 1
>>> outputs = model(input_ids) >>> outputs = model(input_ids)
>>> prediction_logits = outptus.prediction_logits >>> prediction_logits = outputs.prediction_logits
>>> seq_relationship_logits = outputs.seq_relationship_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