{ "best_global_step": 1357, "best_metric": 0.5547252297401428, "best_model_checkpoint": "./sentiment_reddit/checkpoint-1357", "epoch": 2.0, "eval_steps": 500, "global_step": 2714, "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 } ], "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": false }, "attributes": {} } }, "total_flos": 5710876731786240.0, "train_batch_size": 32, "trial_name": null, "trial_params": null }