| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.3893002975984497, | |
| "eval_steps": 100000000, | |
| "global_step": 54000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.0, | |
| "learning_rate": 1.0000000000000001e-07, | |
| "loss": 10.7586, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "learning_rate": 5e-05, | |
| "loss": 6.5292, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "learning_rate": 4.99998214600225e-05, | |
| "loss": 4.7262, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "learning_rate": 4.9999285842640105e-05, | |
| "loss": 4.1997, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "learning_rate": 4.999839315550315e-05, | |
| "loss": 3.8892, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "learning_rate": 4.9997143411362066e-05, | |
| "loss": 3.6618, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "learning_rate": 4.9995536628067196e-05, | |
| "loss": 3.5132, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "learning_rate": 4.9993572828568536e-05, | |
| "loss": 3.4092, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "learning_rate": 4.999125204091544e-05, | |
| "loss": 3.3295, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "learning_rate": 4.998857429825615e-05, | |
| "loss": 3.2596, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "learning_rate": 4.9985539638837424e-05, | |
| "loss": 3.2068, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "learning_rate": 4.998214810600389e-05, | |
| "loss": 3.163, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "learning_rate": 4.997839974819749e-05, | |
| "loss": 3.1256, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "learning_rate": 4.997429461895675e-05, | |
| "loss": 3.0844, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "learning_rate": 4.996983277691606e-05, | |
| "loss": 3.0541, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "learning_rate": 4.996501428580478e-05, | |
| "loss": 3.0293, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "learning_rate": 4.995983921444638e-05, | |
| "loss": 2.999, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "learning_rate": 4.9954307636757434e-05, | |
| "loss": 2.9744, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "learning_rate": 4.994841963174656e-05, | |
| "loss": 2.9525, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "learning_rate": 4.9942175283513294e-05, | |
| "loss": 2.9321, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "learning_rate": 4.993557468124691e-05, | |
| "loss": 2.9237, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "learning_rate": 4.9928617919225106e-05, | |
| "loss": 2.9021, | |
| "step": 10500 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "learning_rate": 4.99213050968127e-05, | |
| "loss": 2.8766, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "learning_rate": 4.991363631846019e-05, | |
| "loss": 2.8691, | |
| "step": 11500 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "learning_rate": 4.990561169370224e-05, | |
| "loss": 2.8531, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "learning_rate": 4.989723133715618e-05, | |
| "loss": 2.8366, | |
| "step": 12500 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "learning_rate": 4.988849536852028e-05, | |
| "loss": 2.8265, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "learning_rate": 4.987940391257212e-05, | |
| "loss": 2.8099, | |
| "step": 13500 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "learning_rate": 4.986995709916677e-05, | |
| "loss": 2.8009, | |
| "step": 14000 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "learning_rate": 4.9860155063234944e-05, | |
| "loss": 2.7881, | |
| "step": 14500 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "learning_rate": 4.9849997944781055e-05, | |
| "loss": 2.7794, | |
| "step": 15000 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "learning_rate": 4.9839485888881235e-05, | |
| "loss": 2.7674, | |
| "step": 15500 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 4.982861904568127e-05, | |
| "loss": 2.7559, | |
| "step": 16000 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 4.981739757039443e-05, | |
| "loss": 2.7455, | |
| "step": 16500 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 4.9805821623299285e-05, | |
| "loss": 2.7351, | |
| "step": 17000 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "learning_rate": 4.979389136973737e-05, | |
| "loss": 2.7317, | |
| "step": 17500 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "learning_rate": 4.978160698011085e-05, | |
| "loss": 2.7192, | |
| "step": 18000 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "learning_rate": 4.976896862988012e-05, | |
| "loss": 2.709, | |
| "step": 18500 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "learning_rate": 4.975597649956122e-05, | |
| "loss": 2.7043, | |
| "step": 19000 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "learning_rate": 4.974263077472334e-05, | |
| "loss": 2.6954, | |
| "step": 19500 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "learning_rate": 4.97289316459861e-05, | |
| "loss": 2.6905, | |
| "step": 20000 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "learning_rate": 4.9714879309016874e-05, | |
| "loss": 2.6782, | |
| "step": 20500 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "learning_rate": 4.970047396452798e-05, | |
| "loss": 2.6751, | |
| "step": 21000 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "learning_rate": 4.968571581827381e-05, | |
| "loss": 2.6659, | |
| "step": 21500 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "learning_rate": 4.9670605081047886e-05, | |
| "loss": 2.6594, | |
| "step": 22000 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "learning_rate": 4.9655141968679866e-05, | |
| "loss": 2.6553, | |
| "step": 22500 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "learning_rate": 4.963932670203245e-05, | |
| "loss": 2.6499, | |
| "step": 23000 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "learning_rate": 4.962315950699822e-05, | |
| "loss": 2.6391, | |
| "step": 23500 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "learning_rate": 4.960664061449643e-05, | |
| "loss": 2.6313, | |
| "step": 24000 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "learning_rate": 4.958977026046971e-05, | |
| "loss": 2.6318, | |
| "step": 24500 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "learning_rate": 4.957254868588065e-05, | |
| "loss": 2.6275, | |
| "step": 25000 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "learning_rate": 4.955497613670842e-05, | |
| "loss": 2.6192, | |
| "step": 25500 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "learning_rate": 4.953705286394522e-05, | |
| "loss": 2.6153, | |
| "step": 26000 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "learning_rate": 4.9518779123592705e-05, | |
| "loss": 2.6102, | |
| "step": 26500 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "learning_rate": 4.9500155176658345e-05, | |
| "loss": 2.601, | |
| "step": 27000 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "learning_rate": 4.9481181289151655e-05, | |
| "loss": 2.5991, | |
| "step": 27500 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "learning_rate": 4.9461857732080436e-05, | |
| "loss": 2.5946, | |
| "step": 28000 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "learning_rate": 4.9442184781446876e-05, | |
| "loss": 2.5911, | |
| "step": 28500 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "learning_rate": 4.9422162718243626e-05, | |
| "loss": 2.5827, | |
| "step": 29000 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "learning_rate": 4.94017918284498e-05, | |
| "loss": 2.5785, | |
| "step": 29500 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "learning_rate": 4.938107240302683e-05, | |
| "loss": 2.5757, | |
| "step": 30000 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "learning_rate": 4.93600047379144e-05, | |
| "loss": 2.5705, | |
| "step": 30500 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "learning_rate": 4.9338589134026124e-05, | |
| "loss": 2.5703, | |
| "step": 31000 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "learning_rate": 4.931682589724534e-05, | |
| "loss": 2.5593, | |
| "step": 31500 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "learning_rate": 4.929471533842065e-05, | |
| "loss": 2.5647, | |
| "step": 32000 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "learning_rate": 4.9272257773361574e-05, | |
| "loss": 2.5564, | |
| "step": 32500 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "learning_rate": 4.924945352283394e-05, | |
| "loss": 2.5492, | |
| "step": 33000 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "learning_rate": 4.922630291255539e-05, | |
| "loss": 2.5509, | |
| "step": 33500 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "learning_rate": 4.920280627319067e-05, | |
| "loss": 2.5466, | |
| "step": 34000 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "learning_rate": 4.917896394034696e-05, | |
| "loss": 2.5412, | |
| "step": 34500 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "learning_rate": 4.915477625456899e-05, | |
| "loss": 2.5383, | |
| "step": 35000 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "learning_rate": 4.91302435613343e-05, | |
| "loss": 2.5364, | |
| "step": 35500 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "learning_rate": 4.91053662110482e-05, | |
| "loss": 2.5292, | |
| "step": 36000 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "learning_rate": 4.908014455903881e-05, | |
| "loss": 2.5304, | |
| "step": 36500 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "learning_rate": 4.905457896555199e-05, | |
| "loss": 2.5257, | |
| "step": 37000 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "learning_rate": 4.902866979574618e-05, | |
| "loss": 2.5189, | |
| "step": 37500 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "learning_rate": 4.9002417419687183e-05, | |
| "loss": 2.522, | |
| "step": 38000 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "learning_rate": 4.897582221234289e-05, | |
| "loss": 2.5141, | |
| "step": 38500 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "learning_rate": 4.8948884553577926e-05, | |
| "loss": 2.517, | |
| "step": 39000 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "learning_rate": 4.8921604828148206e-05, | |
| "loss": 2.5077, | |
| "step": 39500 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 4.8893983425695445e-05, | |
| "loss": 2.5068, | |
| "step": 40000 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 4.8866020740741626e-05, | |
| "loss": 2.5048, | |
| "step": 40500 | |
| }, | |
| { | |
| "epoch": 0.3, | |
| "learning_rate": 4.883771717268331e-05, | |
| "loss": 2.4972, | |
| "step": 41000 | |
| }, | |
| { | |
| "epoch": 0.3, | |
| "learning_rate": 4.8809073125785964e-05, | |
| "loss": 2.4999, | |
| "step": 41500 | |
| }, | |
| { | |
| "epoch": 0.3, | |
| "learning_rate": 4.8780089009178206e-05, | |
| "loss": 2.4941, | |
| "step": 42000 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "learning_rate": 4.8750765236845894e-05, | |
| "loss": 2.4891, | |
| "step": 42500 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "learning_rate": 4.8721102227626304e-05, | |
| "loss": 2.4876, | |
| "step": 43000 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "learning_rate": 4.869110040520206e-05, | |
| "loss": 2.4859, | |
| "step": 43500 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "learning_rate": 4.8660760198095145e-05, | |
| "loss": 2.4833, | |
| "step": 44000 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "learning_rate": 4.8630082039660744e-05, | |
| "loss": 2.4825, | |
| "step": 44500 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "learning_rate": 4.859906636808108e-05, | |
| "loss": 2.4801, | |
| "step": 45000 | |
| }, | |
| { | |
| "epoch": 0.33, | |
| "learning_rate": 4.8567713626359134e-05, | |
| "loss": 2.4763, | |
| "step": 45500 | |
| }, | |
| { | |
| "epoch": 0.33, | |
| "learning_rate": 4.853602426231234e-05, | |
| "loss": 2.4742, | |
| "step": 46000 | |
| }, | |
| { | |
| "epoch": 0.34, | |
| "learning_rate": 4.850399872856615e-05, | |
| "loss": 2.4702, | |
| "step": 46500 | |
| }, | |
| { | |
| "epoch": 0.34, | |
| "learning_rate": 4.847163748254762e-05, | |
| "loss": 2.4679, | |
| "step": 47000 | |
| }, | |
| { | |
| "epoch": 0.34, | |
| "learning_rate": 4.843894098647884e-05, | |
| "loss": 2.4676, | |
| "step": 47500 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "learning_rate": 4.840590970737035e-05, | |
| "loss": 2.4679, | |
| "step": 48000 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "learning_rate": 4.8372544117014436e-05, | |
| "loss": 2.4644, | |
| "step": 48500 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "learning_rate": 4.8338844691978456e-05, | |
| "loss": 2.4622, | |
| "step": 49000 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "learning_rate": 4.830481191359797e-05, | |
| "loss": 2.4605, | |
| "step": 49500 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "learning_rate": 4.82704462679699e-05, | |
| "loss": 2.4552, | |
| "step": 50000 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "learning_rate": 4.823574824594558e-05, | |
| "loss": 2.4501, | |
| "step": 50500 | |
| }, | |
| { | |
| "epoch": 0.37, | |
| "learning_rate": 4.820071834312372e-05, | |
| "loss": 2.4563, | |
| "step": 51000 | |
| }, | |
| { | |
| "epoch": 0.37, | |
| "learning_rate": 4.816535705984337e-05, | |
| "loss": 2.4467, | |
| "step": 51500 | |
| }, | |
| { | |
| "epoch": 0.37, | |
| "learning_rate": 4.812966490117676e-05, | |
| "loss": 2.4434, | |
| "step": 52000 | |
| }, | |
| { | |
| "epoch": 0.38, | |
| "learning_rate": 4.8093642376922057e-05, | |
| "loss": 2.4422, | |
| "step": 52500 | |
| }, | |
| { | |
| "epoch": 0.38, | |
| "learning_rate": 4.8057290001596106e-05, | |
| "loss": 2.4462, | |
| "step": 53000 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "learning_rate": 4.802060829442711e-05, | |
| "loss": 2.4424, | |
| "step": 53500 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "learning_rate": 4.798359777934714e-05, | |
| "loss": 2.4373, | |
| "step": 54000 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 416130, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 3000, | |
| "total_flos": 1.9776864780288e+19, | |
| "train_batch_size": 16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |