{ "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" } ] }