race_cot_16 / trainer_state.json
brettbbb's picture
End of training
9593711
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 20.0,
"eval_steps": 500,
"global_step": 80,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.25,
"learning_rate": 2e-05,
"loss": 2.499,
"step": 1
},
{
"epoch": 0.5,
"learning_rate": 4e-05,
"loss": 2.6363,
"step": 2
},
{
"epoch": 0.75,
"learning_rate": 6e-05,
"loss": 2.3603,
"step": 3
},
{
"epoch": 1.0,
"learning_rate": 8e-05,
"loss": 2.06,
"step": 4
},
{
"epoch": 1.25,
"learning_rate": 0.0001,
"loss": 2.1697,
"step": 5
},
{
"epoch": 1.5,
"learning_rate": 9.866666666666668e-05,
"loss": 2.2317,
"step": 6
},
{
"epoch": 1.75,
"learning_rate": 9.733333333333335e-05,
"loss": 1.9149,
"step": 7
},
{
"epoch": 2.0,
"learning_rate": 9.6e-05,
"loss": 1.9093,
"step": 8
},
{
"epoch": 2.25,
"learning_rate": 9.466666666666667e-05,
"loss": 1.678,
"step": 9
},
{
"epoch": 2.5,
"learning_rate": 9.333333333333334e-05,
"loss": 1.5723,
"step": 10
},
{
"epoch": 2.75,
"learning_rate": 9.200000000000001e-05,
"loss": 1.8143,
"step": 11
},
{
"epoch": 3.0,
"learning_rate": 9.066666666666667e-05,
"loss": 1.5735,
"step": 12
},
{
"epoch": 3.25,
"learning_rate": 8.933333333333334e-05,
"loss": 1.543,
"step": 13
},
{
"epoch": 3.5,
"learning_rate": 8.800000000000001e-05,
"loss": 1.4566,
"step": 14
},
{
"epoch": 3.75,
"learning_rate": 8.666666666666667e-05,
"loss": 1.4143,
"step": 15
},
{
"epoch": 4.0,
"learning_rate": 8.533333333333334e-05,
"loss": 1.168,
"step": 16
},
{
"epoch": 4.25,
"learning_rate": 8.4e-05,
"loss": 1.2837,
"step": 17
},
{
"epoch": 4.5,
"learning_rate": 8.266666666666667e-05,
"loss": 1.3168,
"step": 18
},
{
"epoch": 4.75,
"learning_rate": 8.133333333333334e-05,
"loss": 1.1559,
"step": 19
},
{
"epoch": 5.0,
"learning_rate": 8e-05,
"loss": 0.9572,
"step": 20
},
{
"epoch": 5.25,
"learning_rate": 7.866666666666666e-05,
"loss": 1.0673,
"step": 21
},
{
"epoch": 5.5,
"learning_rate": 7.733333333333333e-05,
"loss": 0.9101,
"step": 22
},
{
"epoch": 5.75,
"learning_rate": 7.6e-05,
"loss": 0.9422,
"step": 23
},
{
"epoch": 6.0,
"learning_rate": 7.466666666666667e-05,
"loss": 0.827,
"step": 24
},
{
"epoch": 6.25,
"learning_rate": 7.333333333333333e-05,
"loss": 0.8863,
"step": 25
},
{
"epoch": 6.5,
"learning_rate": 7.2e-05,
"loss": 0.6619,
"step": 26
},
{
"epoch": 6.75,
"learning_rate": 7.066666666666667e-05,
"loss": 0.627,
"step": 27
},
{
"epoch": 7.0,
"learning_rate": 6.933333333333334e-05,
"loss": 0.5239,
"step": 28
},
{
"epoch": 7.25,
"learning_rate": 6.800000000000001e-05,
"loss": 0.4456,
"step": 29
},
{
"epoch": 7.5,
"learning_rate": 6.666666666666667e-05,
"loss": 0.4669,
"step": 30
},
{
"epoch": 7.75,
"learning_rate": 6.533333333333334e-05,
"loss": 0.3523,
"step": 31
},
{
"epoch": 8.0,
"learning_rate": 6.400000000000001e-05,
"loss": 0.4575,
"step": 32
},
{
"epoch": 8.25,
"learning_rate": 6.266666666666667e-05,
"loss": 0.4323,
"step": 33
},
{
"epoch": 8.5,
"learning_rate": 6.133333333333334e-05,
"loss": 0.2385,
"step": 34
},
{
"epoch": 8.75,
"learning_rate": 6e-05,
"loss": 0.1639,
"step": 35
},
{
"epoch": 9.0,
"learning_rate": 5.866666666666667e-05,
"loss": 0.1429,
"step": 36
},
{
"epoch": 9.25,
"learning_rate": 5.7333333333333336e-05,
"loss": 0.1025,
"step": 37
},
{
"epoch": 9.5,
"learning_rate": 5.6000000000000006e-05,
"loss": 0.1305,
"step": 38
},
{
"epoch": 9.75,
"learning_rate": 5.466666666666666e-05,
"loss": 0.1353,
"step": 39
},
{
"epoch": 10.0,
"learning_rate": 5.333333333333333e-05,
"loss": 0.1031,
"step": 40
},
{
"epoch": 10.25,
"learning_rate": 5.2000000000000004e-05,
"loss": 0.0703,
"step": 41
},
{
"epoch": 10.5,
"learning_rate": 5.0666666666666674e-05,
"loss": 0.0889,
"step": 42
},
{
"epoch": 10.75,
"learning_rate": 4.933333333333334e-05,
"loss": 0.0425,
"step": 43
},
{
"epoch": 11.0,
"learning_rate": 4.8e-05,
"loss": 0.0584,
"step": 44
},
{
"epoch": 11.25,
"learning_rate": 4.666666666666667e-05,
"loss": 0.0461,
"step": 45
},
{
"epoch": 11.5,
"learning_rate": 4.5333333333333335e-05,
"loss": 0.0431,
"step": 46
},
{
"epoch": 11.75,
"learning_rate": 4.4000000000000006e-05,
"loss": 0.0484,
"step": 47
},
{
"epoch": 12.0,
"learning_rate": 4.266666666666667e-05,
"loss": 0.0389,
"step": 48
},
{
"epoch": 12.25,
"learning_rate": 4.133333333333333e-05,
"loss": 0.0303,
"step": 49
},
{
"epoch": 12.5,
"learning_rate": 4e-05,
"loss": 0.0414,
"step": 50
},
{
"epoch": 12.75,
"learning_rate": 3.866666666666667e-05,
"loss": 0.0277,
"step": 51
},
{
"epoch": 13.0,
"learning_rate": 3.733333333333334e-05,
"loss": 0.0459,
"step": 52
},
{
"epoch": 13.25,
"learning_rate": 3.6e-05,
"loss": 0.0446,
"step": 53
},
{
"epoch": 13.5,
"learning_rate": 3.466666666666667e-05,
"loss": 0.0226,
"step": 54
},
{
"epoch": 13.75,
"learning_rate": 3.3333333333333335e-05,
"loss": 0.0299,
"step": 55
},
{
"epoch": 14.0,
"learning_rate": 3.2000000000000005e-05,
"loss": 0.028,
"step": 56
},
{
"epoch": 14.25,
"learning_rate": 3.066666666666667e-05,
"loss": 0.0211,
"step": 57
},
{
"epoch": 14.5,
"learning_rate": 2.9333333333333336e-05,
"loss": 0.0312,
"step": 58
},
{
"epoch": 14.75,
"learning_rate": 2.8000000000000003e-05,
"loss": 0.039,
"step": 59
},
{
"epoch": 15.0,
"learning_rate": 2.6666666666666667e-05,
"loss": 0.0231,
"step": 60
},
{
"epoch": 15.25,
"learning_rate": 2.5333333333333337e-05,
"loss": 0.0211,
"step": 61
},
{
"epoch": 15.5,
"learning_rate": 2.4e-05,
"loss": 0.0217,
"step": 62
},
{
"epoch": 15.75,
"learning_rate": 2.2666666666666668e-05,
"loss": 0.0389,
"step": 63
},
{
"epoch": 16.0,
"learning_rate": 2.1333333333333335e-05,
"loss": 0.0217,
"step": 64
},
{
"epoch": 16.25,
"learning_rate": 2e-05,
"loss": 0.0191,
"step": 65
},
{
"epoch": 16.5,
"learning_rate": 1.866666666666667e-05,
"loss": 0.0305,
"step": 66
},
{
"epoch": 16.75,
"learning_rate": 1.7333333333333336e-05,
"loss": 0.0208,
"step": 67
},
{
"epoch": 17.0,
"learning_rate": 1.6000000000000003e-05,
"loss": 0.0286,
"step": 68
},
{
"epoch": 17.25,
"learning_rate": 1.4666666666666668e-05,
"loss": 0.0201,
"step": 69
},
{
"epoch": 17.5,
"learning_rate": 1.3333333333333333e-05,
"loss": 0.0185,
"step": 70
},
{
"epoch": 17.75,
"learning_rate": 1.2e-05,
"loss": 0.0325,
"step": 71
},
{
"epoch": 18.0,
"learning_rate": 1.0666666666666667e-05,
"loss": 0.0278,
"step": 72
},
{
"epoch": 18.25,
"learning_rate": 9.333333333333334e-06,
"loss": 0.0171,
"step": 73
},
{
"epoch": 18.5,
"learning_rate": 8.000000000000001e-06,
"loss": 0.0205,
"step": 74
},
{
"epoch": 18.75,
"learning_rate": 6.666666666666667e-06,
"loss": 0.0264,
"step": 75
},
{
"epoch": 19.0,
"learning_rate": 5.333333333333334e-06,
"loss": 0.0381,
"step": 76
},
{
"epoch": 19.25,
"learning_rate": 4.000000000000001e-06,
"loss": 0.0316,
"step": 77
},
{
"epoch": 19.5,
"learning_rate": 2.666666666666667e-06,
"loss": 0.0169,
"step": 78
},
{
"epoch": 19.75,
"learning_rate": 1.3333333333333334e-06,
"loss": 0.0176,
"step": 79
},
{
"epoch": 20.0,
"learning_rate": 0.0,
"loss": 0.027,
"step": 80
},
{
"epoch": 20.0,
"step": 80,
"total_flos": 6049593509806080.0,
"train_loss": 0.5706245383480564,
"train_runtime": 196.0958,
"train_samples_per_second": 1.632,
"train_steps_per_second": 0.408
}
],
"logging_steps": 1,
"max_steps": 80,
"num_train_epochs": 20,
"save_steps": 500,
"total_flos": 6049593509806080.0,
"trial_name": null,
"trial_params": null
}