roxy_donut_ring / config.json
kentaromiz's picture
commit files to HF hub
19cc132
{
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"dtype": "float32",
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "1- Frosted",
"1": "10- Chocoramel",
"10": "5- Oreoverload",
"11": "6- Peanut Butter Chocolate",
"12": "7- Signature Chocolate",
"13": "8- Caramel Almond",
"14": "9- Crunch Me Matcha",
"2": "11- BerryCocoa",
"3": "12- Coffee Brulee",
"4": "13- Creme Brulee",
"5": "14- Cheesy Granola",
"6": "15- The Cheese Royale",
"7": "2- Nuttin but Pistachio",
"8": "3- Cheescake",
"9": "4- Burnt Cheese Ring"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"1- Frosted": "0",
"10- Chocoramel": "1",
"11- BerryCocoa": "2",
"12- Coffee Brulee": "3",
"13- Creme Brulee": "4",
"14- Cheesy Granola": "5",
"15- The Cheese Royale": "6",
"2- Nuttin but Pistachio": "7",
"3- Cheescake": "8",
"4- Burnt Cheese Ring": "9",
"5- Oreoverload": "10",
"6- Peanut Butter Chocolate": "11",
"7- Signature Chocolate": "12",
"8- Caramel Almond": "13",
"9- Crunch Me Matcha": "14"
},
"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,
"transformers_version": "4.57.6"
}