Fix fp16_backend field

2020-12-15 17:14:37 -05:00
parent 1551e2dc6d
commit 51adb97cd6
1 changed files with 1 additions and 1 deletions
--- a/src/transformers/training_args.py
+++ b/src/transformers/training_args.py
@@ -384,7 +384,7 @@ class TrainingArguments:
    )
    fp16_backend: str = field(
        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):