| { |
| "config": { |
| "epochs": 50, |
| "lr": 0.0001, |
| "warmup_steps": 100, |
| "grad_accum_steps": 8, |
| "grad_clip": 1.0 |
| }, |
| "train": [ |
| { |
| "epoch": 1, |
| "avg_loss": 48.950583, |
| "lr": 9.99783297632528e-05, |
| "step": 192, |
| "time_s": 1012.5 |
| }, |
| { |
| "epoch": 2, |
| "avg_loss": 39.822198, |
| "lr": 9.979363170168663e-05, |
| "step": 384, |
| "time_s": 1016.4 |
| }, |
| { |
| "epoch": 3, |
| "avg_loss": 36.55349, |
| "lr": 9.942103713803461e-05, |
| "step": 576, |
| "time_s": 1018.1 |
| }, |
| { |
| "epoch": 4, |
| "avg_loss": 34.752199, |
| "lr": 9.88620263818582e-05, |
| "step": 768, |
| "time_s": 1016.3 |
| }, |
| { |
| "epoch": 5, |
| "avg_loss": 33.489609, |
| "lr": 9.811882036992912e-05, |
| "step": 960, |
| "time_s": 1016.9 |
| }, |
| { |
| "epoch": 6, |
| "avg_loss": 32.44456, |
| "lr": 9.719437184249907e-05, |
| "step": 1152, |
| "time_s": 1020.7 |
| }, |
| { |
| "epoch": 7, |
| "avg_loss": 31.508465, |
| "lr": 9.609235361213091e-05, |
| "step": 1344, |
| "time_s": 1017.7 |
| }, |
| { |
| "epoch": 8, |
| "avg_loss": 30.626262, |
| "lr": 9.481714397169986e-05, |
| "step": 1536, |
| "time_s": 1018.6 |
| }, |
| { |
| "epoch": 9, |
| "avg_loss": 29.740664, |
| "lr": 9.337380929953731e-05, |
| "step": 1728, |
| "time_s": 1017.3 |
| }, |
| { |
| "epoch": 10, |
| "avg_loss": 28.876585, |
| "lr": 9.176808393082695e-05, |
| "step": 1920, |
| "time_s": 1015.7 |
| }, |
| { |
| "epoch": 11, |
| "avg_loss": 28.011303, |
| "lr": 9.000634737522459e-05, |
| "step": 2112, |
| "time_s": 1017.1 |
| }, |
| { |
| "epoch": 12, |
| "avg_loss": 27.240841, |
| "lr": 8.80955989712136e-05, |
| "step": 2304, |
| "time_s": 1021.0 |
| }, |
| { |
| "epoch": 13, |
| "avg_loss": 26.574948, |
| "lr": 8.604343007789604e-05, |
| "step": 2496, |
| "time_s": 1018.9 |
| }, |
| { |
| "epoch": 14, |
| "avg_loss": 25.985939, |
| "lr": 8.385799391469841e-05, |
| "step": 2688, |
| "time_s": 1020.0 |
| }, |
| { |
| "epoch": 15, |
| "avg_loss": 25.478551, |
| "lr": 8.154797316882016e-05, |
| "step": 2880, |
| "time_s": 1020.9 |
| } |
| ], |
| "val": [ |
| { |
| "epoch": 1, |
| "avg_loss": 42.255684 |
| }, |
| { |
| "epoch": 2, |
| "avg_loss": 37.861168 |
| }, |
| { |
| "epoch": 3, |
| "avg_loss": 35.591921 |
| }, |
| { |
| "epoch": 4, |
| "avg_loss": 34.174768 |
| }, |
| { |
| "epoch": 5, |
| "avg_loss": 33.054291 |
| }, |
| { |
| "epoch": 6, |
| "avg_loss": 32.198608 |
| }, |
| { |
| "epoch": 7, |
| "avg_loss": 31.356775 |
| }, |
| { |
| "epoch": 8, |
| "avg_loss": 30.324559 |
| }, |
| { |
| "epoch": 9, |
| "avg_loss": 29.332707 |
| }, |
| { |
| "epoch": 10, |
| "avg_loss": 28.455519 |
| }, |
| { |
| "epoch": 11, |
| "avg_loss": 27.58525 |
| }, |
| { |
| "epoch": 12, |
| "avg_loss": 26.99554 |
| }, |
| { |
| "epoch": 13, |
| "avg_loss": 26.37473 |
| }, |
| { |
| "epoch": 14, |
| "avg_loss": 25.988197 |
| }, |
| { |
| "epoch": 15, |
| "avg_loss": 25.571769 |
| } |
| ], |
| "sanity": [ |
| { |
| "epoch": 5, |
| "audio_path": "/kaggle/working/inference_out.wav" |
| }, |
| { |
| "epoch": 10, |
| "audio_path": "/kaggle/working/inference_out.wav" |
| }, |
| { |
| "epoch": 15, |
| "audio_path": "/kaggle/working/inference_out.wav" |
| } |
| ] |
| } |
