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-acid2_techno-colab
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # musicgen-melody-lora-acid2_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,18 +10,18 @@
   "log_history": [
     {
       "epoch": 2.0,
-      "grad_norm": 0.3880564868450165,
+      "grad_norm": 0.45381349325180054,
       "learning_rate": 0.0,
-      "loss": 3.5507,
+      "loss": 3.6167,
       "step": 2
     },
     {
       "epoch": 2.0,
       "step": 2,
-      "total_flos": 739091958720.0,
-      "train_loss": 3.5507075786590576,
-      "train_runtime": 18.2944,
-      "train_samples_per_second": 0.547,
+      "total_flos": 831478453560.0,
+      "train_loss": 3.6166939735412598,
+      "train_runtime": 18.3096,
+      "train_samples_per_second": 0.546,
       "train_steps_per_second": 0.109
     }
   ],
@@ -42,7 +42,7 @@
       "attributes": {}
     }
   },
-  "total_flos": 739091958720.0,
+  "total_flos": 831478453560.0,
   "train_batch_size": 2,
   "trial_name": null,
   "trial_params": null