Llama-2-7b-dpo-2k / trainer_state.json
AmberYifan's picture
Model save
a9c227c verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.992,
"eval_steps": 200,
"global_step": 62,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.016,
"grad_norm": 22.116731061155882,
"learning_rate": 7.142857142857142e-08,
"logits/generated": -0.9303410053253174,
"logits/real": -0.6472625732421875,
"logps/generated": -159.71011352539062,
"logps/real": -190.3052978515625,
"loss": 0.8876,
"rewards/accuracies": 0.0,
"rewards/generated": 0.0,
"rewards/margins": 0.0,
"rewards/real": 0.0,
"step": 1
},
{
"epoch": 0.16,
"grad_norm": 19.00686173642431,
"learning_rate": 4.727272727272727e-07,
"logits/generated": -0.7783747315406799,
"logits/real": -0.7638665437698364,
"logps/generated": -174.0995330810547,
"logps/real": -182.80723571777344,
"loss": 0.8563,
"rewards/accuracies": 0.5277777910232544,
"rewards/generated": 0.09232226759195328,
"rewards/margins": 0.01775742694735527,
"rewards/real": 0.11007969081401825,
"step": 10
},
{
"epoch": 0.32,
"grad_norm": 20.37082961992304,
"learning_rate": 3.818181818181818e-07,
"logits/generated": -0.7393258213996887,
"logits/real": -0.7510851621627808,
"logps/generated": -163.16494750976562,
"logps/real": -164.861572265625,
"loss": 0.8527,
"rewards/accuracies": 0.48750001192092896,
"rewards/generated": 0.45944079756736755,
"rewards/margins": -0.011271441355347633,
"rewards/real": 0.44816938042640686,
"step": 20
},
{
"epoch": 0.48,
"grad_norm": 19.828947810570654,
"learning_rate": 2.909090909090909e-07,
"logits/generated": -0.8125241994857788,
"logits/real": -0.6840296983718872,
"logps/generated": -145.97573852539062,
"logps/real": -154.54061889648438,
"loss": 0.8316,
"rewards/accuracies": 0.625,
"rewards/generated": 0.6635308265686035,
"rewards/margins": 0.1064867228269577,
"rewards/real": 0.7700175046920776,
"step": 30
},
{
"epoch": 0.64,
"grad_norm": 20.449526730384434,
"learning_rate": 2e-07,
"logits/generated": -0.7401232719421387,
"logits/real": -0.6730000972747803,
"logps/generated": -165.48190307617188,
"logps/real": -177.96424865722656,
"loss": 0.8405,
"rewards/accuracies": 0.5625,
"rewards/generated": 0.9031688570976257,
"rewards/margins": 0.03490729257464409,
"rewards/real": 0.9380761384963989,
"step": 40
},
{
"epoch": 0.8,
"grad_norm": 20.801102601242686,
"learning_rate": 1.0909090909090908e-07,
"logits/generated": -0.7878767251968384,
"logits/real": -0.6102726459503174,
"logps/generated": -163.09925842285156,
"logps/real": -180.8894500732422,
"loss": 0.8413,
"rewards/accuracies": 0.5625,
"rewards/generated": 1.0153449773788452,
"rewards/margins": 0.08370596170425415,
"rewards/real": 1.0990509986877441,
"step": 50
},
{
"epoch": 0.96,
"grad_norm": 20.460813305727022,
"learning_rate": 1.818181818181818e-08,
"logits/generated": -0.652385413646698,
"logits/real": -0.5340145230293274,
"logps/generated": -169.20877075195312,
"logps/real": -176.0380859375,
"loss": 0.8323,
"rewards/accuracies": 0.5625,
"rewards/generated": 1.1103785037994385,
"rewards/margins": 0.07993375509977341,
"rewards/real": 1.190312385559082,
"step": 60
},
{
"epoch": 0.992,
"step": 62,
"total_flos": 0.0,
"train_loss": 0.8421394844208995,
"train_runtime": 736.5998,
"train_samples_per_second": 2.715,
"train_steps_per_second": 0.084
}
],
"logging_steps": 10,
"max_steps": 62,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 200,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}