| { | |
| "best_global_step": 1357, | |
| "best_metric": 0.5547252297401428, | |
| "best_model_checkpoint": "./sentiment_reddit/checkpoint-1357", | |
| "epoch": 3.0, | |
| "eval_steps": 500, | |
| "global_step": 4071, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.36845983787767134, | |
| "grad_norm": 11.603619575500488, | |
| "learning_rate": 1.7577990665684107e-05, | |
| "loss": 0.6669, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.7369196757553427, | |
| "grad_norm": 4.453907489776611, | |
| "learning_rate": 1.5121591746499633e-05, | |
| "loss": 0.5746, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.7742351640250645, | |
| "eval_f1": 0.7421323868961363, | |
| "eval_loss": 0.5547252297401428, | |
| "eval_runtime": 9.1842, | |
| "eval_samples_per_second": 590.8, | |
| "eval_steps_per_second": 18.51, | |
| "step": 1357 | |
| }, | |
| { | |
| "epoch": 1.105379513633014, | |
| "grad_norm": 5.306099891662598, | |
| "learning_rate": 1.2665192827315159e-05, | |
| "loss": 0.5286, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.4738393515106853, | |
| "grad_norm": 6.835700511932373, | |
| "learning_rate": 1.021370670596905e-05, | |
| "loss": 0.4861, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 1.8422991893883567, | |
| "grad_norm": 5.161911964416504, | |
| "learning_rate": 7.757307786784575e-06, | |
| "loss": 0.4758, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.7722078879469222, | |
| "eval_f1": 0.7381925712914518, | |
| "eval_loss": 0.5592826008796692, | |
| "eval_runtime": 9.2782, | |
| "eval_samples_per_second": 584.814, | |
| "eval_steps_per_second": 18.323, | |
| "step": 2714 | |
| }, | |
| { | |
| "epoch": 2.210759027266028, | |
| "grad_norm": 9.822325706481934, | |
| "learning_rate": 5.300908867600099e-06, | |
| "loss": 0.4378, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 2.5792188651436994, | |
| "grad_norm": 9.744635581970215, | |
| "learning_rate": 2.844509948415623e-06, | |
| "loss": 0.4143, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 2.9476787030213707, | |
| "grad_norm": 4.211618900299072, | |
| "learning_rate": 3.881110292311472e-07, | |
| "loss": 0.3932, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_accuracy": 0.7688905270917803, | |
| "eval_f1": 0.7422583213361386, | |
| "eval_loss": 0.6015889048576355, | |
| "eval_runtime": 12.4015, | |
| "eval_samples_per_second": 437.528, | |
| "eval_steps_per_second": 13.708, | |
| "step": 4071 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 4071, | |
| "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": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 8566315097679360.0, | |
| "train_batch_size": 32, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |