|
{ |
|
"architectures": [ |
|
"EfficientNetForImageClassification" |
|
], |
|
"batch_norm_eps": 0.001, |
|
"batch_norm_momentum": 0.99, |
|
"depth_coefficient": 3.1, |
|
"depth_divisor": 8, |
|
"depthwise_padding": [], |
|
"drop_connect_rate": 0.2, |
|
"dropout_rate": 0.5, |
|
"expand_ratios": [ |
|
1, |
|
6, |
|
6, |
|
6, |
|
6, |
|
6, |
|
6 |
|
], |
|
"hidden_act": "swish", |
|
"hidden_dim": 2560, |
|
"id2label": { |
|
"0": "Melanoma", |
|
"1": "Melanocytic nevus", |
|
"2": "Basal cell carcinoma", |
|
"3": "Actinic keratosis", |
|
"4": "Benign keratosis", |
|
"5": "Dermatofibroma", |
|
"6": "Vascular lesion", |
|
"7": "Squamous cell carcinoma" |
|
}, |
|
"image_size": 600, |
|
"in_channels": [ |
|
32, |
|
16, |
|
24, |
|
40, |
|
80, |
|
112, |
|
192 |
|
], |
|
"initializer_range": 0.02, |
|
"kernel_sizes": [ |
|
3, |
|
3, |
|
5, |
|
3, |
|
5, |
|
5, |
|
3 |
|
], |
|
"label2id": { |
|
"Actinic keratosis": "3", |
|
"Basal cell carcinoma": "2", |
|
"Benign keratosis": "4", |
|
"Dermatofibroma": "5", |
|
"Melanocytic nevus": "1", |
|
"Melanoma": "0", |
|
"Squamous cell carcinoma": "7", |
|
"Vascular lesion": "6" |
|
}, |
|
"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 |
|
], |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.36.2", |
|
"width_coefficient": 2.0 |
|
} |
|
|