Model save
Browse files- README.md +160 -0
- model.safetensors +1 -1
README.md
ADDED
@@ -0,0 +1,160 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
library_name: transformers
|
3 |
+
license: apache-2.0
|
4 |
+
base_model: google/vit-base-patch16-224
|
5 |
+
tags:
|
6 |
+
- generated_from_trainer
|
7 |
+
metrics:
|
8 |
+
- accuracy
|
9 |
+
model-index:
|
10 |
+
- name: vit-base-cocoa
|
11 |
+
results: []
|
12 |
+
---
|
13 |
+
|
14 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
15 |
+
should probably proofread and complete it, then remove this comment. -->
|
16 |
+
|
17 |
+
# vit-base-cocoa
|
18 |
+
|
19 |
+
This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
|
20 |
+
It achieves the following results on the evaluation set:
|
21 |
+
- Loss: 0.5570
|
22 |
+
- Accuracy: 0.9350
|
23 |
+
|
24 |
+
## Model description
|
25 |
+
|
26 |
+
More information needed
|
27 |
+
|
28 |
+
## Intended uses & limitations
|
29 |
+
|
30 |
+
More information needed
|
31 |
+
|
32 |
+
## Training and evaluation data
|
33 |
+
|
34 |
+
More information needed
|
35 |
+
|
36 |
+
## Training procedure
|
37 |
+
|
38 |
+
### Training hyperparameters
|
39 |
+
|
40 |
+
The following hyperparameters were used during training:
|
41 |
+
- learning_rate: 2e-05
|
42 |
+
- train_batch_size: 8
|
43 |
+
- eval_batch_size: 8
|
44 |
+
- seed: 1337
|
45 |
+
- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
46 |
+
- lr_scheduler_type: linear
|
47 |
+
- num_epochs: 100.0
|
48 |
+
|
49 |
+
### Training results
|
50 |
+
|
51 |
+
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
52 |
+
|:-------------:|:-----:|:-----:|:---------------:|:--------:|
|
53 |
+
| 0.3733 | 1.0 | 196 | 0.3558 | 0.9025 |
|
54 |
+
| 0.3727 | 2.0 | 392 | 0.4098 | 0.8989 |
|
55 |
+
| 0.3901 | 3.0 | 588 | 0.2668 | 0.8989 |
|
56 |
+
| 0.3421 | 4.0 | 784 | 0.2612 | 0.9170 |
|
57 |
+
| 0.2703 | 5.0 | 980 | 0.2061 | 0.9278 |
|
58 |
+
| 0.1734 | 6.0 | 1176 | 0.2568 | 0.9278 |
|
59 |
+
| 0.1385 | 7.0 | 1372 | 0.3242 | 0.9206 |
|
60 |
+
| 0.3237 | 8.0 | 1568 | 0.2922 | 0.9386 |
|
61 |
+
| 0.236 | 9.0 | 1764 | 0.3044 | 0.9386 |
|
62 |
+
| 0.2124 | 10.0 | 1960 | 0.3848 | 0.9061 |
|
63 |
+
| 0.0454 | 11.0 | 2156 | 0.3527 | 0.9350 |
|
64 |
+
| 0.0756 | 12.0 | 2352 | 0.2844 | 0.9350 |
|
65 |
+
| 0.0605 | 13.0 | 2548 | 0.3077 | 0.9314 |
|
66 |
+
| 0.0214 | 14.0 | 2744 | 0.6295 | 0.9025 |
|
67 |
+
| 0.1816 | 15.0 | 2940 | 0.2996 | 0.9386 |
|
68 |
+
| 0.0338 | 16.0 | 3136 | 0.3597 | 0.9278 |
|
69 |
+
| 0.2136 | 17.0 | 3332 | 0.4070 | 0.9314 |
|
70 |
+
| 0.188 | 18.0 | 3528 | 0.3532 | 0.9458 |
|
71 |
+
| 0.0539 | 19.0 | 3724 | 0.3843 | 0.9386 |
|
72 |
+
| 0.0992 | 20.0 | 3920 | 0.3904 | 0.9422 |
|
73 |
+
| 0.0019 | 21.0 | 4116 | 0.3732 | 0.9458 |
|
74 |
+
| 0.0348 | 22.0 | 4312 | 0.4021 | 0.9386 |
|
75 |
+
| 0.0823 | 23.0 | 4508 | 0.4217 | 0.9350 |
|
76 |
+
| 0.1125 | 24.0 | 4704 | 0.4704 | 0.9097 |
|
77 |
+
| 0.0173 | 25.0 | 4900 | 0.3700 | 0.9350 |
|
78 |
+
| 0.0442 | 26.0 | 5096 | 0.3725 | 0.9314 |
|
79 |
+
| 0.0009 | 27.0 | 5292 | 0.4819 | 0.9278 |
|
80 |
+
| 0.0087 | 28.0 | 5488 | 0.6492 | 0.9170 |
|
81 |
+
| 0.0021 | 29.0 | 5684 | 0.5297 | 0.9242 |
|
82 |
+
| 0.2552 | 30.0 | 5880 | 0.4482 | 0.9314 |
|
83 |
+
| 0.0154 | 31.0 | 6076 | 0.6075 | 0.9242 |
|
84 |
+
| 0.0009 | 32.0 | 6272 | 0.4101 | 0.9350 |
|
85 |
+
| 0.1626 | 33.0 | 6468 | 0.4653 | 0.9350 |
|
86 |
+
| 0.0276 | 34.0 | 6664 | 0.4174 | 0.9386 |
|
87 |
+
| 0.0139 | 35.0 | 6860 | 0.3992 | 0.9422 |
|
88 |
+
| 0.0023 | 36.0 | 7056 | 0.6972 | 0.9170 |
|
89 |
+
| 0.1264 | 37.0 | 7252 | 0.4980 | 0.9314 |
|
90 |
+
| 0.0113 | 38.0 | 7448 | 0.7154 | 0.9170 |
|
91 |
+
| 0.0694 | 39.0 | 7644 | 0.5443 | 0.9242 |
|
92 |
+
| 0.0976 | 40.0 | 7840 | 0.3852 | 0.9350 |
|
93 |
+
| 0.1191 | 41.0 | 8036 | 0.5398 | 0.9242 |
|
94 |
+
| 0.1249 | 42.0 | 8232 | 0.6197 | 0.9170 |
|
95 |
+
| 0.0002 | 43.0 | 8428 | 0.6967 | 0.9134 |
|
96 |
+
| 0.1163 | 44.0 | 8624 | 0.5697 | 0.9242 |
|
97 |
+
| 0.0201 | 45.0 | 8820 | 0.7221 | 0.9134 |
|
98 |
+
| 0.0003 | 46.0 | 9016 | 0.5253 | 0.9314 |
|
99 |
+
| 0.0224 | 47.0 | 9212 | 0.3817 | 0.9495 |
|
100 |
+
| 0.0183 | 48.0 | 9408 | 0.4966 | 0.9242 |
|
101 |
+
| 0.0077 | 49.0 | 9604 | 0.4349 | 0.9458 |
|
102 |
+
| 0.0083 | 50.0 | 9800 | 0.5191 | 0.9242 |
|
103 |
+
| 0.0571 | 51.0 | 9996 | 0.5826 | 0.9206 |
|
104 |
+
| 0.0583 | 52.0 | 10192 | 0.5335 | 0.9170 |
|
105 |
+
| 0.0019 | 53.0 | 10388 | 0.5843 | 0.9206 |
|
106 |
+
| 0.0044 | 54.0 | 10584 | 0.5895 | 0.9206 |
|
107 |
+
| 0.0065 | 55.0 | 10780 | 0.4487 | 0.9350 |
|
108 |
+
| 0.0126 | 56.0 | 10976 | 0.6221 | 0.9314 |
|
109 |
+
| 0.0093 | 57.0 | 11172 | 0.5138 | 0.9314 |
|
110 |
+
| 0.0004 | 58.0 | 11368 | 0.5162 | 0.9314 |
|
111 |
+
| 0.0002 | 59.0 | 11564 | 0.4514 | 0.9350 |
|
112 |
+
| 0.1463 | 60.0 | 11760 | 0.4744 | 0.9386 |
|
113 |
+
| 0.0001 | 61.0 | 11956 | 0.5338 | 0.9314 |
|
114 |
+
| 0.0006 | 62.0 | 12152 | 0.5788 | 0.9278 |
|
115 |
+
| 0.0269 | 63.0 | 12348 | 0.5500 | 0.9278 |
|
116 |
+
| 0.1 | 64.0 | 12544 | 0.6467 | 0.9206 |
|
117 |
+
| 0.0004 | 65.0 | 12740 | 0.5828 | 0.9242 |
|
118 |
+
| 0.0001 | 66.0 | 12936 | 0.5283 | 0.9314 |
|
119 |
+
| 0.0001 | 67.0 | 13132 | 0.6212 | 0.9206 |
|
120 |
+
| 0.0002 | 68.0 | 13328 | 0.4973 | 0.9242 |
|
121 |
+
| 0.0058 | 69.0 | 13524 | 0.5021 | 0.9278 |
|
122 |
+
| 0.0605 | 70.0 | 13720 | 0.6982 | 0.9170 |
|
123 |
+
| 0.0006 | 71.0 | 13916 | 0.4602 | 0.9350 |
|
124 |
+
| 0.0021 | 72.0 | 14112 | 0.5595 | 0.9314 |
|
125 |
+
| 0.0004 | 73.0 | 14308 | 0.4366 | 0.9386 |
|
126 |
+
| 0.0124 | 74.0 | 14504 | 0.7612 | 0.9134 |
|
127 |
+
| 0.0284 | 75.0 | 14700 | 0.6054 | 0.9206 |
|
128 |
+
| 0.0001 | 76.0 | 14896 | 0.5922 | 0.9242 |
|
129 |
+
| 0.0119 | 77.0 | 15092 | 0.5496 | 0.9242 |
|
130 |
+
| 0.0006 | 78.0 | 15288 | 0.6327 | 0.9206 |
|
131 |
+
| 0.0711 | 79.0 | 15484 | 0.5177 | 0.9386 |
|
132 |
+
| 0.0001 | 80.0 | 15680 | 0.7391 | 0.9134 |
|
133 |
+
| 0.0985 | 81.0 | 15876 | 0.5683 | 0.9242 |
|
134 |
+
| 0.0001 | 82.0 | 16072 | 0.6106 | 0.9206 |
|
135 |
+
| 0.0 | 83.0 | 16268 | 0.6235 | 0.9242 |
|
136 |
+
| 0.0006 | 84.0 | 16464 | 0.7914 | 0.9061 |
|
137 |
+
| 0.0001 | 85.0 | 16660 | 0.5649 | 0.9314 |
|
138 |
+
| 0.0 | 86.0 | 16856 | 0.5512 | 0.9350 |
|
139 |
+
| 0.066 | 87.0 | 17052 | 0.5473 | 0.9350 |
|
140 |
+
| 0.0189 | 88.0 | 17248 | 0.4866 | 0.9386 |
|
141 |
+
| 0.0 | 89.0 | 17444 | 0.5136 | 0.9386 |
|
142 |
+
| 0.0001 | 90.0 | 17640 | 0.5246 | 0.9350 |
|
143 |
+
| 0.0001 | 91.0 | 17836 | 0.5626 | 0.9314 |
|
144 |
+
| 0.0037 | 92.0 | 18032 | 0.5335 | 0.9350 |
|
145 |
+
| 0.0999 | 93.0 | 18228 | 0.6357 | 0.9242 |
|
146 |
+
| 0.1124 | 94.0 | 18424 | 0.5905 | 0.9278 |
|
147 |
+
| 0.0175 | 95.0 | 18620 | 0.6618 | 0.9206 |
|
148 |
+
| 0.0001 | 96.0 | 18816 | 0.5588 | 0.9386 |
|
149 |
+
| 0.0259 | 97.0 | 19012 | 0.5549 | 0.9350 |
|
150 |
+
| 0.0001 | 98.0 | 19208 | 0.5599 | 0.9350 |
|
151 |
+
| 0.0285 | 99.0 | 19404 | 0.5517 | 0.9350 |
|
152 |
+
| 0.003 | 100.0 | 19600 | 0.5570 | 0.9350 |
|
153 |
+
|
154 |
+
|
155 |
+
### Framework versions
|
156 |
+
|
157 |
+
- Transformers 4.48.0.dev0
|
158 |
+
- Pytorch 2.5.1+cu124
|
159 |
+
- Datasets 3.1.0
|
160 |
+
- Tokenizers 0.21.0
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 343236280
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2c420f35c8d24d2a363f869ee42f80f9417c2175c8361e92f15d0a640722c24a
|
3 |
size 343236280
|