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
+- dnb
 - generated_from_trainer
 model-index:
 - name: musicgen-melody-lora-dnb-colab-gs-3-lr-0.0001
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # musicgen-melody-lora-dnb-colab-gs-3-lr-0.0001
 
-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 istvanj/ncs-dnb dataset.
 
 ## Model description
 

trainer_state.json

@@ -10,82 +10,82 @@
   "log_history": [
     {
       "epoch": 0.26666666666666666,
-      "grad_norm": 8.077447891235352,
+      "grad_norm": 7.614686012268066,
       "learning_rate": 9.047619047619048e-05,
-      "loss": 9.6711,
+      "loss": 9.6721,
       "step": 2
     },
     {
       "epoch": 0.5333333333333333,
-      "grad_norm": 10.438846588134766,
+      "grad_norm": 10.107391357421875,
       "learning_rate": 8.095238095238096e-05,
-      "loss": 9.4926,
+      "loss": 9.4969,
       "step": 4
     },
     {
       "epoch": 0.8,
-      "grad_norm": 11.99673080444336,
+      "grad_norm": 11.758378028869629,
       "learning_rate": 7.142857142857143e-05,
-      "loss": 9.2985,
+      "loss": 9.3036,
       "step": 6
     },
     {
       "epoch": 1.0,
-      "grad_norm": 8.44992733001709,
+      "grad_norm": 8.324995994567871,
       "learning_rate": 6.19047619047619e-05,
-      "loss": 6.8146,
+      "loss": 6.8191,
       "step": 8
     },
     {
       "epoch": 1.2666666666666666,
-      "grad_norm": 15.83218002319336,
+      "grad_norm": 15.543440818786621,
       "learning_rate": 5.2380952380952384e-05,
-      "loss": 8.8302,
+      "loss": 8.8411,
       "step": 10
     },
     {
       "epoch": 1.5333333333333332,
-      "grad_norm": 16.710369110107422,
-      "learning_rate": 4.761904761904762e-05,
-      "loss": 8.5759,
+      "grad_norm": 16.783220291137695,
+      "learning_rate": 4.2857142857142856e-05,
+      "loss": 8.5329,
       "step": 12
     },
     {
       "epoch": 1.8,
-      "grad_norm": 19.962966918945312,
-      "learning_rate": 3.809523809523809e-05,
-      "loss": 8.3464,
+      "grad_norm": 19.108684539794922,
+      "learning_rate": 3.3333333333333335e-05,
+      "loss": 8.2567,
       "step": 14
     },
     {
       "epoch": 2.0,
-      "grad_norm": 9.256256103515625,
-      "learning_rate": 2.857142857142857e-05,
-      "loss": 6.1951,
+      "grad_norm": 8.789897918701172,
+      "learning_rate": 2.380952380952381e-05,
+      "loss": 6.1492,
       "step": 16
     },
     {
       "epoch": 2.2666666666666666,
-      "grad_norm": 18.415756225585938,
-      "learning_rate": 1.9047619047619046e-05,
-      "loss": 8.0719,
+      "grad_norm": 17.98725128173828,
+      "learning_rate": 1.4285714285714285e-05,
+      "loss": 8.0423,
       "step": 18
     },
     {
       "epoch": 2.533333333333333,
-      "grad_norm": 14.65276050567627,
-      "learning_rate": 9.523809523809523e-06,
-      "loss": 7.9604,
+      "grad_norm": 14.338889122009277,
+      "learning_rate": 4.7619047619047615e-06,
+      "loss": 7.9496,
       "step": 20
     },
     {
       "epoch": 2.6666666666666665,
       "step": 21,
       "total_flos": 82187025809664.0,
-      "train_loss": 8.305864515758696,
-      "train_runtime": 507.9977,
-      "train_samples_per_second": 0.709,
-      "train_steps_per_second": 0.041
+      "train_loss": 8.28742908296131,
+      "train_runtime": 577.7205,
+      "train_samples_per_second": 0.623,
+      "train_steps_per_second": 0.036
     }
   ],
   "logging_steps": 2,