Commit
•
068cda0
1
Parent(s):
ae91545
End of training
Browse files- README.md +6 -4
- all_results.json +16 -0
- eval_results.json +10 -0
- train_results.json +9 -0
- trainer_state.json +388 -0
README.md
CHANGED
@@ -2,6 +2,8 @@
|
|
2 |
license: cc-by-nc-4.0
|
3 |
library_name: peft
|
4 |
tags:
|
|
|
|
|
5 |
- generated_from_trainer
|
6 |
base_model: facebook/musicgen-melody-large
|
7 |
model-index:
|
@@ -16,11 +18,11 @@ should probably proofread and complete it, then remove this comment. -->
|
|
16 |
[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/sanchit-gandhi/black-eyed-peas/runs/coi600r4)
|
17 |
# black-eyed-peas-v1-crafted-variable-prompt-8-epochs-piano-prompts
|
18 |
|
19 |
-
This model is a fine-tuned version of [facebook/musicgen-melody-large](https://huggingface.co/facebook/musicgen-melody-large) on
|
20 |
It achieves the following results on the evaluation set:
|
21 |
-
- Loss: 3.
|
22 |
-
- Text Cosine Sim: 0.
|
23 |
-
- Audio Cosine Sim: 0.
|
24 |
|
25 |
## Model description
|
26 |
|
|
|
2 |
license: cc-by-nc-4.0
|
3 |
library_name: peft
|
4 |
tags:
|
5 |
+
- text-to-audio
|
6 |
+
- sweet-dreambooths/black-eyed-peas-v1-piano-prompts
|
7 |
- generated_from_trainer
|
8 |
base_model: facebook/musicgen-melody-large
|
9 |
model-index:
|
|
|
18 |
[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/sanchit-gandhi/black-eyed-peas/runs/coi600r4)
|
19 |
# black-eyed-peas-v1-crafted-variable-prompt-8-epochs-piano-prompts
|
20 |
|
21 |
+
This model is a fine-tuned version of [facebook/musicgen-melody-large](https://huggingface.co/facebook/musicgen-melody-large) on the SWEET-DREAMBOOTHS/BLACK-EYED-PEAS-V1-PIANO-PROMPTS - DEFAULT dataset.
|
22 |
It achieves the following results on the evaluation set:
|
23 |
+
- Loss: 3.0989
|
24 |
+
- Text Cosine Sim: 0.3821
|
25 |
+
- Audio Cosine Sim: 0.5580
|
26 |
|
27 |
## Model description
|
28 |
|
all_results.json
ADDED
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"epoch": 7.917525773195877,
|
3 |
+
"eval_audio_cosine_sim": 0.5579971075057983,
|
4 |
+
"eval_loss": 3.0988662242889404,
|
5 |
+
"eval_runtime": 2019.3746,
|
6 |
+
"eval_samples": 12,
|
7 |
+
"eval_samples_per_second": 0.006,
|
8 |
+
"eval_steps_per_second": 0.006,
|
9 |
+
"eval_text_cosine_sim": 0.3821232318878174,
|
10 |
+
"total_flos": 962957869218000.0,
|
11 |
+
"train_loss": 8.716416617234549,
|
12 |
+
"train_runtime": 7439.5895,
|
13 |
+
"train_samples": 97,
|
14 |
+
"train_samples_per_second": 0.104,
|
15 |
+
"train_steps_per_second": 0.006
|
16 |
+
}
|
eval_results.json
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"epoch": 7.917525773195877,
|
3 |
+
"eval_audio_cosine_sim": 0.5579971075057983,
|
4 |
+
"eval_loss": 3.0988662242889404,
|
5 |
+
"eval_runtime": 2019.3746,
|
6 |
+
"eval_samples": 12,
|
7 |
+
"eval_samples_per_second": 0.006,
|
8 |
+
"eval_steps_per_second": 0.006,
|
9 |
+
"eval_text_cosine_sim": 0.3821232318878174
|
10 |
+
}
|
train_results.json
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"epoch": 7.917525773195877,
|
3 |
+
"total_flos": 962957869218000.0,
|
4 |
+
"train_loss": 8.716416617234549,
|
5 |
+
"train_runtime": 7439.5895,
|
6 |
+
"train_samples": 97,
|
7 |
+
"train_samples_per_second": 0.104,
|
8 |
+
"train_steps_per_second": 0.006
|
9 |
+
}
|
trainer_state.json
ADDED
@@ -0,0 +1,388 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 7.917525773195877,
|
5 |
+
"eval_steps": 30,
|
6 |
+
"global_step": 48,
|
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.16494845360824742,
|
13 |
+
"grad_norm": 1.603933334350586,
|
14 |
+
"learning_rate": 2.9375e-05,
|
15 |
+
"loss": 10.1011,
|
16 |
+
"step": 1
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.32989690721649484,
|
20 |
+
"grad_norm": 1.7179864645004272,
|
21 |
+
"learning_rate": 2.875e-05,
|
22 |
+
"loss": 10.2338,
|
23 |
+
"step": 2
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.4948453608247423,
|
27 |
+
"grad_norm": 1.7488847970962524,
|
28 |
+
"learning_rate": 2.8125e-05,
|
29 |
+
"loss": 10.0559,
|
30 |
+
"step": 3
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.6597938144329897,
|
34 |
+
"grad_norm": 2.1688830852508545,
|
35 |
+
"learning_rate": 2.75e-05,
|
36 |
+
"loss": 10.2016,
|
37 |
+
"step": 4
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.8247422680412371,
|
41 |
+
"grad_norm": 2.111985206604004,
|
42 |
+
"learning_rate": 2.6875000000000003e-05,
|
43 |
+
"loss": 10.0667,
|
44 |
+
"step": 5
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.9896907216494846,
|
48 |
+
"grad_norm": 2.164363145828247,
|
49 |
+
"learning_rate": 2.625e-05,
|
50 |
+
"loss": 9.8519,
|
51 |
+
"step": 6
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 1.1546391752577319,
|
55 |
+
"grad_norm": 2.308406114578247,
|
56 |
+
"learning_rate": 2.5625e-05,
|
57 |
+
"loss": 9.92,
|
58 |
+
"step": 7
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 1.3195876288659794,
|
62 |
+
"grad_norm": 2.625577688217163,
|
63 |
+
"learning_rate": 2.5e-05,
|
64 |
+
"loss": 9.8473,
|
65 |
+
"step": 8
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 1.4845360824742269,
|
69 |
+
"grad_norm": 3.0818679332733154,
|
70 |
+
"learning_rate": 2.4375e-05,
|
71 |
+
"loss": 9.8717,
|
72 |
+
"step": 9
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 1.6494845360824741,
|
76 |
+
"grad_norm": 2.6675806045532227,
|
77 |
+
"learning_rate": 2.3749999999999998e-05,
|
78 |
+
"loss": 9.6087,
|
79 |
+
"step": 10
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 1.8144329896907216,
|
83 |
+
"grad_norm": 2.8696348667144775,
|
84 |
+
"learning_rate": 2.3125000000000003e-05,
|
85 |
+
"loss": 9.6005,
|
86 |
+
"step": 11
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 1.9793814432989691,
|
90 |
+
"grad_norm": 3.16064453125,
|
91 |
+
"learning_rate": 2.25e-05,
|
92 |
+
"loss": 9.5111,
|
93 |
+
"step": 12
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 2.1443298969072164,
|
97 |
+
"grad_norm": 3.2917752265930176,
|
98 |
+
"learning_rate": 2.1875e-05,
|
99 |
+
"loss": 9.4589,
|
100 |
+
"step": 13
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 2.3092783505154637,
|
104 |
+
"grad_norm": 3.362398624420166,
|
105 |
+
"learning_rate": 2.125e-05,
|
106 |
+
"loss": 9.3264,
|
107 |
+
"step": 14
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 2.4742268041237114,
|
111 |
+
"grad_norm": 3.687838315963745,
|
112 |
+
"learning_rate": 2.0625e-05,
|
113 |
+
"loss": 9.3032,
|
114 |
+
"step": 15
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 2.6391752577319587,
|
118 |
+
"grad_norm": 3.6583521366119385,
|
119 |
+
"learning_rate": 1.9999999999999998e-05,
|
120 |
+
"loss": 9.1917,
|
121 |
+
"step": 16
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 2.804123711340206,
|
125 |
+
"grad_norm": 3.846595048904419,
|
126 |
+
"learning_rate": 1.9375e-05,
|
127 |
+
"loss": 9.0754,
|
128 |
+
"step": 17
|
129 |
+
},
|
130 |
+
{
|
131 |
+
"epoch": 2.9690721649484537,
|
132 |
+
"grad_norm": 3.7431483268737793,
|
133 |
+
"learning_rate": 1.8750000000000002e-05,
|
134 |
+
"loss": 9.0118,
|
135 |
+
"step": 18
|
136 |
+
},
|
137 |
+
{
|
138 |
+
"epoch": 3.134020618556701,
|
139 |
+
"grad_norm": 3.54080867767334,
|
140 |
+
"learning_rate": 1.8125e-05,
|
141 |
+
"loss": 8.8953,
|
142 |
+
"step": 19
|
143 |
+
},
|
144 |
+
{
|
145 |
+
"epoch": 3.2989690721649483,
|
146 |
+
"grad_norm": 3.989246368408203,
|
147 |
+
"learning_rate": 1.7500000000000002e-05,
|
148 |
+
"loss": 8.8344,
|
149 |
+
"step": 20
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"epoch": 3.463917525773196,
|
153 |
+
"grad_norm": 3.540311574935913,
|
154 |
+
"learning_rate": 1.6875e-05,
|
155 |
+
"loss": 8.7965,
|
156 |
+
"step": 21
|
157 |
+
},
|
158 |
+
{
|
159 |
+
"epoch": 3.6288659793814433,
|
160 |
+
"grad_norm": 3.735725164413452,
|
161 |
+
"learning_rate": 1.625e-05,
|
162 |
+
"loss": 8.6344,
|
163 |
+
"step": 22
|
164 |
+
},
|
165 |
+
{
|
166 |
+
"epoch": 3.7938144329896906,
|
167 |
+
"grad_norm": 4.285547733306885,
|
168 |
+
"learning_rate": 1.5625e-05,
|
169 |
+
"loss": 8.544,
|
170 |
+
"step": 23
|
171 |
+
},
|
172 |
+
{
|
173 |
+
"epoch": 3.9587628865979383,
|
174 |
+
"grad_norm": 3.555814027786255,
|
175 |
+
"learning_rate": 1.5e-05,
|
176 |
+
"loss": 8.4752,
|
177 |
+
"step": 24
|
178 |
+
},
|
179 |
+
{
|
180 |
+
"epoch": 4.123711340206185,
|
181 |
+
"grad_norm": 3.49817156791687,
|
182 |
+
"learning_rate": 1.4375e-05,
|
183 |
+
"loss": 8.4403,
|
184 |
+
"step": 25
|
185 |
+
},
|
186 |
+
{
|
187 |
+
"epoch": 4.288659793814433,
|
188 |
+
"grad_norm": 3.3520140647888184,
|
189 |
+
"learning_rate": 1.375e-05,
|
190 |
+
"loss": 8.4122,
|
191 |
+
"step": 26
|
192 |
+
},
|
193 |
+
{
|
194 |
+
"epoch": 4.453608247422681,
|
195 |
+
"grad_norm": 3.296865224838257,
|
196 |
+
"learning_rate": 1.3125e-05,
|
197 |
+
"loss": 8.3044,
|
198 |
+
"step": 27
|
199 |
+
},
|
200 |
+
{
|
201 |
+
"epoch": 4.618556701030927,
|
202 |
+
"grad_norm": 3.4501516819000244,
|
203 |
+
"learning_rate": 1.25e-05,
|
204 |
+
"loss": 8.1882,
|
205 |
+
"step": 28
|
206 |
+
},
|
207 |
+
{
|
208 |
+
"epoch": 4.783505154639175,
|
209 |
+
"grad_norm": 3.2213199138641357,
|
210 |
+
"learning_rate": 1.1874999999999999e-05,
|
211 |
+
"loss": 8.2403,
|
212 |
+
"step": 29
|
213 |
+
},
|
214 |
+
{
|
215 |
+
"epoch": 4.948453608247423,
|
216 |
+
"grad_norm": 3.34763765335083,
|
217 |
+
"learning_rate": 1.125e-05,
|
218 |
+
"loss": 8.1814,
|
219 |
+
"step": 30
|
220 |
+
},
|
221 |
+
{
|
222 |
+
"epoch": 4.948453608247423,
|
223 |
+
"eval_audio_cosine_sim": 0.5797536969184875,
|
224 |
+
"eval_loss": 3.095458745956421,
|
225 |
+
"eval_runtime": 2074.7641,
|
226 |
+
"eval_samples_per_second": 0.006,
|
227 |
+
"eval_steps_per_second": 0.006,
|
228 |
+
"eval_text_cosine_sim": 0.38171055912971497,
|
229 |
+
"step": 30
|
230 |
+
},
|
231 |
+
{
|
232 |
+
"epoch": 5.11340206185567,
|
233 |
+
"grad_norm": 3.406193256378174,
|
234 |
+
"learning_rate": 1.0625e-05,
|
235 |
+
"loss": 8.109,
|
236 |
+
"step": 31
|
237 |
+
},
|
238 |
+
{
|
239 |
+
"epoch": 5.278350515463917,
|
240 |
+
"grad_norm": 3.601793050765991,
|
241 |
+
"learning_rate": 9.999999999999999e-06,
|
242 |
+
"loss": 7.9923,
|
243 |
+
"step": 32
|
244 |
+
},
|
245 |
+
{
|
246 |
+
"epoch": 5.443298969072165,
|
247 |
+
"grad_norm": 3.392193555831909,
|
248 |
+
"learning_rate": 9.375000000000001e-06,
|
249 |
+
"loss": 8.202,
|
250 |
+
"step": 33
|
251 |
+
},
|
252 |
+
{
|
253 |
+
"epoch": 5.608247422680412,
|
254 |
+
"grad_norm": 3.5988409519195557,
|
255 |
+
"learning_rate": 8.750000000000001e-06,
|
256 |
+
"loss": 8.0204,
|
257 |
+
"step": 34
|
258 |
+
},
|
259 |
+
{
|
260 |
+
"epoch": 5.77319587628866,
|
261 |
+
"grad_norm": 3.4323904514312744,
|
262 |
+
"learning_rate": 8.125e-06,
|
263 |
+
"loss": 8.008,
|
264 |
+
"step": 35
|
265 |
+
},
|
266 |
+
{
|
267 |
+
"epoch": 5.938144329896907,
|
268 |
+
"grad_norm": 3.5866267681121826,
|
269 |
+
"learning_rate": 7.5e-06,
|
270 |
+
"loss": 7.8909,
|
271 |
+
"step": 36
|
272 |
+
},
|
273 |
+
{
|
274 |
+
"epoch": 6.103092783505154,
|
275 |
+
"grad_norm": 3.682654619216919,
|
276 |
+
"learning_rate": 6.875e-06,
|
277 |
+
"loss": 7.8228,
|
278 |
+
"step": 37
|
279 |
+
},
|
280 |
+
{
|
281 |
+
"epoch": 6.268041237113402,
|
282 |
+
"grad_norm": 3.4344913959503174,
|
283 |
+
"learning_rate": 6.25e-06,
|
284 |
+
"loss": 7.8066,
|
285 |
+
"step": 38
|
286 |
+
},
|
287 |
+
{
|
288 |
+
"epoch": 6.43298969072165,
|
289 |
+
"grad_norm": 3.7810468673706055,
|
290 |
+
"learning_rate": 5.625e-06,
|
291 |
+
"loss": 7.8461,
|
292 |
+
"step": 39
|
293 |
+
},
|
294 |
+
{
|
295 |
+
"epoch": 6.597938144329897,
|
296 |
+
"grad_norm": 3.7566187381744385,
|
297 |
+
"learning_rate": 4.9999999999999996e-06,
|
298 |
+
"loss": 7.8968,
|
299 |
+
"step": 40
|
300 |
+
},
|
301 |
+
{
|
302 |
+
"epoch": 6.762886597938144,
|
303 |
+
"grad_norm": 3.617682695388794,
|
304 |
+
"learning_rate": 4.3750000000000005e-06,
|
305 |
+
"loss": 7.9953,
|
306 |
+
"step": 41
|
307 |
+
},
|
308 |
+
{
|
309 |
+
"epoch": 6.927835051546392,
|
310 |
+
"grad_norm": 3.5307486057281494,
|
311 |
+
"learning_rate": 3.75e-06,
|
312 |
+
"loss": 7.8764,
|
313 |
+
"step": 42
|
314 |
+
},
|
315 |
+
{
|
316 |
+
"epoch": 7.092783505154639,
|
317 |
+
"grad_norm": 3.624812364578247,
|
318 |
+
"learning_rate": 3.125e-06,
|
319 |
+
"loss": 7.7608,
|
320 |
+
"step": 43
|
321 |
+
},
|
322 |
+
{
|
323 |
+
"epoch": 7.257731958762887,
|
324 |
+
"grad_norm": 3.573174476623535,
|
325 |
+
"learning_rate": 2.4999999999999998e-06,
|
326 |
+
"loss": 7.705,
|
327 |
+
"step": 44
|
328 |
+
},
|
329 |
+
{
|
330 |
+
"epoch": 7.422680412371134,
|
331 |
+
"grad_norm": 3.3969686031341553,
|
332 |
+
"learning_rate": 1.875e-06,
|
333 |
+
"loss": 7.9902,
|
334 |
+
"step": 45
|
335 |
+
},
|
336 |
+
{
|
337 |
+
"epoch": 7.587628865979381,
|
338 |
+
"grad_norm": 3.5248804092407227,
|
339 |
+
"learning_rate": 1.2499999999999999e-06,
|
340 |
+
"loss": 7.8566,
|
341 |
+
"step": 46
|
342 |
+
},
|
343 |
+
{
|
344 |
+
"epoch": 7.752577319587629,
|
345 |
+
"grad_norm": 3.8090217113494873,
|
346 |
+
"learning_rate": 6.249999999999999e-07,
|
347 |
+
"loss": 7.5691,
|
348 |
+
"step": 47
|
349 |
+
},
|
350 |
+
{
|
351 |
+
"epoch": 7.917525773195877,
|
352 |
+
"grad_norm": 3.4885289669036865,
|
353 |
+
"learning_rate": 0.0,
|
354 |
+
"loss": 7.8552,
|
355 |
+
"step": 48
|
356 |
+
},
|
357 |
+
{
|
358 |
+
"epoch": 7.917525773195877,
|
359 |
+
"step": 48,
|
360 |
+
"total_flos": 962957869218000.0,
|
361 |
+
"train_loss": 8.716416617234549,
|
362 |
+
"train_runtime": 7439.5895,
|
363 |
+
"train_samples_per_second": 0.104,
|
364 |
+
"train_steps_per_second": 0.006
|
365 |
+
}
|
366 |
+
],
|
367 |
+
"logging_steps": 1.0,
|
368 |
+
"max_steps": 48,
|
369 |
+
"num_input_tokens_seen": 0,
|
370 |
+
"num_train_epochs": 8,
|
371 |
+
"save_steps": 500,
|
372 |
+
"stateful_callbacks": {
|
373 |
+
"TrainerControl": {
|
374 |
+
"args": {
|
375 |
+
"should_epoch_stop": false,
|
376 |
+
"should_evaluate": false,
|
377 |
+
"should_log": false,
|
378 |
+
"should_save": false,
|
379 |
+
"should_training_stop": false
|
380 |
+
},
|
381 |
+
"attributes": {}
|
382 |
+
}
|
383 |
+
},
|
384 |
+
"total_flos": 962957869218000.0,
|
385 |
+
"train_batch_size": 1,
|
386 |
+
"trial_name": null,
|
387 |
+
"trial_params": null
|
388 |
+
}
|