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(): ...@@ -545,7 +545,7 @@ def main():
"run. You can ignore this if you are doing finetuning." "run. You can ignore this if you are doing finetuning."
) )
model.config.label2id = label_to_id 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 elif not is_regression: # classification, but not training
logger.info("using label infos in the model config") logger.info("using label infos in the model config")
logger.info("label2id: {}".format(model.config.label2id)) 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