riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix TrainingArguments help section (#18232)

Browse Source
This commit is contained in:
Sylvain Gugger 2022-07-21 11:03:25 +02:00 committed by GitHub
parent 9f787ce874
commit 43a5375cc1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/training_args.py
View File

@ -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)."
)
},
)