nails-model / config.json
LucaLamacchia's picture
Upload folder using huggingface_hub
0b38cc7 verified
Raw
History Blame Contribute Delete
1.47 kB
{
"architectures": [
"EfficientNetForImageClassification"
],
"batch_norm_eps": 0.001,
"batch_norm_momentum": 0.99,
"depth_coefficient": 1.8,
"depth_divisor": 8,
"depthwise_padding": [
6
],
"drop_connect_rate": 0.2,
"dropout_rate": 0.4,
"dtype": "float32",
"expand_ratios": [
1,
6,
6,
6,
6,
6,
6
],
"hidden_act": "swish",
"hidden_dim": 1792,
"id2label": {
"0": "Acral_Lentiginous_Melanoma",
"1": "Healthy_Nail",
"2": "Onychogryphosis",
"3": "blue_finger",
"4": "clubbing",
"5": "pitting"
},
"image_size": 380,
"in_channels": [
32,
16,
24,
40,
80,
112,
192
],
"initializer_range": 0.02,
"kernel_sizes": [
3,
3,
5,
3,
5,
5,
3
],
"label2id": {
"Acral_Lentiginous_Melanoma": "0",
"Healthy_Nail": "1",
"Onychogryphosis": "2",
"blue_finger": "3",
"clubbing": "4",
"pitting": "5"
},
"model_type": "efficientnet",
"num_block_repeats": [
1,
2,
2,
3,
3,
4,
1
],
"num_channels": 3,
"num_hidden_layers": 64,
"out_channels": [
16,
24,
40,
80,
112,
192,
320
],
"pooling_type": "mean",
"problem_type": "single_label_classification",
"squeeze_expansion_ratio": 0.25,
"strides": [
1,
2,
2,
2,
1,
2,
1
],
"transformers_version": "5.10.1",
"use_cache": false,
"width_coefficient": 1.4
}