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

059bb258 · Yusuke Mori · GitHub · eba418ac · 059bb258
Unverified Commit 059bb258 authored Jan 27, 2021 by Yusuke Mori Committed by GitHub Jan 26, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/text-classification/run_glue.py examples/text-classification/run_glue.py +1 -1

No files found.
--- 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()}