config.json

{
 "_name_or_path": "sail/poolformer_s12",
 "architectures": [
 "PoolFormerForImageClassification"
 ],
 "depths": [
 2,
 2,
 6,
 2
 ],
 "drop_path_rate": 0.0,
 "hidden_act": "gelu",
 "hidden_sizes": [
 64,
 128,
 320,
 512
 ],
 "id2label": {
 "0": "a_noDR",
 "1": "b_mildDR",
 "2": "c_moderateDR",
 "3": "d_severeDR",
 "4": "e_proDR"
 },
 "initializer_range": 0.02,
 "label2id": {
 "a_noDR": 0,
 "b_mildDR": 1,
 "c_moderateDR": 2,
 "d_severeDR": 3,
 "e_proDR": 4
 },
 "layer_scale_init_value": 1e-05,
 "mlp_ratio": 4.0,
 "model_type": "poolformer",
 "num_channels": 3,
 "num_encoder_blocks": 4,
 "num_hidden_layers": 12,
 "padding": [
 2,
 1,
 1,
 1
 ],
 "patch_size": 16,
 "patch_sizes": [
 7,
 3,
 3,
 3
 ],
 "pool_size": 3,
 "problem_type": "single_label_classification",
 "stride": 16,
 "strides": [
 4,
 2,
 2,
 2
 ],
 "torch_dtype": "float32",
 "transformers_version": "4.30.0",
 "use_cache": true,
 "use_layer_scale": true
}