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
8a4f3f1
·
verified ·
1 Parent(s): eb18e8b

Model save

Browse files
Files changed (2) hide show
  1. README.md +66 -34
  2. model.safetensors +1 -1
README.md CHANGED
@@ -17,7 +17,7 @@ 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.2278
21
  - Accuracy: 0.9524
22
 
23
  ## Model description
@@ -46,44 +46,76 @@ 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: 32
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
- | 1.8831 | 1.0 | 147 | 1.2970 | 0.6531 |
56
- | 0.9115 | 2.0 | 294 | 0.8685 | 0.7143 |
57
- | 0.6506 | 3.0 | 441 | 0.7251 | 0.7653 |
58
- | 0.5561 | 4.0 | 588 | 0.6598 | 0.7959 |
59
- | 0.4606 | 5.0 | 735 | 0.6454 | 0.7806 |
60
- | 0.3946 | 6.0 | 882 | 0.5916 | 0.8282 |
61
- | 0.3433 | 7.0 | 1029 | 0.5672 | 0.8265 |
62
- | 0.281 | 8.0 | 1176 | 0.6194 | 0.8265 |
63
- | 0.2735 | 9.0 | 1323 | 0.4681 | 0.8571 |
64
- | 0.2452 | 10.0 | 1470 | 0.4842 | 0.8673 |
65
- | 0.2358 | 11.0 | 1617 | 0.4283 | 0.8776 |
66
- | 0.1693 | 12.0 | 1764 | 0.4548 | 0.8759 |
67
- | 0.1715 | 13.0 | 1911 | 0.5153 | 0.8724 |
68
- | 0.159 | 14.0 | 2058 | 0.3161 | 0.9184 |
69
- | 0.1557 | 15.0 | 2205 | 0.4901 | 0.8861 |
70
- | 0.1137 | 16.0 | 2352 | 0.4257 | 0.9031 |
71
- | 0.1239 | 17.0 | 2499 | 0.3844 | 0.9031 |
72
- | 0.1426 | 18.0 | 2646 | 0.3769 | 0.8980 |
73
- | 0.128 | 19.0 | 2793 | 0.3641 | 0.9082 |
74
- | 0.1168 | 20.0 | 2940 | 0.3241 | 0.9133 |
75
- | 0.1139 | 21.0 | 3087 | 0.2318 | 0.9320 |
76
- | 0.0975 | 22.0 | 3234 | 0.3019 | 0.9286 |
77
- | 0.0989 | 23.0 | 3381 | 0.2984 | 0.9269 |
78
- | 0.0847 | 24.0 | 3528 | 0.2930 | 0.9201 |
79
- | 0.0659 | 25.0 | 3675 | 0.3112 | 0.9371 |
80
- | 0.0789 | 26.0 | 3822 | 0.2835 | 0.9354 |
81
- | 0.0432 | 27.0 | 3969 | 0.3100 | 0.9303 |
82
- | 0.0438 | 28.0 | 4116 | 0.3168 | 0.9320 |
83
- | 0.0371 | 29.0 | 4263 | 0.2361 | 0.9490 |
84
- | 0.039 | 30.0 | 4410 | 0.2714 | 0.9473 |
85
- | 0.0435 | 31.0 | 4557 | 0.2547 | 0.9524 |
86
- | 0.0512 | 32.0 | 4704 | 0.2278 | 0.9524 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
87
 
88
 
89
  ### 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.2844
21
  - Accuracy: 0.9524
22
 
