Hugging Face's logo

bombshelll
/
swin-brain-modality-classification

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

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
@@ -18,15 +18,17 @@
18
  "hidden_size": 768,
19
  "id2label": {
20
  "0": "CT",
21
- "1": "MRI T1",
22
- "2": "MRI T2"
 
23
  },
24
  "image_size": 224,
25
  "initializer_range": 0.02,
26
  "label2id": {
27
  "CT": 0,
28
- "MRI T1": 1,
29
- "MRI T2": 2
 
30
  },
31
  "layer_norm_eps": 1e-05,
32
  "mlp_ratio": 4.0,
 
18
  "hidden_size": 768,
19
  "id2label": {
20
  "0": "CT",
21
+ "1": "MRI FLAIR",
22
+ "2": "MRI T1",
23
+ "3": "MRI T2"
24
  },
25
  "image_size": 224,
26
  "initializer_range": 0.02,
27
  "label2id": {
28
  "CT": 0,
29
+ "MRI FLAIR": 1,
30
+ "MRI T1": 2,
31
+ "MRI T2": 3
32
  },
33
  "layer_norm_eps": 1e-05,
34
  "mlp_ratio": 4.0,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:83ab136ea83b973b30634513252d3b7a4788f34e5b745af9d521a6ce0df95ca1
3
- size 110345908
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81d1aef6d4be03be15fe51f6b8399742c503b2b6845f67da6a5dc5bd57edabd2
3
+ size 110348984
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:07169705d3bceecccfde57a5e0a7d54f6a5e1dbf17f3c761c735117f417b6159
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03354c3552bae3fa7f86e5bd329b3ee4b073249495b11f0f16d4d1ac3201234d
3
  size 5240