{ "best_metric": 0.5495731830596924, "best_model_checkpoint": "Finetuned_distilbert/checkpoint-1000", "epoch": 7.0, "global_step": 7000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.5, "learning_rate": 1.9333333333333333e-05, "loss": 0.7074, "step": 500 }, { "epoch": 1.0, "learning_rate": 1.866666666666667e-05, "loss": 0.5864, "step": 1000 }, { "epoch": 1.0, "eval_accuracy": 0.7885, "eval_loss": 0.5495731830596924, "eval_runtime": 5.6815, "eval_samples_per_second": 352.018, "eval_steps_per_second": 44.002, "step": 1000 }, { "epoch": 1.5, "learning_rate": 1.8e-05, "loss": 0.4665, "step": 1500 }, { "epoch": 2.0, "learning_rate": 1.7333333333333336e-05, "loss": 0.4908, "step": 2000 }, { "epoch": 2.0, "eval_accuracy": 0.791, "eval_loss": 0.6509228944778442, "eval_runtime": 5.7448, "eval_samples_per_second": 348.143, "eval_steps_per_second": 43.518, "step": 2000 }, { "epoch": 2.5, "learning_rate": 1.6666666666666667e-05, "loss": 0.3567, "step": 2500 }, { "epoch": 3.0, "learning_rate": 1.6000000000000003e-05, "loss": 0.3684, "step": 3000 }, { "epoch": 3.0, "eval_accuracy": 0.787, "eval_loss": 0.8537377119064331, "eval_runtime": 5.7003, "eval_samples_per_second": 350.862, "eval_steps_per_second": 43.858, "step": 3000 }, { "epoch": 3.5, "learning_rate": 1.5333333333333334e-05, "loss": 0.2482, "step": 3500 }, { "epoch": 4.0, "learning_rate": 1.4666666666666666e-05, "loss": 0.2601, "step": 4000 }, { "epoch": 4.0, "eval_accuracy": 0.787, "eval_loss": 1.0617473125457764, "eval_runtime": 5.6754, "eval_samples_per_second": 352.397, "eval_steps_per_second": 44.05, "step": 4000 }, { "epoch": 4.5, "learning_rate": 1.4e-05, "loss": 0.1828, "step": 4500 }, { "epoch": 5.0, "learning_rate": 1.3333333333333333e-05, "loss": 0.1671, "step": 5000 }, { "epoch": 5.0, "eval_accuracy": 0.7825, "eval_loss": 1.2580657005310059, "eval_runtime": 5.6568, "eval_samples_per_second": 353.556, "eval_steps_per_second": 44.195, "step": 5000 }, { "epoch": 5.5, "learning_rate": 1.2666666666666667e-05, "loss": 0.1325, "step": 5500 }, { "epoch": 6.0, "learning_rate": 1.2e-05, "loss": 0.1454, "step": 6000 }, { "epoch": 6.0, "eval_accuracy": 0.7765, "eval_loss": 1.3939642906188965, "eval_runtime": 5.745, "eval_samples_per_second": 348.129, "eval_steps_per_second": 43.516, "step": 6000 }, { "epoch": 6.5, "learning_rate": 1.1333333333333334e-05, "loss": 0.0857, "step": 6500 }, { "epoch": 7.0, "learning_rate": 1.0666666666666667e-05, "loss": 0.1154, "step": 7000 }, { "epoch": 7.0, "eval_accuracy": 0.7845, "eval_loss": 1.5688139200210571, "eval_runtime": 5.715, "eval_samples_per_second": 349.955, "eval_steps_per_second": 43.744, "step": 7000 } ], "max_steps": 15000, "num_train_epochs": 15, "total_flos": 1224266732688384.0, "trial_name": null, "trial_params": null }