File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
src/transformers/models/convbert Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -803,12 +803,15 @@ class TFConvBertPreTrainedModel(TFPreTrainedModel):
803
803
vectors than the model's internal embedding lookup matrix.
804
804
output_attentions (:obj:`bool`, `optional`):
805
805
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.
807
808
output_hidden_states (:obj:`bool`, `optional`):
808
809
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.
810
812
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.
812
815
training (:obj:`bool`, `optional`, defaults to :obj:`False`):
813
816
Whether or not to use the model in training mode (some modules like dropout modules have different
814
817
behaviors between training and evaluation).
You can’t perform that action at this time.
0 commit comments