wangyichen25's picture
Training in progress, step 280, checkpoint
9728c40 verified
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.5827263267429761,
"eval_steps": 20,
"global_step": 280,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.04162330905306972,
"grad_norm": 2.6779890060424805,
"learning_rate": 0.00019828326180257511,
"loss": 5.368,
"mean_token_accuracy": 0.8072595901787281,
"num_tokens": 161548.0,
"step": 20
},
{
"epoch": 0.04162330905306972,
"eval_loss": 0.12305498868227005,
"eval_mean_token_accuracy": 0.9847841203212738,
"eval_num_tokens": 161548.0,
"eval_runtime": 68.9772,
"eval_samples_per_second": 2.9,
"eval_steps_per_second": 0.725,
"step": 20
},
{
"epoch": 0.08324661810613944,
"grad_norm": 0.5366652011871338,
"learning_rate": 0.00018969957081545064,
"loss": 0.1516,
"mean_token_accuracy": 0.9917096085846424,
"num_tokens": 323168.0,
"step": 40
},
{
"epoch": 0.08324661810613944,
"eval_loss": 0.013321125879883766,
"eval_mean_token_accuracy": 0.9938336944580078,
"eval_num_tokens": 323168.0,
"eval_runtime": 68.8811,
"eval_samples_per_second": 2.904,
"eval_steps_per_second": 0.726,
"step": 40
},
{
"epoch": 0.12486992715920915,
"grad_norm": 0.4724605977535248,
"learning_rate": 0.0001811158798283262,
"loss": 0.0462,
"mean_token_accuracy": 0.9939016968011856,
"num_tokens": 484780.0,
"step": 60
},
{
"epoch": 0.12486992715920915,
"eval_loss": 0.010577572509646416,
"eval_mean_token_accuracy": 0.99473925948143,
"eval_num_tokens": 484780.0,
"eval_runtime": 68.7899,
"eval_samples_per_second": 2.907,
"eval_steps_per_second": 0.727,
"step": 60
},
{
"epoch": 0.16649323621227888,
"grad_norm": 0.33190304040908813,
"learning_rate": 0.00017253218884120172,
"loss": 0.0388,
"mean_token_accuracy": 0.9948750860989094,
"num_tokens": 646431.0,
"step": 80
},
{
"epoch": 0.16649323621227888,
"eval_loss": 0.00940256379544735,
"eval_mean_token_accuracy": 0.9951227140426636,
"eval_num_tokens": 646431.0,
"eval_runtime": 69.1687,
"eval_samples_per_second": 2.891,
"eval_steps_per_second": 0.723,
"step": 80
},
{
"epoch": 0.2081165452653486,
"grad_norm": 0.20264093577861786,
"learning_rate": 0.00016394849785407727,
"loss": 0.0367,
"mean_token_accuracy": 0.9950500458478928,
"num_tokens": 808054.0,
"step": 100
},
{
"epoch": 0.2081165452653486,
"eval_loss": 0.008246215991675854,
"eval_mean_token_accuracy": 0.9951834440231323,
"eval_num_tokens": 808054.0,
"eval_runtime": 68.7059,
"eval_samples_per_second": 2.911,
"eval_steps_per_second": 0.728,
"step": 100
},
{
"epoch": 0.2497398543184183,
"grad_norm": 0.30675262212753296,
"learning_rate": 0.0001553648068669528,
"loss": 0.0341,
"mean_token_accuracy": 0.994870014488697,
"num_tokens": 969623.0,
"step": 120
},
{
"epoch": 0.2497398543184183,
"eval_loss": 0.00862042885273695,
"eval_mean_token_accuracy": 0.994941633939743,
"eval_num_tokens": 969623.0,
"eval_runtime": 68.8073,
"eval_samples_per_second": 2.907,
"eval_steps_per_second": 0.727,
"step": 120
},
{
"epoch": 0.29136316337148804,
"grad_norm": 0.1850423365831375,
"learning_rate": 0.00014678111587982832,
"loss": 0.034,
"mean_token_accuracy": 0.9949115067720413,
"num_tokens": 1131222.0,
"step": 140
},
{
"epoch": 0.29136316337148804,
"eval_loss": 0.007929541170597076,
"eval_mean_token_accuracy": 0.9951228404045105,
"eval_num_tokens": 1131222.0,
"eval_runtime": 68.7004,
"eval_samples_per_second": 2.911,
"eval_steps_per_second": 0.728,
"step": 140
},
{
"epoch": 0.33298647242455776,
"grad_norm": 0.3047815263271332,
"learning_rate": 0.00013819742489270387,
"loss": 0.0328,
"mean_token_accuracy": 0.9950883395969867,
"num_tokens": 1292839.0,
"step": 160
},
{
"epoch": 0.33298647242455776,
"eval_loss": 0.007993862964212894,
"eval_mean_token_accuracy": 0.9951430022716522,
"eval_num_tokens": 1292839.0,
"eval_runtime": 68.1668,
"eval_samples_per_second": 2.934,
"eval_steps_per_second": 0.733,
"step": 160
},
{
"epoch": 0.37460978147762747,
"grad_norm": 0.13929370045661926,
"learning_rate": 0.0001296137339055794,
"loss": 0.0331,
"mean_token_accuracy": 0.9951471641659737,
"num_tokens": 1454417.0,
"step": 180
},
{
"epoch": 0.37460978147762747,
"eval_loss": 0.007801079656928778,
"eval_mean_token_accuracy": 0.9954658925533295,
"eval_num_tokens": 1454417.0,
"eval_runtime": 68.0889,
"eval_samples_per_second": 2.937,
"eval_steps_per_second": 0.734,
"step": 180
},
{
"epoch": 0.4162330905306972,
"grad_norm": 0.29847878217697144,
"learning_rate": 0.00012103004291845495,
"loss": 0.0308,
"mean_token_accuracy": 0.9953480623662472,
"num_tokens": 1615982.0,
"step": 200
},
{
"epoch": 0.4162330905306972,
"eval_loss": 0.007313677109777927,
"eval_mean_token_accuracy": 0.9954455089569092,
"eval_num_tokens": 1615982.0,
"eval_runtime": 68.4814,
"eval_samples_per_second": 2.921,
"eval_steps_per_second": 0.73,
"step": 200
},
{
"epoch": 0.4578563995837669,
"grad_norm": 0.14924530684947968,
"learning_rate": 0.00011244635193133049,
"loss": 0.0298,
"mean_token_accuracy": 0.9952255949378014,
"num_tokens": 1777586.0,
"step": 220
},
{
"epoch": 0.4578563995837669,
"eval_loss": 0.0077740405686199665,
"eval_mean_token_accuracy": 0.9953651452064514,
"eval_num_tokens": 1777586.0,
"eval_runtime": 68.0902,
"eval_samples_per_second": 2.937,
"eval_steps_per_second": 0.734,
"step": 220
},
{
"epoch": 0.4994797086368366,
"grad_norm": 0.31947973370552063,
"learning_rate": 0.000103862660944206,
"loss": 0.0299,
"mean_token_accuracy": 0.9952647179365158,
"num_tokens": 1939204.0,
"step": 240
},
{
"epoch": 0.4994797086368366,
"eval_loss": 0.007334359921514988,
"eval_mean_token_accuracy": 0.9954258131980896,
"eval_num_tokens": 1939204.0,
"eval_runtime": 68.026,
"eval_samples_per_second": 2.94,
"eval_steps_per_second": 0.735,
"step": 240
},
{
"epoch": 0.5411030176899063,
"grad_norm": 0.15298807621002197,
"learning_rate": 9.527896995708155e-05,
"loss": 0.0305,
"mean_token_accuracy": 0.9951752610504627,
"num_tokens": 2100821.0,
"step": 260
},
{
"epoch": 0.5411030176899063,
"eval_loss": 0.007329077925533056,
"eval_mean_token_accuracy": 0.9955062341690063,
"eval_num_tokens": 2100821.0,
"eval_runtime": 68.3167,
"eval_samples_per_second": 2.928,
"eval_steps_per_second": 0.732,
"step": 260
},
{
"epoch": 0.5827263267429761,
"grad_norm": 0.1927741914987564,
"learning_rate": 8.669527896995709e-05,
"loss": 0.0301,
"mean_token_accuracy": 0.9952726632356643,
"num_tokens": 2262372.0,
"step": 280
},
{
"epoch": 0.5827263267429761,
"eval_loss": 0.007559146732091904,
"eval_mean_token_accuracy": 0.9953450250625611,
"eval_num_tokens": 2262372.0,
"eval_runtime": 68.1437,
"eval_samples_per_second": 2.935,
"eval_steps_per_second": 0.734,
"step": 280
}
],
"logging_steps": 20,
"max_steps": 481,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 40,
"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.907812129062272e+16,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}