|
{ |
|
"_name_or_path": "AutoTrain", |
|
"architectures": [ |
|
"ResNetForImageClassification" |
|
], |
|
"depths": [ |
|
3, |
|
4, |
|
6, |
|
3 |
|
], |
|
"downsample_in_first_stage": false, |
|
"embedding_size": 64, |
|
"hidden_act": "relu", |
|
"hidden_sizes": [ |
|
256, |
|
512, |
|
1024, |
|
2048 |
|
], |
|
"id2label": { |
|
"0": "Covid", |
|
"1": "Covid_test", |
|
"2": "Lung_Opacity", |
|
"3": "Lung_Opacity_test", |
|
"4": "Normal", |
|
"5": "Normal_test" |
|
}, |
|
"label2id": { |
|
"Covid": "0", |
|
"Covid_test": "1", |
|
"Lung_Opacity": "2", |
|
"Lung_Opacity_test": "3", |
|
"Normal": "4", |
|
"Normal_test": "5" |
|
}, |
|
"layer_type": "bottleneck", |
|
"max_length": 128, |
|
"model_type": "resnet", |
|
"num_channels": 3, |
|
"out_features": [ |
|
"stage4" |
|
], |
|
"out_indices": [ |
|
4 |
|
], |
|
"padding": "max_length", |
|
"problem_type": "single_label_classification", |
|
"stage_names": [ |
|
"stem", |
|
"stage1", |
|
"stage2", |
|
"stage3", |
|
"stage4" |
|
], |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.28.1" |
|
} |
|
|