{ "best_metric": 29.2694, "best_model_checkpoint": "./ko-en_mbartLarge_exp20p_linear/checkpoint-60000", "epoch": 6.9613644274277755, "eval_steps": 4000, "global_step": 60000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.06, "learning_rate": 1.25e-05, "loss": 1.9079, "step": 500 }, { "epoch": 0.12, "learning_rate": 2.5e-05, "loss": 1.7987, "step": 1000 }, { "epoch": 0.17, "learning_rate": 3.7500000000000003e-05, "loss": 1.648, "step": 1500 }, { "epoch": 0.23, "learning_rate": 5e-05, "loss": 1.5348, "step": 2000 }, { "epoch": 0.29, "learning_rate": 4.9927062667755864e-05, "loss": 1.4769, "step": 2500 }, { "epoch": 0.35, "learning_rate": 4.985412533551173e-05, "loss": 1.4511, "step": 3000 }, { "epoch": 0.41, "learning_rate": 4.978118800326759e-05, "loss": 1.4201, "step": 3500 }, { "epoch": 0.46, "learning_rate": 4.970825067102346e-05, "loss": 1.3977, "step": 4000 }, { "epoch": 0.46, "eval_bleu": 22.7153, "eval_gen_len": 18.7135, "eval_loss": 1.3720011711120605, "eval_runtime": 1153.7, "eval_samples_per_second": 14.941, "eval_steps_per_second": 0.934, "step": 4000 }, { "epoch": 0.52, "learning_rate": 4.963531333877932e-05, "loss": 1.3792, "step": 4500 }, { "epoch": 0.58, "learning_rate": 4.956237600653519e-05, "loss": 1.3552, "step": 5000 }, { "epoch": 0.64, "learning_rate": 4.948943867429105e-05, "loss": 1.3468, "step": 5500 }, { "epoch": 0.7, "learning_rate": 4.941650134204692e-05, "loss": 1.3468, "step": 6000 }, { "epoch": 0.75, "learning_rate": 4.934356400980278e-05, "loss": 1.3236, "step": 6500 }, { "epoch": 0.81, "learning_rate": 4.927062667755864e-05, "loss": 1.2924, "step": 7000 }, { "epoch": 0.87, "learning_rate": 4.9197689345314504e-05, "loss": 1.3005, "step": 7500 }, { "epoch": 0.93, "learning_rate": 4.912475201307037e-05, "loss": 1.2824, "step": 8000 }, { "epoch": 0.93, "eval_bleu": 24.8579, "eval_gen_len": 18.7821, "eval_loss": 1.2633070945739746, "eval_runtime": 1161.9505, "eval_samples_per_second": 14.835, "eval_steps_per_second": 0.928, "step": 8000 }, { "epoch": 0.99, "learning_rate": 4.905181468082623e-05, "loss": 1.2758, "step": 8500 }, { "epoch": 1.04, "learning_rate": 4.89788773485821e-05, "loss": 1.2295, "step": 9000 }, { "epoch": 1.1, "learning_rate": 4.890594001633796e-05, "loss": 1.2196, "step": 9500 }, { "epoch": 1.16, "learning_rate": 4.883300268409383e-05, "loss": 1.2043, "step": 10000 }, { "epoch": 1.22, "learning_rate": 4.876006535184969e-05, "loss": 1.2144, "step": 10500 }, { "epoch": 1.28, "learning_rate": 4.868712801960555e-05, "loss": 1.195, "step": 11000 }, { "epoch": 1.33, "learning_rate": 4.861419068736142e-05, "loss": 1.184, "step": 11500 }, { "epoch": 1.39, "learning_rate": 4.854125335511728e-05, "loss": 1.1989, "step": 12000 }, { "epoch": 1.39, "eval_bleu": 26.2533, "eval_gen_len": 18.7975, "eval_loss": 1.2069392204284668, "eval_runtime": 1167.3322, "eval_samples_per_second": 14.766, "eval_steps_per_second": 0.923, "step": 12000 }, { "epoch": 1.45, "learning_rate": 4.846831602287315e-05, "loss": 1.186, "step": 12500 }, { "epoch": 1.51, "learning_rate": 4.839537869062901e-05, "loss": 1.1785, "step": 13000 }, { "epoch": 1.57, "learning_rate": 4.832244135838488e-05, "loss": 1.1659, "step": 13500 }, { "epoch": 1.62, "learning_rate": 4.824950402614074e-05, "loss": 1.1624, "step": 14000 }, { "epoch": 1.68, "learning_rate": 4.817656669389661e-05, "loss": 1.1785, "step": 14500 }, { "epoch": 1.74, "learning_rate": 4.810362936165247e-05, "loss": 1.1465, "step": 15000 }, { "epoch": 1.8, "learning_rate": 4.803069202940834e-05, "loss": 1.1617, "step": 15500 }, { "epoch": 1.86, "learning_rate": 4.79577546971642e-05, "loss": 1.1534, "step": 16000 }, { "epoch": 1.86, "eval_bleu": 26.1503, "eval_gen_len": 19.2075, "eval_loss": 1.190694808959961, "eval_runtime": 1177.3192, "eval_samples_per_second": 14.641, "eval_steps_per_second": 0.916, "step": 16000 }, { "epoch": 1.91, "learning_rate": 4.788481736492006e-05, "loss": 1.1745, "step": 16500 }, { "epoch": 1.97, "learning_rate": 4.781188003267593e-05, "loss": 1.1663, "step": 17000 }, { "epoch": 2.03, "learning_rate": 4.773894270043179e-05, "loss": 1.1029, "step": 17500 }, { "epoch": 2.09, "learning_rate": 4.766600536818765e-05, "loss": 1.0464, "step": 18000 }, { "epoch": 2.15, "learning_rate": 4.759306803594352e-05, "loss": 1.0348, "step": 18500 }, { "epoch": 2.2, "learning_rate": 4.752013070369938e-05, "loss": 1.0228, "step": 19000 }, { "epoch": 2.26, "learning_rate": 4.744719337145524e-05, "loss": 1.0503, "step": 19500 }, { "epoch": 2.32, "learning_rate": 4.737425603921111e-05, "loss": 1.0245, "step": 20000 }, { "epoch": 2.32, "eval_bleu": 27.8764, "eval_gen_len": 18.6046, "eval_loss": 1.146370768547058, "eval_runtime": 1146.7204, "eval_samples_per_second": 15.032, "eval_steps_per_second": 0.94, "step": 20000 }, { "epoch": 2.38, "learning_rate": 4.730131870696697e-05, "loss": 1.0493, "step": 20500 }, { "epoch": 2.44, "learning_rate": 4.722838137472284e-05, "loss": 1.0287, "step": 21000 }, { "epoch": 2.49, "learning_rate": 4.71554440424787e-05, "loss": 1.03, "step": 21500 }, { "epoch": 2.55, "learning_rate": 4.708250671023457e-05, "loss": 1.0354, "step": 22000 }, { "epoch": 2.61, "learning_rate": 4.700956937799043e-05, "loss": 1.0268, "step": 22500 }, { "epoch": 2.67, "learning_rate": 4.69366320457463e-05, "loss": 1.039, "step": 23000 }, { "epoch": 2.73, "learning_rate": 4.686369471350216e-05, "loss": 1.0343, "step": 23500 }, { "epoch": 2.78, "learning_rate": 4.679075738125803e-05, "loss": 1.0186, "step": 24000 }, { "epoch": 2.78, "eval_bleu": 28.4585, "eval_gen_len": 18.6731, "eval_loss": 1.1286325454711914, "eval_runtime": 1151.5333, "eval_samples_per_second": 14.969, "eval_steps_per_second": 0.936, "step": 24000 }, { "epoch": 2.84, "learning_rate": 4.671782004901389e-05, "loss": 1.0383, "step": 24500 }, { "epoch": 2.9, "learning_rate": 4.664488271676976e-05, "loss": 1.0488, "step": 25000 }, { "epoch": 2.96, "learning_rate": 4.657194538452562e-05, "loss": 1.0572, "step": 25500 }, { "epoch": 3.02, "learning_rate": 4.649900805228149e-05, "loss": 1.0168, "step": 26000 }, { "epoch": 3.07, "learning_rate": 4.642607072003735e-05, "loss": 0.9238, "step": 26500 }, { "epoch": 3.13, "learning_rate": 4.635313338779321e-05, "loss": 0.9281, "step": 27000 }, { "epoch": 3.19, "learning_rate": 4.628019605554907e-05, "loss": 0.9225, "step": 27500 }, { "epoch": 3.25, "learning_rate": 4.620725872330493e-05, "loss": 0.9245, "step": 28000 }, { "epoch": 3.25, "eval_bleu": 28.4834, "eval_gen_len": 18.5428, "eval_loss": 1.1263633966445923, "eval_runtime": 1147.1298, "eval_samples_per_second": 15.026, "eval_steps_per_second": 0.94, "step": 28000 }, { "epoch": 3.31, "learning_rate": 4.61343213910608e-05, "loss": 0.9145, "step": 28500 }, { "epoch": 3.36, "learning_rate": 4.606138405881666e-05, "loss": 0.9228, "step": 29000 }, { "epoch": 3.42, "learning_rate": 4.598844672657253e-05, "loss": 0.9196, "step": 29500 }, { "epoch": 3.48, "learning_rate": 4.591550939432839e-05, "loss": 0.9138, "step": 30000 }, { "epoch": 3.54, "learning_rate": 4.584257206208426e-05, "loss": 0.9274, "step": 30500 }, { "epoch": 3.6, "learning_rate": 4.576963472984012e-05, "loss": 0.9077, "step": 31000 }, { "epoch": 3.65, "learning_rate": 4.569669739759599e-05, "loss": 0.9312, "step": 31500 }, { "epoch": 3.71, "learning_rate": 4.562376006535185e-05, "loss": 0.9343, "step": 32000 }, { "epoch": 3.71, "eval_bleu": 28.8235, "eval_gen_len": 18.7833, "eval_loss": 1.1182423830032349, "eval_runtime": 1155.993, "eval_samples_per_second": 14.911, "eval_steps_per_second": 0.933, "step": 32000 }, { "epoch": 3.77, "learning_rate": 4.555082273310772e-05, "loss": 0.9232, "step": 32500 }, { "epoch": 3.83, "learning_rate": 4.547788540086358e-05, "loss": 0.9308, "step": 33000 }, { "epoch": 3.89, "learning_rate": 4.540494806861945e-05, "loss": 0.9361, "step": 33500 }, { "epoch": 3.94, "learning_rate": 4.533201073637531e-05, "loss": 0.9139, "step": 34000 }, { "epoch": 4.0, "learning_rate": 4.525907340413118e-05, "loss": 0.919, "step": 34500 }, { "epoch": 4.06, "learning_rate": 4.518613607188704e-05, "loss": 0.8226, "step": 35000 }, { "epoch": 4.12, "learning_rate": 4.5113198739642907e-05, "loss": 0.8242, "step": 35500 }, { "epoch": 4.18, "learning_rate": 4.504026140739877e-05, "loss": 0.8215, "step": 36000 }, { "epoch": 4.18, "eval_bleu": 28.6134, "eval_gen_len": 18.5656, "eval_loss": 1.133116602897644, "eval_runtime": 1144.5462, "eval_samples_per_second": 15.06, "eval_steps_per_second": 0.942, "step": 36000 }, { "epoch": 4.23, "learning_rate": 4.496732407515463e-05, "loss": 0.833, "step": 36500 }, { "epoch": 4.29, "learning_rate": 4.489438674291049e-05, "loss": 0.8175, "step": 37000 }, { "epoch": 4.35, "learning_rate": 4.482144941066635e-05, "loss": 0.8313, "step": 37500 }, { "epoch": 4.41, "learning_rate": 4.474851207842222e-05, "loss": 0.832, "step": 38000 }, { "epoch": 4.47, "learning_rate": 4.467557474617808e-05, "loss": 0.8397, "step": 38500 }, { "epoch": 4.52, "learning_rate": 4.460263741393395e-05, "loss": 0.8324, "step": 39000 }, { "epoch": 4.58, "learning_rate": 4.452970008168981e-05, "loss": 0.8379, "step": 39500 }, { "epoch": 4.64, "learning_rate": 4.445676274944568e-05, "loss": 0.8456, "step": 40000 }, { "epoch": 4.64, "eval_bleu": 28.7324, "eval_gen_len": 18.459, "eval_loss": 1.1203099489212036, "eval_runtime": 1135.006, "eval_samples_per_second": 15.187, "eval_steps_per_second": 0.95, "step": 40000 }, { "epoch": 4.7, "learning_rate": 4.438382541720154e-05, "loss": 0.8481, "step": 40500 }, { "epoch": 4.76, "learning_rate": 4.431088808495741e-05, "loss": 0.8376, "step": 41000 }, { "epoch": 4.81, "learning_rate": 4.423795075271327e-05, "loss": 0.8416, "step": 41500 }, { "epoch": 4.87, "learning_rate": 4.416501342046914e-05, "loss": 0.8462, "step": 42000 }, { "epoch": 4.93, "learning_rate": 4.4092076088225e-05, "loss": 0.846, "step": 42500 }, { "epoch": 4.99, "learning_rate": 4.401913875598087e-05, "loss": 0.8429, "step": 43000 }, { "epoch": 5.05, "learning_rate": 4.394620142373673e-05, "loss": 0.7648, "step": 43500 }, { "epoch": 5.11, "learning_rate": 4.3873264091492596e-05, "loss": 0.7437, "step": 44000 }, { "epoch": 5.11, "eval_bleu": 28.7297, "eval_gen_len": 18.7835, "eval_loss": 1.1458019018173218, "eval_runtime": 1166.3289, "eval_samples_per_second": 14.779, "eval_steps_per_second": 0.924, "step": 44000 }, { "epoch": 5.16, "learning_rate": 4.380032675924846e-05, "loss": 0.7562, "step": 44500 }, { "epoch": 5.22, "learning_rate": 4.3727389427004326e-05, "loss": 0.752, "step": 45000 }, { "epoch": 5.28, "learning_rate": 4.365445209476019e-05, "loss": 0.7526, "step": 45500 }, { "epoch": 5.34, "learning_rate": 4.358151476251605e-05, "loss": 0.7779, "step": 46000 }, { "epoch": 5.4, "learning_rate": 4.350857743027191e-05, "loss": 0.7942, "step": 46500 }, { "epoch": 5.45, "learning_rate": 4.343564009802778e-05, "loss": 0.7954, "step": 47000 }, { "epoch": 5.51, "learning_rate": 4.336270276578364e-05, "loss": 0.7874, "step": 47500 }, { "epoch": 5.57, "learning_rate": 4.32897654335395e-05, "loss": 0.7829, "step": 48000 }, { "epoch": 5.57, "eval_bleu": 28.8328, "eval_gen_len": 18.6052, "eval_loss": 1.1367288827896118, "eval_runtime": 1145.791, "eval_samples_per_second": 15.044, "eval_steps_per_second": 0.941, "step": 48000 }, { "epoch": 5.63, "learning_rate": 4.321682810129537e-05, "loss": 0.7853, "step": 48500 }, { "epoch": 5.69, "learning_rate": 4.314389076905123e-05, "loss": 0.7916, "step": 49000 }, { "epoch": 5.74, "learning_rate": 4.30709534368071e-05, "loss": 0.7914, "step": 49500 }, { "epoch": 5.8, "learning_rate": 4.299801610456296e-05, "loss": 0.8096, "step": 50000 }, { "epoch": 5.86, "learning_rate": 4.292507877231883e-05, "loss": 0.8128, "step": 50500 }, { "epoch": 5.92, "learning_rate": 4.285214144007469e-05, "loss": 0.7987, "step": 51000 }, { "epoch": 5.98, "learning_rate": 4.2779204107830557e-05, "loss": 0.8185, "step": 51500 }, { "epoch": 6.03, "learning_rate": 4.270626677558642e-05, "loss": 0.7434, "step": 52000 }, { "epoch": 6.03, "eval_bleu": 28.2106, "eval_gen_len": 18.4871, "eval_loss": 1.1697230339050293, "eval_runtime": 1139.5757, "eval_samples_per_second": 15.126, "eval_steps_per_second": 0.946, "step": 52000 }, { "epoch": 6.09, "learning_rate": 4.2633329443342286e-05, "loss": 0.6938, "step": 52500 }, { "epoch": 6.15, "learning_rate": 4.256039211109815e-05, "loss": 0.695, "step": 53000 }, { "epoch": 6.21, "learning_rate": 4.2487454778854015e-05, "loss": 0.7016, "step": 53500 }, { "epoch": 6.27, "learning_rate": 4.241451744660988e-05, "loss": 0.6836, "step": 54000 }, { "epoch": 6.32, "learning_rate": 4.234158011436574e-05, "loss": 0.6807, "step": 54500 }, { "epoch": 6.38, "learning_rate": 4.2268642782121606e-05, "loss": 0.674, "step": 55000 }, { "epoch": 6.44, "learning_rate": 4.219570544987747e-05, "loss": 0.6897, "step": 55500 }, { "epoch": 6.5, "learning_rate": 4.2122768117633335e-05, "loss": 0.7153, "step": 56000 }, { "epoch": 6.5, "eval_bleu": 28.1455, "eval_gen_len": 18.7413, "eval_loss": 1.1770974397659302, "eval_runtime": 1156.7175, "eval_samples_per_second": 14.902, "eval_steps_per_second": 0.932, "step": 56000 }, { "epoch": 6.56, "learning_rate": 4.20498307853892e-05, "loss": 0.7177, "step": 56500 }, { "epoch": 6.61, "learning_rate": 4.197689345314506e-05, "loss": 0.716, "step": 57000 }, { "epoch": 6.67, "learning_rate": 4.190395612090092e-05, "loss": 0.721, "step": 57500 }, { "epoch": 6.73, "learning_rate": 4.183101878865679e-05, "loss": 0.7152, "step": 58000 }, { "epoch": 6.79, "learning_rate": 4.175808145641265e-05, "loss": 0.7012, "step": 58500 }, { "epoch": 6.85, "learning_rate": 4.168514412416852e-05, "loss": 0.7069, "step": 59000 }, { "epoch": 6.9, "learning_rate": 4.161220679192438e-05, "loss": 0.7088, "step": 59500 }, { "epoch": 6.96, "learning_rate": 4.1539269459680246e-05, "loss": 0.6996, "step": 60000 }, { "epoch": 6.96, "eval_bleu": 29.2694, "eval_gen_len": 18.5162, "eval_loss": 1.1514174938201904, "eval_runtime": 1140.5069, "eval_samples_per_second": 15.113, "eval_steps_per_second": 0.945, "step": 60000 } ], "logging_steps": 500, "max_steps": 344760, "num_train_epochs": 40, "save_steps": 4000, "total_flos": 2.080583273523708e+18, "trial_name": null, "trial_params": null }