700 lines
14 KiB
JSON
700 lines
14 KiB
JSON
{
|
|
"best_metric": null,
|
|
"best_model_checkpoint": null,
|
|
"epoch": 2.527646129541864,
|
|
"eval_steps": 10,
|
|
"global_step": 100,
|
|
"is_hyper_param_search": false,
|
|
"is_local_process_zero": true,
|
|
"is_world_process_zero": true,
|
|
"log_history": [
|
|
{
|
|
"epoch": 0.03,
|
|
"learning_rate": 2.2e-05,
|
|
"loss": 0.9856,
|
|
"step": 1
|
|
},
|
|
{
|
|
"epoch": 0.05,
|
|
"learning_rate": 2.1998557720321954e-05,
|
|
"loss": 0.8657,
|
|
"step": 2
|
|
},
|
|
{
|
|
"epoch": 0.08,
|
|
"learning_rate": 2.1994231259500664e-05,
|
|
"loss": 0.7599,
|
|
"step": 3
|
|
},
|
|
{
|
|
"epoch": 0.1,
|
|
"learning_rate": 2.1987021752075498e-05,
|
|
"loss": 0.6924,
|
|
"step": 4
|
|
},
|
|
{
|
|
"epoch": 0.13,
|
|
"learning_rate": 2.1976931088614826e-05,
|
|
"loss": 0.818,
|
|
"step": 5
|
|
},
|
|
{
|
|
"epoch": 0.15,
|
|
"learning_rate": 2.1963961915220263e-05,
|
|
"loss": 1.2993,
|
|
"step": 6
|
|
},
|
|
{
|
|
"epoch": 0.18,
|
|
"learning_rate": 2.1948117632832752e-05,
|
|
"loss": 1.4107,
|
|
"step": 7
|
|
},
|
|
{
|
|
"epoch": 0.2,
|
|
"learning_rate": 2.1929402396340756e-05,
|
|
"loss": 0.8539,
|
|
"step": 8
|
|
},
|
|
{
|
|
"epoch": 0.23,
|
|
"learning_rate": 2.1907821113490676e-05,
|
|
"loss": 0.7822,
|
|
"step": 9
|
|
},
|
|
{
|
|
"epoch": 0.25,
|
|
"learning_rate": 2.1883379443599917e-05,
|
|
"loss": 0.6498,
|
|
"step": 10
|
|
},
|
|
{
|
|
"epoch": 0.25,
|
|
"eval_loss": 0.777102530002594,
|
|
"eval_runtime": 26.948,
|
|
"eval_samples_per_second": 30.726,
|
|
"eval_steps_per_second": 0.482,
|
|
"step": 10
|
|
},
|
|
{
|
|
"epoch": 0.28,
|
|
"learning_rate": 2.1856083796072795e-05,
|
|
"loss": 0.7439,
|
|
"step": 11
|
|
},
|
|
{
|
|
"epoch": 0.3,
|
|
"learning_rate": 2.182594132871982e-05,
|
|
"loss": 1.0097,
|
|
"step": 12
|
|
},
|
|
{
|
|
"epoch": 0.33,
|
|
"learning_rate": 2.179295994588063e-05,
|
|
"loss": 1.3189,
|
|
"step": 13
|
|
},
|
|
{
|
|
"epoch": 0.35,
|
|
"learning_rate": 2.175714829635128e-05,
|
|
"loss": 0.7509,
|
|
"step": 14
|
|
},
|
|
{
|
|
"epoch": 0.38,
|
|
"learning_rate": 2.1718515771116186e-05,
|
|
"loss": 0.7273,
|
|
"step": 15
|
|
},
|
|
{
|
|
"epoch": 0.4,
|
|
"learning_rate": 2.167707250088555e-05,
|
|
"loss": 0.6242,
|
|
"step": 16
|
|
},
|
|
{
|
|
"epoch": 0.43,
|
|
"learning_rate": 2.1632829353438724e-05,
|
|
"loss": 0.6135,
|
|
"step": 17
|
|
},
|
|
{
|
|
"epoch": 0.45,
|
|
"learning_rate": 2.1585797930774335e-05,
|
|
"loss": 0.9365,
|
|
"step": 18
|
|
},
|
|
{
|
|
"epoch": 0.48,
|
|
"learning_rate": 2.1535990566067858e-05,
|
|
"loss": 1.4216,
|
|
"step": 19
|
|
},
|
|
{
|
|
"epoch": 0.51,
|
|
"learning_rate": 2.148342032043747e-05,
|
|
"loss": 0.7369,
|
|
"step": 20
|
|
},
|
|
{
|
|
"epoch": 0.51,
|
|
"eval_loss": 0.7404242157936096,
|
|
"eval_runtime": 26.98,
|
|
"eval_samples_per_second": 30.689,
|
|
"eval_steps_per_second": 0.482,
|
|
"step": 20
|
|
},
|
|
{
|
|
"epoch": 0.53,
|
|
"learning_rate": 2.1428100979518976e-05,
|
|
"loss": 0.7605,
|
|
"step": 21
|
|
},
|
|
{
|
|
"epoch": 0.56,
|
|
"learning_rate": 2.1370047049850775e-05,
|
|
"loss": 0.6095,
|
|
"step": 22
|
|
},
|
|
{
|
|
"epoch": 0.58,
|
|
"learning_rate": 2.130927375506978e-05,
|
|
"loss": 0.5955,
|
|
"step": 23
|
|
},
|
|
{
|
|
"epoch": 0.61,
|
|
"learning_rate": 2.1245797031919265e-05,
|
|
"loss": 0.851,
|
|
"step": 24
|
|
},
|
|
{
|
|
"epoch": 0.63,
|
|
"learning_rate": 2.1179633526069748e-05,
|
|
"loss": 1.324,
|
|
"step": 25
|
|
},
|
|
{
|
|
"epoch": 0.66,
|
|
"learning_rate": 2.1110800587753934e-05,
|
|
"loss": 0.9368,
|
|
"step": 26
|
|
},
|
|
{
|
|
"epoch": 0.68,
|
|
"learning_rate": 2.103931626721694e-05,
|
|
"loss": 0.7307,
|
|
"step": 27
|
|
},
|
|
{
|
|
"epoch": 0.71,
|
|
"learning_rate": 2.0965199309982913e-05,
|
|
"loss": 0.6489,
|
|
"step": 28
|
|
},
|
|
{
|
|
"epoch": 0.73,
|
|
"learning_rate": 2.0888469151939347e-05,
|
|
"loss": 0.536,
|
|
"step": 29
|
|
},
|
|
{
|
|
"epoch": 0.76,
|
|
"learning_rate": 2.0809145914240356e-05,
|
|
"loss": 0.732,
|
|
"step": 30
|
|
},
|
|
{
|
|
"epoch": 0.76,
|
|
"eval_loss": 0.7266919612884521,
|
|
"eval_runtime": 26.9346,
|
|
"eval_samples_per_second": 30.741,
|
|
"eval_steps_per_second": 0.483,
|
|
"step": 30
|
|
},
|
|
{
|
|
"epoch": 0.78,
|
|
"learning_rate": 2.072725039803026e-05,
|
|
"loss": 1.1248,
|
|
"step": 31
|
|
},
|
|
{
|
|
"epoch": 0.81,
|
|
"learning_rate": 2.0642804078988827e-05,
|
|
"loss": 1.0917,
|
|
"step": 32
|
|
},
|
|
{
|
|
"epoch": 0.83,
|
|
"learning_rate": 2.055582910169966e-05,
|
|
"loss": 0.735,
|
|
"step": 33
|
|
},
|
|
{
|
|
"epoch": 0.86,
|
|
"learning_rate": 2.0466348273843176e-05,
|
|
"loss": 0.6749,
|
|
"step": 34
|
|
},
|
|
{
|
|
"epoch": 0.88,
|
|
"learning_rate": 2.0374385060215662e-05,
|
|
"loss": 0.5519,
|
|
"step": 35
|
|
},
|
|
{
|
|
"epoch": 0.91,
|
|
"learning_rate": 2.027996357657605e-05,
|
|
"loss": 0.6593,
|
|
"step": 36
|
|
},
|
|
{
|
|
"epoch": 0.94,
|
|
"learning_rate": 2.018310858332198e-05,
|
|
"loss": 0.959,
|
|
"step": 37
|
|
},
|
|
{
|
|
"epoch": 0.96,
|
|
"learning_rate": 2.008384547899682e-05,
|
|
"loss": 1.168,
|
|
"step": 38
|
|
},
|
|
{
|
|
"epoch": 0.99,
|
|
"learning_rate": 1.9982200293629318e-05,
|
|
"loss": 0.6186,
|
|
"step": 39
|
|
},
|
|
{
|
|
"epoch": 1.01,
|
|
"learning_rate": 1.9878199681907702e-05,
|
|
"loss": 1.1207,
|
|
"step": 40
|
|
},
|
|
{
|
|
"epoch": 1.01,
|
|
"eval_loss": 0.7203702926635742,
|
|
"eval_runtime": 26.9926,
|
|
"eval_samples_per_second": 30.675,
|
|
"eval_steps_per_second": 0.482,
|
|
"step": 40
|
|
},
|
|
{
|
|
"epoch": 1.04,
|
|
"learning_rate": 1.977187091618993e-05,
|
|
"loss": 0.7095,
|
|
"step": 41
|
|
},
|
|
{
|
|
"epoch": 1.06,
|
|
"learning_rate": 1.9663241879351998e-05,
|
|
"loss": 0.6781,
|
|
"step": 42
|
|
},
|
|
{
|
|
"epoch": 1.09,
|
|
"learning_rate": 1.9552341057476144e-05,
|
|
"loss": 0.5649,
|
|
"step": 43
|
|
},
|
|
{
|
|
"epoch": 1.11,
|
|
"learning_rate": 1.9439197532380845e-05,
|
|
"loss": 0.5466,
|
|
"step": 44
|
|
},
|
|
{
|
|
"epoch": 1.14,
|
|
"learning_rate": 1.9323840973994647e-05,
|
|
"loss": 0.8721,
|
|
"step": 45
|
|
},
|
|
{
|
|
"epoch": 1.16,
|
|
"learning_rate": 1.9206301632575707e-05,
|
|
"loss": 1.2152,
|
|
"step": 46
|
|
},
|
|
{
|
|
"epoch": 1.19,
|
|
"learning_rate": 1.9086610330779205e-05,
|
|
"loss": 0.6863,
|
|
"step": 47
|
|
},
|
|
{
|
|
"epoch": 1.21,
|
|
"learning_rate": 1.8964798455574637e-05,
|
|
"loss": 0.6792,
|
|
"step": 48
|
|
},
|
|
{
|
|
"epoch": 1.24,
|
|
"learning_rate": 1.884089795001512e-05,
|
|
"loss": 0.5667,
|
|
"step": 49
|
|
},
|
|
{
|
|
"epoch": 1.26,
|
|
"learning_rate": 1.871494130486089e-05,
|
|
"loss": 0.5256,
|
|
"step": 50
|
|
},
|
|
{
|
|
"epoch": 1.26,
|
|
"eval_loss": 0.6949857473373413,
|
|
"eval_runtime": 26.8774,
|
|
"eval_samples_per_second": 30.807,
|
|
"eval_steps_per_second": 0.484,
|
|
"step": 50
|
|
},
|
|
{
|
|
"epoch": 1.29,
|
|
"learning_rate": 1.8586961550059147e-05,
|
|
"loss": 0.7481,
|
|
"step": 51
|
|
},
|
|
{
|
|
"epoch": 1.31,
|
|
"learning_rate": 1.8456992246082534e-05,
|
|
"loss": 1.2028,
|
|
"step": 52
|
|
},
|
|
{
|
|
"epoch": 1.34,
|
|
"learning_rate": 1.8325067475128493e-05,
|
|
"loss": 0.8791,
|
|
"step": 53
|
|
},
|
|
{
|
|
"epoch": 1.36,
|
|
"learning_rate": 1.8191221832181777e-05,
|
|
"loss": 0.6651,
|
|
"step": 54
|
|
},
|
|
{
|
|
"epoch": 1.39,
|
|
"learning_rate": 1.8055490415942538e-05,
|
|
"loss": 0.5875,
|
|
"step": 55
|
|
},
|
|
{
|
|
"epoch": 1.42,
|
|
"learning_rate": 1.791790881962229e-05,
|
|
"loss": 0.5143,
|
|
"step": 56
|
|
},
|
|
{
|
|
"epoch": 1.44,
|
|
"learning_rate": 1.77785131216102e-05,
|
|
"loss": 0.6447,
|
|
"step": 57
|
|
},
|
|
{
|
|
"epoch": 1.47,
|
|
"learning_rate": 1.7637339876012172e-05,
|
|
"loss": 1.0286,
|
|
"step": 58
|
|
},
|
|
{
|
|
"epoch": 1.49,
|
|
"learning_rate": 1.749442610306515e-05,
|
|
"loss": 1.1227,
|
|
"step": 59
|
|
},
|
|
{
|
|
"epoch": 1.52,
|
|
"learning_rate": 1.7349809279429216e-05,
|
|
"loss": 0.6634,
|
|
"step": 60
|
|
},
|
|
{
|
|
"epoch": 1.52,
|
|
"eval_loss": 0.6882586479187012,
|
|
"eval_runtime": 26.9816,
|
|
"eval_samples_per_second": 30.688,
|
|
"eval_steps_per_second": 0.482,
|
|
"step": 60
|
|
},
|
|
{
|
|
"epoch": 1.54,
|
|
"learning_rate": 1.7203527328359972e-05,
|
|
"loss": 0.6145,
|
|
"step": 61
|
|
},
|
|
{
|
|
"epoch": 1.57,
|
|
"learning_rate": 1.705561860976384e-05,
|
|
"loss": 0.5139,
|
|
"step": 62
|
|
},
|
|
{
|
|
"epoch": 1.59,
|
|
"learning_rate": 1.6906121910138824e-05,
|
|
"loss": 0.5819,
|
|
"step": 63
|
|
},
|
|
{
|
|
"epoch": 1.62,
|
|
"learning_rate": 1.675507643240343e-05,
|
|
"loss": 0.8716,
|
|
"step": 64
|
|
},
|
|
{
|
|
"epoch": 1.64,
|
|
"learning_rate": 1.6602521785616394e-05,
|
|
"loss": 1.2125,
|
|
"step": 65
|
|
},
|
|
{
|
|
"epoch": 1.67,
|
|
"learning_rate": 1.6448497974589863e-05,
|
|
"loss": 0.6803,
|
|
"step": 66
|
|
},
|
|
{
|
|
"epoch": 1.69,
|
|
"learning_rate": 1.6293045389398855e-05,
|
|
"loss": 0.6762,
|
|
"step": 67
|
|
},
|
|
{
|
|
"epoch": 1.72,
|
|
"learning_rate": 1.6136204794789646e-05,
|
|
"loss": 0.5324,
|
|
"step": 68
|
|
},
|
|
{
|
|
"epoch": 1.74,
|
|
"learning_rate": 1.597801731948993e-05,
|
|
"loss": 0.541,
|
|
"step": 69
|
|
},
|
|
{
|
|
"epoch": 1.77,
|
|
"learning_rate": 1.5818524445423504e-05,
|
|
"loss": 0.8724,
|
|
"step": 70
|
|
},
|
|
{
|
|
"epoch": 1.77,
|
|
"eval_loss": 0.6789196729660034,
|
|
"eval_runtime": 26.8064,
|
|
"eval_samples_per_second": 30.888,
|
|
"eval_steps_per_second": 0.485,
|
|
"step": 70
|
|
},
|
|
{
|
|
"epoch": 1.79,
|
|
"learning_rate": 1.5657767996832395e-05,
|
|
"loss": 1.204,
|
|
"step": 71
|
|
},
|
|
{
|
|
"epoch": 1.82,
|
|
"learning_rate": 1.549579012930912e-05,
|
|
"loss": 0.6885,
|
|
"step": 72
|
|
},
|
|
{
|
|
"epoch": 1.85,
|
|
"learning_rate": 1.5332633318742157e-05,
|
|
"loss": 0.6975,
|
|
"step": 73
|
|
},
|
|
{
|
|
"epoch": 1.87,
|
|
"learning_rate": 1.5168340350177374e-05,
|
|
"loss": 0.5451,
|
|
"step": 74
|
|
},
|
|
{
|
|
"epoch": 1.9,
|
|
"learning_rate": 1.5002954306598364e-05,
|
|
"loss": 0.4803,
|
|
"step": 75
|
|
},
|
|
{
|
|
"epoch": 1.92,
|
|
"learning_rate": 1.4836518557628703e-05,
|
|
"loss": 0.6776,
|
|
"step": 76
|
|
},
|
|
{
|
|
"epoch": 1.95,
|
|
"learning_rate": 1.4669076748159028e-05,
|
|
"loss": 1.0698,
|
|
"step": 77
|
|
},
|
|
{
|
|
"epoch": 1.97,
|
|
"learning_rate": 1.4500672786901884e-05,
|
|
"loss": 0.7064,
|
|
"step": 78
|
|
},
|
|
{
|
|
"epoch": 2.0,
|
|
"learning_rate": 1.4331350834877457e-05,
|
|
"loss": 0.7407,
|
|
"step": 79
|
|
},
|
|
{
|
|
"epoch": 2.02,
|
|
"learning_rate": 1.4161155293833112e-05,
|
|
"loss": 0.9098,
|
|
"step": 80
|
|
},
|
|
{
|
|
"epoch": 2.02,
|
|
"eval_loss": 0.6750211715698242,
|
|
"eval_runtime": 27.1108,
|
|
"eval_samples_per_second": 30.541,
|
|
"eval_steps_per_second": 0.48,
|
|
"step": 80
|
|
},
|
|
{
|
|
"epoch": 2.05,
|
|
"learning_rate": 1.3990130794599772e-05,
|
|
"loss": 0.6304,
|
|
"step": 81
|
|
},
|
|
{
|
|
"epoch": 2.07,
|
|
"learning_rate": 1.3818322185388293e-05,
|
|
"loss": 0.5601,
|
|
"step": 82
|
|
},
|
|
{
|
|
"epoch": 2.1,
|
|
"learning_rate": 1.3645774520028791e-05,
|
|
"loss": 0.4661,
|
|
"step": 83
|
|
},
|
|
{
|
|
"epoch": 2.12,
|
|
"learning_rate": 1.3472533046156025e-05,
|
|
"loss": 0.5971,
|
|
"step": 84
|
|
},
|
|
{
|
|
"epoch": 2.15,
|
|
"learning_rate": 1.3298643193344028e-05,
|
|
"loss": 0.8505,
|
|
"step": 85
|
|
},
|
|
{
|
|
"epoch": 2.17,
|
|
"learning_rate": 1.3124150561192969e-05,
|
|
"loss": 1.1119,
|
|
"step": 86
|
|
},
|
|
{
|
|
"epoch": 2.2,
|
|
"learning_rate": 1.2949100907371452e-05,
|
|
"loss": 0.6298,
|
|
"step": 87
|
|
},
|
|
{
|
|
"epoch": 2.22,
|
|
"learning_rate": 1.2773540135617358e-05,
|
|
"loss": 0.5901,
|
|
"step": 88
|
|
},
|
|
{
|
|
"epoch": 2.25,
|
|
"learning_rate": 1.2597514283700393e-05,
|
|
"loss": 0.483,
|
|
"step": 89
|
|
},
|
|
{
|
|
"epoch": 2.27,
|
|
"learning_rate": 1.2421069511349471e-05,
|
|
"loss": 0.522,
|
|
"step": 90
|
|
},
|
|
{
|
|
"epoch": 2.27,
|
|
"eval_loss": 0.6687783002853394,
|
|
"eval_runtime": 26.924,
|
|
"eval_samples_per_second": 30.753,
|
|
"eval_steps_per_second": 0.483,
|
|
"step": 90
|
|
},
|
|
{
|
|
"epoch": 2.3,
|
|
"learning_rate": 1.2244252088148128e-05,
|
|
"loss": 0.8623,
|
|
"step": 91
|
|
},
|
|
{
|
|
"epoch": 2.33,
|
|
"learning_rate": 1.2067108381401128e-05,
|
|
"loss": 1.1752,
|
|
"step": 92
|
|
},
|
|
{
|
|
"epoch": 2.35,
|
|
"learning_rate": 1.1889684843975447e-05,
|
|
"loss": 0.6268,
|
|
"step": 93
|
|
},
|
|
{
|
|
"epoch": 2.38,
|
|
"learning_rate": 1.1712028002118794e-05,
|
|
"loss": 0.6146,
|
|
"step": 94
|
|
},
|
|
{
|
|
"epoch": 2.4,
|
|
"learning_rate": 1.1534184443258931e-05,
|
|
"loss": 0.4776,
|
|
"step": 95
|
|
},
|
|
{
|
|
"epoch": 2.43,
|
|
"learning_rate": 1.1356200803786916e-05,
|
|
"loss": 0.4863,
|
|
"step": 96
|
|
},
|
|
{
|
|
"epoch": 2.45,
|
|
"learning_rate": 1.1178123756827518e-05,
|
|
"loss": 0.7674,
|
|
"step": 97
|
|
},
|
|
{
|
|
"epoch": 2.48,
|
|
"learning_rate": 1.1e-05,
|
|
"loss": 1.1923,
|
|
"step": 98
|
|
},
|
|
{
|
|
"epoch": 2.5,
|
|
"learning_rate": 1.0821876243172483e-05,
|
|
"loss": 0.6922,
|
|
"step": 99
|
|
},
|
|
{
|
|
"epoch": 2.53,
|
|
"learning_rate": 1.0643799196213087e-05,
|
|
"loss": 0.6416,
|
|
"step": 100
|
|
},
|
|
{
|
|
"epoch": 2.53,
|
|
"eval_loss": 0.6664387583732605,
|
|
"eval_runtime": 27.0713,
|
|
"eval_samples_per_second": 30.586,
|
|
"eval_steps_per_second": 0.48,
|
|
"step": 100
|
|
}
|
|
],
|
|
"logging_steps": 1,
|
|
"max_steps": 195,
|
|
"num_train_epochs": 5,
|
|
"save_steps": 25,
|
|
"total_flos": 2.5513979468725617e+18,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|