diff --git a/examples/research_projects/wav2vec2/run_asr.py b/examples/research_projects/wav2vec2/run_asr.py index ffb31d7739..f4c2561ccf 100755 --- a/examples/research_projects/wav2vec2/run_asr.py +++ b/examples/research_projects/wav2vec2/run_asr.py @@ -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, diff --git a/examples/research_projects/wav2vec2/run_pretrain.py b/examples/research_projects/wav2vec2/run_pretrain.py index 02dee12953..f0ef04d181 100755 --- a/examples/research_projects/wav2vec2/run_pretrain.py +++ b/examples/research_projects/wav2vec2/run_pretrain.py @@ -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,