{ "best_global_step": 5000, "best_metric": 0.11183962970972061, "best_model_checkpoint": "myanmar_text_segmentation_model/checkpoint-5000", "epoch": 0.4594321418726454, "eval_steps": 1000, "global_step": 5000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.04594321418726454, "grad_norm": 0.577109158039093, "learning_rate": 1.9816594688964442e-05, "loss": 0.1144, "step": 500 }, { "epoch": 0.09188642837452908, "grad_norm": 0.8819814324378967, "learning_rate": 1.9632821832215383e-05, "loss": 0.1133, "step": 1000 }, { "epoch": 0.09188642837452908, "eval_accuracy": 0.9564004775738034, "eval_f1": 0.7581789101913106, "eval_loss": 0.11448723077774048, "eval_precision": 0.7703745996379334, "eval_recall": 0.7463633395484631, "eval_runtime": 465.4607, "eval_samples_per_second": 61.528, "eval_steps_per_second": 1.231, "step": 1000 }, { "epoch": 0.1378296425617936, "grad_norm": 0.5137984156608582, "learning_rate": 1.9449048975466325e-05, "loss": 0.1119, "step": 1500 }, { "epoch": 0.18377285674905816, "grad_norm": 0.6753482222557068, "learning_rate": 1.9265276118717266e-05, "loss": 0.1115, "step": 2000 }, { "epoch": 0.18377285674905816, "eval_accuracy": 0.9565513036953667, "eval_f1": 0.7631813253052621, "eval_loss": 0.11509840190410614, "eval_precision": 0.7619568901058112, "eval_recall": 0.7644097020785157, "eval_runtime": 471.4797, "eval_samples_per_second": 60.743, "eval_steps_per_second": 1.215, "step": 2000 }, { "epoch": 0.2297160709363227, "grad_norm": 0.46862444281578064, "learning_rate": 1.9081503261968207e-05, "loss": 0.1083, "step": 2500 }, { "epoch": 0.2756592851235872, "grad_norm": 0.6514066457748413, "learning_rate": 1.889773040521915e-05, "loss": 0.1085, "step": 3000 }, { "epoch": 0.2756592851235872, "eval_accuracy": 0.955793370748351, "eval_f1": 0.7617613129080717, "eval_loss": 0.11627302318811417, "eval_precision": 0.7514122514122514, "eval_recall": 0.7723994268768838, "eval_runtime": 464.541, "eval_samples_per_second": 61.65, "eval_steps_per_second": 1.233, "step": 3000 }, { "epoch": 0.3216024993108518, "grad_norm": 0.5837205052375793, "learning_rate": 1.8713957548470093e-05, "loss": 0.1068, "step": 3500 }, { "epoch": 0.3675457134981163, "grad_norm": 0.590044379234314, "learning_rate": 1.8530184691721034e-05, "loss": 0.1076, "step": 4000 }, { "epoch": 0.3675457134981163, "eval_accuracy": 0.9567566300121167, "eval_f1": 0.7569129075628852, "eval_loss": 0.11681018769741058, "eval_precision": 0.7790560842692136, "eval_recall": 0.7359936967250493, "eval_runtime": 464.1614, "eval_samples_per_second": 61.701, "eval_steps_per_second": 1.234, "step": 4000 }, { "epoch": 0.41348892768538087, "grad_norm": 0.43107616901397705, "learning_rate": 1.8346411834971975e-05, "loss": 0.1071, "step": 4500 }, { "epoch": 0.4594321418726454, "grad_norm": 0.6132990121841431, "learning_rate": 1.8162638978222916e-05, "loss": 0.1056, "step": 5000 }, { "epoch": 0.4594321418726454, "eval_accuracy": 0.9576032842071108, "eval_f1": 0.7661190672004459, "eval_loss": 0.11183962970972061, "eval_precision": 0.7719609680194174, "eval_recall": 0.7603649207638403, "eval_runtime": 473.1438, "eval_samples_per_second": 60.529, "eval_steps_per_second": 1.211, "step": 5000 } ], "logging_steps": 500, "max_steps": 54415, "num_input_tokens_seen": 0, "num_train_epochs": 5, "save_steps": 1000, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 5.82981898783398e+16, "train_batch_size": 50, "trial_name": null, "trial_params": null }