Hartunka commited on
Commit
efee11c
·
verified ·
1 Parent(s): 4fb260c

End of training

Browse files
Files changed (5) hide show
  1. README.md +16 -4
  2. all_results.json +16 -0
  3. eval_results.json +10 -0
  4. train_results.json +9 -0
  5. trainer_state.json +921 -0
README.md CHANGED
@@ -2,11 +2,23 @@
2
  library_name: transformers
3
  tags:
4
  - generated_from_trainer
 
 
5
  metrics:
6
  - accuracy
7
  model-index:
8
  - name: bert_base_km_20_v2
9
- results: []
 
 
 
 
 
 
 
 
 
 
10
  ---
11
 
12
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -14,10 +26,10 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # bert_base_km_20_v2
16
 
17
- This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 7.0059
20
- - Accuracy: 0.1531
21
 
22
  ## Model description
23
 
 
2
  library_name: transformers
3
  tags:
4
  - generated_from_trainer
5
+ datasets:
6
+ - Hartunka/processed_wikitext-103-raw-v1-km-20
7
  metrics:
8
  - accuracy
9
  model-index:
10
  - name: bert_base_km_20_v2
11
+ results:
12
+ - task:
13
+ name: Masked Language Modeling
14
+ type: fill-mask
15
+ dataset:
16
+ name: Hartunka/processed_wikitext-103-raw-v1-km-20
17
+ type: Hartunka/processed_wikitext-103-raw-v1-km-20
18
+ metrics:
19
+ - name: Accuracy
20
+ type: accuracy
21
+ value: 0.15400915409391422
22
  ---
23
 
24
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
26
 
27
  # bert_base_km_20_v2
28
 
