quip-4k-llama / checkpoint-100 /trainer_state.json
Anis1123's picture
Initial model upload
4894c3a verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.7540056550424128,
"eval_steps": 500,
"global_step": 100,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.03770028275212064,
"grad_norm": 3.988708734512329,
"learning_rate": 4.9995083170283816e-05,
"loss": 4.6192,
"num_input_tokens_seen": 50400,
"step": 5
},
{
"epoch": 0.07540056550424128,
"grad_norm": 2.142688512802124,
"learning_rate": 4.998033461515242e-05,
"loss": 3.9149,
"num_input_tokens_seen": 104016,
"step": 10
},
{
"epoch": 0.11310084825636192,
"grad_norm": 1.5928359031677246,
"learning_rate": 4.9955760135896534e-05,
"loss": 3.6912,
"num_input_tokens_seen": 155584,
"step": 15
},
{
"epoch": 0.15080113100848255,
"grad_norm": 1.5493167638778687,
"learning_rate": 4.992136939879856e-05,
"loss": 3.5556,
"num_input_tokens_seen": 202672,
"step": 20
},
{
"epoch": 0.1885014137606032,
"grad_norm": 1.7764347791671753,
"learning_rate": 4.9877175931330346e-05,
"loss": 3.4256,
"num_input_tokens_seen": 254800,
"step": 25
},
{
"epoch": 0.22620169651272384,
"grad_norm": 1.2482728958129883,
"learning_rate": 4.982319711683221e-05,
"loss": 3.3128,
"num_input_tokens_seen": 306352,
"step": 30
},
{
"epoch": 0.2639019792648445,
"grad_norm": 1.2829065322875977,
"learning_rate": 4.975945418767529e-05,
"loss": 3.2688,
"num_input_tokens_seen": 356352,
"step": 35
},
{
"epoch": 0.3016022620169651,
"grad_norm": 1.513293743133545,
"learning_rate": 4.968597221690986e-05,
"loss": 3.297,
"num_input_tokens_seen": 406672,
"step": 40
},
{
"epoch": 0.3393025447690858,
"grad_norm": 1.883090853691101,
"learning_rate": 4.96027801084029e-05,
"loss": 3.232,
"num_input_tokens_seen": 456160,
"step": 45
},
{
"epoch": 0.3770028275212064,
"grad_norm": 1.402272343635559,
"learning_rate": 4.950991058546893e-05,
"loss": 3.267,
"num_input_tokens_seen": 509680,
"step": 50
},
{
"epoch": 0.41470311027332707,
"grad_norm": 1.5488755702972412,
"learning_rate": 4.940740017799833e-05,
"loss": 3.2148,
"num_input_tokens_seen": 559968,
"step": 55
},
{
"epoch": 0.4524033930254477,
"grad_norm": 1.507287859916687,
"learning_rate": 4.929528920808854e-05,
"loss": 3.1403,
"num_input_tokens_seen": 610000,
"step": 60
},
{
"epoch": 0.49010367577756836,
"grad_norm": 1.9119170904159546,
"learning_rate": 4.917362177418342e-05,
"loss": 3.1515,
"num_input_tokens_seen": 661280,
"step": 65
},
{
"epoch": 0.527803958529689,
"grad_norm": 1.7253235578536987,
"learning_rate": 4.904244573372733e-05,
"loss": 3.1468,
"num_input_tokens_seen": 713264,
"step": 70
},
{
"epoch": 0.5655042412818096,
"grad_norm": 1.7201606035232544,
"learning_rate": 4.8901812684340564e-05,
"loss": 3.196,
"num_input_tokens_seen": 762576,
"step": 75
},
{
"epoch": 0.6032045240339302,
"grad_norm": 1.6135213375091553,
"learning_rate": 4.8751777943523634e-05,
"loss": 3.0593,
"num_input_tokens_seen": 813392,
"step": 80
},
{
"epoch": 0.6409048067860509,
"grad_norm": 1.7381868362426758,
"learning_rate": 4.8592400526898314e-05,
"loss": 3.0676,
"num_input_tokens_seen": 860608,
"step": 85
},
{
"epoch": 0.6786050895381716,
"grad_norm": 1.6142843961715698,
"learning_rate": 4.842374312499405e-05,
"loss": 3.1061,
"num_input_tokens_seen": 909104,
"step": 90
},
{
"epoch": 0.7163053722902922,
"grad_norm": 2.0389633178710938,
"learning_rate": 4.824587207858888e-05,
"loss": 2.9847,
"num_input_tokens_seen": 959600,
"step": 95
},
{
"epoch": 0.7540056550424128,
"grad_norm": 1.923561692237854,
"learning_rate": 4.805885735261454e-05,
"loss": 3.0289,
"num_input_tokens_seen": 1013648,
"step": 100
}
],
"logging_steps": 5,
"max_steps": 792,
"num_input_tokens_seen": 1013648,
"num_train_epochs": 6,
"save_steps": 100,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 4.577165786795213e+16,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}