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

Fix fp16_backend field

Browse Source
This commit is contained in:
Sylvain Gugger
2020-12-15 17:14:37 -05:00
parent 1551e2dc6d
commit 51adb97cd6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/training_args.py
View File

@@ -384,7 +384,7 @@ class TrainingArguments:
) )
fp16_backend: str = field( fp16_backend: str = field(
default="auto", default="auto",
metadata={"help": "The backend to be used for mixed precision. Should be one of 'auto', 'amp' or 'apex'."}, metadata={"help": "The backend to be used for mixed precision.", "choices": ["auto", "amp", "apex"]},
) )
def __post_init__(self): def __post_init__(self):