SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix a typo bug where 'id2label' was incorrectly written as 'i2label' when reading config (#34637)

Browse Source 
fix a bug where 'id2label' was incorrectly written as 'i2label' when reading the config from pretrained config
This commit is contained in:
ZuoChen_BUPT
2024-11-18 21:41:48 +08:00
committed by GitHub
parent eb0ab3ed4b
commit c772d4d91e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/image-classification/run_image_classification_no_trainer.py
View File

@@ -331,7 +331,7 @@ def main():
config = AutoConfig.from_pretrained( config = AutoConfig.from_pretrained(
args.model_name_or_path, args.model_name_or_path,
num_labels=len(labels), num_labels=len(labels),
i2label=id2label, id2label=id2label,
label2id=label2id, label2id=label2id,
finetuning_task="image-classification", finetuning_task="image-classification",
trust_remote_code=args.trust_remote_code, trust_remote_code=args.trust_remote_code,