{ "best_global_step": 3580, "best_metric": 0.9048878669310181, "best_model_checkpoint": "../models/checkpoint-3580", "epoch": 10.0, "eval_steps": 500, "global_step": 3580, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 1.0, "grad_norm": 13.927672386169434, "learning_rate": 1.8005586592178773e-05, "loss": 0.4010912506274005, "step": 358 }, { "epoch": 1.0, "eval_f1_macro": 0.6710618348718376, "eval_f1_micro": 0.6837422485845241, "eval_loss": 0.2920529246330261, "eval_precision": 0.7507400828892836, "eval_recall": 0.6277227722772277, "eval_runtime": 10.5624, "eval_samples_per_second": 131.882, "eval_steps_per_second": 8.331, "step": 358 }, { "epoch": 2.0, "grad_norm": 3.783374309539795, "learning_rate": 1.600558659217877e-05, "loss": 0.2375833735119697, "step": 716 }, { "epoch": 2.0, "eval_f1_macro": 0.7938948225039975, "eval_f1_micro": 0.7958968963703315, "eval_loss": 0.1952921748161316, "eval_precision": 0.8490460157126823, "eval_recall": 0.749009900990099, "eval_runtime": 10.5904, "eval_samples_per_second": 131.534, "eval_steps_per_second": 8.309, "step": 716 }, { "epoch": 3.0, "grad_norm": 2.0740878582000732, "learning_rate": 1.4005586592178772e-05, "loss": 0.16542356906656447, "step": 1074 }, { "epoch": 3.0, "eval_f1_macro": 0.8565937431886845, "eval_f1_micro": 0.8570709612481128, "eval_loss": 0.15338946878910065, "eval_precision": 0.8715455475946776, "eval_recall": 0.843069306930693, "eval_runtime": 10.5905, "eval_samples_per_second": 131.533, "eval_steps_per_second": 8.309, "step": 1074 }, { "epoch": 4.0, "grad_norm": 3.07139253616333, "learning_rate": 1.2005586592178773e-05, "loss": 0.12163071659024201, "step": 1432 }, { "epoch": 4.0, "eval_f1_macro": 0.8791528524021864, "eval_f1_micro": 0.8796576032225579, "eval_loss": 0.13414326310157776, "eval_precision": 0.8949795081967213, "eval_recall": 0.8648514851485148, "eval_runtime": 10.5851, "eval_samples_per_second": 131.601, "eval_steps_per_second": 8.314, "step": 1432 }, { "epoch": 5.0, "grad_norm": 11.252202987670898, "learning_rate": 1.0005586592178771e-05, "loss": 0.09038887876372098, "step": 1790 }, { "epoch": 5.0, "eval_f1_macro": 0.8942549448597952, "eval_f1_micro": 0.8939018344075359, "eval_loss": 0.12078466266393661, "eval_precision": 0.8952333664349553, "eval_recall": 0.8925742574257426, "eval_runtime": 10.5971, "eval_samples_per_second": 131.451, "eval_steps_per_second": 8.304, "step": 1790 }, { "epoch": 6.0, "grad_norm": 0.7471844553947449, "learning_rate": 8.005586592178772e-06, "loss": 0.07169087905457566, "step": 2148 }, { "epoch": 6.0, "eval_f1_macro": 0.89416078614729, "eval_f1_micro": 0.8939319190922546, "eval_loss": 0.12150303274393082, "eval_precision": 0.8908554572271387, "eval_recall": 0.897029702970297, "eval_runtime": 10.5789, "eval_samples_per_second": 131.677, "eval_steps_per_second": 8.318, "step": 2148 }, { "epoch": 7.0, "grad_norm": 10.221022605895996, "learning_rate": 6.005586592178772e-06, "loss": 0.0587206899120821, "step": 2506 }, { "epoch": 7.0, "eval_f1_macro": 0.8976220026825383, "eval_f1_micro": 0.8966528218910335, "eval_loss": 0.12085035443305969, "eval_precision": 0.8851905451037144, "eval_recall": 0.9084158415841584, "eval_runtime": 10.5927, "eval_samples_per_second": 131.506, "eval_steps_per_second": 8.308, "step": 2506 }, { "epoch": 8.0, "grad_norm": 0.19410130381584167, "learning_rate": 4.005586592178771e-06, "loss": 0.048420543777210084, "step": 2864 }, { "epoch": 8.0, "eval_f1_macro": 0.9033916836789936, "eval_f1_micro": 0.902893575282001, "eval_loss": 0.11863750964403152, "eval_precision": 0.8945578231292517, "eval_recall": 0.9113861386138614, "eval_runtime": 10.5889, "eval_samples_per_second": 131.553, "eval_steps_per_second": 8.311, "step": 2864 }, { "epoch": 9.0, "grad_norm": 3.716332197189331, "learning_rate": 2.0055865921787714e-06, "loss": 0.041742356795838424, "step": 3222 }, { "epoch": 9.0, "eval_f1_macro": 0.8996560535399232, "eval_f1_micro": 0.8987775061124694, "eval_loss": 0.1197083443403244, "eval_precision": 0.8879227053140096, "eval_recall": 0.9099009900990099, "eval_runtime": 10.5879, "eval_samples_per_second": 131.565, "eval_steps_per_second": 8.311, "step": 3222 }, { "epoch": 10.0, "grad_norm": 0.6163859367370605, "learning_rate": 5.5865921787709494e-09, "loss": 0.03748961933498276, "step": 3580 }, { "epoch": 10.0, "eval_f1_macro": 0.9048878669310181, "eval_f1_micro": 0.9043775984348251, "eval_loss": 0.11842884868383408, "eval_precision": 0.8936684388593523, "eval_recall": 0.9153465346534654, "eval_runtime": 10.5754, "eval_samples_per_second": 131.72, "eval_steps_per_second": 8.321, "step": 3580 } ], "logging_steps": 500, "max_steps": 3580, "num_input_tokens_seen": 0, "num_train_epochs": 10, "save_steps": 500, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": true }, "attributes": {} } }, "total_flos": 1.503497561751552e+16, "train_batch_size": 16, "trial_name": null, "trial_params": null }