Hugging Face's logo

Hemg
/
Wound-Image-classification

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Model card Files
xet
Metrics Community
Hemg commited on
Commit
33ec79e
·
verified ·
1 Parent(s): cd0487d

Model save

Browse files
Files changed (3) hide show
  1. README.md +35 -11
  2. model.safetensors +1 -1
  3. runs/Mar11_10-14-12_33467ebf2898/events.out.tfevents.1710152053.33467ebf2898.370.8 +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.5404
21
- - Accuracy: 0.8851
22
 
23
  ## Model description
24
 
@@ -46,20 +46,44 @@ The following hyperparameters were used during training:
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_ratio: 0.1
49
- - num_epochs: 8
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
- | 1.8668 | 0.98 | 21 | 1.5875 | 0.4828 |
56
- | 1.1995 | 2.0 | 43 | 1.0690 | 0.7586 |
57
- | 0.8205 | 2.98 | 64 | 0.7598 | 0.8621 |
58
- | 0.5405 | 4.0 | 86 | 0.6630 | 0.8391 |
59
- | 0.4268 | 4.98 | 107 | 0.5655 | 0.8851 |
60
- | 0.3204 | 6.0 | 129 | 0.4704 | 0.9195 |
61
- | 0.2874 | 6.98 | 150 | 0.5273 | 0.8621 |
62
- | 0.2487 | 7.81 | 168 | 0.5404 | 0.8851 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
63
 
64
 
65
  ### 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.3723
21
+ - Accuracy: 0.8966
22
 
23
  ## Model description
24
 
 
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_ratio: 0.1
49
+ - num_epochs: 32
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | 1.9629 | 0.98 | 21 | 1.8499 | 0.3908 |
56
+ | 1.6285 | 2.0 | 43 | 1.5155 | 0.5402 |
57
+ | 1.2198 | 2.98 | 64 | 1.0158 | 0.8161 |
58
+ | 0.7367 | 4.0 | 86 | 0.7362 | 0.8736 |
59
+ | 0.4858 | 4.98 | 107 | 0.5636 | 0.8621 |
60
+ | 0.3395 | 6.0 | 129 | 0.5547 | 0.8621 |
61
+ | 0.2782 | 6.98 | 150 | 0.5086 | 0.8276 |
62
+ | 0.2215 | 8.0 | 172 | 0.5234 | 0.8391 |
63
+ | 0.275 | 8.98 | 193 | 0.4102 | 0.8966 |
64
+ | 0.1995 | 10.0 | 215 | 0.5127 | 0.8621 |
65
+ | 0.2395 | 10.98 | 236 | 0.4507 | 0.8506 |
66
+ | 0.129 | 12.0 | 258 | 0.3436 | 0.8506 |
67
+ | 0.166 | 12.98 | 279 | 0.6932 | 0.8276 |
68
+ | 0.1156 | 14.0 | 301 | 0.4058 | 0.8851 |
69
+ | 0.1466 | 14.98 | 322 | 0.3980 | 0.8621 |
70
+ | 0.1673 | 16.0 | 344 | 0.3390 | 0.9195 |
71
+ | 0.0963 | 16.98 | 365 | 0.3953 | 0.8966 |
72
+ | 0.1058 | 18.0 | 387 | 0.4769 | 0.8621 |
73
+ | 0.1016 | 18.98 | 408 | 0.2004 | 0.9540 |
74
+ | 0.0792 | 20.0 | 430 | 0.2843 | 0.9310 |
75
+ | 0.0858 | 20.98 | 451 | 0.3540 | 0.8966 |
76
+ | 0.1046 | 22.0 | 473 | 0.5916 | 0.8276 |
77
+ | 0.0763 | 22.98 | 494 | 0.5639 | 0.8161 |
78
+ | 0.0934 | 24.0 | 516 | 0.4209 | 0.8736 |
79
+ | 0.1534 | 24.98 | 537 | 0.3895 | 0.8736 |
80
+ | 0.0785 | 26.0 | 559 | 0.4623 | 0.8391 |
81
+ | 0.0905 | 26.98 | 580 | 0.3779 | 0.8966 |
82
+ | 0.0589 | 28.0 | 602 | 0.3099 | 0.8851 |
83
+ | 0.0944 | 28.98 | 623 | 0.2375 | 0.9310 |
84
+ | 0.0567 | 30.0 | 645 | 0.2871 | 0.9195 |
85
+ | 0.0667 | 30.98 | 666 | 0.3315 | 0.9080 |
86
+ | 0.0631 | 31.26 | 672 | 0.3723 | 0.8966 |
87
 
88
 
89
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9d91025725c36aececeb09541604da45b9b8ed0d1c9f00ce81a4d3e7c2111066
3
  size 343239356
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b05a1fa193a56e0e6038dd13320dba92d4a9da4c8f84018151efa961f3ab1cd
3
  size 343239356
runs/Mar11_10-14-12_33467ebf2898/events.out.tfevents.1710152053.33467ebf2898.370.8 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:aac2a3310998d29958b5c837d780efa1aec4d65604a03beca8b60c55a1c4711c
3
- size 21311
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:85ffdb5ca0b4b17fec4f969de7f75a516e8c6d546eb22f8d2de7be7941af5dd9
3
+ size 22199