ziansu's picture
Training in progress, step 100, checkpoint
e96aeb0 verified
raw
history blame
7.34 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.08880994671403197,
"eval_steps": 50,
"global_step": 100,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.008880994671403197,
"grad_norm": 0.045356735587120056,
"learning_rate": 4.999451708687114e-06,
"logits/chosen": 14.352750778198242,
"logits/rejected": 14.82281494140625,
"logps/chosen": -0.2592294216156006,
"logps/rejected": -0.32852867245674133,
"loss": 0.9315,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.38884416222572327,
"rewards/margins": 0.10394889116287231,
"rewards/rejected": -0.4927930235862732,
"step": 10
},
{
"epoch": 0.017761989342806393,
"grad_norm": 0.05255189165472984,
"learning_rate": 4.997807075247147e-06,
"logits/chosen": 14.80792236328125,
"logits/rejected": 15.017183303833008,
"logps/chosen": -0.2825874388217926,
"logps/rejected": -0.36373966932296753,
"loss": 0.9318,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.4238811433315277,
"rewards/margins": 0.12172831594944,
"rewards/rejected": -0.5456094741821289,
"step": 20
},
{
"epoch": 0.02664298401420959,
"grad_norm": 0.056027185171842575,
"learning_rate": 4.9950668210706795e-06,
"logits/chosen": 14.45336627960205,
"logits/rejected": 15.261484146118164,
"logps/chosen": -0.2638034522533417,
"logps/rejected": -0.37216562032699585,
"loss": 0.9211,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.3957051932811737,
"rewards/margins": 0.1625431776046753,
"rewards/rejected": -0.5582484006881714,
"step": 30
},
{
"epoch": 0.035523978685612786,
"grad_norm": 0.06511653959751129,
"learning_rate": 4.9912321481237616e-06,
"logits/chosen": 14.59851360321045,
"logits/rejected": 15.112770080566406,
"logps/chosen": -0.28972768783569336,
"logps/rejected": -0.36043626070022583,
"loss": 0.941,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.43459147214889526,
"rewards/margins": 0.10606291145086288,
"rewards/rejected": -0.5406544208526611,
"step": 40
},
{
"epoch": 0.04440497335701599,
"grad_norm": 0.0637577474117279,
"learning_rate": 4.986304738420684e-06,
"logits/chosen": 14.678179740905762,
"logits/rejected": 15.114399909973145,
"logps/chosen": -0.3033604919910431,
"logps/rejected": -0.3262741267681122,
"loss": 0.929,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -0.45504075288772583,
"rewards/margins": 0.03437047079205513,
"rewards/rejected": -0.4894111752510071,
"step": 50
},
{
"epoch": 0.04440497335701599,
"eval_logits/chosen": 14.936029434204102,
"eval_logits/rejected": 14.780267715454102,
"eval_logps/chosen": -0.29386037588119507,
"eval_logps/rejected": -0.3304942548274994,
"eval_loss": 0.9458721876144409,
"eval_rewards/accuracies": 0.49450549483299255,
"eval_rewards/chosen": -0.4407905340194702,
"eval_rewards/margins": 0.05495081841945648,
"eval_rewards/rejected": -0.49574142694473267,
"eval_runtime": 27.7436,
"eval_samples_per_second": 26.24,
"eval_steps_per_second": 3.28,
"step": 50
},
{
"epoch": 0.05328596802841918,
"grad_norm": 0.06470987200737,
"learning_rate": 4.980286753286196e-06,
"logits/chosen": 14.19079303741455,
"logits/rejected": 14.986845016479492,
"logps/chosen": -0.26126712560653687,
"logps/rejected": -0.31976616382598877,
"loss": 0.9335,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.3919006288051605,
"rewards/margins": 0.08774860948324203,
"rewards/rejected": -0.47964924573898315,
"step": 60
},
{
"epoch": 0.06216696269982238,
"grad_norm": 0.06163545697927475,
"learning_rate": 4.973180832407471e-06,
"logits/chosen": 14.062408447265625,
"logits/rejected": 15.050743103027344,
"logps/chosen": -0.2698076367378235,
"logps/rejected": -0.37131980061531067,
"loss": 0.9234,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.4047114849090576,
"rewards/margins": 0.1522682160139084,
"rewards/rejected": -0.556979775428772,
"step": 70
},
{
"epoch": 0.07104795737122557,
"grad_norm": 0.05943402647972107,
"learning_rate": 4.964990092676263e-06,
"logits/chosen": 14.523780822753906,
"logits/rejected": 15.173608779907227,
"logps/chosen": -0.2835150957107544,
"logps/rejected": -0.35379332304000854,
"loss": 0.9317,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.4252726137638092,
"rewards/margins": 0.10541732609272003,
"rewards/rejected": -0.5306899547576904,
"step": 80
},
{
"epoch": 0.07992895204262877,
"grad_norm": 0.05566830188035965,
"learning_rate": 4.9557181268217225e-06,
"logits/chosen": 14.752237319946289,
"logits/rejected": 15.213434219360352,
"logps/chosen": -0.25490278005599976,
"logps/rejected": -0.31673288345336914,
"loss": 0.9276,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.38235417008399963,
"rewards/margins": 0.09274514764547348,
"rewards/rejected": -0.4750993847846985,
"step": 90
},
{
"epoch": 0.08880994671403197,
"grad_norm": 0.07879115641117096,
"learning_rate": 4.9453690018345144e-06,
"logits/chosen": 14.362078666687012,
"logits/rejected": 14.708786010742188,
"logps/chosen": -0.27969443798065186,
"logps/rejected": -0.3294224143028259,
"loss": 0.9448,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.41954168677330017,
"rewards/margins": 0.0745919868350029,
"rewards/rejected": -0.4941336512565613,
"step": 100
},
{
"epoch": 0.08880994671403197,
"eval_logits/chosen": 14.668691635131836,
"eval_logits/rejected": 14.53217601776123,
"eval_logps/chosen": -0.2837528884410858,
"eval_logps/rejected": -0.33105531334877014,
"eval_loss": 0.9382757544517517,
"eval_rewards/accuracies": 0.5164835453033447,
"eval_rewards/chosen": -0.4256293475627899,
"eval_rewards/margins": 0.07095365226268768,
"eval_rewards/rejected": -0.4965830445289612,
"eval_runtime": 26.9204,
"eval_samples_per_second": 27.043,
"eval_steps_per_second": 3.38,
"step": 100
}
],
"logging_steps": 10,
"max_steps": 1500,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"save_steps": 50,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 2.4411965756748595e+17,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}