Skip to content

GitLab

"vllm/vscode:/vscode.git/clone" did not exist on "29a9e9527c5bd0c3fd8f21a54573ff4d6d4a8ad7"
Unverified Commit f0acf7b6 authored by jheitmann's avatar jheitmann Committed by GitHub
Browse files

Fix id2label assignment in run_classification.py (#28590)

parent 83f9196c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/text-classification/run_classification.py
View file @ f0acf7b6
...@@ -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