| { | |
| "best_metric": 0.92816, | |
| "best_model_checkpoint": "/nvme1/yujiepan/token-dropping-logs/logs-final/train-imdb-baseline/0703-lr5e-05_warm0.1_epoch6/checkpoint-4692", | |
| "epoch": 6.0, | |
| "global_step": 4692, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.5, | |
| "eval_accuracy": 0.91216, | |
| "eval_loss": 0.4330781102180481, | |
| "eval_runtime": 65.4808, | |
| "eval_samples_per_second": 381.791, | |
| "eval_steps_per_second": 5.971, | |
| "step": 391 | |
| }, | |
| { | |
| "epoch": 0.64, | |
| "learning_rate": 4.9680246328754146e-05, | |
| "loss": 0.0385, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.91504, | |
| "eval_loss": 0.3079942464828491, | |
| "eval_runtime": 64.589, | |
| "eval_samples_per_second": 387.063, | |
| "eval_steps_per_second": 6.054, | |
| "step": 782 | |
| }, | |
| { | |
| "epoch": 1.28, | |
| "learning_rate": 4.37588820464235e-05, | |
| "loss": 0.0654, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.5, | |
| "eval_accuracy": 0.91636, | |
| "eval_loss": 0.3285704255104065, | |
| "eval_runtime": 64.8695, | |
| "eval_samples_per_second": 385.389, | |
| "eval_steps_per_second": 6.027, | |
| "step": 1173 | |
| }, | |
| { | |
| "epoch": 1.92, | |
| "learning_rate": 3.783751776409285e-05, | |
| "loss": 0.0636, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.9184, | |
| "eval_loss": 0.42962637543678284, | |
| "eval_runtime": 130.3196, | |
| "eval_samples_per_second": 191.836, | |
| "eval_steps_per_second": 3.0, | |
| "step": 1564 | |
| }, | |
| { | |
| "epoch": 2.5, | |
| "eval_accuracy": 0.91704, | |
| "eval_loss": 0.5365941524505615, | |
| "eval_runtime": 133.611, | |
| "eval_samples_per_second": 187.11, | |
| "eval_steps_per_second": 2.926, | |
| "step": 1955 | |
| }, | |
| { | |
| "epoch": 2.56, | |
| "learning_rate": 3.19161534817622e-05, | |
| "loss": 0.0344, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_accuracy": 0.91724, | |
| "eval_loss": 0.45805788040161133, | |
| "eval_runtime": 130.2627, | |
| "eval_samples_per_second": 191.92, | |
| "eval_steps_per_second": 3.002, | |
| "step": 2346 | |
| }, | |
| { | |
| "epoch": 3.2, | |
| "learning_rate": 2.5994789199431553e-05, | |
| "loss": 0.0285, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 3.5, | |
| "eval_accuracy": 0.91644, | |
| "eval_loss": 0.5552997589111328, | |
| "eval_runtime": 130.5012, | |
| "eval_samples_per_second": 191.569, | |
| "eval_steps_per_second": 2.996, | |
| "step": 2737 | |
| }, | |
| { | |
| "epoch": 3.84, | |
| "learning_rate": 2.008526764566556e-05, | |
| "loss": 0.0163, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.91936, | |
| "eval_loss": 0.5054952502250671, | |
| "eval_runtime": 132.4835, | |
| "eval_samples_per_second": 188.703, | |
| "eval_steps_per_second": 2.951, | |
| "step": 3128 | |
| }, | |
| { | |
| "epoch": 4.48, | |
| "learning_rate": 1.4163903363334913e-05, | |
| "loss": 0.0104, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 4.5, | |
| "eval_accuracy": 0.92644, | |
| "eval_loss": 0.5432029366493225, | |
| "eval_runtime": 130.5202, | |
| "eval_samples_per_second": 191.541, | |
| "eval_steps_per_second": 2.996, | |
| "step": 3519 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "eval_accuracy": 0.92684, | |
| "eval_loss": 0.5357362627983093, | |
| "eval_runtime": 130.5356, | |
| "eval_samples_per_second": 191.519, | |
| "eval_steps_per_second": 2.995, | |
| "step": 3910 | |
| }, | |
| { | |
| "epoch": 5.12, | |
| "learning_rate": 8.242539081004264e-06, | |
| "loss": 0.0034, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 5.5, | |
| "eval_accuracy": 0.92488, | |
| "eval_loss": 0.5456807613372803, | |
| "eval_runtime": 132.2657, | |
| "eval_samples_per_second": 189.014, | |
| "eval_steps_per_second": 2.956, | |
| "step": 4301 | |
| }, | |
| { | |
| "epoch": 5.75, | |
| "learning_rate": 2.3211747986736146e-06, | |
| "loss": 0.0046, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "eval_accuracy": 0.92816, | |
| "eval_loss": 0.5325844883918762, | |
| "eval_runtime": 130.2637, | |
| "eval_samples_per_second": 191.918, | |
| "eval_steps_per_second": 3.002, | |
| "step": 4692 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "step": 4692, | |
| "total_flos": 2.959973061694464e+16, | |
| "train_loss": 0.02827812901810002, | |
| "train_runtime": 3645.2659, | |
| "train_samples_per_second": 41.149, | |
| "train_steps_per_second": 1.287 | |
| } | |
| ], | |
| "max_steps": 4692, | |
| "num_train_epochs": 6, | |
| "total_flos": 2.959973061694464e+16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |