CoolFace
Modelpublic

Devarshi/Brain_Tumor_Classification

sourceHugging Faceapache-2.0updated 4y agoView on Hugging Face
12likes65downloads
config.json53 linesDownload Raw Back to root
1{2  "_name_or_path": "microsoft/swin-tiny-patch4-window7-224",3  "architectures": [4    "SwinForImageClassification"5  ],6  "attention_probs_dropout_prob": 0.0,7  "depths": [8    2,9    2,10    6,11    212  ],13  "drop_path_rate": 0.1,14  "embed_dim": 96,15  "encoder_stride": 32,16  "hidden_act": "gelu",17  "hidden_dropout_prob": 0.0,18  "hidden_size": 768,19  "id2label": {20    "0": "glioma_tumor",21    "1": "meningioma_tumor",22    "2": "no_tumor",23    "3": "pituitary_tumor"24  },25  "image_size": 224,26  "initializer_range": 0.02,27  "label2id": {28    "glioma_tumor": 0,29    "meningioma_tumor": 1,30    "no_tumor": 2,31    "pituitary_tumor": 332  },33  "layer_norm_eps": 1e-05,34  "mlp_ratio": 4.0,35  "model_type": "swin",36  "num_channels": 3,37  "num_heads": [38    3,39    6,40    12,41    2442  ],43  "num_layers": 4,44  "patch_size": 4,45  "path_norm": true,46  "problem_type": "single_label_classification",47  "qkv_bias": true,48  "torch_dtype": "float32",49  "transformers_version": "4.23.1",50  "use_absolute_embeddings": false,51  "window_size": 752}53