Fix a bug in run_glue.py (huggingface#9812) (huggingface#9815)

pnf · Jan 26, 2021 · 059bb25 · 059bb25
1 parent eba418a
commit 059bb25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/text-classification/run_glue.py b/examples/text-classification/run_glue.py
@@ -317,7 +317,7 @@ def main():
     if (
         model.config.label2id != PretrainedConfig(num_labels=num_labels).label2id
         and data_args.task_name is not None
-        and is_regression
+        and not is_regression
     ):
         # Some have all caps in their config, some don't.
         label_name_to_id = {k.lower(): v for k, v in model.config.label2id.items()}