Dhyey8's picture
Training in progress, epoch 0
fc2e2fa verified
{
"_name_or_path": "Dhyey8/swin-tiny-patch4-window7-224-finetuned-teeth_dataset",
"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": "10",
"2": "11",
"3": "12",
"4": "13",
"5": "14",
"6": "15",
"7": "16",
"8": "17",
"9": "18",
"10": "19",
"11": "2",
"12": "20",
"13": "21",
"14": "22",
"15": "23",
"16": "24",
"17": "25",
"18": "26",
"19": "27",
"20": "28",
"21": "29",
"22": "3",
"23": "30",
"24": "31",
"25": "32",
"26": "33",
"27": "34",
"28": "35",
"29": "36",
"30": "37",
"31": "38",
"32": "39",
"33": "4",
"34": "40",
"35": "41",
"36": "42",
"37": "43",
"38": "44",
"39": "45",
"40": "46",
"41": "47",
"42": "48",
"43": "49",
"44": "5",
"45": "50",
"46": "51",
"47": "52",
"48": "53",
"49": "54",
"50": "55",
"51": "56",
"52": "57",
"53": "58",
"54": "59",
"55": "6",
"56": "60",
"57": "61",
"58": "62",
"59": "63",
"60": "64",
"61": "65",
"62": "66",
"63": "67",
"64": "68",
"65": "69",
"66": "7",
"67": "70",
"68": "71",
"69": "72",
"70": "73",
"71": "74",
"72": "75",
"73": "76",
"74": "77",
"75": "78",
"76": "79",
"77": "8",
"78": "80",
"79": "81",
"80": "82",
"81": "83",
"82": "84",
"83": "85",
"84": "86",
"85": "87",
"86": "88",
"87": "89",
"88": "9",
"89": "90",
"90": "91",
"91": "92"
},
"image_size": 224,
"initializer_range": 0.02,
"label2id": {
"1": 0,
"10": 1,
"11": 2,
"12": 3,
"13": 4,
"14": 5,
"15": 6,
"16": 7,
"17": 8,
"18": 9,
"19": 10,
"2": 11,
"20": 12,
"21": 13,
"22": 14,
"23": 15,
"24": 16,
"25": 17,
"26": 18,
"27": 19,
"28": 20,
"29": 21,
"3": 22,
"30": 23,
"31": 24,
"32": 25,
"33": 26,
"34": 27,
"35": 28,
"36": 29,
"37": 30,
"38": 31,
"39": 32,
"4": 33,
"40": 34,
"41": 35,
"42": 36,
"43": 37,
"44": 38,
"45": 39,
"46": 40,
"47": 41,
"48": 42,
"49": 43,
"5": 44,
"50": 45,
"51": 46,
"52": 47,
"53": 48,
"54": 49,
"55": 50,
"56": 51,
"57": 52,
"58": 53,
"59": 54,
"6": 55,
"60": 56,
"61": 57,
"62": 58,
"63": 59,
"64": 60,
"65": 61,
"66": 62,
"67": 63,
"68": 64,
"69": 65,
"7": 66,
"70": 67,
"71": 68,
"72": 69,
"73": 70,
"74": 71,
"75": 72,
"76": 73,
"77": 74,
"78": 75,
"79": 76,
"8": 77,
"80": 78,
"81": 79,
"82": 80,
"83": 81,
"84": 82,
"85": 83,
"86": 84,
"87": 85,
"88": 86,
"89": 87,
"9": 88,
"90": 89,
"91": 90,
"92": 91
},
"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.38.2",
"use_absolute_embeddings": false,
"window_size": 7
}