bryanzhou008
commited on
Commit
•
9f604a4
1
Parent(s):
4aae2b1
Model save
Browse files- README.md +90 -85
- model.safetensors +1 -1
README.md
CHANGED
@@ -23,7 +23,7 @@ model-index:
|
|
23 |
metrics:
|
24 |
- name: Accuracy
|
25 |
type: accuracy
|
26 |
-
value: 0.
|
27 |
---
|
28 |
|
29 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
@@ -33,8 +33,8 @@ should probably proofread and complete it, then remove this comment. -->
|
|
33 |
|
34 |
This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
|
35 |
It achieves the following results on the evaluation set:
|
36 |
-
- Loss:
|
37 |
-
- Accuracy: 0.
|
38 |
|
39 |
## Model description
|
40 |
|
@@ -66,88 +66,93 @@ The following hyperparameters were used during training:
|
|
66 |
|
67 |
### Training results
|
68 |
|
69 |
-
| Training Loss | Epoch
|
70 |
-
|
71 |
-
| No log | 0.
|
72 |
-
| No log |
|
73 |
-
|
|
74 |
-
| 3.
|
75 |
-
| 3.
|
76 |
-
| 3.
|
77 |
-
|
|
78 |
-
|
|
79 |
-
|
|
80 |
-
|
|
81 |
-
|
|
82 |
-
| 2.
|
83 |
-
| 2.
|
84 |
-
| 2.
|
85 |
-
|
|
86 |
-
|
|
87 |
-
|
|
88 |
-
|
|
89 |
-
|
|
90 |
-
|
|
91 |
-
|
|
92 |
-
|
|
93 |
-
|
|
94 |
-
| 1.
|
95 |
-
| 1.
|
96 |
-
| 1.
|
97 |
-
| 1.
|
98 |
-
|
|
99 |
-
|
|
100 |
-
|
|
101 |
-
|
|
102 |
-
|
|
103 |
-
|
|
104 |
-
|
|
105 |
-
|
|
106 |
-
|
|
107 |
-
|
|
108 |
-
|
|
109 |
-
|
|
110 |
-
|
|
111 |
-
|
|
112 |
-
|
|
113 |
-
|
|
114 |
-
|
|
115 |
-
|
|
116 |
-
|
|
117 |
-
|
|
118 |
-
|
|
119 |
-
|
|
120 |
-
|
|
121 |
-
|
|
122 |
-
|
|
123 |
-
|
|
124 |
-
|
|
125 |
-
|
|
126 |
-
|
|
127 |
-
|
|
128 |
-
|
|
129 |
-
|
|
130 |
-
| 0.
|
131 |
-
| 0.
|
132 |
-
| 0.
|
133 |
-
| 0.
|
134 |
-
| 0.
|
135 |
-
| 0.
|
136 |
-
| 0.
|
137 |
-
| 0.
|
138 |
-
| 0.
|
139 |
-
| 0.
|
140 |
-
| 0.
|
141 |
-
| 0.
|
142 |
-
| 0.
|
143 |
-
| 0.
|
144 |
-
| 0.
|
145 |
-
| 0.
|
146 |
-
| 0.
|
147 |
-
| 0.
|
148 |
-
| 0.
|
149 |
-
| 0.
|
150 |
-
| 0.
|
|
|
|
|
|
|
|
|
|
|
151 |
|
152 |
|
153 |
### Framework versions
|
|
|
23 |
metrics:
|
24 |
- name: Accuracy
|
25 |
type: accuracy
|
26 |
+
value: 0.8541666666666666
|
27 |
---
|
28 |
|
29 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
|
33 |
|
34 |
This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
|
35 |
It achieves the following results on the evaluation set:
|
36 |
+
- Loss: 0.7703
|
37 |
+
- Accuracy: 0.8542
|
38 |
|
39 |
## Model description
|
40 |
|
|
|
66 |
|
67 |
### Training results
|
68 |
|
69 |
+
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
70 |
+
|:-------------:|:-------:|:----:|:---------------:|:--------:|
|
71 |
+
| No log | 0.8421 | 4 | 3.1793 | 0.0347 |
|
72 |
+
| No log | 1.8947 | 9 | 3.1647 | 0.0486 |
|
73 |
+
| 3.1648 | 2.9474 | 14 | 3.1382 | 0.0944 |
|
74 |
+
| 3.1648 | 4.0 | 19 | 3.0995 | 0.1556 |
|
75 |
+
| 3.0817 | 4.8421 | 23 | 3.0555 | 0.2639 |
|
76 |
+
| 3.0817 | 5.8947 | 28 | 2.9849 | 0.3889 |
|
77 |
+
| 2.9167 | 6.9474 | 33 | 2.8932 | 0.5139 |
|
78 |
+
| 2.9167 | 8.0 | 38 | 2.7775 | 0.5972 |
|
79 |
+
| 2.6682 | 8.8421 | 42 | 2.6706 | 0.6528 |
|
80 |
+
| 2.6682 | 9.8947 | 47 | 2.5233 | 0.7069 |
|
81 |
+
| 2.3659 | 10.9474 | 52 | 2.3859 | 0.7375 |
|
82 |
+
| 2.3659 | 12.0 | 57 | 2.2546 | 0.75 |
|
83 |
+
| 2.079 | 12.8421 | 61 | 2.1531 | 0.7528 |
|
84 |
+
| 2.079 | 13.8947 | 66 | 2.0372 | 0.75 |
|
85 |
+
| 1.828 | 14.9474 | 71 | 1.9339 | 0.7597 |
|
86 |
+
| 1.828 | 16.0 | 76 | 1.8403 | 0.7694 |
|
87 |
+
| 1.6253 | 16.8421 | 80 | 1.7733 | 0.7764 |
|
88 |
+
| 1.6253 | 17.8947 | 85 | 1.6914 | 0.7903 |
|
89 |
+
| 1.4502 | 18.9474 | 90 | 1.6153 | 0.7875 |
|
90 |
+
| 1.4502 | 20.0 | 95 | 1.5510 | 0.7986 |
|
91 |
+
| 1.4502 | 20.8421 | 99 | 1.5016 | 0.8 |
|
92 |
+
| 1.2959 | 21.8947 | 104 | 1.4454 | 0.8222 |
|
93 |
+
| 1.2959 | 22.9474 | 109 | 1.3912 | 0.8181 |
|
94 |
+
| 1.1802 | 24.0 | 114 | 1.3390 | 0.8333 |
|
95 |
+
| 1.1802 | 24.8421 | 118 | 1.2995 | 0.8333 |
|
96 |
+
| 1.0629 | 25.8947 | 123 | 1.2707 | 0.8389 |
|
97 |
+
| 1.0629 | 26.9474 | 128 | 1.2335 | 0.8361 |
|
98 |
+
| 0.9801 | 28.0 | 133 | 1.1975 | 0.8444 |
|
99 |
+
| 0.9801 | 28.8421 | 137 | 1.1672 | 0.8389 |
|
100 |
+
| 0.9076 | 29.8947 | 142 | 1.1338 | 0.8444 |
|
101 |
+
| 0.9076 | 30.9474 | 147 | 1.1137 | 0.8472 |
|
102 |
+
| 0.8349 | 32.0 | 152 | 1.0855 | 0.8528 |
|
103 |
+
| 0.8349 | 32.8421 | 156 | 1.0717 | 0.8542 |
|
104 |
+
| 0.7782 | 33.8947 | 161 | 1.0483 | 0.8514 |
|
105 |
+
| 0.7782 | 34.9474 | 166 | 1.0352 | 0.85 |
|
106 |
+
| 0.7208 | 36.0 | 171 | 1.0202 | 0.8556 |
|
107 |
+
| 0.7208 | 36.8421 | 175 | 0.9994 | 0.8486 |
|
108 |
+
| 0.6708 | 37.8947 | 180 | 0.9814 | 0.8556 |
|
109 |
+
| 0.6708 | 38.9474 | 185 | 0.9691 | 0.8542 |
|
110 |
+
| 0.6303 | 40.0 | 190 | 0.9599 | 0.8486 |
|
111 |
+
| 0.6303 | 40.8421 | 194 | 0.9422 | 0.8472 |
|
112 |
+
| 0.6303 | 41.8947 | 199 | 0.9278 | 0.8486 |
|
113 |
+
| 0.6018 | 42.9474 | 204 | 0.9172 | 0.8528 |
|
114 |
+
| 0.6018 | 44.0 | 209 | 0.9093 | 0.8514 |
|
115 |
+
| 0.5622 | 44.8421 | 213 | 0.9030 | 0.8583 |
|
116 |
+
| 0.5622 | 45.8947 | 218 | 0.8972 | 0.8625 |
|
117 |
+
| 0.5474 | 46.9474 | 223 | 0.8859 | 0.8569 |
|
118 |
+
| 0.5474 | 48.0 | 228 | 0.8858 | 0.8653 |
|
119 |
+
| 0.5254 | 48.8421 | 232 | 0.8779 | 0.8556 |
|
120 |
+
| 0.5254 | 49.8947 | 237 | 0.8635 | 0.8569 |
|
121 |
+
| 0.5036 | 50.9474 | 242 | 0.8563 | 0.8611 |
|
122 |
+
| 0.5036 | 52.0 | 247 | 0.8613 | 0.8542 |
|
123 |
+
| 0.4855 | 52.8421 | 251 | 0.8546 | 0.8625 |
|
124 |
+
| 0.4855 | 53.8947 | 256 | 0.8469 | 0.8597 |
|
125 |
+
| 0.4697 | 54.9474 | 261 | 0.8327 | 0.8528 |
|
126 |
+
| 0.4697 | 56.0 | 266 | 0.8268 | 0.8597 |
|
127 |
+
| 0.4482 | 56.8421 | 270 | 0.8188 | 0.8556 |
|
128 |
+
| 0.4482 | 57.8947 | 275 | 0.8171 | 0.8653 |
|
129 |
+
| 0.4436 | 58.9474 | 280 | 0.8133 | 0.8486 |
|
130 |
+
| 0.4436 | 60.0 | 285 | 0.8070 | 0.8639 |
|
131 |
+
| 0.4436 | 60.8421 | 289 | 0.7986 | 0.8542 |
|
132 |
+
| 0.4211 | 61.8947 | 294 | 0.7937 | 0.8597 |
|
133 |
+
| 0.4211 | 62.9474 | 299 | 0.7908 | 0.8611 |
|
134 |
+
| 0.4228 | 64.0 | 304 | 0.7952 | 0.8625 |
|
135 |
+
| 0.4228 | 64.8421 | 308 | 0.8010 | 0.8514 |
|
136 |
+
| 0.4046 | 65.8947 | 313 | 0.7975 | 0.8472 |
|
137 |
+
| 0.4046 | 66.9474 | 318 | 0.7927 | 0.8417 |
|
138 |
+
| 0.4048 | 68.0 | 323 | 0.7880 | 0.8556 |
|
139 |
+
| 0.4048 | 68.8421 | 327 | 0.7860 | 0.8514 |
|
140 |
+
| 0.3925 | 69.8947 | 332 | 0.7899 | 0.8403 |
|
141 |
+
| 0.3925 | 70.9474 | 337 | 0.7883 | 0.8417 |
|
142 |
+
| 0.3936 | 72.0 | 342 | 0.7885 | 0.8417 |
|
143 |
+
| 0.3936 | 72.8421 | 346 | 0.7874 | 0.8361 |
|
144 |
+
| 0.3985 | 73.8947 | 351 | 0.7832 | 0.8417 |
|
145 |
+
| 0.3985 | 74.9474 | 356 | 0.7787 | 0.8514 |
|
146 |
+
| 0.3849 | 76.0 | 361 | 0.7753 | 0.8486 |
|
147 |
+
| 0.3849 | 76.8421 | 365 | 0.7746 | 0.8514 |
|
148 |
+
| 0.3796 | 77.8947 | 370 | 0.7736 | 0.8542 |
|
149 |
+
| 0.3796 | 78.9474 | 375 | 0.7731 | 0.8528 |
|
150 |
+
| 0.3717 | 80.0 | 380 | 0.7715 | 0.8556 |
|
151 |
+
| 0.3717 | 80.8421 | 384 | 0.7709 | 0.8556 |
|
152 |
+
| 0.3717 | 81.8947 | 389 | 0.7706 | 0.8569 |
|
153 |
+
| 0.3802 | 82.9474 | 394 | 0.7704 | 0.8556 |
|
154 |
+
| 0.3802 | 84.0 | 399 | 0.7704 | 0.8542 |
|
155 |
+
| 0.3782 | 84.2105 | 400 | 0.7703 | 0.8542 |
|
156 |
|
157 |
|
158 |
### Framework versions
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 343291648
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:66cd604316d329088b4e28b91faeda5434062a63bceea19ff6379a979a77fee7
|
3 |
size 343291648
|