Varifocal-Reranking-Question / trainer_state.json
MoyYuan's picture
Upload folder using huggingface_hub
9cb80a0 verified
{
"best_metric": 0.04804013669490814,
"best_model_checkpoint": "reranker_training_output/focal_question_0.1/checkpoint-1074",
"epoch": 12.0,
"global_step": 4296,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"eval_loss": 0.05196131765842438,
"eval_runtime": 16.6877,
"eval_samples_per_second": 556.038,
"step": 358
},
{
"epoch": 1.4,
"learning_rate": 4.860335195530727e-05,
"loss": 0.0343,
"step": 500
},
{
"epoch": 2.0,
"eval_loss": 0.05120062083005905,
"eval_runtime": 16.6718,
"eval_samples_per_second": 556.569,
"step": 716
},
{
"epoch": 2.79,
"learning_rate": 4.7206703910614525e-05,
"loss": 0.0152,
"step": 1000
},
{
"epoch": 3.0,
"eval_loss": 0.04804013669490814,
"eval_runtime": 16.6811,
"eval_samples_per_second": 556.259,
"step": 1074
},
{
"epoch": 4.0,
"eval_loss": 0.04958715662360191,
"eval_runtime": 16.6857,
"eval_samples_per_second": 556.107,
"step": 1432
},
{
"epoch": 4.19,
"learning_rate": 4.581005586592179e-05,
"loss": 0.0095,
"step": 1500
},
{
"epoch": 5.0,
"eval_loss": 0.05102628469467163,
"eval_runtime": 16.7038,
"eval_samples_per_second": 555.502,
"step": 1790
},
{
"epoch": 5.59,
"learning_rate": 4.4413407821229054e-05,
"loss": 0.0071,
"step": 2000
},
{
"epoch": 6.0,
"eval_loss": 0.05404501408338547,
"eval_runtime": 16.6952,
"eval_samples_per_second": 555.788,
"step": 2148
},
{
"epoch": 6.98,
"learning_rate": 4.301675977653631e-05,
"loss": 0.0062,
"step": 2500
},
{
"epoch": 7.0,
"eval_loss": 0.05861973389983177,
"eval_runtime": 16.6974,
"eval_samples_per_second": 555.716,
"step": 2506
},
{
"epoch": 8.0,
"eval_loss": 0.05649217218160629,
"eval_runtime": 16.6894,
"eval_samples_per_second": 555.983,
"step": 2864
},
{
"epoch": 8.38,
"learning_rate": 4.1620111731843576e-05,
"loss": 0.0052,
"step": 3000
},
{
"epoch": 9.0,
"eval_loss": 0.05835355445742607,
"eval_runtime": 16.6938,
"eval_samples_per_second": 555.836,
"step": 3222
},
{
"epoch": 9.78,
"learning_rate": 4.022346368715084e-05,
"loss": 0.0049,
"step": 3500
},
{
"epoch": 10.0,
"eval_loss": 0.05387857183814049,
"eval_runtime": 16.6873,
"eval_samples_per_second": 556.051,
"step": 3580
},
{
"epoch": 11.0,
"eval_loss": 0.05784677714109421,
"eval_runtime": 16.7026,
"eval_samples_per_second": 555.543,
"step": 3938
},
{
"epoch": 11.17,
"learning_rate": 3.8826815642458106e-05,
"loss": 0.0046,
"step": 4000
},
{
"epoch": 12.0,
"eval_loss": 0.05557448789477348,
"eval_runtime": 16.6808,
"eval_samples_per_second": 556.269,
"step": 4296
}
],
"max_steps": 17900,
"num_train_epochs": 50,
"total_flos": 2.308026552763853e+16,
"trial_name": null,
"trial_params": null
}