jtas commited on
Commit
b9a95d3
·
verified ·
1 Parent(s): 8272b01

End of training

Browse files
README.md ADDED
@@ -0,0 +1,108 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: google/vit-base-patch16-224-in21k
3
+ tags:
4
+ - generated_from_trainer
5
+ datasets:
6
+ - imagefolder
7
+ metrics:
8
+ - accuracy
9
+ - f1
10
+ model-index:
11
+ - name: emotion_classification
12
+ results:
13
+ - task:
14
+ name: Image Classification
15
+ type: image-classification
16
+ dataset:
17
+ name: imagefolder
18
+ type: imagefolder
19
+ config: default
20
+ split: train
21
+ args: default
22
+ metrics:
23
+ - name: Accuracy
24
+ type: accuracy
25
+ value: 0.65
26
+ - name: F1
27
+ type: f1
28
+ value: 0.6231481481481482
29
+ ---
30
+
31
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
32
+ should probably proofread and complete it, then remove this comment. -->
33
+
34
+ # emotion_classification
35
+
36
+ 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.
37
+ It achieves the following results on the evaluation set:
38
+ - Loss: 1.1136
39
+ - Accuracy: 0.65
40
+ - F1: 0.6231
41
+
42
+ ## Model description
43
+
44
+ More information needed
45
+
46
+ ## Intended uses & limitations
47
+
48
+ More information needed
49
+
50
+ ## Training and evaluation data
51
+
52
+ More information needed
53
+
54
+ ## Training procedure
55
+
56
+ ### Training hyperparameters
57
+
58
+ The following hyperparameters were used during training:
59
+ - learning_rate: 0.0001
60
+ - train_batch_size: 16
61
+ - eval_batch_size: 16
62
+ - seed: 45
63
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
64
+ - lr_scheduler_type: cosine_with_restarts
65
+ - num_epochs: 30
66
+
67
+ ### Training results
68
+
69
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
70
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
71
+ | 1.9172 | 1.0 | 43 | 1.5751 | 0.4333 | 0.3263 |
72
+ | 1.4505 | 2.0 | 86 | 1.3041 | 0.5333 | 0.4651 |
73
+ | 1.1121 | 3.0 | 129 | 1.2902 | 0.4833 | 0.4684 |
74
+ | 0.8491 | 4.0 | 172 | 1.2309 | 0.5167 | 0.4916 |
75
+ | 0.6168 | 5.0 | 215 | 1.2573 | 0.5583 | 0.5310 |
76
+ | 0.3953 | 6.0 | 258 | 1.1502 | 0.575 | 0.5401 |
77
+ | 0.3048 | 7.0 | 301 | 1.1136 | 0.65 | 0.6231 |
78
+ | 0.1875 | 8.0 | 344 | 1.4224 | 0.5667 | 0.5598 |
79
+ | 0.1277 | 9.0 | 387 | 1.3467 | 0.6167 | 0.6011 |
80
+ | 0.1123 | 10.0 | 430 | 1.5838 | 0.5833 | 0.5657 |
81
+ | 0.1123 | 11.0 | 473 | 1.5063 | 0.5833 | 0.5550 |
82
+ | 0.0694 | 12.0 | 516 | 1.7733 | 0.55 | 0.5320 |
83
+ | 0.0499 | 13.0 | 559 | 1.6329 | 0.5833 | 0.5536 |
84
+ | 0.0367 | 14.0 | 602 | 1.6878 | 0.5833 | 0.5685 |
85
+ | 0.0291 | 15.0 | 645 | 1.6855 | 0.575 | 0.5392 |
86
+ | 0.0284 | 16.0 | 688 | 1.7869 | 0.6083 | 0.5880 |
87
+ | 0.0316 | 17.0 | 731 | 1.5831 | 0.5917 | 0.5670 |
88
+ | 0.0273 | 18.0 | 774 | 1.5933 | 0.625 | 0.5984 |
89
+ | 0.0234 | 19.0 | 817 | 1.7830 | 0.5833 | 0.5652 |
90
+ | 0.0194 | 20.0 | 860 | 1.6804 | 0.6083 | 0.5878 |
91
+ | 0.0214 | 21.0 | 903 | 1.5962 | 0.6 | 0.5701 |
92
+ | 0.0204 | 22.0 | 946 | 1.5684 | 0.625 | 0.5992 |
93
+ | 0.0178 | 23.0 | 989 | 1.5924 | 0.625 | 0.5992 |
94
+ | 0.0173 | 24.0 | 1032 | 1.6228 | 0.6167 | 0.5933 |
95
+ | 0.016 | 25.0 | 1075 | 1.6177 | 0.6333 | 0.6073 |
96
+ | 0.016 | 26.0 | 1118 | 1.6268 | 0.625 | 0.6009 |
97
+ | 0.016 | 27.0 | 1161 | 1.6387 | 0.625 | 0.6009 |
98
+ | 0.0159 | 28.0 | 1204 | 1.6403 | 0.625 | 0.6009 |
99
+ | 0.0162 | 29.0 | 1247 | 1.6409 | 0.625 | 0.6009 |
100
+ | 0.018 | 30.0 | 1290 | 1.6412 | 0.625 | 0.6009 |
101
+
102
+
103
+ ### Framework versions
104
+
105
+ - Transformers 4.37.2
106
+ - Pytorch 2.1.2
107
+ - Datasets 2.16.1
108
+ - Tokenizers 0.15.1
all_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 30.0,
3
+ "eval_accuracy": 0.65,
4
+ "eval_f1": 0.6231481481481482,
5
+ "eval_loss": 1.1135584115982056,
6
+ "eval_runtime": 4.9712,
7
+ "eval_samples_per_second": 24.139,
8
+ "eval_steps_per_second": 1.609,
9
+ "train_loss": 0.25586533430934877,
10
+ "train_runtime": 2391.6751,
11
+ "train_samples_per_second": 8.53,
12
+ "train_steps_per_second": 0.539
13
+ }
config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.2,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.2,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "anger",
13
+ "1": "contempt",
14
+ "2": "disgust",
15
+ "3": "fear",
16
+ "4": "happy",
17
+ "5": "neutral",
18
+ "6": "sad",
19
+ "7": "surprise"
20
+ },
21
+ "image_size": 224,
22
+ "initializer_range": 0.02,
23
+ "intermediate_size": 3072,
24
+ "label2id": {
25
+ "anger": "0",
26
+ "contempt": "1",
27
+ "disgust": "2",
28
+ "fear": "3",
29
+ "happy": "4",
30
+ "neutral": "5",
31
+ "sad": "6",
32
+ "surprise": "7"
33
+ },
34
+ "layer_norm_eps": 1e-12,
35
+ "model_type": "vit",
36
+ "num_attention_heads": 12,
37
+ "num_channels": 3,
38
+ "num_hidden_layers": 12,
39
+ "patch_size": 16,
40
+ "problem_type": "single_label_classification",
41
+ "qkv_bias": true,
42
+ "torch_dtype": "float32",
43
+ "transformers_version": "4.37.2"
44
+ }
eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 30.0,
3
+ "eval_accuracy": 0.65,
4
+ "eval_f1": 0.6231481481481482,
5
+ "eval_loss": 1.1135584115982056,
6
+ "eval_runtime": 4.9712,
7
+ "eval_samples_per_second": 24.139,
8
+ "eval_steps_per_second": 1.609
9
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0bd0d898a5b7e5edf6f1193d7f3b97c960e7e4f013cd47a0c7190abe1bc6d4dc
3
+ size 343242432
preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_rescale": true,
4
+ "do_resize": true,
5
+ "image_mean": [
6
+ 0.5,
7
+ 0.5,
8
+ 0.5
9
+ ],
10
+ "image_processor_type": "ViTImageProcessor",
11
+ "image_std": [
12
+ 0.5,
13
+ 0.5,
14
+ 0.5
15
+ ],
16
+ "resample": 2,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 224,
20
+ "width": 224
21
+ }
22
+ }
train_results.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 30.0,
3
+ "train_loss": 0.25586533430934877,
4
+ "train_runtime": 2391.6751,
5
+ "train_samples_per_second": 8.53,
6
+ "train_steps_per_second": 0.539
7
+ }
trainer_state.json ADDED
@@ -0,0 +1,510 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 1.1135584115982056,
3
+ "best_model_checkpoint": "./model/emotion_classification/checkpoint-301",
4
+ "epoch": 30.0,
5
+ "eval_steps": 500,
6
+ "global_step": 1290,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 1.0,
13
+ "learning_rate": 9.972609476841367e-05,
14
+ "loss": 1.9172,
15
+ "step": 43
16
+ },
17
+ {
18
+ "epoch": 1.0,
19
+ "eval_accuracy": 0.43333333333333335,
20
+ "eval_f1": 0.32631946371225834,
21
+ "eval_loss": 1.5750515460968018,
22
+ "eval_runtime": 4.0633,
23
+ "eval_samples_per_second": 29.533,
24
+ "eval_steps_per_second": 1.969,
25
+ "step": 43
26
+ },
27
+ {
28
+ "epoch": 2.0,
29
+ "learning_rate": 9.890738003669029e-05,
30
+ "loss": 1.4505,
31
+ "step": 86
32
+ },
33
+ {
34
+ "epoch": 2.0,
35
+ "eval_accuracy": 0.5333333333333333,
36
+ "eval_f1": 0.4650626007978949,
37
+ "eval_loss": 1.3041330575942993,
38
+ "eval_runtime": 4.1206,
39
+ "eval_samples_per_second": 29.122,
40
+ "eval_steps_per_second": 1.941,
41
+ "step": 86
42
+ },
43
+ {
44
+ "epoch": 3.0,
45
+ "learning_rate": 9.755282581475769e-05,
46
+ "loss": 1.1121,
47
+ "step": 129
48
+ },
49
+ {
50
+ "epoch": 3.0,
51
+ "eval_accuracy": 0.48333333333333334,
52
+ "eval_f1": 0.46841408327694595,
53
+ "eval_loss": 1.2902252674102783,
54
+ "eval_runtime": 4.2082,
55
+ "eval_samples_per_second": 28.516,
56
+ "eval_steps_per_second": 1.901,
57
+ "step": 129
58
+ },
59
+ {
60
+ "epoch": 4.0,
61
+ "learning_rate": 9.567727288213005e-05,
62
+ "loss": 0.8491,
63
+ "step": 172
64
+ },
65
+ {
66
+ "epoch": 4.0,
67
+ "eval_accuracy": 0.5166666666666667,
68
+ "eval_f1": 0.49163216814145605,
69
+ "eval_loss": 1.2308896780014038,
70
+ "eval_runtime": 4.2459,
71
+ "eval_samples_per_second": 28.263,
72
+ "eval_steps_per_second": 1.884,
73
+ "step": 172
74
+ },
75
+ {
76
+ "epoch": 5.0,
77
+ "learning_rate": 9.330127018922194e-05,
78
+ "loss": 0.6168,
79
+ "step": 215
80
+ },
81
+ {
82
+ "epoch": 5.0,
83
+ "eval_accuracy": 0.5583333333333333,
84
+ "eval_f1": 0.5309547133900583,
85
+ "eval_loss": 1.2573224306106567,
86
+ "eval_runtime": 4.1217,
87
+ "eval_samples_per_second": 29.114,
88
+ "eval_steps_per_second": 1.941,
89
+ "step": 215
90
+ },
91
+ {
92
+ "epoch": 6.0,
93
+ "learning_rate": 9.045084971874738e-05,
94
+ "loss": 0.3953,
95
+ "step": 258
96
+ },
97
+ {
98
+ "epoch": 6.0,
99
+ "eval_accuracy": 0.575,
100
+ "eval_f1": 0.5400530867590878,
101
+ "eval_loss": 1.1502172946929932,
102
+ "eval_runtime": 4.3017,
103
+ "eval_samples_per_second": 27.896,
104
+ "eval_steps_per_second": 1.86,
105
+ "step": 258
106
+ },
107
+ {
108
+ "epoch": 7.0,
109
+ "learning_rate": 8.715724127386972e-05,
110
+ "loss": 0.3048,
111
+ "step": 301
112
+ },
113
+ {
114
+ "epoch": 7.0,
115
+ "eval_accuracy": 0.65,
116
+ "eval_f1": 0.6231481481481482,
117
+ "eval_loss": 1.1135584115982056,
118
+ "eval_runtime": 4.1683,
119
+ "eval_samples_per_second": 28.789,
120
+ "eval_steps_per_second": 1.919,
121
+ "step": 301
122
+ },
123
+ {
124
+ "epoch": 8.0,
125
+ "learning_rate": 8.345653031794292e-05,
126
+ "loss": 0.1875,
127
+ "step": 344
128
+ },
129
+ {
130
+ "epoch": 8.0,
131
+ "eval_accuracy": 0.5666666666666667,
132
+ "eval_f1": 0.5597741659473293,
133
+ "eval_loss": 1.4224319458007812,
134
+ "eval_runtime": 4.2212,
135
+ "eval_samples_per_second": 28.428,
136
+ "eval_steps_per_second": 1.895,
137
+ "step": 344
138
+ },
139
+ {
140
+ "epoch": 9.0,
141
+ "learning_rate": 7.938926261462366e-05,
142
+ "loss": 0.1277,
143
+ "step": 387
144
+ },
145
+ {
146
+ "epoch": 9.0,
147
+ "eval_accuracy": 0.6166666666666667,
148
+ "eval_f1": 0.6011135939243728,
149
+ "eval_loss": 1.346667766571045,
150
+ "eval_runtime": 3.955,
151
+ "eval_samples_per_second": 30.341,
152
+ "eval_steps_per_second": 2.023,
153
+ "step": 387
154
+ },
155
+ {
156
+ "epoch": 10.0,
157
+ "learning_rate": 7.500000000000001e-05,
158
+ "loss": 0.1123,
159
+ "step": 430
160
+ },
161
+ {
162
+ "epoch": 10.0,
163
+ "eval_accuracy": 0.5833333333333334,
164
+ "eval_f1": 0.5656809749645115,
165
+ "eval_loss": 1.583767294883728,
166
+ "eval_runtime": 4.0704,
167
+ "eval_samples_per_second": 29.481,
168
+ "eval_steps_per_second": 1.965,
169
+ "step": 430
170
+ },
171
+ {
172
+ "epoch": 11.0,
173
+ "learning_rate": 7.033683215379002e-05,
174
+ "loss": 0.1123,
175
+ "step": 473
176
+ },
177
+ {
178
+ "epoch": 11.0,
179
+ "eval_accuracy": 0.5833333333333334,
180
+ "eval_f1": 0.5549914858886633,
181
+ "eval_loss": 1.50627601146698,
182
+ "eval_runtime": 4.0673,
183
+ "eval_samples_per_second": 29.504,
184
+ "eval_steps_per_second": 1.967,
185
+ "step": 473
186
+ },
187
+ {
188
+ "epoch": 12.0,
189
+ "learning_rate": 6.545084971874738e-05,
190
+ "loss": 0.0694,
191
+ "step": 516
192
+ },
193
+ {
194
+ "epoch": 12.0,
195
+ "eval_accuracy": 0.55,
196
+ "eval_f1": 0.5320146001860588,
197
+ "eval_loss": 1.7733001708984375,
198
+ "eval_runtime": 4.0123,
199
+ "eval_samples_per_second": 29.908,
200
+ "eval_steps_per_second": 1.994,
201
+ "step": 516
202
+ },
203
+ {
204
+ "epoch": 13.0,
205
+ "learning_rate": 6.0395584540887963e-05,
206
+ "loss": 0.0499,
207
+ "step": 559
208
+ },
209
+ {
210
+ "epoch": 13.0,
211
+ "eval_accuracy": 0.5833333333333334,
212
+ "eval_f1": 0.5536173850790786,
213
+ "eval_loss": 1.6328585147857666,
214
+ "eval_runtime": 4.0403,
215
+ "eval_samples_per_second": 29.701,
216
+ "eval_steps_per_second": 1.98,
217
+ "step": 559
218
+ },
219
+ {
220
+ "epoch": 14.0,
221
+ "learning_rate": 5.522642316338268e-05,
222
+ "loss": 0.0367,
223
+ "step": 602
224
+ },
225
+ {
226
+ "epoch": 14.0,
227
+ "eval_accuracy": 0.5833333333333334,
228
+ "eval_f1": 0.5684853336495889,
229
+ "eval_loss": 1.6878242492675781,
230
+ "eval_runtime": 3.9845,
231
+ "eval_samples_per_second": 30.116,
232
+ "eval_steps_per_second": 2.008,
233
+ "step": 602
234
+ },
235
+ {
236
+ "epoch": 15.0,
237
+ "learning_rate": 5e-05,
238
+ "loss": 0.0291,
239
+ "step": 645
240
+ },
241
+ {
242
+ "epoch": 15.0,
243
+ "eval_accuracy": 0.575,
244
+ "eval_f1": 0.5392005606664051,
245
+ "eval_loss": 1.685531735420227,
246
+ "eval_runtime": 3.9802,
247
+ "eval_samples_per_second": 30.149,
248
+ "eval_steps_per_second": 2.01,
249
+ "step": 645
250
+ },
251
+ {
252
+ "epoch": 16.0,
253
+ "learning_rate": 4.477357683661734e-05,
254
+ "loss": 0.0284,
255
+ "step": 688
256
+ },
257
+ {
258
+ "epoch": 16.0,
259
+ "eval_accuracy": 0.6083333333333333,
260
+ "eval_f1": 0.5880117663277057,
261
+ "eval_loss": 1.7869312763214111,
262
+ "eval_runtime": 4.0749,
263
+ "eval_samples_per_second": 29.448,
264
+ "eval_steps_per_second": 1.963,
265
+ "step": 688
266
+ },
267
+ {
268
+ "epoch": 17.0,
269
+ "learning_rate": 3.960441545911204e-05,
270
+ "loss": 0.0316,
271
+ "step": 731
272
+ },
273
+ {
274
+ "epoch": 17.0,
275
+ "eval_accuracy": 0.5916666666666667,
276
+ "eval_f1": 0.566969594710963,
277
+ "eval_loss": 1.5830930471420288,
278
+ "eval_runtime": 4.0457,
279
+ "eval_samples_per_second": 29.661,
280
+ "eval_steps_per_second": 1.977,
281
+ "step": 731
282
+ },
283
+ {
284
+ "epoch": 18.0,
285
+ "learning_rate": 3.4549150281252636e-05,
286
+ "loss": 0.0273,
287
+ "step": 774
288
+ },
289
+ {
290
+ "epoch": 18.0,
291
+ "eval_accuracy": 0.625,
292
+ "eval_f1": 0.598442406793843,
293
+ "eval_loss": 1.5932706594467163,
294
+ "eval_runtime": 4.1291,
295
+ "eval_samples_per_second": 29.062,
296
+ "eval_steps_per_second": 1.937,
297
+ "step": 774
298
+ },
299
+ {
300
+ "epoch": 19.0,
301
+ "learning_rate": 2.9663167846209998e-05,
302
+ "loss": 0.0234,
303
+ "step": 817
304
+ },
305
+ {
306
+ "epoch": 19.0,
307
+ "eval_accuracy": 0.5833333333333334,
308
+ "eval_f1": 0.5652118457947398,
309
+ "eval_loss": 1.7830352783203125,
310
+ "eval_runtime": 4.2401,
311
+ "eval_samples_per_second": 28.301,
312
+ "eval_steps_per_second": 1.887,
313
+ "step": 817
314
+ },
315
+ {
316
+ "epoch": 20.0,
317
+ "learning_rate": 2.500000000000001e-05,
318
+ "loss": 0.0194,
319
+ "step": 860
320
+ },
321
+ {
322
+ "epoch": 20.0,
323
+ "eval_accuracy": 0.6083333333333333,
324
+ "eval_f1": 0.5878385871868214,
325
+ "eval_loss": 1.6804471015930176,
326
+ "eval_runtime": 3.9914,
327
+ "eval_samples_per_second": 30.065,
328
+ "eval_steps_per_second": 2.004,
329
+ "step": 860
330
+ },
331
+ {
332
+ "epoch": 21.0,
333
+ "learning_rate": 2.061073738537635e-05,
334
+ "loss": 0.0214,
335
+ "step": 903
336
+ },
337
+ {
338
+ "epoch": 21.0,
339
+ "eval_accuracy": 0.6,
340
+ "eval_f1": 0.5700534489379734,
341
+ "eval_loss": 1.596238374710083,
342
+ "eval_runtime": 4.0432,
343
+ "eval_samples_per_second": 29.68,
344
+ "eval_steps_per_second": 1.979,
345
+ "step": 903
346
+ },
347
+ {
348
+ "epoch": 22.0,
349
+ "learning_rate": 1.6543469682057106e-05,
350
+ "loss": 0.0204,
351
+ "step": 946
352
+ },
353
+ {
354
+ "epoch": 22.0,
355
+ "eval_accuracy": 0.625,
356
+ "eval_f1": 0.5992033196773662,
357
+ "eval_loss": 1.5684361457824707,
358
+ "eval_runtime": 4.0843,
359
+ "eval_samples_per_second": 29.381,
360
+ "eval_steps_per_second": 1.959,
361
+ "step": 946
362
+ },
363
+ {
364
+ "epoch": 23.0,
365
+ "learning_rate": 1.2842758726130283e-05,
366
+ "loss": 0.0178,
367
+ "step": 989
368
+ },
369
+ {
370
+ "epoch": 23.0,
371
+ "eval_accuracy": 0.625,
372
+ "eval_f1": 0.5992033196773662,
373
+ "eval_loss": 1.592431664466858,
374
+ "eval_runtime": 4.0813,
375
+ "eval_samples_per_second": 29.402,
376
+ "eval_steps_per_second": 1.96,
377
+ "step": 989
378
+ },
379
+ {
380
+ "epoch": 24.0,
381
+ "learning_rate": 9.549150281252633e-06,
382
+ "loss": 0.0173,
383
+ "step": 1032
384
+ },
385
+ {
386
+ "epoch": 24.0,
387
+ "eval_accuracy": 0.6166666666666667,
388
+ "eval_f1": 0.5932522595359776,
389
+ "eval_loss": 1.6227874755859375,
390
+ "eval_runtime": 4.2004,
391
+ "eval_samples_per_second": 28.569,
392
+ "eval_steps_per_second": 1.905,
393
+ "step": 1032
394
+ },
395
+ {
396
+ "epoch": 25.0,
397
+ "learning_rate": 6.698729810778065e-06,
398
+ "loss": 0.016,
399
+ "step": 1075
400
+ },
401
+ {
402
+ "epoch": 25.0,
403
+ "eval_accuracy": 0.6333333333333333,
404
+ "eval_f1": 0.6072678358063984,
405
+ "eval_loss": 1.6176973581314087,
406
+ "eval_runtime": 4.0151,
407
+ "eval_samples_per_second": 29.887,
408
+ "eval_steps_per_second": 1.992,
409
+ "step": 1075
410
+ },
411
+ {
412
+ "epoch": 26.0,
413
+ "learning_rate": 4.322727117869951e-06,
414
+ "loss": 0.016,
415
+ "step": 1118
416
+ },
417
+ {
418
+ "epoch": 26.0,
419
+ "eval_accuracy": 0.625,
420
+ "eval_f1": 0.6009141292059813,
421
+ "eval_loss": 1.6267857551574707,
422
+ "eval_runtime": 4.055,
423
+ "eval_samples_per_second": 29.593,
424
+ "eval_steps_per_second": 1.973,
425
+ "step": 1118
426
+ },
427
+ {
428
+ "epoch": 27.0,
429
+ "learning_rate": 2.4471741852423237e-06,
430
+ "loss": 0.016,
431
+ "step": 1161
432
+ },
433
+ {
434
+ "epoch": 27.0,
435
+ "eval_accuracy": 0.625,
436
+ "eval_f1": 0.6009141292059813,
437
+ "eval_loss": 1.6387226581573486,
438
+ "eval_runtime": 4.2902,
439
+ "eval_samples_per_second": 27.971,
440
+ "eval_steps_per_second": 1.865,
441
+ "step": 1161
442
+ },
443
+ {
444
+ "epoch": 28.0,
445
+ "learning_rate": 1.0926199633097157e-06,
446
+ "loss": 0.0159,
447
+ "step": 1204
448
+ },
449
+ {
450
+ "epoch": 28.0,
451
+ "eval_accuracy": 0.625,
452
+ "eval_f1": 0.6009141292059813,
453
+ "eval_loss": 1.6403223276138306,
454
+ "eval_runtime": 4.1003,
455
+ "eval_samples_per_second": 29.266,
456
+ "eval_steps_per_second": 1.951,
457
+ "step": 1204
458
+ },
459
+ {
460
+ "epoch": 29.0,
461
+ "learning_rate": 2.7390523158633554e-07,
462
+ "loss": 0.0162,
463
+ "step": 1247
464
+ },
465
+ {
466
+ "epoch": 29.0,
467
+ "eval_accuracy": 0.625,
468
+ "eval_f1": 0.6009141292059813,
469
+ "eval_loss": 1.6409173011779785,
470
+ "eval_runtime": 4.0023,
471
+ "eval_samples_per_second": 29.983,
472
+ "eval_steps_per_second": 1.999,
473
+ "step": 1247
474
+ },
475
+ {
476
+ "epoch": 30.0,
477
+ "learning_rate": 0.0,
478
+ "loss": 0.018,
479
+ "step": 1290
480
+ },
481
+ {
482
+ "epoch": 30.0,
483
+ "eval_accuracy": 0.625,
484
+ "eval_f1": 0.6009141292059813,
485
+ "eval_loss": 1.6411793231964111,
486
+ "eval_runtime": 4.269,
487
+ "eval_samples_per_second": 28.11,
488
+ "eval_steps_per_second": 1.874,
489
+ "step": 1290
490
+ },
491
+ {
492
+ "epoch": 30.0,
493
+ "step": 1290,
494
+ "total_flos": 1.5809215993675776e+18,
495
+ "train_loss": 0.25586533430934877,
496
+ "train_runtime": 2391.6751,
497
+ "train_samples_per_second": 8.53,
498
+ "train_steps_per_second": 0.539
499
+ }
500
+ ],
501
+ "logging_steps": 500,
502
+ "max_steps": 1290,
503
+ "num_input_tokens_seen": 0,
504
+ "num_train_epochs": 30,
505
+ "save_steps": 500,
506
+ "total_flos": 1.5809215993675776e+18,
507
+ "train_batch_size": 16,
508
+ "trial_name": null,
509
+ "trial_params": null
510
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b69240f8560664eb499d6b60b294aea2460b1992a069feb8f83ff6ee4803bc91
3
+ size 4664