From 51adb97cd644a5840d971868d18c1d436fd6ff5d Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Tue, 15 Dec 2020 17:14:37 -0500 Subject: [PATCH] Fix fp16_backend field --- src/transformers/training_args.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/training_args.py b/src/transformers/training_args.py index 60d80731ed..0e11b9e168 100644 --- 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):