From 5c673efad71026ec820c5101349aa0ae8a95b360 Mon Sep 17 00:00:00 2001 From: 21jun <21jun7654@gmail.com> Date: Fri, 30 Jul 2021 16:06:33 +0900 Subject: [PATCH] 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) --- examples/research_projects/wav2vec2/run_asr.py | 5 ++++- examples/research_projects/wav2vec2/run_pretrain.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) 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,