| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 4.201680672268908, | |
| "eval_steps": 100, | |
| "global_step": 1000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.10504201680672269, | |
| "grad_norm": 8.704133033752441, | |
| "learning_rate": 4.166666666666667e-05, | |
| "loss": 3.5481, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.21008403361344538, | |
| "grad_norm": 10.129899978637695, | |
| "learning_rate": 4.9947570655942796e-05, | |
| "loss": 3.0062, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.31512605042016806, | |
| "grad_norm": 6.111865997314453, | |
| "learning_rate": 4.9734953280908904e-05, | |
| "loss": 2.9095, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.42016806722689076, | |
| "grad_norm": 4.366275787353516, | |
| "learning_rate": 4.936026311617316e-05, | |
| "loss": 2.9062, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.42016806722689076, | |
| "eval_loss": 2.872298240661621, | |
| "eval_runtime": 35.9441, | |
| "eval_samples_per_second": 13.02, | |
| "eval_steps_per_second": 1.641, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.5252100840336135, | |
| "grad_norm": 25.31424903869629, | |
| "learning_rate": 4.882595527372152e-05, | |
| "loss": 2.874, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 0.6302521008403361, | |
| "grad_norm": 53.229515075683594, | |
| "learning_rate": 4.813553074106761e-05, | |
| "loss": 2.887, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.7352941176470589, | |
| "grad_norm": 7.568065166473389, | |
| "learning_rate": 4.7293513441455364e-05, | |
| "loss": 2.8903, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 0.8403361344537815, | |
| "grad_norm": 6.394404888153076, | |
| "learning_rate": 4.630542059139924e-05, | |
| "loss": 2.7455, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.8403361344537815, | |
| "eval_loss": 2.743553638458252, | |
| "eval_runtime": 35.9282, | |
| "eval_samples_per_second": 13.026, | |
| "eval_steps_per_second": 1.642, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.9453781512605042, | |
| "grad_norm": 3.6301393508911133, | |
| "learning_rate": 4.517772654979023e-05, | |
| "loss": 2.7609, | |
| "step": 225 | |
| }, | |
| { | |
| "epoch": 1.050420168067227, | |
| "grad_norm": 21.900226593017578, | |
| "learning_rate": 4.391782039544238e-05, | |
| "loss": 2.7411, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 1.1554621848739495, | |
| "grad_norm": 9.00839900970459, | |
| "learning_rate": 4.253395751104748e-05, | |
| "loss": 2.3212, | |
| "step": 275 | |
| }, | |
| { | |
| "epoch": 1.2605042016806722, | |
| "grad_norm": 3.9047327041625977, | |
| "learning_rate": 4.10352054907785e-05, | |
| "loss": 2.0506, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 1.2605042016806722, | |
| "eval_loss": 2.8556525707244873, | |
| "eval_runtime": 35.8984, | |
| "eval_samples_per_second": 13.037, | |
| "eval_steps_per_second": 1.644, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 1.365546218487395, | |
| "grad_norm": 3.455016613006592, | |
| "learning_rate": 3.943138472597549e-05, | |
| "loss": 1.826, | |
| "step": 325 | |
| }, | |
| { | |
| "epoch": 1.4705882352941178, | |
| "grad_norm": 10.288714408874512, | |
| "learning_rate": 3.773300405821908e-05, | |
| "loss": 1.8779, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 1.5756302521008403, | |
| "grad_norm": 2.957385540008545, | |
| "learning_rate": 3.595119192141706e-05, | |
| "loss": 1.85, | |
| "step": 375 | |
| }, | |
| { | |
| "epoch": 1.680672268907563, | |
| "grad_norm": 3.40120267868042, | |
| "learning_rate": 3.409762342408719e-05, | |
| "loss": 1.835, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 1.680672268907563, | |
| "eval_loss": 2.7827320098876953, | |
| "eval_runtime": 36.0584, | |
| "eval_samples_per_second": 12.979, | |
| "eval_steps_per_second": 1.636, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 1.7857142857142856, | |
| "grad_norm": 3.446842670440674, | |
| "learning_rate": 3.218444384962071e-05, | |
| "loss": 1.8477, | |
| "step": 425 | |
| }, | |
| { | |
| "epoch": 1.8907563025210083, | |
| "grad_norm": 3.1697998046875, | |
| "learning_rate": 3.0224189075781884e-05, | |
| "loss": 1.7934, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 1.995798319327731, | |
| "grad_norm": 2.8750455379486084, | |
| "learning_rate": 2.8229703434885163e-05, | |
| "loss": 1.7526, | |
| "step": 475 | |
| }, | |
| { | |
| "epoch": 2.100840336134454, | |
| "grad_norm": 2.977962017059326, | |
| "learning_rate": 2.621405555286121e-05, | |
| "loss": 0.8246, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 2.100840336134454, | |
| "eval_loss": 3.001242160797119, | |
| "eval_runtime": 35.8998, | |
| "eval_samples_per_second": 13.036, | |
| "eval_steps_per_second": 1.643, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 2.2058823529411766, | |
| "grad_norm": 2.8460147380828857, | |
| "learning_rate": 2.419045271866611e-05, | |
| "loss": 0.7854, | |
| "step": 525 | |
| }, | |
| { | |
| "epoch": 2.310924369747899, | |
| "grad_norm": 3.5654945373535156, | |
| "learning_rate": 2.2172154345117894e-05, | |
| "loss": 0.8065, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 2.4159663865546217, | |
| "grad_norm": 2.3066728115081787, | |
| "learning_rate": 2.0172385088197803e-05, | |
| "loss": 0.7753, | |
| "step": 575 | |
| }, | |
| { | |
| "epoch": 2.5210084033613445, | |
| "grad_norm": 2.5332283973693848, | |
| "learning_rate": 1.820424819409143e-05, | |
| "loss": 0.824, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 2.5210084033613445, | |
| "eval_loss": 2.8449320793151855, | |
| "eval_runtime": 35.8956, | |
| "eval_samples_per_second": 13.038, | |
| "eval_steps_per_second": 1.644, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 2.6260504201680672, | |
| "grad_norm": 3.0348715782165527, | |
| "learning_rate": 1.6280639641752942e-05, | |
| "loss": 0.7481, | |
| "step": 625 | |
| }, | |
| { | |
| "epoch": 2.73109243697479, | |
| "grad_norm": 2.7579290866851807, | |
| "learning_rate": 1.4414163643562755e-05, | |
| "loss": 0.7543, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 2.8361344537815127, | |
| "grad_norm": 2.145768404006958, | |
| "learning_rate": 1.2617050057750322e-05, | |
| "loss": 0.753, | |
| "step": 675 | |
| }, | |
| { | |
| "epoch": 2.9411764705882355, | |
| "grad_norm": 2.6759989261627197, | |
| "learning_rate": 1.0901074253727336e-05, | |
| "loss": 0.7418, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 2.9411764705882355, | |
| "eval_loss": 2.796262741088867, | |
| "eval_runtime": 35.8962, | |
| "eval_samples_per_second": 13.038, | |
| "eval_steps_per_second": 1.644, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 3.046218487394958, | |
| "grad_norm": 1.9286493062973022, | |
| "learning_rate": 9.277479955403887e-06, | |
| "loss": 0.5768, | |
| "step": 725 | |
| }, | |
| { | |
| "epoch": 3.1512605042016806, | |
| "grad_norm": 1.2430107593536377, | |
| "learning_rate": 7.756905568047393e-06, | |
| "loss": 0.2213, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 3.2563025210084033, | |
| "grad_norm": 1.5327359437942505, | |
| "learning_rate": 6.349314471418849e-06, | |
| "loss": 0.2563, | |
| "step": 775 | |
| }, | |
| { | |
| "epoch": 3.361344537815126, | |
| "grad_norm": 1.448890209197998, | |
| "learning_rate": 5.063929735931985e-06, | |
| "loss": 0.2227, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 3.361344537815126, | |
| "eval_loss": 3.311384439468384, | |
| "eval_runtime": 35.9006, | |
| "eval_samples_per_second": 13.036, | |
| "eval_steps_per_second": 1.643, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 3.466386554621849, | |
| "grad_norm": 1.3534725904464722, | |
| "learning_rate": 3.90917368959989e-06, | |
| "loss": 0.2464, | |
| "step": 825 | |
| }, | |
| { | |
| "epoch": 3.571428571428571, | |
| "grad_norm": 1.3151746988296509, | |
| "learning_rate": 2.892612731749414e-06, | |
| "loss": 0.2439, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 3.6764705882352944, | |
| "grad_norm": 1.4781558513641357, | |
| "learning_rate": 2.020907755104698e-06, | |
| "loss": 0.2355, | |
| "step": 875 | |
| }, | |
| { | |
| "epoch": 3.7815126050420167, | |
| "grad_norm": 1.4184162616729736, | |
| "learning_rate": 1.2997705010932393e-06, | |
| "loss": 0.2198, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 3.7815126050420167, | |
| "eval_loss": 3.312718152999878, | |
| "eval_runtime": 35.8966, | |
| "eval_samples_per_second": 13.037, | |
| "eval_steps_per_second": 1.644, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 3.8865546218487395, | |
| "grad_norm": 1.545154333114624, | |
| "learning_rate": 7.339261343510206e-07, | |
| "loss": 0.222, | |
| "step": 925 | |
| }, | |
| { | |
| "epoch": 3.991596638655462, | |
| "grad_norm": 1.5649031400680542, | |
| "learning_rate": 3.270822816527325e-07, | |
| "loss": 0.2069, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 4.0966386554621845, | |
| "grad_norm": 0.9737259149551392, | |
| "learning_rate": 8.190473813576572e-08, | |
| "loss": 0.1163, | |
| "step": 975 | |
| }, | |
| { | |
| "epoch": 4.201680672268908, | |
| "grad_norm": 0.9735974669456482, | |
| "learning_rate": 0.0, | |
| "loss": 0.1478, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 4.201680672268908, | |
| "eval_loss": 3.372861385345459, | |
| "eval_runtime": 35.8981, | |
| "eval_samples_per_second": 13.037, | |
| "eval_steps_per_second": 1.644, | |
| "step": 1000 | |
| } | |
| ], | |
| "logging_steps": 25, | |
| "max_steps": 1000, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 5, | |
| "save_steps": 250, | |
| "total_flos": 9.22205177249792e+16, | |
| "train_batch_size": 1, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |