|
{ |
|
"best_metric": 0.6172745823860168, |
|
"best_model_checkpoint": "hBERTv2_new_pretrain_cola/checkpoint-201", |
|
"epoch": 8.0, |
|
"global_step": 536, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 3.9200000000000004e-05, |
|
"loss": 0.6294, |
|
"step": 67 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_loss": 0.6236423850059509, |
|
"eval_matthews_correlation": 0.0, |
|
"eval_runtime": 1.7907, |
|
"eval_samples_per_second": 582.451, |
|
"eval_steps_per_second": 5.026, |
|
"step": 67 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"learning_rate": 3.8400000000000005e-05, |
|
"loss": 0.6169, |
|
"step": 134 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_loss": 0.6311982870101929, |
|
"eval_matthews_correlation": 0.0, |
|
"eval_runtime": 1.7917, |
|
"eval_samples_per_second": 582.113, |
|
"eval_steps_per_second": 5.023, |
|
"step": 134 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"learning_rate": 3.76e-05, |
|
"loss": 0.6115, |
|
"step": 201 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"eval_loss": 0.6172745823860168, |
|
"eval_matthews_correlation": 0.0, |
|
"eval_runtime": 1.7888, |
|
"eval_samples_per_second": 583.068, |
|
"eval_steps_per_second": 5.031, |
|
"step": 201 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"learning_rate": 3.680000000000001e-05, |
|
"loss": 0.6372, |
|
"step": 268 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"eval_loss": 0.620063841342926, |
|
"eval_matthews_correlation": 0.0, |
|
"eval_runtime": 1.7867, |
|
"eval_samples_per_second": 583.759, |
|
"eval_steps_per_second": 5.037, |
|
"step": 268 |
|
}, |
|
{ |
|
"epoch": 5.0, |
|
"learning_rate": 3.6e-05, |
|
"loss": 0.6087, |
|
"step": 335 |
|
}, |
|
{ |
|
"epoch": 5.0, |
|
"eval_loss": 0.6216873526573181, |
|
"eval_matthews_correlation": 0.0, |
|
"eval_runtime": 1.7924, |
|
"eval_samples_per_second": 581.886, |
|
"eval_steps_per_second": 5.021, |
|
"step": 335 |
|
}, |
|
{ |
|
"epoch": 6.0, |
|
"learning_rate": 3.52e-05, |
|
"loss": 0.6086, |
|
"step": 402 |
|
}, |
|
{ |
|
"epoch": 6.0, |
|
"eval_loss": 0.624801516532898, |
|
"eval_matthews_correlation": 0.0, |
|
"eval_runtime": 1.7896, |
|
"eval_samples_per_second": 582.822, |
|
"eval_steps_per_second": 5.029, |
|
"step": 402 |
|
}, |
|
{ |
|
"epoch": 7.0, |
|
"learning_rate": 3.44e-05, |
|
"loss": 0.6113, |
|
"step": 469 |
|
}, |
|
{ |
|
"epoch": 7.0, |
|
"eval_loss": 0.6282826662063599, |
|
"eval_matthews_correlation": 0.0, |
|
"eval_runtime": 1.7883, |
|
"eval_samples_per_second": 583.234, |
|
"eval_steps_per_second": 5.033, |
|
"step": 469 |
|
}, |
|
{ |
|
"epoch": 8.0, |
|
"learning_rate": 3.3600000000000004e-05, |
|
"loss": 0.6109, |
|
"step": 536 |
|
}, |
|
{ |
|
"epoch": 8.0, |
|
"eval_loss": 0.620032012462616, |
|
"eval_matthews_correlation": 0.0, |
|
"eval_runtime": 1.8037, |
|
"eval_samples_per_second": 578.242, |
|
"eval_steps_per_second": 4.99, |
|
"step": 536 |
|
}, |
|
{ |
|
"epoch": 8.0, |
|
"step": 536, |
|
"total_flos": 1.0010769903058944e+16, |
|
"train_loss": 0.6167977461174353, |
|
"train_runtime": 414.5842, |
|
"train_samples_per_second": 1031.274, |
|
"train_steps_per_second": 8.08 |
|
} |
|
], |
|
"max_steps": 3350, |
|
"num_train_epochs": 50, |
|
"total_flos": 1.0010769903058944e+16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|