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": 10.0,
-    "train_loss": 7.140205179850261,
-    "train_runtime": 114.8228,
+    "epoch": 25.0,
+    "train_loss": 7.2796097819010415,
+    "train_runtime": 437.882,
     "train_samples": 30,
-    "train_samples_per_second": 2.613,
-    "train_steps_per_second": 1.306
+    "train_samples_per_second": 1.713,
+    "train_steps_per_second": 0.856
 }

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 10.0,
-    "train_loss": 7.140205179850261,
-    "train_runtime": 114.8228,
+    "epoch": 25.0,
+    "train_loss": 7.2796097819010415,
+    "train_runtime": 437.882,
     "train_samples": 30,
-    "train_samples_per_second": 2.613,
-    "train_steps_per_second": 1.306
+    "train_samples_per_second": 1.713,
+    "train_steps_per_second": 0.856
 }

trainer_state.json

@@ -1,63 +1,117 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 10.0,
+  "epoch": 25.0,
   "eval_steps": 1000,
-  "global_step": 150,
+  "global_step": 375,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
       "epoch": 1.67,
-      "learning_rate": 8.333333333333334e-06,
-      "loss": 7.5038,
+      "learning_rate": 9.36e-07,
+      "loss": 8.5683,
       "step": 25
     },
     {
       "epoch": 3.33,
-      "learning_rate": 6.666666666666667e-06,
-      "loss": 7.1421,
+      "learning_rate": 8.693333333333333e-07,
+      "loss": 7.5028,
       "step": 50
     },
     {
       "epoch": 5.0,
-      "learning_rate": 5e-06,
-      "loss": 7.1034,
+      "learning_rate": 8.026666666666667e-07,
+      "loss": 7.293,
       "step": 75
     },
     {
       "epoch": 6.67,
-      "learning_rate": 3.3333333333333333e-06,
-      "loss": 7.0666,
+      "learning_rate": 7.359999999999999e-07,
+      "loss": 7.2316,
       "step": 100
     },
     {
       "epoch": 8.33,
-      "learning_rate": 1.6666666666666667e-06,
-      "loss": 7.0238,
+      "learning_rate": 6.693333333333333e-07,
+      "loss": 7.1945,
       "step": 125
     },
     {
       "epoch": 10.0,
-      "learning_rate": 0.0,
-      "loss": 7.0015,
+      "learning_rate": 6.026666666666667e-07,
+      "loss": 7.178,
       "step": 150
     },
     {
-      "epoch": 10.0,
-      "step": 150,
+      "epoch": 11.67,
+      "learning_rate": 5.36e-07,
+      "loss": 7.176,
+      "step": 175
+    },
+    {
+      "epoch": 13.33,
+      "learning_rate": 4.693333333333333e-07,
+      "loss": 7.1478,
+      "step": 200
+    },
+    {
+      "epoch": 15.0,
+      "learning_rate": 4.0799999999999995e-07,
+      "loss": 7.1358,
+      "step": 225
+    },
+    {
+      "epoch": 16.67,
+      "learning_rate": 3.413333333333333e-07,
+      "loss": 7.154,
+      "step": 250
+    },
+    {
+      "epoch": 18.33,
+      "learning_rate": 2.7466666666666664e-07,
+      "loss": 7.1035,
+      "step": 275
+    },
+    {
+      "epoch": 20.0,
+      "learning_rate": 2.0799999999999998e-07,
+      "loss": 7.1441,
+      "step": 300
+    },
+    {
+      "epoch": 21.67,
+      "learning_rate": 1.4133333333333333e-07,
+      "loss": 7.1239,
+      "step": 325
+    },
+    {
+      "epoch": 23.33,
+      "learning_rate": 7.466666666666667e-08,
+      "loss": 7.1274,
+      "step": 350
+    },
+    {
+      "epoch": 25.0,
+      "learning_rate": 8e-09,
+      "loss": 7.1136,
+      "step": 375
+    },
+    {
+      "epoch": 25.0,
+      "step": 375,
       "total_flos": 0.0,
-      "train_loss": 7.140205179850261,
-      "train_runtime": 114.8228,
-      "train_samples_per_second": 2.613,
-      "train_steps_per_second": 1.306
+      "train_loss": 7.2796097819010415,
+      "train_runtime": 437.882,
+      "train_samples_per_second": 1.713,
+      "train_steps_per_second": 0.856
     }
   ],
   "logging_steps": 25,
-  "max_steps": 150,
+  "max_steps": 375,
   "num_input_tokens_seen": 0,
-  "num_train_epochs": 10,
+  "num_train_epochs": 25,
   "save_steps": 500,
   "total_flos": 0.0,
   "train_batch_size": 2,