| { | |
| "best_metric": 0.8709677419354839, | |
| "best_model_checkpoint": "ESMC_plus-finetuned-TP53_201AA/checkpoint-800", | |
| "epoch": 36.36363636363637, | |
| "eval_steps": 50, | |
| "global_step": 800, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 2.2727272727272725, | |
| "grad_norm": 3.7824575901031494, | |
| "learning_rate": 5.6999999999999996e-05, | |
| "loss": 0.6625, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 2.2727272727272725, | |
| "eval_f1_score": 0.7755102040816326, | |
| "eval_loss": 0.5577281713485718, | |
| "eval_precision": 0.6855670103092784, | |
| "eval_recall": 0.8926174496644296, | |
| "eval_roc_auc": 0.7809697045725397, | |
| "eval_runtime": 0.5094, | |
| "eval_samples_per_second": 693.021, | |
| "eval_steps_per_second": 5.89, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 4.545454545454545, | |
| "grad_norm": 10.261860847473145, | |
| "learning_rate": 5.4000000000000005e-05, | |
| "loss": 0.4772, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 4.545454545454545, | |
| "eval_f1_score": 0.7908496732026143, | |
| "eval_loss": 0.5206712484359741, | |
| "eval_precision": 0.7707006369426752, | |
| "eval_recall": 0.8120805369127517, | |
| "eval_roc_auc": 0.80794078906103, | |
| "eval_runtime": 0.5068, | |
| "eval_samples_per_second": 696.586, | |
| "eval_steps_per_second": 5.92, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 6.818181818181818, | |
| "grad_norm": 11.708925247192383, | |
| "learning_rate": 5.1e-05, | |
| "loss": 0.4047, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 6.818181818181818, | |
| "eval_f1_score": 0.7753623188405797, | |
| "eval_loss": 0.49258285760879517, | |
| "eval_precision": 0.84251968503937, | |
| "eval_recall": 0.7181208053691275, | |
| "eval_roc_auc": 0.8690334316000753, | |
| "eval_runtime": 0.5185, | |
| "eval_samples_per_second": 680.81, | |
| "eval_steps_per_second": 5.786, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 9.090909090909092, | |
| "grad_norm": 14.289894104003906, | |
| "learning_rate": 4.8e-05, | |
| "loss": 0.3611, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 9.090909090909092, | |
| "eval_f1_score": 0.7753623188405797, | |
| "eval_loss": 0.46306541562080383, | |
| "eval_precision": 0.84251968503937, | |
| "eval_recall": 0.7181208053691275, | |
| "eval_roc_auc": 0.8886658721696042, | |
| "eval_runtime": 0.487, | |
| "eval_samples_per_second": 724.818, | |
| "eval_steps_per_second": 6.16, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 11.363636363636363, | |
| "grad_norm": 2.303349494934082, | |
| "learning_rate": 4.5e-05, | |
| "loss": 0.3402, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 11.363636363636363, | |
| "eval_f1_score": 0.8417508417508418, | |
| "eval_loss": 0.39569616317749023, | |
| "eval_precision": 0.8445945945945946, | |
| "eval_recall": 0.8389261744966443, | |
| "eval_roc_auc": 0.8988270714420122, | |
| "eval_runtime": 0.4881, | |
| "eval_samples_per_second": 723.205, | |
| "eval_steps_per_second": 6.146, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 13.636363636363637, | |
| "grad_norm": 4.852221488952637, | |
| "learning_rate": 4.2e-05, | |
| "loss": 0.3342, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 13.636363636363637, | |
| "eval_f1_score": 0.8617363344051447, | |
| "eval_loss": 0.42518067359924316, | |
| "eval_precision": 0.8271604938271605, | |
| "eval_recall": 0.8993288590604027, | |
| "eval_roc_auc": 0.8945618766856928, | |
| "eval_runtime": 0.4865, | |
| "eval_samples_per_second": 725.541, | |
| "eval_steps_per_second": 6.166, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 15.909090909090908, | |
| "grad_norm": 5.0347747802734375, | |
| "learning_rate": 3.9e-05, | |
| "loss": 0.2885, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 15.909090909090908, | |
| "eval_f1_score": 0.8571428571428571, | |
| "eval_loss": 0.40493202209472656, | |
| "eval_precision": 0.8486842105263158, | |
| "eval_recall": 0.8657718120805369, | |
| "eval_roc_auc": 0.9029668192937339, | |
| "eval_runtime": 0.4878, | |
| "eval_samples_per_second": 723.589, | |
| "eval_steps_per_second": 6.149, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 18.181818181818183, | |
| "grad_norm": 1.1147843599319458, | |
| "learning_rate": 3.6e-05, | |
| "loss": 0.2654, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 18.181818181818183, | |
| "eval_f1_score": 0.8304498269896193, | |
| "eval_loss": 0.4450823664665222, | |
| "eval_precision": 0.8571428571428571, | |
| "eval_recall": 0.8053691275167785, | |
| "eval_roc_auc": 0.9034058834598256, | |
| "eval_runtime": 0.4878, | |
| "eval_samples_per_second": 723.718, | |
| "eval_steps_per_second": 6.151, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 20.454545454545453, | |
| "grad_norm": 7.004922866821289, | |
| "learning_rate": 3.3e-05, | |
| "loss": 0.2596, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 20.454545454545453, | |
| "eval_f1_score": 0.8395904436860068, | |
| "eval_loss": 0.4173749089241028, | |
| "eval_precision": 0.8541666666666666, | |
| "eval_recall": 0.825503355704698, | |
| "eval_roc_auc": 0.9070438436931568, | |
| "eval_runtime": 0.4865, | |
| "eval_samples_per_second": 725.609, | |
| "eval_steps_per_second": 6.167, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 22.727272727272727, | |
| "grad_norm": 2.1273648738861084, | |
| "learning_rate": 3e-05, | |
| "loss": 0.2366, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 22.727272727272727, | |
| "eval_f1_score": 0.8142857142857143, | |
| "eval_loss": 0.4525018334388733, | |
| "eval_precision": 0.8702290076335878, | |
| "eval_recall": 0.7651006711409396, | |
| "eval_roc_auc": 0.9105563570218905, | |
| "eval_runtime": 0.4875, | |
| "eval_samples_per_second": 724.141, | |
| "eval_steps_per_second": 6.154, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 25.0, | |
| "grad_norm": 4.955398082733154, | |
| "learning_rate": 2.7000000000000002e-05, | |
| "loss": 0.2306, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 25.0, | |
| "eval_f1_score": 0.8350877192982457, | |
| "eval_loss": 0.4550461769104004, | |
| "eval_precision": 0.875, | |
| "eval_recall": 0.7986577181208053, | |
| "eval_roc_auc": 0.9146333814213135, | |
| "eval_runtime": 0.487, | |
| "eval_samples_per_second": 724.913, | |
| "eval_steps_per_second": 6.161, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 27.272727272727273, | |
| "grad_norm": 1.5066548585891724, | |
| "learning_rate": 2.4e-05, | |
| "loss": 0.2214, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 27.272727272727273, | |
| "eval_f1_score": 0.8542372881355932, | |
| "eval_loss": 0.4234406054019928, | |
| "eval_precision": 0.863013698630137, | |
| "eval_recall": 0.8456375838926175, | |
| "eval_roc_auc": 0.9131280185661419, | |
| "eval_runtime": 0.487, | |
| "eval_samples_per_second": 724.849, | |
| "eval_steps_per_second": 6.16, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 29.545454545454547, | |
| "grad_norm": 2.199934482574463, | |
| "learning_rate": 2.1e-05, | |
| "loss": 0.2007, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 29.545454545454547, | |
| "eval_f1_score": 0.8406779661016949, | |
| "eval_loss": 0.4141705632209778, | |
| "eval_precision": 0.8493150684931506, | |
| "eval_recall": 0.8322147651006712, | |
| "eval_roc_auc": 0.9175186602270589, | |
| "eval_runtime": 0.4861, | |
| "eval_samples_per_second": 726.128, | |
| "eval_steps_per_second": 6.171, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 31.818181818181817, | |
| "grad_norm": 3.5876011848449707, | |
| "learning_rate": 1.8e-05, | |
| "loss": 0.1869, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 31.818181818181817, | |
| "eval_f1_score": 0.8533333333333334, | |
| "eval_loss": 0.42080163955688477, | |
| "eval_precision": 0.847682119205298, | |
| "eval_recall": 0.8590604026845637, | |
| "eval_roc_auc": 0.9163896380856803, | |
| "eval_runtime": 0.486, | |
| "eval_samples_per_second": 726.361, | |
| "eval_steps_per_second": 6.173, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 34.09090909090909, | |
| "grad_norm": 2.6010570526123047, | |
| "learning_rate": 1.5e-05, | |
| "loss": 0.1791, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 34.09090909090909, | |
| "eval_f1_score": 0.8655737704918033, | |
| "eval_loss": 0.405379056930542, | |
| "eval_precision": 0.8461538461538461, | |
| "eval_recall": 0.8859060402684564, | |
| "eval_roc_auc": 0.9218465784356771, | |
| "eval_runtime": 0.4871, | |
| "eval_samples_per_second": 724.696, | |
| "eval_steps_per_second": 6.159, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 36.36363636363637, | |
| "grad_norm": 4.041784286499023, | |
| "learning_rate": 1.2e-05, | |
| "loss": 0.1666, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 36.36363636363637, | |
| "eval_f1_score": 0.8709677419354839, | |
| "eval_loss": 0.4173583388328552, | |
| "eval_precision": 0.8385093167701864, | |
| "eval_recall": 0.9060402684563759, | |
| "eval_roc_auc": 0.9189612996299317, | |
| "eval_runtime": 0.4862, | |
| "eval_samples_per_second": 725.969, | |
| "eval_steps_per_second": 6.17, | |
| "step": 800 | |
| } | |
| ], | |
| "logging_steps": 50, | |
| "max_steps": 1000, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 46, | |
| "save_steps": 50, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 1.9066711283712e+16, | |
| "train_batch_size": 128, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |