sedrickkeh commited on
Commit
b8b8c72
·
verified ·
1 Parent(s): 9ca097e

End of training

Browse files
README.md CHANGED
@@ -4,6 +4,7 @@ license: apache-2.0
4
  base_model: mistralai/Mistral-7B-v0.1
5
  tags:
6
  - llama-factory
 
7
  - generated_from_trainer
8
  model-index:
9
  - name: hp_ablations_grid_mistral_bsz4096_lr2e-6_scheduler-cosine-warmup0.05-minlr5e-7
@@ -15,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  # hp_ablations_grid_mistral_bsz4096_lr2e-6_scheduler-cosine-warmup0.05-minlr5e-7
17
 
18
- This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
  - Loss: 0.0622
21
 
 
4
  base_model: mistralai/Mistral-7B-v0.1
5
  tags:
6
  - llama-factory
7
+ - full
8
  - generated_from_trainer
9
  model-index:
10
  - name: hp_ablations_grid_mistral_bsz4096_lr2e-6_scheduler-cosine-warmup0.05-minlr5e-7
 
16
 
17
  # hp_ablations_grid_mistral_bsz4096_lr2e-6_scheduler-cosine-warmup0.05-minlr5e-7
18
 
19
+ This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on the mlfoundations-dev/oh-dcft-v3-llama3.1-nemotron-70b_shareGPT_format dataset.
20
  It achieves the following results on the evaluation set:
21
  - Loss: 0.0622
22
 
