rshrott commited on
Commit
f03988f
1 Parent(s): 2bf220f

🍻 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
@@ -30,9 +31,9 @@ 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: 1.4621
36
  - Accuracy: 0.6667
37
 
38
  ## Model description
 
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
 
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: 1.0025
37
  - Accuracy: 0.6667
38
 
39
  ## Model description
all_results.json CHANGED
@@ -1,13 +1,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
  }
 
1
  {
2
  "epoch": 4.0,
3
+ "eval_accuracy": 0.6666666666666666,
4
+ "eval_loss": 1.0025324821472168,
5
+ "eval_runtime": 8.0414,
6
+ "eval_samples_per_second": 27.234,
7
+ "eval_steps_per_second": 3.482,
8
+ "total_flos": 6.10974224738132e+17,
9
+ "train_loss": 0.18820726821920084,
10
+ "train_runtime": 922.3598,
11
+ "train_samples_per_second": 8.548,
12
+ "train_steps_per_second": 0.538
13
  }
eval_results.json CHANGED
@@ -1,8 +1,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
  }
 
1
  {
2
  "epoch": 4.0,
3
+ "eval_accuracy": 0.6666666666666666,
4
+ "eval_loss": 1.0025324821472168,
5
+ "eval_runtime": 8.0414,
6
+ "eval_samples_per_second": 27.234,
7
+ "eval_steps_per_second": 3.482
8
  }
runs/Mar20_02-17-56_afa5dd1e3938/events.out.tfevents.1710902979.afa5dd1e3938.399.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d783a1629dbb0d6f37e8e3127392c3b2607ab8574d8e93ab9e24d67f8f4785b
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.4817413741181935,
5
- "train_runtime": 646.4292,
6
- "train_samples_per_second": 12.252,
7
- "train_steps_per_second": 0.767
8
  }
 
1
  {
2
  "epoch": 4.0,
3
+ "total_flos": 6.10974224738132e+17,
4
+ "train_loss": 0.18820726821920084,
5
+ "train_runtime": 922.3598,
6
+ "train_samples_per_second": 8.548,
7
+ "train_steps_per_second": 0.538
8
  }
trainer_state.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
- "best_metric": 0.7622188329696655,
3
- "best_model_checkpoint": "./vit-base-renovation/checkpoint-250",
4
  "epoch": 4.0,
5
  "eval_steps": 25,
6
  "global_step": 496,
@@ -10,526 +10,526 @@
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,
@@ -537,7 +537,7 @@
537
  "num_input_tokens_seen": 0,
538
  "num_train_epochs": 4,
539
  "save_steps": 25,
540
- "total_flos": 6.137420584742093e+17,
541
  "train_batch_size": 16,
542
  "trial_name": null,
543
  "trial_params": null
 
