Hugging Face's logo

Dharamanand
/
TransferLearning-BrainTumorClassifier

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
Dharamanand commited on
Commit
8a7b6d6
•
1 Parent(s): 714cbe1

Model save

Browse files
Files changed (4) hide show
  1. README.md +10 -20
  2. model.safetensors +1 -1
  3. runs/May09_09-40-57_484f78f74c09/events.out.tfevents.1715247670.484f78f74c09.693.0 +3 -0
  4. training_args.bin +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.8708
21
- - Accuracy: 0.7713
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: 0.0002
41
  - train_batch_size: 16
42
  - eval_batch_size: 8
43
  - seed: 42
@@ -47,23 +47,13 @@ The following hyperparameters were used during training:
47
 
48
  ### Training results
49
 
50
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
- |:-------------:|:------:|:----:|:---------------:|:--------:|
52
- | 1.3205 | 0.3289 | 100 | 1.2462 | 0.5589 |
53
- | 1.1927 | 0.6579 | 200 | 1.1432 | 0.7083 |
54
- | 1.1171 | 0.9868 | 300 | 1.0771 | 0.6418 |
55
- | 1.0534 | 1.3158 | 400 | 1.0285 | 0.7130 |
56
- | 1.0337 | 1.6447 | 500 | 0.9914 | 0.7223 |
57
- | 0.986 | 1.9737 | 600 | 0.9676 | 0.7445 |
58
- | 0.9606 | 2.3026 | 700 | 0.9419 | 0.7398 |
59
- | 0.9449 | 2.6316 | 800 | 0.9232 | 0.7596 |
60
- | 0.9075 | 2.9605 | 900 | 0.9064 | 0.7491 |
61
- | 0.8954 | 3.2895 | 1000 | 0.8954 | 0.7690 |
62
- | 0.908 | 3.6184 | 1100 | 0.8869 | 0.7608 |
63
- | 0.888 | 3.9474 | 1200 | 0.8808 | 0.7678 |
64
- | 0.8874 | 4.2763 | 1300 | 0.8762 | 0.7748 |
65
- | 0.8679 | 4.6053 | 1400 | 0.8723 | 0.7725 |
66
- | 0.8872 | 4.9342 | 1500 | 0.8708 | 0.7713 |
67
 
68
 
69
  ### 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.7952
21
+ - Accuracy: 0.7818
22
 
23
  ## Model description
24
 
 
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
+ - learning_rate: 0.0003
41
  - train_batch_size: 16
42
  - eval_batch_size: 8
43
  - seed: 42
 
47
 
48
  ### Training results
49
 
50
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
+ | 1.0462 | 1.0 | 304 | 1.0010 | 0.6418 |
53
+ | 0.9093 | 2.0 | 608 | 0.8900 | 0.7433 |
54
+ | 0.8251 | 3.0 | 912 | 0.8286 | 0.7678 |
55
+ | 0.8113 | 4.0 | 1216 | 0.8048 | 0.7841 |
56
+ | 0.8115 | 5.0 | 1520 | 0.7952 | 0.7818 |
 
 
 
 
 
 
 
 
 
 
57
 
58
 
59
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cfbb8373ad0b3d28474e63284e17157d93f2f52389ac9d71ec06d4fe5718b871
3
  size 343230128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b9bb445d66181fdeea6643b599154f5e8b1bc084752b1e498339f4bf7923c6f
3
  size 343230128
runs/May09_09-40-57_484f78f74c09/events.out.tfevents.1715247670.484f78f74c09.693.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e63c8ed073ea70fa5e52e632b2c521f7f98f4a57e53c09bde9b9e12209134f63
3
+ size 9941
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:769c677b5bd6af62c0e181f0d378fb8e75ed33be5381acc39762f3ed59f6474e
3
- size 4984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4ea6571b485fb7f2513c1e361251e026fc2b5684cc41a8266310f614ea7a226
3
+ size 5048