Hugging Face's logo

bombshelll
/
swin-brain-abnormalities-classification

Image Classification
Transformers
Safetensors
swin
Generated from Trainer
Model card Files
xet
Community
bombshelll commited on
Commit
9d95a82
·
verified ·
1 Parent(s): 7dbcada

Training in progress, epoch 0

Browse files
Files changed (3) hide show
  1. config.json +6 -4
  2. model.safetensors +2 -2
  3. training_args.bin +1 -1
config.json CHANGED
@@ -17,14 +17,16 @@
17
  "hidden_dropout_prob": 0.0,
18
  "hidden_size": 768,
19
  "id2label": {
20
- "0": "abnormal",
21
- "1": "normal"
 
22
  },
23
  "image_size": 224,
24
  "initializer_range": 0.02,
25
  "label2id": {
26
- "abnormal": 0,
27
- "normal": 1
 
28
  },
29
  "layer_norm_eps": 1e-05,
30
  "mlp_ratio": 4.0,
 
17
  "hidden_dropout_prob": 0.0,
18
  "hidden_size": 768,
19
  "id2label": {
20
+ "0": "lesion",
21
+ "1": "normal",
22
+ "2": "tumor"
23
  },
24
  "image_size": 224,
25
  "initializer_range": 0.02,
26
  "label2id": {
27
+ "lesion": 0,
28
+ "normal": 1,
29
+ "tumor": 2
30
  },
31
  "layer_norm_eps": 1e-05,
32
  "mlp_ratio": 4.0,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ed991ff42a73ff3e86766498ef8f30af92241f8d09bb2b43c8b0ed8b6985a096
3
- size 110342832
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7ca5a796789a79c468b6f68da0b207bd23aa92da1bcc3b81985ca2582972efe6
3
+ size 110345908
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ce48c691ea161a2aee45e76aca3dffcb97c06f20f00e6edb321b4fadf496d9f5
3
  size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4de15c06e461c7e03a08721b5c8801808e58c2edd0f619a960521222162f35d2
3
  size 5304