SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Documentation fixes (#7585)

Browse Source
This commit is contained in:
Sylvain Gugger
2020-10-05 11:01:03 -04:00
committed by GitHub
parent 9cf7b23b9b
commit 03835af700
5 changed files with 39 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/configuration_utils.py
View File

@@ -114,8 +114,8 @@ class PretrainedConfig(object):
model pretrained weights.
- **finetuning_task** (:obj:`str`, `optional`) -- Name of the task used to fine-tune the model. This can be
used when converting from an original (TensorFlow or PyTorch) checkpoint.
- **id2label** (:obj:`List[str]`, `optional`) -- A map from index (for instance prediction index, or target
index) to label.
- **id2label** (:obj:`Dict[int, str]`, `optional`) -- A map from index (for instance prediction index, or
target index) to label.
- **label2id** (:obj:`Dict[str, int]`, `optional`) -- A map from label to index for the model.
- **num_labels** (:obj:`int`, `optional`) -- Number of labels to use in the last layer added to the model,
typically for a classification task.