| { |
| "config": { |
| "epochs": 30, |
| "lr": 0.0003, |
| "warmup_steps": 500, |
| "grad_accum_steps": 8, |
| "grad_clip": 1.0 |
| }, |
| "train": [ |
| { |
| "epoch": 1, |
| "avg_loss": 6.3824, |
| "lr": 9.24e-05, |
| "step": 154, |
| "time_s": 1992.5 |
| }, |
| { |
| "epoch": 2, |
| "avg_loss": 4.865482, |
| "lr": 0.0001848, |
| "step": 308, |
| "time_s": 1984.5 |
| }, |
| { |
| "epoch": 3, |
| "avg_loss": 4.285284, |
| "lr": 0.0002772, |
| "step": 462, |
| "time_s": 1992.7 |
| }, |
| { |
| "epoch": 4, |
| "avg_loss": 3.949742, |
| "lr": 0.00029940583567001586, |
| "step": 616, |
| "time_s": 1984.4 |
| }, |
| { |
| "epoch": 5, |
| "avg_loss": 3.740897, |
| "lr": 0.00029679040564010837, |
| "step": 770, |
| "time_s": 1975.7 |
| }, |
| { |
| "epoch": 6, |
| "avg_loss": 3.614488, |
| "lr": 0.00029212641108894655, |
| "step": 924, |
| "time_s": 1976.5 |
| }, |
| { |
| "epoch": 7, |
| "avg_loss": 3.530983, |
| "lr": 0.00028547894137794454, |
| "step": 1078, |
| "time_s": 1980.6 |
| } |
| ], |
| "val": [ |
| { |
| "epoch": 1, |
| "avg_loss": 5.362817 |
| }, |
| { |
| "epoch": 2, |
| "avg_loss": 4.490635 |
| }, |
| { |
| "epoch": 3, |
| "avg_loss": 4.07404 |
| }, |
| { |
| "epoch": 4, |
| "avg_loss": 3.811481 |
| }, |
| { |
| "epoch": 5, |
| "avg_loss": 3.647324 |
| }, |
| { |
| "epoch": 6, |
| "avg_loss": 3.559344 |
| }, |
| { |
| "epoch": 7, |
| "avg_loss": 3.5003 |
| } |
| ], |
| "sanity": [ |
| { |
| "epoch": 3, |
| "audio_path": "/kaggle/working/inference_out.wav" |
| }, |
| { |
| "epoch": 6, |
| "audio_path": "/kaggle/working/inference_out.wav" |
| } |
| ] |
| } |
