| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 5.0, | |
| "eval_steps": 500, | |
| "global_step": 12535, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.1994415636218588, | |
| "grad_norm": 1.8192601203918457, | |
| "learning_rate": 4.8005584363781416e-05, | |
| "loss": 7.5114, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.3988831272437176, | |
| "grad_norm": 1.543924331665039, | |
| "learning_rate": 4.601116872756282e-05, | |
| "loss": 6.6557, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.5983246908655764, | |
| "grad_norm": 1.7416774034500122, | |
| "learning_rate": 4.4016753091344236e-05, | |
| "loss": 6.2893, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.7977662544874352, | |
| "grad_norm": 1.72893488407135, | |
| "learning_rate": 4.202233745512565e-05, | |
| "loss": 6.0012, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.9972078181092939, | |
| "grad_norm": 2.0189125537872314, | |
| "learning_rate": 4.002792181890706e-05, | |
| "loss": 5.7968, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 1.1966493817311528, | |
| "grad_norm": 1.9939076900482178, | |
| "learning_rate": 3.803350618268847e-05, | |
| "loss": 5.5695, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 1.3960909453530115, | |
| "grad_norm": 2.0916945934295654, | |
| "learning_rate": 3.6039090546469884e-05, | |
| "loss": 5.4321, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 1.5955325089748702, | |
| "grad_norm": 2.1514716148376465, | |
| "learning_rate": 3.4044674910251304e-05, | |
| "loss": 5.3137, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 1.7949740725967291, | |
| "grad_norm": 2.0663416385650635, | |
| "learning_rate": 3.205025927403271e-05, | |
| "loss": 5.2554, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 1.994415636218588, | |
| "grad_norm": 2.196505546569824, | |
| "learning_rate": 3.0055843637814124e-05, | |
| "loss": 5.171, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 2.193857199840447, | |
| "grad_norm": 2.258173704147339, | |
| "learning_rate": 2.8061428001595534e-05, | |
| "loss": 4.9799, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 2.3932987634623055, | |
| "grad_norm": 2.3806893825531006, | |
| "learning_rate": 2.6067012365376948e-05, | |
| "loss": 4.9378, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 2.5927403270841642, | |
| "grad_norm": 2.332186222076416, | |
| "learning_rate": 2.4072596729158358e-05, | |
| "loss": 4.9042, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 2.792181890706023, | |
| "grad_norm": 2.366901159286499, | |
| "learning_rate": 2.207818109293977e-05, | |
| "loss": 4.8437, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 2.9916234543278817, | |
| "grad_norm": 2.328530788421631, | |
| "learning_rate": 2.008376545672118e-05, | |
| "loss": 4.806, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 3.191065017949741, | |
| "grad_norm": 2.5201773643493652, | |
| "learning_rate": 1.8089349820502595e-05, | |
| "loss": 4.6991, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 3.3905065815715996, | |
| "grad_norm": 2.608670949935913, | |
| "learning_rate": 1.6098923015556443e-05, | |
| "loss": 4.6476, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 3.5899481451934583, | |
| "grad_norm": 2.5904150009155273, | |
| "learning_rate": 1.4104507379337855e-05, | |
| "loss": 4.6051, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 3.789389708815317, | |
| "grad_norm": 2.5216405391693115, | |
| "learning_rate": 1.2110091743119267e-05, | |
| "loss": 4.6082, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 3.988831272437176, | |
| "grad_norm": 2.5698513984680176, | |
| "learning_rate": 1.0115676106900679e-05, | |
| "loss": 4.5688, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 4.188272836059035, | |
| "grad_norm": 2.6962502002716064, | |
| "learning_rate": 8.125249301954529e-06, | |
| "loss": 4.4808, | |
| "step": 10500 | |
| }, | |
| { | |
| "epoch": 4.387714399680894, | |
| "grad_norm": 2.8604938983917236, | |
| "learning_rate": 6.13083366573594e-06, | |
| "loss": 4.4745, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 4.587155963302752, | |
| "grad_norm": 2.820222854614258, | |
| "learning_rate": 4.136418029517352e-06, | |
| "loss": 4.4775, | |
| "step": 11500 | |
| }, | |
| { | |
| "epoch": 4.786597526924611, | |
| "grad_norm": 2.6641643047332764, | |
| "learning_rate": 2.1420023932987634e-06, | |
| "loss": 4.4463, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 4.98603909054647, | |
| "grad_norm": 2.8485090732574463, | |
| "learning_rate": 1.515755883526127e-07, | |
| "loss": 4.456, | |
| "step": 12500 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "step": 12535, | |
| "total_flos": 2.619975204864e+16, | |
| "train_loss": 5.155460170506956, | |
| "train_runtime": 6504.3825, | |
| "train_samples_per_second": 30.832, | |
| "train_steps_per_second": 1.927 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 12535, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 5, | |
| "save_steps": 500, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 2.619975204864e+16, | |
| "train_batch_size": 16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |