{ "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 }