{ "best_metric": 1.8054813146591187, "best_model_checkpoint": "./lora-redpajama/checkpoint-600", "epoch": 2.745995423340961, "global_step": 600, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.05, "learning_rate": 2.9999999999999997e-05, "loss": 3.6041, "step": 10 }, { "epoch": 0.09, "learning_rate": 5.9999999999999995e-05, "loss": 3.462, "step": 20 }, { "epoch": 0.14, "learning_rate": 8.999999999999999e-05, "loss": 3.0984, "step": 30 }, { "epoch": 0.18, "learning_rate": 0.00011999999999999999, "loss": 2.5603, "step": 40 }, { "epoch": 0.23, "learning_rate": 0.00015, "loss": 2.2705, "step": 50 }, { "epoch": 0.27, "learning_rate": 0.00017999999999999998, "loss": 2.2129, "step": 60 }, { "epoch": 0.32, "learning_rate": 0.00020999999999999998, "loss": 2.1456, "step": 70 }, { "epoch": 0.37, "learning_rate": 0.00023999999999999998, "loss": 2.0951, "step": 80 }, { "epoch": 0.41, "learning_rate": 0.00027, "loss": 2.0716, "step": 90 }, { "epoch": 0.46, "learning_rate": 0.0003, "loss": 2.0437, "step": 100 }, { "epoch": 0.5, "learning_rate": 0.0002945848375451263, "loss": 1.9825, "step": 110 }, { "epoch": 0.55, "learning_rate": 0.00028916967509025267, "loss": 2.0074, "step": 120 }, { "epoch": 0.59, "learning_rate": 0.000283754512635379, "loss": 1.9807, "step": 130 }, { "epoch": 0.64, "learning_rate": 0.00027833935018050537, "loss": 1.9581, "step": 140 }, { "epoch": 0.69, "learning_rate": 0.00027292418772563177, "loss": 1.947, "step": 150 }, { "epoch": 0.73, "learning_rate": 0.0002675090252707581, "loss": 1.9234, "step": 160 }, { "epoch": 0.78, "learning_rate": 0.00026209386281588447, "loss": 1.9434, "step": 170 }, { "epoch": 0.82, "learning_rate": 0.0002566787003610108, "loss": 1.8722, "step": 180 }, { "epoch": 0.87, "learning_rate": 0.00025126353790613716, "loss": 1.8902, "step": 190 }, { "epoch": 0.92, "learning_rate": 0.0002458483754512635, "loss": 1.8855, "step": 200 }, { "epoch": 0.92, "eval_loss": 1.9148070812225342, "eval_runtime": 83.9113, "eval_samples_per_second": 23.835, "eval_steps_per_second": 0.596, "step": 200 }, { "epoch": 0.96, "learning_rate": 0.00024043321299638986, "loss": 1.9064, "step": 210 }, { "epoch": 1.01, "learning_rate": 0.00023501805054151624, "loss": 1.9195, "step": 220 }, { "epoch": 1.05, "learning_rate": 0.00022960288808664258, "loss": 1.875, "step": 230 }, { "epoch": 1.1, "learning_rate": 0.00022418772563176893, "loss": 1.8733, "step": 240 }, { "epoch": 1.14, "learning_rate": 0.0002187725631768953, "loss": 1.8652, "step": 250 }, { "epoch": 1.19, "learning_rate": 0.00021335740072202166, "loss": 1.8704, "step": 260 }, { "epoch": 1.24, "learning_rate": 0.00020794223826714798, "loss": 1.8524, "step": 270 }, { "epoch": 1.28, "learning_rate": 0.00020252707581227435, "loss": 1.8322, "step": 280 }, { "epoch": 1.33, "learning_rate": 0.0001971119133574007, "loss": 1.8383, "step": 290 }, { "epoch": 1.37, "learning_rate": 0.00019169675090252705, "loss": 1.846, "step": 300 }, { "epoch": 1.42, "learning_rate": 0.00018628158844765343, "loss": 1.8283, "step": 310 }, { "epoch": 1.46, "learning_rate": 0.00018086642599277977, "loss": 1.8369, "step": 320 }, { "epoch": 1.51, "learning_rate": 0.00017545126353790612, "loss": 1.8398, "step": 330 }, { "epoch": 1.56, "learning_rate": 0.00017003610108303247, "loss": 1.8203, "step": 340 }, { "epoch": 1.6, "learning_rate": 0.00016462093862815885, "loss": 1.8132, "step": 350 }, { "epoch": 1.65, "learning_rate": 0.0001592057761732852, "loss": 1.7873, "step": 360 }, { "epoch": 1.69, "learning_rate": 0.00015379061371841152, "loss": 1.8322, "step": 370 }, { "epoch": 1.74, "learning_rate": 0.0001483754512635379, "loss": 1.809, "step": 380 }, { "epoch": 1.78, "learning_rate": 0.00014296028880866424, "loss": 1.8377, "step": 390 }, { "epoch": 1.83, "learning_rate": 0.00013754512635379062, "loss": 1.8083, "step": 400 }, { "epoch": 1.83, "eval_loss": 1.8361974954605103, "eval_runtime": 81.4531, "eval_samples_per_second": 24.554, "eval_steps_per_second": 0.614, "step": 400 }, { "epoch": 1.88, "learning_rate": 0.00013212996389891696, "loss": 1.8169, "step": 410 }, { "epoch": 1.92, "learning_rate": 0.0001267148014440433, "loss": 1.7913, "step": 420 }, { "epoch": 1.97, "learning_rate": 0.00012129963898916966, "loss": 1.7949, "step": 430 }, { "epoch": 2.01, "learning_rate": 0.00011588447653429602, "loss": 1.7821, "step": 440 }, { "epoch": 2.06, "learning_rate": 0.00011046931407942237, "loss": 1.7765, "step": 450 }, { "epoch": 2.11, "learning_rate": 0.00010505415162454872, "loss": 1.7977, "step": 460 }, { "epoch": 2.15, "learning_rate": 9.963898916967508e-05, "loss": 1.7647, "step": 470 }, { "epoch": 2.2, "learning_rate": 9.422382671480144e-05, "loss": 1.7766, "step": 480 }, { "epoch": 2.24, "learning_rate": 8.880866425992779e-05, "loss": 1.78, "step": 490 }, { "epoch": 2.29, "learning_rate": 8.339350180505415e-05, "loss": 1.7828, "step": 500 }, { "epoch": 2.33, "learning_rate": 7.797833935018049e-05, "loss": 1.7468, "step": 510 }, { "epoch": 2.38, "learning_rate": 7.256317689530685e-05, "loss": 1.7982, "step": 520 }, { "epoch": 2.43, "learning_rate": 6.71480144404332e-05, "loss": 1.7885, "step": 530 }, { "epoch": 2.47, "learning_rate": 6.173285198555956e-05, "loss": 1.7629, "step": 540 }, { "epoch": 2.52, "learning_rate": 5.6317689530685916e-05, "loss": 1.7515, "step": 550 }, { "epoch": 2.56, "learning_rate": 5.090252707581227e-05, "loss": 1.731, "step": 560 }, { "epoch": 2.61, "learning_rate": 4.548736462093862e-05, "loss": 1.7954, "step": 570 }, { "epoch": 2.65, "learning_rate": 4.007220216606498e-05, "loss": 1.7446, "step": 580 }, { "epoch": 2.7, "learning_rate": 3.4657039711191336e-05, "loss": 1.7634, "step": 590 }, { "epoch": 2.75, "learning_rate": 2.9241877256317685e-05, "loss": 1.7523, "step": 600 }, { "epoch": 2.75, "eval_loss": 1.8054813146591187, "eval_runtime": 79.1242, "eval_samples_per_second": 25.277, "eval_steps_per_second": 0.632, "step": 600 } ], "max_steps": 654, "num_train_epochs": 3, "total_flos": 6.284389017861489e+17, "trial_name": null, "trial_params": null }