rshrott commited on
Commit
4647950
1 Parent(s): c2d1640

🍻 cheers

Browse files
README.md CHANGED
@@ -2,6 +2,7 @@
2
  license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
 
5
  - generated_from_trainer
6
  datasets:
7
  - renovation
@@ -14,7 +15,7 @@ model-index:
14
  name: Image Classification
15
  type: image-classification
16
  dataset:
17
- name: renovation
18
  type: renovation
19
  config: default
20
  split: validation
@@ -22,7 +23,7 @@ model-index:
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
- value: 0.6863636363636364
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,10 +31,10 @@ should probably proofread and complete it, then remove this comment. -->
30
 
31
  # vit-base-renovation
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 renovation dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.9761
36
- - Accuracy: 0.6864
37
 
38
  ## Model description
39
 
 
2
  license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
5
+ - image-classification
6
  - generated_from_trainer
7
  datasets:
8
  - renovation
 
15
  name: Image Classification
16
  type: image-classification
17
  dataset:
18
+ name: renovations
19
  type: renovation
20
  config: default
21
  split: validation
 
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
+ value: 0.6545454545454545
27
  ---
28
 
29
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
31
 
32
  # vit-base-renovation
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 renovations dataset.
35
  It achieves the following results on the evaluation set:
36
+ - Loss: 0.7622
37
+ - Accuracy: 0.6545
38
 
39
  ## Model description
40
 
all_results.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
  "epoch": 4.0,
3
- "eval_accuracy": 0.6909090909090909,
4
- "eval_loss": 0.710116982460022,
5
- "eval_runtime": 7.3292,
6
- "eval_samples_per_second": 30.017,
7
- "eval_steps_per_second": 3.82,
8
  "total_flos": 6.137420584742093e+17,
9
- "train_loss": 0.3987359667737638,
10
- "train_runtime": 862.5404,
11
- "train_samples_per_second": 9.182,
12
- "train_steps_per_second": 0.575
13
  }
 
1
  {
2
  "epoch": 4.0,
3
+ "eval_accuracy": 0.6545454545454545,
4
+ "eval_loss": 0.7622188329696655,
5
+ "eval_runtime": 7.8953,
6
+ "eval_samples_per_second": 27.865,
7
+ "eval_steps_per_second": 3.546,
8
  "total_flos": 6.137420584742093e+17,
9
+ "train_loss": 0.4817413741181935,
10
+ "train_runtime": 646.4292,
11
+ "train_samples_per_second": 12.252,
12
+ "train_steps_per_second": 0.767
13
  }
eval_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 4.0,
3
- "eval_accuracy": 0.6909090909090909,
4
- "eval_loss": 0.710116982460022,
5
- "eval_runtime": 7.3292,
6
- "eval_samples_per_second": 30.017,
7
- "eval_steps_per_second": 3.82
8
  }
 
1
  {
2
  "epoch": 4.0,
3
+ "eval_accuracy": 0.6545454545454545,
4
+ "eval_loss": 0.7622188329696655,
5
+ "eval_runtime": 7.8953,
6
+ "eval_samples_per_second": 27.865,
7
+ "eval_steps_per_second": 3.546
8
  }
runs/Mar18_14-26-58_7e4cb5ba27a4/events.out.tfevents.1710772695.7e4cb5ba27a4.347.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ae68c6bd7d5c4fee51e11597a552d9143a351674c988fe3215c62b6af3f73b3
3
+ size 411
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 4.0,
3
  "total_flos": 6.137420584742093e+17,
4
- "train_loss": 0.3987359667737638,
5
- "train_runtime": 862.5404,
6
- "train_samples_per_second": 9.182,
7
- "train_steps_per_second": 0.575
8
  }
 
1
  {
2
  "epoch": 4.0,
3
  "total_flos": 6.137420584742093e+17,
4
+ "train_loss": 0.4817413741181935,
5
+ "train_runtime": 646.4292,
6
+ "train_samples_per_second": 12.252,
7
+ "train_steps_per_second": 0.767
8
  }
trainer_state.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "best_metric": 0.710116982460022,
3
  "best_model_checkpoint": "./vit-base-renovation/checkpoint-250",
4
  "epoch": 4.0,
5
  "eval_steps": 25,
