| { | |
| "best_metric": 0.5472163817533595, | |
| "best_model_checkpoint": "/tmp/mecha-tagger/checkpoint-1573", | |
| "epoch": 13.0, | |
| "global_step": 1573, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.83, | |
| "learning_rate": 2e-05, | |
| "loss": 0.437, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.06247574699262708, | |
| "eval_f1": 0.431281999799217, | |
| "eval_loss": 0.32573798298835754, | |
| "eval_roc_auc": 0.6429553592437186, | |
| "eval_runtime": 50.2381, | |
| "eval_samples_per_second": 51.296, | |
| "eval_steps_per_second": 0.537, | |
| "step": 121 | |
| }, | |
| { | |
| "epoch": 1.65, | |
| "learning_rate": 2e-05, | |
| "loss": 0.3211, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.055490880869227786, | |
| "eval_f1": 0.45432441820076414, | |
| "eval_loss": 0.3099991977214813, | |
| "eval_roc_auc": 0.6526660416613327, | |
| "eval_runtime": 50.3523, | |
| "eval_samples_per_second": 51.179, | |
| "eval_steps_per_second": 0.536, | |
| "step": 242 | |
| }, | |
| { | |
| "epoch": 2.48, | |
| "learning_rate": 2e-05, | |
| "loss": 0.309, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_accuracy": 0.05820721769499418, | |
| "eval_f1": 0.4483176720218248, | |
| "eval_loss": 0.30198442935943604, | |
| "eval_roc_auc": 0.6485457459740307, | |
| "eval_runtime": 50.3568, | |
| "eval_samples_per_second": 51.175, | |
| "eval_steps_per_second": 0.536, | |
| "step": 363 | |
| }, | |
| { | |
| "epoch": 3.31, | |
| "learning_rate": 2e-05, | |
| "loss": 0.303, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.05743112145906092, | |
| "eval_f1": 0.4746113989637306, | |
| "eval_loss": 0.295987069606781, | |
| "eval_roc_auc": 0.6608536486091969, | |
| "eval_runtime": 50.3091, | |
| "eval_samples_per_second": 51.223, | |
| "eval_steps_per_second": 0.537, | |
| "step": 484 | |
| }, | |
| { | |
| "epoch": 4.13, | |
| "learning_rate": 2e-05, | |
| "loss": 0.2965, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 4.96, | |
| "learning_rate": 2e-05, | |
| "loss": 0.294, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "eval_accuracy": 0.0682964687621265, | |
| "eval_f1": 0.4646505308710311, | |
| "eval_loss": 0.28789642453193665, | |
| "eval_roc_auc": 0.6549506720523524, | |
| "eval_runtime": 50.2808, | |
| "eval_samples_per_second": 51.252, | |
| "eval_steps_per_second": 0.537, | |
| "step": 605 | |
| }, | |
| { | |
| "epoch": 5.79, | |
| "learning_rate": 2e-05, | |
| "loss": 0.2879, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "eval_accuracy": 0.06674427629025999, | |
| "eval_f1": 0.4727673207849419, | |
| "eval_loss": 0.2825814187526703, | |
| "eval_roc_auc": 0.658515066442945, | |
| "eval_runtime": 50.324, | |
| "eval_samples_per_second": 51.208, | |
| "eval_steps_per_second": 0.537, | |
| "step": 726 | |
| }, | |
| { | |
| "epoch": 6.61, | |
| "learning_rate": 2e-05, | |
| "loss": 0.2846, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 7.0, | |
| "eval_accuracy": 0.06984866123399301, | |
| "eval_f1": 0.48976385684127843, | |
| "eval_loss": 0.27861288189888, | |
| "eval_roc_auc": 0.6671462002192768, | |
| "eval_runtime": 50.1175, | |
| "eval_samples_per_second": 51.419, | |
| "eval_steps_per_second": 0.539, | |
| "step": 847 | |
| }, | |
| { | |
| "epoch": 7.44, | |
| "learning_rate": 2e-05, | |
| "loss": 0.2781, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 8.0, | |
| "eval_accuracy": 0.06713232440822663, | |
| "eval_f1": 0.4918627450980392, | |
| "eval_loss": 0.27516159415245056, | |
| "eval_roc_auc": 0.6681229472140713, | |
| "eval_runtime": 50.4646, | |
| "eval_samples_per_second": 51.066, | |
| "eval_steps_per_second": 0.535, | |
| "step": 968 | |
| }, | |
| { | |
| "epoch": 8.26, | |
| "learning_rate": 2e-05, | |
| "loss": 0.2626, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 9.0, | |
| "eval_accuracy": 0.06635622817229336, | |
| "eval_f1": 0.5017455391776571, | |
| "eval_loss": 0.26821279525756836, | |
| "eval_roc_auc": 0.6732412875217908, | |
| "eval_runtime": 51.4118, | |
| "eval_samples_per_second": 50.125, | |
| "eval_steps_per_second": 0.525, | |
| "step": 1089 | |
| }, | |
| { | |
| "epoch": 9.09, | |
| "learning_rate": 2e-05, | |
| "loss": 0.2625, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 9.92, | |
| "learning_rate": 2e-05, | |
| "loss": 0.2528, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_accuracy": 0.06674427629025999, | |
| "eval_f1": 0.5120230381569474, | |
| "eval_loss": 0.26574626564979553, | |
| "eval_roc_auc": 0.6785095938529904, | |
| "eval_runtime": 51.4008, | |
| "eval_samples_per_second": 50.135, | |
| "eval_steps_per_second": 0.525, | |
| "step": 1210 | |
| }, | |
| { | |
| "epoch": 10.74, | |
| "learning_rate": 2e-05, | |
| "loss": 0.2451, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 11.0, | |
| "eval_accuracy": 0.0682964687621265, | |
| "eval_f1": 0.5220403321470937, | |
| "eval_loss": 0.2608773112297058, | |
| "eval_roc_auc": 0.6839487133042365, | |
| "eval_runtime": 51.3505, | |
| "eval_samples_per_second": 50.184, | |
| "eval_steps_per_second": 0.526, | |
| "step": 1331 | |
| }, | |
| { | |
| "epoch": 11.57, | |
| "learning_rate": 2e-05, | |
| "loss": 0.2362, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 12.0, | |
| "eval_accuracy": 0.07140085370585952, | |
| "eval_f1": 0.5350377904264253, | |
| "eval_loss": 0.2600255012512207, | |
| "eval_roc_auc": 0.6914645249590691, | |
| "eval_runtime": 51.3565, | |
| "eval_samples_per_second": 50.179, | |
| "eval_steps_per_second": 0.526, | |
| "step": 1452 | |
| }, | |
| { | |
| "epoch": 12.4, | |
| "learning_rate": 2e-05, | |
| "loss": 0.2281, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 13.0, | |
| "eval_accuracy": 0.07023670935195964, | |
| "eval_f1": 0.5472163817533595, | |
| "eval_loss": 0.2586032450199127, | |
| "eval_roc_auc": 0.6994249977444085, | |
| "eval_runtime": 51.2515, | |
| "eval_samples_per_second": 50.281, | |
| "eval_steps_per_second": 0.527, | |
| "step": 1573 | |
| } | |
| ], | |
| "max_steps": 6050, | |
| "num_train_epochs": 50, | |
| "total_flos": 6.882242130850185e+19, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |