c14kevincardenas commited on
Commit
909396d
·
verified ·
1 Parent(s): 1819d20

End of training

Browse files
README.md CHANGED
@@ -1,6 +1,11 @@
1
  ---
2
  library_name: transformers
 
 
3
  tags:
 
 
 
4
  - generated_from_trainer
5
  model-index:
6
  - name: target_hold_hands
@@ -12,9 +17,9 @@ should probably proofread and complete it, then remove this comment. -->
12
 
13
  # target_hold_hands
14
 
15
- This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Loss: 0.6076
18
  - Iou: 0.0000
19
 
20
  ## Model description
 
1
  ---
2
  library_name: transformers
3
+ license: apache-2.0
4
+ base_model: facebook/detr-resnet-50
5
  tags:
6
+ - image-regression
7
+ - human-movement
8
+ - vision
9
  - generated_from_trainer
10
  model-index:
11
  - name: target_hold_hands
 
17
 
18
  # target_hold_hands
19
 
20
+ This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the c14kevincardenas/beta_caller_284_target_hold_hands dataset.
21
  It achieves the following results on the evaluation set:
22
+ - Loss: 0.6006
23
  - Iou: 0.0000
24
 
25
  ## Model description
all_results.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
  "epoch": 20.0,
3
- "eval_IoU": 0.0016660226108877834,
4
- "eval_loss": 0.8514894843101501,
5
- "eval_runtime": 8.7471,
6
- "eval_samples_per_second": 66.308,
7
- "eval_steps_per_second": 1.143,
8
  "total_flos": 0.0,
9
- "train_loss": 1.0301156970170828,
10
- "train_runtime": 3767.5169,
11
- "train_samples_per_second": 17.444,
12
- "train_steps_per_second": 0.276
13
  }
 
1
  {
2
  "epoch": 20.0,
3
+ "eval_IoU": 1.2153273194260533e-07,
4
+ "eval_loss": 0.6006270051002502,
5
+ "eval_runtime": 8.5448,
6
+ "eval_samples_per_second": 67.877,
7
+ "eval_steps_per_second": 1.17,
8
  "total_flos": 0.0,
9
+ "train_loss": 0.6731257475339449,
10
+ "train_runtime": 3556.481,
11
+ "train_samples_per_second": 18.479,
12
+ "train_steps_per_second": 0.292
13
  }
eval_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 20.0,
3
- "eval_IoU": 0.0016660226108877834,
4
- "eval_loss": 0.8514894843101501,
5
- "eval_runtime": 8.7471,
6
- "eval_samples_per_second": 66.308,
7
- "eval_steps_per_second": 1.143
8
  }
 
1
  {
2
  "epoch": 20.0,
3
+ "eval_IoU": 1.2153273194260533e-07,
4
+ "eval_loss": 0.6006270051002502,
5
+ "eval_runtime": 8.5448,
6
+ "eval_samples_per_second": 67.877,
7
+ "eval_steps_per_second": 1.17
8
  }
runs/Oct30_19-31-06_galactica.ad.cirange.net/events.out.tfevents.1730320318.galactica.ad.cirange.net.292105.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60220214345530dd38cbf1079b0617e639f8a56fe2a0986c5eb3a6c50ea2f447
3
+ size 406
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 20.0,
3
  "total_flos": 0.0,
4
- "train_loss": 1.0301156970170828,
5
- "train_runtime": 3767.5169,
6
- "train_samples_per_second": 17.444,
7
- "train_steps_per_second": 0.276
8
  }
 
1
  {
2
  "epoch": 20.0,
3
  "total_flos": 0.0,
4
+ "train_loss": 0.6731257475339449,
5
+ "train_runtime": 3556.481,
6
+ "train_samples_per_second": 18.479,
7
+ "train_steps_per_second": 0.292
8
  }
trainer_state.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
- "best_metric": 0.8514894843101501,
3
- "best_model_checkpoint": "/mnt/ml_drive/kcardenas/target_hold_hands/checkpoint-208",
4
  "epoch": 20.0,
5
  "eval_steps": 500,
6
  "global_step": 1040,