all_results.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.062153831124305725,
4
+ "eval_runtime": 88.9521,
5
+ "eval_samples_per_second": 203.514,
6
+ "eval_steps_per_second": 0.405,
7
+ "total_flos": 3375200049561600.0,
8
+ "train_loss": 0.5371241434699013,
9
+ "train_runtime": 15001.3184,
10
+ "train_samples_per_second": 68.785,
11
+ "train_steps_per_second": 0.017
12
+ }
eval_results.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_loss": 0.062153831124305725,
4
+ "eval_runtime": 88.9521,
5
+ "eval_samples_per_second": 203.514,
6
+ "eval_steps_per_second": 0.405
7
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "total_flos": 3375200049561600.0,
4
+ "train_loss": 0.5371241434699013,
5
+ "train_runtime": 15001.3184,
6
+ "train_samples_per_second": 68.785,
7
+ "train_steps_per_second": 0.017
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,241 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 252,
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.11904761904761904,
13
+ "grad_norm": 6.873670258140521,
14
+ "learning_rate": 1.5384615384615385e-06,
15
+ "loss": 0.8434,
16
+ "step": 10
17
+ },
18
+ {
19
+ "epoch": 0.23809523809523808,
20
+ "grad_norm": 1.799131772835958,
21
+ "learning_rate": 1.9968273303105645e-06,
22
+ "loss": 0.7049,
23
+ "step": 20
24
+ },
25
+ {
26
+ "epoch": 0.35714285714285715,
27
+ "grad_norm": 1.7057448542352618,
28
+ "learning_rate": 1.981352307157077e-06,
29
+ "loss": 0.6308,
30
+ "step": 30
31
+ },
32
+ {
33
+ "epoch": 0.47619047619047616,
34
+ "grad_norm": 3.2000151180883942,
35
+ "learning_rate": 1.9532588586635555e-06,
36
+ "loss": 0.5987,
37
+ "step": 40
38
+ },
39
+ {
40
+ "epoch": 0.5952380952380952,
41
+ "grad_norm": 1.5365760572953298,
42
+ "learning_rate": 1.9130316966151297e-06,
43
+ "loss": 0.5818,
44
+ "step": 50
45
+ },
46
+ {
47
+ "epoch": 0.7142857142857143,
48
+ "grad_norm": 2.3881726368494456,
49
+ "learning_rate": 1.8613648824593137e-06,
50
+ "loss": 0.5689,
51
+ "step": 60
52
+ },
53
+ {
54
+ "epoch": 0.8333333333333334,
55
+ "grad_norm": 1.3104030138998584,
56
+ "learning_rate": 1.7991498522804724e-06,
57
+ "loss": 0.5591,
58
+ "step": 70
59
+ },
60
+ {
61
+ "epoch": 0.9523809523809523,
62
+ "grad_norm": 1.070092075314844,
63
+ "learning_rate": 1.7274600363606417e-06,
64
+ "loss": 0.5488,
65
+ "step": 80
66
+ },
67
+ {
68
+ "epoch": 1.0,
69
+ "eval_loss": 0.06786686182022095,
70
+ "eval_runtime": 89.6525,
71
+ "eval_samples_per_second": 201.924,
72
+ "eval_steps_per_second": 0.402,
73
+ "step": 84
74
+ },
75
+ {
76
+ "epoch": 1.0714285714285714,
77
+ "grad_norm": 1.598258674771628,
78
+ "learning_rate": 1.6475323386939333e-06,
79
+ "loss": 0.5342,
80
+ "step": 90
81
+ },
82
+ {
83
+ "epoch": 1.1904761904761905,
84
+ "grad_norm": 1.6651178950942467,
85
+ "learning_rate": 1.560745795998681e-06,
86
+ "loss": 0.5232,
87
+ "step": 100
88
+ },
89
+ {
90
+ "epoch": 1.3095238095238095,
91
+ "grad_norm": 2.0206003862608646,
92
+ "learning_rate": 1.4685977844351884e-06,
93
+ "loss": 0.5194,
94
+ "step": 110
95
+ },
96
+ {
97
+ "epoch": 1.4285714285714286,
98
+ "grad_norm": 1.5131024957485417,
99
+ "learning_rate": 1.3726781845477123e-06,
100
+ "loss": 0.5147,
101
+ "step": 120
102
+ },
103
+ {
104
+ "epoch": 1.5476190476190477,
105
+ "grad_norm": 1.0648784167624672,
106
+ "learning_rate": 1.274641950177214e-06,
107
+ "loss": 0.5089,
108
+ "step": 130
109
+ },
110
+ {
111
+ "epoch": 1.6666666666666665,
112
+ "grad_norm": 1.0063001149457829,
113
+ "learning_rate": 1.176180554628579e-06,
114
+ "loss": 0.5062,
115
+ "step": 140
116
+ },
117
+ {
118
+ "epoch": 1.7857142857142856,
119
+ "grad_norm": 0.614837213575453,
120
+ "learning_rate": 1.0789928067473494e-06,
121
+ "loss": 0.5027,
122
+ "step": 150
123
+ },
124
+ {
125
+ "epoch": 1.9047619047619047,
126
+ "grad_norm": 0.6264817095760062,
127
+ "learning_rate": 9.847555404323228e-07,
128
+ "loss": 0.4983,
129
+ "step": 160
130
+ },
131
+ {
132
+ "epoch": 2.0,
133
+ "eval_loss": 0.06357834488153458,
134
+ "eval_runtime": 89.3936,
135
+ "eval_samples_per_second": 202.509,
136
+ "eval_steps_per_second": 0.403,
137
+ "step": 168
138
+ },
139
+ {
140
+ "epoch": 2.0238095238095237,
141
+ "grad_norm": 0.7595543213075306,
142
+ "learning_rate": 8.95094683294048e-07,
143
+ "loss": 0.4937,
144
+ "step": 170
145
+ },
146
+ {
147
+ "epoch": 2.142857142857143,
148
+ "grad_norm": 0.8075613915746389,
149
+ "learning_rate": 8.115572036276285e-07,
150
+ "loss": 0.4823,
151
+ "step": 180
152
+ },
153
+ {
154
+ "epoch": 2.261904761904762,
155
+ "grad_norm": 0.7283677919022348,
156
+ "learning_rate": 7.355844197142121e-07,
157
+ "loss": 0.4802,
158
+ "step": 190
159
+ },
160
+ {
161
+ "epoch": 2.380952380952381,
162
+ "grad_norm": 0.5430468412018504,
163
+ "learning_rate": 6.684871319605308e-07,
164
+ "loss": 0.4761,
165
+ "step": 200
166
+ },
167
+ {
168
+ "epoch": 2.5,
169
+ "grad_norm": 0.5606001429431275,
170
+ "learning_rate": 6.114230069354231e-07,
171
+ "loss": 0.4746,
172
+ "step": 210
173
+ },
174
+ {
175
+ "epoch": 2.619047619047619,
176
+ "grad_norm": 0.5550155038305291,
177
+ "learning_rate": 5.653766035090493e-07,
178
+ "loss": 0.4764,
179
+ "step": 220
180
+ },
181
+ {
182
+ "epoch": 2.738095238095238,
183
+ "grad_norm": 0.579557506598742,
184
+ "learning_rate": 5.311423857148533e-07,
185
+ "loss": 0.4747,
186
+ "step": 230
187
+ },
188
+ {
189
+ "epoch": 2.857142857142857,
190
+ "grad_norm": 0.5367862820074156,
191
+ "learning_rate": 5.093110154227483e-07,
192
+ "loss": 0.4734,
193
+ "step": 240
194
+ },
195
+ {
196
+ "epoch": 2.9761904761904763,
197
+ "grad_norm": 0.5135404834401238,
198
+ "learning_rate": 5.002591613236042e-07,
199
+ "loss": 0.4707,
200
+ "step": 250
201
+ },
202
+ {
203
+ "epoch": 3.0,
204
+ "eval_loss": 0.062153831124305725,
205
+ "eval_runtime": 89.1749,
206
+ "eval_samples_per_second": 203.006,
207
+ "eval_steps_per_second": 0.404,
208
+ "step": 252
209
+ },
210
+ {
211
+ "epoch": 3.0,
212
+ "step": 252,
213
+ "total_flos": 3375200049561600.0,
214
+ "train_loss": 0.5371241434699013,
215
+ "train_runtime": 15001.3184,
216
+ "train_samples_per_second": 68.785,
217
+ "train_steps_per_second": 0.017
218
+ }
219
+ ],
220
+ "logging_steps": 10,
221
+ "max_steps": 252,
222
+ "num_input_tokens_seen": 0,
223
+ "num_train_epochs": 3,
224
+ "save_steps": 500,
225
+ "stateful_callbacks": {
226
+ "TrainerControl": {
227
+ "args": {
228
+ "should_epoch_stop": false,
229
+ "should_evaluate": false,
230
+ "should_log": false,
231
+ "should_save": true,
232
+ "should_training_stop": true
233
+ },
234
+ "attributes": {}
235
+ }
236
+ },
237
+ "total_flos": 3375200049561600.0,
238
+ "train_batch_size": 8,
239
+ "trial_name": null,
240
+ "trial_params": null
241
+ }
training_eval_loss.png ADDED
training_loss.png ADDED