End of training

README.md

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

trainer_state.json

@@ -10,54 +10,54 @@
   "log_history": [
     {
       "epoch": 0.64,
-      "grad_norm": 8.356424331665039,
+      "grad_norm": 8.356196403503418,
       "learning_rate": 0.0001666666666666667,
-      "loss": 76.2569,
+      "loss": 76.257,
       "step": 2
     },
     {
       "epoch": 1.0,
-      "grad_norm": 1.7912710905075073,
+      "grad_norm": 1.7907499074935913,
       "learning_rate": 0.00013333333333333334,
-      "loss": 41.8415,
+      "loss": 41.8413,
       "step": 4
     },
     {
       "epoch": 1.6400000000000001,
-      "grad_norm": 16.31438636779785,
+      "grad_norm": 16.30155372619629,
       "learning_rate": 0.0001,
-      "loss": 70.2302,
+      "loss": 70.232,
       "step": 6
     },
     {
       "epoch": 2.0,
-      "grad_norm": 2.897549629211426,
-      "learning_rate": 6.666666666666667e-05,
-      "loss": 38.4848,
+      "grad_norm": 2.569784164428711,
+      "learning_rate": 8.333333333333334e-05,
+      "loss": 38.5374,
       "step": 8
     },
     {
       "epoch": 2.64,
-      "grad_norm": 15.37080192565918,
-      "learning_rate": 3.3333333333333335e-05,
-      "loss": 65.8869,
+      "grad_norm": 15.395219802856445,
+      "learning_rate": 5e-05,
+      "loss": 66.8565,
       "step": 10
     },
     {
       "epoch": 3.0,
-      "grad_norm": 3.5507168769836426,
-      "learning_rate": 0.0,
-      "loss": 36.2084,
+      "grad_norm": 3.543508529663086,
+      "learning_rate": 1.6666666666666667e-05,
+      "loss": 36.567,
       "step": 12
     },
     {
       "epoch": 3.0,
       "step": 12,
       "total_flos": 40557671234760.0,
-      "train_loss": 54.81812349955241,
-      "train_runtime": 48.7104,
-      "train_samples_per_second": 4.106,
-      "train_steps_per_second": 0.246
+      "train_loss": 55.04854393005371,
+      "train_runtime": 49.2052,
+      "train_samples_per_second": 4.065,
+      "train_steps_per_second": 0.244
     }
   ],
   "logging_steps": 2,