{ "best_metric": 0.6069164872169495, "best_model_checkpoint": "Ruster/checkpoint-498", "epoch": 2.0, "eval_steps": 500, "global_step": 498, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.07228915662650602, "grad_norm": 10.613821983337402, "learning_rate": 1.2e-05, "loss": 0.9401, "step": 18 }, { "epoch": 0.14457831325301204, "grad_norm": 6.847902297973633, "learning_rate": 2.4e-05, "loss": 0.4535, "step": 36 }, { "epoch": 0.21686746987951808, "grad_norm": 14.905878067016602, "learning_rate": 3.6e-05, "loss": 0.4731, "step": 54 }, { "epoch": 0.2891566265060241, "grad_norm": 5.860299587249756, "learning_rate": 4.8e-05, "loss": 0.3857, "step": 72 }, { "epoch": 0.3614457831325301, "grad_norm": 8.051860809326172, "learning_rate": 4.888392857142857e-05, "loss": 0.428, "step": 90 }, { "epoch": 0.43373493975903615, "grad_norm": 9.066169738769531, "learning_rate": 4.7544642857142855e-05, "loss": 0.3779, "step": 108 }, { "epoch": 0.5060240963855421, "grad_norm": 0.4762527346611023, "learning_rate": 4.620535714285715e-05, "loss": 0.3174, "step": 126 }, { "epoch": 0.5783132530120482, "grad_norm": 3.7264981269836426, "learning_rate": 4.486607142857143e-05, "loss": 0.3903, "step": 144 }, { "epoch": 0.6506024096385542, "grad_norm": 3.319960117340088, "learning_rate": 4.352678571428572e-05, "loss": 0.2664, "step": 162 }, { "epoch": 0.7228915662650602, "grad_norm": 8.442005157470703, "learning_rate": 4.21875e-05, "loss": 0.4303, "step": 180 }, { "epoch": 0.7951807228915663, "grad_norm": 5.5674729347229, "learning_rate": 4.084821428571429e-05, "loss": 0.3758, "step": 198 }, { "epoch": 0.8674698795180723, "grad_norm": 5.097784042358398, "learning_rate": 3.950892857142857e-05, "loss": 0.4321, "step": 216 }, { "epoch": 0.9397590361445783, "grad_norm": 4.135376930236816, "learning_rate": 3.816964285714286e-05, "loss": 0.3334, "step": 234 }, { "epoch": 1.0, "eval_accuracy": 0.5683646112600537, "eval_f1_macro": 0.34514324233232224, "eval_f1_micro": 0.5683646112600537, "eval_f1_weighted": 0.5027366446460666, "eval_loss": 0.8362799286842346, "eval_precision_macro": 0.4779331485680692, "eval_precision_micro": 0.5683646112600537, "eval_precision_weighted": 0.7304906422900805, "eval_recall_macro": 0.40111757925912955, "eval_recall_micro": 0.5683646112600537, "eval_recall_weighted": 0.5683646112600537, "eval_runtime": 197.9274, "eval_samples_per_second": 3.769, "eval_steps_per_second": 0.237, "step": 249 }, { "epoch": 1.0120481927710843, "grad_norm": 2.6794259548187256, "learning_rate": 3.683035714285715e-05, "loss": 0.3526, "step": 252 }, { "epoch": 1.0843373493975903, "grad_norm": 2.575113296508789, "learning_rate": 3.5491071428571435e-05, "loss": 0.2409, "step": 270 }, { "epoch": 1.1566265060240963, "grad_norm": 20.131845474243164, "learning_rate": 3.415178571428572e-05, "loss": 0.57, "step": 288 }, { "epoch": 1.2289156626506024, "grad_norm": 1.7430191040039062, "learning_rate": 3.2812500000000005e-05, "loss": 0.2363, "step": 306 }, { "epoch": 1.3012048192771084, "grad_norm": 2.6502227783203125, "learning_rate": 3.147321428571428e-05, "loss": 0.4027, "step": 324 }, { "epoch": 1.3734939759036144, "grad_norm": 0.4510114789009094, "learning_rate": 3.013392857142857e-05, "loss": 0.1746, "step": 342 }, { "epoch": 1.4457831325301205, "grad_norm": 1.9941872358322144, "learning_rate": 2.8794642857142857e-05, "loss": 0.3129, "step": 360 }, { "epoch": 1.5180722891566265, "grad_norm": 2.4119579792022705, "learning_rate": 2.7455357142857145e-05, "loss": 0.3607, "step": 378 }, { "epoch": 1.5903614457831325, "grad_norm": 3.2550156116485596, "learning_rate": 2.611607142857143e-05, "loss": 0.2441, "step": 396 }, { "epoch": 1.6626506024096386, "grad_norm": 1.5980372428894043, "learning_rate": 2.4776785714285715e-05, "loss": 0.2501, "step": 414 }, { "epoch": 1.7349397590361446, "grad_norm": 7.059596538543701, "learning_rate": 2.34375e-05, "loss": 0.41, "step": 432 }, { "epoch": 1.8072289156626506, "grad_norm": 1.7965036630630493, "learning_rate": 2.2098214285714286e-05, "loss": 0.323, "step": 450 }, { "epoch": 1.8795180722891565, "grad_norm": 7.194454669952393, "learning_rate": 2.0758928571428574e-05, "loss": 0.3068, "step": 468 }, { "epoch": 1.9518072289156625, "grad_norm": 3.057166814804077, "learning_rate": 1.941964285714286e-05, "loss": 0.3792, "step": 486 }, { "epoch": 2.0, "eval_accuracy": 0.7184986595174263, "eval_f1_macro": 0.47612925723870253, "eval_f1_micro": 0.7184986595174263, "eval_f1_weighted": 0.7080336230828008, "eval_loss": 0.6069164872169495, "eval_precision_macro": 0.5134762379528083, "eval_precision_micro": 0.7184986595174263, "eval_precision_weighted": 0.7786303288525843, "eval_recall_macro": 0.4920582814179348, "eval_recall_micro": 0.7184986595174263, "eval_recall_weighted": 0.7184986595174263, "eval_runtime": 193.5156, "eval_samples_per_second": 3.855, "eval_steps_per_second": 0.243, "step": 498 } ], "logging_steps": 18, "max_steps": 747, "num_input_tokens_seen": 0, "num_train_epochs": 3, "save_steps": 500, "stateful_callbacks": { "EarlyStoppingCallback": { "args": { "early_stopping_patience": 5, "early_stopping_threshold": 0.01 }, "attributes": { "early_stopping_patience_counter": 0 } }, "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 3.076459560786125e+17, "train_batch_size": 8, "trial_name": null, "trial_params": null }