1040 lines
21 KiB
JSON
1040 lines
21 KiB
JSON
{
|
|
"best_metric": null,
|
|
"best_model_checkpoint": null,
|
|
"epoch": 3.7914691943127963,
|
|
"eval_steps": 10,
|
|
"global_step": 150,
|
|
"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
|
|
},
|
|
{
|
|
"epoch": 2.55,
|
|
"learning_rate": 1.0465815556741072e-05,
|
|
"loss": 0.5438,
|
|
"step": 101
|
|
},
|
|
{
|
|
"epoch": 2.58,
|
|
"learning_rate": 1.0287971997881207e-05,
|
|
"loss": 0.4733,
|
|
"step": 102
|
|
},
|
|
{
|
|
"epoch": 2.6,
|
|
"learning_rate": 1.0110315156024555e-05,
|
|
"loss": 0.6894,
|
|
"step": 103
|
|
},
|
|
{
|
|
"epoch": 2.63,
|
|
"learning_rate": 9.932891618598871e-06,
|
|
"loss": 1.0594,
|
|
"step": 104
|
|
},
|
|
{
|
|
"epoch": 2.65,
|
|
"learning_rate": 9.755747911851873e-06,
|
|
"loss": 0.8682,
|
|
"step": 105
|
|
},
|
|
{
|
|
"epoch": 2.68,
|
|
"learning_rate": 9.578930488650532e-06,
|
|
"loss": 0.6275,
|
|
"step": 106
|
|
},
|
|
{
|
|
"epoch": 2.7,
|
|
"learning_rate": 9.40248571629961e-06,
|
|
"loss": 0.5676,
|
|
"step": 107
|
|
},
|
|
{
|
|
"epoch": 2.73,
|
|
"learning_rate": 9.226459864382642e-06,
|
|
"loss": 0.4443,
|
|
"step": 108
|
|
},
|
|
{
|
|
"epoch": 2.76,
|
|
"learning_rate": 9.05089909262855e-06,
|
|
"loss": 0.5907,
|
|
"step": 109
|
|
},
|
|
{
|
|
"epoch": 2.78,
|
|
"learning_rate": 8.875849438807032e-06,
|
|
"loss": 0.9166,
|
|
"step": 110
|
|
},
|
|
{
|
|
"epoch": 2.78,
|
|
"eval_loss": 0.6656625270843506,
|
|
"eval_runtime": 26.8682,
|
|
"eval_samples_per_second": 30.817,
|
|
"eval_steps_per_second": 0.484,
|
|
"step": 110
|
|
},
|
|
{
|
|
"epoch": 2.81,
|
|
"learning_rate": 8.701356806655973e-06,
|
|
"loss": 1.0669,
|
|
"step": 111
|
|
},
|
|
{
|
|
"epoch": 2.83,
|
|
"learning_rate": 8.52746695384398e-06,
|
|
"loss": 0.6286,
|
|
"step": 112
|
|
},
|
|
{
|
|
"epoch": 2.86,
|
|
"learning_rate": 8.354225479971213e-06,
|
|
"loss": 0.5951,
|
|
"step": 113
|
|
},
|
|
{
|
|
"epoch": 2.88,
|
|
"learning_rate": 8.181677814611703e-06,
|
|
"loss": 0.4757,
|
|
"step": 114
|
|
},
|
|
{
|
|
"epoch": 2.91,
|
|
"learning_rate": 8.009869205400227e-06,
|
|
"loss": 0.5694,
|
|
"step": 115
|
|
},
|
|
{
|
|
"epoch": 2.93,
|
|
"learning_rate": 7.83884470616689e-06,
|
|
"loss": 0.8569,
|
|
"step": 116
|
|
},
|
|
{
|
|
"epoch": 2.96,
|
|
"learning_rate": 7.668649165122545e-06,
|
|
"loss": 1.0254,
|
|
"step": 117
|
|
},
|
|
{
|
|
"epoch": 2.98,
|
|
"learning_rate": 7.49932721309812e-06,
|
|
"loss": 0.5457,
|
|
"step": 118
|
|
},
|
|
{
|
|
"epoch": 3.01,
|
|
"learning_rate": 7.330923251840976e-06,
|
|
"loss": 0.9681,
|
|
"step": 119
|
|
},
|
|
{
|
|
"epoch": 3.03,
|
|
"learning_rate": 7.163481442371295e-06,
|
|
"loss": 0.5708,
|
|
"step": 120
|
|
},
|
|
{
|
|
"epoch": 3.03,
|
|
"eval_loss": 0.6612405776977539,
|
|
"eval_runtime": 27.1115,
|
|
"eval_samples_per_second": 30.541,
|
|
"eval_steps_per_second": 0.48,
|
|
"step": 120
|
|
},
|
|
{
|
|
"epoch": 3.06,
|
|
"learning_rate": 6.997045693401637e-06,
|
|
"loss": 0.5819,
|
|
"step": 121
|
|
},
|
|
{
|
|
"epoch": 3.08,
|
|
"learning_rate": 6.831659649822625e-06,
|
|
"loss": 0.4961,
|
|
"step": 122
|
|
},
|
|
{
|
|
"epoch": 3.11,
|
|
"learning_rate": 6.667366681257843e-06,
|
|
"loss": 0.4893,
|
|
"step": 123
|
|
},
|
|
{
|
|
"epoch": 3.13,
|
|
"learning_rate": 6.504209870690885e-06,
|
|
"loss": 0.7579,
|
|
"step": 124
|
|
},
|
|
{
|
|
"epoch": 3.16,
|
|
"learning_rate": 6.342232003167612e-06,
|
|
"loss": 1.0994,
|
|
"step": 125
|
|
},
|
|
{
|
|
"epoch": 3.18,
|
|
"learning_rate": 6.181475554576494e-06,
|
|
"loss": 0.7208,
|
|
"step": 126
|
|
},
|
|
{
|
|
"epoch": 3.21,
|
|
"learning_rate": 6.021982680510073e-06,
|
|
"loss": 0.6245,
|
|
"step": 127
|
|
},
|
|
{
|
|
"epoch": 3.24,
|
|
"learning_rate": 5.863795205210353e-06,
|
|
"loss": 0.4971,
|
|
"step": 128
|
|
},
|
|
{
|
|
"epoch": 3.26,
|
|
"learning_rate": 5.7069546106011436e-06,
|
|
"loss": 0.4555,
|
|
"step": 129
|
|
},
|
|
{
|
|
"epoch": 3.29,
|
|
"learning_rate": 5.551502025410141e-06,
|
|
"loss": 0.6616,
|
|
"step": 130
|
|
},
|
|
{
|
|
"epoch": 3.29,
|
|
"eval_loss": 0.6605544686317444,
|
|
"eval_runtime": 26.9513,
|
|
"eval_samples_per_second": 30.722,
|
|
"eval_steps_per_second": 0.482,
|
|
"step": 130
|
|
},
|
|
{
|
|
"epoch": 3.31,
|
|
"learning_rate": 5.397478214383611e-06,
|
|
"loss": 0.9955,
|
|
"step": 131
|
|
},
|
|
{
|
|
"epoch": 3.34,
|
|
"learning_rate": 5.244923567596569e-06,
|
|
"loss": 0.9027,
|
|
"step": 132
|
|
},
|
|
{
|
|
"epoch": 3.36,
|
|
"learning_rate": 5.093878089861178e-06,
|
|
"loss": 0.5898,
|
|
"step": 133
|
|
},
|
|
{
|
|
"epoch": 3.39,
|
|
"learning_rate": 4.9443813902361605e-06,
|
|
"loss": 0.5661,
|
|
"step": 134
|
|
},
|
|
{
|
|
"epoch": 3.41,
|
|
"learning_rate": 4.796472671640028e-06,
|
|
"loss": 0.4364,
|
|
"step": 135
|
|
},
|
|
{
|
|
"epoch": 3.44,
|
|
"learning_rate": 4.650190720570789e-06,
|
|
"loss": 0.5682,
|
|
"step": 136
|
|
},
|
|
{
|
|
"epoch": 3.46,
|
|
"learning_rate": 4.505573896934854e-06,
|
|
"loss": 0.8235,
|
|
"step": 137
|
|
},
|
|
{
|
|
"epoch": 3.49,
|
|
"learning_rate": 4.3626601239878285e-06,
|
|
"loss": 1.1127,
|
|
"step": 138
|
|
},
|
|
{
|
|
"epoch": 3.51,
|
|
"learning_rate": 4.221486878389799e-06,
|
|
"loss": 0.6107,
|
|
"step": 139
|
|
},
|
|
{
|
|
"epoch": 3.54,
|
|
"learning_rate": 4.08209118037771e-06,
|
|
"loss": 0.5709,
|
|
"step": 140
|
|
},
|
|
{
|
|
"epoch": 3.54,
|
|
"eval_loss": 0.6593899726867676,
|
|
"eval_runtime": 27.0907,
|
|
"eval_samples_per_second": 30.564,
|
|
"eval_steps_per_second": 0.48,
|
|
"step": 140
|
|
},
|
|
{
|
|
"epoch": 3.56,
|
|
"learning_rate": 3.944509584057463e-06,
|
|
"loss": 0.4484,
|
|
"step": 141
|
|
},
|
|
{
|
|
"epoch": 3.59,
|
|
"learning_rate": 3.808778167818225e-06,
|
|
"loss": 0.4906,
|
|
"step": 142
|
|
},
|
|
{
|
|
"epoch": 3.61,
|
|
"learning_rate": 3.67493252487151e-06,
|
|
"loss": 0.8204,
|
|
"step": 143
|
|
},
|
|
{
|
|
"epoch": 3.64,
|
|
"learning_rate": 3.543007753917464e-06,
|
|
"loss": 1.1114,
|
|
"step": 144
|
|
},
|
|
{
|
|
"epoch": 3.67,
|
|
"learning_rate": 3.4130384499408527e-06,
|
|
"loss": 0.6144,
|
|
"step": 145
|
|
},
|
|
{
|
|
"epoch": 3.69,
|
|
"learning_rate": 3.285058695139109e-06,
|
|
"loss": 0.6036,
|
|
"step": 146
|
|
},
|
|
{
|
|
"epoch": 3.72,
|
|
"learning_rate": 3.159102049984881e-06,
|
|
"loss": 0.4777,
|
|
"step": 147
|
|
},
|
|
{
|
|
"epoch": 3.74,
|
|
"learning_rate": 3.0352015444253654e-06,
|
|
"loss": 0.4602,
|
|
"step": 148
|
|
},
|
|
{
|
|
"epoch": 3.77,
|
|
"learning_rate": 2.9133896692207982e-06,
|
|
"loss": 0.6665,
|
|
"step": 149
|
|
},
|
|
{
|
|
"epoch": 3.79,
|
|
"learning_rate": 2.7936983674242933e-06,
|
|
"loss": 1.1254,
|
|
"step": 150
|
|
},
|
|
{
|
|
"epoch": 3.79,
|
|
"eval_loss": 0.6585595607757568,
|
|
"eval_runtime": 26.9155,
|
|
"eval_samples_per_second": 30.763,
|
|
"eval_steps_per_second": 0.483,
|
|
"step": 150
|
|
}
|
|
],
|
|
"logging_steps": 1,
|
|
"max_steps": 195,
|
|
"num_train_epochs": 5,
|
|
"save_steps": 25,
|
|
"total_flos": 3.7484357972760986e+18,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|