| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 3.0, | |
| "eval_steps": 500, | |
| "global_step": 1197, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 1.2109375, | |
| "learning_rate": 1.1111111111111113e-05, | |
| "loss": 0.594, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 0.8125, | |
| "learning_rate": 1.9999414238190722e-05, | |
| "loss": 0.3314, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "grad_norm": 0.765625, | |
| "learning_rate": 1.9978919779420427e-05, | |
| "loss": 0.2863, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 0.7578125, | |
| "learning_rate": 1.9929205817437334e-05, | |
| "loss": 0.2732, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 0.74609375, | |
| "learning_rate": 1.9850417920830427e-05, | |
| "loss": 0.2624, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.3, | |
| "grad_norm": 0.77734375, | |
| "learning_rate": 1.9742786790241548e-05, | |
| "loss": 0.2557, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "grad_norm": 0.70703125, | |
| "learning_rate": 1.9606627582845587e-05, | |
| "loss": 0.2508, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.4, | |
| "grad_norm": 0.7578125, | |
| "learning_rate": 1.944233898953152e-05, | |
| "loss": 0.2474, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.45, | |
| "grad_norm": 0.71484375, | |
| "learning_rate": 1.9250402067486523e-05, | |
| "loss": 0.2438, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.5, | |
| "grad_norm": 0.75, | |
| "learning_rate": 1.9031378831601425e-05, | |
| "loss": 0.2405, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "grad_norm": 0.734375, | |
| "learning_rate": 1.878591060882203e-05, | |
| "loss": 0.2359, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.6, | |
| "grad_norm": 0.74609375, | |
| "learning_rate": 1.851471616026505e-05, | |
| "loss": 0.2347, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.65, | |
| "grad_norm": 0.7421875, | |
| "learning_rate": 1.8218589576597203e-05, | |
| "loss": 0.2305, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "grad_norm": 0.75, | |
| "learning_rate": 1.789839795284022e-05, | |
| "loss": 0.2324, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "grad_norm": 0.7421875, | |
| "learning_rate": 1.7555078849410044e-05, | |
| "loss": 0.23, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "grad_norm": 0.6796875, | |
| "learning_rate": 1.7189637546824752e-05, | |
| "loss": 0.2235, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 0.85, | |
| "grad_norm": 0.70703125, | |
| "learning_rate": 1.680314410211967e-05, | |
| "loss": 0.2236, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 0.9, | |
| "grad_norm": 0.7578125, | |
| "learning_rate": 1.6396730215588913e-05, | |
| "loss": 0.2236, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 0.95, | |
| "grad_norm": 0.72265625, | |
| "learning_rate": 1.5971585917027864e-05, | |
| "loss": 0.2225, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "grad_norm": 0.7421875, | |
| "learning_rate": 1.552895608117971e-05, | |
| "loss": 0.2212, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 1.05, | |
| "grad_norm": 0.7265625, | |
| "learning_rate": 1.5070136782589236e-05, | |
| "loss": 0.1921, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 1.1, | |
| "grad_norm": 0.75, | |
| "learning_rate": 1.4596471500537334e-05, | |
| "loss": 0.1902, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 1.15, | |
| "grad_norm": 0.7265625, | |
| "learning_rate": 1.4109347185168577e-05, | |
| "loss": 0.19, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 1.2, | |
| "grad_norm": 0.7421875, | |
| "learning_rate": 1.3610190196330777e-05, | |
| "loss": 0.1869, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "grad_norm": 0.76171875, | |
| "learning_rate": 1.3100462127018108e-05, | |
| "loss": 0.1902, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 1.3, | |
| "grad_norm": 0.74609375, | |
| "learning_rate": 1.2581655523647234e-05, | |
| "loss": 0.1888, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 1.35, | |
| "grad_norm": 0.75390625, | |
| "learning_rate": 1.2055289515698008e-05, | |
| "loss": 0.1882, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 1.4, | |
| "grad_norm": 0.74609375, | |
| "learning_rate": 1.1522905367515705e-05, | |
| "loss": 0.1881, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 1.45, | |
| "grad_norm": 0.7421875, | |
| "learning_rate": 1.0986061965299646e-05, | |
| "loss": 0.1855, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 1.5, | |
| "grad_norm": 0.73046875, | |
| "learning_rate": 1.0446331252492864e-05, | |
| "loss": 0.1892, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 1.55, | |
| "grad_norm": 0.71484375, | |
| "learning_rate": 9.905293626938564e-06, | |
| "loss": 0.1856, | |
| "step": 620 | |
| }, | |
| { | |
| "epoch": 1.6, | |
| "grad_norm": 0.7890625, | |
| "learning_rate": 9.364533313281056e-06, | |
| "loss": 0.1848, | |
| "step": 640 | |
| }, | |
| { | |
| "epoch": 1.65, | |
| "grad_norm": 0.78125, | |
| "learning_rate": 8.825633724161336e-06, | |
| "loss": 0.1856, | |
| "step": 660 | |
| }, | |
| { | |
| "epoch": 1.7, | |
| "grad_norm": 0.73828125, | |
| "learning_rate": 8.290172823790337e-06, | |
| "loss": 0.1837, | |
| "step": 680 | |
| }, | |
| { | |
| "epoch": 1.75, | |
| "grad_norm": 0.75, | |
| "learning_rate": 7.759718507475822e-06, | |
| "loss": 0.1841, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 1.8, | |
| "grad_norm": 0.72265625, | |
| "learning_rate": 7.235824010632284e-06, | |
| "loss": 0.1856, | |
| "step": 720 | |
| }, | |
| { | |
| "epoch": 1.85, | |
| "grad_norm": 0.77734375, | |
| "learning_rate": 6.720023360716818e-06, | |
| "loss": 0.1849, | |
| "step": 740 | |
| }, | |
| { | |
| "epoch": 1.9, | |
| "grad_norm": 0.76953125, | |
| "learning_rate": 6.2138268854082094e-06, | |
| "loss": 0.1812, | |
| "step": 760 | |
| }, | |
| { | |
| "epoch": 1.95, | |
| "grad_norm": 0.74609375, | |
| "learning_rate": 5.7187167901819665e-06, | |
| "loss": 0.1853, | |
| "step": 780 | |
| }, | |
| { | |
| "epoch": 2.01, | |
| "grad_norm": 0.72265625, | |
| "learning_rate": 5.236142818230518e-06, | |
| "loss": 0.1817, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 2.06, | |
| "grad_norm": 0.76171875, | |
| "learning_rate": 4.767518005437073e-06, | |
| "loss": 0.1694, | |
| "step": 820 | |
| }, | |
| { | |
| "epoch": 2.11, | |
| "grad_norm": 0.76171875, | |
| "learning_rate": 4.314214542832889e-06, | |
| "loss": 0.1679, | |
| "step": 840 | |
| }, | |
| { | |
| "epoch": 2.16, | |
| "grad_norm": 0.73828125, | |
| "learning_rate": 3.877559758653358e-06, | |
| "loss": 0.1724, | |
| "step": 860 | |
| }, | |
| { | |
| "epoch": 2.21, | |
| "grad_norm": 0.77734375, | |
| "learning_rate": 3.4588322317578395e-06, | |
| "loss": 0.1685, | |
| "step": 880 | |
| }, | |
| { | |
| "epoch": 2.26, | |
| "grad_norm": 0.71875, | |
| "learning_rate": 3.0592580477936606e-06, | |
| "loss": 0.1674, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 2.31, | |
| "grad_norm": 0.76171875, | |
| "learning_rate": 2.6800072090667085e-06, | |
| "loss": 0.1687, | |
| "step": 920 | |
| }, | |
| { | |
| "epoch": 2.36, | |
| "grad_norm": 0.80078125, | |
| "learning_rate": 2.3221902086308943e-06, | |
| "loss": 0.1669, | |
| "step": 940 | |
| }, | |
| { | |
| "epoch": 2.41, | |
| "grad_norm": 0.7265625, | |
| "learning_rate": 1.986854778628067e-06, | |
| "loss": 0.1693, | |
| "step": 960 | |
| }, | |
| { | |
| "epoch": 2.46, | |
| "grad_norm": 0.8046875, | |
| "learning_rate": 1.67498282239956e-06, | |
| "loss": 0.169, | |
| "step": 980 | |
| }, | |
| { | |
| "epoch": 2.51, | |
| "grad_norm": 0.73828125, | |
| "learning_rate": 1.3874875393525722e-06, | |
| "loss": 0.1668, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 2.56, | |
| "grad_norm": 0.7421875, | |
| "learning_rate": 1.1252107510000843e-06, | |
| "loss": 0.1684, | |
| "step": 1020 | |
| }, | |
| { | |
| "epoch": 2.61, | |
| "grad_norm": 0.7734375, | |
| "learning_rate": 8.889204360040471e-07, | |
| "loss": 0.1663, | |
| "step": 1040 | |
| }, | |
| { | |
| "epoch": 2.66, | |
| "grad_norm": 0.7421875, | |
| "learning_rate": 6.793084814394945e-07, | |
| "loss": 0.1685, | |
| "step": 1060 | |
| }, | |
| { | |
| "epoch": 2.71, | |
| "grad_norm": 0.75, | |
| "learning_rate": 4.969886568641757e-07, | |
| "loss": 0.1669, | |
| "step": 1080 | |
| }, | |
| { | |
| "epoch": 2.76, | |
| "grad_norm": 0.765625, | |
| "learning_rate": 3.424948171258824e-07, | |
| "loss": 0.1691, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 2.81, | |
| "grad_norm": 0.7890625, | |
| "learning_rate": 2.16279339169847e-07, | |
| "loss": 0.1718, | |
| "step": 1120 | |
| }, | |
| { | |
| "epoch": 2.86, | |
| "grad_norm": 0.78125, | |
| "learning_rate": 1.1871179742348416e-07, | |
| "loss": 0.1648, | |
| "step": 1140 | |
| }, | |
| { | |
| "epoch": 2.91, | |
| "grad_norm": 0.73828125, | |
| "learning_rate": 5.007788163703842e-08, | |
| "loss": 0.1706, | |
| "step": 1160 | |
| }, | |
| { | |
| "epoch": 2.96, | |
| "grad_norm": 0.81640625, | |
| "learning_rate": 1.0578560348905031e-08, | |
| "loss": 0.1694, | |
| "step": 1180 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "step": 1197, | |
| "total_flos": 8.742938292204339e+17, | |
| "train_loss": 0.20622981182216304, | |
| "train_runtime": 3123.6217, | |
| "train_samples_per_second": 55.182, | |
| "train_steps_per_second": 0.383 | |
| } | |
| ], | |
| "logging_steps": 20, | |
| "max_steps": 1197, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 10000, | |
| "total_flos": 8.742938292204339e+17, | |
| "train_batch_size": 36, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |