| { |
| "config": { |
| "epochs": 50, |
| "lr": 0.0003, |
| "warmup_steps": 100, |
| "grad_accum_steps": 8, |
| "grad_clip": 1.0 |
| }, |
| "train": [ |
| { |
| "epoch": 1, |
| "avg_loss": 51.190041, |
| "lr": 0.00028799999999999995, |
| "step": 96, |
| "time_s": 527.2 |
| }, |
| { |
| "epoch": 2, |
| "avg_loss": 34.603503, |
| "lr": 0.00029974034213769383, |
| "step": 192, |
| "time_s": 528.3 |
| }, |
| { |
| "epoch": 3, |
| "avg_loss": 29.146077, |
| "lr": 0.0002989167655490271, |
| "step": 288, |
| "time_s": 528.6 |
| }, |
| { |
| "epoch": 4, |
| "avg_loss": 26.977388, |
| "lr": 0.00029753204719068163, |
| "step": 384, |
| "time_s": 529.0 |
| }, |
| { |
| "epoch": 5, |
| "avg_loss": 25.505867, |
| "lr": 0.0002955916816255344, |
| "step": 480, |
| "time_s": 529.0 |
| }, |
| { |
| "epoch": 6, |
| "avg_loss": 24.266201, |
| "lr": 0.0002931033682247511, |
| "step": 576, |
| "time_s": 528.2 |
| }, |
| { |
| "epoch": 7, |
| "avg_loss": 23.1789, |
| "lr": 0.00029007698061667977, |
| "step": 672, |
| "time_s": 528.3 |
| }, |
| { |
| "epoch": 8, |
| "avg_loss": 22.32955, |
| "lr": 0.00028652452750829054, |
| "step": 768, |
| "time_s": 528.6 |
| }, |
| { |
| "epoch": 9, |
| "avg_loss": 21.523272, |
| "lr": 0.0002824601050346221, |
| "step": 864, |
| "time_s": 529.1 |
| }, |
| { |
| "epoch": 10, |
| "avg_loss": 20.717048, |
| "lr": 0.00027789984082531423, |
| "step": 960, |
| "time_s": 527.3 |
| }, |
| { |
| "epoch": 11, |
| "avg_loss": 19.595496, |
| "lr": 0.0002728618300101674, |
| "step": 1056, |
| "time_s": 527.2 |
| }, |
| { |
| "epoch": 12, |
| "avg_loss": 18.476581, |
| "lr": 0.0002673660634176636, |
| "step": 1152, |
| "time_s": 527.7 |
| }, |
| { |
| "epoch": 13, |
| "avg_loss": 17.646048, |
| "lr": 0.00026143434825135023, |
| "step": 1248, |
| "time_s": 527.9 |
| }, |
| { |
| "epoch": 14, |
| "avg_loss": 17.063148, |
| "lr": 0.0002550902215588531, |
| "step": 1344, |
| "time_s": 528.7 |
| }, |
| { |
| "epoch": 15, |
| "avg_loss": 16.571975, |
| "lr": 0.00024835885683686075, |
| "step": 1440, |
| "time_s": 528.6 |
| } |
| ], |
| "val": [ |
| { |
| "epoch": 1, |
| "avg_loss": 41.392915 |
| }, |
| { |
| "epoch": 2, |
| "avg_loss": 30.623778 |
| }, |
| { |
| "epoch": 3, |
| "avg_loss": 27.911329 |
| }, |
| { |
| "epoch": 4, |
| "avg_loss": 26.286426 |
| }, |
| { |
| "epoch": 5, |
| "avg_loss": 25.004706 |
| }, |
| { |
| "epoch": 6, |
| "avg_loss": 23.84809 |
| }, |
| { |
| "epoch": 7, |
| "avg_loss": 22.920575 |
| }, |
| { |
| "epoch": 8, |
| "avg_loss": 22.398431 |
| }, |
| { |
| "epoch": 9, |
| "avg_loss": 21.322621 |
| }, |
| { |
| "epoch": 10, |
| "avg_loss": 20.282576 |
| }, |
| { |
| "epoch": 11, |
| "avg_loss": 19.124007 |
| }, |
| { |
| "epoch": 12, |
| "avg_loss": 18.138864 |
| }, |
| { |
| "epoch": 13, |
| "avg_loss": 17.611274 |
| }, |
| { |
| "epoch": 14, |
| "avg_loss": 17.186254 |
| }, |
| { |
| "epoch": 15, |
| "avg_loss": 16.819747 |
| } |
| ], |
| "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" |
| } |
| ] |
| } |
