{ "best_metric": null, "best_model_checkpoint": null, "epoch": 1.0, "global_step": 60361, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.01, "learning_rate": 4.9585825284538036e-05, "loss": 2.5736, "step": 500 }, { "epoch": 0.02, "learning_rate": 4.917165056907606e-05, "loss": 2.4354, "step": 1000 }, { "epoch": 0.02, "learning_rate": 4.8757475853614095e-05, "loss": 2.4451, "step": 1500 }, { "epoch": 0.03, "learning_rate": 4.834330113815212e-05, "loss": 2.373, "step": 2000 }, { "epoch": 0.04, "learning_rate": 4.792912642269015e-05, "loss": 2.3647, "step": 2500 }, { "epoch": 0.05, "learning_rate": 4.7514951707228175e-05, "loss": 2.3462, "step": 3000 }, { "epoch": 0.06, "learning_rate": 4.710077699176621e-05, "loss": 2.3625, "step": 3500 }, { "epoch": 0.07, "learning_rate": 4.6686602276304234e-05, "loss": 2.3681, "step": 4000 }, { "epoch": 0.07, "learning_rate": 4.627242756084227e-05, "loss": 2.3366, "step": 4500 }, { "epoch": 0.08, "learning_rate": 4.5858252845380294e-05, "loss": 2.2811, "step": 5000 }, { "epoch": 0.08, "eval_gen_len": 19.0, "eval_loss": 2.1766555309295654, "eval_rouge1": 18.5624, "eval_rouge2": 6.8795, "eval_rougeL": 15.5361, "eval_rougeLsum": 16.6836, "eval_runtime": 1642.1189, "eval_samples_per_second": 2.042, "eval_steps_per_second": 2.042, "step": 5000 }, { "epoch": 0.09, "learning_rate": 4.544407812991833e-05, "loss": 2.3381, "step": 5500 }, { "epoch": 0.1, "learning_rate": 4.502990341445636e-05, "loss": 2.3223, "step": 6000 }, { "epoch": 0.11, "learning_rate": 4.461572869899439e-05, "loss": 2.345, "step": 6500 }, { "epoch": 0.12, "learning_rate": 4.420155398353242e-05, "loss": 2.3104, "step": 7000 }, { "epoch": 0.12, "learning_rate": 4.378737926807045e-05, "loss": 2.2813, "step": 7500 }, { "epoch": 0.13, "learning_rate": 4.337320455260847e-05, "loss": 2.2871, "step": 8000 }, { "epoch": 0.14, "learning_rate": 4.29590298371465e-05, "loss": 2.3268, "step": 8500 }, { "epoch": 0.15, "learning_rate": 4.254485512168453e-05, "loss": 2.2982, "step": 9000 }, { "epoch": 0.16, "learning_rate": 4.213068040622256e-05, "loss": 2.2911, "step": 9500 }, { "epoch": 0.17, "learning_rate": 4.171650569076059e-05, "loss": 2.2551, "step": 10000 }, { "epoch": 0.17, "eval_gen_len": 19.0, "eval_loss": 2.1327288150787354, "eval_rouge1": 19.077, "eval_rouge2": 6.8512, "eval_rougeL": 15.79, "eval_rougeLsum": 17.086, "eval_runtime": 1679.4791, "eval_samples_per_second": 1.996, "eval_steps_per_second": 1.996, "step": 10000 }, { "epoch": 0.17, "learning_rate": 4.130233097529862e-05, "loss": 2.2707, "step": 10500 }, { "epoch": 0.18, "learning_rate": 4.088815625983665e-05, "loss": 2.2714, "step": 11000 }, { "epoch": 0.19, "learning_rate": 4.047398154437468e-05, "loss": 2.2513, "step": 11500 }, { "epoch": 0.2, "learning_rate": 4.005980682891271e-05, "loss": 2.2548, "step": 12000 }, { "epoch": 0.21, "learning_rate": 3.9645632113450745e-05, "loss": 2.2642, "step": 12500 }, { "epoch": 0.22, "learning_rate": 3.923145739798877e-05, "loss": 2.2353, "step": 13000 }, { "epoch": 0.22, "learning_rate": 3.88172826825268e-05, "loss": 2.2724, "step": 13500 }, { "epoch": 0.23, "learning_rate": 3.8403107967064825e-05, "loss": 2.2777, "step": 14000 }, { "epoch": 0.24, "learning_rate": 3.798893325160286e-05, "loss": 2.2624, "step": 14500 }, { "epoch": 0.25, "learning_rate": 3.7574758536140885e-05, "loss": 2.2818, "step": 15000 }, { "epoch": 0.25, "eval_gen_len": 19.0, "eval_loss": 2.10290265083313, "eval_rouge1": 18.8637, "eval_rouge2": 6.9233, "eval_rougeL": 15.7341, "eval_rougeLsum": 16.9717, "eval_runtime": 1688.0673, "eval_samples_per_second": 1.986, "eval_steps_per_second": 1.986, "step": 15000 }, { "epoch": 0.26, "learning_rate": 3.716058382067892e-05, "loss": 2.261, "step": 15500 }, { "epoch": 0.27, "learning_rate": 3.6746409105216944e-05, "loss": 2.2517, "step": 16000 }, { "epoch": 0.27, "learning_rate": 3.633223438975498e-05, "loss": 2.2536, "step": 16500 }, { "epoch": 0.28, "learning_rate": 3.5918059674293004e-05, "loss": 2.2264, "step": 17000 }, { "epoch": 0.29, "learning_rate": 3.550388495883104e-05, "loss": 2.2226, "step": 17500 }, { "epoch": 0.3, "learning_rate": 3.508971024336907e-05, "loss": 2.2261, "step": 18000 }, { "epoch": 0.31, "learning_rate": 3.46755355279071e-05, "loss": 2.2704, "step": 18500 }, { "epoch": 0.31, "learning_rate": 3.4261360812445123e-05, "loss": 2.2111, "step": 19000 }, { "epoch": 0.32, "learning_rate": 3.384718609698315e-05, "loss": 2.1922, "step": 19500 }, { "epoch": 0.33, "learning_rate": 3.343301138152118e-05, "loss": 2.1952, "step": 20000 }, { "epoch": 0.33, "eval_gen_len": 19.0, "eval_loss": 2.080535411834717, "eval_rouge1": 18.962, "eval_rouge2": 7.1157, "eval_rougeL": 15.8297, "eval_rougeLsum": 17.0333, "eval_runtime": 1813.0594, "eval_samples_per_second": 1.849, "eval_steps_per_second": 1.849, "step": 20000 }, { "epoch": 0.34, "learning_rate": 3.301883666605921e-05, "loss": 2.2381, "step": 20500 }, { "epoch": 0.35, "learning_rate": 3.260466195059724e-05, "loss": 2.2409, "step": 21000 }, { "epoch": 0.36, "learning_rate": 3.219048723513527e-05, "loss": 2.2115, "step": 21500 }, { "epoch": 0.36, "learning_rate": 3.17763125196733e-05, "loss": 2.162, "step": 22000 }, { "epoch": 0.37, "learning_rate": 3.136213780421133e-05, "loss": 2.2058, "step": 22500 }, { "epoch": 0.38, "learning_rate": 3.094796308874936e-05, "loss": 2.1927, "step": 23000 }, { "epoch": 0.39, "learning_rate": 3.053378837328739e-05, "loss": 2.1773, "step": 23500 }, { "epoch": 0.4, "learning_rate": 3.011961365782542e-05, "loss": 2.2397, "step": 24000 }, { "epoch": 0.41, "learning_rate": 2.9705438942363452e-05, "loss": 2.1963, "step": 24500 }, { "epoch": 0.41, "learning_rate": 2.929126422690148e-05, "loss": 2.157, "step": 25000 }, { "epoch": 0.41, "eval_gen_len": 19.0, "eval_loss": 2.064121723175049, "eval_rouge1": 19.1418, "eval_rouge2": 7.315, "eval_rougeL": 16.05, "eval_rougeLsum": 17.2551, "eval_runtime": 1736.5991, "eval_samples_per_second": 1.931, "eval_steps_per_second": 1.931, "step": 25000 }, { "epoch": 0.42, "learning_rate": 2.8877089511439508e-05, "loss": 2.2139, "step": 25500 }, { "epoch": 0.43, "learning_rate": 2.8462914795977535e-05, "loss": 2.1927, "step": 26000 }, { "epoch": 0.44, "learning_rate": 2.8048740080515568e-05, "loss": 2.1869, "step": 26500 }, { "epoch": 0.45, "learning_rate": 2.7634565365053594e-05, "loss": 2.19, "step": 27000 }, { "epoch": 0.46, "learning_rate": 2.7220390649591628e-05, "loss": 2.1618, "step": 27500 }, { "epoch": 0.46, "learning_rate": 2.6806215934129654e-05, "loss": 2.2016, "step": 28000 }, { "epoch": 0.47, "learning_rate": 2.6392041218667684e-05, "loss": 2.1833, "step": 28500 }, { "epoch": 0.48, "learning_rate": 2.597786650320571e-05, "loss": 2.1977, "step": 29000 }, { "epoch": 0.49, "learning_rate": 2.5563691787743744e-05, "loss": 2.1588, "step": 29500 }, { "epoch": 0.5, "learning_rate": 2.5149517072281774e-05, "loss": 2.1775, "step": 30000 }, { "epoch": 0.5, "eval_gen_len": 19.0, "eval_loss": 2.045199394226074, "eval_rouge1": 19.2387, "eval_rouge2": 7.3193, "eval_rougeL": 16.0852, "eval_rougeLsum": 17.3563, "eval_runtime": 1710.3484, "eval_samples_per_second": 1.96, "eval_steps_per_second": 1.96, "step": 30000 }, { "epoch": 0.51, "learning_rate": 2.47353423568198e-05, "loss": 2.1526, "step": 30500 }, { "epoch": 0.51, "learning_rate": 2.432116764135783e-05, "loss": 2.2146, "step": 31000 }, { "epoch": 0.52, "learning_rate": 2.390699292589586e-05, "loss": 2.1939, "step": 31500 }, { "epoch": 0.53, "learning_rate": 2.349281821043389e-05, "loss": 2.1838, "step": 32000 }, { "epoch": 0.54, "learning_rate": 2.307864349497192e-05, "loss": 2.1743, "step": 32500 }, { "epoch": 0.55, "learning_rate": 2.266446877950995e-05, "loss": 2.1603, "step": 33000 }, { "epoch": 0.55, "learning_rate": 2.225029406404798e-05, "loss": 2.1586, "step": 33500 }, { "epoch": 0.56, "learning_rate": 2.183611934858601e-05, "loss": 2.1598, "step": 34000 }, { "epoch": 0.57, "learning_rate": 2.142194463312404e-05, "loss": 2.1904, "step": 34500 }, { "epoch": 0.58, "learning_rate": 2.100776991766207e-05, "loss": 2.1376, "step": 35000 }, { "epoch": 0.58, "eval_gen_len": 19.0, "eval_loss": 2.0308475494384766, "eval_rouge1": 19.291, "eval_rouge2": 7.363, "eval_rougeL": 16.1243, "eval_rougeLsum": 17.4151, "eval_runtime": 1709.1474, "eval_samples_per_second": 1.962, "eval_steps_per_second": 1.962, "step": 35000 }, { "epoch": 0.59, "learning_rate": 2.05935952022001e-05, "loss": 2.1989, "step": 35500 }, { "epoch": 0.6, "learning_rate": 2.0179420486738125e-05, "loss": 2.1756, "step": 36000 }, { "epoch": 0.6, "learning_rate": 1.9765245771276155e-05, "loss": 2.142, "step": 36500 }, { "epoch": 0.61, "learning_rate": 1.9351071055814185e-05, "loss": 2.1391, "step": 37000 }, { "epoch": 0.62, "learning_rate": 1.8936896340352215e-05, "loss": 2.1497, "step": 37500 }, { "epoch": 0.63, "learning_rate": 1.8522721624890245e-05, "loss": 2.1181, "step": 38000 }, { "epoch": 0.64, "learning_rate": 1.8108546909428274e-05, "loss": 2.1532, "step": 38500 }, { "epoch": 0.65, "learning_rate": 1.7694372193966304e-05, "loss": 2.158, "step": 39000 }, { "epoch": 0.65, "learning_rate": 1.7280197478504334e-05, "loss": 2.1634, "step": 39500 }, { "epoch": 0.66, "learning_rate": 1.6866022763042364e-05, "loss": 2.1853, "step": 40000 }, { "epoch": 0.66, "eval_gen_len": 19.0, "eval_loss": 2.0206809043884277, "eval_rouge1": 19.2808, "eval_rouge2": 7.4671, "eval_rougeL": 16.1593, "eval_rougeLsum": 17.3836, "eval_runtime": 1688.6655, "eval_samples_per_second": 1.986, "eval_steps_per_second": 1.986, "step": 40000 }, { "epoch": 0.67, "learning_rate": 1.6451848047580394e-05, "loss": 2.1599, "step": 40500 }, { "epoch": 0.68, "learning_rate": 1.603767333211842e-05, "loss": 2.0983, "step": 41000 }, { "epoch": 0.69, "learning_rate": 1.562349861665645e-05, "loss": 2.0822, "step": 41500 }, { "epoch": 0.7, "learning_rate": 1.520932390119448e-05, "loss": 2.1557, "step": 42000 }, { "epoch": 0.7, "learning_rate": 1.479514918573251e-05, "loss": 2.1406, "step": 42500 }, { "epoch": 0.71, "learning_rate": 1.438097447027054e-05, "loss": 2.178, "step": 43000 }, { "epoch": 0.72, "learning_rate": 1.3966799754808568e-05, "loss": 2.1386, "step": 43500 }, { "epoch": 0.73, "learning_rate": 1.3552625039346598e-05, "loss": 2.1576, "step": 44000 }, { "epoch": 0.74, "learning_rate": 1.3138450323884628e-05, "loss": 2.1772, "step": 44500 }, { "epoch": 0.75, "learning_rate": 1.272427560842266e-05, "loss": 2.1416, "step": 45000 }, { "epoch": 0.75, "eval_gen_len": 19.0, "eval_loss": 2.011291265487671, "eval_rouge1": 19.0414, "eval_rouge2": 7.3335, "eval_rougeL": 15.9747, "eval_rougeLsum": 17.1899, "eval_runtime": 1834.2332, "eval_samples_per_second": 1.828, "eval_steps_per_second": 1.828, "step": 45000 }, { "epoch": 0.75, "learning_rate": 1.2310100892960687e-05, "loss": 2.1276, "step": 45500 }, { "epoch": 0.76, "learning_rate": 1.1895926177498716e-05, "loss": 2.1313, "step": 46000 }, { "epoch": 0.77, "learning_rate": 1.1481751462036747e-05, "loss": 2.1463, "step": 46500 }, { "epoch": 0.78, "learning_rate": 1.1067576746574775e-05, "loss": 2.1642, "step": 47000 }, { "epoch": 0.79, "learning_rate": 1.0653402031112805e-05, "loss": 2.1381, "step": 47500 }, { "epoch": 0.8, "learning_rate": 1.0239227315650835e-05, "loss": 2.1237, "step": 48000 }, { "epoch": 0.8, "learning_rate": 9.825052600188863e-06, "loss": 2.1366, "step": 48500 }, { "epoch": 0.81, "learning_rate": 9.410877884726893e-06, "loss": 2.1087, "step": 49000 }, { "epoch": 0.82, "learning_rate": 8.996703169264925e-06, "loss": 2.1091, "step": 49500 }, { "epoch": 0.83, "learning_rate": 8.582528453802953e-06, "loss": 2.1245, "step": 50000 }, { "epoch": 0.83, "eval_gen_len": 19.0, "eval_loss": 2.005504846572876, "eval_rouge1": 19.1445, "eval_rouge2": 7.3715, "eval_rougeL": 16.0166, "eval_rougeLsum": 17.2621, "eval_runtime": 1732.2159, "eval_samples_per_second": 1.936, "eval_steps_per_second": 1.936, "step": 50000 }, { "epoch": 0.84, "learning_rate": 8.168353738340983e-06, "loss": 2.1439, "step": 50500 }, { "epoch": 0.84, "learning_rate": 7.75417902287901e-06, "loss": 2.1713, "step": 51000 }, { "epoch": 0.85, "learning_rate": 7.3400043074170406e-06, "loss": 2.1002, "step": 51500 }, { "epoch": 0.86, "learning_rate": 6.9258295919550704e-06, "loss": 2.124, "step": 52000 }, { "epoch": 0.87, "learning_rate": 6.5116548764931e-06, "loss": 2.1498, "step": 52500 }, { "epoch": 0.88, "learning_rate": 6.09748016103113e-06, "loss": 2.1701, "step": 53000 }, { "epoch": 0.89, "learning_rate": 5.683305445569159e-06, "loss": 2.1739, "step": 53500 }, { "epoch": 0.89, "learning_rate": 5.269130730107188e-06, "loss": 2.1223, "step": 54000 }, { "epoch": 0.9, "learning_rate": 4.854956014645219e-06, "loss": 2.1093, "step": 54500 }, { "epoch": 0.91, "learning_rate": 4.440781299183248e-06, "loss": 2.133, "step": 55000 }, { "epoch": 0.91, "eval_gen_len": 19.0, "eval_loss": 1.999653935432434, "eval_rouge1": 19.3033, "eval_rouge2": 7.4821, "eval_rougeL": 16.1413, "eval_rougeLsum": 17.3949, "eval_runtime": 1687.6858, "eval_samples_per_second": 1.987, "eval_steps_per_second": 1.987, "step": 55000 }, { "epoch": 0.92, "learning_rate": 4.026606583721277e-06, "loss": 2.1221, "step": 55500 }, { "epoch": 0.93, "learning_rate": 3.6124318682593068e-06, "loss": 2.0791, "step": 56000 }, { "epoch": 0.94, "learning_rate": 3.198257152797336e-06, "loss": 2.1306, "step": 56500 }, { "epoch": 0.94, "learning_rate": 2.784082437335366e-06, "loss": 2.1544, "step": 57000 }, { "epoch": 0.95, "learning_rate": 2.369907721873395e-06, "loss": 2.0999, "step": 57500 }, { "epoch": 0.96, "learning_rate": 1.955733006411425e-06, "loss": 2.1143, "step": 58000 }, { "epoch": 0.97, "learning_rate": 1.5415582909494541e-06, "loss": 2.1487, "step": 58500 }, { "epoch": 0.98, "learning_rate": 1.1273835754874836e-06, "loss": 2.1115, "step": 59000 }, { "epoch": 0.99, "learning_rate": 7.132088600255132e-07, "loss": 2.1244, "step": 59500 }, { "epoch": 0.99, "learning_rate": 2.9903414456354266e-07, "loss": 2.1191, "step": 60000 }, { "epoch": 0.99, "eval_gen_len": 19.0, "eval_loss": 1.9973467588424683, "eval_rouge1": 19.4044, "eval_rouge2": 7.5483, "eval_rougeL": 16.2429, "eval_rougeLsum": 17.488, "eval_runtime": 1792.5746, "eval_samples_per_second": 1.87, "eval_steps_per_second": 1.87, "step": 60000 }, { "epoch": 1.0, "step": 60361, "total_flos": 7.327167951508992e+16, "train_loss": 2.2034678317050873, "train_runtime": 36600.7281, "train_samples_per_second": 1.649, "train_steps_per_second": 1.649 } ], "max_steps": 60361, "num_train_epochs": 1, "total_flos": 7.327167951508992e+16, "trial_name": null, "trial_params": null }