| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.391304347826087, | |
| "eval_steps": 50, | |
| "global_step": 200, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.034782608695652174, | |
| "grad_norm": 27.628698806495482, | |
| "learning_rate": 5e-07, | |
| "loss": 1.6404, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.06956521739130435, | |
| "grad_norm": 14.379307335524663, | |
| "learning_rate": 1e-06, | |
| "loss": 1.5079, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.10434782608695652, | |
| "grad_norm": 7.6171415097012805, | |
| "learning_rate": 9.99648681635985e-07, | |
| "loss": 1.2858, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.1391304347826087, | |
| "grad_norm": 5.902380140367801, | |
| "learning_rate": 9.985952202423114e-07, | |
| "loss": 1.2292, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.17391304347826086, | |
| "grad_norm": 5.431235666701782, | |
| "learning_rate": 9.96841096220313e-07, | |
| "loss": 1.2035, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.20869565217391303, | |
| "grad_norm": 5.011022329519287, | |
| "learning_rate": 9.943887745939163e-07, | |
| "loss": 1.1615, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.24347826086956523, | |
| "grad_norm": 4.933604072753844, | |
| "learning_rate": 9.912417015456088e-07, | |
| "loss": 1.1627, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.2782608695652174, | |
| "grad_norm": 4.904429773196176, | |
| "learning_rate": 9.874042995736093e-07, | |
| "loss": 1.1468, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.3130434782608696, | |
| "grad_norm": 4.566916992899494, | |
| "learning_rate": 9.828819612770495e-07, | |
| "loss": 1.1457, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.34782608695652173, | |
| "grad_norm": 4.618513047328418, | |
| "learning_rate": 9.77681041777897e-07, | |
| "loss": 1.1363, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.34782608695652173, | |
| "eval_loss": 1.1214656829833984, | |
| "eval_runtime": 71.1557, | |
| "eval_samples_per_second": 57.423, | |
| "eval_steps_per_second": 0.899, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.3826086956521739, | |
| "grad_norm": 4.482447125747761, | |
| "learning_rate": 9.718088497902707e-07, | |
| "loss": 1.1252, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.41739130434782606, | |
| "grad_norm": 4.746315495183657, | |
| "learning_rate": 9.652736373497e-07, | |
| "loss": 1.1144, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.45217391304347826, | |
| "grad_norm": 4.800644875448408, | |
| "learning_rate": 9.580845882167572e-07, | |
| "loss": 1.1048, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.48695652173913045, | |
| "grad_norm": 5.041563912063608, | |
| "learning_rate": 9.502518049713631e-07, | |
| "loss": 1.0967, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.5217391304347826, | |
| "grad_norm": 4.834159384378619, | |
| "learning_rate": 9.417862948158997e-07, | |
| "loss": 1.0922, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.5565217391304348, | |
| "grad_norm": 4.966276459660724, | |
| "learning_rate": 9.326999541070803e-07, | |
| "loss": 1.1065, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.591304347826087, | |
| "grad_norm": 4.651039844387456, | |
| "learning_rate": 9.23005551638316e-07, | |
| "loss": 1.08, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 0.6260869565217392, | |
| "grad_norm": 4.61712494812361, | |
| "learning_rate": 9.127167106960681e-07, | |
| "loss": 1.0871, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.6608695652173913, | |
| "grad_norm": 4.965649778107791, | |
| "learning_rate": 9.018478899154066e-07, | |
| "loss": 1.0791, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 0.6956521739130435, | |
| "grad_norm": 4.710879346221803, | |
| "learning_rate": 8.904143629616732e-07, | |
| "loss": 1.0856, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.6956521739130435, | |
| "eval_loss": 1.074812412261963, | |
| "eval_runtime": 70.7335, | |
| "eval_samples_per_second": 57.766, | |
| "eval_steps_per_second": 0.905, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.7304347826086957, | |
| "grad_norm": 4.887377540999055, | |
| "learning_rate": 8.784321970668053e-07, | |
| "loss": 1.0638, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 0.7652173913043478, | |
| "grad_norm": 4.804331292468611, | |
| "learning_rate": 8.659182304504808e-07, | |
| "loss": 1.0768, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "grad_norm": 4.78084183802074, | |
| "learning_rate": 8.528900486578158e-07, | |
| "loss": 1.0838, | |
| "step": 115 | |
| }, | |
| { | |
| "epoch": 0.8347826086956521, | |
| "grad_norm": 5.189973322685715, | |
| "learning_rate": 8.393659598468642e-07, | |
| "loss": 1.0633, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.8695652173913043, | |
| "grad_norm": 5.02605525534553, | |
| "learning_rate": 8.253649690606494e-07, | |
| "loss": 1.0748, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 0.9043478260869565, | |
| "grad_norm": 5.433069400847481, | |
| "learning_rate": 8.10906751519882e-07, | |
| "loss": 1.0623, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.9391304347826087, | |
| "grad_norm": 4.770051175238624, | |
| "learning_rate": 7.960116249738937e-07, | |
| "loss": 1.0396, | |
| "step": 135 | |
| }, | |
| { | |
| "epoch": 0.9739130434782609, | |
| "grad_norm": 4.606677098055935, | |
| "learning_rate": 7.807005211486444e-07, | |
| "loss": 1.0662, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 1.008695652173913, | |
| "grad_norm": 5.498117473835195, | |
| "learning_rate": 7.649949563319227e-07, | |
| "loss": 1.0155, | |
| "step": 145 | |
| }, | |
| { | |
| "epoch": 1.0434782608695652, | |
| "grad_norm": 4.9633963344101595, | |
| "learning_rate": 7.489170011370779e-07, | |
| "loss": 0.9426, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.0434782608695652, | |
| "eval_loss": 1.0552960634231567, | |
| "eval_runtime": 70.6918, | |
| "eval_samples_per_second": 57.8, | |
| "eval_steps_per_second": 0.905, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.0782608695652174, | |
| "grad_norm": 5.4731038473793285, | |
| "learning_rate": 7.324892494877733e-07, | |
| "loss": 0.9222, | |
| "step": 155 | |
| }, | |
| { | |
| "epoch": 1.1130434782608696, | |
| "grad_norm": 5.024873028080845, | |
| "learning_rate": 7.15734786867344e-07, | |
| "loss": 0.9223, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 1.1478260869565218, | |
| "grad_norm": 5.363614939434412, | |
| "learning_rate": 6.986771578773811e-07, | |
| "loss": 0.937, | |
| "step": 165 | |
| }, | |
| { | |
| "epoch": 1.182608695652174, | |
| "grad_norm": 5.175874733453326, | |
| "learning_rate": 6.81340333151128e-07, | |
| "loss": 0.9173, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 1.2173913043478262, | |
| "grad_norm": 5.446257822096013, | |
| "learning_rate": 6.637486756681842e-07, | |
| "loss": 0.9232, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 1.2521739130434781, | |
| "grad_norm": 5.0358352027359174, | |
| "learning_rate": 6.459269065178591e-07, | |
| "loss": 0.9146, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 1.2869565217391306, | |
| "grad_norm": 5.287700886552525, | |
| "learning_rate": 6.279000701592794e-07, | |
| "loss": 0.919, | |
| "step": 185 | |
| }, | |
| { | |
| "epoch": 1.3217391304347825, | |
| "grad_norm": 5.280625369358825, | |
| "learning_rate": 6.096934992270767e-07, | |
| "loss": 0.9201, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 1.3565217391304347, | |
| "grad_norm": 5.218195920942847, | |
| "learning_rate": 5.913327789321077e-07, | |
| "loss": 0.9151, | |
| "step": 195 | |
| }, | |
| { | |
| "epoch": 1.391304347826087, | |
| "grad_norm": 5.2780690080724515, | |
| "learning_rate": 5.728437111072375e-07, | |
| "loss": 0.9144, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.391304347826087, | |
| "eval_loss": 1.043231725692749, | |
| "eval_runtime": 70.6101, | |
| "eval_samples_per_second": 57.867, | |
| "eval_steps_per_second": 0.906, | |
| "step": 200 | |
| } | |
| ], | |
| "logging_steps": 5, | |
| "max_steps": 429, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 100, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 1179056703799296.0, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |