| { |
| "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 |
| }, |
| { |
| "epoch": 16, |
| "avg_loss": 15.914811, |
| "lr": 9.684231938481317e-05, |
| "step": 1536, |
| "time_s": 527.2 |
| }, |
| { |
| "epoch": 17, |
| "avg_loss": 15.648117, |
| "lr": 9.358970080957491e-05, |
| "step": 1632, |
| "time_s": 528.0 |
| }, |
| { |
| "epoch": 18, |
| "avg_loss": 15.456193, |
| "lr": 9.022320135942954e-05, |
| "step": 1728, |
| "time_s": 527.0 |
| }, |
| { |
| "epoch": 19, |
| "avg_loss": 15.278314, |
| "lr": 8.675617930524218e-05, |
| "step": 1824, |
| "time_s": 528.2 |
| }, |
| { |
| "epoch": 20, |
| "avg_loss": 15.123071, |
| "lr": 8.320239179160214e-05, |
| "step": 1920, |
| "time_s": 529.3 |
| }, |
| { |
| "epoch": 21, |
| "avg_loss": 14.976132, |
| "lr": 7.957594024846865e-05, |
| "step": 2016, |
| "time_s": 530.0 |
| }, |
| { |
| "epoch": 22, |
| "avg_loss": 14.841149, |
| "lr": 7.589121443669282e-05, |
| "step": 2112, |
| "time_s": 528.2 |
| }, |
| { |
| "epoch": 23, |
| "avg_loss": 14.719751, |
| "lr": 7.21628353494416e-05, |
| "step": 2208, |
| "time_s": 528.5 |
| }, |
| { |
| "epoch": 24, |
| "avg_loss": 14.605327, |
| "lr": 6.840559719609277e-05, |
| "step": 2304, |
| "time_s": 529.3 |
| }, |
| { |
| "epoch": 25, |
| "avg_loss": 14.498531, |
| "lr": 6.46344086988074e-05, |
| "step": 2400, |
| "time_s": 528.0 |
| } |
| ], |
| "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 |
| }, |
| { |
| "epoch": 16, |
| "avg_loss": 16.453496 |
| }, |
| { |
| "epoch": 17, |
| "avg_loss": 16.304954 |
| }, |
| { |
| "epoch": 18, |
| "avg_loss": 16.218734 |
| }, |
| { |
| "epoch": 19, |
| "avg_loss": 16.09561 |
| }, |
| { |
| "epoch": 20, |
| "avg_loss": 16.013735 |
| }, |
| { |
| "epoch": 21, |
| "avg_loss": 15.967682 |
| }, |
| { |
| "epoch": 22, |
| "avg_loss": 15.913238 |
| }, |
| { |
| "epoch": 23, |
| "avg_loss": 15.851707 |
| }, |
| { |
| "epoch": 24, |
| "avg_loss": 15.789699 |
| }, |
| { |
| "epoch": 25, |
| "avg_loss": 15.782429 |
| } |
| ], |
| "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" |
| }, |
| { |
| "epoch": 20, |
| "audio_path": "/kaggle/working/inference_out.wav" |
| }, |
| { |
| "epoch": 25, |
| "audio_path": "/kaggle/working/inference_out.wav" |
| } |
| ] |
| } |
