| { |
| "best_global_step": null, |
| "best_metric": null, |
| "best_model_checkpoint": null, |
| "epoch": 2.0, |
| "eval_steps": 52, |
| "global_step": 52, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.038461538461538464, |
| "grad_norm": 4.689635276794434, |
| "learning_rate": 1e-05, |
| "loss": 2.4853515625, |
| "step": 1 |
| }, |
| { |
| "epoch": 0.07692307692307693, |
| "grad_norm": 3.844747543334961, |
| "learning_rate": 9.990877771116588e-06, |
| "loss": 2.3544921875, |
| "step": 2 |
| }, |
| { |
| "epoch": 0.11538461538461539, |
| "grad_norm": 3.494800090789795, |
| "learning_rate": 9.96354437049027e-06, |
| "loss": 2.2099609375, |
| "step": 3 |
| }, |
| { |
| "epoch": 0.15384615384615385, |
| "grad_norm": 2.4297192096710205, |
| "learning_rate": 9.91809953473572e-06, |
| "loss": 2.0040283203125, |
| "step": 4 |
| }, |
| { |
| "epoch": 0.19230769230769232, |
| "grad_norm": 2.0086750984191895, |
| "learning_rate": 9.854709087130261e-06, |
| "loss": 1.907470703125, |
| "step": 5 |
| }, |
| { |
| "epoch": 0.23076923076923078, |
| "grad_norm": 1.6094433069229126, |
| "learning_rate": 9.77360433254273e-06, |
| "loss": 1.8612060546875, |
| "step": 6 |
| }, |
| { |
| "epoch": 0.2692307692307692, |
| "grad_norm": 1.5231853723526, |
| "learning_rate": 9.675081213427076e-06, |
| "loss": 1.83349609375, |
| "step": 7 |
| }, |
| { |
| "epoch": 0.3076923076923077, |
| "grad_norm": 1.1918054819107056, |
| "learning_rate": 9.55949922996045e-06, |
| "loss": 1.640625, |
| "step": 8 |
| }, |
| { |
| "epoch": 0.34615384615384615, |
| "grad_norm": 0.9854990243911743, |
| "learning_rate": 9.427280128266049e-06, |
| "loss": 1.56982421875, |
| "step": 9 |
| }, |
| { |
| "epoch": 0.38461538461538464, |
| "grad_norm": 0.9586904048919678, |
| "learning_rate": 9.278906361507238e-06, |
| "loss": 1.5654296875, |
| "step": 10 |
| }, |
| { |
| "epoch": 0.4230769230769231, |
| "grad_norm": 0.8120144605636597, |
| "learning_rate": 9.114919329468283e-06, |
| "loss": 1.44482421875, |
| "step": 11 |
| }, |
| { |
| "epoch": 0.46153846153846156, |
| "grad_norm": 0.7073745131492615, |
| "learning_rate": 8.935917403045251e-06, |
| "loss": 1.49560546875, |
| "step": 12 |
| }, |
| { |
| "epoch": 0.5, |
| "grad_norm": 0.6228000521659851, |
| "learning_rate": 8.742553740855507e-06, |
| "loss": 1.47998046875, |
| "step": 13 |
| }, |
| { |
| "epoch": 0.5384615384615384, |
| "grad_norm": 0.544258177280426, |
| "learning_rate": 8.535533905932739e-06, |
| "loss": 1.42950439453125, |
| "step": 14 |
| }, |
| { |
| "epoch": 0.5769230769230769, |
| "grad_norm": 0.44122347235679626, |
| "learning_rate": 8.315613291203977e-06, |
| "loss": 1.392578125, |
| "step": 15 |
| }, |
| { |
| "epoch": 0.6153846153846154, |
| "grad_norm": 0.3860686719417572, |
| "learning_rate": 8.083594363142717e-06, |
| "loss": 1.38311767578125, |
| "step": 16 |
| }, |
| { |
| "epoch": 0.6538461538461539, |
| "grad_norm": 0.35200247168540955, |
| "learning_rate": 7.84032373365578e-06, |
| "loss": 1.3580322265625, |
| "step": 17 |
| }, |
| { |
| "epoch": 0.6923076923076923, |
| "grad_norm": 0.3415108621120453, |
| "learning_rate": 7.586689070888284e-06, |
| "loss": 1.326904296875, |
| "step": 18 |
| }, |
| { |
| "epoch": 0.7307692307692307, |
| "grad_norm": 0.3024138808250427, |
| "learning_rate": 7.323615860218844e-06, |
| "loss": 1.3759765625, |
| "step": 19 |
| }, |
| { |
| "epoch": 0.7692307692307693, |
| "grad_norm": 0.25955742597579956, |
| "learning_rate": 7.052064027263785e-06, |
| "loss": 1.3367919921875, |
| "step": 20 |
| }, |
| { |
| "epoch": 0.8076923076923077, |
| "grad_norm": 0.2363637387752533, |
| "learning_rate": 6.773024435212678e-06, |
| "loss": 1.3818359375, |
| "step": 21 |
| }, |
| { |
| "epoch": 0.8461538461538461, |
| "grad_norm": 0.20735107362270355, |
| "learning_rate": 6.487515269276015e-06, |
| "loss": 1.27813720703125, |
| "step": 22 |
| }, |
| { |
| "epoch": 0.8846153846153846, |
| "grad_norm": 0.19036373496055603, |
| "learning_rate": 6.1965783214377895e-06, |
| "loss": 1.32275390625, |
| "step": 23 |
| }, |
| { |
| "epoch": 0.9230769230769231, |
| "grad_norm": 0.18189361691474915, |
| "learning_rate": 5.90127518906953e-06, |
| "loss": 1.298828125, |
| "step": 24 |
| }, |
| { |
| "epoch": 0.9615384615384616, |
| "grad_norm": 0.17046603560447693, |
| "learning_rate": 5.6026834012766155e-06, |
| "loss": 1.3623046875, |
| "step": 25 |
| }, |
| { |
| "epoch": 1.0, |
| "grad_norm": 0.16392011940479279, |
| "learning_rate": 5.301892487111431e-06, |
| "loss": 1.31689453125, |
| "step": 26 |
| }, |
| { |
| "epoch": 1.0384615384615385, |
| "grad_norm": 0.15912248194217682, |
| "learning_rate": 5e-06, |
| "loss": 1.27880859375, |
| "step": 27 |
| }, |
| { |
| "epoch": 1.0769230769230769, |
| "grad_norm": 0.15796270966529846, |
| "learning_rate": 4.69810751288857e-06, |
| "loss": 1.3271484375, |
| "step": 28 |
| }, |
| { |
| "epoch": 1.1153846153846154, |
| "grad_norm": 0.13745176792144775, |
| "learning_rate": 4.397316598723385e-06, |
| "loss": 1.291015625, |
| "step": 29 |
| }, |
| { |
| "epoch": 1.1538461538461537, |
| "grad_norm": 0.14191485941410065, |
| "learning_rate": 4.098724810930472e-06, |
| "loss": 1.288330078125, |
| "step": 30 |
| }, |
| { |
| "epoch": 1.1923076923076923, |
| "grad_norm": 0.13703800737857819, |
| "learning_rate": 3.803421678562213e-06, |
| "loss": 1.274169921875, |
| "step": 31 |
| }, |
| { |
| "epoch": 1.2307692307692308, |
| "grad_norm": 0.14957869052886963, |
| "learning_rate": 3.5124847307239863e-06, |
| "loss": 1.33819580078125, |
| "step": 32 |
| }, |
| { |
| "epoch": 1.2692307692307692, |
| "grad_norm": 0.14096862077713013, |
| "learning_rate": 3.226975564787322e-06, |
| "loss": 1.39404296875, |
| "step": 33 |
| }, |
| { |
| "epoch": 1.3076923076923077, |
| "grad_norm": 0.13106374442577362, |
| "learning_rate": 2.947935972736217e-06, |
| "loss": 1.29052734375, |
| "step": 34 |
| }, |
| { |
| "epoch": 1.3461538461538463, |
| "grad_norm": 0.1206049919128418, |
| "learning_rate": 2.6763841397811576e-06, |
| "loss": 1.26629638671875, |
| "step": 35 |
| }, |
| { |
| "epoch": 1.3846153846153846, |
| "grad_norm": 0.12056861072778702, |
| "learning_rate": 2.4133109291117156e-06, |
| "loss": 1.2890625, |
| "step": 36 |
| }, |
| { |
| "epoch": 1.4230769230769231, |
| "grad_norm": 0.13918845355510712, |
| "learning_rate": 2.159676266344222e-06, |
| "loss": 1.2119140625, |
| "step": 37 |
| }, |
| { |
| "epoch": 1.4615384615384617, |
| "grad_norm": 0.14604124426841736, |
| "learning_rate": 1.9164056368572847e-06, |
| "loss": 1.28662109375, |
| "step": 38 |
| }, |
| { |
| "epoch": 1.5, |
| "grad_norm": 0.12339212745428085, |
| "learning_rate": 1.6843867087960252e-06, |
| "loss": 1.31494140625, |
| "step": 39 |
| }, |
| { |
| "epoch": 1.5384615384615383, |
| "grad_norm": 0.11346733570098877, |
| "learning_rate": 1.4644660940672628e-06, |
| "loss": 1.284912109375, |
| "step": 40 |
| }, |
| { |
| "epoch": 1.5769230769230769, |
| "grad_norm": 0.10992596298456192, |
| "learning_rate": 1.257446259144494e-06, |
| "loss": 1.265869140625, |
| "step": 41 |
| }, |
| { |
| "epoch": 1.6153846153846154, |
| "grad_norm": 0.1381499022245407, |
| "learning_rate": 1.0640825969547498e-06, |
| "loss": 1.26763916015625, |
| "step": 42 |
| }, |
| { |
| "epoch": 1.6538461538461537, |
| "grad_norm": 0.10350609570741653, |
| "learning_rate": 8.850806705317183e-07, |
| "loss": 1.2626953125, |
| "step": 43 |
| }, |
| { |
| "epoch": 1.6923076923076923, |
| "grad_norm": 0.10742916166782379, |
| "learning_rate": 7.210936384927631e-07, |
| "loss": 1.2371826171875, |
| "step": 44 |
| }, |
| { |
| "epoch": 1.7307692307692308, |
| "grad_norm": 0.11156682670116425, |
| "learning_rate": 5.727198717339511e-07, |
| "loss": 1.30615234375, |
| "step": 45 |
| }, |
| { |
| "epoch": 1.7692307692307692, |
| "grad_norm": 0.10545119643211365, |
| "learning_rate": 4.405007700395497e-07, |
| "loss": 1.271728515625, |
| "step": 46 |
| }, |
| { |
| "epoch": 1.8076923076923077, |
| "grad_norm": 0.1204240471124649, |
| "learning_rate": 3.2491878657292643e-07, |
| "loss": 1.3251953125, |
| "step": 47 |
| }, |
| { |
| "epoch": 1.8461538461538463, |
| "grad_norm": 0.12353746592998505, |
| "learning_rate": 2.2639566745727203e-07, |
| "loss": 1.226898193359375, |
| "step": 48 |
| }, |
| { |
| "epoch": 1.8846153846153846, |
| "grad_norm": 0.11921994388103485, |
| "learning_rate": 1.4529091286973994e-07, |
| "loss": 1.27685546875, |
| "step": 49 |
| }, |
| { |
| "epoch": 1.9230769230769231, |
| "grad_norm": 0.11171863973140717, |
| "learning_rate": 8.190046526428241e-08, |
| "loss": 1.25927734375, |
| "step": 50 |
| }, |
| { |
| "epoch": 1.9615384615384617, |
| "grad_norm": 0.11080524325370789, |
| "learning_rate": 3.645562950973014e-08, |
| "loss": 1.3271484375, |
| "step": 51 |
| }, |
| { |
| "epoch": 2.0, |
| "grad_norm": 0.10683345049619675, |
| "learning_rate": 9.12222888341252e-09, |
| "loss": 1.28271484375, |
| "step": 52 |
| }, |
| { |
| "epoch": 2.0, |
| "eval_loss": 1.32275390625, |
| "eval_runtime": 10.3528, |
| "eval_samples_per_second": 0.29, |
| "eval_steps_per_second": 0.097, |
| "step": 52 |
| } |
| ], |
| "logging_steps": 1.0, |
| "max_steps": 52, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 2, |
| "save_steps": 0, |
| "stateful_callbacks": { |
| "TrainerControl": { |
| "args": { |
| "should_epoch_stop": false, |
| "should_evaluate": false, |
| "should_log": false, |
| "should_save": true, |
| "should_training_stop": true |
| }, |
| "attributes": {} |
| } |
| }, |
| "total_flos": 5.690612945990451e+17, |
| "train_batch_size": 1, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
