yyx123 commited on
Commit
7f216d1
1 Parent(s): 687007e

Model save

Browse files
Files changed (5) hide show
  1. README.md +23 -12
  2. all_results.json +9 -9
  3. eval_results.json +5 -5
  4. train_results.json +5 -5
  5. trainer_state.json +530 -58
README.md CHANGED
@@ -2,13 +2,9 @@
2
  license: other
3
  library_name: peft
4
  tags:
5
- - alignment-handbook
6
- - generated_from_trainer
7
  - trl
8
  - sft
9
  - generated_from_trainer
10
- datasets:
11
- - ruozhiba
12
  base_model: 01-ai/Yi-6B
13
  model-index:
14
  - name: Yi-6B-ruozhiba
@@ -20,9 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  # Yi-6B-ruozhiba
22
 
23
- This model is a fine-tuned version of [01-ai/Yi-6B](https://huggingface.co/01-ai/Yi-6B) on the ruozhiba dataset.
24
  It achieves the following results on the evaluation set:
25
- - Loss: 1.9288
26
 
27
  ## Model description
28
 
@@ -48,17 +44,32 @@ The following hyperparameters were used during training:
48
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
49
  - lr_scheduler_type: cosine
50
  - lr_scheduler_warmup_ratio: 0.1
51
- - num_epochs: 5
52
 
53
  ### Training results
54
 
55
  | Training Loss | Epoch | Step | Validation Loss |
56
  |:-------------:|:-----:|:----:|:---------------:|
57
- | 2.2958 | 1.0 | 55 | 1.9636 |
58
- | 1.8837 | 2.0 | 110 | 1.9193 |
59
- | 1.7174 | 3.0 | 165 | 1.9123 |
60
- | 1.6515 | 4.0 | 220 | 1.9246 |
61
- | 1.6337 | 5.0 | 275 | 1.9288 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
62
 
63
 
64
  ### Framework versions
 
2
  license: other
3
  library_name: peft
4
  tags:
 
 
5
  - trl
6
  - sft
7
  - generated_from_trainer
 
 
8
  base_model: 01-ai/Yi-6B
9
  model-index:
10
  - name: Yi-6B-ruozhiba
 
16
 
17
  # Yi-6B-ruozhiba
18
 
19
+ This model is a fine-tuned version of [01-ai/Yi-6B](https://huggingface.co/01-ai/Yi-6B) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 4.2495
22
 
23
  ## Model description
24
 
 
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: cosine
46
  - lr_scheduler_warmup_ratio: 0.1
47
+ - num_epochs: 20
48
 
49
  ### Training results
50
 
51
  | Training Loss | Epoch | Step | Validation Loss |
52
  |:-------------:|:-----:|:----:|:---------------:|
53
+ | 2.644 | 1.0 | 55 | 2.2811 |
54
+ | 1.9483 | 2.0 | 110 | 1.9384 |
55
+ | 1.7791 | 3.0 | 165 | 1.9109 |
56
+ | 1.6053 | 4.0 | 220 | 1.9560 |
57
+ | 1.3206 | 5.0 | 275 | 2.1674 |
58
+ | 0.9722 | 6.0 | 330 | 2.4236 |
59
+ | 0.7026 | 7.0 | 385 | 2.6935 |
60
+ | 0.4776 | 8.0 | 440 | 3.0005 |
61
+ | 0.3163 | 9.0 | 495 | 3.2703 |
62
+ | 0.2355 | 10.0 | 550 | 3.4524 |
63
+ | 0.1489 | 11.0 | 605 | 3.6805 |
64
+ | 0.1179 | 12.0 | 660 | 3.7960 |
65
+ | 0.0712 | 13.0 | 715 | 3.9715 |
66
+ | 0.0782 | 14.0 | 770 | 4.0284 |
67
+ | 0.0691 | 15.0 | 825 | 4.1576 |
68
+ | 0.0486 | 16.0 | 880 | 4.1690 |
69
+ | 0.0639 | 17.0 | 935 | 4.2094 |
70
+ | 0.0589 | 18.0 | 990 | 4.2436 |
71
+ | 0.042 | 19.0 | 1045 | 4.2491 |
72
+ | 0.0405 | 20.0 | 1100 | 4.2495 |
73
 
74
 
75
  ### Framework versions
all_results.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
- "epoch": 5.0,
3
- "eval_loss": 1.9287786483764648,
4
- "eval_runtime": 1.2455,
5
  "eval_samples": 23,
6
- "eval_samples_per_second": 18.466,
7
- "eval_steps_per_second": 4.817,
8
- "train_loss": 1.8268643422560258,
9
- "train_runtime": 238.6826,
10
  "train_samples": 217,
11
- "train_samples_per_second": 4.546,
12
- "train_steps_per_second": 1.152
13
  }
 
1
  {
2
+ "epoch": 20.0,
3
+ "eval_loss": 4.249512195587158,
4
+ "eval_runtime": 4.9064,
5
  "eval_samples": 23,
6
+ "eval_samples_per_second": 4.688,
7
+ "eval_steps_per_second": 1.223,
8
+ "train_loss": 0.5375886155258526,
9
+ "train_runtime": 2988.6301,
10
  "train_samples": 217,
11
+ "train_samples_per_second": 1.452,
12
+ "train_steps_per_second": 0.368
13
  }
eval_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "epoch": 5.0,
3
- "eval_loss": 1.9287786483764648,
4
- "eval_runtime": 1.2455,
5
  "eval_samples": 23,
6
- "eval_samples_per_second": 18.466,
7
- "eval_steps_per_second": 4.817
8
  }
 
