rshrott commited on
Commit
24a67f1
·
verified ·
1 Parent(s): f6586e1

🍻 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.7031963470319634
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-renovation2
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.9788
36
- - Accuracy: 0.7032
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.6666666666666666
27
  ---
28
 
29
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
31
 
32
  # vit-base-renovation2
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.8273
37
+ - Accuracy: 0.6667
38
 
39
  ## Model description
40
 
all_results.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
  "epoch": 4.0,
3
- "eval_accuracy": 0.6027397260273972,
4
- "eval_loss": 1.2383606433868408,
5
- "eval_runtime": 7.929,
6
- "eval_samples_per_second": 27.62,
7
- "eval_steps_per_second": 3.531,
8
  "total_flos": 6.10974224738132e+17,
9
- "train_loss": 0.14545002643529686,
10
- "train_runtime": 906.411,
11
- "train_samples_per_second": 8.698,
12
- "train_steps_per_second": 0.547
13
  }
 
1
  {
2
  "epoch": 4.0,
3
+ "eval_accuracy": 0.6666666666666666,
4
+ "eval_loss": 0.8272958397865295,
5
+ "eval_runtime": 7.1557,
6
+ "eval_samples_per_second": 30.605,
7
+ "eval_steps_per_second": 3.913,
8
  "total_flos": 6.10974224738132e+17,
9
+ "train_loss": 0.601069396421794,
10
+ "train_runtime": 693.9158,
11
+ "train_samples_per_second": 11.362,
12
+ "train_steps_per_second": 0.715
13
  }
eval_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 4.0,
3
- "eval_accuracy": 0.6027397260273972,
4
- "eval_loss": 1.2383606433868408,
5
- "eval_runtime": 7.929,
6
- "eval_samples_per_second": 27.62,
7
- "eval_steps_per_second": 3.531
8
  }
 
1
  {
2
  "epoch": 4.0,
3
+ "eval_accuracy": 0.6666666666666666,
4
+ "eval_loss": 0.8272958397865295,
5
+ "eval_runtime": 7.1557,
6
+ "eval_samples_per_second": 30.605,
7
+ "eval_steps_per_second": 3.913
8
  }
runs/Mar22_21-16-36_9f0b864d5439/events.out.tfevents.1711142925.9f0b864d5439.318.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a41f191e74b8229e363f237dafd5ee6c71efac8ffb5ab1cc41c09f5f0a5bfed6
3
+ size 411
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 4.0,
3
  "total_flos": 6.10974224738132e+17,
4
- "train_loss": 0.14545002643529686,
5
- "train_runtime": 906.411,
6
- "train_samples_per_second": 8.698,
7
- "train_steps_per_second": 0.547
8
  }
 
1
  {
2
  "epoch": 4.0,
3
  "total_flos": 6.10974224738132e+17,
4
+ "train_loss": 0.601069396421794,
5
+ "train_runtime": 693.9158,
6
+ "train_samples_per_second": 11.362,
7
+ "train_steps_per_second": 0.715
8
  }
trainer_state.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
- "best_metric": 1.2383606433868408,
3
- "best_model_checkpoint": "./vit-base-renovation2/checkpoint-25",
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": 2.575977325439453,
14
  "learning_rate": 0.00019596774193548388,
15
- "loss": 0.3595,
16
  "step": 10
17
  },
18
  {
19
  "epoch": 0.16,
20
- "grad_norm": 0.9806227684020996,
21
  "learning_rate": 0.00019193548387096775,
22
- "loss": 0.273,
23
  "step": 20
24
  },
25
  {
26
  "epoch": 0.2,
27
- "eval_accuracy": 0.6027397260273972,
28
- "eval_loss": 1.2383606433868408,
29
- "eval_runtime": 7.9057,
30
- "eval_samples_per_second": 27.702,
31
- "eval_steps_per_second": 3.542,
32
  "step": 25
33
  },
34
  {
35
  "epoch": 0.24,
36
- "grad_norm": 2.519897699356079,
37
  "learning_rate": 0.00018790322580645164,
38
- "loss": 0.3335,
39
  "step": 30
40
  },
41
  {
42
  "epoch": 0.32,
43
- "grad_norm": 4.2508063316345215,
44
  "learning_rate": 0.00018387096774193548,
45
- "loss": 0.3046,
46
  "step": 40
47
  },
48
  {
49
  "epoch": 0.4,
50
- "grad_norm": 1.2349441051483154,
51
  "learning_rate": 0.00017983870967741935,
52
- "loss": 0.5153,
53
  "step": 50
54
  },
55
  {
56
  "epoch": 0.4,
57
- "eval_accuracy": 0.5844748858447488,
58
- "eval_loss": 1.40598464012146,
59
- "eval_runtime": 8.1033,
60
- "eval_samples_per_second": 27.026,
61
- "eval_steps_per_second": 3.455,
62
  "step": 50
63
  },
