Log update epoch 2

training_log.json

@@ -1,126 +1,25 @@
 {
   "config": {
     "epochs": 50,
-    "lr": 0.0001,
-    "warmup_steps": 100,
+    "lr": 0.0003,
+    "warmup_steps": 1000,
     "grad_accum_steps": 8,
     "grad_clip": 1.0
   },
   "train": [
     {
       "epoch": 1,
-      "avg_loss": 6.414212,
-      "lr": 9.898803064624334e-05,
+      "avg_loss": 6.84537,
+      "lr": 4.62e-05,
       "step": 154,
-      "time_s": 1826.1
-    },
-    {
-      "epoch": 2,
-      "avg_loss": 5.160906,
-      "lr": 9.882726111015644e-05,
-      "step": 308,
-      "time_s": 1835.8
-    },
-    {
-      "epoch": 3,
-      "avg_loss": 4.819508,
-      "lr": 9.847955616807017e-05,
-      "step": 462,
-      "time_s": 1854.7
-    },
-    {
-      "epoch": 4,
-      "avg_loss": 4.637806,
-      "lr": 9.79463038653877e-05,
-      "step": 616,
-      "time_s": 1869.6
-    },
-    {
-      "epoch": 5,
-      "avg_loss": 4.507495,
-      "lr": 9.722963295645685e-05,
-      "step": 770,
-      "time_s": 1852.1
-    },
-    {
-      "epoch": 6,
-      "avg_loss": 4.395226,
-      "lr": 9.633240440653926e-05,
-      "step": 924,
-      "time_s": 1840.7
-    },
-    {
-      "epoch": 7,
-      "avg_loss": 4.292807,
-      "lr": 9.525819997077667e-05,
-      "step": 1078,
-      "time_s": 1840.3
-    },
-    {
-      "epoch": 8,
-      "avg_loss": 4.208474,
-      "lr": 9.40113078957496e-05,
-      "step": 1232,
-      "time_s": 1836.9
-    },
-    {
-      "epoch": 9,
-      "avg_loss": 4.127683,
-      "lr": 9.259670580070702e-05,
-      "step": 1386,
-      "time_s": 1835.7
+      "time_s": 1837.0
     }
   ],
   "val": [
     {
       "epoch": 1,
-      "avg_loss": 5.472925
-    },
-    {
-      "epoch": 2,
-      "avg_loss": 4.957525
-    },
-    {
-      "epoch": 3,
-      "avg_loss": 4.721136
-    },
-    {
-      "epoch": 4,
-      "avg_loss": 4.57539
-    },
-    {
-      "epoch": 5,
-      "avg_loss": 4.47602
-    },
-    {
-      "epoch": 6,
-      "avg_loss": 4.346653
-    },
-    {
-      "epoch": 7,
-      "avg_loss": 4.270272
-    },
-    {
-      "epoch": 8,
-      "avg_loss": 4.168377
-    },
-    {
-      "epoch": 9,
-      "avg_loss": 4.098728
+      "avg_loss": 6.066605
     }
   ],
-  "sanity": [
-    {
-      "epoch": 3,
-      "audio_path": "/kaggle/working/inference_out.wav"
-    },
-    {
-      "epoch": 6,
-      "audio_path": "/kaggle/working/inference_out.wav"
-    },
-    {
-      "epoch": 9,
-      "audio_path": "/kaggle/working/inference_out.wav"
-    }
-  ]
+  "sanity": []
 }