1
  {
2
+ "epoch": 20.0,
3
+ "eval_loss": 4.249512195587158,
4
+ "eval_runtime": 4.9064,
5
  "eval_samples": 23,
6
+ "eval_samples_per_second": 4.688,
7
+ "eval_steps_per_second": 1.223
8
  }
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "epoch": 5.0,
3
- "train_loss": 1.8268643422560258,
4
- "train_runtime": 238.6826,
5
  "train_samples": 217,
6
- "train_samples_per_second": 4.546,
7
- "train_steps_per_second": 1.152
8
  }
 
1
  {
2
+ "epoch": 20.0,
3
+ "train_loss": 0.5375886155258526,
4
+ "train_runtime": 2988.6301,
5
  "train_samples": 217,
6
+ "train_samples_per_second": 1.452,
7
+ "train_steps_per_second": 0.368
8
  }
trainer_state.json CHANGED
@@ -1,153 +1,625 @@
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
- "epoch": 5.0,
5
  "eval_steps": 500,
6
- "global_step": 275,
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.02,
13
- "learning_rate": 1.7857142857142857e-06,
14
  "loss": 2.5611,
15
  "step": 1
16
  },
17
  {
18
  "epoch": 0.36,
19
- "learning_rate": 3.571428571428572e-05,
20
- "loss": 2.5326,
21
  "step": 20
22
  },
23
  {
24
  "epoch": 0.73,
25
- "learning_rate": 4.970937357800635e-05,
26
- "loss": 2.2958,
27
  "step": 40
28
  },
29
  {
30
  "epoch": 1.0,
31
- "eval_loss": 1.9636107683181763,
32
- "eval_runtime": 1.2207,
33
- "eval_samples_per_second": 18.842,
34
- "eval_steps_per_second": 4.915,
 
 
 
 
 
35
  "step": 55
36
  },
37
  {
38
  "epoch": 1.09,
39
- "learning_rate": 4.7957734321334915e-05,
40
- "loss": 1.9545,
41
  "step": 60
42
  },
43
  {
44
  "epoch": 1.45,
45
- "learning_rate": 4.4728512734909844e-05,
46
- "loss": 1.8763,
47
  "step": 80
48
  },
