enoriega commited on
Commit
e5cca2e
·
1 Parent(s): eb150e3

Finished training

Browse files
all_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.41,
3
+ "eval_accuracy": 0.4117903169627308,
4
+ "eval_loss": 4.301416397094727,
5
+ "eval_runtime": 16.4623,
6
+ "eval_samples_per_second": 607.449,
7
+ "eval_steps_per_second": 19.013,
8
+ "perplexity": 73.80425583574277,
9
+ "train_loss": 1.7241914413088844,
10
+ "train_runtime": 16805.6947,
11
+ "train_samples_per_second": 367.816,
12
+ "train_steps_per_second": 0.046
13
+ }
eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.41,
3
+ "eval_accuracy": 0.4117903169627308,
4
+ "eval_loss": 4.301416397094727,
5
+ "eval_runtime": 16.4623,
6
+ "eval_samples_per_second": 607.449,
7
+ "eval_steps_per_second": 19.013,
8
+ "perplexity": 73.80425583574277
9
+ }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1e0b32f9f01a740c61ab65aa539abfa8f151dbad36b104ae0c51dfcf64d129ee
3
  size 438128811
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee80efbe4a45bba1272164ebe358fe6385a00c65b70b698eee5bef6b48b47fae
3
  size 438128811
runs/May11_21-21-57_r5u29n1.puma.hpc.arizona.edu/events.out.tfevents.1652329329.r5u29n1.puma.hpc.arizona.edu.68465.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:461e02a46e61d78165b7958128e69e23dffb9c2d38c732d98e1aeedcbd9e1030
3
- size 31190
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:04f36827533ff937be5ca8d9cc2657e30827639190ecbc8aa6fdbe52c18139a5
3
+ size 31544
runs/May11_21-21-57_r5u29n1.puma.hpc.arizona.edu/events.out.tfevents.1652346151.r5u29n1.puma.hpc.arizona.edu.68465.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a8d9a6f8efb03680e99bb13714b50d669eee6386e285684da21767ca7a7d9ee0
3
+ size 363
train_results.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.41,
3
+ "train_loss": 1.7241914413088844,
4
+ "train_runtime": 16805.6947,
5
+ "train_samples_per_second": 367.816,
6
+ "train_steps_per_second": 0.046
7
+ }
trainer_state.json ADDED
@@ -0,0 +1,979 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.41527342389411354,
3
+ "best_model_checkpoint": "/groups/claytonm/enoriega/kw_pubmed/kw_pubmed_5000_0.0003/checkpoint-375",
4
+ "epoch": 3.407671998757571,
5
+ "global_step": 525,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.03,
12
+ "learning_rate": 0.0002984415584415584,
13
+ "loss": 5.7783,
14
+ "step": 5
15
+ },
16
+ {
17
+ "epoch": 0.06,
18
+ "learning_rate": 0.0002964935064935065,
19
+ "loss": 3.8225,
20
+ "step": 10
21
+ },
22
+ {
23
+ "epoch": 0.1,
24
+ "learning_rate": 0.0002945454545454545,
25
+ "loss": 3.3158,
26
+ "step": 15
27
+ },
28
+ {
29
+ "epoch": 0.1,
30
+ "eval_accuracy": 0.38301114594218044,
31
+ "eval_loss": 3.8663249015808105,
32
+ "eval_runtime": 16.5167,
33
+ "eval_samples_per_second": 605.449,
34
+ "eval_steps_per_second": 18.951,
35
+ "step": 15
36
+ },
37
+ {
38
+ "epoch": 0.13,
39
+ "learning_rate": 0.0002925974025974026,
40
+ "loss": 3.0324,
41
+ "step": 20
42
+ },
43
+ {
44
+ "epoch": 0.16,
45
+ "learning_rate": 0.0002906493506493506,
46
+ "loss": 2.8348,
47
+ "step": 25
48
+ },
49
+ {
50
+ "epoch": 0.19,
51
+ "learning_rate": 0.0002887012987012987,
52
+ "loss": 2.7061,
53
+ "step": 30
54
+ },
55
+ {
56
+ "epoch": 0.19,
57
+ "eval_accuracy": 0.38962904911180773,
58
+ "eval_loss": 3.9842898845672607,
59
+ "eval_runtime": 16.7828,
60
+ "eval_samples_per_second": 595.847,
61
+ "eval_steps_per_second": 18.65,
62
+ "step": 30
63
+ },
64
+ {
65
+ "epoch": 0.23,
66
+ "learning_rate": 0.00028675324675324674,
67
+ "loss": 2.6148,
68
+ "step": 35
69
+ },
70
+ {
71
+ "epoch": 0.26,
72
+ "learning_rate": 0.00028480519480519477,
73
+ "loss": 2.5071,
74
+ "step": 40
75
+ },
76
+ {
77
+ "epoch": 0.29,
78
+ "learning_rate": 0.0002828571428571428,
79
+ "loss": 2.4347,
80
+ "step": 45
81
+ },
82
+ {
83
+ "epoch": 0.29,
84
+ "eval_accuracy": 0.40025252525252525,
85
+ "eval_loss": 3.96134614944458,
86
+ "eval_runtime": 16.7186,
87
+ "eval_samples_per_second": 598.137,
88
+ "eval_steps_per_second": 18.722,
89
+ "step": 45
90
+ },
91
+ {
92
+ "epoch": 0.32,
93
+ "learning_rate": 0.0002809090909090909,
94
+ "loss": 2.386,
95
+ "step": 50
96
+ },
97
+ {
98
+ "epoch": 0.36,
99
+ "learning_rate": 0.0002789610389610389,
100
+ "loss": 2.3579,
101
+ "step": 55
102
+ },
103
+ {
104
+ "epoch": 0.39,
105
+ "learning_rate": 0.000277012987012987,
106
+ "loss": 2.2818,
107
+ "step": 60
108
+ },
109
+ {
110
+ "epoch": 0.39,
111
+ "eval_accuracy": 0.4000348310693138,
112
+ "eval_loss": 4.00076961517334,
113
+ "eval_runtime": 16.4241,
114
+ "eval_samples_per_second": 608.861,
115
+ "eval_steps_per_second": 19.057,
116
+ "step": 60
117
+ },
118
+ {
119
+ "epoch": 0.42,
120
+ "learning_rate": 0.0002754545454545454,
121
+ "loss": 2.2853,
122
+ "step": 65
123
+ },
124
+ {
125
+ "epoch": 0.45,
126
+ "learning_rate": 0.0002738961038961039,
127
+ "loss": 2.4897,
128
+ "step": 70
129
+ },
130
+ {
131
+ "epoch": 0.49,
132
+ "learning_rate": 0.0002719480519480519,
133
+ "loss": 2.2269,
134
+ "step": 75
135
+ },
136
+ {
137
+ "epoch": 0.49,
138
+ "eval_accuracy": 0.39672587948450017,
139
+ "eval_loss": 4.018723964691162,
140
+ "eval_runtime": 16.4842,
141
+ "eval_samples_per_second": 606.64,
142
+ "eval_steps_per_second": 18.988,
143
+ "step": 75
144
+ },
145
+ {
146
+ "epoch": 0.52,
147
+ "learning_rate": 0.00027,
148
+ "loss": 2.1896,
149
+ "step": 80
150
+ },
151
+ {
152
+ "epoch": 0.55,
153
+ "learning_rate": 0.00026805194805194803,
154
+ "loss": 2.2399,
155
+ "step": 85
156
+ },
157
+ {
158
+ "epoch": 0.58,
159
+ "learning_rate": 0.00026610389610389607,
160
+ "loss": 2.1639,
161
+ "step": 90
162
+ },
163
+ {
164
+ "epoch": 0.58,
165
+ "eval_accuracy": 0.39946882619296414,
166
+ "eval_loss": 4.048261642456055,
167
+ "eval_runtime": 16.4769,
168
+ "eval_samples_per_second": 606.909,
169
+ "eval_steps_per_second": 18.996,
170
+ "step": 90
171
+ },
172
+ {
173
+ "epoch": 0.61,
174
+ "learning_rate": 0.0002641558441558441,
175
+ "loss": 2.2196,
176
+ "step": 95
177
+ },
178
+ {
179
+ "epoch": 0.65,
180
+ "learning_rate": 0.0002622077922077922,
181
+ "loss": 2.1771,
182
+ "step": 100
183
+ },
184
+ {
185
+ "epoch": 0.68,
186
+ "learning_rate": 0.0002602597402597402,
187
+ "loss": 2.096,
188
+ "step": 105
189
+ },
190
+ {
191
+ "epoch": 0.68,
192
+ "eval_accuracy": 0.40103622431208635,
193
+ "eval_loss": 4.030768871307373,
194
+ "eval_runtime": 16.4452,
195
+ "eval_samples_per_second": 608.079,
196
+ "eval_steps_per_second": 19.033,
197
+ "step": 105
198
+ },
199
+ {
200
+ "epoch": 0.71,
201
+ "learning_rate": 0.0002583116883116883,
202
+ "loss": 2.0654,
203
+ "step": 110
204
+ },
205
+ {
206
+ "epoch": 0.74,
207
+ "learning_rate": 0.00025636363636363633,
208
+ "loss": 2.1061,
209
+ "step": 115
210
+ },
211
+ {
212
+ "epoch": 0.78,
213
+ "learning_rate": 0.0002544155844155844,
214
+ "loss": 2.0405,
215
+ "step": 120
216
+ },
217
+ {
218
+ "epoch": 0.78,
219
+ "eval_accuracy": 0.3975095785440613,
220
+ "eval_loss": 4.094750881195068,
221
+ "eval_runtime": 16.44,
222
+ "eval_samples_per_second": 608.272,
223
+ "eval_steps_per_second": 19.039,
224
+ "step": 120
225
+ },
226
+ {
227
+ "epoch": 0.81,
228
+ "learning_rate": 0.00025246753246753245,
229
+ "loss": 2.0486,
230
+ "step": 125
231
+ },
232
+ {
233
+ "epoch": 0.84,
234
+ "learning_rate": 0.0002505194805194805,
235
+ "loss": 2.0271,
236
+ "step": 130
237
+ },
238
+ {
239
+ "epoch": 0.87,
240
+ "learning_rate": 0.00024857142857142857,
241
+ "loss": 2.0288,
242
+ "step": 135
243
+ },
244
+ {
245
+ "epoch": 0.87,
246
+ "eval_accuracy": 0.4073058167885754,
247
+ "eval_loss": 4.073542594909668,
248
+ "eval_runtime": 16.4284,
249
+ "eval_samples_per_second": 608.702,
250
+ "eval_steps_per_second": 19.052,
251
+ "step": 135
252
+ },
253
+ {
254
+ "epoch": 0.91,
255
+ "learning_rate": 0.0002466233766233766,
256
+ "loss": 1.9626,
257
+ "step": 140
258
+ },
259
+ {
260
+ "epoch": 0.94,
261
+ "learning_rate": 0.0002446753246753247,
262
+ "loss": 1.9723,
263
+ "step": 145
264
+ },
265
+ {
266
+ "epoch": 0.97,
267
+ "learning_rate": 0.0002427272727272727,
268
+ "loss": 1.9442,
269
+ "step": 150
270
+ },
271
+ {
272
+ "epoch": 0.97,
273
+ "eval_accuracy": 0.4082636711947057,
274
+ "eval_loss": 4.0442705154418945,
275
+ "eval_runtime": 16.4893,
276
+ "eval_samples_per_second": 606.453,
277
+ "eval_steps_per_second": 18.982,
278
+ "step": 150
279
+ },
280
+ {
281
+ "epoch": 1.01,
282
+ "learning_rate": 0.00024077922077922077,
283
+ "loss": 2.0944,
284
+ "step": 155
285
+ },
286
+ {
287
+ "epoch": 1.04,
288
+ "learning_rate": 0.0002388311688311688,
289
+ "loss": 1.8005,
290
+ "step": 160
291
+ },
292
+ {
293
+ "epoch": 1.07,
294
+ "learning_rate": 0.00023688311688311686,
295
+ "loss": 1.7948,
296
+ "step": 165
297
+ },
298
+ {
299
+ "epoch": 1.07,
300
+ "eval_accuracy": 0.4029519331243469,
301
+ "eval_loss": 4.067967414855957,
302
+ "eval_runtime": 16.8157,
303
+ "eval_samples_per_second": 594.681,
304
+ "eval_steps_per_second": 18.614,
305
+ "step": 165
306
+ },
307
+ {
308
+ "epoch": 1.1,
309
+ "learning_rate": 0.00023493506493506492,
310
+ "loss": 1.7747,
311
+ "step": 170
312
+ },
313
+ {
314
+ "epoch": 1.14,
315
+ "learning_rate": 0.00023298701298701298,
316
+ "loss": 1.7695,
317
+ "step": 175
318
+ },
319
+ {
320
+ "epoch": 1.17,
321
+ "learning_rate": 0.000231038961038961,
322
+ "loss": 1.7509,
323
+ "step": 180
324
+ },
325
+ {
326
+ "epoch": 1.17,
327
+ "eval_accuracy": 0.40434517589690006,
328
+ "eval_loss": 4.150224685668945,
329
+ "eval_runtime": 16.8831,
330
+ "eval_samples_per_second": 592.308,
331
+ "eval_steps_per_second": 18.539,
332
+ "step": 180
333
+ },
334
+ {
335
+ "epoch": 1.2,
336
+ "learning_rate": 0.00022909090909090907,
337
+ "loss": 1.708,
338
+ "step": 185
339
+ },
340
+ {
341
+ "epoch": 1.23,
342
+ "learning_rate": 0.00022714285714285713,
343
+ "loss": 1.699,
344
+ "step": 190
345
+ },
346
+ {
347
+ "epoch": 1.27,
348
+ "learning_rate": 0.00022519480519480519,
349
+ "loss": 1.706,
350
+ "step": 195
351
+ },
352
+ {
353
+ "epoch": 1.27,
354
+ "eval_accuracy": 0.40917798676419365,
355
+ "eval_loss": 4.179904460906982,
356
+ "eval_runtime": 16.415,
357
+ "eval_samples_per_second": 609.2,
358
+ "eval_steps_per_second": 19.068,
359
+ "step": 195
360
+ },
361
+ {
362
+ "epoch": 1.3,
363
+ "learning_rate": 0.00022324675324675322,
364
+ "loss": 1.6846,
365
+ "step": 200
366
+ },
367
+ {
368
+ "epoch": 1.33,
369
+ "learning_rate": 0.00022129870129870128,
370
+ "loss": 1.6696,
371
+ "step": 205
372
+ },
373
+ {
374
+ "epoch": 1.36,
375
+ "learning_rate": 0.00021935064935064933,
376
+ "loss": 1.6689,
377
+ "step": 210
378
+ },
379
+ {
380
+ "epoch": 1.36,
381
+ "eval_accuracy": 0.4087861372344131,
382
+ "eval_loss": 4.159827709197998,
383
+ "eval_runtime": 16.4109,
384
+ "eval_samples_per_second": 609.351,
385
+ "eval_steps_per_second": 19.073,
386
+ "step": 210
387
+ },
388
+ {
389
+ "epoch": 1.39,
390
+ "learning_rate": 0.0002174025974025974,
391
+ "loss": 1.687,
392
+ "step": 215
393
+ },
394
+ {
395
+ "epoch": 1.43,
396
+ "learning_rate": 0.00021545454545454542,
397
+ "loss": 1.65,
398
+ "step": 220
399
+ },
400
+ {
401
+ "epoch": 1.46,
402
+ "learning_rate": 0.00021350649350649348,
403
+ "loss": 1.6472,
404
+ "step": 225
405
+ },
406
+ {
407
+ "epoch": 1.46,
408
+ "eval_accuracy": 0.40599965168930685,
409
+ "eval_loss": 4.144142150878906,
410
+ "eval_runtime": 16.4801,
411
+ "eval_samples_per_second": 606.792,
412
+ "eval_steps_per_second": 18.993,
413
+ "step": 225
414
+ },
415
+ {
416
+ "epoch": 1.49,
417
+ "learning_rate": 0.00021155844155844154,
418
+ "loss": 1.634,
419
+ "step": 230
420
+ },
421
+ {
422
+ "epoch": 1.52,
423
+ "learning_rate": 0.0002096103896103896,
424
+ "loss": 1.6136,
425
+ "step": 235
426
+ },
427
+ {
428
+ "epoch": 1.56,
429
+ "learning_rate": 0.00020766233766233763,
430
+ "loss": 1.6103,
431
+ "step": 240
432
+ },
433
+ {
434
+ "epoch": 1.56,
435
+ "eval_accuracy": 0.41266109369557646,
436
+ "eval_loss": 4.268022537231445,
437
+ "eval_runtime": 16.5084,
438
+ "eval_samples_per_second": 605.753,
439
+ "eval_steps_per_second": 18.96,
440
+ "step": 240
441
+ },
442
+ {
443
+ "epoch": 1.59,
444
+ "learning_rate": 0.0002057142857142857,
445
+ "loss": 1.6145,
446
+ "step": 245
447
+ },
448
+ {
449
+ "epoch": 1.62,
450
+ "learning_rate": 0.00020376623376623372,
451
+ "loss": 1.5818,
452
+ "step": 250
453
+ },
454
+ {
455
+ "epoch": 1.65,
456
+ "learning_rate": 0.0002018181818181818,
457
+ "loss": 1.5862,
458
+ "step": 255
459
+ },
460
+ {
461
+ "epoch": 1.65,
462
+ "eval_accuracy": 0.41431556948798326,
463
+ "eval_loss": 4.115560531616211,
464
+ "eval_runtime": 16.3845,
465
+ "eval_samples_per_second": 610.333,
466
+ "eval_steps_per_second": 19.103,
467
+ "step": 255
468
+ },
469
+ {
470
+ "epoch": 1.69,
471
+ "learning_rate": 0.00019987012987012986,
472
+ "loss": 1.5816,
473
+ "step": 260
474
+ },
475
+ {
476
+ "epoch": 1.72,
477
+ "learning_rate": 0.0001979220779220779,
478
+ "loss": 1.5614,
479
+ "step": 265
480
+ },
481
+ {
482
+ "epoch": 1.75,
483
+ "learning_rate": 0.00019597402597402598,
484
+ "loss": 1.5722,
485
+ "step": 270
486
+ },
487
+ {
488
+ "epoch": 1.75,
489
+ "eval_accuracy": 0.4119644723092999,
490
+ "eval_loss": 4.146672248840332,
491
+ "eval_runtime": 16.4614,
492
+ "eval_samples_per_second": 607.482,
493
+ "eval_steps_per_second": 19.014,
494
+ "step": 270
495
+ },
496
+ {
497
+ "epoch": 1.78,
498
+ "learning_rate": 0.000194025974025974,
499
+ "loss": 1.5387,
500
+ "step": 275
501
+ },
502
+ {
503
+ "epoch": 1.82,
504
+ "learning_rate": 0.00019207792207792207,
505
+ "loss": 1.5353,
506
+ "step": 280
507
+ },
508
+ {
509
+ "epoch": 1.85,
510
+ "learning_rate": 0.0001901298701298701,
511
+ "loss": 1.5182,
512
+ "step": 285
513
+ },
514
+ {
515
+ "epoch": 1.85,
516
+ "eval_accuracy": 0.414272030651341,
517
+ "eval_loss": 4.154214382171631,
518
+ "eval_runtime": 16.4446,
519
+ "eval_samples_per_second": 608.101,
520
+ "eval_steps_per_second": 19.034,
521
+ "step": 285
522
+ },
523
+ {
524
+ "epoch": 1.88,
525
+ "learning_rate": 0.0001881818181818182,
526
+ "loss": 1.5075,
527
+ "step": 290
528
+ },
529
+ {
530
+ "epoch": 1.91,
531
+ "learning_rate": 0.00018623376623376622,
532
+ "loss": 1.5203,
533
+ "step": 295
534
+ },
535
+ {
536
+ "epoch": 1.94,
537
+ "learning_rate": 0.00018428571428571428,
538
+ "loss": 1.5195,
539
+ "step": 300
540
+ },
541
+ {
542
+ "epoch": 1.94,
543
+ "eval_accuracy": 0.41157262277951934,
544
+ "eval_loss": 4.271289825439453,
545
+ "eval_runtime": 16.4758,
546
+ "eval_samples_per_second": 606.95,
547
+ "eval_steps_per_second": 18.998,
548
+ "step": 300
549
+ },
550
+ {
551
+ "epoch": 1.98,
552
+ "learning_rate": 0.0001823376623376623,
553
+ "loss": 1.4973,
554
+ "step": 305
555
+ },
556
+ {
557
+ "epoch": 2.01,
558
+ "learning_rate": 0.0001803896103896104,
559
+ "loss": 1.6131,
560
+ "step": 310
561
+ },
562
+ {
563
+ "epoch": 2.05,
564
+ "learning_rate": 0.00017844155844155843,
565
+ "loss": 1.3535,
566
+ "step": 315
567
+ },
568
+ {
569
+ "epoch": 2.05,
570
+ "eval_accuracy": 0.41322709857192613,
571
+ "eval_loss": 4.2467217445373535,
572
+ "eval_runtime": 16.3639,
573
+ "eval_samples_per_second": 611.102,
574
+ "eval_steps_per_second": 19.127,
575
+ "step": 315
576
+ },
577
+ {
578
+ "epoch": 2.08,
579
+ "learning_rate": 0.00017649350649350648,
580
+ "loss": 1.3275,
581
+ "step": 320
582
+ },
583
+ {
584
+ "epoch": 2.11,
585
+ "learning_rate": 0.00017454545454545452,
586
+ "loss": 1.3251,
587
+ "step": 325
588
+ },
589
+ {
590
+ "epoch": 2.14,
591
+ "learning_rate": 0.0001725974025974026,
592
+ "loss": 1.3322,
593
+ "step": 330
594
+ },
595
+ {
596
+ "epoch": 2.14,
597
+ "eval_accuracy": 0.40691396725879486,
598
+ "eval_loss": 4.347814083099365,
599
+ "eval_runtime": 16.4522,
600
+ "eval_samples_per_second": 607.822,
601
+ "eval_steps_per_second": 19.025,
602
+ "step": 330
603
+ },
604
+ {
605
+ "epoch": 2.17,
606
+ "learning_rate": 0.00017064935064935063,
607
+ "loss": 1.3317,
608
+ "step": 335
609
+ },
610
+ {
611
+ "epoch": 2.21,
612
+ "learning_rate": 0.0001687012987012987,
613
+ "loss": 1.3386,
614
+ "step": 340
615
+ },
616
+ {
617
+ "epoch": 2.24,
618
+ "learning_rate": 0.00016675324675324672,
619
+ "loss": 1.3253,
620
+ "step": 345
621
+ },
622
+ {
623
+ "epoch": 2.24,
624
+ "eval_accuracy": 0.40747997213514453,
625
+ "eval_loss": 4.348999500274658,
626
+ "eval_runtime": 16.4842,
627
+ "eval_samples_per_second": 606.643,
628
+ "eval_steps_per_second": 18.988,
629
+ "step": 345
630
+ },
631
+ {
632
+ "epoch": 2.27,
633
+ "learning_rate": 0.0001648051948051948,
634
+ "loss": 1.3325,
635
+ "step": 350
636
+ },
637
+ {
638
+ "epoch": 2.3,
639
+ "learning_rate": 0.00016285714285714284,
640
+ "loss": 1.3262,
641
+ "step": 355
642
+ },
643
+ {
644
+ "epoch": 2.34,
645
+ "learning_rate": 0.0001609090909090909,
646
+ "loss": 1.3052,
647
+ "step": 360
648
+ },
649
+ {
650
+ "epoch": 2.34,
651
+ "eval_accuracy": 0.41448972483455243,
652
+ "eval_loss": 4.312870025634766,
653
+ "eval_runtime": 16.4771,
654
+ "eval_samples_per_second": 606.902,
655
+ "eval_steps_per_second": 18.996,
656
+ "step": 360
657
+ },
658
+ {
659
+ "epoch": 2.37,
660
+ "learning_rate": 0.00015896103896103893,
661
+ "loss": 1.3058,
662
+ "step": 365
663
+ },
664
+ {
665
+ "epoch": 2.4,
666
+ "learning_rate": 0.00015701298701298701,
667
+ "loss": 1.297,
668
+ "step": 370
669
+ },
670
+ {
671
+ "epoch": 2.43,
672
+ "learning_rate": 0.00015506493506493505,
673
+ "loss": 1.3093,
674
+ "step": 375
675
+ },
676
+ {
677
+ "epoch": 2.43,
678
+ "eval_accuracy": 0.41527342389411354,
679
+ "eval_loss": 4.299270153045654,
680
+ "eval_runtime": 16.3482,
681
+ "eval_samples_per_second": 611.689,
682
+ "eval_steps_per_second": 19.146,
683
+ "step": 375
684
+ },
685
+ {
686
+ "epoch": 2.47,
687
+ "learning_rate": 0.0001531168831168831,
688
+ "loss": 1.2818,
689
+ "step": 380
690
+ },
691
+ {
692
+ "epoch": 2.5,
693
+ "learning_rate": 0.00015116883116883114,
694
+ "loss": 1.3043,
695
+ "step": 385
696
+ },
697
+ {
698
+ "epoch": 2.53,
699
+ "learning_rate": 0.0001492207792207792,
700
+ "loss": 1.2865,
701
+ "step": 390
702
+ },
703
+ {
704
+ "epoch": 2.53,
705
+ "eval_accuracy": 0.4116597004528039,
706
+ "eval_loss": 4.3583269119262695,
707
+ "eval_runtime": 17.7946,
708
+ "eval_samples_per_second": 561.97,
709
+ "eval_steps_per_second": 17.59,
710
+ "step": 390
711
+ },
712
+ {
713
+ "epoch": 2.56,
714
+ "learning_rate": 0.00014727272727272725,
715
+ "loss": 1.2959,
716
+ "step": 395
717
+ },
718
+ {
719
+ "epoch": 2.6,
720
+ "learning_rate": 0.0001453246753246753,
721
+ "loss": 1.2998,
722
+ "step": 400
723
+ },
724
+ {
725
+ "epoch": 2.63,
726
+ "learning_rate": 0.00014337662337662337,
727
+ "loss": 1.2934,
728
+ "step": 405
729
+ },
730
+ {
731
+ "epoch": 2.63,
732
+ "eval_accuracy": 0.4146203413444793,
733
+ "eval_loss": 4.293191909790039,
734
+ "eval_runtime": 16.4542,
735
+ "eval_samples_per_second": 607.749,
736
+ "eval_steps_per_second": 19.023,
737
+ "step": 405
738
+ },
739
+ {
740
+ "epoch": 2.66,
741
+ "learning_rate": 0.0001414285714285714,
742
+ "loss": 1.2983,
743
+ "step": 410
744
+ },
745
+ {
746
+ "epoch": 2.69,
747
+ "learning_rate": 0.00013948051948051946,
748
+ "loss": 1.2949,
749
+ "step": 415
750
+ },
751
+ {
752
+ "epoch": 2.72,
753
+ "learning_rate": 0.00013753246753246752,
754
+ "loss": 1.2967,
755
+ "step": 420
756
+ },
757
+ {
758
+ "epoch": 2.72,
759
+ "eval_accuracy": 0.4104841518634622,
760
+ "eval_loss": 4.251837730407715,
761
+ "eval_runtime": 16.3722,
762
+ "eval_samples_per_second": 610.79,
763
+ "eval_steps_per_second": 19.118,
764
+ "step": 420
765
+ },
766
+ {
767
+ "epoch": 2.76,
768
+ "learning_rate": 0.00013558441558441558,
769
+ "loss": 1.2745,
770
+ "step": 425
771
+ },
772
+ {
773
+ "epoch": 2.79,
774
+ "learning_rate": 0.0001336363636363636,
775
+ "loss": 1.276,
776
+ "step": 430
777
+ },
778
+ {
779
+ "epoch": 2.82,
780
+ "learning_rate": 0.00013168831168831167,
781
+ "loss": 1.2706,
782
+ "step": 435
783
+ },
784
+ {
785
+ "epoch": 2.82,
786
+ "eval_accuracy": 0.4111807732497388,
787
+ "eval_loss": 4.352819919586182,
788
+ "eval_runtime": 16.376,
789
+ "eval_samples_per_second": 610.648,
790
+ "eval_steps_per_second": 19.113,
791
+ "step": 435
792
+ },
793
+ {
794
+ "epoch": 2.85,
795
+ "learning_rate": 0.00012974025974025972,
796
+ "loss": 1.2837,
797
+ "step": 440
798
+ },
799
+ {
800
+ "epoch": 2.89,
801
+ "learning_rate": 0.00012779220779220778,
802
+ "loss": 1.2779,
803
+ "step": 445
804
+ },
805
+ {
806
+ "epoch": 2.92,
807
+ "learning_rate": 0.00012584415584415581,
808
+ "loss": 1.2585,
809
+ "step": 450
810
+ },
811
+ {
812
+ "epoch": 2.92,
813
+ "eval_accuracy": 0.4151428073841867,
814
+ "eval_loss": 4.317609786987305,
815
+ "eval_runtime": 16.4566,
816
+ "eval_samples_per_second": 607.659,
817
+ "eval_steps_per_second": 19.02,
818
+ "step": 450
819
+ },
820
+ {
821
+ "epoch": 2.95,
822
+ "learning_rate": 0.0001238961038961039,
823
+ "loss": 1.2726,
824
+ "step": 455
825
+ },
826
+ {
827
+ "epoch": 2.98,
828
+ "learning_rate": 0.00012194805194805194,
829
+ "loss": 1.2667,
830
+ "step": 460
831
+ },
832
+ {
833
+ "epoch": 3.02,
834
+ "learning_rate": 0.00011999999999999999,
835
+ "loss": 1.3254,
836
+ "step": 465
837
+ },
838
+ {
839
+ "epoch": 3.02,
840
+ "eval_accuracy": 0.40573841866945315,
841
+ "eval_loss": 4.5338358879089355,
842
+ "eval_runtime": 16.961,
843
+ "eval_samples_per_second": 589.587,
844
+ "eval_steps_per_second": 18.454,
845
+ "step": 465
846
+ },
847
+ {
848
+ "epoch": 3.05,
849
+ "learning_rate": 0.00011805194805194805,
850
+ "loss": 1.1489,
851
+ "step": 470
852
+ },
853
+ {
854
+ "epoch": 3.08,
855
+ "learning_rate": 0.00011610389610389609,
856
+ "loss": 1.1305,
857
+ "step": 475
858
+ },
859
+ {
860
+ "epoch": 3.12,
861
+ "learning_rate": 0.00011415584415584415,
862
+ "loss": 1.1389,
863
+ "step": 480
864
+ },
865
+ {
866
+ "epoch": 3.12,
867
+ "eval_accuracy": 0.4110066179031696,
868
+ "eval_loss": 4.393136978149414,
869
+ "eval_runtime": 16.4067,
870
+ "eval_samples_per_second": 609.508,
871
+ "eval_steps_per_second": 19.078,
872
+ "step": 480
873
+ },
874
+ {
875
+ "epoch": 3.15,
876
+ "learning_rate": 0.0001122077922077922,
877
+ "loss": 1.138,
878
+ "step": 485
879
+ },
880
+ {
881
+ "epoch": 3.18,
882
+ "learning_rate": 0.00011025974025974025,
883
+ "loss": 1.1166,
884
+ "step": 490
885
+ },
886
+ {
887
+ "epoch": 3.21,
888
+ "learning_rate": 0.0001083116883116883,
889
+ "loss": 1.149,
890
+ "step": 495
891
+ },
892
+ {
893
+ "epoch": 3.21,
894
+ "eval_accuracy": 0.4105712295367468,
895
+ "eval_loss": 4.424084663391113,
896
+ "eval_runtime": 16.3916,
897
+ "eval_samples_per_second": 610.068,
898
+ "eval_steps_per_second": 19.095,
899
+ "step": 495
900
+ },
901
+ {
902
+ "epoch": 3.25,
903
+ "learning_rate": 0.00010636363636363636,
904
+ "loss": 1.1305,
905
+ "step": 500
906
+ },
907
+ {
908
+ "epoch": 3.28,
909
+ "learning_rate": 0.0001044155844155844,
910
+ "loss": 1.1262,
911
+ "step": 505
912
+ },
913
+ {
914
+ "epoch": 3.31,
915
+ "learning_rate": 0.00010246753246753246,
916
+ "loss": 1.1432,
917
+ "step": 510
918
+ },
919
+ {
920
+ "epoch": 3.31,
921
+ "eval_accuracy": 0.41122431208638105,
922
+ "eval_loss": 4.4350810050964355,
923
+ "eval_runtime": 16.3705,
924
+ "eval_samples_per_second": 610.854,
925
+ "eval_steps_per_second": 19.12,
926
+ "step": 510
927
+ },
928
+ {
929
+ "epoch": 3.34,
930
+ "learning_rate": 0.0001005194805194805,
931
+ "loss": 1.1285,
932
+ "step": 515
933
+ },
934
+ {
935
+ "epoch": 3.38,
936
+ "learning_rate": 9.857142857142856e-05,
937
+ "loss": 1.1175,
938
+ "step": 520
939
+ },
940
+ {
941
+ "epoch": 3.41,
942
+ "learning_rate": 9.662337662337661e-05,
943
+ "loss": 1.1383,
944
+ "step": 525
945
+ },
946
+ {
947
+ "epoch": 3.41,
948
+ "eval_accuracy": 0.40865552072448624,
949
+ "eval_loss": 4.405033111572266,
950
+ "eval_runtime": 16.4548,
951
+ "eval_samples_per_second": 607.724,
952
+ "eval_steps_per_second": 19.022,
953
+ "step": 525
954
+ },
955
+ {
956
+ "epoch": 3.41,
957
+ "step": 525,
958
+ "total_flos": 3.4290279623915635e+17,
959
+ "train_loss": 1.7241914413088844,
960
+ "train_runtime": 16805.6947,
961
+ "train_samples_per_second": 367.816,
962
+ "train_steps_per_second": 0.046
963
+ },
964
+ {
965
+ "epoch": 3.41,
966
+ "eval_accuracy": 0.4117903169627308,
967
+ "eval_loss": 4.301416397094727,
968
+ "eval_runtime": 16.4623,
969
+ "eval_samples_per_second": 607.449,
970
+ "eval_steps_per_second": 19.013,
971
+ "step": 525
972
+ }
973
+ ],
974
+ "max_steps": 770,
975
+ "num_train_epochs": 5,
976
+ "total_flos": 3.4290279623915635e+17,
977
+ "trial_name": null,
978
+ "trial_params": null
979
+ }