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

fix a bug where 'id2label' was incorrectly written as 'i2label' when reading the config from pretrained config

fix a typo bug where 'id2label' was incorrectly written as 'i2label' when reading config (#34637)
fix a bug where 'id2label' was incorrectly written as 'i2label' when reading the config from pretrained config
c772d4d9 · ZuoChen_BUPT · GitHub · eb0ab3ed · c772d4d9
Unverified Commit c772d4d9 authored 7 months ago by ZuoChen_BUPT Committed by GitHub 7 months ago
Hide whitespace changes
Inline Side-by-side

Showing

with 1 addition and 1 deletion
+1 -1
--- a/examples/pytorch/image-classification/run_image_classification_no_trainer.py
+++ b/examples/pytorch/image-classification/run_image_classification_no_trainer.py
@@ -331,7 +331,7 @@ def main():
    config = AutoConfig.from_pretrained(
        args.model_name_or_path,
        num_labels=len(labels),
-        i2label=id2label,
+        id2label=id2label,
        label2id=label2id,
        finetuning_task="image-classification",
        trust_remote_code=args.trust_remote_code,