@@ -10,479 +10,479 @@
10
  "log_history": [
11
  {
12
  "epoch": 0.4807692307692308,
13
- "grad_norm": 470173.21875,
14
  "learning_rate": 5e-06,
15
- "loss": 1.6382,
16
  "step": 25
17
  },
18
  {
19
  "epoch": 0.9615384615384616,
20
- "grad_norm": 186560.1875,
21
  "learning_rate": 1e-05,
22
- "loss": 1.2249,
23
  "step": 50
24
  },
25
  {
26
  "epoch": 1.0,
27
- "eval_IoU": 0.00368653304382784,
28
- "eval_loss": 1.0920462608337402,
29
- "eval_runtime": 8.6853,
30
- "eval_samples_per_second": 66.78,
31
- "eval_steps_per_second": 1.151,
32
  "step": 52
33
  },
34
  {
35
  "epoch": 1.4423076923076923,
36
- "grad_norm": 192981.125,
37
  "learning_rate": 1.5e-05,
38
- "loss": 1.0355,
39
  "step": 75
40
  },
41
  {
42
  "epoch": 1.9230769230769231,
43
- "grad_norm": 44049.89453125,
44
  "learning_rate": 2e-05,
45
- "loss": 0.9394,
46
  "step": 100
47
  },
48
  {
49
  "epoch": 2.0,
50
- "eval_IoU": 0.0017734083008499365,
51
- "eval_loss": 0.8837252855300903,
52
- "eval_runtime": 8.4489,
53
- "eval_samples_per_second": 68.648,
54
- "eval_steps_per_second": 1.184,
55
  "step": 104
56
  },
57
  {
58
  "epoch": 2.4038461538461537,
59
- "grad_norm": 17287.9609375,
60
  "learning_rate": 2.5e-05,
61
- "loss": 0.8803,
62
  "step": 125
63
  },
64
  {
65
  "epoch": 2.8846153846153846,
66
- "grad_norm": 27958.970703125,
67
  "learning_rate": 3e-05,
68
- "loss": 0.8744,
69
  "step": 150
70
  },
71
  {
72
  "epoch": 3.0,
73
- "eval_IoU": 0.0014925809955409412,
74
- "eval_loss": 0.8666122555732727,
75
- "eval_runtime": 8.1483,
76
- "eval_samples_per_second": 71.18,
77
- "eval_steps_per_second": 1.227,
78
  "step": 156
79
  },
80
  {
81
  "epoch": 3.3653846153846154,
82
- "grad_norm": 35543.01953125,
83
  "learning_rate": 3.5e-05,
84
- "loss": 0.8684,
85
  "step": 175
86
  },
87
  {
88
  "epoch": 3.8461538461538463,
89
- "grad_norm": 57596.91015625,
90
  "learning_rate": 4e-05,
91
- "loss": 0.8594,
92
  "step": 200
93
  },
94
  {
95
  "epoch": 4.0,
96
- "eval_IoU": 0.0016660226108877834,
97
- "eval_loss": 0.8514894843101501,
98
- "eval_runtime": 8.2606,
99
- "eval_samples_per_second": 70.213,
100
- "eval_steps_per_second": 1.211,
101
  "step": 208
102
  },
103
  {
104
  "epoch": 4.326923076923077,
105
- "grad_norm": 253015.703125,
106
  "learning_rate": 4.5e-05,
107
- "loss": 0.9143,
108
  "step": 225
109
  },
110
  {
111
  "epoch": 4.8076923076923075,
112
- "grad_norm": 84673.3203125,
113
  "learning_rate": 5e-05,
114
- "loss": 1.1734,
115
  "step": 250
116
  },
117
  {
118
  "epoch": 5.0,
119
- "eval_IoU": 0.00037444951155319575,
120
- "eval_loss": 1.1195136308670044,
121
- "eval_runtime": 8.4182,
122
- "eval_samples_per_second": 68.898,
123
- "eval_steps_per_second": 1.188,
124
  "step": 260
125
  },
126
  {
127
  "epoch": 5.288461538461538,
128
- "grad_norm": 83865.71875,
129
  "learning_rate": 4.8417721518987346e-05,
130
- "loss": 1.1296,
131
  "step": 275
132
  },
133
  {
134
  "epoch": 5.769230769230769,
135
- "grad_norm": 56211.53515625,
136
  "learning_rate": 4.683544303797468e-05,
137
- "loss": 1.0918,
138
  "step": 300
139
  },
140
  {
141
  "epoch": 6.0,
142
- "eval_IoU": 0.0,
143
- "eval_loss": 1.073646068572998,
144
- "eval_runtime": 8.1504,
145
- "eval_samples_per_second": 71.162,
146
- "eval_steps_per_second": 1.227,
147
  "step": 312
148
  },
149
  {
150
  "epoch": 6.25,
151
- "grad_norm": 113868.4296875,
152
  "learning_rate": 4.525316455696203e-05,
153
- "loss": 1.0703,
154
  "step": 325
155
  },
156
  {
157
  "epoch": 6.730769230769231,
158
- "grad_norm": 58733.578125,
159
  "learning_rate": 4.367088607594937e-05,
160
- "loss": 1.0759,
161
  "step": 350
162
  },
163
  {
164
  "epoch": 7.0,
165
- "eval_IoU": 0.0,
166
- "eval_loss": 1.0613230466842651,
167
- "eval_runtime": 8.178,
168
- "eval_samples_per_second": 70.922,
169
- "eval_steps_per_second": 1.223,
170
  "step": 364
171
  },
172
  {
173
  "epoch": 7.211538461538462,
174
- "grad_norm": 50991.9609375,
175
  "learning_rate": 4.208860759493671e-05,
176
- "loss": 1.0565,
177
  "step": 375
178
  },
179
  {
180
  "epoch": 7.6923076923076925,
181
- "grad_norm": 148489.953125,
182
  "learning_rate": 4.050632911392405e-05,
183
- "loss": 1.0629,
184
  "step": 400
185
  },
186
  {
187
  "epoch": 8.0,
188
- "eval_IoU": 0.0,
189
- "eval_loss": 1.0518534183502197,
190
- "eval_runtime": 8.2742,
191
- "eval_samples_per_second": 70.097,
192
- "eval_steps_per_second": 1.209,
193
  "step": 416
194
  },
195
  {
196
  "epoch": 8.173076923076923,
197
- "grad_norm": 59951.16796875,
198
  "learning_rate": 3.89240506329114e-05,
199
- "loss": 1.0674,
200
  "step": 425
201
  },
202
  {
203
  "epoch": 8.653846153846153,
204
- "grad_norm": 66094.6484375,
205
  "learning_rate": 3.7341772151898736e-05,
206
- "loss": 1.0504,
207
  "step": 450
208
  },
209
  {
210
  "epoch": 9.0,
211
- "eval_IoU": 0.0,
212
- "eval_loss": 1.0394742488861084,
213
- "eval_runtime": 8.1128,
214
- "eval_samples_per_second": 71.492,
215
- "eval_steps_per_second": 1.233,
216
  "step": 468
217
  },
218
  {
219
  "epoch": 9.134615384615385,
220
- "grad_norm": 87932.0,
221
  "learning_rate": 3.575949367088608e-05,
222
- "loss": 1.0514,
223
  "step": 475
224
  },
225
  {
226
  "epoch": 9.615384615384615,
227
- "grad_norm": 95270.5703125,
228
  "learning_rate": 3.4177215189873416e-05,
229
- "loss": 1.0395,
230
  "step": 500
231
  },
232
  {
233
  "epoch": 10.0,
234
  "eval_IoU": 0.0,
235
- "eval_loss": 1.0350908041000366,
236
- "eval_runtime": 8.6091,
237
- "eval_samples_per_second": 67.371,
238
- "eval_steps_per_second": 1.162,
239
  "step": 520
240
  },
241
  {
242
  "epoch": 10.096153846153847,
243
- "grad_norm": 80274.7890625,
244
  "learning_rate": 3.2594936708860766e-05,
245
- "loss": 1.0496,
246
  "step": 525
247
  },
248
  {
249
  "epoch": 10.576923076923077,
250
- "grad_norm": 84925.7578125,
251
  "learning_rate": 3.10126582278481e-05,
252
- "loss": 1.0352,
253
  "step": 550
254
  },
255
  {
256
  "epoch": 11.0,
257
- "eval_IoU": 0.0,
258
- "eval_loss": 1.0399140119552612,
259
- "eval_runtime": 8.4569,
260
- "eval_samples_per_second": 68.583,
261
- "eval_steps_per_second": 1.182,
262
  "step": 572
263
  },
264
  {
265
  "epoch": 11.057692307692308,
266
- "grad_norm": 79746.3984375,
267
  "learning_rate": 2.9430379746835446e-05,
268
- "loss": 1.0338,
269
  "step": 575
270
  },
271
  {
272
  "epoch": 11.538461538461538,
273
- "grad_norm": 124261.1328125,
274
  "learning_rate": 2.7848101265822786e-05,
275
- "loss": 1.0295,
276
  "step": 600
277
  },
278
  {
279
  "epoch": 12.0,
280
- "eval_IoU": 0.0,
281
- "eval_loss": 1.0345797538757324,
282
- "eval_runtime": 8.5716,
283
- "eval_samples_per_second": 67.666,
284
- "eval_steps_per_second": 1.167,
285
  "step": 624
286
  },
287
  {
288
  "epoch": 12.01923076923077,
289
- "grad_norm": 62221.9375,
290
  "learning_rate": 2.626582278481013e-05,
291
- "loss": 1.0282,
292
  "step": 625
293
  },
294
  {
295
  "epoch": 12.5,
296
- "grad_norm": 75017.7421875,
297
  "learning_rate": 2.468354430379747e-05,
298
- "loss": 1.0263,
299
  "step": 650
300
  },
301
  {
302
  "epoch": 12.98076923076923,
303
- "grad_norm": 88052.2578125,
304
  "learning_rate": 2.3101265822784813e-05,
305
- "loss": 1.023,
306
  "step": 675
307
  },
308
  {
309
  "epoch": 13.0,
310
- "eval_IoU": 0.0,
311
- "eval_loss": 1.0258605480194092,
312
- "eval_runtime": 8.0982,
313
- "eval_samples_per_second": 71.621,
314
- "eval_steps_per_second": 1.235,
315
  "step": 676
316
  },
317
  {
318
  "epoch": 13.461538461538462,
319
- "grad_norm": 84983.6171875,
320
  "learning_rate": 2.1518987341772153e-05,
321
- "loss": 1.0144,
322
  "step": 700
323
  },
324
  {
325
  "epoch": 13.942307692307692,
326
- "grad_norm": 83458.8984375,
327
  "learning_rate": 1.9936708860759496e-05,
328
- "loss": 1.011,
329
  "step": 725
330
  },
331
  {
332
  "epoch": 14.0,
333
- "eval_IoU": 0.0,
334
- "eval_loss": 1.0342016220092773,
335
- "eval_runtime": 8.2238,
336
- "eval_samples_per_second": 70.527,
337
- "eval_steps_per_second": 1.216,
338
  "step": 728
339
  },
340
  {
341
  "epoch": 14.423076923076923,
342
- "grad_norm": 149109.84375,
343
  "learning_rate": 1.8354430379746836e-05,
344
- "loss": 1.0155,
345
  "step": 750
346
  },
347
  {
348
  "epoch": 14.903846153846153,
349
- "grad_norm": 84930.2890625,
350
  "learning_rate": 1.677215189873418e-05,
351
- "loss": 0.9963,
352
  "step": 775
353
  },
354
  {
355
  "epoch": 15.0,
356
  "eval_IoU": 0.0,
357
- "eval_loss": 1.0392385721206665,
358
- "eval_runtime": 8.4881,
359
- "eval_samples_per_second": 68.331,
360
- "eval_steps_per_second": 1.178,
361
  "step": 780
362
  },
363
  {
364
  "epoch": 15.384615384615385,
365
- "grad_norm": 161351.4375,
366
  "learning_rate": 1.5189873417721521e-05,
367
- "loss": 1.006,
368
  "step": 800
369
  },
370
  {
371
  "epoch": 15.865384615384615,
372
- "grad_norm": 119519.0,
373
  "learning_rate": 1.3607594936708861e-05,
374
- "loss": 0.9975,
375
  "step": 825
376
  },
377
  {
378
  "epoch": 16.0,
379
- "eval_IoU": 0.0,
380
- "eval_loss": 1.0270028114318848,
381
- "eval_runtime": 8.9467,
382
- "eval_samples_per_second": 64.829,
383
- "eval_steps_per_second": 1.118,
384
  "step": 832
385
  },
386
  {
387
  "epoch": 16.346153846153847,
388
- "grad_norm": 122443.3671875,
389
  "learning_rate": 1.2025316455696203e-05,
390
- "loss": 0.9927,
391
  "step": 850
392
  },
393
  {
394
  "epoch": 16.826923076923077,
395
- "grad_norm": 142480.4375,
396
  "learning_rate": 1.0443037974683544e-05,
397
- "loss": 0.9879,
398
  "step": 875
399
  },
400
  {
401
  "epoch": 17.0,
402
- "eval_IoU": 0.0,
403
- "eval_loss": 1.0335967540740967,
404
- "eval_runtime": 8.3944,
405
- "eval_samples_per_second": 69.094,
406
- "eval_steps_per_second": 1.191,
407
  "step": 884
408
  },
409
  {
410
  "epoch": 17.307692307692307,
411
- "grad_norm": 347530.125,
412
  "learning_rate": 8.860759493670886e-06,
413
- "loss": 0.9942,
414
  "step": 900
415
  },
416
  {
417
  "epoch": 17.78846153846154,
418
- "grad_norm": 98790.3359375,
419
  "learning_rate": 7.2784810126582285e-06,
420
- "loss": 0.9956,
421
  "step": 925
422
  },
423
  {
424
  "epoch": 18.0,
425
- "eval_IoU": 0.0,
426
- "eval_loss": 1.0295637845993042,
427
- "eval_runtime": 8.6175,
428
- "eval_samples_per_second": 67.305,
429
- "eval_steps_per_second": 1.16,
430
  "step": 936
431
  },
432
  {
433
  "epoch": 18.26923076923077,
434
- "grad_norm": 102745.3046875,
435
  "learning_rate": 5.69620253164557e-06,
436
- "loss": 0.9814,
437
  "step": 950
438
  },
439
  {
440
  "epoch": 18.75,
441
- "grad_norm": 109978.796875,
442
  "learning_rate": 4.113924050632911e-06,
443
- "loss": 0.9753,
444
  "step": 975
445
  },
446
  {
447
  "epoch": 19.0,
448
- "eval_IoU": 0.0,
449
- "eval_loss": 1.0286283493041992,
450
- "eval_runtime": 8.2667,
451
- "eval_samples_per_second": 70.161,
452
- "eval_steps_per_second": 1.21,
453
  "step": 988
454
  },
455
  {
456
  "epoch": 19.23076923076923,
457
- "grad_norm": 98892.78125,
458
  "learning_rate": 2.531645569620253e-06,
459
- "loss": 0.986,
460
  "step": 1000
461
  },
462
  {
463
  "epoch": 19.71153846153846,
464
- "grad_norm": 67631.1015625,
465
  "learning_rate": 9.493670886075951e-07,
466
- "loss": 0.9818,
467
  "step": 1025
468
  },
469
  {
470
  "epoch": 20.0,
471
- "eval_IoU": 0.0,
472
- "eval_loss": 1.0302703380584717,
473
- "eval_runtime": 8.3565,
474
- "eval_samples_per_second": 69.407,
475
- "eval_steps_per_second": 1.197,
476
  "step": 1040
477
  },
478
  {
479
  "epoch": 20.0,
480
  "step": 1040,
481
  "total_flos": 0.0,
482
- "train_loss": 1.0301156970170828,
483
- "train_runtime": 3767.5169,
484
- "train_samples_per_second": 17.444,
485
- "train_steps_per_second": 0.276
486
  }
487
  ],
