[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/test_seq2seq_examples.py
+++ b/examples/seq2seq/test_seq2seq_examples.py
@@ -31,6 +31,8 @@ logger = logging.getLogger()
 CUDA_AVAILABLE = torch.cuda.is_available()
 CHEAP_ARGS = {
    "label_smoothing": 0.2,
+    "eval_beams": 1,
+    "val_metric": None,
    "adafactor": True,
    "early_stopping_patience": 2,
    "logger_name": "default",