@@ -10,526 +10,526 @@
10
  "log_history": [
11
  {
12
  "epoch": 0.08,
13
- "grad_norm": 1.7348554134368896,
14
  "learning_rate": 0.00019596774193548388,
15
- "loss": 1.02,
16
  "step": 10
17
  },
18
  {
19
  "epoch": 0.16,
20
- "grad_norm": 0.9956100583076477,
21
  "learning_rate": 0.00019193548387096775,
22
- "loss": 1.0036,
23
  "step": 20
24
  },
25
  {
26
  "epoch": 0.2,
27
- "eval_accuracy": 0.5,
28
- "eval_loss": 0.9849038124084473,
29
- "eval_runtime": 8.3244,
30
- "eval_samples_per_second": 26.428,
31
- "eval_steps_per_second": 3.364,
32
  "step": 25
33
  },
34
  {
35
  "epoch": 0.24,
36
- "grad_norm": 1.1454524993896484,
37
  "learning_rate": 0.00018790322580645164,
38
- "loss": 0.9917,
39
  "step": 30
40
  },
41
  {
42
  "epoch": 0.32,
43
- "grad_norm": 2.6789638996124268,
44
  "learning_rate": 0.00018387096774193548,
45
- "loss": 0.8835,
46
  "step": 40
47
  },
48
  {
49
  "epoch": 0.4,
50
- "grad_norm": 2.0885393619537354,
51
  "learning_rate": 0.00017983870967741935,
52
- "loss": 0.8051,
53
  "step": 50
54
  },
55
  {
56
  "epoch": 0.4,
57
- "eval_accuracy": 0.5545454545454546,
58
- "eval_loss": 0.9105865359306335,
59
- "eval_runtime": 8.5151,
60
- "eval_samples_per_second": 25.836,
61
- "eval_steps_per_second": 3.288,
62
  "step": 50
63
  },
64
  {
65
  "epoch": 0.48,
66
- "grad_norm": 1.2426408529281616,
67
  "learning_rate": 0.00017580645161290325,
68
- "loss": 0.7545,
69
  "step": 60
70
  },
71
  {
72
  "epoch": 0.56,
73
- "grad_norm": 3.0708746910095215,
74
  "learning_rate": 0.00017177419354838711,
75
- "loss": 0.8336,
76
  "step": 70
77
  },
78
  {
79
  "epoch": 0.6,
80
- "eval_accuracy": 0.5954545454545455,
81
- "eval_loss": 0.9004097580909729,
82
- "eval_runtime": 8.3678,
83
- "eval_samples_per_second": 26.291,
84
- "eval_steps_per_second": 3.346,
85
  "step": 75
86
  },
87
  {
88
  "epoch": 0.65,
89
- "grad_norm": 1.7403985261917114,
90
- "learning_rate": 0.0001681451612903226,
91
- "loss": 0.8614,
92
  "step": 80
93
  },
94
  {
95
  "epoch": 0.73,
96
- "grad_norm": 1.7942191362380981,
97
- "learning_rate": 0.00016411290322580646,
98
- "loss": 0.8141,
99
  "step": 90
100
  },
101
  {
102
  "epoch": 0.81,
103
- "grad_norm": 2.56789231300354,
104
- "learning_rate": 0.00016008064516129033,
105
- "loss": 0.786,
106
  "step": 100
107
  },
108
  {
109
  "epoch": 0.81,
110
- "eval_accuracy": 0.6454545454545455,
111
- "eval_loss": 0.7700986266136169,
112
- "eval_runtime": 8.4023,
113
- "eval_samples_per_second": 26.183,
114
- "eval_steps_per_second": 3.332,
115
  "step": 100
116
  },
117
  {
118
  "epoch": 0.89,
119
- "grad_norm": 2.265244722366333,
120
- "learning_rate": 0.0001560483870967742,
121
- "loss": 0.7118,
122
  "step": 110
123
  },
124
  {
125
  "epoch": 0.97,
126
- "grad_norm": 2.001025915145874,
127
- "learning_rate": 0.0001520161290322581,
128
- "loss": 0.7854,
129
  "step": 120
130
  },
131
  {
132
  "epoch": 1.01,
133
- "eval_accuracy": 0.6227272727272727,
134
- "eval_loss": 0.7561001777648926,
135
- "eval_runtime": 8.1474,
136
- "eval_samples_per_second": 27.002,
137
- "eval_steps_per_second": 3.437,
138
  "step": 125
139
  },
140
  {
141
  "epoch": 1.05,
142
- "grad_norm": 1.243610143661499,
143
- "learning_rate": 0.00014798387096774193,
144
- "loss": 0.5415,
145
  "step": 130
146
  },
147
  {
148
  "epoch": 1.13,
149
- "grad_norm": 1.9543741941452026,
150
- "learning_rate": 0.0001439516129032258,
151
- "loss": 0.5246,
152
  "step": 140
153
  },
154
  {
155
  "epoch": 1.21,
156
- "grad_norm": 1.4787832498550415,
157
- "learning_rate": 0.00013991935483870967,
158
- "loss": 0.4603,
159
  "step": 150
160
  },
161
  {
162
  "epoch": 1.21,
163
  "eval_accuracy": 0.6409090909090909,
164
- "eval_loss": 0.8104726672172546,
165
- "eval_runtime": 7.4392,
166
- "eval_samples_per_second": 29.573,
167
- "eval_steps_per_second": 3.764,
168
  "step": 150
169
  },
170
  {
171
  "epoch": 1.29,
172
- "grad_norm": 1.9429636001586914,
173
- "learning_rate": 0.00013588709677419357,
174
- "loss": 0.442,
175
  "step": 160
176
  },
177
  {
178
  "epoch": 1.37,
179
- "grad_norm": 2.558549642562866,
180
- "learning_rate": 0.00013185483870967743,
181
- "loss": 0.4934,
182
  "step": 170
183
  },
184
  {
185
  "epoch": 1.41,
186
- "eval_accuracy": 0.6181818181818182,
187
- "eval_loss": 0.874618411064148,
188
- "eval_runtime": 7.375,
189
- "eval_samples_per_second": 29.831,
190
- "eval_steps_per_second": 3.797,
191
  "step": 175
192
  },
193
  {
194
  "epoch": 1.45,
195
- "grad_norm": 3.0746328830718994,
196
- "learning_rate": 0.0001278225806451613,
197
- "loss": 0.5744,
198
  "step": 180
199
  },
200
  {
201
  "epoch": 1.53,
202
- "grad_norm": 2.0470433235168457,
203
  "learning_rate": 0.00012379032258064514,
204
- "loss": 0.5933,
205
  "step": 190
206
  },
207
  {
208
  "epoch": 1.61,
209
- "grad_norm": 2.0045182704925537,
210
  "learning_rate": 0.00011975806451612903,
211
- "loss": 0.5315,
212
  "step": 200
213
  },
214
  {
215
  "epoch": 1.61,
216
- "eval_accuracy": 0.6636363636363637,
217
- "eval_loss": 0.8266905546188354,
218
- "eval_runtime": 7.745,
219
- "eval_samples_per_second": 28.405,
220
- "eval_steps_per_second": 3.615,
221
  "step": 200
222
  },
223
  {
224
  "epoch": 1.69,
225
- "grad_norm": 2.0231032371520996,
226
  "learning_rate": 0.00011572580645161291,
227
- "loss": 0.5538,
228
  "step": 210
229
  },
230
  {
231
  "epoch": 1.77,
232
- "grad_norm": 2.176417589187622,
233
  "learning_rate": 0.00011169354838709678,
234
- "loss": 0.5251,
235
  "step": 220
236
  },
237
  {
238
  "epoch": 1.81,
239
- "eval_accuracy": 0.65,
240
- "eval_loss": 0.8585435152053833,
241
- "eval_runtime": 7.3957,
242
- "eval_samples_per_second": 29.747,
243
- "eval_steps_per_second": 3.786,
244
  "step": 225
245
  },
246
  {
247
  "epoch": 1.85,
248
- "grad_norm": 3.1730520725250244,
249
- "learning_rate": 0.00010806451612903225,
250
- "loss": 0.5472,
251
  "step": 230
252
  },
253
  {
254
  "epoch": 1.94,
255
- "grad_norm": 2.2670953273773193,
256
- "learning_rate": 0.00010403225806451614,
257
- "loss": 0.5488,
258
  "step": 240
259
  },
260
  {
261
  "epoch": 2.02,
262
- "grad_norm": 0.9752326607704163,
263
- "learning_rate": 0.0001,
264
- "loss": 0.4386,
265
  "step": 250
266
  },
267
  {
268
  "epoch": 2.02,
269
- "eval_accuracy": 0.6909090909090909,
270
- "eval_loss": 0.710116982460022,
271
- "eval_runtime": 8.4962,
272
- "eval_samples_per_second": 25.894,
273
- "eval_steps_per_second": 3.296,
274
  "step": 250
275
  },
276
  {
277
  "epoch": 2.1,
278
- "grad_norm": 0.6374533176422119,
279
- "learning_rate": 9.596774193548387e-05,
280
- "loss": 0.241,
281
  "step": 260
282
  },
283
  {
284
  "epoch": 2.18,
285
- "grad_norm": 0.9005341529846191,
286
- "learning_rate": 9.193548387096774e-05,
287
- "loss": 0.2627,
288
  "step": 270
289
  },
290
  {
291
  "epoch": 2.22,
292
- "eval_accuracy": 0.6409090909090909,
293
- "eval_loss": 1.0041698217391968,
294
- "eval_runtime": 7.4719,
295
- "eval_samples_per_second": 29.443,
296
- "eval_steps_per_second": 3.747,
297
  "step": 275
298
  },
299
  {
300
  "epoch": 2.26,
301
- "grad_norm": 2.1455209255218506,
302
- "learning_rate": 8.790322580645162e-05,
303
- "loss": 0.1868,
304
  "step": 280
305
  },
306
  {
307
  "epoch": 2.34,
308
- "grad_norm": 0.3361818790435791,
309
- "learning_rate": 8.387096774193549e-05,
310
- "loss": 0.1521,
311
  "step": 290
312
  },
313
  {
314
  "epoch": 2.42,
315
- "grad_norm": 5.28161096572876,
316
- "learning_rate": 7.983870967741936e-05,
317
- "loss": 0.1524,
318
  "step": 300
319
  },
320
  {
321
  "epoch": 2.42,
322
- "eval_accuracy": 0.6545454545454545,
323
- "eval_loss": 0.9489237666130066,
324
- "eval_runtime": 8.3095,
325
- "eval_samples_per_second": 26.476,
326
- "eval_steps_per_second": 3.37,
327
  "step": 300
328
  },
329
  {
330
  "epoch": 2.5,
331
- "grad_norm": 2.760798215866089,
332
- "learning_rate": 7.580645161290323e-05,
333
- "loss": 0.1971,
334
  "step": 310
335
  },
336
  {
337
  "epoch": 2.58,
338
- "grad_norm": 1.529565453529358,
339
- "learning_rate": 7.177419354838711e-05,
340
- "loss": 0.1272,
341
  "step": 320
342
  },
343
  {
344
  "epoch": 2.62,
345
- "eval_accuracy": 0.65,
346
- "eval_loss": 1.066292405128479,
347
- "eval_runtime": 8.3186,
348
- "eval_samples_per_second": 26.447,
349
- "eval_steps_per_second": 3.366,
350
  "step": 325
351
  },
352
  {
353
  "epoch": 2.66,
354
- "grad_norm": 0.5179287791252136,
355
- "learning_rate": 6.774193548387096e-05,
356
- "loss": 0.1676,
357
  "step": 330
358
  },
359
  {
360
  "epoch": 2.74,
361
- "grad_norm": 1.027869701385498,
362
- "learning_rate": 6.370967741935485e-05,
363
- "loss": 0.1194,
364
  "step": 340
365
  },
366
  {
367
  "epoch": 2.82,
368
- "grad_norm": 1.2000805139541626,
369
- "learning_rate": 5.9677419354838715e-05,
370
- "loss": 0.186,
371
  "step": 350
372
  },
373
  {
374
  "epoch": 2.82,
375
- "eval_accuracy": 0.6545454545454545,
376
- "eval_loss": 1.0830607414245605,
377
- "eval_runtime": 8.4011,
378
- "eval_samples_per_second": 26.187,
379
- "eval_steps_per_second": 3.333,
380
  "step": 350
381
  },
382
  {
383
  "epoch": 2.9,
384
- "grad_norm": 3.6447389125823975,
385
- "learning_rate": 5.5645161290322576e-05,
386
- "loss": 0.2194,
387
  "step": 360
388
  },
389
  {
390
  "epoch": 2.98,
391
- "grad_norm": 4.36700439453125,
392
- "learning_rate": 5.161290322580645e-05,
393
- "loss": 0.1544,
394
  "step": 370
395
  },
396
  {
397
  "epoch": 3.02,
398
- "eval_accuracy": 0.6363636363636364,
399
- "eval_loss": 1.115280032157898,
400
- "eval_runtime": 7.5342,
401
- "eval_samples_per_second": 29.2,
402
- "eval_steps_per_second": 3.716,
403
  "step": 375
404
  },
405
  {
406
  "epoch": 3.06,
407
- "grad_norm": 0.32941731810569763,
408
- "learning_rate": 4.7580645161290326e-05,
409
- "loss": 0.0438,
410
  "step": 380
411
  },
412
  {
413
  "epoch": 3.15,
414
- "grad_norm": 0.40441280603408813,
415
- "learning_rate": 4.3548387096774194e-05,
416
- "loss": 0.0432,
417
  "step": 390
418
  },
419
  {
420
  "epoch": 3.23,
421
- "grad_norm": 0.3993894159793854,
422
- "learning_rate": 3.951612903225806e-05,
423
- "loss": 0.0803,
424
  "step": 400
425
  },
426
  {
427
  "epoch": 3.23,
428
- "eval_accuracy": 0.6409090909090909,
429
- "eval_loss": 1.039893388748169,
430
- "eval_runtime": 7.6391,
431
- "eval_samples_per_second": 28.799,
432
- "eval_steps_per_second": 3.665,
433
  "step": 400
434
  },
435
  {
436
  "epoch": 3.31,
437
- "grad_norm": 0.0989086702466011,
438
- "learning_rate": 3.548387096774194e-05,
439
- "loss": 0.0298,
440
  "step": 410
441
  },
442
  {
443
  "epoch": 3.39,
444
- "grad_norm": 0.30518239736557007,
445
- "learning_rate": 3.1451612903225806e-05,
446
- "loss": 0.041,
447
  "step": 420
448
  },
449
  {
450
  "epoch": 3.43,
451
- "eval_accuracy": 0.6818181818181818,
452
- "eval_loss": 1.0910764932632446,
453
- "eval_runtime": 7.8435,
454
- "eval_samples_per_second": 28.049,
455
- "eval_steps_per_second": 3.57,
456
  "step": 425
457
  },
458
  {
459
  "epoch": 3.47,
460
- "grad_norm": 0.383977472782135,
461
- "learning_rate": 2.7419354838709678e-05,
462
- "loss": 0.0702,
463
  "step": 430
464
  },
465
  {
466
  "epoch": 3.55,
467
- "grad_norm": 0.12995119392871857,
468
- "learning_rate": 2.338709677419355e-05,
469
- "loss": 0.0211,
470
  "step": 440
471
  },
472
  {
473
  "epoch": 3.63,
474
- "grad_norm": 0.09603479504585266,
475
- "learning_rate": 1.935483870967742e-05,
476
- "loss": 0.0685,
477
  "step": 450
478
  },
479
  {
480
  "epoch": 3.63,
481
- "eval_accuracy": 0.6590909090909091,
482
- "eval_loss": 1.1889867782592773,
483
- "eval_runtime": 7.401,
484
- "eval_samples_per_second": 29.726,
485
- "eval_steps_per_second": 3.783,
486
  "step": 450
487
  },
488
  {
489
  "epoch": 3.71,
490
- "grad_norm": 0.13990958034992218,
491
- "learning_rate": 1.5322580645161292e-05,
492
- "loss": 0.0406,
493
  "step": 460
494
  },
495
  {
496
  "epoch": 3.79,
497
- "grad_norm": 0.11549140512943268,
498
- "learning_rate": 1.129032258064516e-05,
499
- "loss": 0.0475,
500
  "step": 470
501
  },
502
  {
503
  "epoch": 3.83,
504
- "eval_accuracy": 0.6681818181818182,
505
- "eval_loss": 1.1724765300750732,
506
- "eval_runtime": 8.3023,
507
- "eval_samples_per_second": 26.499,
508
- "eval_steps_per_second": 3.373,
509
  "step": 475
510
  },
511
  {
512
  "epoch": 3.87,
513
- "grad_norm": 2.7816131114959717,
514
- "learning_rate": 7.258064516129033e-06,
515
- "loss": 0.0376,
516
  "step": 480
517
  },
518
  {
519
  "epoch": 3.95,
520
- "grad_norm": 0.09355681389570236,
521
  "learning_rate": 3.225806451612903e-06,
522
- "loss": 0.0452,
523
  "step": 490
524
  },
525
  {
526
  "epoch": 4.0,
527
  "step": 496,
528
  "total_flos": 6.137420584742093e+17,
529
- "train_loss": 0.3987359667737638,
530
- "train_runtime": 862.5404,
531
- "train_samples_per_second": 9.182,
532
- "train_steps_per_second": 0.575
533
  }
534
  ],
