sghirardelli's picture
Training in progress epoch 0
f7b64b1
raw
history blame contribute delete
No virus
2.75 kB
{
"_name_or_path": "google/vit-base-patch16-224",
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "apple",
"1": "ball",
"10": "cereal_box",
"11": "coffee_mug",
"12": "comb",
"13": "dry_battery",
"14": "flashlight",
"15": "food_bag",
"16": "food_box",
"17": "food_can",
"18": "food_cup",
"19": "food_jar",
"2": "banana",
"20": "garlic",
"21": "glue_stick",
"22": "greens",
"23": "hand_towel",
"24": "instant_noodles",
"25": "keyboard",
"26": "kleenex",
"27": "lemon",
"28": "lightbulb",
"29": "lime",
"3": "bell_pepper",
"30": "marker",
"31": "mushroom",
"32": "notebook",
"33": "onion",
"34": "orange",
"35": "peach",
"36": "pear",
"37": "pitcher",
"38": "plate",
"39": "pliers",
"4": "binder",
"40": "potato",
"41": "rubber_eraser",
"42": "scissors",
"43": "shampoo",
"44": "soda_can",
"45": "sponge",
"46": "stapler",
"47": "tomato",
"48": "toothbrush",
"49": "toothpaste",
"5": "bowl",
"50": "water_bottle",
"6": "calculator",
"7": "camera",
"8": "cap",
"9": "cell_phone"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"apple": "0",
"ball": "1",
"banana": "2",
"bell_pepper": "3",
"binder": "4",
"bowl": "5",
"calculator": "6",
"camera": "7",
"cap": "8",
"cell_phone": "9",
"cereal_box": "10",
"coffee_mug": "11",
"comb": "12",
"dry_battery": "13",
"flashlight": "14",
"food_bag": "15",
"food_box": "16",
"food_can": "17",
"food_cup": "18",
"food_jar": "19",
"garlic": "20",
"glue_stick": "21",
"greens": "22",
"hand_towel": "23",
"instant_noodles": "24",
"keyboard": "25",
"kleenex": "26",
"lemon": "27",
"lightbulb": "28",
"lime": "29",
"marker": "30",
"mushroom": "31",
"notebook": "32",
"onion": "33",
"orange": "34",
"peach": "35",
"pear": "36",
"pitcher": "37",
"plate": "38",
"pliers": "39",
"potato": "40",
"rubber_eraser": "41",
"scissors": "42",
"shampoo": "43",
"soda_can": "44",
"sponge": "45",
"stapler": "46",
"tomato": "47",
"toothbrush": "48",
"toothpaste": "49",
"water_bottle": "50"
},
"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,
"transformers_version": "4.31.0"
}