End of training

README.md

@@ -1,5 +1,8 @@
 ---
+base_model: ylacombe/musicgen-melody
 tags:
+- text-to-audio
+- ylacombe/bella_ciao
 - generated_from_trainer
 model-index:
 - name: musicgen-melody-bella-ciao
@@ -11,7 +14,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 an unknown dataset.
+This model is a fine-tuned version of [ylacombe/musicgen-melody](https://huggingface.co/ylacombe/musicgen-melody) on the YLACOMBE/BELLA_CIAO - DEFAULT dataset.
 
 ## Model description
 

all_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 20.0,
-    "train_loss": 7.698295211791992,
-    "train_runtime": 141.1997,
+    "train_loss": 7.524070739746094,
+    "train_runtime": 153.2004,
     "train_samples": 10,
-    "train_samples_per_second": 1.416,
-    "train_steps_per_second": 1.416
+    "train_samples_per_second": 1.305,
+    "train_steps_per_second": 1.305
 }

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 20.0,
-    "train_loss": 7.698295211791992,
-    "train_runtime": 141.1997,
+    "train_loss": 7.524070739746094,
+    "train_runtime": 153.2004,
     "train_samples": 10,
-    "train_samples_per_second": 1.416,
-    "train_steps_per_second": 1.416
+    "train_samples_per_second": 1.305,
+    "train_steps_per_second": 1.305
 }

trainer_state.json

@@ -10,60 +10,60 @@
   "log_history": [
     {
       "epoch": 2.5,
-      "learning_rate": 4.3999999999999997e-07,
-      "loss": 9.1366,
+      "learning_rate": 4.425e-07,
+      "loss": 9.0808,
       "step": 25
     },
     {
       "epoch": 5.0,
-      "learning_rate": 3.775e-07,
-      "loss": 8.1903,
+      "learning_rate": 3.7999999999999996e-07,
+      "loss": 8.0507,
       "step": 50
     },
     {
       "epoch": 7.5,
-      "learning_rate": 3.15e-07,
-      "loss": 7.7121,
+      "learning_rate": 3.175e-07,
+      "loss": 7.5948,
       "step": 75
     },
     {
       "epoch": 10.0,
-      "learning_rate": 2.5249999999999996e-07,
-      "loss": 7.4457,
+      "learning_rate": 2.55e-07,
+      "loss": 7.2855,
       "step": 100
     },
     {
       "epoch": 12.5,
       "learning_rate": 1.9249999999999998e-07,
-      "loss": 7.3392,
+      "loss": 7.1565,
       "step": 125
     },
     {
       "epoch": 15.0,
       "learning_rate": 1.3e-07,
-      "loss": 7.2883,
+      "loss": 7.0459,
       "step": 150
     },
     {
       "epoch": 17.5,
       "learning_rate": 6.75e-08,
-      "loss": 7.2359,
+      "loss": 7.0015,
       "step": 175
     },
     {
       "epoch": 20.0,
       "learning_rate": 5e-09,
-      "loss": 7.2382,
+      "loss": 6.9769,
       "step": 200
     },
     {
       "epoch": 20.0,
       "step": 200,
       "total_flos": 0.0,
-      "train_loss": 7.698295211791992,
-      "train_runtime": 141.1997,
-      "train_samples_per_second": 1.416,
-      "train_steps_per_second": 1.416
+      "train_loss": 7.524070739746094,
+      "train_runtime": 153.2004,
+      "train_samples_per_second": 1.305,
+      "train_steps_per_second": 1.305
     }
   ],
   "logging_steps": 25,