Mistral-7B-v0.1-gen-dpo-2k-hhrlhf / trainer_state.json
AmberYifan's picture
Model save
07aaf09 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": 187.93083613602994,
"learning_rate": 7.142857142857142e-08,
"logits/generated": -2.834711790084839,
"logits/real": -3.0148138999938965,
"logps/generated": -199.41073608398438,
"logps/real": -229.6434326171875,
"loss": 0.9151,
"rewards/accuracies": 0.0,
"rewards/generated": 0.0,
"rewards/margins": 0.0,
"rewards/real": 0.0,
"step": 1
},
{
"epoch": 0.16,
"grad_norm": 137.95931851523642,
"learning_rate": 4.727272727272727e-07,
"logits/generated": -2.88613224029541,
"logits/real": -2.9299046993255615,
"logps/generated": -221.88681030273438,
"logps/real": -211.3094940185547,
"loss": 0.872,
"rewards/accuracies": 0.5416666865348816,
"rewards/generated": 0.013987368904054165,
"rewards/margins": 0.1755019873380661,
"rewards/real": 0.18948933482170105,
"step": 10
},
{
"epoch": 0.32,
"grad_norm": 111.51574233912503,
"learning_rate": 3.818181818181818e-07,
"logits/generated": -2.764463186264038,
"logits/real": -2.8043417930603027,
"logps/generated": -241.7828369140625,
"logps/real": -226.93673706054688,
"loss": 0.8295,
"rewards/accuracies": 0.7250000238418579,
"rewards/generated": -1.7762424945831299,
"rewards/margins": 0.9314602017402649,
"rewards/real": -0.8447820544242859,
"step": 20
},
{
"epoch": 0.48,
"grad_norm": 85.31778142133322,
"learning_rate": 2.909090909090909e-07,
"logits/generated": -2.688220262527466,
"logits/real": -2.6674160957336426,
"logps/generated": -256.388427734375,
"logps/real": -225.5462188720703,
"loss": 0.6307,
"rewards/accuracies": 0.824999988079071,
"rewards/generated": -2.2042856216430664,
"rewards/margins": 1.7098881006240845,
"rewards/real": -0.4943977892398834,
"step": 30
},
{
"epoch": 0.64,
"grad_norm": 99.66362503152259,
"learning_rate": 2e-07,
"logits/generated": -2.6126770973205566,
"logits/real": -2.4645614624023438,
"logps/generated": -249.4606475830078,
"logps/real": -212.3032684326172,
"loss": 0.5484,
"rewards/accuracies": 0.887499988079071,
"rewards/generated": -2.520421266555786,
"rewards/margins": 2.207958459854126,
"rewards/real": -0.3124631941318512,
"step": 40
},
{
"epoch": 0.8,
"grad_norm": 117.4078873382634,
"learning_rate": 1.0909090909090908e-07,
"logits/generated": -2.496239185333252,
"logits/real": -2.617043972015381,
"logps/generated": -253.89047241210938,
"logps/real": -228.7570037841797,
"loss": 0.51,
"rewards/accuracies": 0.8374999761581421,
"rewards/generated": -3.078453302383423,
"rewards/margins": 2.4696271419525146,
"rewards/real": -0.6088260412216187,
"step": 50
},
{
"epoch": 0.96,
"grad_norm": 120.65322478839221,
"learning_rate": 1.818181818181818e-08,
"logits/generated": -2.538569688796997,
"logits/real": -2.4903368949890137,
"logps/generated": -258.3146057128906,
"logps/real": -220.8245849609375,
"loss": 0.545,
"rewards/accuracies": 0.862500011920929,
"rewards/generated": -3.1993417739868164,
"rewards/margins": 2.3859920501708984,
"rewards/real": -0.8133499026298523,
"step": 60
},
{
"epoch": 0.992,
"step": 62,
"total_flos": 0.0,
"train_loss": 0.6561146609244808,
"train_runtime": 788.6021,
"train_samples_per_second": 2.535,
"train_steps_per_second": 0.079
}
],
"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
}