End of training

README.md

@@ -3,6 +3,8 @@ library_name: peft
 license: cc-by-nc-4.0
 base_model: facebook/musicgen-melody
 tags:
+- text-to-audio
+- tiny-punk
 - generated_from_trainer
 model-index:
 - name: musicgen-electronic-colab
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # musicgen-electronic-colab
 
-This model is a fine-tuned version of [facebook/musicgen-melody](https://huggingface.co/facebook/musicgen-melody) on an unknown dataset.
+This model is a fine-tuned version of [facebook/musicgen-melody](https://huggingface.co/facebook/musicgen-melody) on the PlutoG99001/MusicGen-Electronic-Clean dataset.
 
 ## Model description
 

trainer_state.json

@@ -9,55 +9,55 @@
   "is_world_process_zero": true,
   "log_history": [
     {
-      "epoch": 0.6153846153846154,
-      "grad_norm": NaN,
-      "learning_rate": 0.00018333333333333334,
-      "loss": 76.0981,
+      "epoch": 0.64,
+      "grad_norm": 8.356424331665039,
+      "learning_rate": 0.0001666666666666667,
+      "loss": 76.2569,
       "step": 2
     },
     {
       "epoch": 1.0,
-      "grad_norm": 2.3868422508239746,
-      "learning_rate": 0.00015000000000000001,
-      "loss": 47.3904,
+      "grad_norm": 1.7912710905075073,
+      "learning_rate": 0.00013333333333333334,
+      "loss": 41.8415,
       "step": 4
     },
     {
-      "epoch": 1.6153846153846154,
-      "grad_norm": 14.68977165222168,
-      "learning_rate": 0.00011666666666666668,
-      "loss": 72.3613,
+      "epoch": 1.6400000000000001,
+      "grad_norm": 16.31438636779785,
+      "learning_rate": 0.0001,
+      "loss": 70.2302,
       "step": 6
     },
     {
       "epoch": 2.0,
-      "grad_norm": 5.613176345825195,
-      "learning_rate": 8.333333333333334e-05,
-      "loss": 43.5572,
+      "grad_norm": 2.897549629211426,
+      "learning_rate": 6.666666666666667e-05,
+      "loss": 38.4848,
       "step": 8
     },
     {
-      "epoch": 2.6153846153846154,
-      "grad_norm": 17.510847091674805,
-      "learning_rate": 5e-05,
-      "loss": 66.3821,
+      "epoch": 2.64,
+      "grad_norm": 15.37080192565918,
+      "learning_rate": 3.3333333333333335e-05,
+      "loss": 65.8869,
       "step": 10
     },
     {
       "epoch": 3.0,
-      "grad_norm": 4.0115838050842285,
-      "learning_rate": 1.6666666666666667e-05,
-      "loss": 40.0605,
+      "grad_norm": 3.5507168769836426,
+      "learning_rate": 0.0,
+      "loss": 36.2084,
       "step": 12
     },
     {
       "epoch": 3.0,
       "step": 12,
-      "total_flos": 37471962307104.0,
-      "train_loss": 57.6416072845459,
-      "train_runtime": 73.9506,
-      "train_samples_per_second": 2.813,
-      "train_steps_per_second": 0.162
+      "total_flos": 40557671234760.0,
+      "train_loss": 54.81812349955241,
+      "train_runtime": 48.7104,
+      "train_samples_per_second": 4.106,
+      "train_steps_per_second": 0.246
     }
   ],
   "logging_steps": 2,
@@ -77,7 +77,7 @@
       "attributes": {}
     }
   },
-  "total_flos": 37471962307104.0,
+  "total_flos": 40557671234760.0,
   "train_batch_size": 2,
   "trial_name": null,
   "trial_params": null