fix typo in gradient_checkpointing arg (#12855)

help for `ModelArguments.gradient_checkpointing` should be
"If True, use gradient checkpointing to save memory
at the expense of slower backward pass."
not "Whether to freeze the feature extractor layers of the model."
(which is duplicated from `freeze_feature_extractor` arg)
This commit is contained in:
21jun
2021-07-30 16:06:33 +09:00
committed by GitHub
parent fd0255b41d
commit 5c673efad7
2 changed files with 8 additions and 2 deletions

View File

@@ -55,7 +55,10 @@ class ModelArguments:
default=True, metadata={"help": "Whether to freeze the feature extractor layers of the model."}
)
gradient_checkpointing: Optional[bool] = field(
default=False, metadata={"help": "Whether to freeze the feature extractor layers of the model."}
default=False,
metadata={
"help": "If True, use gradient checkpointing to save memory at the expense of slower backward pass."
},
)
verbose_logging: Optional[bool] = field(
default=False,

View File

@@ -51,7 +51,10 @@ class ModelArguments:
default=True, metadata={"help": "Whether to freeze the feature extractor layers of the model."}
)
gradient_checkpointing: Optional[bool] = field(
default=False, metadata={"help": "Whether to freeze the feature extractor layers of the model."}
default=False,
metadata={
"help": "If True, use gradient checkpointing to save memory at the expense of slower backward pass."
},
)
verbose_logging: Optional[bool] = field(
default=False,