Skip to content

GitLab

"torchvision/vscode:/vscode.git/clone" did not exist on "c991db82abba12e664eeac14c9b643d0f1f1a7df"
Commit bed83905 authored by Maxim Neumann's avatar Maxim Neumann Committed by A. Unique TensorFlower
Browse files

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

PiperOrigin-RevId: 319475767
parent 2c810902
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/nlp/data/create_finetuning_data.py
View file @ bed83905
...@@ -50,8 +50,8 @@ flags.DEFINE_string( ...@@ -50,8 +50,8 @@ flags.DEFINE_string(
"for the task.") "for the task.")
flags.DEFINE_enum("classification_task_name", "MNLI", flags.DEFINE_enum("classification_task_name", "MNLI",
["COLA", "MNLI", "MRPC", "QNLI", "QQP", "SST-2", "XNLI", ["COLA", "MNLI", "MRPC", "PAWS-X", "QNLI", "QQP", "RTE",
"PAWS-X", "XTREME-XNLI", "XTREME-PAWS-X"], "SST-2", "WNLI", "XNLI", "XTREME-XNLI", "XTREME-PAWS-X"],
"The name of the task to train BERT classifier. The " "The name of the task to train BERT classifier. The "
"difference between XTREME-XNLI and XNLI is: 1. the format " "difference between XTREME-XNLI and XNLI is: 1. the format "
"of input tsv files; 2. the dev set for XTREME is english " "of input tsv files; 2. the dev set for XTREME is english "
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment