1056 lines
24 KiB
JSON
1056 lines
24 KiB
JSON
{
|
|
"best_metric": null,
|
|
"best_model_checkpoint": null,
|
|
"epoch": 3.0,
|
|
"eval_steps": 500,
|
|
"global_step": 711,
|
|
"is_hyper_param_search": false,
|
|
"is_local_process_zero": true,
|
|
"is_world_process_zero": true,
|
|
"log_history": [
|
|
{
|
|
"epoch": 0.0,
|
|
"grad_norm": 6.304776350824905,
|
|
"learning_rate": 2.7777777777777776e-07,
|
|
"loss": 1.2835,
|
|
"step": 1
|
|
},
|
|
{
|
|
"epoch": 0.02,
|
|
"grad_norm": 6.1444154892612906,
|
|
"learning_rate": 1.3888888888888892e-06,
|
|
"loss": 1.2052,
|
|
"step": 5
|
|
},
|
|
{
|
|
"epoch": 0.04,
|
|
"grad_norm": 3.829285889985009,
|
|
"learning_rate": 2.7777777777777783e-06,
|
|
"loss": 1.2292,
|
|
"step": 10
|
|
},
|
|
{
|
|
"epoch": 0.06,
|
|
"grad_norm": 1.1679705762984034,
|
|
"learning_rate": 4.166666666666667e-06,
|
|
"loss": 1.1597,
|
|
"step": 15
|
|
},
|
|
{
|
|
"epoch": 0.08,
|
|
"grad_norm": 1.4357027628580525,
|
|
"learning_rate": 5.555555555555557e-06,
|
|
"loss": 1.1284,
|
|
"step": 20
|
|
},
|
|
{
|
|
"epoch": 0.11,
|
|
"grad_norm": 0.9452863213714234,
|
|
"learning_rate": 6.944444444444445e-06,
|
|
"loss": 1.1531,
|
|
"step": 25
|
|
},
|
|
{
|
|
"epoch": 0.13,
|
|
"grad_norm": 0.8873033736663324,
|
|
"learning_rate": 8.333333333333334e-06,
|
|
"loss": 1.1549,
|
|
"step": 30
|
|
},
|
|
{
|
|
"epoch": 0.15,
|
|
"grad_norm": 0.7881640401882983,
|
|
"learning_rate": 9.722222222222223e-06,
|
|
"loss": 1.1306,
|
|
"step": 35
|
|
},
|
|
{
|
|
"epoch": 0.17,
|
|
"grad_norm": 0.7249086340452754,
|
|
"learning_rate": 1.1111111111111113e-05,
|
|
"loss": 1.0851,
|
|
"step": 40
|
|
},
|
|
{
|
|
"epoch": 0.19,
|
|
"grad_norm": 0.6764364880756644,
|
|
"learning_rate": 1.25e-05,
|
|
"loss": 1.139,
|
|
"step": 45
|
|
},
|
|
{
|
|
"epoch": 0.21,
|
|
"grad_norm": 0.6897544761050797,
|
|
"learning_rate": 1.388888888888889e-05,
|
|
"loss": 1.1317,
|
|
"step": 50
|
|
},
|
|
{
|
|
"epoch": 0.23,
|
|
"grad_norm": 0.6816142474216428,
|
|
"learning_rate": 1.5277777777777777e-05,
|
|
"loss": 1.1313,
|
|
"step": 55
|
|
},
|
|
{
|
|
"epoch": 0.25,
|
|
"grad_norm": 0.6811212493739357,
|
|
"learning_rate": 1.6666666666666667e-05,
|
|
"loss": 1.138,
|
|
"step": 60
|
|
},
|
|
{
|
|
"epoch": 0.27,
|
|
"grad_norm": 0.6862844341255663,
|
|
"learning_rate": 1.8055555555555558e-05,
|
|
"loss": 1.1483,
|
|
"step": 65
|
|
},
|
|
{
|
|
"epoch": 0.3,
|
|
"grad_norm": 1.4787128965985026,
|
|
"learning_rate": 1.9444444444444445e-05,
|
|
"loss": 1.1395,
|
|
"step": 70
|
|
},
|
|
{
|
|
"epoch": 0.32,
|
|
"grad_norm": 0.7775487986058592,
|
|
"learning_rate": 1.999891231617599e-05,
|
|
"loss": 1.1161,
|
|
"step": 75
|
|
},
|
|
{
|
|
"epoch": 0.34,
|
|
"grad_norm": 0.7115088691454458,
|
|
"learning_rate": 1.9992266216318037e-05,
|
|
"loss": 1.0932,
|
|
"step": 80
|
|
},
|
|
{
|
|
"epoch": 0.36,
|
|
"grad_norm": 0.6818793389006765,
|
|
"learning_rate": 1.997958229642588e-05,
|
|
"loss": 1.1178,
|
|
"step": 85
|
|
},
|
|
{
|
|
"epoch": 0.38,
|
|
"grad_norm": 0.6928106880338162,
|
|
"learning_rate": 1.996086822074945e-05,
|
|
"loss": 1.1298,
|
|
"step": 90
|
|
},
|
|
{
|
|
"epoch": 0.4,
|
|
"grad_norm": 0.6676196732940871,
|
|
"learning_rate": 1.9936135297256183e-05,
|
|
"loss": 1.1094,
|
|
"step": 95
|
|
},
|
|
{
|
|
"epoch": 0.42,
|
|
"grad_norm": 0.713377771129619,
|
|
"learning_rate": 1.9905398470798207e-05,
|
|
"loss": 1.1224,
|
|
"step": 100
|
|
},
|
|
{
|
|
"epoch": 0.44,
|
|
"grad_norm": 0.6855059849013185,
|
|
"learning_rate": 1.9868676314081907e-05,
|
|
"loss": 1.1123,
|
|
"step": 105
|
|
},
|
|
{
|
|
"epoch": 0.46,
|
|
"grad_norm": 0.660806750689474,
|
|
"learning_rate": 1.9825991016445387e-05,
|
|
"loss": 1.1044,
|
|
"step": 110
|
|
},
|
|
{
|
|
"epoch": 0.49,
|
|
"grad_norm": 0.6638994109236709,
|
|
"learning_rate": 1.9777368370450582e-05,
|
|
"loss": 1.0811,
|
|
"step": 115
|
|
},
|
|
{
|
|
"epoch": 0.51,
|
|
"grad_norm": 0.6862841124125318,
|
|
"learning_rate": 1.9722837756298112e-05,
|
|
"loss": 1.1057,
|
|
"step": 120
|
|
},
|
|
{
|
|
"epoch": 0.53,
|
|
"grad_norm": 0.6713561495513073,
|
|
"learning_rate": 1.9662432124074325e-05,
|
|
"loss": 1.1247,
|
|
"step": 125
|
|
},
|
|
{
|
|
"epoch": 0.55,
|
|
"grad_norm": 0.6865654618803174,
|
|
"learning_rate": 1.9596187973841218e-05,
|
|
"loss": 1.0959,
|
|
"step": 130
|
|
},
|
|
{
|
|
"epoch": 0.57,
|
|
"grad_norm": 0.6512909442458682,
|
|
"learning_rate": 1.9524145333581315e-05,
|
|
"loss": 1.1178,
|
|
"step": 135
|
|
},
|
|
{
|
|
"epoch": 0.59,
|
|
"grad_norm": 1.542422418836288,
|
|
"learning_rate": 1.9446347735010765e-05,
|
|
"loss": 1.0984,
|
|
"step": 140
|
|
},
|
|
{
|
|
"epoch": 0.61,
|
|
"grad_norm": 0.7206951642675766,
|
|
"learning_rate": 1.9362842187275354e-05,
|
|
"loss": 1.0812,
|
|
"step": 145
|
|
},
|
|
{
|
|
"epoch": 0.63,
|
|
"grad_norm": 0.6984462641482525,
|
|
"learning_rate": 1.9273679148545246e-05,
|
|
"loss": 1.1129,
|
|
"step": 150
|
|
},
|
|
{
|
|
"epoch": 0.65,
|
|
"grad_norm": 0.6657063460800421,
|
|
"learning_rate": 1.917891249552568e-05,
|
|
"loss": 1.1147,
|
|
"step": 155
|
|
},
|
|
{
|
|
"epoch": 0.68,
|
|
"grad_norm": 0.6732684805189473,
|
|
"learning_rate": 1.9078599490901984e-05,
|
|
"loss": 1.0988,
|
|
"step": 160
|
|
},
|
|
{
|
|
"epoch": 0.7,
|
|
"grad_norm": 0.6969660113279658,
|
|
"learning_rate": 1.897280074873868e-05,
|
|
"loss": 1.0901,
|
|
"step": 165
|
|
},
|
|
{
|
|
"epoch": 0.72,
|
|
"grad_norm": 0.6545636951842201,
|
|
"learning_rate": 1.8861580197853423e-05,
|
|
"loss": 1.0808,
|
|
"step": 170
|
|
},
|
|
{
|
|
"epoch": 0.74,
|
|
"grad_norm": 0.728879426031119,
|
|
"learning_rate": 1.8745005043188104e-05,
|
|
"loss": 1.1007,
|
|
"step": 175
|
|
},
|
|
{
|
|
"epoch": 0.76,
|
|
"grad_norm": 0.6989476047236611,
|
|
"learning_rate": 1.862314572520028e-05,
|
|
"loss": 1.0972,
|
|
"step": 180
|
|
},
|
|
{
|
|
"epoch": 0.78,
|
|
"grad_norm": 0.6937493330837292,
|
|
"learning_rate": 1.8496075877299585e-05,
|
|
"loss": 1.0751,
|
|
"step": 185
|
|
},
|
|
{
|
|
"epoch": 0.8,
|
|
"grad_norm": 0.7237715465824409,
|
|
"learning_rate": 1.8363872281354796e-05,
|
|
"loss": 1.1005,
|
|
"step": 190
|
|
},
|
|
{
|
|
"epoch": 0.82,
|
|
"grad_norm": 0.665001160804127,
|
|
"learning_rate": 1.8226614821298444e-05,
|
|
"loss": 1.0979,
|
|
"step": 195
|
|
},
|
|
{
|
|
"epoch": 0.84,
|
|
"grad_norm": 0.6512333420285555,
|
|
"learning_rate": 1.808438643485698e-05,
|
|
"loss": 1.0663,
|
|
"step": 200
|
|
},
|
|
{
|
|
"epoch": 0.86,
|
|
"grad_norm": 0.6670845393437954,
|
|
"learning_rate": 1.793727306343574e-05,
|
|
"loss": 1.0479,
|
|
"step": 205
|
|
},
|
|
{
|
|
"epoch": 0.89,
|
|
"grad_norm": 0.660996369149025,
|
|
"learning_rate": 1.7785363600188894e-05,
|
|
"loss": 1.0631,
|
|
"step": 210
|
|
},
|
|
{
|
|
"epoch": 0.91,
|
|
"grad_norm": 0.6383025721204546,
|
|
"learning_rate": 1.762874983630582e-05,
|
|
"loss": 1.0594,
|
|
"step": 215
|
|
},
|
|
{
|
|
"epoch": 0.93,
|
|
"grad_norm": 0.6481663495293073,
|
|
"learning_rate": 1.7467526405546344e-05,
|
|
"loss": 1.1117,
|
|
"step": 220
|
|
},
|
|
{
|
|
"epoch": 0.95,
|
|
"grad_norm": 0.6526574717781632,
|
|
"learning_rate": 1.7301790727058344e-05,
|
|
"loss": 1.074,
|
|
"step": 225
|
|
},
|
|
{
|
|
"epoch": 0.97,
|
|
"grad_norm": 0.6602485216423672,
|
|
"learning_rate": 1.7131642946512314e-05,
|
|
"loss": 1.1002,
|
|
"step": 230
|
|
},
|
|
{
|
|
"epoch": 0.99,
|
|
"grad_norm": 0.703340110707448,
|
|
"learning_rate": 1.6957185875588403e-05,
|
|
"loss": 1.0963,
|
|
"step": 235
|
|
},
|
|
{
|
|
"epoch": 1.0,
|
|
"eval_loss": 1.0788308382034302,
|
|
"eval_runtime": 5.3118,
|
|
"eval_samples_per_second": 277.497,
|
|
"eval_steps_per_second": 4.518,
|
|
"step": 237
|
|
},
|
|
{
|
|
"epoch": 1.01,
|
|
"grad_norm": 0.7492129279971655,
|
|
"learning_rate": 1.6778524929852513e-05,
|
|
"loss": 1.028,
|
|
"step": 240
|
|
},
|
|
{
|
|
"epoch": 1.03,
|
|
"grad_norm": 0.8102380476622988,
|
|
"learning_rate": 1.659576806505905e-05,
|
|
"loss": 0.9794,
|
|
"step": 245
|
|
},
|
|
{
|
|
"epoch": 1.05,
|
|
"grad_norm": 0.7121110613571635,
|
|
"learning_rate": 1.640902571191869e-05,
|
|
"loss": 0.9648,
|
|
"step": 250
|
|
},
|
|
{
|
|
"epoch": 1.08,
|
|
"grad_norm": 0.7426962668349942,
|
|
"learning_rate": 1.6218410709370735e-05,
|
|
"loss": 0.9826,
|
|
"step": 255
|
|
},
|
|
{
|
|
"epoch": 1.1,
|
|
"grad_norm": 0.675584007232439,
|
|
"learning_rate": 1.6024038236400246e-05,
|
|
"loss": 0.9454,
|
|
"step": 260
|
|
},
|
|
{
|
|
"epoch": 1.12,
|
|
"grad_norm": 0.7129690280611058,
|
|
"learning_rate": 1.582602574244121e-05,
|
|
"loss": 0.9403,
|
|
"step": 265
|
|
},
|
|
{
|
|
"epoch": 1.14,
|
|
"grad_norm": 0.6747033998323487,
|
|
"learning_rate": 1.562449287640781e-05,
|
|
"loss": 0.9864,
|
|
"step": 270
|
|
},
|
|
{
|
|
"epoch": 1.16,
|
|
"grad_norm": 0.6837814023465153,
|
|
"learning_rate": 1.5419561414396657e-05,
|
|
"loss": 0.9864,
|
|
"step": 275
|
|
},
|
|
{
|
|
"epoch": 1.18,
|
|
"grad_norm": 0.6969571002742431,
|
|
"learning_rate": 1.5211355186103655e-05,
|
|
"loss": 0.9567,
|
|
"step": 280
|
|
},
|
|
{
|
|
"epoch": 1.2,
|
|
"grad_norm": 0.6630210250243682,
|
|
"learning_rate": 1.5000000000000002e-05,
|
|
"loss": 0.9811,
|
|
"step": 285
|
|
},
|
|
{
|
|
"epoch": 1.22,
|
|
"grad_norm": 0.6589068053142794,
|
|
"learning_rate": 1.4785623567312492e-05,
|
|
"loss": 0.9389,
|
|
"step": 290
|
|
},
|
|
{
|
|
"epoch": 1.24,
|
|
"grad_norm": 0.6878317017661794,
|
|
"learning_rate": 1.4568355424854113e-05,
|
|
"loss": 1.0157,
|
|
"step": 295
|
|
},
|
|
{
|
|
"epoch": 1.27,
|
|
"grad_norm": 0.7179535464777137,
|
|
"learning_rate": 1.4348326856751496e-05,
|
|
"loss": 0.9787,
|
|
"step": 300
|
|
},
|
|
{
|
|
"epoch": 1.29,
|
|
"grad_norm": 0.6829129430840285,
|
|
"learning_rate": 1.412567081511659e-05,
|
|
"loss": 0.9639,
|
|
"step": 305
|
|
},
|
|
{
|
|
"epoch": 1.31,
|
|
"grad_norm": 0.7030815356191583,
|
|
"learning_rate": 1.3900521839710428e-05,
|
|
"loss": 0.9486,
|
|
"step": 310
|
|
},
|
|
{
|
|
"epoch": 1.33,
|
|
"grad_norm": 0.67559393512531,
|
|
"learning_rate": 1.367301597664757e-05,
|
|
"loss": 0.9795,
|
|
"step": 315
|
|
},
|
|
{
|
|
"epoch": 1.35,
|
|
"grad_norm": 0.6780812345565891,
|
|
"learning_rate": 1.3443290696190335e-05,
|
|
"loss": 0.9785,
|
|
"step": 320
|
|
},
|
|
{
|
|
"epoch": 1.37,
|
|
"grad_norm": 0.6710329710920228,
|
|
"learning_rate": 1.3211484809682482e-05,
|
|
"loss": 0.992,
|
|
"step": 325
|
|
},
|
|
{
|
|
"epoch": 1.39,
|
|
"grad_norm": 0.6741452885754833,
|
|
"learning_rate": 1.2977738385672558e-05,
|
|
"loss": 1.0023,
|
|
"step": 330
|
|
},
|
|
{
|
|
"epoch": 1.41,
|
|
"grad_norm": 0.6632729564225439,
|
|
"learning_rate": 1.2742192665277569e-05,
|
|
"loss": 0.995,
|
|
"step": 335
|
|
},
|
|
{
|
|
"epoch": 1.43,
|
|
"grad_norm": 0.6601370650300007,
|
|
"learning_rate": 1.250498997683813e-05,
|
|
"loss": 0.9764,
|
|
"step": 340
|
|
},
|
|
{
|
|
"epoch": 1.46,
|
|
"grad_norm": 0.6652011155656283,
|
|
"learning_rate": 1.2266273649916669e-05,
|
|
"loss": 0.9537,
|
|
"step": 345
|
|
},
|
|
{
|
|
"epoch": 1.48,
|
|
"grad_norm": 0.6446717094616893,
|
|
"learning_rate": 1.202618792869063e-05,
|
|
"loss": 0.9525,
|
|
"step": 350
|
|
},
|
|
{
|
|
"epoch": 1.5,
|
|
"grad_norm": 0.6721085026757567,
|
|
"learning_rate": 1.178487788479303e-05,
|
|
"loss": 0.9629,
|
|
"step": 355
|
|
},
|
|
{
|
|
"epoch": 1.52,
|
|
"grad_norm": 0.6629783449328003,
|
|
"learning_rate": 1.1542489329653024e-05,
|
|
"loss": 0.9842,
|
|
"step": 360
|
|
},
|
|
{
|
|
"epoch": 1.54,
|
|
"grad_norm": 0.6522335344513943,
|
|
"learning_rate": 1.129916872638945e-05,
|
|
"loss": 1.0048,
|
|
"step": 365
|
|
},
|
|
{
|
|
"epoch": 1.56,
|
|
"grad_norm": 0.6781899139396206,
|
|
"learning_rate": 1.1055063101310581e-05,
|
|
"loss": 0.9685,
|
|
"step": 370
|
|
},
|
|
{
|
|
"epoch": 1.58,
|
|
"grad_norm": 0.6691172096591039,
|
|
"learning_rate": 1.08103199550736e-05,
|
|
"loss": 0.9747,
|
|
"step": 375
|
|
},
|
|
{
|
|
"epoch": 1.6,
|
|
"grad_norm": 0.6373869560949246,
|
|
"learning_rate": 1.0565087173557396e-05,
|
|
"loss": 0.9686,
|
|
"step": 380
|
|
},
|
|
{
|
|
"epoch": 1.62,
|
|
"grad_norm": 0.6517373978992131,
|
|
"learning_rate": 1.0319512938502654e-05,
|
|
"loss": 0.9714,
|
|
"step": 385
|
|
},
|
|
{
|
|
"epoch": 1.65,
|
|
"grad_norm": 0.6580577126773488,
|
|
"learning_rate": 1.0073745637973125e-05,
|
|
"loss": 0.9747,
|
|
"step": 390
|
|
},
|
|
{
|
|
"epoch": 1.67,
|
|
"grad_norm": 0.6489932524871953,
|
|
"learning_rate": 9.827933776692236e-06,
|
|
"loss": 0.9874,
|
|
"step": 395
|
|
},
|
|
{
|
|
"epoch": 1.69,
|
|
"grad_norm": 0.6407459560571601,
|
|
"learning_rate": 9.582225886309217e-06,
|
|
"loss": 0.9618,
|
|
"step": 400
|
|
},
|
|
{
|
|
"epoch": 1.71,
|
|
"grad_norm": 0.6629610117016943,
|
|
"learning_rate": 9.336770435648963e-06,
|
|
"loss": 0.992,
|
|
"step": 405
|
|
},
|
|
{
|
|
"epoch": 1.73,
|
|
"grad_norm": 0.6667119859349727,
|
|
"learning_rate": 9.091715740999829e-06,
|
|
"loss": 0.971,
|
|
"step": 410
|
|
},
|
|
{
|
|
"epoch": 1.75,
|
|
"grad_norm": 0.6886862002448931,
|
|
"learning_rate": 8.84720987649363e-06,
|
|
"loss": 0.9556,
|
|
"step": 415
|
|
},
|
|
{
|
|
"epoch": 1.77,
|
|
"grad_norm": 0.6426160780174832,
|
|
"learning_rate": 8.60340058463194e-06,
|
|
"loss": 0.9879,
|
|
"step": 420
|
|
},
|
|
{
|
|
"epoch": 1.79,
|
|
"grad_norm": 0.6346609133677081,
|
|
"learning_rate": 8.360435187012789e-06,
|
|
"loss": 1.0061,
|
|
"step": 425
|
|
},
|
|
{
|
|
"epoch": 1.81,
|
|
"grad_norm": 0.6483257723288812,
|
|
"learning_rate": 8.118460495311687e-06,
|
|
"loss": 1.0141,
|
|
"step": 430
|
|
},
|
|
{
|
|
"epoch": 1.84,
|
|
"grad_norm": 0.7171449794590815,
|
|
"learning_rate": 7.877622722570772e-06,
|
|
"loss": 0.9675,
|
|
"step": 435
|
|
},
|
|
{
|
|
"epoch": 1.86,
|
|
"grad_norm": 0.650114145091982,
|
|
"learning_rate": 7.638067394849672e-06,
|
|
"loss": 0.9742,
|
|
"step": 440
|
|
},
|
|
{
|
|
"epoch": 1.88,
|
|
"grad_norm": 0.6929157157409924,
|
|
"learning_rate": 7.3999392632914936e-06,
|
|
"loss": 0.9872,
|
|
"step": 445
|
|
},
|
|
{
|
|
"epoch": 1.9,
|
|
"grad_norm": 0.6361399427239406,
|
|
"learning_rate": 7.163382216657033e-06,
|
|
"loss": 0.9685,
|
|
"step": 450
|
|
},
|
|
{
|
|
"epoch": 1.92,
|
|
"grad_norm": 0.6216240484506025,
|
|
"learning_rate": 6.928539194380101e-06,
|
|
"loss": 0.9648,
|
|
"step": 455
|
|
},
|
|
{
|
|
"epoch": 1.94,
|
|
"grad_norm": 0.6162980824439727,
|
|
"learning_rate": 6.6955521001964526e-06,
|
|
"loss": 0.9608,
|
|
"step": 460
|
|
},
|
|
{
|
|
"epoch": 1.96,
|
|
"grad_norm": 0.662006443614462,
|
|
"learning_rate": 6.464561716398565e-06,
|
|
"loss": 0.9555,
|
|
"step": 465
|
|
},
|
|
{
|
|
"epoch": 1.98,
|
|
"grad_norm": 0.6525504751262218,
|
|
"learning_rate": 6.2357076187680325e-06,
|
|
"loss": 0.9779,
|
|
"step": 470
|
|
},
|
|
{
|
|
"epoch": 2.0,
|
|
"eval_loss": 1.0727574825286865,
|
|
"eval_runtime": 4.9769,
|
|
"eval_samples_per_second": 296.167,
|
|
"eval_steps_per_second": 4.822,
|
|
"step": 474
|
|
},
|
|
{
|
|
"epoch": 2.0,
|
|
"grad_norm": 0.9211908456070792,
|
|
"learning_rate": 6.009128092236983e-06,
|
|
"loss": 0.9304,
|
|
"step": 475
|
|
},
|
|
{
|
|
"epoch": 2.03,
|
|
"grad_norm": 0.8532374349387694,
|
|
"learning_rate": 5.78496004732952e-06,
|
|
"loss": 0.8546,
|
|
"step": 480
|
|
},
|
|
{
|
|
"epoch": 2.05,
|
|
"grad_norm": 0.7234550887455584,
|
|
"learning_rate": 5.563338937433622e-06,
|
|
"loss": 0.8819,
|
|
"step": 485
|
|
},
|
|
{
|
|
"epoch": 2.07,
|
|
"grad_norm": 0.7126403603814161,
|
|
"learning_rate": 5.344398676953526e-06,
|
|
"loss": 0.9011,
|
|
"step": 490
|
|
},
|
|
{
|
|
"epoch": 2.09,
|
|
"grad_norm": 0.640441506166026,
|
|
"learning_rate": 5.128271560392037e-06,
|
|
"loss": 0.8714,
|
|
"step": 495
|
|
},
|
|
{
|
|
"epoch": 2.11,
|
|
"grad_norm": 0.6599554320566688,
|
|
"learning_rate": 4.915088182411674e-06,
|
|
"loss": 0.8918,
|
|
"step": 500
|
|
},
|
|
{
|
|
"epoch": 2.13,
|
|
"grad_norm": 0.6420656415982475,
|
|
"learning_rate": 4.7049773589229306e-06,
|
|
"loss": 0.8936,
|
|
"step": 505
|
|
},
|
|
{
|
|
"epoch": 2.15,
|
|
"grad_norm": 0.6624281801845274,
|
|
"learning_rate": 4.498066049247344e-06,
|
|
"loss": 0.9076,
|
|
"step": 510
|
|
},
|
|
{
|
|
"epoch": 2.17,
|
|
"grad_norm": 0.6494624453990643,
|
|
"learning_rate": 4.29447927940242e-06,
|
|
"loss": 0.874,
|
|
"step": 515
|
|
},
|
|
{
|
|
"epoch": 2.19,
|
|
"grad_norm": 0.62980438510625,
|
|
"learning_rate": 4.094340066554742e-06,
|
|
"loss": 0.8797,
|
|
"step": 520
|
|
},
|
|
{
|
|
"epoch": 2.22,
|
|
"grad_norm": 0.6496965620768959,
|
|
"learning_rate": 3.897769344686929e-06,
|
|
"loss": 0.8783,
|
|
"step": 525
|
|
},
|
|
{
|
|
"epoch": 2.24,
|
|
"grad_norm": 0.623988823063145,
|
|
"learning_rate": 3.7048858915233665e-06,
|
|
"loss": 0.8991,
|
|
"step": 530
|
|
},
|
|
{
|
|
"epoch": 2.26,
|
|
"grad_norm": 0.6235243813957997,
|
|
"learning_rate": 3.5158062567588468e-06,
|
|
"loss": 0.8587,
|
|
"step": 535
|
|
},
|
|
{
|
|
"epoch": 2.28,
|
|
"grad_norm": 0.6224524258972475,
|
|
"learning_rate": 3.330644691633492e-06,
|
|
"loss": 0.8742,
|
|
"step": 540
|
|
},
|
|
{
|
|
"epoch": 2.3,
|
|
"grad_norm": 0.6267167014433769,
|
|
"learning_rate": 3.149513079896521e-06,
|
|
"loss": 0.9013,
|
|
"step": 545
|
|
},
|
|
{
|
|
"epoch": 2.32,
|
|
"grad_norm": 0.6252594161210878,
|
|
"learning_rate": 2.9725208702005736e-06,
|
|
"loss": 0.8905,
|
|
"step": 550
|
|
},
|
|
{
|
|
"epoch": 2.34,
|
|
"grad_norm": 0.6204178434773648,
|
|
"learning_rate": 2.7997750099674282e-06,
|
|
"loss": 0.8819,
|
|
"step": 555
|
|
},
|
|
{
|
|
"epoch": 2.36,
|
|
"grad_norm": 0.6234939504584522,
|
|
"learning_rate": 2.631379880765107e-06,
|
|
"loss": 0.8844,
|
|
"step": 560
|
|
},
|
|
{
|
|
"epoch": 2.38,
|
|
"grad_norm": 0.6404757813962814,
|
|
"learning_rate": 2.467437235235378e-06,
|
|
"loss": 0.8643,
|
|
"step": 565
|
|
},
|
|
{
|
|
"epoch": 2.41,
|
|
"grad_norm": 0.6154655956232923,
|
|
"learning_rate": 2.3080461356097938e-06,
|
|
"loss": 0.8835,
|
|
"step": 570
|
|
},
|
|
{
|
|
"epoch": 2.43,
|
|
"grad_norm": 0.6226597834743854,
|
|
"learning_rate": 2.153302893851401e-06,
|
|
"loss": 0.8547,
|
|
"step": 575
|
|
},
|
|
{
|
|
"epoch": 2.45,
|
|
"grad_norm": 0.635711672458865,
|
|
"learning_rate": 2.0033010134583085e-06,
|
|
"loss": 0.8799,
|
|
"step": 580
|
|
},
|
|
{
|
|
"epoch": 2.47,
|
|
"grad_norm": 0.6222216060732768,
|
|
"learning_rate": 1.8581311329642592e-06,
|
|
"loss": 0.8614,
|
|
"step": 585
|
|
},
|
|
{
|
|
"epoch": 2.49,
|
|
"grad_norm": 0.6211538083313471,
|
|
"learning_rate": 1.7178809711703525e-06,
|
|
"loss": 0.8851,
|
|
"step": 590
|
|
},
|
|
{
|
|
"epoch": 2.51,
|
|
"grad_norm": 0.6264778347828893,
|
|
"learning_rate": 1.5826352741410333e-06,
|
|
"loss": 0.8841,
|
|
"step": 595
|
|
},
|
|
{
|
|
"epoch": 2.53,
|
|
"grad_norm": 0.6194165165737189,
|
|
"learning_rate": 1.452475763996326e-06,
|
|
"loss": 0.8519,
|
|
"step": 600
|
|
},
|
|
{
|
|
"epoch": 2.55,
|
|
"grad_norm": 0.6156200040809161,
|
|
"learning_rate": 1.3274810895313083e-06,
|
|
"loss": 0.8595,
|
|
"step": 605
|
|
},
|
|
{
|
|
"epoch": 2.57,
|
|
"grad_norm": 0.6268920763538539,
|
|
"learning_rate": 1.207726778692625e-06,
|
|
"loss": 0.8668,
|
|
"step": 610
|
|
},
|
|
{
|
|
"epoch": 2.59,
|
|
"grad_norm": 0.6237602437342175,
|
|
"learning_rate": 1.0932851929407828e-06,
|
|
"loss": 0.8984,
|
|
"step": 615
|
|
},
|
|
{
|
|
"epoch": 2.62,
|
|
"grad_norm": 0.6098041198161336,
|
|
"learning_rate": 9.842254835257792e-07,
|
|
"loss": 0.8889,
|
|
"step": 620
|
|
},
|
|
{
|
|
"epoch": 2.64,
|
|
"grad_norm": 0.6047803478738842,
|
|
"learning_rate": 8.806135497025181e-07,
|
|
"loss": 0.8712,
|
|
"step": 625
|
|
},
|
|
{
|
|
"epoch": 2.66,
|
|
"grad_norm": 0.6161378090035984,
|
|
"learning_rate": 7.825119989112173e-07,
|
|
"loss": 0.8885,
|
|
"step": 630
|
|
},
|
|
{
|
|
"epoch": 2.68,
|
|
"grad_norm": 0.6233431696221546,
|
|
"learning_rate": 6.899801089469205e-07,
|
|
"loss": 0.8852,
|
|
"step": 635
|
|
},
|
|
{
|
|
"epoch": 2.7,
|
|
"grad_norm": 0.6391554220139581,
|
|
"learning_rate": 6.030737921409169e-07,
|
|
"loss": 0.8773,
|
|
"step": 640
|
|
},
|
|
{
|
|
"epoch": 2.72,
|
|
"grad_norm": 0.6245630190412572,
|
|
"learning_rate": 5.218455615757601e-07,
|
|
"loss": 0.8805,
|
|
"step": 645
|
|
},
|
|
{
|
|
"epoch": 2.74,
|
|
"grad_norm": 0.6100529800808214,
|
|
"learning_rate": 4.4634449935427203e-07,
|
|
"loss": 0.8605,
|
|
"step": 650
|
|
},
|
|
{
|
|
"epoch": 2.76,
|
|
"grad_norm": 0.6121972838225812,
|
|
"learning_rate": 3.7661622694171393e-07,
|
|
"loss": 0.9044,
|
|
"step": 655
|
|
},
|
|
{
|
|
"epoch": 2.78,
|
|
"grad_norm": 0.6334525599084514,
|
|
"learning_rate": 3.127028775990515e-07,
|
|
"loss": 0.8752,
|
|
"step": 660
|
|
},
|
|
{
|
|
"epoch": 2.81,
|
|
"grad_norm": 0.6246892070368814,
|
|
"learning_rate": 2.546430709239578e-07,
|
|
"loss": 0.873,
|
|
"step": 665
|
|
},
|
|
{
|
|
"epoch": 2.83,
|
|
"grad_norm": 0.6189951160404621,
|
|
"learning_rate": 2.02471889514948e-07,
|
|
"loss": 0.8616,
|
|
"step": 670
|
|
},
|
|
{
|
|
"epoch": 2.85,
|
|
"grad_norm": 0.612318312224187,
|
|
"learning_rate": 1.562208577727442e-07,
|
|
"loss": 0.8419,
|
|
"step": 675
|
|
},
|
|
{
|
|
"epoch": 2.87,
|
|
"grad_norm": 0.597268698517341,
|
|
"learning_rate": 1.1591792285167603e-07,
|
|
"loss": 0.863,
|
|
"step": 680
|
|
},
|
|
{
|
|
"epoch": 2.89,
|
|
"grad_norm": 0.6176692124122442,
|
|
"learning_rate": 8.158743777263334e-08,
|
|
"loss": 0.8801,
|
|
"step": 685
|
|
},
|
|
{
|
|
"epoch": 2.91,
|
|
"grad_norm": 0.6020045294431875,
|
|
"learning_rate": 5.325014670776951e-08,
|
|
"loss": 0.8632,
|
|
"step": 690
|
|
},
|
|
{
|
|
"epoch": 2.93,
|
|
"grad_norm": 0.6387655106971034,
|
|
"learning_rate": 3.092317244584919e-08,
|
|
"loss": 0.9029,
|
|
"step": 695
|
|
},
|
|
{
|
|
"epoch": 2.95,
|
|
"grad_norm": 0.6272250942520349,
|
|
"learning_rate": 1.4620006045816814e-08,
|
|
"loss": 0.9094,
|
|
"step": 700
|
|
},
|
|
{
|
|
"epoch": 2.97,
|
|
"grad_norm": 0.6245247048439003,
|
|
"learning_rate": 4.3504986848297295e-09,
|
|
"loss": 0.8741,
|
|
"step": 705
|
|
},
|
|
{
|
|
"epoch": 3.0,
|
|
"grad_norm": 0.6085774534918146,
|
|
"learning_rate": 1.2085570569642102e-10,
|
|
"loss": 0.8734,
|
|
"step": 710
|
|
},
|
|
{
|
|
"epoch": 3.0,
|
|
"eval_loss": 1.0886437892913818,
|
|
"eval_runtime": 5.0486,
|
|
"eval_samples_per_second": 291.961,
|
|
"eval_steps_per_second": 4.754,
|
|
"step": 711
|
|
},
|
|
{
|
|
"epoch": 3.0,
|
|
"step": 711,
|
|
"total_flos": 277125248581632.0,
|
|
"train_loss": 0.9885472514458347,
|
|
"train_runtime": 1164.5291,
|
|
"train_samples_per_second": 77.885,
|
|
"train_steps_per_second": 0.611
|
|
}
|
|
],
|
|
"logging_steps": 5,
|
|
"max_steps": 711,
|
|
"num_input_tokens_seen": 0,
|
|
"num_train_epochs": 3,
|
|
"save_steps": 100,
|
|
"total_flos": 277125248581632.0,
|
|
"train_batch_size": 16,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|