29
+ This model is a fine-tuned version of [](https://huggingface.co/) on the Hartunka/processed_wikitext-103-raw-v1-km-20 dataset.
30
  It achieves the following results on the evaluation set:
31
+ - Loss: 6.7230
32
+ - Accuracy: 0.1540
33
 
34
  ## Model description
35
 
all_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 25.0,
3
+ "eval_accuracy": 0.15400915409391422,
4
+ "eval_loss": 6.723025321960449,
5
+ "eval_runtime": 1.0283,
6
+ "eval_samples": 479,
7
+ "eval_samples_per_second": 465.802,
8
+ "eval_steps_per_second": 4.862,
9
+ "perplexity": 831.3287480343491,
10
+ "total_flos": 1.5151126801503744e+18,
11
+ "train_loss": 6.378149575490495,
12
+ "train_runtime": 24807.4865,
13
+ "train_samples": 228639,
14
+ "train_samples_per_second": 230.413,
15
+ "train_steps_per_second": 2.4
16
+ }
eval_results.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 25.0,
3
+ "eval_accuracy": 0.15400915409391422,
4
+ "eval_loss": 6.723025321960449,
5
+ "eval_runtime": 1.0283,
6
+ "eval_samples": 479,
7
+ "eval_samples_per_second": 465.802,
8
+ "eval_steps_per_second": 4.862,
9
+ "perplexity": 831.3287480343491
10
+ }
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 25.0,
3
+ "total_flos": 1.5151126801503744e+18,
4
+ "train_loss": 6.378149575490495,
5
+ "train_runtime": 24807.4865,
6
+ "train_samples": 228639,
7
+ "train_samples_per_second": 230.413,
8
+ "train_steps_per_second": 2.4
9
+ }
trainer_state.json ADDED
@@ -0,0 +1,921 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 20000,
3
+ "best_metric": 6.701448440551758,
4
+ "best_model_checkpoint": "bert_base_km_20_v2/checkpoint-20000",
5
+ "epoch": 25.0,
6
+ "eval_steps": 10000,
7
+ "global_step": 59550,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.20990764063811923,
14
+ "grad_norm": 3.2586686611175537,
15
+ "learning_rate": 5e-06,
16
+ "loss": 12.1717,
17
+ "step": 500
18
+ },
19
+ {
20
+ "epoch": 0.41981528127623846,
21
+ "grad_norm": 4.922037601470947,
22
+ "learning_rate": 1e-05,
23
+ "loss": 10.0311,
24
+ "step": 1000
25
+ },
26
+ {
27
+ "epoch": 0.6297229219143576,
28
+ "grad_norm": 28.822181701660156,
29
+ "learning_rate": 1.5e-05,
30
+ "loss": 8.5295,
31
+ "step": 1500
32
+ },
33
+ {
34
+ "epoch": 0.8396305625524769,
35
+ "grad_norm": 13.167506217956543,
36
+ "learning_rate": 2e-05,
37
+ "loss": 8.0722,
38
+ "step": 2000
39
+ },
40
+ {
41
+ "epoch": 1.0495382031905962,
42
+ "grad_norm": 16.0736141204834,
43
+ "learning_rate": 2.5e-05,
44
+ "loss": 7.8283,
45
+ "step": 2500
46
+ },
47
+ {
48
+ "epoch": 1.2594458438287153,
49
+ "grad_norm": 14.928688049316406,
50
+ "learning_rate": 3e-05,
51
+ "loss": 7.6626,
52
+ "step": 3000
53
+ },
54
+ {
55
+ "epoch": 1.4693534844668346,
56
+ "grad_norm": 6.039346218109131,
57
+ "learning_rate": 3.5e-05,
58
+ "loss": 7.5058,
59
+ "step": 3500
60
+ },
61
+ {
62
+ "epoch": 1.6792611251049538,
63
+ "grad_norm": 12.346041679382324,
64
+ "learning_rate": 4e-05,
65
+ "loss": 7.3977,
66
+ "step": 4000
67
+ },
68
+ {
69
+ "epoch": 1.8891687657430731,
70
+ "grad_norm": 6.854773044586182,
71
+ "learning_rate": 4.5e-05,
72
+ "loss": 7.3026,
73
+ "step": 4500
74
+ },
75
+ {
76
+ "epoch": 2.0990764063811924,
77
+ "grad_norm": 7.6127848625183105,
78
+ "learning_rate": 5e-05,
79
+ "loss": 7.1895,
80
+ "step": 5000
81
+ },
82
+ {
83
+ "epoch": 2.3089840470193117,
84
+ "grad_norm": 7.826136589050293,
85
+ "learning_rate": 5.500000000000001e-05,
86
+ "loss": 7.1128,
87
+ "step": 5500
88
+ },
89
+ {
90
+ "epoch": 2.5188916876574305,
91
+ "grad_norm": 8.711495399475098,
92
+ "learning_rate": 6e-05,
93
+ "loss": 7.0718,
94
+ "step": 6000
95
+ },
96
+ {
97
+ "epoch": 2.72879932829555,
98
+ "grad_norm": 4.3706536293029785,
99
+ "learning_rate": 6.500000000000001e-05,
100
+ "loss": 7.0242,
101
+ "step": 6500
102
+ },
103
+ {
104
+ "epoch": 2.938706968933669,
105
+ "grad_norm": 7.588512420654297,
106
+ "learning_rate": 7e-05,
107
+ "loss": 6.9993,
108
+ "step": 7000
109
+ },
110
+ {
111
+ "epoch": 3.1486146095717884,
112
+ "grad_norm": 6.372596740722656,
113
+ "learning_rate": 7.500000000000001e-05,
114
+ "loss": 6.9168,
115
+ "step": 7500
116
+ },
117
+ {
118
+ "epoch": 3.3585222502099077,
119
+ "grad_norm": 5.352084159851074,
120
+ "learning_rate": 8e-05,
121
+ "loss": 6.8948,
122
+ "step": 8000
123
+ },
124
+ {
125
+ "epoch": 3.568429890848027,
126
+ "grad_norm": 4.005120277404785,
127
+ "learning_rate": 8.5e-05,
128
+ "loss": 6.8742,
129
+ "step": 8500
130
+ },
131
+ {
132
+ "epoch": 3.7783375314861463,
133
+ "grad_norm": 3.1480915546417236,
134
+ "learning_rate": 9e-05,
135
+ "loss": 6.8502,
136
+ "step": 9000
137
+ },
138
+ {
139
+ "epoch": 3.988245172124265,
140
+ "grad_norm": 2.2671101093292236,
141
+ "learning_rate": 9.5e-05,
142
+ "loss": 6.8378,
143
+ "step": 9500
144
+ },
145
+ {
146
+ "epoch": 4.198152812762385,
147
+ "grad_norm": 3.7820088863372803,
148
+ "learning_rate": 0.0001,
149
+ "loss": 6.741,
150
+ "step": 10000
151
+ },
152
+ {
153
+ "epoch": 4.198152812762385,
154
+ "eval_accuracy": 0.15047719747349675,
155
+ "eval_loss": 6.86174201965332,
156
+ "eval_runtime": 1.0657,
157
+ "eval_samples_per_second": 449.453,
158
+ "eval_steps_per_second": 4.692,
159
+ "step": 10000
160
+ },
161
+ {
162
+ "epoch": 4.408060453400504,
163
+ "grad_norm": 2.66556978225708,
164
+ "learning_rate": 9.899091826437942e-05,
165
+ "loss": 6.741,
166
+ "step": 10500
167
+ },
168
+ {
169
+ "epoch": 4.617968094038623,
170
+ "grad_norm": 3.0062286853790283,
171
+ "learning_rate": 9.798183652875883e-05,
172
+ "loss": 6.7367,
173
+ "step": 11000
174
+ },
175
+ {
176
+ "epoch": 4.827875734676742,
177
+ "grad_norm": 2.927842855453491,
178
+ "learning_rate": 9.697275479313825e-05,
179
+ "loss": 6.7229,
180
+ "step": 11500
181
+ },
182
+ {
183
+ "epoch": 5.037783375314861,
184
+ "grad_norm": 3.1912167072296143,
185
+ "learning_rate": 9.596367305751767e-05,
186
+ "loss": 6.6724,
187
+ "step": 12000
188
+ },
189
+ {
190
+ "epoch": 5.247691015952981,
191
+ "grad_norm": 3.3745853900909424,
192
+ "learning_rate": 9.495459132189707e-05,
193
+ "loss": 6.6049,
194
+ "step": 12500
195
+ },
196
+ {
197
+ "epoch": 5.4575986565911,
198
+ "grad_norm": 2.804530382156372,
199
+ "learning_rate": 9.394550958627649e-05,
200
+ "loss": 6.5968,
201
+ "step": 13000
202
+ },
203
+ {
204
+ "epoch": 5.667506297229219,
205
+ "grad_norm": 3.8129560947418213,
206
+ "learning_rate": 9.293642785065591e-05,
207
+ "loss": 6.5992,
208
+ "step": 13500
209
+ },
210
+ {
211
+ "epoch": 5.877413937867338,
212
+ "grad_norm": 2.3966245651245117,
213
+ "learning_rate": 9.192734611503532e-05,
214
+ "loss": 6.5829,
215
+ "step": 14000
216
+ },
217
+ {
218
+ "epoch": 6.087321578505458,
219
+ "grad_norm": 3.265836477279663,
220
+ "learning_rate": 9.091826437941474e-05,
221
+ "loss": 6.5406,
222
+ "step": 14500
223
+ },
224
+ {
225
+ "epoch": 6.297229219143577,
226
+ "grad_norm": 2.9976537227630615,
227
+ "learning_rate": 8.990918264379415e-05,
228
+ "loss": 6.4934,
229
+ "step": 15000
230
+ },
231
+ {
232
+ "epoch": 6.507136859781696,
233
+ "grad_norm": 1.8768919706344604,
234
+ "learning_rate": 8.890010090817356e-05,
235
+ "loss": 6.4939,
236
+ "step": 15500
237
+ },
238
+ {
239
+ "epoch": 6.717044500419815,
240
+ "grad_norm": 3.4282023906707764,
241
+ "learning_rate": 8.789101917255298e-05,
242
+ "loss": 6.4879,
243
+ "step": 16000
244
+ },
245
+ {
246
+ "epoch": 6.926952141057934,
247
+ "grad_norm": 3.0579543113708496,
248
+ "learning_rate": 8.68819374369324e-05,
249
+ "loss": 6.4843,
250
+ "step": 16500
251
+ },
252
+ {
253
+ "epoch": 7.136859781696054,
254
+ "grad_norm": 2.4434196949005127,
255
+ "learning_rate": 8.58728557013118e-05,
256
+ "loss": 6.4169,
257
+ "step": 17000
258
+ },
259
+ {
260
+ "epoch": 7.346767422334173,
261
+ "grad_norm": 2.620242118835449,
262
+ "learning_rate": 8.486377396569122e-05,
263
+ "loss": 6.3983,
264
+ "step": 17500
265
+ },
266
+ {
267
+ "epoch": 7.5566750629722925,
268
+ "grad_norm": 2.902372121810913,
269
+ "learning_rate": 8.385469223007064e-05,
270
+ "loss": 6.4067,
271
+ "step": 18000
272
+ },
273
+ {
274
+ "epoch": 7.766582703610411,
275
+ "grad_norm": 2.9418728351593018,
276
+ "learning_rate": 8.284561049445006e-05,
277
+ "loss": 6.4046,
278
+ "step": 18500
279
+ },
280
+ {
281
+ "epoch": 7.97649034424853,
282
+ "grad_norm": 3.1700100898742676,
283
+ "learning_rate": 8.183652875882947e-05,
284
+ "loss": 6.4043,
285
+ "step": 19000
286
+ },
287
+ {
288
+ "epoch": 8.18639798488665,
289
+ "grad_norm": 2.1963798999786377,
290
+ "learning_rate": 8.082744702320889e-05,
291
+ "loss": 6.3264,
292
+ "step": 19500
293
+ },
294
+ {
295
+ "epoch": 8.39630562552477,
296
+ "grad_norm": 2.7740824222564697,
297
+ "learning_rate": 7.98183652875883e-05,
298
+ "loss": 6.3291,
299
+ "step": 20000
300
+ },
301
+ {
302
+ "epoch": 8.39630562552477,
303
+ "eval_accuracy": 0.15233311821319342,
304
+ "eval_loss": 6.701448440551758,
305
+ "eval_runtime": 1.0499,
306
+ "eval_samples_per_second": 456.231,
307
+ "eval_steps_per_second": 4.762,
308
+ "step": 20000
309
+ },
310
+ {
311
+ "epoch": 8.606213266162888,
312
+ "grad_norm": 2.7811079025268555,
313
+ "learning_rate": 7.880928355196771e-05,
314
+ "loss": 6.3322,
315
+ "step": 20500
316
+ },
317
+ {
318
+ "epoch": 8.816120906801007,
319
+ "grad_norm": 2.257051706314087,
320
+ "learning_rate": 7.780020181634713e-05,
321
+ "loss": 6.3292,
322
+ "step": 21000
323
+ },
324
+ {
325
+ "epoch": 9.026028547439127,
326
+ "grad_norm": 2.737684726715088,
327
+ "learning_rate": 7.679112008072655e-05,
328
+ "loss": 6.3188,
329
+ "step": 21500
330
+ },
331
+ {
332
+ "epoch": 9.235936188077247,
333
+ "grad_norm": 3.8119022846221924,
334
+ "learning_rate": 7.578203834510595e-05,
335
+ "loss": 6.2548,
336
+ "step": 22000
337
+ },
338
+ {
339
+ "epoch": 9.445843828715365,
340
+ "grad_norm": 2.7535812854766846,
341
+ "learning_rate": 7.477295660948537e-05,
342
+ "loss": 6.2632,
343
+ "step": 22500
344
+ },
345
+ {
346
+ "epoch": 9.655751469353484,
347
+ "grad_norm": 5.819505214691162,
348
+ "learning_rate": 7.376387487386479e-05,
349
+ "loss": 6.2604,
350
+ "step": 23000
351
+ },
352
+ {
353
+ "epoch": 9.865659109991604,
354
+ "grad_norm": 3.6783864498138428,
355
+ "learning_rate": 7.27547931382442e-05,
356
+ "loss": 6.2687,
357
+ "step": 23500
358
+ },
359
+ {
360
+ "epoch": 10.075566750629722,
361
+ "grad_norm": 5.783772945404053,
362
+ "learning_rate": 7.174571140262362e-05,
363
+ "loss": 6.231,
364
+ "step": 24000
365
+ },
366
+ {
367
+ "epoch": 10.285474391267842,
368
+ "grad_norm": 2.723422050476074,
369
+ "learning_rate": 7.073662966700304e-05,
370
+ "loss": 6.2044,
371
+ "step": 24500
372
+ },
373
+ {
374
+ "epoch": 10.495382031905962,
375
+ "grad_norm": 3.333406448364258,
376
+ "learning_rate": 6.972754793138244e-05,
377
+ "loss": 6.205,
378
+ "step": 25000
379
+ },
380
+ {
381
+ "epoch": 10.705289672544081,
382
+ "grad_norm": 3.7053959369659424,
383
+ "learning_rate": 6.871846619576186e-05,
384
+ "loss": 6.2037,
385
+ "step": 25500
386
+ },
387
+ {
388
+ "epoch": 10.9151973131822,
389
+ "grad_norm": 3.2251012325286865,
390
+ "learning_rate": 6.770938446014128e-05,
391
+ "loss": 6.2136,
392
+ "step": 26000
393
+ },
394
+ {
395
+ "epoch": 11.125104953820319,
396
+ "grad_norm": 2.389143466949463,
397
+ "learning_rate": 6.670030272452068e-05,
398
+ "loss": 6.1707,
399
+ "step": 26500
400
+ },
401
+ {
402
+ "epoch": 11.335012594458439,
403
+ "grad_norm": 3.476702928543091,
404
+ "learning_rate": 6.56912209889001e-05,
405
+ "loss": 6.1547,
406
+ "step": 27000
407
+ },
408
+ {
409
+ "epoch": 11.544920235096557,
410
+ "grad_norm": 3.1944007873535156,
411
+ "learning_rate": 6.468213925327952e-05,
412
+ "loss": 6.1514,
413
+ "step": 27500
414
+ },
415
+ {
416
+ "epoch": 11.754827875734676,
417
+ "grad_norm": 3.1503257751464844,
418
+ "learning_rate": 6.367305751765893e-05,
419
+ "loss": 6.155,
420
+ "step": 28000
421
+ },
422
+ {
423
+ "epoch": 11.964735516372796,
424
+ "grad_norm": 2.4318435192108154,
425
+ "learning_rate": 6.266397578203835e-05,
426
+ "loss": 6.1554,
427
+ "step": 28500
428
+ },
429
+ {
430
+ "epoch": 12.174643157010916,
431
+ "grad_norm": 2.668941020965576,
432
+ "learning_rate": 6.165489404641777e-05,
433
+ "loss": 6.1185,
434
+ "step": 29000
435
+ },
436
+ {
437
+ "epoch": 12.384550797649034,
438
+ "grad_norm": 2.188319206237793,
439
+ "learning_rate": 6.064581231079718e-05,
440
+ "loss": 6.1029,
441
+ "step": 29500
442
+ },
443
+ {
444
+ "epoch": 12.594458438287154,
445
+ "grad_norm": 2.9347007274627686,
446
+ "learning_rate": 5.963673057517659e-05,
447
+ "loss": 6.1153,
448
+ "step": 30000
449
+ },
450
+ {
451
+ "epoch": 12.594458438287154,
452
+ "eval_accuracy": 0.1539580826594286,
453
+ "eval_loss": 6.706422805786133,
454
+ "eval_runtime": 1.033,
455
+ "eval_samples_per_second": 463.676,
456
+ "eval_steps_per_second": 4.84,
457
+ "step": 30000
458
+ },
459
+ {
460
+ "epoch": 12.804366078925273,
461
+ "grad_norm": 3.4366564750671387,
462
+ "learning_rate": 5.8627648839556e-05,
463
+ "loss": 6.1044,
464
+ "step": 30500
465
+ },
466
+ {
467
+ "epoch": 13.014273719563391,
468
+ "grad_norm": 3.167576313018799,
469
+ "learning_rate": 5.7618567103935416e-05,
470
+ "loss": 6.1106,
471
+ "step": 31000
472
+ },
473
+ {
474
+ "epoch": 13.224181360201511,
475
+ "grad_norm": 2.0107972621917725,
476
+ "learning_rate": 5.660948536831484e-05,
477
+ "loss": 6.0733,
478
+ "step": 31500
479
+ },
480
+ {
481
+ "epoch": 13.43408900083963,
482
+ "grad_norm": 4.030771255493164,
483
+ "learning_rate": 5.5600403632694254e-05,
484
+ "loss": 6.072,
485
+ "step": 32000
486
+ },
487
+ {
488
+ "epoch": 13.64399664147775,
489
+ "grad_norm": 3.8046839237213135,
490
+ "learning_rate": 5.4591321897073666e-05,
491
+ "loss": 6.0786,
492
+ "step": 32500
493
+ },
494
+ {
495
+ "epoch": 13.853904282115868,
496
+ "grad_norm": 2.889934778213501,
497
+ "learning_rate": 5.358224016145308e-05,
498
+ "loss": 6.0757,
499
+ "step": 33000
500
+ },
501
+ {
502
+ "epoch": 14.063811922753988,
503
+ "grad_norm": 4.883760929107666,
504
+ "learning_rate": 5.257315842583249e-05,
505
+ "loss": 6.0633,
506
+ "step": 33500
507
+ },
508
+ {
509
+ "epoch": 14.273719563392108,
510
+ "grad_norm": 0.8915508389472961,
511
+ "learning_rate": 5.15640766902119e-05,
512
+ "loss": 6.0436,
513
+ "step": 34000
514
+ },
515
+ {
516
+ "epoch": 14.483627204030226,
517
+ "grad_norm": 2.643712043762207,
518
+ "learning_rate": 5.055499495459133e-05,
519
+ "loss": 6.0413,
520
+ "step": 34500
521
+ },
522
+ {
523
+ "epoch": 14.693534844668346,
524
+ "grad_norm": 2.5184438228607178,
525
+ "learning_rate": 4.954591321897074e-05,
526
+ "loss": 6.0432,
527
+ "step": 35000
528
+ },
529
+ {
530
+ "epoch": 14.903442485306465,
531
+ "grad_norm": 2.847947835922241,
532
+ "learning_rate": 4.853683148335015e-05,
533
+ "loss": 6.0409,
534
+ "step": 35500
535
+ },
536
+ {
537
+ "epoch": 15.113350125944585,
538
+ "grad_norm": 1.147056221961975,
539
+ "learning_rate": 4.7527749747729566e-05,
540
+ "loss": 6.0247,
541
+ "step": 36000
542
+ },
543
+ {
544
+ "epoch": 15.323257766582703,
545
+ "grad_norm": 2.816908597946167,
546
+ "learning_rate": 4.6518668012108985e-05,
547
+ "loss": 6.0205,
548
+ "step": 36500
549
+ },
550
+ {
551
+ "epoch": 15.533165407220823,
552
+ "grad_norm": 3.443824052810669,
553
+ "learning_rate": 4.55095862764884e-05,
554
+ "loss": 6.0149,
555
+ "step": 37000
556
+ },
557
+ {
558
+ "epoch": 15.743073047858942,
559
+ "grad_norm": 2.6054351329803467,
560
+ "learning_rate": 4.450050454086781e-05,
561
+ "loss": 6.0135,
562
+ "step": 37500
563
+ },
564
+ {
565
+ "epoch": 15.95298068849706,
566
+ "grad_norm": 3.3810739517211914,
567
+ "learning_rate": 4.349142280524723e-05,
568
+ "loss": 6.0131,
569
+ "step": 38000
570
+ },
571
+ {
572
+ "epoch": 16.162888329135182,
573
+ "grad_norm": 2.275756597518921,
574
+ "learning_rate": 4.248234106962664e-05,
575
+ "loss": 5.9965,
576
+ "step": 38500
577
+ },
578
+ {
579
+ "epoch": 16.3727959697733,
580
+ "grad_norm": 2.3946337699890137,
581
+ "learning_rate": 4.147325933400605e-05,
582
+ "loss": 5.9914,
583
+ "step": 39000
584
+ },
585
+ {
586
+ "epoch": 16.582703610411418,
587
+ "grad_norm": 4.954773426055908,
588
+ "learning_rate": 4.046417759838547e-05,
589
+ "loss": 5.9909,
590
+ "step": 39500
591
+ },
592
+ {
593
+ "epoch": 16.79261125104954,
594
+ "grad_norm": 4.085302829742432,
595
+ "learning_rate": 3.9455095862764884e-05,
596
+ "loss": 5.9862,
597
+ "step": 40000
598
+ },
599
+ {
600
+ "epoch": 16.79261125104954,
601
+ "eval_accuracy": 0.1519849725243916,
602
+ "eval_loss": 7.014549255371094,
603
+ "eval_runtime": 1.0149,
604
+ "eval_samples_per_second": 471.967,
605
+ "eval_steps_per_second": 4.927,
606
+ "step": 40000
607
+ },
608
+ {
609
+ "epoch": 17.002518891687657,
610
+ "grad_norm": 3.3821780681610107,
611
+ "learning_rate": 3.84460141271443e-05,
612
+ "loss": 5.9872,
613
+ "step": 40500
614
+ },
615
+ {
616
+ "epoch": 17.212426532325775,
617
+ "grad_norm": 1.8607794046401978,
618
+ "learning_rate": 3.7436932391523715e-05,
619
+ "loss": 5.9711,
620
+ "step": 41000
621
+ },
622
+ {
623
+ "epoch": 17.422334172963897,
624
+ "grad_norm": 2.3415207862854004,
625
+ "learning_rate": 3.642785065590313e-05,
626
+ "loss": 5.9652,
627
+ "step": 41500
628
+ },
629
+ {
630
+ "epoch": 17.632241813602015,
631
+ "grad_norm": 2.3748843669891357,
632
+ "learning_rate": 3.541876892028255e-05,
633
+ "loss": 5.97,
634
+ "step": 42000
635
+ },
636
+ {
637
+ "epoch": 17.842149454240133,
638
+ "grad_norm": 1.7207521200180054,
639
+ "learning_rate": 3.440968718466196e-05,
640
+ "loss": 5.9688,
641
+ "step": 42500
642
+ },
643
+ {
644
+ "epoch": 18.052057094878254,
645
+ "grad_norm": 3.5838589668273926,
646
+ "learning_rate": 3.340060544904137e-05,
647
+ "loss": 5.9576,
648
+ "step": 43000
649
+ },
650
+ {
651
+ "epoch": 18.261964735516372,
652
+ "grad_norm": 1.1076291799545288,
653
+ "learning_rate": 3.239152371342079e-05,
654
+ "loss": 5.9478,
655
+ "step": 43500
656
+ },
657
+ {
658
+ "epoch": 18.471872376154494,
659
+ "grad_norm": 2.307013750076294,
660
+ "learning_rate": 3.13824419778002e-05,
661
+ "loss": 5.9504,
662
+ "step": 44000
663
+ },
664
+ {
665
+ "epoch": 18.68178001679261,
666
+ "grad_norm": 1.877256989479065,
667
+ "learning_rate": 3.0373360242179615e-05,
668
+ "loss": 5.95,
669
+ "step": 44500
670
+ },
671
+ {
672
+ "epoch": 18.89168765743073,
673
+ "grad_norm": 2.667221784591675,
674
+ "learning_rate": 2.9364278506559034e-05,
675
+ "loss": 5.9522,
676
+ "step": 45000
677
+ },
678
+ {
679
+ "epoch": 19.10159529806885,
680
+ "grad_norm": 1.781567096710205,
681
+ "learning_rate": 2.8355196770938446e-05,
682
+ "loss": 5.9428,
683
+ "step": 45500
684
+ },
685
+ {
686
+ "epoch": 19.31150293870697,
687
+ "grad_norm": 2.600219488143921,
688
+ "learning_rate": 2.734611503531786e-05,
689
+ "loss": 5.9273,
690
+ "step": 46000
691
+ },
692
+ {
693
+ "epoch": 19.521410579345087,
694
+ "grad_norm": 3.2653560638427734,
695
+ "learning_rate": 2.6337033299697278e-05,
696
+ "loss": 5.9327,
697
+ "step": 46500
698
+ },
699
+ {
700
+ "epoch": 19.73131821998321,
701
+ "grad_norm": 0.8226433992385864,
702
+ "learning_rate": 2.532795156407669e-05,
703
+ "loss": 5.9303,
704
+ "step": 47000
705
+ },
706
+ {
707
+ "epoch": 19.941225860621326,
708
+ "grad_norm": 2.6203231811523438,
709
+ "learning_rate": 2.4318869828456106e-05,
710
+ "loss": 5.9259,
711
+ "step": 47500
712
+ },
713
+ {
714
+ "epoch": 20.151133501259444,
715
+ "grad_norm": 3.7636687755584717,
716
+ "learning_rate": 2.330978809283552e-05,
717
+ "loss": 5.9187,
718
+ "step": 48000
719
+ },
720
+ {
721
+ "epoch": 20.361041141897566,
722
+ "grad_norm": 2.3655290603637695,
723
+ "learning_rate": 2.2300706357214934e-05,
724
+ "loss": 5.9176,
725
+ "step": 48500
726
+ },
727
+ {
728
+ "epoch": 20.570948782535684,
729
+ "grad_norm": 1.55186927318573,
730
+ "learning_rate": 2.129162462159435e-05,
731
+ "loss": 5.9103,
732
+ "step": 49000
733
+ },
734
+ {
735
+ "epoch": 20.7808564231738,
736
+ "grad_norm": 4.357689380645752,
737
+ "learning_rate": 2.0282542885973765e-05,
738
+ "loss": 5.9109,
739
+ "step": 49500
740
+ },
741
+ {
742
+ "epoch": 20.990764063811923,
743
+ "grad_norm": 3.323455333709717,
744
+ "learning_rate": 1.927346115035318e-05,
745
+ "loss": 5.9088,
746
+ "step": 50000
747
+ },
748
+ {
749
+ "epoch": 20.990764063811923,
750
+ "eval_accuracy": 0.15313875571256344,
751
+ "eval_loss": 7.005898952484131,
752
+ "eval_runtime": 1.0145,
753
+ "eval_samples_per_second": 472.149,
754
+ "eval_steps_per_second": 4.928,
755
+ "step": 50000
756
+ },
757
+ {
758
+ "epoch": 21.20067170445004,
759
+ "grad_norm": 0.8441404104232788,
760
+ "learning_rate": 1.8264379414732593e-05,
761
+ "loss": 5.9017,
762
+ "step": 50500
763
+ },
764
+ {
765
+ "epoch": 21.410579345088163,
766
+ "grad_norm": 2.293954372406006,
767
+ "learning_rate": 1.725529767911201e-05,
768
+ "loss": 5.9007,
769
+ "step": 51000
770
+ },
771
+ {
772
+ "epoch": 21.62048698572628,
773
+ "grad_norm": 2.9091224670410156,
774
+ "learning_rate": 1.6246215943491424e-05,
775
+ "loss": 5.9001,
776
+ "step": 51500
777
+ },
778
+ {
779
+ "epoch": 21.8303946263644,
780
+ "grad_norm": 3.189171075820923,
781
+ "learning_rate": 1.5237134207870837e-05,
782
+ "loss": 5.8969,
783
+ "step": 52000
784
+ },
785
+ {
786
+ "epoch": 22.04030226700252,
787
+ "grad_norm": 0.8301982283592224,
788
+ "learning_rate": 1.4228052472250252e-05,
789
+ "loss": 5.8956,
790
+ "step": 52500
791
+ },
792
+ {
793
+ "epoch": 22.250209907640638,
794
+ "grad_norm": 3.955568552017212,
795
+ "learning_rate": 1.3218970736629668e-05,
796
+ "loss": 5.8826,
797
+ "step": 53000
798
+ },
799
+ {
800
+ "epoch": 22.460117548278756,
801
+ "grad_norm": 1.694311261177063,
802
+ "learning_rate": 1.2209889001009082e-05,
803
+ "loss": 5.8871,
804
+ "step": 53500
805
+ },
806
+ {
807
+ "epoch": 22.670025188916878,
808
+ "grad_norm": 1.507860541343689,
809
+ "learning_rate": 1.1200807265388496e-05,
810
+ "loss": 5.89,
811
+ "step": 54000
812
+ },
813
+ {
814
+ "epoch": 22.879932829554996,
815
+ "grad_norm": 0.8334353566169739,
816
+ "learning_rate": 1.0191725529767912e-05,
817
+ "loss": 5.8817,
818
+ "step": 54500
819
+ },
820
+ {
821
+ "epoch": 23.089840470193113,
822
+ "grad_norm": 0.8316900134086609,
823
+ "learning_rate": 9.182643794147326e-06,
824
+ "loss": 5.8804,
825
+ "step": 55000
826
+ },
827
+ {
828
+ "epoch": 23.299748110831235,
829
+ "grad_norm": 2.653756618499756,
830
+ "learning_rate": 8.173562058526741e-06,
831
+ "loss": 5.8749,
832
+ "step": 55500
833
+ },
834
+ {
835
+ "epoch": 23.509655751469353,
836
+ "grad_norm": 3.7355833053588867,
837
+ "learning_rate": 7.164480322906156e-06,
838
+ "loss": 5.8769,
839
+ "step": 56000
840
+ },
841
+ {
842
+ "epoch": 23.719563392107474,
843
+ "grad_norm": 0.8122338652610779,
844
+ "learning_rate": 6.155398587285571e-06,
845
+ "loss": 5.8699,
846
+ "step": 56500
847
+ },
848
+ {
849
+ "epoch": 23.929471032745592,
850
+ "grad_norm": 0.8836779594421387,
851
+ "learning_rate": 5.146316851664986e-06,
852
+ "loss": 5.8733,
853
+ "step": 57000
854
+ },
855
+ {
856
+ "epoch": 24.13937867338371,
857
+ "grad_norm": 3.144986152648926,
858
+ "learning_rate": 4.1372351160444e-06,
859
+ "loss": 5.8685,
860
+ "step": 57500
861
+ },
862
+ {
863
+ "epoch": 24.349286314021832,
864
+ "grad_norm": 3.1895604133605957,
865
+ "learning_rate": 3.1281533804238145e-06,
866
+ "loss": 5.8702,
867
+ "step": 58000
868
+ },
869
+ {
870
+ "epoch": 24.55919395465995,
871
+ "grad_norm": 0.9001981019973755,
872
+ "learning_rate": 2.1190716448032293e-06,
873
+ "loss": 5.8651,
874
+ "step": 58500
875
+ },
876
+ {
877
+ "epoch": 24.769101595298068,
878
+ "grad_norm": 1.6653518676757812,
879
+ "learning_rate": 1.109989909182644e-06,
880
+ "loss": 5.8625,
881
+ "step": 59000
882
+ },
883
+ {
884
+ "epoch": 24.97900923593619,
885
+ "grad_norm": 1.8964729309082031,
886
+ "learning_rate": 1.0090817356205853e-07,
887
+ "loss": 5.8636,
888
+ "step": 59500
889
+ },
890
+ {
891
+ "epoch": 25.0,
892
+ "step": 59550,
893
+ "total_flos": 1.5151126801503744e+18,
894
+ "train_loss": 6.378149575490495,
895
+ "train_runtime": 24807.4865,
896
+ "train_samples_per_second": 230.413,
897
+ "train_steps_per_second": 2.4
898
+ }
899
+ ],
900
+ "logging_steps": 500,
901
+ "max_steps": 59550,
902
+ "num_input_tokens_seen": 0,
903
+ "num_train_epochs": 25,
904
+ "save_steps": 10000,
905
+ "stateful_callbacks": {
906
+ "TrainerControl": {
907
+ "args": {
908
+ "should_epoch_stop": false,
909
+ "should_evaluate": false,
910
+ "should_log": false,
911
+ "should_save": true,
912
+ "should_training_stop": true
913
+ },
914
+ "attributes": {}
915
+ }
916
+ },
917
+ "total_flos": 1.5151126801503744e+18,
918
+ "train_batch_size": 96,
919
+ "trial_name": null,
920
+ "trial_params": null
921
+ }