Dhahlan2000's picture
commit files to HF hub
0881abf
{
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "Bellpepper_fresh",
"1": "Bellpepper_intermediate_fresh",
"10": "Potato_intermediate_fresh",
"11": "Potato_rotten",
"12": "Tomato_fresh",
"13": "Tomato_intermediate_fresh",
"14": "Tomato_rotten",
"15": "ripe_apple",
"16": "ripe_banana",
"17": "ripe_mango",
"18": "ripe_oranges",
"19": "ripe_strawberry",
"2": "Bellpepper_rotten",
"20": "rotten_apple",
"21": "rotten_banana",
"22": "rotten_mango",
"23": "rotten_oranges",
"24": "rotten_strawberry",
"25": "unripe_apple",
"26": "unripe_banana",
"27": "unripe_mango",
"28": "unripe_oranges",
"29": "unripe_strawberry",
"3": "Carrot_fresh",
"4": "Carrot_intermediate_fresh",
"5": "Carrot_rotten",
"6": "Cucumber_fresh",
"7": "Cucumber_intermediate_fresh",
"8": "Cucumber_rotten",
"9": "Potato_fresh"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Bellpepper_fresh": "0",
"Bellpepper_intermediate_fresh": "1",
"Bellpepper_rotten": "2",
"Carrot_fresh": "3",
"Carrot_intermediate_fresh": "4",
"Carrot_rotten": "5",
"Cucumber_fresh": "6",
"Cucumber_intermediate_fresh": "7",
"Cucumber_rotten": "8",
"Potato_fresh": "9",
"Potato_intermediate_fresh": "10",
"Potato_rotten": "11",
"Tomato_fresh": "12",
"Tomato_intermediate_fresh": "13",
"Tomato_rotten": "14",
"ripe_apple": "15",
"ripe_banana": "16",
"ripe_mango": "17",
"ripe_oranges": "18",
"ripe_strawberry": "19",
"rotten_apple": "20",
"rotten_banana": "21",
"rotten_mango": "22",
"rotten_oranges": "23",
"rotten_strawberry": "24",
"unripe_apple": "25",
"unripe_banana": "26",
"unripe_mango": "27",
"unripe_oranges": "28",
"unripe_strawberry": "29"
},
"layer_norm_eps": 1e-12,
"model_type": "vit",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"patch_size": 16,
"pooler_act": "tanh",
"pooler_output_size": 768,
"problem_type": "single_label_classification",
"qkv_bias": true,
"torch_dtype": "float32",
"transformers_version": "4.52.3"
}