{ "best_metric": 0.6611698269844055, "best_model_checkpoint": "./amazon-reviews-distilbert-base-sentiment/checkpoint-1500", "epoch": 0.2555366269165247, "eval_steps": 500, "global_step": 1500, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.09, "learning_rate": 1.943214082907439e-05, "loss": 0.0657, "step": 500 }, { "epoch": 0.09, "eval_accuracy": 0.8683166745993971, "eval_f1": 0.7203595462721784, "eval_loss": 0.8777859807014465, "eval_precision": 0.7011145822442907, "eval_recall": 0.7614722170387864, "eval_runtime": 100.8555, "eval_samples_per_second": 62.495, "eval_steps_per_second": 3.907, "step": 500 }, { "epoch": 0.17, "learning_rate": 1.886428165814878e-05, "loss": 0.1101, "step": 1000 }, { "epoch": 0.17, "eval_accuracy": 0.8534031413612565, "eval_f1": 0.7086122982138194, "eval_loss": 0.7233699560165405, "eval_precision": 0.6832545369711506, "eval_recall": 0.76652546254752, "eval_runtime": 100.9638, "eval_samples_per_second": 62.428, "eval_steps_per_second": 3.902, "step": 1000 }, { "epoch": 0.26, "learning_rate": 1.829642248722317e-05, "loss": 0.1218, "step": 1500 }, { "epoch": 0.26, "eval_accuracy": 0.8714897667777249, "eval_f1": 0.724263640834165, "eval_loss": 0.6611698269844055, "eval_precision": 0.7057228828259511, "eval_recall": 0.7653889499971643, "eval_runtime": 100.7702, "eval_samples_per_second": 62.548, "eval_steps_per_second": 3.91, "step": 1500 } ], "logging_steps": 500, "max_steps": 17610, "num_train_epochs": 3, "save_steps": 500, "total_flos": 3179274264576000.0, "trial_name": null, "trial_params": null }