{ "best_metric": 2.3194692134857178, "best_model_checkpoint": "output/coldplay/checkpoint-376", "epoch": 8.0, "global_step": 376, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.1, "learning_rate": 0.00013355940688336427, "loss": 3.2771, "step": 5 }, { "epoch": 0.21, "learning_rate": 0.00012302403914397873, "loss": 2.8643, "step": 10 }, { "epoch": 0.31, "learning_rate": 0.00010671211798514472, "loss": 2.6223, "step": 15 }, { "epoch": 0.42, "learning_rate": 8.635498649403293e-05, "loss": 2.9051, "step": 20 }, { "epoch": 0.52, "learning_rate": 6.411334533481218e-05, "loss": 2.8302, "step": 25 }, { "epoch": 0.62, "learning_rate": 4.2347916539754844e-05, "loss": 2.8412, "step": 30 }, { "epoch": 0.73, "learning_rate": 2.3368877084135277e-05, "loss": 2.7828, "step": 35 }, { "epoch": 0.83, "learning_rate": 9.190657300387505e-06, "loss": 2.6878, "step": 40 }, { "epoch": 0.94, "learning_rate": 1.3181297643383925e-06, "loss": 2.57, "step": 45 }, { "epoch": 1.0, "eval_loss": 2.627840757369995, "eval_runtime": 3.4104, "eval_samples_per_second": 21.405, "eval_steps_per_second": 2.932, "step": 48 }, { "epoch": 1.06, "learning_rate": 1.3746270344901413e-06, "loss": 2.7251, "step": 50 }, { "epoch": 1.17, "learning_rate": 9.576451662754438e-06, "loss": 2.4207, "step": 55 }, { "epoch": 1.28, "learning_rate": 2.4309929383066146e-05, "loss": 2.7058, "step": 60 }, { "epoch": 1.38, "learning_rate": 4.3944626783346644e-05, "loss": 2.6908, "step": 65 }, { "epoch": 1.49, "learning_rate": 6.630773257727353e-05, "loss": 2.7345, "step": 70 }, { "epoch": 1.6, "learning_rate": 8.892450484875447e-05, "loss": 2.4105, "step": 75 }, { "epoch": 1.7, "learning_rate": 0.00010929213048843373, "loss": 2.5926, "step": 80 }, { "epoch": 1.81, "learning_rate": 0.00012515669103944476, "loss": 2.4909, "step": 85 }, { "epoch": 1.91, "learning_rate": 0.00013476258540873022, "loss": 2.5389, "step": 90 }, { "epoch": 2.0, "eval_loss": 2.446333646774292, "eval_runtime": 1.0636, "eval_samples_per_second": 76.156, "eval_steps_per_second": 10.342, "step": 94 }, { "epoch": 2.02, "learning_rate": 0.00013704680787354832, "loss": 2.3598, "step": 95 }, { "epoch": 2.13, "learning_rate": 0.00013175658222600302, "loss": 2.2124, "step": 100 }, { "epoch": 2.23, "learning_rate": 0.00011947733444744994, "loss": 2.1658, "step": 105 }, { "epoch": 2.34, "learning_rate": 0.0001015679084058065, "loss": 2.1519, "step": 110 }, { "epoch": 2.45, "learning_rate": 8.001019372440279e-05, "loss": 2.243, "step": 115 }, { "epoch": 2.55, "learning_rate": 5.718980627559731e-05, "loss": 2.2952, "step": 120 }, { "epoch": 2.66, "learning_rate": 3.563209159419354e-05, "loss": 2.2394, "step": 125 }, { "epoch": 2.77, "learning_rate": 1.772266555255008e-05, "loss": 2.2106, "step": 130 }, { "epoch": 2.87, "learning_rate": 5.443417773996978e-06, "loss": 2.2222, "step": 135 }, { "epoch": 2.98, "learning_rate": 1.5319212645169297e-07, "loss": 2.1897, "step": 140 }, { "epoch": 3.0, "eval_loss": 2.3795111179351807, "eval_runtime": 1.0634, "eval_samples_per_second": 76.169, "eval_steps_per_second": 10.344, "step": 141 }, { "epoch": 3.09, "learning_rate": 2.4374145912697595e-06, "loss": 2.0277, "step": 145 }, { "epoch": 3.19, "learning_rate": 1.204330896055522e-05, "loss": 2.0337, "step": 150 }, { "epoch": 3.3, "learning_rate": 2.790786951156628e-05, "loss": 2.0818, "step": 155 }, { "epoch": 3.4, "learning_rate": 4.8275495151245426e-05, "loss": 2.1041, "step": 160 }, { "epoch": 3.51, "learning_rate": 7.089226742272638e-05, "loss": 2.1708, "step": 165 }, { "epoch": 3.62, "learning_rate": 9.325537321665337e-05, "loss": 2.0552, "step": 170 }, { "epoch": 3.72, "learning_rate": 0.00011289007061693382, "loss": 2.1205, "step": 175 }, { "epoch": 3.83, "learning_rate": 0.00012762354833724553, "loss": 1.9704, "step": 180 }, { "epoch": 3.94, "learning_rate": 0.00013582537296550986, "loss": 1.9882, "step": 185 }, { "epoch": 4.0, "eval_loss": 2.3495991230010986, "eval_runtime": 1.0619, "eval_samples_per_second": 76.281, "eval_steps_per_second": 10.359, "step": 188 }, { "epoch": 4.04, "learning_rate": 0.0001365879156874179, "loss": 2.1933, "step": 190 }, { "epoch": 4.15, "learning_rate": 0.00012982679213998792, "loss": 1.8172, "step": 195 }, { "epoch": 4.26, "learning_rate": 0.00011629020053848047, "loss": 1.6238, "step": 200 }, { "epoch": 4.36, "learning_rate": 9.747612470258382e-05, "loss": 1.8408, "step": 205 }, { "epoch": 4.47, "learning_rate": 7.546656444541333e-05, "loss": 1.8867, "step": 210 }, { "epoch": 4.57, "learning_rate": 5.2697137691647635e-05, "loss": 2.0636, "step": 215 }, { "epoch": 4.68, "learning_rate": 3.1687550572992616e-05, "loss": 1.8887, "step": 220 }, { "epoch": 4.79, "learning_rate": 1.4762762169883855e-05, "loss": 1.9152, "step": 225 }, { "epoch": 4.89, "learning_rate": 3.795700315696817e-06, "loss": 1.8236, "step": 230 }, { "epoch": 5.0, "learning_rate": 0.0, "loss": 1.6923, "step": 235 }, { "epoch": 5.0, "eval_loss": 2.332808017730713, "eval_runtime": 1.0629, "eval_samples_per_second": 76.208, "eval_steps_per_second": 10.349, "step": 235 }, { "epoch": 5.11, "learning_rate": 3.7957003156967485e-06, "loss": 1.6147, "step": 240 }, { "epoch": 5.21, "learning_rate": 1.4762762169883802e-05, "loss": 1.7048, "step": 245 }, { "epoch": 5.32, "learning_rate": 3.168755057299255e-05, "loss": 1.5544, "step": 250 }, { "epoch": 5.43, "learning_rate": 5.269713769164743e-05, "loss": 1.5129, "step": 255 }, { "epoch": 5.53, "learning_rate": 7.546656444541325e-05, "loss": 1.7164, "step": 260 }, { "epoch": 5.64, "learning_rate": 9.747612470258363e-05, "loss": 1.6792, "step": 265 }, { "epoch": 5.74, "learning_rate": 0.0001162902005384805, "loss": 1.7173, "step": 270 }, { "epoch": 5.85, "learning_rate": 0.00012982679213998787, "loss": 1.7795, "step": 275 }, { "epoch": 5.96, "learning_rate": 0.00013658791568741792, "loss": 1.9717, "step": 280 }, { "epoch": 6.0, "eval_loss": 2.336082935333252, "eval_runtime": 1.0719, "eval_samples_per_second": 75.567, "eval_steps_per_second": 10.262, "step": 282 }, { "epoch": 6.06, "learning_rate": 0.00013582537296550986, "loss": 1.6967, "step": 285 }, { "epoch": 6.17, "learning_rate": 0.00012762354833724559, "loss": 1.4993, "step": 290 }, { "epoch": 6.28, "learning_rate": 0.0001128900706169339, "loss": 1.3665, "step": 295 }, { "epoch": 6.38, "learning_rate": 9.325537321665346e-05, "loss": 1.5098, "step": 300 }, { "epoch": 6.49, "learning_rate": 7.089226742272658e-05, "loss": 1.6949, "step": 305 }, { "epoch": 6.6, "learning_rate": 4.827549515124539e-05, "loss": 1.5465, "step": 310 }, { "epoch": 6.7, "learning_rate": 2.7907869511566348e-05, "loss": 1.506, "step": 315 }, { "epoch": 6.81, "learning_rate": 1.2043308960555334e-05, "loss": 1.6257, "step": 320 }, { "epoch": 6.91, "learning_rate": 2.437414591269752e-06, "loss": 1.3859, "step": 325 }, { "epoch": 7.0, "eval_loss": 2.334944009780884, "eval_runtime": 1.0709, "eval_samples_per_second": 75.638, "eval_steps_per_second": 10.272, "step": 329 }, { "epoch": 7.02, "learning_rate": 1.5319212645167772e-07, "loss": 1.5912, "step": 330 }, { "epoch": 7.13, "learning_rate": 5.443417773996994e-06, "loss": 1.4726, "step": 335 }, { "epoch": 7.23, "learning_rate": 1.772266555255011e-05, "loss": 1.2431, "step": 340 }, { "epoch": 7.34, "learning_rate": 3.563209159419346e-05, "loss": 1.3668, "step": 345 }, { "epoch": 7.45, "learning_rate": 5.718980627559723e-05, "loss": 1.3476, "step": 350 }, { "epoch": 7.55, "learning_rate": 8.001019372440265e-05, "loss": 1.2561, "step": 355 }, { "epoch": 7.66, "learning_rate": 0.00010156790840580641, "loss": 1.6616, "step": 360 }, { "epoch": 7.77, "learning_rate": 0.0001194773344474498, "loss": 1.4607, "step": 365 }, { "epoch": 7.87, "learning_rate": 0.00013175658222600294, "loss": 1.3461, "step": 370 }, { "epoch": 7.98, "learning_rate": 0.00013704680787354832, "loss": 1.3842, "step": 375 }, { "epoch": 8.0, "eval_loss": 2.3194692134857178, "eval_runtime": 1.0661, "eval_samples_per_second": 75.98, "eval_steps_per_second": 10.318, "step": 376 } ], "max_steps": 2867, "num_train_epochs": 61, "total_flos": 391938048000000.0, "trial_name": null, "trial_params": null }