ntr-base-qrecc / old /trainer_state.json
3v324v23's picture
add updated checkpoints
d191936
raw
history blame
6.55 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.5195263290501386,
"global_step": 10000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.13,
"learning_rate": 9.5e-05,
"loss": 0.8107,
"step": 500
},
{
"epoch": 0.13,
"eval_loss": 0.6702454090118408,
"eval_runtime": 4.5337,
"eval_samples_per_second": 66.17,
"eval_steps_per_second": 33.085,
"step": 500
},
{
"epoch": 0.25,
"learning_rate": 9e-05,
"loss": 0.717,
"step": 1000
},
{
"epoch": 0.25,
"eval_loss": 0.6662179827690125,
"eval_runtime": 5.109,
"eval_samples_per_second": 58.719,
"eval_steps_per_second": 29.36,
"step": 1000
},
{
"epoch": 0.38,
"learning_rate": 8.5e-05,
"loss": 0.7212,
"step": 1500
},
{
"epoch": 0.38,
"eval_loss": 0.6485440135002136,
"eval_runtime": 4.6475,
"eval_samples_per_second": 64.551,
"eval_steps_per_second": 32.276,
"step": 1500
},
{
"epoch": 0.5,
"learning_rate": 8e-05,
"loss": 0.7094,
"step": 2000
},
{
"epoch": 0.5,
"eval_loss": 0.6365593671798706,
"eval_runtime": 4.9661,
"eval_samples_per_second": 60.41,
"eval_steps_per_second": 30.205,
"step": 2000
},
{
"epoch": 0.63,
"learning_rate": 7.500000000000001e-05,
"loss": 0.7081,
"step": 2500
},
{
"epoch": 0.63,
"eval_loss": 0.6411292552947998,
"eval_runtime": 5.0449,
"eval_samples_per_second": 59.466,
"eval_steps_per_second": 29.733,
"step": 2500
},
{
"epoch": 0.76,
"learning_rate": 7e-05,
"loss": 0.6803,
"step": 3000
},
{
"epoch": 0.76,
"eval_loss": 0.629453718662262,
"eval_runtime": 5.1787,
"eval_samples_per_second": 57.929,
"eval_steps_per_second": 28.965,
"step": 3000
},
{
"epoch": 0.88,
"learning_rate": 6.500000000000001e-05,
"loss": 0.6886,
"step": 3500
},
{
"epoch": 0.88,
"eval_loss": 0.6171028017997742,
"eval_runtime": 6.0926,
"eval_samples_per_second": 49.24,
"eval_steps_per_second": 24.62,
"step": 3500
},
{
"epoch": 1.01,
"learning_rate": 6e-05,
"loss": 0.673,
"step": 4000
},
{
"epoch": 1.01,
"eval_loss": 0.6199834942817688,
"eval_runtime": 4.5562,
"eval_samples_per_second": 65.844,
"eval_steps_per_second": 32.922,
"step": 4000
},
{
"epoch": 1.13,
"learning_rate": 5.500000000000001e-05,
"loss": 0.6123,
"step": 4500
},
{
"epoch": 1.13,
"eval_loss": 0.615871250629425,
"eval_runtime": 4.3914,
"eval_samples_per_second": 68.315,
"eval_steps_per_second": 34.158,
"step": 4500
},
{
"epoch": 1.26,
"learning_rate": 5e-05,
"loss": 0.6036,
"step": 5000
},
{
"epoch": 1.26,
"eval_loss": 0.6081268191337585,
"eval_runtime": 4.202,
"eval_samples_per_second": 71.395,
"eval_steps_per_second": 35.697,
"step": 5000
},
{
"epoch": 1.39,
"learning_rate": 4.5e-05,
"loss": 0.5854,
"step": 5500
},
{
"epoch": 1.39,
"eval_loss": 0.610887885093689,
"eval_runtime": 4.5399,
"eval_samples_per_second": 66.08,
"eval_steps_per_second": 33.04,
"step": 5500
},
{
"epoch": 1.51,
"learning_rate": 4e-05,
"loss": 0.5936,
"step": 6000
},
{
"epoch": 1.51,
"eval_loss": 0.6057543158531189,
"eval_runtime": 5.4963,
"eval_samples_per_second": 54.582,
"eval_steps_per_second": 27.291,
"step": 6000
},
{
"epoch": 1.64,
"learning_rate": 3.5e-05,
"loss": 0.5917,
"step": 6500
},
{
"epoch": 1.64,
"eval_loss": 0.6038711071014404,
"eval_runtime": 4.4545,
"eval_samples_per_second": 67.348,
"eval_steps_per_second": 33.674,
"step": 6500
},
{
"epoch": 1.76,
"learning_rate": 3e-05,
"loss": 0.5848,
"step": 7000
},
{
"epoch": 1.76,
"eval_loss": 0.6017142534255981,
"eval_runtime": 4.4106,
"eval_samples_per_second": 68.018,
"eval_steps_per_second": 34.009,
"step": 7000
},
{
"epoch": 1.89,
"learning_rate": 2.5e-05,
"loss": 0.5875,
"step": 7500
},
{
"epoch": 1.89,
"eval_loss": 0.6017369627952576,
"eval_runtime": 4.4904,
"eval_samples_per_second": 66.809,
"eval_steps_per_second": 33.404,
"step": 7500
},
{
"epoch": 2.02,
"learning_rate": 2e-05,
"loss": 0.5975,
"step": 8000
},
{
"epoch": 2.02,
"eval_loss": 0.5987575054168701,
"eval_runtime": 4.1962,
"eval_samples_per_second": 71.493,
"eval_steps_per_second": 35.747,
"step": 8000
},
{
"epoch": 2.14,
"learning_rate": 1.5e-05,
"loss": 0.5551,
"step": 8500
},
{
"epoch": 2.14,
"eval_loss": 0.6018974781036377,
"eval_runtime": 4.1728,
"eval_samples_per_second": 71.894,
"eval_steps_per_second": 35.947,
"step": 8500
},
{
"epoch": 2.27,
"learning_rate": 1e-05,
"loss": 0.5338,
"step": 9000
},
{
"epoch": 2.27,
"eval_loss": 0.600771963596344,
"eval_runtime": 4.5111,
"eval_samples_per_second": 66.502,
"eval_steps_per_second": 33.251,
"step": 9000
},
{
"epoch": 2.39,
"learning_rate": 5e-06,
"loss": 0.5648,
"step": 9500
},
{
"epoch": 2.39,
"eval_loss": 0.599320650100708,
"eval_runtime": 4.3392,
"eval_samples_per_second": 69.138,
"eval_steps_per_second": 34.569,
"step": 9500
},
{
"epoch": 2.52,
"learning_rate": 0.0,
"loss": 0.5328,
"step": 10000
},
{
"epoch": 2.52,
"eval_loss": 0.6002569794654846,
"eval_runtime": 5.4729,
"eval_samples_per_second": 54.816,
"eval_steps_per_second": 27.408,
"step": 10000
}
],
"max_steps": 10000,
"num_train_epochs": 3,
"total_flos": 3.963545541632102e+16,
"trial_name": null,
"trial_params": null
}