23
  ## Model description
 
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: 64
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | 2.0494 | 1.0 | 147 | 1.6122 | 0.5612 |
56
+ | 1.1533 | 2.0 | 294 | 0.9492 | 0.7330 |
57
+ | 0.7367 | 3.0 | 441 | 0.6831 | 0.8129 |
58
+ | 0.5485 | 4.0 | 588 | 0.6085 | 0.8112 |
59
+ | 0.4602 | 5.0 | 735 | 0.6417 | 0.8010 |
60
+ | 0.4421 | 6.0 | 882 | 0.6718 | 0.7806 |
61
+ | 0.4397 | 7.0 | 1029 | 0.5751 | 0.8248 |
62
+ | 0.3958 | 8.0 | 1176 | 0.4965 | 0.8503 |
63
+ | 0.3675 | 9.0 | 1323 | 0.4673 | 0.8571 |
64
+ | 0.3292 | 10.0 | 1470 | 0.5529 | 0.8248 |
65
+ | 0.2941 | 11.0 | 1617 | 0.3971 | 0.8759 |
66
+ | 0.23 | 12.0 | 1764 | 0.6137 | 0.8469 |
67
+ | 0.239 | 13.0 | 1911 | 0.5268 | 0.8588 |
68
+ | 0.2151 | 14.0 | 2058 | 0.4803 | 0.8759 |
69
+ | 0.2418 | 15.0 | 2205 | 0.4425 | 0.8605 |
70
+ | 0.1874 | 16.0 | 2352 | 0.4986 | 0.8571 |
71
+ | 0.2018 | 17.0 | 2499 | 0.4748 | 0.8878 |
72
+ | 0.1967 | 18.0 | 2646 | 0.3726 | 0.9065 |
73
+ | 0.1679 | 19.0 | 2793 | 0.4188 | 0.8912 |
74
+ | 0.167 | 20.0 | 2940 | 0.5495 | 0.8537 |
75
+ | 0.1584 | 21.0 | 3087 | 0.3908 | 0.9048 |
76
+ | 0.1569 | 22.0 | 3234 | 0.4526 | 0.8980 |
77
+ | 0.171 | 23.0 | 3381 | 0.4073 | 0.8844 |
78
+ | 0.1615 | 24.0 | 3528 | 0.4611 | 0.8759 |
79
+ | 0.1471 | 25.0 | 3675 | 0.3808 | 0.9048 |
80
+ | 0.1401 | 26.0 | 3822 | 0.4670 | 0.9014 |
81
+ | 0.1213 | 27.0 | 3969 | 0.4816 | 0.8844 |
82
+ | 0.1162 | 28.0 | 4116 | 0.4832 | 0.8827 |
83
+ | 0.1125 | 29.0 | 4263 | 0.4539 | 0.8929 |
84
+ | 0.109 | 30.0 | 4410 | 0.3661 | 0.9133 |
85
+ | 0.0996 | 31.0 | 4557 | 0.3400 | 0.9201 |
86
+ | 0.0973 | 32.0 | 4704 | 0.3441 | 0.9099 |
87
+ | 0.1181 | 33.0 | 4851 | 0.3713 | 0.9167 |
88
+ | 0.0903 | 34.0 | 4998 | 0.3351 | 0.9252 |
89
+ | 0.1021 | 35.0 | 5145 | 0.4256 | 0.9048 |
90
+ | 0.0682 | 36.0 | 5292 | 0.3798 | 0.9235 |
91
+ | 0.1128 | 37.0 | 5439 | 0.3761 | 0.9099 |
92
+ | 0.0851 | 38.0 | 5586 | 0.3434 | 0.9218 |
93
+ | 0.0838 | 39.0 | 5733 | 0.3656 | 0.9184 |
94
+ | 0.0746 | 40.0 | 5880 | 0.3743 | 0.9235 |
95
+ | 0.0776 | 41.0 | 6027 | 0.3182 | 0.9371 |
96
+ | 0.0727 | 42.0 | 6174 | 0.3861 | 0.9269 |
97
+ | 0.0839 | 43.0 | 6321 | 0.3226 | 0.9303 |
98
+ | 0.0619 | 44.0 | 6468 | 0.3421 | 0.9303 |
99
+ | 0.0689 | 45.0 | 6615 | 0.3497 | 0.9286 |
100
+ | 0.0603 | 46.0 | 6762 | 0.2965 | 0.9371 |
101
+ | 0.0492 | 47.0 | 6909 | 0.3269 | 0.9354 |
102
+ | 0.0552 | 48.0 | 7056 | 0.3348 | 0.9473 |
103
+ | 0.0501 | 49.0 | 7203 | 0.3621 | 0.9303 |
104
+ | 0.0708 | 50.0 | 7350 | 0.3247 | 0.9371 |
105
+ | 0.0596 | 51.0 | 7497 | 0.2823 | 0.9405 |
106
+ | 0.0463 | 52.0 | 7644 | 0.3110 | 0.9371 |
107
+ | 0.0438 | 53.0 | 7791 | 0.3717 | 0.9286 |
108
+ | 0.0348 | 54.0 | 7938 | 0.3130 | 0.9388 |
109
+ | 0.0384 | 55.0 | 8085 | 0.2841 | 0.9456 |
110
+ | 0.0551 | 56.0 | 8232 | 0.3086 | 0.9456 |
111
+ | 0.0353 | 57.0 | 8379 | 0.2653 | 0.9490 |
112
+ | 0.0456 | 58.0 | 8526 | 0.3161 | 0.9269 |
113
+ | 0.0238 | 59.0 | 8673 | 0.2408 | 0.9558 |
114
+ | 0.0244 | 60.0 | 8820 | 0.2389 | 0.9558 |
115
+ | 0.0333 | 61.0 | 8967 | 0.2798 | 0.9473 |
116
+ | 0.0274 | 62.0 | 9114 | 0.2832 | 0.9456 |
117
+ | 0.0259 | 63.0 | 9261 | 0.2414 | 0.9507 |
118
+ | 0.0223 | 64.0 | 9408 | 0.2844 | 0.9524 |
119
 
120
 
121
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:20bed482536844392cd5815a5ca9cb77d7e9a53e43b8cc57cac41992e2656e25
3
  size 343248584
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:27ebb95737aae47efbe80fc62cbc8db73d6d224ac74e77ebcef38d8b303aafc0
3
  size 343248584