chansung commited on
Commit
7acb8e9
1 Parent(s): e7a95c1

Model save

Browse files
README.md ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: gemma
3
+ library_name: peft
4
+ tags:
5
+ - trl
6
+ - sft
7
+ - generated_from_trainer
8
+ base_model: google/gemma-2b
9
+ datasets:
10
+ - generator
11
+ model-index:
12
+ - name: gemma2b-summarize-claude3sonnet-16k
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # gemma2b-summarize-claude3sonnet-16k
20
+
21
+ This model is a fine-tuned version of [google/gemma-2b](https://huggingface.co/google/gemma-2b) on the generator dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 2.5454
24
+
25
+ ## Model description
26
+
27
+ More information needed
28
+
29
+ ## Intended uses & limitations
30
+
31
+ More information needed
32
+
33
+ ## Training and evaluation data
34
+
35
+ More information needed
36
+
37
+ ## Training procedure
38
+
39
+ ### Training hyperparameters
40
+
41
+ The following hyperparameters were used during training:
42
+ - learning_rate: 0.0002
43
+ - train_batch_size: 8
44
+ - eval_batch_size: 8
45
+ - seed: 42
46
+ - distributed_type: multi-GPU
47
+ - num_devices: 3
48
+ - gradient_accumulation_steps: 2
49
+ - total_train_batch_size: 48
50
+ - total_eval_batch_size: 24
51
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
52
+ - lr_scheduler_type: cosine
53
+ - lr_scheduler_warmup_ratio: 0.1
54
+ - num_epochs: 10
55
+
56
+ ### Training results
57
+
58
+ | Training Loss | Epoch | Step | Validation Loss |
59
+ |:-------------:|:-----:|:----:|:---------------:|
60
+ | 1.4889 | 1.0 | 51 | 2.5875 |
61
+ | 1.1695 | 2.0 | 102 | 2.5044 |
62
+ | 1.1003 | 3.0 | 153 | 2.5000 |
63
+ | 1.0537 | 4.0 | 204 | 2.5040 |
64
+ | 1.0131 | 5.0 | 255 | 2.5131 |
65
+ | 0.995 | 6.0 | 306 | 2.5208 |
66
+ | 0.9739 | 7.0 | 357 | 2.5337 |
67
+ | 0.9626 | 8.0 | 408 | 2.5445 |
68
+ | 0.9572 | 9.0 | 459 | 2.5443 |
69
+ | 0.9554 | 10.0 | 510 | 2.5454 |
70
+
71
+
72
+ ### Framework versions
73
+
74
+ - PEFT 0.10.0
75
+ - Transformers 4.40.0
76
+ - Pytorch 2.2.2+cu121
77
+ - Datasets 2.19.1
78
+ - Tokenizers 0.19.1
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:34f67c4f1e59b159b445dff20db0f11867e2174a81a90d1fc2453270490834e9
3
  size 19644912
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6392527e0fe55c2c920d96a4c9d6113ebe5eb2e8f180bff1b1319e7422dc431
3
  size 19644912
all_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "total_flos": 2.9956040277203354e+17,
4
+ "train_loss": 1.144676014956306,
5
+ "train_runtime": 1752.8789,
6
+ "train_samples": 15873,
7
+ "train_samples_per_second": 13.857,
8
+ "train_steps_per_second": 0.291
9
+ }
runs/Jun05_14-01-48_user-HP-Z8-Fury-G5-Workstation-Desktop-PC/events.out.tfevents.1717563730.user-HP-Z8-Fury-G5-Workstation-Desktop-PC.934985.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ce4ee9ff13ca3c8ba64115a2698111a74e75c90c636d3a0ca3a2bf4e767309b3
3
- size 29112
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd60e47fb0e54d9025b630ec1ce604b523051b10f107bebe071e0c074bd39227
3
+ size 30159
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "total_flos": 2.9956040277203354e+17,
4
+ "train_loss": 1.144676014956306,
5
+ "train_runtime": 1752.8789,
6
+ "train_samples": 15873,
7
+ "train_samples_per_second": 13.857,
8
+ "train_steps_per_second": 0.291
9
+ }
trainer_state.json ADDED
@@ -0,0 +1,831 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 10.0,
5
+ "eval_steps": 500,
6
+ "global_step": 510,
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.0196078431372549,
13
+ "grad_norm": 2.359375,
14
+ "learning_rate": 3.92156862745098e-06,
15
+ "loss": 2.7922,
16
+ "step": 1
17
+ },
18
+ {
19
+ "epoch": 0.09803921568627451,
20
+ "grad_norm": 1.8984375,
21
+ "learning_rate": 1.9607843137254903e-05,
22
+ "loss": 2.8181,
23
+ "step": 5
24
+ },
25
+ {
26
+ "epoch": 0.19607843137254902,
27
+ "grad_norm": 1.5,
28
+ "learning_rate": 3.9215686274509805e-05,
29
+ "loss": 2.7665,
30
+ "step": 10
31
+ },
32
+ {
33
+ "epoch": 0.29411764705882354,
34
+ "grad_norm": 1.953125,
35
+ "learning_rate": 5.882352941176471e-05,
36
+ "loss": 2.6609,
37
+ "step": 15
38
+ },
39
+ {
40
+ "epoch": 0.39215686274509803,
41
+ "grad_norm": 1.6796875,
42
+ "learning_rate": 7.843137254901961e-05,
43
+ "loss": 2.4693,
44
+ "step": 20
45
+ },
46
+ {
47
+ "epoch": 0.49019607843137253,
48
+ "grad_norm": 1.8125,
49
+ "learning_rate": 9.80392156862745e-05,
50
+ "loss": 2.2357,
51
+ "step": 25
52
+ },
53
+ {
54
+ "epoch": 0.5882352941176471,
55
+ "grad_norm": 2.3125,
56
+ "learning_rate": 0.00011764705882352942,
57
+ "loss": 2.0763,
58
+ "step": 30
59
+ },
60
+ {
61
+ "epoch": 0.6862745098039216,
62
+ "grad_norm": 0.98046875,
63
+ "learning_rate": 0.0001372549019607843,
64
+ "loss": 1.8956,
65
+ "step": 35
66
+ },
67
+ {
68
+ "epoch": 0.7843137254901961,
69
+ "grad_norm": 0.8515625,
70
+ "learning_rate": 0.00015686274509803922,
71
+ "loss": 1.7407,
72
+ "step": 40
73
+ },
74
+ {
75
+ "epoch": 0.8823529411764706,
76
+ "grad_norm": 0.6328125,
77
+ "learning_rate": 0.00017647058823529413,
78
+ "loss": 1.5856,
79
+ "step": 45
80
+ },
81
+ {
82
+ "epoch": 0.9803921568627451,
83
+ "grad_norm": 1.6171875,
84
+ "learning_rate": 0.000196078431372549,
85
+ "loss": 1.4889,
86
+ "step": 50
87
+ },
88
+ {
89
+ "epoch": 1.0,
90
+ "eval_loss": 2.5875163078308105,
91
+ "eval_runtime": 0.5475,
92
+ "eval_samples_per_second": 18.266,
93
+ "eval_steps_per_second": 1.827,
94
+ "step": 51
95
+ },
96
+ {
97
+ "epoch": 1.0784313725490196,
98
+ "grad_norm": 0.52734375,
99
+ "learning_rate": 0.00019996252538028507,
100
+ "loss": 1.3908,
101
+ "step": 55
102
+ },
103
+ {
104
+ "epoch": 1.1764705882352942,
105
+ "grad_norm": 0.412109375,
106
+ "learning_rate": 0.00019981033287370443,
107
+ "loss": 1.336,
108
+ "step": 60
109
+ },
110
+ {
111
+ "epoch": 1.2745098039215685,
112
+ "grad_norm": 0.5078125,
113
+ "learning_rate": 0.00019954125840299165,
114
+ "loss": 1.3043,
115
+ "step": 65
116
+ },
117
+ {
118
+ "epoch": 1.3725490196078431,
119
+ "grad_norm": 0.69140625,
120
+ "learning_rate": 0.00019915561706530883,
121
+ "loss": 1.2637,
122
+ "step": 70
123
+ },
124
+ {
125
+ "epoch": 1.4705882352941178,
126
+ "grad_norm": 0.3203125,
127
+ "learning_rate": 0.00019865386046236596,
128
+ "loss": 1.2476,
129
+ "step": 75
130
+ },
131
+ {
132
+ "epoch": 1.5686274509803921,
133
+ "grad_norm": 0.56640625,
134
+ "learning_rate": 0.0001980365761715769,
135
+ "loss": 1.2254,
136
+ "step": 80
137
+ },
138
+ {
139
+ "epoch": 1.6666666666666665,
140
+ "grad_norm": 0.26953125,
141
+ "learning_rate": 0.00019730448705798239,
142
+ "loss": 1.2052,
143
+ "step": 85
144
+ },
145
+ {
146
+ "epoch": 1.7647058823529411,
147
+ "grad_norm": 0.345703125,
148
+ "learning_rate": 0.00019645845042774553,
149
+ "loss": 1.1999,
150
+ "step": 90
151
+ },
152
+ {
153
+ "epoch": 1.8627450980392157,
154
+ "grad_norm": 0.291015625,
155
+ "learning_rate": 0.00019549945702421144,
156
+ "loss": 1.1781,
157
+ "step": 95
158
+ },
159
+ {
160
+ "epoch": 1.9607843137254903,
161
+ "grad_norm": 0.427734375,
162
+ "learning_rate": 0.00019442862986770646,
163
+ "loss": 1.1695,
164
+ "step": 100
165
+ },
166
+ {
167
+ "epoch": 2.0,
168
+ "eval_loss": 2.5043954849243164,
169
+ "eval_runtime": 0.539,
170
+ "eval_samples_per_second": 18.554,
171
+ "eval_steps_per_second": 1.855,
172
+ "step": 102
173
+ },
174
+ {
175
+ "epoch": 2.0588235294117645,
176
+ "grad_norm": 0.52734375,
177
+ "learning_rate": 0.00019324722294043558,
178
+ "loss": 1.1426,
179
+ "step": 105
180
+ },
181
+ {
182
+ "epoch": 2.156862745098039,
183
+ "grad_norm": 0.37109375,
184
+ "learning_rate": 0.00019195661971801827,
185
+ "loss": 1.1312,
186
+ "step": 110
187
+ },
188
+ {
189
+ "epoch": 2.2549019607843137,
190
+ "grad_norm": 0.52734375,
191
+ "learning_rate": 0.00019055833154938207,
192
+ "loss": 1.1323,
193
+ "step": 115
194
+ },
195
+ {
196
+ "epoch": 2.3529411764705883,
197
+ "grad_norm": 0.5078125,
198
+ "learning_rate": 0.00018905399588691163,
199
+ "loss": 1.1302,
200
+ "step": 120
201
+ },
202
+ {
203
+ "epoch": 2.450980392156863,
204
+ "grad_norm": 0.390625,
205
+ "learning_rate": 0.00018744537436892516,
206
+ "loss": 1.1181,
207
+ "step": 125
208
+ },
209
+ {
210
+ "epoch": 2.549019607843137,
211
+ "grad_norm": 0.365234375,
212
+ "learning_rate": 0.00018573435075672424,
213
+ "loss": 1.1019,
214
+ "step": 130
215
+ },
216
+ {
217
+ "epoch": 2.6470588235294117,
218
+ "grad_norm": 0.375,
219
+ "learning_rate": 0.00018392292872863267,
220
+ "loss": 1.119,
221
+ "step": 135
222
+ },
223
+ {
224
+ "epoch": 2.7450980392156863,
225
+ "grad_norm": 0.458984375,
226
+ "learning_rate": 0.00018201322953360758,
227
+ "loss": 1.095,
228
+ "step": 140
229
+ },
230
+ {
231
+ "epoch": 2.843137254901961,
232
+ "grad_norm": 0.421875,
233
+ "learning_rate": 0.00018000748950717038,
234
+ "loss": 1.0959,
235
+ "step": 145
236
+ },
237
+ {
238
+ "epoch": 2.9411764705882355,
239
+ "grad_norm": 0.45703125,
240
+ "learning_rate": 0.00017790805745256704,
241
+ "loss": 1.1003,
242
+ "step": 150
243
+ },
244
+ {
245
+ "epoch": 3.0,
246
+ "eval_loss": 2.5000319480895996,
247
+ "eval_runtime": 0.5409,
248
+ "eval_samples_per_second": 18.489,
249
+ "eval_steps_per_second": 1.849,
250
+ "step": 153
251
+ },
252
+ {
253
+ "epoch": 3.0392156862745097,
254
+ "grad_norm": 0.609375,
255
+ "learning_rate": 0.00017571739189022365,
256
+ "loss": 1.0867,
257
+ "step": 155
258
+ },
259
+ {
260
+ "epoch": 3.1372549019607843,
261
+ "grad_norm": 0.6171875,
262
+ "learning_rate": 0.00017343805817871886,
263
+ "loss": 1.0606,
264
+ "step": 160
265
+ },
266
+ {
267
+ "epoch": 3.235294117647059,
268
+ "grad_norm": 0.58984375,
269
+ "learning_rate": 0.00017107272551064473,
270
+ "loss": 1.0521,
271
+ "step": 165
272
+ },
273
+ {
274
+ "epoch": 3.3333333333333335,
275
+ "grad_norm": 0.51171875,
276
+ "learning_rate": 0.0001686241637868734,
277
+ "loss": 1.0691,
278
+ "step": 170
279
+ },
280
+ {
281
+ "epoch": 3.431372549019608,
282
+ "grad_norm": 0.65625,
283
+ "learning_rate": 0.00016609524037289019,
284
+ "loss": 1.0573,
285
+ "step": 175
286
+ },
287
+ {
288
+ "epoch": 3.5294117647058822,
289
+ "grad_norm": 0.470703125,
290
+ "learning_rate": 0.0001634889167409923,
291
+ "loss": 1.063,
292
+ "step": 180
293
+ },
294
+ {
295
+ "epoch": 3.627450980392157,
296
+ "grad_norm": 0.44921875,
297
+ "learning_rate": 0.00016080824500228367,
298
+ "loss": 1.0503,
299
+ "step": 185
300
+ },
301
+ {
302
+ "epoch": 3.7254901960784315,
303
+ "grad_norm": 0.5234375,
304
+ "learning_rate": 0.00015805636433252891,
305
+ "loss": 1.051,
306
+ "step": 190
307
+ },
308
+ {
309
+ "epoch": 3.8235294117647056,
310
+ "grad_norm": 0.431640625,
311
+ "learning_rate": 0.0001552364972960506,
312
+ "loss": 1.0651,
313
+ "step": 195
314
+ },
315
+ {
316
+ "epoch": 3.9215686274509802,
317
+ "grad_norm": 0.5,
318
+ "learning_rate": 0.00015235194607197508,
319
+ "loss": 1.0537,
320
+ "step": 200
321
+ },
322
+ {
323
+ "epoch": 4.0,
324
+ "eval_loss": 2.503964900970459,
325
+ "eval_runtime": 0.5385,
326
+ "eval_samples_per_second": 18.568,
327
+ "eval_steps_per_second": 1.857,
328
+ "step": 204
329
+ },
330
+ {
331
+ "epoch": 4.019607843137255,
332
+ "grad_norm": 0.41796875,
333
+ "learning_rate": 0.0001494060885872464,
334
+ "loss": 1.0477,
335
+ "step": 205
336
+ },
337
+ {
338
+ "epoch": 4.117647058823529,
339
+ "grad_norm": 0.486328125,
340
+ "learning_rate": 0.00014640237456093634,
341
+ "loss": 1.0205,
342
+ "step": 210
343
+ },
344
+ {
345
+ "epoch": 4.215686274509804,
346
+ "grad_norm": 0.43359375,
347
+ "learning_rate": 0.00014334432146448272,
348
+ "loss": 1.0166,
349
+ "step": 215
350
+ },
351
+ {
352
+ "epoch": 4.313725490196078,
353
+ "grad_norm": 0.4609375,
354
+ "learning_rate": 0.00014023551040258725,
355
+ "loss": 1.0248,
356
+ "step": 220
357
+ },
358
+ {
359
+ "epoch": 4.411764705882353,
360
+ "grad_norm": 0.56640625,
361
+ "learning_rate": 0.00013707958191959608,
362
+ "loss": 1.0165,
363
+ "step": 225
364
+ },
365
+ {
366
+ "epoch": 4.509803921568627,
367
+ "grad_norm": 0.59765625,
368
+ "learning_rate": 0.00013388023173627414,
369
+ "loss": 1.0233,
370
+ "step": 230
371
+ },
372
+ {
373
+ "epoch": 4.607843137254902,
374
+ "grad_norm": 0.5625,
375
+ "learning_rate": 0.00013064120642196548,
376
+ "loss": 1.0261,
377
+ "step": 235
378
+ },
379
+ {
380
+ "epoch": 4.705882352941177,
381
+ "grad_norm": 0.53515625,
382
+ "learning_rate": 0.0001273662990072083,
383
+ "loss": 1.0329,
384
+ "step": 240
385
+ },
386
+ {
387
+ "epoch": 4.803921568627451,
388
+ "grad_norm": 0.484375,
389
+ "learning_rate": 0.00012405934454194146,
390
+ "loss": 1.0197,
391
+ "step": 245
392
+ },
393
+ {
394
+ "epoch": 4.901960784313726,
395
+ "grad_norm": 0.44921875,
396
+ "learning_rate": 0.00012072421560450497,
397
+ "loss": 1.0312,
398
+ "step": 250
399
+ },
400
+ {
401
+ "epoch": 5.0,
402
+ "grad_norm": 0.53515625,
403
+ "learning_rate": 0.00011736481776669306,
404
+ "loss": 1.0131,
405
+ "step": 255
406
+ },
407
+ {
408
+ "epoch": 5.0,
409
+ "eval_loss": 2.513137102127075,
410
+ "eval_runtime": 0.537,
411
+ "eval_samples_per_second": 18.622,
412
+ "eval_steps_per_second": 1.862,
413
+ "step": 255
414
+ },
415
+ {
416
+ "epoch": 5.098039215686274,
417
+ "grad_norm": 0.474609375,
418
+ "learning_rate": 0.00011398508502017047,
419
+ "loss": 0.9953,
420
+ "step": 260
421
+ },
422
+ {
423
+ "epoch": 5.196078431372549,
424
+ "grad_norm": 0.482421875,
425
+ "learning_rate": 0.00011058897516960816,
426
+ "loss": 1.0078,
427
+ "step": 265
428
+ },
429
+ {
430
+ "epoch": 5.294117647058823,
431
+ "grad_norm": 0.51171875,
432
+ "learning_rate": 0.00010718046519793276,
433
+ "loss": 1.0003,
434
+ "step": 270
435
+ },
436
+ {
437
+ "epoch": 5.392156862745098,
438
+ "grad_norm": 0.474609375,
439
+ "learning_rate": 0.00010376354660911771,
440
+ "loss": 1.0081,
441
+ "step": 275
442
+ },
443
+ {
444
+ "epoch": 5.490196078431373,
445
+ "grad_norm": 0.4765625,
446
+ "learning_rate": 0.00010034222075396953,
447
+ "loss": 0.9872,
448
+ "step": 280
449
+ },
450
+ {
451
+ "epoch": 5.588235294117647,
452
+ "grad_norm": 0.466796875,
453
+ "learning_rate": 9.692049414438299e-05,
454
+ "loss": 0.988,
455
+ "step": 285
456
+ },
457
+ {
458
+ "epoch": 5.686274509803922,
459
+ "grad_norm": 0.451171875,
460
+ "learning_rate": 9.350237376155269e-05,
461
+ "loss": 0.9944,
462
+ "step": 290
463
+ },
464
+ {
465
+ "epoch": 5.784313725490196,
466
+ "grad_norm": 0.4921875,
467
+ "learning_rate": 9.00918623636349e-05,
468
+ "loss": 0.985,
469
+ "step": 295
470
+ },
471
+ {
472
+ "epoch": 5.882352941176471,
473
+ "grad_norm": 0.4921875,
474
+ "learning_rate": 8.669295379835467e-05,
475
+ "loss": 1.0003,
476
+ "step": 300
477
+ },
478
+ {
479
+ "epoch": 5.980392156862745,
480
+ "grad_norm": 0.46875,
481
+ "learning_rate": 8.330962832604747e-05,
482
+ "loss": 0.995,
483
+ "step": 305
484
+ },
485
+ {
486
+ "epoch": 6.0,
487
+ "eval_loss": 2.5207793712615967,
488
+ "eval_runtime": 0.5386,
489
+ "eval_samples_per_second": 18.566,
490
+ "eval_steps_per_second": 1.857,
491
+ "step": 306
492
+ },
493
+ {
494
+ "epoch": 6.078431372549019,
495
+ "grad_norm": 0.56640625,
496
+ "learning_rate": 7.994584795861249e-05,
497
+ "loss": 0.9828,
498
+ "step": 310
499
+ },
500
+ {
501
+ "epoch": 6.176470588235294,
502
+ "grad_norm": 0.48828125,
503
+ "learning_rate": 7.660555181983518e-05,
504
+ "loss": 0.9815,
505
+ "step": 315
506
+ },
507
+ {
508
+ "epoch": 6.2745098039215685,
509
+ "grad_norm": 0.515625,
510
+ "learning_rate": 7.329265153251285e-05,
511
+ "loss": 0.9688,
512
+ "step": 320
513
+ },
514
+ {
515
+ "epoch": 6.372549019607844,
516
+ "grad_norm": 0.48828125,
517
+ "learning_rate": 7.001102663778533e-05,
518
+ "loss": 0.9786,
519
+ "step": 325
520
+ },
521
+ {
522
+ "epoch": 6.470588235294118,
523
+ "grad_norm": 0.50390625,
524
+ "learning_rate": 6.676452005203406e-05,
525
+ "loss": 0.9787,
526
+ "step": 330
527
+ },
528
+ {
529
+ "epoch": 6.568627450980392,
530
+ "grad_norm": 0.5546875,
531
+ "learning_rate": 6.355693356667065e-05,
532
+ "loss": 0.9836,
533
+ "step": 335
534
+ },
535
+ {
536
+ "epoch": 6.666666666666667,
537
+ "grad_norm": 0.490234375,
538
+ "learning_rate": 6.039202339608432e-05,
539
+ "loss": 0.9728,
540
+ "step": 340
541
+ },
542
+ {
543
+ "epoch": 6.764705882352941,
544
+ "grad_norm": 0.486328125,
545
+ "learning_rate": 5.727349577896194e-05,
546
+ "loss": 0.9746,
547
+ "step": 345
548
+ },
549
+ {
550
+ "epoch": 6.862745098039216,
551
+ "grad_norm": 0.4921875,
552
+ "learning_rate": 5.4205002638131406e-05,
553
+ "loss": 0.9733,
554
+ "step": 350
555
+ },
556
+ {
557
+ "epoch": 6.96078431372549,
558
+ "grad_norm": 0.4453125,
559
+ "learning_rate": 5.119013730401152e-05,
560
+ "loss": 0.9739,
561
+ "step": 355
562
+ },
563
+ {
564
+ "epoch": 7.0,
565
+ "eval_loss": 2.5337109565734863,
566
+ "eval_runtime": 0.5399,
567
+ "eval_samples_per_second": 18.521,
568
+ "eval_steps_per_second": 1.852,
569
+ "step": 357
570
+ },
571
+ {
572
+ "epoch": 7.0588235294117645,
573
+ "grad_norm": 0.451171875,
574
+ "learning_rate": 4.823243030667576e-05,
575
+ "loss": 0.9639,
576
+ "step": 360
577
+ },
578
+ {
579
+ "epoch": 7.1568627450980395,
580
+ "grad_norm": 0.484375,
581
+ "learning_rate": 4.533534524145756e-05,
582
+ "loss": 0.9543,
583
+ "step": 365
584
+ },
585
+ {
586
+ "epoch": 7.254901960784314,
587
+ "grad_norm": 0.4140625,
588
+ "learning_rate": 4.250227471293935e-05,
589
+ "loss": 0.9674,
590
+ "step": 370
591
+ },
592
+ {
593
+ "epoch": 7.352941176470588,
594
+ "grad_norm": 0.44921875,
595
+ "learning_rate": 3.973653636207437e-05,
596
+ "loss": 0.9617,
597
+ "step": 375
598
+ },
599
+ {
600
+ "epoch": 7.450980392156863,
601
+ "grad_norm": 0.439453125,
602
+ "learning_rate": 3.7041368981094024e-05,
603
+ "loss": 0.9616,
604
+ "step": 380
605
+ },
606
+ {
607
+ "epoch": 7.549019607843137,
608
+ "grad_norm": 0.431640625,
609
+ "learning_rate": 3.441992872075027e-05,
610
+ "loss": 0.9685,
611
+ "step": 385
612
+ },
613
+ {
614
+ "epoch": 7.647058823529412,
615
+ "grad_norm": 0.458984375,
616
+ "learning_rate": 3.187528539433458e-05,
617
+ "loss": 0.975,
618
+ "step": 390
619
+ },
620
+ {
621
+ "epoch": 7.745098039215686,
622
+ "grad_norm": 0.466796875,
623
+ "learning_rate": 2.9410418882801683e-05,
624
+ "loss": 0.9707,
625
+ "step": 395
626
+ },
627
+ {
628
+ "epoch": 7.8431372549019605,
629
+ "grad_norm": 0.443359375,
630
+ "learning_rate": 2.702821564520732e-05,
631
+ "loss": 0.9593,
632
+ "step": 400
633
+ },
634
+ {
635
+ "epoch": 7.9411764705882355,
636
+ "grad_norm": 0.408203125,
637
+ "learning_rate": 2.4731465338547556e-05,
638
+ "loss": 0.9626,
639
+ "step": 405
640
+ },
641
+ {
642
+ "epoch": 8.0,
643
+ "eval_loss": 2.5445075035095215,
644
+ "eval_runtime": 0.5398,
645
+ "eval_samples_per_second": 18.525,
646
+ "eval_steps_per_second": 1.852,
647
+ "step": 408
648
+ },
649
+ {
650
+ "epoch": 8.03921568627451,
651
+ "grad_norm": 0.40625,
652
+ "learning_rate": 2.252285755095652e-05,
653
+ "loss": 0.9591,
654
+ "step": 410
655
+ },
656
+ {
657
+ "epoch": 8.137254901960784,
658
+ "grad_norm": 0.40625,
659
+ "learning_rate": 2.0404978652089323e-05,
660
+ "loss": 0.9516,
661
+ "step": 415
662
+ },
663
+ {
664
+ "epoch": 8.235294117647058,
665
+ "grad_norm": 0.412109375,
666
+ "learning_rate": 1.8380308764377842e-05,
667
+ "loss": 0.951,
668
+ "step": 420
669
+ },
670
+ {
671
+ "epoch": 8.333333333333334,
672
+ "grad_norm": 0.447265625,
673
+ "learning_rate": 1.6451218858706374e-05,
674
+ "loss": 0.9559,
675
+ "step": 425
676
+ },
677
+ {
678
+ "epoch": 8.431372549019608,
679
+ "grad_norm": 0.427734375,
680
+ "learning_rate": 1.4619967977908156e-05,
681
+ "loss": 0.9495,
682
+ "step": 430
683
+ },
684
+ {
685
+ "epoch": 8.529411764705882,
686
+ "grad_norm": 0.404296875,
687
+ "learning_rate": 1.2888700591334223e-05,
688
+ "loss": 0.9549,
689
+ "step": 435
690
+ },
691
+ {
692
+ "epoch": 8.627450980392156,
693
+ "grad_norm": 0.41796875,
694
+ "learning_rate": 1.1259444083592585e-05,
695
+ "loss": 0.9636,
696
+ "step": 440
697
+ },
698
+ {
699
+ "epoch": 8.72549019607843,
700
+ "grad_norm": 0.427734375,
701
+ "learning_rate": 9.734106380398023e-06,
702
+ "loss": 0.9547,
703
+ "step": 445
704
+ },
705
+ {
706
+ "epoch": 8.823529411764707,
707
+ "grad_norm": 0.40625,
708
+ "learning_rate": 8.314473714313719e-06,
709
+ "loss": 0.9643,
710
+ "step": 450
711
+ },
712
+ {
713
+ "epoch": 8.92156862745098,
714
+ "grad_norm": 0.4296875,
715
+ "learning_rate": 7.002208532999932e-06,
716
+ "loss": 0.9572,
717
+ "step": 455
718
+ },
719
+ {
720
+ "epoch": 9.0,
721
+ "eval_loss": 2.544271469116211,
722
+ "eval_runtime": 0.5384,
723
+ "eval_samples_per_second": 18.575,
724
+ "eval_steps_per_second": 1.857,
725
+ "step": 459
726
+ },
727
+ {
728
+ "epoch": 9.019607843137255,
729
+ "grad_norm": 0.412109375,
730
+ "learning_rate": 5.798847552420183e-06,
731
+ "loss": 0.9625,
732
+ "step": 460
733
+ },
734
+ {
735
+ "epoch": 9.117647058823529,
736
+ "grad_norm": 0.421875,
737
+ "learning_rate": 4.705799957284351e-06,
738
+ "loss": 0.9482,
739
+ "step": 465
740
+ },
741
+ {
742
+ "epoch": 9.215686274509803,
743
+ "grad_norm": 0.408203125,
744
+ "learning_rate": 3.724345750835878e-06,
745
+ "loss": 0.943,
746
+ "step": 470
747
+ },
748
+ {
749
+ "epoch": 9.313725490196079,
750
+ "grad_norm": 0.41796875,
751
+ "learning_rate": 2.855634255915951e-06,
752
+ "loss": 0.9568,
753
+ "step": 475
754
+ },
755
+ {
756
+ "epoch": 9.411764705882353,
757
+ "grad_norm": 0.421875,
758
+ "learning_rate": 2.100682769059548e-06,
759
+ "loss": 0.9561,
760
+ "step": 480
761
+ },
762
+ {
763
+ "epoch": 9.509803921568627,
764
+ "grad_norm": 0.404296875,
765
+ "learning_rate": 1.4603753691998735e-06,
766
+ "loss": 0.9634,
767
+ "step": 485
768
+ },
769
+ {
770
+ "epoch": 9.607843137254902,
771
+ "grad_norm": 0.400390625,
772
+ "learning_rate": 9.354618823758655e-07,
773
+ "loss": 0.9588,
774
+ "step": 490
775
+ },
776
+ {
777
+ "epoch": 9.705882352941176,
778
+ "grad_norm": 0.408203125,
779
+ "learning_rate": 5.265570036553813e-07,
780
+ "loss": 0.9482,
781
+ "step": 495
782
+ },
783
+ {
784
+ "epoch": 9.803921568627452,
785
+ "grad_norm": 0.427734375,
786
+ "learning_rate": 2.3413957730226144e-07,
787
+ "loss": 0.9663,
788
+ "step": 500
789
+ },
790
+ {
791
+ "epoch": 9.901960784313726,
792
+ "grad_norm": 0.41796875,
793
+ "learning_rate": 5.8552036030179445e-08,
794
+ "loss": 0.9606,
795
+ "step": 505
796
+ },
797
+ {
798
+ "epoch": 10.0,
799
+ "grad_norm": 0.421875,
800
+ "learning_rate": 0.0,
801
+ "loss": 0.9554,
802
+ "step": 510
803
+ },
804
+ {
805
+ "epoch": 10.0,
806
+ "eval_loss": 2.545398235321045,
807
+ "eval_runtime": 0.5358,
808
+ "eval_samples_per_second": 18.663,
809
+ "eval_steps_per_second": 1.866,
810
+ "step": 510
811
+ },
812
+ {
813
+ "epoch": 10.0,
814
+ "step": 510,
815
+ "total_flos": 2.9956040277203354e+17,
816
+ "train_loss": 1.144676014956306,
817
+ "train_runtime": 1752.8789,
818
+ "train_samples_per_second": 13.857,
819
+ "train_steps_per_second": 0.291
820
+ }
821
+ ],
822
+ "logging_steps": 5,
823
+ "max_steps": 510,
824
+ "num_input_tokens_seen": 0,
825
+ "num_train_epochs": 10,
826
+ "save_steps": 100,
827
+ "total_flos": 2.9956040277203354e+17,
828
+ "train_batch_size": 8,
829
+ "trial_name": null,
830
+ "trial_params": null
831
+ }