{ "best_metric": null, "best_model_checkpoint": null, "epoch": 3.0, "eval_steps": 500, "global_step": 22500, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.07, "learning_rate": 1.9333333333333333e-05, "loss": 0.0827, "step": 500 }, { "epoch": 0.13, "learning_rate": 1.866666666666667e-05, "loss": 0.0397, "step": 1000 }, { "epoch": 0.2, "learning_rate": 1.8e-05, "loss": 0.0337, "step": 1500 }, { "epoch": 0.27, "learning_rate": 1.7333333333333336e-05, "loss": 0.0327, "step": 2000 }, { "epoch": 0.33, "learning_rate": 1.6666666666666667e-05, "loss": 0.0313, "step": 2500 }, { "epoch": 0.4, "learning_rate": 1.6000000000000003e-05, "loss": 0.0272, "step": 3000 }, { "epoch": 0.47, "learning_rate": 1.5333333333333334e-05, "loss": 0.0236, "step": 3500 }, { "epoch": 0.53, "learning_rate": 1.4666666666666666e-05, "loss": 0.0263, "step": 4000 }, { "epoch": 0.6, "learning_rate": 1.4e-05, "loss": 0.0233, "step": 4500 }, { "epoch": 0.67, "learning_rate": 1.3333333333333333e-05, "loss": 0.0229, "step": 5000 }, { "epoch": 0.73, "learning_rate": 1.2666666666666667e-05, "loss": 0.022, "step": 5500 }, { "epoch": 0.8, "learning_rate": 1.2e-05, "loss": 0.0215, "step": 6000 }, { "epoch": 0.87, "learning_rate": 1.1333333333333334e-05, "loss": 0.0194, "step": 6500 }, { "epoch": 0.93, "learning_rate": 1.0666666666666667e-05, "loss": 0.0234, "step": 7000 }, { "epoch": 1.0, "learning_rate": 1e-05, "loss": 0.0182, "step": 7500 }, { "epoch": 1.07, "learning_rate": 9.333333333333334e-06, "loss": 0.0136, "step": 8000 }, { "epoch": 1.13, "learning_rate": 8.666666666666668e-06, "loss": 0.0104, "step": 8500 }, { "epoch": 1.2, "learning_rate": 8.000000000000001e-06, "loss": 0.0121, "step": 9000 }, { "epoch": 1.27, "learning_rate": 7.333333333333333e-06, "loss": 0.0122, "step": 9500 }, { "epoch": 1.33, "learning_rate": 6.666666666666667e-06, "loss": 0.0108, "step": 10000 }, { "epoch": 1.4, "learning_rate": 6e-06, "loss": 0.0107, "step": 10500 }, { "epoch": 1.47, "learning_rate": 5.333333333333334e-06, "loss": 0.0121, "step": 11000 }, { "epoch": 1.53, "learning_rate": 4.666666666666667e-06, "loss": 0.0085, "step": 11500 }, { "epoch": 1.6, "learning_rate": 4.000000000000001e-06, "loss": 0.0079, "step": 12000 }, { "epoch": 1.67, "learning_rate": 3.3333333333333333e-06, "loss": 0.0086, "step": 12500 }, { "epoch": 1.73, "learning_rate": 2.666666666666667e-06, "loss": 0.0095, "step": 13000 }, { "epoch": 1.8, "learning_rate": 2.0000000000000003e-06, "loss": 0.0092, "step": 13500 }, { "epoch": 1.87, "learning_rate": 1.3333333333333334e-06, "loss": 0.0103, "step": 14000 }, { "epoch": 1.93, "learning_rate": 6.666666666666667e-07, "loss": 0.0077, "step": 14500 }, { "epoch": 2.0, "learning_rate": 0.0, "loss": 0.009, "step": 15000 }, { "epoch": 2.07, "learning_rate": 6.222222222222223e-06, "loss": 0.0065, "step": 15500 }, { "epoch": 2.13, "learning_rate": 5.777777777777778e-06, "loss": 0.0075, "step": 16000 }, { "epoch": 2.2, "learning_rate": 5.333333333333334e-06, "loss": 0.0068, "step": 16500 }, { "epoch": 2.27, "learning_rate": 4.888888888888889e-06, "loss": 0.008, "step": 17000 }, { "epoch": 2.33, "learning_rate": 4.444444444444444e-06, "loss": 0.0074, "step": 17500 }, { "epoch": 2.4, "learning_rate": 4.000000000000001e-06, "loss": 0.0071, "step": 18000 }, { "epoch": 2.47, "learning_rate": 3.555555555555556e-06, "loss": 0.0076, "step": 18500 }, { "epoch": 2.53, "learning_rate": 3.1111111111111116e-06, "loss": 0.0057, "step": 19000 }, { "epoch": 2.6, "learning_rate": 2.666666666666667e-06, "loss": 0.0058, "step": 19500 }, { "epoch": 2.67, "learning_rate": 2.222222222222222e-06, "loss": 0.0058, "step": 20000 }, { "epoch": 2.73, "learning_rate": 1.777777777777778e-06, "loss": 0.0066, "step": 20500 }, { "epoch": 2.8, "learning_rate": 1.3333333333333334e-06, "loss": 0.0063, "step": 21000 }, { "epoch": 2.87, "learning_rate": 8.88888888888889e-07, "loss": 0.0045, "step": 21500 }, { "epoch": 2.93, "learning_rate": 4.444444444444445e-07, "loss": 0.0058, "step": 22000 }, { "epoch": 3.0, "learning_rate": 0.0, "loss": 0.0044, "step": 22500 }, { "epoch": 3.0, "step": 22500, "total_flos": 1.578694643712e+17, "train_loss": 0.0021281812138027613, "train_runtime": 4797.7609, "train_samples_per_second": 300.14, "train_steps_per_second": 4.69 } ], "logging_steps": 500, "max_steps": 22500, "num_input_tokens_seen": 0, "num_train_epochs": 3, "save_steps": 500, "total_flos": 1.578694643712e+17, "train_batch_size": 64, "trial_name": null, "trial_params": null }