| { |
| "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 |
| }, |
| { |
| "epoch": 8, |
| "avg_loss": 3.470683, |
| "lr": 0.000276940766675198, |
| "step": 1232, |
| "time_s": 1976.5 |
| }, |
| { |
| "epoch": 9, |
| "avg_loss": 3.420985, |
| "lr": 0.00026663104328180195, |
| "step": 1386, |
| "time_s": 1972.6 |
| }, |
| { |
| "epoch": 10, |
| "avg_loss": 3.378666, |
| "lr": 0.0002546936507209144, |
| "step": 1540, |
| "time_s": 1979.8 |
| }, |
| { |
| "epoch": 11, |
| "avg_loss": 3.339928, |
| "lr": 0.0002412951837966703, |
| "step": 1694, |
| "time_s": 1973.9 |
| }, |
| { |
| "epoch": 12, |
| "avg_loss": 3.303412, |
| "lr": 0.00022662262764519306, |
| "step": 1848, |
| "time_s": 1981.5 |
| }, |
| { |
| "epoch": 13, |
| "avg_loss": 3.270916, |
| "lr": 0.00021088074822401884, |
| "step": 2002, |
| "time_s": 1974.9 |
| }, |
| { |
| "epoch": 14, |
| "avg_loss": 3.241058, |
| "lr": 0.00019428923465751146, |
| "step": 2156, |
| "time_s": 1969.6 |
| }, |
| { |
| "epoch": 15, |
| "avg_loss": 3.21413, |
| "lr": 0.00017707963331886904, |
| "step": 2310, |
| "time_s": 1968.4 |
| } |
| ], |
| "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 |
| }, |
| { |
| "epoch": 8, |
| "avg_loss": 3.45047 |
| }, |
| { |
| "epoch": 9, |
| "avg_loss": 3.420838 |
| }, |
| { |
| "epoch": 10, |
| "avg_loss": 3.387246 |
| }, |
| { |
| "epoch": 11, |
| "avg_loss": 3.363138 |
| }, |
| { |
| "epoch": 12, |
| "avg_loss": 3.336315 |
| }, |
| { |
| "epoch": 13, |
| "avg_loss": 3.319501 |
| }, |
| { |
| "epoch": 14, |
| "avg_loss": 3.298631 |
| }, |
| { |
| "epoch": 15, |
| "avg_loss": 3.285041 |
| } |
| ], |
| "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" |
| } |
| ] |
| } |
