{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 50.0,
  "eval_steps": 500,
  "global_step": 10000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.05,
      "grad_norm": 8.31881332397461,
      "learning_rate": 1.8e-06,
      "loss": 1.069,
      "step": 10
    },
    {
      "epoch": 0.1,
      "grad_norm": 2.0894038677215576,
      "learning_rate": 3.8e-06,
      "loss": 0.8352,
      "step": 20
    },
    {
      "epoch": 0.15,
      "grad_norm": 1.1884596347808838,
      "learning_rate": 5.8e-06,
      "loss": 0.5511,
      "step": 30
    },
    {
      "epoch": 0.2,
      "grad_norm": 1.5132265090942383,
      "learning_rate": 7.8e-06,
      "loss": 0.4146,
      "step": 40
    },
    {
      "epoch": 0.25,
      "grad_norm": 1.5936781167984009,
      "learning_rate": 9.800000000000001e-06,
      "loss": 0.3387,
      "step": 50
    },
    {
      "epoch": 0.3,
      "grad_norm": 1.5522421598434448,
      "learning_rate": 1.18e-05,
      "loss": 0.3109,
      "step": 60
    },
    {
      "epoch": 0.35,
      "grad_norm": 0.7835227847099304,
      "learning_rate": 1.3800000000000002e-05,
      "loss": 0.2827,
      "step": 70
    },
    {
      "epoch": 0.4,
      "grad_norm": 0.7390315532684326,
      "learning_rate": 1.58e-05,
      "loss": 0.2718,
      "step": 80
    },
    {
      "epoch": 0.45,
      "grad_norm": 0.8529846668243408,
      "learning_rate": 1.78e-05,
      "loss": 0.2501,
      "step": 90
    },
    {
      "epoch": 0.5,
      "grad_norm": 0.7633126974105835,
      "learning_rate": 1.9800000000000004e-05,
      "loss": 0.2345,
      "step": 100
    },
    {
      "epoch": 0.55,
      "grad_norm": 0.6488248109817505,
      "learning_rate": 2.18e-05,
      "loss": 0.2312,
      "step": 110
    },
    {
      "epoch": 0.6,
      "grad_norm": 0.6527178883552551,
      "learning_rate": 2.38e-05,
      "loss": 0.226,
      "step": 120
    },
    {
      "epoch": 0.65,
      "grad_norm": 0.5850367546081543,
      "learning_rate": 2.58e-05,
      "loss": 0.209,
      "step": 130
    },
    {
      "epoch": 0.7,
      "grad_norm": 0.5257017016410828,
      "learning_rate": 2.7800000000000005e-05,
      "loss": 0.1967,
      "step": 140
    },
    {
      "epoch": 0.75,
      "grad_norm": 0.8061370849609375,
      "learning_rate": 2.98e-05,
      "loss": 0.1936,
      "step": 150
    },
    {
      "epoch": 0.8,
      "grad_norm": 0.5481591820716858,
      "learning_rate": 3.18e-05,
      "loss": 0.1945,
      "step": 160
    },
    {
      "epoch": 0.85,
      "grad_norm": 0.6668545603752136,
      "learning_rate": 3.38e-05,
      "loss": 0.1955,
      "step": 170
    },
    {
      "epoch": 0.9,
      "grad_norm": 0.6164191961288452,
      "learning_rate": 3.58e-05,
      "loss": 0.1872,
      "step": 180
    },
    {
      "epoch": 0.95,
      "grad_norm": 0.6763240694999695,
      "learning_rate": 3.7800000000000004e-05,
      "loss": 0.1727,
      "step": 190
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.9940104484558105,
      "learning_rate": 3.9800000000000005e-05,
      "loss": 0.1648,
      "step": 200
    },
    {
      "epoch": 1.05,
      "grad_norm": 0.6939016580581665,
      "learning_rate": 4.18e-05,
      "loss": 0.1715,
      "step": 210
    },
    {
      "epoch": 1.1,
      "grad_norm": 0.6544582843780518,
      "learning_rate": 4.38e-05,
      "loss": 0.1669,
      "step": 220
    },
    {
      "epoch": 1.15,
      "grad_norm": 0.604867160320282,
      "learning_rate": 4.58e-05,
      "loss": 0.1651,
      "step": 230
    },
    {
      "epoch": 1.2,
      "grad_norm": 0.53764808177948,
      "learning_rate": 4.78e-05,
      "loss": 0.1586,
      "step": 240
    },
    {
      "epoch": 1.25,
      "grad_norm": 0.6055057644844055,
      "learning_rate": 4.9800000000000004e-05,
      "loss": 0.1608,
      "step": 250
    },
    {
      "epoch": 1.3,
      "grad_norm": 0.5422135591506958,
      "learning_rate": 5.1800000000000005e-05,
      "loss": 0.1546,
      "step": 260
    },
    {
      "epoch": 1.35,
      "grad_norm": 0.47950276732444763,
      "learning_rate": 5.380000000000001e-05,
      "loss": 0.1575,
      "step": 270
    },
    {
      "epoch": 1.4,
      "grad_norm": 0.8649070858955383,
      "learning_rate": 5.580000000000001e-05,
      "loss": 0.1514,
      "step": 280
    },
    {
      "epoch": 1.45,
      "grad_norm": 0.7543731927871704,
      "learning_rate": 5.7799999999999995e-05,
      "loss": 0.1636,
      "step": 290
    },
    {
      "epoch": 1.5,
      "grad_norm": 0.6467975974082947,
      "learning_rate": 5.9800000000000003e-05,
      "loss": 0.1581,
      "step": 300
    },
    {
      "epoch": 1.55,
      "grad_norm": 0.5864694118499756,
      "learning_rate": 6.18e-05,
      "loss": 0.1531,
      "step": 310
    },
    {
      "epoch": 1.6,
      "grad_norm": 0.6489605903625488,
      "learning_rate": 6.38e-05,
      "loss": 0.1556,
      "step": 320
    },
    {
      "epoch": 1.65,
      "grad_norm": 0.9031095504760742,
      "learning_rate": 6.58e-05,
      "loss": 0.1557,
      "step": 330
    },
    {
      "epoch": 1.7,
      "grad_norm": 0.5535191297531128,
      "learning_rate": 6.780000000000001e-05,
      "loss": 0.1423,
      "step": 340
    },
    {
      "epoch": 1.75,
      "grad_norm": 0.6151514053344727,
      "learning_rate": 6.98e-05,
      "loss": 0.1398,
      "step": 350
    },
    {
      "epoch": 1.8,
      "grad_norm": 0.6400225758552551,
      "learning_rate": 7.18e-05,
      "loss": 0.1509,
      "step": 360
    },
    {
      "epoch": 1.85,
      "grad_norm": 0.4071156680583954,
      "learning_rate": 7.38e-05,
      "loss": 0.1491,
      "step": 370
    },
    {
      "epoch": 1.9,
      "grad_norm": 0.5420860052108765,
      "learning_rate": 7.58e-05,
      "loss": 0.1416,
      "step": 380
    },
    {
      "epoch": 1.95,
      "grad_norm": 0.5800616145133972,
      "learning_rate": 7.780000000000001e-05,
      "loss": 0.1488,
      "step": 390
    },
    {
      "epoch": 2.0,
      "grad_norm": 0.5919883847236633,
      "learning_rate": 7.98e-05,
      "loss": 0.1366,
      "step": 400
    },
    {
      "epoch": 2.05,
      "grad_norm": 0.6108524203300476,
      "learning_rate": 8.18e-05,
      "loss": 0.1336,
      "step": 410
    },
    {
      "epoch": 2.1,
      "grad_norm": 0.5259194374084473,
      "learning_rate": 8.38e-05,
      "loss": 0.1361,
      "step": 420
    },
    {
      "epoch": 2.15,
      "grad_norm": 0.6181365847587585,
      "learning_rate": 8.58e-05,
      "loss": 0.1377,
      "step": 430
    },
    {
      "epoch": 2.2,
      "grad_norm": 0.3939558267593384,
      "learning_rate": 8.78e-05,
      "loss": 0.1376,
      "step": 440
    },
    {
      "epoch": 2.25,
      "grad_norm": 0.6277884840965271,
      "learning_rate": 8.98e-05,
      "loss": 0.1389,
      "step": 450
    },
    {
      "epoch": 2.3,
      "grad_norm": 0.5376912951469421,
      "learning_rate": 9.180000000000001e-05,
      "loss": 0.1342,
      "step": 460
    },
    {
      "epoch": 2.35,
      "grad_norm": 0.41411763429641724,
      "learning_rate": 9.38e-05,
      "loss": 0.1341,
      "step": 470
    },
    {
      "epoch": 2.4,
      "grad_norm": 0.4782755970954895,
      "learning_rate": 9.58e-05,
      "loss": 0.1365,
      "step": 480
    },
    {
      "epoch": 2.45,
      "grad_norm": 0.4729580283164978,
      "learning_rate": 9.78e-05,
      "loss": 0.1311,
      "step": 490
    },
    {
      "epoch": 2.5,
      "grad_norm": 0.49572497606277466,
      "learning_rate": 9.98e-05,
      "loss": 0.1346,
      "step": 500
    },
    {
      "epoch": 2.55,
      "grad_norm": 0.5231636166572571,
      "learning_rate": 9.9999778549206e-05,
      "loss": 0.1317,
      "step": 510
    },
    {
      "epoch": 2.6,
      "grad_norm": 0.4779125452041626,
      "learning_rate": 9.999901304280685e-05,
      "loss": 0.128,
      "step": 520
    },
    {
      "epoch": 2.65,
      "grad_norm": 0.4100823700428009,
      "learning_rate": 9.999770075521164e-05,
      "loss": 0.1214,
      "step": 530
    },
    {
      "epoch": 2.7,
      "grad_norm": 0.3725630044937134,
      "learning_rate": 9.99958417007713e-05,
      "loss": 0.1252,
      "step": 540
    },
    {
      "epoch": 2.75,
      "grad_norm": 0.511320948600769,
      "learning_rate": 9.999343589981615e-05,
      "loss": 0.1211,
      "step": 550
    },
    {
      "epoch": 2.8,
      "grad_norm": 0.478473037481308,
      "learning_rate": 9.999048337865568e-05,
      "loss": 0.1203,
      "step": 560
    },
    {
      "epoch": 2.85,
      "grad_norm": 0.3342892825603485,
      "learning_rate": 9.998698416957815e-05,
      "loss": 0.123,
      "step": 570
    },
    {
      "epoch": 2.9,
      "grad_norm": 0.3094799518585205,
      "learning_rate": 9.998293831085037e-05,
      "loss": 0.1177,
      "step": 580
    },
    {
      "epoch": 2.95,
      "grad_norm": 0.5041730403900146,
      "learning_rate": 9.997834584671719e-05,
      "loss": 0.1257,
      "step": 590
    },
    {
      "epoch": 3.0,
      "grad_norm": 0.7954579591751099,
      "learning_rate": 9.997320682740107e-05,
      "loss": 0.1174,
      "step": 600
    },
    {
      "epoch": 3.05,
      "grad_norm": 0.374284565448761,
      "learning_rate": 9.996752130910149e-05,
      "loss": 0.1195,
      "step": 610
    },
    {
      "epoch": 3.1,
      "grad_norm": 0.5732347369194031,
      "learning_rate": 9.99612893539944e-05,
      "loss": 0.1247,
      "step": 620
    },
    {
      "epoch": 3.15,
      "grad_norm": 0.4805952310562134,
      "learning_rate": 9.995451103023144e-05,
      "loss": 0.1183,
      "step": 630
    },
    {
      "epoch": 3.2,
      "grad_norm": 0.3452022969722748,
      "learning_rate": 9.994718641193928e-05,
      "loss": 0.1206,
      "step": 640
    },
    {
      "epoch": 3.25,
      "grad_norm": 0.5182231068611145,
      "learning_rate": 9.993931557921874e-05,
      "loss": 0.1178,
      "step": 650
    },
    {
      "epoch": 3.3,
      "grad_norm": 0.5714840292930603,
      "learning_rate": 9.993089861814402e-05,
      "loss": 0.111,
      "step": 660
    },
    {
      "epoch": 3.35,
      "grad_norm": 0.5582299828529358,
      "learning_rate": 9.992193562076166e-05,
      "loss": 0.1204,
      "step": 670
    },
    {
      "epoch": 3.4,
      "grad_norm": 0.3832492232322693,
      "learning_rate": 9.991242668508954e-05,
      "loss": 0.1185,
      "step": 680
    },
    {
      "epoch": 3.45,
      "grad_norm": 0.594416618347168,
      "learning_rate": 9.990237191511587e-05,
      "loss": 0.1161,
      "step": 690
    },
    {
      "epoch": 3.5,
      "grad_norm": 0.4631149470806122,
      "learning_rate": 9.989177142079802e-05,
      "loss": 0.1093,
      "step": 700
    },
    {
      "epoch": 3.55,
      "grad_norm": 0.329464852809906,
      "learning_rate": 9.988062531806126e-05,
      "loss": 0.1088,
      "step": 710
    },
    {
      "epoch": 3.6,
      "grad_norm": 0.4083748459815979,
      "learning_rate": 9.986893372879762e-05,
      "loss": 0.1084,
      "step": 720
    },
    {
      "epoch": 3.65,
      "grad_norm": 0.4153082072734833,
      "learning_rate": 9.985669678086443e-05,
      "loss": 0.1182,
      "step": 730
    },
    {
      "epoch": 3.7,
      "grad_norm": 0.36365047097206116,
      "learning_rate": 9.984391460808298e-05,
      "loss": 0.1104,
      "step": 740
    },
    {
      "epoch": 3.75,
      "grad_norm": 0.3429168164730072,
      "learning_rate": 9.983058735023709e-05,
      "loss": 0.1105,
      "step": 750
    },
    {
      "epoch": 3.8,
      "grad_norm": 0.2749885022640228,
      "learning_rate": 9.98167151530715e-05,
      "loss": 0.1124,
      "step": 760
    },
    {
      "epoch": 3.85,
      "grad_norm": 0.29668092727661133,
      "learning_rate": 9.980229816829034e-05,
      "loss": 0.1071,
      "step": 770
    },
    {
      "epoch": 3.9,
      "grad_norm": 0.483247846364975,
      "learning_rate": 9.978733655355544e-05,
      "loss": 0.1108,
      "step": 780
    },
    {
      "epoch": 3.95,
      "grad_norm": 0.5551722645759583,
      "learning_rate": 9.977183047248464e-05,
      "loss": 0.1045,
      "step": 790
    },
    {
      "epoch": 4.0,
      "grad_norm": 0.51619952917099,
      "learning_rate": 9.975578009464992e-05,
      "loss": 0.1128,
      "step": 800
    },
    {
      "epoch": 4.05,
      "grad_norm": 0.3293454647064209,
      "learning_rate": 9.97391855955757e-05,
      "loss": 0.1047,
      "step": 810
    },
    {
      "epoch": 4.1,
      "grad_norm": 0.3791157007217407,
      "learning_rate": 9.972204715673669e-05,
      "loss": 0.104,
      "step": 820
    },
    {
      "epoch": 4.15,
      "grad_norm": 0.4593803584575653,
      "learning_rate": 9.970436496555617e-05,
      "loss": 0.0996,
      "step": 830
    },
    {
      "epoch": 4.2,
      "grad_norm": 0.3820911943912506,
      "learning_rate": 9.968613921540373e-05,
      "loss": 0.1069,
      "step": 840
    },
    {
      "epoch": 4.25,
      "grad_norm": 0.3202776312828064,
      "learning_rate": 9.966737010559326e-05,
      "loss": 0.106,
      "step": 850
    },
    {
      "epoch": 4.3,
      "grad_norm": 0.334024578332901,
      "learning_rate": 9.964805784138072e-05,
      "loss": 0.0997,
      "step": 860
    },
    {
      "epoch": 4.35,
      "grad_norm": 0.3140077590942383,
      "learning_rate": 9.962820263396195e-05,
      "loss": 0.0984,
      "step": 870
    },
    {
      "epoch": 4.4,
      "grad_norm": 0.3355635404586792,
      "learning_rate": 9.960780470047033e-05,
      "loss": 0.1017,
      "step": 880
    },
    {
      "epoch": 4.45,
      "grad_norm": 0.3279488682746887,
      "learning_rate": 9.958686426397437e-05,
      "loss": 0.1002,
      "step": 890
    },
    {
      "epoch": 4.5,
      "grad_norm": 0.38120609521865845,
      "learning_rate": 9.956538155347534e-05,
      "loss": 0.1026,
      "step": 900
    },
    {
      "epoch": 4.55,
      "grad_norm": 0.4751589000225067,
      "learning_rate": 9.95433568039047e-05,
      "loss": 0.1019,
      "step": 910
    },
    {
      "epoch": 4.6,
      "grad_norm": 0.44802504777908325,
      "learning_rate": 9.952079025612162e-05,
      "loss": 0.0975,
      "step": 920
    },
    {
      "epoch": 4.65,
      "grad_norm": 0.4823198616504669,
      "learning_rate": 9.949768215691022e-05,
      "loss": 0.1028,
      "step": 930
    },
    {
      "epoch": 4.7,
      "grad_norm": 0.37554022669792175,
      "learning_rate": 9.9474032758977e-05,
      "loss": 0.0984,
      "step": 940
    },
    {
      "epoch": 4.75,
      "grad_norm": 0.32095158100128174,
      "learning_rate": 9.944984232094794e-05,
      "loss": 0.1,
      "step": 950
    },
    {
      "epoch": 4.8,
      "grad_norm": 0.31493982672691345,
      "learning_rate": 9.942511110736584e-05,
      "loss": 0.0985,
      "step": 960
    },
    {
      "epoch": 4.85,
      "grad_norm": 0.4463932514190674,
      "learning_rate": 9.939983938868726e-05,
      "loss": 0.0955,
      "step": 970
    },
    {
      "epoch": 4.9,
      "grad_norm": 0.4392613470554352,
      "learning_rate": 9.93740274412797e-05,
      "loss": 0.0974,
      "step": 980
    },
    {
      "epoch": 4.95,
      "grad_norm": 0.3794855773448944,
      "learning_rate": 9.934767554741846e-05,
      "loss": 0.1001,
      "step": 990
    },
    {
      "epoch": 5.0,
      "grad_norm": 0.49750006198883057,
      "learning_rate": 9.932078399528361e-05,
      "loss": 0.0974,
      "step": 1000
    },
    {
      "mimicgen/eval/stack_three/success_rate": 0.02,
      "step": 1000
    },
    {
      "epoch": 5.05,
      "grad_norm": 0.31231364607810974,
      "learning_rate": 9.929335307895689e-05,
      "loss": 0.0958,
      "step": 1010
    },
    {
      "epoch": 5.1,
      "grad_norm": 0.34459710121154785,
      "learning_rate": 9.926538309841839e-05,
      "loss": 0.0962,
      "step": 1020
    },
    {
      "epoch": 5.15,
      "grad_norm": 0.42152777314186096,
      "learning_rate": 9.923687435954334e-05,
      "loss": 0.0996,
      "step": 1030
    },
    {
      "epoch": 5.2,
      "grad_norm": 0.4169371724128723,
      "learning_rate": 9.920782717409873e-05,
      "loss": 0.0977,
      "step": 1040
    },
    {
      "epoch": 5.25,
      "grad_norm": 0.3752817213535309,
      "learning_rate": 9.917824185973994e-05,
      "loss": 0.0987,
      "step": 1050
    },
    {
      "epoch": 5.3,
      "grad_norm": 0.27668800950050354,
      "learning_rate": 9.914811874000723e-05,
      "loss": 0.0886,
      "step": 1060
    },
    {
      "epoch": 5.35,
      "grad_norm": 0.44182252883911133,
      "learning_rate": 9.911745814432218e-05,
      "loss": 0.098,
      "step": 1070
    },
    {
      "epoch": 5.4,
      "grad_norm": 0.3632456064224243,
      "learning_rate": 9.90862604079842e-05,
      "loss": 0.093,
      "step": 1080
    },
    {
      "epoch": 5.45,
      "grad_norm": 0.3520803451538086,
      "learning_rate": 9.90545258721667e-05,
      "loss": 0.0883,
      "step": 1090
    },
    {
      "epoch": 5.5,
      "grad_norm": 0.31671738624572754,
      "learning_rate": 9.90222548839135e-05,
      "loss": 0.0943,
      "step": 1100
    },
    {
      "epoch": 5.55,
      "grad_norm": 0.3459419012069702,
      "learning_rate": 9.898944779613495e-05,
      "loss": 0.0942,
      "step": 1110
    },
    {
      "epoch": 5.6,
      "grad_norm": 0.2997388541698456,
      "learning_rate": 9.89561049676041e-05,
      "loss": 0.0913,
      "step": 1120
    },
    {
      "epoch": 5.65,
      "grad_norm": 0.30470800399780273,
      "learning_rate": 9.89222267629528e-05,
      "loss": 0.095,
      "step": 1130
    },
    {
      "epoch": 5.7,
      "grad_norm": 0.3047671318054199,
      "learning_rate": 9.888781355266763e-05,
      "loss": 0.0931,
      "step": 1140
    },
    {
      "epoch": 5.75,
      "grad_norm": 0.2551024556159973,
      "learning_rate": 9.885286571308598e-05,
      "loss": 0.0914,
      "step": 1150
    },
    {
      "epoch": 5.8,
      "grad_norm": 0.3499239981174469,
      "learning_rate": 9.881738362639182e-05,
      "loss": 0.0942,
      "step": 1160
    },
    {
      "epoch": 5.85,
      "grad_norm": 0.28688231110572815,
      "learning_rate": 9.878136768061154e-05,
      "loss": 0.0982,
      "step": 1170
    },
    {
      "epoch": 5.9,
      "grad_norm": 0.2133469134569168,
      "learning_rate": 9.874481826960979e-05,
      "loss": 0.0887,
      "step": 1180
    },
    {
      "epoch": 5.95,
      "grad_norm": 0.280164510011673,
      "learning_rate": 9.870773579308503e-05,
      "loss": 0.0909,
      "step": 1190
    },
    {
      "epoch": 6.0,
      "grad_norm": 0.31730639934539795,
      "learning_rate": 9.867012065656533e-05,
      "loss": 0.0855,
      "step": 1200
    },
    {
      "epoch": 6.05,
      "grad_norm": 0.37769395112991333,
      "learning_rate": 9.863197327140376e-05,
      "loss": 0.0882,
      "step": 1210
    },
    {
      "epoch": 6.1,
      "grad_norm": 0.283701092004776,
      "learning_rate": 9.859329405477403e-05,
      "loss": 0.0902,
      "step": 1220
    },
    {
      "epoch": 6.15,
      "grad_norm": 0.3801388442516327,
      "learning_rate": 9.855408342966585e-05,
      "loss": 0.0863,
      "step": 1230
    },
    {
      "epoch": 6.2,
      "grad_norm": 0.25928717851638794,
      "learning_rate": 9.851434182488033e-05,
      "loss": 0.0909,
      "step": 1240
    },
    {
      "epoch": 6.25,
      "grad_norm": 0.33434000611305237,
      "learning_rate": 9.84740696750253e-05,
      "loss": 0.0879,
      "step": 1250
    },
    {
      "epoch": 6.3,
      "grad_norm": 0.35260024666786194,
      "learning_rate": 9.843326742051055e-05,
      "loss": 0.0901,
      "step": 1260
    },
    {
      "epoch": 6.35,
      "grad_norm": 0.33025026321411133,
      "learning_rate": 9.839193550754297e-05,
      "loss": 0.0875,
      "step": 1270
    },
    {
      "epoch": 6.4,
      "grad_norm": 0.29212647676467896,
      "learning_rate": 9.835007438812177e-05,
      "loss": 0.0921,
      "step": 1280
    },
    {
      "epoch": 6.45,
      "grad_norm": 0.37930721044540405,
      "learning_rate": 9.830768452003341e-05,
      "loss": 0.0878,
      "step": 1290
    },
    {
      "epoch": 6.5,
      "grad_norm": 0.3575275242328644,
      "learning_rate": 9.826476636684671e-05,
      "loss": 0.0859,
      "step": 1300
    },
    {
      "epoch": 6.55,
      "grad_norm": 0.33191609382629395,
      "learning_rate": 9.822132039790773e-05,
      "loss": 0.0907,
      "step": 1310
    },
    {
      "epoch": 6.6,
      "grad_norm": 0.2645428478717804,
      "learning_rate": 9.817734708833461e-05,
      "loss": 0.0863,
      "step": 1320
    },
    {
      "epoch": 6.65,
      "grad_norm": 0.20752987265586853,
      "learning_rate": 9.813284691901243e-05,
      "loss": 0.0926,
      "step": 1330
    },
    {
      "epoch": 6.7,
      "grad_norm": 0.2601943910121918,
      "learning_rate": 9.808782037658792e-05,
      "loss": 0.0864,
      "step": 1340
    },
    {
      "epoch": 6.75,
      "grad_norm": 0.4597942531108856,
      "learning_rate": 9.804226795346411e-05,
      "loss": 0.0896,
      "step": 1350
    },
    {
      "epoch": 6.8,
      "grad_norm": 0.25841736793518066,
      "learning_rate": 9.799619014779503e-05,
      "loss": 0.0863,
      "step": 1360
    },
    {
      "epoch": 6.85,
      "grad_norm": 0.3548521101474762,
      "learning_rate": 9.794958746348013e-05,
      "loss": 0.0886,
      "step": 1370
    },
    {
      "epoch": 6.9,
      "grad_norm": 0.43177253007888794,
      "learning_rate": 9.790246041015896e-05,
      "loss": 0.0839,
      "step": 1380
    },
    {
      "epoch": 6.95,
      "grad_norm": 0.613959014415741,
      "learning_rate": 9.785480950320538e-05,
      "loss": 0.0817,
      "step": 1390
    },
    {
      "epoch": 7.0,
      "grad_norm": 0.5007033348083496,
      "learning_rate": 9.78066352637221e-05,
      "loss": 0.0839,
      "step": 1400
    },
    {
      "epoch": 7.05,
      "grad_norm": 0.37768879532814026,
      "learning_rate": 9.775793821853488e-05,
      "loss": 0.0914,
      "step": 1410
    },
    {
      "epoch": 7.1,
      "grad_norm": 0.30859270691871643,
      "learning_rate": 9.77087189001868e-05,
      "loss": 0.0896,
      "step": 1420
    },
    {
      "epoch": 7.15,
      "grad_norm": 0.28924018144607544,
      "learning_rate": 9.765897784693243e-05,
      "loss": 0.0897,
      "step": 1430
    },
    {
      "epoch": 7.2,
      "grad_norm": 0.3883243203163147,
      "learning_rate": 9.760871560273197e-05,
      "loss": 0.09,
      "step": 1440
    },
    {
      "epoch": 7.25,
      "grad_norm": 0.26877668499946594,
      "learning_rate": 9.755793271724526e-05,
      "loss": 0.0872,
      "step": 1450
    },
    {
      "epoch": 7.3,
      "grad_norm": 0.26304537057876587,
      "learning_rate": 9.750662974582584e-05,
      "loss": 0.0839,
      "step": 1460
    },
    {
      "epoch": 7.35,
      "grad_norm": 0.24647530913352966,
      "learning_rate": 9.745480724951473e-05,
      "loss": 0.0867,
      "step": 1470
    },
    {
      "epoch": 7.4,
      "grad_norm": 0.6214026808738708,
      "learning_rate": 9.740246579503447e-05,
      "loss": 0.0897,
      "step": 1480
    },
    {
      "epoch": 7.45,
      "grad_norm": 0.4697532057762146,
      "learning_rate": 9.734960595478284e-05,
      "loss": 0.0872,
      "step": 1490
    },
    {
      "epoch": 7.5,
      "grad_norm": 0.23033514618873596,
      "learning_rate": 9.729622830682657e-05,
      "loss": 0.0763,
      "step": 1500
    },
    {
      "epoch": 7.55,
      "grad_norm": 0.3114646077156067,
      "learning_rate": 9.724233343489504e-05,
      "loss": 0.0818,
      "step": 1510
    },
    {
      "epoch": 7.6,
      "grad_norm": 0.2765814960002899,
      "learning_rate": 9.718792192837396e-05,
      "loss": 0.0767,
      "step": 1520
    },
    {
      "epoch": 7.65,
      "grad_norm": 0.2841011881828308,
      "learning_rate": 9.713299438229886e-05,
      "loss": 0.0811,
      "step": 1530
    },
    {
      "epoch": 7.7,
      "grad_norm": 0.2892618179321289,
      "learning_rate": 9.707755139734855e-05,
      "loss": 0.0802,
      "step": 1540
    },
    {
      "epoch": 7.75,
      "grad_norm": 0.27881234884262085,
      "learning_rate": 9.702159357983866e-05,
      "loss": 0.0858,
      "step": 1550
    },
    {
      "epoch": 7.8,
      "grad_norm": 0.3085121810436249,
      "learning_rate": 9.696512154171492e-05,
      "loss": 0.0819,
      "step": 1560
    },
    {
      "epoch": 7.85,
      "grad_norm": 0.2841378450393677,
      "learning_rate": 9.690813590054645e-05,
      "loss": 0.082,
      "step": 1570
    },
    {
      "epoch": 7.9,
      "grad_norm": 0.30185776948928833,
      "learning_rate": 9.685063727951914e-05,
      "loss": 0.0833,
      "step": 1580
    },
    {
      "epoch": 7.95,
      "grad_norm": 0.2795591950416565,
      "learning_rate": 9.679262630742865e-05,
      "loss": 0.0749,
      "step": 1590
    },
    {
      "epoch": 8.0,
      "grad_norm": 0.45515498518943787,
      "learning_rate": 9.673410361867373e-05,
      "loss": 0.0847,
      "step": 1600
    },
    {
      "epoch": 8.05,
      "grad_norm": 0.34494543075561523,
      "learning_rate": 9.667506985324909e-05,
      "loss": 0.0797,
      "step": 1610
    },
    {
      "epoch": 8.1,
      "grad_norm": 0.33676546812057495,
      "learning_rate": 9.661552565673855e-05,
      "loss": 0.0766,
      "step": 1620
    },
    {
      "epoch": 8.15,
      "grad_norm": 0.30524742603302,
      "learning_rate": 9.655547168030789e-05,
      "loss": 0.0799,
      "step": 1630
    },
    {
      "epoch": 8.2,
      "grad_norm": 0.27557501196861267,
      "learning_rate": 9.649490858069777e-05,
      "loss": 0.082,
      "step": 1640
    },
    {
      "epoch": 8.25,
      "grad_norm": 0.3024428188800812,
      "learning_rate": 9.643383702021658e-05,
      "loss": 0.0859,
      "step": 1650
    },
    {
      "epoch": 8.3,
      "grad_norm": 0.30283573269844055,
      "learning_rate": 9.637225766673307e-05,
      "loss": 0.0797,
      "step": 1660
    },
    {
      "epoch": 8.35,
      "grad_norm": 0.35822656750679016,
      "learning_rate": 9.631017119366922e-05,
      "loss": 0.0776,
      "step": 1670
    },
    {
      "epoch": 8.4,
      "grad_norm": 0.2784837484359741,
      "learning_rate": 9.624757827999273e-05,
      "loss": 0.0783,
      "step": 1680
    },
    {
      "epoch": 8.45,
      "grad_norm": 0.2793690860271454,
      "learning_rate": 9.618447961020971e-05,
      "loss": 0.0813,
      "step": 1690
    },
    {
      "epoch": 8.5,
      "grad_norm": 0.2518792152404785,
      "learning_rate": 9.612087587435707e-05,
      "loss": 0.0728,
      "step": 1700
    },
    {
      "epoch": 8.55,
      "grad_norm": 0.2861917316913605,
      "learning_rate": 9.605676776799508e-05,
      "loss": 0.0739,
      "step": 1710
    },
    {
      "epoch": 8.6,
      "grad_norm": 0.23168425261974335,
      "learning_rate": 9.599215599219973e-05,
      "loss": 0.0771,
      "step": 1720
    },
    {
      "epoch": 8.65,
      "grad_norm": 0.2521399259567261,
      "learning_rate": 9.592704125355505e-05,
      "loss": 0.0821,
      "step": 1730
    },
    {
      "epoch": 8.7,
      "grad_norm": 0.3736461102962494,
      "learning_rate": 9.586142426414538e-05,
      "loss": 0.0785,
      "step": 1740
    },
    {
      "epoch": 8.75,
      "grad_norm": 0.27999347448349,
      "learning_rate": 9.57953057415476e-05,
      "loss": 0.074,
      "step": 1750
    },
    {
      "epoch": 8.8,
      "grad_norm": 0.3226291239261627,
      "learning_rate": 9.572868640882328e-05,
      "loss": 0.074,
      "step": 1760
    },
    {
      "epoch": 8.85,
      "grad_norm": 0.27279022336006165,
      "learning_rate": 9.56615669945108e-05,
      "loss": 0.081,
      "step": 1770
    },
    {
      "epoch": 8.9,
      "grad_norm": 0.41367554664611816,
      "learning_rate": 9.55939482326173e-05,
      "loss": 0.0727,
      "step": 1780
    },
    {
      "epoch": 8.95,
      "grad_norm": 0.3348133862018585,
      "learning_rate": 9.552583086261069e-05,
      "loss": 0.0817,
      "step": 1790
    },
    {
      "epoch": 9.0,
      "grad_norm": 0.39675143361091614,
      "learning_rate": 9.545721562941168e-05,
      "loss": 0.0777,
      "step": 1800
    },
    {
      "epoch": 9.05,
      "grad_norm": 0.331039696931839,
      "learning_rate": 9.538810328338543e-05,
      "loss": 0.0767,
      "step": 1810
    },
    {
      "epoch": 9.1,
      "grad_norm": 0.2716194987297058,
      "learning_rate": 9.531849458033349e-05,
      "loss": 0.0777,
      "step": 1820
    },
    {
      "epoch": 9.15,
      "grad_norm": 0.2176772505044937,
      "learning_rate": 9.524839028148547e-05,
      "loss": 0.0766,
      "step": 1830
    },
    {
      "epoch": 9.2,
      "grad_norm": 0.3065796196460724,
      "learning_rate": 9.517779115349077e-05,
      "loss": 0.0737,
      "step": 1840
    },
    {
      "epoch": 9.25,
      "grad_norm": 0.33548703789711,
      "learning_rate": 9.510669796841014e-05,
      "loss": 0.0743,
      "step": 1850
    },
    {
      "epoch": 9.3,
      "grad_norm": 0.356990247964859,
      "learning_rate": 9.503511150370727e-05,
      "loss": 0.0704,
      "step": 1860
    },
    {
      "epoch": 9.35,
      "grad_norm": 0.305056631565094,
      "learning_rate": 9.496303254224024e-05,
      "loss": 0.0778,
      "step": 1870
    },
    {
      "epoch": 9.4,
      "grad_norm": 0.21457155048847198,
      "learning_rate": 9.489046187225306e-05,
      "loss": 0.0758,
      "step": 1880
    },
    {
      "epoch": 9.45,
      "grad_norm": 0.26219892501831055,
      "learning_rate": 9.481740028736692e-05,
      "loss": 0.0771,
      "step": 1890
    },
    {
      "epoch": 9.5,
      "grad_norm": 0.21693368256092072,
      "learning_rate": 9.474384858657164e-05,
      "loss": 0.0747,
      "step": 1900
    },
    {
      "epoch": 9.55,
      "grad_norm": 0.26491591334342957,
      "learning_rate": 9.466980757421679e-05,
      "loss": 0.0721,
      "step": 1910
    },
    {
      "epoch": 9.6,
      "grad_norm": 0.25640758872032166,
      "learning_rate": 9.459527806000305e-05,
      "loss": 0.076,
      "step": 1920
    },
    {
      "epoch": 9.65,
      "grad_norm": 0.30470454692840576,
      "learning_rate": 9.452026085897325e-05,
      "loss": 0.0735,
      "step": 1930
    },
    {
      "epoch": 9.7,
      "grad_norm": 0.203828826546669,
      "learning_rate": 9.444475679150348e-05,
      "loss": 0.0759,
      "step": 1940
    },
    {
      "epoch": 9.75,
      "grad_norm": 0.32379215955734253,
      "learning_rate": 9.436876668329411e-05,
      "loss": 0.0712,
      "step": 1950
    },
    {
      "epoch": 9.8,
      "grad_norm": 0.31410378217697144,
      "learning_rate": 9.429229136536079e-05,
      "loss": 0.0757,
      "step": 1960
    },
    {
      "epoch": 9.85,
      "grad_norm": 0.3540605306625366,
      "learning_rate": 9.421533167402534e-05,
      "loss": 0.0732,
      "step": 1970
    },
    {
      "epoch": 9.9,
      "grad_norm": 0.36735063791275024,
      "learning_rate": 9.413788845090666e-05,
      "loss": 0.0794,
      "step": 1980
    },
    {
      "epoch": 9.95,
      "grad_norm": 0.2954480051994324,
      "learning_rate": 9.405996254291136e-05,
      "loss": 0.0802,
      "step": 1990
    },
    {
      "epoch": 10.0,
      "grad_norm": 0.37434396147727966,
      "learning_rate": 9.398155480222474e-05,
      "loss": 0.0786,
      "step": 2000
    },
    {
      "mimicgen/eval/stack_three/success_rate": 0.16,
      "step": 2000
    },
    {
      "epoch": 10.05,
      "grad_norm": 0.315305233001709,
      "learning_rate": 9.390266608630128e-05,
      "loss": 0.0784,
      "step": 2010
    },
    {
      "epoch": 10.1,
      "grad_norm": 0.30153176188468933,
      "learning_rate": 9.38232972578553e-05,
      "loss": 0.0728,
      "step": 2020
    },
    {
      "epoch": 10.15,
      "grad_norm": 0.2791180908679962,
      "learning_rate": 9.374344918485164e-05,
      "loss": 0.0667,
      "step": 2030
    },
    {
      "epoch": 10.2,
      "grad_norm": 0.3643386662006378,
      "learning_rate": 9.366312274049602e-05,
      "loss": 0.0733,
      "step": 2040
    },
    {
      "epoch": 10.25,
      "grad_norm": 0.28011080622673035,
      "learning_rate": 9.358231880322554e-05,
      "loss": 0.075,
      "step": 2050
    },
    {
      "epoch": 10.3,
      "grad_norm": 0.26816317439079285,
      "learning_rate": 9.350103825669916e-05,
      "loss": 0.0745,
      "step": 2060
    },
    {
      "epoch": 10.35,
      "grad_norm": 0.22580718994140625,
      "learning_rate": 9.341928198978787e-05,
      "loss": 0.0745,
      "step": 2070
    },
    {
      "epoch": 10.4,
      "grad_norm": 0.18741004168987274,
      "learning_rate": 9.333705089656512e-05,
      "loss": 0.0725,
      "step": 2080
    },
    {
      "epoch": 10.45,
      "grad_norm": 0.2767433524131775,
      "learning_rate": 9.325434587629698e-05,
      "loss": 0.0752,
      "step": 2090
    },
    {
      "epoch": 10.5,
      "grad_norm": 0.22656016051769257,
      "learning_rate": 9.31711678334323e-05,
      "loss": 0.0717,
      "step": 2100
    },
    {
      "epoch": 10.55,
      "grad_norm": 0.2846101224422455,
      "learning_rate": 9.308751767759282e-05,
      "loss": 0.0709,
      "step": 2110
    },
    {
      "epoch": 10.6,
      "grad_norm": 0.2476411610841751,
      "learning_rate": 9.300339632356325e-05,
      "loss": 0.0694,
      "step": 2120
    },
    {
      "epoch": 10.65,
      "grad_norm": 0.24221138656139374,
      "learning_rate": 9.291880469128124e-05,
      "loss": 0.0727,
      "step": 2130
    },
    {
      "epoch": 10.7,
      "grad_norm": 0.27998360991477966,
      "learning_rate": 9.283374370582732e-05,
      "loss": 0.0741,
      "step": 2140
    },
    {
      "epoch": 10.75,
      "grad_norm": 0.2537428140640259,
      "learning_rate": 9.274821429741482e-05,
      "loss": 0.0693,
      "step": 2150
    },
    {
      "epoch": 10.8,
      "grad_norm": 0.2726587653160095,
      "learning_rate": 9.266221740137961e-05,
      "loss": 0.0761,
      "step": 2160
    },
    {
      "epoch": 10.85,
      "grad_norm": 0.3610733449459076,
      "learning_rate": 9.257575395817001e-05,
      "loss": 0.0702,
      "step": 2170
    },
    {
      "epoch": 10.9,
      "grad_norm": 0.4682345986366272,
      "learning_rate": 9.248882491333637e-05,
      "loss": 0.0721,
      "step": 2180
    },
    {
      "epoch": 10.95,
      "grad_norm": 0.2736009955406189,
      "learning_rate": 9.240143121752076e-05,
      "loss": 0.0729,
      "step": 2190
    },
    {
      "epoch": 11.0,
      "grad_norm": 0.3351520299911499,
      "learning_rate": 9.23135738264467e-05,
      "loss": 0.0749,
      "step": 2200
    },
    {
      "epoch": 11.05,
      "grad_norm": 0.34215888381004333,
      "learning_rate": 9.222525370090849e-05,
      "loss": 0.0777,
      "step": 2210
    },
    {
      "epoch": 11.1,
      "grad_norm": 0.32070812582969666,
      "learning_rate": 9.213647180676088e-05,
      "loss": 0.0731,
      "step": 2220
    },
    {
      "epoch": 11.15,
      "grad_norm": 0.3347787857055664,
      "learning_rate": 9.204722911490846e-05,
      "loss": 0.0715,
      "step": 2230
    },
    {
      "epoch": 11.2,
      "grad_norm": 0.28402262926101685,
      "learning_rate": 9.1957526601295e-05,
      "loss": 0.0669,
      "step": 2240
    },
    {
      "epoch": 11.25,
      "grad_norm": 0.34232524037361145,
      "learning_rate": 9.186736524689281e-05,
      "loss": 0.0699,
      "step": 2250
    },
    {
      "epoch": 11.3,
      "grad_norm": 0.41841474175453186,
      "learning_rate": 9.177674603769204e-05,
      "loss": 0.0695,
      "step": 2260
    },
    {
      "epoch": 11.35,
      "grad_norm": 0.3261333405971527,
      "learning_rate": 9.168566996468983e-05,
      "loss": 0.0652,
      "step": 2270
    },
    {
      "epoch": 11.4,
      "grad_norm": 0.3083556890487671,
      "learning_rate": 9.159413802387951e-05,
      "loss": 0.0688,
      "step": 2280
    },
    {
      "epoch": 11.45,
      "grad_norm": 0.31208536028862,
      "learning_rate": 9.150215121623974e-05,
      "loss": 0.068,
      "step": 2290
    },
    {
      "epoch": 11.5,
      "grad_norm": 0.2832343578338623,
      "learning_rate": 9.140971054772349e-05,
      "loss": 0.077,
      "step": 2300
    },
    {
      "epoch": 11.55,
      "grad_norm": 0.30020684003829956,
      "learning_rate": 9.131681702924713e-05,
      "loss": 0.07,
      "step": 2310
    },
    {
      "epoch": 11.6,
      "grad_norm": 0.28911808133125305,
      "learning_rate": 9.122347167667926e-05,
      "loss": 0.0722,
      "step": 2320
    },
    {
      "epoch": 11.65,
      "grad_norm": 0.4736734628677368,
      "learning_rate": 9.112967551082973e-05,
      "loss": 0.0664,
      "step": 2330
    },
    {
      "epoch": 11.7,
      "grad_norm": 0.3362722098827362,
      "learning_rate": 9.103542955743835e-05,
      "loss": 0.075,
      "step": 2340
    },
    {
      "epoch": 11.75,
      "grad_norm": 0.21794366836547852,
      "learning_rate": 9.094073484716381e-05,
      "loss": 0.0716,
      "step": 2350
    },
    {
      "epoch": 11.8,
      "grad_norm": 0.26827946305274963,
      "learning_rate": 9.084559241557226e-05,
      "loss": 0.0698,
      "step": 2360
    },
    {
      "epoch": 11.85,
      "grad_norm": 0.4168849587440491,
      "learning_rate": 9.075000330312608e-05,
      "loss": 0.0709,
      "step": 2370
    },
    {
      "epoch": 11.9,
      "grad_norm": 0.32936570048332214,
      "learning_rate": 9.065396855517253e-05,
      "loss": 0.0738,
      "step": 2380
    },
    {
      "epoch": 11.95,
      "grad_norm": 0.25441545248031616,
      "learning_rate": 9.055748922193219e-05,
      "loss": 0.0723,
      "step": 2390
    },
    {
      "epoch": 12.0,
      "grad_norm": 0.4008967876434326,
      "learning_rate": 9.046056635848761e-05,
      "loss": 0.0732,
      "step": 2400
    },
    {
      "epoch": 12.05,
      "grad_norm": 0.3557305335998535,
      "learning_rate": 9.036320102477169e-05,
      "loss": 0.077,
      "step": 2410
    },
    {
      "epoch": 12.1,
      "grad_norm": 0.22185827791690826,
      "learning_rate": 9.02653942855561e-05,
      "loss": 0.0706,
      "step": 2420
    },
    {
      "epoch": 12.15,
      "grad_norm": 0.2042110115289688,
      "learning_rate": 9.016714721043971e-05,
      "loss": 0.0704,
      "step": 2430
    },
    {
      "epoch": 12.2,
      "grad_norm": 0.294768363237381,
      "learning_rate": 9.006846087383675e-05,
      "loss": 0.0704,
      "step": 2440
    },
    {
      "epoch": 12.25,
      "grad_norm": 0.33640989661216736,
      "learning_rate": 8.996933635496523e-05,
      "loss": 0.0694,
      "step": 2450
    },
    {
      "epoch": 12.3,
      "grad_norm": 0.28535789251327515,
      "learning_rate": 8.986977473783498e-05,
      "loss": 0.0685,
      "step": 2460
    },
    {
      "epoch": 12.35,
      "grad_norm": 0.3316439390182495,
      "learning_rate": 8.97697771112359e-05,
      "loss": 0.0765,
      "step": 2470
    },
    {
      "epoch": 12.4,
      "grad_norm": 0.22241409122943878,
      "learning_rate": 8.966934456872602e-05,
      "loss": 0.0727,
      "step": 2480
    },
    {
      "epoch": 12.45,
      "grad_norm": 0.2602394223213196,
      "learning_rate": 8.95684782086195e-05,
      "loss": 0.0691,
      "step": 2490
    },
    {
      "epoch": 12.5,
      "grad_norm": 0.26225581765174866,
      "learning_rate": 8.946717913397476e-05,
      "loss": 0.0705,
      "step": 2500
    },
    {
      "epoch": 12.55,
      "grad_norm": 0.36338135600090027,
      "learning_rate": 8.93654484525822e-05,
      "loss": 0.0694,
      "step": 2510
    },
    {
      "epoch": 12.6,
      "grad_norm": 0.3044951558113098,
      "learning_rate": 8.926328727695226e-05,
      "loss": 0.0707,
      "step": 2520
    },
    {
      "epoch": 12.65,
      "grad_norm": 0.3144603967666626,
      "learning_rate": 8.916069672430319e-05,
      "loss": 0.0636,
      "step": 2530
    },
    {
      "epoch": 12.7,
      "grad_norm": 0.5535969734191895,
      "learning_rate": 8.905767791654884e-05,
      "loss": 0.0669,
      "step": 2540
    },
    {
      "epoch": 12.75,
      "grad_norm": 0.3151090145111084,
      "learning_rate": 8.895423198028638e-05,
      "loss": 0.0706,
      "step": 2550
    },
    {
      "epoch": 12.8,
      "grad_norm": 0.3438064754009247,
      "learning_rate": 8.885036004678402e-05,
      "loss": 0.0705,
      "step": 2560
    },
    {
      "epoch": 12.85,
      "grad_norm": 0.3122187852859497,
      "learning_rate": 8.874606325196857e-05,
      "loss": 0.0698,
      "step": 2570
    },
    {
      "epoch": 12.9,
      "grad_norm": 0.32961440086364746,
      "learning_rate": 8.864134273641304e-05,
      "loss": 0.0726,
      "step": 2580
    },
    {
      "epoch": 12.95,
      "grad_norm": 0.21093641221523285,
      "learning_rate": 8.853619964532427e-05,
      "loss": 0.0659,
      "step": 2590
    },
    {
      "epoch": 13.0,
      "grad_norm": 0.3746702969074249,
      "learning_rate": 8.843063512853019e-05,
      "loss": 0.0675,
      "step": 2600
    },
    {
      "epoch": 13.05,
      "grad_norm": 0.4066503942012787,
      "learning_rate": 8.832465034046749e-05,
      "loss": 0.0658,
      "step": 2610
    },
    {
      "epoch": 13.1,
      "grad_norm": 0.25783559679985046,
      "learning_rate": 8.821824644016882e-05,
      "loss": 0.0681,
      "step": 2620
    },
    {
      "epoch": 13.15,
      "grad_norm": 0.253420889377594,
      "learning_rate": 8.811142459125019e-05,
      "loss": 0.0657,
      "step": 2630
    },
    {
      "epoch": 13.2,
      "grad_norm": 0.2950253188610077,
      "learning_rate": 8.800418596189822e-05,
      "loss": 0.0697,
      "step": 2640
    },
    {
      "epoch": 13.25,
      "grad_norm": 0.2699006497859955,
      "learning_rate": 8.789653172485737e-05,
      "loss": 0.0637,
      "step": 2650
    },
    {
      "epoch": 13.3,
      "grad_norm": 0.1790885329246521,
      "learning_rate": 8.778846305741715e-05,
      "loss": 0.0631,
      "step": 2660
    },
    {
      "epoch": 13.35,
      "grad_norm": 0.19111207127571106,
      "learning_rate": 8.767998114139918e-05,
      "loss": 0.0631,
      "step": 2670
    },
    {
      "epoch": 13.4,
      "grad_norm": 0.28023600578308105,
      "learning_rate": 8.757108716314429e-05,
      "loss": 0.0608,
      "step": 2680
    },
    {
      "epoch": 13.45,
      "grad_norm": 0.38140079379081726,
      "learning_rate": 8.746178231349962e-05,
      "loss": 0.0628,
      "step": 2690
    },
    {
      "epoch": 13.5,
      "grad_norm": 0.23159313201904297,
      "learning_rate": 8.735206778780549e-05,
      "loss": 0.0672,
      "step": 2700
    },
    {
      "epoch": 13.55,
      "grad_norm": 0.34765559434890747,
      "learning_rate": 8.724194478588234e-05,
      "loss": 0.0667,
      "step": 2710
    },
    {
      "epoch": 13.6,
      "grad_norm": 0.40895354747772217,
      "learning_rate": 8.713141451201772e-05,
      "loss": 0.066,
      "step": 2720
    },
    {
      "epoch": 13.65,
      "grad_norm": 0.27039822936058044,
      "learning_rate": 8.702047817495295e-05,
      "loss": 0.0622,
      "step": 2730
    },
    {
      "epoch": 13.7,
      "grad_norm": 0.33033761382102966,
      "learning_rate": 8.69091369878701e-05,
      "loss": 0.0628,
      "step": 2740
    },
    {
      "epoch": 13.75,
      "grad_norm": 0.3204769492149353,
      "learning_rate": 8.679739216837849e-05,
      "loss": 0.0676,
      "step": 2750
    },
    {
      "epoch": 13.8,
      "grad_norm": 0.32906439900398254,
      "learning_rate": 8.66852449385016e-05,
      "loss": 0.0694,
      "step": 2760
    },
    {
      "epoch": 13.85,
      "grad_norm": 0.279493123292923,
      "learning_rate": 8.657269652466356e-05,
      "loss": 0.0647,
      "step": 2770
    },
    {
      "epoch": 13.9,
      "grad_norm": 0.31332072615623474,
      "learning_rate": 8.645974815767577e-05,
      "loss": 0.0665,
      "step": 2780
    },
    {
      "epoch": 13.95,
      "grad_norm": 0.30920207500457764,
      "learning_rate": 8.634640107272351e-05,
      "loss": 0.0702,
      "step": 2790
    },
    {
      "epoch": 14.0,
      "grad_norm": 0.39714086055755615,
      "learning_rate": 8.623265650935234e-05,
      "loss": 0.0721,
      "step": 2800
    },
    {
      "epoch": 14.05,
      "grad_norm": 0.344400018453598,
      "learning_rate": 8.611851571145456e-05,
      "loss": 0.0705,
      "step": 2810
    },
    {
      "epoch": 14.1,
      "grad_norm": 0.27441877126693726,
      "learning_rate": 8.600397992725566e-05,
      "loss": 0.0689,
      "step": 2820
    },
    {
      "epoch": 14.15,
      "grad_norm": 0.29878881573677063,
      "learning_rate": 8.588905040930061e-05,
      "loss": 0.0632,
      "step": 2830
    },
    {
      "epoch": 14.2,
      "grad_norm": 0.29191383719444275,
      "learning_rate": 8.577372841444022e-05,
      "loss": 0.0657,
      "step": 2840
    },
    {
      "epoch": 14.25,
      "grad_norm": 0.3657996952533722,
      "learning_rate": 8.565801520381736e-05,
      "loss": 0.065,
      "step": 2850
    },
    {
      "epoch": 14.3,
      "grad_norm": 0.2532500624656677,
      "learning_rate": 8.554191204285313e-05,
      "loss": 0.0632,
      "step": 2860
    },
    {
      "epoch": 14.35,
      "grad_norm": 0.3889797329902649,
      "learning_rate": 8.542542020123315e-05,
      "loss": 0.0674,
      "step": 2870
    },
    {
      "epoch": 14.4,
      "grad_norm": 0.2500646710395813,
      "learning_rate": 8.530854095289347e-05,
      "loss": 0.0658,
      "step": 2880
    },
    {
      "epoch": 14.45,
      "grad_norm": 0.2977831959724426,
      "learning_rate": 8.519127557600688e-05,
      "loss": 0.061,
      "step": 2890
    },
    {
      "epoch": 14.5,
      "grad_norm": 0.27110958099365234,
      "learning_rate": 8.507362535296871e-05,
      "loss": 0.064,
      "step": 2900
    },
    {
      "epoch": 14.55,
      "grad_norm": 0.28830111026763916,
      "learning_rate": 8.495559157038299e-05,
      "loss": 0.0664,
      "step": 2910
    },
    {
      "epoch": 14.6,
      "grad_norm": 0.258830189704895,
      "learning_rate": 8.483717551904823e-05,
      "loss": 0.0632,
      "step": 2920
    },
    {
      "epoch": 14.65,
      "grad_norm": 0.29505327343940735,
      "learning_rate": 8.47183784939434e-05,
      "loss": 0.0638,
      "step": 2930
    },
    {
      "epoch": 14.7,
      "grad_norm": 0.3200320303440094,
      "learning_rate": 8.459920179421374e-05,
      "loss": 0.0643,
      "step": 2940
    },
    {
      "epoch": 14.75,
      "grad_norm": 0.34875309467315674,
      "learning_rate": 8.447964672315656e-05,
      "loss": 0.0586,
      "step": 2950
    },
    {
      "epoch": 14.8,
      "grad_norm": 0.4562538266181946,
      "learning_rate": 8.435971458820692e-05,
      "loss": 0.0648,
      "step": 2960
    },
    {
      "epoch": 14.85,
      "grad_norm": 0.511403501033783,
      "learning_rate": 8.423940670092345e-05,
      "loss": 0.0662,
      "step": 2970
    },
    {
      "epoch": 14.9,
      "grad_norm": 0.39596399664878845,
      "learning_rate": 8.411872437697394e-05,
      "loss": 0.0652,
      "step": 2980
    },
    {
      "epoch": 14.95,
      "grad_norm": 0.206265389919281,
      "learning_rate": 8.399766893612096e-05,
      "loss": 0.0644,
      "step": 2990
    },
    {
      "epoch": 15.0,
      "grad_norm": 0.4106707274913788,
      "learning_rate": 8.38762417022074e-05,
      "loss": 0.0632,
      "step": 3000
    },
    {
      "mimicgen/eval/stack_three/success_rate": 0.2,
      "step": 3000
    },
    {
      "epoch": 15.05,
      "grad_norm": 0.296929270029068,
      "learning_rate": 8.375444400314204e-05,
      "loss": 0.0632,
      "step": 3010
    },
    {
      "epoch": 15.1,
      "grad_norm": 0.2239900827407837,
      "learning_rate": 8.3632277170885e-05,
      "loss": 0.0641,
      "step": 3020
    },
    {
      "epoch": 15.15,
      "grad_norm": 0.2684023380279541,
      "learning_rate": 8.350974254143318e-05,
      "loss": 0.0634,
      "step": 3030
    },
    {
      "epoch": 15.2,
      "grad_norm": 0.20909832417964935,
      "learning_rate": 8.338684145480566e-05,
      "loss": 0.061,
      "step": 3040
    },
    {
      "epoch": 15.25,
      "grad_norm": 0.5397847890853882,
      "learning_rate": 8.326357525502904e-05,
      "loss": 0.0671,
      "step": 3050
    },
    {
      "epoch": 15.3,
      "grad_norm": 0.42903128266334534,
      "learning_rate": 8.313994529012273e-05,
      "loss": 0.0643,
      "step": 3060
    },
    {
      "epoch": 15.35,
      "grad_norm": 0.3745482563972473,
      "learning_rate": 8.301595291208422e-05,
      "loss": 0.0664,
      "step": 3070
    },
    {
      "epoch": 15.4,
      "grad_norm": 0.320884644985199,
      "learning_rate": 8.289159947687427e-05,
      "loss": 0.0672,
      "step": 3080
    },
    {
      "epoch": 15.45,
      "grad_norm": 0.34070494771003723,
      "learning_rate": 8.276688634440216e-05,
      "loss": 0.0604,
      "step": 3090
    },
    {
      "epoch": 15.5,
      "grad_norm": 0.36394786834716797,
      "learning_rate": 8.26418148785107e-05,
      "loss": 0.059,
      "step": 3100
    },
    {
      "epoch": 15.55,
      "grad_norm": 0.17489610612392426,
      "learning_rate": 8.251638644696141e-05,
      "loss": 0.0669,
      "step": 3110
    },
    {
      "epoch": 15.6,
      "grad_norm": 0.21970877051353455,
      "learning_rate": 8.23906024214195e-05,
      "loss": 0.0624,
      "step": 3120
    },
    {
      "epoch": 15.65,
      "grad_norm": 0.25887778401374817,
      "learning_rate": 8.226446417743897e-05,
      "loss": 0.0607,
      "step": 3130
    },
    {
      "epoch": 15.7,
      "grad_norm": 0.23011456429958344,
      "learning_rate": 8.213797309444742e-05,
      "loss": 0.059,
      "step": 3140
    },
    {
      "epoch": 15.75,
      "grad_norm": 0.21193788945674896,
      "learning_rate": 8.201113055573105e-05,
      "loss": 0.0617,
      "step": 3150
    },
    {
      "epoch": 15.8,
      "grad_norm": 0.20553408563137054,
      "learning_rate": 8.188393794841958e-05,
      "loss": 0.0611,
      "step": 3160
    },
    {
      "epoch": 15.85,
      "grad_norm": 0.28941473364830017,
      "learning_rate": 8.175639666347094e-05,
      "loss": 0.0593,
      "step": 3170
    },
    {
      "epoch": 15.9,
      "grad_norm": 0.23488271236419678,
      "learning_rate": 8.162850809565623e-05,
      "loss": 0.0656,
      "step": 3180
    },
    {
      "epoch": 15.95,
      "grad_norm": 0.2775566875934601,
      "learning_rate": 8.150027364354431e-05,
      "loss": 0.0601,
      "step": 3190
    },
    {
      "epoch": 16.0,
      "grad_norm": 0.4618174433708191,
      "learning_rate": 8.137169470948662e-05,
      "loss": 0.0622,
      "step": 3200
    },
    {
      "epoch": 16.05,
      "grad_norm": 0.2531496584415436,
      "learning_rate": 8.124277269960179e-05,
      "loss": 0.0661,
      "step": 3210
    },
    {
      "epoch": 16.1,
      "grad_norm": 0.3925548195838928,
      "learning_rate": 8.111350902376023e-05,
      "loss": 0.065,
      "step": 3220
    },
    {
      "epoch": 16.15,
      "grad_norm": 0.33551856875419617,
      "learning_rate": 8.098390509556883e-05,
      "loss": 0.0663,
      "step": 3230
    },
    {
      "epoch": 16.2,
      "grad_norm": 0.4993094205856323,
      "learning_rate": 8.085396233235536e-05,
      "loss": 0.062,
      "step": 3240
    },
    {
      "epoch": 16.25,
      "grad_norm": 0.24952338635921478,
      "learning_rate": 8.072368215515306e-05,
      "loss": 0.0627,
      "step": 3250
    },
    {
      "epoch": 16.3,
      "grad_norm": 0.31293413043022156,
      "learning_rate": 8.059306598868506e-05,
      "loss": 0.0671,
      "step": 3260
    },
    {
      "epoch": 16.35,
      "grad_norm": 0.2769198417663574,
      "learning_rate": 8.046211526134888e-05,
      "loss": 0.0632,
      "step": 3270
    },
    {
      "epoch": 16.4,
      "grad_norm": 0.23442985117435455,
      "learning_rate": 8.033083140520065e-05,
      "loss": 0.0575,
      "step": 3280
    },
    {
      "epoch": 16.45,
      "grad_norm": 0.4090781509876251,
      "learning_rate": 8.019921585593962e-05,
      "loss": 0.0629,
      "step": 3290
    },
    {
      "epoch": 16.5,
      "grad_norm": 0.25748008489608765,
      "learning_rate": 8.006727005289232e-05,
      "loss": 0.0603,
      "step": 3300
    },
    {
      "epoch": 16.55,
      "grad_norm": 0.2900031507015228,
      "learning_rate": 7.993499543899692e-05,
      "loss": 0.0635,
      "step": 3310
    },
    {
      "epoch": 16.6,
      "grad_norm": 0.23611591756343842,
      "learning_rate": 7.980239346078742e-05,
      "loss": 0.0664,
      "step": 3320
    },
    {
      "epoch": 16.65,
      "grad_norm": 0.2712796628475189,
      "learning_rate": 7.966946556837778e-05,
      "loss": 0.0723,
      "step": 3330
    },
    {
      "epoch": 16.7,
      "grad_norm": 0.22548086941242218,
      "learning_rate": 7.953621321544616e-05,
      "loss": 0.0631,
      "step": 3340
    },
    {
      "epoch": 16.75,
      "grad_norm": 0.29517921805381775,
      "learning_rate": 7.940263785921896e-05,
      "loss": 0.0565,
      "step": 3350
    },
    {
      "epoch": 16.8,
      "grad_norm": 0.24349242448806763,
      "learning_rate": 7.926874096045482e-05,
      "loss": 0.0595,
      "step": 3360
    },
    {
      "epoch": 16.85,
      "grad_norm": 0.31428682804107666,
      "learning_rate": 7.913452398342881e-05,
      "loss": 0.0667,
      "step": 3370
    },
    {
      "epoch": 16.9,
      "grad_norm": 0.39727914333343506,
      "learning_rate": 7.89999883959163e-05,
      "loss": 0.0646,
      "step": 3380
    },
    {
      "epoch": 16.95,
      "grad_norm": 0.19261136651039124,
      "learning_rate": 7.886513566917687e-05,
      "loss": 0.0621,
      "step": 3390
    },
    {
      "epoch": 17.0,
      "grad_norm": 0.3027864396572113,
      "learning_rate": 7.872996727793838e-05,
      "loss": 0.0617,
      "step": 3400
    },
    {
      "epoch": 17.05,
      "grad_norm": 0.23397517204284668,
      "learning_rate": 7.859448470038069e-05,
      "loss": 0.0604,
      "step": 3410
    },
    {
      "epoch": 17.1,
      "grad_norm": 0.22506684064865112,
      "learning_rate": 7.845868941811956e-05,
      "loss": 0.0628,
      "step": 3420
    },
    {
      "epoch": 17.15,
      "grad_norm": 0.22739958763122559,
      "learning_rate": 7.832258291619043e-05,
      "loss": 0.0622,
      "step": 3430
    },
    {
      "epoch": 17.2,
      "grad_norm": 0.3005223274230957,
      "learning_rate": 7.81861666830322e-05,
      "loss": 0.06,
      "step": 3440
    },
    {
      "epoch": 17.25,
      "grad_norm": 0.32206326723098755,
      "learning_rate": 7.804944221047097e-05,
      "loss": 0.0621,
      "step": 3450
    },
    {
      "epoch": 17.3,
      "grad_norm": 0.20813407003879547,
      "learning_rate": 7.791241099370364e-05,
      "loss": 0.0666,
      "step": 3460
    },
    {
      "epoch": 17.35,
      "grad_norm": 0.2243848443031311,
      "learning_rate": 7.777507453128163e-05,
      "loss": 0.0638,
      "step": 3470
    },
    {
      "epoch": 17.4,
      "grad_norm": 0.2873692214488983,
      "learning_rate": 7.763743432509451e-05,
      "loss": 0.0622,
      "step": 3480
    },
    {
      "epoch": 17.45,
      "grad_norm": 0.26742789149284363,
      "learning_rate": 7.749949188035353e-05,
      "loss": 0.0627,
      "step": 3490
    },
    {
      "epoch": 17.5,
      "grad_norm": 0.2441941350698471,
      "learning_rate": 7.736124870557516e-05,
      "loss": 0.0617,
      "step": 3500
    },
    {
      "epoch": 17.55,
      "grad_norm": 0.3393511474132538,
      "learning_rate": 7.722270631256459e-05,
      "loss": 0.0588,
      "step": 3510
    },
    {
      "epoch": 17.6,
      "grad_norm": 0.26802194118499756,
      "learning_rate": 7.708386621639925e-05,
      "loss": 0.0593,
      "step": 3520
    },
    {
      "epoch": 17.65,
      "grad_norm": 0.24832294881343842,
      "learning_rate": 7.694472993541219e-05,
      "loss": 0.0576,
      "step": 3530
    },
    {
      "epoch": 17.7,
      "grad_norm": 0.40361255407333374,
      "learning_rate": 7.680529899117547e-05,
      "loss": 0.0597,
      "step": 3540
    },
    {
      "epoch": 17.75,
      "grad_norm": 0.19824928045272827,
      "learning_rate": 7.666557490848358e-05,
      "loss": 0.0585,
      "step": 3550
    },
    {
      "epoch": 17.8,
      "grad_norm": 0.2692924439907074,
      "learning_rate": 7.65255592153367e-05,
      "loss": 0.0621,
      "step": 3560
    },
    {
      "epoch": 17.85,
      "grad_norm": 0.2544858753681183,
      "learning_rate": 7.638525344292402e-05,
      "loss": 0.0624,
      "step": 3570
    },
    {
      "epoch": 17.9,
      "grad_norm": 0.2764870524406433,
      "learning_rate": 7.624465912560697e-05,
      "loss": 0.0532,
      "step": 3580
    },
    {
      "epoch": 17.95,
      "grad_norm": 0.3781887888908386,
      "learning_rate": 7.610377780090249e-05,
      "loss": 0.0614,
      "step": 3590
    },
    {
      "epoch": 18.0,
      "grad_norm": 0.24595384299755096,
      "learning_rate": 7.596261100946618e-05,
      "loss": 0.061,
      "step": 3600
    },
    {
      "epoch": 18.05,
      "grad_norm": 0.23278765380382538,
      "learning_rate": 7.582116029507542e-05,
      "loss": 0.0586,
      "step": 3610
    },
    {
      "epoch": 18.1,
      "grad_norm": 0.3567298352718353,
      "learning_rate": 7.56794272046126e-05,
      "loss": 0.0565,
      "step": 3620
    },
    {
      "epoch": 18.15,
      "grad_norm": 0.19411662220954895,
      "learning_rate": 7.55374132880481e-05,
      "loss": 0.0607,
      "step": 3630
    },
    {
      "epoch": 18.2,
      "grad_norm": 0.32690006494522095,
      "learning_rate": 7.539512009842333e-05,
      "loss": 0.0566,
      "step": 3640
    },
    {
      "epoch": 18.25,
      "grad_norm": 0.29931947588920593,
      "learning_rate": 7.525254919183382e-05,
      "loss": 0.0607,
      "step": 3650
    },
    {
      "epoch": 18.3,
      "grad_norm": 0.23588475584983826,
      "learning_rate": 7.510970212741215e-05,
      "loss": 0.0602,
      "step": 3660
    },
    {
      "epoch": 18.35,
      "grad_norm": 0.5744988322257996,
      "learning_rate": 7.496658046731096e-05,
      "loss": 0.0611,
      "step": 3670
    },
    {
      "epoch": 18.4,
      "grad_norm": 0.2993015646934509,
      "learning_rate": 7.482318577668578e-05,
      "loss": 0.0542,
      "step": 3680
    },
    {
      "epoch": 18.45,
      "grad_norm": 0.24922674894332886,
      "learning_rate": 7.467951962367796e-05,
      "loss": 0.0643,
      "step": 3690
    },
    {
      "epoch": 18.5,
      "grad_norm": 0.30454349517822266,
      "learning_rate": 7.453558357939755e-05,
      "loss": 0.0695,
      "step": 3700
    },
    {
      "epoch": 18.55,
      "grad_norm": 0.28865817189216614,
      "learning_rate": 7.439137921790606e-05,
      "loss": 0.0553,
      "step": 3710
    },
    {
      "epoch": 18.6,
      "grad_norm": 0.39939433336257935,
      "learning_rate": 7.42469081161993e-05,
      "loss": 0.0603,
      "step": 3720
    },
    {
      "epoch": 18.65,
      "grad_norm": 0.2864205837249756,
      "learning_rate": 7.410217185419006e-05,
      "loss": 0.0592,
      "step": 3730
    },
    {
      "epoch": 18.7,
      "grad_norm": 0.2871643900871277,
      "learning_rate": 7.395717201469095e-05,
      "loss": 0.0576,
      "step": 3740
    },
    {
      "epoch": 18.75,
      "grad_norm": 0.21617856621742249,
      "learning_rate": 7.381191018339696e-05,
      "loss": 0.0549,
      "step": 3750
    },
    {
      "epoch": 18.8,
      "grad_norm": 0.33022254705429077,
      "learning_rate": 7.36663879488682e-05,
      "loss": 0.0588,
      "step": 3760
    },
    {
      "epoch": 18.85,
      "grad_norm": 0.3078994154930115,
      "learning_rate": 7.352060690251254e-05,
      "loss": 0.0575,
      "step": 3770
    },
    {
      "epoch": 18.9,
      "grad_norm": 0.2696463167667389,
      "learning_rate": 7.337456863856811e-05,
      "loss": 0.0536,
      "step": 3780
    },
    {
      "epoch": 18.95,
      "grad_norm": 0.3746519386768341,
      "learning_rate": 7.3228274754086e-05,
      "loss": 0.061,
      "step": 3790
    },
    {
      "epoch": 19.0,
      "grad_norm": 0.41565945744514465,
      "learning_rate": 7.308172684891267e-05,
      "loss": 0.0549,
      "step": 3800
    },
    {
      "epoch": 19.05,
      "grad_norm": 0.3366476595401764,
      "learning_rate": 7.293492652567255e-05,
      "loss": 0.0578,
      "step": 3810
    },
    {
      "epoch": 19.1,
      "grad_norm": 0.3361624777317047,
      "learning_rate": 7.278787538975043e-05,
      "loss": 0.0582,
      "step": 3820
    },
    {
      "epoch": 19.15,
      "grad_norm": 0.3842703402042389,
      "learning_rate": 7.2640575049274e-05,
      "loss": 0.0604,
      "step": 3830
    },
    {
      "epoch": 19.2,
      "grad_norm": 0.2732186019420624,
      "learning_rate": 7.249302711509616e-05,
      "loss": 0.065,
      "step": 3840
    },
    {
      "epoch": 19.25,
      "grad_norm": 0.619268000125885,
      "learning_rate": 7.23452332007775e-05,
      "loss": 0.0536,
      "step": 3850
    },
    {
      "epoch": 19.3,
      "grad_norm": 0.24627473950386047,
      "learning_rate": 7.219719492256858e-05,
      "loss": 0.0608,
      "step": 3860
    },
    {
      "epoch": 19.35,
      "grad_norm": 0.24951502680778503,
      "learning_rate": 7.20489138993923e-05,
      "loss": 0.0567,
      "step": 3870
    },
    {
      "epoch": 19.4,
      "grad_norm": 0.25133541226387024,
      "learning_rate": 7.190039175282614e-05,
      "loss": 0.0596,
      "step": 3880
    },
    {
      "epoch": 19.45,
      "grad_norm": 0.3201720118522644,
      "learning_rate": 7.175163010708455e-05,
      "loss": 0.0611,
      "step": 3890
    },
    {
      "epoch": 19.5,
      "grad_norm": 0.2819395065307617,
      "learning_rate": 7.1602630589001e-05,
      "loss": 0.0551,
      "step": 3900
    },
    {
      "epoch": 19.55,
      "grad_norm": 0.16655567288398743,
      "learning_rate": 7.14533948280104e-05,
      "loss": 0.0569,
      "step": 3910
    },
    {
      "epoch": 19.6,
      "grad_norm": 0.2197456955909729,
      "learning_rate": 7.130392445613109e-05,
      "loss": 0.0551,
      "step": 3920
    },
    {
      "epoch": 19.65,
      "grad_norm": 0.4014874994754791,
      "learning_rate": 7.115422110794711e-05,
      "loss": 0.0558,
      "step": 3930
    },
    {
      "epoch": 19.7,
      "grad_norm": 0.2744712233543396,
      "learning_rate": 7.100428642059033e-05,
      "loss": 0.0598,
      "step": 3940
    },
    {
      "epoch": 19.75,
      "grad_norm": 0.2635567784309387,
      "learning_rate": 7.08541220337224e-05,
      "loss": 0.0609,
      "step": 3950
    },
    {
      "epoch": 19.8,
      "grad_norm": 0.3199650049209595,
      "learning_rate": 7.070372958951706e-05,
      "loss": 0.0603,
      "step": 3960
    },
    {
      "epoch": 19.85,
      "grad_norm": 0.2338671237230301,
      "learning_rate": 7.055311073264194e-05,
      "loss": 0.0582,
      "step": 3970
    },
    {
      "epoch": 19.9,
      "grad_norm": 0.3215521275997162,
      "learning_rate": 7.040226711024077e-05,
      "loss": 0.0611,
      "step": 3980
    },
    {
      "epoch": 19.95,
      "grad_norm": 0.23548661172389984,
      "learning_rate": 7.02512003719152e-05,
      "loss": 0.0659,
      "step": 3990
    },
    {
      "epoch": 20.0,
      "grad_norm": 0.2608540654182434,
      "learning_rate": 7.00999121697069e-05,
      "loss": 0.0565,
      "step": 4000
    },
    {
      "mimicgen/eval/stack_three/success_rate": 0.28,
      "step": 4000
    },
    {
      "epoch": 20.05,
      "grad_norm": 0.2713872492313385,
      "learning_rate": 6.99484041580794e-05,
      "loss": 0.0563,
      "step": 4010
    },
    {
      "epoch": 20.1,
      "grad_norm": 0.20759469270706177,
      "learning_rate": 6.979667799390004e-05,
      "loss": 0.0529,
      "step": 4020
    },
    {
      "epoch": 20.15,
      "grad_norm": 0.549977719783783,
      "learning_rate": 6.964473533642185e-05,
      "loss": 0.0574,
      "step": 4030
    },
    {
      "epoch": 20.2,
      "grad_norm": 0.383088618516922,
      "learning_rate": 6.949257784726539e-05,
      "loss": 0.0533,
      "step": 4040
    },
    {
      "epoch": 20.25,
      "grad_norm": 0.45215800404548645,
      "learning_rate": 6.934020719040056e-05,
      "loss": 0.058,
      "step": 4050
    },
    {
      "epoch": 20.3,
      "grad_norm": 0.23660048842430115,
      "learning_rate": 6.918762503212848e-05,
      "loss": 0.0621,
      "step": 4060
    },
    {
      "epoch": 20.35,
      "grad_norm": 0.22827062010765076,
      "learning_rate": 6.903483304106319e-05,
      "loss": 0.0585,
      "step": 4070
    },
    {
      "epoch": 20.4,
      "grad_norm": 0.2525210380554199,
      "learning_rate": 6.888183288811341e-05,
      "loss": 0.0588,
      "step": 4080
    },
    {
      "epoch": 20.45,
      "grad_norm": 0.40943118929862976,
      "learning_rate": 6.87286262464643e-05,
      "loss": 0.0601,
      "step": 4090
    },
    {
      "epoch": 20.5,
      "grad_norm": 0.45140525698661804,
      "learning_rate": 6.857521479155915e-05,
      "loss": 0.0585,
      "step": 4100
    },
    {
      "epoch": 20.55,
      "grad_norm": 0.22838418185710907,
      "learning_rate": 6.842160020108104e-05,
      "loss": 0.0605,
      "step": 4110
    },
    {
      "epoch": 20.6,
      "grad_norm": 0.2599889039993286,
      "learning_rate": 6.826778415493455e-05,
      "loss": 0.0517,
      "step": 4120
    },
    {
      "epoch": 20.65,
      "grad_norm": 0.3993876576423645,
      "learning_rate": 6.811376833522729e-05,
      "loss": 0.0582,
      "step": 4130
    },
    {
      "epoch": 20.7,
      "grad_norm": 0.2476586103439331,
      "learning_rate": 6.795955442625159e-05,
      "loss": 0.0551,
      "step": 4140
    },
    {
      "epoch": 20.75,
      "grad_norm": 0.32274284958839417,
      "learning_rate": 6.780514411446608e-05,
      "loss": 0.0566,
      "step": 4150
    },
    {
      "epoch": 20.8,
      "grad_norm": 0.23713870346546173,
      "learning_rate": 6.765053908847716e-05,
      "loss": 0.0577,
      "step": 4160
    },
    {
      "epoch": 20.85,
      "grad_norm": 0.203489750623703,
      "learning_rate": 6.749574103902064e-05,
      "loss": 0.0545,
      "step": 4170
    },
    {
      "epoch": 20.9,
      "grad_norm": 0.2523726522922516,
      "learning_rate": 6.734075165894317e-05,
      "loss": 0.0551,
      "step": 4180
    },
    {
      "epoch": 20.95,
      "grad_norm": 0.2219790667295456,
      "learning_rate": 6.71855726431838e-05,
      "loss": 0.0531,
      "step": 4190
    },
    {
      "epoch": 21.0,
      "grad_norm": 0.35188865661621094,
      "learning_rate": 6.703020568875538e-05,
      "loss": 0.0545,
      "step": 4200
    },
    {
      "epoch": 21.05,
      "grad_norm": 0.20831525325775146,
      "learning_rate": 6.687465249472603e-05,
      "loss": 0.0561,
      "step": 4210
    },
    {
      "epoch": 21.1,
      "grad_norm": 0.23723992705345154,
      "learning_rate": 6.671891476220055e-05,
      "loss": 0.056,
      "step": 4220
    },
    {
      "epoch": 21.15,
      "grad_norm": 0.3784637749195099,
      "learning_rate": 6.656299419430183e-05,
      "loss": 0.0557,
      "step": 4230
    },
    {
      "epoch": 21.2,
      "grad_norm": 0.29352110624313354,
      "learning_rate": 6.640689249615223e-05,
      "loss": 0.0596,
      "step": 4240
    },
    {
      "epoch": 21.25,
      "grad_norm": 0.3758820593357086,
      "learning_rate": 6.625061137485491e-05,
      "loss": 0.0575,
      "step": 4250
    },
    {
      "epoch": 21.3,
      "grad_norm": 0.42453479766845703,
      "learning_rate": 6.609415253947517e-05,
      "loss": 0.0594,
      "step": 4260
    },
    {
      "epoch": 21.35,
      "grad_norm": 0.46256381273269653,
      "learning_rate": 6.593751770102178e-05,
      "loss": 0.0562,
      "step": 4270
    },
    {
      "epoch": 21.4,
      "grad_norm": 0.454553484916687,
      "learning_rate": 6.578070857242823e-05,
      "loss": 0.0591,
      "step": 4280
    },
    {
      "epoch": 21.45,
      "grad_norm": 0.34887054562568665,
      "learning_rate": 6.562372686853402e-05,
      "loss": 0.0626,
      "step": 4290
    },
    {
      "epoch": 21.5,
      "grad_norm": 0.27456000447273254,
      "learning_rate": 6.546657430606593e-05,
      "loss": 0.06,
      "step": 4300
    },
    {
      "epoch": 21.55,
      "grad_norm": 0.2411060631275177,
      "learning_rate": 6.530925260361918e-05,
      "loss": 0.0564,
      "step": 4310
    },
    {
      "epoch": 21.6,
      "grad_norm": 0.25875720381736755,
      "learning_rate": 6.515176348163871e-05,
      "loss": 0.0554,
      "step": 4320
    },
    {
      "epoch": 21.65,
      "grad_norm": 0.23697493970394135,
      "learning_rate": 6.499410866240032e-05,
      "loss": 0.0575,
      "step": 4330
    },
    {
      "epoch": 21.7,
      "grad_norm": 0.3276313245296478,
      "learning_rate": 6.48362898699919e-05,
      "loss": 0.0595,
      "step": 4340
    },
    {
      "epoch": 21.75,
      "grad_norm": 0.2486778348684311,
      "learning_rate": 6.467830883029443e-05,
      "loss": 0.061,
      "step": 4350
    },
    {
      "epoch": 21.8,
      "grad_norm": 0.18433448672294617,
      "learning_rate": 6.452016727096326e-05,
      "loss": 0.0498,
      "step": 4360
    },
    {
      "epoch": 21.85,
      "grad_norm": 0.4209209680557251,
      "learning_rate": 6.436186692140916e-05,
      "loss": 0.0588,
      "step": 4370
    },
    {
      "epoch": 21.9,
      "grad_norm": 0.28192031383514404,
      "learning_rate": 6.420340951277938e-05,
      "loss": 0.0558,
      "step": 4380
    },
    {
      "epoch": 21.95,
      "grad_norm": 0.29865923523902893,
      "learning_rate": 6.404479677793874e-05,
      "loss": 0.0486,
      "step": 4390
    },
    {
      "epoch": 22.0,
      "grad_norm": 0.4114896357059479,
      "learning_rate": 6.388603045145075e-05,
      "loss": 0.061,
      "step": 4400
    },
    {
      "epoch": 22.05,
      "grad_norm": 0.40894582867622375,
      "learning_rate": 6.372711226955843e-05,
      "loss": 0.0502,
      "step": 4410
    },
    {
      "epoch": 22.1,
      "grad_norm": 0.27326732873916626,
      "learning_rate": 6.356804397016564e-05,
      "loss": 0.0574,
      "step": 4420
    },
    {
      "epoch": 22.15,
      "grad_norm": 0.3570876717567444,
      "learning_rate": 6.340882729281779e-05,
      "loss": 0.0583,
      "step": 4430
    },
    {
      "epoch": 22.2,
      "grad_norm": 0.20742526650428772,
      "learning_rate": 6.324946397868294e-05,
      "loss": 0.0603,
      "step": 4440
    },
    {
      "epoch": 22.25,
      "grad_norm": 0.38835686445236206,
      "learning_rate": 6.308995577053276e-05,
      "loss": 0.0594,
      "step": 4450
    },
    {
      "epoch": 22.3,
      "grad_norm": 0.4050792455673218,
      "learning_rate": 6.293030441272347e-05,
      "loss": 0.0588,
      "step": 4460
    },
    {
      "epoch": 22.35,
      "grad_norm": 0.4545459449291229,
      "learning_rate": 6.277051165117677e-05,
      "loss": 0.0616,
      "step": 4470
    },
    {
      "epoch": 22.4,
      "grad_norm": 0.25926971435546875,
      "learning_rate": 6.261057923336064e-05,
      "loss": 0.0564,
      "step": 4480
    },
    {
      "epoch": 22.45,
      "grad_norm": 0.6452457904815674,
      "learning_rate": 6.245050890827042e-05,
      "loss": 0.0579,
      "step": 4490
    },
    {
      "epoch": 22.5,
      "grad_norm": 0.3403329849243164,
      "learning_rate": 6.229030242640952e-05,
      "loss": 0.0581,
      "step": 4500
    },
    {
      "epoch": 22.55,
      "grad_norm": 0.36902835965156555,
      "learning_rate": 6.212996153977037e-05,
      "loss": 0.0592,
      "step": 4510
    },
    {
      "epoch": 22.6,
      "grad_norm": 0.41927993297576904,
      "learning_rate": 6.196948800181523e-05,
      "loss": 0.0596,
      "step": 4520
    },
    {
      "epoch": 22.65,
      "grad_norm": 0.33769503235816956,
      "learning_rate": 6.180888356745695e-05,
      "loss": 0.052,
      "step": 4530
    },
    {
      "epoch": 22.7,
      "grad_norm": 0.26099061965942383,
      "learning_rate": 6.164814999303995e-05,
      "loss": 0.0567,
      "step": 4540
    },
    {
      "epoch": 22.75,
      "grad_norm": 0.30975306034088135,
      "learning_rate": 6.148728903632081e-05,
      "loss": 0.0552,
      "step": 4550
    },
    {
      "epoch": 22.8,
      "grad_norm": 0.3618062734603882,
      "learning_rate": 6.132630245644921e-05,
      "loss": 0.0523,
      "step": 4560
    },
    {
      "epoch": 22.85,
      "grad_norm": 0.19721676409244537,
      "learning_rate": 6.116519201394857e-05,
      "loss": 0.0567,
      "step": 4570
    },
    {
      "epoch": 22.9,
      "grad_norm": 0.36515527963638306,
      "learning_rate": 6.10039594706969e-05,
      "loss": 0.0548,
      "step": 4580
    },
    {
      "epoch": 22.95,
      "grad_norm": 0.25335949659347534,
      "learning_rate": 6.084260658990744e-05,
      "loss": 0.0538,
      "step": 4590
    },
    {
      "epoch": 23.0,
      "grad_norm": 0.7224768996238708,
      "learning_rate": 6.068113513610943e-05,
      "loss": 0.0525,
      "step": 4600
    },
    {
      "epoch": 23.05,
      "grad_norm": 0.46208998560905457,
      "learning_rate": 6.0519546875128876e-05,
      "loss": 0.0578,
      "step": 4610
    },
    {
      "epoch": 23.1,
      "grad_norm": 0.32195109128952026,
      "learning_rate": 6.035784357406906e-05,
      "loss": 0.0558,
      "step": 4620
    },
    {
      "epoch": 23.15,
      "grad_norm": 0.2665095329284668,
      "learning_rate": 6.01960270012914e-05,
      "loss": 0.0549,
      "step": 4630
    },
    {
      "epoch": 23.2,
      "grad_norm": 0.4071931540966034,
      "learning_rate": 6.003409892639599e-05,
      "loss": 0.0561,
      "step": 4640
    },
    {
      "epoch": 23.25,
      "grad_norm": 0.24154309928417206,
      "learning_rate": 5.9872061120202336e-05,
      "loss": 0.0595,
      "step": 4650
    },
    {
      "epoch": 23.3,
      "grad_norm": 0.22523197531700134,
      "learning_rate": 5.9709915354729914e-05,
      "loss": 0.0582,
      "step": 4660
    },
    {
      "epoch": 23.35,
      "grad_norm": 0.2103099375963211,
      "learning_rate": 5.9547663403178824e-05,
      "loss": 0.0511,
      "step": 4670
    },
    {
      "epoch": 23.4,
      "grad_norm": 0.22890040278434753,
      "learning_rate": 5.9385307039910445e-05,
      "loss": 0.0584,
      "step": 4680
    },
    {
      "epoch": 23.45,
      "grad_norm": 0.25025373697280884,
      "learning_rate": 5.922284804042792e-05,
      "loss": 0.0536,
      "step": 4690
    },
    {
      "epoch": 23.5,
      "grad_norm": 0.3406077027320862,
      "learning_rate": 5.906028818135687e-05,
      "loss": 0.0521,
      "step": 4700
    },
    {
      "epoch": 23.55,
      "grad_norm": 0.30768176913261414,
      "learning_rate": 5.889762924042585e-05,
      "loss": 0.0544,
      "step": 4710
    },
    {
      "epoch": 23.6,
      "grad_norm": 0.20302532613277435,
      "learning_rate": 5.873487299644699e-05,
      "loss": 0.0498,
      "step": 4720
    },
    {
      "epoch": 23.65,
      "grad_norm": 0.23531799018383026,
      "learning_rate": 5.857202122929649e-05,
      "loss": 0.0499,
      "step": 4730
    },
    {
      "epoch": 23.7,
      "grad_norm": 0.5547714829444885,
      "learning_rate": 5.840907571989518e-05,
      "loss": 0.0555,
      "step": 4740
    },
    {
      "epoch": 23.75,
      "grad_norm": 0.20475926995277405,
      "learning_rate": 5.824603825018904e-05,
      "loss": 0.0517,
      "step": 4750
    },
    {
      "epoch": 23.8,
      "grad_norm": 0.25496530532836914,
      "learning_rate": 5.808291060312975e-05,
      "loss": 0.047,
      "step": 4760
    },
    {
      "epoch": 23.85,
      "grad_norm": 0.28856706619262695,
      "learning_rate": 5.7919694562655083e-05,
      "loss": 0.0547,
      "step": 4770
    },
    {
      "epoch": 23.9,
      "grad_norm": 0.3011395037174225,
      "learning_rate": 5.775639191366954e-05,
      "loss": 0.0484,
      "step": 4780
    },
    {
      "epoch": 23.95,
      "grad_norm": 0.25094273686408997,
      "learning_rate": 5.75930044420247e-05,
      "loss": 0.0511,
      "step": 4790
    },
    {
      "epoch": 24.0,
      "grad_norm": 0.29298165440559387,
      "learning_rate": 5.74295339344998e-05,
      "loss": 0.0557,
      "step": 4800
    },
    {
      "epoch": 24.05,
      "grad_norm": 0.2739546298980713,
      "learning_rate": 5.726598217878211e-05,
      "loss": 0.0498,
      "step": 4810
    },
    {
      "epoch": 24.1,
      "grad_norm": 0.24546706676483154,
      "learning_rate": 5.71023509634474e-05,
      "loss": 0.0529,
      "step": 4820
    },
    {
      "epoch": 24.15,
      "grad_norm": 0.3511584401130676,
      "learning_rate": 5.693864207794049e-05,
      "loss": 0.0573,
      "step": 4830
    },
    {
      "epoch": 24.2,
      "grad_norm": 0.2364063560962677,
      "learning_rate": 5.677485731255545e-05,
      "loss": 0.0481,
      "step": 4840
    },
    {
      "epoch": 24.25,
      "grad_norm": 0.2150588482618332,
      "learning_rate": 5.6610998458416296e-05,
      "loss": 0.051,
      "step": 4850
    },
    {
      "epoch": 24.3,
      "grad_norm": 0.30544859170913696,
      "learning_rate": 5.644706730745716e-05,
      "loss": 0.0531,
      "step": 4860
    },
    {
      "epoch": 24.35,
      "grad_norm": 0.3106234669685364,
      "learning_rate": 5.628306565240287e-05,
      "loss": 0.0554,
      "step": 4870
    },
    {
      "epoch": 24.4,
      "grad_norm": 0.2785274386405945,
      "learning_rate": 5.611899528674923e-05,
      "loss": 0.0569,
      "step": 4880
    },
    {
      "epoch": 24.45,
      "grad_norm": 0.24085290729999542,
      "learning_rate": 5.595485800474349e-05,
      "loss": 0.0503,
      "step": 4890
    },
    {
      "epoch": 24.5,
      "grad_norm": 0.37234413623809814,
      "learning_rate": 5.579065560136467e-05,
      "loss": 0.0543,
      "step": 4900
    },
    {
      "epoch": 24.55,
      "grad_norm": 0.19865863025188446,
      "learning_rate": 5.562638987230392e-05,
      "loss": 0.0521,
      "step": 4910
    },
    {
      "epoch": 24.6,
      "grad_norm": 0.5244201421737671,
      "learning_rate": 5.546206261394498e-05,
      "loss": 0.0471,
      "step": 4920
    },
    {
      "epoch": 24.65,
      "grad_norm": 0.5496284365653992,
      "learning_rate": 5.529767562334437e-05,
      "loss": 0.0527,
      "step": 4930
    },
    {
      "epoch": 24.7,
      "grad_norm": 0.21160632371902466,
      "learning_rate": 5.5133230698211926e-05,
      "loss": 0.0527,
      "step": 4940
    },
    {
      "epoch": 24.75,
      "grad_norm": 0.25491249561309814,
      "learning_rate": 5.496872963689096e-05,
      "loss": 0.0528,
      "step": 4950
    },
    {
      "epoch": 24.8,
      "grad_norm": 0.23031413555145264,
      "learning_rate": 5.4804174238338756e-05,
      "loss": 0.0539,
      "step": 4960
    },
    {
      "epoch": 24.85,
      "grad_norm": 0.2366831749677658,
      "learning_rate": 5.463956630210678e-05,
      "loss": 0.0534,
      "step": 4970
    },
    {
      "epoch": 24.9,
      "grad_norm": 0.4210018217563629,
      "learning_rate": 5.4474907628321046e-05,
      "loss": 0.0586,
      "step": 4980
    },
    {
      "epoch": 24.95,
      "grad_norm": 0.40865975618362427,
      "learning_rate": 5.431020001766244e-05,
      "loss": 0.0494,
      "step": 4990
    },
    {
      "epoch": 25.0,
      "grad_norm": 0.5061337947845459,
      "learning_rate": 5.4145445271346986e-05,
      "loss": 0.0518,
      "step": 5000
    },
    {
      "mimicgen/eval/stack_three/success_rate": 0.48,
      "step": 5000
    },
    {
      "epoch": 25.05,
      "grad_norm": 0.3156657814979553,
      "learning_rate": 5.398064519110622e-05,
      "loss": 0.0513,
      "step": 5010
    },
    {
      "epoch": 25.1,
      "grad_norm": 0.21172654628753662,
      "learning_rate": 5.3815801579167394e-05,
      "loss": 0.0506,
      "step": 5020
    },
    {
      "epoch": 25.15,
      "grad_norm": 0.16904295980930328,
      "learning_rate": 5.365091623823382e-05,
      "loss": 0.0502,
      "step": 5030
    },
    {
      "epoch": 25.2,
      "grad_norm": 0.26419973373413086,
      "learning_rate": 5.348599097146521e-05,
      "loss": 0.0471,
      "step": 5040
    },
    {
      "epoch": 25.25,
      "grad_norm": 0.22036412358283997,
      "learning_rate": 5.3321027582457836e-05,
      "loss": 0.0547,
      "step": 5050
    },
    {
      "epoch": 25.3,
      "grad_norm": 0.41291096806526184,
      "learning_rate": 5.315602787522491e-05,
      "loss": 0.053,
      "step": 5060
    },
    {
      "epoch": 25.35,
      "grad_norm": 0.4199090898036957,
      "learning_rate": 5.299099365417678e-05,
      "loss": 0.0519,
      "step": 5070
    },
    {
      "epoch": 25.4,
      "grad_norm": 0.2975972592830658,
      "learning_rate": 5.2825926724101236e-05,
      "loss": 0.0477,
      "step": 5080
    },
    {
      "epoch": 25.45,
      "grad_norm": 0.33958539366722107,
      "learning_rate": 5.26608288901438e-05,
      "loss": 0.0469,
      "step": 5090
    },
    {
      "epoch": 25.5,
      "grad_norm": 0.5836455821990967,
      "learning_rate": 5.24957019577879e-05,
      "loss": 0.0567,
      "step": 5100
    },
    {
      "epoch": 25.55,
      "grad_norm": 0.4513375759124756,
      "learning_rate": 5.2330547732835266e-05,
      "loss": 0.0464,
      "step": 5110
    },
    {
      "epoch": 25.6,
      "grad_norm": 0.34597229957580566,
      "learning_rate": 5.2165368021385996e-05,
      "loss": 0.0537,
      "step": 5120
    },
    {
      "epoch": 25.65,
      "grad_norm": 0.41008517146110535,
      "learning_rate": 5.200016462981897e-05,
      "loss": 0.0493,
      "step": 5130
    },
    {
      "epoch": 25.7,
      "grad_norm": 0.2872490882873535,
      "learning_rate": 5.1834939364772015e-05,
      "loss": 0.0488,
      "step": 5140
    },
    {
      "epoch": 25.75,
      "grad_norm": 0.2286832183599472,
      "learning_rate": 5.166969403312214e-05,
      "loss": 0.0535,
      "step": 5150
    },
    {
      "epoch": 25.8,
      "grad_norm": 0.3824242651462555,
      "learning_rate": 5.1504430441965844e-05,
      "loss": 0.0516,
      "step": 5160
    },
    {
      "epoch": 25.85,
      "grad_norm": 0.16388070583343506,
      "learning_rate": 5.133915039859923e-05,
      "loss": 0.0488,
      "step": 5170
    },
    {
      "epoch": 25.9,
      "grad_norm": 0.210269495844841,
      "learning_rate": 5.1173855710498444e-05,
      "loss": 0.0508,
      "step": 5180
    },
    {
      "epoch": 25.95,
      "grad_norm": 0.2874531149864197,
      "learning_rate": 5.100854818529967e-05,
      "loss": 0.0505,
      "step": 5190
    },
    {
      "epoch": 26.0,
      "grad_norm": 0.44283297657966614,
      "learning_rate": 5.084322963077951e-05,
      "loss": 0.049,
      "step": 5200
    },
    {
      "epoch": 26.05,
      "grad_norm": 0.2826021909713745,
      "learning_rate": 5.067790185483522e-05,
      "loss": 0.0545,
      "step": 5210
    },
    {
      "epoch": 26.1,
      "grad_norm": 0.3794805109500885,
      "learning_rate": 5.0512566665464844e-05,
      "loss": 0.052,
      "step": 5220
    },
    {
      "epoch": 26.15,
      "grad_norm": 0.21447333693504333,
      "learning_rate": 5.034722587074755e-05,
      "loss": 0.048,
      "step": 5230
    },
    {
      "epoch": 26.2,
      "grad_norm": 0.23619022965431213,
      "learning_rate": 5.018188127882375e-05,
      "loss": 0.0503,
      "step": 5240
    },
    {
      "epoch": 26.25,
      "grad_norm": 0.3903311789035797,
      "learning_rate": 5.0016534697875417e-05,
      "loss": 0.0486,
      "step": 5250
    },
    {
      "epoch": 26.3,
      "grad_norm": 0.36223429441452026,
      "learning_rate": 4.9851187936106294e-05,
      "loss": 0.0549,
      "step": 5260
    },
    {
      "epoch": 26.35,
      "grad_norm": 0.27641722559928894,
      "learning_rate": 4.968584280172206e-05,
      "loss": 0.0451,
      "step": 5270
    },
    {
      "epoch": 26.4,
      "grad_norm": 0.23055115342140198,
      "learning_rate": 4.95205011029106e-05,
      "loss": 0.0522,
      "step": 5280
    },
    {
      "epoch": 26.45,
      "grad_norm": 0.2710167467594147,
      "learning_rate": 4.935516464782227e-05,
      "loss": 0.051,
      "step": 5290
    },
    {
      "epoch": 26.5,
      "grad_norm": 0.40663960576057434,
      "learning_rate": 4.918983524455003e-05,
      "loss": 0.0486,
      "step": 5300
    },
    {
      "epoch": 26.55,
      "grad_norm": 0.2940742075443268,
      "learning_rate": 4.9024514701109766e-05,
      "loss": 0.05,
      "step": 5310
    },
    {
      "epoch": 26.6,
      "grad_norm": 0.21932756900787354,
      "learning_rate": 4.885920482542043e-05,
      "loss": 0.0502,
      "step": 5320
    },
    {
      "epoch": 26.65,
      "grad_norm": 0.3388858139514923,
      "learning_rate": 4.869390742528438e-05,
      "loss": 0.0532,
      "step": 5330
    },
    {
      "epoch": 26.7,
      "grad_norm": 0.3895670771598816,
      "learning_rate": 4.852862430836744e-05,
      "loss": 0.0483,
      "step": 5340
    },
    {
      "epoch": 26.75,
      "grad_norm": 0.22904324531555176,
      "learning_rate": 4.836335728217933e-05,
      "loss": 0.0511,
      "step": 5350
    },
    {
      "epoch": 26.8,
      "grad_norm": 0.3420020341873169,
      "learning_rate": 4.819810815405379e-05,
      "loss": 0.0521,
      "step": 5360
    },
    {
      "epoch": 26.85,
      "grad_norm": 0.3138860762119293,
      "learning_rate": 4.803287873112877e-05,
      "loss": 0.0511,
      "step": 5370
    },
    {
      "epoch": 26.9,
      "grad_norm": 0.22803205251693726,
      "learning_rate": 4.786767082032681e-05,
      "loss": 0.0505,
      "step": 5380
    },
    {
      "epoch": 26.95,
      "grad_norm": 0.22282983362674713,
      "learning_rate": 4.77024862283351e-05,
      "loss": 0.0489,
      "step": 5390
    },
    {
      "epoch": 27.0,
      "grad_norm": 0.5086653232574463,
      "learning_rate": 4.753732676158593e-05,
      "loss": 0.0428,
      "step": 5400
    },
    {
      "epoch": 27.05,
      "grad_norm": 0.43237000703811646,
      "learning_rate": 4.737219422623672e-05,
      "loss": 0.0501,
      "step": 5410
    },
    {
      "epoch": 27.1,
      "grad_norm": 0.41998669505119324,
      "learning_rate": 4.720709042815044e-05,
      "loss": 0.0483,
      "step": 5420
    },
    {
      "epoch": 27.15,
      "grad_norm": 0.32148078083992004,
      "learning_rate": 4.704201717287578e-05,
      "loss": 0.0459,
      "step": 5430
    },
    {
      "epoch": 27.2,
      "grad_norm": 0.2794801592826843,
      "learning_rate": 4.6876976265627404e-05,
      "loss": 0.0516,
      "step": 5440
    },
    {
      "epoch": 27.25,
      "grad_norm": 0.2555250823497772,
      "learning_rate": 4.671196951126626e-05,
      "loss": 0.0472,
      "step": 5450
    },
    {
      "epoch": 27.3,
      "grad_norm": 0.436898797750473,
      "learning_rate": 4.654699871427971e-05,
      "loss": 0.0476,
      "step": 5460
    },
    {
      "epoch": 27.35,
      "grad_norm": 0.23393304646015167,
      "learning_rate": 4.6382065678762034e-05,
      "loss": 0.0496,
      "step": 5470
    },
    {
      "epoch": 27.4,
      "grad_norm": 0.33094361424446106,
      "learning_rate": 4.6217172208394424e-05,
      "loss": 0.052,
      "step": 5480
    },
    {
      "epoch": 27.45,
      "grad_norm": 0.31259191036224365,
      "learning_rate": 4.605232010642549e-05,
      "loss": 0.0456,
      "step": 5490
    },
    {
      "epoch": 27.5,
      "grad_norm": 0.27039188146591187,
      "learning_rate": 4.588751117565142e-05,
      "loss": 0.0541,
      "step": 5500
    },
    {
      "epoch": 27.55,
      "grad_norm": 0.19947443902492523,
      "learning_rate": 4.5722747218396214e-05,
      "loss": 0.0491,
      "step": 5510
    },
    {
      "epoch": 27.6,
      "grad_norm": 0.25906801223754883,
      "learning_rate": 4.5558030036492194e-05,
      "loss": 0.0479,
      "step": 5520
    },
    {
      "epoch": 27.65,
      "grad_norm": 0.4609508216381073,
      "learning_rate": 4.539336143125999e-05,
      "loss": 0.0476,
      "step": 5530
    },
    {
      "epoch": 27.7,
      "grad_norm": 0.5646840333938599,
      "learning_rate": 4.522874320348916e-05,
      "loss": 0.0506,
      "step": 5540
    },
    {
      "epoch": 27.75,
      "grad_norm": 0.4694201648235321,
      "learning_rate": 4.506417715341821e-05,
      "loss": 0.0479,
      "step": 5550
    },
    {
      "epoch": 27.8,
      "grad_norm": 0.36954089999198914,
      "learning_rate": 4.489966508071511e-05,
      "loss": 0.0474,
      "step": 5560
    },
    {
      "epoch": 27.85,
      "grad_norm": 0.3262273073196411,
      "learning_rate": 4.4735208784457575e-05,
      "loss": 0.048,
      "step": 5570
    },
    {
      "epoch": 27.9,
      "grad_norm": 0.3336296081542969,
      "learning_rate": 4.457081006311325e-05,
      "loss": 0.0504,
      "step": 5580
    },
    {
      "epoch": 27.95,
      "grad_norm": 0.24268998205661774,
      "learning_rate": 4.440647071452027e-05,
      "loss": 0.0437,
      "step": 5590
    },
    {
      "epoch": 28.0,
      "grad_norm": 0.6531052589416504,
      "learning_rate": 4.424219253586737e-05,
      "loss": 0.05,
      "step": 5600
    },
    {
      "epoch": 28.05,
      "grad_norm": 0.4362652599811554,
      "learning_rate": 4.407797732367443e-05,
      "loss": 0.0524,
      "step": 5610
    },
    {
      "epoch": 28.1,
      "grad_norm": 0.24813103675842285,
      "learning_rate": 4.391382687377268e-05,
      "loss": 0.0502,
      "step": 5620
    },
    {
      "epoch": 28.15,
      "grad_norm": 0.2442522794008255,
      "learning_rate": 4.374974298128512e-05,
      "loss": 0.0507,
      "step": 5630
    },
    {
      "epoch": 28.2,
      "grad_norm": 0.3636161684989929,
      "learning_rate": 4.358572744060699e-05,
      "loss": 0.0487,
      "step": 5640
    },
    {
      "epoch": 28.25,
      "grad_norm": 0.22177453339099884,
      "learning_rate": 4.342178204538588e-05,
      "loss": 0.044,
      "step": 5650
    },
    {
      "epoch": 28.3,
      "grad_norm": 0.27626025676727295,
      "learning_rate": 4.325790858850241e-05,
      "loss": 0.0446,
      "step": 5660
    },
    {
      "epoch": 28.35,
      "grad_norm": 0.3013800084590912,
      "learning_rate": 4.309410886205043e-05,
      "loss": 0.0455,
      "step": 5670
    },
    {
      "epoch": 28.4,
      "grad_norm": 0.25736096501350403,
      "learning_rate": 4.293038465731752e-05,
      "loss": 0.0476,
      "step": 5680
    },
    {
      "epoch": 28.45,
      "grad_norm": 0.6885226964950562,
      "learning_rate": 4.276673776476533e-05,
      "loss": 0.0515,
      "step": 5690
    },
    {
      "epoch": 28.5,
      "grad_norm": 0.5512219667434692,
      "learning_rate": 4.260316997401007e-05,
      "loss": 0.0509,
      "step": 5700
    },
    {
      "epoch": 28.55,
      "grad_norm": 0.2252778857946396,
      "learning_rate": 4.243968307380293e-05,
      "loss": 0.0503,
      "step": 5710
    },
    {
      "epoch": 28.6,
      "grad_norm": 0.6579379439353943,
      "learning_rate": 4.22762788520104e-05,
      "loss": 0.0481,
      "step": 5720
    },
    {
      "epoch": 28.65,
      "grad_norm": 0.1377784162759781,
      "learning_rate": 4.211295909559491e-05,
      "loss": 0.0463,
      "step": 5730
    },
    {
      "epoch": 28.7,
      "grad_norm": 0.19235268235206604,
      "learning_rate": 4.194972559059511e-05,
      "loss": 0.0495,
      "step": 5740
    },
    {
      "epoch": 28.75,
      "grad_norm": 0.19346018135547638,
      "learning_rate": 4.178658012210651e-05,
      "loss": 0.0453,
      "step": 5750
    },
    {
      "epoch": 28.8,
      "grad_norm": 0.5356079339981079,
      "learning_rate": 4.162352447426177e-05,
      "loss": 0.0454,
      "step": 5760
    },
    {
      "epoch": 28.85,
      "grad_norm": 0.46095407009124756,
      "learning_rate": 4.146056043021135e-05,
      "loss": 0.0516,
      "step": 5770
    },
    {
      "epoch": 28.9,
      "grad_norm": 0.2556172311306,
      "learning_rate": 4.1297689772103944e-05,
      "loss": 0.0533,
      "step": 5780
    },
    {
      "epoch": 28.95,
      "grad_norm": 0.304859459400177,
      "learning_rate": 4.113491428106694e-05,
      "loss": 0.0511,
      "step": 5790
    },
    {
      "epoch": 29.0,
      "grad_norm": 0.2528996169567108,
      "learning_rate": 4.0972235737187055e-05,
      "loss": 0.0472,
      "step": 5800
    },
    {
      "epoch": 29.05,
      "grad_norm": 0.3128699064254761,
      "learning_rate": 4.080965591949076e-05,
      "loss": 0.0471,
      "step": 5810
    },
    {
      "epoch": 29.1,
      "grad_norm": 0.5651791095733643,
      "learning_rate": 4.0647176605924924e-05,
      "loss": 0.0533,
      "step": 5820
    },
    {
      "epoch": 29.15,
      "grad_norm": 0.2908657491207123,
      "learning_rate": 4.0484799573337255e-05,
      "loss": 0.0503,
      "step": 5830
    },
    {
      "epoch": 29.2,
      "grad_norm": 0.30977728962898254,
      "learning_rate": 4.032252659745699e-05,
      "loss": 0.0507,
      "step": 5840
    },
    {
      "epoch": 29.25,
      "grad_norm": 0.3490520417690277,
      "learning_rate": 4.016035945287539e-05,
      "loss": 0.0455,
      "step": 5850
    },
    {
      "epoch": 29.3,
      "grad_norm": 0.232571080327034,
      "learning_rate": 3.999829991302635e-05,
      "loss": 0.0483,
      "step": 5860
    },
    {
      "epoch": 29.35,
      "grad_norm": 0.23217721283435822,
      "learning_rate": 3.983634975016707e-05,
      "loss": 0.0457,
      "step": 5870
    },
    {
      "epoch": 29.4,
      "grad_norm": 0.23421244323253632,
      "learning_rate": 3.967451073535854e-05,
      "loss": 0.0481,
      "step": 5880
    },
    {
      "epoch": 29.45,
      "grad_norm": 0.2898552417755127,
      "learning_rate": 3.951278463844633e-05,
      "loss": 0.0497,
      "step": 5890
    },
    {
      "epoch": 29.5,
      "grad_norm": 0.41422924399375916,
      "learning_rate": 3.935117322804111e-05,
      "loss": 0.0451,
      "step": 5900
    },
    {
      "epoch": 29.55,
      "grad_norm": 0.4588993787765503,
      "learning_rate": 3.918967827149938e-05,
      "loss": 0.0487,
      "step": 5910
    },
    {
      "epoch": 29.6,
      "grad_norm": 0.25742068886756897,
      "learning_rate": 3.9028301534904094e-05,
      "loss": 0.0486,
      "step": 5920
    },
    {
      "epoch": 29.65,
      "grad_norm": 0.18322889506816864,
      "learning_rate": 3.88670447830454e-05,
      "loss": 0.0533,
      "step": 5930
    },
    {
      "epoch": 29.7,
      "grad_norm": 0.3035212457180023,
      "learning_rate": 3.870590977940132e-05,
      "loss": 0.0472,
      "step": 5940
    },
    {
      "epoch": 29.75,
      "grad_norm": 0.3117741048336029,
      "learning_rate": 3.8544898286118404e-05,
      "loss": 0.0498,
      "step": 5950
    },
    {
      "epoch": 29.8,
      "grad_norm": 0.24405276775360107,
      "learning_rate": 3.838401206399257e-05,
      "loss": 0.0504,
      "step": 5960
    },
    {
      "epoch": 29.85,
      "grad_norm": 0.5126343965530396,
      "learning_rate": 3.822325287244975e-05,
      "loss": 0.0497,
      "step": 5970
    },
    {
      "epoch": 29.9,
      "grad_norm": 0.6243114471435547,
      "learning_rate": 3.8062622469526725e-05,
      "loss": 0.0565,
      "step": 5980
    },
    {
      "epoch": 29.95,
      "grad_norm": 0.28896650671958923,
      "learning_rate": 3.790212261185183e-05,
      "loss": 0.0492,
      "step": 5990
    },
    {
      "epoch": 30.0,
      "grad_norm": 0.3882354199886322,
      "learning_rate": 3.7741755054625794e-05,
      "loss": 0.0476,
      "step": 6000
    },
    {
      "mimicgen/eval/stack_three/success_rate": 0.54,
      "step": 6000
    },
    {
      "epoch": 30.05,
      "grad_norm": 0.2529790997505188,
      "learning_rate": 3.758152155160255e-05,
      "loss": 0.0475,
      "step": 6010
    },
    {
      "epoch": 30.1,
      "grad_norm": 0.2379978895187378,
      "learning_rate": 3.742142385506999e-05,
      "loss": 0.0483,
      "step": 6020
    },
    {
      "epoch": 30.15,
      "grad_norm": 0.3732977509498596,
      "learning_rate": 3.72614637158309e-05,
      "loss": 0.0449,
      "step": 6030
    },
    {
      "epoch": 30.2,
      "grad_norm": 0.44576072692871094,
      "learning_rate": 3.710164288318371e-05,
      "loss": 0.0466,
      "step": 6040
    },
    {
      "epoch": 30.25,
      "grad_norm": 0.18610893189907074,
      "learning_rate": 3.694196310490345e-05,
      "loss": 0.0444,
      "step": 6050
    },
    {
      "epoch": 30.3,
      "grad_norm": 0.4956027865409851,
      "learning_rate": 3.678242612722259e-05,
      "loss": 0.0474,
      "step": 6060
    },
    {
      "epoch": 30.35,
      "grad_norm": 0.3176361918449402,
      "learning_rate": 3.6623033694811953e-05,
      "loss": 0.0469,
      "step": 6070
    },
    {
      "epoch": 30.4,
      "grad_norm": 0.4559582769870758,
      "learning_rate": 3.6463787550761665e-05,
      "loss": 0.0431,
      "step": 6080
    },
    {
      "epoch": 30.45,
      "grad_norm": 0.3999873399734497,
      "learning_rate": 3.630468943656202e-05,
      "loss": 0.0462,
      "step": 6090
    },
    {
      "epoch": 30.5,
      "grad_norm": 0.6342704892158508,
      "learning_rate": 3.6145741092084523e-05,
      "loss": 0.0467,
      "step": 6100
    },
    {
      "epoch": 30.55,
      "grad_norm": 0.368694007396698,
      "learning_rate": 3.598694425556278e-05,
      "loss": 0.0473,
      "step": 6110
    },
    {
      "epoch": 30.6,
      "grad_norm": 0.283324658870697,
      "learning_rate": 3.58283006635736e-05,
      "loss": 0.0488,
      "step": 6120
    },
    {
      "epoch": 30.65,
      "grad_norm": 0.41913390159606934,
      "learning_rate": 3.566981205101781e-05,
      "loss": 0.0508,
      "step": 6130
    },
    {
      "epoch": 30.7,
      "grad_norm": 0.5377837419509888,
      "learning_rate": 3.5511480151101556e-05,
      "loss": 0.0434,
      "step": 6140
    },
    {
      "epoch": 30.75,
      "grad_norm": 0.6758276224136353,
      "learning_rate": 3.5353306695317104e-05,
      "loss": 0.0486,
      "step": 6150
    },
    {
      "epoch": 30.8,
      "grad_norm": 0.3431497812271118,
      "learning_rate": 3.519529341342402e-05,
      "loss": 0.0461,
      "step": 6160
    },
    {
      "epoch": 30.85,
      "grad_norm": 0.34173500537872314,
      "learning_rate": 3.503744203343026e-05,
      "loss": 0.0502,
      "step": 6170
    },
    {
      "epoch": 30.9,
      "grad_norm": 0.7766184210777283,
      "learning_rate": 3.487975428157318e-05,
      "loss": 0.0489,
      "step": 6180
    },
    {
      "epoch": 30.95,
      "grad_norm": 0.5298252105712891,
      "learning_rate": 3.472223188230083e-05,
      "loss": 0.0489,
      "step": 6190
    },
    {
      "epoch": 31.0,
      "grad_norm": 0.3954032361507416,
      "learning_rate": 3.4564876558252866e-05,
      "loss": 0.0416,
      "step": 6200
    },
    {
      "epoch": 31.05,
      "grad_norm": 0.27368101477622986,
      "learning_rate": 3.440769003024195e-05,
      "loss": 0.0409,
      "step": 6210
    },
    {
      "epoch": 31.1,
      "grad_norm": 0.29750439524650574,
      "learning_rate": 3.425067401723477e-05,
      "loss": 0.0415,
      "step": 6220
    },
    {
      "epoch": 31.15,
      "grad_norm": 0.6667307019233704,
      "learning_rate": 3.409383023633325e-05,
      "loss": 0.0441,
      "step": 6230
    },
    {
      "epoch": 31.2,
      "grad_norm": 0.4117032587528229,
      "learning_rate": 3.3937160402755894e-05,
      "loss": 0.0418,
      "step": 6240
    },
    {
      "epoch": 31.25,
      "grad_norm": 0.591961681842804,
      "learning_rate": 3.378066622981885e-05,
      "loss": 0.0453,
      "step": 6250
    },
    {
      "epoch": 31.3,
      "grad_norm": 0.33909693360328674,
      "learning_rate": 3.362434942891738e-05,
      "loss": 0.047,
      "step": 6260
    },
    {
      "epoch": 31.35,
      "grad_norm": 0.8566231727600098,
      "learning_rate": 3.346821170950693e-05,
      "loss": 0.0447,
      "step": 6270
    },
    {
      "epoch": 31.4,
      "grad_norm": 0.6507214903831482,
      "learning_rate": 3.3312254779084585e-05,
      "loss": 0.0424,
      "step": 6280
    },
    {
      "epoch": 31.45,
      "grad_norm": 0.24364884197711945,
      "learning_rate": 3.315648034317039e-05,
      "loss": 0.0478,
      "step": 6290
    },
    {
      "epoch": 31.5,
      "grad_norm": 0.24964404106140137,
      "learning_rate": 3.3000890105288564e-05,
      "loss": 0.0461,
      "step": 6300
    },
    {
      "epoch": 31.55,
      "grad_norm": 0.2656809091567993,
      "learning_rate": 3.284548576694908e-05,
      "loss": 0.0422,
      "step": 6310
    },
    {
      "epoch": 31.6,
      "grad_norm": 0.3868388831615448,
      "learning_rate": 3.2690269027628815e-05,
      "loss": 0.0425,
      "step": 6320
    },
    {
      "epoch": 31.65,
      "grad_norm": 0.7454066276550293,
      "learning_rate": 3.253524158475324e-05,
      "loss": 0.0477,
      "step": 6330
    },
    {
      "epoch": 31.7,
      "grad_norm": 0.6047732830047607,
      "learning_rate": 3.238040513367757e-05,
      "loss": 0.0472,
      "step": 6340
    },
    {
      "epoch": 31.75,
      "grad_norm": 0.31720399856567383,
      "learning_rate": 3.222576136766843e-05,
      "loss": 0.0458,
      "step": 6350
    },
    {
      "epoch": 31.8,
      "grad_norm": 0.20385631918907166,
      "learning_rate": 3.2071311977885324e-05,
      "loss": 0.0446,
      "step": 6360
    },
    {
      "epoch": 31.85,
      "grad_norm": 0.28796133399009705,
      "learning_rate": 3.191705865336197e-05,
      "loss": 0.0451,
      "step": 6370
    },
    {
      "epoch": 31.9,
      "grad_norm": 0.2851928174495697,
      "learning_rate": 3.1763003080988075e-05,
      "loss": 0.0495,
      "step": 6380
    },
    {
      "epoch": 31.95,
      "grad_norm": 0.37570512294769287,
      "learning_rate": 3.160914694549063e-05,
      "loss": 0.0474,
      "step": 6390
    },
    {
      "epoch": 32.0,
      "grad_norm": 0.42759668827056885,
      "learning_rate": 3.145549192941573e-05,
      "loss": 0.0479,
      "step": 6400
    },
    {
      "epoch": 32.05,
      "grad_norm": 0.25117647647857666,
      "learning_rate": 3.130203971310999e-05,
      "loss": 0.0484,
      "step": 6410
    },
    {
      "epoch": 32.1,
      "grad_norm": 0.33110061287879944,
      "learning_rate": 3.114879197470225e-05,
      "loss": 0.0456,
      "step": 6420
    },
    {
      "epoch": 32.15,
      "grad_norm": 0.6213118433952332,
      "learning_rate": 3.0995750390085285e-05,
      "loss": 0.0387,
      "step": 6430
    },
    {
      "epoch": 32.2,
      "grad_norm": 0.2257700264453888,
      "learning_rate": 3.084291663289728e-05,
      "loss": 0.0454,
      "step": 6440
    },
    {
      "epoch": 32.25,
      "grad_norm": 0.2271321564912796,
      "learning_rate": 3.069029237450375e-05,
      "loss": 0.0451,
      "step": 6450
    },
    {
      "epoch": 32.3,
      "grad_norm": 0.31200265884399414,
      "learning_rate": 3.053787928397911e-05,
      "loss": 0.0458,
      "step": 6460
    },
    {
      "epoch": 32.35,
      "grad_norm": 0.6247195601463318,
      "learning_rate": 3.0385679028088526e-05,
      "loss": 0.0412,
      "step": 6470
    },
    {
      "epoch": 32.4,
      "grad_norm": 0.3253103494644165,
      "learning_rate": 3.023369327126959e-05,
      "loss": 0.0474,
      "step": 6480
    },
    {
      "epoch": 32.45,
      "grad_norm": 0.25428399443626404,
      "learning_rate": 3.0081923675614198e-05,
      "loss": 0.0471,
      "step": 6490
    },
    {
      "epoch": 32.5,
      "grad_norm": 0.2929300367832184,
      "learning_rate": 2.993037190085034e-05,
      "loss": 0.0412,
      "step": 6500
    },
    {
      "epoch": 32.55,
      "grad_norm": 0.42548292875289917,
      "learning_rate": 2.977903960432392e-05,
      "loss": 0.0434,
      "step": 6510
    },
    {
      "epoch": 32.6,
      "grad_norm": 0.1924518495798111,
      "learning_rate": 2.9627928440980722e-05,
      "loss": 0.0478,
      "step": 6520
    },
    {
      "epoch": 32.65,
      "grad_norm": 0.2286495417356491,
      "learning_rate": 2.9477040063348183e-05,
      "loss": 0.0381,
      "step": 6530
    },
    {
      "epoch": 32.7,
      "grad_norm": 0.40596309304237366,
      "learning_rate": 2.9326376121517456e-05,
      "loss": 0.047,
      "step": 6540
    },
    {
      "epoch": 32.75,
      "grad_norm": 0.4043411612510681,
      "learning_rate": 2.9175938263125236e-05,
      "loss": 0.0441,
      "step": 6550
    },
    {
      "epoch": 32.8,
      "grad_norm": 0.2415008693933487,
      "learning_rate": 2.9025728133335873e-05,
      "loss": 0.0444,
      "step": 6560
    },
    {
      "epoch": 32.85,
      "grad_norm": 0.3759835958480835,
      "learning_rate": 2.8875747374823288e-05,
      "loss": 0.0485,
      "step": 6570
    },
    {
      "epoch": 32.9,
      "grad_norm": 0.38236716389656067,
      "learning_rate": 2.872599762775298e-05,
      "loss": 0.0428,
      "step": 6580
    },
    {
      "epoch": 32.95,
      "grad_norm": 0.2508070170879364,
      "learning_rate": 2.857648052976425e-05,
      "loss": 0.042,
      "step": 6590
    },
    {
      "epoch": 33.0,
      "grad_norm": 0.363592267036438,
      "learning_rate": 2.8427197715952047e-05,
      "loss": 0.0458,
      "step": 6600
    },
    {
      "epoch": 33.05,
      "grad_norm": 0.23147256672382355,
      "learning_rate": 2.8278150818849393e-05,
      "loss": 0.0436,
      "step": 6610
    },
    {
      "epoch": 33.1,
      "grad_norm": 0.31120434403419495,
      "learning_rate": 2.812934146840922e-05,
      "loss": 0.0427,
      "step": 6620
    },
    {
      "epoch": 33.15,
      "grad_norm": 0.2634255886077881,
      "learning_rate": 2.7980771291986764e-05,
      "loss": 0.0412,
      "step": 6630
    },
    {
      "epoch": 33.2,
      "grad_norm": 0.5908904671669006,
      "learning_rate": 2.783244191432167e-05,
      "loss": 0.0411,
      "step": 6640
    },
    {
      "epoch": 33.25,
      "grad_norm": 0.5316810011863708,
      "learning_rate": 2.768435495752022e-05,
      "loss": 0.0479,
      "step": 6650
    },
    {
      "epoch": 33.3,
      "grad_norm": 0.22751127183437347,
      "learning_rate": 2.753651204103771e-05,
      "loss": 0.044,
      "step": 6660
    },
    {
      "epoch": 33.35,
      "grad_norm": 0.3375546932220459,
      "learning_rate": 2.7388914781660523e-05,
      "loss": 0.0433,
      "step": 6670
    },
    {
      "epoch": 33.4,
      "grad_norm": 0.18735012412071228,
      "learning_rate": 2.7241564793488693e-05,
      "loss": 0.0454,
      "step": 6680
    },
    {
      "epoch": 33.45,
      "grad_norm": 0.21282385289669037,
      "learning_rate": 2.7094463687918037e-05,
      "loss": 0.0447,
      "step": 6690
    },
    {
      "epoch": 33.5,
      "grad_norm": 0.23620350658893585,
      "learning_rate": 2.694761307362268e-05,
      "loss": 0.0427,
      "step": 6700
    },
    {
      "epoch": 33.55,
      "grad_norm": 0.3497268855571747,
      "learning_rate": 2.6801014556537467e-05,
      "loss": 0.0422,
      "step": 6710
    },
    {
      "epoch": 33.6,
      "grad_norm": 0.22555963695049286,
      "learning_rate": 2.6654669739840243e-05,
      "loss": 0.0443,
      "step": 6720
    },
    {
      "epoch": 33.65,
      "grad_norm": 0.20421560108661652,
      "learning_rate": 2.650858022393451e-05,
      "loss": 0.0489,
      "step": 6730
    },
    {
      "epoch": 33.7,
      "grad_norm": 0.5181028842926025,
      "learning_rate": 2.6362747606431747e-05,
      "loss": 0.0436,
      "step": 6740
    },
    {
      "epoch": 33.75,
      "grad_norm": 0.3222913146018982,
      "learning_rate": 2.6217173482134172e-05,
      "loss": 0.0412,
      "step": 6750
    },
    {
      "epoch": 33.8,
      "grad_norm": 0.26537737250328064,
      "learning_rate": 2.6071859443017044e-05,
      "loss": 0.0423,
      "step": 6760
    },
    {
      "epoch": 33.85,
      "grad_norm": 0.2742117941379547,
      "learning_rate": 2.5926807078211414e-05,
      "loss": 0.045,
      "step": 6770
    },
    {
      "epoch": 33.9,
      "grad_norm": 0.24554044008255005,
      "learning_rate": 2.5782017973986728e-05,
      "loss": 0.0404,
      "step": 6780
    },
    {
      "epoch": 33.95,
      "grad_norm": 0.4262136220932007,
      "learning_rate": 2.5637493713733374e-05,
      "loss": 0.0457,
      "step": 6790
    },
    {
      "epoch": 34.0,
      "grad_norm": 0.8119643330574036,
      "learning_rate": 2.549323587794559e-05,
      "loss": 0.0442,
      "step": 6800
    },
    {
      "epoch": 34.05,
      "grad_norm": 0.8580737709999084,
      "learning_rate": 2.5349246044203895e-05,
      "loss": 0.0447,
      "step": 6810
    },
    {
      "epoch": 34.1,
      "grad_norm": 0.23737110197544098,
      "learning_rate": 2.520552578715808e-05,
      "loss": 0.0394,
      "step": 6820
    },
    {
      "epoch": 34.15,
      "grad_norm": 0.6481374502182007,
      "learning_rate": 2.506207667850981e-05,
      "loss": 0.0412,
      "step": 6830
    },
    {
      "epoch": 34.2,
      "grad_norm": 0.4944464862346649,
      "learning_rate": 2.4918900286995555e-05,
      "loss": 0.0442,
      "step": 6840
    },
    {
      "epoch": 34.25,
      "grad_norm": 0.282958984375,
      "learning_rate": 2.4775998178369458e-05,
      "loss": 0.0439,
      "step": 6850
    },
    {
      "epoch": 34.3,
      "grad_norm": 0.2667924761772156,
      "learning_rate": 2.4633371915386017e-05,
      "loss": 0.0487,
      "step": 6860
    },
    {
      "epoch": 34.35,
      "grad_norm": 0.44736137986183167,
      "learning_rate": 2.4491023057783235e-05,
      "loss": 0.0442,
      "step": 6870
    },
    {
      "epoch": 34.4,
      "grad_norm": 0.39403706789016724,
      "learning_rate": 2.4348953162265375e-05,
      "loss": 0.0416,
      "step": 6880
    },
    {
      "epoch": 34.45,
      "grad_norm": 0.3815446197986603,
      "learning_rate": 2.420716378248607e-05,
      "loss": 0.0471,
      "step": 6890
    },
    {
      "epoch": 34.5,
      "grad_norm": 0.3031497299671173,
      "learning_rate": 2.4065656469031266e-05,
      "loss": 0.0402,
      "step": 6900
    },
    {
      "epoch": 34.55,
      "grad_norm": 0.3887488842010498,
      "learning_rate": 2.3924432769402268e-05,
      "loss": 0.0403,
      "step": 6910
    },
    {
      "epoch": 34.6,
      "grad_norm": 0.2753234803676605,
      "learning_rate": 2.3783494227998844e-05,
      "loss": 0.0411,
      "step": 6920
    },
    {
      "epoch": 34.65,
      "grad_norm": 0.2736208140850067,
      "learning_rate": 2.3642842386102264e-05,
      "loss": 0.0456,
      "step": 6930
    },
    {
      "epoch": 34.7,
      "grad_norm": 0.3083034157752991,
      "learning_rate": 2.3502478781858567e-05,
      "loss": 0.0396,
      "step": 6940
    },
    {
      "epoch": 34.75,
      "grad_norm": 0.2571602463722229,
      "learning_rate": 2.3362404950261628e-05,
      "loss": 0.0451,
      "step": 6950
    },
    {
      "epoch": 34.8,
      "grad_norm": 0.30380338430404663,
      "learning_rate": 2.3222622423136458e-05,
      "loss": 0.043,
      "step": 6960
    },
    {
      "epoch": 34.85,
      "grad_norm": 0.3656725287437439,
      "learning_rate": 2.3083132729122332e-05,
      "loss": 0.044,
      "step": 6970
    },
    {
      "epoch": 34.9,
      "grad_norm": 0.5429569482803345,
      "learning_rate": 2.294393739365621e-05,
      "loss": 0.0387,
      "step": 6980
    },
    {
      "epoch": 34.95,
      "grad_norm": 0.47947391867637634,
      "learning_rate": 2.2805037938956e-05,
      "loss": 0.043,
      "step": 6990
    },
    {
      "epoch": 35.0,
      "grad_norm": 0.7154796123504639,
      "learning_rate": 2.266643588400386e-05,
      "loss": 0.0475,
      "step": 7000
    },
    {
      "mimicgen/eval/stack_three/success_rate": 0.54,
      "step": 7000
    },
    {
      "epoch": 35.05,
      "grad_norm": 0.36626002192497253,
      "learning_rate": 2.252813274452969e-05,
      "loss": 0.0389,
      "step": 7010
    },
    {
      "epoch": 35.1,
      "grad_norm": 0.22212912142276764,
      "learning_rate": 2.2390130032994427e-05,
      "loss": 0.04,
      "step": 7020
    },
    {
      "epoch": 35.15,
      "grad_norm": 0.4438019096851349,
      "learning_rate": 2.2252429258573633e-05,
      "loss": 0.0407,
      "step": 7030
    },
    {
      "epoch": 35.2,
      "grad_norm": 0.3675675690174103,
      "learning_rate": 2.2115031927140904e-05,
      "loss": 0.0426,
      "step": 7040
    },
    {
      "epoch": 35.25,
      "grad_norm": 0.3226551413536072,
      "learning_rate": 2.1977939541251463e-05,
      "loss": 0.0369,
      "step": 7050
    },
    {
      "epoch": 35.3,
      "grad_norm": 0.28358566761016846,
      "learning_rate": 2.1841153600125684e-05,
      "loss": 0.0357,
      "step": 7060
    },
    {
      "epoch": 35.35,
      "grad_norm": 0.2983461022377014,
      "learning_rate": 2.170467559963267e-05,
      "loss": 0.0439,
      "step": 7070
    },
    {
      "epoch": 35.4,
      "grad_norm": 0.29088157415390015,
      "learning_rate": 2.1568507032273982e-05,
      "loss": 0.0431,
      "step": 7080
    },
    {
      "epoch": 35.45,
      "grad_norm": 0.22399871051311493,
      "learning_rate": 2.1432649387167264e-05,
      "loss": 0.037,
      "step": 7090
    },
    {
      "epoch": 35.5,
      "grad_norm": 0.17647090554237366,
      "learning_rate": 2.1297104150029973e-05,
      "loss": 0.0394,
      "step": 7100
    },
    {
      "epoch": 35.55,
      "grad_norm": 0.7062708735466003,
      "learning_rate": 2.116187280316307e-05,
      "loss": 0.0402,
      "step": 7110
    },
    {
      "epoch": 35.6,
      "grad_norm": 0.2555898427963257,
      "learning_rate": 2.1026956825434908e-05,
      "loss": 0.0437,
      "step": 7120
    },
    {
      "epoch": 35.65,
      "grad_norm": 0.40502646565437317,
      "learning_rate": 2.0892357692265017e-05,
      "loss": 0.0417,
      "step": 7130
    },
    {
      "epoch": 35.7,
      "grad_norm": 0.217569500207901,
      "learning_rate": 2.0758076875607947e-05,
      "loss": 0.0445,
      "step": 7140
    },
    {
      "epoch": 35.75,
      "grad_norm": 0.3940875828266144,
      "learning_rate": 2.0624115843937207e-05,
      "loss": 0.0437,
      "step": 7150
    },
    {
      "epoch": 35.8,
      "grad_norm": 0.3169999122619629,
      "learning_rate": 2.0490476062229157e-05,
      "loss": 0.0383,
      "step": 7160
    },
    {
      "epoch": 35.85,
      "grad_norm": 0.5646221041679382,
      "learning_rate": 2.035715899194704e-05,
      "loss": 0.0461,
      "step": 7170
    },
    {
      "epoch": 35.9,
      "grad_norm": 0.38779908418655396,
      "learning_rate": 2.022416609102499e-05,
      "loss": 0.0404,
      "step": 7180
    },
    {
      "epoch": 35.95,
      "grad_norm": 0.3517095744609833,
      "learning_rate": 2.009149881385205e-05,
      "loss": 0.0417,
      "step": 7190
    },
    {
      "epoch": 36.0,
      "grad_norm": 0.4080592393875122,
      "learning_rate": 1.995915861125634e-05,
      "loss": 0.0392,
      "step": 7200
    },
    {
      "epoch": 36.05,
      "grad_norm": 0.2731874883174896,
      "learning_rate": 1.9827146930489065e-05,
      "loss": 0.0447,
      "step": 7210
    },
    {
      "epoch": 36.1,
      "grad_norm": 0.8624924421310425,
      "learning_rate": 1.9695465215208848e-05,
      "loss": 0.0374,
      "step": 7220
    },
    {
      "epoch": 36.15,
      "grad_norm": 0.2675189673900604,
      "learning_rate": 1.9564114905465813e-05,
      "loss": 0.0397,
      "step": 7230
    },
    {
      "epoch": 36.2,
      "grad_norm": 0.433937132358551,
      "learning_rate": 1.9433097437685936e-05,
      "loss": 0.0407,
      "step": 7240
    },
    {
      "epoch": 36.25,
      "grad_norm": 0.907551109790802,
      "learning_rate": 1.930241424465521e-05,
      "loss": 0.0409,
      "step": 7250
    },
    {
      "epoch": 36.3,
      "grad_norm": 0.23497632145881653,
      "learning_rate": 1.9172066755504115e-05,
      "loss": 0.0378,
      "step": 7260
    },
    {
      "epoch": 36.35,
      "grad_norm": 0.5989136099815369,
      "learning_rate": 1.9042056395691914e-05,
      "loss": 0.042,
      "step": 7270
    },
    {
      "epoch": 36.4,
      "grad_norm": 0.37530848383903503,
      "learning_rate": 1.8912384586991066e-05,
      "loss": 0.0407,
      "step": 7280
    },
    {
      "epoch": 36.45,
      "grad_norm": 0.5285875797271729,
      "learning_rate": 1.8783052747471717e-05,
      "loss": 0.0443,
      "step": 7290
    },
    {
      "epoch": 36.5,
      "grad_norm": 0.3829076290130615,
      "learning_rate": 1.865406229148611e-05,
      "loss": 0.0371,
      "step": 7300
    },
    {
      "epoch": 36.55,
      "grad_norm": 0.3283442258834839,
      "learning_rate": 1.8525414629653233e-05,
      "loss": 0.0377,
      "step": 7310
    },
    {
      "epoch": 36.6,
      "grad_norm": 0.22921478748321533,
      "learning_rate": 1.8397111168843255e-05,
      "loss": 0.0437,
      "step": 7320
    },
    {
      "epoch": 36.65,
      "grad_norm": 0.35409313440322876,
      "learning_rate": 1.8269153312162323e-05,
      "loss": 0.0399,
      "step": 7330
    },
    {
      "epoch": 36.7,
      "grad_norm": 0.40798836946487427,
      "learning_rate": 1.8141542458937054e-05,
      "loss": 0.0375,
      "step": 7340
    },
    {
      "epoch": 36.75,
      "grad_norm": 0.4877980053424835,
      "learning_rate": 1.8014280004699268e-05,
      "loss": 0.0414,
      "step": 7350
    },
    {
      "epoch": 36.8,
      "grad_norm": 0.23646283149719238,
      "learning_rate": 1.788736734117078e-05,
      "loss": 0.0385,
      "step": 7360
    },
    {
      "epoch": 36.85,
      "grad_norm": 0.2852775454521179,
      "learning_rate": 1.7760805856248152e-05,
      "loss": 0.043,
      "step": 7370
    },
    {
      "epoch": 36.9,
      "grad_norm": 0.15556122362613678,
      "learning_rate": 1.7634596933987518e-05,
      "loss": 0.0435,
      "step": 7380
    },
    {
      "epoch": 36.95,
      "grad_norm": 0.22426344454288483,
      "learning_rate": 1.7508741954589404e-05,
      "loss": 0.0368,
      "step": 7390
    },
    {
      "epoch": 37.0,
      "grad_norm": 0.41222405433654785,
      "learning_rate": 1.7383242294383717e-05,
      "loss": 0.0415,
      "step": 7400
    },
    {
      "epoch": 37.05,
      "grad_norm": 0.5053587555885315,
      "learning_rate": 1.7258099325814632e-05,
      "loss": 0.04,
      "step": 7410
    },
    {
      "epoch": 37.1,
      "grad_norm": 0.2500326633453369,
      "learning_rate": 1.7133314417425594e-05,
      "loss": 0.0381,
      "step": 7420
    },
    {
      "epoch": 37.15,
      "grad_norm": 0.20612084865570068,
      "learning_rate": 1.7008888933844408e-05,
      "loss": 0.0412,
      "step": 7430
    },
    {
      "epoch": 37.2,
      "grad_norm": 0.5284052491188049,
      "learning_rate": 1.6884824235768172e-05,
      "loss": 0.0374,
      "step": 7440
    },
    {
      "epoch": 37.25,
      "grad_norm": 0.371955931186676,
      "learning_rate": 1.6761121679948592e-05,
      "loss": 0.0413,
      "step": 7450
    },
    {
      "epoch": 37.3,
      "grad_norm": 0.325240820646286,
      "learning_rate": 1.663778261917695e-05,
      "loss": 0.0408,
      "step": 7460
    },
    {
      "epoch": 37.35,
      "grad_norm": 0.2626557946205139,
      "learning_rate": 1.651480840226952e-05,
      "loss": 0.045,
      "step": 7470
    },
    {
      "epoch": 37.4,
      "grad_norm": 0.3781748414039612,
      "learning_rate": 1.639220037405258e-05,
      "loss": 0.0434,
      "step": 7480
    },
    {
      "epoch": 37.45,
      "grad_norm": 0.2682613134384155,
      "learning_rate": 1.6269959875347906e-05,
      "loss": 0.0371,
      "step": 7490
    },
    {
      "epoch": 37.5,
      "grad_norm": 0.303692489862442,
      "learning_rate": 1.614808824295802e-05,
      "loss": 0.0387,
      "step": 7500
    },
    {
      "epoch": 37.55,
      "grad_norm": 0.2576656639575958,
      "learning_rate": 1.602658680965152e-05,
      "loss": 0.0399,
      "step": 7510
    },
    {
      "epoch": 37.6,
      "grad_norm": 0.2551245093345642,
      "learning_rate": 1.5905456904148686e-05,
      "loss": 0.0395,
      "step": 7520
    },
    {
      "epoch": 37.65,
      "grad_norm": 0.21612882614135742,
      "learning_rate": 1.57846998511067e-05,
      "loss": 0.0409,
      "step": 7530
    },
    {
      "epoch": 37.7,
      "grad_norm": 0.31360912322998047,
      "learning_rate": 1.566431697110538e-05,
      "loss": 0.0401,
      "step": 7540
    },
    {
      "epoch": 37.75,
      "grad_norm": 0.5262354612350464,
      "learning_rate": 1.554430958063259e-05,
      "loss": 0.0412,
      "step": 7550
    },
    {
      "epoch": 37.8,
      "grad_norm": 0.4412863552570343,
      "learning_rate": 1.5424678992069912e-05,
      "loss": 0.0471,
      "step": 7560
    },
    {
      "epoch": 37.85,
      "grad_norm": 0.3521061837673187,
      "learning_rate": 1.5305426513678362e-05,
      "loss": 0.0422,
      "step": 7570
    },
    {
      "epoch": 37.9,
      "grad_norm": 0.8592746257781982,
      "learning_rate": 1.518655344958388e-05,
      "loss": 0.0367,
      "step": 7580
    },
    {
      "epoch": 37.95,
      "grad_norm": 0.527859628200531,
      "learning_rate": 1.5068061099763275e-05,
      "loss": 0.0389,
      "step": 7590
    },
    {
      "epoch": 38.0,
      "grad_norm": 1.0624752044677734,
      "learning_rate": 1.494995076002988e-05,
      "loss": 0.0417,
      "step": 7600
    },
    {
      "epoch": 38.05,
      "grad_norm": 0.47344064712524414,
      "learning_rate": 1.4832223722019456e-05,
      "loss": 0.0399,
      "step": 7610
    },
    {
      "epoch": 38.1,
      "grad_norm": 0.4575490951538086,
      "learning_rate": 1.4714881273176035e-05,
      "loss": 0.0405,
      "step": 7620
    },
    {
      "epoch": 38.15,
      "grad_norm": 0.33690276741981506,
      "learning_rate": 1.4597924696737835e-05,
      "loss": 0.0409,
      "step": 7630
    },
    {
      "epoch": 38.2,
      "grad_norm": 0.3590207099914551,
      "learning_rate": 1.4481355271723252e-05,
      "loss": 0.0414,
      "step": 7640
    },
    {
      "epoch": 38.25,
      "grad_norm": 0.26785987615585327,
      "learning_rate": 1.4365174272916809e-05,
      "loss": 0.0383,
      "step": 7650
    },
    {
      "epoch": 38.3,
      "grad_norm": 0.4456406831741333,
      "learning_rate": 1.4249382970855319e-05,
      "loss": 0.0442,
      "step": 7660
    },
    {
      "epoch": 38.35,
      "grad_norm": 0.2839334011077881,
      "learning_rate": 1.4133982631813903e-05,
      "loss": 0.0379,
      "step": 7670
    },
    {
      "epoch": 38.4,
      "grad_norm": 0.3741433322429657,
      "learning_rate": 1.4018974517792194e-05,
      "loss": 0.0338,
      "step": 7680
    },
    {
      "epoch": 38.45,
      "grad_norm": 0.4899735748767853,
      "learning_rate": 1.390435988650048e-05,
      "loss": 0.0384,
      "step": 7690
    },
    {
      "epoch": 38.5,
      "grad_norm": 0.29834485054016113,
      "learning_rate": 1.3790139991346006e-05,
      "loss": 0.0426,
      "step": 7700
    },
    {
      "epoch": 38.55,
      "grad_norm": 0.36579495668411255,
      "learning_rate": 1.367631608141926e-05,
      "loss": 0.0394,
      "step": 7710
    },
    {
      "epoch": 38.6,
      "grad_norm": 0.3422141373157501,
      "learning_rate": 1.3562889401480278e-05,
      "loss": 0.0348,
      "step": 7720
    },
    {
      "epoch": 38.65,
      "grad_norm": 1.197156310081482,
      "learning_rate": 1.3449861191945074e-05,
      "loss": 0.0382,
      "step": 7730
    },
    {
      "epoch": 38.7,
      "grad_norm": 0.47582194209098816,
      "learning_rate": 1.3337232688872009e-05,
      "loss": 0.0397,
      "step": 7740
    },
    {
      "epoch": 38.75,
      "grad_norm": 0.21331538259983063,
      "learning_rate": 1.3225005123948364e-05,
      "loss": 0.034,
      "step": 7750
    },
    {
      "epoch": 38.8,
      "grad_norm": 0.6290726661682129,
      "learning_rate": 1.311317972447681e-05,
      "loss": 0.0426,
      "step": 7760
    },
    {
      "epoch": 38.85,
      "grad_norm": 0.3253229856491089,
      "learning_rate": 1.3001757713361996e-05,
      "loss": 0.041,
      "step": 7770
    },
    {
      "epoch": 38.9,
      "grad_norm": 0.39687755703926086,
      "learning_rate": 1.2890740309097204e-05,
      "loss": 0.0381,
      "step": 7780
    },
    {
      "epoch": 38.95,
      "grad_norm": 0.4150811433792114,
      "learning_rate": 1.2780128725750944e-05,
      "loss": 0.0426,
      "step": 7790
    },
    {
      "epoch": 39.0,
      "grad_norm": 0.6048423051834106,
      "learning_rate": 1.266992417295379e-05,
      "loss": 0.0421,
      "step": 7800
    },
    {
      "epoch": 39.05,
      "grad_norm": 0.40411150455474854,
      "learning_rate": 1.2560127855885073e-05,
      "loss": 0.0371,
      "step": 7810
    },
    {
      "epoch": 39.1,
      "grad_norm": 0.2806643843650818,
      "learning_rate": 1.2450740975259745e-05,
      "loss": 0.0387,
      "step": 7820
    },
    {
      "epoch": 39.15,
      "grad_norm": 0.4676588773727417,
      "learning_rate": 1.234176472731517e-05,
      "loss": 0.0349,
      "step": 7830
    },
    {
      "epoch": 39.2,
      "grad_norm": 0.853940486907959,
      "learning_rate": 1.2233200303798158e-05,
      "loss": 0.0377,
      "step": 7840
    },
    {
      "epoch": 39.25,
      "grad_norm": 0.36107537150382996,
      "learning_rate": 1.2125048891951846e-05,
      "loss": 0.0372,
      "step": 7850
    },
    {
      "epoch": 39.3,
      "grad_norm": 0.437497615814209,
      "learning_rate": 1.2017311674502745e-05,
      "loss": 0.036,
      "step": 7860
    },
    {
      "epoch": 39.35,
      "grad_norm": 0.5499402284622192,
      "learning_rate": 1.1909989829647822e-05,
      "loss": 0.0387,
      "step": 7870
    },
    {
      "epoch": 39.4,
      "grad_norm": 0.5952790379524231,
      "learning_rate": 1.1803084531041553e-05,
      "loss": 0.0339,
      "step": 7880
    },
    {
      "epoch": 39.45,
      "grad_norm": 0.3826034963130951,
      "learning_rate": 1.1696596947783162e-05,
      "loss": 0.0416,
      "step": 7890
    },
    {
      "epoch": 39.5,
      "grad_norm": 0.7212252020835876,
      "learning_rate": 1.1590528244403803e-05,
      "loss": 0.0392,
      "step": 7900
    },
    {
      "epoch": 39.55,
      "grad_norm": 0.4670110046863556,
      "learning_rate": 1.148487958085382e-05,
      "loss": 0.0441,
      "step": 7910
    },
    {
      "epoch": 39.6,
      "grad_norm": 0.6097583174705505,
      "learning_rate": 1.1379652112490086e-05,
      "loss": 0.0409,
      "step": 7920
    },
    {
      "epoch": 39.65,
      "grad_norm": 0.43858572840690613,
      "learning_rate": 1.1274846990063315e-05,
      "loss": 0.0378,
      "step": 7930
    },
    {
      "epoch": 39.7,
      "grad_norm": 0.4082120954990387,
      "learning_rate": 1.117046535970554e-05,
      "loss": 0.0414,
      "step": 7940
    },
    {
      "epoch": 39.75,
      "grad_norm": 0.45165523886680603,
      "learning_rate": 1.106650836291755e-05,
      "loss": 0.0417,
      "step": 7950
    },
    {
      "epoch": 39.8,
      "grad_norm": 0.5497769713401794,
      "learning_rate": 1.0962977136556418e-05,
      "loss": 0.0414,
      "step": 7960
    },
    {
      "epoch": 39.85,
      "grad_norm": 0.3740074336528778,
      "learning_rate": 1.0859872812823024e-05,
      "loss": 0.0388,
      "step": 7970
    },
    {
      "epoch": 39.9,
      "grad_norm": 0.4368816614151001,
      "learning_rate": 1.0757196519249747e-05,
      "loss": 0.0396,
      "step": 7980
    },
    {
      "epoch": 39.95,
      "grad_norm": 0.45498383045196533,
      "learning_rate": 1.0654949378688077e-05,
      "loss": 0.0389,
      "step": 7990
    },
    {
      "epoch": 40.0,
      "grad_norm": 1.775468349456787,
      "learning_rate": 1.0553132509296376e-05,
      "loss": 0.0387,
      "step": 8000
    },
    {
      "mimicgen/eval/stack_three/success_rate": 0.4,
      "step": 8000
    },
    {
      "epoch": 40.05,
      "grad_norm": 0.5041416883468628,
      "learning_rate": 1.0451747024527613e-05,
      "loss": 0.0401,
      "step": 8010
    },
    {
      "epoch": 40.1,
      "grad_norm": 0.20094169676303864,
      "learning_rate": 1.0350794033117189e-05,
      "loss": 0.035,
      "step": 8020
    },
    {
      "epoch": 40.15,
      "grad_norm": 0.42335760593414307,
      "learning_rate": 1.0250274639070856e-05,
      "loss": 0.0368,
      "step": 8030
    },
    {
      "epoch": 40.2,
      "grad_norm": 0.34476685523986816,
      "learning_rate": 1.0150189941652599e-05,
      "loss": 0.0372,
      "step": 8040
    },
    {
      "epoch": 40.25,
      "grad_norm": 0.38736361265182495,
      "learning_rate": 1.0050541035372635e-05,
      "loss": 0.0396,
      "step": 8050
    },
    {
      "epoch": 40.3,
      "grad_norm": 0.20145465433597565,
      "learning_rate": 9.951329009975458e-06,
      "loss": 0.0401,
      "step": 8060
    },
    {
      "epoch": 40.35,
      "grad_norm": 0.33405452966690063,
      "learning_rate": 9.852554950427845e-06,
      "loss": 0.0334,
      "step": 8070
    },
    {
      "epoch": 40.4,
      "grad_norm": 0.5580013990402222,
      "learning_rate": 9.754219936907105e-06,
      "loss": 0.0358,
      "step": 8080
    },
    {
      "epoch": 40.45,
      "grad_norm": 0.4656769633293152,
      "learning_rate": 9.656325044789194e-06,
      "loss": 0.0378,
      "step": 8090
    },
    {
      "epoch": 40.5,
      "grad_norm": 0.36446264386177063,
      "learning_rate": 9.55887134463697e-06,
      "loss": 0.038,
      "step": 8100
    },
    {
      "epoch": 40.55,
      "grad_norm": 0.6851218342781067,
      "learning_rate": 9.461859902188475e-06,
      "loss": 0.0394,
      "step": 8110
    },
    {
      "epoch": 40.6,
      "grad_norm": 0.7170999646186829,
      "learning_rate": 9.365291778345303e-06,
      "loss": 0.0447,
      "step": 8120
    },
    {
      "epoch": 40.65,
      "grad_norm": 0.3872458338737488,
      "learning_rate": 9.269168029160991e-06,
      "loss": 0.0394,
      "step": 8130
    },
    {
      "epoch": 40.7,
      "grad_norm": 0.41478461027145386,
      "learning_rate": 9.173489705829447e-06,
      "loss": 0.036,
      "step": 8140
    },
    {
      "epoch": 40.75,
      "grad_norm": 0.5186578631401062,
      "learning_rate": 9.078257854673516e-06,
      "loss": 0.0421,
      "step": 8150
    },
    {
      "epoch": 40.8,
      "grad_norm": 1.0792628526687622,
      "learning_rate": 8.983473517133429e-06,
      "loss": 0.0431,
      "step": 8160
    },
    {
      "epoch": 40.85,
      "grad_norm": 0.37399011850357056,
      "learning_rate": 8.889137729755537e-06,
      "loss": 0.0382,
      "step": 8170
    },
    {
      "epoch": 40.9,
      "grad_norm": 0.1968294233083725,
      "learning_rate": 8.79525152418087e-06,
      "loss": 0.0371,
      "step": 8180
    },
    {
      "epoch": 40.95,
      "grad_norm": 0.1968257874250412,
      "learning_rate": 8.701815927133961e-06,
      "loss": 0.0374,
      "step": 8190
    },
    {
      "epoch": 41.0,
      "grad_norm": 0.29792192578315735,
      "learning_rate": 8.608831960411534e-06,
      "loss": 0.0369,
      "step": 8200
    },
    {
      "epoch": 41.05,
      "grad_norm": 0.23982441425323486,
      "learning_rate": 8.516300640871321e-06,
      "loss": 0.0372,
      "step": 8210
    },
    {
      "epoch": 41.1,
      "grad_norm": 0.8326598405838013,
      "learning_rate": 8.424222980421038e-06,
      "loss": 0.0375,
      "step": 8220
    },
    {
      "epoch": 41.15,
      "grad_norm": 0.20053526759147644,
      "learning_rate": 8.332599986007184e-06,
      "loss": 0.0376,
      "step": 8230
    },
    {
      "epoch": 41.2,
      "grad_norm": 0.3552948534488678,
      "learning_rate": 8.241432659604203e-06,
      "loss": 0.0394,
      "step": 8240
    },
    {
      "epoch": 41.25,
      "grad_norm": 0.46144750714302063,
      "learning_rate": 8.150721998203331e-06,
      "loss": 0.0328,
      "step": 8250
    },
    {
      "epoch": 41.3,
      "grad_norm": 0.24308092892169952,
      "learning_rate": 8.06046899380184e-06,
      "loss": 0.0412,
      "step": 8260
    },
    {
      "epoch": 41.35,
      "grad_norm": 0.36071622371673584,
      "learning_rate": 7.970674633392133e-06,
      "loss": 0.0349,
      "step": 8270
    },
    {
      "epoch": 41.4,
      "grad_norm": 0.24819619953632355,
      "learning_rate": 7.881339898950924e-06,
      "loss": 0.0358,
      "step": 8280
    },
    {
      "epoch": 41.45,
      "grad_norm": 0.4055517315864563,
      "learning_rate": 7.792465767428597e-06,
      "loss": 0.0351,
      "step": 8290
    },
    {
      "epoch": 41.5,
      "grad_norm": 0.31506070494651794,
      "learning_rate": 7.704053210738376e-06,
      "loss": 0.0347,
      "step": 8300
    },
    {
      "epoch": 41.55,
      "grad_norm": 0.6063005924224854,
      "learning_rate": 7.6161031957458494e-06,
      "loss": 0.0394,
      "step": 8310
    },
    {
      "epoch": 41.6,
      "grad_norm": 0.4536493122577667,
      "learning_rate": 7.5286166842582605e-06,
      "loss": 0.0391,
      "step": 8320
    },
    {
      "epoch": 41.65,
      "grad_norm": 0.3163391351699829,
      "learning_rate": 7.4415946330140814e-06,
      "loss": 0.0337,
      "step": 8330
    },
    {
      "epoch": 41.7,
      "grad_norm": 0.27775922417640686,
      "learning_rate": 7.3550379936725644e-06,
      "loss": 0.035,
      "step": 8340
    },
    {
      "epoch": 41.75,
      "grad_norm": 0.4332856237888336,
      "learning_rate": 7.2689477128032035e-06,
      "loss": 0.0414,
      "step": 8350
    },
    {
      "epoch": 41.8,
      "grad_norm": 0.22718693315982819,
      "learning_rate": 7.183324731875551e-06,
      "loss": 0.0411,
      "step": 8360
    },
    {
      "epoch": 41.85,
      "grad_norm": 0.21669556200504303,
      "learning_rate": 7.098169987248782e-06,
      "loss": 0.0354,
      "step": 8370
    },
    {
      "epoch": 41.9,
      "grad_norm": 0.2533864974975586,
      "learning_rate": 7.013484410161553e-06,
      "loss": 0.0342,
      "step": 8380
    },
    {
      "epoch": 41.95,
      "grad_norm": 0.4258666932582855,
      "learning_rate": 6.92926892672176e-06,
      "loss": 0.0384,
      "step": 8390
    },
    {
      "epoch": 42.0,
      "grad_norm": 0.8555409908294678,
      "learning_rate": 6.845524457896446e-06,
      "loss": 0.0354,
      "step": 8400
    },
    {
      "epoch": 42.05,
      "grad_norm": 0.32595473527908325,
      "learning_rate": 6.7622519195017165e-06,
      "loss": 0.039,
      "step": 8410
    },
    {
      "epoch": 42.1,
      "grad_norm": 0.21946340799331665,
      "learning_rate": 6.679452222192684e-06,
      "loss": 0.0383,
      "step": 8420
    },
    {
      "epoch": 42.15,
      "grad_norm": 0.6571286916732788,
      "learning_rate": 6.597126271453579e-06,
      "loss": 0.0388,
      "step": 8430
    },
    {
      "epoch": 42.2,
      "grad_norm": 0.644687831401825,
      "learning_rate": 6.51527496758782e-06,
      "loss": 0.0368,
      "step": 8440
    },
    {
      "epoch": 42.25,
      "grad_norm": 0.3112056255340576,
      "learning_rate": 6.433899205708155e-06,
      "loss": 0.0339,
      "step": 8450
    },
    {
      "epoch": 42.3,
      "grad_norm": 0.30681154131889343,
      "learning_rate": 6.352999875726856e-06,
      "loss": 0.0372,
      "step": 8460
    },
    {
      "epoch": 42.35,
      "grad_norm": 0.3423842489719391,
      "learning_rate": 6.272577862346052e-06,
      "loss": 0.0349,
      "step": 8470
    },
    {
      "epoch": 42.4,
      "grad_norm": 0.31989842653274536,
      "learning_rate": 6.192634045047996e-06,
      "loss": 0.0358,
      "step": 8480
    },
    {
      "epoch": 42.45,
      "grad_norm": 0.20057649910449982,
      "learning_rate": 6.113169298085458e-06,
      "loss": 0.0323,
      "step": 8490
    },
    {
      "epoch": 42.5,
      "grad_norm": 0.6665655374526978,
      "learning_rate": 6.034184490472195e-06,
      "loss": 0.0342,
      "step": 8500
    },
    {
      "epoch": 42.55,
      "grad_norm": 1.1850807666778564,
      "learning_rate": 5.955680485973386e-06,
      "loss": 0.0376,
      "step": 8510
    },
    {
      "epoch": 42.6,
      "grad_norm": 0.3246558904647827,
      "learning_rate": 5.877658143096265e-06,
      "loss": 0.0395,
      "step": 8520
    },
    {
      "epoch": 42.65,
      "grad_norm": 0.20522993803024292,
      "learning_rate": 5.800118315080661e-06,
      "loss": 0.0367,
      "step": 8530
    },
    {
      "epoch": 42.7,
      "grad_norm": 0.3548533320426941,
      "learning_rate": 5.723061849889716e-06,
      "loss": 0.0364,
      "step": 8540
    },
    {
      "epoch": 42.75,
      "grad_norm": 0.26185810565948486,
      "learning_rate": 5.646489590200604e-06,
      "loss": 0.0349,
      "step": 8550
    },
    {
      "epoch": 42.8,
      "grad_norm": 1.0496854782104492,
      "learning_rate": 5.570402373395256e-06,
      "loss": 0.039,
      "step": 8560
    },
    {
      "epoch": 42.85,
      "grad_norm": 0.4175034165382385,
      "learning_rate": 5.494801031551305e-06,
      "loss": 0.0425,
      "step": 8570
    },
    {
      "epoch": 42.9,
      "grad_norm": 0.5267932415008545,
      "learning_rate": 5.41968639143291e-06,
      "loss": 0.0342,
      "step": 8580
    },
    {
      "epoch": 42.95,
      "grad_norm": 0.19436116516590118,
      "learning_rate": 5.345059274481751e-06,
      "loss": 0.0399,
      "step": 8590
    },
    {
      "epoch": 43.0,
      "grad_norm": 0.4495581388473511,
      "learning_rate": 5.270920496808002e-06,
      "loss": 0.0349,
      "step": 8600
    },
    {
      "epoch": 43.05,
      "grad_norm": 0.3514513373374939,
      "learning_rate": 5.1972708691814695e-06,
      "loss": 0.0381,
      "step": 8610
    },
    {
      "epoch": 43.1,
      "grad_norm": 0.3388993740081787,
      "learning_rate": 5.124111197022674e-06,
      "loss": 0.0376,
      "step": 8620
    },
    {
      "epoch": 43.15,
      "grad_norm": 0.3133702874183655,
      "learning_rate": 5.051442280394081e-06,
      "loss": 0.0408,
      "step": 8630
    },
    {
      "epoch": 43.2,
      "grad_norm": 0.20314310491085052,
      "learning_rate": 4.979264913991322e-06,
      "loss": 0.0368,
      "step": 8640
    },
    {
      "epoch": 43.25,
      "grad_norm": 0.23970478773117065,
      "learning_rate": 4.907579887134489e-06,
      "loss": 0.0341,
      "step": 8650
    },
    {
      "epoch": 43.3,
      "grad_norm": 0.7166557312011719,
      "learning_rate": 4.836387983759572e-06,
      "loss": 0.035,
      "step": 8660
    },
    {
      "epoch": 43.35,
      "grad_norm": 0.3872759938240051,
      "learning_rate": 4.765689982409816e-06,
      "loss": 0.0362,
      "step": 8670
    },
    {
      "epoch": 43.4,
      "grad_norm": 0.5117878913879395,
      "learning_rate": 4.695486656227233e-06,
      "loss": 0.0369,
      "step": 8680
    },
    {
      "epoch": 43.45,
      "grad_norm": 0.5676246881484985,
      "learning_rate": 4.625778772944156e-06,
      "loss": 0.0327,
      "step": 8690
    },
    {
      "epoch": 43.5,
      "grad_norm": 0.49997153878211975,
      "learning_rate": 4.556567094874825e-06,
      "loss": 0.0346,
      "step": 8700
    },
    {
      "epoch": 43.55,
      "grad_norm": 0.47356289625167847,
      "learning_rate": 4.487852378907059e-06,
      "loss": 0.0344,
      "step": 8710
    },
    {
      "epoch": 43.6,
      "grad_norm": 0.36372750997543335,
      "learning_rate": 4.419635376493986e-06,
      "loss": 0.0397,
      "step": 8720
    },
    {
      "epoch": 43.65,
      "grad_norm": 0.23237326741218567,
      "learning_rate": 4.351916833645825e-06,
      "loss": 0.0352,
      "step": 8730
    },
    {
      "epoch": 43.7,
      "grad_norm": 0.45296064019203186,
      "learning_rate": 4.284697490921691e-06,
      "loss": 0.0301,
      "step": 8740
    },
    {
      "epoch": 43.75,
      "grad_norm": 0.37872350215911865,
      "learning_rate": 4.2179780834215585e-06,
      "loss": 0.04,
      "step": 8750
    },
    {
      "epoch": 43.8,
      "grad_norm": 0.2786872982978821,
      "learning_rate": 4.151759340778178e-06,
      "loss": 0.0366,
      "step": 8760
    },
    {
      "epoch": 43.85,
      "grad_norm": 0.6333494186401367,
      "learning_rate": 4.086041987149109e-06,
      "loss": 0.0327,
      "step": 8770
    },
    {
      "epoch": 43.9,
      "grad_norm": 0.5084655284881592,
      "learning_rate": 4.020826741208811e-06,
      "loss": 0.041,
      "step": 8780
    },
    {
      "epoch": 43.95,
      "grad_norm": 0.3386616110801697,
      "learning_rate": 3.956114316140746e-06,
      "loss": 0.0381,
      "step": 8790
    },
    {
      "epoch": 44.0,
      "grad_norm": 0.410385400056839,
      "learning_rate": 3.891905419629643e-06,
      "loss": 0.0391,
      "step": 8800
    },
    {
      "epoch": 44.05,
      "grad_norm": 1.0861644744873047,
      "learning_rate": 3.8282007538536946e-06,
      "loss": 0.0404,
      "step": 8810
    },
    {
      "epoch": 44.1,
      "grad_norm": 0.3386635482311249,
      "learning_rate": 3.7650010154769265e-06,
      "loss": 0.0372,
      "step": 8820
    },
    {
      "epoch": 44.15,
      "grad_norm": 0.688351035118103,
      "learning_rate": 3.7023068956415608e-06,
      "loss": 0.0339,
      "step": 8830
    },
    {
      "epoch": 44.2,
      "grad_norm": 0.24957528710365295,
      "learning_rate": 3.6401190799604303e-06,
      "loss": 0.0412,
      "step": 8840
    },
    {
      "epoch": 44.25,
      "grad_norm": 0.3447701334953308,
      "learning_rate": 3.578438248509536e-06,
      "loss": 0.0359,
      "step": 8850
    },
    {
      "epoch": 44.3,
      "grad_norm": 0.2675687074661255,
      "learning_rate": 3.5172650758205583e-06,
      "loss": 0.0383,
      "step": 8860
    },
    {
      "epoch": 44.35,
      "grad_norm": 0.3373061418533325,
      "learning_rate": 3.45660023087353e-06,
      "loss": 0.033,
      "step": 8870
    },
    {
      "epoch": 44.4,
      "grad_norm": 0.7125996947288513,
      "learning_rate": 3.3964443770894528e-06,
      "loss": 0.0337,
      "step": 8880
    },
    {
      "epoch": 44.45,
      "grad_norm": 0.31101202964782715,
      "learning_rate": 3.3367981723231245e-06,
      "loss": 0.0344,
      "step": 8890
    },
    {
      "epoch": 44.5,
      "grad_norm": 0.7502497434616089,
      "learning_rate": 3.2776622688558746e-06,
      "loss": 0.0324,
      "step": 8900
    },
    {
      "epoch": 44.55,
      "grad_norm": 0.685218095779419,
      "learning_rate": 3.2190373133884677e-06,
      "loss": 0.0371,
      "step": 8910
    },
    {
      "epoch": 44.6,
      "grad_norm": 0.4297686517238617,
      "learning_rate": 3.1609239470340446e-06,
      "loss": 0.0346,
      "step": 8920
    },
    {
      "epoch": 44.65,
      "grad_norm": 0.46178340911865234,
      "learning_rate": 3.1033228053110373e-06,
      "loss": 0.0365,
      "step": 8930
    },
    {
      "epoch": 44.7,
      "grad_norm": 0.7480696439743042,
      "learning_rate": 3.0462345181363314e-06,
      "loss": 0.0376,
      "step": 8940
    },
    {
      "epoch": 44.75,
      "grad_norm": 0.5215370059013367,
      "learning_rate": 2.9896597098182654e-06,
      "loss": 0.038,
      "step": 8950
    },
    {
      "epoch": 44.8,
      "grad_norm": 0.5113524794578552,
      "learning_rate": 2.933598999049891e-06,
      "loss": 0.0346,
      "step": 8960
    },
    {
      "epoch": 44.85,
      "grad_norm": 0.41008108854293823,
      "learning_rate": 2.8780529989021697e-06,
      "loss": 0.0333,
      "step": 8970
    },
    {
      "epoch": 44.9,
      "grad_norm": 0.7631787657737732,
      "learning_rate": 2.823022316817242e-06,
      "loss": 0.0389,
      "step": 8980
    },
    {
      "epoch": 44.95,
      "grad_norm": 0.31746816635131836,
      "learning_rate": 2.7685075546018456e-06,
      "loss": 0.0327,
      "step": 8990
    },
    {
      "epoch": 45.0,
      "grad_norm": 0.9666823148727417,
      "learning_rate": 2.7145093084206598e-06,
      "loss": 0.0373,
      "step": 9000
    },
    {
      "mimicgen/eval/stack_three/success_rate": 0.4,
      "step": 9000
    },
    {
      "epoch": 45.05,
      "grad_norm": 1.244088053703308,
      "learning_rate": 2.661028168789892e-06,
      "loss": 0.03,
      "step": 9010
    },
    {
      "epoch": 45.1,
      "grad_norm": 0.4780507981777191,
      "learning_rate": 2.6080647205706855e-06,
      "loss": 0.0339,
      "step": 9020
    },
    {
      "epoch": 45.15,
      "grad_norm": 0.2992382049560547,
      "learning_rate": 2.555619542962834e-06,
      "loss": 0.0346,
      "step": 9030
    },
    {
      "epoch": 45.2,
      "grad_norm": 0.35845303535461426,
      "learning_rate": 2.503693209498409e-06,
      "loss": 0.0393,
      "step": 9040
    },
    {
      "epoch": 45.25,
      "grad_norm": 0.44914886355400085,
      "learning_rate": 2.452286288035449e-06,
      "loss": 0.0345,
      "step": 9050
    },
    {
      "epoch": 45.3,
      "grad_norm": 0.3233708143234253,
      "learning_rate": 2.4013993407518363e-06,
      "loss": 0.0344,
      "step": 9060
    },
    {
      "epoch": 45.35,
      "grad_norm": 0.2433205246925354,
      "learning_rate": 2.351032924139063e-06,
      "loss": 0.0338,
      "step": 9070
    },
    {
      "epoch": 45.4,
      "grad_norm": 0.547118067741394,
      "learning_rate": 2.30118758899619e-06,
      "loss": 0.0318,
      "step": 9080
    },
    {
      "epoch": 45.45,
      "grad_norm": 0.507278561592102,
      "learning_rate": 2.2518638804238157e-06,
      "loss": 0.0429,
      "step": 9090
    },
    {
      "epoch": 45.5,
      "grad_norm": 0.9034734964370728,
      "learning_rate": 2.203062337818118e-06,
      "loss": 0.0354,
      "step": 9100
    },
    {
      "epoch": 45.55,
      "grad_norm": 0.22559940814971924,
      "learning_rate": 2.1547834948649483e-06,
      "loss": 0.0353,
      "step": 9110
    },
    {
      "epoch": 45.6,
      "grad_norm": 0.2447793334722519,
      "learning_rate": 2.1070278795340017e-06,
      "loss": 0.0323,
      "step": 9120
    },
    {
      "epoch": 45.65,
      "grad_norm": 0.2207028865814209,
      "learning_rate": 2.059796014073029e-06,
      "loss": 0.0359,
      "step": 9130
    },
    {
      "epoch": 45.7,
      "grad_norm": 0.3441169261932373,
      "learning_rate": 2.01308841500214e-06,
      "loss": 0.0358,
      "step": 9140
    },
    {
      "epoch": 45.75,
      "grad_norm": 0.493439257144928,
      "learning_rate": 1.9669055931081704e-06,
      "loss": 0.0368,
      "step": 9150
    },
    {
      "epoch": 45.8,
      "grad_norm": 0.4438033401966095,
      "learning_rate": 1.9212480534390507e-06,
      "loss": 0.0374,
      "step": 9160
    },
    {
      "epoch": 45.85,
      "grad_norm": 0.39255279302597046,
      "learning_rate": 1.8761162952983246e-06,
      "loss": 0.0341,
      "step": 9170
    },
    {
      "epoch": 45.9,
      "grad_norm": 0.41188180446624756,
      "learning_rate": 1.8315108122396618e-06,
      "loss": 0.0377,
      "step": 9180
    },
    {
      "epoch": 45.95,
      "grad_norm": 0.2564510703086853,
      "learning_rate": 1.787432092061475e-06,
      "loss": 0.0394,
      "step": 9190
    },
    {
      "epoch": 46.0,
      "grad_norm": 0.545372724533081,
      "learning_rate": 1.743880616801602e-06,
      "loss": 0.0357,
      "step": 9200
    },
    {
      "epoch": 46.05,
      "grad_norm": 0.5007700324058533,
      "learning_rate": 1.7008568627319865e-06,
      "loss": 0.0383,
      "step": 9210
    },
    {
      "epoch": 46.1,
      "grad_norm": 0.33164748549461365,
      "learning_rate": 1.6583613003535226e-06,
      "loss": 0.0386,
      "step": 9220
    },
    {
      "epoch": 46.15,
      "grad_norm": 0.27492326498031616,
      "learning_rate": 1.6163943943908522e-06,
      "loss": 0.0367,
      "step": 9230
    },
    {
      "epoch": 46.2,
      "grad_norm": 0.3220658302307129,
      "learning_rate": 1.5749566037873476e-06,
      "loss": 0.035,
      "step": 9240
    },
    {
      "epoch": 46.25,
      "grad_norm": 0.1789606660604477,
      "learning_rate": 1.5340483817000428e-06,
      "loss": 0.0358,
      "step": 9250
    },
    {
      "epoch": 46.3,
      "grad_norm": 0.3349335193634033,
      "learning_rate": 1.4936701754947101e-06,
      "loss": 0.0327,
      "step": 9260
    },
    {
      "epoch": 46.35,
      "grad_norm": 0.426546573638916,
      "learning_rate": 1.4538224267409361e-06,
      "loss": 0.0367,
      "step": 9270
    },
    {
      "epoch": 46.4,
      "grad_norm": 0.5170747637748718,
      "learning_rate": 1.414505571207314e-06,
      "loss": 0.0365,
      "step": 9280
    },
    {
      "epoch": 46.45,
      "grad_norm": 0.2526771128177643,
      "learning_rate": 1.3757200388566816e-06,
      "loss": 0.0362,
      "step": 9290
    },
    {
      "epoch": 46.5,
      "grad_norm": 0.33976858854293823,
      "learning_rate": 1.3374662538414074e-06,
      "loss": 0.0369,
      "step": 9300
    },
    {
      "epoch": 46.55,
      "grad_norm": 0.6400748491287231,
      "learning_rate": 1.2997446344987617e-06,
      "loss": 0.0334,
      "step": 9310
    },
    {
      "epoch": 46.6,
      "grad_norm": 0.2609960734844208,
      "learning_rate": 1.262555593346315e-06,
      "loss": 0.0389,
      "step": 9320
    },
    {
      "epoch": 46.65,
      "grad_norm": 0.24406810104846954,
      "learning_rate": 1.2258995370774685e-06,
      "loss": 0.034,
      "step": 9330
    },
    {
      "epoch": 46.7,
      "grad_norm": 0.3102216422557831,
      "learning_rate": 1.1897768665569798e-06,
      "loss": 0.0314,
      "step": 9340
    },
    {
      "epoch": 46.75,
      "grad_norm": 0.8757972121238708,
      "learning_rate": 1.1541879768165954e-06,
      "loss": 0.0282,
      "step": 9350
    },
    {
      "epoch": 46.8,
      "grad_norm": 0.3995205760002136,
      "learning_rate": 1.1191332570507085e-06,
      "loss": 0.0356,
      "step": 9360
    },
    {
      "epoch": 46.85,
      "grad_norm": 0.2676095962524414,
      "learning_rate": 1.0846130906121132e-06,
      "loss": 0.0356,
      "step": 9370
    },
    {
      "epoch": 46.9,
      "grad_norm": 0.8241976499557495,
      "learning_rate": 1.0506278550078131e-06,
      "loss": 0.0367,
      "step": 9380
    },
    {
      "epoch": 46.95,
      "grad_norm": 0.3184134066104889,
      "learning_rate": 1.0171779218949185e-06,
      "loss": 0.0363,
      "step": 9390
    },
    {
      "epoch": 47.0,
      "grad_norm": 1.0161032676696777,
      "learning_rate": 9.842636570765174e-07,
      "loss": 0.0336,
      "step": 9400
    },
    {
      "epoch": 47.05,
      "grad_norm": 0.6136456727981567,
      "learning_rate": 9.518854204977612e-07,
      "loss": 0.0357,
      "step": 9410
    },
    {
      "epoch": 47.1,
      "grad_norm": 0.4089236855506897,
      "learning_rate": 9.200435662418349e-07,
      "loss": 0.0345,
      "step": 9420
    },
    {
      "epoch": 47.15,
      "grad_norm": 0.24304096400737762,
      "learning_rate": 8.887384425261658e-07,
      "loss": 0.0344,
      "step": 9430
    },
    {
      "epoch": 47.2,
      "grad_norm": 0.19187456369400024,
      "learning_rate": 8.579703916985648e-07,
      "loss": 0.0359,
      "step": 9440
    },
    {
      "epoch": 47.25,
      "grad_norm": 0.2230997383594513,
      "learning_rate": 8.277397502335194e-07,
      "loss": 0.0378,
      "step": 9450
    },
    {
      "epoch": 47.3,
      "grad_norm": 0.25574415922164917,
      "learning_rate": 7.980468487284675e-07,
      "loss": 0.0341,
      "step": 9460
    },
    {
      "epoch": 47.35,
      "grad_norm": 0.27565649151802063,
      "learning_rate": 7.688920119002297e-07,
      "loss": 0.0291,
      "step": 9470
    },
    {
      "epoch": 47.4,
      "grad_norm": 0.3576205372810364,
      "learning_rate": 7.402755585814269e-07,
      "loss": 0.0371,
      "step": 9480
    },
    {
      "epoch": 47.45,
      "grad_norm": 0.32880523800849915,
      "learning_rate": 7.121978017170073e-07,
      "loss": 0.0311,
      "step": 9490
    },
    {
      "epoch": 47.5,
      "grad_norm": 0.47625529766082764,
      "learning_rate": 6.846590483608306e-07,
      "loss": 0.0375,
      "step": 9500
    },
    {
      "epoch": 47.55,
      "grad_norm": 0.4562666714191437,
      "learning_rate": 6.576595996722834e-07,
      "loss": 0.037,
      "step": 9510
    },
    {
      "epoch": 47.6,
      "grad_norm": 0.41174784302711487,
      "learning_rate": 6.311997509130141e-07,
      "loss": 0.0317,
      "step": 9520
    },
    {
      "epoch": 47.65,
      "grad_norm": 0.42239701747894287,
      "learning_rate": 6.052797914436803e-07,
      "loss": 0.0325,
      "step": 9530
    },
    {
      "epoch": 47.7,
      "grad_norm": 0.3325954079627991,
      "learning_rate": 5.799000047208181e-07,
      "loss": 0.0397,
      "step": 9540
    },
    {
      "epoch": 47.75,
      "grad_norm": 1.2204281091690063,
      "learning_rate": 5.550606682937054e-07,
      "loss": 0.0372,
      "step": 9550
    },
    {
      "epoch": 47.8,
      "grad_norm": 0.4140435755252838,
      "learning_rate": 5.307620538013481e-07,
      "loss": 0.0346,
      "step": 9560
    },
    {
      "epoch": 47.85,
      "grad_norm": 0.30938148498535156,
      "learning_rate": 5.070044269694874e-07,
      "loss": 0.0388,
      "step": 9570
    },
    {
      "epoch": 47.9,
      "grad_norm": 0.5286203622817993,
      "learning_rate": 4.837880476077417e-07,
      "loss": 0.0385,
      "step": 9580
    },
    {
      "epoch": 47.95,
      "grad_norm": 0.24596525728702545,
      "learning_rate": 4.6111316960670835e-07,
      "loss": 0.0399,
      "step": 9590
    },
    {
      "epoch": 48.0,
      "grad_norm": 0.2804517149925232,
      "learning_rate": 4.389800409352218e-07,
      "loss": 0.032,
      "step": 9600
    },
    {
      "epoch": 48.05,
      "grad_norm": 0.5426785349845886,
      "learning_rate": 4.173889036376277e-07,
      "loss": 0.0353,
      "step": 9610
    },
    {
      "epoch": 48.1,
      "grad_norm": 0.731410562992096,
      "learning_rate": 3.963399938311463e-07,
      "loss": 0.0384,
      "step": 9620
    },
    {
      "epoch": 48.15,
      "grad_norm": 0.28942984342575073,
      "learning_rate": 3.7583354170328545e-07,
      "loss": 0.0369,
      "step": 9630
    },
    {
      "epoch": 48.2,
      "grad_norm": 0.5436448454856873,
      "learning_rate": 3.558697715093207e-07,
      "loss": 0.0349,
      "step": 9640
    },
    {
      "epoch": 48.25,
      "grad_norm": 0.35626929998397827,
      "learning_rate": 3.3644890156983576e-07,
      "loss": 0.0392,
      "step": 9650
    },
    {
      "epoch": 48.3,
      "grad_norm": 0.4438352882862091,
      "learning_rate": 3.175711442683638e-07,
      "loss": 0.0378,
      "step": 9660
    },
    {
      "epoch": 48.35,
      "grad_norm": 0.5708608627319336,
      "learning_rate": 2.9923670604902197e-07,
      "loss": 0.0365,
      "step": 9670
    },
    {
      "epoch": 48.4,
      "grad_norm": 0.29777252674102783,
      "learning_rate": 2.814457874143028e-07,
      "loss": 0.0372,
      "step": 9680
    },
    {
      "epoch": 48.45,
      "grad_norm": 0.32535362243652344,
      "learning_rate": 2.641985829228366e-07,
      "loss": 0.0358,
      "step": 9690
    },
    {
      "epoch": 48.5,
      "grad_norm": 0.2530880272388458,
      "learning_rate": 2.474952811872877e-07,
      "loss": 0.0309,
      "step": 9700
    },
    {
      "epoch": 48.55,
      "grad_norm": 0.13639384508132935,
      "learning_rate": 2.3133606487228397e-07,
      "loss": 0.0399,
      "step": 9710
    },
    {
      "epoch": 48.6,
      "grad_norm": 0.15416240692138672,
      "learning_rate": 2.157211106924295e-07,
      "loss": 0.0352,
      "step": 9720
    },
    {
      "epoch": 48.65,
      "grad_norm": 0.6885636448860168,
      "learning_rate": 2.006505894103672e-07,
      "loss": 0.0352,
      "step": 9730
    },
    {
      "epoch": 48.7,
      "grad_norm": 0.24868763983249664,
      "learning_rate": 1.8612466583489696e-07,
      "loss": 0.0374,
      "step": 9740
    },
    {
      "epoch": 48.75,
      "grad_norm": 0.3589503765106201,
      "learning_rate": 1.7214349881918834e-07,
      "loss": 0.0325,
      "step": 9750
    },
    {
      "epoch": 48.8,
      "grad_norm": 0.5011966824531555,
      "learning_rate": 1.5870724125904845e-07,
      "loss": 0.0304,
      "step": 9760
    },
    {
      "epoch": 48.85,
      "grad_norm": 0.8502700924873352,
      "learning_rate": 1.4581604009124006e-07,
      "loss": 0.0323,
      "step": 9770
    },
    {
      "epoch": 48.9,
      "grad_norm": 0.23579604923725128,
      "learning_rate": 1.334700362918717e-07,
      "loss": 0.0406,
      "step": 9780
    },
    {
      "epoch": 48.95,
      "grad_norm": 0.25013628602027893,
      "learning_rate": 1.2166936487486015e-07,
      "loss": 0.0312,
      "step": 9790
    },
    {
      "epoch": 49.0,
      "grad_norm": 0.3596136271953583,
      "learning_rate": 1.1041415489045914e-07,
      "loss": 0.037,
      "step": 9800
    },
    {
      "epoch": 49.05,
      "grad_norm": 0.44026029109954834,
      "learning_rate": 9.970452942384412e-08,
      "loss": 0.0354,
      "step": 9810
    },
    {
      "epoch": 49.1,
      "grad_norm": 0.17597776651382446,
      "learning_rate": 8.954060559375754e-08,
      "loss": 0.035,
      "step": 9820
    },
    {
      "epoch": 49.15,
      "grad_norm": 0.3062511384487152,
      "learning_rate": 7.99224945512489e-08,
      "loss": 0.0354,
      "step": 9830
    },
    {
      "epoch": 49.2,
      "grad_norm": 0.26529982686042786,
      "learning_rate": 7.085030147843675e-08,
      "loss": 0.0315,
      "step": 9840
    },
    {
      "epoch": 49.25,
      "grad_norm": 0.20628805458545685,
      "learning_rate": 6.232412558736523e-08,
      "loss": 0.0383,
      "step": 9850
    },
    {
      "epoch": 49.3,
      "grad_norm": 0.5765173435211182,
      "learning_rate": 5.434406011893822e-08,
      "loss": 0.0364,
      "step": 9860
    },
    {
      "epoch": 49.35,
      "grad_norm": 0.312528133392334,
      "learning_rate": 4.6910192341864664e-08,
      "loss": 0.0319,
      "step": 9870
    },
    {
      "epoch": 49.4,
      "grad_norm": 0.6173632144927979,
      "learning_rate": 4.0022603551737035e-08,
      "loss": 0.0321,
      "step": 9880
    },
    {
      "epoch": 49.45,
      "grad_norm": 0.3045026659965515,
      "learning_rate": 3.3681369070120985e-08,
      "loss": 0.0351,
      "step": 9890
    },
    {
      "epoch": 49.5,
      "grad_norm": 1.0620832443237305,
      "learning_rate": 2.7886558243744866e-08,
      "loss": 0.037,
      "step": 9900
    },
    {
      "epoch": 49.55,
      "grad_norm": 0.41756436228752136,
      "learning_rate": 2.2638234443722596e-08,
      "loss": 0.0342,
      "step": 9910
    },
    {
      "epoch": 49.6,
      "grad_norm": 0.28733766078948975,
      "learning_rate": 1.7936455064887504e-08,
      "loss": 0.039,
      "step": 9920
    },
    {
      "epoch": 49.65,
      "grad_norm": 0.3406146466732025,
      "learning_rate": 1.378127152514841e-08,
      "loss": 0.0327,
      "step": 9930
    },
    {
      "epoch": 49.7,
      "grad_norm": 0.6264913082122803,
      "learning_rate": 1.0172729264917857e-08,
      "loss": 0.0365,
      "step": 9940
    },
    {
      "epoch": 49.75,
      "grad_norm": 0.545315146446228,
      "learning_rate": 7.1108677466458215e-09,
      "loss": 0.0361,
      "step": 9950
    },
    {
      "epoch": 49.8,
      "grad_norm": 0.3003397583961487,
      "learning_rate": 4.595720454353414e-09,
      "loss": 0.0369,
      "step": 9960
    },
    {
      "epoch": 49.85,
      "grad_norm": 0.39046424627304077,
      "learning_rate": 2.627314893294264e-09,
      "loss": 0.036,
      "step": 9970
    },
    {
      "epoch": 49.9,
      "grad_norm": 0.4943208396434784,
      "learning_rate": 1.2056725896270048e-09,
      "loss": 0.0363,
      "step": 9980
    },
    {
      "epoch": 49.95,
      "grad_norm": 0.4176075756549835,
      "learning_rate": 3.308090902098826e-10,
      "loss": 0.038,
      "step": 9990
    },
    {
      "epoch": 50.0,
      "grad_norm": 0.4269272983074188,
      "learning_rate": 2.7339624120159555e-12,
      "loss": 0.0375,
      "step": 10000
    }
  ],
  "logging_steps": 10,
  "max_steps": 10000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 50,
  "save_steps": 1000,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 0.0,
  "train_batch_size": 64,
  "trial_name": null,
  "trial_params": null
}