{ "best_metric": 0.6828581526810108, "best_model_checkpoint": "trained/hebban-reviews/bert-base-multilingual-cased/checkpoint-1500", "epoch": 4.382997370727432, "global_step": 5001, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.44, "learning_rate": 4.504099180163968e-05, "loss": 0.8498, "step": 500 }, { "epoch": 0.44, "eval_accuracy": 0.6671597633136095, "eval_f1": 0.6926558253677099, "eval_loss": 0.7364796996116638, "eval_precision": 0.7520529030378761, "eval_qwk": 0.5224288418460314, "eval_recall": 0.6671597633136095, "eval_runtime": 23.7829, "eval_samples_per_second": 682.171, "eval_steps_per_second": 5.34, "step": 500 }, { "epoch": 0.88, "learning_rate": 4.0041991601679665e-05, "loss": 0.6918, "step": 1000 }, { "epoch": 0.88, "eval_accuracy": 0.7271942800788954, "eval_f1": 0.7414418295002464, "eval_loss": 0.6605744957923889, "eval_precision": 0.7743449767911329, "eval_qwk": 0.5976543204298063, "eval_recall": 0.7271942800788954, "eval_runtime": 23.6177, "eval_samples_per_second": 686.943, "eval_steps_per_second": 5.377, "step": 1000 }, { "epoch": 1.31, "learning_rate": 3.504299140171966e-05, "loss": 0.621, "step": 1500 }, { "epoch": 1.31, "eval_accuracy": 0.7785996055226825, "eval_f1": 0.7848033042674865, "eval_loss": 0.6530709266662598, "eval_precision": 0.7942290484031055, "eval_qwk": 0.6828581526810108, "eval_recall": 0.7785996055226825, "eval_runtime": 23.6972, "eval_samples_per_second": 684.637, "eval_steps_per_second": 5.359, "step": 1500 }, { "epoch": 1.75, "learning_rate": 3.0043991201759648e-05, "loss": 0.5936, "step": 2000 }, { "epoch": 1.75, "eval_accuracy": 0.7501849112426036, "eval_f1": 0.7649773762321463, "eval_loss": 0.6148771643638611, "eval_precision": 0.7952109430358816, "eval_qwk": 0.6613885538385067, "eval_recall": 0.7501849112426036, "eval_runtime": 23.8294, "eval_samples_per_second": 680.838, "eval_steps_per_second": 5.33, "step": 2000 }, { "epoch": 2.19, "learning_rate": 2.504499100179964e-05, "loss": 0.5434, "step": 2500 }, { "epoch": 2.19, "eval_accuracy": 0.7548076923076923, "eval_f1": 0.7709100781310829, "eval_loss": 0.6947299242019653, "eval_precision": 0.8080766037849636, "eval_qwk": 0.6597495777741245, "eval_recall": 0.7548076923076923, "eval_runtime": 23.6425, "eval_samples_per_second": 686.22, "eval_steps_per_second": 5.372, "step": 2500 }, { "epoch": 2.63, "learning_rate": 2.0045990801839633e-05, "loss": 0.4808, "step": 3000 }, { "epoch": 2.63, "eval_accuracy": 0.7347140039447732, "eval_f1": 0.7545089976363922, "eval_loss": 0.645616352558136, "eval_precision": 0.8030403955873724, "eval_qwk": 0.6513487818562542, "eval_recall": 0.7347140039447732, "eval_runtime": 23.8892, "eval_samples_per_second": 679.136, "eval_steps_per_second": 5.316, "step": 3000 }, { "epoch": 3.07, "learning_rate": 1.5046990601879624e-05, "loss": 0.4643, "step": 3500 }, { "epoch": 3.07, "eval_accuracy": 0.7427268244575936, "eval_f1": 0.7611673940281857, "eval_loss": 0.7119439244270325, "eval_precision": 0.8053303788937228, "eval_qwk": 0.6549274566801152, "eval_recall": 0.7427268244575936, "eval_runtime": 23.6156, "eval_samples_per_second": 687.004, "eval_steps_per_second": 5.378, "step": 3500 }, { "epoch": 3.51, "learning_rate": 1.0047990401919617e-05, "loss": 0.3854, "step": 4000 }, { "epoch": 3.51, "eval_accuracy": 0.7685527613412229, "eval_f1": 0.7819273980616418, "eval_loss": 0.7430822253227234, "eval_precision": 0.8091264019259906, "eval_qwk": 0.6812961348271991, "eval_recall": 0.7685527613412229, "eval_runtime": 23.6906, "eval_samples_per_second": 684.828, "eval_steps_per_second": 5.361, "step": 4000 }, { "epoch": 3.94, "learning_rate": 5.048990201959608e-06, "loss": 0.3642, "step": 4500 }, { "epoch": 3.94, "eval_accuracy": 0.7693540433925049, "eval_f1": 0.7827174333765768, "eval_loss": 0.7832001447677612, "eval_precision": 0.8104759531984178, "eval_qwk": 0.6799731235431681, "eval_recall": 0.7693540433925049, "eval_runtime": 23.5733, "eval_samples_per_second": 688.237, "eval_steps_per_second": 5.387, "step": 4500 }, { "epoch": 4.38, "learning_rate": 4.999000199960008e-08, "loss": 0.3199, "step": 5000 }, { "epoch": 4.38, "eval_accuracy": 0.7682445759368837, "eval_f1": 0.7820013322186519, "eval_loss": 0.8646416664123535, "eval_precision": 0.8117448424467829, "eval_qwk": 0.6776167553082479, "eval_recall": 0.7682445759368837, "eval_runtime": 23.7003, "eval_samples_per_second": 684.547, "eval_steps_per_second": 5.359, "step": 5000 }, { "epoch": 4.38, "step": 5001, "total_flos": 1.6838837781764506e+17, "train_loss": 0.5313903366928219, "train_runtime": 2667.6757, "train_samples_per_second": 239.957, "train_steps_per_second": 1.875 } ], "max_steps": 5001, "num_train_epochs": 5, "total_flos": 1.6838837781764506e+17, "trial_name": null, "trial_params": null }