[s2s] command line args for faster val steps (#6833)

2020-08-31 16:16:10 -04:00
parent 8af1970e45
commit b9772897ec
3 changed files with 10 additions and 3 deletions
--- a/examples/seq2seq/distillation.py
+++ b/examples/seq2seq/distillation.py
@@ -262,7 +262,7 @@ class BartTranslationDistiller(BartSummarizationDistiller):

    mode = "translation"
    metric_names = ["bleu"]
-    val_metric = "bleu"
+    default_val_metric = "bleu"

    def __init__(self, hparams, **kwargs):
        super().__init__(hparams, **kwargs)