Qwen2.5-7B-spin-2k / trainer_state.json
AmberYifan's picture
Model save
91da415 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": 171.73661439403955,
"learning_rate": 7.142857142857142e-08,
"logits/generated": -0.7590318322181702,
"logits/real": -1.0396332740783691,
"logps/generated": -289.2199401855469,
"logps/real": -197.24859619140625,
"loss": 0.9187,
"rewards/accuracies": 0.0,
"rewards/generated": 0.0,
"rewards/margins": 0.0,
"rewards/real": 0.0,
"step": 1
},
{
"epoch": 0.16,
"grad_norm": 34.20735933667791,
"learning_rate": 4.727272727272727e-07,
"logits/generated": -0.952869176864624,
"logits/real": -0.9400660991668701,
"logps/generated": -267.64935302734375,
"logps/real": -216.30641174316406,
"loss": 0.6785,
"rewards/accuracies": 0.8611111044883728,
"rewards/generated": -0.6007769107818604,
"rewards/margins": 0.7688150405883789,
"rewards/real": 0.16803814470767975,
"step": 10
},
{
"epoch": 0.32,
"grad_norm": 1.89230891359901,
"learning_rate": 3.818181818181818e-07,
"logits/generated": -0.8387147188186646,
"logits/real": -0.6731612086296082,
"logps/generated": -285.69097900390625,
"logps/real": -183.71739196777344,
"loss": 0.1842,
"rewards/accuracies": 1.0,
"rewards/generated": -2.0443050861358643,
"rewards/margins": 6.490469455718994,
"rewards/real": 4.446164608001709,
"step": 20
},
{
"epoch": 0.48,
"grad_norm": 4.462947343445185,
"learning_rate": 2.909090909090909e-07,
"logits/generated": -0.7116909027099609,
"logits/real": -0.5934397578239441,
"logps/generated": -282.5917053222656,
"logps/real": -128.06210327148438,
"loss": 0.1377,
"rewards/accuracies": 1.0,
"rewards/generated": -2.1207313537597656,
"rewards/margins": 9.267239570617676,
"rewards/real": 7.146508693695068,
"step": 30
},
{
"epoch": 0.64,
"grad_norm": 1.017306581553629,
"learning_rate": 2e-07,
"logits/generated": -0.5800212621688843,
"logits/real": -0.6825695633888245,
"logps/generated": -289.5108337402344,
"logps/real": -159.77926635742188,
"loss": 0.1543,
"rewards/accuracies": 1.0,
"rewards/generated": -2.36224627494812,
"rewards/margins": 9.751206398010254,
"rewards/real": 7.3889594078063965,
"step": 40
},
{
"epoch": 0.8,
"grad_norm": 0.9620294445274639,
"learning_rate": 1.0909090909090908e-07,
"logits/generated": -0.6641489267349243,
"logits/real": -0.6202989220619202,
"logps/generated": -291.93603515625,
"logps/real": -158.97694396972656,
"loss": 0.1462,
"rewards/accuracies": 1.0,
"rewards/generated": -2.7126903533935547,
"rewards/margins": 10.294529914855957,
"rewards/real": 7.581839561462402,
"step": 50
},
{
"epoch": 0.96,
"grad_norm": 1.0600372275387984,
"learning_rate": 1.818181818181818e-08,
"logits/generated": -0.4919801354408264,
"logits/real": -0.810921311378479,
"logps/generated": -301.64776611328125,
"logps/real": -126.55671691894531,
"loss": 0.1347,
"rewards/accuracies": 1.0,
"rewards/generated": -3.056795835494995,
"rewards/margins": 10.84618854522705,
"rewards/real": 7.78939151763916,
"step": 60
},
{
"epoch": 0.992,
"step": 62,
"total_flos": 0.0,
"train_loss": 0.24013032932435313,
"train_runtime": 796.2749,
"train_samples_per_second": 2.502,
"train_steps_per_second": 0.078
}
],
"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
}