|
{ |
|
"best_metric": 5.520716190338135, |
|
"best_model_checkpoint": "/temp/t5_base-qg-aas/checkpoint-4040", |
|
"epoch": 5.0, |
|
"eval_steps": 500, |
|
"global_step": 4040, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 0.004876237623762376, |
|
"loss": 8.1261, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 0.004752475247524752, |
|
"loss": 7.6878, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 0.004628712871287129, |
|
"loss": 7.3657, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 0.0045049504950495055, |
|
"loss": 7.0658, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 0.004381188118811882, |
|
"loss": 6.8105, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 0.004257425742574258, |
|
"loss": 6.6166, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"learning_rate": 0.004133663366336634, |
|
"loss": 6.4796, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 0.00400990099009901, |
|
"loss": 6.307, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_loss": 7.262293338775635, |
|
"eval_runtime": 213.2733, |
|
"eval_samples_per_second": 29.708, |
|
"eval_steps_per_second": 7.427, |
|
"step": 808 |
|
}, |
|
{ |
|
"epoch": 1.11, |
|
"learning_rate": 0.0038861386138613866, |
|
"loss": 6.1902, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 1.24, |
|
"learning_rate": 0.0037623762376237627, |
|
"loss": 6.0651, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 1.36, |
|
"learning_rate": 0.0036386138613861384, |
|
"loss": 5.9582, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 1.49, |
|
"learning_rate": 0.0035148514851485145, |
|
"loss": 5.9095, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 1.61, |
|
"learning_rate": 0.0033910891089108915, |
|
"loss": 5.7845, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 1.73, |
|
"learning_rate": 0.0032673267326732676, |
|
"loss": 5.6799, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 1.86, |
|
"learning_rate": 0.0031435643564356438, |
|
"loss": 5.6105, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 1.98, |
|
"learning_rate": 0.00301980198019802, |
|
"loss": 5.5213, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_loss": 6.364130973815918, |
|
"eval_runtime": 212.9865, |
|
"eval_samples_per_second": 29.748, |
|
"eval_steps_per_second": 7.437, |
|
"step": 1616 |
|
}, |
|
{ |
|
"epoch": 2.1, |
|
"learning_rate": 0.002896039603960396, |
|
"loss": 5.4604, |
|
"step": 1700 |
|
}, |
|
{ |
|
"epoch": 2.23, |
|
"learning_rate": 0.0027722772277227726, |
|
"loss": 5.3793, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 2.35, |
|
"learning_rate": 0.0026485148514851487, |
|
"loss": 5.3105, |
|
"step": 1900 |
|
}, |
|
{ |
|
"epoch": 2.48, |
|
"learning_rate": 0.002524752475247525, |
|
"loss": 5.2991, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 2.6, |
|
"learning_rate": 0.002400990099009901, |
|
"loss": 5.2497, |
|
"step": 2100 |
|
}, |
|
{ |
|
"epoch": 2.72, |
|
"learning_rate": 0.0022772277227722775, |
|
"loss": 5.1817, |
|
"step": 2200 |
|
}, |
|
{ |
|
"epoch": 2.85, |
|
"learning_rate": 0.0021534653465346536, |
|
"loss": 5.1606, |
|
"step": 2300 |
|
}, |
|
{ |
|
"epoch": 2.97, |
|
"learning_rate": 0.0020297029702970298, |
|
"loss": 5.1108, |
|
"step": 2400 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"eval_loss": 5.862464904785156, |
|
"eval_runtime": 213.0715, |
|
"eval_samples_per_second": 29.736, |
|
"eval_steps_per_second": 7.434, |
|
"step": 2424 |
|
}, |
|
{ |
|
"epoch": 3.09, |
|
"learning_rate": 0.0019059405940594061, |
|
"loss": 5.0772, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 3.22, |
|
"learning_rate": 0.001782178217821782, |
|
"loss": 5.0112, |
|
"step": 2600 |
|
}, |
|
{ |
|
"epoch": 3.34, |
|
"learning_rate": 0.0016584158415841586, |
|
"loss": 4.9832, |
|
"step": 2700 |
|
}, |
|
{ |
|
"epoch": 3.47, |
|
"learning_rate": 0.0015346534653465347, |
|
"loss": 4.9874, |
|
"step": 2800 |
|
}, |
|
{ |
|
"epoch": 3.59, |
|
"learning_rate": 0.0014108910891089108, |
|
"loss": 4.955, |
|
"step": 2900 |
|
}, |
|
{ |
|
"epoch": 3.71, |
|
"learning_rate": 0.0012871287128712872, |
|
"loss": 4.9129, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 3.84, |
|
"learning_rate": 0.0011633663366336635, |
|
"loss": 4.8824, |
|
"step": 3100 |
|
}, |
|
{ |
|
"epoch": 3.96, |
|
"learning_rate": 0.0010396039603960397, |
|
"loss": 4.8497, |
|
"step": 3200 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"eval_loss": 5.601846218109131, |
|
"eval_runtime": 213.6741, |
|
"eval_samples_per_second": 29.653, |
|
"eval_steps_per_second": 7.413, |
|
"step": 3232 |
|
}, |
|
{ |
|
"epoch": 4.08, |
|
"learning_rate": 0.0009158415841584158, |
|
"loss": 4.8512, |
|
"step": 3300 |
|
}, |
|
{ |
|
"epoch": 4.21, |
|
"learning_rate": 0.0007920792079207921, |
|
"loss": 4.8049, |
|
"step": 3400 |
|
}, |
|
{ |
|
"epoch": 4.33, |
|
"learning_rate": 0.0006683168316831684, |
|
"loss": 4.8225, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 4.46, |
|
"learning_rate": 0.0005445544554455446, |
|
"loss": 4.8119, |
|
"step": 3600 |
|
}, |
|
{ |
|
"epoch": 4.58, |
|
"learning_rate": 0.00042079207920792084, |
|
"loss": 4.7827, |
|
"step": 3700 |
|
}, |
|
{ |
|
"epoch": 4.7, |
|
"learning_rate": 0.000297029702970297, |
|
"loss": 4.7438, |
|
"step": 3800 |
|
}, |
|
{ |
|
"epoch": 4.83, |
|
"learning_rate": 0.00017326732673267329, |
|
"loss": 4.7972, |
|
"step": 3900 |
|
}, |
|
{ |
|
"epoch": 4.95, |
|
"learning_rate": 4.950495049504951e-05, |
|
"loss": 4.8246, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 5.0, |
|
"eval_loss": 5.520716190338135, |
|
"eval_runtime": 213.5177, |
|
"eval_samples_per_second": 29.674, |
|
"eval_steps_per_second": 7.419, |
|
"step": 4040 |
|
}, |
|
{ |
|
"epoch": 5.0, |
|
"step": 4040, |
|
"total_flos": 1.332945784406016e+17, |
|
"train_loss": 5.578122329711914, |
|
"train_runtime": 9884.2344, |
|
"train_samples_per_second": 26.155, |
|
"train_steps_per_second": 0.409 |
|
} |
|
], |
|
"logging_steps": 100, |
|
"max_steps": 4040, |
|
"num_train_epochs": 5, |
|
"save_steps": 500, |
|
"total_flos": 1.332945784406016e+17, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|