Mistral-7B-Instruct-v0.2-GP.../io-chatbot-v2/checkpoint-54/trainer_state.json

184 lines
3.6 KiB
JSON

{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 18.0,
"eval_steps": 5,
"global_step": 54,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.67,
"learning_rate": 2.9999177540482684e-05,
"loss": 2.6576,
"step": 2
},
{
"epoch": 1.33,
"learning_rate": 2.9996710252122685e-05,
"loss": 2.1353,
"step": 4
},
{
"epoch": 2.0,
"learning_rate": 2.9992598405485974e-05,
"loss": 1.8307,
"step": 6
},
{
"epoch": 2.67,
"learning_rate": 2.9986842451482876e-05,
"loss": 1.5685,
"step": 8
},
{
"epoch": 3.33,
"learning_rate": 2.9979443021318607e-05,
"loss": 1.3489,
"step": 10
},
{
"epoch": 4.0,
"learning_rate": 2.9970400926424075e-05,
"loss": 1.1757,
"step": 12
},
{
"epoch": 4.67,
"learning_rate": 2.995971715836687e-05,
"loss": 1.0709,
"step": 14
},
{
"epoch": 5.33,
"learning_rate": 2.9947392888742566e-05,
"loss": 1.0194,
"step": 16
},
{
"epoch": 6.0,
"learning_rate": 2.9933429469046202e-05,
"loss": 0.9694,
"step": 18
},
{
"epoch": 6.67,
"learning_rate": 2.99178284305241e-05,
"loss": 0.9486,
"step": 20
},
{
"epoch": 7.33,
"learning_rate": 2.9900591484005944e-05,
"loss": 0.9138,
"step": 22
},
{
"epoch": 8.0,
"learning_rate": 2.988172051971717e-05,
"loss": 0.8702,
"step": 24
},
{
"epoch": 8.67,
"learning_rate": 2.9861217607071655e-05,
"loss": 0.8556,
"step": 26
},
{
"epoch": 9.33,
"learning_rate": 2.983908499444483e-05,
"loss": 0.7949,
"step": 28
},
{
"epoch": 10.0,
"learning_rate": 2.981532510892707e-05,
"loss": 0.785,
"step": 30
},
{
"epoch": 10.67,
"learning_rate": 2.9789940556057574e-05,
"loss": 0.7599,
"step": 32
},
{
"epoch": 11.33,
"learning_rate": 2.9762934119538628e-05,
"loss": 0.7527,
"step": 34
},
{
"epoch": 12.0,
"learning_rate": 2.9734308760930333e-05,
"loss": 0.7187,
"step": 36
},
{
"epoch": 12.67,
"learning_rate": 2.9704067619325828e-05,
"loss": 0.7241,
"step": 38
},
{
"epoch": 13.33,
"learning_rate": 2.9672214011007087e-05,
"loss": 0.7075,
"step": 40
},
{
"epoch": 14.0,
"learning_rate": 2.9638751429081213e-05,
"loss": 0.6973,
"step": 42
},
{
"epoch": 14.67,
"learning_rate": 2.9603683543097406e-05,
"loss": 0.6908,
"step": 44
},
{
"epoch": 15.33,
"learning_rate": 2.9567014198644542e-05,
"loss": 0.6889,
"step": 46
},
{
"epoch": 16.0,
"learning_rate": 2.9528747416929467e-05,
"loss": 0.675,
"step": 48
},
{
"epoch": 16.67,
"learning_rate": 2.9488887394336025e-05,
"loss": 0.6694,
"step": 50
},
{
"epoch": 17.33,
"learning_rate": 2.9447438501964873e-05,
"loss": 0.6485,
"step": 52
},
{
"epoch": 18.0,
"learning_rate": 2.9404405285154146e-05,
"loss": 0.6777,
"step": 54
}
],
"logging_steps": 2,
"max_steps": 600,
"num_input_tokens_seen": 0,
"num_train_epochs": 200,
"save_steps": 500,
"total_flos": 2951182590935040.0,
"train_batch_size": 52,
"trial_name": null,
"trial_params": null
}