Hugging Face's logo

SynthCode-Project
/
rv2-qwen8b-g2_3_plan_oldcode_newcode

Safetensors
Model card Files
xet
 Community
rv2-qwen8b-g2_3_plan_oldcode_newcode / final /trainer_state.json
rqadri's picture
rqadri
Upload folder using huggingface_hub
604488c verified
Raw
History Blame Contribute Delete
96.2 kB
{
 "best_global_step": null,
 "best_metric": null,
 "best_model_checkpoint": null,
 "epoch": 0.31061914481603603,
 "eval_steps": 500,
 "global_step": 5935,
 "is_hyper_param_search": false,
 "is_local_process_zero": true,
 "is_world_process_zero": true,
 "log_history": [
 {
 "epoch": 0.0005233683990160674,
 "grad_norm": 1.1370724439620972,
 "learning_rate": 2.910112359550562e-06,
 "loss": 0.8003,
 "step": 10
 },
 {
 "epoch": 0.0010467367980321349,
 "grad_norm": 0.9877656102180481,
 "learning_rate": 3.921348314606742e-06,
 "loss": 0.7587,
 "step": 20
 },
 {
 "epoch": 0.0015701051970482022,
 "grad_norm": 0.8628589510917664,
 "learning_rate": 4.932584269662922e-06,
 "loss": 0.7476,
 "step": 30
 },
 {
 "epoch": 0.0020934735960642698,
 "grad_norm": 0.7837583422660828,
 "learning_rate": 5.943820224719102e-06,
 "loss": 0.7334,
 "step": 40
 },
 {
 "epoch": 0.002616841995080337,
 "grad_norm": 0.7342932224273682,
 "learning_rate": 6.955056179775282e-06,
 "loss": 0.7397,
 "step": 50
 },
 {
 "epoch": 0.0031402103940964044,
 "grad_norm": 0.9271304607391357,
 "learning_rate": 7.966292134831462e-06,
 "loss": 0.7296,
 "step": 60
 },
 {
 "epoch": 0.0036635787931124718,
 "grad_norm": 0.8066033720970154,
 "learning_rate": 8.977528089887641e-06,
 "loss": 0.6864,
 "step": 70
 },
 {
 "epoch": 0.0041869471921285395,
 "grad_norm": 0.8114457726478577,
 "learning_rate": 9.988764044943822e-06,
 "loss": 0.6792,
 "step": 80
 },
 {
 "epoch": 0.004710315591144607,
 "grad_norm": 0.8880093097686768,
 "learning_rate": 1.1000000000000001e-05,
 "loss": 0.7508,
 "step": 90
 },
 {
 "epoch": 0.005233683990160674,
 "grad_norm": 1.1273611783981323,
 "learning_rate": 1.2011235955056182e-05,
 "loss": 0.6849,
 "step": 100
 },
 {
 "epoch": 0.0057570523891767415,
 "grad_norm": 0.7791317701339722,
 "learning_rate": 1.3022471910112362e-05,
 "loss": 0.698,
 "step": 110
 },
 {
 "epoch": 0.006280420788192809,
 "grad_norm": 0.8140974044799805,
 "learning_rate": 1.403370786516854e-05,
 "loss": 0.6947,
 "step": 120
 },
 {
 "epoch": 0.006803789187208876,
 "grad_norm": 0.8656443357467651,
 "learning_rate": 1.504494382022472e-05,
 "loss": 0.7158,
 "step": 130
 },
 {
 "epoch": 0.0073271575862249435,
 "grad_norm": 0.744526743888855,
 "learning_rate": 1.60561797752809e-05,
 "loss": 0.6701,
 "step": 140
 },
 {
 "epoch": 0.00785052598524101,
 "grad_norm": 0.7715036273002625,
 "learning_rate": 1.706741573033708e-05,
 "loss": 0.6794,
 "step": 150
 },
 {
 "epoch": 0.008373894384257079,
 "grad_norm": 0.7816882729530334,
 "learning_rate": 1.8078651685393256e-05,
 "loss": 0.7059,
 "step": 160
 },
 {
 "epoch": 0.008897262783273146,
 "grad_norm": 0.7964625954627991,
 "learning_rate": 1.908988764044944e-05,
 "loss": 0.6954,
 "step": 170
 },
 {
 "epoch": 0.009420631182289214,
 "grad_norm": 1.0829215049743652,
 "learning_rate": 2e-05,
 "loss": 0.6888,
 "step": 180
 },
 {
 "epoch": 0.00994399958130528,
 "grad_norm": 0.6891264319419861,
 "learning_rate": 2e-05,
 "loss": 0.6539,
 "step": 190
 },
 {
 "epoch": 0.010467367980321348,
 "grad_norm": 0.7293897867202759,
 "learning_rate": 2e-05,
 "loss": 0.7164,
 "step": 200
 },
 {
 "epoch": 0.010990736379337415,
 "grad_norm": 0.6318921446800232,
 "learning_rate": 2e-05,
 "loss": 0.7084,
 "step": 210
 },
 {
 "epoch": 0.011514104778353483,
 "grad_norm": 0.6405470371246338,
 "learning_rate": 2e-05,
 "loss": 0.6812,
 "step": 220
 },
 {
 "epoch": 0.012037473177369551,
 "grad_norm": 0.6632219552993774,
 "learning_rate": 2e-05,
 "loss": 0.6733,
 "step": 230
 },
 {
 "epoch": 0.012560841576385618,
 "grad_norm": 0.7196585536003113,
 "learning_rate": 2e-05,
 "loss": 0.6835,
 "step": 240
 },
 {
 "epoch": 0.013084209975401686,
 "grad_norm": 0.5942214131355286,
 "learning_rate": 2e-05,
 "loss": 0.6613,
 "step": 250
 },
 {
 "epoch": 0.013607578374417752,
 "grad_norm": 0.5716996788978577,
 "learning_rate": 2e-05,
 "loss": 0.6759,
 "step": 260
 },
 {
 "epoch": 0.01413094677343382,
 "grad_norm": 0.5713549256324768,
 "learning_rate": 2e-05,
 "loss": 0.7041,
 "step": 270
 },
 {
 "epoch": 0.014654315172449887,
 "grad_norm": 0.5658433437347412,
 "learning_rate": 2e-05,
 "loss": 0.6866,
 "step": 280
 },
 {
 "epoch": 0.015177683571465955,
 "grad_norm": 0.5381125211715698,
 "learning_rate": 2e-05,
 "loss": 0.7024,
 "step": 290
 },
 {
 "epoch": 0.01570105197048202,
 "grad_norm": 0.5079479217529297,
 "learning_rate": 2e-05,
 "loss": 0.686,
 "step": 300
 },
 {
 "epoch": 0.016224420369498088,
 "grad_norm": 0.5530030727386475,
 "learning_rate": 2e-05,
 "loss": 0.6912,
 "step": 310
 },
 {
 "epoch": 0.016747788768514158,
 "grad_norm": 0.5396811962127686,
 "learning_rate": 2e-05,
 "loss": 0.6751,
 "step": 320
 },
 {
 "epoch": 0.017271157167530225,
 "grad_norm": 0.4894005060195923,
 "learning_rate": 2e-05,
 "loss": 0.6649,
 "step": 330
 },
 {
 "epoch": 0.01779452556654629,
 "grad_norm": 0.5230134129524231,
 "learning_rate": 2e-05,
 "loss": 0.684,
 "step": 340
 },
 {
 "epoch": 0.01831789396556236,
 "grad_norm": 0.5332402586936951,
 "learning_rate": 2e-05,
 "loss": 0.6767,
 "step": 350
 },
 {
 "epoch": 0.018841262364578427,
 "grad_norm": 0.5048417448997498,
 "learning_rate": 2e-05,
 "loss": 0.6922,
 "step": 360
 },
 {
 "epoch": 0.019364630763594494,
 "grad_norm": 0.5345087051391602,
 "learning_rate": 2e-05,
 "loss": 0.6976,
 "step": 370
 },
 {
 "epoch": 0.01988799916261056,
 "grad_norm": 0.511880099773407,
 "learning_rate": 2e-05,
 "loss": 0.6814,
 "step": 380
 },
 {
 "epoch": 0.02041136756162663,
 "grad_norm": 0.46280574798583984,
 "learning_rate": 2e-05,
 "loss": 0.6566,
 "step": 390
 },
 {
 "epoch": 0.020934735960642697,
 "grad_norm": 0.464054137468338,
 "learning_rate": 2e-05,
 "loss": 0.6483,
 "step": 400
 },
 {
 "epoch": 0.021458104359658763,
 "grad_norm": 0.5449008941650391,
 "learning_rate": 2e-05,
 "loss": 0.6841,
 "step": 410
 },
 {
 "epoch": 0.02198147275867483,
 "grad_norm": 0.5352079272270203,
 "learning_rate": 2e-05,
 "loss": 0.6604,
 "step": 420
 },
 {
 "epoch": 0.0225048411576909,
 "grad_norm": 0.5088271498680115,
 "learning_rate": 2e-05,
 "loss": 0.6683,
 "step": 430
 },
 {
 "epoch": 0.023028209556706966,
 "grad_norm": 0.5593430399894714,
 "learning_rate": 2e-05,
 "loss": 0.6795,
 "step": 440
 },
 {
 "epoch": 0.023551577955723033,
 "grad_norm": 0.507481575012207,
 "learning_rate": 2e-05,
 "loss": 0.6654,
 "step": 450
 },
 {
 "epoch": 0.024074946354739103,
 "grad_norm": 0.4815616011619568,
 "learning_rate": 2e-05,
 "loss": 0.6843,
 "step": 460
 },
 {
 "epoch": 0.02459831475375517,
 "grad_norm": 0.5703353881835938,
 "learning_rate": 2e-05,
 "loss": 0.6598,
 "step": 470
 },
 {
 "epoch": 0.025121683152771235,
 "grad_norm": 0.48705339431762695,
 "learning_rate": 2e-05,
 "loss": 0.6614,
 "step": 480
 },
 {
 "epoch": 0.025645051551787302,
 "grad_norm": 0.547533392906189,
 "learning_rate": 2e-05,
 "loss": 0.6585,
 "step": 490
 },
 {
 "epoch": 0.026168419950803372,
 "grad_norm": 0.6569129228591919,
 "learning_rate": 2e-05,
 "loss": 0.6716,
 "step": 500
 },
 {
 "epoch": 0.02669178834981944,
 "grad_norm": 0.5733374357223511,
 "learning_rate": 2e-05,
 "loss": 0.6465,
 "step": 510
 },
 {
 "epoch": 0.027215156748835505,
 "grad_norm": 0.4885842502117157,
 "learning_rate": 2e-05,
 "loss": 0.682,
 "step": 520
 },
 {
 "epoch": 0.02773852514785157,
 "grad_norm": 0.42785435914993286,
 "learning_rate": 2e-05,
 "loss": 0.6594,
 "step": 530
 },
 {
 "epoch": 0.02826189354686764,
 "grad_norm": 0.4955022931098938,
 "learning_rate": 2e-05,
 "loss": 0.6674,
 "step": 540
 },
 {
 "epoch": 0.028785261945883708,
 "grad_norm": 0.48021453619003296,
 "learning_rate": 2e-05,
 "loss": 0.6796,
 "step": 550
 },
 {
 "epoch": 0.029308630344899774,
 "grad_norm": 0.4767313301563263,
 "learning_rate": 2e-05,
 "loss": 0.6865,
 "step": 560
 },
 {
 "epoch": 0.029831998743915844,
 "grad_norm": 0.5284311771392822,
 "learning_rate": 2e-05,
 "loss": 0.6489,
 "step": 570
 },
 {
 "epoch": 0.03035536714293191,
 "grad_norm": 0.4461924731731415,
 "learning_rate": 2e-05,
 "loss": 0.686,
 "step": 580
 },
 {
 "epoch": 0.030878735541947977,
 "grad_norm": 0.45115983486175537,
 "learning_rate": 2e-05,
 "loss": 0.6539,
 "step": 590
 },
 {
 "epoch": 0.03140210394096404,
 "grad_norm": 0.5064119100570679,
 "learning_rate": 2e-05,
 "loss": 0.6946,
 "step": 600
 },
 {
 "epoch": 0.03192547233998011,
 "grad_norm": 0.5319550037384033,
 "learning_rate": 2e-05,
 "loss": 0.645,
 "step": 610
 },
 {
 "epoch": 0.032448840738996176,
 "grad_norm": 0.4419703781604767,
 "learning_rate": 2e-05,
 "loss": 0.7003,
 "step": 620
 },
 {
 "epoch": 0.03297220913801225,
 "grad_norm": 0.42887452244758606,
 "learning_rate": 2e-05,
 "loss": 0.6783,
 "step": 630
 },
 {
 "epoch": 0.033495577537028316,
 "grad_norm": 0.46661144495010376,
 "learning_rate": 2e-05,
 "loss": 0.6686,
 "step": 640
 },
 {
 "epoch": 0.03401894593604438,
 "grad_norm": 0.5283986330032349,
 "learning_rate": 2e-05,
 "loss": 0.6679,
 "step": 650
 },
 {
 "epoch": 0.03454231433506045,
 "grad_norm": 0.5113454461097717,
 "learning_rate": 2e-05,
 "loss": 0.6506,
 "step": 660
 },
 {
 "epoch": 0.035065682734076516,
 "grad_norm": 0.4744948148727417,
 "learning_rate": 2e-05,
 "loss": 0.6724,
 "step": 670
 },
 {
 "epoch": 0.03558905113309258,
 "grad_norm": 0.49696066975593567,
 "learning_rate": 2e-05,
 "loss": 0.6652,
 "step": 680
 },
 {
 "epoch": 0.03611241953210865,
 "grad_norm": 0.7813011407852173,
 "learning_rate": 2e-05,
 "loss": 0.6621,
 "step": 690
 },
 {
 "epoch": 0.03663578793112472,
 "grad_norm": 0.49424248933792114,
 "learning_rate": 2e-05,
 "loss": 0.6727,
 "step": 700
 },
 {
 "epoch": 0.03715915633014079,
 "grad_norm": 0.4447154104709625,
 "learning_rate": 2e-05,
 "loss": 0.6659,
 "step": 710
 },
 {
 "epoch": 0.037682524729156855,
 "grad_norm": 0.44892066717147827,
 "learning_rate": 2e-05,
 "loss": 0.6554,
 "step": 720
 },
 {
 "epoch": 0.03820589312817292,
 "grad_norm": 0.49734535813331604,
 "learning_rate": 2e-05,
 "loss": 0.6779,
 "step": 730
 },
 {
 "epoch": 0.03872926152718899,
 "grad_norm": 0.5591990947723389,
 "learning_rate": 2e-05,
 "loss": 0.6927,
 "step": 740
 },
 {
 "epoch": 0.039252629926205054,
 "grad_norm": 0.45224255323410034,
 "learning_rate": 2e-05,
 "loss": 0.6398,
 "step": 750
 },
 {
 "epoch": 0.03977599832522112,
 "grad_norm": 0.41793182492256165,
 "learning_rate": 2e-05,
 "loss": 0.6668,
 "step": 760
 },
 {
 "epoch": 0.04029936672423719,
 "grad_norm": 0.43993815779685974,
 "learning_rate": 2e-05,
 "loss": 0.6456,
 "step": 770
 },
 {
 "epoch": 0.04082273512325326,
 "grad_norm": 0.44372230768203735,
 "learning_rate": 2e-05,
 "loss": 0.6669,
 "step": 780
 },
 {
 "epoch": 0.04134610352226933,
 "grad_norm": 0.4827944040298462,
 "learning_rate": 2e-05,
 "loss": 0.657,
 "step": 790
 },
 {
 "epoch": 0.041869471921285394,
 "grad_norm": 0.4187488257884979,
 "learning_rate": 2e-05,
 "loss": 0.6456,
 "step": 800
 },
 {
 "epoch": 0.04239284032030146,
 "grad_norm": 0.47213008999824524,
 "learning_rate": 2e-05,
 "loss": 0.6555,
 "step": 810
 },
 {
 "epoch": 0.042916208719317527,
 "grad_norm": 0.43557852506637573,
 "learning_rate": 2e-05,
 "loss": 0.6545,
 "step": 820
 },
 {
 "epoch": 0.04343957711833359,
 "grad_norm": 0.45871806144714355,
 "learning_rate": 2e-05,
 "loss": 0.6538,
 "step": 830
 },
 {
 "epoch": 0.04396294551734966,
 "grad_norm": 0.46028947830200195,
 "learning_rate": 2e-05,
 "loss": 0.6356,
 "step": 840
 },
 {
 "epoch": 0.04448631391636573,
 "grad_norm": 0.548116147518158,
 "learning_rate": 2e-05,
 "loss": 0.6872,
 "step": 850
 },
 {
 "epoch": 0.0450096823153818,
 "grad_norm": 0.49427151679992676,
 "learning_rate": 2e-05,
 "loss": 0.6372,
 "step": 860
 },
 {
 "epoch": 0.045533050714397866,
 "grad_norm": 0.46103739738464355,
 "learning_rate": 2e-05,
 "loss": 0.6991,
 "step": 870
 },
 {
 "epoch": 0.04605641911341393,
 "grad_norm": 0.4400480091571808,
 "learning_rate": 2e-05,
 "loss": 0.667,
 "step": 880
 },
 {
 "epoch": 0.04657978751243,
 "grad_norm": 0.45002153515815735,
 "learning_rate": 2e-05,
 "loss": 0.6582,
 "step": 890
 },
 {
 "epoch": 0.047103155911446065,
 "grad_norm": 0.44472604990005493,
 "learning_rate": 2e-05,
 "loss": 0.6706,
 "step": 900
 },
 {
 "epoch": 0.04762652431046213,
 "grad_norm": 0.4121211767196655,
 "learning_rate": 2e-05,
 "loss": 0.6602,
 "step": 910
 },
 {
 "epoch": 0.048149892709478205,
 "grad_norm": 0.5010108351707458,
 "learning_rate": 2e-05,
 "loss": 0.6761,
 "step": 920
 },
 {
 "epoch": 0.04867326110849427,
 "grad_norm": 0.45920342206954956,
 "learning_rate": 2e-05,
 "loss": 0.6671,
 "step": 930
 },
 {
 "epoch": 0.04919662950751034,
 "grad_norm": 0.5177173018455505,
 "learning_rate": 2e-05,
 "loss": 0.652,
 "step": 940
 },
 {
 "epoch": 0.049719997906526404,
 "grad_norm": 0.49086499214172363,
 "learning_rate": 2e-05,
 "loss": 0.6795,
 "step": 950
 },
 {
 "epoch": 0.05024336630554247,
 "grad_norm": 0.582841157913208,
 "learning_rate": 2e-05,
 "loss": 0.6376,
 "step": 960
 },
 {
 "epoch": 0.05076673470455854,
 "grad_norm": 0.4690440595149994,
 "learning_rate": 2e-05,
 "loss": 0.6465,
 "step": 970
 },
 {
 "epoch": 0.051290103103574604,
 "grad_norm": 0.45363402366638184,
 "learning_rate": 2e-05,
 "loss": 0.6511,
 "step": 980
 },
 {
 "epoch": 0.05181347150259067,
 "grad_norm": 0.43920376896858215,
 "learning_rate": 2e-05,
 "loss": 0.6943,
 "step": 990
 },
 {
 "epoch": 0.052336839901606744,
 "grad_norm": 0.421144038438797,
 "learning_rate": 2e-05,
 "loss": 0.6434,
 "step": 1000
 },
 {
 "epoch": 0.05286020830062281,
 "grad_norm": 0.45188188552856445,
 "learning_rate": 2e-05,
 "loss": 0.6467,
 "step": 1010
 },
 {
 "epoch": 0.05338357669963888,
 "grad_norm": 0.4339853823184967,
 "learning_rate": 2e-05,
 "loss": 0.6596,
 "step": 1020
 },
 {
 "epoch": 0.05390694509865494,
 "grad_norm": 0.4234607517719269,
 "learning_rate": 2e-05,
 "loss": 0.674,
 "step": 1030
 },
 {
 "epoch": 0.05443031349767101,
 "grad_norm": 0.46552717685699463,
 "learning_rate": 2e-05,
 "loss": 0.6649,
 "step": 1040
 },
 {
 "epoch": 0.054953681896687076,
 "grad_norm": 0.4405843913555145,
 "learning_rate": 2e-05,
 "loss": 0.6577,
 "step": 1050
 },
 {
 "epoch": 0.05547705029570314,
 "grad_norm": 0.5044958591461182,
 "learning_rate": 2e-05,
 "loss": 0.6152,
 "step": 1060
 },
 {
 "epoch": 0.056000418694719216,
 "grad_norm": 0.43265852332115173,
 "learning_rate": 2e-05,
 "loss": 0.6317,
 "step": 1070
 },
 {
 "epoch": 0.05652378709373528,
 "grad_norm": 0.41526952385902405,
 "learning_rate": 2e-05,
 "loss": 0.6501,
 "step": 1080
 },
 {
 "epoch": 0.05704715549275135,
 "grad_norm": 0.43947693705558777,
 "learning_rate": 2e-05,
 "loss": 0.6542,
 "step": 1090
 },
 {
 "epoch": 0.057570523891767415,
 "grad_norm": 0.4375130534172058,
 "learning_rate": 2e-05,
 "loss": 0.671,
 "step": 1100
 },
 {
 "epoch": 0.05809389229078348,
 "grad_norm": 0.4554004371166229,
 "learning_rate": 2e-05,
 "loss": 0.6507,
 "step": 1110
 },
 {
 "epoch": 0.05861726068979955,
 "grad_norm": 0.5064834356307983,
 "learning_rate": 2e-05,
 "loss": 0.6467,
 "step": 1120
 },
 {
 "epoch": 0.059140629088815615,
 "grad_norm": 0.446809321641922,
 "learning_rate": 2e-05,
 "loss": 0.6751,
 "step": 1130
 },
 {
 "epoch": 0.05966399748783169,
 "grad_norm": 0.4745899736881256,
 "learning_rate": 2e-05,
 "loss": 0.6564,
 "step": 1140
 },
 {
 "epoch": 0.060187365886847755,
 "grad_norm": 0.5119066834449768,
 "learning_rate": 2e-05,
 "loss": 0.6618,
 "step": 1150
 },
 {
 "epoch": 0.06071073428586382,
 "grad_norm": 0.41898113489151,
 "learning_rate": 2e-05,
 "loss": 0.6622,
 "step": 1160
 },
 {
 "epoch": 0.06123410268487989,
 "grad_norm": 0.6409226059913635,
 "learning_rate": 2e-05,
 "loss": 0.6708,
 "step": 1170
 },
 {
 "epoch": 0.061757471083895954,
 "grad_norm": 0.439377099275589,
 "learning_rate": 2e-05,
 "loss": 0.6473,
 "step": 1180
 },
 {
 "epoch": 0.06228083948291202,
 "grad_norm": 0.4160749316215515,
 "learning_rate": 2e-05,
 "loss": 0.6798,
 "step": 1190
 },
 {
 "epoch": 0.06280420788192809,
 "grad_norm": 0.5204553604125977,
 "learning_rate": 2e-05,
 "loss": 0.6866,
 "step": 1200
 },
 {
 "epoch": 0.06332757628094415,
 "grad_norm": 0.48897305130958557,
 "learning_rate": 2e-05,
 "loss": 0.6618,
 "step": 1210
 },
 {
 "epoch": 0.06385094467996022,
 "grad_norm": 0.45014750957489014,
 "learning_rate": 2e-05,
 "loss": 0.6647,
 "step": 1220
 },
 {
 "epoch": 0.06437431307897629,
 "grad_norm": 0.4595189690589905,
 "learning_rate": 2e-05,
 "loss": 0.6625,
 "step": 1230
 },
 {
 "epoch": 0.06489768147799235,
 "grad_norm": 0.49671754240989685,
 "learning_rate": 2e-05,
 "loss": 0.663,
 "step": 1240
 },
 {
 "epoch": 0.06542104987700843,
 "grad_norm": 0.44070637226104736,
 "learning_rate": 2e-05,
 "loss": 0.6613,
 "step": 1250
 },
 {
 "epoch": 0.0659444182760245,
 "grad_norm": 0.45897576212882996,
 "learning_rate": 2e-05,
 "loss": 0.6516,
 "step": 1260
 },
 {
 "epoch": 0.06646778667504057,
 "grad_norm": 0.4450547695159912,
 "learning_rate": 2e-05,
 "loss": 0.641,
 "step": 1270
 },
 {
 "epoch": 0.06699115507405663,
 "grad_norm": 0.4288750886917114,
 "learning_rate": 2e-05,
 "loss": 0.6656,
 "step": 1280
 },
 {
 "epoch": 0.0675145234730727,
 "grad_norm": 0.47525206208229065,
 "learning_rate": 2e-05,
 "loss": 0.6915,
 "step": 1290
 },
 {
 "epoch": 0.06803789187208877,
 "grad_norm": 0.41285258531570435,
 "learning_rate": 2e-05,
 "loss": 0.6401,
 "step": 1300
 },
 {
 "epoch": 0.06856126027110483,
 "grad_norm": 0.46445295214653015,
 "learning_rate": 2e-05,
 "loss": 0.6692,
 "step": 1310
 },
 {
 "epoch": 0.0690846286701209,
 "grad_norm": 0.44153061509132385,
 "learning_rate": 2e-05,
 "loss": 0.6291,
 "step": 1320
 },
 {
 "epoch": 0.06960799706913696,
 "grad_norm": 0.7016943693161011,
 "learning_rate": 2e-05,
 "loss": 0.6405,
 "step": 1330
 },
 {
 "epoch": 0.07013136546815303,
 "grad_norm": 0.498710572719574,
 "learning_rate": 2e-05,
 "loss": 0.6433,
 "step": 1340
 },
 {
 "epoch": 0.0706547338671691,
 "grad_norm": 0.4415338337421417,
 "learning_rate": 2e-05,
 "loss": 0.673,
 "step": 1350
 },
 {
 "epoch": 0.07117810226618516,
 "grad_norm": 0.5377183556556702,
 "learning_rate": 2e-05,
 "loss": 0.6556,
 "step": 1360
 },
 {
 "epoch": 0.07170147066520123,
 "grad_norm": 0.40461525321006775,
 "learning_rate": 2e-05,
 "loss": 0.6384,
 "step": 1370
 },
 {
 "epoch": 0.0722248390642173,
 "grad_norm": 0.4240797758102417,
 "learning_rate": 2e-05,
 "loss": 0.6467,
 "step": 1380
 },
 {
 "epoch": 0.07274820746323336,
 "grad_norm": 0.4052056074142456,
 "learning_rate": 2e-05,
 "loss": 0.6448,
 "step": 1390
 },
 {
 "epoch": 0.07327157586224944,
 "grad_norm": 0.41825127601623535,
 "learning_rate": 2e-05,
 "loss": 0.6465,
 "step": 1400
 },
 {
 "epoch": 0.07379494426126551,
 "grad_norm": 0.41024652123451233,
 "learning_rate": 2e-05,
 "loss": 0.6428,
 "step": 1410
 },
 {
 "epoch": 0.07431831266028158,
 "grad_norm": 0.38469892740249634,
 "learning_rate": 2e-05,
 "loss": 0.6652,
 "step": 1420
 },
 {
 "epoch": 0.07484168105929764,
 "grad_norm": 0.4639175236225128,
 "learning_rate": 2e-05,
 "loss": 0.6503,
 "step": 1430
 },
 {
 "epoch": 0.07536504945831371,
 "grad_norm": 0.4508153200149536,
 "learning_rate": 2e-05,
 "loss": 0.6464,
 "step": 1440
 },
 {
 "epoch": 0.07588841785732978,
 "grad_norm": 0.5173973441123962,
 "learning_rate": 2e-05,
 "loss": 0.6475,
 "step": 1450
 },
 {
 "epoch": 0.07641178625634584,
 "grad_norm": 0.47181883454322815,
 "learning_rate": 2e-05,
 "loss": 0.6662,
 "step": 1460
 },
 {
 "epoch": 0.07693515465536191,
 "grad_norm": 0.43690988421440125,
 "learning_rate": 2e-05,
 "loss": 0.6459,
 "step": 1470
 },
 {
 "epoch": 0.07745852305437798,
 "grad_norm": 0.40047961473464966,
 "learning_rate": 2e-05,
 "loss": 0.6808,
 "step": 1480
 },
 {
 "epoch": 0.07798189145339404,
 "grad_norm": 0.3962308466434479,
 "learning_rate": 2e-05,
 "loss": 0.6249,
 "step": 1490
 },
 {
 "epoch": 0.07850525985241011,
 "grad_norm": 0.5075916051864624,
 "learning_rate": 2e-05,
 "loss": 0.6562,
 "step": 1500
 },
 {
 "epoch": 0.07902862825142618,
 "grad_norm": 0.44439756870269775,
 "learning_rate": 2e-05,
 "loss": 0.6744,
 "step": 1510
 },
 {
 "epoch": 0.07955199665044224,
 "grad_norm": 0.4172062575817108,
 "learning_rate": 2e-05,
 "loss": 0.6328,
 "step": 1520
 },
 {
 "epoch": 0.08007536504945831,
 "grad_norm": 0.9272112250328064,
 "learning_rate": 2e-05,
 "loss": 0.6554,
 "step": 1530
 },
 {
 "epoch": 0.08059873344847437,
 "grad_norm": 0.44365498423576355,
 "learning_rate": 2e-05,
 "loss": 0.6567,
 "step": 1540
 },
 {
 "epoch": 0.08112210184749045,
 "grad_norm": 0.5808505415916443,
 "learning_rate": 2e-05,
 "loss": 0.6671,
 "step": 1550
 },
 {
 "epoch": 0.08164547024650652,
 "grad_norm": 0.4238007068634033,
 "learning_rate": 2e-05,
 "loss": 0.6655,
 "step": 1560
 },
 {
 "epoch": 0.08216883864552259,
 "grad_norm": 0.4158059358596802,
 "learning_rate": 2e-05,
 "loss": 0.6502,
 "step": 1570
 },
 {
 "epoch": 0.08269220704453865,
 "grad_norm": 0.4266833961009979,
 "learning_rate": 2e-05,
 "loss": 0.6389,
 "step": 1580
 },
 {
 "epoch": 0.08321557544355472,
 "grad_norm": 0.4183817505836487,
 "learning_rate": 2e-05,
 "loss": 0.6297,
 "step": 1590
 },
 {
 "epoch": 0.08373894384257079,
 "grad_norm": 0.4242572784423828,
 "learning_rate": 2e-05,
 "loss": 0.6575,
 "step": 1600
 },
 {
 "epoch": 0.08426231224158685,
 "grad_norm": 0.42805418372154236,
 "learning_rate": 2e-05,
 "loss": 0.6316,
 "step": 1610
 },
 {
 "epoch": 0.08478568064060292,
 "grad_norm": 0.441866010427475,
 "learning_rate": 2e-05,
 "loss": 0.6702,
 "step": 1620
 },
 {
 "epoch": 0.08530904903961899,
 "grad_norm": 0.4371151626110077,
 "learning_rate": 2e-05,
 "loss": 0.6623,
 "step": 1630
 },
 {
 "epoch": 0.08583241743863505,
 "grad_norm": 0.43131211400032043,
 "learning_rate": 2e-05,
 "loss": 0.6448,
 "step": 1640
 },
 {
 "epoch": 0.08635578583765112,
 "grad_norm": 0.4316512644290924,
 "learning_rate": 2e-05,
 "loss": 0.6374,
 "step": 1650
 },
 {
 "epoch": 0.08687915423666719,
 "grad_norm": 0.48419904708862305,
 "learning_rate": 2e-05,
 "loss": 0.6224,
 "step": 1660
 },
 {
 "epoch": 0.08740252263568325,
 "grad_norm": 0.3933804929256439,
 "learning_rate": 2e-05,
 "loss": 0.6522,
 "step": 1670
 },
 {
 "epoch": 0.08792589103469932,
 "grad_norm": 0.4437639117240906,
 "learning_rate": 2e-05,
 "loss": 0.6237,
 "step": 1680
 },
 {
 "epoch": 0.08844925943371539,
 "grad_norm": 0.4899040460586548,
 "learning_rate": 2e-05,
 "loss": 0.6321,
 "step": 1690
 },
 {
 "epoch": 0.08897262783273147,
 "grad_norm": 0.4382249712944031,
 "learning_rate": 2e-05,
 "loss": 0.6469,
 "step": 1700
 },
 {
 "epoch": 0.08949599623174753,
 "grad_norm": 0.4095931053161621,
 "learning_rate": 2e-05,
 "loss": 0.6588,
 "step": 1710
 },
 {
 "epoch": 0.0900193646307636,
 "grad_norm": 0.43513184785842896,
 "learning_rate": 2e-05,
 "loss": 0.6569,
 "step": 1720
 },
 {
 "epoch": 0.09054273302977967,
 "grad_norm": 0.46793821454048157,
 "learning_rate": 2e-05,
 "loss": 0.6322,
 "step": 1730
 },
 {
 "epoch": 0.09106610142879573,
 "grad_norm": 0.430164098739624,
 "learning_rate": 2e-05,
 "loss": 0.6224,
 "step": 1740
 },
 {
 "epoch": 0.0915894698278118,
 "grad_norm": 0.3761891722679138,
 "learning_rate": 2e-05,
 "loss": 0.606,
 "step": 1750
 },
 {
 "epoch": 0.09211283822682786,
 "grad_norm": 0.373146116733551,
 "learning_rate": 2e-05,
 "loss": 0.6329,
 "step": 1760
 },
 {
 "epoch": 0.09263620662584393,
 "grad_norm": 0.42644017934799194,
 "learning_rate": 2e-05,
 "loss": 0.6466,
 "step": 1770
 },
 {
 "epoch": 0.09315957502486,
 "grad_norm": 0.40171223878860474,
 "learning_rate": 2e-05,
 "loss": 0.6567,
 "step": 1780
 },
 {
 "epoch": 0.09368294342387606,
 "grad_norm": 0.5002732872962952,
 "learning_rate": 2e-05,
 "loss": 0.6645,
 "step": 1790
 },
 {
 "epoch": 0.09420631182289213,
 "grad_norm": 0.45148345828056335,
 "learning_rate": 2e-05,
 "loss": 0.6735,
 "step": 1800
 },
 {
 "epoch": 0.0947296802219082,
 "grad_norm": 0.4408950209617615,
 "learning_rate": 2e-05,
 "loss": 0.64,
 "step": 1810
 },
 {
 "epoch": 0.09525304862092426,
 "grad_norm": 0.41224032640457153,
 "learning_rate": 2e-05,
 "loss": 0.6483,
 "step": 1820
 },
 {
 "epoch": 0.09577641701994033,
 "grad_norm": 0.3857000768184662,
 "learning_rate": 2e-05,
 "loss": 0.6591,
 "step": 1830
 },
 {
 "epoch": 0.09629978541895641,
 "grad_norm": 0.5142344832420349,
 "learning_rate": 2e-05,
 "loss": 0.6613,
 "step": 1840
 },
 {
 "epoch": 0.09682315381797248,
 "grad_norm": 0.43707653880119324,
 "learning_rate": 2e-05,
 "loss": 0.6463,
 "step": 1850
 },
 {
 "epoch": 0.09734652221698854,
 "grad_norm": 0.5951546430587769,
 "learning_rate": 2e-05,
 "loss": 0.6634,
 "step": 1860
 },
 {
 "epoch": 0.09786989061600461,
 "grad_norm": 0.451498419046402,
 "learning_rate": 2e-05,
 "loss": 0.6228,
 "step": 1870
 },
 {
 "epoch": 0.09839325901502068,
 "grad_norm": 0.43872302770614624,
 "learning_rate": 2e-05,
 "loss": 0.6417,
 "step": 1880
 },
 {
 "epoch": 0.09891662741403674,
 "grad_norm": 0.3905869722366333,
 "learning_rate": 2e-05,
 "loss": 0.6195,
 "step": 1890
 },
 {
 "epoch": 0.09943999581305281,
 "grad_norm": 0.46226704120635986,
 "learning_rate": 2e-05,
 "loss": 0.652,
 "step": 1900
 },
 {
 "epoch": 0.09996336421206888,
 "grad_norm": 0.4350249469280243,
 "learning_rate": 2e-05,
 "loss": 0.6814,
 "step": 1910
 },
 {
 "epoch": 0.10048673261108494,
 "grad_norm": 0.547003448009491,
 "learning_rate": 2e-05,
 "loss": 0.6376,
 "step": 1920
 },
 {
 "epoch": 0.10101010101010101,
 "grad_norm": 0.40520697832107544,
 "learning_rate": 2e-05,
 "loss": 0.6281,
 "step": 1930
 },
 {
 "epoch": 0.10153346940911707,
 "grad_norm": 0.3886966109275818,
 "learning_rate": 2e-05,
 "loss": 0.6349,
 "step": 1940
 },
 {
 "epoch": 0.10205683780813314,
 "grad_norm": 0.4416722059249878,
 "learning_rate": 2e-05,
 "loss": 0.6626,
 "step": 1950
 },
 {
 "epoch": 0.10258020620714921,
 "grad_norm": 0.4272310137748718,
 "learning_rate": 2e-05,
 "loss": 0.627,
 "step": 1960
 },
 {
 "epoch": 0.10310357460616527,
 "grad_norm": 0.4342178404331207,
 "learning_rate": 2e-05,
 "loss": 0.6616,
 "step": 1970
 },
 {
 "epoch": 0.10362694300518134,
 "grad_norm": 0.4413588047027588,
 "learning_rate": 2e-05,
 "loss": 0.6602,
 "step": 1980
 },
 {
 "epoch": 0.10415031140419742,
 "grad_norm": 0.39807382225990295,
 "learning_rate": 2e-05,
 "loss": 0.6184,
 "step": 1990
 },
 {
 "epoch": 0.10467367980321349,
 "grad_norm": 0.39922189712524414,
 "learning_rate": 2e-05,
 "loss": 0.6348,
 "step": 2000
 },
 {
 "epoch": 0.10519704820222955,
 "grad_norm": 0.4405508041381836,
 "learning_rate": 2e-05,
 "loss": 0.6527,
 "step": 2010
 },
 {
 "epoch": 0.10572041660124562,
 "grad_norm": 0.4154644310474396,
 "learning_rate": 2e-05,
 "loss": 0.6383,
 "step": 2020
 },
 {
 "epoch": 0.10624378500026169,
 "grad_norm": 0.4121659994125366,
 "learning_rate": 2e-05,
 "loss": 0.6337,
 "step": 2030
 },
 {
 "epoch": 0.10676715339927775,
 "grad_norm": 0.4231645166873932,
 "learning_rate": 2e-05,
 "loss": 0.6648,
 "step": 2040
 },
 {
 "epoch": 0.10729052179829382,
 "grad_norm": 0.5495308041572571,
 "learning_rate": 2e-05,
 "loss": 0.6655,
 "step": 2050
 },
 {
 "epoch": 0.10781389019730989,
 "grad_norm": 0.4083961546421051,
 "learning_rate": 2e-05,
 "loss": 0.6091,
 "step": 2060
 },
 {
 "epoch": 0.10833725859632595,
 "grad_norm": 0.394467830657959,
 "learning_rate": 2e-05,
 "loss": 0.6472,
 "step": 2070
 },
 {
 "epoch": 0.10886062699534202,
 "grad_norm": 0.4051300585269928,
 "learning_rate": 2e-05,
 "loss": 0.6295,
 "step": 2080
 },
 {
 "epoch": 0.10938399539435809,
 "grad_norm": 0.44103720784187317,
 "learning_rate": 2e-05,
 "loss": 0.6316,
 "step": 2090
 },
 {
 "epoch": 0.10990736379337415,
 "grad_norm": 0.4452396333217621,
 "learning_rate": 2e-05,
 "loss": 0.6367,
 "step": 2100
 },
 {
 "epoch": 0.11043073219239022,
 "grad_norm": 0.5510708689689636,
 "learning_rate": 2e-05,
 "loss": 0.6258,
 "step": 2110
 },
 {
 "epoch": 0.11095410059140629,
 "grad_norm": 0.5372933149337769,
 "learning_rate": 2e-05,
 "loss": 0.6476,
 "step": 2120
 },
 {
 "epoch": 0.11147746899042235,
 "grad_norm": 0.4383063316345215,
 "learning_rate": 2e-05,
 "loss": 0.6605,
 "step": 2130
 },
 {
 "epoch": 0.11200083738943843,
 "grad_norm": 0.4241836667060852,
 "learning_rate": 2e-05,
 "loss": 0.6362,
 "step": 2140
 },
 {
 "epoch": 0.1125242057884545,
 "grad_norm": 0.442451149225235,
 "learning_rate": 2e-05,
 "loss": 0.6433,
 "step": 2150
 },
 {
 "epoch": 0.11304757418747056,
 "grad_norm": 0.41505861282348633,
 "learning_rate": 2e-05,
 "loss": 0.6239,
 "step": 2160
 },
 {
 "epoch": 0.11357094258648663,
 "grad_norm": 0.43439918756484985,
 "learning_rate": 2e-05,
 "loss": 0.6274,
 "step": 2170
 },
 {
 "epoch": 0.1140943109855027,
 "grad_norm": 0.43937069177627563,
 "learning_rate": 2e-05,
 "loss": 0.6414,
 "step": 2180
 },
 {
 "epoch": 0.11461767938451876,
 "grad_norm": 0.43210020661354065,
 "learning_rate": 2e-05,
 "loss": 0.6257,
 "step": 2190
 },
 {
 "epoch": 0.11514104778353483,
 "grad_norm": 0.49416911602020264,
 "learning_rate": 2e-05,
 "loss": 0.6442,
 "step": 2200
 },
 {
 "epoch": 0.1156644161825509,
 "grad_norm": 0.40978503227233887,
 "learning_rate": 2e-05,
 "loss": 0.6329,
 "step": 2210
 },
 {
 "epoch": 0.11618778458156696,
 "grad_norm": 0.4340236485004425,
 "learning_rate": 2e-05,
 "loss": 0.6347,
 "step": 2220
 },
 {
 "epoch": 0.11671115298058303,
 "grad_norm": 0.41441839933395386,
 "learning_rate": 2e-05,
 "loss": 0.6205,
 "step": 2230
 },
 {
 "epoch": 0.1172345213795991,
 "grad_norm": 0.4365447461605072,
 "learning_rate": 2e-05,
 "loss": 0.6329,
 "step": 2240
 },
 {
 "epoch": 0.11775788977861516,
 "grad_norm": 0.44118615984916687,
 "learning_rate": 2e-05,
 "loss": 0.6549,
 "step": 2250
 },
 {
 "epoch": 0.11828125817763123,
 "grad_norm": 0.40321072936058044,
 "learning_rate": 2e-05,
 "loss": 0.6284,
 "step": 2260
 },
 {
 "epoch": 0.1188046265766473,
 "grad_norm": 0.4358213543891907,
 "learning_rate": 2e-05,
 "loss": 0.6375,
 "step": 2270
 },
 {
 "epoch": 0.11932799497566338,
 "grad_norm": 0.42884060740470886,
 "learning_rate": 2e-05,
 "loss": 0.6452,
 "step": 2280
 },
 {
 "epoch": 0.11985136337467944,
 "grad_norm": 0.43688151240348816,
 "learning_rate": 2e-05,
 "loss": 0.6433,
 "step": 2290
 },
 {
 "epoch": 0.12037473177369551,
 "grad_norm": 0.42834022641181946,
 "learning_rate": 2e-05,
 "loss": 0.6461,
 "step": 2300
 },
 {
 "epoch": 0.12089810017271158,
 "grad_norm": 0.44969525933265686,
 "learning_rate": 2e-05,
 "loss": 0.6704,
 "step": 2310
 },
 {
 "epoch": 0.12142146857172764,
 "grad_norm": 0.4053316116333008,
 "learning_rate": 2e-05,
 "loss": 0.6188,
 "step": 2320
 },
 {
 "epoch": 0.12194483697074371,
 "grad_norm": 0.3716852366924286,
 "learning_rate": 2e-05,
 "loss": 0.6249,
 "step": 2330
 },
 {
 "epoch": 0.12246820536975978,
 "grad_norm": 0.4359619617462158,
 "learning_rate": 2e-05,
 "loss": 0.6571,
 "step": 2340
 },
 {
 "epoch": 0.12299157376877584,
 "grad_norm": 0.38862133026123047,
 "learning_rate": 2e-05,
 "loss": 0.641,
 "step": 2350
 },
 {
 "epoch": 0.12351494216779191,
 "grad_norm": 0.4286133646965027,
 "learning_rate": 2e-05,
 "loss": 0.6514,
 "step": 2360
 },
 {
 "epoch": 0.12403831056680797,
 "grad_norm": 0.4758356809616089,
 "learning_rate": 2e-05,
 "loss": 0.628,
 "step": 2370
 },
 {
 "epoch": 0.12456167896582404,
 "grad_norm": 0.4074471592903137,
 "learning_rate": 2e-05,
 "loss": 0.6161,
 "step": 2380
 },
 {
 "epoch": 0.1250850473648401,
 "grad_norm": 0.4395975172519684,
 "learning_rate": 2e-05,
 "loss": 0.6514,
 "step": 2390
 },
 {
 "epoch": 0.12560841576385617,
 "grad_norm": 0.4212956130504608,
 "learning_rate": 2e-05,
 "loss": 0.6459,
 "step": 2400
 },
 {
 "epoch": 0.12613178416287224,
 "grad_norm": 0.43518325686454773,
 "learning_rate": 2e-05,
 "loss": 0.6631,
 "step": 2410
 },
 {
 "epoch": 0.1266551525618883,
 "grad_norm": 0.41669315099716187,
 "learning_rate": 2e-05,
 "loss": 0.6433,
 "step": 2420
 },
 {
 "epoch": 0.12717852096090437,
 "grad_norm": 0.47521886229515076,
 "learning_rate": 2e-05,
 "loss": 0.6008,
 "step": 2430
 },
 {
 "epoch": 0.12770188935992044,
 "grad_norm": 0.5306794047355652,
 "learning_rate": 2e-05,
 "loss": 0.6499,
 "step": 2440
 },
 {
 "epoch": 0.1282252577589365,
 "grad_norm": 0.4078957438468933,
 "learning_rate": 2e-05,
 "loss": 0.632,
 "step": 2450
 },
 {
 "epoch": 0.12874862615795257,
 "grad_norm": 0.4263654351234436,
 "learning_rate": 2e-05,
 "loss": 0.6367,
 "step": 2460
 },
 {
 "epoch": 0.12927199455696864,
 "grad_norm": 0.4139572083950043,
 "learning_rate": 2e-05,
 "loss": 0.642,
 "step": 2470
 },
 {
 "epoch": 0.1297953629559847,
 "grad_norm": 0.38871797919273376,
 "learning_rate": 2e-05,
 "loss": 0.6342,
 "step": 2480
 },
 {
 "epoch": 0.13031873135500077,
 "grad_norm": 0.429225891828537,
 "learning_rate": 2e-05,
 "loss": 0.6249,
 "step": 2490
 },
 {
 "epoch": 0.13084209975401687,
 "grad_norm": 0.45410633087158203,
 "learning_rate": 2e-05,
 "loss": 0.6395,
 "step": 2500
 },
 {
 "epoch": 0.13136546815303293,
 "grad_norm": 0.45400431752204895,
 "learning_rate": 2e-05,
 "loss": 0.6244,
 "step": 2510
 },
 {
 "epoch": 0.131888836552049,
 "grad_norm": 0.4065890312194824,
 "learning_rate": 2e-05,
 "loss": 0.6394,
 "step": 2520
 },
 {
 "epoch": 0.13241220495106507,
 "grad_norm": 0.4191737771034241,
 "learning_rate": 2e-05,
 "loss": 0.6409,
 "step": 2530
 },
 {
 "epoch": 0.13293557335008113,
 "grad_norm": 0.45096731185913086,
 "learning_rate": 2e-05,
 "loss": 0.6208,
 "step": 2540
 },
 {
 "epoch": 0.1334589417490972,
 "grad_norm": 0.46258044242858887,
 "learning_rate": 2e-05,
 "loss": 0.6395,
 "step": 2550
 },
 {
 "epoch": 0.13398231014811327,
 "grad_norm": 0.4282276928424835,
 "learning_rate": 2e-05,
 "loss": 0.601,
 "step": 2560
 },
 {
 "epoch": 0.13450567854712933,
 "grad_norm": 0.41605663299560547,
 "learning_rate": 2e-05,
 "loss": 0.6501,
 "step": 2570
 },
 {
 "epoch": 0.1350290469461454,
 "grad_norm": 0.41660454869270325,
 "learning_rate": 2e-05,
 "loss": 0.6546,
 "step": 2580
 },
 {
 "epoch": 0.13555241534516146,
 "grad_norm": 0.4275190234184265,
 "learning_rate": 2e-05,
 "loss": 0.6458,
 "step": 2590
 },
 {
 "epoch": 0.13607578374417753,
 "grad_norm": 0.4305119514465332,
 "learning_rate": 2e-05,
 "loss": 0.6403,
 "step": 2600
 },
 {
 "epoch": 0.1365991521431936,
 "grad_norm": 0.42454832792282104,
 "learning_rate": 2e-05,
 "loss": 0.6368,
 "step": 2610
 },
 {
 "epoch": 0.13712252054220966,
 "grad_norm": 0.44257181882858276,
 "learning_rate": 2e-05,
 "loss": 0.6343,
 "step": 2620
 },
 {
 "epoch": 0.13764588894122573,
 "grad_norm": 0.3995290696620941,
 "learning_rate": 2e-05,
 "loss": 0.6616,
 "step": 2630
 },
 {
 "epoch": 0.1381692573402418,
 "grad_norm": 0.4199754297733307,
 "learning_rate": 2e-05,
 "loss": 0.6286,
 "step": 2640
 },
 {
 "epoch": 0.13869262573925786,
 "grad_norm": 0.42020726203918457,
 "learning_rate": 2e-05,
 "loss": 0.6371,
 "step": 2650
 },
 {
 "epoch": 0.13921599413827393,
 "grad_norm": 0.39368361234664917,
 "learning_rate": 2e-05,
 "loss": 0.6517,
 "step": 2660
 },
 {
 "epoch": 0.13973936253729,
 "grad_norm": 0.4302023947238922,
 "learning_rate": 2e-05,
 "loss": 0.6139,
 "step": 2670
 },
 {
 "epoch": 0.14026273093630606,
 "grad_norm": 0.49658092856407166,
 "learning_rate": 2e-05,
 "loss": 0.6565,
 "step": 2680
 },
 {
 "epoch": 0.14078609933532213,
 "grad_norm": 0.3941155672073364,
 "learning_rate": 2e-05,
 "loss": 0.6527,
 "step": 2690
 },
 {
 "epoch": 0.1413094677343382,
 "grad_norm": 0.43608132004737854,
 "learning_rate": 2e-05,
 "loss": 0.6309,
 "step": 2700
 },
 {
 "epoch": 0.14183283613335426,
 "grad_norm": 0.4120804965496063,
 "learning_rate": 2e-05,
 "loss": 0.6326,
 "step": 2710
 },
 {
 "epoch": 0.14235620453237033,
 "grad_norm": 0.4378657639026642,
 "learning_rate": 2e-05,
 "loss": 0.6479,
 "step": 2720
 },
 {
 "epoch": 0.1428795729313864,
 "grad_norm": 0.39626598358154297,
 "learning_rate": 2e-05,
 "loss": 0.6537,
 "step": 2730
 },
 {
 "epoch": 0.14340294133040246,
 "grad_norm": 0.40195828676223755,
 "learning_rate": 2e-05,
 "loss": 0.6075,
 "step": 2740
 },
 {
 "epoch": 0.14392630972941853,
 "grad_norm": 0.3953752815723419,
 "learning_rate": 2e-05,
 "loss": 0.6179,
 "step": 2750
 },
 {
 "epoch": 0.1444496781284346,
 "grad_norm": 0.4433000385761261,
 "learning_rate": 2e-05,
 "loss": 0.6743,
 "step": 2760
 },
 {
 "epoch": 0.14497304652745066,
 "grad_norm": 0.42893922328948975,
 "learning_rate": 2e-05,
 "loss": 0.6272,
 "step": 2770
 },
 {
 "epoch": 0.14549641492646673,
 "grad_norm": 0.4065708518028259,
 "learning_rate": 2e-05,
 "loss": 0.6417,
 "step": 2780
 },
 {
 "epoch": 0.1460197833254828,
 "grad_norm": 0.4181337058544159,
 "learning_rate": 2e-05,
 "loss": 0.6392,
 "step": 2790
 },
 {
 "epoch": 0.1465431517244989,
 "grad_norm": 0.4179243743419647,
 "learning_rate": 2e-05,
 "loss": 0.6026,
 "step": 2800
 },
 {
 "epoch": 0.14706652012351495,
 "grad_norm": 0.46388933062553406,
 "learning_rate": 2e-05,
 "loss": 0.6335,
 "step": 2810
 },
 {
 "epoch": 0.14758988852253102,
 "grad_norm": 0.4350208640098572,
 "learning_rate": 2e-05,
 "loss": 0.6349,
 "step": 2820
 },
 {
 "epoch": 0.1481132569215471,
 "grad_norm": 0.4281371831893921,
 "learning_rate": 2e-05,
 "loss": 0.5927,
 "step": 2830
 },
 {
 "epoch": 0.14863662532056315,
 "grad_norm": 0.3796765208244324,
 "learning_rate": 2e-05,
 "loss": 0.6244,
 "step": 2840
 },
 {
 "epoch": 0.14915999371957922,
 "grad_norm": 0.400057315826416,
 "learning_rate": 2e-05,
 "loss": 0.6083,
 "step": 2850
 },
 {
 "epoch": 0.1496833621185953,
 "grad_norm": 0.5580277442932129,
 "learning_rate": 2e-05,
 "loss": 0.6088,
 "step": 2860
 },
 {
 "epoch": 0.15020673051761135,
 "grad_norm": 0.4220695197582245,
 "learning_rate": 2e-05,
 "loss": 0.6339,
 "step": 2870
 },
 {
 "epoch": 0.15073009891662742,
 "grad_norm": 0.4196612536907196,
 "learning_rate": 2e-05,
 "loss": 0.6415,
 "step": 2880
 },
 {
 "epoch": 0.1512534673156435,
 "grad_norm": 0.4167044162750244,
 "learning_rate": 2e-05,
 "loss": 0.6276,
 "step": 2890
 },
 {
 "epoch": 0.15177683571465955,
 "grad_norm": 0.4121061861515045,
 "learning_rate": 2e-05,
 "loss": 0.6311,
 "step": 2900
 },
 {
 "epoch": 0.15230020411367562,
 "grad_norm": 0.4583412706851959,
 "learning_rate": 2e-05,
 "loss": 0.6415,
 "step": 2910
 },
 {
 "epoch": 0.15282357251269169,
 "grad_norm": 0.4203738272190094,
 "learning_rate": 2e-05,
 "loss": 0.657,
 "step": 2920
 },
 {
 "epoch": 0.15334694091170775,
 "grad_norm": 0.4278572201728821,
 "learning_rate": 2e-05,
 "loss": 0.6457,
 "step": 2930
 },
 {
 "epoch": 0.15387030931072382,
 "grad_norm": 0.41517725586891174,
 "learning_rate": 2e-05,
 "loss": 0.6506,
 "step": 2940
 },
 {
 "epoch": 0.15439367770973988,
 "grad_norm": 0.39440619945526123,
 "learning_rate": 2e-05,
 "loss": 0.6247,
 "step": 2950
 },
 {
 "epoch": 0.15491704610875595,
 "grad_norm": 0.3762326240539551,
 "learning_rate": 2e-05,
 "loss": 0.6464,
 "step": 2960
 },
 {
 "epoch": 0.15544041450777202,
 "grad_norm": 0.43617796897888184,
 "learning_rate": 2e-05,
 "loss": 0.6162,
 "step": 2970
 },
 {
 "epoch": 0.15596378290678808,
 "grad_norm": 0.41636911034584045,
 "learning_rate": 2e-05,
 "loss": 0.6327,
 "step": 2980
 },
 {
 "epoch": 0.15648715130580415,
 "grad_norm": 0.4136853814125061,
 "learning_rate": 2e-05,
 "loss": 0.6292,
 "step": 2990
 },
 {
 "epoch": 0.15701051970482022,
 "grad_norm": 0.44505634903907776,
 "learning_rate": 2e-05,
 "loss": 0.646,
 "step": 3000
 },
 {
 "epoch": 0.15753388810383628,
 "grad_norm": 0.42681798338890076,
 "learning_rate": 2e-05,
 "loss": 0.6274,
 "step": 3010
 },
 {
 "epoch": 0.15805725650285235,
 "grad_norm": 0.3740960657596588,
 "learning_rate": 2e-05,
 "loss": 0.6233,
 "step": 3020
 },
 {
 "epoch": 0.15858062490186842,
 "grad_norm": 0.4044531285762787,
 "learning_rate": 2e-05,
 "loss": 0.655,
 "step": 3030
 },
 {
 "epoch": 0.15910399330088448,
 "grad_norm": 0.38664835691452026,
 "learning_rate": 2e-05,
 "loss": 0.6201,
 "step": 3040
 },
 {
 "epoch": 0.15962736169990055,
 "grad_norm": 0.5274628400802612,
 "learning_rate": 2e-05,
 "loss": 0.6374,
 "step": 3050
 },
 {
 "epoch": 0.16015073009891662,
 "grad_norm": 0.4279114305973053,
 "learning_rate": 2e-05,
 "loss": 0.6433,
 "step": 3060
 },
 {
 "epoch": 0.16067409849793268,
 "grad_norm": 0.396634578704834,
 "learning_rate": 2e-05,
 "loss": 0.6155,
 "step": 3070
 },
 {
 "epoch": 0.16119746689694875,
 "grad_norm": 0.41256171464920044,
 "learning_rate": 2e-05,
 "loss": 0.6274,
 "step": 3080
 },
 {
 "epoch": 0.16172083529596484,
 "grad_norm": 0.47806742787361145,
 "learning_rate": 2e-05,
 "loss": 0.6288,
 "step": 3090
 },
 {
 "epoch": 0.1622442036949809,
 "grad_norm": 0.41482001543045044,
 "learning_rate": 2e-05,
 "loss": 0.6097,
 "step": 3100
 },
 {
 "epoch": 0.16276757209399698,
 "grad_norm": 0.4124285876750946,
 "learning_rate": 2e-05,
 "loss": 0.6381,
 "step": 3110
 },
 {
 "epoch": 0.16329094049301304,
 "grad_norm": 0.39308762550354004,
 "learning_rate": 2e-05,
 "loss": 0.6418,
 "step": 3120
 },
 {
 "epoch": 0.1638143088920291,
 "grad_norm": 0.5976954698562622,
 "learning_rate": 2e-05,
 "loss": 0.6403,
 "step": 3130
 },
 {
 "epoch": 0.16433767729104518,
 "grad_norm": 0.4556901752948761,
 "learning_rate": 2e-05,
 "loss": 0.6415,
 "step": 3140
 },
 {
 "epoch": 0.16486104569006124,
 "grad_norm": 0.4634074866771698,
 "learning_rate": 2e-05,
 "loss": 0.6142,
 "step": 3150
 },
 {
 "epoch": 0.1653844140890773,
 "grad_norm": 0.4238816797733307,
 "learning_rate": 2e-05,
 "loss": 0.6528,
 "step": 3160
 },
 {
 "epoch": 0.16590778248809337,
 "grad_norm": 0.5388637185096741,
 "learning_rate": 2e-05,
 "loss": 0.6723,
 "step": 3170
 },
 {
 "epoch": 0.16643115088710944,
 "grad_norm": 0.3934621512889862,
 "learning_rate": 2e-05,
 "loss": 0.6347,
 "step": 3180
 },
 {
 "epoch": 0.1669545192861255,
 "grad_norm": 0.3961109519004822,
 "learning_rate": 2e-05,
 "loss": 0.6345,
 "step": 3190
 },
 {
 "epoch": 0.16747788768514157,
 "grad_norm": 0.41134336590766907,
 "learning_rate": 2e-05,
 "loss": 0.6227,
 "step": 3200
 },
 {
 "epoch": 0.16800125608415764,
 "grad_norm": 0.4064910411834717,
 "learning_rate": 2e-05,
 "loss": 0.6087,
 "step": 3210
 },
 {
 "epoch": 0.1685246244831737,
 "grad_norm": 0.411873459815979,
 "learning_rate": 2e-05,
 "loss": 0.6291,
 "step": 3220
 },
 {
 "epoch": 0.16904799288218977,
 "grad_norm": 0.41122299432754517,
 "learning_rate": 2e-05,
 "loss": 0.6217,
 "step": 3230
 },
 {
 "epoch": 0.16957136128120584,
 "grad_norm": 0.4552735686302185,
 "learning_rate": 2e-05,
 "loss": 0.6334,
 "step": 3240
 },
 {
 "epoch": 0.1700947296802219,
 "grad_norm": 0.457132488489151,
 "learning_rate": 2e-05,
 "loss": 0.6213,
 "step": 3250
 },
 {
 "epoch": 0.17061809807923797,
 "grad_norm": 0.38603243231773376,
 "learning_rate": 2e-05,
 "loss": 0.6363,
 "step": 3260
 },
 {
 "epoch": 0.17114146647825404,
 "grad_norm": 0.4321597218513489,
 "learning_rate": 2e-05,
 "loss": 0.6194,
 "step": 3270
 },
 {
 "epoch": 0.1716648348772701,
 "grad_norm": 0.43034639954566956,
 "learning_rate": 2e-05,
 "loss": 0.6409,
 "step": 3280
 },
 {
 "epoch": 0.17218820327628617,
 "grad_norm": 0.42103973031044006,
 "learning_rate": 2e-05,
 "loss": 0.6154,
 "step": 3290
 },
 {
 "epoch": 0.17271157167530224,
 "grad_norm": 0.3927100896835327,
 "learning_rate": 2e-05,
 "loss": 0.6313,
 "step": 3300
 },
 {
 "epoch": 0.1732349400743183,
 "grad_norm": 0.4339587688446045,
 "learning_rate": 2e-05,
 "loss": 0.614,
 "step": 3310
 },
 {
 "epoch": 0.17375830847333437,
 "grad_norm": 0.40991151332855225,
 "learning_rate": 2e-05,
 "loss": 0.6456,
 "step": 3320
 },
 {
 "epoch": 0.17428167687235044,
 "grad_norm": 0.4325578808784485,
 "learning_rate": 2e-05,
 "loss": 0.6041,
 "step": 3330
 },
 {
 "epoch": 0.1748050452713665,
 "grad_norm": 0.41213691234588623,
 "learning_rate": 2e-05,
 "loss": 0.6349,
 "step": 3340
 },
 {
 "epoch": 0.17532841367038257,
 "grad_norm": 0.47782137989997864,
 "learning_rate": 2e-05,
 "loss": 0.636,
 "step": 3350
 },
 {
 "epoch": 0.17585178206939864,
 "grad_norm": 0.42506250739097595,
 "learning_rate": 2e-05,
 "loss": 0.626,
 "step": 3360
 },
 {
 "epoch": 0.1763751504684147,
 "grad_norm": 0.3940761983394623,
 "learning_rate": 2e-05,
 "loss": 0.6193,
 "step": 3370
 },
 {
 "epoch": 0.17689851886743077,
 "grad_norm": 0.4069708287715912,
 "learning_rate": 2e-05,
 "loss": 0.6365,
 "step": 3380
 },
 {
 "epoch": 0.17742188726644686,
 "grad_norm": 0.504355788230896,
 "learning_rate": 2e-05,
 "loss": 0.5985,
 "step": 3390
 },
 {
 "epoch": 0.17794525566546293,
 "grad_norm": 0.3881029486656189,
 "learning_rate": 2e-05,
 "loss": 0.6328,
 "step": 3400
 },
 {
 "epoch": 0.178468624064479,
 "grad_norm": 2.036409854888916,
 "learning_rate": 2e-05,
 "loss": 0.6237,
 "step": 3410
 },
 {
 "epoch": 0.17899199246349506,
 "grad_norm": 0.41256919503211975,
 "learning_rate": 2e-05,
 "loss": 0.6066,
 "step": 3420
 },
 {
 "epoch": 0.17951536086251113,
 "grad_norm": 0.4488495886325836,
 "learning_rate": 2e-05,
 "loss": 0.6247,
 "step": 3430
 },
 {
 "epoch": 0.1800387292615272,
 "grad_norm": 0.4205232560634613,
 "learning_rate": 2e-05,
 "loss": 0.6094,
 "step": 3440
 },
 {
 "epoch": 0.18056209766054326,
 "grad_norm": 0.4089616537094116,
 "learning_rate": 2e-05,
 "loss": 0.6264,
 "step": 3450
 },
 {
 "epoch": 0.18108546605955933,
 "grad_norm": 0.37557995319366455,
 "learning_rate": 2e-05,
 "loss": 0.6322,
 "step": 3460
 },
 {
 "epoch": 0.1816088344585754,
 "grad_norm": 0.39397501945495605,
 "learning_rate": 2e-05,
 "loss": 0.6647,
 "step": 3470
 },
 {
 "epoch": 0.18213220285759146,
 "grad_norm": 0.4009655714035034,
 "learning_rate": 2e-05,
 "loss": 0.6203,
 "step": 3480
 },
 {
 "epoch": 0.18265557125660753,
 "grad_norm": 0.40983593463897705,
 "learning_rate": 2e-05,
 "loss": 0.6684,
 "step": 3490
 },
 {
 "epoch": 0.1831789396556236,
 "grad_norm": 0.41632992029190063,
 "learning_rate": 2e-05,
 "loss": 0.6055,
 "step": 3500
 },
 {
 "epoch": 0.18370230805463966,
 "grad_norm": 0.42826464772224426,
 "learning_rate": 2e-05,
 "loss": 0.6114,
 "step": 3510
 },
 {
 "epoch": 0.18422567645365573,
 "grad_norm": 0.4451156556606293,
 "learning_rate": 2e-05,
 "loss": 0.6164,
 "step": 3520
 },
 {
 "epoch": 0.1847490448526718,
 "grad_norm": 0.7244383692741394,
 "learning_rate": 2e-05,
 "loss": 0.643,
 "step": 3530
 },
 {
 "epoch": 0.18527241325168786,
 "grad_norm": 0.41192492842674255,
 "learning_rate": 2e-05,
 "loss": 0.6389,
 "step": 3540
 },
 {
 "epoch": 0.18579578165070393,
 "grad_norm": 0.42937004566192627,
 "learning_rate": 2e-05,
 "loss": 0.6449,
 "step": 3550
 },
 {
 "epoch": 0.18631915004972,
 "grad_norm": 0.4111078977584839,
 "learning_rate": 2e-05,
 "loss": 0.6202,
 "step": 3560
 },
 {
 "epoch": 0.18684251844873606,
 "grad_norm": 0.4172969460487366,
 "learning_rate": 2e-05,
 "loss": 0.6188,
 "step": 3570
 },
 {
 "epoch": 0.18736588684775213,
 "grad_norm": 0.38246381282806396,
 "learning_rate": 2e-05,
 "loss": 0.609,
 "step": 3580
 },
 {
 "epoch": 0.1878892552467682,
 "grad_norm": 0.467737078666687,
 "learning_rate": 2e-05,
 "loss": 0.6177,
 "step": 3590
 },
 {
 "epoch": 0.18841262364578426,
 "grad_norm": 0.3787391185760498,
 "learning_rate": 2e-05,
 "loss": 0.6156,
 "step": 3600
 },
 {
 "epoch": 0.18893599204480033,
 "grad_norm": 0.39074644446372986,
 "learning_rate": 2e-05,
 "loss": 0.6275,
 "step": 3610
 },
 {
 "epoch": 0.1894593604438164,
 "grad_norm": 0.4353400468826294,
 "learning_rate": 2e-05,
 "loss": 0.613,
 "step": 3620
 },
 {
 "epoch": 0.18998272884283246,
 "grad_norm": 0.4572504758834839,
 "learning_rate": 2e-05,
 "loss": 0.6243,
 "step": 3630
 },
 {
 "epoch": 0.19050609724184853,
 "grad_norm": 0.3868942856788635,
 "learning_rate": 2e-05,
 "loss": 0.6303,
 "step": 3640
 },
 {
 "epoch": 0.1910294656408646,
 "grad_norm": 0.41320866346359253,
 "learning_rate": 2e-05,
 "loss": 0.6255,
 "step": 3650
 },
 {
 "epoch": 0.19155283403988066,
 "grad_norm": 0.39754143357276917,
 "learning_rate": 2e-05,
 "loss": 0.6064,
 "step": 3660
 },
 {
 "epoch": 0.19207620243889673,
 "grad_norm": 0.45954880118370056,
 "learning_rate": 2e-05,
 "loss": 0.6355,
 "step": 3670
 },
 {
 "epoch": 0.19259957083791282,
 "grad_norm": 0.42518895864486694,
 "learning_rate": 2e-05,
 "loss": 0.6339,
 "step": 3680
 },
 {
 "epoch": 0.1931229392369289,
 "grad_norm": 0.40126708149909973,
 "learning_rate": 2e-05,
 "loss": 0.6226,
 "step": 3690
 },
 {
 "epoch": 0.19364630763594495,
 "grad_norm": 0.5110933184623718,
 "learning_rate": 2e-05,
 "loss": 0.6236,
 "step": 3700
 },
 {
 "epoch": 0.19416967603496102,
 "grad_norm": 0.4530631899833679,
 "learning_rate": 2e-05,
 "loss": 0.6132,
 "step": 3710
 },
 {
 "epoch": 0.19469304443397709,
 "grad_norm": 0.40570759773254395,
 "learning_rate": 2e-05,
 "loss": 0.6205,
 "step": 3720
 },
 {
 "epoch": 0.19521641283299315,
 "grad_norm": 0.4424304962158203,
 "learning_rate": 2e-05,
 "loss": 0.6318,
 "step": 3730
 },
 {
 "epoch": 0.19573978123200922,
 "grad_norm": 0.4104612171649933,
 "learning_rate": 2e-05,
 "loss": 0.6112,
 "step": 3740
 },
 {
 "epoch": 0.19626314963102529,
 "grad_norm": 0.4605982303619385,
 "learning_rate": 2e-05,
 "loss": 0.63,
 "step": 3750
 },
 {
 "epoch": 0.19678651803004135,
 "grad_norm": 0.41476568579673767,
 "learning_rate": 2e-05,
 "loss": 0.6301,
 "step": 3760
 },
 {
 "epoch": 0.19730988642905742,
 "grad_norm": 0.4510325491428375,
 "learning_rate": 2e-05,
 "loss": 0.6311,
 "step": 3770
 },
 {
 "epoch": 0.19783325482807348,
 "grad_norm": 0.3622564673423767,
 "learning_rate": 2e-05,
 "loss": 0.6028,
 "step": 3780
 },
 {
 "epoch": 0.19835662322708955,
 "grad_norm": 0.40758174657821655,
 "learning_rate": 2e-05,
 "loss": 0.6328,
 "step": 3790
 },
 {
 "epoch": 0.19887999162610562,
 "grad_norm": 0.42536821961402893,
 "learning_rate": 2e-05,
 "loss": 0.6343,
 "step": 3800
 },
 {
 "epoch": 0.19940336002512168,
 "grad_norm": 0.4246145486831665,
 "learning_rate": 2e-05,
 "loss": 0.617,
 "step": 3810
 },
 {
 "epoch": 0.19992672842413775,
 "grad_norm": 0.4054490327835083,
 "learning_rate": 2e-05,
 "loss": 0.6374,
 "step": 3820
 },
 {
 "epoch": 0.20045009682315382,
 "grad_norm": 0.4975704848766327,
 "learning_rate": 2e-05,
 "loss": 0.6006,
 "step": 3830
 },
 {
 "epoch": 0.20097346522216988,
 "grad_norm": 0.5053114891052246,
 "learning_rate": 2e-05,
 "loss": 0.6319,
 "step": 3840
 },
 {
 "epoch": 0.20149683362118595,
 "grad_norm": 0.41113340854644775,
 "learning_rate": 2e-05,
 "loss": 0.6342,
 "step": 3850
 },
 {
 "epoch": 0.20202020202020202,
 "grad_norm": 0.3836223781108856,
 "learning_rate": 2e-05,
 "loss": 0.6196,
 "step": 3860
 },
 {
 "epoch": 0.20254357041921808,
 "grad_norm": 0.3937544822692871,
 "learning_rate": 2e-05,
 "loss": 0.641,
 "step": 3870
 },
 {
 "epoch": 0.20306693881823415,
 "grad_norm": 0.4237481951713562,
 "learning_rate": 2e-05,
 "loss": 0.6125,
 "step": 3880
 },
 {
 "epoch": 0.20359030721725022,
 "grad_norm": 0.3751864731311798,
 "learning_rate": 2e-05,
 "loss": 0.6197,
 "step": 3890
 },
 {
 "epoch": 0.20411367561626628,
 "grad_norm": 0.5081929564476013,
 "learning_rate": 2e-05,
 "loss": 0.6203,
 "step": 3900
 },
 {
 "epoch": 0.20463704401528235,
 "grad_norm": 0.4540235698223114,
 "learning_rate": 2e-05,
 "loss": 0.6168,
 "step": 3910
 },
 {
 "epoch": 0.20516041241429842,
 "grad_norm": 0.41905754804611206,
 "learning_rate": 2e-05,
 "loss": 0.616,
 "step": 3920
 },
 {
 "epoch": 0.20568378081331448,
 "grad_norm": 0.5326279997825623,
 "learning_rate": 2e-05,
 "loss": 0.6211,
 "step": 3930
 },
 {
 "epoch": 0.20620714921233055,
 "grad_norm": 0.41054078936576843,
 "learning_rate": 2e-05,
 "loss": 0.6113,
 "step": 3940
 },
 {
 "epoch": 0.20673051761134661,
 "grad_norm": 0.4346046447753906,
 "learning_rate": 2e-05,
 "loss": 0.6495,
 "step": 3950
 },
 {
 "epoch": 0.20725388601036268,
 "grad_norm": 0.410095751285553,
 "learning_rate": 2e-05,
 "loss": 0.62,
 "step": 3960
 },
 {
 "epoch": 0.20777725440937878,
 "grad_norm": 0.40288132429122925,
 "learning_rate": 2e-05,
 "loss": 0.6214,
 "step": 3970
 },
 {
 "epoch": 0.20830062280839484,
 "grad_norm": 0.40218082070350647,
 "learning_rate": 2e-05,
 "loss": 0.6318,
 "step": 3980
 },
 {
 "epoch": 0.2088239912074109,
 "grad_norm": 0.43279343843460083,
 "learning_rate": 2e-05,
 "loss": 0.6341,
 "step": 3990
 },
 {
 "epoch": 0.20934735960642697,
 "grad_norm": 0.39592084288597107,
 "learning_rate": 2e-05,
 "loss": 0.6042,
 "step": 4000
 },
 {
 "epoch": 0.20987072800544304,
 "grad_norm": 0.4071950912475586,
 "learning_rate": 2e-05,
 "loss": 0.5843,
 "step": 4010
 },
 {
 "epoch": 0.2103940964044591,
 "grad_norm": 0.3897649347782135,
 "learning_rate": 2e-05,
 "loss": 0.6269,
 "step": 4020
 },
 {
 "epoch": 0.21091746480347517,
 "grad_norm": 0.7160203456878662,
 "learning_rate": 2e-05,
 "loss": 0.6499,
 "step": 4030
 },
 {
 "epoch": 0.21144083320249124,
 "grad_norm": 0.38851386308670044,
 "learning_rate": 2e-05,
 "loss": 0.6142,
 "step": 4040
 },
 {
 "epoch": 0.2119642016015073,
 "grad_norm": 0.3943970799446106,
 "learning_rate": 2e-05,
 "loss": 0.5983,
 "step": 4050
 },
 {
 "epoch": 0.21248757000052337,
 "grad_norm": 0.41590937972068787,
 "learning_rate": 2e-05,
 "loss": 0.6363,
 "step": 4060
 },
 {
 "epoch": 0.21301093839953944,
 "grad_norm": 0.38856077194213867,
 "learning_rate": 2e-05,
 "loss": 0.611,
 "step": 4070
 },
 {
 "epoch": 0.2135343067985555,
 "grad_norm": 0.4386836886405945,
 "learning_rate": 2e-05,
 "loss": 0.6145,
 "step": 4080
 },
 {
 "epoch": 0.21405767519757157,
 "grad_norm": 0.46340975165367126,
 "learning_rate": 2e-05,
 "loss": 0.5949,
 "step": 4090
 },
 {
 "epoch": 0.21458104359658764,
 "grad_norm": 0.4160849153995514,
 "learning_rate": 2e-05,
 "loss": 0.6446,
 "step": 4100
 },
 {
 "epoch": 0.2151044119956037,
 "grad_norm": 0.4251480996608734,
 "learning_rate": 2e-05,
 "loss": 0.6031,
 "step": 4110
 },
 {
 "epoch": 0.21562778039461977,
 "grad_norm": 0.41039806604385376,
 "learning_rate": 2e-05,
 "loss": 0.6165,
 "step": 4120
 },
 {
 "epoch": 0.21615114879363584,
 "grad_norm": 0.4157627522945404,
 "learning_rate": 2e-05,
 "loss": 0.6219,
 "step": 4130
 },
 {
 "epoch": 0.2166745171926519,
 "grad_norm": 0.3525838553905487,
 "learning_rate": 2e-05,
 "loss": 0.6062,
 "step": 4140
 },
 {
 "epoch": 0.21719788559166797,
 "grad_norm": 0.4129687249660492,
 "learning_rate": 2e-05,
 "loss": 0.6104,
 "step": 4150
 },
 {
 "epoch": 0.21772125399068404,
 "grad_norm": 0.4055493175983429,
 "learning_rate": 2e-05,
 "loss": 0.5947,
 "step": 4160
 },
 {
 "epoch": 0.2182446223897001,
 "grad_norm": 0.42468446493148804,
 "learning_rate": 2e-05,
 "loss": 0.6507,
 "step": 4170
 },
 {
 "epoch": 0.21876799078871617,
 "grad_norm": 0.4344474673271179,
 "learning_rate": 2e-05,
 "loss": 0.6062,
 "step": 4180
 },
 {
 "epoch": 0.21929135918773224,
 "grad_norm": 0.4484303295612335,
 "learning_rate": 2e-05,
 "loss": 0.6128,
 "step": 4190
 },
 {
 "epoch": 0.2198147275867483,
 "grad_norm": 0.38424041867256165,
 "learning_rate": 2e-05,
 "loss": 0.611,
 "step": 4200
 },
 {
 "epoch": 0.22033809598576437,
 "grad_norm": 0.40774214267730713,
 "learning_rate": 2e-05,
 "loss": 0.6128,
 "step": 4210
 },
 {
 "epoch": 0.22086146438478044,
 "grad_norm": 0.39545294642448425,
 "learning_rate": 2e-05,
 "loss": 0.6196,
 "step": 4220
 },
 {
 "epoch": 0.2213848327837965,
 "grad_norm": 0.42367228865623474,
 "learning_rate": 2e-05,
 "loss": 0.6324,
 "step": 4230
 },
 {
 "epoch": 0.22190820118281257,
 "grad_norm": 0.4583156406879425,
 "learning_rate": 2e-05,
 "loss": 0.6011,
 "step": 4240
 },
 {
 "epoch": 0.22243156958182864,
 "grad_norm": 0.36509275436401367,
 "learning_rate": 2e-05,
 "loss": 0.6227,
 "step": 4250
 },
 {
 "epoch": 0.2229549379808447,
 "grad_norm": 0.41607242822647095,
 "learning_rate": 2e-05,
 "loss": 0.625,
 "step": 4260
 },
 {
 "epoch": 0.2234783063798608,
 "grad_norm": 0.41028332710266113,
 "learning_rate": 2e-05,
 "loss": 0.5968,
 "step": 4270
 },
 {
 "epoch": 0.22400167477887686,
 "grad_norm": 0.4013286530971527,
 "learning_rate": 2e-05,
 "loss": 0.6143,
 "step": 4280
 },
 {
 "epoch": 0.22452504317789293,
 "grad_norm": 0.42334598302841187,
 "learning_rate": 2e-05,
 "loss": 0.6036,
 "step": 4290
 },
 {
 "epoch": 0.225048411576909,
 "grad_norm": 0.47744619846343994,
 "learning_rate": 2e-05,
 "loss": 0.6329,
 "step": 4300
 },
 {
 "epoch": 0.22557177997592506,
 "grad_norm": 0.378061443567276,
 "learning_rate": 2e-05,
 "loss": 0.6191,
 "step": 4310
 },
 {
 "epoch": 0.22609514837494113,
 "grad_norm": 0.4232480823993683,
 "learning_rate": 2e-05,
 "loss": 0.6167,
 "step": 4320
 },
 {
 "epoch": 0.2266185167739572,
 "grad_norm": 0.3921084702014923,
 "learning_rate": 2e-05,
 "loss": 0.6363,
 "step": 4330
 },
 {
 "epoch": 0.22714188517297326,
 "grad_norm": 0.3987734913825989,
 "learning_rate": 2e-05,
 "loss": 0.611,
 "step": 4340
 },
 {
 "epoch": 0.22766525357198933,
 "grad_norm": 0.4001140296459198,
 "learning_rate": 2e-05,
 "loss": 0.6025,
 "step": 4350
 },
 {
 "epoch": 0.2281886219710054,
 "grad_norm": 0.4162245988845825,
 "learning_rate": 2e-05,
 "loss": 0.6151,
 "step": 4360
 },
 {
 "epoch": 0.22871199037002146,
 "grad_norm": 0.4258664548397064,
 "learning_rate": 2e-05,
 "loss": 0.6107,
 "step": 4370
 },
 {
 "epoch": 0.22923535876903753,
 "grad_norm": 0.39507806301116943,
 "learning_rate": 2e-05,
 "loss": 0.6155,
 "step": 4380
 },
 {
 "epoch": 0.2297587271680536,
 "grad_norm": 0.49936944246292114,
 "learning_rate": 2e-05,
 "loss": 0.6367,
 "step": 4390
 },
 {
 "epoch": 0.23028209556706966,
 "grad_norm": 0.4109865128993988,
 "learning_rate": 2e-05,
 "loss": 0.6163,
 "step": 4400
 },
 {
 "epoch": 0.23080546396608573,
 "grad_norm": 0.4059872031211853,
 "learning_rate": 2e-05,
 "loss": 0.6087,
 "step": 4410
 },
 {
 "epoch": 0.2313288323651018,
 "grad_norm": 0.41509491205215454,
 "learning_rate": 2e-05,
 "loss": 0.6034,
 "step": 4420
 },
 {
 "epoch": 0.23185220076411786,
 "grad_norm": 0.4001082181930542,
 "learning_rate": 2e-05,
 "loss": 0.5977,
 "step": 4430
 },
 {
 "epoch": 0.23237556916313393,
 "grad_norm": 0.4086252748966217,
 "learning_rate": 2e-05,
 "loss": 0.606,
 "step": 4440
 },
 {
 "epoch": 0.23289893756215,
 "grad_norm": 0.3729661703109741,
 "learning_rate": 2e-05,
 "loss": 0.6201,
 "step": 4450
 },
 {
 "epoch": 0.23342230596116606,
 "grad_norm": 0.39218300580978394,
 "learning_rate": 2e-05,
 "loss": 0.6174,
 "step": 4460
 },
 {
 "epoch": 0.23394567436018213,
 "grad_norm": 0.3822868764400482,
 "learning_rate": 2e-05,
 "loss": 0.6395,
 "step": 4470
 },
 {
 "epoch": 0.2344690427591982,
 "grad_norm": 0.4116714596748352,
 "learning_rate": 2e-05,
 "loss": 0.6101,
 "step": 4480
 },
 {
 "epoch": 0.23499241115821426,
 "grad_norm": 0.4040875732898712,
 "learning_rate": 2e-05,
 "loss": 0.626,
 "step": 4490
 },
 {
 "epoch": 0.23551577955723033,
 "grad_norm": 0.3950350880622864,
 "learning_rate": 2e-05,
 "loss": 0.5975,
 "step": 4500
 },
 {
 "epoch": 0.2360391479562464,
 "grad_norm": 0.39547741413116455,
 "learning_rate": 2e-05,
 "loss": 0.6092,
 "step": 4510
 },
 {
 "epoch": 0.23656251635526246,
 "grad_norm": 0.4103192985057831,
 "learning_rate": 2e-05,
 "loss": 0.5902,
 "step": 4520
 },
 {
 "epoch": 0.23708588475427853,
 "grad_norm": 0.4051879048347473,
 "learning_rate": 2e-05,
 "loss": 0.6064,
 "step": 4530
 },
 {
 "epoch": 0.2376092531532946,
 "grad_norm": 0.4002724885940552,
 "learning_rate": 2e-05,
 "loss": 0.618,
 "step": 4540
 },
 {
 "epoch": 0.23813262155231066,
 "grad_norm": 0.4022727310657501,
 "learning_rate": 2e-05,
 "loss": 0.6145,
 "step": 4550
 },
 {
 "epoch": 0.23865598995132675,
 "grad_norm": 0.4159553050994873,
 "learning_rate": 2e-05,
 "loss": 0.5922,
 "step": 4560
 },
 {
 "epoch": 0.23917935835034282,
 "grad_norm": 0.3999551236629486,
 "learning_rate": 2e-05,
 "loss": 0.5905,
 "step": 4570
 },
 {
 "epoch": 0.23970272674935889,
 "grad_norm": 0.4089633822441101,
 "learning_rate": 2e-05,
 "loss": 0.6132,
 "step": 4580
 },
 {
 "epoch": 0.24022609514837495,
 "grad_norm": 0.37393105030059814,
 "learning_rate": 2e-05,
 "loss": 0.6094,
 "step": 4590
 },
 {
 "epoch": 0.24074946354739102,
 "grad_norm": 0.6190505623817444,
 "learning_rate": 2e-05,
 "loss": 0.6066,
 "step": 4600
 },
 {
 "epoch": 0.24127283194640708,
 "grad_norm": 0.42357298731803894,
 "learning_rate": 2e-05,
 "loss": 0.63,
 "step": 4610
 },
 {
 "epoch": 0.24179620034542315,
 "grad_norm": 0.7327708601951599,
 "learning_rate": 2e-05,
 "loss": 0.5942,
 "step": 4620
 },
 {
 "epoch": 0.24231956874443922,
 "grad_norm": 0.4425099790096283,
 "learning_rate": 2e-05,
 "loss": 0.6078,
 "step": 4630
 },
 {
 "epoch": 0.24284293714345528,
 "grad_norm": 0.3585580289363861,
 "learning_rate": 2e-05,
 "loss": 0.5886,
 "step": 4640
 },
 {
 "epoch": 0.24336630554247135,
 "grad_norm": 1.4051371812820435,
 "learning_rate": 2e-05,
 "loss": 0.6031,
 "step": 4650
 },
 {
 "epoch": 0.24388967394148742,
 "grad_norm": 0.4339183270931244,
 "learning_rate": 2e-05,
 "loss": 0.6104,
 "step": 4660
 },
 {
 "epoch": 0.24441304234050348,
 "grad_norm": 0.40716227889060974,
 "learning_rate": 2e-05,
 "loss": 0.6532,
 "step": 4670
 },
 {
 "epoch": 0.24493641073951955,
 "grad_norm": 0.3990173041820526,
 "learning_rate": 2e-05,
 "loss": 0.6159,
 "step": 4680
 },
 {
 "epoch": 0.24545977913853562,
 "grad_norm": 0.5576123595237732,
 "learning_rate": 2e-05,
 "loss": 0.6279,
 "step": 4690
 },
 {
 "epoch": 0.24598314753755168,
 "grad_norm": 0.3955235183238983,
 "learning_rate": 2e-05,
 "loss": 0.6232,
 "step": 4700
 },
 {
 "epoch": 0.24650651593656775,
 "grad_norm": 0.4591265320777893,
 "learning_rate": 2e-05,
 "loss": 0.6311,
 "step": 4710
 },
 {
 "epoch": 0.24702988433558382,
 "grad_norm": 0.3967943787574768,
 "learning_rate": 2e-05,
 "loss": 0.6285,
 "step": 4720
 },
 {
 "epoch": 0.24755325273459988,
 "grad_norm": 0.5398264527320862,
 "learning_rate": 2e-05,
 "loss": 0.6331,
 "step": 4730
 },
 {
 "epoch": 0.24807662113361595,
 "grad_norm": 0.41231897473335266,
 "learning_rate": 2e-05,
 "loss": 0.6024,
 "step": 4740
 },
 {
 "epoch": 0.24859998953263202,
 "grad_norm": 0.40629324316978455,
 "learning_rate": 2e-05,
 "loss": 0.6271,
 "step": 4750
 },
 {
 "epoch": 0.24912335793164808,
 "grad_norm": 0.42986491322517395,
 "learning_rate": 2e-05,
 "loss": 0.5869,
 "step": 4760
 },
 {
 "epoch": 0.24964672633066415,
 "grad_norm": 0.45604678988456726,
 "learning_rate": 2e-05,
 "loss": 0.632,
 "step": 4770
 },
 {
 "epoch": 0.2501700947296802,
 "grad_norm": 0.4645533859729767,
 "learning_rate": 2e-05,
 "loss": 0.6167,
 "step": 4780
 },
 {
 "epoch": 0.2506934631286963,
 "grad_norm": 0.3783343732357025,
 "learning_rate": 2e-05,
 "loss": 0.6157,
 "step": 4790
 },
 {
 "epoch": 0.25121683152771235,
 "grad_norm": 0.38526099920272827,
 "learning_rate": 2e-05,
 "loss": 0.5896,
 "step": 4800
 },
 {
 "epoch": 0.2517401999267284,
 "grad_norm": 0.3877682685852051,
 "learning_rate": 2e-05,
 "loss": 0.6102,
 "step": 4810
 },
 {
 "epoch": 0.2522635683257445,
 "grad_norm": 0.39466989040374756,
 "learning_rate": 2e-05,
 "loss": 0.5981,
 "step": 4820
 },
 {
 "epoch": 0.25278693672476055,
 "grad_norm": 0.4054771065711975,
 "learning_rate": 2e-05,
 "loss": 0.5967,
 "step": 4830
 },
 {
 "epoch": 0.2533103051237766,
 "grad_norm": 0.37173131108283997,
 "learning_rate": 2e-05,
 "loss": 0.5909,
 "step": 4840
 },
 {
 "epoch": 0.2538336735227927,
 "grad_norm": 0.41072943806648254,
 "learning_rate": 2e-05,
 "loss": 0.6111,
 "step": 4850
 },
 {
 "epoch": 0.25435704192180875,
 "grad_norm": 0.41547060012817383,
 "learning_rate": 2e-05,
 "loss": 0.6222,
 "step": 4860
 },
 {
 "epoch": 0.2548804103208248,
 "grad_norm": 0.409974604845047,
 "learning_rate": 2e-05,
 "loss": 0.6307,
 "step": 4870
 },
 {
 "epoch": 0.2554037787198409,
 "grad_norm": 0.38040807843208313,
 "learning_rate": 2e-05,
 "loss": 0.6125,
 "step": 4880
 },
 {
 "epoch": 0.25592714711885695,
 "grad_norm": 0.4161960482597351,
 "learning_rate": 2e-05,
 "loss": 0.6311,
 "step": 4890
 },
 {
 "epoch": 0.256450515517873,
 "grad_norm": 0.41063639521598816,
 "learning_rate": 2e-05,
 "loss": 0.6335,
 "step": 4900
 },
 {
 "epoch": 0.2569738839168891,
 "grad_norm": 0.4094699025154114,
 "learning_rate": 2e-05,
 "loss": 0.6166,
 "step": 4910
 },
 {
 "epoch": 0.25749725231590515,
 "grad_norm": 0.4404178559780121,
 "learning_rate": 2e-05,
 "loss": 0.6109,
 "step": 4920
 },
 {
 "epoch": 0.2580206207149212,
 "grad_norm": 0.42314210534095764,
 "learning_rate": 2e-05,
 "loss": 0.6089,
 "step": 4930
 },
 {
 "epoch": 0.2585439891139373,
 "grad_norm": 0.4365096688270569,
 "learning_rate": 2e-05,
 "loss": 0.6193,
 "step": 4940
 },
 {
 "epoch": 0.25906735751295334,
 "grad_norm": 0.4479396939277649,
 "learning_rate": 2e-05,
 "loss": 0.6254,
 "step": 4950
 },
 {
 "epoch": 0.2595907259119694,
 "grad_norm": 0.3912433385848999,
 "learning_rate": 2e-05,
 "loss": 0.6169,
 "step": 4960
 },
 {
 "epoch": 0.2601140943109855,
 "grad_norm": 0.3662976622581482,
 "learning_rate": 2e-05,
 "loss": 0.6037,
 "step": 4970
 },
 {
 "epoch": 0.26063746271000154,
 "grad_norm": 0.4002629518508911,
 "learning_rate": 2e-05,
 "loss": 0.5935,
 "step": 4980
 },
 {
 "epoch": 0.2611608311090176,
 "grad_norm": 0.39754584431648254,
 "learning_rate": 2e-05,
 "loss": 0.6045,
 "step": 4990
 },
 {
 "epoch": 0.26168419950803373,
 "grad_norm": 0.44900912046432495,
 "learning_rate": 2e-05,
 "loss": 0.6234,
 "step": 5000
 },
 {
 "epoch": 0.2622075679070498,
 "grad_norm": 0.46902185678482056,
 "learning_rate": 2e-05,
 "loss": 0.5961,
 "step": 5010
 },
 {
 "epoch": 0.26273093630606587,
 "grad_norm": 0.41309309005737305,
 "learning_rate": 2e-05,
 "loss": 0.6275,
 "step": 5020
 },
 {
 "epoch": 0.26325430470508193,
 "grad_norm": 0.396864652633667,
 "learning_rate": 2e-05,
 "loss": 0.638,
 "step": 5030
 },
 {
 "epoch": 0.263777673104098,
 "grad_norm": 0.39725908637046814,
 "learning_rate": 2e-05,
 "loss": 0.638,
 "step": 5040
 },
 {
 "epoch": 0.26430104150311406,
 "grad_norm": 0.40473729372024536,
 "learning_rate": 2e-05,
 "loss": 0.6166,
 "step": 5050
 },
 {
 "epoch": 0.26482440990213013,
 "grad_norm": 0.45297154784202576,
 "learning_rate": 2e-05,
 "loss": 0.6167,
 "step": 5060
 },
 {
 "epoch": 0.2653477783011462,
 "grad_norm": 0.3637174367904663,
 "learning_rate": 2e-05,
 "loss": 0.59,
 "step": 5070
 },
 {
 "epoch": 0.26587114670016226,
 "grad_norm": 0.3888295590877533,
 "learning_rate": 2e-05,
 "loss": 0.5946,
 "step": 5080
 },
 {
 "epoch": 0.26639451509917833,
 "grad_norm": 0.3925114870071411,
 "learning_rate": 2e-05,
 "loss": 0.6454,
 "step": 5090
 },
 {
 "epoch": 0.2669178834981944,
 "grad_norm": 0.42430034279823303,
 "learning_rate": 2e-05,
 "loss": 0.598,
 "step": 5100
 },
 {
 "epoch": 0.26744125189721046,
 "grad_norm": 0.4356481432914734,
 "learning_rate": 2e-05,
 "loss": 0.5841,
 "step": 5110
 },
 {
 "epoch": 0.26796462029622653,
 "grad_norm": 0.40845903754234314,
 "learning_rate": 2e-05,
 "loss": 0.6214,
 "step": 5120
 },
 {
 "epoch": 0.2684879886952426,
 "grad_norm": 0.37573105096817017,
 "learning_rate": 2e-05,
 "loss": 0.6283,
 "step": 5130
 },
 {
 "epoch": 0.26901135709425866,
 "grad_norm": 0.40981823205947876,
 "learning_rate": 2e-05,
 "loss": 0.5933,
 "step": 5140
 },
 {
 "epoch": 0.26953472549327473,
 "grad_norm": 0.39033955335617065,
 "learning_rate": 2e-05,
 "loss": 0.5921,
 "step": 5150
 },
 {
 "epoch": 0.2700580938922908,
 "grad_norm": 0.4205704927444458,
 "learning_rate": 2e-05,
 "loss": 0.6128,
 "step": 5160
 },
 {
 "epoch": 0.27058146229130686,
 "grad_norm": 0.37669286131858826,
 "learning_rate": 2e-05,
 "loss": 0.6054,
 "step": 5170
 },
 {
 "epoch": 0.27110483069032293,
 "grad_norm": 0.40791818499565125,
 "learning_rate": 2e-05,
 "loss": 0.6309,
 "step": 5180
 },
 {
 "epoch": 0.271628199089339,
 "grad_norm": 0.42482030391693115,
 "learning_rate": 2e-05,
 "loss": 0.6297,
 "step": 5190
 },
 {
 "epoch": 0.27215156748835506,
 "grad_norm": 0.4918680787086487,
 "learning_rate": 2e-05,
 "loss": 0.5974,
 "step": 5200
 },
 {
 "epoch": 0.27267493588737113,
 "grad_norm": 0.4212997257709503,
 "learning_rate": 2e-05,
 "loss": 0.6251,
 "step": 5210
 },
 {
 "epoch": 0.2731983042863872,
 "grad_norm": 0.40279462933540344,
 "learning_rate": 2e-05,
 "loss": 0.5907,
 "step": 5220
 },
 {
 "epoch": 0.27372167268540326,
 "grad_norm": 0.4512042999267578,
 "learning_rate": 2e-05,
 "loss": 0.6082,
 "step": 5230
 },
 {
 "epoch": 0.2742450410844193,
 "grad_norm": 0.4068576693534851,
 "learning_rate": 2e-05,
 "loss": 0.6062,
 "step": 5240
 },
 {
 "epoch": 0.2747684094834354,
 "grad_norm": 0.4271692931652069,
 "learning_rate": 2e-05,
 "loss": 0.6326,
 "step": 5250
 },
 {
 "epoch": 0.27529177788245146,
 "grad_norm": 0.41401100158691406,
 "learning_rate": 2e-05,
 "loss": 0.6105,
 "step": 5260
 },
 {
 "epoch": 0.2758151462814675,
 "grad_norm": 0.4413628578186035,
 "learning_rate": 2e-05,
 "loss": 0.5898,
 "step": 5270
 },
 {
 "epoch": 0.2763385146804836,
 "grad_norm": 0.41365042328834534,
 "learning_rate": 2e-05,
 "loss": 0.5958,
 "step": 5280
 },
 {
 "epoch": 0.27686188307949966,
 "grad_norm": 0.45695561170578003,
 "learning_rate": 2e-05,
 "loss": 0.6283,
 "step": 5290
 },
 {
 "epoch": 0.2773852514785157,
 "grad_norm": 0.4205687940120697,
 "learning_rate": 2e-05,
 "loss": 0.6257,
 "step": 5300
 },
 {
 "epoch": 0.2779086198775318,
 "grad_norm": 0.3920809030532837,
 "learning_rate": 2e-05,
 "loss": 0.6518,
 "step": 5310
 },
 {
 "epoch": 0.27843198827654786,
 "grad_norm": 0.34857314825057983,
 "learning_rate": 2e-05,
 "loss": 0.5915,
 "step": 5320
 },
 {
 "epoch": 0.2789553566755639,
 "grad_norm": 0.47157859802246094,
 "learning_rate": 2e-05,
 "loss": 0.6315,
 "step": 5330
 },
 {
 "epoch": 0.27947872507458,
 "grad_norm": 0.4518488943576813,
 "learning_rate": 2e-05,
 "loss": 0.6172,
 "step": 5340
 },
 {
 "epoch": 0.28000209347359606,
 "grad_norm": 0.44076552987098694,
 "learning_rate": 1.999194520388485e-05,
 "loss": 0.6251,
 "step": 5350
 },
 {
 "epoch": 0.2805254618726121,
 "grad_norm": 0.38890475034713745,
 "learning_rate": 1.9959247303157763e-05,
 "loss": 0.6194,
 "step": 5360
 },
 {
 "epoch": 0.2810488302716282,
 "grad_norm": 0.3902517855167389,
 "learning_rate": 1.990149424459817e-05,
 "loss": 0.6009,
 "step": 5370
 },
 {
 "epoch": 0.28157219867064426,
 "grad_norm": 0.39967110753059387,
 "learning_rate": 1.9818847538648688e-05,
 "loss": 0.6081,
 "step": 5380
 },
 {
 "epoch": 0.2820955670696603,
 "grad_norm": 0.3647516965866089,
 "learning_rate": 1.971153831256721e-05,
 "loss": 0.5785,
 "step": 5390
 },
 {
 "epoch": 0.2826189354686764,
 "grad_norm": 0.4686444401741028,
 "learning_rate": 1.9579866664063435e-05,
 "loss": 0.635,
 "step": 5400
 },
 {
 "epoch": 0.28314230386769246,
 "grad_norm": 0.4206537902355194,
 "learning_rate": 1.9424200822054782e-05,
 "loss": 0.628,
 "step": 5410
 },
 {
 "epoch": 0.2836656722667085,
 "grad_norm": 0.43939080834388733,
 "learning_rate": 1.9244976116888626e-05,
 "loss": 0.6335,
 "step": 5420
 },
 {
 "epoch": 0.2841890406657246,
 "grad_norm": 0.3906099796295166,
 "learning_rate": 1.904269376291071e-05,
 "loss": 0.5958,
 "step": 5430
 },
 {
 "epoch": 0.28471240906474066,
 "grad_norm": 0.3985750377178192,
 "learning_rate": 1.88179194567844e-05,
 "loss": 0.6213,
 "step": 5440
 },
 {
 "epoch": 0.2852357774637567,
 "grad_norm": 0.4336455762386322,
 "learning_rate": 1.8571281795480632e-05,
 "loss": 0.604,
 "step": 5450
 },
 {
 "epoch": 0.2857591458627728,
 "grad_norm": 0.4126925468444824,
 "learning_rate": 1.8303470518362808e-05,
 "loss": 0.5991,
 "step": 5460
 },
 {
 "epoch": 0.28628251426178886,
 "grad_norm": 0.4074404835700989,
 "learning_rate": 1.801523457828271e-05,
 "loss": 0.6045,
 "step": 5470
 },
 {
 "epoch": 0.2868058826608049,
 "grad_norm": 0.40034711360931396,
 "learning_rate": 1.7707380047081796e-05,
 "loss": 0.6052,
 "step": 5480
 },
 {
 "epoch": 0.287329251059821,
 "grad_norm": 0.41930365562438965,
 "learning_rate": 1.7380767861355252e-05,
 "loss": 0.6316,
 "step": 5490
 },
 {
 "epoch": 0.28785261945883706,
 "grad_norm": 0.42970430850982666,
 "learning_rate": 1.703631141478299e-05,
 "loss": 0.643,
 "step": 5500
 },
 {
 "epoch": 0.2883759878578531,
 "grad_norm": 0.6285122632980347,
 "learning_rate": 1.6674974003760705e-05,
 "loss": 0.5952,
 "step": 5510
 },
 {
 "epoch": 0.2888993562568692,
 "grad_norm": 0.4414973556995392,
 "learning_rate": 1.6297766133474556e-05,
 "loss": 0.5934,
 "step": 5520
 },
 {
 "epoch": 0.28942272465588526,
 "grad_norm": 0.3714768886566162,
 "learning_rate": 1.5905742691953186e-05,
 "loss": 0.5846,
 "step": 5530
 },
 {
 "epoch": 0.2899460930549013,
 "grad_norm": 0.3532920181751251,
 "learning_rate": 1.55e-05,
 "loss": 0.5995,
 "step": 5540
 },
 {
 "epoch": 0.2904694614539174,
 "grad_norm": 0.4924233853816986,
 "learning_rate": 1.5081672745255862e-05,
 "loss": 0.5818,
 "step": 5550
 },
 {
 "epoch": 0.29099282985293345,
 "grad_norm": 0.4407850503921509,
 "learning_rate": 1.4651930808966206e-05,
 "loss": 0.6016,
 "step": 5560
 },
 {
 "epoch": 0.2915161982519495,
 "grad_norm": 0.42552539706230164,
 "learning_rate": 1.4211975994326848e-05,
 "loss": 0.603,
 "step": 5570
 },
 {
 "epoch": 0.2920395666509656,
 "grad_norm": 0.36129358410835266,
 "learning_rate": 1.376303866555779e-05,
 "loss": 0.6101,
 "step": 5580
 },
 {
 "epoch": 0.2925629350499817,
 "grad_norm": 0.406896710395813,
 "learning_rate": 1.3306374307104164e-05,
 "loss": 0.6115,
 "step": 5590
 },
 {
 "epoch": 0.2930863034489978,
 "grad_norm": 0.43325328826904297,
 "learning_rate": 1.2843260012586718e-05,
 "loss": 0.6105,
 "step": 5600
 },
 {
 "epoch": 0.29360967184801384,
 "grad_norm": 0.40178170800209045,
 "learning_rate": 1.2374990913320726e-05,
 "loss": 0.6258,
 "step": 5610
 },
 {
 "epoch": 0.2941330402470299,
 "grad_norm": 0.37656369805336,
 "learning_rate": 1.1902876556391247e-05,
 "loss": 0.6063,
 "step": 5620
 },
 {
 "epoch": 0.294656408646046,
 "grad_norm": 0.4082496166229248,
 "learning_rate": 1.1428237242413685e-05,
 "loss": 0.579,
 "step": 5630
 },
 {
 "epoch": 0.29517977704506204,
 "grad_norm": 0.40694573521614075,
 "learning_rate": 1.095240033322132e-05,
 "loss": 0.5992,
 "step": 5640
 },
 {
 "epoch": 0.2957031454440781,
 "grad_norm": 0.4091090261936188,
 "learning_rate": 1.047669653980572e-05,
 "loss": 0.5843,
 "step": 5650
 },
 {
 "epoch": 0.2962265138430942,
 "grad_norm": 0.44866329431533813,
 "learning_rate": 1.0002456200890902e-05,
 "loss": 0.6118,
 "step": 5660
 },
 {
 "epoch": 0.29674988224211024,
 "grad_norm": 0.4145185649394989,
 "learning_rate": 9.531005562548668e-06,
 "loss": 0.5998,
 "step": 5670
 },
 {
 "epoch": 0.2972732506411263,
 "grad_norm": 0.4247947037220001,
 "learning_rate": 9.06366306925925e-06,
 "loss": 0.5885,
 "step": 5680
 },
 {
 "epoch": 0.2977966190401424,
 "grad_norm": 0.3585066795349121,
 "learning_rate": 8.601735676789686e-06,
 "loss": 0.6015,
 "step": 5690
 },
 {
 "epoch": 0.29831998743915844,
 "grad_norm": 0.3997299373149872,
 "learning_rate": 8.14651519720118e-06,
 "loss": 0.6087,
 "step": 5700
 },
 {
 "epoch": 0.2988433558381745,
 "grad_norm": 0.4697336256504059,
 "learning_rate": 7.699274686206866e-06,
 "loss": 0.6262,
 "step": 5710
 },
 {
 "epoch": 0.2993667242371906,
 "grad_norm": 0.3803713023662567,
 "learning_rate": 7.261264882983024e-06,
 "loss": 0.6104,
 "step": 5720
 },
 {
 "epoch": 0.29989009263620664,
 "grad_norm": 0.382207989692688,
 "learning_rate": 6.8337107123900386e-06,
 "loss": 0.5738,
 "step": 5730
 },
 {
 "epoch": 0.3004134610352227,
 "grad_norm": 0.39442846179008484,
 "learning_rate": 6.417807859384881e-06,
 "loss": 0.6163,
 "step": 5740
 },
 {
 "epoch": 0.3009368294342388,
 "grad_norm": 0.3684907555580139,
 "learning_rate": 6.014719425205009e-06,
 "loss": 0.6046,
 "step": 5750
 },
 {
 "epoch": 0.30146019783325484,
 "grad_norm": 0.37866348028182983,
 "learning_rate": 5.6255726746749275e-06,
 "loss": 0.5828,
 "step": 5760
 },
 {
 "epoch": 0.3019835662322709,
 "grad_norm": 0.38500234484672546,
 "learning_rate": 5.251455883731728e-06,
 "loss": 0.6056,
 "step": 5770
 },
 {
 "epoch": 0.302506934631287,
 "grad_norm": 0.4282638430595398,
 "learning_rate": 4.893415295985812e-06,
 "loss": 0.6071,
 "step": 5780
 },
 {
 "epoch": 0.30303030303030304,
 "grad_norm": 0.4080576002597809,
 "learning_rate": 4.552452196827885e-06,
 "loss": 0.5835,
 "step": 5790
 },
 {
 "epoch": 0.3035536714293191,
 "grad_norm": 0.3698746860027313,
 "learning_rate": 4.229520113264785e-06,
 "loss": 0.6342,
 "step": 5800
 },
 {
 "epoch": 0.30407703982833517,
 "grad_norm": 0.388528972864151,
 "learning_rate": 3.925522147314915e-06,
 "loss": 0.6107,
 "step": 5810
 },
 {
 "epoch": 0.30460040822735124,
 "grad_norm": 0.4334981441497803,
 "learning_rate": 3.641308450420675e-06,
 "loss": 0.5793,
 "step": 5820
 },
 {
 "epoch": 0.3051237766263673,
 "grad_norm": 0.4103776514530182,
 "learning_rate": 3.37767384594087e-06,
 "loss": 0.6098,
 "step": 5830
 },
 {
 "epoch": 0.30564714502538337,
 "grad_norm": 0.386342316865921,
 "learning_rate": 3.1353556063719357e-06,
 "loss": 0.5895,
 "step": 5840
 },
 {
 "epoch": 0.30617051342439944,
 "grad_norm": 0.39483997225761414,
 "learning_rate": 2.9150313915141474e-06,
 "loss": 0.5973,
 "step": 5850
 },
 {
 "epoch": 0.3066938818234155,
 "grad_norm": 0.3758963942527771,
 "learning_rate": 2.717317353348891e-06,
 "loss": 0.5851,
 "step": 5860
 },
 {
 "epoch": 0.30721725022243157,
 "grad_norm": 0.37651148438453674,
 "learning_rate": 2.5427664129268253e-06,
 "loss": 0.5944,
 "step": 5870
 },
 {
 "epoch": 0.30774061862144764,
 "grad_norm": 0.3909638226032257,
 "learning_rate": 2.391866714085717e-06,
 "loss": 0.5946,
 "step": 5880
 },
 {
 "epoch": 0.3082639870204637,
 "grad_norm": 0.3788841664791107,
 "learning_rate": 2.2650402583222753e-06,
 "loss": 0.6036,
 "step": 5890
 },
 {
 "epoch": 0.30878735541947977,
 "grad_norm": 0.3904835283756256,
 "learning_rate": 2.1626417246356398e-06,
 "loss": 0.5993,
 "step": 5900
 },
 {
 "epoch": 0.30931072381849584,
 "grad_norm": 0.4416792690753937,
 "learning_rate": 2.084957477642894e-06,
 "loss": 0.5919,
 "step": 5910
 },
 {
 "epoch": 0.3098340922175119,
 "grad_norm": 0.4406580626964569,
 "learning_rate": 2.0322047667405147e-06,
 "loss": 0.5997,
 "step": 5920
 },
 {
 "epoch": 0.31035746061652797,
 "grad_norm": 0.39945074915885925,
 "learning_rate": 2.0045311185513342e-06,
 "loss": 0.6132,
 "step": 5930
 }
 ],
 "logging_steps": 10,
 "max_steps": 5935,
 "num_input_tokens_seen": 0,
 "num_train_epochs": 1,
 "save_steps": 2035,
 "stateful_callbacks": {
 "TrainerControl": {
 "args": {
 "should_epoch_stop": false,
 "should_evaluate": false,
 "should_log": false,
 "should_save": true,
 "should_training_stop": true
 },
 "attributes": {}
 }
 },
 "total_flos": 2.6494000762504348e+19,
 "train_batch_size": 1,
 "trial_name": null,
 "trial_params": null
}