riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 11:11:05 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

[docstring] misc arg doc corrections (#6932)

Browse Source 
* correct bool types

fix docstring s/int/bool/

* fix description

* fix num_labels to match reality
This commit is contained in:
Stas Bekman 2020-09-04 07:09:42 -07:00 committed by GitHub
parent e3990d137a
commit c5d43a872f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/transformers/configuration_bart.py
View File

@ -65,7 +65,7 @@ BART_CONFIG_ARGS_DOC = r"""
Typically set this to something large just in case (e.g., 512 or 1024 or 2048). Typically set this to something large just in case (e.g., 512 or 1024 or 2048).
init_std (:obj:`float`, optional, defaults to 0.02): init_std (:obj:`float`, optional, defaults to 0.02):
The standard deviation of the truncated_normal_initializer for initializing all weight matrices. The standard deviation of the truncated_normal_initializer for initializing all weight matrices.
add_bias_logits (:obj:`int`, optional, defaults to False): add_bias_logits (:obj:`bool`, optional, defaults to False):
True for marian only. True for marian only.
normalize_before (:obj:`bool`, optional, defaults to False): normalize_before (:obj:`bool`, optional, defaults to False):
Call layernorm before attention ops. True for pegasus, mbart. False for bart. FIXME: marian? Call layernorm before attention ops. True for pegasus, mbart. False for bart. FIXME: marian?
@ -89,10 +89,10 @@ BART_CONFIG_ARGS_DOC = r"""
Google "layerdrop arxiv", as its not explainable in one line. Google "layerdrop arxiv", as its not explainable in one line.
extra_pos_embeddings: (:obj:`int`, optional, defaults to 2): extra_pos_embeddings: (:obj:`int`, optional, defaults to 2):
How many extra learned positional embeddings to use. Should be pad_token_id+1 for bart. How many extra learned positional embeddings to use. Should be pad_token_id+1 for bart.
num_labels: (:obj:`int`, optional, defaults to 2): num_labels: (:obj:`int`, optional, defaults to 3):
for SequenceClassification for SequenceClassification
is_encoder_decoder (:obj:`int`, optional, defaults to True): is_encoder_decoder (:obj:`bool`, optional, defaults to True):
True Whether this is an encoder/decoder model
force_bos_token_to_be_generated (:obj:`bool`, `optional`, defaults to :obj:`False`): force_bos_token_to_be_generated (:obj:`bool`, `optional`, defaults to :obj:`False`):
Whether or not to force BOS token to be generated at step 1 (after ``decoder_start_token_id``), only true for `bart-large-cnn`. Whether or not to force BOS token to be generated at step 1 (after ``decoder_start_token_id``), only true for `bart-large-cnn`.