1
  {
2
+ "best_metric": 1.0025324821472168,
3
+ "best_model_checkpoint": "./vit-base-renovation/checkpoint-75",
4
  "epoch": 4.0,
5
  "eval_steps": 25,
6
  "global_step": 496,
 
10
  "log_history": [
11
  {
12
  "epoch": 0.08,
13
+ "grad_norm": 4.653707504272461,
14
  "learning_rate": 0.00019596774193548388,
15
+ "loss": 0.4901,
16
  "step": 10
17
  },
18
  {
19
  "epoch": 0.16,
20
+ "grad_norm": 1.4146682024002075,
21
  "learning_rate": 0.00019193548387096775,
22
+ "loss": 0.382,
23
  "step": 20
24
  },
25
  {
26
  "epoch": 0.2,
27
+ "eval_accuracy": 0.6073059360730594,
28
+ "eval_loss": 1.1102608442306519,
29
+ "eval_runtime": 7.3313,
30
+ "eval_samples_per_second": 29.872,
31
+ "eval_steps_per_second": 3.819,
32
  "step": 25
33
  },
34
  {
35
  "epoch": 0.24,
36
+ "grad_norm": 4.3104939460754395,
37
  "learning_rate": 0.00018790322580645164,
38
+ "loss": 0.3537,
39
  "step": 30
40
  },
41
  {
42
  "epoch": 0.32,
43
+ "grad_norm": 6.281633377075195,
44
  "learning_rate": 0.00018387096774193548,
45
+ "loss": 0.5919,
46
  "step": 40
47
  },
48
  {
49
  "epoch": 0.4,
50
+ "grad_norm": 1.1501330137252808,
51
  "learning_rate": 0.00017983870967741935,
52
+ "loss": 0.5741,
53
  "step": 50
54
  },
55
  {
56
  "epoch": 0.4,
57
+ "eval_accuracy": 0.6210045662100456,
58
+ "eval_loss": 1.0627731084823608,
59
+ "eval_runtime": 7.7376,
60
+ "eval_samples_per_second": 28.303,
61
+ "eval_steps_per_second": 3.619,
62
  "step": 50
63
  },
64
  {
65
  "epoch": 0.48,
66
+ "grad_norm": 4.414377689361572,
67
  "learning_rate": 0.00017580645161290325,
68
+ "loss": 0.4974,
69
  "step": 60
70
  },
71
  {
72
  "epoch": 0.56,
73
+ "grad_norm": 2.614995241165161,
74
  "learning_rate": 0.00017177419354838711,
75
+ "loss": 0.5589,
76
  "step": 70
77
  },
78
  {
79
  "epoch": 0.6,
80
+ "eval_accuracy": 0.6666666666666666,
81
+ "eval_loss": 1.0025324821472168,
82
+ "eval_runtime": 8.1199,
83
+ "eval_samples_per_second": 26.971,
84
+ "eval_steps_per_second": 3.448,
85
  "step": 75
86
  },
87
  {
88
  "epoch": 0.65,
89
+ "grad_norm": 2.475475549697876,
90
  "learning_rate": 0.00016774193548387098,
91
+ "loss": 0.3703,
92
  "step": 80
93
  },
94
  {
95
  "epoch": 0.73,
96
+ "grad_norm": 5.414321422576904,
97
  "learning_rate": 0.00016370967741935485,
98
+ "loss": 0.4059,
99
  "step": 90
100
  },
101
  {
102
  "epoch": 0.81,
103
+ "grad_norm": 2.9260778427124023,
104
  "learning_rate": 0.00015967741935483872,
105
+ "loss": 0.4074,
106
  "step": 100
107
  },
108
  {
109
  "epoch": 0.81,
110
+ "eval_accuracy": 0.6073059360730594,
111
+ "eval_loss": 1.1324039697647095,
112
+ "eval_runtime": 7.9656,
113
+ "eval_samples_per_second": 27.493,
114
+ "eval_steps_per_second": 3.515,
115
  "step": 100
116
  },
117
  {
118
  "epoch": 0.89,
119
+ "grad_norm": 1.745337724685669,
120
  "learning_rate": 0.0001556451612903226,
121
+ "loss": 0.2648,
122
  "step": 110
123
  },
124
  {
125
  "epoch": 0.97,
126
+ "grad_norm": 0.7676700353622437,
127
  "learning_rate": 0.00015161290322580646,
128
+ "loss": 0.3581,
129
  "step": 120
130
  },
131
  {
132
  "epoch": 1.01,
133
+ "eval_accuracy": 0.6438356164383562,
134
+ "eval_loss": 1.1934621334075928,
135
+ "eval_runtime": 7.3834,
136
+ "eval_samples_per_second": 29.661,
137
+ "eval_steps_per_second": 3.792,
138
  "step": 125
139
  },
140
  {
141
  "epoch": 1.05,
142
+ "grad_norm": 0.40987706184387207,
143
  "learning_rate": 0.00014758064516129032,
144
+ "loss": 0.1496,
145
  "step": 130
146
  },
147
  {
148
  "epoch": 1.13,
149
+ "grad_norm": 0.4426233768463135,
150
  "learning_rate": 0.00014354838709677422,
151
+ "loss": 0.1734,
152
  "step": 140
153
  },
154
  {
155
  "epoch": 1.21,
156
+ "grad_norm": 4.183903694152832,
157
  "learning_rate": 0.0001395161290322581,
158
+ "loss": 0.2618,
159
  "step": 150
160
  },
161
  {
162
  "epoch": 1.21,
163
+ "eval_accuracy": 0.502283105022831,
164
+ "eval_loss": 1.8300055265426636,
165
+ "eval_runtime": 8.5064,
166
+ "eval_samples_per_second": 25.745,
167
+ "eval_steps_per_second": 3.292,
168
  "step": 150
169
  },
170
  {
171
  "epoch": 1.29,
172
+ "grad_norm": 5.4273681640625,
173
  "learning_rate": 0.00013548387096774193,
174
+ "loss": 0.4102,
175
  "step": 160
176
  },
177
  {
178
  "epoch": 1.37,
179
+ "grad_norm": 0.8633623123168945,
180
  "learning_rate": 0.0001314516129032258,
181
+ "loss": 0.1299,
182
  "step": 170
183
  },
184
  {
185
  "epoch": 1.41,
186
+ "eval_accuracy": 0.6301369863013698,
187
+ "eval_loss": 1.2576818466186523,
188
+ "eval_runtime": 8.1086,
189
+ "eval_samples_per_second": 27.008,
190
+ "eval_steps_per_second": 3.453,
191
  "step": 175
192
  },
193
  {
194
  "epoch": 1.45,
195
+ "grad_norm": 4.694212436676025,
196
  "learning_rate": 0.0001274193548387097,
197
+ "loss": 0.2189,
198
  "step": 180
199
  },
200
  {
201
  "epoch": 1.53,
202
+ "grad_norm": 3.287609338760376,
203
+ "learning_rate": 0.00012338709677419356,
204
+ "loss": 0.1625,
205
  "step": 190
206
  },
207
  {
208
  "epoch": 1.61,
209
+ "grad_norm": 3.42903470993042,
210
+ "learning_rate": 0.00011935483870967743,
211
+ "loss": 0.2562,
212
  "step": 200
213
  },
214
  {
215
  "epoch": 1.61,
216
+ "eval_accuracy": 0.6894977168949772,
217
+ "eval_loss": 1.092441439628601,
218
+ "eval_runtime": 8.405,
219
+ "eval_samples_per_second": 26.056,
220
+ "eval_steps_per_second": 3.331,
221
  "step": 200
222
  },
223
  {
224
  "epoch": 1.69,
225
+ "grad_norm": 3.9116501808166504,
226
+ "learning_rate": 0.00011532258064516131,
227
+ "loss": 0.3386,
228
  "step": 210
229
  },
230
  {
231
  "epoch": 1.77,
232
+ "grad_norm": 1.250923752784729,
233
+ "learning_rate": 0.00011129032258064515,
234
+ "loss": 0.2573,
235
  "step": 220
236
  },
237
  {
238
  "epoch": 1.81,
239
+ "eval_accuracy": 0.684931506849315,
240
+ "eval_loss": 1.1285330057144165,
241
+ "eval_runtime": 8.1457,
242
+ "eval_samples_per_second": 26.885,
243
+ "eval_steps_per_second": 3.437,
244
  "step": 225
245
  },
246
  {
247
  "epoch": 1.85,
248
+ "grad_norm": 0.8015751242637634,
249
+ "learning_rate": 0.00010725806451612903,
250
+ "loss": 0.1359,
251
  "step": 230
252
  },
253
  {
254
  "epoch": 1.94,
255
+ "grad_norm": 3.1585347652435303,
256
+ "learning_rate": 0.0001032258064516129,
257
+ "loss": 0.2325,
258
  "step": 240
259
  },
260
  {
261
  "epoch": 2.02,
262
+ "grad_norm": 1.220850944519043,
263
+ "learning_rate": 9.919354838709678e-05,
264
+ "loss": 0.2471,
265
  "step": 250
266
  },
267
  {
268
  "epoch": 2.02,
269
+ "eval_accuracy": 0.6255707762557078,
270
+ "eval_loss": 1.3386634588241577,
271
+ "eval_runtime": 8.4514,
272
+ "eval_samples_per_second": 25.913,
273
+ "eval_steps_per_second": 3.313,
274
  "step": 250
275
  },
276
  {
277
  "epoch": 2.1,
278
+ "grad_norm": 0.14146128296852112,
279
+ "learning_rate": 9.516129032258065e-05,
280
+ "loss": 0.0773,
281
  "step": 260
282
  },
283
  {
284
  "epoch": 2.18,
285
+ "grad_norm": 0.13016600906848907,
286
+ "learning_rate": 9.112903225806452e-05,
287
+ "loss": 0.0618,
288
  "step": 270
289
  },
290
  {
291
  "epoch": 2.22,
292
+ "eval_accuracy": 0.6666666666666666,
293
+ "eval_loss": 1.2246029376983643,
294
+ "eval_runtime": 7.483,
295
+ "eval_samples_per_second": 29.267,
296
+ "eval_steps_per_second": 3.742,
297
  "step": 275
298
  },
299
  {
300
  "epoch": 2.26,
301
+ "grad_norm": 0.1758795827627182,
302
+ "learning_rate": 8.709677419354839e-05,
303
+ "loss": 0.108,
304
  "step": 280
305
  },
306
  {
307
  "epoch": 2.34,
308
+ "grad_norm": 1.5025098323822021,
309
+ "learning_rate": 8.306451612903227e-05,
310
+ "loss": 0.0662,
311
  "step": 290
312
  },
313
  {
314
  "epoch": 2.42,
315
+ "grad_norm": 8.820311546325684,
316
+ "learning_rate": 7.903225806451613e-05,
317
+ "loss": 0.0658,
318
  "step": 300
319
  },
320
  {
321
  "epoch": 2.42,
322
+ "eval_accuracy": 0.634703196347032,
323
+ "eval_loss": 1.4131733179092407,
324
+ "eval_runtime": 8.0515,
325
+ "eval_samples_per_second": 27.2,
326
+ "eval_steps_per_second": 3.478,
327
  "step": 300
328
  },
329
  {
330
  "epoch": 2.5,
331
+ "grad_norm": 0.2889515161514282,
332
+ "learning_rate": 7.500000000000001e-05,
333
+ "loss": 0.1059,
334
  "step": 310
335
  },
336
  {
337
  "epoch": 2.58,
338
+ "grad_norm": 0.25573813915252686,
339
+ "learning_rate": 7.096774193548388e-05,
340
+ "loss": 0.0592,
341
  "step": 320
342
  },
343
  {
344
  "epoch": 2.62,
345
+ "eval_accuracy": 0.6529680365296804,
346
+ "eval_loss": 1.4326461553573608,
347
+ "eval_runtime": 8.3973,
348
+ "eval_samples_per_second": 26.08,
349
+ "eval_steps_per_second": 3.334,
350
  "step": 325
351
  },
352
  {
353
  "epoch": 2.66,
354
+ "grad_norm": 4.445276260375977,
355
+ "learning_rate": 6.693548387096774e-05,
356
+ "loss": 0.0717,
357
  "step": 330
358
  },
359
  {
360
  "epoch": 2.74,
361
+ "grad_norm": 0.5493516325950623,
362
+ "learning_rate": 6.290322580645161e-05,
363
+ "loss": 0.0981,
364
  "step": 340
365
  },
366
  {
367
  "epoch": 2.82,
368
+ "grad_norm": 3.7721645832061768,
369
+ "learning_rate": 5.887096774193549e-05,
370
+ "loss": 0.0464,
371
  "step": 350
372
  },
373
  {
374
  "epoch": 2.82,
375
+ "eval_accuracy": 0.684931506849315,
376
+ "eval_loss": 1.2483667135238647,
377
+ "eval_runtime": 7.3404,
378
+ "eval_samples_per_second": 29.835,
379
+ "eval_steps_per_second": 3.815,
380
  "step": 350
381
  },
382
  {
383
  "epoch": 2.9,
384
+ "grad_norm": 0.07467525452375412,
385
+ "learning_rate": 5.4838709677419355e-05,
386
+ "loss": 0.0249,
387
  "step": 360
388
  },
389
  {
390
  "epoch": 2.98,
391
+ "grad_norm": 1.5583500862121582,
392
+ "learning_rate": 5.080645161290323e-05,
393
+ "loss": 0.0567,
394
  "step": 370
395
  },
396
  {
397
  "epoch": 3.02,
398
+ "eval_accuracy": 0.634703196347032,
399
+ "eval_loss": 1.5350439548492432,
400
+ "eval_runtime": 7.9868,
401
+ "eval_samples_per_second": 27.42,
402
+ "eval_steps_per_second": 3.506,
403
  "step": 375
404
  },
405
  {
406
  "epoch": 3.06,
407
+ "grad_norm": 0.06753262132406235,
408
+ "learning_rate": 4.67741935483871e-05,
409
+ "loss": 0.0185,
410
  "step": 380
411
  },
412
  {
413
  "epoch": 3.15,
414
+ "grad_norm": 0.0861305296421051,
415
+ "learning_rate": 4.2741935483870973e-05,
416
+ "loss": 0.0179,
417
  "step": 390
418
  },
419
  {
420
  "epoch": 3.23,
421
+ "grad_norm": 0.07843153923749924,
422
+ "learning_rate": 3.870967741935484e-05,
423
+ "loss": 0.0269,
424
  "step": 400
425
  },
426
  {
427
  "epoch": 3.23,
428
+ "eval_accuracy": 0.6666666666666666,
429
+ "eval_loss": 1.479716181755066,
430
+ "eval_runtime": 10.862,
431
+ "eval_samples_per_second": 20.162,
432
+ "eval_steps_per_second": 2.578,
433
  "step": 400
434
  },
435
  {
436
  "epoch": 3.31,
437
+ "grad_norm": 0.5874373912811279,
438
+ "learning_rate": 3.467741935483872e-05,
439
+ "loss": 0.0181,
440
  "step": 410
441
  },
442
  {
443
  "epoch": 3.39,
444
+ "grad_norm": 0.15166939795017242,
445
+ "learning_rate": 3.0645161290322585e-05,
446
+ "loss": 0.0239,
447
  "step": 420
448
  },
449
  {
450
  "epoch": 3.43,
451
+ "eval_accuracy": 0.6529680365296804,
452
+ "eval_loss": 1.444357991218567,
453
+ "eval_runtime": 8.4171,
454
+ "eval_samples_per_second": 26.019,
455
+ "eval_steps_per_second": 3.327,
456
  "step": 425
457
  },
458
  {
459
  "epoch": 3.47,
460
+ "grad_norm": 0.2967914938926697,
461
+ "learning_rate": 2.661290322580645e-05,
462
+ "loss": 0.0217,
463
  "step": 430
464
  },
465
  {
466
  "epoch": 3.55,
467
+ "grad_norm": 0.084845632314682,
468
+ "learning_rate": 2.258064516129032e-05,
469
+ "loss": 0.021,
470
  "step": 440
471
  },
472
  {
473
  "epoch": 3.63,
474
+ "grad_norm": 0.06501659750938416,
475
+ "learning_rate": 1.8548387096774193e-05,
476
+ "loss": 0.0184,
477
  "step": 450
478
  },
479
  {
480
  "epoch": 3.63,
481
+ "eval_accuracy": 0.6575342465753424,
482
+ "eval_loss": 1.4473841190338135,
483
+ "eval_runtime": 8.0514,
484
+ "eval_samples_per_second": 27.2,
485
+ "eval_steps_per_second": 3.478,
486
  "step": 450
487
  },
488
  {
489
  "epoch": 3.71,
490
+ "grad_norm": 0.09078585356473923,
491
+ "learning_rate": 1.4516129032258066e-05,
492
+ "loss": 0.0162,
493
  "step": 460
494
  },
495
  {
496
  "epoch": 3.79,
497
+ "grad_norm": 0.06605294346809387,
498
+ "learning_rate": 1.0483870967741936e-05,
499
+ "loss": 0.0286,
500
  "step": 470
501
  },
502
  {
503
  "epoch": 3.83,
504
+ "eval_accuracy": 0.6666666666666666,
505
+ "eval_loss": 1.4620742797851562,
506
+ "eval_runtime": 8.3261,
507
+ "eval_samples_per_second": 26.303,
508
+ "eval_steps_per_second": 3.363,
509
  "step": 475
510
  },
511
  {
512
  "epoch": 3.87,
513
+ "grad_norm": 0.05681619793176651,
514
+ "learning_rate": 6.451612903225806e-06,
515
+ "loss": 0.0162,
516
  "step": 480
517
  },
518
  {
519
  "epoch": 3.95,
520
+ "grad_norm": 0.07769191265106201,
521
+ "learning_rate": 2.4193548387096776e-06,
522
+ "loss": 0.0261,
523
  "step": 490
524
  },
525
  {
526
  "epoch": 4.0,
527
  "step": 496,
528
+ "total_flos": 6.10974224738132e+17,
529
+ "train_loss": 0.18820726821920084,
530
+ "train_runtime": 922.3598,
531
+ "train_samples_per_second": 8.548,
532
+ "train_steps_per_second": 0.538
533
  }
534
  ],
535
  "logging_steps": 10,
 
537
  "num_input_tokens_seen": 0,
538
  "num_train_epochs": 4,
539
  "save_steps": 25,
540
+ "total_flos": 6.10974224738132e+17,
541
  "train_batch_size": 16,
542
  "trial_name": null,
543
  "trial_params": null