sumeyya's picture
End of training
d407196 verified
{
"_name_or_path": "nvidia/mit-b0",
"architectures": [
"SegformerForSemanticSegmentation"
],
"attention_probs_dropout_prob": 0.0,
"classifier_dropout_prob": 0.1,
"decoder_hidden_size": 256,
"depths": [
2,
2,
2,
2
],
"downsampling_rates": [
1,
4,
8,
16
],
"drop_path_rate": 0.1,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_sizes": [
32,
64,
160,
256
],
"id2label": {
"0": "background",
"1": "candy",
"2": "egg tart",
"3": "french fries",
"4": "chocolate",
"5": "biscuit",
"6": "popcorn",
"7": "pudding",
"8": "ice cream",
"9": "cheese butter",
"10": "cake",
"11": "wine",
"12": "milkshake",
"13": "coffee",
"14": "juice",
"15": "milk",
"16": "tea",
"17": "almond",
"18": "red beans",
"19": "cashew",
"20": "dried cranberries",
"21": "soy",
"22": "walnut",
"23": "peanut",
"24": "egg",
"25": "apple",
"26": "date",
"27": "apricot",
"28": "avocado",
"29": "banana",
"30": "strawberry",
"31": "cherry",
"32": "blueberry",
"33": "raspberry",
"34": "mango",
"35": "olives",
"36": "peach",
"37": "lemon",
"38": "pear",
"39": "fig",
"40": "pineapple",
"41": "grape",
"42": "kiwi",
"43": "melon",
"44": "orange",
"45": "watermelon",
"46": "steak",
"47": "pork",
"48": "chicken duck",
"49": "sausage",
"50": "fried meat",
"51": "lamb",
"52": "sauce",
"53": "crab",
"54": "fish",
"55": "shellfish",
"56": "shrimp",
"57": "soup",
"58": "bread",
"59": "corn",
"60": "hamburg",
"61": "pizza",
"62": " hanamaki baozi",
"63": "wonton dumplings",
"64": "pasta",
"65": "noodles",
"66": "rice",
"67": "pie",
"68": "tofu",
"69": "eggplant",
"70": "potato",
"71": "garlic",
"72": "cauliflower",
"73": "tomato",
"74": "kelp",
"75": "seaweed",
"76": "spring onion",
"77": "rape",
"78": "ginger",
"79": "okra",
"80": "lettuce",
"81": "pumpkin",
"82": "cucumber",
"83": "white radish",
"84": "carrot",
"85": "asparagus",
"86": "bamboo shoots",
"87": "broccoli",
"88": "celery stick",
"89": "cilantro mint",
"90": "snow peas",
"91": " cabbage",
"92": "bean sprouts",
"93": "onion",
"94": "pepper",
"95": "green beans",
"96": "French beans",
"97": "king oyster mushroom",
"98": "shiitake",
"99": "enoki mushroom",
"100": "oyster mushroom",
"101": "white button mushroom",
"102": "salad",
"103": "other ingredients"
},
"image_size": 224,
"initializer_range": 0.02,
"label2id": {
" cabbage": 91,
" hanamaki baozi": 62,
"French beans": 96,
"almond": 17,
"apple": 25,
"apricot": 27,
"asparagus": 85,
"avocado": 28,
"background": 0,
"bamboo shoots": 86,
"banana": 29,
"bean sprouts": 92,
"biscuit": 5,
"blueberry": 32,
"bread": 58,
"broccoli": 87,
"cake": 10,
"candy": 1,
"carrot": 84,
"cashew": 19,
"cauliflower": 72,
"celery stick": 88,
"cheese butter": 9,
"cherry": 31,
"chicken duck": 48,
"chocolate": 4,
"cilantro mint": 89,
"coffee": 13,
"corn": 59,
"crab": 53,
"cucumber": 82,
"date": 26,
"dried cranberries": 20,
"egg": 24,
"egg tart": 2,
"eggplant": 69,
"enoki mushroom": 99,
"fig": 39,
"fish": 54,
"french fries": 3,
"fried meat": 50,
"garlic": 71,
"ginger": 78,
"grape": 41,
"green beans": 95,
"hamburg": 60,
"ice cream": 8,
"juice": 14,
"kelp": 74,
"king oyster mushroom": 97,
"kiwi": 42,
"lamb": 51,
"lemon": 37,
"lettuce": 80,
"mango": 34,
"melon": 43,
"milk": 15,
"milkshake": 12,
"noodles": 65,
"okra": 79,
"olives": 35,
"onion": 93,
"orange": 44,
"other ingredients": 103,
"oyster mushroom": 100,
"pasta": 64,
"peach": 36,
"peanut": 23,
"pear": 38,
"pepper": 94,
"pie": 67,
"pineapple": 40,
"pizza": 61,
"popcorn": 6,
"pork": 47,
"potato": 70,
"pudding": 7,
"pumpkin": 81,
"rape": 77,
"raspberry": 33,
"red beans": 18,
"rice": 66,
"salad": 102,
"sauce": 52,
"sausage": 49,
"seaweed": 75,
"shellfish": 55,
"shiitake": 98,
"shrimp": 56,
"snow peas": 90,
"soup": 57,
"soy": 21,
"spring onion": 76,
"steak": 46,
"strawberry": 30,
"tea": 16,
"tofu": 68,
"tomato": 73,
"walnut": 22,
"watermelon": 45,
"white button mushroom": 101,
"white radish": 83,
"wine": 11,
"wonton dumplings": 63
},
"layer_norm_eps": 1e-06,
"mlp_ratios": [
4,
4,
4,
4
],
"model_type": "segformer",
"num_attention_heads": [
1,
2,
5,
8
],
"num_channels": 3,
"num_encoder_blocks": 4,
"patch_sizes": [
7,
3,
3,
3
],
"reshape_last_stage": true,
"semantic_loss_ignore_index": 255,
"sr_ratios": [
8,
4,
2,
1
],
"strides": [
4,
2,
2,
2
],
"torch_dtype": "float32",
"transformers_version": "4.46.3"
}