Skip to content

GitLab

Commit 51adb97c authored by Sylvain Gugger's avatar Sylvain Gugger
Browse files

Fix fp16_backend field

parent 1551e2dc
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/training_args.py
View file @ 51adb97c
...@@ -384,7 +384,7 @@ class TrainingArguments: ...@@ -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):
......
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