diff --git a/examples/text-classification/run_glue.py b/examples/text-classification/run_glue.py index 269d0dd03..2ba141185 100644 --- 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()}