baselines / checkpoint-500 /trainer_state.json
fisherman611's picture
Upload folder using huggingface_hub
dd59dd2 verified
Raw
History Blame Contribute Delete
4.43 kB
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.2834285714285714,
"eval_steps": 500,
"global_step": 500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.004571428571428572,
"grad_norm": 8.67539865837776,
"learning_rate": 0.0,
"logps_logqs/chosen": -0.0693359375,
"logqs/chosen": -2.375,
"logqs/rejected": -2.375,
"logqs_logps/rejected": 0.0693359375,
"loss": 0.6643,
"step": 1
},
{
"epoch": 0.22857142857142856,
"grad_norm": 0.14019721630763493,
"learning_rate": 8.90909090909091e-05,
"logps_logqs/chosen": 0.018798828125,
"logqs/chosen": -4.15625,
"logqs/rejected": -4.15625,
"logqs_logps/rejected": -0.018310546875,
"loss": 0.4195,
"step": 50
},
{
"epoch": 0.45714285714285713,
"grad_norm": 0.45905529641569304,
"learning_rate": 9.955899960329547e-05,
"logps_logqs/chosen": 0.0380859375,
"logqs/chosen": -4.8125,
"logqs/rejected": -4.78125,
"logqs_logps/rejected": -0.037841796875,
"loss": 0.2379,
"step": 100
},
{
"epoch": 0.6857142857142857,
"grad_norm": 0.24812229199411728,
"learning_rate": 9.799779392165715e-05,
"logps_logqs/chosen": 0.014404296875,
"logqs/chosen": -3.703125,
"logqs/rejected": -3.6875,
"logqs_logps/rejected": -0.0140380859375,
"loss": 0.2077,
"step": 150
},
{
"epoch": 0.9142857142857143,
"grad_norm": 0.3582955582094429,
"learning_rate": 9.534371804252728e-05,
"logps_logqs/chosen": 0.01251220703125,
"logqs/chosen": -3.578125,
"logqs/rejected": -3.578125,
"logqs_logps/rejected": -0.01220703125,
"loss": 0.1949,
"step": 200
},
{
"epoch": 1.1417142857142857,
"grad_norm": 0.3905038311837086,
"learning_rate": 9.165720308561348e-05,
"logps_logqs/chosen": 0.0089111328125,
"logqs/chosen": -3.671875,
"logqs/rejected": -3.671875,
"logqs_logps/rejected": -0.0087890625,
"loss": 0.1899,
"step": 250
},
{
"epoch": 1.3702857142857143,
"grad_norm": 0.3292403853943784,
"learning_rate": 8.702218795726617e-05,
"logps_logqs/chosen": 0.006988525390625,
"logqs/chosen": -3.609375,
"logqs/rejected": -3.609375,
"logqs_logps/rejected": -0.00689697265625,
"loss": 0.1779,
"step": 300
},
{
"epoch": 1.5988571428571428,
"grad_norm": 0.44189530758031564,
"learning_rate": 8.154420813078153e-05,
"logps_logqs/chosen": -1.621246337890625e-05,
"logqs/chosen": -3.453125,
"logqs/rejected": -3.453125,
"logqs_logps/rejected": 0.0002841949462890625,
"loss": 0.1688,
"step": 350
},
{
"epoch": 1.8274285714285714,
"grad_norm": 0.29224407333605207,
"learning_rate": 7.534799269067953e-05,
"logps_logqs/chosen": -0.0031890869140625,
"logqs/chosen": -3.28125,
"logqs/rejected": -3.296875,
"logqs_logps/rejected": 0.003509521484375,
"loss": 0.1546,
"step": 400
},
{
"epoch": 2.0548571428571427,
"grad_norm": 0.3845748871005478,
"learning_rate": 6.857462435427786e-05,
"logps_logqs/chosen": 0.0002040863037109375,
"logqs/chosen": -3.4375,
"logqs/rejected": -3.4375,
"logqs_logps/rejected": -2.276897430419922e-05,
"loss": 0.1534,
"step": 450
},
{
"epoch": 2.2834285714285714,
"grad_norm": 0.5119536467282476,
"learning_rate": 6.13783271345295e-05,
"logps_logqs/chosen": 0.002655029296875,
"logqs/chosen": -3.453125,
"logqs/rejected": -3.453125,
"logqs_logps/rejected": -0.0024871826171875,
"loss": 0.1458,
"step": 500
}
],
"logging_steps": 50,
"max_steps": 1095,
"num_input_tokens_seen": 0,
"num_train_epochs": 5,
"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": 0.0,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}