64
  {
65
  "epoch": 0.48,
66
- "grad_norm": 2.6730687618255615,
67
  "learning_rate": 0.00017580645161290325,
68
- "loss": 0.3137,
69
  "step": 60
70
  },
71
  {
72
  "epoch": 0.56,
73
- "grad_norm": 4.323659420013428,
74
  "learning_rate": 0.00017177419354838711,
75
- "loss": 0.2792,
76
  "step": 70
77
  },
78
  {
79
  "epoch": 0.6,
80
- "eval_accuracy": 0.593607305936073,
81
- "eval_loss": 1.30255126953125,
82
- "eval_runtime": 7.4143,
83
- "eval_samples_per_second": 29.538,
84
- "eval_steps_per_second": 3.776,
85
  "step": 75
86
  },
87
  {
88
  "epoch": 0.65,
89
- "grad_norm": 1.2822186946868896,
90
  "learning_rate": 0.00016774193548387098,
91
- "loss": 0.3751,
92
  "step": 80
93
  },
94
  {
95
  "epoch": 0.73,
96
- "grad_norm": 3.15390944480896,
97
  "learning_rate": 0.00016370967741935485,
98
- "loss": 0.5094,
99
  "step": 90
100
  },
101
  {
102
  "epoch": 0.81,
103
- "grad_norm": 3.2424371242523193,
104
- "learning_rate": 0.00016008064516129033,
105
- "loss": 0.5516,
106
  "step": 100
107
  },
108
  {
109
  "epoch": 0.81,
110
- "eval_accuracy": 0.6027397260273972,
111
- "eval_loss": 1.3998719453811646,
112
- "eval_runtime": 7.9973,
113
- "eval_samples_per_second": 27.384,
114
- "eval_steps_per_second": 3.501,
115
  "step": 100
116
  },
117
  {
118
  "epoch": 0.89,
119
- "grad_norm": 1.8185967206954956,
120
- "learning_rate": 0.0001560483870967742,
121
- "loss": 0.2684,
122
  "step": 110
123
  },
124
  {
125
  "epoch": 0.97,
126
- "grad_norm": 3.6117029190063477,
127
- "learning_rate": 0.0001520161290322581,
128
- "loss": 0.4247,
129
  "step": 120
130
  },
131
  {
132
  "epoch": 1.01,
133
- "eval_accuracy": 0.5981735159817352,
134
- "eval_loss": 1.2620913982391357,
135
- "eval_runtime": 8.1833,
136
- "eval_samples_per_second": 26.762,
137
- "eval_steps_per_second": 3.422,
138
  "step": 125
139
  },
140
  {
141
  "epoch": 1.05,
142
- "grad_norm": 0.7001488208770752,
143
- "learning_rate": 0.00014798387096774193,
144
- "loss": 0.1693,
145
  "step": 130
146
  },
147
  {
148
  "epoch": 1.13,
149
- "grad_norm": 3.214643716812134,
150
- "learning_rate": 0.0001439516129032258,
151
- "loss": 0.2085,
152
  "step": 140
153
  },
154
  {
155
  "epoch": 1.21,
156
- "grad_norm": 2.9545109272003174,
157
- "learning_rate": 0.00013991935483870967,
158
- "loss": 0.1556,
159
  "step": 150
160
  },
161
  {
162
  "epoch": 1.21,
163
- "eval_accuracy": 0.5570776255707762,
164
- "eval_loss": 1.5660632848739624,
165
- "eval_runtime": 8.1924,
166
- "eval_samples_per_second": 26.732,
167
- "eval_steps_per_second": 3.418,
168
  "step": 150
169
  },
170
  {
171
  "epoch": 1.29,
172
- "grad_norm": 3.702380418777466,
173
- "learning_rate": 0.00013588709677419357,
174
- "loss": 0.1262,
175
  "step": 160
176
  },
177
  {
178
  "epoch": 1.37,
179
- "grad_norm": 2.634195327758789,
180
- "learning_rate": 0.00013185483870967743,
181
- "loss": 0.1458,
182
  "step": 170
183
  },
184
  {
185
  "epoch": 1.41,
186
- "eval_accuracy": 0.634703196347032,
187
- "eval_loss": 1.3458657264709473,
188
- "eval_runtime": 7.2989,
189
- "eval_samples_per_second": 30.005,
190
- "eval_steps_per_second": 3.836,
191
  "step": 175
192
  },
193
  {
194
  "epoch": 1.45,
195
- "grad_norm": 1.4022108316421509,
196
- "learning_rate": 0.0001278225806451613,
197
- "loss": 0.0859,
198
  "step": 180
199
  },
200
  {
201
  "epoch": 1.53,
202
- "grad_norm": 6.311899185180664,
203
- "learning_rate": 0.00012379032258064514,
204
- "loss": 0.0868,
205
  "step": 190
206
  },
207
  {
208
  "epoch": 1.61,
209
- "grad_norm": 1.674466609954834,
210
- "learning_rate": 0.00011975806451612903,
211
- "loss": 0.1595,
212
  "step": 200
213
  },
