| { | |
| "best_metric": 0.36799120903015137, | |
| "best_model_checkpoint": "/tmp/model/checkpoint-4000", | |
| "epoch": 2.0, | |
| "eval_steps": 500, | |
| "global_step": 4000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.05, | |
| "learning_rate": 5e-06, | |
| "loss": 0.581, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "learning_rate": 1e-05, | |
| "loss": 0.5379, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "learning_rate": 1.5e-05, | |
| "loss": 0.4945, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "learning_rate": 2e-05, | |
| "loss": 0.4066, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "learning_rate": 2.5e-05, | |
| "loss": 0.4182, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.3, | |
| "learning_rate": 3e-05, | |
| "loss": 0.4407, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "learning_rate": 3.5e-05, | |
| "loss": 0.418, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.4, | |
| "learning_rate": 4e-05, | |
| "loss": 0.4258, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 0.45, | |
| "learning_rate": 4.5e-05, | |
| "loss": 0.3885, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 0.5, | |
| "learning_rate": 5e-05, | |
| "loss": 0.4283, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "learning_rate": 4.9444444444444446e-05, | |
| "loss": 0.4148, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 0.6, | |
| "learning_rate": 4.888888888888889e-05, | |
| "loss": 0.3951, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 0.65, | |
| "learning_rate": 4.8333333333333334e-05, | |
| "loss": 0.4255, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "learning_rate": 4.7777777777777784e-05, | |
| "loss": 0.3842, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "learning_rate": 4.722222222222222e-05, | |
| "loss": 0.4461, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "learning_rate": 4.666666666666667e-05, | |
| "loss": 0.3789, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 0.85, | |
| "learning_rate": 4.6111111111111115e-05, | |
| "loss": 0.3893, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 0.9, | |
| "learning_rate": 4.555555555555556e-05, | |
| "loss": 0.3833, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 0.95, | |
| "learning_rate": 4.5e-05, | |
| "loss": 0.3817, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "learning_rate": 4.4444444444444447e-05, | |
| "loss": 0.3654, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.84475, | |
| "eval_auc": 0.8835736098526796, | |
| "eval_f1": 0.689344672336168, | |
| "eval_loss": 0.40313366055488586, | |
| "eval_precision": 0.6828543111992071, | |
| "eval_recall": 0.695959595959596, | |
| "eval_runtime": 738.7835, | |
| "eval_samples_per_second": 5.414, | |
| "eval_steps_per_second": 0.338, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 1.05, | |
| "learning_rate": 4.388888888888889e-05, | |
| "loss": 0.3236, | |
| "step": 2100 | |
| }, | |
| { | |
| "epoch": 1.1, | |
| "learning_rate": 4.3333333333333334e-05, | |
| "loss": 0.3882, | |
| "step": 2200 | |
| }, | |
| { | |
| "epoch": 1.15, | |
| "learning_rate": 4.277777777777778e-05, | |
| "loss": 0.3413, | |
| "step": 2300 | |
| }, | |
| { | |
| "epoch": 1.2, | |
| "learning_rate": 4.222222222222222e-05, | |
| "loss": 0.303, | |
| "step": 2400 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "learning_rate": 4.166666666666667e-05, | |
| "loss": 0.352, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 1.3, | |
| "learning_rate": 4.111111111111111e-05, | |
| "loss": 0.323, | |
| "step": 2600 | |
| }, | |
| { | |
| "epoch": 1.35, | |
| "learning_rate": 4.055555555555556e-05, | |
| "loss": 0.3494, | |
| "step": 2700 | |
| }, | |
| { | |
| "epoch": 1.4, | |
| "learning_rate": 4e-05, | |
| "loss": 0.3125, | |
| "step": 2800 | |
| }, | |
| { | |
| "epoch": 1.45, | |
| "learning_rate": 3.944444444444445e-05, | |
| "loss": 0.314, | |
| "step": 2900 | |
| }, | |
| { | |
| "epoch": 1.5, | |
| "learning_rate": 3.888888888888889e-05, | |
| "loss": 0.394, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 1.55, | |
| "learning_rate": 3.8333333333333334e-05, | |
| "loss": 0.3142, | |
| "step": 3100 | |
| }, | |
| { | |
| "epoch": 1.6, | |
| "learning_rate": 3.777777777777778e-05, | |
| "loss": 0.3179, | |
| "step": 3200 | |
| }, | |
| { | |
| "epoch": 1.65, | |
| "learning_rate": 3.722222222222222e-05, | |
| "loss": 0.2756, | |
| "step": 3300 | |
| }, | |
| { | |
| "epoch": 1.7, | |
| "learning_rate": 3.6666666666666666e-05, | |
| "loss": 0.3178, | |
| "step": 3400 | |
| }, | |
| { | |
| "epoch": 1.75, | |
| "learning_rate": 3.611111111111111e-05, | |
| "loss": 0.3101, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 1.8, | |
| "learning_rate": 3.555555555555556e-05, | |
| "loss": 0.3615, | |
| "step": 3600 | |
| }, | |
| { | |
| "epoch": 1.85, | |
| "learning_rate": 3.5e-05, | |
| "loss": 0.3084, | |
| "step": 3700 | |
| }, | |
| { | |
| "epoch": 1.9, | |
| "learning_rate": 3.444444444444445e-05, | |
| "loss": 0.308, | |
| "step": 3800 | |
| }, | |
| { | |
| "epoch": 1.95, | |
| "learning_rate": 3.388888888888889e-05, | |
| "loss": 0.3686, | |
| "step": 3900 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "learning_rate": 3.3333333333333335e-05, | |
| "loss": 0.3642, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.85775, | |
| "eval_auc": 0.8958072418537535, | |
| "eval_f1": 0.7044155844155844, | |
| "eval_loss": 0.36799120903015137, | |
| "eval_precision": 0.7251336898395722, | |
| "eval_recall": 0.6848484848484848, | |
| "eval_runtime": 782.6408, | |
| "eval_samples_per_second": 5.111, | |
| "eval_steps_per_second": 0.319, | |
| "step": 4000 | |
| } | |
| ], | |
| "logging_steps": 100, | |
| "max_steps": 10000, | |
| "num_train_epochs": 5, | |
| "save_steps": 500, | |
| "total_flos": 2104888442880000.0, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |