Hugging Face's logo

Hemg
/
Brain-Tumor-Classification

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
Hemg commited on
Commit
fa1da00
•
1 Parent(s): 9bdd6e3

Model save

Browse files
Files changed (3) hide show
  1. README.md +19 -19
  2. model.safetensors +1 -1
  3. runs/Mar06_09-34-11_65f3a70606cc/events.out.tfevents.1709717652.65f3a70606cc.34.6 +2 -2
README.md CHANGED
@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.1912
21
- - Accuracy: 0.9568
22
 
23
  ## Model description
24
 
@@ -37,7 +37,7 @@ More information needed
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
- - learning_rate: 4e-05
41
  - train_batch_size: 32
42
  - eval_batch_size: 32
43
  - seed: 42
@@ -52,22 +52,22 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
- | 1.3193 | 0.98 | 13 | 1.1484 | 0.5468 |
56
- | 1.0232 | 1.96 | 26 | 0.8361 | 0.7074 |
57
- | 0.7191 | 2.94 | 39 | 0.5765 | 0.8681 |
58
- | 0.4619 | 4.0 | 53 | 0.4373 | 0.8873 |
59
- | 0.3721 | 4.98 | 66 | 0.3590 | 0.9233 |
60
- | 0.3025 | 5.96 | 79 | 0.3022 | 0.9281 |
61
- | 0.2727 | 6.94 | 92 | 0.2658 | 0.9448 |
62
- | 0.2123 | 8.0 | 106 | 0.2307 | 0.9472 |
63
- | 0.2221 | 8.98 | 119 | 0.2353 | 0.9448 |
64
- | 0.1892 | 9.96 | 132 | 0.2064 | 0.9592 |
65
- | 0.1875 | 10.94 | 145 | 0.2105 | 0.9496 |
66
- | 0.154 | 12.0 | 159 | 0.2149 | 0.9520 |
67
- | 0.1589 | 12.98 | 172 | 0.2041 | 0.9496 |
68
- | 0.155 | 13.96 | 185 | 0.2056 | 0.9472 |
69
- | 0.1497 | 14.94 | 198 | 0.2009 | 0.9496 |
70
- | 0.1371 | 15.7 | 208 | 0.1912 | 0.9568 |
71
 
72
 
73
  ### Framework versions
 
17
 
18
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.0872
21
+ - Accuracy: 0.9758
22
 
23
  ## Model description
24
 
 
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
+ - learning_rate: 5e-05
41
  - train_batch_size: 32
42
  - eval_batch_size: 32
43
  - seed: 42
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | 1.2074 | 1.0 | 44 | 0.8060 | 0.8128 |
56
+ | 0.4897 | 2.0 | 88 | 0.3008 | 0.9274 |
57
+ | 0.2462 | 3.0 | 132 | 0.2464 | 0.9331 |
58
+ | 0.1937 | 4.0 | 176 | 0.1918 | 0.9502 |
59
+ | 0.1523 | 5.0 | 220 | 0.1699 | 0.9502 |
60
+ | 0.1371 | 6.0 | 264 | 0.1372 | 0.9644 |
61
+ | 0.1104 | 7.0 | 308 | 0.1121 | 0.9708 |
62
+ | 0.1097 | 8.0 | 352 | 0.1220 | 0.9651 |
63
+ | 0.1015 | 9.0 | 396 | 0.1053 | 0.9737 |
64
+ | 0.0841 | 10.0 | 440 | 0.1142 | 0.9708 |
65
+ | 0.0839 | 11.0 | 484 | 0.1073 | 0.9708 |
66
+ | 0.0771 | 12.0 | 528 | 0.1156 | 0.9665 |
67
+ | 0.074 | 13.0 | 572 | 0.1203 | 0.9644 |
68
+ | 0.0652 | 14.0 | 616 | 0.0706 | 0.9858 |
69
+ | 0.0694 | 15.0 | 660 | 0.0984 | 0.9744 |
70
+ | 0.0596 | 16.0 | 704 | 0.0872 | 0.9758 |
71
 
72
 
73
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:83edbff4a79320a78b955faa8412726fc21962d910fa93ec7e2645ca14325079
3
  size 343230128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:51d06d0a5fc79184da94f8badd53f513dda7f8f5ae8d9d9a53aa55d9bd87481b
3
  size 343230128
runs/Mar06_09-34-11_65f3a70606cc/events.out.tfevents.1709717652.65f3a70606cc.34.6 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a23320a83bb3f96dbe19f650a29ff1fba905af6ae93721f422cd7c8d9bab8de8
3
- size 13204
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:254e081e461c58bebc8b658193f1b9bf8aa59068d28099e325bdc75d7c489f75
3
+ size 13558