candy-first / config.json
ajanco
commit files to HF hub
9e3c26f
raw
history blame
1.96 kB
{
"_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": "airheads",
"1": "candy corn",
"10": "m&ms",
"11": "marshmallow",
"12": "mentos",
"13": "mint",
"14": "nerds",
"15": "peeps",
"16": "pez",
"17": "popcorn",
"18": "pretzel",
"19": "reeses",
"2": "caramel",
"20": "seeds",
"21": "skittles",
"22": "snickers",
"23": "soda",
"24": "sour",
"25": "swedish fish",
"26": "taffy",
"27": "tootsie",
"28": "twix",
"29": "twizzlers",
"3": "chips",
"30": "warheads",
"31": "whoppers",
"4": "chocolate",
"5": "fruit",
"6": "gum",
"7": "haribo",
"8": "jelly beans",
"9": "lollipop"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"airheads": "0",
"candy corn": "1",
"caramel": "2",
"chips": "3",
"chocolate": "4",
"fruit": "5",
"gum": "6",
"haribo": "7",
"jelly beans": "8",
"lollipop": "9",
"m&ms": "10",
"marshmallow": "11",
"mentos": "12",
"mint": "13",
"nerds": "14",
"peeps": "15",
"pez": "16",
"popcorn": "17",
"pretzel": "18",
"reeses": "19",
"seeds": "20",
"skittles": "21",
"snickers": "22",
"soda": "23",
"sour": "24",
"swedish fish": "25",
"taffy": "26",
"tootsie": "27",
"twix": "28",
"twizzlers": "29",
"warheads": "30",
"whoppers": "31"
},
"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.20.1"
}