252 lines
5.9 KiB
JSON
252 lines
5.9 KiB
JSON
{
|
|
"best_global_step": 1885,
|
|
"best_metric": 2.8720178604125977,
|
|
"best_model_checkpoint": "/home/p318482/syntactic-bootstrapping/model_trained/cds_o_fr_13/checkpoint-1885",
|
|
"epoch": 14.0,
|
|
"eval_steps": 500,
|
|
"global_step": 2030,
|
|
"is_hyper_param_search": false,
|
|
"is_local_process_zero": true,
|
|
"is_world_process_zero": true,
|
|
"log_history": [
|
|
{
|
|
"epoch": 0.006896551724137931,
|
|
"grad_norm": 32.2354736328125,
|
|
"learning_rate": 0.0,
|
|
"loss": 9.7326,
|
|
"step": 1
|
|
},
|
|
{
|
|
"epoch": 1.0,
|
|
"grad_norm": 1.26012122631073,
|
|
"learning_rate": 2.88e-05,
|
|
"loss": 5.8188,
|
|
"step": 145
|
|
},
|
|
{
|
|
"epoch": 1.0,
|
|
"eval_loss": 4.1781511306762695,
|
|
"eval_runtime": 4.3842,
|
|
"eval_samples_per_second": 803.562,
|
|
"eval_steps_per_second": 3.193,
|
|
"step": 145
|
|
},
|
|
{
|
|
"epoch": 2.0,
|
|
"grad_norm": 1.0639448165893555,
|
|
"learning_rate": 5.7799999999999995e-05,
|
|
"loss": 3.6451,
|
|
"step": 290
|
|
},
|
|
{
|
|
"epoch": 2.0,
|
|
"eval_loss": 3.4042274951934814,
|
|
"eval_runtime": 4.6134,
|
|
"eval_samples_per_second": 763.65,
|
|
"eval_steps_per_second": 3.035,
|
|
"step": 290
|
|
},
|
|
{
|
|
"epoch": 3.0,
|
|
"grad_norm": 0.9253916144371033,
|
|
"learning_rate": 8.680000000000001e-05,
|
|
"loss": 3.2742,
|
|
"step": 435
|
|
},
|
|
{
|
|
"epoch": 3.0,
|
|
"eval_loss": 3.1871676445007324,
|
|
"eval_runtime": 4.7952,
|
|
"eval_samples_per_second": 734.687,
|
|
"eval_steps_per_second": 2.92,
|
|
"step": 435
|
|
},
|
|
{
|
|
"epoch": 4.0,
|
|
"grad_norm": 0.8254706263542175,
|
|
"learning_rate": 9.668067226890756e-05,
|
|
"loss": 3.1065,
|
|
"step": 580
|
|
},
|
|
{
|
|
"epoch": 4.0,
|
|
"eval_loss": 3.0691733360290527,
|
|
"eval_runtime": 4.4368,
|
|
"eval_samples_per_second": 794.047,
|
|
"eval_steps_per_second": 3.155,
|
|
"step": 580
|
|
},
|
|
{
|
|
"epoch": 5.0,
|
|
"grad_norm": 0.7847094535827637,
|
|
"learning_rate": 9.058823529411765e-05,
|
|
"loss": 3.0059,
|
|
"step": 725
|
|
},
|
|
{
|
|
"epoch": 5.0,
|
|
"eval_loss": 3.00063157081604,
|
|
"eval_runtime": 4.5738,
|
|
"eval_samples_per_second": 770.265,
|
|
"eval_steps_per_second": 3.061,
|
|
"step": 725
|
|
},
|
|
{
|
|
"epoch": 6.0,
|
|
"grad_norm": 0.82077956199646,
|
|
"learning_rate": 8.449579831932774e-05,
|
|
"loss": 2.9407,
|
|
"step": 870
|
|
},
|
|
{
|
|
"epoch": 6.0,
|
|
"eval_loss": 2.95955753326416,
|
|
"eval_runtime": 4.664,
|
|
"eval_samples_per_second": 755.368,
|
|
"eval_steps_per_second": 3.002,
|
|
"step": 870
|
|
},
|
|
{
|
|
"epoch": 7.0,
|
|
"grad_norm": 0.7905252575874329,
|
|
"learning_rate": 7.840336134453782e-05,
|
|
"loss": 2.8903,
|
|
"step": 1015
|
|
},
|
|
{
|
|
"epoch": 7.0,
|
|
"eval_loss": 2.929441452026367,
|
|
"eval_runtime": 4.4917,
|
|
"eval_samples_per_second": 784.335,
|
|
"eval_steps_per_second": 3.117,
|
|
"step": 1015
|
|
},
|
|
{
|
|
"epoch": 8.0,
|
|
"grad_norm": 0.8253563642501831,
|
|
"learning_rate": 7.23109243697479e-05,
|
|
"loss": 2.8492,
|
|
"step": 1160
|
|
},
|
|
{
|
|
"epoch": 8.0,
|
|
"eval_loss": 2.907557487487793,
|
|
"eval_runtime": 4.4334,
|
|
"eval_samples_per_second": 794.644,
|
|
"eval_steps_per_second": 3.158,
|
|
"step": 1160
|
|
},
|
|
{
|
|
"epoch": 9.0,
|
|
"grad_norm": 0.8737186789512634,
|
|
"learning_rate": 6.621848739495798e-05,
|
|
"loss": 2.8124,
|
|
"step": 1305
|
|
},
|
|
{
|
|
"epoch": 9.0,
|
|
"eval_loss": 2.894392728805542,
|
|
"eval_runtime": 4.3693,
|
|
"eval_samples_per_second": 806.31,
|
|
"eval_steps_per_second": 3.204,
|
|
"step": 1305
|
|
},
|
|
{
|
|
"epoch": 10.0,
|
|
"grad_norm": 0.9210653901100159,
|
|
"learning_rate": 6.012605042016807e-05,
|
|
"loss": 2.7787,
|
|
"step": 1450
|
|
},
|
|
{
|
|
"epoch": 10.0,
|
|
"eval_loss": 2.8836522102355957,
|
|
"eval_runtime": 4.3707,
|
|
"eval_samples_per_second": 806.047,
|
|
"eval_steps_per_second": 3.203,
|
|
"step": 1450
|
|
},
|
|
{
|
|
"epoch": 11.0,
|
|
"grad_norm": 0.9565634727478027,
|
|
"learning_rate": 5.403361344537815e-05,
|
|
"loss": 2.7466,
|
|
"step": 1595
|
|
},
|
|
{
|
|
"epoch": 11.0,
|
|
"eval_loss": 2.8761401176452637,
|
|
"eval_runtime": 4.46,
|
|
"eval_samples_per_second": 789.914,
|
|
"eval_steps_per_second": 3.139,
|
|
"step": 1595
|
|
},
|
|
{
|
|
"epoch": 12.0,
|
|
"grad_norm": 1.0066618919372559,
|
|
"learning_rate": 4.794117647058824e-05,
|
|
"loss": 2.7144,
|
|
"step": 1740
|
|
},
|
|
{
|
|
"epoch": 12.0,
|
|
"eval_loss": 2.8736512660980225,
|
|
"eval_runtime": 4.4277,
|
|
"eval_samples_per_second": 795.673,
|
|
"eval_steps_per_second": 3.162,
|
|
"step": 1740
|
|
},
|
|
{
|
|
"epoch": 13.0,
|
|
"grad_norm": 1.0534671545028687,
|
|
"learning_rate": 4.184873949579832e-05,
|
|
"loss": 2.6832,
|
|
"step": 1885
|
|
},
|
|
{
|
|
"epoch": 13.0,
|
|
"eval_loss": 2.8720178604125977,
|
|
"eval_runtime": 4.4634,
|
|
"eval_samples_per_second": 789.313,
|
|
"eval_steps_per_second": 3.137,
|
|
"step": 1885
|
|
},
|
|
{
|
|
"epoch": 14.0,
|
|
"grad_norm": 1.1330785751342773,
|
|
"learning_rate": 3.575630252100841e-05,
|
|
"loss": 2.6516,
|
|
"step": 2030
|
|
},
|
|
{
|
|
"epoch": 14.0,
|
|
"eval_loss": 2.874244451522827,
|
|
"eval_runtime": 4.764,
|
|
"eval_samples_per_second": 739.504,
|
|
"eval_steps_per_second": 2.939,
|
|
"step": 2030
|
|
}
|
|
],
|
|
"logging_steps": 500,
|
|
"max_steps": 2900,
|
|
"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": 3.3822685790208e+16,
|
|
"train_batch_size": 256,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|