1315 lines
29 KiB
JSON
1315 lines
29 KiB
JSON
{
|
|
"best_metric": null,
|
|
"best_model_checkpoint": null,
|
|
"epoch": 2.994991652754591,
|
|
"eval_steps": 500,
|
|
"global_step": 897,
|
|
"is_hyper_param_search": false,
|
|
"is_local_process_zero": true,
|
|
"is_world_process_zero": true,
|
|
"log_history": [
|
|
{
|
|
"epoch": 0.0,
|
|
"grad_norm": 1554.4859195078177,
|
|
"learning_rate": 2.2222222222222224e-07,
|
|
"loss": 13.9835,
|
|
"step": 1
|
|
},
|
|
{
|
|
"epoch": 0.02,
|
|
"grad_norm": 1214.7457310015889,
|
|
"learning_rate": 1.111111111111111e-06,
|
|
"loss": 13.3423,
|
|
"step": 5
|
|
},
|
|
{
|
|
"epoch": 0.03,
|
|
"grad_norm": 319.25263582700893,
|
|
"learning_rate": 2.222222222222222e-06,
|
|
"loss": 7.1598,
|
|
"step": 10
|
|
},
|
|
{
|
|
"epoch": 0.05,
|
|
"grad_norm": 101.74884085779856,
|
|
"learning_rate": 3.3333333333333333e-06,
|
|
"loss": 3.257,
|
|
"step": 15
|
|
},
|
|
{
|
|
"epoch": 0.07,
|
|
"grad_norm": 39.750398897218815,
|
|
"learning_rate": 4.444444444444444e-06,
|
|
"loss": 2.3046,
|
|
"step": 20
|
|
},
|
|
{
|
|
"epoch": 0.08,
|
|
"grad_norm": 39.73415841304385,
|
|
"learning_rate": 5.555555555555557e-06,
|
|
"loss": 2.0193,
|
|
"step": 25
|
|
},
|
|
{
|
|
"epoch": 0.1,
|
|
"grad_norm": 22.603785465724318,
|
|
"learning_rate": 6.666666666666667e-06,
|
|
"loss": 1.803,
|
|
"step": 30
|
|
},
|
|
{
|
|
"epoch": 0.12,
|
|
"grad_norm": 13.57277760359388,
|
|
"learning_rate": 7.77777777777778e-06,
|
|
"loss": 1.6723,
|
|
"step": 35
|
|
},
|
|
{
|
|
"epoch": 0.13,
|
|
"grad_norm": 10.745675701951754,
|
|
"learning_rate": 8.888888888888888e-06,
|
|
"loss": 1.6164,
|
|
"step": 40
|
|
},
|
|
{
|
|
"epoch": 0.15,
|
|
"grad_norm": 11.48936807248645,
|
|
"learning_rate": 1e-05,
|
|
"loss": 1.5651,
|
|
"step": 45
|
|
},
|
|
{
|
|
"epoch": 0.17,
|
|
"grad_norm": 6.759355981336981,
|
|
"learning_rate": 1.1111111111111113e-05,
|
|
"loss": 1.5031,
|
|
"step": 50
|
|
},
|
|
{
|
|
"epoch": 0.18,
|
|
"grad_norm": 9.542982258251046,
|
|
"learning_rate": 1.2222222222222224e-05,
|
|
"loss": 1.4295,
|
|
"step": 55
|
|
},
|
|
{
|
|
"epoch": 0.2,
|
|
"grad_norm": 7.960957372064927,
|
|
"learning_rate": 1.3333333333333333e-05,
|
|
"loss": 1.3752,
|
|
"step": 60
|
|
},
|
|
{
|
|
"epoch": 0.22,
|
|
"grad_norm": 65.24876161518675,
|
|
"learning_rate": 1.4444444444444446e-05,
|
|
"loss": 1.3768,
|
|
"step": 65
|
|
},
|
|
{
|
|
"epoch": 0.23,
|
|
"grad_norm": 127.017819160354,
|
|
"learning_rate": 1.555555555555556e-05,
|
|
"loss": 1.3837,
|
|
"step": 70
|
|
},
|
|
{
|
|
"epoch": 0.25,
|
|
"grad_norm": 66.80311438981465,
|
|
"learning_rate": 1.6666666666666667e-05,
|
|
"loss": 1.2844,
|
|
"step": 75
|
|
},
|
|
{
|
|
"epoch": 0.27,
|
|
"grad_norm": 45.06752723588582,
|
|
"learning_rate": 1.7777777777777777e-05,
|
|
"loss": 1.1568,
|
|
"step": 80
|
|
},
|
|
{
|
|
"epoch": 0.28,
|
|
"grad_norm": 16.607018601484466,
|
|
"learning_rate": 1.888888888888889e-05,
|
|
"loss": 1.1208,
|
|
"step": 85
|
|
},
|
|
{
|
|
"epoch": 0.3,
|
|
"grad_norm": 6.20855793840101,
|
|
"learning_rate": 2e-05,
|
|
"loss": 1.0982,
|
|
"step": 90
|
|
},
|
|
{
|
|
"epoch": 0.32,
|
|
"grad_norm": 5.970187327270322,
|
|
"learning_rate": 1.9998105699049984e-05,
|
|
"loss": 1.087,
|
|
"step": 95
|
|
},
|
|
{
|
|
"epoch": 0.33,
|
|
"grad_norm": 9.207660040111582,
|
|
"learning_rate": 1.9992423513875158e-05,
|
|
"loss": 1.0611,
|
|
"step": 100
|
|
},
|
|
{
|
|
"epoch": 0.35,
|
|
"grad_norm": 11.032251244972851,
|
|
"learning_rate": 1.9982955597229275e-05,
|
|
"loss": 1.055,
|
|
"step": 105
|
|
},
|
|
{
|
|
"epoch": 0.37,
|
|
"grad_norm": 7.70794959419036,
|
|
"learning_rate": 1.9969705536129033e-05,
|
|
"loss": 1.0494,
|
|
"step": 110
|
|
},
|
|
{
|
|
"epoch": 0.38,
|
|
"grad_norm": 5.115775243109966,
|
|
"learning_rate": 1.9952678350495104e-05,
|
|
"loss": 1.03,
|
|
"step": 115
|
|
},
|
|
{
|
|
"epoch": 0.4,
|
|
"grad_norm": 6.185703416946629,
|
|
"learning_rate": 1.9931880491250263e-05,
|
|
"loss": 1.0341,
|
|
"step": 120
|
|
},
|
|
{
|
|
"epoch": 0.42,
|
|
"grad_norm": 5.719162959123087,
|
|
"learning_rate": 1.990731983787542e-05,
|
|
"loss": 1.0433,
|
|
"step": 125
|
|
},
|
|
{
|
|
"epoch": 0.43,
|
|
"grad_norm": 12.405446493626323,
|
|
"learning_rate": 1.987900569542438e-05,
|
|
"loss": 1.0496,
|
|
"step": 130
|
|
},
|
|
{
|
|
"epoch": 0.45,
|
|
"grad_norm": 13.49655125076312,
|
|
"learning_rate": 1.9846948790998532e-05,
|
|
"loss": 1.0155,
|
|
"step": 135
|
|
},
|
|
{
|
|
"epoch": 0.47,
|
|
"grad_norm": 7.942566346935867,
|
|
"learning_rate": 1.9811161269682776e-05,
|
|
"loss": 1.0249,
|
|
"step": 140
|
|
},
|
|
{
|
|
"epoch": 0.48,
|
|
"grad_norm": 4.540779868090953,
|
|
"learning_rate": 1.9771656689944238e-05,
|
|
"loss": 1.0094,
|
|
"step": 145
|
|
},
|
|
{
|
|
"epoch": 0.5,
|
|
"grad_norm": 5.71935614128467,
|
|
"learning_rate": 1.9728450018495506e-05,
|
|
"loss": 1.0363,
|
|
"step": 150
|
|
},
|
|
{
|
|
"epoch": 0.52,
|
|
"grad_norm": 8.301612111722024,
|
|
"learning_rate": 1.968155762462433e-05,
|
|
"loss": 0.9804,
|
|
"step": 155
|
|
},
|
|
{
|
|
"epoch": 0.53,
|
|
"grad_norm": 4.611128256903672,
|
|
"learning_rate": 1.9630997273991964e-05,
|
|
"loss": 0.99,
|
|
"step": 160
|
|
},
|
|
{
|
|
"epoch": 0.55,
|
|
"grad_norm": 6.349166471933137,
|
|
"learning_rate": 1.9576788121902457e-05,
|
|
"loss": 0.974,
|
|
"step": 165
|
|
},
|
|
{
|
|
"epoch": 0.57,
|
|
"grad_norm": 4.709847602929964,
|
|
"learning_rate": 1.951895070604547e-05,
|
|
"loss": 0.9771,
|
|
"step": 170
|
|
},
|
|
{
|
|
"epoch": 0.58,
|
|
"grad_norm": 4.952845884539213,
|
|
"learning_rate": 1.9457506938715357e-05,
|
|
"loss": 1.0052,
|
|
"step": 175
|
|
},
|
|
{
|
|
"epoch": 0.6,
|
|
"grad_norm": 5.4487385859558914,
|
|
"learning_rate": 1.9392480098509488e-05,
|
|
"loss": 0.9863,
|
|
"step": 180
|
|
},
|
|
{
|
|
"epoch": 0.62,
|
|
"grad_norm": 5.258252581584813,
|
|
"learning_rate": 1.93238948215089e-05,
|
|
"loss": 0.9766,
|
|
"step": 185
|
|
},
|
|
{
|
|
"epoch": 0.63,
|
|
"grad_norm": 4.480190590593033,
|
|
"learning_rate": 1.9251777091944665e-05,
|
|
"loss": 0.9981,
|
|
"step": 190
|
|
},
|
|
{
|
|
"epoch": 0.65,
|
|
"grad_norm": 7.433883383226472,
|
|
"learning_rate": 1.9176154232353513e-05,
|
|
"loss": 0.9985,
|
|
"step": 195
|
|
},
|
|
{
|
|
"epoch": 0.67,
|
|
"grad_norm": 4.227680033826242,
|
|
"learning_rate": 1.9097054893226395e-05,
|
|
"loss": 0.9709,
|
|
"step": 200
|
|
},
|
|
{
|
|
"epoch": 0.68,
|
|
"grad_norm": 3.5498806463564025,
|
|
"learning_rate": 1.9014509042153964e-05,
|
|
"loss": 1.0064,
|
|
"step": 205
|
|
},
|
|
{
|
|
"epoch": 0.7,
|
|
"grad_norm": 3.8662258274644032,
|
|
"learning_rate": 1.8928547952473037e-05,
|
|
"loss": 0.9557,
|
|
"step": 210
|
|
},
|
|
{
|
|
"epoch": 0.72,
|
|
"grad_norm": 6.516418186408566,
|
|
"learning_rate": 1.8839204191418386e-05,
|
|
"loss": 0.9499,
|
|
"step": 215
|
|
},
|
|
{
|
|
"epoch": 0.73,
|
|
"grad_norm": 6.638325575394558,
|
|
"learning_rate": 1.8746511607784298e-05,
|
|
"loss": 0.9832,
|
|
"step": 220
|
|
},
|
|
{
|
|
"epoch": 0.75,
|
|
"grad_norm": 7.49893501725385,
|
|
"learning_rate": 1.865050531910062e-05,
|
|
"loss": 0.9535,
|
|
"step": 225
|
|
},
|
|
{
|
|
"epoch": 0.77,
|
|
"grad_norm": 6.04469353811688,
|
|
"learning_rate": 1.855122169832813e-05,
|
|
"loss": 0.947,
|
|
"step": 230
|
|
},
|
|
{
|
|
"epoch": 0.78,
|
|
"grad_norm": 3.3984302418422536,
|
|
"learning_rate": 1.844869836007825e-05,
|
|
"loss": 0.9592,
|
|
"step": 235
|
|
},
|
|
{
|
|
"epoch": 0.8,
|
|
"grad_norm": 3.599537518194927,
|
|
"learning_rate": 1.8342974146362397e-05,
|
|
"loss": 0.9938,
|
|
"step": 240
|
|
},
|
|
{
|
|
"epoch": 0.82,
|
|
"grad_norm": 3.5743133672120875,
|
|
"learning_rate": 1.8234089111876256e-05,
|
|
"loss": 0.9375,
|
|
"step": 245
|
|
},
|
|
{
|
|
"epoch": 0.83,
|
|
"grad_norm": 4.153578666049746,
|
|
"learning_rate": 1.8122084508824692e-05,
|
|
"loss": 0.9503,
|
|
"step": 250
|
|
},
|
|
{
|
|
"epoch": 0.85,
|
|
"grad_norm": 4.358976477421789,
|
|
"learning_rate": 1.80070027712929e-05,
|
|
"loss": 0.9405,
|
|
"step": 255
|
|
},
|
|
{
|
|
"epoch": 0.87,
|
|
"grad_norm": 4.715023563708574,
|
|
"learning_rate": 1.7888887499169816e-05,
|
|
"loss": 0.9448,
|
|
"step": 260
|
|
},
|
|
{
|
|
"epoch": 0.88,
|
|
"grad_norm": 3.540308197945189,
|
|
"learning_rate": 1.7767783441629883e-05,
|
|
"loss": 0.9518,
|
|
"step": 265
|
|
},
|
|
{
|
|
"epoch": 0.9,
|
|
"grad_norm": 4.61899118162512,
|
|
"learning_rate": 1.7643736480179353e-05,
|
|
"loss": 0.9331,
|
|
"step": 270
|
|
},
|
|
{
|
|
"epoch": 0.92,
|
|
"grad_norm": 3.8933823492498987,
|
|
"learning_rate": 1.7516793611273614e-05,
|
|
"loss": 0.9402,
|
|
"step": 275
|
|
},
|
|
{
|
|
"epoch": 0.93,
|
|
"grad_norm": 4.402077716843347,
|
|
"learning_rate": 1.7387002928512093e-05,
|
|
"loss": 0.9427,
|
|
"step": 280
|
|
},
|
|
{
|
|
"epoch": 0.95,
|
|
"grad_norm": 4.439064477908216,
|
|
"learning_rate": 1.725441360441752e-05,
|
|
"loss": 0.9689,
|
|
"step": 285
|
|
},
|
|
{
|
|
"epoch": 0.97,
|
|
"grad_norm": 4.228994650655801,
|
|
"learning_rate": 1.711907587180642e-05,
|
|
"loss": 0.922,
|
|
"step": 290
|
|
},
|
|
{
|
|
"epoch": 0.98,
|
|
"grad_norm": 4.352694277486445,
|
|
"learning_rate": 1.698104100475788e-05,
|
|
"loss": 0.9482,
|
|
"step": 295
|
|
},
|
|
{
|
|
"epoch": 1.0,
|
|
"eval_loss": 0.9848138093948364,
|
|
"eval_runtime": 21.922,
|
|
"eval_samples_per_second": 87.264,
|
|
"eval_steps_per_second": 1.368,
|
|
"step": 299
|
|
},
|
|
{
|
|
"epoch": 1.0,
|
|
"grad_norm": 3.7344611256762605,
|
|
"learning_rate": 1.684036129918786e-05,
|
|
"loss": 0.9147,
|
|
"step": 300
|
|
},
|
|
{
|
|
"epoch": 1.02,
|
|
"grad_norm": 7.808442765171335,
|
|
"learning_rate": 1.6697090053036344e-05,
|
|
"loss": 0.872,
|
|
"step": 305
|
|
},
|
|
{
|
|
"epoch": 1.04,
|
|
"grad_norm": 4.265041320808819,
|
|
"learning_rate": 1.6551281546074863e-05,
|
|
"loss": 0.8276,
|
|
"step": 310
|
|
},
|
|
{
|
|
"epoch": 1.05,
|
|
"grad_norm": 11.609187861418906,
|
|
"learning_rate": 1.6402991019342073e-05,
|
|
"loss": 0.8501,
|
|
"step": 315
|
|
},
|
|
{
|
|
"epoch": 1.07,
|
|
"grad_norm": 6.4050483054707765,
|
|
"learning_rate": 1.625227465421511e-05,
|
|
"loss": 0.8722,
|
|
"step": 320
|
|
},
|
|
{
|
|
"epoch": 1.09,
|
|
"grad_norm": 4.007163655718817,
|
|
"learning_rate": 1.60991895511247e-05,
|
|
"loss": 0.8791,
|
|
"step": 325
|
|
},
|
|
{
|
|
"epoch": 1.1,
|
|
"grad_norm": 6.169584899822902,
|
|
"learning_rate": 1.5943793707922086e-05,
|
|
"loss": 0.8378,
|
|
"step": 330
|
|
},
|
|
{
|
|
"epoch": 1.12,
|
|
"grad_norm": 5.702556304345057,
|
|
"learning_rate": 1.5786145997905952e-05,
|
|
"loss": 0.8529,
|
|
"step": 335
|
|
},
|
|
{
|
|
"epoch": 1.14,
|
|
"grad_norm": 3.666829882865862,
|
|
"learning_rate": 1.5626306147517665e-05,
|
|
"loss": 0.8625,
|
|
"step": 340
|
|
},
|
|
{
|
|
"epoch": 1.15,
|
|
"grad_norm": 7.196219177199569,
|
|
"learning_rate": 1.5464334713713312e-05,
|
|
"loss": 0.846,
|
|
"step": 345
|
|
},
|
|
{
|
|
"epoch": 1.17,
|
|
"grad_norm": 4.360948957874904,
|
|
"learning_rate": 1.5300293061021084e-05,
|
|
"loss": 0.8504,
|
|
"step": 350
|
|
},
|
|
{
|
|
"epoch": 1.19,
|
|
"grad_norm": 3.985490385252694,
|
|
"learning_rate": 1.5134243338292686e-05,
|
|
"loss": 0.8722,
|
|
"step": 355
|
|
},
|
|
{
|
|
"epoch": 1.2,
|
|
"grad_norm": 4.45100120259278,
|
|
"learning_rate": 1.4966248455157622e-05,
|
|
"loss": 0.8322,
|
|
"step": 360
|
|
},
|
|
{
|
|
"epoch": 1.22,
|
|
"grad_norm": 5.266623346917125,
|
|
"learning_rate": 1.4796372058189235e-05,
|
|
"loss": 0.8284,
|
|
"step": 365
|
|
},
|
|
{
|
|
"epoch": 1.24,
|
|
"grad_norm": 3.8641643965703647,
|
|
"learning_rate": 1.4624678506791556e-05,
|
|
"loss": 0.8482,
|
|
"step": 370
|
|
},
|
|
{
|
|
"epoch": 1.25,
|
|
"grad_norm": 5.041022068055208,
|
|
"learning_rate": 1.445123284881609e-05,
|
|
"loss": 0.8301,
|
|
"step": 375
|
|
},
|
|
{
|
|
"epoch": 1.27,
|
|
"grad_norm": 3.3180915430698654,
|
|
"learning_rate": 1.4276100795917777e-05,
|
|
"loss": 0.8208,
|
|
"step": 380
|
|
},
|
|
{
|
|
"epoch": 1.29,
|
|
"grad_norm": 6.201419840904921,
|
|
"learning_rate": 1.409934869865945e-05,
|
|
"loss": 0.8571,
|
|
"step": 385
|
|
},
|
|
{
|
|
"epoch": 1.3,
|
|
"grad_norm": 7.459788747214111,
|
|
"learning_rate": 1.392104352137426e-05,
|
|
"loss": 0.8569,
|
|
"step": 390
|
|
},
|
|
{
|
|
"epoch": 1.32,
|
|
"grad_norm": 6.709652163052175,
|
|
"learning_rate": 1.3741252816795552e-05,
|
|
"loss": 0.8622,
|
|
"step": 395
|
|
},
|
|
{
|
|
"epoch": 1.34,
|
|
"grad_norm": 3.473788351536053,
|
|
"learning_rate": 1.3560044700463824e-05,
|
|
"loss": 0.8577,
|
|
"step": 400
|
|
},
|
|
{
|
|
"epoch": 1.35,
|
|
"grad_norm": 3.5872538636448805,
|
|
"learning_rate": 1.3377487824920459e-05,
|
|
"loss": 0.86,
|
|
"step": 405
|
|
},
|
|
{
|
|
"epoch": 1.37,
|
|
"grad_norm": 5.307258675746272,
|
|
"learning_rate": 1.3193651353698012e-05,
|
|
"loss": 0.8229,
|
|
"step": 410
|
|
},
|
|
{
|
|
"epoch": 1.39,
|
|
"grad_norm": 4.804376190916538,
|
|
"learning_rate": 1.30086049351169e-05,
|
|
"loss": 0.8249,
|
|
"step": 415
|
|
},
|
|
{
|
|
"epoch": 1.4,
|
|
"grad_norm": 4.476642767728941,
|
|
"learning_rate": 1.2822418675898428e-05,
|
|
"loss": 0.815,
|
|
"step": 420
|
|
},
|
|
{
|
|
"epoch": 1.42,
|
|
"grad_norm": 3.662676271025836,
|
|
"learning_rate": 1.2635163114604131e-05,
|
|
"loss": 0.8322,
|
|
"step": 425
|
|
},
|
|
{
|
|
"epoch": 1.44,
|
|
"grad_norm": 5.33760518542071,
|
|
"learning_rate": 1.2446909194911552e-05,
|
|
"loss": 0.8283,
|
|
"step": 430
|
|
},
|
|
{
|
|
"epoch": 1.45,
|
|
"grad_norm": 3.6241603881416093,
|
|
"learning_rate": 1.2257728238736468e-05,
|
|
"loss": 0.8419,
|
|
"step": 435
|
|
},
|
|
{
|
|
"epoch": 1.47,
|
|
"grad_norm": 8.345200463380893,
|
|
"learning_rate": 1.2067691919211879e-05,
|
|
"loss": 0.8431,
|
|
"step": 440
|
|
},
|
|
{
|
|
"epoch": 1.49,
|
|
"grad_norm": 5.157020256989829,
|
|
"learning_rate": 1.1876872233533909e-05,
|
|
"loss": 0.8341,
|
|
"step": 445
|
|
},
|
|
{
|
|
"epoch": 1.5,
|
|
"grad_norm": 3.22959672388336,
|
|
"learning_rate": 1.1685341475684935e-05,
|
|
"loss": 0.8238,
|
|
"step": 450
|
|
},
|
|
{
|
|
"epoch": 1.52,
|
|
"grad_norm": 4.075617092942542,
|
|
"learning_rate": 1.1493172209044259e-05,
|
|
"loss": 0.8246,
|
|
"step": 455
|
|
},
|
|
{
|
|
"epoch": 1.54,
|
|
"grad_norm": 4.130669786993403,
|
|
"learning_rate": 1.1300437238896758e-05,
|
|
"loss": 0.8587,
|
|
"step": 460
|
|
},
|
|
{
|
|
"epoch": 1.55,
|
|
"grad_norm": 3.8344057909806653,
|
|
"learning_rate": 1.1107209584849845e-05,
|
|
"loss": 0.8405,
|
|
"step": 465
|
|
},
|
|
{
|
|
"epoch": 1.57,
|
|
"grad_norm": 4.419191495305873,
|
|
"learning_rate": 1.0913562453169241e-05,
|
|
"loss": 0.8513,
|
|
"step": 470
|
|
},
|
|
{
|
|
"epoch": 1.59,
|
|
"grad_norm": 3.639847517010915,
|
|
"learning_rate": 1.0719569209044047e-05,
|
|
"loss": 0.8268,
|
|
"step": 475
|
|
},
|
|
{
|
|
"epoch": 1.6,
|
|
"grad_norm": 6.0850406807943695,
|
|
"learning_rate": 1.0525303348791599e-05,
|
|
"loss": 0.835,
|
|
"step": 480
|
|
},
|
|
{
|
|
"epoch": 1.62,
|
|
"grad_norm": 4.663234575938155,
|
|
"learning_rate": 1.0330838472012617e-05,
|
|
"loss": 0.8297,
|
|
"step": 485
|
|
},
|
|
{
|
|
"epoch": 1.64,
|
|
"grad_norm": 2.952245938431757,
|
|
"learning_rate": 1.0136248253707267e-05,
|
|
"loss": 0.8435,
|
|
"step": 490
|
|
},
|
|
{
|
|
"epoch": 1.65,
|
|
"grad_norm": 3.1679752592105435,
|
|
"learning_rate": 9.94160641636263e-06,
|
|
"loss": 0.8409,
|
|
"step": 495
|
|
},
|
|
{
|
|
"epoch": 1.67,
|
|
"grad_norm": 3.2886951326936753,
|
|
"learning_rate": 9.74698670202218e-06,
|
|
"loss": 0.811,
|
|
"step": 500
|
|
},
|
|
{
|
|
"epoch": 1.69,
|
|
"grad_norm": 3.114045037749334,
|
|
"learning_rate": 9.552462844347883e-06,
|
|
"loss": 0.847,
|
|
"step": 505
|
|
},
|
|
{
|
|
"epoch": 1.7,
|
|
"grad_norm": 3.537647563076529,
|
|
"learning_rate": 9.358108540685406e-06,
|
|
"loss": 0.8272,
|
|
"step": 510
|
|
},
|
|
{
|
|
"epoch": 1.72,
|
|
"grad_norm": 3.41342205194864,
|
|
"learning_rate": 9.163997424143167e-06,
|
|
"loss": 0.8435,
|
|
"step": 515
|
|
},
|
|
{
|
|
"epoch": 1.74,
|
|
"grad_norm": 4.382615756695022,
|
|
"learning_rate": 8.970203035695662e-06,
|
|
"loss": 0.8241,
|
|
"step": 520
|
|
},
|
|
{
|
|
"epoch": 1.75,
|
|
"grad_norm": 6.06775300616703,
|
|
"learning_rate": 8.776798796321715e-06,
|
|
"loss": 0.8328,
|
|
"step": 525
|
|
},
|
|
{
|
|
"epoch": 1.77,
|
|
"grad_norm": 6.410082659319438,
|
|
"learning_rate": 8.583857979188203e-06,
|
|
"loss": 0.832,
|
|
"step": 530
|
|
},
|
|
{
|
|
"epoch": 1.79,
|
|
"grad_norm": 5.6018541648497715,
|
|
"learning_rate": 8.391453681889772e-06,
|
|
"loss": 0.8477,
|
|
"step": 535
|
|
},
|
|
{
|
|
"epoch": 1.8,
|
|
"grad_norm": 4.110209003561718,
|
|
"learning_rate": 8.199658798755048e-06,
|
|
"loss": 0.8365,
|
|
"step": 540
|
|
},
|
|
{
|
|
"epoch": 1.82,
|
|
"grad_norm": 3.2444930006535078,
|
|
"learning_rate": 8.008545993229897e-06,
|
|
"loss": 0.8519,
|
|
"step": 545
|
|
},
|
|
{
|
|
"epoch": 1.84,
|
|
"grad_norm": 3.25930172152866,
|
|
"learning_rate": 7.818187670348133e-06,
|
|
"loss": 0.8455,
|
|
"step": 550
|
|
},
|
|
{
|
|
"epoch": 1.85,
|
|
"grad_norm": 3.1738586241269706,
|
|
"learning_rate": 7.628655949300133e-06,
|
|
"loss": 0.8451,
|
|
"step": 555
|
|
},
|
|
{
|
|
"epoch": 1.87,
|
|
"grad_norm": 3.1977554967249144,
|
|
"learning_rate": 7.440022636109742e-06,
|
|
"loss": 0.8263,
|
|
"step": 560
|
|
},
|
|
{
|
|
"epoch": 1.89,
|
|
"grad_norm": 5.738948608123997,
|
|
"learning_rate": 7.2523591964298345e-06,
|
|
"loss": 0.8246,
|
|
"step": 565
|
|
},
|
|
{
|
|
"epoch": 1.9,
|
|
"grad_norm": 3.650496523890181,
|
|
"learning_rate": 7.065736728466832e-06,
|
|
"loss": 0.821,
|
|
"step": 570
|
|
},
|
|
{
|
|
"epoch": 1.92,
|
|
"grad_norm": 3.3610021866091073,
|
|
"learning_rate": 6.880225936044402e-06,
|
|
"loss": 0.8039,
|
|
"step": 575
|
|
},
|
|
{
|
|
"epoch": 1.94,
|
|
"grad_norm": 3.4144579406106996,
|
|
"learning_rate": 6.695897101816606e-06,
|
|
"loss": 0.8081,
|
|
"step": 580
|
|
},
|
|
{
|
|
"epoch": 1.95,
|
|
"grad_norm": 4.538437452063136,
|
|
"learning_rate": 6.512820060640608e-06,
|
|
"loss": 0.8334,
|
|
"step": 585
|
|
},
|
|
{
|
|
"epoch": 1.97,
|
|
"grad_norm": 4.895649774301149,
|
|
"learning_rate": 6.331064173119008e-06,
|
|
"loss": 0.8063,
|
|
"step": 590
|
|
},
|
|
{
|
|
"epoch": 1.99,
|
|
"grad_norm": 3.373177420856697,
|
|
"learning_rate": 6.150698299321889e-06,
|
|
"loss": 0.8139,
|
|
"step": 595
|
|
},
|
|
{
|
|
"epoch": 2.0,
|
|
"eval_loss": 0.9609715938568115,
|
|
"eval_runtime": 23.563,
|
|
"eval_samples_per_second": 81.187,
|
|
"eval_steps_per_second": 1.273,
|
|
"step": 599
|
|
},
|
|
{
|
|
"epoch": 2.0,
|
|
"grad_norm": 4.9256303288824705,
|
|
"learning_rate": 5.971790772698467e-06,
|
|
"loss": 0.7915,
|
|
"step": 600
|
|
},
|
|
{
|
|
"epoch": 2.02,
|
|
"grad_norm": 4.385016578281952,
|
|
"learning_rate": 5.794409374188272e-06,
|
|
"loss": 0.7106,
|
|
"step": 605
|
|
},
|
|
{
|
|
"epoch": 2.04,
|
|
"grad_norm": 4.602412318513612,
|
|
"learning_rate": 5.61862130654165e-06,
|
|
"loss": 0.701,
|
|
"step": 610
|
|
},
|
|
{
|
|
"epoch": 2.05,
|
|
"grad_norm": 4.466214253217951,
|
|
"learning_rate": 5.444493168859304e-06,
|
|
"loss": 0.7118,
|
|
"step": 615
|
|
},
|
|
{
|
|
"epoch": 2.07,
|
|
"grad_norm": 3.3100514145049678,
|
|
"learning_rate": 5.272090931360564e-06,
|
|
"loss": 0.7004,
|
|
"step": 620
|
|
},
|
|
{
|
|
"epoch": 2.09,
|
|
"grad_norm": 2.9902764365530023,
|
|
"learning_rate": 5.101479910389888e-06,
|
|
"loss": 0.7177,
|
|
"step": 625
|
|
},
|
|
{
|
|
"epoch": 2.1,
|
|
"grad_norm": 2.960756149194766,
|
|
"learning_rate": 4.932724743671089e-06,
|
|
"loss": 0.7061,
|
|
"step": 630
|
|
},
|
|
{
|
|
"epoch": 2.12,
|
|
"grad_norm": 3.3796322175614093,
|
|
"learning_rate": 4.765889365818708e-06,
|
|
"loss": 0.7243,
|
|
"step": 635
|
|
},
|
|
{
|
|
"epoch": 2.14,
|
|
"grad_norm": 3.068706242207435,
|
|
"learning_rate": 4.601036984115684e-06,
|
|
"loss": 0.7138,
|
|
"step": 640
|
|
},
|
|
{
|
|
"epoch": 2.15,
|
|
"grad_norm": 2.949462217015265,
|
|
"learning_rate": 4.438230054566678e-06,
|
|
"loss": 0.6921,
|
|
"step": 645
|
|
},
|
|
{
|
|
"epoch": 2.17,
|
|
"grad_norm": 2.9285837942249935,
|
|
"learning_rate": 4.277530258235955e-06,
|
|
"loss": 0.7087,
|
|
"step": 650
|
|
},
|
|
{
|
|
"epoch": 2.19,
|
|
"grad_norm": 2.8241960382183517,
|
|
"learning_rate": 4.118998477878879e-06,
|
|
"loss": 0.692,
|
|
"step": 655
|
|
},
|
|
{
|
|
"epoch": 2.2,
|
|
"grad_norm": 3.754111606827016,
|
|
"learning_rate": 3.96269477487588e-06,
|
|
"loss": 0.7036,
|
|
"step": 660
|
|
},
|
|
{
|
|
"epoch": 2.22,
|
|
"grad_norm": 2.801260790695992,
|
|
"learning_rate": 3.8086783664775827e-06,
|
|
"loss": 0.703,
|
|
"step": 665
|
|
},
|
|
{
|
|
"epoch": 2.24,
|
|
"grad_norm": 3.5205014160146484,
|
|
"learning_rate": 3.657007603369728e-06,
|
|
"loss": 0.6837,
|
|
"step": 670
|
|
},
|
|
{
|
|
"epoch": 2.25,
|
|
"grad_norm": 2.957347255235954,
|
|
"learning_rate": 3.5077399475664474e-06,
|
|
"loss": 0.716,
|
|
"step": 675
|
|
},
|
|
{
|
|
"epoch": 2.27,
|
|
"grad_norm": 3.0699352110382394,
|
|
"learning_rate": 3.360931950640185e-06,
|
|
"loss": 0.7159,
|
|
"step": 680
|
|
},
|
|
{
|
|
"epoch": 2.29,
|
|
"grad_norm": 2.8837481146108077,
|
|
"learning_rate": 3.2166392322965423e-06,
|
|
"loss": 0.688,
|
|
"step": 685
|
|
},
|
|
{
|
|
"epoch": 2.3,
|
|
"grad_norm": 2.763237253733171,
|
|
"learning_rate": 3.074916459302211e-06,
|
|
"loss": 0.7158,
|
|
"step": 690
|
|
},
|
|
{
|
|
"epoch": 2.32,
|
|
"grad_norm": 3.355328029300809,
|
|
"learning_rate": 2.935817324773893e-06,
|
|
"loss": 0.7046,
|
|
"step": 695
|
|
},
|
|
{
|
|
"epoch": 2.34,
|
|
"grad_norm": 2.8914814523547587,
|
|
"learning_rate": 2.799394527836129e-06,
|
|
"loss": 0.7096,
|
|
"step": 700
|
|
},
|
|
{
|
|
"epoch": 2.35,
|
|
"grad_norm": 3.5718163510342014,
|
|
"learning_rate": 2.665699753655684e-06,
|
|
"loss": 0.6984,
|
|
"step": 705
|
|
},
|
|
{
|
|
"epoch": 2.37,
|
|
"grad_norm": 3.2223677480030157,
|
|
"learning_rate": 2.5347836538601113e-06,
|
|
"loss": 0.6906,
|
|
"step": 710
|
|
},
|
|
{
|
|
"epoch": 2.39,
|
|
"grad_norm": 2.7101631013279373,
|
|
"learning_rate": 2.406695827347848e-06,
|
|
"loss": 0.7229,
|
|
"step": 715
|
|
},
|
|
{
|
|
"epoch": 2.4,
|
|
"grad_norm": 2.9181738000190482,
|
|
"learning_rate": 2.281484801497186e-06,
|
|
"loss": 0.7161,
|
|
"step": 720
|
|
},
|
|
{
|
|
"epoch": 2.42,
|
|
"grad_norm": 2.7148342264509067,
|
|
"learning_rate": 2.1591980137811684e-06,
|
|
"loss": 0.6911,
|
|
"step": 725
|
|
},
|
|
{
|
|
"epoch": 2.44,
|
|
"grad_norm": 2.645866837308739,
|
|
"learning_rate": 2.0398817937954275e-06,
|
|
"loss": 0.6935,
|
|
"step": 730
|
|
},
|
|
{
|
|
"epoch": 2.45,
|
|
"grad_norm": 2.923721171986238,
|
|
"learning_rate": 1.923581345705736e-06,
|
|
"loss": 0.7012,
|
|
"step": 735
|
|
},
|
|
{
|
|
"epoch": 2.47,
|
|
"grad_norm": 2.5730130782483767,
|
|
"learning_rate": 1.8103407311219523e-06,
|
|
"loss": 0.7001,
|
|
"step": 740
|
|
},
|
|
{
|
|
"epoch": 2.49,
|
|
"grad_norm": 3.1294946465675535,
|
|
"learning_rate": 1.7002028524048354e-06,
|
|
"loss": 0.6849,
|
|
"step": 745
|
|
},
|
|
{
|
|
"epoch": 2.5,
|
|
"grad_norm": 2.9432875746786955,
|
|
"learning_rate": 1.5932094364120453e-06,
|
|
"loss": 0.6834,
|
|
"step": 750
|
|
},
|
|
{
|
|
"epoch": 2.52,
|
|
"grad_norm": 2.951328262763639,
|
|
"learning_rate": 1.489401018689488e-06,
|
|
"loss": 0.6893,
|
|
"step": 755
|
|
},
|
|
{
|
|
"epoch": 2.54,
|
|
"grad_norm": 3.0873225694591886,
|
|
"learning_rate": 1.3888169281140284e-06,
|
|
"loss": 0.6672,
|
|
"step": 760
|
|
},
|
|
{
|
|
"epoch": 2.55,
|
|
"grad_norm": 2.9951086396524285,
|
|
"learning_rate": 1.291495271993337e-06,
|
|
"loss": 0.7005,
|
|
"step": 765
|
|
},
|
|
{
|
|
"epoch": 2.57,
|
|
"grad_norm": 2.6961455349482057,
|
|
"learning_rate": 1.1974729216285386e-06,
|
|
"loss": 0.7047,
|
|
"step": 770
|
|
},
|
|
{
|
|
"epoch": 2.59,
|
|
"grad_norm": 2.7063424423455777,
|
|
"learning_rate": 1.1067854983451575e-06,
|
|
"loss": 0.6874,
|
|
"step": 775
|
|
},
|
|
{
|
|
"epoch": 2.6,
|
|
"grad_norm": 2.969350173180754,
|
|
"learning_rate": 1.0194673599976134e-06,
|
|
"loss": 0.6954,
|
|
"step": 780
|
|
},
|
|
{
|
|
"epoch": 2.62,
|
|
"grad_norm": 2.8241124873176804,
|
|
"learning_rate": 9.355515879523858e-07,
|
|
"loss": 0.7151,
|
|
"step": 785
|
|
},
|
|
{
|
|
"epoch": 2.64,
|
|
"grad_norm": 2.703842291084545,
|
|
"learning_rate": 8.550699745548196e-07,
|
|
"loss": 0.7051,
|
|
"step": 790
|
|
},
|
|
{
|
|
"epoch": 2.65,
|
|
"grad_norm": 2.52014475737678,
|
|
"learning_rate": 7.780530110842566e-07,
|
|
"loss": 0.7032,
|
|
"step": 795
|
|
},
|
|
{
|
|
"epoch": 2.67,
|
|
"grad_norm": 2.8940329465111967,
|
|
"learning_rate": 7.045298762021125e-07,
|
|
"loss": 0.696,
|
|
"step": 800
|
|
},
|
|
{
|
|
"epoch": 2.69,
|
|
"grad_norm": 2.715730102462268,
|
|
"learning_rate": 6.345284248972383e-07,
|
|
"loss": 0.6594,
|
|
"step": 805
|
|
},
|
|
{
|
|
"epoch": 2.7,
|
|
"grad_norm": 2.562152086335658,
|
|
"learning_rate": 5.680751779327742e-07,
|
|
"loss": 0.6882,
|
|
"step": 810
|
|
},
|
|
{
|
|
"epoch": 2.72,
|
|
"grad_norm": 2.753866693209901,
|
|
"learning_rate": 5.05195311798491e-07,
|
|
"loss": 0.6929,
|
|
"step": 815
|
|
},
|
|
{
|
|
"epoch": 2.74,
|
|
"grad_norm": 2.5848331829143225,
|
|
"learning_rate": 4.4591264917242195e-07,
|
|
"loss": 0.6914,
|
|
"step": 820
|
|
},
|
|
{
|
|
"epoch": 2.75,
|
|
"grad_norm": 2.656019725897738,
|
|
"learning_rate": 3.9024964989539227e-07,
|
|
"loss": 0.6997,
|
|
"step": 825
|
|
},
|
|
{
|
|
"epoch": 2.77,
|
|
"grad_norm": 2.6710040716803927,
|
|
"learning_rate": 3.3822740246188477e-07,
|
|
"loss": 0.7042,
|
|
"step": 830
|
|
},
|
|
{
|
|
"epoch": 2.79,
|
|
"grad_norm": 2.797959120368719,
|
|
"learning_rate": 2.8986561603044694e-07,
|
|
"loss": 0.717,
|
|
"step": 835
|
|
},
|
|
{
|
|
"epoch": 2.8,
|
|
"grad_norm": 2.6446312241054803,
|
|
"learning_rate": 2.4518261295667255e-07,
|
|
"loss": 0.6838,
|
|
"step": 840
|
|
},
|
|
{
|
|
"epoch": 2.82,
|
|
"grad_norm": 2.6673940520178836,
|
|
"learning_rate": 2.0419532185159796e-07,
|
|
"loss": 0.7003,
|
|
"step": 845
|
|
},
|
|
{
|
|
"epoch": 2.84,
|
|
"grad_norm": 2.697047820136764,
|
|
"learning_rate": 1.6691927116812002e-07,
|
|
"loss": 0.6936,
|
|
"step": 850
|
|
},
|
|
{
|
|
"epoch": 2.85,
|
|
"grad_norm": 2.5372070358560843,
|
|
"learning_rate": 1.3336858331787993e-07,
|
|
"loss": 0.6913,
|
|
"step": 855
|
|
},
|
|
{
|
|
"epoch": 2.87,
|
|
"grad_norm": 2.5976068799798187,
|
|
"learning_rate": 1.0355596932085432e-07,
|
|
"loss": 0.694,
|
|
"step": 860
|
|
},
|
|
{
|
|
"epoch": 2.89,
|
|
"grad_norm": 2.922996849182434,
|
|
"learning_rate": 7.749272398964613e-08,
|
|
"loss": 0.703,
|
|
"step": 865
|
|
},
|
|
{
|
|
"epoch": 2.9,
|
|
"grad_norm": 2.53597875366406,
|
|
"learning_rate": 5.518872165033329e-08,
|
|
"loss": 0.6936,
|
|
"step": 870
|
|
},
|
|
{
|
|
"epoch": 2.92,
|
|
"grad_norm": 2.7025680089043584,
|
|
"learning_rate": 3.6652412401478875e-08,
|
|
"loss": 0.6911,
|
|
"step": 875
|
|
},
|
|
{
|
|
"epoch": 2.94,
|
|
"grad_norm": 2.6197435490839873,
|
|
"learning_rate": 2.1890818912728706e-08,
|
|
"loss": 0.7137,
|
|
"step": 880
|
|
},
|
|
{
|
|
"epoch": 2.95,
|
|
"grad_norm": 2.6370913398207825,
|
|
"learning_rate": 1.0909533764194013e-08,
|
|
"loss": 0.7064,
|
|
"step": 885
|
|
},
|
|
{
|
|
"epoch": 2.97,
|
|
"grad_norm": 2.7175545973145687,
|
|
"learning_rate": 3.7127173276563234e-09,
|
|
"loss": 0.7116,
|
|
"step": 890
|
|
},
|
|
{
|
|
"epoch": 2.99,
|
|
"grad_norm": 2.796693677065229,
|
|
"learning_rate": 3.0309619035495675e-10,
|
|
"loss": 0.722,
|
|
"step": 895
|
|
},
|
|
{
|
|
"epoch": 2.99,
|
|
"eval_loss": 0.973169207572937,
|
|
"eval_runtime": 24.5227,
|
|
"eval_samples_per_second": 78.009,
|
|
"eval_steps_per_second": 1.223,
|
|
"step": 897
|
|
},
|
|
{
|
|
"epoch": 2.99,
|
|
"step": 897,
|
|
"total_flos": 246909348347904.0,
|
|
"train_loss": 1.0022602648772259,
|
|
"train_runtime": 5184.357,
|
|
"train_samples_per_second": 22.178,
|
|
"train_steps_per_second": 0.173
|
|
}
|
|
],
|
|
"logging_steps": 5,
|
|
"max_steps": 897,
|
|
"num_input_tokens_seen": 0,
|
|
"num_train_epochs": 3,
|
|
"save_steps": 500,
|
|
"total_flos": 246909348347904.0,
|
|
"train_batch_size": 4,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|