rshrott commited on
Commit
d4d76e0
1 Parent(s): ab52a27

🍻 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.6529680365296804
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: 1.6878
36
- - Accuracy: 0.6530
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.6027397260273972
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: 1.2384
37
+ - Accuracy: 0.6027
38
 
39
  ## Model description
40
 
all_results.json ADDED
@@ -0,0 +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
+ }
eval_results.json ADDED
@@ -0,0 +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
+ }
runs/Mar20_02-51-07_afa5dd1e3938/events.out.tfevents.1710904009.afa5dd1e3938.399.4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0cd7291fa42cb2140583152703a5b9128a8b4bb974654ce6c8e010089bbefa3
3
+ size 411
train_results.json ADDED
@@ -0,0 +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
+ }
trainer_state.json ADDED
@@ -0,0 +1,544 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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,
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.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,
536
+ "max_steps": 496,
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
544
+ }