|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 4.581551618814905, |
|
"global_step": 15000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 1.0070493454179256e-06, |
|
"loss": 0.7017, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 2.0140986908358513e-06, |
|
"loss": 0.6899, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 3.0211480362537765e-06, |
|
"loss": 0.5926, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 4.028197381671703e-06, |
|
"loss": 0.4967, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 5.035246727089627e-06, |
|
"loss": 0.4748, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"eval_accuracy": 0.8350723045945451, |
|
"eval_loss": 0.3857437074184418, |
|
"eval_runtime": 24.3432, |
|
"eval_samples_per_second": 224.416, |
|
"eval_steps_per_second": 7.025, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 6.042296072507553e-06, |
|
"loss": 0.4313, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 7.0493454179254795e-06, |
|
"loss": 0.4277, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 8.056394763343405e-06, |
|
"loss": 0.4182, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 9.06344410876133e-06, |
|
"loss": 0.3942, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 9.995447746634584e-06, |
|
"loss": 0.3548, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"eval_accuracy": 0.8813838550247117, |
|
"eval_loss": 0.3002248704433441, |
|
"eval_runtime": 23.5094, |
|
"eval_samples_per_second": 232.375, |
|
"eval_steps_per_second": 7.274, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 9.930415555700073e-06, |
|
"loss": 0.3842, |
|
"step": 2200 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 9.86538336476556e-06, |
|
"loss": 0.3613, |
|
"step": 2400 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 9.800351173831047e-06, |
|
"loss": 0.357, |
|
"step": 2600 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 9.735318982896535e-06, |
|
"loss": 0.3459, |
|
"step": 2800 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 9.670286791962022e-06, |
|
"loss": 0.3244, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"eval_accuracy": 0.8949295259015193, |
|
"eval_loss": 0.2588460445404053, |
|
"eval_runtime": 29.4325, |
|
"eval_samples_per_second": 185.611, |
|
"eval_steps_per_second": 5.81, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 9.60525460102751e-06, |
|
"loss": 0.3215, |
|
"step": 3200 |
|
}, |
|
{ |
|
"epoch": 1.04, |
|
"learning_rate": 9.540222410092996e-06, |
|
"loss": 0.3189, |
|
"step": 3400 |
|
}, |
|
{ |
|
"epoch": 1.1, |
|
"learning_rate": 9.475190219158484e-06, |
|
"loss": 0.3007, |
|
"step": 3600 |
|
}, |
|
{ |
|
"epoch": 1.16, |
|
"learning_rate": 9.410158028223972e-06, |
|
"loss": 0.3032, |
|
"step": 3800 |
|
}, |
|
{ |
|
"epoch": 1.22, |
|
"learning_rate": 9.34512583728946e-06, |
|
"loss": 0.2853, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 1.22, |
|
"eval_accuracy": 0.8989566172432729, |
|
"eval_loss": 0.2614916265010834, |
|
"eval_runtime": 26.1105, |
|
"eval_samples_per_second": 209.226, |
|
"eval_steps_per_second": 6.549, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 1.28, |
|
"learning_rate": 9.280093646354947e-06, |
|
"loss": 0.2773, |
|
"step": 4200 |
|
}, |
|
{ |
|
"epoch": 1.34, |
|
"learning_rate": 9.215061455420435e-06, |
|
"loss": 0.277, |
|
"step": 4400 |
|
}, |
|
{ |
|
"epoch": 1.41, |
|
"learning_rate": 9.15002926448592e-06, |
|
"loss": 0.2913, |
|
"step": 4600 |
|
}, |
|
{ |
|
"epoch": 1.47, |
|
"learning_rate": 9.084997073551408e-06, |
|
"loss": 0.2767, |
|
"step": 4800 |
|
}, |
|
{ |
|
"epoch": 1.53, |
|
"learning_rate": 9.019964882616896e-06, |
|
"loss": 0.277, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 1.53, |
|
"eval_accuracy": 0.9011532125205931, |
|
"eval_loss": 0.24724067747592926, |
|
"eval_runtime": 24.2716, |
|
"eval_samples_per_second": 225.078, |
|
"eval_steps_per_second": 7.045, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 1.59, |
|
"learning_rate": 8.954932691682384e-06, |
|
"loss": 0.2723, |
|
"step": 5200 |
|
}, |
|
{ |
|
"epoch": 1.65, |
|
"learning_rate": 8.889900500747872e-06, |
|
"loss": 0.2767, |
|
"step": 5400 |
|
}, |
|
{ |
|
"epoch": 1.71, |
|
"learning_rate": 8.824868309813358e-06, |
|
"loss": 0.2747, |
|
"step": 5600 |
|
}, |
|
{ |
|
"epoch": 1.77, |
|
"learning_rate": 8.759836118878845e-06, |
|
"loss": 0.264, |
|
"step": 5800 |
|
}, |
|
{ |
|
"epoch": 1.83, |
|
"learning_rate": 8.694803927944333e-06, |
|
"loss": 0.2604, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 1.83, |
|
"eval_accuracy": 0.9104887424492037, |
|
"eval_loss": 0.22219303250312805, |
|
"eval_runtime": 23.7933, |
|
"eval_samples_per_second": 229.602, |
|
"eval_steps_per_second": 7.187, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 1.89, |
|
"learning_rate": 8.62977173700982e-06, |
|
"loss": 0.2717, |
|
"step": 6200 |
|
}, |
|
{ |
|
"epoch": 1.95, |
|
"learning_rate": 8.564739546075308e-06, |
|
"loss": 0.2733, |
|
"step": 6400 |
|
}, |
|
{ |
|
"epoch": 2.02, |
|
"learning_rate": 8.499707355140794e-06, |
|
"loss": 0.245, |
|
"step": 6600 |
|
}, |
|
{ |
|
"epoch": 2.08, |
|
"learning_rate": 8.434675164206282e-06, |
|
"loss": 0.2251, |
|
"step": 6800 |
|
}, |
|
{ |
|
"epoch": 2.14, |
|
"learning_rate": 8.36964297327177e-06, |
|
"loss": 0.229, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 2.14, |
|
"eval_accuracy": 0.9139666849716274, |
|
"eval_loss": 0.21798129379749298, |
|
"eval_runtime": 24.6881, |
|
"eval_samples_per_second": 221.28, |
|
"eval_steps_per_second": 6.926, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 2.2, |
|
"learning_rate": 8.304610782337258e-06, |
|
"loss": 0.2268, |
|
"step": 7200 |
|
}, |
|
{ |
|
"epoch": 2.26, |
|
"learning_rate": 8.239578591402745e-06, |
|
"loss": 0.2317, |
|
"step": 7400 |
|
}, |
|
{ |
|
"epoch": 2.32, |
|
"learning_rate": 8.174546400468233e-06, |
|
"loss": 0.2306, |
|
"step": 7600 |
|
}, |
|
{ |
|
"epoch": 2.38, |
|
"learning_rate": 8.109514209533719e-06, |
|
"loss": 0.236, |
|
"step": 7800 |
|
}, |
|
{ |
|
"epoch": 2.44, |
|
"learning_rate": 8.044482018599207e-06, |
|
"loss": 0.2257, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 2.44, |
|
"eval_accuracy": 0.918542925132711, |
|
"eval_loss": 0.2288324385881424, |
|
"eval_runtime": 24.6216, |
|
"eval_samples_per_second": 221.878, |
|
"eval_steps_per_second": 6.945, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 2.5, |
|
"learning_rate": 7.979449827664695e-06, |
|
"loss": 0.2265, |
|
"step": 8200 |
|
}, |
|
{ |
|
"epoch": 2.57, |
|
"learning_rate": 7.914417636730182e-06, |
|
"loss": 0.2137, |
|
"step": 8400 |
|
}, |
|
{ |
|
"epoch": 2.63, |
|
"learning_rate": 7.84938544579567e-06, |
|
"loss": 0.2331, |
|
"step": 8600 |
|
}, |
|
{ |
|
"epoch": 2.69, |
|
"learning_rate": 7.784353254861156e-06, |
|
"loss": 0.2212, |
|
"step": 8800 |
|
}, |
|
{ |
|
"epoch": 2.75, |
|
"learning_rate": 7.719321063926645e-06, |
|
"loss": 0.2292, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 2.75, |
|
"eval_accuracy": 0.919092073952041, |
|
"eval_loss": 0.2137669026851654, |
|
"eval_runtime": 25.2669, |
|
"eval_samples_per_second": 216.212, |
|
"eval_steps_per_second": 6.768, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 2.81, |
|
"learning_rate": 7.654288872992131e-06, |
|
"loss": 0.2285, |
|
"step": 9200 |
|
}, |
|
{ |
|
"epoch": 2.87, |
|
"learning_rate": 7.58925668205762e-06, |
|
"loss": 0.2292, |
|
"step": 9400 |
|
}, |
|
{ |
|
"epoch": 2.93, |
|
"learning_rate": 7.524224491123107e-06, |
|
"loss": 0.2153, |
|
"step": 9600 |
|
}, |
|
{ |
|
"epoch": 2.99, |
|
"learning_rate": 7.459192300188594e-06, |
|
"loss": 0.2208, |
|
"step": 9800 |
|
}, |
|
{ |
|
"epoch": 3.05, |
|
"learning_rate": 7.394160109254081e-06, |
|
"loss": 0.1947, |
|
"step": 10000 |
|
}, |
|
{ |
|
"epoch": 3.05, |
|
"eval_accuracy": 0.914332784184514, |
|
"eval_loss": 0.23720914125442505, |
|
"eval_runtime": 24.8628, |
|
"eval_samples_per_second": 219.726, |
|
"eval_steps_per_second": 6.878, |
|
"step": 10000 |
|
}, |
|
{ |
|
"epoch": 3.12, |
|
"learning_rate": 7.329127918319568e-06, |
|
"loss": 0.1857, |
|
"step": 10200 |
|
}, |
|
{ |
|
"epoch": 3.18, |
|
"learning_rate": 7.264095727385057e-06, |
|
"loss": 0.202, |
|
"step": 10400 |
|
}, |
|
{ |
|
"epoch": 3.24, |
|
"learning_rate": 7.199063536450544e-06, |
|
"loss": 0.1804, |
|
"step": 10600 |
|
}, |
|
{ |
|
"epoch": 3.3, |
|
"learning_rate": 7.134031345516031e-06, |
|
"loss": 0.1893, |
|
"step": 10800 |
|
}, |
|
{ |
|
"epoch": 3.36, |
|
"learning_rate": 7.068999154581518e-06, |
|
"loss": 0.1946, |
|
"step": 11000 |
|
}, |
|
{ |
|
"epoch": 3.36, |
|
"eval_accuracy": 0.924400512538898, |
|
"eval_loss": 0.21092592179775238, |
|
"eval_runtime": 23.8519, |
|
"eval_samples_per_second": 229.038, |
|
"eval_steps_per_second": 7.169, |
|
"step": 11000 |
|
}, |
|
{ |
|
"epoch": 3.42, |
|
"learning_rate": 7.003966963647005e-06, |
|
"loss": 0.1889, |
|
"step": 11200 |
|
}, |
|
{ |
|
"epoch": 3.48, |
|
"learning_rate": 6.938934772712494e-06, |
|
"loss": 0.1834, |
|
"step": 11400 |
|
}, |
|
{ |
|
"epoch": 3.54, |
|
"learning_rate": 6.873902581777981e-06, |
|
"loss": 0.1887, |
|
"step": 11600 |
|
}, |
|
{ |
|
"epoch": 3.6, |
|
"learning_rate": 6.808870390843468e-06, |
|
"loss": 0.1865, |
|
"step": 11800 |
|
}, |
|
{ |
|
"epoch": 3.67, |
|
"learning_rate": 6.743838199908955e-06, |
|
"loss": 0.1975, |
|
"step": 12000 |
|
}, |
|
{ |
|
"epoch": 3.67, |
|
"eval_accuracy": 0.9178107267069375, |
|
"eval_loss": 0.21930640935897827, |
|
"eval_runtime": 24.5268, |
|
"eval_samples_per_second": 222.736, |
|
"eval_steps_per_second": 6.972, |
|
"step": 12000 |
|
}, |
|
{ |
|
"epoch": 3.73, |
|
"learning_rate": 6.678806008974443e-06, |
|
"loss": 0.1905, |
|
"step": 12200 |
|
}, |
|
{ |
|
"epoch": 3.79, |
|
"learning_rate": 6.613773818039931e-06, |
|
"loss": 0.1925, |
|
"step": 12400 |
|
}, |
|
{ |
|
"epoch": 3.85, |
|
"learning_rate": 6.5487416271054175e-06, |
|
"loss": 0.1962, |
|
"step": 12600 |
|
}, |
|
{ |
|
"epoch": 3.91, |
|
"learning_rate": 6.483709436170905e-06, |
|
"loss": 0.1821, |
|
"step": 12800 |
|
}, |
|
{ |
|
"epoch": 3.97, |
|
"learning_rate": 6.418677245236392e-06, |
|
"loss": 0.1832, |
|
"step": 13000 |
|
}, |
|
{ |
|
"epoch": 3.97, |
|
"eval_accuracy": 0.9207395204100312, |
|
"eval_loss": 0.22072818875312805, |
|
"eval_runtime": 25.907, |
|
"eval_samples_per_second": 210.87, |
|
"eval_steps_per_second": 6.601, |
|
"step": 13000 |
|
}, |
|
{ |
|
"epoch": 4.03, |
|
"learning_rate": 6.35364505430188e-06, |
|
"loss": 0.1752, |
|
"step": 13200 |
|
}, |
|
{ |
|
"epoch": 4.09, |
|
"learning_rate": 6.2886128633673675e-06, |
|
"loss": 0.1565, |
|
"step": 13400 |
|
}, |
|
{ |
|
"epoch": 4.15, |
|
"learning_rate": 6.223580672432855e-06, |
|
"loss": 0.1647, |
|
"step": 13600 |
|
}, |
|
{ |
|
"epoch": 4.22, |
|
"learning_rate": 6.158548481498342e-06, |
|
"loss": 0.1488, |
|
"step": 13800 |
|
}, |
|
{ |
|
"epoch": 4.28, |
|
"learning_rate": 6.09351629056383e-06, |
|
"loss": 0.1639, |
|
"step": 14000 |
|
}, |
|
{ |
|
"epoch": 4.28, |
|
"eval_accuracy": 0.9207395204100312, |
|
"eval_loss": 0.2469523698091507, |
|
"eval_runtime": 24.3069, |
|
"eval_samples_per_second": 224.751, |
|
"eval_steps_per_second": 7.035, |
|
"step": 14000 |
|
}, |
|
{ |
|
"epoch": 4.34, |
|
"learning_rate": 6.028484099629317e-06, |
|
"loss": 0.1652, |
|
"step": 14200 |
|
}, |
|
{ |
|
"epoch": 4.4, |
|
"learning_rate": 5.9634519086948035e-06, |
|
"loss": 0.1675, |
|
"step": 14400 |
|
}, |
|
{ |
|
"epoch": 4.46, |
|
"learning_rate": 5.898419717760292e-06, |
|
"loss": 0.1517, |
|
"step": 14600 |
|
}, |
|
{ |
|
"epoch": 4.52, |
|
"learning_rate": 5.833387526825779e-06, |
|
"loss": 0.1543, |
|
"step": 14800 |
|
}, |
|
{ |
|
"epoch": 4.58, |
|
"learning_rate": 5.768355335891267e-06, |
|
"loss": 0.1717, |
|
"step": 15000 |
|
}, |
|
{ |
|
"epoch": 4.58, |
|
"eval_accuracy": 0.9231191652937946, |
|
"eval_loss": 0.23586037755012512, |
|
"eval_runtime": 24.1493, |
|
"eval_samples_per_second": 226.218, |
|
"eval_steps_per_second": 7.081, |
|
"step": 15000 |
|
} |
|
], |
|
"max_steps": 32740, |
|
"num_train_epochs": 10, |
|
"total_flos": 3.6794803116288e+16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|