End of training

README.md

@@ -2,6 +2,8 @@
 license: cc-by-nc-4.0
 library_name: peft
 tags:
+- text-to-audio
+- ylacombe/tiny-punk
 - generated_from_trainer
 base_model: facebook/musicgen-melody
 model-index:
@@ -14,7 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # musicgen-melody-punk-lora
 
-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 YLACOMBE/TINY-PUNK - DEFAULT dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.2634
+- Clap: nan
 
 ## Model description
 

all_results.json

@@ -1,15 +1,15 @@
 {
     "epoch": 3.764705882352941,
-    "eval_clap": 0.3365415632724762,
-    "eval_loss": 4.704524993896484,
-    "eval_runtime": 147.6762,
+    "eval_clap": NaN,
+    "eval_loss": 0.2634366452693939,
+    "eval_runtime": 177.4407,
     "eval_samples": 5,
-    "eval_samples_per_second": 0.034,
-    "eval_steps_per_second": 0.034,
-    "total_flos": 59875624258368.0,
-    "train_loss": 8.819526672363281,
-    "train_runtime": 269.5748,
+    "eval_samples_per_second": 0.028,
+    "eval_steps_per_second": 0.028,
+    "total_flos": 35328595626816.0,
+    "train_loss": 6.771849155426025,
+    "train_runtime": 276.9653,
     "train_samples": 33,
-    "train_samples_per_second": 0.49,
-    "train_steps_per_second": 0.03
+    "train_samples_per_second": 0.477,
+    "train_steps_per_second": 0.029
 }

eval_results.json

@@ -1,9 +1,9 @@
 {
     "epoch": 3.764705882352941,
-    "eval_clap": 0.3365415632724762,
-    "eval_loss": 4.704524993896484,
-    "eval_runtime": 147.6762,
+    "eval_clap": NaN,
+    "eval_loss": 0.2634366452693939,
+    "eval_runtime": 177.4407,
     "eval_samples": 5,
-    "eval_samples_per_second": 0.034,
-    "eval_steps_per_second": 0.034
+    "eval_samples_per_second": 0.028,
+    "eval_steps_per_second": 0.028
 }

train_results.json

@@ -1,9 +1,9 @@
 {
     "epoch": 3.764705882352941,
-    "total_flos": 59875624258368.0,
-    "train_loss": 8.819526672363281,
-    "train_runtime": 269.5748,
+    "total_flos": 35328595626816.0,
+    "train_loss": 6.771849155426025,
+    "train_runtime": 276.9653,
     "train_samples": 33,
-    "train_samples_per_second": 0.49,
-    "train_steps_per_second": 0.03
+    "train_samples_per_second": 0.477,
+    "train_steps_per_second": 0.029
 }

trainer_state.json

@@ -11,11 +11,11 @@
     {
       "epoch": 3.764705882352941,
       "step": 8,
-      "total_flos": 59875624258368.0,
-      "train_loss": 8.819526672363281,
-      "train_runtime": 269.5748,
-      "train_samples_per_second": 0.49,
-      "train_steps_per_second": 0.03
+      "total_flos": 35328595626816.0,
+      "train_loss": 6.771849155426025,
+      "train_runtime": 276.9653,
+      "train_samples_per_second": 0.477,
+      "train_steps_per_second": 0.029
     }
   ],
   "logging_steps": 25,
@@ -23,7 +23,7 @@
   "num_input_tokens_seen": 0,
   "num_train_epochs": 4,
   "save_steps": 500,
-  "total_flos": 59875624258368.0,
+  "total_flos": 35328595626816.0,
   "train_batch_size": 2,
   "trial_name": null,
   "trial_params": null