Skip to content

Commit 429fad8

Browse files
committed
Update convbert doc
1 parent a183844 commit 429fad8

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/transformers/models/convbert/modeling_tf_convbert.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -803,12 +803,15 @@ class TFConvBertPreTrainedModel(TFPreTrainedModel):
803803
vectors than the model's internal embedding lookup matrix.
804804
output_attentions (:obj:`bool`, `optional`):
805805
Whether or not to return the attentions tensors of all attention layers. See ``attentions`` under returned
806-
tensors for more detail.
806+
tensors for more detail. This argument can be used only in eager mode, in graph mode the value in the
807+
config will be used instead.
807808
output_hidden_states (:obj:`bool`, `optional`):
808809
Whether or not to return the hidden states of all layers. See ``hidden_states`` under returned tensors for
809-
more detail.
810+
more detail. This argument can be used only in eager mode, in graph mode the value in the config will be
811+
used instead.
810812
return_dict (:obj:`bool`, `optional`):
811-
Whether or not to return a :class:`~transformers.file_utils.ModelOutput` instead of a plain tuple.
813+
Whether or not to return a :class:`~transformers.file_utils.ModelOutput` instead of a plain tuple. This
814+
argument can be used in eager mode, in graph mode the value will always be set to True.
812815
training (:obj:`bool`, `optional`, defaults to :obj:`False`):
813816
Whether or not to use the model in training mode (some modules like dropout modules have different
814817
behaviors between training and evaluation).

0 commit comments

Comments
 (0)