Unverified Commit f0acf7b6 authored by jheitmann's avatar jheitmann Committed by GitHub
Browse files

Fix id2label assignment in run_classification.py (#28590)

parent 83f9196c
......@@ -545,7 +545,7 @@ def main():
"run. You can ignore this if you are doing finetuning."
)
model.config.label2id = label_to_id
model.config.id2label = {id: label for label, id in config.label2id.items()}
model.config.id2label = {id: label for label, id in label_to_id.items()}
elif not is_regression: # classification, but not training
logger.info("using label infos in the model config")
logger.info("label2id: {}".format(model.config.label2id))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment