farleyknight's picture
End of training
97084b1
{
"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
}