| { | |
| "best_global_step": null, | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 2.57516339869281, | |
| "eval_steps": 10, | |
| "global_step": 100, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.26143790849673204, | |
| "grad_norm": 80.0328369140625, | |
| "learning_rate": 1.9682539682539684e-05, | |
| "loss": 22.6724, | |
| "mean_token_accuracy": 0.49049999862909316, | |
| "num_tokens": 49120.0, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.26143790849673204, | |
| "eval_loss": 4.981190204620361, | |
| "eval_mean_token_accuracy": 0.5223076802033645, | |
| "eval_num_tokens": 49120.0, | |
| "eval_runtime": 17.5192, | |
| "eval_samples_per_second": 2.854, | |
| "eval_steps_per_second": 0.742, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.5228758169934641, | |
| "grad_norm": 57.1182746887207, | |
| "learning_rate": 1.8624338624338625e-05, | |
| "loss": 17.5658, | |
| "mean_token_accuracy": 0.5591249987483025, | |
| "num_tokens": 98240.0, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.5228758169934641, | |
| "eval_loss": 3.730682134628296, | |
| "eval_mean_token_accuracy": 0.5742307580434359, | |
| "eval_num_tokens": 98240.0, | |
| "eval_runtime": 17.4511, | |
| "eval_samples_per_second": 2.865, | |
| "eval_steps_per_second": 0.745, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.7843137254901961, | |
| "grad_norm": 58.24352264404297, | |
| "learning_rate": 1.7566137566137566e-05, | |
| "loss": 13.113, | |
| "mean_token_accuracy": 0.6388749912381172, | |
| "num_tokens": 147360.0, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.7843137254901961, | |
| "eval_loss": 2.7529046535491943, | |
| "eval_mean_token_accuracy": 0.7115384431985708, | |
| "eval_num_tokens": 147360.0, | |
| "eval_runtime": 17.6786, | |
| "eval_samples_per_second": 2.828, | |
| "eval_steps_per_second": 0.735, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 1.026143790849673, | |
| "grad_norm": 30.73020362854004, | |
| "learning_rate": 1.6507936507936507e-05, | |
| "loss": 8.7369, | |
| "mean_token_accuracy": 0.7555405417004147, | |
| "num_tokens": 192796.0, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 1.026143790849673, | |
| "eval_loss": 1.836286187171936, | |
| "eval_mean_token_accuracy": 0.8476923199800345, | |
| "eval_num_tokens": 192796.0, | |
| "eval_runtime": 17.4538, | |
| "eval_samples_per_second": 2.865, | |
| "eval_steps_per_second": 0.745, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 1.287581699346405, | |
| "grad_norm": 21.440105438232422, | |
| "learning_rate": 1.544973544973545e-05, | |
| "loss": 6.2989, | |
| "mean_token_accuracy": 0.8789999932050705, | |
| "num_tokens": 241916.0, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 1.287581699346405, | |
| "eval_loss": 1.3245856761932373, | |
| "eval_mean_token_accuracy": 0.8942307508908786, | |
| "eval_num_tokens": 241916.0, | |
| "eval_runtime": 17.6384, | |
| "eval_samples_per_second": 2.835, | |
| "eval_steps_per_second": 0.737, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 1.5490196078431373, | |
| "grad_norm": 10.908184051513672, | |
| "learning_rate": 1.4391534391534392e-05, | |
| "loss": 4.5427, | |
| "mean_token_accuracy": 0.9057500049471855, | |
| "num_tokens": 291036.0, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 1.5490196078431373, | |
| "eval_loss": 0.9362902045249939, | |
| "eval_mean_token_accuracy": 0.9126923221808213, | |
| "eval_num_tokens": 291036.0, | |
| "eval_runtime": 17.472, | |
| "eval_samples_per_second": 2.862, | |
| "eval_steps_per_second": 0.744, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 1.8104575163398693, | |
| "grad_norm": 11.274917602539062, | |
| "learning_rate": 1.3333333333333333e-05, | |
| "loss": 3.4538, | |
| "mean_token_accuracy": 0.9126250177621842, | |
| "num_tokens": 340156.0, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 1.8104575163398693, | |
| "eval_loss": 0.7567419409751892, | |
| "eval_mean_token_accuracy": 0.9103846183189979, | |
| "eval_num_tokens": 340156.0, | |
| "eval_runtime": 17.5544, | |
| "eval_samples_per_second": 2.848, | |
| "eval_steps_per_second": 0.741, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 2.052287581699346, | |
| "grad_norm": 13.183298110961914, | |
| "learning_rate": 1.2275132275132276e-05, | |
| "loss": 2.5315, | |
| "mean_token_accuracy": 0.9293243321212562, | |
| "num_tokens": 385592.0, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 2.052287581699346, | |
| "eval_loss": 0.6147329807281494, | |
| "eval_mean_token_accuracy": 0.9292307771169223, | |
| "eval_num_tokens": 385592.0, | |
| "eval_runtime": 17.582, | |
| "eval_samples_per_second": 2.844, | |
| "eval_steps_per_second": 0.739, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 2.313725490196078, | |
| "grad_norm": 12.541313171386719, | |
| "learning_rate": 1.1216931216931217e-05, | |
| "loss": 2.2552, | |
| "mean_token_accuracy": 0.9353750020265579, | |
| "num_tokens": 434712.0, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 2.313725490196078, | |
| "eval_loss": 0.5065863132476807, | |
| "eval_mean_token_accuracy": 0.9465384529187129, | |
| "eval_num_tokens": 434712.0, | |
| "eval_runtime": 17.6818, | |
| "eval_samples_per_second": 2.828, | |
| "eval_steps_per_second": 0.735, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 2.57516339869281, | |
| "grad_norm": 7.007925033569336, | |
| "learning_rate": 1.015873015873016e-05, | |
| "loss": 1.9547, | |
| "mean_token_accuracy": 0.9474999904632568, | |
| "num_tokens": 483832.0, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 2.57516339869281, | |
| "eval_loss": 0.4519544243812561, | |
| "eval_mean_token_accuracy": 0.9511538331325238, | |
| "eval_num_tokens": 483832.0, | |
| "eval_runtime": 17.5399, | |
| "eval_samples_per_second": 2.851, | |
| "eval_steps_per_second": 0.741, | |
| "step": 100 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 195, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 5, | |
| "save_steps": 20, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 1.2581157053946624e+16, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |