atuo commited on
Commit
eb8f313
1 Parent(s): d0b4ec8

End of training

Browse files
README.md CHANGED
@@ -22,7 +22,7 @@ model-index:
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
- value: 0.7460682777138473
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +32,8 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  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.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.6395
36
- - Accuracy: 0.7461
37
 
38
  ## Model description
39
 
 
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
+ value: 0.7472190257000384
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
32
 
33
  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.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.6450
36
+ - Accuracy: 0.7472
37
 
38
  ## Model description
39
 
all_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.97,
3
+ "eval_accuracy": 0.7472190257000384,
4
+ "eval_loss": 0.6449919939041138,
5
+ "eval_runtime": 104.9261,
6
+ "eval_samples_per_second": 24.846,
7
+ "eval_steps_per_second": 0.782,
8
+ "total_flos": 1.8133380652001642e+19,
9
+ "train_loss": 0.6611971782204883,
10
+ "train_runtime": 13354.5549,
11
+ "train_samples_per_second": 17.568,
12
+ "train_steps_per_second": 0.137
13
+ }
eval_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.97,
3
+ "eval_accuracy": 0.7472190257000384,
4
+ "eval_loss": 0.6449919939041138,
5
+ "eval_runtime": 104.9261,
6
+ "eval_samples_per_second": 24.846,
7
+ "eval_steps_per_second": 0.782
8
+ }
runs/Jan22_11-44-19_55bf1a6762ea/events.out.tfevents.1705939369.55bf1a6762ea.11542.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4330468032d5e768a2e09bdb57fb94401c210bf1f47ce533a9e4d6ec53253d8f
3
+ size 411
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.97,
3
+ "total_flos": 1.8133380652001642e+19,
4
+ "train_loss": 0.6611971782204883,
5
+ "train_runtime": 13354.5549,
6
+ "train_samples_per_second": 17.568,
7
+ "train_steps_per_second": 0.137
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,1218 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.7472190257000384,
3
+ "best_model_checkpoint": "vit-base-patch16-224-in21k-finetuned-crop-classification/checkpoint-1468",
4
+ "epoch": 9.97275204359673,
5
+ "eval_steps": 500,
6
+ "global_step": 1830,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.05,
13
+ "learning_rate": 2.73224043715847e-06,
14
+ "loss": 1.6369,
15
+ "step": 10
16
+ },
17
+ {
18
+ "epoch": 0.11,
19
+ "learning_rate": 5.46448087431694e-06,
20
+ "loss": 1.6111,
21
+ "step": 20
22
+ },
23
+ {
24
+ "epoch": 0.16,
25
+ "learning_rate": 8.196721311475409e-06,
26
+ "loss": 1.5398,
27
+ "step": 30
28
+ },
29
+ {
30
+ "epoch": 0.22,
31
+ "learning_rate": 1.092896174863388e-05,
32
+ "loss": 1.4457,
33
+ "step": 40
34
+ },
35
+ {
36
+ "epoch": 0.27,
37
+ "learning_rate": 1.3661202185792351e-05,
38
+ "loss": 1.3015,
39
+ "step": 50
40
+ },
41
+ {
42
+ "epoch": 0.33,
43
+ "learning_rate": 1.6393442622950818e-05,
44
+ "loss": 1.1684,
45
+ "step": 60
46
+ },
47
+ {
48
+ "epoch": 0.38,
49
+ "learning_rate": 1.912568306010929e-05,
50
+ "loss": 1.0962,
51
+ "step": 70
52
+ },
53
+ {
54
+ "epoch": 0.44,
55
+ "learning_rate": 2.185792349726776e-05,
56
+ "loss": 1.0177,
57
+ "step": 80
58
+ },
59
+ {
60
+ "epoch": 0.49,
61
+ "learning_rate": 2.459016393442623e-05,
62
+ "loss": 0.9627,
63
+ "step": 90
64
+ },
65
+ {
66
+ "epoch": 0.54,
67
+ "learning_rate": 2.7322404371584703e-05,
68
+ "loss": 0.8997,
69
+ "step": 100
70
+ },
71
+ {
72
+ "epoch": 0.6,
73
+ "learning_rate": 3.005464480874317e-05,
74
+ "loss": 0.8822,
75
+ "step": 110
76
+ },
77
+ {
78
+ "epoch": 0.65,
79
+ "learning_rate": 3.2786885245901635e-05,
80
+ "loss": 0.8726,
81
+ "step": 120
82
+ },
83
+ {
84
+ "epoch": 0.71,
85
+ "learning_rate": 3.551912568306011e-05,
86
+ "loss": 0.8288,
87
+ "step": 130
88
+ },
89
+ {
90
+ "epoch": 0.76,
91
+ "learning_rate": 3.825136612021858e-05,
92
+ "loss": 0.8337,
93
+ "step": 140
94
+ },
95
+ {
96
+ "epoch": 0.82,
97
+ "learning_rate": 4.098360655737705e-05,
98
+ "loss": 0.829,
99
+ "step": 150
100
+ },
101
+ {
102
+ "epoch": 0.87,
103
+ "learning_rate": 4.371584699453552e-05,
104
+ "loss": 0.7949,
105
+ "step": 160
106
+ },
107
+ {
108
+ "epoch": 0.93,
109
+ "learning_rate": 4.644808743169399e-05,
110
+ "loss": 0.7856,
111
+ "step": 170
112
+ },
113
+ {
114
+ "epoch": 0.98,
115
+ "learning_rate": 4.918032786885246e-05,
116
+ "loss": 0.8031,
117
+ "step": 180
118
+ },
119
+ {
120
+ "epoch": 1.0,
121
+ "eval_accuracy": 0.7050249328730341,
122
+ "eval_loss": 0.7602581977844238,
123
+ "eval_runtime": 93.216,
124
+ "eval_samples_per_second": 27.967,
125
+ "eval_steps_per_second": 0.88,
126
+ "step": 183
127
+ },
128
+ {
129
+ "epoch": 1.04,
130
+ "learning_rate": 4.9787492410443234e-05,
131
+ "loss": 0.78,
132
+ "step": 190
133
+ },
134
+ {
135
+ "epoch": 1.09,
136
+ "learning_rate": 4.948391013964785e-05,
137
+ "loss": 0.7994,
138
+ "step": 200
139
+ },
140
+ {
141
+ "epoch": 1.14,
142
+ "learning_rate": 4.918032786885246e-05,
143
+ "loss": 0.7665,
144
+ "step": 210
145
+ },
146
+ {
147
+ "epoch": 1.2,
148
+ "learning_rate": 4.8876745598057074e-05,
149
+ "loss": 0.7866,
150
+ "step": 220
151
+ },
152
+ {
153
+ "epoch": 1.25,
154
+ "learning_rate": 4.857316332726169e-05,
155
+ "loss": 0.7733,
156
+ "step": 230
157
+ },
158
+ {
159
+ "epoch": 1.31,
160
+ "learning_rate": 4.8269581056466304e-05,
161
+ "loss": 0.7504,
162
+ "step": 240
163
+ },
164
+ {
165
+ "epoch": 1.36,
166
+ "learning_rate": 4.7965998785670915e-05,
167
+ "loss": 0.7805,
168
+ "step": 250
169
+ },
170
+ {
171
+ "epoch": 1.42,
172
+ "learning_rate": 4.766241651487553e-05,
173
+ "loss": 0.7671,
174
+ "step": 260
175
+ },
176
+ {
177
+ "epoch": 1.47,
178
+ "learning_rate": 4.7358834244080144e-05,
179
+ "loss": 0.7012,
180
+ "step": 270
181
+ },
182
+ {
183
+ "epoch": 1.53,
184
+ "learning_rate": 4.705525197328476e-05,
185
+ "loss": 0.7305,
186
+ "step": 280
187
+ },
188
+ {
189
+ "epoch": 1.58,
190
+ "learning_rate": 4.6751669702489374e-05,
191
+ "loss": 0.7441,
192
+ "step": 290
193
+ },
194
+ {
195
+ "epoch": 1.63,
196
+ "learning_rate": 4.644808743169399e-05,
197
+ "loss": 0.7758,
198
+ "step": 300
199
+ },
200
+ {
201
+ "epoch": 1.69,
202
+ "learning_rate": 4.61445051608986e-05,
203
+ "loss": 0.731,
204
+ "step": 310
205
+ },
206
+ {
207
+ "epoch": 1.74,
208
+ "learning_rate": 4.584092289010322e-05,
209
+ "loss": 0.7303,
210
+ "step": 320
211
+ },
212
+ {
213
+ "epoch": 1.8,
214
+ "learning_rate": 4.553734061930783e-05,
215
+ "loss": 0.7265,
216
+ "step": 330
217
+ },
218
+ {
219
+ "epoch": 1.85,
220
+ "learning_rate": 4.523375834851245e-05,
221
+ "loss": 0.7385,
222
+ "step": 340
223
+ },
224
+ {
225
+ "epoch": 1.91,
226
+ "learning_rate": 4.493017607771706e-05,
227
+ "loss": 0.7396,
228
+ "step": 350
229
+ },
230
+ {
231
+ "epoch": 1.96,
232
+ "learning_rate": 4.462659380692168e-05,
233
+ "loss": 0.7311,
234
+ "step": 360
235
+ },
236
+ {
237
+ "epoch": 2.0,
238
+ "eval_accuracy": 0.7249712313003452,
239
+ "eval_loss": 0.7046979069709778,
240
+ "eval_runtime": 93.2263,
241
+ "eval_samples_per_second": 27.964,
242
+ "eval_steps_per_second": 0.88,
243
+ "step": 367
244
+ },
245
+ {
246
+ "epoch": 2.02,
247
+ "learning_rate": 4.432301153612629e-05,
248
+ "loss": 0.7129,
249
+ "step": 370
250
+ },
251
+ {
252
+ "epoch": 2.07,
253
+ "learning_rate": 4.401942926533091e-05,
254
+ "loss": 0.7103,
255
+ "step": 380
256
+ },
257
+ {
258
+ "epoch": 2.13,
259
+ "learning_rate": 4.371584699453552e-05,
260
+ "loss": 0.6994,
261
+ "step": 390
262
+ },
263
+ {
264
+ "epoch": 2.18,
265
+ "learning_rate": 4.341226472374014e-05,
266
+ "loss": 0.6637,
267
+ "step": 400
268
+ },
269
+ {
270
+ "epoch": 2.23,
271
+ "learning_rate": 4.310868245294475e-05,
272
+ "loss": 0.7107,
273
+ "step": 410
274
+ },
275
+ {
276
+ "epoch": 2.29,
277
+ "learning_rate": 4.280510018214937e-05,
278
+ "loss": 0.7326,
279
+ "step": 420
280
+ },
281
+ {
282
+ "epoch": 2.34,
283
+ "learning_rate": 4.250151791135398e-05,
284
+ "loss": 0.7229,
285
+ "step": 430
286
+ },
287
+ {
288
+ "epoch": 2.4,
289
+ "learning_rate": 4.21979356405586e-05,
290
+ "loss": 0.7064,
291
+ "step": 440
292
+ },
293
+ {
294
+ "epoch": 2.45,
295
+ "learning_rate": 4.189435336976321e-05,
296
+ "loss": 0.6781,
297
+ "step": 450
298
+ },
299
+ {
300
+ "epoch": 2.51,
301
+ "learning_rate": 4.1590771098967827e-05,
302
+ "loss": 0.7207,
303
+ "step": 460
304
+ },
305
+ {
306
+ "epoch": 2.56,
307
+ "learning_rate": 4.128718882817244e-05,
308
+ "loss": 0.7244,
309
+ "step": 470
310
+ },
311
+ {
312
+ "epoch": 2.62,
313
+ "learning_rate": 4.098360655737705e-05,
314
+ "loss": 0.6972,
315
+ "step": 480
316
+ },
317
+ {
318
+ "epoch": 2.67,
319
+ "learning_rate": 4.068002428658167e-05,
320
+ "loss": 0.7148,
321
+ "step": 490
322
+ },
323
+ {
324
+ "epoch": 2.72,
325
+ "learning_rate": 4.037644201578628e-05,
326
+ "loss": 0.7168,
327
+ "step": 500
328
+ },
329
+ {
330
+ "epoch": 2.78,
331
+ "learning_rate": 4.007285974499089e-05,
332
+ "loss": 0.7146,
333
+ "step": 510
334
+ },
335
+ {
336
+ "epoch": 2.83,
337
+ "learning_rate": 3.976927747419551e-05,
338
+ "loss": 0.6583,
339
+ "step": 520
340
+ },
341
+ {
342
+ "epoch": 2.89,
343
+ "learning_rate": 3.946569520340012e-05,
344
+ "loss": 0.7119,
345
+ "step": 530
346
+ },
347
+ {
348
+ "epoch": 2.94,
349
+ "learning_rate": 3.916211293260474e-05,
350
+ "loss": 0.6891,
351
+ "step": 540
352
+ },
353
+ {
354
+ "epoch": 3.0,
355
+ "learning_rate": 3.885853066180935e-05,
356
+ "loss": 0.7144,
357
+ "step": 550
358
+ },
359
+ {
360
+ "epoch": 3.0,
361
+ "eval_accuracy": 0.7211354046797085,
362
+ "eval_loss": 0.6968041062355042,
363
+ "eval_runtime": 91.8623,
364
+ "eval_samples_per_second": 28.379,
365
+ "eval_steps_per_second": 0.893,
366
+ "step": 550
367
+ },
368
+ {
369
+ "epoch": 3.05,
370
+ "learning_rate": 3.8554948391013967e-05,
371
+ "loss": 0.7037,
372
+ "step": 560
373
+ },
374
+ {
375
+ "epoch": 3.11,
376
+ "learning_rate": 3.825136612021858e-05,
377
+ "loss": 0.6613,
378
+ "step": 570
379
+ },
380
+ {
381
+ "epoch": 3.16,
382
+ "learning_rate": 3.7947783849423196e-05,
383
+ "loss": 0.6462,
384
+ "step": 580
385
+ },
386
+ {
387
+ "epoch": 3.22,
388
+ "learning_rate": 3.764420157862781e-05,
389
+ "loss": 0.6484,
390
+ "step": 590
391
+ },
392
+ {
393
+ "epoch": 3.27,
394
+ "learning_rate": 3.7340619307832425e-05,
395
+ "loss": 0.6747,
396
+ "step": 600
397
+ },
398
+ {
399
+ "epoch": 3.32,
400
+ "learning_rate": 3.7037037037037037e-05,
401
+ "loss": 0.7036,
402
+ "step": 610
403
+ },
404
+ {
405
+ "epoch": 3.38,
406
+ "learning_rate": 3.6733454766241655e-05,
407
+ "loss": 0.6444,
408
+ "step": 620
409
+ },
410
+ {
411
+ "epoch": 3.43,
412
+ "learning_rate": 3.6429872495446266e-05,
413
+ "loss": 0.6337,
414
+ "step": 630
415
+ },
416
+ {
417
+ "epoch": 3.49,
418
+ "learning_rate": 3.6126290224650884e-05,
419
+ "loss": 0.6715,
420
+ "step": 640
421
+ },
422
+ {
423
+ "epoch": 3.54,
424
+ "learning_rate": 3.5822707953855495e-05,
425
+ "loss": 0.6583,
426
+ "step": 650
427
+ },
428
+ {
429
+ "epoch": 3.6,
430
+ "learning_rate": 3.551912568306011e-05,
431
+ "loss": 0.663,
432
+ "step": 660
433
+ },
434
+ {
435
+ "epoch": 3.65,
436
+ "learning_rate": 3.5215543412264725e-05,
437
+ "loss": 0.6901,
438
+ "step": 670
439
+ },
440
+ {
441
+ "epoch": 3.71,
442
+ "learning_rate": 3.491196114146934e-05,
443
+ "loss": 0.6582,
444
+ "step": 680
445
+ },
446
+ {
447
+ "epoch": 3.76,
448
+ "learning_rate": 3.4608378870673954e-05,
449
+ "loss": 0.6868,
450
+ "step": 690
451
+ },
452
+ {
453
+ "epoch": 3.81,
454
+ "learning_rate": 3.430479659987857e-05,
455
+ "loss": 0.7168,
456
+ "step": 700
457
+ },
458
+ {
459
+ "epoch": 3.87,
460
+ "learning_rate": 3.400121432908318e-05,
461
+ "loss": 0.6522,
462
+ "step": 710
463
+ },
464
+ {
465
+ "epoch": 3.92,
466
+ "learning_rate": 3.36976320582878e-05,
467
+ "loss": 0.6337,
468
+ "step": 720
469
+ },
470
+ {
471
+ "epoch": 3.98,
472
+ "learning_rate": 3.339404978749241e-05,
473
+ "loss": 0.6516,
474
+ "step": 730
475
+ },
476
+ {
477
+ "epoch": 4.0,
478
+ "eval_accuracy": 0.7376294591484465,
479
+ "eval_loss": 0.6568663716316223,
480
+ "eval_runtime": 93.5303,
481
+ "eval_samples_per_second": 27.873,
482
+ "eval_steps_per_second": 0.877,
483
+ "step": 734
484
+ },
485
+ {
486
+ "epoch": 4.03,
487
+ "learning_rate": 3.3090467516697024e-05,
488
+ "loss": 0.6547,
489
+ "step": 740
490
+ },
491
+ {
492
+ "epoch": 4.09,
493
+ "learning_rate": 3.2786885245901635e-05,
494
+ "loss": 0.6444,
495
+ "step": 750
496
+ },
497
+ {
498
+ "epoch": 4.14,
499
+ "learning_rate": 3.248330297510625e-05,
500
+ "loss": 0.6687,
501
+ "step": 760
502
+ },
503
+ {
504
+ "epoch": 4.2,
505
+ "learning_rate": 3.2179720704310865e-05,
506
+ "loss": 0.6634,
507
+ "step": 770
508
+ },
509
+ {
510
+ "epoch": 4.25,
511
+ "learning_rate": 3.187613843351548e-05,
512
+ "loss": 0.6334,
513
+ "step": 780
514
+ },
515
+ {
516
+ "epoch": 4.31,
517
+ "learning_rate": 3.1572556162720094e-05,
518
+ "loss": 0.6095,
519
+ "step": 790
520
+ },
521
+ {
522
+ "epoch": 4.36,
523
+ "learning_rate": 3.126897389192471e-05,
524
+ "loss": 0.6467,
525
+ "step": 800
526
+ },
527
+ {
528
+ "epoch": 4.41,
529
+ "learning_rate": 3.096539162112932e-05,
530
+ "loss": 0.6532,
531
+ "step": 810
532
+ },
533
+ {
534
+ "epoch": 4.47,
535
+ "learning_rate": 3.066180935033394e-05,
536
+ "loss": 0.6544,
537
+ "step": 820
538
+ },
539
+ {
540
+ "epoch": 4.52,
541
+ "learning_rate": 3.0358227079538553e-05,
542
+ "loss": 0.6161,
543
+ "step": 830
544
+ },
545
+ {
546
+ "epoch": 4.58,
547
+ "learning_rate": 3.005464480874317e-05,
548
+ "loss": 0.6092,
549
+ "step": 840
550
+ },
551
+ {
552
+ "epoch": 4.63,
553
+ "learning_rate": 2.9751062537947782e-05,
554
+ "loss": 0.6323,
555
+ "step": 850
556
+ },
557
+ {
558
+ "epoch": 4.69,
559
+ "learning_rate": 2.94474802671524e-05,
560
+ "loss": 0.6593,
561
+ "step": 860
562
+ },
563
+ {
564
+ "epoch": 4.74,
565
+ "learning_rate": 2.9143897996357018e-05,
566
+ "loss": 0.6493,
567
+ "step": 870
568
+ },
569
+ {
570
+ "epoch": 4.8,
571
+ "learning_rate": 2.884031572556163e-05,
572
+ "loss": 0.6475,
573
+ "step": 880
574
+ },
575
+ {
576
+ "epoch": 4.85,
577
+ "learning_rate": 2.8536733454766244e-05,
578
+ "loss": 0.6196,
579
+ "step": 890
580
+ },
581
+ {
582
+ "epoch": 4.9,
583
+ "learning_rate": 2.823315118397086e-05,
584
+ "loss": 0.6257,
585
+ "step": 900
586
+ },
587
+ {
588
+ "epoch": 4.96,
589
+ "learning_rate": 2.7929568913175473e-05,
590
+ "loss": 0.6371,
591
+ "step": 910
592
+ },
593
+ {
594
+ "epoch": 5.0,
595
+ "eval_accuracy": 0.7376294591484465,
596
+ "eval_loss": 0.6482810974121094,
597
+ "eval_runtime": 93.1536,
598
+ "eval_samples_per_second": 27.986,
599
+ "eval_steps_per_second": 0.88,
600
+ "step": 917
601
+ },
602
+ {
603
+ "epoch": 5.01,
604
+ "learning_rate": 2.7625986642380085e-05,
605
+ "loss": 0.6049,
606
+ "step": 920
607
+ },
608
+ {
609
+ "epoch": 5.07,
610
+ "learning_rate": 2.7322404371584703e-05,
611
+ "loss": 0.6072,
612
+ "step": 930
613
+ },
614
+ {
615
+ "epoch": 5.12,
616
+ "learning_rate": 2.7018822100789314e-05,
617
+ "loss": 0.5548,
618
+ "step": 940
619
+ },
620
+ {
621
+ "epoch": 5.18,
622
+ "learning_rate": 2.6715239829993932e-05,
623
+ "loss": 0.5689,
624
+ "step": 950
625
+ },
626
+ {
627
+ "epoch": 5.23,
628
+ "learning_rate": 2.6411657559198543e-05,
629
+ "loss": 0.6459,
630
+ "step": 960
631
+ },
632
+ {
633
+ "epoch": 5.29,
634
+ "learning_rate": 2.610807528840316e-05,
635
+ "loss": 0.6033,
636
+ "step": 970
637
+ },
638
+ {
639
+ "epoch": 5.34,
640
+ "learning_rate": 2.5804493017607773e-05,
641
+ "loss": 0.5981,
642
+ "step": 980
643
+ },
644
+ {
645
+ "epoch": 5.4,
646
+ "learning_rate": 2.550091074681239e-05,
647
+ "loss": 0.6201,
648
+ "step": 990
649
+ },
650
+ {
651
+ "epoch": 5.45,
652
+ "learning_rate": 2.5197328476017002e-05,
653
+ "loss": 0.6121,
654
+ "step": 1000
655
+ },
656
+ {
657
+ "epoch": 5.5,
658
+ "learning_rate": 2.4893746205221617e-05,
659
+ "loss": 0.6147,
660
+ "step": 1010
661
+ },
662
+ {
663
+ "epoch": 5.56,
664
+ "learning_rate": 2.459016393442623e-05,
665
+ "loss": 0.6007,
666
+ "step": 1020
667
+ },
668
+ {
669
+ "epoch": 5.61,
670
+ "learning_rate": 2.4286581663630846e-05,
671
+ "loss": 0.6087,
672
+ "step": 1030
673
+ },
674
+ {
675
+ "epoch": 5.67,
676
+ "learning_rate": 2.3982999392835457e-05,
677
+ "loss": 0.6195,
678
+ "step": 1040
679
+ },
680
+ {
681
+ "epoch": 5.72,
682
+ "learning_rate": 2.3679417122040072e-05,
683
+ "loss": 0.5848,
684
+ "step": 1050
685
+ },
686
+ {
687
+ "epoch": 5.78,
688
+ "learning_rate": 2.3375834851244687e-05,
689
+ "loss": 0.6066,
690
+ "step": 1060
691
+ },
692
+ {
693
+ "epoch": 5.83,
694
+ "learning_rate": 2.30722525804493e-05,
695
+ "loss": 0.6386,
696
+ "step": 1070
697
+ },
698
+ {
699
+ "epoch": 5.89,
700
+ "learning_rate": 2.2768670309653916e-05,
701
+ "loss": 0.6196,
702
+ "step": 1080
703
+ },
704
+ {
705
+ "epoch": 5.94,
706
+ "learning_rate": 2.246508803885853e-05,
707
+ "loss": 0.6127,
708
+ "step": 1090
709
+ },
710
+ {
711
+ "epoch": 5.99,
712
+ "learning_rate": 2.2161505768063146e-05,
713
+ "loss": 0.6246,
714
+ "step": 1100
715
+ },
716
+ {
717
+ "epoch": 6.0,
718
+ "eval_accuracy": 0.7364787111622555,
719
+ "eval_loss": 0.6492410898208618,
720
+ "eval_runtime": 92.6582,
721
+ "eval_samples_per_second": 28.136,
722
+ "eval_steps_per_second": 0.885,
723
+ "step": 1101
724
+ },
725
+ {
726
+ "epoch": 6.05,
727
+ "learning_rate": 2.185792349726776e-05,
728
+ "loss": 0.5467,
729
+ "step": 1110
730
+ },
731
+ {
732
+ "epoch": 6.1,
733
+ "learning_rate": 2.1554341226472375e-05,
734
+ "loss": 0.577,
735
+ "step": 1120
736
+ },
737
+ {
738
+ "epoch": 6.16,
739
+ "learning_rate": 2.125075895567699e-05,
740
+ "loss": 0.6047,
741
+ "step": 1130
742
+ },
743
+ {
744
+ "epoch": 6.21,
745
+ "learning_rate": 2.0947176684881604e-05,
746
+ "loss": 0.5822,
747
+ "step": 1140
748
+ },
749
+ {
750
+ "epoch": 6.27,
751
+ "learning_rate": 2.064359441408622e-05,
752
+ "loss": 0.605,
753
+ "step": 1150
754
+ },
755
+ {
756
+ "epoch": 6.32,
757
+ "learning_rate": 2.0340012143290834e-05,
758
+ "loss": 0.5743,
759
+ "step": 1160
760
+ },
761
+ {
762
+ "epoch": 6.38,
763
+ "learning_rate": 2.0036429872495445e-05,
764
+ "loss": 0.5658,
765
+ "step": 1170
766
+ },
767
+ {
768
+ "epoch": 6.43,
769
+ "learning_rate": 1.973284760170006e-05,
770
+ "loss": 0.5686,
771
+ "step": 1180
772
+ },
773
+ {
774
+ "epoch": 6.49,
775
+ "learning_rate": 1.9429265330904674e-05,
776
+ "loss": 0.5884,
777
+ "step": 1190
778
+ },
779
+ {
780
+ "epoch": 6.54,
781
+ "learning_rate": 1.912568306010929e-05,
782
+ "loss": 0.544,
783
+ "step": 1200
784
+ },
785
+ {
786
+ "epoch": 6.59,
787
+ "learning_rate": 1.8822100789313904e-05,
788
+ "loss": 0.5888,
789
+ "step": 1210
790
+ },
791
+ {
792
+ "epoch": 6.65,
793
+ "learning_rate": 1.8518518518518518e-05,
794
+ "loss": 0.5377,
795
+ "step": 1220
796
+ },
797
+ {
798
+ "epoch": 6.7,
799
+ "learning_rate": 1.8214936247723133e-05,
800
+ "loss": 0.5665,
801
+ "step": 1230
802
+ },
803
+ {
804
+ "epoch": 6.76,
805
+ "learning_rate": 1.7911353976927748e-05,
806
+ "loss": 0.6032,
807
+ "step": 1240
808
+ },
809
+ {
810
+ "epoch": 6.81,
811
+ "learning_rate": 1.7607771706132362e-05,
812
+ "loss": 0.5905,
813
+ "step": 1250
814
+ },
815
+ {
816
+ "epoch": 6.87,
817
+ "learning_rate": 1.7304189435336977e-05,
818
+ "loss": 0.5831,
819
+ "step": 1260
820
+ },
821
+ {
822
+ "epoch": 6.92,
823
+ "learning_rate": 1.700060716454159e-05,
824
+ "loss": 0.5762,
825
+ "step": 1270
826
+ },
827
+ {
828
+ "epoch": 6.98,
829
+ "learning_rate": 1.6697024893746206e-05,
830
+ "loss": 0.5659,
831
+ "step": 1280
832
+ },
833
+ {
834
+ "epoch": 7.0,
835
+ "eval_accuracy": 0.7410817031070196,
836
+ "eval_loss": 0.6481292843818665,
837
+ "eval_runtime": 93.3242,
838
+ "eval_samples_per_second": 27.935,
839
+ "eval_steps_per_second": 0.879,
840
+ "step": 1284
841
+ },
842
+ {
843
+ "epoch": 7.03,
844
+ "learning_rate": 1.6393442622950818e-05,
845
+ "loss": 0.5738,
846
+ "step": 1290
847
+ },
848
+ {
849
+ "epoch": 7.08,
850
+ "learning_rate": 1.6089860352155432e-05,
851
+ "loss": 0.5339,
852
+ "step": 1300
853
+ },
854
+ {
855
+ "epoch": 7.14,
856
+ "learning_rate": 1.5786278081360047e-05,
857
+ "loss": 0.5405,
858
+ "step": 1310
859
+ },
860
+ {
861
+ "epoch": 7.19,
862
+ "learning_rate": 1.548269581056466e-05,
863
+ "loss": 0.5792,
864
+ "step": 1320
865
+ },
866
+ {
867
+ "epoch": 7.25,
868
+ "learning_rate": 1.5179113539769276e-05,
869
+ "loss": 0.5677,
870
+ "step": 1330
871
+ },
872
+ {
873
+ "epoch": 7.3,
874
+ "learning_rate": 1.4875531268973891e-05,
875
+ "loss": 0.5319,
876
+ "step": 1340
877
+ },
878
+ {
879
+ "epoch": 7.36,
880
+ "learning_rate": 1.4571948998178509e-05,
881
+ "loss": 0.5693,
882
+ "step": 1350
883
+ },
884
+ {
885
+ "epoch": 7.41,
886
+ "learning_rate": 1.4268366727383122e-05,
887
+ "loss": 0.5532,
888
+ "step": 1360
889
+ },
890
+ {
891
+ "epoch": 7.47,
892
+ "learning_rate": 1.3964784456587737e-05,
893
+ "loss": 0.565,
894
+ "step": 1370
895
+ },
896
+ {
897
+ "epoch": 7.52,
898
+ "learning_rate": 1.3661202185792351e-05,
899
+ "loss": 0.536,
900
+ "step": 1380
901
+ },
902
+ {
903
+ "epoch": 7.57,
904
+ "learning_rate": 1.3357619914996966e-05,
905
+ "loss": 0.5408,
906
+ "step": 1390
907
+ },
908
+ {
909
+ "epoch": 7.63,
910
+ "learning_rate": 1.305403764420158e-05,
911
+ "loss": 0.5835,
912
+ "step": 1400
913
+ },
914
+ {
915
+ "epoch": 7.68,
916
+ "learning_rate": 1.2750455373406195e-05,
917
+ "loss": 0.5353,
918
+ "step": 1410
919
+ },
920
+ {
921
+ "epoch": 7.74,
922
+ "learning_rate": 1.2446873102610808e-05,
923
+ "loss": 0.5568,
924
+ "step": 1420
925
+ },
926
+ {
927
+ "epoch": 7.79,
928
+ "learning_rate": 1.2143290831815423e-05,
929
+ "loss": 0.5531,
930
+ "step": 1430
931
+ },
932
+ {
933
+ "epoch": 7.85,
934
+ "learning_rate": 1.1839708561020036e-05,
935
+ "loss": 0.555,
936
+ "step": 1440
937
+ },
938
+ {
939
+ "epoch": 7.9,
940
+ "learning_rate": 1.153612629022465e-05,
941
+ "loss": 0.5478,
942
+ "step": 1450
943
+ },
944
+ {
945
+ "epoch": 7.96,
946
+ "learning_rate": 1.1232544019429265e-05,
947
+ "loss": 0.533,
948
+ "step": 1460
949
+ },
950
+ {
951
+ "epoch": 8.0,
952
+ "eval_accuracy": 0.7472190257000384,
953
+ "eval_loss": 0.6449919939041138,
954
+ "eval_runtime": 93.8292,
955
+ "eval_samples_per_second": 27.785,
956
+ "eval_steps_per_second": 0.874,
957
+ "step": 1468
958
+ },
959
+ {
960
+ "epoch": 8.01,
961
+ "learning_rate": 1.092896174863388e-05,
962
+ "loss": 0.5692,
963
+ "step": 1470
964
+ },
965
+ {
966
+ "epoch": 8.07,
967
+ "learning_rate": 1.0625379477838495e-05,
968
+ "loss": 0.5422,
969
+ "step": 1480
970
+ },
971
+ {
972
+ "epoch": 8.12,
973
+ "learning_rate": 1.032179720704311e-05,
974
+ "loss": 0.5154,
975
+ "step": 1490
976
+ },
977
+ {
978
+ "epoch": 8.17,
979
+ "learning_rate": 1.0018214936247722e-05,
980
+ "loss": 0.5471,
981
+ "step": 1500
982
+ },
983
+ {
984
+ "epoch": 8.23,
985
+ "learning_rate": 9.714632665452337e-06,
986
+ "loss": 0.5253,
987
+ "step": 1510
988
+ },
989
+ {
990
+ "epoch": 8.28,
991
+ "learning_rate": 9.411050394656952e-06,
992
+ "loss": 0.526,
993
+ "step": 1520
994
+ },
995
+ {
996
+ "epoch": 8.34,
997
+ "learning_rate": 9.107468123861566e-06,
998
+ "loss": 0.5038,
999
+ "step": 1530
1000
+ },
1001
+ {
1002
+ "epoch": 8.39,
1003
+ "learning_rate": 8.803885853066181e-06,
1004
+ "loss": 0.5109,
1005
+ "step": 1540
1006
+ },
1007
+ {
1008
+ "epoch": 8.45,
1009
+ "learning_rate": 8.500303582270796e-06,
1010
+ "loss": 0.4941,
1011
+ "step": 1550
1012
+ },
1013
+ {
1014
+ "epoch": 8.5,
1015
+ "learning_rate": 8.196721311475409e-06,
1016
+ "loss": 0.5616,
1017
+ "step": 1560
1018
+ },
1019
+ {
1020
+ "epoch": 8.56,
1021
+ "learning_rate": 7.893139040680023e-06,
1022
+ "loss": 0.5514,
1023
+ "step": 1570
1024
+ },
1025
+ {
1026
+ "epoch": 8.61,
1027
+ "learning_rate": 7.589556769884638e-06,
1028
+ "loss": 0.5203,
1029
+ "step": 1580
1030
+ },
1031
+ {
1032
+ "epoch": 8.66,
1033
+ "learning_rate": 7.2859744990892545e-06,
1034
+ "loss": 0.5351,
1035
+ "step": 1590
1036
+ },
1037
+ {
1038
+ "epoch": 8.72,
1039
+ "learning_rate": 6.982392228293868e-06,
1040
+ "loss": 0.5334,
1041
+ "step": 1600
1042
+ },
1043
+ {
1044
+ "epoch": 8.77,
1045
+ "learning_rate": 6.678809957498483e-06,
1046
+ "loss": 0.5162,
1047
+ "step": 1610
1048
+ },
1049
+ {
1050
+ "epoch": 8.83,
1051
+ "learning_rate": 6.375227686703098e-06,
1052
+ "loss": 0.5482,
1053
+ "step": 1620
1054
+ },
1055
+ {
1056
+ "epoch": 8.88,
1057
+ "learning_rate": 6.0716454159077115e-06,
1058
+ "loss": 0.5077,
1059
+ "step": 1630
1060
+ },
1061
+ {
1062
+ "epoch": 8.94,
1063
+ "learning_rate": 5.768063145112325e-06,
1064
+ "loss": 0.5392,
1065
+ "step": 1640
1066
+ },
1067
+ {
1068
+ "epoch": 8.99,
1069
+ "learning_rate": 5.46448087431694e-06,
1070
+ "loss": 0.5416,
1071
+ "step": 1650
1072
+ },
1073
+ {
1074
+ "epoch": 9.0,
1075
+ "eval_accuracy": 0.74530111238972,
1076
+ "eval_loss": 0.638206422328949,
1077
+ "eval_runtime": 94.5986,
1078
+ "eval_samples_per_second": 27.559,
1079
+ "eval_steps_per_second": 0.867,
1080
+ "step": 1651
1081
+ },
1082
+ {
1083
+ "epoch": 9.05,
1084
+ "learning_rate": 5.160898603521555e-06,
1085
+ "loss": 0.541,
1086
+ "step": 1660
1087
+ },
1088
+ {
1089
+ "epoch": 9.1,
1090
+ "learning_rate": 4.8573163327261686e-06,
1091
+ "loss": 0.5393,
1092
+ "step": 1670
1093
+ },
1094
+ {
1095
+ "epoch": 9.16,
1096
+ "learning_rate": 4.553734061930783e-06,
1097
+ "loss": 0.488,
1098
+ "step": 1680
1099
+ },
1100
+ {
1101
+ "epoch": 9.21,
1102
+ "learning_rate": 4.250151791135398e-06,
1103
+ "loss": 0.5104,
1104
+ "step": 1690
1105
+ },
1106
+ {
1107
+ "epoch": 9.26,
1108
+ "learning_rate": 3.946569520340012e-06,
1109
+ "loss": 0.5151,
1110
+ "step": 1700
1111
+ },
1112
+ {
1113
+ "epoch": 9.32,
1114
+ "learning_rate": 3.6429872495446273e-06,
1115
+ "loss": 0.5165,
1116
+ "step": 1710
1117
+ },
1118
+ {
1119
+ "epoch": 9.37,
1120
+ "learning_rate": 3.3394049787492415e-06,
1121
+ "loss": 0.5116,
1122
+ "step": 1720
1123
+ },
1124
+ {
1125
+ "epoch": 9.43,
1126
+ "learning_rate": 3.0358227079538558e-06,
1127
+ "loss": 0.5174,
1128
+ "step": 1730
1129
+ },
1130
+ {
1131
+ "epoch": 9.48,
1132
+ "learning_rate": 2.73224043715847e-06,
1133
+ "loss": 0.5321,
1134
+ "step": 1740
1135
+ },
1136
+ {
1137
+ "epoch": 9.54,
1138
+ "learning_rate": 2.4286581663630843e-06,
1139
+ "loss": 0.4799,
1140
+ "step": 1750
1141
+ },
1142
+ {
1143
+ "epoch": 9.59,
1144
+ "learning_rate": 2.125075895567699e-06,
1145
+ "loss": 0.4766,
1146
+ "step": 1760
1147
+ },
1148
+ {
1149
+ "epoch": 9.65,
1150
+ "learning_rate": 1.8214936247723136e-06,
1151
+ "loss": 0.5425,
1152
+ "step": 1770
1153
+ },
1154
+ {
1155
+ "epoch": 9.7,
1156
+ "learning_rate": 1.5179113539769279e-06,
1157
+ "loss": 0.4936,
1158
+ "step": 1780
1159
+ },
1160
+ {
1161
+ "epoch": 9.75,
1162
+ "learning_rate": 1.2143290831815421e-06,
1163
+ "loss": 0.5408,
1164
+ "step": 1790
1165
+ },
1166
+ {
1167
+ "epoch": 9.81,
1168
+ "learning_rate": 9.107468123861568e-07,
1169
+ "loss": 0.5223,
1170
+ "step": 1800
1171
+ },
1172
+ {
1173
+ "epoch": 9.86,
1174
+ "learning_rate": 6.071645415907711e-07,
1175
+ "loss": 0.5074,
1176
+ "step": 1810
1177
+ },
1178
+ {
1179
+ "epoch": 9.92,
1180
+ "learning_rate": 3.0358227079538554e-07,
1181
+ "loss": 0.5151,
1182
+ "step": 1820
1183
+ },
1184
+ {
1185
+ "epoch": 9.97,
1186
+ "learning_rate": 0.0,
1187
+ "loss": 0.5062,
1188
+ "step": 1830
1189
+ },
1190
+ {
1191
+ "epoch": 9.97,
1192
+ "eval_accuracy": 0.7460682777138473,
1193
+ "eval_loss": 0.6395400166511536,
1194
+ "eval_runtime": 94.039,
1195
+ "eval_samples_per_second": 27.723,
1196
+ "eval_steps_per_second": 0.872,
1197
+ "step": 1830
1198
+ },
1199
+ {
1200
+ "epoch": 9.97,
1201
+ "step": 1830,
1202
+ "total_flos": 1.8133380652001642e+19,
1203
+ "train_loss": 0.6611971782204883,
1204
+ "train_runtime": 13354.5549,
1205
+ "train_samples_per_second": 17.568,
1206
+ "train_steps_per_second": 0.137
1207
+ }
1208
+ ],
1209
+ "logging_steps": 10,
1210
+ "max_steps": 1830,
1211
+ "num_input_tokens_seen": 0,
1212
+ "num_train_epochs": 10,
1213
+ "save_steps": 500,
1214
+ "total_flos": 1.8133380652001642e+19,
1215
+ "train_batch_size": 32,
1216
+ "trial_name": null,
1217
+ "trial_params": null
1218
+ }