{ "config": { "epochs": 50, "lr": 0.0003, "warmup_steps": 1000, "grad_accum_steps": 8, "grad_clip": 1.0 }, "train": [ { "epoch": 1, "avg_loss": 6.84537, "lr": 4.62e-05, "step": 154, "time_s": 1837.0 }, { "epoch": 2, "avg_loss": 5.575614, "lr": 9.24e-05, "step": 308, "time_s": 1839.1 }, { "epoch": 3, "avg_loss": 4.989896, "lr": 0.0001386, "step": 462, "time_s": 1835.3 }, { "epoch": 4, "avg_loss": 4.691752, "lr": 0.0001848, "step": 616, "time_s": 1840.6 }, { "epoch": 5, "avg_loss": 4.482983, "lr": 0.00023099999999999998, "step": 770, "time_s": 1834.2 }, { "epoch": 6, "avg_loss": 4.299778, "lr": 0.0002772, "step": 924, "time_s": 1833.6 }, { "epoch": 7, "avg_loss": 4.1192, "lr": 0.0002998983576882524, "step": 1078, "time_s": 1838.9 }, { "epoch": 8, "avg_loss": 3.908664, "lr": 0.000299101586340815, "step": 1232, "time_s": 1838.9 }, { "epoch": 9, "avg_loss": 3.715396, "lr": 0.00029751739651339516, "step": 1386, "time_s": 1834.4 }, { "epoch": 10, "avg_loss": 3.58282, "lr": 0.0002951541544506219, "step": 1540, "time_s": 1837.1 }, { "epoch": 11, "avg_loss": 3.473428, "lr": 0.00029202434063991704, "step": 1694, "time_s": 1835.9 }, { "epoch": 12, "avg_loss": 3.386997, "lr": 0.0002881444839009527, "step": 1848, "time_s": 1831.1 }, { "epoch": 13, "avg_loss": 3.314629, "lr": 0.00028353507409551327, "step": 2002, "time_s": 1839.7 }, { "epoch": 14, "avg_loss": 3.264942, "lr": 0.0002782204539187482, "step": 2156, "time_s": 1949.1 }, { "epoch": 15, "avg_loss": 3.203215, "lr": 0.0002722286903432766, "step": 2310, "time_s": 1955.6 }, { "epoch": 16, "avg_loss": 3.149329, "lr": 0.0002655914263950567, "step": 2464, "time_s": 1950.9 }, { "epoch": 17, "avg_loss": 3.100616, "lr": 0.0002583437140438068, "step": 2618, "time_s": 1949.1 }, { "epoch": 18, "avg_loss": 3.070203, "lr": 0.00025052382909049596, "step": 2772, "time_s": 1907.7 }, { "epoch": 19, "avg_loss": 3.024378, "lr": 0.0002421730690295016, "step": 2926, "time_s": 1908.8 }, { "epoch": 20, "avg_loss": 2.993083, "lr": 0.0002333355349529403, "step": 3080, "time_s": 1885.1 } ], "val": [ { "epoch": 1, "avg_loss": 6.066605 }, { "epoch": 2, "avg_loss": 5.205319 }, { "epoch": 3, "avg_loss": 4.828915 }, { "epoch": 4, "avg_loss": 4.60438 }, { "epoch": 5, "avg_loss": 4.435558 }, { "epoch": 6, "avg_loss": 4.259781 }, { "epoch": 7, "avg_loss": 4.034241 }, { "epoch": 8, "avg_loss": 3.795881 }, { "epoch": 9, "avg_loss": 3.69397 }, { "epoch": 10, "avg_loss": 3.575521 }, { "epoch": 11, "avg_loss": 3.480529 }, { "epoch": 12, "avg_loss": 3.427576 }, { "epoch": 13, "avg_loss": 3.405988 }, { "epoch": 14, "avg_loss": 3.222532 }, { "epoch": 15, "avg_loss": 3.198101 }, { "epoch": 16, "avg_loss": 3.186287 }, { "epoch": 17, "avg_loss": 3.174198 }, { "epoch": 18, "avg_loss": 3.035027 }, { "epoch": 19, "avg_loss": 3.036608 }, { "epoch": 20, "avg_loss": 2.939511 } ], "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" }, { "epoch": 12, "audio_path": "/kaggle/working/inference_out.wav" }, { "epoch": 15, "audio_path": "/kaggle/working/inference_out.wav" }, { "epoch": 18, "audio_path": "/kaggle/working/inference_out.wav" } ] }