| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 2.9955686853766617, | |
| "eval_steps": 500, | |
| "global_step": 507, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.29542097488921715, | |
| "grad_norm": 1.8936359882354736, | |
| "learning_rate": 2.5e-05, | |
| "loss": 2.2777, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.5908419497784343, | |
| "grad_norm": 0.42867106199264526, | |
| "learning_rate": 5e-05, | |
| "loss": 0.9288, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.8862629246676514, | |
| "grad_norm": 0.5983214378356934, | |
| "learning_rate": 4.726477024070022e-05, | |
| "loss": 0.6469, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.1816838995568686, | |
| "grad_norm": 0.5473693013191223, | |
| "learning_rate": 4.452954048140044e-05, | |
| "loss": 0.5595, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.4771048744460857, | |
| "grad_norm": 0.5971556901931763, | |
| "learning_rate": 4.179431072210066e-05, | |
| "loss": 0.537, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 1.7725258493353029, | |
| "grad_norm": 0.6025377511978149, | |
| "learning_rate": 3.9059080962800876e-05, | |
| "loss": 0.5271, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.06794682422452, | |
| "grad_norm": 0.9687663912773132, | |
| "learning_rate": 3.6323851203501094e-05, | |
| "loss": 0.5318, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 2.363367799113737, | |
| "grad_norm": 0.673847496509552, | |
| "learning_rate": 3.358862144420131e-05, | |
| "loss": 0.5116, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 2.658788774002954, | |
| "grad_norm": 0.7841825485229492, | |
| "learning_rate": 3.085339168490153e-05, | |
| "loss": 0.5114, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 2.9542097488921715, | |
| "grad_norm": 0.8053774237632751, | |
| "learning_rate": 2.811816192560175e-05, | |
| "loss": 0.5049, | |
| "step": 500 | |
| } | |
| ], | |
| "logging_steps": 50, | |
| "max_steps": 1014, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 6, | |
| "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": 8.019802466942976e+16, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |