Rose-Kimiko-20B / checkpoint-108 /trainer_state.json
nRuaif
Upload folder using huggingface_hub
913db72
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.0334928229665072,
"eval_steps": 500,
"global_step": 108,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.01,
"learning_rate": 2e-05,
"loss": 2.3634,
"step": 1
},
{
"epoch": 0.02,
"learning_rate": 4e-05,
"loss": 2.4605,
"step": 2
},
{
"epoch": 0.03,
"learning_rate": 6e-05,
"loss": 2.3783,
"step": 3
},
{
"epoch": 0.04,
"learning_rate": 8e-05,
"loss": 2.5611,
"step": 4
},
{
"epoch": 0.05,
"learning_rate": 0.0001,
"loss": 2.525,
"step": 5
},
{
"epoch": 0.06,
"learning_rate": 0.00012,
"loss": 2.4104,
"step": 6
},
{
"epoch": 0.07,
"learning_rate": 0.00014,
"loss": 2.3589,
"step": 7
},
{
"epoch": 0.08,
"learning_rate": 0.00016,
"loss": 2.4133,
"step": 8
},
{
"epoch": 0.09,
"learning_rate": 0.00018,
"loss": 2.1582,
"step": 9
},
{
"epoch": 0.1,
"learning_rate": 0.0002,
"loss": 2.1394,
"step": 10
},
{
"epoch": 0.11,
"learning_rate": 0.00019998741276738754,
"loss": 2.1466,
"step": 11
},
{
"epoch": 0.11,
"learning_rate": 0.00019994965423831854,
"loss": 2.0672,
"step": 12
},
{
"epoch": 0.12,
"learning_rate": 0.0001998867339183008,
"loss": 2.1323,
"step": 13
},
{
"epoch": 0.13,
"learning_rate": 0.00019979866764718843,
"loss": 2.0887,
"step": 14
},
{
"epoch": 0.14,
"learning_rate": 0.00019968547759519425,
"loss": 2.1057,
"step": 15
},
{
"epoch": 0.15,
"learning_rate": 0.00019954719225730847,
"loss": 2.0863,
"step": 16
},
{
"epoch": 0.16,
"learning_rate": 0.00019938384644612543,
"loss": 2.09,
"step": 17
},
{
"epoch": 0.17,
"learning_rate": 0.00019919548128307954,
"loss": 2.0731,
"step": 18
},
{
"epoch": 0.18,
"learning_rate": 0.0001989821441880933,
"loss": 2.1503,
"step": 19
},
{
"epoch": 0.19,
"learning_rate": 0.00019874388886763944,
"loss": 1.9755,
"step": 20
},
{
"epoch": 0.2,
"learning_rate": 0.00019848077530122083,
"loss": 2.117,
"step": 21
},
{
"epoch": 0.21,
"learning_rate": 0.00019819286972627066,
"loss": 1.9935,
"step": 22
},
{
"epoch": 0.22,
"learning_rate": 0.00019788024462147788,
"loss": 2.0685,
"step": 23
},
{
"epoch": 0.23,
"learning_rate": 0.00019754297868854073,
"loss": 2.0086,
"step": 24
},
{
"epoch": 0.24,
"learning_rate": 0.00019718115683235417,
"loss": 2.0013,
"step": 25
},
{
"epoch": 0.25,
"learning_rate": 0.00019679487013963564,
"loss": 2.0066,
"step": 26
},
{
"epoch": 0.26,
"learning_rate": 0.00019638421585599423,
"loss": 1.9823,
"step": 27
},
{
"epoch": 0.27,
"learning_rate": 0.00019594929736144976,
"loss": 2.0394,
"step": 28
},
{
"epoch": 0.28,
"learning_rate": 0.0001954902241444074,
"loss": 2.0804,
"step": 29
},
{
"epoch": 0.29,
"learning_rate": 0.00019500711177409454,
"loss": 1.997,
"step": 30
},
{
"epoch": 0.3,
"learning_rate": 0.00019450008187146684,
"loss": 2.0794,
"step": 31
},
{
"epoch": 0.31,
"learning_rate": 0.00019396926207859084,
"loss": 1.9769,
"step": 32
},
{
"epoch": 0.32,
"learning_rate": 0.00019341478602651069,
"loss": 1.9643,
"step": 33
},
{
"epoch": 0.33,
"learning_rate": 0.00019283679330160726,
"loss": 2.041,
"step": 34
},
{
"epoch": 0.33,
"learning_rate": 0.00019223542941045817,
"loss": 1.9873,
"step": 35
},
{
"epoch": 0.34,
"learning_rate": 0.00019161084574320696,
"loss": 2.0358,
"step": 36
},
{
"epoch": 0.35,
"learning_rate": 0.00019096319953545185,
"loss": 2.021,
"step": 37
},
{
"epoch": 0.36,
"learning_rate": 0.00019029265382866214,
"loss": 2.0764,
"step": 38
},
{
"epoch": 0.37,
"learning_rate": 0.00018959937742913359,
"loss": 2.1639,
"step": 39
},
{
"epoch": 0.38,
"learning_rate": 0.00018888354486549237,
"loss": 2.0892,
"step": 40
},
{
"epoch": 0.39,
"learning_rate": 0.00018814533634475822,
"loss": 1.9749,
"step": 41
},
{
"epoch": 0.4,
"learning_rate": 0.00018738493770697852,
"loss": 2.0521,
"step": 42
},
{
"epoch": 0.41,
"learning_rate": 0.00018660254037844388,
"loss": 2.0751,
"step": 43
},
{
"epoch": 0.42,
"learning_rate": 0.00018579834132349772,
"loss": 2.0852,
"step": 44
},
{
"epoch": 0.43,
"learning_rate": 0.00018497254299495146,
"loss": 1.9871,
"step": 45
},
{
"epoch": 0.44,
"learning_rate": 0.00018412535328311814,
"loss": 1.9199,
"step": 46
},
{
"epoch": 0.45,
"learning_rate": 0.00018325698546347715,
"loss": 1.9627,
"step": 47
},
{
"epoch": 0.46,
"learning_rate": 0.0001823676581429833,
"loss": 2.0465,
"step": 48
},
{
"epoch": 0.47,
"learning_rate": 0.00018145759520503358,
"loss": 2.0033,
"step": 49
},
{
"epoch": 0.48,
"learning_rate": 0.00018052702575310588,
"loss": 1.8835,
"step": 50
},
{
"epoch": 0.49,
"learning_rate": 0.00017957618405308324,
"loss": 1.9823,
"step": 51
},
{
"epoch": 0.5,
"learning_rate": 0.00017860530947427875,
"loss": 2.0887,
"step": 52
},
{
"epoch": 0.51,
"learning_rate": 0.0001776146464291757,
"loss": 2.0936,
"step": 53
},
{
"epoch": 0.52,
"learning_rate": 0.0001766044443118978,
"loss": 2.0576,
"step": 54
},
{
"epoch": 0.53,
"learning_rate": 0.00017557495743542585,
"loss": 1.8788,
"step": 55
},
{
"epoch": 0.54,
"learning_rate": 0.0001745264449675755,
"loss": 2.0452,
"step": 56
},
{
"epoch": 0.55,
"learning_rate": 0.00017345917086575332,
"loss": 2.0411,
"step": 57
},
{
"epoch": 0.56,
"learning_rate": 0.00017237340381050703,
"loss": 2.0834,
"step": 58
},
{
"epoch": 0.56,
"learning_rate": 0.00017126941713788632,
"loss": 2.113,
"step": 59
},
{
"epoch": 0.57,
"learning_rate": 0.00017014748877063214,
"loss": 2.0797,
"step": 60
},
{
"epoch": 0.58,
"learning_rate": 0.00016900790114821122,
"loss": 2.0783,
"step": 61
},
{
"epoch": 0.59,
"learning_rate": 0.00016785094115571322,
"loss": 1.9576,
"step": 62
},
{
"epoch": 0.6,
"learning_rate": 0.00016667690005162916,
"loss": 1.9651,
"step": 63
},
{
"epoch": 0.61,
"learning_rate": 0.00016548607339452853,
"loss": 1.9927,
"step": 64
},
{
"epoch": 0.62,
"learning_rate": 0.00016427876096865394,
"loss": 2.0967,
"step": 65
},
{
"epoch": 0.63,
"learning_rate": 0.00016305526670845226,
"loss": 2.0661,
"step": 66
},
{
"epoch": 0.64,
"learning_rate": 0.00016181589862206052,
"loss": 1.9413,
"step": 67
},
{
"epoch": 0.65,
"learning_rate": 0.00016056096871376667,
"loss": 1.8223,
"step": 68
},
{
"epoch": 0.66,
"learning_rate": 0.00015929079290546408,
"loss": 2.0162,
"step": 69
},
{
"epoch": 0.67,
"learning_rate": 0.00015800569095711982,
"loss": 1.9682,
"step": 70
},
{
"epoch": 0.68,
"learning_rate": 0.00015670598638627706,
"loss": 1.8836,
"step": 71
},
{
"epoch": 0.69,
"learning_rate": 0.00015539200638661104,
"loss": 1.9973,
"step": 72
},
{
"epoch": 0.7,
"learning_rate": 0.00015406408174555976,
"loss": 2.1492,
"step": 73
},
{
"epoch": 0.71,
"learning_rate": 0.00015272254676105025,
"loss": 1.9856,
"step": 74
},
{
"epoch": 0.72,
"learning_rate": 0.00015136773915734066,
"loss": 1.9407,
"step": 75
},
{
"epoch": 0.73,
"learning_rate": 0.00015000000000000001,
"loss": 2.0559,
"step": 76
},
{
"epoch": 0.74,
"learning_rate": 0.00014861967361004687,
"loss": 1.9553,
"step": 77
},
{
"epoch": 0.75,
"learning_rate": 0.0001472271074772683,
"loss": 1.996,
"step": 78
},
{
"epoch": 0.76,
"learning_rate": 0.00014582265217274104,
"loss": 2.2306,
"step": 79
},
{
"epoch": 0.77,
"learning_rate": 0.00014440666126057744,
"loss": 2.0002,
"step": 80
},
{
"epoch": 0.78,
"learning_rate": 0.00014297949120891718,
"loss": 2.068,
"step": 81
},
{
"epoch": 0.78,
"learning_rate": 0.00014154150130018866,
"loss": 2.0615,
"step": 82
},
{
"epoch": 0.79,
"learning_rate": 0.00014009305354066137,
"loss": 2.0083,
"step": 83
},
{
"epoch": 0.8,
"learning_rate": 0.00013863451256931287,
"loss": 1.9979,
"step": 84
},
{
"epoch": 0.81,
"learning_rate": 0.00013716624556603274,
"loss": 1.8525,
"step": 85
},
{
"epoch": 0.82,
"learning_rate": 0.00013568862215918717,
"loss": 2.045,
"step": 86
},
{
"epoch": 0.83,
"learning_rate": 0.00013420201433256689,
"loss": 1.9144,
"step": 87
},
{
"epoch": 0.84,
"learning_rate": 0.00013270679633174218,
"loss": 1.9353,
"step": 88
},
{
"epoch": 0.85,
"learning_rate": 0.0001312033445698487,
"loss": 2.1257,
"step": 89
},
{
"epoch": 0.86,
"learning_rate": 0.0001296920375328275,
"loss": 1.906,
"step": 90
},
{
"epoch": 0.87,
"learning_rate": 0.00012817325568414297,
"loss": 1.9963,
"step": 91
},
{
"epoch": 0.88,
"learning_rate": 0.00012664738136900348,
"loss": 2.0734,
"step": 92
},
{
"epoch": 0.89,
"learning_rate": 0.0001251147987181079,
"loss": 2.0182,
"step": 93
},
{
"epoch": 0.9,
"learning_rate": 0.00012357589355094275,
"loss": 1.9574,
"step": 94
},
{
"epoch": 0.91,
"learning_rate": 0.00012203105327865407,
"loss": 2.1562,
"step": 95
},
{
"epoch": 0.92,
"learning_rate": 0.00012048066680651908,
"loss": 2.082,
"step": 96
},
{
"epoch": 0.93,
"learning_rate": 0.00011892512443604102,
"loss": 1.9546,
"step": 97
},
{
"epoch": 0.94,
"learning_rate": 0.00011736481776669306,
"loss": 2.0957,
"step": 98
},
{
"epoch": 0.95,
"learning_rate": 0.000115800139597335,
"loss": 1.9348,
"step": 99
},
{
"epoch": 0.96,
"learning_rate": 0.00011423148382732853,
"loss": 2.0772,
"step": 100
},
{
"epoch": 0.97,
"learning_rate": 0.00011265924535737493,
"loss": 2.0298,
"step": 101
},
{
"epoch": 0.98,
"learning_rate": 0.00011108381999010111,
"loss": 2.0272,
"step": 102
},
{
"epoch": 0.99,
"learning_rate": 0.00010950560433041826,
"loss": 2.0293,
"step": 103
},
{
"epoch": 1.0,
"learning_rate": 0.00010792499568567884,
"loss": 1.8576,
"step": 104
},
{
"epoch": 1.0,
"learning_rate": 0.00010634239196565646,
"loss": 1.9792,
"step": 105
},
{
"epoch": 1.01,
"learning_rate": 0.00010475819158237425,
"loss": 1.8629,
"step": 106
},
{
"epoch": 1.02,
"learning_rate": 0.00010317279334980678,
"loss": 2.0062,
"step": 107
},
{
"epoch": 1.03,
"learning_rate": 0.00010158659638348081,
"loss": 1.937,
"step": 108
}
],
"logging_steps": 1,
"max_steps": 208,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"save_steps": 500,
"total_flos": 4.245474393540526e+17,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}