Add WNLI and RTE to the valid data creation tasks and keep list alphabetical.

PiperOrigin-RevId: 319475767

Add WNLI and RTE to the valid data creation tasks and keep list alphabetical.
PiperOrigin-RevId: 319475767
4701bfbc · Maxim Neumann · A. Unique TensorFlower · 8a19350e · 4701bfbc
Commit 4701bfbc authored Jul 02, 2020 by Maxim Neumann Committed by A. Unique TensorFlower Jul 02, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

official/nlp/data/create_finetuning_data.py official/nlp/data/create_finetuning_data.py +2 -2

No files found.
--- a/official/nlp/data/create_finetuning_data.py
+++ b/official/nlp/data/create_finetuning_data.py
@@ -50,8 +50,8 @@ flags.DEFINE_string(
    "for the task.")

 flags.DEFINE_enum("classification_task_name", "MNLI",
-                  ["COLA", "MNLI", "MRPC", "QNLI", "QQP", "SST-2", "XNLI",
-                   "PAWS-X", "XTREME-XNLI", "XTREME-PAWS-X"],
+                  ["COLA", "MNLI", "MRPC", "PAWS-X", "QNLI", "QQP", "RTE",
+                   "SST-2", "WNLI", "XNLI", "XTREME-XNLI", "XTREME-PAWS-X"],
                  "The name of the task to train BERT classifier. The "
                  "difference between XTREME-XNLI and XNLI is: 1. the format "
                  "of input tsv files; 2. the dev set for XTREME is english "