segformer-webots-grasp / config.json
badinkajink's picture
End of training
1e46cc5
{
"_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": "dummy0",
"1": "dummy1",
"2": "dummy2",
"3": "dummy3",
"4": "dummy4",
"5": "dummy5",
"6": "dummy6",
"7": "dummy7",
"8": "dummy8",
"9": "dummy9",
"10": "dummy10",
"11": "dummy11",
"12": "dummy12",
"13": "dummy13",
"14": "dummy14",
"15": "dummy15",
"16": "dummy16",
"17": "dummy17",
"18": "dummy18",
"19": "dummy19",
"20": "dummy20",
"21": "dummy21",
"22": "dummy22",
"23": "dummy23",
"24": "dummy24",
"25": "dummy25",
"26": "dummy26",
"27": "dummy27",
"28": "dummy28",
"29": "dummy29",
"30": "dummy30",
"31": "dummy31",
"32": "dummy32",
"33": "dummy33",
"34": "dummy34",
"35": "dummy35",
"36": "dummy36",
"37": "dummy37",
"38": "dummy38",
"39": "dummy39",
"40": "dummy40",
"41": "dummy41",
"42": "dummy42",
"43": "dummy43",
"44": "dummy44",
"45": "dummy45",
"46": "dummy46",
"47": "dummy47",
"48": "dummy48",
"49": "dummy49",
"50": "dummy50",
"51": "dummy51",
"52": "dummy52",
"53": "dummy53",
"54": "dummy54",
"55": "dummy55",
"56": "dummy56",
"57": "dummy57",
"58": "dummy58",
"59": "dummy59",
"60": "dummy60",
"61": "dummy61",
"62": "dummy62",
"63": "dummy63",
"64": "dummy64",
"65": "dummy65",
"66": "dummy66",
"67": "dummy67",
"68": "dummy68",
"69": "dummy69",
"70": "dummy70",
"71": "dummy71",
"72": "dummy72",
"73": "dummy73",
"74": "dummy74",
"75": "dummy75",
"76": "dummy76",
"77": "dummy77",
"78": "dummy78",
"79": "dummy79",
"80": "dummy80",
"81": "dummy81",
"82": "dummy82",
"83": "dummy83",
"84": "dummy84",
"85": "dummy85",
"86": "dummy86",
"87": "dummy87",
"88": "dummy88",
"89": "dummy89",
"90": "dummy90",
"91": "dummy91",
"92": "dummy92",
"93": "dummy93",
"94": "dummy94",
"95": "dummy95",
"96": "dummy96",
"97": "dummy97",
"98": "dummy98",
"99": "dummy99",
"100": "dummy100",
"101": "dummy101",
"102": "dummy102",
"103": "dummy103",
"104": "dummy104",
"105": "dummy105",
"106": "dummy106",
"107": "dummy107",
"108": "dummy108",
"109": "dummy109",
"110": "dummy110",
"111": "dummy111",
"112": "dummy112",
"113": "dummy113",
"114": "dummy114",
"115": "dummy115",
"116": "dummy116",
"117": "dummy117",
"118": "dummy118",
"119": "dummy119",
"120": "dummy120",
"121": "dummy121",
"122": "dummy122",
"123": "dummy123",
"124": "dummy124",
"125": "dummy125",
"126": "dummy126",
"127": "dummy127",
"128": "dummy128",
"129": "dummy129",
"130": "dummy130",
"131": "dummy131",
"132": "dummy132",
"133": "dummy133",
"134": "dummy134",
"135": "dummy135",
"136": "dummy136",
"137": "dummy137",
"138": "dummy138",
"139": "dummy139",
"140": "dummy140",
"141": "dummy141",
"142": "dummy142",
"143": "dummy143",
"144": "dummy144",
"145": "dummy145",
"146": "dummy146",
"147": "dummy147",
"148": "dummy148",
"149": "dummy149",
"150": "dummy150",
"151": "dummy151",
"152": "dummy152",
"153": "dummy153",
"154": "dummy154",
"155": "dummy155",
"156": "dummy156",
"157": "dummy157",
"158": "dummy158",
"159": "dummy159",
"160": "dummy160",
"161": "dummy161",
"162": "dummy162",
"163": "dummy163",
"164": "dummy164",
"165": "dummy165",
"166": "dummy166",
"167": "dummy167",
"168": "dummy168",
"169": "dummy169",
"170": "dummy170",
"171": "dummy171",
"172": "dummy172",
"173": "dummy173",
"174": "dummy174",
"175": "dummy175",
"176": "dummy176",
"177": "dummy177",
"178": "dummy178",
"179": "dummy179",
"180": "dummy180",
"181": "dummy181",
"182": "dummy182",
"183": "dummy183",
"184": "dummy184",
"185": "dummy185",
"186": "dummy186",
"187": "dummy187",
"188": "dummy188",
"189": "dummy189",
"190": "dummy190",
"191": "dummy191",
"192": "dummy192",
"193": "dummy193",
"194": "dummy194",
"195": "dummy195",
"196": "dummy196",
"197": "dummy197",
"198": "dummy198",
"199": "dummy199",
"200": "dummy200",
"201": "dummy201",
"202": "dummy202",
"203": "dummy203",
"204": "dummy204",
"205": "dummy205",
"206": "dummy206",
"207": "dummy207",
"208": "dummy208",
"209": "dummy209",
"210": "dummy210",
"211": "dummy211",
"212": "dummy212",
"213": "dummy213",
"214": "dummy214",
"215": "dummy215",
"216": "dummy216",
"217": "dummy217",
"218": "dummy218",
"219": "dummy219",
"220": "dummy220",
"221": "dummy221",
"222": "dummy222",
"223": "dummy223",
"224": "dummy224"
},
"image_size": 224,
"initializer_range": 0.02,
"label2id": {
"dummy0": 0,
"dummy1": 1,
"dummy10": 10,
"dummy100": 100,
"dummy101": 101,
"dummy102": 102,
"dummy103": 103,
"dummy104": 104,
"dummy105": 105,
"dummy106": 106,
"dummy107": 107,
"dummy108": 108,
"dummy109": 109,
"dummy11": 11,
"dummy110": 110,
"dummy111": 111,
"dummy112": 112,
"dummy113": 113,
"dummy114": 114,
"dummy115": 115,
"dummy116": 116,
"dummy117": 117,
"dummy118": 118,
"dummy119": 119,
"dummy12": 12,
"dummy120": 120,
"dummy121": 121,
"dummy122": 122,
"dummy123": 123,
"dummy124": 124,
"dummy125": 125,
"dummy126": 126,
"dummy127": 127,
"dummy128": 128,
"dummy129": 129,
"dummy13": 13,
"dummy130": 130,
"dummy131": 131,
"dummy132": 132,
"dummy133": 133,
"dummy134": 134,
"dummy135": 135,
"dummy136": 136,
"dummy137": 137,
"dummy138": 138,
"dummy139": 139,
"dummy14": 14,
"dummy140": 140,
"dummy141": 141,
"dummy142": 142,
"dummy143": 143,
"dummy144": 144,
"dummy145": 145,
"dummy146": 146,
"dummy147": 147,
"dummy148": 148,
"dummy149": 149,
"dummy15": 15,
"dummy150": 150,
"dummy151": 151,
"dummy152": 152,
"dummy153": 153,
"dummy154": 154,
"dummy155": 155,
"dummy156": 156,
"dummy157": 157,
"dummy158": 158,
"dummy159": 159,
"dummy16": 16,
"dummy160": 160,
"dummy161": 161,
"dummy162": 162,
"dummy163": 163,
"dummy164": 164,
"dummy165": 165,
"dummy166": 166,
"dummy167": 167,
"dummy168": 168,
"dummy169": 169,
"dummy17": 17,
"dummy170": 170,
"dummy171": 171,
"dummy172": 172,
"dummy173": 173,
"dummy174": 174,
"dummy175": 175,
"dummy176": 176,
"dummy177": 177,
"dummy178": 178,
"dummy179": 179,
"dummy18": 18,
"dummy180": 180,
"dummy181": 181,
"dummy182": 182,
"dummy183": 183,
"dummy184": 184,
"dummy185": 185,
"dummy186": 186,
"dummy187": 187,
"dummy188": 188,
"dummy189": 189,
"dummy19": 19,
"dummy190": 190,
"dummy191": 191,
"dummy192": 192,
"dummy193": 193,
"dummy194": 194,
"dummy195": 195,
"dummy196": 196,
"dummy197": 197,
"dummy198": 198,
"dummy199": 199,
"dummy2": 2,
"dummy20": 20,
"dummy200": 200,
"dummy201": 201,
"dummy202": 202,
"dummy203": 203,
"dummy204": 204,
"dummy205": 205,
"dummy206": 206,
"dummy207": 207,
"dummy208": 208,
"dummy209": 209,
"dummy21": 21,
"dummy210": 210,
"dummy211": 211,
"dummy212": 212,
"dummy213": 213,
"dummy214": 214,
"dummy215": 215,
"dummy216": 216,
"dummy217": 217,
"dummy218": 218,
"dummy219": 219,
"dummy22": 22,
"dummy220": 220,
"dummy221": 221,
"dummy222": 222,
"dummy223": 223,
"dummy224": 224,
"dummy23": 23,
"dummy24": 24,
"dummy25": 25,
"dummy26": 26,
"dummy27": 27,
"dummy28": 28,
"dummy29": 29,
"dummy3": 3,
"dummy30": 30,
"dummy31": 31,
"dummy32": 32,
"dummy33": 33,
"dummy34": 34,
"dummy35": 35,
"dummy36": 36,
"dummy37": 37,
"dummy38": 38,
"dummy39": 39,
"dummy4": 4,
"dummy40": 40,
"dummy41": 41,
"dummy42": 42,
"dummy43": 43,
"dummy44": 44,
"dummy45": 45,
"dummy46": 46,
"dummy47": 47,
"dummy48": 48,
"dummy49": 49,
"dummy5": 5,
"dummy50": 50,
"dummy51": 51,
"dummy52": 52,
"dummy53": 53,
"dummy54": 54,
"dummy55": 55,
"dummy56": 56,
"dummy57": 57,
"dummy58": 58,
"dummy59": 59,
"dummy6": 6,
"dummy60": 60,
"dummy61": 61,
"dummy62": 62,
"dummy63": 63,
"dummy64": 64,
"dummy65": 65,
"dummy66": 66,
"dummy67": 67,
"dummy68": 68,
"dummy69": 69,
"dummy7": 7,
"dummy70": 70,
"dummy71": 71,
"dummy72": 72,
"dummy73": 73,
"dummy74": 74,
"dummy75": 75,
"dummy76": 76,
"dummy77": 77,
"dummy78": 78,
"dummy79": 79,
"dummy8": 8,
"dummy80": 80,
"dummy81": 81,
"dummy82": 82,
"dummy83": 83,
"dummy84": 84,
"dummy85": 85,
"dummy86": 86,
"dummy87": 87,
"dummy88": 88,
"dummy89": 89,
"dummy9": 9,
"dummy90": 90,
"dummy91": 91,
"dummy92": 92,
"dummy93": 93,
"dummy94": 94,
"dummy95": 95,
"dummy96": 96,
"dummy97": 97,
"dummy98": 98,
"dummy99": 99
},
"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.35.2"
}