Merge pull request #7 from ElementAI/fix-4-gradient-checkpointing

fix #4
ServiceNow · Oct 18, 2021 · 675ba18 · 675ba18
2 parents 94b14b3 + a8b9e49
commit 675ba18
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/seq2seq/run_seq2seq.py b/seq2seq/run_seq2seq.py
@@ -122,8 +122,8 @@ def main() -> None:
         num_beams=data_training_args.num_beams,
         num_beam_groups=data_training_args.num_beam_groups,
         diversity_penalty=data_training_args.diversity_penalty,
-        gradient_checkpointing=model_args.gradient_checkpointing,
-        use_cache=not model_args.gradient_checkpointing,
+        gradient_checkpointing=training_args.gradient_checkpointing,
+        use_cache=not training_args.gradient_checkpointing,
     )
 
     # Initialize tokenizer

diff --git a/seq2seq/utils/args.py b/seq2seq/utils/args.py
@@ -37,7 +37,3 @@ class ModelArguments:
             "with private models)."
         },
     )
-    gradient_checkpointing: bool = field(
-        default=False,
-        metadata={"help": "Will enable gradient / activation checkpointing"},
-    )