| { |
| "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" |
| } |
| ] |
| } |
