swinv2-plantclef / config.json
n1hal's picture
End of training
e6d0ed5 verified
{
"_name_or_path": "microsoft/swinv2-base-patch4-window16-256",
"architectures": [
"Swinv2ForImageClassification"
],
"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": "1355868",
"1": "1355869",
"10": "1355885",
"11": "1355886",
"12": "1355894",
"13": "1355897",
"14": "1355898",
"15": "1355899",
"16": "1355900",
"17": "1355901",
"18": "1355902",
"19": "1355903",
"2": "1355870",
"20": "1355907",
"21": "1355908",
"22": "1355914",
"23": "1355926",
"24": "1355927",
"25": "1355928",
"26": "1355932",
"27": "1355934",
"28": "1355935",
"29": "1355936",
"3": "1355871",
"30": "1355937",
"31": "1355941",
"32": "1355948",
"33": "1355950",
"34": "1355952",
"35": "1355953",
"36": "1355964",
"37": "1355967",
"38": "1355968",
"39": "1355969",
"4": "1355872",
"40": "1355970",
"41": "1355971",
"42": "1355972",
"43": "1355977",
"44": "1355978",
"45": "1355984",
"46": "1355986",
"47": "1355987",
"48": "1355989",
"49": "1355990",
"5": "1355873",
"50": "1355991",
"51": "1355992",
"52": "1355993",
"53": "1355994",
"54": "1355995",
"55": "1355997",
"56": "1355998",
"57": "1356001",
"58": "1356007",
"59": "1356008",
"6": "1355880",
"60": "1356012",
"61": "1356013",
"62": "1356017",
"63": "1356022",
"64": "1356023",
"65": "1356024",
"66": "1356033",
"67": "1356040",
"68": "1356042",
"69": "1356044",
"7": "1355881",
"70": "1356045",
"71": "1356046",
"72": "1356052",
"73": "1356054",
"74": "1356055",
"75": "1356058",
"76": "1356062",
"77": "1356063",
"78": "1356064",
"79": "1356065",
"8": "1355882",
"80": "1356066",
"81": "1356067",
"82": "1356069",
"83": "1356070",
"84": "1356072",
"85": "1356075",
"86": "1356078",
"87": "1356079",
"88": "1356081",
"89": "1356082",
"9": "1355884",
"90": "1356083",
"91": "1356084",
"92": "1356086",
"93": "1356089",
"94": "1356091",
"95": "1356094",
"96": "1356095",
"97": "1356105",
"98": "1356106",
"99": "1356107"
},
"image_size": 256,
"initializer_range": 0.02,
"label2id": {
"1355868": "0",
"1355869": "1",
"1355870": "2",
"1355871": "3",
"1355872": "4",
"1355873": "5",
"1355880": "6",
"1355881": "7",
"1355882": "8",
"1355884": "9",
"1355885": "10",
"1355886": "11",
"1355894": "12",
"1355897": "13",
"1355898": "14",
"1355899": "15",
"1355900": "16",
"1355901": "17",
"1355902": "18",
"1355903": "19",
"1355907": "20",
"1355908": "21",
"1355914": "22",
"1355926": "23",
"1355927": "24",
"1355928": "25",
"1355932": "26",
"1355934": "27",
"1355935": "28",
"1355936": "29",
"1355937": "30",
"1355941": "31",
"1355948": "32",
"1355950": "33",
"1355952": "34",
"1355953": "35",
"1355964": "36",
"1355967": "37",
"1355968": "38",
"1355969": "39",
"1355970": "40",
"1355971": "41",
"1355972": "42",
"1355977": "43",
"1355978": "44",
"1355984": "45",
"1355986": "46",
"1355987": "47",
"1355989": "48",
"1355990": "49",
"1355991": "50",
"1355992": "51",
"1355993": "52",
"1355994": "53",
"1355995": "54",
"1355997": "55",
"1355998": "56",
"1356001": "57",
"1356007": "58",
"1356008": "59",
"1356012": "60",
"1356013": "61",
"1356017": "62",
"1356022": "63",
"1356023": "64",
"1356024": "65",
"1356033": "66",
"1356040": "67",
"1356042": "68",
"1356044": "69",
"1356045": "70",
"1356046": "71",
"1356052": "72",
"1356054": "73",
"1356055": "74",
"1356058": "75",
"1356062": "76",
"1356063": "77",
"1356064": "78",
"1356065": "79",
"1356066": "80",
"1356067": "81",
"1356069": "82",
"1356070": "83",
"1356072": "84",
"1356075": "85",
"1356078": "86",
"1356079": "87",
"1356081": "88",
"1356082": "89",
"1356083": "90",
"1356084": "91",
"1356086": "92",
"1356089": "93",
"1356091": "94",
"1356094": "95",
"1356095": "96",
"1356105": "97",
"1356106": "98",
"1356107": "99"
},
"layer_norm_eps": 1e-05,
"mlp_ratio": 4.0,
"model_type": "swinv2",
"num_channels": 3,
"num_heads": [
4,
8,
16,
32
],
"num_layers": 4,
"out_features": [
"stage4"
],
"out_indices": [
4
],
"patch_size": 4,
"path_norm": true,
"pretrained_window_sizes": [
0,
0,
0,
0
],
"problem_type": "single_label_classification",
"qkv_bias": true,
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4"
],
"torch_dtype": "float32",
"transformers_version": "4.46.2",
"use_absolute_embeddings": false,
"window_size": 16
}