PedroSampaio's picture
Training in progress, epoch 0
99f90c0
raw
history blame
5.77 kB
{
"_name_or_path": "microsoft/swin-base-patch4-window7-224",
"architectures": [
"SwinForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"depths": [
2,
2,
18,
2
],
"drop_path_rate": 0.1,
"embed_dim": 128,
"encoder_stride": 32,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 1024,
"id2label": {
"0": "apple_pie",
"1": "baby_back_ribs",
"2": "baklava",
"3": "beef_carpaccio",
"4": "beef_tartare",
"5": "beet_salad",
"6": "beignets",
"7": "bibimbap",
"8": "bread_pudding",
"9": "breakfast_burrito",
"10": "bruschetta",
"11": "caesar_salad",
"12": "cannoli",
"13": "caprese_salad",
"14": "carrot_cake",
"15": "ceviche",
"16": "cheesecake",
"17": "cheese_plate",
"18": "chicken_curry",
"19": "chicken_quesadilla",
"20": "chicken_wings",
"21": "chocolate_cake",
"22": "chocolate_mousse",
"23": "churros",
"24": "clam_chowder",
"25": "club_sandwich",
"26": "crab_cakes",
"27": "creme_brulee",
"28": "croque_madame",
"29": "cup_cakes",
"30": "deviled_eggs",
"31": "donuts",
"32": "dumplings",
"33": "edamame",
"34": "eggs_benedict",
"35": "escargots",
"36": "falafel",
"37": "filet_mignon",
"38": "fish_and_chips",
"39": "foie_gras",
"40": "french_fries",
"41": "french_onion_soup",
"42": "french_toast",
"43": "fried_calamari",
"44": "fried_rice",
"45": "frozen_yogurt",
"46": "garlic_bread",
"47": "gnocchi",
"48": "greek_salad",
"49": "grilled_cheese_sandwich",
"50": "grilled_salmon",
"51": "guacamole",
"52": "gyoza",
"53": "hamburger",
"54": "hot_and_sour_soup",
"55": "hot_dog",
"56": "huevos_rancheros",
"57": "hummus",
"58": "ice_cream",
"59": "lasagna",
"60": "lobster_bisque",
"61": "lobster_roll_sandwich",
"62": "macaroni_and_cheese",
"63": "macarons",
"64": "miso_soup",
"65": "mussels",
"66": "nachos",
"67": "omelette",
"68": "onion_rings",
"69": "oysters",
"70": "pad_thai",
"71": "paella",
"72": "pancakes",
"73": "panna_cotta",
"74": "peking_duck",
"75": "pho",
"76": "pizza",
"77": "pork_chop",
"78": "poutine",
"79": "prime_rib",
"80": "pulled_pork_sandwich",
"81": "ramen",
"82": "ravioli",
"83": "red_velvet_cake",
"84": "risotto",
"85": "samosa",
"86": "sashimi",
"87": "scallops",
"88": "seaweed_salad",
"89": "shrimp_and_grits",
"90": "spaghetti_bolognese",
"91": "spaghetti_carbonara",
"92": "spring_rolls",
"93": "steak",
"94": "strawberry_shortcake",
"95": "sushi",
"96": "tacos",
"97": "takoyaki",
"98": "tiramisu",
"99": "tuna_tartare",
"100": "waffles"
},
"image_size": 224,
"initializer_range": 0.02,
"label2id": {
"apple_pie": 0,
"baby_back_ribs": 1,
"baklava": 2,
"beef_carpaccio": 3,
"beef_tartare": 4,
"beet_salad": 5,
"beignets": 6,
"bibimbap": 7,
"bread_pudding": 8,
"breakfast_burrito": 9,
"bruschetta": 10,
"caesar_salad": 11,
"cannoli": 12,
"caprese_salad": 13,
"carrot_cake": 14,
"ceviche": 15,
"cheese_plate": 17,
"cheesecake": 16,
"chicken_curry": 18,
"chicken_quesadilla": 19,
"chicken_wings": 20,
"chocolate_cake": 21,
"chocolate_mousse": 22,
"churros": 23,
"clam_chowder": 24,
"club_sandwich": 25,
"crab_cakes": 26,
"creme_brulee": 27,
"croque_madame": 28,
"cup_cakes": 29,
"deviled_eggs": 30,
"donuts": 31,
"dumplings": 32,
"edamame": 33,
"eggs_benedict": 34,
"escargots": 35,
"falafel": 36,
"filet_mignon": 37,
"fish_and_chips": 38,
"foie_gras": 39,
"french_fries": 40,
"french_onion_soup": 41,
"french_toast": 42,
"fried_calamari": 43,
"fried_rice": 44,
"frozen_yogurt": 45,
"garlic_bread": 46,
"gnocchi": 47,
"greek_salad": 48,
"grilled_cheese_sandwich": 49,
"grilled_salmon": 50,
"guacamole": 51,
"gyoza": 52,
"hamburger": 53,
"hot_and_sour_soup": 54,
"hot_dog": 55,
"huevos_rancheros": 56,
"hummus": 57,
"ice_cream": 58,
"lasagna": 59,
"lobster_bisque": 60,
"lobster_roll_sandwich": 61,
"macaroni_and_cheese": 62,
"macarons": 63,
"miso_soup": 64,
"mussels": 65,
"nachos": 66,
"omelette": 67,
"onion_rings": 68,
"oysters": 69,
"pad_thai": 70,
"paella": 71,
"pancakes": 72,
"panna_cotta": 73,
"peking_duck": 74,
"pho": 75,
"pizza": 76,
"pork_chop": 77,
"poutine": 78,
"prime_rib": 79,
"pulled_pork_sandwich": 80,
"ramen": 81,
"ravioli": 82,
"red_velvet_cake": 83,
"risotto": 84,
"samosa": 85,
"sashimi": 86,
"scallops": 87,
"seaweed_salad": 88,
"shrimp_and_grits": 89,
"spaghetti_bolognese": 90,
"spaghetti_carbonara": 91,
"spring_rolls": 92,
"steak": 93,
"strawberry_shortcake": 94,
"sushi": 95,
"tacos": 96,
"takoyaki": 97,
"tiramisu": 98,
"tuna_tartare": 99,
"waffles": 100
},
"layer_norm_eps": 1e-05,
"mlp_ratio": 4.0,
"model_type": "swin",
"num_channels": 3,
"num_heads": [
4,
8,
16,
32
],
"num_layers": 4,
"out_features": [
"stage4"
],
"out_indices": [
4
],
"patch_size": 4,
"path_norm": true,
"problem_type": "single_label_classification",
"qkv_bias": true,
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4"
],
"torch_dtype": "float32",
"transformers_version": "4.34.1",
"use_absolute_embeddings": false,
"window_size": 7
}