From c4336ecbbdbb8bddcbfb31e0fcc4d382b430a9a5 Mon Sep 17 00:00:00 2001
From: VictorSanh <victorsanh@gmail.com>
Date: Tue, 29 Oct 2019 12:04:20 -0400
Subject: [PATCH] xnli - output_mode consistency

---
 examples/run_xnli.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/run_xnli.py b/examples/run_xnli.py
index ee37296832..a9b2e46c13 100644
--- a/examples/run_xnli.py
+++ b/examples/run_xnli.py
@@ -247,8 +247,8 @@ def evaluate(args, model, tokenizer, prefix=""):
         eval_loss = eval_loss / nb_eval_steps
         if args.output_mode == "classification":
             preds = np.argmax(preds, axis=1)
-        elif args.output_mode == "regression":
-            preds = np.squeeze(preds)
+        else:
+            raise ValueError(f'No other `output_mode` for XNLI.')
         result = compute_metrics(eval_task, preds, out_label_ids)
         results.update(result)