488
  "logging_steps": 25,
 
1
  {
2
+ "best_metric": 0.6006270051002502,
3
+ "best_model_checkpoint": "/mnt/ml_drive/kcardenas/target_hold_hands/checkpoint-624",
4
  "epoch": 20.0,
5
  "eval_steps": 500,
6
  "global_step": 1040,
 
10
  "log_history": [
11
  {
12
  "epoch": 0.4807692307692308,
13
+ "grad_norm": 772638.0,
14
  "learning_rate": 5e-06,
15
+ "loss": 2.0721,
16
  "step": 25
17
  },
18
  {
19
  "epoch": 0.9615384615384616,
20
+ "grad_norm": 57469.69140625,
21
  "learning_rate": 1e-05,
22
+ "loss": 0.957,
23
  "step": 50
24
  },
25
  {
26
  "epoch": 1.0,
27
+ "eval_IoU": 0.0,
28
+ "eval_loss": 0.8556730151176453,
29
+ "eval_runtime": 8.489,
30
+ "eval_samples_per_second": 68.324,
31
+ "eval_steps_per_second": 1.178,
32
  "step": 52
33
  },
34
  {
35
  "epoch": 1.4423076923076923,
36
+ "grad_norm": 77164.875,
37
  "learning_rate": 1.5e-05,
38
+ "loss": 0.8264,
39
  "step": 75
40
  },
41
  {
42
  "epoch": 1.9230769230769231,
43
+ "grad_norm": 95215.3828125,
44
  "learning_rate": 2e-05,
45
+ "loss": 0.7261,
46
  "step": 100
47
  },
48
  {
49
  "epoch": 2.0,
50
+ "eval_IoU": 7.7851798261353e-09,
51
+ "eval_loss": 0.6803418397903442,
52
+ "eval_runtime": 8.5727,
53
+ "eval_samples_per_second": 67.657,
54
+ "eval_steps_per_second": 1.166,
55
  "step": 104
56
  },
57
  {
58
  "epoch": 2.4038461538461537,
59
+ "grad_norm": 78873.3359375,
60
  "learning_rate": 2.5e-05,
61
+ "loss": 0.6825,
62
  "step": 125
63
  },
64
  {
65
  "epoch": 2.8846153846153846,
66
+ "grad_norm": 65023.43359375,
67
  "learning_rate": 3e-05,
68
+ "loss": 0.6689,
69
  "step": 150
70
  },
71
  {
72
  "epoch": 3.0,
73
+ "eval_IoU": 3.1203175320106754e-07,
74
+ "eval_loss": 0.6544784903526306,
75
+ "eval_runtime": 8.4506,
76
+ "eval_samples_per_second": 68.634,
77
+ "eval_steps_per_second": 1.183,
78
  "step": 156
79
  },
80
  {
81
  "epoch": 3.3653846153846154,
82
+ "grad_norm": 187234.5,
83
  "learning_rate": 3.5e-05,
84
+ "loss": 0.7322,
85
  "step": 175
86
  },
87
  {
88
  "epoch": 3.8461538461538463,
89
+ "grad_norm": 207238.46875,
90
  "learning_rate": 4e-05,
91
+ "loss": 0.7126,
92
  "step": 200
93
  },
94
  {
95
  "epoch": 4.0,
96
+ "eval_IoU": 7.16540375536177e-08,
97
+ "eval_loss": 0.702006459236145,
98
+ "eval_runtime": 8.1834,
99
+ "eval_samples_per_second": 70.875,
100
+ "eval_steps_per_second": 1.222,
101
  "step": 208
102
  },
103
  {
104
  "epoch": 4.326923076923077,
105
+ "grad_norm": 70302.0625,
106
  "learning_rate": 4.5e-05,
107
+ "loss": 0.7118,
108
  "step": 225
109
  },
110
  {
111
  "epoch": 4.8076923076923075,
112
+ "grad_norm": 106005.375,
113
  "learning_rate": 5e-05,
114
+ "loss": 0.6688,
115
  "step": 250
116
  },
117
  {
118
  "epoch": 5.0,
119
+ "eval_IoU": 5.327572640987815e-07,
120
+ "eval_loss": 0.6712496876716614,
121
+ "eval_runtime": 8.4417,
122
+ "eval_samples_per_second": 68.707,
123
+ "eval_steps_per_second": 1.185,
124
  "step": 260
125
  },
126
  {
127
  "epoch": 5.288461538461538,
128
+ "grad_norm": 181192.4375,
129
  "learning_rate": 4.8417721518987346e-05,
130
+ "loss": 0.729,
131
  "step": 275
132
  },
133
  {
134
  "epoch": 5.769230769230769,
135
+ "grad_norm": 527063.125,
136
  "learning_rate": 4.683544303797468e-05,
137
+ "loss": 0.7126,
138
  "step": 300
139
  },
140
  {
141
  "epoch": 6.0,
142
+ "eval_IoU": 8.267290868345697e-08,
143
+ "eval_loss": 0.6633245944976807,
144
+ "eval_runtime": 8.2611,
145
+ "eval_samples_per_second": 70.208,
146
+ "eval_steps_per_second": 1.21,
147
  "step": 312
148
  },
149
  {
150
  "epoch": 6.25,
151
+ "grad_norm": 36219.71875,
152
  "learning_rate": 4.525316455696203e-05,
153
+ "loss": 0.6747,
154
  "step": 325
155
  },
156
  {
157
  "epoch": 6.730769230769231,
158
+ "grad_norm": 89007.9921875,
159
  "learning_rate": 4.367088607594937e-05,
160
+ "loss": 0.6633,
161
  "step": 350
162
  },
163
  {
164
  "epoch": 7.0,
165
+ "eval_IoU": 1.3435934524923367e-07,
166
+ "eval_loss": 0.6083313226699829,
167
+ "eval_runtime": 8.3505,
168
+ "eval_samples_per_second": 69.457,
169
+ "eval_steps_per_second": 1.198,
170
  "step": 364
171
  },
172
  {
173
  "epoch": 7.211538461538462,
174
+ "grad_norm": 32232.62109375,
175
  "learning_rate": 4.208860759493671e-05,
176
+ "loss": 0.611,
177
  "step": 375
178
  },
179
  {
180
  "epoch": 7.6923076923076925,
181
+ "grad_norm": 54438.1484375,
182
  "learning_rate": 4.050632911392405e-05,
183
+ "loss": 0.6113,
184
  "step": 400
185
  },
186
  {
187
  "epoch": 8.0,
188
+ "eval_IoU": 2.002574751254223e-07,
189
+ "eval_loss": 0.6060946583747864,
190
+ "eval_runtime": 8.5578,
191
+ "eval_samples_per_second": 67.774,
192
+ "eval_steps_per_second": 1.169,
193
  "step": 416
194
  },
195
  {
196
  "epoch": 8.173076923076923,
197
+ "grad_norm": 47105.93359375,
198
  "learning_rate": 3.89240506329114e-05,
199
+ "loss": 0.6072,
200
  "step": 425
201
  },
202
  {
203
  "epoch": 8.653846153846153,
204
+ "grad_norm": 37276.02734375,
205
  "learning_rate": 3.7341772151898736e-05,
206
+ "loss": 0.6101,
207
  "step": 450
208
  },
209
  {
210
  "epoch": 9.0,
211
+ "eval_IoU": 1.3619726913206721e-07,
212
+ "eval_loss": 0.6026849150657654,
213
+ "eval_runtime": 8.1149,
214
+ "eval_samples_per_second": 71.473,
215
+ "eval_steps_per_second": 1.232,
216
  "step": 468
217
  },
218
  {
219
  "epoch": 9.134615384615385,
220
+ "grad_norm": 27913.052734375,
221
  "learning_rate": 3.575949367088608e-05,
222
+ "loss": 0.6014,
223
  "step": 475
224
  },
225
  {
226
  "epoch": 9.615384615384615,
227
+ "grad_norm": 44065.1796875,
228
  "learning_rate": 3.4177215189873416e-05,
229
+ "loss": 0.6028,
230
  "step": 500
231
  },
232
  {
233
  "epoch": 10.0,
234
  "eval_IoU": 0.0,
235
+ "eval_loss": 0.6006531715393066,
236
+ "eval_runtime": 8.1108,
237
+ "eval_samples_per_second": 71.509,
238
+ "eval_steps_per_second": 1.233,
239
  "step": 520
240
  },
241
  {
242
  "epoch": 10.096153846153847,
243
+ "grad_norm": 54902.17578125,
244
  "learning_rate": 3.2594936708860766e-05,
245
+ "loss": 0.6069,
246
  "step": 525
247
  },
248
  {
249
  "epoch": 10.576923076923077,
250
+ "grad_norm": 47223.47265625,
251
  "learning_rate": 3.10126582278481e-05,
252
+ "loss": 0.5983,
253
  "step": 550
254
  },
255
  {
256
  "epoch": 11.0,
257
+ "eval_IoU": 3.227043016392884e-09,
258
+ "eval_loss": 0.6019152402877808,
259
+ "eval_runtime": 8.2699,
260
+ "eval_samples_per_second": 70.134,
261
+ "eval_steps_per_second": 1.209,
262
  "step": 572
263
  },
264
  {
265
  "epoch": 11.057692307692308,
266
+ "grad_norm": 44127.5,
267
  "learning_rate": 2.9430379746835446e-05,
268
+ "loss": 0.5984,
269
  "step": 575
270
  },
271
  {
272
  "epoch": 11.538461538461538,
273
+ "grad_norm": 65407.2578125,
274
  "learning_rate": 2.7848101265822786e-05,
275
+ "loss": 0.6014,
276
  "step": 600
277
  },
278
  {
279
  "epoch": 12.0,
280
+ "eval_IoU": 1.2153273194260533e-07,
281
+ "eval_loss": 0.6006270051002502,
282
+ "eval_runtime": 8.3368,
283
+ "eval_samples_per_second": 69.571,
284
+ "eval_steps_per_second": 1.199,
285
  "step": 624
286
  },
287
  {
288
  "epoch": 12.01923076923077,
289
+ "grad_norm": 72088.734375,
290
  "learning_rate": 2.626582278481013e-05,
291
+ "loss": 0.5996,
292
  "step": 625
293
  },
294
  {
295
  "epoch": 12.5,
296
+ "grad_norm": 54947.15234375,
297
  "learning_rate": 2.468354430379747e-05,
298
+ "loss": 0.5988,
299
  "step": 650
300
  },
301
  {
302
  "epoch": 12.98076923076923,
303
+ "grad_norm": 36718.6796875,
304
  "learning_rate": 2.3101265822784813e-05,
305
+ "loss": 0.5968,
306
  "step": 675
307
  },
308
  {
309
  "epoch": 13.0,
310
+ "eval_IoU": 1.0654944237144431e-07,
311
+ "eval_loss": 0.6013975739479065,
312
+ "eval_runtime": 8.4006,
313
+ "eval_samples_per_second": 69.043,
314
+ "eval_steps_per_second": 1.19,
315
  "step": 676
316
  },
317
  {
318
  "epoch": 13.461538461538462,
319
+ "grad_norm": 44921.89453125,
320
  "learning_rate": 2.1518987341772153e-05,
321
+ "loss": 0.599,
322
  "step": 700
323
  },
324
  {
325
  "epoch": 13.942307692307692,
326
+ "grad_norm": 45908.14453125,
327
  "learning_rate": 1.9936708860759496e-05,
328
+ "loss": 0.5932,
329
  "step": 725
330
  },
331
  {
332
  "epoch": 14.0,
333
+ "eval_IoU": 4.530420583131857e-07,
334
+ "eval_loss": 0.6021161675453186,
335
+ "eval_runtime": 7.9829,
336
+ "eval_samples_per_second": 72.656,
337
+ "eval_steps_per_second": 1.253,
338
  "step": 728
339
  },
340
  {
341
  "epoch": 14.423076923076923,
342
+ "grad_norm": 51494.0,
343
  "learning_rate": 1.8354430379746836e-05,
344
+ "loss": 0.5967,
345
  "step": 750
346
  },
347
  {
348
  "epoch": 14.903846153846153,
349
+ "grad_norm": 122982.484375,
350
  "learning_rate": 1.677215189873418e-05,
351
+ "loss": 0.592,
352
  "step": 775
353
  },
354
  {
355
  "epoch": 15.0,
356
  "eval_IoU": 0.0,
357
+ "eval_loss": 0.6046732664108276,
358
+ "eval_runtime": 8.5652,
359
+ "eval_samples_per_second": 67.716,
360
+ "eval_steps_per_second": 1.168,
361
  "step": 780
362
  },
363
  {
364
  "epoch": 15.384615384615385,
365
+ "grad_norm": 51938.21484375,
366
  "learning_rate": 1.5189873417721521e-05,
367
+ "loss": 0.5907,
368
  "step": 800
369
  },
370
  {
371
  "epoch": 15.865384615384615,
372
+ "grad_norm": 316787.84375,
373
  "learning_rate": 1.3607594936708861e-05,
374
+ "loss": 0.5935,
375
  "step": 825
376
  },
377
  {
378
  "epoch": 16.0,
379
+ "eval_IoU": 1.7956697673278814e-07,
380
+ "eval_loss": 0.6020213961601257,
381
+ "eval_runtime": 8.4008,
382
+ "eval_samples_per_second": 69.041,
383
+ "eval_steps_per_second": 1.19,
384
  "step": 832
385
  },
386
  {
387
  "epoch": 16.346153846153847,
388
+ "grad_norm": 31160.560546875,
389
  "learning_rate": 1.2025316455696203e-05,
390
+ "loss": 0.59,
391
  "step": 850
392
  },
393
  {
394
  "epoch": 16.826923076923077,
395
+ "grad_norm": 61832.81640625,
396
  "learning_rate": 1.0443037974683544e-05,
397
+ "loss": 0.5873,
398
  "step": 875
399
  },
400
  {
401
  "epoch": 17.0,
402
+ "eval_IoU": 2.442868828912477e-08,
403
+ "eval_loss": 0.6026484966278076,
404
+ "eval_runtime": 8.4706,
405
+ "eval_samples_per_second": 68.472,
406
+ "eval_steps_per_second": 1.181,
407
  "step": 884
408
  },
409
  {
410
  "epoch": 17.307692307692307,
411
+ "grad_norm": 37131.21484375,
412
  "learning_rate": 8.860759493670886e-06,
413
+ "loss": 0.591,
414
  "step": 900
415
  },
416
  {
417
  "epoch": 17.78846153846154,
418
+ "grad_norm": 42635.22265625,
419
  "learning_rate": 7.2784810126582285e-06,
420
+ "loss": 0.5853,
421
  "step": 925
422
  },
423
  {
424
  "epoch": 18.0,
425
+ "eval_IoU": 6.648389312912634e-09,
426
+ "eval_loss": 0.6114510893821716,
427
+ "eval_runtime": 8.1171,
428
+ "eval_samples_per_second": 71.454,
429
+ "eval_steps_per_second": 1.232,
430
  "step": 936
431
  },
432
  {
433
  "epoch": 18.26923076923077,
434
+ "grad_norm": 40143.515625,
435
  "learning_rate": 5.69620253164557e-06,
436
+ "loss": 0.5918,
437
  "step": 950
438
  },
439
  {
440
  "epoch": 18.75,
441
+ "grad_norm": 38210.26953125,
442
  "learning_rate": 4.113924050632911e-06,
443
+ "loss": 0.5835,
444
  "step": 975
445
  },
446
  {
447
  "epoch": 19.0,
448
+ "eval_IoU": 1.853719873045982e-07,
449
+ "eval_loss": 0.6087284088134766,
450
+ "eval_runtime": 8.3102,
451
+ "eval_samples_per_second": 69.794,
452
+ "eval_steps_per_second": 1.203,
453
  "step": 988
454
  },
455
  {
456
  "epoch": 19.23076923076923,
457
+ "grad_norm": 54590.265625,
458
  "learning_rate": 2.531645569620253e-06,
459
+ "loss": 0.5897,
460
  "step": 1000
461
  },
462
  {
463
  "epoch": 19.71153846153846,
464
+ "grad_norm": 47306.3046875,
465
  "learning_rate": 9.493670886075951e-07,
466
+ "loss": 0.5868,
467
  "step": 1025
468
  },
469
  {
470
  "epoch": 20.0,
471
+ "eval_IoU": 1.5521619976116134e-07,
472
+ "eval_loss": 0.6075738072395325,
473
+ "eval_runtime": 8.5226,
474
+ "eval_samples_per_second": 68.055,
475
+ "eval_steps_per_second": 1.173,
476
  "step": 1040
477
  },
478
  {
479
  "epoch": 20.0,
480
  "step": 1040,
481
  "total_flos": 0.0,
482
+ "train_loss": 0.6731257475339449,
483
+ "train_runtime": 3556.481,
484
+ "train_samples_per_second": 18.479,
485
+ "train_steps_per_second": 0.292
486
  }
487
  ],
488
  "logging_steps": 25,