Files
general_knowledge_model/checkpoint-1000/trainer_state.json

235 lines
6.5 KiB
JSON
Raw Normal View History

{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.16024998998437562,
"eval_steps": 500,
"global_step": 1000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"entropy": 2.223189059495926,
"epoch": 0.008012499499218782,
"grad_norm": 2.875,
"learning_rate": 1.9842973882390643e-05,
"loss": 2.3428109741210936,
"mean_token_accuracy": 0.4963963569700718,
"num_tokens": 305366.0,
"step": 50
},
{
"entropy": 2.188130117058754,
"epoch": 0.016024998998437564,
"grad_norm": 3.328125,
"learning_rate": 1.96827431501362e-05,
"loss": 2.225358734130859,
"mean_token_accuracy": 0.5110921229422093,
"num_tokens": 605042.0,
"step": 100
},
{
"entropy": 2.197347580194473,
"epoch": 0.024037498497656344,
"grad_norm": 3.453125,
"learning_rate": 1.952251241788175e-05,
"loss": 2.2299420166015627,
"mean_token_accuracy": 0.5093490639328957,
"num_tokens": 906888.0,
"step": 150
},
{
"entropy": 2.1913950395584108,
"epoch": 0.03204999799687513,
"grad_norm": 2.90625,
"learning_rate": 1.9362281685627303e-05,
"loss": 2.210995941162109,
"mean_token_accuracy": 0.5097909501194954,
"num_tokens": 1213125.0,
"step": 200
},
{
"entropy": 2.1611998677253723,
"epoch": 0.040062497496093905,
"grad_norm": 3.25,
"learning_rate": 1.920205095337286e-05,
"loss": 2.1867474365234374,
"mean_token_accuracy": 0.5154410660266876,
"num_tokens": 1509026.0,
"step": 250
},
{
"entropy": 2.168051454424858,
"epoch": 0.04807499699531269,
"grad_norm": 2.9375,
"learning_rate": 1.904182022111841e-05,
"loss": 2.190071716308594,
"mean_token_accuracy": 0.5156884534657001,
"num_tokens": 1806850.0,
"step": 300
},
{
"entropy": 2.1737154543399813,
"epoch": 0.05608749649453147,
"grad_norm": 3.421875,
"learning_rate": 1.8881589488863967e-05,
"loss": 2.199147186279297,
"mean_token_accuracy": 0.514045044630766,
"num_tokens": 2104505.0,
"step": 350
},
{
"entropy": 2.148039159178734,
"epoch": 0.06409999599375026,
"grad_norm": 3.109375,
"learning_rate": 1.872135875660952e-05,
"loss": 2.176946105957031,
"mean_token_accuracy": 0.5177112428843975,
"num_tokens": 2391088.0,
"step": 400
},
{
"entropy": 2.194890711903572,
"epoch": 0.07211249549296903,
"grad_norm": 3.28125,
"learning_rate": 1.856112802435507e-05,
"loss": 2.216288757324219,
"mean_token_accuracy": 0.5110335703194141,
"num_tokens": 2691272.0,
"step": 450
},
{
"entropy": 2.1557786524295808,
"epoch": 0.08012499499218781,
"grad_norm": 4.15625,
"learning_rate": 1.8400897292100627e-05,
"loss": 2.1812680053710936,
"mean_token_accuracy": 0.5176697050035,
"num_tokens": 2991855.0,
"step": 500
},
{
"entropy": 2.1708167713880537,
"epoch": 0.08813749449140659,
"grad_norm": 4.34375,
"learning_rate": 1.824066655984618e-05,
"loss": 2.1921002197265627,
"mean_token_accuracy": 0.5156507430970669,
"num_tokens": 3295049.0,
"step": 550
},
{
"entropy": 2.1686314886808393,
"epoch": 0.09614999399062538,
"grad_norm": 2.875,
"learning_rate": 1.8080435827591735e-05,
"loss": 2.1914617919921877,
"mean_token_accuracy": 0.5139613692462445,
"num_tokens": 3597206.0,
"step": 600
},
{
"entropy": 2.15742906332016,
"epoch": 0.10416249348984416,
"grad_norm": 3.109375,
"learning_rate": 1.7920205095337287e-05,
"loss": 2.174853973388672,
"mean_token_accuracy": 0.5159518288075924,
"num_tokens": 3905268.0,
"step": 650
},
{
"entropy": 2.162363375425339,
"epoch": 0.11217499298906294,
"grad_norm": 3.171875,
"learning_rate": 1.7759974363082843e-05,
"loss": 2.184231872558594,
"mean_token_accuracy": 0.516631413847208,
"num_tokens": 4207934.0,
"step": 700
},
{
"entropy": 2.1516289287805557,
"epoch": 0.12018749248828171,
"grad_norm": 3.609375,
"learning_rate": 1.7599743630828395e-05,
"loss": 2.167020263671875,
"mean_token_accuracy": 0.516358617246151,
"num_tokens": 4515774.0,
"step": 750
},
{
"entropy": 2.1549544781446457,
"epoch": 0.1281999919875005,
"grad_norm": 3.28125,
"learning_rate": 1.7439512898573947e-05,
"loss": 2.181917877197266,
"mean_token_accuracy": 0.5182220858335495,
"num_tokens": 4814709.0,
"step": 800
},
{
"entropy": 2.1532615756988527,
"epoch": 0.1362124914867193,
"grad_norm": 3.078125,
"learning_rate": 1.7279282166319503e-05,
"loss": 2.1659007263183594,
"mean_token_accuracy": 0.5180425970256328,
"num_tokens": 5112585.0,
"step": 850
},
{
"entropy": 2.1704017293453215,
"epoch": 0.14422499098593805,
"grad_norm": 2.921875,
"learning_rate": 1.7119051434065055e-05,
"loss": 2.1934996032714844,
"mean_token_accuracy": 0.5154964691400528,
"num_tokens": 5411369.0,
"step": 900
},
{
"entropy": 2.1349107217788696,
"epoch": 0.15223749048515683,
"grad_norm": 2.984375,
"learning_rate": 1.6958820701810608e-05,
"loss": 2.1598545837402345,
"mean_token_accuracy": 0.5208920259773732,
"num_tokens": 5707060.0,
"step": 950
},
{
"entropy": 2.1541705918312073,
"epoch": 0.16024998998437562,
"grad_norm": 3.203125,
"learning_rate": 1.6798589969556163e-05,
"loss": 2.170956573486328,
"mean_token_accuracy": 0.5204938448965549,
"num_tokens": 6004079.0,
"step": 1000
}
],
"logging_steps": 50,
"max_steps": 6241,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"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": 7.095551986099814e+16,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}