{ "best_metric": 0.7213629796675549, "best_model_checkpoint": "bart-base/checkpoint-1286", "epoch": 2.0, "eval_steps": 500, "global_step": 1286, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.78, "learning_rate": 4.871176775531364e-05, "loss": 1.2028, "step": 500 }, { "epoch": 1.0, "eval_accuracy": 0.7598760650658405, "eval_f1_macro": 0.6063644681337579, "eval_f1_micro": 0.7598760650658405, "eval_f1_weighted": 0.7489266744747745, "eval_loss": 0.8429557681083679, "eval_macro_fpr": 0.023226157303183535, "eval_macro_sensitivity": 0.6366904609623596, "eval_macro_specificity": 0.981694425945952, "eval_precision": 0.7600880594489849, "eval_precision_macro": 0.6003620126022547, "eval_recall": 0.7598760650658405, "eval_recall_macro": 0.6366904609623596, "eval_runtime": 43.5934, "eval_samples_per_second": 29.615, "eval_steps_per_second": 3.716, "eval_weighted_fpr": 0.02207348333808032, "eval_weighted_sensitivity": 0.7598760650658405, "eval_weighted_specificity": 0.9655403241234398, "step": 643 }, { "epoch": 1.56, "learning_rate": 4.741575946086055e-05, "loss": 0.715, "step": 1000 }, { "epoch": 2.0, "eval_accuracy": 0.7931835786212239, "eval_f1_macro": 0.7213629796675549, "eval_f1_micro": 0.793183578621224, "eval_f1_weighted": 0.7853477061503886, "eval_loss": 0.7331996560096741, "eval_macro_fpr": 0.018982515099395962, "eval_macro_sensitivity": 0.7320748785168393, "eval_macro_specificity": 0.9845140063777027, "eval_precision": 0.8019650926098448, "eval_precision_macro": 0.7386000214414028, "eval_recall": 0.7931835786212239, "eval_recall_macro": 0.7320748785168393, "eval_runtime": 77.2393, "eval_samples_per_second": 16.714, "eval_steps_per_second": 2.097, "eval_weighted_fpr": 0.018283914264192288, "eval_weighted_sensitivity": 0.7931835786212239, "eval_weighted_specificity": 0.9745265170443169, "step": 1286 } ], "logging_steps": 500, "max_steps": 19290, "num_train_epochs": 30, "save_steps": 500, "total_flos": 3151833790500864.0, "trial_name": null, "trial_params": null }