{ | |
"_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 | |
} | |