49
  {
50
  "epoch": 1.82,
51
- "learning_rate": 4.022954469173914e-05,
52
- "loss": 1.8837,
53
  "step": 100
54
  },
55
  {
56
  "epoch": 2.0,
57
- "eval_loss": 1.9193000793457031,
58
- "eval_runtime": 1.2166,
59
- "eval_samples_per_second": 18.905,
60
- "eval_steps_per_second": 4.932,
 
 
 
 
 
61
  "step": 110
62
  },
63
  {
64
  "epoch": 2.18,
65
- "learning_rate": 3.475038819391789e-05,
66
- "loss": 1.7375,
67
  "step": 120
68
  },
69
  {
70
  "epoch": 2.55,
71
- "learning_rate": 2.864368713521274e-05,
72
- "loss": 1.7612,
73
  "step": 140
74
  },
75
  {
76
  "epoch": 2.91,
77
- "learning_rate": 2.2302474793696117e-05,
78
- "loss": 1.7174,
79
  "step": 160
80
  },
81
  {
82
  "epoch": 3.0,
83
- "eval_loss": 1.9123048782348633,
84
- "eval_runtime": 1.2199,
85
- "eval_samples_per_second": 18.853,
86
- "eval_steps_per_second": 4.918,
 
 
 
 
 
87
  "step": 165
88
  },
89
  {
90
  "epoch": 3.27,
91
- "learning_rate": 1.613487782393661e-05,
92
- "loss": 1.6455,
93
  "step": 180
94
  },
95
  {
96
  "epoch": 3.64,
97
- "learning_rate": 1.0537848824653418e-05,
98
- "loss": 1.6657,
99
  "step": 200
100
  },
101
  {
102
  "epoch": 4.0,
103
- "learning_rate": 5.871618079580327e-06,
104
- "loss": 1.6515,
 
 
 
 
 
105
  "step": 220
106
  },
107
  {
108
  "epoch": 4.0,
109
- "eval_loss": 1.9245940446853638,
110
- "eval_runtime": 1.269,
111
- "eval_samples_per_second": 18.125,
112
- "eval_steps_per_second": 4.728,
113
  "step": 220
114
  },
115
  {
116
  "epoch": 4.36,
117
- "learning_rate": 2.4365087825904355e-06,
118
- "loss": 1.5988,
119
  "step": 240
120
  },
121
  {
122
  "epoch": 4.73,
123
- "learning_rate": 4.536079419643374e-07,
124
- "loss": 1.6337,
125
  "step": 260
126
  },
127
  {
128
  "epoch": 5.0,
129
- "eval_loss": 1.9287786483764648,
130
- "eval_runtime": 1.2179,
131
- "eval_samples_per_second": 18.886,
132
- "eval_steps_per_second": 4.927,
133
  "step": 275
134
  },
135
  {
136
  "epoch": 5.0,
137
- "step": 275,
138
- "total_flos": 9506378566434816.0,
139
- "train_loss": 1.8268643422560258,
140
- "train_runtime": 238.6826,
141
- "train_samples_per_second": 4.546,
142
- "train_steps_per_second": 1.152
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
143
  }
144
  ],
145
  "logging_steps": 20,
146
- "max_steps": 275,
147
  "num_input_tokens_seen": 0,
148
- "num_train_epochs": 5,
149
  "save_steps": 20,
150
- "total_flos": 9506378566434816.0,
151
  "train_batch_size": 4,
152
  "trial_name": null,
153
  "trial_params": null
 
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
+ "epoch": 20.0,
5
  "eval_steps": 500,
6
+ "global_step": 1100,
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.02,
13
+ "learning_rate": 4.545454545454545e-07,
14
  "loss": 2.5611,
15
  "step": 1
16
  },
17
  {
18
  "epoch": 0.36,
19
+ "learning_rate": 9.090909090909091e-06,
20
+ "loss": 2.5692,
21
  "step": 20
22
  },
