{ "best_metric": 0.32759252190589905, "best_model_checkpoint": "electra-base-emotion-Tweet_About_Disaster_Or_Not/checkpoint-572", "epoch": 5.0, "global_step": 715, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.01, "learning_rate": 1.9972027972027975e-05, "loss": 0.651, "step": 1 }, { "epoch": 1.0, "learning_rate": 1.6000000000000003e-05, "loss": 0.4633, "step": 143 }, { "epoch": 1.0, "eval_F1": 0.6489563567362429, "eval_Precision": 0.5463258785942492, "eval_Recall": 0.7990654205607477, "eval_accuracy": 0.8372911169744943, "eval_loss": 0.3937690556049347, "eval_runtime": 16.8905, "eval_samples_per_second": 67.316, "eval_steps_per_second": 1.066, "step": 143 }, { "epoch": 2.0, "learning_rate": 1.2e-05, "loss": 0.3566, "step": 286 }, { "epoch": 2.0, "eval_F1": 0.6860465116279069, "eval_Precision": 0.5860927152317881, "eval_Recall": 0.8271028037383178, "eval_accuracy": 0.8575197889182058, "eval_loss": 0.35505905747413635, "eval_runtime": 17.7237, "eval_samples_per_second": 64.152, "eval_steps_per_second": 1.016, "step": 286 }, { "epoch": 3.0, "learning_rate": 8.000000000000001e-06, "loss": 0.3078, "step": 429 }, { "epoch": 3.0, "eval_F1": 0.7219730941704037, "eval_Precision": 0.6939655172413793, "eval_Recall": 0.7523364485981309, "eval_accuracy": 0.8909410729991205, "eval_loss": 0.34157443046569824, "eval_runtime": 17.809, "eval_samples_per_second": 63.844, "eval_steps_per_second": 1.011, "step": 429 }, { "epoch": 4.0, "learning_rate": 4.000000000000001e-06, "loss": 0.2813, "step": 572 }, { "epoch": 4.0, "eval_F1": 0.7245762711864407, "eval_Precision": 0.6627906976744186, "eval_Recall": 0.7990654205607477, "eval_accuracy": 0.8856640281442393, "eval_loss": 0.32759252190589905, "eval_runtime": 17.6367, "eval_samples_per_second": 64.468, "eval_steps_per_second": 1.021, "step": 572 }, { "epoch": 5.0, "learning_rate": 0.0, "loss": 0.2592, "step": 715 }, { "epoch": 5.0, "eval_F1": 0.7272727272727273, "eval_Precision": 0.6774193548387096, "eval_Recall": 0.7850467289719626, "eval_accuracy": 0.8891820580474934, "eval_loss": 0.3278656005859375, "eval_runtime": 18.9663, "eval_samples_per_second": 59.948, "eval_steps_per_second": 0.949, "step": 715 }, { "epoch": 5.0, "step": 715, "total_flos": 202034560394880.0, "train_loss": 0.3338941712479491, "train_runtime": 2347.1794, "train_samples_per_second": 19.376, "train_steps_per_second": 0.305 } ], "max_steps": 715, "num_train_epochs": 5, "total_flos": 202034560394880.0, "trial_name": null, "trial_params": null }