Unverified Commit 380280d9 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `BigBirdForMaskedLM` doctest (#23369)



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 96ae83a0
......@@ -2453,7 +2453,7 @@ class BigBirdForMaskedLM(BigBirdPreTrainedModel):
>>> labels = torch.where(inputs.input_ids == tokenizer.mask_token_id, labels, -100)
>>> outputs = model(**inputs, labels=labels)
>>> round(outputs.loss.item(), 2)
1.08
1.99
```
"""
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
......
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