147 lines
3.4 KiB
JSON
147 lines
3.4 KiB
JSON
{
|
|
"best_global_step": 2744,
|
|
"best_metric": 5.620563507080078,
|
|
"best_model_checkpoint": null,
|
|
"epoch": 7.653061224489796,
|
|
"eval_steps": 500,
|
|
"global_step": 3000,
|
|
"is_hyper_param_search": false,
|
|
"is_local_process_zero": true,
|
|
"is_world_process_zero": true,
|
|
"log_history": [
|
|
{
|
|
"epoch": 0.002551020408163265,
|
|
"grad_norm": 16.32339096069336,
|
|
"learning_rate": 0.0,
|
|
"loss": 10.3351,
|
|
"step": 1
|
|
},
|
|
{
|
|
"epoch": 1.0,
|
|
"grad_norm": 0.6908162236213684,
|
|
"learning_rate": 7.82e-05,
|
|
"loss": 7.1094,
|
|
"step": 392
|
|
},
|
|
{
|
|
"epoch": 1.0,
|
|
"eval_loss": 6.317445755004883,
|
|
"eval_runtime": 15.5824,
|
|
"eval_samples_per_second": 1616.887,
|
|
"eval_steps_per_second": 6.353,
|
|
"step": 392
|
|
},
|
|
{
|
|
"epoch": 2.0,
|
|
"grad_norm": 0.7368313074111938,
|
|
"learning_rate": 9.613387978142077e-05,
|
|
"loss": 6.0963,
|
|
"step": 784
|
|
},
|
|
{
|
|
"epoch": 2.0,
|
|
"eval_loss": 6.024065017700195,
|
|
"eval_runtime": 15.5591,
|
|
"eval_samples_per_second": 1619.314,
|
|
"eval_steps_per_second": 6.363,
|
|
"step": 784
|
|
},
|
|
{
|
|
"epoch": 3.0,
|
|
"grad_norm": 0.7423638701438904,
|
|
"learning_rate": 9.077868852459017e-05,
|
|
"loss": 5.8833,
|
|
"step": 1176
|
|
},
|
|
{
|
|
"epoch": 3.0,
|
|
"eval_loss": 5.884053707122803,
|
|
"eval_runtime": 15.6179,
|
|
"eval_samples_per_second": 1613.212,
|
|
"eval_steps_per_second": 6.339,
|
|
"step": 1176
|
|
},
|
|
{
|
|
"epoch": 4.0,
|
|
"grad_norm": 0.707794725894928,
|
|
"learning_rate": 8.542349726775956e-05,
|
|
"loss": 5.7525,
|
|
"step": 1568
|
|
},
|
|
{
|
|
"epoch": 4.0,
|
|
"eval_loss": 5.788422584533691,
|
|
"eval_runtime": 15.5176,
|
|
"eval_samples_per_second": 1623.64,
|
|
"eval_steps_per_second": 6.38,
|
|
"step": 1568
|
|
},
|
|
{
|
|
"epoch": 5.0,
|
|
"grad_norm": 0.7633333802223206,
|
|
"learning_rate": 8.006830601092896e-05,
|
|
"loss": 5.6482,
|
|
"step": 1960
|
|
},
|
|
{
|
|
"epoch": 5.0,
|
|
"eval_loss": 5.718640327453613,
|
|
"eval_runtime": 15.5828,
|
|
"eval_samples_per_second": 1616.851,
|
|
"eval_steps_per_second": 6.353,
|
|
"step": 1960
|
|
},
|
|
{
|
|
"epoch": 6.0,
|
|
"grad_norm": 0.7929214239120483,
|
|
"learning_rate": 7.471311475409837e-05,
|
|
"loss": 5.5583,
|
|
"step": 2352
|
|
},
|
|
{
|
|
"epoch": 6.0,
|
|
"eval_loss": 5.6633148193359375,
|
|
"eval_runtime": 15.6437,
|
|
"eval_samples_per_second": 1610.551,
|
|
"eval_steps_per_second": 6.328,
|
|
"step": 2352
|
|
},
|
|
{
|
|
"epoch": 7.0,
|
|
"grad_norm": 0.8233454823493958,
|
|
"learning_rate": 6.935792349726776e-05,
|
|
"loss": 5.4779,
|
|
"step": 2744
|
|
},
|
|
{
|
|
"epoch": 7.0,
|
|
"eval_loss": 5.620563507080078,
|
|
"eval_runtime": 15.5999,
|
|
"eval_samples_per_second": 1615.079,
|
|
"eval_steps_per_second": 6.346,
|
|
"step": 2744
|
|
}
|
|
],
|
|
"logging_steps": 500,
|
|
"max_steps": 7840,
|
|
"num_input_tokens_seen": 0,
|
|
"num_train_epochs": 20,
|
|
"save_steps": 500,
|
|
"stateful_callbacks": {
|
|
"TrainerControl": {
|
|
"args": {
|
|
"should_epoch_stop": false,
|
|
"should_evaluate": false,
|
|
"should_log": false,
|
|
"should_save": true,
|
|
"should_training_stop": false
|
|
},
|
|
"attributes": {}
|
|
}
|
|
},
|
|
"total_flos": 5.011548512256e+16,
|
|
"train_batch_size": 256,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|