214
  {
215
  "epoch": 1.61,
216
- "eval_accuracy": 0.5981735159817352,
217
- "eval_loss": 1.5277855396270752,
218
- "eval_runtime": 8.087,
219
- "eval_samples_per_second": 27.081,
220
- "eval_steps_per_second": 3.462,
221
  "step": 200
222
  },
223
  {
224
  "epoch": 1.69,
225
- "grad_norm": 1.2903810739517212,
226
- "learning_rate": 0.00011572580645161291,
227
- "loss": 0.1657,
228
  "step": 210
229
  },
230
  {
231
  "epoch": 1.77,
232
- "grad_norm": 1.2086151838302612,
233
- "learning_rate": 0.00011169354838709678,
234
- "loss": 0.1195,
235
  "step": 220
236
  },
237
  {
238
  "epoch": 1.81,
239
- "eval_accuracy": 0.6255707762557078,
240
- "eval_loss": 1.5303133726119995,
241
- "eval_runtime": 7.5329,
242
- "eval_samples_per_second": 29.072,
243
- "eval_steps_per_second": 3.717,
244
  "step": 225
245
  },
246
  {
247
  "epoch": 1.85,
248
- "grad_norm": 7.646343231201172,
249
- "learning_rate": 0.00010766129032258066,
250
- "loss": 0.1381,
251
  "step": 230
252
  },
253
  {
254
  "epoch": 1.94,
255
- "grad_norm": 5.495790481567383,
256
- "learning_rate": 0.00010362903225806453,
257
- "loss": 0.1761,
258
  "step": 240
259
  },
260
  {
261
  "epoch": 2.02,
262
- "grad_norm": 0.2532494366168976,
263
- "learning_rate": 9.95967741935484e-05,
264
- "loss": 0.1507,
265
  "step": 250
266
  },
267
  {
268
  "epoch": 2.02,
269
- "eval_accuracy": 0.5844748858447488,
270
- "eval_loss": 1.7701371908187866,
271
- "eval_runtime": 8.229,
272
- "eval_samples_per_second": 26.613,
273
- "eval_steps_per_second": 3.403,
274
  "step": 250
275
  },
276
  {
277
  "epoch": 2.1,
278
- "grad_norm": 2.001723527908325,
279
- "learning_rate": 9.556451612903226e-05,
280
- "loss": 0.141,
281
  "step": 260
282
  },
283
  {
284
  "epoch": 2.18,
285
- "grad_norm": 0.07203235477209091,
286
- "learning_rate": 9.153225806451613e-05,
287
- "loss": 0.023,
288
  "step": 270
289
  },
290
  {
291
  "epoch": 2.22,
292
- "eval_accuracy": 0.6301369863013698,
293
- "eval_loss": 1.535449743270874,
294
- "eval_runtime": 7.2739,
295
- "eval_samples_per_second": 30.108,
296
- "eval_steps_per_second": 3.849,
297
  "step": 275
298
  },
299
  {
300
  "epoch": 2.26,
301
- "grad_norm": 0.10443231463432312,
302
- "learning_rate": 8.75e-05,
303
- "loss": 0.0434,
304
  "step": 280
305
  },
306
  {
307
  "epoch": 2.34,
308
- "grad_norm": 1.9513520002365112,
309
- "learning_rate": 8.346774193548388e-05,
310
- "loss": 0.0221,
311
  "step": 290
312
  },
313
  {
314
  "epoch": 2.42,
315
- "grad_norm": 4.193704605102539,
316
- "learning_rate": 7.943548387096774e-05,
317
- "loss": 0.028,
318
  "step": 300
319
  },
320
  {
321
  "epoch": 2.42,
322
- "eval_accuracy": 0.6301369863013698,
323
- "eval_loss": 1.6535084247589111,
324
- "eval_runtime": 8.3156,
325
- "eval_samples_per_second": 26.336,
326
- "eval_steps_per_second": 3.367,
327
  "step": 300
328
  },
329
  {
330
  "epoch": 2.5,
331
- "grad_norm": 0.11177629232406616,
332
- "learning_rate": 7.540322580645162e-05,
333
- "loss": 0.0573,
334
  "step": 310
335
  },
336
  {
337
  "epoch": 2.58,
338
- "grad_norm": 0.0510709173977375,
339
- "learning_rate": 7.137096774193549e-05,
340
- "loss": 0.0698,
341
  "step": 320
342
  },
343
  {
344
  "epoch": 2.62,
345
- "eval_accuracy": 0.6438356164383562,
346
- "eval_loss": 1.6771588325500488,
347
- "eval_runtime": 7.5025,
348
- "eval_samples_per_second": 29.19,
349
- "eval_steps_per_second": 3.732,
350
  "step": 325
351
  },
352
  {
353
  "epoch": 2.66,
354
- "grad_norm": 4.947625160217285,
355
  "learning_rate": 6.733870967741935e-05,
356
- "loss": 0.0964,
357
  "step": 330
358
  },
