sanchit-gandhi commited on
Commit
ca12e87
·
verified ·
1 Parent(s): fd9cee9

End of training

Browse files
Files changed (5) hide show
  1. README.md +5 -3
  2. all_results.json +15 -0
  3. eval_results.json +9 -0
  4. train_results.json +9 -0
  5. trainer_state.json +320 -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,10 +18,10 @@ 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/lq30qz9w)
17
  # black-eyed-peas-v1-autotuned
18
 
19
- This model is a fine-tuned version of [facebook/musicgen-melody-large](https://huggingface.co/facebook/musicgen-melody-large) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 5.1036
22
- - Clap: 0.0850
23
 
24
  ## Model description
25
 
 
2
  license: cc-by-nc-4.0
3
  library_name: peft
4
  tags:
5
+ - text-to-audio
6
+ - sweet-dreambooths/black-eyed-peas-v1-autotuned
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/lq30qz9w)
19
  # black-eyed-peas-v1-autotuned
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-AUTOTUNED - DEFAULT dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 5.1025
24
+ - Clap: 0.0905
25
 
26
  ## Model description
27
 
all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.0,
3
+ "eval_clap": 0.0905328094959259,
4
+ "eval_loss": 5.102526664733887,
5
+ "eval_runtime": 303.1993,
6
+ "eval_samples": 8,
7
+ "eval_samples_per_second": 0.026,
8
+ "eval_steps_per_second": 0.026,
9
+ "total_flos": 152209974742572.0,
10
+ "train_loss": 6.930231094360352,
11
+ "train_runtime": 3130.5046,
12
+ "train_samples": 121,
13
+ "train_samples_per_second": 0.077,
14
+ "train_steps_per_second": 0.01
15
+ }
eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.0,
3
+ "eval_clap": 0.0905328094959259,
4
+ "eval_loss": 5.102526664733887,
5
+ "eval_runtime": 303.1993,
6
+ "eval_samples": 8,
7
+ "eval_samples_per_second": 0.026,
8
+ "eval_steps_per_second": 0.026
9
+ }
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.0,
3
+ "total_flos": 152209974742572.0,
4
+ "train_loss": 6.930231094360352,
5
+ "train_runtime": 3130.5046,
6
+ "train_samples": 121,
7
+ "train_samples_per_second": 0.077,
8
+ "train_steps_per_second": 0.01
9
+ }
trainer_state.json ADDED
@@ -0,0 +1,320 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 2.0,
5
+ "eval_steps": 5,
6
+ "global_step": 32,
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.0625,
13
+ "grad_norm": 1.8187159299850464,
14
+ "learning_rate": 0.00019375000000000002,
15
+ "loss": 10.0878,
16
+ "step": 1
17
+ },
18
+ {
19
+ "epoch": 0.125,
20
+ "grad_norm": 1.9498610496520996,
21
+ "learning_rate": 0.0001875,
22
+ "loss": 9.9916,
23
+ "step": 2
24
+ },
25
+ {
26
+ "epoch": 0.1875,
27
+ "grad_norm": 2.7474117279052734,
28
+ "learning_rate": 0.00018125000000000001,
29
+ "loss": 9.7044,
30
+ "step": 3
31
+ },
32
+ {
33
+ "epoch": 0.25,
34
+ "grad_norm": 3.391249418258667,
35
+ "learning_rate": 0.000175,
36
+ "loss": 8.9738,
37
+ "step": 4
38
+ },
39
+ {
40
+ "epoch": 0.3125,
41
+ "grad_norm": 3.6622118949890137,
42
+ "learning_rate": 0.00016875,
43
+ "loss": 8.2705,
44
+ "step": 5
45
+ },
46
+ {
47
+ "epoch": 0.3125,
48
+ "eval_clap": 0.31716692447662354,
49
+ "eval_loss": 3.0108680725097656,
50
+ "eval_runtime": 265.8708,
51
+ "eval_samples_per_second": 0.03,
52
+ "eval_steps_per_second": 0.03,
53
+ "step": 5
54
+ },
55
+ {
56
+ "epoch": 0.375,
57
+ "grad_norm": 4.368948936462402,
58
+ "learning_rate": 0.00016250000000000002,
59
+ "loss": 7.8107,
60
+ "step": 6
61
+ },
62
+ {
63
+ "epoch": 0.4375,
64
+ "grad_norm": 6.55327033996582,
65
+ "learning_rate": 0.00015625,
66
+ "loss": 7.507,
67
+ "step": 7
68
+ },
69
+ {
70
+ "epoch": 0.5,
71
+ "grad_norm": 5.326369762420654,
72
+ "learning_rate": 0.00015000000000000001,
73
+ "loss": 7.0676,
74
+ "step": 8
75
+ },
76
+ {
77
+ "epoch": 0.5625,
78
+ "grad_norm": 3.2963500022888184,
79
+ "learning_rate": 0.00014375,
80
+ "loss": 7.0271,
81
+ "step": 9
82
+ },
83
+ {
84
+ "epoch": 0.625,
85
+ "grad_norm": 2.0462214946746826,
86
+ "learning_rate": 0.0001375,
87
+ "loss": 6.4565,
88
+ "step": 10
89
+ },
90
+ {
91
+ "epoch": 0.625,
92
+ "eval_clap": 0.20497097074985504,
93
+ "eval_loss": 3.3557252883911133,
94
+ "eval_runtime": 503.1747,
95
+ "eval_samples_per_second": 0.016,
96
+ "eval_steps_per_second": 0.016,
97
+ "step": 10
98
+ },
99
+ {
100
+ "epoch": 0.6875,
101
+ "grad_norm": 2.9796876907348633,
102
+ "learning_rate": 0.00013125000000000002,
103
+ "loss": 7.025,
104
+ "step": 11
105
+ },
106
+ {
107
+ "epoch": 0.75,
108
+ "grad_norm": 1.7379363775253296,
109
+ "learning_rate": 0.000125,
110
+ "loss": 6.4613,
111
+ "step": 12
112
+ },
113
+ {
114
+ "epoch": 0.8125,
115
+ "grad_norm": 2.6363821029663086,
116
+ "learning_rate": 0.00011875,
117
+ "loss": 6.8828,
118
+ "step": 13
119
+ },
120
+ {
121
+ "epoch": 0.875,
122
+ "grad_norm": 1.376232624053955,
123
+ "learning_rate": 0.00011250000000000001,
124
+ "loss": 6.3572,
125
+ "step": 14
126
+ },
127
+ {
128
+ "epoch": 0.9375,
129
+ "grad_norm": 1.8433048725128174,
130
+ "learning_rate": 0.00010625000000000001,
131
+ "loss": 6.0972,
132
+ "step": 15
133
+ },
134
+ {
135
+ "epoch": 0.9375,
136
+ "eval_clap": 0.12289813160896301,
137
+ "eval_loss": 4.520351409912109,
138
+ "eval_runtime": 513.434,
139
+ "eval_samples_per_second": 0.016,
140
+ "eval_steps_per_second": 0.016,
141
+ "step": 15
142
+ },
143
+ {
144
+ "epoch": 1.0,
145
+ "grad_norm": 2.2514262199401855,
146
+ "learning_rate": 0.0001,
147
+ "loss": 6.3686,
148
+ "step": 16
149
+ },
150
+ {
151
+ "epoch": 1.0625,
152
+ "grad_norm": 0.9838012456893921,
153
+ "learning_rate": 9.375e-05,
154
+ "loss": 6.2712,
155
+ "step": 17
156
+ },
157
+ {
158
+ "epoch": 1.125,
159
+ "grad_norm": 0.983974277973175,
160
+ "learning_rate": 8.75e-05,
161
+ "loss": 6.1015,
162
+ "step": 18
163
+ },
164
+ {
165
+ "epoch": 1.1875,
166
+ "grad_norm": 0.8950835466384888,
167
+ "learning_rate": 8.125000000000001e-05,
168
+ "loss": 6.1462,
169
+ "step": 19
170
+ },
171
+ {
172
+ "epoch": 1.25,
173
+ "grad_norm": 1.661649465560913,
174
+ "learning_rate": 7.500000000000001e-05,
175
+ "loss": 6.3825,
176
+ "step": 20
177
+ },
178
+ {
179
+ "epoch": 1.25,
180
+ "eval_clap": 0.11188629269599915,
181
+ "eval_loss": 4.979422092437744,
182
+ "eval_runtime": 516.4364,
183
+ "eval_samples_per_second": 0.015,
184
+ "eval_steps_per_second": 0.015,
185
+ "step": 20
186
+ },
187
+ {
188
+ "epoch": 1.3125,
189
+ "grad_norm": 1.2889989614486694,
190
+ "learning_rate": 6.875e-05,
191
+ "loss": 6.2617,
192
+ "step": 21
193
+ },
194
+ {
195
+ "epoch": 1.375,
196
+ "grad_norm": 0.9493317008018494,
197
+ "learning_rate": 6.25e-05,
198
+ "loss": 5.9844,
199
+ "step": 22
200
+ },
201
+ {
202
+ "epoch": 1.4375,
203
+ "grad_norm": 0.9931596517562866,
204
+ "learning_rate": 5.6250000000000005e-05,
205
+ "loss": 6.2795,
206
+ "step": 23
207
+ },
208
+ {
209
+ "epoch": 1.5,
210
+ "grad_norm": 1.8261293172836304,
211
+ "learning_rate": 5e-05,
212
+ "loss": 5.7055,
213
+ "step": 24
214
+ },
215
+ {
216
+ "epoch": 1.5625,
217
+ "grad_norm": 1.0377460718154907,
218
+ "learning_rate": 4.375e-05,
219
+ "loss": 6.3178,
220
+ "step": 25
221
+ },
222
+ {
223
+ "epoch": 1.5625,
224
+ "eval_clap": 0.09158371388912201,
225
+ "eval_loss": 5.108123779296875,
226
+ "eval_runtime": 516.4832,
227
+ "eval_samples_per_second": 0.015,
228
+ "eval_steps_per_second": 0.015,
229
+ "step": 25
230
+ },
231
+ {
232
+ "epoch": 1.625,
233
+ "grad_norm": 1.5140869617462158,
234
+ "learning_rate": 3.7500000000000003e-05,
235
+ "loss": 6.4832,
236
+ "step": 26
237
+ },
238
+ {
239
+ "epoch": 1.6875,
240
+ "grad_norm": 0.9569382667541504,
241
+ "learning_rate": 3.125e-05,
242
+ "loss": 6.3163,
243
+ "step": 27
244
+ },
245
+ {
246
+ "epoch": 1.75,
247
+ "grad_norm": 1.586517095565796,
248
+ "learning_rate": 2.5e-05,
249
+ "loss": 6.2888,
250
+ "step": 28
251
+ },
252
+ {
253
+ "epoch": 1.8125,
254
+ "grad_norm": 1.238013744354248,
255
+ "learning_rate": 1.8750000000000002e-05,
256
+ "loss": 6.1361,
257
+ "step": 29
258
+ },
259
+ {
260
+ "epoch": 1.875,
261
+ "grad_norm": 1.0142874717712402,
262
+ "learning_rate": 1.25e-05,
263
+ "loss": 6.1301,
264
+ "step": 30
265
+ },
266
+ {
267
+ "epoch": 1.875,
268
+ "eval_clap": 0.08499659597873688,
269
+ "eval_loss": 5.103597640991211,
270
+ "eval_runtime": 424.9013,
271
+ "eval_samples_per_second": 0.019,
272
+ "eval_steps_per_second": 0.019,
273
+ "step": 30
274
+ },
275
+ {
276
+ "epoch": 1.9375,
277
+ "grad_norm": 0.9664702415466309,
278
+ "learning_rate": 6.25e-06,
279
+ "loss": 6.3296,
280
+ "step": 31
281
+ },
282
+ {
283
+ "epoch": 2.0,
284
+ "grad_norm": 1.8057743310928345,
285
+ "learning_rate": 0.0,
286
+ "loss": 6.544,
287
+ "step": 32
288
+ },
289
+ {
290
+ "epoch": 2.0,
291
+ "step": 32,
292
+ "total_flos": 152209974742572.0,
293
+ "train_loss": 6.930231094360352,
294
+ "train_runtime": 3130.5046,
295
+ "train_samples_per_second": 0.077,
296
+ "train_steps_per_second": 0.01
297
+ }
298
+ ],
299
+ "logging_steps": 1.0,
300
+ "max_steps": 32,
301
+ "num_input_tokens_seen": 0,
302
+ "num_train_epochs": 2,
303
+ "save_steps": 500,
304
+ "stateful_callbacks": {
305
+ "TrainerControl": {
306
+ "args": {
307
+ "should_epoch_stop": false,
308
+ "should_evaluate": false,
309
+ "should_log": false,
310
+ "should_save": false,
311
+ "should_training_stop": false
312
+ },
313
+ "attributes": {}
314
+ }
315
+ },
316
+ "total_flos": 152209974742572.0,
317
+ "train_batch_size": 8,
318
+ "trial_name": null,
319
+ "trial_params": null
320
+ }