End of training

README.md

@@ -2,6 +2,8 @@
 license: cc-by-nc-4.0
 library_name: peft
 tags:
+- text-to-audio
+- techno_nan
 - generated_from_trainer
 base_model: facebook/musicgen-melody
 model-index:
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # musicgen-melody-lora-techno-nan-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 xiaosh9527/nan_music dataset.
 
 ## Model description
 

trainer_state.json

@@ -10,96 +10,96 @@
   "log_history": [
     {
       "epoch": 0.32,
-      "grad_norm": 1.016900658607483,
-      "learning_rate": 0.00018333333333333334,
-      "loss": 9.5861,
+      "grad_norm": 0.20171859860420227,
+      "learning_rate": 9.166666666666667e-05,
+      "loss": 6.7889,
       "step": 2
     },
     {
       "epoch": 0.64,
-      "grad_norm": 1.7372517585754395,
-      "learning_rate": 0.0001666666666666667,
-      "loss": 9.2056,
+      "grad_norm": 0.2258663922548294,
+      "learning_rate": 8.333333333333334e-05,
+      "loss": 6.8518,
       "step": 4
     },
     {
       "epoch": 0.96,
-      "grad_norm": 2.0075902938842773,
-      "learning_rate": 0.00015000000000000001,
-      "loss": 8.5072,
+      "grad_norm": 0.2266591638326645,
+      "learning_rate": 7.500000000000001e-05,
+      "loss": 6.8562,
       "step": 6
     },
     {
       "epoch": 1.28,
-      "grad_norm": 2.0754647254943848,
-      "learning_rate": 0.00013333333333333334,
-      "loss": 8.0374,
+      "grad_norm": 0.19837072491645813,
+      "learning_rate": 6.666666666666667e-05,
+      "loss": 6.8282,
       "step": 8
     },
     {
       "epoch": 1.6,
-      "grad_norm": 1.2964591979980469,
-      "learning_rate": 0.00011666666666666668,
-      "loss": 7.6874,
+      "grad_norm": 0.17619071900844574,
+      "learning_rate": 5.833333333333334e-05,
+      "loss": 6.8528,
       "step": 10
     },
     {
       "epoch": 1.92,
-      "grad_norm": 0.9893752336502075,
-      "learning_rate": 0.0001,
-      "loss": 7.4668,
+      "grad_norm": 0.12639087438583374,
+      "learning_rate": 5e-05,
+      "loss": 6.8231,
       "step": 12
     },
     {
       "epoch": 2.24,
-      "grad_norm": 0.9852508306503296,
-      "learning_rate": 8.333333333333334e-05,
-      "loss": 7.4002,
+      "grad_norm": 0.13687172532081604,
+      "learning_rate": 4.166666666666667e-05,
+      "loss": 6.7815,
       "step": 14
     },
     {
       "epoch": 2.56,
-      "grad_norm": 0.9375394582748413,
-      "learning_rate": 6.666666666666667e-05,
-      "loss": 7.3329,
+      "grad_norm": 0.18898847699165344,
+      "learning_rate": 3.3333333333333335e-05,
+      "loss": 6.7792,
       "step": 16
     },
     {
       "epoch": 2.88,
-      "grad_norm": 0.8562221527099609,
-      "learning_rate": 5e-05,
-      "loss": 7.2785,
+      "grad_norm": 0.1648869514465332,
+      "learning_rate": 2.5e-05,
+      "loss": 6.8046,
       "step": 18
     },
     {
       "epoch": 3.2,
-      "grad_norm": 0.663221538066864,
-      "learning_rate": 3.3333333333333335e-05,
-      "loss": 7.2907,
+      "grad_norm": 0.16822992265224457,
+      "learning_rate": 1.6666666666666667e-05,
+      "loss": 6.8128,
       "step": 20
     },
     {
       "epoch": 3.52,
-      "grad_norm": 1.024126648902893,
-      "learning_rate": 1.6666666666666667e-05,
-      "loss": 7.2383,
+      "grad_norm": 0.16933274269104004,
+      "learning_rate": 8.333333333333334e-06,
+      "loss": 6.8081,
       "step": 22
     },
     {
       "epoch": 3.84,
-      "grad_norm": 0.7755734920501709,
+      "grad_norm": 0.1729530692100525,
       "learning_rate": 0.0,
-      "loss": 7.2692,
+      "loss": 6.8425,
       "step": 24
     },
     {
       "epoch": 3.84,
       "step": 24,
       "total_flos": 92571267829680.0,
-      "train_loss": 7.858357032140096,
-      "train_runtime": 664.0524,
-      "train_samples_per_second": 0.602,
-      "train_steps_per_second": 0.036
+      "train_loss": 6.819143931070964,
+      "train_runtime": 117.2771,
+      "train_samples_per_second": 3.411,
+      "train_steps_per_second": 0.205
     }
   ],
   "logging_steps": 2,