359
  {
360
  "epoch": 2.74,
361
- "grad_norm": 0.07066596299409866,
362
  "learning_rate": 6.330645161290322e-05,
363
- "loss": 0.0792,
364
  "step": 340
365
  },
366
  {
367
  "epoch": 2.82,
368
- "grad_norm": 6.108673572540283,
369
  "learning_rate": 5.9274193548387104e-05,
370
- "loss": 0.0516,
371
  "step": 350
372
  },
373
  {
374
  "epoch": 2.82,
375
- "eval_accuracy": 0.680365296803653,
376
- "eval_loss": 1.4380197525024414,
377
- "eval_runtime": 8.9956,
378
- "eval_samples_per_second": 24.345,
379
- "eval_steps_per_second": 3.113,
380
  "step": 350
381
  },
382
  {
383
  "epoch": 2.9,
384
- "grad_norm": 0.04943337291479111,
385
  "learning_rate": 5.5241935483870966e-05,
386
- "loss": 0.024,
387
  "step": 360
388
  },
389
  {
390
  "epoch": 2.98,
391
- "grad_norm": 0.2530857026576996,
392
  "learning_rate": 5.120967741935484e-05,
393
- "loss": 0.0136,
394
  "step": 370
395
  },
396
  {
397
  "epoch": 3.02,
398
- "eval_accuracy": 0.6484018264840182,
399
- "eval_loss": 1.6561062335968018,
400
- "eval_runtime": 7.9464,
401
- "eval_samples_per_second": 27.56,
402
- "eval_steps_per_second": 3.524,
403
  "step": 375
404
  },
405
  {
406
  "epoch": 3.06,
407
- "grad_norm": 0.07669265568256378,
408
  "learning_rate": 4.7177419354838716e-05,
409
- "loss": 0.0114,
410
  "step": 380
411
  },
412
  {
413
  "epoch": 3.15,
414
- "grad_norm": 0.04155677929520607,
415
  "learning_rate": 4.3145161290322584e-05,
416
- "loss": 0.0109,
417
  "step": 390
418
  },
419
  {
420
  "epoch": 3.23,
421
- "grad_norm": 0.8200215101242065,
422
  "learning_rate": 3.911290322580645e-05,
423
- "loss": 0.0325,
424
  "step": 400
425
  },
426
  {
427
  "epoch": 3.23,
428
- "eval_accuracy": 0.6621004566210046,
429
- "eval_loss": 1.6027804613113403,
430
- "eval_runtime": 8.0216,
431
- "eval_samples_per_second": 27.301,
432
- "eval_steps_per_second": 3.491,
433
  "step": 400
434
  },
435
  {
436
  "epoch": 3.31,
437
- "grad_norm": 0.028287330642342567,
438
  "learning_rate": 3.508064516129033e-05,
439
- "loss": 0.0085,
440
  "step": 410
441
  },
442
  {
443
  "epoch": 3.39,
444
- "grad_norm": 0.05001891031861305,
445
  "learning_rate": 3.1048387096774195e-05,
446
- "loss": 0.0149,
447
  "step": 420
448
  },
449
  {
450
  "epoch": 3.43,
451
- "eval_accuracy": 0.6621004566210046,
452
- "eval_loss": 1.6261166334152222,
453
- "eval_runtime": 7.3858,
454
- "eval_samples_per_second": 29.652,
455
- "eval_steps_per_second": 3.791,
456
  "step": 425
457
  },
458
  {
459
  "epoch": 3.47,
460
- "grad_norm": 0.07340113818645477,
461
  "learning_rate": 2.7016129032258064e-05,
462
- "loss": 0.0104,
463
  "step": 430
464
  },
465
  {
466
  "epoch": 3.55,
467
- "grad_norm": 0.04165619611740112,
468
  "learning_rate": 2.2983870967741935e-05,
469
- "loss": 0.0119,
470
  "step": 440
471
  },
472
  {
473
  "epoch": 3.63,
474
- "grad_norm": 0.032720282673835754,
475
  "learning_rate": 1.8951612903225807e-05,
476
- "loss": 0.0082,
477
  "step": 450
478
  },
479
  {
480
  "epoch": 3.63,
481
- "eval_accuracy": 0.6621004566210046,
482
- "eval_loss": 1.6614654064178467,
483
- "eval_runtime": 7.1739,
484
- "eval_samples_per_second": 30.527,
485
- "eval_steps_per_second": 3.903,
486
  "step": 450
487
  },
488
  {
489
  "epoch": 3.71,
490
- "grad_norm": 0.07252111285924911,
491
  "learning_rate": 1.4919354838709679e-05,
492
- "loss": 0.0079,
493
  "step": 460
494
  },
495
  {
496
  "epoch": 3.79,
497
- "grad_norm": 0.029246076941490173,
498
  "learning_rate": 1.0887096774193549e-05,
499
- "loss": 0.0093,
500
  "step": 470
501
  },
