| { | |
| "best_metric": 0.8600789143532662, | |
| "best_model_checkpoint": "training_with_callbacks/checkpoint-7880", | |
| "epoch": 10.0, | |
| "global_step": 15760, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.32, | |
| "learning_rate": 1.9368020304568528e-05, | |
| "loss": 0.5589, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.63, | |
| "learning_rate": 1.8733502538071068e-05, | |
| "loss": 0.4001, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.95, | |
| "learning_rate": 1.81002538071066e-05, | |
| "loss": 0.3671, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.8556491163135005, | |
| "eval_loss": 0.33663880825042725, | |
| "eval_macro_f1": 0.8274802013644363, | |
| "eval_micro_f1": 0.8556491163135005, | |
| "eval_runtime": 45.4196, | |
| "eval_samples_per_second": 237.937, | |
| "eval_steps_per_second": 14.883, | |
| "step": 1576 | |
| }, | |
| { | |
| "epoch": 1.27, | |
| "learning_rate": 1.7465736040609138e-05, | |
| "loss": 0.3096, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 1.59, | |
| "learning_rate": 1.6831218274111674e-05, | |
| "loss": 0.2976, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 1.9, | |
| "learning_rate": 1.6196700507614215e-05, | |
| "loss": 0.3043, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.8647173128527806, | |
| "eval_loss": 0.3377376198768616, | |
| "eval_macro_f1": 0.8410857961930669, | |
| "eval_micro_f1": 0.8647173128527806, | |
| "eval_runtime": 45.0937, | |
| "eval_samples_per_second": 239.656, | |
| "eval_steps_per_second": 14.991, | |
| "step": 3152 | |
| }, | |
| { | |
| "epoch": 2.22, | |
| "learning_rate": 1.5562182741116755e-05, | |
| "loss": 0.2492, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 2.54, | |
| "learning_rate": 1.4927664974619292e-05, | |
| "loss": 0.2329, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 2.86, | |
| "learning_rate": 1.4293147208121828e-05, | |
| "loss": 0.2413, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_accuracy": 0.8739705746275562, | |
| "eval_loss": 0.32710105180740356, | |
| "eval_macro_f1": 0.8518710972157342, | |
| "eval_micro_f1": 0.8739705746275562, | |
| "eval_runtime": 45.1151, | |
| "eval_samples_per_second": 239.543, | |
| "eval_steps_per_second": 14.984, | |
| "step": 4728 | |
| }, | |
| { | |
| "epoch": 3.17, | |
| "learning_rate": 1.3658629441624367e-05, | |
| "loss": 0.2021, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 3.49, | |
| "learning_rate": 1.30253807106599e-05, | |
| "loss": 0.1933, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 3.81, | |
| "learning_rate": 1.2392131979695431e-05, | |
| "loss": 0.1887, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.872119922272601, | |
| "eval_loss": 0.367958664894104, | |
| "eval_macro_f1": 0.8499428993945666, | |
| "eval_micro_f1": 0.8721199222726009, | |
| "eval_runtime": 45.2612, | |
| "eval_samples_per_second": 238.77, | |
| "eval_steps_per_second": 14.936, | |
| "step": 6304 | |
| }, | |
| { | |
| "epoch": 4.12, | |
| "learning_rate": 1.175761421319797e-05, | |
| "loss": 0.1702, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 4.44, | |
| "learning_rate": 1.1123096446700508e-05, | |
| "loss": 0.1574, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 4.76, | |
| "learning_rate": 1.0488578680203047e-05, | |
| "loss": 0.1494, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "eval_accuracy": 0.8798926621634126, | |
| "eval_loss": 0.3825337588787079, | |
| "eval_macro_f1": 0.8600789143532662, | |
| "eval_micro_f1": 0.8798926621634126, | |
| "eval_runtime": 44.9299, | |
| "eval_samples_per_second": 240.53, | |
| "eval_steps_per_second": 15.046, | |
| "step": 7880 | |
| }, | |
| { | |
| "epoch": 5.08, | |
| "learning_rate": 9.855329949238578e-06, | |
| "loss": 0.1449, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 5.39, | |
| "learning_rate": 9.220812182741118e-06, | |
| "loss": 0.1263, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 5.71, | |
| "learning_rate": 8.586294416243655e-06, | |
| "loss": 0.1255, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "eval_accuracy": 0.8775793467197187, | |
| "eval_loss": 0.471343457698822, | |
| "eval_macro_f1": 0.8558476837008455, | |
| "eval_micro_f1": 0.8775793467197187, | |
| "eval_runtime": 45.3421, | |
| "eval_samples_per_second": 238.343, | |
| "eval_steps_per_second": 14.909, | |
| "step": 9456 | |
| }, | |
| { | |
| "epoch": 6.03, | |
| "learning_rate": 7.951776649746194e-06, | |
| "loss": 0.1259, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 6.35, | |
| "learning_rate": 7.319796954314721e-06, | |
| "loss": 0.0964, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 6.66, | |
| "learning_rate": 6.68527918781726e-06, | |
| "loss": 0.1086, | |
| "step": 10500 | |
| }, | |
| { | |
| "epoch": 6.98, | |
| "learning_rate": 6.050761421319798e-06, | |
| "loss": 0.1077, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 7.0, | |
| "eval_accuracy": 0.8795225316924216, | |
| "eval_loss": 0.5118919610977173, | |
| "eval_macro_f1": 0.8578917168529726, | |
| "eval_micro_f1": 0.8795225316924216, | |
| "eval_runtime": 45.4537, | |
| "eval_samples_per_second": 237.759, | |
| "eval_steps_per_second": 14.872, | |
| "step": 11032 | |
| }, | |
| { | |
| "epoch": 7.3, | |
| "learning_rate": 5.416243654822335e-06, | |
| "loss": 0.0846, | |
| "step": 11500 | |
| }, | |
| { | |
| "epoch": 7.61, | |
| "learning_rate": 4.781725888324874e-06, | |
| "loss": 0.0851, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 7.93, | |
| "learning_rate": 4.1472081218274114e-06, | |
| "loss": 0.0875, | |
| "step": 12500 | |
| }, | |
| { | |
| "epoch": 8.0, | |
| "eval_accuracy": 0.8807254557231424, | |
| "eval_loss": 0.5281736254692078, | |
| "eval_macro_f1": 0.8573342434559597, | |
| "eval_micro_f1": 0.8807254557231424, | |
| "eval_runtime": 45.6116, | |
| "eval_samples_per_second": 236.935, | |
| "eval_steps_per_second": 14.821, | |
| "step": 12608 | |
| }, | |
| { | |
| "epoch": 8.25, | |
| "learning_rate": 3.5126903553299495e-06, | |
| "loss": 0.0705, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 8.57, | |
| "learning_rate": 2.8781725888324876e-06, | |
| "loss": 0.0711, | |
| "step": 13500 | |
| }, | |
| { | |
| "epoch": 8.88, | |
| "learning_rate": 2.2449238578680207e-06, | |
| "loss": 0.0702, | |
| "step": 14000 | |
| }, | |
| { | |
| "epoch": 9.0, | |
| "eval_accuracy": 0.8808179883408902, | |
| "eval_loss": 0.591612696647644, | |
| "eval_macro_f1": 0.8563906267141775, | |
| "eval_micro_f1": 0.8808179883408902, | |
| "eval_runtime": 45.3025, | |
| "eval_samples_per_second": 238.552, | |
| "eval_steps_per_second": 14.922, | |
| "step": 14184 | |
| }, | |
| { | |
| "epoch": 9.2, | |
| "learning_rate": 1.6116751269035535e-06, | |
| "loss": 0.063, | |
| "step": 14500 | |
| }, | |
| { | |
| "epoch": 9.52, | |
| "learning_rate": 9.771573604060914e-07, | |
| "loss": 0.0609, | |
| "step": 15000 | |
| }, | |
| { | |
| "epoch": 9.84, | |
| "learning_rate": 3.4263959390862946e-07, | |
| "loss": 0.0539, | |
| "step": 15500 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_accuracy": 0.8817433145183677, | |
| "eval_loss": 0.6316967010498047, | |
| "eval_macro_f1": 0.8560485298070233, | |
| "eval_micro_f1": 0.8817433145183678, | |
| "eval_runtime": 45.5653, | |
| "eval_samples_per_second": 237.176, | |
| "eval_steps_per_second": 14.836, | |
| "step": 15760 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "step": 15760, | |
| "total_flos": 1.3269191906021376e+17, | |
| "train_loss": 0.1755784906106552, | |
| "train_runtime": 5214.2121, | |
| "train_samples_per_second": 96.718, | |
| "train_steps_per_second": 3.023 | |
| } | |
| ], | |
| "max_steps": 15760, | |
| "num_train_epochs": 10, | |
| "total_flos": 1.3269191906021376e+17, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |