[seq2seq] distillation.py accepts trainer arguments (#5865)

2020-07-18 07:43:57 -04:00
parent ba2400189b
commit dad5e12e54
2 changed files with 2 additions and 1 deletions
--- a/examples/seq2seq/distillation.py
+++ b/examples/seq2seq/distillation.py
@@ -446,6 +446,7 @@ def distill_main(args):

 if __name__ == "__main__":
    parser = argparse.ArgumentParser()
+    parser = pl.Trainer.add_argparse_args(parser)
    parser = BartSummarizationDistiller.add_model_specific_args(parser, os.getcwd())
    args = parser.parse_args()