535
  "logging_steps": 10,
 
1
  {
2
+ "best_metric": 0.7622188329696655,
3
  "best_model_checkpoint": "./vit-base-renovation/checkpoint-250",
4
  "epoch": 4.0,
5
  "eval_steps": 25,
 
10
  "log_history": [
11
  {
12
  "epoch": 0.08,
13
+ "grad_norm": 1.4599734544754028,
14
  "learning_rate": 0.00019596774193548388,
15
+ "loss": 0.9874,
16
  "step": 10
17
  },
18
  {
19
  "epoch": 0.16,
20
+ "grad_norm": 1.6008720397949219,
21
  "learning_rate": 0.00019193548387096775,
22
+ "loss": 0.9737,
23
  "step": 20
24
  },
25
  {
26
  "epoch": 0.2,
27
+ "eval_accuracy": 0.5045454545454545,
28
+ "eval_loss": 1.0076260566711426,
29
+ "eval_runtime": 8.1182,
30
+ "eval_samples_per_second": 27.1,
31
+ "eval_steps_per_second": 3.449,
32
  "step": 25
33
  },
34
  {
35
  "epoch": 0.24,
36
+ "grad_norm": 1.4715768098831177,
37
  "learning_rate": 0.00018790322580645164,
38
+ "loss": 1.0314,
39
  "step": 30
40
  },
41
  {
42
  "epoch": 0.32,
43
+ "grad_norm": 3.563518762588501,
44
  "learning_rate": 0.00018387096774193548,
45
+ "loss": 0.9846,
46
  "step": 40
47
  },
48
  {
49
  "epoch": 0.4,
50
+ "grad_norm": 2.6104962825775146,
51
  "learning_rate": 0.00017983870967741935,
52
+ "loss": 0.862,
53
  "step": 50
54
  },
55
  {
56
  "epoch": 0.4,
57
+ "eval_accuracy": 0.5045454545454545,
58
+ "eval_loss": 1.0219964981079102,
59
+ "eval_runtime": 8.0163,
60
+ "eval_samples_per_second": 27.444,
61
+ "eval_steps_per_second": 3.493,
62
  "step": 50
63
  },
64
  {
65
  "epoch": 0.48,
66
+ "grad_norm": 1.1494791507720947,
67
  "learning_rate": 0.00017580645161290325,
68
+ "loss": 0.7918,
69
  "step": 60
70
  },
71
  {
72
  "epoch": 0.56,
73
+ "grad_norm": 2.8406925201416016,
74
  "learning_rate": 0.00017177419354838711,
75
+ "loss": 0.9064,
76
  "step": 70
77
  },
78
  {
79
  "epoch": 0.6,
80
+ "eval_accuracy": 0.5590909090909091,
81
+ "eval_loss": 0.9076008200645447,
82
+ "eval_runtime": 7.2828,
83
+ "eval_samples_per_second": 30.208,
84
+ "eval_steps_per_second": 3.845,
85
  "step": 75
86
  },
87
  {
88
  "epoch": 0.65,
89
+ "grad_norm": 1.1309199333190918,
90
+ "learning_rate": 0.00016774193548387098,
91
+ "loss": 0.944,
92
  "step": 80
93
  },
94
  {
95
  "epoch": 0.73,
96
+ "grad_norm": 1.6623650789260864,
97
+ "learning_rate": 0.00016370967741935485,
98
+ "loss": 0.865,
99
  "step": 90
100
  },
101
  {
102
  "epoch": 0.81,
103
+ "grad_norm": 2.2675681114196777,
104
+ "learning_rate": 0.00015967741935483872,
105
+ "loss": 0.8528,
106
  "step": 100
107
  },
108
  {
109
  "epoch": 0.81,
110
+ "eval_accuracy": 0.65,
111
+ "eval_loss": 0.8157132267951965,
112
+ "eval_runtime": 7.8906,
113
+ "eval_samples_per_second": 27.881,
114
+ "eval_steps_per_second": 3.549,
115
  "step": 100
116
  },
117
  {
118
  "epoch": 0.89,
119
+ "grad_norm": 2.036137819290161,
120
+ "learning_rate": 0.0001556451612903226,
121
+ "loss": 0.7899,
122
  "step": 110
123
  },
124
  {
125
  "epoch": 0.97,
126
+ "grad_norm": 1.0894314050674438,
127
+ "learning_rate": 0.00015161290322580646,
128
+ "loss": 0.8848,
129
  "step": 120
130
  },
131
  {
132
  "epoch": 1.01,
133
+ "eval_accuracy": 0.6272727272727273,
134
+ "eval_loss": 0.8089344501495361,
135
+ "eval_runtime": 7.2662,
136
+ "eval_samples_per_second": 30.277,
137
+ "eval_steps_per_second": 3.853,
138
  "step": 125
139
  },
140
  {
141
  "epoch": 1.05,
142
+ "grad_norm": 1.2654739618301392,
143
+ "learning_rate": 0.00014758064516129032,
144
+ "loss": 0.6029,
145
  "step": 130
146
  },
147
  {
148
  "epoch": 1.13,
149
+ "grad_norm": 2.1379916667938232,
150
+ "learning_rate": 0.00014354838709677422,
151
+ "loss": 0.6292,
152
  "step": 140
153
  },
154
  {
155
  "epoch": 1.21,
156
+ "grad_norm": 3.7936744689941406,
157
+ "learning_rate": 0.0001395161290322581,
158
+ "loss": 0.6608,
159
  "step": 150
160
  },
161
  {
162
  "epoch": 1.21,
163
  "eval_accuracy": 0.6409090909090909,
164
+ "eval_loss": 0.8614596128463745,
165
+ "eval_runtime": 8.173,
166
+ "eval_samples_per_second": 26.918,
167
+ "eval_steps_per_second": 3.426,
168
  "step": 150
169
  },
170
  {
171
  "epoch": 1.29,
172
+ "grad_norm": 1.3360966444015503,
173
+ "learning_rate": 0.00013548387096774193,
174
+ "loss": 0.539,
175
  "step": 160
176
  },
177
  {
178
  "epoch": 1.37,
179
+ "grad_norm": 4.338820934295654,
180
+ "learning_rate": 0.0001314516129032258,
181
+ "loss": 0.6748,
182
  "step": 170
183
  },
184
  {
185
  "epoch": 1.41,
186
+ "eval_accuracy": 0.6318181818181818,
187
+ "eval_loss": 0.8425928354263306,
188
+ "eval_runtime": 7.2996,
189
+ "eval_samples_per_second": 30.139,
190
+ "eval_steps_per_second": 3.836,
191
  "step": 175
192
  },
193
  {
194
  "epoch": 1.45,
195
+ "grad_norm": 2.611848831176758,
196
+ "learning_rate": 0.0001274193548387097,
197
+ "loss": 0.6317,
198
  "step": 180
199
  },
200
  {
201
  "epoch": 1.53,
202
+ "grad_norm": 2.4912662506103516,
203
  "learning_rate": 0.00012379032258064514,
204
+ "loss": 0.6935,
205
  "step": 190
206
  },
207
  {
208
  "epoch": 1.61,
209
+ "grad_norm": 2.0737533569335938,
210
  "learning_rate": 0.00011975806451612903,
211
+ "loss": 0.6559,
212
  "step": 200
213
  },
214
  {
215
  "epoch": 1.61,
216
+ "eval_accuracy": 0.6090909090909091,
217
+ "eval_loss": 0.8427345752716064,
218
+ "eval_runtime": 8.531,
219
+ "eval_samples_per_second": 25.788,
220
+ "eval_steps_per_second": 3.282,
221
  "step": 200
222
  },
223
  {
224
  "epoch": 1.69,
225
+ "grad_norm": 2.8581531047821045,
226
  "learning_rate": 0.00011572580645161291,
227
+ "loss": 0.5985,
228
  "step": 210
229
  },
230
  {
231
  "epoch": 1.77,
232
+ "grad_norm": 1.7107830047607422,
233
  "learning_rate": 0.00011169354838709678,
234
+ "loss": 0.5654,
235
  "step": 220
236
  },
237
  {
238
  "epoch": 1.81,
239
+ "eval_accuracy": 0.6681818181818182,
240
+ "eval_loss": 0.8266812562942505,
241
+ "eval_runtime": 7.3084,
242
+ "eval_samples_per_second": 30.102,
243
+ "eval_steps_per_second": 3.831,
244
  "step": 225
245
  },
246
  {
247
  "epoch": 1.85,
248
+ "grad_norm": 2.7065391540527344,
249
+ "learning_rate": 0.00010766129032258066,
250
+ "loss": 0.6948,
251
  "step": 230
252
  },
253
  {
254
  "epoch": 1.94,
255
+ "grad_norm": 1.9006155729293823,
256
+ "learning_rate": 0.00010362903225806453,
257
+ "loss": 0.5896,
258
  "step": 240
259
  },
260
  {
261
  "epoch": 2.02,
262
+ "grad_norm": 1.5013632774353027,
263
+ "learning_rate": 9.95967741935484e-05,
264
+ "loss": 0.5254,
265
  "step": 250
266
  },
267
  {
268
  "epoch": 2.02,
269
+ "eval_accuracy": 0.6545454545454545,
270
+ "eval_loss": 0.7622188329696655,
271
+ "eval_runtime": 8.0247,
272
+ "eval_samples_per_second": 27.415,
273
+ "eval_steps_per_second": 3.489,
274
  "step": 250
275
  },
276
  {
277
  "epoch": 2.1,
278
+ "grad_norm": 3.849762439727783,
279
+ "learning_rate": 9.556451612903226e-05,
280
+ "loss": 0.3218,
281
  "step": 260
282
  },
283
  {
284
  "epoch": 2.18,
285
+ "grad_norm": 1.2341018915176392,
286
+ "learning_rate": 9.153225806451613e-05,
287
+ "loss": 0.2778,
288
  "step": 270
289
  },
290
  {
291
  "epoch": 2.22,
292
+ "eval_accuracy": 0.6636363636363637,
293
+ "eval_loss": 0.9480502009391785,
294
+ "eval_runtime": 7.5019,
295
+ "eval_samples_per_second": 29.326,
296
+ "eval_steps_per_second": 3.732,
297
  "step": 275
298
  },
299
  {
300
  "epoch": 2.26,
301
+ "grad_norm": 2.990760087966919,
302
+ "learning_rate": 8.75e-05,
303
+ "loss": 0.3018,
304
  "step": 280
305
  },
306
  {
307
  "epoch": 2.34,
308
+ "grad_norm": 3.7092368602752686,
309
+ "learning_rate": 8.346774193548388e-05,
310
+ "loss": 0.3696,
311
  "step": 290
312
  },
313
  {
314
  "epoch": 2.42,
315
+ "grad_norm": 4.312985420227051,
316
+ "learning_rate": 7.943548387096774e-05,
317
+ "loss": 0.309,
318
  "step": 300
319
  },
320
  {
321
  "epoch": 2.42,
322
+ "eval_accuracy": 0.6409090909090909,
323
+ "eval_loss": 0.8998208045959473,
324
+ "eval_runtime": 8.0465,
325
+ "eval_samples_per_second": 27.341,
326
+ "eval_steps_per_second": 3.48,
327
  "step": 300
328
  },
329
  {
330
  "epoch": 2.5,
331
+ "grad_norm": 3.9428765773773193,
332
+ "learning_rate": 7.540322580645162e-05,
333
+ "loss": 0.4113,
334
  "step": 310
335
  },
336
  {
337
  "epoch": 2.58,
338
+ "grad_norm": 1.6300395727157593,
339
+ "learning_rate": 7.137096774193549e-05,
340
+ "loss": 0.2396,
341
  "step": 320
342
  },
343
  {
344
  "epoch": 2.62,
345
+ "eval_accuracy": 0.6409090909090909,
346
+ "eval_loss": 0.9171214699745178,
347
+ "eval_runtime": 7.6722,
348
+ "eval_samples_per_second": 28.675,
349
+ "eval_steps_per_second": 3.65,
350
  "step": 325
351
  },
352
  {
353
  "epoch": 2.66,
354
+ "grad_norm": 1.3225387334823608,
355
+ "learning_rate": 6.733870967741935e-05,
356
+ "loss": 0.3451,
357
  "step": 330
358
  },
359
  {
360
  "epoch": 2.74,
361
+ "grad_norm": 2.227588176727295,
362
+ "learning_rate": 6.330645161290322e-05,
363
+ "loss": 0.206,
364
  "step": 340
365
  },
366
  {
367
  "epoch": 2.82,
368
+ "grad_norm": 3.282959222793579,
369
+ "learning_rate": 5.9274193548387104e-05,
370
+ "loss": 0.2773,
371
  "step": 350
372
  },
373
  {
374
  "epoch": 2.82,
375
+ "eval_accuracy": 0.6090909090909091,
376
+ "eval_loss": 1.0582060813903809,
377
+ "eval_runtime": 7.9345,
378
+ "eval_samples_per_second": 27.727,
379
+ "eval_steps_per_second": 3.529,
380
  "step": 350
381
  },
382
  {
383
  "epoch": 2.9,
384
+ "grad_norm": 4.272079944610596,
385
+ "learning_rate": 5.5241935483870966e-05,
386
+ "loss": 0.4049,
387
  "step": 360
388
  },
389
  {
390
  "epoch": 2.98,
391
+ "grad_norm": 4.28529167175293,
392
+ "learning_rate": 5.120967741935484e-05,
393
+ "loss": 0.2516,
394
  "step": 370
395
  },
396
  {
397
  "epoch": 3.02,
398
+ "eval_accuracy": 0.6454545454545455,
399
+ "eval_loss": 0.9362202286720276,
400
+ "eval_runtime": 7.9239,
401
+ "eval_samples_per_second": 27.764,
402
+ "eval_steps_per_second": 3.534,
403
  "step": 375
404
  },
405
  {
406
  "epoch": 3.06,
407
+ "grad_norm": 0.6417772173881531,
408
+ "learning_rate": 4.7177419354838716e-05,
409
+ "loss": 0.1439,
410
  "step": 380
411
  },
412
  {
413
  "epoch": 3.15,
414
+ "grad_norm": 0.6277107000350952,
415
+ "learning_rate": 4.3145161290322584e-05,
416
+ "loss": 0.1053,
417
  "step": 390
418
  },
419
  {
420
  "epoch": 3.23,
421
+ "grad_norm": 4.842502593994141,
422
+ "learning_rate": 3.911290322580645e-05,
423
+ "loss": 0.1578,
424
  "step": 400
425
  },
426
  {
427
  "epoch": 3.23,
428
+ "eval_accuracy": 0.6772727272727272,
429
+ "eval_loss": 0.9263622760772705,
430
+ "eval_runtime": 8.037,
431
+ "eval_samples_per_second": 27.373,
432
+ "eval_steps_per_second": 3.484,
433
  "step": 400
434
  },
435
  {
436
  "epoch": 3.31,
437
+ "grad_norm": 3.4976587295532227,
438
+ "learning_rate": 3.508064516129033e-05,
439
+ "loss": 0.1233,
440
  "step": 410
441
  },
442
  {
443
  "epoch": 3.39,
444
+ "grad_norm": 0.3740604519844055,
445
+ "learning_rate": 3.1048387096774195e-05,
446
+ "loss": 0.0979,
447
  "step": 420
448
  },
449
  {
450
  "epoch": 3.43,
451
+ "eval_accuracy": 0.6772727272727272,
452
+ "eval_loss": 0.946960985660553,
453
+ "eval_runtime": 7.5496,
454
+ "eval_samples_per_second": 29.141,
455
+ "eval_steps_per_second": 3.709,
456
  "step": 425
457
  },
458
  {
459
  "epoch": 3.47,
460
+ "grad_norm": 0.2739836275577545,
461
+ "learning_rate": 2.7016129032258064e-05,
462
+ "loss": 0.083,
463
  "step": 430
464
  },
465
  {
466
  "epoch": 3.55,
467
+ "grad_norm": 2.7154436111450195,
468
+ "learning_rate": 2.2983870967741935e-05,
469
+ "loss": 0.085,
470
  "step": 440
471
  },
472
  {
473
  "epoch": 3.63,
474
+ "grad_norm": 1.2921925783157349,
475
+ "learning_rate": 1.8951612903225807e-05,
476
+ "loss": 0.0836,
477
  "step": 450
478
  },
479
  {
480
  "epoch": 3.63,
481
+ "eval_accuracy": 0.6681818181818182,
482
+ "eval_loss": 0.9940941333770752,
483
+ "eval_runtime": 8.0024,
484
+ "eval_samples_per_second": 27.492,
485
+ "eval_steps_per_second": 3.499,
486
  "step": 450
487
  },
488
  {
489
  "epoch": 3.71,
490
+ "grad_norm": 0.7346110939979553,
491
+ "learning_rate": 1.4919354838709679e-05,
492
+ "loss": 0.0661,
493
  "step": 460
494
  },
495
  {
496
  "epoch": 3.79,
497
+ "grad_norm": 0.540484607219696,
498
+ "learning_rate": 1.0887096774193549e-05,
499
+ "loss": 0.126,
500
  "step": 470
501
  },
502
  {
503
  "epoch": 3.83,
504
+ "eval_accuracy": 0.6863636363636364,
505
+ "eval_loss": 0.9761253595352173,
506
+ "eval_runtime": 7.4163,
507
+ "eval_samples_per_second": 29.665,
508
+ "eval_steps_per_second": 3.775,
509
  "step": 475
510
  },
511
  {
512
  "epoch": 3.87,
513
+ "grad_norm": 1.4092530012130737,
514
+ "learning_rate": 6.854838709677419e-06,
515
+ "loss": 0.0657,
516
  "step": 480
517
  },
518
  {
519
  "epoch": 3.95,
520
+ "grad_norm": 1.824102759361267,
521
  "learning_rate": 3.225806451612903e-06,
522
+ "loss": 0.075,
523
  "step": 490
524
  },
525
  {
526
  "epoch": 4.0,
527
  "step": 496,
528
  "total_flos": 6.137420584742093e+17,
529
+ "train_loss": 0.4817413741181935,
530
+ "train_runtime": 646.4292,
531
+ "train_samples_per_second": 12.252,
532
+ "train_steps_per_second": 0.767
533
  }
534
  ],
535
  "logging_steps": 10,