{ "_name_or_path": "google/vit-base-patch16-224-in21k", "architectures": [ "ViTForImageClassification" ], "attention_probs_dropout_prob": 0.0, "encoder_stride": 16, "hidden_act": "gelu", "hidden_dropout_prob": 0.0, "hidden_size": 768, "id2label": { "0": "burger", "1": "butter_naan", "2": "chai", "3": "chapati", "4": "chole_bhature", "5": "dal_makhani", "6": "dhokla", "7": "fried_rice", "8": "idli", "9": "jalebi", "10": "kaathi_rolls", "11": "kadai_paneer", "12": "kulfi", "13": "masala_dosa", "14": "momos", "15": "paani_puri", "16": "pakode", "17": "pav_bhaji", "18": "pizza", "19": "samosa" }, "image_size": 224, "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { "burger": 0, "butter_naan": 1, "chai": 2, "chapati": 3, "chole_bhature": 4, "dal_makhani": 5, "dhokla": 6, "fried_rice": 7, "idli": 8, "jalebi": 9, "kaathi_rolls": 10, "kadai_paneer": 11, "kulfi": 12, "masala_dosa": 13, "momos": 14, "paani_puri": 15, "pakode": 16, "pav_bhaji": 17, "pizza": 18, "samosa": 19 }, "layer_norm_eps": 1e-12, "model_type": "vit", "num_attention_heads": 12, "num_channels": 3, "num_hidden_layers": 12, "patch_size": 16, "problem_type": "single_label_classification", "qkv_bias": true, "torch_dtype": "float32", "transformers_version": "4.46.3" }