[s2s] Adafactor support for builtin trainer (#7522)

2020-10-01 17:27:45 -04:00
parent d3a9601a11
commit de4d7b004a
3 changed files with 40 additions and 0 deletions
--- a/examples/seq2seq/test_finetune_trainer.py
+++ b/examples/seq2seq/test_finetune_trainer.py
@@ -91,6 +91,7 @@ def run_trainer(eval_steps: int, max_len: str, model_name: str, num_train_epochs
        "0.1",
        # "--eval_beams",
        # "2",
+        "--adafactor",
        "--task",
        "translation",
        "--tgt_lang",