502
  {
503
  "epoch": 3.83,
504
- "eval_accuracy": 0.6529680365296804,
505
- "eval_loss": 1.6878423690795898,
506
- "eval_runtime": 7.9237,
507
- "eval_samples_per_second": 27.639,
508
- "eval_steps_per_second": 3.534,
509
  "step": 475
510
  },
511
  {
512
  "epoch": 3.87,
513
- "grad_norm": 0.029909249395132065,
514
  "learning_rate": 6.854838709677419e-06,
515
- "loss": 0.0087,
516
  "step": 480
517
  },
518
  {
519
  "epoch": 3.95,
520
- "grad_norm": 0.042575616389513016,
521
  "learning_rate": 2.82258064516129e-06,
522
- "loss": 0.0293,
523
  "step": 490
524
  },
525
  {
526
  "epoch": 4.0,
527
  "step": 496,
528
  "total_flos": 6.10974224738132e+17,
529
- "train_loss": 0.14545002643529686,
530
- "train_runtime": 906.411,
531
- "train_samples_per_second": 8.698,
532
- "train_steps_per_second": 0.547
533
  }
534
  ],
535
  "logging_steps": 10,
 
1
  {
2
+ "best_metric": 0.8272958397865295,
3
+ "best_model_checkpoint": "./vit-base-renovation2/checkpoint-250",
4
  "epoch": 4.0,
5
  "eval_steps": 25,
6
  "global_step": 496,
 
10
  "log_history": [
11
  {
12
  "epoch": 0.08,
13
+ "grad_norm": 1.815894365310669,
14
  "learning_rate": 0.00019596774193548388,
15
+ "loss": 1.5846,
16
  "step": 10
17
  },
18
  {
19
  "epoch": 0.16,
20
+ "grad_norm": 2.0016415119171143,
21
  "learning_rate": 0.00019193548387096775,
22
+ "loss": 1.359,
23
  "step": 20
24
  },
25
  {
26
  "epoch": 0.2,
27
+ "eval_accuracy": 0.4657534246575342,
28
+ "eval_loss": 1.2074170112609863,
29
+ "eval_runtime": 7.6692,
30
+ "eval_samples_per_second": 28.556,
31
+ "eval_steps_per_second": 3.651,
32
  "step": 25
33
  },
34
  {
35
  "epoch": 0.24,
36
+ "grad_norm": 1.6700384616851807,
37
  "learning_rate": 0.00018790322580645164,
38
+ "loss": 1.182,
39
  "step": 30
40
  },
41
  {
42
  "epoch": 0.32,
43
+ "grad_norm": 2.4473767280578613,
44
  "learning_rate": 0.00018387096774193548,
45
+ "loss": 1.1559,
46
  "step": 40
47
  },
48
  {
49
  "epoch": 0.4,
50
+ "grad_norm": 2.757627010345459,
51
  "learning_rate": 0.00017983870967741935,
52
+ "loss": 1.1384,
53
  "step": 50
54
  },
55
  {
56
  "epoch": 0.4,
57
+ "eval_accuracy": 0.5205479452054794,
58
+ "eval_loss": 1.1213266849517822,
59
+ "eval_runtime": 8.2359,
60
+ "eval_samples_per_second": 26.591,
61
+ "eval_steps_per_second": 3.4,
62
  "step": 50
63
  },
64
  {
65
  "epoch": 0.48,
66
+ "grad_norm": 1.8923784494400024,
67
  "learning_rate": 0.00017580645161290325,
68
+ "loss": 1.147,
69
  "step": 60
70
  },
71
  {
72
  "epoch": 0.56,
73
+ "grad_norm": 2.0101828575134277,
74
  "learning_rate": 0.00017177419354838711,
75
+ "loss": 1.0866,
76
  "step": 70
77
  },
78
  {
79
  "epoch": 0.6,
80
+ "eval_accuracy": 0.6301369863013698,
81
+ "eval_loss": 0.9746416807174683,
82
+ "eval_runtime": 8.0785,
83
+ "eval_samples_per_second": 27.109,
84
+ "eval_steps_per_second": 3.466,
85
  "step": 75
86
  },
87
  {
88
  "epoch": 0.65,
89
+ "grad_norm": 3.479001522064209,
90
  "learning_rate": 0.00016774193548387098,
91
+ "loss": 0.875,
92
  "step": 80
93
  },
94
  {
95
  "epoch": 0.73,
96
+ "grad_norm": 3.4011082649230957,
97
  "learning_rate": 0.00016370967741935485,
98
+ "loss": 1.1928,
99
  "step": 90
100
  },
101
  {
102
  "epoch": 0.81,
103
+ "grad_norm": 2.3756794929504395,
104
+ "learning_rate": 0.00015967741935483872,
105
+ "loss": 1.1787,
106
  "step": 100
107
  },
108
  {
109
  "epoch": 0.81,
110
+ "eval_accuracy": 0.5662100456621004,
111
+ "eval_loss": 1.0523391962051392,
112
+ "eval_runtime": 9.1559,
113
+ "eval_samples_per_second": 23.919,
114
+ "eval_steps_per_second": 3.058,
115
  "step": 100
116
  },
117
  {
118
  "epoch": 0.89,
119
+ "grad_norm": 1.5179095268249512,
120
+ "learning_rate": 0.0001556451612903226,
121
+ "loss": 0.9827,
122
  "step": 110
123
  },
124
  {
125
  "epoch": 0.97,
126
+ "grad_norm": 2.3550424575805664,
127
+ "learning_rate": 0.00015161290322580646,
128
+ "loss": 0.9242,
129
  "step": 120
130
  },
131
  {
132
  "epoch": 1.01,
133
+ "eval_accuracy": 0.6255707762557078,
134
+ "eval_loss": 0.9543189406394958,
135
+ "eval_runtime": 7.3462,
136
+ "eval_samples_per_second": 29.812,
137
+ "eval_steps_per_second": 3.812,
138
  "step": 125
139
  },
140
  {
141
  "epoch": 1.05,
142
+ "grad_norm": 1.8633378744125366,
143
+ "learning_rate": 0.00014758064516129032,
144
+ "loss": 0.8227,
145
  "step": 130
146
  },
147
  {
148
  "epoch": 1.13,
149
+ "grad_norm": 2.0424883365631104,
150
+ "learning_rate": 0.00014354838709677422,
151
+ "loss": 0.747,
152
  "step": 140
153
  },
154
  {
155
  "epoch": 1.21,
156
+ "grad_norm": 2.2865993976593018,
157
+ "learning_rate": 0.0001395161290322581,
158
+ "loss": 0.7945,
159
  "step": 150
160
  },
161
  {
162
  "epoch": 1.21,
163
+ "eval_accuracy": 0.6118721461187214,
164
+ "eval_loss": 0.9199973940849304,
165
+ "eval_runtime": 7.7884,
166
+ "eval_samples_per_second": 28.119,
167
+ "eval_steps_per_second": 3.595,
168
  "step": 150
169
  },
170
  {
171
  "epoch": 1.29,
172
+ "grad_norm": 1.884401798248291,
173
+ "learning_rate": 0.00013548387096774193,
174
+ "loss": 0.9408,
175
  "step": 160
176
  },
177
  {
178
  "epoch": 1.37,
179
+ "grad_norm": 2.085380792617798,
180
+ "learning_rate": 0.0001314516129032258,
181
+ "loss": 0.8379,
182
  "step": 170
183
  },
184
  {
185
  "epoch": 1.41,
186
+ "eval_accuracy": 0.6712328767123288,
187
+ "eval_loss": 0.8447285294532776,
188
+ "eval_runtime": 7.8794,
189
+ "eval_samples_per_second": 27.794,
190
+ "eval_steps_per_second": 3.554,
191
  "step": 175
192
  },
193
  {
194
  "epoch": 1.45,
195
+ "grad_norm": 2.1437325477600098,
196
+ "learning_rate": 0.0001274193548387097,
197
+ "loss": 0.8348,
198
  "step": 180
199
  },
200
  {
201
  "epoch": 1.53,
202
+ "grad_norm": 1.6616462469100952,
203
+ "learning_rate": 0.00012338709677419356,
204
+ "loss": 0.6521,
205
  "step": 190
206
  },
207
  {
208
  "epoch": 1.61,
209
+ "grad_norm": 2.3329973220825195,
210
+ "learning_rate": 0.00011935483870967743,
211
+ "loss": 0.7253,
212
  "step": 200
213
  },
214
  {
215
  "epoch": 1.61,
216
+ "eval_accuracy": 0.6575342465753424,
217
+ "eval_loss": 0.8642416596412659,
218
+ "eval_runtime": 7.2392,
219
+ "eval_samples_per_second": 30.252,
220
+ "eval_steps_per_second": 3.868,
221
  "step": 200
222
  },
223
  {
224
  "epoch": 1.69,
225
+ "grad_norm": 3.4008474349975586,
226
+ "learning_rate": 0.00011532258064516131,
227
+ "loss": 0.6683,
228
  "step": 210
229
  },
230
  {
231
  "epoch": 1.77,
232
+ "grad_norm": 2.7154157161712646,
233
+ "learning_rate": 0.00011129032258064515,
234
+ "loss": 0.6344,
235
  "step": 220
236
  },
237
  {
238
  "epoch": 1.81,
239
+ "eval_accuracy": 0.6438356164383562,
240
+ "eval_loss": 0.8443467020988464,
241
+ "eval_runtime": 7.7267,
242
+ "eval_samples_per_second": 28.343,
243
+ "eval_steps_per_second": 3.624,
244
  "step": 225
245
  },
246
  {
247
  "epoch": 1.85,
248
+ "grad_norm": 2.2653236389160156,
249
+ "learning_rate": 0.00010725806451612903,
250
+ "loss": 0.768,
251
  "step": 230
252
  },
253
  {
254
  "epoch": 1.94,
255
+ "grad_norm": 2.3153140544891357,
256
+ "learning_rate": 0.0001032258064516129,
257
+ "loss": 0.8194,
258
  "step": 240
259
  },
260
  {
261
  "epoch": 2.02,
262
+ "grad_norm": 1.3728018999099731,
263
+ "learning_rate": 9.919354838709678e-05,
264
+ "loss": 0.6521,
265
  "step": 250
266
  },
267
  {
268
  "epoch": 2.02,
269
+ "eval_accuracy": 0.6666666666666666,
270
+ "eval_loss": 0.8272958397865295,
271
+ "eval_runtime": 7.9935,
272
+ "eval_samples_per_second": 27.397,
273
+ "eval_steps_per_second": 3.503,
274
  "step": 250
275
  },
276
  {
277
  "epoch": 2.1,
278
+ "grad_norm": 1.0636628866195679,
279
+ "learning_rate": 9.516129032258065e-05,
280
+ "loss": 0.3979,
281
  "step": 260
282
  },
283
  {
284
  "epoch": 2.18,
285
+ "grad_norm": 1.3106963634490967,
286
+ "learning_rate": 9.112903225806452e-05,
287
+ "loss": 0.3627,
288
  "step": 270
289
  },
290
  {
291
  "epoch": 2.22,
292
+ "eval_accuracy": 0.6712328767123288,
293
+ "eval_loss": 0.865295946598053,
294
+ "eval_runtime": 7.9627,
295
+ "eval_samples_per_second": 27.503,
296
+ "eval_steps_per_second": 3.516,
297
  "step": 275
298
  },
299
  {
300
  "epoch": 2.26,
301
+ "grad_norm": 1.7618376016616821,
302
+ "learning_rate": 8.709677419354839e-05,
303
+ "loss": 0.3624,
304
  "step": 280
305
  },
306
  {
307
  "epoch": 2.34,
308
+ "grad_norm": 1.3475382328033447,
309
+ "learning_rate": 8.306451612903227e-05,
310
+ "loss": 0.4034,
311
  "step": 290
312
  },
313
  {
314
  "epoch": 2.42,
315
+ "grad_norm": 3.0443389415740967,
316
+ "learning_rate": 7.903225806451613e-05,
317
+ "loss": 0.2523,
318
  "step": 300
319
  },
320
  {
321
  "epoch": 2.42,
322
+ "eval_accuracy": 0.6894977168949772,
323
+ "eval_loss": 0.874794065952301,
324
+ "eval_runtime": 7.2976,
325
+ "eval_samples_per_second": 30.01,
326
+ "eval_steps_per_second": 3.837,
327
  "step": 300
328
  },
329
  {
330
  "epoch": 2.5,
331
+ "grad_norm": 2.961167812347412,
332
+ "learning_rate": 7.500000000000001e-05,
333
+ "loss": 0.3129,
334
  "step": 310
335
  },
336
  {
337
  "epoch": 2.58,
338
+ "grad_norm": 1.5659066438674927,
339
+ "learning_rate": 7.096774193548388e-05,
340
+ "loss": 0.363,
341
  "step": 320
342
  },
343
  {
344
  "epoch": 2.62,
345
+ "eval_accuracy": 0.684931506849315,
346
+ "eval_loss": 0.8407436609268188,
347
+ "eval_runtime": 7.6113,
348
+ "eval_samples_per_second": 28.773,
349
+ "eval_steps_per_second": 3.679,
350
  "step": 325
351
  },
352
  {
353
  "epoch": 2.66,
354
+ "grad_norm": 1.1653140783309937,
355
  "learning_rate": 6.733870967741935e-05,
356
+ "loss": 0.3924,
357
  "step": 330
358
  },
359
  {
360
  "epoch": 2.74,
361
+ "grad_norm": 3.533656358718872,
362
  "learning_rate": 6.330645161290322e-05,
363
+ "loss": 0.3786,
364
  "step": 340
365
  },
366
  {
367
  "epoch": 2.82,
368
+ "grad_norm": 3.4866058826446533,
369
  "learning_rate": 5.9274193548387104e-05,
370
+ "loss": 0.3433,
371
  "step": 350
372
  },
373
  {
374
  "epoch": 2.82,
375
+ "eval_accuracy": 0.6484018264840182,
376
+ "eval_loss": 0.969610333442688,
377
+ "eval_runtime": 8.4636,
378
+ "eval_samples_per_second": 25.875,
379
+ "eval_steps_per_second": 3.308,
380
  "step": 350
381
  },
382
  {
383
  "epoch": 2.9,
384
+ "grad_norm": 0.8572694659233093,
385
  "learning_rate": 5.5241935483870966e-05,
386
+ "loss": 0.3991,
387
  "step": 360
388
  },
389
  {
390
  "epoch": 2.98,
391
+ "grad_norm": 2.2343902587890625,
392
  "learning_rate": 5.120967741935484e-05,
393
+ "loss": 0.2874,
394
  "step": 370
395
  },
396
  {
397
  "epoch": 3.02,
398
+ "eval_accuracy": 0.680365296803653,
399
+ "eval_loss": 0.9290387034416199,
400
+ "eval_runtime": 7.9323,
401
+ "eval_samples_per_second": 27.608,
402
+ "eval_steps_per_second": 3.53,
403
  "step": 375
404
  },
405
  {
406
  "epoch": 3.06,
407
+ "grad_norm": 0.40827465057373047,
408
  "learning_rate": 4.7177419354838716e-05,
409
+ "loss": 0.1757,
410
  "step": 380
411
  },
412
  {
413
  "epoch": 3.15,
414
+ "grad_norm": 0.28673678636550903,
415
  "learning_rate": 4.3145161290322584e-05,
416
+ "loss": 0.122,
417
  "step": 390
418
  },
419
  {
420
  "epoch": 3.23,
421
+ "grad_norm": 1.3551135063171387,
422
  "learning_rate": 3.911290322580645e-05,
423
+ "loss": 0.1682,
424
  "step": 400
425
  },
426
  {
427
  "epoch": 3.23,
428
+ "eval_accuracy": 0.6575342465753424,
429
+ "eval_loss": 0.9713302254676819,
430
+ "eval_runtime": 7.4128,
431
+ "eval_samples_per_second": 29.543,
432
+ "eval_steps_per_second": 3.777,
433
  "step": 400
434
  },
435
  {
436
  "epoch": 3.31,
437
+ "grad_norm": 0.3541167676448822,
438
  "learning_rate": 3.508064516129033e-05,
439
+ "loss": 0.1371,
440
  "step": 410
441
  },
442
  {
443
  "epoch": 3.39,
444
+ "grad_norm": 1.2407793998718262,
445
  "learning_rate": 3.1048387096774195e-05,
446
+ "loss": 0.1575,
447
  "step": 420
448
  },
449
  {
450
  "epoch": 3.43,
451
+ "eval_accuracy": 0.680365296803653,
452
+ "eval_loss": 0.9963440299034119,
453
+ "eval_runtime": 7.3009,
454
+ "eval_samples_per_second": 29.996,
455
+ "eval_steps_per_second": 3.835,
456
  "step": 425
457
  },
458
  {
459
  "epoch": 3.47,
460
+ "grad_norm": 2.0023341178894043,
461
  "learning_rate": 2.7016129032258064e-05,
462
+ "loss": 0.1432,
463
  "step": 430
464
  },
465
  {
466
  "epoch": 3.55,
467
+ "grad_norm": 0.3904053568840027,
468
  "learning_rate": 2.2983870967741935e-05,
469
+ "loss": 0.1827,
470
  "step": 440
471
  },
472
  {
473
  "epoch": 3.63,
474
+ "grad_norm": 0.42743560671806335,
475
  "learning_rate": 1.8951612903225807e-05,
476
+ "loss": 0.0822,
477
  "step": 450
478
  },
479
  {
480
  "epoch": 3.63,
481
+ "eval_accuracy": 0.7123287671232876,
482
+ "eval_loss": 0.9472664594650269,
483
+ "eval_runtime": 7.4585,
484
+ "eval_samples_per_second": 29.362,
485
+ "eval_steps_per_second": 3.754,
486
  "step": 450
487
  },
488
  {
489
  "epoch": 3.71,
490
+ "grad_norm": 0.7994292974472046,
491
  "learning_rate": 1.4919354838709679e-05,
492
+ "loss": 0.133,
493
  "step": 460
494
  },
495
  {
496
  "epoch": 3.79,
497
+ "grad_norm": 0.4317198097705841,
498
  "learning_rate": 1.0887096774193549e-05,
499
+ "loss": 0.1678,
500
  "step": 470
501
  },
502
  {
503
  "epoch": 3.83,
504
+ "eval_accuracy": 0.7031963470319634,
505
+ "eval_loss": 0.9787965416908264,
506
+ "eval_runtime": 7.434,
507
+ "eval_samples_per_second": 29.459,
508
+ "eval_steps_per_second": 3.766,
509
  "step": 475
510
  },
511
  {
512
  "epoch": 3.87,
513
+ "grad_norm": 0.2247464805841446,
514
  "learning_rate": 6.854838709677419e-06,
515
+ "loss": 0.148,
516
  "step": 480
517
  },
518
  {
519
  "epoch": 3.95,
520
+ "grad_norm": 0.28201037645339966,
521
  "learning_rate": 2.82258064516129e-06,
522
+ "loss": 0.1406,
523
  "step": 490
524
  },
525
  {
526
  "epoch": 4.0,
527
  "step": 496,
528
  "total_flos": 6.10974224738132e+17,
529
+ "train_loss": 0.601069396421794,
530
+ "train_runtime": 693.9158,
531
+ "train_samples_per_second": 11.362,
532
+ "train_steps_per_second": 0.715
533
  }
534
  ],
535
  "logging_steps": 10,