Hugging Face's logo

dhritic99
/
vit-base-brain-alzheimer-detection

Safetensors
vit
Generated from Trainer
Model card Files Community
dhritic99 commited on
Commit
63775d7
•
1 Parent(s): 23d37fe

Model save

Browse files
Files changed (2) hide show
  1. README.md +10 -19
  2. model.safetensors +1 -1
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.2428
21
- - Accuracy: 0.9508
22
 
23
  ## Model description
24
 
@@ -51,23 +51,14 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
  |:-------------:|:-------:|:----:|:---------------:|:--------:|
54
- | 0.9772 | 0.7812 | 200 | 0.9400 | 0.5801 |
55
- | 0.7451 | 1.5625 | 400 | 0.7947 | 0.6553 |
56
- | 0.5701 | 2.3438 | 600 | 0.7642 | 0.7236 |
57
- | 0.3704 | 3.125 | 800 | 0.5532 | 0.7744 |
58
- | 0.2906 | 3.9062 | 1000 | 0.4423 | 0.8555 |
59
- | 0.1636 | 4.6875 | 1200 | 0.3226 | 0.9004 |
60
- | 0.0837 | 5.4688 | 1400 | 0.3483 | 0.9023 |
61
- | 0.0368 | 6.25 | 1600 | 0.2423 | 0.9395 |
62
- | 0.063 | 7.0312 | 1800 | 0.3091 | 0.9277 |
63
- | 0.047 | 7.8125 | 2000 | 0.3907 | 0.9023 |
64
- | 0.0127 | 8.5938 | 2200 | 0.2002 | 0.9561 |
65
- | 0.0102 | 9.375 | 2400 | 0.3001 | 0.9307 |
66
- | 0.0086 | 10.1562 | 2600 | 0.1998 | 0.9512 |
67
- | 0.0073 | 10.9375 | 2800 | 0.1932 | 0.9590 |
68
- | 0.0064 | 11.7188 | 3000 | 0.1988 | 0.9561 |
69
- | 0.0056 | 12.5 | 3200 | 0.1993 | 0.9580 |
70
- | 0.0049 | 13.2812 | 3400 | 0.2047 | 0.9590 |
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.1892
21
+ - Accuracy: 0.9590
22
 
23
  ## Model description
24
 
 
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
  |:-------------:|:-------:|:----:|:---------------:|:--------:|
54
+ | 0.4285 | 1.9531 | 500 | 0.4633 | 0.8311 |
55
+ | 0.171 | 3.9062 | 1000 | 0.3237 | 0.8994 |
56
+ | 0.0622 | 5.8594 | 1500 | 0.2032 | 0.9414 |
57
+ | 0.0162 | 7.8125 | 2000 | 0.2413 | 0.9512 |
58
+ | 0.0044 | 9.7656 | 2500 | 0.1623 | 0.9668 |
59
+ | 0.003 | 11.7188 | 3000 | 0.1641 | 0.9668 |
60
+ | 0.0025 | 13.6719 | 3500 | 0.1796 | 0.9619 |
61
+ | 0.0019 | 15.625 | 4000 | 0.1892 | 0.9590 |
 
 
 
 
 
 
 
 
 
62
 
63
 
64
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:594b232445ee48c8118baeb6b629f6835131e60bab12b6913375f025f4982a39
3
  size 343230128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad7d25badfac8196f48af0ad69a456305d0e35162b2fe7014cff19639e3c4252
3
  size 343230128