End of training

README.md

@@ -3,6 +3,8 @@ base_model: facebook/musicgen-melody
 library_name: peft
 license: cc-by-nc-4.0
 tags:
+- text-to-audio
+- acid_data
 - generated_from_trainer
 model-index:
 - name: musicgen-melody-lora-acid3_techno-colab
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # musicgen-melody-lora-acid3_techno-colab
 
-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 ShaySha/acid_data dataset.
 
 ## Model description
 

trainer_state.json

@@ -10,19 +10,19 @@
   "log_history": [
     {
       "epoch": 2.0,
-      "grad_norm": 0.45381349325180054,
+      "grad_norm": 0.5394480228424072,
       "learning_rate": 0.0,
-      "loss": 3.6167,
+      "loss": 4.8066,
       "step": 2
     },
     {
       "epoch": 2.0,
       "step": 2,
-      "total_flos": 831478453560.0,
-      "train_loss": 3.6166939735412598,
-      "train_runtime": 18.3096,
-      "train_samples_per_second": 0.546,
-      "train_steps_per_second": 0.109
+      "total_flos": 1182547133952.0,
+      "train_loss": 4.806564807891846,
+      "train_runtime": 27.6339,
+      "train_samples_per_second": 0.579,
+      "train_steps_per_second": 0.072
     }
   ],
   "logging_steps": 2,
@@ -42,7 +42,7 @@
       "attributes": {}
     }
   },
-  "total_flos": 831478453560.0,
+  "total_flos": 1182547133952.0,
   "train_batch_size": 2,
   "trial_name": null,
   "trial_params": null