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

2021-01-27 04:32:19 +09:00
parent eba418ac5d
commit 059bb25817
1 changed files with 1 additions and 1 deletions
--- 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()}