Hugging Face's logo

andrei-teodor
/
vit-base-brain-mri

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
andrei-teodor commited on
Commit
e578574
•
1 Parent(s): b406854

Model save

Browse files
Files changed (6) hide show
  1. README.md +7 -8
  2. config.json +0 -1
  3. model.safetensors +1 -1
  4. runs/Aug25_15-47-00_DESKTOP-4SGMSGR/events.out.tfevents.1724590021.DESKTOP-4SGMSGR.19616.0 +2 -2
  5. runs/Aug25_15-54-39_DESKTOP-4SGMSGR/events.out.tfevents.1724590480.DESKTOP-4SGMSGR.27212.0 +3 -0
  6. training_args.bin +1 -1
README.md CHANGED
@@ -3,7 +3,6 @@ library_name: transformers
3
  license: apache-2.0
4
  base_model: google/vit-base-patch16-224
5
  tags:
6
- - image-classification
7
  - generated_from_trainer
8
  metrics:
9
  - accuracy
@@ -17,10 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  # vit-base-brain-mri
19
 
20
- This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the BrainMRI dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 1.1297
23
- - Accuracy: 0.5685
24
 
25
  ## Model description
26
 
@@ -61,10 +60,10 @@ The following hyperparameters were used during training:
61
  | 0.6307 | 8.0 | 576 | 0.7822 | 0.6363 |
62
  | 0.5857 | 9.0 | 648 | 0.7822 | 0.6089 |
63
  | 0.576 | 10.0 | 720 | 0.7770 | 0.6249 |
64
- | 0.576 | 11.0 | 792 | 0.6184 | 0.7840 |
65
- | 0.5733 | 12.0 | 864 | 0.6006 | 0.7944 |
66
- | 0.5555 | 13.0 | 936 | 0.5898 | 0.8014 |
67
- | 0.5481 | 14.0 | 1008 | 0.5857 | 0.8223 |
68
 
69
 
70
  ### Framework versions
 
3
  license: apache-2.0
4
  base_model: google/vit-base-patch16-224
5
  tags:
 
6
  - generated_from_trainer
7
  metrics:
8
  - accuracy
 
16
 
17
  # vit-base-brain-mri
18
 
19
+ This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Accuracy: 0.8223
22
+ - Loss: 0.5857
23
 
24
  ## Model description
25
 
 
60
  | 0.6307 | 8.0 | 576 | 0.7822 | 0.6363 |
61
  | 0.5857 | 9.0 | 648 | 0.7822 | 0.6089 |
62
  | 0.576 | 10.0 | 720 | 0.7770 | 0.6249 |
63
+ | 0.576 | 11.0 | 792 | 0.7840 | 0.6184 |
64
+ | 0.5733 | 12.0 | 864 | 0.7944 | 0.6006 |
65
+ | 0.5555 | 13.0 | 936 | 0.8014 | 0.5898 |
66
+ | 0.5481 | 14.0 | 1008 | 0.8223 | 0.5857 |
67
 
68
 
69
  ### Framework versions
config.json CHANGED
@@ -29,7 +29,6 @@
29
  "num_channels": 3,
30
  "num_hidden_layers": 12,
31
  "patch_size": 16,
32
- "problem_type": "single_label_classification",
33
  "qkv_bias": true,
34
  "torch_dtype": "float32",
35
  "transformers_version": "4.44.2"
 
29
  "num_channels": 3,
30
  "num_hidden_layers": 12,
31
  "patch_size": 16,
 
32
  "qkv_bias": true,
33
  "torch_dtype": "float32",
34
  "transformers_version": "4.44.2"
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1e8b2807e569ef21dcb3c6327db498080012919c55e94660df17657ebb694bb0
3
  size 343230128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:30ee3e9f0f561fd607211c791628bbf90a3534be82191709c8a92e8ab4acd17c
3
  size 343230128
runs/Aug25_15-47-00_DESKTOP-4SGMSGR/events.out.tfevents.1724590021.DESKTOP-4SGMSGR.19616.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9c761bc81054800b151f2431139150f95651ff51ce81b6973e474fa0620c5001
3
- size 10554
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77ae4bf195626269b8136434dc30fd9dbe1f9d11ccdd858522332b09a0786272
3
+ size 11088
runs/Aug25_15-54-39_DESKTOP-4SGMSGR/events.out.tfevents.1724590480.DESKTOP-4SGMSGR.27212.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4cfd23407344f539d812ede735b8d694e403f3e8ab765e8d5ab6e2526071cfb1
3
+ size 5351
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ec7d046b69a14b32bae467d94129d8d179caa2e45b65c0978fe134e84e5e35f9
3
  size 5176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3bb2984d0c2cf21b6334e6f59abfa302ad9229d1cbab7df929a2d90f3bbbb6b1
3
  size 5176