indian_food_model / config.json
AdamCod's picture
End of training
492f82e
{
"_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",
"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",
"2": "chai",
"3": "chapati",
"4": "chole_bhature",
"5": "dal_makhani",
"6": "dhokla",
"7": "fried_rice",
"8": "idli",
"9": "jalebi"
},
"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,
"qkv_bias": true,
"torch_dtype": "float32",
"transformers_version": "4.29.2"
}