mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-14 20:58:08 +00:00
parent
eba418ac5d
commit
059bb25817
1 changed files with 1 additions and 1 deletions
|
|
@ -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()}
|
||||
|
|
|
|||
Loading…
Reference in a new issue