quip-4k-gemma / checkpoint-100 /trainer_state.json
Anis1123's picture
Initial model upload
c8d7b76 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.2918148040771484,
"learning_rate": 4.9995083170283816e-05,
"loss": 4.3742,
"num_input_tokens_seen": 50544,
"step": 5
},
{
"epoch": 0.07540056550424128,
"grad_norm": 3.0331127643585205,
"learning_rate": 4.998033461515242e-05,
"loss": 3.8434,
"num_input_tokens_seen": 104080,
"step": 10
},
{
"epoch": 0.11310084825636192,
"grad_norm": 3.322950601577759,
"learning_rate": 4.9955760135896534e-05,
"loss": 3.6027,
"num_input_tokens_seen": 155776,
"step": 15
},
{
"epoch": 0.15080113100848255,
"grad_norm": 2.7705531120300293,
"learning_rate": 4.992136939879856e-05,
"loss": 3.4102,
"num_input_tokens_seen": 203184,
"step": 20
},
{
"epoch": 0.1885014137606032,
"grad_norm": 3.4572091102600098,
"learning_rate": 4.9877175931330346e-05,
"loss": 3.153,
"num_input_tokens_seen": 255744,
"step": 25
},
{
"epoch": 0.22620169651272384,
"grad_norm": 2.0412893295288086,
"learning_rate": 4.982319711683221e-05,
"loss": 3.0949,
"num_input_tokens_seen": 307808,
"step": 30
},
{
"epoch": 0.2639019792648445,
"grad_norm": 2.0765998363494873,
"learning_rate": 4.975945418767529e-05,
"loss": 3.0392,
"num_input_tokens_seen": 358192,
"step": 35
},
{
"epoch": 0.3016022620169651,
"grad_norm": 2.2433698177337646,
"learning_rate": 4.968597221690986e-05,
"loss": 3.0685,
"num_input_tokens_seen": 408832,
"step": 40
},
{
"epoch": 0.3393025447690858,
"grad_norm": 2.7639973163604736,
"learning_rate": 4.96027801084029e-05,
"loss": 3.0385,
"num_input_tokens_seen": 458368,
"step": 45
},
{
"epoch": 0.3770028275212064,
"grad_norm": 2.0680019855499268,
"learning_rate": 4.950991058546893e-05,
"loss": 3.1079,
"num_input_tokens_seen": 512160,
"step": 50
},
{
"epoch": 0.41470311027332707,
"grad_norm": 2.181554079055786,
"learning_rate": 4.940740017799833e-05,
"loss": 2.9883,
"num_input_tokens_seen": 562928,
"step": 55
},
{
"epoch": 0.4524033930254477,
"grad_norm": 2.0724411010742188,
"learning_rate": 4.929528920808854e-05,
"loss": 2.969,
"num_input_tokens_seen": 613072,
"step": 60
},
{
"epoch": 0.49010367577756836,
"grad_norm": 2.5529158115386963,
"learning_rate": 4.917362177418342e-05,
"loss": 3.0019,
"num_input_tokens_seen": 664640,
"step": 65
},
{
"epoch": 0.527803958529689,
"grad_norm": 2.3983142375946045,
"learning_rate": 4.904244573372733e-05,
"loss": 2.9984,
"num_input_tokens_seen": 716800,
"step": 70
},
{
"epoch": 0.5655042412818096,
"grad_norm": 2.206209182739258,
"learning_rate": 4.8901812684340564e-05,
"loss": 2.9848,
"num_input_tokens_seen": 766528,
"step": 75
},
{
"epoch": 0.6032045240339302,
"grad_norm": 2.22698974609375,
"learning_rate": 4.8751777943523634e-05,
"loss": 2.9169,
"num_input_tokens_seen": 817376,
"step": 80
},
{
"epoch": 0.6409048067860509,
"grad_norm": 2.4569709300994873,
"learning_rate": 4.8592400526898314e-05,
"loss": 2.8844,
"num_input_tokens_seen": 864688,
"step": 85
},
{
"epoch": 0.6786050895381716,
"grad_norm": 2.195237636566162,
"learning_rate": 4.842374312499405e-05,
"loss": 2.9422,
"num_input_tokens_seen": 913216,
"step": 90
},
{
"epoch": 0.7163053722902922,
"grad_norm": 2.54585599899292,
"learning_rate": 4.824587207858888e-05,
"loss": 2.8431,
"num_input_tokens_seen": 964144,
"step": 95
},
{
"epoch": 0.7540056550424128,
"grad_norm": 2.3841867446899414,
"learning_rate": 4.805885735261454e-05,
"loss": 2.8377,
"num_input_tokens_seen": 1018768,
"step": 100
}
],
"logging_steps": 5,
"max_steps": 792,
"num_input_tokens_seen": 1018768,
"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": 1.243855378710528e+16,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}