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: 4.7288
+- Clap: 0.2611
 
 ## Model description
 

all_results.json

@@ -1,15 +1,15 @@
 {
     "epoch": 3.764705882352941,
-    "eval_clap": 0.3319118022918701,
-    "eval_loss": 4.704505920410156,
-    "eval_runtime": 149.2288,
+    "eval_clap": 0.261121928691864,
+    "eval_loss": 4.728769302368164,
+    "eval_runtime": 140.567,
     "eval_samples": 5,
-    "eval_samples_per_second": 0.034,
-    "eval_steps_per_second": 0.034,
-    "total_flos": 59875624258368.0,
-    "train_loss": 8.819560050964355,
-    "train_runtime": 362.7337,
+    "eval_samples_per_second": 0.036,
+    "eval_steps_per_second": 0.036,
+    "total_flos": 35328595626816.0,
+    "train_loss": 8.988505363464355,
+    "train_runtime": 310.1402,
     "train_samples": 33,
-    "train_samples_per_second": 0.364,
-    "train_steps_per_second": 0.022
+    "train_samples_per_second": 0.426,
+    "train_steps_per_second": 0.026
 }

eval_results.json

@@ -1,9 +1,9 @@
 {
     "epoch": 3.764705882352941,
-    "eval_clap": 0.3319118022918701,
-    "eval_loss": 4.704505920410156,
-    "eval_runtime": 149.2288,
+    "eval_clap": 0.261121928691864,
+    "eval_loss": 4.728769302368164,
+    "eval_runtime": 140.567,
     "eval_samples": 5,
-    "eval_samples_per_second": 0.034,
-    "eval_steps_per_second": 0.034
+    "eval_samples_per_second": 0.036,
+    "eval_steps_per_second": 0.036
 }

train_results.json

@@ -1,9 +1,9 @@
 {
     "epoch": 3.764705882352941,
-    "total_flos": 59875624258368.0,
-    "train_loss": 8.819560050964355,
-    "train_runtime": 362.7337,
+    "total_flos": 35328595626816.0,
+    "train_loss": 8.988505363464355,
+    "train_runtime": 310.1402,
     "train_samples": 33,
-    "train_samples_per_second": 0.364,
-    "train_steps_per_second": 0.022
+    "train_samples_per_second": 0.426,
+    "train_steps_per_second": 0.026
 }

trainer_state.json

@@ -11,11 +11,11 @@
     {
       "epoch": 3.764705882352941,
       "step": 8,
-      "total_flos": 59875624258368.0,
-      "train_loss": 8.819560050964355,
-      "train_runtime": 362.7337,
-      "train_samples_per_second": 0.364,
-      "train_steps_per_second": 0.022
+      "total_flos": 35328595626816.0,
+      "train_loss": 8.988505363464355,
+      "train_runtime": 310.1402,
+      "train_samples_per_second": 0.426,
+      "train_steps_per_second": 0.026
     }
   ],
   "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