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
+- classic_techno_clips
 - generated_from_trainer
 model-index:
 - name: musicgen-melody-lora-regular3_techno-colab
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # musicgen-melody-lora-regular3_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/classic_techno_clips dataset.
 
 ## Model description
 

trainer_state.json

@@ -3,100 +3,30 @@
   "best_model_checkpoint": null,
   "epoch": 2.0,
   "eval_steps": 500,
-  "global_step": 22,
+  "global_step": 2,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
-    {
-      "epoch": 0.18181818181818182,
-      "grad_norm": 0.9673512578010559,
-      "learning_rate": 0.00018181818181818183,
-      "loss": 9.4696,
-      "step": 2
-    },
-    {
-      "epoch": 0.36363636363636365,
-      "grad_norm": 1.2611587047576904,
-      "learning_rate": 0.00016363636363636366,
-      "loss": 9.1349,
-      "step": 4
-    },
-    {
-      "epoch": 0.5454545454545454,
-      "grad_norm": 2.2909739017486572,
-      "learning_rate": 0.00014545454545454546,
-      "loss": 8.631,
-      "step": 6
-    },
-    {
-      "epoch": 0.7272727272727273,
-      "grad_norm": 2.0307061672210693,
-      "learning_rate": 0.00012727272727272728,
-      "loss": 8.0002,
-      "step": 8
-    },
-    {
-      "epoch": 0.9090909090909091,
-      "grad_norm": 1.1030476093292236,
-      "learning_rate": 0.00010909090909090909,
-      "loss": 7.5263,
-      "step": 10
-    },
-    {
-      "epoch": 1.0909090909090908,
-      "grad_norm": 0.8558670878410339,
-      "learning_rate": 9.090909090909092e-05,
-      "loss": 7.4314,
-      "step": 12
-    },
-    {
-      "epoch": 1.2727272727272727,
-      "grad_norm": 0.947796642780304,
-      "learning_rate": 7.272727272727273e-05,
-      "loss": 7.3711,
-      "step": 14
-    },
-    {
-      "epoch": 1.4545454545454546,
-      "grad_norm": 0.7418118715286255,
-      "learning_rate": 5.4545454545454546e-05,
-      "loss": 7.2495,
-      "step": 16
-    },
-    {
-      "epoch": 1.6363636363636362,
-      "grad_norm": 0.7568270564079285,
-      "learning_rate": 3.6363636363636364e-05,
-      "loss": 7.2816,
-      "step": 18
-    },
-    {
-      "epoch": 1.8181818181818183,
-      "grad_norm": 0.955255389213562,
-      "learning_rate": 1.8181818181818182e-05,
-      "loss": 7.3431,
-      "step": 20
-    },
     {
       "epoch": 2.0,
-      "grad_norm": 0.8418056964874268,
+      "grad_norm": 0.3880564868450165,
       "learning_rate": 0.0,
-      "loss": 7.1887,
-      "step": 22
+      "loss": 3.5507,
+      "step": 2
     },
     {
       "epoch": 2.0,
-      "step": 22,
-      "total_flos": 26016036946944.0,
-      "train_loss": 7.875224590301514,
-      "train_runtime": 563.2063,
-      "train_samples_per_second": 0.625,
-      "train_steps_per_second": 0.039
+      "step": 2,
+      "total_flos": 739091958720.0,
+      "train_loss": 3.5507075786590576,
+      "train_runtime": 18.2944,
+      "train_samples_per_second": 0.547,
+      "train_steps_per_second": 0.109
     }
   ],
   "logging_steps": 2,
-  "max_steps": 22,
+  "max_steps": 2,
   "num_input_tokens_seen": 0,
   "num_train_epochs": 2,
   "save_steps": 500,
@@ -112,7 +42,7 @@
       "attributes": {}
     }
   },
-  "total_flos": 26016036946944.0,
+  "total_flos": 739091958720.0,
   "train_batch_size": 2,
   "trial_name": null,
   "trial_params": null