Fix TrainingArguments help section (#18232)

43a5375c · Sylvain Gugger · GitHub · 9f787ce8 · 43a5375c
Unverified Commit 43a5375c authored Jul 21, 2022 by Sylvain Gugger Committed by GitHub Jul 21, 2022
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/transformers/training_args.py src/transformers/training_args.py +2 -2

No files found.
--- a/src/transformers/training_args.py
+++ b/src/transformers/training_args.py
@@ -803,12 +803,12 @@ class TrainingArguments:
            )
        },
    )
-    fsdp_transformer_layer_cls_to_wrap: str = field(
+    fsdp_transformer_layer_cls_to_wrap: Optional[str] = field(
        default=None,
        metadata={
            "help": (
                "Transformer layer class name (case-sensitive) to wrap ,e.g, `BertLayer`, `GPTJBlock`, `T5Block` .... "
-                "(useful only when `fsdp` flag is passed).",
+                "(useful only when `fsdp` flag is passed)."
            )
        },
    )