RewardModelPT / trainer_state.json
nicholasKluge's picture
Upload 14 files
5d5bd89
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.5424164524421595,
"eval_steps": 500,
"global_step": 1200,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.26,
"learning_rate": 4.166666666666667e-05,
"loss": 0.0396,
"step": 200
},
{
"epoch": 0.51,
"learning_rate": 3.3333333333333335e-05,
"loss": 0.0083,
"step": 400
},
{
"epoch": 0.77,
"learning_rate": 2.5e-05,
"loss": 0.0074,
"step": 600
},
{
"epoch": 1.03,
"learning_rate": 1.6666666666666667e-05,
"loss": 0.0062,
"step": 800
},
{
"epoch": 1.29,
"learning_rate": 8.333333333333334e-06,
"loss": 0.002,
"step": 1000
},
{
"epoch": 1.54,
"learning_rate": 0.0,
"loss": 0.0013,
"step": 1200
}
],
"logging_steps": 200,
"max_steps": 1200,
"num_train_epochs": 2,
"save_steps": 400,
"total_flos": 0.0,
"trial_name": null,
"trial_params": null
}