| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.050420168067227, | |
| "eval_steps": 100, | |
| "global_step": 250, | |
| "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 | |
| } | |
| ], | |
| "logging_steps": 25, | |
| "max_steps": 1000, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 5, | |
| "save_steps": 250, | |
| "total_flos": 2.30551294312448e+16, | |
| "train_batch_size": 1, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |