{ "best_metric": 22.9493, "best_model_checkpoint": "./ko-en_mbartLarge_mid3/checkpoint-22000", "epoch": 2.552500290056851, "eval_steps": 2000, "global_step": 22000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.06, "learning_rate": 2.5e-05, "loss": 1.9104, "step": 500 }, { "epoch": 0.12, "learning_rate": 5e-05, "loss": 1.7744, "step": 1000 }, { "epoch": 0.17, "learning_rate": 4.99997390013085e-05, "loss": 1.6281, "step": 1500 }, { "epoch": 0.23, "learning_rate": 4.9998956010683606e-05, "loss": 1.5377, "step": 2000 }, { "epoch": 0.23, "eval_bleu": 17.2009, "eval_gen_len": 18.7106, "eval_loss": 1.6122242212295532, "eval_runtime": 1146.4034, "eval_samples_per_second": 15.036, "eval_steps_per_second": 0.94, "step": 2000 }, { "epoch": 0.29, "learning_rate": 4.9997651044474085e-05, "loss": 1.4666, "step": 2500 }, { "epoch": 0.35, "learning_rate": 4.999582412992751e-05, "loss": 1.4294, "step": 3000 }, { "epoch": 0.41, "learning_rate": 4.999347530518964e-05, "loss": 1.3956, "step": 3500 }, { "epoch": 0.46, "learning_rate": 4.999060461930371e-05, "loss": 1.3891, "step": 4000 }, { "epoch": 0.46, "eval_bleu": 19.3345, "eval_gen_len": 18.7688, "eval_loss": 1.5059210062026978, "eval_runtime": 1145.6172, "eval_samples_per_second": 15.046, "eval_steps_per_second": 0.941, "step": 4000 }, { "epoch": 0.52, "learning_rate": 4.998721213220933e-05, "loss": 1.3651, "step": 4500 }, { "epoch": 0.58, "learning_rate": 4.9983297914741276e-05, "loss": 1.3337, "step": 5000 }, { "epoch": 0.64, "learning_rate": 4.9978862048628e-05, "loss": 1.3017, "step": 5500 }, { "epoch": 0.7, "learning_rate": 4.997390462648993e-05, "loss": 1.2812, "step": 6000 }, { "epoch": 0.7, "eval_bleu": 20.6032, "eval_gen_len": 18.9022, "eval_loss": 1.4347867965698242, "eval_runtime": 1146.1733, "eval_samples_per_second": 15.039, "eval_steps_per_second": 0.941, "step": 6000 }, { "epoch": 0.75, "learning_rate": 4.996842575183751e-05, "loss": 1.2664, "step": 6500 }, { "epoch": 0.81, "learning_rate": 4.996242553906908e-05, "loss": 1.2599, "step": 7000 }, { "epoch": 0.87, "learning_rate": 4.9955904113468444e-05, "loss": 1.2513, "step": 7500 }, { "epoch": 0.93, "learning_rate": 4.994886161120229e-05, "loss": 1.2374, "step": 8000 }, { "epoch": 0.93, "eval_bleu": 21.2391, "eval_gen_len": 18.8434, "eval_loss": 1.4034597873687744, "eval_runtime": 1157.6007, "eval_samples_per_second": 14.89, "eval_steps_per_second": 0.931, "step": 8000 }, { "epoch": 0.99, "learning_rate": 4.994129817931734e-05, "loss": 1.2306, "step": 8500 }, { "epoch": 1.04, "learning_rate": 4.993321397573724e-05, "loss": 1.1716, "step": 9000 }, { "epoch": 1.1, "learning_rate": 4.9924609169259325e-05, "loss": 1.1718, "step": 9500 }, { "epoch": 1.16, "learning_rate": 4.991548393955104e-05, "loss": 1.1734, "step": 10000 }, { "epoch": 1.16, "eval_bleu": 21.304, "eval_gen_len": 18.9964, "eval_loss": 1.4039467573165894, "eval_runtime": 1151.81, "eval_samples_per_second": 14.965, "eval_steps_per_second": 0.936, "step": 10000 }, { "epoch": 1.22, "learning_rate": 4.990583847714626e-05, "loss": 1.1702, "step": 10500 }, { "epoch": 1.28, "learning_rate": 4.9895672983441194e-05, "loss": 1.1688, "step": 11000 }, { "epoch": 1.33, "learning_rate": 4.988498767069031e-05, "loss": 1.1581, "step": 11500 }, { "epoch": 1.39, "learning_rate": 4.98737827620018e-05, "loss": 1.1531, "step": 12000 }, { "epoch": 1.39, "eval_bleu": 21.9087, "eval_gen_len": 18.8573, "eval_loss": 1.3693655729293823, "eval_runtime": 1144.9147, "eval_samples_per_second": 15.055, "eval_steps_per_second": 0.942, "step": 12000 }, { "epoch": 1.45, "learning_rate": 4.9862058491333e-05, "loss": 1.1184, "step": 12500 }, { "epoch": 1.51, "learning_rate": 4.984981510348545e-05, "loss": 1.1171, "step": 13000 }, { "epoch": 1.57, "learning_rate": 4.9837052854099804e-05, "loss": 1.1149, "step": 13500 }, { "epoch": 1.62, "learning_rate": 4.9823772009650496e-05, "loss": 1.1158, "step": 14000 }, { "epoch": 1.62, "eval_bleu": 22.004, "eval_gen_len": 18.5485, "eval_loss": 1.357371211051941, "eval_runtime": 1129.6253, "eval_samples_per_second": 15.259, "eval_steps_per_second": 0.954, "step": 14000 }, { "epoch": 1.68, "learning_rate": 4.980997284744017e-05, "loss": 1.1163, "step": 14500 }, { "epoch": 1.74, "learning_rate": 4.979565565559388e-05, "loss": 1.1147, "step": 15000 }, { "epoch": 1.8, "learning_rate": 4.97808207330531e-05, "loss": 1.0988, "step": 15500 }, { "epoch": 1.86, "learning_rate": 4.9765468389569455e-05, "loss": 1.0941, "step": 16000 }, { "epoch": 1.86, "eval_bleu": 21.9785, "eval_gen_len": 18.7119, "eval_loss": 1.3456674814224243, "eval_runtime": 1149.9948, "eval_samples_per_second": 14.989, "eval_steps_per_second": 0.937, "step": 16000 }, { "epoch": 1.91, "learning_rate": 4.9749598945698276e-05, "loss": 1.1112, "step": 16500 }, { "epoch": 1.97, "learning_rate": 4.973321273279189e-05, "loss": 1.1, "step": 17000 }, { "epoch": 2.03, "learning_rate": 4.97163100929927e-05, "loss": 1.0405, "step": 17500 }, { "epoch": 2.09, "learning_rate": 4.9698891379226064e-05, "loss": 0.9809, "step": 18000 }, { "epoch": 2.09, "eval_bleu": 22.7983, "eval_gen_len": 18.8011, "eval_loss": 1.3494887351989746, "eval_runtime": 1141.8247, "eval_samples_per_second": 15.096, "eval_steps_per_second": 0.944, "step": 18000 }, { "epoch": 2.15, "learning_rate": 4.9680956955192896e-05, "loss": 0.9747, "step": 18500 }, { "epoch": 2.2, "learning_rate": 4.966250719536209e-05, "loss": 0.9933, "step": 19000 }, { "epoch": 2.26, "learning_rate": 4.964354248496271e-05, "loss": 0.9972, "step": 19500 }, { "epoch": 2.32, "learning_rate": 4.962406321997592e-05, "loss": 0.9834, "step": 20000 }, { "epoch": 2.32, "eval_bleu": 22.5654, "eval_gen_len": 18.9416, "eval_loss": 1.3429056406021118, "eval_runtime": 1150.5664, "eval_samples_per_second": 14.981, "eval_steps_per_second": 0.937, "step": 20000 }, { "epoch": 2.38, "learning_rate": 4.960406980712672e-05, "loss": 0.9855, "step": 20500 }, { "epoch": 2.44, "learning_rate": 4.9583562663875496e-05, "loss": 0.9939, "step": 21000 }, { "epoch": 2.49, "learning_rate": 4.956254221840925e-05, "loss": 0.985, "step": 21500 }, { "epoch": 2.55, "learning_rate": 4.9541008909632674e-05, "loss": 0.9981, "step": 22000 }, { "epoch": 2.55, "eval_bleu": 22.9493, "eval_gen_len": 18.7364, "eval_loss": 1.324600100517273, "eval_runtime": 1151.8793, "eval_samples_per_second": 14.964, "eval_steps_per_second": 0.936, "step": 22000 } ], "logging_steps": 500, "max_steps": 344760, "num_train_epochs": 40, "save_steps": 2000, "total_flos": 7.629337002738975e+17, "trial_name": null, "trial_params": null }