23
  {
24
  "epoch": 0.73,
25
+ "learning_rate": 1.8181818181818182e-05,
26
+ "loss": 2.644,
27
  "step": 40
28
  },
29
  {
30
  "epoch": 1.0,
31
+ "gpt4_scores": 0.0,
32
+ "step": 55
33
+ },
34
+ {
35
+ "epoch": 1.0,
36
+ "eval_loss": 2.2810959815979004,
37
+ "eval_runtime": 4.9148,
38
+ "eval_samples_per_second": 4.68,
39
+ "eval_steps_per_second": 1.221,
40
  "step": 55
41
  },
42
  {
43
  "epoch": 1.09,
44
+ "learning_rate": 2.7272727272727273e-05,
45
+ "loss": 2.3589,
46
  "step": 60
47
  },
48
  {
49
  "epoch": 1.45,
50
+ "learning_rate": 3.6363636363636364e-05,
51
+ "loss": 2.0614,
52
  "step": 80
53
  },
54
  {
55
  "epoch": 1.82,
56
+ "learning_rate": 4.545454545454546e-05,
57
+ "loss": 1.9483,
58
  "step": 100
59
  },
60
  {
61
  "epoch": 2.0,
62
+ "gpt4_scores": 0.0,
63
+ "step": 110
64
+ },
65
+ {
66
+ "epoch": 2.0,
67
+ "eval_loss": 1.938421607017517,
68
+ "eval_runtime": 4.8876,
69
+ "eval_samples_per_second": 4.706,
70
+ "eval_steps_per_second": 1.228,
71
  "step": 110
72
  },
73
  {
74
  "epoch": 2.18,
75
+ "learning_rate": 4.9987413559579636e-05,
76
+ "loss": 1.7952,
77
  "step": 120
78
  },
79
  {
80
  "epoch": 2.55,
81
+ "learning_rate": 4.988679806432712e-05,
82
+ "loss": 1.8206,
83
  "step": 140
84
  },
85
  {
86
  "epoch": 2.91,
87
+ "learning_rate": 4.968597221690986e-05,
88
+ "loss": 1.7791,
89
  "step": 160
90
  },
91
  {
92
  "epoch": 3.0,
93
+ "gpt4_scores": 0.0,
94
+ "step": 165
95
+ },
96
+ {
97
+ "epoch": 3.0,
98
+ "eval_loss": 1.910915493965149,
99
+ "eval_runtime": 4.8939,
100
+ "eval_samples_per_second": 4.7,
101
+ "eval_steps_per_second": 1.226,
102
  "step": 165
103
  },
104
  {
105
  "epoch": 3.27,
106
+ "learning_rate": 4.938574467213518e-05,
107
+ "loss": 1.6577,
108
  "step": 180
109
  },
110
  {
111
  "epoch": 3.64,
112
+ "learning_rate": 4.898732434036244e-05,
113
+ "loss": 1.6283,
114
  "step": 200
115
  },
116
  {
117
  "epoch": 4.0,
118
+ "learning_rate": 4.849231551964771e-05,
119
+ "loss": 1.6053,
120
+ "step": 220
121
+ },
122
+ {
123
+ "epoch": 4.0,
124
+ "gpt4_scores": 0.0,
125
  "step": 220
126
  },
127
  {
128
  "epoch": 4.0,
129
+ "eval_loss": 1.9559897184371948,
130
+ "eval_runtime": 4.8861,
131
+ "eval_samples_per_second": 4.707,
132
+ "eval_steps_per_second": 1.228,
133
  "step": 220
134
  },
135
  {
136
  "epoch": 4.36,
137
+ "learning_rate": 4.790271143580174e-05,
138
+ "loss": 1.332,
139
  "step": 240
140
  },
141
  {
142
  "epoch": 4.73,
143
+ "learning_rate": 4.722088621637309e-05,
144
+ "loss": 1.3206,
145
  "step": 260
146
  },
147
  {
148
  "epoch": 5.0,
149
+ "gpt4_scores": 0.0,
 
 
 
150
  "step": 275
151
  },
