| { | |
| "best_metric": 0.4918627450980392, | |
| "best_model_checkpoint": "mecha-tagger/checkpoint-968", | |
| "epoch": 8.0, | |
| "global_step": 968, | |
| "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 | |
| } | |
| ], | |
| "max_steps": 6050, | |
| "num_train_epochs": 50, | |
| "total_flos": 4.235225926677037e+19, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |