End of training

README.md

@@ -1,5 +1,8 @@
 ---
+base_model: ylacombe/musicgen-melody
 tags:
+- text-to-audio
+- patrickvonplaten/bella_ciao
 - generated_from_trainer
 datasets:
 - audiofolder
@@ -13,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # musicgen-melody-bella-ciao
 
-This model is a fine-tuned version of [](https://huggingface.co/) on the audiofolder dataset.
+This model is a fine-tuned version of [ylacombe/musicgen-melody](https://huggingface.co/ylacombe/musicgen-melody) on the PATRICKVONPLATEN/BELLA_CIAO - DEFAULT dataset.
 
 ## Model description
 

all_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 30.0,
-    "train_loss": 6.948778754340278,
-    "train_runtime": 493.0316,
+    "epoch": 10.0,
+    "train_loss": 7.3888854471842444,
+    "train_runtime": 149.4092,
     "train_samples": 30,
-    "train_samples_per_second": 1.825,
-    "train_steps_per_second": 1.825
+    "train_samples_per_second": 2.008,
+    "train_steps_per_second": 2.008
 }

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 30.0,
-    "train_loss": 6.948778754340278,
-    "train_runtime": 493.0316,
+    "epoch": 10.0,
+    "train_loss": 7.3888854471842444,
+    "train_runtime": 149.4092,
     "train_samples": 30,
-    "train_samples_per_second": 1.825,
-    "train_steps_per_second": 1.825
+    "train_samples_per_second": 2.008,
+    "train_steps_per_second": 2.008
 }

trainer_state.json

@@ -1,243 +1,99 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 30.0,
+  "epoch": 10.0,
   "eval_steps": 1000,
-  "global_step": 900,
+  "global_step": 300,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
       "epoch": 0.83,
-      "learning_rate": 4.861111111111111e-06,
-      "loss": 7.7904,
+      "learning_rate": 9.2e-07,
+      "loss": 8.7857,
       "step": 25
     },
     {
       "epoch": 1.67,
-      "learning_rate": 4.722222222222222e-06,
-      "loss": 7.224,
+      "learning_rate": 8.366666666666667e-07,
+      "loss": 7.6115,
       "step": 50
     },
     {
       "epoch": 2.5,
-      "learning_rate": 4.583333333333333e-06,
-      "loss": 7.1605,
+      "learning_rate": 7.533333333333332e-07,
+      "loss": 7.3706,
       "step": 75
     },
     {
       "epoch": 3.33,
-      "learning_rate": 4.444444444444444e-06,
-      "loss": 7.162,
+      "learning_rate": 6.733333333333333e-07,
+      "loss": 7.2883,
       "step": 100
     },
     {
       "epoch": 4.17,
-      "learning_rate": 4.305555555555556e-06,
-      "loss": 7.1535,
+      "learning_rate": 5.9e-07,
+      "loss": 7.2673,
       "step": 125
     },
     {
       "epoch": 5.0,
-      "learning_rate": 4.166666666666667e-06,
-      "loss": 7.1078,
+      "learning_rate": 5.066666666666667e-07,
+      "loss": 7.2174,
       "step": 150
     },
     {
       "epoch": 5.83,
-      "learning_rate": 4.027777777777779e-06,
-      "loss": 7.0931,
+      "learning_rate": 4.2333333333333334e-07,
+      "loss": 7.2089,
       "step": 175
     },
     {
       "epoch": 6.67,
-      "learning_rate": 3.894444444444444e-06,
-      "loss": 7.0935,
+      "learning_rate": 3.4000000000000003e-07,
+      "loss": 7.1989,
       "step": 200
     },
     {
       "epoch": 7.5,
-      "learning_rate": 3.7555555555555557e-06,
-      "loss": 7.0416,
+      "learning_rate": 2.5666666666666666e-07,
+      "loss": 7.1801,
       "step": 225
     },
     {
       "epoch": 8.33,
-      "learning_rate": 3.616666666666667e-06,
-      "loss": 7.059,
+      "learning_rate": 1.7333333333333332e-07,
+      "loss": 7.1842,
       "step": 250
     },
     {
       "epoch": 9.17,
-      "learning_rate": 3.4777777777777784e-06,
-      "loss": 7.0239,
+      "learning_rate": 9e-08,
+      "loss": 7.1852,
       "step": 275
     },
     {
       "epoch": 10.0,
-      "learning_rate": 3.3388888888888893e-06,
-      "loss": 7.0133,
+      "learning_rate": 1e-08,
+      "loss": 7.1686,
       "step": 300
     },
     {
-      "epoch": 10.83,
-      "learning_rate": 3.2000000000000003e-06,
-      "loss": 7.0166,
-      "step": 325
-    },
-    {
-      "epoch": 11.67,
-      "learning_rate": 3.0611111111111112e-06,
-      "loss": 6.9859,
-      "step": 350
-    },
-    {
-      "epoch": 12.5,
-      "learning_rate": 2.9222222222222226e-06,
-      "loss": 6.9105,
-      "step": 375
-    },
-    {
-      "epoch": 13.33,
-      "learning_rate": 2.7833333333333335e-06,
-      "loss": 6.9786,
-      "step": 400
-    },
-    {
-      "epoch": 14.17,
-      "learning_rate": 2.6444444444444444e-06,
-      "loss": 6.8962,
-      "step": 425
-    },
-    {
-      "epoch": 15.0,
-      "learning_rate": 2.5055555555555554e-06,
-      "loss": 6.9087,
-      "step": 450
-    },
-    {
-      "epoch": 15.83,
-      "learning_rate": 2.3666666666666667e-06,
-      "loss": 6.8853,
-      "step": 475
-    },
-    {
-      "epoch": 16.67,
-      "learning_rate": 2.2277777777777777e-06,
-      "loss": 6.908,
-      "step": 500
-    },
-    {
-      "epoch": 17.5,
-      "learning_rate": 2.088888888888889e-06,
-      "loss": 6.8508,
-      "step": 525
-    },
-    {
-      "epoch": 18.33,
-      "learning_rate": 1.9500000000000004e-06,
-      "loss": 6.7925,
-      "step": 550
-    },
-    {
-      "epoch": 19.17,
-      "learning_rate": 1.8111111111111113e-06,
-      "loss": 6.9076,
-      "step": 575
-    },
-    {
-      "epoch": 20.0,
-      "learning_rate": 1.6722222222222223e-06,
-      "loss": 6.8269,
-      "step": 600
-    },
-    {
-      "epoch": 20.83,
-      "learning_rate": 1.5333333333333334e-06,
-      "loss": 6.8281,
-      "step": 625
-    },
-    {
-      "epoch": 21.67,
-      "learning_rate": 1.3944444444444446e-06,
-      "loss": 6.8085,
-      "step": 650
-    },
-    {
-      "epoch": 22.5,
-      "learning_rate": 1.2555555555555557e-06,
-      "loss": 6.8046,
-      "step": 675
-    },
-    {
-      "epoch": 23.33,
-      "learning_rate": 1.1166666666666666e-06,
-      "loss": 6.8158,
-      "step": 700
-    },
-    {
-      "epoch": 24.17,
-      "learning_rate": 9.77777777777778e-07,
-      "loss": 6.7809,
-      "step": 725
-    },
-    {
-      "epoch": 25.0,
-      "learning_rate": 8.388888888888889e-07,
-      "loss": 6.7924,
-      "step": 750
-    },
-    {
-      "epoch": 25.83,
-      "learning_rate": 7.000000000000001e-07,
-      "loss": 6.7673,
-      "step": 775
-    },
-    {
-      "epoch": 26.67,
-      "learning_rate": 5.611111111111111e-07,
-      "loss": 6.7454,
-      "step": 800
-    },
-    {
-      "epoch": 27.5,
-      "learning_rate": 4.2222222222222226e-07,
-      "loss": 6.7967,
-      "step": 825
-    },
-    {
-      "epoch": 28.33,
-      "learning_rate": 2.8333333333333336e-07,
-      "loss": 6.7509,
-      "step": 850
-    },
-    {
-      "epoch": 29.17,
-      "learning_rate": 1.4444444444444445e-07,
-      "loss": 6.7423,
-      "step": 875
-    },
-    {
-      "epoch": 30.0,
-      "learning_rate": 5.555555555555556e-09,
-      "loss": 6.7329,
-      "step": 900
-    },
-    {
-      "epoch": 30.0,
-      "step": 900,
+      "epoch": 10.0,
+      "step": 300,
       "total_flos": 0.0,
-      "train_loss": 6.948778754340278,
-      "train_runtime": 493.0316,
-      "train_samples_per_second": 1.825,
-      "train_steps_per_second": 1.825
+      "train_loss": 7.3888854471842444,
+      "train_runtime": 149.4092,
+      "train_samples_per_second": 2.008,
+      "train_steps_per_second": 2.008
     }
   ],
   "logging_steps": 25,
-  "max_steps": 900,
+  "max_steps": 300,
   "num_input_tokens_seen": 0,
-  "num_train_epochs": 30,
+  "num_train_epochs": 10,
   "save_steps": 500,
   "total_flos": 0.0,
   "train_batch_size": 1,