152
  {
153
  "epoch": 5.0,
154
+ "eval_loss": 2.167389154434204,
155
+ "eval_runtime": 4.8826,
156
+ "eval_samples_per_second": 4.711,
157
+ "eval_steps_per_second": 1.229,
158
+ "step": 275
159
+ },
160
+ {
161
+ "epoch": 5.09,
162
+ "learning_rate": 4.644958533087443e-05,
163
+ "loss": 1.1859,
164
+ "step": 280
165
+ },
166
+ {
167
+ "epoch": 5.45,
168
+ "learning_rate": 4.559191453574582e-05,
169
+ "loss": 0.9584,
170
+ "step": 300
171
+ },
172
+ {
173
+ "epoch": 5.82,
174
+ "learning_rate": 4.465132736856969e-05,
175
+ "loss": 0.9722,
176
+ "step": 320
177
+ },
178
+ {
179
+ "epoch": 6.0,
180
+ "gpt4_scores": 0.0,
181
+ "step": 330
182
+ },
183
+ {
184
+ "epoch": 6.0,
185
+ "eval_loss": 2.423626661300659,
186
+ "eval_runtime": 4.8846,
187
+ "eval_samples_per_second": 4.709,
188
+ "eval_steps_per_second": 1.228,
189
+ "step": 330
190
+ },
191
+ {
192
+ "epoch": 6.18,
193
+ "learning_rate": 4.3631611241893874e-05,
194
+ "loss": 0.8037,
195
+ "step": 340
196
+ },
197
+ {
198
+ "epoch": 6.55,
199
+ "learning_rate": 4.2536872192658036e-05,
200
+ "loss": 0.6705,
201
+ "step": 360
202
+ },
203
+ {
204
+ "epoch": 6.91,
205
+ "learning_rate": 4.137151834863213e-05,
206
+ "loss": 0.7026,
207
+ "step": 380
208
+ },
209
+ {
210
+ "epoch": 7.0,
211
+ "gpt4_scores": 0.0,
212
+ "step": 385
213
+ },
214
+ {
215
+ "epoch": 7.0,
216
+ "eval_loss": 2.693521499633789,
217
+ "eval_runtime": 4.8871,
218
+ "eval_samples_per_second": 4.706,
219
+ "eval_steps_per_second": 1.228,
220
+ "step": 385
221
+ },
222
+ {
223
+ "epoch": 7.27,
224
+ "learning_rate": 4.014024217844167e-05,
225
+ "loss": 0.5485,
226
+ "step": 400
227
+ },
228
+ {
229
+ "epoch": 7.64,
230
+ "learning_rate": 3.884800159665276e-05,
231
+ "loss": 0.4682,
232
+ "step": 420
233
+ },
234
+ {
235
+ "epoch": 8.0,
236
+ "learning_rate": 3.7500000000000003e-05,
237
+ "loss": 0.4776,
238
+ "step": 440
239
+ },
240
+ {
241
+ "epoch": 8.0,
242
+ "gpt4_scores": 0.0,
243
+ "step": 440
244
+ },
245
+ {
246
+ "epoch": 8.0,
247
+ "eval_loss": 3.0005459785461426,
248
+ "eval_runtime": 4.8865,
249
+ "eval_samples_per_second": 4.707,
250
+ "eval_steps_per_second": 1.228,
251
+ "step": 440
252
+ },
253
+ {
254
+ "epoch": 8.36,
255
+ "learning_rate": 3.610166531514436e-05,
256
+ "loss": 0.3172,
257
+ "step": 460
258
+ },
259
+ {
260
+ "epoch": 8.73,
261
+ "learning_rate": 3.465862814232822e-05,
262
+ "loss": 0.3163,
263
+ "step": 480
264
+ },
265
+ {
266
+ "epoch": 9.0,
267
+ "gpt4_scores": 0.0,
268
+ "step": 495
269
+ },
270
+ {
271
+ "epoch": 9.0,
272
+ "eval_loss": 3.270254611968994,
273
+ "eval_runtime": 4.8817,
274
+ "eval_samples_per_second": 4.711,
275
+ "eval_steps_per_second": 1.229,
276
+ "step": 495
277
+ },
278
+ {
279
+ "epoch": 9.09,
280
+ "learning_rate": 3.3176699082935545e-05,
281
+ "loss": 0.3132,
282
+ "step": 500
283
+ },
284
+ {
285
+ "epoch": 9.45,
286
+ "learning_rate": 3.166184534225087e-05,
287
+ "loss": 0.213,
288
+ "step": 520
289
+ },
290
+ {
291
+ "epoch": 9.82,
292
+ "learning_rate": 3.012016670162977e-05,
293
+ "loss": 0.2355,
294
+ "step": 540
295
+ },
296
+ {
297
+ "epoch": 10.0,
298
+ "gpt4_scores": 0.0,
299
+ "step": 550
300
+ },
301
+ {
302
+ "epoch": 10.0,
303
+ "eval_loss": 3.4523885250091553,
304
+ "eval_runtime": 4.8835,
305
+ "eval_samples_per_second": 4.71,
306
+ "eval_steps_per_second": 1.229,
307
+ "step": 550
308
+ },
309
+ {
310
+ "epoch": 10.18,
311
+ "learning_rate": 2.8557870956832132e-05,
312
+ "loss": 0.194,
313
+ "step": 560
314
+ },
315
+ {
316
+ "epoch": 10.55,
317
+ "learning_rate": 2.698124892141971e-05,
318
+ "loss": 0.1577,
319
+ "step": 580
320
+ },
321
+ {
322
+ "epoch": 10.91,
323
+ "learning_rate": 2.5396649095870202e-05,
324
+ "loss": 0.1489,
325
+ "step": 600
326
+ },
327
+ {
328
+ "epoch": 11.0,
329
+ "gpt4_scores": 0.0,
330
+ "step": 605
331
+ },
332
+ {
333
+ "epoch": 11.0,
334
+ "eval_loss": 3.6804909706115723,
335
+ "eval_runtime": 4.8815,
336
+ "eval_samples_per_second": 4.712,
337
+ "eval_steps_per_second": 1.229,
338
+ "step": 605
339
+ },
340
+ {
341
+ "epoch": 11.27,
342
+ "learning_rate": 2.3810452104406444e-05,
343
+ "loss": 0.1045,
344
+ "step": 620
345
+ },
346
+ {
347
+ "epoch": 11.64,
348
+ "learning_rate": 2.222904500247473e-05,
349
+ "loss": 0.0945,
350
+ "step": 640
351
+ },
352
+ {
353
+ "epoch": 12.0,
354
+ "learning_rate": 2.0658795558326743e-05,
355
+ "loss": 0.1179,
356
+ "step": 660
357
+ },
358
+ {
359
+ "epoch": 12.0,
360
+ "gpt4_scores": 0.0,
361
+ "step": 660
362
+ },
363
+ {
364
+ "epoch": 12.0,
365
+ "eval_loss": 3.7960402965545654,
366
+ "eval_runtime": 4.8861,
367
+ "eval_samples_per_second": 4.707,
368
+ "eval_steps_per_second": 1.228,
369
+ "step": 660
370
+ },
371
+ {
372
+ "epoch": 12.36,
373
+ "learning_rate": 1.9106026612264316e-05,
374
+ "loss": 0.0938,
375
+ "step": 680
376
+ },
377
+ {
378
+ "epoch": 12.73,
379
+ "learning_rate": 1.7576990616793137e-05,
380
+ "loss": 0.0712,
381
+ "step": 700
382
+ },
383
+ {
384
+ "epoch": 13.0,
385
+ "gpt4_scores": 0.0,
386
+ "step": 715
387
+ },
388
+ {
389
+ "epoch": 13.0,
390
+ "eval_loss": 3.971480369567871,
391
+ "eval_runtime": 4.8837,
392
+ "eval_samples_per_second": 4.71,
393
+ "eval_steps_per_second": 1.229,
394
+ "step": 715
395
+ },
396
+ {
397
+ "epoch": 13.09,
398
+ "learning_rate": 1.6077844460203206e-05,
399
+ "loss": 0.0715,
400
+ "step": 720
401
+ },
402
+ {
403
+ "epoch": 13.45,
404
+ "learning_rate": 1.4614624674952842e-05,
405
+ "loss": 0.056,
406
+ "step": 740
407
+ },
408
+ {
409
+ "epoch": 13.82,
410
+ "learning_rate": 1.3193223130682936e-05,
411
+ "loss": 0.0782,
412
+ "step": 760
413
+ },
414
+ {
415
+ "epoch": 14.0,
416
+ "gpt4_scores": 0.0,
417
+ "step": 770
418
+ },
419
+ {
420
+ "epoch": 14.0,
421
+ "eval_loss": 4.028377532958984,
422
+ "eval_runtime": 4.8846,
423
+ "eval_samples_per_second": 4.709,
424
+ "eval_steps_per_second": 1.228,
425
+ "step": 770
426
+ },
427
+ {
428
+ "epoch": 14.18,
429
+ "learning_rate": 1.181936330973744e-05,
430
+ "loss": 0.0582,
431
+ "step": 780
432
+ },
433
+ {
434
+ "epoch": 14.55,
435
+ "learning_rate": 1.049857726072005e-05,
436
+ "loss": 0.0518,
437
+ "step": 800
438
+ },
439
+ {
440
+ "epoch": 14.91,
441
+ "learning_rate": 9.236183322886945e-06,
442
+ "loss": 0.0691,
443
+ "step": 820
444
+ },
445
+ {
446
+ "epoch": 15.0,
447
+ "gpt4_scores": 0.0,
448
+ "step": 825
449
+ },
450
+ {
451
+ "epoch": 15.0,
452
+ "eval_loss": 4.157599925994873,
453
+ "eval_runtime": 4.8853,
454
+ "eval_samples_per_second": 4.708,
455
+ "eval_steps_per_second": 1.228,
456
+ "step": 825
457
+ },
458
+ {
459
+ "epoch": 15.27,
460
+ "learning_rate": 8.0372647110717e-06,
461
+ "loss": 0.0483,
462
+ "step": 840
463
+ },
464
+ {
465
+ "epoch": 15.64,
466
+ "learning_rate": 6.906649047373246e-06,
467
+ "loss": 0.0672,
468
+ "step": 860
469
+ },
470
+ {
471
+ "epoch": 16.0,
472
+ "learning_rate": 5.848888922025553e-06,
473
+ "loss": 0.0486,
474
+ "step": 880
475
+ },
476
+ {
477
+ "epoch": 16.0,
478
+ "gpt4_scores": 0.0,
479
+ "step": 880
480
+ },
481
+ {
482
+ "epoch": 16.0,
483
+ "eval_loss": 4.169044017791748,
484
+ "eval_runtime": 4.8903,
485
+ "eval_samples_per_second": 4.703,
486
+ "eval_steps_per_second": 1.227,
487
+ "step": 880
488
+ },
489
+ {
490
+ "epoch": 16.36,
491
+ "learning_rate": 4.868243561723535e-06,
492
+ "loss": 0.044,
493
+ "step": 900
494
+ },
495
+ {
496
+ "epoch": 16.73,
497
+ "learning_rate": 3.968661679220468e-06,
498
+ "loss": 0.0639,
499
+ "step": 920
500
+ },
501
+ {
502
+ "epoch": 17.0,
503
+ "gpt4_scores": 0.0,
504
+ "step": 935
505
+ },
506
+ {
507
+ "epoch": 17.0,
508
+ "eval_loss": 4.209411144256592,
509
+ "eval_runtime": 4.8828,
510
+ "eval_samples_per_second": 4.71,
511
+ "eval_steps_per_second": 1.229,
512
+ "step": 935
513
+ },
514
+ {
515
+ "epoch": 17.09,
516
+ "learning_rate": 3.1537655732553768e-06,
517
+ "loss": 0.0469,
518
+ "step": 940
519
+ },
520
+ {
521
+ "epoch": 17.45,
522
+ "learning_rate": 2.4268365428344736e-06,
523
+ "loss": 0.0427,
524
+ "step": 960
525
+ },
526
+ {
527
+ "epoch": 17.82,
528
+ "learning_rate": 1.790801674598186e-06,
529
+ "loss": 0.0589,
530
+ "step": 980
531
+ },
532
+ {
533
+ "epoch": 18.0,
534
+ "gpt4_scores": 0.0,
535
+ "step": 990
536
+ },
537
+ {
538
+ "epoch": 18.0,
539
+ "eval_loss": 4.243592262268066,
540
+ "eval_runtime": 4.8851,
541
+ "eval_samples_per_second": 4.708,
542
+ "eval_steps_per_second": 1.228,
543
+ "step": 990
544
+ },
545
+ {
546
+ "epoch": 18.18,
547
+ "learning_rate": 1.248222056476367e-06,
548
+ "loss": 0.0453,
549
+ "step": 1000
550
+ },
551
+ {
552
+ "epoch": 18.55,
553
+ "learning_rate": 8.012824650910938e-07,
554
+ "loss": 0.0582,
555
+ "step": 1020
556
+ },
557
+ {
558
+ "epoch": 18.91,
559
+ "learning_rate": 4.517825684323324e-07,
560
+ "loss": 0.042,
561
+ "step": 1040
562
+ },
563
+ {
564
+ "epoch": 19.0,
565
+ "gpt4_scores": 0.0,
566
+ "step": 1045
567
+ },
568
+ {
569
+ "epoch": 19.0,
570
+ "eval_loss": 4.24906063079834,
571
+ "eval_runtime": 4.8836,
572
+ "eval_samples_per_second": 4.71,
573
+ "eval_steps_per_second": 1.229,
574
+ "step": 1045
575
+ },
576
+ {
577
+ "epoch": 19.27,
578
+ "learning_rate": 2.011296792301165e-07,
579
+ "loss": 0.0426,
580
+ "step": 1060
581
+ },
582
+ {
583
+ "epoch": 19.64,
584
+ "learning_rate": 5.033308820289184e-08,
585
+ "loss": 0.0611,
586
+ "step": 1080
587
+ },
588
+ {
589
+ "epoch": 20.0,
590
+ "learning_rate": 0.0,
591
+ "loss": 0.0405,
592
+ "step": 1100
593
+ },
594
+ {
595
+ "epoch": 20.0,
596
+ "gpt4_scores": 0.0,
597
+ "step": 1100
598
+ },
599
+ {
600
+ "epoch": 20.0,
601
+ "eval_loss": 4.249512195587158,
602
+ "eval_runtime": 4.8886,
603
+ "eval_samples_per_second": 4.705,
604
+ "eval_steps_per_second": 1.227,
605
+ "step": 1100
606
+ },
607
+ {
608
+ "epoch": 20.0,
609
+ "step": 1100,
610
+ "total_flos": 3.807078373542298e+16,
611
+ "train_loss": 0.5375886155258526,
612
+ "train_runtime": 2988.6301,
613
+ "train_samples_per_second": 1.452,
614
+ "train_steps_per_second": 0.368
615
  }
616
  ],
617
  "logging_steps": 20,
618
+ "max_steps": 1100,
619
  "num_input_tokens_seen": 0,
620
+ "num_train_epochs": 20,
621
  "save_steps": 20,
622
+ "total_flos": 3.807078373542298e+16,
623
  "train_batch_size": 4,
624
  "trial_name": null,
625
  "trial_params": null