Skip to content

GitLab

"examples/run_ner.py" did not exist on "3d47a7f8ab8504a98fba8c8ed1397c28e68cca8d"
Commit 520e7f21 authored by Lysandre's avatar Lysandre
Browse files

Correct docstring for xlnet

parent dd288303
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/modeling_xlnet.py
View file @ 520e7f21
...@@ -550,7 +550,7 @@ XLNET_INPUTS_DOCSTRING = r""" ...@@ -550,7 +550,7 @@ XLNET_INPUTS_DOCSTRING = r"""
token_type_ids (:obj:`torch.LongTensor` of shape :obj:`(batch_size, sequence_length)`, `optional`, defaults to :obj:`None`): token_type_ids (:obj:`torch.LongTensor` of shape :obj:`(batch_size, sequence_length)`, `optional`, defaults to :obj:`None`):
Segment token indices to indicate first and second portions of the inputs. Segment token indices to indicate first and second portions of the inputs.
Indices are selected in ``[0, 1]``: ``0`` corresponds to a `sentence A` token, ``1`` Indices are selected in ``[0, 1]``: ``0`` corresponds to a `sentence A` token, ``1``
corresponds to a `sentence B` token corresponds to a `sentence B` token. The classifier token should be represented by a ``2``.
`What are token type IDs? <../glossary.html#token-type-ids>`_ `What are token type IDs? <../glossary.html#token-type-ids>`_
input_mask (:obj:`torch.FloatTensor` of shape :obj:`(batch_size, sequence_length)`, `optional`, defaults to :obj:`None`): input_mask (:obj:`torch.FloatTensor` of shape :obj:`(batch_size, sequence_length)`, `optional`, defaults to :obj:`None`):
......
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