Log update epoch 6

training_log.json

@@ -0,0 +1,74 @@
+{
+  "config": {
+    "epochs": 50,
+    "lr": 0.0001,
+    "warmup_steps": 100,
+    "grad_accum_steps": 8,
+    "grad_clip": 1.0
+  },
+  "train": [
+    {
+      "epoch": 1,
+      "avg_loss": 44.475887,
+      "lr": 9.996020008796059e-05,
+      "step": 287,
+      "time_s": 1010.2
+    },
+    {
+      "epoch": 2,
+      "avg_loss": 32.809112,
+      "lr": 9.974447906798636e-05,
+      "step": 574,
+      "time_s": 1011.2
+    },
+    {
+      "epoch": 3,
+      "avg_loss": 29.81465,
+      "lr": 9.934230567308004e-05,
+      "step": 861,
+      "time_s": 1012.4
+    },
+    {
+      "epoch": 4,
+      "avg_loss": 28.016445,
+      "lr": 9.87552670978959e-05,
+      "step": 1148,
+      "time_s": 1011.7
+    },
+    {
+      "epoch": 5,
+      "avg_loss": 26.626537,
+      "lr": 9.79856801154978e-05,
+      "step": 1435,
+      "time_s": 1012.5
+    }
+  ],
+  "val": [
+    {
+      "epoch": 1,
+      "avg_loss": 35.319888
+    },
+    {
+      "epoch": 2,
+      "avg_loss": 30.780966
+    },
+    {
+      "epoch": 3,
+      "avg_loss": 28.72085
+    },
+    {
+      "epoch": 4,
+      "avg_loss": 27.274619
+    },
+    {
+      "epoch": 5,
+      "avg_loss": 26.087392
+    }
+  ],
+  "sanity": [
+    {
+      "epoch": 5,
+      "audio_path": "/kaggle/working/inference_out.wav"
+    }
+  ]
+}