pythia-160m / last-checkpoint /trainer_state.json
SystemAdmin123's picture
Training in progress, step 100, checkpoint
b90a2e5 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 16.666666666666668,
"eval_steps": 20,
"global_step": 100,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.16666666666666666,
"eval_loss": 3.877518653869629,
"eval_runtime": 5.1478,
"eval_samples_per_second": 291.579,
"eval_steps_per_second": 2.525,
"step": 1
},
{
"epoch": 1.6666666666666665,
"grad_norm": 50.75,
"learning_rate": 0.00019863613034027224,
"loss": 8.3623,
"step": 10
},
{
"epoch": 3.3333333333333335,
"grad_norm": 70.5,
"learning_rate": 0.0001879473751206489,
"loss": 7.7135,
"step": 20
},
{
"epoch": 3.3333333333333335,
"eval_loss": 7.355209827423096,
"eval_runtime": 5.3218,
"eval_samples_per_second": 282.048,
"eval_steps_per_second": 2.443,
"step": 20
},
{
"epoch": 5.0,
"grad_norm": 57.0,
"learning_rate": 0.00016772815716257412,
"loss": 7.2433,
"step": 30
},
{
"epoch": 6.666666666666667,
"grad_norm": 34.0,
"learning_rate": 0.00014016954246529696,
"loss": 7.4558,
"step": 40
},
{
"epoch": 6.666666666666667,
"eval_loss": 7.126556873321533,
"eval_runtime": 5.3539,
"eval_samples_per_second": 280.354,
"eval_steps_per_second": 2.428,
"step": 40
},
{
"epoch": 8.333333333333334,
"grad_norm": 15.8125,
"learning_rate": 0.00010825793454723325,
"loss": 7.077,
"step": 50
},
{
"epoch": 10.0,
"grad_norm": 19.75,
"learning_rate": 7.54514512859201e-05,
"loss": 6.9453,
"step": 60
},
{
"epoch": 10.0,
"eval_loss": 6.862760543823242,
"eval_runtime": 5.2014,
"eval_samples_per_second": 288.578,
"eval_steps_per_second": 2.499,
"step": 60
},
{
"epoch": 11.666666666666666,
"grad_norm": 23.125,
"learning_rate": 4.530518418775733e-05,
"loss": 6.8316,
"step": 70
},
{
"epoch": 13.333333333333334,
"grad_norm": 22.375,
"learning_rate": 2.1085949060360654e-05,
"loss": 6.7644,
"step": 80
},
{
"epoch": 13.333333333333334,
"eval_loss": 6.84028434753418,
"eval_runtime": 5.2665,
"eval_samples_per_second": 285.007,
"eval_steps_per_second": 2.468,
"step": 80
},
{
"epoch": 15.0,
"grad_norm": 17.375,
"learning_rate": 5.418275829936537e-06,
"loss": 6.7252,
"step": 90
},
{
"epoch": 16.666666666666668,
"grad_norm": 32.5,
"learning_rate": 0.0,
"loss": 6.7186,
"step": 100
},
{
"epoch": 16.666666666666668,
"eval_loss": 6.840234279632568,
"eval_runtime": 5.339,
"eval_samples_per_second": 281.138,
"eval_steps_per_second": 2.435,
"step": 100
}
],
"logging_steps": 10,
"max_steps": 100,
"num_input_tokens_seen": 0,
"num_train_epochs": 17,
"save_steps": 20,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 1.823581105291264e+16,
"train_batch_size": 30,
"trial_name": null,
"trial_params": null
}