{ "best_metric": null, "best_model_checkpoint": null, "epoch": 2.5600921541021373, "eval_steps": 500, "global_step": 10000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.03, "learning_rate": 0.0004957330602491893, "loss": 2.0441, "step": 100 }, { "epoch": 0.05, "learning_rate": 0.0004914661204983786, "loss": 1.713, "step": 200 }, { "epoch": 0.08, "learning_rate": 0.00048719918074756785, "loss": 1.6007, "step": 300 }, { "epoch": 0.1, "learning_rate": 0.00048293224099675715, "loss": 1.5181, "step": 400 }, { "epoch": 0.13, "learning_rate": 0.00047866530124594645, "loss": 1.4493, "step": 500 }, { "epoch": 0.13, "eval_bleu": 3.9997520454971136, "eval_loss": 1.3540712594985962, "eval_runtime": 10.3322, "eval_samples_per_second": 193.569, "eval_steps_per_second": 1.549, "step": 500 }, { "epoch": 0.15, "learning_rate": 0.0004743983614951357, "loss": 1.4096, "step": 600 }, { "epoch": 0.18, "learning_rate": 0.000470131421744325, "loss": 1.3729, "step": 700 }, { "epoch": 0.2, "learning_rate": 0.0004658644819935143, "loss": 1.3389, "step": 800 }, { "epoch": 0.23, "learning_rate": 0.00046159754224270354, "loss": 1.3166, "step": 900 }, { "epoch": 0.26, "learning_rate": 0.00045733060249189284, "loss": 1.2915, "step": 1000 }, { "epoch": 0.26, "eval_bleu": 5.393565315342523, "eval_loss": 1.2113043069839478, "eval_runtime": 8.5631, "eval_samples_per_second": 233.562, "eval_steps_per_second": 1.868, "step": 1000 }, { "epoch": 0.28, "learning_rate": 0.00045306366274108214, "loss": 1.2669, "step": 1100 }, { "epoch": 0.31, "learning_rate": 0.0004487967229902714, "loss": 1.2492, "step": 1200 }, { "epoch": 0.33, "learning_rate": 0.0004445297832394607, "loss": 1.2371, "step": 1300 }, { "epoch": 0.36, "learning_rate": 0.00044026284348864993, "loss": 1.2231, "step": 1400 }, { "epoch": 0.38, "learning_rate": 0.00043599590373783923, "loss": 1.2059, "step": 1500 }, { "epoch": 0.38, "eval_bleu": 5.838067918496751, "eval_loss": 1.1367273330688477, "eval_runtime": 9.2404, "eval_samples_per_second": 216.442, "eval_steps_per_second": 1.732, "step": 1500 }, { "epoch": 0.41, "learning_rate": 0.0004317289639870285, "loss": 1.1931, "step": 1600 }, { "epoch": 0.44, "learning_rate": 0.0004274620242362178, "loss": 1.1761, "step": 1700 }, { "epoch": 0.46, "learning_rate": 0.0004231950844854071, "loss": 1.1792, "step": 1800 }, { "epoch": 0.49, "learning_rate": 0.0004189281447345963, "loss": 1.1614, "step": 1900 }, { "epoch": 0.51, "learning_rate": 0.0004146612049837856, "loss": 1.1573, "step": 2000 }, { "epoch": 0.51, "eval_bleu": 6.242227124133382, "eval_loss": 1.0900604724884033, "eval_runtime": 8.3953, "eval_samples_per_second": 238.228, "eval_steps_per_second": 1.906, "step": 2000 }, { "epoch": 0.54, "learning_rate": 0.0004103942652329749, "loss": 1.1449, "step": 2100 }, { "epoch": 0.56, "learning_rate": 0.00040612732548216417, "loss": 1.1379, "step": 2200 }, { "epoch": 0.59, "learning_rate": 0.00040186038573135347, "loss": 1.1281, "step": 2300 }, { "epoch": 0.61, "learning_rate": 0.00039759344598054277, "loss": 1.1226, "step": 2400 }, { "epoch": 0.64, "learning_rate": 0.00039332650622973206, "loss": 1.1121, "step": 2500 }, { "epoch": 0.64, "eval_bleu": 6.627148578471305, "eval_loss": 1.0542418956756592, "eval_runtime": 9.2835, "eval_samples_per_second": 215.436, "eval_steps_per_second": 1.723, "step": 2500 }, { "epoch": 0.67, "learning_rate": 0.0003890595664789213, "loss": 1.1076, "step": 2600 }, { "epoch": 0.69, "learning_rate": 0.0003847926267281106, "loss": 1.1001, "step": 2700 }, { "epoch": 0.72, "learning_rate": 0.0003805256869772999, "loss": 1.0996, "step": 2800 }, { "epoch": 0.74, "learning_rate": 0.00037625874722648915, "loss": 1.088, "step": 2900 }, { "epoch": 0.77, "learning_rate": 0.00037199180747567845, "loss": 1.0867, "step": 3000 }, { "epoch": 0.77, "eval_bleu": 6.879580310573366, "eval_loss": 1.025155782699585, "eval_runtime": 8.5706, "eval_samples_per_second": 233.355, "eval_steps_per_second": 1.867, "step": 3000 }, { "epoch": 0.79, "learning_rate": 0.00036772486772486775, "loss": 1.0784, "step": 3100 }, { "epoch": 0.82, "learning_rate": 0.000363457927974057, "loss": 1.0794, "step": 3200 }, { "epoch": 0.84, "learning_rate": 0.0003591909882232463, "loss": 1.0736, "step": 3300 }, { "epoch": 0.87, "learning_rate": 0.0003549240484724356, "loss": 1.0684, "step": 3400 }, { "epoch": 0.9, "learning_rate": 0.0003506571087216249, "loss": 1.0623, "step": 3500 }, { "epoch": 0.9, "eval_bleu": 7.039305128716119, "eval_loss": 1.0067821741104126, "eval_runtime": 9.3223, "eval_samples_per_second": 214.539, "eval_steps_per_second": 1.716, "step": 3500 }, { "epoch": 0.92, "learning_rate": 0.00034639016897081414, "loss": 1.0609, "step": 3600 }, { "epoch": 0.95, "learning_rate": 0.00034212322922000344, "loss": 1.0579, "step": 3700 }, { "epoch": 0.97, "learning_rate": 0.00033785628946919274, "loss": 1.0517, "step": 3800 }, { "epoch": 1.0, "learning_rate": 0.000333589349718382, "loss": 1.054, "step": 3900 }, { "epoch": 1.02, "learning_rate": 0.0003293224099675713, "loss": 1.0408, "step": 4000 }, { "epoch": 1.02, "eval_bleu": 7.2660174031875915, "eval_loss": 0.9882246255874634, "eval_runtime": 8.4135, "eval_samples_per_second": 237.713, "eval_steps_per_second": 1.902, "step": 4000 }, { "epoch": 1.05, "learning_rate": 0.0003250554702167606, "loss": 1.0317, "step": 4100 }, { "epoch": 1.08, "learning_rate": 0.0003207885304659498, "loss": 1.0344, "step": 4200 }, { "epoch": 1.1, "learning_rate": 0.0003165215907151391, "loss": 1.0274, "step": 4300 }, { "epoch": 1.13, "learning_rate": 0.0003122546509643284, "loss": 1.0216, "step": 4400 }, { "epoch": 1.15, "learning_rate": 0.0003079877112135177, "loss": 1.0203, "step": 4500 }, { "epoch": 1.15, "eval_bleu": 7.055297387993883, "eval_loss": 0.9723305702209473, "eval_runtime": 9.3412, "eval_samples_per_second": 214.105, "eval_steps_per_second": 1.713, "step": 4500 }, { "epoch": 1.18, "learning_rate": 0.0003037207714627069, "loss": 1.0217, "step": 4600 }, { "epoch": 1.2, "learning_rate": 0.0002994538317118962, "loss": 1.0184, "step": 4700 }, { "epoch": 1.23, "learning_rate": 0.0002951868919610855, "loss": 1.015, "step": 4800 }, { "epoch": 1.25, "learning_rate": 0.00029091995221027477, "loss": 1.011, "step": 4900 }, { "epoch": 1.28, "learning_rate": 0.00028665301245946406, "loss": 1.0054, "step": 5000 }, { "epoch": 1.28, "eval_bleu": 7.455525040183996, "eval_loss": 0.9624391198158264, "eval_runtime": 8.4826, "eval_samples_per_second": 235.778, "eval_steps_per_second": 1.886, "step": 5000 }, { "epoch": 1.31, "learning_rate": 0.00028238607270865336, "loss": 1.0062, "step": 5100 }, { "epoch": 1.33, "learning_rate": 0.0002781191329578426, "loss": 1.0081, "step": 5200 }, { "epoch": 1.36, "learning_rate": 0.0002738521932070319, "loss": 1.0007, "step": 5300 }, { "epoch": 1.38, "learning_rate": 0.0002695852534562212, "loss": 0.9973, "step": 5400 }, { "epoch": 1.41, "learning_rate": 0.0002653183137054105, "loss": 0.9977, "step": 5500 }, { "epoch": 1.41, "eval_bleu": 7.426000467701662, "eval_loss": 0.9526358842849731, "eval_runtime": 9.3659, "eval_samples_per_second": 213.542, "eval_steps_per_second": 1.708, "step": 5500 }, { "epoch": 1.43, "learning_rate": 0.00026105137395459975, "loss": 0.9982, "step": 5600 }, { "epoch": 1.46, "learning_rate": 0.00025678443420378905, "loss": 0.9956, "step": 5700 }, { "epoch": 1.48, "learning_rate": 0.00025251749445297835, "loss": 0.9941, "step": 5800 }, { "epoch": 1.51, "learning_rate": 0.0002482505547021676, "loss": 0.9881, "step": 5900 }, { "epoch": 1.54, "learning_rate": 0.0002439836149513569, "loss": 0.9931, "step": 6000 }, { "epoch": 1.54, "eval_bleu": 7.523072732517067, "eval_loss": 0.9395684003829956, "eval_runtime": 9.0823, "eval_samples_per_second": 220.209, "eval_steps_per_second": 1.762, "step": 6000 }, { "epoch": 1.56, "learning_rate": 0.0002397166752005462, "loss": 0.9834, "step": 6100 }, { "epoch": 1.59, "learning_rate": 0.00023544973544973544, "loss": 0.9821, "step": 6200 }, { "epoch": 1.61, "learning_rate": 0.00023118279569892471, "loss": 0.9859, "step": 6300 }, { "epoch": 1.64, "learning_rate": 0.00022691585594811401, "loss": 0.9762, "step": 6400 }, { "epoch": 1.66, "learning_rate": 0.0002226489161973033, "loss": 0.9804, "step": 6500 }, { "epoch": 1.66, "eval_bleu": 7.43758431482021, "eval_loss": 0.9323887825012207, "eval_runtime": 9.0344, "eval_samples_per_second": 221.376, "eval_steps_per_second": 1.771, "step": 6500 }, { "epoch": 1.69, "learning_rate": 0.00021838197644649259, "loss": 0.9794, "step": 6600 }, { "epoch": 1.72, "learning_rate": 0.00021411503669568186, "loss": 0.9737, "step": 6700 }, { "epoch": 1.74, "learning_rate": 0.00020984809694487113, "loss": 0.9696, "step": 6800 }, { "epoch": 1.77, "learning_rate": 0.00020558115719406043, "loss": 0.9735, "step": 6900 }, { "epoch": 1.79, "learning_rate": 0.0002013142174432497, "loss": 0.9691, "step": 7000 }, { "epoch": 1.79, "eval_bleu": 7.52270860836675, "eval_loss": 0.9264442324638367, "eval_runtime": 9.3377, "eval_samples_per_second": 214.186, "eval_steps_per_second": 1.713, "step": 7000 }, { "epoch": 1.82, "learning_rate": 0.000197047277692439, "loss": 0.9754, "step": 7100 }, { "epoch": 1.84, "learning_rate": 0.00019278033794162827, "loss": 0.9688, "step": 7200 }, { "epoch": 1.87, "learning_rate": 0.00018851339819081755, "loss": 0.968, "step": 7300 }, { "epoch": 1.89, "learning_rate": 0.00018424645844000685, "loss": 0.9628, "step": 7400 }, { "epoch": 1.92, "learning_rate": 0.00017997951868919612, "loss": 0.9645, "step": 7500 }, { "epoch": 1.92, "eval_bleu": 7.685904345266889, "eval_loss": 0.9192501306533813, "eval_runtime": 8.7414, "eval_samples_per_second": 228.796, "eval_steps_per_second": 1.83, "step": 7500 }, { "epoch": 1.95, "learning_rate": 0.0001757125789383854, "loss": 0.9674, "step": 7600 }, { "epoch": 1.97, "learning_rate": 0.00017144563918757466, "loss": 0.9659, "step": 7700 }, { "epoch": 2.0, "learning_rate": 0.00016717869943676394, "loss": 0.9646, "step": 7800 }, { "epoch": 2.02, "learning_rate": 0.00016291175968595324, "loss": 0.9477, "step": 7900 }, { "epoch": 2.05, "learning_rate": 0.0001586448199351425, "loss": 0.9509, "step": 8000 }, { "epoch": 2.05, "eval_bleu": 7.647325378973634, "eval_loss": 0.9144095778465271, "eval_runtime": 9.1538, "eval_samples_per_second": 218.489, "eval_steps_per_second": 1.748, "step": 8000 }, { "epoch": 2.07, "learning_rate": 0.00024078341013824886, "loss": 0.9532, "step": 8100 }, { "epoch": 2.1, "learning_rate": 0.00023758320532514082, "loss": 0.949, "step": 8200 }, { "epoch": 2.12, "learning_rate": 0.00023438300051203275, "loss": 0.9506, "step": 8300 }, { "epoch": 2.15, "learning_rate": 0.00023118279569892471, "loss": 0.9539, "step": 8400 }, { "epoch": 2.18, "learning_rate": 0.00022798259088581668, "loss": 0.9485, "step": 8500 }, { "epoch": 2.18, "eval_bleu": 7.654834398763605, "eval_loss": 0.9117684364318848, "eval_runtime": 11.8365, "eval_samples_per_second": 168.969, "eval_steps_per_second": 1.352, "step": 8500 }, { "epoch": 2.2, "learning_rate": 0.00022478238607270864, "loss": 0.9495, "step": 8600 }, { "epoch": 2.23, "learning_rate": 0.00022158218125960063, "loss": 0.9484, "step": 8700 }, { "epoch": 2.25, "learning_rate": 0.00021838197644649259, "loss": 0.9486, "step": 8800 }, { "epoch": 2.28, "learning_rate": 0.00021518177163338455, "loss": 0.9445, "step": 8900 }, { "epoch": 2.3, "learning_rate": 0.0002119815668202765, "loss": 0.9437, "step": 9000 }, { "epoch": 2.3, "eval_bleu": 7.606584348844775, "eval_loss": 0.9072746634483337, "eval_runtime": 9.3821, "eval_samples_per_second": 213.171, "eval_steps_per_second": 1.705, "step": 9000 }, { "epoch": 2.33, "learning_rate": 0.00020878136200716847, "loss": 0.9406, "step": 9100 }, { "epoch": 2.36, "learning_rate": 0.00020558115719406043, "loss": 0.9418, "step": 9200 }, { "epoch": 2.38, "learning_rate": 0.0002023809523809524, "loss": 0.9402, "step": 9300 }, { "epoch": 2.41, "learning_rate": 0.00019918074756784435, "loss": 0.94, "step": 9400 }, { "epoch": 2.43, "learning_rate": 0.00019598054275473631, "loss": 0.9393, "step": 9500 }, { "epoch": 2.43, "eval_bleu": 7.714033503089272, "eval_loss": 0.9019351601600647, "eval_runtime": 9.2531, "eval_samples_per_second": 216.144, "eval_steps_per_second": 1.729, "step": 9500 }, { "epoch": 2.46, "learning_rate": 0.00019278033794162827, "loss": 0.9378, "step": 9600 }, { "epoch": 2.48, "learning_rate": 0.00018958013312852024, "loss": 0.9345, "step": 9700 }, { "epoch": 2.51, "learning_rate": 0.0001863799283154122, "loss": 0.936, "step": 9800 }, { "epoch": 2.53, "learning_rate": 0.00018317972350230416, "loss": 0.9389, "step": 9900 }, { "epoch": 2.56, "learning_rate": 0.00017997951868919612, "loss": 0.9336, "step": 10000 }, { "epoch": 2.56, "eval_bleu": 7.80952091204729, "eval_loss": 0.897022545337677, "eval_runtime": 9.0616, "eval_samples_per_second": 220.711, "eval_steps_per_second": 1.766, "step": 10000 } ], "logging_steps": 100, "max_steps": 15624, "num_train_epochs": 4, "save_steps": 2000, "total_flos": 2.921158230933504e+16, "trial_name": null, "trial_params": null }