test_triage / config.json
arunboss's picture
Training in progress, epoch 0
922859a
raw
history blame
No virus
4.56 kB
{
"_name_or_path": "microsoft/swin-tiny-patch4-window7-224",
"architectures": [
"SwinForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"depths": [
2,
2,
6,
2
],
"drop_path_rate": 0.1,
"embed_dim": 96,
"encoder_stride": 32,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "1",
"1": "100",
"2": "102",
"3": "104",
"4": "105",
"5": "108",
"6": "109",
"7": "113",
"8": "115",
"9": "117",
"10": "118",
"11": "119",
"12": "121",
"13": "122",
"14": "124",
"15": "126",
"16": "128",
"17": "129",
"18": "131",
"19": "132",
"20": "133",
"21": "136",
"22": "14",
"23": "142",
"24": "143",
"25": "148",
"26": "15",
"27": "153",
"28": "154",
"29": "155",
"30": "157",
"31": "158",
"32": "159",
"33": "16",
"34": "172",
"35": "173",
"36": "18",
"37": "181",
"38": "182",
"39": "183",
"40": "184",
"41": "189",
"42": "19",
"43": "20",
"44": "201",
"45": "202",
"46": "213",
"47": "216",
"48": "219",
"49": "22",
"50": "225",
"51": "23",
"52": "233",
"53": "234",
"54": "237",
"55": "238",
"56": "239",
"57": "24",
"58": "240",
"59": "242",
"60": "244",
"61": "249",
"62": "250",
"63": "251",
"64": "252",
"65": "253",
"66": "27",
"67": "274",
"68": "275",
"69": "3",
"70": "31",
"71": "32",
"72": "34",
"73": "36",
"74": "37",
"75": "38",
"76": "39",
"77": "41",
"78": "42",
"79": "44",
"80": "45",
"81": "46",
"82": "47",
"83": "50",
"84": "51",
"85": "52",
"86": "53",
"87": "54",
"88": "55",
"89": "57",
"90": "58",
"91": "59",
"92": "60",
"93": "63",
"94": "64",
"95": "66",
"96": "67",
"97": "68",
"98": "70",
"99": "71",
"100": "72",
"101": "73",
"102": "75",
"103": "77",
"104": "78",
"105": "79",
"106": "81",
"107": "82",
"108": "83",
"109": "84",
"110": "86",
"111": "89",
"112": "92",
"113": "93",
"114": "98"
},
"image_size": 224,
"initializer_range": 0.02,
"label2id": {
"1": 0,
"100": 1,
"102": 2,
"104": 3,
"105": 4,
"108": 5,
"109": 6,
"113": 7,
"115": 8,
"117": 9,
"118": 10,
"119": 11,
"121": 12,
"122": 13,
"124": 14,
"126": 15,
"128": 16,
"129": 17,
"131": 18,
"132": 19,
"133": 20,
"136": 21,
"14": 22,
"142": 23,
"143": 24,
"148": 25,
"15": 26,
"153": 27,
"154": 28,
"155": 29,
"157": 30,
"158": 31,
"159": 32,
"16": 33,
"172": 34,
"173": 35,
"18": 36,
"181": 37,
"182": 38,
"183": 39,
"184": 40,
"189": 41,
"19": 42,
"20": 43,
"201": 44,
"202": 45,
"213": 46,
"216": 47,
"219": 48,
"22": 49,
"225": 50,
"23": 51,
"233": 52,
"234": 53,
"237": 54,
"238": 55,
"239": 56,
"24": 57,
"240": 58,
"242": 59,
"244": 60,
"249": 61,
"250": 62,
"251": 63,
"252": 64,
"253": 65,
"27": 66,
"274": 67,
"275": 68,
"3": 69,
"31": 70,
"32": 71,
"34": 72,
"36": 73,
"37": 74,
"38": 75,
"39": 76,
"41": 77,
"42": 78,
"44": 79,
"45": 80,
"46": 81,
"47": 82,
"50": 83,
"51": 84,
"52": 85,
"53": 86,
"54": 87,
"55": 88,
"57": 89,
"58": 90,
"59": 91,
"60": 92,
"63": 93,
"64": 94,
"66": 95,
"67": 96,
"68": 97,
"70": 98,
"71": 99,
"72": 100,
"73": 101,
"75": 102,
"77": 103,
"78": 104,
"79": 105,
"81": 106,
"82": 107,
"83": 108,
"84": 109,
"86": 110,
"89": 111,
"92": 112,
"93": 113,
"98": 114
},
"layer_norm_eps": 1e-05,
"mlp_ratio": 4.0,
"model_type": "swin",
"num_channels": 3,
"num_heads": [
3,
6,
12,
24
],
"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.30.2",
"use_absolute_embeddings": false,
"window_size": 7
}