| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 2.8054862842892767, | |
| "eval_steps": 500, | |
| "global_step": 13500, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.10390689941812137, | |
| "grad_norm": 1.8855979442596436, | |
| "learning_rate": 1.9310058187863676e-05, | |
| "loss": 0.8022, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.20781379883624274, | |
| "grad_norm": 0.8622527122497559, | |
| "learning_rate": 1.86173455250762e-05, | |
| "loss": 0.5316, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.3117206982543641, | |
| "grad_norm": 1.404678225517273, | |
| "learning_rate": 1.7924632862288724e-05, | |
| "loss": 0.4832, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.41562759767248547, | |
| "grad_norm": 1.3559819459915161, | |
| "learning_rate": 1.7233305624826823e-05, | |
| "loss": 0.4518, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.5195344970906068, | |
| "grad_norm": 0.8163271546363831, | |
| "learning_rate": 1.6540592962039347e-05, | |
| "loss": 0.4389, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.6234413965087282, | |
| "grad_norm": 0.8109046816825867, | |
| "learning_rate": 1.584788029925187e-05, | |
| "loss": 0.4193, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.7273482959268496, | |
| "grad_norm": 1.0217444896697998, | |
| "learning_rate": 1.5155167636464397e-05, | |
| "loss": 0.4215, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 0.8312551953449709, | |
| "grad_norm": 1.6476292610168457, | |
| "learning_rate": 1.446245497367692e-05, | |
| "loss": 0.4062, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.9351620947630923, | |
| "grad_norm": 1.4694277048110962, | |
| "learning_rate": 1.3769742310889445e-05, | |
| "loss": 0.4029, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_loss": 0.3212089240550995, | |
| "eval_runtime": 31.4577, | |
| "eval_samples_per_second": 15.704, | |
| "eval_steps_per_second": 7.852, | |
| "step": 4812 | |
| }, | |
| { | |
| "epoch": 1.0390689941812137, | |
| "grad_norm": 0.6561925411224365, | |
| "learning_rate": 1.3077029648101969e-05, | |
| "loss": 0.4059, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 1.142975893599335, | |
| "grad_norm": 0.6741281747817993, | |
| "learning_rate": 1.2384316985314493e-05, | |
| "loss": 0.3594, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 1.2468827930174564, | |
| "grad_norm": 0.48619207739830017, | |
| "learning_rate": 1.1691604322527017e-05, | |
| "loss": 0.3736, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 1.3507896924355778, | |
| "grad_norm": 1.1009119749069214, | |
| "learning_rate": 1.099889165973954e-05, | |
| "loss": 0.3624, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 1.4546965918536992, | |
| "grad_norm": 0.3497615456581116, | |
| "learning_rate": 1.0306178996952066e-05, | |
| "loss": 0.3516, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 1.5586034912718203, | |
| "grad_norm": 1.4209001064300537, | |
| "learning_rate": 9.61346633416459e-06, | |
| "loss": 0.3565, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 1.6625103906899419, | |
| "grad_norm": 0.8116744160652161, | |
| "learning_rate": 8.920753671377114e-06, | |
| "loss": 0.3635, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 1.766417290108063, | |
| "grad_norm": 0.8015578985214233, | |
| "learning_rate": 8.228041008589638e-06, | |
| "loss": 0.3549, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 1.8703241895261846, | |
| "grad_norm": 0.7980790734291077, | |
| "learning_rate": 7.536713771127737e-06, | |
| "loss": 0.3495, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 1.9742310889443058, | |
| "grad_norm": 1.4501579999923706, | |
| "learning_rate": 6.845386533665836e-06, | |
| "loss": 0.3385, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_loss": 0.29613471031188965, | |
| "eval_runtime": 31.4504, | |
| "eval_samples_per_second": 15.707, | |
| "eval_steps_per_second": 7.854, | |
| "step": 9624 | |
| }, | |
| { | |
| "epoch": 2.0781379883624274, | |
| "grad_norm": 0.6130263209342957, | |
| "learning_rate": 6.15267387087836e-06, | |
| "loss": 0.3293, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 2.1820448877805485, | |
| "grad_norm": 1.2724053859710693, | |
| "learning_rate": 5.459961208090885e-06, | |
| "loss": 0.3369, | |
| "step": 10500 | |
| }, | |
| { | |
| "epoch": 2.28595178719867, | |
| "grad_norm": 0.7700533270835876, | |
| "learning_rate": 4.767248545303408e-06, | |
| "loss": 0.3387, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 2.3898586866167912, | |
| "grad_norm": 1.4450799226760864, | |
| "learning_rate": 4.0759213078415074e-06, | |
| "loss": 0.3411, | |
| "step": 11500 | |
| }, | |
| { | |
| "epoch": 2.493765586034913, | |
| "grad_norm": 0.8265316486358643, | |
| "learning_rate": 3.3832086450540318e-06, | |
| "loss": 0.317, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 2.597672485453034, | |
| "grad_norm": 1.4168757200241089, | |
| "learning_rate": 2.690495982266556e-06, | |
| "loss": 0.3176, | |
| "step": 12500 | |
| }, | |
| { | |
| "epoch": 2.7015793848711556, | |
| "grad_norm": 1.0274338722229004, | |
| "learning_rate": 1.99778331947908e-06, | |
| "loss": 0.3389, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 2.8054862842892767, | |
| "grad_norm": 1.6529736518859863, | |
| "learning_rate": 1.3050706566916044e-06, | |
| "loss": 0.3163, | |
| "step": 13500 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 14436, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 500, | |
| "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.644064512933888e+16, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |