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
761cc8f
•
1 Parent(s): 706d16f

Model save

Browse files
Files changed (2) hide show
  1. README.md +26 -22
  2. runs/Aug28_15-52-04_DESKTOP-4SGMSGR/events.out.tfevents.1724849525.DESKTOP-4SGMSGR.2900.0 +2 -2
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
 
@@ -45,27 +44,32 @@ The following hyperparameters were used during training:
45
  - seed: 42
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
- - num_epochs: 15
49
 
50
  ### Training results
51
 
52
- | Training Loss | Epoch | Step | Accuracy | Validation Loss |
53
- |:-------------:|:-----:|:----:|:--------:|:---------------:|
54
- | No log | 1.0 | 72 | 0.6289 | 0.9818 |
55
- | 1.0744 | 2.0 | 144 | 0.6864 | 0.8287 |
56
- | 0.7716 | 3.0 | 216 | 0.7160 | 0.7535 |
57
- | 0.7716 | 4.0 | 288 | 0.7404 | 0.7140 |
58
- | 0.6975 | 5.0 | 360 | 0.7491 | 0.7015 |
59
- | 0.6651 | 6.0 | 432 | 0.7631 | 0.6839 |
60
- | 0.6307 | 7.0 | 504 | 0.7700 | 0.6624 |
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.7840 | 0.6184 |
65
- | 0.5733 | 12.0 | 864 | 0.7944 | 0.6006 |
66
- | 0.5555 | 13.0 | 936 | 0.8014 | 0.5898 |
67
- | 0.5481 | 14.0 | 1008 | 0.8223 | 0.5857 |
68
- | 0.5481 | 15.0 | 1080 | 0.5992 | 0.7857 |
 
 
 
 
 
69
 
70
 
71
  ### 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
+ - Loss: 0.4598
22
+ - Accuracy: 0.8659
23
 
24
  ## Model description
25
 
 
44
  - seed: 42
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
+ - num_epochs: 20
48
 
49
  ### Training results
50
 
51
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
+ | No log | 1.0 | 72 | 0.9986 | 0.6098 |
54
+ | 1.098 | 2.0 | 144 | 0.8445 | 0.7003 |
55
+ | 0.7895 | 3.0 | 216 | 0.7318 | 0.7526 |
56
+ | 0.7895 | 4.0 | 288 | 0.6842 | 0.7474 |
57
+ | 0.6629 | 5.0 | 360 | 0.6328 | 0.7857 |
58
+ | 0.5966 | 6.0 | 432 | 0.5957 | 0.8101 |
59
+ | 0.5546 | 7.0 | 504 | 0.5646 | 0.8118 |
60
+ | 0.5546 | 8.0 | 576 | 0.5647 | 0.8049 |
61
+ | 0.5113 | 9.0 | 648 | 0.5340 | 0.8275 |
62
+ | 0.4882 | 10.0 | 720 | 0.5190 | 0.8328 |
63
+ | 0.4882 | 11.0 | 792 | 0.5197 | 0.8328 |
64
+ | 0.4789 | 12.0 | 864 | 0.5002 | 0.8258 |
65
+ | 0.4582 | 13.0 | 936 | 0.4957 | 0.8310 |
66
+ | 0.4426 | 14.0 | 1008 | 0.4821 | 0.8310 |
67
+ | 0.4426 | 15.0 | 1080 | 0.4706 | 0.8467 |
68
+ | 0.4328 | 16.0 | 1152 | 0.4821 | 0.8153 |
69
+ | 0.432 | 17.0 | 1224 | 0.4992 | 0.8275 |
70
+ | 0.432 | 18.0 | 1296 | 0.4799 | 0.8345 |
71
+ | 0.4196 | 19.0 | 1368 | 0.4838 | 0.8310 |
72
+ | 0.4287 | 20.0 | 1440 | 0.4598 | 0.8659 |
73
 
74
 
75
  ### Framework versions
runs/Aug28_15-52-04_DESKTOP-4SGMSGR/events.out.tfevents.1724849525.DESKTOP-4SGMSGR.2900.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:228460c2a313d4b21c05134eb326507a7e6e90619af52908ed29f7e8deb5700c
3
- size 14078
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c0fb8ad34d715c6b819756e8a50151b5a3dd7dfed058baad1fcc0c84bba53e5
3
+ size 14755