|
{ |
|
"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 |
|
} |
|
|