{ "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": 3.6144, "step": 500 }, { "epoch": 0.02, "learning_rate": 4.917165056907606e-05, "loss": 3.4028, "step": 1000 }, { "epoch": 0.02, "learning_rate": 4.8757475853614095e-05, "loss": 3.3486, "step": 1500 }, { "epoch": 0.03, "learning_rate": 4.834330113815212e-05, "loss": 3.2449, "step": 2000 }, { "epoch": 0.04, "learning_rate": 4.792912642269015e-05, "loss": 3.2114, "step": 2500 }, { "epoch": 0.05, "learning_rate": 4.7514951707228175e-05, "loss": 3.1917, "step": 3000 }, { "epoch": 0.06, "learning_rate": 4.710077699176621e-05, "loss": 3.1649, "step": 3500 }, { "epoch": 0.07, "learning_rate": 4.6686602276304234e-05, "loss": 3.146, "step": 4000 }, { "epoch": 0.07, "learning_rate": 4.627242756084227e-05, "loss": 3.1357, "step": 4500 }, { "epoch": 0.08, "learning_rate": 4.5858252845380294e-05, "loss": 3.0567, "step": 5000 }, { "epoch": 0.08, "eval_gen_len": 19.997912317327767, "eval_loss": 2.886395215988159, "eval_rouge1": 18.9387, "eval_rouge2": 7.1014, "eval_rougeL": 15.4506, "eval_rougeLsum": 16.8377, "eval_runtime": 859.9366, "eval_samples_per_second": 3.899, "eval_steps_per_second": 3.899, "step": 5000 }, { "epoch": 0.09, "learning_rate": 4.544407812991833e-05, "loss": 3.1082, "step": 5500 }, { "epoch": 0.1, "learning_rate": 4.502990341445636e-05, "loss": 3.088, "step": 6000 }, { "epoch": 0.11, "learning_rate": 4.461572869899439e-05, "loss": 3.0837, "step": 6500 }, { "epoch": 0.12, "learning_rate": 4.420155398353242e-05, "loss": 3.0459, "step": 7000 }, { "epoch": 0.12, "learning_rate": 4.378737926807045e-05, "loss": 2.9948, "step": 7500 }, { "epoch": 0.13, "learning_rate": 4.337320455260847e-05, "loss": 2.9943, "step": 8000 }, { "epoch": 0.14, "learning_rate": 4.29590298371465e-05, "loss": 3.0398, "step": 8500 }, { "epoch": 0.15, "learning_rate": 4.254485512168453e-05, "loss": 2.9949, "step": 9000 }, { "epoch": 0.16, "learning_rate": 4.213068040622256e-05, "loss": 2.9824, "step": 9500 }, { "epoch": 0.17, "learning_rate": 4.171650569076059e-05, "loss": 2.9285, "step": 10000 }, { "epoch": 0.17, "eval_gen_len": 20.0, "eval_loss": 2.7799806594848633, "eval_rouge1": 19.8983, "eval_rouge2": 7.3258, "eval_rougeL": 16.0823, "eval_rougeLsum": 17.7019, "eval_runtime": 854.1045, "eval_samples_per_second": 3.926, "eval_steps_per_second": 3.926, "step": 10000 }, { "epoch": 0.17, "learning_rate": 4.130233097529862e-05, "loss": 2.9611, "step": 10500 }, { "epoch": 0.18, "learning_rate": 4.088815625983665e-05, "loss": 2.9401, "step": 11000 }, { "epoch": 0.19, "learning_rate": 4.047398154437468e-05, "loss": 2.9217, "step": 11500 }, { "epoch": 0.2, "learning_rate": 4.005980682891271e-05, "loss": 2.9318, "step": 12000 }, { "epoch": 0.21, "learning_rate": 3.9645632113450745e-05, "loss": 2.8956, "step": 12500 }, { "epoch": 0.22, "learning_rate": 3.923145739798877e-05, "loss": 2.902, "step": 13000 }, { "epoch": 0.22, "learning_rate": 3.88172826825268e-05, "loss": 2.8895, "step": 13500 }, { "epoch": 0.23, "learning_rate": 3.8403107967064825e-05, "loss": 2.9131, "step": 14000 }, { "epoch": 0.24, "learning_rate": 3.798893325160286e-05, "loss": 2.9017, "step": 14500 }, { "epoch": 0.25, "learning_rate": 3.7574758536140885e-05, "loss": 2.9252, "step": 15000 }, { "epoch": 0.25, "eval_gen_len": 20.0, "eval_loss": 2.7080156803131104, "eval_rouge1": 19.6623, "eval_rouge2": 7.4627, "eval_rougeL": 16.0153, "eval_rougeLsum": 17.4485, "eval_runtime": 853.6576, "eval_samples_per_second": 3.928, "eval_steps_per_second": 3.928, "step": 15000 }, { "epoch": 0.26, "learning_rate": 3.716058382067892e-05, "loss": 2.8971, "step": 15500 }, { "epoch": 0.27, "learning_rate": 3.6746409105216944e-05, "loss": 2.8831, "step": 16000 }, { "epoch": 0.27, "learning_rate": 3.633223438975498e-05, "loss": 2.8911, "step": 16500 }, { "epoch": 0.28, "learning_rate": 3.5918059674293004e-05, "loss": 2.8266, "step": 17000 }, { "epoch": 0.29, "learning_rate": 3.550388495883104e-05, "loss": 2.8465, "step": 17500 }, { "epoch": 0.3, "learning_rate": 3.508971024336907e-05, "loss": 2.8168, "step": 18000 }, { "epoch": 0.31, "learning_rate": 3.46755355279071e-05, "loss": 2.8807, "step": 18500 }, { "epoch": 0.31, "learning_rate": 3.4261360812445123e-05, "loss": 2.8521, "step": 19000 }, { "epoch": 0.32, "learning_rate": 3.384718609698315e-05, "loss": 2.8083, "step": 19500 }, { "epoch": 0.33, "learning_rate": 3.343301138152118e-05, "loss": 2.8123, "step": 20000 }, { "epoch": 0.33, "eval_gen_len": 20.0, "eval_loss": 2.658450126647949, "eval_rouge1": 19.7414, "eval_rouge2": 7.5251, "eval_rougeL": 15.8166, "eval_rougeLsum": 17.4668, "eval_runtime": 853.1021, "eval_samples_per_second": 3.93, "eval_steps_per_second": 3.93, "step": 20000 }, { "epoch": 0.34, "learning_rate": 3.301883666605921e-05, "loss": 2.8339, "step": 20500 }, { "epoch": 0.35, "learning_rate": 3.260466195059724e-05, "loss": 2.8566, "step": 21000 }, { "epoch": 0.36, "learning_rate": 3.219048723513527e-05, "loss": 2.804, "step": 21500 }, { "epoch": 0.36, "learning_rate": 3.17763125196733e-05, "loss": 2.7706, "step": 22000 }, { "epoch": 0.37, "learning_rate": 3.136213780421133e-05, "loss": 2.8014, "step": 22500 }, { "epoch": 0.38, "learning_rate": 3.094796308874936e-05, "loss": 2.7753, "step": 23000 }, { "epoch": 0.39, "learning_rate": 3.053378837328739e-05, "loss": 2.7491, "step": 23500 }, { "epoch": 0.4, "learning_rate": 3.011961365782542e-05, "loss": 2.8093, "step": 24000 }, { "epoch": 0.41, "learning_rate": 2.9705438942363452e-05, "loss": 2.7746, "step": 24500 }, { "epoch": 0.41, "learning_rate": 2.929126422690148e-05, "loss": 2.7117, "step": 25000 }, { "epoch": 0.41, "eval_gen_len": 20.0, "eval_loss": 2.607012987136841, "eval_rouge1": 19.7661, "eval_rouge2": 7.7193, "eval_rougeL": 16.2795, "eval_rougeLsum": 17.7884, "eval_runtime": 852.4977, "eval_samples_per_second": 3.933, "eval_steps_per_second": 3.933, "step": 25000 }, { "epoch": 0.42, "learning_rate": 2.8877089511439508e-05, "loss": 2.7638, "step": 25500 }, { "epoch": 0.43, "learning_rate": 2.8462914795977535e-05, "loss": 2.7299, "step": 26000 }, { "epoch": 0.44, "learning_rate": 2.8048740080515568e-05, "loss": 2.7383, "step": 26500 }, { "epoch": 0.45, "learning_rate": 2.7634565365053594e-05, "loss": 2.7634, "step": 27000 }, { "epoch": 0.46, "learning_rate": 2.7220390649591628e-05, "loss": 2.7074, "step": 27500 }, { "epoch": 0.46, "learning_rate": 2.6806215934129654e-05, "loss": 2.7268, "step": 28000 }, { "epoch": 0.47, "learning_rate": 2.6392041218667684e-05, "loss": 2.709, "step": 28500 }, { "epoch": 0.48, "learning_rate": 2.597786650320571e-05, "loss": 2.7451, "step": 29000 }, { "epoch": 0.49, "learning_rate": 2.5563691787743744e-05, "loss": 2.6797, "step": 29500 }, { "epoch": 0.5, "learning_rate": 2.5149517072281774e-05, "loss": 2.7131, "step": 30000 }, { "epoch": 0.5, "eval_gen_len": 20.0, "eval_loss": 2.5616209506988525, "eval_rouge1": 19.6706, "eval_rouge2": 7.4229, "eval_rougeL": 15.7998, "eval_rougeLsum": 17.4324, "eval_runtime": 853.8437, "eval_samples_per_second": 3.927, "eval_steps_per_second": 3.927, "step": 30000 }, { "epoch": 0.51, "learning_rate": 2.47353423568198e-05, "loss": 2.6996, "step": 30500 }, { "epoch": 0.51, "learning_rate": 2.432116764135783e-05, "loss": 2.7442, "step": 31000 }, { "epoch": 0.52, "learning_rate": 2.390699292589586e-05, "loss": 2.7131, "step": 31500 }, { "epoch": 0.53, "learning_rate": 2.349281821043389e-05, "loss": 2.7108, "step": 32000 }, { "epoch": 0.54, "learning_rate": 2.307864349497192e-05, "loss": 2.7011, "step": 32500 }, { "epoch": 0.55, "learning_rate": 2.266446877950995e-05, "loss": 2.6768, "step": 33000 }, { "epoch": 0.55, "learning_rate": 2.225029406404798e-05, "loss": 2.6671, "step": 33500 }, { "epoch": 0.56, "learning_rate": 2.183611934858601e-05, "loss": 2.6739, "step": 34000 }, { "epoch": 0.57, "learning_rate": 2.142194463312404e-05, "loss": 2.7291, "step": 34500 }, { "epoch": 0.58, "learning_rate": 2.100776991766207e-05, "loss": 2.6373, "step": 35000 }, { "epoch": 0.58, "eval_gen_len": 20.0, "eval_loss": 2.525042772293091, "eval_rouge1": 20.0155, "eval_rouge2": 7.6811, "eval_rougeL": 16.1231, "eval_rougeLsum": 17.7578, "eval_runtime": 855.0145, "eval_samples_per_second": 3.922, "eval_steps_per_second": 3.922, "step": 35000 }, { "epoch": 0.59, "learning_rate": 2.05935952022001e-05, "loss": 2.711, "step": 35500 }, { "epoch": 0.6, "learning_rate": 2.0179420486738125e-05, "loss": 2.663, "step": 36000 }, { "epoch": 0.6, "learning_rate": 1.9765245771276155e-05, "loss": 2.6486, "step": 36500 }, { "epoch": 0.61, "learning_rate": 1.9351071055814185e-05, "loss": 2.6442, "step": 37000 }, { "epoch": 0.62, "learning_rate": 1.8936896340352215e-05, "loss": 2.6583, "step": 37500 }, { "epoch": 0.63, "learning_rate": 1.8522721624890245e-05, "loss": 2.6227, "step": 38000 }, { "epoch": 0.64, "learning_rate": 1.8108546909428274e-05, "loss": 2.631, "step": 38500 }, { "epoch": 0.65, "learning_rate": 1.7694372193966304e-05, "loss": 2.6675, "step": 39000 }, { "epoch": 0.65, "learning_rate": 1.7280197478504334e-05, "loss": 2.6396, "step": 39500 }, { "epoch": 0.66, "learning_rate": 1.6866022763042364e-05, "loss": 2.6785, "step": 40000 }, { "epoch": 0.66, "eval_gen_len": 20.0, "eval_loss": 2.497694253921509, "eval_rouge1": 20.0974, "eval_rouge2": 7.9578, "eval_rougeL": 16.543, "eval_rougeLsum": 18.0242, "eval_runtime": 854.2144, "eval_samples_per_second": 3.925, "eval_steps_per_second": 3.925, "step": 40000 }, { "epoch": 0.67, "learning_rate": 1.6451848047580394e-05, "loss": 2.6492, "step": 40500 }, { "epoch": 0.68, "learning_rate": 1.603767333211842e-05, "loss": 2.5702, "step": 41000 }, { "epoch": 0.69, "learning_rate": 1.562349861665645e-05, "loss": 2.5583, "step": 41500 }, { "epoch": 0.7, "learning_rate": 1.520932390119448e-05, "loss": 2.6215, "step": 42000 }, { "epoch": 0.7, "learning_rate": 1.479514918573251e-05, "loss": 2.6198, "step": 42500 }, { "epoch": 0.71, "learning_rate": 1.438097447027054e-05, "loss": 2.6482, "step": 43000 }, { "epoch": 0.72, "learning_rate": 1.3966799754808568e-05, "loss": 2.6153, "step": 43500 }, { "epoch": 0.73, "learning_rate": 1.3552625039346598e-05, "loss": 2.6112, "step": 44000 }, { "epoch": 0.74, "learning_rate": 1.3138450323884628e-05, "loss": 2.6555, "step": 44500 }, { "epoch": 0.75, "learning_rate": 1.272427560842266e-05, "loss": 2.6265, "step": 45000 }, { "epoch": 0.75, "eval_gen_len": 20.0, "eval_loss": 2.4701449871063232, "eval_rouge1": 19.994, "eval_rouge2": 7.9114, "eval_rougeL": 16.3501, "eval_rougeLsum": 17.8786, "eval_runtime": 853.6465, "eval_samples_per_second": 3.928, "eval_steps_per_second": 3.928, "step": 45000 }, { "epoch": 0.75, "learning_rate": 1.2310100892960687e-05, "loss": 2.588, "step": 45500 }, { "epoch": 0.76, "learning_rate": 1.1895926177498716e-05, "loss": 2.5818, "step": 46000 }, { "epoch": 0.77, "learning_rate": 1.1481751462036747e-05, "loss": 2.5937, "step": 46500 }, { "epoch": 0.78, "learning_rate": 1.1067576746574775e-05, "loss": 2.6259, "step": 47000 }, { "epoch": 0.79, "learning_rate": 1.0653402031112805e-05, "loss": 2.6067, "step": 47500 }, { "epoch": 0.8, "learning_rate": 1.0239227315650835e-05, "loss": 2.5841, "step": 48000 }, { "epoch": 0.8, "learning_rate": 9.825052600188863e-06, "loss": 2.581, "step": 48500 }, { "epoch": 0.81, "learning_rate": 9.410877884726893e-06, "loss": 2.552, "step": 49000 }, { "epoch": 0.82, "learning_rate": 8.996703169264925e-06, "loss": 2.5549, "step": 49500 }, { "epoch": 0.83, "learning_rate": 8.582528453802953e-06, "loss": 2.5833, "step": 50000 }, { "epoch": 0.83, "eval_gen_len": 20.0, "eval_loss": 2.444098472595215, "eval_rouge1": 19.9981, "eval_rouge2": 7.934, "eval_rougeL": 16.3033, "eval_rougeLsum": 17.8674, "eval_runtime": 854.286, "eval_samples_per_second": 3.925, "eval_steps_per_second": 3.925, "step": 50000 }, { "epoch": 0.84, "learning_rate": 8.168353738340983e-06, "loss": 2.5983, "step": 50500 }, { "epoch": 0.84, "learning_rate": 7.75417902287901e-06, "loss": 2.6161, "step": 51000 }, { "epoch": 0.85, "learning_rate": 7.3400043074170406e-06, "loss": 2.5425, "step": 51500 }, { "epoch": 0.86, "learning_rate": 6.9258295919550704e-06, "loss": 2.5936, "step": 52000 }, { "epoch": 0.87, "learning_rate": 6.5116548764931e-06, "loss": 2.5658, "step": 52500 }, { "epoch": 0.88, "learning_rate": 6.09748016103113e-06, "loss": 2.6075, "step": 53000 }, { "epoch": 0.89, "learning_rate": 5.683305445569159e-06, "loss": 2.6051, "step": 53500 }, { "epoch": 0.89, "learning_rate": 5.269130730107188e-06, "loss": 2.542, "step": 54000 }, { "epoch": 0.9, "learning_rate": 4.854956014645219e-06, "loss": 2.5377, "step": 54500 }, { "epoch": 0.91, "learning_rate": 4.440781299183248e-06, "loss": 2.5579, "step": 55000 }, { "epoch": 0.91, "eval_gen_len": 20.0, "eval_loss": 2.4251036643981934, "eval_rouge1": 20.0544, "eval_rouge2": 7.8966, "eval_rougeL": 16.3889, "eval_rougeLsum": 17.9491, "eval_runtime": 854.2468, "eval_samples_per_second": 3.925, "eval_steps_per_second": 3.925, "step": 55000 }, { "epoch": 0.92, "learning_rate": 4.026606583721277e-06, "loss": 2.5384, "step": 55500 }, { "epoch": 0.93, "learning_rate": 3.6124318682593068e-06, "loss": 2.5131, "step": 56000 }, { "epoch": 0.94, "learning_rate": 3.198257152797336e-06, "loss": 2.5346, "step": 56500 }, { "epoch": 0.94, "learning_rate": 2.784082437335366e-06, "loss": 2.5782, "step": 57000 }, { "epoch": 0.95, "learning_rate": 2.369907721873395e-06, "loss": 2.523, "step": 57500 }, { "epoch": 0.96, "learning_rate": 1.955733006411425e-06, "loss": 2.5384, "step": 58000 }, { "epoch": 0.97, "learning_rate": 1.5415582909494541e-06, "loss": 2.5758, "step": 58500 }, { "epoch": 0.98, "learning_rate": 1.1273835754874836e-06, "loss": 2.5242, "step": 59000 }, { "epoch": 0.99, "learning_rate": 7.132088600255132e-07, "loss": 2.5284, "step": 59500 }, { "epoch": 0.99, "learning_rate": 2.9903414456354266e-07, "loss": 2.5242, "step": 60000 }, { "epoch": 0.99, "eval_gen_len": 20.0, "eval_loss": 2.409696340560913, "eval_rouge1": 20.1093, "eval_rouge2": 8.0572, "eval_rougeL": 16.4935, "eval_rougeLsum": 17.9823, "eval_runtime": 854.1023, "eval_samples_per_second": 3.926, "eval_steps_per_second": 3.926, "step": 60000 }, { "epoch": 1.0, "step": 60361, "total_flos": 3.65352746024448e+16, "train_loss": 2.76828055864371, "train_runtime": 19432.47, "train_samples_per_second": 3.106, "train_steps_per_second": 3.106 } ], "max_steps": 60361, "num_train_epochs": 1, "total_flos": 3.65352746024448e+16, "trial_name": null, "trial_params": null }