End of training

README.md

@@ -3,6 +3,8 @@ base_model: facebook/musicgen-large
 library_name: peft
 license: cc-by-nc-4.0
 tags:
+- text-to-audio
+- acid_data
 - generated_from_trainer
 model-index:
 - name: musicgen-large-lora-acid_techno_2ep_20da_v4-colab
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # musicgen-large-lora-acid_techno_2ep_20da_v4-colab
 
-This model is a fine-tuned version of [facebook/musicgen-large](https://huggingface.co/facebook/musicgen-large) on an unknown dataset.
+This model is a fine-tuned version of [facebook/musicgen-large](https://huggingface.co/facebook/musicgen-large) on the ShaySha/acid_data dataset.
 
 ## Model description
 

trainer_state.json

@@ -10,26 +10,26 @@
   "log_history": [
     {
       "epoch": 0.8,
-      "grad_norm": 0.9572145938873291,
+      "grad_norm": 1.5954439640045166,
       "learning_rate": 0.0001,
-      "loss": 8.9787,
+      "loss": 8.9222,
       "step": 1
     },
     {
       "epoch": 1.0,
-      "grad_norm": 0.2813553214073181,
+      "grad_norm": 0.6375588178634644,
       "learning_rate": 0.0,
-      "loss": 2.2209,
+      "loss": 2.1991,
       "step": 2
     },
     {
       "epoch": 1.0,
       "step": 2,
       "total_flos": 3290561496960.0,
-      "train_loss": 5.599770545959473,
-      "train_runtime": 11.2689,
-      "train_samples_per_second": 3.55,
-      "train_steps_per_second": 0.177
+      "train_loss": 5.560663461685181,
+      "train_runtime": 11.683,
+      "train_samples_per_second": 3.424,
+      "train_steps_per_second": 0.171
     }
   ],
   "logging_steps": 1,