{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 1.9073486328125e-05,
  "eval_steps": 500,
  "global_step": 20000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0,
      "model_forward_time": 7.404147624969482,
      "step": 0
    },
    {
      "dynamics_loss_avg": 0.0649011954665184,
      "epoch": 0,
      "step": 0
    },
    {
      "action_loss_avg": 0.03883320838212967,
      "epoch": 0,
      "step": 0
    },
    {
      "epoch": 0,
      "step": 0,
      "training_step_time": 18.968783617019653
    },
    {
      "epoch": 9.5367431640625e-10,
      "model_forward_time": 5.522135496139526,
      "step": 1
    },
    {
      "epoch": 9.5367431640625e-10,
      "step": 1,
      "training_step_time": 16.77615976333618
    },
    {
      "epoch": 1.9073486328125e-09,
      "model_forward_time": 5.5198752880096436,
      "step": 2
    },
    {
      "epoch": 1.9073486328125e-09,
      "step": 2,
      "training_step_time": 16.811382293701172
    },
    {
      "epoch": 2.86102294921875e-09,
      "model_forward_time": 5.554849147796631,
      "step": 3
    },
    {
      "epoch": 2.86102294921875e-09,
      "step": 3,
      "training_step_time": 16.879974365234375
    },
    {
      "epoch": 3.814697265625e-09,
      "model_forward_time": 5.572574138641357,
      "step": 4
    },
    {
      "epoch": 3.814697265625e-09,
      "step": 4,
      "training_step_time": 17.319549322128296
    },
    {
      "epoch": 4.76837158203125e-09,
      "model_forward_time": 5.5796332359313965,
      "step": 5
    },
    {
      "epoch": 4.76837158203125e-09,
      "step": 5,
      "training_step_time": 16.9878671169281
    },
    {
      "epoch": 5.7220458984375e-09,
      "model_forward_time": 5.621728420257568,
      "step": 6
    },
    {
      "epoch": 5.7220458984375e-09,
      "step": 6,
      "training_step_time": 17.03107714653015
    },
    {
      "epoch": 6.67572021484375e-09,
      "model_forward_time": 5.572687864303589,
      "step": 7
    },
    {
      "epoch": 6.67572021484375e-09,
      "step": 7,
      "training_step_time": 16.977957487106323
    },
    {
      "epoch": 7.62939453125e-09,
      "model_forward_time": 5.591064929962158,
      "step": 8
    },
    {
      "epoch": 7.62939453125e-09,
      "step": 8,
      "training_step_time": 17.02068519592285
    },
    {
      "epoch": 8.58306884765625e-09,
      "model_forward_time": 5.592599868774414,
      "step": 9
    },
    {
      "epoch": 8.58306884765625e-09,
      "step": 9,
      "training_step_time": 17.067595958709717
    },
    {
      "epoch": 9.5367431640625e-09,
      "grad_norm": 2.9324047565460205,
      "learning_rate": 1e-05,
      "loss": 0.1749,
      "step": 10
    },
    {
      "epoch": 9.5367431640625e-09,
      "model_forward_time": 5.976618051528931,
      "step": 10
    },
    {
      "dynamics_loss_avg": 0.07815264947712422,
      "epoch": 9.5367431640625e-09,
      "step": 10
    },
    {
      "action_loss_avg": 0.09755489230155945,
      "epoch": 9.5367431640625e-09,
      "step": 10
    },
    {
      "epoch": 9.5367431640625e-09,
      "step": 10,
      "training_step_time": 17.504717111587524
    },
    {
      "epoch": 1.049041748046875e-08,
      "model_forward_time": 5.63182520866394,
      "step": 11
    },
    {
      "epoch": 1.049041748046875e-08,
      "step": 11,
      "training_step_time": 17.1242618560791
    },
    {
      "epoch": 1.1444091796875e-08,
      "model_forward_time": 5.602146148681641,
      "step": 12
    },
    {
      "epoch": 1.1444091796875e-08,
      "step": 12,
      "training_step_time": 17.048158645629883
    },
    {
      "epoch": 1.239776611328125e-08,
      "model_forward_time": 5.593994855880737,
      "step": 13
    },
    {
      "epoch": 1.239776611328125e-08,
      "step": 13,
      "training_step_time": 17.04441738128662
    },
    {
      "epoch": 1.33514404296875e-08,
      "model_forward_time": 5.594303131103516,
      "step": 14
    },
    {
      "epoch": 1.33514404296875e-08,
      "step": 14,
      "training_step_time": 17.018332719802856
    },
    {
      "epoch": 1.430511474609375e-08,
      "model_forward_time": 5.595613956451416,
      "step": 15
    },
    {
      "epoch": 1.430511474609375e-08,
      "step": 15,
      "training_step_time": 17.423380613327026
    },
    {
      "epoch": 1.52587890625e-08,
      "model_forward_time": 5.597571849822998,
      "step": 16
    },
    {
      "epoch": 1.52587890625e-08,
      "step": 16,
      "training_step_time": 17.05902123451233
    },
    {
      "epoch": 1.621246337890625e-08,
      "model_forward_time": 5.610524415969849,
      "step": 17
    },
    {
      "epoch": 1.621246337890625e-08,
      "step": 17,
      "training_step_time": 17.123729944229126
    },
    {
      "epoch": 1.71661376953125e-08,
      "model_forward_time": 5.6246349811553955,
      "step": 18
    },
    {
      "epoch": 1.71661376953125e-08,
      "step": 18,
      "training_step_time": 17.153099536895752
    },
    {
      "epoch": 1.811981201171875e-08,
      "model_forward_time": 5.6252851486206055,
      "step": 19
    },
    {
      "epoch": 1.811981201171875e-08,
      "step": 19,
      "training_step_time": 17.150875091552734
    },
    {
      "epoch": 1.9073486328125e-08,
      "grad_norm": 3.481780529022217,
      "learning_rate": 1e-05,
      "loss": 0.1605,
      "step": 20
    },
    {
      "epoch": 1.9073486328125e-08,
      "model_forward_time": 5.62933874130249,
      "step": 20
    },
    {
      "dynamics_loss_avg": 0.0899870179593563,
      "epoch": 1.9073486328125e-08,
      "step": 20
    },
    {
      "action_loss_avg": 0.06896097883582115,
      "epoch": 1.9073486328125e-08,
      "step": 20
    },
    {
      "epoch": 1.9073486328125e-08,
      "step": 20,
      "training_step_time": 17.151204586029053
    },
    {
      "epoch": 2.002716064453125e-08,
      "model_forward_time": 5.993120431900024,
      "step": 21
    },
    {
      "epoch": 2.002716064453125e-08,
      "step": 21,
      "training_step_time": 17.498188257217407
    },
    {
      "epoch": 2.09808349609375e-08,
      "model_forward_time": 5.6196815967559814,
      "step": 22
    },
    {
      "epoch": 2.09808349609375e-08,
      "step": 22,
      "training_step_time": 17.11999535560608
    },
    {
      "epoch": 2.193450927734375e-08,
      "model_forward_time": 5.624630928039551,
      "step": 23
    },
    {
      "epoch": 2.193450927734375e-08,
      "step": 23,
      "training_step_time": 17.102427005767822
    },
    {
      "epoch": 2.288818359375e-08,
      "model_forward_time": 5.597720146179199,
      "step": 24
    },
    {
      "epoch": 2.288818359375e-08,
      "step": 24,
      "training_step_time": 17.027849912643433
    },
    {
      "epoch": 2.384185791015625e-08,
      "model_forward_time": 5.568884611129761,
      "step": 25
    },
    {
      "epoch": 2.384185791015625e-08,
      "step": 25,
      "training_step_time": 16.981927394866943
    },
    {
      "epoch": 2.47955322265625e-08,
      "model_forward_time": 5.936374664306641,
      "step": 26
    },
    {
      "epoch": 2.47955322265625e-08,
      "step": 26,
      "training_step_time": 17.342462062835693
    },
    {
      "epoch": 2.574920654296875e-08,
      "model_forward_time": 5.570954084396362,
      "step": 27
    },
    {
      "epoch": 2.574920654296875e-08,
      "step": 27,
      "training_step_time": 17.006794452667236
    },
    {
      "epoch": 2.6702880859375e-08,
      "model_forward_time": 5.5771002769470215,
      "step": 28
    },
    {
      "epoch": 2.6702880859375e-08,
      "step": 28,
      "training_step_time": 16.989881992340088
    },
    {
      "epoch": 2.765655517578125e-08,
      "model_forward_time": 5.5828845500946045,
      "step": 29
    },
    {
      "epoch": 2.765655517578125e-08,
      "step": 29,
      "training_step_time": 17.00499391555786
    },
    {
      "epoch": 2.86102294921875e-08,
      "grad_norm": 2.25717830657959,
      "learning_rate": 1e-05,
      "loss": 0.1694,
      "step": 30
    },
    {
      "epoch": 2.86102294921875e-08,
      "model_forward_time": 5.587762355804443,
      "step": 30
    },
    {
      "dynamics_loss_avg": 0.09149161390960217,
      "epoch": 2.86102294921875e-08,
      "step": 30
    },
    {
      "action_loss_avg": 0.0944710586220026,
      "epoch": 2.86102294921875e-08,
      "step": 30
    },
    {
      "epoch": 2.86102294921875e-08,
      "step": 30,
      "training_step_time": 16.981801986694336
    },
    {
      "epoch": 2.956390380859375e-08,
      "model_forward_time": 5.587629795074463,
      "step": 31
    },
    {
      "epoch": 2.956390380859375e-08,
      "step": 31,
      "training_step_time": 17.37498116493225
    },
    {
      "epoch": 3.0517578125e-08,
      "model_forward_time": 5.600824356079102,
      "step": 32
    },
    {
      "epoch": 3.0517578125e-08,
      "step": 32,
      "training_step_time": 17.086120128631592
    },
    {
      "epoch": 3.147125244140625e-08,
      "model_forward_time": 5.62825345993042,
      "step": 33
    },
    {
      "epoch": 3.147125244140625e-08,
      "step": 33,
      "training_step_time": 17.13467049598694
    },
    {
      "epoch": 3.24249267578125e-08,
      "model_forward_time": 5.6253135204315186,
      "step": 34
    },
    {
      "epoch": 3.24249267578125e-08,
      "step": 34,
      "training_step_time": 17.110946893692017
    },
    {
      "epoch": 3.337860107421875e-08,
      "model_forward_time": 5.607084035873413,
      "step": 35
    },
    {
      "epoch": 3.337860107421875e-08,
      "step": 35,
      "training_step_time": 17.06308937072754
    },
    {
      "epoch": 3.4332275390625e-08,
      "model_forward_time": 5.5814454555511475,
      "step": 36
    },
    {
      "epoch": 3.4332275390625e-08,
      "step": 36,
      "training_step_time": 16.988475561141968
    },
    {
      "epoch": 3.528594970703125e-08,
      "model_forward_time": 5.940773010253906,
      "step": 37
    },
    {
      "epoch": 3.528594970703125e-08,
      "step": 37,
      "training_step_time": 17.34658455848694
    },
    {
      "epoch": 3.62396240234375e-08,
      "model_forward_time": 5.588025331497192,
      "step": 38
    },
    {
      "epoch": 3.62396240234375e-08,
      "step": 38,
      "training_step_time": 17.009358167648315
    },
    {
      "epoch": 3.719329833984375e-08,
      "model_forward_time": 5.58552360534668,
      "step": 39
    },
    {
      "epoch": 3.719329833984375e-08,
      "step": 39,
      "training_step_time": 17.022035598754883
    },
    {
      "epoch": 3.814697265625e-08,
      "grad_norm": 2.6665501594543457,
      "learning_rate": 1e-05,
      "loss": 0.1496,
      "step": 40
    },
    {
      "epoch": 3.814697265625e-08,
      "model_forward_time": 5.59173846244812,
      "step": 40
    },
    {
      "dynamics_loss_avg": 0.0773721432313323,
      "epoch": 3.814697265625e-08,
      "step": 40
    },
    {
      "action_loss_avg": 0.058738372987136245,
      "epoch": 3.814697265625e-08,
      "step": 40
    },
    {
      "epoch": 3.814697265625e-08,
      "step": 40,
      "training_step_time": 17.043782711029053
    },
    {
      "epoch": 3.910064697265625e-08,
      "model_forward_time": 5.6105334758758545,
      "step": 41
    },
    {
      "epoch": 3.910064697265625e-08,
      "step": 41,
      "training_step_time": 17.118997812271118
    },
    {
      "epoch": 4.00543212890625e-08,
      "model_forward_time": 5.995128870010376,
      "step": 42
    },
    {
      "epoch": 4.00543212890625e-08,
      "step": 42,
      "training_step_time": 17.51010274887085
    },
    {
      "epoch": 4.100799560546875e-08,
      "model_forward_time": 5.625436067581177,
      "step": 43
    },
    {
      "epoch": 4.100799560546875e-08,
      "step": 43,
      "training_step_time": 17.13625478744507
    },
    {
      "epoch": 4.1961669921875e-08,
      "model_forward_time": 5.617766857147217,
      "step": 44
    },
    {
      "epoch": 4.1961669921875e-08,
      "step": 44,
      "training_step_time": 17.121216535568237
    },
    {
      "epoch": 4.291534423828125e-08,
      "model_forward_time": 5.6325132846832275,
      "step": 45
    },
    {
      "epoch": 4.291534423828125e-08,
      "step": 45,
      "training_step_time": 17.112922191619873
    },
    {
      "epoch": 4.38690185546875e-08,
      "model_forward_time": 5.59778356552124,
      "step": 46
    },
    {
      "epoch": 4.38690185546875e-08,
      "step": 46,
      "training_step_time": 17.03479266166687
    },
    {
      "epoch": 4.482269287109375e-08,
      "model_forward_time": 5.570957660675049,
      "step": 47
    },
    {
      "epoch": 4.482269287109375e-08,
      "step": 47,
      "training_step_time": 16.995689630508423
    },
    {
      "epoch": 4.57763671875e-08,
      "model_forward_time": 5.936127662658691,
      "step": 48
    },
    {
      "epoch": 4.57763671875e-08,
      "step": 48,
      "training_step_time": 17.335299015045166
    },
    {
      "epoch": 4.673004150390625e-08,
      "model_forward_time": 5.568790912628174,
      "step": 49
    },
    {
      "epoch": 4.673004150390625e-08,
      "step": 49,
      "training_step_time": 16.981577157974243
    },
    {
      "epoch": 4.76837158203125e-08,
      "grad_norm": 2.6380159854888916,
      "learning_rate": 1e-05,
      "loss": 0.157,
      "step": 50
    },
    {
      "epoch": 4.76837158203125e-08,
      "model_forward_time": 5.570084095001221,
      "step": 50
    },
    {
      "dynamics_loss_avg": 0.08533320967108012,
      "epoch": 4.76837158203125e-08,
      "step": 50
    },
    {
      "action_loss_avg": 0.0712424511089921,
      "epoch": 4.76837158203125e-08,
      "step": 50
    },
    {
      "epoch": 4.76837158203125e-08,
      "step": 50,
      "training_step_time": 17.00242829322815
    },
    {
      "epoch": 4.863739013671875e-08,
      "model_forward_time": 5.568849802017212,
      "step": 51
    },
    {
      "epoch": 4.863739013671875e-08,
      "step": 51,
      "training_step_time": 16.995311498641968
    },
    {
      "epoch": 4.9591064453125e-08,
      "model_forward_time": 4.256653785705566,
      "step": 52
    },
    {
      "epoch": 4.9591064453125e-08,
      "step": 52,
      "training_step_time": 12.740020036697388
    },
    {
      "epoch": 5.054473876953125e-08,
      "model_forward_time": 5.951001405715942,
      "step": 53
    },
    {
      "epoch": 5.054473876953125e-08,
      "step": 53,
      "training_step_time": 17.37908935546875
    },
    {
      "epoch": 5.14984130859375e-08,
      "model_forward_time": 5.583484172821045,
      "step": 54
    },
    {
      "epoch": 5.14984130859375e-08,
      "step": 54,
      "training_step_time": 17.0379056930542
    },
    {
      "epoch": 5.245208740234375e-08,
      "model_forward_time": 5.597090482711792,
      "step": 55
    },
    {
      "epoch": 5.245208740234375e-08,
      "step": 55,
      "training_step_time": 17.05932331085205
    },
    {
      "epoch": 5.340576171875e-08,
      "model_forward_time": 5.600346326828003,
      "step": 56
    },
    {
      "epoch": 5.340576171875e-08,
      "step": 56,
      "training_step_time": 17.086033582687378
    },
    {
      "epoch": 5.435943603515625e-08,
      "model_forward_time": 5.624179124832153,
      "step": 57
    },
    {
      "epoch": 5.435943603515625e-08,
      "step": 57,
      "training_step_time": 17.116217851638794
    },
    {
      "epoch": 5.53131103515625e-08,
      "model_forward_time": 5.605840444564819,
      "step": 58
    },
    {
      "epoch": 5.53131103515625e-08,
      "step": 58,
      "training_step_time": 17.04880166053772
    },
    {
      "epoch": 5.626678466796875e-08,
      "model_forward_time": 5.947382688522339,
      "step": 59
    },
    {
      "epoch": 5.626678466796875e-08,
      "step": 59,
      "training_step_time": 17.3749520778656
    },
    {
      "epoch": 5.7220458984375e-08,
      "grad_norm": 0.9019680619239807,
      "learning_rate": 1e-05,
      "loss": 0.0949,
      "step": 60
    },
    {
      "epoch": 5.7220458984375e-08,
      "model_forward_time": 5.595783948898315,
      "step": 60
    },
    {
      "dynamics_loss_avg": 0.06131114698946476,
      "epoch": 5.7220458984375e-08,
      "step": 60
    },
    {
      "action_loss_avg": 0.03154488829895854,
      "epoch": 5.7220458984375e-08,
      "step": 60
    },
    {
      "epoch": 5.7220458984375e-08,
      "step": 60,
      "training_step_time": 17.0197491645813
    },
    {
      "epoch": 5.817413330078125e-08,
      "model_forward_time": 5.579700231552124,
      "step": 61
    },
    {
      "epoch": 5.817413330078125e-08,
      "step": 61,
      "training_step_time": 16.986732721328735
    },
    {
      "epoch": 5.91278076171875e-08,
      "model_forward_time": 5.583734035491943,
      "step": 62
    },
    {
      "epoch": 5.91278076171875e-08,
      "step": 62,
      "training_step_time": 17.022071838378906
    },
    {
      "epoch": 6.008148193359375e-08,
      "model_forward_time": 5.606449842453003,
      "step": 63
    },
    {
      "epoch": 6.008148193359375e-08,
      "step": 63,
      "training_step_time": 17.075413703918457
    },
    {
      "epoch": 6.103515625e-08,
      "model_forward_time": 5.9633190631866455,
      "step": 64
    },
    {
      "epoch": 6.103515625e-08,
      "step": 64,
      "training_step_time": 17.465396642684937
    },
    {
      "epoch": 6.198883056640625e-08,
      "model_forward_time": 5.6399829387664795,
      "step": 65
    },
    {
      "epoch": 6.198883056640625e-08,
      "step": 65,
      "training_step_time": 17.144006490707397
    },
    {
      "epoch": 6.29425048828125e-08,
      "model_forward_time": 5.6549577713012695,
      "step": 66
    },
    {
      "epoch": 6.29425048828125e-08,
      "step": 66,
      "training_step_time": 17.165140390396118
    },
    {
      "epoch": 6.389617919921875e-08,
      "model_forward_time": 5.61392068862915,
      "step": 67
    },
    {
      "epoch": 6.389617919921875e-08,
      "step": 67,
      "training_step_time": 17.110894441604614
    },
    {
      "epoch": 6.4849853515625e-08,
      "model_forward_time": 5.615699291229248,
      "step": 68
    },
    {
      "epoch": 6.4849853515625e-08,
      "step": 68,
      "training_step_time": 17.08616352081299
    },
    {
      "epoch": 6.580352783203126e-08,
      "model_forward_time": 5.587852954864502,
      "step": 69
    },
    {
      "epoch": 6.580352783203126e-08,
      "step": 69,
      "training_step_time": 17.022611141204834
    },
    {
      "epoch": 6.67572021484375e-08,
      "grad_norm": 2.922215700149536,
      "learning_rate": 1e-05,
      "loss": 0.1724,
      "step": 70
    },
    {
      "epoch": 6.67572021484375e-08,
      "model_forward_time": 5.952557563781738,
      "step": 70
    },
    {
      "dynamics_loss_avg": 0.08843596167862415,
      "epoch": 6.67572021484375e-08,
      "step": 70
    },
    {
      "action_loss_avg": 0.08556531704962253,
      "epoch": 6.67572021484375e-08,
      "step": 70
    },
    {
      "epoch": 6.67572021484375e-08,
      "step": 70,
      "training_step_time": 17.391374826431274
    },
    {
      "epoch": 6.771087646484374e-08,
      "model_forward_time": 5.585414171218872,
      "step": 71
    },
    {
      "epoch": 6.771087646484374e-08,
      "step": 71,
      "training_step_time": 16.9951229095459
    },
    {
      "epoch": 6.866455078125e-08,
      "model_forward_time": 5.591109752655029,
      "step": 72
    },
    {
      "epoch": 6.866455078125e-08,
      "step": 72,
      "training_step_time": 17.008846282958984
    },
    {
      "epoch": 6.961822509765625e-08,
      "model_forward_time": 5.589650869369507,
      "step": 73
    },
    {
      "epoch": 6.961822509765625e-08,
      "step": 73,
      "training_step_time": 17.027228355407715
    },
    {
      "epoch": 7.05718994140625e-08,
      "model_forward_time": 5.594601631164551,
      "step": 74
    },
    {
      "epoch": 7.05718994140625e-08,
      "step": 74,
      "training_step_time": 17.058642148971558
    },
    {
      "epoch": 7.152557373046875e-08,
      "model_forward_time": 5.613351345062256,
      "step": 75
    },
    {
      "epoch": 7.152557373046875e-08,
      "step": 75,
      "training_step_time": 17.462204217910767
    },
    {
      "epoch": 7.2479248046875e-08,
      "model_forward_time": 5.620312213897705,
      "step": 76
    },
    {
      "epoch": 7.2479248046875e-08,
      "step": 76,
      "training_step_time": 17.110782384872437
    },
    {
      "epoch": 7.343292236328125e-08,
      "model_forward_time": 5.618015766143799,
      "step": 77
    },
    {
      "epoch": 7.343292236328125e-08,
      "step": 77,
      "training_step_time": 17.11786675453186
    },
    {
      "epoch": 7.43865966796875e-08,
      "model_forward_time": 5.621709585189819,
      "step": 78
    },
    {
      "epoch": 7.43865966796875e-08,
      "step": 78,
      "training_step_time": 17.123344898223877
    },
    {
      "epoch": 7.534027099609375e-08,
      "model_forward_time": 5.621900796890259,
      "step": 79
    },
    {
      "epoch": 7.534027099609375e-08,
      "step": 79,
      "training_step_time": 17.128108739852905
    },
    {
      "epoch": 7.62939453125e-08,
      "grad_norm": 1.8280235528945923,
      "learning_rate": 1e-05,
      "loss": 0.1513,
      "step": 80
    },
    {
      "epoch": 7.62939453125e-08,
      "model_forward_time": 5.619303941726685,
      "step": 80
    },
    {
      "dynamics_loss_avg": 0.08864242509007454,
      "epoch": 7.62939453125e-08,
      "step": 80
    },
    {
      "action_loss_avg": 0.06677919328212738,
      "epoch": 7.62939453125e-08,
      "step": 80
    },
    {
      "epoch": 7.62939453125e-08,
      "step": 80,
      "training_step_time": 17.12136197090149
    },
    {
      "epoch": 7.724761962890625e-08,
      "model_forward_time": 5.998715877532959,
      "step": 81
    },
    {
      "epoch": 7.724761962890625e-08,
      "step": 81,
      "training_step_time": 17.506337881088257
    },
    {
      "epoch": 7.82012939453125e-08,
      "model_forward_time": 5.622274398803711,
      "step": 82
    },
    {
      "epoch": 7.82012939453125e-08,
      "step": 82,
      "training_step_time": 17.11471438407898
    },
    {
      "epoch": 7.915496826171875e-08,
      "model_forward_time": 5.618281841278076,
      "step": 83
    },
    {
      "epoch": 7.915496826171875e-08,
      "step": 83,
      "training_step_time": 17.111189603805542
    },
    {
      "epoch": 8.0108642578125e-08,
      "model_forward_time": 5.6336915493011475,
      "step": 84
    },
    {
      "epoch": 8.0108642578125e-08,
      "step": 84,
      "training_step_time": 17.130810737609863
    },
    {
      "epoch": 8.106231689453126e-08,
      "model_forward_time": 5.618281364440918,
      "step": 85
    },
    {
      "epoch": 8.106231689453126e-08,
      "step": 85,
      "training_step_time": 17.099099159240723
    },
    {
      "epoch": 8.20159912109375e-08,
      "model_forward_time": 5.93895149230957,
      "step": 86
    },
    {
      "epoch": 8.20159912109375e-08,
      "step": 86,
      "training_step_time": 17.334192991256714
    },
    {
      "epoch": 8.296966552734374e-08,
      "model_forward_time": 5.583041429519653,
      "step": 87
    },
    {
      "epoch": 8.296966552734374e-08,
      "step": 87,
      "training_step_time": 17.0082528591156
    },
    {
      "epoch": 8.392333984375e-08,
      "model_forward_time": 5.583116054534912,
      "step": 88
    },
    {
      "epoch": 8.392333984375e-08,
      "step": 88,
      "training_step_time": 17.031860828399658
    },
    {
      "epoch": 8.487701416015625e-08,
      "model_forward_time": 5.578646898269653,
      "step": 89
    },
    {
      "epoch": 8.487701416015625e-08,
      "step": 89,
      "training_step_time": 16.994219541549683
    },
    {
      "epoch": 8.58306884765625e-08,
      "grad_norm": 2.061774253845215,
      "learning_rate": 1e-05,
      "loss": 0.1255,
      "step": 90
    },
    {
      "epoch": 8.58306884765625e-08,
      "model_forward_time": 5.579021692276001,
      "step": 90
    },
    {
      "dynamics_loss_avg": 0.07048528138548135,
      "epoch": 8.58306884765625e-08,
      "step": 90
    },
    {
      "action_loss_avg": 0.04858455546200276,
      "epoch": 8.58306884765625e-08,
      "step": 90
    },
    {
      "epoch": 8.58306884765625e-08,
      "step": 90,
      "training_step_time": 16.999523878097534
    },
    {
      "epoch": 8.678436279296875e-08,
      "model_forward_time": 5.57997465133667,
      "step": 91
    },
    {
      "epoch": 8.678436279296875e-08,
      "step": 91,
      "training_step_time": 16.992339849472046
    },
    {
      "epoch": 8.7738037109375e-08,
      "model_forward_time": 5.934932708740234,
      "step": 92
    },
    {
      "epoch": 8.7738037109375e-08,
      "step": 92,
      "training_step_time": 17.371366262435913
    },
    {
      "epoch": 8.869171142578125e-08,
      "model_forward_time": 5.60008692741394,
      "step": 93
    },
    {
      "epoch": 8.869171142578125e-08,
      "step": 93,
      "training_step_time": 17.078848123550415
    },
    {
      "epoch": 8.96453857421875e-08,
      "model_forward_time": 5.615258455276489,
      "step": 94
    },
    {
      "epoch": 8.96453857421875e-08,
      "step": 94,
      "training_step_time": 17.142696619033813
    },
    {
      "epoch": 9.059906005859375e-08,
      "model_forward_time": 5.626536846160889,
      "step": 95
    },
    {
      "epoch": 9.059906005859375e-08,
      "step": 95,
      "training_step_time": 17.1572368144989
    },
    {
      "epoch": 9.1552734375e-08,
      "model_forward_time": 5.643907308578491,
      "step": 96
    },
    {
      "epoch": 9.1552734375e-08,
      "step": 96,
      "training_step_time": 17.1584050655365
    },
    {
      "epoch": 9.250640869140625e-08,
      "model_forward_time": 5.96495795249939,
      "step": 97
    },
    {
      "epoch": 9.250640869140625e-08,
      "step": 97,
      "training_step_time": 17.46081018447876
    },
    {
      "epoch": 9.34600830078125e-08,
      "model_forward_time": 5.613002777099609,
      "step": 98
    },
    {
      "epoch": 9.34600830078125e-08,
      "step": 98,
      "training_step_time": 17.10563325881958
    },
    {
      "epoch": 9.441375732421875e-08,
      "model_forward_time": 5.6186134815216064,
      "step": 99
    },
    {
      "epoch": 9.441375732421875e-08,
      "step": 99,
      "training_step_time": 17.11480164527893
    },
    {
      "epoch": 9.5367431640625e-08,
      "grad_norm": 2.5843684673309326,
      "learning_rate": 1e-05,
      "loss": 0.1189,
      "step": 100
    },
    {
      "epoch": 9.5367431640625e-08,
      "model_forward_time": 5.615823268890381,
      "step": 100
    },
    {
      "dynamics_loss_avg": 0.07388069555163383,
      "epoch": 9.5367431640625e-08,
      "step": 100
    },
    {
      "action_loss_avg": 0.0515537703409791,
      "epoch": 9.5367431640625e-08,
      "step": 100
    },
    {
      "epoch": 9.5367431640625e-08,
      "step": 100,
      "training_step_time": 17.097065925598145
    },
    {
      "epoch": 9.632110595703126e-08,
      "model_forward_time": 5.603729963302612,
      "step": 101
    },
    {
      "epoch": 9.632110595703126e-08,
      "step": 101,
      "training_step_time": 17.092025995254517
    },
    {
      "epoch": 9.72747802734375e-08,
      "model_forward_time": 5.6057775020599365,
      "step": 102
    },
    {
      "epoch": 9.72747802734375e-08,
      "step": 102,
      "training_step_time": 17.44489550590515
    },
    {
      "epoch": 9.822845458984374e-08,
      "model_forward_time": 5.625262260437012,
      "step": 103
    },
    {
      "epoch": 9.822845458984374e-08,
      "step": 103,
      "training_step_time": 17.130916118621826
    },
    {
      "epoch": 9.918212890625e-08,
      "model_forward_time": 5.618953704833984,
      "step": 104
    },
    {
      "epoch": 9.918212890625e-08,
      "step": 104,
      "training_step_time": 17.089444875717163
    },
    {
      "epoch": 1.0013580322265625e-07,
      "model_forward_time": 5.599619150161743,
      "step": 105
    },
    {
      "epoch": 1.0013580322265625e-07,
      "step": 105,
      "training_step_time": 17.01015305519104
    },
    {
      "epoch": 1.010894775390625e-07,
      "model_forward_time": 5.579010009765625,
      "step": 106
    },
    {
      "epoch": 1.010894775390625e-07,
      "step": 106,
      "training_step_time": 16.985439777374268
    },
    {
      "epoch": 1.0204315185546875e-07,
      "model_forward_time": 5.562278747558594,
      "step": 107
    },
    {
      "epoch": 1.0204315185546875e-07,
      "step": 107,
      "training_step_time": 17.00095844268799
    },
    {
      "epoch": 1.02996826171875e-07,
      "model_forward_time": 5.927168607711792,
      "step": 108
    },
    {
      "epoch": 1.02996826171875e-07,
      "step": 108,
      "training_step_time": 17.35067105293274
    },
    {
      "epoch": 1.0395050048828125e-07,
      "model_forward_time": 5.573602199554443,
      "step": 109
    },
    {
      "epoch": 1.0395050048828125e-07,
      "step": 109,
      "training_step_time": 16.991777896881104
    },
    {
      "epoch": 1.049041748046875e-07,
      "grad_norm": 2.9782183170318604,
      "learning_rate": 1e-05,
      "loss": 0.1169,
      "step": 110
    },
    {
      "epoch": 1.049041748046875e-07,
      "model_forward_time": 5.575784683227539,
      "step": 110
    },
    {
      "dynamics_loss_avg": 0.07066806815564633,
      "epoch": 1.049041748046875e-07,
      "step": 110
    },
    {
      "action_loss_avg": 0.04612231701612472,
      "epoch": 1.049041748046875e-07,
      "step": 110
    },
    {
      "epoch": 1.049041748046875e-07,
      "step": 110,
      "training_step_time": 16.98777437210083
    },
    {
      "epoch": 1.0585784912109375e-07,
      "model_forward_time": 5.574217796325684,
      "step": 111
    },
    {
      "epoch": 1.0585784912109375e-07,
      "step": 111,
      "training_step_time": 17.01040554046631
    },
    {
      "epoch": 1.068115234375e-07,
      "model_forward_time": 5.584629058837891,
      "step": 112
    },
    {
      "epoch": 1.068115234375e-07,
      "step": 112,
      "training_step_time": 17.004778385162354
    },
    {
      "epoch": 1.0776519775390625e-07,
      "model_forward_time": 5.9756529331207275,
      "step": 113
    },
    {
      "epoch": 1.0776519775390625e-07,
      "step": 113,
      "training_step_time": 17.438083171844482
    },
    {
      "epoch": 1.087188720703125e-07,
      "model_forward_time": 5.617786884307861,
      "step": 114
    },
    {
      "epoch": 1.087188720703125e-07,
      "step": 114,
      "training_step_time": 17.114013671875
    },
    {
      "epoch": 1.0967254638671875e-07,
      "model_forward_time": 5.634376049041748,
      "step": 115
    },
    {
      "epoch": 1.0967254638671875e-07,
      "step": 115,
      "training_step_time": 17.16213846206665
    },
    {
      "epoch": 1.10626220703125e-07,
      "model_forward_time": 5.632700681686401,
      "step": 116
    },
    {
      "epoch": 1.10626220703125e-07,
      "step": 116,
      "training_step_time": 17.12896728515625
    },
    {
      "epoch": 1.1157989501953126e-07,
      "model_forward_time": 5.600438356399536,
      "step": 117
    },
    {
      "epoch": 1.1157989501953126e-07,
      "step": 117,
      "training_step_time": 17.055405139923096
    },
    {
      "epoch": 1.125335693359375e-07,
      "model_forward_time": 5.5780980587005615,
      "step": 118
    },
    {
      "epoch": 1.125335693359375e-07,
      "step": 118,
      "training_step_time": 17.000640630722046
    },
    {
      "epoch": 1.1348724365234374e-07,
      "model_forward_time": 5.929272413253784,
      "step": 119
    },
    {
      "epoch": 1.1348724365234374e-07,
      "step": 119,
      "training_step_time": 17.374207735061646
    },
    {
      "epoch": 1.1444091796875e-07,
      "grad_norm": 2.0271353721618652,
      "learning_rate": 1e-05,
      "loss": 0.1214,
      "step": 120
    },
    {
      "epoch": 1.1444091796875e-07,
      "model_forward_time": 5.589640140533447,
      "step": 120
    },
    {
      "dynamics_loss_avg": 0.06617306768894196,
      "epoch": 1.1444091796875e-07,
      "step": 120
    },
    {
      "action_loss_avg": 0.04593704603612423,
      "epoch": 1.1444091796875e-07,
      "step": 120
    },
    {
      "epoch": 1.1444091796875e-07,
      "step": 120,
      "training_step_time": 17.01862335205078
    },
    {
      "epoch": 1.1539459228515625e-07,
      "model_forward_time": 5.604982614517212,
      "step": 121
    },
    {
      "epoch": 1.1539459228515625e-07,
      "step": 121,
      "training_step_time": 17.073968172073364
    },
    {
      "epoch": 1.163482666015625e-07,
      "model_forward_time": 5.601877212524414,
      "step": 122
    },
    {
      "epoch": 1.163482666015625e-07,
      "step": 122,
      "training_step_time": 17.07988214492798
    },
    {
      "epoch": 1.1730194091796875e-07,
      "model_forward_time": 5.619926452636719,
      "step": 123
    },
    {
      "epoch": 1.1730194091796875e-07,
      "step": 123,
      "training_step_time": 17.12134599685669
    },
    {
      "epoch": 1.18255615234375e-07,
      "model_forward_time": 5.97954535484314,
      "step": 124
    },
    {
      "epoch": 1.18255615234375e-07,
      "step": 124,
      "training_step_time": 17.47329068183899
    },
    {
      "epoch": 1.1920928955078125e-07,
      "model_forward_time": 5.617776393890381,
      "step": 125
    },
    {
      "epoch": 1.1920928955078125e-07,
      "step": 125,
      "training_step_time": 17.11531686782837
    },
    {
      "epoch": 1.201629638671875e-07,
      "model_forward_time": 5.626497745513916,
      "step": 126
    },
    {
      "epoch": 1.201629638671875e-07,
      "step": 126,
      "training_step_time": 17.1122407913208
    },
    {
      "epoch": 1.2111663818359375e-07,
      "model_forward_time": 5.610496282577515,
      "step": 127
    },
    {
      "epoch": 1.2111663818359375e-07,
      "step": 127,
      "training_step_time": 17.090770959854126
    },
    {
      "epoch": 1.220703125e-07,
      "model_forward_time": 5.6107728481292725,
      "step": 128
    },
    {
      "epoch": 1.220703125e-07,
      "step": 128,
      "training_step_time": 17.086337089538574
    },
    {
      "epoch": 1.2302398681640625e-07,
      "model_forward_time": 5.605630159378052,
      "step": 129
    },
    {
      "epoch": 1.2302398681640625e-07,
      "step": 129,
      "training_step_time": 17.08961009979248
    },
    {
      "epoch": 1.239776611328125e-07,
      "grad_norm": 0.3489076495170593,
      "learning_rate": 1e-05,
      "loss": 0.0985,
      "step": 130
    },
    {
      "epoch": 1.239776611328125e-07,
      "model_forward_time": 5.969388246536255,
      "step": 130
    },
    {
      "dynamics_loss_avg": 0.0657358741387725,
      "epoch": 1.239776611328125e-07,
      "step": 130
    },
    {
      "action_loss_avg": 0.038248634757474066,
      "epoch": 1.239776611328125e-07,
      "step": 130
    },
    {
      "epoch": 1.239776611328125e-07,
      "step": 130,
      "training_step_time": 17.430496215820312
    },
    {
      "epoch": 1.2493133544921876e-07,
      "model_forward_time": 5.601787805557251,
      "step": 131
    },
    {
      "epoch": 1.2493133544921876e-07,
      "step": 131,
      "training_step_time": 17.071939706802368
    },
    {
      "epoch": 1.25885009765625e-07,
      "model_forward_time": 5.5945258140563965,
      "step": 132
    },
    {
      "epoch": 1.25885009765625e-07,
      "step": 132,
      "training_step_time": 17.05671977996826
    },
    {
      "epoch": 1.2683868408203126e-07,
      "model_forward_time": 5.606276512145996,
      "step": 133
    },
    {
      "epoch": 1.2683868408203126e-07,
      "step": 133,
      "training_step_time": 17.0854594707489
    },
    {
      "epoch": 1.277923583984375e-07,
      "model_forward_time": 5.612694501876831,
      "step": 134
    },
    {
      "epoch": 1.277923583984375e-07,
      "step": 134,
      "training_step_time": 17.103304862976074
    },
    {
      "epoch": 1.2874603271484376e-07,
      "model_forward_time": 6.002877950668335,
      "step": 135
    },
    {
      "epoch": 1.2874603271484376e-07,
      "step": 135,
      "training_step_time": 17.511454343795776
    },
    {
      "epoch": 1.2969970703125e-07,
      "model_forward_time": 5.629975318908691,
      "step": 136
    },
    {
      "epoch": 1.2969970703125e-07,
      "step": 136,
      "training_step_time": 17.121077060699463
    },
    {
      "epoch": 1.3065338134765626e-07,
      "model_forward_time": 5.596906661987305,
      "step": 137
    },
    {
      "epoch": 1.3065338134765626e-07,
      "step": 137,
      "training_step_time": 17.020152807235718
    },
    {
      "epoch": 1.316070556640625e-07,
      "model_forward_time": 5.581537961959839,
      "step": 138
    },
    {
      "epoch": 1.316070556640625e-07,
      "step": 138,
      "training_step_time": 17.01257061958313
    },
    {
      "epoch": 1.3256072998046876e-07,
      "model_forward_time": 5.583458185195923,
      "step": 139
    },
    {
      "epoch": 1.3256072998046876e-07,
      "step": 139,
      "training_step_time": 17.008997201919556
    },
    {
      "epoch": 1.33514404296875e-07,
      "grad_norm": 2.412825107574463,
      "learning_rate": 1e-05,
      "loss": 0.1002,
      "step": 140
    },
    {
      "epoch": 1.33514404296875e-07,
      "model_forward_time": 5.5817272663116455,
      "step": 140
    },
    {
      "dynamics_loss_avg": 0.06666639167815447,
      "epoch": 1.33514404296875e-07,
      "step": 140
    },
    {
      "action_loss_avg": 0.04626455837860703,
      "epoch": 1.33514404296875e-07,
      "step": 140
    },
    {
      "epoch": 1.33514404296875e-07,
      "step": 140,
      "training_step_time": 16.9952871799469
    },
    {
      "epoch": 1.3446807861328124e-07,
      "model_forward_time": 5.9408087730407715,
      "step": 141
    },
    {
      "epoch": 1.3446807861328124e-07,
      "step": 141,
      "training_step_time": 17.379520416259766
    },
    {
      "epoch": 1.354217529296875e-07,
      "model_forward_time": 5.599747896194458,
      "step": 142
    },
    {
      "epoch": 1.354217529296875e-07,
      "step": 142,
      "training_step_time": 17.039044618606567
    },
    {
      "epoch": 1.3637542724609374e-07,
      "model_forward_time": 5.596493721008301,
      "step": 143
    },
    {
      "epoch": 1.3637542724609374e-07,
      "step": 143,
      "training_step_time": 17.057661771774292
    },
    {
      "epoch": 1.373291015625e-07,
      "model_forward_time": 5.597535610198975,
      "step": 144
    },
    {
      "epoch": 1.373291015625e-07,
      "step": 144,
      "training_step_time": 17.08879780769348
    },
    {
      "epoch": 1.3828277587890624e-07,
      "model_forward_time": 5.6166791915893555,
      "step": 145
    },
    {
      "epoch": 1.3828277587890624e-07,
      "step": 145,
      "training_step_time": 17.13599181175232
    },
    {
      "epoch": 1.392364501953125e-07,
      "model_forward_time": 6.000133752822876,
      "step": 146
    },
    {
      "epoch": 1.392364501953125e-07,
      "step": 146,
      "training_step_time": 17.508206367492676
    },
    {
      "epoch": 1.4019012451171874e-07,
      "model_forward_time": 5.623027086257935,
      "step": 147
    },
    {
      "epoch": 1.4019012451171874e-07,
      "step": 147,
      "training_step_time": 17.117671728134155
    },
    {
      "epoch": 1.41143798828125e-07,
      "model_forward_time": 5.603341102600098,
      "step": 148
    },
    {
      "epoch": 1.41143798828125e-07,
      "step": 148,
      "training_step_time": 17.04575276374817
    },
    {
      "epoch": 1.4209747314453124e-07,
      "model_forward_time": 5.5943968296051025,
      "step": 149
    },
    {
      "epoch": 1.4209747314453124e-07,
      "step": 149,
      "training_step_time": 17.042855978012085
    },
    {
      "epoch": 1.430511474609375e-07,
      "grad_norm": 2.5831706523895264,
      "learning_rate": 1e-05,
      "loss": 0.1658,
      "step": 150
    },
    {
      "epoch": 1.430511474609375e-07,
      "model_forward_time": 5.592408180236816,
      "step": 150
    },
    {
      "dynamics_loss_avg": 0.08673129864037037,
      "epoch": 1.430511474609375e-07,
      "step": 150
    },
    {
      "action_loss_avg": 0.06780891586095095,
      "epoch": 1.430511474609375e-07,
      "step": 150
    },
    {
      "epoch": 1.430511474609375e-07,
      "step": 150,
      "training_step_time": 17.01962447166443
    },
    {
      "epoch": 1.4400482177734375e-07,
      "model_forward_time": 5.942994594573975,
      "step": 151
    },
    {
      "epoch": 1.4400482177734375e-07,
      "step": 151,
      "training_step_time": 17.392841815948486
    },
    {
      "epoch": 1.4495849609375e-07,
      "model_forward_time": 5.594034910202026,
      "step": 152
    },
    {
      "epoch": 1.4495849609375e-07,
      "step": 152,
      "training_step_time": 17.050387382507324
    },
    {
      "epoch": 1.4591217041015625e-07,
      "model_forward_time": 5.603398561477661,
      "step": 153
    },
    {
      "epoch": 1.4591217041015625e-07,
      "step": 153,
      "training_step_time": 17.085932970046997
    },
    {
      "epoch": 1.468658447265625e-07,
      "model_forward_time": 5.608724355697632,
      "step": 154
    },
    {
      "epoch": 1.468658447265625e-07,
      "step": 154,
      "training_step_time": 17.112812757492065
    },
    {
      "epoch": 1.4781951904296875e-07,
      "model_forward_time": 5.613775968551636,
      "step": 155
    },
    {
      "epoch": 1.4781951904296875e-07,
      "step": 155,
      "training_step_time": 17.11452078819275
    },
    {
      "epoch": 1.48773193359375e-07,
      "model_forward_time": 5.617646217346191,
      "step": 156
    },
    {
      "epoch": 1.48773193359375e-07,
      "step": 156,
      "training_step_time": 17.105773448944092
    },
    {
      "epoch": 1.4972686767578125e-07,
      "model_forward_time": 5.976882696151733,
      "step": 157
    },
    {
      "epoch": 1.4972686767578125e-07,
      "step": 157,
      "training_step_time": 17.437976360321045
    },
    {
      "epoch": 1.506805419921875e-07,
      "model_forward_time": 5.593473196029663,
      "step": 158
    },
    {
      "epoch": 1.506805419921875e-07,
      "step": 158,
      "training_step_time": 17.02190852165222
    },
    {
      "epoch": 1.5163421630859375e-07,
      "model_forward_time": 5.5942065715789795,
      "step": 159
    },
    {
      "epoch": 1.5163421630859375e-07,
      "step": 159,
      "training_step_time": 17.02304434776306
    },
    {
      "epoch": 1.52587890625e-07,
      "grad_norm": 1.8997080326080322,
      "learning_rate": 1e-05,
      "loss": 0.1459,
      "step": 160
    },
    {
      "epoch": 1.52587890625e-07,
      "model_forward_time": 4.269421339035034,
      "step": 160
    },
    {
      "dynamics_loss_avg": 0.08627236112952233,
      "epoch": 1.52587890625e-07,
      "step": 160
    },
    {
      "action_loss_avg": 0.061650623381137845,
      "epoch": 1.52587890625e-07,
      "step": 160
    },
    {
      "epoch": 1.52587890625e-07,
      "step": 160,
      "training_step_time": 12.763325691223145
    },
    {
      "epoch": 1.5354156494140625e-07,
      "model_forward_time": 5.587313175201416,
      "step": 161
    },
    {
      "epoch": 1.5354156494140625e-07,
      "step": 161,
      "training_step_time": 17.01651430130005
    },
    {
      "epoch": 1.544952392578125e-07,
      "model_forward_time": 5.598422288894653,
      "step": 162
    },
    {
      "epoch": 1.544952392578125e-07,
      "step": 162,
      "training_step_time": 17.431644439697266
    },
    {
      "epoch": 1.5544891357421876e-07,
      "model_forward_time": 5.59174370765686,
      "step": 163
    },
    {
      "epoch": 1.5544891357421876e-07,
      "step": 163,
      "training_step_time": 17.036934852600098
    },
    {
      "epoch": 1.56402587890625e-07,
      "model_forward_time": 5.596264362335205,
      "step": 164
    },
    {
      "epoch": 1.56402587890625e-07,
      "step": 164,
      "training_step_time": 17.074551820755005
    },
    {
      "epoch": 1.5735626220703126e-07,
      "model_forward_time": 5.659414052963257,
      "step": 165
    },
    {
      "epoch": 1.5735626220703126e-07,
      "step": 165,
      "training_step_time": 17.188085556030273
    },
    {
      "epoch": 1.583099365234375e-07,
      "model_forward_time": 5.632307052612305,
      "step": 166
    },
    {
      "epoch": 1.583099365234375e-07,
      "step": 166,
      "training_step_time": 17.135499954223633
    },
    {
      "epoch": 1.5926361083984376e-07,
      "model_forward_time": 5.619110584259033,
      "step": 167
    },
    {
      "epoch": 1.5926361083984376e-07,
      "step": 167,
      "training_step_time": 17.11414074897766
    },
    {
      "epoch": 1.6021728515625e-07,
      "model_forward_time": 5.982302904129028,
      "step": 168
    },
    {
      "epoch": 1.6021728515625e-07,
      "step": 168,
      "training_step_time": 17.465250730514526
    },
    {
      "epoch": 1.6117095947265626e-07,
      "model_forward_time": 5.6178834438323975,
      "step": 169
    },
    {
      "epoch": 1.6117095947265626e-07,
      "step": 169,
      "training_step_time": 17.11181116104126
    },
    {
      "epoch": 1.621246337890625e-07,
      "grad_norm": 2.2680089473724365,
      "learning_rate": 1e-05,
      "loss": 0.1282,
      "step": 170
    },
    {
      "epoch": 1.621246337890625e-07,
      "model_forward_time": 5.611957311630249,
      "step": 170
    },
    {
      "dynamics_loss_avg": 0.07196397706866264,
      "epoch": 1.621246337890625e-07,
      "step": 170
    },
    {
      "action_loss_avg": 0.04997931588441133,
      "epoch": 1.621246337890625e-07,
      "step": 170
    },
    {
      "epoch": 1.621246337890625e-07,
      "step": 170,
      "training_step_time": 17.11851716041565
    },
    {
      "epoch": 1.6307830810546876e-07,
      "model_forward_time": 5.612863540649414,
      "step": 171
    },
    {
      "epoch": 1.6307830810546876e-07,
      "step": 171,
      "training_step_time": 17.108092308044434
    },
    {
      "epoch": 1.64031982421875e-07,
      "model_forward_time": 5.618607759475708,
      "step": 172
    },
    {
      "epoch": 1.64031982421875e-07,
      "step": 172,
      "training_step_time": 17.12138605117798
    },
    {
      "epoch": 1.6498565673828124e-07,
      "model_forward_time": 5.967632055282593,
      "step": 173
    },
    {
      "epoch": 1.6498565673828124e-07,
      "step": 173,
      "training_step_time": 17.451032638549805
    },
    {
      "epoch": 1.659393310546875e-07,
      "model_forward_time": 5.625074625015259,
      "step": 174
    },
    {
      "epoch": 1.659393310546875e-07,
      "step": 174,
      "training_step_time": 17.124858617782593
    },
    {
      "epoch": 1.6689300537109374e-07,
      "model_forward_time": 5.601170778274536,
      "step": 175
    },
    {
      "epoch": 1.6689300537109374e-07,
      "step": 175,
      "training_step_time": 17.107131958007812
    },
    {
      "epoch": 1.678466796875e-07,
      "model_forward_time": 5.612768650054932,
      "step": 176
    },
    {
      "epoch": 1.678466796875e-07,
      "step": 176,
      "training_step_time": 17.14441680908203
    },
    {
      "epoch": 1.6880035400390624e-07,
      "model_forward_time": 5.618987321853638,
      "step": 177
    },
    {
      "epoch": 1.6880035400390624e-07,
      "step": 177,
      "training_step_time": 17.128589630126953
    },
    {
      "epoch": 1.697540283203125e-07,
      "model_forward_time": 5.619208335876465,
      "step": 178
    },
    {
      "epoch": 1.697540283203125e-07,
      "step": 178,
      "training_step_time": 17.106338024139404
    },
    {
      "epoch": 1.7070770263671874e-07,
      "model_forward_time": 5.978659629821777,
      "step": 179
    },
    {
      "epoch": 1.7070770263671874e-07,
      "step": 179,
      "training_step_time": 17.470324993133545
    },
    {
      "epoch": 1.71661376953125e-07,
      "grad_norm": 1.3197126388549805,
      "learning_rate": 1e-05,
      "loss": 0.0965,
      "step": 180
    },
    {
      "epoch": 1.71661376953125e-07,
      "model_forward_time": 5.616020917892456,
      "step": 180
    },
    {
      "dynamics_loss_avg": 0.06575686018913984,
      "epoch": 1.71661376953125e-07,
      "step": 180
    },
    {
      "action_loss_avg": 0.0401048514759168,
      "epoch": 1.71661376953125e-07,
      "step": 180
    },
    {
      "epoch": 1.71661376953125e-07,
      "step": 180,
      "training_step_time": 17.100203275680542
    },
    {
      "epoch": 1.7261505126953124e-07,
      "model_forward_time": 5.603033065795898,
      "step": 181
    },
    {
      "epoch": 1.7261505126953124e-07,
      "step": 181,
      "training_step_time": 17.091928720474243
    },
    {
      "epoch": 1.735687255859375e-07,
      "model_forward_time": 5.605882406234741,
      "step": 182
    },
    {
      "epoch": 1.735687255859375e-07,
      "step": 182,
      "training_step_time": 17.09157419204712
    },
    {
      "epoch": 1.7452239990234375e-07,
      "model_forward_time": 5.605503082275391,
      "step": 183
    },
    {
      "epoch": 1.7452239990234375e-07,
      "step": 183,
      "training_step_time": 17.076306581497192
    },
    {
      "epoch": 1.7547607421875e-07,
      "model_forward_time": 5.967817068099976,
      "step": 184
    },
    {
      "epoch": 1.7547607421875e-07,
      "step": 184,
      "training_step_time": 17.446988344192505
    },
    {
      "epoch": 1.7642974853515625e-07,
      "model_forward_time": 5.600744962692261,
      "step": 185
    },
    {
      "epoch": 1.7642974853515625e-07,
      "step": 185,
      "training_step_time": 17.087635278701782
    },
    {
      "epoch": 1.773834228515625e-07,
      "model_forward_time": 5.598136901855469,
      "step": 186
    },
    {
      "epoch": 1.773834228515625e-07,
      "step": 186,
      "training_step_time": 17.069210529327393
    },
    {
      "epoch": 1.7833709716796875e-07,
      "model_forward_time": 5.6045310497283936,
      "step": 187
    },
    {
      "epoch": 1.7833709716796875e-07,
      "step": 187,
      "training_step_time": 17.080683946609497
    },
    {
      "epoch": 1.79290771484375e-07,
      "model_forward_time": 5.603019952774048,
      "step": 188
    },
    {
      "epoch": 1.79290771484375e-07,
      "step": 188,
      "training_step_time": 17.06918501853943
    },
    {
      "epoch": 1.8024444580078125e-07,
      "model_forward_time": 5.595450401306152,
      "step": 189
    },
    {
      "epoch": 1.8024444580078125e-07,
      "step": 189,
      "training_step_time": 17.419142246246338
    },
    {
      "epoch": 1.811981201171875e-07,
      "grad_norm": 0.44932255148887634,
      "learning_rate": 1e-05,
      "loss": 0.1027,
      "step": 190
    },
    {
      "epoch": 1.811981201171875e-07,
      "model_forward_time": 5.591164827346802,
      "step": 190
    },
    {
      "dynamics_loss_avg": 0.05767967589199543,
      "epoch": 1.811981201171875e-07,
      "step": 190
    },
    {
      "action_loss_avg": 0.03447057381272316,
      "epoch": 1.811981201171875e-07,
      "step": 190
    },
    {
      "epoch": 1.811981201171875e-07,
      "step": 190,
      "training_step_time": 17.02064323425293
    },
    {
      "epoch": 1.8215179443359375e-07,
      "model_forward_time": 5.590381383895874,
      "step": 191
    },
    {
      "epoch": 1.8215179443359375e-07,
      "step": 191,
      "training_step_time": 17.03775930404663
    },
    {
      "epoch": 1.8310546875e-07,
      "model_forward_time": 5.615260362625122,
      "step": 192
    },
    {
      "epoch": 1.8310546875e-07,
      "step": 192,
      "training_step_time": 17.120606184005737
    },
    {
      "epoch": 1.8405914306640625e-07,
      "model_forward_time": 5.619052410125732,
      "step": 193
    },
    {
      "epoch": 1.8405914306640625e-07,
      "step": 193,
      "training_step_time": 17.100674867630005
    },
    {
      "epoch": 1.850128173828125e-07,
      "model_forward_time": 5.597915410995483,
      "step": 194
    },
    {
      "epoch": 1.850128173828125e-07,
      "step": 194,
      "training_step_time": 17.040138244628906
    },
    {
      "epoch": 1.8596649169921876e-07,
      "model_forward_time": 5.928178071975708,
      "step": 195
    },
    {
      "epoch": 1.8596649169921876e-07,
      "step": 195,
      "training_step_time": 17.358542919158936
    },
    {
      "epoch": 1.86920166015625e-07,
      "model_forward_time": 5.596522092819214,
      "step": 196
    },
    {
      "epoch": 1.86920166015625e-07,
      "step": 196,
      "training_step_time": 17.057849168777466
    },
    {
      "epoch": 1.8787384033203126e-07,
      "model_forward_time": 5.608537197113037,
      "step": 197
    },
    {
      "epoch": 1.8787384033203126e-07,
      "step": 197,
      "training_step_time": 17.094459533691406
    },
    {
      "epoch": 1.888275146484375e-07,
      "model_forward_time": 5.61134934425354,
      "step": 198
    },
    {
      "epoch": 1.888275146484375e-07,
      "step": 198,
      "training_step_time": 17.102195262908936
    },
    {
      "epoch": 1.8978118896484376e-07,
      "model_forward_time": 5.622015953063965,
      "step": 199
    },
    {
      "epoch": 1.8978118896484376e-07,
      "step": 199,
      "training_step_time": 17.114651679992676
    },
    {
      "epoch": 1.9073486328125e-07,
      "grad_norm": 1.979188084602356,
      "learning_rate": 1e-05,
      "loss": 0.1222,
      "step": 200
    },
    {
      "epoch": 1.9073486328125e-07,
      "model_forward_time": 5.620561122894287,
      "step": 200
    },
    {
      "dynamics_loss_avg": 0.0767642904073,
      "epoch": 1.9073486328125e-07,
      "step": 200
    },
    {
      "action_loss_avg": 0.05080116782337427,
      "epoch": 1.9073486328125e-07,
      "step": 200
    },
    {
      "epoch": 1.9073486328125e-07,
      "step": 200,
      "training_step_time": 17.460115909576416
    },
    {
      "epoch": 1.9168853759765626e-07,
      "model_forward_time": 5.613499879837036,
      "step": 201
    },
    {
      "epoch": 1.9168853759765626e-07,
      "step": 201,
      "training_step_time": 17.077190399169922
    },
    {
      "epoch": 1.926422119140625e-07,
      "model_forward_time": 5.59907865524292,
      "step": 202
    },
    {
      "epoch": 1.926422119140625e-07,
      "step": 202,
      "training_step_time": 17.073105573654175
    },
    {
      "epoch": 1.9359588623046876e-07,
      "model_forward_time": 5.608891487121582,
      "step": 203
    },
    {
      "epoch": 1.9359588623046876e-07,
      "step": 203,
      "training_step_time": 17.11198091506958
    },
    {
      "epoch": 1.94549560546875e-07,
      "model_forward_time": 5.639033079147339,
      "step": 204
    },
    {
      "epoch": 1.94549560546875e-07,
      "step": 204,
      "training_step_time": 17.124775409698486
    },
    {
      "epoch": 1.9550323486328124e-07,
      "model_forward_time": 5.600820779800415,
      "step": 205
    },
    {
      "epoch": 1.9550323486328124e-07,
      "step": 205,
      "training_step_time": 17.05144190788269
    },
    {
      "epoch": 1.964569091796875e-07,
      "model_forward_time": 5.937396049499512,
      "step": 206
    },
    {
      "epoch": 1.964569091796875e-07,
      "step": 206,
      "training_step_time": 17.376744747161865
    },
    {
      "epoch": 1.9741058349609374e-07,
      "model_forward_time": 5.585180282592773,
      "step": 207
    },
    {
      "epoch": 1.9741058349609374e-07,
      "step": 207,
      "training_step_time": 17.020911693572998
    },
    {
      "epoch": 1.983642578125e-07,
      "model_forward_time": 5.599476337432861,
      "step": 208
    },
    {
      "epoch": 1.983642578125e-07,
      "step": 208,
      "training_step_time": 17.056224822998047
    },
    {
      "epoch": 1.9931793212890624e-07,
      "model_forward_time": 5.596649169921875,
      "step": 209
    },
    {
      "epoch": 1.9931793212890624e-07,
      "step": 209,
      "training_step_time": 17.067695140838623
    },
    {
      "epoch": 2.002716064453125e-07,
      "grad_norm": 3.48921799659729,
      "learning_rate": 1e-05,
      "loss": 0.124,
      "step": 210
    },
    {
      "epoch": 2.002716064453125e-07,
      "model_forward_time": 5.609497785568237,
      "step": 210
    },
    {
      "dynamics_loss_avg": 0.07179250866174698,
      "epoch": 2.002716064453125e-07,
      "step": 210
    },
    {
      "action_loss_avg": 0.04783525727689266,
      "epoch": 2.002716064453125e-07,
      "step": 210
    },
    {
      "epoch": 2.002716064453125e-07,
      "step": 210,
      "training_step_time": 17.116711378097534
    },
    {
      "epoch": 2.0122528076171874e-07,
      "model_forward_time": 6.003995895385742,
      "step": 211
    },
    {
      "epoch": 2.0122528076171874e-07,
      "step": 211,
      "training_step_time": 17.502394437789917
    },
    {
      "epoch": 2.02178955078125e-07,
      "model_forward_time": 5.6046106815338135,
      "step": 212
    },
    {
      "epoch": 2.02178955078125e-07,
      "step": 212,
      "training_step_time": 17.07647442817688
    },
    {
      "epoch": 2.0313262939453125e-07,
      "model_forward_time": 5.604447364807129,
      "step": 213
    },
    {
      "epoch": 2.0313262939453125e-07,
      "step": 213,
      "training_step_time": 17.046313285827637
    },
    {
      "epoch": 2.040863037109375e-07,
      "model_forward_time": 5.581603765487671,
      "step": 214
    },
    {
      "epoch": 2.040863037109375e-07,
      "step": 214,
      "training_step_time": 16.99937629699707
    },
    {
      "epoch": 2.0503997802734375e-07,
      "model_forward_time": 5.585303544998169,
      "step": 215
    },
    {
      "epoch": 2.0503997802734375e-07,
      "step": 215,
      "training_step_time": 17.024492025375366
    },
    {
      "epoch": 2.0599365234375e-07,
      "model_forward_time": 5.606135368347168,
      "step": 216
    },
    {
      "epoch": 2.0599365234375e-07,
      "step": 216,
      "training_step_time": 17.41546392440796
    },
    {
      "epoch": 2.0694732666015625e-07,
      "model_forward_time": 5.593871593475342,
      "step": 217
    },
    {
      "epoch": 2.0694732666015625e-07,
      "step": 217,
      "training_step_time": 17.06522226333618
    },
    {
      "epoch": 2.079010009765625e-07,
      "model_forward_time": 5.606517314910889,
      "step": 218
    },
    {
      "epoch": 2.079010009765625e-07,
      "step": 218,
      "training_step_time": 17.087825059890747
    },
    {
      "epoch": 2.0885467529296875e-07,
      "model_forward_time": 5.611430644989014,
      "step": 219
    },
    {
      "epoch": 2.0885467529296875e-07,
      "step": 219,
      "training_step_time": 17.116047859191895
    },
    {
      "epoch": 2.09808349609375e-07,
      "grad_norm": 2.0128185749053955,
      "learning_rate": 1e-05,
      "loss": 0.1405,
      "step": 220
    },
    {
      "epoch": 2.09808349609375e-07,
      "model_forward_time": 5.612933397293091,
      "step": 220
    },
    {
      "dynamics_loss_avg": 0.08975301124155521,
      "epoch": 2.09808349609375e-07,
      "step": 220
    },
    {
      "action_loss_avg": 0.054640472680330274,
      "epoch": 2.09808349609375e-07,
      "step": 220
    },
    {
      "epoch": 2.09808349609375e-07,
      "step": 220,
      "training_step_time": 17.10082197189331
    },
    {
      "epoch": 2.1076202392578125e-07,
      "model_forward_time": 5.6130335330963135,
      "step": 221
    },
    {
      "epoch": 2.1076202392578125e-07,
      "step": 221,
      "training_step_time": 17.087596654891968
    },
    {
      "epoch": 2.117156982421875e-07,
      "model_forward_time": 5.98846960067749,
      "step": 222
    },
    {
      "epoch": 2.117156982421875e-07,
      "step": 222,
      "training_step_time": 17.45145058631897
    },
    {
      "epoch": 2.1266937255859375e-07,
      "model_forward_time": 5.596600294113159,
      "step": 223
    },
    {
      "epoch": 2.1266937255859375e-07,
      "step": 223,
      "training_step_time": 17.05972909927368
    },
    {
      "epoch": 2.13623046875e-07,
      "model_forward_time": 5.601090669631958,
      "step": 224
    },
    {
      "epoch": 2.13623046875e-07,
      "step": 224,
      "training_step_time": 17.06662893295288
    },
    {
      "epoch": 2.1457672119140626e-07,
      "model_forward_time": 5.605543851852417,
      "step": 225
    },
    {
      "epoch": 2.1457672119140626e-07,
      "step": 225,
      "training_step_time": 17.076900482177734
    },
    {
      "epoch": 2.155303955078125e-07,
      "model_forward_time": 5.60306978225708,
      "step": 226
    },
    {
      "epoch": 2.155303955078125e-07,
      "step": 226,
      "training_step_time": 17.074257135391235
    },
    {
      "epoch": 2.1648406982421876e-07,
      "model_forward_time": 5.956277132034302,
      "step": 227
    },
    {
      "epoch": 2.1648406982421876e-07,
      "step": 227,
      "training_step_time": 17.40352964401245
    },
    {
      "epoch": 2.17437744140625e-07,
      "model_forward_time": 5.594989776611328,
      "step": 228
    },
    {
      "epoch": 2.17437744140625e-07,
      "step": 228,
      "training_step_time": 17.017683506011963
    },
    {
      "epoch": 2.1839141845703126e-07,
      "model_forward_time": 5.585497140884399,
      "step": 229
    },
    {
      "epoch": 2.1839141845703126e-07,
      "step": 229,
      "training_step_time": 17.01555824279785
    },
    {
      "epoch": 2.193450927734375e-07,
      "grad_norm": 1.7824900150299072,
      "learning_rate": 1e-05,
      "loss": 0.1018,
      "step": 230
    },
    {
      "epoch": 2.193450927734375e-07,
      "model_forward_time": 5.5886852741241455,
      "step": 230
    },
    {
      "dynamics_loss_avg": 0.06420558020472526,
      "epoch": 2.193450927734375e-07,
      "step": 230
    },
    {
      "action_loss_avg": 0.029937134496867655,
      "epoch": 2.193450927734375e-07,
      "step": 230
    },
    {
      "epoch": 2.193450927734375e-07,
      "step": 230,
      "training_step_time": 17.050981998443604
    },
    {
      "epoch": 2.2029876708984376e-07,
      "model_forward_time": 5.6027162075042725,
      "step": 231
    },
    {
      "epoch": 2.2029876708984376e-07,
      "step": 231,
      "training_step_time": 17.054308891296387
    },
    {
      "epoch": 2.2125244140625e-07,
      "model_forward_time": 5.5967981815338135,
      "step": 232
    },
    {
      "epoch": 2.2125244140625e-07,
      "step": 232,
      "training_step_time": 17.03212070465088
    },
    {
      "epoch": 2.2220611572265626e-07,
      "model_forward_time": 5.967787027359009,
      "step": 233
    },
    {
      "epoch": 2.2220611572265626e-07,
      "step": 233,
      "training_step_time": 17.43043828010559
    },
    {
      "epoch": 2.231597900390625e-07,
      "model_forward_time": 5.600518226623535,
      "step": 234
    },
    {
      "epoch": 2.231597900390625e-07,
      "step": 234,
      "training_step_time": 17.056878566741943
    },
    {
      "epoch": 2.2411346435546874e-07,
      "model_forward_time": 5.611528635025024,
      "step": 235
    },
    {
      "epoch": 2.2411346435546874e-07,
      "step": 235,
      "training_step_time": 17.101916074752808
    },
    {
      "epoch": 2.25067138671875e-07,
      "model_forward_time": 5.6146979331970215,
      "step": 236
    },
    {
      "epoch": 2.25067138671875e-07,
      "step": 236,
      "training_step_time": 17.116347074508667
    },
    {
      "epoch": 2.2602081298828124e-07,
      "model_forward_time": 5.607531547546387,
      "step": 237
    },
    {
      "epoch": 2.2602081298828124e-07,
      "step": 237,
      "training_step_time": 17.09160566329956
    },
    {
      "epoch": 2.269744873046875e-07,
      "model_forward_time": 5.9600489139556885,
      "step": 238
    },
    {
      "epoch": 2.269744873046875e-07,
      "step": 238,
      "training_step_time": 17.418473482131958
    },
    {
      "epoch": 2.2792816162109374e-07,
      "model_forward_time": 5.5964195728302,
      "step": 239
    },
    {
      "epoch": 2.2792816162109374e-07,
      "step": 239,
      "training_step_time": 17.042383909225464
    },
    {
      "epoch": 2.288818359375e-07,
      "grad_norm": 0.815731942653656,
      "learning_rate": 1e-05,
      "loss": 0.1085,
      "step": 240
    },
    {
      "epoch": 2.288818359375e-07,
      "model_forward_time": 5.581787347793579,
      "step": 240
    },
    {
      "dynamics_loss_avg": 0.06751575134694576,
      "epoch": 2.288818359375e-07,
      "step": 240
    },
    {
      "action_loss_avg": 0.0383166913408786,
      "epoch": 2.288818359375e-07,
      "step": 240
    },
    {
      "epoch": 2.288818359375e-07,
      "step": 240,
      "training_step_time": 17.02783203125
    },
    {
      "epoch": 2.2983551025390624e-07,
      "model_forward_time": 5.59224271774292,
      "step": 241
    },
    {
      "epoch": 2.2983551025390624e-07,
      "step": 241,
      "training_step_time": 17.065092086791992
    },
    {
      "epoch": 2.307891845703125e-07,
      "model_forward_time": 5.597123861312866,
      "step": 242
    },
    {
      "epoch": 2.307891845703125e-07,
      "step": 242,
      "training_step_time": 17.088340759277344
    },
    {
      "epoch": 2.3174285888671874e-07,
      "model_forward_time": 5.616132736206055,
      "step": 243
    },
    {
      "epoch": 2.3174285888671874e-07,
      "step": 243,
      "training_step_time": 17.153554677963257
    },
    {
      "epoch": 2.32696533203125e-07,
      "model_forward_time": 5.980180025100708,
      "step": 244
    },
    {
      "epoch": 2.32696533203125e-07,
      "step": 244,
      "training_step_time": 17.43187165260315
    },
    {
      "epoch": 2.3365020751953125e-07,
      "model_forward_time": 5.580890893936157,
      "step": 245
    },
    {
      "epoch": 2.3365020751953125e-07,
      "step": 245,
      "training_step_time": 16.977755308151245
    },
    {
      "epoch": 2.346038818359375e-07,
      "model_forward_time": 5.570791006088257,
      "step": 246
    },
    {
      "epoch": 2.346038818359375e-07,
      "step": 246,
      "training_step_time": 16.988791704177856
    },
    {
      "epoch": 2.3555755615234375e-07,
      "model_forward_time": 5.5639331340789795,
      "step": 247
    },
    {
      "epoch": 2.3555755615234375e-07,
      "step": 247,
      "training_step_time": 16.991425037384033
    },
    {
      "epoch": 2.3651123046875e-07,
      "model_forward_time": 5.56477427482605,
      "step": 248
    },
    {
      "epoch": 2.3651123046875e-07,
      "step": 248,
      "training_step_time": 16.9951012134552
    },
    {
      "epoch": 2.3746490478515625e-07,
      "model_forward_time": 5.924005746841431,
      "step": 249
    },
    {
      "epoch": 2.3746490478515625e-07,
      "step": 249,
      "training_step_time": 17.33001399040222
    },
    {
      "epoch": 2.384185791015625e-07,
      "grad_norm": 1.694961667060852,
      "learning_rate": 1e-05,
      "loss": 0.1061,
      "step": 250
    },
    {
      "epoch": 2.384185791015625e-07,
      "model_forward_time": 5.594616174697876,
      "step": 250
    },
    {
      "dynamics_loss_avg": 0.07188496328890323,
      "epoch": 2.384185791015625e-07,
      "step": 250
    },
    {
      "action_loss_avg": 0.04600810166448355,
      "epoch": 2.384185791015625e-07,
      "step": 250
    },
    {
      "epoch": 2.384185791015625e-07,
      "step": 250,
      "training_step_time": 17.00856590270996
    },
    {
      "epoch": 2.3937225341796875e-07,
      "model_forward_time": 5.582397222518921,
      "step": 251
    },
    {
      "epoch": 2.3937225341796875e-07,
      "step": 251,
      "training_step_time": 17.026217937469482
    },
    {
      "epoch": 2.40325927734375e-07,
      "model_forward_time": 5.591115951538086,
      "step": 252
    },
    {
      "epoch": 2.40325927734375e-07,
      "step": 252,
      "training_step_time": 17.02333664894104
    },
    {
      "epoch": 2.4127960205078125e-07,
      "model_forward_time": 5.591325044631958,
      "step": 253
    },
    {
      "epoch": 2.4127960205078125e-07,
      "step": 253,
      "training_step_time": 17.031644582748413
    },
    {
      "epoch": 2.422332763671875e-07,
      "model_forward_time": 5.602702617645264,
      "step": 254
    },
    {
      "epoch": 2.422332763671875e-07,
      "step": 254,
      "training_step_time": 17.451000213623047
    },
    {
      "epoch": 2.4318695068359375e-07,
      "model_forward_time": 5.602764844894409,
      "step": 255
    },
    {
      "epoch": 2.4318695068359375e-07,
      "step": 255,
      "training_step_time": 17.067627668380737
    },
    {
      "epoch": 2.44140625e-07,
      "model_forward_time": 5.602911949157715,
      "step": 256
    },
    {
      "epoch": 2.44140625e-07,
      "step": 256,
      "training_step_time": 17.06487464904785
    },
    {
      "epoch": 2.4509429931640626e-07,
      "model_forward_time": 5.5963358879089355,
      "step": 257
    },
    {
      "epoch": 2.4509429931640626e-07,
      "step": 257,
      "training_step_time": 17.022968530654907
    },
    {
      "epoch": 2.460479736328125e-07,
      "model_forward_time": 5.5904762744903564,
      "step": 258
    },
    {
      "epoch": 2.460479736328125e-07,
      "step": 258,
      "training_step_time": 17.035993814468384
    },
    {
      "epoch": 2.4700164794921876e-07,
      "model_forward_time": 5.592491626739502,
      "step": 259
    },
    {
      "epoch": 2.4700164794921876e-07,
      "step": 259,
      "training_step_time": 17.058645009994507
    },
    {
      "epoch": 2.47955322265625e-07,
      "grad_norm": 1.943295955657959,
      "learning_rate": 1e-05,
      "loss": 0.1189,
      "step": 260
    },
    {
      "epoch": 2.47955322265625e-07,
      "model_forward_time": 5.975651979446411,
      "step": 260
    },
    {
      "dynamics_loss_avg": 0.07788472780957818,
      "epoch": 2.47955322265625e-07,
      "step": 260
    },
    {
      "action_loss_avg": 0.04049905370920896,
      "epoch": 2.47955322265625e-07,
      "step": 260
    },
    {
      "epoch": 2.47955322265625e-07,
      "step": 260,
      "training_step_time": 17.490822792053223
    },
    {
      "epoch": 2.4890899658203126e-07,
      "model_forward_time": 5.6114654541015625,
      "step": 261
    },
    {
      "epoch": 2.4890899658203126e-07,
      "step": 261,
      "training_step_time": 17.080394744873047
    },
    {
      "epoch": 2.498626708984375e-07,
      "model_forward_time": 5.589546203613281,
      "step": 262
    },
    {
      "epoch": 2.498626708984375e-07,
      "step": 262,
      "training_step_time": 17.005521059036255
    },
    {
      "epoch": 2.5081634521484376e-07,
      "model_forward_time": 5.590474843978882,
      "step": 263
    },
    {
      "epoch": 2.5081634521484376e-07,
      "step": 263,
      "training_step_time": 17.031410694122314
    },
    {
      "epoch": 2.5177001953125e-07,
      "model_forward_time": 5.5907673835754395,
      "step": 264
    },
    {
      "epoch": 2.5177001953125e-07,
      "step": 264,
      "training_step_time": 17.03397297859192
    },
    {
      "epoch": 2.5272369384765626e-07,
      "model_forward_time": 5.97426438331604,
      "step": 265
    },
    {
      "epoch": 2.5272369384765626e-07,
      "step": 265,
      "training_step_time": 17.426306009292603
    },
    {
      "epoch": 2.536773681640625e-07,
      "model_forward_time": 5.608394622802734,
      "step": 266
    },
    {
      "epoch": 2.536773681640625e-07,
      "step": 266,
      "training_step_time": 17.070871353149414
    },
    {
      "epoch": 2.5463104248046876e-07,
      "model_forward_time": 5.606316566467285,
      "step": 267
    },
    {
      "epoch": 2.5463104248046876e-07,
      "step": 267,
      "training_step_time": 17.086058855056763
    },
    {
      "epoch": 2.55584716796875e-07,
      "model_forward_time": 5.6016058921813965,
      "step": 268
    },
    {
      "epoch": 2.55584716796875e-07,
      "step": 268,
      "training_step_time": 17.086066246032715
    },
    {
      "epoch": 2.5653839111328127e-07,
      "model_forward_time": 5.61359715461731,
      "step": 269
    },
    {
      "epoch": 2.5653839111328127e-07,
      "step": 269,
      "training_step_time": 17.059858560562134
    },
    {
      "epoch": 2.574920654296875e-07,
      "grad_norm": 2.1475934982299805,
      "learning_rate": 1e-05,
      "loss": 0.1272,
      "step": 270
    },
    {
      "epoch": 2.574920654296875e-07,
      "model_forward_time": 5.59052038192749,
      "step": 270
    },
    {
      "dynamics_loss_avg": 0.08025160320103168,
      "epoch": 2.574920654296875e-07,
      "step": 270
    },
    {
      "action_loss_avg": 0.04842926766723395,
      "epoch": 2.574920654296875e-07,
      "step": 270
    },
    {
      "epoch": 2.574920654296875e-07,
      "step": 270,
      "training_step_time": 17.03550410270691
    },
    {
      "epoch": 2.5844573974609377e-07,
      "model_forward_time": 5.92849588394165,
      "step": 271
    },
    {
      "epoch": 2.5844573974609377e-07,
      "step": 271,
      "training_step_time": 17.332490921020508
    },
    {
      "epoch": 2.593994140625e-07,
      "model_forward_time": 5.58753514289856,
      "step": 272
    },
    {
      "epoch": 2.593994140625e-07,
      "step": 272,
      "training_step_time": 17.021122455596924
    },
    {
      "epoch": 2.6035308837890627e-07,
      "model_forward_time": 5.592071533203125,
      "step": 273
    },
    {
      "epoch": 2.6035308837890627e-07,
      "step": 273,
      "training_step_time": 17.027200937271118
    },
    {
      "epoch": 2.613067626953125e-07,
      "model_forward_time": 5.591593503952026,
      "step": 274
    },
    {
      "epoch": 2.613067626953125e-07,
      "step": 274,
      "training_step_time": 17.02534317970276
    },
    {
      "epoch": 2.6226043701171877e-07,
      "model_forward_time": 5.600379228591919,
      "step": 275
    },
    {
      "epoch": 2.6226043701171877e-07,
      "step": 275,
      "training_step_time": 17.081436157226562
    },
    {
      "epoch": 2.63214111328125e-07,
      "model_forward_time": 5.980151414871216,
      "step": 276
    },
    {
      "epoch": 2.63214111328125e-07,
      "step": 276,
      "training_step_time": 17.486363410949707
    },
    {
      "epoch": 2.6416778564453127e-07,
      "model_forward_time": 5.615906000137329,
      "step": 277
    },
    {
      "epoch": 2.6416778564453127e-07,
      "step": 277,
      "training_step_time": 17.09386706352234
    },
    {
      "epoch": 2.651214599609375e-07,
      "model_forward_time": 5.596087217330933,
      "step": 278
    },
    {
      "epoch": 2.651214599609375e-07,
      "step": 278,
      "training_step_time": 17.03489065170288
    },
    {
      "epoch": 2.660751342773438e-07,
      "model_forward_time": 5.58189058303833,
      "step": 279
    },
    {
      "epoch": 2.660751342773438e-07,
      "step": 279,
      "training_step_time": 17.01270627975464
    },
    {
      "epoch": 2.6702880859375e-07,
      "grad_norm": 2.6546053886413574,
      "learning_rate": 1e-05,
      "loss": 0.1414,
      "step": 280
    },
    {
      "epoch": 2.6702880859375e-07,
      "model_forward_time": 5.587113618850708,
      "step": 280
    },
    {
      "dynamics_loss_avg": 0.08792693186551333,
      "epoch": 2.6702880859375e-07,
      "step": 280
    },
    {
      "action_loss_avg": 0.04679098529741168,
      "epoch": 2.6702880859375e-07,
      "step": 280
    },
    {
      "epoch": 2.6702880859375e-07,
      "step": 280,
      "training_step_time": 17.002458572387695
    },
    {
      "epoch": 2.679824829101563e-07,
      "model_forward_time": 5.581991910934448,
      "step": 281
    },
    {
      "epoch": 2.679824829101563e-07,
      "step": 281,
      "training_step_time": 17.01447367668152
    },
    {
      "epoch": 2.689361572265625e-07,
      "model_forward_time": 5.954323053359985,
      "step": 282
    },
    {
      "epoch": 2.689361572265625e-07,
      "step": 282,
      "training_step_time": 17.404338836669922
    },
    {
      "epoch": 2.698898315429687e-07,
      "model_forward_time": 5.604142189025879,
      "step": 283
    },
    {
      "epoch": 2.698898315429687e-07,
      "step": 283,
      "training_step_time": 17.112542867660522
    },
    {
      "epoch": 2.70843505859375e-07,
      "model_forward_time": 5.635445833206177,
      "step": 284
    },
    {
      "epoch": 2.70843505859375e-07,
      "step": 284,
      "training_step_time": 17.156084060668945
    },
    {
      "epoch": 2.7179718017578123e-07,
      "model_forward_time": 5.620648384094238,
      "step": 285
    },
    {
      "epoch": 2.7179718017578123e-07,
      "step": 285,
      "training_step_time": 17.085015058517456
    },
    {
      "epoch": 2.727508544921875e-07,
      "model_forward_time": 5.598835468292236,
      "step": 286
    },
    {
      "epoch": 2.727508544921875e-07,
      "step": 286,
      "training_step_time": 17.03766942024231
    },
    {
      "epoch": 2.7370452880859373e-07,
      "model_forward_time": 5.949311017990112,
      "step": 287
    },
    {
      "epoch": 2.7370452880859373e-07,
      "step": 287,
      "training_step_time": 17.36637258529663
    },
    {
      "epoch": 2.74658203125e-07,
      "model_forward_time": 5.593549728393555,
      "step": 288
    },
    {
      "epoch": 2.74658203125e-07,
      "step": 288,
      "training_step_time": 17.049559593200684
    },
    {
      "epoch": 2.7561187744140623e-07,
      "model_forward_time": 5.604710340499878,
      "step": 289
    },
    {
      "epoch": 2.7561187744140623e-07,
      "step": 289,
      "training_step_time": 17.070480585098267
    },
    {
      "epoch": 2.765655517578125e-07,
      "grad_norm": 1.3907634019851685,
      "learning_rate": 1e-05,
      "loss": 0.115,
      "step": 290
    },
    {
      "epoch": 2.765655517578125e-07,
      "model_forward_time": 5.628479957580566,
      "step": 290
    },
    {
      "dynamics_loss_avg": 0.07960245348513126,
      "epoch": 2.765655517578125e-07,
      "step": 290
    },
    {
      "action_loss_avg": 0.042633849754929544,
      "epoch": 2.765655517578125e-07,
      "step": 290
    },
    {
      "epoch": 2.765655517578125e-07,
      "step": 290,
      "training_step_time": 17.127748727798462
    },
    {
      "epoch": 2.7751922607421873e-07,
      "model_forward_time": 5.615041732788086,
      "step": 291
    },
    {
      "epoch": 2.7751922607421873e-07,
      "step": 291,
      "training_step_time": 17.103742361068726
    },
    {
      "epoch": 2.78472900390625e-07,
      "model_forward_time": 5.616617441177368,
      "step": 292
    },
    {
      "epoch": 2.78472900390625e-07,
      "step": 292,
      "training_step_time": 17.108824253082275
    },
    {
      "epoch": 2.7942657470703123e-07,
      "model_forward_time": 5.9509875774383545,
      "step": 293
    },
    {
      "epoch": 2.7942657470703123e-07,
      "step": 293,
      "training_step_time": 17.37803030014038
    },
    {
      "epoch": 2.803802490234375e-07,
      "model_forward_time": 5.605287551879883,
      "step": 294
    },
    {
      "epoch": 2.803802490234375e-07,
      "step": 294,
      "training_step_time": 17.089163303375244
    },
    {
      "epoch": 2.8133392333984374e-07,
      "model_forward_time": 5.607502460479736,
      "step": 295
    },
    {
      "epoch": 2.8133392333984374e-07,
      "step": 295,
      "training_step_time": 17.077168941497803
    },
    {
      "epoch": 2.8228759765625e-07,
      "model_forward_time": 5.608356237411499,
      "step": 296
    },
    {
      "epoch": 2.8228759765625e-07,
      "step": 296,
      "training_step_time": 17.057217597961426
    },
    {
      "epoch": 2.8324127197265624e-07,
      "model_forward_time": 5.594308853149414,
      "step": 297
    },
    {
      "epoch": 2.8324127197265624e-07,
      "step": 297,
      "training_step_time": 17.035280466079712
    },
    {
      "epoch": 2.841949462890625e-07,
      "model_forward_time": 5.957638740539551,
      "step": 298
    },
    {
      "epoch": 2.841949462890625e-07,
      "step": 298,
      "training_step_time": 17.416232109069824
    },
    {
      "epoch": 2.8514862060546874e-07,
      "model_forward_time": 5.605282306671143,
      "step": 299
    },
    {
      "epoch": 2.8514862060546874e-07,
      "step": 299,
      "training_step_time": 17.09113597869873
    },
    {
      "epoch": 2.86102294921875e-07,
      "grad_norm": 1.0871576070785522,
      "learning_rate": 1e-05,
      "loss": 0.1199,
      "step": 300
    },
    {
      "epoch": 2.86102294921875e-07,
      "model_forward_time": 5.6093456745147705,
      "step": 300
    },
    {
      "dynamics_loss_avg": 0.07225110344588756,
      "epoch": 2.86102294921875e-07,
      "step": 300
    },
    {
      "action_loss_avg": 0.042492296732962134,
      "epoch": 2.86102294921875e-07,
      "step": 300
    },
    {
      "epoch": 2.86102294921875e-07,
      "step": 300,
      "training_step_time": 17.072463750839233
    },
    {
      "epoch": 2.8705596923828124e-07,
      "model_forward_time": 5.604180097579956,
      "step": 301
    },
    {
      "epoch": 2.8705596923828124e-07,
      "step": 301,
      "training_step_time": 17.05655598640442
    },
    {
      "epoch": 2.880096435546875e-07,
      "model_forward_time": 5.598243474960327,
      "step": 302
    },
    {
      "epoch": 2.880096435546875e-07,
      "step": 302,
      "training_step_time": 17.035034656524658
    },
    {
      "epoch": 2.8896331787109374e-07,
      "model_forward_time": 5.952453374862671,
      "step": 303
    },
    {
      "epoch": 2.8896331787109374e-07,
      "step": 303,
      "training_step_time": 17.4037926197052
    },
    {
      "epoch": 2.899169921875e-07,
      "model_forward_time": 5.600820064544678,
      "step": 304
    },
    {
      "epoch": 2.899169921875e-07,
      "step": 304,
      "training_step_time": 17.07973623275757
    },
    {
      "epoch": 2.9087066650390624e-07,
      "model_forward_time": 5.621901750564575,
      "step": 305
    },
    {
      "epoch": 2.9087066650390624e-07,
      "step": 305,
      "training_step_time": 17.119747400283813
    },
    {
      "epoch": 2.918243408203125e-07,
      "model_forward_time": 5.620824337005615,
      "step": 306
    },
    {
      "epoch": 2.918243408203125e-07,
      "step": 306,
      "training_step_time": 17.12924313545227
    },
    {
      "epoch": 2.9277801513671875e-07,
      "model_forward_time": 5.602719306945801,
      "step": 307
    },
    {
      "epoch": 2.9277801513671875e-07,
      "step": 307,
      "training_step_time": 17.07754397392273
    },
    {
      "epoch": 2.93731689453125e-07,
      "model_forward_time": 5.605270624160767,
      "step": 308
    },
    {
      "epoch": 2.93731689453125e-07,
      "step": 308,
      "training_step_time": 17.073800802230835
    },
    {
      "epoch": 2.9468536376953125e-07,
      "model_forward_time": 5.973750114440918,
      "step": 309
    },
    {
      "epoch": 2.9468536376953125e-07,
      "step": 309,
      "training_step_time": 17.432127237319946
    },
    {
      "epoch": 2.956390380859375e-07,
      "grad_norm": 2.8067479133605957,
      "learning_rate": 1e-05,
      "loss": 0.1317,
      "step": 310
    },
    {
      "epoch": 2.956390380859375e-07,
      "model_forward_time": 5.60139012336731,
      "step": 310
    },
    {
      "dynamics_loss_avg": 0.08409080430865287,
      "epoch": 2.956390380859375e-07,
      "step": 310
    },
    {
      "action_loss_avg": 0.047078983625397086,
      "epoch": 2.956390380859375e-07,
      "step": 310
    },
    {
      "epoch": 2.956390380859375e-07,
      "step": 310,
      "training_step_time": 17.055782318115234
    },
    {
      "epoch": 2.9659271240234375e-07,
      "model_forward_time": 5.5971033573150635,
      "step": 311
    },
    {
      "epoch": 2.9659271240234375e-07,
      "step": 311,
      "training_step_time": 17.058936834335327
    },
    {
      "epoch": 2.9754638671875e-07,
      "model_forward_time": 5.595714569091797,
      "step": 312
    },
    {
      "epoch": 2.9754638671875e-07,
      "step": 312,
      "training_step_time": 17.063572883605957
    },
    {
      "epoch": 2.9850006103515625e-07,
      "model_forward_time": 5.609219312667847,
      "step": 313
    },
    {
      "epoch": 2.9850006103515625e-07,
      "step": 313,
      "training_step_time": 17.106091499328613
    },
    {
      "epoch": 2.994537353515625e-07,
      "model_forward_time": 5.612901210784912,
      "step": 314
    },
    {
      "epoch": 2.994537353515625e-07,
      "step": 314,
      "training_step_time": 17.419898748397827
    },
    {
      "epoch": 3.0040740966796875e-07,
      "model_forward_time": 5.620344877243042,
      "step": 315
    },
    {
      "epoch": 3.0040740966796875e-07,
      "step": 315,
      "training_step_time": 17.10828948020935
    },
    {
      "epoch": 3.01361083984375e-07,
      "model_forward_time": 5.606201171875,
      "step": 316
    },
    {
      "epoch": 3.01361083984375e-07,
      "step": 316,
      "training_step_time": 17.0818030834198
    },
    {
      "epoch": 3.0231475830078125e-07,
      "model_forward_time": 5.612060785293579,
      "step": 317
    },
    {
      "epoch": 3.0231475830078125e-07,
      "step": 317,
      "training_step_time": 17.111778736114502
    },
    {
      "epoch": 3.032684326171875e-07,
      "model_forward_time": 5.622238636016846,
      "step": 318
    },
    {
      "epoch": 3.032684326171875e-07,
      "step": 318,
      "training_step_time": 17.119714498519897
    },
    {
      "epoch": 3.0422210693359376e-07,
      "model_forward_time": 5.615356206893921,
      "step": 319
    },
    {
      "epoch": 3.0422210693359376e-07,
      "step": 319,
      "training_step_time": 17.09915566444397
    },
    {
      "epoch": 3.0517578125e-07,
      "grad_norm": 0.2673991918563843,
      "learning_rate": 1e-05,
      "loss": 0.1085,
      "step": 320
    },
    {
      "epoch": 3.0517578125e-07,
      "model_forward_time": 5.946145296096802,
      "step": 320
    },
    {
      "dynamics_loss_avg": 0.0641440587118268,
      "epoch": 3.0517578125e-07,
      "step": 320
    },
    {
      "action_loss_avg": 0.04074212065897882,
      "epoch": 3.0517578125e-07,
      "step": 320
    },
    {
      "epoch": 3.0517578125e-07,
      "step": 320,
      "training_step_time": 17.369829177856445
    },
    {
      "epoch": 3.0612945556640626e-07,
      "model_forward_time": 5.578166723251343,
      "step": 321
    },
    {
      "epoch": 3.0612945556640626e-07,
      "step": 321,
      "training_step_time": 16.989697694778442
    },
    {
      "epoch": 3.070831298828125e-07,
      "model_forward_time": 5.572685241699219,
      "step": 322
    },
    {
      "epoch": 3.070831298828125e-07,
      "step": 322,
      "training_step_time": 16.971193552017212
    },
    {
      "epoch": 3.0803680419921876e-07,
      "model_forward_time": 5.573858737945557,
      "step": 323
    },
    {
      "epoch": 3.0803680419921876e-07,
      "step": 323,
      "training_step_time": 17.003729343414307
    },
    {
      "epoch": 3.08990478515625e-07,
      "model_forward_time": 5.5804619789123535,
      "step": 324
    },
    {
      "epoch": 3.08990478515625e-07,
      "step": 324,
      "training_step_time": 16.99650263786316
    },
    {
      "epoch": 3.0994415283203126e-07,
      "model_forward_time": 5.943991661071777,
      "step": 325
    },
    {
      "epoch": 3.0994415283203126e-07,
      "step": 325,
      "training_step_time": 17.35979962348938
    },
    {
      "epoch": 3.108978271484375e-07,
      "model_forward_time": 5.606292724609375,
      "step": 326
    },
    {
      "epoch": 3.108978271484375e-07,
      "step": 326,
      "training_step_time": 17.03225541114807
    },
    {
      "epoch": 3.1185150146484376e-07,
      "model_forward_time": 5.578652858734131,
      "step": 327
    },
    {
      "epoch": 3.1185150146484376e-07,
      "step": 327,
      "training_step_time": 17.029969215393066
    },
    {
      "epoch": 3.1280517578125e-07,
      "model_forward_time": 5.587142467498779,
      "step": 328
    },
    {
      "epoch": 3.1280517578125e-07,
      "step": 328,
      "training_step_time": 17.023709297180176
    },
    {
      "epoch": 3.1375885009765626e-07,
      "model_forward_time": 5.59787130355835,
      "step": 329
    },
    {
      "epoch": 3.1375885009765626e-07,
      "step": 329,
      "training_step_time": 17.067554235458374
    },
    {
      "epoch": 3.147125244140625e-07,
      "grad_norm": 1.4198534488677979,
      "learning_rate": 1e-05,
      "loss": 0.1167,
      "step": 330
    },
    {
      "epoch": 3.147125244140625e-07,
      "model_forward_time": 5.614817142486572,
      "step": 330
    },
    {
      "dynamics_loss_avg": 0.07748199198395014,
      "epoch": 3.147125244140625e-07,
      "step": 330
    },
    {
      "action_loss_avg": 0.0363432802259922,
      "epoch": 3.147125244140625e-07,
      "step": 330
    },
    {
      "epoch": 3.147125244140625e-07,
      "step": 330,
      "training_step_time": 17.12643051147461
    },
    {
      "epoch": 3.1566619873046877e-07,
      "model_forward_time": 5.954533100128174,
      "step": 331
    },
    {
      "epoch": 3.1566619873046877e-07,
      "step": 331,
      "training_step_time": 17.429604053497314
    },
    {
      "epoch": 3.16619873046875e-07,
      "model_forward_time": 5.614620208740234,
      "step": 332
    },
    {
      "epoch": 3.16619873046875e-07,
      "step": 332,
      "training_step_time": 17.08944320678711
    },
    {
      "epoch": 3.1757354736328127e-07,
      "model_forward_time": 5.597102642059326,
      "step": 333
    },
    {
      "epoch": 3.1757354736328127e-07,
      "step": 333,
      "training_step_time": 17.034236431121826
    },
    {
      "epoch": 3.185272216796875e-07,
      "model_forward_time": 5.588478088378906,
      "step": 334
    },
    {
      "epoch": 3.185272216796875e-07,
      "step": 334,
      "training_step_time": 17.019750595092773
    },
    {
      "epoch": 3.1948089599609377e-07,
      "model_forward_time": 5.583276271820068,
      "step": 335
    },
    {
      "epoch": 3.1948089599609377e-07,
      "step": 335,
      "training_step_time": 17.04328680038452
    },
    {
      "epoch": 3.204345703125e-07,
      "model_forward_time": 5.945930480957031,
      "step": 336
    },
    {
      "epoch": 3.204345703125e-07,
      "step": 336,
      "training_step_time": 17.427277088165283
    },
    {
      "epoch": 3.2138824462890627e-07,
      "model_forward_time": 5.618454694747925,
      "step": 337
    },
    {
      "epoch": 3.2138824462890627e-07,
      "step": 337,
      "training_step_time": 17.131093978881836
    },
    {
      "epoch": 3.223419189453125e-07,
      "model_forward_time": 5.619553804397583,
      "step": 338
    },
    {
      "epoch": 3.223419189453125e-07,
      "step": 338,
      "training_step_time": 17.112713098526
    },
    {
      "epoch": 3.2329559326171877e-07,
      "model_forward_time": 5.599277496337891,
      "step": 339
    },
    {
      "epoch": 3.2329559326171877e-07,
      "step": 339,
      "training_step_time": 17.050367832183838
    },
    {
      "epoch": 3.24249267578125e-07,
      "grad_norm": 1.1056430339813232,
      "learning_rate": 1e-05,
      "loss": 0.1094,
      "step": 340
    },
    {
      "epoch": 3.24249267578125e-07,
      "model_forward_time": 5.589550733566284,
      "step": 340
    },
    {
      "dynamics_loss_avg": 0.078439150005579,
      "epoch": 3.24249267578125e-07,
      "step": 340
    },
    {
      "action_loss_avg": 0.044992228876799344,
      "epoch": 3.24249267578125e-07,
      "step": 340
    },
    {
      "epoch": 3.24249267578125e-07,
      "step": 340,
      "training_step_time": 17.038951635360718
    },
    {
      "epoch": 3.252029418945313e-07,
      "model_forward_time": 5.597823619842529,
      "step": 341
    },
    {
      "epoch": 3.252029418945313e-07,
      "step": 341,
      "training_step_time": 17.426952123641968
    },
    {
      "epoch": 3.261566162109375e-07,
      "model_forward_time": 5.6048362255096436,
      "step": 342
    },
    {
      "epoch": 3.261566162109375e-07,
      "step": 342,
      "training_step_time": 17.081634759902954
    },
    {
      "epoch": 3.271102905273438e-07,
      "model_forward_time": 5.600597620010376,
      "step": 343
    },
    {
      "epoch": 3.271102905273438e-07,
      "step": 343,
      "training_step_time": 17.073065042495728
    },
    {
      "epoch": 3.2806396484375e-07,
      "model_forward_time": 5.602693557739258,
      "step": 344
    },
    {
      "epoch": 3.2806396484375e-07,
      "step": 344,
      "training_step_time": 17.082104206085205
    },
    {
      "epoch": 3.290176391601562e-07,
      "model_forward_time": 5.603657245635986,
      "step": 345
    },
    {
      "epoch": 3.290176391601562e-07,
      "step": 345,
      "training_step_time": 17.0821590423584
    },
    {
      "epoch": 3.299713134765625e-07,
      "model_forward_time": 5.602646112442017,
      "step": 346
    },
    {
      "epoch": 3.299713134765625e-07,
      "step": 346,
      "training_step_time": 17.080182313919067
    },
    {
      "epoch": 3.309249877929687e-07,
      "model_forward_time": 5.94417929649353,
      "step": 347
    },
    {
      "epoch": 3.309249877929687e-07,
      "step": 347,
      "training_step_time": 17.42571473121643
    },
    {
      "epoch": 3.31878662109375e-07,
      "model_forward_time": 5.60689377784729,
      "step": 348
    },
    {
      "epoch": 3.31878662109375e-07,
      "step": 348,
      "training_step_time": 17.09152626991272
    },
    {
      "epoch": 3.3283233642578123e-07,
      "model_forward_time": 5.620347499847412,
      "step": 349
    },
    {
      "epoch": 3.3283233642578123e-07,
      "step": 349,
      "training_step_time": 17.10755491256714
    },
    {
      "epoch": 3.337860107421875e-07,
      "grad_norm": 1.943939447402954,
      "learning_rate": 1e-05,
      "loss": 0.1274,
      "step": 350
    },
    {
      "epoch": 3.337860107421875e-07,
      "model_forward_time": 5.618785858154297,
      "step": 350
    },
    {
      "dynamics_loss_avg": 0.07324151769280433,
      "epoch": 3.337860107421875e-07,
      "step": 350
    },
    {
      "action_loss_avg": 0.048130860552191734,
      "epoch": 3.337860107421875e-07,
      "step": 350
    },
    {
      "epoch": 3.337860107421875e-07,
      "step": 350,
      "training_step_time": 17.09413504600525
    },
    {
      "epoch": 3.3473968505859373e-07,
      "model_forward_time": 5.590093374252319,
      "step": 351
    },
    {
      "epoch": 3.3473968505859373e-07,
      "step": 351,
      "training_step_time": 17.037535190582275
    },
    {
      "epoch": 3.35693359375e-07,
      "model_forward_time": 5.588232040405273,
      "step": 352
    },
    {
      "epoch": 3.35693359375e-07,
      "step": 352,
      "training_step_time": 17.372549772262573
    },
    {
      "epoch": 3.3664703369140623e-07,
      "model_forward_time": 5.595671653747559,
      "step": 353
    },
    {
      "epoch": 3.3664703369140623e-07,
      "step": 353,
      "training_step_time": 17.067819118499756
    },
    {
      "epoch": 3.376007080078125e-07,
      "model_forward_time": 5.614426612854004,
      "step": 354
    },
    {
      "epoch": 3.376007080078125e-07,
      "step": 354,
      "training_step_time": 17.1376211643219
    },
    {
      "epoch": 3.3855438232421873e-07,
      "model_forward_time": 5.639795541763306,
      "step": 355
    },
    {
      "epoch": 3.3855438232421873e-07,
      "step": 355,
      "training_step_time": 17.14620351791382
    },
    {
      "epoch": 3.39508056640625e-07,
      "model_forward_time": 5.59555983543396,
      "step": 356
    },
    {
      "epoch": 3.39508056640625e-07,
      "step": 356,
      "training_step_time": 17.058629274368286
    },
    {
      "epoch": 3.4046173095703123e-07,
      "model_forward_time": 5.585467100143433,
      "step": 357
    },
    {
      "epoch": 3.4046173095703123e-07,
      "step": 357,
      "training_step_time": 17.02792501449585
    },
    {
      "epoch": 3.414154052734375e-07,
      "model_forward_time": 5.921554327011108,
      "step": 358
    },
    {
      "epoch": 3.414154052734375e-07,
      "step": 358,
      "training_step_time": 17.34466552734375
    },
    {
      "epoch": 3.4236907958984374e-07,
      "model_forward_time": 5.579931020736694,
      "step": 359
    },
    {
      "epoch": 3.4236907958984374e-07,
      "step": 359,
      "training_step_time": 16.98289179801941
    },
    {
      "epoch": 3.4332275390625e-07,
      "grad_norm": 1.9374604225158691,
      "learning_rate": 1e-05,
      "loss": 0.1293,
      "step": 360
    },
    {
      "epoch": 3.4332275390625e-07,
      "model_forward_time": 5.580068588256836,
      "step": 360
    },
    {
      "dynamics_loss_avg": 0.08070078007876873,
      "epoch": 3.4332275390625e-07,
      "step": 360
    },
    {
      "action_loss_avg": 0.04781445600092411,
      "epoch": 3.4332275390625e-07,
      "step": 360
    },
    {
      "epoch": 3.4332275390625e-07,
      "step": 360,
      "training_step_time": 17.00235152244568
    },
    {
      "epoch": 3.4427642822265624e-07,
      "model_forward_time": 5.584229469299316,
      "step": 361
    },
    {
      "epoch": 3.4427642822265624e-07,
      "step": 361,
      "training_step_time": 17.03398585319519
    },
    {
      "epoch": 3.452301025390625e-07,
      "model_forward_time": 5.586492538452148,
      "step": 362
    },
    {
      "epoch": 3.452301025390625e-07,
      "step": 362,
      "training_step_time": 17.031816005706787
    },
    {
      "epoch": 3.4618377685546874e-07,
      "model_forward_time": 5.955261945724487,
      "step": 363
    },
    {
      "epoch": 3.4618377685546874e-07,
      "step": 363,
      "training_step_time": 17.38978600502014
    },
    {
      "epoch": 3.47137451171875e-07,
      "model_forward_time": 5.597552061080933,
      "step": 364
    },
    {
      "epoch": 3.47137451171875e-07,
      "step": 364,
      "training_step_time": 17.066899299621582
    },
    {
      "epoch": 3.4809112548828124e-07,
      "model_forward_time": 5.609162330627441,
      "step": 365
    },
    {
      "epoch": 3.4809112548828124e-07,
      "step": 365,
      "training_step_time": 17.105030298233032
    },
    {
      "epoch": 3.490447998046875e-07,
      "model_forward_time": 5.616037368774414,
      "step": 366
    },
    {
      "epoch": 3.490447998046875e-07,
      "step": 366,
      "training_step_time": 17.07775902748108
    },
    {
      "epoch": 3.4999847412109374e-07,
      "model_forward_time": 5.581974983215332,
      "step": 367
    },
    {
      "epoch": 3.4999847412109374e-07,
      "step": 367,
      "training_step_time": 16.994128465652466
    },
    {
      "epoch": 3.509521484375e-07,
      "model_forward_time": 5.57740330696106,
      "step": 368
    },
    {
      "epoch": 3.509521484375e-07,
      "step": 368,
      "training_step_time": 17.37011432647705
    },
    {
      "epoch": 3.5190582275390624e-07,
      "model_forward_time": 5.584179401397705,
      "step": 369
    },
    {
      "epoch": 3.5190582275390624e-07,
      "step": 369,
      "training_step_time": 17.034438133239746
    },
    {
      "epoch": 3.528594970703125e-07,
      "grad_norm": 1.898307204246521,
      "learning_rate": 1e-05,
      "loss": 0.1139,
      "step": 370
    },
    {
      "epoch": 3.528594970703125e-07,
      "model_forward_time": 5.579282283782959,
      "step": 370
    },
    {
      "dynamics_loss_avg": 0.07612124048173427,
      "epoch": 3.528594970703125e-07,
      "step": 370
    },
    {
      "action_loss_avg": 0.040068901516497134,
      "epoch": 3.528594970703125e-07,
      "step": 370
    },
    {
      "epoch": 3.528594970703125e-07,
      "step": 370,
      "training_step_time": 16.98650097846985
    },
    {
      "epoch": 3.5381317138671875e-07,
      "model_forward_time": 5.583621501922607,
      "step": 371
    },
    {
      "epoch": 3.5381317138671875e-07,
      "step": 371,
      "training_step_time": 17.015478372573853
    },
    {
      "epoch": 3.54766845703125e-07,
      "model_forward_time": 5.578587055206299,
      "step": 372
    },
    {
      "epoch": 3.54766845703125e-07,
      "step": 372,
      "training_step_time": 17.01657795906067
    },
    {
      "epoch": 3.5572052001953125e-07,
      "model_forward_time": 5.59442138671875,
      "step": 373
    },
    {
      "epoch": 3.5572052001953125e-07,
      "step": 373,
      "training_step_time": 17.07735013961792
    },
    {
      "epoch": 3.566741943359375e-07,
      "model_forward_time": 5.992326974868774,
      "step": 374
    },
    {
      "epoch": 3.566741943359375e-07,
      "step": 374,
      "training_step_time": 17.51848864555359
    },
    {
      "epoch": 3.5762786865234375e-07,
      "model_forward_time": 5.625614404678345,
      "step": 375
    },
    {
      "epoch": 3.5762786865234375e-07,
      "step": 375,
      "training_step_time": 17.093932628631592
    },
    {
      "epoch": 3.5858154296875e-07,
      "model_forward_time": 5.617163181304932,
      "step": 376
    },
    {
      "epoch": 3.5858154296875e-07,
      "step": 376,
      "training_step_time": 17.1053729057312
    },
    {
      "epoch": 3.5953521728515625e-07,
      "model_forward_time": 5.610530853271484,
      "step": 377
    },
    {
      "epoch": 3.5953521728515625e-07,
      "step": 377,
      "training_step_time": 17.109931707382202
    },
    {
      "epoch": 3.604888916015625e-07,
      "model_forward_time": 5.6054956912994385,
      "step": 378
    },
    {
      "epoch": 3.604888916015625e-07,
      "step": 378,
      "training_step_time": 17.070064783096313
    },
    {
      "epoch": 3.6144256591796875e-07,
      "model_forward_time": 5.9406657218933105,
      "step": 379
    },
    {
      "epoch": 3.6144256591796875e-07,
      "step": 379,
      "training_step_time": 17.41495108604431
    },
    {
      "epoch": 3.62396240234375e-07,
      "grad_norm": 2.173955202102661,
      "learning_rate": 1e-05,
      "loss": 0.1146,
      "step": 380
    },
    {
      "epoch": 3.62396240234375e-07,
      "model_forward_time": 5.595640659332275,
      "step": 380
    },
    {
      "dynamics_loss_avg": 0.07192523293197155,
      "epoch": 3.62396240234375e-07,
      "step": 380
    },
    {
      "action_loss_avg": 0.040517967008054254,
      "epoch": 3.62396240234375e-07,
      "step": 380
    },
    {
      "epoch": 3.62396240234375e-07,
      "step": 380,
      "training_step_time": 17.039990663528442
    },
    {
      "epoch": 3.6334991455078126e-07,
      "model_forward_time": 5.60043478012085,
      "step": 381
    },
    {
      "epoch": 3.6334991455078126e-07,
      "step": 381,
      "training_step_time": 17.036750555038452
    },
    {
      "epoch": 3.643035888671875e-07,
      "model_forward_time": 5.580901384353638,
      "step": 382
    },
    {
      "epoch": 3.643035888671875e-07,
      "step": 382,
      "training_step_time": 17.018833875656128
    },
    {
      "epoch": 3.6525726318359376e-07,
      "model_forward_time": 5.59175705909729,
      "step": 383
    },
    {
      "epoch": 3.6525726318359376e-07,
      "step": 383,
      "training_step_time": 17.05478310585022
    },
    {
      "epoch": 3.662109375e-07,
      "model_forward_time": 5.595940351486206,
      "step": 384
    },
    {
      "epoch": 3.662109375e-07,
      "step": 384,
      "training_step_time": 17.053250074386597
    },
    {
      "epoch": 3.6716461181640626e-07,
      "model_forward_time": 5.967647552490234,
      "step": 385
    },
    {
      "epoch": 3.6716461181640626e-07,
      "step": 385,
      "training_step_time": 17.421070098876953
    },
    {
      "epoch": 3.681182861328125e-07,
      "model_forward_time": 5.588942289352417,
      "step": 386
    },
    {
      "epoch": 3.681182861328125e-07,
      "step": 386,
      "training_step_time": 17.010396718978882
    },
    {
      "epoch": 3.6907196044921876e-07,
      "model_forward_time": 5.593878746032715,
      "step": 387
    },
    {
      "epoch": 3.6907196044921876e-07,
      "step": 387,
      "training_step_time": 17.029677629470825
    },
    {
      "epoch": 3.70025634765625e-07,
      "model_forward_time": 5.582335948944092,
      "step": 388
    },
    {
      "epoch": 3.70025634765625e-07,
      "step": 388,
      "training_step_time": 17.03785538673401
    },
    {
      "epoch": 3.7097930908203126e-07,
      "model_forward_time": 5.596714496612549,
      "step": 389
    },
    {
      "epoch": 3.7097930908203126e-07,
      "step": 389,
      "training_step_time": 17.064443349838257
    },
    {
      "epoch": 3.719329833984375e-07,
      "grad_norm": 2.6108596324920654,
      "learning_rate": 1e-05,
      "loss": 0.1364,
      "step": 390
    },
    {
      "epoch": 3.719329833984375e-07,
      "model_forward_time": 5.947711944580078,
      "step": 390
    },
    {
      "dynamics_loss_avg": 0.08299174793064594,
      "epoch": 3.719329833984375e-07,
      "step": 390
    },
    {
      "action_loss_avg": 0.048725799471139905,
      "epoch": 3.719329833984375e-07,
      "step": 390
    },
    {
      "epoch": 3.719329833984375e-07,
      "step": 390,
      "training_step_time": 17.39935564994812
    },
    {
      "epoch": 3.7288665771484376e-07,
      "model_forward_time": 5.596961736679077,
      "step": 391
    },
    {
      "epoch": 3.7288665771484376e-07,
      "step": 391,
      "training_step_time": 17.033881902694702
    },
    {
      "epoch": 3.7384033203125e-07,
      "model_forward_time": 5.599355697631836,
      "step": 392
    },
    {
      "epoch": 3.7384033203125e-07,
      "step": 392,
      "training_step_time": 17.071845769882202
    },
    {
      "epoch": 3.7479400634765627e-07,
      "model_forward_time": 5.60984468460083,
      "step": 393
    },
    {
      "epoch": 3.7479400634765627e-07,
      "step": 393,
      "training_step_time": 17.089698791503906
    },
    {
      "epoch": 3.757476806640625e-07,
      "model_forward_time": 5.612102746963501,
      "step": 394
    },
    {
      "epoch": 3.757476806640625e-07,
      "step": 394,
      "training_step_time": 17.09540891647339
    },
    {
      "epoch": 3.7670135498046877e-07,
      "model_forward_time": 5.613178253173828,
      "step": 395
    },
    {
      "epoch": 3.7670135498046877e-07,
      "step": 395,
      "training_step_time": 17.092180252075195
    },
    {
      "epoch": 3.77655029296875e-07,
      "model_forward_time": 5.97557520866394,
      "step": 396
    },
    {
      "epoch": 3.77655029296875e-07,
      "step": 396,
      "training_step_time": 17.458683490753174
    },
    {
      "epoch": 3.7860870361328127e-07,
      "model_forward_time": 5.610397815704346,
      "step": 397
    },
    {
      "epoch": 3.7860870361328127e-07,
      "step": 397,
      "training_step_time": 17.102257013320923
    },
    {
      "epoch": 3.795623779296875e-07,
      "model_forward_time": 5.6086976528167725,
      "step": 398
    },
    {
      "epoch": 3.795623779296875e-07,
      "step": 398,
      "training_step_time": 17.099091291427612
    },
    {
      "epoch": 3.8051605224609377e-07,
      "model_forward_time": 5.605505466461182,
      "step": 399
    },
    {
      "epoch": 3.8051605224609377e-07,
      "step": 399,
      "training_step_time": 17.08067226409912
    },
    {
      "epoch": 3.814697265625e-07,
      "grad_norm": 0.7716083526611328,
      "learning_rate": 1e-05,
      "loss": 0.1113,
      "step": 400
    },
    {
      "epoch": 3.814697265625e-07,
      "model_forward_time": 5.596590518951416,
      "step": 400
    },
    {
      "dynamics_loss_avg": 0.06847628802061081,
      "epoch": 3.814697265625e-07,
      "step": 400
    },
    {
      "action_loss_avg": 0.04067210694774985,
      "epoch": 3.814697265625e-07,
      "step": 400
    },
    {
      "epoch": 3.814697265625e-07,
      "step": 400,
      "training_step_time": 17.05895233154297
    },
    {
      "epoch": 3.8242340087890627e-07,
      "model_forward_time": 5.957905530929565,
      "step": 401
    },
    {
      "epoch": 3.8242340087890627e-07,
      "step": 401,
      "training_step_time": 17.406567811965942
    },
    {
      "epoch": 3.833770751953125e-07,
      "model_forward_time": 5.595866918563843,
      "step": 402
    },
    {
      "epoch": 3.833770751953125e-07,
      "step": 402,
      "training_step_time": 17.047045469284058
    },
    {
      "epoch": 3.843307495117188e-07,
      "model_forward_time": 5.608618497848511,
      "step": 403
    },
    {
      "epoch": 3.843307495117188e-07,
      "step": 403,
      "training_step_time": 17.088382959365845
    },
    {
      "epoch": 3.85284423828125e-07,
      "model_forward_time": 5.6044182777404785,
      "step": 404
    },
    {
      "epoch": 3.85284423828125e-07,
      "step": 404,
      "training_step_time": 17.09802007675171
    },
    {
      "epoch": 3.862380981445313e-07,
      "model_forward_time": 5.602186679840088,
      "step": 405
    },
    {
      "epoch": 3.862380981445313e-07,
      "step": 405,
      "training_step_time": 17.0929696559906
    },
    {
      "epoch": 3.871917724609375e-07,
      "model_forward_time": 5.623464822769165,
      "step": 406
    },
    {
      "epoch": 3.871917724609375e-07,
      "step": 406,
      "training_step_time": 17.469547271728516
    },
    {
      "epoch": 3.881454467773437e-07,
      "model_forward_time": 5.616232633590698,
      "step": 407
    },
    {
      "epoch": 3.881454467773437e-07,
      "step": 407,
      "training_step_time": 17.1007821559906
    },
    {
      "epoch": 3.8909912109375e-07,
      "model_forward_time": 5.619393587112427,
      "step": 408
    },
    {
      "epoch": 3.8909912109375e-07,
      "step": 408,
      "training_step_time": 17.10140323638916
    },
    {
      "epoch": 3.900527954101562e-07,
      "model_forward_time": 5.6009745597839355,
      "step": 409
    },
    {
      "epoch": 3.900527954101562e-07,
      "step": 409,
      "training_step_time": 17.053142309188843
    },
    {
      "epoch": 3.910064697265625e-07,
      "grad_norm": 2.5430407524108887,
      "learning_rate": 1e-05,
      "loss": 0.1134,
      "step": 410
    },
    {
      "epoch": 3.910064697265625e-07,
      "model_forward_time": 5.600423097610474,
      "step": 410
    },
    {
      "dynamics_loss_avg": 0.07706639189273119,
      "epoch": 3.910064697265625e-07,
      "step": 410
    },
    {
      "action_loss_avg": 0.04316465351730585,
      "epoch": 3.910064697265625e-07,
      "step": 410
    },
    {
      "epoch": 3.910064697265625e-07,
      "step": 410,
      "training_step_time": 17.038384437561035
    },
    {
      "epoch": 3.9196014404296873e-07,
      "model_forward_time": 5.606509447097778,
      "step": 411
    },
    {
      "epoch": 3.9196014404296873e-07,
      "step": 411,
      "training_step_time": 17.08405327796936
    },
    {
      "epoch": 3.92913818359375e-07,
      "model_forward_time": 5.942118167877197,
      "step": 412
    },
    {
      "epoch": 3.92913818359375e-07,
      "step": 412,
      "training_step_time": 17.400113821029663
    },
    {
      "epoch": 3.9386749267578123e-07,
      "model_forward_time": 5.5890748500823975,
      "step": 413
    },
    {
      "epoch": 3.9386749267578123e-07,
      "step": 413,
      "training_step_time": 17.01620101928711
    },
    {
      "epoch": 3.948211669921875e-07,
      "model_forward_time": 5.596976280212402,
      "step": 414
    },
    {
      "epoch": 3.948211669921875e-07,
      "step": 414,
      "training_step_time": 17.024485111236572
    },
    {
      "epoch": 3.9577484130859373e-07,
      "model_forward_time": 5.593471050262451,
      "step": 415
    },
    {
      "epoch": 3.9577484130859373e-07,
      "step": 415,
      "training_step_time": 17.047276973724365
    },
    {
      "epoch": 3.96728515625e-07,
      "model_forward_time": 5.6116392612457275,
      "step": 416
    },
    {
      "epoch": 3.96728515625e-07,
      "step": 416,
      "training_step_time": 17.103204250335693
    },
    {
      "epoch": 3.9768218994140623e-07,
      "model_forward_time": 5.995280027389526,
      "step": 417
    },
    {
      "epoch": 3.9768218994140623e-07,
      "step": 417,
      "training_step_time": 17.489434480667114
    },
    {
      "epoch": 3.986358642578125e-07,
      "model_forward_time": 5.6052985191345215,
      "step": 418
    },
    {
      "epoch": 3.986358642578125e-07,
      "step": 418,
      "training_step_time": 17.030910968780518
    },
    {
      "epoch": 3.9958953857421873e-07,
      "model_forward_time": 5.587802410125732,
      "step": 419
    },
    {
      "epoch": 3.9958953857421873e-07,
      "step": 419,
      "training_step_time": 16.999663591384888
    },
    {
      "epoch": 4.00543212890625e-07,
      "grad_norm": 1.1138719320297241,
      "learning_rate": 1e-05,
      "loss": 0.118,
      "step": 420
    },
    {
      "epoch": 4.00543212890625e-07,
      "model_forward_time": 5.584768533706665,
      "step": 420
    },
    {
      "dynamics_loss_avg": 0.07117796652019023,
      "epoch": 4.00543212890625e-07,
      "step": 420
    },
    {
      "action_loss_avg": 0.04234424270689487,
      "epoch": 4.00543212890625e-07,
      "step": 420
    },
    {
      "epoch": 4.00543212890625e-07,
      "step": 420,
      "training_step_time": 17.007360219955444
    },
    {
      "epoch": 4.0149688720703124e-07,
      "model_forward_time": 5.57953667640686,
      "step": 421
    },
    {
      "epoch": 4.0149688720703124e-07,
      "step": 421,
      "training_step_time": 17.01552677154541
    },
    {
      "epoch": 4.024505615234375e-07,
      "model_forward_time": 5.570747137069702,
      "step": 422
    },
    {
      "epoch": 4.024505615234375e-07,
      "step": 422,
      "training_step_time": 16.997902154922485
    },
    {
      "epoch": 4.0340423583984374e-07,
      "model_forward_time": 5.939239263534546,
      "step": 423
    },
    {
      "epoch": 4.0340423583984374e-07,
      "step": 423,
      "training_step_time": 17.342153310775757
    },
    {
      "epoch": 4.0435791015625e-07,
      "model_forward_time": 5.5931150913238525,
      "step": 424
    },
    {
      "epoch": 4.0435791015625e-07,
      "step": 424,
      "training_step_time": 17.043622255325317
    },
    {
      "epoch": 4.0531158447265624e-07,
      "model_forward_time": 5.608745336532593,
      "step": 425
    },
    {
      "epoch": 4.0531158447265624e-07,
      "step": 425,
      "training_step_time": 17.096710681915283
    },
    {
      "epoch": 4.062652587890625e-07,
      "model_forward_time": 5.633427619934082,
      "step": 426
    },
    {
      "epoch": 4.062652587890625e-07,
      "step": 426,
      "training_step_time": 17.152858018875122
    },
    {
      "epoch": 4.0721893310546874e-07,
      "model_forward_time": 5.624749422073364,
      "step": 427
    },
    {
      "epoch": 4.0721893310546874e-07,
      "step": 427,
      "training_step_time": 17.10771346092224
    },
    {
      "epoch": 4.08172607421875e-07,
      "model_forward_time": 5.962222099304199,
      "step": 428
    },
    {
      "epoch": 4.08172607421875e-07,
      "step": 428,
      "training_step_time": 17.37659788131714
    },
    {
      "epoch": 4.0912628173828124e-07,
      "model_forward_time": 5.568232774734497,
      "step": 429
    },
    {
      "epoch": 4.0912628173828124e-07,
      "step": 429,
      "training_step_time": 17.00497531890869
    },
    {
      "epoch": 4.100799560546875e-07,
      "grad_norm": 2.271038055419922,
      "learning_rate": 1e-05,
      "loss": 0.1016,
      "step": 430
    },
    {
      "epoch": 4.100799560546875e-07,
      "model_forward_time": 5.569753646850586,
      "step": 430
    },
    {
      "dynamics_loss_avg": 0.06523771956562996,
      "epoch": 4.100799560546875e-07,
      "step": 430
    },
    {
      "action_loss_avg": 0.03400124562904239,
      "epoch": 4.100799560546875e-07,
      "step": 430
    },
    {
      "epoch": 4.100799560546875e-07,
      "step": 430,
      "training_step_time": 16.981454849243164
    },
    {
      "epoch": 4.1103363037109374e-07,
      "model_forward_time": 5.570985794067383,
      "step": 431
    },
    {
      "epoch": 4.1103363037109374e-07,
      "step": 431,
      "training_step_time": 16.98231291770935
    },
    {
      "epoch": 4.119873046875e-07,
      "model_forward_time": 5.582546949386597,
      "step": 432
    },
    {
      "epoch": 4.119873046875e-07,
      "step": 432,
      "training_step_time": 16.996522665023804
    },
    {
      "epoch": 4.1294097900390625e-07,
      "model_forward_time": 5.57907247543335,
      "step": 433
    },
    {
      "epoch": 4.1294097900390625e-07,
      "step": 433,
      "training_step_time": 17.010034799575806
    },
    {
      "epoch": 4.138946533203125e-07,
      "model_forward_time": 5.92802357673645,
      "step": 434
    },
    {
      "epoch": 4.138946533203125e-07,
      "step": 434,
      "training_step_time": 17.40307331085205
    },
    {
      "epoch": 4.1484832763671875e-07,
      "model_forward_time": 5.610742807388306,
      "step": 435
    },
    {
      "epoch": 4.1484832763671875e-07,
      "step": 435,
      "training_step_time": 17.128278255462646
    },
    {
      "epoch": 4.15802001953125e-07,
      "model_forward_time": 5.635641813278198,
      "step": 436
    },
    {
      "epoch": 4.15802001953125e-07,
      "step": 436,
      "training_step_time": 17.1326904296875
    },
    {
      "epoch": 4.1675567626953125e-07,
      "model_forward_time": 5.610973358154297,
      "step": 437
    },
    {
      "epoch": 4.1675567626953125e-07,
      "step": 437,
      "training_step_time": 17.064094066619873
    },
    {
      "epoch": 4.177093505859375e-07,
      "model_forward_time": 5.603577375411987,
      "step": 438
    },
    {
      "epoch": 4.177093505859375e-07,
      "step": 438,
      "training_step_time": 17.079550743103027
    },
    {
      "epoch": 4.1866302490234375e-07,
      "model_forward_time": 5.995667219161987,
      "step": 439
    },
    {
      "epoch": 4.1866302490234375e-07,
      "step": 439,
      "training_step_time": 17.475492238998413
    },
    {
      "epoch": 4.1961669921875e-07,
      "grad_norm": 2.778494358062744,
      "learning_rate": 1e-05,
      "loss": 0.102,
      "step": 440
    },
    {
      "epoch": 4.1961669921875e-07,
      "model_forward_time": 5.606098175048828,
      "step": 440
    },
    {
      "dynamics_loss_avg": 0.07179645523428917,
      "epoch": 4.1961669921875e-07,
      "step": 440
    },
    {
      "action_loss_avg": 0.04132894007489085,
      "epoch": 4.1961669921875e-07,
      "step": 440
    },
    {
      "epoch": 4.1961669921875e-07,
      "step": 440,
      "training_step_time": 17.093462467193604
    },
    {
      "epoch": 4.2057037353515625e-07,
      "model_forward_time": 5.616779088973999,
      "step": 441
    },
    {
      "epoch": 4.2057037353515625e-07,
      "step": 441,
      "training_step_time": 17.107365131378174
    },
    {
      "epoch": 4.215240478515625e-07,
      "model_forward_time": 5.5975728034973145,
      "step": 442
    },
    {
      "epoch": 4.215240478515625e-07,
      "step": 442,
      "training_step_time": 17.06166982650757
    },
    {
      "epoch": 4.2247772216796875e-07,
      "model_forward_time": 5.5918190479278564,
      "step": 443
    },
    {
      "epoch": 4.2247772216796875e-07,
      "step": 443,
      "training_step_time": 17.023868799209595
    },
    {
      "epoch": 4.23431396484375e-07,
      "model_forward_time": 5.58853006362915,
      "step": 444
    },
    {
      "epoch": 4.23431396484375e-07,
      "step": 444,
      "training_step_time": 17.024528741836548
    },
    {
      "epoch": 4.2438507080078126e-07,
      "model_forward_time": 5.966403484344482,
      "step": 445
    },
    {
      "epoch": 4.2438507080078126e-07,
      "step": 445,
      "training_step_time": 17.409137964248657
    },
    {
      "epoch": 4.253387451171875e-07,
      "model_forward_time": 5.595148086547852,
      "step": 446
    },
    {
      "epoch": 4.253387451171875e-07,
      "step": 446,
      "training_step_time": 17.0730619430542
    },
    {
      "epoch": 4.2629241943359376e-07,
      "model_forward_time": 5.616210460662842,
      "step": 447
    },
    {
      "epoch": 4.2629241943359376e-07,
      "step": 447,
      "training_step_time": 17.121087551116943
    },
    {
      "epoch": 4.2724609375e-07,
      "model_forward_time": 5.621835708618164,
      "step": 448
    },
    {
      "epoch": 4.2724609375e-07,
      "step": 448,
      "training_step_time": 17.144612073898315
    },
    {
      "epoch": 4.2819976806640626e-07,
      "model_forward_time": 5.626211404800415,
      "step": 449
    },
    {
      "epoch": 4.2819976806640626e-07,
      "step": 449,
      "training_step_time": 17.13956117630005
    },
    {
      "epoch": 4.291534423828125e-07,
      "grad_norm": 1.9024603366851807,
      "learning_rate": 1e-05,
      "loss": 0.1075,
      "step": 450
    },
    {
      "epoch": 4.291534423828125e-07,
      "model_forward_time": 5.979674816131592,
      "step": 450
    },
    {
      "dynamics_loss_avg": 0.06935918480157852,
      "epoch": 4.291534423828125e-07,
      "step": 450
    },
    {
      "action_loss_avg": 0.032748748734593394,
      "epoch": 4.291534423828125e-07,
      "step": 450
    },
    {
      "epoch": 4.291534423828125e-07,
      "step": 450,
      "training_step_time": 17.483304977416992
    },
    {
      "epoch": 4.3010711669921876e-07,
      "model_forward_time": 5.6270716190338135,
      "step": 451
    },
    {
      "epoch": 4.3010711669921876e-07,
      "step": 451,
      "training_step_time": 17.102686882019043
    },
    {
      "epoch": 4.31060791015625e-07,
      "model_forward_time": 5.600527286529541,
      "step": 452
    },
    {
      "epoch": 4.31060791015625e-07,
      "step": 452,
      "training_step_time": 17.034569263458252
    },
    {
      "epoch": 4.3201446533203126e-07,
      "model_forward_time": 5.5798704624176025,
      "step": 453
    },
    {
      "epoch": 4.3201446533203126e-07,
      "step": 453,
      "training_step_time": 16.985607385635376
    },
    {
      "epoch": 4.329681396484375e-07,
      "model_forward_time": 5.577766418457031,
      "step": 454
    },
    {
      "epoch": 4.329681396484375e-07,
      "step": 454,
      "training_step_time": 16.98786997795105
    },
    {
      "epoch": 4.3392181396484376e-07,
      "model_forward_time": 5.9290502071380615,
      "step": 455
    },
    {
      "epoch": 4.3392181396484376e-07,
      "step": 455,
      "training_step_time": 17.353118658065796
    },
    {
      "epoch": 4.3487548828125e-07,
      "model_forward_time": 5.582648038864136,
      "step": 456
    },
    {
      "epoch": 4.3487548828125e-07,
      "step": 456,
      "training_step_time": 17.028616428375244
    },
    {
      "epoch": 4.3582916259765627e-07,
      "model_forward_time": 5.5851311683654785,
      "step": 457
    },
    {
      "epoch": 4.3582916259765627e-07,
      "step": 457,
      "training_step_time": 16.998619556427002
    },
    {
      "epoch": 4.367828369140625e-07,
      "model_forward_time": 5.583526372909546,
      "step": 458
    },
    {
      "epoch": 4.367828369140625e-07,
      "step": 458,
      "training_step_time": 17.015312910079956
    },
    {
      "epoch": 4.3773651123046877e-07,
      "model_forward_time": 5.596277952194214,
      "step": 459
    },
    {
      "epoch": 4.3773651123046877e-07,
      "step": 459,
      "training_step_time": 17.052536010742188
    },
    {
      "epoch": 4.38690185546875e-07,
      "grad_norm": 1.4424549341201782,
      "learning_rate": 1e-05,
      "loss": 0.12,
      "step": 460
    },
    {
      "epoch": 4.38690185546875e-07,
      "model_forward_time": 5.61637806892395,
      "step": 460
    },
    {
      "dynamics_loss_avg": 0.07858547903597354,
      "epoch": 4.38690185546875e-07,
      "step": 460
    },
    {
      "action_loss_avg": 0.043759002536535266,
      "epoch": 4.38690185546875e-07,
      "step": 460
    },
    {
      "epoch": 4.38690185546875e-07,
      "step": 460,
      "training_step_time": 17.09530234336853
    },
    {
      "epoch": 4.3964385986328127e-07,
      "model_forward_time": 5.985904693603516,
      "step": 461
    },
    {
      "epoch": 4.3964385986328127e-07,
      "step": 461,
      "training_step_time": 17.483134508132935
    },
    {
      "epoch": 4.405975341796875e-07,
      "model_forward_time": 5.617387533187866,
      "step": 462
    },
    {
      "epoch": 4.405975341796875e-07,
      "step": 462,
      "training_step_time": 17.09356713294983
    },
    {
      "epoch": 4.4155120849609377e-07,
      "model_forward_time": 5.597471237182617,
      "step": 463
    },
    {
      "epoch": 4.4155120849609377e-07,
      "step": 463,
      "training_step_time": 17.039196014404297
    },
    {
      "epoch": 4.425048828125e-07,
      "model_forward_time": 5.5840370655059814,
      "step": 464
    },
    {
      "epoch": 4.425048828125e-07,
      "step": 464,
      "training_step_time": 17.003960371017456
    },
    {
      "epoch": 4.4345855712890627e-07,
      "model_forward_time": 5.593360900878906,
      "step": 465
    },
    {
      "epoch": 4.4345855712890627e-07,
      "step": 465,
      "training_step_time": 17.02178192138672
    },
    {
      "epoch": 4.444122314453125e-07,
      "model_forward_time": 5.598447799682617,
      "step": 466
    },
    {
      "epoch": 4.444122314453125e-07,
      "step": 466,
      "training_step_time": 17.409537315368652
    },
    {
      "epoch": 4.453659057617188e-07,
      "model_forward_time": 5.607002258300781,
      "step": 467
    },
    {
      "epoch": 4.453659057617188e-07,
      "step": 467,
      "training_step_time": 17.06959581375122
    },
    {
      "epoch": 4.46319580078125e-07,
      "model_forward_time": 5.63431715965271,
      "step": 468
    },
    {
      "epoch": 4.46319580078125e-07,
      "step": 468,
      "training_step_time": 17.08950972557068
    },
    {
      "epoch": 4.472732543945312e-07,
      "model_forward_time": 5.592742919921875,
      "step": 469
    },
    {
      "epoch": 4.472732543945312e-07,
      "step": 469,
      "training_step_time": 17.042443990707397
    },
    {
      "epoch": 4.482269287109375e-07,
      "grad_norm": 0.8911119699478149,
      "learning_rate": 1e-05,
      "loss": 0.1282,
      "step": 470
    },
    {
      "epoch": 4.482269287109375e-07,
      "model_forward_time": 5.603128671646118,
      "step": 470
    },
    {
      "dynamics_loss_avg": 0.09006702899932861,
      "epoch": 4.482269287109375e-07,
      "step": 470
    },
    {
      "action_loss_avg": 0.04324385654181242,
      "epoch": 4.482269287109375e-07,
      "step": 470
    },
    {
      "epoch": 4.482269287109375e-07,
      "step": 470,
      "training_step_time": 17.082709074020386
    },
    {
      "epoch": 4.491806030273437e-07,
      "model_forward_time": 5.610854864120483,
      "step": 471
    },
    {
      "epoch": 4.491806030273437e-07,
      "step": 471,
      "training_step_time": 17.107661724090576
    },
    {
      "epoch": 4.5013427734375e-07,
      "model_forward_time": 6.007529258728027,
      "step": 472
    },
    {
      "epoch": 4.5013427734375e-07,
      "step": 472,
      "training_step_time": 17.508260250091553
    },
    {
      "epoch": 4.5108795166015623e-07,
      "model_forward_time": 5.62762188911438,
      "step": 473
    },
    {
      "epoch": 4.5108795166015623e-07,
      "step": 473,
      "training_step_time": 17.137087106704712
    },
    {
      "epoch": 4.520416259765625e-07,
      "model_forward_time": 5.625940561294556,
      "step": 474
    },
    {
      "epoch": 4.520416259765625e-07,
      "step": 474,
      "training_step_time": 17.130455017089844
    },
    {
      "epoch": 4.5299530029296873e-07,
      "model_forward_time": 5.622917413711548,
      "step": 475
    },
    {
      "epoch": 4.5299530029296873e-07,
      "step": 475,
      "training_step_time": 17.118232250213623
    },
    {
      "epoch": 4.53948974609375e-07,
      "model_forward_time": 5.621124982833862,
      "step": 476
    },
    {
      "epoch": 4.53948974609375e-07,
      "step": 476,
      "training_step_time": 17.097469091415405
    },
    {
      "epoch": 4.5490264892578123e-07,
      "model_forward_time": 5.5968852043151855,
      "step": 477
    },
    {
      "epoch": 4.5490264892578123e-07,
      "step": 477,
      "training_step_time": 17.364100217819214
    },
    {
      "epoch": 4.558563232421875e-07,
      "model_forward_time": 5.572577953338623,
      "step": 478
    },
    {
      "epoch": 4.558563232421875e-07,
      "step": 478,
      "training_step_time": 17.013031721115112
    },
    {
      "epoch": 4.5680999755859373e-07,
      "model_forward_time": 5.568183422088623,
      "step": 479
    },
    {
      "epoch": 4.5680999755859373e-07,
      "step": 479,
      "training_step_time": 16.986241817474365
    },
    {
      "epoch": 4.57763671875e-07,
      "grad_norm": 1.775735855102539,
      "learning_rate": 1e-05,
      "loss": 0.1184,
      "step": 480
    },
    {
      "epoch": 4.57763671875e-07,
      "model_forward_time": 5.5644447803497314,
      "step": 480
    },
    {
      "dynamics_loss_avg": 0.07200150527060031,
      "epoch": 4.57763671875e-07,
      "step": 480
    },
    {
      "action_loss_avg": 0.040643921867012975,
      "epoch": 4.57763671875e-07,
      "step": 480
    },
    {
      "epoch": 4.57763671875e-07,
      "step": 480,
      "training_step_time": 16.97177028656006
    },
    {
      "epoch": 4.5871734619140623e-07,
      "model_forward_time": 5.569270133972168,
      "step": 481
    },
    {
      "epoch": 4.5871734619140623e-07,
      "step": 481,
      "training_step_time": 16.98740816116333
    },
    {
      "epoch": 4.596710205078125e-07,
      "model_forward_time": 5.57281494140625,
      "step": 482
    },
    {
      "epoch": 4.596710205078125e-07,
      "step": 482,
      "training_step_time": 17.002669095993042
    },
    {
      "epoch": 4.6062469482421874e-07,
      "model_forward_time": 5.949306488037109,
      "step": 483
    },
    {
      "epoch": 4.6062469482421874e-07,
      "step": 483,
      "training_step_time": 17.37355923652649
    },
    {
      "epoch": 4.61578369140625e-07,
      "model_forward_time": 5.593043565750122,
      "step": 484
    },
    {
      "epoch": 4.61578369140625e-07,
      "step": 484,
      "training_step_time": 17.054414749145508
    },
    {
      "epoch": 4.6253204345703124e-07,
      "model_forward_time": 5.597353935241699,
      "step": 485
    },
    {
      "epoch": 4.6253204345703124e-07,
      "step": 485,
      "training_step_time": 17.066559076309204
    },
    {
      "epoch": 4.634857177734375e-07,
      "model_forward_time": 5.604955673217773,
      "step": 486
    },
    {
      "epoch": 4.634857177734375e-07,
      "step": 486,
      "training_step_time": 17.087488651275635
    },
    {
      "epoch": 4.6443939208984374e-07,
      "model_forward_time": 5.604806661605835,
      "step": 487
    },
    {
      "epoch": 4.6443939208984374e-07,
      "step": 487,
      "training_step_time": 17.046701431274414
    },
    {
      "epoch": 4.6539306640625e-07,
      "model_forward_time": 5.936918497085571,
      "step": 488
    },
    {
      "epoch": 4.6539306640625e-07,
      "step": 488,
      "training_step_time": 17.370556354522705
    },
    {
      "epoch": 4.6634674072265624e-07,
      "model_forward_time": 5.596365451812744,
      "step": 489
    },
    {
      "epoch": 4.6634674072265624e-07,
      "step": 489,
      "training_step_time": 17.057937145233154
    },
    {
      "epoch": 4.673004150390625e-07,
      "grad_norm": 1.6574381589889526,
      "learning_rate": 1e-05,
      "loss": 0.1275,
      "step": 490
    },
    {
      "epoch": 4.673004150390625e-07,
      "model_forward_time": 5.603854417800903,
      "step": 490
    },
    {
      "dynamics_loss_avg": 0.08145049177110195,
      "epoch": 4.673004150390625e-07,
      "step": 490
    },
    {
      "action_loss_avg": 0.04124371279031038,
      "epoch": 4.673004150390625e-07,
      "step": 490
    },
    {
      "epoch": 4.673004150390625e-07,
      "step": 490,
      "training_step_time": 17.06993842124939
    },
    {
      "epoch": 4.6825408935546874e-07,
      "model_forward_time": 5.604434013366699,
      "step": 491
    },
    {
      "epoch": 4.6825408935546874e-07,
      "step": 491,
      "training_step_time": 17.08207869529724
    },
    {
      "epoch": 4.69207763671875e-07,
      "model_forward_time": 5.6061692237854,
      "step": 492
    },
    {
      "epoch": 4.69207763671875e-07,
      "step": 492,
      "training_step_time": 17.10389518737793
    },
    {
      "epoch": 4.7016143798828124e-07,
      "model_forward_time": 5.617032051086426,
      "step": 493
    },
    {
      "epoch": 4.7016143798828124e-07,
      "step": 493,
      "training_step_time": 17.104538917541504
    },
    {
      "epoch": 4.711151123046875e-07,
      "model_forward_time": 5.980209827423096,
      "step": 494
    },
    {
      "epoch": 4.711151123046875e-07,
      "step": 494,
      "training_step_time": 17.44815158843994
    },
    {
      "epoch": 4.7206878662109375e-07,
      "model_forward_time": 5.588275909423828,
      "step": 495
    },
    {
      "epoch": 4.7206878662109375e-07,
      "step": 495,
      "training_step_time": 17.00178074836731
    },
    {
      "epoch": 4.730224609375e-07,
      "model_forward_time": 5.566215991973877,
      "step": 496
    },
    {
      "epoch": 4.730224609375e-07,
      "step": 496,
      "training_step_time": 16.976256132125854
    },
    {
      "epoch": 4.7397613525390625e-07,
      "model_forward_time": 5.581497669219971,
      "step": 497
    },
    {
      "epoch": 4.7397613525390625e-07,
      "step": 497,
      "training_step_time": 17.017110109329224
    },
    {
      "epoch": 4.749298095703125e-07,
      "model_forward_time": 5.571829080581665,
      "step": 498
    },
    {
      "epoch": 4.749298095703125e-07,
      "step": 498,
      "training_step_time": 16.992891788482666
    },
    {
      "epoch": 4.7588348388671875e-07,
      "model_forward_time": 5.580824136734009,
      "step": 499
    },
    {
      "epoch": 4.7588348388671875e-07,
      "step": 499,
      "training_step_time": 17.35400414466858
    },
    {
      "epoch": 4.76837158203125e-07,
      "grad_norm": 1.9924147129058838,
      "learning_rate": 1e-05,
      "loss": 0.1147,
      "step": 500
    },
    {
      "epoch": 4.76837158203125e-07,
      "model_forward_time": 5.5802903175354,
      "step": 500
    },
    {
      "dynamics_loss_avg": 0.07657814845442772,
      "epoch": 4.76837158203125e-07,
      "step": 500
    },
    {
      "action_loss_avg": 0.04472140967845917,
      "epoch": 4.76837158203125e-07,
      "step": 500
    },
    {
      "epoch": 4.76837158203125e-07,
      "step": 500,
      "training_step_time": 16.987775802612305
    },
    {
      "epoch": 4.777908325195313e-07,
      "model_forward_time": 5.585555553436279,
      "step": 501
    },
    {
      "epoch": 4.777908325195313e-07,
      "step": 501,
      "training_step_time": 17.026606798171997
    },
    {
      "epoch": 4.787445068359375e-07,
      "model_forward_time": 5.597659587860107,
      "step": 502
    },
    {
      "epoch": 4.787445068359375e-07,
      "step": 502,
      "training_step_time": 17.083839416503906
    },
    {
      "epoch": 4.796981811523438e-07,
      "model_forward_time": 5.609550476074219,
      "step": 503
    },
    {
      "epoch": 4.796981811523438e-07,
      "step": 503,
      "training_step_time": 17.12269139289856
    },
    {
      "epoch": 4.8065185546875e-07,
      "model_forward_time": 5.621304750442505,
      "step": 504
    },
    {
      "epoch": 4.8065185546875e-07,
      "step": 504,
      "training_step_time": 17.126453161239624
    },
    {
      "epoch": 4.816055297851563e-07,
      "model_forward_time": 5.984508752822876,
      "step": 505
    },
    {
      "epoch": 4.816055297851563e-07,
      "step": 505,
      "training_step_time": 17.456693410873413
    },
    {
      "epoch": 4.825592041015625e-07,
      "model_forward_time": 5.598014831542969,
      "step": 506
    },
    {
      "epoch": 4.825592041015625e-07,
      "step": 506,
      "training_step_time": 17.00552010536194
    },
    {
      "epoch": 4.835128784179688e-07,
      "model_forward_time": 5.588842868804932,
      "step": 507
    },
    {
      "epoch": 4.835128784179688e-07,
      "step": 507,
      "training_step_time": 17.02829360961914
    },
    {
      "epoch": 4.84466552734375e-07,
      "model_forward_time": 5.591099500656128,
      "step": 508
    },
    {
      "epoch": 4.84466552734375e-07,
      "step": 508,
      "training_step_time": 17.046460151672363
    },
    {
      "epoch": 4.854202270507813e-07,
      "model_forward_time": 5.612640380859375,
      "step": 509
    },
    {
      "epoch": 4.854202270507813e-07,
      "step": 509,
      "training_step_time": 17.107019424438477
    },
    {
      "epoch": 4.863739013671875e-07,
      "grad_norm": 2.1853549480438232,
      "learning_rate": 1e-05,
      "loss": 0.1314,
      "step": 510
    },
    {
      "epoch": 4.863739013671875e-07,
      "model_forward_time": 5.973844289779663,
      "step": 510
    },
    {
      "dynamics_loss_avg": 0.08875537440180778,
      "epoch": 4.863739013671875e-07,
      "step": 510
    },
    {
      "action_loss_avg": 0.042768124490976334,
      "epoch": 4.863739013671875e-07,
      "step": 510
    },
    {
      "epoch": 4.863739013671875e-07,
      "step": 510,
      "training_step_time": 17.474159717559814
    },
    {
      "epoch": 4.873275756835938e-07,
      "model_forward_time": 5.626838445663452,
      "step": 511
    },
    {
      "epoch": 4.873275756835938e-07,
      "step": 511,
      "training_step_time": 17.142691135406494
    },
    {
      "epoch": 4.8828125e-07,
      "model_forward_time": 5.61917519569397,
      "step": 512
    },
    {
      "epoch": 4.8828125e-07,
      "step": 512,
      "training_step_time": 17.14109516143799
    },
    {
      "epoch": 4.892349243164063e-07,
      "model_forward_time": 5.619571924209595,
      "step": 513
    },
    {
      "epoch": 4.892349243164063e-07,
      "step": 513,
      "training_step_time": 17.122715950012207
    },
    {
      "epoch": 4.901885986328125e-07,
      "model_forward_time": 5.6186065673828125,
      "step": 514
    },
    {
      "epoch": 4.901885986328125e-07,
      "step": 514,
      "training_step_time": 17.117187023162842
    },
    {
      "epoch": 4.911422729492188e-07,
      "model_forward_time": 5.615200042724609,
      "step": 515
    },
    {
      "epoch": 4.911422729492188e-07,
      "step": 515,
      "training_step_time": 17.088601112365723
    },
    {
      "epoch": 4.92095947265625e-07,
      "model_forward_time": 5.9877471923828125,
      "step": 516
    },
    {
      "epoch": 4.92095947265625e-07,
      "step": 516,
      "training_step_time": 17.41871404647827
    },
    {
      "epoch": 4.930496215820313e-07,
      "model_forward_time": 5.575128555297852,
      "step": 517
    },
    {
      "epoch": 4.930496215820313e-07,
      "step": 517,
      "training_step_time": 16.98338031768799
    },
    {
      "epoch": 4.940032958984375e-07,
      "model_forward_time": 5.57555890083313,
      "step": 518
    },
    {
      "epoch": 4.940032958984375e-07,
      "step": 518,
      "training_step_time": 16.991579055786133
    },
    {
      "epoch": 4.949569702148438e-07,
      "model_forward_time": 5.575857877731323,
      "step": 519
    },
    {
      "epoch": 4.949569702148438e-07,
      "step": 519,
      "training_step_time": 16.977983951568604
    },
    {
      "epoch": 4.9591064453125e-07,
      "grad_norm": 1.2560971975326538,
      "learning_rate": 1e-05,
      "loss": 0.0989,
      "step": 520
    },
    {
      "epoch": 4.9591064453125e-07,
      "model_forward_time": 5.5775063037872314,
      "step": 520
    },
    {
      "dynamics_loss_avg": 0.062438165210187434,
      "epoch": 4.9591064453125e-07,
      "step": 520
    },
    {
      "action_loss_avg": 0.03187190340831876,
      "epoch": 4.9591064453125e-07,
      "step": 520
    },
    {
      "epoch": 4.9591064453125e-07,
      "step": 520,
      "training_step_time": 17.01379632949829
    },
    {
      "epoch": 4.968643188476563e-07,
      "model_forward_time": 5.92793345451355,
      "step": 521
    },
    {
      "epoch": 4.968643188476563e-07,
      "step": 521,
      "training_step_time": 17.349945545196533
    },
    {
      "epoch": 4.978179931640625e-07,
      "model_forward_time": 5.584918975830078,
      "step": 522
    },
    {
      "epoch": 4.978179931640625e-07,
      "step": 522,
      "training_step_time": 17.01728630065918
    },
    {
      "epoch": 4.987716674804688e-07,
      "model_forward_time": 5.598847150802612,
      "step": 523
    },
    {
      "epoch": 4.987716674804688e-07,
      "step": 523,
      "training_step_time": 17.053397178649902
    },
    {
      "epoch": 4.99725341796875e-07,
      "model_forward_time": 5.6096343994140625,
      "step": 524
    },
    {
      "epoch": 4.99725341796875e-07,
      "step": 524,
      "training_step_time": 17.122952699661255
    },
    {
      "epoch": 5.006790161132813e-07,
      "model_forward_time": 5.623891592025757,
      "step": 525
    },
    {
      "epoch": 5.006790161132813e-07,
      "step": 525,
      "training_step_time": 17.143537282943726
    },
    {
      "epoch": 5.016326904296875e-07,
      "model_forward_time": 5.621872425079346,
      "step": 526
    },
    {
      "epoch": 5.016326904296875e-07,
      "step": 526,
      "training_step_time": 17.491457223892212
    },
    {
      "epoch": 5.025863647460938e-07,
      "model_forward_time": 5.614690542221069,
      "step": 527
    },
    {
      "epoch": 5.025863647460938e-07,
      "step": 527,
      "training_step_time": 17.090392351150513
    },
    {
      "epoch": 5.035400390625e-07,
      "model_forward_time": 5.59438419342041,
      "step": 528
    },
    {
      "epoch": 5.035400390625e-07,
      "step": 528,
      "training_step_time": 17.058815002441406
    },
    {
      "epoch": 5.044937133789063e-07,
      "model_forward_time": 5.601871728897095,
      "step": 529
    },
    {
      "epoch": 5.044937133789063e-07,
      "step": 529,
      "training_step_time": 17.05395483970642
    },
    {
      "epoch": 5.054473876953125e-07,
      "grad_norm": 1.596875548362732,
      "learning_rate": 1e-05,
      "loss": 0.0963,
      "step": 530
    },
    {
      "epoch": 5.054473876953125e-07,
      "model_forward_time": 5.6045660972595215,
      "step": 530
    },
    {
      "dynamics_loss_avg": 0.06504865605384111,
      "epoch": 5.054473876953125e-07,
      "step": 530
    },
    {
      "action_loss_avg": 0.027873299503698944,
      "epoch": 5.054473876953125e-07,
      "step": 530
    },
    {
      "epoch": 5.054473876953125e-07,
      "step": 530,
      "training_step_time": 17.067589282989502
    },
    {
      "epoch": 5.064010620117188e-07,
      "model_forward_time": 5.610079765319824,
      "step": 531
    },
    {
      "epoch": 5.064010620117188e-07,
      "step": 531,
      "training_step_time": 17.097698211669922
    },
    {
      "epoch": 5.07354736328125e-07,
      "model_forward_time": 5.965182065963745,
      "step": 532
    },
    {
      "epoch": 5.07354736328125e-07,
      "step": 532,
      "training_step_time": 17.468806505203247
    },
    {
      "epoch": 5.083084106445313e-07,
      "model_forward_time": 5.61917781829834,
      "step": 533
    },
    {
      "epoch": 5.083084106445313e-07,
      "step": 533,
      "training_step_time": 17.126474857330322
    },
    {
      "epoch": 5.092620849609375e-07,
      "model_forward_time": 5.608654260635376,
      "step": 534
    },
    {
      "epoch": 5.092620849609375e-07,
      "step": 534,
      "training_step_time": 17.082293272018433
    },
    {
      "epoch": 5.102157592773438e-07,
      "model_forward_time": 5.589629173278809,
      "step": 535
    },
    {
      "epoch": 5.102157592773438e-07,
      "step": 535,
      "training_step_time": 17.015633583068848
    },
    {
      "epoch": 5.1116943359375e-07,
      "model_forward_time": 5.586554765701294,
      "step": 536
    },
    {
      "epoch": 5.1116943359375e-07,
      "step": 536,
      "training_step_time": 17.003663063049316
    },
    {
      "epoch": 5.121231079101563e-07,
      "model_forward_time": 5.588385343551636,
      "step": 537
    },
    {
      "epoch": 5.121231079101563e-07,
      "step": 537,
      "training_step_time": 17.364745140075684
    },
    {
      "epoch": 5.130767822265625e-07,
      "model_forward_time": 5.594558238983154,
      "step": 538
    },
    {
      "epoch": 5.130767822265625e-07,
      "step": 538,
      "training_step_time": 17.040696382522583
    },
    {
      "epoch": 5.140304565429688e-07,
      "model_forward_time": 5.612666368484497,
      "step": 539
    },
    {
      "epoch": 5.140304565429688e-07,
      "step": 539,
      "training_step_time": 17.084676504135132
    },
    {
      "epoch": 5.14984130859375e-07,
      "grad_norm": 2.378451347351074,
      "learning_rate": 1e-05,
      "loss": 0.0981,
      "step": 540
    },
    {
      "epoch": 5.14984130859375e-07,
      "model_forward_time": 5.622057676315308,
      "step": 540
    },
    {
      "dynamics_loss_avg": 0.07316150274127722,
      "epoch": 5.14984130859375e-07,
      "step": 540
    },
    {
      "action_loss_avg": 0.0329498004168272,
      "epoch": 5.14984130859375e-07,
      "step": 540
    },
    {
      "epoch": 5.14984130859375e-07,
      "step": 540,
      "training_step_time": 17.132275819778442
    },
    {
      "epoch": 5.159378051757813e-07,
      "model_forward_time": 5.619291543960571,
      "step": 541
    },
    {
      "epoch": 5.159378051757813e-07,
      "step": 541,
      "training_step_time": 17.107771635055542
    },
    {
      "epoch": 5.168914794921875e-07,
      "model_forward_time": 5.598813533782959,
      "step": 542
    },
    {
      "epoch": 5.168914794921875e-07,
      "step": 542,
      "training_step_time": 17.03277564048767
    },
    {
      "epoch": 5.178451538085938e-07,
      "model_forward_time": 5.927793741226196,
      "step": 543
    },
    {
      "epoch": 5.178451538085938e-07,
      "step": 543,
      "training_step_time": 17.365412950515747
    },
    {
      "epoch": 5.18798828125e-07,
      "model_forward_time": 5.572945594787598,
      "step": 544
    },
    {
      "epoch": 5.18798828125e-07,
      "step": 544,
      "training_step_time": 16.991020679473877
    },
    {
      "epoch": 5.197525024414063e-07,
      "model_forward_time": 5.572066068649292,
      "step": 545
    },
    {
      "epoch": 5.197525024414063e-07,
      "step": 545,
      "training_step_time": 16.998119831085205
    },
    {
      "epoch": 5.207061767578125e-07,
      "model_forward_time": 5.589353561401367,
      "step": 546
    },
    {
      "epoch": 5.207061767578125e-07,
      "step": 546,
      "training_step_time": 17.02741765975952
    },
    {
      "epoch": 5.216598510742188e-07,
      "model_forward_time": 5.578131198883057,
      "step": 547
    },
    {
      "epoch": 5.216598510742188e-07,
      "step": 547,
      "training_step_time": 17.00645160675049
    },
    {
      "epoch": 5.22613525390625e-07,
      "model_forward_time": 5.948681831359863,
      "step": 548
    },
    {
      "epoch": 5.22613525390625e-07,
      "step": 548,
      "training_step_time": 17.381482362747192
    },
    {
      "epoch": 5.235671997070313e-07,
      "model_forward_time": 5.599571466445923,
      "step": 549
    },
    {
      "epoch": 5.235671997070313e-07,
      "step": 549,
      "training_step_time": 17.0859215259552
    },
    {
      "epoch": 5.245208740234375e-07,
      "grad_norm": 1.1317259073257446,
      "learning_rate": 1e-05,
      "loss": 0.1272,
      "step": 550
    },
    {
      "epoch": 5.245208740234375e-07,
      "model_forward_time": 5.6224329471588135,
      "step": 550
    },
    {
      "dynamics_loss_avg": 0.0835259772837162,
      "epoch": 5.245208740234375e-07,
      "step": 550
    },
    {
      "action_loss_avg": 0.040057852119207385,
      "epoch": 5.245208740234375e-07,
      "step": 550
    },
    {
      "epoch": 5.245208740234375e-07,
      "step": 550,
      "training_step_time": 17.11950373649597
    },
    {
      "epoch": 5.254745483398438e-07,
      "model_forward_time": 5.606196880340576,
      "step": 551
    },
    {
      "epoch": 5.254745483398438e-07,
      "step": 551,
      "training_step_time": 17.07090997695923
    },
    {
      "epoch": 5.2642822265625e-07,
      "model_forward_time": 5.59125280380249,
      "step": 552
    },
    {
      "epoch": 5.2642822265625e-07,
      "step": 552,
      "training_step_time": 17.046163320541382
    },
    {
      "epoch": 5.273818969726563e-07,
      "model_forward_time": 5.598268508911133,
      "step": 553
    },
    {
      "epoch": 5.273818969726563e-07,
      "step": 553,
      "training_step_time": 17.401309967041016
    },
    {
      "epoch": 5.283355712890625e-07,
      "model_forward_time": 5.601338148117065,
      "step": 554
    },
    {
      "epoch": 5.283355712890625e-07,
      "step": 554,
      "training_step_time": 17.03825330734253
    },
    {
      "epoch": 5.292892456054688e-07,
      "model_forward_time": 5.594842195510864,
      "step": 555
    },
    {
      "epoch": 5.292892456054688e-07,
      "step": 555,
      "training_step_time": 17.030879497528076
    },
    {
      "epoch": 5.30242919921875e-07,
      "model_forward_time": 5.597942590713501,
      "step": 556
    },
    {
      "epoch": 5.30242919921875e-07,
      "step": 556,
      "training_step_time": 17.06758689880371
    },
    {
      "epoch": 5.311965942382813e-07,
      "model_forward_time": 5.607752799987793,
      "step": 557
    },
    {
      "epoch": 5.311965942382813e-07,
      "step": 557,
      "training_step_time": 17.09090566635132
    },
    {
      "epoch": 5.321502685546875e-07,
      "model_forward_time": 5.605133295059204,
      "step": 558
    },
    {
      "epoch": 5.321502685546875e-07,
      "step": 558,
      "training_step_time": 17.094508171081543
    },
    {
      "epoch": 5.331039428710938e-07,
      "model_forward_time": 5.970140218734741,
      "step": 559
    },
    {
      "epoch": 5.331039428710938e-07,
      "step": 559,
      "training_step_time": 17.468925714492798
    },
    {
      "epoch": 5.340576171875e-07,
      "grad_norm": 1.3254735469818115,
      "learning_rate": 1e-05,
      "loss": 0.1117,
      "step": 560
    },
    {
      "epoch": 5.340576171875e-07,
      "model_forward_time": 5.612865686416626,
      "step": 560
    },
    {
      "dynamics_loss_avg": 0.07318138144910336,
      "epoch": 5.340576171875e-07,
      "step": 560
    },
    {
      "action_loss_avg": 0.03979517305269838,
      "epoch": 5.340576171875e-07,
      "step": 560
    },
    {
      "epoch": 5.340576171875e-07,
      "step": 560,
      "training_step_time": 17.103585720062256
    },
    {
      "epoch": 5.350112915039063e-07,
      "model_forward_time": 5.603909492492676,
      "step": 561
    },
    {
      "epoch": 5.350112915039063e-07,
      "step": 561,
      "training_step_time": 17.02732825279236
    },
    {
      "epoch": 5.359649658203126e-07,
      "model_forward_time": 5.5892744064331055,
      "step": 562
    },
    {
      "epoch": 5.359649658203126e-07,
      "step": 562,
      "training_step_time": 17.016748428344727
    },
    {
      "epoch": 5.369186401367187e-07,
      "model_forward_time": 5.584195375442505,
      "step": 563
    },
    {
      "epoch": 5.369186401367187e-07,
      "step": 563,
      "training_step_time": 16.981163263320923
    },
    {
      "epoch": 5.37872314453125e-07,
      "model_forward_time": 5.933314561843872,
      "step": 564
    },
    {
      "epoch": 5.37872314453125e-07,
      "step": 564,
      "training_step_time": 17.351824045181274
    },
    {
      "epoch": 5.388259887695312e-07,
      "model_forward_time": 5.592909097671509,
      "step": 565
    },
    {
      "epoch": 5.388259887695312e-07,
      "step": 565,
      "training_step_time": 17.026607513427734
    },
    {
      "epoch": 5.397796630859375e-07,
      "model_forward_time": 5.598347425460815,
      "step": 566
    },
    {
      "epoch": 5.397796630859375e-07,
      "step": 566,
      "training_step_time": 17.066056966781616
    },
    {
      "epoch": 5.407333374023437e-07,
      "model_forward_time": 5.61690092086792,
      "step": 567
    },
    {
      "epoch": 5.407333374023437e-07,
      "step": 567,
      "training_step_time": 17.125439405441284
    },
    {
      "epoch": 5.4168701171875e-07,
      "model_forward_time": 5.62917160987854,
      "step": 568
    },
    {
      "epoch": 5.4168701171875e-07,
      "step": 568,
      "training_step_time": 17.125990629196167
    },
    {
      "epoch": 5.426406860351562e-07,
      "model_forward_time": 5.602643728256226,
      "step": 569
    },
    {
      "epoch": 5.426406860351562e-07,
      "step": 569,
      "training_step_time": 17.056297063827515
    },
    {
      "epoch": 5.435943603515625e-07,
      "grad_norm": 2.0449256896972656,
      "learning_rate": 1e-05,
      "loss": 0.1119,
      "step": 570
    },
    {
      "epoch": 5.435943603515625e-07,
      "model_forward_time": 5.959978103637695,
      "step": 570
    },
    {
      "dynamics_loss_avg": 0.07473896145820617,
      "epoch": 5.435943603515625e-07,
      "step": 570
    },
    {
      "action_loss_avg": 0.03961593499407172,
      "epoch": 5.435943603515625e-07,
      "step": 570
    },
    {
      "epoch": 5.435943603515625e-07,
      "step": 570,
      "training_step_time": 17.37816286087036
    },
    {
      "epoch": 5.445480346679687e-07,
      "model_forward_time": 5.594894170761108,
      "step": 571
    },
    {
      "epoch": 5.445480346679687e-07,
      "step": 571,
      "training_step_time": 17.014308214187622
    },
    {
      "epoch": 5.45501708984375e-07,
      "model_forward_time": 5.590451240539551,
      "step": 572
    },
    {
      "epoch": 5.45501708984375e-07,
      "step": 572,
      "training_step_time": 17.027911901474
    },
    {
      "epoch": 5.464553833007812e-07,
      "model_forward_time": 5.609383821487427,
      "step": 573
    },
    {
      "epoch": 5.464553833007812e-07,
      "step": 573,
      "training_step_time": 17.068305730819702
    },
    {
      "epoch": 5.474090576171875e-07,
      "model_forward_time": 5.620680570602417,
      "step": 574
    },
    {
      "epoch": 5.474090576171875e-07,
      "step": 574,
      "training_step_time": 17.110084295272827
    },
    {
      "epoch": 5.483627319335937e-07,
      "model_forward_time": 5.969238042831421,
      "step": 575
    },
    {
      "epoch": 5.483627319335937e-07,
      "step": 575,
      "training_step_time": 17.483470678329468
    },
    {
      "epoch": 5.4931640625e-07,
      "model_forward_time": 5.623978853225708,
      "step": 576
    },
    {
      "epoch": 5.4931640625e-07,
      "step": 576,
      "training_step_time": 17.12934184074402
    },
    {
      "epoch": 5.502700805664062e-07,
      "model_forward_time": 5.621719598770142,
      "step": 577
    },
    {
      "epoch": 5.502700805664062e-07,
      "step": 577,
      "training_step_time": 17.12434434890747
    },
    {
      "epoch": 5.512237548828125e-07,
      "model_forward_time": 5.620795726776123,
      "step": 578
    },
    {
      "epoch": 5.512237548828125e-07,
      "step": 578,
      "training_step_time": 17.105467319488525
    },
    {
      "epoch": 5.521774291992187e-07,
      "model_forward_time": 5.619779348373413,
      "step": 579
    },
    {
      "epoch": 5.521774291992187e-07,
      "step": 579,
      "training_step_time": 17.116675853729248
    },
    {
      "epoch": 5.53131103515625e-07,
      "grad_norm": 1.670231819152832,
      "learning_rate": 1e-05,
      "loss": 0.1344,
      "step": 580
    },
    {
      "epoch": 5.53131103515625e-07,
      "model_forward_time": 5.605622053146362,
      "step": 580
    },
    {
      "dynamics_loss_avg": 0.08838388286530971,
      "epoch": 5.53131103515625e-07,
      "step": 580
    },
    {
      "action_loss_avg": 0.037592968344688414,
      "epoch": 5.53131103515625e-07,
      "step": 580
    },
    {
      "epoch": 5.53131103515625e-07,
      "step": 580,
      "training_step_time": 17.097352504730225
    },
    {
      "epoch": 5.540847778320312e-07,
      "model_forward_time": 5.970191955566406,
      "step": 581
    },
    {
      "epoch": 5.540847778320312e-07,
      "step": 581,
      "training_step_time": 17.459831714630127
    },
    {
      "epoch": 5.550384521484375e-07,
      "model_forward_time": 5.62748384475708,
      "step": 582
    },
    {
      "epoch": 5.550384521484375e-07,
      "step": 582,
      "training_step_time": 17.125558137893677
    },
    {
      "epoch": 5.559921264648437e-07,
      "model_forward_time": 5.602869272232056,
      "step": 583
    },
    {
      "epoch": 5.559921264648437e-07,
      "step": 583,
      "training_step_time": 17.08247685432434
    },
    {
      "epoch": 5.5694580078125e-07,
      "model_forward_time": 4.277154922485352,
      "step": 584
    },
    {
      "epoch": 5.5694580078125e-07,
      "step": 584,
      "training_step_time": 12.785662412643433
    },
    {
      "epoch": 5.578994750976562e-07,
      "model_forward_time": 4.259653806686401,
      "step": 585
    },
    {
      "epoch": 5.578994750976562e-07,
      "step": 585,
      "training_step_time": 12.752119064331055
    },
    {
      "epoch": 5.588531494140625e-07,
      "model_forward_time": 5.940475702285767,
      "step": 586
    },
    {
      "epoch": 5.588531494140625e-07,
      "step": 586,
      "training_step_time": 17.36960506439209
    },
    {
      "epoch": 5.598068237304687e-07,
      "model_forward_time": 5.592021465301514,
      "step": 587
    },
    {
      "epoch": 5.598068237304687e-07,
      "step": 587,
      "training_step_time": 17.030436277389526
    },
    {
      "epoch": 5.60760498046875e-07,
      "model_forward_time": 5.601689338684082,
      "step": 588
    },
    {
      "epoch": 5.60760498046875e-07,
      "step": 588,
      "training_step_time": 17.103370428085327
    },
    {
      "epoch": 5.617141723632812e-07,
      "model_forward_time": 5.627993583679199,
      "step": 589
    },
    {
      "epoch": 5.617141723632812e-07,
      "step": 589,
      "training_step_time": 17.146883487701416
    },
    {
      "epoch": 5.626678466796875e-07,
      "grad_norm": 2.0990257263183594,
      "learning_rate": 1e-05,
      "loss": 0.1054,
      "step": 590
    },
    {
      "epoch": 5.626678466796875e-07,
      "model_forward_time": 5.626682281494141,
      "step": 590
    },
    {
      "dynamics_loss_avg": 0.06951635256409645,
      "epoch": 5.626678466796875e-07,
      "step": 590
    },
    {
      "action_loss_avg": 0.03643495626747608,
      "epoch": 5.626678466796875e-07,
      "step": 590
    },
    {
      "epoch": 5.626678466796875e-07,
      "step": 590,
      "training_step_time": 17.109054803848267
    },
    {
      "epoch": 5.636215209960937e-07,
      "model_forward_time": 5.5983054637908936,
      "step": 591
    },
    {
      "epoch": 5.636215209960937e-07,
      "step": 591,
      "training_step_time": 17.046924352645874
    },
    {
      "epoch": 5.645751953125e-07,
      "model_forward_time": 5.946259021759033,
      "step": 592
    },
    {
      "epoch": 5.645751953125e-07,
      "step": 592,
      "training_step_time": 17.386050939559937
    },
    {
      "epoch": 5.655288696289062e-07,
      "model_forward_time": 5.579942226409912,
      "step": 593
    },
    {
      "epoch": 5.655288696289062e-07,
      "step": 593,
      "training_step_time": 17.03101897239685
    },
    {
      "epoch": 5.664825439453125e-07,
      "model_forward_time": 5.592306137084961,
      "step": 594
    },
    {
      "epoch": 5.664825439453125e-07,
      "step": 594,
      "training_step_time": 17.061476230621338
    },
    {
      "epoch": 5.674362182617187e-07,
      "model_forward_time": 5.611386775970459,
      "step": 595
    },
    {
      "epoch": 5.674362182617187e-07,
      "step": 595,
      "training_step_time": 17.099954843521118
    },
    {
      "epoch": 5.68389892578125e-07,
      "model_forward_time": 5.6203272342681885,
      "step": 596
    },
    {
      "epoch": 5.68389892578125e-07,
      "step": 596,
      "training_step_time": 17.113160371780396
    },
    {
      "epoch": 5.693435668945312e-07,
      "model_forward_time": 5.967419385910034,
      "step": 597
    },
    {
      "epoch": 5.693435668945312e-07,
      "step": 597,
      "training_step_time": 17.43950915336609
    },
    {
      "epoch": 5.702972412109375e-07,
      "model_forward_time": 5.601563215255737,
      "step": 598
    },
    {
      "epoch": 5.702972412109375e-07,
      "step": 598,
      "training_step_time": 17.033780097961426
    },
    {
      "epoch": 5.712509155273437e-07,
      "model_forward_time": 5.592378854751587,
      "step": 599
    },
    {
      "epoch": 5.712509155273437e-07,
      "step": 599,
      "training_step_time": 17.039703607559204
    },
    {
      "epoch": 5.7220458984375e-07,
      "grad_norm": 0.7852856516838074,
      "learning_rate": 1e-05,
      "loss": 0.0971,
      "step": 600
    },
    {
      "epoch": 5.7220458984375e-07,
      "model_forward_time": 5.595813512802124,
      "step": 600
    },
    {
      "dynamics_loss_avg": 0.06565323323011399,
      "epoch": 5.7220458984375e-07,
      "step": 600
    },
    {
      "action_loss_avg": 0.03254586635157466,
      "epoch": 5.7220458984375e-07,
      "step": 600
    },
    {
      "epoch": 5.7220458984375e-07,
      "step": 600,
      "training_step_time": 17.02664351463318
    },
    {
      "epoch": 5.731582641601562e-07,
      "model_forward_time": 5.596338272094727,
      "step": 601
    },
    {
      "epoch": 5.731582641601562e-07,
      "step": 601,
      "training_step_time": 17.07986879348755
    },
    {
      "epoch": 5.741119384765625e-07,
      "model_forward_time": 5.609363079071045,
      "step": 602
    },
    {
      "epoch": 5.741119384765625e-07,
      "step": 602,
      "training_step_time": 17.4705593585968
    },
    {
      "epoch": 5.750656127929687e-07,
      "model_forward_time": 5.620516300201416,
      "step": 603
    },
    {
      "epoch": 5.750656127929687e-07,
      "step": 603,
      "training_step_time": 17.1276695728302
    },
    {
      "epoch": 5.76019287109375e-07,
      "model_forward_time": 5.6262125968933105,
      "step": 604
    },
    {
      "epoch": 5.76019287109375e-07,
      "step": 604,
      "training_step_time": 17.12968683242798
    },
    {
      "epoch": 5.769729614257812e-07,
      "model_forward_time": 5.621087551116943,
      "step": 605
    },
    {
      "epoch": 5.769729614257812e-07,
      "step": 605,
      "training_step_time": 17.12259078025818
    },
    {
      "epoch": 5.779266357421875e-07,
      "model_forward_time": 5.616328477859497,
      "step": 606
    },
    {
      "epoch": 5.779266357421875e-07,
      "step": 606,
      "training_step_time": 17.1284658908844
    },
    {
      "epoch": 5.788803100585937e-07,
      "model_forward_time": 5.613451242446899,
      "step": 607
    },
    {
      "epoch": 5.788803100585937e-07,
      "step": 607,
      "training_step_time": 17.101058959960938
    },
    {
      "epoch": 5.79833984375e-07,
      "model_forward_time": 5.954967498779297,
      "step": 608
    },
    {
      "epoch": 5.79833984375e-07,
      "step": 608,
      "training_step_time": 17.413290977478027
    },
    {
      "epoch": 5.807876586914062e-07,
      "model_forward_time": 5.596168756484985,
      "step": 609
    },
    {
      "epoch": 5.807876586914062e-07,
      "step": 609,
      "training_step_time": 17.042718648910522
    },
    {
      "epoch": 5.817413330078125e-07,
      "grad_norm": 1.1968309879302979,
      "learning_rate": 1e-05,
      "loss": 0.1008,
      "step": 610
    },
    {
      "epoch": 5.817413330078125e-07,
      "model_forward_time": 5.592349529266357,
      "step": 610
    },
    {
      "dynamics_loss_avg": 0.06691386587917805,
      "epoch": 5.817413330078125e-07,
      "step": 610
    },
    {
      "action_loss_avg": 0.03226322866976261,
      "epoch": 5.817413330078125e-07,
      "step": 610
    },
    {
      "epoch": 5.817413330078125e-07,
      "step": 610,
      "training_step_time": 17.017030954360962
    },
    {
      "epoch": 5.826950073242187e-07,
      "model_forward_time": 5.590609788894653,
      "step": 611
    },
    {
      "epoch": 5.826950073242187e-07,
      "step": 611,
      "training_step_time": 17.03791832923889
    },
    {
      "epoch": 5.83648681640625e-07,
      "model_forward_time": 5.6018712520599365,
      "step": 612
    },
    {
      "epoch": 5.83648681640625e-07,
      "step": 612,
      "training_step_time": 17.04907512664795
    },
    {
      "epoch": 5.846023559570312e-07,
      "model_forward_time": 5.9545371532440186,
      "step": 613
    },
    {
      "epoch": 5.846023559570312e-07,
      "step": 613,
      "training_step_time": 17.409193992614746
    },
    {
      "epoch": 5.855560302734375e-07,
      "model_forward_time": 5.604228973388672,
      "step": 614
    },
    {
      "epoch": 5.855560302734375e-07,
      "step": 614,
      "training_step_time": 17.082462310791016
    },
    {
      "epoch": 5.865097045898437e-07,
      "model_forward_time": 5.605851650238037,
      "step": 615
    },
    {
      "epoch": 5.865097045898437e-07,
      "step": 615,
      "training_step_time": 17.03360390663147
    },
    {
      "epoch": 5.8746337890625e-07,
      "model_forward_time": 5.592261791229248,
      "step": 616
    },
    {
      "epoch": 5.8746337890625e-07,
      "step": 616,
      "training_step_time": 17.03487467765808
    },
    {
      "epoch": 5.884170532226562e-07,
      "model_forward_time": 5.583861827850342,
      "step": 617
    },
    {
      "epoch": 5.884170532226562e-07,
      "step": 617,
      "training_step_time": 17.00869369506836
    },
    {
      "epoch": 5.893707275390625e-07,
      "model_forward_time": 5.5958943367004395,
      "step": 618
    },
    {
      "epoch": 5.893707275390625e-07,
      "step": 618,
      "training_step_time": 17.055902004241943
    },
    {
      "epoch": 5.903244018554687e-07,
      "model_forward_time": 5.96308445930481,
      "step": 619
    },
    {
      "epoch": 5.903244018554687e-07,
      "step": 619,
      "training_step_time": 17.4516658782959
    },
    {
      "epoch": 5.91278076171875e-07,
      "grad_norm": 0.6274821162223816,
      "learning_rate": 1e-05,
      "loss": 0.0956,
      "step": 620
    },
    {
      "epoch": 5.91278076171875e-07,
      "model_forward_time": 5.602372169494629,
      "step": 620
    },
    {
      "dynamics_loss_avg": 0.06399942487478257,
      "epoch": 5.91278076171875e-07,
      "step": 620
    },
    {
      "action_loss_avg": 0.03331329650245607,
      "epoch": 5.91278076171875e-07,
      "step": 620
    },
    {
      "epoch": 5.91278076171875e-07,
      "step": 620,
      "training_step_time": 17.09219980239868
    },
    {
      "epoch": 5.922317504882812e-07,
      "model_forward_time": 5.615926265716553,
      "step": 621
    },
    {
      "epoch": 5.922317504882812e-07,
      "step": 621,
      "training_step_time": 17.10922622680664
    },
    {
      "epoch": 5.931854248046875e-07,
      "model_forward_time": 5.617345809936523,
      "step": 622
    },
    {
      "epoch": 5.931854248046875e-07,
      "step": 622,
      "training_step_time": 17.11634135246277
    },
    {
      "epoch": 5.941390991210937e-07,
      "model_forward_time": 5.605565071105957,
      "step": 623
    },
    {
      "epoch": 5.941390991210937e-07,
      "step": 623,
      "training_step_time": 17.081138849258423
    },
    {
      "epoch": 5.950927734375e-07,
      "model_forward_time": 5.981844663619995,
      "step": 624
    },
    {
      "epoch": 5.950927734375e-07,
      "step": 624,
      "training_step_time": 17.43607997894287
    },
    {
      "epoch": 5.960464477539062e-07,
      "model_forward_time": 5.602394342422485,
      "step": 625
    },
    {
      "epoch": 5.960464477539062e-07,
      "step": 625,
      "training_step_time": 17.054346323013306
    },
    {
      "epoch": 5.970001220703125e-07,
      "model_forward_time": 5.588301420211792,
      "step": 626
    },
    {
      "epoch": 5.970001220703125e-07,
      "step": 626,
      "training_step_time": 17.020923852920532
    },
    {
      "epoch": 5.979537963867188e-07,
      "model_forward_time": 5.613487720489502,
      "step": 627
    },
    {
      "epoch": 5.979537963867188e-07,
      "step": 627,
      "training_step_time": 17.063862800598145
    },
    {
      "epoch": 5.98907470703125e-07,
      "model_forward_time": 5.592061996459961,
      "step": 628
    },
    {
      "epoch": 5.98907470703125e-07,
      "step": 628,
      "training_step_time": 17.057941913604736
    },
    {
      "epoch": 5.998611450195313e-07,
      "model_forward_time": 5.59916877746582,
      "step": 629
    },
    {
      "epoch": 5.998611450195313e-07,
      "step": 629,
      "training_step_time": 17.0574369430542
    },
    {
      "epoch": 6.008148193359375e-07,
      "grad_norm": 2.163902759552002,
      "learning_rate": 1e-05,
      "loss": 0.0992,
      "step": 630
    },
    {
      "epoch": 6.008148193359375e-07,
      "model_forward_time": 5.940722703933716,
      "step": 630
    },
    {
      "dynamics_loss_avg": 0.07089761234819889,
      "epoch": 6.008148193359375e-07,
      "step": 630
    },
    {
      "action_loss_avg": 0.033928290009498596,
      "epoch": 6.008148193359375e-07,
      "step": 630
    },
    {
      "epoch": 6.008148193359375e-07,
      "step": 630,
      "training_step_time": 17.3967387676239
    },
    {
      "epoch": 6.017684936523438e-07,
      "model_forward_time": 5.590991258621216,
      "step": 631
    },
    {
      "epoch": 6.017684936523438e-07,
      "step": 631,
      "training_step_time": 17.052931785583496
    },
    {
      "epoch": 6.0272216796875e-07,
      "model_forward_time": 5.596532106399536,
      "step": 632
    },
    {
      "epoch": 6.0272216796875e-07,
      "step": 632,
      "training_step_time": 17.070653200149536
    },
    {
      "epoch": 6.036758422851563e-07,
      "model_forward_time": 5.603273630142212,
      "step": 633
    },
    {
      "epoch": 6.036758422851563e-07,
      "step": 633,
      "training_step_time": 17.08328866958618
    },
    {
      "epoch": 6.046295166015625e-07,
      "model_forward_time": 5.6074583530426025,
      "step": 634
    },
    {
      "epoch": 6.046295166015625e-07,
      "step": 634,
      "training_step_time": 17.076634407043457
    },
    {
      "epoch": 6.055831909179688e-07,
      "model_forward_time": 5.975558519363403,
      "step": 635
    },
    {
      "epoch": 6.055831909179688e-07,
      "step": 635,
      "training_step_time": 17.469412326812744
    },
    {
      "epoch": 6.06536865234375e-07,
      "model_forward_time": 5.612764596939087,
      "step": 636
    },
    {
      "epoch": 6.06536865234375e-07,
      "step": 636,
      "training_step_time": 17.115018606185913
    },
    {
      "epoch": 6.074905395507813e-07,
      "model_forward_time": 5.623314142227173,
      "step": 637
    },
    {
      "epoch": 6.074905395507813e-07,
      "step": 637,
      "training_step_time": 17.11667537689209
    },
    {
      "epoch": 6.084442138671875e-07,
      "model_forward_time": 5.60553765296936,
      "step": 638
    },
    {
      "epoch": 6.084442138671875e-07,
      "step": 638,
      "training_step_time": 17.08230495452881
    },
    {
      "epoch": 6.093978881835938e-07,
      "model_forward_time": 5.6043922901153564,
      "step": 639
    },
    {
      "epoch": 6.093978881835938e-07,
      "step": 639,
      "training_step_time": 17.09276795387268
    },
    {
      "epoch": 6.103515625e-07,
      "grad_norm": 1.7307569980621338,
      "learning_rate": 1e-05,
      "loss": 0.1124,
      "step": 640
    },
    {
      "epoch": 6.103515625e-07,
      "model_forward_time": 5.971481800079346,
      "step": 640
    },
    {
      "dynamics_loss_avg": 0.06799092404544353,
      "epoch": 6.103515625e-07,
      "step": 640
    },
    {
      "action_loss_avg": 0.04091675877571106,
      "epoch": 6.103515625e-07,
      "step": 640
    },
    {
      "epoch": 6.103515625e-07,
      "step": 640,
      "training_step_time": 17.45122766494751
    },
    {
      "epoch": 6.113052368164063e-07,
      "model_forward_time": 5.609447240829468,
      "step": 641
    },
    {
      "epoch": 6.113052368164063e-07,
      "step": 641,
      "training_step_time": 17.094808340072632
    },
    {
      "epoch": 6.122589111328125e-07,
      "model_forward_time": 5.6067893505096436,
      "step": 642
    },
    {
      "epoch": 6.122589111328125e-07,
      "step": 642,
      "training_step_time": 17.090473175048828
    },
    {
      "epoch": 6.132125854492188e-07,
      "model_forward_time": 5.611481189727783,
      "step": 643
    },
    {
      "epoch": 6.132125854492188e-07,
      "step": 643,
      "training_step_time": 17.099519729614258
    },
    {
      "epoch": 6.14166259765625e-07,
      "model_forward_time": 5.604613304138184,
      "step": 644
    },
    {
      "epoch": 6.14166259765625e-07,
      "step": 644,
      "training_step_time": 17.102497577667236
    },
    {
      "epoch": 6.151199340820313e-07,
      "model_forward_time": 5.609381675720215,
      "step": 645
    },
    {
      "epoch": 6.151199340820313e-07,
      "step": 645,
      "training_step_time": 17.09874725341797
    },
    {
      "epoch": 6.160736083984375e-07,
      "model_forward_time": 5.985091686248779,
      "step": 646
    },
    {
      "epoch": 6.160736083984375e-07,
      "step": 646,
      "training_step_time": 17.480387926101685
    },
    {
      "epoch": 6.170272827148438e-07,
      "model_forward_time": 5.625203371047974,
      "step": 647
    },
    {
      "epoch": 6.170272827148438e-07,
      "step": 647,
      "training_step_time": 17.103217124938965
    },
    {
      "epoch": 6.1798095703125e-07,
      "model_forward_time": 5.600256681442261,
      "step": 648
    },
    {
      "epoch": 6.1798095703125e-07,
      "step": 648,
      "training_step_time": 17.057323694229126
    },
    {
      "epoch": 6.189346313476563e-07,
      "model_forward_time": 5.600374460220337,
      "step": 649
    },
    {
      "epoch": 6.189346313476563e-07,
      "step": 649,
      "training_step_time": 17.058812379837036
    },
    {
      "epoch": 6.198883056640625e-07,
      "grad_norm": 2.141730546951294,
      "learning_rate": 1e-05,
      "loss": 0.0916,
      "step": 650
    },
    {
      "epoch": 6.198883056640625e-07,
      "model_forward_time": 5.600920677185059,
      "step": 650
    },
    {
      "dynamics_loss_avg": 0.06504296641796828,
      "epoch": 6.198883056640625e-07,
      "step": 650
    },
    {
      "action_loss_avg": 0.03560885819606483,
      "epoch": 6.198883056640625e-07,
      "step": 650
    },
    {
      "epoch": 6.198883056640625e-07,
      "step": 650,
      "training_step_time": 17.050659894943237
    },
    {
      "epoch": 6.208419799804688e-07,
      "model_forward_time": 5.593500375747681,
      "step": 651
    },
    {
      "epoch": 6.208419799804688e-07,
      "step": 651,
      "training_step_time": 17.368455410003662
    },
    {
      "epoch": 6.21795654296875e-07,
      "model_forward_time": 5.589773416519165,
      "step": 652
    },
    {
      "epoch": 6.21795654296875e-07,
      "step": 652,
      "training_step_time": 16.999438762664795
    },
    {
      "epoch": 6.227493286132813e-07,
      "model_forward_time": 5.597997665405273,
      "step": 653
    },
    {
      "epoch": 6.227493286132813e-07,
      "step": 653,
      "training_step_time": 17.07230281829834
    },
    {
      "epoch": 6.237030029296875e-07,
      "model_forward_time": 5.609451770782471,
      "step": 654
    },
    {
      "epoch": 6.237030029296875e-07,
      "step": 654,
      "training_step_time": 17.086117267608643
    },
    {
      "epoch": 6.246566772460938e-07,
      "model_forward_time": 5.611790895462036,
      "step": 655
    },
    {
      "epoch": 6.246566772460938e-07,
      "step": 655,
      "training_step_time": 17.097704648971558
    },
    {
      "epoch": 6.256103515625e-07,
      "model_forward_time": 5.607733726501465,
      "step": 656
    },
    {
      "epoch": 6.256103515625e-07,
      "step": 656,
      "training_step_time": 17.095059156417847
    },
    {
      "epoch": 6.265640258789063e-07,
      "model_forward_time": 5.974505662918091,
      "step": 657
    },
    {
      "epoch": 6.265640258789063e-07,
      "step": 657,
      "training_step_time": 17.454768896102905
    },
    {
      "epoch": 6.275177001953125e-07,
      "model_forward_time": 5.612444162368774,
      "step": 658
    },
    {
      "epoch": 6.275177001953125e-07,
      "step": 658,
      "training_step_time": 17.100178003311157
    },
    {
      "epoch": 6.284713745117188e-07,
      "model_forward_time": 5.615504264831543,
      "step": 659
    },
    {
      "epoch": 6.284713745117188e-07,
      "step": 659,
      "training_step_time": 17.09786891937256
    },
    {
      "epoch": 6.29425048828125e-07,
      "grad_norm": 1.651818871498108,
      "learning_rate": 1e-05,
      "loss": 0.1196,
      "step": 660
    },
    {
      "epoch": 6.29425048828125e-07,
      "model_forward_time": 5.596841812133789,
      "step": 660
    },
    {
      "dynamics_loss_avg": 0.07240798007696866,
      "epoch": 6.29425048828125e-07,
      "step": 660
    },
    {
      "action_loss_avg": 0.03764743506908417,
      "epoch": 6.29425048828125e-07,
      "step": 660
    },
    {
      "epoch": 6.29425048828125e-07,
      "step": 660,
      "training_step_time": 17.069445610046387
    },
    {
      "epoch": 6.303787231445313e-07,
      "model_forward_time": 5.596953392028809,
      "step": 661
    },
    {
      "epoch": 6.303787231445313e-07,
      "step": 661,
      "training_step_time": 17.066671133041382
    },
    {
      "epoch": 6.313323974609375e-07,
      "model_forward_time": 5.6040942668914795,
      "step": 662
    },
    {
      "epoch": 6.313323974609375e-07,
      "step": 662,
      "training_step_time": 17.441851377487183
    },
    {
      "epoch": 6.322860717773438e-07,
      "model_forward_time": 5.600828170776367,
      "step": 663
    },
    {
      "epoch": 6.322860717773438e-07,
      "step": 663,
      "training_step_time": 17.069098949432373
    },
    {
      "epoch": 6.3323974609375e-07,
      "model_forward_time": 5.606163263320923,
      "step": 664
    },
    {
      "epoch": 6.3323974609375e-07,
      "step": 664,
      "training_step_time": 17.0794038772583
    },
    {
      "epoch": 6.341934204101563e-07,
      "model_forward_time": 5.605687618255615,
      "step": 665
    },
    {
      "epoch": 6.341934204101563e-07,
      "step": 665,
      "training_step_time": 17.0793399810791
    },
    {
      "epoch": 6.351470947265625e-07,
      "model_forward_time": 5.605741500854492,
      "step": 666
    },
    {
      "epoch": 6.351470947265625e-07,
      "step": 666,
      "training_step_time": 17.069101572036743
    },
    {
      "epoch": 6.361007690429688e-07,
      "model_forward_time": 5.592899322509766,
      "step": 667
    },
    {
      "epoch": 6.361007690429688e-07,
      "step": 667,
      "training_step_time": 17.017953157424927
    },
    {
      "epoch": 6.37054443359375e-07,
      "model_forward_time": 5.95351505279541,
      "step": 668
    },
    {
      "epoch": 6.37054443359375e-07,
      "step": 668,
      "training_step_time": 17.39211392402649
    },
    {
      "epoch": 6.380081176757813e-07,
      "model_forward_time": 5.606299638748169,
      "step": 669
    },
    {
      "epoch": 6.380081176757813e-07,
      "step": 669,
      "training_step_time": 17.091776132583618
    },
    {
      "epoch": 6.389617919921875e-07,
      "grad_norm": 1.5777528285980225,
      "learning_rate": 1e-05,
      "loss": 0.0949,
      "step": 670
    },
    {
      "epoch": 6.389617919921875e-07,
      "model_forward_time": 5.611095905303955,
      "step": 670
    },
    {
      "dynamics_loss_avg": 0.06724888943135739,
      "epoch": 6.389617919921875e-07,
      "step": 670
    },
    {
      "action_loss_avg": 0.031001314334571362,
      "epoch": 6.389617919921875e-07,
      "step": 670
    },
    {
      "epoch": 6.389617919921875e-07,
      "step": 670,
      "training_step_time": 17.086254119873047
    },
    {
      "epoch": 6.399154663085938e-07,
      "model_forward_time": 5.603059530258179,
      "step": 671
    },
    {
      "epoch": 6.399154663085938e-07,
      "step": 671,
      "training_step_time": 17.068530797958374
    },
    {
      "epoch": 6.40869140625e-07,
      "model_forward_time": 5.589035272598267,
      "step": 672
    },
    {
      "epoch": 6.40869140625e-07,
      "step": 672,
      "training_step_time": 17.04590082168579
    },
    {
      "epoch": 6.418228149414063e-07,
      "model_forward_time": 5.94667387008667,
      "step": 673
    },
    {
      "epoch": 6.418228149414063e-07,
      "step": 673,
      "training_step_time": 17.321167707443237
    },
    {
      "epoch": 6.427764892578125e-07,
      "model_forward_time": 5.585063934326172,
      "step": 674
    },
    {
      "epoch": 6.427764892578125e-07,
      "step": 674,
      "training_step_time": 17.03100347518921
    },
    {
      "epoch": 6.437301635742188e-07,
      "model_forward_time": 5.5985565185546875,
      "step": 675
    },
    {
      "epoch": 6.437301635742188e-07,
      "step": 675,
      "training_step_time": 17.039308071136475
    },
    {
      "epoch": 6.44683837890625e-07,
      "model_forward_time": 5.600575923919678,
      "step": 676
    },
    {
      "epoch": 6.44683837890625e-07,
      "step": 676,
      "training_step_time": 17.065335988998413
    },
    {
      "epoch": 6.456375122070313e-07,
      "model_forward_time": 5.597217321395874,
      "step": 677
    },
    {
      "epoch": 6.456375122070313e-07,
      "step": 677,
      "training_step_time": 17.03972625732422
    },
    {
      "epoch": 6.465911865234375e-07,
      "model_forward_time": 5.609105348587036,
      "step": 678
    },
    {
      "epoch": 6.465911865234375e-07,
      "step": 678,
      "training_step_time": 17.095797538757324
    },
    {
      "epoch": 6.475448608398438e-07,
      "model_forward_time": 5.965397596359253,
      "step": 679
    },
    {
      "epoch": 6.475448608398438e-07,
      "step": 679,
      "training_step_time": 17.453920602798462
    },
    {
      "epoch": 6.4849853515625e-07,
      "grad_norm": 1.2163965702056885,
      "learning_rate": 1e-05,
      "loss": 0.0814,
      "step": 680
    },
    {
      "epoch": 6.4849853515625e-07,
      "model_forward_time": 5.618013620376587,
      "step": 680
    },
    {
      "dynamics_loss_avg": 0.04822235051542521,
      "epoch": 6.4849853515625e-07,
      "step": 680
    },
    {
      "action_loss_avg": 0.025026361085474493,
      "epoch": 6.4849853515625e-07,
      "step": 680
    },
    {
      "epoch": 6.4849853515625e-07,
      "step": 680,
      "training_step_time": 17.087908506393433
    },
    {
      "epoch": 6.494522094726563e-07,
      "model_forward_time": 5.6005637645721436,
      "step": 681
    },
    {
      "epoch": 6.494522094726563e-07,
      "step": 681,
      "training_step_time": 17.06959891319275
    },
    {
      "epoch": 6.504058837890625e-07,
      "model_forward_time": 5.6046271324157715,
      "step": 682
    },
    {
      "epoch": 6.504058837890625e-07,
      "step": 682,
      "training_step_time": 17.06665563583374
    },
    {
      "epoch": 6.513595581054688e-07,
      "model_forward_time": 5.589216947555542,
      "step": 683
    },
    {
      "epoch": 6.513595581054688e-07,
      "step": 683,
      "training_step_time": 17.031273126602173
    },
    {
      "epoch": 6.52313232421875e-07,
      "model_forward_time": 5.602293014526367,
      "step": 684
    },
    {
      "epoch": 6.52313232421875e-07,
      "step": 684,
      "training_step_time": 17.427889108657837
    },
    {
      "epoch": 6.532669067382813e-07,
      "model_forward_time": 5.60089373588562,
      "step": 685
    },
    {
      "epoch": 6.532669067382813e-07,
      "step": 685,
      "training_step_time": 17.08441185951233
    },
    {
      "epoch": 6.542205810546876e-07,
      "model_forward_time": 5.601747751235962,
      "step": 686
    },
    {
      "epoch": 6.542205810546876e-07,
      "step": 686,
      "training_step_time": 17.056753873825073
    },
    {
      "epoch": 6.551742553710938e-07,
      "model_forward_time": 5.5916759967803955,
      "step": 687
    },
    {
      "epoch": 6.551742553710938e-07,
      "step": 687,
      "training_step_time": 17.06023645401001
    },
    {
      "epoch": 6.561279296875e-07,
      "model_forward_time": 5.597945690155029,
      "step": 688
    },
    {
      "epoch": 6.561279296875e-07,
      "step": 688,
      "training_step_time": 17.057748794555664
    },
    {
      "epoch": 6.570816040039062e-07,
      "model_forward_time": 5.599879503250122,
      "step": 689
    },
    {
      "epoch": 6.570816040039062e-07,
      "step": 689,
      "training_step_time": 17.091427087783813
    },
    {
      "epoch": 6.580352783203124e-07,
      "grad_norm": 1.4038387537002563,
      "learning_rate": 1e-05,
      "loss": 0.1082,
      "step": 690
    },
    {
      "epoch": 6.580352783203124e-07,
      "model_forward_time": 5.968189716339111,
      "step": 690
    },
    {
      "dynamics_loss_avg": 0.07680377084761858,
      "epoch": 6.580352783203124e-07,
      "step": 690
    },
    {
      "action_loss_avg": 0.0368189724162221,
      "epoch": 6.580352783203124e-07,
      "step": 690
    },
    {
      "epoch": 6.580352783203124e-07,
      "step": 690,
      "training_step_time": 17.45352602005005
    },
    {
      "epoch": 6.589889526367187e-07,
      "model_forward_time": 5.627529621124268,
      "step": 691
    },
    {
      "epoch": 6.589889526367187e-07,
      "step": 691,
      "training_step_time": 17.125195264816284
    },
    {
      "epoch": 6.59942626953125e-07,
      "model_forward_time": 5.624860525131226,
      "step": 692
    },
    {
      "epoch": 6.59942626953125e-07,
      "step": 692,
      "training_step_time": 17.12928032875061
    },
    {
      "epoch": 6.608963012695312e-07,
      "model_forward_time": 5.6092283725738525,
      "step": 693
    },
    {
      "epoch": 6.608963012695312e-07,
      "step": 693,
      "training_step_time": 17.079676389694214
    },
    {
      "epoch": 6.618499755859375e-07,
      "model_forward_time": 5.600755214691162,
      "step": 694
    },
    {
      "epoch": 6.618499755859375e-07,
      "step": 694,
      "training_step_time": 17.077059030532837
    },
    {
      "epoch": 6.628036499023437e-07,
      "model_forward_time": 5.6159584522247314,
      "step": 695
    },
    {
      "epoch": 6.628036499023437e-07,
      "step": 695,
      "training_step_time": 17.470550298690796
    },
    {
      "epoch": 6.6375732421875e-07,
      "model_forward_time": 5.629366397857666,
      "step": 696
    },
    {
      "epoch": 6.6375732421875e-07,
      "step": 696,
      "training_step_time": 17.13445782661438
    },
    {
      "epoch": 6.647109985351562e-07,
      "model_forward_time": 5.619012832641602,
      "step": 697
    },
    {
      "epoch": 6.647109985351562e-07,
      "step": 697,
      "training_step_time": 17.134976625442505
    },
    {
      "epoch": 6.656646728515625e-07,
      "model_forward_time": 5.6203460693359375,
      "step": 698
    },
    {
      "epoch": 6.656646728515625e-07,
      "step": 698,
      "training_step_time": 17.113646030426025
    },
    {
      "epoch": 6.666183471679687e-07,
      "model_forward_time": 5.596371173858643,
      "step": 699
    },
    {
      "epoch": 6.666183471679687e-07,
      "step": 699,
      "training_step_time": 17.019957780838013
    },
    {
      "epoch": 6.67572021484375e-07,
      "grad_norm": 1.0820434093475342,
      "learning_rate": 1e-05,
      "loss": 0.0847,
      "step": 700
    },
    {
      "epoch": 6.67572021484375e-07,
      "model_forward_time": 5.586803436279297,
      "step": 700
    },
    {
      "dynamics_loss_avg": 0.05969731789082289,
      "epoch": 6.67572021484375e-07,
      "step": 700
    },
    {
      "action_loss_avg": 0.027906857803463934,
      "epoch": 6.67572021484375e-07,
      "step": 700
    },
    {
      "epoch": 6.67572021484375e-07,
      "step": 700,
      "training_step_time": 17.00929093360901
    },
    {
      "epoch": 6.685256958007812e-07,
      "model_forward_time": 5.962340354919434,
      "step": 701
    },
    {
      "epoch": 6.685256958007812e-07,
      "step": 701,
      "training_step_time": 17.375823497772217
    },
    {
      "epoch": 6.694793701171875e-07,
      "model_forward_time": 5.569398403167725,
      "step": 702
    },
    {
      "epoch": 6.694793701171875e-07,
      "step": 702,
      "training_step_time": 17.010863065719604
    },
    {
      "epoch": 6.704330444335937e-07,
      "model_forward_time": 5.583025932312012,
      "step": 703
    },
    {
      "epoch": 6.704330444335937e-07,
      "step": 703,
      "training_step_time": 17.033034801483154
    },
    {
      "epoch": 6.7138671875e-07,
      "model_forward_time": 5.600006818771362,
      "step": 704
    },
    {
      "epoch": 6.7138671875e-07,
      "step": 704,
      "training_step_time": 17.081089973449707
    },
    {
      "epoch": 6.723403930664062e-07,
      "model_forward_time": 5.6092329025268555,
      "step": 705
    },
    {
      "epoch": 6.723403930664062e-07,
      "step": 705,
      "training_step_time": 17.100399255752563
    },
    {
      "epoch": 6.732940673828125e-07,
      "model_forward_time": 5.980920791625977,
      "step": 706
    },
    {
      "epoch": 6.732940673828125e-07,
      "step": 706,
      "training_step_time": 17.46971821784973
    },
    {
      "epoch": 6.742477416992187e-07,
      "model_forward_time": 5.616300821304321,
      "step": 707
    },
    {
      "epoch": 6.742477416992187e-07,
      "step": 707,
      "training_step_time": 17.1083562374115
    },
    {
      "epoch": 6.75201416015625e-07,
      "model_forward_time": 5.6089866161346436,
      "step": 708
    },
    {
      "epoch": 6.75201416015625e-07,
      "step": 708,
      "training_step_time": 17.094386100769043
    },
    {
      "epoch": 6.761550903320312e-07,
      "model_forward_time": 5.604366302490234,
      "step": 709
    },
    {
      "epoch": 6.761550903320312e-07,
      "step": 709,
      "training_step_time": 17.07818841934204
    },
    {
      "epoch": 6.771087646484375e-07,
      "grad_norm": 0.8363240361213684,
      "learning_rate": 1e-05,
      "loss": 0.1073,
      "step": 710
    },
    {
      "epoch": 6.771087646484375e-07,
      "model_forward_time": 5.6050238609313965,
      "step": 710
    },
    {
      "dynamics_loss_avg": 0.08171819839626551,
      "epoch": 6.771087646484375e-07,
      "step": 710
    },
    {
      "action_loss_avg": 0.035752090718597175,
      "epoch": 6.771087646484375e-07,
      "step": 710
    },
    {
      "epoch": 6.771087646484375e-07,
      "step": 710,
      "training_step_time": 17.088327407836914
    },
    {
      "epoch": 6.780624389648437e-07,
      "model_forward_time": 5.5971174240112305,
      "step": 711
    },
    {
      "epoch": 6.780624389648437e-07,
      "step": 711,
      "training_step_time": 17.04736876487732
    },
    {
      "epoch": 6.7901611328125e-07,
      "model_forward_time": 5.954291820526123,
      "step": 712
    },
    {
      "epoch": 6.7901611328125e-07,
      "step": 712,
      "training_step_time": 17.405055046081543
    },
    {
      "epoch": 6.799697875976562e-07,
      "model_forward_time": 5.599053859710693,
      "step": 713
    },
    {
      "epoch": 6.799697875976562e-07,
      "step": 713,
      "training_step_time": 17.070462465286255
    },
    {
      "epoch": 6.809234619140625e-07,
      "model_forward_time": 5.6088783740997314,
      "step": 714
    },
    {
      "epoch": 6.809234619140625e-07,
      "step": 714,
      "training_step_time": 17.09949779510498
    },
    {
      "epoch": 6.818771362304687e-07,
      "model_forward_time": 5.615526914596558,
      "step": 715
    },
    {
      "epoch": 6.818771362304687e-07,
      "step": 715,
      "training_step_time": 17.116480350494385
    },
    {
      "epoch": 6.82830810546875e-07,
      "model_forward_time": 5.605265855789185,
      "step": 716
    },
    {
      "epoch": 6.82830810546875e-07,
      "step": 716,
      "training_step_time": 17.092308282852173
    },
    {
      "epoch": 6.837844848632812e-07,
      "model_forward_time": 5.606391906738281,
      "step": 717
    },
    {
      "epoch": 6.837844848632812e-07,
      "step": 717,
      "training_step_time": 17.433713674545288
    },
    {
      "epoch": 6.847381591796875e-07,
      "model_forward_time": 5.608845233917236,
      "step": 718
    },
    {
      "epoch": 6.847381591796875e-07,
      "step": 718,
      "training_step_time": 17.089807510375977
    },
    {
      "epoch": 6.856918334960937e-07,
      "model_forward_time": 5.610620975494385,
      "step": 719
    },
    {
      "epoch": 6.856918334960937e-07,
      "step": 719,
      "training_step_time": 17.110357761383057
    },
    {
      "epoch": 6.866455078125e-07,
      "grad_norm": 2.1349527835845947,
      "learning_rate": 1e-05,
      "loss": 0.1262,
      "step": 720
    },
    {
      "epoch": 6.866455078125e-07,
      "model_forward_time": 5.619175672531128,
      "step": 720
    },
    {
      "dynamics_loss_avg": 0.07746376805007457,
      "epoch": 6.866455078125e-07,
      "step": 720
    },
    {
      "action_loss_avg": 0.03796137273311615,
      "epoch": 6.866455078125e-07,
      "step": 720
    },
    {
      "epoch": 6.866455078125e-07,
      "step": 720,
      "training_step_time": 17.11260199546814
    },
    {
      "epoch": 6.875991821289062e-07,
      "model_forward_time": 5.616736888885498,
      "step": 721
    },
    {
      "epoch": 6.875991821289062e-07,
      "step": 721,
      "training_step_time": 17.09899926185608
    },
    {
      "epoch": 6.885528564453125e-07,
      "model_forward_time": 5.610821723937988,
      "step": 722
    },
    {
      "epoch": 6.885528564453125e-07,
      "step": 722,
      "training_step_time": 17.076643705368042
    },
    {
      "epoch": 6.895065307617187e-07,
      "model_forward_time": 5.9562788009643555,
      "step": 723
    },
    {
      "epoch": 6.895065307617187e-07,
      "step": 723,
      "training_step_time": 17.408331632614136
    },
    {
      "epoch": 6.90460205078125e-07,
      "model_forward_time": 5.591896295547485,
      "step": 724
    },
    {
      "epoch": 6.90460205078125e-07,
      "step": 724,
      "training_step_time": 17.033397912979126
    },
    {
      "epoch": 6.914138793945312e-07,
      "model_forward_time": 5.5797929763793945,
      "step": 725
    },
    {
      "epoch": 6.914138793945312e-07,
      "step": 725,
      "training_step_time": 17.02791428565979
    },
    {
      "epoch": 6.923675537109375e-07,
      "model_forward_time": 5.594738245010376,
      "step": 726
    },
    {
      "epoch": 6.923675537109375e-07,
      "step": 726,
      "training_step_time": 17.061856508255005
    },
    {
      "epoch": 6.933212280273437e-07,
      "model_forward_time": 5.607733964920044,
      "step": 727
    },
    {
      "epoch": 6.933212280273437e-07,
      "step": 727,
      "training_step_time": 17.075291395187378
    },
    {
      "epoch": 6.9427490234375e-07,
      "model_forward_time": 5.595452547073364,
      "step": 728
    },
    {
      "epoch": 6.9427490234375e-07,
      "step": 728,
      "training_step_time": 17.417383909225464
    },
    {
      "epoch": 6.952285766601562e-07,
      "model_forward_time": 5.607696294784546,
      "step": 729
    },
    {
      "epoch": 6.952285766601562e-07,
      "step": 729,
      "training_step_time": 17.093711137771606
    },
    {
      "epoch": 6.961822509765625e-07,
      "grad_norm": 1.560908317565918,
      "learning_rate": 1e-05,
      "loss": 0.098,
      "step": 730
    },
    {
      "epoch": 6.961822509765625e-07,
      "model_forward_time": 5.6243486404418945,
      "step": 730
    },
    {
      "dynamics_loss_avg": 0.06193110439926386,
      "epoch": 6.961822509765625e-07,
      "step": 730
    },
    {
      "action_loss_avg": 0.03441426679491997,
      "epoch": 6.961822509765625e-07,
      "step": 730
    },
    {
      "epoch": 6.961822509765625e-07,
      "step": 730,
      "training_step_time": 17.12222456932068
    },
    {
      "epoch": 6.971359252929687e-07,
      "model_forward_time": 5.619488954544067,
      "step": 731
    },
    {
      "epoch": 6.971359252929687e-07,
      "step": 731,
      "training_step_time": 17.093684911727905
    },
    {
      "epoch": 6.98089599609375e-07,
      "model_forward_time": 5.6056437492370605,
      "step": 732
    },
    {
      "epoch": 6.98089599609375e-07,
      "step": 732,
      "training_step_time": 17.047173738479614
    },
    {
      "epoch": 6.990432739257812e-07,
      "model_forward_time": 5.605666160583496,
      "step": 733
    },
    {
      "epoch": 6.990432739257812e-07,
      "step": 733,
      "training_step_time": 17.05527901649475
    },
    {
      "epoch": 6.999969482421875e-07,
      "model_forward_time": 5.948263168334961,
      "step": 734
    },
    {
      "epoch": 6.999969482421875e-07,
      "step": 734,
      "training_step_time": 17.388531923294067
    },
    {
      "epoch": 7.009506225585937e-07,
      "model_forward_time": 5.596432447433472,
      "step": 735
    },
    {
      "epoch": 7.009506225585937e-07,
      "step": 735,
      "training_step_time": 17.048731803894043
    },
    {
      "epoch": 7.01904296875e-07,
      "model_forward_time": 5.610074758529663,
      "step": 736
    },
    {
      "epoch": 7.01904296875e-07,
      "step": 736,
      "training_step_time": 17.091645002365112
    },
    {
      "epoch": 7.028579711914062e-07,
      "model_forward_time": 5.601391077041626,
      "step": 737
    },
    {
      "epoch": 7.028579711914062e-07,
      "step": 737,
      "training_step_time": 17.07515835762024
    },
    {
      "epoch": 7.038116455078125e-07,
      "model_forward_time": 5.6033618450164795,
      "step": 738
    },
    {
      "epoch": 7.038116455078125e-07,
      "step": 738,
      "training_step_time": 17.07618236541748
    },
    {
      "epoch": 7.047653198242187e-07,
      "model_forward_time": 5.9668896198272705,
      "step": 739
    },
    {
      "epoch": 7.047653198242187e-07,
      "step": 739,
      "training_step_time": 17.4559006690979
    },
    {
      "epoch": 7.05718994140625e-07,
      "grad_norm": 2.7523200511932373,
      "learning_rate": 1e-05,
      "loss": 0.131,
      "step": 740
    },
    {
      "epoch": 7.05718994140625e-07,
      "model_forward_time": 5.612494230270386,
      "step": 740
    },
    {
      "dynamics_loss_avg": 0.08036640174686908,
      "epoch": 7.05718994140625e-07,
      "step": 740
    },
    {
      "action_loss_avg": 0.04501852821558714,
      "epoch": 7.05718994140625e-07,
      "step": 740
    },
    {
      "epoch": 7.05718994140625e-07,
      "step": 740,
      "training_step_time": 17.101440906524658
    },
    {
      "epoch": 7.066726684570312e-07,
      "model_forward_time": 5.618849754333496,
      "step": 741
    },
    {
      "epoch": 7.066726684570312e-07,
      "step": 741,
      "training_step_time": 17.106106758117676
    },
    {
      "epoch": 7.076263427734375e-07,
      "model_forward_time": 5.5972068309783936,
      "step": 742
    },
    {
      "epoch": 7.076263427734375e-07,
      "step": 742,
      "training_step_time": 17.02655267715454
    },
    {
      "epoch": 7.085800170898437e-07,
      "model_forward_time": 5.583998918533325,
      "step": 743
    },
    {
      "epoch": 7.085800170898437e-07,
      "step": 743,
      "training_step_time": 17.024015188217163
    },
    {
      "epoch": 7.0953369140625e-07,
      "model_forward_time": 5.584993124008179,
      "step": 744
    },
    {
      "epoch": 7.0953369140625e-07,
      "step": 744,
      "training_step_time": 17.02666711807251
    },
    {
      "epoch": 7.104873657226562e-07,
      "model_forward_time": 5.9504311084747314,
      "step": 745
    },
    {
      "epoch": 7.104873657226562e-07,
      "step": 745,
      "training_step_time": 17.391372203826904
    },
    {
      "epoch": 7.114410400390625e-07,
      "model_forward_time": 5.587895393371582,
      "step": 746
    },
    {
      "epoch": 7.114410400390625e-07,
      "step": 746,
      "training_step_time": 17.05859398841858
    },
    {
      "epoch": 7.123947143554687e-07,
      "model_forward_time": 5.606567859649658,
      "step": 747
    },
    {
      "epoch": 7.123947143554687e-07,
      "step": 747,
      "training_step_time": 17.095011234283447
    },
    {
      "epoch": 7.13348388671875e-07,
      "model_forward_time": 5.617410898208618,
      "step": 748
    },
    {
      "epoch": 7.13348388671875e-07,
      "step": 748,
      "training_step_time": 17.122191905975342
    },
    {
      "epoch": 7.143020629882812e-07,
      "model_forward_time": 5.626811265945435,
      "step": 749
    },
    {
      "epoch": 7.143020629882812e-07,
      "step": 749,
      "training_step_time": 17.111087560653687
    },
    {
      "epoch": 7.152557373046875e-07,
      "grad_norm": 0.5464767217636108,
      "learning_rate": 1e-05,
      "loss": 0.0764,
      "step": 750
    },
    {
      "epoch": 7.152557373046875e-07,
      "model_forward_time": 5.601794958114624,
      "step": 750
    },
    {
      "dynamics_loss_avg": 0.05692431665956974,
      "epoch": 7.152557373046875e-07,
      "step": 750
    },
    {
      "action_loss_avg": 0.026801869133487345,
      "epoch": 7.152557373046875e-07,
      "step": 750
    },
    {
      "epoch": 7.152557373046875e-07,
      "step": 750,
      "training_step_time": 17.412841081619263
    },
    {
      "epoch": 7.162094116210938e-07,
      "model_forward_time": 5.583707332611084,
      "step": 751
    },
    {
      "epoch": 7.162094116210938e-07,
      "step": 751,
      "training_step_time": 17.03983974456787
    },
    {
      "epoch": 7.171630859375e-07,
      "model_forward_time": 5.587002992630005,
      "step": 752
    },
    {
      "epoch": 7.171630859375e-07,
      "step": 752,
      "training_step_time": 17.042244911193848
    },
    {
      "epoch": 7.181167602539063e-07,
      "model_forward_time": 5.589303016662598,
      "step": 753
    },
    {
      "epoch": 7.181167602539063e-07,
      "step": 753,
      "training_step_time": 17.051620483398438
    },
    {
      "epoch": 7.190704345703125e-07,
      "model_forward_time": 5.596238136291504,
      "step": 754
    },
    {
      "epoch": 7.190704345703125e-07,
      "step": 754,
      "training_step_time": 17.08280634880066
    },
    {
      "epoch": 7.200241088867188e-07,
      "model_forward_time": 5.612109422683716,
      "step": 755
    },
    {
      "epoch": 7.200241088867188e-07,
      "step": 755,
      "training_step_time": 17.113573789596558
    },
    {
      "epoch": 7.20977783203125e-07,
      "model_forward_time": 5.9785261154174805,
      "step": 756
    },
    {
      "epoch": 7.20977783203125e-07,
      "step": 756,
      "training_step_time": 17.46280837059021
    },
    {
      "epoch": 7.219314575195313e-07,
      "model_forward_time": 5.619527816772461,
      "step": 757
    },
    {
      "epoch": 7.219314575195313e-07,
      "step": 757,
      "training_step_time": 17.09497833251953
    },
    {
      "epoch": 7.228851318359375e-07,
      "model_forward_time": 5.605347633361816,
      "step": 758
    },
    {
      "epoch": 7.228851318359375e-07,
      "step": 758,
      "training_step_time": 17.05769109725952
    },
    {
      "epoch": 7.238388061523438e-07,
      "model_forward_time": 5.59577488899231,
      "step": 759
    },
    {
      "epoch": 7.238388061523438e-07,
      "step": 759,
      "training_step_time": 17.068313360214233
    },
    {
      "epoch": 7.2479248046875e-07,
      "grad_norm": 0.7871508598327637,
      "learning_rate": 1e-05,
      "loss": 0.0844,
      "step": 760
    },
    {
      "epoch": 7.2479248046875e-07,
      "model_forward_time": 5.599776983261108,
      "step": 760
    },
    {
      "dynamics_loss_avg": 0.059533681720495224,
      "epoch": 7.2479248046875e-07,
      "step": 760
    },
    {
      "action_loss_avg": 0.027385299652814867,
      "epoch": 7.2479248046875e-07,
      "step": 760
    },
    {
      "epoch": 7.2479248046875e-07,
      "step": 760,
      "training_step_time": 17.0559504032135
    },
    {
      "epoch": 7.257461547851563e-07,
      "model_forward_time": 5.9380104541778564,
      "step": 761
    },
    {
      "epoch": 7.257461547851563e-07,
      "step": 761,
      "training_step_time": 17.416892766952515
    },
    {
      "epoch": 7.266998291015625e-07,
      "model_forward_time": 5.599855661392212,
      "step": 762
    },
    {
      "epoch": 7.266998291015625e-07,
      "step": 762,
      "training_step_time": 17.083876848220825
    },
    {
      "epoch": 7.276535034179688e-07,
      "model_forward_time": 5.625086069107056,
      "step": 763
    },
    {
      "epoch": 7.276535034179688e-07,
      "step": 763,
      "training_step_time": 17.11123490333557
    },
    {
      "epoch": 7.28607177734375e-07,
      "model_forward_time": 5.617228746414185,
      "step": 764
    },
    {
      "epoch": 7.28607177734375e-07,
      "step": 764,
      "training_step_time": 17.11146593093872
    },
    {
      "epoch": 7.295608520507813e-07,
      "model_forward_time": 5.609588384628296,
      "step": 765
    },
    {
      "epoch": 7.295608520507813e-07,
      "step": 765,
      "training_step_time": 17.110547304153442
    },
    {
      "epoch": 7.305145263671875e-07,
      "model_forward_time": 5.605602502822876,
      "step": 766
    },
    {
      "epoch": 7.305145263671875e-07,
      "step": 766,
      "training_step_time": 17.081072330474854
    },
    {
      "epoch": 7.314682006835938e-07,
      "model_forward_time": 5.972374439239502,
      "step": 767
    },
    {
      "epoch": 7.314682006835938e-07,
      "step": 767,
      "training_step_time": 17.434014558792114
    },
    {
      "epoch": 7.32421875e-07,
      "model_forward_time": 5.602832078933716,
      "step": 768
    },
    {
      "epoch": 7.32421875e-07,
      "step": 768,
      "training_step_time": 17.07427477836609
    },
    {
      "epoch": 7.333755493164063e-07,
      "model_forward_time": 5.603912830352783,
      "step": 769
    },
    {
      "epoch": 7.333755493164063e-07,
      "step": 769,
      "training_step_time": 17.081183910369873
    },
    {
      "epoch": 7.343292236328125e-07,
      "grad_norm": 1.220334529876709,
      "learning_rate": 1e-05,
      "loss": 0.1048,
      "step": 770
    },
    {
      "epoch": 7.343292236328125e-07,
      "model_forward_time": 5.607928991317749,
      "step": 770
    },
    {
      "dynamics_loss_avg": 0.0667471831664443,
      "epoch": 7.343292236328125e-07,
      "step": 770
    },
    {
      "action_loss_avg": 0.029329240042716264,
      "epoch": 7.343292236328125e-07,
      "step": 770
    },
    {
      "epoch": 7.343292236328125e-07,
      "step": 770,
      "training_step_time": 17.060401678085327
    },
    {
      "epoch": 7.352828979492188e-07,
      "model_forward_time": 5.605108976364136,
      "step": 771
    },
    {
      "epoch": 7.352828979492188e-07,
      "step": 771,
      "training_step_time": 17.0776629447937
    },
    {
      "epoch": 7.36236572265625e-07,
      "model_forward_time": 5.958683013916016,
      "step": 772
    },
    {
      "epoch": 7.36236572265625e-07,
      "step": 772,
      "training_step_time": 17.441951751708984
    },
    {
      "epoch": 7.371902465820313e-07,
      "model_forward_time": 5.6137895584106445,
      "step": 773
    },
    {
      "epoch": 7.371902465820313e-07,
      "step": 773,
      "training_step_time": 17.11710214614868
    },
    {
      "epoch": 7.381439208984375e-07,
      "model_forward_time": 5.6095662117004395,
      "step": 774
    },
    {
      "epoch": 7.381439208984375e-07,
      "step": 774,
      "training_step_time": 17.08582067489624
    },
    {
      "epoch": 7.390975952148438e-07,
      "model_forward_time": 5.606618881225586,
      "step": 775
    },
    {
      "epoch": 7.390975952148438e-07,
      "step": 775,
      "training_step_time": 17.086655139923096
    },
    {
      "epoch": 7.4005126953125e-07,
      "model_forward_time": 5.603890657424927,
      "step": 776
    },
    {
      "epoch": 7.4005126953125e-07,
      "step": 776,
      "training_step_time": 17.06636643409729
    },
    {
      "epoch": 7.410049438476563e-07,
      "model_forward_time": 5.597526788711548,
      "step": 777
    },
    {
      "epoch": 7.410049438476563e-07,
      "step": 777,
      "training_step_time": 17.453892707824707
    },
    {
      "epoch": 7.419586181640625e-07,
      "model_forward_time": 5.60637903213501,
      "step": 778
    },
    {
      "epoch": 7.419586181640625e-07,
      "step": 778,
      "training_step_time": 17.073435068130493
    },
    {
      "epoch": 7.429122924804688e-07,
      "model_forward_time": 5.594564914703369,
      "step": 779
    },
    {
      "epoch": 7.429122924804688e-07,
      "step": 779,
      "training_step_time": 17.060499906539917
    },
    {
      "epoch": 7.43865966796875e-07,
      "grad_norm": 3.1355950832366943,
      "learning_rate": 1e-05,
      "loss": 0.0965,
      "step": 780
    },
    {
      "epoch": 7.43865966796875e-07,
      "model_forward_time": 5.596926927566528,
      "step": 780
    },
    {
      "dynamics_loss_avg": 0.07191886436194181,
      "epoch": 7.43865966796875e-07,
      "step": 780
    },
    {
      "action_loss_avg": 0.030219715926796197,
      "epoch": 7.43865966796875e-07,
      "step": 780
    },
    {
      "epoch": 7.43865966796875e-07,
      "step": 780,
      "training_step_time": 17.080535411834717
    },
    {
      "epoch": 7.448196411132813e-07,
      "model_forward_time": 5.592758655548096,
      "step": 781
    },
    {
      "epoch": 7.448196411132813e-07,
      "step": 781,
      "training_step_time": 17.068042993545532
    },
    {
      "epoch": 7.457733154296875e-07,
      "model_forward_time": 5.618113994598389,
      "step": 782
    },
    {
      "epoch": 7.457733154296875e-07,
      "step": 782,
      "training_step_time": 17.10146689414978
    },
    {
      "epoch": 7.467269897460938e-07,
      "model_forward_time": 5.9492409229278564,
      "step": 783
    },
    {
      "epoch": 7.467269897460938e-07,
      "step": 783,
      "training_step_time": 17.389383554458618
    },
    {
      "epoch": 7.476806640625e-07,
      "model_forward_time": 5.580016374588013,
      "step": 784
    },
    {
      "epoch": 7.476806640625e-07,
      "step": 784,
      "training_step_time": 16.990076780319214
    },
    {
      "epoch": 7.486343383789063e-07,
      "model_forward_time": 5.576340675354004,
      "step": 785
    },
    {
      "epoch": 7.486343383789063e-07,
      "step": 785,
      "training_step_time": 17.014781951904297
    },
    {
      "epoch": 7.495880126953125e-07,
      "model_forward_time": 5.5823071002960205,
      "step": 786
    },
    {
      "epoch": 7.495880126953125e-07,
      "step": 786,
      "training_step_time": 17.029335975646973
    },
    {
      "epoch": 7.505416870117188e-07,
      "model_forward_time": 5.598050832748413,
      "step": 787
    },
    {
      "epoch": 7.505416870117188e-07,
      "step": 787,
      "training_step_time": 17.06958031654358
    },
    {
      "epoch": 7.51495361328125e-07,
      "model_forward_time": 5.606089115142822,
      "step": 788
    },
    {
      "epoch": 7.51495361328125e-07,
      "step": 788,
      "training_step_time": 17.456279516220093
    },
    {
      "epoch": 7.524490356445313e-07,
      "model_forward_time": 5.611734628677368,
      "step": 789
    },
    {
      "epoch": 7.524490356445313e-07,
      "step": 789,
      "training_step_time": 17.12348771095276
    },
    {
      "epoch": 7.534027099609375e-07,
      "grad_norm": 0.33809995651245117,
      "learning_rate": 1e-05,
      "loss": 0.0799,
      "step": 790
    },
    {
      "epoch": 7.534027099609375e-07,
      "model_forward_time": 5.6239564418792725,
      "step": 790
    },
    {
      "dynamics_loss_avg": 0.05473757777363062,
      "epoch": 7.534027099609375e-07,
      "step": 790
    },
    {
      "action_loss_avg": 0.02556721302680671,
      "epoch": 7.534027099609375e-07,
      "step": 790
    },
    {
      "epoch": 7.534027099609375e-07,
      "step": 790,
      "training_step_time": 17.129401445388794
    },
    {
      "epoch": 7.543563842773438e-07,
      "model_forward_time": 5.616208791732788,
      "step": 791
    },
    {
      "epoch": 7.543563842773438e-07,
      "step": 791,
      "training_step_time": 17.121209144592285
    },
    {
      "epoch": 7.5531005859375e-07,
      "model_forward_time": 5.617938995361328,
      "step": 792
    },
    {
      "epoch": 7.5531005859375e-07,
      "step": 792,
      "training_step_time": 17.08376979827881
    },
    {
      "epoch": 7.562637329101563e-07,
      "model_forward_time": 5.595463752746582,
      "step": 793
    },
    {
      "epoch": 7.562637329101563e-07,
      "step": 793,
      "training_step_time": 17.0308780670166
    },
    {
      "epoch": 7.572174072265625e-07,
      "model_forward_time": 5.929090261459351,
      "step": 794
    },
    {
      "epoch": 7.572174072265625e-07,
      "step": 794,
      "training_step_time": 17.331892490386963
    },
    {
      "epoch": 7.581710815429688e-07,
      "model_forward_time": 5.587334394454956,
      "step": 795
    },
    {
      "epoch": 7.581710815429688e-07,
      "step": 795,
      "training_step_time": 17.04661989212036
    },
    {
      "epoch": 7.59124755859375e-07,
      "model_forward_time": 5.592158794403076,
      "step": 796
    },
    {
      "epoch": 7.59124755859375e-07,
      "step": 796,
      "training_step_time": 17.031882524490356
    },
    {
      "epoch": 7.600784301757813e-07,
      "model_forward_time": 5.6029040813446045,
      "step": 797
    },
    {
      "epoch": 7.600784301757813e-07,
      "step": 797,
      "training_step_time": 17.096890926361084
    },
    {
      "epoch": 7.610321044921875e-07,
      "model_forward_time": 5.611914157867432,
      "step": 798
    },
    {
      "epoch": 7.610321044921875e-07,
      "step": 798,
      "training_step_time": 17.097238302230835
    },
    {
      "epoch": 7.619857788085938e-07,
      "model_forward_time": 5.984187126159668,
      "step": 799
    },
    {
      "epoch": 7.619857788085938e-07,
      "step": 799,
      "training_step_time": 17.449029684066772
    },
    {
      "epoch": 7.62939453125e-07,
      "grad_norm": 1.2035728693008423,
      "learning_rate": 1e-05,
      "loss": 0.1002,
      "step": 800
    },
    {
      "epoch": 7.62939453125e-07,
      "model_forward_time": 5.5864644050598145,
      "step": 800
    },
    {
      "dynamics_loss_avg": 0.06822140384465455,
      "epoch": 7.62939453125e-07,
      "step": 800
    },
    {
      "action_loss_avg": 0.02892828909680247,
      "epoch": 7.62939453125e-07,
      "step": 800
    },
    {
      "epoch": 7.62939453125e-07,
      "step": 800,
      "training_step_time": 17.011424779891968
    },
    {
      "epoch": 7.638931274414063e-07,
      "model_forward_time": 5.582099437713623,
      "step": 801
    },
    {
      "epoch": 7.638931274414063e-07,
      "step": 801,
      "training_step_time": 17.00499987602234
    },
    {
      "epoch": 7.648468017578125e-07,
      "model_forward_time": 5.591078042984009,
      "step": 802
    },
    {
      "epoch": 7.648468017578125e-07,
      "step": 802,
      "training_step_time": 17.0648295879364
    },
    {
      "epoch": 7.658004760742188e-07,
      "model_forward_time": 5.607881784439087,
      "step": 803
    },
    {
      "epoch": 7.658004760742188e-07,
      "step": 803,
      "training_step_time": 17.115784645080566
    },
    {
      "epoch": 7.66754150390625e-07,
      "model_forward_time": 5.618537664413452,
      "step": 804
    },
    {
      "epoch": 7.66754150390625e-07,
      "step": 804,
      "training_step_time": 17.476325273513794
    },
    {
      "epoch": 7.677078247070313e-07,
      "model_forward_time": 5.616516351699829,
      "step": 805
    },
    {
      "epoch": 7.677078247070313e-07,
      "step": 805,
      "training_step_time": 17.12545657157898
    },
    {
      "epoch": 7.686614990234375e-07,
      "model_forward_time": 5.6185219287872314,
      "step": 806
    },
    {
      "epoch": 7.686614990234375e-07,
      "step": 806,
      "training_step_time": 17.098362922668457
    },
    {
      "epoch": 7.696151733398438e-07,
      "model_forward_time": 5.606277942657471,
      "step": 807
    },
    {
      "epoch": 7.696151733398438e-07,
      "step": 807,
      "training_step_time": 17.077392578125
    },
    {
      "epoch": 7.7056884765625e-07,
      "model_forward_time": 5.600022315979004,
      "step": 808
    },
    {
      "epoch": 7.7056884765625e-07,
      "step": 808,
      "training_step_time": 17.050915718078613
    },
    {
      "epoch": 7.715225219726563e-07,
      "model_forward_time": 5.603508949279785,
      "step": 809
    },
    {
      "epoch": 7.715225219726563e-07,
      "step": 809,
      "training_step_time": 17.061828136444092
    },
    {
      "epoch": 7.724761962890626e-07,
      "grad_norm": 2.482452869415283,
      "learning_rate": 1e-05,
      "loss": 0.1116,
      "step": 810
    },
    {
      "epoch": 7.724761962890626e-07,
      "model_forward_time": 5.98170280456543,
      "step": 810
    },
    {
      "dynamics_loss_avg": 0.07485683634877205,
      "epoch": 7.724761962890626e-07,
      "step": 810
    },
    {
      "action_loss_avg": 0.03894344121217728,
      "epoch": 7.724761962890626e-07,
      "step": 810
    },
    {
      "epoch": 7.724761962890626e-07,
      "step": 810,
      "training_step_time": 17.466505765914917
    },
    {
      "epoch": 7.734298706054688e-07,
      "model_forward_time": 5.620690584182739,
      "step": 811
    },
    {
      "epoch": 7.734298706054688e-07,
      "step": 811,
      "training_step_time": 17.12797498703003
    },
    {
      "epoch": 7.74383544921875e-07,
      "model_forward_time": 5.634830474853516,
      "step": 812
    },
    {
      "epoch": 7.74383544921875e-07,
      "step": 812,
      "training_step_time": 17.124027013778687
    },
    {
      "epoch": 7.753372192382812e-07,
      "model_forward_time": 5.604724168777466,
      "step": 813
    },
    {
      "epoch": 7.753372192382812e-07,
      "step": 813,
      "training_step_time": 17.0576753616333
    },
    {
      "epoch": 7.762908935546874e-07,
      "model_forward_time": 5.594561338424683,
      "step": 814
    },
    {
      "epoch": 7.762908935546874e-07,
      "step": 814,
      "training_step_time": 16.999784231185913
    },
    {
      "epoch": 7.772445678710937e-07,
      "model_forward_time": 5.936705827713013,
      "step": 815
    },
    {
      "epoch": 7.772445678710937e-07,
      "step": 815,
      "training_step_time": 17.370378971099854
    },
    {
      "epoch": 7.781982421875e-07,
      "model_forward_time": 5.588096857070923,
      "step": 816
    },
    {
      "epoch": 7.781982421875e-07,
      "step": 816,
      "training_step_time": 17.014960050582886
    },
    {
      "epoch": 7.791519165039062e-07,
      "model_forward_time": 5.589982748031616,
      "step": 817
    },
    {
      "epoch": 7.791519165039062e-07,
      "step": 817,
      "training_step_time": 16.99368667602539
    },
    {
      "epoch": 7.801055908203125e-07,
      "model_forward_time": 5.594781160354614,
      "step": 818
    },
    {
      "epoch": 7.801055908203125e-07,
      "step": 818,
      "training_step_time": 17.03668522834778
    },
    {
      "epoch": 7.810592651367187e-07,
      "model_forward_time": 5.5991082191467285,
      "step": 819
    },
    {
      "epoch": 7.810592651367187e-07,
      "step": 819,
      "training_step_time": 17.070433855056763
    },
    {
      "epoch": 7.82012939453125e-07,
      "grad_norm": 1.5502674579620361,
      "learning_rate": 1e-05,
      "loss": 0.0849,
      "step": 820
    },
    {
      "epoch": 7.82012939453125e-07,
      "model_forward_time": 5.6100547313690186,
      "step": 820
    },
    {
      "dynamics_loss_avg": 0.05940895043313503,
      "epoch": 7.82012939453125e-07,
      "step": 820
    },
    {
      "action_loss_avg": 0.027787596825510262,
      "epoch": 7.82012939453125e-07,
      "step": 820
    },
    {
      "epoch": 7.82012939453125e-07,
      "step": 820,
      "training_step_time": 17.089044332504272
    },
    {
      "epoch": 7.829666137695312e-07,
      "model_forward_time": 5.967227935791016,
      "step": 821
    },
    {
      "epoch": 7.829666137695312e-07,
      "step": 821,
      "training_step_time": 17.42306160926819
    },
    {
      "epoch": 7.839202880859375e-07,
      "model_forward_time": 5.597430467605591,
      "step": 822
    },
    {
      "epoch": 7.839202880859375e-07,
      "step": 822,
      "training_step_time": 17.048937797546387
    },
    {
      "epoch": 7.848739624023437e-07,
      "model_forward_time": 5.585299491882324,
      "step": 823
    },
    {
      "epoch": 7.848739624023437e-07,
      "step": 823,
      "training_step_time": 17.022182941436768
    },
    {
      "epoch": 7.8582763671875e-07,
      "model_forward_time": 5.589641809463501,
      "step": 824
    },
    {
      "epoch": 7.8582763671875e-07,
      "step": 824,
      "training_step_time": 17.05750560760498
    },
    {
      "epoch": 7.867813110351562e-07,
      "model_forward_time": 5.615913391113281,
      "step": 825
    },
    {
      "epoch": 7.867813110351562e-07,
      "step": 825,
      "training_step_time": 17.101539373397827
    },
    {
      "epoch": 7.877349853515625e-07,
      "model_forward_time": 5.9671618938446045,
      "step": 826
    },
    {
      "epoch": 7.877349853515625e-07,
      "step": 826,
      "training_step_time": 17.45640730857849
    },
    {
      "epoch": 7.886886596679687e-07,
      "model_forward_time": 5.611545085906982,
      "step": 827
    },
    {
      "epoch": 7.886886596679687e-07,
      "step": 827,
      "training_step_time": 17.1007559299469
    },
    {
      "epoch": 7.89642333984375e-07,
      "model_forward_time": 5.617573499679565,
      "step": 828
    },
    {
      "epoch": 7.89642333984375e-07,
      "step": 828,
      "training_step_time": 17.118673086166382
    },
    {
      "epoch": 7.905960083007812e-07,
      "model_forward_time": 5.615434408187866,
      "step": 829
    },
    {
      "epoch": 7.905960083007812e-07,
      "step": 829,
      "training_step_time": 17.116380214691162
    },
    {
      "epoch": 7.915496826171875e-07,
      "grad_norm": 1.4394227266311646,
      "learning_rate": 1e-05,
      "loss": 0.1035,
      "step": 830
    },
    {
      "epoch": 7.915496826171875e-07,
      "model_forward_time": 5.610934495925903,
      "step": 830
    },
    {
      "dynamics_loss_avg": 0.07014708090573549,
      "epoch": 7.915496826171875e-07,
      "step": 830
    },
    {
      "action_loss_avg": 0.03303912999108434,
      "epoch": 7.915496826171875e-07,
      "step": 830
    },
    {
      "epoch": 7.915496826171875e-07,
      "step": 830,
      "training_step_time": 17.08280611038208
    },
    {
      "epoch": 7.925033569335937e-07,
      "model_forward_time": 5.606256484985352,
      "step": 831
    },
    {
      "epoch": 7.925033569335937e-07,
      "step": 831,
      "training_step_time": 17.076392650604248
    },
    {
      "epoch": 7.9345703125e-07,
      "model_forward_time": 5.954892158508301,
      "step": 832
    },
    {
      "epoch": 7.9345703125e-07,
      "step": 832,
      "training_step_time": 17.39205813407898
    },
    {
      "epoch": 7.944107055664062e-07,
      "model_forward_time": 5.5997724533081055,
      "step": 833
    },
    {
      "epoch": 7.944107055664062e-07,
      "step": 833,
      "training_step_time": 17.086854934692383
    },
    {
      "epoch": 7.953643798828125e-07,
      "model_forward_time": 5.604699611663818,
      "step": 834
    },
    {
      "epoch": 7.953643798828125e-07,
      "step": 834,
      "training_step_time": 17.079643487930298
    },
    {
      "epoch": 7.963180541992187e-07,
      "model_forward_time": 5.608872175216675,
      "step": 835
    },
    {
      "epoch": 7.963180541992187e-07,
      "step": 835,
      "training_step_time": 17.070123195648193
    },
    {
      "epoch": 7.97271728515625e-07,
      "model_forward_time": 5.5973920822143555,
      "step": 836
    },
    {
      "epoch": 7.97271728515625e-07,
      "step": 836,
      "training_step_time": 17.064836978912354
    },
    {
      "epoch": 7.982254028320312e-07,
      "model_forward_time": 5.9268798828125,
      "step": 837
    },
    {
      "epoch": 7.982254028320312e-07,
      "step": 837,
      "training_step_time": 17.395902156829834
    },
    {
      "epoch": 7.991790771484375e-07,
      "model_forward_time": 5.601691484451294,
      "step": 838
    },
    {
      "epoch": 7.991790771484375e-07,
      "step": 838,
      "training_step_time": 17.085538148880005
    },
    {
      "epoch": 8.001327514648437e-07,
      "model_forward_time": 5.60843300819397,
      "step": 839
    },
    {
      "epoch": 8.001327514648437e-07,
      "step": 839,
      "training_step_time": 17.108935832977295
    },
    {
      "epoch": 8.0108642578125e-07,
      "grad_norm": 1.8224868774414062,
      "learning_rate": 1e-05,
      "loss": 0.1195,
      "step": 840
    },
    {
      "epoch": 8.0108642578125e-07,
      "model_forward_time": 5.623893976211548,
      "step": 840
    },
    {
      "dynamics_loss_avg": 0.08310736753046513,
      "epoch": 8.0108642578125e-07,
      "step": 840
    },
    {
      "action_loss_avg": 0.0312357597053051,
      "epoch": 8.0108642578125e-07,
      "step": 840
    },
    {
      "epoch": 8.0108642578125e-07,
      "step": 840,
      "training_step_time": 17.130168199539185
    },
    {
      "epoch": 8.020401000976562e-07,
      "model_forward_time": 5.619497776031494,
      "step": 841
    },
    {
      "epoch": 8.020401000976562e-07,
      "step": 841,
      "training_step_time": 17.099809408187866
    },
    {
      "epoch": 8.029937744140625e-07,
      "model_forward_time": 5.601051568984985,
      "step": 842
    },
    {
      "epoch": 8.029937744140625e-07,
      "step": 842,
      "training_step_time": 17.393271923065186
    },
    {
      "epoch": 8.039474487304687e-07,
      "model_forward_time": 5.581958770751953,
      "step": 843
    },
    {
      "epoch": 8.039474487304687e-07,
      "step": 843,
      "training_step_time": 16.995506763458252
    },
    {
      "epoch": 8.04901123046875e-07,
      "model_forward_time": 5.583328485488892,
      "step": 844
    },
    {
      "epoch": 8.04901123046875e-07,
      "step": 844,
      "training_step_time": 17.00540566444397
    },
    {
      "epoch": 8.058547973632812e-07,
      "model_forward_time": 5.584792613983154,
      "step": 845
    },
    {
      "epoch": 8.058547973632812e-07,
      "step": 845,
      "training_step_time": 17.02918314933777
    },
    {
      "epoch": 8.068084716796875e-07,
      "model_forward_time": 5.5850560665130615,
      "step": 846
    },
    {
      "epoch": 8.068084716796875e-07,
      "step": 846,
      "training_step_time": 17.01979160308838
    },
    {
      "epoch": 8.077621459960937e-07,
      "model_forward_time": 5.590533971786499,
      "step": 847
    },
    {
      "epoch": 8.077621459960937e-07,
      "step": 847,
      "training_step_time": 17.01651406288147
    },
    {
      "epoch": 8.087158203125e-07,
      "model_forward_time": 5.9327850341796875,
      "step": 848
    },
    {
      "epoch": 8.087158203125e-07,
      "step": 848,
      "training_step_time": 17.39974570274353
    },
    {
      "epoch": 8.096694946289062e-07,
      "model_forward_time": 5.610834121704102,
      "step": 849
    },
    {
      "epoch": 8.096694946289062e-07,
      "step": 849,
      "training_step_time": 17.117876291275024
    },
    {
      "epoch": 8.106231689453125e-07,
      "grad_norm": 2.331690549850464,
      "learning_rate": 1e-05,
      "loss": 0.1098,
      "step": 850
    },
    {
      "epoch": 8.106231689453125e-07,
      "model_forward_time": 5.620593070983887,
      "step": 850
    },
    {
      "dynamics_loss_avg": 0.07759007811546326,
      "epoch": 8.106231689453125e-07,
      "step": 850
    },
    {
      "action_loss_avg": 0.03721771035343409,
      "epoch": 8.106231689453125e-07,
      "step": 850
    },
    {
      "epoch": 8.106231689453125e-07,
      "step": 850,
      "training_step_time": 17.12504553794861
    },
    {
      "epoch": 8.115768432617187e-07,
      "model_forward_time": 5.617682695388794,
      "step": 851
    },
    {
      "epoch": 8.115768432617187e-07,
      "step": 851,
      "training_step_time": 17.11064624786377
    },
    {
      "epoch": 8.12530517578125e-07,
      "model_forward_time": 5.61461067199707,
      "step": 852
    },
    {
      "epoch": 8.12530517578125e-07,
      "step": 852,
      "training_step_time": 17.104719877243042
    },
    {
      "epoch": 8.134841918945312e-07,
      "model_forward_time": 5.9766013622283936,
      "step": 853
    },
    {
      "epoch": 8.134841918945312e-07,
      "step": 853,
      "training_step_time": 17.462162494659424
    },
    {
      "epoch": 8.144378662109375e-07,
      "model_forward_time": 5.61545467376709,
      "step": 854
    },
    {
      "epoch": 8.144378662109375e-07,
      "step": 854,
      "training_step_time": 17.109113693237305
    },
    {
      "epoch": 8.153915405273437e-07,
      "model_forward_time": 5.619632244110107,
      "step": 855
    },
    {
      "epoch": 8.153915405273437e-07,
      "step": 855,
      "training_step_time": 17.085227012634277
    },
    {
      "epoch": 8.1634521484375e-07,
      "model_forward_time": 5.602182149887085,
      "step": 856
    },
    {
      "epoch": 8.1634521484375e-07,
      "step": 856,
      "training_step_time": 17.045753955841064
    },
    {
      "epoch": 8.172988891601562e-07,
      "model_forward_time": 5.589383602142334,
      "step": 857
    },
    {
      "epoch": 8.172988891601562e-07,
      "step": 857,
      "training_step_time": 17.0310640335083
    },
    {
      "epoch": 8.182525634765625e-07,
      "model_forward_time": 5.595195055007935,
      "step": 858
    },
    {
      "epoch": 8.182525634765625e-07,
      "step": 858,
      "training_step_time": 17.06033182144165
    },
    {
      "epoch": 8.192062377929687e-07,
      "model_forward_time": 5.953321933746338,
      "step": 859
    },
    {
      "epoch": 8.192062377929687e-07,
      "step": 859,
      "training_step_time": 17.444684982299805
    },
    {
      "epoch": 8.20159912109375e-07,
      "grad_norm": 2.594183921813965,
      "learning_rate": 1e-05,
      "loss": 0.1071,
      "step": 860
    },
    {
      "epoch": 8.20159912109375e-07,
      "model_forward_time": 5.622119903564453,
      "step": 860
    },
    {
      "dynamics_loss_avg": 0.0811663568019867,
      "epoch": 8.20159912109375e-07,
      "step": 860
    },
    {
      "action_loss_avg": 0.03286467166617513,
      "epoch": 8.20159912109375e-07,
      "step": 860
    },
    {
      "epoch": 8.20159912109375e-07,
      "step": 860,
      "training_step_time": 17.138779163360596
    },
    {
      "epoch": 8.211135864257812e-07,
      "model_forward_time": 5.6249473094940186,
      "step": 861
    },
    {
      "epoch": 8.211135864257812e-07,
      "step": 861,
      "training_step_time": 17.117680072784424
    },
    {
      "epoch": 8.220672607421875e-07,
      "model_forward_time": 5.606762886047363,
      "step": 862
    },
    {
      "epoch": 8.220672607421875e-07,
      "step": 862,
      "training_step_time": 17.053048133850098
    },
    {
      "epoch": 8.230209350585937e-07,
      "model_forward_time": 5.5833611488342285,
      "step": 863
    },
    {
      "epoch": 8.230209350585937e-07,
      "step": 863,
      "training_step_time": 17.030881643295288
    },
    {
      "epoch": 8.23974609375e-07,
      "model_forward_time": 5.949687242507935,
      "step": 864
    },
    {
      "epoch": 8.23974609375e-07,
      "step": 864,
      "training_step_time": 17.41964340209961
    },
    {
      "epoch": 8.249282836914062e-07,
      "model_forward_time": 5.616888046264648,
      "step": 865
    },
    {
      "epoch": 8.249282836914062e-07,
      "step": 865,
      "training_step_time": 17.12841486930847
    },
    {
      "epoch": 8.258819580078125e-07,
      "model_forward_time": 5.625107288360596,
      "step": 866
    },
    {
      "epoch": 8.258819580078125e-07,
      "step": 866,
      "training_step_time": 17.11311650276184
    },
    {
      "epoch": 8.268356323242187e-07,
      "model_forward_time": 5.617733001708984,
      "step": 867
    },
    {
      "epoch": 8.268356323242187e-07,
      "step": 867,
      "training_step_time": 17.106683254241943
    },
    {
      "epoch": 8.27789306640625e-07,
      "model_forward_time": 5.613573789596558,
      "step": 868
    },
    {
      "epoch": 8.27789306640625e-07,
      "step": 868,
      "training_step_time": 17.101664066314697
    },
    {
      "epoch": 8.287429809570312e-07,
      "model_forward_time": 5.611528635025024,
      "step": 869
    },
    {
      "epoch": 8.287429809570312e-07,
      "step": 869,
      "training_step_time": 17.101659297943115
    },
    {
      "epoch": 8.296966552734375e-07,
      "grad_norm": 0.30871763825416565,
      "learning_rate": 1e-05,
      "loss": 0.1205,
      "step": 870
    },
    {
      "epoch": 8.296966552734375e-07,
      "model_forward_time": 5.9471824169158936,
      "step": 870
    },
    {
      "dynamics_loss_avg": 0.08097790032625199,
      "epoch": 8.296966552734375e-07,
      "step": 870
    },
    {
      "action_loss_avg": 0.03529223469085992,
      "epoch": 8.296966552734375e-07,
      "step": 870
    },
    {
      "epoch": 8.296966552734375e-07,
      "step": 870,
      "training_step_time": 17.41843032836914
    },
    {
      "epoch": 8.306503295898437e-07,
      "model_forward_time": 5.592171669006348,
      "step": 871
    },
    {
      "epoch": 8.306503295898437e-07,
      "step": 871,
      "training_step_time": 17.04190230369568
    },
    {
      "epoch": 8.3160400390625e-07,
      "model_forward_time": 5.593835115432739,
      "step": 872
    },
    {
      "epoch": 8.3160400390625e-07,
      "step": 872,
      "training_step_time": 17.04998016357422
    },
    {
      "epoch": 8.325576782226562e-07,
      "model_forward_time": 5.605483770370483,
      "step": 873
    },
    {
      "epoch": 8.325576782226562e-07,
      "step": 873,
      "training_step_time": 17.09308886528015
    },
    {
      "epoch": 8.335113525390625e-07,
      "model_forward_time": 5.6207544803619385,
      "step": 874
    },
    {
      "epoch": 8.335113525390625e-07,
      "step": 874,
      "training_step_time": 17.11917519569397
    },
    {
      "epoch": 8.344650268554688e-07,
      "model_forward_time": 5.9637134075164795,
      "step": 875
    },
    {
      "epoch": 8.344650268554688e-07,
      "step": 875,
      "training_step_time": 17.44093346595764
    },
    {
      "epoch": 8.35418701171875e-07,
      "model_forward_time": 5.600019931793213,
      "step": 876
    },
    {
      "epoch": 8.35418701171875e-07,
      "step": 876,
      "training_step_time": 17.063507556915283
    },
    {
      "epoch": 8.363723754882813e-07,
      "model_forward_time": 5.598779916763306,
      "step": 877
    },
    {
      "epoch": 8.363723754882813e-07,
      "step": 877,
      "training_step_time": 17.023491144180298
    },
    {
      "epoch": 8.373260498046875e-07,
      "model_forward_time": 5.584840536117554,
      "step": 878
    },
    {
      "epoch": 8.373260498046875e-07,
      "step": 878,
      "training_step_time": 17.023135900497437
    },
    {
      "epoch": 8.382797241210938e-07,
      "model_forward_time": 5.596566438674927,
      "step": 879
    },
    {
      "epoch": 8.382797241210938e-07,
      "step": 879,
      "training_step_time": 17.043609619140625
    },
    {
      "epoch": 8.392333984375e-07,
      "grad_norm": 1.4451411962509155,
      "learning_rate": 1e-05,
      "loss": 0.0913,
      "step": 880
    },
    {
      "epoch": 8.392333984375e-07,
      "model_forward_time": 5.592734336853027,
      "step": 880
    },
    {
      "dynamics_loss_avg": 0.06208086460828781,
      "epoch": 8.392333984375e-07,
      "step": 880
    },
    {
      "action_loss_avg": 0.023352243937551975,
      "epoch": 8.392333984375e-07,
      "step": 880
    },
    {
      "epoch": 8.392333984375e-07,
      "step": 880,
      "training_step_time": 17.057576656341553
    },
    {
      "epoch": 8.401870727539063e-07,
      "model_forward_time": 5.945918083190918,
      "step": 881
    },
    {
      "epoch": 8.401870727539063e-07,
      "step": 881,
      "training_step_time": 17.439561367034912
    },
    {
      "epoch": 8.411407470703125e-07,
      "model_forward_time": 5.6140456199646,
      "step": 882
    },
    {
      "epoch": 8.411407470703125e-07,
      "step": 882,
      "training_step_time": 17.064378261566162
    },
    {
      "epoch": 8.420944213867188e-07,
      "model_forward_time": 5.595370769500732,
      "step": 883
    },
    {
      "epoch": 8.420944213867188e-07,
      "step": 883,
      "training_step_time": 17.03501272201538
    },
    {
      "epoch": 8.43048095703125e-07,
      "model_forward_time": 5.587140083312988,
      "step": 884
    },
    {
      "epoch": 8.43048095703125e-07,
      "step": 884,
      "training_step_time": 16.99379801750183
    },
    {
      "epoch": 8.440017700195313e-07,
      "model_forward_time": 5.582539796829224,
      "step": 885
    },
    {
      "epoch": 8.440017700195313e-07,
      "step": 885,
      "training_step_time": 16.99200749397278
    },
    {
      "epoch": 8.449554443359375e-07,
      "model_forward_time": 5.94741153717041,
      "step": 886
    },
    {
      "epoch": 8.449554443359375e-07,
      "step": 886,
      "training_step_time": 17.356980323791504
    },
    {
      "epoch": 8.459091186523438e-07,
      "model_forward_time": 5.609988212585449,
      "step": 887
    },
    {
      "epoch": 8.459091186523438e-07,
      "step": 887,
      "training_step_time": 17.042978525161743
    },
    {
      "epoch": 8.4686279296875e-07,
      "model_forward_time": 5.594897270202637,
      "step": 888
    },
    {
      "epoch": 8.4686279296875e-07,
      "step": 888,
      "training_step_time": 17.045400381088257
    },
    {
      "epoch": 8.478164672851563e-07,
      "model_forward_time": 5.592925548553467,
      "step": 889
    },
    {
      "epoch": 8.478164672851563e-07,
      "step": 889,
      "training_step_time": 17.07549238204956
    },
    {
      "epoch": 8.487701416015625e-07,
      "grad_norm": 3.024719476699829,
      "learning_rate": 1e-05,
      "loss": 0.1094,
      "step": 890
    },
    {
      "epoch": 8.487701416015625e-07,
      "model_forward_time": 5.617943048477173,
      "step": 890
    },
    {
      "dynamics_loss_avg": 0.07532855942845344,
      "epoch": 8.487701416015625e-07,
      "step": 890
    },
    {
      "action_loss_avg": 0.031397513672709466,
      "epoch": 8.487701416015625e-07,
      "step": 890
    },
    {
      "epoch": 8.487701416015625e-07,
      "step": 890,
      "training_step_time": 17.11794662475586
    },
    {
      "epoch": 8.497238159179688e-07,
      "model_forward_time": 5.954319000244141,
      "step": 891
    },
    {
      "epoch": 8.497238159179688e-07,
      "step": 891,
      "training_step_time": 17.40486764907837
    },
    {
      "epoch": 8.50677490234375e-07,
      "model_forward_time": 5.602641582489014,
      "step": 892
    },
    {
      "epoch": 8.50677490234375e-07,
      "step": 892,
      "training_step_time": 17.070595741271973
    },
    {
      "epoch": 8.516311645507813e-07,
      "model_forward_time": 5.604943513870239,
      "step": 893
    },
    {
      "epoch": 8.516311645507813e-07,
      "step": 893,
      "training_step_time": 17.085256338119507
    },
    {
      "epoch": 8.525848388671875e-07,
      "model_forward_time": 5.5952839851379395,
      "step": 894
    },
    {
      "epoch": 8.525848388671875e-07,
      "step": 894,
      "training_step_time": 17.05834674835205
    },
    {
      "epoch": 8.535385131835938e-07,
      "model_forward_time": 5.602445602416992,
      "step": 895
    },
    {
      "epoch": 8.535385131835938e-07,
      "step": 895,
      "training_step_time": 17.051482915878296
    },
    {
      "epoch": 8.544921875e-07,
      "model_forward_time": 5.605368614196777,
      "step": 896
    },
    {
      "epoch": 8.544921875e-07,
      "step": 896,
      "training_step_time": 17.07175922393799
    },
    {
      "epoch": 8.554458618164063e-07,
      "model_forward_time": 5.9549479484558105,
      "step": 897
    },
    {
      "epoch": 8.554458618164063e-07,
      "step": 897,
      "training_step_time": 17.400012731552124
    },
    {
      "epoch": 8.563995361328125e-07,
      "model_forward_time": 5.597649574279785,
      "step": 898
    },
    {
      "epoch": 8.563995361328125e-07,
      "step": 898,
      "training_step_time": 17.08762526512146
    },
    {
      "epoch": 8.573532104492188e-07,
      "model_forward_time": 5.6252281665802,
      "step": 899
    },
    {
      "epoch": 8.573532104492188e-07,
      "step": 899,
      "training_step_time": 17.142502307891846
    },
    {
      "epoch": 8.58306884765625e-07,
      "grad_norm": 2.071617364883423,
      "learning_rate": 1e-05,
      "loss": 0.0983,
      "step": 900
    },
    {
      "epoch": 8.58306884765625e-07,
      "model_forward_time": 5.622058868408203,
      "step": 900
    },
    {
      "dynamics_loss_avg": 0.07175751477479934,
      "epoch": 8.58306884765625e-07,
      "step": 900
    },
    {
      "action_loss_avg": 0.028138335142284632,
      "epoch": 8.58306884765625e-07,
      "step": 900
    },
    {
      "epoch": 8.58306884765625e-07,
      "step": 900,
      "training_step_time": 17.10083818435669
    },
    {
      "epoch": 8.592605590820313e-07,
      "model_forward_time": 5.609132766723633,
      "step": 901
    },
    {
      "epoch": 8.592605590820313e-07,
      "step": 901,
      "training_step_time": 17.088937997817993
    },
    {
      "epoch": 8.602142333984375e-07,
      "model_forward_time": 5.596412420272827,
      "step": 902
    },
    {
      "epoch": 8.602142333984375e-07,
      "step": 902,
      "training_step_time": 17.409851551055908
    },
    {
      "epoch": 8.611679077148438e-07,
      "model_forward_time": 5.595025539398193,
      "step": 903
    },
    {
      "epoch": 8.611679077148438e-07,
      "step": 903,
      "training_step_time": 17.051578521728516
    },
    {
      "epoch": 8.6212158203125e-07,
      "model_forward_time": 5.597488880157471,
      "step": 904
    },
    {
      "epoch": 8.6212158203125e-07,
      "step": 904,
      "training_step_time": 17.07371759414673
    },
    {
      "epoch": 8.630752563476563e-07,
      "model_forward_time": 5.601090908050537,
      "step": 905
    },
    {
      "epoch": 8.630752563476563e-07,
      "step": 905,
      "training_step_time": 17.04048466682434
    },
    {
      "epoch": 8.640289306640625e-07,
      "model_forward_time": 5.661089658737183,
      "step": 906
    },
    {
      "epoch": 8.640289306640625e-07,
      "step": 906,
      "training_step_time": 17.163736581802368
    },
    {
      "epoch": 8.649826049804688e-07,
      "model_forward_time": 5.639760971069336,
      "step": 907
    },
    {
      "epoch": 8.649826049804688e-07,
      "step": 907,
      "training_step_time": 17.125585794448853
    },
    {
      "epoch": 8.65936279296875e-07,
      "model_forward_time": 6.080518960952759,
      "step": 908
    },
    {
      "epoch": 8.65936279296875e-07,
      "step": 908,
      "training_step_time": 17.54813241958618
    },
    {
      "epoch": 8.668899536132813e-07,
      "model_forward_time": 5.614075660705566,
      "step": 909
    },
    {
      "epoch": 8.668899536132813e-07,
      "step": 909,
      "training_step_time": 17.118966817855835
    },
    {
      "epoch": 8.678436279296875e-07,
      "grad_norm": 1.5532315969467163,
      "learning_rate": 1e-05,
      "loss": 0.0982,
      "step": 910
    },
    {
      "epoch": 8.678436279296875e-07,
      "model_forward_time": 5.640245199203491,
      "step": 910
    },
    {
      "dynamics_loss_avg": 0.06730675622820854,
      "epoch": 8.678436279296875e-07,
      "step": 910
    },
    {
      "action_loss_avg": 0.028969980869442226,
      "epoch": 8.678436279296875e-07,
      "step": 910
    },
    {
      "epoch": 8.678436279296875e-07,
      "step": 910,
      "training_step_time": 17.151474237442017
    },
    {
      "epoch": 8.687973022460938e-07,
      "model_forward_time": 5.618433475494385,
      "step": 911
    },
    {
      "epoch": 8.687973022460938e-07,
      "step": 911,
      "training_step_time": 17.07342839241028
    },
    {
      "epoch": 8.697509765625e-07,
      "model_forward_time": 5.5866312980651855,
      "step": 912
    },
    {
      "epoch": 8.697509765625e-07,
      "step": 912,
      "training_step_time": 16.99953007698059
    },
    {
      "epoch": 8.707046508789063e-07,
      "model_forward_time": 5.580838441848755,
      "step": 913
    },
    {
      "epoch": 8.707046508789063e-07,
      "step": 913,
      "training_step_time": 17.38211989402771
    },
    {
      "epoch": 8.716583251953125e-07,
      "model_forward_time": 5.5663511753082275,
      "step": 914
    },
    {
      "epoch": 8.716583251953125e-07,
      "step": 914,
      "training_step_time": 16.9893639087677
    },
    {
      "epoch": 8.726119995117188e-07,
      "model_forward_time": 5.561511278152466,
      "step": 915
    },
    {
      "epoch": 8.726119995117188e-07,
      "step": 915,
      "training_step_time": 16.979177236557007
    },
    {
      "epoch": 8.73565673828125e-07,
      "model_forward_time": 5.569768190383911,
      "step": 916
    },
    {
      "epoch": 8.73565673828125e-07,
      "step": 916,
      "training_step_time": 16.987020254135132
    },
    {
      "epoch": 8.745193481445313e-07,
      "model_forward_time": 5.597668647766113,
      "step": 917
    },
    {
      "epoch": 8.745193481445313e-07,
      "step": 917,
      "training_step_time": 17.019206047058105
    },
    {
      "epoch": 8.754730224609375e-07,
      "model_forward_time": 5.572957754135132,
      "step": 918
    },
    {
      "epoch": 8.754730224609375e-07,
      "step": 918,
      "training_step_time": 17.02171039581299
    },
    {
      "epoch": 8.764266967773438e-07,
      "model_forward_time": 5.95085334777832,
      "step": 919
    },
    {
      "epoch": 8.764266967773438e-07,
      "step": 919,
      "training_step_time": 17.38226079940796
    },
    {
      "epoch": 8.7738037109375e-07,
      "grad_norm": 1.5489147901535034,
      "learning_rate": 1e-05,
      "loss": 0.1217,
      "step": 920
    },
    {
      "epoch": 8.7738037109375e-07,
      "model_forward_time": 5.5869481563568115,
      "step": 920
    },
    {
      "dynamics_loss_avg": 0.08575975336134434,
      "epoch": 8.7738037109375e-07,
      "step": 920
    },
    {
      "action_loss_avg": 0.03648517979308963,
      "epoch": 8.7738037109375e-07,
      "step": 920
    },
    {
      "epoch": 8.7738037109375e-07,
      "step": 920,
      "training_step_time": 17.04244875907898
    },
    {
      "epoch": 8.783340454101563e-07,
      "model_forward_time": 5.614730358123779,
      "step": 921
    },
    {
      "epoch": 8.783340454101563e-07,
      "step": 921,
      "training_step_time": 17.112783193588257
    },
    {
      "epoch": 8.792877197265625e-07,
      "model_forward_time": 5.625761270523071,
      "step": 922
    },
    {
      "epoch": 8.792877197265625e-07,
      "step": 922,
      "training_step_time": 17.136403799057007
    },
    {
      "epoch": 8.802413940429688e-07,
      "model_forward_time": 5.627568244934082,
      "step": 923
    },
    {
      "epoch": 8.802413940429688e-07,
      "step": 923,
      "training_step_time": 17.140968322753906
    },
    {
      "epoch": 8.81195068359375e-07,
      "model_forward_time": 5.967700719833374,
      "step": 924
    },
    {
      "epoch": 8.81195068359375e-07,
      "step": 924,
      "training_step_time": 17.45127558708191
    },
    {
      "epoch": 8.821487426757813e-07,
      "model_forward_time": 5.590178489685059,
      "step": 925
    },
    {
      "epoch": 8.821487426757813e-07,
      "step": 925,
      "training_step_time": 17.022135972976685
    },
    {
      "epoch": 8.831024169921875e-07,
      "model_forward_time": 5.568838357925415,
      "step": 926
    },
    {
      "epoch": 8.831024169921875e-07,
      "step": 926,
      "training_step_time": 16.99972105026245
    },
    {
      "epoch": 8.840560913085938e-07,
      "model_forward_time": 5.566625118255615,
      "step": 927
    },
    {
      "epoch": 8.840560913085938e-07,
      "step": 927,
      "training_step_time": 16.98204517364502
    },
    {
      "epoch": 8.85009765625e-07,
      "model_forward_time": 5.570538282394409,
      "step": 928
    },
    {
      "epoch": 8.85009765625e-07,
      "step": 928,
      "training_step_time": 16.985069274902344
    },
    {
      "epoch": 8.859634399414063e-07,
      "model_forward_time": 5.568406820297241,
      "step": 929
    },
    {
      "epoch": 8.859634399414063e-07,
      "step": 929,
      "training_step_time": 17.000375986099243
    },
    {
      "epoch": 8.869171142578125e-07,
      "grad_norm": 0.8811934590339661,
      "learning_rate": 1e-05,
      "loss": 0.0941,
      "step": 930
    },
    {
      "epoch": 8.869171142578125e-07,
      "model_forward_time": 5.936558961868286,
      "step": 930
    },
    {
      "dynamics_loss_avg": 0.06790360771119594,
      "epoch": 8.869171142578125e-07,
      "step": 930
    },
    {
      "action_loss_avg": 0.029241542145609856,
      "epoch": 8.869171142578125e-07,
      "step": 930
    },
    {
      "epoch": 8.869171142578125e-07,
      "step": 930,
      "training_step_time": 17.348936796188354
    },
    {
      "epoch": 8.878707885742188e-07,
      "model_forward_time": 5.584569931030273,
      "step": 931
    },
    {
      "epoch": 8.878707885742188e-07,
      "step": 931,
      "training_step_time": 17.007502555847168
    },
    {
      "epoch": 8.88824462890625e-07,
      "model_forward_time": 5.603008031845093,
      "step": 932
    },
    {
      "epoch": 8.88824462890625e-07,
      "step": 932,
      "training_step_time": 17.074201345443726
    },
    {
      "epoch": 8.897781372070313e-07,
      "model_forward_time": 5.604749917984009,
      "step": 933
    },
    {
      "epoch": 8.897781372070313e-07,
      "step": 933,
      "training_step_time": 17.089439868927002
    },
    {
      "epoch": 8.907318115234375e-07,
      "model_forward_time": 5.600212097167969,
      "step": 934
    },
    {
      "epoch": 8.907318115234375e-07,
      "step": 934,
      "training_step_time": 17.064959049224854
    },
    {
      "epoch": 8.916854858398438e-07,
      "model_forward_time": 5.593743562698364,
      "step": 935
    },
    {
      "epoch": 8.916854858398438e-07,
      "step": 935,
      "training_step_time": 17.398903369903564
    },
    {
      "epoch": 8.9263916015625e-07,
      "model_forward_time": 5.592809438705444,
      "step": 936
    },
    {
      "epoch": 8.9263916015625e-07,
      "step": 936,
      "training_step_time": 17.05492925643921
    },
    {
      "epoch": 8.935928344726563e-07,
      "model_forward_time": 5.600707530975342,
      "step": 937
    },
    {
      "epoch": 8.935928344726563e-07,
      "step": 937,
      "training_step_time": 17.069438219070435
    },
    {
      "epoch": 8.945465087890624e-07,
      "model_forward_time": 5.595383644104004,
      "step": 938
    },
    {
      "epoch": 8.945465087890624e-07,
      "step": 938,
      "training_step_time": 17.023402214050293
    },
    {
      "epoch": 8.955001831054687e-07,
      "model_forward_time": 5.595118284225464,
      "step": 939
    },
    {
      "epoch": 8.955001831054687e-07,
      "step": 939,
      "training_step_time": 17.050164461135864
    },
    {
      "epoch": 8.96453857421875e-07,
      "grad_norm": 1.8045711517333984,
      "learning_rate": 1e-05,
      "loss": 0.092,
      "step": 940
    },
    {
      "epoch": 8.96453857421875e-07,
      "model_forward_time": 5.5971457958221436,
      "step": 940
    },
    {
      "dynamics_loss_avg": 0.06776414178311825,
      "epoch": 8.96453857421875e-07,
      "step": 940
    },
    {
      "action_loss_avg": 0.026793814613483846,
      "epoch": 8.96453857421875e-07,
      "step": 940
    },
    {
      "epoch": 8.96453857421875e-07,
      "step": 940,
      "training_step_time": 17.08020329475403
    },
    {
      "epoch": 8.974075317382812e-07,
      "model_forward_time": 5.989783048629761,
      "step": 941
    },
    {
      "epoch": 8.974075317382812e-07,
      "step": 941,
      "training_step_time": 17.49849772453308
    },
    {
      "epoch": 8.983612060546875e-07,
      "model_forward_time": 5.615435361862183,
      "step": 942
    },
    {
      "epoch": 8.983612060546875e-07,
      "step": 942,
      "training_step_time": 17.12338638305664
    },
    {
      "epoch": 8.993148803710937e-07,
      "model_forward_time": 5.614883184432983,
      "step": 943
    },
    {
      "epoch": 8.993148803710937e-07,
      "step": 943,
      "training_step_time": 17.116880178451538
    },
    {
      "epoch": 9.002685546875e-07,
      "model_forward_time": 5.607093811035156,
      "step": 944
    },
    {
      "epoch": 9.002685546875e-07,
      "step": 944,
      "training_step_time": 17.09333109855652
    },
    {
      "epoch": 9.012222290039062e-07,
      "model_forward_time": 5.607401371002197,
      "step": 945
    },
    {
      "epoch": 9.012222290039062e-07,
      "step": 945,
      "training_step_time": 17.09717297554016
    },
    {
      "epoch": 9.021759033203125e-07,
      "model_forward_time": 5.603790044784546,
      "step": 946
    },
    {
      "epoch": 9.021759033203125e-07,
      "step": 946,
      "training_step_time": 17.439680099487305
    },
    {
      "epoch": 9.031295776367187e-07,
      "model_forward_time": 5.606228590011597,
      "step": 947
    },
    {
      "epoch": 9.031295776367187e-07,
      "step": 947,
      "training_step_time": 17.086161851882935
    },
    {
      "epoch": 9.04083251953125e-07,
      "model_forward_time": 5.601015329360962,
      "step": 948
    },
    {
      "epoch": 9.04083251953125e-07,
      "step": 948,
      "training_step_time": 17.04379892349243
    },
    {
      "epoch": 9.050369262695312e-07,
      "model_forward_time": 5.580278396606445,
      "step": 949
    },
    {
      "epoch": 9.050369262695312e-07,
      "step": 949,
      "training_step_time": 17.029247045516968
    },
    {
      "epoch": 9.059906005859375e-07,
      "grad_norm": 1.0754114389419556,
      "learning_rate": 1e-05,
      "loss": 0.0873,
      "step": 950
    },
    {
      "epoch": 9.059906005859375e-07,
      "model_forward_time": 5.595031261444092,
      "step": 950
    },
    {
      "dynamics_loss_avg": 0.05755225606262684,
      "epoch": 9.059906005859375e-07,
      "step": 950
    },
    {
      "action_loss_avg": 0.023539384268224238,
      "epoch": 9.059906005859375e-07,
      "step": 950
    },
    {
      "epoch": 9.059906005859375e-07,
      "step": 950,
      "training_step_time": 17.050891876220703
    },
    {
      "epoch": 9.069442749023437e-07,
      "model_forward_time": 5.5971901416778564,
      "step": 951
    },
    {
      "epoch": 9.069442749023437e-07,
      "step": 951,
      "training_step_time": 17.05645251274109
    },
    {
      "epoch": 9.0789794921875e-07,
      "model_forward_time": 5.96404242515564,
      "step": 952
    },
    {
      "epoch": 9.0789794921875e-07,
      "step": 952,
      "training_step_time": 17.404531002044678
    },
    {
      "epoch": 9.088516235351562e-07,
      "model_forward_time": 5.576531648635864,
      "step": 953
    },
    {
      "epoch": 9.088516235351562e-07,
      "step": 953,
      "training_step_time": 16.994613647460938
    },
    {
      "epoch": 9.098052978515625e-07,
      "model_forward_time": 5.592449903488159,
      "step": 954
    },
    {
      "epoch": 9.098052978515625e-07,
      "step": 954,
      "training_step_time": 17.06342601776123
    },
    {
      "epoch": 9.107589721679687e-07,
      "model_forward_time": 5.597185373306274,
      "step": 955
    },
    {
      "epoch": 9.107589721679687e-07,
      "step": 955,
      "training_step_time": 17.075157642364502
    },
    {
      "epoch": 9.11712646484375e-07,
      "model_forward_time": 5.61449408531189,
      "step": 956
    },
    {
      "epoch": 9.11712646484375e-07,
      "step": 956,
      "training_step_time": 17.097152948379517
    },
    {
      "epoch": 9.126663208007812e-07,
      "model_forward_time": 5.967119216918945,
      "step": 957
    },
    {
      "epoch": 9.126663208007812e-07,
      "step": 957,
      "training_step_time": 17.465712070465088
    },
    {
      "epoch": 9.136199951171875e-07,
      "model_forward_time": 5.616621255874634,
      "step": 958
    },
    {
      "epoch": 9.136199951171875e-07,
      "step": 958,
      "training_step_time": 17.110249280929565
    },
    {
      "epoch": 9.145736694335937e-07,
      "model_forward_time": 5.60628867149353,
      "step": 959
    },
    {
      "epoch": 9.145736694335937e-07,
      "step": 959,
      "training_step_time": 17.092110872268677
    },
    {
      "epoch": 9.1552734375e-07,
      "grad_norm": 1.9739787578582764,
      "learning_rate": 1e-05,
      "loss": 0.0836,
      "step": 960
    },
    {
      "epoch": 9.1552734375e-07,
      "model_forward_time": 5.605801820755005,
      "step": 960
    },
    {
      "dynamics_loss_avg": 0.062164004519581796,
      "epoch": 9.1552734375e-07,
      "step": 960
    },
    {
      "action_loss_avg": 0.024905206728726627,
      "epoch": 9.1552734375e-07,
      "step": 960
    },
    {
      "epoch": 9.1552734375e-07,
      "step": 960,
      "training_step_time": 17.082230806350708
    },
    {
      "epoch": 9.164810180664062e-07,
      "model_forward_time": 5.6079747676849365,
      "step": 961
    },
    {
      "epoch": 9.164810180664062e-07,
      "step": 961,
      "training_step_time": 17.089025497436523
    },
    {
      "epoch": 9.174346923828125e-07,
      "model_forward_time": 5.611504793167114,
      "step": 962
    },
    {
      "epoch": 9.174346923828125e-07,
      "step": 962,
      "training_step_time": 17.090086698532104
    },
    {
      "epoch": 9.183883666992187e-07,
      "model_forward_time": 5.992034196853638,
      "step": 963
    },
    {
      "epoch": 9.183883666992187e-07,
      "step": 963,
      "training_step_time": 17.467978954315186
    },
    {
      "epoch": 9.19342041015625e-07,
      "model_forward_time": 5.619885206222534,
      "step": 964
    },
    {
      "epoch": 9.19342041015625e-07,
      "step": 964,
      "training_step_time": 17.109739303588867
    },
    {
      "epoch": 9.202957153320312e-07,
      "model_forward_time": 5.6060450077056885,
      "step": 965
    },
    {
      "epoch": 9.202957153320312e-07,
      "step": 965,
      "training_step_time": 17.103026628494263
    },
    {
      "epoch": 9.212493896484375e-07,
      "model_forward_time": 5.621550559997559,
      "step": 966
    },
    {
      "epoch": 9.212493896484375e-07,
      "step": 966,
      "training_step_time": 17.11837935447693
    },
    {
      "epoch": 9.222030639648437e-07,
      "model_forward_time": 5.614540338516235,
      "step": 967
    },
    {
      "epoch": 9.222030639648437e-07,
      "step": 967,
      "training_step_time": 17.09190320968628
    },
    {
      "epoch": 9.2315673828125e-07,
      "model_forward_time": 5.616649389266968,
      "step": 968
    },
    {
      "epoch": 9.2315673828125e-07,
      "step": 968,
      "training_step_time": 17.4364116191864
    },
    {
      "epoch": 9.241104125976562e-07,
      "model_forward_time": 5.598785877227783,
      "step": 969
    },
    {
      "epoch": 9.241104125976562e-07,
      "step": 969,
      "training_step_time": 17.01827836036682
    },
    {
      "epoch": 9.250640869140625e-07,
      "grad_norm": 2.0753042697906494,
      "learning_rate": 1e-05,
      "loss": 0.0965,
      "step": 970
    },
    {
      "epoch": 9.250640869140625e-07,
      "model_forward_time": 5.591946125030518,
      "step": 970
    },
    {
      "dynamics_loss_avg": 0.06990581266582012,
      "epoch": 9.250640869140625e-07,
      "step": 970
    },
    {
      "action_loss_avg": 0.021975538600236176,
      "epoch": 9.250640869140625e-07,
      "step": 970
    },
    {
      "epoch": 9.250640869140625e-07,
      "step": 970,
      "training_step_time": 17.026963710784912
    },
    {
      "epoch": 9.260177612304687e-07,
      "model_forward_time": 5.59780478477478,
      "step": 971
    },
    {
      "epoch": 9.260177612304687e-07,
      "step": 971,
      "training_step_time": 17.092036247253418
    },
    {
      "epoch": 9.26971435546875e-07,
      "model_forward_time": 5.622776985168457,
      "step": 972
    },
    {
      "epoch": 9.26971435546875e-07,
      "step": 972,
      "training_step_time": 17.142817735671997
    },
    {
      "epoch": 9.279251098632812e-07,
      "model_forward_time": 5.619745492935181,
      "step": 973
    },
    {
      "epoch": 9.279251098632812e-07,
      "step": 973,
      "training_step_time": 17.12903118133545
    },
    {
      "epoch": 9.288787841796875e-07,
      "model_forward_time": 5.967742681503296,
      "step": 974
    },
    {
      "epoch": 9.288787841796875e-07,
      "step": 974,
      "training_step_time": 17.425237894058228
    },
    {
      "epoch": 9.298324584960937e-07,
      "model_forward_time": 5.583709716796875,
      "step": 975
    },
    {
      "epoch": 9.298324584960937e-07,
      "step": 975,
      "training_step_time": 17.009817361831665
    },
    {
      "epoch": 9.307861328125e-07,
      "model_forward_time": 5.5779290199279785,
      "step": 976
    },
    {
      "epoch": 9.307861328125e-07,
      "step": 976,
      "training_step_time": 16.992409706115723
    },
    {
      "epoch": 9.317398071289062e-07,
      "model_forward_time": 5.563563585281372,
      "step": 977
    },
    {
      "epoch": 9.317398071289062e-07,
      "step": 977,
      "training_step_time": 16.962319135665894
    },
    {
      "epoch": 9.326934814453125e-07,
      "model_forward_time": 5.580997467041016,
      "step": 978
    },
    {
      "epoch": 9.326934814453125e-07,
      "step": 978,
      "training_step_time": 17.0202693939209
    },
    {
      "epoch": 9.336471557617187e-07,
      "model_forward_time": 5.915348291397095,
      "step": 979
    },
    {
      "epoch": 9.336471557617187e-07,
      "step": 979,
      "training_step_time": 17.33508539199829
    },
    {
      "epoch": 9.34600830078125e-07,
      "grad_norm": 1.0974041223526,
      "learning_rate": 1e-05,
      "loss": 0.0931,
      "step": 980
    },
    {
      "epoch": 9.34600830078125e-07,
      "model_forward_time": 5.584020614624023,
      "step": 980
    },
    {
      "dynamics_loss_avg": 0.07517103962600231,
      "epoch": 9.34600830078125e-07,
      "step": 980
    },
    {
      "action_loss_avg": 0.02662748787552118,
      "epoch": 9.34600830078125e-07,
      "step": 980
    },
    {
      "epoch": 9.34600830078125e-07,
      "step": 980,
      "training_step_time": 17.021106004714966
    },
    {
      "epoch": 9.355545043945312e-07,
      "model_forward_time": 5.599820852279663,
      "step": 981
    },
    {
      "epoch": 9.355545043945312e-07,
      "step": 981,
      "training_step_time": 17.07781982421875
    },
    {
      "epoch": 9.365081787109375e-07,
      "model_forward_time": 5.621808767318726,
      "step": 982
    },
    {
      "epoch": 9.365081787109375e-07,
      "step": 982,
      "training_step_time": 17.13126564025879
    },
    {
      "epoch": 9.374618530273437e-07,
      "model_forward_time": 5.618195533752441,
      "step": 983
    },
    {
      "epoch": 9.374618530273437e-07,
      "step": 983,
      "training_step_time": 17.100892782211304
    },
    {
      "epoch": 9.3841552734375e-07,
      "model_forward_time": 5.599469423294067,
      "step": 984
    },
    {
      "epoch": 9.3841552734375e-07,
      "step": 984,
      "training_step_time": 17.034716844558716
    },
    {
      "epoch": 9.393692016601562e-07,
      "model_forward_time": 5.936985731124878,
      "step": 985
    },
    {
      "epoch": 9.393692016601562e-07,
      "step": 985,
      "training_step_time": 17.34439468383789
    },
    {
      "epoch": 9.403228759765625e-07,
      "model_forward_time": 5.5762412548065186,
      "step": 986
    },
    {
      "epoch": 9.403228759765625e-07,
      "step": 986,
      "training_step_time": 17.004395484924316
    },
    {
      "epoch": 9.412765502929687e-07,
      "model_forward_time": 5.581926345825195,
      "step": 987
    },
    {
      "epoch": 9.412765502929687e-07,
      "step": 987,
      "training_step_time": 16.989278078079224
    },
    {
      "epoch": 9.42230224609375e-07,
      "model_forward_time": 5.591959238052368,
      "step": 988
    },
    {
      "epoch": 9.42230224609375e-07,
      "step": 988,
      "training_step_time": 17.03756880760193
    },
    {
      "epoch": 9.431838989257812e-07,
      "model_forward_time": 5.59609317779541,
      "step": 989
    },
    {
      "epoch": 9.431838989257812e-07,
      "step": 989,
      "training_step_time": 17.060858964920044
    },
    {
      "epoch": 9.441375732421875e-07,
      "grad_norm": 1.3704663515090942,
      "learning_rate": 1e-05,
      "loss": 0.114,
      "step": 990
    },
    {
      "epoch": 9.441375732421875e-07,
      "model_forward_time": 5.962547540664673,
      "step": 990
    },
    {
      "dynamics_loss_avg": 0.07076330855488777,
      "epoch": 9.441375732421875e-07,
      "step": 990
    },
    {
      "action_loss_avg": 0.034508184995502234,
      "epoch": 9.441375732421875e-07,
      "step": 990
    },
    {
      "epoch": 9.441375732421875e-07,
      "step": 990,
      "training_step_time": 17.442424535751343
    },
    {
      "epoch": 9.450912475585937e-07,
      "model_forward_time": 5.604442834854126,
      "step": 991
    },
    {
      "epoch": 9.450912475585937e-07,
      "step": 991,
      "training_step_time": 17.07888126373291
    },
    {
      "epoch": 9.46044921875e-07,
      "model_forward_time": 5.613363265991211,
      "step": 992
    },
    {
      "epoch": 9.46044921875e-07,
      "step": 992,
      "training_step_time": 17.11695694923401
    },
    {
      "epoch": 9.469985961914062e-07,
      "model_forward_time": 5.619887590408325,
      "step": 993
    },
    {
      "epoch": 9.469985961914062e-07,
      "step": 993,
      "training_step_time": 17.119774103164673
    },
    {
      "epoch": 9.479522705078125e-07,
      "model_forward_time": 5.61333441734314,
      "step": 994
    },
    {
      "epoch": 9.479522705078125e-07,
      "step": 994,
      "training_step_time": 17.118955850601196
    },
    {
      "epoch": 9.489059448242187e-07,
      "model_forward_time": 5.610289573669434,
      "step": 995
    },
    {
      "epoch": 9.489059448242187e-07,
      "step": 995,
      "training_step_time": 17.46768546104431
    },
    {
      "epoch": 9.49859619140625e-07,
      "model_forward_time": 5.617230415344238,
      "step": 996
    },
    {
      "epoch": 9.49859619140625e-07,
      "step": 996,
      "training_step_time": 17.100963592529297
    },
    {
      "epoch": 9.508132934570312e-07,
      "model_forward_time": 5.6141297817230225,
      "step": 997
    },
    {
      "epoch": 9.508132934570312e-07,
      "step": 997,
      "training_step_time": 17.094218254089355
    },
    {
      "epoch": 9.517669677734375e-07,
      "model_forward_time": 5.612694501876831,
      "step": 998
    },
    {
      "epoch": 9.517669677734375e-07,
      "step": 998,
      "training_step_time": 17.10472059249878
    },
    {
      "epoch": 9.527206420898437e-07,
      "model_forward_time": 5.618385314941406,
      "step": 999
    },
    {
      "epoch": 9.527206420898437e-07,
      "step": 999,
      "training_step_time": 17.110783100128174
    },
    {
      "epoch": 9.5367431640625e-07,
      "grad_norm": 1.2318884134292603,
      "learning_rate": 1e-05,
      "loss": 0.1072,
      "step": 1000
    },
    {
      "epoch": 9.5367431640625e-07,
      "model_forward_time": 5.665853500366211,
      "step": 1000
    },
    {
      "dynamics_loss_avg": 0.07623286917805672,
      "epoch": 9.5367431640625e-07,
      "step": 1000
    },
    {
      "action_loss_avg": 0.03276725523173809,
      "epoch": 9.5367431640625e-07,
      "step": 1000
    },
    {
      "epoch": 9.5367431640625e-07,
      "step": 1000,
      "training_step_time": 17.165422439575195
    },
    {
      "epoch": 9.546279907226561e-07,
      "model_forward_time": 6.033774375915527,
      "step": 1001
    },
    {
      "epoch": 9.546279907226561e-07,
      "step": 1001,
      "training_step_time": 17.52300763130188
    },
    {
      "epoch": 9.555816650390625e-07,
      "model_forward_time": 5.650704860687256,
      "step": 1002
    },
    {
      "epoch": 9.555816650390625e-07,
      "step": 1002,
      "training_step_time": 17.129776000976562
    },
    {
      "epoch": 9.565353393554686e-07,
      "model_forward_time": 5.683681488037109,
      "step": 1003
    },
    {
      "epoch": 9.565353393554686e-07,
      "step": 1003,
      "training_step_time": 17.105867385864258
    },
    {
      "epoch": 9.57489013671875e-07,
      "model_forward_time": 5.59341287612915,
      "step": 1004
    },
    {
      "epoch": 9.57489013671875e-07,
      "step": 1004,
      "training_step_time": 17.016294717788696
    },
    {
      "epoch": 9.584426879882811e-07,
      "model_forward_time": 5.586260080337524,
      "step": 1005
    },
    {
      "epoch": 9.584426879882811e-07,
      "step": 1005,
      "training_step_time": 17.010443210601807
    },
    {
      "epoch": 9.593963623046875e-07,
      "model_forward_time": 5.954156875610352,
      "step": 1006
    },
    {
      "epoch": 9.593963623046875e-07,
      "step": 1006,
      "training_step_time": 17.431724786758423
    },
    {
      "epoch": 9.603500366210937e-07,
      "model_forward_time": 5.60625147819519,
      "step": 1007
    },
    {
      "epoch": 9.603500366210937e-07,
      "step": 1007,
      "training_step_time": 17.101216793060303
    },
    {
      "epoch": 9.613037109375e-07,
      "model_forward_time": 5.6018712520599365,
      "step": 1008
    },
    {
      "epoch": 9.613037109375e-07,
      "step": 1008,
      "training_step_time": 17.07625460624695
    },
    {
      "epoch": 9.622573852539062e-07,
      "model_forward_time": 5.60405969619751,
      "step": 1009
    },
    {
      "epoch": 9.622573852539062e-07,
      "step": 1009,
      "training_step_time": 17.07671594619751
    },
    {
      "epoch": 9.632110595703125e-07,
      "grad_norm": 1.658150553703308,
      "learning_rate": 1e-05,
      "loss": 0.1067,
      "step": 1010
    },
    {
      "epoch": 9.632110595703125e-07,
      "model_forward_time": 5.608660936355591,
      "step": 1010
    },
    {
      "dynamics_loss_avg": 0.07913651149719954,
      "epoch": 9.632110595703125e-07,
      "step": 1010
    },
    {
      "action_loss_avg": 0.029704955872148275,
      "epoch": 9.632110595703125e-07,
      "step": 1010
    },
    {
      "epoch": 9.632110595703125e-07,
      "step": 1010,
      "training_step_time": 17.06928539276123
    },
    {
      "epoch": 9.641647338867187e-07,
      "model_forward_time": 5.600531339645386,
      "step": 1011
    },
    {
      "epoch": 9.641647338867187e-07,
      "step": 1011,
      "training_step_time": 17.0337131023407
    },
    {
      "epoch": 9.65118408203125e-07,
      "model_forward_time": 5.959836959838867,
      "step": 1012
    },
    {
      "epoch": 9.65118408203125e-07,
      "step": 1012,
      "training_step_time": 17.39739155769348
    },
    {
      "epoch": 9.660720825195312e-07,
      "model_forward_time": 5.59660005569458,
      "step": 1013
    },
    {
      "epoch": 9.660720825195312e-07,
      "step": 1013,
      "training_step_time": 17.05921721458435
    },
    {
      "epoch": 9.670257568359375e-07,
      "model_forward_time": 5.602498531341553,
      "step": 1014
    },
    {
      "epoch": 9.670257568359375e-07,
      "step": 1014,
      "training_step_time": 17.09356665611267
    },
    {
      "epoch": 9.679794311523437e-07,
      "model_forward_time": 5.61101508140564,
      "step": 1015
    },
    {
      "epoch": 9.679794311523437e-07,
      "step": 1015,
      "training_step_time": 17.110037565231323
    },
    {
      "epoch": 9.6893310546875e-07,
      "model_forward_time": 5.611303091049194,
      "step": 1016
    },
    {
      "epoch": 9.6893310546875e-07,
      "step": 1016,
      "training_step_time": 17.07493567466736
    },
    {
      "epoch": 9.698867797851562e-07,
      "model_forward_time": 5.933389902114868,
      "step": 1017
    },
    {
      "epoch": 9.698867797851562e-07,
      "step": 1017,
      "training_step_time": 17.3543438911438
    },
    {
      "epoch": 9.708404541015625e-07,
      "model_forward_time": 5.586977005004883,
      "step": 1018
    },
    {
      "epoch": 9.708404541015625e-07,
      "step": 1018,
      "training_step_time": 17.009081840515137
    },
    {
      "epoch": 9.717941284179687e-07,
      "model_forward_time": 5.581618070602417,
      "step": 1019
    },
    {
      "epoch": 9.717941284179687e-07,
      "step": 1019,
      "training_step_time": 17.002751350402832
    },
    {
      "epoch": 9.72747802734375e-07,
      "grad_norm": 0.5692186951637268,
      "learning_rate": 1e-05,
      "loss": 0.111,
      "step": 1020
    },
    {
      "epoch": 9.72747802734375e-07,
      "model_forward_time": 5.579861640930176,
      "step": 1020
    },
    {
      "dynamics_loss_avg": 0.07292125299572945,
      "epoch": 9.72747802734375e-07,
      "step": 1020
    },
    {
      "action_loss_avg": 0.037005748692899944,
      "epoch": 9.72747802734375e-07,
      "step": 1020
    },
    {
      "epoch": 9.72747802734375e-07,
      "step": 1020,
      "training_step_time": 17.005897760391235
    },
    {
      "epoch": 9.737014770507812e-07,
      "model_forward_time": 5.5792717933654785,
      "step": 1021
    },
    {
      "epoch": 9.737014770507812e-07,
      "step": 1021,
      "training_step_time": 17.015308141708374
    },
    {
      "epoch": 9.746551513671875e-07,
      "model_forward_time": 5.5949249267578125,
      "step": 1022
    },
    {
      "epoch": 9.746551513671875e-07,
      "step": 1022,
      "training_step_time": 17.042320013046265
    },
    {
      "epoch": 9.756088256835937e-07,
      "model_forward_time": 5.949614763259888,
      "step": 1023
    },
    {
      "epoch": 9.756088256835937e-07,
      "step": 1023,
      "training_step_time": 17.405478477478027
    },
    {
      "epoch": 9.765625e-07,
      "model_forward_time": 5.600083589553833,
      "step": 1024
    },
    {
      "epoch": 9.765625e-07,
      "step": 1024,
      "training_step_time": 17.073917388916016
    },
    {
      "epoch": 9.775161743164062e-07,
      "model_forward_time": 5.594661712646484,
      "step": 1025
    },
    {
      "epoch": 9.775161743164062e-07,
      "step": 1025,
      "training_step_time": 17.052903175354004
    },
    {
      "epoch": 9.784698486328125e-07,
      "model_forward_time": 5.590449571609497,
      "step": 1026
    },
    {
      "epoch": 9.784698486328125e-07,
      "step": 1026,
      "training_step_time": 17.048446655273438
    },
    {
      "epoch": 9.794235229492187e-07,
      "model_forward_time": 5.604611396789551,
      "step": 1027
    },
    {
      "epoch": 9.794235229492187e-07,
      "step": 1027,
      "training_step_time": 17.043662309646606
    },
    {
      "epoch": 9.80377197265625e-07,
      "model_forward_time": 5.940013408660889,
      "step": 1028
    },
    {
      "epoch": 9.80377197265625e-07,
      "step": 1028,
      "training_step_time": 17.40998935699463
    },
    {
      "epoch": 9.813308715820312e-07,
      "model_forward_time": 5.616394758224487,
      "step": 1029
    },
    {
      "epoch": 9.813308715820312e-07,
      "step": 1029,
      "training_step_time": 17.109834909439087
    },
    {
      "epoch": 9.822845458984375e-07,
      "grad_norm": 0.36669960618019104,
      "learning_rate": 1e-05,
      "loss": 0.0887,
      "step": 1030
    },
    {
      "epoch": 9.822845458984375e-07,
      "model_forward_time": 5.604011297225952,
      "step": 1030
    },
    {
      "dynamics_loss_avg": 0.06399496383965016,
      "epoch": 9.822845458984375e-07,
      "step": 1030
    },
    {
      "action_loss_avg": 0.02567543354816735,
      "epoch": 9.822845458984375e-07,
      "step": 1030
    },
    {
      "epoch": 9.822845458984375e-07,
      "step": 1030,
      "training_step_time": 17.096000909805298
    },
    {
      "epoch": 9.832382202148437e-07,
      "model_forward_time": 5.606524467468262,
      "step": 1031
    },
    {
      "epoch": 9.832382202148437e-07,
      "step": 1031,
      "training_step_time": 17.08370018005371
    },
    {
      "epoch": 9.8419189453125e-07,
      "model_forward_time": 5.5809385776519775,
      "step": 1032
    },
    {
      "epoch": 9.8419189453125e-07,
      "step": 1032,
      "training_step_time": 16.996827125549316
    },
    {
      "epoch": 9.851455688476562e-07,
      "model_forward_time": 5.943794012069702,
      "step": 1033
    },
    {
      "epoch": 9.851455688476562e-07,
      "step": 1033,
      "training_step_time": 17.356536626815796
    },
    {
      "epoch": 9.860992431640625e-07,
      "model_forward_time": 5.576663017272949,
      "step": 1034
    },
    {
      "epoch": 9.860992431640625e-07,
      "step": 1034,
      "training_step_time": 16.99695897102356
    },
    {
      "epoch": 9.870529174804687e-07,
      "model_forward_time": 5.584352970123291,
      "step": 1035
    },
    {
      "epoch": 9.870529174804687e-07,
      "step": 1035,
      "training_step_time": 17.005213737487793
    },
    {
      "epoch": 9.88006591796875e-07,
      "model_forward_time": 5.573677062988281,
      "step": 1036
    },
    {
      "epoch": 9.88006591796875e-07,
      "step": 1036,
      "training_step_time": 16.990717887878418
    },
    {
      "epoch": 9.889602661132812e-07,
      "model_forward_time": 5.590507745742798,
      "step": 1037
    },
    {
      "epoch": 9.889602661132812e-07,
      "step": 1037,
      "training_step_time": 17.0387020111084
    },
    {
      "epoch": 9.899139404296875e-07,
      "model_forward_time": 5.596148490905762,
      "step": 1038
    },
    {
      "epoch": 9.899139404296875e-07,
      "step": 1038,
      "training_step_time": 17.041789054870605
    },
    {
      "epoch": 9.908676147460937e-07,
      "model_forward_time": 5.947723627090454,
      "step": 1039
    },
    {
      "epoch": 9.908676147460937e-07,
      "step": 1039,
      "training_step_time": 17.40820026397705
    },
    {
      "epoch": 9.918212890625e-07,
      "grad_norm": 0.7071210741996765,
      "learning_rate": 1e-05,
      "loss": 0.104,
      "step": 1040
    },
    {
      "epoch": 9.918212890625e-07,
      "model_forward_time": 5.594080209732056,
      "step": 1040
    },
    {
      "dynamics_loss_avg": 0.0714408352971077,
      "epoch": 9.918212890625e-07,
      "step": 1040
    },
    {
      "action_loss_avg": 0.02822629394941032,
      "epoch": 9.918212890625e-07,
      "step": 1040
    },
    {
      "epoch": 9.918212890625e-07,
      "step": 1040,
      "training_step_time": 17.091675758361816
    },
    {
      "epoch": 9.927749633789062e-07,
      "model_forward_time": 5.622321605682373,
      "step": 1041
    },
    {
      "epoch": 9.927749633789062e-07,
      "step": 1041,
      "training_step_time": 17.144461154937744
    },
    {
      "epoch": 9.937286376953125e-07,
      "model_forward_time": 5.621269464492798,
      "step": 1042
    },
    {
      "epoch": 9.937286376953125e-07,
      "step": 1042,
      "training_step_time": 17.124693155288696
    },
    {
      "epoch": 9.946823120117187e-07,
      "model_forward_time": 5.607078790664673,
      "step": 1043
    },
    {
      "epoch": 9.946823120117187e-07,
      "step": 1043,
      "training_step_time": 17.07681131362915
    },
    {
      "epoch": 9.95635986328125e-07,
      "model_forward_time": 5.956783294677734,
      "step": 1044
    },
    {
      "epoch": 9.95635986328125e-07,
      "step": 1044,
      "training_step_time": 17.358911514282227
    },
    {
      "epoch": 9.965896606445312e-07,
      "model_forward_time": 5.57390284538269,
      "step": 1045
    },
    {
      "epoch": 9.965896606445312e-07,
      "step": 1045,
      "training_step_time": 16.972270011901855
    },
    {
      "epoch": 9.975433349609375e-07,
      "model_forward_time": 5.578664302825928,
      "step": 1046
    },
    {
      "epoch": 9.975433349609375e-07,
      "step": 1046,
      "training_step_time": 17.02590537071228
    },
    {
      "epoch": 9.984970092773437e-07,
      "model_forward_time": 5.58474326133728,
      "step": 1047
    },
    {
      "epoch": 9.984970092773437e-07,
      "step": 1047,
      "training_step_time": 17.033836126327515
    },
    {
      "epoch": 9.9945068359375e-07,
      "model_forward_time": 5.583190202713013,
      "step": 1048
    },
    {
      "epoch": 9.9945068359375e-07,
      "step": 1048,
      "training_step_time": 17.01044464111328
    },
    {
      "epoch": 1.0004043579101562e-06,
      "model_forward_time": 5.58763861656189,
      "step": 1049
    },
    {
      "epoch": 1.0004043579101562e-06,
      "step": 1049,
      "training_step_time": 16.99516272544861
    },
    {
      "epoch": 1.0013580322265625e-06,
      "grad_norm": 3.284152030944824,
      "learning_rate": 1e-05,
      "loss": 0.0804,
      "step": 1050
    },
    {
      "epoch": 1.0013580322265625e-06,
      "model_forward_time": 5.937809228897095,
      "step": 1050
    },
    {
      "dynamics_loss_avg": 0.05815275385975838,
      "epoch": 1.0013580322265625e-06,
      "step": 1050
    },
    {
      "action_loss_avg": 0.02187018282711506,
      "epoch": 1.0013580322265625e-06,
      "step": 1050
    },
    {
      "epoch": 1.0013580322265625e-06,
      "step": 1050,
      "training_step_time": 17.39580488204956
    },
    {
      "epoch": 1.0023117065429687e-06,
      "model_forward_time": 5.609051704406738,
      "step": 1051
    },
    {
      "epoch": 1.0023117065429687e-06,
      "step": 1051,
      "training_step_time": 17.1097571849823
    },
    {
      "epoch": 1.003265380859375e-06,
      "model_forward_time": 5.622312545776367,
      "step": 1052
    },
    {
      "epoch": 1.003265380859375e-06,
      "step": 1052,
      "training_step_time": 17.130102157592773
    },
    {
      "epoch": 1.0042190551757812e-06,
      "model_forward_time": 5.622856616973877,
      "step": 1053
    },
    {
      "epoch": 1.0042190551757812e-06,
      "step": 1053,
      "training_step_time": 17.114959001541138
    },
    {
      "epoch": 1.0051727294921875e-06,
      "model_forward_time": 5.600127696990967,
      "step": 1054
    },
    {
      "epoch": 1.0051727294921875e-06,
      "step": 1054,
      "training_step_time": 17.052096843719482
    },
    {
      "epoch": 1.0061264038085937e-06,
      "model_forward_time": 5.982797622680664,
      "step": 1055
    },
    {
      "epoch": 1.0061264038085937e-06,
      "step": 1055,
      "training_step_time": 17.40965723991394
    },
    {
      "epoch": 1.007080078125e-06,
      "model_forward_time": 5.590013027191162,
      "step": 1056
    },
    {
      "epoch": 1.007080078125e-06,
      "step": 1056,
      "training_step_time": 17.04772973060608
    },
    {
      "epoch": 1.0080337524414062e-06,
      "model_forward_time": 5.5970542430877686,
      "step": 1057
    },
    {
      "epoch": 1.0080337524414062e-06,
      "step": 1057,
      "training_step_time": 17.050156831741333
    },
    {
      "epoch": 1.0089874267578125e-06,
      "model_forward_time": 5.59464955329895,
      "step": 1058
    },
    {
      "epoch": 1.0089874267578125e-06,
      "step": 1058,
      "training_step_time": 17.0301616191864
    },
    {
      "epoch": 1.0099411010742187e-06,
      "model_forward_time": 5.5961081981658936,
      "step": 1059
    },
    {
      "epoch": 1.0099411010742187e-06,
      "step": 1059,
      "training_step_time": 17.064871788024902
    },
    {
      "epoch": 1.010894775390625e-06,
      "grad_norm": 0.9873877763748169,
      "learning_rate": 1e-05,
      "loss": 0.1068,
      "step": 1060
    },
    {
      "epoch": 1.010894775390625e-06,
      "model_forward_time": 5.59818172454834,
      "step": 1060
    },
    {
      "dynamics_loss_avg": 0.07630092725157737,
      "epoch": 1.010894775390625e-06,
      "step": 1060
    },
    {
      "action_loss_avg": 0.035633868910372256,
      "epoch": 1.010894775390625e-06,
      "step": 1060
    },
    {
      "epoch": 1.010894775390625e-06,
      "step": 1060,
      "training_step_time": 17.41429877281189
    },
    {
      "epoch": 1.0118484497070312e-06,
      "model_forward_time": 5.602216482162476,
      "step": 1061
    },
    {
      "epoch": 1.0118484497070312e-06,
      "step": 1061,
      "training_step_time": 17.036157846450806
    },
    {
      "epoch": 1.0128021240234376e-06,
      "model_forward_time": 5.583468198776245,
      "step": 1062
    },
    {
      "epoch": 1.0128021240234376e-06,
      "step": 1062,
      "training_step_time": 17.01346731185913
    },
    {
      "epoch": 1.0137557983398437e-06,
      "model_forward_time": 5.590306282043457,
      "step": 1063
    },
    {
      "epoch": 1.0137557983398437e-06,
      "step": 1063,
      "training_step_time": 17.040330410003662
    },
    {
      "epoch": 1.01470947265625e-06,
      "model_forward_time": 5.593006134033203,
      "step": 1064
    },
    {
      "epoch": 1.01470947265625e-06,
      "step": 1064,
      "training_step_time": 17.06114935874939
    },
    {
      "epoch": 1.0156631469726562e-06,
      "model_forward_time": 5.6063783168792725,
      "step": 1065
    },
    {
      "epoch": 1.0156631469726562e-06,
      "step": 1065,
      "training_step_time": 17.091968774795532
    },
    {
      "epoch": 1.0166168212890626e-06,
      "model_forward_time": 5.975464820861816,
      "step": 1066
    },
    {
      "epoch": 1.0166168212890626e-06,
      "step": 1066,
      "training_step_time": 17.430705547332764
    },
    {
      "epoch": 1.0175704956054687e-06,
      "model_forward_time": 5.600888252258301,
      "step": 1067
    },
    {
      "epoch": 1.0175704956054687e-06,
      "step": 1067,
      "training_step_time": 17.068967580795288
    },
    {
      "epoch": 1.018524169921875e-06,
      "model_forward_time": 5.604154109954834,
      "step": 1068
    },
    {
      "epoch": 1.018524169921875e-06,
      "step": 1068,
      "training_step_time": 17.046080350875854
    },
    {
      "epoch": 1.0194778442382812e-06,
      "model_forward_time": 5.594825267791748,
      "step": 1069
    },
    {
      "epoch": 1.0194778442382812e-06,
      "step": 1069,
      "training_step_time": 17.012077569961548
    },
    {
      "epoch": 1.0204315185546876e-06,
      "grad_norm": 1.356656551361084,
      "learning_rate": 1e-05,
      "loss": 0.0928,
      "step": 1070
    },
    {
      "epoch": 1.0204315185546876e-06,
      "model_forward_time": 5.584721803665161,
      "step": 1070
    },
    {
      "dynamics_loss_avg": 0.06744749583303929,
      "epoch": 1.0204315185546876e-06,
      "step": 1070
    },
    {
      "action_loss_avg": 0.025672740023583173,
      "epoch": 1.0204315185546876e-06,
      "step": 1070
    },
    {
      "epoch": 1.0204315185546876e-06,
      "step": 1070,
      "training_step_time": 17.03640079498291
    },
    {
      "epoch": 1.0213851928710937e-06,
      "model_forward_time": 5.9492011070251465,
      "step": 1071
    },
    {
      "epoch": 1.0213851928710937e-06,
      "step": 1071,
      "training_step_time": 17.418996334075928
    },
    {
      "epoch": 1.0223388671875e-06,
      "model_forward_time": 5.606598615646362,
      "step": 1072
    },
    {
      "epoch": 1.0223388671875e-06,
      "step": 1072,
      "training_step_time": 17.08711051940918
    },
    {
      "epoch": 1.0232925415039062e-06,
      "model_forward_time": 5.612871408462524,
      "step": 1073
    },
    {
      "epoch": 1.0232925415039062e-06,
      "step": 1073,
      "training_step_time": 17.08744215965271
    },
    {
      "epoch": 1.0242462158203126e-06,
      "model_forward_time": 5.601048707962036,
      "step": 1074
    },
    {
      "epoch": 1.0242462158203126e-06,
      "step": 1074,
      "training_step_time": 17.038898944854736
    },
    {
      "epoch": 1.0251998901367187e-06,
      "model_forward_time": 5.590337753295898,
      "step": 1075
    },
    {
      "epoch": 1.0251998901367187e-06,
      "step": 1075,
      "training_step_time": 17.005876541137695
    },
    {
      "epoch": 1.026153564453125e-06,
      "model_forward_time": 5.585791826248169,
      "step": 1076
    },
    {
      "epoch": 1.026153564453125e-06,
      "step": 1076,
      "training_step_time": 17.002230644226074
    },
    {
      "epoch": 1.0271072387695312e-06,
      "model_forward_time": 5.97050142288208,
      "step": 1077
    },
    {
      "epoch": 1.0271072387695312e-06,
      "step": 1077,
      "training_step_time": 17.394718408584595
    },
    {
      "epoch": 1.0280609130859376e-06,
      "model_forward_time": 5.602715492248535,
      "step": 1078
    },
    {
      "epoch": 1.0280609130859376e-06,
      "step": 1078,
      "training_step_time": 17.086183547973633
    },
    {
      "epoch": 1.0290145874023437e-06,
      "model_forward_time": 5.612802028656006,
      "step": 1079
    },
    {
      "epoch": 1.0290145874023437e-06,
      "step": 1079,
      "training_step_time": 17.117772579193115
    },
    {
      "epoch": 1.02996826171875e-06,
      "grad_norm": 0.9653003811836243,
      "learning_rate": 1e-05,
      "loss": 0.1102,
      "step": 1080
    },
    {
      "epoch": 1.02996826171875e-06,
      "model_forward_time": 5.620919704437256,
      "step": 1080
    },
    {
      "dynamics_loss_avg": 0.07873181477189065,
      "epoch": 1.02996826171875e-06,
      "step": 1080
    },
    {
      "action_loss_avg": 0.03137734979391098,
      "epoch": 1.02996826171875e-06,
      "step": 1080
    },
    {
      "epoch": 1.02996826171875e-06,
      "step": 1080,
      "training_step_time": 17.097700119018555
    },
    {
      "epoch": 1.0309219360351562e-06,
      "model_forward_time": 5.594109535217285,
      "step": 1081
    },
    {
      "epoch": 1.0309219360351562e-06,
      "step": 1081,
      "training_step_time": 17.023043155670166
    },
    {
      "epoch": 1.0318756103515626e-06,
      "model_forward_time": 5.579837799072266,
      "step": 1082
    },
    {
      "epoch": 1.0318756103515626e-06,
      "step": 1082,
      "training_step_time": 17.351367473602295
    },
    {
      "epoch": 1.0328292846679687e-06,
      "model_forward_time": 5.586904048919678,
      "step": 1083
    },
    {
      "epoch": 1.0328292846679687e-06,
      "step": 1083,
      "training_step_time": 17.011489391326904
    },
    {
      "epoch": 1.033782958984375e-06,
      "model_forward_time": 5.5860960483551025,
      "step": 1084
    },
    {
      "epoch": 1.033782958984375e-06,
      "step": 1084,
      "training_step_time": 17.011270999908447
    },
    {
      "epoch": 1.0347366333007812e-06,
      "model_forward_time": 5.59400486946106,
      "step": 1085
    },
    {
      "epoch": 1.0347366333007812e-06,
      "step": 1085,
      "training_step_time": 17.033302307128906
    },
    {
      "epoch": 1.0356903076171876e-06,
      "model_forward_time": 5.595252275466919,
      "step": 1086
    },
    {
      "epoch": 1.0356903076171876e-06,
      "step": 1086,
      "training_step_time": 17.066410541534424
    },
    {
      "epoch": 1.0366439819335937e-06,
      "model_forward_time": 5.602933883666992,
      "step": 1087
    },
    {
      "epoch": 1.0366439819335937e-06,
      "step": 1087,
      "training_step_time": 17.081897497177124
    },
    {
      "epoch": 1.03759765625e-06,
      "model_forward_time": 5.96350359916687,
      "step": 1088
    },
    {
      "epoch": 1.03759765625e-06,
      "step": 1088,
      "training_step_time": 17.434381008148193
    },
    {
      "epoch": 1.0385513305664062e-06,
      "model_forward_time": 5.598262548446655,
      "step": 1089
    },
    {
      "epoch": 1.0385513305664062e-06,
      "step": 1089,
      "training_step_time": 17.061038494110107
    },
    {
      "epoch": 1.0395050048828126e-06,
      "grad_norm": 1.2170767784118652,
      "learning_rate": 1e-05,
      "loss": 0.0943,
      "step": 1090
    },
    {
      "epoch": 1.0395050048828126e-06,
      "model_forward_time": 5.590310573577881,
      "step": 1090
    },
    {
      "dynamics_loss_avg": 0.07181701548397541,
      "epoch": 1.0395050048828126e-06,
      "step": 1090
    },
    {
      "action_loss_avg": 0.02229562634602189,
      "epoch": 1.0395050048828126e-06,
      "step": 1090
    },
    {
      "epoch": 1.0395050048828126e-06,
      "step": 1090,
      "training_step_time": 17.02824306488037
    },
    {
      "epoch": 1.0404586791992187e-06,
      "model_forward_time": 5.589348793029785,
      "step": 1091
    },
    {
      "epoch": 1.0404586791992187e-06,
      "step": 1091,
      "training_step_time": 17.03334379196167
    },
    {
      "epoch": 1.041412353515625e-06,
      "model_forward_time": 5.602860689163208,
      "step": 1092
    },
    {
      "epoch": 1.041412353515625e-06,
      "step": 1092,
      "training_step_time": 17.048624515533447
    },
    {
      "epoch": 1.0423660278320312e-06,
      "model_forward_time": 5.9477620124816895,
      "step": 1093
    },
    {
      "epoch": 1.0423660278320312e-06,
      "step": 1093,
      "training_step_time": 17.42202615737915
    },
    {
      "epoch": 1.0433197021484376e-06,
      "model_forward_time": 5.605731964111328,
      "step": 1094
    },
    {
      "epoch": 1.0433197021484376e-06,
      "step": 1094,
      "training_step_time": 17.09499979019165
    },
    {
      "epoch": 1.0442733764648437e-06,
      "model_forward_time": 5.5965306758880615,
      "step": 1095
    },
    {
      "epoch": 1.0442733764648437e-06,
      "step": 1095,
      "training_step_time": 17.068975925445557
    },
    {
      "epoch": 1.04522705078125e-06,
      "model_forward_time": 5.597944259643555,
      "step": 1096
    },
    {
      "epoch": 1.04522705078125e-06,
      "step": 1096,
      "training_step_time": 17.023653984069824
    },
    {
      "epoch": 1.0461807250976562e-06,
      "model_forward_time": 5.578910827636719,
      "step": 1097
    },
    {
      "epoch": 1.0461807250976562e-06,
      "step": 1097,
      "training_step_time": 17.002243518829346
    },
    {
      "epoch": 1.0471343994140626e-06,
      "model_forward_time": 5.586470603942871,
      "step": 1098
    },
    {
      "epoch": 1.0471343994140626e-06,
      "step": 1098,
      "training_step_time": 17.039401054382324
    },
    {
      "epoch": 1.0480880737304687e-06,
      "model_forward_time": 5.962259292602539,
      "step": 1099
    },
    {
      "epoch": 1.0480880737304687e-06,
      "step": 1099,
      "training_step_time": 17.44432806968689
    },
    {
      "epoch": 1.049041748046875e-06,
      "grad_norm": 1.4053651094436646,
      "learning_rate": 1e-05,
      "loss": 0.0813,
      "step": 1100
    },
    {
      "epoch": 1.049041748046875e-06,
      "model_forward_time": 5.618964433670044,
      "step": 1100
    },
    {
      "dynamics_loss_avg": 0.05864292737096548,
      "epoch": 1.049041748046875e-06,
      "step": 1100
    },
    {
      "action_loss_avg": 0.019613927160389722,
      "epoch": 1.049041748046875e-06,
      "step": 1100
    },
    {
      "epoch": 1.049041748046875e-06,
      "step": 1100,
      "training_step_time": 17.119489192962646
    },
    {
      "epoch": 1.0499954223632812e-06,
      "model_forward_time": 5.612155199050903,
      "step": 1101
    },
    {
      "epoch": 1.0499954223632812e-06,
      "step": 1101,
      "training_step_time": 17.094552040100098
    },
    {
      "epoch": 1.0509490966796876e-06,
      "model_forward_time": 5.622186660766602,
      "step": 1102
    },
    {
      "epoch": 1.0509490966796876e-06,
      "step": 1102,
      "training_step_time": 17.11436653137207
    },
    {
      "epoch": 1.0519027709960937e-06,
      "model_forward_time": 5.61642050743103,
      "step": 1103
    },
    {
      "epoch": 1.0519027709960937e-06,
      "step": 1103,
      "training_step_time": 17.10522150993347
    },
    {
      "epoch": 1.0528564453125e-06,
      "model_forward_time": 5.955403804779053,
      "step": 1104
    },
    {
      "epoch": 1.0528564453125e-06,
      "step": 1104,
      "training_step_time": 17.430789947509766
    },
    {
      "epoch": 1.0538101196289062e-06,
      "model_forward_time": 5.610955238342285,
      "step": 1105
    },
    {
      "epoch": 1.0538101196289062e-06,
      "step": 1105,
      "training_step_time": 17.099174976348877
    },
    {
      "epoch": 1.0547637939453126e-06,
      "model_forward_time": 5.603537559509277,
      "step": 1106
    },
    {
      "epoch": 1.0547637939453126e-06,
      "step": 1106,
      "training_step_time": 17.07270073890686
    },
    {
      "epoch": 1.0557174682617187e-06,
      "model_forward_time": 5.599250078201294,
      "step": 1107
    },
    {
      "epoch": 1.0557174682617187e-06,
      "step": 1107,
      "training_step_time": 17.078927278518677
    },
    {
      "epoch": 1.056671142578125e-06,
      "model_forward_time": 5.6119701862335205,
      "step": 1108
    },
    {
      "epoch": 1.056671142578125e-06,
      "step": 1108,
      "training_step_time": 17.11678671836853
    },
    {
      "epoch": 1.0576248168945312e-06,
      "model_forward_time": 5.613335609436035,
      "step": 1109
    },
    {
      "epoch": 1.0576248168945312e-06,
      "step": 1109,
      "training_step_time": 17.440086364746094
    },
    {
      "epoch": 1.0585784912109376e-06,
      "grad_norm": 1.1858540773391724,
      "learning_rate": 1e-05,
      "loss": 0.0905,
      "step": 1110
    },
    {
      "epoch": 1.0585784912109376e-06,
      "model_forward_time": 5.59270715713501,
      "step": 1110
    },
    {
      "dynamics_loss_avg": 0.07588890418410302,
      "epoch": 1.0585784912109376e-06,
      "step": 1110
    },
    {
      "action_loss_avg": 0.02591268392279744,
      "epoch": 1.0585784912109376e-06,
      "step": 1110
    },
    {
      "epoch": 1.0585784912109376e-06,
      "step": 1110,
      "training_step_time": 17.025681018829346
    },
    {
      "epoch": 1.0595321655273437e-06,
      "model_forward_time": 5.580126523971558,
      "step": 1111
    },
    {
      "epoch": 1.0595321655273437e-06,
      "step": 1111,
      "training_step_time": 17.00350594520569
    },
    {
      "epoch": 1.06048583984375e-06,
      "model_forward_time": 5.58539342880249,
      "step": 1112
    },
    {
      "epoch": 1.06048583984375e-06,
      "step": 1112,
      "training_step_time": 17.030702829360962
    },
    {
      "epoch": 1.0614395141601562e-06,
      "model_forward_time": 5.598445177078247,
      "step": 1113
    },
    {
      "epoch": 1.0614395141601562e-06,
      "step": 1113,
      "training_step_time": 17.04350996017456
    },
    {
      "epoch": 1.0623931884765626e-06,
      "model_forward_time": 5.598259449005127,
      "step": 1114
    },
    {
      "epoch": 1.0623931884765626e-06,
      "step": 1114,
      "training_step_time": 17.052319288253784
    },
    {
      "epoch": 1.0633468627929687e-06,
      "model_forward_time": 5.963724374771118,
      "step": 1115
    },
    {
      "epoch": 1.0633468627929687e-06,
      "step": 1115,
      "training_step_time": 17.391746044158936
    },
    {
      "epoch": 1.064300537109375e-06,
      "model_forward_time": 5.589806079864502,
      "step": 1116
    },
    {
      "epoch": 1.064300537109375e-06,
      "step": 1116,
      "training_step_time": 17.018869161605835
    },
    {
      "epoch": 1.0652542114257812e-06,
      "model_forward_time": 5.5839128494262695,
      "step": 1117
    },
    {
      "epoch": 1.0652542114257812e-06,
      "step": 1117,
      "training_step_time": 17.004592418670654
    },
    {
      "epoch": 1.0662078857421876e-06,
      "model_forward_time": 5.59640097618103,
      "step": 1118
    },
    {
      "epoch": 1.0662078857421876e-06,
      "step": 1118,
      "training_step_time": 17.05600905418396
    },
    {
      "epoch": 1.0671615600585937e-06,
      "model_forward_time": 5.6020941734313965,
      "step": 1119
    },
    {
      "epoch": 1.0671615600585937e-06,
      "step": 1119,
      "training_step_time": 17.091838121414185
    },
    {
      "epoch": 1.068115234375e-06,
      "grad_norm": 1.2006551027297974,
      "learning_rate": 1e-05,
      "loss": 0.1134,
      "step": 1120
    },
    {
      "epoch": 1.068115234375e-06,
      "model_forward_time": 5.987655401229858,
      "step": 1120
    },
    {
      "dynamics_loss_avg": 0.07876896560192108,
      "epoch": 1.068115234375e-06,
      "step": 1120
    },
    {
      "action_loss_avg": 0.03183706430718303,
      "epoch": 1.068115234375e-06,
      "step": 1120
    },
    {
      "epoch": 1.068115234375e-06,
      "step": 1120,
      "training_step_time": 17.498772621154785
    },
    {
      "epoch": 1.0690689086914062e-06,
      "model_forward_time": 5.618924617767334,
      "step": 1121
    },
    {
      "epoch": 1.0690689086914062e-06,
      "step": 1121,
      "training_step_time": 17.130147218704224
    },
    {
      "epoch": 1.0700225830078126e-06,
      "model_forward_time": 5.616003751754761,
      "step": 1122
    },
    {
      "epoch": 1.0700225830078126e-06,
      "step": 1122,
      "training_step_time": 17.10252332687378
    },
    {
      "epoch": 1.0709762573242187e-06,
      "model_forward_time": 5.5897417068481445,
      "step": 1123
    },
    {
      "epoch": 1.0709762573242187e-06,
      "step": 1123,
      "training_step_time": 17.02898335456848
    },
    {
      "epoch": 1.071929931640625e-06,
      "model_forward_time": 5.575504302978516,
      "step": 1124
    },
    {
      "epoch": 1.071929931640625e-06,
      "step": 1124,
      "training_step_time": 16.991247177124023
    },
    {
      "epoch": 1.0728836059570312e-06,
      "model_forward_time": 5.585094928741455,
      "step": 1125
    },
    {
      "epoch": 1.0728836059570312e-06,
      "step": 1125,
      "training_step_time": 17.005940675735474
    },
    {
      "epoch": 1.0738372802734374e-06,
      "model_forward_time": 5.9309186935424805,
      "step": 1126
    },
    {
      "epoch": 1.0738372802734374e-06,
      "step": 1126,
      "training_step_time": 17.354206800460815
    },
    {
      "epoch": 1.0747909545898438e-06,
      "model_forward_time": 5.602313995361328,
      "step": 1127
    },
    {
      "epoch": 1.0747909545898438e-06,
      "step": 1127,
      "training_step_time": 17.088757038116455
    },
    {
      "epoch": 1.07574462890625e-06,
      "model_forward_time": 5.606518507003784,
      "step": 1128
    },
    {
      "epoch": 1.07574462890625e-06,
      "step": 1128,
      "training_step_time": 17.064886569976807
    },
    {
      "epoch": 1.0766983032226563e-06,
      "model_forward_time": 5.585988759994507,
      "step": 1129
    },
    {
      "epoch": 1.0766983032226563e-06,
      "step": 1129,
      "training_step_time": 17.011635541915894
    },
    {
      "epoch": 1.0776519775390624e-06,
      "grad_norm": 0.6697179079055786,
      "learning_rate": 1e-05,
      "loss": 0.0952,
      "step": 1130
    },
    {
      "epoch": 1.0776519775390624e-06,
      "model_forward_time": 5.575880289077759,
      "step": 1130
    },
    {
      "dynamics_loss_avg": 0.07078451607376338,
      "epoch": 1.0776519775390624e-06,
      "step": 1130
    },
    {
      "action_loss_avg": 0.02534282556734979,
      "epoch": 1.0776519775390624e-06,
      "step": 1130
    },
    {
      "epoch": 1.0776519775390624e-06,
      "step": 1130,
      "training_step_time": 17.017694234848022
    },
    {
      "epoch": 1.0786056518554688e-06,
      "model_forward_time": 5.948629379272461,
      "step": 1131
    },
    {
      "epoch": 1.0786056518554688e-06,
      "step": 1131,
      "training_step_time": 17.365511417388916
    },
    {
      "epoch": 1.079559326171875e-06,
      "model_forward_time": 5.580777645111084,
      "step": 1132
    },
    {
      "epoch": 1.079559326171875e-06,
      "step": 1132,
      "training_step_time": 16.99139714241028
    },
    {
      "epoch": 1.0805130004882813e-06,
      "model_forward_time": 5.5843727588653564,
      "step": 1133
    },
    {
      "epoch": 1.0805130004882813e-06,
      "step": 1133,
      "training_step_time": 17.017525911331177
    },
    {
      "epoch": 1.0814666748046874e-06,
      "model_forward_time": 5.577566623687744,
      "step": 1134
    },
    {
      "epoch": 1.0814666748046874e-06,
      "step": 1134,
      "training_step_time": 17.016942501068115
    },
    {
      "epoch": 1.0824203491210938e-06,
      "model_forward_time": 5.5814220905303955,
      "step": 1135
    },
    {
      "epoch": 1.0824203491210938e-06,
      "step": 1135,
      "training_step_time": 17.005824089050293
    },
    {
      "epoch": 1.0833740234375e-06,
      "model_forward_time": 5.592589378356934,
      "step": 1136
    },
    {
      "epoch": 1.0833740234375e-06,
      "step": 1136,
      "training_step_time": 17.081268548965454
    },
    {
      "epoch": 1.0843276977539063e-06,
      "model_forward_time": 5.9579949378967285,
      "step": 1137
    },
    {
      "epoch": 1.0843276977539063e-06,
      "step": 1137,
      "training_step_time": 17.439491271972656
    },
    {
      "epoch": 1.0852813720703124e-06,
      "model_forward_time": 5.5933332443237305,
      "step": 1138
    },
    {
      "epoch": 1.0852813720703124e-06,
      "step": 1138,
      "training_step_time": 17.037546634674072
    },
    {
      "epoch": 1.0862350463867188e-06,
      "model_forward_time": 5.604494333267212,
      "step": 1139
    },
    {
      "epoch": 1.0862350463867188e-06,
      "step": 1139,
      "training_step_time": 17.074265480041504
    },
    {
      "epoch": 1.087188720703125e-06,
      "grad_norm": 1.1424356698989868,
      "learning_rate": 1e-05,
      "loss": 0.0959,
      "step": 1140
    },
    {
      "epoch": 1.087188720703125e-06,
      "model_forward_time": 5.604321718215942,
      "step": 1140
    },
    {
      "dynamics_loss_avg": 0.06735350154340267,
      "epoch": 1.087188720703125e-06,
      "step": 1140
    },
    {
      "action_loss_avg": 0.020543298730626703,
      "epoch": 1.087188720703125e-06,
      "step": 1140
    },
    {
      "epoch": 1.087188720703125e-06,
      "step": 1140,
      "training_step_time": 17.044604063034058
    },
    {
      "epoch": 1.0881423950195313e-06,
      "model_forward_time": 5.594532489776611,
      "step": 1141
    },
    {
      "epoch": 1.0881423950195313e-06,
      "step": 1141,
      "training_step_time": 17.04653573036194
    },
    {
      "epoch": 1.0890960693359374e-06,
      "model_forward_time": 5.963591814041138,
      "step": 1142
    },
    {
      "epoch": 1.0890960693359374e-06,
      "step": 1142,
      "training_step_time": 17.442601203918457
    },
    {
      "epoch": 1.0900497436523438e-06,
      "model_forward_time": 5.615708112716675,
      "step": 1143
    },
    {
      "epoch": 1.0900497436523438e-06,
      "step": 1143,
      "training_step_time": 17.119604349136353
    },
    {
      "epoch": 1.09100341796875e-06,
      "model_forward_time": 5.62690544128418,
      "step": 1144
    },
    {
      "epoch": 1.09100341796875e-06,
      "step": 1144,
      "training_step_time": 17.140822649002075
    },
    {
      "epoch": 1.0919570922851563e-06,
      "model_forward_time": 5.616986274719238,
      "step": 1145
    },
    {
      "epoch": 1.0919570922851563e-06,
      "step": 1145,
      "training_step_time": 17.07725238800049
    },
    {
      "epoch": 1.0929107666015624e-06,
      "model_forward_time": 5.5857532024383545,
      "step": 1146
    },
    {
      "epoch": 1.0929107666015624e-06,
      "step": 1146,
      "training_step_time": 17.02755117416382
    },
    {
      "epoch": 1.0938644409179688e-06,
      "model_forward_time": 5.585647106170654,
      "step": 1147
    },
    {
      "epoch": 1.0938644409179688e-06,
      "step": 1147,
      "training_step_time": 17.012766122817993
    },
    {
      "epoch": 1.094818115234375e-06,
      "model_forward_time": 5.9385552406311035,
      "step": 1148
    },
    {
      "epoch": 1.094818115234375e-06,
      "step": 1148,
      "training_step_time": 17.36747145652771
    },
    {
      "epoch": 1.0957717895507813e-06,
      "model_forward_time": 5.595489025115967,
      "step": 1149
    },
    {
      "epoch": 1.0957717895507813e-06,
      "step": 1149,
      "training_step_time": 17.038119554519653
    },
    {
      "epoch": 1.0967254638671874e-06,
      "grad_norm": 2.162452459335327,
      "learning_rate": 1e-05,
      "loss": 0.0955,
      "step": 1150
    },
    {
      "epoch": 1.0967254638671874e-06,
      "model_forward_time": 5.599528551101685,
      "step": 1150
    },
    {
      "dynamics_loss_avg": 0.07224548049271107,
      "epoch": 1.0967254638671874e-06,
      "step": 1150
    },
    {
      "action_loss_avg": 0.027417072374373675,
      "epoch": 1.0967254638671874e-06,
      "step": 1150
    },
    {
      "epoch": 1.0967254638671874e-06,
      "step": 1150,
      "training_step_time": 17.076354026794434
    },
    {
      "epoch": 1.0976791381835938e-06,
      "model_forward_time": 5.613846302032471,
      "step": 1151
    },
    {
      "epoch": 1.0976791381835938e-06,
      "step": 1151,
      "training_step_time": 17.09319806098938
    },
    {
      "epoch": 1.0986328125e-06,
      "model_forward_time": 5.58337140083313,
      "step": 1152
    },
    {
      "epoch": 1.0986328125e-06,
      "step": 1152,
      "training_step_time": 16.988451957702637
    },
    {
      "epoch": 1.0995864868164063e-06,
      "model_forward_time": 5.928481578826904,
      "step": 1153
    },
    {
      "epoch": 1.0995864868164063e-06,
      "step": 1153,
      "training_step_time": 17.35532784461975
    },
    {
      "epoch": 1.1005401611328124e-06,
      "model_forward_time": 5.578182697296143,
      "step": 1154
    },
    {
      "epoch": 1.1005401611328124e-06,
      "step": 1154,
      "training_step_time": 16.985586881637573
    },
    {
      "epoch": 1.1014938354492188e-06,
      "model_forward_time": 5.585666656494141,
      "step": 1155
    },
    {
      "epoch": 1.1014938354492188e-06,
      "step": 1155,
      "training_step_time": 17.015416622161865
    },
    {
      "epoch": 1.102447509765625e-06,
      "model_forward_time": 5.596662521362305,
      "step": 1156
    },
    {
      "epoch": 1.102447509765625e-06,
      "step": 1156,
      "training_step_time": 17.077725172042847
    },
    {
      "epoch": 1.1034011840820313e-06,
      "model_forward_time": 5.634331703186035,
      "step": 1157
    },
    {
      "epoch": 1.1034011840820313e-06,
      "step": 1157,
      "training_step_time": 17.15055537223816
    },
    {
      "epoch": 1.1043548583984374e-06,
      "model_forward_time": 5.980623006820679,
      "step": 1158
    },
    {
      "epoch": 1.1043548583984374e-06,
      "step": 1158,
      "training_step_time": 17.49243402481079
    },
    {
      "epoch": 1.1053085327148438e-06,
      "model_forward_time": 5.6145501136779785,
      "step": 1159
    },
    {
      "epoch": 1.1053085327148438e-06,
      "step": 1159,
      "training_step_time": 17.102766036987305
    },
    {
      "epoch": 1.10626220703125e-06,
      "grad_norm": 2.0924034118652344,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 1160
    },
    {
      "epoch": 1.10626220703125e-06,
      "model_forward_time": 5.604074478149414,
      "step": 1160
    },
    {
      "dynamics_loss_avg": 0.0605822766199708,
      "epoch": 1.10626220703125e-06,
      "step": 1160
    },
    {
      "action_loss_avg": 0.01910497238859534,
      "epoch": 1.10626220703125e-06,
      "step": 1160
    },
    {
      "epoch": 1.10626220703125e-06,
      "step": 1160,
      "training_step_time": 17.082713842391968
    },
    {
      "epoch": 1.1072158813476563e-06,
      "model_forward_time": 5.621005296707153,
      "step": 1161
    },
    {
      "epoch": 1.1072158813476563e-06,
      "step": 1161,
      "training_step_time": 17.0852952003479
    },
    {
      "epoch": 1.1081695556640624e-06,
      "model_forward_time": 5.623276472091675,
      "step": 1162
    },
    {
      "epoch": 1.1081695556640624e-06,
      "step": 1162,
      "training_step_time": 17.089629650115967
    },
    {
      "epoch": 1.1091232299804688e-06,
      "model_forward_time": 5.615360975265503,
      "step": 1163
    },
    {
      "epoch": 1.1091232299804688e-06,
      "step": 1163,
      "training_step_time": 17.062747716903687
    },
    {
      "epoch": 1.110076904296875e-06,
      "model_forward_time": 5.95822548866272,
      "step": 1164
    },
    {
      "epoch": 1.110076904296875e-06,
      "step": 1164,
      "training_step_time": 17.40284776687622
    },
    {
      "epoch": 1.1110305786132813e-06,
      "model_forward_time": 5.593998670578003,
      "step": 1165
    },
    {
      "epoch": 1.1110305786132813e-06,
      "step": 1165,
      "training_step_time": 17.0408833026886
    },
    {
      "epoch": 1.1119842529296874e-06,
      "model_forward_time": 5.604634046554565,
      "step": 1166
    },
    {
      "epoch": 1.1119842529296874e-06,
      "step": 1166,
      "training_step_time": 17.08290147781372
    },
    {
      "epoch": 1.1129379272460938e-06,
      "model_forward_time": 5.599013805389404,
      "step": 1167
    },
    {
      "epoch": 1.1129379272460938e-06,
      "step": 1167,
      "training_step_time": 17.068650245666504
    },
    {
      "epoch": 1.1138916015625e-06,
      "model_forward_time": 5.601745128631592,
      "step": 1168
    },
    {
      "epoch": 1.1138916015625e-06,
      "step": 1168,
      "training_step_time": 17.068187713623047
    },
    {
      "epoch": 1.1148452758789063e-06,
      "model_forward_time": 5.608053684234619,
      "step": 1169
    },
    {
      "epoch": 1.1148452758789063e-06,
      "step": 1169,
      "training_step_time": 17.415852546691895
    },
    {
      "epoch": 1.1157989501953124e-06,
      "grad_norm": 1.8984791040420532,
      "learning_rate": 1e-05,
      "loss": 0.1015,
      "step": 1170
    },
    {
      "epoch": 1.1157989501953124e-06,
      "model_forward_time": 5.595646858215332,
      "step": 1170
    },
    {
      "dynamics_loss_avg": 0.07964642122387885,
      "epoch": 1.1157989501953124e-06,
      "step": 1170
    },
    {
      "action_loss_avg": 0.029798766039311884,
      "epoch": 1.1157989501953124e-06,
      "step": 1170
    },
    {
      "epoch": 1.1157989501953124e-06,
      "step": 1170,
      "training_step_time": 17.057449102401733
    },
    {
      "epoch": 1.1167526245117188e-06,
      "model_forward_time": 5.596156358718872,
      "step": 1171
    },
    {
      "epoch": 1.1167526245117188e-06,
      "step": 1171,
      "training_step_time": 17.054832220077515
    },
    {
      "epoch": 1.117706298828125e-06,
      "model_forward_time": 5.599807977676392,
      "step": 1172
    },
    {
      "epoch": 1.117706298828125e-06,
      "step": 1172,
      "training_step_time": 17.068382263183594
    },
    {
      "epoch": 1.1186599731445313e-06,
      "model_forward_time": 5.608366250991821,
      "step": 1173
    },
    {
      "epoch": 1.1186599731445313e-06,
      "step": 1173,
      "training_step_time": 17.083929300308228
    },
    {
      "epoch": 1.1196136474609374e-06,
      "model_forward_time": 5.604277610778809,
      "step": 1174
    },
    {
      "epoch": 1.1196136474609374e-06,
      "step": 1174,
      "training_step_time": 17.09917688369751
    },
    {
      "epoch": 1.1205673217773438e-06,
      "model_forward_time": 5.978312253952026,
      "step": 1175
    },
    {
      "epoch": 1.1205673217773438e-06,
      "step": 1175,
      "training_step_time": 17.451821327209473
    },
    {
      "epoch": 1.12152099609375e-06,
      "model_forward_time": 5.593801498413086,
      "step": 1176
    },
    {
      "epoch": 1.12152099609375e-06,
      "step": 1176,
      "training_step_time": 17.029866218566895
    },
    {
      "epoch": 1.1224746704101563e-06,
      "model_forward_time": 5.572296142578125,
      "step": 1177
    },
    {
      "epoch": 1.1224746704101563e-06,
      "step": 1177,
      "training_step_time": 16.97569990158081
    },
    {
      "epoch": 1.1234283447265624e-06,
      "model_forward_time": 5.5636560916900635,
      "step": 1178
    },
    {
      "epoch": 1.1234283447265624e-06,
      "step": 1178,
      "training_step_time": 16.979357719421387
    },
    {
      "epoch": 1.1243820190429688e-06,
      "model_forward_time": 5.56606388092041,
      "step": 1179
    },
    {
      "epoch": 1.1243820190429688e-06,
      "step": 1179,
      "training_step_time": 16.982995748519897
    },
    {
      "epoch": 1.125335693359375e-06,
      "grad_norm": 1.8634237051010132,
      "learning_rate": 1e-05,
      "loss": 0.0796,
      "step": 1180
    },
    {
      "epoch": 1.125335693359375e-06,
      "model_forward_time": 5.565872669219971,
      "step": 1180
    },
    {
      "dynamics_loss_avg": 0.05277664884924889,
      "epoch": 1.125335693359375e-06,
      "step": 1180
    },
    {
      "action_loss_avg": 0.018457038188353182,
      "epoch": 1.125335693359375e-06,
      "step": 1180
    },
    {
      "epoch": 1.125335693359375e-06,
      "step": 1180,
      "training_step_time": 17.388795614242554
    },
    {
      "epoch": 1.1262893676757813e-06,
      "model_forward_time": 5.574473142623901,
      "step": 1181
    },
    {
      "epoch": 1.1262893676757813e-06,
      "step": 1181,
      "training_step_time": 16.99153232574463
    },
    {
      "epoch": 1.1272430419921874e-06,
      "model_forward_time": 5.591947793960571,
      "step": 1182
    },
    {
      "epoch": 1.1272430419921874e-06,
      "step": 1182,
      "training_step_time": 17.030102729797363
    },
    {
      "epoch": 1.1281967163085938e-06,
      "model_forward_time": 5.5914928913116455,
      "step": 1183
    },
    {
      "epoch": 1.1281967163085938e-06,
      "step": 1183,
      "training_step_time": 17.050708293914795
    },
    {
      "epoch": 1.129150390625e-06,
      "model_forward_time": 5.6003899574279785,
      "step": 1184
    },
    {
      "epoch": 1.129150390625e-06,
      "step": 1184,
      "training_step_time": 17.07074737548828
    },
    {
      "epoch": 1.1301040649414063e-06,
      "model_forward_time": 5.5968804359436035,
      "step": 1185
    },
    {
      "epoch": 1.1301040649414063e-06,
      "step": 1185,
      "training_step_time": 17.04783868789673
    },
    {
      "epoch": 1.1310577392578124e-06,
      "model_forward_time": 5.9561498165130615,
      "step": 1186
    },
    {
      "epoch": 1.1310577392578124e-06,
      "step": 1186,
      "training_step_time": 17.40308952331543
    },
    {
      "epoch": 1.1320114135742188e-06,
      "model_forward_time": 5.598376750946045,
      "step": 1187
    },
    {
      "epoch": 1.1320114135742188e-06,
      "step": 1187,
      "training_step_time": 16.964655876159668
    },
    {
      "epoch": 1.132965087890625e-06,
      "model_forward_time": 5.586026191711426,
      "step": 1188
    },
    {
      "epoch": 1.132965087890625e-06,
      "step": 1188,
      "training_step_time": 17.01098394393921
    },
    {
      "epoch": 1.1339187622070313e-06,
      "model_forward_time": 5.590216398239136,
      "step": 1189
    },
    {
      "epoch": 1.1339187622070313e-06,
      "step": 1189,
      "training_step_time": 17.03183102607727
    },
    {
      "epoch": 1.1348724365234374e-06,
      "grad_norm": 1.732839584350586,
      "learning_rate": 1e-05,
      "loss": 0.0869,
      "step": 1190
    },
    {
      "epoch": 1.1348724365234374e-06,
      "model_forward_time": 5.5948803424835205,
      "step": 1190
    },
    {
      "dynamics_loss_avg": 0.06610061265528203,
      "epoch": 1.1348724365234374e-06,
      "step": 1190
    },
    {
      "action_loss_avg": 0.02348934058099985,
      "epoch": 1.1348724365234374e-06,
      "step": 1190
    },
    {
      "epoch": 1.1348724365234374e-06,
      "step": 1190,
      "training_step_time": 17.058455228805542
    },
    {
      "epoch": 1.1358261108398438e-06,
      "model_forward_time": 5.95046067237854,
      "step": 1191
    },
    {
      "epoch": 1.1358261108398438e-06,
      "step": 1191,
      "training_step_time": 17.438488721847534
    },
    {
      "epoch": 1.13677978515625e-06,
      "model_forward_time": 5.6256091594696045,
      "step": 1192
    },
    {
      "epoch": 1.13677978515625e-06,
      "step": 1192,
      "training_step_time": 17.026155948638916
    },
    {
      "epoch": 1.1377334594726563e-06,
      "model_forward_time": 5.616949558258057,
      "step": 1193
    },
    {
      "epoch": 1.1377334594726563e-06,
      "step": 1193,
      "training_step_time": 17.105188608169556
    },
    {
      "epoch": 1.1386871337890625e-06,
      "model_forward_time": 5.61209774017334,
      "step": 1194
    },
    {
      "epoch": 1.1386871337890625e-06,
      "step": 1194,
      "training_step_time": 17.094483852386475
    },
    {
      "epoch": 1.1396408081054688e-06,
      "model_forward_time": 5.602380037307739,
      "step": 1195
    },
    {
      "epoch": 1.1396408081054688e-06,
      "step": 1195,
      "training_step_time": 17.04137134552002
    },
    {
      "epoch": 1.140594482421875e-06,
      "model_forward_time": 5.580157518386841,
      "step": 1196
    },
    {
      "epoch": 1.140594482421875e-06,
      "step": 1196,
      "training_step_time": 17.01172661781311
    },
    {
      "epoch": 1.1415481567382813e-06,
      "model_forward_time": 5.952543258666992,
      "step": 1197
    },
    {
      "epoch": 1.1415481567382813e-06,
      "step": 1197,
      "training_step_time": 17.379295587539673
    },
    {
      "epoch": 1.1425018310546875e-06,
      "model_forward_time": 5.593563795089722,
      "step": 1198
    },
    {
      "epoch": 1.1425018310546875e-06,
      "step": 1198,
      "training_step_time": 17.02661895751953
    },
    {
      "epoch": 1.1434555053710938e-06,
      "model_forward_time": 5.602207660675049,
      "step": 1199
    },
    {
      "epoch": 1.1434555053710938e-06,
      "step": 1199,
      "training_step_time": 17.063848733901978
    },
    {
      "epoch": 1.1444091796875e-06,
      "grad_norm": 1.5041258335113525,
      "learning_rate": 1e-05,
      "loss": 0.0995,
      "step": 1200
    },
    {
      "epoch": 1.1444091796875e-06,
      "model_forward_time": 5.600026845932007,
      "step": 1200
    },
    {
      "dynamics_loss_avg": 0.07564058750867844,
      "epoch": 1.1444091796875e-06,
      "step": 1200
    },
    {
      "action_loss_avg": 0.02766162035986781,
      "epoch": 1.1444091796875e-06,
      "step": 1200
    },
    {
      "epoch": 1.1444091796875e-06,
      "step": 1200,
      "training_step_time": 17.065698623657227
    },
    {
      "epoch": 1.1453628540039063e-06,
      "model_forward_time": 5.597033977508545,
      "step": 1201
    },
    {
      "epoch": 1.1453628540039063e-06,
      "step": 1201,
      "training_step_time": 17.045361042022705
    },
    {
      "epoch": 1.1463165283203125e-06,
      "model_forward_time": 5.591616153717041,
      "step": 1202
    },
    {
      "epoch": 1.1463165283203125e-06,
      "step": 1202,
      "training_step_time": 17.361867666244507
    },
    {
      "epoch": 1.1472702026367188e-06,
      "model_forward_time": 5.588441610336304,
      "step": 1203
    },
    {
      "epoch": 1.1472702026367188e-06,
      "step": 1203,
      "training_step_time": 17.03058910369873
    },
    {
      "epoch": 1.148223876953125e-06,
      "model_forward_time": 5.603678226470947,
      "step": 1204
    },
    {
      "epoch": 1.148223876953125e-06,
      "step": 1204,
      "training_step_time": 17.07399868965149
    },
    {
      "epoch": 1.1491775512695313e-06,
      "model_forward_time": 5.606896638870239,
      "step": 1205
    },
    {
      "epoch": 1.1491775512695313e-06,
      "step": 1205,
      "training_step_time": 17.087069988250732
    },
    {
      "epoch": 1.1501312255859375e-06,
      "model_forward_time": 5.600082874298096,
      "step": 1206
    },
    {
      "epoch": 1.1501312255859375e-06,
      "step": 1206,
      "training_step_time": 17.0751690864563
    },
    {
      "epoch": 1.1510848999023438e-06,
      "model_forward_time": 5.599328279495239,
      "step": 1207
    },
    {
      "epoch": 1.1510848999023438e-06,
      "step": 1207,
      "training_step_time": 17.05476665496826
    },
    {
      "epoch": 1.15203857421875e-06,
      "model_forward_time": 5.951728343963623,
      "step": 1208
    },
    {
      "epoch": 1.15203857421875e-06,
      "step": 1208,
      "training_step_time": 17.3771493434906
    },
    {
      "epoch": 1.1529922485351563e-06,
      "model_forward_time": 5.584867477416992,
      "step": 1209
    },
    {
      "epoch": 1.1529922485351563e-06,
      "step": 1209,
      "training_step_time": 17.034677982330322
    },
    {
      "epoch": 1.1539459228515625e-06,
      "grad_norm": 1.6548351049423218,
      "learning_rate": 1e-05,
      "loss": 0.1045,
      "step": 1210
    },
    {
      "epoch": 1.1539459228515625e-06,
      "model_forward_time": 5.58856463432312,
      "step": 1210
    },
    {
      "dynamics_loss_avg": 0.07637450210750103,
      "epoch": 1.1539459228515625e-06,
      "step": 1210
    },
    {
      "action_loss_avg": 0.03328447658568621,
      "epoch": 1.1539459228515625e-06,
      "step": 1210
    },
    {
      "epoch": 1.1539459228515625e-06,
      "step": 1210,
      "training_step_time": 17.079484939575195
    },
    {
      "epoch": 1.1548995971679688e-06,
      "model_forward_time": 5.620693922042847,
      "step": 1211
    },
    {
      "epoch": 1.1548995971679688e-06,
      "step": 1211,
      "training_step_time": 17.1282799243927
    },
    {
      "epoch": 1.155853271484375e-06,
      "model_forward_time": 5.620854616165161,
      "step": 1212
    },
    {
      "epoch": 1.155853271484375e-06,
      "step": 1212,
      "training_step_time": 17.12590980529785
    },
    {
      "epoch": 1.1568069458007813e-06,
      "model_forward_time": 5.624156713485718,
      "step": 1213
    },
    {
      "epoch": 1.1568069458007813e-06,
      "step": 1213,
      "training_step_time": 17.473437547683716
    },
    {
      "epoch": 1.1577606201171875e-06,
      "model_forward_time": 5.6175408363342285,
      "step": 1214
    },
    {
      "epoch": 1.1577606201171875e-06,
      "step": 1214,
      "training_step_time": 17.089900016784668
    },
    {
      "epoch": 1.1587142944335938e-06,
      "model_forward_time": 5.604997158050537,
      "step": 1215
    },
    {
      "epoch": 1.1587142944335938e-06,
      "step": 1215,
      "training_step_time": 17.087491512298584
    },
    {
      "epoch": 1.15966796875e-06,
      "model_forward_time": 5.599009990692139,
      "step": 1216
    },
    {
      "epoch": 1.15966796875e-06,
      "step": 1216,
      "training_step_time": 17.079288721084595
    },
    {
      "epoch": 1.1606216430664063e-06,
      "model_forward_time": 5.608132839202881,
      "step": 1217
    },
    {
      "epoch": 1.1606216430664063e-06,
      "step": 1217,
      "training_step_time": 17.09956455230713
    },
    {
      "epoch": 1.1615753173828125e-06,
      "model_forward_time": 5.607793807983398,
      "step": 1218
    },
    {
      "epoch": 1.1615753173828125e-06,
      "step": 1218,
      "training_step_time": 17.0736825466156
    },
    {
      "epoch": 1.1625289916992188e-06,
      "model_forward_time": 5.96812891960144,
      "step": 1219
    },
    {
      "epoch": 1.1625289916992188e-06,
      "step": 1219,
      "training_step_time": 17.417664527893066
    },
    {
      "epoch": 1.163482666015625e-06,
      "grad_norm": 1.6931687593460083,
      "learning_rate": 1e-05,
      "loss": 0.1033,
      "step": 1220
    },
    {
      "epoch": 1.163482666015625e-06,
      "model_forward_time": 5.602428197860718,
      "step": 1220
    },
    {
      "dynamics_loss_avg": 0.07231778167188167,
      "epoch": 1.163482666015625e-06,
      "step": 1220
    },
    {
      "action_loss_avg": 0.027910447027534247,
      "epoch": 1.163482666015625e-06,
      "step": 1220
    },
    {
      "epoch": 1.163482666015625e-06,
      "step": 1220,
      "training_step_time": 17.055381774902344
    },
    {
      "epoch": 1.1644363403320313e-06,
      "model_forward_time": 5.598360776901245,
      "step": 1221
    },
    {
      "epoch": 1.1644363403320313e-06,
      "step": 1221,
      "training_step_time": 17.03940987586975
    },
    {
      "epoch": 1.1653900146484375e-06,
      "model_forward_time": 5.592724084854126,
      "step": 1222
    },
    {
      "epoch": 1.1653900146484375e-06,
      "step": 1222,
      "training_step_time": 17.018113374710083
    },
    {
      "epoch": 1.1663436889648438e-06,
      "model_forward_time": 5.5920937061309814,
      "step": 1223
    },
    {
      "epoch": 1.1663436889648438e-06,
      "step": 1223,
      "training_step_time": 17.032063007354736
    },
    {
      "epoch": 1.16729736328125e-06,
      "model_forward_time": 5.94104528427124,
      "step": 1224
    },
    {
      "epoch": 1.16729736328125e-06,
      "step": 1224,
      "training_step_time": 17.39507484436035
    },
    {
      "epoch": 1.1682510375976563e-06,
      "model_forward_time": 5.610835313796997,
      "step": 1225
    },
    {
      "epoch": 1.1682510375976563e-06,
      "step": 1225,
      "training_step_time": 17.089030742645264
    },
    {
      "epoch": 1.1692047119140625e-06,
      "model_forward_time": 5.6201491355896,
      "step": 1226
    },
    {
      "epoch": 1.1692047119140625e-06,
      "step": 1226,
      "training_step_time": 17.088970184326172
    },
    {
      "epoch": 1.1701583862304688e-06,
      "model_forward_time": 5.597114086151123,
      "step": 1227
    },
    {
      "epoch": 1.1701583862304688e-06,
      "step": 1227,
      "training_step_time": 17.02512240409851
    },
    {
      "epoch": 1.171112060546875e-06,
      "model_forward_time": 5.581345319747925,
      "step": 1228
    },
    {
      "epoch": 1.171112060546875e-06,
      "step": 1228,
      "training_step_time": 16.997528553009033
    },
    {
      "epoch": 1.1720657348632813e-06,
      "model_forward_time": 5.5788586139678955,
      "step": 1229
    },
    {
      "epoch": 1.1720657348632813e-06,
      "step": 1229,
      "training_step_time": 17.009644269943237
    },
    {
      "epoch": 1.1730194091796875e-06,
      "grad_norm": 1.4668779373168945,
      "learning_rate": 1e-05,
      "loss": 0.093,
      "step": 1230
    },
    {
      "epoch": 1.1730194091796875e-06,
      "model_forward_time": 5.943209886550903,
      "step": 1230
    },
    {
      "dynamics_loss_avg": 0.06965752169489861,
      "epoch": 1.1730194091796875e-06,
      "step": 1230
    },
    {
      "action_loss_avg": 0.024949998036026953,
      "epoch": 1.1730194091796875e-06,
      "step": 1230
    },
    {
      "epoch": 1.1730194091796875e-06,
      "step": 1230,
      "training_step_time": 17.35326337814331
    },
    {
      "epoch": 1.1739730834960938e-06,
      "model_forward_time": 5.5882368087768555,
      "step": 1231
    },
    {
      "epoch": 1.1739730834960938e-06,
      "step": 1231,
      "training_step_time": 17.016358852386475
    },
    {
      "epoch": 1.1749267578125e-06,
      "model_forward_time": 5.583476305007935,
      "step": 1232
    },
    {
      "epoch": 1.1749267578125e-06,
      "step": 1232,
      "training_step_time": 17.023521184921265
    },
    {
      "epoch": 1.1758804321289063e-06,
      "model_forward_time": 5.589804172515869,
      "step": 1233
    },
    {
      "epoch": 1.1758804321289063e-06,
      "step": 1233,
      "training_step_time": 17.066354513168335
    },
    {
      "epoch": 1.1768341064453125e-06,
      "model_forward_time": 5.6062703132629395,
      "step": 1234
    },
    {
      "epoch": 1.1768341064453125e-06,
      "step": 1234,
      "training_step_time": 17.091370820999146
    },
    {
      "epoch": 1.1777877807617188e-06,
      "model_forward_time": 5.609121084213257,
      "step": 1235
    },
    {
      "epoch": 1.1777877807617188e-06,
      "step": 1235,
      "training_step_time": 17.443730354309082
    },
    {
      "epoch": 1.178741455078125e-06,
      "model_forward_time": 5.586062669754028,
      "step": 1236
    },
    {
      "epoch": 1.178741455078125e-06,
      "step": 1236,
      "training_step_time": 17.01309299468994
    },
    {
      "epoch": 1.1796951293945313e-06,
      "model_forward_time": 5.575901031494141,
      "step": 1237
    },
    {
      "epoch": 1.1796951293945313e-06,
      "step": 1237,
      "training_step_time": 17.02087903022766
    },
    {
      "epoch": 1.1806488037109375e-06,
      "model_forward_time": 5.57391095161438,
      "step": 1238
    },
    {
      "epoch": 1.1806488037109375e-06,
      "step": 1238,
      "training_step_time": 16.991655826568604
    },
    {
      "epoch": 1.1816024780273438e-06,
      "model_forward_time": 5.575950384140015,
      "step": 1239
    },
    {
      "epoch": 1.1816024780273438e-06,
      "step": 1239,
      "training_step_time": 16.99864172935486
    },
    {
      "epoch": 1.18255615234375e-06,
      "grad_norm": 2.4851152896881104,
      "learning_rate": 1e-05,
      "loss": 0.1044,
      "step": 1240
    },
    {
      "epoch": 1.18255615234375e-06,
      "model_forward_time": 5.586919069290161,
      "step": 1240
    },
    {
      "dynamics_loss_avg": 0.07437539771199227,
      "epoch": 1.18255615234375e-06,
      "step": 1240
    },
    {
      "action_loss_avg": 0.025374167039990424,
      "epoch": 1.18255615234375e-06,
      "step": 1240
    },
    {
      "epoch": 1.18255615234375e-06,
      "step": 1240,
      "training_step_time": 17.029613494873047
    },
    {
      "epoch": 1.1835098266601563e-06,
      "model_forward_time": 5.976536989212036,
      "step": 1241
    },
    {
      "epoch": 1.1835098266601563e-06,
      "step": 1241,
      "training_step_time": 17.426527738571167
    },
    {
      "epoch": 1.1844635009765625e-06,
      "model_forward_time": 5.5990636348724365,
      "step": 1242
    },
    {
      "epoch": 1.1844635009765625e-06,
      "step": 1242,
      "training_step_time": 17.07134437561035
    },
    {
      "epoch": 1.1854171752929688e-06,
      "model_forward_time": 5.6070876121521,
      "step": 1243
    },
    {
      "epoch": 1.1854171752929688e-06,
      "step": 1243,
      "training_step_time": 17.105878591537476
    },
    {
      "epoch": 1.186370849609375e-06,
      "model_forward_time": 5.6091108322143555,
      "step": 1244
    },
    {
      "epoch": 1.186370849609375e-06,
      "step": 1244,
      "training_step_time": 17.10768961906433
    },
    {
      "epoch": 1.1873245239257814e-06,
      "model_forward_time": 5.6167027950286865,
      "step": 1245
    },
    {
      "epoch": 1.1873245239257814e-06,
      "step": 1245,
      "training_step_time": 17.116086959838867
    },
    {
      "epoch": 1.1882781982421875e-06,
      "model_forward_time": 5.618102788925171,
      "step": 1246
    },
    {
      "epoch": 1.1882781982421875e-06,
      "step": 1246,
      "training_step_time": 17.470758199691772
    },
    {
      "epoch": 1.1892318725585939e-06,
      "model_forward_time": 5.609045028686523,
      "step": 1247
    },
    {
      "epoch": 1.1892318725585939e-06,
      "step": 1247,
      "training_step_time": 17.081819534301758
    },
    {
      "epoch": 1.190185546875e-06,
      "model_forward_time": 5.606184482574463,
      "step": 1248
    },
    {
      "epoch": 1.190185546875e-06,
      "step": 1248,
      "training_step_time": 17.090601205825806
    },
    {
      "epoch": 1.1911392211914064e-06,
      "model_forward_time": 5.609712600708008,
      "step": 1249
    },
    {
      "epoch": 1.1911392211914064e-06,
      "step": 1249,
      "training_step_time": 17.1016788482666
    },
    {
      "epoch": 1.1920928955078125e-06,
      "grad_norm": 0.8929726481437683,
      "learning_rate": 1e-05,
      "loss": 0.091,
      "step": 1250
    },
    {
      "epoch": 1.1920928955078125e-06,
      "model_forward_time": 5.5972888469696045,
      "step": 1250
    },
    {
      "dynamics_loss_avg": 0.0685901839286089,
      "epoch": 1.1920928955078125e-06,
      "step": 1250
    },
    {
      "action_loss_avg": 0.024285950418561696,
      "epoch": 1.1920928955078125e-06,
      "step": 1250
    },
    {
      "epoch": 1.1920928955078125e-06,
      "step": 1250,
      "training_step_time": 17.040817499160767
    },
    {
      "epoch": 1.1930465698242186e-06,
      "model_forward_time": 5.591794013977051,
      "step": 1251
    },
    {
      "epoch": 1.1930465698242186e-06,
      "step": 1251,
      "training_step_time": 17.003363847732544
    },
    {
      "epoch": 1.194000244140625e-06,
      "model_forward_time": 5.938674211502075,
      "step": 1252
    },
    {
      "epoch": 1.194000244140625e-06,
      "step": 1252,
      "training_step_time": 17.38334321975708
    },
    {
      "epoch": 1.1949539184570311e-06,
      "model_forward_time": 5.578593492507935,
      "step": 1253
    },
    {
      "epoch": 1.1949539184570311e-06,
      "step": 1253,
      "training_step_time": 17.02351474761963
    },
    {
      "epoch": 1.1959075927734375e-06,
      "model_forward_time": 5.599271059036255,
      "step": 1254
    },
    {
      "epoch": 1.1959075927734375e-06,
      "step": 1254,
      "training_step_time": 17.086400508880615
    },
    {
      "epoch": 1.1968612670898436e-06,
      "model_forward_time": 5.609832525253296,
      "step": 1255
    },
    {
      "epoch": 1.1968612670898436e-06,
      "step": 1255,
      "training_step_time": 17.117668867111206
    },
    {
      "epoch": 1.19781494140625e-06,
      "model_forward_time": 5.628127813339233,
      "step": 1256
    },
    {
      "epoch": 1.19781494140625e-06,
      "step": 1256,
      "training_step_time": 17.102611303329468
    },
    {
      "epoch": 1.1987686157226562e-06,
      "model_forward_time": 5.965941905975342,
      "step": 1257
    },
    {
      "epoch": 1.1987686157226562e-06,
      "step": 1257,
      "training_step_time": 17.421263933181763
    },
    {
      "epoch": 1.1997222900390625e-06,
      "model_forward_time": 5.598726749420166,
      "step": 1258
    },
    {
      "epoch": 1.1997222900390625e-06,
      "step": 1258,
      "training_step_time": 17.052446842193604
    },
    {
      "epoch": 1.2006759643554687e-06,
      "model_forward_time": 5.619046449661255,
      "step": 1259
    },
    {
      "epoch": 1.2006759643554687e-06,
      "step": 1259,
      "training_step_time": 17.102378368377686
    },
    {
      "epoch": 1.201629638671875e-06,
      "grad_norm": 0.6478079557418823,
      "learning_rate": 1e-05,
      "loss": 0.104,
      "step": 1260
    },
    {
      "epoch": 1.201629638671875e-06,
      "model_forward_time": 5.611713171005249,
      "step": 1260
    },
    {
      "dynamics_loss_avg": 0.07537568621337414,
      "epoch": 1.201629638671875e-06,
      "step": 1260
    },
    {
      "action_loss_avg": 0.02822289764881134,
      "epoch": 1.201629638671875e-06,
      "step": 1260
    },
    {
      "epoch": 1.201629638671875e-06,
      "step": 1260,
      "training_step_time": 17.1125910282135
    },
    {
      "epoch": 1.2025833129882812e-06,
      "model_forward_time": 5.621954679489136,
      "step": 1261
    },
    {
      "epoch": 1.2025833129882812e-06,
      "step": 1261,
      "training_step_time": 17.126409769058228
    },
    {
      "epoch": 1.2035369873046875e-06,
      "model_forward_time": 5.606920003890991,
      "step": 1262
    },
    {
      "epoch": 1.2035369873046875e-06,
      "step": 1262,
      "training_step_time": 17.10442352294922
    },
    {
      "epoch": 1.2044906616210937e-06,
      "model_forward_time": 5.996821880340576,
      "step": 1263
    },
    {
      "epoch": 1.2044906616210937e-06,
      "step": 1263,
      "training_step_time": 17.486454725265503
    },
    {
      "epoch": 1.2054443359375e-06,
      "model_forward_time": 5.611070156097412,
      "step": 1264
    },
    {
      "epoch": 1.2054443359375e-06,
      "step": 1264,
      "training_step_time": 17.091909885406494
    },
    {
      "epoch": 1.2063980102539062e-06,
      "model_forward_time": 5.612128257751465,
      "step": 1265
    },
    {
      "epoch": 1.2063980102539062e-06,
      "step": 1265,
      "training_step_time": 17.099223852157593
    },
    {
      "epoch": 1.2073516845703125e-06,
      "model_forward_time": 5.625770330429077,
      "step": 1266
    },
    {
      "epoch": 1.2073516845703125e-06,
      "step": 1266,
      "training_step_time": 17.11929988861084
    },
    {
      "epoch": 1.2083053588867187e-06,
      "model_forward_time": 5.6126673221588135,
      "step": 1267
    },
    {
      "epoch": 1.2083053588867187e-06,
      "step": 1267,
      "training_step_time": 17.07485270500183
    },
    {
      "epoch": 1.209259033203125e-06,
      "model_forward_time": 5.608313322067261,
      "step": 1268
    },
    {
      "epoch": 1.209259033203125e-06,
      "step": 1268,
      "training_step_time": 17.4271342754364
    },
    {
      "epoch": 1.2102127075195312e-06,
      "model_forward_time": 5.612549781799316,
      "step": 1269
    },
    {
      "epoch": 1.2102127075195312e-06,
      "step": 1269,
      "training_step_time": 17.07656168937683
    },
    {
      "epoch": 1.2111663818359375e-06,
      "grad_norm": 1.599737524986267,
      "learning_rate": 1e-05,
      "loss": 0.0947,
      "step": 1270
    },
    {
      "epoch": 1.2111663818359375e-06,
      "model_forward_time": 5.591836929321289,
      "step": 1270
    },
    {
      "dynamics_loss_avg": 0.07028107549995184,
      "epoch": 1.2111663818359375e-06,
      "step": 1270
    },
    {
      "action_loss_avg": 0.024964731466025115,
      "epoch": 1.2111663818359375e-06,
      "step": 1270
    },
    {
      "epoch": 1.2111663818359375e-06,
      "step": 1270,
      "training_step_time": 17.05661416053772
    },
    {
      "epoch": 1.2121200561523437e-06,
      "model_forward_time": 5.594195127487183,
      "step": 1271
    },
    {
      "epoch": 1.2121200561523437e-06,
      "step": 1271,
      "training_step_time": 17.056461095809937
    },
    {
      "epoch": 1.21307373046875e-06,
      "model_forward_time": 5.5946877002716064,
      "step": 1272
    },
    {
      "epoch": 1.21307373046875e-06,
      "step": 1272,
      "training_step_time": 17.069429636001587
    },
    {
      "epoch": 1.2140274047851562e-06,
      "model_forward_time": 5.5998029708862305,
      "step": 1273
    },
    {
      "epoch": 1.2140274047851562e-06,
      "step": 1273,
      "training_step_time": 17.091556072235107
    },
    {
      "epoch": 1.2149810791015625e-06,
      "model_forward_time": 5.988771677017212,
      "step": 1274
    },
    {
      "epoch": 1.2149810791015625e-06,
      "step": 1274,
      "training_step_time": 17.48212194442749
    },
    {
      "epoch": 1.2159347534179687e-06,
      "model_forward_time": 5.615345239639282,
      "step": 1275
    },
    {
      "epoch": 1.2159347534179687e-06,
      "step": 1275,
      "training_step_time": 17.11270809173584
    },
    {
      "epoch": 1.216888427734375e-06,
      "model_forward_time": 5.612975835800171,
      "step": 1276
    },
    {
      "epoch": 1.216888427734375e-06,
      "step": 1276,
      "training_step_time": 17.10644292831421
    },
    {
      "epoch": 1.2178421020507812e-06,
      "model_forward_time": 5.615614175796509,
      "step": 1277
    },
    {
      "epoch": 1.2178421020507812e-06,
      "step": 1277,
      "training_step_time": 17.090298414230347
    },
    {
      "epoch": 1.2187957763671875e-06,
      "model_forward_time": 5.581579685211182,
      "step": 1278
    },
    {
      "epoch": 1.2187957763671875e-06,
      "step": 1278,
      "training_step_time": 17.015844106674194
    },
    {
      "epoch": 1.2197494506835937e-06,
      "model_forward_time": 5.559351444244385,
      "step": 1279
    },
    {
      "epoch": 1.2197494506835937e-06,
      "step": 1279,
      "training_step_time": 17.346639394760132
    },
    {
      "epoch": 1.220703125e-06,
      "grad_norm": 1.5565662384033203,
      "learning_rate": 1e-05,
      "loss": 0.1,
      "step": 1280
    },
    {
      "epoch": 1.220703125e-06,
      "model_forward_time": 5.560257434844971,
      "step": 1280
    },
    {
      "dynamics_loss_avg": 0.06797417551279068,
      "epoch": 1.220703125e-06,
      "step": 1280
    },
    {
      "action_loss_avg": 0.024064819747582077,
      "epoch": 1.220703125e-06,
      "step": 1280
    },
    {
      "epoch": 1.220703125e-06,
      "step": 1280,
      "training_step_time": 16.966070890426636
    },
    {
      "epoch": 1.2216567993164062e-06,
      "model_forward_time": 5.57453179359436,
      "step": 1281
    },
    {
      "epoch": 1.2216567993164062e-06,
      "step": 1281,
      "training_step_time": 16.980932235717773
    },
    {
      "epoch": 1.2226104736328125e-06,
      "model_forward_time": 5.563254117965698,
      "step": 1282
    },
    {
      "epoch": 1.2226104736328125e-06,
      "step": 1282,
      "training_step_time": 16.99534296989441
    },
    {
      "epoch": 1.2235641479492187e-06,
      "model_forward_time": 5.564183712005615,
      "step": 1283
    },
    {
      "epoch": 1.2235641479492187e-06,
      "step": 1283,
      "training_step_time": 16.98598337173462
    },
    {
      "epoch": 1.224517822265625e-06,
      "model_forward_time": 5.5787506103515625,
      "step": 1284
    },
    {
      "epoch": 1.224517822265625e-06,
      "step": 1284,
      "training_step_time": 17.003760814666748
    },
    {
      "epoch": 1.2254714965820312e-06,
      "model_forward_time": 5.952395915985107,
      "step": 1285
    },
    {
      "epoch": 1.2254714965820312e-06,
      "step": 1285,
      "training_step_time": 17.39059543609619
    },
    {
      "epoch": 1.2264251708984375e-06,
      "model_forward_time": 5.592081546783447,
      "step": 1286
    },
    {
      "epoch": 1.2264251708984375e-06,
      "step": 1286,
      "training_step_time": 17.045193433761597
    },
    {
      "epoch": 1.2273788452148437e-06,
      "model_forward_time": 5.5975048542022705,
      "step": 1287
    },
    {
      "epoch": 1.2273788452148437e-06,
      "step": 1287,
      "training_step_time": 17.077969789505005
    },
    {
      "epoch": 1.22833251953125e-06,
      "model_forward_time": 5.598245143890381,
      "step": 1288
    },
    {
      "epoch": 1.22833251953125e-06,
      "step": 1288,
      "training_step_time": 17.057552814483643
    },
    {
      "epoch": 1.2292861938476562e-06,
      "model_forward_time": 5.593814134597778,
      "step": 1289
    },
    {
      "epoch": 1.2292861938476562e-06,
      "step": 1289,
      "training_step_time": 17.048457145690918
    },
    {
      "epoch": 1.2302398681640625e-06,
      "grad_norm": 1.2051442861557007,
      "learning_rate": 1e-05,
      "loss": 0.0941,
      "step": 1290
    },
    {
      "epoch": 1.2302398681640625e-06,
      "model_forward_time": 5.9539361000061035,
      "step": 1290
    },
    {
      "dynamics_loss_avg": 0.07514215670526028,
      "epoch": 1.2302398681640625e-06,
      "step": 1290
    },
    {
      "action_loss_avg": 0.027268578112125397,
      "epoch": 1.2302398681640625e-06,
      "step": 1290
    },
    {
      "epoch": 1.2302398681640625e-06,
      "step": 1290,
      "training_step_time": 17.4289128780365
    },
    {
      "epoch": 1.2311935424804687e-06,
      "model_forward_time": 5.623953819274902,
      "step": 1291
    },
    {
      "epoch": 1.2311935424804687e-06,
      "step": 1291,
      "training_step_time": 17.124366521835327
    },
    {
      "epoch": 1.232147216796875e-06,
      "model_forward_time": 5.597602367401123,
      "step": 1292
    },
    {
      "epoch": 1.232147216796875e-06,
      "step": 1292,
      "training_step_time": 17.045307874679565
    },
    {
      "epoch": 1.2331008911132812e-06,
      "model_forward_time": 5.586564540863037,
      "step": 1293
    },
    {
      "epoch": 1.2331008911132812e-06,
      "step": 1293,
      "training_step_time": 17.012048721313477
    },
    {
      "epoch": 1.2340545654296875e-06,
      "model_forward_time": 5.581613540649414,
      "step": 1294
    },
    {
      "epoch": 1.2340545654296875e-06,
      "step": 1294,
      "training_step_time": 17.016156435012817
    },
    {
      "epoch": 1.2350082397460937e-06,
      "model_forward_time": 5.593199968338013,
      "step": 1295
    },
    {
      "epoch": 1.2350082397460937e-06,
      "step": 1295,
      "training_step_time": 17.0358407497406
    },
    {
      "epoch": 1.2359619140625e-06,
      "model_forward_time": 5.982070684432983,
      "step": 1296
    },
    {
      "epoch": 1.2359619140625e-06,
      "step": 1296,
      "training_step_time": 17.456852674484253
    },
    {
      "epoch": 1.2369155883789062e-06,
      "model_forward_time": 5.613462209701538,
      "step": 1297
    },
    {
      "epoch": 1.2369155883789062e-06,
      "step": 1297,
      "training_step_time": 17.077263355255127
    },
    {
      "epoch": 1.2378692626953125e-06,
      "model_forward_time": 5.607845783233643,
      "step": 1298
    },
    {
      "epoch": 1.2378692626953125e-06,
      "step": 1298,
      "training_step_time": 17.06012749671936
    },
    {
      "epoch": 1.2388229370117187e-06,
      "model_forward_time": 5.593578338623047,
      "step": 1299
    },
    {
      "epoch": 1.2388229370117187e-06,
      "step": 1299,
      "training_step_time": 17.064718008041382
    },
    {
      "epoch": 1.239776611328125e-06,
      "grad_norm": 2.104570150375366,
      "learning_rate": 1e-05,
      "loss": 0.0908,
      "step": 1300
    },
    {
      "epoch": 1.239776611328125e-06,
      "model_forward_time": 5.613752841949463,
      "step": 1300
    },
    {
      "dynamics_loss_avg": 0.06394335776567459,
      "epoch": 1.239776611328125e-06,
      "step": 1300
    },
    {
      "action_loss_avg": 0.026921534445136786,
      "epoch": 1.239776611328125e-06,
      "step": 1300
    },
    {
      "epoch": 1.239776611328125e-06,
      "step": 1300,
      "training_step_time": 17.10101628303528
    },
    {
      "epoch": 1.2407302856445312e-06,
      "model_forward_time": 5.625575542449951,
      "step": 1301
    },
    {
      "epoch": 1.2407302856445312e-06,
      "step": 1301,
      "training_step_time": 17.4835422039032
    },
    {
      "epoch": 1.2416839599609375e-06,
      "model_forward_time": 5.606329679489136,
      "step": 1302
    },
    {
      "epoch": 1.2416839599609375e-06,
      "step": 1302,
      "training_step_time": 17.09034299850464
    },
    {
      "epoch": 1.2426376342773437e-06,
      "model_forward_time": 5.607934474945068,
      "step": 1303
    },
    {
      "epoch": 1.2426376342773437e-06,
      "step": 1303,
      "training_step_time": 17.06916308403015
    },
    {
      "epoch": 1.24359130859375e-06,
      "model_forward_time": 5.594174861907959,
      "step": 1304
    },
    {
      "epoch": 1.24359130859375e-06,
      "step": 1304,
      "training_step_time": 17.046900272369385
    },
    {
      "epoch": 1.2445449829101562e-06,
      "model_forward_time": 5.597422361373901,
      "step": 1305
    },
    {
      "epoch": 1.2445449829101562e-06,
      "step": 1305,
      "training_step_time": 17.056446313858032
    },
    {
      "epoch": 1.2454986572265625e-06,
      "model_forward_time": 5.60482931137085,
      "step": 1306
    },
    {
      "epoch": 1.2454986572265625e-06,
      "step": 1306,
      "training_step_time": 17.060051918029785
    },
    {
      "epoch": 1.2464523315429687e-06,
      "model_forward_time": 5.953420877456665,
      "step": 1307
    },
    {
      "epoch": 1.2464523315429687e-06,
      "step": 1307,
      "training_step_time": 17.39258122444153
    },
    {
      "epoch": 1.247406005859375e-06,
      "model_forward_time": 5.5860700607299805,
      "step": 1308
    },
    {
      "epoch": 1.247406005859375e-06,
      "step": 1308,
      "training_step_time": 17.001700162887573
    },
    {
      "epoch": 1.2483596801757812e-06,
      "model_forward_time": 5.589903831481934,
      "step": 1309
    },
    {
      "epoch": 1.2483596801757812e-06,
      "step": 1309,
      "training_step_time": 17.034494400024414
    },
    {
      "epoch": 1.2493133544921876e-06,
      "grad_norm": 0.9439994096755981,
      "learning_rate": 1e-05,
      "loss": 0.0914,
      "step": 1310
    },
    {
      "epoch": 1.2493133544921876e-06,
      "model_forward_time": 5.59754490852356,
      "step": 1310
    },
    {
      "dynamics_loss_avg": 0.06948802955448627,
      "epoch": 1.2493133544921876e-06,
      "step": 1310
    },
    {
      "action_loss_avg": 0.02592944698408246,
      "epoch": 1.2493133544921876e-06,
      "step": 1310
    },
    {
      "epoch": 1.2493133544921876e-06,
      "step": 1310,
      "training_step_time": 17.06629991531372
    },
    {
      "epoch": 1.2502670288085937e-06,
      "model_forward_time": 5.608937740325928,
      "step": 1311
    },
    {
      "epoch": 1.2502670288085937e-06,
      "step": 1311,
      "training_step_time": 17.08597493171692
    },
    {
      "epoch": 1.251220703125e-06,
      "model_forward_time": 5.602908134460449,
      "step": 1312
    },
    {
      "epoch": 1.251220703125e-06,
      "step": 1312,
      "training_step_time": 17.442418813705444
    },
    {
      "epoch": 1.2521743774414062e-06,
      "model_forward_time": 5.616362810134888,
      "step": 1313
    },
    {
      "epoch": 1.2521743774414062e-06,
      "step": 1313,
      "training_step_time": 17.139100074768066
    },
    {
      "epoch": 1.2531280517578126e-06,
      "model_forward_time": 5.622605085372925,
      "step": 1314
    },
    {
      "epoch": 1.2531280517578126e-06,
      "step": 1314,
      "training_step_time": 17.13640332221985
    },
    {
      "epoch": 1.2540817260742187e-06,
      "model_forward_time": 5.6141276359558105,
      "step": 1315
    },
    {
      "epoch": 1.2540817260742187e-06,
      "step": 1315,
      "training_step_time": 17.119555711746216
    },
    {
      "epoch": 1.255035400390625e-06,
      "model_forward_time": 5.621883869171143,
      "step": 1316
    },
    {
      "epoch": 1.255035400390625e-06,
      "step": 1316,
      "training_step_time": 17.12471890449524
    },
    {
      "epoch": 1.2559890747070312e-06,
      "model_forward_time": 5.611056089401245,
      "step": 1317
    },
    {
      "epoch": 1.2559890747070312e-06,
      "step": 1317,
      "training_step_time": 17.10854148864746
    },
    {
      "epoch": 1.2569427490234376e-06,
      "model_forward_time": 5.999401092529297,
      "step": 1318
    },
    {
      "epoch": 1.2569427490234376e-06,
      "step": 1318,
      "training_step_time": 17.479608297348022
    },
    {
      "epoch": 1.2578964233398437e-06,
      "model_forward_time": 5.601255655288696,
      "step": 1319
    },
    {
      "epoch": 1.2578964233398437e-06,
      "step": 1319,
      "training_step_time": 17.062660455703735
    },
    {
      "epoch": 1.25885009765625e-06,
      "grad_norm": 2.0751326084136963,
      "learning_rate": 1e-05,
      "loss": 0.1016,
      "step": 1320
    },
    {
      "epoch": 1.25885009765625e-06,
      "model_forward_time": 5.597212314605713,
      "step": 1320
    },
    {
      "dynamics_loss_avg": 0.06701375171542168,
      "epoch": 1.25885009765625e-06,
      "step": 1320
    },
    {
      "action_loss_avg": 0.028544477187097073,
      "epoch": 1.25885009765625e-06,
      "step": 1320
    },
    {
      "epoch": 1.25885009765625e-06,
      "step": 1320,
      "training_step_time": 17.045768976211548
    },
    {
      "epoch": 1.2598037719726562e-06,
      "model_forward_time": 5.597483396530151,
      "step": 1321
    },
    {
      "epoch": 1.2598037719726562e-06,
      "step": 1321,
      "training_step_time": 17.079469919204712
    },
    {
      "epoch": 1.2607574462890626e-06,
      "model_forward_time": 5.61777925491333,
      "step": 1322
    },
    {
      "epoch": 1.2607574462890626e-06,
      "step": 1322,
      "training_step_time": 17.137773275375366
    },
    {
      "epoch": 1.2617111206054687e-06,
      "model_forward_time": 5.992555856704712,
      "step": 1323
    },
    {
      "epoch": 1.2617111206054687e-06,
      "step": 1323,
      "training_step_time": 17.479425191879272
    },
    {
      "epoch": 1.262664794921875e-06,
      "model_forward_time": 5.596971273422241,
      "step": 1324
    },
    {
      "epoch": 1.262664794921875e-06,
      "step": 1324,
      "training_step_time": 17.02796959877014
    },
    {
      "epoch": 1.2636184692382812e-06,
      "model_forward_time": 5.574084043502808,
      "step": 1325
    },
    {
      "epoch": 1.2636184692382812e-06,
      "step": 1325,
      "training_step_time": 16.98693537712097
    },
    {
      "epoch": 1.2645721435546876e-06,
      "model_forward_time": 5.569013357162476,
      "step": 1326
    },
    {
      "epoch": 1.2645721435546876e-06,
      "step": 1326,
      "training_step_time": 16.977405309677124
    },
    {
      "epoch": 1.2655258178710937e-06,
      "model_forward_time": 5.56910514831543,
      "step": 1327
    },
    {
      "epoch": 1.2655258178710937e-06,
      "step": 1327,
      "training_step_time": 16.977264404296875
    },
    {
      "epoch": 1.2664794921875e-06,
      "model_forward_time": 5.5790064334869385,
      "step": 1328
    },
    {
      "epoch": 1.2664794921875e-06,
      "step": 1328,
      "training_step_time": 17.018359184265137
    },
    {
      "epoch": 1.2674331665039062e-06,
      "model_forward_time": 5.962926387786865,
      "step": 1329
    },
    {
      "epoch": 1.2674331665039062e-06,
      "step": 1329,
      "training_step_time": 17.379491090774536
    },
    {
      "epoch": 1.2683868408203126e-06,
      "grad_norm": 1.7219798564910889,
      "learning_rate": 1e-05,
      "loss": 0.0897,
      "step": 1330
    },
    {
      "epoch": 1.2683868408203126e-06,
      "model_forward_time": 5.587417125701904,
      "step": 1330
    },
    {
      "dynamics_loss_avg": 0.06768462136387825,
      "epoch": 1.2683868408203126e-06,
      "step": 1330
    },
    {
      "action_loss_avg": 0.021017582155764103,
      "epoch": 1.2683868408203126e-06,
      "step": 1330
    },
    {
      "epoch": 1.2683868408203126e-06,
      "step": 1330,
      "training_step_time": 17.00614047050476
    },
    {
      "epoch": 1.2693405151367187e-06,
      "model_forward_time": 5.59679102897644,
      "step": 1331
    },
    {
      "epoch": 1.2693405151367187e-06,
      "step": 1331,
      "training_step_time": 17.075052499771118
    },
    {
      "epoch": 1.270294189453125e-06,
      "model_forward_time": 5.630521774291992,
      "step": 1332
    },
    {
      "epoch": 1.270294189453125e-06,
      "step": 1332,
      "training_step_time": 17.158854246139526
    },
    {
      "epoch": 1.2712478637695312e-06,
      "model_forward_time": 5.632890939712524,
      "step": 1333
    },
    {
      "epoch": 1.2712478637695312e-06,
      "step": 1333,
      "training_step_time": 17.141452312469482
    },
    {
      "epoch": 1.2722015380859376e-06,
      "model_forward_time": 5.622430801391602,
      "step": 1334
    },
    {
      "epoch": 1.2722015380859376e-06,
      "step": 1334,
      "training_step_time": 17.481895685195923
    },
    {
      "epoch": 1.2731552124023437e-06,
      "model_forward_time": 5.607868909835815,
      "step": 1335
    },
    {
      "epoch": 1.2731552124023437e-06,
      "step": 1335,
      "training_step_time": 17.082660913467407
    },
    {
      "epoch": 1.27410888671875e-06,
      "model_forward_time": 5.598358869552612,
      "step": 1336
    },
    {
      "epoch": 1.27410888671875e-06,
      "step": 1336,
      "training_step_time": 17.06804919242859
    },
    {
      "epoch": 1.2750625610351562e-06,
      "model_forward_time": 5.601416826248169,
      "step": 1337
    },
    {
      "epoch": 1.2750625610351562e-06,
      "step": 1337,
      "training_step_time": 17.093327522277832
    },
    {
      "epoch": 1.2760162353515626e-06,
      "model_forward_time": 5.613909721374512,
      "step": 1338
    },
    {
      "epoch": 1.2760162353515626e-06,
      "step": 1338,
      "training_step_time": 17.100210189819336
    },
    {
      "epoch": 1.2769699096679687e-06,
      "model_forward_time": 5.597041845321655,
      "step": 1339
    },
    {
      "epoch": 1.2769699096679687e-06,
      "step": 1339,
      "training_step_time": 17.053643226623535
    },
    {
      "epoch": 1.277923583984375e-06,
      "grad_norm": 1.1532455682754517,
      "learning_rate": 1e-05,
      "loss": 0.0889,
      "step": 1340
    },
    {
      "epoch": 1.277923583984375e-06,
      "model_forward_time": 5.959536552429199,
      "step": 1340
    },
    {
      "dynamics_loss_avg": 0.06469252705574036,
      "epoch": 1.277923583984375e-06,
      "step": 1340
    },
    {
      "action_loss_avg": 0.02369825541973114,
      "epoch": 1.277923583984375e-06,
      "step": 1340
    },
    {
      "epoch": 1.277923583984375e-06,
      "step": 1340,
      "training_step_time": 17.40497660636902
    },
    {
      "epoch": 1.2788772583007812e-06,
      "model_forward_time": 5.584456920623779,
      "step": 1341
    },
    {
      "epoch": 1.2788772583007812e-06,
      "step": 1341,
      "training_step_time": 17.015741109848022
    },
    {
      "epoch": 1.2798309326171876e-06,
      "model_forward_time": 5.591371536254883,
      "step": 1342
    },
    {
      "epoch": 1.2798309326171876e-06,
      "step": 1342,
      "training_step_time": 17.033839225769043
    },
    {
      "epoch": 1.2807846069335937e-06,
      "model_forward_time": 5.590074062347412,
      "step": 1343
    },
    {
      "epoch": 1.2807846069335937e-06,
      "step": 1343,
      "training_step_time": 17.04619073867798
    },
    {
      "epoch": 1.28173828125e-06,
      "model_forward_time": 5.606754302978516,
      "step": 1344
    },
    {
      "epoch": 1.28173828125e-06,
      "step": 1344,
      "training_step_time": 17.09518265724182
    },
    {
      "epoch": 1.2826919555664062e-06,
      "model_forward_time": 5.610330820083618,
      "step": 1345
    },
    {
      "epoch": 1.2826919555664062e-06,
      "step": 1345,
      "training_step_time": 17.458553314208984
    },
    {
      "epoch": 1.2836456298828126e-06,
      "model_forward_time": 5.597515106201172,
      "step": 1346
    },
    {
      "epoch": 1.2836456298828126e-06,
      "step": 1346,
      "training_step_time": 17.05754780769348
    },
    {
      "epoch": 1.2845993041992187e-06,
      "model_forward_time": 5.5876312255859375,
      "step": 1347
    },
    {
      "epoch": 1.2845993041992187e-06,
      "step": 1347,
      "training_step_time": 17.015811443328857
    },
    {
      "epoch": 1.285552978515625e-06,
      "model_forward_time": 5.58003306388855,
      "step": 1348
    },
    {
      "epoch": 1.285552978515625e-06,
      "step": 1348,
      "training_step_time": 17.027977466583252
    },
    {
      "epoch": 1.2865066528320312e-06,
      "model_forward_time": 5.5977537631988525,
      "step": 1349
    },
    {
      "epoch": 1.2865066528320312e-06,
      "step": 1349,
      "training_step_time": 17.057480812072754
    },
    {
      "epoch": 1.2874603271484376e-06,
      "grad_norm": 1.936672329902649,
      "learning_rate": 1e-05,
      "loss": 0.083,
      "step": 1350
    },
    {
      "epoch": 1.2874603271484376e-06,
      "model_forward_time": 5.602775812149048,
      "step": 1350
    },
    {
      "dynamics_loss_avg": 0.06263908930122852,
      "epoch": 1.2874603271484376e-06,
      "step": 1350
    },
    {
      "action_loss_avg": 0.021128211123868822,
      "epoch": 1.2874603271484376e-06,
      "step": 1350
    },
    {
      "epoch": 1.2874603271484376e-06,
      "step": 1350,
      "training_step_time": 17.08193612098694
    },
    {
      "epoch": 1.2884140014648437e-06,
      "model_forward_time": 5.970695972442627,
      "step": 1351
    },
    {
      "epoch": 1.2884140014648437e-06,
      "step": 1351,
      "training_step_time": 17.44506049156189
    },
    {
      "epoch": 1.28936767578125e-06,
      "model_forward_time": 5.611573934555054,
      "step": 1352
    },
    {
      "epoch": 1.28936767578125e-06,
      "step": 1352,
      "training_step_time": 17.110758781433105
    },
    {
      "epoch": 1.2903213500976562e-06,
      "model_forward_time": 5.613971471786499,
      "step": 1353
    },
    {
      "epoch": 1.2903213500976562e-06,
      "step": 1353,
      "training_step_time": 17.129528045654297
    },
    {
      "epoch": 1.2912750244140626e-06,
      "model_forward_time": 5.611907482147217,
      "step": 1354
    },
    {
      "epoch": 1.2912750244140626e-06,
      "step": 1354,
      "training_step_time": 17.086577653884888
    },
    {
      "epoch": 1.2922286987304687e-06,
      "model_forward_time": 5.592209577560425,
      "step": 1355
    },
    {
      "epoch": 1.2922286987304687e-06,
      "step": 1355,
      "training_step_time": 17.028281450271606
    },
    {
      "epoch": 1.293182373046875e-06,
      "model_forward_time": 5.923288106918335,
      "step": 1356
    },
    {
      "epoch": 1.293182373046875e-06,
      "step": 1356,
      "training_step_time": 17.324342727661133
    },
    {
      "epoch": 1.2941360473632812e-06,
      "model_forward_time": 5.570666551589966,
      "step": 1357
    },
    {
      "epoch": 1.2941360473632812e-06,
      "step": 1357,
      "training_step_time": 16.96985626220703
    },
    {
      "epoch": 1.2950897216796876e-06,
      "model_forward_time": 5.578720569610596,
      "step": 1358
    },
    {
      "epoch": 1.2950897216796876e-06,
      "step": 1358,
      "training_step_time": 17.016783952713013
    },
    {
      "epoch": 1.2960433959960937e-06,
      "model_forward_time": 5.572477340698242,
      "step": 1359
    },
    {
      "epoch": 1.2960433959960937e-06,
      "step": 1359,
      "training_step_time": 17.00695562362671
    },
    {
      "epoch": 1.2969970703125e-06,
      "grad_norm": 1.3182930946350098,
      "learning_rate": 1e-05,
      "loss": 0.0986,
      "step": 1360
    },
    {
      "epoch": 1.2969970703125e-06,
      "model_forward_time": 5.588665723800659,
      "step": 1360
    },
    {
      "dynamics_loss_avg": 0.07432229369878769,
      "epoch": 1.2969970703125e-06,
      "step": 1360
    },
    {
      "action_loss_avg": 0.026580468844622375,
      "epoch": 1.2969970703125e-06,
      "step": 1360
    },
    {
      "epoch": 1.2969970703125e-06,
      "step": 1360,
      "training_step_time": 17.00604748725891
    },
    {
      "epoch": 1.2979507446289062e-06,
      "model_forward_time": 5.585018873214722,
      "step": 1361
    },
    {
      "epoch": 1.2979507446289062e-06,
      "step": 1361,
      "training_step_time": 17.05266308784485
    },
    {
      "epoch": 1.2989044189453126e-06,
      "model_forward_time": 5.959396839141846,
      "step": 1362
    },
    {
      "epoch": 1.2989044189453126e-06,
      "step": 1362,
      "training_step_time": 17.428521633148193
    },
    {
      "epoch": 1.2998580932617187e-06,
      "model_forward_time": 5.620180606842041,
      "step": 1363
    },
    {
      "epoch": 1.2998580932617187e-06,
      "step": 1363,
      "training_step_time": 17.114433526992798
    },
    {
      "epoch": 1.300811767578125e-06,
      "model_forward_time": 5.604895114898682,
      "step": 1364
    },
    {
      "epoch": 1.300811767578125e-06,
      "step": 1364,
      "training_step_time": 17.083498239517212
    },
    {
      "epoch": 1.3017654418945312e-06,
      "model_forward_time": 5.595007658004761,
      "step": 1365
    },
    {
      "epoch": 1.3017654418945312e-06,
      "step": 1365,
      "training_step_time": 17.036458015441895
    },
    {
      "epoch": 1.3027191162109376e-06,
      "model_forward_time": 5.587914943695068,
      "step": 1366
    },
    {
      "epoch": 1.3027191162109376e-06,
      "step": 1366,
      "training_step_time": 17.03228497505188
    },
    {
      "epoch": 1.3036727905273437e-06,
      "model_forward_time": 5.579970359802246,
      "step": 1367
    },
    {
      "epoch": 1.3036727905273437e-06,
      "step": 1367,
      "training_step_time": 17.338785409927368
    },
    {
      "epoch": 1.30462646484375e-06,
      "model_forward_time": 5.584892988204956,
      "step": 1368
    },
    {
      "epoch": 1.30462646484375e-06,
      "step": 1368,
      "training_step_time": 17.001693725585938
    },
    {
      "epoch": 1.3055801391601562e-06,
      "model_forward_time": 5.596776962280273,
      "step": 1369
    },
    {
      "epoch": 1.3055801391601562e-06,
      "step": 1369,
      "training_step_time": 17.07005548477173
    },
    {
      "epoch": 1.3065338134765626e-06,
      "grad_norm": 1.2192414999008179,
      "learning_rate": 1e-05,
      "loss": 0.0927,
      "step": 1370
    },
    {
      "epoch": 1.3065338134765626e-06,
      "model_forward_time": 5.618798732757568,
      "step": 1370
    },
    {
      "dynamics_loss_avg": 0.0673732165247202,
      "epoch": 1.3065338134765626e-06,
      "step": 1370
    },
    {
      "action_loss_avg": 0.0207814522087574,
      "epoch": 1.3065338134765626e-06,
      "step": 1370
    },
    {
      "epoch": 1.3065338134765626e-06,
      "step": 1370,
      "training_step_time": 17.121156215667725
    },
    {
      "epoch": 1.3074874877929687e-06,
      "model_forward_time": 5.621509313583374,
      "step": 1371
    },
    {
      "epoch": 1.3074874877929687e-06,
      "step": 1371,
      "training_step_time": 17.11077094078064
    },
    {
      "epoch": 1.308441162109375e-06,
      "model_forward_time": 5.611366033554077,
      "step": 1372
    },
    {
      "epoch": 1.308441162109375e-06,
      "step": 1372,
      "training_step_time": 17.09481120109558
    },
    {
      "epoch": 1.3093948364257812e-06,
      "model_forward_time": 5.956582546234131,
      "step": 1373
    },
    {
      "epoch": 1.3093948364257812e-06,
      "step": 1373,
      "training_step_time": 17.39718794822693
    },
    {
      "epoch": 1.3103485107421876e-06,
      "model_forward_time": 5.584827661514282,
      "step": 1374
    },
    {
      "epoch": 1.3103485107421876e-06,
      "step": 1374,
      "training_step_time": 17.027125597000122
    },
    {
      "epoch": 1.3113021850585938e-06,
      "model_forward_time": 5.587955951690674,
      "step": 1375
    },
    {
      "epoch": 1.3113021850585938e-06,
      "step": 1375,
      "training_step_time": 17.022194862365723
    },
    {
      "epoch": 1.312255859375e-06,
      "model_forward_time": 5.596673488616943,
      "step": 1376
    },
    {
      "epoch": 1.312255859375e-06,
      "step": 1376,
      "training_step_time": 17.0587477684021
    },
    {
      "epoch": 1.3132095336914063e-06,
      "model_forward_time": 5.614466667175293,
      "step": 1377
    },
    {
      "epoch": 1.3132095336914063e-06,
      "step": 1377,
      "training_step_time": 17.066532611846924
    },
    {
      "epoch": 1.3141632080078124e-06,
      "model_forward_time": 5.607226133346558,
      "step": 1378
    },
    {
      "epoch": 1.3141632080078124e-06,
      "step": 1378,
      "training_step_time": 17.420133590698242
    },
    {
      "epoch": 1.3151168823242188e-06,
      "model_forward_time": 5.628031253814697,
      "step": 1379
    },
    {
      "epoch": 1.3151168823242188e-06,
      "step": 1379,
      "training_step_time": 17.09561777114868
    },
    {
      "epoch": 1.316070556640625e-06,
      "grad_norm": 1.331565022468567,
      "learning_rate": 1e-05,
      "loss": 0.101,
      "step": 1380
    },
    {
      "epoch": 1.316070556640625e-06,
      "model_forward_time": 5.599383115768433,
      "step": 1380
    },
    {
      "dynamics_loss_avg": 0.07787715382874012,
      "epoch": 1.316070556640625e-06,
      "step": 1380
    },
    {
      "action_loss_avg": 0.02445230996236205,
      "epoch": 1.316070556640625e-06,
      "step": 1380
    },
    {
      "epoch": 1.316070556640625e-06,
      "step": 1380,
      "training_step_time": 17.069767475128174
    },
    {
      "epoch": 1.3170242309570313e-06,
      "model_forward_time": 5.609548091888428,
      "step": 1381
    },
    {
      "epoch": 1.3170242309570313e-06,
      "step": 1381,
      "training_step_time": 17.051783323287964
    },
    {
      "epoch": 1.3179779052734374e-06,
      "model_forward_time": 5.59518027305603,
      "step": 1382
    },
    {
      "epoch": 1.3179779052734374e-06,
      "step": 1382,
      "training_step_time": 17.04348921775818
    },
    {
      "epoch": 1.3189315795898438e-06,
      "model_forward_time": 5.601776599884033,
      "step": 1383
    },
    {
      "epoch": 1.3189315795898438e-06,
      "step": 1383,
      "training_step_time": 17.065134525299072
    },
    {
      "epoch": 1.31988525390625e-06,
      "model_forward_time": 5.995264768600464,
      "step": 1384
    },
    {
      "epoch": 1.31988525390625e-06,
      "step": 1384,
      "training_step_time": 17.44534420967102
    },
    {
      "epoch": 1.3208389282226563e-06,
      "model_forward_time": 5.587446689605713,
      "step": 1385
    },
    {
      "epoch": 1.3208389282226563e-06,
      "step": 1385,
      "training_step_time": 17.031500816345215
    },
    {
      "epoch": 1.3217926025390624e-06,
      "model_forward_time": 5.584197759628296,
      "step": 1386
    },
    {
      "epoch": 1.3217926025390624e-06,
      "step": 1386,
      "training_step_time": 17.052030324935913
    },
    {
      "epoch": 1.3227462768554688e-06,
      "model_forward_time": 5.617108345031738,
      "step": 1387
    },
    {
      "epoch": 1.3227462768554688e-06,
      "step": 1387,
      "training_step_time": 17.11223840713501
    },
    {
      "epoch": 1.323699951171875e-06,
      "model_forward_time": 5.6193366050720215,
      "step": 1388
    },
    {
      "epoch": 1.323699951171875e-06,
      "step": 1388,
      "training_step_time": 17.11294722557068
    },
    {
      "epoch": 1.3246536254882813e-06,
      "model_forward_time": 5.960928440093994,
      "step": 1389
    },
    {
      "epoch": 1.3246536254882813e-06,
      "step": 1389,
      "training_step_time": 17.445605278015137
    },
    {
      "epoch": 1.3256072998046874e-06,
      "grad_norm": 2.0890729427337646,
      "learning_rate": 1e-05,
      "loss": 0.0841,
      "step": 1390
    },
    {
      "epoch": 1.3256072998046874e-06,
      "model_forward_time": 5.620229721069336,
      "step": 1390
    },
    {
      "dynamics_loss_avg": 0.06266673095524311,
      "epoch": 1.3256072998046874e-06,
      "step": 1390
    },
    {
      "action_loss_avg": 0.018780673481523992,
      "epoch": 1.3256072998046874e-06,
      "step": 1390
    },
    {
      "epoch": 1.3256072998046874e-06,
      "step": 1390,
      "training_step_time": 17.108155488967896
    },
    {
      "epoch": 1.3265609741210938e-06,
      "model_forward_time": 5.607930660247803,
      "step": 1391
    },
    {
      "epoch": 1.3265609741210938e-06,
      "step": 1391,
      "training_step_time": 17.109017372131348
    },
    {
      "epoch": 1.3275146484375e-06,
      "model_forward_time": 5.612914800643921,
      "step": 1392
    },
    {
      "epoch": 1.3275146484375e-06,
      "step": 1392,
      "training_step_time": 17.11454439163208
    },
    {
      "epoch": 1.3284683227539063e-06,
      "model_forward_time": 5.6144514083862305,
      "step": 1393
    },
    {
      "epoch": 1.3284683227539063e-06,
      "step": 1393,
      "training_step_time": 17.10474920272827
    },
    {
      "epoch": 1.3294219970703124e-06,
      "model_forward_time": 5.607189416885376,
      "step": 1394
    },
    {
      "epoch": 1.3294219970703124e-06,
      "step": 1394,
      "training_step_time": 17.085654735565186
    },
    {
      "epoch": 1.3303756713867188e-06,
      "model_forward_time": 5.971123695373535,
      "step": 1395
    },
    {
      "epoch": 1.3303756713867188e-06,
      "step": 1395,
      "training_step_time": 17.427706718444824
    },
    {
      "epoch": 1.331329345703125e-06,
      "model_forward_time": 5.59528374671936,
      "step": 1396
    },
    {
      "epoch": 1.331329345703125e-06,
      "step": 1396,
      "training_step_time": 17.034642934799194
    },
    {
      "epoch": 1.3322830200195313e-06,
      "model_forward_time": 5.593500852584839,
      "step": 1397
    },
    {
      "epoch": 1.3322830200195313e-06,
      "step": 1397,
      "training_step_time": 17.030853748321533
    },
    {
      "epoch": 1.3332366943359374e-06,
      "model_forward_time": 5.593181610107422,
      "step": 1398
    },
    {
      "epoch": 1.3332366943359374e-06,
      "step": 1398,
      "training_step_time": 17.04723882675171
    },
    {
      "epoch": 1.3341903686523438e-06,
      "model_forward_time": 5.592043399810791,
      "step": 1399
    },
    {
      "epoch": 1.3341903686523438e-06,
      "step": 1399,
      "training_step_time": 17.02717351913452
    },
    {
      "epoch": 1.33514404296875e-06,
      "grad_norm": 0.9230601191520691,
      "learning_rate": 1e-05,
      "loss": 0.0788,
      "step": 1400
    },
    {
      "epoch": 1.33514404296875e-06,
      "model_forward_time": 5.592849493026733,
      "step": 1400
    },
    {
      "dynamics_loss_avg": 0.06511453874409198,
      "epoch": 1.33514404296875e-06,
      "step": 1400
    },
    {
      "action_loss_avg": 0.014342654007486999,
      "epoch": 1.33514404296875e-06,
      "step": 1400
    },
    {
      "epoch": 1.33514404296875e-06,
      "step": 1400,
      "training_step_time": 17.37626361846924
    },
    {
      "epoch": 1.3360977172851563e-06,
      "model_forward_time": 5.603891134262085,
      "step": 1401
    },
    {
      "epoch": 1.3360977172851563e-06,
      "step": 1401,
      "training_step_time": 17.078990697860718
    },
    {
      "epoch": 1.3370513916015624e-06,
      "model_forward_time": 5.612358093261719,
      "step": 1402
    },
    {
      "epoch": 1.3370513916015624e-06,
      "step": 1402,
      "training_step_time": 17.099202394485474
    },
    {
      "epoch": 1.3380050659179688e-06,
      "model_forward_time": 5.604381799697876,
      "step": 1403
    },
    {
      "epoch": 1.3380050659179688e-06,
      "step": 1403,
      "training_step_time": 17.0783908367157
    },
    {
      "epoch": 1.338958740234375e-06,
      "model_forward_time": 5.604944229125977,
      "step": 1404
    },
    {
      "epoch": 1.338958740234375e-06,
      "step": 1404,
      "training_step_time": 17.068941831588745
    },
    {
      "epoch": 1.3399124145507813e-06,
      "model_forward_time": 5.608835458755493,
      "step": 1405
    },
    {
      "epoch": 1.3399124145507813e-06,
      "step": 1405,
      "training_step_time": 17.0786349773407
    },
    {
      "epoch": 1.3408660888671874e-06,
      "model_forward_time": 5.9591028690338135,
      "step": 1406
    },
    {
      "epoch": 1.3408660888671874e-06,
      "step": 1406,
      "training_step_time": 17.425175666809082
    },
    {
      "epoch": 1.3418197631835938e-06,
      "model_forward_time": 5.607861042022705,
      "step": 1407
    },
    {
      "epoch": 1.3418197631835938e-06,
      "step": 1407,
      "training_step_time": 17.08397889137268
    },
    {
      "epoch": 1.3427734375e-06,
      "model_forward_time": 5.614609956741333,
      "step": 1408
    },
    {
      "epoch": 1.3427734375e-06,
      "step": 1408,
      "training_step_time": 17.083839654922485
    },
    {
      "epoch": 1.3437271118164063e-06,
      "model_forward_time": 5.600635766983032,
      "step": 1409
    },
    {
      "epoch": 1.3437271118164063e-06,
      "step": 1409,
      "training_step_time": 17.067955493927002
    },
    {
      "epoch": 1.3446807861328124e-06,
      "grad_norm": 1.4488780498504639,
      "learning_rate": 1e-05,
      "loss": 0.0832,
      "step": 1410
    },
    {
      "epoch": 1.3446807861328124e-06,
      "model_forward_time": 5.596512317657471,
      "step": 1410
    },
    {
      "dynamics_loss_avg": 0.059090980514884,
      "epoch": 1.3446807861328124e-06,
      "step": 1410
    },
    {
      "action_loss_avg": 0.02545750653371215,
      "epoch": 1.3446807861328124e-06,
      "step": 1410
    },
    {
      "epoch": 1.3446807861328124e-06,
      "step": 1410,
      "training_step_time": 17.05163598060608
    },
    {
      "epoch": 1.3456344604492188e-06,
      "model_forward_time": 5.594351530075073,
      "step": 1411
    },
    {
      "epoch": 1.3456344604492188e-06,
      "step": 1411,
      "training_step_time": 17.37489652633667
    },
    {
      "epoch": 1.346588134765625e-06,
      "model_forward_time": 5.575741529464722,
      "step": 1412
    },
    {
      "epoch": 1.346588134765625e-06,
      "step": 1412,
      "training_step_time": 17.02796506881714
    },
    {
      "epoch": 1.3475418090820313e-06,
      "model_forward_time": 5.5825889110565186,
      "step": 1413
    },
    {
      "epoch": 1.3475418090820313e-06,
      "step": 1413,
      "training_step_time": 17.045268774032593
    },
    {
      "epoch": 1.3484954833984374e-06,
      "model_forward_time": 5.610051870346069,
      "step": 1414
    },
    {
      "epoch": 1.3484954833984374e-06,
      "step": 1414,
      "training_step_time": 17.11553955078125
    },
    {
      "epoch": 1.3494491577148438e-06,
      "model_forward_time": 5.624273777008057,
      "step": 1415
    },
    {
      "epoch": 1.3494491577148438e-06,
      "step": 1415,
      "training_step_time": 17.14431595802307
    },
    {
      "epoch": 1.35040283203125e-06,
      "model_forward_time": 5.617396354675293,
      "step": 1416
    },
    {
      "epoch": 1.35040283203125e-06,
      "step": 1416,
      "training_step_time": 17.10088849067688
    },
    {
      "epoch": 1.3513565063476563e-06,
      "model_forward_time": 5.997320175170898,
      "step": 1417
    },
    {
      "epoch": 1.3513565063476563e-06,
      "step": 1417,
      "training_step_time": 17.46894335746765
    },
    {
      "epoch": 1.3523101806640624e-06,
      "model_forward_time": 5.602589845657349,
      "step": 1418
    },
    {
      "epoch": 1.3523101806640624e-06,
      "step": 1418,
      "training_step_time": 17.050097942352295
    },
    {
      "epoch": 1.3532638549804688e-06,
      "model_forward_time": 5.584827899932861,
      "step": 1419
    },
    {
      "epoch": 1.3532638549804688e-06,
      "step": 1419,
      "training_step_time": 17.02630114555359
    },
    {
      "epoch": 1.354217529296875e-06,
      "grad_norm": 0.8215503692626953,
      "learning_rate": 1e-05,
      "loss": 0.0831,
      "step": 1420
    },
    {
      "epoch": 1.354217529296875e-06,
      "model_forward_time": 5.5828163623809814,
      "step": 1420
    },
    {
      "dynamics_loss_avg": 0.06079891212284565,
      "epoch": 1.354217529296875e-06,
      "step": 1420
    },
    {
      "action_loss_avg": 0.022232536505907775,
      "epoch": 1.354217529296875e-06,
      "step": 1420
    },
    {
      "epoch": 1.354217529296875e-06,
      "step": 1420,
      "training_step_time": 17.019678115844727
    },
    {
      "epoch": 1.3551712036132813e-06,
      "model_forward_time": 5.591458797454834,
      "step": 1421
    },
    {
      "epoch": 1.3551712036132813e-06,
      "step": 1421,
      "training_step_time": 17.019086360931396
    },
    {
      "epoch": 1.3561248779296874e-06,
      "model_forward_time": 5.955806732177734,
      "step": 1422
    },
    {
      "epoch": 1.3561248779296874e-06,
      "step": 1422,
      "training_step_time": 17.404240608215332
    },
    {
      "epoch": 1.3570785522460938e-06,
      "model_forward_time": 5.598576545715332,
      "step": 1423
    },
    {
      "epoch": 1.3570785522460938e-06,
      "step": 1423,
      "training_step_time": 17.070761680603027
    },
    {
      "epoch": 1.3580322265625e-06,
      "model_forward_time": 5.6079559326171875,
      "step": 1424
    },
    {
      "epoch": 1.3580322265625e-06,
      "step": 1424,
      "training_step_time": 17.078994035720825
    },
    {
      "epoch": 1.3589859008789063e-06,
      "model_forward_time": 5.607294797897339,
      "step": 1425
    },
    {
      "epoch": 1.3589859008789063e-06,
      "step": 1425,
      "training_step_time": 17.082668781280518
    },
    {
      "epoch": 1.3599395751953124e-06,
      "model_forward_time": 5.605772972106934,
      "step": 1426
    },
    {
      "epoch": 1.3599395751953124e-06,
      "step": 1426,
      "training_step_time": 17.092620611190796
    },
    {
      "epoch": 1.3608932495117188e-06,
      "model_forward_time": 5.604935169219971,
      "step": 1427
    },
    {
      "epoch": 1.3608932495117188e-06,
      "step": 1427,
      "training_step_time": 17.05579972267151
    },
    {
      "epoch": 1.361846923828125e-06,
      "model_forward_time": 5.959484577178955,
      "step": 1428
    },
    {
      "epoch": 1.361846923828125e-06,
      "step": 1428,
      "training_step_time": 17.4293212890625
    },
    {
      "epoch": 1.3628005981445313e-06,
      "model_forward_time": 5.601660490036011,
      "step": 1429
    },
    {
      "epoch": 1.3628005981445313e-06,
      "step": 1429,
      "training_step_time": 17.072007417678833
    },
    {
      "epoch": 1.3637542724609374e-06,
      "grad_norm": 1.3849481344223022,
      "learning_rate": 1e-05,
      "loss": 0.0837,
      "step": 1430
    },
    {
      "epoch": 1.3637542724609374e-06,
      "model_forward_time": 5.60108494758606,
      "step": 1430
    },
    {
      "dynamics_loss_avg": 0.06832680944353342,
      "epoch": 1.3637542724609374e-06,
      "step": 1430
    },
    {
      "action_loss_avg": 0.018145068688318132,
      "epoch": 1.3637542724609374e-06,
      "step": 1430
    },
    {
      "epoch": 1.3637542724609374e-06,
      "step": 1430,
      "training_step_time": 17.067561864852905
    },
    {
      "epoch": 1.3647079467773438e-06,
      "model_forward_time": 5.600274562835693,
      "step": 1431
    },
    {
      "epoch": 1.3647079467773438e-06,
      "step": 1431,
      "training_step_time": 17.052242755889893
    },
    {
      "epoch": 1.36566162109375e-06,
      "model_forward_time": 5.600139617919922,
      "step": 1432
    },
    {
      "epoch": 1.36566162109375e-06,
      "step": 1432,
      "training_step_time": 17.09391212463379
    },
    {
      "epoch": 1.3666152954101563e-06,
      "model_forward_time": 5.624066114425659,
      "step": 1433
    },
    {
      "epoch": 1.3666152954101563e-06,
      "step": 1433,
      "training_step_time": 17.498890161514282
    },
    {
      "epoch": 1.3675689697265624e-06,
      "model_forward_time": 5.629641532897949,
      "step": 1434
    },
    {
      "epoch": 1.3675689697265624e-06,
      "step": 1434,
      "training_step_time": 17.141784191131592
    },
    {
      "epoch": 1.3685226440429688e-06,
      "model_forward_time": 5.608900308609009,
      "step": 1435
    },
    {
      "epoch": 1.3685226440429688e-06,
      "step": 1435,
      "training_step_time": 17.05577325820923
    },
    {
      "epoch": 1.369476318359375e-06,
      "model_forward_time": 5.59494948387146,
      "step": 1436
    },
    {
      "epoch": 1.369476318359375e-06,
      "step": 1436,
      "training_step_time": 17.021419763565063
    },
    {
      "epoch": 1.3704299926757813e-06,
      "model_forward_time": 5.582002401351929,
      "step": 1437
    },
    {
      "epoch": 1.3704299926757813e-06,
      "step": 1437,
      "training_step_time": 16.985819816589355
    },
    {
      "epoch": 1.3713836669921874e-06,
      "model_forward_time": 5.5690531730651855,
      "step": 1438
    },
    {
      "epoch": 1.3713836669921874e-06,
      "step": 1438,
      "training_step_time": 17.002097368240356
    },
    {
      "epoch": 1.3723373413085938e-06,
      "model_forward_time": 5.929304838180542,
      "step": 1439
    },
    {
      "epoch": 1.3723373413085938e-06,
      "step": 1439,
      "training_step_time": 17.362977981567383
    },
    {
      "epoch": 1.373291015625e-06,
      "grad_norm": 0.7230544686317444,
      "learning_rate": 1e-05,
      "loss": 0.0993,
      "step": 1440
    },
    {
      "epoch": 1.373291015625e-06,
      "model_forward_time": 5.582334041595459,
      "step": 1440
    },
    {
      "dynamics_loss_avg": 0.06554932780563831,
      "epoch": 1.373291015625e-06,
      "step": 1440
    },
    {
      "action_loss_avg": 0.029920877236872912,
      "epoch": 1.373291015625e-06,
      "step": 1440
    },
    {
      "epoch": 1.373291015625e-06,
      "step": 1440,
      "training_step_time": 17.001934051513672
    },
    {
      "epoch": 1.3742446899414063e-06,
      "model_forward_time": 5.589483976364136,
      "step": 1441
    },
    {
      "epoch": 1.3742446899414063e-06,
      "step": 1441,
      "training_step_time": 17.004218339920044
    },
    {
      "epoch": 1.3751983642578125e-06,
      "model_forward_time": 5.584879636764526,
      "step": 1442
    },
    {
      "epoch": 1.3751983642578125e-06,
      "step": 1442,
      "training_step_time": 17.031375885009766
    },
    {
      "epoch": 1.3761520385742188e-06,
      "model_forward_time": 5.58520770072937,
      "step": 1443
    },
    {
      "epoch": 1.3761520385742188e-06,
      "step": 1443,
      "training_step_time": 17.058720350265503
    },
    {
      "epoch": 1.377105712890625e-06,
      "model_forward_time": 5.609721422195435,
      "step": 1444
    },
    {
      "epoch": 1.377105712890625e-06,
      "step": 1444,
      "training_step_time": 17.47692608833313
    },
    {
      "epoch": 1.3780593872070313e-06,
      "model_forward_time": 5.617760896682739,
      "step": 1445
    },
    {
      "epoch": 1.3780593872070313e-06,
      "step": 1445,
      "training_step_time": 17.105366230010986
    },
    {
      "epoch": 1.3790130615234375e-06,
      "model_forward_time": 5.608333587646484,
      "step": 1446
    },
    {
      "epoch": 1.3790130615234375e-06,
      "step": 1446,
      "training_step_time": 17.1298086643219
    },
    {
      "epoch": 1.3799667358398438e-06,
      "model_forward_time": 5.619919776916504,
      "step": 1447
    },
    {
      "epoch": 1.3799667358398438e-06,
      "step": 1447,
      "training_step_time": 17.112979650497437
    },
    {
      "epoch": 1.38092041015625e-06,
      "model_forward_time": 5.601171493530273,
      "step": 1448
    },
    {
      "epoch": 1.38092041015625e-06,
      "step": 1448,
      "training_step_time": 17.097104787826538
    },
    {
      "epoch": 1.3818740844726563e-06,
      "model_forward_time": 5.611514568328857,
      "step": 1449
    },
    {
      "epoch": 1.3818740844726563e-06,
      "step": 1449,
      "training_step_time": 17.10755681991577
    },
    {
      "epoch": 1.3828277587890625e-06,
      "grad_norm": 1.2553586959838867,
      "learning_rate": 1e-05,
      "loss": 0.072,
      "step": 1450
    },
    {
      "epoch": 1.3828277587890625e-06,
      "model_forward_time": 5.983250379562378,
      "step": 1450
    },
    {
      "dynamics_loss_avg": 0.05635244231671095,
      "epoch": 1.3828277587890625e-06,
      "step": 1450
    },
    {
      "action_loss_avg": 0.020762948505580425,
      "epoch": 1.3828277587890625e-06,
      "step": 1450
    },
    {
      "epoch": 1.3828277587890625e-06,
      "step": 1450,
      "training_step_time": 17.46004605293274
    },
    {
      "epoch": 1.3837814331054688e-06,
      "model_forward_time": 5.586800813674927,
      "step": 1451
    },
    {
      "epoch": 1.3837814331054688e-06,
      "step": 1451,
      "training_step_time": 17.045877695083618
    },
    {
      "epoch": 1.384735107421875e-06,
      "model_forward_time": 5.594606161117554,
      "step": 1452
    },
    {
      "epoch": 1.384735107421875e-06,
      "step": 1452,
      "training_step_time": 17.050261974334717
    },
    {
      "epoch": 1.3856887817382813e-06,
      "model_forward_time": 5.6044840812683105,
      "step": 1453
    },
    {
      "epoch": 1.3856887817382813e-06,
      "step": 1453,
      "training_step_time": 17.072474479675293
    },
    {
      "epoch": 1.3866424560546875e-06,
      "model_forward_time": 5.586712598800659,
      "step": 1454
    },
    {
      "epoch": 1.3866424560546875e-06,
      "step": 1454,
      "training_step_time": 17.007039070129395
    },
    {
      "epoch": 1.3875961303710938e-06,
      "model_forward_time": 5.950893402099609,
      "step": 1455
    },
    {
      "epoch": 1.3875961303710938e-06,
      "step": 1455,
      "training_step_time": 17.39228582382202
    },
    {
      "epoch": 1.3885498046875e-06,
      "model_forward_time": 5.607651233673096,
      "step": 1456
    },
    {
      "epoch": 1.3885498046875e-06,
      "step": 1456,
      "training_step_time": 17.06380867958069
    },
    {
      "epoch": 1.3895034790039063e-06,
      "model_forward_time": 5.593355417251587,
      "step": 1457
    },
    {
      "epoch": 1.3895034790039063e-06,
      "step": 1457,
      "training_step_time": 17.05288529396057
    },
    {
      "epoch": 1.3904571533203125e-06,
      "model_forward_time": 5.6038432121276855,
      "step": 1458
    },
    {
      "epoch": 1.3904571533203125e-06,
      "step": 1458,
      "training_step_time": 17.058837175369263
    },
    {
      "epoch": 1.3914108276367188e-06,
      "model_forward_time": 5.61522364616394,
      "step": 1459
    },
    {
      "epoch": 1.3914108276367188e-06,
      "step": 1459,
      "training_step_time": 17.099842309951782
    },
    {
      "epoch": 1.392364501953125e-06,
      "grad_norm": 0.6774768233299255,
      "learning_rate": 1e-05,
      "loss": 0.0703,
      "step": 1460
    },
    {
      "epoch": 1.392364501953125e-06,
      "model_forward_time": 5.604983568191528,
      "step": 1460
    },
    {
      "dynamics_loss_avg": 0.048563769459724425,
      "epoch": 1.392364501953125e-06,
      "step": 1460
    },
    {
      "action_loss_avg": 0.015810948237776755,
      "epoch": 1.392364501953125e-06,
      "step": 1460
    },
    {
      "epoch": 1.392364501953125e-06,
      "step": 1460,
      "training_step_time": 17.094680547714233
    },
    {
      "epoch": 1.3933181762695313e-06,
      "model_forward_time": 5.985588550567627,
      "step": 1461
    },
    {
      "epoch": 1.3933181762695313e-06,
      "step": 1461,
      "training_step_time": 17.45985507965088
    },
    {
      "epoch": 1.3942718505859375e-06,
      "model_forward_time": 5.605888366699219,
      "step": 1462
    },
    {
      "epoch": 1.3942718505859375e-06,
      "step": 1462,
      "training_step_time": 17.084067344665527
    },
    {
      "epoch": 1.3952255249023438e-06,
      "model_forward_time": 5.613280773162842,
      "step": 1463
    },
    {
      "epoch": 1.3952255249023438e-06,
      "step": 1463,
      "training_step_time": 17.083303689956665
    },
    {
      "epoch": 1.39617919921875e-06,
      "model_forward_time": 5.597893238067627,
      "step": 1464
    },
    {
      "epoch": 1.39617919921875e-06,
      "step": 1464,
      "training_step_time": 17.047215461730957
    },
    {
      "epoch": 1.3971328735351563e-06,
      "model_forward_time": 5.590430498123169,
      "step": 1465
    },
    {
      "epoch": 1.3971328735351563e-06,
      "step": 1465,
      "training_step_time": 17.028911113739014
    },
    {
      "epoch": 1.3980865478515625e-06,
      "model_forward_time": 5.595033645629883,
      "step": 1466
    },
    {
      "epoch": 1.3980865478515625e-06,
      "step": 1466,
      "training_step_time": 17.435356616973877
    },
    {
      "epoch": 1.3990402221679688e-06,
      "model_forward_time": 5.609270334243774,
      "step": 1467
    },
    {
      "epoch": 1.3990402221679688e-06,
      "step": 1467,
      "training_step_time": 17.120468378067017
    },
    {
      "epoch": 1.399993896484375e-06,
      "model_forward_time": 5.624797344207764,
      "step": 1468
    },
    {
      "epoch": 1.399993896484375e-06,
      "step": 1468,
      "training_step_time": 17.15261697769165
    },
    {
      "epoch": 1.4009475708007813e-06,
      "model_forward_time": 5.614286661148071,
      "step": 1469
    },
    {
      "epoch": 1.4009475708007813e-06,
      "step": 1469,
      "training_step_time": 17.10477042198181
    },
    {
      "epoch": 1.4019012451171875e-06,
      "grad_norm": 1.563166856765747,
      "learning_rate": 1e-05,
      "loss": 0.0902,
      "step": 1470
    },
    {
      "epoch": 1.4019012451171875e-06,
      "model_forward_time": 5.609995126724243,
      "step": 1470
    },
    {
      "dynamics_loss_avg": 0.07329537197947503,
      "epoch": 1.4019012451171875e-06,
      "step": 1470
    },
    {
      "action_loss_avg": 0.023343078326433897,
      "epoch": 1.4019012451171875e-06,
      "step": 1470
    },
    {
      "epoch": 1.4019012451171875e-06,
      "step": 1470,
      "training_step_time": 17.084152698516846
    },
    {
      "epoch": 1.4028549194335938e-06,
      "model_forward_time": 5.59756326675415,
      "step": 1471
    },
    {
      "epoch": 1.4028549194335938e-06,
      "step": 1471,
      "training_step_time": 17.07951283454895
    },
    {
      "epoch": 1.40380859375e-06,
      "model_forward_time": 5.974167108535767,
      "step": 1472
    },
    {
      "epoch": 1.40380859375e-06,
      "step": 1472,
      "training_step_time": 17.473381519317627
    },
    {
      "epoch": 1.4047622680664063e-06,
      "model_forward_time": 5.615321397781372,
      "step": 1473
    },
    {
      "epoch": 1.4047622680664063e-06,
      "step": 1473,
      "training_step_time": 17.092750549316406
    },
    {
      "epoch": 1.4057159423828125e-06,
      "model_forward_time": 5.616360664367676,
      "step": 1474
    },
    {
      "epoch": 1.4057159423828125e-06,
      "step": 1474,
      "training_step_time": 17.11423349380493
    },
    {
      "epoch": 1.4066696166992188e-06,
      "model_forward_time": 5.613865375518799,
      "step": 1475
    },
    {
      "epoch": 1.4066696166992188e-06,
      "step": 1475,
      "training_step_time": 17.114216566085815
    },
    {
      "epoch": 1.407623291015625e-06,
      "model_forward_time": 5.609759092330933,
      "step": 1476
    },
    {
      "epoch": 1.407623291015625e-06,
      "step": 1476,
      "training_step_time": 17.095430374145508
    },
    {
      "epoch": 1.4085769653320313e-06,
      "model_forward_time": 5.612509727478027,
      "step": 1477
    },
    {
      "epoch": 1.4085769653320313e-06,
      "step": 1477,
      "training_step_time": 17.43662977218628
    },
    {
      "epoch": 1.4095306396484375e-06,
      "model_forward_time": 5.613327503204346,
      "step": 1478
    },
    {
      "epoch": 1.4095306396484375e-06,
      "step": 1478,
      "training_step_time": 17.084651708602905
    },
    {
      "epoch": 1.4104843139648438e-06,
      "model_forward_time": 5.602799654006958,
      "step": 1479
    },
    {
      "epoch": 1.4104843139648438e-06,
      "step": 1479,
      "training_step_time": 17.065497636795044
    },
    {
      "epoch": 1.41143798828125e-06,
      "grad_norm": 1.4699684381484985,
      "learning_rate": 1e-05,
      "loss": 0.0878,
      "step": 1480
    },
    {
      "epoch": 1.41143798828125e-06,
      "model_forward_time": 5.5875022411346436,
      "step": 1480
    },
    {
      "dynamics_loss_avg": 0.060273229889571664,
      "epoch": 1.41143798828125e-06,
      "step": 1480
    },
    {
      "action_loss_avg": 0.01991350520402193,
      "epoch": 1.41143798828125e-06,
      "step": 1480
    },
    {
      "epoch": 1.41143798828125e-06,
      "step": 1480,
      "training_step_time": 17.023417472839355
    },
    {
      "epoch": 1.4123916625976563e-06,
      "model_forward_time": 5.576023817062378,
      "step": 1481
    },
    {
      "epoch": 1.4123916625976563e-06,
      "step": 1481,
      "training_step_time": 17.01268434524536
    },
    {
      "epoch": 1.4133453369140625e-06,
      "model_forward_time": 5.590420961380005,
      "step": 1482
    },
    {
      "epoch": 1.4133453369140625e-06,
      "step": 1482,
      "training_step_time": 17.034966468811035
    },
    {
      "epoch": 1.4142990112304688e-06,
      "model_forward_time": 6.004347801208496,
      "step": 1483
    },
    {
      "epoch": 1.4142990112304688e-06,
      "step": 1483,
      "training_step_time": 17.457342624664307
    },
    {
      "epoch": 1.415252685546875e-06,
      "model_forward_time": 5.601778984069824,
      "step": 1484
    },
    {
      "epoch": 1.415252685546875e-06,
      "step": 1484,
      "training_step_time": 17.069870471954346
    },
    {
      "epoch": 1.4162063598632813e-06,
      "model_forward_time": 5.6094982624053955,
      "step": 1485
    },
    {
      "epoch": 1.4162063598632813e-06,
      "step": 1485,
      "training_step_time": 17.07499098777771
    },
    {
      "epoch": 1.4171600341796875e-06,
      "model_forward_time": 5.600210905075073,
      "step": 1486
    },
    {
      "epoch": 1.4171600341796875e-06,
      "step": 1486,
      "training_step_time": 17.056636810302734
    },
    {
      "epoch": 1.4181137084960938e-06,
      "model_forward_time": 5.594789743423462,
      "step": 1487
    },
    {
      "epoch": 1.4181137084960938e-06,
      "step": 1487,
      "training_step_time": 17.06055736541748
    },
    {
      "epoch": 1.4190673828125e-06,
      "model_forward_time": 5.954984188079834,
      "step": 1488
    },
    {
      "epoch": 1.4190673828125e-06,
      "step": 1488,
      "training_step_time": 17.4295597076416
    },
    {
      "epoch": 1.4200210571289063e-06,
      "model_forward_time": 5.622369289398193,
      "step": 1489
    },
    {
      "epoch": 1.4200210571289063e-06,
      "step": 1489,
      "training_step_time": 17.128936290740967
    },
    {
      "epoch": 1.4209747314453125e-06,
      "grad_norm": 0.42352205514907837,
      "learning_rate": 1e-05,
      "loss": 0.0858,
      "step": 1490
    },
    {
      "epoch": 1.4209747314453125e-06,
      "model_forward_time": 5.631174802780151,
      "step": 1490
    },
    {
      "dynamics_loss_avg": 0.0668613962829113,
      "epoch": 1.4209747314453125e-06,
      "step": 1490
    },
    {
      "action_loss_avg": 0.022197570372372865,
      "epoch": 1.4209747314453125e-06,
      "step": 1490
    },
    {
      "epoch": 1.4209747314453125e-06,
      "step": 1490,
      "training_step_time": 17.155978202819824
    },
    {
      "epoch": 1.4219284057617188e-06,
      "model_forward_time": 5.632737874984741,
      "step": 1491
    },
    {
      "epoch": 1.4219284057617188e-06,
      "step": 1491,
      "training_step_time": 17.134571075439453
    },
    {
      "epoch": 1.422882080078125e-06,
      "model_forward_time": 5.604653835296631,
      "step": 1492
    },
    {
      "epoch": 1.422882080078125e-06,
      "step": 1492,
      "training_step_time": 17.04797601699829
    },
    {
      "epoch": 1.4238357543945313e-06,
      "model_forward_time": 5.578922748565674,
      "step": 1493
    },
    {
      "epoch": 1.4238357543945313e-06,
      "step": 1493,
      "training_step_time": 16.987083911895752
    },
    {
      "epoch": 1.4247894287109375e-06,
      "model_forward_time": 5.944008111953735,
      "step": 1494
    },
    {
      "epoch": 1.4247894287109375e-06,
      "step": 1494,
      "training_step_time": 17.36026692390442
    },
    {
      "epoch": 1.4257431030273439e-06,
      "model_forward_time": 5.568824052810669,
      "step": 1495
    },
    {
      "epoch": 1.4257431030273439e-06,
      "step": 1495,
      "training_step_time": 17.001718997955322
    },
    {
      "epoch": 1.42669677734375e-06,
      "model_forward_time": 5.569071054458618,
      "step": 1496
    },
    {
      "epoch": 1.42669677734375e-06,
      "step": 1496,
      "training_step_time": 16.988598108291626
    },
    {
      "epoch": 1.4276504516601564e-06,
      "model_forward_time": 5.58154559135437,
      "step": 1497
    },
    {
      "epoch": 1.4276504516601564e-06,
      "step": 1497,
      "training_step_time": 17.00602412223816
    },
    {
      "epoch": 1.4286041259765625e-06,
      "model_forward_time": 5.589333772659302,
      "step": 1498
    },
    {
      "epoch": 1.4286041259765625e-06,
      "step": 1498,
      "training_step_time": 17.047990322113037
    },
    {
      "epoch": 1.4295578002929689e-06,
      "model_forward_time": 5.609955072402954,
      "step": 1499
    },
    {
      "epoch": 1.4295578002929689e-06,
      "step": 1499,
      "training_step_time": 17.442888736724854
    },
    {
      "epoch": 1.430511474609375e-06,
      "grad_norm": 0.4561484754085541,
      "learning_rate": 1e-05,
      "loss": 0.0673,
      "step": 1500
    },
    {
      "epoch": 1.430511474609375e-06,
      "model_forward_time": 5.612767934799194,
      "step": 1500
    },
    {
      "dynamics_loss_avg": 0.04810604322701693,
      "epoch": 1.430511474609375e-06,
      "step": 1500
    },
    {
      "action_loss_avg": 0.01877118954434991,
      "epoch": 1.430511474609375e-06,
      "step": 1500
    },
    {
      "epoch": 1.430511474609375e-06,
      "step": 1500,
      "training_step_time": 17.107952117919922
    },
    {
      "epoch": 1.4314651489257811e-06,
      "model_forward_time": 5.612745761871338,
      "step": 1501
    },
    {
      "epoch": 1.4314651489257811e-06,
      "step": 1501,
      "training_step_time": 17.101266622543335
    },
    {
      "epoch": 1.4324188232421875e-06,
      "model_forward_time": 5.606881618499756,
      "step": 1502
    },
    {
      "epoch": 1.4324188232421875e-06,
      "step": 1502,
      "training_step_time": 17.097015619277954
    },
    {
      "epoch": 1.4333724975585936e-06,
      "model_forward_time": 5.606057405471802,
      "step": 1503
    },
    {
      "epoch": 1.4333724975585936e-06,
      "step": 1503,
      "training_step_time": 17.085301399230957
    },
    {
      "epoch": 1.434326171875e-06,
      "model_forward_time": 5.604034900665283,
      "step": 1504
    },
    {
      "epoch": 1.434326171875e-06,
      "step": 1504,
      "training_step_time": 17.079487562179565
    },
    {
      "epoch": 1.4352798461914061e-06,
      "model_forward_time": 5.975969076156616,
      "step": 1505
    },
    {
      "epoch": 1.4352798461914061e-06,
      "step": 1505,
      "training_step_time": 17.450759649276733
    },
    {
      "epoch": 1.4362335205078125e-06,
      "model_forward_time": 5.615978717803955,
      "step": 1506
    },
    {
      "epoch": 1.4362335205078125e-06,
      "step": 1506,
      "training_step_time": 17.10934329032898
    },
    {
      "epoch": 1.4371871948242187e-06,
      "model_forward_time": 5.606693506240845,
      "step": 1507
    },
    {
      "epoch": 1.4371871948242187e-06,
      "step": 1507,
      "training_step_time": 17.044180631637573
    },
    {
      "epoch": 1.438140869140625e-06,
      "model_forward_time": 5.56807804107666,
      "step": 1508
    },
    {
      "epoch": 1.438140869140625e-06,
      "step": 1508,
      "training_step_time": 16.96921992301941
    },
    {
      "epoch": 1.4390945434570312e-06,
      "model_forward_time": 5.566065311431885,
      "step": 1509
    },
    {
      "epoch": 1.4390945434570312e-06,
      "step": 1509,
      "training_step_time": 16.98436212539673
    },
    {
      "epoch": 1.4400482177734375e-06,
      "grad_norm": 1.7405797243118286,
      "learning_rate": 1e-05,
      "loss": 0.0973,
      "step": 1510
    },
    {
      "epoch": 1.4400482177734375e-06,
      "model_forward_time": 5.923998832702637,
      "step": 1510
    },
    {
      "dynamics_loss_avg": 0.07507823631167412,
      "epoch": 1.4400482177734375e-06,
      "step": 1510
    },
    {
      "action_loss_avg": 0.024180589802563192,
      "epoch": 1.4400482177734375e-06,
      "step": 1510
    },
    {
      "epoch": 1.4400482177734375e-06,
      "step": 1510,
      "training_step_time": 17.344536066055298
    },
    {
      "epoch": 1.4410018920898437e-06,
      "model_forward_time": 5.579034090042114,
      "step": 1511
    },
    {
      "epoch": 1.4410018920898437e-06,
      "step": 1511,
      "training_step_time": 17.01055884361267
    },
    {
      "epoch": 1.44195556640625e-06,
      "model_forward_time": 5.572027921676636,
      "step": 1512
    },
    {
      "epoch": 1.44195556640625e-06,
      "step": 1512,
      "training_step_time": 16.98512315750122
    },
    {
      "epoch": 1.4429092407226562e-06,
      "model_forward_time": 5.576883554458618,
      "step": 1513
    },
    {
      "epoch": 1.4429092407226562e-06,
      "step": 1513,
      "training_step_time": 16.984875917434692
    },
    {
      "epoch": 1.4438629150390625e-06,
      "model_forward_time": 5.57636284828186,
      "step": 1514
    },
    {
      "epoch": 1.4438629150390625e-06,
      "step": 1514,
      "training_step_time": 17.010324954986572
    },
    {
      "epoch": 1.4448165893554687e-06,
      "model_forward_time": 5.577912092208862,
      "step": 1515
    },
    {
      "epoch": 1.4448165893554687e-06,
      "step": 1515,
      "training_step_time": 17.014830112457275
    },
    {
      "epoch": 1.445770263671875e-06,
      "model_forward_time": 5.961372375488281,
      "step": 1516
    },
    {
      "epoch": 1.445770263671875e-06,
      "step": 1516,
      "training_step_time": 17.40310764312744
    },
    {
      "epoch": 1.4467239379882812e-06,
      "model_forward_time": 5.604893207550049,
      "step": 1517
    },
    {
      "epoch": 1.4467239379882812e-06,
      "step": 1517,
      "training_step_time": 17.084616661071777
    },
    {
      "epoch": 1.4476776123046875e-06,
      "model_forward_time": 5.614236831665039,
      "step": 1518
    },
    {
      "epoch": 1.4476776123046875e-06,
      "step": 1518,
      "training_step_time": 17.09067130088806
    },
    {
      "epoch": 1.4486312866210937e-06,
      "model_forward_time": 5.641597270965576,
      "step": 1519
    },
    {
      "epoch": 1.4486312866210937e-06,
      "step": 1519,
      "training_step_time": 17.101778268814087
    },
    {
      "epoch": 1.4495849609375e-06,
      "grad_norm": 1.692610263824463,
      "learning_rate": 1e-05,
      "loss": 0.1034,
      "step": 1520
    },
    {
      "epoch": 1.4495849609375e-06,
      "model_forward_time": 5.606346368789673,
      "step": 1520
    },
    {
      "dynamics_loss_avg": 0.07593188360333443,
      "epoch": 1.4495849609375e-06,
      "step": 1520
    },
    {
      "action_loss_avg": 0.027287405729293824,
      "epoch": 1.4495849609375e-06,
      "step": 1520
    },
    {
      "epoch": 1.4495849609375e-06,
      "step": 1520,
      "training_step_time": 17.05867624282837
    },
    {
      "epoch": 1.4505386352539062e-06,
      "model_forward_time": 5.9336652755737305,
      "step": 1521
    },
    {
      "epoch": 1.4505386352539062e-06,
      "step": 1521,
      "training_step_time": 17.357628345489502
    },
    {
      "epoch": 1.4514923095703125e-06,
      "model_forward_time": 5.587782144546509,
      "step": 1522
    },
    {
      "epoch": 1.4514923095703125e-06,
      "step": 1522,
      "training_step_time": 17.033705711364746
    },
    {
      "epoch": 1.4524459838867187e-06,
      "model_forward_time": 5.606122970581055,
      "step": 1523
    },
    {
      "epoch": 1.4524459838867187e-06,
      "step": 1523,
      "training_step_time": 17.10005235671997
    },
    {
      "epoch": 1.453399658203125e-06,
      "model_forward_time": 5.620751619338989,
      "step": 1524
    },
    {
      "epoch": 1.453399658203125e-06,
      "step": 1524,
      "training_step_time": 17.118828535079956
    },
    {
      "epoch": 1.4543533325195312e-06,
      "model_forward_time": 5.617259979248047,
      "step": 1525
    },
    {
      "epoch": 1.4543533325195312e-06,
      "step": 1525,
      "training_step_time": 17.08396291732788
    },
    {
      "epoch": 1.4553070068359375e-06,
      "model_forward_time": 5.584760665893555,
      "step": 1526
    },
    {
      "epoch": 1.4553070068359375e-06,
      "step": 1526,
      "training_step_time": 17.413114309310913
    },
    {
      "epoch": 1.4562606811523437e-06,
      "model_forward_time": 5.563512563705444,
      "step": 1527
    },
    {
      "epoch": 1.4562606811523437e-06,
      "step": 1527,
      "training_step_time": 16.976136922836304
    },
    {
      "epoch": 1.45721435546875e-06,
      "model_forward_time": 5.567373037338257,
      "step": 1528
    },
    {
      "epoch": 1.45721435546875e-06,
      "step": 1528,
      "training_step_time": 16.9813334941864
    },
    {
      "epoch": 1.4581680297851562e-06,
      "model_forward_time": 5.578988552093506,
      "step": 1529
    },
    {
      "epoch": 1.4581680297851562e-06,
      "step": 1529,
      "training_step_time": 16.990995407104492
    },
    {
      "epoch": 1.4591217041015625e-06,
      "grad_norm": 0.4329611361026764,
      "learning_rate": 1e-05,
      "loss": 0.0688,
      "step": 1530
    },
    {
      "epoch": 1.4591217041015625e-06,
      "model_forward_time": 5.586091756820679,
      "step": 1530
    },
    {
      "dynamics_loss_avg": 0.05619749762117863,
      "epoch": 1.4591217041015625e-06,
      "step": 1530
    },
    {
      "action_loss_avg": 0.016198427928611637,
      "epoch": 1.4591217041015625e-06,
      "step": 1530
    },
    {
      "epoch": 1.4591217041015625e-06,
      "step": 1530,
      "training_step_time": 17.02605676651001
    },
    {
      "epoch": 1.4600753784179687e-06,
      "model_forward_time": 5.579866170883179,
      "step": 1531
    },
    {
      "epoch": 1.4600753784179687e-06,
      "step": 1531,
      "training_step_time": 17.021612644195557
    },
    {
      "epoch": 1.461029052734375e-06,
      "model_forward_time": 5.946579217910767,
      "step": 1532
    },
    {
      "epoch": 1.461029052734375e-06,
      "step": 1532,
      "training_step_time": 17.40293002128601
    },
    {
      "epoch": 1.4619827270507812e-06,
      "model_forward_time": 5.595267057418823,
      "step": 1533
    },
    {
      "epoch": 1.4619827270507812e-06,
      "step": 1533,
      "training_step_time": 17.056815147399902
    },
    {
      "epoch": 1.4629364013671875e-06,
      "model_forward_time": 5.606712102890015,
      "step": 1534
    },
    {
      "epoch": 1.4629364013671875e-06,
      "step": 1534,
      "training_step_time": 17.070117950439453
    },
    {
      "epoch": 1.4638900756835937e-06,
      "model_forward_time": 5.600931882858276,
      "step": 1535
    },
    {
      "epoch": 1.4638900756835937e-06,
      "step": 1535,
      "training_step_time": 17.080554008483887
    },
    {
      "epoch": 1.46484375e-06,
      "model_forward_time": 5.609839916229248,
      "step": 1536
    },
    {
      "epoch": 1.46484375e-06,
      "step": 1536,
      "training_step_time": 17.0986111164093
    },
    {
      "epoch": 1.4657974243164062e-06,
      "model_forward_time": 5.613932132720947,
      "step": 1537
    },
    {
      "epoch": 1.4657974243164062e-06,
      "step": 1537,
      "training_step_time": 17.464844703674316
    },
    {
      "epoch": 1.4667510986328125e-06,
      "model_forward_time": 5.612747430801392,
      "step": 1538
    },
    {
      "epoch": 1.4667510986328125e-06,
      "step": 1538,
      "training_step_time": 17.104451656341553
    },
    {
      "epoch": 1.4677047729492187e-06,
      "model_forward_time": 5.606096506118774,
      "step": 1539
    },
    {
      "epoch": 1.4677047729492187e-06,
      "step": 1539,
      "training_step_time": 17.104052782058716
    },
    {
      "epoch": 1.468658447265625e-06,
      "grad_norm": 0.6971932649612427,
      "learning_rate": 1e-05,
      "loss": 0.0835,
      "step": 1540
    },
    {
      "epoch": 1.468658447265625e-06,
      "model_forward_time": 5.611685037612915,
      "step": 1540
    },
    {
      "dynamics_loss_avg": 0.06104459129273891,
      "epoch": 1.468658447265625e-06,
      "step": 1540
    },
    {
      "action_loss_avg": 0.01780947307124734,
      "epoch": 1.468658447265625e-06,
      "step": 1540
    },
    {
      "epoch": 1.468658447265625e-06,
      "step": 1540,
      "training_step_time": 17.136680126190186
    },
    {
      "epoch": 1.4696121215820312e-06,
      "model_forward_time": 5.613023281097412,
      "step": 1541
    },
    {
      "epoch": 1.4696121215820312e-06,
      "step": 1541,
      "training_step_time": 17.110649585723877
    },
    {
      "epoch": 1.4705657958984375e-06,
      "model_forward_time": 5.617234945297241,
      "step": 1542
    },
    {
      "epoch": 1.4705657958984375e-06,
      "step": 1542,
      "training_step_time": 17.109501838684082
    },
    {
      "epoch": 1.4715194702148437e-06,
      "model_forward_time": 5.929660797119141,
      "step": 1543
    },
    {
      "epoch": 1.4715194702148437e-06,
      "step": 1543,
      "training_step_time": 17.338295698165894
    },
    {
      "epoch": 1.47247314453125e-06,
      "model_forward_time": 5.571425437927246,
      "step": 1544
    },
    {
      "epoch": 1.47247314453125e-06,
      "step": 1544,
      "training_step_time": 16.9738347530365
    },
    {
      "epoch": 1.4734268188476562e-06,
      "model_forward_time": 5.567342042922974,
      "step": 1545
    },
    {
      "epoch": 1.4734268188476562e-06,
      "step": 1545,
      "training_step_time": 17.003596544265747
    },
    {
      "epoch": 1.4743804931640625e-06,
      "model_forward_time": 5.572691440582275,
      "step": 1546
    },
    {
      "epoch": 1.4743804931640625e-06,
      "step": 1546,
      "training_step_time": 16.99125361442566
    },
    {
      "epoch": 1.4753341674804687e-06,
      "model_forward_time": 5.576744556427002,
      "step": 1547
    },
    {
      "epoch": 1.4753341674804687e-06,
      "step": 1547,
      "training_step_time": 16.98790740966797
    },
    {
      "epoch": 1.476287841796875e-06,
      "model_forward_time": 5.950143575668335,
      "step": 1548
    },
    {
      "epoch": 1.476287841796875e-06,
      "step": 1548,
      "training_step_time": 17.37202215194702
    },
    {
      "epoch": 1.4772415161132812e-06,
      "model_forward_time": 5.593020915985107,
      "step": 1549
    },
    {
      "epoch": 1.4772415161132812e-06,
      "step": 1549,
      "training_step_time": 17.063310384750366
    },
    {
      "epoch": 1.4781951904296875e-06,
      "grad_norm": 1.4277091026306152,
      "learning_rate": 1e-05,
      "loss": 0.069,
      "step": 1550
    },
    {
      "epoch": 1.4781951904296875e-06,
      "model_forward_time": 5.599002361297607,
      "step": 1550
    },
    {
      "dynamics_loss_avg": 0.05433096326887608,
      "epoch": 1.4781951904296875e-06,
      "step": 1550
    },
    {
      "action_loss_avg": 0.018105349456891417,
      "epoch": 1.4781951904296875e-06,
      "step": 1550
    },
    {
      "epoch": 1.4781951904296875e-06,
      "step": 1550,
      "training_step_time": 17.084991931915283
    },
    {
      "epoch": 1.4791488647460937e-06,
      "model_forward_time": 5.6274094581604,
      "step": 1551
    },
    {
      "epoch": 1.4791488647460937e-06,
      "step": 1551,
      "training_step_time": 17.14054846763611
    },
    {
      "epoch": 1.4801025390625e-06,
      "model_forward_time": 5.610320806503296,
      "step": 1552
    },
    {
      "epoch": 1.4801025390625e-06,
      "step": 1552,
      "training_step_time": 17.11759066581726
    },
    {
      "epoch": 1.4810562133789062e-06,
      "model_forward_time": 5.616767883300781,
      "step": 1553
    },
    {
      "epoch": 1.4810562133789062e-06,
      "step": 1553,
      "training_step_time": 17.45173978805542
    },
    {
      "epoch": 1.4820098876953125e-06,
      "model_forward_time": 5.590007543563843,
      "step": 1554
    },
    {
      "epoch": 1.4820098876953125e-06,
      "step": 1554,
      "training_step_time": 16.995062828063965
    },
    {
      "epoch": 1.4829635620117187e-06,
      "model_forward_time": 5.573533296585083,
      "step": 1555
    },
    {
      "epoch": 1.4829635620117187e-06,
      "step": 1555,
      "training_step_time": 16.977455139160156
    },
    {
      "epoch": 1.483917236328125e-06,
      "model_forward_time": 5.565342903137207,
      "step": 1556
    },
    {
      "epoch": 1.483917236328125e-06,
      "step": 1556,
      "training_step_time": 17.00459098815918
    },
    {
      "epoch": 1.4848709106445312e-06,
      "model_forward_time": 5.559155464172363,
      "step": 1557
    },
    {
      "epoch": 1.4848709106445312e-06,
      "step": 1557,
      "training_step_time": 16.96788001060486
    },
    {
      "epoch": 1.4858245849609375e-06,
      "model_forward_time": 5.57506251335144,
      "step": 1558
    },
    {
      "epoch": 1.4858245849609375e-06,
      "step": 1558,
      "training_step_time": 16.978172063827515
    },
    {
      "epoch": 1.4867782592773437e-06,
      "model_forward_time": 5.938916921615601,
      "step": 1559
    },
    {
      "epoch": 1.4867782592773437e-06,
      "step": 1559,
      "training_step_time": 17.344078540802002
    },
    {
      "epoch": 1.48773193359375e-06,
      "grad_norm": 1.4291729927062988,
      "learning_rate": 1e-05,
      "loss": 0.0796,
      "step": 1560
    },
    {
      "epoch": 1.48773193359375e-06,
      "model_forward_time": 5.590505123138428,
      "step": 1560
    },
    {
      "dynamics_loss_avg": 0.059776948764920235,
      "epoch": 1.48773193359375e-06,
      "step": 1560
    },
    {
      "action_loss_avg": 0.018551695672795177,
      "epoch": 1.48773193359375e-06,
      "step": 1560
    },
    {
      "epoch": 1.48773193359375e-06,
      "step": 1560,
      "training_step_time": 17.047280311584473
    },
    {
      "epoch": 1.4886856079101562e-06,
      "model_forward_time": 5.596821069717407,
      "step": 1561
    },
    {
      "epoch": 1.4886856079101562e-06,
      "step": 1561,
      "training_step_time": 17.03773856163025
    },
    {
      "epoch": 1.4896392822265626e-06,
      "model_forward_time": 5.607382297515869,
      "step": 1562
    },
    {
      "epoch": 1.4896392822265626e-06,
      "step": 1562,
      "training_step_time": 17.107445240020752
    },
    {
      "epoch": 1.4905929565429687e-06,
      "model_forward_time": 5.6124725341796875,
      "step": 1563
    },
    {
      "epoch": 1.4905929565429687e-06,
      "step": 1563,
      "training_step_time": 17.103094816207886
    },
    {
      "epoch": 1.491546630859375e-06,
      "model_forward_time": 5.976920127868652,
      "step": 1564
    },
    {
      "epoch": 1.491546630859375e-06,
      "step": 1564,
      "training_step_time": 17.463231086730957
    },
    {
      "epoch": 1.4925003051757812e-06,
      "model_forward_time": 5.618849992752075,
      "step": 1565
    },
    {
      "epoch": 1.4925003051757812e-06,
      "step": 1565,
      "training_step_time": 17.11833667755127
    },
    {
      "epoch": 1.4934539794921876e-06,
      "model_forward_time": 5.608551740646362,
      "step": 1566
    },
    {
      "epoch": 1.4934539794921876e-06,
      "step": 1566,
      "training_step_time": 17.095062494277954
    },
    {
      "epoch": 1.4944076538085937e-06,
      "model_forward_time": 5.6145031452178955,
      "step": 1567
    },
    {
      "epoch": 1.4944076538085937e-06,
      "step": 1567,
      "training_step_time": 17.094468116760254
    },
    {
      "epoch": 1.495361328125e-06,
      "model_forward_time": 5.60609769821167,
      "step": 1568
    },
    {
      "epoch": 1.495361328125e-06,
      "step": 1568,
      "training_step_time": 17.06189799308777
    },
    {
      "epoch": 1.4963150024414062e-06,
      "model_forward_time": 5.58258318901062,
      "step": 1569
    },
    {
      "epoch": 1.4963150024414062e-06,
      "step": 1569,
      "training_step_time": 16.99104380607605
    },
    {
      "epoch": 1.4972686767578126e-06,
      "grad_norm": 1.3381620645523071,
      "learning_rate": 1e-05,
      "loss": 0.0934,
      "step": 1570
    },
    {
      "epoch": 1.4972686767578126e-06,
      "model_forward_time": 5.945866107940674,
      "step": 1570
    },
    {
      "dynamics_loss_avg": 0.06694075986742973,
      "epoch": 1.4972686767578126e-06,
      "step": 1570
    },
    {
      "action_loss_avg": 0.025956328539177776,
      "epoch": 1.4972686767578126e-06,
      "step": 1570
    },
    {
      "epoch": 1.4972686767578126e-06,
      "step": 1570,
      "training_step_time": 17.349159955978394
    },
    {
      "epoch": 1.4982223510742187e-06,
      "model_forward_time": 5.593486547470093,
      "step": 1571
    },
    {
      "epoch": 1.4982223510742187e-06,
      "step": 1571,
      "training_step_time": 17.030656576156616
    },
    {
      "epoch": 1.499176025390625e-06,
      "model_forward_time": 5.596039772033691,
      "step": 1572
    },
    {
      "epoch": 1.499176025390625e-06,
      "step": 1572,
      "training_step_time": 17.088537216186523
    },
    {
      "epoch": 1.5001296997070312e-06,
      "model_forward_time": 5.616868734359741,
      "step": 1573
    },
    {
      "epoch": 1.5001296997070312e-06,
      "step": 1573,
      "training_step_time": 17.129778146743774
    },
    {
      "epoch": 1.5010833740234376e-06,
      "model_forward_time": 5.624483346939087,
      "step": 1574
    },
    {
      "epoch": 1.5010833740234376e-06,
      "step": 1574,
      "training_step_time": 17.124531745910645
    },
    {
      "epoch": 1.5020370483398437e-06,
      "model_forward_time": 5.963633060455322,
      "step": 1575
    },
    {
      "epoch": 1.5020370483398437e-06,
      "step": 1575,
      "training_step_time": 17.44122624397278
    },
    {
      "epoch": 1.50299072265625e-06,
      "model_forward_time": 5.6097400188446045,
      "step": 1576
    },
    {
      "epoch": 1.50299072265625e-06,
      "step": 1576,
      "training_step_time": 17.102824926376343
    },
    {
      "epoch": 1.5039443969726562e-06,
      "model_forward_time": 5.603147745132446,
      "step": 1577
    },
    {
      "epoch": 1.5039443969726562e-06,
      "step": 1577,
      "training_step_time": 17.05378794670105
    },
    {
      "epoch": 1.5048980712890626e-06,
      "model_forward_time": 5.589188575744629,
      "step": 1578
    },
    {
      "epoch": 1.5048980712890626e-06,
      "step": 1578,
      "training_step_time": 17.037803649902344
    },
    {
      "epoch": 1.5058517456054687e-06,
      "model_forward_time": 5.589381694793701,
      "step": 1579
    },
    {
      "epoch": 1.5058517456054687e-06,
      "step": 1579,
      "training_step_time": 17.02500009536743
    },
    {
      "epoch": 1.506805419921875e-06,
      "grad_norm": 1.501268744468689,
      "learning_rate": 1e-05,
      "loss": 0.0952,
      "step": 1580
    },
    {
      "epoch": 1.506805419921875e-06,
      "model_forward_time": 5.591887712478638,
      "step": 1580
    },
    {
      "dynamics_loss_avg": 0.07287818603217602,
      "epoch": 1.506805419921875e-06,
      "step": 1580
    },
    {
      "action_loss_avg": 0.024900022894144058,
      "epoch": 1.506805419921875e-06,
      "step": 1580
    },
    {
      "epoch": 1.506805419921875e-06,
      "step": 1580,
      "training_step_time": 17.04934287071228
    },
    {
      "epoch": 1.5077590942382812e-06,
      "model_forward_time": 5.962072134017944,
      "step": 1581
    },
    {
      "epoch": 1.5077590942382812e-06,
      "step": 1581,
      "training_step_time": 17.43065047264099
    },
    {
      "epoch": 1.5087127685546876e-06,
      "model_forward_time": 5.611337661743164,
      "step": 1582
    },
    {
      "epoch": 1.5087127685546876e-06,
      "step": 1582,
      "training_step_time": 17.119142055511475
    },
    {
      "epoch": 1.5096664428710937e-06,
      "model_forward_time": 5.622099876403809,
      "step": 1583
    },
    {
      "epoch": 1.5096664428710937e-06,
      "step": 1583,
      "training_step_time": 17.081349849700928
    },
    {
      "epoch": 1.5106201171875e-06,
      "model_forward_time": 5.600662469863892,
      "step": 1584
    },
    {
      "epoch": 1.5106201171875e-06,
      "step": 1584,
      "training_step_time": 17.03214144706726
    },
    {
      "epoch": 1.5115737915039062e-06,
      "model_forward_time": 5.576812028884888,
      "step": 1585
    },
    {
      "epoch": 1.5115737915039062e-06,
      "step": 1585,
      "training_step_time": 16.985628604888916
    },
    {
      "epoch": 1.5125274658203126e-06,
      "model_forward_time": 5.934234857559204,
      "step": 1586
    },
    {
      "epoch": 1.5125274658203126e-06,
      "step": 1586,
      "training_step_time": 17.355260848999023
    },
    {
      "epoch": 1.5134811401367187e-06,
      "model_forward_time": 5.5727667808532715,
      "step": 1587
    },
    {
      "epoch": 1.5134811401367187e-06,
      "step": 1587,
      "training_step_time": 17.006967306137085
    },
    {
      "epoch": 1.514434814453125e-06,
      "model_forward_time": 5.578892707824707,
      "step": 1588
    },
    {
      "epoch": 1.514434814453125e-06,
      "step": 1588,
      "training_step_time": 16.98723816871643
    },
    {
      "epoch": 1.5153884887695312e-06,
      "model_forward_time": 5.592259883880615,
      "step": 1589
    },
    {
      "epoch": 1.5153884887695312e-06,
      "step": 1589,
      "training_step_time": 17.02364230155945
    },
    {
      "epoch": 1.5163421630859376e-06,
      "grad_norm": 2.414538621902466,
      "learning_rate": 1e-05,
      "loss": 0.0994,
      "step": 1590
    },
    {
      "epoch": 1.5163421630859376e-06,
      "model_forward_time": 5.603297710418701,
      "step": 1590
    },
    {
      "dynamics_loss_avg": 0.06756284162402153,
      "epoch": 1.5163421630859376e-06,
      "step": 1590
    },
    {
      "action_loss_avg": 0.02822255790233612,
      "epoch": 1.5163421630859376e-06,
      "step": 1590
    },
    {
      "epoch": 1.5163421630859376e-06,
      "step": 1590,
      "training_step_time": 17.083399772644043
    },
    {
      "epoch": 1.5172958374023437e-06,
      "model_forward_time": 5.607856512069702,
      "step": 1591
    },
    {
      "epoch": 1.5172958374023437e-06,
      "step": 1591,
      "training_step_time": 17.458441257476807
    },
    {
      "epoch": 1.51824951171875e-06,
      "model_forward_time": 5.611372709274292,
      "step": 1592
    },
    {
      "epoch": 1.51824951171875e-06,
      "step": 1592,
      "training_step_time": 17.096303462982178
    },
    {
      "epoch": 1.5192031860351562e-06,
      "model_forward_time": 5.596982717514038,
      "step": 1593
    },
    {
      "epoch": 1.5192031860351562e-06,
      "step": 1593,
      "training_step_time": 17.052825450897217
    },
    {
      "epoch": 1.5201568603515626e-06,
      "model_forward_time": 5.585917234420776,
      "step": 1594
    },
    {
      "epoch": 1.5201568603515626e-06,
      "step": 1594,
      "training_step_time": 17.020069122314453
    },
    {
      "epoch": 1.5211105346679687e-06,
      "model_forward_time": 5.58200740814209,
      "step": 1595
    },
    {
      "epoch": 1.5211105346679687e-06,
      "step": 1595,
      "training_step_time": 17.01033592224121
    },
    {
      "epoch": 1.522064208984375e-06,
      "model_forward_time": 5.588299036026001,
      "step": 1596
    },
    {
      "epoch": 1.522064208984375e-06,
      "step": 1596,
      "training_step_time": 17.034693241119385
    },
    {
      "epoch": 1.5230178833007812e-06,
      "model_forward_time": 5.939826250076294,
      "step": 1597
    },
    {
      "epoch": 1.5230178833007812e-06,
      "step": 1597,
      "training_step_time": 17.41270899772644
    },
    {
      "epoch": 1.5239715576171876e-06,
      "model_forward_time": 5.610223293304443,
      "step": 1598
    },
    {
      "epoch": 1.5239715576171876e-06,
      "step": 1598,
      "training_step_time": 17.082682132720947
    },
    {
      "epoch": 1.5249252319335937e-06,
      "model_forward_time": 5.600521087646484,
      "step": 1599
    },
    {
      "epoch": 1.5249252319335937e-06,
      "step": 1599,
      "training_step_time": 17.07077407836914
    },
    {
      "epoch": 1.52587890625e-06,
      "grad_norm": 1.7332537174224854,
      "learning_rate": 1e-05,
      "loss": 0.0881,
      "step": 1600
    },
    {
      "epoch": 1.52587890625e-06,
      "model_forward_time": 5.603912830352783,
      "step": 1600
    },
    {
      "dynamics_loss_avg": 0.06651818566024303,
      "epoch": 1.52587890625e-06,
      "step": 1600
    },
    {
      "action_loss_avg": 0.02071985835209489,
      "epoch": 1.52587890625e-06,
      "step": 1600
    },
    {
      "epoch": 1.52587890625e-06,
      "step": 1600,
      "training_step_time": 17.056645154953003
    },
    {
      "epoch": 1.5268325805664062e-06,
      "model_forward_time": 5.5947980880737305,
      "step": 1601
    },
    {
      "epoch": 1.5268325805664062e-06,
      "step": 1601,
      "training_step_time": 17.04496479034424
    },
    {
      "epoch": 1.5277862548828126e-06,
      "model_forward_time": 5.957691669464111,
      "step": 1602
    },
    {
      "epoch": 1.5277862548828126e-06,
      "step": 1602,
      "training_step_time": 17.391950368881226
    },
    {
      "epoch": 1.5287399291992187e-06,
      "model_forward_time": 5.588972806930542,
      "step": 1603
    },
    {
      "epoch": 1.5287399291992187e-06,
      "step": 1603,
      "training_step_time": 17.05430269241333
    },
    {
      "epoch": 1.529693603515625e-06,
      "model_forward_time": 5.600311994552612,
      "step": 1604
    },
    {
      "epoch": 1.529693603515625e-06,
      "step": 1604,
      "training_step_time": 17.046523332595825
    },
    {
      "epoch": 1.5306472778320312e-06,
      "model_forward_time": 5.6176676750183105,
      "step": 1605
    },
    {
      "epoch": 1.5306472778320312e-06,
      "step": 1605,
      "training_step_time": 17.117149591445923
    },
    {
      "epoch": 1.5316009521484376e-06,
      "model_forward_time": 5.600653171539307,
      "step": 1606
    },
    {
      "epoch": 1.5316009521484376e-06,
      "step": 1606,
      "training_step_time": 17.100870609283447
    },
    {
      "epoch": 1.5325546264648437e-06,
      "model_forward_time": 5.630834102630615,
      "step": 1607
    },
    {
      "epoch": 1.5325546264648437e-06,
      "step": 1607,
      "training_step_time": 17.13155508041382
    },
    {
      "epoch": 1.53350830078125e-06,
      "model_forward_time": 5.960390329360962,
      "step": 1608
    },
    {
      "epoch": 1.53350830078125e-06,
      "step": 1608,
      "training_step_time": 17.46769618988037
    },
    {
      "epoch": 1.5344619750976562e-06,
      "model_forward_time": 5.620260715484619,
      "step": 1609
    },
    {
      "epoch": 1.5344619750976562e-06,
      "step": 1609,
      "training_step_time": 17.11618947982788
    },
    {
      "epoch": 1.5354156494140626e-06,
      "grad_norm": 1.204938292503357,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 1610
    },
    {
      "epoch": 1.5354156494140626e-06,
      "model_forward_time": 5.612297773361206,
      "step": 1610
    },
    {
      "dynamics_loss_avg": 0.05874115638434887,
      "epoch": 1.5354156494140626e-06,
      "step": 1610
    },
    {
      "action_loss_avg": 0.017330134753137827,
      "epoch": 1.5354156494140626e-06,
      "step": 1610
    },
    {
      "epoch": 1.5354156494140626e-06,
      "step": 1610,
      "training_step_time": 17.105718851089478
    },
    {
      "epoch": 1.5363693237304687e-06,
      "model_forward_time": 5.6011643409729,
      "step": 1611
    },
    {
      "epoch": 1.5363693237304687e-06,
      "step": 1611,
      "training_step_time": 17.083617687225342
    },
    {
      "epoch": 1.537322998046875e-06,
      "model_forward_time": 5.602293491363525,
      "step": 1612
    },
    {
      "epoch": 1.537322998046875e-06,
      "step": 1612,
      "training_step_time": 17.071938514709473
    },
    {
      "epoch": 1.5382766723632812e-06,
      "model_forward_time": 5.96733021736145,
      "step": 1613
    },
    {
      "epoch": 1.5382766723632812e-06,
      "step": 1613,
      "training_step_time": 17.43226671218872
    },
    {
      "epoch": 1.5392303466796876e-06,
      "model_forward_time": 5.615089654922485,
      "step": 1614
    },
    {
      "epoch": 1.5392303466796876e-06,
      "step": 1614,
      "training_step_time": 17.1222562789917
    },
    {
      "epoch": 1.5401840209960937e-06,
      "model_forward_time": 5.6185479164123535,
      "step": 1615
    },
    {
      "epoch": 1.5401840209960937e-06,
      "step": 1615,
      "training_step_time": 17.136576890945435
    },
    {
      "epoch": 1.5411376953125e-06,
      "model_forward_time": 5.620175838470459,
      "step": 1616
    },
    {
      "epoch": 1.5411376953125e-06,
      "step": 1616,
      "training_step_time": 17.117445707321167
    },
    {
      "epoch": 1.5420913696289062e-06,
      "model_forward_time": 5.617905855178833,
      "step": 1617
    },
    {
      "epoch": 1.5420913696289062e-06,
      "step": 1617,
      "training_step_time": 17.097195386886597
    },
    {
      "epoch": 1.5430450439453126e-06,
      "model_forward_time": 5.627084970474243,
      "step": 1618
    },
    {
      "epoch": 1.5430450439453126e-06,
      "step": 1618,
      "training_step_time": 17.1304190158844
    },
    {
      "epoch": 1.5439987182617187e-06,
      "model_forward_time": 5.969823598861694,
      "step": 1619
    },
    {
      "epoch": 1.5439987182617187e-06,
      "step": 1619,
      "training_step_time": 17.44177794456482
    },
    {
      "epoch": 1.544952392578125e-06,
      "grad_norm": 1.0305140018463135,
      "learning_rate": 1e-05,
      "loss": 0.0927,
      "step": 1620
    },
    {
      "epoch": 1.544952392578125e-06,
      "model_forward_time": 5.603923082351685,
      "step": 1620
    },
    {
      "dynamics_loss_avg": 0.06874711513519287,
      "epoch": 1.544952392578125e-06,
      "step": 1620
    },
    {
      "action_loss_avg": 0.021069794707000256,
      "epoch": 1.544952392578125e-06,
      "step": 1620
    },
    {
      "epoch": 1.544952392578125e-06,
      "step": 1620,
      "training_step_time": 17.068175792694092
    },
    {
      "epoch": 1.5459060668945312e-06,
      "model_forward_time": 5.593944549560547,
      "step": 1621
    },
    {
      "epoch": 1.5459060668945312e-06,
      "step": 1621,
      "training_step_time": 17.027690172195435
    },
    {
      "epoch": 1.5468597412109376e-06,
      "model_forward_time": 5.5776989459991455,
      "step": 1622
    },
    {
      "epoch": 1.5468597412109376e-06,
      "step": 1622,
      "training_step_time": 17.01495909690857
    },
    {
      "epoch": 1.5478134155273437e-06,
      "model_forward_time": 5.593823432922363,
      "step": 1623
    },
    {
      "epoch": 1.5478134155273437e-06,
      "step": 1623,
      "training_step_time": 17.023277759552002
    },
    {
      "epoch": 1.54876708984375e-06,
      "model_forward_time": 5.957764387130737,
      "step": 1624
    },
    {
      "epoch": 1.54876708984375e-06,
      "step": 1624,
      "training_step_time": 17.419724941253662
    },
    {
      "epoch": 1.5497207641601562e-06,
      "model_forward_time": 5.592472076416016,
      "step": 1625
    },
    {
      "epoch": 1.5497207641601562e-06,
      "step": 1625,
      "training_step_time": 17.03422474861145
    },
    {
      "epoch": 1.5506744384765624e-06,
      "model_forward_time": 5.594497919082642,
      "step": 1626
    },
    {
      "epoch": 1.5506744384765624e-06,
      "step": 1626,
      "training_step_time": 17.055012702941895
    },
    {
      "epoch": 1.5516281127929688e-06,
      "model_forward_time": 5.594755411148071,
      "step": 1627
    },
    {
      "epoch": 1.5516281127929688e-06,
      "step": 1627,
      "training_step_time": 17.05445432662964
    },
    {
      "epoch": 1.552581787109375e-06,
      "model_forward_time": 5.604476690292358,
      "step": 1628
    },
    {
      "epoch": 1.552581787109375e-06,
      "step": 1628,
      "training_step_time": 17.062325954437256
    },
    {
      "epoch": 1.5535354614257813e-06,
      "model_forward_time": 5.607899904251099,
      "step": 1629
    },
    {
      "epoch": 1.5535354614257813e-06,
      "step": 1629,
      "training_step_time": 17.086694955825806
    },
    {
      "epoch": 1.5544891357421874e-06,
      "grad_norm": 1.2747479677200317,
      "learning_rate": 1e-05,
      "loss": 0.0766,
      "step": 1630
    },
    {
      "epoch": 1.5544891357421874e-06,
      "model_forward_time": 5.933743000030518,
      "step": 1630
    },
    {
      "dynamics_loss_avg": 0.05983915776014328,
      "epoch": 1.5544891357421874e-06,
      "step": 1630
    },
    {
      "action_loss_avg": 0.01762278685346246,
      "epoch": 1.5544891357421874e-06,
      "step": 1630
    },
    {
      "epoch": 1.5544891357421874e-06,
      "step": 1630,
      "training_step_time": 17.397435665130615
    },
    {
      "epoch": 1.5554428100585938e-06,
      "model_forward_time": 5.581878662109375,
      "step": 1631
    },
    {
      "epoch": 1.5554428100585938e-06,
      "step": 1631,
      "training_step_time": 17.012770175933838
    },
    {
      "epoch": 1.556396484375e-06,
      "model_forward_time": 5.584856986999512,
      "step": 1632
    },
    {
      "epoch": 1.556396484375e-06,
      "step": 1632,
      "training_step_time": 17.02278470993042
    },
    {
      "epoch": 1.5573501586914063e-06,
      "model_forward_time": 5.593059539794922,
      "step": 1633
    },
    {
      "epoch": 1.5573501586914063e-06,
      "step": 1633,
      "training_step_time": 17.05814504623413
    },
    {
      "epoch": 1.5583038330078124e-06,
      "model_forward_time": 5.594212532043457,
      "step": 1634
    },
    {
      "epoch": 1.5583038330078124e-06,
      "step": 1634,
      "training_step_time": 17.07409143447876
    },
    {
      "epoch": 1.5592575073242188e-06,
      "model_forward_time": 5.987576484680176,
      "step": 1635
    },
    {
      "epoch": 1.5592575073242188e-06,
      "step": 1635,
      "training_step_time": 17.48751139640808
    },
    {
      "epoch": 1.560211181640625e-06,
      "model_forward_time": 5.613900661468506,
      "step": 1636
    },
    {
      "epoch": 1.560211181640625e-06,
      "step": 1636,
      "training_step_time": 17.107735872268677
    },
    {
      "epoch": 1.5611648559570313e-06,
      "model_forward_time": 5.616432189941406,
      "step": 1637
    },
    {
      "epoch": 1.5611648559570313e-06,
      "step": 1637,
      "training_step_time": 17.09407925605774
    },
    {
      "epoch": 1.5621185302734374e-06,
      "model_forward_time": 5.599172830581665,
      "step": 1638
    },
    {
      "epoch": 1.5621185302734374e-06,
      "step": 1638,
      "training_step_time": 17.030062913894653
    },
    {
      "epoch": 1.5630722045898438e-06,
      "model_forward_time": 5.577911615371704,
      "step": 1639
    },
    {
      "epoch": 1.5630722045898438e-06,
      "step": 1639,
      "training_step_time": 16.99245262145996
    },
    {
      "epoch": 1.56402587890625e-06,
      "grad_norm": 1.8319463729858398,
      "learning_rate": 1e-05,
      "loss": 0.1003,
      "step": 1640
    },
    {
      "epoch": 1.56402587890625e-06,
      "model_forward_time": 5.938848257064819,
      "step": 1640
    },
    {
      "dynamics_loss_avg": 0.07687062844634056,
      "epoch": 1.56402587890625e-06,
      "step": 1640
    },
    {
      "action_loss_avg": 0.025505608506500722,
      "epoch": 1.56402587890625e-06,
      "step": 1640
    },
    {
      "epoch": 1.56402587890625e-06,
      "step": 1640,
      "training_step_time": 17.381781816482544
    },
    {
      "epoch": 1.5649795532226563e-06,
      "model_forward_time": 5.574368238449097,
      "step": 1641
    },
    {
      "epoch": 1.5649795532226563e-06,
      "step": 1641,
      "training_step_time": 16.97561502456665
    },
    {
      "epoch": 1.5659332275390624e-06,
      "model_forward_time": 5.592215538024902,
      "step": 1642
    },
    {
      "epoch": 1.5659332275390624e-06,
      "step": 1642,
      "training_step_time": 17.03403329849243
    },
    {
      "epoch": 1.5668869018554688e-06,
      "model_forward_time": 5.598730802536011,
      "step": 1643
    },
    {
      "epoch": 1.5668869018554688e-06,
      "step": 1643,
      "training_step_time": 17.05396866798401
    },
    {
      "epoch": 1.567840576171875e-06,
      "model_forward_time": 5.606250524520874,
      "step": 1644
    },
    {
      "epoch": 1.567840576171875e-06,
      "step": 1644,
      "training_step_time": 17.040013074874878
    },
    {
      "epoch": 1.5687942504882813e-06,
      "model_forward_time": 5.597843647003174,
      "step": 1645
    },
    {
      "epoch": 1.5687942504882813e-06,
      "step": 1645,
      "training_step_time": 17.040918588638306
    },
    {
      "epoch": 1.5697479248046874e-06,
      "model_forward_time": 5.977355241775513,
      "step": 1646
    },
    {
      "epoch": 1.5697479248046874e-06,
      "step": 1646,
      "training_step_time": 17.424283742904663
    },
    {
      "epoch": 1.5707015991210938e-06,
      "model_forward_time": 5.592496633529663,
      "step": 1647
    },
    {
      "epoch": 1.5707015991210938e-06,
      "step": 1647,
      "training_step_time": 17.024719715118408
    },
    {
      "epoch": 1.5716552734375e-06,
      "model_forward_time": 5.593469858169556,
      "step": 1648
    },
    {
      "epoch": 1.5716552734375e-06,
      "step": 1648,
      "training_step_time": 17.07305884361267
    },
    {
      "epoch": 1.5726089477539063e-06,
      "model_forward_time": 5.605470180511475,
      "step": 1649
    },
    {
      "epoch": 1.5726089477539063e-06,
      "step": 1649,
      "training_step_time": 17.08349871635437
    },
    {
      "epoch": 1.5735626220703124e-06,
      "grad_norm": 1.1141763925552368,
      "learning_rate": 1e-05,
      "loss": 0.0775,
      "step": 1650
    },
    {
      "epoch": 1.5735626220703124e-06,
      "model_forward_time": 5.608829736709595,
      "step": 1650
    },
    {
      "dynamics_loss_avg": 0.05127138569951058,
      "epoch": 1.5735626220703124e-06,
      "step": 1650
    },
    {
      "action_loss_avg": 0.015602567885071039,
      "epoch": 1.5735626220703124e-06,
      "step": 1650
    },
    {
      "epoch": 1.5735626220703124e-06,
      "step": 1650,
      "training_step_time": 17.062310218811035
    },
    {
      "epoch": 1.5745162963867188e-06,
      "model_forward_time": 5.5926148891448975,
      "step": 1651
    },
    {
      "epoch": 1.5745162963867188e-06,
      "step": 1651,
      "training_step_time": 17.386746644973755
    },
    {
      "epoch": 1.575469970703125e-06,
      "model_forward_time": 5.572753429412842,
      "step": 1652
    },
    {
      "epoch": 1.575469970703125e-06,
      "step": 1652,
      "training_step_time": 17.028387784957886
    },
    {
      "epoch": 1.5764236450195313e-06,
      "model_forward_time": 5.593424081802368,
      "step": 1653
    },
    {
      "epoch": 1.5764236450195313e-06,
      "step": 1653,
      "training_step_time": 17.057994842529297
    },
    {
      "epoch": 1.5773773193359374e-06,
      "model_forward_time": 5.598496437072754,
      "step": 1654
    },
    {
      "epoch": 1.5773773193359374e-06,
      "step": 1654,
      "training_step_time": 17.06385612487793
    },
    {
      "epoch": 1.5783309936523438e-06,
      "model_forward_time": 5.601763725280762,
      "step": 1655
    },
    {
      "epoch": 1.5783309936523438e-06,
      "step": 1655,
      "training_step_time": 17.05269193649292
    },
    {
      "epoch": 1.57928466796875e-06,
      "model_forward_time": 5.59607458114624,
      "step": 1656
    },
    {
      "epoch": 1.57928466796875e-06,
      "step": 1656,
      "training_step_time": 17.049186944961548
    },
    {
      "epoch": 1.5802383422851563e-06,
      "model_forward_time": 5.968242645263672,
      "step": 1657
    },
    {
      "epoch": 1.5802383422851563e-06,
      "step": 1657,
      "training_step_time": 17.41157627105713
    },
    {
      "epoch": 1.5811920166015624e-06,
      "model_forward_time": 5.5957324504852295,
      "step": 1658
    },
    {
      "epoch": 1.5811920166015624e-06,
      "step": 1658,
      "training_step_time": 17.063329696655273
    },
    {
      "epoch": 1.5821456909179688e-06,
      "model_forward_time": 5.595636606216431,
      "step": 1659
    },
    {
      "epoch": 1.5821456909179688e-06,
      "step": 1659,
      "training_step_time": 17.021294355392456
    },
    {
      "epoch": 1.583099365234375e-06,
      "grad_norm": 1.469256043434143,
      "learning_rate": 1e-05,
      "loss": 0.0779,
      "step": 1660
    },
    {
      "epoch": 1.583099365234375e-06,
      "model_forward_time": 5.595615863800049,
      "step": 1660
    },
    {
      "dynamics_loss_avg": 0.060426528379321096,
      "epoch": 1.583099365234375e-06,
      "step": 1660
    },
    {
      "action_loss_avg": 0.020877213031053544,
      "epoch": 1.583099365234375e-06,
      "step": 1660
    },
    {
      "epoch": 1.583099365234375e-06,
      "step": 1660,
      "training_step_time": 17.05970001220703
    },
    {
      "epoch": 1.5840530395507813e-06,
      "model_forward_time": 5.598189353942871,
      "step": 1661
    },
    {
      "epoch": 1.5840530395507813e-06,
      "step": 1661,
      "training_step_time": 17.06320023536682
    },
    {
      "epoch": 1.5850067138671874e-06,
      "model_forward_time": 5.60493803024292,
      "step": 1662
    },
    {
      "epoch": 1.5850067138671874e-06,
      "step": 1662,
      "training_step_time": 17.46348237991333
    },
    {
      "epoch": 1.5859603881835938e-06,
      "model_forward_time": 5.604578495025635,
      "step": 1663
    },
    {
      "epoch": 1.5859603881835938e-06,
      "step": 1663,
      "training_step_time": 17.10383701324463
    },
    {
      "epoch": 1.5869140625e-06,
      "model_forward_time": 5.615831613540649,
      "step": 1664
    },
    {
      "epoch": 1.5869140625e-06,
      "step": 1664,
      "training_step_time": 17.09937596321106
    },
    {
      "epoch": 1.5878677368164063e-06,
      "model_forward_time": 5.594854116439819,
      "step": 1665
    },
    {
      "epoch": 1.5878677368164063e-06,
      "step": 1665,
      "training_step_time": 17.03085947036743
    },
    {
      "epoch": 1.5888214111328124e-06,
      "model_forward_time": 5.576177597045898,
      "step": 1666
    },
    {
      "epoch": 1.5888214111328124e-06,
      "step": 1666,
      "training_step_time": 16.997639179229736
    },
    {
      "epoch": 1.5897750854492188e-06,
      "model_forward_time": 5.567614316940308,
      "step": 1667
    },
    {
      "epoch": 1.5897750854492188e-06,
      "step": 1667,
      "training_step_time": 16.992440700531006
    },
    {
      "epoch": 1.590728759765625e-06,
      "model_forward_time": 5.968945264816284,
      "step": 1668
    },
    {
      "epoch": 1.590728759765625e-06,
      "step": 1668,
      "training_step_time": 17.4212486743927
    },
    {
      "epoch": 1.5916824340820313e-06,
      "model_forward_time": 5.604738235473633,
      "step": 1669
    },
    {
      "epoch": 1.5916824340820313e-06,
      "step": 1669,
      "training_step_time": 17.09504461288452
    },
    {
      "epoch": 1.5926361083984374e-06,
      "grad_norm": 1.6251579523086548,
      "learning_rate": 1e-05,
      "loss": 0.0916,
      "step": 1670
    },
    {
      "epoch": 1.5926361083984374e-06,
      "model_forward_time": 5.62131929397583,
      "step": 1670
    },
    {
      "dynamics_loss_avg": 0.07466685138642788,
      "epoch": 1.5926361083984374e-06,
      "step": 1670
    },
    {
      "action_loss_avg": 0.025122064258903266,
      "epoch": 1.5926361083984374e-06,
      "step": 1670
    },
    {
      "epoch": 1.5926361083984374e-06,
      "step": 1670,
      "training_step_time": 17.09648585319519
    },
    {
      "epoch": 1.5935897827148438e-06,
      "model_forward_time": 5.602538347244263,
      "step": 1671
    },
    {
      "epoch": 1.5935897827148438e-06,
      "step": 1671,
      "training_step_time": 17.069337129592896
    },
    {
      "epoch": 1.59454345703125e-06,
      "model_forward_time": 4.274318695068359,
      "step": 1672
    },
    {
      "epoch": 1.59454345703125e-06,
      "step": 1672,
      "training_step_time": 12.806114673614502
    },
    {
      "epoch": 1.5954971313476563e-06,
      "model_forward_time": 5.599551677703857,
      "step": 1673
    },
    {
      "epoch": 1.5954971313476563e-06,
      "step": 1673,
      "training_step_time": 17.417221069335938
    },
    {
      "epoch": 1.5964508056640624e-06,
      "model_forward_time": 5.601998329162598,
      "step": 1674
    },
    {
      "epoch": 1.5964508056640624e-06,
      "step": 1674,
      "training_step_time": 17.043769121170044
    },
    {
      "epoch": 1.5974044799804688e-06,
      "model_forward_time": 5.593741416931152,
      "step": 1675
    },
    {
      "epoch": 1.5974044799804688e-06,
      "step": 1675,
      "training_step_time": 17.044956922531128
    },
    {
      "epoch": 1.598358154296875e-06,
      "model_forward_time": 5.5895678997039795,
      "step": 1676
    },
    {
      "epoch": 1.598358154296875e-06,
      "step": 1676,
      "training_step_time": 17.028318166732788
    },
    {
      "epoch": 1.5993118286132813e-06,
      "model_forward_time": 5.580373764038086,
      "step": 1677
    },
    {
      "epoch": 1.5993118286132813e-06,
      "step": 1677,
      "training_step_time": 16.994707345962524
    },
    {
      "epoch": 1.6002655029296874e-06,
      "model_forward_time": 5.5914084911346436,
      "step": 1678
    },
    {
      "epoch": 1.6002655029296874e-06,
      "step": 1678,
      "training_step_time": 17.04283118247986
    },
    {
      "epoch": 1.6012191772460938e-06,
      "model_forward_time": 5.963799715042114,
      "step": 1679
    },
    {
      "epoch": 1.6012191772460938e-06,
      "step": 1679,
      "training_step_time": 17.470052003860474
    },
    {
      "epoch": 1.6021728515625e-06,
      "grad_norm": 1.8809707164764404,
      "learning_rate": 1e-05,
      "loss": 0.0961,
      "step": 1680
    },
    {
      "epoch": 1.6021728515625e-06,
      "model_forward_time": 5.623195648193359,
      "step": 1680
    },
    {
      "dynamics_loss_avg": 0.0767152737826109,
      "epoch": 1.6021728515625e-06,
      "step": 1680
    },
    {
      "action_loss_avg": 0.02179887667298317,
      "epoch": 1.6021728515625e-06,
      "step": 1680
    },
    {
      "epoch": 1.6021728515625e-06,
      "step": 1680,
      "training_step_time": 17.15453338623047
    },
    {
      "epoch": 1.6031265258789063e-06,
      "model_forward_time": 5.625983953475952,
      "step": 1681
    },
    {
      "epoch": 1.6031265258789063e-06,
      "step": 1681,
      "training_step_time": 17.135517597198486
    },
    {
      "epoch": 1.6040802001953124e-06,
      "model_forward_time": 5.612828016281128,
      "step": 1682
    },
    {
      "epoch": 1.6040802001953124e-06,
      "step": 1682,
      "training_step_time": 17.09888005256653
    },
    {
      "epoch": 1.6050338745117188e-06,
      "model_forward_time": 5.607950925827026,
      "step": 1683
    },
    {
      "epoch": 1.6050338745117188e-06,
      "step": 1683,
      "training_step_time": 17.076131582260132
    },
    {
      "epoch": 1.605987548828125e-06,
      "model_forward_time": 5.597115755081177,
      "step": 1684
    },
    {
      "epoch": 1.605987548828125e-06,
      "step": 1684,
      "training_step_time": 17.414965629577637
    },
    {
      "epoch": 1.6069412231445313e-06,
      "model_forward_time": 5.599932432174683,
      "step": 1685
    },
    {
      "epoch": 1.6069412231445313e-06,
      "step": 1685,
      "training_step_time": 17.03732657432556
    },
    {
      "epoch": 1.6078948974609374e-06,
      "model_forward_time": 5.590217590332031,
      "step": 1686
    },
    {
      "epoch": 1.6078948974609374e-06,
      "step": 1686,
      "training_step_time": 17.02589511871338
    },
    {
      "epoch": 1.6088485717773438e-06,
      "model_forward_time": 5.587953567504883,
      "step": 1687
    },
    {
      "epoch": 1.6088485717773438e-06,
      "step": 1687,
      "training_step_time": 17.030598402023315
    },
    {
      "epoch": 1.60980224609375e-06,
      "model_forward_time": 5.5943615436553955,
      "step": 1688
    },
    {
      "epoch": 1.60980224609375e-06,
      "step": 1688,
      "training_step_time": 17.056992530822754
    },
    {
      "epoch": 1.6107559204101563e-06,
      "model_forward_time": 5.603490352630615,
      "step": 1689
    },
    {
      "epoch": 1.6107559204101563e-06,
      "step": 1689,
      "training_step_time": 17.050886154174805
    },
    {
      "epoch": 1.6117095947265624e-06,
      "grad_norm": 1.0051372051239014,
      "learning_rate": 1e-05,
      "loss": 0.0899,
      "step": 1690
    },
    {
      "epoch": 1.6117095947265624e-06,
      "model_forward_time": 5.949490547180176,
      "step": 1690
    },
    {
      "dynamics_loss_avg": 0.06352638117969037,
      "epoch": 1.6117095947265624e-06,
      "step": 1690
    },
    {
      "action_loss_avg": 0.020038618240505457,
      "epoch": 1.6117095947265624e-06,
      "step": 1690
    },
    {
      "epoch": 1.6117095947265624e-06,
      "step": 1690,
      "training_step_time": 17.377655744552612
    },
    {
      "epoch": 1.6126632690429688e-06,
      "model_forward_time": 5.595337152481079,
      "step": 1691
    },
    {
      "epoch": 1.6126632690429688e-06,
      "step": 1691,
      "training_step_time": 17.059609174728394
    },
    {
      "epoch": 1.613616943359375e-06,
      "model_forward_time": 5.601722717285156,
      "step": 1692
    },
    {
      "epoch": 1.613616943359375e-06,
      "step": 1692,
      "training_step_time": 17.079984188079834
    },
    {
      "epoch": 1.6145706176757813e-06,
      "model_forward_time": 5.609821796417236,
      "step": 1693
    },
    {
      "epoch": 1.6145706176757813e-06,
      "step": 1693,
      "training_step_time": 17.067850828170776
    },
    {
      "epoch": 1.6155242919921875e-06,
      "model_forward_time": 5.596774578094482,
      "step": 1694
    },
    {
      "epoch": 1.6155242919921875e-06,
      "step": 1694,
      "training_step_time": 17.057113647460938
    },
    {
      "epoch": 1.6164779663085938e-06,
      "model_forward_time": 5.602863550186157,
      "step": 1695
    },
    {
      "epoch": 1.6164779663085938e-06,
      "step": 1695,
      "training_step_time": 17.42920732498169
    },
    {
      "epoch": 1.617431640625e-06,
      "model_forward_time": 5.602030992507935,
      "step": 1696
    },
    {
      "epoch": 1.617431640625e-06,
      "step": 1696,
      "training_step_time": 17.094385623931885
    },
    {
      "epoch": 1.6183853149414063e-06,
      "model_forward_time": 5.605045318603516,
      "step": 1697
    },
    {
      "epoch": 1.6183853149414063e-06,
      "step": 1697,
      "training_step_time": 17.068292140960693
    },
    {
      "epoch": 1.6193389892578125e-06,
      "model_forward_time": 5.591639041900635,
      "step": 1698
    },
    {
      "epoch": 1.6193389892578125e-06,
      "step": 1698,
      "training_step_time": 17.029471397399902
    },
    {
      "epoch": 1.6202926635742188e-06,
      "model_forward_time": 5.593628644943237,
      "step": 1699
    },
    {
      "epoch": 1.6202926635742188e-06,
      "step": 1699,
      "training_step_time": 17.029062032699585
    },
    {
      "epoch": 1.621246337890625e-06,
      "grad_norm": 0.7605403661727905,
      "learning_rate": 1e-05,
      "loss": 0.0976,
      "step": 1700
    },
    {
      "epoch": 1.621246337890625e-06,
      "model_forward_time": 5.593194961547852,
      "step": 1700
    },
    {
      "dynamics_loss_avg": 0.0693306103348732,
      "epoch": 1.621246337890625e-06,
      "step": 1700
    },
    {
      "action_loss_avg": 0.0263607248198241,
      "epoch": 1.621246337890625e-06,
      "step": 1700
    },
    {
      "epoch": 1.621246337890625e-06,
      "step": 1700,
      "training_step_time": 17.05994486808777
    },
    {
      "epoch": 1.6222000122070313e-06,
      "model_forward_time": 5.966579437255859,
      "step": 1701
    },
    {
      "epoch": 1.6222000122070313e-06,
      "step": 1701,
      "training_step_time": 17.4210422039032
    },
    {
      "epoch": 1.6231536865234375e-06,
      "model_forward_time": 5.602120876312256,
      "step": 1702
    },
    {
      "epoch": 1.6231536865234375e-06,
      "step": 1702,
      "training_step_time": 17.104284048080444
    },
    {
      "epoch": 1.6241073608398438e-06,
      "model_forward_time": 5.609401226043701,
      "step": 1703
    },
    {
      "epoch": 1.6241073608398438e-06,
      "step": 1703,
      "training_step_time": 17.095159769058228
    },
    {
      "epoch": 1.62506103515625e-06,
      "model_forward_time": 5.599334716796875,
      "step": 1704
    },
    {
      "epoch": 1.62506103515625e-06,
      "step": 1704,
      "training_step_time": 17.025714635849
    },
    {
      "epoch": 1.6260147094726563e-06,
      "model_forward_time": 5.57249903678894,
      "step": 1705
    },
    {
      "epoch": 1.6260147094726563e-06,
      "step": 1705,
      "training_step_time": 17.00626301765442
    },
    {
      "epoch": 1.6269683837890625e-06,
      "model_forward_time": 5.9363038539886475,
      "step": 1706
    },
    {
      "epoch": 1.6269683837890625e-06,
      "step": 1706,
      "training_step_time": 17.364643096923828
    },
    {
      "epoch": 1.6279220581054688e-06,
      "model_forward_time": 5.578419923782349,
      "step": 1707
    },
    {
      "epoch": 1.6279220581054688e-06,
      "step": 1707,
      "training_step_time": 16.995291233062744
    },
    {
      "epoch": 1.628875732421875e-06,
      "model_forward_time": 5.5741071701049805,
      "step": 1708
    },
    {
      "epoch": 1.628875732421875e-06,
      "step": 1708,
      "training_step_time": 17.021183490753174
    },
    {
      "epoch": 1.6298294067382813e-06,
      "model_forward_time": 5.5829291343688965,
      "step": 1709
    },
    {
      "epoch": 1.6298294067382813e-06,
      "step": 1709,
      "training_step_time": 17.014289617538452
    },
    {
      "epoch": 1.6307830810546875e-06,
      "grad_norm": 2.2141525745391846,
      "learning_rate": 1e-05,
      "loss": 0.088,
      "step": 1710
    },
    {
      "epoch": 1.6307830810546875e-06,
      "model_forward_time": 5.584976434707642,
      "step": 1710
    },
    {
      "dynamics_loss_avg": 0.06599726174026728,
      "epoch": 1.6307830810546875e-06,
      "step": 1710
    },
    {
      "action_loss_avg": 0.022444836143404246,
      "epoch": 1.6307830810546875e-06,
      "step": 1710
    },
    {
      "epoch": 1.6307830810546875e-06,
      "step": 1710,
      "training_step_time": 17.023427486419678
    },
    {
      "epoch": 1.6317367553710938e-06,
      "model_forward_time": 5.601115942001343,
      "step": 1711
    },
    {
      "epoch": 1.6317367553710938e-06,
      "step": 1711,
      "training_step_time": 17.050837993621826
    },
    {
      "epoch": 1.6326904296875e-06,
      "model_forward_time": 5.963510036468506,
      "step": 1712
    },
    {
      "epoch": 1.6326904296875e-06,
      "step": 1712,
      "training_step_time": 17.438117265701294
    },
    {
      "epoch": 1.6336441040039063e-06,
      "model_forward_time": 5.606492757797241,
      "step": 1713
    },
    {
      "epoch": 1.6336441040039063e-06,
      "step": 1713,
      "training_step_time": 17.07390546798706
    },
    {
      "epoch": 1.6345977783203125e-06,
      "model_forward_time": 5.604060649871826,
      "step": 1714
    },
    {
      "epoch": 1.6345977783203125e-06,
      "step": 1714,
      "training_step_time": 17.053905725479126
    },
    {
      "epoch": 1.6355514526367188e-06,
      "model_forward_time": 5.59466290473938,
      "step": 1715
    },
    {
      "epoch": 1.6355514526367188e-06,
      "step": 1715,
      "training_step_time": 17.034667491912842
    },
    {
      "epoch": 1.636505126953125e-06,
      "model_forward_time": 5.596134424209595,
      "step": 1716
    },
    {
      "epoch": 1.636505126953125e-06,
      "step": 1716,
      "training_step_time": 17.047566413879395
    },
    {
      "epoch": 1.6374588012695313e-06,
      "model_forward_time": 5.585718631744385,
      "step": 1717
    },
    {
      "epoch": 1.6374588012695313e-06,
      "step": 1717,
      "training_step_time": 17.398985385894775
    },
    {
      "epoch": 1.6384124755859375e-06,
      "model_forward_time": 5.591270446777344,
      "step": 1718
    },
    {
      "epoch": 1.6384124755859375e-06,
      "step": 1718,
      "training_step_time": 17.049644947052002
    },
    {
      "epoch": 1.6393661499023438e-06,
      "model_forward_time": 5.611116647720337,
      "step": 1719
    },
    {
      "epoch": 1.6393661499023438e-06,
      "step": 1719,
      "training_step_time": 17.079305410385132
    },
    {
      "epoch": 1.64031982421875e-06,
      "grad_norm": 0.8628849387168884,
      "learning_rate": 1e-05,
      "loss": 0.0695,
      "step": 1720
    },
    {
      "epoch": 1.64031982421875e-06,
      "model_forward_time": 5.601449251174927,
      "step": 1720
    },
    {
      "dynamics_loss_avg": 0.05399707034230232,
      "epoch": 1.64031982421875e-06,
      "step": 1720
    },
    {
      "action_loss_avg": 0.01600180994719267,
      "epoch": 1.64031982421875e-06,
      "step": 1720
    },
    {
      "epoch": 1.64031982421875e-06,
      "step": 1720,
      "training_step_time": 17.047999620437622
    },
    {
      "epoch": 1.6412734985351563e-06,
      "model_forward_time": 5.59572172164917,
      "step": 1721
    },
    {
      "epoch": 1.6412734985351563e-06,
      "step": 1721,
      "training_step_time": 17.01934576034546
    },
    {
      "epoch": 1.6422271728515625e-06,
      "model_forward_time": 5.5898520946502686,
      "step": 1722
    },
    {
      "epoch": 1.6422271728515625e-06,
      "step": 1722,
      "training_step_time": 17.035580158233643
    },
    {
      "epoch": 1.6431808471679688e-06,
      "model_forward_time": 5.970623731613159,
      "step": 1723
    },
    {
      "epoch": 1.6431808471679688e-06,
      "step": 1723,
      "training_step_time": 17.468265533447266
    },
    {
      "epoch": 1.644134521484375e-06,
      "model_forward_time": 5.62896466255188,
      "step": 1724
    },
    {
      "epoch": 1.644134521484375e-06,
      "step": 1724,
      "training_step_time": 17.132044315338135
    },
    {
      "epoch": 1.6450881958007813e-06,
      "model_forward_time": 5.617481470108032,
      "step": 1725
    },
    {
      "epoch": 1.6450881958007813e-06,
      "step": 1725,
      "training_step_time": 17.107335329055786
    },
    {
      "epoch": 1.6460418701171875e-06,
      "model_forward_time": 5.604911804199219,
      "step": 1726
    },
    {
      "epoch": 1.6460418701171875e-06,
      "step": 1726,
      "training_step_time": 17.095114707946777
    },
    {
      "epoch": 1.6469955444335938e-06,
      "model_forward_time": 5.60874080657959,
      "step": 1727
    },
    {
      "epoch": 1.6469955444335938e-06,
      "step": 1727,
      "training_step_time": 17.09822368621826
    },
    {
      "epoch": 1.64794921875e-06,
      "model_forward_time": 5.612700462341309,
      "step": 1728
    },
    {
      "epoch": 1.64794921875e-06,
      "step": 1728,
      "training_step_time": 17.4467556476593
    },
    {
      "epoch": 1.6489028930664063e-06,
      "model_forward_time": 5.604080677032471,
      "step": 1729
    },
    {
      "epoch": 1.6489028930664063e-06,
      "step": 1729,
      "training_step_time": 17.089263916015625
    },
    {
      "epoch": 1.6498565673828125e-06,
      "grad_norm": 1.3074740171432495,
      "learning_rate": 1e-05,
      "loss": 0.0865,
      "step": 1730
    },
    {
      "epoch": 1.6498565673828125e-06,
      "model_forward_time": 5.603634357452393,
      "step": 1730
    },
    {
      "dynamics_loss_avg": 0.06655626390129328,
      "epoch": 1.6498565673828125e-06,
      "step": 1730
    },
    {
      "action_loss_avg": 0.02248655925504863,
      "epoch": 1.6498565673828125e-06,
      "step": 1730
    },
    {
      "epoch": 1.6498565673828125e-06,
      "step": 1730,
      "training_step_time": 17.09689164161682
    },
    {
      "epoch": 1.6508102416992188e-06,
      "model_forward_time": 5.614360094070435,
      "step": 1731
    },
    {
      "epoch": 1.6508102416992188e-06,
      "step": 1731,
      "training_step_time": 17.088440418243408
    },
    {
      "epoch": 1.651763916015625e-06,
      "model_forward_time": 5.608522653579712,
      "step": 1732
    },
    {
      "epoch": 1.651763916015625e-06,
      "step": 1732,
      "training_step_time": 17.07215666770935
    },
    {
      "epoch": 1.6527175903320313e-06,
      "model_forward_time": 5.600959300994873,
      "step": 1733
    },
    {
      "epoch": 1.6527175903320313e-06,
      "step": 1733,
      "training_step_time": 17.03770875930786
    },
    {
      "epoch": 1.6536712646484375e-06,
      "model_forward_time": 5.961915969848633,
      "step": 1734
    },
    {
      "epoch": 1.6536712646484375e-06,
      "step": 1734,
      "training_step_time": 17.39848256111145
    },
    {
      "epoch": 1.6546249389648438e-06,
      "model_forward_time": 5.585339069366455,
      "step": 1735
    },
    {
      "epoch": 1.6546249389648438e-06,
      "step": 1735,
      "training_step_time": 17.017143487930298
    },
    {
      "epoch": 1.65557861328125e-06,
      "model_forward_time": 5.591806411743164,
      "step": 1736
    },
    {
      "epoch": 1.65557861328125e-06,
      "step": 1736,
      "training_step_time": 17.065383911132812
    },
    {
      "epoch": 1.6565322875976563e-06,
      "model_forward_time": 5.609923362731934,
      "step": 1737
    },
    {
      "epoch": 1.6565322875976563e-06,
      "step": 1737,
      "training_step_time": 17.102948665618896
    },
    {
      "epoch": 1.6574859619140625e-06,
      "model_forward_time": 5.623699903488159,
      "step": 1738
    },
    {
      "epoch": 1.6574859619140625e-06,
      "step": 1738,
      "training_step_time": 17.10363745689392
    },
    {
      "epoch": 1.6584396362304688e-06,
      "model_forward_time": 5.957876205444336,
      "step": 1739
    },
    {
      "epoch": 1.6584396362304688e-06,
      "step": 1739,
      "training_step_time": 17.391021013259888
    },
    {
      "epoch": 1.659393310546875e-06,
      "grad_norm": 0.7143338918685913,
      "learning_rate": 1e-05,
      "loss": 0.0811,
      "step": 1740
    },
    {
      "epoch": 1.659393310546875e-06,
      "model_forward_time": 5.571341514587402,
      "step": 1740
    },
    {
      "dynamics_loss_avg": 0.06658409163355827,
      "epoch": 1.659393310546875e-06,
      "step": 1740
    },
    {
      "action_loss_avg": 0.015575116313993932,
      "epoch": 1.659393310546875e-06,
      "step": 1740
    },
    {
      "epoch": 1.659393310546875e-06,
      "step": 1740,
      "training_step_time": 16.99515724182129
    },
    {
      "epoch": 1.6603469848632813e-06,
      "model_forward_time": 5.574572801589966,
      "step": 1741
    },
    {
      "epoch": 1.6603469848632813e-06,
      "step": 1741,
      "training_step_time": 16.996033430099487
    },
    {
      "epoch": 1.6613006591796875e-06,
      "model_forward_time": 5.577419996261597,
      "step": 1742
    },
    {
      "epoch": 1.6613006591796875e-06,
      "step": 1742,
      "training_step_time": 16.989673376083374
    },
    {
      "epoch": 1.6622543334960938e-06,
      "model_forward_time": 5.578535556793213,
      "step": 1743
    },
    {
      "epoch": 1.6622543334960938e-06,
      "step": 1743,
      "training_step_time": 17.020681858062744
    },
    {
      "epoch": 1.6632080078125e-06,
      "model_forward_time": 5.604177236557007,
      "step": 1744
    },
    {
      "epoch": 1.6632080078125e-06,
      "step": 1744,
      "training_step_time": 17.079442501068115
    },
    {
      "epoch": 1.6641616821289064e-06,
      "model_forward_time": 5.997311115264893,
      "step": 1745
    },
    {
      "epoch": 1.6641616821289064e-06,
      "step": 1745,
      "training_step_time": 17.48399329185486
    },
    {
      "epoch": 1.6651153564453125e-06,
      "model_forward_time": 5.611245632171631,
      "step": 1746
    },
    {
      "epoch": 1.6651153564453125e-06,
      "step": 1746,
      "training_step_time": 17.096774578094482
    },
    {
      "epoch": 1.6660690307617189e-06,
      "model_forward_time": 5.617930173873901,
      "step": 1747
    },
    {
      "epoch": 1.6660690307617189e-06,
      "step": 1747,
      "training_step_time": 17.090731143951416
    },
    {
      "epoch": 1.667022705078125e-06,
      "model_forward_time": 5.618172645568848,
      "step": 1748
    },
    {
      "epoch": 1.667022705078125e-06,
      "step": 1748,
      "training_step_time": 17.131096363067627
    },
    {
      "epoch": 1.6679763793945314e-06,
      "model_forward_time": 5.625028371810913,
      "step": 1749
    },
    {
      "epoch": 1.6679763793945314e-06,
      "step": 1749,
      "training_step_time": 17.119765996932983
    },
    {
      "epoch": 1.6689300537109375e-06,
      "grad_norm": 1.33991277217865,
      "learning_rate": 1e-05,
      "loss": 0.0903,
      "step": 1750
    },
    {
      "epoch": 1.6689300537109375e-06,
      "model_forward_time": 5.600841760635376,
      "step": 1750
    },
    {
      "dynamics_loss_avg": 0.06824408955872059,
      "epoch": 1.6689300537109375e-06,
      "step": 1750
    },
    {
      "action_loss_avg": 0.02183255264535546,
      "epoch": 1.6689300537109375e-06,
      "step": 1750
    },
    {
      "epoch": 1.6689300537109375e-06,
      "step": 1750,
      "training_step_time": 17.421793937683105
    },
    {
      "epoch": 1.6698837280273436e-06,
      "model_forward_time": 5.615455389022827,
      "step": 1751
    },
    {
      "epoch": 1.6698837280273436e-06,
      "step": 1751,
      "training_step_time": 17.112565517425537
    },
    {
      "epoch": 1.67083740234375e-06,
      "model_forward_time": 5.615023136138916,
      "step": 1752
    },
    {
      "epoch": 1.67083740234375e-06,
      "step": 1752,
      "training_step_time": 17.11204957962036
    },
    {
      "epoch": 1.6717910766601561e-06,
      "model_forward_time": 5.613384485244751,
      "step": 1753
    },
    {
      "epoch": 1.6717910766601561e-06,
      "step": 1753,
      "training_step_time": 17.116649389266968
    },
    {
      "epoch": 1.6727447509765625e-06,
      "model_forward_time": 5.607910633087158,
      "step": 1754
    },
    {
      "epoch": 1.6727447509765625e-06,
      "step": 1754,
      "training_step_time": 17.06786346435547
    },
    {
      "epoch": 1.6736984252929686e-06,
      "model_forward_time": 5.598142385482788,
      "step": 1755
    },
    {
      "epoch": 1.6736984252929686e-06,
      "step": 1755,
      "training_step_time": 17.018463134765625
    },
    {
      "epoch": 1.674652099609375e-06,
      "model_forward_time": 5.970558166503906,
      "step": 1756
    },
    {
      "epoch": 1.674652099609375e-06,
      "step": 1756,
      "training_step_time": 17.388802528381348
    },
    {
      "epoch": 1.6756057739257812e-06,
      "model_forward_time": 5.573582410812378,
      "step": 1757
    },
    {
      "epoch": 1.6756057739257812e-06,
      "step": 1757,
      "training_step_time": 16.97891116142273
    },
    {
      "epoch": 1.6765594482421875e-06,
      "model_forward_time": 5.57223105430603,
      "step": 1758
    },
    {
      "epoch": 1.6765594482421875e-06,
      "step": 1758,
      "training_step_time": 17.01720690727234
    },
    {
      "epoch": 1.6775131225585937e-06,
      "model_forward_time": 5.57280158996582,
      "step": 1759
    },
    {
      "epoch": 1.6775131225585937e-06,
      "step": 1759,
      "training_step_time": 17.009948015213013
    },
    {
      "epoch": 1.678466796875e-06,
      "grad_norm": 0.6703776121139526,
      "learning_rate": 1e-05,
      "loss": 0.0814,
      "step": 1760
    },
    {
      "epoch": 1.678466796875e-06,
      "model_forward_time": 5.584303617477417,
      "step": 1760
    },
    {
      "dynamics_loss_avg": 0.059096505865454674,
      "epoch": 1.678466796875e-06,
      "step": 1760
    },
    {
      "action_loss_avg": 0.01497373627498746,
      "epoch": 1.678466796875e-06,
      "step": 1760
    },
    {
      "epoch": 1.678466796875e-06,
      "step": 1760,
      "training_step_time": 17.021750450134277
    },
    {
      "epoch": 1.6794204711914062e-06,
      "model_forward_time": 5.941830396652222,
      "step": 1761
    },
    {
      "epoch": 1.6794204711914062e-06,
      "step": 1761,
      "training_step_time": 17.37699007987976
    },
    {
      "epoch": 1.6803741455078125e-06,
      "model_forward_time": 5.586811780929565,
      "step": 1762
    },
    {
      "epoch": 1.6803741455078125e-06,
      "step": 1762,
      "training_step_time": 17.018118619918823
    },
    {
      "epoch": 1.6813278198242187e-06,
      "model_forward_time": 5.59801173210144,
      "step": 1763
    },
    {
      "epoch": 1.6813278198242187e-06,
      "step": 1763,
      "training_step_time": 17.07277774810791
    },
    {
      "epoch": 1.682281494140625e-06,
      "model_forward_time": 5.612043857574463,
      "step": 1764
    },
    {
      "epoch": 1.682281494140625e-06,
      "step": 1764,
      "training_step_time": 17.09117341041565
    },
    {
      "epoch": 1.6832351684570312e-06,
      "model_forward_time": 5.5982654094696045,
      "step": 1765
    },
    {
      "epoch": 1.6832351684570312e-06,
      "step": 1765,
      "training_step_time": 17.05667018890381
    },
    {
      "epoch": 1.6841888427734375e-06,
      "model_forward_time": 5.589184522628784,
      "step": 1766
    },
    {
      "epoch": 1.6841888427734375e-06,
      "step": 1766,
      "training_step_time": 17.0459942817688
    },
    {
      "epoch": 1.6851425170898437e-06,
      "model_forward_time": 5.974243879318237,
      "step": 1767
    },
    {
      "epoch": 1.6851425170898437e-06,
      "step": 1767,
      "training_step_time": 17.450865507125854
    },
    {
      "epoch": 1.68609619140625e-06,
      "model_forward_time": 5.608729600906372,
      "step": 1768
    },
    {
      "epoch": 1.68609619140625e-06,
      "step": 1768,
      "training_step_time": 17.088881731033325
    },
    {
      "epoch": 1.6870498657226562e-06,
      "model_forward_time": 5.632959365844727,
      "step": 1769
    },
    {
      "epoch": 1.6870498657226562e-06,
      "step": 1769,
      "training_step_time": 17.119455099105835
    },
    {
      "epoch": 1.6880035400390625e-06,
      "grad_norm": 1.1614792346954346,
      "learning_rate": 1e-05,
      "loss": 0.0754,
      "step": 1770
    },
    {
      "epoch": 1.6880035400390625e-06,
      "model_forward_time": 5.59824538230896,
      "step": 1770
    },
    {
      "dynamics_loss_avg": 0.061469847336411476,
      "epoch": 1.6880035400390625e-06,
      "step": 1770
    },
    {
      "action_loss_avg": 0.0212996409740299,
      "epoch": 1.6880035400390625e-06,
      "step": 1770
    },
    {
      "epoch": 1.6880035400390625e-06,
      "step": 1770,
      "training_step_time": 17.069809198379517
    },
    {
      "epoch": 1.6889572143554687e-06,
      "model_forward_time": 5.598830938339233,
      "step": 1771
    },
    {
      "epoch": 1.6889572143554687e-06,
      "step": 1771,
      "training_step_time": 17.050432205200195
    },
    {
      "epoch": 1.689910888671875e-06,
      "model_forward_time": 5.943494081497192,
      "step": 1772
    },
    {
      "epoch": 1.689910888671875e-06,
      "step": 1772,
      "training_step_time": 17.3768150806427
    },
    {
      "epoch": 1.6908645629882812e-06,
      "model_forward_time": 5.596898078918457,
      "step": 1773
    },
    {
      "epoch": 1.6908645629882812e-06,
      "step": 1773,
      "training_step_time": 17.058773040771484
    },
    {
      "epoch": 1.6918182373046875e-06,
      "model_forward_time": 5.614842176437378,
      "step": 1774
    },
    {
      "epoch": 1.6918182373046875e-06,
      "step": 1774,
      "training_step_time": 17.107551336288452
    },
    {
      "epoch": 1.6927719116210937e-06,
      "model_forward_time": 5.614161968231201,
      "step": 1775
    },
    {
      "epoch": 1.6927719116210937e-06,
      "step": 1775,
      "training_step_time": 17.109369039535522
    },
    {
      "epoch": 1.6937255859375e-06,
      "model_forward_time": 5.61501932144165,
      "step": 1776
    },
    {
      "epoch": 1.6937255859375e-06,
      "step": 1776,
      "training_step_time": 17.11139750480652
    },
    {
      "epoch": 1.6946792602539062e-06,
      "model_forward_time": 5.609270811080933,
      "step": 1777
    },
    {
      "epoch": 1.6946792602539062e-06,
      "step": 1777,
      "training_step_time": 17.464479684829712
    },
    {
      "epoch": 1.6956329345703125e-06,
      "model_forward_time": 5.612812280654907,
      "step": 1778
    },
    {
      "epoch": 1.6956329345703125e-06,
      "step": 1778,
      "training_step_time": 17.102391719818115
    },
    {
      "epoch": 1.6965866088867187e-06,
      "model_forward_time": 5.608808994293213,
      "step": 1779
    },
    {
      "epoch": 1.6965866088867187e-06,
      "step": 1779,
      "training_step_time": 17.08720350265503
    },
    {
      "epoch": 1.697540283203125e-06,
      "grad_norm": 0.8510586619377136,
      "learning_rate": 1e-05,
      "loss": 0.0793,
      "step": 1780
    },
    {
      "epoch": 1.697540283203125e-06,
      "model_forward_time": 5.594355344772339,
      "step": 1780
    },
    {
      "dynamics_loss_avg": 0.055099793151021,
      "epoch": 1.697540283203125e-06,
      "step": 1780
    },
    {
      "action_loss_avg": 0.016390569414943455,
      "epoch": 1.697540283203125e-06,
      "step": 1780
    },
    {
      "epoch": 1.697540283203125e-06,
      "step": 1780,
      "training_step_time": 17.060020446777344
    },
    {
      "epoch": 1.6984939575195312e-06,
      "model_forward_time": 5.602289199829102,
      "step": 1781
    },
    {
      "epoch": 1.6984939575195312e-06,
      "step": 1781,
      "training_step_time": 17.056174516677856
    },
    {
      "epoch": 1.6994476318359375e-06,
      "model_forward_time": 5.606018304824829,
      "step": 1782
    },
    {
      "epoch": 1.6994476318359375e-06,
      "step": 1782,
      "training_step_time": 17.066564798355103
    },
    {
      "epoch": 1.7004013061523437e-06,
      "model_forward_time": 5.953035116195679,
      "step": 1783
    },
    {
      "epoch": 1.7004013061523437e-06,
      "step": 1783,
      "training_step_time": 17.407918691635132
    },
    {
      "epoch": 1.70135498046875e-06,
      "model_forward_time": 5.592679500579834,
      "step": 1784
    },
    {
      "epoch": 1.70135498046875e-06,
      "step": 1784,
      "training_step_time": 17.043281078338623
    },
    {
      "epoch": 1.7023086547851562e-06,
      "model_forward_time": 5.585180044174194,
      "step": 1785
    },
    {
      "epoch": 1.7023086547851562e-06,
      "step": 1785,
      "training_step_time": 17.03566002845764
    },
    {
      "epoch": 1.7032623291015625e-06,
      "model_forward_time": 5.598536491394043,
      "step": 1786
    },
    {
      "epoch": 1.7032623291015625e-06,
      "step": 1786,
      "training_step_time": 17.064470052719116
    },
    {
      "epoch": 1.7042160034179687e-06,
      "model_forward_time": 5.603134632110596,
      "step": 1787
    },
    {
      "epoch": 1.7042160034179687e-06,
      "step": 1787,
      "training_step_time": 17.09479832649231
    },
    {
      "epoch": 1.705169677734375e-06,
      "model_forward_time": 5.613045930862427,
      "step": 1788
    },
    {
      "epoch": 1.705169677734375e-06,
      "step": 1788,
      "training_step_time": 17.455244779586792
    },
    {
      "epoch": 1.7061233520507812e-06,
      "model_forward_time": 5.610759973526001,
      "step": 1789
    },
    {
      "epoch": 1.7061233520507812e-06,
      "step": 1789,
      "training_step_time": 17.093408346176147
    },
    {
      "epoch": 1.7070770263671875e-06,
      "grad_norm": 0.9729554653167725,
      "learning_rate": 1e-05,
      "loss": 0.0648,
      "step": 1790
    },
    {
      "epoch": 1.7070770263671875e-06,
      "model_forward_time": 5.5992231369018555,
      "step": 1790
    },
    {
      "dynamics_loss_avg": 0.05151410289108753,
      "epoch": 1.7070770263671875e-06,
      "step": 1790
    },
    {
      "action_loss_avg": 0.013029574463143945,
      "epoch": 1.7070770263671875e-06,
      "step": 1790
    },
    {
      "epoch": 1.7070770263671875e-06,
      "step": 1790,
      "training_step_time": 17.000130653381348
    },
    {
      "epoch": 1.7080307006835937e-06,
      "model_forward_time": 5.588952302932739,
      "step": 1791
    },
    {
      "epoch": 1.7080307006835937e-06,
      "step": 1791,
      "training_step_time": 16.99745750427246
    },
    {
      "epoch": 1.708984375e-06,
      "model_forward_time": 5.578047275543213,
      "step": 1792
    },
    {
      "epoch": 1.708984375e-06,
      "step": 1792,
      "training_step_time": 17.01024580001831
    },
    {
      "epoch": 1.7099380493164062e-06,
      "model_forward_time": 5.567260980606079,
      "step": 1793
    },
    {
      "epoch": 1.7099380493164062e-06,
      "step": 1793,
      "training_step_time": 17.0080623626709
    },
    {
      "epoch": 1.7108917236328125e-06,
      "model_forward_time": 5.9134862422943115,
      "step": 1794
    },
    {
      "epoch": 1.7108917236328125e-06,
      "step": 1794,
      "training_step_time": 17.315610647201538
    },
    {
      "epoch": 1.7118453979492187e-06,
      "model_forward_time": 5.581900596618652,
      "step": 1795
    },
    {
      "epoch": 1.7118453979492187e-06,
      "step": 1795,
      "training_step_time": 17.01896619796753
    },
    {
      "epoch": 1.712799072265625e-06,
      "model_forward_time": 5.593508243560791,
      "step": 1796
    },
    {
      "epoch": 1.712799072265625e-06,
      "step": 1796,
      "training_step_time": 17.036571741104126
    },
    {
      "epoch": 1.7137527465820312e-06,
      "model_forward_time": 5.600517272949219,
      "step": 1797
    },
    {
      "epoch": 1.7137527465820312e-06,
      "step": 1797,
      "training_step_time": 17.082787036895752
    },
    {
      "epoch": 1.7147064208984375e-06,
      "model_forward_time": 5.613354444503784,
      "step": 1798
    },
    {
      "epoch": 1.7147064208984375e-06,
      "step": 1798,
      "training_step_time": 17.108443021774292
    },
    {
      "epoch": 1.7156600952148437e-06,
      "model_forward_time": 5.978383779525757,
      "step": 1799
    },
    {
      "epoch": 1.7156600952148437e-06,
      "step": 1799,
      "training_step_time": 17.47050189971924
    },
    {
      "epoch": 1.71661376953125e-06,
      "grad_norm": 1.4385145902633667,
      "learning_rate": 1e-05,
      "loss": 0.0708,
      "step": 1800
    },
    {
      "epoch": 1.71661376953125e-06,
      "model_forward_time": 5.612359285354614,
      "step": 1800
    },
    {
      "dynamics_loss_avg": 0.057386955805122854,
      "epoch": 1.71661376953125e-06,
      "step": 1800
    },
    {
      "action_loss_avg": 0.01614668683614582,
      "epoch": 1.71661376953125e-06,
      "step": 1800
    },
    {
      "epoch": 1.71661376953125e-06,
      "step": 1800,
      "training_step_time": 17.099712133407593
    },
    {
      "epoch": 1.7175674438476562e-06,
      "model_forward_time": 5.614821434020996,
      "step": 1801
    },
    {
      "epoch": 1.7175674438476562e-06,
      "step": 1801,
      "training_step_time": 17.11040472984314
    },
    {
      "epoch": 1.7185211181640625e-06,
      "model_forward_time": 5.605914354324341,
      "step": 1802
    },
    {
      "epoch": 1.7185211181640625e-06,
      "step": 1802,
      "training_step_time": 17.09858727455139
    },
    {
      "epoch": 1.7194747924804687e-06,
      "model_forward_time": 5.6112141609191895,
      "step": 1803
    },
    {
      "epoch": 1.7194747924804687e-06,
      "step": 1803,
      "training_step_time": 17.0918447971344
    },
    {
      "epoch": 1.720428466796875e-06,
      "model_forward_time": 5.607925891876221,
      "step": 1804
    },
    {
      "epoch": 1.720428466796875e-06,
      "step": 1804,
      "training_step_time": 17.455123901367188
    },
    {
      "epoch": 1.7213821411132812e-06,
      "model_forward_time": 5.610820293426514,
      "step": 1805
    },
    {
      "epoch": 1.7213821411132812e-06,
      "step": 1805,
      "training_step_time": 17.08763027191162
    },
    {
      "epoch": 1.7223358154296875e-06,
      "model_forward_time": 5.596380233764648,
      "step": 1806
    },
    {
      "epoch": 1.7223358154296875e-06,
      "step": 1806,
      "training_step_time": 17.04222559928894
    },
    {
      "epoch": 1.7232894897460937e-06,
      "model_forward_time": 5.59267258644104,
      "step": 1807
    },
    {
      "epoch": 1.7232894897460937e-06,
      "step": 1807,
      "training_step_time": 17.01788830757141
    },
    {
      "epoch": 1.7242431640625e-06,
      "model_forward_time": 5.594836235046387,
      "step": 1808
    },
    {
      "epoch": 1.7242431640625e-06,
      "step": 1808,
      "training_step_time": 17.05057382583618
    },
    {
      "epoch": 1.7251968383789062e-06,
      "model_forward_time": 5.596137046813965,
      "step": 1809
    },
    {
      "epoch": 1.7251968383789062e-06,
      "step": 1809,
      "training_step_time": 17.084266424179077
    },
    {
      "epoch": 1.7261505126953126e-06,
      "grad_norm": 1.073880672454834,
      "learning_rate": 1e-05,
      "loss": 0.0892,
      "step": 1810
    },
    {
      "epoch": 1.7261505126953126e-06,
      "model_forward_time": 5.960195302963257,
      "step": 1810
    },
    {
      "dynamics_loss_avg": 0.07150448746979236,
      "epoch": 1.7261505126953126e-06,
      "step": 1810
    },
    {
      "action_loss_avg": 0.02055705259554088,
      "epoch": 1.7261505126953126e-06,
      "step": 1810
    },
    {
      "epoch": 1.7261505126953126e-06,
      "step": 1810,
      "training_step_time": 17.471601009368896
    },
    {
      "epoch": 1.7271041870117187e-06,
      "model_forward_time": 5.609716415405273,
      "step": 1811
    },
    {
      "epoch": 1.7271041870117187e-06,
      "step": 1811,
      "training_step_time": 17.110017776489258
    },
    {
      "epoch": 1.728057861328125e-06,
      "model_forward_time": 5.620495080947876,
      "step": 1812
    },
    {
      "epoch": 1.728057861328125e-06,
      "step": 1812,
      "training_step_time": 17.134215831756592
    },
    {
      "epoch": 1.7290115356445312e-06,
      "model_forward_time": 5.617928981781006,
      "step": 1813
    },
    {
      "epoch": 1.7290115356445312e-06,
      "step": 1813,
      "training_step_time": 17.11211585998535
    },
    {
      "epoch": 1.7299652099609376e-06,
      "model_forward_time": 5.616857051849365,
      "step": 1814
    },
    {
      "epoch": 1.7299652099609376e-06,
      "step": 1814,
      "training_step_time": 17.112418174743652
    },
    {
      "epoch": 1.7309188842773437e-06,
      "model_forward_time": 5.965937614440918,
      "step": 1815
    },
    {
      "epoch": 1.7309188842773437e-06,
      "step": 1815,
      "training_step_time": 17.448102474212646
    },
    {
      "epoch": 1.73187255859375e-06,
      "model_forward_time": 5.60065221786499,
      "step": 1816
    },
    {
      "epoch": 1.73187255859375e-06,
      "step": 1816,
      "training_step_time": 17.072284936904907
    },
    {
      "epoch": 1.7328262329101562e-06,
      "model_forward_time": 5.601800203323364,
      "step": 1817
    },
    {
      "epoch": 1.7328262329101562e-06,
      "step": 1817,
      "training_step_time": 17.081230401992798
    },
    {
      "epoch": 1.7337799072265626e-06,
      "model_forward_time": 5.599191188812256,
      "step": 1818
    },
    {
      "epoch": 1.7337799072265626e-06,
      "step": 1818,
      "training_step_time": 17.063176155090332
    },
    {
      "epoch": 1.7347335815429687e-06,
      "model_forward_time": 5.598146438598633,
      "step": 1819
    },
    {
      "epoch": 1.7347335815429687e-06,
      "step": 1819,
      "training_step_time": 17.045579433441162
    },
    {
      "epoch": 1.735687255859375e-06,
      "grad_norm": 1.2943538427352905,
      "learning_rate": 1e-05,
      "loss": 0.0894,
      "step": 1820
    },
    {
      "epoch": 1.735687255859375e-06,
      "model_forward_time": 5.58268404006958,
      "step": 1820
    },
    {
      "dynamics_loss_avg": 0.061531894654035565,
      "epoch": 1.735687255859375e-06,
      "step": 1820
    },
    {
      "action_loss_avg": 0.021931294444948436,
      "epoch": 1.735687255859375e-06,
      "step": 1820
    },
    {
      "epoch": 1.735687255859375e-06,
      "step": 1820,
      "training_step_time": 17.033211946487427
    },
    {
      "epoch": 1.7366409301757812e-06,
      "model_forward_time": 5.956955671310425,
      "step": 1821
    },
    {
      "epoch": 1.7366409301757812e-06,
      "step": 1821,
      "training_step_time": 17.421348810195923
    },
    {
      "epoch": 1.7375946044921876e-06,
      "model_forward_time": 5.607226848602295,
      "step": 1822
    },
    {
      "epoch": 1.7375946044921876e-06,
      "step": 1822,
      "training_step_time": 17.093949556350708
    },
    {
      "epoch": 1.7385482788085937e-06,
      "model_forward_time": 5.608392000198364,
      "step": 1823
    },
    {
      "epoch": 1.7385482788085937e-06,
      "step": 1823,
      "training_step_time": 17.10217308998108
    },
    {
      "epoch": 1.739501953125e-06,
      "model_forward_time": 5.614684820175171,
      "step": 1824
    },
    {
      "epoch": 1.739501953125e-06,
      "step": 1824,
      "training_step_time": 17.09123706817627
    },
    {
      "epoch": 1.7404556274414062e-06,
      "model_forward_time": 5.591387987136841,
      "step": 1825
    },
    {
      "epoch": 1.7404556274414062e-06,
      "step": 1825,
      "training_step_time": 17.002887725830078
    },
    {
      "epoch": 1.7414093017578126e-06,
      "model_forward_time": 5.931191921234131,
      "step": 1826
    },
    {
      "epoch": 1.7414093017578126e-06,
      "step": 1826,
      "training_step_time": 17.372801303863525
    },
    {
      "epoch": 1.7423629760742187e-06,
      "model_forward_time": 5.594919681549072,
      "step": 1827
    },
    {
      "epoch": 1.7423629760742187e-06,
      "step": 1827,
      "training_step_time": 17.02131676673889
    },
    {
      "epoch": 1.743316650390625e-06,
      "model_forward_time": 5.588167428970337,
      "step": 1828
    },
    {
      "epoch": 1.743316650390625e-06,
      "step": 1828,
      "training_step_time": 17.043860912322998
    },
    {
      "epoch": 1.7442703247070312e-06,
      "model_forward_time": 5.595597505569458,
      "step": 1829
    },
    {
      "epoch": 1.7442703247070312e-06,
      "step": 1829,
      "training_step_time": 17.0672824382782
    },
    {
      "epoch": 1.7452239990234376e-06,
      "grad_norm": 1.765832543373108,
      "learning_rate": 1e-05,
      "loss": 0.0914,
      "step": 1830
    },
    {
      "epoch": 1.7452239990234376e-06,
      "model_forward_time": 5.6147685050964355,
      "step": 1830
    },
    {
      "dynamics_loss_avg": 0.07954539321362972,
      "epoch": 1.7452239990234376e-06,
      "step": 1830
    },
    {
      "action_loss_avg": 0.022583203576505183,
      "epoch": 1.7452239990234376e-06,
      "step": 1830
    },
    {
      "epoch": 1.7452239990234376e-06,
      "step": 1830,
      "training_step_time": 17.107875108718872
    },
    {
      "epoch": 1.7461776733398437e-06,
      "model_forward_time": 5.6200432777404785,
      "step": 1831
    },
    {
      "epoch": 1.7461776733398437e-06,
      "step": 1831,
      "training_step_time": 17.12244987487793
    },
    {
      "epoch": 1.74713134765625e-06,
      "model_forward_time": 5.975140333175659,
      "step": 1832
    },
    {
      "epoch": 1.74713134765625e-06,
      "step": 1832,
      "training_step_time": 17.478841066360474
    },
    {
      "epoch": 1.7480850219726562e-06,
      "model_forward_time": 5.619770526885986,
      "step": 1833
    },
    {
      "epoch": 1.7480850219726562e-06,
      "step": 1833,
      "training_step_time": 17.127429485321045
    },
    {
      "epoch": 1.7490386962890626e-06,
      "model_forward_time": 5.605296611785889,
      "step": 1834
    },
    {
      "epoch": 1.7490386962890626e-06,
      "step": 1834,
      "training_step_time": 17.093416452407837
    },
    {
      "epoch": 1.7499923706054687e-06,
      "model_forward_time": 5.607513189315796,
      "step": 1835
    },
    {
      "epoch": 1.7499923706054687e-06,
      "step": 1835,
      "training_step_time": 17.07991099357605
    },
    {
      "epoch": 1.750946044921875e-06,
      "model_forward_time": 5.607096433639526,
      "step": 1836
    },
    {
      "epoch": 1.750946044921875e-06,
      "step": 1836,
      "training_step_time": 17.085740566253662
    },
    {
      "epoch": 1.7518997192382812e-06,
      "model_forward_time": 5.942755699157715,
      "step": 1837
    },
    {
      "epoch": 1.7518997192382812e-06,
      "step": 1837,
      "training_step_time": 17.40743374824524
    },
    {
      "epoch": 1.7528533935546876e-06,
      "model_forward_time": 5.5945892333984375,
      "step": 1838
    },
    {
      "epoch": 1.7528533935546876e-06,
      "step": 1838,
      "training_step_time": 17.027564525604248
    },
    {
      "epoch": 1.7538070678710937e-06,
      "model_forward_time": 5.587900161743164,
      "step": 1839
    },
    {
      "epoch": 1.7538070678710937e-06,
      "step": 1839,
      "training_step_time": 17.023133277893066
    },
    {
      "epoch": 1.7547607421875e-06,
      "grad_norm": 1.1116477251052856,
      "learning_rate": 1e-05,
      "loss": 0.0944,
      "step": 1840
    },
    {
      "epoch": 1.7547607421875e-06,
      "model_forward_time": 5.598162651062012,
      "step": 1840
    },
    {
      "dynamics_loss_avg": 0.06757661178708077,
      "epoch": 1.7547607421875e-06,
      "step": 1840
    },
    {
      "action_loss_avg": 0.02332581374794245,
      "epoch": 1.7547607421875e-06,
      "step": 1840
    },
    {
      "epoch": 1.7547607421875e-06,
      "step": 1840,
      "training_step_time": 17.058037042617798
    },
    {
      "epoch": 1.7557144165039062e-06,
      "model_forward_time": 5.611665487289429,
      "step": 1841
    },
    {
      "epoch": 1.7557144165039062e-06,
      "step": 1841,
      "training_step_time": 17.09143567085266
    },
    {
      "epoch": 1.7566680908203126e-06,
      "model_forward_time": 5.610542058944702,
      "step": 1842
    },
    {
      "epoch": 1.7566680908203126e-06,
      "step": 1842,
      "training_step_time": 17.421735048294067
    },
    {
      "epoch": 1.7576217651367187e-06,
      "model_forward_time": 5.596369981765747,
      "step": 1843
    },
    {
      "epoch": 1.7576217651367187e-06,
      "step": 1843,
      "training_step_time": 17.075395345687866
    },
    {
      "epoch": 1.758575439453125e-06,
      "model_forward_time": 5.606994867324829,
      "step": 1844
    },
    {
      "epoch": 1.758575439453125e-06,
      "step": 1844,
      "training_step_time": 17.05257773399353
    },
    {
      "epoch": 1.7595291137695312e-06,
      "model_forward_time": 5.607952356338501,
      "step": 1845
    },
    {
      "epoch": 1.7595291137695312e-06,
      "step": 1845,
      "training_step_time": 17.10387873649597
    },
    {
      "epoch": 1.7604827880859376e-06,
      "model_forward_time": 5.607506513595581,
      "step": 1846
    },
    {
      "epoch": 1.7604827880859376e-06,
      "step": 1846,
      "training_step_time": 17.105815410614014
    },
    {
      "epoch": 1.7614364624023437e-06,
      "model_forward_time": 5.61041522026062,
      "step": 1847
    },
    {
      "epoch": 1.7614364624023437e-06,
      "step": 1847,
      "training_step_time": 17.117339372634888
    },
    {
      "epoch": 1.76239013671875e-06,
      "model_forward_time": 5.952373504638672,
      "step": 1848
    },
    {
      "epoch": 1.76239013671875e-06,
      "step": 1848,
      "training_step_time": 17.441019296646118
    },
    {
      "epoch": 1.7633438110351562e-06,
      "model_forward_time": 5.601345539093018,
      "step": 1849
    },
    {
      "epoch": 1.7633438110351562e-06,
      "step": 1849,
      "training_step_time": 17.08250641822815
    },
    {
      "epoch": 1.7642974853515626e-06,
      "grad_norm": 0.5006253719329834,
      "learning_rate": 1e-05,
      "loss": 0.0716,
      "step": 1850
    },
    {
      "epoch": 1.7642974853515626e-06,
      "model_forward_time": 5.612658977508545,
      "step": 1850
    },
    {
      "dynamics_loss_avg": 0.05621335636824369,
      "epoch": 1.7642974853515626e-06,
      "step": 1850
    },
    {
      "action_loss_avg": 0.015394182270392775,
      "epoch": 1.7642974853515626e-06,
      "step": 1850
    },
    {
      "epoch": 1.7642974853515626e-06,
      "step": 1850,
      "training_step_time": 17.09920573234558
    },
    {
      "epoch": 1.7652511596679687e-06,
      "model_forward_time": 5.602306842803955,
      "step": 1851
    },
    {
      "epoch": 1.7652511596679687e-06,
      "step": 1851,
      "training_step_time": 17.06943702697754
    },
    {
      "epoch": 1.766204833984375e-06,
      "model_forward_time": 5.594723224639893,
      "step": 1852
    },
    {
      "epoch": 1.766204833984375e-06,
      "step": 1852,
      "training_step_time": 17.03304171562195
    },
    {
      "epoch": 1.7671585083007812e-06,
      "model_forward_time": 5.96907377243042,
      "step": 1853
    },
    {
      "epoch": 1.7671585083007812e-06,
      "step": 1853,
      "training_step_time": 17.399317741394043
    },
    {
      "epoch": 1.7681121826171876e-06,
      "model_forward_time": 5.589099407196045,
      "step": 1854
    },
    {
      "epoch": 1.7681121826171876e-06,
      "step": 1854,
      "training_step_time": 17.042335748672485
    },
    {
      "epoch": 1.7690658569335937e-06,
      "model_forward_time": 5.5965282917022705,
      "step": 1855
    },
    {
      "epoch": 1.7690658569335937e-06,
      "step": 1855,
      "training_step_time": 17.057570219039917
    },
    {
      "epoch": 1.77001953125e-06,
      "model_forward_time": 5.614925861358643,
      "step": 1856
    },
    {
      "epoch": 1.77001953125e-06,
      "step": 1856,
      "training_step_time": 17.09080481529236
    },
    {
      "epoch": 1.7709732055664062e-06,
      "model_forward_time": 5.605602264404297,
      "step": 1857
    },
    {
      "epoch": 1.7709732055664062e-06,
      "step": 1857,
      "training_step_time": 17.07326579093933
    },
    {
      "epoch": 1.7719268798828126e-06,
      "model_forward_time": 5.602914810180664,
      "step": 1858
    },
    {
      "epoch": 1.7719268798828126e-06,
      "step": 1858,
      "training_step_time": 17.05653405189514
    },
    {
      "epoch": 1.7728805541992187e-06,
      "model_forward_time": 5.942248582839966,
      "step": 1859
    },
    {
      "epoch": 1.7728805541992187e-06,
      "step": 1859,
      "training_step_time": 17.393877029418945
    },
    {
      "epoch": 1.773834228515625e-06,
      "grad_norm": 0.2709331810474396,
      "learning_rate": 1e-05,
      "loss": 0.0786,
      "step": 1860
    },
    {
      "epoch": 1.773834228515625e-06,
      "model_forward_time": 5.59619927406311,
      "step": 1860
    },
    {
      "dynamics_loss_avg": 0.060475196689367294,
      "epoch": 1.773834228515625e-06,
      "step": 1860
    },
    {
      "action_loss_avg": 0.013902868796139955,
      "epoch": 1.773834228515625e-06,
      "step": 1860
    },
    {
      "epoch": 1.773834228515625e-06,
      "step": 1860,
      "training_step_time": 17.05353045463562
    },
    {
      "epoch": 1.7747879028320312e-06,
      "model_forward_time": 5.603753089904785,
      "step": 1861
    },
    {
      "epoch": 1.7747879028320312e-06,
      "step": 1861,
      "training_step_time": 17.06919765472412
    },
    {
      "epoch": 1.7757415771484376e-06,
      "model_forward_time": 5.6052329540252686,
      "step": 1862
    },
    {
      "epoch": 1.7757415771484376e-06,
      "step": 1862,
      "training_step_time": 17.101807355880737
    },
    {
      "epoch": 1.7766952514648437e-06,
      "model_forward_time": 5.610049247741699,
      "step": 1863
    },
    {
      "epoch": 1.7766952514648437e-06,
      "step": 1863,
      "training_step_time": 17.097548723220825
    },
    {
      "epoch": 1.77764892578125e-06,
      "model_forward_time": 5.981693744659424,
      "step": 1864
    },
    {
      "epoch": 1.77764892578125e-06,
      "step": 1864,
      "training_step_time": 17.451581478118896
    },
    {
      "epoch": 1.7786026000976562e-06,
      "model_forward_time": 5.604621171951294,
      "step": 1865
    },
    {
      "epoch": 1.7786026000976562e-06,
      "step": 1865,
      "training_step_time": 17.07542610168457
    },
    {
      "epoch": 1.7795562744140626e-06,
      "model_forward_time": 5.601081371307373,
      "step": 1866
    },
    {
      "epoch": 1.7795562744140626e-06,
      "step": 1866,
      "training_step_time": 17.066195011138916
    },
    {
      "epoch": 1.7805099487304687e-06,
      "model_forward_time": 5.606940031051636,
      "step": 1867
    },
    {
      "epoch": 1.7805099487304687e-06,
      "step": 1867,
      "training_step_time": 17.055760383605957
    },
    {
      "epoch": 1.781463623046875e-06,
      "model_forward_time": 5.589433193206787,
      "step": 1868
    },
    {
      "epoch": 1.781463623046875e-06,
      "step": 1868,
      "training_step_time": 17.035391330718994
    },
    {
      "epoch": 1.7824172973632812e-06,
      "model_forward_time": 5.601453065872192,
      "step": 1869
    },
    {
      "epoch": 1.7824172973632812e-06,
      "step": 1869,
      "training_step_time": 17.061622858047485
    },
    {
      "epoch": 1.7833709716796876e-06,
      "grad_norm": 1.585254430770874,
      "learning_rate": 1e-05,
      "loss": 0.0885,
      "step": 1870
    },
    {
      "epoch": 1.7833709716796876e-06,
      "model_forward_time": 5.946524381637573,
      "step": 1870
    },
    {
      "dynamics_loss_avg": 0.06865072194486857,
      "epoch": 1.7833709716796876e-06,
      "step": 1870
    },
    {
      "action_loss_avg": 0.020068002527114003,
      "epoch": 1.7833709716796876e-06,
      "step": 1870
    },
    {
      "epoch": 1.7833709716796876e-06,
      "step": 1870,
      "training_step_time": 17.448514223098755
    },
    {
      "epoch": 1.7843246459960937e-06,
      "model_forward_time": 5.631012201309204,
      "step": 1871
    },
    {
      "epoch": 1.7843246459960937e-06,
      "step": 1871,
      "training_step_time": 17.15599012374878
    },
    {
      "epoch": 1.7852783203125e-06,
      "model_forward_time": 5.629082202911377,
      "step": 1872
    },
    {
      "epoch": 1.7852783203125e-06,
      "step": 1872,
      "training_step_time": 17.112623691558838
    },
    {
      "epoch": 1.7862319946289062e-06,
      "model_forward_time": 5.5952630043029785,
      "step": 1873
    },
    {
      "epoch": 1.7862319946289062e-06,
      "step": 1873,
      "training_step_time": 17.01616406440735
    },
    {
      "epoch": 1.7871856689453126e-06,
      "model_forward_time": 5.585816144943237,
      "step": 1874
    },
    {
      "epoch": 1.7871856689453126e-06,
      "step": 1874,
      "training_step_time": 17.011850118637085
    },
    {
      "epoch": 1.7881393432617188e-06,
      "model_forward_time": 5.96883749961853,
      "step": 1875
    },
    {
      "epoch": 1.7881393432617188e-06,
      "step": 1875,
      "training_step_time": 17.40855574607849
    },
    {
      "epoch": 1.789093017578125e-06,
      "model_forward_time": 5.605601072311401,
      "step": 1876
    },
    {
      "epoch": 1.789093017578125e-06,
      "step": 1876,
      "training_step_time": 17.086543798446655
    },
    {
      "epoch": 1.7900466918945313e-06,
      "model_forward_time": 5.600637674331665,
      "step": 1877
    },
    {
      "epoch": 1.7900466918945313e-06,
      "step": 1877,
      "training_step_time": 17.084028005599976
    },
    {
      "epoch": 1.7910003662109374e-06,
      "model_forward_time": 5.598837375640869,
      "step": 1878
    },
    {
      "epoch": 1.7910003662109374e-06,
      "step": 1878,
      "training_step_time": 17.05146551132202
    },
    {
      "epoch": 1.7919540405273438e-06,
      "model_forward_time": 5.5941362380981445,
      "step": 1879
    },
    {
      "epoch": 1.7919540405273438e-06,
      "step": 1879,
      "training_step_time": 17.02311897277832
    },
    {
      "epoch": 1.79290771484375e-06,
      "grad_norm": 0.35435232520103455,
      "learning_rate": 1e-05,
      "loss": 0.0731,
      "step": 1880
    },
    {
      "epoch": 1.79290771484375e-06,
      "model_forward_time": 5.584429979324341,
      "step": 1880
    },
    {
      "dynamics_loss_avg": 0.05716000813990831,
      "epoch": 1.79290771484375e-06,
      "step": 1880
    },
    {
      "action_loss_avg": 0.01900853170081973,
      "epoch": 1.79290771484375e-06,
      "step": 1880
    },
    {
      "epoch": 1.79290771484375e-06,
      "step": 1880,
      "training_step_time": 17.034268140792847
    },
    {
      "epoch": 1.7938613891601563e-06,
      "model_forward_time": 5.935553550720215,
      "step": 1881
    },
    {
      "epoch": 1.7938613891601563e-06,
      "step": 1881,
      "training_step_time": 17.39536714553833
    },
    {
      "epoch": 1.7948150634765624e-06,
      "model_forward_time": 5.597761392593384,
      "step": 1882
    },
    {
      "epoch": 1.7948150634765624e-06,
      "step": 1882,
      "training_step_time": 17.055864572525024
    },
    {
      "epoch": 1.7957687377929688e-06,
      "model_forward_time": 5.606854200363159,
      "step": 1883
    },
    {
      "epoch": 1.7957687377929688e-06,
      "step": 1883,
      "training_step_time": 17.0986065864563
    },
    {
      "epoch": 1.796722412109375e-06,
      "model_forward_time": 5.614740371704102,
      "step": 1884
    },
    {
      "epoch": 1.796722412109375e-06,
      "step": 1884,
      "training_step_time": 17.09622883796692
    },
    {
      "epoch": 1.7976760864257813e-06,
      "model_forward_time": 5.5986974239349365,
      "step": 1885
    },
    {
      "epoch": 1.7976760864257813e-06,
      "step": 1885,
      "training_step_time": 17.07530689239502
    },
    {
      "epoch": 1.7986297607421874e-06,
      "model_forward_time": 5.974087715148926,
      "step": 1886
    },
    {
      "epoch": 1.7986297607421874e-06,
      "step": 1886,
      "training_step_time": 17.43216586112976
    },
    {
      "epoch": 1.7995834350585938e-06,
      "model_forward_time": 5.607853412628174,
      "step": 1887
    },
    {
      "epoch": 1.7995834350585938e-06,
      "step": 1887,
      "training_step_time": 17.073315382003784
    },
    {
      "epoch": 1.800537109375e-06,
      "model_forward_time": 5.611284971237183,
      "step": 1888
    },
    {
      "epoch": 1.800537109375e-06,
      "step": 1888,
      "training_step_time": 17.10086965560913
    },
    {
      "epoch": 1.8014907836914063e-06,
      "model_forward_time": 5.611493349075317,
      "step": 1889
    },
    {
      "epoch": 1.8014907836914063e-06,
      "step": 1889,
      "training_step_time": 17.11112666130066
    },
    {
      "epoch": 1.8024444580078124e-06,
      "grad_norm": 1.0744123458862305,
      "learning_rate": 1e-05,
      "loss": 0.0847,
      "step": 1890
    },
    {
      "epoch": 1.8024444580078124e-06,
      "model_forward_time": 5.610360383987427,
      "step": 1890
    },
    {
      "dynamics_loss_avg": 0.06040958203375339,
      "epoch": 1.8024444580078124e-06,
      "step": 1890
    },
    {
      "action_loss_avg": 0.017890438577160238,
      "epoch": 1.8024444580078124e-06,
      "step": 1890
    },
    {
      "epoch": 1.8024444580078124e-06,
      "step": 1890,
      "training_step_time": 17.100057363510132
    },
    {
      "epoch": 1.8033981323242188e-06,
      "model_forward_time": 5.967177391052246,
      "step": 1891
    },
    {
      "epoch": 1.8033981323242188e-06,
      "step": 1891,
      "training_step_time": 17.421475887298584
    },
    {
      "epoch": 1.804351806640625e-06,
      "model_forward_time": 5.602450132369995,
      "step": 1892
    },
    {
      "epoch": 1.804351806640625e-06,
      "step": 1892,
      "training_step_time": 17.062125205993652
    },
    {
      "epoch": 1.8053054809570313e-06,
      "model_forward_time": 5.5883917808532715,
      "step": 1893
    },
    {
      "epoch": 1.8053054809570313e-06,
      "step": 1893,
      "training_step_time": 17.028631925582886
    },
    {
      "epoch": 1.8062591552734374e-06,
      "model_forward_time": 5.5910186767578125,
      "step": 1894
    },
    {
      "epoch": 1.8062591552734374e-06,
      "step": 1894,
      "training_step_time": 17.021122455596924
    },
    {
      "epoch": 1.8072128295898438e-06,
      "model_forward_time": 5.595940589904785,
      "step": 1895
    },
    {
      "epoch": 1.8072128295898438e-06,
      "step": 1895,
      "training_step_time": 17.025375604629517
    },
    {
      "epoch": 1.80816650390625e-06,
      "model_forward_time": 5.604051351547241,
      "step": 1896
    },
    {
      "epoch": 1.80816650390625e-06,
      "step": 1896,
      "training_step_time": 17.0592942237854
    },
    {
      "epoch": 1.8091201782226563e-06,
      "model_forward_time": 5.968304872512817,
      "step": 1897
    },
    {
      "epoch": 1.8091201782226563e-06,
      "step": 1897,
      "training_step_time": 17.44481921195984
    },
    {
      "epoch": 1.8100738525390624e-06,
      "model_forward_time": 5.611130237579346,
      "step": 1898
    },
    {
      "epoch": 1.8100738525390624e-06,
      "step": 1898,
      "training_step_time": 17.104159116744995
    },
    {
      "epoch": 1.8110275268554688e-06,
      "model_forward_time": 5.618419885635376,
      "step": 1899
    },
    {
      "epoch": 1.8110275268554688e-06,
      "step": 1899,
      "training_step_time": 17.117294311523438
    },
    {
      "epoch": 1.811981201171875e-06,
      "grad_norm": 1.3629289865493774,
      "learning_rate": 1e-05,
      "loss": 0.077,
      "step": 1900
    },
    {
      "epoch": 1.811981201171875e-06,
      "model_forward_time": 5.62367582321167,
      "step": 1900
    },
    {
      "dynamics_loss_avg": 0.05937362648546696,
      "epoch": 1.811981201171875e-06,
      "step": 1900
    },
    {
      "action_loss_avg": 0.019315622467547654,
      "epoch": 1.811981201171875e-06,
      "step": 1900
    },
    {
      "epoch": 1.811981201171875e-06,
      "step": 1900,
      "training_step_time": 17.109049797058105
    },
    {
      "epoch": 1.8129348754882813e-06,
      "model_forward_time": 5.611749649047852,
      "step": 1901
    },
    {
      "epoch": 1.8129348754882813e-06,
      "step": 1901,
      "training_step_time": 17.10870885848999
    },
    {
      "epoch": 1.8138885498046874e-06,
      "model_forward_time": 5.605737686157227,
      "step": 1902
    },
    {
      "epoch": 1.8138885498046874e-06,
      "step": 1902,
      "training_step_time": 17.467427730560303
    },
    {
      "epoch": 1.8148422241210938e-06,
      "model_forward_time": 5.61856484413147,
      "step": 1903
    },
    {
      "epoch": 1.8148422241210938e-06,
      "step": 1903,
      "training_step_time": 17.11486005783081
    },
    {
      "epoch": 1.8157958984375e-06,
      "model_forward_time": 5.609575510025024,
      "step": 1904
    },
    {
      "epoch": 1.8157958984375e-06,
      "step": 1904,
      "training_step_time": 17.107537269592285
    },
    {
      "epoch": 1.8167495727539063e-06,
      "model_forward_time": 5.614884853363037,
      "step": 1905
    },
    {
      "epoch": 1.8167495727539063e-06,
      "step": 1905,
      "training_step_time": 17.107025146484375
    },
    {
      "epoch": 1.8177032470703124e-06,
      "model_forward_time": 5.611638307571411,
      "step": 1906
    },
    {
      "epoch": 1.8177032470703124e-06,
      "step": 1906,
      "training_step_time": 17.089775562286377
    },
    {
      "epoch": 1.8186569213867188e-06,
      "model_forward_time": 5.608200311660767,
      "step": 1907
    },
    {
      "epoch": 1.8186569213867188e-06,
      "step": 1907,
      "training_step_time": 17.097888946533203
    },
    {
      "epoch": 1.819610595703125e-06,
      "model_forward_time": 5.976360321044922,
      "step": 1908
    },
    {
      "epoch": 1.819610595703125e-06,
      "step": 1908,
      "training_step_time": 17.464134454727173
    },
    {
      "epoch": 1.8205642700195313e-06,
      "model_forward_time": 5.5956971645355225,
      "step": 1909
    },
    {
      "epoch": 1.8205642700195313e-06,
      "step": 1909,
      "training_step_time": 17.082418203353882
    },
    {
      "epoch": 1.8215179443359374e-06,
      "grad_norm": 1.093293309211731,
      "learning_rate": 1e-05,
      "loss": 0.0722,
      "step": 1910
    },
    {
      "epoch": 1.8215179443359374e-06,
      "model_forward_time": 5.603272914886475,
      "step": 1910
    },
    {
      "dynamics_loss_avg": 0.05650934148579836,
      "epoch": 1.8215179443359374e-06,
      "step": 1910
    },
    {
      "action_loss_avg": 0.01757852416485548,
      "epoch": 1.8215179443359374e-06,
      "step": 1910
    },
    {
      "epoch": 1.8215179443359374e-06,
      "step": 1910,
      "training_step_time": 17.073006629943848
    },
    {
      "epoch": 1.8224716186523438e-06,
      "model_forward_time": 5.59562087059021,
      "step": 1911
    },
    {
      "epoch": 1.8224716186523438e-06,
      "step": 1911,
      "training_step_time": 17.073401927947998
    },
    {
      "epoch": 1.82342529296875e-06,
      "model_forward_time": 5.595203399658203,
      "step": 1912
    },
    {
      "epoch": 1.82342529296875e-06,
      "step": 1912,
      "training_step_time": 17.043142557144165
    },
    {
      "epoch": 1.8243789672851563e-06,
      "model_forward_time": 5.5874269008636475,
      "step": 1913
    },
    {
      "epoch": 1.8243789672851563e-06,
      "step": 1913,
      "training_step_time": 17.38432216644287
    },
    {
      "epoch": 1.8253326416015624e-06,
      "model_forward_time": 5.582846403121948,
      "step": 1914
    },
    {
      "epoch": 1.8253326416015624e-06,
      "step": 1914,
      "training_step_time": 17.03153419494629
    },
    {
      "epoch": 1.8262863159179688e-06,
      "model_forward_time": 5.583696603775024,
      "step": 1915
    },
    {
      "epoch": 1.8262863159179688e-06,
      "step": 1915,
      "training_step_time": 17.014458179473877
    },
    {
      "epoch": 1.827239990234375e-06,
      "model_forward_time": 5.593910217285156,
      "step": 1916
    },
    {
      "epoch": 1.827239990234375e-06,
      "step": 1916,
      "training_step_time": 17.061784267425537
    },
    {
      "epoch": 1.8281936645507813e-06,
      "model_forward_time": 5.614869832992554,
      "step": 1917
    },
    {
      "epoch": 1.8281936645507813e-06,
      "step": 1917,
      "training_step_time": 17.090700387954712
    },
    {
      "epoch": 1.8291473388671874e-06,
      "model_forward_time": 5.600505113601685,
      "step": 1918
    },
    {
      "epoch": 1.8291473388671874e-06,
      "step": 1918,
      "training_step_time": 17.040343761444092
    },
    {
      "epoch": 1.8301010131835938e-06,
      "model_forward_time": 5.953602075576782,
      "step": 1919
    },
    {
      "epoch": 1.8301010131835938e-06,
      "step": 1919,
      "training_step_time": 17.383092164993286
    },
    {
      "epoch": 1.8310546875e-06,
      "grad_norm": 0.40702226758003235,
      "learning_rate": 1e-05,
      "loss": 0.0835,
      "step": 1920
    },
    {
      "epoch": 1.8310546875e-06,
      "model_forward_time": 5.58365797996521,
      "step": 1920
    },
    {
      "dynamics_loss_avg": 0.06023184861987829,
      "epoch": 1.8310546875e-06,
      "step": 1920
    },
    {
      "action_loss_avg": 0.01995097999460995,
      "epoch": 1.8310546875e-06,
      "step": 1920
    },
    {
      "epoch": 1.8310546875e-06,
      "step": 1920,
      "training_step_time": 17.017961025238037
    },
    {
      "epoch": 1.8320083618164063e-06,
      "model_forward_time": 5.586472272872925,
      "step": 1921
    },
    {
      "epoch": 1.8320083618164063e-06,
      "step": 1921,
      "training_step_time": 17.012572288513184
    },
    {
      "epoch": 1.8329620361328124e-06,
      "model_forward_time": 5.605081081390381,
      "step": 1922
    },
    {
      "epoch": 1.8329620361328124e-06,
      "step": 1922,
      "training_step_time": 17.075763940811157
    },
    {
      "epoch": 1.8339157104492188e-06,
      "model_forward_time": 5.623979330062866,
      "step": 1923
    },
    {
      "epoch": 1.8339157104492188e-06,
      "step": 1923,
      "training_step_time": 17.115890741348267
    },
    {
      "epoch": 1.834869384765625e-06,
      "model_forward_time": 5.9874207973480225,
      "step": 1924
    },
    {
      "epoch": 1.834869384765625e-06,
      "step": 1924,
      "training_step_time": 17.49515676498413
    },
    {
      "epoch": 1.8358230590820313e-06,
      "model_forward_time": 5.628743886947632,
      "step": 1925
    },
    {
      "epoch": 1.8358230590820313e-06,
      "step": 1925,
      "training_step_time": 17.10149884223938
    },
    {
      "epoch": 1.8367767333984374e-06,
      "model_forward_time": 5.585856914520264,
      "step": 1926
    },
    {
      "epoch": 1.8367767333984374e-06,
      "step": 1926,
      "training_step_time": 17.004756212234497
    },
    {
      "epoch": 1.8377304077148438e-06,
      "model_forward_time": 5.579223394393921,
      "step": 1927
    },
    {
      "epoch": 1.8377304077148438e-06,
      "step": 1927,
      "training_step_time": 16.994510650634766
    },
    {
      "epoch": 1.83868408203125e-06,
      "model_forward_time": 5.5754475593566895,
      "step": 1928
    },
    {
      "epoch": 1.83868408203125e-06,
      "step": 1928,
      "training_step_time": 16.979681968688965
    },
    {
      "epoch": 1.8396377563476563e-06,
      "model_forward_time": 5.570980072021484,
      "step": 1929
    },
    {
      "epoch": 1.8396377563476563e-06,
      "step": 1929,
      "training_step_time": 17.004469394683838
    },
    {
      "epoch": 1.8405914306640624e-06,
      "grad_norm": 0.9369627833366394,
      "learning_rate": 1e-05,
      "loss": 0.0681,
      "step": 1930
    },
    {
      "epoch": 1.8405914306640624e-06,
      "model_forward_time": 5.933247089385986,
      "step": 1930
    },
    {
      "dynamics_loss_avg": 0.05317197535187006,
      "epoch": 1.8405914306640624e-06,
      "step": 1930
    },
    {
      "action_loss_avg": 0.01613466371782124,
      "epoch": 1.8405914306640624e-06,
      "step": 1930
    },
    {
      "epoch": 1.8405914306640624e-06,
      "step": 1930,
      "training_step_time": 17.359763383865356
    },
    {
      "epoch": 1.8415451049804688e-06,
      "model_forward_time": 5.596045732498169,
      "step": 1931
    },
    {
      "epoch": 1.8415451049804688e-06,
      "step": 1931,
      "training_step_time": 17.01758599281311
    },
    {
      "epoch": 1.842498779296875e-06,
      "model_forward_time": 5.582545757293701,
      "step": 1932
    },
    {
      "epoch": 1.842498779296875e-06,
      "step": 1932,
      "training_step_time": 17.01099443435669
    },
    {
      "epoch": 1.8434524536132813e-06,
      "model_forward_time": 5.5974507331848145,
      "step": 1933
    },
    {
      "epoch": 1.8434524536132813e-06,
      "step": 1933,
      "training_step_time": 17.04201078414917
    },
    {
      "epoch": 1.8444061279296874e-06,
      "model_forward_time": 5.606290340423584,
      "step": 1934
    },
    {
      "epoch": 1.8444061279296874e-06,
      "step": 1934,
      "training_step_time": 17.101335763931274
    },
    {
      "epoch": 1.8453598022460938e-06,
      "model_forward_time": 5.645545959472656,
      "step": 1935
    },
    {
      "epoch": 1.8453598022460938e-06,
      "step": 1935,
      "training_step_time": 17.509603261947632
    },
    {
      "epoch": 1.8463134765625e-06,
      "model_forward_time": 5.616369962692261,
      "step": 1936
    },
    {
      "epoch": 1.8463134765625e-06,
      "step": 1936,
      "training_step_time": 17.11527442932129
    },
    {
      "epoch": 1.8472671508789063e-06,
      "model_forward_time": 5.610056400299072,
      "step": 1937
    },
    {
      "epoch": 1.8472671508789063e-06,
      "step": 1937,
      "training_step_time": 17.07253646850586
    },
    {
      "epoch": 1.8482208251953124e-06,
      "model_forward_time": 5.59691596031189,
      "step": 1938
    },
    {
      "epoch": 1.8482208251953124e-06,
      "step": 1938,
      "training_step_time": 17.071637868881226
    },
    {
      "epoch": 1.8491744995117188e-06,
      "model_forward_time": 5.606248140335083,
      "step": 1939
    },
    {
      "epoch": 1.8491744995117188e-06,
      "step": 1939,
      "training_step_time": 17.05596661567688
    },
    {
      "epoch": 1.850128173828125e-06,
      "grad_norm": 1.2249855995178223,
      "learning_rate": 1e-05,
      "loss": 0.0882,
      "step": 1940
    },
    {
      "epoch": 1.850128173828125e-06,
      "model_forward_time": 5.601675987243652,
      "step": 1940
    },
    {
      "dynamics_loss_avg": 0.0679645549505949,
      "epoch": 1.850128173828125e-06,
      "step": 1940
    },
    {
      "action_loss_avg": 0.02441154969856143,
      "epoch": 1.850128173828125e-06,
      "step": 1940
    },
    {
      "epoch": 1.850128173828125e-06,
      "step": 1940,
      "training_step_time": 17.024436950683594
    },
    {
      "epoch": 1.8510818481445313e-06,
      "model_forward_time": 5.959826707839966,
      "step": 1941
    },
    {
      "epoch": 1.8510818481445313e-06,
      "step": 1941,
      "training_step_time": 17.405699968338013
    },
    {
      "epoch": 1.8520355224609375e-06,
      "model_forward_time": 5.587077617645264,
      "step": 1942
    },
    {
      "epoch": 1.8520355224609375e-06,
      "step": 1942,
      "training_step_time": 17.01498794555664
    },
    {
      "epoch": 1.8529891967773438e-06,
      "model_forward_time": 5.587049961090088,
      "step": 1943
    },
    {
      "epoch": 1.8529891967773438e-06,
      "step": 1943,
      "training_step_time": 17.033274173736572
    },
    {
      "epoch": 1.85394287109375e-06,
      "model_forward_time": 5.6022515296936035,
      "step": 1944
    },
    {
      "epoch": 1.85394287109375e-06,
      "step": 1944,
      "training_step_time": 17.10007882118225
    },
    {
      "epoch": 1.8548965454101563e-06,
      "model_forward_time": 5.602144956588745,
      "step": 1945
    },
    {
      "epoch": 1.8548965454101563e-06,
      "step": 1945,
      "training_step_time": 17.033650159835815
    },
    {
      "epoch": 1.8558502197265625e-06,
      "model_forward_time": 5.605051755905151,
      "step": 1946
    },
    {
      "epoch": 1.8558502197265625e-06,
      "step": 1946,
      "training_step_time": 17.461061239242554
    },
    {
      "epoch": 1.8568038940429688e-06,
      "model_forward_time": 5.616213321685791,
      "step": 1947
    },
    {
      "epoch": 1.8568038940429688e-06,
      "step": 1947,
      "training_step_time": 17.101207494735718
    },
    {
      "epoch": 1.857757568359375e-06,
      "model_forward_time": 5.605334997177124,
      "step": 1948
    },
    {
      "epoch": 1.857757568359375e-06,
      "step": 1948,
      "training_step_time": 17.083845615386963
    },
    {
      "epoch": 1.8587112426757813e-06,
      "model_forward_time": 5.606014728546143,
      "step": 1949
    },
    {
      "epoch": 1.8587112426757813e-06,
      "step": 1949,
      "training_step_time": 17.102252960205078
    },
    {
      "epoch": 1.8596649169921875e-06,
      "grad_norm": 1.2431708574295044,
      "learning_rate": 1e-05,
      "loss": 0.0666,
      "step": 1950
    },
    {
      "epoch": 1.8596649169921875e-06,
      "model_forward_time": 5.61698579788208,
      "step": 1950
    },
    {
      "dynamics_loss_avg": 0.05026345048099756,
      "epoch": 1.8596649169921875e-06,
      "step": 1950
    },
    {
      "action_loss_avg": 0.013724030228331685,
      "epoch": 1.8596649169921875e-06,
      "step": 1950
    },
    {
      "epoch": 1.8596649169921875e-06,
      "step": 1950,
      "training_step_time": 17.10617184638977
    },
    {
      "epoch": 1.8606185913085938e-06,
      "model_forward_time": 5.5953874588012695,
      "step": 1951
    },
    {
      "epoch": 1.8606185913085938e-06,
      "step": 1951,
      "training_step_time": 17.061346530914307
    },
    {
      "epoch": 1.861572265625e-06,
      "model_forward_time": 5.974792957305908,
      "step": 1952
    },
    {
      "epoch": 1.861572265625e-06,
      "step": 1952,
      "training_step_time": 17.44225788116455
    },
    {
      "epoch": 1.8625259399414063e-06,
      "model_forward_time": 5.607264041900635,
      "step": 1953
    },
    {
      "epoch": 1.8625259399414063e-06,
      "step": 1953,
      "training_step_time": 17.0907621383667
    },
    {
      "epoch": 1.8634796142578125e-06,
      "model_forward_time": 5.608428001403809,
      "step": 1954
    },
    {
      "epoch": 1.8634796142578125e-06,
      "step": 1954,
      "training_step_time": 17.103139877319336
    },
    {
      "epoch": 1.8644332885742188e-06,
      "model_forward_time": 5.613882541656494,
      "step": 1955
    },
    {
      "epoch": 1.8644332885742188e-06,
      "step": 1955,
      "training_step_time": 17.10077953338623
    },
    {
      "epoch": 1.865386962890625e-06,
      "model_forward_time": 5.604485273361206,
      "step": 1956
    },
    {
      "epoch": 1.865386962890625e-06,
      "step": 1956,
      "training_step_time": 17.050329208374023
    },
    {
      "epoch": 1.8663406372070313e-06,
      "model_forward_time": 5.929195404052734,
      "step": 1957
    },
    {
      "epoch": 1.8663406372070313e-06,
      "step": 1957,
      "training_step_time": 17.337072372436523
    },
    {
      "epoch": 1.8672943115234375e-06,
      "model_forward_time": 5.569621801376343,
      "step": 1958
    },
    {
      "epoch": 1.8672943115234375e-06,
      "step": 1958,
      "training_step_time": 16.98985767364502
    },
    {
      "epoch": 1.8682479858398438e-06,
      "model_forward_time": 5.571774005889893,
      "step": 1959
    },
    {
      "epoch": 1.8682479858398438e-06,
      "step": 1959,
      "training_step_time": 17.009055137634277
    },
    {
      "epoch": 1.86920166015625e-06,
      "grad_norm": 1.6590758562088013,
      "learning_rate": 1e-05,
      "loss": 0.0792,
      "step": 1960
    },
    {
      "epoch": 1.86920166015625e-06,
      "model_forward_time": 5.575715780258179,
      "step": 1960
    },
    {
      "dynamics_loss_avg": 0.06597959585487842,
      "epoch": 1.86920166015625e-06,
      "step": 1960
    },
    {
      "action_loss_avg": 0.021106382180005313,
      "epoch": 1.86920166015625e-06,
      "step": 1960
    },
    {
      "epoch": 1.86920166015625e-06,
      "step": 1960,
      "training_step_time": 17.01055359840393
    },
    {
      "epoch": 1.8701553344726563e-06,
      "model_forward_time": 5.580137014389038,
      "step": 1961
    },
    {
      "epoch": 1.8701553344726563e-06,
      "step": 1961,
      "training_step_time": 17.021022081375122
    },
    {
      "epoch": 1.8711090087890625e-06,
      "model_forward_time": 5.587834119796753,
      "step": 1962
    },
    {
      "epoch": 1.8711090087890625e-06,
      "step": 1962,
      "training_step_time": 17.035208225250244
    },
    {
      "epoch": 1.8720626831054688e-06,
      "model_forward_time": 5.979027986526489,
      "step": 1963
    },
    {
      "epoch": 1.8720626831054688e-06,
      "step": 1963,
      "training_step_time": 17.466283798217773
    },
    {
      "epoch": 1.873016357421875e-06,
      "model_forward_time": 5.624507904052734,
      "step": 1964
    },
    {
      "epoch": 1.873016357421875e-06,
      "step": 1964,
      "training_step_time": 17.15376305580139
    },
    {
      "epoch": 1.8739700317382813e-06,
      "model_forward_time": 5.620124340057373,
      "step": 1965
    },
    {
      "epoch": 1.8739700317382813e-06,
      "step": 1965,
      "training_step_time": 17.12627625465393
    },
    {
      "epoch": 1.8749237060546875e-06,
      "model_forward_time": 5.6044416427612305,
      "step": 1966
    },
    {
      "epoch": 1.8749237060546875e-06,
      "step": 1966,
      "training_step_time": 17.06676435470581
    },
    {
      "epoch": 1.8758773803710938e-06,
      "model_forward_time": 5.586665391921997,
      "step": 1967
    },
    {
      "epoch": 1.8758773803710938e-06,
      "step": 1967,
      "training_step_time": 17.025362253189087
    },
    {
      "epoch": 1.8768310546875e-06,
      "model_forward_time": 5.577404022216797,
      "step": 1968
    },
    {
      "epoch": 1.8768310546875e-06,
      "step": 1968,
      "training_step_time": 17.354153156280518
    },
    {
      "epoch": 1.8777847290039063e-06,
      "model_forward_time": 5.583399534225464,
      "step": 1969
    },
    {
      "epoch": 1.8777847290039063e-06,
      "step": 1969,
      "training_step_time": 16.994622468948364
    },
    {
      "epoch": 1.8787384033203125e-06,
      "grad_norm": 0.9657015204429626,
      "learning_rate": 1e-05,
      "loss": 0.0729,
      "step": 1970
    },
    {
      "epoch": 1.8787384033203125e-06,
      "model_forward_time": 5.588171005249023,
      "step": 1970
    },
    {
      "dynamics_loss_avg": 0.05497757904231548,
      "epoch": 1.8787384033203125e-06,
      "step": 1970
    },
    {
      "action_loss_avg": 0.01582382107153535,
      "epoch": 1.8787384033203125e-06,
      "step": 1970
    },
    {
      "epoch": 1.8787384033203125e-06,
      "step": 1970,
      "training_step_time": 17.03232979774475
    },
    {
      "epoch": 1.8796920776367188e-06,
      "model_forward_time": 5.601515293121338,
      "step": 1971
    },
    {
      "epoch": 1.8796920776367188e-06,
      "step": 1971,
      "training_step_time": 17.054079294204712
    },
    {
      "epoch": 1.880645751953125e-06,
      "model_forward_time": 5.604841947555542,
      "step": 1972
    },
    {
      "epoch": 1.880645751953125e-06,
      "step": 1972,
      "training_step_time": 17.073261260986328
    },
    {
      "epoch": 1.8815994262695313e-06,
      "model_forward_time": 5.59940505027771,
      "step": 1973
    },
    {
      "epoch": 1.8815994262695313e-06,
      "step": 1973,
      "training_step_time": 17.060012340545654
    },
    {
      "epoch": 1.8825531005859375e-06,
      "model_forward_time": 5.9706809520721436,
      "step": 1974
    },
    {
      "epoch": 1.8825531005859375e-06,
      "step": 1974,
      "training_step_time": 17.430521726608276
    },
    {
      "epoch": 1.8835067749023438e-06,
      "model_forward_time": 5.605151653289795,
      "step": 1975
    },
    {
      "epoch": 1.8835067749023438e-06,
      "step": 1975,
      "training_step_time": 17.085832595825195
    },
    {
      "epoch": 1.88446044921875e-06,
      "model_forward_time": 5.612879753112793,
      "step": 1976
    },
    {
      "epoch": 1.88446044921875e-06,
      "step": 1976,
      "training_step_time": 17.112006902694702
    },
    {
      "epoch": 1.8854141235351563e-06,
      "model_forward_time": 5.609498977661133,
      "step": 1977
    },
    {
      "epoch": 1.8854141235351563e-06,
      "step": 1977,
      "training_step_time": 17.087963104248047
    },
    {
      "epoch": 1.8863677978515625e-06,
      "model_forward_time": 5.605503797531128,
      "step": 1978
    },
    {
      "epoch": 1.8863677978515625e-06,
      "step": 1978,
      "training_step_time": 17.082155466079712
    },
    {
      "epoch": 1.8873214721679688e-06,
      "model_forward_time": 5.948463678359985,
      "step": 1979
    },
    {
      "epoch": 1.8873214721679688e-06,
      "step": 1979,
      "training_step_time": 17.423392057418823
    },
    {
      "epoch": 1.888275146484375e-06,
      "grad_norm": 0.9158557057380676,
      "learning_rate": 1e-05,
      "loss": 0.078,
      "step": 1980
    },
    {
      "epoch": 1.888275146484375e-06,
      "model_forward_time": 5.604876279830933,
      "step": 1980
    },
    {
      "dynamics_loss_avg": 0.05508088618516922,
      "epoch": 1.888275146484375e-06,
      "step": 1980
    },
    {
      "action_loss_avg": 0.016679865075275302,
      "epoch": 1.888275146484375e-06,
      "step": 1980
    },
    {
      "epoch": 1.888275146484375e-06,
      "step": 1980,
      "training_step_time": 17.09743881225586
    },
    {
      "epoch": 1.8892288208007813e-06,
      "model_forward_time": 5.61046576499939,
      "step": 1981
    },
    {
      "epoch": 1.8892288208007813e-06,
      "step": 1981,
      "training_step_time": 17.111185789108276
    },
    {
      "epoch": 1.8901824951171875e-06,
      "model_forward_time": 5.626043081283569,
      "step": 1982
    },
    {
      "epoch": 1.8901824951171875e-06,
      "step": 1982,
      "training_step_time": 17.118475914001465
    },
    {
      "epoch": 1.8911361694335938e-06,
      "model_forward_time": 5.603294849395752,
      "step": 1983
    },
    {
      "epoch": 1.8911361694335938e-06,
      "step": 1983,
      "training_step_time": 17.050403833389282
    },
    {
      "epoch": 1.89208984375e-06,
      "model_forward_time": 5.586453199386597,
      "step": 1984
    },
    {
      "epoch": 1.89208984375e-06,
      "step": 1984,
      "training_step_time": 17.0046865940094
    },
    {
      "epoch": 1.8930435180664063e-06,
      "model_forward_time": 5.94635796546936,
      "step": 1985
    },
    {
      "epoch": 1.8930435180664063e-06,
      "step": 1985,
      "training_step_time": 17.365195751190186
    },
    {
      "epoch": 1.8939971923828125e-06,
      "model_forward_time": 5.577345371246338,
      "step": 1986
    },
    {
      "epoch": 1.8939971923828125e-06,
      "step": 1986,
      "training_step_time": 17.01756000518799
    },
    {
      "epoch": 1.8949508666992188e-06,
      "model_forward_time": 5.580199480056763,
      "step": 1987
    },
    {
      "epoch": 1.8949508666992188e-06,
      "step": 1987,
      "training_step_time": 17.026423692703247
    },
    {
      "epoch": 1.895904541015625e-06,
      "model_forward_time": 5.6014721393585205,
      "step": 1988
    },
    {
      "epoch": 1.895904541015625e-06,
      "step": 1988,
      "training_step_time": 17.080482482910156
    },
    {
      "epoch": 1.8968582153320313e-06,
      "model_forward_time": 5.608477592468262,
      "step": 1989
    },
    {
      "epoch": 1.8968582153320313e-06,
      "step": 1989,
      "training_step_time": 17.089324474334717
    },
    {
      "epoch": 1.8978118896484375e-06,
      "grad_norm": 0.7963810563087463,
      "learning_rate": 1e-05,
      "loss": 0.075,
      "step": 1990
    },
    {
      "epoch": 1.8978118896484375e-06,
      "model_forward_time": 5.980987310409546,
      "step": 1990
    },
    {
      "dynamics_loss_avg": 0.05751292742788792,
      "epoch": 1.8978118896484375e-06,
      "step": 1990
    },
    {
      "action_loss_avg": 0.015671580843627454,
      "epoch": 1.8978118896484375e-06,
      "step": 1990
    },
    {
      "epoch": 1.8978118896484375e-06,
      "step": 1990,
      "training_step_time": 17.4696307182312
    },
    {
      "epoch": 1.8987655639648438e-06,
      "model_forward_time": 5.599291086196899,
      "step": 1991
    },
    {
      "epoch": 1.8987655639648438e-06,
      "step": 1991,
      "training_step_time": 17.060570240020752
    },
    {
      "epoch": 1.89971923828125e-06,
      "model_forward_time": 5.591749906539917,
      "step": 1992
    },
    {
      "epoch": 1.89971923828125e-06,
      "step": 1992,
      "training_step_time": 17.03053379058838
    },
    {
      "epoch": 1.9006729125976563e-06,
      "model_forward_time": 5.590371131896973,
      "step": 1993
    },
    {
      "epoch": 1.9006729125976563e-06,
      "step": 1993,
      "training_step_time": 17.043394327163696
    },
    {
      "epoch": 1.9016265869140625e-06,
      "model_forward_time": 5.604087829589844,
      "step": 1994
    },
    {
      "epoch": 1.9016265869140625e-06,
      "step": 1994,
      "training_step_time": 17.08616304397583
    },
    {
      "epoch": 1.9025802612304689e-06,
      "model_forward_time": 5.616748809814453,
      "step": 1995
    },
    {
      "epoch": 1.9025802612304689e-06,
      "step": 1995,
      "training_step_time": 17.464092016220093
    },
    {
      "epoch": 1.903533935546875e-06,
      "model_forward_time": 5.611379861831665,
      "step": 1996
    },
    {
      "epoch": 1.903533935546875e-06,
      "step": 1996,
      "training_step_time": 17.098581790924072
    },
    {
      "epoch": 1.9044876098632814e-06,
      "model_forward_time": 5.613332033157349,
      "step": 1997
    },
    {
      "epoch": 1.9044876098632814e-06,
      "step": 1997,
      "training_step_time": 17.086640119552612
    },
    {
      "epoch": 1.9054412841796875e-06,
      "model_forward_time": 5.603274345397949,
      "step": 1998
    },
    {
      "epoch": 1.9054412841796875e-06,
      "step": 1998,
      "training_step_time": 17.0343074798584
    },
    {
      "epoch": 1.9063949584960939e-06,
      "model_forward_time": 5.584512948989868,
      "step": 1999
    },
    {
      "epoch": 1.9063949584960939e-06,
      "step": 1999,
      "training_step_time": 16.985033750534058
    },
    {
      "epoch": 1.9073486328125e-06,
      "grad_norm": 1.9663352966308594,
      "learning_rate": 1e-05,
      "loss": 0.0901,
      "step": 2000
    },
    {
      "epoch": 1.9073486328125e-06,
      "model_forward_time": 5.617415904998779,
      "step": 2000
    },
    {
      "dynamics_loss_avg": 0.07394531667232514,
      "epoch": 1.9073486328125e-06,
      "step": 2000
    },
    {
      "action_loss_avg": 0.021248415298759937,
      "epoch": 1.9073486328125e-06,
      "step": 2000
    },
    {
      "epoch": 1.9073486328125e-06,
      "step": 2000,
      "training_step_time": 17.043957710266113
    },
    {
      "epoch": 1.908302307128906e-06,
      "model_forward_time": 5.9975502490997314,
      "step": 2001
    },
    {
      "epoch": 1.908302307128906e-06,
      "step": 2001,
      "training_step_time": 17.511157751083374
    },
    {
      "epoch": 1.9092559814453123e-06,
      "model_forward_time": 5.674611806869507,
      "step": 2002
    },
    {
      "epoch": 1.9092559814453123e-06,
      "step": 2002,
      "training_step_time": 17.184847593307495
    },
    {
      "epoch": 1.910209655761719e-06,
      "model_forward_time": 5.664215803146362,
      "step": 2003
    },
    {
      "epoch": 1.910209655761719e-06,
      "step": 2003,
      "training_step_time": 17.186525583267212
    },
    {
      "epoch": 1.911163330078125e-06,
      "model_forward_time": 5.6930577754974365,
      "step": 2004
    },
    {
      "epoch": 1.911163330078125e-06,
      "step": 2004,
      "training_step_time": 17.183230876922607
    },
    {
      "epoch": 1.912117004394531e-06,
      "model_forward_time": 5.608337163925171,
      "step": 2005
    },
    {
      "epoch": 1.912117004394531e-06,
      "step": 2005,
      "training_step_time": 17.047935724258423
    },
    {
      "epoch": 1.9130706787109373e-06,
      "model_forward_time": 5.9314398765563965,
      "step": 2006
    },
    {
      "epoch": 1.9130706787109373e-06,
      "step": 2006,
      "training_step_time": 17.353065729141235
    },
    {
      "epoch": 1.914024353027344e-06,
      "model_forward_time": 5.5788702964782715,
      "step": 2007
    },
    {
      "epoch": 1.914024353027344e-06,
      "step": 2007,
      "training_step_time": 17.008652925491333
    },
    {
      "epoch": 1.91497802734375e-06,
      "model_forward_time": 5.591000318527222,
      "step": 2008
    },
    {
      "epoch": 1.91497802734375e-06,
      "step": 2008,
      "training_step_time": 17.04437279701233
    },
    {
      "epoch": 1.915931701660156e-06,
      "model_forward_time": 5.606053113937378,
      "step": 2009
    },
    {
      "epoch": 1.915931701660156e-06,
      "step": 2009,
      "training_step_time": 17.060519456863403
    },
    {
      "epoch": 1.9168853759765623e-06,
      "grad_norm": 1.6584970951080322,
      "learning_rate": 1e-05,
      "loss": 0.0851,
      "step": 2010
    },
    {
      "epoch": 1.9168853759765623e-06,
      "model_forward_time": 5.602184295654297,
      "step": 2010
    },
    {
      "dynamics_loss_avg": 0.06687413267791271,
      "epoch": 1.9168853759765623e-06,
      "step": 2010
    },
    {
      "action_loss_avg": 0.020920541789382695,
      "epoch": 1.9168853759765623e-06,
      "step": 2010
    },
    {
      "epoch": 1.9168853759765623e-06,
      "step": 2010,
      "training_step_time": 17.063329696655273
    },
    {
      "epoch": 1.917839050292969e-06,
      "model_forward_time": 5.596333265304565,
      "step": 2011
    },
    {
      "epoch": 1.917839050292969e-06,
      "step": 2011,
      "training_step_time": 17.04278874397278
    },
    {
      "epoch": 1.918792724609375e-06,
      "model_forward_time": 5.954978942871094,
      "step": 2012
    },
    {
      "epoch": 1.918792724609375e-06,
      "step": 2012,
      "training_step_time": 17.385599374771118
    },
    {
      "epoch": 1.919746398925781e-06,
      "model_forward_time": 5.595270872116089,
      "step": 2013
    },
    {
      "epoch": 1.919746398925781e-06,
      "step": 2013,
      "training_step_time": 17.06132674217224
    },
    {
      "epoch": 1.9207000732421873e-06,
      "model_forward_time": 5.629546403884888,
      "step": 2014
    },
    {
      "epoch": 1.9207000732421873e-06,
      "step": 2014,
      "training_step_time": 17.100098371505737
    },
    {
      "epoch": 1.921653747558594e-06,
      "model_forward_time": 5.609220743179321,
      "step": 2015
    },
    {
      "epoch": 1.921653747558594e-06,
      "step": 2015,
      "training_step_time": 17.059534549713135
    },
    {
      "epoch": 1.922607421875e-06,
      "model_forward_time": 5.59394907951355,
      "step": 2016
    },
    {
      "epoch": 1.922607421875e-06,
      "step": 2016,
      "training_step_time": 17.03450059890747
    },
    {
      "epoch": 1.923561096191406e-06,
      "model_forward_time": 5.935792446136475,
      "step": 2017
    },
    {
      "epoch": 1.923561096191406e-06,
      "step": 2017,
      "training_step_time": 17.37616229057312
    },
    {
      "epoch": 1.9245147705078123e-06,
      "model_forward_time": 5.606027126312256,
      "step": 2018
    },
    {
      "epoch": 1.9245147705078123e-06,
      "step": 2018,
      "training_step_time": 17.082493543624878
    },
    {
      "epoch": 1.925468444824219e-06,
      "model_forward_time": 5.605107545852661,
      "step": 2019
    },
    {
      "epoch": 1.925468444824219e-06,
      "step": 2019,
      "training_step_time": 17.08034920692444
    },
    {
      "epoch": 1.926422119140625e-06,
      "grad_norm": 1.1886765956878662,
      "learning_rate": 1e-05,
      "loss": 0.0938,
      "step": 2020
    },
    {
      "epoch": 1.926422119140625e-06,
      "model_forward_time": 5.608006000518799,
      "step": 2020
    },
    {
      "dynamics_loss_avg": 0.06735272221267223,
      "epoch": 1.926422119140625e-06,
      "step": 2020
    },
    {
      "action_loss_avg": 0.02026484007947147,
      "epoch": 1.926422119140625e-06,
      "step": 2020
    },
    {
      "epoch": 1.926422119140625e-06,
      "step": 2020,
      "training_step_time": 17.066793203353882
    },
    {
      "epoch": 1.927375793457031e-06,
      "model_forward_time": 5.604067087173462,
      "step": 2021
    },
    {
      "epoch": 1.927375793457031e-06,
      "step": 2021,
      "training_step_time": 17.08021092414856
    },
    {
      "epoch": 1.9283294677734373e-06,
      "model_forward_time": 5.59726357460022,
      "step": 2022
    },
    {
      "epoch": 1.9283294677734373e-06,
      "step": 2022,
      "training_step_time": 17.073638439178467
    },
    {
      "epoch": 1.929283142089844e-06,
      "model_forward_time": 5.96600341796875,
      "step": 2023
    },
    {
      "epoch": 1.929283142089844e-06,
      "step": 2023,
      "training_step_time": 17.45734429359436
    },
    {
      "epoch": 1.93023681640625e-06,
      "model_forward_time": 5.606794357299805,
      "step": 2024
    },
    {
      "epoch": 1.93023681640625e-06,
      "step": 2024,
      "training_step_time": 17.10230302810669
    },
    {
      "epoch": 1.931190490722656e-06,
      "model_forward_time": 5.625829696655273,
      "step": 2025
    },
    {
      "epoch": 1.931190490722656e-06,
      "step": 2025,
      "training_step_time": 17.131349563598633
    },
    {
      "epoch": 1.9321441650390623e-06,
      "model_forward_time": 5.615208864212036,
      "step": 2026
    },
    {
      "epoch": 1.9321441650390623e-06,
      "step": 2026,
      "training_step_time": 17.089934825897217
    },
    {
      "epoch": 1.933097839355469e-06,
      "model_forward_time": 5.590886116027832,
      "step": 2027
    },
    {
      "epoch": 1.933097839355469e-06,
      "step": 2027,
      "training_step_time": 17.022614002227783
    },
    {
      "epoch": 1.934051513671875e-06,
      "model_forward_time": 5.907072067260742,
      "step": 2028
    },
    {
      "epoch": 1.934051513671875e-06,
      "step": 2028,
      "training_step_time": 17.33453917503357
    },
    {
      "epoch": 1.935005187988281e-06,
      "model_forward_time": 5.567033290863037,
      "step": 2029
    },
    {
      "epoch": 1.935005187988281e-06,
      "step": 2029,
      "training_step_time": 16.967501401901245
    },
    {
      "epoch": 1.9359588623046873e-06,
      "grad_norm": 0.5475432276725769,
      "learning_rate": 1e-05,
      "loss": 0.0637,
      "step": 2030
    },
    {
      "epoch": 1.9359588623046873e-06,
      "model_forward_time": 5.573256969451904,
      "step": 2030
    },
    {
      "dynamics_loss_avg": 0.053185096569359304,
      "epoch": 1.9359588623046873e-06,
      "step": 2030
    },
    {
      "action_loss_avg": 0.015492659620940685,
      "epoch": 1.9359588623046873e-06,
      "step": 2030
    },
    {
      "epoch": 1.9359588623046873e-06,
      "step": 2030,
      "training_step_time": 16.998064279556274
    },
    {
      "epoch": 1.936912536621094e-06,
      "model_forward_time": 5.573827505111694,
      "step": 2031
    },
    {
      "epoch": 1.936912536621094e-06,
      "step": 2031,
      "training_step_time": 16.993735313415527
    },
    {
      "epoch": 1.9378662109375e-06,
      "model_forward_time": 5.569855213165283,
      "step": 2032
    },
    {
      "epoch": 1.9378662109375e-06,
      "step": 2032,
      "training_step_time": 17.012914896011353
    },
    {
      "epoch": 1.938819885253906e-06,
      "model_forward_time": 5.95775842666626,
      "step": 2033
    },
    {
      "epoch": 1.938819885253906e-06,
      "step": 2033,
      "training_step_time": 17.405008792877197
    },
    {
      "epoch": 1.9397735595703123e-06,
      "model_forward_time": 5.596723318099976,
      "step": 2034
    },
    {
      "epoch": 1.9397735595703123e-06,
      "step": 2034,
      "training_step_time": 17.092409372329712
    },
    {
      "epoch": 1.940727233886719e-06,
      "model_forward_time": 5.622894763946533,
      "step": 2035
    },
    {
      "epoch": 1.940727233886719e-06,
      "step": 2035,
      "training_step_time": 17.142563343048096
    },
    {
      "epoch": 1.941680908203125e-06,
      "model_forward_time": 5.629651308059692,
      "step": 2036
    },
    {
      "epoch": 1.941680908203125e-06,
      "step": 2036,
      "training_step_time": 17.139323472976685
    },
    {
      "epoch": 1.942634582519531e-06,
      "model_forward_time": 5.615142345428467,
      "step": 2037
    },
    {
      "epoch": 1.942634582519531e-06,
      "step": 2037,
      "training_step_time": 17.07584571838379
    },
    {
      "epoch": 1.9435882568359373e-06,
      "model_forward_time": 5.585662603378296,
      "step": 2038
    },
    {
      "epoch": 1.9435882568359373e-06,
      "step": 2038,
      "training_step_time": 17.006985902786255
    },
    {
      "epoch": 1.944541931152344e-06,
      "model_forward_time": 5.9291369915008545,
      "step": 2039
    },
    {
      "epoch": 1.944541931152344e-06,
      "step": 2039,
      "training_step_time": 17.358039379119873
    },
    {
      "epoch": 1.94549560546875e-06,
      "grad_norm": 1.0814224481582642,
      "learning_rate": 1e-05,
      "loss": 0.0711,
      "step": 2040
    },
    {
      "epoch": 1.94549560546875e-06,
      "model_forward_time": 5.57566237449646,
      "step": 2040
    },
    {
      "dynamics_loss_avg": 0.0513758048415184,
      "epoch": 1.94549560546875e-06,
      "step": 2040
    },
    {
      "action_loss_avg": 0.01308274599723518,
      "epoch": 1.94549560546875e-06,
      "step": 2040
    },
    {
      "epoch": 1.94549560546875e-06,
      "step": 2040,
      "training_step_time": 16.997335195541382
    },
    {
      "epoch": 1.946449279785156e-06,
      "model_forward_time": 5.607075452804565,
      "step": 2041
    },
    {
      "epoch": 1.946449279785156e-06,
      "step": 2041,
      "training_step_time": 17.007920026779175
    },
    {
      "epoch": 1.9474029541015623e-06,
      "model_forward_time": 5.5735392570495605,
      "step": 2042
    },
    {
      "epoch": 1.9474029541015623e-06,
      "step": 2042,
      "training_step_time": 16.99659776687622
    },
    {
      "epoch": 1.948356628417969e-06,
      "model_forward_time": 5.576460361480713,
      "step": 2043
    },
    {
      "epoch": 1.948356628417969e-06,
      "step": 2043,
      "training_step_time": 17.020223379135132
    },
    {
      "epoch": 1.949310302734375e-06,
      "model_forward_time": 5.9611146450042725,
      "step": 2044
    },
    {
      "epoch": 1.949310302734375e-06,
      "step": 2044,
      "training_step_time": 17.399014949798584
    },
    {
      "epoch": 1.950263977050781e-06,
      "model_forward_time": 5.625004768371582,
      "step": 2045
    },
    {
      "epoch": 1.950263977050781e-06,
      "step": 2045,
      "training_step_time": 17.07451820373535
    },
    {
      "epoch": 1.9512176513671873e-06,
      "model_forward_time": 5.6075615882873535,
      "step": 2046
    },
    {
      "epoch": 1.9512176513671873e-06,
      "step": 2046,
      "training_step_time": 17.0700204372406
    },
    {
      "epoch": 1.952171325683594e-06,
      "model_forward_time": 5.596928596496582,
      "step": 2047
    },
    {
      "epoch": 1.952171325683594e-06,
      "step": 2047,
      "training_step_time": 17.037936449050903
    },
    {
      "epoch": 1.953125e-06,
      "model_forward_time": 5.591370344161987,
      "step": 2048
    },
    {
      "epoch": 1.953125e-06,
      "step": 2048,
      "training_step_time": 17.041932106018066
    },
    {
      "epoch": 1.954078674316406e-06,
      "model_forward_time": 5.5954742431640625,
      "step": 2049
    },
    {
      "epoch": 1.954078674316406e-06,
      "step": 2049,
      "training_step_time": 17.06653642654419
    },
    {
      "epoch": 1.9550323486328123e-06,
      "grad_norm": 1.1957134008407593,
      "learning_rate": 1e-05,
      "loss": 0.0896,
      "step": 2050
    },
    {
      "epoch": 1.9550323486328123e-06,
      "model_forward_time": 5.941931962966919,
      "step": 2050
    },
    {
      "dynamics_loss_avg": 0.07001891322433948,
      "epoch": 1.9550323486328123e-06,
      "step": 2050
    },
    {
      "action_loss_avg": 0.022258834820240735,
      "epoch": 1.9550323486328123e-06,
      "step": 2050
    },
    {
      "epoch": 1.9550323486328123e-06,
      "step": 2050,
      "training_step_time": 17.452057123184204
    },
    {
      "epoch": 1.955986022949219e-06,
      "model_forward_time": 5.625622749328613,
      "step": 2051
    },
    {
      "epoch": 1.955986022949219e-06,
      "step": 2051,
      "training_step_time": 17.11577010154724
    },
    {
      "epoch": 1.956939697265625e-06,
      "model_forward_time": 5.59141731262207,
      "step": 2052
    },
    {
      "epoch": 1.956939697265625e-06,
      "step": 2052,
      "training_step_time": 17.011963844299316
    },
    {
      "epoch": 1.957893371582031e-06,
      "model_forward_time": 5.573776006698608,
      "step": 2053
    },
    {
      "epoch": 1.957893371582031e-06,
      "step": 2053,
      "training_step_time": 16.97651219367981
    },
    {
      "epoch": 1.9588470458984373e-06,
      "model_forward_time": 5.578815698623657,
      "step": 2054
    },
    {
      "epoch": 1.9588470458984373e-06,
      "step": 2054,
      "training_step_time": 17.010629892349243
    },
    {
      "epoch": 1.959800720214844e-06,
      "model_forward_time": 5.936546802520752,
      "step": 2055
    },
    {
      "epoch": 1.959800720214844e-06,
      "step": 2055,
      "training_step_time": 17.358561277389526
    },
    {
      "epoch": 1.96075439453125e-06,
      "model_forward_time": 5.583416223526001,
      "step": 2056
    },
    {
      "epoch": 1.96075439453125e-06,
      "step": 2056,
      "training_step_time": 16.997631549835205
    },
    {
      "epoch": 1.961708068847656e-06,
      "model_forward_time": 5.571548700332642,
      "step": 2057
    },
    {
      "epoch": 1.961708068847656e-06,
      "step": 2057,
      "training_step_time": 17.005897760391235
    },
    {
      "epoch": 1.9626617431640623e-06,
      "model_forward_time": 5.5745460987091064,
      "step": 2058
    },
    {
      "epoch": 1.9626617431640623e-06,
      "step": 2058,
      "training_step_time": 17.004155158996582
    },
    {
      "epoch": 1.963615417480469e-06,
      "model_forward_time": 5.635586500167847,
      "step": 2059
    },
    {
      "epoch": 1.963615417480469e-06,
      "step": 2059,
      "training_step_time": 17.10431957244873
    },
    {
      "epoch": 1.964569091796875e-06,
      "grad_norm": 0.6340070366859436,
      "learning_rate": 1e-05,
      "loss": 0.0683,
      "step": 2060
    },
    {
      "epoch": 1.964569091796875e-06,
      "model_forward_time": 5.6068150997161865,
      "step": 2060
    },
    {
      "dynamics_loss_avg": 0.05704137682914734,
      "epoch": 1.964569091796875e-06,
      "step": 2060
    },
    {
      "action_loss_avg": 0.015147905820049346,
      "epoch": 1.964569091796875e-06,
      "step": 2060
    },
    {
      "epoch": 1.964569091796875e-06,
      "step": 2060,
      "training_step_time": 17.428663730621338
    },
    {
      "epoch": 1.965522766113281e-06,
      "model_forward_time": 5.595760822296143,
      "step": 2061
    },
    {
      "epoch": 1.965522766113281e-06,
      "step": 2061,
      "training_step_time": 17.065488576889038
    },
    {
      "epoch": 1.9664764404296873e-06,
      "model_forward_time": 5.606378793716431,
      "step": 2062
    },
    {
      "epoch": 1.9664764404296873e-06,
      "step": 2062,
      "training_step_time": 17.05381417274475
    },
    {
      "epoch": 1.967430114746094e-06,
      "model_forward_time": 5.58671236038208,
      "step": 2063
    },
    {
      "epoch": 1.967430114746094e-06,
      "step": 2063,
      "training_step_time": 17.02416729927063
    },
    {
      "epoch": 1.9683837890625e-06,
      "model_forward_time": 5.611387491226196,
      "step": 2064
    },
    {
      "epoch": 1.9683837890625e-06,
      "step": 2064,
      "training_step_time": 17.050659656524658
    },
    {
      "epoch": 1.969337463378906e-06,
      "model_forward_time": 5.629126310348511,
      "step": 2065
    },
    {
      "epoch": 1.969337463378906e-06,
      "step": 2065,
      "training_step_time": 17.124314069747925
    },
    {
      "epoch": 1.9702911376953123e-06,
      "model_forward_time": 5.986748695373535,
      "step": 2066
    },
    {
      "epoch": 1.9702911376953123e-06,
      "step": 2066,
      "training_step_time": 17.469444036483765
    },
    {
      "epoch": 1.971244812011719e-06,
      "model_forward_time": 5.614747047424316,
      "step": 2067
    },
    {
      "epoch": 1.971244812011719e-06,
      "step": 2067,
      "training_step_time": 17.100564002990723
    },
    {
      "epoch": 1.972198486328125e-06,
      "model_forward_time": 5.613249063491821,
      "step": 2068
    },
    {
      "epoch": 1.972198486328125e-06,
      "step": 2068,
      "training_step_time": 17.116766929626465
    },
    {
      "epoch": 1.973152160644531e-06,
      "model_forward_time": 5.613346338272095,
      "step": 2069
    },
    {
      "epoch": 1.973152160644531e-06,
      "step": 2069,
      "training_step_time": 17.096020936965942
    },
    {
      "epoch": 1.9741058349609373e-06,
      "grad_norm": 1.3696588277816772,
      "learning_rate": 1e-05,
      "loss": 0.0982,
      "step": 2070
    },
    {
      "epoch": 1.9741058349609373e-06,
      "model_forward_time": 5.614301681518555,
      "step": 2070
    },
    {
      "dynamics_loss_avg": 0.07484669648110867,
      "epoch": 1.9741058349609373e-06,
      "step": 2070
    },
    {
      "action_loss_avg": 0.020119983656331896,
      "epoch": 1.9741058349609373e-06,
      "step": 2070
    },
    {
      "epoch": 1.9741058349609373e-06,
      "step": 2070,
      "training_step_time": 17.102094888687134
    },
    {
      "epoch": 1.975059509277344e-06,
      "model_forward_time": 5.983132839202881,
      "step": 2071
    },
    {
      "epoch": 1.975059509277344e-06,
      "step": 2071,
      "training_step_time": 17.48040771484375
    },
    {
      "epoch": 1.97601318359375e-06,
      "model_forward_time": 5.6064276695251465,
      "step": 2072
    },
    {
      "epoch": 1.97601318359375e-06,
      "step": 2072,
      "training_step_time": 17.097310543060303
    },
    {
      "epoch": 1.976966857910156e-06,
      "model_forward_time": 5.610613584518433,
      "step": 2073
    },
    {
      "epoch": 1.976966857910156e-06,
      "step": 2073,
      "training_step_time": 17.074703693389893
    },
    {
      "epoch": 1.9779205322265624e-06,
      "model_forward_time": 5.603351354598999,
      "step": 2074
    },
    {
      "epoch": 1.9779205322265624e-06,
      "step": 2074,
      "training_step_time": 17.06920027732849
    },
    {
      "epoch": 1.978874206542969e-06,
      "model_forward_time": 5.5903708934783936,
      "step": 2075
    },
    {
      "epoch": 1.978874206542969e-06,
      "step": 2075,
      "training_step_time": 17.033167839050293
    },
    {
      "epoch": 1.979827880859375e-06,
      "model_forward_time": 5.576455354690552,
      "step": 2076
    },
    {
      "epoch": 1.979827880859375e-06,
      "step": 2076,
      "training_step_time": 16.998525857925415
    },
    {
      "epoch": 1.980781555175781e-06,
      "model_forward_time": 5.9467549324035645,
      "step": 2077
    },
    {
      "epoch": 1.980781555175781e-06,
      "step": 2077,
      "training_step_time": 17.384283304214478
    },
    {
      "epoch": 1.9817352294921874e-06,
      "model_forward_time": 5.576272249221802,
      "step": 2078
    },
    {
      "epoch": 1.9817352294921874e-06,
      "step": 2078,
      "training_step_time": 17.001575469970703
    },
    {
      "epoch": 1.982688903808594e-06,
      "model_forward_time": 5.576411485671997,
      "step": 2079
    },
    {
      "epoch": 1.982688903808594e-06,
      "step": 2079,
      "training_step_time": 17.006532430648804
    },
    {
      "epoch": 1.983642578125e-06,
      "grad_norm": 1.6616599559783936,
      "learning_rate": 1e-05,
      "loss": 0.0819,
      "step": 2080
    },
    {
      "epoch": 1.983642578125e-06,
      "model_forward_time": 5.58516001701355,
      "step": 2080
    },
    {
      "dynamics_loss_avg": 0.059382332302629945,
      "epoch": 1.983642578125e-06,
      "step": 2080
    },
    {
      "action_loss_avg": 0.019899480883032082,
      "epoch": 1.983642578125e-06,
      "step": 2080
    },
    {
      "epoch": 1.983642578125e-06,
      "step": 2080,
      "training_step_time": 17.055402994155884
    },
    {
      "epoch": 1.9845962524414062e-06,
      "model_forward_time": 5.6112518310546875,
      "step": 2081
    },
    {
      "epoch": 1.9845962524414062e-06,
      "step": 2081,
      "training_step_time": 17.122264623641968
    },
    {
      "epoch": 1.9855499267578124e-06,
      "model_forward_time": 5.629991769790649,
      "step": 2082
    },
    {
      "epoch": 1.9855499267578124e-06,
      "step": 2082,
      "training_step_time": 17.514724493026733
    },
    {
      "epoch": 1.986503601074219e-06,
      "model_forward_time": 5.626185894012451,
      "step": 2083
    },
    {
      "epoch": 1.986503601074219e-06,
      "step": 2083,
      "training_step_time": 17.123958587646484
    },
    {
      "epoch": 1.987457275390625e-06,
      "model_forward_time": 5.6085524559021,
      "step": 2084
    },
    {
      "epoch": 1.987457275390625e-06,
      "step": 2084,
      "training_step_time": 17.082884073257446
    },
    {
      "epoch": 1.9884109497070312e-06,
      "model_forward_time": 5.6028571128845215,
      "step": 2085
    },
    {
      "epoch": 1.9884109497070312e-06,
      "step": 2085,
      "training_step_time": 17.06163716316223
    },
    {
      "epoch": 1.9893646240234374e-06,
      "model_forward_time": 5.5960533618927,
      "step": 2086
    },
    {
      "epoch": 1.9893646240234374e-06,
      "step": 2086,
      "training_step_time": 17.03202486038208
    },
    {
      "epoch": 1.990318298339844e-06,
      "model_forward_time": 5.587233304977417,
      "step": 2087
    },
    {
      "epoch": 1.990318298339844e-06,
      "step": 2087,
      "training_step_time": 17.008684396743774
    },
    {
      "epoch": 1.99127197265625e-06,
      "model_forward_time": 5.951656341552734,
      "step": 2088
    },
    {
      "epoch": 1.99127197265625e-06,
      "step": 2088,
      "training_step_time": 17.3719322681427
    },
    {
      "epoch": 1.9922256469726562e-06,
      "model_forward_time": 5.600297927856445,
      "step": 2089
    },
    {
      "epoch": 1.9922256469726562e-06,
      "step": 2089,
      "training_step_time": 17.069703578948975
    },
    {
      "epoch": 1.9931793212890624e-06,
      "grad_norm": 1.661020040512085,
      "learning_rate": 1e-05,
      "loss": 0.0853,
      "step": 2090
    },
    {
      "epoch": 1.9931793212890624e-06,
      "model_forward_time": 5.599327802658081,
      "step": 2090
    },
    {
      "dynamics_loss_avg": 0.06652072370052338,
      "epoch": 1.9931793212890624e-06,
      "step": 2090
    },
    {
      "action_loss_avg": 0.01809469163417816,
      "epoch": 1.9931793212890624e-06,
      "step": 2090
    },
    {
      "epoch": 1.9931793212890624e-06,
      "step": 2090,
      "training_step_time": 17.068909406661987
    },
    {
      "epoch": 1.994132995605469e-06,
      "model_forward_time": 5.598198890686035,
      "step": 2091
    },
    {
      "epoch": 1.994132995605469e-06,
      "step": 2091,
      "training_step_time": 17.07825231552124
    },
    {
      "epoch": 1.995086669921875e-06,
      "model_forward_time": 5.607751131057739,
      "step": 2092
    },
    {
      "epoch": 1.995086669921875e-06,
      "step": 2092,
      "training_step_time": 17.089035272598267
    },
    {
      "epoch": 1.9960403442382812e-06,
      "model_forward_time": 5.96324896812439,
      "step": 2093
    },
    {
      "epoch": 1.9960403442382812e-06,
      "step": 2093,
      "training_step_time": 17.457844018936157
    },
    {
      "epoch": 1.9969940185546874e-06,
      "model_forward_time": 5.611225366592407,
      "step": 2094
    },
    {
      "epoch": 1.9969940185546874e-06,
      "step": 2094,
      "training_step_time": 17.102670431137085
    },
    {
      "epoch": 1.997947692871094e-06,
      "model_forward_time": 5.614508867263794,
      "step": 2095
    },
    {
      "epoch": 1.997947692871094e-06,
      "step": 2095,
      "training_step_time": 17.087700843811035
    },
    {
      "epoch": 1.9989013671875e-06,
      "model_forward_time": 5.595184803009033,
      "step": 2096
    },
    {
      "epoch": 1.9989013671875e-06,
      "step": 2096,
      "training_step_time": 17.029167890548706
    },
    {
      "epoch": 1.9998550415039062e-06,
      "model_forward_time": 5.578771591186523,
      "step": 2097
    },
    {
      "epoch": 1.9998550415039062e-06,
      "step": 2097,
      "training_step_time": 16.997296571731567
    },
    {
      "epoch": 2.0008087158203124e-06,
      "model_forward_time": 5.586655139923096,
      "step": 2098
    },
    {
      "epoch": 2.0008087158203124e-06,
      "step": 2098,
      "training_step_time": 17.015897512435913
    },
    {
      "epoch": 2.001762390136719e-06,
      "model_forward_time": 5.953185319900513,
      "step": 2099
    },
    {
      "epoch": 2.001762390136719e-06,
      "step": 2099,
      "training_step_time": 17.419245958328247
    },
    {
      "epoch": 2.002716064453125e-06,
      "grad_norm": 1.2626343965530396,
      "learning_rate": 1e-05,
      "loss": 0.0849,
      "step": 2100
    },
    {
      "epoch": 2.002716064453125e-06,
      "model_forward_time": 5.606971979141235,
      "step": 2100
    },
    {
      "dynamics_loss_avg": 0.06656989455223083,
      "epoch": 2.002716064453125e-06,
      "step": 2100
    },
    {
      "action_loss_avg": 0.024835567362606524,
      "epoch": 2.002716064453125e-06,
      "step": 2100
    },
    {
      "epoch": 2.002716064453125e-06,
      "step": 2100,
      "training_step_time": 17.09244418144226
    },
    {
      "epoch": 2.0036697387695312e-06,
      "model_forward_time": 5.615339517593384,
      "step": 2101
    },
    {
      "epoch": 2.0036697387695312e-06,
      "step": 2101,
      "training_step_time": 17.080700159072876
    },
    {
      "epoch": 2.0046234130859374e-06,
      "model_forward_time": 5.596335172653198,
      "step": 2102
    },
    {
      "epoch": 2.0046234130859374e-06,
      "step": 2102,
      "training_step_time": 17.020902633666992
    },
    {
      "epoch": 2.005577087402344e-06,
      "model_forward_time": 5.58715295791626,
      "step": 2103
    },
    {
      "epoch": 2.005577087402344e-06,
      "step": 2103,
      "training_step_time": 17.008639097213745
    },
    {
      "epoch": 2.00653076171875e-06,
      "model_forward_time": 5.925396919250488,
      "step": 2104
    },
    {
      "epoch": 2.00653076171875e-06,
      "step": 2104,
      "training_step_time": 17.370170831680298
    },
    {
      "epoch": 2.0074844360351562e-06,
      "model_forward_time": 5.5811474323272705,
      "step": 2105
    },
    {
      "epoch": 2.0074844360351562e-06,
      "step": 2105,
      "training_step_time": 17.012861251831055
    },
    {
      "epoch": 2.0084381103515624e-06,
      "model_forward_time": 5.587087869644165,
      "step": 2106
    },
    {
      "epoch": 2.0084381103515624e-06,
      "step": 2106,
      "training_step_time": 17.06965160369873
    },
    {
      "epoch": 2.009391784667969e-06,
      "model_forward_time": 5.6142494678497314,
      "step": 2107
    },
    {
      "epoch": 2.009391784667969e-06,
      "step": 2107,
      "training_step_time": 17.12347412109375
    },
    {
      "epoch": 2.010345458984375e-06,
      "model_forward_time": 5.616372585296631,
      "step": 2108
    },
    {
      "epoch": 2.010345458984375e-06,
      "step": 2108,
      "training_step_time": 17.101510047912598
    },
    {
      "epoch": 2.0112991333007812e-06,
      "model_forward_time": 5.5938146114349365,
      "step": 2109
    },
    {
      "epoch": 2.0112991333007812e-06,
      "step": 2109,
      "training_step_time": 17.372453451156616
    },
    {
      "epoch": 2.0122528076171874e-06,
      "grad_norm": 2.126481533050537,
      "learning_rate": 1e-05,
      "loss": 0.0924,
      "step": 2110
    },
    {
      "epoch": 2.0122528076171874e-06,
      "model_forward_time": 5.582964897155762,
      "step": 2110
    },
    {
      "dynamics_loss_avg": 0.0723502453416586,
      "epoch": 2.0122528076171874e-06,
      "step": 2110
    },
    {
      "action_loss_avg": 0.02540592704899609,
      "epoch": 2.0122528076171874e-06,
      "step": 2110
    },
    {
      "epoch": 2.0122528076171874e-06,
      "step": 2110,
      "training_step_time": 17.009995222091675
    },
    {
      "epoch": 2.013206481933594e-06,
      "model_forward_time": 5.5843260288238525,
      "step": 2111
    },
    {
      "epoch": 2.013206481933594e-06,
      "step": 2111,
      "training_step_time": 17.0146963596344
    },
    {
      "epoch": 2.01416015625e-06,
      "model_forward_time": 5.572048187255859,
      "step": 2112
    },
    {
      "epoch": 2.01416015625e-06,
      "step": 2112,
      "training_step_time": 17.008975505828857
    },
    {
      "epoch": 2.0151138305664062e-06,
      "model_forward_time": 5.583057641983032,
      "step": 2113
    },
    {
      "epoch": 2.0151138305664062e-06,
      "step": 2113,
      "training_step_time": 17.037614822387695
    },
    {
      "epoch": 2.0160675048828124e-06,
      "model_forward_time": 5.592524290084839,
      "step": 2114
    },
    {
      "epoch": 2.0160675048828124e-06,
      "step": 2114,
      "training_step_time": 17.051799535751343
    },
    {
      "epoch": 2.017021179199219e-06,
      "model_forward_time": 5.943899154663086,
      "step": 2115
    },
    {
      "epoch": 2.017021179199219e-06,
      "step": 2115,
      "training_step_time": 17.375653982162476
    },
    {
      "epoch": 2.017974853515625e-06,
      "model_forward_time": 5.590718746185303,
      "step": 2116
    },
    {
      "epoch": 2.017974853515625e-06,
      "step": 2116,
      "training_step_time": 17.05474829673767
    },
    {
      "epoch": 2.0189285278320312e-06,
      "model_forward_time": 5.593997478485107,
      "step": 2117
    },
    {
      "epoch": 2.0189285278320312e-06,
      "step": 2117,
      "training_step_time": 17.037954807281494
    },
    {
      "epoch": 2.0198822021484374e-06,
      "model_forward_time": 5.594741582870483,
      "step": 2118
    },
    {
      "epoch": 2.0198822021484374e-06,
      "step": 2118,
      "training_step_time": 17.059632301330566
    },
    {
      "epoch": 2.020835876464844e-06,
      "model_forward_time": 5.603735446929932,
      "step": 2119
    },
    {
      "epoch": 2.020835876464844e-06,
      "step": 2119,
      "training_step_time": 17.087975025177002
    },
    {
      "epoch": 2.02178955078125e-06,
      "grad_norm": 0.5518291592597961,
      "learning_rate": 1e-05,
      "loss": 0.0857,
      "step": 2120
    },
    {
      "epoch": 2.02178955078125e-06,
      "model_forward_time": 5.991204738616943,
      "step": 2120
    },
    {
      "dynamics_loss_avg": 0.05990472435951233,
      "epoch": 2.02178955078125e-06,
      "step": 2120
    },
    {
      "action_loss_avg": 0.018158507044427098,
      "epoch": 2.02178955078125e-06,
      "step": 2120
    },
    {
      "epoch": 2.02178955078125e-06,
      "step": 2120,
      "training_step_time": 17.47727060317993
    },
    {
      "epoch": 2.0227432250976562e-06,
      "model_forward_time": 5.6007466316223145,
      "step": 2121
    },
    {
      "epoch": 2.0227432250976562e-06,
      "step": 2121,
      "training_step_time": 17.02911925315857
    },
    {
      "epoch": 2.0236968994140624e-06,
      "model_forward_time": 5.570096254348755,
      "step": 2122
    },
    {
      "epoch": 2.0236968994140624e-06,
      "step": 2122,
      "training_step_time": 16.98195171356201
    },
    {
      "epoch": 2.024650573730469e-06,
      "model_forward_time": 5.5707268714904785,
      "step": 2123
    },
    {
      "epoch": 2.024650573730469e-06,
      "step": 2123,
      "training_step_time": 17.011295080184937
    },
    {
      "epoch": 2.025604248046875e-06,
      "model_forward_time": 5.578616380691528,
      "step": 2124
    },
    {
      "epoch": 2.025604248046875e-06,
      "step": 2124,
      "training_step_time": 17.000359773635864
    },
    {
      "epoch": 2.0265579223632812e-06,
      "model_forward_time": 5.574931383132935,
      "step": 2125
    },
    {
      "epoch": 2.0265579223632812e-06,
      "step": 2125,
      "training_step_time": 17.005330801010132
    },
    {
      "epoch": 2.0275115966796874e-06,
      "model_forward_time": 5.947427272796631,
      "step": 2126
    },
    {
      "epoch": 2.0275115966796874e-06,
      "step": 2126,
      "training_step_time": 17.408417463302612
    },
    {
      "epoch": 2.0284652709960935e-06,
      "model_forward_time": 5.6176512241363525,
      "step": 2127
    },
    {
      "epoch": 2.0284652709960935e-06,
      "step": 2127,
      "training_step_time": 17.12569499015808
    },
    {
      "epoch": 2.0294189453125e-06,
      "model_forward_time": 5.634967088699341,
      "step": 2128
    },
    {
      "epoch": 2.0294189453125e-06,
      "step": 2128,
      "training_step_time": 17.16693878173828
    },
    {
      "epoch": 2.0303726196289063e-06,
      "model_forward_time": 5.61868143081665,
      "step": 2129
    },
    {
      "epoch": 2.0303726196289063e-06,
      "step": 2129,
      "training_step_time": 17.103551626205444
    },
    {
      "epoch": 2.0313262939453124e-06,
      "grad_norm": 2.2566215991973877,
      "learning_rate": 1e-05,
      "loss": 0.1041,
      "step": 2130
    },
    {
      "epoch": 2.0313262939453124e-06,
      "model_forward_time": 5.5901830196380615,
      "step": 2130
    },
    {
      "dynamics_loss_avg": 0.07819609493017196,
      "epoch": 2.0313262939453124e-06,
      "step": 2130
    },
    {
      "action_loss_avg": 0.025190891372039913,
      "epoch": 2.0313262939453124e-06,
      "step": 2130
    },
    {
      "epoch": 2.0313262939453124e-06,
      "step": 2130,
      "training_step_time": 17.018946170806885
    },
    {
      "epoch": 2.0322799682617185e-06,
      "model_forward_time": 5.958258628845215,
      "step": 2131
    },
    {
      "epoch": 2.0322799682617185e-06,
      "step": 2131,
      "training_step_time": 17.386043787002563
    },
    {
      "epoch": 2.033233642578125e-06,
      "model_forward_time": 5.576777458190918,
      "step": 2132
    },
    {
      "epoch": 2.033233642578125e-06,
      "step": 2132,
      "training_step_time": 16.981356143951416
    },
    {
      "epoch": 2.0341873168945313e-06,
      "model_forward_time": 5.589494466781616,
      "step": 2133
    },
    {
      "epoch": 2.0341873168945313e-06,
      "step": 2133,
      "training_step_time": 17.034082412719727
    },
    {
      "epoch": 2.0351409912109374e-06,
      "model_forward_time": 5.591506481170654,
      "step": 2134
    },
    {
      "epoch": 2.0351409912109374e-06,
      "step": 2134,
      "training_step_time": 17.056280612945557
    },
    {
      "epoch": 2.0360946655273435e-06,
      "model_forward_time": 5.6015729904174805,
      "step": 2135
    },
    {
      "epoch": 2.0360946655273435e-06,
      "step": 2135,
      "training_step_time": 17.075424909591675
    },
    {
      "epoch": 2.03704833984375e-06,
      "model_forward_time": 5.59959864616394,
      "step": 2136
    },
    {
      "epoch": 2.03704833984375e-06,
      "step": 2136,
      "training_step_time": 17.08069133758545
    },
    {
      "epoch": 2.0380020141601563e-06,
      "model_forward_time": 5.951476097106934,
      "step": 2137
    },
    {
      "epoch": 2.0380020141601563e-06,
      "step": 2137,
      "training_step_time": 17.413780450820923
    },
    {
      "epoch": 2.0389556884765624e-06,
      "model_forward_time": 5.5895044803619385,
      "step": 2138
    },
    {
      "epoch": 2.0389556884765624e-06,
      "step": 2138,
      "training_step_time": 17.01369285583496
    },
    {
      "epoch": 2.0399093627929686e-06,
      "model_forward_time": 5.589494228363037,
      "step": 2139
    },
    {
      "epoch": 2.0399093627929686e-06,
      "step": 2139,
      "training_step_time": 17.030742168426514
    },
    {
      "epoch": 2.040863037109375e-06,
      "grad_norm": 1.9683610200881958,
      "learning_rate": 1e-05,
      "loss": 0.0856,
      "step": 2140
    },
    {
      "epoch": 2.040863037109375e-06,
      "model_forward_time": 5.596445798873901,
      "step": 2140
    },
    {
      "dynamics_loss_avg": 0.06887709517031908,
      "epoch": 2.040863037109375e-06,
      "step": 2140
    },
    {
      "action_loss_avg": 0.021847125981003045,
      "epoch": 2.040863037109375e-06,
      "step": 2140
    },
    {
      "epoch": 2.040863037109375e-06,
      "step": 2140,
      "training_step_time": 17.07066512107849
    },
    {
      "epoch": 2.0418167114257813e-06,
      "model_forward_time": 5.602924585342407,
      "step": 2141
    },
    {
      "epoch": 2.0418167114257813e-06,
      "step": 2141,
      "training_step_time": 17.083563566207886
    },
    {
      "epoch": 2.0427703857421874e-06,
      "model_forward_time": 5.98665976524353,
      "step": 2142
    },
    {
      "epoch": 2.0427703857421874e-06,
      "step": 2142,
      "training_step_time": 17.474849939346313
    },
    {
      "epoch": 2.0437240600585936e-06,
      "model_forward_time": 5.605324983596802,
      "step": 2143
    },
    {
      "epoch": 2.0437240600585936e-06,
      "step": 2143,
      "training_step_time": 17.104441165924072
    },
    {
      "epoch": 2.044677734375e-06,
      "model_forward_time": 5.611202716827393,
      "step": 2144
    },
    {
      "epoch": 2.044677734375e-06,
      "step": 2144,
      "training_step_time": 17.10057759284973
    },
    {
      "epoch": 2.0456314086914063e-06,
      "model_forward_time": 5.610446453094482,
      "step": 2145
    },
    {
      "epoch": 2.0456314086914063e-06,
      "step": 2145,
      "training_step_time": 17.104051113128662
    },
    {
      "epoch": 2.0465850830078124e-06,
      "model_forward_time": 5.611797094345093,
      "step": 2146
    },
    {
      "epoch": 2.0465850830078124e-06,
      "step": 2146,
      "training_step_time": 17.090843677520752
    },
    {
      "epoch": 2.0475387573242186e-06,
      "model_forward_time": 5.61080265045166,
      "step": 2147
    },
    {
      "epoch": 2.0475387573242186e-06,
      "step": 2147,
      "training_step_time": 17.109142065048218
    },
    {
      "epoch": 2.048492431640625e-06,
      "model_forward_time": 5.949729919433594,
      "step": 2148
    },
    {
      "epoch": 2.048492431640625e-06,
      "step": 2148,
      "training_step_time": 17.44245457649231
    },
    {
      "epoch": 2.0494461059570313e-06,
      "model_forward_time": 5.620496034622192,
      "step": 2149
    },
    {
      "epoch": 2.0494461059570313e-06,
      "step": 2149,
      "training_step_time": 17.1186945438385
    },
    {
      "epoch": 2.0503997802734374e-06,
      "grad_norm": 1.8600072860717773,
      "learning_rate": 1e-05,
      "loss": 0.0699,
      "step": 2150
    },
    {
      "epoch": 2.0503997802734374e-06,
      "model_forward_time": 5.614555835723877,
      "step": 2150
    },
    {
      "dynamics_loss_avg": 0.049831597972661255,
      "epoch": 2.0503997802734374e-06,
      "step": 2150
    },
    {
      "action_loss_avg": 0.01543968110345304,
      "epoch": 2.0503997802734374e-06,
      "step": 2150
    },
    {
      "epoch": 2.0503997802734374e-06,
      "step": 2150,
      "training_step_time": 17.116385221481323
    },
    {
      "epoch": 2.0513534545898436e-06,
      "model_forward_time": 5.618591070175171,
      "step": 2151
    },
    {
      "epoch": 2.0513534545898436e-06,
      "step": 2151,
      "training_step_time": 17.127066612243652
    },
    {
      "epoch": 2.05230712890625e-06,
      "model_forward_time": 5.611385822296143,
      "step": 2152
    },
    {
      "epoch": 2.05230712890625e-06,
      "step": 2152,
      "training_step_time": 17.117056369781494
    },
    {
      "epoch": 2.0532608032226563e-06,
      "model_forward_time": 5.988739490509033,
      "step": 2153
    },
    {
      "epoch": 2.0532608032226563e-06,
      "step": 2153,
      "training_step_time": 17.48418951034546
    },
    {
      "epoch": 2.0542144775390624e-06,
      "model_forward_time": 5.620873212814331,
      "step": 2154
    },
    {
      "epoch": 2.0542144775390624e-06,
      "step": 2154,
      "training_step_time": 17.130295991897583
    },
    {
      "epoch": 2.0551681518554686e-06,
      "model_forward_time": 5.622361660003662,
      "step": 2155
    },
    {
      "epoch": 2.0551681518554686e-06,
      "step": 2155,
      "training_step_time": 17.120247840881348
    },
    {
      "epoch": 2.056121826171875e-06,
      "model_forward_time": 5.6024816036224365,
      "step": 2156
    },
    {
      "epoch": 2.056121826171875e-06,
      "step": 2156,
      "training_step_time": 17.08850407600403
    },
    {
      "epoch": 2.0570755004882813e-06,
      "model_forward_time": 5.607809543609619,
      "step": 2157
    },
    {
      "epoch": 2.0570755004882813e-06,
      "step": 2157,
      "training_step_time": 17.082194328308105
    },
    {
      "epoch": 2.0580291748046874e-06,
      "model_forward_time": 5.95077109336853,
      "step": 2158
    },
    {
      "epoch": 2.0580291748046874e-06,
      "step": 2158,
      "training_step_time": 17.39146852493286
    },
    {
      "epoch": 2.0589828491210936e-06,
      "model_forward_time": 5.598378419876099,
      "step": 2159
    },
    {
      "epoch": 2.0589828491210936e-06,
      "step": 2159,
      "training_step_time": 17.072650909423828
    },
    {
      "epoch": 2.0599365234375e-06,
      "grad_norm": 1.1383696794509888,
      "learning_rate": 1e-05,
      "loss": 0.063,
      "step": 2160
    },
    {
      "epoch": 2.0599365234375e-06,
      "model_forward_time": 5.60925817489624,
      "step": 2160
    },
    {
      "dynamics_loss_avg": 0.056645781919360164,
      "epoch": 2.0599365234375e-06,
      "step": 2160
    },
    {
      "action_loss_avg": 0.012947650672867895,
      "epoch": 2.0599365234375e-06,
      "step": 2160
    },
    {
      "epoch": 2.0599365234375e-06,
      "step": 2160,
      "training_step_time": 17.08512806892395
    },
    {
      "epoch": 2.0608901977539063e-06,
      "model_forward_time": 5.609379529953003,
      "step": 2161
    },
    {
      "epoch": 2.0608901977539063e-06,
      "step": 2161,
      "training_step_time": 17.0913245677948
    },
    {
      "epoch": 2.0618438720703124e-06,
      "model_forward_time": 5.612377405166626,
      "step": 2162
    },
    {
      "epoch": 2.0618438720703124e-06,
      "step": 2162,
      "training_step_time": 17.100281715393066
    },
    {
      "epoch": 2.0627975463867186e-06,
      "model_forward_time": 5.612408876419067,
      "step": 2163
    },
    {
      "epoch": 2.0627975463867186e-06,
      "step": 2163,
      "training_step_time": 17.114426612854004
    },
    {
      "epoch": 2.063751220703125e-06,
      "model_forward_time": 5.985273599624634,
      "step": 2164
    },
    {
      "epoch": 2.063751220703125e-06,
      "step": 2164,
      "training_step_time": 17.46271538734436
    },
    {
      "epoch": 2.0647048950195313e-06,
      "model_forward_time": 5.602855920791626,
      "step": 2165
    },
    {
      "epoch": 2.0647048950195313e-06,
      "step": 2165,
      "training_step_time": 17.0915424823761
    },
    {
      "epoch": 2.0656585693359374e-06,
      "model_forward_time": 5.612093210220337,
      "step": 2166
    },
    {
      "epoch": 2.0656585693359374e-06,
      "step": 2166,
      "training_step_time": 17.082452535629272
    },
    {
      "epoch": 2.0666122436523436e-06,
      "model_forward_time": 5.589773893356323,
      "step": 2167
    },
    {
      "epoch": 2.0666122436523436e-06,
      "step": 2167,
      "training_step_time": 17.00079607963562
    },
    {
      "epoch": 2.06756591796875e-06,
      "model_forward_time": 5.56555962562561,
      "step": 2168
    },
    {
      "epoch": 2.06756591796875e-06,
      "step": 2168,
      "training_step_time": 16.995051622390747
    },
    {
      "epoch": 2.0685195922851563e-06,
      "model_forward_time": 5.5668768882751465,
      "step": 2169
    },
    {
      "epoch": 2.0685195922851563e-06,
      "step": 2169,
      "training_step_time": 17.34433937072754
    },
    {
      "epoch": 2.0694732666015624e-06,
      "grad_norm": 1.0242253541946411,
      "learning_rate": 1e-05,
      "loss": 0.0752,
      "step": 2170
    },
    {
      "epoch": 2.0694732666015624e-06,
      "model_forward_time": 5.571137189865112,
      "step": 2170
    },
    {
      "dynamics_loss_avg": 0.054498161934316156,
      "epoch": 2.0694732666015624e-06,
      "step": 2170
    },
    {
      "action_loss_avg": 0.014861783105880022,
      "epoch": 2.0694732666015624e-06,
      "step": 2170
    },
    {
      "epoch": 2.0694732666015624e-06,
      "step": 2170,
      "training_step_time": 16.97705078125
    },
    {
      "epoch": 2.0704269409179686e-06,
      "model_forward_time": 5.561697483062744,
      "step": 2171
    },
    {
      "epoch": 2.0704269409179686e-06,
      "step": 2171,
      "training_step_time": 16.97380518913269
    },
    {
      "epoch": 2.071380615234375e-06,
      "model_forward_time": 5.581418991088867,
      "step": 2172
    },
    {
      "epoch": 2.071380615234375e-06,
      "step": 2172,
      "training_step_time": 17.022202491760254
    },
    {
      "epoch": 2.0723342895507813e-06,
      "model_forward_time": 5.570635557174683,
      "step": 2173
    },
    {
      "epoch": 2.0723342895507813e-06,
      "step": 2173,
      "training_step_time": 17.023524284362793
    },
    {
      "epoch": 2.0732879638671874e-06,
      "model_forward_time": 5.598999977111816,
      "step": 2174
    },
    {
      "epoch": 2.0732879638671874e-06,
      "step": 2174,
      "training_step_time": 17.081937551498413
    },
    {
      "epoch": 2.0742416381835936e-06,
      "model_forward_time": 5.975257873535156,
      "step": 2175
    },
    {
      "epoch": 2.0742416381835936e-06,
      "step": 2175,
      "training_step_time": 17.476338386535645
    },
    {
      "epoch": 2.0751953125e-06,
      "model_forward_time": 5.618509292602539,
      "step": 2176
    },
    {
      "epoch": 2.0751953125e-06,
      "step": 2176,
      "training_step_time": 17.114556074142456
    },
    {
      "epoch": 2.0761489868164063e-06,
      "model_forward_time": 5.620693683624268,
      "step": 2177
    },
    {
      "epoch": 2.0761489868164063e-06,
      "step": 2177,
      "training_step_time": 17.13093090057373
    },
    {
      "epoch": 2.0771026611328124e-06,
      "model_forward_time": 5.626757383346558,
      "step": 2178
    },
    {
      "epoch": 2.0771026611328124e-06,
      "step": 2178,
      "training_step_time": 17.133341073989868
    },
    {
      "epoch": 2.0780563354492186e-06,
      "model_forward_time": 5.608179807662964,
      "step": 2179
    },
    {
      "epoch": 2.0780563354492186e-06,
      "step": 2179,
      "training_step_time": 17.076056241989136
    },
    {
      "epoch": 2.079010009765625e-06,
      "grad_norm": 1.5190119743347168,
      "learning_rate": 1e-05,
      "loss": 0.0805,
      "step": 2180
    },
    {
      "epoch": 2.079010009765625e-06,
      "model_forward_time": 5.604453086853027,
      "step": 2180
    },
    {
      "dynamics_loss_avg": 0.059036165848374365,
      "epoch": 2.079010009765625e-06,
      "step": 2180
    },
    {
      "action_loss_avg": 0.01920118648558855,
      "epoch": 2.079010009765625e-06,
      "step": 2180
    },
    {
      "epoch": 2.079010009765625e-06,
      "step": 2180,
      "training_step_time": 17.40265130996704
    },
    {
      "epoch": 2.0799636840820313e-06,
      "model_forward_time": 5.5907301902771,
      "step": 2181
    },
    {
      "epoch": 2.0799636840820313e-06,
      "step": 2181,
      "training_step_time": 17.036503076553345
    },
    {
      "epoch": 2.0809173583984374e-06,
      "model_forward_time": 5.5990777015686035,
      "step": 2182
    },
    {
      "epoch": 2.0809173583984374e-06,
      "step": 2182,
      "training_step_time": 17.039987564086914
    },
    {
      "epoch": 2.0818710327148436e-06,
      "model_forward_time": 5.589608907699585,
      "step": 2183
    },
    {
      "epoch": 2.0818710327148436e-06,
      "step": 2183,
      "training_step_time": 17.049018383026123
    },
    {
      "epoch": 2.08282470703125e-06,
      "model_forward_time": 5.599602937698364,
      "step": 2184
    },
    {
      "epoch": 2.08282470703125e-06,
      "step": 2184,
      "training_step_time": 17.06287431716919
    },
    {
      "epoch": 2.0837783813476563e-06,
      "model_forward_time": 5.61204719543457,
      "step": 2185
    },
    {
      "epoch": 2.0837783813476563e-06,
      "step": 2185,
      "training_step_time": 17.102200031280518
    },
    {
      "epoch": 2.0847320556640624e-06,
      "model_forward_time": 5.980714797973633,
      "step": 2186
    },
    {
      "epoch": 2.0847320556640624e-06,
      "step": 2186,
      "training_step_time": 17.490954875946045
    },
    {
      "epoch": 2.0856857299804686e-06,
      "model_forward_time": 5.617225408554077,
      "step": 2187
    },
    {
      "epoch": 2.0856857299804686e-06,
      "step": 2187,
      "training_step_time": 17.119953632354736
    },
    {
      "epoch": 2.086639404296875e-06,
      "model_forward_time": 5.608513355255127,
      "step": 2188
    },
    {
      "epoch": 2.086639404296875e-06,
      "step": 2188,
      "training_step_time": 17.100639820098877
    },
    {
      "epoch": 2.0875930786132813e-06,
      "model_forward_time": 5.607353448867798,
      "step": 2189
    },
    {
      "epoch": 2.0875930786132813e-06,
      "step": 2189,
      "training_step_time": 17.092665672302246
    },
    {
      "epoch": 2.0885467529296874e-06,
      "grad_norm": 0.9383519291877747,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 2190
    },
    {
      "epoch": 2.0885467529296874e-06,
      "model_forward_time": 5.621855974197388,
      "step": 2190
    },
    {
      "dynamics_loss_avg": 0.05468380376696587,
      "epoch": 2.0885467529296874e-06,
      "step": 2190
    },
    {
      "action_loss_avg": 0.012480558338575065,
      "epoch": 2.0885467529296874e-06,
      "step": 2190
    },
    {
      "epoch": 2.0885467529296874e-06,
      "step": 2190,
      "training_step_time": 17.089450120925903
    },
    {
      "epoch": 2.0895004272460936e-06,
      "model_forward_time": 5.946813106536865,
      "step": 2191
    },
    {
      "epoch": 2.0895004272460936e-06,
      "step": 2191,
      "training_step_time": 17.39072012901306
    },
    {
      "epoch": 2.0904541015625e-06,
      "model_forward_time": 5.582315683364868,
      "step": 2192
    },
    {
      "epoch": 2.0904541015625e-06,
      "step": 2192,
      "training_step_time": 17.020764112472534
    },
    {
      "epoch": 2.0914077758789063e-06,
      "model_forward_time": 5.5947020053863525,
      "step": 2193
    },
    {
      "epoch": 2.0914077758789063e-06,
      "step": 2193,
      "training_step_time": 17.07190227508545
    },
    {
      "epoch": 2.0923614501953125e-06,
      "model_forward_time": 5.607213258743286,
      "step": 2194
    },
    {
      "epoch": 2.0923614501953125e-06,
      "step": 2194,
      "training_step_time": 17.097996473312378
    },
    {
      "epoch": 2.0933151245117186e-06,
      "model_forward_time": 5.61389684677124,
      "step": 2195
    },
    {
      "epoch": 2.0933151245117186e-06,
      "step": 2195,
      "training_step_time": 17.081848859786987
    },
    {
      "epoch": 2.094268798828125e-06,
      "model_forward_time": 5.576931953430176,
      "step": 2196
    },
    {
      "epoch": 2.094268798828125e-06,
      "step": 2196,
      "training_step_time": 16.99608087539673
    },
    {
      "epoch": 2.0952224731445313e-06,
      "model_forward_time": 5.941158056259155,
      "step": 2197
    },
    {
      "epoch": 2.0952224731445313e-06,
      "step": 2197,
      "training_step_time": 17.356444358825684
    },
    {
      "epoch": 2.0961761474609375e-06,
      "model_forward_time": 5.579656600952148,
      "step": 2198
    },
    {
      "epoch": 2.0961761474609375e-06,
      "step": 2198,
      "training_step_time": 17.005616188049316
    },
    {
      "epoch": 2.0971298217773436e-06,
      "model_forward_time": 5.568625450134277,
      "step": 2199
    },
    {
      "epoch": 2.0971298217773436e-06,
      "step": 2199,
      "training_step_time": 17.005765676498413
    },
    {
      "epoch": 2.09808349609375e-06,
      "grad_norm": 2.169245481491089,
      "learning_rate": 1e-05,
      "loss": 0.0904,
      "step": 2200
    },
    {
      "epoch": 2.09808349609375e-06,
      "model_forward_time": 5.576742172241211,
      "step": 2200
    },
    {
      "dynamics_loss_avg": 0.07181354202330112,
      "epoch": 2.09808349609375e-06,
      "step": 2200
    },
    {
      "action_loss_avg": 0.023282105335965753,
      "epoch": 2.09808349609375e-06,
      "step": 2200
    },
    {
      "epoch": 2.09808349609375e-06,
      "step": 2200,
      "training_step_time": 16.981919527053833
    },
    {
      "epoch": 2.0990371704101563e-06,
      "model_forward_time": 5.5822577476501465,
      "step": 2201
    },
    {
      "epoch": 2.0990371704101563e-06,
      "step": 2201,
      "training_step_time": 17.031155824661255
    },
    {
      "epoch": 2.0999908447265625e-06,
      "model_forward_time": 5.5895960330963135,
      "step": 2202
    },
    {
      "epoch": 2.0999908447265625e-06,
      "step": 2202,
      "training_step_time": 17.385146379470825
    },
    {
      "epoch": 2.1009445190429686e-06,
      "model_forward_time": 5.600390672683716,
      "step": 2203
    },
    {
      "epoch": 2.1009445190429686e-06,
      "step": 2203,
      "training_step_time": 17.05455732345581
    },
    {
      "epoch": 2.101898193359375e-06,
      "model_forward_time": 5.602344751358032,
      "step": 2204
    },
    {
      "epoch": 2.101898193359375e-06,
      "step": 2204,
      "training_step_time": 17.072781085968018
    },
    {
      "epoch": 2.1028518676757813e-06,
      "model_forward_time": 5.5958192348480225,
      "step": 2205
    },
    {
      "epoch": 2.1028518676757813e-06,
      "step": 2205,
      "training_step_time": 17.05831480026245
    },
    {
      "epoch": 2.1038055419921875e-06,
      "model_forward_time": 5.599685192108154,
      "step": 2206
    },
    {
      "epoch": 2.1038055419921875e-06,
      "step": 2206,
      "training_step_time": 17.06601309776306
    },
    {
      "epoch": 2.1047592163085936e-06,
      "model_forward_time": 5.586487770080566,
      "step": 2207
    },
    {
      "epoch": 2.1047592163085936e-06,
      "step": 2207,
      "training_step_time": 17.042750358581543
    },
    {
      "epoch": 2.105712890625e-06,
      "model_forward_time": 5.967611789703369,
      "step": 2208
    },
    {
      "epoch": 2.105712890625e-06,
      "step": 2208,
      "training_step_time": 17.45500111579895
    },
    {
      "epoch": 2.1066665649414063e-06,
      "model_forward_time": 5.615272045135498,
      "step": 2209
    },
    {
      "epoch": 2.1066665649414063e-06,
      "step": 2209,
      "training_step_time": 17.1114559173584
    },
    {
      "epoch": 2.1076202392578125e-06,
      "grad_norm": 0.9340147376060486,
      "learning_rate": 1e-05,
      "loss": 0.0973,
      "step": 2210
    },
    {
      "epoch": 2.1076202392578125e-06,
      "model_forward_time": 5.608694314956665,
      "step": 2210
    },
    {
      "dynamics_loss_avg": 0.07654805593192578,
      "epoch": 2.1076202392578125e-06,
      "step": 2210
    },
    {
      "action_loss_avg": 0.022185444459319116,
      "epoch": 2.1076202392578125e-06,
      "step": 2210
    },
    {
      "epoch": 2.1076202392578125e-06,
      "step": 2210,
      "training_step_time": 17.04290270805359
    },
    {
      "epoch": 2.1085739135742186e-06,
      "model_forward_time": 5.578976154327393,
      "step": 2211
    },
    {
      "epoch": 2.1085739135742186e-06,
      "step": 2211,
      "training_step_time": 16.986526489257812
    },
    {
      "epoch": 2.109527587890625e-06,
      "model_forward_time": 5.59035325050354,
      "step": 2212
    },
    {
      "epoch": 2.109527587890625e-06,
      "step": 2212,
      "training_step_time": 17.017653465270996
    },
    {
      "epoch": 2.1104812622070313e-06,
      "model_forward_time": 5.582958221435547,
      "step": 2213
    },
    {
      "epoch": 2.1104812622070313e-06,
      "step": 2213,
      "training_step_time": 17.396890878677368
    },
    {
      "epoch": 2.1114349365234375e-06,
      "model_forward_time": 5.580654621124268,
      "step": 2214
    },
    {
      "epoch": 2.1114349365234375e-06,
      "step": 2214,
      "training_step_time": 17.00901222229004
    },
    {
      "epoch": 2.1123886108398436e-06,
      "model_forward_time": 5.57779598236084,
      "step": 2215
    },
    {
      "epoch": 2.1123886108398436e-06,
      "step": 2215,
      "training_step_time": 17.02344012260437
    },
    {
      "epoch": 2.11334228515625e-06,
      "model_forward_time": 5.605578422546387,
      "step": 2216
    },
    {
      "epoch": 2.11334228515625e-06,
      "step": 2216,
      "training_step_time": 17.08858895301819
    },
    {
      "epoch": 2.1142959594726563e-06,
      "model_forward_time": 5.607747554779053,
      "step": 2217
    },
    {
      "epoch": 2.1142959594726563e-06,
      "step": 2217,
      "training_step_time": 17.08832859992981
    },
    {
      "epoch": 2.1152496337890625e-06,
      "model_forward_time": 5.608134746551514,
      "step": 2218
    },
    {
      "epoch": 2.1152496337890625e-06,
      "step": 2218,
      "training_step_time": 17.094932794570923
    },
    {
      "epoch": 2.1162033081054686e-06,
      "model_forward_time": 5.9662415981292725,
      "step": 2219
    },
    {
      "epoch": 2.1162033081054686e-06,
      "step": 2219,
      "training_step_time": 17.452028512954712
    },
    {
      "epoch": 2.117156982421875e-06,
      "grad_norm": 1.4210395812988281,
      "learning_rate": 1e-05,
      "loss": 0.0776,
      "step": 2220
    },
    {
      "epoch": 2.117156982421875e-06,
      "model_forward_time": 5.612822771072388,
      "step": 2220
    },
    {
      "dynamics_loss_avg": 0.05437321923673153,
      "epoch": 2.117156982421875e-06,
      "step": 2220
    },
    {
      "action_loss_avg": 0.014528384106233716,
      "epoch": 2.117156982421875e-06,
      "step": 2220
    },
    {
      "epoch": 2.117156982421875e-06,
      "step": 2220,
      "training_step_time": 17.103785514831543
    },
    {
      "epoch": 2.1181106567382813e-06,
      "model_forward_time": 5.609232425689697,
      "step": 2221
    },
    {
      "epoch": 2.1181106567382813e-06,
      "step": 2221,
      "training_step_time": 17.103744745254517
    },
    {
      "epoch": 2.1190643310546875e-06,
      "model_forward_time": 5.61122989654541,
      "step": 2222
    },
    {
      "epoch": 2.1190643310546875e-06,
      "step": 2222,
      "training_step_time": 17.083409309387207
    },
    {
      "epoch": 2.1200180053710936e-06,
      "model_forward_time": 5.594753980636597,
      "step": 2223
    },
    {
      "epoch": 2.1200180053710936e-06,
      "step": 2223,
      "training_step_time": 17.03691267967224
    },
    {
      "epoch": 2.1209716796875e-06,
      "model_forward_time": 5.929994583129883,
      "step": 2224
    },
    {
      "epoch": 2.1209716796875e-06,
      "step": 2224,
      "training_step_time": 17.355491161346436
    },
    {
      "epoch": 2.1219253540039063e-06,
      "model_forward_time": 5.573101997375488,
      "step": 2225
    },
    {
      "epoch": 2.1219253540039063e-06,
      "step": 2225,
      "training_step_time": 17.00833797454834
    },
    {
      "epoch": 2.1228790283203125e-06,
      "model_forward_time": 5.572848558425903,
      "step": 2226
    },
    {
      "epoch": 2.1228790283203125e-06,
      "step": 2226,
      "training_step_time": 16.98458766937256
    },
    {
      "epoch": 2.1238327026367186e-06,
      "model_forward_time": 5.578548431396484,
      "step": 2227
    },
    {
      "epoch": 2.1238327026367186e-06,
      "step": 2227,
      "training_step_time": 17.002779245376587
    },
    {
      "epoch": 2.124786376953125e-06,
      "model_forward_time": 5.580458164215088,
      "step": 2228
    },
    {
      "epoch": 2.124786376953125e-06,
      "step": 2228,
      "training_step_time": 17.010146617889404
    },
    {
      "epoch": 2.1257400512695313e-06,
      "model_forward_time": 5.600552082061768,
      "step": 2229
    },
    {
      "epoch": 2.1257400512695313e-06,
      "step": 2229,
      "training_step_time": 17.09031844139099
    },
    {
      "epoch": 2.1266937255859375e-06,
      "grad_norm": 1.6557025909423828,
      "learning_rate": 1e-05,
      "loss": 0.0791,
      "step": 2230
    },
    {
      "epoch": 2.1266937255859375e-06,
      "model_forward_time": 5.992475271224976,
      "step": 2230
    },
    {
      "dynamics_loss_avg": 0.06171432621777058,
      "epoch": 2.1266937255859375e-06,
      "step": 2230
    },
    {
      "action_loss_avg": 0.019064740044996142,
      "epoch": 2.1266937255859375e-06,
      "step": 2230
    },
    {
      "epoch": 2.1266937255859375e-06,
      "step": 2230,
      "training_step_time": 17.503787994384766
    },
    {
      "epoch": 2.1276473999023436e-06,
      "model_forward_time": 5.626554727554321,
      "step": 2231
    },
    {
      "epoch": 2.1276473999023436e-06,
      "step": 2231,
      "training_step_time": 17.107468128204346
    },
    {
      "epoch": 2.12860107421875e-06,
      "model_forward_time": 5.591230630874634,
      "step": 2232
    },
    {
      "epoch": 2.12860107421875e-06,
      "step": 2232,
      "training_step_time": 17.011746644973755
    },
    {
      "epoch": 2.1295547485351563e-06,
      "model_forward_time": 5.5705766677856445,
      "step": 2233
    },
    {
      "epoch": 2.1295547485351563e-06,
      "step": 2233,
      "training_step_time": 17.01175308227539
    },
    {
      "epoch": 2.1305084228515625e-06,
      "model_forward_time": 5.572281360626221,
      "step": 2234
    },
    {
      "epoch": 2.1305084228515625e-06,
      "step": 2234,
      "training_step_time": 16.98069953918457
    },
    {
      "epoch": 2.1314620971679686e-06,
      "model_forward_time": 5.58669638633728,
      "step": 2235
    },
    {
      "epoch": 2.1314620971679686e-06,
      "step": 2235,
      "training_step_time": 17.369905710220337
    },
    {
      "epoch": 2.132415771484375e-06,
      "model_forward_time": 5.5727174282073975,
      "step": 2236
    },
    {
      "epoch": 2.132415771484375e-06,
      "step": 2236,
      "training_step_time": 16.984779596328735
    },
    {
      "epoch": 2.1333694458007813e-06,
      "model_forward_time": 5.577014207839966,
      "step": 2237
    },
    {
      "epoch": 2.1333694458007813e-06,
      "step": 2237,
      "training_step_time": 17.016024112701416
    },
    {
      "epoch": 2.1343231201171875e-06,
      "model_forward_time": 5.586681842803955,
      "step": 2238
    },
    {
      "epoch": 2.1343231201171875e-06,
      "step": 2238,
      "training_step_time": 17.015442848205566
    },
    {
      "epoch": 2.1352767944335936e-06,
      "model_forward_time": 5.59479832649231,
      "step": 2239
    },
    {
      "epoch": 2.1352767944335936e-06,
      "step": 2239,
      "training_step_time": 17.049927949905396
    },
    {
      "epoch": 2.13623046875e-06,
      "grad_norm": 0.5016259551048279,
      "learning_rate": 1e-05,
      "loss": 0.0921,
      "step": 2240
    },
    {
      "epoch": 2.13623046875e-06,
      "model_forward_time": 5.598102331161499,
      "step": 2240
    },
    {
      "dynamics_loss_avg": 0.06865062899887561,
      "epoch": 2.13623046875e-06,
      "step": 2240
    },
    {
      "action_loss_avg": 0.020778086176142095,
      "epoch": 2.13623046875e-06,
      "step": 2240
    },
    {
      "epoch": 2.13623046875e-06,
      "step": 2240,
      "training_step_time": 17.089925289154053
    },
    {
      "epoch": 2.1371841430664063e-06,
      "model_forward_time": 5.986040830612183,
      "step": 2241
    },
    {
      "epoch": 2.1371841430664063e-06,
      "step": 2241,
      "training_step_time": 17.481093883514404
    },
    {
      "epoch": 2.1381378173828125e-06,
      "model_forward_time": 5.604259014129639,
      "step": 2242
    },
    {
      "epoch": 2.1381378173828125e-06,
      "step": 2242,
      "training_step_time": 17.07146406173706
    },
    {
      "epoch": 2.1390914916992186e-06,
      "model_forward_time": 5.596123456954956,
      "step": 2243
    },
    {
      "epoch": 2.1390914916992186e-06,
      "step": 2243,
      "training_step_time": 17.045064687728882
    },
    {
      "epoch": 2.140045166015625e-06,
      "model_forward_time": 5.576333999633789,
      "step": 2244
    },
    {
      "epoch": 2.140045166015625e-06,
      "step": 2244,
      "training_step_time": 17.01455593109131
    },
    {
      "epoch": 2.1409988403320314e-06,
      "model_forward_time": 5.582854270935059,
      "step": 2245
    },
    {
      "epoch": 2.1409988403320314e-06,
      "step": 2245,
      "training_step_time": 17.047406673431396
    },
    {
      "epoch": 2.1419525146484375e-06,
      "model_forward_time": 5.602087497711182,
      "step": 2246
    },
    {
      "epoch": 2.1419525146484375e-06,
      "step": 2246,
      "training_step_time": 17.45694351196289
    },
    {
      "epoch": 2.1429061889648436e-06,
      "model_forward_time": 5.627985000610352,
      "step": 2247
    },
    {
      "epoch": 2.1429061889648436e-06,
      "step": 2247,
      "training_step_time": 17.13531470298767
    },
    {
      "epoch": 2.14385986328125e-06,
      "model_forward_time": 5.613511323928833,
      "step": 2248
    },
    {
      "epoch": 2.14385986328125e-06,
      "step": 2248,
      "training_step_time": 17.107963800430298
    },
    {
      "epoch": 2.1448135375976564e-06,
      "model_forward_time": 5.6154515743255615,
      "step": 2249
    },
    {
      "epoch": 2.1448135375976564e-06,
      "step": 2249,
      "training_step_time": 17.100215435028076
    },
    {
      "epoch": 2.1457672119140625e-06,
      "grad_norm": 1.2624722719192505,
      "learning_rate": 1e-05,
      "loss": 0.0754,
      "step": 2250
    },
    {
      "epoch": 2.1457672119140625e-06,
      "model_forward_time": 5.601046800613403,
      "step": 2250
    },
    {
      "dynamics_loss_avg": 0.06010808553546667,
      "epoch": 2.1457672119140625e-06,
      "step": 2250
    },
    {
      "action_loss_avg": 0.018529020063579083,
      "epoch": 2.1457672119140625e-06,
      "step": 2250
    },
    {
      "epoch": 2.1457672119140625e-06,
      "step": 2250,
      "training_step_time": 17.05839490890503
    },
    {
      "epoch": 2.1467208862304686e-06,
      "model_forward_time": 5.597934722900391,
      "step": 2251
    },
    {
      "epoch": 2.1467208862304686e-06,
      "step": 2251,
      "training_step_time": 17.05381178855896
    },
    {
      "epoch": 2.147674560546875e-06,
      "model_forward_time": 5.949889421463013,
      "step": 2252
    },
    {
      "epoch": 2.147674560546875e-06,
      "step": 2252,
      "training_step_time": 17.41397786140442
    },
    {
      "epoch": 2.1486282348632814e-06,
      "model_forward_time": 5.617022275924683,
      "step": 2253
    },
    {
      "epoch": 2.1486282348632814e-06,
      "step": 2253,
      "training_step_time": 17.097244262695312
    },
    {
      "epoch": 2.1495819091796875e-06,
      "model_forward_time": 5.611863851547241,
      "step": 2254
    },
    {
      "epoch": 2.1495819091796875e-06,
      "step": 2254,
      "training_step_time": 17.07675814628601
    },
    {
      "epoch": 2.1505355834960936e-06,
      "model_forward_time": 5.587683200836182,
      "step": 2255
    },
    {
      "epoch": 2.1505355834960936e-06,
      "step": 2255,
      "training_step_time": 17.014894247055054
    },
    {
      "epoch": 2.1514892578125e-06,
      "model_forward_time": 5.587600946426392,
      "step": 2256
    },
    {
      "epoch": 2.1514892578125e-06,
      "step": 2256,
      "training_step_time": 17.031035900115967
    },
    {
      "epoch": 2.1524429321289064e-06,
      "model_forward_time": 5.930980205535889,
      "step": 2257
    },
    {
      "epoch": 2.1524429321289064e-06,
      "step": 2257,
      "training_step_time": 17.366599082946777
    },
    {
      "epoch": 2.1533966064453125e-06,
      "model_forward_time": 5.596254825592041,
      "step": 2258
    },
    {
      "epoch": 2.1533966064453125e-06,
      "step": 2258,
      "training_step_time": 17.03687310218811
    },
    {
      "epoch": 2.1543502807617187e-06,
      "model_forward_time": 5.5977089405059814,
      "step": 2259
    },
    {
      "epoch": 2.1543502807617187e-06,
      "step": 2259,
      "training_step_time": 17.035288095474243
    },
    {
      "epoch": 2.155303955078125e-06,
      "grad_norm": 1.2573330402374268,
      "learning_rate": 1e-05,
      "loss": 0.0844,
      "step": 2260
    },
    {
      "epoch": 2.155303955078125e-06,
      "model_forward_time": 5.592751502990723,
      "step": 2260
    },
    {
      "dynamics_loss_avg": 0.06830989848822355,
      "epoch": 2.155303955078125e-06,
      "step": 2260
    },
    {
      "action_loss_avg": 0.023384025087580085,
      "epoch": 2.155303955078125e-06,
      "step": 2260
    },
    {
      "epoch": 2.155303955078125e-06,
      "step": 2260,
      "training_step_time": 17.05251955986023
    },
    {
      "epoch": 2.1562576293945314e-06,
      "model_forward_time": 5.597455024719238,
      "step": 2261
    },
    {
      "epoch": 2.1562576293945314e-06,
      "step": 2261,
      "training_step_time": 17.06309461593628
    },
    {
      "epoch": 2.1572113037109375e-06,
      "model_forward_time": 5.6075921058654785,
      "step": 2262
    },
    {
      "epoch": 2.1572113037109375e-06,
      "step": 2262,
      "training_step_time": 17.090434789657593
    },
    {
      "epoch": 2.1581649780273437e-06,
      "model_forward_time": 5.984216213226318,
      "step": 2263
    },
    {
      "epoch": 2.1581649780273437e-06,
      "step": 2263,
      "training_step_time": 17.459943294525146
    },
    {
      "epoch": 2.15911865234375e-06,
      "model_forward_time": 5.600835800170898,
      "step": 2264
    },
    {
      "epoch": 2.15911865234375e-06,
      "step": 2264,
      "training_step_time": 17.070434093475342
    },
    {
      "epoch": 2.1600723266601564e-06,
      "model_forward_time": 5.598244905471802,
      "step": 2265
    },
    {
      "epoch": 2.1600723266601564e-06,
      "step": 2265,
      "training_step_time": 17.07028865814209
    },
    {
      "epoch": 2.1610260009765625e-06,
      "model_forward_time": 5.609182834625244,
      "step": 2266
    },
    {
      "epoch": 2.1610260009765625e-06,
      "step": 2266,
      "training_step_time": 17.106797456741333
    },
    {
      "epoch": 2.1619796752929687e-06,
      "model_forward_time": 5.6175642013549805,
      "step": 2267
    },
    {
      "epoch": 2.1619796752929687e-06,
      "step": 2267,
      "training_step_time": 17.08752465248108
    },
    {
      "epoch": 2.162933349609375e-06,
      "model_forward_time": 5.626819849014282,
      "step": 2268
    },
    {
      "epoch": 2.162933349609375e-06,
      "step": 2268,
      "training_step_time": 17.466127634048462
    },
    {
      "epoch": 2.1638870239257814e-06,
      "model_forward_time": 5.59661602973938,
      "step": 2269
    },
    {
      "epoch": 2.1638870239257814e-06,
      "step": 2269,
      "training_step_time": 17.02007746696472
    },
    {
      "epoch": 2.1648406982421875e-06,
      "grad_norm": 1.2859679460525513,
      "learning_rate": 1e-05,
      "loss": 0.0849,
      "step": 2270
    },
    {
      "epoch": 2.1648406982421875e-06,
      "model_forward_time": 5.5898802280426025,
      "step": 2270
    },
    {
      "dynamics_loss_avg": 0.06086038909852505,
      "epoch": 2.1648406982421875e-06,
      "step": 2270
    },
    {
      "action_loss_avg": 0.01477574286982417,
      "epoch": 2.1648406982421875e-06,
      "step": 2270
    },
    {
      "epoch": 2.1648406982421875e-06,
      "step": 2270,
      "training_step_time": 17.024980068206787
    },
    {
      "epoch": 2.1657943725585937e-06,
      "model_forward_time": 5.586164951324463,
      "step": 2271
    },
    {
      "epoch": 2.1657943725585937e-06,
      "step": 2271,
      "training_step_time": 17.023526430130005
    },
    {
      "epoch": 2.166748046875e-06,
      "model_forward_time": 5.581455707550049,
      "step": 2272
    },
    {
      "epoch": 2.166748046875e-06,
      "step": 2272,
      "training_step_time": 16.99901032447815
    },
    {
      "epoch": 2.1677017211914064e-06,
      "model_forward_time": 5.597059726715088,
      "step": 2273
    },
    {
      "epoch": 2.1677017211914064e-06,
      "step": 2273,
      "training_step_time": 17.043131589889526
    },
    {
      "epoch": 2.1686553955078125e-06,
      "model_forward_time": 5.954877853393555,
      "step": 2274
    },
    {
      "epoch": 2.1686553955078125e-06,
      "step": 2274,
      "training_step_time": 17.416691064834595
    },
    {
      "epoch": 2.1696090698242187e-06,
      "model_forward_time": 5.606230020523071,
      "step": 2275
    },
    {
      "epoch": 2.1696090698242187e-06,
      "step": 2275,
      "training_step_time": 17.09706711769104
    },
    {
      "epoch": 2.170562744140625e-06,
      "model_forward_time": 5.605712652206421,
      "step": 2276
    },
    {
      "epoch": 2.170562744140625e-06,
      "step": 2276,
      "training_step_time": 17.103426933288574
    },
    {
      "epoch": 2.1715164184570314e-06,
      "model_forward_time": 5.613344192504883,
      "step": 2277
    },
    {
      "epoch": 2.1715164184570314e-06,
      "step": 2277,
      "training_step_time": 17.097564697265625
    },
    {
      "epoch": 2.1724700927734375e-06,
      "model_forward_time": 5.59869122505188,
      "step": 2278
    },
    {
      "epoch": 2.1724700927734375e-06,
      "step": 2278,
      "training_step_time": 17.039108276367188
    },
    {
      "epoch": 2.1734237670898437e-06,
      "model_forward_time": 5.584851503372192,
      "step": 2279
    },
    {
      "epoch": 2.1734237670898437e-06,
      "step": 2279,
      "training_step_time": 17.38461399078369
    },
    {
      "epoch": 2.17437744140625e-06,
      "grad_norm": 1.0105258226394653,
      "learning_rate": 1e-05,
      "loss": 0.0716,
      "step": 2280
    },
    {
      "epoch": 2.17437744140625e-06,
      "model_forward_time": 5.594677686691284,
      "step": 2280
    },
    {
      "dynamics_loss_avg": 0.06161257773637772,
      "epoch": 2.17437744140625e-06,
      "step": 2280
    },
    {
      "action_loss_avg": 0.017940676654689014,
      "epoch": 2.17437744140625e-06,
      "step": 2280
    },
    {
      "epoch": 2.17437744140625e-06,
      "step": 2280,
      "training_step_time": 17.062151432037354
    },
    {
      "epoch": 2.1753311157226564e-06,
      "model_forward_time": 5.609049081802368,
      "step": 2281
    },
    {
      "epoch": 2.1753311157226564e-06,
      "step": 2281,
      "training_step_time": 17.106050491333008
    },
    {
      "epoch": 2.1762847900390625e-06,
      "model_forward_time": 5.6134114265441895,
      "step": 2282
    },
    {
      "epoch": 2.1762847900390625e-06,
      "step": 2282,
      "training_step_time": 17.095163345336914
    },
    {
      "epoch": 2.1772384643554687e-06,
      "model_forward_time": 5.595052242279053,
      "step": 2283
    },
    {
      "epoch": 2.1772384643554687e-06,
      "step": 2283,
      "training_step_time": 17.017574548721313
    },
    {
      "epoch": 2.178192138671875e-06,
      "model_forward_time": 5.58838677406311,
      "step": 2284
    },
    {
      "epoch": 2.178192138671875e-06,
      "step": 2284,
      "training_step_time": 17.018036127090454
    },
    {
      "epoch": 2.1791458129882814e-06,
      "model_forward_time": 5.960570812225342,
      "step": 2285
    },
    {
      "epoch": 2.1791458129882814e-06,
      "step": 2285,
      "training_step_time": 17.415090799331665
    },
    {
      "epoch": 2.1800994873046875e-06,
      "model_forward_time": 5.603941917419434,
      "step": 2286
    },
    {
      "epoch": 2.1800994873046875e-06,
      "step": 2286,
      "training_step_time": 17.08045482635498
    },
    {
      "epoch": 2.1810531616210937e-06,
      "model_forward_time": 5.619379281997681,
      "step": 2287
    },
    {
      "epoch": 2.1810531616210937e-06,
      "step": 2287,
      "training_step_time": 17.123432636260986
    },
    {
      "epoch": 2.1820068359375e-06,
      "model_forward_time": 5.624701023101807,
      "step": 2288
    },
    {
      "epoch": 2.1820068359375e-06,
      "step": 2288,
      "training_step_time": 17.138922214508057
    },
    {
      "epoch": 2.1829605102539064e-06,
      "model_forward_time": 5.613243103027344,
      "step": 2289
    },
    {
      "epoch": 2.1829605102539064e-06,
      "step": 2289,
      "training_step_time": 17.104821920394897
    },
    {
      "epoch": 2.1839141845703125e-06,
      "grad_norm": 1.2165974378585815,
      "learning_rate": 1e-05,
      "loss": 0.0834,
      "step": 2290
    },
    {
      "epoch": 2.1839141845703125e-06,
      "model_forward_time": 5.953359603881836,
      "step": 2290
    },
    {
      "dynamics_loss_avg": 0.06532170567661524,
      "epoch": 2.1839141845703125e-06,
      "step": 2290
    },
    {
      "action_loss_avg": 0.014779622713103891,
      "epoch": 2.1839141845703125e-06,
      "step": 2290
    },
    {
      "epoch": 2.1839141845703125e-06,
      "step": 2290,
      "training_step_time": 17.43090319633484
    },
    {
      "epoch": 2.1848678588867187e-06,
      "model_forward_time": 5.591280221939087,
      "step": 2291
    },
    {
      "epoch": 2.1848678588867187e-06,
      "step": 2291,
      "training_step_time": 17.05126690864563
    },
    {
      "epoch": 2.185821533203125e-06,
      "model_forward_time": 5.599072217941284,
      "step": 2292
    },
    {
      "epoch": 2.185821533203125e-06,
      "step": 2292,
      "training_step_time": 17.054367303848267
    },
    {
      "epoch": 2.1867752075195314e-06,
      "model_forward_time": 5.5956645011901855,
      "step": 2293
    },
    {
      "epoch": 2.1867752075195314e-06,
      "step": 2293,
      "training_step_time": 17.045326232910156
    },
    {
      "epoch": 2.1877288818359375e-06,
      "model_forward_time": 5.581188678741455,
      "step": 2294
    },
    {
      "epoch": 2.1877288818359375e-06,
      "step": 2294,
      "training_step_time": 17.030272960662842
    },
    {
      "epoch": 2.1886825561523437e-06,
      "model_forward_time": 5.59366250038147,
      "step": 2295
    },
    {
      "epoch": 2.1886825561523437e-06,
      "step": 2295,
      "training_step_time": 17.043912887573242
    },
    {
      "epoch": 2.18963623046875e-06,
      "model_forward_time": 5.970652103424072,
      "step": 2296
    },
    {
      "epoch": 2.18963623046875e-06,
      "step": 2296,
      "training_step_time": 17.440314531326294
    },
    {
      "epoch": 2.1905899047851564e-06,
      "model_forward_time": 5.607024908065796,
      "step": 2297
    },
    {
      "epoch": 2.1905899047851564e-06,
      "step": 2297,
      "training_step_time": 17.11042094230652
    },
    {
      "epoch": 2.1915435791015625e-06,
      "model_forward_time": 5.62071681022644,
      "step": 2298
    },
    {
      "epoch": 2.1915435791015625e-06,
      "step": 2298,
      "training_step_time": 17.11199975013733
    },
    {
      "epoch": 2.1924972534179687e-06,
      "model_forward_time": 5.603920936584473,
      "step": 2299
    },
    {
      "epoch": 2.1924972534179687e-06,
      "step": 2299,
      "training_step_time": 17.060033082962036
    },
    {
      "epoch": 2.193450927734375e-06,
      "grad_norm": 0.6947450041770935,
      "learning_rate": 1e-05,
      "loss": 0.0748,
      "step": 2300
    },
    {
      "epoch": 2.193450927734375e-06,
      "model_forward_time": 5.588507175445557,
      "step": 2300
    },
    {
      "dynamics_loss_avg": 0.06166243180632591,
      "epoch": 2.193450927734375e-06,
      "step": 2300
    },
    {
      "action_loss_avg": 0.014570534927770495,
      "epoch": 2.193450927734375e-06,
      "step": 2300
    },
    {
      "epoch": 2.193450927734375e-06,
      "step": 2300,
      "training_step_time": 17.01767635345459
    },
    {
      "epoch": 2.1944046020507814e-06,
      "model_forward_time": 5.586286544799805,
      "step": 2301
    },
    {
      "epoch": 2.1944046020507814e-06,
      "step": 2301,
      "training_step_time": 17.387169122695923
    },
    {
      "epoch": 2.1953582763671875e-06,
      "model_forward_time": 5.583670616149902,
      "step": 2302
    },
    {
      "epoch": 2.1953582763671875e-06,
      "step": 2302,
      "training_step_time": 17.023868560791016
    },
    {
      "epoch": 2.1963119506835937e-06,
      "model_forward_time": 5.605654001235962,
      "step": 2303
    },
    {
      "epoch": 2.1963119506835937e-06,
      "step": 2303,
      "training_step_time": 17.106338500976562
    },
    {
      "epoch": 2.197265625e-06,
      "model_forward_time": 5.6286609172821045,
      "step": 2304
    },
    {
      "epoch": 2.197265625e-06,
      "step": 2304,
      "training_step_time": 17.160860300064087
    },
    {
      "epoch": 2.1982192993164064e-06,
      "model_forward_time": 5.626323938369751,
      "step": 2305
    },
    {
      "epoch": 2.1982192993164064e-06,
      "step": 2305,
      "training_step_time": 17.125651359558105
    },
    {
      "epoch": 2.1991729736328125e-06,
      "model_forward_time": 5.615024566650391,
      "step": 2306
    },
    {
      "epoch": 2.1991729736328125e-06,
      "step": 2306,
      "training_step_time": 17.10537552833557
    },
    {
      "epoch": 2.2001266479492187e-06,
      "model_forward_time": 5.97428822517395,
      "step": 2307
    },
    {
      "epoch": 2.2001266479492187e-06,
      "step": 2307,
      "training_step_time": 17.445845365524292
    },
    {
      "epoch": 2.201080322265625e-06,
      "model_forward_time": 5.601294994354248,
      "step": 2308
    },
    {
      "epoch": 2.201080322265625e-06,
      "step": 2308,
      "training_step_time": 17.055686712265015
    },
    {
      "epoch": 2.2020339965820314e-06,
      "model_forward_time": 5.593064069747925,
      "step": 2309
    },
    {
      "epoch": 2.2020339965820314e-06,
      "step": 2309,
      "training_step_time": 17.031164169311523
    },
    {
      "epoch": 2.2029876708984376e-06,
      "grad_norm": 2.152479648590088,
      "learning_rate": 1e-05,
      "loss": 0.0918,
      "step": 2310
    },
    {
      "epoch": 2.2029876708984376e-06,
      "model_forward_time": 5.575366258621216,
      "step": 2310
    },
    {
      "dynamics_loss_avg": 0.06871494594961405,
      "epoch": 2.2029876708984376e-06,
      "step": 2310
    },
    {
      "action_loss_avg": 0.01776623916812241,
      "epoch": 2.2029876708984376e-06,
      "step": 2310
    },
    {
      "epoch": 2.2029876708984376e-06,
      "step": 2310,
      "training_step_time": 16.994348526000977
    },
    {
      "epoch": 2.2039413452148437e-06,
      "model_forward_time": 5.5942418575286865,
      "step": 2311
    },
    {
      "epoch": 2.2039413452148437e-06,
      "step": 2311,
      "training_step_time": 17.043532848358154
    },
    {
      "epoch": 2.20489501953125e-06,
      "model_forward_time": 5.593393802642822,
      "step": 2312
    },
    {
      "epoch": 2.20489501953125e-06,
      "step": 2312,
      "training_step_time": 17.434956312179565
    },
    {
      "epoch": 2.2058486938476564e-06,
      "model_forward_time": 5.612027168273926,
      "step": 2313
    },
    {
      "epoch": 2.2058486938476564e-06,
      "step": 2313,
      "training_step_time": 17.14252734184265
    },
    {
      "epoch": 2.2068023681640626e-06,
      "model_forward_time": 5.6222615242004395,
      "step": 2314
    },
    {
      "epoch": 2.2068023681640626e-06,
      "step": 2314,
      "training_step_time": 17.123934984207153
    },
    {
      "epoch": 2.2077560424804687e-06,
      "model_forward_time": 5.614266395568848,
      "step": 2315
    },
    {
      "epoch": 2.2077560424804687e-06,
      "step": 2315,
      "training_step_time": 17.10689377784729
    },
    {
      "epoch": 2.208709716796875e-06,
      "model_forward_time": 5.608617782592773,
      "step": 2316
    },
    {
      "epoch": 2.208709716796875e-06,
      "step": 2316,
      "training_step_time": 17.087990045547485
    },
    {
      "epoch": 2.2096633911132814e-06,
      "model_forward_time": 5.595239877700806,
      "step": 2317
    },
    {
      "epoch": 2.2096633911132814e-06,
      "step": 2317,
      "training_step_time": 17.04622483253479
    },
    {
      "epoch": 2.2106170654296876e-06,
      "model_forward_time": 5.969874858856201,
      "step": 2318
    },
    {
      "epoch": 2.2106170654296876e-06,
      "step": 2318,
      "training_step_time": 17.417466163635254
    },
    {
      "epoch": 2.2115707397460937e-06,
      "model_forward_time": 5.591119766235352,
      "step": 2319
    },
    {
      "epoch": 2.2115707397460937e-06,
      "step": 2319,
      "training_step_time": 17.05118703842163
    },
    {
      "epoch": 2.2125244140625e-06,
      "grad_norm": 1.049503207206726,
      "learning_rate": 1e-05,
      "loss": 0.1101,
      "step": 2320
    },
    {
      "epoch": 2.2125244140625e-06,
      "model_forward_time": 5.598816156387329,
      "step": 2320
    },
    {
      "dynamics_loss_avg": 0.08303344920277596,
      "epoch": 2.2125244140625e-06,
      "step": 2320
    },
    {
      "action_loss_avg": 0.026061197835952045,
      "epoch": 2.2125244140625e-06,
      "step": 2320
    },
    {
      "epoch": 2.2125244140625e-06,
      "step": 2320,
      "training_step_time": 17.07799792289734
    },
    {
      "epoch": 2.2134780883789064e-06,
      "model_forward_time": 5.602248191833496,
      "step": 2321
    },
    {
      "epoch": 2.2134780883789064e-06,
      "step": 2321,
      "training_step_time": 17.069955348968506
    },
    {
      "epoch": 2.2144317626953126e-06,
      "model_forward_time": 5.597323894500732,
      "step": 2322
    },
    {
      "epoch": 2.2144317626953126e-06,
      "step": 2322,
      "training_step_time": 17.054086446762085
    },
    {
      "epoch": 2.2153854370117187e-06,
      "model_forward_time": 5.95550274848938,
      "step": 2323
    },
    {
      "epoch": 2.2153854370117187e-06,
      "step": 2323,
      "training_step_time": 17.416992902755737
    },
    {
      "epoch": 2.216339111328125e-06,
      "model_forward_time": 5.603100776672363,
      "step": 2324
    },
    {
      "epoch": 2.216339111328125e-06,
      "step": 2324,
      "training_step_time": 17.064703702926636
    },
    {
      "epoch": 2.2172927856445314e-06,
      "model_forward_time": 5.611437559127808,
      "step": 2325
    },
    {
      "epoch": 2.2172927856445314e-06,
      "step": 2325,
      "training_step_time": 17.090110063552856
    },
    {
      "epoch": 2.2182464599609376e-06,
      "model_forward_time": 5.603555917739868,
      "step": 2326
    },
    {
      "epoch": 2.2182464599609376e-06,
      "step": 2326,
      "training_step_time": 17.103693962097168
    },
    {
      "epoch": 2.2192001342773437e-06,
      "model_forward_time": 5.611928462982178,
      "step": 2327
    },
    {
      "epoch": 2.2192001342773437e-06,
      "step": 2327,
      "training_step_time": 17.079577445983887
    },
    {
      "epoch": 2.22015380859375e-06,
      "model_forward_time": 5.585184574127197,
      "step": 2328
    },
    {
      "epoch": 2.22015380859375e-06,
      "step": 2328,
      "training_step_time": 17.028070211410522
    },
    {
      "epoch": 2.2211074829101564e-06,
      "model_forward_time": 5.93238639831543,
      "step": 2329
    },
    {
      "epoch": 2.2211074829101564e-06,
      "step": 2329,
      "training_step_time": 17.349925756454468
    },
    {
      "epoch": 2.2220611572265626e-06,
      "grad_norm": 0.900049090385437,
      "learning_rate": 1e-05,
      "loss": 0.08,
      "step": 2330
    },
    {
      "epoch": 2.2220611572265626e-06,
      "model_forward_time": 5.564645051956177,
      "step": 2330
    },
    {
      "dynamics_loss_avg": 0.06741048078984022,
      "epoch": 2.2220611572265626e-06,
      "step": 2330
    },
    {
      "action_loss_avg": 0.017048165667802095,
      "epoch": 2.2220611572265626e-06,
      "step": 2330
    },
    {
      "epoch": 2.2220611572265626e-06,
      "step": 2330,
      "training_step_time": 16.982314109802246
    },
    {
      "epoch": 2.2230148315429687e-06,
      "model_forward_time": 5.567178010940552,
      "step": 2331
    },
    {
      "epoch": 2.2230148315429687e-06,
      "step": 2331,
      "training_step_time": 16.99963617324829
    },
    {
      "epoch": 2.223968505859375e-06,
      "model_forward_time": 5.576737880706787,
      "step": 2332
    },
    {
      "epoch": 2.223968505859375e-06,
      "step": 2332,
      "training_step_time": 17.024440050125122
    },
    {
      "epoch": 2.2249221801757814e-06,
      "model_forward_time": 5.587639093399048,
      "step": 2333
    },
    {
      "epoch": 2.2249221801757814e-06,
      "step": 2333,
      "training_step_time": 17.00015163421631
    },
    {
      "epoch": 2.2258758544921876e-06,
      "model_forward_time": 5.583291053771973,
      "step": 2334
    },
    {
      "epoch": 2.2258758544921876e-06,
      "step": 2334,
      "training_step_time": 17.38355040550232
    },
    {
      "epoch": 2.2268295288085937e-06,
      "model_forward_time": 5.590975046157837,
      "step": 2335
    },
    {
      "epoch": 2.2268295288085937e-06,
      "step": 2335,
      "training_step_time": 17.036803007125854
    },
    {
      "epoch": 2.227783203125e-06,
      "model_forward_time": 5.586873292922974,
      "step": 2336
    },
    {
      "epoch": 2.227783203125e-06,
      "step": 2336,
      "training_step_time": 17.023842096328735
    },
    {
      "epoch": 2.2287368774414064e-06,
      "model_forward_time": 5.594014406204224,
      "step": 2337
    },
    {
      "epoch": 2.2287368774414064e-06,
      "step": 2337,
      "training_step_time": 17.073813438415527
    },
    {
      "epoch": 2.2296905517578126e-06,
      "model_forward_time": 5.6052258014678955,
      "step": 2338
    },
    {
      "epoch": 2.2296905517578126e-06,
      "step": 2338,
      "training_step_time": 17.100403308868408
    },
    {
      "epoch": 2.2306442260742187e-06,
      "model_forward_time": 5.613606929779053,
      "step": 2339
    },
    {
      "epoch": 2.2306442260742187e-06,
      "step": 2339,
      "training_step_time": 17.09489417076111
    },
    {
      "epoch": 2.231597900390625e-06,
      "grad_norm": 0.9001427292823792,
      "learning_rate": 1e-05,
      "loss": 0.0772,
      "step": 2340
    },
    {
      "epoch": 2.231597900390625e-06,
      "model_forward_time": 5.967752695083618,
      "step": 2340
    },
    {
      "dynamics_loss_avg": 0.0640630591660738,
      "epoch": 2.231597900390625e-06,
      "step": 2340
    },
    {
      "action_loss_avg": 0.015541571518406272,
      "epoch": 2.231597900390625e-06,
      "step": 2340
    },
    {
      "epoch": 2.231597900390625e-06,
      "step": 2340,
      "training_step_time": 17.43132472038269
    },
    {
      "epoch": 2.2325515747070314e-06,
      "model_forward_time": 5.607698678970337,
      "step": 2341
    },
    {
      "epoch": 2.2325515747070314e-06,
      "step": 2341,
      "training_step_time": 17.070037841796875
    },
    {
      "epoch": 2.2335052490234376e-06,
      "model_forward_time": 5.605118751525879,
      "step": 2342
    },
    {
      "epoch": 2.2335052490234376e-06,
      "step": 2342,
      "training_step_time": 17.064218997955322
    },
    {
      "epoch": 2.2344589233398437e-06,
      "model_forward_time": 5.605026483535767,
      "step": 2343
    },
    {
      "epoch": 2.2344589233398437e-06,
      "step": 2343,
      "training_step_time": 17.061492919921875
    },
    {
      "epoch": 2.23541259765625e-06,
      "model_forward_time": 5.604630947113037,
      "step": 2344
    },
    {
      "epoch": 2.23541259765625e-06,
      "step": 2344,
      "training_step_time": 17.08012294769287
    },
    {
      "epoch": 2.2363662719726564e-06,
      "model_forward_time": 5.602356672286987,
      "step": 2345
    },
    {
      "epoch": 2.2363662719726564e-06,
      "step": 2345,
      "training_step_time": 17.45199751853943
    },
    {
      "epoch": 2.2373199462890626e-06,
      "model_forward_time": 5.587684869766235,
      "step": 2346
    },
    {
      "epoch": 2.2373199462890626e-06,
      "step": 2346,
      "training_step_time": 17.00855851173401
    },
    {
      "epoch": 2.2382736206054687e-06,
      "model_forward_time": 5.570941209793091,
      "step": 2347
    },
    {
      "epoch": 2.2382736206054687e-06,
      "step": 2347,
      "training_step_time": 17.004854679107666
    },
    {
      "epoch": 2.239227294921875e-06,
      "model_forward_time": 5.564697504043579,
      "step": 2348
    },
    {
      "epoch": 2.239227294921875e-06,
      "step": 2348,
      "training_step_time": 16.979504823684692
    },
    {
      "epoch": 2.2401809692382814e-06,
      "model_forward_time": 5.57686710357666,
      "step": 2349
    },
    {
      "epoch": 2.2401809692382814e-06,
      "step": 2349,
      "training_step_time": 17.013042449951172
    },
    {
      "epoch": 2.2411346435546876e-06,
      "grad_norm": 1.1051658391952515,
      "learning_rate": 1e-05,
      "loss": 0.0686,
      "step": 2350
    },
    {
      "epoch": 2.2411346435546876e-06,
      "model_forward_time": 5.577110767364502,
      "step": 2350
    },
    {
      "dynamics_loss_avg": 0.052282802388072015,
      "epoch": 2.2411346435546876e-06,
      "step": 2350
    },
    {
      "action_loss_avg": 0.013146218657493592,
      "epoch": 2.2411346435546876e-06,
      "step": 2350
    },
    {
      "epoch": 2.2411346435546876e-06,
      "step": 2350,
      "training_step_time": 17.002923488616943
    },
    {
      "epoch": 2.2420883178710937e-06,
      "model_forward_time": 5.956966876983643,
      "step": 2351
    },
    {
      "epoch": 2.2420883178710937e-06,
      "step": 2351,
      "training_step_time": 17.42612934112549
    },
    {
      "epoch": 2.2430419921875e-06,
      "model_forward_time": 5.604814529418945,
      "step": 2352
    },
    {
      "epoch": 2.2430419921875e-06,
      "step": 2352,
      "training_step_time": 17.104241609573364
    },
    {
      "epoch": 2.2439956665039064e-06,
      "model_forward_time": 5.617549657821655,
      "step": 2353
    },
    {
      "epoch": 2.2439956665039064e-06,
      "step": 2353,
      "training_step_time": 17.125199556350708
    },
    {
      "epoch": 2.2449493408203126e-06,
      "model_forward_time": 5.656369209289551,
      "step": 2354
    },
    {
      "epoch": 2.2449493408203126e-06,
      "step": 2354,
      "training_step_time": 17.170827388763428
    },
    {
      "epoch": 2.2459030151367187e-06,
      "model_forward_time": 5.601422071456909,
      "step": 2355
    },
    {
      "epoch": 2.2459030151367187e-06,
      "step": 2355,
      "training_step_time": 17.043630599975586
    },
    {
      "epoch": 2.246856689453125e-06,
      "model_forward_time": 5.930510759353638,
      "step": 2356
    },
    {
      "epoch": 2.246856689453125e-06,
      "step": 2356,
      "training_step_time": 17.362887382507324
    },
    {
      "epoch": 2.2478103637695314e-06,
      "model_forward_time": 5.5739006996154785,
      "step": 2357
    },
    {
      "epoch": 2.2478103637695314e-06,
      "step": 2357,
      "training_step_time": 16.99973225593567
    },
    {
      "epoch": 2.2487640380859376e-06,
      "model_forward_time": 5.5876007080078125,
      "step": 2358
    },
    {
      "epoch": 2.2487640380859376e-06,
      "step": 2358,
      "training_step_time": 17.022855520248413
    },
    {
      "epoch": 2.2497177124023437e-06,
      "model_forward_time": 5.603439569473267,
      "step": 2359
    },
    {
      "epoch": 2.2497177124023437e-06,
      "step": 2359,
      "training_step_time": 17.053699016571045
    },
    {
      "epoch": 2.25067138671875e-06,
      "grad_norm": 1.8333914279937744,
      "learning_rate": 1e-05,
      "loss": 0.0752,
      "step": 2360
    },
    {
      "epoch": 2.25067138671875e-06,
      "model_forward_time": 5.601857662200928,
      "step": 2360
    },
    {
      "dynamics_loss_avg": 0.058242480456829074,
      "epoch": 2.25067138671875e-06,
      "step": 2360
    },
    {
      "action_loss_avg": 0.01812801561318338,
      "epoch": 2.25067138671875e-06,
      "step": 2360
    },
    {
      "epoch": 2.25067138671875e-06,
      "step": 2360,
      "training_step_time": 17.10243320465088
    },
    {
      "epoch": 2.2516250610351564e-06,
      "model_forward_time": 5.6115124225616455,
      "step": 2361
    },
    {
      "epoch": 2.2516250610351564e-06,
      "step": 2361,
      "training_step_time": 17.09123682975769
    },
    {
      "epoch": 2.2525787353515626e-06,
      "model_forward_time": 5.9562668800354,
      "step": 2362
    },
    {
      "epoch": 2.2525787353515626e-06,
      "step": 2362,
      "training_step_time": 17.392958402633667
    },
    {
      "epoch": 2.2535324096679687e-06,
      "model_forward_time": 5.5942699909210205,
      "step": 2363
    },
    {
      "epoch": 2.2535324096679687e-06,
      "step": 2363,
      "training_step_time": 17.021892786026
    },
    {
      "epoch": 2.254486083984375e-06,
      "model_forward_time": 5.595242500305176,
      "step": 2364
    },
    {
      "epoch": 2.254486083984375e-06,
      "step": 2364,
      "training_step_time": 17.051820993423462
    },
    {
      "epoch": 2.2554397583007815e-06,
      "model_forward_time": 5.595947504043579,
      "step": 2365
    },
    {
      "epoch": 2.2554397583007815e-06,
      "step": 2365,
      "training_step_time": 17.053552865982056
    },
    {
      "epoch": 2.2563934326171876e-06,
      "model_forward_time": 5.604478359222412,
      "step": 2366
    },
    {
      "epoch": 2.2563934326171876e-06,
      "step": 2366,
      "training_step_time": 17.082075357437134
    },
    {
      "epoch": 2.2573471069335937e-06,
      "model_forward_time": 5.608532428741455,
      "step": 2367
    },
    {
      "epoch": 2.2573471069335937e-06,
      "step": 2367,
      "training_step_time": 17.446422338485718
    },
    {
      "epoch": 2.25830078125e-06,
      "model_forward_time": 5.593049764633179,
      "step": 2368
    },
    {
      "epoch": 2.25830078125e-06,
      "step": 2368,
      "training_step_time": 17.014293909072876
    },
    {
      "epoch": 2.2592544555664065e-06,
      "model_forward_time": 5.5759437084198,
      "step": 2369
    },
    {
      "epoch": 2.2592544555664065e-06,
      "step": 2369,
      "training_step_time": 16.983153581619263
    },
    {
      "epoch": 2.2602081298828126e-06,
      "grad_norm": 0.9757922291755676,
      "learning_rate": 1e-05,
      "loss": 0.0828,
      "step": 2370
    },
    {
      "epoch": 2.2602081298828126e-06,
      "model_forward_time": 5.581430673599243,
      "step": 2370
    },
    {
      "dynamics_loss_avg": 0.06268599722534418,
      "epoch": 2.2602081298828126e-06,
      "step": 2370
    },
    {
      "action_loss_avg": 0.017261285940185188,
      "epoch": 2.2602081298828126e-06,
      "step": 2370
    },
    {
      "epoch": 2.2602081298828126e-06,
      "step": 2370,
      "training_step_time": 17.014343976974487
    },
    {
      "epoch": 2.2611618041992187e-06,
      "model_forward_time": 5.592885494232178,
      "step": 2371
    },
    {
      "epoch": 2.2611618041992187e-06,
      "step": 2371,
      "training_step_time": 17.0663161277771
    },
    {
      "epoch": 2.262115478515625e-06,
      "model_forward_time": 5.600250959396362,
      "step": 2372
    },
    {
      "epoch": 2.262115478515625e-06,
      "step": 2372,
      "training_step_time": 17.102645874023438
    },
    {
      "epoch": 2.2630691528320315e-06,
      "model_forward_time": 5.977388858795166,
      "step": 2373
    },
    {
      "epoch": 2.2630691528320315e-06,
      "step": 2373,
      "training_step_time": 17.486390590667725
    },
    {
      "epoch": 2.2640228271484376e-06,
      "model_forward_time": 5.611395597457886,
      "step": 2374
    },
    {
      "epoch": 2.2640228271484376e-06,
      "step": 2374,
      "training_step_time": 17.098535299301147
    },
    {
      "epoch": 2.2649765014648438e-06,
      "model_forward_time": 5.608227014541626,
      "step": 2375
    },
    {
      "epoch": 2.2649765014648438e-06,
      "step": 2375,
      "training_step_time": 17.096277713775635
    },
    {
      "epoch": 2.26593017578125e-06,
      "model_forward_time": 5.607812881469727,
      "step": 2376
    },
    {
      "epoch": 2.26593017578125e-06,
      "step": 2376,
      "training_step_time": 17.097489595413208
    },
    {
      "epoch": 2.266883850097656e-06,
      "model_forward_time": 5.598386526107788,
      "step": 2377
    },
    {
      "epoch": 2.266883850097656e-06,
      "step": 2377,
      "training_step_time": 17.059624195098877
    },
    {
      "epoch": 2.2678375244140626e-06,
      "model_forward_time": 5.5935139656066895,
      "step": 2378
    },
    {
      "epoch": 2.2678375244140626e-06,
      "step": 2378,
      "training_step_time": 17.40750217437744
    },
    {
      "epoch": 2.2687911987304688e-06,
      "model_forward_time": 5.604219198226929,
      "step": 2379
    },
    {
      "epoch": 2.2687911987304688e-06,
      "step": 2379,
      "training_step_time": 17.066908597946167
    },
    {
      "epoch": 2.269744873046875e-06,
      "grad_norm": 1.2037345170974731,
      "learning_rate": 1e-05,
      "loss": 0.088,
      "step": 2380
    },
    {
      "epoch": 2.269744873046875e-06,
      "model_forward_time": 5.6004602909088135,
      "step": 2380
    },
    {
      "dynamics_loss_avg": 0.07121677724644541,
      "epoch": 2.269744873046875e-06,
      "step": 2380
    },
    {
      "action_loss_avg": 0.021430783369578422,
      "epoch": 2.269744873046875e-06,
      "step": 2380
    },
    {
      "epoch": 2.269744873046875e-06,
      "step": 2380,
      "training_step_time": 17.062816619873047
    },
    {
      "epoch": 2.270698547363281e-06,
      "model_forward_time": 5.599059820175171,
      "step": 2381
    },
    {
      "epoch": 2.270698547363281e-06,
      "step": 2381,
      "training_step_time": 17.02288031578064
    },
    {
      "epoch": 2.2716522216796876e-06,
      "model_forward_time": 5.590263605117798,
      "step": 2382
    },
    {
      "epoch": 2.2716522216796876e-06,
      "step": 2382,
      "training_step_time": 17.050368785858154
    },
    {
      "epoch": 2.2726058959960938e-06,
      "model_forward_time": 5.596904516220093,
      "step": 2383
    },
    {
      "epoch": 2.2726058959960938e-06,
      "step": 2383,
      "training_step_time": 17.091418266296387
    },
    {
      "epoch": 2.2735595703125e-06,
      "model_forward_time": 5.97716212272644,
      "step": 2384
    },
    {
      "epoch": 2.2735595703125e-06,
      "step": 2384,
      "training_step_time": 17.46913480758667
    },
    {
      "epoch": 2.274513244628906e-06,
      "model_forward_time": 5.609656095504761,
      "step": 2385
    },
    {
      "epoch": 2.274513244628906e-06,
      "step": 2385,
      "training_step_time": 17.08191156387329
    },
    {
      "epoch": 2.2754669189453126e-06,
      "model_forward_time": 5.576427936553955,
      "step": 2386
    },
    {
      "epoch": 2.2754669189453126e-06,
      "step": 2386,
      "training_step_time": 16.99862813949585
    },
    {
      "epoch": 2.2764205932617188e-06,
      "model_forward_time": 5.571985244750977,
      "step": 2387
    },
    {
      "epoch": 2.2764205932617188e-06,
      "step": 2387,
      "training_step_time": 16.98103356361389
    },
    {
      "epoch": 2.277374267578125e-06,
      "model_forward_time": 5.553851366043091,
      "step": 2388
    },
    {
      "epoch": 2.277374267578125e-06,
      "step": 2388,
      "training_step_time": 16.96315336227417
    },
    {
      "epoch": 2.278327941894531e-06,
      "model_forward_time": 5.926240682601929,
      "step": 2389
    },
    {
      "epoch": 2.278327941894531e-06,
      "step": 2389,
      "training_step_time": 17.35691237449646
    },
    {
      "epoch": 2.2792816162109376e-06,
      "grad_norm": 1.277761697769165,
      "learning_rate": 1e-05,
      "loss": 0.0982,
      "step": 2390
    },
    {
      "epoch": 2.2792816162109376e-06,
      "model_forward_time": 5.567090034484863,
      "step": 2390
    },
    {
      "dynamics_loss_avg": 0.07188833244144917,
      "epoch": 2.2792816162109376e-06,
      "step": 2390
    },
    {
      "action_loss_avg": 0.018992189969867467,
      "epoch": 2.2792816162109376e-06,
      "step": 2390
    },
    {
      "epoch": 2.2792816162109376e-06,
      "step": 2390,
      "training_step_time": 16.976078271865845
    },
    {
      "epoch": 2.2802352905273438e-06,
      "model_forward_time": 5.571113109588623,
      "step": 2391
    },
    {
      "epoch": 2.2802352905273438e-06,
      "step": 2391,
      "training_step_time": 16.98068642616272
    },
    {
      "epoch": 2.28118896484375e-06,
      "model_forward_time": 5.567638397216797,
      "step": 2392
    },
    {
      "epoch": 2.28118896484375e-06,
      "step": 2392,
      "training_step_time": 16.998142957687378
    },
    {
      "epoch": 2.282142639160156e-06,
      "model_forward_time": 5.569777965545654,
      "step": 2393
    },
    {
      "epoch": 2.282142639160156e-06,
      "step": 2393,
      "training_step_time": 17.01216411590576
    },
    {
      "epoch": 2.2830963134765626e-06,
      "model_forward_time": 5.57390832901001,
      "step": 2394
    },
    {
      "epoch": 2.2830963134765626e-06,
      "step": 2394,
      "training_step_time": 16.99529004096985
    },
    {
      "epoch": 2.2840499877929688e-06,
      "model_forward_time": 5.944450855255127,
      "step": 2395
    },
    {
      "epoch": 2.2840499877929688e-06,
      "step": 2395,
      "training_step_time": 17.395092010498047
    },
    {
      "epoch": 2.285003662109375e-06,
      "model_forward_time": 5.598478555679321,
      "step": 2396
    },
    {
      "epoch": 2.285003662109375e-06,
      "step": 2396,
      "training_step_time": 17.061304807662964
    },
    {
      "epoch": 2.285957336425781e-06,
      "model_forward_time": 5.60085129737854,
      "step": 2397
    },
    {
      "epoch": 2.285957336425781e-06,
      "step": 2397,
      "training_step_time": 17.080195665359497
    },
    {
      "epoch": 2.2869110107421876e-06,
      "model_forward_time": 5.602475881576538,
      "step": 2398
    },
    {
      "epoch": 2.2869110107421876e-06,
      "step": 2398,
      "training_step_time": 17.08446478843689
    },
    {
      "epoch": 2.2878646850585938e-06,
      "model_forward_time": 5.604938507080078,
      "step": 2399
    },
    {
      "epoch": 2.2878646850585938e-06,
      "step": 2399,
      "training_step_time": 17.077239990234375
    },
    {
      "epoch": 2.288818359375e-06,
      "grad_norm": 1.6957350969314575,
      "learning_rate": 1e-05,
      "loss": 0.0948,
      "step": 2400
    },
    {
      "epoch": 2.288818359375e-06,
      "model_forward_time": 5.609755039215088,
      "step": 2400
    },
    {
      "dynamics_loss_avg": 0.07041844949126244,
      "epoch": 2.288818359375e-06,
      "step": 2400
    },
    {
      "action_loss_avg": 0.024513297202065586,
      "epoch": 2.288818359375e-06,
      "step": 2400
    },
    {
      "epoch": 2.288818359375e-06,
      "step": 2400,
      "training_step_time": 17.42767310142517
    },
    {
      "epoch": 2.289772033691406e-06,
      "model_forward_time": 5.607732534408569,
      "step": 2401
    },
    {
      "epoch": 2.289772033691406e-06,
      "step": 2401,
      "training_step_time": 17.118672370910645
    },
    {
      "epoch": 2.2907257080078126e-06,
      "model_forward_time": 5.6171252727508545,
      "step": 2402
    },
    {
      "epoch": 2.2907257080078126e-06,
      "step": 2402,
      "training_step_time": 17.110955953598022
    },
    {
      "epoch": 2.2916793823242188e-06,
      "model_forward_time": 5.594740629196167,
      "step": 2403
    },
    {
      "epoch": 2.2916793823242188e-06,
      "step": 2403,
      "training_step_time": 17.025529623031616
    },
    {
      "epoch": 2.292633056640625e-06,
      "model_forward_time": 5.5828046798706055,
      "step": 2404
    },
    {
      "epoch": 2.292633056640625e-06,
      "step": 2404,
      "training_step_time": 17.014611959457397
    },
    {
      "epoch": 2.293586730957031e-06,
      "model_forward_time": 5.575602293014526,
      "step": 2405
    },
    {
      "epoch": 2.293586730957031e-06,
      "step": 2405,
      "training_step_time": 17.00350332260132
    },
    {
      "epoch": 2.2945404052734376e-06,
      "model_forward_time": 5.9730894565582275,
      "step": 2406
    },
    {
      "epoch": 2.2945404052734376e-06,
      "step": 2406,
      "training_step_time": 17.411083698272705
    },
    {
      "epoch": 2.2954940795898438e-06,
      "model_forward_time": 5.5995564460754395,
      "step": 2407
    },
    {
      "epoch": 2.2954940795898438e-06,
      "step": 2407,
      "training_step_time": 17.041078567504883
    },
    {
      "epoch": 2.29644775390625e-06,
      "model_forward_time": 5.595976114273071,
      "step": 2408
    },
    {
      "epoch": 2.29644775390625e-06,
      "step": 2408,
      "training_step_time": 17.032034635543823
    },
    {
      "epoch": 2.297401428222656e-06,
      "model_forward_time": 5.593356609344482,
      "step": 2409
    },
    {
      "epoch": 2.297401428222656e-06,
      "step": 2409,
      "training_step_time": 17.029813766479492
    },
    {
      "epoch": 2.2983551025390626e-06,
      "grad_norm": 1.8393582105636597,
      "learning_rate": 1e-05,
      "loss": 0.0797,
      "step": 2410
    },
    {
      "epoch": 2.2983551025390626e-06,
      "model_forward_time": 5.593883752822876,
      "step": 2410
    },
    {
      "dynamics_loss_avg": 0.06451183669269085,
      "epoch": 2.2983551025390626e-06,
      "step": 2410
    },
    {
      "action_loss_avg": 0.017380022443830967,
      "epoch": 2.2983551025390626e-06,
      "step": 2410
    },
    {
      "epoch": 2.2983551025390626e-06,
      "step": 2410,
      "training_step_time": 17.032021522521973
    },
    {
      "epoch": 2.2993087768554688e-06,
      "model_forward_time": 5.590417861938477,
      "step": 2411
    },
    {
      "epoch": 2.2993087768554688e-06,
      "step": 2411,
      "training_step_time": 17.406999826431274
    },
    {
      "epoch": 2.300262451171875e-06,
      "model_forward_time": 5.604548931121826,
      "step": 2412
    },
    {
      "epoch": 2.300262451171875e-06,
      "step": 2412,
      "training_step_time": 17.08523464202881
    },
    {
      "epoch": 2.301216125488281e-06,
      "model_forward_time": 5.613471031188965,
      "step": 2413
    },
    {
      "epoch": 2.301216125488281e-06,
      "step": 2413,
      "training_step_time": 17.073259830474854
    },
    {
      "epoch": 2.3021697998046876e-06,
      "model_forward_time": 5.615909099578857,
      "step": 2414
    },
    {
      "epoch": 2.3021697998046876e-06,
      "step": 2414,
      "training_step_time": 17.094005584716797
    },
    {
      "epoch": 2.3031234741210938e-06,
      "model_forward_time": 5.604777097702026,
      "step": 2415
    },
    {
      "epoch": 2.3031234741210938e-06,
      "step": 2415,
      "training_step_time": 17.074002981185913
    },
    {
      "epoch": 2.3040771484375e-06,
      "model_forward_time": 5.5994250774383545,
      "step": 2416
    },
    {
      "epoch": 2.3040771484375e-06,
      "step": 2416,
      "training_step_time": 17.08009433746338
    },
    {
      "epoch": 2.305030822753906e-06,
      "model_forward_time": 5.978516578674316,
      "step": 2417
    },
    {
      "epoch": 2.305030822753906e-06,
      "step": 2417,
      "training_step_time": 17.473644256591797
    },
    {
      "epoch": 2.3059844970703126e-06,
      "model_forward_time": 5.614299774169922,
      "step": 2418
    },
    {
      "epoch": 2.3059844970703126e-06,
      "step": 2418,
      "training_step_time": 17.09575605392456
    },
    {
      "epoch": 2.3069381713867188e-06,
      "model_forward_time": 5.608065128326416,
      "step": 2419
    },
    {
      "epoch": 2.3069381713867188e-06,
      "step": 2419,
      "training_step_time": 17.1102614402771
    },
    {
      "epoch": 2.307891845703125e-06,
      "grad_norm": 1.5229511260986328,
      "learning_rate": 1e-05,
      "loss": 0.0884,
      "step": 2420
    },
    {
      "epoch": 2.307891845703125e-06,
      "model_forward_time": 5.624667167663574,
      "step": 2420
    },
    {
      "dynamics_loss_avg": 0.07127478495240211,
      "epoch": 2.307891845703125e-06,
      "step": 2420
    },
    {
      "action_loss_avg": 0.01803801776841283,
      "epoch": 2.307891845703125e-06,
      "step": 2420
    },
    {
      "epoch": 2.307891845703125e-06,
      "step": 2420,
      "training_step_time": 17.13090753555298
    },
    {
      "epoch": 2.308845520019531e-06,
      "model_forward_time": 5.622042179107666,
      "step": 2421
    },
    {
      "epoch": 2.308845520019531e-06,
      "step": 2421,
      "training_step_time": 17.122822046279907
    },
    {
      "epoch": 2.3097991943359376e-06,
      "model_forward_time": 5.965176582336426,
      "step": 2422
    },
    {
      "epoch": 2.3097991943359376e-06,
      "step": 2422,
      "training_step_time": 17.46771478652954
    },
    {
      "epoch": 2.310752868652344e-06,
      "model_forward_time": 5.611603021621704,
      "step": 2423
    },
    {
      "epoch": 2.310752868652344e-06,
      "step": 2423,
      "training_step_time": 17.109496355056763
    },
    {
      "epoch": 2.31170654296875e-06,
      "model_forward_time": 5.622400522232056,
      "step": 2424
    },
    {
      "epoch": 2.31170654296875e-06,
      "step": 2424,
      "training_step_time": 17.12625765800476
    },
    {
      "epoch": 2.312660217285156e-06,
      "model_forward_time": 5.625092506408691,
      "step": 2425
    },
    {
      "epoch": 2.312660217285156e-06,
      "step": 2425,
      "training_step_time": 17.103625774383545
    },
    {
      "epoch": 2.3136138916015626e-06,
      "model_forward_time": 5.586191415786743,
      "step": 2426
    },
    {
      "epoch": 2.3136138916015626e-06,
      "step": 2426,
      "training_step_time": 17.00393271446228
    },
    {
      "epoch": 2.314567565917969e-06,
      "model_forward_time": 5.572583913803101,
      "step": 2427
    },
    {
      "epoch": 2.314567565917969e-06,
      "step": 2427,
      "training_step_time": 17.001909494400024
    },
    {
      "epoch": 2.315521240234375e-06,
      "model_forward_time": 5.936865329742432,
      "step": 2428
    },
    {
      "epoch": 2.315521240234375e-06,
      "step": 2428,
      "training_step_time": 17.367467164993286
    },
    {
      "epoch": 2.316474914550781e-06,
      "model_forward_time": 5.5773468017578125,
      "step": 2429
    },
    {
      "epoch": 2.316474914550781e-06,
      "step": 2429,
      "training_step_time": 16.99913001060486
    },
    {
      "epoch": 2.3174285888671877e-06,
      "grad_norm": 0.781236469745636,
      "learning_rate": 1e-05,
      "loss": 0.0697,
      "step": 2430
    },
    {
      "epoch": 2.3174285888671877e-06,
      "model_forward_time": 5.570573806762695,
      "step": 2430
    },
    {
      "dynamics_loss_avg": 0.054426465183496475,
      "epoch": 2.3174285888671877e-06,
      "step": 2430
    },
    {
      "action_loss_avg": 0.014010398555547,
      "epoch": 2.3174285888671877e-06,
      "step": 2430
    },
    {
      "epoch": 2.3174285888671877e-06,
      "step": 2430,
      "training_step_time": 16.98561692237854
    },
    {
      "epoch": 2.318382263183594e-06,
      "model_forward_time": 5.578463077545166,
      "step": 2431
    },
    {
      "epoch": 2.318382263183594e-06,
      "step": 2431,
      "training_step_time": 17.011563777923584
    },
    {
      "epoch": 2.3193359375e-06,
      "model_forward_time": 5.586144208908081,
      "step": 2432
    },
    {
      "epoch": 2.3193359375e-06,
      "step": 2432,
      "training_step_time": 17.03291368484497
    },
    {
      "epoch": 2.320289611816406e-06,
      "model_forward_time": 5.598111391067505,
      "step": 2433
    },
    {
      "epoch": 2.320289611816406e-06,
      "step": 2433,
      "training_step_time": 17.449771642684937
    },
    {
      "epoch": 2.3212432861328127e-06,
      "model_forward_time": 5.612538814544678,
      "step": 2434
    },
    {
      "epoch": 2.3212432861328127e-06,
      "step": 2434,
      "training_step_time": 17.113567352294922
    },
    {
      "epoch": 2.322196960449219e-06,
      "model_forward_time": 5.623812913894653,
      "step": 2435
    },
    {
      "epoch": 2.322196960449219e-06,
      "step": 2435,
      "training_step_time": 17.086947441101074
    },
    {
      "epoch": 2.323150634765625e-06,
      "model_forward_time": 5.594439744949341,
      "step": 2436
    },
    {
      "epoch": 2.323150634765625e-06,
      "step": 2436,
      "training_step_time": 17.011544704437256
    },
    {
      "epoch": 2.324104309082031e-06,
      "model_forward_time": 5.577110290527344,
      "step": 2437
    },
    {
      "epoch": 2.324104309082031e-06,
      "step": 2437,
      "training_step_time": 16.99529767036438
    },
    {
      "epoch": 2.3250579833984377e-06,
      "model_forward_time": 5.572256565093994,
      "step": 2438
    },
    {
      "epoch": 2.3250579833984377e-06,
      "step": 2438,
      "training_step_time": 17.004209756851196
    },
    {
      "epoch": 2.326011657714844e-06,
      "model_forward_time": 5.926275253295898,
      "step": 2439
    },
    {
      "epoch": 2.326011657714844e-06,
      "step": 2439,
      "training_step_time": 17.349152326583862
    },
    {
      "epoch": 2.32696533203125e-06,
      "grad_norm": 2.8327510356903076,
      "learning_rate": 1e-05,
      "loss": 0.0896,
      "step": 2440
    },
    {
      "epoch": 2.32696533203125e-06,
      "model_forward_time": 5.571662664413452,
      "step": 2440
    },
    {
      "dynamics_loss_avg": 0.07020653178915381,
      "epoch": 2.32696533203125e-06,
      "step": 2440
    },
    {
      "action_loss_avg": 0.022204510460142046,
      "epoch": 2.32696533203125e-06,
      "step": 2440
    },
    {
      "epoch": 2.32696533203125e-06,
      "step": 2440,
      "training_step_time": 17.000872373580933
    },
    {
      "epoch": 2.327919006347656e-06,
      "model_forward_time": 5.5901947021484375,
      "step": 2441
    },
    {
      "epoch": 2.327919006347656e-06,
      "step": 2441,
      "training_step_time": 17.074763298034668
    },
    {
      "epoch": 2.3288726806640627e-06,
      "model_forward_time": 5.623499155044556,
      "step": 2442
    },
    {
      "epoch": 2.3288726806640627e-06,
      "step": 2442,
      "training_step_time": 17.142152309417725
    },
    {
      "epoch": 2.329826354980469e-06,
      "model_forward_time": 5.617950439453125,
      "step": 2443
    },
    {
      "epoch": 2.329826354980469e-06,
      "step": 2443,
      "training_step_time": 17.098021984100342
    },
    {
      "epoch": 2.330780029296875e-06,
      "model_forward_time": 5.583792686462402,
      "step": 2444
    },
    {
      "epoch": 2.330780029296875e-06,
      "step": 2444,
      "training_step_time": 17.35665225982666
    },
    {
      "epoch": 2.331733703613281e-06,
      "model_forward_time": 5.564143657684326,
      "step": 2445
    },
    {
      "epoch": 2.331733703613281e-06,
      "step": 2445,
      "training_step_time": 16.97855496406555
    },
    {
      "epoch": 2.3326873779296877e-06,
      "model_forward_time": 5.5553247928619385,
      "step": 2446
    },
    {
      "epoch": 2.3326873779296877e-06,
      "step": 2446,
      "training_step_time": 16.982072591781616
    },
    {
      "epoch": 2.333641052246094e-06,
      "model_forward_time": 5.55812931060791,
      "step": 2447
    },
    {
      "epoch": 2.333641052246094e-06,
      "step": 2447,
      "training_step_time": 16.960063219070435
    },
    {
      "epoch": 2.3345947265625e-06,
      "model_forward_time": 5.563259601593018,
      "step": 2448
    },
    {
      "epoch": 2.3345947265625e-06,
      "step": 2448,
      "training_step_time": 16.967310905456543
    },
    {
      "epoch": 2.335548400878906e-06,
      "model_forward_time": 5.562221050262451,
      "step": 2449
    },
    {
      "epoch": 2.335548400878906e-06,
      "step": 2449,
      "training_step_time": 16.980724096298218
    },
    {
      "epoch": 2.3365020751953127e-06,
      "grad_norm": 0.672088086605072,
      "learning_rate": 1e-05,
      "loss": 0.0752,
      "step": 2450
    },
    {
      "epoch": 2.3365020751953127e-06,
      "model_forward_time": 5.941052675247192,
      "step": 2450
    },
    {
      "dynamics_loss_avg": 0.06384293902665376,
      "epoch": 2.3365020751953127e-06,
      "step": 2450
    },
    {
      "action_loss_avg": 0.016757471673190593,
      "epoch": 2.3365020751953127e-06,
      "step": 2450
    },
    {
      "epoch": 2.3365020751953127e-06,
      "step": 2450,
      "training_step_time": 17.379993200302124
    },
    {
      "epoch": 2.337455749511719e-06,
      "model_forward_time": 5.576296806335449,
      "step": 2451
    },
    {
      "epoch": 2.337455749511719e-06,
      "step": 2451,
      "training_step_time": 16.995803594589233
    },
    {
      "epoch": 2.338409423828125e-06,
      "model_forward_time": 5.593092441558838,
      "step": 2452
    },
    {
      "epoch": 2.338409423828125e-06,
      "step": 2452,
      "training_step_time": 17.0427303314209
    },
    {
      "epoch": 2.339363098144531e-06,
      "model_forward_time": 5.607924461364746,
      "step": 2453
    },
    {
      "epoch": 2.339363098144531e-06,
      "step": 2453,
      "training_step_time": 17.09447717666626
    },
    {
      "epoch": 2.3403167724609377e-06,
      "model_forward_time": 4.280741214752197,
      "step": 2454
    },
    {
      "epoch": 2.3403167724609377e-06,
      "step": 2454,
      "training_step_time": 12.822872400283813
    },
    {
      "epoch": 2.341270446777344e-06,
      "model_forward_time": 5.605733633041382,
      "step": 2455
    },
    {
      "epoch": 2.341270446777344e-06,
      "step": 2455,
      "training_step_time": 17.40765690803528
    },
    {
      "epoch": 2.34222412109375e-06,
      "model_forward_time": 5.582643747329712,
      "step": 2456
    },
    {
      "epoch": 2.34222412109375e-06,
      "step": 2456,
      "training_step_time": 17.016358613967896
    },
    {
      "epoch": 2.343177795410156e-06,
      "model_forward_time": 5.576390981674194,
      "step": 2457
    },
    {
      "epoch": 2.343177795410156e-06,
      "step": 2457,
      "training_step_time": 16.997747898101807
    },
    {
      "epoch": 2.3441314697265627e-06,
      "model_forward_time": 5.5908427238464355,
      "step": 2458
    },
    {
      "epoch": 2.3441314697265627e-06,
      "step": 2458,
      "training_step_time": 17.03831458091736
    },
    {
      "epoch": 2.345085144042969e-06,
      "model_forward_time": 5.597407817840576,
      "step": 2459
    },
    {
      "epoch": 2.345085144042969e-06,
      "step": 2459,
      "training_step_time": 17.070440769195557
    },
    {
      "epoch": 2.346038818359375e-06,
      "grad_norm": 2.032010316848755,
      "learning_rate": 1e-05,
      "loss": 0.1005,
      "step": 2460
    },
    {
      "epoch": 2.346038818359375e-06,
      "model_forward_time": 5.612636089324951,
      "step": 2460
    },
    {
      "dynamics_loss_avg": 0.07861189395189286,
      "epoch": 2.346038818359375e-06,
      "step": 2460
    },
    {
      "action_loss_avg": 0.02164721693843603,
      "epoch": 2.346038818359375e-06,
      "step": 2460
    },
    {
      "epoch": 2.346038818359375e-06,
      "step": 2460,
      "training_step_time": 17.099928617477417
    },
    {
      "epoch": 2.346992492675781e-06,
      "model_forward_time": 5.976729869842529,
      "step": 2461
    },
    {
      "epoch": 2.346992492675781e-06,
      "step": 2461,
      "training_step_time": 17.41329336166382
    },
    {
      "epoch": 2.3479461669921877e-06,
      "model_forward_time": 5.59045934677124,
      "step": 2462
    },
    {
      "epoch": 2.3479461669921877e-06,
      "step": 2462,
      "training_step_time": 17.019497632980347
    },
    {
      "epoch": 2.348899841308594e-06,
      "model_forward_time": 5.5714945793151855,
      "step": 2463
    },
    {
      "epoch": 2.348899841308594e-06,
      "step": 2463,
      "training_step_time": 16.98525309562683
    },
    {
      "epoch": 2.349853515625e-06,
      "model_forward_time": 5.578824281692505,
      "step": 2464
    },
    {
      "epoch": 2.349853515625e-06,
      "step": 2464,
      "training_step_time": 16.98457384109497
    },
    {
      "epoch": 2.350807189941406e-06,
      "model_forward_time": 5.573394060134888,
      "step": 2465
    },
    {
      "epoch": 2.350807189941406e-06,
      "step": 2465,
      "training_step_time": 17.004273414611816
    },
    {
      "epoch": 2.3517608642578127e-06,
      "model_forward_time": 5.592391014099121,
      "step": 2466
    },
    {
      "epoch": 2.3517608642578127e-06,
      "step": 2466,
      "training_step_time": 17.409559965133667
    },
    {
      "epoch": 2.352714538574219e-06,
      "model_forward_time": 5.601110219955444,
      "step": 2467
    },
    {
      "epoch": 2.352714538574219e-06,
      "step": 2467,
      "training_step_time": 17.085469722747803
    },
    {
      "epoch": 2.353668212890625e-06,
      "model_forward_time": 5.609277725219727,
      "step": 2468
    },
    {
      "epoch": 2.353668212890625e-06,
      "step": 2468,
      "training_step_time": 17.089357376098633
    },
    {
      "epoch": 2.354621887207031e-06,
      "model_forward_time": 5.611843824386597,
      "step": 2469
    },
    {
      "epoch": 2.354621887207031e-06,
      "step": 2469,
      "training_step_time": 17.08513641357422
    },
    {
      "epoch": 2.3555755615234377e-06,
      "grad_norm": 1.6259119510650635,
      "learning_rate": 1e-05,
      "loss": 0.0851,
      "step": 2470
    },
    {
      "epoch": 2.3555755615234377e-06,
      "model_forward_time": 5.605083703994751,
      "step": 2470
    },
    {
      "dynamics_loss_avg": 0.063002026733011,
      "epoch": 2.3555755615234377e-06,
      "step": 2470
    },
    {
      "action_loss_avg": 0.01839370548259467,
      "epoch": 2.3555755615234377e-06,
      "step": 2470
    },
    {
      "epoch": 2.3555755615234377e-06,
      "step": 2470,
      "training_step_time": 17.084950923919678
    },
    {
      "epoch": 2.356529235839844e-06,
      "model_forward_time": 5.612749099731445,
      "step": 2471
    },
    {
      "epoch": 2.356529235839844e-06,
      "step": 2471,
      "training_step_time": 17.101358652114868
    },
    {
      "epoch": 2.35748291015625e-06,
      "model_forward_time": 5.978825092315674,
      "step": 2472
    },
    {
      "epoch": 2.35748291015625e-06,
      "step": 2472,
      "training_step_time": 17.470137119293213
    },
    {
      "epoch": 2.358436584472656e-06,
      "model_forward_time": 5.61365270614624,
      "step": 2473
    },
    {
      "epoch": 2.358436584472656e-06,
      "step": 2473,
      "training_step_time": 17.071920156478882
    },
    {
      "epoch": 2.3593902587890627e-06,
      "model_forward_time": 5.589986801147461,
      "step": 2474
    },
    {
      "epoch": 2.3593902587890627e-06,
      "step": 2474,
      "training_step_time": 16.99950861930847
    },
    {
      "epoch": 2.360343933105469e-06,
      "model_forward_time": 5.56748366355896,
      "step": 2475
    },
    {
      "epoch": 2.360343933105469e-06,
      "step": 2475,
      "training_step_time": 16.988232135772705
    },
    {
      "epoch": 2.361297607421875e-06,
      "model_forward_time": 5.580253601074219,
      "step": 2476
    },
    {
      "epoch": 2.361297607421875e-06,
      "step": 2476,
      "training_step_time": 17.009753704071045
    },
    {
      "epoch": 2.362251281738281e-06,
      "model_forward_time": 5.569375991821289,
      "step": 2477
    },
    {
      "epoch": 2.362251281738281e-06,
      "step": 2477,
      "training_step_time": 17.371378660202026
    },
    {
      "epoch": 2.3632049560546877e-06,
      "model_forward_time": 5.5787129402160645,
      "step": 2478
    },
    {
      "epoch": 2.3632049560546877e-06,
      "step": 2478,
      "training_step_time": 17.00147843360901
    },
    {
      "epoch": 2.364158630371094e-06,
      "model_forward_time": 5.587601184844971,
      "step": 2479
    },
    {
      "epoch": 2.364158630371094e-06,
      "step": 2479,
      "training_step_time": 17.04157018661499
    },
    {
      "epoch": 2.3651123046875e-06,
      "grad_norm": 1.7342275381088257,
      "learning_rate": 1e-05,
      "loss": 0.0735,
      "step": 2480
    },
    {
      "epoch": 2.3651123046875e-06,
      "model_forward_time": 5.599032878875732,
      "step": 2480
    },
    {
      "dynamics_loss_avg": 0.05758821740746498,
      "epoch": 2.3651123046875e-06,
      "step": 2480
    },
    {
      "action_loss_avg": 0.0131347949616611,
      "epoch": 2.3651123046875e-06,
      "step": 2480
    },
    {
      "epoch": 2.3651123046875e-06,
      "step": 2480,
      "training_step_time": 17.067139863967896
    },
    {
      "epoch": 2.366065979003906e-06,
      "model_forward_time": 5.598405122756958,
      "step": 2481
    },
    {
      "epoch": 2.366065979003906e-06,
      "step": 2481,
      "training_step_time": 17.07150959968567
    },
    {
      "epoch": 2.3670196533203127e-06,
      "model_forward_time": 5.598004341125488,
      "step": 2482
    },
    {
      "epoch": 2.3670196533203127e-06,
      "step": 2482,
      "training_step_time": 17.05778431892395
    },
    {
      "epoch": 2.367973327636719e-06,
      "model_forward_time": 5.9634246826171875,
      "step": 2483
    },
    {
      "epoch": 2.367973327636719e-06,
      "step": 2483,
      "training_step_time": 17.431777000427246
    },
    {
      "epoch": 2.368927001953125e-06,
      "model_forward_time": 5.608835458755493,
      "step": 2484
    },
    {
      "epoch": 2.368927001953125e-06,
      "step": 2484,
      "training_step_time": 17.0719997882843
    },
    {
      "epoch": 2.369880676269531e-06,
      "model_forward_time": 5.600909233093262,
      "step": 2485
    },
    {
      "epoch": 2.369880676269531e-06,
      "step": 2485,
      "training_step_time": 17.078655242919922
    },
    {
      "epoch": 2.3708343505859377e-06,
      "model_forward_time": 5.602942228317261,
      "step": 2486
    },
    {
      "epoch": 2.3708343505859377e-06,
      "step": 2486,
      "training_step_time": 17.091774702072144
    },
    {
      "epoch": 2.371788024902344e-06,
      "model_forward_time": 5.613932371139526,
      "step": 2487
    },
    {
      "epoch": 2.371788024902344e-06,
      "step": 2487,
      "training_step_time": 17.107609748840332
    },
    {
      "epoch": 2.37274169921875e-06,
      "model_forward_time": 5.953110933303833,
      "step": 2488
    },
    {
      "epoch": 2.37274169921875e-06,
      "step": 2488,
      "training_step_time": 17.454959869384766
    },
    {
      "epoch": 2.373695373535156e-06,
      "model_forward_time": 5.6116416454315186,
      "step": 2489
    },
    {
      "epoch": 2.373695373535156e-06,
      "step": 2489,
      "training_step_time": 17.096156120300293
    },
    {
      "epoch": 2.3746490478515627e-06,
      "grad_norm": 1.5738933086395264,
      "learning_rate": 1e-05,
      "loss": 0.0694,
      "step": 2490
    },
    {
      "epoch": 2.3746490478515627e-06,
      "model_forward_time": 5.6180174350738525,
      "step": 2490
    },
    {
      "dynamics_loss_avg": 0.0516342906281352,
      "epoch": 2.3746490478515627e-06,
      "step": 2490
    },
    {
      "action_loss_avg": 0.014071960095316171,
      "epoch": 2.3746490478515627e-06,
      "step": 2490
    },
    {
      "epoch": 2.3746490478515627e-06,
      "step": 2490,
      "training_step_time": 17.105319023132324
    },
    {
      "epoch": 2.375602722167969e-06,
      "model_forward_time": 5.604219198226929,
      "step": 2491
    },
    {
      "epoch": 2.375602722167969e-06,
      "step": 2491,
      "training_step_time": 17.070242166519165
    },
    {
      "epoch": 2.376556396484375e-06,
      "model_forward_time": 5.593058347702026,
      "step": 2492
    },
    {
      "epoch": 2.376556396484375e-06,
      "step": 2492,
      "training_step_time": 17.04541826248169
    },
    {
      "epoch": 2.377510070800781e-06,
      "model_forward_time": 5.592877149581909,
      "step": 2493
    },
    {
      "epoch": 2.377510070800781e-06,
      "step": 2493,
      "training_step_time": 17.050858736038208
    },
    {
      "epoch": 2.3784637451171877e-06,
      "model_forward_time": 5.976007461547852,
      "step": 2494
    },
    {
      "epoch": 2.3784637451171877e-06,
      "step": 2494,
      "training_step_time": 17.444578647613525
    },
    {
      "epoch": 2.379417419433594e-06,
      "model_forward_time": 5.608694553375244,
      "step": 2495
    },
    {
      "epoch": 2.379417419433594e-06,
      "step": 2495,
      "training_step_time": 17.08859658241272
    },
    {
      "epoch": 2.38037109375e-06,
      "model_forward_time": 5.6080193519592285,
      "step": 2496
    },
    {
      "epoch": 2.38037109375e-06,
      "step": 2496,
      "training_step_time": 17.052669048309326
    },
    {
      "epoch": 2.381324768066406e-06,
      "model_forward_time": 5.601185083389282,
      "step": 2497
    },
    {
      "epoch": 2.381324768066406e-06,
      "step": 2497,
      "training_step_time": 17.056182861328125
    },
    {
      "epoch": 2.3822784423828127e-06,
      "model_forward_time": 5.588564395904541,
      "step": 2498
    },
    {
      "epoch": 2.3822784423828127e-06,
      "step": 2498,
      "training_step_time": 17.026496648788452
    },
    {
      "epoch": 2.383232116699219e-06,
      "model_forward_time": 5.592974424362183,
      "step": 2499
    },
    {
      "epoch": 2.383232116699219e-06,
      "step": 2499,
      "training_step_time": 17.391951084136963
    },
    {
      "epoch": 2.384185791015625e-06,
      "grad_norm": 1.2865333557128906,
      "learning_rate": 1e-05,
      "loss": 0.0887,
      "step": 2500
    },
    {
      "epoch": 2.384185791015625e-06,
      "model_forward_time": 5.597162485122681,
      "step": 2500
    },
    {
      "dynamics_loss_avg": 0.07564935758709908,
      "epoch": 2.384185791015625e-06,
      "step": 2500
    },
    {
      "action_loss_avg": 0.01706512235105038,
      "epoch": 2.384185791015625e-06,
      "step": 2500
    },
    {
      "epoch": 2.384185791015625e-06,
      "step": 2500,
      "training_step_time": 17.041543006896973
    },
    {
      "epoch": 2.385139465332031e-06,
      "model_forward_time": 5.5944788455963135,
      "step": 2501
    },
    {
      "epoch": 2.385139465332031e-06,
      "step": 2501,
      "training_step_time": 17.05924940109253
    },
    {
      "epoch": 2.3860931396484373e-06,
      "model_forward_time": 5.592501878738403,
      "step": 2502
    },
    {
      "epoch": 2.3860931396484373e-06,
      "step": 2502,
      "training_step_time": 17.05926251411438
    },
    {
      "epoch": 2.387046813964844e-06,
      "model_forward_time": 5.608994722366333,
      "step": 2503
    },
    {
      "epoch": 2.387046813964844e-06,
      "step": 2503,
      "training_step_time": 17.087549686431885
    },
    {
      "epoch": 2.38800048828125e-06,
      "model_forward_time": 5.606424570083618,
      "step": 2504
    },
    {
      "epoch": 2.38800048828125e-06,
      "step": 2504,
      "training_step_time": 17.07492733001709
    },
    {
      "epoch": 2.388954162597656e-06,
      "model_forward_time": 5.9669318199157715,
      "step": 2505
    },
    {
      "epoch": 2.388954162597656e-06,
      "step": 2505,
      "training_step_time": 17.44352412223816
    },
    {
      "epoch": 2.3899078369140623e-06,
      "model_forward_time": 5.615166425704956,
      "step": 2506
    },
    {
      "epoch": 2.3899078369140623e-06,
      "step": 2506,
      "training_step_time": 17.097866535186768
    },
    {
      "epoch": 2.390861511230469e-06,
      "model_forward_time": 5.6068713665008545,
      "step": 2507
    },
    {
      "epoch": 2.390861511230469e-06,
      "step": 2507,
      "training_step_time": 17.087911367416382
    },
    {
      "epoch": 2.391815185546875e-06,
      "model_forward_time": 5.614677906036377,
      "step": 2508
    },
    {
      "epoch": 2.391815185546875e-06,
      "step": 2508,
      "training_step_time": 17.091641902923584
    },
    {
      "epoch": 2.392768859863281e-06,
      "model_forward_time": 5.599883794784546,
      "step": 2509
    },
    {
      "epoch": 2.392768859863281e-06,
      "step": 2509,
      "training_step_time": 17.070250988006592
    },
    {
      "epoch": 2.3937225341796873e-06,
      "grad_norm": 2.1426351070404053,
      "learning_rate": 1e-05,
      "loss": 0.1011,
      "step": 2510
    },
    {
      "epoch": 2.3937225341796873e-06,
      "model_forward_time": 5.970048904418945,
      "step": 2510
    },
    {
      "dynamics_loss_avg": 0.07456743530929089,
      "epoch": 2.3937225341796873e-06,
      "step": 2510
    },
    {
      "action_loss_avg": 0.02286598472855985,
      "epoch": 2.3937225341796873e-06,
      "step": 2510
    },
    {
      "epoch": 2.3937225341796873e-06,
      "step": 2510,
      "training_step_time": 17.438255548477173
    },
    {
      "epoch": 2.394676208496094e-06,
      "model_forward_time": 5.605121374130249,
      "step": 2511
    },
    {
      "epoch": 2.394676208496094e-06,
      "step": 2511,
      "training_step_time": 17.07718563079834
    },
    {
      "epoch": 2.3956298828125e-06,
      "model_forward_time": 5.6124114990234375,
      "step": 2512
    },
    {
      "epoch": 2.3956298828125e-06,
      "step": 2512,
      "training_step_time": 17.090747117996216
    },
    {
      "epoch": 2.396583557128906e-06,
      "model_forward_time": 5.610296249389648,
      "step": 2513
    },
    {
      "epoch": 2.396583557128906e-06,
      "step": 2513,
      "training_step_time": 17.088099241256714
    },
    {
      "epoch": 2.3975372314453123e-06,
      "model_forward_time": 5.596046686172485,
      "step": 2514
    },
    {
      "epoch": 2.3975372314453123e-06,
      "step": 2514,
      "training_step_time": 17.044981718063354
    },
    {
      "epoch": 2.398490905761719e-06,
      "model_forward_time": 5.592061996459961,
      "step": 2515
    },
    {
      "epoch": 2.398490905761719e-06,
      "step": 2515,
      "training_step_time": 17.031341314315796
    },
    {
      "epoch": 2.399444580078125e-06,
      "model_forward_time": 5.9395599365234375,
      "step": 2516
    },
    {
      "epoch": 2.399444580078125e-06,
      "step": 2516,
      "training_step_time": 17.38552474975586
    },
    {
      "epoch": 2.400398254394531e-06,
      "model_forward_time": 5.599279403686523,
      "step": 2517
    },
    {
      "epoch": 2.400398254394531e-06,
      "step": 2517,
      "training_step_time": 17.077139854431152
    },
    {
      "epoch": 2.4013519287109373e-06,
      "model_forward_time": 5.6027991771698,
      "step": 2518
    },
    {
      "epoch": 2.4013519287109373e-06,
      "step": 2518,
      "training_step_time": 17.054338216781616
    },
    {
      "epoch": 2.402305603027344e-06,
      "model_forward_time": 5.593801736831665,
      "step": 2519
    },
    {
      "epoch": 2.402305603027344e-06,
      "step": 2519,
      "training_step_time": 17.068832874298096
    },
    {
      "epoch": 2.40325927734375e-06,
      "grad_norm": 0.8076092600822449,
      "learning_rate": 1e-05,
      "loss": 0.0679,
      "step": 2520
    },
    {
      "epoch": 2.40325927734375e-06,
      "model_forward_time": 5.604388475418091,
      "step": 2520
    },
    {
      "dynamics_loss_avg": 0.05448228102177381,
      "epoch": 2.40325927734375e-06,
      "step": 2520
    },
    {
      "action_loss_avg": 0.016838645935058592,
      "epoch": 2.40325927734375e-06,
      "step": 2520
    },
    {
      "epoch": 2.40325927734375e-06,
      "step": 2520,
      "training_step_time": 17.089682579040527
    },
    {
      "epoch": 2.404212951660156e-06,
      "model_forward_time": 5.978556871414185,
      "step": 2521
    },
    {
      "epoch": 2.404212951660156e-06,
      "step": 2521,
      "training_step_time": 17.4673969745636
    },
    {
      "epoch": 2.4051666259765623e-06,
      "model_forward_time": 5.602978944778442,
      "step": 2522
    },
    {
      "epoch": 2.4051666259765623e-06,
      "step": 2522,
      "training_step_time": 17.085761785507202
    },
    {
      "epoch": 2.406120300292969e-06,
      "model_forward_time": 5.6173529624938965,
      "step": 2523
    },
    {
      "epoch": 2.406120300292969e-06,
      "step": 2523,
      "training_step_time": 17.113783836364746
    },
    {
      "epoch": 2.407073974609375e-06,
      "model_forward_time": 5.6125006675720215,
      "step": 2524
    },
    {
      "epoch": 2.407073974609375e-06,
      "step": 2524,
      "training_step_time": 17.118446588516235
    },
    {
      "epoch": 2.408027648925781e-06,
      "model_forward_time": 5.619882822036743,
      "step": 2525
    },
    {
      "epoch": 2.408027648925781e-06,
      "step": 2525,
      "training_step_time": 17.112046480178833
    },
    {
      "epoch": 2.4089813232421873e-06,
      "model_forward_time": 5.608112573623657,
      "step": 2526
    },
    {
      "epoch": 2.4089813232421873e-06,
      "step": 2526,
      "training_step_time": 17.455139636993408
    },
    {
      "epoch": 2.409934997558594e-06,
      "model_forward_time": 5.577115535736084,
      "step": 2527
    },
    {
      "epoch": 2.409934997558594e-06,
      "step": 2527,
      "training_step_time": 16.995298862457275
    },
    {
      "epoch": 2.410888671875e-06,
      "model_forward_time": 5.567253589630127,
      "step": 2528
    },
    {
      "epoch": 2.410888671875e-06,
      "step": 2528,
      "training_step_time": 16.974920511245728
    },
    {
      "epoch": 2.411842346191406e-06,
      "model_forward_time": 5.571352481842041,
      "step": 2529
    },
    {
      "epoch": 2.411842346191406e-06,
      "step": 2529,
      "training_step_time": 16.98928189277649
    },
    {
      "epoch": 2.4127960205078123e-06,
      "grad_norm": 1.0363916158676147,
      "learning_rate": 1e-05,
      "loss": 0.0603,
      "step": 2530
    },
    {
      "epoch": 2.4127960205078123e-06,
      "model_forward_time": 5.5660176277160645,
      "step": 2530
    },
    {
      "dynamics_loss_avg": 0.046766752190887925,
      "epoch": 2.4127960205078123e-06,
      "step": 2530
    },
    {
      "action_loss_avg": 0.010990408947691321,
      "epoch": 2.4127960205078123e-06,
      "step": 2530
    },
    {
      "epoch": 2.4127960205078123e-06,
      "step": 2530,
      "training_step_time": 16.990322589874268
    },
    {
      "epoch": 2.413749694824219e-06,
      "model_forward_time": 5.5690693855285645,
      "step": 2531
    },
    {
      "epoch": 2.413749694824219e-06,
      "step": 2531,
      "training_step_time": 16.972641944885254
    },
    {
      "epoch": 2.414703369140625e-06,
      "model_forward_time": 5.918592691421509,
      "step": 2532
    },
    {
      "epoch": 2.414703369140625e-06,
      "step": 2532,
      "training_step_time": 17.333443641662598
    },
    {
      "epoch": 2.415657043457031e-06,
      "model_forward_time": 5.585752487182617,
      "step": 2533
    },
    {
      "epoch": 2.415657043457031e-06,
      "step": 2533,
      "training_step_time": 16.996400356292725
    },
    {
      "epoch": 2.4166107177734373e-06,
      "model_forward_time": 5.597841501235962,
      "step": 2534
    },
    {
      "epoch": 2.4166107177734373e-06,
      "step": 2534,
      "training_step_time": 17.07790493965149
    },
    {
      "epoch": 2.417564392089844e-06,
      "model_forward_time": 5.614447832107544,
      "step": 2535
    },
    {
      "epoch": 2.417564392089844e-06,
      "step": 2535,
      "training_step_time": 17.131952047348022
    },
    {
      "epoch": 2.41851806640625e-06,
      "model_forward_time": 5.627792596817017,
      "step": 2536
    },
    {
      "epoch": 2.41851806640625e-06,
      "step": 2536,
      "training_step_time": 17.139079093933105
    },
    {
      "epoch": 2.419471740722656e-06,
      "model_forward_time": 5.632314443588257,
      "step": 2537
    },
    {
      "epoch": 2.419471740722656e-06,
      "step": 2537,
      "training_step_time": 17.48197102546692
    },
    {
      "epoch": 2.4204254150390623e-06,
      "model_forward_time": 5.634825229644775,
      "step": 2538
    },
    {
      "epoch": 2.4204254150390623e-06,
      "step": 2538,
      "training_step_time": 17.13191056251526
    },
    {
      "epoch": 2.421379089355469e-06,
      "model_forward_time": 5.589934825897217,
      "step": 2539
    },
    {
      "epoch": 2.421379089355469e-06,
      "step": 2539,
      "training_step_time": 17.008023262023926
    },
    {
      "epoch": 2.422332763671875e-06,
      "grad_norm": 2.0306284427642822,
      "learning_rate": 1e-05,
      "loss": 0.0794,
      "step": 2540
    },
    {
      "epoch": 2.422332763671875e-06,
      "model_forward_time": 5.575640678405762,
      "step": 2540
    },
    {
      "dynamics_loss_avg": 0.06383640449494124,
      "epoch": 2.422332763671875e-06,
      "step": 2540
    },
    {
      "action_loss_avg": 0.020680366549640894,
      "epoch": 2.422332763671875e-06,
      "step": 2540
    },
    {
      "epoch": 2.422332763671875e-06,
      "step": 2540,
      "training_step_time": 16.99996542930603
    },
    {
      "epoch": 2.423286437988281e-06,
      "model_forward_time": 5.574774503707886,
      "step": 2541
    },
    {
      "epoch": 2.423286437988281e-06,
      "step": 2541,
      "training_step_time": 17.005895614624023
    },
    {
      "epoch": 2.4242401123046873e-06,
      "model_forward_time": 5.5720438957214355,
      "step": 2542
    },
    {
      "epoch": 2.4242401123046873e-06,
      "step": 2542,
      "training_step_time": 16.981221199035645
    },
    {
      "epoch": 2.425193786621094e-06,
      "model_forward_time": 5.920393466949463,
      "step": 2543
    },
    {
      "epoch": 2.425193786621094e-06,
      "step": 2543,
      "training_step_time": 17.327370166778564
    },
    {
      "epoch": 2.4261474609375e-06,
      "model_forward_time": 5.587672233581543,
      "step": 2544
    },
    {
      "epoch": 2.4261474609375e-06,
      "step": 2544,
      "training_step_time": 17.015522003173828
    },
    {
      "epoch": 2.427101135253906e-06,
      "model_forward_time": 5.592951774597168,
      "step": 2545
    },
    {
      "epoch": 2.427101135253906e-06,
      "step": 2545,
      "training_step_time": 17.039485216140747
    },
    {
      "epoch": 2.4280548095703123e-06,
      "model_forward_time": 5.60544490814209,
      "step": 2546
    },
    {
      "epoch": 2.4280548095703123e-06,
      "step": 2546,
      "training_step_time": 17.090259075164795
    },
    {
      "epoch": 2.429008483886719e-06,
      "model_forward_time": 5.608174562454224,
      "step": 2547
    },
    {
      "epoch": 2.429008483886719e-06,
      "step": 2547,
      "training_step_time": 17.097584009170532
    },
    {
      "epoch": 2.429962158203125e-06,
      "model_forward_time": 5.96504545211792,
      "step": 2548
    },
    {
      "epoch": 2.429962158203125e-06,
      "step": 2548,
      "training_step_time": 17.39534854888916
    },
    {
      "epoch": 2.430915832519531e-06,
      "model_forward_time": 5.5769829750061035,
      "step": 2549
    },
    {
      "epoch": 2.430915832519531e-06,
      "step": 2549,
      "training_step_time": 17.014262914657593
    },
    {
      "epoch": 2.4318695068359373e-06,
      "grad_norm": 0.49305588006973267,
      "learning_rate": 1e-05,
      "loss": 0.0902,
      "step": 2550
    },
    {
      "epoch": 2.4318695068359373e-06,
      "model_forward_time": 5.568625450134277,
      "step": 2550
    },
    {
      "dynamics_loss_avg": 0.06714189350605011,
      "epoch": 2.4318695068359373e-06,
      "step": 2550
    },
    {
      "action_loss_avg": 0.017024003667756915,
      "epoch": 2.4318695068359373e-06,
      "step": 2550
    },
    {
      "epoch": 2.4318695068359373e-06,
      "step": 2550,
      "training_step_time": 16.985387563705444
    },
    {
      "epoch": 2.432823181152344e-06,
      "model_forward_time": 5.578609943389893,
      "step": 2551
    },
    {
      "epoch": 2.432823181152344e-06,
      "step": 2551,
      "training_step_time": 16.994168996810913
    },
    {
      "epoch": 2.43377685546875e-06,
      "model_forward_time": 5.582104444503784,
      "step": 2552
    },
    {
      "epoch": 2.43377685546875e-06,
      "step": 2552,
      "training_step_time": 17.009270906448364
    },
    {
      "epoch": 2.434730529785156e-06,
      "model_forward_time": 5.573647737503052,
      "step": 2553
    },
    {
      "epoch": 2.434730529785156e-06,
      "step": 2553,
      "training_step_time": 17.38438367843628
    },
    {
      "epoch": 2.4356842041015623e-06,
      "model_forward_time": 5.586763143539429,
      "step": 2554
    },
    {
      "epoch": 2.4356842041015623e-06,
      "step": 2554,
      "training_step_time": 17.031694173812866
    },
    {
      "epoch": 2.436637878417969e-06,
      "model_forward_time": 5.593315839767456,
      "step": 2555
    },
    {
      "epoch": 2.436637878417969e-06,
      "step": 2555,
      "training_step_time": 17.060928106307983
    },
    {
      "epoch": 2.437591552734375e-06,
      "model_forward_time": 5.599113464355469,
      "step": 2556
    },
    {
      "epoch": 2.437591552734375e-06,
      "step": 2556,
      "training_step_time": 17.06594944000244
    },
    {
      "epoch": 2.438545227050781e-06,
      "model_forward_time": 5.593164443969727,
      "step": 2557
    },
    {
      "epoch": 2.438545227050781e-06,
      "step": 2557,
      "training_step_time": 17.064820289611816
    },
    {
      "epoch": 2.4394989013671873e-06,
      "model_forward_time": 5.598336458206177,
      "step": 2558
    },
    {
      "epoch": 2.4394989013671873e-06,
      "step": 2558,
      "training_step_time": 17.073365449905396
    },
    {
      "epoch": 2.440452575683594e-06,
      "model_forward_time": 5.974184513092041,
      "step": 2559
    },
    {
      "epoch": 2.440452575683594e-06,
      "step": 2559,
      "training_step_time": 17.453243494033813
    },
    {
      "epoch": 2.44140625e-06,
      "grad_norm": 0.6514803767204285,
      "learning_rate": 1e-05,
      "loss": 0.076,
      "step": 2560
    },
    {
      "epoch": 2.44140625e-06,
      "model_forward_time": 5.602416276931763,
      "step": 2560
    },
    {
      "dynamics_loss_avg": 0.06379355806857348,
      "epoch": 2.44140625e-06,
      "step": 2560
    },
    {
      "action_loss_avg": 0.01697263792157173,
      "epoch": 2.44140625e-06,
      "step": 2560
    },
    {
      "epoch": 2.44140625e-06,
      "step": 2560,
      "training_step_time": 17.06287693977356
    },
    {
      "epoch": 2.442359924316406e-06,
      "model_forward_time": 5.602638244628906,
      "step": 2561
    },
    {
      "epoch": 2.442359924316406e-06,
      "step": 2561,
      "training_step_time": 17.081281423568726
    },
    {
      "epoch": 2.4433135986328123e-06,
      "model_forward_time": 5.5996222496032715,
      "step": 2562
    },
    {
      "epoch": 2.4433135986328123e-06,
      "step": 2562,
      "training_step_time": 17.04953145980835
    },
    {
      "epoch": 2.444267272949219e-06,
      "model_forward_time": 5.593276262283325,
      "step": 2563
    },
    {
      "epoch": 2.444267272949219e-06,
      "step": 2563,
      "training_step_time": 17.045027256011963
    },
    {
      "epoch": 2.445220947265625e-06,
      "model_forward_time": 5.948846340179443,
      "step": 2564
    },
    {
      "epoch": 2.445220947265625e-06,
      "step": 2564,
      "training_step_time": 17.378028392791748
    },
    {
      "epoch": 2.446174621582031e-06,
      "model_forward_time": 5.591074705123901,
      "step": 2565
    },
    {
      "epoch": 2.446174621582031e-06,
      "step": 2565,
      "training_step_time": 17.033879280090332
    },
    {
      "epoch": 2.4471282958984373e-06,
      "model_forward_time": 5.595256805419922,
      "step": 2566
    },
    {
      "epoch": 2.4471282958984373e-06,
      "step": 2566,
      "training_step_time": 17.06254005432129
    },
    {
      "epoch": 2.448081970214844e-06,
      "model_forward_time": 5.605130434036255,
      "step": 2567
    },
    {
      "epoch": 2.448081970214844e-06,
      "step": 2567,
      "training_step_time": 17.07404112815857
    },
    {
      "epoch": 2.44903564453125e-06,
      "model_forward_time": 5.606077671051025,
      "step": 2568
    },
    {
      "epoch": 2.44903564453125e-06,
      "step": 2568,
      "training_step_time": 17.093263626098633
    },
    {
      "epoch": 2.449989318847656e-06,
      "model_forward_time": 5.604945421218872,
      "step": 2569
    },
    {
      "epoch": 2.449989318847656e-06,
      "step": 2569,
      "training_step_time": 17.09943652153015
    },
    {
      "epoch": 2.4509429931640623e-06,
      "grad_norm": 1.0203487873077393,
      "learning_rate": 1e-05,
      "loss": 0.0821,
      "step": 2570
    },
    {
      "epoch": 2.4509429931640623e-06,
      "model_forward_time": 5.963953495025635,
      "step": 2570
    },
    {
      "dynamics_loss_avg": 0.06051913686096668,
      "epoch": 2.4509429931640623e-06,
      "step": 2570
    },
    {
      "action_loss_avg": 0.02058967910706997,
      "epoch": 2.4509429931640623e-06,
      "step": 2570
    },
    {
      "epoch": 2.4509429931640623e-06,
      "step": 2570,
      "training_step_time": 17.442850351333618
    },
    {
      "epoch": 2.451896667480469e-06,
      "model_forward_time": 5.612018346786499,
      "step": 2571
    },
    {
      "epoch": 2.451896667480469e-06,
      "step": 2571,
      "training_step_time": 17.107725858688354
    },
    {
      "epoch": 2.452850341796875e-06,
      "model_forward_time": 5.618382692337036,
      "step": 2572
    },
    {
      "epoch": 2.452850341796875e-06,
      "step": 2572,
      "training_step_time": 17.11823534965515
    },
    {
      "epoch": 2.453804016113281e-06,
      "model_forward_time": 5.622642278671265,
      "step": 2573
    },
    {
      "epoch": 2.453804016113281e-06,
      "step": 2573,
      "training_step_time": 17.126908779144287
    },
    {
      "epoch": 2.4547576904296874e-06,
      "model_forward_time": 5.61513352394104,
      "step": 2574
    },
    {
      "epoch": 2.4547576904296874e-06,
      "step": 2574,
      "training_step_time": 17.090141773223877
    },
    {
      "epoch": 2.455711364746094e-06,
      "model_forward_time": 5.955167770385742,
      "step": 2575
    },
    {
      "epoch": 2.455711364746094e-06,
      "step": 2575,
      "training_step_time": 17.39267110824585
    },
    {
      "epoch": 2.4566650390625e-06,
      "model_forward_time": 5.593760251998901,
      "step": 2576
    },
    {
      "epoch": 2.4566650390625e-06,
      "step": 2576,
      "training_step_time": 17.024063110351562
    },
    {
      "epoch": 2.457618713378906e-06,
      "model_forward_time": 5.5857157707214355,
      "step": 2577
    },
    {
      "epoch": 2.457618713378906e-06,
      "step": 2577,
      "training_step_time": 17.027708053588867
    },
    {
      "epoch": 2.4585723876953124e-06,
      "model_forward_time": 5.594689607620239,
      "step": 2578
    },
    {
      "epoch": 2.4585723876953124e-06,
      "step": 2578,
      "training_step_time": 17.07191252708435
    },
    {
      "epoch": 2.459526062011719e-06,
      "model_forward_time": 5.610548257827759,
      "step": 2579
    },
    {
      "epoch": 2.459526062011719e-06,
      "step": 2579,
      "training_step_time": 17.111241579055786
    },
    {
      "epoch": 2.460479736328125e-06,
      "grad_norm": 0.8552390933036804,
      "learning_rate": 1e-05,
      "loss": 0.0779,
      "step": 2580
    },
    {
      "epoch": 2.460479736328125e-06,
      "model_forward_time": 5.624622583389282,
      "step": 2580
    },
    {
      "dynamics_loss_avg": 0.0605888195335865,
      "epoch": 2.460479736328125e-06,
      "step": 2580
    },
    {
      "action_loss_avg": 0.016633998276665805,
      "epoch": 2.460479736328125e-06,
      "step": 2580
    },
    {
      "epoch": 2.460479736328125e-06,
      "step": 2580,
      "training_step_time": 17.13235330581665
    },
    {
      "epoch": 2.4614334106445312e-06,
      "model_forward_time": 5.9738450050354,
      "step": 2581
    },
    {
      "epoch": 2.4614334106445312e-06,
      "step": 2581,
      "training_step_time": 17.446784257888794
    },
    {
      "epoch": 2.4623870849609374e-06,
      "model_forward_time": 5.592393159866333,
      "step": 2582
    },
    {
      "epoch": 2.4623870849609374e-06,
      "step": 2582,
      "training_step_time": 17.039524793624878
    },
    {
      "epoch": 2.463340759277344e-06,
      "model_forward_time": 5.591526746749878,
      "step": 2583
    },
    {
      "epoch": 2.463340759277344e-06,
      "step": 2583,
      "training_step_time": 17.032431602478027
    },
    {
      "epoch": 2.46429443359375e-06,
      "model_forward_time": 5.597501277923584,
      "step": 2584
    },
    {
      "epoch": 2.46429443359375e-06,
      "step": 2584,
      "training_step_time": 17.043688774108887
    },
    {
      "epoch": 2.4652481079101562e-06,
      "model_forward_time": 5.596335411071777,
      "step": 2585
    },
    {
      "epoch": 2.4652481079101562e-06,
      "step": 2585,
      "training_step_time": 17.074455499649048
    },
    {
      "epoch": 2.4662017822265624e-06,
      "model_forward_time": 5.965911388397217,
      "step": 2586
    },
    {
      "epoch": 2.4662017822265624e-06,
      "step": 2586,
      "training_step_time": 17.457519054412842
    },
    {
      "epoch": 2.467155456542969e-06,
      "model_forward_time": 5.615194797515869,
      "step": 2587
    },
    {
      "epoch": 2.467155456542969e-06,
      "step": 2587,
      "training_step_time": 17.10053849220276
    },
    {
      "epoch": 2.468109130859375e-06,
      "model_forward_time": 5.608439683914185,
      "step": 2588
    },
    {
      "epoch": 2.468109130859375e-06,
      "step": 2588,
      "training_step_time": 17.098947048187256
    },
    {
      "epoch": 2.4690628051757812e-06,
      "model_forward_time": 5.606674432754517,
      "step": 2589
    },
    {
      "epoch": 2.4690628051757812e-06,
      "step": 2589,
      "training_step_time": 17.102141857147217
    },
    {
      "epoch": 2.4700164794921874e-06,
      "grad_norm": 0.7566547989845276,
      "learning_rate": 1e-05,
      "loss": 0.0824,
      "step": 2590
    },
    {
      "epoch": 2.4700164794921874e-06,
      "model_forward_time": 5.610873222351074,
      "step": 2590
    },
    {
      "dynamics_loss_avg": 0.06395695749670267,
      "epoch": 2.4700164794921874e-06,
      "step": 2590
    },
    {
      "action_loss_avg": 0.013983303494751454,
      "epoch": 2.4700164794921874e-06,
      "step": 2590
    },
    {
      "epoch": 2.4700164794921874e-06,
      "step": 2590,
      "training_step_time": 17.09294891357422
    },
    {
      "epoch": 2.470970153808594e-06,
      "model_forward_time": 5.6116461753845215,
      "step": 2591
    },
    {
      "epoch": 2.470970153808594e-06,
      "step": 2591,
      "training_step_time": 17.46834135055542
    },
    {
      "epoch": 2.471923828125e-06,
      "model_forward_time": 5.604639530181885,
      "step": 2592
    },
    {
      "epoch": 2.471923828125e-06,
      "step": 2592,
      "training_step_time": 17.10217571258545
    },
    {
      "epoch": 2.4728775024414062e-06,
      "model_forward_time": 5.619041919708252,
      "step": 2593
    },
    {
      "epoch": 2.4728775024414062e-06,
      "step": 2593,
      "training_step_time": 17.09997844696045
    },
    {
      "epoch": 2.4738311767578124e-06,
      "model_forward_time": 5.598082065582275,
      "step": 2594
    },
    {
      "epoch": 2.4738311767578124e-06,
      "step": 2594,
      "training_step_time": 17.037025451660156
    },
    {
      "epoch": 2.474784851074219e-06,
      "model_forward_time": 5.5794453620910645,
      "step": 2595
    },
    {
      "epoch": 2.474784851074219e-06,
      "step": 2595,
      "training_step_time": 17.022445678710938
    },
    {
      "epoch": 2.475738525390625e-06,
      "model_forward_time": 5.580349683761597,
      "step": 2596
    },
    {
      "epoch": 2.475738525390625e-06,
      "step": 2596,
      "training_step_time": 17.013737201690674
    },
    {
      "epoch": 2.4766921997070312e-06,
      "model_forward_time": 5.942066192626953,
      "step": 2597
    },
    {
      "epoch": 2.4766921997070312e-06,
      "step": 2597,
      "training_step_time": 17.412821531295776
    },
    {
      "epoch": 2.4776458740234374e-06,
      "model_forward_time": 5.6065497398376465,
      "step": 2598
    },
    {
      "epoch": 2.4776458740234374e-06,
      "step": 2598,
      "training_step_time": 17.104416847229004
    },
    {
      "epoch": 2.478599548339844e-06,
      "model_forward_time": 5.620805501937866,
      "step": 2599
    },
    {
      "epoch": 2.478599548339844e-06,
      "step": 2599,
      "training_step_time": 17.117233991622925
    },
    {
      "epoch": 2.47955322265625e-06,
      "grad_norm": 1.5053731203079224,
      "learning_rate": 1e-05,
      "loss": 0.0746,
      "step": 2600
    },
    {
      "epoch": 2.47955322265625e-06,
      "model_forward_time": 5.621260166168213,
      "step": 2600
    },
    {
      "dynamics_loss_avg": 0.07111380510032177,
      "epoch": 2.47955322265625e-06,
      "step": 2600
    },
    {
      "action_loss_avg": 0.016536259558051825,
      "epoch": 2.47955322265625e-06,
      "step": 2600
    },
    {
      "epoch": 2.47955322265625e-06,
      "step": 2600,
      "training_step_time": 17.12673783302307
    },
    {
      "epoch": 2.4805068969726562e-06,
      "model_forward_time": 5.618954181671143,
      "step": 2601
    },
    {
      "epoch": 2.4805068969726562e-06,
      "step": 2601,
      "training_step_time": 17.10529613494873
    },
    {
      "epoch": 2.4814605712890624e-06,
      "model_forward_time": 5.9649176597595215,
      "step": 2602
    },
    {
      "epoch": 2.4814605712890624e-06,
      "step": 2602,
      "training_step_time": 17.409805297851562
    },
    {
      "epoch": 2.482414245605469e-06,
      "model_forward_time": 5.5719664096832275,
      "step": 2603
    },
    {
      "epoch": 2.482414245605469e-06,
      "step": 2603,
      "training_step_time": 16.98422646522522
    },
    {
      "epoch": 2.483367919921875e-06,
      "model_forward_time": 5.587044954299927,
      "step": 2604
    },
    {
      "epoch": 2.483367919921875e-06,
      "step": 2604,
      "training_step_time": 17.01139497756958
    },
    {
      "epoch": 2.4843215942382812e-06,
      "model_forward_time": 5.56649374961853,
      "step": 2605
    },
    {
      "epoch": 2.4843215942382812e-06,
      "step": 2605,
      "training_step_time": 16.981744050979614
    },
    {
      "epoch": 2.4852752685546874e-06,
      "model_forward_time": 5.5836944580078125,
      "step": 2606
    },
    {
      "epoch": 2.4852752685546874e-06,
      "step": 2606,
      "training_step_time": 17.02816605567932
    },
    {
      "epoch": 2.486228942871094e-06,
      "model_forward_time": 5.573496580123901,
      "step": 2607
    },
    {
      "epoch": 2.486228942871094e-06,
      "step": 2607,
      "training_step_time": 16.987478256225586
    },
    {
      "epoch": 2.4871826171875e-06,
      "model_forward_time": 5.9345831871032715,
      "step": 2608
    },
    {
      "epoch": 2.4871826171875e-06,
      "step": 2608,
      "training_step_time": 17.38731360435486
    },
    {
      "epoch": 2.4881362915039062e-06,
      "model_forward_time": 5.5967857837677,
      "step": 2609
    },
    {
      "epoch": 2.4881362915039062e-06,
      "step": 2609,
      "training_step_time": 17.089031219482422
    },
    {
      "epoch": 2.4890899658203124e-06,
      "grad_norm": 1.0081520080566406,
      "learning_rate": 1e-05,
      "loss": 0.0814,
      "step": 2610
    },
    {
      "epoch": 2.4890899658203124e-06,
      "model_forward_time": 5.617396593093872,
      "step": 2610
    },
    {
      "dynamics_loss_avg": 0.06231098286807537,
      "epoch": 2.4890899658203124e-06,
      "step": 2610
    },
    {
      "action_loss_avg": 0.01652014101855457,
      "epoch": 2.4890899658203124e-06,
      "step": 2610
    },
    {
      "epoch": 2.4890899658203124e-06,
      "step": 2610,
      "training_step_time": 17.133400678634644
    },
    {
      "epoch": 2.490043640136719e-06,
      "model_forward_time": 5.622507810592651,
      "step": 2611
    },
    {
      "epoch": 2.490043640136719e-06,
      "step": 2611,
      "training_step_time": 17.133986949920654
    },
    {
      "epoch": 2.490997314453125e-06,
      "model_forward_time": 5.606847763061523,
      "step": 2612
    },
    {
      "epoch": 2.490997314453125e-06,
      "step": 2612,
      "training_step_time": 17.105600833892822
    },
    {
      "epoch": 2.4919509887695312e-06,
      "model_forward_time": 5.985036849975586,
      "step": 2613
    },
    {
      "epoch": 2.4919509887695312e-06,
      "step": 2613,
      "training_step_time": 17.45562434196472
    },
    {
      "epoch": 2.4929046630859374e-06,
      "model_forward_time": 5.6047985553741455,
      "step": 2614
    },
    {
      "epoch": 2.4929046630859374e-06,
      "step": 2614,
      "training_step_time": 17.079669713974
    },
    {
      "epoch": 2.493858337402344e-06,
      "model_forward_time": 5.596731901168823,
      "step": 2615
    },
    {
      "epoch": 2.493858337402344e-06,
      "step": 2615,
      "training_step_time": 17.066291332244873
    },
    {
      "epoch": 2.49481201171875e-06,
      "model_forward_time": 5.597033262252808,
      "step": 2616
    },
    {
      "epoch": 2.49481201171875e-06,
      "step": 2616,
      "training_step_time": 17.057416439056396
    },
    {
      "epoch": 2.4957656860351562e-06,
      "model_forward_time": 5.594509601593018,
      "step": 2617
    },
    {
      "epoch": 2.4957656860351562e-06,
      "step": 2617,
      "training_step_time": 17.020092725753784
    },
    {
      "epoch": 2.4967193603515624e-06,
      "model_forward_time": 5.588140964508057,
      "step": 2618
    },
    {
      "epoch": 2.4967193603515624e-06,
      "step": 2618,
      "training_step_time": 17.040963649749756
    },
    {
      "epoch": 2.497673034667969e-06,
      "model_forward_time": 5.949080228805542,
      "step": 2619
    },
    {
      "epoch": 2.497673034667969e-06,
      "step": 2619,
      "training_step_time": 17.39663028717041
    },
    {
      "epoch": 2.498626708984375e-06,
      "grad_norm": 2.013338565826416,
      "learning_rate": 1e-05,
      "loss": 0.0765,
      "step": 2620
    },
    {
      "epoch": 2.498626708984375e-06,
      "model_forward_time": 5.598554372787476,
      "step": 2620
    },
    {
      "dynamics_loss_avg": 0.054594510793685914,
      "epoch": 2.498626708984375e-06,
      "step": 2620
    },
    {
      "action_loss_avg": 0.015179119061212987,
      "epoch": 2.498626708984375e-06,
      "step": 2620
    },
    {
      "epoch": 2.498626708984375e-06,
      "step": 2620,
      "training_step_time": 17.076109886169434
    },
    {
      "epoch": 2.4995803833007812e-06,
      "model_forward_time": 5.618740797042847,
      "step": 2621
    },
    {
      "epoch": 2.4995803833007812e-06,
      "step": 2621,
      "training_step_time": 17.112229585647583
    },
    {
      "epoch": 2.5005340576171874e-06,
      "model_forward_time": 5.622880935668945,
      "step": 2622
    },
    {
      "epoch": 2.5005340576171874e-06,
      "step": 2622,
      "training_step_time": 17.13421368598938
    },
    {
      "epoch": 2.501487731933594e-06,
      "model_forward_time": 5.618049621582031,
      "step": 2623
    },
    {
      "epoch": 2.501487731933594e-06,
      "step": 2623,
      "training_step_time": 17.129112720489502
    },
    {
      "epoch": 2.50244140625e-06,
      "model_forward_time": 6.019360065460205,
      "step": 2624
    },
    {
      "epoch": 2.50244140625e-06,
      "step": 2624,
      "training_step_time": 17.518011808395386
    },
    {
      "epoch": 2.5033950805664062e-06,
      "model_forward_time": 5.603916168212891,
      "step": 2625
    },
    {
      "epoch": 2.5033950805664062e-06,
      "step": 2625,
      "training_step_time": 17.05040454864502
    },
    {
      "epoch": 2.5043487548828124e-06,
      "model_forward_time": 5.602916955947876,
      "step": 2626
    },
    {
      "epoch": 2.5043487548828124e-06,
      "step": 2626,
      "training_step_time": 17.061323404312134
    },
    {
      "epoch": 2.5053024291992185e-06,
      "model_forward_time": 5.5915687084198,
      "step": 2627
    },
    {
      "epoch": 2.5053024291992185e-06,
      "step": 2627,
      "training_step_time": 17.045104265213013
    },
    {
      "epoch": 2.506256103515625e-06,
      "model_forward_time": 5.59921932220459,
      "step": 2628
    },
    {
      "epoch": 2.506256103515625e-06,
      "step": 2628,
      "training_step_time": 17.071112871170044
    },
    {
      "epoch": 2.5072097778320313e-06,
      "model_forward_time": 5.607430458068848,
      "step": 2629
    },
    {
      "epoch": 2.5072097778320313e-06,
      "step": 2629,
      "training_step_time": 17.050644636154175
    },
    {
      "epoch": 2.5081634521484374e-06,
      "grad_norm": 1.5483590364456177,
      "learning_rate": 1e-05,
      "loss": 0.0768,
      "step": 2630
    },
    {
      "epoch": 2.5081634521484374e-06,
      "model_forward_time": 5.9286415576934814,
      "step": 2630
    },
    {
      "dynamics_loss_avg": 0.062131761386990544,
      "epoch": 2.5081634521484374e-06,
      "step": 2630
    },
    {
      "action_loss_avg": 0.01705900551751256,
      "epoch": 2.5081634521484374e-06,
      "step": 2630
    },
    {
      "epoch": 2.5081634521484374e-06,
      "step": 2630,
      "training_step_time": 17.402338981628418
    },
    {
      "epoch": 2.5091171264648435e-06,
      "model_forward_time": 5.608201265335083,
      "step": 2631
    },
    {
      "epoch": 2.5091171264648435e-06,
      "step": 2631,
      "training_step_time": 17.11077880859375
    },
    {
      "epoch": 2.51007080078125e-06,
      "model_forward_time": 5.623731374740601,
      "step": 2632
    },
    {
      "epoch": 2.51007080078125e-06,
      "step": 2632,
      "training_step_time": 17.1267991065979
    },
    {
      "epoch": 2.5110244750976563e-06,
      "model_forward_time": 5.628504991531372,
      "step": 2633
    },
    {
      "epoch": 2.5110244750976563e-06,
      "step": 2633,
      "training_step_time": 17.140477418899536
    },
    {
      "epoch": 2.5119781494140624e-06,
      "model_forward_time": 5.6153833866119385,
      "step": 2634
    },
    {
      "epoch": 2.5119781494140624e-06,
      "step": 2634,
      "training_step_time": 17.101215600967407
    },
    {
      "epoch": 2.5129318237304685e-06,
      "model_forward_time": 5.974319934844971,
      "step": 2635
    },
    {
      "epoch": 2.5129318237304685e-06,
      "step": 2635,
      "training_step_time": 17.41119146347046
    },
    {
      "epoch": 2.513885498046875e-06,
      "model_forward_time": 5.597212553024292,
      "step": 2636
    },
    {
      "epoch": 2.513885498046875e-06,
      "step": 2636,
      "training_step_time": 17.031493425369263
    },
    {
      "epoch": 2.5148391723632813e-06,
      "model_forward_time": 5.592570543289185,
      "step": 2637
    },
    {
      "epoch": 2.5148391723632813e-06,
      "step": 2637,
      "training_step_time": 17.062100410461426
    },
    {
      "epoch": 2.5157928466796874e-06,
      "model_forward_time": 5.597775936126709,
      "step": 2638
    },
    {
      "epoch": 2.5157928466796874e-06,
      "step": 2638,
      "training_step_time": 17.072004318237305
    },
    {
      "epoch": 2.5167465209960936e-06,
      "model_forward_time": 5.609617233276367,
      "step": 2639
    },
    {
      "epoch": 2.5167465209960936e-06,
      "step": 2639,
      "training_step_time": 17.028437614440918
    },
    {
      "epoch": 2.5177001953125e-06,
      "grad_norm": 0.45329275727272034,
      "learning_rate": 1e-05,
      "loss": 0.0821,
      "step": 2640
    },
    {
      "epoch": 2.5177001953125e-06,
      "model_forward_time": 5.958292722702026,
      "step": 2640
    },
    {
      "dynamics_loss_avg": 0.06729382481426001,
      "epoch": 2.5177001953125e-06,
      "step": 2640
    },
    {
      "action_loss_avg": 0.01644217499997467,
      "epoch": 2.5177001953125e-06,
      "step": 2640
    },
    {
      "epoch": 2.5177001953125e-06,
      "step": 2640,
      "training_step_time": 17.451961517333984
    },
    {
      "epoch": 2.5186538696289063e-06,
      "model_forward_time": 5.610339403152466,
      "step": 2641
    },
    {
      "epoch": 2.5186538696289063e-06,
      "step": 2641,
      "training_step_time": 17.087576627731323
    },
    {
      "epoch": 2.5196075439453124e-06,
      "model_forward_time": 5.6069581508636475,
      "step": 2642
    },
    {
      "epoch": 2.5196075439453124e-06,
      "step": 2642,
      "training_step_time": 17.070338010787964
    },
    {
      "epoch": 2.5205612182617186e-06,
      "model_forward_time": 5.594323396682739,
      "step": 2643
    },
    {
      "epoch": 2.5205612182617186e-06,
      "step": 2643,
      "training_step_time": 17.040176391601562
    },
    {
      "epoch": 2.521514892578125e-06,
      "model_forward_time": 5.601053953170776,
      "step": 2644
    },
    {
      "epoch": 2.521514892578125e-06,
      "step": 2644,
      "training_step_time": 17.05105495452881
    },
    {
      "epoch": 2.5224685668945313e-06,
      "model_forward_time": 5.591610431671143,
      "step": 2645
    },
    {
      "epoch": 2.5224685668945313e-06,
      "step": 2645,
      "training_step_time": 17.076045989990234
    },
    {
      "epoch": 2.5234222412109374e-06,
      "model_forward_time": 5.963130712509155,
      "step": 2646
    },
    {
      "epoch": 2.5234222412109374e-06,
      "step": 2646,
      "training_step_time": 17.429088592529297
    },
    {
      "epoch": 2.5243759155273436e-06,
      "model_forward_time": 5.590651750564575,
      "step": 2647
    },
    {
      "epoch": 2.5243759155273436e-06,
      "step": 2647,
      "training_step_time": 17.022055625915527
    },
    {
      "epoch": 2.52532958984375e-06,
      "model_forward_time": 5.578173398971558,
      "step": 2648
    },
    {
      "epoch": 2.52532958984375e-06,
      "step": 2648,
      "training_step_time": 17.020896911621094
    },
    {
      "epoch": 2.5262832641601563e-06,
      "model_forward_time": 5.574988842010498,
      "step": 2649
    },
    {
      "epoch": 2.5262832641601563e-06,
      "step": 2649,
      "training_step_time": 16.985086679458618
    },
    {
      "epoch": 2.5272369384765624e-06,
      "grad_norm": 1.389100193977356,
      "learning_rate": 1e-05,
      "loss": 0.0859,
      "step": 2650
    },
    {
      "epoch": 2.5272369384765624e-06,
      "model_forward_time": 5.577797174453735,
      "step": 2650
    },
    {
      "dynamics_loss_avg": 0.064682780764997,
      "epoch": 2.5272369384765624e-06,
      "step": 2650
    },
    {
      "action_loss_avg": 0.016036241594702006,
      "epoch": 2.5272369384765624e-06,
      "step": 2650
    },
    {
      "epoch": 2.5272369384765624e-06,
      "step": 2650,
      "training_step_time": 16.986207485198975
    },
    {
      "epoch": 2.5281906127929686e-06,
      "model_forward_time": 5.579902172088623,
      "step": 2651
    },
    {
      "epoch": 2.5281906127929686e-06,
      "step": 2651,
      "training_step_time": 17.383204221725464
    },
    {
      "epoch": 2.529144287109375e-06,
      "model_forward_time": 5.586807489395142,
      "step": 2652
    },
    {
      "epoch": 2.529144287109375e-06,
      "step": 2652,
      "training_step_time": 17.060341358184814
    },
    {
      "epoch": 2.5300979614257813e-06,
      "model_forward_time": 5.615135908126831,
      "step": 2653
    },
    {
      "epoch": 2.5300979614257813e-06,
      "step": 2653,
      "training_step_time": 17.12487506866455
    },
    {
      "epoch": 2.5310516357421874e-06,
      "model_forward_time": 5.644196510314941,
      "step": 2654
    },
    {
      "epoch": 2.5310516357421874e-06,
      "step": 2654,
      "training_step_time": 17.162740230560303
    },
    {
      "epoch": 2.5320053100585936e-06,
      "model_forward_time": 5.617121696472168,
      "step": 2655
    },
    {
      "epoch": 2.5320053100585936e-06,
      "step": 2655,
      "training_step_time": 17.10553503036499
    },
    {
      "epoch": 2.532958984375e-06,
      "model_forward_time": 5.608955383300781,
      "step": 2656
    },
    {
      "epoch": 2.532958984375e-06,
      "step": 2656,
      "training_step_time": 17.087414741516113
    },
    {
      "epoch": 2.5339126586914063e-06,
      "model_forward_time": 5.966227769851685,
      "step": 2657
    },
    {
      "epoch": 2.5339126586914063e-06,
      "step": 2657,
      "training_step_time": 17.419718980789185
    },
    {
      "epoch": 2.5348663330078124e-06,
      "model_forward_time": 5.587800979614258,
      "step": 2658
    },
    {
      "epoch": 2.5348663330078124e-06,
      "step": 2658,
      "training_step_time": 17.03429627418518
    },
    {
      "epoch": 2.5358200073242186e-06,
      "model_forward_time": 5.583116292953491,
      "step": 2659
    },
    {
      "epoch": 2.5358200073242186e-06,
      "step": 2659,
      "training_step_time": 17.0254864692688
    },
    {
      "epoch": 2.536773681640625e-06,
      "grad_norm": 1.244279146194458,
      "learning_rate": 1e-05,
      "loss": 0.068,
      "step": 2660
    },
    {
      "epoch": 2.536773681640625e-06,
      "model_forward_time": 5.585575103759766,
      "step": 2660
    },
    {
      "dynamics_loss_avg": 0.06399653237313033,
      "epoch": 2.536773681640625e-06,
      "step": 2660
    },
    {
      "action_loss_avg": 0.012498407694511115,
      "epoch": 2.536773681640625e-06,
      "step": 2660
    },
    {
      "epoch": 2.536773681640625e-06,
      "step": 2660,
      "training_step_time": 17.04101014137268
    },
    {
      "epoch": 2.5377273559570313e-06,
      "model_forward_time": 5.593277215957642,
      "step": 2661
    },
    {
      "epoch": 2.5377273559570313e-06,
      "step": 2661,
      "training_step_time": 17.06972885131836
    },
    {
      "epoch": 2.5386810302734374e-06,
      "model_forward_time": 5.608531475067139,
      "step": 2662
    },
    {
      "epoch": 2.5386810302734374e-06,
      "step": 2662,
      "training_step_time": 17.4568829536438
    },
    {
      "epoch": 2.5396347045898436e-06,
      "model_forward_time": 5.60809326171875,
      "step": 2663
    },
    {
      "epoch": 2.5396347045898436e-06,
      "step": 2663,
      "training_step_time": 17.109057426452637
    },
    {
      "epoch": 2.54058837890625e-06,
      "model_forward_time": 5.618780136108398,
      "step": 2664
    },
    {
      "epoch": 2.54058837890625e-06,
      "step": 2664,
      "training_step_time": 17.11474370956421
    },
    {
      "epoch": 2.5415420532226563e-06,
      "model_forward_time": 5.594776153564453,
      "step": 2665
    },
    {
      "epoch": 2.5415420532226563e-06,
      "step": 2665,
      "training_step_time": 17.053855180740356
    },
    {
      "epoch": 2.5424957275390624e-06,
      "model_forward_time": 5.584499359130859,
      "step": 2666
    },
    {
      "epoch": 2.5424957275390624e-06,
      "step": 2666,
      "training_step_time": 16.999313831329346
    },
    {
      "epoch": 2.5434494018554686e-06,
      "model_forward_time": 5.5963664054870605,
      "step": 2667
    },
    {
      "epoch": 2.5434494018554686e-06,
      "step": 2667,
      "training_step_time": 17.044459342956543
    },
    {
      "epoch": 2.544403076171875e-06,
      "model_forward_time": 5.983582258224487,
      "step": 2668
    },
    {
      "epoch": 2.544403076171875e-06,
      "step": 2668,
      "training_step_time": 17.45170569419861
    },
    {
      "epoch": 2.5453567504882813e-06,
      "model_forward_time": 5.611183404922485,
      "step": 2669
    },
    {
      "epoch": 2.5453567504882813e-06,
      "step": 2669,
      "training_step_time": 17.117250204086304
    },
    {
      "epoch": 2.5463104248046874e-06,
      "grad_norm": 2.23824143409729,
      "learning_rate": 1e-05,
      "loss": 0.0711,
      "step": 2670
    },
    {
      "epoch": 2.5463104248046874e-06,
      "model_forward_time": 5.617654323577881,
      "step": 2670
    },
    {
      "dynamics_loss_avg": 0.054061229526996615,
      "epoch": 2.5463104248046874e-06,
      "step": 2670
    },
    {
      "action_loss_avg": 0.014355834806337953,
      "epoch": 2.5463104248046874e-06,
      "step": 2670
    },
    {
      "epoch": 2.5463104248046874e-06,
      "step": 2670,
      "training_step_time": 17.12967276573181
    },
    {
      "epoch": 2.5472640991210936e-06,
      "model_forward_time": 5.6150336265563965,
      "step": 2671
    },
    {
      "epoch": 2.5472640991210936e-06,
      "step": 2671,
      "training_step_time": 17.11811137199402
    },
    {
      "epoch": 2.5482177734375e-06,
      "model_forward_time": 5.624894618988037,
      "step": 2672
    },
    {
      "epoch": 2.5482177734375e-06,
      "step": 2672,
      "training_step_time": 17.139063119888306
    },
    {
      "epoch": 2.5491714477539063e-06,
      "model_forward_time": 5.9707372188568115,
      "step": 2673
    },
    {
      "epoch": 2.5491714477539063e-06,
      "step": 2673,
      "training_step_time": 17.470070838928223
    },
    {
      "epoch": 2.5501251220703124e-06,
      "model_forward_time": 5.6217124462127686,
      "step": 2674
    },
    {
      "epoch": 2.5501251220703124e-06,
      "step": 2674,
      "training_step_time": 17.119441986083984
    },
    {
      "epoch": 2.5510787963867186e-06,
      "model_forward_time": 5.61838173866272,
      "step": 2675
    },
    {
      "epoch": 2.5510787963867186e-06,
      "step": 2675,
      "training_step_time": 17.089542865753174
    },
    {
      "epoch": 2.552032470703125e-06,
      "model_forward_time": 5.590556383132935,
      "step": 2676
    },
    {
      "epoch": 2.552032470703125e-06,
      "step": 2676,
      "training_step_time": 17.020493507385254
    },
    {
      "epoch": 2.5529861450195313e-06,
      "model_forward_time": 5.579942941665649,
      "step": 2677
    },
    {
      "epoch": 2.5529861450195313e-06,
      "step": 2677,
      "training_step_time": 16.987765550613403
    },
    {
      "epoch": 2.5539398193359374e-06,
      "model_forward_time": 5.58447003364563,
      "step": 2678
    },
    {
      "epoch": 2.5539398193359374e-06,
      "step": 2678,
      "training_step_time": 17.026827573776245
    },
    {
      "epoch": 2.5548934936523436e-06,
      "model_forward_time": 5.942332029342651,
      "step": 2679
    },
    {
      "epoch": 2.5548934936523436e-06,
      "step": 2679,
      "training_step_time": 17.374224424362183
    },
    {
      "epoch": 2.55584716796875e-06,
      "grad_norm": 0.4344628155231476,
      "learning_rate": 1e-05,
      "loss": 0.0667,
      "step": 2680
    },
    {
      "epoch": 2.55584716796875e-06,
      "model_forward_time": 5.591200828552246,
      "step": 2680
    },
    {
      "dynamics_loss_avg": 0.049622312374413016,
      "epoch": 2.55584716796875e-06,
      "step": 2680
    },
    {
      "action_loss_avg": 0.011649132706224918,
      "epoch": 2.55584716796875e-06,
      "step": 2680
    },
    {
      "epoch": 2.55584716796875e-06,
      "step": 2680,
      "training_step_time": 17.02885937690735
    },
    {
      "epoch": 2.5568008422851563e-06,
      "model_forward_time": 5.613122463226318,
      "step": 2681
    },
    {
      "epoch": 2.5568008422851563e-06,
      "step": 2681,
      "training_step_time": 17.05300211906433
    },
    {
      "epoch": 2.5577545166015624e-06,
      "model_forward_time": 5.603592157363892,
      "step": 2682
    },
    {
      "epoch": 2.5577545166015624e-06,
      "step": 2682,
      "training_step_time": 17.089393377304077
    },
    {
      "epoch": 2.5587081909179686e-06,
      "model_forward_time": 5.6230762004852295,
      "step": 2683
    },
    {
      "epoch": 2.5587081909179686e-06,
      "step": 2683,
      "training_step_time": 17.122100353240967
    },
    {
      "epoch": 2.559661865234375e-06,
      "model_forward_time": 5.62838077545166,
      "step": 2684
    },
    {
      "epoch": 2.559661865234375e-06,
      "step": 2684,
      "training_step_time": 17.478413820266724
    },
    {
      "epoch": 2.5606155395507813e-06,
      "model_forward_time": 5.622301816940308,
      "step": 2685
    },
    {
      "epoch": 2.5606155395507813e-06,
      "step": 2685,
      "training_step_time": 17.091960430145264
    },
    {
      "epoch": 2.5615692138671874e-06,
      "model_forward_time": 5.593614816665649,
      "step": 2686
    },
    {
      "epoch": 2.5615692138671874e-06,
      "step": 2686,
      "training_step_time": 17.035436391830444
    },
    {
      "epoch": 2.5625228881835936e-06,
      "model_forward_time": 5.597020387649536,
      "step": 2687
    },
    {
      "epoch": 2.5625228881835936e-06,
      "step": 2687,
      "training_step_time": 17.029858827590942
    },
    {
      "epoch": 2.5634765625e-06,
      "model_forward_time": 5.589454889297485,
      "step": 2688
    },
    {
      "epoch": 2.5634765625e-06,
      "step": 2688,
      "training_step_time": 17.02316117286682
    },
    {
      "epoch": 2.5644302368164063e-06,
      "model_forward_time": 5.589358329772949,
      "step": 2689
    },
    {
      "epoch": 2.5644302368164063e-06,
      "step": 2689,
      "training_step_time": 17.04556369781494
    },
    {
      "epoch": 2.5653839111328124e-06,
      "grad_norm": 1.5792361497879028,
      "learning_rate": 1e-05,
      "loss": 0.0966,
      "step": 2690
    },
    {
      "epoch": 2.5653839111328124e-06,
      "model_forward_time": 5.9629857540130615,
      "step": 2690
    },
    {
      "dynamics_loss_avg": 0.07800607774406672,
      "epoch": 2.5653839111328124e-06,
      "step": 2690
    },
    {
      "action_loss_avg": 0.021140426211059094,
      "epoch": 2.5653839111328124e-06,
      "step": 2690
    },
    {
      "epoch": 2.5653839111328124e-06,
      "step": 2690,
      "training_step_time": 17.412617921829224
    },
    {
      "epoch": 2.5663375854492186e-06,
      "model_forward_time": 5.6021223068237305,
      "step": 2691
    },
    {
      "epoch": 2.5663375854492186e-06,
      "step": 2691,
      "training_step_time": 17.008041381835938
    },
    {
      "epoch": 2.567291259765625e-06,
      "model_forward_time": 5.605305910110474,
      "step": 2692
    },
    {
      "epoch": 2.567291259765625e-06,
      "step": 2692,
      "training_step_time": 17.070352792739868
    },
    {
      "epoch": 2.5682449340820313e-06,
      "model_forward_time": 5.598685264587402,
      "step": 2693
    },
    {
      "epoch": 2.5682449340820313e-06,
      "step": 2693,
      "training_step_time": 17.06795048713684
    },
    {
      "epoch": 2.5691986083984375e-06,
      "model_forward_time": 5.597334861755371,
      "step": 2694
    },
    {
      "epoch": 2.5691986083984375e-06,
      "step": 2694,
      "training_step_time": 17.04000949859619
    },
    {
      "epoch": 2.5701522827148436e-06,
      "model_forward_time": 5.5921714305877686,
      "step": 2695
    },
    {
      "epoch": 2.5701522827148436e-06,
      "step": 2695,
      "training_step_time": 17.42292809486389
    },
    {
      "epoch": 2.57110595703125e-06,
      "model_forward_time": 5.604165315628052,
      "step": 2696
    },
    {
      "epoch": 2.57110595703125e-06,
      "step": 2696,
      "training_step_time": 17.09137725830078
    },
    {
      "epoch": 2.5720596313476563e-06,
      "model_forward_time": 5.602344751358032,
      "step": 2697
    },
    {
      "epoch": 2.5720596313476563e-06,
      "step": 2697,
      "training_step_time": 17.103895902633667
    },
    {
      "epoch": 2.5730133056640625e-06,
      "model_forward_time": 5.613284111022949,
      "step": 2698
    },
    {
      "epoch": 2.5730133056640625e-06,
      "step": 2698,
      "training_step_time": 17.11400556564331
    },
    {
      "epoch": 2.5739669799804686e-06,
      "model_forward_time": 5.600677728652954,
      "step": 2699
    },
    {
      "epoch": 2.5739669799804686e-06,
      "step": 2699,
      "training_step_time": 17.10637378692627
    },
    {
      "epoch": 2.574920654296875e-06,
      "grad_norm": 0.6356464624404907,
      "learning_rate": 1e-05,
      "loss": 0.0845,
      "step": 2700
    },
    {
      "epoch": 2.574920654296875e-06,
      "model_forward_time": 5.605792760848999,
      "step": 2700
    },
    {
      "dynamics_loss_avg": 0.07139370124787092,
      "epoch": 2.574920654296875e-06,
      "step": 2700
    },
    {
      "action_loss_avg": 0.013987787626683712,
      "epoch": 2.574920654296875e-06,
      "step": 2700
    },
    {
      "epoch": 2.574920654296875e-06,
      "step": 2700,
      "training_step_time": 17.077943801879883
    },
    {
      "epoch": 2.5758743286132813e-06,
      "model_forward_time": 5.944730520248413,
      "step": 2701
    },
    {
      "epoch": 2.5758743286132813e-06,
      "step": 2701,
      "training_step_time": 17.388811111450195
    },
    {
      "epoch": 2.5768280029296875e-06,
      "model_forward_time": 5.582675933837891,
      "step": 2702
    },
    {
      "epoch": 2.5768280029296875e-06,
      "step": 2702,
      "training_step_time": 17.002243518829346
    },
    {
      "epoch": 2.5777816772460936e-06,
      "model_forward_time": 5.580811023712158,
      "step": 2703
    },
    {
      "epoch": 2.5777816772460936e-06,
      "step": 2703,
      "training_step_time": 16.995715141296387
    },
    {
      "epoch": 2.5787353515625e-06,
      "model_forward_time": 5.583183526992798,
      "step": 2704
    },
    {
      "epoch": 2.5787353515625e-06,
      "step": 2704,
      "training_step_time": 17.009978771209717
    },
    {
      "epoch": 2.5796890258789063e-06,
      "model_forward_time": 5.583017587661743,
      "step": 2705
    },
    {
      "epoch": 2.5796890258789063e-06,
      "step": 2705,
      "training_step_time": 17.02077579498291
    },
    {
      "epoch": 2.5806427001953125e-06,
      "model_forward_time": 5.944655656814575,
      "step": 2706
    },
    {
      "epoch": 2.5806427001953125e-06,
      "step": 2706,
      "training_step_time": 17.372015476226807
    },
    {
      "epoch": 2.5815963745117186e-06,
      "model_forward_time": 5.596362113952637,
      "step": 2707
    },
    {
      "epoch": 2.5815963745117186e-06,
      "step": 2707,
      "training_step_time": 17.05342149734497
    },
    {
      "epoch": 2.582550048828125e-06,
      "model_forward_time": 5.599777460098267,
      "step": 2708
    },
    {
      "epoch": 2.582550048828125e-06,
      "step": 2708,
      "training_step_time": 17.093645572662354
    },
    {
      "epoch": 2.5835037231445313e-06,
      "model_forward_time": 5.616459131240845,
      "step": 2709
    },
    {
      "epoch": 2.5835037231445313e-06,
      "step": 2709,
      "training_step_time": 17.09971308708191
    },
    {
      "epoch": 2.5844573974609375e-06,
      "grad_norm": 1.4996271133422852,
      "learning_rate": 1e-05,
      "loss": 0.0873,
      "step": 2710
    },
    {
      "epoch": 2.5844573974609375e-06,
      "model_forward_time": 5.59827184677124,
      "step": 2710
    },
    {
      "dynamics_loss_avg": 0.06956152878701687,
      "epoch": 2.5844573974609375e-06,
      "step": 2710
    },
    {
      "action_loss_avg": 0.022094758972525597,
      "epoch": 2.5844573974609375e-06,
      "step": 2710
    },
    {
      "epoch": 2.5844573974609375e-06,
      "step": 2710,
      "training_step_time": 17.052282333374023
    },
    {
      "epoch": 2.5854110717773436e-06,
      "model_forward_time": 5.592013359069824,
      "step": 2711
    },
    {
      "epoch": 2.5854110717773436e-06,
      "step": 2711,
      "training_step_time": 17.017387866973877
    },
    {
      "epoch": 2.58636474609375e-06,
      "model_forward_time": 5.946134328842163,
      "step": 2712
    },
    {
      "epoch": 2.58636474609375e-06,
      "step": 2712,
      "training_step_time": 17.383748292922974
    },
    {
      "epoch": 2.5873184204101563e-06,
      "model_forward_time": 5.5875184535980225,
      "step": 2713
    },
    {
      "epoch": 2.5873184204101563e-06,
      "step": 2713,
      "training_step_time": 17.048208236694336
    },
    {
      "epoch": 2.5882720947265625e-06,
      "model_forward_time": 5.59326171875,
      "step": 2714
    },
    {
      "epoch": 2.5882720947265625e-06,
      "step": 2714,
      "training_step_time": 17.06978440284729
    },
    {
      "epoch": 2.5892257690429686e-06,
      "model_forward_time": 5.597944021224976,
      "step": 2715
    },
    {
      "epoch": 2.5892257690429686e-06,
      "step": 2715,
      "training_step_time": 17.064796924591064
    },
    {
      "epoch": 2.590179443359375e-06,
      "model_forward_time": 5.585612058639526,
      "step": 2716
    },
    {
      "epoch": 2.590179443359375e-06,
      "step": 2716,
      "training_step_time": 17.031394958496094
    },
    {
      "epoch": 2.5911331176757813e-06,
      "model_forward_time": 5.599416255950928,
      "step": 2717
    },
    {
      "epoch": 2.5911331176757813e-06,
      "step": 2717,
      "training_step_time": 17.43002200126648
    },
    {
      "epoch": 2.5920867919921875e-06,
      "model_forward_time": 5.602109670639038,
      "step": 2718
    },
    {
      "epoch": 2.5920867919921875e-06,
      "step": 2718,
      "training_step_time": 17.06300687789917
    },
    {
      "epoch": 2.5930404663085936e-06,
      "model_forward_time": 5.596321105957031,
      "step": 2719
    },
    {
      "epoch": 2.5930404663085936e-06,
      "step": 2719,
      "training_step_time": 17.066712617874146
    },
    {
      "epoch": 2.593994140625e-06,
      "grad_norm": 1.483879804611206,
      "learning_rate": 1e-05,
      "loss": 0.0856,
      "step": 2720
    },
    {
      "epoch": 2.593994140625e-06,
      "model_forward_time": 5.5938966274261475,
      "step": 2720
    },
    {
      "dynamics_loss_avg": 0.06251761130988598,
      "epoch": 2.593994140625e-06,
      "step": 2720
    },
    {
      "action_loss_avg": 0.017415143130347133,
      "epoch": 2.593994140625e-06,
      "step": 2720
    },
    {
      "epoch": 2.593994140625e-06,
      "step": 2720,
      "training_step_time": 17.037641525268555
    },
    {
      "epoch": 2.5949478149414063e-06,
      "model_forward_time": 5.59231972694397,
      "step": 2721
    },
    {
      "epoch": 2.5949478149414063e-06,
      "step": 2721,
      "training_step_time": 17.023594617843628
    },
    {
      "epoch": 2.5959014892578125e-06,
      "model_forward_time": 5.581385850906372,
      "step": 2722
    },
    {
      "epoch": 2.5959014892578125e-06,
      "step": 2722,
      "training_step_time": 17.005419492721558
    },
    {
      "epoch": 2.5968551635742186e-06,
      "model_forward_time": 5.974357843399048,
      "step": 2723
    },
    {
      "epoch": 2.5968551635742186e-06,
      "step": 2723,
      "training_step_time": 17.415691614151
    },
    {
      "epoch": 2.597808837890625e-06,
      "model_forward_time": 5.588854551315308,
      "step": 2724
    },
    {
      "epoch": 2.597808837890625e-06,
      "step": 2724,
      "training_step_time": 17.02352547645569
    },
    {
      "epoch": 2.5987625122070313e-06,
      "model_forward_time": 5.591522693634033,
      "step": 2725
    },
    {
      "epoch": 2.5987625122070313e-06,
      "step": 2725,
      "training_step_time": 17.026286602020264
    },
    {
      "epoch": 2.5997161865234375e-06,
      "model_forward_time": 5.597378253936768,
      "step": 2726
    },
    {
      "epoch": 2.5997161865234375e-06,
      "step": 2726,
      "training_step_time": 17.06219792366028
    },
    {
      "epoch": 2.6006698608398436e-06,
      "model_forward_time": 5.611578941345215,
      "step": 2727
    },
    {
      "epoch": 2.6006698608398436e-06,
      "step": 2727,
      "training_step_time": 17.095851182937622
    },
    {
      "epoch": 2.60162353515625e-06,
      "model_forward_time": 5.614514589309692,
      "step": 2728
    },
    {
      "epoch": 2.60162353515625e-06,
      "step": 2728,
      "training_step_time": 17.486622095108032
    },
    {
      "epoch": 2.6025772094726563e-06,
      "model_forward_time": 5.628244161605835,
      "step": 2729
    },
    {
      "epoch": 2.6025772094726563e-06,
      "step": 2729,
      "training_step_time": 17.142256498336792
    },
    {
      "epoch": 2.6035308837890625e-06,
      "grad_norm": 0.5710092782974243,
      "learning_rate": 1e-05,
      "loss": 0.0689,
      "step": 2730
    },
    {
      "epoch": 2.6035308837890625e-06,
      "model_forward_time": 5.62141489982605,
      "step": 2730
    },
    {
      "dynamics_loss_avg": 0.053014243580400945,
      "epoch": 2.6035308837890625e-06,
      "step": 2730
    },
    {
      "action_loss_avg": 0.014928370760753751,
      "epoch": 2.6035308837890625e-06,
      "step": 2730
    },
    {
      "epoch": 2.6035308837890625e-06,
      "step": 2730,
      "training_step_time": 17.122023820877075
    },
    {
      "epoch": 2.6044845581054686e-06,
      "model_forward_time": 5.626364469528198,
      "step": 2731
    },
    {
      "epoch": 2.6044845581054686e-06,
      "step": 2731,
      "training_step_time": 17.11216688156128
    },
    {
      "epoch": 2.605438232421875e-06,
      "model_forward_time": 5.617387056350708,
      "step": 2732
    },
    {
      "epoch": 2.605438232421875e-06,
      "step": 2732,
      "training_step_time": 17.096402168273926
    },
    {
      "epoch": 2.6063919067382813e-06,
      "model_forward_time": 5.603975534439087,
      "step": 2733
    },
    {
      "epoch": 2.6063919067382813e-06,
      "step": 2733,
      "training_step_time": 17.087058305740356
    },
    {
      "epoch": 2.6073455810546875e-06,
      "model_forward_time": 5.98207688331604,
      "step": 2734
    },
    {
      "epoch": 2.6073455810546875e-06,
      "step": 2734,
      "training_step_time": 17.460798025131226
    },
    {
      "epoch": 2.6082992553710936e-06,
      "model_forward_time": 5.626728296279907,
      "step": 2735
    },
    {
      "epoch": 2.6082992553710936e-06,
      "step": 2735,
      "training_step_time": 17.085310697555542
    },
    {
      "epoch": 2.6092529296875e-06,
      "model_forward_time": 5.594570636749268,
      "step": 2736
    },
    {
      "epoch": 2.6092529296875e-06,
      "step": 2736,
      "training_step_time": 17.064406394958496
    },
    {
      "epoch": 2.6102066040039063e-06,
      "model_forward_time": 4.282132148742676,
      "step": 2737
    },
    {
      "epoch": 2.6102066040039063e-06,
      "step": 2737,
      "training_step_time": 12.805123567581177
    },
    {
      "epoch": 2.6111602783203125e-06,
      "model_forward_time": 5.6046881675720215,
      "step": 2738
    },
    {
      "epoch": 2.6111602783203125e-06,
      "step": 2738,
      "training_step_time": 17.09696316719055
    },
    {
      "epoch": 2.6121139526367186e-06,
      "model_forward_time": 5.967001438140869,
      "step": 2739
    },
    {
      "epoch": 2.6121139526367186e-06,
      "step": 2739,
      "training_step_time": 17.45165491104126
    },
    {
      "epoch": 2.613067626953125e-06,
      "grad_norm": 2.1089797019958496,
      "learning_rate": 1e-05,
      "loss": 0.0852,
      "step": 2740
    },
    {
      "epoch": 2.613067626953125e-06,
      "model_forward_time": 5.602176189422607,
      "step": 2740
    },
    {
      "dynamics_loss_avg": 0.061772657185792924,
      "epoch": 2.613067626953125e-06,
      "step": 2740
    },
    {
      "action_loss_avg": 0.019184819795191287,
      "epoch": 2.613067626953125e-06,
      "step": 2740
    },
    {
      "epoch": 2.613067626953125e-06,
      "step": 2740,
      "training_step_time": 17.076900720596313
    },
    {
      "epoch": 2.6140213012695313e-06,
      "model_forward_time": 5.603511810302734,
      "step": 2741
    },
    {
      "epoch": 2.6140213012695313e-06,
      "step": 2741,
      "training_step_time": 17.06856894493103
    },
    {
      "epoch": 2.6149749755859375e-06,
      "model_forward_time": 5.5993812084198,
      "step": 2742
    },
    {
      "epoch": 2.6149749755859375e-06,
      "step": 2742,
      "training_step_time": 17.061538696289062
    },
    {
      "epoch": 2.6159286499023436e-06,
      "model_forward_time": 5.594053506851196,
      "step": 2743
    },
    {
      "epoch": 2.6159286499023436e-06,
      "step": 2743,
      "training_step_time": 17.035567045211792
    },
    {
      "epoch": 2.61688232421875e-06,
      "model_forward_time": 5.606360673904419,
      "step": 2744
    },
    {
      "epoch": 2.61688232421875e-06,
      "step": 2744,
      "training_step_time": 17.09447932243347
    },
    {
      "epoch": 2.6178359985351564e-06,
      "model_forward_time": 5.975698709487915,
      "step": 2745
    },
    {
      "epoch": 2.6178359985351564e-06,
      "step": 2745,
      "training_step_time": 17.468692541122437
    },
    {
      "epoch": 2.6187896728515625e-06,
      "model_forward_time": 5.607764482498169,
      "step": 2746
    },
    {
      "epoch": 2.6187896728515625e-06,
      "step": 2746,
      "training_step_time": 17.09012484550476
    },
    {
      "epoch": 2.6197433471679686e-06,
      "model_forward_time": 5.615151882171631,
      "step": 2747
    },
    {
      "epoch": 2.6197433471679686e-06,
      "step": 2747,
      "training_step_time": 17.10914969444275
    },
    {
      "epoch": 2.620697021484375e-06,
      "model_forward_time": 5.612270355224609,
      "step": 2748
    },
    {
      "epoch": 2.620697021484375e-06,
      "step": 2748,
      "training_step_time": 17.108518362045288
    },
    {
      "epoch": 2.6216506958007814e-06,
      "model_forward_time": 5.598994970321655,
      "step": 2749
    },
    {
      "epoch": 2.6216506958007814e-06,
      "step": 2749,
      "training_step_time": 17.07649278640747
    },
    {
      "epoch": 2.6226043701171875e-06,
      "grad_norm": 0.890259325504303,
      "learning_rate": 1e-05,
      "loss": 0.0658,
      "step": 2750
    },
    {
      "epoch": 2.6226043701171875e-06,
      "model_forward_time": 5.598550319671631,
      "step": 2750
    },
    {
      "dynamics_loss_avg": 0.05804004911333323,
      "epoch": 2.6226043701171875e-06,
      "step": 2750
    },
    {
      "action_loss_avg": 0.013930420740507543,
      "epoch": 2.6226043701171875e-06,
      "step": 2750
    },
    {
      "epoch": 2.6226043701171875e-06,
      "step": 2750,
      "training_step_time": 17.394312620162964
    },
    {
      "epoch": 2.6235580444335936e-06,
      "model_forward_time": 5.586942195892334,
      "step": 2751
    },
    {
      "epoch": 2.6235580444335936e-06,
      "step": 2751,
      "training_step_time": 17.041122436523438
    },
    {
      "epoch": 2.62451171875e-06,
      "model_forward_time": 5.608994483947754,
      "step": 2752
    },
    {
      "epoch": 2.62451171875e-06,
      "step": 2752,
      "training_step_time": 17.106205940246582
    },
    {
      "epoch": 2.6254653930664064e-06,
      "model_forward_time": 5.610419988632202,
      "step": 2753
    },
    {
      "epoch": 2.6254653930664064e-06,
      "step": 2753,
      "training_step_time": 17.10880947113037
    },
    {
      "epoch": 2.6264190673828125e-06,
      "model_forward_time": 5.6203773021698,
      "step": 2754
    },
    {
      "epoch": 2.6264190673828125e-06,
      "step": 2754,
      "training_step_time": 17.116824865341187
    },
    {
      "epoch": 2.6273727416992186e-06,
      "model_forward_time": 5.615442752838135,
      "step": 2755
    },
    {
      "epoch": 2.6273727416992186e-06,
      "step": 2755,
      "training_step_time": 17.129119157791138
    },
    {
      "epoch": 2.628326416015625e-06,
      "model_forward_time": 5.9738450050354,
      "step": 2756
    },
    {
      "epoch": 2.628326416015625e-06,
      "step": 2756,
      "training_step_time": 17.459986209869385
    },
    {
      "epoch": 2.6292800903320314e-06,
      "model_forward_time": 5.601589679718018,
      "step": 2757
    },
    {
      "epoch": 2.6292800903320314e-06,
      "step": 2757,
      "training_step_time": 17.091104984283447
    },
    {
      "epoch": 2.6302337646484375e-06,
      "model_forward_time": 5.60720682144165,
      "step": 2758
    },
    {
      "epoch": 2.6302337646484375e-06,
      "step": 2758,
      "training_step_time": 17.079298496246338
    },
    {
      "epoch": 2.6311874389648437e-06,
      "model_forward_time": 5.6123716831207275,
      "step": 2759
    },
    {
      "epoch": 2.6311874389648437e-06,
      "step": 2759,
      "training_step_time": 17.099239826202393
    },
    {
      "epoch": 2.63214111328125e-06,
      "grad_norm": 1.0012308359146118,
      "learning_rate": 1e-05,
      "loss": 0.0775,
      "step": 2760
    },
    {
      "epoch": 2.63214111328125e-06,
      "model_forward_time": 5.601781845092773,
      "step": 2760
    },
    {
      "dynamics_loss_avg": 0.06359531022608281,
      "epoch": 2.63214111328125e-06,
      "step": 2760
    },
    {
      "action_loss_avg": 0.014747558697126805,
      "epoch": 2.63214111328125e-06,
      "step": 2760
    },
    {
      "epoch": 2.63214111328125e-06,
      "step": 2760,
      "training_step_time": 17.08717155456543
    },
    {
      "epoch": 2.6330947875976564e-06,
      "model_forward_time": 5.958489179611206,
      "step": 2761
    },
    {
      "epoch": 2.6330947875976564e-06,
      "step": 2761,
      "training_step_time": 17.456307411193848
    },
    {
      "epoch": 2.6340484619140625e-06,
      "model_forward_time": 5.611990690231323,
      "step": 2762
    },
    {
      "epoch": 2.6340484619140625e-06,
      "step": 2762,
      "training_step_time": 17.06201672554016
    },
    {
      "epoch": 2.6350021362304687e-06,
      "model_forward_time": 5.574888467788696,
      "step": 2763
    },
    {
      "epoch": 2.6350021362304687e-06,
      "step": 2763,
      "training_step_time": 16.991657495498657
    },
    {
      "epoch": 2.635955810546875e-06,
      "model_forward_time": 5.586021184921265,
      "step": 2764
    },
    {
      "epoch": 2.635955810546875e-06,
      "step": 2764,
      "training_step_time": 16.999003887176514
    },
    {
      "epoch": 2.6369094848632814e-06,
      "model_forward_time": 5.617583990097046,
      "step": 2765
    },
    {
      "epoch": 2.6369094848632814e-06,
      "step": 2765,
      "training_step_time": 17.05045437812805
    },
    {
      "epoch": 2.6378631591796875e-06,
      "model_forward_time": 5.579814434051514,
      "step": 2766
    },
    {
      "epoch": 2.6378631591796875e-06,
      "step": 2766,
      "training_step_time": 17.02048087120056
    },
    {
      "epoch": 2.6388168334960937e-06,
      "model_forward_time": 5.95876145362854,
      "step": 2767
    },
    {
      "epoch": 2.6388168334960937e-06,
      "step": 2767,
      "training_step_time": 17.403538465499878
    },
    {
      "epoch": 2.6397705078125e-06,
      "model_forward_time": 5.596538305282593,
      "step": 2768
    },
    {
      "epoch": 2.6397705078125e-06,
      "step": 2768,
      "training_step_time": 17.05469560623169
    },
    {
      "epoch": 2.6407241821289064e-06,
      "model_forward_time": 5.611296892166138,
      "step": 2769
    },
    {
      "epoch": 2.6407241821289064e-06,
      "step": 2769,
      "training_step_time": 17.099451541900635
    },
    {
      "epoch": 2.6416778564453125e-06,
      "grad_norm": 1.368938684463501,
      "learning_rate": 1e-05,
      "loss": 0.0615,
      "step": 2770
    },
    {
      "epoch": 2.6416778564453125e-06,
      "model_forward_time": 5.603705644607544,
      "step": 2770
    },
    {
      "dynamics_loss_avg": 0.051526978984475134,
      "epoch": 2.6416778564453125e-06,
      "step": 2770
    },
    {
      "action_loss_avg": 0.010745452577248215,
      "epoch": 2.6416778564453125e-06,
      "step": 2770
    },
    {
      "epoch": 2.6416778564453125e-06,
      "step": 2770,
      "training_step_time": 17.072444915771484
    },
    {
      "epoch": 2.6426315307617187e-06,
      "model_forward_time": 5.595398902893066,
      "step": 2771
    },
    {
      "epoch": 2.6426315307617187e-06,
      "step": 2771,
      "training_step_time": 17.00664782524109
    },
    {
      "epoch": 2.643585205078125e-06,
      "model_forward_time": 5.944301605224609,
      "step": 2772
    },
    {
      "epoch": 2.643585205078125e-06,
      "step": 2772,
      "training_step_time": 17.363900899887085
    },
    {
      "epoch": 2.6445388793945314e-06,
      "model_forward_time": 5.5884459018707275,
      "step": 2773
    },
    {
      "epoch": 2.6445388793945314e-06,
      "step": 2773,
      "training_step_time": 17.01667833328247
    },
    {
      "epoch": 2.6454925537109375e-06,
      "model_forward_time": 5.590388536453247,
      "step": 2774
    },
    {
      "epoch": 2.6454925537109375e-06,
      "step": 2774,
      "training_step_time": 17.04299783706665
    },
    {
      "epoch": 2.6464462280273437e-06,
      "model_forward_time": 5.593712091445923,
      "step": 2775
    },
    {
      "epoch": 2.6464462280273437e-06,
      "step": 2775,
      "training_step_time": 17.053171396255493
    },
    {
      "epoch": 2.64739990234375e-06,
      "model_forward_time": 5.6022419929504395,
      "step": 2776
    },
    {
      "epoch": 2.64739990234375e-06,
      "step": 2776,
      "training_step_time": 17.06450653076172
    },
    {
      "epoch": 2.6483535766601564e-06,
      "model_forward_time": 5.593858003616333,
      "step": 2777
    },
    {
      "epoch": 2.6483535766601564e-06,
      "step": 2777,
      "training_step_time": 17.415923357009888
    },
    {
      "epoch": 2.6493072509765625e-06,
      "model_forward_time": 5.583951473236084,
      "step": 2778
    },
    {
      "epoch": 2.6493072509765625e-06,
      "step": 2778,
      "training_step_time": 17.035918712615967
    },
    {
      "epoch": 2.6502609252929687e-06,
      "model_forward_time": 5.597992658615112,
      "step": 2779
    },
    {
      "epoch": 2.6502609252929687e-06,
      "step": 2779,
      "training_step_time": 17.06173801422119
    },
    {
      "epoch": 2.651214599609375e-06,
      "grad_norm": 0.26685094833374023,
      "learning_rate": 1e-05,
      "loss": 0.0674,
      "step": 2780
    },
    {
      "epoch": 2.651214599609375e-06,
      "model_forward_time": 5.601114749908447,
      "step": 2780
    },
    {
      "dynamics_loss_avg": 0.05383771434426308,
      "epoch": 2.651214599609375e-06,
      "step": 2780
    },
    {
      "action_loss_avg": 0.012479314836673439,
      "epoch": 2.651214599609375e-06,
      "step": 2780
    },
    {
      "epoch": 2.651214599609375e-06,
      "step": 2780,
      "training_step_time": 17.096040725708008
    },
    {
      "epoch": 2.6521682739257814e-06,
      "model_forward_time": 5.619800567626953,
      "step": 2781
    },
    {
      "epoch": 2.6521682739257814e-06,
      "step": 2781,
      "training_step_time": 17.099554300308228
    },
    {
      "epoch": 2.6531219482421875e-06,
      "model_forward_time": 5.5915207862854,
      "step": 2782
    },
    {
      "epoch": 2.6531219482421875e-06,
      "step": 2782,
      "training_step_time": 17.05621337890625
    },
    {
      "epoch": 2.6540756225585937e-06,
      "model_forward_time": 5.942970275878906,
      "step": 2783
    },
    {
      "epoch": 2.6540756225585937e-06,
      "step": 2783,
      "training_step_time": 17.37469792366028
    },
    {
      "epoch": 2.655029296875e-06,
      "model_forward_time": 5.58636999130249,
      "step": 2784
    },
    {
      "epoch": 2.655029296875e-06,
      "step": 2784,
      "training_step_time": 17.042377471923828
    },
    {
      "epoch": 2.6559829711914064e-06,
      "model_forward_time": 5.609166145324707,
      "step": 2785
    },
    {
      "epoch": 2.6559829711914064e-06,
      "step": 2785,
      "training_step_time": 17.079252243041992
    },
    {
      "epoch": 2.6569366455078125e-06,
      "model_forward_time": 5.606585741043091,
      "step": 2786
    },
    {
      "epoch": 2.6569366455078125e-06,
      "step": 2786,
      "training_step_time": 17.102452754974365
    },
    {
      "epoch": 2.6578903198242187e-06,
      "model_forward_time": 5.607686281204224,
      "step": 2787
    },
    {
      "epoch": 2.6578903198242187e-06,
      "step": 2787,
      "training_step_time": 17.07459807395935
    },
    {
      "epoch": 2.658843994140625e-06,
      "model_forward_time": 5.596670150756836,
      "step": 2788
    },
    {
      "epoch": 2.658843994140625e-06,
      "step": 2788,
      "training_step_time": 17.397212028503418
    },
    {
      "epoch": 2.6597976684570314e-06,
      "model_forward_time": 5.583184719085693,
      "step": 2789
    },
    {
      "epoch": 2.6597976684570314e-06,
      "step": 2789,
      "training_step_time": 17.03334879875183
    },
    {
      "epoch": 2.6607513427734375e-06,
      "grad_norm": 0.6593564748764038,
      "learning_rate": 1e-05,
      "loss": 0.0693,
      "step": 2790
    },
    {
      "epoch": 2.6607513427734375e-06,
      "model_forward_time": 5.592194318771362,
      "step": 2790
    },
    {
      "dynamics_loss_avg": 0.052471373416483405,
      "epoch": 2.6607513427734375e-06,
      "step": 2790
    },
    {
      "action_loss_avg": 0.01381098162382841,
      "epoch": 2.6607513427734375e-06,
      "step": 2790
    },
    {
      "epoch": 2.6607513427734375e-06,
      "step": 2790,
      "training_step_time": 17.070703983306885
    },
    {
      "epoch": 2.6617050170898437e-06,
      "model_forward_time": 5.604002237319946,
      "step": 2791
    },
    {
      "epoch": 2.6617050170898437e-06,
      "step": 2791,
      "training_step_time": 17.0930335521698
    },
    {
      "epoch": 2.66265869140625e-06,
      "model_forward_time": 5.6023149490356445,
      "step": 2792
    },
    {
      "epoch": 2.66265869140625e-06,
      "step": 2792,
      "training_step_time": 17.076547861099243
    },
    {
      "epoch": 2.6636123657226564e-06,
      "model_forward_time": 5.615161657333374,
      "step": 2793
    },
    {
      "epoch": 2.6636123657226564e-06,
      "step": 2793,
      "training_step_time": 17.099870681762695
    },
    {
      "epoch": 2.6645660400390625e-06,
      "model_forward_time": 5.959989547729492,
      "step": 2794
    },
    {
      "epoch": 2.6645660400390625e-06,
      "step": 2794,
      "training_step_time": 17.456834077835083
    },
    {
      "epoch": 2.6655197143554687e-06,
      "model_forward_time": 5.6155126094818115,
      "step": 2795
    },
    {
      "epoch": 2.6655197143554687e-06,
      "step": 2795,
      "training_step_time": 17.085132122039795
    },
    {
      "epoch": 2.666473388671875e-06,
      "model_forward_time": 5.583632469177246,
      "step": 2796
    },
    {
      "epoch": 2.666473388671875e-06,
      "step": 2796,
      "training_step_time": 17.013142824172974
    },
    {
      "epoch": 2.6674270629882814e-06,
      "model_forward_time": 5.571739196777344,
      "step": 2797
    },
    {
      "epoch": 2.6674270629882814e-06,
      "step": 2797,
      "training_step_time": 16.992908477783203
    },
    {
      "epoch": 2.6683807373046875e-06,
      "model_forward_time": 5.563489675521851,
      "step": 2798
    },
    {
      "epoch": 2.6683807373046875e-06,
      "step": 2798,
      "training_step_time": 16.983354568481445
    },
    {
      "epoch": 2.6693344116210937e-06,
      "model_forward_time": 5.933314323425293,
      "step": 2799
    },
    {
      "epoch": 2.6693344116210937e-06,
      "step": 2799,
      "training_step_time": 17.351341485977173
    },
    {
      "epoch": 2.6702880859375e-06,
      "grad_norm": 1.6326584815979004,
      "learning_rate": 1e-05,
      "loss": 0.0886,
      "step": 2800
    },
    {
      "epoch": 2.6702880859375e-06,
      "model_forward_time": 5.5725085735321045,
      "step": 2800
    },
    {
      "dynamics_loss_avg": 0.07426226492971182,
      "epoch": 2.6702880859375e-06,
      "step": 2800
    },
    {
      "action_loss_avg": 0.014219646877609193,
      "epoch": 2.6702880859375e-06,
      "step": 2800
    },
    {
      "epoch": 2.6702880859375e-06,
      "step": 2800,
      "training_step_time": 17.007352590560913
    },
    {
      "epoch": 2.6712417602539064e-06,
      "model_forward_time": 5.577643156051636,
      "step": 2801
    },
    {
      "epoch": 2.6712417602539064e-06,
      "step": 2801,
      "training_step_time": 16.980780601501465
    },
    {
      "epoch": 2.6721954345703125e-06,
      "model_forward_time": 5.58114218711853,
      "step": 2802
    },
    {
      "epoch": 2.6721954345703125e-06,
      "step": 2802,
      "training_step_time": 17.01461696624756
    },
    {
      "epoch": 2.6731491088867187e-06,
      "model_forward_time": 5.5959343910217285,
      "step": 2803
    },
    {
      "epoch": 2.6731491088867187e-06,
      "step": 2803,
      "training_step_time": 17.01802110671997
    },
    {
      "epoch": 2.674102783203125e-06,
      "model_forward_time": 5.596897840499878,
      "step": 2804
    },
    {
      "epoch": 2.674102783203125e-06,
      "step": 2804,
      "training_step_time": 17.401673316955566
    },
    {
      "epoch": 2.6750564575195314e-06,
      "model_forward_time": 5.594125509262085,
      "step": 2805
    },
    {
      "epoch": 2.6750564575195314e-06,
      "step": 2805,
      "training_step_time": 17.059547185897827
    },
    {
      "epoch": 2.6760101318359375e-06,
      "model_forward_time": 5.612130403518677,
      "step": 2806
    },
    {
      "epoch": 2.6760101318359375e-06,
      "step": 2806,
      "training_step_time": 17.07552170753479
    },
    {
      "epoch": 2.6769638061523437e-06,
      "model_forward_time": 5.599437952041626,
      "step": 2807
    },
    {
      "epoch": 2.6769638061523437e-06,
      "step": 2807,
      "training_step_time": 17.051660776138306
    },
    {
      "epoch": 2.67791748046875e-06,
      "model_forward_time": 5.600875377655029,
      "step": 2808
    },
    {
      "epoch": 2.67791748046875e-06,
      "step": 2808,
      "training_step_time": 17.049269676208496
    },
    {
      "epoch": 2.6788711547851564e-06,
      "model_forward_time": 5.595411539077759,
      "step": 2809
    },
    {
      "epoch": 2.6788711547851564e-06,
      "step": 2809,
      "training_step_time": 17.028207540512085
    },
    {
      "epoch": 2.6798248291015626e-06,
      "grad_norm": 1.4877713918685913,
      "learning_rate": 1e-05,
      "loss": 0.0795,
      "step": 2810
    },
    {
      "epoch": 2.6798248291015626e-06,
      "model_forward_time": 5.950738191604614,
      "step": 2810
    },
    {
      "dynamics_loss_avg": 0.06362244375050068,
      "epoch": 2.6798248291015626e-06,
      "step": 2810
    },
    {
      "action_loss_avg": 0.015570289501920343,
      "epoch": 2.6798248291015626e-06,
      "step": 2810
    },
    {
      "epoch": 2.6798248291015626e-06,
      "step": 2810,
      "training_step_time": 17.405476093292236
    },
    {
      "epoch": 2.6807785034179687e-06,
      "model_forward_time": 5.617332935333252,
      "step": 2811
    },
    {
      "epoch": 2.6807785034179687e-06,
      "step": 2811,
      "training_step_time": 17.113901138305664
    },
    {
      "epoch": 2.681732177734375e-06,
      "model_forward_time": 5.619619607925415,
      "step": 2812
    },
    {
      "epoch": 2.681732177734375e-06,
      "step": 2812,
      "training_step_time": 17.12981605529785
    },
    {
      "epoch": 2.6826858520507814e-06,
      "model_forward_time": 5.614192247390747,
      "step": 2813
    },
    {
      "epoch": 2.6826858520507814e-06,
      "step": 2813,
      "training_step_time": 17.107327938079834
    },
    {
      "epoch": 2.6836395263671876e-06,
      "model_forward_time": 5.598394155502319,
      "step": 2814
    },
    {
      "epoch": 2.6836395263671876e-06,
      "step": 2814,
      "training_step_time": 17.089224338531494
    },
    {
      "epoch": 2.6845932006835937e-06,
      "model_forward_time": 5.949897050857544,
      "step": 2815
    },
    {
      "epoch": 2.6845932006835937e-06,
      "step": 2815,
      "training_step_time": 17.423386573791504
    },
    {
      "epoch": 2.685546875e-06,
      "model_forward_time": 5.619957208633423,
      "step": 2816
    },
    {
      "epoch": 2.685546875e-06,
      "step": 2816,
      "training_step_time": 17.080971479415894
    },
    {
      "epoch": 2.6865005493164064e-06,
      "model_forward_time": 5.590227127075195,
      "step": 2817
    },
    {
      "epoch": 2.6865005493164064e-06,
      "step": 2817,
      "training_step_time": 17.024430990219116
    },
    {
      "epoch": 2.6874542236328126e-06,
      "model_forward_time": 5.589319229125977,
      "step": 2818
    },
    {
      "epoch": 2.6874542236328126e-06,
      "step": 2818,
      "training_step_time": 17.013655424118042
    },
    {
      "epoch": 2.6884078979492187e-06,
      "model_forward_time": 5.58693265914917,
      "step": 2819
    },
    {
      "epoch": 2.6884078979492187e-06,
      "step": 2819,
      "training_step_time": 17.02045464515686
    },
    {
      "epoch": 2.689361572265625e-06,
      "grad_norm": 1.119150161743164,
      "learning_rate": 1e-05,
      "loss": 0.0705,
      "step": 2820
    },
    {
      "epoch": 2.689361572265625e-06,
      "model_forward_time": 5.576252460479736,
      "step": 2820
    },
    {
      "dynamics_loss_avg": 0.056561294943094254,
      "epoch": 2.689361572265625e-06,
      "step": 2820
    },
    {
      "action_loss_avg": 0.016819389211013912,
      "epoch": 2.689361572265625e-06,
      "step": 2820
    },
    {
      "epoch": 2.689361572265625e-06,
      "step": 2820,
      "training_step_time": 17.02738332748413
    },
    {
      "epoch": 2.6903152465820314e-06,
      "model_forward_time": 5.964659929275513,
      "step": 2821
    },
    {
      "epoch": 2.6903152465820314e-06,
      "step": 2821,
      "training_step_time": 17.402384519577026
    },
    {
      "epoch": 2.6912689208984376e-06,
      "model_forward_time": 5.590792417526245,
      "step": 2822
    },
    {
      "epoch": 2.6912689208984376e-06,
      "step": 2822,
      "training_step_time": 17.04933786392212
    },
    {
      "epoch": 2.6922225952148437e-06,
      "model_forward_time": 5.608551502227783,
      "step": 2823
    },
    {
      "epoch": 2.6922225952148437e-06,
      "step": 2823,
      "training_step_time": 17.11527442932129
    },
    {
      "epoch": 2.69317626953125e-06,
      "model_forward_time": 5.624216318130493,
      "step": 2824
    },
    {
      "epoch": 2.69317626953125e-06,
      "step": 2824,
      "training_step_time": 17.120229959487915
    },
    {
      "epoch": 2.6941299438476564e-06,
      "model_forward_time": 5.600085973739624,
      "step": 2825
    },
    {
      "epoch": 2.6941299438476564e-06,
      "step": 2825,
      "training_step_time": 17.063992261886597
    },
    {
      "epoch": 2.6950836181640626e-06,
      "model_forward_time": 5.955352306365967,
      "step": 2826
    },
    {
      "epoch": 2.6950836181640626e-06,
      "step": 2826,
      "training_step_time": 17.404983282089233
    },
    {
      "epoch": 2.6960372924804687e-06,
      "model_forward_time": 5.58249044418335,
      "step": 2827
    },
    {
      "epoch": 2.6960372924804687e-06,
      "step": 2827,
      "training_step_time": 17.02375841140747
    },
    {
      "epoch": 2.696990966796875e-06,
      "model_forward_time": 5.59207820892334,
      "step": 2828
    },
    {
      "epoch": 2.696990966796875e-06,
      "step": 2828,
      "training_step_time": 17.038609743118286
    },
    {
      "epoch": 2.6979446411132814e-06,
      "model_forward_time": 5.601090669631958,
      "step": 2829
    },
    {
      "epoch": 2.6979446411132814e-06,
      "step": 2829,
      "training_step_time": 17.073456048965454
    },
    {
      "epoch": 2.6988983154296876e-06,
      "grad_norm": 1.2424015998840332,
      "learning_rate": 1e-05,
      "loss": 0.076,
      "step": 2830
    },
    {
      "epoch": 2.6988983154296876e-06,
      "model_forward_time": 5.622931480407715,
      "step": 2830
    },
    {
      "dynamics_loss_avg": 0.055722229182720184,
      "epoch": 2.6988983154296876e-06,
      "step": 2830
    },
    {
      "action_loss_avg": 0.0151432940736413,
      "epoch": 2.6988983154296876e-06,
      "step": 2830
    },
    {
      "epoch": 2.6988983154296876e-06,
      "step": 2830,
      "training_step_time": 17.139238595962524
    },
    {
      "epoch": 2.6998519897460937e-06,
      "model_forward_time": 5.624377965927124,
      "step": 2831
    },
    {
      "epoch": 2.6998519897460937e-06,
      "step": 2831,
      "training_step_time": 17.123217821121216
    },
    {
      "epoch": 2.7008056640625e-06,
      "model_forward_time": 5.9717302322387695,
      "step": 2832
    },
    {
      "epoch": 2.7008056640625e-06,
      "step": 2832,
      "training_step_time": 17.441317081451416
    },
    {
      "epoch": 2.7017593383789064e-06,
      "model_forward_time": 5.607638359069824,
      "step": 2833
    },
    {
      "epoch": 2.7017593383789064e-06,
      "step": 2833,
      "training_step_time": 17.080315351486206
    },
    {
      "epoch": 2.7027130126953126e-06,
      "model_forward_time": 5.596014738082886,
      "step": 2834
    },
    {
      "epoch": 2.7027130126953126e-06,
      "step": 2834,
      "training_step_time": 17.058791637420654
    },
    {
      "epoch": 2.7036666870117187e-06,
      "model_forward_time": 5.615963459014893,
      "step": 2835
    },
    {
      "epoch": 2.7036666870117187e-06,
      "step": 2835,
      "training_step_time": 17.053128957748413
    },
    {
      "epoch": 2.704620361328125e-06,
      "model_forward_time": 5.590964078903198,
      "step": 2836
    },
    {
      "epoch": 2.704620361328125e-06,
      "step": 2836,
      "training_step_time": 17.035682439804077
    },
    {
      "epoch": 2.7055740356445314e-06,
      "model_forward_time": 5.946499586105347,
      "step": 2837
    },
    {
      "epoch": 2.7055740356445314e-06,
      "step": 2837,
      "training_step_time": 17.412397861480713
    },
    {
      "epoch": 2.7065277099609376e-06,
      "model_forward_time": 5.608757257461548,
      "step": 2838
    },
    {
      "epoch": 2.7065277099609376e-06,
      "step": 2838,
      "training_step_time": 17.094918251037598
    },
    {
      "epoch": 2.7074813842773437e-06,
      "model_forward_time": 5.611098289489746,
      "step": 2839
    },
    {
      "epoch": 2.7074813842773437e-06,
      "step": 2839,
      "training_step_time": 17.094318389892578
    },
    {
      "epoch": 2.70843505859375e-06,
      "grad_norm": 0.4611390233039856,
      "learning_rate": 1e-05,
      "loss": 0.0664,
      "step": 2840
    },
    {
      "epoch": 2.70843505859375e-06,
      "model_forward_time": 5.6151816844940186,
      "step": 2840
    },
    {
      "dynamics_loss_avg": 0.05763272605836391,
      "epoch": 2.70843505859375e-06,
      "step": 2840
    },
    {
      "action_loss_avg": 0.013879145542159677,
      "epoch": 2.70843505859375e-06,
      "step": 2840
    },
    {
      "epoch": 2.70843505859375e-06,
      "step": 2840,
      "training_step_time": 17.10954761505127
    },
    {
      "epoch": 2.7093887329101564e-06,
      "model_forward_time": 5.612638235092163,
      "step": 2841
    },
    {
      "epoch": 2.7093887329101564e-06,
      "step": 2841,
      "training_step_time": 17.101569414138794
    },
    {
      "epoch": 2.7103424072265626e-06,
      "model_forward_time": 5.600304365158081,
      "step": 2842
    },
    {
      "epoch": 2.7103424072265626e-06,
      "step": 2842,
      "training_step_time": 17.40652370452881
    },
    {
      "epoch": 2.7112960815429687e-06,
      "model_forward_time": 5.591541051864624,
      "step": 2843
    },
    {
      "epoch": 2.7112960815429687e-06,
      "step": 2843,
      "training_step_time": 17.045494318008423
    },
    {
      "epoch": 2.712249755859375e-06,
      "model_forward_time": 5.600773811340332,
      "step": 2844
    },
    {
      "epoch": 2.712249755859375e-06,
      "step": 2844,
      "training_step_time": 17.07306146621704
    },
    {
      "epoch": 2.7132034301757814e-06,
      "model_forward_time": 5.598641872406006,
      "step": 2845
    },
    {
      "epoch": 2.7132034301757814e-06,
      "step": 2845,
      "training_step_time": 17.084636926651
    },
    {
      "epoch": 2.7141571044921876e-06,
      "model_forward_time": 5.605654239654541,
      "step": 2846
    },
    {
      "epoch": 2.7141571044921876e-06,
      "step": 2846,
      "training_step_time": 17.077693939208984
    },
    {
      "epoch": 2.7151107788085937e-06,
      "model_forward_time": 5.597284555435181,
      "step": 2847
    },
    {
      "epoch": 2.7151107788085937e-06,
      "step": 2847,
      "training_step_time": 17.062475204467773
    },
    {
      "epoch": 2.716064453125e-06,
      "model_forward_time": 5.953387498855591,
      "step": 2848
    },
    {
      "epoch": 2.716064453125e-06,
      "step": 2848,
      "training_step_time": 17.42201852798462
    },
    {
      "epoch": 2.7170181274414064e-06,
      "model_forward_time": 5.598315238952637,
      "step": 2849
    },
    {
      "epoch": 2.7170181274414064e-06,
      "step": 2849,
      "training_step_time": 17.017067670822144
    },
    {
      "epoch": 2.7179718017578126e-06,
      "grad_norm": 0.4110797345638275,
      "learning_rate": 1e-05,
      "loss": 0.084,
      "step": 2850
    },
    {
      "epoch": 2.7179718017578126e-06,
      "model_forward_time": 5.605963945388794,
      "step": 2850
    },
    {
      "dynamics_loss_avg": 0.06839141510426998,
      "epoch": 2.7179718017578126e-06,
      "step": 2850
    },
    {
      "action_loss_avg": 0.017609290033578872,
      "epoch": 2.7179718017578126e-06,
      "step": 2850
    },
    {
      "epoch": 2.7179718017578126e-06,
      "step": 2850,
      "training_step_time": 17.083019971847534
    },
    {
      "epoch": 2.7189254760742187e-06,
      "model_forward_time": 5.601819276809692,
      "step": 2851
    },
    {
      "epoch": 2.7189254760742187e-06,
      "step": 2851,
      "training_step_time": 17.071568489074707
    },
    {
      "epoch": 2.719879150390625e-06,
      "model_forward_time": 5.5964035987854,
      "step": 2852
    },
    {
      "epoch": 2.719879150390625e-06,
      "step": 2852,
      "training_step_time": 17.02948498725891
    },
    {
      "epoch": 2.7208328247070314e-06,
      "model_forward_time": 5.958034992218018,
      "step": 2853
    },
    {
      "epoch": 2.7208328247070314e-06,
      "step": 2853,
      "training_step_time": 17.400163412094116
    },
    {
      "epoch": 2.7217864990234376e-06,
      "model_forward_time": 5.599717855453491,
      "step": 2854
    },
    {
      "epoch": 2.7217864990234376e-06,
      "step": 2854,
      "training_step_time": 17.06696081161499
    },
    {
      "epoch": 2.7227401733398437e-06,
      "model_forward_time": 5.603968381881714,
      "step": 2855
    },
    {
      "epoch": 2.7227401733398437e-06,
      "step": 2855,
      "training_step_time": 17.08457899093628
    },
    {
      "epoch": 2.72369384765625e-06,
      "model_forward_time": 5.614404916763306,
      "step": 2856
    },
    {
      "epoch": 2.72369384765625e-06,
      "step": 2856,
      "training_step_time": 17.12320613861084
    },
    {
      "epoch": 2.7246475219726564e-06,
      "model_forward_time": 5.6192615032196045,
      "step": 2857
    },
    {
      "epoch": 2.7246475219726564e-06,
      "step": 2857,
      "training_step_time": 17.104306936264038
    },
    {
      "epoch": 2.7256011962890626e-06,
      "model_forward_time": 5.603411436080933,
      "step": 2858
    },
    {
      "epoch": 2.7256011962890626e-06,
      "step": 2858,
      "training_step_time": 17.06793999671936
    },
    {
      "epoch": 2.7265548706054687e-06,
      "model_forward_time": 5.938819408416748,
      "step": 2859
    },
    {
      "epoch": 2.7265548706054687e-06,
      "step": 2859,
      "training_step_time": 17.37886071205139
    },
    {
      "epoch": 2.727508544921875e-06,
      "grad_norm": 0.6750516295433044,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 2860
    },
    {
      "epoch": 2.727508544921875e-06,
      "model_forward_time": 5.5949742794036865,
      "step": 2860
    },
    {
      "dynamics_loss_avg": 0.05968083366751671,
      "epoch": 2.727508544921875e-06,
      "step": 2860
    },
    {
      "action_loss_avg": 0.01734394021332264,
      "epoch": 2.727508544921875e-06,
      "step": 2860
    },
    {
      "epoch": 2.727508544921875e-06,
      "step": 2860,
      "training_step_time": 17.045368671417236
    },
    {
      "epoch": 2.7284622192382814e-06,
      "model_forward_time": 5.605754613876343,
      "step": 2861
    },
    {
      "epoch": 2.7284622192382814e-06,
      "step": 2861,
      "training_step_time": 17.092043161392212
    },
    {
      "epoch": 2.7294158935546876e-06,
      "model_forward_time": 5.619410037994385,
      "step": 2862
    },
    {
      "epoch": 2.7294158935546876e-06,
      "step": 2862,
      "training_step_time": 17.103453636169434
    },
    {
      "epoch": 2.7303695678710937e-06,
      "model_forward_time": 5.600383520126343,
      "step": 2863
    },
    {
      "epoch": 2.7303695678710937e-06,
      "step": 2863,
      "training_step_time": 17.056127071380615
    },
    {
      "epoch": 2.7313232421875e-06,
      "model_forward_time": 5.962235927581787,
      "step": 2864
    },
    {
      "epoch": 2.7313232421875e-06,
      "step": 2864,
      "training_step_time": 17.394087553024292
    },
    {
      "epoch": 2.7322769165039065e-06,
      "model_forward_time": 5.587021112442017,
      "step": 2865
    },
    {
      "epoch": 2.7322769165039065e-06,
      "step": 2865,
      "training_step_time": 17.029629230499268
    },
    {
      "epoch": 2.7332305908203126e-06,
      "model_forward_time": 5.592346668243408,
      "step": 2866
    },
    {
      "epoch": 2.7332305908203126e-06,
      "step": 2866,
      "training_step_time": 17.040164709091187
    },
    {
      "epoch": 2.7341842651367187e-06,
      "model_forward_time": 5.5993006229400635,
      "step": 2867
    },
    {
      "epoch": 2.7341842651367187e-06,
      "step": 2867,
      "training_step_time": 17.07624840736389
    },
    {
      "epoch": 2.735137939453125e-06,
      "model_forward_time": 5.6089277267456055,
      "step": 2868
    },
    {
      "epoch": 2.735137939453125e-06,
      "step": 2868,
      "training_step_time": 17.11140251159668
    },
    {
      "epoch": 2.7360916137695315e-06,
      "model_forward_time": 5.620970726013184,
      "step": 2869
    },
    {
      "epoch": 2.7360916137695315e-06,
      "step": 2869,
      "training_step_time": 17.12086844444275
    },
    {
      "epoch": 2.7370452880859376e-06,
      "grad_norm": 0.6805873513221741,
      "learning_rate": 1e-05,
      "loss": 0.0996,
      "step": 2870
    },
    {
      "epoch": 2.7370452880859376e-06,
      "model_forward_time": 5.96150016784668,
      "step": 2870
    },
    {
      "dynamics_loss_avg": 0.07735882364213467,
      "epoch": 2.7370452880859376e-06,
      "step": 2870
    },
    {
      "action_loss_avg": 0.021124101150780915,
      "epoch": 2.7370452880859376e-06,
      "step": 2870
    },
    {
      "epoch": 2.7370452880859376e-06,
      "step": 2870,
      "training_step_time": 17.46676802635193
    },
    {
      "epoch": 2.7379989624023437e-06,
      "model_forward_time": 5.62985634803772,
      "step": 2871
    },
    {
      "epoch": 2.7379989624023437e-06,
      "step": 2871,
      "training_step_time": 17.12764549255371
    },
    {
      "epoch": 2.73895263671875e-06,
      "model_forward_time": 5.61267352104187,
      "step": 2872
    },
    {
      "epoch": 2.73895263671875e-06,
      "step": 2872,
      "training_step_time": 17.11326766014099
    },
    {
      "epoch": 2.7399063110351565e-06,
      "model_forward_time": 5.617063760757446,
      "step": 2873
    },
    {
      "epoch": 2.7399063110351565e-06,
      "step": 2873,
      "training_step_time": 17.115290641784668
    },
    {
      "epoch": 2.7408599853515626e-06,
      "model_forward_time": 5.613993883132935,
      "step": 2874
    },
    {
      "epoch": 2.7408599853515626e-06,
      "step": 2874,
      "training_step_time": 17.09708547592163
    },
    {
      "epoch": 2.7418136596679688e-06,
      "model_forward_time": 5.977107524871826,
      "step": 2875
    },
    {
      "epoch": 2.7418136596679688e-06,
      "step": 2875,
      "training_step_time": 17.449124813079834
    },
    {
      "epoch": 2.742767333984375e-06,
      "model_forward_time": 5.599130153656006,
      "step": 2876
    },
    {
      "epoch": 2.742767333984375e-06,
      "step": 2876,
      "training_step_time": 17.07562017440796
    },
    {
      "epoch": 2.743721008300781e-06,
      "model_forward_time": 5.59993052482605,
      "step": 2877
    },
    {
      "epoch": 2.743721008300781e-06,
      "step": 2877,
      "training_step_time": 17.065840244293213
    },
    {
      "epoch": 2.7446746826171876e-06,
      "model_forward_time": 5.5990025997161865,
      "step": 2878
    },
    {
      "epoch": 2.7446746826171876e-06,
      "step": 2878,
      "training_step_time": 17.066110849380493
    },
    {
      "epoch": 2.7456283569335938e-06,
      "model_forward_time": 5.603466033935547,
      "step": 2879
    },
    {
      "epoch": 2.7456283569335938e-06,
      "step": 2879,
      "training_step_time": 17.074724912643433
    },
    {
      "epoch": 2.74658203125e-06,
      "grad_norm": 1.83604097366333,
      "learning_rate": 1e-05,
      "loss": 0.0929,
      "step": 2880
    },
    {
      "epoch": 2.74658203125e-06,
      "model_forward_time": 5.606765270233154,
      "step": 2880
    },
    {
      "dynamics_loss_avg": 0.06805291324853897,
      "epoch": 2.74658203125e-06,
      "step": 2880
    },
    {
      "action_loss_avg": 0.021966355852782726,
      "epoch": 2.74658203125e-06,
      "step": 2880
    },
    {
      "epoch": 2.74658203125e-06,
      "step": 2880,
      "training_step_time": 17.08342719078064
    },
    {
      "epoch": 2.747535705566406e-06,
      "model_forward_time": 5.957509756088257,
      "step": 2881
    },
    {
      "epoch": 2.747535705566406e-06,
      "step": 2881,
      "training_step_time": 17.440682411193848
    },
    {
      "epoch": 2.7484893798828126e-06,
      "model_forward_time": 5.608949422836304,
      "step": 2882
    },
    {
      "epoch": 2.7484893798828126e-06,
      "step": 2882,
      "training_step_time": 17.09682536125183
    },
    {
      "epoch": 2.7494430541992188e-06,
      "model_forward_time": 5.610594272613525,
      "step": 2883
    },
    {
      "epoch": 2.7494430541992188e-06,
      "step": 2883,
      "training_step_time": 17.108388423919678
    },
    {
      "epoch": 2.750396728515625e-06,
      "model_forward_time": 5.611079454421997,
      "step": 2884
    },
    {
      "epoch": 2.750396728515625e-06,
      "step": 2884,
      "training_step_time": 17.09699773788452
    },
    {
      "epoch": 2.751350402832031e-06,
      "model_forward_time": 5.614703178405762,
      "step": 2885
    },
    {
      "epoch": 2.751350402832031e-06,
      "step": 2885,
      "training_step_time": 17.107942581176758
    },
    {
      "epoch": 2.7523040771484376e-06,
      "model_forward_time": 5.971526145935059,
      "step": 2886
    },
    {
      "epoch": 2.7523040771484376e-06,
      "step": 2886,
      "training_step_time": 17.42283868789673
    },
    {
      "epoch": 2.7532577514648438e-06,
      "model_forward_time": 5.590911388397217,
      "step": 2887
    },
    {
      "epoch": 2.7532577514648438e-06,
      "step": 2887,
      "training_step_time": 17.048978090286255
    },
    {
      "epoch": 2.75421142578125e-06,
      "model_forward_time": 5.611023664474487,
      "step": 2888
    },
    {
      "epoch": 2.75421142578125e-06,
      "step": 2888,
      "training_step_time": 17.084006547927856
    },
    {
      "epoch": 2.755165100097656e-06,
      "model_forward_time": 5.618941068649292,
      "step": 2889
    },
    {
      "epoch": 2.755165100097656e-06,
      "step": 2889,
      "training_step_time": 17.08009672164917
    },
    {
      "epoch": 2.7561187744140626e-06,
      "grad_norm": 0.20983576774597168,
      "learning_rate": 1e-05,
      "loss": 0.0615,
      "step": 2890
    },
    {
      "epoch": 2.7561187744140626e-06,
      "model_forward_time": 5.593996524810791,
      "step": 2890
    },
    {
      "dynamics_loss_avg": 0.0516206756234169,
      "epoch": 2.7561187744140626e-06,
      "step": 2890
    },
    {
      "action_loss_avg": 0.011902155890129506,
      "epoch": 2.7561187744140626e-06,
      "step": 2890
    },
    {
      "epoch": 2.7561187744140626e-06,
      "step": 2890,
      "training_step_time": 17.04731059074402
    },
    {
      "epoch": 2.7570724487304688e-06,
      "model_forward_time": 5.948766231536865,
      "step": 2891
    },
    {
      "epoch": 2.7570724487304688e-06,
      "step": 2891,
      "training_step_time": 17.39464044570923
    },
    {
      "epoch": 2.758026123046875e-06,
      "model_forward_time": 5.587661981582642,
      "step": 2892
    },
    {
      "epoch": 2.758026123046875e-06,
      "step": 2892,
      "training_step_time": 17.038374423980713
    },
    {
      "epoch": 2.758979797363281e-06,
      "model_forward_time": 5.5939857959747314,
      "step": 2893
    },
    {
      "epoch": 2.758979797363281e-06,
      "step": 2893,
      "training_step_time": 17.056385278701782
    },
    {
      "epoch": 2.7599334716796876e-06,
      "model_forward_time": 5.596015930175781,
      "step": 2894
    },
    {
      "epoch": 2.7599334716796876e-06,
      "step": 2894,
      "training_step_time": 17.064591646194458
    },
    {
      "epoch": 2.7608871459960938e-06,
      "model_forward_time": 5.608414649963379,
      "step": 2895
    },
    {
      "epoch": 2.7608871459960938e-06,
      "step": 2895,
      "training_step_time": 17.084240198135376
    },
    {
      "epoch": 2.7618408203125e-06,
      "model_forward_time": 5.611022710800171,
      "step": 2896
    },
    {
      "epoch": 2.7618408203125e-06,
      "step": 2896,
      "training_step_time": 17.09772276878357
    },
    {
      "epoch": 2.762794494628906e-06,
      "model_forward_time": 5.973954916000366,
      "step": 2897
    },
    {
      "epoch": 2.762794494628906e-06,
      "step": 2897,
      "training_step_time": 17.46190333366394
    },
    {
      "epoch": 2.7637481689453126e-06,
      "model_forward_time": 5.605272531509399,
      "step": 2898
    },
    {
      "epoch": 2.7637481689453126e-06,
      "step": 2898,
      "training_step_time": 17.096033096313477
    },
    {
      "epoch": 2.7647018432617188e-06,
      "model_forward_time": 5.6243226528167725,
      "step": 2899
    },
    {
      "epoch": 2.7647018432617188e-06,
      "step": 2899,
      "training_step_time": 17.11129093170166
    },
    {
      "epoch": 2.765655517578125e-06,
      "grad_norm": 0.718311071395874,
      "learning_rate": 1e-05,
      "loss": 0.0712,
      "step": 2900
    },
    {
      "epoch": 2.765655517578125e-06,
      "model_forward_time": 5.608481168746948,
      "step": 2900
    },
    {
      "dynamics_loss_avg": 0.05646102018654346,
      "epoch": 2.765655517578125e-06,
      "step": 2900
    },
    {
      "action_loss_avg": 0.016314213164150715,
      "epoch": 2.765655517578125e-06,
      "step": 2900
    },
    {
      "epoch": 2.765655517578125e-06,
      "step": 2900,
      "training_step_time": 17.080475330352783
    },
    {
      "epoch": 2.766609191894531e-06,
      "model_forward_time": 5.58652138710022,
      "step": 2901
    },
    {
      "epoch": 2.766609191894531e-06,
      "step": 2901,
      "training_step_time": 17.00726318359375
    },
    {
      "epoch": 2.7675628662109376e-06,
      "model_forward_time": 5.586140155792236,
      "step": 2902
    },
    {
      "epoch": 2.7675628662109376e-06,
      "step": 2902,
      "training_step_time": 17.38033962249756
    },
    {
      "epoch": 2.7685165405273438e-06,
      "model_forward_time": 5.584361791610718,
      "step": 2903
    },
    {
      "epoch": 2.7685165405273438e-06,
      "step": 2903,
      "training_step_time": 17.022516012191772
    },
    {
      "epoch": 2.76947021484375e-06,
      "model_forward_time": 5.581701993942261,
      "step": 2904
    },
    {
      "epoch": 2.76947021484375e-06,
      "step": 2904,
      "training_step_time": 16.994969367980957
    },
    {
      "epoch": 2.770423889160156e-06,
      "model_forward_time": 5.587879657745361,
      "step": 2905
    },
    {
      "epoch": 2.770423889160156e-06,
      "step": 2905,
      "training_step_time": 17.003726482391357
    },
    {
      "epoch": 2.7713775634765626e-06,
      "model_forward_time": 5.587095737457275,
      "step": 2906
    },
    {
      "epoch": 2.7713775634765626e-06,
      "step": 2906,
      "training_step_time": 17.027342319488525
    },
    {
      "epoch": 2.7723312377929688e-06,
      "model_forward_time": 5.583259582519531,
      "step": 2907
    },
    {
      "epoch": 2.7723312377929688e-06,
      "step": 2907,
      "training_step_time": 17.01714062690735
    },
    {
      "epoch": 2.773284912109375e-06,
      "model_forward_time": 5.942343473434448,
      "step": 2908
    },
    {
      "epoch": 2.773284912109375e-06,
      "step": 2908,
      "training_step_time": 17.383740663528442
    },
    {
      "epoch": 2.774238586425781e-06,
      "model_forward_time": 5.611300468444824,
      "step": 2909
    },
    {
      "epoch": 2.774238586425781e-06,
      "step": 2909,
      "training_step_time": 17.095293045043945
    },
    {
      "epoch": 2.7751922607421876e-06,
      "grad_norm": 1.371052622795105,
      "learning_rate": 1e-05,
      "loss": 0.0678,
      "step": 2910
    },
    {
      "epoch": 2.7751922607421876e-06,
      "model_forward_time": 5.617337703704834,
      "step": 2910
    },
    {
      "dynamics_loss_avg": 0.055698657594621184,
      "epoch": 2.7751922607421876e-06,
      "step": 2910
    },
    {
      "action_loss_avg": 0.014079432655125856,
      "epoch": 2.7751922607421876e-06,
      "step": 2910
    },
    {
      "epoch": 2.7751922607421876e-06,
      "step": 2910,
      "training_step_time": 17.107308864593506
    },
    {
      "epoch": 2.7761459350585938e-06,
      "model_forward_time": 5.615979433059692,
      "step": 2911
    },
    {
      "epoch": 2.7761459350585938e-06,
      "step": 2911,
      "training_step_time": 17.117004871368408
    },
    {
      "epoch": 2.777099609375e-06,
      "model_forward_time": 5.622981309890747,
      "step": 2912
    },
    {
      "epoch": 2.777099609375e-06,
      "step": 2912,
      "training_step_time": 17.122636318206787
    },
    {
      "epoch": 2.778053283691406e-06,
      "model_forward_time": 5.620753765106201,
      "step": 2913
    },
    {
      "epoch": 2.778053283691406e-06,
      "step": 2913,
      "training_step_time": 17.46768879890442
    },
    {
      "epoch": 2.7790069580078126e-06,
      "model_forward_time": 5.60418176651001,
      "step": 2914
    },
    {
      "epoch": 2.7790069580078126e-06,
      "step": 2914,
      "training_step_time": 17.063697576522827
    },
    {
      "epoch": 2.7799606323242188e-06,
      "model_forward_time": 5.6082727909088135,
      "step": 2915
    },
    {
      "epoch": 2.7799606323242188e-06,
      "step": 2915,
      "training_step_time": 17.071135997772217
    },
    {
      "epoch": 2.780914306640625e-06,
      "model_forward_time": 5.593570947647095,
      "step": 2916
    },
    {
      "epoch": 2.780914306640625e-06,
      "step": 2916,
      "training_step_time": 17.05753803253174
    },
    {
      "epoch": 2.781867980957031e-06,
      "model_forward_time": 5.597541093826294,
      "step": 2917
    },
    {
      "epoch": 2.781867980957031e-06,
      "step": 2917,
      "training_step_time": 17.066802978515625
    },
    {
      "epoch": 2.7828216552734376e-06,
      "model_forward_time": 5.596539258956909,
      "step": 2918
    },
    {
      "epoch": 2.7828216552734376e-06,
      "step": 2918,
      "training_step_time": 17.042239904403687
    },
    {
      "epoch": 2.7837753295898438e-06,
      "model_forward_time": 5.9526965618133545,
      "step": 2919
    },
    {
      "epoch": 2.7837753295898438e-06,
      "step": 2919,
      "training_step_time": 17.38406467437744
    },
    {
      "epoch": 2.78472900390625e-06,
      "grad_norm": 0.6348021626472473,
      "learning_rate": 1e-05,
      "loss": 0.0909,
      "step": 2920
    },
    {
      "epoch": 2.78472900390625e-06,
      "model_forward_time": 5.593183755874634,
      "step": 2920
    },
    {
      "dynamics_loss_avg": 0.06948502156883478,
      "epoch": 2.78472900390625e-06,
      "step": 2920
    },
    {
      "action_loss_avg": 0.019046223955228925,
      "epoch": 2.78472900390625e-06,
      "step": 2920
    },
    {
      "epoch": 2.78472900390625e-06,
      "step": 2920,
      "training_step_time": 17.04070258140564
    },
    {
      "epoch": 2.785682678222656e-06,
      "model_forward_time": 5.602266788482666,
      "step": 2921
    },
    {
      "epoch": 2.785682678222656e-06,
      "step": 2921,
      "training_step_time": 17.078091621398926
    },
    {
      "epoch": 2.7866363525390626e-06,
      "model_forward_time": 5.604321479797363,
      "step": 2922
    },
    {
      "epoch": 2.7866363525390626e-06,
      "step": 2922,
      "training_step_time": 17.069511890411377
    },
    {
      "epoch": 2.787590026855469e-06,
      "model_forward_time": 5.599312782287598,
      "step": 2923
    },
    {
      "epoch": 2.787590026855469e-06,
      "step": 2923,
      "training_step_time": 17.053115129470825
    },
    {
      "epoch": 2.788543701171875e-06,
      "model_forward_time": 5.961167573928833,
      "step": 2924
    },
    {
      "epoch": 2.788543701171875e-06,
      "step": 2924,
      "training_step_time": 17.428529262542725
    },
    {
      "epoch": 2.789497375488281e-06,
      "model_forward_time": 5.6041786670684814,
      "step": 2925
    },
    {
      "epoch": 2.789497375488281e-06,
      "step": 2925,
      "training_step_time": 17.054489850997925
    },
    {
      "epoch": 2.7904510498046876e-06,
      "model_forward_time": 5.606889963150024,
      "step": 2926
    },
    {
      "epoch": 2.7904510498046876e-06,
      "step": 2926,
      "training_step_time": 17.07133436203003
    },
    {
      "epoch": 2.791404724121094e-06,
      "model_forward_time": 5.59737229347229,
      "step": 2927
    },
    {
      "epoch": 2.791404724121094e-06,
      "step": 2927,
      "training_step_time": 17.0541250705719
    },
    {
      "epoch": 2.7923583984375e-06,
      "model_forward_time": 5.595592498779297,
      "step": 2928
    },
    {
      "epoch": 2.7923583984375e-06,
      "step": 2928,
      "training_step_time": 17.01914930343628
    },
    {
      "epoch": 2.793312072753906e-06,
      "model_forward_time": 5.600343227386475,
      "step": 2929
    },
    {
      "epoch": 2.793312072753906e-06,
      "step": 2929,
      "training_step_time": 17.068782329559326
    },
    {
      "epoch": 2.7942657470703127e-06,
      "grad_norm": 2.1088645458221436,
      "learning_rate": 1e-05,
      "loss": 0.0918,
      "step": 2930
    },
    {
      "epoch": 2.7942657470703127e-06,
      "model_forward_time": 5.984036207199097,
      "step": 2930
    },
    {
      "dynamics_loss_avg": 0.06524906512349844,
      "epoch": 2.7942657470703127e-06,
      "step": 2930
    },
    {
      "action_loss_avg": 0.02079092541243881,
      "epoch": 2.7942657470703127e-06,
      "step": 2930
    },
    {
      "epoch": 2.7942657470703127e-06,
      "step": 2930,
      "training_step_time": 17.49396324157715
    },
    {
      "epoch": 2.795219421386719e-06,
      "model_forward_time": 5.621316432952881,
      "step": 2931
    },
    {
      "epoch": 2.795219421386719e-06,
      "step": 2931,
      "training_step_time": 17.134052276611328
    },
    {
      "epoch": 2.796173095703125e-06,
      "model_forward_time": 5.636352062225342,
      "step": 2932
    },
    {
      "epoch": 2.796173095703125e-06,
      "step": 2932,
      "training_step_time": 17.130383729934692
    },
    {
      "epoch": 2.797126770019531e-06,
      "model_forward_time": 5.612860202789307,
      "step": 2933
    },
    {
      "epoch": 2.797126770019531e-06,
      "step": 2933,
      "training_step_time": 17.088534355163574
    },
    {
      "epoch": 2.7980804443359377e-06,
      "model_forward_time": 5.598196268081665,
      "step": 2934
    },
    {
      "epoch": 2.7980804443359377e-06,
      "step": 2934,
      "training_step_time": 17.0647189617157
    },
    {
      "epoch": 2.799034118652344e-06,
      "model_forward_time": 5.596476316452026,
      "step": 2935
    },
    {
      "epoch": 2.799034118652344e-06,
      "step": 2935,
      "training_step_time": 17.413185834884644
    },
    {
      "epoch": 2.79998779296875e-06,
      "model_forward_time": 5.603107929229736,
      "step": 2936
    },
    {
      "epoch": 2.79998779296875e-06,
      "step": 2936,
      "training_step_time": 17.089282274246216
    },
    {
      "epoch": 2.800941467285156e-06,
      "model_forward_time": 5.62021017074585,
      "step": 2937
    },
    {
      "epoch": 2.800941467285156e-06,
      "step": 2937,
      "training_step_time": 17.104642391204834
    },
    {
      "epoch": 2.8018951416015627e-06,
      "model_forward_time": 5.595848083496094,
      "step": 2938
    },
    {
      "epoch": 2.8018951416015627e-06,
      "step": 2938,
      "training_step_time": 17.03311800956726
    },
    {
      "epoch": 2.802848815917969e-06,
      "model_forward_time": 5.580467700958252,
      "step": 2939
    },
    {
      "epoch": 2.802848815917969e-06,
      "step": 2939,
      "training_step_time": 16.990145921707153
    },
    {
      "epoch": 2.803802490234375e-06,
      "grad_norm": 0.650088369846344,
      "learning_rate": 1e-05,
      "loss": 0.076,
      "step": 2940
    },
    {
      "epoch": 2.803802490234375e-06,
      "model_forward_time": 5.574125289916992,
      "step": 2940
    },
    {
      "dynamics_loss_avg": 0.06104919072240591,
      "epoch": 2.803802490234375e-06,
      "step": 2940
    },
    {
      "action_loss_avg": 0.016133796563372015,
      "epoch": 2.803802490234375e-06,
      "step": 2940
    },
    {
      "epoch": 2.803802490234375e-06,
      "step": 2940,
      "training_step_time": 17.014889001846313
    },
    {
      "epoch": 2.804756164550781e-06,
      "model_forward_time": 5.9584009647369385,
      "step": 2941
    },
    {
      "epoch": 2.804756164550781e-06,
      "step": 2941,
      "training_step_time": 17.385344743728638
    },
    {
      "epoch": 2.8057098388671877e-06,
      "model_forward_time": 5.581873655319214,
      "step": 2942
    },
    {
      "epoch": 2.8057098388671877e-06,
      "step": 2942,
      "training_step_time": 17.035117387771606
    },
    {
      "epoch": 2.806663513183594e-06,
      "model_forward_time": 5.598800420761108,
      "step": 2943
    },
    {
      "epoch": 2.806663513183594e-06,
      "step": 2943,
      "training_step_time": 17.092182159423828
    },
    {
      "epoch": 2.8076171875e-06,
      "model_forward_time": 5.614765644073486,
      "step": 2944
    },
    {
      "epoch": 2.8076171875e-06,
      "step": 2944,
      "training_step_time": 17.112755060195923
    },
    {
      "epoch": 2.808570861816406e-06,
      "model_forward_time": 5.61824369430542,
      "step": 2945
    },
    {
      "epoch": 2.808570861816406e-06,
      "step": 2945,
      "training_step_time": 17.123519897460938
    },
    {
      "epoch": 2.8095245361328127e-06,
      "model_forward_time": 5.613905191421509,
      "step": 2946
    },
    {
      "epoch": 2.8095245361328127e-06,
      "step": 2946,
      "training_step_time": 17.46909761428833
    },
    {
      "epoch": 2.810478210449219e-06,
      "model_forward_time": 5.595454454421997,
      "step": 2947
    },
    {
      "epoch": 2.810478210449219e-06,
      "step": 2947,
      "training_step_time": 17.050318717956543
    },
    {
      "epoch": 2.811431884765625e-06,
      "model_forward_time": 5.591735363006592,
      "step": 2948
    },
    {
      "epoch": 2.811431884765625e-06,
      "step": 2948,
      "training_step_time": 17.02537989616394
    },
    {
      "epoch": 2.812385559082031e-06,
      "model_forward_time": 5.591741323471069,
      "step": 2949
    },
    {
      "epoch": 2.812385559082031e-06,
      "step": 2949,
      "training_step_time": 17.051998138427734
    },
    {
      "epoch": 2.8133392333984377e-06,
      "grad_norm": 0.8921341300010681,
      "learning_rate": 1e-05,
      "loss": 0.0692,
      "step": 2950
    },
    {
      "epoch": 2.8133392333984377e-06,
      "model_forward_time": 5.595930576324463,
      "step": 2950
    },
    {
      "dynamics_loss_avg": 0.05954812318086624,
      "epoch": 2.8133392333984377e-06,
      "step": 2950
    },
    {
      "action_loss_avg": 0.014729231968522072,
      "epoch": 2.8133392333984377e-06,
      "step": 2950
    },
    {
      "epoch": 2.8133392333984377e-06,
      "step": 2950,
      "training_step_time": 17.038960456848145
    },
    {
      "epoch": 2.814292907714844e-06,
      "model_forward_time": 5.599217176437378,
      "step": 2951
    },
    {
      "epoch": 2.814292907714844e-06,
      "step": 2951,
      "training_step_time": 17.05812358856201
    },
    {
      "epoch": 2.81524658203125e-06,
      "model_forward_time": 5.957622289657593,
      "step": 2952
    },
    {
      "epoch": 2.81524658203125e-06,
      "step": 2952,
      "training_step_time": 17.411274194717407
    },
    {
      "epoch": 2.816200256347656e-06,
      "model_forward_time": 5.601940870285034,
      "step": 2953
    },
    {
      "epoch": 2.816200256347656e-06,
      "step": 2953,
      "training_step_time": 17.080182552337646
    },
    {
      "epoch": 2.8171539306640627e-06,
      "model_forward_time": 5.614229917526245,
      "step": 2954
    },
    {
      "epoch": 2.8171539306640627e-06,
      "step": 2954,
      "training_step_time": 17.102816104888916
    },
    {
      "epoch": 2.818107604980469e-06,
      "model_forward_time": 5.609074354171753,
      "step": 2955
    },
    {
      "epoch": 2.818107604980469e-06,
      "step": 2955,
      "training_step_time": 17.098284244537354
    },
    {
      "epoch": 2.819061279296875e-06,
      "model_forward_time": 5.608924865722656,
      "step": 2956
    },
    {
      "epoch": 2.819061279296875e-06,
      "step": 2956,
      "training_step_time": 17.056946992874146
    },
    {
      "epoch": 2.820014953613281e-06,
      "model_forward_time": 5.944006681442261,
      "step": 2957
    },
    {
      "epoch": 2.820014953613281e-06,
      "step": 2957,
      "training_step_time": 17.37312889099121
    },
    {
      "epoch": 2.8209686279296877e-06,
      "model_forward_time": 5.595056772232056,
      "step": 2958
    },
    {
      "epoch": 2.8209686279296877e-06,
      "step": 2958,
      "training_step_time": 17.05504536628723
    },
    {
      "epoch": 2.821922302246094e-06,
      "model_forward_time": 5.595317125320435,
      "step": 2959
    },
    {
      "epoch": 2.821922302246094e-06,
      "step": 2959,
      "training_step_time": 17.05344033241272
    },
    {
      "epoch": 2.8228759765625e-06,
      "grad_norm": 1.0028893947601318,
      "learning_rate": 1e-05,
      "loss": 0.0729,
      "step": 2960
    },
    {
      "epoch": 2.8228759765625e-06,
      "model_forward_time": 5.6006269454956055,
      "step": 2960
    },
    {
      "dynamics_loss_avg": 0.05758942384272814,
      "epoch": 2.8228759765625e-06,
      "step": 2960
    },
    {
      "action_loss_avg": 0.015428546769544482,
      "epoch": 2.8228759765625e-06,
      "step": 2960
    },
    {
      "epoch": 2.8228759765625e-06,
      "step": 2960,
      "training_step_time": 17.038320541381836
    },
    {
      "epoch": 2.823829650878906e-06,
      "model_forward_time": 5.614434719085693,
      "step": 2961
    },
    {
      "epoch": 2.823829650878906e-06,
      "step": 2961,
      "training_step_time": 17.07602095603943
    },
    {
      "epoch": 2.8247833251953127e-06,
      "model_forward_time": 5.591834306716919,
      "step": 2962
    },
    {
      "epoch": 2.8247833251953127e-06,
      "step": 2962,
      "training_step_time": 17.046504974365234
    },
    {
      "epoch": 2.825736999511719e-06,
      "model_forward_time": 5.964134216308594,
      "step": 2963
    },
    {
      "epoch": 2.825736999511719e-06,
      "step": 2963,
      "training_step_time": 17.392804861068726
    },
    {
      "epoch": 2.826690673828125e-06,
      "model_forward_time": 5.598036050796509,
      "step": 2964
    },
    {
      "epoch": 2.826690673828125e-06,
      "step": 2964,
      "training_step_time": 17.038845539093018
    },
    {
      "epoch": 2.827644348144531e-06,
      "model_forward_time": 5.587177991867065,
      "step": 2965
    },
    {
      "epoch": 2.827644348144531e-06,
      "step": 2965,
      "training_step_time": 17.052491188049316
    },
    {
      "epoch": 2.8285980224609377e-06,
      "model_forward_time": 5.606831312179565,
      "step": 2966
    },
    {
      "epoch": 2.8285980224609377e-06,
      "step": 2966,
      "training_step_time": 17.096113204956055
    },
    {
      "epoch": 2.829551696777344e-06,
      "model_forward_time": 5.616417407989502,
      "step": 2967
    },
    {
      "epoch": 2.829551696777344e-06,
      "step": 2967,
      "training_step_time": 17.100651741027832
    },
    {
      "epoch": 2.83050537109375e-06,
      "model_forward_time": 5.612807512283325,
      "step": 2968
    },
    {
      "epoch": 2.83050537109375e-06,
      "step": 2968,
      "training_step_time": 17.418160438537598
    },
    {
      "epoch": 2.831459045410156e-06,
      "model_forward_time": 5.5886523723602295,
      "step": 2969
    },
    {
      "epoch": 2.831459045410156e-06,
      "step": 2969,
      "training_step_time": 17.004799127578735
    },
    {
      "epoch": 2.8324127197265627e-06,
      "grad_norm": 1.503273606300354,
      "learning_rate": 1e-05,
      "loss": 0.0858,
      "step": 2970
    },
    {
      "epoch": 2.8324127197265627e-06,
      "model_forward_time": 5.577877521514893,
      "step": 2970
    },
    {
      "dynamics_loss_avg": 0.06567172445356846,
      "epoch": 2.8324127197265627e-06,
      "step": 2970
    },
    {
      "action_loss_avg": 0.014556875871494412,
      "epoch": 2.8324127197265627e-06,
      "step": 2970
    },
    {
      "epoch": 2.8324127197265627e-06,
      "step": 2970,
      "training_step_time": 17.00309133529663
    },
    {
      "epoch": 2.833366394042969e-06,
      "model_forward_time": 5.568984746932983,
      "step": 2971
    },
    {
      "epoch": 2.833366394042969e-06,
      "step": 2971,
      "training_step_time": 17.007838010787964
    },
    {
      "epoch": 2.834320068359375e-06,
      "model_forward_time": 5.572895050048828,
      "step": 2972
    },
    {
      "epoch": 2.834320068359375e-06,
      "step": 2972,
      "training_step_time": 16.98380708694458
    },
    {
      "epoch": 2.835273742675781e-06,
      "model_forward_time": 5.576002597808838,
      "step": 2973
    },
    {
      "epoch": 2.835273742675781e-06,
      "step": 2973,
      "training_step_time": 16.995609045028687
    },
    {
      "epoch": 2.8362274169921877e-06,
      "model_forward_time": 5.950277090072632,
      "step": 2974
    },
    {
      "epoch": 2.8362274169921877e-06,
      "step": 2974,
      "training_step_time": 17.37967324256897
    },
    {
      "epoch": 2.837181091308594e-06,
      "model_forward_time": 5.581657648086548,
      "step": 2975
    },
    {
      "epoch": 2.837181091308594e-06,
      "step": 2975,
      "training_step_time": 17.02096700668335
    },
    {
      "epoch": 2.838134765625e-06,
      "model_forward_time": 5.595752000808716,
      "step": 2976
    },
    {
      "epoch": 2.838134765625e-06,
      "step": 2976,
      "training_step_time": 17.07066059112549
    },
    {
      "epoch": 2.839088439941406e-06,
      "model_forward_time": 5.606377124786377,
      "step": 2977
    },
    {
      "epoch": 2.839088439941406e-06,
      "step": 2977,
      "training_step_time": 17.100752592086792
    },
    {
      "epoch": 2.8400421142578127e-06,
      "model_forward_time": 5.614914417266846,
      "step": 2978
    },
    {
      "epoch": 2.8400421142578127e-06,
      "step": 2978,
      "training_step_time": 17.097266912460327
    },
    {
      "epoch": 2.840995788574219e-06,
      "model_forward_time": 5.95824408531189,
      "step": 2979
    },
    {
      "epoch": 2.840995788574219e-06,
      "step": 2979,
      "training_step_time": 17.409695148468018
    },
    {
      "epoch": 2.841949462890625e-06,
      "grad_norm": 0.4456499516963959,
      "learning_rate": 1e-05,
      "loss": 0.0723,
      "step": 2980
    },
    {
      "epoch": 2.841949462890625e-06,
      "model_forward_time": 5.587687015533447,
      "step": 2980
    },
    {
      "dynamics_loss_avg": 0.06569342594593763,
      "epoch": 2.841949462890625e-06,
      "step": 2980
    },
    {
      "action_loss_avg": 0.01521865869872272,
      "epoch": 2.841949462890625e-06,
      "step": 2980
    },
    {
      "epoch": 2.841949462890625e-06,
      "step": 2980,
      "training_step_time": 17.009379148483276
    },
    {
      "epoch": 2.842903137207031e-06,
      "model_forward_time": 5.5839948654174805,
      "step": 2981
    },
    {
      "epoch": 2.842903137207031e-06,
      "step": 2981,
      "training_step_time": 17.007911920547485
    },
    {
      "epoch": 2.8438568115234377e-06,
      "model_forward_time": 5.577173948287964,
      "step": 2982
    },
    {
      "epoch": 2.8438568115234377e-06,
      "step": 2982,
      "training_step_time": 17.0145046710968
    },
    {
      "epoch": 2.844810485839844e-06,
      "model_forward_time": 5.587958335876465,
      "step": 2983
    },
    {
      "epoch": 2.844810485839844e-06,
      "step": 2983,
      "training_step_time": 17.041295289993286
    },
    {
      "epoch": 2.84576416015625e-06,
      "model_forward_time": 5.603732585906982,
      "step": 2984
    },
    {
      "epoch": 2.84576416015625e-06,
      "step": 2984,
      "training_step_time": 17.077253818511963
    },
    {
      "epoch": 2.846717834472656e-06,
      "model_forward_time": 5.980411529541016,
      "step": 2985
    },
    {
      "epoch": 2.846717834472656e-06,
      "step": 2985,
      "training_step_time": 17.48457622528076
    },
    {
      "epoch": 2.8476715087890627e-06,
      "model_forward_time": 5.622361183166504,
      "step": 2986
    },
    {
      "epoch": 2.8476715087890627e-06,
      "step": 2986,
      "training_step_time": 17.134739637374878
    },
    {
      "epoch": 2.848625183105469e-06,
      "model_forward_time": 5.6225879192352295,
      "step": 2987
    },
    {
      "epoch": 2.848625183105469e-06,
      "step": 2987,
      "training_step_time": 17.12175416946411
    },
    {
      "epoch": 2.849578857421875e-06,
      "model_forward_time": 5.61524510383606,
      "step": 2988
    },
    {
      "epoch": 2.849578857421875e-06,
      "step": 2988,
      "training_step_time": 17.108649015426636
    },
    {
      "epoch": 2.850532531738281e-06,
      "model_forward_time": 5.612108469009399,
      "step": 2989
    },
    {
      "epoch": 2.850532531738281e-06,
      "step": 2989,
      "training_step_time": 17.073521375656128
    },
    {
      "epoch": 2.8514862060546877e-06,
      "grad_norm": 0.9334211945533752,
      "learning_rate": 1e-05,
      "loss": 0.082,
      "step": 2990
    },
    {
      "epoch": 2.8514862060546877e-06,
      "model_forward_time": 5.952013254165649,
      "step": 2990
    },
    {
      "dynamics_loss_avg": 0.06402297914028168,
      "epoch": 2.8514862060546877e-06,
      "step": 2990
    },
    {
      "action_loss_avg": 0.012176730670034885,
      "epoch": 2.8514862060546877e-06,
      "step": 2990
    },
    {
      "epoch": 2.8514862060546877e-06,
      "step": 2990,
      "training_step_time": 17.408075094223022
    },
    {
      "epoch": 2.852439880371094e-06,
      "model_forward_time": 5.5935516357421875,
      "step": 2991
    },
    {
      "epoch": 2.852439880371094e-06,
      "step": 2991,
      "training_step_time": 17.0394127368927
    },
    {
      "epoch": 2.8533935546875e-06,
      "model_forward_time": 5.594072580337524,
      "step": 2992
    },
    {
      "epoch": 2.8533935546875e-06,
      "step": 2992,
      "training_step_time": 17.048424243927002
    },
    {
      "epoch": 2.854347229003906e-06,
      "model_forward_time": 4.268275737762451,
      "step": 2993
    },
    {
      "epoch": 2.854347229003906e-06,
      "step": 2993,
      "training_step_time": 12.783553838729858
    },
    {
      "epoch": 2.8553009033203127e-06,
      "model_forward_time": 5.6012139320373535,
      "step": 2994
    },
    {
      "epoch": 2.8553009033203127e-06,
      "step": 2994,
      "training_step_time": 17.049147129058838
    },
    {
      "epoch": 2.856254577636719e-06,
      "model_forward_time": 5.603033065795898,
      "step": 2995
    },
    {
      "epoch": 2.856254577636719e-06,
      "step": 2995,
      "training_step_time": 17.442096948623657
    },
    {
      "epoch": 2.857208251953125e-06,
      "model_forward_time": 5.605274677276611,
      "step": 2996
    },
    {
      "epoch": 2.857208251953125e-06,
      "step": 2996,
      "training_step_time": 17.080763339996338
    },
    {
      "epoch": 2.858161926269531e-06,
      "model_forward_time": 5.595992088317871,
      "step": 2997
    },
    {
      "epoch": 2.858161926269531e-06,
      "step": 2997,
      "training_step_time": 17.06891393661499
    },
    {
      "epoch": 2.8591156005859377e-06,
      "model_forward_time": 5.596897602081299,
      "step": 2998
    },
    {
      "epoch": 2.8591156005859377e-06,
      "step": 2998,
      "training_step_time": 17.045045614242554
    },
    {
      "epoch": 2.860069274902344e-06,
      "model_forward_time": 5.594949007034302,
      "step": 2999
    },
    {
      "epoch": 2.860069274902344e-06,
      "step": 2999,
      "training_step_time": 17.036409616470337
    },
    {
      "epoch": 2.86102294921875e-06,
      "grad_norm": 0.812512218952179,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 3000
    },
    {
      "epoch": 2.86102294921875e-06,
      "model_forward_time": 5.637332201004028,
      "step": 3000
    },
    {
      "dynamics_loss_avg": 0.055815395154058936,
      "epoch": 2.86102294921875e-06,
      "step": 3000
    },
    {
      "action_loss_avg": 0.017905012611299752,
      "epoch": 2.86102294921875e-06,
      "step": 3000
    },
    {
      "epoch": 2.86102294921875e-06,
      "step": 3000,
      "training_step_time": 17.131494998931885
    },
    {
      "epoch": 2.861976623535156e-06,
      "model_forward_time": 6.008355140686035,
      "step": 3001
    },
    {
      "epoch": 2.861976623535156e-06,
      "step": 3001,
      "training_step_time": 17.52009391784668
    },
    {
      "epoch": 2.8629302978515623e-06,
      "model_forward_time": 5.656423330307007,
      "step": 3002
    },
    {
      "epoch": 2.8629302978515623e-06,
      "step": 3002,
      "training_step_time": 17.15009307861328
    },
    {
      "epoch": 2.863883972167969e-06,
      "model_forward_time": 5.656443357467651,
      "step": 3003
    },
    {
      "epoch": 2.863883972167969e-06,
      "step": 3003,
      "training_step_time": 17.115477323532104
    },
    {
      "epoch": 2.864837646484375e-06,
      "model_forward_time": 5.6044182777404785,
      "step": 3004
    },
    {
      "epoch": 2.864837646484375e-06,
      "step": 3004,
      "training_step_time": 17.065244913101196
    },
    {
      "epoch": 2.865791320800781e-06,
      "model_forward_time": 5.593705892562866,
      "step": 3005
    },
    {
      "epoch": 2.865791320800781e-06,
      "step": 3005,
      "training_step_time": 17.03435230255127
    },
    {
      "epoch": 2.8667449951171873e-06,
      "model_forward_time": 5.959939479827881,
      "step": 3006
    },
    {
      "epoch": 2.8667449951171873e-06,
      "step": 3006,
      "training_step_time": 17.39258599281311
    },
    {
      "epoch": 2.867698669433594e-06,
      "model_forward_time": 5.584690809249878,
      "step": 3007
    },
    {
      "epoch": 2.867698669433594e-06,
      "step": 3007,
      "training_step_time": 17.0049045085907
    },
    {
      "epoch": 2.86865234375e-06,
      "model_forward_time": 5.590042591094971,
      "step": 3008
    },
    {
      "epoch": 2.86865234375e-06,
      "step": 3008,
      "training_step_time": 17.03149676322937
    },
    {
      "epoch": 2.869606018066406e-06,
      "model_forward_time": 5.6010096073150635,
      "step": 3009
    },
    {
      "epoch": 2.869606018066406e-06,
      "step": 3009,
      "training_step_time": 17.068832635879517
    },
    {
      "epoch": 2.8705596923828123e-06,
      "grad_norm": 2.5228793621063232,
      "learning_rate": 1e-05,
      "loss": 0.0869,
      "step": 3010
    },
    {
      "epoch": 2.8705596923828123e-06,
      "model_forward_time": 5.603160381317139,
      "step": 3010
    },
    {
      "dynamics_loss_avg": 0.06906436625868081,
      "epoch": 2.8705596923828123e-06,
      "step": 3010
    },
    {
      "action_loss_avg": 0.01851870180107653,
      "epoch": 2.8705596923828123e-06,
      "step": 3010
    },
    {
      "epoch": 2.8705596923828123e-06,
      "step": 3010,
      "training_step_time": 17.091919422149658
    },
    {
      "epoch": 2.871513366699219e-06,
      "model_forward_time": 5.606098651885986,
      "step": 3011
    },
    {
      "epoch": 2.871513366699219e-06,
      "step": 3011,
      "training_step_time": 17.098434925079346
    },
    {
      "epoch": 2.872467041015625e-06,
      "model_forward_time": 5.965886831283569,
      "step": 3012
    },
    {
      "epoch": 2.872467041015625e-06,
      "step": 3012,
      "training_step_time": 17.472946405410767
    },
    {
      "epoch": 2.873420715332031e-06,
      "model_forward_time": 5.606330871582031,
      "step": 3013
    },
    {
      "epoch": 2.873420715332031e-06,
      "step": 3013,
      "training_step_time": 17.097639322280884
    },
    {
      "epoch": 2.8743743896484373e-06,
      "model_forward_time": 5.6177589893341064,
      "step": 3014
    },
    {
      "epoch": 2.8743743896484373e-06,
      "step": 3014,
      "training_step_time": 17.112770795822144
    },
    {
      "epoch": 2.875328063964844e-06,
      "model_forward_time": 5.608944416046143,
      "step": 3015
    },
    {
      "epoch": 2.875328063964844e-06,
      "step": 3015,
      "training_step_time": 17.090839624404907
    },
    {
      "epoch": 2.87628173828125e-06,
      "model_forward_time": 5.6034135818481445,
      "step": 3016
    },
    {
      "epoch": 2.87628173828125e-06,
      "step": 3016,
      "training_step_time": 17.067511558532715
    },
    {
      "epoch": 2.877235412597656e-06,
      "model_forward_time": 5.943057060241699,
      "step": 3017
    },
    {
      "epoch": 2.877235412597656e-06,
      "step": 3017,
      "training_step_time": 17.400164365768433
    },
    {
      "epoch": 2.8781890869140623e-06,
      "model_forward_time": 5.592801332473755,
      "step": 3018
    },
    {
      "epoch": 2.8781890869140623e-06,
      "step": 3018,
      "training_step_time": 17.05607533454895
    },
    {
      "epoch": 2.879142761230469e-06,
      "model_forward_time": 5.60208797454834,
      "step": 3019
    },
    {
      "epoch": 2.879142761230469e-06,
      "step": 3019,
      "training_step_time": 17.091822147369385
    },
    {
      "epoch": 2.880096435546875e-06,
      "grad_norm": 0.319094717502594,
      "learning_rate": 1e-05,
      "loss": 0.0868,
      "step": 3020
    },
    {
      "epoch": 2.880096435546875e-06,
      "model_forward_time": 5.603024482727051,
      "step": 3020
    },
    {
      "dynamics_loss_avg": 0.06429032925516368,
      "epoch": 2.880096435546875e-06,
      "step": 3020
    },
    {
      "action_loss_avg": 0.022433590493164957,
      "epoch": 2.880096435546875e-06,
      "step": 3020
    },
    {
      "epoch": 2.880096435546875e-06,
      "step": 3020,
      "training_step_time": 17.089569568634033
    },
    {
      "epoch": 2.881050109863281e-06,
      "model_forward_time": 5.609328269958496,
      "step": 3021
    },
    {
      "epoch": 2.881050109863281e-06,
      "step": 3021,
      "training_step_time": 17.08286476135254
    },
    {
      "epoch": 2.8820037841796873e-06,
      "model_forward_time": 5.600013971328735,
      "step": 3022
    },
    {
      "epoch": 2.8820037841796873e-06,
      "step": 3022,
      "training_step_time": 17.449342727661133
    },
    {
      "epoch": 2.882957458496094e-06,
      "model_forward_time": 5.602755784988403,
      "step": 3023
    },
    {
      "epoch": 2.882957458496094e-06,
      "step": 3023,
      "training_step_time": 17.07926297187805
    },
    {
      "epoch": 2.8839111328125e-06,
      "model_forward_time": 5.6035332679748535,
      "step": 3024
    },
    {
      "epoch": 2.8839111328125e-06,
      "step": 3024,
      "training_step_time": 17.080790042877197
    },
    {
      "epoch": 2.884864807128906e-06,
      "model_forward_time": 5.596749544143677,
      "step": 3025
    },
    {
      "epoch": 2.884864807128906e-06,
      "step": 3025,
      "training_step_time": 17.062328815460205
    },
    {
      "epoch": 2.8858184814453123e-06,
      "model_forward_time": 5.589740753173828,
      "step": 3026
    },
    {
      "epoch": 2.8858184814453123e-06,
      "step": 3026,
      "training_step_time": 17.02311658859253
    },
    {
      "epoch": 2.886772155761719e-06,
      "model_forward_time": 5.584331035614014,
      "step": 3027
    },
    {
      "epoch": 2.886772155761719e-06,
      "step": 3027,
      "training_step_time": 17.03683352470398
    },
    {
      "epoch": 2.887725830078125e-06,
      "model_forward_time": 5.960835933685303,
      "step": 3028
    },
    {
      "epoch": 2.887725830078125e-06,
      "step": 3028,
      "training_step_time": 17.416064739227295
    },
    {
      "epoch": 2.888679504394531e-06,
      "model_forward_time": 5.611614227294922,
      "step": 3029
    },
    {
      "epoch": 2.888679504394531e-06,
      "step": 3029,
      "training_step_time": 17.11279296875
    },
    {
      "epoch": 2.8896331787109373e-06,
      "grad_norm": 1.5648664236068726,
      "learning_rate": 1e-05,
      "loss": 0.0842,
      "step": 3030
    },
    {
      "epoch": 2.8896331787109373e-06,
      "model_forward_time": 5.622279167175293,
      "step": 3030
    },
    {
      "dynamics_loss_avg": 0.058675164543092254,
      "epoch": 2.8896331787109373e-06,
      "step": 3030
    },
    {
      "action_loss_avg": 0.019323015632107855,
      "epoch": 2.8896331787109373e-06,
      "step": 3030
    },
    {
      "epoch": 2.8896331787109373e-06,
      "step": 3030,
      "training_step_time": 17.14332127571106
    },
    {
      "epoch": 2.890586853027344e-06,
      "model_forward_time": 5.6304168701171875,
      "step": 3031
    },
    {
      "epoch": 2.890586853027344e-06,
      "step": 3031,
      "training_step_time": 17.13290286064148
    },
    {
      "epoch": 2.89154052734375e-06,
      "model_forward_time": 5.6104278564453125,
      "step": 3032
    },
    {
      "epoch": 2.89154052734375e-06,
      "step": 3032,
      "training_step_time": 17.11092758178711
    },
    {
      "epoch": 2.892494201660156e-06,
      "model_forward_time": 5.970871686935425,
      "step": 3033
    },
    {
      "epoch": 2.892494201660156e-06,
      "step": 3033,
      "training_step_time": 17.462340354919434
    },
    {
      "epoch": 2.8934478759765623e-06,
      "model_forward_time": 5.6088550090789795,
      "step": 3034
    },
    {
      "epoch": 2.8934478759765623e-06,
      "step": 3034,
      "training_step_time": 17.089680671691895
    },
    {
      "epoch": 2.894401550292969e-06,
      "model_forward_time": 5.607002258300781,
      "step": 3035
    },
    {
      "epoch": 2.894401550292969e-06,
      "step": 3035,
      "training_step_time": 17.078068256378174
    },
    {
      "epoch": 2.895355224609375e-06,
      "model_forward_time": 5.60578465461731,
      "step": 3036
    },
    {
      "epoch": 2.895355224609375e-06,
      "step": 3036,
      "training_step_time": 17.0556161403656
    },
    {
      "epoch": 2.896308898925781e-06,
      "model_forward_time": 5.6029722690582275,
      "step": 3037
    },
    {
      "epoch": 2.896308898925781e-06,
      "step": 3037,
      "training_step_time": 17.05412244796753
    },
    {
      "epoch": 2.8972625732421873e-06,
      "model_forward_time": 5.595723867416382,
      "step": 3038
    },
    {
      "epoch": 2.8972625732421873e-06,
      "step": 3038,
      "training_step_time": 17.043286323547363
    },
    {
      "epoch": 2.898216247558594e-06,
      "model_forward_time": 5.936501502990723,
      "step": 3039
    },
    {
      "epoch": 2.898216247558594e-06,
      "step": 3039,
      "training_step_time": 17.36152458190918
    },
    {
      "epoch": 2.899169921875e-06,
      "grad_norm": 1.192044973373413,
      "learning_rate": 1e-05,
      "loss": 0.0659,
      "step": 3040
    },
    {
      "epoch": 2.899169921875e-06,
      "model_forward_time": 5.605062484741211,
      "step": 3040
    },
    {
      "dynamics_loss_avg": 0.05347681641578674,
      "epoch": 2.899169921875e-06,
      "step": 3040
    },
    {
      "action_loss_avg": 0.017269317689351738,
      "epoch": 2.899169921875e-06,
      "step": 3040
    },
    {
      "epoch": 2.899169921875e-06,
      "step": 3040,
      "training_step_time": 17.10501003265381
    },
    {
      "epoch": 2.900123596191406e-06,
      "model_forward_time": 5.6117682456970215,
      "step": 3041
    },
    {
      "epoch": 2.900123596191406e-06,
      "step": 3041,
      "training_step_time": 17.109862804412842
    },
    {
      "epoch": 2.9010772705078123e-06,
      "model_forward_time": 5.613362550735474,
      "step": 3042
    },
    {
      "epoch": 2.9010772705078123e-06,
      "step": 3042,
      "training_step_time": 17.102293968200684
    },
    {
      "epoch": 2.902030944824219e-06,
      "model_forward_time": 5.600167751312256,
      "step": 3043
    },
    {
      "epoch": 2.902030944824219e-06,
      "step": 3043,
      "training_step_time": 17.04582691192627
    },
    {
      "epoch": 2.902984619140625e-06,
      "model_forward_time": 5.95844030380249,
      "step": 3044
    },
    {
      "epoch": 2.902984619140625e-06,
      "step": 3044,
      "training_step_time": 17.37881326675415
    },
    {
      "epoch": 2.903938293457031e-06,
      "model_forward_time": 5.595994234085083,
      "step": 3045
    },
    {
      "epoch": 2.903938293457031e-06,
      "step": 3045,
      "training_step_time": 16.998226165771484
    },
    {
      "epoch": 2.9048919677734373e-06,
      "model_forward_time": 5.5909974575042725,
      "step": 3046
    },
    {
      "epoch": 2.9048919677734373e-06,
      "step": 3046,
      "training_step_time": 17.039730310440063
    },
    {
      "epoch": 2.905845642089844e-06,
      "model_forward_time": 5.590472459793091,
      "step": 3047
    },
    {
      "epoch": 2.905845642089844e-06,
      "step": 3047,
      "training_step_time": 17.07212233543396
    },
    {
      "epoch": 2.90679931640625e-06,
      "model_forward_time": 5.622971296310425,
      "step": 3048
    },
    {
      "epoch": 2.90679931640625e-06,
      "step": 3048,
      "training_step_time": 17.119823932647705
    },
    {
      "epoch": 2.907752990722656e-06,
      "model_forward_time": 5.601909399032593,
      "step": 3049
    },
    {
      "epoch": 2.907752990722656e-06,
      "step": 3049,
      "training_step_time": 17.069748640060425
    },
    {
      "epoch": 2.9087066650390623e-06,
      "grad_norm": 0.7613796591758728,
      "learning_rate": 1e-05,
      "loss": 0.0583,
      "step": 3050
    },
    {
      "epoch": 2.9087066650390623e-06,
      "model_forward_time": 5.937125205993652,
      "step": 3050
    },
    {
      "dynamics_loss_avg": 0.04416163237765432,
      "epoch": 2.9087066650390623e-06,
      "step": 3050
    },
    {
      "action_loss_avg": 0.012233994854614138,
      "epoch": 2.9087066650390623e-06,
      "step": 3050
    },
    {
      "epoch": 2.9087066650390623e-06,
      "step": 3050,
      "training_step_time": 17.380384922027588
    },
    {
      "epoch": 2.909660339355469e-06,
      "model_forward_time": 5.593666315078735,
      "step": 3051
    },
    {
      "epoch": 2.909660339355469e-06,
      "step": 3051,
      "training_step_time": 17.019365549087524
    },
    {
      "epoch": 2.910614013671875e-06,
      "model_forward_time": 5.589592218399048,
      "step": 3052
    },
    {
      "epoch": 2.910614013671875e-06,
      "step": 3052,
      "training_step_time": 17.054044723510742
    },
    {
      "epoch": 2.911567687988281e-06,
      "model_forward_time": 5.598454713821411,
      "step": 3053
    },
    {
      "epoch": 2.911567687988281e-06,
      "step": 3053,
      "training_step_time": 17.063315629959106
    },
    {
      "epoch": 2.9125213623046873e-06,
      "model_forward_time": 5.596246957778931,
      "step": 3054
    },
    {
      "epoch": 2.9125213623046873e-06,
      "step": 3054,
      "training_step_time": 17.072968244552612
    },
    {
      "epoch": 2.913475036621094e-06,
      "model_forward_time": 5.99081826210022,
      "step": 3055
    },
    {
      "epoch": 2.913475036621094e-06,
      "step": 3055,
      "training_step_time": 17.449410676956177
    },
    {
      "epoch": 2.9144287109375e-06,
      "model_forward_time": 5.584501028060913,
      "step": 3056
    },
    {
      "epoch": 2.9144287109375e-06,
      "step": 3056,
      "training_step_time": 17.01228094100952
    },
    {
      "epoch": 2.915382385253906e-06,
      "model_forward_time": 5.584594488143921,
      "step": 3057
    },
    {
      "epoch": 2.915382385253906e-06,
      "step": 3057,
      "training_step_time": 17.020697355270386
    },
    {
      "epoch": 2.9163360595703123e-06,
      "model_forward_time": 5.585286378860474,
      "step": 3058
    },
    {
      "epoch": 2.9163360595703123e-06,
      "step": 3058,
      "training_step_time": 17.003875970840454
    },
    {
      "epoch": 2.917289733886719e-06,
      "model_forward_time": 5.573665142059326,
      "step": 3059
    },
    {
      "epoch": 2.917289733886719e-06,
      "step": 3059,
      "training_step_time": 17.00112223625183
    },
    {
      "epoch": 2.918243408203125e-06,
      "grad_norm": 1.4737426042556763,
      "learning_rate": 1e-05,
      "loss": 0.0943,
      "step": 3060
    },
    {
      "epoch": 2.918243408203125e-06,
      "model_forward_time": 5.5925188064575195,
      "step": 3060
    },
    {
      "dynamics_loss_avg": 0.07549333684146405,
      "epoch": 2.918243408203125e-06,
      "step": 3060
    },
    {
      "action_loss_avg": 0.021672930940985678,
      "epoch": 2.918243408203125e-06,
      "step": 3060
    },
    {
      "epoch": 2.918243408203125e-06,
      "step": 3060,
      "training_step_time": 17.012555837631226
    },
    {
      "epoch": 2.919197082519531e-06,
      "model_forward_time": 5.924053192138672,
      "step": 3061
    },
    {
      "epoch": 2.919197082519531e-06,
      "step": 3061,
      "training_step_time": 17.363611698150635
    },
    {
      "epoch": 2.9201507568359373e-06,
      "model_forward_time": 5.584827899932861,
      "step": 3062
    },
    {
      "epoch": 2.9201507568359373e-06,
      "step": 3062,
      "training_step_time": 17.01067614555359
    },
    {
      "epoch": 2.921104431152344e-06,
      "model_forward_time": 5.606464624404907,
      "step": 3063
    },
    {
      "epoch": 2.921104431152344e-06,
      "step": 3063,
      "training_step_time": 17.10305690765381
    },
    {
      "epoch": 2.92205810546875e-06,
      "model_forward_time": 5.619290113449097,
      "step": 3064
    },
    {
      "epoch": 2.92205810546875e-06,
      "step": 3064,
      "training_step_time": 17.134642362594604
    },
    {
      "epoch": 2.923011779785156e-06,
      "model_forward_time": 5.63707160949707,
      "step": 3065
    },
    {
      "epoch": 2.923011779785156e-06,
      "step": 3065,
      "training_step_time": 17.157204151153564
    },
    {
      "epoch": 2.9239654541015623e-06,
      "model_forward_time": 5.9969165325164795,
      "step": 3066
    },
    {
      "epoch": 2.9239654541015623e-06,
      "step": 3066,
      "training_step_time": 17.499619245529175
    },
    {
      "epoch": 2.924919128417969e-06,
      "model_forward_time": 5.617126703262329,
      "step": 3067
    },
    {
      "epoch": 2.924919128417969e-06,
      "step": 3067,
      "training_step_time": 17.1033034324646
    },
    {
      "epoch": 2.925872802734375e-06,
      "model_forward_time": 5.606148719787598,
      "step": 3068
    },
    {
      "epoch": 2.925872802734375e-06,
      "step": 3068,
      "training_step_time": 17.078261137008667
    },
    {
      "epoch": 2.926826477050781e-06,
      "model_forward_time": 5.606998920440674,
      "step": 3069
    },
    {
      "epoch": 2.926826477050781e-06,
      "step": 3069,
      "training_step_time": 17.07135009765625
    },
    {
      "epoch": 2.9277801513671873e-06,
      "grad_norm": 0.9397907257080078,
      "learning_rate": 1e-05,
      "loss": 0.1054,
      "step": 3070
    },
    {
      "epoch": 2.9277801513671873e-06,
      "model_forward_time": 5.599406480789185,
      "step": 3070
    },
    {
      "dynamics_loss_avg": 0.08047128319740296,
      "epoch": 2.9277801513671873e-06,
      "step": 3070
    },
    {
      "action_loss_avg": 0.02298600282520056,
      "epoch": 2.9277801513671873e-06,
      "step": 3070
    },
    {
      "epoch": 2.9277801513671873e-06,
      "step": 3070,
      "training_step_time": 17.038688898086548
    },
    {
      "epoch": 2.928733825683594e-06,
      "model_forward_time": 5.946760892868042,
      "step": 3071
    },
    {
      "epoch": 2.928733825683594e-06,
      "step": 3071,
      "training_step_time": 17.381540775299072
    },
    {
      "epoch": 2.9296875e-06,
      "model_forward_time": 5.598954916000366,
      "step": 3072
    },
    {
      "epoch": 2.9296875e-06,
      "step": 3072,
      "training_step_time": 17.074641942977905
    },
    {
      "epoch": 2.930641174316406e-06,
      "model_forward_time": 5.616979360580444,
      "step": 3073
    },
    {
      "epoch": 2.930641174316406e-06,
      "step": 3073,
      "training_step_time": 17.115572929382324
    },
    {
      "epoch": 2.9315948486328124e-06,
      "model_forward_time": 5.612714052200317,
      "step": 3074
    },
    {
      "epoch": 2.9315948486328124e-06,
      "step": 3074,
      "training_step_time": 17.093506574630737
    },
    {
      "epoch": 2.932548522949219e-06,
      "model_forward_time": 5.6022233963012695,
      "step": 3075
    },
    {
      "epoch": 2.932548522949219e-06,
      "step": 3075,
      "training_step_time": 17.052355527877808
    },
    {
      "epoch": 2.933502197265625e-06,
      "model_forward_time": 5.590764999389648,
      "step": 3076
    },
    {
      "epoch": 2.933502197265625e-06,
      "step": 3076,
      "training_step_time": 17.04093909263611
    },
    {
      "epoch": 2.934455871582031e-06,
      "model_forward_time": 5.9533233642578125,
      "step": 3077
    },
    {
      "epoch": 2.934455871582031e-06,
      "step": 3077,
      "training_step_time": 17.40655517578125
    },
    {
      "epoch": 2.9354095458984374e-06,
      "model_forward_time": 5.596873044967651,
      "step": 3078
    },
    {
      "epoch": 2.9354095458984374e-06,
      "step": 3078,
      "training_step_time": 17.06942367553711
    },
    {
      "epoch": 2.936363220214844e-06,
      "model_forward_time": 5.614358901977539,
      "step": 3079
    },
    {
      "epoch": 2.936363220214844e-06,
      "step": 3079,
      "training_step_time": 17.128724813461304
    },
    {
      "epoch": 2.93731689453125e-06,
      "grad_norm": 1.0612980127334595,
      "learning_rate": 1e-05,
      "loss": 0.0868,
      "step": 3080
    },
    {
      "epoch": 2.93731689453125e-06,
      "model_forward_time": 5.623635530471802,
      "step": 3080
    },
    {
      "dynamics_loss_avg": 0.06907120998948812,
      "epoch": 2.93731689453125e-06,
      "step": 3080
    },
    {
      "action_loss_avg": 0.015450242348015309,
      "epoch": 2.93731689453125e-06,
      "step": 3080
    },
    {
      "epoch": 2.93731689453125e-06,
      "step": 3080,
      "training_step_time": 17.141637802124023
    },
    {
      "epoch": 2.9382705688476562e-06,
      "model_forward_time": 5.6226725578308105,
      "step": 3081
    },
    {
      "epoch": 2.9382705688476562e-06,
      "step": 3081,
      "training_step_time": 17.133853912353516
    },
    {
      "epoch": 2.9392242431640624e-06,
      "model_forward_time": 5.6201536655426025,
      "step": 3082
    },
    {
      "epoch": 2.9392242431640624e-06,
      "step": 3082,
      "training_step_time": 17.45720911026001
    },
    {
      "epoch": 2.940177917480469e-06,
      "model_forward_time": 5.594379663467407,
      "step": 3083
    },
    {
      "epoch": 2.940177917480469e-06,
      "step": 3083,
      "training_step_time": 17.019484281539917
    },
    {
      "epoch": 2.941131591796875e-06,
      "model_forward_time": 5.574842929840088,
      "step": 3084
    },
    {
      "epoch": 2.941131591796875e-06,
      "step": 3084,
      "training_step_time": 16.980213403701782
    },
    {
      "epoch": 2.9420852661132812e-06,
      "model_forward_time": 5.573506593704224,
      "step": 3085
    },
    {
      "epoch": 2.9420852661132812e-06,
      "step": 3085,
      "training_step_time": 17.01026487350464
    },
    {
      "epoch": 2.9430389404296874e-06,
      "model_forward_time": 5.567865610122681,
      "step": 3086
    },
    {
      "epoch": 2.9430389404296874e-06,
      "step": 3086,
      "training_step_time": 16.98862051963806
    },
    {
      "epoch": 2.943992614746094e-06,
      "model_forward_time": 5.58562707901001,
      "step": 3087
    },
    {
      "epoch": 2.943992614746094e-06,
      "step": 3087,
      "training_step_time": 16.997055053710938
    },
    {
      "epoch": 2.9449462890625e-06,
      "model_forward_time": 5.97326135635376,
      "step": 3088
    },
    {
      "epoch": 2.9449462890625e-06,
      "step": 3088,
      "training_step_time": 17.408141374588013
    },
    {
      "epoch": 2.9458999633789062e-06,
      "model_forward_time": 5.581650018692017,
      "step": 3089
    },
    {
      "epoch": 2.9458999633789062e-06,
      "step": 3089,
      "training_step_time": 17.02073907852173
    },
    {
      "epoch": 2.9468536376953124e-06,
      "grad_norm": 1.5187333822250366,
      "learning_rate": 1e-05,
      "loss": 0.1046,
      "step": 3090
    },
    {
      "epoch": 2.9468536376953124e-06,
      "model_forward_time": 5.587876319885254,
      "step": 3090
    },
    {
      "dynamics_loss_avg": 0.08383873272687196,
      "epoch": 2.9468536376953124e-06,
      "step": 3090
    },
    {
      "action_loss_avg": 0.02284901519306004,
      "epoch": 2.9468536376953124e-06,
      "step": 3090
    },
    {
      "epoch": 2.9468536376953124e-06,
      "step": 3090,
      "training_step_time": 17.019549131393433
    },
    {
      "epoch": 2.947807312011719e-06,
      "model_forward_time": 5.596467971801758,
      "step": 3091
    },
    {
      "epoch": 2.947807312011719e-06,
      "step": 3091,
      "training_step_time": 17.083744525909424
    },
    {
      "epoch": 2.948760986328125e-06,
      "model_forward_time": 5.62920880317688,
      "step": 3092
    },
    {
      "epoch": 2.948760986328125e-06,
      "step": 3092,
      "training_step_time": 17.15062928199768
    },
    {
      "epoch": 2.9497146606445312e-06,
      "model_forward_time": 5.630530118942261,
      "step": 3093
    },
    {
      "epoch": 2.9497146606445312e-06,
      "step": 3093,
      "training_step_time": 17.510890245437622
    },
    {
      "epoch": 2.9506683349609374e-06,
      "model_forward_time": 5.622823238372803,
      "step": 3094
    },
    {
      "epoch": 2.9506683349609374e-06,
      "step": 3094,
      "training_step_time": 17.12183928489685
    },
    {
      "epoch": 2.951622009277344e-06,
      "model_forward_time": 5.621560335159302,
      "step": 3095
    },
    {
      "epoch": 2.951622009277344e-06,
      "step": 3095,
      "training_step_time": 17.119044303894043
    },
    {
      "epoch": 2.95257568359375e-06,
      "model_forward_time": 5.609521150588989,
      "step": 3096
    },
    {
      "epoch": 2.95257568359375e-06,
      "step": 3096,
      "training_step_time": 17.104081392288208
    },
    {
      "epoch": 2.9535293579101562e-06,
      "model_forward_time": 5.619998455047607,
      "step": 3097
    },
    {
      "epoch": 2.9535293579101562e-06,
      "step": 3097,
      "training_step_time": 17.102554082870483
    },
    {
      "epoch": 2.9544830322265624e-06,
      "model_forward_time": 5.611133098602295,
      "step": 3098
    },
    {
      "epoch": 2.9544830322265624e-06,
      "step": 3098,
      "training_step_time": 17.112028121948242
    },
    {
      "epoch": 2.955436706542969e-06,
      "model_forward_time": 5.965329647064209,
      "step": 3099
    },
    {
      "epoch": 2.955436706542969e-06,
      "step": 3099,
      "training_step_time": 17.443382501602173
    },
    {
      "epoch": 2.956390380859375e-06,
      "grad_norm": 0.2729990482330322,
      "learning_rate": 1e-05,
      "loss": 0.0945,
      "step": 3100
    },
    {
      "epoch": 2.956390380859375e-06,
      "model_forward_time": 5.604507684707642,
      "step": 3100
    },
    {
      "dynamics_loss_avg": 0.07348739691078662,
      "epoch": 2.956390380859375e-06,
      "step": 3100
    },
    {
      "action_loss_avg": 0.020157822966575624,
      "epoch": 2.956390380859375e-06,
      "step": 3100
    },
    {
      "epoch": 2.956390380859375e-06,
      "step": 3100,
      "training_step_time": 17.081374406814575
    },
    {
      "epoch": 2.9573440551757812e-06,
      "model_forward_time": 5.601423978805542,
      "step": 3101
    },
    {
      "epoch": 2.9573440551757812e-06,
      "step": 3101,
      "training_step_time": 17.064167976379395
    },
    {
      "epoch": 2.9582977294921874e-06,
      "model_forward_time": 5.594381809234619,
      "step": 3102
    },
    {
      "epoch": 2.9582977294921874e-06,
      "step": 3102,
      "training_step_time": 17.05042815208435
    },
    {
      "epoch": 2.959251403808594e-06,
      "model_forward_time": 5.586402893066406,
      "step": 3103
    },
    {
      "epoch": 2.959251403808594e-06,
      "step": 3103,
      "training_step_time": 17.02407956123352
    },
    {
      "epoch": 2.960205078125e-06,
      "model_forward_time": 5.95146107673645,
      "step": 3104
    },
    {
      "epoch": 2.960205078125e-06,
      "step": 3104,
      "training_step_time": 17.412298917770386
    },
    {
      "epoch": 2.9611587524414062e-06,
      "model_forward_time": 5.601456642150879,
      "step": 3105
    },
    {
      "epoch": 2.9611587524414062e-06,
      "step": 3105,
      "training_step_time": 17.107991933822632
    },
    {
      "epoch": 2.9621124267578124e-06,
      "model_forward_time": 5.620380163192749,
      "step": 3106
    },
    {
      "epoch": 2.9621124267578124e-06,
      "step": 3106,
      "training_step_time": 17.128516674041748
    },
    {
      "epoch": 2.963066101074219e-06,
      "model_forward_time": 5.618455648422241,
      "step": 3107
    },
    {
      "epoch": 2.963066101074219e-06,
      "step": 3107,
      "training_step_time": 17.11707353591919
    },
    {
      "epoch": 2.964019775390625e-06,
      "model_forward_time": 5.618476629257202,
      "step": 3108
    },
    {
      "epoch": 2.964019775390625e-06,
      "step": 3108,
      "training_step_time": 17.11259961128235
    },
    {
      "epoch": 2.9649734497070312e-06,
      "model_forward_time": 5.617110967636108,
      "step": 3109
    },
    {
      "epoch": 2.9649734497070312e-06,
      "step": 3109,
      "training_step_time": 17.113986015319824
    },
    {
      "epoch": 2.9659271240234374e-06,
      "grad_norm": 0.4212581515312195,
      "learning_rate": 1e-05,
      "loss": 0.091,
      "step": 3110
    },
    {
      "epoch": 2.9659271240234374e-06,
      "model_forward_time": 5.974766254425049,
      "step": 3110
    },
    {
      "dynamics_loss_avg": 0.07014628946781158,
      "epoch": 2.9659271240234374e-06,
      "step": 3110
    },
    {
      "action_loss_avg": 0.02212314261123538,
      "epoch": 2.9659271240234374e-06,
      "step": 3110
    },
    {
      "epoch": 2.9659271240234374e-06,
      "step": 3110,
      "training_step_time": 17.4615375995636
    },
    {
      "epoch": 2.966880798339844e-06,
      "model_forward_time": 5.613454580307007,
      "step": 3111
    },
    {
      "epoch": 2.966880798339844e-06,
      "step": 3111,
      "training_step_time": 17.090083837509155
    },
    {
      "epoch": 2.96783447265625e-06,
      "model_forward_time": 5.606254816055298,
      "step": 3112
    },
    {
      "epoch": 2.96783447265625e-06,
      "step": 3112,
      "training_step_time": 17.06813073158264
    },
    {
      "epoch": 2.9687881469726562e-06,
      "model_forward_time": 5.588169097900391,
      "step": 3113
    },
    {
      "epoch": 2.9687881469726562e-06,
      "step": 3113,
      "training_step_time": 17.042685747146606
    },
    {
      "epoch": 2.9697418212890624e-06,
      "model_forward_time": 5.589253902435303,
      "step": 3114
    },
    {
      "epoch": 2.9697418212890624e-06,
      "step": 3114,
      "training_step_time": 17.026665687561035
    },
    {
      "epoch": 2.970695495605469e-06,
      "model_forward_time": 5.596808910369873,
      "step": 3115
    },
    {
      "epoch": 2.970695495605469e-06,
      "step": 3115,
      "training_step_time": 17.39132332801819
    },
    {
      "epoch": 2.971649169921875e-06,
      "model_forward_time": 5.5879807472229,
      "step": 3116
    },
    {
      "epoch": 2.971649169921875e-06,
      "step": 3116,
      "training_step_time": 17.036476135253906
    },
    {
      "epoch": 2.9726028442382812e-06,
      "model_forward_time": 5.59497332572937,
      "step": 3117
    },
    {
      "epoch": 2.9726028442382812e-06,
      "step": 3117,
      "training_step_time": 17.046852350234985
    },
    {
      "epoch": 2.9735565185546874e-06,
      "model_forward_time": 5.6151204109191895,
      "step": 3118
    },
    {
      "epoch": 2.9735565185546874e-06,
      "step": 3118,
      "training_step_time": 17.076743364334106
    },
    {
      "epoch": 2.974510192871094e-06,
      "model_forward_time": 5.597572088241577,
      "step": 3119
    },
    {
      "epoch": 2.974510192871094e-06,
      "step": 3119,
      "training_step_time": 17.039225339889526
    },
    {
      "epoch": 2.9754638671875e-06,
      "grad_norm": 1.2500673532485962,
      "learning_rate": 1e-05,
      "loss": 0.1015,
      "step": 3120
    },
    {
      "epoch": 2.9754638671875e-06,
      "model_forward_time": 5.589081764221191,
      "step": 3120
    },
    {
      "dynamics_loss_avg": 0.07970191165804863,
      "epoch": 2.9754638671875e-06,
      "step": 3120
    },
    {
      "action_loss_avg": 0.026018880307674408,
      "epoch": 2.9754638671875e-06,
      "step": 3120
    },
    {
      "epoch": 2.9754638671875e-06,
      "step": 3120,
      "training_step_time": 17.043352842330933
    },
    {
      "epoch": 2.9764175415039062e-06,
      "model_forward_time": 5.938843488693237,
      "step": 3121
    },
    {
      "epoch": 2.9764175415039062e-06,
      "step": 3121,
      "training_step_time": 17.37508511543274
    },
    {
      "epoch": 2.9773712158203124e-06,
      "model_forward_time": 5.592352628707886,
      "step": 3122
    },
    {
      "epoch": 2.9773712158203124e-06,
      "step": 3122,
      "training_step_time": 17.0658016204834
    },
    {
      "epoch": 2.978324890136719e-06,
      "model_forward_time": 5.601452589035034,
      "step": 3123
    },
    {
      "epoch": 2.978324890136719e-06,
      "step": 3123,
      "training_step_time": 17.0785813331604
    },
    {
      "epoch": 2.979278564453125e-06,
      "model_forward_time": 5.623946666717529,
      "step": 3124
    },
    {
      "epoch": 2.979278564453125e-06,
      "step": 3124,
      "training_step_time": 17.120656490325928
    },
    {
      "epoch": 2.9802322387695312e-06,
      "model_forward_time": 5.628848314285278,
      "step": 3125
    },
    {
      "epoch": 2.9802322387695312e-06,
      "step": 3125,
      "training_step_time": 17.150694131851196
    },
    {
      "epoch": 2.9811859130859374e-06,
      "model_forward_time": 5.62247109413147,
      "step": 3126
    },
    {
      "epoch": 2.9811859130859374e-06,
      "step": 3126,
      "training_step_time": 17.477822065353394
    },
    {
      "epoch": 2.9821395874023435e-06,
      "model_forward_time": 5.604100465774536,
      "step": 3127
    },
    {
      "epoch": 2.9821395874023435e-06,
      "step": 3127,
      "training_step_time": 17.07141900062561
    },
    {
      "epoch": 2.98309326171875e-06,
      "model_forward_time": 5.607431888580322,
      "step": 3128
    },
    {
      "epoch": 2.98309326171875e-06,
      "step": 3128,
      "training_step_time": 17.060811042785645
    },
    {
      "epoch": 2.9840469360351563e-06,
      "model_forward_time": 5.588139772415161,
      "step": 3129
    },
    {
      "epoch": 2.9840469360351563e-06,
      "step": 3129,
      "training_step_time": 17.018421173095703
    },
    {
      "epoch": 2.9850006103515624e-06,
      "grad_norm": 1.2039545774459839,
      "learning_rate": 1e-05,
      "loss": 0.0919,
      "step": 3130
    },
    {
      "epoch": 2.9850006103515624e-06,
      "model_forward_time": 5.585412263870239,
      "step": 3130
    },
    {
      "dynamics_loss_avg": 0.06708725988864898,
      "epoch": 2.9850006103515624e-06,
      "step": 3130
    },
    {
      "action_loss_avg": 0.021718240482732652,
      "epoch": 2.9850006103515624e-06,
      "step": 3130
    },
    {
      "epoch": 2.9850006103515624e-06,
      "step": 3130,
      "training_step_time": 17.02170467376709
    },
    {
      "epoch": 2.9859542846679685e-06,
      "model_forward_time": 5.586240768432617,
      "step": 3131
    },
    {
      "epoch": 2.9859542846679685e-06,
      "step": 3131,
      "training_step_time": 17.0392005443573
    },
    {
      "epoch": 2.986907958984375e-06,
      "model_forward_time": 5.972160816192627,
      "step": 3132
    },
    {
      "epoch": 2.986907958984375e-06,
      "step": 3132,
      "training_step_time": 17.45328974723816
    },
    {
      "epoch": 2.9878616333007813e-06,
      "model_forward_time": 5.601048231124878,
      "step": 3133
    },
    {
      "epoch": 2.9878616333007813e-06,
      "step": 3133,
      "training_step_time": 17.085214376449585
    },
    {
      "epoch": 2.9888153076171874e-06,
      "model_forward_time": 5.611879825592041,
      "step": 3134
    },
    {
      "epoch": 2.9888153076171874e-06,
      "step": 3134,
      "training_step_time": 17.10515284538269
    },
    {
      "epoch": 2.9897689819335935e-06,
      "model_forward_time": 5.60994291305542,
      "step": 3135
    },
    {
      "epoch": 2.9897689819335935e-06,
      "step": 3135,
      "training_step_time": 17.082074880599976
    },
    {
      "epoch": 2.99072265625e-06,
      "model_forward_time": 5.604953289031982,
      "step": 3136
    },
    {
      "epoch": 2.99072265625e-06,
      "step": 3136,
      "training_step_time": 17.08827495574951
    },
    {
      "epoch": 2.9916763305664063e-06,
      "model_forward_time": 5.964464902877808,
      "step": 3137
    },
    {
      "epoch": 2.9916763305664063e-06,
      "step": 3137,
      "training_step_time": 17.439225912094116
    },
    {
      "epoch": 2.9926300048828124e-06,
      "model_forward_time": 5.6123809814453125,
      "step": 3138
    },
    {
      "epoch": 2.9926300048828124e-06,
      "step": 3138,
      "training_step_time": 17.093331575393677
    },
    {
      "epoch": 2.9935836791992186e-06,
      "model_forward_time": 5.602736711502075,
      "step": 3139
    },
    {
      "epoch": 2.9935836791992186e-06,
      "step": 3139,
      "training_step_time": 17.049404859542847
    },
    {
      "epoch": 2.994537353515625e-06,
      "grad_norm": 1.2546521425247192,
      "learning_rate": 1e-05,
      "loss": 0.0726,
      "step": 3140
    },
    {
      "epoch": 2.994537353515625e-06,
      "model_forward_time": 5.599249839782715,
      "step": 3140
    },
    {
      "dynamics_loss_avg": 0.05303402151912451,
      "epoch": 2.994537353515625e-06,
      "step": 3140
    },
    {
      "action_loss_avg": 0.013639612484257668,
      "epoch": 2.994537353515625e-06,
      "step": 3140
    },
    {
      "epoch": 2.994537353515625e-06,
      "step": 3140,
      "training_step_time": 17.070879459381104
    },
    {
      "epoch": 2.9954910278320313e-06,
      "model_forward_time": 5.588553190231323,
      "step": 3141
    },
    {
      "epoch": 2.9954910278320313e-06,
      "step": 3141,
      "training_step_time": 17.026678323745728
    },
    {
      "epoch": 2.9964447021484374e-06,
      "model_forward_time": 5.585789442062378,
      "step": 3142
    },
    {
      "epoch": 2.9964447021484374e-06,
      "step": 3142,
      "training_step_time": 17.023757457733154
    },
    {
      "epoch": 2.9973983764648436e-06,
      "model_forward_time": 5.957133769989014,
      "step": 3143
    },
    {
      "epoch": 2.9973983764648436e-06,
      "step": 3143,
      "training_step_time": 17.423073768615723
    },
    {
      "epoch": 2.99835205078125e-06,
      "model_forward_time": 5.607730865478516,
      "step": 3144
    },
    {
      "epoch": 2.99835205078125e-06,
      "step": 3144,
      "training_step_time": 17.09632682800293
    },
    {
      "epoch": 2.9993057250976563e-06,
      "model_forward_time": 5.604584217071533,
      "step": 3145
    },
    {
      "epoch": 2.9993057250976563e-06,
      "step": 3145,
      "training_step_time": 17.096592903137207
    },
    {
      "epoch": 3.0002593994140624e-06,
      "model_forward_time": 5.603580713272095,
      "step": 3146
    },
    {
      "epoch": 3.0002593994140624e-06,
      "step": 3146,
      "training_step_time": 17.071247816085815
    },
    {
      "epoch": 3.0012130737304686e-06,
      "model_forward_time": 5.58268928527832,
      "step": 3147
    },
    {
      "epoch": 3.0012130737304686e-06,
      "step": 3147,
      "training_step_time": 17.00703191757202
    },
    {
      "epoch": 3.002166748046875e-06,
      "model_forward_time": 5.580778121948242,
      "step": 3148
    },
    {
      "epoch": 3.002166748046875e-06,
      "step": 3148,
      "training_step_time": 17.34791851043701
    },
    {
      "epoch": 3.0031204223632813e-06,
      "model_forward_time": 5.585095167160034,
      "step": 3149
    },
    {
      "epoch": 3.0031204223632813e-06,
      "step": 3149,
      "training_step_time": 17.004725694656372
    },
    {
      "epoch": 3.0040740966796874e-06,
      "grad_norm": 1.5588934421539307,
      "learning_rate": 1e-05,
      "loss": 0.0761,
      "step": 3150
    },
    {
      "epoch": 3.0040740966796874e-06,
      "model_forward_time": 5.582712888717651,
      "step": 3150
    },
    {
      "dynamics_loss_avg": 0.061910825036466124,
      "epoch": 3.0040740966796874e-06,
      "step": 3150
    },
    {
      "action_loss_avg": 0.01783652459271252,
      "epoch": 3.0040740966796874e-06,
      "step": 3150
    },
    {
      "epoch": 3.0040740966796874e-06,
      "step": 3150,
      "training_step_time": 17.026775360107422
    },
    {
      "epoch": 3.0050277709960936e-06,
      "model_forward_time": 5.573720932006836,
      "step": 3151
    },
    {
      "epoch": 3.0050277709960936e-06,
      "step": 3151,
      "training_step_time": 17.004826307296753
    },
    {
      "epoch": 3.0059814453125e-06,
      "model_forward_time": 5.595083236694336,
      "step": 3152
    },
    {
      "epoch": 3.0059814453125e-06,
      "step": 3152,
      "training_step_time": 17.028700828552246
    },
    {
      "epoch": 3.0069351196289063e-06,
      "model_forward_time": 5.586479425430298,
      "step": 3153
    },
    {
      "epoch": 3.0069351196289063e-06,
      "step": 3153,
      "training_step_time": 17.021760940551758
    },
    {
      "epoch": 3.0078887939453124e-06,
      "model_forward_time": 5.979719400405884,
      "step": 3154
    },
    {
      "epoch": 3.0078887939453124e-06,
      "step": 3154,
      "training_step_time": 17.438576459884644
    },
    {
      "epoch": 3.0088424682617186e-06,
      "model_forward_time": 5.6076061725616455,
      "step": 3155
    },
    {
      "epoch": 3.0088424682617186e-06,
      "step": 3155,
      "training_step_time": 17.12084650993347
    },
    {
      "epoch": 3.009796142578125e-06,
      "model_forward_time": 5.619763374328613,
      "step": 3156
    },
    {
      "epoch": 3.009796142578125e-06,
      "step": 3156,
      "training_step_time": 17.115408182144165
    },
    {
      "epoch": 3.0107498168945313e-06,
      "model_forward_time": 5.601520538330078,
      "step": 3157
    },
    {
      "epoch": 3.0107498168945313e-06,
      "step": 3157,
      "training_step_time": 17.047847032546997
    },
    {
      "epoch": 3.0117034912109374e-06,
      "model_forward_time": 5.595278024673462,
      "step": 3158
    },
    {
      "epoch": 3.0117034912109374e-06,
      "step": 3158,
      "training_step_time": 17.02367901802063
    },
    {
      "epoch": 3.0126571655273436e-06,
      "model_forward_time": 5.587118625640869,
      "step": 3159
    },
    {
      "epoch": 3.0126571655273436e-06,
      "step": 3159,
      "training_step_time": 17.398432970046997
    },
    {
      "epoch": 3.01361083984375e-06,
      "grad_norm": 0.42942070960998535,
      "learning_rate": 1e-05,
      "loss": 0.0993,
      "step": 3160
    },
    {
      "epoch": 3.01361083984375e-06,
      "model_forward_time": 5.60447096824646,
      "step": 3160
    },
    {
      "dynamics_loss_avg": 0.07983470503240823,
      "epoch": 3.01361083984375e-06,
      "step": 3160
    },
    {
      "action_loss_avg": 0.018922333675436675,
      "epoch": 3.01361083984375e-06,
      "step": 3160
    },
    {
      "epoch": 3.01361083984375e-06,
      "step": 3160,
      "training_step_time": 17.089494466781616
    },
    {
      "epoch": 3.0145645141601563e-06,
      "model_forward_time": 5.60584020614624,
      "step": 3161
    },
    {
      "epoch": 3.0145645141601563e-06,
      "step": 3161,
      "training_step_time": 17.09034514427185
    },
    {
      "epoch": 3.0155181884765624e-06,
      "model_forward_time": 5.614813566207886,
      "step": 3162
    },
    {
      "epoch": 3.0155181884765624e-06,
      "step": 3162,
      "training_step_time": 17.100036144256592
    },
    {
      "epoch": 3.0164718627929686e-06,
      "model_forward_time": 5.597303867340088,
      "step": 3163
    },
    {
      "epoch": 3.0164718627929686e-06,
      "step": 3163,
      "training_step_time": 17.029704809188843
    },
    {
      "epoch": 3.017425537109375e-06,
      "model_forward_time": 5.5794432163238525,
      "step": 3164
    },
    {
      "epoch": 3.017425537109375e-06,
      "step": 3164,
      "training_step_time": 16.99748706817627
    },
    {
      "epoch": 3.0183792114257813e-06,
      "model_forward_time": 5.9613776206970215,
      "step": 3165
    },
    {
      "epoch": 3.0183792114257813e-06,
      "step": 3165,
      "training_step_time": 17.39969754219055
    },
    {
      "epoch": 3.0193328857421874e-06,
      "model_forward_time": 5.582706689834595,
      "step": 3166
    },
    {
      "epoch": 3.0193328857421874e-06,
      "step": 3166,
      "training_step_time": 17.028449296951294
    },
    {
      "epoch": 3.0202865600585936e-06,
      "model_forward_time": 5.5916748046875,
      "step": 3167
    },
    {
      "epoch": 3.0202865600585936e-06,
      "step": 3167,
      "training_step_time": 17.029104709625244
    },
    {
      "epoch": 3.021240234375e-06,
      "model_forward_time": 5.591371774673462,
      "step": 3168
    },
    {
      "epoch": 3.021240234375e-06,
      "step": 3168,
      "training_step_time": 17.052857398986816
    },
    {
      "epoch": 3.0221939086914063e-06,
      "model_forward_time": 5.602040767669678,
      "step": 3169
    },
    {
      "epoch": 3.0221939086914063e-06,
      "step": 3169,
      "training_step_time": 17.06320309638977
    },
    {
      "epoch": 3.0231475830078124e-06,
      "grad_norm": 0.7707942128181458,
      "learning_rate": 1e-05,
      "loss": 0.0847,
      "step": 3170
    },
    {
      "epoch": 3.0231475830078124e-06,
      "model_forward_time": 5.978082656860352,
      "step": 3170
    },
    {
      "dynamics_loss_avg": 0.0646067576482892,
      "epoch": 3.0231475830078124e-06,
      "step": 3170
    },
    {
      "action_loss_avg": 0.015527631156146526,
      "epoch": 3.0231475830078124e-06,
      "step": 3170
    },
    {
      "epoch": 3.0231475830078124e-06,
      "step": 3170,
      "training_step_time": 17.47079849243164
    },
    {
      "epoch": 3.0241012573242186e-06,
      "model_forward_time": 5.605952501296997,
      "step": 3171
    },
    {
      "epoch": 3.0241012573242186e-06,
      "step": 3171,
      "training_step_time": 17.100489377975464
    },
    {
      "epoch": 3.025054931640625e-06,
      "model_forward_time": 5.606152772903442,
      "step": 3172
    },
    {
      "epoch": 3.025054931640625e-06,
      "step": 3172,
      "training_step_time": 17.083996534347534
    },
    {
      "epoch": 3.0260086059570313e-06,
      "model_forward_time": 5.609652519226074,
      "step": 3173
    },
    {
      "epoch": 3.0260086059570313e-06,
      "step": 3173,
      "training_step_time": 17.09492540359497
    },
    {
      "epoch": 3.0269622802734374e-06,
      "model_forward_time": 5.6137611865997314,
      "step": 3174
    },
    {
      "epoch": 3.0269622802734374e-06,
      "step": 3174,
      "training_step_time": 17.11194610595703
    },
    {
      "epoch": 3.0279159545898436e-06,
      "model_forward_time": 5.611971855163574,
      "step": 3175
    },
    {
      "epoch": 3.0279159545898436e-06,
      "step": 3175,
      "training_step_time": 17.09703040122986
    },
    {
      "epoch": 3.02886962890625e-06,
      "model_forward_time": 5.946775913238525,
      "step": 3176
    },
    {
      "epoch": 3.02886962890625e-06,
      "step": 3176,
      "training_step_time": 17.36788845062256
    },
    {
      "epoch": 3.0298233032226563e-06,
      "model_forward_time": 5.572584629058838,
      "step": 3177
    },
    {
      "epoch": 3.0298233032226563e-06,
      "step": 3177,
      "training_step_time": 17.003904104232788
    },
    {
      "epoch": 3.0307769775390624e-06,
      "model_forward_time": 5.571248292922974,
      "step": 3178
    },
    {
      "epoch": 3.0307769775390624e-06,
      "step": 3178,
      "training_step_time": 16.988964080810547
    },
    {
      "epoch": 3.0317306518554686e-06,
      "model_forward_time": 5.568532466888428,
      "step": 3179
    },
    {
      "epoch": 3.0317306518554686e-06,
      "step": 3179,
      "training_step_time": 16.984046697616577
    },
    {
      "epoch": 3.032684326171875e-06,
      "grad_norm": 1.6378719806671143,
      "learning_rate": 1e-05,
      "loss": 0.0711,
      "step": 3180
    },
    {
      "epoch": 3.032684326171875e-06,
      "model_forward_time": 5.574169874191284,
      "step": 3180
    },
    {
      "dynamics_loss_avg": 0.05793289374560118,
      "epoch": 3.032684326171875e-06,
      "step": 3180
    },
    {
      "action_loss_avg": 0.01592048080638051,
      "epoch": 3.032684326171875e-06,
      "step": 3180
    },
    {
      "epoch": 3.032684326171875e-06,
      "step": 3180,
      "training_step_time": 17.007532119750977
    },
    {
      "epoch": 3.0336380004882813e-06,
      "model_forward_time": 5.572058916091919,
      "step": 3181
    },
    {
      "epoch": 3.0336380004882813e-06,
      "step": 3181,
      "training_step_time": 17.371940851211548
    },
    {
      "epoch": 3.0345916748046874e-06,
      "model_forward_time": 5.589730978012085,
      "step": 3182
    },
    {
      "epoch": 3.0345916748046874e-06,
      "step": 3182,
      "training_step_time": 17.021921157836914
    },
    {
      "epoch": 3.0355453491210936e-06,
      "model_forward_time": 5.590062856674194,
      "step": 3183
    },
    {
      "epoch": 3.0355453491210936e-06,
      "step": 3183,
      "training_step_time": 16.997058868408203
    },
    {
      "epoch": 3.0364990234375e-06,
      "model_forward_time": 5.5825865268707275,
      "step": 3184
    },
    {
      "epoch": 3.0364990234375e-06,
      "step": 3184,
      "training_step_time": 17.033032178878784
    },
    {
      "epoch": 3.0374526977539063e-06,
      "model_forward_time": 5.600109815597534,
      "step": 3185
    },
    {
      "epoch": 3.0374526977539063e-06,
      "step": 3185,
      "training_step_time": 17.075127601623535
    },
    {
      "epoch": 3.0384063720703124e-06,
      "model_forward_time": 5.61506724357605,
      "step": 3186
    },
    {
      "epoch": 3.0384063720703124e-06,
      "step": 3186,
      "training_step_time": 17.08966064453125
    },
    {
      "epoch": 3.0393600463867186e-06,
      "model_forward_time": 5.969427585601807,
      "step": 3187
    },
    {
      "epoch": 3.0393600463867186e-06,
      "step": 3187,
      "training_step_time": 17.429181575775146
    },
    {
      "epoch": 3.040313720703125e-06,
      "model_forward_time": 5.600198745727539,
      "step": 3188
    },
    {
      "epoch": 3.040313720703125e-06,
      "step": 3188,
      "training_step_time": 17.037938833236694
    },
    {
      "epoch": 3.0412673950195313e-06,
      "model_forward_time": 5.591518402099609,
      "step": 3189
    },
    {
      "epoch": 3.0412673950195313e-06,
      "step": 3189,
      "training_step_time": 17.032936573028564
    },
    {
      "epoch": 3.0422210693359374e-06,
      "grad_norm": 0.9067350029945374,
      "learning_rate": 1e-05,
      "loss": 0.0615,
      "step": 3190
    },
    {
      "epoch": 3.0422210693359374e-06,
      "model_forward_time": 5.590033769607544,
      "step": 3190
    },
    {
      "dynamics_loss_avg": 0.049363061971962455,
      "epoch": 3.0422210693359374e-06,
      "step": 3190
    },
    {
      "action_loss_avg": 0.012592043448239564,
      "epoch": 3.0422210693359374e-06,
      "step": 3190
    },
    {
      "epoch": 3.0422210693359374e-06,
      "step": 3190,
      "training_step_time": 17.033278465270996
    },
    {
      "epoch": 3.0431747436523436e-06,
      "model_forward_time": 5.596735239028931,
      "step": 3191
    },
    {
      "epoch": 3.0431747436523436e-06,
      "step": 3191,
      "training_step_time": 17.046261310577393
    },
    {
      "epoch": 3.04412841796875e-06,
      "model_forward_time": 5.606090068817139,
      "step": 3192
    },
    {
      "epoch": 3.04412841796875e-06,
      "step": 3192,
      "training_step_time": 17.45208501815796
    },
    {
      "epoch": 3.0450820922851563e-06,
      "model_forward_time": 5.642459392547607,
      "step": 3193
    },
    {
      "epoch": 3.0450820922851563e-06,
      "step": 3193,
      "training_step_time": 17.173730611801147
    },
    {
      "epoch": 3.0460357666015625e-06,
      "model_forward_time": 5.626376152038574,
      "step": 3194
    },
    {
      "epoch": 3.0460357666015625e-06,
      "step": 3194,
      "training_step_time": 17.13733696937561
    },
    {
      "epoch": 3.0469894409179686e-06,
      "model_forward_time": 5.62015962600708,
      "step": 3195
    },
    {
      "epoch": 3.0469894409179686e-06,
      "step": 3195,
      "training_step_time": 17.09575843811035
    },
    {
      "epoch": 3.047943115234375e-06,
      "model_forward_time": 5.610986709594727,
      "step": 3196
    },
    {
      "epoch": 3.047943115234375e-06,
      "step": 3196,
      "training_step_time": 17.080656051635742
    },
    {
      "epoch": 3.0488967895507813e-06,
      "model_forward_time": 5.607289791107178,
      "step": 3197
    },
    {
      "epoch": 3.0488967895507813e-06,
      "step": 3197,
      "training_step_time": 17.081276893615723
    },
    {
      "epoch": 3.0498504638671875e-06,
      "model_forward_time": 5.982971429824829,
      "step": 3198
    },
    {
      "epoch": 3.0498504638671875e-06,
      "step": 3198,
      "training_step_time": 17.444671392440796
    },
    {
      "epoch": 3.0508041381835936e-06,
      "model_forward_time": 5.59568452835083,
      "step": 3199
    },
    {
      "epoch": 3.0508041381835936e-06,
      "step": 3199,
      "training_step_time": 17.03503966331482
    },
    {
      "epoch": 3.0517578125e-06,
      "grad_norm": 1.2030678987503052,
      "learning_rate": 1e-05,
      "loss": 0.0872,
      "step": 3200
    },
    {
      "epoch": 3.0517578125e-06,
      "model_forward_time": 5.596566677093506,
      "step": 3200
    },
    {
      "dynamics_loss_avg": 0.06411709282547236,
      "epoch": 3.0517578125e-06,
      "step": 3200
    },
    {
      "action_loss_avg": 0.020869740657508372,
      "epoch": 3.0517578125e-06,
      "step": 3200
    },
    {
      "epoch": 3.0517578125e-06,
      "step": 3200,
      "training_step_time": 17.032966136932373
    },
    {
      "epoch": 3.0527114868164063e-06,
      "model_forward_time": 5.59692120552063,
      "step": 3201
    },
    {
      "epoch": 3.0527114868164063e-06,
      "step": 3201,
      "training_step_time": 17.07403063774109
    },
    {
      "epoch": 3.0536651611328125e-06,
      "model_forward_time": 5.602099657058716,
      "step": 3202
    },
    {
      "epoch": 3.0536651611328125e-06,
      "step": 3202,
      "training_step_time": 17.09037494659424
    },
    {
      "epoch": 3.0546188354492186e-06,
      "model_forward_time": 5.986742734909058,
      "step": 3203
    },
    {
      "epoch": 3.0546188354492186e-06,
      "step": 3203,
      "training_step_time": 17.506842136383057
    },
    {
      "epoch": 3.055572509765625e-06,
      "model_forward_time": 5.629178047180176,
      "step": 3204
    },
    {
      "epoch": 3.055572509765625e-06,
      "step": 3204,
      "training_step_time": 17.139488697052002
    },
    {
      "epoch": 3.0565261840820313e-06,
      "model_forward_time": 5.600327730178833,
      "step": 3205
    },
    {
      "epoch": 3.0565261840820313e-06,
      "step": 3205,
      "training_step_time": 17.062098264694214
    },
    {
      "epoch": 3.0574798583984375e-06,
      "model_forward_time": 5.58889102935791,
      "step": 3206
    },
    {
      "epoch": 3.0574798583984375e-06,
      "step": 3206,
      "training_step_time": 17.021647691726685
    },
    {
      "epoch": 3.0584335327148436e-06,
      "model_forward_time": 5.587232351303101,
      "step": 3207
    },
    {
      "epoch": 3.0584335327148436e-06,
      "step": 3207,
      "training_step_time": 17.023324012756348
    },
    {
      "epoch": 3.05938720703125e-06,
      "model_forward_time": 5.591023206710815,
      "step": 3208
    },
    {
      "epoch": 3.05938720703125e-06,
      "step": 3208,
      "training_step_time": 17.07151770591736
    },
    {
      "epoch": 3.0603408813476563e-06,
      "model_forward_time": 5.97886323928833,
      "step": 3209
    },
    {
      "epoch": 3.0603408813476563e-06,
      "step": 3209,
      "training_step_time": 17.48849892616272
    },
    {
      "epoch": 3.0612945556640625e-06,
      "grad_norm": 1.381842017173767,
      "learning_rate": 1e-05,
      "loss": 0.058,
      "step": 3210
    },
    {
      "epoch": 3.0612945556640625e-06,
      "model_forward_time": 5.629655838012695,
      "step": 3210
    },
    {
      "dynamics_loss_avg": 0.050060960464179516,
      "epoch": 3.0612945556640625e-06,
      "step": 3210
    },
    {
      "action_loss_avg": 0.013492909434717149,
      "epoch": 3.0612945556640625e-06,
      "step": 3210
    },
    {
      "epoch": 3.0612945556640625e-06,
      "step": 3210,
      "training_step_time": 17.12695002555847
    },
    {
      "epoch": 3.0622482299804686e-06,
      "model_forward_time": 5.610170602798462,
      "step": 3211
    },
    {
      "epoch": 3.0622482299804686e-06,
      "step": 3211,
      "training_step_time": 17.100653171539307
    },
    {
      "epoch": 3.063201904296875e-06,
      "model_forward_time": 5.609511137008667,
      "step": 3212
    },
    {
      "epoch": 3.063201904296875e-06,
      "step": 3212,
      "training_step_time": 17.08922266960144
    },
    {
      "epoch": 3.0641555786132813e-06,
      "model_forward_time": 5.603844165802002,
      "step": 3213
    },
    {
      "epoch": 3.0641555786132813e-06,
      "step": 3213,
      "training_step_time": 17.074549436569214
    },
    {
      "epoch": 3.0651092529296875e-06,
      "model_forward_time": 5.600534677505493,
      "step": 3214
    },
    {
      "epoch": 3.0651092529296875e-06,
      "step": 3214,
      "training_step_time": 17.455625534057617
    },
    {
      "epoch": 3.0660629272460936e-06,
      "model_forward_time": 5.601460695266724,
      "step": 3215
    },
    {
      "epoch": 3.0660629272460936e-06,
      "step": 3215,
      "training_step_time": 17.056103467941284
    },
    {
      "epoch": 3.0670166015625e-06,
      "model_forward_time": 5.603485822677612,
      "step": 3216
    },
    {
      "epoch": 3.0670166015625e-06,
      "step": 3216,
      "training_step_time": 17.06668734550476
    },
    {
      "epoch": 3.0679702758789063e-06,
      "model_forward_time": 5.596102476119995,
      "step": 3217
    },
    {
      "epoch": 3.0679702758789063e-06,
      "step": 3217,
      "training_step_time": 17.06121253967285
    },
    {
      "epoch": 3.0689239501953125e-06,
      "model_forward_time": 5.589234113693237,
      "step": 3218
    },
    {
      "epoch": 3.0689239501953125e-06,
      "step": 3218,
      "training_step_time": 17.040687322616577
    },
    {
      "epoch": 3.0698776245117186e-06,
      "model_forward_time": 5.614229679107666,
      "step": 3219
    },
    {
      "epoch": 3.0698776245117186e-06,
      "step": 3219,
      "training_step_time": 17.06755566596985
    },
    {
      "epoch": 3.070831298828125e-06,
      "grad_norm": 0.6810414791107178,
      "learning_rate": 1e-05,
      "loss": 0.0728,
      "step": 3220
    },
    {
      "epoch": 3.070831298828125e-06,
      "model_forward_time": 5.965754270553589,
      "step": 3220
    },
    {
      "dynamics_loss_avg": 0.05951081607490778,
      "epoch": 3.070831298828125e-06,
      "step": 3220
    },
    {
      "action_loss_avg": 0.015293584344908595,
      "epoch": 3.070831298828125e-06,
      "step": 3220
    },
    {
      "epoch": 3.070831298828125e-06,
      "step": 3220,
      "training_step_time": 17.42866325378418
    },
    {
      "epoch": 3.0717849731445313e-06,
      "model_forward_time": 5.601401090621948,
      "step": 3221
    },
    {
      "epoch": 3.0717849731445313e-06,
      "step": 3221,
      "training_step_time": 17.081507444381714
    },
    {
      "epoch": 3.0727386474609375e-06,
      "model_forward_time": 5.605248212814331,
      "step": 3222
    },
    {
      "epoch": 3.0727386474609375e-06,
      "step": 3222,
      "training_step_time": 17.075366735458374
    },
    {
      "epoch": 3.0736923217773436e-06,
      "model_forward_time": 5.604033708572388,
      "step": 3223
    },
    {
      "epoch": 3.0736923217773436e-06,
      "step": 3223,
      "training_step_time": 17.084283351898193
    },
    {
      "epoch": 3.07464599609375e-06,
      "model_forward_time": 5.595615863800049,
      "step": 3224
    },
    {
      "epoch": 3.07464599609375e-06,
      "step": 3224,
      "training_step_time": 17.050902605056763
    },
    {
      "epoch": 3.0755996704101563e-06,
      "model_forward_time": 5.598579168319702,
      "step": 3225
    },
    {
      "epoch": 3.0755996704101563e-06,
      "step": 3225,
      "training_step_time": 17.41072940826416
    },
    {
      "epoch": 3.0765533447265625e-06,
      "model_forward_time": 5.59703803062439,
      "step": 3226
    },
    {
      "epoch": 3.0765533447265625e-06,
      "step": 3226,
      "training_step_time": 17.055097579956055
    },
    {
      "epoch": 3.0775070190429686e-06,
      "model_forward_time": 5.596601963043213,
      "step": 3227
    },
    {
      "epoch": 3.0775070190429686e-06,
      "step": 3227,
      "training_step_time": 17.076456546783447
    },
    {
      "epoch": 3.078460693359375e-06,
      "model_forward_time": 5.607391595840454,
      "step": 3228
    },
    {
      "epoch": 3.078460693359375e-06,
      "step": 3228,
      "training_step_time": 17.090437650680542
    },
    {
      "epoch": 3.0794143676757813e-06,
      "model_forward_time": 5.605638027191162,
      "step": 3229
    },
    {
      "epoch": 3.0794143676757813e-06,
      "step": 3229,
      "training_step_time": 17.074338912963867
    },
    {
      "epoch": 3.0803680419921875e-06,
      "grad_norm": 0.8358275890350342,
      "learning_rate": 1e-05,
      "loss": 0.0769,
      "step": 3230
    },
    {
      "epoch": 3.0803680419921875e-06,
      "model_forward_time": 5.610452890396118,
      "step": 3230
    },
    {
      "dynamics_loss_avg": 0.05453513190150261,
      "epoch": 3.0803680419921875e-06,
      "step": 3230
    },
    {
      "action_loss_avg": 0.016736174281686545,
      "epoch": 3.0803680419921875e-06,
      "step": 3230
    },
    {
      "epoch": 3.0803680419921875e-06,
      "step": 3230,
      "training_step_time": 17.090402603149414
    },
    {
      "epoch": 3.0813217163085936e-06,
      "model_forward_time": 5.963189125061035,
      "step": 3231
    },
    {
      "epoch": 3.0813217163085936e-06,
      "step": 3231,
      "training_step_time": 17.44587779045105
    },
    {
      "epoch": 3.082275390625e-06,
      "model_forward_time": 5.6027607917785645,
      "step": 3232
    },
    {
      "epoch": 3.082275390625e-06,
      "step": 3232,
      "training_step_time": 17.083139657974243
    },
    {
      "epoch": 3.0832290649414063e-06,
      "model_forward_time": 5.6045918464660645,
      "step": 3233
    },
    {
      "epoch": 3.0832290649414063e-06,
      "step": 3233,
      "training_step_time": 17.078834533691406
    },
    {
      "epoch": 3.0841827392578125e-06,
      "model_forward_time": 5.599002838134766,
      "step": 3234
    },
    {
      "epoch": 3.0841827392578125e-06,
      "step": 3234,
      "training_step_time": 17.053012371063232
    },
    {
      "epoch": 3.0851364135742186e-06,
      "model_forward_time": 5.586793661117554,
      "step": 3235
    },
    {
      "epoch": 3.0851364135742186e-06,
      "step": 3235,
      "training_step_time": 17.026805877685547
    },
    {
      "epoch": 3.086090087890625e-06,
      "model_forward_time": 5.942636251449585,
      "step": 3236
    },
    {
      "epoch": 3.086090087890625e-06,
      "step": 3236,
      "training_step_time": 17.39323139190674
    },
    {
      "epoch": 3.0870437622070313e-06,
      "model_forward_time": 5.604412794113159,
      "step": 3237
    },
    {
      "epoch": 3.0870437622070313e-06,
      "step": 3237,
      "training_step_time": 17.04266667366028
    },
    {
      "epoch": 3.0879974365234375e-06,
      "model_forward_time": 5.5928566455841064,
      "step": 3238
    },
    {
      "epoch": 3.0879974365234375e-06,
      "step": 3238,
      "training_step_time": 17.051383018493652
    },
    {
      "epoch": 3.0889511108398436e-06,
      "model_forward_time": 5.608161687850952,
      "step": 3239
    },
    {
      "epoch": 3.0889511108398436e-06,
      "step": 3239,
      "training_step_time": 17.080525398254395
    },
    {
      "epoch": 3.08990478515625e-06,
      "grad_norm": 1.1652299165725708,
      "learning_rate": 1e-05,
      "loss": 0.0645,
      "step": 3240
    },
    {
      "epoch": 3.08990478515625e-06,
      "model_forward_time": 5.5972900390625,
      "step": 3240
    },
    {
      "dynamics_loss_avg": 0.05246199108660221,
      "epoch": 3.08990478515625e-06,
      "step": 3240
    },
    {
      "action_loss_avg": 0.014339813217520713,
      "epoch": 3.08990478515625e-06,
      "step": 3240
    },
    {
      "epoch": 3.08990478515625e-06,
      "step": 3240,
      "training_step_time": 17.06981635093689
    },
    {
      "epoch": 3.0908584594726563e-06,
      "model_forward_time": 5.601588010787964,
      "step": 3241
    },
    {
      "epoch": 3.0908584594726563e-06,
      "step": 3241,
      "training_step_time": 17.059844493865967
    },
    {
      "epoch": 3.0918121337890625e-06,
      "model_forward_time": 5.958109617233276,
      "step": 3242
    },
    {
      "epoch": 3.0918121337890625e-06,
      "step": 3242,
      "training_step_time": 17.409480810165405
    },
    {
      "epoch": 3.0927658081054686e-06,
      "model_forward_time": 5.5938804149627686,
      "step": 3243
    },
    {
      "epoch": 3.0927658081054686e-06,
      "step": 3243,
      "training_step_time": 17.00999879837036
    },
    {
      "epoch": 3.093719482421875e-06,
      "model_forward_time": 5.599167823791504,
      "step": 3244
    },
    {
      "epoch": 3.093719482421875e-06,
      "step": 3244,
      "training_step_time": 17.06198525428772
    },
    {
      "epoch": 3.0946731567382814e-06,
      "model_forward_time": 5.594666004180908,
      "step": 3245
    },
    {
      "epoch": 3.0946731567382814e-06,
      "step": 3245,
      "training_step_time": 17.07345461845398
    },
    {
      "epoch": 3.0956268310546875e-06,
      "model_forward_time": 5.6159844398498535,
      "step": 3246
    },
    {
      "epoch": 3.0956268310546875e-06,
      "step": 3246,
      "training_step_time": 17.129496335983276
    },
    {
      "epoch": 3.0965805053710936e-06,
      "model_forward_time": 5.627606153488159,
      "step": 3247
    },
    {
      "epoch": 3.0965805053710936e-06,
      "step": 3247,
      "training_step_time": 17.5100839138031
    },
    {
      "epoch": 3.0975341796875e-06,
      "model_forward_time": 5.618107557296753,
      "step": 3248
    },
    {
      "epoch": 3.0975341796875e-06,
      "step": 3248,
      "training_step_time": 17.134718894958496
    },
    {
      "epoch": 3.0984878540039064e-06,
      "model_forward_time": 5.613516569137573,
      "step": 3249
    },
    {
      "epoch": 3.0984878540039064e-06,
      "step": 3249,
      "training_step_time": 17.064374208450317
    },
    {
      "epoch": 3.0994415283203125e-06,
      "grad_norm": 1.3568670749664307,
      "learning_rate": 1e-05,
      "loss": 0.088,
      "step": 3250
    },
    {
      "epoch": 3.0994415283203125e-06,
      "model_forward_time": 5.613981246948242,
      "step": 3250
    },
    {
      "dynamics_loss_avg": 0.06989767886698246,
      "epoch": 3.0994415283203125e-06,
      "step": 3250
    },
    {
      "action_loss_avg": 0.014623677730560303,
      "epoch": 3.0994415283203125e-06,
      "step": 3250
    },
    {
      "epoch": 3.0994415283203125e-06,
      "step": 3250,
      "training_step_time": 17.108200550079346
    },
    {
      "epoch": 3.1003952026367186e-06,
      "model_forward_time": 5.619662523269653,
      "step": 3251
    },
    {
      "epoch": 3.1003952026367186e-06,
      "step": 3251,
      "training_step_time": 17.12405824661255
    },
    {
      "epoch": 3.101348876953125e-06,
      "model_forward_time": 5.612950563430786,
      "step": 3252
    },
    {
      "epoch": 3.101348876953125e-06,
      "step": 3252,
      "training_step_time": 17.105987071990967
    },
    {
      "epoch": 3.1023025512695314e-06,
      "model_forward_time": 5.973012447357178,
      "step": 3253
    },
    {
      "epoch": 3.1023025512695314e-06,
      "step": 3253,
      "training_step_time": 17.444696187973022
    },
    {
      "epoch": 3.1032562255859375e-06,
      "model_forward_time": 5.581641674041748,
      "step": 3254
    },
    {
      "epoch": 3.1032562255859375e-06,
      "step": 3254,
      "training_step_time": 17.025306940078735
    },
    {
      "epoch": 3.1042098999023436e-06,
      "model_forward_time": 5.570146560668945,
      "step": 3255
    },
    {
      "epoch": 3.1042098999023436e-06,
      "step": 3255,
      "training_step_time": 16.996281385421753
    },
    {
      "epoch": 3.10516357421875e-06,
      "model_forward_time": 5.574394941329956,
      "step": 3256
    },
    {
      "epoch": 3.10516357421875e-06,
      "step": 3256,
      "training_step_time": 16.99105715751648
    },
    {
      "epoch": 3.1061172485351564e-06,
      "model_forward_time": 5.566565036773682,
      "step": 3257
    },
    {
      "epoch": 3.1061172485351564e-06,
      "step": 3257,
      "training_step_time": 16.991591930389404
    },
    {
      "epoch": 3.1070709228515625e-06,
      "model_forward_time": 5.572359085083008,
      "step": 3258
    },
    {
      "epoch": 3.1070709228515625e-06,
      "step": 3258,
      "training_step_time": 17.371593236923218
    },
    {
      "epoch": 3.1080245971679687e-06,
      "model_forward_time": 5.573897361755371,
      "step": 3259
    },
    {
      "epoch": 3.1080245971679687e-06,
      "step": 3259,
      "training_step_time": 16.981963634490967
    },
    {
      "epoch": 3.108978271484375e-06,
      "grad_norm": 1.2781270742416382,
      "learning_rate": 1e-05,
      "loss": 0.085,
      "step": 3260
    },
    {
      "epoch": 3.108978271484375e-06,
      "model_forward_time": 5.580275297164917,
      "step": 3260
    },
    {
      "dynamics_loss_avg": 0.07554789260029793,
      "epoch": 3.108978271484375e-06,
      "step": 3260
    },
    {
      "action_loss_avg": 0.018371000001206995,
      "epoch": 3.108978271484375e-06,
      "step": 3260
    },
    {
      "epoch": 3.108978271484375e-06,
      "step": 3260,
      "training_step_time": 17.008790731430054
    },
    {
      "epoch": 3.1099319458007814e-06,
      "model_forward_time": 5.591135025024414,
      "step": 3261
    },
    {
      "epoch": 3.1099319458007814e-06,
      "step": 3261,
      "training_step_time": 17.044235944747925
    },
    {
      "epoch": 3.1108856201171875e-06,
      "model_forward_time": 5.594096660614014,
      "step": 3262
    },
    {
      "epoch": 3.1108856201171875e-06,
      "step": 3262,
      "training_step_time": 17.068620443344116
    },
    {
      "epoch": 3.1118392944335937e-06,
      "model_forward_time": 5.59893536567688,
      "step": 3263
    },
    {
      "epoch": 3.1118392944335937e-06,
      "step": 3263,
      "training_step_time": 17.063291788101196
    },
    {
      "epoch": 3.11279296875e-06,
      "model_forward_time": 5.975793361663818,
      "step": 3264
    },
    {
      "epoch": 3.11279296875e-06,
      "step": 3264,
      "training_step_time": 17.427183389663696
    },
    {
      "epoch": 3.1137466430664064e-06,
      "model_forward_time": 5.601870059967041,
      "step": 3265
    },
    {
      "epoch": 3.1137466430664064e-06,
      "step": 3265,
      "training_step_time": 17.08252477645874
    },
    {
      "epoch": 3.1147003173828125e-06,
      "model_forward_time": 5.591327667236328,
      "step": 3266
    },
    {
      "epoch": 3.1147003173828125e-06,
      "step": 3266,
      "training_step_time": 17.048675775527954
    },
    {
      "epoch": 3.1156539916992187e-06,
      "model_forward_time": 5.595680475234985,
      "step": 3267
    },
    {
      "epoch": 3.1156539916992187e-06,
      "step": 3267,
      "training_step_time": 17.06668257713318
    },
    {
      "epoch": 3.116607666015625e-06,
      "model_forward_time": 5.614513397216797,
      "step": 3268
    },
    {
      "epoch": 3.116607666015625e-06,
      "step": 3268,
      "training_step_time": 17.11577272415161
    },
    {
      "epoch": 3.1175613403320314e-06,
      "model_forward_time": 5.963533401489258,
      "step": 3269
    },
    {
      "epoch": 3.1175613403320314e-06,
      "step": 3269,
      "training_step_time": 17.457930088043213
    },
    {
      "epoch": 3.1185150146484375e-06,
      "grad_norm": 1.357853889465332,
      "learning_rate": 1e-05,
      "loss": 0.0981,
      "step": 3270
    },
    {
      "epoch": 3.1185150146484375e-06,
      "model_forward_time": 5.611986398696899,
      "step": 3270
    },
    {
      "dynamics_loss_avg": 0.08310099355876446,
      "epoch": 3.1185150146484375e-06,
      "step": 3270
    },
    {
      "action_loss_avg": 0.021009934693574907,
      "epoch": 3.1185150146484375e-06,
      "step": 3270
    },
    {
      "epoch": 3.1185150146484375e-06,
      "step": 3270,
      "training_step_time": 17.122549057006836
    },
    {
      "epoch": 3.1194686889648437e-06,
      "model_forward_time": 5.623032331466675,
      "step": 3271
    },
    {
      "epoch": 3.1194686889648437e-06,
      "step": 3271,
      "training_step_time": 17.1058087348938
    },
    {
      "epoch": 3.12042236328125e-06,
      "model_forward_time": 5.59329628944397,
      "step": 3272
    },
    {
      "epoch": 3.12042236328125e-06,
      "step": 3272,
      "training_step_time": 17.04847264289856
    },
    {
      "epoch": 3.1213760375976564e-06,
      "model_forward_time": 5.57407021522522,
      "step": 3273
    },
    {
      "epoch": 3.1213760375976564e-06,
      "step": 3273,
      "training_step_time": 16.98745584487915
    },
    {
      "epoch": 3.1223297119140625e-06,
      "model_forward_time": 5.579632759094238,
      "step": 3274
    },
    {
      "epoch": 3.1223297119140625e-06,
      "step": 3274,
      "training_step_time": 17.0028018951416
    },
    {
      "epoch": 3.1232833862304687e-06,
      "model_forward_time": 5.968447685241699,
      "step": 3275
    },
    {
      "epoch": 3.1232833862304687e-06,
      "step": 3275,
      "training_step_time": 17.44349479675293
    },
    {
      "epoch": 3.124237060546875e-06,
      "model_forward_time": 5.6372435092926025,
      "step": 3276
    },
    {
      "epoch": 3.124237060546875e-06,
      "step": 3276,
      "training_step_time": 17.167577743530273
    },
    {
      "epoch": 3.1251907348632814e-06,
      "model_forward_time": 5.6208319664001465,
      "step": 3277
    },
    {
      "epoch": 3.1251907348632814e-06,
      "step": 3277,
      "training_step_time": 17.1128191947937
    },
    {
      "epoch": 3.1261444091796875e-06,
      "model_forward_time": 5.59331488609314,
      "step": 3278
    },
    {
      "epoch": 3.1261444091796875e-06,
      "step": 3278,
      "training_step_time": 17.028930187225342
    },
    {
      "epoch": 3.1270980834960937e-06,
      "model_forward_time": 5.599287509918213,
      "step": 3279
    },
    {
      "epoch": 3.1270980834960937e-06,
      "step": 3279,
      "training_step_time": 17.029577016830444
    },
    {
      "epoch": 3.1280517578125e-06,
      "grad_norm": 2.0833256244659424,
      "learning_rate": 1e-05,
      "loss": 0.0887,
      "step": 3280
    },
    {
      "epoch": 3.1280517578125e-06,
      "model_forward_time": 5.578678369522095,
      "step": 3280
    },
    {
      "dynamics_loss_avg": 0.06170221194624901,
      "epoch": 3.1280517578125e-06,
      "step": 3280
    },
    {
      "action_loss_avg": 0.01731913350522518,
      "epoch": 3.1280517578125e-06,
      "step": 3280
    },
    {
      "epoch": 3.1280517578125e-06,
      "step": 3280,
      "training_step_time": 17.37937569618225
    },
    {
      "epoch": 3.1290054321289064e-06,
      "model_forward_time": 5.591387748718262,
      "step": 3281
    },
    {
      "epoch": 3.1290054321289064e-06,
      "step": 3281,
      "training_step_time": 17.04395842552185
    },
    {
      "epoch": 3.1299591064453125e-06,
      "model_forward_time": 5.593847990036011,
      "step": 3282
    },
    {
      "epoch": 3.1299591064453125e-06,
      "step": 3282,
      "training_step_time": 17.06519889831543
    },
    {
      "epoch": 3.1309127807617187e-06,
      "model_forward_time": 5.600921869277954,
      "step": 3283
    },
    {
      "epoch": 3.1309127807617187e-06,
      "step": 3283,
      "training_step_time": 17.08679485321045
    },
    {
      "epoch": 3.131866455078125e-06,
      "model_forward_time": 5.609977960586548,
      "step": 3284
    },
    {
      "epoch": 3.131866455078125e-06,
      "step": 3284,
      "training_step_time": 17.118059158325195
    },
    {
      "epoch": 3.1328201293945314e-06,
      "model_forward_time": 5.619610071182251,
      "step": 3285
    },
    {
      "epoch": 3.1328201293945314e-06,
      "step": 3285,
      "training_step_time": 17.096967935562134
    },
    {
      "epoch": 3.1337738037109375e-06,
      "model_forward_time": 5.979888677597046,
      "step": 3286
    },
    {
      "epoch": 3.1337738037109375e-06,
      "step": 3286,
      "training_step_time": 17.465763568878174
    },
    {
      "epoch": 3.1347274780273437e-06,
      "model_forward_time": 5.60785174369812,
      "step": 3287
    },
    {
      "epoch": 3.1347274780273437e-06,
      "step": 3287,
      "training_step_time": 17.09133529663086
    },
    {
      "epoch": 3.13568115234375e-06,
      "model_forward_time": 5.616960763931274,
      "step": 3288
    },
    {
      "epoch": 3.13568115234375e-06,
      "step": 3288,
      "training_step_time": 17.115742921829224
    },
    {
      "epoch": 3.1366348266601564e-06,
      "model_forward_time": 5.618982791900635,
      "step": 3289
    },
    {
      "epoch": 3.1366348266601564e-06,
      "step": 3289,
      "training_step_time": 17.02473735809326
    },
    {
      "epoch": 3.1375885009765625e-06,
      "grad_norm": 0.9005701541900635,
      "learning_rate": 1e-05,
      "loss": 0.076,
      "step": 3290
    },
    {
      "epoch": 3.1375885009765625e-06,
      "model_forward_time": 5.594661712646484,
      "step": 3290
    },
    {
      "dynamics_loss_avg": 0.059397993795573714,
      "epoch": 3.1375885009765625e-06,
      "step": 3290
    },
    {
      "action_loss_avg": 0.014882892044261098,
      "epoch": 3.1375885009765625e-06,
      "step": 3290
    },
    {
      "epoch": 3.1375885009765625e-06,
      "step": 3290,
      "training_step_time": 17.038257837295532
    },
    {
      "epoch": 3.1385421752929687e-06,
      "model_forward_time": 5.583563804626465,
      "step": 3291
    },
    {
      "epoch": 3.1385421752929687e-06,
      "step": 3291,
      "training_step_time": 17.383414268493652
    },
    {
      "epoch": 3.139495849609375e-06,
      "model_forward_time": 5.579828500747681,
      "step": 3292
    },
    {
      "epoch": 3.139495849609375e-06,
      "step": 3292,
      "training_step_time": 17.004304885864258
    },
    {
      "epoch": 3.1404495239257814e-06,
      "model_forward_time": 5.593911170959473,
      "step": 3293
    },
    {
      "epoch": 3.1404495239257814e-06,
      "step": 3293,
      "training_step_time": 17.046678066253662
    },
    {
      "epoch": 3.1414031982421875e-06,
      "model_forward_time": 5.6008195877075195,
      "step": 3294
    },
    {
      "epoch": 3.1414031982421875e-06,
      "step": 3294,
      "training_step_time": 17.06569266319275
    },
    {
      "epoch": 3.1423568725585937e-06,
      "model_forward_time": 5.595614910125732,
      "step": 3295
    },
    {
      "epoch": 3.1423568725585937e-06,
      "step": 3295,
      "training_step_time": 17.05327558517456
    },
    {
      "epoch": 3.143310546875e-06,
      "model_forward_time": 5.596839189529419,
      "step": 3296
    },
    {
      "epoch": 3.143310546875e-06,
      "step": 3296,
      "training_step_time": 17.06690287590027
    },
    {
      "epoch": 3.1442642211914064e-06,
      "model_forward_time": 5.995169162750244,
      "step": 3297
    },
    {
      "epoch": 3.1442642211914064e-06,
      "step": 3297,
      "training_step_time": 17.427611589431763
    },
    {
      "epoch": 3.1452178955078125e-06,
      "model_forward_time": 5.581411123275757,
      "step": 3298
    },
    {
      "epoch": 3.1452178955078125e-06,
      "step": 3298,
      "training_step_time": 17.020777463912964
    },
    {
      "epoch": 3.1461715698242187e-06,
      "model_forward_time": 5.578176975250244,
      "step": 3299
    },
    {
      "epoch": 3.1461715698242187e-06,
      "step": 3299,
      "training_step_time": 17.025371074676514
    },
    {
      "epoch": 3.147125244140625e-06,
      "grad_norm": 0.7632082104682922,
      "learning_rate": 1e-05,
      "loss": 0.0782,
      "step": 3300
    },
    {
      "epoch": 3.147125244140625e-06,
      "model_forward_time": 5.591696500778198,
      "step": 3300
    },
    {
      "dynamics_loss_avg": 0.05961990281939507,
      "epoch": 3.147125244140625e-06,
      "step": 3300
    },
    {
      "action_loss_avg": 0.017471670312806963,
      "epoch": 3.147125244140625e-06,
      "step": 3300
    },
    {
      "epoch": 3.147125244140625e-06,
      "step": 3300,
      "training_step_time": 17.060041189193726
    },
    {
      "epoch": 3.1480789184570314e-06,
      "model_forward_time": 5.603002071380615,
      "step": 3301
    },
    {
      "epoch": 3.1480789184570314e-06,
      "step": 3301,
      "training_step_time": 17.078749895095825
    },
    {
      "epoch": 3.1490325927734375e-06,
      "model_forward_time": 5.9406418800354,
      "step": 3302
    },
    {
      "epoch": 3.1490325927734375e-06,
      "step": 3302,
      "training_step_time": 17.401214599609375
    },
    {
      "epoch": 3.1499862670898437e-06,
      "model_forward_time": 5.592396259307861,
      "step": 3303
    },
    {
      "epoch": 3.1499862670898437e-06,
      "step": 3303,
      "training_step_time": 17.03805685043335
    },
    {
      "epoch": 3.15093994140625e-06,
      "model_forward_time": 5.592310428619385,
      "step": 3304
    },
    {
      "epoch": 3.15093994140625e-06,
      "step": 3304,
      "training_step_time": 17.02853751182556
    },
    {
      "epoch": 3.1518936157226564e-06,
      "model_forward_time": 5.575037002563477,
      "step": 3305
    },
    {
      "epoch": 3.1518936157226564e-06,
      "step": 3305,
      "training_step_time": 17.011629343032837
    },
    {
      "epoch": 3.1528472900390625e-06,
      "model_forward_time": 5.584320545196533,
      "step": 3306
    },
    {
      "epoch": 3.1528472900390625e-06,
      "step": 3306,
      "training_step_time": 17.053826093673706
    },
    {
      "epoch": 3.1538009643554687e-06,
      "model_forward_time": 5.608032703399658,
      "step": 3307
    },
    {
      "epoch": 3.1538009643554687e-06,
      "step": 3307,
      "training_step_time": 17.110326766967773
    },
    {
      "epoch": 3.154754638671875e-06,
      "model_forward_time": 5.976144552230835,
      "step": 3308
    },
    {
      "epoch": 3.154754638671875e-06,
      "step": 3308,
      "training_step_time": 17.461819410324097
    },
    {
      "epoch": 3.1557083129882814e-06,
      "model_forward_time": 5.598339796066284,
      "step": 3309
    },
    {
      "epoch": 3.1557083129882814e-06,
      "step": 3309,
      "training_step_time": 17.04972004890442
    },
    {
      "epoch": 3.1566619873046876e-06,
      "grad_norm": 1.8692243099212646,
      "learning_rate": 1e-05,
      "loss": 0.0747,
      "step": 3310
    },
    {
      "epoch": 3.1566619873046876e-06,
      "model_forward_time": 5.583172798156738,
      "step": 3310
    },
    {
      "dynamics_loss_avg": 0.06215442828834057,
      "epoch": 3.1566619873046876e-06,
      "step": 3310
    },
    {
      "action_loss_avg": 0.015316299605183303,
      "epoch": 3.1566619873046876e-06,
      "step": 3310
    },
    {
      "epoch": 3.1566619873046876e-06,
      "step": 3310,
      "training_step_time": 17.030391216278076
    },
    {
      "epoch": 3.1576156616210937e-06,
      "model_forward_time": 5.590081691741943,
      "step": 3311
    },
    {
      "epoch": 3.1576156616210937e-06,
      "step": 3311,
      "training_step_time": 17.05510711669922
    },
    {
      "epoch": 3.1585693359375e-06,
      "model_forward_time": 5.609065532684326,
      "step": 3312
    },
    {
      "epoch": 3.1585693359375e-06,
      "step": 3312,
      "training_step_time": 17.10043954849243
    },
    {
      "epoch": 3.1595230102539064e-06,
      "model_forward_time": 5.606073379516602,
      "step": 3313
    },
    {
      "epoch": 3.1595230102539064e-06,
      "step": 3313,
      "training_step_time": 17.465482234954834
    },
    {
      "epoch": 3.1604766845703126e-06,
      "model_forward_time": 5.617368936538696,
      "step": 3314
    },
    {
      "epoch": 3.1604766845703126e-06,
      "step": 3314,
      "training_step_time": 17.113451719284058
    },
    {
      "epoch": 3.1614303588867187e-06,
      "model_forward_time": 5.610980749130249,
      "step": 3315
    },
    {
      "epoch": 3.1614303588867187e-06,
      "step": 3315,
      "training_step_time": 17.095752239227295
    },
    {
      "epoch": 3.162384033203125e-06,
      "model_forward_time": 5.5955119132995605,
      "step": 3316
    },
    {
      "epoch": 3.162384033203125e-06,
      "step": 3316,
      "training_step_time": 17.029723644256592
    },
    {
      "epoch": 3.1633377075195314e-06,
      "model_forward_time": 5.575009822845459,
      "step": 3317
    },
    {
      "epoch": 3.1633377075195314e-06,
      "step": 3317,
      "training_step_time": 16.994384288787842
    },
    {
      "epoch": 3.1642913818359376e-06,
      "model_forward_time": 5.5714380741119385,
      "step": 3318
    },
    {
      "epoch": 3.1642913818359376e-06,
      "step": 3318,
      "training_step_time": 17.002171277999878
    },
    {
      "epoch": 3.1652450561523437e-06,
      "model_forward_time": 5.931001424789429,
      "step": 3319
    },
    {
      "epoch": 3.1652450561523437e-06,
      "step": 3319,
      "training_step_time": 17.379648208618164
    },
    {
      "epoch": 3.16619873046875e-06,
      "grad_norm": 1.1358752250671387,
      "learning_rate": 1e-05,
      "loss": 0.0821,
      "step": 3320
    },
    {
      "epoch": 3.16619873046875e-06,
      "model_forward_time": 5.599068880081177,
      "step": 3320
    },
    {
      "dynamics_loss_avg": 0.06499296352267266,
      "epoch": 3.16619873046875e-06,
      "step": 3320
    },
    {
      "action_loss_avg": 0.01556147113442421,
      "epoch": 3.16619873046875e-06,
      "step": 3320
    },
    {
      "epoch": 3.16619873046875e-06,
      "step": 3320,
      "training_step_time": 17.07296395301819
    },
    {
      "epoch": 3.1671524047851564e-06,
      "model_forward_time": 5.605945110321045,
      "step": 3321
    },
    {
      "epoch": 3.1671524047851564e-06,
      "step": 3321,
      "training_step_time": 17.07554006576538
    },
    {
      "epoch": 3.1681060791015626e-06,
      "model_forward_time": 5.617406606674194,
      "step": 3322
    },
    {
      "epoch": 3.1681060791015626e-06,
      "step": 3322,
      "training_step_time": 17.088540077209473
    },
    {
      "epoch": 3.1690597534179687e-06,
      "model_forward_time": 5.600114107131958,
      "step": 3323
    },
    {
      "epoch": 3.1690597534179687e-06,
      "step": 3323,
      "training_step_time": 17.080906867980957
    },
    {
      "epoch": 3.170013427734375e-06,
      "model_forward_time": 5.942410707473755,
      "step": 3324
    },
    {
      "epoch": 3.170013427734375e-06,
      "step": 3324,
      "training_step_time": 17.418375968933105
    },
    {
      "epoch": 3.1709671020507814e-06,
      "model_forward_time": 5.611282587051392,
      "step": 3325
    },
    {
      "epoch": 3.1709671020507814e-06,
      "step": 3325,
      "training_step_time": 17.096207857131958
    },
    {
      "epoch": 3.1719207763671876e-06,
      "model_forward_time": 5.608754396438599,
      "step": 3326
    },
    {
      "epoch": 3.1719207763671876e-06,
      "step": 3326,
      "training_step_time": 17.103508710861206
    },
    {
      "epoch": 3.1728744506835937e-06,
      "model_forward_time": 5.6074793338775635,
      "step": 3327
    },
    {
      "epoch": 3.1728744506835937e-06,
      "step": 3327,
      "training_step_time": 17.069072008132935
    },
    {
      "epoch": 3.173828125e-06,
      "model_forward_time": 5.5942676067352295,
      "step": 3328
    },
    {
      "epoch": 3.173828125e-06,
      "step": 3328,
      "training_step_time": 17.04211711883545
    },
    {
      "epoch": 3.1747817993164064e-06,
      "model_forward_time": 5.593778371810913,
      "step": 3329
    },
    {
      "epoch": 3.1747817993164064e-06,
      "step": 3329,
      "training_step_time": 17.032036066055298
    },
    {
      "epoch": 3.1757354736328126e-06,
      "grad_norm": 0.9919289350509644,
      "learning_rate": 1e-05,
      "loss": 0.0756,
      "step": 3330
    },
    {
      "epoch": 3.1757354736328126e-06,
      "model_forward_time": 5.946632385253906,
      "step": 3330
    },
    {
      "dynamics_loss_avg": 0.061595755629241467,
      "epoch": 3.1757354736328126e-06,
      "step": 3330
    },
    {
      "action_loss_avg": 0.0180233642924577,
      "epoch": 3.1757354736328126e-06,
      "step": 3330
    },
    {
      "epoch": 3.1757354736328126e-06,
      "step": 3330,
      "training_step_time": 17.370426416397095
    },
    {
      "epoch": 3.1766891479492187e-06,
      "model_forward_time": 5.58633828163147,
      "step": 3331
    },
    {
      "epoch": 3.1766891479492187e-06,
      "step": 3331,
      "training_step_time": 17.051364183425903
    },
    {
      "epoch": 3.177642822265625e-06,
      "model_forward_time": 5.599890232086182,
      "step": 3332
    },
    {
      "epoch": 3.177642822265625e-06,
      "step": 3332,
      "training_step_time": 17.094059705734253
    },
    {
      "epoch": 3.1785964965820314e-06,
      "model_forward_time": 5.6127941608428955,
      "step": 3333
    },
    {
      "epoch": 3.1785964965820314e-06,
      "step": 3333,
      "training_step_time": 17.110738277435303
    },
    {
      "epoch": 3.1795501708984376e-06,
      "model_forward_time": 5.620946884155273,
      "step": 3334
    },
    {
      "epoch": 3.1795501708984376e-06,
      "step": 3334,
      "training_step_time": 17.133098602294922
    },
    {
      "epoch": 3.1805038452148437e-06,
      "model_forward_time": 5.9590394496917725,
      "step": 3335
    },
    {
      "epoch": 3.1805038452148437e-06,
      "step": 3335,
      "training_step_time": 17.451972723007202
    },
    {
      "epoch": 3.18145751953125e-06,
      "model_forward_time": 5.60393214225769,
      "step": 3336
    },
    {
      "epoch": 3.18145751953125e-06,
      "step": 3336,
      "training_step_time": 17.09091305732727
    },
    {
      "epoch": 3.1824111938476564e-06,
      "model_forward_time": 5.608281373977661,
      "step": 3337
    },
    {
      "epoch": 3.1824111938476564e-06,
      "step": 3337,
      "training_step_time": 17.063633680343628
    },
    {
      "epoch": 3.1833648681640626e-06,
      "model_forward_time": 5.597098350524902,
      "step": 3338
    },
    {
      "epoch": 3.1833648681640626e-06,
      "step": 3338,
      "training_step_time": 17.07149362564087
    },
    {
      "epoch": 3.1843185424804687e-06,
      "model_forward_time": 5.597609758377075,
      "step": 3339
    },
    {
      "epoch": 3.1843185424804687e-06,
      "step": 3339,
      "training_step_time": 17.061532258987427
    },
    {
      "epoch": 3.185272216796875e-06,
      "grad_norm": 1.0118120908737183,
      "learning_rate": 1e-05,
      "loss": 0.0828,
      "step": 3340
    },
    {
      "epoch": 3.185272216796875e-06,
      "model_forward_time": 5.599379301071167,
      "step": 3340
    },
    {
      "dynamics_loss_avg": 0.06289210747927428,
      "epoch": 3.185272216796875e-06,
      "step": 3340
    },
    {
      "action_loss_avg": 0.015000129118561745,
      "epoch": 3.185272216796875e-06,
      "step": 3340
    },
    {
      "epoch": 3.185272216796875e-06,
      "step": 3340,
      "training_step_time": 17.432647466659546
    },
    {
      "epoch": 3.1862258911132814e-06,
      "model_forward_time": 5.578594923019409,
      "step": 3341
    },
    {
      "epoch": 3.1862258911132814e-06,
      "step": 3341,
      "training_step_time": 17.015610933303833
    },
    {
      "epoch": 3.1871795654296876e-06,
      "model_forward_time": 5.597249507904053,
      "step": 3342
    },
    {
      "epoch": 3.1871795654296876e-06,
      "step": 3342,
      "training_step_time": 17.02807641029358
    },
    {
      "epoch": 3.1881332397460937e-06,
      "model_forward_time": 5.591590166091919,
      "step": 3343
    },
    {
      "epoch": 3.1881332397460937e-06,
      "step": 3343,
      "training_step_time": 17.052314281463623
    },
    {
      "epoch": 3.1890869140625e-06,
      "model_forward_time": 5.6031653881073,
      "step": 3344
    },
    {
      "epoch": 3.1890869140625e-06,
      "step": 3344,
      "training_step_time": 17.05576992034912
    },
    {
      "epoch": 3.1900405883789064e-06,
      "model_forward_time": 5.587157726287842,
      "step": 3345
    },
    {
      "epoch": 3.1900405883789064e-06,
      "step": 3345,
      "training_step_time": 17.049436807632446
    },
    {
      "epoch": 3.1909942626953126e-06,
      "model_forward_time": 5.9439263343811035,
      "step": 3346
    },
    {
      "epoch": 3.1909942626953126e-06,
      "step": 3346,
      "training_step_time": 17.397682905197144
    },
    {
      "epoch": 3.1919479370117187e-06,
      "model_forward_time": 5.601767063140869,
      "step": 3347
    },
    {
      "epoch": 3.1919479370117187e-06,
      "step": 3347,
      "training_step_time": 17.063538312911987
    },
    {
      "epoch": 3.192901611328125e-06,
      "model_forward_time": 5.604182481765747,
      "step": 3348
    },
    {
      "epoch": 3.192901611328125e-06,
      "step": 3348,
      "training_step_time": 17.08526039123535
    },
    {
      "epoch": 3.1938552856445314e-06,
      "model_forward_time": 5.608187675476074,
      "step": 3349
    },
    {
      "epoch": 3.1938552856445314e-06,
      "step": 3349,
      "training_step_time": 17.094099521636963
    },
    {
      "epoch": 3.1948089599609376e-06,
      "grad_norm": 1.9905489683151245,
      "learning_rate": 1e-05,
      "loss": 0.0733,
      "step": 3350
    },
    {
      "epoch": 3.1948089599609376e-06,
      "model_forward_time": 5.609436750411987,
      "step": 3350
    },
    {
      "dynamics_loss_avg": 0.05623464249074459,
      "epoch": 3.1948089599609376e-06,
      "step": 3350
    },
    {
      "action_loss_avg": 0.018284285767003895,
      "epoch": 3.1948089599609376e-06,
      "step": 3350
    },
    {
      "epoch": 3.1948089599609376e-06,
      "step": 3350,
      "training_step_time": 17.08614945411682
    },
    {
      "epoch": 3.1957626342773437e-06,
      "model_forward_time": 5.60770845413208,
      "step": 3351
    },
    {
      "epoch": 3.1957626342773437e-06,
      "step": 3351,
      "training_step_time": 17.458414793014526
    },
    {
      "epoch": 3.19671630859375e-06,
      "model_forward_time": 5.608235597610474,
      "step": 3352
    },
    {
      "epoch": 3.19671630859375e-06,
      "step": 3352,
      "training_step_time": 17.066936254501343
    },
    {
      "epoch": 3.1976699829101564e-06,
      "model_forward_time": 5.601855516433716,
      "step": 3353
    },
    {
      "epoch": 3.1976699829101564e-06,
      "step": 3353,
      "training_step_time": 17.0746009349823
    },
    {
      "epoch": 3.1986236572265626e-06,
      "model_forward_time": 5.604669094085693,
      "step": 3354
    },
    {
      "epoch": 3.1986236572265626e-06,
      "step": 3354,
      "training_step_time": 17.09339928627014
    },
    {
      "epoch": 3.1995773315429687e-06,
      "model_forward_time": 4.281526803970337,
      "step": 3355
    },
    {
      "epoch": 3.1995773315429687e-06,
      "step": 3355,
      "training_step_time": 12.819190263748169
    },
    {
      "epoch": 3.200531005859375e-06,
      "model_forward_time": 5.611706495285034,
      "step": 3356
    },
    {
      "epoch": 3.200531005859375e-06,
      "step": 3356,
      "training_step_time": 17.104898691177368
    },
    {
      "epoch": 3.2014846801757814e-06,
      "model_forward_time": 5.97224235534668,
      "step": 3357
    },
    {
      "epoch": 3.2014846801757814e-06,
      "step": 3357,
      "training_step_time": 17.475986003875732
    },
    {
      "epoch": 3.2024383544921876e-06,
      "model_forward_time": 5.642374515533447,
      "step": 3358
    },
    {
      "epoch": 3.2024383544921876e-06,
      "step": 3358,
      "training_step_time": 17.147141218185425
    },
    {
      "epoch": 3.2033920288085937e-06,
      "model_forward_time": 5.617491006851196,
      "step": 3359
    },
    {
      "epoch": 3.2033920288085937e-06,
      "step": 3359,
      "training_step_time": 17.12950372695923
    },
    {
      "epoch": 3.204345703125e-06,
      "grad_norm": 0.7808769941329956,
      "learning_rate": 1e-05,
      "loss": 0.0891,
      "step": 3360
    },
    {
      "epoch": 3.204345703125e-06,
      "model_forward_time": 5.614220380783081,
      "step": 3360
    },
    {
      "dynamics_loss_avg": 0.06813712250441313,
      "epoch": 3.204345703125e-06,
      "step": 3360
    },
    {
      "action_loss_avg": 0.02119479328393936,
      "epoch": 3.204345703125e-06,
      "step": 3360
    },
    {
      "epoch": 3.204345703125e-06,
      "step": 3360,
      "training_step_time": 17.11507248878479
    },
    {
      "epoch": 3.2052993774414064e-06,
      "model_forward_time": 5.638104200363159,
      "step": 3361
    },
    {
      "epoch": 3.2052993774414064e-06,
      "step": 3361,
      "training_step_time": 17.138676643371582
    },
    {
      "epoch": 3.2062530517578126e-06,
      "model_forward_time": 5.995408535003662,
      "step": 3362
    },
    {
      "epoch": 3.2062530517578126e-06,
      "step": 3362,
      "training_step_time": 17.49377179145813
    },
    {
      "epoch": 3.2072067260742187e-06,
      "model_forward_time": 5.604690790176392,
      "step": 3363
    },
    {
      "epoch": 3.2072067260742187e-06,
      "step": 3363,
      "training_step_time": 17.075338125228882
    },
    {
      "epoch": 3.208160400390625e-06,
      "model_forward_time": 5.595135688781738,
      "step": 3364
    },
    {
      "epoch": 3.208160400390625e-06,
      "step": 3364,
      "training_step_time": 17.057903289794922
    },
    {
      "epoch": 3.2091140747070315e-06,
      "model_forward_time": 5.617739915847778,
      "step": 3365
    },
    {
      "epoch": 3.2091140747070315e-06,
      "step": 3365,
      "training_step_time": 17.054332733154297
    },
    {
      "epoch": 3.2100677490234376e-06,
      "model_forward_time": 5.584305047988892,
      "step": 3366
    },
    {
      "epoch": 3.2100677490234376e-06,
      "step": 3366,
      "training_step_time": 17.012644052505493
    },
    {
      "epoch": 3.2110214233398437e-06,
      "model_forward_time": 5.593059539794922,
      "step": 3367
    },
    {
      "epoch": 3.2110214233398437e-06,
      "step": 3367,
      "training_step_time": 17.03447198867798
    },
    {
      "epoch": 3.21197509765625e-06,
      "model_forward_time": 5.927116870880127,
      "step": 3368
    },
    {
      "epoch": 3.21197509765625e-06,
      "step": 3368,
      "training_step_time": 17.39827036857605
    },
    {
      "epoch": 3.2129287719726565e-06,
      "model_forward_time": 5.604147911071777,
      "step": 3369
    },
    {
      "epoch": 3.2129287719726565e-06,
      "step": 3369,
      "training_step_time": 17.083488941192627
    },
    {
      "epoch": 3.2138824462890626e-06,
      "grad_norm": 0.6323313117027283,
      "learning_rate": 1e-05,
      "loss": 0.0945,
      "step": 3370
    },
    {
      "epoch": 3.2138824462890626e-06,
      "model_forward_time": 5.6120898723602295,
      "step": 3370
    },
    {
      "dynamics_loss_avg": 0.07416929751634598,
      "epoch": 3.2138824462890626e-06,
      "step": 3370
    },
    {
      "action_loss_avg": 0.019396076817065476,
      "epoch": 3.2138824462890626e-06,
      "step": 3370
    },
    {
      "epoch": 3.2138824462890626e-06,
      "step": 3370,
      "training_step_time": 17.095820426940918
    },
    {
      "epoch": 3.2148361206054687e-06,
      "model_forward_time": 5.618140935897827,
      "step": 3371
    },
    {
      "epoch": 3.2148361206054687e-06,
      "step": 3371,
      "training_step_time": 17.10264301300049
    },
    {
      "epoch": 3.215789794921875e-06,
      "model_forward_time": 5.604619264602661,
      "step": 3372
    },
    {
      "epoch": 3.215789794921875e-06,
      "step": 3372,
      "training_step_time": 17.09024953842163
    },
    {
      "epoch": 3.2167434692382815e-06,
      "model_forward_time": 5.9811975955963135,
      "step": 3373
    },
    {
      "epoch": 3.2167434692382815e-06,
      "step": 3373,
      "training_step_time": 17.458602905273438
    },
    {
      "epoch": 3.2176971435546876e-06,
      "model_forward_time": 5.6040143966674805,
      "step": 3374
    },
    {
      "epoch": 3.2176971435546876e-06,
      "step": 3374,
      "training_step_time": 17.07836675643921
    },
    {
      "epoch": 3.2186508178710938e-06,
      "model_forward_time": 5.605534076690674,
      "step": 3375
    },
    {
      "epoch": 3.2186508178710938e-06,
      "step": 3375,
      "training_step_time": 17.080754280090332
    },
    {
      "epoch": 3.2196044921875e-06,
      "model_forward_time": 5.607341527938843,
      "step": 3376
    },
    {
      "epoch": 3.2196044921875e-06,
      "step": 3376,
      "training_step_time": 17.101718425750732
    },
    {
      "epoch": 3.220558166503906e-06,
      "model_forward_time": 5.613820791244507,
      "step": 3377
    },
    {
      "epoch": 3.220558166503906e-06,
      "step": 3377,
      "training_step_time": 17.084675073623657
    },
    {
      "epoch": 3.2215118408203126e-06,
      "model_forward_time": 5.593872547149658,
      "step": 3378
    },
    {
      "epoch": 3.2215118408203126e-06,
      "step": 3378,
      "training_step_time": 17.02193570137024
    },
    {
      "epoch": 3.2224655151367188e-06,
      "model_forward_time": 5.921724319458008,
      "step": 3379
    },
    {
      "epoch": 3.2224655151367188e-06,
      "step": 3379,
      "training_step_time": 17.35812282562256
    },
    {
      "epoch": 3.223419189453125e-06,
      "grad_norm": 1.015230417251587,
      "learning_rate": 1e-05,
      "loss": 0.0826,
      "step": 3380
    },
    {
      "epoch": 3.223419189453125e-06,
      "model_forward_time": 5.564486503601074,
      "step": 3380
    },
    {
      "dynamics_loss_avg": 0.06216182596981525,
      "epoch": 3.223419189453125e-06,
      "step": 3380
    },
    {
      "action_loss_avg": 0.016551112243905663,
      "epoch": 3.223419189453125e-06,
      "step": 3380
    },
    {
      "epoch": 3.223419189453125e-06,
      "step": 3380,
      "training_step_time": 16.98443365097046
    },
    {
      "epoch": 3.224372863769531e-06,
      "model_forward_time": 5.569122076034546,
      "step": 3381
    },
    {
      "epoch": 3.224372863769531e-06,
      "step": 3381,
      "training_step_time": 16.983028173446655
    },
    {
      "epoch": 3.2253265380859376e-06,
      "model_forward_time": 5.574164867401123,
      "step": 3382
    },
    {
      "epoch": 3.2253265380859376e-06,
      "step": 3382,
      "training_step_time": 17.00794219970703
    },
    {
      "epoch": 3.2262802124023438e-06,
      "model_forward_time": 5.56928825378418,
      "step": 3383
    },
    {
      "epoch": 3.2262802124023438e-06,
      "step": 3383,
      "training_step_time": 16.99401307106018
    },
    {
      "epoch": 3.22723388671875e-06,
      "model_forward_time": 5.953813076019287,
      "step": 3384
    },
    {
      "epoch": 3.22723388671875e-06,
      "step": 3384,
      "training_step_time": 17.37756896018982
    },
    {
      "epoch": 3.228187561035156e-06,
      "model_forward_time": 5.592540979385376,
      "step": 3385
    },
    {
      "epoch": 3.228187561035156e-06,
      "step": 3385,
      "training_step_time": 17.05098795890808
    },
    {
      "epoch": 3.2291412353515626e-06,
      "model_forward_time": 5.600661754608154,
      "step": 3386
    },
    {
      "epoch": 3.2291412353515626e-06,
      "step": 3386,
      "training_step_time": 17.080810070037842
    },
    {
      "epoch": 3.2300949096679688e-06,
      "model_forward_time": 5.612252473831177,
      "step": 3387
    },
    {
      "epoch": 3.2300949096679688e-06,
      "step": 3387,
      "training_step_time": 17.115633249282837
    },
    {
      "epoch": 3.231048583984375e-06,
      "model_forward_time": 5.6133201122283936,
      "step": 3388
    },
    {
      "epoch": 3.231048583984375e-06,
      "step": 3388,
      "training_step_time": 17.11886215209961
    },
    {
      "epoch": 3.232002258300781e-06,
      "model_forward_time": 5.613693475723267,
      "step": 3389
    },
    {
      "epoch": 3.232002258300781e-06,
      "step": 3389,
      "training_step_time": 17.44924831390381
    },
    {
      "epoch": 3.2329559326171876e-06,
      "grad_norm": 1.1642074584960938,
      "learning_rate": 1e-05,
      "loss": 0.0736,
      "step": 3390
    },
    {
      "epoch": 3.2329559326171876e-06,
      "model_forward_time": 5.584176540374756,
      "step": 3390
    },
    {
      "dynamics_loss_avg": 0.0627559993416071,
      "epoch": 3.2329559326171876e-06,
      "step": 3390
    },
    {
      "action_loss_avg": 0.01525215720757842,
      "epoch": 3.2329559326171876e-06,
      "step": 3390
    },
    {
      "epoch": 3.2329559326171876e-06,
      "step": 3390,
      "training_step_time": 17.025991201400757
    },
    {
      "epoch": 3.2339096069335938e-06,
      "model_forward_time": 5.567612648010254,
      "step": 3391
    },
    {
      "epoch": 3.2339096069335938e-06,
      "step": 3391,
      "training_step_time": 16.989490032196045
    },
    {
      "epoch": 3.23486328125e-06,
      "model_forward_time": 5.565579175949097,
      "step": 3392
    },
    {
      "epoch": 3.23486328125e-06,
      "step": 3392,
      "training_step_time": 16.98246693611145
    },
    {
      "epoch": 3.235816955566406e-06,
      "model_forward_time": 5.5657734870910645,
      "step": 3393
    },
    {
      "epoch": 3.235816955566406e-06,
      "step": 3393,
      "training_step_time": 16.97615361213684
    },
    {
      "epoch": 3.2367706298828126e-06,
      "model_forward_time": 5.566926002502441,
      "step": 3394
    },
    {
      "epoch": 3.2367706298828126e-06,
      "step": 3394,
      "training_step_time": 17.00556707382202
    },
    {
      "epoch": 3.2377243041992188e-06,
      "model_forward_time": 5.936751127243042,
      "step": 3395
    },
    {
      "epoch": 3.2377243041992188e-06,
      "step": 3395,
      "training_step_time": 17.35560417175293
    },
    {
      "epoch": 3.238677978515625e-06,
      "model_forward_time": 5.594942331314087,
      "step": 3396
    },
    {
      "epoch": 3.238677978515625e-06,
      "step": 3396,
      "training_step_time": 17.060920476913452
    },
    {
      "epoch": 3.239631652832031e-06,
      "model_forward_time": 5.610456466674805,
      "step": 3397
    },
    {
      "epoch": 3.239631652832031e-06,
      "step": 3397,
      "training_step_time": 17.107512712478638
    },
    {
      "epoch": 3.2405853271484376e-06,
      "model_forward_time": 5.625313997268677,
      "step": 3398
    },
    {
      "epoch": 3.2405853271484376e-06,
      "step": 3398,
      "training_step_time": 17.136225700378418
    },
    {
      "epoch": 3.2415390014648438e-06,
      "model_forward_time": 5.621963024139404,
      "step": 3399
    },
    {
      "epoch": 3.2415390014648438e-06,
      "step": 3399,
      "training_step_time": 17.104915618896484
    },
    {
      "epoch": 3.24249267578125e-06,
      "grad_norm": 1.2982192039489746,
      "learning_rate": 1e-05,
      "loss": 0.083,
      "step": 3400
    },
    {
      "epoch": 3.24249267578125e-06,
      "model_forward_time": 5.954552888870239,
      "step": 3400
    },
    {
      "dynamics_loss_avg": 0.06665087677538395,
      "epoch": 3.24249267578125e-06,
      "step": 3400
    },
    {
      "action_loss_avg": 0.017701718816533686,
      "epoch": 3.24249267578125e-06,
      "step": 3400
    },
    {
      "epoch": 3.24249267578125e-06,
      "step": 3400,
      "training_step_time": 17.36690354347229
    },
    {
      "epoch": 3.243446350097656e-06,
      "model_forward_time": 5.574463129043579,
      "step": 3401
    },
    {
      "epoch": 3.243446350097656e-06,
      "step": 3401,
      "training_step_time": 17.00351881980896
    },
    {
      "epoch": 3.2444000244140626e-06,
      "model_forward_time": 5.576798439025879,
      "step": 3402
    },
    {
      "epoch": 3.2444000244140626e-06,
      "step": 3402,
      "training_step_time": 17.02693772315979
    },
    {
      "epoch": 3.2453536987304688e-06,
      "model_forward_time": 5.572669982910156,
      "step": 3403
    },
    {
      "epoch": 3.2453536987304688e-06,
      "step": 3403,
      "training_step_time": 16.987160205841064
    },
    {
      "epoch": 3.246307373046875e-06,
      "model_forward_time": 5.581547975540161,
      "step": 3404
    },
    {
      "epoch": 3.246307373046875e-06,
      "step": 3404,
      "training_step_time": 17.018941164016724
    },
    {
      "epoch": 3.247261047363281e-06,
      "model_forward_time": 5.6015050411224365,
      "step": 3405
    },
    {
      "epoch": 3.247261047363281e-06,
      "step": 3405,
      "training_step_time": 17.057294845581055
    },
    {
      "epoch": 3.2482147216796876e-06,
      "model_forward_time": 5.961424112319946,
      "step": 3406
    },
    {
      "epoch": 3.2482147216796876e-06,
      "step": 3406,
      "training_step_time": 17.435115814208984
    },
    {
      "epoch": 3.2491683959960938e-06,
      "model_forward_time": 5.595512390136719,
      "step": 3407
    },
    {
      "epoch": 3.2491683959960938e-06,
      "step": 3407,
      "training_step_time": 17.062166690826416
    },
    {
      "epoch": 3.2501220703125e-06,
      "model_forward_time": 5.597129821777344,
      "step": 3408
    },
    {
      "epoch": 3.2501220703125e-06,
      "step": 3408,
      "training_step_time": 17.056535720825195
    },
    {
      "epoch": 3.251075744628906e-06,
      "model_forward_time": 5.603900671005249,
      "step": 3409
    },
    {
      "epoch": 3.251075744628906e-06,
      "step": 3409,
      "training_step_time": 17.07144260406494
    },
    {
      "epoch": 3.2520294189453126e-06,
      "grad_norm": 0.6709189414978027,
      "learning_rate": 1e-05,
      "loss": 0.0679,
      "step": 3410
    },
    {
      "epoch": 3.2520294189453126e-06,
      "model_forward_time": 5.597474098205566,
      "step": 3410
    },
    {
      "dynamics_loss_avg": 0.05322093386203051,
      "epoch": 3.2520294189453126e-06,
      "step": 3410
    },
    {
      "action_loss_avg": 0.012199577153660357,
      "epoch": 3.2520294189453126e-06,
      "step": 3410
    },
    {
      "epoch": 3.2520294189453126e-06,
      "step": 3410,
      "training_step_time": 17.05923056602478
    },
    {
      "epoch": 3.2529830932617188e-06,
      "model_forward_time": 5.590384483337402,
      "step": 3411
    },
    {
      "epoch": 3.2529830932617188e-06,
      "step": 3411,
      "training_step_time": 17.40351939201355
    },
    {
      "epoch": 3.253936767578125e-06,
      "model_forward_time": 5.584856986999512,
      "step": 3412
    },
    {
      "epoch": 3.253936767578125e-06,
      "step": 3412,
      "training_step_time": 17.00648593902588
    },
    {
      "epoch": 3.254890441894531e-06,
      "model_forward_time": 5.581242322921753,
      "step": 3413
    },
    {
      "epoch": 3.254890441894531e-06,
      "step": 3413,
      "training_step_time": 17.03909206390381
    },
    {
      "epoch": 3.2558441162109376e-06,
      "model_forward_time": 5.595844030380249,
      "step": 3414
    },
    {
      "epoch": 3.2558441162109376e-06,
      "step": 3414,
      "training_step_time": 17.07320237159729
    },
    {
      "epoch": 3.2567977905273438e-06,
      "model_forward_time": 5.612353801727295,
      "step": 3415
    },
    {
      "epoch": 3.2567977905273438e-06,
      "step": 3415,
      "training_step_time": 17.09247374534607
    },
    {
      "epoch": 3.25775146484375e-06,
      "model_forward_time": 5.601150989532471,
      "step": 3416
    },
    {
      "epoch": 3.25775146484375e-06,
      "step": 3416,
      "training_step_time": 17.09006929397583
    },
    {
      "epoch": 3.258705139160156e-06,
      "model_forward_time": 5.988196611404419,
      "step": 3417
    },
    {
      "epoch": 3.258705139160156e-06,
      "step": 3417,
      "training_step_time": 17.473838090896606
    },
    {
      "epoch": 3.2596588134765626e-06,
      "model_forward_time": 5.604284048080444,
      "step": 3418
    },
    {
      "epoch": 3.2596588134765626e-06,
      "step": 3418,
      "training_step_time": 17.07293152809143
    },
    {
      "epoch": 3.2606124877929688e-06,
      "model_forward_time": 5.5979273319244385,
      "step": 3419
    },
    {
      "epoch": 3.2606124877929688e-06,
      "step": 3419,
      "training_step_time": 17.077618837356567
    },
    {
      "epoch": 3.261566162109375e-06,
      "grad_norm": 1.1644055843353271,
      "learning_rate": 1e-05,
      "loss": 0.0809,
      "step": 3420
    },
    {
      "epoch": 3.261566162109375e-06,
      "model_forward_time": 5.5989556312561035,
      "step": 3420
    },
    {
      "dynamics_loss_avg": 0.0675888292491436,
      "epoch": 3.261566162109375e-06,
      "step": 3420
    },
    {
      "action_loss_avg": 0.017743308283388614,
      "epoch": 3.261566162109375e-06,
      "step": 3420
    },
    {
      "epoch": 3.261566162109375e-06,
      "step": 3420,
      "training_step_time": 17.079832792282104
    },
    {
      "epoch": 3.262519836425781e-06,
      "model_forward_time": 5.608508825302124,
      "step": 3421
    },
    {
      "epoch": 3.262519836425781e-06,
      "step": 3421,
      "training_step_time": 17.085134029388428
    },
    {
      "epoch": 3.2634735107421876e-06,
      "model_forward_time": 5.942922830581665,
      "step": 3422
    },
    {
      "epoch": 3.2634735107421876e-06,
      "step": 3422,
      "training_step_time": 17.421523332595825
    },
    {
      "epoch": 3.264427185058594e-06,
      "model_forward_time": 5.595909118652344,
      "step": 3423
    },
    {
      "epoch": 3.264427185058594e-06,
      "step": 3423,
      "training_step_time": 17.052370071411133
    },
    {
      "epoch": 3.265380859375e-06,
      "model_forward_time": 5.592684268951416,
      "step": 3424
    },
    {
      "epoch": 3.265380859375e-06,
      "step": 3424,
      "training_step_time": 17.029052257537842
    },
    {
      "epoch": 3.266334533691406e-06,
      "model_forward_time": 5.577336072921753,
      "step": 3425
    },
    {
      "epoch": 3.266334533691406e-06,
      "step": 3425,
      "training_step_time": 17.01908278465271
    },
    {
      "epoch": 3.2672882080078126e-06,
      "model_forward_time": 5.589422941207886,
      "step": 3426
    },
    {
      "epoch": 3.2672882080078126e-06,
      "step": 3426,
      "training_step_time": 17.03591012954712
    },
    {
      "epoch": 3.268241882324219e-06,
      "model_forward_time": 5.605635404586792,
      "step": 3427
    },
    {
      "epoch": 3.268241882324219e-06,
      "step": 3427,
      "training_step_time": 17.094959497451782
    },
    {
      "epoch": 3.269195556640625e-06,
      "model_forward_time": 5.9856908321380615,
      "step": 3428
    },
    {
      "epoch": 3.269195556640625e-06,
      "step": 3428,
      "training_step_time": 17.46241855621338
    },
    {
      "epoch": 3.270149230957031e-06,
      "model_forward_time": 5.596025466918945,
      "step": 3429
    },
    {
      "epoch": 3.270149230957031e-06,
      "step": 3429,
      "training_step_time": 17.049197912216187
    },
    {
      "epoch": 3.2711029052734377e-06,
      "grad_norm": 0.9748673439025879,
      "learning_rate": 1e-05,
      "loss": 0.097,
      "step": 3430
    },
    {
      "epoch": 3.2711029052734377e-06,
      "model_forward_time": 5.5803797245025635,
      "step": 3430
    },
    {
      "dynamics_loss_avg": 0.07445223983377218,
      "epoch": 3.2711029052734377e-06,
      "step": 3430
    },
    {
      "action_loss_avg": 0.019173936219885945,
      "epoch": 3.2711029052734377e-06,
      "step": 3430
    },
    {
      "epoch": 3.2711029052734377e-06,
      "step": 3430,
      "training_step_time": 17.008751153945923
    },
    {
      "epoch": 3.272056579589844e-06,
      "model_forward_time": 5.594036817550659,
      "step": 3431
    },
    {
      "epoch": 3.272056579589844e-06,
      "step": 3431,
      "training_step_time": 17.020124435424805
    },
    {
      "epoch": 3.27301025390625e-06,
      "model_forward_time": 5.585728645324707,
      "step": 3432
    },
    {
      "epoch": 3.27301025390625e-06,
      "step": 3432,
      "training_step_time": 17.051652431488037
    },
    {
      "epoch": 3.273963928222656e-06,
      "model_forward_time": 5.960453748703003,
      "step": 3433
    },
    {
      "epoch": 3.273963928222656e-06,
      "step": 3433,
      "training_step_time": 17.45993399620056
    },
    {
      "epoch": 3.2749176025390627e-06,
      "model_forward_time": 5.6232452392578125,
      "step": 3434
    },
    {
      "epoch": 3.2749176025390627e-06,
      "step": 3434,
      "training_step_time": 17.141525745391846
    },
    {
      "epoch": 3.275871276855469e-06,
      "model_forward_time": 5.608891487121582,
      "step": 3435
    },
    {
      "epoch": 3.275871276855469e-06,
      "step": 3435,
      "training_step_time": 17.09572196006775
    },
    {
      "epoch": 3.276824951171875e-06,
      "model_forward_time": 5.600481748580933,
      "step": 3436
    },
    {
      "epoch": 3.276824951171875e-06,
      "step": 3436,
      "training_step_time": 17.07405996322632
    },
    {
      "epoch": 3.277778625488281e-06,
      "model_forward_time": 5.597976446151733,
      "step": 3437
    },
    {
      "epoch": 3.277778625488281e-06,
      "step": 3437,
      "training_step_time": 17.06981635093689
    },
    {
      "epoch": 3.2787322998046877e-06,
      "model_forward_time": 5.5911290645599365,
      "step": 3438
    },
    {
      "epoch": 3.2787322998046877e-06,
      "step": 3438,
      "training_step_time": 17.374757766723633
    },
    {
      "epoch": 3.279685974121094e-06,
      "model_forward_time": 5.597127199172974,
      "step": 3439
    },
    {
      "epoch": 3.279685974121094e-06,
      "step": 3439,
      "training_step_time": 17.07818865776062
    },
    {
      "epoch": 3.2806396484375e-06,
      "grad_norm": 1.131205439567566,
      "learning_rate": 1e-05,
      "loss": 0.0837,
      "step": 3440
    },
    {
      "epoch": 3.2806396484375e-06,
      "model_forward_time": 5.60906720161438,
      "step": 3440
    },
    {
      "dynamics_loss_avg": 0.06684248559176922,
      "epoch": 3.2806396484375e-06,
      "step": 3440
    },
    {
      "action_loss_avg": 0.021158484183251856,
      "epoch": 3.2806396484375e-06,
      "step": 3440
    },
    {
      "epoch": 3.2806396484375e-06,
      "step": 3440,
      "training_step_time": 17.08245539665222
    },
    {
      "epoch": 3.281593322753906e-06,
      "model_forward_time": 5.608227252960205,
      "step": 3441
    },
    {
      "epoch": 3.281593322753906e-06,
      "step": 3441,
      "training_step_time": 17.100390195846558
    },
    {
      "epoch": 3.2825469970703127e-06,
      "model_forward_time": 5.608770847320557,
      "step": 3442
    },
    {
      "epoch": 3.2825469970703127e-06,
      "step": 3442,
      "training_step_time": 17.09321904182434
    },
    {
      "epoch": 3.283500671386719e-06,
      "model_forward_time": 5.606164455413818,
      "step": 3443
    },
    {
      "epoch": 3.283500671386719e-06,
      "step": 3443,
      "training_step_time": 17.071330547332764
    },
    {
      "epoch": 3.284454345703125e-06,
      "model_forward_time": 5.948925018310547,
      "step": 3444
    },
    {
      "epoch": 3.284454345703125e-06,
      "step": 3444,
      "training_step_time": 17.42955183982849
    },
    {
      "epoch": 3.285408020019531e-06,
      "model_forward_time": 5.609208583831787,
      "step": 3445
    },
    {
      "epoch": 3.285408020019531e-06,
      "step": 3445,
      "training_step_time": 17.104877471923828
    },
    {
      "epoch": 3.2863616943359377e-06,
      "model_forward_time": 5.620949983596802,
      "step": 3446
    },
    {
      "epoch": 3.2863616943359377e-06,
      "step": 3446,
      "training_step_time": 17.111929416656494
    },
    {
      "epoch": 3.287315368652344e-06,
      "model_forward_time": 5.619311094284058,
      "step": 3447
    },
    {
      "epoch": 3.287315368652344e-06,
      "step": 3447,
      "training_step_time": 17.106592416763306
    },
    {
      "epoch": 3.28826904296875e-06,
      "model_forward_time": 5.620930433273315,
      "step": 3448
    },
    {
      "epoch": 3.28826904296875e-06,
      "step": 3448,
      "training_step_time": 17.11741280555725
    },
    {
      "epoch": 3.289222717285156e-06,
      "model_forward_time": 5.995365142822266,
      "step": 3449
    },
    {
      "epoch": 3.289222717285156e-06,
      "step": 3449,
      "training_step_time": 17.461530208587646
    },
    {
      "epoch": 3.2901763916015627e-06,
      "grad_norm": 1.145883560180664,
      "learning_rate": 1e-05,
      "loss": 0.1005,
      "step": 3450
    },
    {
      "epoch": 3.2901763916015627e-06,
      "model_forward_time": 5.588327169418335,
      "step": 3450
    },
    {
      "dynamics_loss_avg": 0.07462740354239941,
      "epoch": 3.2901763916015627e-06,
      "step": 3450
    },
    {
      "action_loss_avg": 0.01820851964876056,
      "epoch": 3.2901763916015627e-06,
      "step": 3450
    },
    {
      "epoch": 3.2901763916015627e-06,
      "step": 3450,
      "training_step_time": 17.008148670196533
    },
    {
      "epoch": 3.291130065917969e-06,
      "model_forward_time": 5.563370943069458,
      "step": 3451
    },
    {
      "epoch": 3.291130065917969e-06,
      "step": 3451,
      "training_step_time": 17.004435539245605
    },
    {
      "epoch": 3.292083740234375e-06,
      "model_forward_time": 5.570903301239014,
      "step": 3452
    },
    {
      "epoch": 3.292083740234375e-06,
      "step": 3452,
      "training_step_time": 16.97096061706543
    },
    {
      "epoch": 3.293037414550781e-06,
      "model_forward_time": 5.581013202667236,
      "step": 3453
    },
    {
      "epoch": 3.293037414550781e-06,
      "step": 3453,
      "training_step_time": 17.005016088485718
    },
    {
      "epoch": 3.2939910888671877e-06,
      "model_forward_time": 5.601904392242432,
      "step": 3454
    },
    {
      "epoch": 3.2939910888671877e-06,
      "step": 3454,
      "training_step_time": 17.103068828582764
    },
    {
      "epoch": 3.294944763183594e-06,
      "model_forward_time": 5.976499319076538,
      "step": 3455
    },
    {
      "epoch": 3.294944763183594e-06,
      "step": 3455,
      "training_step_time": 17.468404054641724
    },
    {
      "epoch": 3.2958984375e-06,
      "model_forward_time": 5.607503175735474,
      "step": 3456
    },
    {
      "epoch": 3.2958984375e-06,
      "step": 3456,
      "training_step_time": 17.06987977027893
    },
    {
      "epoch": 3.296852111816406e-06,
      "model_forward_time": 5.587350368499756,
      "step": 3457
    },
    {
      "epoch": 3.296852111816406e-06,
      "step": 3457,
      "training_step_time": 17.026302337646484
    },
    {
      "epoch": 3.2978057861328127e-06,
      "model_forward_time": 5.5923333168029785,
      "step": 3458
    },
    {
      "epoch": 3.2978057861328127e-06,
      "step": 3458,
      "training_step_time": 17.035256147384644
    },
    {
      "epoch": 3.298759460449219e-06,
      "model_forward_time": 5.586492300033569,
      "step": 3459
    },
    {
      "epoch": 3.298759460449219e-06,
      "step": 3459,
      "training_step_time": 17.03329610824585
    },
    {
      "epoch": 3.299713134765625e-06,
      "grad_norm": 1.135843276977539,
      "learning_rate": 1e-05,
      "loss": 0.097,
      "step": 3460
    },
    {
      "epoch": 3.299713134765625e-06,
      "model_forward_time": 5.991382837295532,
      "step": 3460
    },
    {
      "dynamics_loss_avg": 0.0865710842423141,
      "epoch": 3.299713134765625e-06,
      "step": 3460
    },
    {
      "action_loss_avg": 0.02363488047849387,
      "epoch": 3.299713134765625e-06,
      "step": 3460
    },
    {
      "epoch": 3.299713134765625e-06,
      "step": 3460,
      "training_step_time": 17.480011224746704
    },
    {
      "epoch": 3.300666809082031e-06,
      "model_forward_time": 5.604394912719727,
      "step": 3461
    },
    {
      "epoch": 3.300666809082031e-06,
      "step": 3461,
      "training_step_time": 17.058337450027466
    },
    {
      "epoch": 3.3016204833984377e-06,
      "model_forward_time": 5.588273763656616,
      "step": 3462
    },
    {
      "epoch": 3.3016204833984377e-06,
      "step": 3462,
      "training_step_time": 17.030995845794678
    },
    {
      "epoch": 3.302574157714844e-06,
      "model_forward_time": 5.5948944091796875,
      "step": 3463
    },
    {
      "epoch": 3.302574157714844e-06,
      "step": 3463,
      "training_step_time": 17.05259418487549
    },
    {
      "epoch": 3.30352783203125e-06,
      "model_forward_time": 5.596449851989746,
      "step": 3464
    },
    {
      "epoch": 3.30352783203125e-06,
      "step": 3464,
      "training_step_time": 17.022010326385498
    },
    {
      "epoch": 3.304481506347656e-06,
      "model_forward_time": 5.5997633934021,
      "step": 3465
    },
    {
      "epoch": 3.304481506347656e-06,
      "step": 3465,
      "training_step_time": 17.035129070281982
    },
    {
      "epoch": 3.3054351806640627e-06,
      "model_forward_time": 5.927381753921509,
      "step": 3466
    },
    {
      "epoch": 3.3054351806640627e-06,
      "step": 3466,
      "training_step_time": 17.361744165420532
    },
    {
      "epoch": 3.306388854980469e-06,
      "model_forward_time": 5.598249435424805,
      "step": 3467
    },
    {
      "epoch": 3.306388854980469e-06,
      "step": 3467,
      "training_step_time": 17.070849657058716
    },
    {
      "epoch": 3.307342529296875e-06,
      "model_forward_time": 5.614943742752075,
      "step": 3468
    },
    {
      "epoch": 3.307342529296875e-06,
      "step": 3468,
      "training_step_time": 17.09971284866333
    },
    {
      "epoch": 3.308296203613281e-06,
      "model_forward_time": 5.60052227973938,
      "step": 3469
    },
    {
      "epoch": 3.308296203613281e-06,
      "step": 3469,
      "training_step_time": 17.074987411499023
    },
    {
      "epoch": 3.3092498779296877e-06,
      "grad_norm": 1.5721927881240845,
      "learning_rate": 1e-05,
      "loss": 0.0876,
      "step": 3470
    },
    {
      "epoch": 3.3092498779296877e-06,
      "model_forward_time": 5.602039813995361,
      "step": 3470
    },
    {
      "dynamics_loss_avg": 0.06496016532182694,
      "epoch": 3.3092498779296877e-06,
      "step": 3470
    },
    {
      "action_loss_avg": 0.01634368486702442,
      "epoch": 3.3092498779296877e-06,
      "step": 3470
    },
    {
      "epoch": 3.3092498779296877e-06,
      "step": 3470,
      "training_step_time": 17.054184913635254
    },
    {
      "epoch": 3.310203552246094e-06,
      "model_forward_time": 5.964690923690796,
      "step": 3471
    },
    {
      "epoch": 3.310203552246094e-06,
      "step": 3471,
      "training_step_time": 17.429040670394897
    },
    {
      "epoch": 3.3111572265625e-06,
      "model_forward_time": 5.598324775695801,
      "step": 3472
    },
    {
      "epoch": 3.3111572265625e-06,
      "step": 3472,
      "training_step_time": 17.090189933776855
    },
    {
      "epoch": 3.312110900878906e-06,
      "model_forward_time": 5.622272491455078,
      "step": 3473
    },
    {
      "epoch": 3.312110900878906e-06,
      "step": 3473,
      "training_step_time": 17.115444898605347
    },
    {
      "epoch": 3.3130645751953127e-06,
      "model_forward_time": 5.61030650138855,
      "step": 3474
    },
    {
      "epoch": 3.3130645751953127e-06,
      "step": 3474,
      "training_step_time": 17.114610195159912
    },
    {
      "epoch": 3.314018249511719e-06,
      "model_forward_time": 5.615333557128906,
      "step": 3475
    },
    {
      "epoch": 3.314018249511719e-06,
      "step": 3475,
      "training_step_time": 17.107351064682007
    },
    {
      "epoch": 3.314971923828125e-06,
      "model_forward_time": 5.609992742538452,
      "step": 3476
    },
    {
      "epoch": 3.314971923828125e-06,
      "step": 3476,
      "training_step_time": 17.071022033691406
    },
    {
      "epoch": 3.315925598144531e-06,
      "model_forward_time": 5.933651685714722,
      "step": 3477
    },
    {
      "epoch": 3.315925598144531e-06,
      "step": 3477,
      "training_step_time": 17.395925998687744
    },
    {
      "epoch": 3.3168792724609377e-06,
      "model_forward_time": 5.599306583404541,
      "step": 3478
    },
    {
      "epoch": 3.3168792724609377e-06,
      "step": 3478,
      "training_step_time": 17.063637733459473
    },
    {
      "epoch": 3.317832946777344e-06,
      "model_forward_time": 5.600970029830933,
      "step": 3479
    },
    {
      "epoch": 3.317832946777344e-06,
      "step": 3479,
      "training_step_time": 17.08234167098999
    },
    {
      "epoch": 3.31878662109375e-06,
      "grad_norm": 0.9735801219940186,
      "learning_rate": 1e-05,
      "loss": 0.0756,
      "step": 3480
    },
    {
      "epoch": 3.31878662109375e-06,
      "model_forward_time": 5.606752872467041,
      "step": 3480
    },
    {
      "dynamics_loss_avg": 0.05600418150424957,
      "epoch": 3.31878662109375e-06,
      "step": 3480
    },
    {
      "action_loss_avg": 0.012639552401378751,
      "epoch": 3.31878662109375e-06,
      "step": 3480
    },
    {
      "epoch": 3.31878662109375e-06,
      "step": 3480,
      "training_step_time": 17.071871280670166
    },
    {
      "epoch": 3.319740295410156e-06,
      "model_forward_time": 5.595471620559692,
      "step": 3481
    },
    {
      "epoch": 3.319740295410156e-06,
      "step": 3481,
      "training_step_time": 17.046324253082275
    },
    {
      "epoch": 3.3206939697265627e-06,
      "model_forward_time": 5.940096855163574,
      "step": 3482
    },
    {
      "epoch": 3.3206939697265627e-06,
      "step": 3482,
      "training_step_time": 17.378554344177246
    },
    {
      "epoch": 3.321647644042969e-06,
      "model_forward_time": 5.5967724323272705,
      "step": 3483
    },
    {
      "epoch": 3.321647644042969e-06,
      "step": 3483,
      "training_step_time": 17.0411696434021
    },
    {
      "epoch": 3.322601318359375e-06,
      "model_forward_time": 5.594976902008057,
      "step": 3484
    },
    {
      "epoch": 3.322601318359375e-06,
      "step": 3484,
      "training_step_time": 17.056532382965088
    },
    {
      "epoch": 3.323554992675781e-06,
      "model_forward_time": 5.605425596237183,
      "step": 3485
    },
    {
      "epoch": 3.323554992675781e-06,
      "step": 3485,
      "training_step_time": 17.062254190444946
    },
    {
      "epoch": 3.3245086669921877e-06,
      "model_forward_time": 5.5961830615997314,
      "step": 3486
    },
    {
      "epoch": 3.3245086669921877e-06,
      "step": 3486,
      "training_step_time": 17.050110340118408
    },
    {
      "epoch": 3.325462341308594e-06,
      "model_forward_time": 5.953181266784668,
      "step": 3487
    },
    {
      "epoch": 3.325462341308594e-06,
      "step": 3487,
      "training_step_time": 17.409732580184937
    },
    {
      "epoch": 3.326416015625e-06,
      "model_forward_time": 5.597063064575195,
      "step": 3488
    },
    {
      "epoch": 3.326416015625e-06,
      "step": 3488,
      "training_step_time": 17.048391819000244
    },
    {
      "epoch": 3.327369689941406e-06,
      "model_forward_time": 5.5961012840271,
      "step": 3489
    },
    {
      "epoch": 3.327369689941406e-06,
      "step": 3489,
      "training_step_time": 17.018083095550537
    },
    {
      "epoch": 3.3283233642578127e-06,
      "grad_norm": 1.3035160303115845,
      "learning_rate": 1e-05,
      "loss": 0.0581,
      "step": 3490
    },
    {
      "epoch": 3.3283233642578127e-06,
      "model_forward_time": 5.595689058303833,
      "step": 3490
    },
    {
      "dynamics_loss_avg": 0.052390816062688826,
      "epoch": 3.3283233642578127e-06,
      "step": 3490
    },
    {
      "action_loss_avg": 0.011646767752245069,
      "epoch": 3.3283233642578127e-06,
      "step": 3490
    },
    {
      "epoch": 3.3283233642578127e-06,
      "step": 3490,
      "training_step_time": 17.036006450653076
    },
    {
      "epoch": 3.329277038574219e-06,
      "model_forward_time": 5.598406553268433,
      "step": 3491
    },
    {
      "epoch": 3.329277038574219e-06,
      "step": 3491,
      "training_step_time": 17.0733904838562
    },
    {
      "epoch": 3.330230712890625e-06,
      "model_forward_time": 5.608458995819092,
      "step": 3492
    },
    {
      "epoch": 3.330230712890625e-06,
      "step": 3492,
      "training_step_time": 17.06713104248047
    },
    {
      "epoch": 3.331184387207031e-06,
      "model_forward_time": 5.980128049850464,
      "step": 3493
    },
    {
      "epoch": 3.331184387207031e-06,
      "step": 3493,
      "training_step_time": 17.426435708999634
    },
    {
      "epoch": 3.3321380615234377e-06,
      "model_forward_time": 5.586866140365601,
      "step": 3494
    },
    {
      "epoch": 3.3321380615234377e-06,
      "step": 3494,
      "training_step_time": 17.033528804779053
    },
    {
      "epoch": 3.333091735839844e-06,
      "model_forward_time": 5.579259157180786,
      "step": 3495
    },
    {
      "epoch": 3.333091735839844e-06,
      "step": 3495,
      "training_step_time": 17.037880420684814
    },
    {
      "epoch": 3.33404541015625e-06,
      "model_forward_time": 5.597239255905151,
      "step": 3496
    },
    {
      "epoch": 3.33404541015625e-06,
      "step": 3496,
      "training_step_time": 17.070945501327515
    },
    {
      "epoch": 3.334999084472656e-06,
      "model_forward_time": 5.607870101928711,
      "step": 3497
    },
    {
      "epoch": 3.334999084472656e-06,
      "step": 3497,
      "training_step_time": 17.073723554611206
    },
    {
      "epoch": 3.3359527587890627e-06,
      "model_forward_time": 5.589025497436523,
      "step": 3498
    },
    {
      "epoch": 3.3359527587890627e-06,
      "step": 3498,
      "training_step_time": 17.384520053863525
    },
    {
      "epoch": 3.336906433105469e-06,
      "model_forward_time": 5.586917400360107,
      "step": 3499
    },
    {
      "epoch": 3.336906433105469e-06,
      "step": 3499,
      "training_step_time": 17.004626035690308
    },
    {
      "epoch": 3.337860107421875e-06,
      "grad_norm": 1.605940580368042,
      "learning_rate": 1e-05,
      "loss": 0.0831,
      "step": 3500
    },
    {
      "epoch": 3.337860107421875e-06,
      "model_forward_time": 5.577751874923706,
      "step": 3500
    },
    {
      "dynamics_loss_avg": 0.06115956120193004,
      "epoch": 3.337860107421875e-06,
      "step": 3500
    },
    {
      "action_loss_avg": 0.01568256616592407,
      "epoch": 3.337860107421875e-06,
      "step": 3500
    },
    {
      "epoch": 3.337860107421875e-06,
      "step": 3500,
      "training_step_time": 17.01693058013916
    },
    {
      "epoch": 3.338813781738281e-06,
      "model_forward_time": 5.589225769042969,
      "step": 3501
    },
    {
      "epoch": 3.338813781738281e-06,
      "step": 3501,
      "training_step_time": 17.038159132003784
    },
    {
      "epoch": 3.3397674560546873e-06,
      "model_forward_time": 5.608539819717407,
      "step": 3502
    },
    {
      "epoch": 3.3397674560546873e-06,
      "step": 3502,
      "training_step_time": 17.114643573760986
    },
    {
      "epoch": 3.340721130371094e-06,
      "model_forward_time": 5.608366012573242,
      "step": 3503
    },
    {
      "epoch": 3.340721130371094e-06,
      "step": 3503,
      "training_step_time": 17.115620851516724
    },
    {
      "epoch": 3.3416748046875e-06,
      "model_forward_time": 5.9757843017578125,
      "step": 3504
    },
    {
      "epoch": 3.3416748046875e-06,
      "step": 3504,
      "training_step_time": 17.470266103744507
    },
    {
      "epoch": 3.342628479003906e-06,
      "model_forward_time": 5.588972091674805,
      "step": 3505
    },
    {
      "epoch": 3.342628479003906e-06,
      "step": 3505,
      "training_step_time": 17.036117792129517
    },
    {
      "epoch": 3.3435821533203123e-06,
      "model_forward_time": 5.570321798324585,
      "step": 3506
    },
    {
      "epoch": 3.3435821533203123e-06,
      "step": 3506,
      "training_step_time": 16.982001304626465
    },
    {
      "epoch": 3.344535827636719e-06,
      "model_forward_time": 5.56565260887146,
      "step": 3507
    },
    {
      "epoch": 3.344535827636719e-06,
      "step": 3507,
      "training_step_time": 16.984212160110474
    },
    {
      "epoch": 3.345489501953125e-06,
      "model_forward_time": 5.565769910812378,
      "step": 3508
    },
    {
      "epoch": 3.345489501953125e-06,
      "step": 3508,
      "training_step_time": 16.997347354888916
    },
    {
      "epoch": 3.346443176269531e-06,
      "model_forward_time": 5.566933870315552,
      "step": 3509
    },
    {
      "epoch": 3.346443176269531e-06,
      "step": 3509,
      "training_step_time": 17.34523916244507
    },
    {
      "epoch": 3.3473968505859373e-06,
      "grad_norm": 1.335598111152649,
      "learning_rate": 1e-05,
      "loss": 0.0954,
      "step": 3510
    },
    {
      "epoch": 3.3473968505859373e-06,
      "model_forward_time": 5.5850090980529785,
      "step": 3510
    },
    {
      "dynamics_loss_avg": 0.08229109942913056,
      "epoch": 3.3473968505859373e-06,
      "step": 3510
    },
    {
      "action_loss_avg": 0.021059786528348924,
      "epoch": 3.3473968505859373e-06,
      "step": 3510
    },
    {
      "epoch": 3.3473968505859373e-06,
      "step": 3510,
      "training_step_time": 17.0385684967041
    },
    {
      "epoch": 3.348350524902344e-06,
      "model_forward_time": 5.598492622375488,
      "step": 3511
    },
    {
      "epoch": 3.348350524902344e-06,
      "step": 3511,
      "training_step_time": 17.079068660736084
    },
    {
      "epoch": 3.34930419921875e-06,
      "model_forward_time": 5.618403434753418,
      "step": 3512
    },
    {
      "epoch": 3.34930419921875e-06,
      "step": 3512,
      "training_step_time": 17.104423761367798
    },
    {
      "epoch": 3.350257873535156e-06,
      "model_forward_time": 5.60902738571167,
      "step": 3513
    },
    {
      "epoch": 3.350257873535156e-06,
      "step": 3513,
      "training_step_time": 17.104844093322754
    },
    {
      "epoch": 3.3512115478515623e-06,
      "model_forward_time": 5.609307765960693,
      "step": 3514
    },
    {
      "epoch": 3.3512115478515623e-06,
      "step": 3514,
      "training_step_time": 17.113210678100586
    },
    {
      "epoch": 3.352165222167969e-06,
      "model_forward_time": 5.969758033752441,
      "step": 3515
    },
    {
      "epoch": 3.352165222167969e-06,
      "step": 3515,
      "training_step_time": 17.438099145889282
    },
    {
      "epoch": 3.353118896484375e-06,
      "model_forward_time": 5.589231729507446,
      "step": 3516
    },
    {
      "epoch": 3.353118896484375e-06,
      "step": 3516,
      "training_step_time": 17.05109429359436
    },
    {
      "epoch": 3.354072570800781e-06,
      "model_forward_time": 5.579703092575073,
      "step": 3517
    },
    {
      "epoch": 3.354072570800781e-06,
      "step": 3517,
      "training_step_time": 17.022300004959106
    },
    {
      "epoch": 3.3550262451171873e-06,
      "model_forward_time": 5.601792812347412,
      "step": 3518
    },
    {
      "epoch": 3.3550262451171873e-06,
      "step": 3518,
      "training_step_time": 17.081918716430664
    },
    {
      "epoch": 3.355979919433594e-06,
      "model_forward_time": 5.604725360870361,
      "step": 3519
    },
    {
      "epoch": 3.355979919433594e-06,
      "step": 3519,
      "training_step_time": 17.07009983062744
    },
    {
      "epoch": 3.35693359375e-06,
      "grad_norm": 1.483139991760254,
      "learning_rate": 1e-05,
      "loss": 0.0951,
      "step": 3520
    },
    {
      "epoch": 3.35693359375e-06,
      "model_forward_time": 5.9629528522491455,
      "step": 3520
    },
    {
      "dynamics_loss_avg": 0.07724562324583531,
      "epoch": 3.35693359375e-06,
      "step": 3520
    },
    {
      "action_loss_avg": 0.017593686981126667,
      "epoch": 3.35693359375e-06,
      "step": 3520
    },
    {
      "epoch": 3.35693359375e-06,
      "step": 3520,
      "training_step_time": 17.39660382270813
    },
    {
      "epoch": 3.357887268066406e-06,
      "model_forward_time": 5.591085195541382,
      "step": 3521
    },
    {
      "epoch": 3.357887268066406e-06,
      "step": 3521,
      "training_step_time": 17.04764676094055
    },
    {
      "epoch": 3.3588409423828123e-06,
      "model_forward_time": 5.591894149780273,
      "step": 3522
    },
    {
      "epoch": 3.3588409423828123e-06,
      "step": 3522,
      "training_step_time": 17.092376232147217
    },
    {
      "epoch": 3.359794616699219e-06,
      "model_forward_time": 5.623008966445923,
      "step": 3523
    },
    {
      "epoch": 3.359794616699219e-06,
      "step": 3523,
      "training_step_time": 17.13232946395874
    },
    {
      "epoch": 3.360748291015625e-06,
      "model_forward_time": 5.620097637176514,
      "step": 3524
    },
    {
      "epoch": 3.360748291015625e-06,
      "step": 3524,
      "training_step_time": 17.135141849517822
    },
    {
      "epoch": 3.361701965332031e-06,
      "model_forward_time": 5.606100797653198,
      "step": 3525
    },
    {
      "epoch": 3.361701965332031e-06,
      "step": 3525,
      "training_step_time": 17.104446411132812
    },
    {
      "epoch": 3.3626556396484373e-06,
      "model_forward_time": 5.9781389236450195,
      "step": 3526
    },
    {
      "epoch": 3.3626556396484373e-06,
      "step": 3526,
      "training_step_time": 17.473334789276123
    },
    {
      "epoch": 3.363609313964844e-06,
      "model_forward_time": 5.6117844581604,
      "step": 3527
    },
    {
      "epoch": 3.363609313964844e-06,
      "step": 3527,
      "training_step_time": 17.09801983833313
    },
    {
      "epoch": 3.36456298828125e-06,
      "model_forward_time": 5.600008964538574,
      "step": 3528
    },
    {
      "epoch": 3.36456298828125e-06,
      "step": 3528,
      "training_step_time": 17.091243028640747
    },
    {
      "epoch": 3.365516662597656e-06,
      "model_forward_time": 5.608938455581665,
      "step": 3529
    },
    {
      "epoch": 3.365516662597656e-06,
      "step": 3529,
      "training_step_time": 17.078948497772217
    },
    {
      "epoch": 3.3664703369140623e-06,
      "grad_norm": 1.4945894479751587,
      "learning_rate": 1e-05,
      "loss": 0.1052,
      "step": 3530
    },
    {
      "epoch": 3.3664703369140623e-06,
      "model_forward_time": 5.595888137817383,
      "step": 3530
    },
    {
      "dynamics_loss_avg": 0.0851038597524166,
      "epoch": 3.3664703369140623e-06,
      "step": 3530
    },
    {
      "action_loss_avg": 0.01953773628920317,
      "epoch": 3.3664703369140623e-06,
      "step": 3530
    },
    {
      "epoch": 3.3664703369140623e-06,
      "step": 3530,
      "training_step_time": 17.069296836853027
    },
    {
      "epoch": 3.367424011230469e-06,
      "model_forward_time": 5.602383136749268,
      "step": 3531
    },
    {
      "epoch": 3.367424011230469e-06,
      "step": 3531,
      "training_step_time": 17.432841062545776
    },
    {
      "epoch": 3.368377685546875e-06,
      "model_forward_time": 5.6025190353393555,
      "step": 3532
    },
    {
      "epoch": 3.368377685546875e-06,
      "step": 3532,
      "training_step_time": 17.076927661895752
    },
    {
      "epoch": 3.369331359863281e-06,
      "model_forward_time": 5.600569009780884,
      "step": 3533
    },
    {
      "epoch": 3.369331359863281e-06,
      "step": 3533,
      "training_step_time": 17.091492891311646
    },
    {
      "epoch": 3.3702850341796873e-06,
      "model_forward_time": 5.613965749740601,
      "step": 3534
    },
    {
      "epoch": 3.3702850341796873e-06,
      "step": 3534,
      "training_step_time": 17.09716296195984
    },
    {
      "epoch": 3.371238708496094e-06,
      "model_forward_time": 5.6060545444488525,
      "step": 3535
    },
    {
      "epoch": 3.371238708496094e-06,
      "step": 3535,
      "training_step_time": 17.101741552352905
    },
    {
      "epoch": 3.3721923828125e-06,
      "model_forward_time": 5.608293533325195,
      "step": 3536
    },
    {
      "epoch": 3.3721923828125e-06,
      "step": 3536,
      "training_step_time": 17.09627628326416
    },
    {
      "epoch": 3.373146057128906e-06,
      "model_forward_time": 5.9823291301727295,
      "step": 3537
    },
    {
      "epoch": 3.373146057128906e-06,
      "step": 3537,
      "training_step_time": 17.464762687683105
    },
    {
      "epoch": 3.3740997314453123e-06,
      "model_forward_time": 5.603023052215576,
      "step": 3538
    },
    {
      "epoch": 3.3740997314453123e-06,
      "step": 3538,
      "training_step_time": 17.051209688186646
    },
    {
      "epoch": 3.375053405761719e-06,
      "model_forward_time": 5.596871376037598,
      "step": 3539
    },
    {
      "epoch": 3.375053405761719e-06,
      "step": 3539,
      "training_step_time": 17.051830291748047
    },
    {
      "epoch": 3.376007080078125e-06,
      "grad_norm": 1.6753482818603516,
      "learning_rate": 1e-05,
      "loss": 0.0682,
      "step": 3540
    },
    {
      "epoch": 3.376007080078125e-06,
      "model_forward_time": 5.581770181655884,
      "step": 3540
    },
    {
      "dynamics_loss_avg": 0.051048201881349085,
      "epoch": 3.376007080078125e-06,
      "step": 3540
    },
    {
      "action_loss_avg": 0.010712608636822551,
      "epoch": 3.376007080078125e-06,
      "step": 3540
    },
    {
      "epoch": 3.376007080078125e-06,
      "step": 3540,
      "training_step_time": 17.005007028579712
    },
    {
      "epoch": 3.376960754394531e-06,
      "model_forward_time": 5.595435857772827,
      "step": 3541
    },
    {
      "epoch": 3.376960754394531e-06,
      "step": 3541,
      "training_step_time": 17.04598569869995
    },
    {
      "epoch": 3.3779144287109373e-06,
      "model_forward_time": 5.959747791290283,
      "step": 3542
    },
    {
      "epoch": 3.3779144287109373e-06,
      "step": 3542,
      "training_step_time": 17.414733409881592
    },
    {
      "epoch": 3.378868103027344e-06,
      "model_forward_time": 5.604723930358887,
      "step": 3543
    },
    {
      "epoch": 3.378868103027344e-06,
      "step": 3543,
      "training_step_time": 17.067764043807983
    },
    {
      "epoch": 3.37982177734375e-06,
      "model_forward_time": 5.5928449630737305,
      "step": 3544
    },
    {
      "epoch": 3.37982177734375e-06,
      "step": 3544,
      "training_step_time": 17.04453945159912
    },
    {
      "epoch": 3.380775451660156e-06,
      "model_forward_time": 5.592277765274048,
      "step": 3545
    },
    {
      "epoch": 3.380775451660156e-06,
      "step": 3545,
      "training_step_time": 17.05520796775818
    },
    {
      "epoch": 3.3817291259765623e-06,
      "model_forward_time": 5.599872827529907,
      "step": 3546
    },
    {
      "epoch": 3.3817291259765623e-06,
      "step": 3546,
      "training_step_time": 17.060242652893066
    },
    {
      "epoch": 3.382682800292969e-06,
      "model_forward_time": 5.608869314193726,
      "step": 3547
    },
    {
      "epoch": 3.382682800292969e-06,
      "step": 3547,
      "training_step_time": 17.07291007041931
    },
    {
      "epoch": 3.383636474609375e-06,
      "model_forward_time": 5.972220182418823,
      "step": 3548
    },
    {
      "epoch": 3.383636474609375e-06,
      "step": 3548,
      "training_step_time": 17.46138620376587
    },
    {
      "epoch": 3.384590148925781e-06,
      "model_forward_time": 5.608657121658325,
      "step": 3549
    },
    {
      "epoch": 3.384590148925781e-06,
      "step": 3549,
      "training_step_time": 17.099536418914795
    },
    {
      "epoch": 3.3855438232421873e-06,
      "grad_norm": 0.648715615272522,
      "learning_rate": 1e-05,
      "loss": 0.0636,
      "step": 3550
    },
    {
      "epoch": 3.3855438232421873e-06,
      "model_forward_time": 5.611172914505005,
      "step": 3550
    },
    {
      "dynamics_loss_avg": 0.055065279453992845,
      "epoch": 3.3855438232421873e-06,
      "step": 3550
    },
    {
      "action_loss_avg": 0.010632142005488277,
      "epoch": 3.3855438232421873e-06,
      "step": 3550
    },
    {
      "epoch": 3.3855438232421873e-06,
      "step": 3550,
      "training_step_time": 17.08651113510132
    },
    {
      "epoch": 3.386497497558594e-06,
      "model_forward_time": 5.62198805809021,
      "step": 3551
    },
    {
      "epoch": 3.386497497558594e-06,
      "step": 3551,
      "training_step_time": 17.11749577522278
    },
    {
      "epoch": 3.387451171875e-06,
      "model_forward_time": 5.61769437789917,
      "step": 3552
    },
    {
      "epoch": 3.387451171875e-06,
      "step": 3552,
      "training_step_time": 17.123574018478394
    },
    {
      "epoch": 3.388404846191406e-06,
      "model_forward_time": 5.954481363296509,
      "step": 3553
    },
    {
      "epoch": 3.388404846191406e-06,
      "step": 3553,
      "training_step_time": 17.448279857635498
    },
    {
      "epoch": 3.3893585205078123e-06,
      "model_forward_time": 5.607049465179443,
      "step": 3554
    },
    {
      "epoch": 3.3893585205078123e-06,
      "step": 3554,
      "training_step_time": 17.073330640792847
    },
    {
      "epoch": 3.390312194824219e-06,
      "model_forward_time": 5.593250513076782,
      "step": 3555
    },
    {
      "epoch": 3.390312194824219e-06,
      "step": 3555,
      "training_step_time": 17.0313777923584
    },
    {
      "epoch": 3.391265869140625e-06,
      "model_forward_time": 5.5764381885528564,
      "step": 3556
    },
    {
      "epoch": 3.391265869140625e-06,
      "step": 3556,
      "training_step_time": 16.99686861038208
    },
    {
      "epoch": 3.392219543457031e-06,
      "model_forward_time": 5.574980735778809,
      "step": 3557
    },
    {
      "epoch": 3.392219543457031e-06,
      "step": 3557,
      "training_step_time": 17.003516912460327
    },
    {
      "epoch": 3.3931732177734373e-06,
      "model_forward_time": 5.570683002471924,
      "step": 3558
    },
    {
      "epoch": 3.3931732177734373e-06,
      "step": 3558,
      "training_step_time": 17.38587975502014
    },
    {
      "epoch": 3.394126892089844e-06,
      "model_forward_time": 5.565075874328613,
      "step": 3559
    },
    {
      "epoch": 3.394126892089844e-06,
      "step": 3559,
      "training_step_time": 16.981664896011353
    },
    {
      "epoch": 3.39508056640625e-06,
      "grad_norm": 0.15418055653572083,
      "learning_rate": 1e-05,
      "loss": 0.0513,
      "step": 3560
    },
    {
      "epoch": 3.39508056640625e-06,
      "model_forward_time": 5.5764429569244385,
      "step": 3560
    },
    {
      "dynamics_loss_avg": 0.04227983709424734,
      "epoch": 3.39508056640625e-06,
      "step": 3560
    },
    {
      "action_loss_avg": 0.0069349403493106365,
      "epoch": 3.39508056640625e-06,
      "step": 3560
    },
    {
      "epoch": 3.39508056640625e-06,
      "step": 3560,
      "training_step_time": 16.98993968963623
    },
    {
      "epoch": 3.396034240722656e-06,
      "model_forward_time": 5.584901571273804,
      "step": 3561
    },
    {
      "epoch": 3.396034240722656e-06,
      "step": 3561,
      "training_step_time": 16.998978853225708
    },
    {
      "epoch": 3.3969879150390623e-06,
      "model_forward_time": 5.586889266967773,
      "step": 3562
    },
    {
      "epoch": 3.3969879150390623e-06,
      "step": 3562,
      "training_step_time": 17.053589820861816
    },
    {
      "epoch": 3.397941589355469e-06,
      "model_forward_time": 5.59775710105896,
      "step": 3563
    },
    {
      "epoch": 3.397941589355469e-06,
      "step": 3563,
      "training_step_time": 17.05991554260254
    },
    {
      "epoch": 3.398895263671875e-06,
      "model_forward_time": 5.954335451126099,
      "step": 3564
    },
    {
      "epoch": 3.398895263671875e-06,
      "step": 3564,
      "training_step_time": 17.411876440048218
    },
    {
      "epoch": 3.399848937988281e-06,
      "model_forward_time": 5.606755971908569,
      "step": 3565
    },
    {
      "epoch": 3.399848937988281e-06,
      "step": 3565,
      "training_step_time": 17.065468072891235
    },
    {
      "epoch": 3.4008026123046873e-06,
      "model_forward_time": 5.6362481117248535,
      "step": 3566
    },
    {
      "epoch": 3.4008026123046873e-06,
      "step": 3566,
      "training_step_time": 17.11995005607605
    },
    {
      "epoch": 3.401756286621094e-06,
      "model_forward_time": 5.6028783321380615,
      "step": 3567
    },
    {
      "epoch": 3.401756286621094e-06,
      "step": 3567,
      "training_step_time": 17.076625108718872
    },
    {
      "epoch": 3.4027099609375e-06,
      "model_forward_time": 5.601106643676758,
      "step": 3568
    },
    {
      "epoch": 3.4027099609375e-06,
      "step": 3568,
      "training_step_time": 17.066136598587036
    },
    {
      "epoch": 3.403663635253906e-06,
      "model_forward_time": 5.60483717918396,
      "step": 3569
    },
    {
      "epoch": 3.403663635253906e-06,
      "step": 3569,
      "training_step_time": 17.437302589416504
    },
    {
      "epoch": 3.4046173095703123e-06,
      "grad_norm": 1.6996822357177734,
      "learning_rate": 1e-05,
      "loss": 0.0759,
      "step": 3570
    },
    {
      "epoch": 3.4046173095703123e-06,
      "model_forward_time": 5.595771074295044,
      "step": 3570
    },
    {
      "dynamics_loss_avg": 0.06200261041522026,
      "epoch": 3.4046173095703123e-06,
      "step": 3570
    },
    {
      "action_loss_avg": 0.013461630558595061,
      "epoch": 3.4046173095703123e-06,
      "step": 3570
    },
    {
      "epoch": 3.4046173095703123e-06,
      "step": 3570,
      "training_step_time": 17.06949234008789
    },
    {
      "epoch": 3.405570983886719e-06,
      "model_forward_time": 5.598340034484863,
      "step": 3571
    },
    {
      "epoch": 3.405570983886719e-06,
      "step": 3571,
      "training_step_time": 17.055023431777954
    },
    {
      "epoch": 3.406524658203125e-06,
      "model_forward_time": 5.597140550613403,
      "step": 3572
    },
    {
      "epoch": 3.406524658203125e-06,
      "step": 3572,
      "training_step_time": 17.046155214309692
    },
    {
      "epoch": 3.407478332519531e-06,
      "model_forward_time": 5.592805862426758,
      "step": 3573
    },
    {
      "epoch": 3.407478332519531e-06,
      "step": 3573,
      "training_step_time": 17.034075498580933
    },
    {
      "epoch": 3.4084320068359374e-06,
      "model_forward_time": 5.590313673019409,
      "step": 3574
    },
    {
      "epoch": 3.4084320068359374e-06,
      "step": 3574,
      "training_step_time": 17.03195357322693
    },
    {
      "epoch": 3.409385681152344e-06,
      "model_forward_time": 5.932544231414795,
      "step": 3575
    },
    {
      "epoch": 3.409385681152344e-06,
      "step": 3575,
      "training_step_time": 17.37176537513733
    },
    {
      "epoch": 3.41033935546875e-06,
      "model_forward_time": 5.590990781784058,
      "step": 3576
    },
    {
      "epoch": 3.41033935546875e-06,
      "step": 3576,
      "training_step_time": 17.023799419403076
    },
    {
      "epoch": 3.411293029785156e-06,
      "model_forward_time": 5.593527793884277,
      "step": 3577
    },
    {
      "epoch": 3.411293029785156e-06,
      "step": 3577,
      "training_step_time": 17.029876708984375
    },
    {
      "epoch": 3.4122467041015624e-06,
      "model_forward_time": 5.626510858535767,
      "step": 3578
    },
    {
      "epoch": 3.4122467041015624e-06,
      "step": 3578,
      "training_step_time": 17.07512879371643
    },
    {
      "epoch": 3.413200378417969e-06,
      "model_forward_time": 5.589181900024414,
      "step": 3579
    },
    {
      "epoch": 3.413200378417969e-06,
      "step": 3579,
      "training_step_time": 17.05793857574463
    },
    {
      "epoch": 3.414154052734375e-06,
      "grad_norm": 0.41769298911094666,
      "learning_rate": 1e-05,
      "loss": 0.0847,
      "step": 3580
    },
    {
      "epoch": 3.414154052734375e-06,
      "model_forward_time": 5.984778165817261,
      "step": 3580
    },
    {
      "dynamics_loss_avg": 0.07517358995974063,
      "epoch": 3.414154052734375e-06,
      "step": 3580
    },
    {
      "action_loss_avg": 0.018703609332442285,
      "epoch": 3.414154052734375e-06,
      "step": 3580
    },
    {
      "epoch": 3.414154052734375e-06,
      "step": 3580,
      "training_step_time": 17.453401803970337
    },
    {
      "epoch": 3.4151077270507812e-06,
      "model_forward_time": 5.610008239746094,
      "step": 3581
    },
    {
      "epoch": 3.4151077270507812e-06,
      "step": 3581,
      "training_step_time": 17.1053626537323
    },
    {
      "epoch": 3.4160614013671874e-06,
      "model_forward_time": 5.615829706192017,
      "step": 3582
    },
    {
      "epoch": 3.4160614013671874e-06,
      "step": 3582,
      "training_step_time": 17.10246753692627
    },
    {
      "epoch": 3.417015075683594e-06,
      "model_forward_time": 5.623856544494629,
      "step": 3583
    },
    {
      "epoch": 3.417015075683594e-06,
      "step": 3583,
      "training_step_time": 17.112813234329224
    },
    {
      "epoch": 3.41796875e-06,
      "model_forward_time": 5.602742910385132,
      "step": 3584
    },
    {
      "epoch": 3.41796875e-06,
      "step": 3584,
      "training_step_time": 17.06927490234375
    },
    {
      "epoch": 3.4189224243164062e-06,
      "model_forward_time": 5.5893449783325195,
      "step": 3585
    },
    {
      "epoch": 3.4189224243164062e-06,
      "step": 3585,
      "training_step_time": 17.407890558242798
    },
    {
      "epoch": 3.4198760986328124e-06,
      "model_forward_time": 5.621306896209717,
      "step": 3586
    },
    {
      "epoch": 3.4198760986328124e-06,
      "step": 3586,
      "training_step_time": 17.106070280075073
    },
    {
      "epoch": 3.420829772949219e-06,
      "model_forward_time": 5.600621938705444,
      "step": 3587
    },
    {
      "epoch": 3.420829772949219e-06,
      "step": 3587,
      "training_step_time": 17.07709312438965
    },
    {
      "epoch": 3.421783447265625e-06,
      "model_forward_time": 5.6009297370910645,
      "step": 3588
    },
    {
      "epoch": 3.421783447265625e-06,
      "step": 3588,
      "training_step_time": 17.05518913269043
    },
    {
      "epoch": 3.4227371215820312e-06,
      "model_forward_time": 5.580893278121948,
      "step": 3589
    },
    {
      "epoch": 3.4227371215820312e-06,
      "step": 3589,
      "training_step_time": 17.014440774917603
    },
    {
      "epoch": 3.4236907958984374e-06,
      "grad_norm": 0.7673053741455078,
      "learning_rate": 1e-05,
      "loss": 0.091,
      "step": 3590
    },
    {
      "epoch": 3.4236907958984374e-06,
      "model_forward_time": 5.590529203414917,
      "step": 3590
    },
    {
      "dynamics_loss_avg": 0.06649495325982571,
      "epoch": 3.4236907958984374e-06,
      "step": 3590
    },
    {
      "action_loss_avg": 0.0179061371833086,
      "epoch": 3.4236907958984374e-06,
      "step": 3590
    },
    {
      "epoch": 3.4236907958984374e-06,
      "step": 3590,
      "training_step_time": 17.002022981643677
    },
    {
      "epoch": 3.424644470214844e-06,
      "model_forward_time": 5.960843801498413,
      "step": 3591
    },
    {
      "epoch": 3.424644470214844e-06,
      "step": 3591,
      "training_step_time": 17.431641340255737
    },
    {
      "epoch": 3.42559814453125e-06,
      "model_forward_time": 5.610949754714966,
      "step": 3592
    },
    {
      "epoch": 3.42559814453125e-06,
      "step": 3592,
      "training_step_time": 17.1249783039093
    },
    {
      "epoch": 3.4265518188476562e-06,
      "model_forward_time": 5.620631694793701,
      "step": 3593
    },
    {
      "epoch": 3.4265518188476562e-06,
      "step": 3593,
      "training_step_time": 17.11630916595459
    },
    {
      "epoch": 3.4275054931640624e-06,
      "model_forward_time": 5.602706432342529,
      "step": 3594
    },
    {
      "epoch": 3.4275054931640624e-06,
      "step": 3594,
      "training_step_time": 17.060616731643677
    },
    {
      "epoch": 3.428459167480469e-06,
      "model_forward_time": 5.585627555847168,
      "step": 3595
    },
    {
      "epoch": 3.428459167480469e-06,
      "step": 3595,
      "training_step_time": 17.02990961074829
    },
    {
      "epoch": 3.429412841796875e-06,
      "model_forward_time": 5.948427438735962,
      "step": 3596
    },
    {
      "epoch": 3.429412841796875e-06,
      "step": 3596,
      "training_step_time": 17.38389277458191
    },
    {
      "epoch": 3.4303665161132812e-06,
      "model_forward_time": 5.598974943161011,
      "step": 3597
    },
    {
      "epoch": 3.4303665161132812e-06,
      "step": 3597,
      "training_step_time": 17.07891345024109
    },
    {
      "epoch": 3.4313201904296874e-06,
      "model_forward_time": 5.596491813659668,
      "step": 3598
    },
    {
      "epoch": 3.4313201904296874e-06,
      "step": 3598,
      "training_step_time": 17.075517416000366
    },
    {
      "epoch": 3.432273864746094e-06,
      "model_forward_time": 5.614826440811157,
      "step": 3599
    },
    {
      "epoch": 3.432273864746094e-06,
      "step": 3599,
      "training_step_time": 17.10196590423584
    },
    {
      "epoch": 3.4332275390625e-06,
      "grad_norm": 1.3408300876617432,
      "learning_rate": 1e-05,
      "loss": 0.0799,
      "step": 3600
    },
    {
      "epoch": 3.4332275390625e-06,
      "model_forward_time": 5.618149280548096,
      "step": 3600
    },
    {
      "dynamics_loss_avg": 0.06362243443727493,
      "epoch": 3.4332275390625e-06,
      "step": 3600
    },
    {
      "action_loss_avg": 0.01610429468564689,
      "epoch": 3.4332275390625e-06,
      "step": 3600
    },
    {
      "epoch": 3.4332275390625e-06,
      "step": 3600,
      "training_step_time": 17.10021996498108
    },
    {
      "epoch": 3.4341812133789062e-06,
      "model_forward_time": 5.59344220161438,
      "step": 3601
    },
    {
      "epoch": 3.4341812133789062e-06,
      "step": 3601,
      "training_step_time": 17.043628931045532
    },
    {
      "epoch": 3.4351348876953124e-06,
      "model_forward_time": 5.927227973937988,
      "step": 3602
    },
    {
      "epoch": 3.4351348876953124e-06,
      "step": 3602,
      "training_step_time": 17.34235453605652
    },
    {
      "epoch": 3.436088562011719e-06,
      "model_forward_time": 5.578307151794434,
      "step": 3603
    },
    {
      "epoch": 3.436088562011719e-06,
      "step": 3603,
      "training_step_time": 17.001704216003418
    },
    {
      "epoch": 3.437042236328125e-06,
      "model_forward_time": 5.577747344970703,
      "step": 3604
    },
    {
      "epoch": 3.437042236328125e-06,
      "step": 3604,
      "training_step_time": 16.999915838241577
    },
    {
      "epoch": 3.4379959106445312e-06,
      "model_forward_time": 5.577713966369629,
      "step": 3605
    },
    {
      "epoch": 3.4379959106445312e-06,
      "step": 3605,
      "training_step_time": 17.003265857696533
    },
    {
      "epoch": 3.4389495849609374e-06,
      "model_forward_time": 5.58345627784729,
      "step": 3606
    },
    {
      "epoch": 3.4389495849609374e-06,
      "step": 3606,
      "training_step_time": 17.01130700111389
    },
    {
      "epoch": 3.439903259277344e-06,
      "model_forward_time": 5.9453794956207275,
      "step": 3607
    },
    {
      "epoch": 3.439903259277344e-06,
      "step": 3607,
      "training_step_time": 17.42442274093628
    },
    {
      "epoch": 3.44085693359375e-06,
      "model_forward_time": 5.624148607254028,
      "step": 3608
    },
    {
      "epoch": 3.44085693359375e-06,
      "step": 3608,
      "training_step_time": 17.144709587097168
    },
    {
      "epoch": 3.4418106079101562e-06,
      "model_forward_time": 5.6265459060668945,
      "step": 3609
    },
    {
      "epoch": 3.4418106079101562e-06,
      "step": 3609,
      "training_step_time": 17.128886938095093
    },
    {
      "epoch": 3.4427642822265624e-06,
      "grad_norm": 2.1181886196136475,
      "learning_rate": 1e-05,
      "loss": 0.078,
      "step": 3610
    },
    {
      "epoch": 3.4427642822265624e-06,
      "model_forward_time": 5.628391981124878,
      "step": 3610
    },
    {
      "dynamics_loss_avg": 0.05897921156138182,
      "epoch": 3.4427642822265624e-06,
      "step": 3610
    },
    {
      "action_loss_avg": 0.016176802571862936,
      "epoch": 3.4427642822265624e-06,
      "step": 3610
    },
    {
      "epoch": 3.4427642822265624e-06,
      "step": 3610,
      "training_step_time": 17.104182720184326
    },
    {
      "epoch": 3.443717956542969e-06,
      "model_forward_time": 5.602032423019409,
      "step": 3611
    },
    {
      "epoch": 3.443717956542969e-06,
      "step": 3611,
      "training_step_time": 17.05125856399536
    },
    {
      "epoch": 3.444671630859375e-06,
      "model_forward_time": 5.59104061126709,
      "step": 3612
    },
    {
      "epoch": 3.444671630859375e-06,
      "step": 3612,
      "training_step_time": 17.057180881500244
    },
    {
      "epoch": 3.4456253051757812e-06,
      "model_forward_time": 5.961536884307861,
      "step": 3613
    },
    {
      "epoch": 3.4456253051757812e-06,
      "step": 3613,
      "training_step_time": 17.421122074127197
    },
    {
      "epoch": 3.4465789794921874e-06,
      "model_forward_time": 5.604591131210327,
      "step": 3614
    },
    {
      "epoch": 3.4465789794921874e-06,
      "step": 3614,
      "training_step_time": 17.075515031814575
    },
    {
      "epoch": 3.447532653808594e-06,
      "model_forward_time": 5.606217861175537,
      "step": 3615
    },
    {
      "epoch": 3.447532653808594e-06,
      "step": 3615,
      "training_step_time": 17.066057682037354
    },
    {
      "epoch": 3.448486328125e-06,
      "model_forward_time": 5.585850715637207,
      "step": 3616
    },
    {
      "epoch": 3.448486328125e-06,
      "step": 3616,
      "training_step_time": 17.037437677383423
    },
    {
      "epoch": 3.4494400024414062e-06,
      "model_forward_time": 5.5857932567596436,
      "step": 3617
    },
    {
      "epoch": 3.4494400024414062e-06,
      "step": 3617,
      "training_step_time": 17.02420926094055
    },
    {
      "epoch": 3.4503936767578124e-06,
      "model_forward_time": 5.951207399368286,
      "step": 3618
    },
    {
      "epoch": 3.4503936767578124e-06,
      "step": 3618,
      "training_step_time": 17.408387422561646
    },
    {
      "epoch": 3.451347351074219e-06,
      "model_forward_time": 5.598111152648926,
      "step": 3619
    },
    {
      "epoch": 3.451347351074219e-06,
      "step": 3619,
      "training_step_time": 17.065407276153564
    },
    {
      "epoch": 3.452301025390625e-06,
      "grad_norm": 1.1277230978012085,
      "learning_rate": 1e-05,
      "loss": 0.0732,
      "step": 3620
    },
    {
      "epoch": 3.452301025390625e-06,
      "model_forward_time": 5.619774103164673,
      "step": 3620
    },
    {
      "dynamics_loss_avg": 0.06130542811006308,
      "epoch": 3.452301025390625e-06,
      "step": 3620
    },
    {
      "action_loss_avg": 0.012857417250052095,
      "epoch": 3.452301025390625e-06,
      "step": 3620
    },
    {
      "epoch": 3.452301025390625e-06,
      "step": 3620,
      "training_step_time": 17.10674738883972
    },
    {
      "epoch": 3.4532546997070312e-06,
      "model_forward_time": 5.602282285690308,
      "step": 3621
    },
    {
      "epoch": 3.4532546997070312e-06,
      "step": 3621,
      "training_step_time": 17.091883182525635
    },
    {
      "epoch": 3.4542083740234374e-06,
      "model_forward_time": 5.615037679672241,
      "step": 3622
    },
    {
      "epoch": 3.4542083740234374e-06,
      "step": 3622,
      "training_step_time": 17.100006103515625
    },
    {
      "epoch": 3.455162048339844e-06,
      "model_forward_time": 5.618349313735962,
      "step": 3623
    },
    {
      "epoch": 3.455162048339844e-06,
      "step": 3623,
      "training_step_time": 17.44599986076355
    },
    {
      "epoch": 3.45611572265625e-06,
      "model_forward_time": 5.600704669952393,
      "step": 3624
    },
    {
      "epoch": 3.45611572265625e-06,
      "step": 3624,
      "training_step_time": 17.050421237945557
    },
    {
      "epoch": 3.4570693969726562e-06,
      "model_forward_time": 5.58728814125061,
      "step": 3625
    },
    {
      "epoch": 3.4570693969726562e-06,
      "step": 3625,
      "training_step_time": 17.007423400878906
    },
    {
      "epoch": 3.4580230712890624e-06,
      "model_forward_time": 5.586889743804932,
      "step": 3626
    },
    {
      "epoch": 3.4580230712890624e-06,
      "step": 3626,
      "training_step_time": 17.021485328674316
    },
    {
      "epoch": 3.4589767456054685e-06,
      "model_forward_time": 5.581237554550171,
      "step": 3627
    },
    {
      "epoch": 3.4589767456054685e-06,
      "step": 3627,
      "training_step_time": 17.00408625602722
    },
    {
      "epoch": 3.459930419921875e-06,
      "model_forward_time": 5.595659494400024,
      "step": 3628
    },
    {
      "epoch": 3.459930419921875e-06,
      "step": 3628,
      "training_step_time": 17.060096502304077
    },
    {
      "epoch": 3.4608840942382813e-06,
      "model_forward_time": 5.966733455657959,
      "step": 3629
    },
    {
      "epoch": 3.4608840942382813e-06,
      "step": 3629,
      "training_step_time": 17.435627698898315
    },
    {
      "epoch": 3.4618377685546874e-06,
      "grad_norm": 1.6486492156982422,
      "learning_rate": 1e-05,
      "loss": 0.0695,
      "step": 3630
    },
    {
      "epoch": 3.4618377685546874e-06,
      "model_forward_time": 5.593990802764893,
      "step": 3630
    },
    {
      "dynamics_loss_avg": 0.06280913427472115,
      "epoch": 3.4618377685546874e-06,
      "step": 3630
    },
    {
      "action_loss_avg": 0.014239953039214015,
      "epoch": 3.4618377685546874e-06,
      "step": 3630
    },
    {
      "epoch": 3.4618377685546874e-06,
      "step": 3630,
      "training_step_time": 17.02558445930481
    },
    {
      "epoch": 3.4627914428710935e-06,
      "model_forward_time": 5.58901572227478,
      "step": 3631
    },
    {
      "epoch": 3.4627914428710935e-06,
      "step": 3631,
      "training_step_time": 17.036988735198975
    },
    {
      "epoch": 3.4637451171875e-06,
      "model_forward_time": 5.585740327835083,
      "step": 3632
    },
    {
      "epoch": 3.4637451171875e-06,
      "step": 3632,
      "training_step_time": 17.030733108520508
    },
    {
      "epoch": 3.4646987915039063e-06,
      "model_forward_time": 5.595278739929199,
      "step": 3633
    },
    {
      "epoch": 3.4646987915039063e-06,
      "step": 3633,
      "training_step_time": 17.07559895515442
    },
    {
      "epoch": 3.4656524658203124e-06,
      "model_forward_time": 6.008716106414795,
      "step": 3634
    },
    {
      "epoch": 3.4656524658203124e-06,
      "step": 3634,
      "training_step_time": 17.500298023223877
    },
    {
      "epoch": 3.4666061401367185e-06,
      "model_forward_time": 5.628270626068115,
      "step": 3635
    },
    {
      "epoch": 3.4666061401367185e-06,
      "step": 3635,
      "training_step_time": 17.12613821029663
    },
    {
      "epoch": 3.467559814453125e-06,
      "model_forward_time": 5.600130081176758,
      "step": 3636
    },
    {
      "epoch": 3.467559814453125e-06,
      "step": 3636,
      "training_step_time": 17.098165035247803
    },
    {
      "epoch": 3.4685134887695313e-06,
      "model_forward_time": 5.624475717544556,
      "step": 3637
    },
    {
      "epoch": 3.4685134887695313e-06,
      "step": 3637,
      "training_step_time": 17.13530421257019
    },
    {
      "epoch": 3.4694671630859374e-06,
      "model_forward_time": 5.612315654754639,
      "step": 3638
    },
    {
      "epoch": 3.4694671630859374e-06,
      "step": 3638,
      "training_step_time": 17.097370624542236
    },
    {
      "epoch": 3.4704208374023436e-06,
      "model_forward_time": 5.621264696121216,
      "step": 3639
    },
    {
      "epoch": 3.4704208374023436e-06,
      "step": 3639,
      "training_step_time": 17.124457836151123
    },
    {
      "epoch": 3.47137451171875e-06,
      "grad_norm": 0.8155897855758667,
      "learning_rate": 1e-05,
      "loss": 0.0843,
      "step": 3640
    },
    {
      "epoch": 3.47137451171875e-06,
      "model_forward_time": 5.989800453186035,
      "step": 3640
    },
    {
      "dynamics_loss_avg": 0.06528069451451302,
      "epoch": 3.47137451171875e-06,
      "step": 3640
    },
    {
      "action_loss_avg": 0.01539174634963274,
      "epoch": 3.47137451171875e-06,
      "step": 3640
    },
    {
      "epoch": 3.47137451171875e-06,
      "step": 3640,
      "training_step_time": 17.486948251724243
    },
    {
      "epoch": 3.4723281860351563e-06,
      "model_forward_time": 5.641912221908569,
      "step": 3641
    },
    {
      "epoch": 3.4723281860351563e-06,
      "step": 3641,
      "training_step_time": 17.141260147094727
    },
    {
      "epoch": 3.4732818603515624e-06,
      "model_forward_time": 5.614081859588623,
      "step": 3642
    },
    {
      "epoch": 3.4732818603515624e-06,
      "step": 3642,
      "training_step_time": 17.108266353607178
    },
    {
      "epoch": 3.4742355346679686e-06,
      "model_forward_time": 5.621049165725708,
      "step": 3643
    },
    {
      "epoch": 3.4742355346679686e-06,
      "step": 3643,
      "training_step_time": 17.11364722251892
    },
    {
      "epoch": 3.475189208984375e-06,
      "model_forward_time": 5.605839729309082,
      "step": 3644
    },
    {
      "epoch": 3.475189208984375e-06,
      "step": 3644,
      "training_step_time": 17.081369400024414
    },
    {
      "epoch": 3.4761428833007813e-06,
      "model_forward_time": 5.9777891635894775,
      "step": 3645
    },
    {
      "epoch": 3.4761428833007813e-06,
      "step": 3645,
      "training_step_time": 17.452911853790283
    },
    {
      "epoch": 3.4770965576171874e-06,
      "model_forward_time": 5.602979421615601,
      "step": 3646
    },
    {
      "epoch": 3.4770965576171874e-06,
      "step": 3646,
      "training_step_time": 17.07523012161255
    },
    {
      "epoch": 3.4780502319335936e-06,
      "model_forward_time": 5.625028610229492,
      "step": 3647
    },
    {
      "epoch": 3.4780502319335936e-06,
      "step": 3647,
      "training_step_time": 17.098647832870483
    },
    {
      "epoch": 3.47900390625e-06,
      "model_forward_time": 5.617154598236084,
      "step": 3648
    },
    {
      "epoch": 3.47900390625e-06,
      "step": 3648,
      "training_step_time": 17.111347436904907
    },
    {
      "epoch": 3.4799575805664063e-06,
      "model_forward_time": 5.605488061904907,
      "step": 3649
    },
    {
      "epoch": 3.4799575805664063e-06,
      "step": 3649,
      "training_step_time": 17.08736515045166
    },
    {
      "epoch": 3.4809112548828124e-06,
      "grad_norm": 0.881534993648529,
      "learning_rate": 1e-05,
      "loss": 0.088,
      "step": 3650
    },
    {
      "epoch": 3.4809112548828124e-06,
      "model_forward_time": 5.613005876541138,
      "step": 3650
    },
    {
      "dynamics_loss_avg": 0.07241007052361965,
      "epoch": 3.4809112548828124e-06,
      "step": 3650
    },
    {
      "action_loss_avg": 0.018190575111657382,
      "epoch": 3.4809112548828124e-06,
      "step": 3650
    },
    {
      "epoch": 3.4809112548828124e-06,
      "step": 3650,
      "training_step_time": 17.10983443260193
    },
    {
      "epoch": 3.4818649291992186e-06,
      "model_forward_time": 5.950978994369507,
      "step": 3651
    },
    {
      "epoch": 3.4818649291992186e-06,
      "step": 3651,
      "training_step_time": 17.42997694015503
    },
    {
      "epoch": 3.482818603515625e-06,
      "model_forward_time": 5.607430934906006,
      "step": 3652
    },
    {
      "epoch": 3.482818603515625e-06,
      "step": 3652,
      "training_step_time": 17.071643352508545
    },
    {
      "epoch": 3.4837722778320313e-06,
      "model_forward_time": 5.601499080657959,
      "step": 3653
    },
    {
      "epoch": 3.4837722778320313e-06,
      "step": 3653,
      "training_step_time": 17.04664921760559
    },
    {
      "epoch": 3.4847259521484374e-06,
      "model_forward_time": 5.592478513717651,
      "step": 3654
    },
    {
      "epoch": 3.4847259521484374e-06,
      "step": 3654,
      "training_step_time": 17.02456283569336
    },
    {
      "epoch": 3.4856796264648436e-06,
      "model_forward_time": 5.589204549789429,
      "step": 3655
    },
    {
      "epoch": 3.4856796264648436e-06,
      "step": 3655,
      "training_step_time": 17.00540566444397
    },
    {
      "epoch": 3.48663330078125e-06,
      "model_forward_time": 5.948056221008301,
      "step": 3656
    },
    {
      "epoch": 3.48663330078125e-06,
      "step": 3656,
      "training_step_time": 17.3973069190979
    },
    {
      "epoch": 3.4875869750976563e-06,
      "model_forward_time": 5.608934164047241,
      "step": 3657
    },
    {
      "epoch": 3.4875869750976563e-06,
      "step": 3657,
      "training_step_time": 17.111613750457764
    },
    {
      "epoch": 3.4885406494140624e-06,
      "model_forward_time": 5.615656614303589,
      "step": 3658
    },
    {
      "epoch": 3.4885406494140624e-06,
      "step": 3658,
      "training_step_time": 17.124687433242798
    },
    {
      "epoch": 3.4894943237304686e-06,
      "model_forward_time": 5.6167378425598145,
      "step": 3659
    },
    {
      "epoch": 3.4894943237304686e-06,
      "step": 3659,
      "training_step_time": 17.14809250831604
    },
    {
      "epoch": 3.490447998046875e-06,
      "grad_norm": 1.4812405109405518,
      "learning_rate": 1e-05,
      "loss": 0.0843,
      "step": 3660
    },
    {
      "epoch": 3.490447998046875e-06,
      "model_forward_time": 5.618628978729248,
      "step": 3660
    },
    {
      "dynamics_loss_avg": 0.06799383424222469,
      "epoch": 3.490447998046875e-06,
      "step": 3660
    },
    {
      "action_loss_avg": 0.012860776018351316,
      "epoch": 3.490447998046875e-06,
      "step": 3660
    },
    {
      "epoch": 3.490447998046875e-06,
      "step": 3660,
      "training_step_time": 17.10309100151062
    },
    {
      "epoch": 3.4914016723632813e-06,
      "model_forward_time": 5.59749174118042,
      "step": 3661
    },
    {
      "epoch": 3.4914016723632813e-06,
      "step": 3661,
      "training_step_time": 17.04488730430603
    },
    {
      "epoch": 3.4923553466796874e-06,
      "model_forward_time": 5.915209770202637,
      "step": 3662
    },
    {
      "epoch": 3.4923553466796874e-06,
      "step": 3662,
      "training_step_time": 17.343384504318237
    },
    {
      "epoch": 3.4933090209960936e-06,
      "model_forward_time": 5.592841148376465,
      "step": 3663
    },
    {
      "epoch": 3.4933090209960936e-06,
      "step": 3663,
      "training_step_time": 17.061060428619385
    },
    {
      "epoch": 3.4942626953125e-06,
      "model_forward_time": 5.612987756729126,
      "step": 3664
    },
    {
      "epoch": 3.4942626953125e-06,
      "step": 3664,
      "training_step_time": 17.10428285598755
    },
    {
      "epoch": 3.4952163696289063e-06,
      "model_forward_time": 5.626845359802246,
      "step": 3665
    },
    {
      "epoch": 3.4952163696289063e-06,
      "step": 3665,
      "training_step_time": 17.13573694229126
    },
    {
      "epoch": 3.4961700439453124e-06,
      "model_forward_time": 5.6252546310424805,
      "step": 3666
    },
    {
      "epoch": 3.4961700439453124e-06,
      "step": 3666,
      "training_step_time": 17.140808820724487
    },
    {
      "epoch": 3.4971237182617186e-06,
      "model_forward_time": 5.973819017410278,
      "step": 3667
    },
    {
      "epoch": 3.4971237182617186e-06,
      "step": 3667,
      "training_step_time": 17.452728271484375
    },
    {
      "epoch": 3.498077392578125e-06,
      "model_forward_time": 5.604937791824341,
      "step": 3668
    },
    {
      "epoch": 3.498077392578125e-06,
      "step": 3668,
      "training_step_time": 17.056771516799927
    },
    {
      "epoch": 3.4990310668945313e-06,
      "model_forward_time": 5.596242427825928,
      "step": 3669
    },
    {
      "epoch": 3.4990310668945313e-06,
      "step": 3669,
      "training_step_time": 17.03580927848816
    },
    {
      "epoch": 3.4999847412109374e-06,
      "grad_norm": 1.1545462608337402,
      "learning_rate": 1e-05,
      "loss": 0.092,
      "step": 3670
    },
    {
      "epoch": 3.4999847412109374e-06,
      "model_forward_time": 5.578485488891602,
      "step": 3670
    },
    {
      "dynamics_loss_avg": 0.07489184755831957,
      "epoch": 3.4999847412109374e-06,
      "step": 3670
    },
    {
      "action_loss_avg": 0.01692790782544762,
      "epoch": 3.4999847412109374e-06,
      "step": 3670
    },
    {
      "epoch": 3.4999847412109374e-06,
      "step": 3670,
      "training_step_time": 16.98920202255249
    },
    {
      "epoch": 3.5009384155273436e-06,
      "model_forward_time": 5.581552743911743,
      "step": 3671
    },
    {
      "epoch": 3.5009384155273436e-06,
      "step": 3671,
      "training_step_time": 16.994436979293823
    },
    {
      "epoch": 3.50189208984375e-06,
      "model_forward_time": 5.930018186569214,
      "step": 3672
    },
    {
      "epoch": 3.50189208984375e-06,
      "step": 3672,
      "training_step_time": 17.371052742004395
    },
    {
      "epoch": 3.5028457641601563e-06,
      "model_forward_time": 5.5782630443573,
      "step": 3673
    },
    {
      "epoch": 3.5028457641601563e-06,
      "step": 3673,
      "training_step_time": 17.013856649398804
    },
    {
      "epoch": 3.5037994384765624e-06,
      "model_forward_time": 5.590353488922119,
      "step": 3674
    },
    {
      "epoch": 3.5037994384765624e-06,
      "step": 3674,
      "training_step_time": 17.04083752632141
    },
    {
      "epoch": 3.5047531127929686e-06,
      "model_forward_time": 5.604074716567993,
      "step": 3675
    },
    {
      "epoch": 3.5047531127929686e-06,
      "step": 3675,
      "training_step_time": 17.088074684143066
    },
    {
      "epoch": 3.505706787109375e-06,
      "model_forward_time": 5.620429515838623,
      "step": 3676
    },
    {
      "epoch": 3.505706787109375e-06,
      "step": 3676,
      "training_step_time": 17.092130184173584
    },
    {
      "epoch": 3.5066604614257813e-06,
      "model_forward_time": 5.590973615646362,
      "step": 3677
    },
    {
      "epoch": 3.5066604614257813e-06,
      "step": 3677,
      "training_step_time": 17.019203901290894
    },
    {
      "epoch": 3.5076141357421874e-06,
      "model_forward_time": 5.953977108001709,
      "step": 3678
    },
    {
      "epoch": 3.5076141357421874e-06,
      "step": 3678,
      "training_step_time": 17.39244294166565
    },
    {
      "epoch": 3.5085678100585936e-06,
      "model_forward_time": 5.597743511199951,
      "step": 3679
    },
    {
      "epoch": 3.5085678100585936e-06,
      "step": 3679,
      "training_step_time": 17.053842067718506
    },
    {
      "epoch": 3.509521484375e-06,
      "grad_norm": 1.3082892894744873,
      "learning_rate": 1e-05,
      "loss": 0.0966,
      "step": 3680
    },
    {
      "epoch": 3.509521484375e-06,
      "model_forward_time": 5.598462343215942,
      "step": 3680
    },
    {
      "dynamics_loss_avg": 0.07666918598115444,
      "epoch": 3.509521484375e-06,
      "step": 3680
    },
    {
      "action_loss_avg": 0.02012123856693506,
      "epoch": 3.509521484375e-06,
      "step": 3680
    },
    {
      "epoch": 3.509521484375e-06,
      "step": 3680,
      "training_step_time": 17.045808792114258
    },
    {
      "epoch": 3.5104751586914063e-06,
      "model_forward_time": 5.593828201293945,
      "step": 3681
    },
    {
      "epoch": 3.5104751586914063e-06,
      "step": 3681,
      "training_step_time": 17.055285692214966
    },
    {
      "epoch": 3.5114288330078124e-06,
      "model_forward_time": 5.595368385314941,
      "step": 3682
    },
    {
      "epoch": 3.5114288330078124e-06,
      "step": 3682,
      "training_step_time": 17.05270218849182
    },
    {
      "epoch": 3.5123825073242186e-06,
      "model_forward_time": 5.591705560684204,
      "step": 3683
    },
    {
      "epoch": 3.5123825073242186e-06,
      "step": 3683,
      "training_step_time": 17.380830764770508
    },
    {
      "epoch": 3.513336181640625e-06,
      "model_forward_time": 5.581003665924072,
      "step": 3684
    },
    {
      "epoch": 3.513336181640625e-06,
      "step": 3684,
      "training_step_time": 17.037564516067505
    },
    {
      "epoch": 3.5142898559570313e-06,
      "model_forward_time": 5.597466468811035,
      "step": 3685
    },
    {
      "epoch": 3.5142898559570313e-06,
      "step": 3685,
      "training_step_time": 17.066120624542236
    },
    {
      "epoch": 3.5152435302734374e-06,
      "model_forward_time": 5.60775089263916,
      "step": 3686
    },
    {
      "epoch": 3.5152435302734374e-06,
      "step": 3686,
      "training_step_time": 17.080636501312256
    },
    {
      "epoch": 3.5161972045898436e-06,
      "model_forward_time": 4.2692742347717285,
      "step": 3687
    },
    {
      "epoch": 3.5161972045898436e-06,
      "step": 3687,
      "training_step_time": 12.776155948638916
    },
    {
      "epoch": 3.51715087890625e-06,
      "model_forward_time": 5.586284399032593,
      "step": 3688
    },
    {
      "epoch": 3.51715087890625e-06,
      "step": 3688,
      "training_step_time": 17.03792381286621
    },
    {
      "epoch": 3.5181045532226563e-06,
      "model_forward_time": 5.954186916351318,
      "step": 3689
    },
    {
      "epoch": 3.5181045532226563e-06,
      "step": 3689,
      "training_step_time": 17.37954807281494
    },
    {
      "epoch": 3.5190582275390624e-06,
      "grad_norm": 1.3157461881637573,
      "learning_rate": 1e-05,
      "loss": 0.0747,
      "step": 3690
    },
    {
      "epoch": 3.5190582275390624e-06,
      "model_forward_time": 5.59604811668396,
      "step": 3690
    },
    {
      "dynamics_loss_avg": 0.05622016750276089,
      "epoch": 3.5190582275390624e-06,
      "step": 3690
    },
    {
      "action_loss_avg": 0.014430920965969563,
      "epoch": 3.5190582275390624e-06,
      "step": 3690
    },
    {
      "epoch": 3.5190582275390624e-06,
      "step": 3690,
      "training_step_time": 17.062968969345093
    },
    {
      "epoch": 3.5200119018554686e-06,
      "model_forward_time": 5.610233545303345,
      "step": 3691
    },
    {
      "epoch": 3.5200119018554686e-06,
      "step": 3691,
      "training_step_time": 17.097168445587158
    },
    {
      "epoch": 3.520965576171875e-06,
      "model_forward_time": 5.618273019790649,
      "step": 3692
    },
    {
      "epoch": 3.520965576171875e-06,
      "step": 3692,
      "training_step_time": 17.10102152824402
    },
    {
      "epoch": 3.5219192504882813e-06,
      "model_forward_time": 5.606367588043213,
      "step": 3693
    },
    {
      "epoch": 3.5219192504882813e-06,
      "step": 3693,
      "training_step_time": 17.07767152786255
    },
    {
      "epoch": 3.5228729248046875e-06,
      "model_forward_time": 5.596614837646484,
      "step": 3694
    },
    {
      "epoch": 3.5228729248046875e-06,
      "step": 3694,
      "training_step_time": 17.39718461036682
    },
    {
      "epoch": 3.5238265991210936e-06,
      "model_forward_time": 5.602652311325073,
      "step": 3695
    },
    {
      "epoch": 3.5238265991210936e-06,
      "step": 3695,
      "training_step_time": 17.06844401359558
    },
    {
      "epoch": 3.5247802734375e-06,
      "model_forward_time": 5.587967872619629,
      "step": 3696
    },
    {
      "epoch": 3.5247802734375e-06,
      "step": 3696,
      "training_step_time": 17.061967134475708
    },
    {
      "epoch": 3.5257339477539063e-06,
      "model_forward_time": 5.599438428878784,
      "step": 3697
    },
    {
      "epoch": 3.5257339477539063e-06,
      "step": 3697,
      "training_step_time": 17.07420825958252
    },
    {
      "epoch": 3.5266876220703125e-06,
      "model_forward_time": 5.5923919677734375,
      "step": 3698
    },
    {
      "epoch": 3.5266876220703125e-06,
      "step": 3698,
      "training_step_time": 17.04337430000305
    },
    {
      "epoch": 3.5276412963867186e-06,
      "model_forward_time": 5.594827175140381,
      "step": 3699
    },
    {
      "epoch": 3.5276412963867186e-06,
      "step": 3699,
      "training_step_time": 17.043540716171265
    },
    {
      "epoch": 3.528594970703125e-06,
      "grad_norm": 1.6185317039489746,
      "learning_rate": 1e-05,
      "loss": 0.0905,
      "step": 3700
    },
    {
      "epoch": 3.528594970703125e-06,
      "model_forward_time": 5.970443487167358,
      "step": 3700
    },
    {
      "dynamics_loss_avg": 0.07332135066390037,
      "epoch": 3.528594970703125e-06,
      "step": 3700
    },
    {
      "action_loss_avg": 0.02089678728953004,
      "epoch": 3.528594970703125e-06,
      "step": 3700
    },
    {
      "epoch": 3.528594970703125e-06,
      "step": 3700,
      "training_step_time": 17.430173635482788
    },
    {
      "epoch": 3.5295486450195313e-06,
      "model_forward_time": 5.591593027114868,
      "step": 3701
    },
    {
      "epoch": 3.5295486450195313e-06,
      "step": 3701,
      "training_step_time": 17.048403024673462
    },
    {
      "epoch": 3.5305023193359375e-06,
      "model_forward_time": 5.59424901008606,
      "step": 3702
    },
    {
      "epoch": 3.5305023193359375e-06,
      "step": 3702,
      "training_step_time": 17.022743225097656
    },
    {
      "epoch": 3.5314559936523436e-06,
      "model_forward_time": 5.587148427963257,
      "step": 3703
    },
    {
      "epoch": 3.5314559936523436e-06,
      "step": 3703,
      "training_step_time": 17.02068066596985
    },
    {
      "epoch": 3.53240966796875e-06,
      "model_forward_time": 5.593393087387085,
      "step": 3704
    },
    {
      "epoch": 3.53240966796875e-06,
      "step": 3704,
      "training_step_time": 17.04843258857727
    },
    {
      "epoch": 3.5333633422851563e-06,
      "model_forward_time": 5.957879066467285,
      "step": 3705
    },
    {
      "epoch": 3.5333633422851563e-06,
      "step": 3705,
      "training_step_time": 17.450920820236206
    },
    {
      "epoch": 3.5343170166015625e-06,
      "model_forward_time": 5.616763591766357,
      "step": 3706
    },
    {
      "epoch": 3.5343170166015625e-06,
      "step": 3706,
      "training_step_time": 17.116539239883423
    },
    {
      "epoch": 3.5352706909179686e-06,
      "model_forward_time": 5.618101596832275,
      "step": 3707
    },
    {
      "epoch": 3.5352706909179686e-06,
      "step": 3707,
      "training_step_time": 17.126091241836548
    },
    {
      "epoch": 3.536224365234375e-06,
      "model_forward_time": 5.615209341049194,
      "step": 3708
    },
    {
      "epoch": 3.536224365234375e-06,
      "step": 3708,
      "training_step_time": 17.117830276489258
    },
    {
      "epoch": 3.5371780395507813e-06,
      "model_forward_time": 5.610545873641968,
      "step": 3709
    },
    {
      "epoch": 3.5371780395507813e-06,
      "step": 3709,
      "training_step_time": 17.120935440063477
    },
    {
      "epoch": 3.5381317138671875e-06,
      "grad_norm": 1.8448151350021362,
      "learning_rate": 1e-05,
      "loss": 0.0853,
      "step": 3710
    },
    {
      "epoch": 3.5381317138671875e-06,
      "model_forward_time": 5.610914468765259,
      "step": 3710
    },
    {
      "dynamics_loss_avg": 0.06933171413838864,
      "epoch": 3.5381317138671875e-06,
      "step": 3710
    },
    {
      "action_loss_avg": 0.01643812358379364,
      "epoch": 3.5381317138671875e-06,
      "step": 3710
    },
    {
      "epoch": 3.5381317138671875e-06,
      "step": 3710,
      "training_step_time": 17.11212706565857
    },
    {
      "epoch": 3.5390853881835936e-06,
      "model_forward_time": 5.991130590438843,
      "step": 3711
    },
    {
      "epoch": 3.5390853881835936e-06,
      "step": 3711,
      "training_step_time": 17.464229106903076
    },
    {
      "epoch": 3.5400390625e-06,
      "model_forward_time": 5.598429918289185,
      "step": 3712
    },
    {
      "epoch": 3.5400390625e-06,
      "step": 3712,
      "training_step_time": 17.056550979614258
    },
    {
      "epoch": 3.5409927368164063e-06,
      "model_forward_time": 5.604133129119873,
      "step": 3713
    },
    {
      "epoch": 3.5409927368164063e-06,
      "step": 3713,
      "training_step_time": 17.078280210494995
    },
    {
      "epoch": 3.5419464111328125e-06,
      "model_forward_time": 5.618814706802368,
      "step": 3714
    },
    {
      "epoch": 3.5419464111328125e-06,
      "step": 3714,
      "training_step_time": 17.087764978408813
    },
    {
      "epoch": 3.5429000854492186e-06,
      "model_forward_time": 5.604337930679321,
      "step": 3715
    },
    {
      "epoch": 3.5429000854492186e-06,
      "step": 3715,
      "training_step_time": 17.09800124168396
    },
    {
      "epoch": 3.543853759765625e-06,
      "model_forward_time": 5.606142997741699,
      "step": 3716
    },
    {
      "epoch": 3.543853759765625e-06,
      "step": 3716,
      "training_step_time": 17.432705640792847
    },
    {
      "epoch": 3.5448074340820313e-06,
      "model_forward_time": 5.598437786102295,
      "step": 3717
    },
    {
      "epoch": 3.5448074340820313e-06,
      "step": 3717,
      "training_step_time": 17.083662509918213
    },
    {
      "epoch": 3.5457611083984375e-06,
      "model_forward_time": 5.587573528289795,
      "step": 3718
    },
    {
      "epoch": 3.5457611083984375e-06,
      "step": 3718,
      "training_step_time": 17.029950857162476
    },
    {
      "epoch": 3.5467147827148436e-06,
      "model_forward_time": 5.598994255065918,
      "step": 3719
    },
    {
      "epoch": 3.5467147827148436e-06,
      "step": 3719,
      "training_step_time": 17.082037925720215
    },
    {
      "epoch": 3.54766845703125e-06,
      "grad_norm": 1.609351396560669,
      "learning_rate": 1e-05,
      "loss": 0.0845,
      "step": 3720
    },
    {
      "epoch": 3.54766845703125e-06,
      "model_forward_time": 5.626210451126099,
      "step": 3720
    },
    {
      "dynamics_loss_avg": 0.06586496345698833,
      "epoch": 3.54766845703125e-06,
      "step": 3720
    },
    {
      "action_loss_avg": 0.017265816358849405,
      "epoch": 3.54766845703125e-06,
      "step": 3720
    },
    {
      "epoch": 3.54766845703125e-06,
      "step": 3720,
      "training_step_time": 17.146319150924683
    },
    {
      "epoch": 3.5486221313476563e-06,
      "model_forward_time": 5.623838663101196,
      "step": 3721
    },
    {
      "epoch": 3.5486221313476563e-06,
      "step": 3721,
      "training_step_time": 17.110671043395996
    },
    {
      "epoch": 3.5495758056640625e-06,
      "model_forward_time": 5.9594738483428955,
      "step": 3722
    },
    {
      "epoch": 3.5495758056640625e-06,
      "step": 3722,
      "training_step_time": 17.402707815170288
    },
    {
      "epoch": 3.5505294799804686e-06,
      "model_forward_time": 5.575901746749878,
      "step": 3723
    },
    {
      "epoch": 3.5505294799804686e-06,
      "step": 3723,
      "training_step_time": 17.016507387161255
    },
    {
      "epoch": 3.551483154296875e-06,
      "model_forward_time": 5.575654745101929,
      "step": 3724
    },
    {
      "epoch": 3.551483154296875e-06,
      "step": 3724,
      "training_step_time": 16.993717193603516
    },
    {
      "epoch": 3.5524368286132813e-06,
      "model_forward_time": 5.574464797973633,
      "step": 3725
    },
    {
      "epoch": 3.5524368286132813e-06,
      "step": 3725,
      "training_step_time": 17.00373411178589
    },
    {
      "epoch": 3.5533905029296875e-06,
      "model_forward_time": 5.578359365463257,
      "step": 3726
    },
    {
      "epoch": 3.5533905029296875e-06,
      "step": 3726,
      "training_step_time": 17.036731243133545
    },
    {
      "epoch": 3.5543441772460936e-06,
      "model_forward_time": 5.583230972290039,
      "step": 3727
    },
    {
      "epoch": 3.5543441772460936e-06,
      "step": 3727,
      "training_step_time": 17.3873553276062
    },
    {
      "epoch": 3.5552978515625e-06,
      "model_forward_time": 5.589078664779663,
      "step": 3728
    },
    {
      "epoch": 3.5552978515625e-06,
      "step": 3728,
      "training_step_time": 17.029958248138428
    },
    {
      "epoch": 3.5562515258789063e-06,
      "model_forward_time": 5.5903167724609375,
      "step": 3729
    },
    {
      "epoch": 3.5562515258789063e-06,
      "step": 3729,
      "training_step_time": 17.04235291481018
    },
    {
      "epoch": 3.5572052001953125e-06,
      "grad_norm": 1.6320650577545166,
      "learning_rate": 1e-05,
      "loss": 0.0727,
      "step": 3730
    },
    {
      "epoch": 3.5572052001953125e-06,
      "model_forward_time": 5.596620082855225,
      "step": 3730
    },
    {
      "dynamics_loss_avg": 0.05738307163119316,
      "epoch": 3.5572052001953125e-06,
      "step": 3730
    },
    {
      "action_loss_avg": 0.013448320236057043,
      "epoch": 3.5572052001953125e-06,
      "step": 3730
    },
    {
      "epoch": 3.5572052001953125e-06,
      "step": 3730,
      "training_step_time": 17.055423259735107
    },
    {
      "epoch": 3.5581588745117186e-06,
      "model_forward_time": 5.594813108444214,
      "step": 3731
    },
    {
      "epoch": 3.5581588745117186e-06,
      "step": 3731,
      "training_step_time": 17.053734302520752
    },
    {
      "epoch": 3.559112548828125e-06,
      "model_forward_time": 5.593423128128052,
      "step": 3732
    },
    {
      "epoch": 3.559112548828125e-06,
      "step": 3732,
      "training_step_time": 17.003427982330322
    },
    {
      "epoch": 3.5600662231445313e-06,
      "model_forward_time": 5.949472188949585,
      "step": 3733
    },
    {
      "epoch": 3.5600662231445313e-06,
      "step": 3733,
      "training_step_time": 17.389414310455322
    },
    {
      "epoch": 3.5610198974609375e-06,
      "model_forward_time": 5.589195013046265,
      "step": 3734
    },
    {
      "epoch": 3.5610198974609375e-06,
      "step": 3734,
      "training_step_time": 17.026752471923828
    },
    {
      "epoch": 3.5619735717773436e-06,
      "model_forward_time": 5.597496509552002,
      "step": 3735
    },
    {
      "epoch": 3.5619735717773436e-06,
      "step": 3735,
      "training_step_time": 17.03583598136902
    },
    {
      "epoch": 3.56292724609375e-06,
      "model_forward_time": 5.593767166137695,
      "step": 3736
    },
    {
      "epoch": 3.56292724609375e-06,
      "step": 3736,
      "training_step_time": 17.03876566886902
    },
    {
      "epoch": 3.5638809204101563e-06,
      "model_forward_time": 5.606733083724976,
      "step": 3737
    },
    {
      "epoch": 3.5638809204101563e-06,
      "step": 3737,
      "training_step_time": 17.08115863800049
    },
    {
      "epoch": 3.5648345947265625e-06,
      "model_forward_time": 5.959714889526367,
      "step": 3738
    },
    {
      "epoch": 3.5648345947265625e-06,
      "step": 3738,
      "training_step_time": 17.446673154830933
    },
    {
      "epoch": 3.5657882690429686e-06,
      "model_forward_time": 5.601613759994507,
      "step": 3739
    },
    {
      "epoch": 3.5657882690429686e-06,
      "step": 3739,
      "training_step_time": 17.058913469314575
    },
    {
      "epoch": 3.566741943359375e-06,
      "grad_norm": 1.1552338600158691,
      "learning_rate": 1e-05,
      "loss": 0.0834,
      "step": 3740
    },
    {
      "epoch": 3.566741943359375e-06,
      "model_forward_time": 5.594497442245483,
      "step": 3740
    },
    {
      "dynamics_loss_avg": 0.0716297060251236,
      "epoch": 3.566741943359375e-06,
      "step": 3740
    },
    {
      "action_loss_avg": 0.017576344264671206,
      "epoch": 3.566741943359375e-06,
      "step": 3740
    },
    {
      "epoch": 3.566741943359375e-06,
      "step": 3740,
      "training_step_time": 17.028786659240723
    },
    {
      "epoch": 3.5676956176757813e-06,
      "model_forward_time": 5.587392807006836,
      "step": 3741
    },
    {
      "epoch": 3.5676956176757813e-06,
      "step": 3741,
      "training_step_time": 17.03770875930786
    },
    {
      "epoch": 3.5686492919921875e-06,
      "model_forward_time": 5.579353094100952,
      "step": 3742
    },
    {
      "epoch": 3.5686492919921875e-06,
      "step": 3742,
      "training_step_time": 17.00294780731201
    },
    {
      "epoch": 3.5696029663085936e-06,
      "model_forward_time": 5.577147006988525,
      "step": 3743
    },
    {
      "epoch": 3.5696029663085936e-06,
      "step": 3743,
      "training_step_time": 16.992119789123535
    },
    {
      "epoch": 3.570556640625e-06,
      "model_forward_time": 5.9482340812683105,
      "step": 3744
    },
    {
      "epoch": 3.570556640625e-06,
      "step": 3744,
      "training_step_time": 17.384719371795654
    },
    {
      "epoch": 3.5715103149414064e-06,
      "model_forward_time": 5.584188222885132,
      "step": 3745
    },
    {
      "epoch": 3.5715103149414064e-06,
      "step": 3745,
      "training_step_time": 17.03958010673523
    },
    {
      "epoch": 3.5724639892578125e-06,
      "model_forward_time": 5.5929460525512695,
      "step": 3746
    },
    {
      "epoch": 3.5724639892578125e-06,
      "step": 3746,
      "training_step_time": 17.0421085357666
    },
    {
      "epoch": 3.5734176635742186e-06,
      "model_forward_time": 5.603221416473389,
      "step": 3747
    },
    {
      "epoch": 3.5734176635742186e-06,
      "step": 3747,
      "training_step_time": 17.082740306854248
    },
    {
      "epoch": 3.574371337890625e-06,
      "model_forward_time": 5.615607261657715,
      "step": 3748
    },
    {
      "epoch": 3.574371337890625e-06,
      "step": 3748,
      "training_step_time": 17.09229850769043
    },
    {
      "epoch": 3.5753250122070314e-06,
      "model_forward_time": 5.611106872558594,
      "step": 3749
    },
    {
      "epoch": 3.5753250122070314e-06,
      "step": 3749,
      "training_step_time": 17.460334539413452
    },
    {
      "epoch": 3.5762786865234375e-06,
      "grad_norm": 1.0493637323379517,
      "learning_rate": 1e-05,
      "loss": 0.0585,
      "step": 3750
    },
    {
      "epoch": 3.5762786865234375e-06,
      "model_forward_time": 5.598314046859741,
      "step": 3750
    },
    {
      "dynamics_loss_avg": 0.048606077954173085,
      "epoch": 3.5762786865234375e-06,
      "step": 3750
    },
    {
      "action_loss_avg": 0.010749497497454286,
      "epoch": 3.5762786865234375e-06,
      "step": 3750
    },
    {
      "epoch": 3.5762786865234375e-06,
      "step": 3750,
      "training_step_time": 17.047709226608276
    },
    {
      "epoch": 3.5772323608398436e-06,
      "model_forward_time": 5.591250419616699,
      "step": 3751
    },
    {
      "epoch": 3.5772323608398436e-06,
      "step": 3751,
      "training_step_time": 17.05082368850708
    },
    {
      "epoch": 3.57818603515625e-06,
      "model_forward_time": 5.5992279052734375,
      "step": 3752
    },
    {
      "epoch": 3.57818603515625e-06,
      "step": 3752,
      "training_step_time": 17.065795183181763
    },
    {
      "epoch": 3.5791397094726564e-06,
      "model_forward_time": 5.601231575012207,
      "step": 3753
    },
    {
      "epoch": 3.5791397094726564e-06,
      "step": 3753,
      "training_step_time": 17.07893943786621
    },
    {
      "epoch": 3.5800933837890625e-06,
      "model_forward_time": 5.61023736000061,
      "step": 3754
    },
    {
      "epoch": 3.5800933837890625e-06,
      "step": 3754,
      "training_step_time": 17.091248512268066
    },
    {
      "epoch": 3.5810470581054686e-06,
      "model_forward_time": 5.961161851882935,
      "step": 3755
    },
    {
      "epoch": 3.5810470581054686e-06,
      "step": 3755,
      "training_step_time": 17.433076858520508
    },
    {
      "epoch": 3.582000732421875e-06,
      "model_forward_time": 5.619230508804321,
      "step": 3756
    },
    {
      "epoch": 3.582000732421875e-06,
      "step": 3756,
      "training_step_time": 17.094536304473877
    },
    {
      "epoch": 3.5829544067382814e-06,
      "model_forward_time": 5.606291770935059,
      "step": 3757
    },
    {
      "epoch": 3.5829544067382814e-06,
      "step": 3757,
      "training_step_time": 17.093427658081055
    },
    {
      "epoch": 3.5839080810546875e-06,
      "model_forward_time": 5.606114625930786,
      "step": 3758
    },
    {
      "epoch": 3.5839080810546875e-06,
      "step": 3758,
      "training_step_time": 17.0887131690979
    },
    {
      "epoch": 3.5848617553710937e-06,
      "model_forward_time": 5.602863073348999,
      "step": 3759
    },
    {
      "epoch": 3.5848617553710937e-06,
      "step": 3759,
      "training_step_time": 17.070229053497314
    },
    {
      "epoch": 3.5858154296875e-06,
      "grad_norm": 1.9301197528839111,
      "learning_rate": 1e-05,
      "loss": 0.0807,
      "step": 3760
    },
    {
      "epoch": 3.5858154296875e-06,
      "model_forward_time": 5.945894956588745,
      "step": 3760
    },
    {
      "dynamics_loss_avg": 0.06276087388396263,
      "epoch": 3.5858154296875e-06,
      "step": 3760
    },
    {
      "action_loss_avg": 0.013364174962043762,
      "epoch": 3.5858154296875e-06,
      "step": 3760
    },
    {
      "epoch": 3.5858154296875e-06,
      "step": 3760,
      "training_step_time": 17.400050163269043
    },
    {
      "epoch": 3.5867691040039064e-06,
      "model_forward_time": 5.5879807472229,
      "step": 3761
    },
    {
      "epoch": 3.5867691040039064e-06,
      "step": 3761,
      "training_step_time": 17.04432463645935
    },
    {
      "epoch": 3.5877227783203125e-06,
      "model_forward_time": 5.594446182250977,
      "step": 3762
    },
    {
      "epoch": 3.5877227783203125e-06,
      "step": 3762,
      "training_step_time": 17.053993225097656
    },
    {
      "epoch": 3.5886764526367187e-06,
      "model_forward_time": 5.596330165863037,
      "step": 3763
    },
    {
      "epoch": 3.5886764526367187e-06,
      "step": 3763,
      "training_step_time": 17.08022379875183
    },
    {
      "epoch": 3.589630126953125e-06,
      "model_forward_time": 5.607290029525757,
      "step": 3764
    },
    {
      "epoch": 3.589630126953125e-06,
      "step": 3764,
      "training_step_time": 17.076382160186768
    },
    {
      "epoch": 3.5905838012695314e-06,
      "model_forward_time": 5.603532552719116,
      "step": 3765
    },
    {
      "epoch": 3.5905838012695314e-06,
      "step": 3765,
      "training_step_time": 17.049896717071533
    },
    {
      "epoch": 3.5915374755859375e-06,
      "model_forward_time": 5.975814342498779,
      "step": 3766
    },
    {
      "epoch": 3.5915374755859375e-06,
      "step": 3766,
      "training_step_time": 17.40950846672058
    },
    {
      "epoch": 3.5924911499023437e-06,
      "model_forward_time": 5.584547996520996,
      "step": 3767
    },
    {
      "epoch": 3.5924911499023437e-06,
      "step": 3767,
      "training_step_time": 17.033499717712402
    },
    {
      "epoch": 3.59344482421875e-06,
      "model_forward_time": 5.595671892166138,
      "step": 3768
    },
    {
      "epoch": 3.59344482421875e-06,
      "step": 3768,
      "training_step_time": 17.053604125976562
    },
    {
      "epoch": 3.5943984985351564e-06,
      "model_forward_time": 5.60775899887085,
      "step": 3769
    },
    {
      "epoch": 3.5943984985351564e-06,
      "step": 3769,
      "training_step_time": 17.095958709716797
    },
    {
      "epoch": 3.5953521728515625e-06,
      "grad_norm": 0.6596070528030396,
      "learning_rate": 1e-05,
      "loss": 0.0905,
      "step": 3770
    },
    {
      "epoch": 3.5953521728515625e-06,
      "model_forward_time": 5.624078273773193,
      "step": 3770
    },
    {
      "dynamics_loss_avg": 0.07155886590480805,
      "epoch": 3.5953521728515625e-06,
      "step": 3770
    },
    {
      "action_loss_avg": 0.017898558638989925,
      "epoch": 3.5953521728515625e-06,
      "step": 3770
    },
    {
      "epoch": 3.5953521728515625e-06,
      "step": 3770,
      "training_step_time": 17.12032651901245
    },
    {
      "epoch": 3.5963058471679687e-06,
      "model_forward_time": 5.960412502288818,
      "step": 3771
    },
    {
      "epoch": 3.5963058471679687e-06,
      "step": 3771,
      "training_step_time": 17.427751302719116
    },
    {
      "epoch": 3.597259521484375e-06,
      "model_forward_time": 5.612249135971069,
      "step": 3772
    },
    {
      "epoch": 3.597259521484375e-06,
      "step": 3772,
      "training_step_time": 17.078401565551758
    },
    {
      "epoch": 3.5982131958007814e-06,
      "model_forward_time": 5.599607229232788,
      "step": 3773
    },
    {
      "epoch": 3.5982131958007814e-06,
      "step": 3773,
      "training_step_time": 17.054808616638184
    },
    {
      "epoch": 3.5991668701171875e-06,
      "model_forward_time": 5.599133014678955,
      "step": 3774
    },
    {
      "epoch": 3.5991668701171875e-06,
      "step": 3774,
      "training_step_time": 17.048696279525757
    },
    {
      "epoch": 3.6001205444335937e-06,
      "model_forward_time": 5.585541248321533,
      "step": 3775
    },
    {
      "epoch": 3.6001205444335937e-06,
      "step": 3775,
      "training_step_time": 17.043455123901367
    },
    {
      "epoch": 3.60107421875e-06,
      "model_forward_time": 5.603681325912476,
      "step": 3776
    },
    {
      "epoch": 3.60107421875e-06,
      "step": 3776,
      "training_step_time": 17.46186375617981
    },
    {
      "epoch": 3.6020278930664064e-06,
      "model_forward_time": 5.606816291809082,
      "step": 3777
    },
    {
      "epoch": 3.6020278930664064e-06,
      "step": 3777,
      "training_step_time": 17.095488786697388
    },
    {
      "epoch": 3.6029815673828125e-06,
      "model_forward_time": 5.618711233139038,
      "step": 3778
    },
    {
      "epoch": 3.6029815673828125e-06,
      "step": 3778,
      "training_step_time": 17.116798877716064
    },
    {
      "epoch": 3.6039352416992187e-06,
      "model_forward_time": 5.61263370513916,
      "step": 3779
    },
    {
      "epoch": 3.6039352416992187e-06,
      "step": 3779,
      "training_step_time": 17.109553337097168
    },
    {
      "epoch": 3.604888916015625e-06,
      "grad_norm": 0.8249773979187012,
      "learning_rate": 1e-05,
      "loss": 0.063,
      "step": 3780
    },
    {
      "epoch": 3.604888916015625e-06,
      "model_forward_time": 5.6148576736450195,
      "step": 3780
    },
    {
      "dynamics_loss_avg": 0.054776512831449506,
      "epoch": 3.604888916015625e-06,
      "step": 3780
    },
    {
      "action_loss_avg": 0.012918884051032364,
      "epoch": 3.604888916015625e-06,
      "step": 3780
    },
    {
      "epoch": 3.604888916015625e-06,
      "step": 3780,
      "training_step_time": 17.08645725250244
    },
    {
      "epoch": 3.6058425903320314e-06,
      "model_forward_time": 5.5971808433532715,
      "step": 3781
    },
    {
      "epoch": 3.6058425903320314e-06,
      "step": 3781,
      "training_step_time": 17.02763271331787
    },
    {
      "epoch": 3.6067962646484375e-06,
      "model_forward_time": 5.925248146057129,
      "step": 3782
    },
    {
      "epoch": 3.6067962646484375e-06,
      "step": 3782,
      "training_step_time": 17.343852519989014
    },
    {
      "epoch": 3.6077499389648437e-06,
      "model_forward_time": 5.5804688930511475,
      "step": 3783
    },
    {
      "epoch": 3.6077499389648437e-06,
      "step": 3783,
      "training_step_time": 17.026780128479004
    },
    {
      "epoch": 3.60870361328125e-06,
      "model_forward_time": 5.6102471351623535,
      "step": 3784
    },
    {
      "epoch": 3.60870361328125e-06,
      "step": 3784,
      "training_step_time": 17.05190920829773
    },
    {
      "epoch": 3.6096572875976564e-06,
      "model_forward_time": 5.637262344360352,
      "step": 3785
    },
    {
      "epoch": 3.6096572875976564e-06,
      "step": 3785,
      "training_step_time": 17.074742794036865
    },
    {
      "epoch": 3.6106109619140625e-06,
      "model_forward_time": 5.590583801269531,
      "step": 3786
    },
    {
      "epoch": 3.6106109619140625e-06,
      "step": 3786,
      "training_step_time": 17.058469772338867
    },
    {
      "epoch": 3.6115646362304687e-06,
      "model_forward_time": 5.606962203979492,
      "step": 3787
    },
    {
      "epoch": 3.6115646362304687e-06,
      "step": 3787,
      "training_step_time": 17.478723764419556
    },
    {
      "epoch": 3.612518310546875e-06,
      "model_forward_time": 5.617361307144165,
      "step": 3788
    },
    {
      "epoch": 3.612518310546875e-06,
      "step": 3788,
      "training_step_time": 17.127362966537476
    },
    {
      "epoch": 3.6134719848632814e-06,
      "model_forward_time": 5.6220481395721436,
      "step": 3789
    },
    {
      "epoch": 3.6134719848632814e-06,
      "step": 3789,
      "training_step_time": 17.135863542556763
    },
    {
      "epoch": 3.6144256591796875e-06,
      "grad_norm": 0.9654287695884705,
      "learning_rate": 1e-05,
      "loss": 0.0578,
      "step": 3790
    },
    {
      "epoch": 3.6144256591796875e-06,
      "model_forward_time": 5.626257658004761,
      "step": 3790
    },
    {
      "dynamics_loss_avg": 0.0472394865937531,
      "epoch": 3.6144256591796875e-06,
      "step": 3790
    },
    {
      "action_loss_avg": 0.010931866231840105,
      "epoch": 3.6144256591796875e-06,
      "step": 3790
    },
    {
      "epoch": 3.6144256591796875e-06,
      "step": 3790,
      "training_step_time": 17.105202198028564
    },
    {
      "epoch": 3.6153793334960937e-06,
      "model_forward_time": 5.586677312850952,
      "step": 3791
    },
    {
      "epoch": 3.6153793334960937e-06,
      "step": 3791,
      "training_step_time": 17.002662897109985
    },
    {
      "epoch": 3.6163330078125e-06,
      "model_forward_time": 5.585870027542114,
      "step": 3792
    },
    {
      "epoch": 3.6163330078125e-06,
      "step": 3792,
      "training_step_time": 16.985364198684692
    },
    {
      "epoch": 3.6172866821289064e-06,
      "model_forward_time": 5.922251224517822,
      "step": 3793
    },
    {
      "epoch": 3.6172866821289064e-06,
      "step": 3793,
      "training_step_time": 17.339131355285645
    },
    {
      "epoch": 3.6182403564453125e-06,
      "model_forward_time": 5.570437431335449,
      "step": 3794
    },
    {
      "epoch": 3.6182403564453125e-06,
      "step": 3794,
      "training_step_time": 17.004215478897095
    },
    {
      "epoch": 3.6191940307617187e-06,
      "model_forward_time": 5.5795934200286865,
      "step": 3795
    },
    {
      "epoch": 3.6191940307617187e-06,
      "step": 3795,
      "training_step_time": 16.983769178390503
    },
    {
      "epoch": 3.620147705078125e-06,
      "model_forward_time": 5.580113410949707,
      "step": 3796
    },
    {
      "epoch": 3.620147705078125e-06,
      "step": 3796,
      "training_step_time": 16.994983673095703
    },
    {
      "epoch": 3.6211013793945314e-06,
      "model_forward_time": 5.5882885456085205,
      "step": 3797
    },
    {
      "epoch": 3.6211013793945314e-06,
      "step": 3797,
      "training_step_time": 17.009363889694214
    },
    {
      "epoch": 3.6220550537109375e-06,
      "model_forward_time": 5.9525580406188965,
      "step": 3798
    },
    {
      "epoch": 3.6220550537109375e-06,
      "step": 3798,
      "training_step_time": 17.393747091293335
    },
    {
      "epoch": 3.6230087280273437e-06,
      "model_forward_time": 5.595012187957764,
      "step": 3799
    },
    {
      "epoch": 3.6230087280273437e-06,
      "step": 3799,
      "training_step_time": 17.06404995918274
    },
    {
      "epoch": 3.62396240234375e-06,
      "grad_norm": 1.8109277486801147,
      "learning_rate": 1e-05,
      "loss": 0.1012,
      "step": 3800
    },
    {
      "epoch": 3.62396240234375e-06,
      "model_forward_time": 5.610772132873535,
      "step": 3800
    },
    {
      "dynamics_loss_avg": 0.07407565116882324,
      "epoch": 3.62396240234375e-06,
      "step": 3800
    },
    {
      "action_loss_avg": 0.022526671644300222,
      "epoch": 3.62396240234375e-06,
      "step": 3800
    },
    {
      "epoch": 3.62396240234375e-06,
      "step": 3800,
      "training_step_time": 17.10648512840271
    },
    {
      "epoch": 3.6249160766601564e-06,
      "model_forward_time": 5.610686779022217,
      "step": 3801
    },
    {
      "epoch": 3.6249160766601564e-06,
      "step": 3801,
      "training_step_time": 17.102170944213867
    },
    {
      "epoch": 3.6258697509765625e-06,
      "model_forward_time": 5.606325149536133,
      "step": 3802
    },
    {
      "epoch": 3.6258697509765625e-06,
      "step": 3802,
      "training_step_time": 17.072789430618286
    },
    {
      "epoch": 3.6268234252929687e-06,
      "model_forward_time": 5.601955413818359,
      "step": 3803
    },
    {
      "epoch": 3.6268234252929687e-06,
      "step": 3803,
      "training_step_time": 17.41663694381714
    },
    {
      "epoch": 3.627777099609375e-06,
      "model_forward_time": 5.585368394851685,
      "step": 3804
    },
    {
      "epoch": 3.627777099609375e-06,
      "step": 3804,
      "training_step_time": 17.03459882736206
    },
    {
      "epoch": 3.6287307739257814e-06,
      "model_forward_time": 5.593237400054932,
      "step": 3805
    },
    {
      "epoch": 3.6287307739257814e-06,
      "step": 3805,
      "training_step_time": 17.040809392929077
    },
    {
      "epoch": 3.6296844482421875e-06,
      "model_forward_time": 5.589803457260132,
      "step": 3806
    },
    {
      "epoch": 3.6296844482421875e-06,
      "step": 3806,
      "training_step_time": 17.061854600906372
    },
    {
      "epoch": 3.6306381225585937e-06,
      "model_forward_time": 5.607584714889526,
      "step": 3807
    },
    {
      "epoch": 3.6306381225585937e-06,
      "step": 3807,
      "training_step_time": 17.08691906929016
    },
    {
      "epoch": 3.631591796875e-06,
      "model_forward_time": 5.602200031280518,
      "step": 3808
    },
    {
      "epoch": 3.631591796875e-06,
      "step": 3808,
      "training_step_time": 17.07787537574768
    },
    {
      "epoch": 3.6325454711914064e-06,
      "model_forward_time": 5.9978859424591064,
      "step": 3809
    },
    {
      "epoch": 3.6325454711914064e-06,
      "step": 3809,
      "training_step_time": 17.479763746261597
    },
    {
      "epoch": 3.6334991455078126e-06,
      "grad_norm": 1.9003701210021973,
      "learning_rate": 1e-05,
      "loss": 0.0807,
      "step": 3810
    },
    {
      "epoch": 3.6334991455078126e-06,
      "model_forward_time": 5.616687536239624,
      "step": 3810
    },
    {
      "dynamics_loss_avg": 0.06278862450271845,
      "epoch": 3.6334991455078126e-06,
      "step": 3810
    },
    {
      "action_loss_avg": 0.014669666066765785,
      "epoch": 3.6334991455078126e-06,
      "step": 3810
    },
    {
      "epoch": 3.6334991455078126e-06,
      "step": 3810,
      "training_step_time": 17.10682439804077
    },
    {
      "epoch": 3.6344528198242187e-06,
      "model_forward_time": 5.614923477172852,
      "step": 3811
    },
    {
      "epoch": 3.6344528198242187e-06,
      "step": 3811,
      "training_step_time": 17.077029943466187
    },
    {
      "epoch": 3.635406494140625e-06,
      "model_forward_time": 5.595540761947632,
      "step": 3812
    },
    {
      "epoch": 3.635406494140625e-06,
      "step": 3812,
      "training_step_time": 17.02207064628601
    },
    {
      "epoch": 3.6363601684570314e-06,
      "model_forward_time": 5.584485292434692,
      "step": 3813
    },
    {
      "epoch": 3.6363601684570314e-06,
      "step": 3813,
      "training_step_time": 17.02076506614685
    },
    {
      "epoch": 3.6373138427734376e-06,
      "model_forward_time": 5.920206069946289,
      "step": 3814
    },
    {
      "epoch": 3.6373138427734376e-06,
      "step": 3814,
      "training_step_time": 17.328377723693848
    },
    {
      "epoch": 3.6382675170898437e-06,
      "model_forward_time": 5.592473745346069,
      "step": 3815
    },
    {
      "epoch": 3.6382675170898437e-06,
      "step": 3815,
      "training_step_time": 17.037131547927856
    },
    {
      "epoch": 3.63922119140625e-06,
      "model_forward_time": 5.592653751373291,
      "step": 3816
    },
    {
      "epoch": 3.63922119140625e-06,
      "step": 3816,
      "training_step_time": 17.044493675231934
    },
    {
      "epoch": 3.6401748657226564e-06,
      "model_forward_time": 5.591026306152344,
      "step": 3817
    },
    {
      "epoch": 3.6401748657226564e-06,
      "step": 3817,
      "training_step_time": 17.018900156021118
    },
    {
      "epoch": 3.6411285400390626e-06,
      "model_forward_time": 5.603709936141968,
      "step": 3818
    },
    {
      "epoch": 3.6411285400390626e-06,
      "step": 3818,
      "training_step_time": 17.071839809417725
    },
    {
      "epoch": 3.6420822143554687e-06,
      "model_forward_time": 5.601375341415405,
      "step": 3819
    },
    {
      "epoch": 3.6420822143554687e-06,
      "step": 3819,
      "training_step_time": 17.068342447280884
    },
    {
      "epoch": 3.643035888671875e-06,
      "grad_norm": 1.7034498453140259,
      "learning_rate": 1e-05,
      "loss": 0.0743,
      "step": 3820
    },
    {
      "epoch": 3.643035888671875e-06,
      "model_forward_time": 5.986374378204346,
      "step": 3820
    },
    {
      "dynamics_loss_avg": 0.0653412587940693,
      "epoch": 3.643035888671875e-06,
      "step": 3820
    },
    {
      "action_loss_avg": 0.01713882377371192,
      "epoch": 3.643035888671875e-06,
      "step": 3820
    },
    {
      "epoch": 3.643035888671875e-06,
      "step": 3820,
      "training_step_time": 17.464675188064575
    },
    {
      "epoch": 3.6439895629882814e-06,
      "model_forward_time": 5.610823631286621,
      "step": 3821
    },
    {
      "epoch": 3.6439895629882814e-06,
      "step": 3821,
      "training_step_time": 17.088531732559204
    },
    {
      "epoch": 3.6449432373046876e-06,
      "model_forward_time": 5.610690116882324,
      "step": 3822
    },
    {
      "epoch": 3.6449432373046876e-06,
      "step": 3822,
      "training_step_time": 17.095618724822998
    },
    {
      "epoch": 3.6458969116210937e-06,
      "model_forward_time": 5.610375642776489,
      "step": 3823
    },
    {
      "epoch": 3.6458969116210937e-06,
      "step": 3823,
      "training_step_time": 17.075040578842163
    },
    {
      "epoch": 3.6468505859375e-06,
      "model_forward_time": 5.6075756549835205,
      "step": 3824
    },
    {
      "epoch": 3.6468505859375e-06,
      "step": 3824,
      "training_step_time": 17.08474373817444
    },
    {
      "epoch": 3.6478042602539064e-06,
      "model_forward_time": 5.977405309677124,
      "step": 3825
    },
    {
      "epoch": 3.6478042602539064e-06,
      "step": 3825,
      "training_step_time": 17.45378565788269
    },
    {
      "epoch": 3.6487579345703126e-06,
      "model_forward_time": 5.599346876144409,
      "step": 3826
    },
    {
      "epoch": 3.6487579345703126e-06,
      "step": 3826,
      "training_step_time": 17.055187225341797
    },
    {
      "epoch": 3.6497116088867187e-06,
      "model_forward_time": 5.5950706005096436,
      "step": 3827
    },
    {
      "epoch": 3.6497116088867187e-06,
      "step": 3827,
      "training_step_time": 17.05484414100647
    },
    {
      "epoch": 3.650665283203125e-06,
      "model_forward_time": 5.597675800323486,
      "step": 3828
    },
    {
      "epoch": 3.650665283203125e-06,
      "step": 3828,
      "training_step_time": 17.05182456970215
    },
    {
      "epoch": 3.6516189575195314e-06,
      "model_forward_time": 5.594518423080444,
      "step": 3829
    },
    {
      "epoch": 3.6516189575195314e-06,
      "step": 3829,
      "training_step_time": 17.071372270584106
    },
    {
      "epoch": 3.6525726318359376e-06,
      "grad_norm": 1.2810564041137695,
      "learning_rate": 1e-05,
      "loss": 0.0797,
      "step": 3830
    },
    {
      "epoch": 3.6525726318359376e-06,
      "model_forward_time": 5.610689163208008,
      "step": 3830
    },
    {
      "dynamics_loss_avg": 0.06054490953683853,
      "epoch": 3.6525726318359376e-06,
      "step": 3830
    },
    {
      "action_loss_avg": 0.013619836186990142,
      "epoch": 3.6525726318359376e-06,
      "step": 3830
    },
    {
      "epoch": 3.6525726318359376e-06,
      "step": 3830,
      "training_step_time": 17.09117603302002
    },
    {
      "epoch": 3.6535263061523437e-06,
      "model_forward_time": 5.967309951782227,
      "step": 3831
    },
    {
      "epoch": 3.6535263061523437e-06,
      "step": 3831,
      "training_step_time": 17.45286989212036
    },
    {
      "epoch": 3.65447998046875e-06,
      "model_forward_time": 5.610890626907349,
      "step": 3832
    },
    {
      "epoch": 3.65447998046875e-06,
      "step": 3832,
      "training_step_time": 17.09743046760559
    },
    {
      "epoch": 3.6554336547851564e-06,
      "model_forward_time": 5.655901908874512,
      "step": 3833
    },
    {
      "epoch": 3.6554336547851564e-06,
      "step": 3833,
      "training_step_time": 17.1828556060791
    },
    {
      "epoch": 3.6563873291015626e-06,
      "model_forward_time": 5.622629404067993,
      "step": 3834
    },
    {
      "epoch": 3.6563873291015626e-06,
      "step": 3834,
      "training_step_time": 17.11423087120056
    },
    {
      "epoch": 3.6573410034179687e-06,
      "model_forward_time": 5.597546577453613,
      "step": 3835
    },
    {
      "epoch": 3.6573410034179687e-06,
      "step": 3835,
      "training_step_time": 17.028369188308716
    },
    {
      "epoch": 3.658294677734375e-06,
      "model_forward_time": 5.936688661575317,
      "step": 3836
    },
    {
      "epoch": 3.658294677734375e-06,
      "step": 3836,
      "training_step_time": 17.369770050048828
    },
    {
      "epoch": 3.6592483520507814e-06,
      "model_forward_time": 5.594476699829102,
      "step": 3837
    },
    {
      "epoch": 3.6592483520507814e-06,
      "step": 3837,
      "training_step_time": 17.048547744750977
    },
    {
      "epoch": 3.6602020263671876e-06,
      "model_forward_time": 5.611665964126587,
      "step": 3838
    },
    {
      "epoch": 3.6602020263671876e-06,
      "step": 3838,
      "training_step_time": 17.105199813842773
    },
    {
      "epoch": 3.6611557006835937e-06,
      "model_forward_time": 5.614370822906494,
      "step": 3839
    },
    {
      "epoch": 3.6611557006835937e-06,
      "step": 3839,
      "training_step_time": 17.142929077148438
    },
    {
      "epoch": 3.662109375e-06,
      "grad_norm": 1.6757965087890625,
      "learning_rate": 1e-05,
      "loss": 0.0695,
      "step": 3840
    },
    {
      "epoch": 3.662109375e-06,
      "model_forward_time": 5.620082855224609,
      "step": 3840
    },
    {
      "dynamics_loss_avg": 0.055258896201848984,
      "epoch": 3.662109375e-06,
      "step": 3840
    },
    {
      "action_loss_avg": 0.012598832650110126,
      "epoch": 3.662109375e-06,
      "step": 3840
    },
    {
      "epoch": 3.662109375e-06,
      "step": 3840,
      "training_step_time": 17.13775873184204
    },
    {
      "epoch": 3.6630630493164064e-06,
      "model_forward_time": 5.641517639160156,
      "step": 3841
    },
    {
      "epoch": 3.6630630493164064e-06,
      "step": 3841,
      "training_step_time": 17.491875410079956
    },
    {
      "epoch": 3.6640167236328126e-06,
      "model_forward_time": 5.599934101104736,
      "step": 3842
    },
    {
      "epoch": 3.6640167236328126e-06,
      "step": 3842,
      "training_step_time": 17.04690456390381
    },
    {
      "epoch": 3.6649703979492187e-06,
      "model_forward_time": 5.583151817321777,
      "step": 3843
    },
    {
      "epoch": 3.6649703979492187e-06,
      "step": 3843,
      "training_step_time": 17.015576124191284
    },
    {
      "epoch": 3.665924072265625e-06,
      "model_forward_time": 5.58639669418335,
      "step": 3844
    },
    {
      "epoch": 3.665924072265625e-06,
      "step": 3844,
      "training_step_time": 17.02913498878479
    },
    {
      "epoch": 3.6668777465820314e-06,
      "model_forward_time": 5.631210803985596,
      "step": 3845
    },
    {
      "epoch": 3.6668777465820314e-06,
      "step": 3845,
      "training_step_time": 17.08573889732361
    },
    {
      "epoch": 3.6678314208984376e-06,
      "model_forward_time": 5.601376295089722,
      "step": 3846
    },
    {
      "epoch": 3.6678314208984376e-06,
      "step": 3846,
      "training_step_time": 17.09234380722046
    },
    {
      "epoch": 3.6687850952148437e-06,
      "model_forward_time": 5.947274684906006,
      "step": 3847
    },
    {
      "epoch": 3.6687850952148437e-06,
      "step": 3847,
      "training_step_time": 17.396190881729126
    },
    {
      "epoch": 3.66973876953125e-06,
      "model_forward_time": 5.595929861068726,
      "step": 3848
    },
    {
      "epoch": 3.66973876953125e-06,
      "step": 3848,
      "training_step_time": 17.053923845291138
    },
    {
      "epoch": 3.6706924438476564e-06,
      "model_forward_time": 5.595021486282349,
      "step": 3849
    },
    {
      "epoch": 3.6706924438476564e-06,
      "step": 3849,
      "training_step_time": 17.031076908111572
    },
    {
      "epoch": 3.6716461181640626e-06,
      "grad_norm": 1.4542357921600342,
      "learning_rate": 1e-05,
      "loss": 0.0798,
      "step": 3850
    },
    {
      "epoch": 3.6716461181640626e-06,
      "model_forward_time": 5.590914964675903,
      "step": 3850
    },
    {
      "dynamics_loss_avg": 0.06377359423786402,
      "epoch": 3.6716461181640626e-06,
      "step": 3850
    },
    {
      "action_loss_avg": 0.015430569590535015,
      "epoch": 3.6716461181640626e-06,
      "step": 3850
    },
    {
      "epoch": 3.6716461181640626e-06,
      "step": 3850,
      "training_step_time": 17.03368353843689
    },
    {
      "epoch": 3.6725997924804687e-06,
      "model_forward_time": 5.5939483642578125,
      "step": 3851
    },
    {
      "epoch": 3.6725997924804687e-06,
      "step": 3851,
      "training_step_time": 17.049855709075928
    },
    {
      "epoch": 3.673553466796875e-06,
      "model_forward_time": 5.981143474578857,
      "step": 3852
    },
    {
      "epoch": 3.673553466796875e-06,
      "step": 3852,
      "training_step_time": 17.47044038772583
    },
    {
      "epoch": 3.6745071411132814e-06,
      "model_forward_time": 5.617178916931152,
      "step": 3853
    },
    {
      "epoch": 3.6745071411132814e-06,
      "step": 3853,
      "training_step_time": 17.118531703948975
    },
    {
      "epoch": 3.6754608154296876e-06,
      "model_forward_time": 5.63122296333313,
      "step": 3854
    },
    {
      "epoch": 3.6754608154296876e-06,
      "step": 3854,
      "training_step_time": 17.084707736968994
    },
    {
      "epoch": 3.6764144897460937e-06,
      "model_forward_time": 5.588466644287109,
      "step": 3855
    },
    {
      "epoch": 3.6764144897460937e-06,
      "step": 3855,
      "training_step_time": 17.02772283554077
    },
    {
      "epoch": 3.6773681640625e-06,
      "model_forward_time": 5.588536024093628,
      "step": 3856
    },
    {
      "epoch": 3.6773681640625e-06,
      "step": 3856,
      "training_step_time": 17.059735536575317
    },
    {
      "epoch": 3.6783218383789064e-06,
      "model_forward_time": 5.617429971694946,
      "step": 3857
    },
    {
      "epoch": 3.6783218383789064e-06,
      "step": 3857,
      "training_step_time": 17.09813094139099
    },
    {
      "epoch": 3.6792755126953126e-06,
      "model_forward_time": 5.945879220962524,
      "step": 3858
    },
    {
      "epoch": 3.6792755126953126e-06,
      "step": 3858,
      "training_step_time": 17.428637981414795
    },
    {
      "epoch": 3.6802291870117187e-06,
      "model_forward_time": 5.618841648101807,
      "step": 3859
    },
    {
      "epoch": 3.6802291870117187e-06,
      "step": 3859,
      "training_step_time": 17.11492681503296
    },
    {
      "epoch": 3.681182861328125e-06,
      "grad_norm": 1.42606520652771,
      "learning_rate": 1e-05,
      "loss": 0.0683,
      "step": 3860
    },
    {
      "epoch": 3.681182861328125e-06,
      "model_forward_time": 5.625274419784546,
      "step": 3860
    },
    {
      "dynamics_loss_avg": 0.059794805757701394,
      "epoch": 3.681182861328125e-06,
      "step": 3860
    },
    {
      "action_loss_avg": 0.01305595701560378,
      "epoch": 3.681182861328125e-06,
      "step": 3860
    },
    {
      "epoch": 3.681182861328125e-06,
      "step": 3860,
      "training_step_time": 17.132211446762085
    },
    {
      "epoch": 3.6821365356445314e-06,
      "model_forward_time": 5.607114791870117,
      "step": 3861
    },
    {
      "epoch": 3.6821365356445314e-06,
      "step": 3861,
      "training_step_time": 17.08159065246582
    },
    {
      "epoch": 3.6830902099609376e-06,
      "model_forward_time": 5.577815532684326,
      "step": 3862
    },
    {
      "epoch": 3.6830902099609376e-06,
      "step": 3862,
      "training_step_time": 17.014735221862793
    },
    {
      "epoch": 3.6840438842773437e-06,
      "model_forward_time": 5.9365012645721436,
      "step": 3863
    },
    {
      "epoch": 3.6840438842773437e-06,
      "step": 3863,
      "training_step_time": 17.36526346206665
    },
    {
      "epoch": 3.68499755859375e-06,
      "model_forward_time": 5.579380750656128,
      "step": 3864
    },
    {
      "epoch": 3.68499755859375e-06,
      "step": 3864,
      "training_step_time": 17.01479411125183
    },
    {
      "epoch": 3.6859512329101565e-06,
      "model_forward_time": 5.596762418746948,
      "step": 3865
    },
    {
      "epoch": 3.6859512329101565e-06,
      "step": 3865,
      "training_step_time": 17.026211500167847
    },
    {
      "epoch": 3.6869049072265626e-06,
      "model_forward_time": 5.60977578163147,
      "step": 3866
    },
    {
      "epoch": 3.6869049072265626e-06,
      "step": 3866,
      "training_step_time": 17.11391520500183
    },
    {
      "epoch": 3.6878585815429687e-06,
      "model_forward_time": 4.293224811553955,
      "step": 3867
    },
    {
      "epoch": 3.6878585815429687e-06,
      "step": 3867,
      "training_step_time": 12.842852354049683
    },
    {
      "epoch": 3.688812255859375e-06,
      "model_forward_time": 5.619245529174805,
      "step": 3868
    },
    {
      "epoch": 3.688812255859375e-06,
      "step": 3868,
      "training_step_time": 17.113568544387817
    },
    {
      "epoch": 3.6897659301757815e-06,
      "model_forward_time": 5.942307949066162,
      "step": 3869
    },
    {
      "epoch": 3.6897659301757815e-06,
      "step": 3869,
      "training_step_time": 17.421931982040405
    },
    {
      "epoch": 3.6907196044921876e-06,
      "grad_norm": 1.3176538944244385,
      "learning_rate": 1e-05,
      "loss": 0.0788,
      "step": 3870
    },
    {
      "epoch": 3.6907196044921876e-06,
      "model_forward_time": 5.602663516998291,
      "step": 3870
    },
    {
      "dynamics_loss_avg": 0.0628182403743267,
      "epoch": 3.6907196044921876e-06,
      "step": 3870
    },
    {
      "action_loss_avg": 0.011985390074551105,
      "epoch": 3.6907196044921876e-06,
      "step": 3870
    },
    {
      "epoch": 3.6907196044921876e-06,
      "step": 3870,
      "training_step_time": 17.0488862991333
    },
    {
      "epoch": 3.6916732788085937e-06,
      "model_forward_time": 5.59603214263916,
      "step": 3871
    },
    {
      "epoch": 3.6916732788085937e-06,
      "step": 3871,
      "training_step_time": 17.06322932243347
    },
    {
      "epoch": 3.692626953125e-06,
      "model_forward_time": 5.602541923522949,
      "step": 3872
    },
    {
      "epoch": 3.692626953125e-06,
      "step": 3872,
      "training_step_time": 17.077943086624146
    },
    {
      "epoch": 3.6935806274414065e-06,
      "model_forward_time": 5.604291200637817,
      "step": 3873
    },
    {
      "epoch": 3.6935806274414065e-06,
      "step": 3873,
      "training_step_time": 17.053943634033203
    },
    {
      "epoch": 3.6945343017578126e-06,
      "model_forward_time": 5.979860067367554,
      "step": 3874
    },
    {
      "epoch": 3.6945343017578126e-06,
      "step": 3874,
      "training_step_time": 17.417845964431763
    },
    {
      "epoch": 3.6954879760742188e-06,
      "model_forward_time": 5.58451771736145,
      "step": 3875
    },
    {
      "epoch": 3.6954879760742188e-06,
      "step": 3875,
      "training_step_time": 17.041728258132935
    },
    {
      "epoch": 3.696441650390625e-06,
      "model_forward_time": 5.605713129043579,
      "step": 3876
    },
    {
      "epoch": 3.696441650390625e-06,
      "step": 3876,
      "training_step_time": 17.061100482940674
    },
    {
      "epoch": 3.697395324707031e-06,
      "model_forward_time": 5.604893207550049,
      "step": 3877
    },
    {
      "epoch": 3.697395324707031e-06,
      "step": 3877,
      "training_step_time": 17.07238268852234
    },
    {
      "epoch": 3.6983489990234376e-06,
      "model_forward_time": 5.601020574569702,
      "step": 3878
    },
    {
      "epoch": 3.6983489990234376e-06,
      "step": 3878,
      "training_step_time": 17.06363296508789
    },
    {
      "epoch": 3.6993026733398438e-06,
      "model_forward_time": 5.620253562927246,
      "step": 3879
    },
    {
      "epoch": 3.6993026733398438e-06,
      "step": 3879,
      "training_step_time": 17.119641065597534
    },
    {
      "epoch": 3.70025634765625e-06,
      "grad_norm": 0.7251248359680176,
      "learning_rate": 1e-05,
      "loss": 0.0602,
      "step": 3880
    },
    {
      "epoch": 3.70025634765625e-06,
      "model_forward_time": 5.957849025726318,
      "step": 3880
    },
    {
      "dynamics_loss_avg": 0.049732841365039346,
      "epoch": 3.70025634765625e-06,
      "step": 3880
    },
    {
      "action_loss_avg": 0.014381806016899645,
      "epoch": 3.70025634765625e-06,
      "step": 3880
    },
    {
      "epoch": 3.70025634765625e-06,
      "step": 3880,
      "training_step_time": 17.457411527633667
    },
    {
      "epoch": 3.701210021972656e-06,
      "model_forward_time": 5.616246938705444,
      "step": 3881
    },
    {
      "epoch": 3.701210021972656e-06,
      "step": 3881,
      "training_step_time": 17.102901458740234
    },
    {
      "epoch": 3.7021636962890626e-06,
      "model_forward_time": 5.592983961105347,
      "step": 3882
    },
    {
      "epoch": 3.7021636962890626e-06,
      "step": 3882,
      "training_step_time": 17.054306507110596
    },
    {
      "epoch": 3.7031173706054688e-06,
      "model_forward_time": 5.579085111618042,
      "step": 3883
    },
    {
      "epoch": 3.7031173706054688e-06,
      "step": 3883,
      "training_step_time": 17.019710302352905
    },
    {
      "epoch": 3.704071044921875e-06,
      "model_forward_time": 5.584223508834839,
      "step": 3884
    },
    {
      "epoch": 3.704071044921875e-06,
      "step": 3884,
      "training_step_time": 17.026121139526367
    },
    {
      "epoch": 3.705024719238281e-06,
      "model_forward_time": 5.953880786895752,
      "step": 3885
    },
    {
      "epoch": 3.705024719238281e-06,
      "step": 3885,
      "training_step_time": 17.396260023117065
    },
    {
      "epoch": 3.7059783935546876e-06,
      "model_forward_time": 5.603701591491699,
      "step": 3886
    },
    {
      "epoch": 3.7059783935546876e-06,
      "step": 3886,
      "training_step_time": 17.061970233917236
    },
    {
      "epoch": 3.7069320678710938e-06,
      "model_forward_time": 5.611467123031616,
      "step": 3887
    },
    {
      "epoch": 3.7069320678710938e-06,
      "step": 3887,
      "training_step_time": 17.09626340866089
    },
    {
      "epoch": 3.7078857421875e-06,
      "model_forward_time": 5.596214532852173,
      "step": 3888
    },
    {
      "epoch": 3.7078857421875e-06,
      "step": 3888,
      "training_step_time": 17.042036771774292
    },
    {
      "epoch": 3.708839416503906e-06,
      "model_forward_time": 5.591156721115112,
      "step": 3889
    },
    {
      "epoch": 3.708839416503906e-06,
      "step": 3889,
      "training_step_time": 17.045629262924194
    },
    {
      "epoch": 3.7097930908203126e-06,
      "grad_norm": 1.2878299951553345,
      "learning_rate": 1e-05,
      "loss": 0.0841,
      "step": 3890
    },
    {
      "epoch": 3.7097930908203126e-06,
      "model_forward_time": 5.607096910476685,
      "step": 3890
    },
    {
      "dynamics_loss_avg": 0.06669705025851727,
      "epoch": 3.7097930908203126e-06,
      "step": 3890
    },
    {
      "action_loss_avg": 0.01411544606089592,
      "epoch": 3.7097930908203126e-06,
      "step": 3890
    },
    {
      "epoch": 3.7097930908203126e-06,
      "step": 3890,
      "training_step_time": 17.439791440963745
    },
    {
      "epoch": 3.7107467651367188e-06,
      "model_forward_time": 5.596029996871948,
      "step": 3891
    },
    {
      "epoch": 3.7107467651367188e-06,
      "step": 3891,
      "training_step_time": 17.076239824295044
    },
    {
      "epoch": 3.711700439453125e-06,
      "model_forward_time": 5.59802508354187,
      "step": 3892
    },
    {
      "epoch": 3.711700439453125e-06,
      "step": 3892,
      "training_step_time": 17.040014028549194
    },
    {
      "epoch": 3.712654113769531e-06,
      "model_forward_time": 5.58463454246521,
      "step": 3893
    },
    {
      "epoch": 3.712654113769531e-06,
      "step": 3893,
      "training_step_time": 17.00991916656494
    },
    {
      "epoch": 3.7136077880859376e-06,
      "model_forward_time": 5.584322690963745,
      "step": 3894
    },
    {
      "epoch": 3.7136077880859376e-06,
      "step": 3894,
      "training_step_time": 17.027326583862305
    },
    {
      "epoch": 3.7145614624023438e-06,
      "model_forward_time": 5.5943522453308105,
      "step": 3895
    },
    {
      "epoch": 3.7145614624023438e-06,
      "step": 3895,
      "training_step_time": 17.03021264076233
    },
    {
      "epoch": 3.71551513671875e-06,
      "model_forward_time": 5.94254732131958,
      "step": 3896
    },
    {
      "epoch": 3.71551513671875e-06,
      "step": 3896,
      "training_step_time": 17.391223907470703
    },
    {
      "epoch": 3.716468811035156e-06,
      "model_forward_time": 5.606792449951172,
      "step": 3897
    },
    {
      "epoch": 3.716468811035156e-06,
      "step": 3897,
      "training_step_time": 17.048046588897705
    },
    {
      "epoch": 3.7174224853515626e-06,
      "model_forward_time": 5.598546028137207,
      "step": 3898
    },
    {
      "epoch": 3.7174224853515626e-06,
      "step": 3898,
      "training_step_time": 17.06206965446472
    },
    {
      "epoch": 3.7183761596679688e-06,
      "model_forward_time": 5.610979795455933,
      "step": 3899
    },
    {
      "epoch": 3.7183761596679688e-06,
      "step": 3899,
      "training_step_time": 17.091789960861206
    },
    {
      "epoch": 3.719329833984375e-06,
      "grad_norm": 1.1566165685653687,
      "learning_rate": 1e-05,
      "loss": 0.0799,
      "step": 3900
    },
    {
      "epoch": 3.719329833984375e-06,
      "model_forward_time": 5.6071436405181885,
      "step": 3900
    },
    {
      "dynamics_loss_avg": 0.06448029242455959,
      "epoch": 3.719329833984375e-06,
      "step": 3900
    },
    {
      "action_loss_avg": 0.014466240163892507,
      "epoch": 3.719329833984375e-06,
      "step": 3900
    },
    {
      "epoch": 3.719329833984375e-06,
      "step": 3900,
      "training_step_time": 17.099924564361572
    },
    {
      "epoch": 3.720283508300781e-06,
      "model_forward_time": 5.952083349227905,
      "step": 3901
    },
    {
      "epoch": 3.720283508300781e-06,
      "step": 3901,
      "training_step_time": 17.440612316131592
    },
    {
      "epoch": 3.7212371826171876e-06,
      "model_forward_time": 5.615606069564819,
      "step": 3902
    },
    {
      "epoch": 3.7212371826171876e-06,
      "step": 3902,
      "training_step_time": 17.089070558547974
    },
    {
      "epoch": 3.7221908569335938e-06,
      "model_forward_time": 5.6070544719696045,
      "step": 3903
    },
    {
      "epoch": 3.7221908569335938e-06,
      "step": 3903,
      "training_step_time": 17.081626176834106
    },
    {
      "epoch": 3.72314453125e-06,
      "model_forward_time": 5.6126344203948975,
      "step": 3904
    },
    {
      "epoch": 3.72314453125e-06,
      "step": 3904,
      "training_step_time": 17.100618839263916
    },
    {
      "epoch": 3.724098205566406e-06,
      "model_forward_time": 5.604576110839844,
      "step": 3905
    },
    {
      "epoch": 3.724098205566406e-06,
      "step": 3905,
      "training_step_time": 17.059922218322754
    },
    {
      "epoch": 3.7250518798828126e-06,
      "model_forward_time": 5.602601528167725,
      "step": 3906
    },
    {
      "epoch": 3.7250518798828126e-06,
      "step": 3906,
      "training_step_time": 17.06190037727356
    },
    {
      "epoch": 3.7260055541992188e-06,
      "model_forward_time": 5.947117567062378,
      "step": 3907
    },
    {
      "epoch": 3.7260055541992188e-06,
      "step": 3907,
      "training_step_time": 17.390276193618774
    },
    {
      "epoch": 3.726959228515625e-06,
      "model_forward_time": 5.591929912567139,
      "step": 3908
    },
    {
      "epoch": 3.726959228515625e-06,
      "step": 3908,
      "training_step_time": 17.03559970855713
    },
    {
      "epoch": 3.727912902832031e-06,
      "model_forward_time": 5.586510181427002,
      "step": 3909
    },
    {
      "epoch": 3.727912902832031e-06,
      "step": 3909,
      "training_step_time": 17.057440519332886
    },
    {
      "epoch": 3.7288665771484376e-06,
      "grad_norm": 0.5104532837867737,
      "learning_rate": 1e-05,
      "loss": 0.0947,
      "step": 3910
    },
    {
      "epoch": 3.7288665771484376e-06,
      "model_forward_time": 5.605311870574951,
      "step": 3910
    },
    {
      "dynamics_loss_avg": 0.07871464509516954,
      "epoch": 3.7288665771484376e-06,
      "step": 3910
    },
    {
      "action_loss_avg": 0.022608994133770467,
      "epoch": 3.7288665771484376e-06,
      "step": 3910
    },
    {
      "epoch": 3.7288665771484376e-06,
      "step": 3910,
      "training_step_time": 17.100510835647583
    },
    {
      "epoch": 3.7298202514648438e-06,
      "model_forward_time": 5.6051671504974365,
      "step": 3911
    },
    {
      "epoch": 3.7298202514648438e-06,
      "step": 3911,
      "training_step_time": 17.088597536087036
    },
    {
      "epoch": 3.73077392578125e-06,
      "model_forward_time": 5.970087051391602,
      "step": 3912
    },
    {
      "epoch": 3.73077392578125e-06,
      "step": 3912,
      "training_step_time": 17.461721420288086
    },
    {
      "epoch": 3.731727600097656e-06,
      "model_forward_time": 5.6117002964019775,
      "step": 3913
    },
    {
      "epoch": 3.731727600097656e-06,
      "step": 3913,
      "training_step_time": 17.098905086517334
    },
    {
      "epoch": 3.7326812744140626e-06,
      "model_forward_time": 5.605591773986816,
      "step": 3914
    },
    {
      "epoch": 3.7326812744140626e-06,
      "step": 3914,
      "training_step_time": 17.082647800445557
    },
    {
      "epoch": 3.7336349487304688e-06,
      "model_forward_time": 5.608433961868286,
      "step": 3915
    },
    {
      "epoch": 3.7336349487304688e-06,
      "step": 3915,
      "training_step_time": 17.092564344406128
    },
    {
      "epoch": 3.734588623046875e-06,
      "model_forward_time": 5.602351665496826,
      "step": 3916
    },
    {
      "epoch": 3.734588623046875e-06,
      "step": 3916,
      "training_step_time": 17.07741951942444
    },
    {
      "epoch": 3.735542297363281e-06,
      "model_forward_time": 5.594033479690552,
      "step": 3917
    },
    {
      "epoch": 3.735542297363281e-06,
      "step": 3917,
      "training_step_time": 17.03637146949768
    },
    {
      "epoch": 3.7364959716796876e-06,
      "model_forward_time": 5.942840576171875,
      "step": 3918
    },
    {
      "epoch": 3.7364959716796876e-06,
      "step": 3918,
      "training_step_time": 17.350782871246338
    },
    {
      "epoch": 3.7374496459960938e-06,
      "model_forward_time": 5.5823400020599365,
      "step": 3919
    },
    {
      "epoch": 3.7374496459960938e-06,
      "step": 3919,
      "training_step_time": 17.00298023223877
    },
    {
      "epoch": 3.7384033203125e-06,
      "grad_norm": 1.4503663778305054,
      "learning_rate": 1e-05,
      "loss": 0.0941,
      "step": 3920
    },
    {
      "epoch": 3.7384033203125e-06,
      "model_forward_time": 5.579789400100708,
      "step": 3920
    },
    {
      "dynamics_loss_avg": 0.06788924969732761,
      "epoch": 3.7384033203125e-06,
      "step": 3920
    },
    {
      "action_loss_avg": 0.02050201368983835,
      "epoch": 3.7384033203125e-06,
      "step": 3920
    },
    {
      "epoch": 3.7384033203125e-06,
      "step": 3920,
      "training_step_time": 17.01961374282837
    },
    {
      "epoch": 3.739356994628906e-06,
      "model_forward_time": 5.59093165397644,
      "step": 3921
    },
    {
      "epoch": 3.739356994628906e-06,
      "step": 3921,
      "training_step_time": 17.03542160987854
    },
    {
      "epoch": 3.7403106689453126e-06,
      "model_forward_time": 5.600895166397095,
      "step": 3922
    },
    {
      "epoch": 3.7403106689453126e-06,
      "step": 3922,
      "training_step_time": 17.036820888519287
    },
    {
      "epoch": 3.741264343261719e-06,
      "model_forward_time": 5.943085193634033,
      "step": 3923
    },
    {
      "epoch": 3.741264343261719e-06,
      "step": 3923,
      "training_step_time": 17.412015914916992
    },
    {
      "epoch": 3.742218017578125e-06,
      "model_forward_time": 5.5964531898498535,
      "step": 3924
    },
    {
      "epoch": 3.742218017578125e-06,
      "step": 3924,
      "training_step_time": 17.085017681121826
    },
    {
      "epoch": 3.743171691894531e-06,
      "model_forward_time": 5.612135887145996,
      "step": 3925
    },
    {
      "epoch": 3.743171691894531e-06,
      "step": 3925,
      "training_step_time": 17.092891216278076
    },
    {
      "epoch": 3.7441253662109376e-06,
      "model_forward_time": 5.608522653579712,
      "step": 3926
    },
    {
      "epoch": 3.7441253662109376e-06,
      "step": 3926,
      "training_step_time": 17.084407806396484
    },
    {
      "epoch": 3.745079040527344e-06,
      "model_forward_time": 5.586367845535278,
      "step": 3927
    },
    {
      "epoch": 3.745079040527344e-06,
      "step": 3927,
      "training_step_time": 17.01030421257019
    },
    {
      "epoch": 3.74603271484375e-06,
      "model_forward_time": 5.582639455795288,
      "step": 3928
    },
    {
      "epoch": 3.74603271484375e-06,
      "step": 3928,
      "training_step_time": 17.379821300506592
    },
    {
      "epoch": 3.746986389160156e-06,
      "model_forward_time": 5.577639818191528,
      "step": 3929
    },
    {
      "epoch": 3.746986389160156e-06,
      "step": 3929,
      "training_step_time": 16.99509072303772
    },
    {
      "epoch": 3.7479400634765627e-06,
      "grad_norm": 1.0815622806549072,
      "learning_rate": 1e-05,
      "loss": 0.0883,
      "step": 3930
    },
    {
      "epoch": 3.7479400634765627e-06,
      "model_forward_time": 5.573749542236328,
      "step": 3930
    },
    {
      "dynamics_loss_avg": 0.06945686154067517,
      "epoch": 3.7479400634765627e-06,
      "step": 3930
    },
    {
      "action_loss_avg": 0.019601393863558768,
      "epoch": 3.7479400634765627e-06,
      "step": 3930
    },
    {
      "epoch": 3.7479400634765627e-06,
      "step": 3930,
      "training_step_time": 16.96587085723877
    },
    {
      "epoch": 3.748893737792969e-06,
      "model_forward_time": 5.585667133331299,
      "step": 3931
    },
    {
      "epoch": 3.748893737792969e-06,
      "step": 3931,
      "training_step_time": 17.022846937179565
    },
    {
      "epoch": 3.749847412109375e-06,
      "model_forward_time": 5.582593202590942,
      "step": 3932
    },
    {
      "epoch": 3.749847412109375e-06,
      "step": 3932,
      "training_step_time": 17.01744508743286
    },
    {
      "epoch": 3.750801086425781e-06,
      "model_forward_time": 5.591843366622925,
      "step": 3933
    },
    {
      "epoch": 3.750801086425781e-06,
      "step": 3933,
      "training_step_time": 17.048349380493164
    },
    {
      "epoch": 3.7517547607421877e-06,
      "model_forward_time": 5.958926677703857,
      "step": 3934
    },
    {
      "epoch": 3.7517547607421877e-06,
      "step": 3934,
      "training_step_time": 17.45525550842285
    },
    {
      "epoch": 3.752708435058594e-06,
      "model_forward_time": 5.61554741859436,
      "step": 3935
    },
    {
      "epoch": 3.752708435058594e-06,
      "step": 3935,
      "training_step_time": 17.129566192626953
    },
    {
      "epoch": 3.753662109375e-06,
      "model_forward_time": 5.624183654785156,
      "step": 3936
    },
    {
      "epoch": 3.753662109375e-06,
      "step": 3936,
      "training_step_time": 17.137222051620483
    },
    {
      "epoch": 3.754615783691406e-06,
      "model_forward_time": 5.616812467575073,
      "step": 3937
    },
    {
      "epoch": 3.754615783691406e-06,
      "step": 3937,
      "training_step_time": 17.13150429725647
    },
    {
      "epoch": 3.7555694580078127e-06,
      "model_forward_time": 5.630162477493286,
      "step": 3938
    },
    {
      "epoch": 3.7555694580078127e-06,
      "step": 3938,
      "training_step_time": 17.12893319129944
    },
    {
      "epoch": 3.756523132324219e-06,
      "model_forward_time": 5.625849485397339,
      "step": 3939
    },
    {
      "epoch": 3.756523132324219e-06,
      "step": 3939,
      "training_step_time": 17.456986665725708
    },
    {
      "epoch": 3.757476806640625e-06,
      "grad_norm": 1.1544054746627808,
      "learning_rate": 1e-05,
      "loss": 0.0812,
      "step": 3940
    },
    {
      "epoch": 3.757476806640625e-06,
      "model_forward_time": 5.61100959777832,
      "step": 3940
    },
    {
      "dynamics_loss_avg": 0.06814595498144627,
      "epoch": 3.757476806640625e-06,
      "step": 3940
    },
    {
      "action_loss_avg": 0.015370346186682583,
      "epoch": 3.757476806640625e-06,
      "step": 3940
    },
    {
      "epoch": 3.757476806640625e-06,
      "step": 3940,
      "training_step_time": 17.096149444580078
    },
    {
      "epoch": 3.758430480957031e-06,
      "model_forward_time": 5.6058149337768555,
      "step": 3941
    },
    {
      "epoch": 3.758430480957031e-06,
      "step": 3941,
      "training_step_time": 17.09935975074768
    },
    {
      "epoch": 3.7593841552734377e-06,
      "model_forward_time": 5.6051881313323975,
      "step": 3942
    },
    {
      "epoch": 3.7593841552734377e-06,
      "step": 3942,
      "training_step_time": 17.080626726150513
    },
    {
      "epoch": 3.760337829589844e-06,
      "model_forward_time": 5.623027801513672,
      "step": 3943
    },
    {
      "epoch": 3.760337829589844e-06,
      "step": 3943,
      "training_step_time": 17.08942222595215
    },
    {
      "epoch": 3.76129150390625e-06,
      "model_forward_time": 5.59226655960083,
      "step": 3944
    },
    {
      "epoch": 3.76129150390625e-06,
      "step": 3944,
      "training_step_time": 17.03568410873413
    },
    {
      "epoch": 3.762245178222656e-06,
      "model_forward_time": 5.946736812591553,
      "step": 3945
    },
    {
      "epoch": 3.762245178222656e-06,
      "step": 3945,
      "training_step_time": 17.36350917816162
    },
    {
      "epoch": 3.7631988525390627e-06,
      "model_forward_time": 5.5836241245269775,
      "step": 3946
    },
    {
      "epoch": 3.7631988525390627e-06,
      "step": 3946,
      "training_step_time": 17.007286071777344
    },
    {
      "epoch": 3.764152526855469e-06,
      "model_forward_time": 5.600463151931763,
      "step": 3947
    },
    {
      "epoch": 3.764152526855469e-06,
      "step": 3947,
      "training_step_time": 17.096909761428833
    },
    {
      "epoch": 3.765106201171875e-06,
      "model_forward_time": 5.618604421615601,
      "step": 3948
    },
    {
      "epoch": 3.765106201171875e-06,
      "step": 3948,
      "training_step_time": 17.124078512191772
    },
    {
      "epoch": 3.766059875488281e-06,
      "model_forward_time": 5.614863634109497,
      "step": 3949
    },
    {
      "epoch": 3.766059875488281e-06,
      "step": 3949,
      "training_step_time": 17.095499753952026
    },
    {
      "epoch": 3.7670135498046877e-06,
      "grad_norm": 1.3018345832824707,
      "learning_rate": 1e-05,
      "loss": 0.0801,
      "step": 3950
    },
    {
      "epoch": 3.7670135498046877e-06,
      "model_forward_time": 5.958265066146851,
      "step": 3950
    },
    {
      "dynamics_loss_avg": 0.05892946254462004,
      "epoch": 3.7670135498046877e-06,
      "step": 3950
    },
    {
      "action_loss_avg": 0.017028874019160867,
      "epoch": 3.7670135498046877e-06,
      "step": 3950
    },
    {
      "epoch": 3.7670135498046877e-06,
      "step": 3950,
      "training_step_time": 17.390905380249023
    },
    {
      "epoch": 3.767967224121094e-06,
      "model_forward_time": 5.587239980697632,
      "step": 3951
    },
    {
      "epoch": 3.767967224121094e-06,
      "step": 3951,
      "training_step_time": 17.01848578453064
    },
    {
      "epoch": 3.7689208984375e-06,
      "model_forward_time": 5.57707953453064,
      "step": 3952
    },
    {
      "epoch": 3.7689208984375e-06,
      "step": 3952,
      "training_step_time": 16.996137142181396
    },
    {
      "epoch": 3.769874572753906e-06,
      "model_forward_time": 5.578343868255615,
      "step": 3953
    },
    {
      "epoch": 3.769874572753906e-06,
      "step": 3953,
      "training_step_time": 16.98592185974121
    },
    {
      "epoch": 3.7708282470703127e-06,
      "model_forward_time": 5.575613498687744,
      "step": 3954
    },
    {
      "epoch": 3.7708282470703127e-06,
      "step": 3954,
      "training_step_time": 17.016719341278076
    },
    {
      "epoch": 3.771781921386719e-06,
      "model_forward_time": 4.255622148513794,
      "step": 3955
    },
    {
      "epoch": 3.771781921386719e-06,
      "step": 3955,
      "training_step_time": 12.752636671066284
    },
    {
      "epoch": 3.772735595703125e-06,
      "model_forward_time": 5.948569297790527,
      "step": 3956
    },
    {
      "epoch": 3.772735595703125e-06,
      "step": 3956,
      "training_step_time": 17.379443168640137
    },
    {
      "epoch": 3.773689270019531e-06,
      "model_forward_time": 5.597531080245972,
      "step": 3957
    },
    {
      "epoch": 3.773689270019531e-06,
      "step": 3957,
      "training_step_time": 17.07211136817932
    },
    {
      "epoch": 3.7746429443359377e-06,
      "model_forward_time": 5.615810394287109,
      "step": 3958
    },
    {
      "epoch": 3.7746429443359377e-06,
      "step": 3958,
      "training_step_time": 17.10279107093811
    },
    {
      "epoch": 3.775596618652344e-06,
      "model_forward_time": 5.6283183097839355,
      "step": 3959
    },
    {
      "epoch": 3.775596618652344e-06,
      "step": 3959,
      "training_step_time": 17.13590121269226
    },
    {
      "epoch": 3.77655029296875e-06,
      "grad_norm": 1.9208351373672485,
      "learning_rate": 1e-05,
      "loss": 0.0854,
      "step": 3960
    },
    {
      "epoch": 3.77655029296875e-06,
      "model_forward_time": 5.622087478637695,
      "step": 3960
    },
    {
      "dynamics_loss_avg": 0.06832802295684814,
      "epoch": 3.77655029296875e-06,
      "step": 3960
    },
    {
      "action_loss_avg": 0.020361830992624164,
      "epoch": 3.77655029296875e-06,
      "step": 3960
    },
    {
      "epoch": 3.77655029296875e-06,
      "step": 3960,
      "training_step_time": 17.141732931137085
    },
    {
      "epoch": 3.777503967285156e-06,
      "model_forward_time": 5.990673780441284,
      "step": 3961
    },
    {
      "epoch": 3.777503967285156e-06,
      "step": 3961,
      "training_step_time": 17.470500946044922
    },
    {
      "epoch": 3.7784576416015627e-06,
      "model_forward_time": 5.589626789093018,
      "step": 3962
    },
    {
      "epoch": 3.7784576416015627e-06,
      "step": 3962,
      "training_step_time": 17.023231506347656
    },
    {
      "epoch": 3.779411315917969e-06,
      "model_forward_time": 5.5767529010772705,
      "step": 3963
    },
    {
      "epoch": 3.779411315917969e-06,
      "step": 3963,
      "training_step_time": 17.00961685180664
    },
    {
      "epoch": 3.780364990234375e-06,
      "model_forward_time": 5.573798656463623,
      "step": 3964
    },
    {
      "epoch": 3.780364990234375e-06,
      "step": 3964,
      "training_step_time": 16.981197834014893
    },
    {
      "epoch": 3.781318664550781e-06,
      "model_forward_time": 5.5929787158966064,
      "step": 3965
    },
    {
      "epoch": 3.781318664550781e-06,
      "step": 3965,
      "training_step_time": 17.012550592422485
    },
    {
      "epoch": 3.7822723388671877e-06,
      "model_forward_time": 5.584140777587891,
      "step": 3966
    },
    {
      "epoch": 3.7822723388671877e-06,
      "step": 3966,
      "training_step_time": 17.02303910255432
    },
    {
      "epoch": 3.783226013183594e-06,
      "model_forward_time": 5.926462173461914,
      "step": 3967
    },
    {
      "epoch": 3.783226013183594e-06,
      "step": 3967,
      "training_step_time": 17.389681577682495
    },
    {
      "epoch": 3.7841796875e-06,
      "model_forward_time": 5.604521036148071,
      "step": 3968
    },
    {
      "epoch": 3.7841796875e-06,
      "step": 3968,
      "training_step_time": 17.071487426757812
    },
    {
      "epoch": 3.785133361816406e-06,
      "model_forward_time": 5.600137233734131,
      "step": 3969
    },
    {
      "epoch": 3.785133361816406e-06,
      "step": 3969,
      "training_step_time": 17.073595762252808
    },
    {
      "epoch": 3.7860870361328127e-06,
      "grad_norm": 1.3411728143692017,
      "learning_rate": 1e-05,
      "loss": 0.0787,
      "step": 3970
    },
    {
      "epoch": 3.7860870361328127e-06,
      "model_forward_time": 5.607275724411011,
      "step": 3970
    },
    {
      "dynamics_loss_avg": 0.06648949943482876,
      "epoch": 3.7860870361328127e-06,
      "step": 3970
    },
    {
      "action_loss_avg": 0.011843897216022015,
      "epoch": 3.7860870361328127e-06,
      "step": 3970
    },
    {
      "epoch": 3.7860870361328127e-06,
      "step": 3970,
      "training_step_time": 17.072272300720215
    },
    {
      "epoch": 3.787040710449219e-06,
      "model_forward_time": 5.6011223793029785,
      "step": 3971
    },
    {
      "epoch": 3.787040710449219e-06,
      "step": 3971,
      "training_step_time": 17.051443338394165
    },
    {
      "epoch": 3.787994384765625e-06,
      "model_forward_time": 5.958346843719482,
      "step": 3972
    },
    {
      "epoch": 3.787994384765625e-06,
      "step": 3972,
      "training_step_time": 17.405028581619263
    },
    {
      "epoch": 3.788948059082031e-06,
      "model_forward_time": 5.585421085357666,
      "step": 3973
    },
    {
      "epoch": 3.788948059082031e-06,
      "step": 3973,
      "training_step_time": 17.029281854629517
    },
    {
      "epoch": 3.7899017333984377e-06,
      "model_forward_time": 5.594356060028076,
      "step": 3974
    },
    {
      "epoch": 3.7899017333984377e-06,
      "step": 3974,
      "training_step_time": 17.07258367538452
    },
    {
      "epoch": 3.790855407714844e-06,
      "model_forward_time": 5.6088480949401855,
      "step": 3975
    },
    {
      "epoch": 3.790855407714844e-06,
      "step": 3975,
      "training_step_time": 17.110050201416016
    },
    {
      "epoch": 3.79180908203125e-06,
      "model_forward_time": 5.61144232749939,
      "step": 3976
    },
    {
      "epoch": 3.79180908203125e-06,
      "step": 3976,
      "training_step_time": 17.097837209701538
    },
    {
      "epoch": 3.792762756347656e-06,
      "model_forward_time": 5.931723594665527,
      "step": 3977
    },
    {
      "epoch": 3.792762756347656e-06,
      "step": 3977,
      "training_step_time": 17.378346920013428
    },
    {
      "epoch": 3.7937164306640627e-06,
      "model_forward_time": 5.580519676208496,
      "step": 3978
    },
    {
      "epoch": 3.7937164306640627e-06,
      "step": 3978,
      "training_step_time": 16.984904527664185
    },
    {
      "epoch": 3.794670104980469e-06,
      "model_forward_time": 5.597750663757324,
      "step": 3979
    },
    {
      "epoch": 3.794670104980469e-06,
      "step": 3979,
      "training_step_time": 17.016414642333984
    },
    {
      "epoch": 3.795623779296875e-06,
      "grad_norm": 1.2223776578903198,
      "learning_rate": 1e-05,
      "loss": 0.0802,
      "step": 3980
    },
    {
      "epoch": 3.795623779296875e-06,
      "model_forward_time": 5.59275484085083,
      "step": 3980
    },
    {
      "dynamics_loss_avg": 0.06294652707874775,
      "epoch": 3.795623779296875e-06,
      "step": 3980
    },
    {
      "action_loss_avg": 0.01677410863339901,
      "epoch": 3.795623779296875e-06,
      "step": 3980
    },
    {
      "epoch": 3.795623779296875e-06,
      "step": 3980,
      "training_step_time": 17.026853561401367
    },
    {
      "epoch": 3.796577453613281e-06,
      "model_forward_time": 5.585888147354126,
      "step": 3981
    },
    {
      "epoch": 3.796577453613281e-06,
      "step": 3981,
      "training_step_time": 17.03084969520569
    },
    {
      "epoch": 3.7975311279296877e-06,
      "model_forward_time": 5.588062763214111,
      "step": 3982
    },
    {
      "epoch": 3.7975311279296877e-06,
      "step": 3982,
      "training_step_time": 17.0524799823761
    },
    {
      "epoch": 3.798484802246094e-06,
      "model_forward_time": 5.973235368728638,
      "step": 3983
    },
    {
      "epoch": 3.798484802246094e-06,
      "step": 3983,
      "training_step_time": 17.46221661567688
    },
    {
      "epoch": 3.7994384765625e-06,
      "model_forward_time": 5.6236748695373535,
      "step": 3984
    },
    {
      "epoch": 3.7994384765625e-06,
      "step": 3984,
      "training_step_time": 17.1448016166687
    },
    {
      "epoch": 3.800392150878906e-06,
      "model_forward_time": 5.622481822967529,
      "step": 3985
    },
    {
      "epoch": 3.800392150878906e-06,
      "step": 3985,
      "training_step_time": 17.13432478904724
    },
    {
      "epoch": 3.8013458251953127e-06,
      "model_forward_time": 5.610132455825806,
      "step": 3986
    },
    {
      "epoch": 3.8013458251953127e-06,
      "step": 3986,
      "training_step_time": 17.105588912963867
    },
    {
      "epoch": 3.802299499511719e-06,
      "model_forward_time": 5.599991083145142,
      "step": 3987
    },
    {
      "epoch": 3.802299499511719e-06,
      "step": 3987,
      "training_step_time": 17.088756561279297
    },
    {
      "epoch": 3.803253173828125e-06,
      "model_forward_time": 5.945855140686035,
      "step": 3988
    },
    {
      "epoch": 3.803253173828125e-06,
      "step": 3988,
      "training_step_time": 17.416982412338257
    },
    {
      "epoch": 3.804206848144531e-06,
      "model_forward_time": 5.600934028625488,
      "step": 3989
    },
    {
      "epoch": 3.804206848144531e-06,
      "step": 3989,
      "training_step_time": 17.05270528793335
    },
    {
      "epoch": 3.8051605224609377e-06,
      "grad_norm": 1.6318726539611816,
      "learning_rate": 1e-05,
      "loss": 0.0778,
      "step": 3990
    },
    {
      "epoch": 3.8051605224609377e-06,
      "model_forward_time": 5.6027891635894775,
      "step": 3990
    },
    {
      "dynamics_loss_avg": 0.06951734200119972,
      "epoch": 3.8051605224609377e-06,
      "step": 3990
    },
    {
      "action_loss_avg": 0.017867300752550365,
      "epoch": 3.8051605224609377e-06,
      "step": 3990
    },
    {
      "epoch": 3.8051605224609377e-06,
      "step": 3990,
      "training_step_time": 17.059383392333984
    },
    {
      "epoch": 3.806114196777344e-06,
      "model_forward_time": 5.599411249160767,
      "step": 3991
    },
    {
      "epoch": 3.806114196777344e-06,
      "step": 3991,
      "training_step_time": 17.091124296188354
    },
    {
      "epoch": 3.80706787109375e-06,
      "model_forward_time": 5.618673801422119,
      "step": 3992
    },
    {
      "epoch": 3.80706787109375e-06,
      "step": 3992,
      "training_step_time": 17.137259006500244
    },
    {
      "epoch": 3.808021545410156e-06,
      "model_forward_time": 5.628342390060425,
      "step": 3993
    },
    {
      "epoch": 3.808021545410156e-06,
      "step": 3993,
      "training_step_time": 17.497424840927124
    },
    {
      "epoch": 3.8089752197265627e-06,
      "model_forward_time": 5.610315322875977,
      "step": 3994
    },
    {
      "epoch": 3.8089752197265627e-06,
      "step": 3994,
      "training_step_time": 17.094696760177612
    },
    {
      "epoch": 3.809928894042969e-06,
      "model_forward_time": 5.607517242431641,
      "step": 3995
    },
    {
      "epoch": 3.809928894042969e-06,
      "step": 3995,
      "training_step_time": 17.097380876541138
    },
    {
      "epoch": 3.810882568359375e-06,
      "model_forward_time": 5.610018730163574,
      "step": 3996
    },
    {
      "epoch": 3.810882568359375e-06,
      "step": 3996,
      "training_step_time": 17.08863377571106
    },
    {
      "epoch": 3.811836242675781e-06,
      "model_forward_time": 5.605930328369141,
      "step": 3997
    },
    {
      "epoch": 3.811836242675781e-06,
      "step": 3997,
      "training_step_time": 17.074910879135132
    },
    {
      "epoch": 3.8127899169921877e-06,
      "model_forward_time": 5.599153757095337,
      "step": 3998
    },
    {
      "epoch": 3.8127899169921877e-06,
      "step": 3998,
      "training_step_time": 17.03861665725708
    },
    {
      "epoch": 3.813743591308594e-06,
      "model_forward_time": 5.936726331710815,
      "step": 3999
    },
    {
      "epoch": 3.813743591308594e-06,
      "step": 3999,
      "training_step_time": 17.377362966537476
    },
    {
      "epoch": 3.814697265625e-06,
      "grad_norm": 0.549163281917572,
      "learning_rate": 1e-05,
      "loss": 0.0765,
      "step": 4000
    },
    {
      "epoch": 3.814697265625e-06,
      "model_forward_time": 5.637278318405151,
      "step": 4000
    },
    {
      "dynamics_loss_avg": 0.05148556139320135,
      "epoch": 3.814697265625e-06,
      "step": 4000
    },
    {
      "action_loss_avg": 0.0157812122721225,
      "epoch": 3.814697265625e-06,
      "step": 4000
    },
    {
      "epoch": 3.814697265625e-06,
      "step": 4000,
      "training_step_time": 17.097097873687744
    },
    {
      "epoch": 3.8156509399414066e-06,
      "model_forward_time": 5.64143180847168,
      "step": 4001
    },
    {
      "epoch": 3.8156509399414066e-06,
      "step": 4001,
      "training_step_time": 17.12959885597229
    },
    {
      "epoch": 3.816604614257812e-06,
      "model_forward_time": 5.657400131225586,
      "step": 4002
    },
    {
      "epoch": 3.816604614257812e-06,
      "step": 4002,
      "training_step_time": 17.168849229812622
    },
    {
      "epoch": 3.817558288574219e-06,
      "model_forward_time": 5.707019329071045,
      "step": 4003
    },
    {
      "epoch": 3.817558288574219e-06,
      "step": 4003,
      "training_step_time": 17.225906133651733
    },
    {
      "epoch": 3.818511962890625e-06,
      "model_forward_time": 6.171561241149902,
      "step": 4004
    },
    {
      "epoch": 3.818511962890625e-06,
      "step": 4004,
      "training_step_time": 17.638387441635132
    },
    {
      "epoch": 3.819465637207031e-06,
      "model_forward_time": 5.6137731075286865,
      "step": 4005
    },
    {
      "epoch": 3.819465637207031e-06,
      "step": 4005,
      "training_step_time": 17.09928870201111
    },
    {
      "epoch": 3.820419311523438e-06,
      "model_forward_time": 5.606202602386475,
      "step": 4006
    },
    {
      "epoch": 3.820419311523438e-06,
      "step": 4006,
      "training_step_time": 17.08770775794983
    },
    {
      "epoch": 3.8213729858398434e-06,
      "model_forward_time": 5.610462665557861,
      "step": 4007
    },
    {
      "epoch": 3.8213729858398434e-06,
      "step": 4007,
      "training_step_time": 17.085684299468994
    },
    {
      "epoch": 3.82232666015625e-06,
      "model_forward_time": 5.611296653747559,
      "step": 4008
    },
    {
      "epoch": 3.82232666015625e-06,
      "step": 4008,
      "training_step_time": 17.088062286376953
    },
    {
      "epoch": 3.8232803344726566e-06,
      "model_forward_time": 5.607251167297363,
      "step": 4009
    },
    {
      "epoch": 3.8232803344726566e-06,
      "step": 4009,
      "training_step_time": 17.077937602996826
    },
    {
      "epoch": 3.824234008789062e-06,
      "grad_norm": 1.623713493347168,
      "learning_rate": 1e-05,
      "loss": 0.0999,
      "step": 4010
    },
    {
      "epoch": 3.824234008789062e-06,
      "model_forward_time": 6.026623964309692,
      "step": 4010
    },
    {
      "dynamics_loss_avg": 0.07325713448226452,
      "epoch": 3.824234008789062e-06,
      "step": 4010
    },
    {
      "action_loss_avg": 0.037799152359366414,
      "epoch": 3.824234008789062e-06,
      "step": 4010
    },
    {
      "epoch": 3.824234008789062e-06,
      "step": 4010,
      "training_step_time": 17.518227338790894
    },
    {
      "epoch": 3.825187683105469e-06,
      "model_forward_time": 5.615304470062256,
      "step": 4011
    },
    {
      "epoch": 3.825187683105469e-06,
      "step": 4011,
      "training_step_time": 17.104072332382202
    },
    {
      "epoch": 3.826141357421875e-06,
      "model_forward_time": 5.6094770431518555,
      "step": 4012
    },
    {
      "epoch": 3.826141357421875e-06,
      "step": 4012,
      "training_step_time": 17.109330892562866
    },
    {
      "epoch": 3.827095031738281e-06,
      "model_forward_time": 5.602422714233398,
      "step": 4013
    },
    {
      "epoch": 3.827095031738281e-06,
      "step": 4013,
      "training_step_time": 17.08221197128296
    },
    {
      "epoch": 3.828048706054688e-06,
      "model_forward_time": 5.598847150802612,
      "step": 4014
    },
    {
      "epoch": 3.828048706054688e-06,
      "step": 4014,
      "training_step_time": 17.00892996788025
    },
    {
      "epoch": 3.8290023803710934e-06,
      "model_forward_time": 5.998331308364868,
      "step": 4015
    },
    {
      "epoch": 3.8290023803710934e-06,
      "step": 4015,
      "training_step_time": 17.444987773895264
    },
    {
      "epoch": 3.8299560546875e-06,
      "model_forward_time": 5.581852912902832,
      "step": 4016
    },
    {
      "epoch": 3.8299560546875e-06,
      "step": 4016,
      "training_step_time": 17.03470516204834
    },
    {
      "epoch": 3.830909729003907e-06,
      "model_forward_time": 5.596025466918945,
      "step": 4017
    },
    {
      "epoch": 3.830909729003907e-06,
      "step": 4017,
      "training_step_time": 17.05549907684326
    },
    {
      "epoch": 3.831863403320312e-06,
      "model_forward_time": 5.598947763442993,
      "step": 4018
    },
    {
      "epoch": 3.831863403320312e-06,
      "step": 4018,
      "training_step_time": 17.05246925354004
    },
    {
      "epoch": 3.832817077636719e-06,
      "model_forward_time": 5.597060203552246,
      "step": 4019
    },
    {
      "epoch": 3.832817077636719e-06,
      "step": 4019,
      "training_step_time": 17.05301547050476
    },
    {
      "epoch": 3.833770751953125e-06,
      "grad_norm": 2.251391887664795,
      "learning_rate": 1e-05,
      "loss": 0.0972,
      "step": 4020
    },
    {
      "epoch": 3.833770751953125e-06,
      "model_forward_time": 5.594199895858765,
      "step": 4020
    },
    {
      "dynamics_loss_avg": 0.06149273421615362,
      "epoch": 3.833770751953125e-06,
      "step": 4020
    },
    {
      "action_loss_avg": 0.02543005347251892,
      "epoch": 3.833770751953125e-06,
      "step": 4020
    },
    {
      "epoch": 3.833770751953125e-06,
      "step": 4020,
      "training_step_time": 17.049318075180054
    },
    {
      "epoch": 3.834724426269531e-06,
      "model_forward_time": 5.971890687942505,
      "step": 4021
    },
    {
      "epoch": 3.834724426269531e-06,
      "step": 4021,
      "training_step_time": 17.425742149353027
    },
    {
      "epoch": 3.835678100585938e-06,
      "model_forward_time": 5.593903541564941,
      "step": 4022
    },
    {
      "epoch": 3.835678100585938e-06,
      "step": 4022,
      "training_step_time": 17.016330242156982
    },
    {
      "epoch": 3.8366317749023435e-06,
      "model_forward_time": 5.5968177318573,
      "step": 4023
    },
    {
      "epoch": 3.8366317749023435e-06,
      "step": 4023,
      "training_step_time": 17.05175280570984
    },
    {
      "epoch": 3.83758544921875e-06,
      "model_forward_time": 5.593152761459351,
      "step": 4024
    },
    {
      "epoch": 3.83758544921875e-06,
      "step": 4024,
      "training_step_time": 17.033734560012817
    },
    {
      "epoch": 3.838539123535157e-06,
      "model_forward_time": 5.590565204620361,
      "step": 4025
    },
    {
      "epoch": 3.838539123535157e-06,
      "step": 4025,
      "training_step_time": 17.03826880455017
    },
    {
      "epoch": 3.839492797851562e-06,
      "model_forward_time": 5.9578235149383545,
      "step": 4026
    },
    {
      "epoch": 3.839492797851562e-06,
      "step": 4026,
      "training_step_time": 17.420082569122314
    },
    {
      "epoch": 3.840446472167969e-06,
      "model_forward_time": 5.621449708938599,
      "step": 4027
    },
    {
      "epoch": 3.840446472167969e-06,
      "step": 4027,
      "training_step_time": 17.118958711624146
    },
    {
      "epoch": 3.841400146484375e-06,
      "model_forward_time": 5.606513023376465,
      "step": 4028
    },
    {
      "epoch": 3.841400146484375e-06,
      "step": 4028,
      "training_step_time": 17.09502410888672
    },
    {
      "epoch": 3.842353820800781e-06,
      "model_forward_time": 5.603078842163086,
      "step": 4029
    },
    {
      "epoch": 3.842353820800781e-06,
      "step": 4029,
      "training_step_time": 17.046805381774902
    },
    {
      "epoch": 3.843307495117188e-06,
      "grad_norm": 2.2469499111175537,
      "learning_rate": 1e-05,
      "loss": 0.1063,
      "step": 4030
    },
    {
      "epoch": 3.843307495117188e-06,
      "model_forward_time": 5.579110860824585,
      "step": 4030
    },
    {
      "dynamics_loss_avg": 0.07261497154831886,
      "epoch": 3.843307495117188e-06,
      "step": 4030
    },
    {
      "action_loss_avg": 0.032475425861775875,
      "epoch": 3.843307495117188e-06,
      "step": 4030
    },
    {
      "epoch": 3.843307495117188e-06,
      "step": 4030,
      "training_step_time": 17.021620512008667
    },
    {
      "epoch": 3.8442611694335935e-06,
      "model_forward_time": 5.988964319229126,
      "step": 4031
    },
    {
      "epoch": 3.8442611694335935e-06,
      "step": 4031,
      "training_step_time": 17.443814039230347
    },
    {
      "epoch": 3.84521484375e-06,
      "model_forward_time": 5.588183403015137,
      "step": 4032
    },
    {
      "epoch": 3.84521484375e-06,
      "step": 4032,
      "training_step_time": 17.04012942314148
    },
    {
      "epoch": 3.846168518066407e-06,
      "model_forward_time": 5.604986906051636,
      "step": 4033
    },
    {
      "epoch": 3.846168518066407e-06,
      "step": 4033,
      "training_step_time": 17.092230081558228
    },
    {
      "epoch": 3.847122192382812e-06,
      "model_forward_time": 5.604416608810425,
      "step": 4034
    },
    {
      "epoch": 3.847122192382812e-06,
      "step": 4034,
      "training_step_time": 17.095109462738037
    },
    {
      "epoch": 3.848075866699219e-06,
      "model_forward_time": 5.601407289505005,
      "step": 4035
    },
    {
      "epoch": 3.848075866699219e-06,
      "step": 4035,
      "training_step_time": 17.050801992416382
    },
    {
      "epoch": 3.849029541015625e-06,
      "model_forward_time": 5.588908672332764,
      "step": 4036
    },
    {
      "epoch": 3.849029541015625e-06,
      "step": 4036,
      "training_step_time": 17.04406714439392
    },
    {
      "epoch": 3.849983215332031e-06,
      "model_forward_time": 5.942901372909546,
      "step": 4037
    },
    {
      "epoch": 3.849983215332031e-06,
      "step": 4037,
      "training_step_time": 17.428448915481567
    },
    {
      "epoch": 3.850936889648438e-06,
      "model_forward_time": 5.6140453815460205,
      "step": 4038
    },
    {
      "epoch": 3.850936889648438e-06,
      "step": 4038,
      "training_step_time": 17.123969793319702
    },
    {
      "epoch": 3.8518905639648435e-06,
      "model_forward_time": 5.6267173290252686,
      "step": 4039
    },
    {
      "epoch": 3.8518905639648435e-06,
      "step": 4039,
      "training_step_time": 17.139848947525024
    },
    {
      "epoch": 3.85284423828125e-06,
      "grad_norm": 1.0706350803375244,
      "learning_rate": 1e-05,
      "loss": 0.0977,
      "step": 4040
    },
    {
      "epoch": 3.85284423828125e-06,
      "model_forward_time": 5.614295721054077,
      "step": 4040
    },
    {
      "dynamics_loss_avg": 0.06884696036577224,
      "epoch": 3.85284423828125e-06,
      "step": 4040
    },
    {
      "action_loss_avg": 0.03178869448602199,
      "epoch": 3.85284423828125e-06,
      "step": 4040
    },
    {
      "epoch": 3.85284423828125e-06,
      "step": 4040,
      "training_step_time": 17.098373413085938
    },
    {
      "epoch": 3.853797912597657e-06,
      "model_forward_time": 5.598445415496826,
      "step": 4041
    },
    {
      "epoch": 3.853797912597657e-06,
      "step": 4041,
      "training_step_time": 17.076183557510376
    },
    {
      "epoch": 3.854751586914062e-06,
      "model_forward_time": 5.965424299240112,
      "step": 4042
    },
    {
      "epoch": 3.854751586914062e-06,
      "step": 4042,
      "training_step_time": 17.43113112449646
    },
    {
      "epoch": 3.855705261230469e-06,
      "model_forward_time": 5.591874837875366,
      "step": 4043
    },
    {
      "epoch": 3.855705261230469e-06,
      "step": 4043,
      "training_step_time": 17.007054328918457
    },
    {
      "epoch": 3.856658935546875e-06,
      "model_forward_time": 5.582660913467407,
      "step": 4044
    },
    {
      "epoch": 3.856658935546875e-06,
      "step": 4044,
      "training_step_time": 17.012694120407104
    },
    {
      "epoch": 3.857612609863281e-06,
      "model_forward_time": 5.592613220214844,
      "step": 4045
    },
    {
      "epoch": 3.857612609863281e-06,
      "step": 4045,
      "training_step_time": 17.044926643371582
    },
    {
      "epoch": 3.858566284179688e-06,
      "model_forward_time": 5.605388402938843,
      "step": 4046
    },
    {
      "epoch": 3.858566284179688e-06,
      "step": 4046,
      "training_step_time": 17.090805292129517
    },
    {
      "epoch": 3.8595199584960935e-06,
      "model_forward_time": 5.599937200546265,
      "step": 4047
    },
    {
      "epoch": 3.8595199584960935e-06,
      "step": 4047,
      "training_step_time": 17.040688514709473
    },
    {
      "epoch": 3.8604736328125e-06,
      "model_forward_time": 5.94800591468811,
      "step": 4048
    },
    {
      "epoch": 3.8604736328125e-06,
      "step": 4048,
      "training_step_time": 17.388391733169556
    },
    {
      "epoch": 3.861427307128907e-06,
      "model_forward_time": 5.587732315063477,
      "step": 4049
    },
    {
      "epoch": 3.861427307128907e-06,
      "step": 4049,
      "training_step_time": 17.01947832107544
    },
    {
      "epoch": 3.862380981445312e-06,
      "grad_norm": 1.9860316514968872,
      "learning_rate": 1e-05,
      "loss": 0.0649,
      "step": 4050
    },
    {
      "epoch": 3.862380981445312e-06,
      "model_forward_time": 5.600798606872559,
      "step": 4050
    },
    {
      "dynamics_loss_avg": 0.04553944617509842,
      "epoch": 3.862380981445312e-06,
      "step": 4050
    },
    {
      "action_loss_avg": 0.016897549945861103,
      "epoch": 3.862380981445312e-06,
      "step": 4050
    },
    {
      "epoch": 3.862380981445312e-06,
      "step": 4050,
      "training_step_time": 17.07904553413391
    },
    {
      "epoch": 3.863334655761719e-06,
      "model_forward_time": 5.6181275844573975,
      "step": 4051
    },
    {
      "epoch": 3.863334655761719e-06,
      "step": 4051,
      "training_step_time": 17.115782022476196
    },
    {
      "epoch": 3.864288330078125e-06,
      "model_forward_time": 5.615461587905884,
      "step": 4052
    },
    {
      "epoch": 3.864288330078125e-06,
      "step": 4052,
      "training_step_time": 17.121299982070923
    },
    {
      "epoch": 3.865242004394531e-06,
      "model_forward_time": 6.0027618408203125,
      "step": 4053
    },
    {
      "epoch": 3.865242004394531e-06,
      "step": 4053,
      "training_step_time": 17.467726230621338
    },
    {
      "epoch": 3.866195678710938e-06,
      "model_forward_time": 5.5744099617004395,
      "step": 4054
    },
    {
      "epoch": 3.866195678710938e-06,
      "step": 4054,
      "training_step_time": 16.99664068222046
    },
    {
      "epoch": 3.8671493530273435e-06,
      "model_forward_time": 5.55055570602417,
      "step": 4055
    },
    {
      "epoch": 3.8671493530273435e-06,
      "step": 4055,
      "training_step_time": 16.961642742156982
    },
    {
      "epoch": 3.86810302734375e-06,
      "model_forward_time": 5.549717903137207,
      "step": 4056
    },
    {
      "epoch": 3.86810302734375e-06,
      "step": 4056,
      "training_step_time": 16.945263385772705
    },
    {
      "epoch": 3.869056701660157e-06,
      "model_forward_time": 5.540003061294556,
      "step": 4057
    },
    {
      "epoch": 3.869056701660157e-06,
      "step": 4057,
      "training_step_time": 16.936249017715454
    },
    {
      "epoch": 3.870010375976562e-06,
      "model_forward_time": 5.54662561416626,
      "step": 4058
    },
    {
      "epoch": 3.870010375976562e-06,
      "step": 4058,
      "training_step_time": 17.227084636688232
    },
    {
      "epoch": 3.870964050292969e-06,
      "model_forward_time": 5.565884113311768,
      "step": 4059
    },
    {
      "epoch": 3.870964050292969e-06,
      "step": 4059,
      "training_step_time": 16.985775232315063
    },
    {
      "epoch": 3.871917724609375e-06,
      "grad_norm": 0.4875239133834839,
      "learning_rate": 1e-05,
      "loss": 0.0735,
      "step": 4060
    },
    {
      "epoch": 3.871917724609375e-06,
      "model_forward_time": 5.5594987869262695,
      "step": 4060
    },
    {
      "dynamics_loss_avg": 0.05373175870627165,
      "epoch": 3.871917724609375e-06,
      "step": 4060
    },
    {
      "action_loss_avg": 0.02102017344441265,
      "epoch": 3.871917724609375e-06,
      "step": 4060
    },
    {
      "epoch": 3.871917724609375e-06,
      "step": 4060,
      "training_step_time": 16.970364570617676
    },
    {
      "epoch": 3.872871398925781e-06,
      "model_forward_time": 5.561188459396362,
      "step": 4061
    },
    {
      "epoch": 3.872871398925781e-06,
      "step": 4061,
      "training_step_time": 16.98252248764038
    },
    {
      "epoch": 3.873825073242188e-06,
      "model_forward_time": 5.567285537719727,
      "step": 4062
    },
    {
      "epoch": 3.873825073242188e-06,
      "step": 4062,
      "training_step_time": 16.98091745376587
    },
    {
      "epoch": 3.8747787475585935e-06,
      "model_forward_time": 5.569468259811401,
      "step": 4063
    },
    {
      "epoch": 3.8747787475585935e-06,
      "step": 4063,
      "training_step_time": 16.986754655838013
    },
    {
      "epoch": 3.875732421875e-06,
      "model_forward_time": 5.957141876220703,
      "step": 4064
    },
    {
      "epoch": 3.875732421875e-06,
      "step": 4064,
      "training_step_time": 17.39490008354187
    },
    {
      "epoch": 3.876686096191407e-06,
      "model_forward_time": 5.603865385055542,
      "step": 4065
    },
    {
      "epoch": 3.876686096191407e-06,
      "step": 4065,
      "training_step_time": 17.066698789596558
    },
    {
      "epoch": 3.877639770507812e-06,
      "model_forward_time": 5.592973232269287,
      "step": 4066
    },
    {
      "epoch": 3.877639770507812e-06,
      "step": 4066,
      "training_step_time": 17.049511671066284
    },
    {
      "epoch": 3.878593444824219e-06,
      "model_forward_time": 5.589651823043823,
      "step": 4067
    },
    {
      "epoch": 3.878593444824219e-06,
      "step": 4067,
      "training_step_time": 17.033479928970337
    },
    {
      "epoch": 3.879547119140625e-06,
      "model_forward_time": 5.5779314041137695,
      "step": 4068
    },
    {
      "epoch": 3.879547119140625e-06,
      "step": 4068,
      "training_step_time": 17.02278184890747
    },
    {
      "epoch": 3.880500793457031e-06,
      "model_forward_time": 5.92495584487915,
      "step": 4069
    },
    {
      "epoch": 3.880500793457031e-06,
      "step": 4069,
      "training_step_time": 17.372315168380737
    },
    {
      "epoch": 3.881454467773438e-06,
      "grad_norm": 1.3348685503005981,
      "learning_rate": 1e-05,
      "loss": 0.0574,
      "step": 4070
    },
    {
      "epoch": 3.881454467773438e-06,
      "model_forward_time": 5.594344615936279,
      "step": 4070
    },
    {
      "dynamics_loss_avg": 0.03969673914834857,
      "epoch": 3.881454467773438e-06,
      "step": 4070
    },
    {
      "action_loss_avg": 0.014478079497348518,
      "epoch": 3.881454467773438e-06,
      "step": 4070
    },
    {
      "epoch": 3.881454467773438e-06,
      "step": 4070,
      "training_step_time": 17.072781324386597
    },
    {
      "epoch": 3.8824081420898435e-06,
      "model_forward_time": 5.628094673156738,
      "step": 4071
    },
    {
      "epoch": 3.8824081420898435e-06,
      "step": 4071,
      "training_step_time": 17.136501789093018
    },
    {
      "epoch": 3.88336181640625e-06,
      "model_forward_time": 5.619011163711548,
      "step": 4072
    },
    {
      "epoch": 3.88336181640625e-06,
      "step": 4072,
      "training_step_time": 17.08941340446472
    },
    {
      "epoch": 3.884315490722657e-06,
      "model_forward_time": 5.583896636962891,
      "step": 4073
    },
    {
      "epoch": 3.884315490722657e-06,
      "step": 4073,
      "training_step_time": 16.99730658531189
    },
    {
      "epoch": 3.885269165039062e-06,
      "model_forward_time": 5.56219482421875,
      "step": 4074
    },
    {
      "epoch": 3.885269165039062e-06,
      "step": 4074,
      "training_step_time": 16.970842599868774
    },
    {
      "epoch": 3.886222839355469e-06,
      "model_forward_time": 5.943665266036987,
      "step": 4075
    },
    {
      "epoch": 3.886222839355469e-06,
      "step": 4075,
      "training_step_time": 17.35208749771118
    },
    {
      "epoch": 3.887176513671875e-06,
      "model_forward_time": 5.5569446086883545,
      "step": 4076
    },
    {
      "epoch": 3.887176513671875e-06,
      "step": 4076,
      "training_step_time": 16.971418142318726
    },
    {
      "epoch": 3.888130187988281e-06,
      "model_forward_time": 5.5624778270721436,
      "step": 4077
    },
    {
      "epoch": 3.888130187988281e-06,
      "step": 4077,
      "training_step_time": 16.981923580169678
    },
    {
      "epoch": 3.889083862304688e-06,
      "model_forward_time": 5.558552503585815,
      "step": 4078
    },
    {
      "epoch": 3.889083862304688e-06,
      "step": 4078,
      "training_step_time": 16.971070289611816
    },
    {
      "epoch": 3.8900375366210935e-06,
      "model_forward_time": 5.565401792526245,
      "step": 4079
    },
    {
      "epoch": 3.8900375366210935e-06,
      "step": 4079,
      "training_step_time": 16.982015371322632
    },
    {
      "epoch": 3.8909912109375e-06,
      "grad_norm": 0.8776333928108215,
      "learning_rate": 1e-05,
      "loss": 0.0831,
      "step": 4080
    },
    {
      "epoch": 3.8909912109375e-06,
      "model_forward_time": 5.566734075546265,
      "step": 4080
    },
    {
      "dynamics_loss_avg": 0.06428814791142941,
      "epoch": 3.8909912109375e-06,
      "step": 4080
    },
    {
      "action_loss_avg": 0.02668308634310961,
      "epoch": 3.8909912109375e-06,
      "step": 4080
    },
    {
      "epoch": 3.8909912109375e-06,
      "step": 4080,
      "training_step_time": 17.377818822860718
    },
    {
      "epoch": 3.891944885253907e-06,
      "model_forward_time": 5.571102142333984,
      "step": 4081
    },
    {
      "epoch": 3.891944885253907e-06,
      "step": 4081,
      "training_step_time": 17.017743349075317
    },
    {
      "epoch": 3.892898559570312e-06,
      "model_forward_time": 5.600002288818359,
      "step": 4082
    },
    {
      "epoch": 3.892898559570312e-06,
      "step": 4082,
      "training_step_time": 17.075092315673828
    },
    {
      "epoch": 3.893852233886719e-06,
      "model_forward_time": 5.6032328605651855,
      "step": 4083
    },
    {
      "epoch": 3.893852233886719e-06,
      "step": 4083,
      "training_step_time": 17.093029022216797
    },
    {
      "epoch": 3.894805908203125e-06,
      "model_forward_time": 5.613298177719116,
      "step": 4084
    },
    {
      "epoch": 3.894805908203125e-06,
      "step": 4084,
      "training_step_time": 17.105226278305054
    },
    {
      "epoch": 3.895759582519531e-06,
      "model_forward_time": 5.620600938796997,
      "step": 4085
    },
    {
      "epoch": 3.895759582519531e-06,
      "step": 4085,
      "training_step_time": 17.118969440460205
    },
    {
      "epoch": 3.896713256835938e-06,
      "model_forward_time": 5.972956657409668,
      "step": 4086
    },
    {
      "epoch": 3.896713256835938e-06,
      "step": 4086,
      "training_step_time": 17.441667318344116
    },
    {
      "epoch": 3.8976669311523435e-06,
      "model_forward_time": 5.59714674949646,
      "step": 4087
    },
    {
      "epoch": 3.8976669311523435e-06,
      "step": 4087,
      "training_step_time": 17.033390522003174
    },
    {
      "epoch": 3.89862060546875e-06,
      "model_forward_time": 5.566410064697266,
      "step": 4088
    },
    {
      "epoch": 3.89862060546875e-06,
      "step": 4088,
      "training_step_time": 16.999889612197876
    },
    {
      "epoch": 3.899574279785157e-06,
      "model_forward_time": 5.571990013122559,
      "step": 4089
    },
    {
      "epoch": 3.899574279785157e-06,
      "step": 4089,
      "training_step_time": 16.9798686504364
    },
    {
      "epoch": 3.900527954101562e-06,
      "grad_norm": 1.0663471221923828,
      "learning_rate": 1e-05,
      "loss": 0.0753,
      "step": 4090
    },
    {
      "epoch": 3.900527954101562e-06,
      "model_forward_time": 5.566646337509155,
      "step": 4090
    },
    {
      "dynamics_loss_avg": 0.06280697658658027,
      "epoch": 3.900527954101562e-06,
      "step": 4090
    },
    {
      "action_loss_avg": 0.017290635081008077,
      "epoch": 3.900527954101562e-06,
      "step": 4090
    },
    {
      "epoch": 3.900527954101562e-06,
      "step": 4090,
      "training_step_time": 16.984581232070923
    },
    {
      "epoch": 3.901481628417969e-06,
      "model_forward_time": 5.936948776245117,
      "step": 4091
    },
    {
      "epoch": 3.901481628417969e-06,
      "step": 4091,
      "training_step_time": 17.374511003494263
    },
    {
      "epoch": 3.902435302734375e-06,
      "model_forward_time": 5.58424973487854,
      "step": 4092
    },
    {
      "epoch": 3.902435302734375e-06,
      "step": 4092,
      "training_step_time": 17.005359888076782
    },
    {
      "epoch": 3.903388977050781e-06,
      "model_forward_time": 5.5887675285339355,
      "step": 4093
    },
    {
      "epoch": 3.903388977050781e-06,
      "step": 4093,
      "training_step_time": 17.009220838546753
    },
    {
      "epoch": 3.904342651367188e-06,
      "model_forward_time": 5.593624591827393,
      "step": 4094
    },
    {
      "epoch": 3.904342651367188e-06,
      "step": 4094,
      "training_step_time": 17.048449277877808
    },
    {
      "epoch": 3.9052963256835935e-06,
      "model_forward_time": 5.603857040405273,
      "step": 4095
    },
    {
      "epoch": 3.9052963256835935e-06,
      "step": 4095,
      "training_step_time": 17.086781978607178
    },
    {
      "epoch": 3.90625e-06,
      "model_forward_time": 5.606167554855347,
      "step": 4096
    },
    {
      "epoch": 3.90625e-06,
      "step": 4096,
      "training_step_time": 17.115849256515503
    },
    {
      "epoch": 3.907203674316407e-06,
      "model_forward_time": 5.964753866195679,
      "step": 4097
    },
    {
      "epoch": 3.907203674316407e-06,
      "step": 4097,
      "training_step_time": 17.466362714767456
    },
    {
      "epoch": 3.908157348632812e-06,
      "model_forward_time": 5.61237096786499,
      "step": 4098
    },
    {
      "epoch": 3.908157348632812e-06,
      "step": 4098,
      "training_step_time": 17.051701068878174
    },
    {
      "epoch": 3.909111022949219e-06,
      "model_forward_time": 5.610354423522949,
      "step": 4099
    },
    {
      "epoch": 3.909111022949219e-06,
      "step": 4099,
      "training_step_time": 17.081827878952026
    },
    {
      "epoch": 3.910064697265625e-06,
      "grad_norm": 2.0512688159942627,
      "learning_rate": 1e-05,
      "loss": 0.0949,
      "step": 4100
    },
    {
      "epoch": 3.910064697265625e-06,
      "model_forward_time": 5.602877616882324,
      "step": 4100
    },
    {
      "dynamics_loss_avg": 0.04874577559530735,
      "epoch": 3.910064697265625e-06,
      "step": 4100
    },
    {
      "action_loss_avg": 0.032424163445830344,
      "epoch": 3.910064697265625e-06,
      "step": 4100
    },
    {
      "epoch": 3.910064697265625e-06,
      "step": 4100,
      "training_step_time": 17.072198152542114
    },
    {
      "epoch": 3.911018371582031e-06,
      "model_forward_time": 5.593003749847412,
      "step": 4101
    },
    {
      "epoch": 3.911018371582031e-06,
      "step": 4101,
      "training_step_time": 17.037975311279297
    },
    {
      "epoch": 3.911972045898438e-06,
      "model_forward_time": 5.933389186859131,
      "step": 4102
    },
    {
      "epoch": 3.911972045898438e-06,
      "step": 4102,
      "training_step_time": 17.370944261550903
    },
    {
      "epoch": 3.9129257202148435e-06,
      "model_forward_time": 5.598965883255005,
      "step": 4103
    },
    {
      "epoch": 3.9129257202148435e-06,
      "step": 4103,
      "training_step_time": 17.06790781021118
    },
    {
      "epoch": 3.91387939453125e-06,
      "model_forward_time": 5.611226797103882,
      "step": 4104
    },
    {
      "epoch": 3.91387939453125e-06,
      "step": 4104,
      "training_step_time": 17.11182928085327
    },
    {
      "epoch": 3.914833068847657e-06,
      "model_forward_time": 5.6230080127716064,
      "step": 4105
    },
    {
      "epoch": 3.914833068847657e-06,
      "step": 4105,
      "training_step_time": 17.13884949684143
    },
    {
      "epoch": 3.915786743164062e-06,
      "model_forward_time": 5.622653484344482,
      "step": 4106
    },
    {
      "epoch": 3.915786743164062e-06,
      "step": 4106,
      "training_step_time": 17.09222960472107
    },
    {
      "epoch": 3.916740417480469e-06,
      "model_forward_time": 5.591577529907227,
      "step": 4107
    },
    {
      "epoch": 3.916740417480469e-06,
      "step": 4107,
      "training_step_time": 17.375882625579834
    },
    {
      "epoch": 3.917694091796875e-06,
      "model_forward_time": 5.5671706199646,
      "step": 4108
    },
    {
      "epoch": 3.917694091796875e-06,
      "step": 4108,
      "training_step_time": 16.98750376701355
    },
    {
      "epoch": 3.918647766113281e-06,
      "model_forward_time": 5.571603775024414,
      "step": 4109
    },
    {
      "epoch": 3.918647766113281e-06,
      "step": 4109,
      "training_step_time": 16.982341527938843
    },
    {
      "epoch": 3.919601440429688e-06,
      "grad_norm": 0.379833847284317,
      "learning_rate": 1e-05,
      "loss": 0.0685,
      "step": 4110
    },
    {
      "epoch": 3.919601440429688e-06,
      "model_forward_time": 5.571479797363281,
      "step": 4110
    },
    {
      "dynamics_loss_avg": 0.05258462876081467,
      "epoch": 3.919601440429688e-06,
      "step": 4110
    },
    {
      "action_loss_avg": 0.019259579246863723,
      "epoch": 3.919601440429688e-06,
      "step": 4110
    },
    {
      "epoch": 3.919601440429688e-06,
      "step": 4110,
      "training_step_time": 17.012913703918457
    },
    {
      "epoch": 3.9205551147460935e-06,
      "model_forward_time": 5.565411329269409,
      "step": 4111
    },
    {
      "epoch": 3.9205551147460935e-06,
      "step": 4111,
      "training_step_time": 16.969855070114136
    },
    {
      "epoch": 3.9215087890625e-06,
      "model_forward_time": 5.5725038051605225,
      "step": 4112
    },
    {
      "epoch": 3.9215087890625e-06,
      "step": 4112,
      "training_step_time": 16.99311590194702
    },
    {
      "epoch": 3.922462463378907e-06,
      "model_forward_time": 5.942081689834595,
      "step": 4113
    },
    {
      "epoch": 3.922462463378907e-06,
      "step": 4113,
      "training_step_time": 17.395123958587646
    },
    {
      "epoch": 3.923416137695312e-06,
      "model_forward_time": 5.605118751525879,
      "step": 4114
    },
    {
      "epoch": 3.923416137695312e-06,
      "step": 4114,
      "training_step_time": 17.107332468032837
    },
    {
      "epoch": 3.924369812011719e-06,
      "model_forward_time": 5.625825643539429,
      "step": 4115
    },
    {
      "epoch": 3.924369812011719e-06,
      "step": 4115,
      "training_step_time": 17.149821043014526
    },
    {
      "epoch": 3.925323486328125e-06,
      "model_forward_time": 5.627967119216919,
      "step": 4116
    },
    {
      "epoch": 3.925323486328125e-06,
      "step": 4116,
      "training_step_time": 17.112104177474976
    },
    {
      "epoch": 3.926277160644531e-06,
      "model_forward_time": 5.584559440612793,
      "step": 4117
    },
    {
      "epoch": 3.926277160644531e-06,
      "step": 4117,
      "training_step_time": 17.00969958305359
    },
    {
      "epoch": 3.927230834960938e-06,
      "model_forward_time": 5.951741695404053,
      "step": 4118
    },
    {
      "epoch": 3.927230834960938e-06,
      "step": 4118,
      "training_step_time": 17.395990133285522
    },
    {
      "epoch": 3.9281845092773435e-06,
      "model_forward_time": 5.569936037063599,
      "step": 4119
    },
    {
      "epoch": 3.9281845092773435e-06,
      "step": 4119,
      "training_step_time": 16.988384246826172
    },
    {
      "epoch": 3.92913818359375e-06,
      "grad_norm": 2.134841203689575,
      "learning_rate": 1e-05,
      "loss": 0.0773,
      "step": 4120
    },
    {
      "epoch": 3.92913818359375e-06,
      "model_forward_time": 5.590131759643555,
      "step": 4120
    },
    {
      "dynamics_loss_avg": 0.052270596474409105,
      "epoch": 3.92913818359375e-06,
      "step": 4120
    },
    {
      "action_loss_avg": 0.022901507723145188,
      "epoch": 3.92913818359375e-06,
      "step": 4120
    },
    {
      "epoch": 3.92913818359375e-06,
      "step": 4120,
      "training_step_time": 17.023724794387817
    },
    {
      "epoch": 3.930091857910157e-06,
      "model_forward_time": 5.58127498626709,
      "step": 4121
    },
    {
      "epoch": 3.930091857910157e-06,
      "step": 4121,
      "training_step_time": 17.03892421722412
    },
    {
      "epoch": 3.931045532226562e-06,
      "model_forward_time": 5.603732347488403,
      "step": 4122
    },
    {
      "epoch": 3.931045532226562e-06,
      "step": 4122,
      "training_step_time": 17.067816257476807
    },
    {
      "epoch": 3.931999206542969e-06,
      "model_forward_time": 5.619481325149536,
      "step": 4123
    },
    {
      "epoch": 3.931999206542969e-06,
      "step": 4123,
      "training_step_time": 17.13418412208557
    },
    {
      "epoch": 3.932952880859375e-06,
      "model_forward_time": 5.96002721786499,
      "step": 4124
    },
    {
      "epoch": 3.932952880859375e-06,
      "step": 4124,
      "training_step_time": 17.451956510543823
    },
    {
      "epoch": 3.933906555175781e-06,
      "model_forward_time": 5.61255407333374,
      "step": 4125
    },
    {
      "epoch": 3.933906555175781e-06,
      "step": 4125,
      "training_step_time": 17.102017879486084
    },
    {
      "epoch": 3.934860229492188e-06,
      "model_forward_time": 5.6009111404418945,
      "step": 4126
    },
    {
      "epoch": 3.934860229492188e-06,
      "step": 4126,
      "training_step_time": 17.08977723121643
    },
    {
      "epoch": 3.9358139038085935e-06,
      "model_forward_time": 5.600616931915283,
      "step": 4127
    },
    {
      "epoch": 3.9358139038085935e-06,
      "step": 4127,
      "training_step_time": 17.08853769302368
    },
    {
      "epoch": 3.936767578125e-06,
      "model_forward_time": 5.6027467250823975,
      "step": 4128
    },
    {
      "epoch": 3.936767578125e-06,
      "step": 4128,
      "training_step_time": 17.071523904800415
    },
    {
      "epoch": 3.937721252441406e-06,
      "model_forward_time": 5.975654602050781,
      "step": 4129
    },
    {
      "epoch": 3.937721252441406e-06,
      "step": 4129,
      "training_step_time": 17.434226512908936
    },
    {
      "epoch": 3.938674926757812e-06,
      "grad_norm": 0.6053023934364319,
      "learning_rate": 1e-05,
      "loss": 0.0875,
      "step": 4130
    },
    {
      "epoch": 3.938674926757812e-06,
      "model_forward_time": 5.594287633895874,
      "step": 4130
    },
    {
      "dynamics_loss_avg": 0.06312137506902218,
      "epoch": 3.938674926757812e-06,
      "step": 4130
    },
    {
      "action_loss_avg": 0.0235473764128983,
      "epoch": 3.938674926757812e-06,
      "step": 4130
    },
    {
      "epoch": 3.938674926757812e-06,
      "step": 4130,
      "training_step_time": 17.07674789428711
    },
    {
      "epoch": 3.939628601074219e-06,
      "model_forward_time": 5.6079161167144775,
      "step": 4131
    },
    {
      "epoch": 3.939628601074219e-06,
      "step": 4131,
      "training_step_time": 17.08533787727356
    },
    {
      "epoch": 3.940582275390625e-06,
      "model_forward_time": 5.609633207321167,
      "step": 4132
    },
    {
      "epoch": 3.940582275390625e-06,
      "step": 4132,
      "training_step_time": 17.094608545303345
    },
    {
      "epoch": 3.941535949707031e-06,
      "model_forward_time": 5.600030183792114,
      "step": 4133
    },
    {
      "epoch": 3.941535949707031e-06,
      "step": 4133,
      "training_step_time": 17.03467607498169
    },
    {
      "epoch": 3.942489624023438e-06,
      "model_forward_time": 5.581127643585205,
      "step": 4134
    },
    {
      "epoch": 3.942489624023438e-06,
      "step": 4134,
      "training_step_time": 17.004870653152466
    },
    {
      "epoch": 3.9434432983398435e-06,
      "model_forward_time": 5.926271677017212,
      "step": 4135
    },
    {
      "epoch": 3.9434432983398435e-06,
      "step": 4135,
      "training_step_time": 17.3351948261261
    },
    {
      "epoch": 3.94439697265625e-06,
      "model_forward_time": 5.5711541175842285,
      "step": 4136
    },
    {
      "epoch": 3.94439697265625e-06,
      "step": 4136,
      "training_step_time": 16.998641967773438
    },
    {
      "epoch": 3.945350646972656e-06,
      "model_forward_time": 5.581909656524658,
      "step": 4137
    },
    {
      "epoch": 3.945350646972656e-06,
      "step": 4137,
      "training_step_time": 17.02570867538452
    },
    {
      "epoch": 3.946304321289062e-06,
      "model_forward_time": 5.588866233825684,
      "step": 4138
    },
    {
      "epoch": 3.946304321289062e-06,
      "step": 4138,
      "training_step_time": 17.02764081954956
    },
    {
      "epoch": 3.947257995605469e-06,
      "model_forward_time": 5.590592384338379,
      "step": 4139
    },
    {
      "epoch": 3.947257995605469e-06,
      "step": 4139,
      "training_step_time": 17.033577919006348
    },
    {
      "epoch": 3.948211669921875e-06,
      "grad_norm": 1.4227575063705444,
      "learning_rate": 1e-05,
      "loss": 0.0642,
      "step": 4140
    },
    {
      "epoch": 3.948211669921875e-06,
      "model_forward_time": 5.963258981704712,
      "step": 4140
    },
    {
      "dynamics_loss_avg": 0.04936956241726875,
      "epoch": 3.948211669921875e-06,
      "step": 4140
    },
    {
      "action_loss_avg": 0.016640031524002553,
      "epoch": 3.948211669921875e-06,
      "step": 4140
    },
    {
      "epoch": 3.948211669921875e-06,
      "step": 4140,
      "training_step_time": 17.448296785354614
    },
    {
      "epoch": 3.949165344238281e-06,
      "model_forward_time": 5.610602378845215,
      "step": 4141
    },
    {
      "epoch": 3.949165344238281e-06,
      "step": 4141,
      "training_step_time": 17.117679119110107
    },
    {
      "epoch": 3.950119018554688e-06,
      "model_forward_time": 5.628469944000244,
      "step": 4142
    },
    {
      "epoch": 3.950119018554688e-06,
      "step": 4142,
      "training_step_time": 17.122548818588257
    },
    {
      "epoch": 3.9510726928710936e-06,
      "model_forward_time": 5.627918004989624,
      "step": 4143
    },
    {
      "epoch": 3.9510726928710936e-06,
      "step": 4143,
      "training_step_time": 17.125876426696777
    },
    {
      "epoch": 3.9520263671875e-06,
      "model_forward_time": 5.60415244102478,
      "step": 4144
    },
    {
      "epoch": 3.9520263671875e-06,
      "step": 4144,
      "training_step_time": 17.079625129699707
    },
    {
      "epoch": 3.952980041503906e-06,
      "model_forward_time": 5.6068809032440186,
      "step": 4145
    },
    {
      "epoch": 3.952980041503906e-06,
      "step": 4145,
      "training_step_time": 17.088109731674194
    },
    {
      "epoch": 3.953933715820312e-06,
      "model_forward_time": 5.9478349685668945,
      "step": 4146
    },
    {
      "epoch": 3.953933715820312e-06,
      "step": 4146,
      "training_step_time": 17.43699550628662
    },
    {
      "epoch": 3.954887390136719e-06,
      "model_forward_time": 5.6067399978637695,
      "step": 4147
    },
    {
      "epoch": 3.954887390136719e-06,
      "step": 4147,
      "training_step_time": 17.068923473358154
    },
    {
      "epoch": 3.955841064453125e-06,
      "model_forward_time": 5.6008827686309814,
      "step": 4148
    },
    {
      "epoch": 3.955841064453125e-06,
      "step": 4148,
      "training_step_time": 17.04284167289734
    },
    {
      "epoch": 3.956794738769531e-06,
      "model_forward_time": 5.592556715011597,
      "step": 4149
    },
    {
      "epoch": 3.956794738769531e-06,
      "step": 4149,
      "training_step_time": 17.03341317176819
    },
    {
      "epoch": 3.957748413085938e-06,
      "grad_norm": 0.8261658549308777,
      "learning_rate": 1e-05,
      "loss": 0.0679,
      "step": 4150
    },
    {
      "epoch": 3.957748413085938e-06,
      "model_forward_time": 5.602823734283447,
      "step": 4150
    },
    {
      "dynamics_loss_avg": 0.047860400285571814,
      "epoch": 3.957748413085938e-06,
      "step": 4150
    },
    {
      "action_loss_avg": 0.01628040524665266,
      "epoch": 3.957748413085938e-06,
      "step": 4150
    },
    {
      "epoch": 3.957748413085938e-06,
      "step": 4150,
      "training_step_time": 17.09377694129944
    },
    {
      "epoch": 3.9587020874023436e-06,
      "model_forward_time": 5.96588659286499,
      "step": 4151
    },
    {
      "epoch": 3.9587020874023436e-06,
      "step": 4151,
      "training_step_time": 17.45061159133911
    },
    {
      "epoch": 3.95965576171875e-06,
      "model_forward_time": 5.608561992645264,
      "step": 4152
    },
    {
      "epoch": 3.95965576171875e-06,
      "step": 4152,
      "training_step_time": 17.091026306152344
    },
    {
      "epoch": 3.960609436035156e-06,
      "model_forward_time": 5.616730213165283,
      "step": 4153
    },
    {
      "epoch": 3.960609436035156e-06,
      "step": 4153,
      "training_step_time": 17.10228943824768
    },
    {
      "epoch": 3.961563110351562e-06,
      "model_forward_time": 5.6100850105285645,
      "step": 4154
    },
    {
      "epoch": 3.961563110351562e-06,
      "step": 4154,
      "training_step_time": 17.08133316040039
    },
    {
      "epoch": 3.962516784667969e-06,
      "model_forward_time": 5.599956512451172,
      "step": 4155
    },
    {
      "epoch": 3.962516784667969e-06,
      "step": 4155,
      "training_step_time": 17.02986240386963
    },
    {
      "epoch": 3.963470458984375e-06,
      "model_forward_time": 5.93517541885376,
      "step": 4156
    },
    {
      "epoch": 3.963470458984375e-06,
      "step": 4156,
      "training_step_time": 17.37400460243225
    },
    {
      "epoch": 3.964424133300781e-06,
      "model_forward_time": 5.595848083496094,
      "step": 4157
    },
    {
      "epoch": 3.964424133300781e-06,
      "step": 4157,
      "training_step_time": 17.066477060317993
    },
    {
      "epoch": 3.965377807617188e-06,
      "model_forward_time": 5.598581314086914,
      "step": 4158
    },
    {
      "epoch": 3.965377807617188e-06,
      "step": 4158,
      "training_step_time": 17.086716413497925
    },
    {
      "epoch": 3.9663314819335936e-06,
      "model_forward_time": 5.6189610958099365,
      "step": 4159
    },
    {
      "epoch": 3.9663314819335936e-06,
      "step": 4159,
      "training_step_time": 17.111223220825195
    },
    {
      "epoch": 3.96728515625e-06,
      "grad_norm": 1.154711365699768,
      "learning_rate": 1e-05,
      "loss": 0.0715,
      "step": 4160
    },
    {
      "epoch": 3.96728515625e-06,
      "model_forward_time": 5.599998235702515,
      "step": 4160
    },
    {
      "dynamics_loss_avg": 0.06580087970942258,
      "epoch": 3.96728515625e-06,
      "step": 4160
    },
    {
      "action_loss_avg": 0.01790576484054327,
      "epoch": 3.96728515625e-06,
      "step": 4160
    },
    {
      "epoch": 3.96728515625e-06,
      "step": 4160,
      "training_step_time": 17.065199851989746
    },
    {
      "epoch": 3.968238830566406e-06,
      "model_forward_time": 5.587592601776123,
      "step": 4161
    },
    {
      "epoch": 3.968238830566406e-06,
      "step": 4161,
      "training_step_time": 17.016947269439697
    },
    {
      "epoch": 3.9691925048828124e-06,
      "model_forward_time": 5.951806306838989,
      "step": 4162
    },
    {
      "epoch": 3.9691925048828124e-06,
      "step": 4162,
      "training_step_time": 17.385843515396118
    },
    {
      "epoch": 3.970146179199219e-06,
      "model_forward_time": 5.591429233551025,
      "step": 4163
    },
    {
      "epoch": 3.970146179199219e-06,
      "step": 4163,
      "training_step_time": 17.023421049118042
    },
    {
      "epoch": 3.971099853515625e-06,
      "model_forward_time": 5.591655015945435,
      "step": 4164
    },
    {
      "epoch": 3.971099853515625e-06,
      "step": 4164,
      "training_step_time": 17.0432345867157
    },
    {
      "epoch": 3.972053527832031e-06,
      "model_forward_time": 5.598872184753418,
      "step": 4165
    },
    {
      "epoch": 3.972053527832031e-06,
      "step": 4165,
      "training_step_time": 17.066497564315796
    },
    {
      "epoch": 3.973007202148438e-06,
      "model_forward_time": 5.6057915687561035,
      "step": 4166
    },
    {
      "epoch": 3.973007202148438e-06,
      "step": 4166,
      "training_step_time": 17.094402313232422
    },
    {
      "epoch": 3.9739608764648436e-06,
      "model_forward_time": 5.605681419372559,
      "step": 4167
    },
    {
      "epoch": 3.9739608764648436e-06,
      "step": 4167,
      "training_step_time": 17.453418493270874
    },
    {
      "epoch": 3.97491455078125e-06,
      "model_forward_time": 5.613061189651489,
      "step": 4168
    },
    {
      "epoch": 3.97491455078125e-06,
      "step": 4168,
      "training_step_time": 17.110119342803955
    },
    {
      "epoch": 3.975868225097656e-06,
      "model_forward_time": 5.61256742477417,
      "step": 4169
    },
    {
      "epoch": 3.975868225097656e-06,
      "step": 4169,
      "training_step_time": 17.098761320114136
    },
    {
      "epoch": 3.9768218994140624e-06,
      "grad_norm": 0.9664953947067261,
      "learning_rate": 1e-05,
      "loss": 0.0745,
      "step": 4170
    },
    {
      "epoch": 3.9768218994140624e-06,
      "model_forward_time": 5.595979690551758,
      "step": 4170
    },
    {
      "dynamics_loss_avg": 0.05554791819304228,
      "epoch": 3.9768218994140624e-06,
      "step": 4170
    },
    {
      "action_loss_avg": 0.014897032966837287,
      "epoch": 3.9768218994140624e-06,
      "step": 4170
    },
    {
      "epoch": 3.9768218994140624e-06,
      "step": 4170,
      "training_step_time": 17.031121253967285
    },
    {
      "epoch": 3.977775573730469e-06,
      "model_forward_time": 5.584995746612549,
      "step": 4171
    },
    {
      "epoch": 3.977775573730469e-06,
      "step": 4171,
      "training_step_time": 17.016945838928223
    },
    {
      "epoch": 3.978729248046875e-06,
      "model_forward_time": 5.589223861694336,
      "step": 4172
    },
    {
      "epoch": 3.978729248046875e-06,
      "step": 4172,
      "training_step_time": 17.026275873184204
    },
    {
      "epoch": 3.979682922363281e-06,
      "model_forward_time": 5.949004411697388,
      "step": 4173
    },
    {
      "epoch": 3.979682922363281e-06,
      "step": 4173,
      "training_step_time": 17.384862422943115
    },
    {
      "epoch": 3.980636596679688e-06,
      "model_forward_time": 5.592857360839844,
      "step": 4174
    },
    {
      "epoch": 3.980636596679688e-06,
      "step": 4174,
      "training_step_time": 17.059608936309814
    },
    {
      "epoch": 3.981590270996094e-06,
      "model_forward_time": 5.606676816940308,
      "step": 4175
    },
    {
      "epoch": 3.981590270996094e-06,
      "step": 4175,
      "training_step_time": 17.067317247390747
    },
    {
      "epoch": 3.9825439453125e-06,
      "model_forward_time": 5.611940383911133,
      "step": 4176
    },
    {
      "epoch": 3.9825439453125e-06,
      "step": 4176,
      "training_step_time": 17.081725597381592
    },
    {
      "epoch": 3.983497619628906e-06,
      "model_forward_time": 5.607123374938965,
      "step": 4177
    },
    {
      "epoch": 3.983497619628906e-06,
      "step": 4177,
      "training_step_time": 17.069532871246338
    },
    {
      "epoch": 3.9844512939453124e-06,
      "model_forward_time": 5.607348680496216,
      "step": 4178
    },
    {
      "epoch": 3.9844512939453124e-06,
      "step": 4178,
      "training_step_time": 17.4255952835083
    },
    {
      "epoch": 3.985404968261719e-06,
      "model_forward_time": 5.588764429092407,
      "step": 4179
    },
    {
      "epoch": 3.985404968261719e-06,
      "step": 4179,
      "training_step_time": 17.03768825531006
    },
    {
      "epoch": 3.986358642578125e-06,
      "grad_norm": 1.2885223627090454,
      "learning_rate": 1e-05,
      "loss": 0.0858,
      "step": 4180
    },
    {
      "epoch": 3.986358642578125e-06,
      "model_forward_time": 5.580869674682617,
      "step": 4180
    },
    {
      "dynamics_loss_avg": 0.061601484566926955,
      "epoch": 3.986358642578125e-06,
      "step": 4180
    },
    {
      "action_loss_avg": 0.020956796733662485,
      "epoch": 3.986358642578125e-06,
      "step": 4180
    },
    {
      "epoch": 3.986358642578125e-06,
      "step": 4180,
      "training_step_time": 17.030704975128174
    },
    {
      "epoch": 3.987312316894531e-06,
      "model_forward_time": 5.60107946395874,
      "step": 4181
    },
    {
      "epoch": 3.987312316894531e-06,
      "step": 4181,
      "training_step_time": 17.05887007713318
    },
    {
      "epoch": 3.988265991210938e-06,
      "model_forward_time": 5.603322267532349,
      "step": 4182
    },
    {
      "epoch": 3.988265991210938e-06,
      "step": 4182,
      "training_step_time": 17.08446979522705
    },
    {
      "epoch": 3.989219665527344e-06,
      "model_forward_time": 5.621089935302734,
      "step": 4183
    },
    {
      "epoch": 3.989219665527344e-06,
      "step": 4183,
      "training_step_time": 17.12939143180847
    },
    {
      "epoch": 3.99017333984375e-06,
      "model_forward_time": 5.987838268280029,
      "step": 4184
    },
    {
      "epoch": 3.99017333984375e-06,
      "step": 4184,
      "training_step_time": 17.483803272247314
    },
    {
      "epoch": 3.991127014160156e-06,
      "model_forward_time": 5.604642868041992,
      "step": 4185
    },
    {
      "epoch": 3.991127014160156e-06,
      "step": 4185,
      "training_step_time": 17.093977212905884
    },
    {
      "epoch": 3.9920806884765624e-06,
      "model_forward_time": 5.6088902950286865,
      "step": 4186
    },
    {
      "epoch": 3.9920806884765624e-06,
      "step": 4186,
      "training_step_time": 17.091678619384766
    },
    {
      "epoch": 3.993034362792969e-06,
      "model_forward_time": 5.608818531036377,
      "step": 4187
    },
    {
      "epoch": 3.993034362792969e-06,
      "step": 4187,
      "training_step_time": 17.10064458847046
    },
    {
      "epoch": 3.993988037109375e-06,
      "model_forward_time": 5.610711574554443,
      "step": 4188
    },
    {
      "epoch": 3.993988037109375e-06,
      "step": 4188,
      "training_step_time": 17.118833303451538
    },
    {
      "epoch": 3.994941711425781e-06,
      "model_forward_time": 5.967352390289307,
      "step": 4189
    },
    {
      "epoch": 3.994941711425781e-06,
      "step": 4189,
      "training_step_time": 17.45539903640747
    },
    {
      "epoch": 3.995895385742188e-06,
      "grad_norm": 0.34987547993659973,
      "learning_rate": 1e-05,
      "loss": 0.0753,
      "step": 4190
    },
    {
      "epoch": 3.995895385742188e-06,
      "model_forward_time": 5.604928970336914,
      "step": 4190
    },
    {
      "dynamics_loss_avg": 0.0654927721247077,
      "epoch": 3.995895385742188e-06,
      "step": 4190
    },
    {
      "action_loss_avg": 0.017592640966176985,
      "epoch": 3.995895385742188e-06,
      "step": 4190
    },
    {
      "epoch": 3.995895385742188e-06,
      "step": 4190,
      "training_step_time": 17.08647394180298
    },
    {
      "epoch": 3.996849060058594e-06,
      "model_forward_time": 5.610537528991699,
      "step": 4191
    },
    {
      "epoch": 3.996849060058594e-06,
      "step": 4191,
      "training_step_time": 17.099185466766357
    },
    {
      "epoch": 3.997802734375e-06,
      "model_forward_time": 5.609918117523193,
      "step": 4192
    },
    {
      "epoch": 3.997802734375e-06,
      "step": 4192,
      "training_step_time": 17.112513065338135
    },
    {
      "epoch": 3.998756408691406e-06,
      "model_forward_time": 5.61053729057312,
      "step": 4193
    },
    {
      "epoch": 3.998756408691406e-06,
      "step": 4193,
      "training_step_time": 17.104324102401733
    },
    {
      "epoch": 3.9997100830078125e-06,
      "model_forward_time": 5.609896659851074,
      "step": 4194
    },
    {
      "epoch": 3.9997100830078125e-06,
      "step": 4194,
      "training_step_time": 17.09288239479065
    },
    {
      "epoch": 4.000663757324219e-06,
      "model_forward_time": 5.970376968383789,
      "step": 4195
    },
    {
      "epoch": 4.000663757324219e-06,
      "step": 4195,
      "training_step_time": 17.453354358673096
    },
    {
      "epoch": 4.001617431640625e-06,
      "model_forward_time": 5.606013536453247,
      "step": 4196
    },
    {
      "epoch": 4.001617431640625e-06,
      "step": 4196,
      "training_step_time": 17.057785749435425
    },
    {
      "epoch": 4.002571105957031e-06,
      "model_forward_time": 5.5944178104400635,
      "step": 4197
    },
    {
      "epoch": 4.002571105957031e-06,
      "step": 4197,
      "training_step_time": 17.040770053863525
    },
    {
      "epoch": 4.003524780273438e-06,
      "model_forward_time": 5.578660726547241,
      "step": 4198
    },
    {
      "epoch": 4.003524780273438e-06,
      "step": 4198,
      "training_step_time": 17.02975630760193
    },
    {
      "epoch": 4.004478454589844e-06,
      "model_forward_time": 5.585911750793457,
      "step": 4199
    },
    {
      "epoch": 4.004478454589844e-06,
      "step": 4199,
      "training_step_time": 17.04030966758728
    },
    {
      "epoch": 4.00543212890625e-06,
      "grad_norm": 0.8732214570045471,
      "learning_rate": 1e-05,
      "loss": 0.0736,
      "step": 4200
    },
    {
      "epoch": 4.00543212890625e-06,
      "model_forward_time": 5.5940539836883545,
      "step": 4200
    },
    {
      "dynamics_loss_avg": 0.05340275932103396,
      "epoch": 4.00543212890625e-06,
      "step": 4200
    },
    {
      "action_loss_avg": 0.018985177390277386,
      "epoch": 4.00543212890625e-06,
      "step": 4200
    },
    {
      "epoch": 4.00543212890625e-06,
      "step": 4200,
      "training_step_time": 17.435171842575073
    },
    {
      "epoch": 4.006385803222656e-06,
      "model_forward_time": 5.607433557510376,
      "step": 4201
    },
    {
      "epoch": 4.006385803222656e-06,
      "step": 4201,
      "training_step_time": 17.118447303771973
    },
    {
      "epoch": 4.0073394775390625e-06,
      "model_forward_time": 5.615504503250122,
      "step": 4202
    },
    {
      "epoch": 4.0073394775390625e-06,
      "step": 4202,
      "training_step_time": 17.131202459335327
    },
    {
      "epoch": 4.008293151855469e-06,
      "model_forward_time": 5.617387294769287,
      "step": 4203
    },
    {
      "epoch": 4.008293151855469e-06,
      "step": 4203,
      "training_step_time": 17.11212468147278
    },
    {
      "epoch": 4.009246826171875e-06,
      "model_forward_time": 5.61270546913147,
      "step": 4204
    },
    {
      "epoch": 4.009246826171875e-06,
      "step": 4204,
      "training_step_time": 17.103680849075317
    },
    {
      "epoch": 4.010200500488281e-06,
      "model_forward_time": 5.606507301330566,
      "step": 4205
    },
    {
      "epoch": 4.010200500488281e-06,
      "step": 4205,
      "training_step_time": 17.0722553730011
    },
    {
      "epoch": 4.011154174804688e-06,
      "model_forward_time": 5.96475887298584,
      "step": 4206
    },
    {
      "epoch": 4.011154174804688e-06,
      "step": 4206,
      "training_step_time": 17.444390058517456
    },
    {
      "epoch": 4.012107849121094e-06,
      "model_forward_time": 5.598586082458496,
      "step": 4207
    },
    {
      "epoch": 4.012107849121094e-06,
      "step": 4207,
      "training_step_time": 17.052814960479736
    },
    {
      "epoch": 4.0130615234375e-06,
      "model_forward_time": 5.5939953327178955,
      "step": 4208
    },
    {
      "epoch": 4.0130615234375e-06,
      "step": 4208,
      "training_step_time": 17.046319723129272
    },
    {
      "epoch": 4.014015197753906e-06,
      "model_forward_time": 5.599618196487427,
      "step": 4209
    },
    {
      "epoch": 4.014015197753906e-06,
      "step": 4209,
      "training_step_time": 17.07967257499695
    },
    {
      "epoch": 4.0149688720703125e-06,
      "grad_norm": 0.553063690662384,
      "learning_rate": 1e-05,
      "loss": 0.0817,
      "step": 4210
    },
    {
      "epoch": 4.0149688720703125e-06,
      "model_forward_time": 5.618419885635376,
      "step": 4210
    },
    {
      "dynamics_loss_avg": 0.0580106626264751,
      "epoch": 4.0149688720703125e-06,
      "step": 4210
    },
    {
      "action_loss_avg": 0.01768672143807635,
      "epoch": 4.0149688720703125e-06,
      "step": 4210
    },
    {
      "epoch": 4.0149688720703125e-06,
      "step": 4210,
      "training_step_time": 17.120001316070557
    },
    {
      "epoch": 4.015922546386719e-06,
      "model_forward_time": 5.620526075363159,
      "step": 4211
    },
    {
      "epoch": 4.015922546386719e-06,
      "step": 4211,
      "training_step_time": 17.501548528671265
    },
    {
      "epoch": 4.016876220703125e-06,
      "model_forward_time": 5.622798204421997,
      "step": 4212
    },
    {
      "epoch": 4.016876220703125e-06,
      "step": 4212,
      "training_step_time": 17.121124029159546
    },
    {
      "epoch": 4.017829895019531e-06,
      "model_forward_time": 5.6083080768585205,
      "step": 4213
    },
    {
      "epoch": 4.017829895019531e-06,
      "step": 4213,
      "training_step_time": 17.10589027404785
    },
    {
      "epoch": 4.018783569335938e-06,
      "model_forward_time": 5.608374118804932,
      "step": 4214
    },
    {
      "epoch": 4.018783569335938e-06,
      "step": 4214,
      "training_step_time": 17.08677101135254
    },
    {
      "epoch": 4.019737243652344e-06,
      "model_forward_time": 5.6030848026275635,
      "step": 4215
    },
    {
      "epoch": 4.019737243652344e-06,
      "step": 4215,
      "training_step_time": 17.078924655914307
    },
    {
      "epoch": 4.02069091796875e-06,
      "model_forward_time": 5.602721452713013,
      "step": 4216
    },
    {
      "epoch": 4.02069091796875e-06,
      "step": 4216,
      "training_step_time": 17.054703950881958
    },
    {
      "epoch": 4.021644592285156e-06,
      "model_forward_time": 5.966429948806763,
      "step": 4217
    },
    {
      "epoch": 4.021644592285156e-06,
      "step": 4217,
      "training_step_time": 17.40040636062622
    },
    {
      "epoch": 4.0225982666015625e-06,
      "model_forward_time": 5.58808159828186,
      "step": 4218
    },
    {
      "epoch": 4.0225982666015625e-06,
      "step": 4218,
      "training_step_time": 17.01462960243225
    },
    {
      "epoch": 4.023551940917969e-06,
      "model_forward_time": 5.591702699661255,
      "step": 4219
    },
    {
      "epoch": 4.023551940917969e-06,
      "step": 4219,
      "training_step_time": 17.037768840789795
    },
    {
      "epoch": 4.024505615234375e-06,
      "grad_norm": 0.40111833810806274,
      "learning_rate": 1e-05,
      "loss": 0.0733,
      "step": 4220
    },
    {
      "epoch": 4.024505615234375e-06,
      "model_forward_time": 5.593188762664795,
      "step": 4220
    },
    {
      "dynamics_loss_avg": 0.054365018382668495,
      "epoch": 4.024505615234375e-06,
      "step": 4220
    },
    {
      "action_loss_avg": 0.021479146694764494,
      "epoch": 4.024505615234375e-06,
      "step": 4220
    },
    {
      "epoch": 4.024505615234375e-06,
      "step": 4220,
      "training_step_time": 17.057136058807373
    },
    {
      "epoch": 4.025459289550781e-06,
      "model_forward_time": 5.596748352050781,
      "step": 4221
    },
    {
      "epoch": 4.025459289550781e-06,
      "step": 4221,
      "training_step_time": 17.060481548309326
    },
    {
      "epoch": 4.026412963867188e-06,
      "model_forward_time": 5.9590113162994385,
      "step": 4222
    },
    {
      "epoch": 4.026412963867188e-06,
      "step": 4222,
      "training_step_time": 17.436277866363525
    },
    {
      "epoch": 4.027366638183594e-06,
      "model_forward_time": 5.610644578933716,
      "step": 4223
    },
    {
      "epoch": 4.027366638183594e-06,
      "step": 4223,
      "training_step_time": 17.08340048789978
    },
    {
      "epoch": 4.0283203125e-06,
      "model_forward_time": 5.5865983963012695,
      "step": 4224
    },
    {
      "epoch": 4.0283203125e-06,
      "step": 4224,
      "training_step_time": 17.0219464302063
    },
    {
      "epoch": 4.029273986816406e-06,
      "model_forward_time": 5.574106931686401,
      "step": 4225
    },
    {
      "epoch": 4.029273986816406e-06,
      "step": 4225,
      "training_step_time": 17.0022451877594
    },
    {
      "epoch": 4.0302276611328125e-06,
      "model_forward_time": 5.590684175491333,
      "step": 4226
    },
    {
      "epoch": 4.0302276611328125e-06,
      "step": 4226,
      "training_step_time": 17.008867740631104
    },
    {
      "epoch": 4.031181335449219e-06,
      "model_forward_time": 5.596746921539307,
      "step": 4227
    },
    {
      "epoch": 4.031181335449219e-06,
      "step": 4227,
      "training_step_time": 17.05827260017395
    },
    {
      "epoch": 4.032135009765625e-06,
      "model_forward_time": 5.964820861816406,
      "step": 4228
    },
    {
      "epoch": 4.032135009765625e-06,
      "step": 4228,
      "training_step_time": 17.432687759399414
    },
    {
      "epoch": 4.033088684082031e-06,
      "model_forward_time": 5.6073126792907715,
      "step": 4229
    },
    {
      "epoch": 4.033088684082031e-06,
      "step": 4229,
      "training_step_time": 17.073684453964233
    },
    {
      "epoch": 4.034042358398438e-06,
      "grad_norm": 1.1812719106674194,
      "learning_rate": 1e-05,
      "loss": 0.0805,
      "step": 4230
    },
    {
      "epoch": 4.034042358398438e-06,
      "model_forward_time": 5.604870080947876,
      "step": 4230
    },
    {
      "dynamics_loss_avg": 0.06035957336425781,
      "epoch": 4.034042358398438e-06,
      "step": 4230
    },
    {
      "action_loss_avg": 0.018190334644168617,
      "epoch": 4.034042358398438e-06,
      "step": 4230
    },
    {
      "epoch": 4.034042358398438e-06,
      "step": 4230,
      "training_step_time": 17.082765579223633
    },
    {
      "epoch": 4.034996032714844e-06,
      "model_forward_time": 5.614556074142456,
      "step": 4231
    },
    {
      "epoch": 4.034996032714844e-06,
      "step": 4231,
      "training_step_time": 17.10477900505066
    },
    {
      "epoch": 4.03594970703125e-06,
      "model_forward_time": 5.6028218269348145,
      "step": 4232
    },
    {
      "epoch": 4.03594970703125e-06,
      "step": 4232,
      "training_step_time": 17.080809354782104
    },
    {
      "epoch": 4.036903381347656e-06,
      "model_forward_time": 5.6140828132629395,
      "step": 4233
    },
    {
      "epoch": 4.036903381347656e-06,
      "step": 4233,
      "training_step_time": 17.477365016937256
    },
    {
      "epoch": 4.0378570556640625e-06,
      "model_forward_time": 5.605398893356323,
      "step": 4234
    },
    {
      "epoch": 4.0378570556640625e-06,
      "step": 4234,
      "training_step_time": 17.101377964019775
    },
    {
      "epoch": 4.038810729980469e-06,
      "model_forward_time": 5.606285572052002,
      "step": 4235
    },
    {
      "epoch": 4.038810729980469e-06,
      "step": 4235,
      "training_step_time": 17.105735778808594
    },
    {
      "epoch": 4.039764404296875e-06,
      "model_forward_time": 5.608640432357788,
      "step": 4236
    },
    {
      "epoch": 4.039764404296875e-06,
      "step": 4236,
      "training_step_time": 17.099863052368164
    },
    {
      "epoch": 4.040718078613281e-06,
      "model_forward_time": 5.606200456619263,
      "step": 4237
    },
    {
      "epoch": 4.040718078613281e-06,
      "step": 4237,
      "training_step_time": 17.079315662384033
    },
    {
      "epoch": 4.041671752929688e-06,
      "model_forward_time": 5.6056602001190186,
      "step": 4238
    },
    {
      "epoch": 4.041671752929688e-06,
      "step": 4238,
      "training_step_time": 17.079219341278076
    },
    {
      "epoch": 4.042625427246094e-06,
      "model_forward_time": 5.973820209503174,
      "step": 4239
    },
    {
      "epoch": 4.042625427246094e-06,
      "step": 4239,
      "training_step_time": 17.445545196533203
    },
    {
      "epoch": 4.0435791015625e-06,
      "grad_norm": 1.345526099205017,
      "learning_rate": 1e-05,
      "loss": 0.0603,
      "step": 4240
    },
    {
      "epoch": 4.0435791015625e-06,
      "model_forward_time": 5.605602741241455,
      "step": 4240
    },
    {
      "dynamics_loss_avg": 0.04669672884047031,
      "epoch": 4.0435791015625e-06,
      "step": 4240
    },
    {
      "action_loss_avg": 0.01548418290913105,
      "epoch": 4.0435791015625e-06,
      "step": 4240
    },
    {
      "epoch": 4.0435791015625e-06,
      "step": 4240,
      "training_step_time": 17.08232808113098
    },
    {
      "epoch": 4.044532775878906e-06,
      "model_forward_time": 5.607630491256714,
      "step": 4241
    },
    {
      "epoch": 4.044532775878906e-06,
      "step": 4241,
      "training_step_time": 17.09642267227173
    },
    {
      "epoch": 4.0454864501953125e-06,
      "model_forward_time": 5.608163833618164,
      "step": 4242
    },
    {
      "epoch": 4.0454864501953125e-06,
      "step": 4242,
      "training_step_time": 17.096404552459717
    },
    {
      "epoch": 4.046440124511719e-06,
      "model_forward_time": 5.6113715171813965,
      "step": 4243
    },
    {
      "epoch": 4.046440124511719e-06,
      "step": 4243,
      "training_step_time": 17.08943772315979
    },
    {
      "epoch": 4.047393798828125e-06,
      "model_forward_time": 5.602006196975708,
      "step": 4244
    },
    {
      "epoch": 4.047393798828125e-06,
      "step": 4244,
      "training_step_time": 17.43020486831665
    },
    {
      "epoch": 4.048347473144531e-06,
      "model_forward_time": 5.5905983448028564,
      "step": 4245
    },
    {
      "epoch": 4.048347473144531e-06,
      "step": 4245,
      "training_step_time": 17.02419352531433
    },
    {
      "epoch": 4.049301147460938e-06,
      "model_forward_time": 5.577720642089844,
      "step": 4246
    },
    {
      "epoch": 4.049301147460938e-06,
      "step": 4246,
      "training_step_time": 17.014514684677124
    },
    {
      "epoch": 4.050254821777344e-06,
      "model_forward_time": 5.597415924072266,
      "step": 4247
    },
    {
      "epoch": 4.050254821777344e-06,
      "step": 4247,
      "training_step_time": 17.042525053024292
    },
    {
      "epoch": 4.05120849609375e-06,
      "model_forward_time": 5.596583127975464,
      "step": 4248
    },
    {
      "epoch": 4.05120849609375e-06,
      "step": 4248,
      "training_step_time": 17.056633472442627
    },
    {
      "epoch": 4.052162170410156e-06,
      "model_forward_time": 5.598047494888306,
      "step": 4249
    },
    {
      "epoch": 4.052162170410156e-06,
      "step": 4249,
      "training_step_time": 17.06048274040222
    },
    {
      "epoch": 4.0531158447265625e-06,
      "grad_norm": 1.2234153747558594,
      "learning_rate": 1e-05,
      "loss": 0.0759,
      "step": 4250
    },
    {
      "epoch": 4.0531158447265625e-06,
      "model_forward_time": 5.958560466766357,
      "step": 4250
    },
    {
      "dynamics_loss_avg": 0.05393535643815994,
      "epoch": 4.0531158447265625e-06,
      "step": 4250
    },
    {
      "action_loss_avg": 0.02179248807951808,
      "epoch": 4.0531158447265625e-06,
      "step": 4250
    },
    {
      "epoch": 4.0531158447265625e-06,
      "step": 4250,
      "training_step_time": 17.41611957550049
    },
    {
      "epoch": 4.054069519042969e-06,
      "model_forward_time": 5.594675064086914,
      "step": 4251
    },
    {
      "epoch": 4.054069519042969e-06,
      "step": 4251,
      "training_step_time": 17.025363206863403
    },
    {
      "epoch": 4.055023193359375e-06,
      "model_forward_time": 5.585770606994629,
      "step": 4252
    },
    {
      "epoch": 4.055023193359375e-06,
      "step": 4252,
      "training_step_time": 17.01025366783142
    },
    {
      "epoch": 4.055976867675781e-06,
      "model_forward_time": 5.577908515930176,
      "step": 4253
    },
    {
      "epoch": 4.055976867675781e-06,
      "step": 4253,
      "training_step_time": 17.014770030975342
    },
    {
      "epoch": 4.056930541992187e-06,
      "model_forward_time": 5.6081154346466064,
      "step": 4254
    },
    {
      "epoch": 4.056930541992187e-06,
      "step": 4254,
      "training_step_time": 17.09458565711975
    },
    {
      "epoch": 4.057884216308594e-06,
      "model_forward_time": 5.983726263046265,
      "step": 4255
    },
    {
      "epoch": 4.057884216308594e-06,
      "step": 4255,
      "training_step_time": 17.47417950630188
    },
    {
      "epoch": 4.058837890625e-06,
      "model_forward_time": 5.60728120803833,
      "step": 4256
    },
    {
      "epoch": 4.058837890625e-06,
      "step": 4256,
      "training_step_time": 17.076424837112427
    },
    {
      "epoch": 4.059791564941406e-06,
      "model_forward_time": 5.604836463928223,
      "step": 4257
    },
    {
      "epoch": 4.059791564941406e-06,
      "step": 4257,
      "training_step_time": 17.066328525543213
    },
    {
      "epoch": 4.0607452392578125e-06,
      "model_forward_time": 5.595311164855957,
      "step": 4258
    },
    {
      "epoch": 4.0607452392578125e-06,
      "step": 4258,
      "training_step_time": 17.04177951812744
    },
    {
      "epoch": 4.061698913574219e-06,
      "model_forward_time": 4.265632390975952,
      "step": 4259
    },
    {
      "epoch": 4.061698913574219e-06,
      "step": 4259,
      "training_step_time": 12.777074098587036
    },
    {
      "epoch": 4.062652587890625e-06,
      "grad_norm": 0.9258387088775635,
      "learning_rate": 1e-05,
      "loss": 0.0763,
      "step": 4260
    },
    {
      "epoch": 4.062652587890625e-06,
      "model_forward_time": 5.592637062072754,
      "step": 4260
    },
    {
      "dynamics_loss_avg": 0.052486881241202356,
      "epoch": 4.062652587890625e-06,
      "step": 4260
    },
    {
      "action_loss_avg": 0.019367590919137,
      "epoch": 4.062652587890625e-06,
      "step": 4260
    },
    {
      "epoch": 4.062652587890625e-06,
      "step": 4260,
      "training_step_time": 17.059065341949463
    },
    {
      "epoch": 4.063606262207031e-06,
      "model_forward_time": 5.982841730117798,
      "step": 4261
    },
    {
      "epoch": 4.063606262207031e-06,
      "step": 4261,
      "training_step_time": 17.47088074684143
    },
    {
      "epoch": 4.064559936523437e-06,
      "model_forward_time": 5.61682391166687,
      "step": 4262
    },
    {
      "epoch": 4.064559936523437e-06,
      "step": 4262,
      "training_step_time": 17.089423179626465
    },
    {
      "epoch": 4.065513610839844e-06,
      "model_forward_time": 5.588050603866577,
      "step": 4263
    },
    {
      "epoch": 4.065513610839844e-06,
      "step": 4263,
      "training_step_time": 17.02364492416382
    },
    {
      "epoch": 4.06646728515625e-06,
      "model_forward_time": 5.5744123458862305,
      "step": 4264
    },
    {
      "epoch": 4.06646728515625e-06,
      "step": 4264,
      "training_step_time": 16.99159574508667
    },
    {
      "epoch": 4.067420959472656e-06,
      "model_forward_time": 5.573922395706177,
      "step": 4265
    },
    {
      "epoch": 4.067420959472656e-06,
      "step": 4265,
      "training_step_time": 16.97615337371826
    },
    {
      "epoch": 4.0683746337890625e-06,
      "model_forward_time": 5.572387456893921,
      "step": 4266
    },
    {
      "epoch": 4.0683746337890625e-06,
      "step": 4266,
      "training_step_time": 17.367039680480957
    },
    {
      "epoch": 4.069328308105469e-06,
      "model_forward_time": 5.574823379516602,
      "step": 4267
    },
    {
      "epoch": 4.069328308105469e-06,
      "step": 4267,
      "training_step_time": 17.00533151626587
    },
    {
      "epoch": 4.070281982421875e-06,
      "model_forward_time": 5.589047908782959,
      "step": 4268
    },
    {
      "epoch": 4.070281982421875e-06,
      "step": 4268,
      "training_step_time": 17.045233964920044
    },
    {
      "epoch": 4.071235656738281e-06,
      "model_forward_time": 5.600043535232544,
      "step": 4269
    },
    {
      "epoch": 4.071235656738281e-06,
      "step": 4269,
      "training_step_time": 17.07635807991028
    },
    {
      "epoch": 4.072189331054687e-06,
      "grad_norm": 0.7727416157722473,
      "learning_rate": 1e-05,
      "loss": 0.0677,
      "step": 4270
    },
    {
      "epoch": 4.072189331054687e-06,
      "model_forward_time": 5.599423885345459,
      "step": 4270
    },
    {
      "dynamics_loss_avg": 0.04909635465592146,
      "epoch": 4.072189331054687e-06,
      "step": 4270
    },
    {
      "action_loss_avg": 0.018060593400150537,
      "epoch": 4.072189331054687e-06,
      "step": 4270
    },
    {
      "epoch": 4.072189331054687e-06,
      "step": 4270,
      "training_step_time": 17.0457022190094
    },
    {
      "epoch": 4.073143005371094e-06,
      "model_forward_time": 5.588347434997559,
      "step": 4271
    },
    {
      "epoch": 4.073143005371094e-06,
      "step": 4271,
      "training_step_time": 17.05423641204834
    },
    {
      "epoch": 4.0740966796875e-06,
      "model_forward_time": 5.971200942993164,
      "step": 4272
    },
    {
      "epoch": 4.0740966796875e-06,
      "step": 4272,
      "training_step_time": 17.422687768936157
    },
    {
      "epoch": 4.075050354003906e-06,
      "model_forward_time": 5.596686840057373,
      "step": 4273
    },
    {
      "epoch": 4.075050354003906e-06,
      "step": 4273,
      "training_step_time": 17.05109429359436
    },
    {
      "epoch": 4.0760040283203125e-06,
      "model_forward_time": 5.600613117218018,
      "step": 4274
    },
    {
      "epoch": 4.0760040283203125e-06,
      "step": 4274,
      "training_step_time": 17.046746253967285
    },
    {
      "epoch": 4.076957702636719e-06,
      "model_forward_time": 5.608228921890259,
      "step": 4275
    },
    {
      "epoch": 4.076957702636719e-06,
      "step": 4275,
      "training_step_time": 17.080991744995117
    },
    {
      "epoch": 4.077911376953125e-06,
      "model_forward_time": 5.599566459655762,
      "step": 4276
    },
    {
      "epoch": 4.077911376953125e-06,
      "step": 4276,
      "training_step_time": 17.06423783302307
    },
    {
      "epoch": 4.078865051269531e-06,
      "model_forward_time": 5.6076319217681885,
      "step": 4277
    },
    {
      "epoch": 4.078865051269531e-06,
      "step": 4277,
      "training_step_time": 17.444347858428955
    },
    {
      "epoch": 4.079818725585937e-06,
      "model_forward_time": 5.609879493713379,
      "step": 4278
    },
    {
      "epoch": 4.079818725585937e-06,
      "step": 4278,
      "training_step_time": 17.084542989730835
    },
    {
      "epoch": 4.080772399902344e-06,
      "model_forward_time": 5.5981764793396,
      "step": 4279
    },
    {
      "epoch": 4.080772399902344e-06,
      "step": 4279,
      "training_step_time": 17.065044164657593
    },
    {
      "epoch": 4.08172607421875e-06,
      "grad_norm": 2.425896644592285,
      "learning_rate": 1e-05,
      "loss": 0.0804,
      "step": 4280
    },
    {
      "epoch": 4.08172607421875e-06,
      "model_forward_time": 5.602768898010254,
      "step": 4280
    },
    {
      "dynamics_loss_avg": 0.058888182044029236,
      "epoch": 4.08172607421875e-06,
      "step": 4280
    },
    {
      "action_loss_avg": 0.023610978201031686,
      "epoch": 4.08172607421875e-06,
      "step": 4280
    },
    {
      "epoch": 4.08172607421875e-06,
      "step": 4280,
      "training_step_time": 17.08552646636963
    },
    {
      "epoch": 4.082679748535156e-06,
      "model_forward_time": 5.608977794647217,
      "step": 4281
    },
    {
      "epoch": 4.082679748535156e-06,
      "step": 4281,
      "training_step_time": 17.09167170524597
    },
    {
      "epoch": 4.0836334228515625e-06,
      "model_forward_time": 5.610862731933594,
      "step": 4282
    },
    {
      "epoch": 4.0836334228515625e-06,
      "step": 4282,
      "training_step_time": 17.0793297290802
    },
    {
      "epoch": 4.084587097167969e-06,
      "model_forward_time": 5.981509447097778,
      "step": 4283
    },
    {
      "epoch": 4.084587097167969e-06,
      "step": 4283,
      "training_step_time": 17.46190905570984
    },
    {
      "epoch": 4.085540771484375e-06,
      "model_forward_time": 5.600420236587524,
      "step": 4284
    },
    {
      "epoch": 4.085540771484375e-06,
      "step": 4284,
      "training_step_time": 17.092654943466187
    },
    {
      "epoch": 4.086494445800781e-06,
      "model_forward_time": 5.602208852767944,
      "step": 4285
    },
    {
      "epoch": 4.086494445800781e-06,
      "step": 4285,
      "training_step_time": 17.059206247329712
    },
    {
      "epoch": 4.087448120117187e-06,
      "model_forward_time": 5.582520484924316,
      "step": 4286
    },
    {
      "epoch": 4.087448120117187e-06,
      "step": 4286,
      "training_step_time": 17.018065690994263
    },
    {
      "epoch": 4.088401794433594e-06,
      "model_forward_time": 5.571628093719482,
      "step": 4287
    },
    {
      "epoch": 4.088401794433594e-06,
      "step": 4287,
      "training_step_time": 16.974573373794556
    },
    {
      "epoch": 4.08935546875e-06,
      "model_forward_time": 5.9425458908081055,
      "step": 4288
    },
    {
      "epoch": 4.08935546875e-06,
      "step": 4288,
      "training_step_time": 17.357910871505737
    },
    {
      "epoch": 4.090309143066406e-06,
      "model_forward_time": 5.579526662826538,
      "step": 4289
    },
    {
      "epoch": 4.090309143066406e-06,
      "step": 4289,
      "training_step_time": 17.0115487575531
    },
    {
      "epoch": 4.0912628173828125e-06,
      "grad_norm": 1.9953559637069702,
      "learning_rate": 1e-05,
      "loss": 0.0888,
      "step": 4290
    },
    {
      "epoch": 4.0912628173828125e-06,
      "model_forward_time": 5.5791168212890625,
      "step": 4290
    },
    {
      "dynamics_loss_avg": 0.06737328059971333,
      "epoch": 4.0912628173828125e-06,
      "step": 4290
    },
    {
      "action_loss_avg": 0.027530250092968344,
      "epoch": 4.0912628173828125e-06,
      "step": 4290
    },
    {
      "epoch": 4.0912628173828125e-06,
      "step": 4290,
      "training_step_time": 16.99352264404297
    },
    {
      "epoch": 4.092216491699219e-06,
      "model_forward_time": 5.5827882289886475,
      "step": 4291
    },
    {
      "epoch": 4.092216491699219e-06,
      "step": 4291,
      "training_step_time": 17.008044242858887
    },
    {
      "epoch": 4.093170166015625e-06,
      "model_forward_time": 5.586214542388916,
      "step": 4292
    },
    {
      "epoch": 4.093170166015625e-06,
      "step": 4292,
      "training_step_time": 16.99726939201355
    },
    {
      "epoch": 4.094123840332031e-06,
      "model_forward_time": 5.580036401748657,
      "step": 4293
    },
    {
      "epoch": 4.094123840332031e-06,
      "step": 4293,
      "training_step_time": 17.01340889930725
    },
    {
      "epoch": 4.095077514648437e-06,
      "model_forward_time": 5.9654741287231445,
      "step": 4294
    },
    {
      "epoch": 4.095077514648437e-06,
      "step": 4294,
      "training_step_time": 17.41536283493042
    },
    {
      "epoch": 4.096031188964844e-06,
      "model_forward_time": 5.5933518409729,
      "step": 4295
    },
    {
      "epoch": 4.096031188964844e-06,
      "step": 4295,
      "training_step_time": 17.066771268844604
    },
    {
      "epoch": 4.09698486328125e-06,
      "model_forward_time": 5.5943145751953125,
      "step": 4296
    },
    {
      "epoch": 4.09698486328125e-06,
      "step": 4296,
      "training_step_time": 17.070816040039062
    },
    {
      "epoch": 4.097938537597656e-06,
      "model_forward_time": 5.60659384727478,
      "step": 4297
    },
    {
      "epoch": 4.097938537597656e-06,
      "step": 4297,
      "training_step_time": 17.06789207458496
    },
    {
      "epoch": 4.0988922119140625e-06,
      "model_forward_time": 5.593201398849487,
      "step": 4298
    },
    {
      "epoch": 4.0988922119140625e-06,
      "step": 4298,
      "training_step_time": 17.049139261245728
    },
    {
      "epoch": 4.099845886230469e-06,
      "model_forward_time": 5.599349498748779,
      "step": 4299
    },
    {
      "epoch": 4.099845886230469e-06,
      "step": 4299,
      "training_step_time": 17.43754768371582
    },
    {
      "epoch": 4.100799560546875e-06,
      "grad_norm": 1.2595916986465454,
      "learning_rate": 1e-05,
      "loss": 0.0843,
      "step": 4300
    },
    {
      "epoch": 4.100799560546875e-06,
      "model_forward_time": 5.604015588760376,
      "step": 4300
    },
    {
      "dynamics_loss_avg": 0.061955093033611776,
      "epoch": 4.100799560546875e-06,
      "step": 4300
    },
    {
      "action_loss_avg": 0.02180427119601518,
      "epoch": 4.100799560546875e-06,
      "step": 4300
    },
    {
      "epoch": 4.100799560546875e-06,
      "step": 4300,
      "training_step_time": 17.088266849517822
    },
    {
      "epoch": 4.101753234863281e-06,
      "model_forward_time": 5.609071969985962,
      "step": 4301
    },
    {
      "epoch": 4.101753234863281e-06,
      "step": 4301,
      "training_step_time": 17.099896669387817
    },
    {
      "epoch": 4.102706909179687e-06,
      "model_forward_time": 5.612060070037842,
      "step": 4302
    },
    {
      "epoch": 4.102706909179687e-06,
      "step": 4302,
      "training_step_time": 17.097689867019653
    },
    {
      "epoch": 4.103660583496094e-06,
      "model_forward_time": 5.61130166053772,
      "step": 4303
    },
    {
      "epoch": 4.103660583496094e-06,
      "step": 4303,
      "training_step_time": 17.099700212478638
    },
    {
      "epoch": 4.1046142578125e-06,
      "model_forward_time": 5.599120378494263,
      "step": 4304
    },
    {
      "epoch": 4.1046142578125e-06,
      "step": 4304,
      "training_step_time": 17.077527284622192
    },
    {
      "epoch": 4.105567932128906e-06,
      "model_forward_time": 5.96660852432251,
      "step": 4305
    },
    {
      "epoch": 4.105567932128906e-06,
      "step": 4305,
      "training_step_time": 17.43558120727539
    },
    {
      "epoch": 4.1065216064453125e-06,
      "model_forward_time": 5.602885007858276,
      "step": 4306
    },
    {
      "epoch": 4.1065216064453125e-06,
      "step": 4306,
      "training_step_time": 17.093465328216553
    },
    {
      "epoch": 4.107475280761719e-06,
      "model_forward_time": 5.5998430252075195,
      "step": 4307
    },
    {
      "epoch": 4.107475280761719e-06,
      "step": 4307,
      "training_step_time": 17.081472396850586
    },
    {
      "epoch": 4.108428955078125e-06,
      "model_forward_time": 5.603972434997559,
      "step": 4308
    },
    {
      "epoch": 4.108428955078125e-06,
      "step": 4308,
      "training_step_time": 17.08517837524414
    },
    {
      "epoch": 4.109382629394531e-06,
      "model_forward_time": 5.6171674728393555,
      "step": 4309
    },
    {
      "epoch": 4.109382629394531e-06,
      "step": 4309,
      "training_step_time": 17.11552119255066
    },
    {
      "epoch": 4.110336303710937e-06,
      "grad_norm": 1.4051786661148071,
      "learning_rate": 1e-05,
      "loss": 0.0664,
      "step": 4310
    },
    {
      "epoch": 4.110336303710937e-06,
      "model_forward_time": 5.612084865570068,
      "step": 4310
    },
    {
      "dynamics_loss_avg": 0.05176601894199848,
      "epoch": 4.110336303710937e-06,
      "step": 4310
    },
    {
      "action_loss_avg": 0.014060188084840775,
      "epoch": 4.110336303710937e-06,
      "step": 4310
    },
    {
      "epoch": 4.110336303710937e-06,
      "step": 4310,
      "training_step_time": 17.443806886672974
    },
    {
      "epoch": 4.111289978027344e-06,
      "model_forward_time": 5.596535682678223,
      "step": 4311
    },
    {
      "epoch": 4.111289978027344e-06,
      "step": 4311,
      "training_step_time": 17.034627676010132
    },
    {
      "epoch": 4.11224365234375e-06,
      "model_forward_time": 5.5773608684539795,
      "step": 4312
    },
    {
      "epoch": 4.11224365234375e-06,
      "step": 4312,
      "training_step_time": 17.02128291130066
    },
    {
      "epoch": 4.113197326660156e-06,
      "model_forward_time": 5.58177924156189,
      "step": 4313
    },
    {
      "epoch": 4.113197326660156e-06,
      "step": 4313,
      "training_step_time": 17.022715091705322
    },
    {
      "epoch": 4.1141510009765626e-06,
      "model_forward_time": 5.583216428756714,
      "step": 4314
    },
    {
      "epoch": 4.1141510009765626e-06,
      "step": 4314,
      "training_step_time": 17.006737232208252
    },
    {
      "epoch": 4.115104675292969e-06,
      "model_forward_time": 5.585062503814697,
      "step": 4315
    },
    {
      "epoch": 4.115104675292969e-06,
      "step": 4315,
      "training_step_time": 17.03599500656128
    },
    {
      "epoch": 4.116058349609375e-06,
      "model_forward_time": 5.975029230117798,
      "step": 4316
    },
    {
      "epoch": 4.116058349609375e-06,
      "step": 4316,
      "training_step_time": 17.42923402786255
    },
    {
      "epoch": 4.117012023925781e-06,
      "model_forward_time": 5.6048057079315186,
      "step": 4317
    },
    {
      "epoch": 4.117012023925781e-06,
      "step": 4317,
      "training_step_time": 17.070270776748657
    },
    {
      "epoch": 4.117965698242187e-06,
      "model_forward_time": 5.597084999084473,
      "step": 4318
    },
    {
      "epoch": 4.117965698242187e-06,
      "step": 4318,
      "training_step_time": 17.046876668930054
    },
    {
      "epoch": 4.118919372558594e-06,
      "model_forward_time": 5.601824760437012,
      "step": 4319
    },
    {
      "epoch": 4.118919372558594e-06,
      "step": 4319,
      "training_step_time": 17.039824724197388
    },
    {
      "epoch": 4.119873046875e-06,
      "grad_norm": 1.492274522781372,
      "learning_rate": 1e-05,
      "loss": 0.0927,
      "step": 4320
    },
    {
      "epoch": 4.119873046875e-06,
      "model_forward_time": 5.582397222518921,
      "step": 4320
    },
    {
      "dynamics_loss_avg": 0.07247309945523739,
      "epoch": 4.119873046875e-06,
      "step": 4320
    },
    {
      "action_loss_avg": 0.01921047465875745,
      "epoch": 4.119873046875e-06,
      "step": 4320
    },
    {
      "epoch": 4.119873046875e-06,
      "step": 4320,
      "training_step_time": 17.028198957443237
    },
    {
      "epoch": 4.120826721191406e-06,
      "model_forward_time": 5.93747615814209,
      "step": 4321
    },
    {
      "epoch": 4.120826721191406e-06,
      "step": 4321,
      "training_step_time": 17.373939275741577
    },
    {
      "epoch": 4.1217803955078126e-06,
      "model_forward_time": 5.599583387374878,
      "step": 4322
    },
    {
      "epoch": 4.1217803955078126e-06,
      "step": 4322,
      "training_step_time": 17.07903242111206
    },
    {
      "epoch": 4.122734069824219e-06,
      "model_forward_time": 5.601528644561768,
      "step": 4323
    },
    {
      "epoch": 4.122734069824219e-06,
      "step": 4323,
      "training_step_time": 17.07936668395996
    },
    {
      "epoch": 4.123687744140625e-06,
      "model_forward_time": 5.605759143829346,
      "step": 4324
    },
    {
      "epoch": 4.123687744140625e-06,
      "step": 4324,
      "training_step_time": 17.04199242591858
    },
    {
      "epoch": 4.124641418457031e-06,
      "model_forward_time": 5.582292079925537,
      "step": 4325
    },
    {
      "epoch": 4.124641418457031e-06,
      "step": 4325,
      "training_step_time": 16.986189603805542
    },
    {
      "epoch": 4.125595092773437e-06,
      "model_forward_time": 5.578262805938721,
      "step": 4326
    },
    {
      "epoch": 4.125595092773437e-06,
      "step": 4326,
      "training_step_time": 17.01330327987671
    },
    {
      "epoch": 4.126548767089844e-06,
      "model_forward_time": 5.959188461303711,
      "step": 4327
    },
    {
      "epoch": 4.126548767089844e-06,
      "step": 4327,
      "training_step_time": 17.389679193496704
    },
    {
      "epoch": 4.12750244140625e-06,
      "model_forward_time": 5.601346015930176,
      "step": 4328
    },
    {
      "epoch": 4.12750244140625e-06,
      "step": 4328,
      "training_step_time": 17.088820695877075
    },
    {
      "epoch": 4.128456115722656e-06,
      "model_forward_time": 5.619556665420532,
      "step": 4329
    },
    {
      "epoch": 4.128456115722656e-06,
      "step": 4329,
      "training_step_time": 17.1336829662323
    },
    {
      "epoch": 4.1294097900390626e-06,
      "grad_norm": 1.2940956354141235,
      "learning_rate": 1e-05,
      "loss": 0.0906,
      "step": 4330
    },
    {
      "epoch": 4.1294097900390626e-06,
      "model_forward_time": 5.6056578159332275,
      "step": 4330
    },
    {
      "dynamics_loss_avg": 0.07094719596207141,
      "epoch": 4.1294097900390626e-06,
      "step": 4330
    },
    {
      "action_loss_avg": 0.014240814838558435,
      "epoch": 4.1294097900390626e-06,
      "step": 4330
    },
    {
      "epoch": 4.1294097900390626e-06,
      "step": 4330,
      "training_step_time": 17.079764366149902
    },
    {
      "epoch": 4.130363464355469e-06,
      "model_forward_time": 5.602849721908569,
      "step": 4331
    },
    {
      "epoch": 4.130363464355469e-06,
      "step": 4331,
      "training_step_time": 17.081725597381592
    },
    {
      "epoch": 4.131317138671875e-06,
      "model_forward_time": 5.607542276382446,
      "step": 4332
    },
    {
      "epoch": 4.131317138671875e-06,
      "step": 4332,
      "training_step_time": 17.459164142608643
    },
    {
      "epoch": 4.1322708129882814e-06,
      "model_forward_time": 5.601725816726685,
      "step": 4333
    },
    {
      "epoch": 4.1322708129882814e-06,
      "step": 4333,
      "training_step_time": 17.073116302490234
    },
    {
      "epoch": 4.133224487304687e-06,
      "model_forward_time": 5.599490404129028,
      "step": 4334
    },
    {
      "epoch": 4.133224487304687e-06,
      "step": 4334,
      "training_step_time": 17.076197862625122
    },
    {
      "epoch": 4.134178161621094e-06,
      "model_forward_time": 5.597767353057861,
      "step": 4335
    },
    {
      "epoch": 4.134178161621094e-06,
      "step": 4335,
      "training_step_time": 17.04808521270752
    },
    {
      "epoch": 4.1351318359375e-06,
      "model_forward_time": 5.60252833366394,
      "step": 4336
    },
    {
      "epoch": 4.1351318359375e-06,
      "step": 4336,
      "training_step_time": 17.071569681167603
    },
    {
      "epoch": 4.136085510253906e-06,
      "model_forward_time": 5.599575042724609,
      "step": 4337
    },
    {
      "epoch": 4.136085510253906e-06,
      "step": 4337,
      "training_step_time": 17.07062864303589
    },
    {
      "epoch": 4.1370391845703126e-06,
      "model_forward_time": 5.96045184135437,
      "step": 4338
    },
    {
      "epoch": 4.1370391845703126e-06,
      "step": 4338,
      "training_step_time": 17.430203437805176
    },
    {
      "epoch": 4.137992858886719e-06,
      "model_forward_time": 5.597492694854736,
      "step": 4339
    },
    {
      "epoch": 4.137992858886719e-06,
      "step": 4339,
      "training_step_time": 17.047684907913208
    },
    {
      "epoch": 4.138946533203125e-06,
      "grad_norm": 1.7260547876358032,
      "learning_rate": 1e-05,
      "loss": 0.0666,
      "step": 4340
    },
    {
      "epoch": 4.138946533203125e-06,
      "model_forward_time": 5.594757080078125,
      "step": 4340
    },
    {
      "dynamics_loss_avg": 0.05333880223333835,
      "epoch": 4.138946533203125e-06,
      "step": 4340
    },
    {
      "action_loss_avg": 0.016844804515130818,
      "epoch": 4.138946533203125e-06,
      "step": 4340
    },
    {
      "epoch": 4.138946533203125e-06,
      "step": 4340,
      "training_step_time": 17.032931804656982
    },
    {
      "epoch": 4.1399002075195314e-06,
      "model_forward_time": 5.582707166671753,
      "step": 4341
    },
    {
      "epoch": 4.1399002075195314e-06,
      "step": 4341,
      "training_step_time": 17.033323049545288
    },
    {
      "epoch": 4.140853881835937e-06,
      "model_forward_time": 5.595439672470093,
      "step": 4342
    },
    {
      "epoch": 4.140853881835937e-06,
      "step": 4342,
      "training_step_time": 17.069682359695435
    },
    {
      "epoch": 4.141807556152344e-06,
      "model_forward_time": 5.607440233230591,
      "step": 4343
    },
    {
      "epoch": 4.141807556152344e-06,
      "step": 4343,
      "training_step_time": 17.453115701675415
    },
    {
      "epoch": 4.14276123046875e-06,
      "model_forward_time": 5.600319862365723,
      "step": 4344
    },
    {
      "epoch": 4.14276123046875e-06,
      "step": 4344,
      "training_step_time": 17.06835675239563
    },
    {
      "epoch": 4.143714904785156e-06,
      "model_forward_time": 5.596876621246338,
      "step": 4345
    },
    {
      "epoch": 4.143714904785156e-06,
      "step": 4345,
      "training_step_time": 17.05613613128662
    },
    {
      "epoch": 4.144668579101563e-06,
      "model_forward_time": 5.598651647567749,
      "step": 4346
    },
    {
      "epoch": 4.144668579101563e-06,
      "step": 4346,
      "training_step_time": 17.05952262878418
    },
    {
      "epoch": 4.145622253417969e-06,
      "model_forward_time": 5.587854862213135,
      "step": 4347
    },
    {
      "epoch": 4.145622253417969e-06,
      "step": 4347,
      "training_step_time": 17.033947944641113
    },
    {
      "epoch": 4.146575927734375e-06,
      "model_forward_time": 5.584935188293457,
      "step": 4348
    },
    {
      "epoch": 4.146575927734375e-06,
      "step": 4348,
      "training_step_time": 17.010512351989746
    },
    {
      "epoch": 4.1475296020507814e-06,
      "model_forward_time": 5.944509029388428,
      "step": 4349
    },
    {
      "epoch": 4.1475296020507814e-06,
      "step": 4349,
      "training_step_time": 17.357744455337524
    },
    {
      "epoch": 4.148483276367187e-06,
      "grad_norm": 2.0344762802124023,
      "learning_rate": 1e-05,
      "loss": 0.0948,
      "step": 4350
    },
    {
      "epoch": 4.148483276367187e-06,
      "model_forward_time": 5.5851335525512695,
      "step": 4350
    },
    {
      "dynamics_loss_avg": 0.07025025002658367,
      "epoch": 4.148483276367187e-06,
      "step": 4350
    },
    {
      "action_loss_avg": 0.02764587290585041,
      "epoch": 4.148483276367187e-06,
      "step": 4350
    },
    {
      "epoch": 4.148483276367187e-06,
      "step": 4350,
      "training_step_time": 17.03925347328186
    },
    {
      "epoch": 4.149436950683594e-06,
      "model_forward_time": 5.603858709335327,
      "step": 4351
    },
    {
      "epoch": 4.149436950683594e-06,
      "step": 4351,
      "training_step_time": 17.074565172195435
    },
    {
      "epoch": 4.150390625e-06,
      "model_forward_time": 5.606469631195068,
      "step": 4352
    },
    {
      "epoch": 4.150390625e-06,
      "step": 4352,
      "training_step_time": 17.07901930809021
    },
    {
      "epoch": 4.151344299316406e-06,
      "model_forward_time": 5.611496448516846,
      "step": 4353
    },
    {
      "epoch": 4.151344299316406e-06,
      "step": 4353,
      "training_step_time": 17.095734119415283
    },
    {
      "epoch": 4.152297973632813e-06,
      "model_forward_time": 5.943004846572876,
      "step": 4354
    },
    {
      "epoch": 4.152297973632813e-06,
      "step": 4354,
      "training_step_time": 17.4010648727417
    },
    {
      "epoch": 4.153251647949219e-06,
      "model_forward_time": 5.598161220550537,
      "step": 4355
    },
    {
      "epoch": 4.153251647949219e-06,
      "step": 4355,
      "training_step_time": 17.066462516784668
    },
    {
      "epoch": 4.154205322265625e-06,
      "model_forward_time": 5.6000816822052,
      "step": 4356
    },
    {
      "epoch": 4.154205322265625e-06,
      "step": 4356,
      "training_step_time": 17.065518856048584
    },
    {
      "epoch": 4.1551589965820314e-06,
      "model_forward_time": 5.6049816608428955,
      "step": 4357
    },
    {
      "epoch": 4.1551589965820314e-06,
      "step": 4357,
      "training_step_time": 17.06293535232544
    },
    {
      "epoch": 4.156112670898437e-06,
      "model_forward_time": 5.589748382568359,
      "step": 4358
    },
    {
      "epoch": 4.156112670898437e-06,
      "step": 4358,
      "training_step_time": 17.03729248046875
    },
    {
      "epoch": 4.157066345214844e-06,
      "model_forward_time": 5.591116428375244,
      "step": 4359
    },
    {
      "epoch": 4.157066345214844e-06,
      "step": 4359,
      "training_step_time": 17.043734312057495
    },
    {
      "epoch": 4.15802001953125e-06,
      "grad_norm": 0.4946834444999695,
      "learning_rate": 1e-05,
      "loss": 0.0806,
      "step": 4360
    },
    {
      "epoch": 4.15802001953125e-06,
      "model_forward_time": 5.968374729156494,
      "step": 4360
    },
    {
      "dynamics_loss_avg": 0.06314036529511213,
      "epoch": 4.15802001953125e-06,
      "step": 4360
    },
    {
      "action_loss_avg": 0.018761957553215323,
      "epoch": 4.15802001953125e-06,
      "step": 4360
    },
    {
      "epoch": 4.15802001953125e-06,
      "step": 4360,
      "training_step_time": 17.412566900253296
    },
    {
      "epoch": 4.158973693847656e-06,
      "model_forward_time": 5.592623710632324,
      "step": 4361
    },
    {
      "epoch": 4.158973693847656e-06,
      "step": 4361,
      "training_step_time": 17.058332920074463
    },
    {
      "epoch": 4.159927368164063e-06,
      "model_forward_time": 5.614112377166748,
      "step": 4362
    },
    {
      "epoch": 4.159927368164063e-06,
      "step": 4362,
      "training_step_time": 17.08773899078369
    },
    {
      "epoch": 4.160881042480469e-06,
      "model_forward_time": 5.6112658977508545,
      "step": 4363
    },
    {
      "epoch": 4.160881042480469e-06,
      "step": 4363,
      "training_step_time": 17.101221084594727
    },
    {
      "epoch": 4.161834716796875e-06,
      "model_forward_time": 5.609578371047974,
      "step": 4364
    },
    {
      "epoch": 4.161834716796875e-06,
      "step": 4364,
      "training_step_time": 17.107990741729736
    },
    {
      "epoch": 4.1627883911132815e-06,
      "model_forward_time": 5.616714239120483,
      "step": 4365
    },
    {
      "epoch": 4.1627883911132815e-06,
      "step": 4365,
      "training_step_time": 17.468855381011963
    },
    {
      "epoch": 4.163742065429687e-06,
      "model_forward_time": 5.624009847640991,
      "step": 4366
    },
    {
      "epoch": 4.163742065429687e-06,
      "step": 4366,
      "training_step_time": 17.126649379730225
    },
    {
      "epoch": 4.164695739746094e-06,
      "model_forward_time": 5.609996318817139,
      "step": 4367
    },
    {
      "epoch": 4.164695739746094e-06,
      "step": 4367,
      "training_step_time": 17.036566972732544
    },
    {
      "epoch": 4.1656494140625e-06,
      "model_forward_time": 5.605163097381592,
      "step": 4368
    },
    {
      "epoch": 4.1656494140625e-06,
      "step": 4368,
      "training_step_time": 17.080795764923096
    },
    {
      "epoch": 4.166603088378906e-06,
      "model_forward_time": 5.609808444976807,
      "step": 4369
    },
    {
      "epoch": 4.166603088378906e-06,
      "step": 4369,
      "training_step_time": 17.080005407333374
    },
    {
      "epoch": 4.167556762695313e-06,
      "grad_norm": 1.0862529277801514,
      "learning_rate": 1e-05,
      "loss": 0.0756,
      "step": 4370
    },
    {
      "epoch": 4.167556762695313e-06,
      "model_forward_time": 5.59472131729126,
      "step": 4370
    },
    {
      "dynamics_loss_avg": 0.05168577320873737,
      "epoch": 4.167556762695313e-06,
      "step": 4370
    },
    {
      "action_loss_avg": 0.014924456574954092,
      "epoch": 4.167556762695313e-06,
      "step": 4370
    },
    {
      "epoch": 4.167556762695313e-06,
      "step": 4370,
      "training_step_time": 17.068178415298462
    },
    {
      "epoch": 4.168510437011719e-06,
      "model_forward_time": 5.970468521118164,
      "step": 4371
    },
    {
      "epoch": 4.168510437011719e-06,
      "step": 4371,
      "training_step_time": 17.465975046157837
    },
    {
      "epoch": 4.169464111328125e-06,
      "model_forward_time": 5.620058536529541,
      "step": 4372
    },
    {
      "epoch": 4.169464111328125e-06,
      "step": 4372,
      "training_step_time": 17.11052107810974
    },
    {
      "epoch": 4.1704177856445315e-06,
      "model_forward_time": 5.607335329055786,
      "step": 4373
    },
    {
      "epoch": 4.1704177856445315e-06,
      "step": 4373,
      "training_step_time": 17.078799724578857
    },
    {
      "epoch": 4.171371459960937e-06,
      "model_forward_time": 5.608611822128296,
      "step": 4374
    },
    {
      "epoch": 4.171371459960937e-06,
      "step": 4374,
      "training_step_time": 17.086667776107788
    },
    {
      "epoch": 4.172325134277344e-06,
      "model_forward_time": 5.603157043457031,
      "step": 4375
    },
    {
      "epoch": 4.172325134277344e-06,
      "step": 4375,
      "training_step_time": 17.092238664627075
    },
    {
      "epoch": 4.17327880859375e-06,
      "model_forward_time": 5.606986045837402,
      "step": 4376
    },
    {
      "epoch": 4.17327880859375e-06,
      "step": 4376,
      "training_step_time": 17.464625597000122
    },
    {
      "epoch": 4.174232482910156e-06,
      "model_forward_time": 5.611817121505737,
      "step": 4377
    },
    {
      "epoch": 4.174232482910156e-06,
      "step": 4377,
      "training_step_time": 17.102588176727295
    },
    {
      "epoch": 4.175186157226563e-06,
      "model_forward_time": 5.60261082649231,
      "step": 4378
    },
    {
      "epoch": 4.175186157226563e-06,
      "step": 4378,
      "training_step_time": 17.08786368370056
    },
    {
      "epoch": 4.176139831542968e-06,
      "model_forward_time": 5.617300271987915,
      "step": 4379
    },
    {
      "epoch": 4.176139831542968e-06,
      "step": 4379,
      "training_step_time": 17.104651927947998
    },
    {
      "epoch": 4.177093505859375e-06,
      "grad_norm": 1.944765567779541,
      "learning_rate": 1e-05,
      "loss": 0.0901,
      "step": 4380
    },
    {
      "epoch": 4.177093505859375e-06,
      "model_forward_time": 5.604128122329712,
      "step": 4380
    },
    {
      "dynamics_loss_avg": 0.06765219233930111,
      "epoch": 4.177093505859375e-06,
      "step": 4380
    },
    {
      "action_loss_avg": 0.030506178550422192,
      "epoch": 4.177093505859375e-06,
      "step": 4380
    },
    {
      "epoch": 4.177093505859375e-06,
      "step": 4380,
      "training_step_time": 17.05378532409668
    },
    {
      "epoch": 4.1780471801757815e-06,
      "model_forward_time": 5.598294973373413,
      "step": 4381
    },
    {
      "epoch": 4.1780471801757815e-06,
      "step": 4381,
      "training_step_time": 17.04392910003662
    },
    {
      "epoch": 4.179000854492187e-06,
      "model_forward_time": 5.96464467048645,
      "step": 4382
    },
    {
      "epoch": 4.179000854492187e-06,
      "step": 4382,
      "training_step_time": 17.38861656188965
    },
    {
      "epoch": 4.179954528808594e-06,
      "model_forward_time": 5.584468364715576,
      "step": 4383
    },
    {
      "epoch": 4.179954528808594e-06,
      "step": 4383,
      "training_step_time": 17.034575939178467
    },
    {
      "epoch": 4.180908203125e-06,
      "model_forward_time": 5.597249507904053,
      "step": 4384
    },
    {
      "epoch": 4.180908203125e-06,
      "step": 4384,
      "training_step_time": 17.02744174003601
    },
    {
      "epoch": 4.181861877441406e-06,
      "model_forward_time": 5.600666046142578,
      "step": 4385
    },
    {
      "epoch": 4.181861877441406e-06,
      "step": 4385,
      "training_step_time": 17.067430019378662
    },
    {
      "epoch": 4.182815551757813e-06,
      "model_forward_time": 5.596338748931885,
      "step": 4386
    },
    {
      "epoch": 4.182815551757813e-06,
      "step": 4386,
      "training_step_time": 17.058006763458252
    },
    {
      "epoch": 4.183769226074218e-06,
      "model_forward_time": 5.960951566696167,
      "step": 4387
    },
    {
      "epoch": 4.183769226074218e-06,
      "step": 4387,
      "training_step_time": 17.40808415412903
    },
    {
      "epoch": 4.184722900390625e-06,
      "model_forward_time": 5.5918591022491455,
      "step": 4388
    },
    {
      "epoch": 4.184722900390625e-06,
      "step": 4388,
      "training_step_time": 17.034817695617676
    },
    {
      "epoch": 4.1856765747070315e-06,
      "model_forward_time": 5.59394383430481,
      "step": 4389
    },
    {
      "epoch": 4.1856765747070315e-06,
      "step": 4389,
      "training_step_time": 17.052207946777344
    },
    {
      "epoch": 4.186630249023437e-06,
      "grad_norm": 1.44413423538208,
      "learning_rate": 1e-05,
      "loss": 0.0709,
      "step": 4390
    },
    {
      "epoch": 4.186630249023437e-06,
      "model_forward_time": 5.604472875595093,
      "step": 4390
    },
    {
      "dynamics_loss_avg": 0.049290372617542745,
      "epoch": 4.186630249023437e-06,
      "step": 4390
    },
    {
      "action_loss_avg": 0.014455654844641685,
      "epoch": 4.186630249023437e-06,
      "step": 4390
    },
    {
      "epoch": 4.186630249023437e-06,
      "step": 4390,
      "training_step_time": 17.09196424484253
    },
    {
      "epoch": 4.187583923339844e-06,
      "model_forward_time": 5.62567663192749,
      "step": 4391
    },
    {
      "epoch": 4.187583923339844e-06,
      "step": 4391,
      "training_step_time": 17.13092041015625
    },
    {
      "epoch": 4.18853759765625e-06,
      "model_forward_time": 5.642274379730225,
      "step": 4392
    },
    {
      "epoch": 4.18853759765625e-06,
      "step": 4392,
      "training_step_time": 17.138640642166138
    },
    {
      "epoch": 4.189491271972656e-06,
      "model_forward_time": 5.975800037384033,
      "step": 4393
    },
    {
      "epoch": 4.189491271972656e-06,
      "step": 4393,
      "training_step_time": 17.451677322387695
    },
    {
      "epoch": 4.190444946289063e-06,
      "model_forward_time": 5.616994857788086,
      "step": 4394
    },
    {
      "epoch": 4.190444946289063e-06,
      "step": 4394,
      "training_step_time": 17.06163763999939
    },
    {
      "epoch": 4.191398620605468e-06,
      "model_forward_time": 5.5836021900177,
      "step": 4395
    },
    {
      "epoch": 4.191398620605468e-06,
      "step": 4395,
      "training_step_time": 17.008957624435425
    },
    {
      "epoch": 4.192352294921875e-06,
      "model_forward_time": 5.576982021331787,
      "step": 4396
    },
    {
      "epoch": 4.192352294921875e-06,
      "step": 4396,
      "training_step_time": 17.019211530685425
    },
    {
      "epoch": 4.1933059692382815e-06,
      "model_forward_time": 5.593029975891113,
      "step": 4397
    },
    {
      "epoch": 4.1933059692382815e-06,
      "step": 4397,
      "training_step_time": 17.062901258468628
    },
    {
      "epoch": 4.194259643554687e-06,
      "model_forward_time": 5.611637592315674,
      "step": 4398
    },
    {
      "epoch": 4.194259643554687e-06,
      "step": 4398,
      "training_step_time": 17.468467712402344
    },
    {
      "epoch": 4.195213317871094e-06,
      "model_forward_time": 5.605238437652588,
      "step": 4399
    },
    {
      "epoch": 4.195213317871094e-06,
      "step": 4399,
      "training_step_time": 17.093195915222168
    },
    {
      "epoch": 4.1961669921875e-06,
      "grad_norm": 0.7419040203094482,
      "learning_rate": 1e-05,
      "loss": 0.0769,
      "step": 4400
    },
    {
      "epoch": 4.1961669921875e-06,
      "model_forward_time": 5.606442451477051,
      "step": 4400
    },
    {
      "dynamics_loss_avg": 0.06298020258545875,
      "epoch": 4.1961669921875e-06,
      "step": 4400
    },
    {
      "action_loss_avg": 0.016627045720815657,
      "epoch": 4.1961669921875e-06,
      "step": 4400
    },
    {
      "epoch": 4.1961669921875e-06,
      "step": 4400,
      "training_step_time": 17.09302544593811
    },
    {
      "epoch": 4.197120666503906e-06,
      "model_forward_time": 5.610357284545898,
      "step": 4401
    },
    {
      "epoch": 4.197120666503906e-06,
      "step": 4401,
      "training_step_time": 17.07165837287903
    },
    {
      "epoch": 4.198074340820313e-06,
      "model_forward_time": 5.598852634429932,
      "step": 4402
    },
    {
      "epoch": 4.198074340820313e-06,
      "step": 4402,
      "training_step_time": 17.046456336975098
    },
    {
      "epoch": 4.199028015136718e-06,
      "model_forward_time": 5.596654415130615,
      "step": 4403
    },
    {
      "epoch": 4.199028015136718e-06,
      "step": 4403,
      "training_step_time": 17.035370349884033
    },
    {
      "epoch": 4.199981689453125e-06,
      "model_forward_time": 5.943432092666626,
      "step": 4404
    },
    {
      "epoch": 4.199981689453125e-06,
      "step": 4404,
      "training_step_time": 17.37638783454895
    },
    {
      "epoch": 4.2009353637695315e-06,
      "model_forward_time": 5.5810699462890625,
      "step": 4405
    },
    {
      "epoch": 4.2009353637695315e-06,
      "step": 4405,
      "training_step_time": 17.021440505981445
    },
    {
      "epoch": 4.201889038085937e-06,
      "model_forward_time": 5.6085288524627686,
      "step": 4406
    },
    {
      "epoch": 4.201889038085937e-06,
      "step": 4406,
      "training_step_time": 17.08958625793457
    },
    {
      "epoch": 4.202842712402344e-06,
      "model_forward_time": 5.608598709106445,
      "step": 4407
    },
    {
      "epoch": 4.202842712402344e-06,
      "step": 4407,
      "training_step_time": 17.078300952911377
    },
    {
      "epoch": 4.20379638671875e-06,
      "model_forward_time": 5.5893614292144775,
      "step": 4408
    },
    {
      "epoch": 4.20379638671875e-06,
      "step": 4408,
      "training_step_time": 17.05374550819397
    },
    {
      "epoch": 4.204750061035156e-06,
      "model_forward_time": 5.595711946487427,
      "step": 4409
    },
    {
      "epoch": 4.204750061035156e-06,
      "step": 4409,
      "training_step_time": 17.40515398979187
    },
    {
      "epoch": 4.205703735351563e-06,
      "grad_norm": 1.1086393594741821,
      "learning_rate": 1e-05,
      "loss": 0.068,
      "step": 4410
    },
    {
      "epoch": 4.205703735351563e-06,
      "model_forward_time": 5.576246023178101,
      "step": 4410
    },
    {
      "dynamics_loss_avg": 0.055788596533238886,
      "epoch": 4.205703735351563e-06,
      "step": 4410
    },
    {
      "action_loss_avg": 0.017437399132177234,
      "epoch": 4.205703735351563e-06,
      "step": 4410
    },
    {
      "epoch": 4.205703735351563e-06,
      "step": 4410,
      "training_step_time": 16.986128330230713
    },
    {
      "epoch": 4.206657409667968e-06,
      "model_forward_time": 5.593809127807617,
      "step": 4411
    },
    {
      "epoch": 4.206657409667968e-06,
      "step": 4411,
      "training_step_time": 17.040337324142456
    },
    {
      "epoch": 4.207611083984375e-06,
      "model_forward_time": 5.604377031326294,
      "step": 4412
    },
    {
      "epoch": 4.207611083984375e-06,
      "step": 4412,
      "training_step_time": 17.10563826560974
    },
    {
      "epoch": 4.2085647583007815e-06,
      "model_forward_time": 5.62052059173584,
      "step": 4413
    },
    {
      "epoch": 4.2085647583007815e-06,
      "step": 4413,
      "training_step_time": 17.106024980545044
    },
    {
      "epoch": 4.209518432617187e-06,
      "model_forward_time": 5.595430850982666,
      "step": 4414
    },
    {
      "epoch": 4.209518432617187e-06,
      "step": 4414,
      "training_step_time": 17.03251314163208
    },
    {
      "epoch": 4.210472106933594e-06,
      "model_forward_time": 5.946146488189697,
      "step": 4415
    },
    {
      "epoch": 4.210472106933594e-06,
      "step": 4415,
      "training_step_time": 17.397316932678223
    },
    {
      "epoch": 4.21142578125e-06,
      "model_forward_time": 5.586632251739502,
      "step": 4416
    },
    {
      "epoch": 4.21142578125e-06,
      "step": 4416,
      "training_step_time": 16.99363112449646
    },
    {
      "epoch": 4.212379455566406e-06,
      "model_forward_time": 5.586647272109985,
      "step": 4417
    },
    {
      "epoch": 4.212379455566406e-06,
      "step": 4417,
      "training_step_time": 17.04581332206726
    },
    {
      "epoch": 4.213333129882813e-06,
      "model_forward_time": 5.598718881607056,
      "step": 4418
    },
    {
      "epoch": 4.213333129882813e-06,
      "step": 4418,
      "training_step_time": 17.059324264526367
    },
    {
      "epoch": 4.214286804199218e-06,
      "model_forward_time": 5.600674152374268,
      "step": 4419
    },
    {
      "epoch": 4.214286804199218e-06,
      "step": 4419,
      "training_step_time": 17.052767753601074
    },
    {
      "epoch": 4.215240478515625e-06,
      "grad_norm": 2.331829786300659,
      "learning_rate": 1e-05,
      "loss": 0.1141,
      "step": 4420
    },
    {
      "epoch": 4.215240478515625e-06,
      "model_forward_time": 5.946481943130493,
      "step": 4420
    },
    {
      "dynamics_loss_avg": 0.07331138998270034,
      "epoch": 4.215240478515625e-06,
      "step": 4420
    },
    {
      "action_loss_avg": 0.03317441013641655,
      "epoch": 4.215240478515625e-06,
      "step": 4420
    },
    {
      "epoch": 4.215240478515625e-06,
      "step": 4420,
      "training_step_time": 17.412730932235718
    },
    {
      "epoch": 4.2161941528320315e-06,
      "model_forward_time": 5.602747678756714,
      "step": 4421
    },
    {
      "epoch": 4.2161941528320315e-06,
      "step": 4421,
      "training_step_time": 17.071915864944458
    },
    {
      "epoch": 4.217147827148437e-06,
      "model_forward_time": 5.600586891174316,
      "step": 4422
    },
    {
      "epoch": 4.217147827148437e-06,
      "step": 4422,
      "training_step_time": 17.07423496246338
    },
    {
      "epoch": 4.218101501464844e-06,
      "model_forward_time": 5.613211154937744,
      "step": 4423
    },
    {
      "epoch": 4.218101501464844e-06,
      "step": 4423,
      "training_step_time": 17.089146375656128
    },
    {
      "epoch": 4.21905517578125e-06,
      "model_forward_time": 5.605550050735474,
      "step": 4424
    },
    {
      "epoch": 4.21905517578125e-06,
      "step": 4424,
      "training_step_time": 17.07976460456848
    },
    {
      "epoch": 4.220008850097656e-06,
      "model_forward_time": 5.596745252609253,
      "step": 4425
    },
    {
      "epoch": 4.220008850097656e-06,
      "step": 4425,
      "training_step_time": 17.07732367515564
    },
    {
      "epoch": 4.220962524414063e-06,
      "model_forward_time": 4.596179485321045,
      "step": 4426
    },
    {
      "epoch": 4.220962524414063e-06,
      "step": 4426,
      "training_step_time": 13.113150119781494
    },
    {
      "epoch": 4.221916198730468e-06,
      "model_forward_time": 5.589754819869995,
      "step": 4427
    },
    {
      "epoch": 4.221916198730468e-06,
      "step": 4427,
      "training_step_time": 17.06175661087036
    },
    {
      "epoch": 4.222869873046875e-06,
      "model_forward_time": 5.5999860763549805,
      "step": 4428
    },
    {
      "epoch": 4.222869873046875e-06,
      "step": 4428,
      "training_step_time": 17.084578037261963
    },
    {
      "epoch": 4.2238235473632815e-06,
      "model_forward_time": 5.607801198959351,
      "step": 4429
    },
    {
      "epoch": 4.2238235473632815e-06,
      "step": 4429,
      "training_step_time": 17.08897113800049
    },
    {
      "epoch": 4.224777221679687e-06,
      "grad_norm": 3.0656206607818604,
      "learning_rate": 1e-05,
      "loss": 0.0784,
      "step": 4430
    },
    {
      "epoch": 4.224777221679687e-06,
      "model_forward_time": 5.599733591079712,
      "step": 4430
    },
    {
      "dynamics_loss_avg": 0.0631888872012496,
      "epoch": 4.224777221679687e-06,
      "step": 4430
    },
    {
      "action_loss_avg": 0.02230292079038918,
      "epoch": 4.224777221679687e-06,
      "step": 4430
    },
    {
      "epoch": 4.224777221679687e-06,
      "step": 4430,
      "training_step_time": 17.099448204040527
    },
    {
      "epoch": 4.225730895996094e-06,
      "model_forward_time": 5.616146564483643,
      "step": 4431
    },
    {
      "epoch": 4.225730895996094e-06,
      "step": 4431,
      "training_step_time": 17.465553045272827
    },
    {
      "epoch": 4.2266845703125e-06,
      "model_forward_time": 5.617002725601196,
      "step": 4432
    },
    {
      "epoch": 4.2266845703125e-06,
      "step": 4432,
      "training_step_time": 17.099422454833984
    },
    {
      "epoch": 4.227638244628906e-06,
      "model_forward_time": 5.598554849624634,
      "step": 4433
    },
    {
      "epoch": 4.227638244628906e-06,
      "step": 4433,
      "training_step_time": 17.038230180740356
    },
    {
      "epoch": 4.228591918945313e-06,
      "model_forward_time": 5.574598550796509,
      "step": 4434
    },
    {
      "epoch": 4.228591918945313e-06,
      "step": 4434,
      "training_step_time": 16.99928331375122
    },
    {
      "epoch": 4.229545593261718e-06,
      "model_forward_time": 5.572109699249268,
      "step": 4435
    },
    {
      "epoch": 4.229545593261718e-06,
      "step": 4435,
      "training_step_time": 16.983360052108765
    },
    {
      "epoch": 4.230499267578125e-06,
      "model_forward_time": 5.585648536682129,
      "step": 4436
    },
    {
      "epoch": 4.230499267578125e-06,
      "step": 4436,
      "training_step_time": 16.99859118461609
    },
    {
      "epoch": 4.2314529418945315e-06,
      "model_forward_time": 5.956003189086914,
      "step": 4437
    },
    {
      "epoch": 4.2314529418945315e-06,
      "step": 4437,
      "training_step_time": 17.425022840499878
    },
    {
      "epoch": 4.232406616210937e-06,
      "model_forward_time": 5.60686731338501,
      "step": 4438
    },
    {
      "epoch": 4.232406616210937e-06,
      "step": 4438,
      "training_step_time": 17.062258005142212
    },
    {
      "epoch": 4.233360290527344e-06,
      "model_forward_time": 5.60789942741394,
      "step": 4439
    },
    {
      "epoch": 4.233360290527344e-06,
      "step": 4439,
      "training_step_time": 17.099305629730225
    },
    {
      "epoch": 4.23431396484375e-06,
      "grad_norm": 1.5995216369628906,
      "learning_rate": 1e-05,
      "loss": 0.0788,
      "step": 4440
    },
    {
      "epoch": 4.23431396484375e-06,
      "model_forward_time": 5.615737199783325,
      "step": 4440
    },
    {
      "dynamics_loss_avg": 0.055868376046419144,
      "epoch": 4.23431396484375e-06,
      "step": 4440
    },
    {
      "action_loss_avg": 0.01487069963477552,
      "epoch": 4.23431396484375e-06,
      "step": 4440
    },
    {
      "epoch": 4.23431396484375e-06,
      "step": 4440,
      "training_step_time": 17.112669944763184
    },
    {
      "epoch": 4.235267639160156e-06,
      "model_forward_time": 5.607069492340088,
      "step": 4441
    },
    {
      "epoch": 4.235267639160156e-06,
      "step": 4441,
      "training_step_time": 17.08548355102539
    },
    {
      "epoch": 4.236221313476563e-06,
      "model_forward_time": 5.962910890579224,
      "step": 4442
    },
    {
      "epoch": 4.236221313476563e-06,
      "step": 4442,
      "training_step_time": 17.43795657157898
    },
    {
      "epoch": 4.237174987792968e-06,
      "model_forward_time": 5.59920072555542,
      "step": 4443
    },
    {
      "epoch": 4.237174987792968e-06,
      "step": 4443,
      "training_step_time": 17.07202935218811
    },
    {
      "epoch": 4.238128662109375e-06,
      "model_forward_time": 5.595098495483398,
      "step": 4444
    },
    {
      "epoch": 4.238128662109375e-06,
      "step": 4444,
      "training_step_time": 17.047252655029297
    },
    {
      "epoch": 4.2390823364257815e-06,
      "model_forward_time": 5.596653938293457,
      "step": 4445
    },
    {
      "epoch": 4.2390823364257815e-06,
      "step": 4445,
      "training_step_time": 17.046218872070312
    },
    {
      "epoch": 4.240036010742187e-06,
      "model_forward_time": 5.6129772663116455,
      "step": 4446
    },
    {
      "epoch": 4.240036010742187e-06,
      "step": 4446,
      "training_step_time": 17.085936784744263
    },
    {
      "epoch": 4.240989685058594e-06,
      "model_forward_time": 5.620591402053833,
      "step": 4447
    },
    {
      "epoch": 4.240989685058594e-06,
      "step": 4447,
      "training_step_time": 17.124991416931152
    },
    {
      "epoch": 4.241943359375e-06,
      "model_forward_time": 6.035610675811768,
      "step": 4448
    },
    {
      "epoch": 4.241943359375e-06,
      "step": 4448,
      "training_step_time": 17.551015377044678
    },
    {
      "epoch": 4.242897033691406e-06,
      "model_forward_time": 5.618501901626587,
      "step": 4449
    },
    {
      "epoch": 4.242897033691406e-06,
      "step": 4449,
      "training_step_time": 17.077573537826538
    },
    {
      "epoch": 4.243850708007813e-06,
      "grad_norm": 0.36620017886161804,
      "learning_rate": 1e-05,
      "loss": 0.0699,
      "step": 4450
    },
    {
      "epoch": 4.243850708007813e-06,
      "model_forward_time": 5.621630907058716,
      "step": 4450
    },
    {
      "dynamics_loss_avg": 0.05582945458590984,
      "epoch": 4.243850708007813e-06,
      "step": 4450
    },
    {
      "action_loss_avg": 0.018309444189071655,
      "epoch": 4.243850708007813e-06,
      "step": 4450
    },
    {
      "epoch": 4.243850708007813e-06,
      "step": 4450,
      "training_step_time": 17.079976558685303
    },
    {
      "epoch": 4.244804382324218e-06,
      "model_forward_time": 5.58414363861084,
      "step": 4451
    },
    {
      "epoch": 4.244804382324218e-06,
      "step": 4451,
      "training_step_time": 17.004607439041138
    },
    {
      "epoch": 4.245758056640625e-06,
      "model_forward_time": 5.581649541854858,
      "step": 4452
    },
    {
      "epoch": 4.245758056640625e-06,
      "step": 4452,
      "training_step_time": 17.02448081970215
    },
    {
      "epoch": 4.2467117309570315e-06,
      "model_forward_time": 5.938020467758179,
      "step": 4453
    },
    {
      "epoch": 4.2467117309570315e-06,
      "step": 4453,
      "training_step_time": 17.401899099349976
    },
    {
      "epoch": 4.247665405273437e-06,
      "model_forward_time": 5.6010730266571045,
      "step": 4454
    },
    {
      "epoch": 4.247665405273437e-06,
      "step": 4454,
      "training_step_time": 17.066966772079468
    },
    {
      "epoch": 4.248619079589844e-06,
      "model_forward_time": 5.599537372589111,
      "step": 4455
    },
    {
      "epoch": 4.248619079589844e-06,
      "step": 4455,
      "training_step_time": 17.0668728351593
    },
    {
      "epoch": 4.24957275390625e-06,
      "model_forward_time": 5.598348140716553,
      "step": 4456
    },
    {
      "epoch": 4.24957275390625e-06,
      "step": 4456,
      "training_step_time": 17.055243730545044
    },
    {
      "epoch": 4.250526428222656e-06,
      "model_forward_time": 5.603512763977051,
      "step": 4457
    },
    {
      "epoch": 4.250526428222656e-06,
      "step": 4457,
      "training_step_time": 17.05604076385498
    },
    {
      "epoch": 4.251480102539063e-06,
      "model_forward_time": 5.623732566833496,
      "step": 4458
    },
    {
      "epoch": 4.251480102539063e-06,
      "step": 4458,
      "training_step_time": 17.462360858917236
    },
    {
      "epoch": 4.252433776855468e-06,
      "model_forward_time": 5.599390506744385,
      "step": 4459
    },
    {
      "epoch": 4.252433776855468e-06,
      "step": 4459,
      "training_step_time": 17.051273822784424
    },
    {
      "epoch": 4.253387451171875e-06,
      "grad_norm": 1.3019514083862305,
      "learning_rate": 1e-05,
      "loss": 0.0885,
      "step": 4460
    },
    {
      "epoch": 4.253387451171875e-06,
      "model_forward_time": 5.58979606628418,
      "step": 4460
    },
    {
      "dynamics_loss_avg": 0.06746187135577202,
      "epoch": 4.253387451171875e-06,
      "step": 4460
    },
    {
      "action_loss_avg": 0.024668281711637974,
      "epoch": 4.253387451171875e-06,
      "step": 4460
    },
    {
      "epoch": 4.253387451171875e-06,
      "step": 4460,
      "training_step_time": 17.012574195861816
    },
    {
      "epoch": 4.2543411254882815e-06,
      "model_forward_time": 5.584986925125122,
      "step": 4461
    },
    {
      "epoch": 4.2543411254882815e-06,
      "step": 4461,
      "training_step_time": 17.034062385559082
    },
    {
      "epoch": 4.255294799804687e-06,
      "model_forward_time": 5.587772846221924,
      "step": 4462
    },
    {
      "epoch": 4.255294799804687e-06,
      "step": 4462,
      "training_step_time": 17.041611433029175
    },
    {
      "epoch": 4.256248474121094e-06,
      "model_forward_time": 5.597729444503784,
      "step": 4463
    },
    {
      "epoch": 4.256248474121094e-06,
      "step": 4463,
      "training_step_time": 17.052558422088623
    },
    {
      "epoch": 4.2572021484375e-06,
      "model_forward_time": 5.935357093811035,
      "step": 4464
    },
    {
      "epoch": 4.2572021484375e-06,
      "step": 4464,
      "training_step_time": 17.36528754234314
    },
    {
      "epoch": 4.258155822753906e-06,
      "model_forward_time": 5.596933126449585,
      "step": 4465
    },
    {
      "epoch": 4.258155822753906e-06,
      "step": 4465,
      "training_step_time": 17.059168577194214
    },
    {
      "epoch": 4.259109497070313e-06,
      "model_forward_time": 5.599982976913452,
      "step": 4466
    },
    {
      "epoch": 4.259109497070313e-06,
      "step": 4466,
      "training_step_time": 17.061734914779663
    },
    {
      "epoch": 4.260063171386718e-06,
      "model_forward_time": 5.592586040496826,
      "step": 4467
    },
    {
      "epoch": 4.260063171386718e-06,
      "step": 4467,
      "training_step_time": 17.043500661849976
    },
    {
      "epoch": 4.261016845703125e-06,
      "model_forward_time": 5.59614109992981,
      "step": 4468
    },
    {
      "epoch": 4.261016845703125e-06,
      "step": 4468,
      "training_step_time": 17.036094188690186
    },
    {
      "epoch": 4.2619705200195315e-06,
      "model_forward_time": 5.597304105758667,
      "step": 4469
    },
    {
      "epoch": 4.2619705200195315e-06,
      "step": 4469,
      "training_step_time": 17.417884349822998
    },
    {
      "epoch": 4.262924194335937e-06,
      "grad_norm": 1.8864514827728271,
      "learning_rate": 1e-05,
      "loss": 0.0842,
      "step": 4470
    },
    {
      "epoch": 4.262924194335937e-06,
      "model_forward_time": 5.605836868286133,
      "step": 4470
    },
    {
      "dynamics_loss_avg": 0.0602811312302947,
      "epoch": 4.262924194335937e-06,
      "step": 4470
    },
    {
      "action_loss_avg": 0.021028408920392393,
      "epoch": 4.262924194335937e-06,
      "step": 4470
    },
    {
      "epoch": 4.262924194335937e-06,
      "step": 4470,
      "training_step_time": 17.075710773468018
    },
    {
      "epoch": 4.263877868652344e-06,
      "model_forward_time": 5.603148460388184,
      "step": 4471
    },
    {
      "epoch": 4.263877868652344e-06,
      "step": 4471,
      "training_step_time": 17.067803621292114
    },
    {
      "epoch": 4.26483154296875e-06,
      "model_forward_time": 5.6044840812683105,
      "step": 4472
    },
    {
      "epoch": 4.26483154296875e-06,
      "step": 4472,
      "training_step_time": 17.067349672317505
    },
    {
      "epoch": 4.265785217285156e-06,
      "model_forward_time": 5.601405620574951,
      "step": 4473
    },
    {
      "epoch": 4.265785217285156e-06,
      "step": 4473,
      "training_step_time": 17.08240818977356
    },
    {
      "epoch": 4.266738891601563e-06,
      "model_forward_time": 5.599849462509155,
      "step": 4474
    },
    {
      "epoch": 4.266738891601563e-06,
      "step": 4474,
      "training_step_time": 17.092763662338257
    },
    {
      "epoch": 4.267692565917968e-06,
      "model_forward_time": 5.975963354110718,
      "step": 4475
    },
    {
      "epoch": 4.267692565917968e-06,
      "step": 4475,
      "training_step_time": 17.472806692123413
    },
    {
      "epoch": 4.268646240234375e-06,
      "model_forward_time": 5.6047892570495605,
      "step": 4476
    },
    {
      "epoch": 4.268646240234375e-06,
      "step": 4476,
      "training_step_time": 17.064801454544067
    },
    {
      "epoch": 4.2695999145507815e-06,
      "model_forward_time": 5.599574565887451,
      "step": 4477
    },
    {
      "epoch": 4.2695999145507815e-06,
      "step": 4477,
      "training_step_time": 17.06643271446228
    },
    {
      "epoch": 4.270553588867187e-06,
      "model_forward_time": 5.593251943588257,
      "step": 4478
    },
    {
      "epoch": 4.270553588867187e-06,
      "step": 4478,
      "training_step_time": 17.036396503448486
    },
    {
      "epoch": 4.271507263183594e-06,
      "model_forward_time": 5.592015981674194,
      "step": 4479
    },
    {
      "epoch": 4.271507263183594e-06,
      "step": 4479,
      "training_step_time": 17.044594764709473
    },
    {
      "epoch": 4.2724609375e-06,
      "grad_norm": 0.49450942873954773,
      "learning_rate": 1e-05,
      "loss": 0.0633,
      "step": 4480
    },
    {
      "epoch": 4.2724609375e-06,
      "model_forward_time": 5.968076467514038,
      "step": 4480
    },
    {
      "dynamics_loss_avg": 0.043912660144269464,
      "epoch": 4.2724609375e-06,
      "step": 4480
    },
    {
      "action_loss_avg": 0.016508842771872877,
      "epoch": 4.2724609375e-06,
      "step": 4480
    },
    {
      "epoch": 4.2724609375e-06,
      "step": 4480,
      "training_step_time": 17.452152490615845
    },
    {
      "epoch": 4.273414611816406e-06,
      "model_forward_time": 5.605879545211792,
      "step": 4481
    },
    {
      "epoch": 4.273414611816406e-06,
      "step": 4481,
      "training_step_time": 17.073859930038452
    },
    {
      "epoch": 4.274368286132813e-06,
      "model_forward_time": 5.588316202163696,
      "step": 4482
    },
    {
      "epoch": 4.274368286132813e-06,
      "step": 4482,
      "training_step_time": 16.99712300300598
    },
    {
      "epoch": 4.275321960449218e-06,
      "model_forward_time": 5.584127426147461,
      "step": 4483
    },
    {
      "epoch": 4.275321960449218e-06,
      "step": 4483,
      "training_step_time": 17.015538454055786
    },
    {
      "epoch": 4.276275634765625e-06,
      "model_forward_time": 5.585184812545776,
      "step": 4484
    },
    {
      "epoch": 4.276275634765625e-06,
      "step": 4484,
      "training_step_time": 17.02207088470459
    },
    {
      "epoch": 4.2772293090820316e-06,
      "model_forward_time": 5.601141452789307,
      "step": 4485
    },
    {
      "epoch": 4.2772293090820316e-06,
      "step": 4485,
      "training_step_time": 17.44225835800171
    },
    {
      "epoch": 4.278182983398437e-06,
      "model_forward_time": 5.605695009231567,
      "step": 4486
    },
    {
      "epoch": 4.278182983398437e-06,
      "step": 4486,
      "training_step_time": 17.101705074310303
    },
    {
      "epoch": 4.279136657714844e-06,
      "model_forward_time": 5.61503267288208,
      "step": 4487
    },
    {
      "epoch": 4.279136657714844e-06,
      "step": 4487,
      "training_step_time": 17.1077823638916
    },
    {
      "epoch": 4.28009033203125e-06,
      "model_forward_time": 5.6229941844940186,
      "step": 4488
    },
    {
      "epoch": 4.28009033203125e-06,
      "step": 4488,
      "training_step_time": 17.106027126312256
    },
    {
      "epoch": 4.281044006347656e-06,
      "model_forward_time": 5.594897985458374,
      "step": 4489
    },
    {
      "epoch": 4.281044006347656e-06,
      "step": 4489,
      "training_step_time": 17.038509845733643
    },
    {
      "epoch": 4.281997680664063e-06,
      "grad_norm": 1.722452998161316,
      "learning_rate": 1e-05,
      "loss": 0.0898,
      "step": 4490
    },
    {
      "epoch": 4.281997680664063e-06,
      "model_forward_time": 5.587023019790649,
      "step": 4490
    },
    {
      "dynamics_loss_avg": 0.06934582628309727,
      "epoch": 4.281997680664063e-06,
      "step": 4490
    },
    {
      "action_loss_avg": 0.02181004509329796,
      "epoch": 4.281997680664063e-06,
      "step": 4490
    },
    {
      "epoch": 4.281997680664063e-06,
      "step": 4490,
      "training_step_time": 17.020231008529663
    },
    {
      "epoch": 4.2829513549804684e-06,
      "model_forward_time": 5.935773134231567,
      "step": 4491
    },
    {
      "epoch": 4.2829513549804684e-06,
      "step": 4491,
      "training_step_time": 17.372968912124634
    },
    {
      "epoch": 4.283905029296875e-06,
      "model_forward_time": 5.582936525344849,
      "step": 4492
    },
    {
      "epoch": 4.283905029296875e-06,
      "step": 4492,
      "training_step_time": 17.00434398651123
    },
    {
      "epoch": 4.2848587036132816e-06,
      "model_forward_time": 5.589716196060181,
      "step": 4493
    },
    {
      "epoch": 4.2848587036132816e-06,
      "step": 4493,
      "training_step_time": 17.064600944519043
    },
    {
      "epoch": 4.285812377929687e-06,
      "model_forward_time": 5.611166715621948,
      "step": 4494
    },
    {
      "epoch": 4.285812377929687e-06,
      "step": 4494,
      "training_step_time": 17.129018783569336
    },
    {
      "epoch": 4.286766052246094e-06,
      "model_forward_time": 5.628816366195679,
      "step": 4495
    },
    {
      "epoch": 4.286766052246094e-06,
      "step": 4495,
      "training_step_time": 17.16085457801819
    },
    {
      "epoch": 4.2877197265625e-06,
      "model_forward_time": 5.996673583984375,
      "step": 4496
    },
    {
      "epoch": 4.2877197265625e-06,
      "step": 4496,
      "training_step_time": 17.464463472366333
    },
    {
      "epoch": 4.288673400878906e-06,
      "model_forward_time": 5.591312408447266,
      "step": 4497
    },
    {
      "epoch": 4.288673400878906e-06,
      "step": 4497,
      "training_step_time": 17.03920555114746
    },
    {
      "epoch": 4.289627075195313e-06,
      "model_forward_time": 5.587700128555298,
      "step": 4498
    },
    {
      "epoch": 4.289627075195313e-06,
      "step": 4498,
      "training_step_time": 17.011796951293945
    },
    {
      "epoch": 4.2905807495117184e-06,
      "model_forward_time": 5.5833306312561035,
      "step": 4499
    },
    {
      "epoch": 4.2905807495117184e-06,
      "step": 4499,
      "training_step_time": 17.036206245422363
    },
    {
      "epoch": 4.291534423828125e-06,
      "grad_norm": 0.862276017665863,
      "learning_rate": 1e-05,
      "loss": 0.0681,
      "step": 4500
    },
    {
      "epoch": 4.291534423828125e-06,
      "model_forward_time": 5.586797714233398,
      "step": 4500
    },
    {
      "dynamics_loss_avg": 0.05171565655618906,
      "epoch": 4.291534423828125e-06,
      "step": 4500
    },
    {
      "action_loss_avg": 0.015394919412210583,
      "epoch": 4.291534423828125e-06,
      "step": 4500
    },
    {
      "epoch": 4.291534423828125e-06,
      "step": 4500,
      "training_step_time": 17.056071519851685
    },
    {
      "epoch": 4.2924880981445316e-06,
      "model_forward_time": 5.604619741439819,
      "step": 4501
    },
    {
      "epoch": 4.2924880981445316e-06,
      "step": 4501,
      "training_step_time": 17.09808659553528
    },
    {
      "epoch": 4.293441772460937e-06,
      "model_forward_time": 5.9803948402404785,
      "step": 4502
    },
    {
      "epoch": 4.293441772460937e-06,
      "step": 4502,
      "training_step_time": 17.477775812149048
    },
    {
      "epoch": 4.294395446777344e-06,
      "model_forward_time": 5.608191967010498,
      "step": 4503
    },
    {
      "epoch": 4.294395446777344e-06,
      "step": 4503,
      "training_step_time": 17.09859585762024
    },
    {
      "epoch": 4.29534912109375e-06,
      "model_forward_time": 5.606703281402588,
      "step": 4504
    },
    {
      "epoch": 4.29534912109375e-06,
      "step": 4504,
      "training_step_time": 17.091699838638306
    },
    {
      "epoch": 4.296302795410156e-06,
      "model_forward_time": 5.599792718887329,
      "step": 4505
    },
    {
      "epoch": 4.296302795410156e-06,
      "step": 4505,
      "training_step_time": 17.08637046813965
    },
    {
      "epoch": 4.297256469726563e-06,
      "model_forward_time": 5.610533237457275,
      "step": 4506
    },
    {
      "epoch": 4.297256469726563e-06,
      "step": 4506,
      "training_step_time": 17.083855628967285
    },
    {
      "epoch": 4.2982101440429684e-06,
      "model_forward_time": 5.943073511123657,
      "step": 4507
    },
    {
      "epoch": 4.2982101440429684e-06,
      "step": 4507,
      "training_step_time": 17.40680432319641
    },
    {
      "epoch": 4.299163818359375e-06,
      "model_forward_time": 5.606353044509888,
      "step": 4508
    },
    {
      "epoch": 4.299163818359375e-06,
      "step": 4508,
      "training_step_time": 17.05726718902588
    },
    {
      "epoch": 4.3001174926757816e-06,
      "model_forward_time": 5.592409372329712,
      "step": 4509
    },
    {
      "epoch": 4.3001174926757816e-06,
      "step": 4509,
      "training_step_time": 17.034626960754395
    },
    {
      "epoch": 4.301071166992187e-06,
      "grad_norm": 1.0146355628967285,
      "learning_rate": 1e-05,
      "loss": 0.09,
      "step": 4510
    },
    {
      "epoch": 4.301071166992187e-06,
      "model_forward_time": 5.582062005996704,
      "step": 4510
    },
    {
      "dynamics_loss_avg": 0.06396012492477894,
      "epoch": 4.301071166992187e-06,
      "step": 4510
    },
    {
      "action_loss_avg": 0.026361556723713875,
      "epoch": 4.301071166992187e-06,
      "step": 4510
    },
    {
      "epoch": 4.301071166992187e-06,
      "step": 4510,
      "training_step_time": 17.0298855304718
    },
    {
      "epoch": 4.302024841308594e-06,
      "model_forward_time": 5.593839883804321,
      "step": 4511
    },
    {
      "epoch": 4.302024841308594e-06,
      "step": 4511,
      "training_step_time": 17.073193311691284
    },
    {
      "epoch": 4.302978515625e-06,
      "model_forward_time": 5.610234022140503,
      "step": 4512
    },
    {
      "epoch": 4.302978515625e-06,
      "step": 4512,
      "training_step_time": 17.100412368774414
    },
    {
      "epoch": 4.303932189941406e-06,
      "model_forward_time": 5.966290712356567,
      "step": 4513
    },
    {
      "epoch": 4.303932189941406e-06,
      "step": 4513,
      "training_step_time": 17.436439037322998
    },
    {
      "epoch": 4.304885864257813e-06,
      "model_forward_time": 5.592653751373291,
      "step": 4514
    },
    {
      "epoch": 4.304885864257813e-06,
      "step": 4514,
      "training_step_time": 17.039179801940918
    },
    {
      "epoch": 4.3058395385742184e-06,
      "model_forward_time": 5.583346843719482,
      "step": 4515
    },
    {
      "epoch": 4.3058395385742184e-06,
      "step": 4515,
      "training_step_time": 16.996553897857666
    },
    {
      "epoch": 4.306793212890625e-06,
      "model_forward_time": 5.5720720291137695,
      "step": 4516
    },
    {
      "epoch": 4.306793212890625e-06,
      "step": 4516,
      "training_step_time": 16.982166290283203
    },
    {
      "epoch": 4.307746887207032e-06,
      "model_forward_time": 5.591028451919556,
      "step": 4517
    },
    {
      "epoch": 4.307746887207032e-06,
      "step": 4517,
      "training_step_time": 17.016085624694824
    },
    {
      "epoch": 4.308700561523437e-06,
      "model_forward_time": 5.915913105010986,
      "step": 4518
    },
    {
      "epoch": 4.308700561523437e-06,
      "step": 4518,
      "training_step_time": 17.353471279144287
    },
    {
      "epoch": 4.309654235839844e-06,
      "model_forward_time": 5.58367133140564,
      "step": 4519
    },
    {
      "epoch": 4.309654235839844e-06,
      "step": 4519,
      "training_step_time": 17.00102949142456
    },
    {
      "epoch": 4.31060791015625e-06,
      "grad_norm": 2.129189968109131,
      "learning_rate": 1e-05,
      "loss": 0.084,
      "step": 4520
    },
    {
      "epoch": 4.31060791015625e-06,
      "model_forward_time": 5.581805229187012,
      "step": 4520
    },
    {
      "dynamics_loss_avg": 0.06108038276433945,
      "epoch": 4.31060791015625e-06,
      "step": 4520
    },
    {
      "action_loss_avg": 0.01949073055293411,
      "epoch": 4.31060791015625e-06,
      "step": 4520
    },
    {
      "epoch": 4.31060791015625e-06,
      "step": 4520,
      "training_step_time": 16.992053985595703
    },
    {
      "epoch": 4.311561584472656e-06,
      "model_forward_time": 5.592120885848999,
      "step": 4521
    },
    {
      "epoch": 4.311561584472656e-06,
      "step": 4521,
      "training_step_time": 17.031994342803955
    },
    {
      "epoch": 4.312515258789063e-06,
      "model_forward_time": 5.6068360805511475,
      "step": 4522
    },
    {
      "epoch": 4.312515258789063e-06,
      "step": 4522,
      "training_step_time": 17.058695316314697
    },
    {
      "epoch": 4.3134689331054685e-06,
      "model_forward_time": 5.600739479064941,
      "step": 4523
    },
    {
      "epoch": 4.3134689331054685e-06,
      "step": 4523,
      "training_step_time": 17.434414386749268
    },
    {
      "epoch": 4.314422607421875e-06,
      "model_forward_time": 5.606838226318359,
      "step": 4524
    },
    {
      "epoch": 4.314422607421875e-06,
      "step": 4524,
      "training_step_time": 17.105159521102905
    },
    {
      "epoch": 4.315376281738282e-06,
      "model_forward_time": 5.616543531417847,
      "step": 4525
    },
    {
      "epoch": 4.315376281738282e-06,
      "step": 4525,
      "training_step_time": 17.11996579170227
    },
    {
      "epoch": 4.316329956054687e-06,
      "model_forward_time": 5.613231658935547,
      "step": 4526
    },
    {
      "epoch": 4.316329956054687e-06,
      "step": 4526,
      "training_step_time": 17.10246205329895
    },
    {
      "epoch": 4.317283630371094e-06,
      "model_forward_time": 5.610900640487671,
      "step": 4527
    },
    {
      "epoch": 4.317283630371094e-06,
      "step": 4527,
      "training_step_time": 17.078656435012817
    },
    {
      "epoch": 4.3182373046875e-06,
      "model_forward_time": 5.59284520149231,
      "step": 4528
    },
    {
      "epoch": 4.3182373046875e-06,
      "step": 4528,
      "training_step_time": 17.03828454017639
    },
    {
      "epoch": 4.319190979003906e-06,
      "model_forward_time": 5.941185712814331,
      "step": 4529
    },
    {
      "epoch": 4.319190979003906e-06,
      "step": 4529,
      "training_step_time": 17.38264513015747
    },
    {
      "epoch": 4.320144653320313e-06,
      "grad_norm": 1.0802390575408936,
      "learning_rate": 1e-05,
      "loss": 0.0715,
      "step": 4530
    },
    {
      "epoch": 4.320144653320313e-06,
      "model_forward_time": 5.567361354827881,
      "step": 4530
    },
    {
      "dynamics_loss_avg": 0.06213307473808527,
      "epoch": 4.320144653320313e-06,
      "step": 4530
    },
    {
      "action_loss_avg": 0.013962259190157056,
      "epoch": 4.320144653320313e-06,
      "step": 4530
    },
    {
      "epoch": 4.320144653320313e-06,
      "step": 4530,
      "training_step_time": 16.987637042999268
    },
    {
      "epoch": 4.3210983276367185e-06,
      "model_forward_time": 5.574092864990234,
      "step": 4531
    },
    {
      "epoch": 4.3210983276367185e-06,
      "step": 4531,
      "training_step_time": 16.989797830581665
    },
    {
      "epoch": 4.322052001953125e-06,
      "model_forward_time": 5.570071220397949,
      "step": 4532
    },
    {
      "epoch": 4.322052001953125e-06,
      "step": 4532,
      "training_step_time": 16.99648690223694
    },
    {
      "epoch": 4.323005676269532e-06,
      "model_forward_time": 5.586303472518921,
      "step": 4533
    },
    {
      "epoch": 4.323005676269532e-06,
      "step": 4533,
      "training_step_time": 17.027111053466797
    },
    {
      "epoch": 4.323959350585937e-06,
      "model_forward_time": 5.984293222427368,
      "step": 4534
    },
    {
      "epoch": 4.323959350585937e-06,
      "step": 4534,
      "training_step_time": 17.41038990020752
    },
    {
      "epoch": 4.324913024902344e-06,
      "model_forward_time": 5.58870267868042,
      "step": 4535
    },
    {
      "epoch": 4.324913024902344e-06,
      "step": 4535,
      "training_step_time": 17.05930256843567
    },
    {
      "epoch": 4.32586669921875e-06,
      "model_forward_time": 5.594656467437744,
      "step": 4536
    },
    {
      "epoch": 4.32586669921875e-06,
      "step": 4536,
      "training_step_time": 17.06448745727539
    },
    {
      "epoch": 4.326820373535156e-06,
      "model_forward_time": 5.604975700378418,
      "step": 4537
    },
    {
      "epoch": 4.326820373535156e-06,
      "step": 4537,
      "training_step_time": 17.056241750717163
    },
    {
      "epoch": 4.327774047851563e-06,
      "model_forward_time": 5.604308366775513,
      "step": 4538
    },
    {
      "epoch": 4.327774047851563e-06,
      "step": 4538,
      "training_step_time": 17.069246292114258
    },
    {
      "epoch": 4.3287277221679685e-06,
      "model_forward_time": 5.598516941070557,
      "step": 4539
    },
    {
      "epoch": 4.3287277221679685e-06,
      "step": 4539,
      "training_step_time": 17.064117670059204
    },
    {
      "epoch": 4.329681396484375e-06,
      "grad_norm": 1.1112192869186401,
      "learning_rate": 1e-05,
      "loss": 0.0777,
      "step": 4540
    },
    {
      "epoch": 4.329681396484375e-06,
      "model_forward_time": 5.960276126861572,
      "step": 4540
    },
    {
      "dynamics_loss_avg": 0.053953233920037744,
      "epoch": 4.329681396484375e-06,
      "step": 4540
    },
    {
      "action_loss_avg": 0.019377458747476338,
      "epoch": 4.329681396484375e-06,
      "step": 4540
    },
    {
      "epoch": 4.329681396484375e-06,
      "step": 4540,
      "training_step_time": 17.42164134979248
    },
    {
      "epoch": 4.330635070800782e-06,
      "model_forward_time": 5.604896306991577,
      "step": 4541
    },
    {
      "epoch": 4.330635070800782e-06,
      "step": 4541,
      "training_step_time": 17.06086540222168
    },
    {
      "epoch": 4.331588745117187e-06,
      "model_forward_time": 5.597285270690918,
      "step": 4542
    },
    {
      "epoch": 4.331588745117187e-06,
      "step": 4542,
      "training_step_time": 17.04780912399292
    },
    {
      "epoch": 4.332542419433594e-06,
      "model_forward_time": 5.60231614112854,
      "step": 4543
    },
    {
      "epoch": 4.332542419433594e-06,
      "step": 4543,
      "training_step_time": 17.089282274246216
    },
    {
      "epoch": 4.33349609375e-06,
      "model_forward_time": 5.607855796813965,
      "step": 4544
    },
    {
      "epoch": 4.33349609375e-06,
      "step": 4544,
      "training_step_time": 17.057300329208374
    },
    {
      "epoch": 4.334449768066406e-06,
      "model_forward_time": 5.958110332489014,
      "step": 4545
    },
    {
      "epoch": 4.334449768066406e-06,
      "step": 4545,
      "training_step_time": 17.402867555618286
    },
    {
      "epoch": 4.335403442382813e-06,
      "model_forward_time": 5.5981199741363525,
      "step": 4546
    },
    {
      "epoch": 4.335403442382813e-06,
      "step": 4546,
      "training_step_time": 17.026782512664795
    },
    {
      "epoch": 4.3363571166992185e-06,
      "model_forward_time": 5.600055456161499,
      "step": 4547
    },
    {
      "epoch": 4.3363571166992185e-06,
      "step": 4547,
      "training_step_time": 17.08152985572815
    },
    {
      "epoch": 4.337310791015625e-06,
      "model_forward_time": 5.623104810714722,
      "step": 4548
    },
    {
      "epoch": 4.337310791015625e-06,
      "step": 4548,
      "training_step_time": 17.119292736053467
    },
    {
      "epoch": 4.338264465332032e-06,
      "model_forward_time": 5.6061975955963135,
      "step": 4549
    },
    {
      "epoch": 4.338264465332032e-06,
      "step": 4549,
      "training_step_time": 17.088679313659668
    },
    {
      "epoch": 4.339218139648437e-06,
      "grad_norm": 0.6013240218162537,
      "learning_rate": 1e-05,
      "loss": 0.0861,
      "step": 4550
    },
    {
      "epoch": 4.339218139648437e-06,
      "model_forward_time": 5.597895860671997,
      "step": 4550
    },
    {
      "dynamics_loss_avg": 0.07169755157083273,
      "epoch": 4.339218139648437e-06,
      "step": 4550
    },
    {
      "action_loss_avg": 0.01883881303947419,
      "epoch": 4.339218139648437e-06,
      "step": 4550
    },
    {
      "epoch": 4.339218139648437e-06,
      "step": 4550,
      "training_step_time": 17.059281587600708
    },
    {
      "epoch": 4.340171813964844e-06,
      "model_forward_time": 5.942669153213501,
      "step": 4551
    },
    {
      "epoch": 4.340171813964844e-06,
      "step": 4551,
      "training_step_time": 17.38102889060974
    },
    {
      "epoch": 4.34112548828125e-06,
      "model_forward_time": 5.5798797607421875,
      "step": 4552
    },
    {
      "epoch": 4.34112548828125e-06,
      "step": 4552,
      "training_step_time": 17.025129795074463
    },
    {
      "epoch": 4.342079162597656e-06,
      "model_forward_time": 5.584608316421509,
      "step": 4553
    },
    {
      "epoch": 4.342079162597656e-06,
      "step": 4553,
      "training_step_time": 17.040327548980713
    },
    {
      "epoch": 4.343032836914063e-06,
      "model_forward_time": 5.592209100723267,
      "step": 4554
    },
    {
      "epoch": 4.343032836914063e-06,
      "step": 4554,
      "training_step_time": 17.061911582946777
    },
    {
      "epoch": 4.3439865112304685e-06,
      "model_forward_time": 5.601232290267944,
      "step": 4555
    },
    {
      "epoch": 4.3439865112304685e-06,
      "step": 4555,
      "training_step_time": 17.06970477104187
    },
    {
      "epoch": 4.344940185546875e-06,
      "model_forward_time": 5.985761642456055,
      "step": 4556
    },
    {
      "epoch": 4.344940185546875e-06,
      "step": 4556,
      "training_step_time": 17.458892107009888
    },
    {
      "epoch": 4.345893859863282e-06,
      "model_forward_time": 5.611751556396484,
      "step": 4557
    },
    {
      "epoch": 4.345893859863282e-06,
      "step": 4557,
      "training_step_time": 17.10513401031494
    },
    {
      "epoch": 4.346847534179687e-06,
      "model_forward_time": 5.624492645263672,
      "step": 4558
    },
    {
      "epoch": 4.346847534179687e-06,
      "step": 4558,
      "training_step_time": 17.109344959259033
    },
    {
      "epoch": 4.347801208496094e-06,
      "model_forward_time": 5.595176696777344,
      "step": 4559
    },
    {
      "epoch": 4.347801208496094e-06,
      "step": 4559,
      "training_step_time": 17.044689178466797
    },
    {
      "epoch": 4.3487548828125e-06,
      "grad_norm": 2.0206143856048584,
      "learning_rate": 1e-05,
      "loss": 0.0785,
      "step": 4560
    },
    {
      "epoch": 4.3487548828125e-06,
      "model_forward_time": 5.575980186462402,
      "step": 4560
    },
    {
      "dynamics_loss_avg": 0.0585964860394597,
      "epoch": 4.3487548828125e-06,
      "step": 4560
    },
    {
      "action_loss_avg": 0.01628845385275781,
      "epoch": 4.3487548828125e-06,
      "step": 4560
    },
    {
      "epoch": 4.3487548828125e-06,
      "step": 4560,
      "training_step_time": 17.024819135665894
    },
    {
      "epoch": 4.349708557128906e-06,
      "model_forward_time": 5.583478212356567,
      "step": 4561
    },
    {
      "epoch": 4.349708557128906e-06,
      "step": 4561,
      "training_step_time": 17.027156352996826
    },
    {
      "epoch": 4.350662231445313e-06,
      "model_forward_time": 5.925292730331421,
      "step": 4562
    },
    {
      "epoch": 4.350662231445313e-06,
      "step": 4562,
      "training_step_time": 17.383074283599854
    },
    {
      "epoch": 4.3516159057617185e-06,
      "model_forward_time": 5.6114501953125,
      "step": 4563
    },
    {
      "epoch": 4.3516159057617185e-06,
      "step": 4563,
      "training_step_time": 17.090537071228027
    },
    {
      "epoch": 4.352569580078125e-06,
      "model_forward_time": 5.610923767089844,
      "step": 4564
    },
    {
      "epoch": 4.352569580078125e-06,
      "step": 4564,
      "training_step_time": 17.10055446624756
    },
    {
      "epoch": 4.353523254394532e-06,
      "model_forward_time": 5.606713533401489,
      "step": 4565
    },
    {
      "epoch": 4.353523254394532e-06,
      "step": 4565,
      "training_step_time": 17.091376543045044
    },
    {
      "epoch": 4.354476928710937e-06,
      "model_forward_time": 5.603038549423218,
      "step": 4566
    },
    {
      "epoch": 4.354476928710937e-06,
      "step": 4566,
      "training_step_time": 17.08552861213684
    },
    {
      "epoch": 4.355430603027344e-06,
      "model_forward_time": 5.979169130325317,
      "step": 4567
    },
    {
      "epoch": 4.355430603027344e-06,
      "step": 4567,
      "training_step_time": 17.446895837783813
    },
    {
      "epoch": 4.35638427734375e-06,
      "model_forward_time": 5.5920140743255615,
      "step": 4568
    },
    {
      "epoch": 4.35638427734375e-06,
      "step": 4568,
      "training_step_time": 17.04652738571167
    },
    {
      "epoch": 4.357337951660156e-06,
      "model_forward_time": 5.593195199966431,
      "step": 4569
    },
    {
      "epoch": 4.357337951660156e-06,
      "step": 4569,
      "training_step_time": 17.027753591537476
    },
    {
      "epoch": 4.358291625976563e-06,
      "grad_norm": 0.700936496257782,
      "learning_rate": 1e-05,
      "loss": 0.0702,
      "step": 4570
    },
    {
      "epoch": 4.358291625976563e-06,
      "model_forward_time": 5.59000563621521,
      "step": 4570
    },
    {
      "dynamics_loss_avg": 0.055676520243287085,
      "epoch": 4.358291625976563e-06,
      "step": 4570
    },
    {
      "action_loss_avg": 0.020023247529752552,
      "epoch": 4.358291625976563e-06,
      "step": 4570
    },
    {
      "epoch": 4.358291625976563e-06,
      "step": 4570,
      "training_step_time": 17.022063970565796
    },
    {
      "epoch": 4.3592453002929685e-06,
      "model_forward_time": 5.604971647262573,
      "step": 4571
    },
    {
      "epoch": 4.3592453002929685e-06,
      "step": 4571,
      "training_step_time": 17.07219362258911
    },
    {
      "epoch": 4.360198974609375e-06,
      "model_forward_time": 5.961150646209717,
      "step": 4572
    },
    {
      "epoch": 4.360198974609375e-06,
      "step": 4572,
      "training_step_time": 17.44305443763733
    },
    {
      "epoch": 4.361152648925782e-06,
      "model_forward_time": 5.602548599243164,
      "step": 4573
    },
    {
      "epoch": 4.361152648925782e-06,
      "step": 4573,
      "training_step_time": 17.086552381515503
    },
    {
      "epoch": 4.362106323242187e-06,
      "model_forward_time": 5.610278844833374,
      "step": 4574
    },
    {
      "epoch": 4.362106323242187e-06,
      "step": 4574,
      "training_step_time": 17.09957003593445
    },
    {
      "epoch": 4.363059997558594e-06,
      "model_forward_time": 5.60490608215332,
      "step": 4575
    },
    {
      "epoch": 4.363059997558594e-06,
      "step": 4575,
      "training_step_time": 17.073492527008057
    },
    {
      "epoch": 4.364013671875e-06,
      "model_forward_time": 5.601280927658081,
      "step": 4576
    },
    {
      "epoch": 4.364013671875e-06,
      "step": 4576,
      "training_step_time": 17.049519062042236
    },
    {
      "epoch": 4.364967346191406e-06,
      "model_forward_time": 5.597089767456055,
      "step": 4577
    },
    {
      "epoch": 4.364967346191406e-06,
      "step": 4577,
      "training_step_time": 17.059783697128296
    },
    {
      "epoch": 4.365921020507813e-06,
      "model_forward_time": 5.952017307281494,
      "step": 4578
    },
    {
      "epoch": 4.365921020507813e-06,
      "step": 4578,
      "training_step_time": 17.379297971725464
    },
    {
      "epoch": 4.3668746948242185e-06,
      "model_forward_time": 5.591834306716919,
      "step": 4579
    },
    {
      "epoch": 4.3668746948242185e-06,
      "step": 4579,
      "training_step_time": 17.0501606464386
    },
    {
      "epoch": 4.367828369140625e-06,
      "grad_norm": 0.6881329417228699,
      "learning_rate": 1e-05,
      "loss": 0.0798,
      "step": 4580
    },
    {
      "epoch": 4.367828369140625e-06,
      "model_forward_time": 5.603968143463135,
      "step": 4580
    },
    {
      "dynamics_loss_avg": 0.061026794649660585,
      "epoch": 4.367828369140625e-06,
      "step": 4580
    },
    {
      "action_loss_avg": 0.017399217141792177,
      "epoch": 4.367828369140625e-06,
      "step": 4580
    },
    {
      "epoch": 4.367828369140625e-06,
      "step": 4580,
      "training_step_time": 17.103485107421875
    },
    {
      "epoch": 4.368782043457032e-06,
      "model_forward_time": 5.612618446350098,
      "step": 4581
    },
    {
      "epoch": 4.368782043457032e-06,
      "step": 4581,
      "training_step_time": 17.127885580062866
    },
    {
      "epoch": 4.369735717773437e-06,
      "model_forward_time": 5.623121500015259,
      "step": 4582
    },
    {
      "epoch": 4.369735717773437e-06,
      "step": 4582,
      "training_step_time": 17.138593673706055
    },
    {
      "epoch": 4.370689392089844e-06,
      "model_forward_time": 5.620359182357788,
      "step": 4583
    },
    {
      "epoch": 4.370689392089844e-06,
      "step": 4583,
      "training_step_time": 17.491069555282593
    },
    {
      "epoch": 4.37164306640625e-06,
      "model_forward_time": 5.61702299118042,
      "step": 4584
    },
    {
      "epoch": 4.37164306640625e-06,
      "step": 4584,
      "training_step_time": 17.104063987731934
    },
    {
      "epoch": 4.372596740722656e-06,
      "model_forward_time": 5.602577447891235,
      "step": 4585
    },
    {
      "epoch": 4.372596740722656e-06,
      "step": 4585,
      "training_step_time": 17.05157232284546
    },
    {
      "epoch": 4.373550415039063e-06,
      "model_forward_time": 5.599039316177368,
      "step": 4586
    },
    {
      "epoch": 4.373550415039063e-06,
      "step": 4586,
      "training_step_time": 17.048238039016724
    },
    {
      "epoch": 4.3745040893554685e-06,
      "model_forward_time": 5.602404356002808,
      "step": 4587
    },
    {
      "epoch": 4.3745040893554685e-06,
      "step": 4587,
      "training_step_time": 17.06920337677002
    },
    {
      "epoch": 4.375457763671875e-06,
      "model_forward_time": 5.601047039031982,
      "step": 4588
    },
    {
      "epoch": 4.375457763671875e-06,
      "step": 4588,
      "training_step_time": 17.07517123222351
    },
    {
      "epoch": 4.376411437988282e-06,
      "model_forward_time": 5.9577014446258545,
      "step": 4589
    },
    {
      "epoch": 4.376411437988282e-06,
      "step": 4589,
      "training_step_time": 17.41478729248047
    },
    {
      "epoch": 4.377365112304687e-06,
      "grad_norm": 0.6714656352996826,
      "learning_rate": 1e-05,
      "loss": 0.0718,
      "step": 4590
    },
    {
      "epoch": 4.377365112304687e-06,
      "model_forward_time": 5.598883390426636,
      "step": 4590
    },
    {
      "dynamics_loss_avg": 0.04860725980252027,
      "epoch": 4.377365112304687e-06,
      "step": 4590
    },
    {
      "action_loss_avg": 0.017323570651933552,
      "epoch": 4.377365112304687e-06,
      "step": 4590
    },
    {
      "epoch": 4.377365112304687e-06,
      "step": 4590,
      "training_step_time": 17.072299480438232
    },
    {
      "epoch": 4.378318786621094e-06,
      "model_forward_time": 5.601714134216309,
      "step": 4591
    },
    {
      "epoch": 4.378318786621094e-06,
      "step": 4591,
      "training_step_time": 17.051634073257446
    },
    {
      "epoch": 4.3792724609375e-06,
      "model_forward_time": 5.589984893798828,
      "step": 4592
    },
    {
      "epoch": 4.3792724609375e-06,
      "step": 4592,
      "training_step_time": 17.028866052627563
    },
    {
      "epoch": 4.380226135253906e-06,
      "model_forward_time": 5.591523170471191,
      "step": 4593
    },
    {
      "epoch": 4.380226135253906e-06,
      "step": 4593,
      "training_step_time": 17.01860237121582
    },
    {
      "epoch": 4.381179809570313e-06,
      "model_forward_time": 5.594649791717529,
      "step": 4594
    },
    {
      "epoch": 4.381179809570313e-06,
      "step": 4594,
      "training_step_time": 17.40868043899536
    },
    {
      "epoch": 4.3821334838867185e-06,
      "model_forward_time": 5.595820903778076,
      "step": 4595
    },
    {
      "epoch": 4.3821334838867185e-06,
      "step": 4595,
      "training_step_time": 17.07658314704895
    },
    {
      "epoch": 4.383087158203125e-06,
      "model_forward_time": 5.6093926429748535,
      "step": 4596
    },
    {
      "epoch": 4.383087158203125e-06,
      "step": 4596,
      "training_step_time": 17.098002433776855
    },
    {
      "epoch": 4.384040832519532e-06,
      "model_forward_time": 5.605618953704834,
      "step": 4597
    },
    {
      "epoch": 4.384040832519532e-06,
      "step": 4597,
      "training_step_time": 17.042438745498657
    },
    {
      "epoch": 4.384994506835937e-06,
      "model_forward_time": 5.583552360534668,
      "step": 4598
    },
    {
      "epoch": 4.384994506835937e-06,
      "step": 4598,
      "training_step_time": 17.023892164230347
    },
    {
      "epoch": 4.385948181152344e-06,
      "model_forward_time": 5.604733228683472,
      "step": 4599
    },
    {
      "epoch": 4.385948181152344e-06,
      "step": 4599,
      "training_step_time": 17.096231698989868
    },
    {
      "epoch": 4.38690185546875e-06,
      "grad_norm": 0.8520225286483765,
      "learning_rate": 1e-05,
      "loss": 0.073,
      "step": 4600
    },
    {
      "epoch": 4.38690185546875e-06,
      "model_forward_time": 6.002598762512207,
      "step": 4600
    },
    {
      "dynamics_loss_avg": 0.05853969603776932,
      "epoch": 4.38690185546875e-06,
      "step": 4600
    },
    {
      "action_loss_avg": 0.017616474675014614,
      "epoch": 4.38690185546875e-06,
      "step": 4600
    },
    {
      "epoch": 4.38690185546875e-06,
      "step": 4600,
      "training_step_time": 17.52983045578003
    },
    {
      "epoch": 4.387855529785156e-06,
      "model_forward_time": 5.633091926574707,
      "step": 4601
    },
    {
      "epoch": 4.387855529785156e-06,
      "step": 4601,
      "training_step_time": 17.138255834579468
    },
    {
      "epoch": 4.388809204101563e-06,
      "model_forward_time": 5.609922885894775,
      "step": 4602
    },
    {
      "epoch": 4.388809204101563e-06,
      "step": 4602,
      "training_step_time": 17.1020348072052
    },
    {
      "epoch": 4.3897628784179685e-06,
      "model_forward_time": 5.615589618682861,
      "step": 4603
    },
    {
      "epoch": 4.3897628784179685e-06,
      "step": 4603,
      "training_step_time": 17.081252813339233
    },
    {
      "epoch": 4.390716552734375e-06,
      "model_forward_time": 5.597910165786743,
      "step": 4604
    },
    {
      "epoch": 4.390716552734375e-06,
      "step": 4604,
      "training_step_time": 17.049792528152466
    },
    {
      "epoch": 4.391670227050782e-06,
      "model_forward_time": 5.954268455505371,
      "step": 4605
    },
    {
      "epoch": 4.391670227050782e-06,
      "step": 4605,
      "training_step_time": 17.3920738697052
    },
    {
      "epoch": 4.392623901367187e-06,
      "model_forward_time": 5.596741437911987,
      "step": 4606
    },
    {
      "epoch": 4.392623901367187e-06,
      "step": 4606,
      "training_step_time": 17.079405069351196
    },
    {
      "epoch": 4.393577575683594e-06,
      "model_forward_time": 5.618607044219971,
      "step": 4607
    },
    {
      "epoch": 4.393577575683594e-06,
      "step": 4607,
      "training_step_time": 17.116581678390503
    },
    {
      "epoch": 4.39453125e-06,
      "model_forward_time": 5.612689256668091,
      "step": 4608
    },
    {
      "epoch": 4.39453125e-06,
      "step": 4608,
      "training_step_time": 17.092814922332764
    },
    {
      "epoch": 4.395484924316406e-06,
      "model_forward_time": 5.604864120483398,
      "step": 4609
    },
    {
      "epoch": 4.395484924316406e-06,
      "step": 4609,
      "training_step_time": 17.062906503677368
    },
    {
      "epoch": 4.396438598632813e-06,
      "grad_norm": 1.4584318399429321,
      "learning_rate": 1e-05,
      "loss": 0.097,
      "step": 4610
    },
    {
      "epoch": 4.396438598632813e-06,
      "model_forward_time": 5.601635932922363,
      "step": 4610
    },
    {
      "dynamics_loss_avg": 0.07471624873578549,
      "epoch": 4.396438598632813e-06,
      "step": 4610
    },
    {
      "action_loss_avg": 0.0217735075391829,
      "epoch": 4.396438598632813e-06,
      "step": 4610
    },
    {
      "epoch": 4.396438598632813e-06,
      "step": 4610,
      "training_step_time": 17.057068347930908
    },
    {
      "epoch": 4.3973922729492185e-06,
      "model_forward_time": 5.986415386199951,
      "step": 4611
    },
    {
      "epoch": 4.3973922729492185e-06,
      "step": 4611,
      "training_step_time": 17.432923555374146
    },
    {
      "epoch": 4.398345947265625e-06,
      "model_forward_time": 5.590044975280762,
      "step": 4612
    },
    {
      "epoch": 4.398345947265625e-06,
      "step": 4612,
      "training_step_time": 17.057495594024658
    },
    {
      "epoch": 4.399299621582032e-06,
      "model_forward_time": 5.609151124954224,
      "step": 4613
    },
    {
      "epoch": 4.399299621582032e-06,
      "step": 4613,
      "training_step_time": 17.084800720214844
    },
    {
      "epoch": 4.400253295898437e-06,
      "model_forward_time": 5.603265762329102,
      "step": 4614
    },
    {
      "epoch": 4.400253295898437e-06,
      "step": 4614,
      "training_step_time": 17.07964324951172
    },
    {
      "epoch": 4.401206970214844e-06,
      "model_forward_time": 5.60245680809021,
      "step": 4615
    },
    {
      "epoch": 4.401206970214844e-06,
      "step": 4615,
      "training_step_time": 17.068259716033936
    },
    {
      "epoch": 4.40216064453125e-06,
      "model_forward_time": 5.596721649169922,
      "step": 4616
    },
    {
      "epoch": 4.40216064453125e-06,
      "step": 4616,
      "training_step_time": 17.43047046661377
    },
    {
      "epoch": 4.403114318847656e-06,
      "model_forward_time": 5.628353118896484,
      "step": 4617
    },
    {
      "epoch": 4.403114318847656e-06,
      "step": 4617,
      "training_step_time": 17.070911169052124
    },
    {
      "epoch": 4.404067993164063e-06,
      "model_forward_time": 5.5881311893463135,
      "step": 4618
    },
    {
      "epoch": 4.404067993164063e-06,
      "step": 4618,
      "training_step_time": 17.031691551208496
    },
    {
      "epoch": 4.4050216674804685e-06,
      "model_forward_time": 5.603230953216553,
      "step": 4619
    },
    {
      "epoch": 4.4050216674804685e-06,
      "step": 4619,
      "training_step_time": 17.066758155822754
    },
    {
      "epoch": 4.405975341796875e-06,
      "grad_norm": 1.5104714632034302,
      "learning_rate": 1e-05,
      "loss": 0.1045,
      "step": 4620
    },
    {
      "epoch": 4.405975341796875e-06,
      "model_forward_time": 5.5940752029418945,
      "step": 4620
    },
    {
      "dynamics_loss_avg": 0.0806131612509489,
      "epoch": 4.405975341796875e-06,
      "step": 4620
    },
    {
      "action_loss_avg": 0.022320290794596076,
      "epoch": 4.405975341796875e-06,
      "step": 4620
    },
    {
      "epoch": 4.405975341796875e-06,
      "step": 4620,
      "training_step_time": 17.057231903076172
    },
    {
      "epoch": 4.406929016113282e-06,
      "model_forward_time": 5.614463567733765,
      "step": 4621
    },
    {
      "epoch": 4.406929016113282e-06,
      "step": 4621,
      "training_step_time": 17.08386540412903
    },
    {
      "epoch": 4.407882690429687e-06,
      "model_forward_time": 5.969204902648926,
      "step": 4622
    },
    {
      "epoch": 4.407882690429687e-06,
      "step": 4622,
      "training_step_time": 17.44845485687256
    },
    {
      "epoch": 4.408836364746094e-06,
      "model_forward_time": 5.6091179847717285,
      "step": 4623
    },
    {
      "epoch": 4.408836364746094e-06,
      "step": 4623,
      "training_step_time": 17.088515758514404
    },
    {
      "epoch": 4.4097900390625e-06,
      "model_forward_time": 5.614638566970825,
      "step": 4624
    },
    {
      "epoch": 4.4097900390625e-06,
      "step": 4624,
      "training_step_time": 17.11096954345703
    },
    {
      "epoch": 4.410743713378906e-06,
      "model_forward_time": 5.633335828781128,
      "step": 4625
    },
    {
      "epoch": 4.410743713378906e-06,
      "step": 4625,
      "training_step_time": 17.119442224502563
    },
    {
      "epoch": 4.411697387695313e-06,
      "model_forward_time": 5.609516382217407,
      "step": 4626
    },
    {
      "epoch": 4.411697387695313e-06,
      "step": 4626,
      "training_step_time": 17.103128910064697
    },
    {
      "epoch": 4.4126510620117185e-06,
      "model_forward_time": 5.605840682983398,
      "step": 4627
    },
    {
      "epoch": 4.4126510620117185e-06,
      "step": 4627,
      "training_step_time": 17.442403078079224
    },
    {
      "epoch": 4.413604736328125e-06,
      "model_forward_time": 5.604579210281372,
      "step": 4628
    },
    {
      "epoch": 4.413604736328125e-06,
      "step": 4628,
      "training_step_time": 17.080941200256348
    },
    {
      "epoch": 4.414558410644531e-06,
      "model_forward_time": 5.605187654495239,
      "step": 4629
    },
    {
      "epoch": 4.414558410644531e-06,
      "step": 4629,
      "training_step_time": 17.09999656677246
    },
    {
      "epoch": 4.415512084960937e-06,
      "grad_norm": 1.0833778381347656,
      "learning_rate": 1e-05,
      "loss": 0.0857,
      "step": 4630
    },
    {
      "epoch": 4.415512084960937e-06,
      "model_forward_time": 5.6110358238220215,
      "step": 4630
    },
    {
      "dynamics_loss_avg": 0.07253838181495667,
      "epoch": 4.415512084960937e-06,
      "step": 4630
    },
    {
      "action_loss_avg": 0.017848652973771095,
      "epoch": 4.415512084960937e-06,
      "step": 4630
    },
    {
      "epoch": 4.415512084960937e-06,
      "step": 4630,
      "training_step_time": 17.098443031311035
    },
    {
      "epoch": 4.416465759277344e-06,
      "model_forward_time": 5.606011629104614,
      "step": 4631
    },
    {
      "epoch": 4.416465759277344e-06,
      "step": 4631,
      "training_step_time": 17.0915470123291
    },
    {
      "epoch": 4.41741943359375e-06,
      "model_forward_time": 5.606677055358887,
      "step": 4632
    },
    {
      "epoch": 4.41741943359375e-06,
      "step": 4632,
      "training_step_time": 17.06401491165161
    },
    {
      "epoch": 4.418373107910156e-06,
      "model_forward_time": 5.982618093490601,
      "step": 4633
    },
    {
      "epoch": 4.418373107910156e-06,
      "step": 4633,
      "training_step_time": 17.4487144947052
    },
    {
      "epoch": 4.419326782226563e-06,
      "model_forward_time": 5.589317560195923,
      "step": 4634
    },
    {
      "epoch": 4.419326782226563e-06,
      "step": 4634,
      "training_step_time": 17.026857137680054
    },
    {
      "epoch": 4.4202804565429685e-06,
      "model_forward_time": 5.59067177772522,
      "step": 4635
    },
    {
      "epoch": 4.4202804565429685e-06,
      "step": 4635,
      "training_step_time": 17.020063638687134
    },
    {
      "epoch": 4.421234130859375e-06,
      "model_forward_time": 5.5837531089782715,
      "step": 4636
    },
    {
      "epoch": 4.421234130859375e-06,
      "step": 4636,
      "training_step_time": 17.02767515182495
    },
    {
      "epoch": 4.422187805175781e-06,
      "model_forward_time": 5.595045804977417,
      "step": 4637
    },
    {
      "epoch": 4.422187805175781e-06,
      "step": 4637,
      "training_step_time": 17.040658235549927
    },
    {
      "epoch": 4.423141479492187e-06,
      "model_forward_time": 5.961462736129761,
      "step": 4638
    },
    {
      "epoch": 4.423141479492187e-06,
      "step": 4638,
      "training_step_time": 17.41611623764038
    },
    {
      "epoch": 4.424095153808594e-06,
      "model_forward_time": 5.59402322769165,
      "step": 4639
    },
    {
      "epoch": 4.424095153808594e-06,
      "step": 4639,
      "training_step_time": 17.024962186813354
    },
    {
      "epoch": 4.425048828125e-06,
      "grad_norm": 1.6626540422439575,
      "learning_rate": 1e-05,
      "loss": 0.0712,
      "step": 4640
    },
    {
      "epoch": 4.425048828125e-06,
      "model_forward_time": 5.593416929244995,
      "step": 4640
    },
    {
      "dynamics_loss_avg": 0.053095747623592614,
      "epoch": 4.425048828125e-06,
      "step": 4640
    },
    {
      "action_loss_avg": 0.014221194107085467,
      "epoch": 4.425048828125e-06,
      "step": 4640
    },
    {
      "epoch": 4.425048828125e-06,
      "step": 4640,
      "training_step_time": 17.04328203201294
    },
    {
      "epoch": 4.426002502441406e-06,
      "model_forward_time": 5.597314357757568,
      "step": 4641
    },
    {
      "epoch": 4.426002502441406e-06,
      "step": 4641,
      "training_step_time": 17.0459406375885
    },
    {
      "epoch": 4.426956176757813e-06,
      "model_forward_time": 5.6020872592926025,
      "step": 4642
    },
    {
      "epoch": 4.426956176757813e-06,
      "step": 4642,
      "training_step_time": 17.090147972106934
    },
    {
      "epoch": 4.4279098510742186e-06,
      "model_forward_time": 5.611083984375,
      "step": 4643
    },
    {
      "epoch": 4.4279098510742186e-06,
      "step": 4643,
      "training_step_time": 17.098804712295532
    },
    {
      "epoch": 4.428863525390625e-06,
      "model_forward_time": 5.981923341751099,
      "step": 4644
    },
    {
      "epoch": 4.428863525390625e-06,
      "step": 4644,
      "training_step_time": 17.483966588974
    },
    {
      "epoch": 4.429817199707031e-06,
      "model_forward_time": 5.611554861068726,
      "step": 4645
    },
    {
      "epoch": 4.429817199707031e-06,
      "step": 4645,
      "training_step_time": 17.111066341400146
    },
    {
      "epoch": 4.430770874023437e-06,
      "model_forward_time": 5.618869304656982,
      "step": 4646
    },
    {
      "epoch": 4.430770874023437e-06,
      "step": 4646,
      "training_step_time": 17.11532735824585
    },
    {
      "epoch": 4.431724548339844e-06,
      "model_forward_time": 5.606813430786133,
      "step": 4647
    },
    {
      "epoch": 4.431724548339844e-06,
      "step": 4647,
      "training_step_time": 17.097163915634155
    },
    {
      "epoch": 4.43267822265625e-06,
      "model_forward_time": 5.5994555950164795,
      "step": 4648
    },
    {
      "epoch": 4.43267822265625e-06,
      "step": 4648,
      "training_step_time": 17.08539652824402
    },
    {
      "epoch": 4.433631896972656e-06,
      "model_forward_time": 5.615588188171387,
      "step": 4649
    },
    {
      "epoch": 4.433631896972656e-06,
      "step": 4649,
      "training_step_time": 17.477511405944824
    },
    {
      "epoch": 4.434585571289063e-06,
      "grad_norm": 1.2421425580978394,
      "learning_rate": 1e-05,
      "loss": 0.0616,
      "step": 4650
    },
    {
      "epoch": 4.434585571289063e-06,
      "model_forward_time": 5.606165409088135,
      "step": 4650
    },
    {
      "dynamics_loss_avg": 0.054105729423463346,
      "epoch": 4.434585571289063e-06,
      "step": 4650
    },
    {
      "action_loss_avg": 0.015453584655188025,
      "epoch": 4.434585571289063e-06,
      "step": 4650
    },
    {
      "epoch": 4.434585571289063e-06,
      "step": 4650,
      "training_step_time": 17.09604811668396
    },
    {
      "epoch": 4.4355392456054686e-06,
      "model_forward_time": 5.611787557601929,
      "step": 4651
    },
    {
      "epoch": 4.4355392456054686e-06,
      "step": 4651,
      "training_step_time": 17.07586431503296
    },
    {
      "epoch": 4.436492919921875e-06,
      "model_forward_time": 5.595539093017578,
      "step": 4652
    },
    {
      "epoch": 4.436492919921875e-06,
      "step": 4652,
      "training_step_time": 17.064521074295044
    },
    {
      "epoch": 4.437446594238281e-06,
      "model_forward_time": 5.598502159118652,
      "step": 4653
    },
    {
      "epoch": 4.437446594238281e-06,
      "step": 4653,
      "training_step_time": 17.054354190826416
    },
    {
      "epoch": 4.438400268554687e-06,
      "model_forward_time": 5.584424018859863,
      "step": 4654
    },
    {
      "epoch": 4.438400268554687e-06,
      "step": 4654,
      "training_step_time": 17.017014503479004
    },
    {
      "epoch": 4.439353942871094e-06,
      "model_forward_time": 5.949716567993164,
      "step": 4655
    },
    {
      "epoch": 4.439353942871094e-06,
      "step": 4655,
      "training_step_time": 17.37498664855957
    },
    {
      "epoch": 4.4403076171875e-06,
      "model_forward_time": 5.599048137664795,
      "step": 4656
    },
    {
      "epoch": 4.4403076171875e-06,
      "step": 4656,
      "training_step_time": 17.09275507926941
    },
    {
      "epoch": 4.441261291503906e-06,
      "model_forward_time": 5.607990026473999,
      "step": 4657
    },
    {
      "epoch": 4.441261291503906e-06,
      "step": 4657,
      "training_step_time": 17.116173267364502
    },
    {
      "epoch": 4.442214965820313e-06,
      "model_forward_time": 5.610281944274902,
      "step": 4658
    },
    {
      "epoch": 4.442214965820313e-06,
      "step": 4658,
      "training_step_time": 17.114975690841675
    },
    {
      "epoch": 4.4431686401367186e-06,
      "model_forward_time": 5.613829135894775,
      "step": 4659
    },
    {
      "epoch": 4.4431686401367186e-06,
      "step": 4659,
      "training_step_time": 17.084168672561646
    },
    {
      "epoch": 4.444122314453125e-06,
      "grad_norm": 1.7612420320510864,
      "learning_rate": 1e-05,
      "loss": 0.0985,
      "step": 4660
    },
    {
      "epoch": 4.444122314453125e-06,
      "model_forward_time": 5.591062068939209,
      "step": 4660
    },
    {
      "dynamics_loss_avg": 0.06928931437432766,
      "epoch": 4.444122314453125e-06,
      "step": 4660
    },
    {
      "action_loss_avg": 0.020184840820729733,
      "epoch": 4.444122314453125e-06,
      "step": 4660
    },
    {
      "epoch": 4.444122314453125e-06,
      "step": 4660,
      "training_step_time": 17.374861240386963
    },
    {
      "epoch": 4.445075988769531e-06,
      "model_forward_time": 5.580084800720215,
      "step": 4661
    },
    {
      "epoch": 4.445075988769531e-06,
      "step": 4661,
      "training_step_time": 16.994489908218384
    },
    {
      "epoch": 4.4460296630859374e-06,
      "model_forward_time": 5.55973219871521,
      "step": 4662
    },
    {
      "epoch": 4.4460296630859374e-06,
      "step": 4662,
      "training_step_time": 16.990575790405273
    },
    {
      "epoch": 4.446983337402344e-06,
      "model_forward_time": 5.563408613204956,
      "step": 4663
    },
    {
      "epoch": 4.446983337402344e-06,
      "step": 4663,
      "training_step_time": 16.985675811767578
    },
    {
      "epoch": 4.44793701171875e-06,
      "model_forward_time": 5.574153661727905,
      "step": 4664
    },
    {
      "epoch": 4.44793701171875e-06,
      "step": 4664,
      "training_step_time": 16.99028754234314
    },
    {
      "epoch": 4.448890686035156e-06,
      "model_forward_time": 5.576236963272095,
      "step": 4665
    },
    {
      "epoch": 4.448890686035156e-06,
      "step": 4665,
      "training_step_time": 16.996222019195557
    },
    {
      "epoch": 4.449844360351563e-06,
      "model_forward_time": 5.9468066692352295,
      "step": 4666
    },
    {
      "epoch": 4.449844360351563e-06,
      "step": 4666,
      "training_step_time": 17.388378143310547
    },
    {
      "epoch": 4.4507980346679686e-06,
      "model_forward_time": 5.5887229442596436,
      "step": 4667
    },
    {
      "epoch": 4.4507980346679686e-06,
      "step": 4667,
      "training_step_time": 17.024108171463013
    },
    {
      "epoch": 4.451751708984375e-06,
      "model_forward_time": 5.592292070388794,
      "step": 4668
    },
    {
      "epoch": 4.451751708984375e-06,
      "step": 4668,
      "training_step_time": 17.046082019805908
    },
    {
      "epoch": 4.452705383300781e-06,
      "model_forward_time": 5.605506420135498,
      "step": 4669
    },
    {
      "epoch": 4.452705383300781e-06,
      "step": 4669,
      "training_step_time": 17.06123399734497
    },
    {
      "epoch": 4.4536590576171874e-06,
      "grad_norm": 1.7234740257263184,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 4670
    },
    {
      "epoch": 4.4536590576171874e-06,
      "model_forward_time": 5.5889294147491455,
      "step": 4670
    },
    {
      "dynamics_loss_avg": 0.0531217398121953,
      "epoch": 4.4536590576171874e-06,
      "step": 4670
    },
    {
      "action_loss_avg": 0.014467167062684893,
      "epoch": 4.4536590576171874e-06,
      "step": 4670
    },
    {
      "epoch": 4.4536590576171874e-06,
      "step": 4670,
      "training_step_time": 17.029101133346558
    },
    {
      "epoch": 4.454612731933594e-06,
      "model_forward_time": 5.93698525428772,
      "step": 4671
    },
    {
      "epoch": 4.454612731933594e-06,
      "step": 4671,
      "training_step_time": 17.38120412826538
    },
    {
      "epoch": 4.45556640625e-06,
      "model_forward_time": 5.591092348098755,
      "step": 4672
    },
    {
      "epoch": 4.45556640625e-06,
      "step": 4672,
      "training_step_time": 17.065776586532593
    },
    {
      "epoch": 4.456520080566406e-06,
      "model_forward_time": 5.6031458377838135,
      "step": 4673
    },
    {
      "epoch": 4.456520080566406e-06,
      "step": 4673,
      "training_step_time": 17.07692313194275
    },
    {
      "epoch": 4.457473754882813e-06,
      "model_forward_time": 5.608989238739014,
      "step": 4674
    },
    {
      "epoch": 4.457473754882813e-06,
      "step": 4674,
      "training_step_time": 17.084569692611694
    },
    {
      "epoch": 4.458427429199219e-06,
      "model_forward_time": 5.60116720199585,
      "step": 4675
    },
    {
      "epoch": 4.458427429199219e-06,
      "step": 4675,
      "training_step_time": 17.060522079467773
    },
    {
      "epoch": 4.459381103515625e-06,
      "model_forward_time": 5.596523761749268,
      "step": 4676
    },
    {
      "epoch": 4.459381103515625e-06,
      "step": 4676,
      "training_step_time": 17.05527687072754
    },
    {
      "epoch": 4.460334777832031e-06,
      "model_forward_time": 5.961175918579102,
      "step": 4677
    },
    {
      "epoch": 4.460334777832031e-06,
      "step": 4677,
      "training_step_time": 17.38336491584778
    },
    {
      "epoch": 4.4612884521484374e-06,
      "model_forward_time": 5.589228868484497,
      "step": 4678
    },
    {
      "epoch": 4.4612884521484374e-06,
      "step": 4678,
      "training_step_time": 16.972289323806763
    },
    {
      "epoch": 4.462242126464844e-06,
      "model_forward_time": 5.591385364532471,
      "step": 4679
    },
    {
      "epoch": 4.462242126464844e-06,
      "step": 4679,
      "training_step_time": 17.06534743309021
    },
    {
      "epoch": 4.46319580078125e-06,
      "grad_norm": 0.35855066776275635,
      "learning_rate": 1e-05,
      "loss": 0.0613,
      "step": 4680
    },
    {
      "epoch": 4.46319580078125e-06,
      "model_forward_time": 5.609762907028198,
      "step": 4680
    },
    {
      "dynamics_loss_avg": 0.04773284401744604,
      "epoch": 4.46319580078125e-06,
      "step": 4680
    },
    {
      "action_loss_avg": 0.01481505618430674,
      "epoch": 4.46319580078125e-06,
      "step": 4680
    },
    {
      "epoch": 4.46319580078125e-06,
      "step": 4680,
      "training_step_time": 17.10296869277954
    },
    {
      "epoch": 4.464149475097656e-06,
      "model_forward_time": 5.6172263622283936,
      "step": 4681
    },
    {
      "epoch": 4.464149475097656e-06,
      "step": 4681,
      "training_step_time": 17.10997700691223
    },
    {
      "epoch": 4.465103149414063e-06,
      "model_forward_time": 5.589670181274414,
      "step": 4682
    },
    {
      "epoch": 4.465103149414063e-06,
      "step": 4682,
      "training_step_time": 17.412744522094727
    },
    {
      "epoch": 4.466056823730469e-06,
      "model_forward_time": 5.589036703109741,
      "step": 4683
    },
    {
      "epoch": 4.466056823730469e-06,
      "step": 4683,
      "training_step_time": 17.02250051498413
    },
    {
      "epoch": 4.467010498046875e-06,
      "model_forward_time": 5.60033655166626,
      "step": 4684
    },
    {
      "epoch": 4.467010498046875e-06,
      "step": 4684,
      "training_step_time": 17.056983709335327
    },
    {
      "epoch": 4.467964172363281e-06,
      "model_forward_time": 5.592907905578613,
      "step": 4685
    },
    {
      "epoch": 4.467964172363281e-06,
      "step": 4685,
      "training_step_time": 17.052491426467896
    },
    {
      "epoch": 4.4689178466796874e-06,
      "model_forward_time": 5.5970618724823,
      "step": 4686
    },
    {
      "epoch": 4.4689178466796874e-06,
      "step": 4686,
      "training_step_time": 17.072513818740845
    },
    {
      "epoch": 4.469871520996094e-06,
      "model_forward_time": 5.621049880981445,
      "step": 4687
    },
    {
      "epoch": 4.469871520996094e-06,
      "step": 4687,
      "training_step_time": 17.12478232383728
    },
    {
      "epoch": 4.4708251953125e-06,
      "model_forward_time": 5.980130434036255,
      "step": 4688
    },
    {
      "epoch": 4.4708251953125e-06,
      "step": 4688,
      "training_step_time": 17.49377155303955
    },
    {
      "epoch": 4.471778869628906e-06,
      "model_forward_time": 5.6151509284973145,
      "step": 4689
    },
    {
      "epoch": 4.471778869628906e-06,
      "step": 4689,
      "training_step_time": 17.11402654647827
    },
    {
      "epoch": 4.472732543945313e-06,
      "grad_norm": 1.328221321105957,
      "learning_rate": 1e-05,
      "loss": 0.0755,
      "step": 4690
    },
    {
      "epoch": 4.472732543945313e-06,
      "model_forward_time": 5.61503529548645,
      "step": 4690
    },
    {
      "dynamics_loss_avg": 0.0529466524720192,
      "epoch": 4.472732543945313e-06,
      "step": 4690
    },
    {
      "action_loss_avg": 0.0219987406861037,
      "epoch": 4.472732543945313e-06,
      "step": 4690
    },
    {
      "epoch": 4.472732543945313e-06,
      "step": 4690,
      "training_step_time": 17.11941695213318
    },
    {
      "epoch": 4.473686218261719e-06,
      "model_forward_time": 5.6170854568481445,
      "step": 4691
    },
    {
      "epoch": 4.473686218261719e-06,
      "step": 4691,
      "training_step_time": 17.120789289474487
    },
    {
      "epoch": 4.474639892578125e-06,
      "model_forward_time": 5.609944105148315,
      "step": 4692
    },
    {
      "epoch": 4.474639892578125e-06,
      "step": 4692,
      "training_step_time": 17.105009078979492
    },
    {
      "epoch": 4.475593566894531e-06,
      "model_forward_time": 5.606573820114136,
      "step": 4693
    },
    {
      "epoch": 4.475593566894531e-06,
      "step": 4693,
      "training_step_time": 17.455031633377075
    },
    {
      "epoch": 4.4765472412109375e-06,
      "model_forward_time": 5.604295969009399,
      "step": 4694
    },
    {
      "epoch": 4.4765472412109375e-06,
      "step": 4694,
      "training_step_time": 17.065513610839844
    },
    {
      "epoch": 4.477500915527344e-06,
      "model_forward_time": 5.594145059585571,
      "step": 4695
    },
    {
      "epoch": 4.477500915527344e-06,
      "step": 4695,
      "training_step_time": 17.060999870300293
    },
    {
      "epoch": 4.47845458984375e-06,
      "model_forward_time": 5.595637321472168,
      "step": 4696
    },
    {
      "epoch": 4.47845458984375e-06,
      "step": 4696,
      "training_step_time": 17.051514148712158
    },
    {
      "epoch": 4.479408264160156e-06,
      "model_forward_time": 5.588892459869385,
      "step": 4697
    },
    {
      "epoch": 4.479408264160156e-06,
      "step": 4697,
      "training_step_time": 17.03654718399048
    },
    {
      "epoch": 4.480361938476563e-06,
      "model_forward_time": 5.5933473110198975,
      "step": 4698
    },
    {
      "epoch": 4.480361938476563e-06,
      "step": 4698,
      "training_step_time": 17.033965587615967
    },
    {
      "epoch": 4.481315612792969e-06,
      "model_forward_time": 5.953962802886963,
      "step": 4699
    },
    {
      "epoch": 4.481315612792969e-06,
      "step": 4699,
      "training_step_time": 17.42887020111084
    },
    {
      "epoch": 4.482269287109375e-06,
      "grad_norm": 0.5717409253120422,
      "learning_rate": 1e-05,
      "loss": 0.0667,
      "step": 4700
    },
    {
      "epoch": 4.482269287109375e-06,
      "model_forward_time": 5.60542893409729,
      "step": 4700
    },
    {
      "dynamics_loss_avg": 0.05334753766655922,
      "epoch": 4.482269287109375e-06,
      "step": 4700
    },
    {
      "action_loss_avg": 0.015469210990704595,
      "epoch": 4.482269287109375e-06,
      "step": 4700
    },
    {
      "epoch": 4.482269287109375e-06,
      "step": 4700,
      "training_step_time": 17.102001428604126
    },
    {
      "epoch": 4.483222961425781e-06,
      "model_forward_time": 5.6027140617370605,
      "step": 4701
    },
    {
      "epoch": 4.483222961425781e-06,
      "step": 4701,
      "training_step_time": 17.07410216331482
    },
    {
      "epoch": 4.4841766357421875e-06,
      "model_forward_time": 5.597202301025391,
      "step": 4702
    },
    {
      "epoch": 4.4841766357421875e-06,
      "step": 4702,
      "training_step_time": 17.04287338256836
    },
    {
      "epoch": 4.485130310058594e-06,
      "model_forward_time": 5.594294309616089,
      "step": 4703
    },
    {
      "epoch": 4.485130310058594e-06,
      "step": 4703,
      "training_step_time": 17.045905113220215
    },
    {
      "epoch": 4.486083984375e-06,
      "model_forward_time": 5.9539220333099365,
      "step": 4704
    },
    {
      "epoch": 4.486083984375e-06,
      "step": 4704,
      "training_step_time": 17.386833667755127
    },
    {
      "epoch": 4.487037658691406e-06,
      "model_forward_time": 5.5849010944366455,
      "step": 4705
    },
    {
      "epoch": 4.487037658691406e-06,
      "step": 4705,
      "training_step_time": 17.04056453704834
    },
    {
      "epoch": 4.487991333007813e-06,
      "model_forward_time": 5.601792573928833,
      "step": 4706
    },
    {
      "epoch": 4.487991333007813e-06,
      "step": 4706,
      "training_step_time": 17.082855939865112
    },
    {
      "epoch": 4.488945007324219e-06,
      "model_forward_time": 5.613949298858643,
      "step": 4707
    },
    {
      "epoch": 4.488945007324219e-06,
      "step": 4707,
      "training_step_time": 17.11482572555542
    },
    {
      "epoch": 4.489898681640625e-06,
      "model_forward_time": 5.610953092575073,
      "step": 4708
    },
    {
      "epoch": 4.489898681640625e-06,
      "step": 4708,
      "training_step_time": 17.11014986038208
    },
    {
      "epoch": 4.490852355957031e-06,
      "model_forward_time": 5.607580900192261,
      "step": 4709
    },
    {
      "epoch": 4.490852355957031e-06,
      "step": 4709,
      "training_step_time": 17.101378679275513
    },
    {
      "epoch": 4.4918060302734375e-06,
      "grad_norm": 1.2268224954605103,
      "learning_rate": 1e-05,
      "loss": 0.0654,
      "step": 4710
    },
    {
      "epoch": 4.4918060302734375e-06,
      "model_forward_time": 5.976533651351929,
      "step": 4710
    },
    {
      "dynamics_loss_avg": 0.05203145239502192,
      "epoch": 4.4918060302734375e-06,
      "step": 4710
    },
    {
      "action_loss_avg": 0.01617475552484393,
      "epoch": 4.4918060302734375e-06,
      "step": 4710
    },
    {
      "epoch": 4.4918060302734375e-06,
      "step": 4710,
      "training_step_time": 17.46455717086792
    },
    {
      "epoch": 4.492759704589844e-06,
      "model_forward_time": 5.60478663444519,
      "step": 4711
    },
    {
      "epoch": 4.492759704589844e-06,
      "step": 4711,
      "training_step_time": 17.09407877922058
    },
    {
      "epoch": 4.49371337890625e-06,
      "model_forward_time": 5.608643293380737,
      "step": 4712
    },
    {
      "epoch": 4.49371337890625e-06,
      "step": 4712,
      "training_step_time": 17.10304880142212
    },
    {
      "epoch": 4.494667053222656e-06,
      "model_forward_time": 5.612378120422363,
      "step": 4713
    },
    {
      "epoch": 4.494667053222656e-06,
      "step": 4713,
      "training_step_time": 17.10994029045105
    },
    {
      "epoch": 4.495620727539063e-06,
      "model_forward_time": 5.612897872924805,
      "step": 4714
    },
    {
      "epoch": 4.495620727539063e-06,
      "step": 4714,
      "training_step_time": 17.104626417160034
    },
    {
      "epoch": 4.496574401855469e-06,
      "model_forward_time": 5.608721017837524,
      "step": 4715
    },
    {
      "epoch": 4.496574401855469e-06,
      "step": 4715,
      "training_step_time": 17.45832133293152
    },
    {
      "epoch": 4.497528076171875e-06,
      "model_forward_time": 5.614690542221069,
      "step": 4716
    },
    {
      "epoch": 4.497528076171875e-06,
      "step": 4716,
      "training_step_time": 17.10931944847107
    },
    {
      "epoch": 4.498481750488281e-06,
      "model_forward_time": 5.601944208145142,
      "step": 4717
    },
    {
      "epoch": 4.498481750488281e-06,
      "step": 4717,
      "training_step_time": 17.094937562942505
    },
    {
      "epoch": 4.4994354248046875e-06,
      "model_forward_time": 5.619563102722168,
      "step": 4718
    },
    {
      "epoch": 4.4994354248046875e-06,
      "step": 4718,
      "training_step_time": 17.100457668304443
    },
    {
      "epoch": 4.500389099121094e-06,
      "model_forward_time": 5.600887298583984,
      "step": 4719
    },
    {
      "epoch": 4.500389099121094e-06,
      "step": 4719,
      "training_step_time": 17.070300579071045
    },
    {
      "epoch": 4.5013427734375e-06,
      "grad_norm": 0.7963244318962097,
      "learning_rate": 1e-05,
      "loss": 0.0562,
      "step": 4720
    },
    {
      "epoch": 4.5013427734375e-06,
      "model_forward_time": 5.613297939300537,
      "step": 4720
    },
    {
      "dynamics_loss_avg": 0.04270941913127899,
      "epoch": 4.5013427734375e-06,
      "step": 4720
    },
    {
      "action_loss_avg": 0.010576836112886668,
      "epoch": 4.5013427734375e-06,
      "step": 4720
    },
    {
      "epoch": 4.5013427734375e-06,
      "step": 4720,
      "training_step_time": 17.094763040542603
    },
    {
      "epoch": 4.502296447753906e-06,
      "model_forward_time": 5.959221839904785,
      "step": 4721
    },
    {
      "epoch": 4.502296447753906e-06,
      "step": 4721,
      "training_step_time": 17.435565948486328
    },
    {
      "epoch": 4.503250122070313e-06,
      "model_forward_time": 5.598867654800415,
      "step": 4722
    },
    {
      "epoch": 4.503250122070313e-06,
      "step": 4722,
      "training_step_time": 17.07390594482422
    },
    {
      "epoch": 4.504203796386719e-06,
      "model_forward_time": 5.598113536834717,
      "step": 4723
    },
    {
      "epoch": 4.504203796386719e-06,
      "step": 4723,
      "training_step_time": 17.05941104888916
    },
    {
      "epoch": 4.505157470703125e-06,
      "model_forward_time": 5.593736171722412,
      "step": 4724
    },
    {
      "epoch": 4.505157470703125e-06,
      "step": 4724,
      "training_step_time": 17.051445722579956
    },
    {
      "epoch": 4.506111145019531e-06,
      "model_forward_time": 5.59040093421936,
      "step": 4725
    },
    {
      "epoch": 4.506111145019531e-06,
      "step": 4725,
      "training_step_time": 17.020457983016968
    },
    {
      "epoch": 4.5070648193359375e-06,
      "model_forward_time": 5.584951162338257,
      "step": 4726
    },
    {
      "epoch": 4.5070648193359375e-06,
      "step": 4726,
      "training_step_time": 17.410053730010986
    },
    {
      "epoch": 4.508018493652344e-06,
      "model_forward_time": 5.598876237869263,
      "step": 4727
    },
    {
      "epoch": 4.508018493652344e-06,
      "step": 4727,
      "training_step_time": 17.100940465927124
    },
    {
      "epoch": 4.50897216796875e-06,
      "model_forward_time": 5.608621597290039,
      "step": 4728
    },
    {
      "epoch": 4.50897216796875e-06,
      "step": 4728,
      "training_step_time": 17.08488965034485
    },
    {
      "epoch": 4.509925842285156e-06,
      "model_forward_time": 5.593465089797974,
      "step": 4729
    },
    {
      "epoch": 4.509925842285156e-06,
      "step": 4729,
      "training_step_time": 17.04981517791748
    },
    {
      "epoch": 4.510879516601563e-06,
      "grad_norm": 1.1751655340194702,
      "learning_rate": 1e-05,
      "loss": 0.0802,
      "step": 4730
    },
    {
      "epoch": 4.510879516601563e-06,
      "model_forward_time": 5.591246604919434,
      "step": 4730
    },
    {
      "dynamics_loss_avg": 0.06326510664075613,
      "epoch": 4.510879516601563e-06,
      "step": 4730
    },
    {
      "action_loss_avg": 0.016711630439385772,
      "epoch": 4.510879516601563e-06,
      "step": 4730
    },
    {
      "epoch": 4.510879516601563e-06,
      "step": 4730,
      "training_step_time": 17.034652948379517
    },
    {
      "epoch": 4.511833190917969e-06,
      "model_forward_time": 5.607303619384766,
      "step": 4731
    },
    {
      "epoch": 4.511833190917969e-06,
      "step": 4731,
      "training_step_time": 17.052627563476562
    },
    {
      "epoch": 4.512786865234375e-06,
      "model_forward_time": 5.95722222328186,
      "step": 4732
    },
    {
      "epoch": 4.512786865234375e-06,
      "step": 4732,
      "training_step_time": 17.411155223846436
    },
    {
      "epoch": 4.513740539550781e-06,
      "model_forward_time": 5.602771282196045,
      "step": 4733
    },
    {
      "epoch": 4.513740539550781e-06,
      "step": 4733,
      "training_step_time": 17.08761715888977
    },
    {
      "epoch": 4.5146942138671875e-06,
      "model_forward_time": 5.618795394897461,
      "step": 4734
    },
    {
      "epoch": 4.5146942138671875e-06,
      "step": 4734,
      "training_step_time": 17.13716197013855
    },
    {
      "epoch": 4.515647888183594e-06,
      "model_forward_time": 5.614936590194702,
      "step": 4735
    },
    {
      "epoch": 4.515647888183594e-06,
      "step": 4735,
      "training_step_time": 17.119250774383545
    },
    {
      "epoch": 4.5166015625e-06,
      "model_forward_time": 5.607353448867798,
      "step": 4736
    },
    {
      "epoch": 4.5166015625e-06,
      "step": 4736,
      "training_step_time": 17.094648361206055
    },
    {
      "epoch": 4.517555236816406e-06,
      "model_forward_time": 5.96149754524231,
      "step": 4737
    },
    {
      "epoch": 4.517555236816406e-06,
      "step": 4737,
      "training_step_time": 17.442149877548218
    },
    {
      "epoch": 4.518508911132813e-06,
      "model_forward_time": 5.609280347824097,
      "step": 4738
    },
    {
      "epoch": 4.518508911132813e-06,
      "step": 4738,
      "training_step_time": 17.073437690734863
    },
    {
      "epoch": 4.519462585449219e-06,
      "model_forward_time": 5.593357563018799,
      "step": 4739
    },
    {
      "epoch": 4.519462585449219e-06,
      "step": 4739,
      "training_step_time": 17.046032190322876
    },
    {
      "epoch": 4.520416259765625e-06,
      "grad_norm": 2.622532844543457,
      "learning_rate": 1e-05,
      "loss": 0.0694,
      "step": 4740
    },
    {
      "epoch": 4.520416259765625e-06,
      "model_forward_time": 5.593411684036255,
      "step": 4740
    },
    {
      "dynamics_loss_avg": 0.05623738225549459,
      "epoch": 4.520416259765625e-06,
      "step": 4740
    },
    {
      "action_loss_avg": 0.01317109630908817,
      "epoch": 4.520416259765625e-06,
      "step": 4740
    },
    {
      "epoch": 4.520416259765625e-06,
      "step": 4740,
      "training_step_time": 17.02541756629944
    },
    {
      "epoch": 4.521369934082031e-06,
      "model_forward_time": 5.585423707962036,
      "step": 4741
    },
    {
      "epoch": 4.521369934082031e-06,
      "step": 4741,
      "training_step_time": 17.035595893859863
    },
    {
      "epoch": 4.5223236083984375e-06,
      "model_forward_time": 5.597129583358765,
      "step": 4742
    },
    {
      "epoch": 4.5223236083984375e-06,
      "step": 4742,
      "training_step_time": 17.050813913345337
    },
    {
      "epoch": 4.523277282714844e-06,
      "model_forward_time": 5.968013525009155,
      "step": 4743
    },
    {
      "epoch": 4.523277282714844e-06,
      "step": 4743,
      "training_step_time": 17.441882371902466
    },
    {
      "epoch": 4.52423095703125e-06,
      "model_forward_time": 5.612788915634155,
      "step": 4744
    },
    {
      "epoch": 4.52423095703125e-06,
      "step": 4744,
      "training_step_time": 17.11195158958435
    },
    {
      "epoch": 4.525184631347656e-06,
      "model_forward_time": 5.604994535446167,
      "step": 4745
    },
    {
      "epoch": 4.525184631347656e-06,
      "step": 4745,
      "training_step_time": 17.084638118743896
    },
    {
      "epoch": 4.526138305664063e-06,
      "model_forward_time": 5.606340646743774,
      "step": 4746
    },
    {
      "epoch": 4.526138305664063e-06,
      "step": 4746,
      "training_step_time": 17.087700843811035
    },
    {
      "epoch": 4.527091979980469e-06,
      "model_forward_time": 5.624607086181641,
      "step": 4747
    },
    {
      "epoch": 4.527091979980469e-06,
      "step": 4747,
      "training_step_time": 17.124046802520752
    },
    {
      "epoch": 4.528045654296875e-06,
      "model_forward_time": 5.644402503967285,
      "step": 4748
    },
    {
      "epoch": 4.528045654296875e-06,
      "step": 4748,
      "training_step_time": 17.49389100074768
    },
    {
      "epoch": 4.528999328613281e-06,
      "model_forward_time": 5.598286867141724,
      "step": 4749
    },
    {
      "epoch": 4.528999328613281e-06,
      "step": 4749,
      "training_step_time": 17.0296471118927
    },
    {
      "epoch": 4.5299530029296875e-06,
      "grad_norm": 1.4675617218017578,
      "learning_rate": 1e-05,
      "loss": 0.0791,
      "step": 4750
    },
    {
      "epoch": 4.5299530029296875e-06,
      "model_forward_time": 5.5813586711883545,
      "step": 4750
    },
    {
      "dynamics_loss_avg": 0.06466200863942503,
      "epoch": 4.5299530029296875e-06,
      "step": 4750
    },
    {
      "action_loss_avg": 0.01336283665150404,
      "epoch": 4.5299530029296875e-06,
      "step": 4750
    },
    {
      "epoch": 4.5299530029296875e-06,
      "step": 4750,
      "training_step_time": 17.036699295043945
    },
    {
      "epoch": 4.530906677246094e-06,
      "model_forward_time": 5.6015424728393555,
      "step": 4751
    },
    {
      "epoch": 4.530906677246094e-06,
      "step": 4751,
      "training_step_time": 17.09934973716736
    },
    {
      "epoch": 4.5318603515625e-06,
      "model_forward_time": 5.612713575363159,
      "step": 4752
    },
    {
      "epoch": 4.5318603515625e-06,
      "step": 4752,
      "training_step_time": 17.10332155227661
    },
    {
      "epoch": 4.532814025878906e-06,
      "model_forward_time": 5.60456657409668,
      "step": 4753
    },
    {
      "epoch": 4.532814025878906e-06,
      "step": 4753,
      "training_step_time": 17.061517000198364
    },
    {
      "epoch": 4.533767700195312e-06,
      "model_forward_time": 5.9402759075164795,
      "step": 4754
    },
    {
      "epoch": 4.533767700195312e-06,
      "step": 4754,
      "training_step_time": 17.377715826034546
    },
    {
      "epoch": 4.534721374511719e-06,
      "model_forward_time": 5.579425573348999,
      "step": 4755
    },
    {
      "epoch": 4.534721374511719e-06,
      "step": 4755,
      "training_step_time": 16.991896867752075
    },
    {
      "epoch": 4.535675048828125e-06,
      "model_forward_time": 5.576961517333984,
      "step": 4756
    },
    {
      "epoch": 4.535675048828125e-06,
      "step": 4756,
      "training_step_time": 16.99480152130127
    },
    {
      "epoch": 4.536628723144531e-06,
      "model_forward_time": 5.566015720367432,
      "step": 4757
    },
    {
      "epoch": 4.536628723144531e-06,
      "step": 4757,
      "training_step_time": 16.999035835266113
    },
    {
      "epoch": 4.5375823974609375e-06,
      "model_forward_time": 5.5693199634552,
      "step": 4758
    },
    {
      "epoch": 4.5375823974609375e-06,
      "step": 4758,
      "training_step_time": 16.983576774597168
    },
    {
      "epoch": 4.538536071777344e-06,
      "model_forward_time": 5.578332901000977,
      "step": 4759
    },
    {
      "epoch": 4.538536071777344e-06,
      "step": 4759,
      "training_step_time": 17.36639976501465
    },
    {
      "epoch": 4.53948974609375e-06,
      "grad_norm": 1.2383098602294922,
      "learning_rate": 1e-05,
      "loss": 0.0537,
      "step": 4760
    },
    {
      "epoch": 4.53948974609375e-06,
      "model_forward_time": 5.579313516616821,
      "step": 4760
    },
    {
      "dynamics_loss_avg": 0.04122291561216116,
      "epoch": 4.53948974609375e-06,
      "step": 4760
    },
    {
      "action_loss_avg": 0.011451323702931403,
      "epoch": 4.53948974609375e-06,
      "step": 4760
    },
    {
      "epoch": 4.53948974609375e-06,
      "step": 4760,
      "training_step_time": 17.00292158126831
    },
    {
      "epoch": 4.540443420410156e-06,
      "model_forward_time": 5.580784320831299,
      "step": 4761
    },
    {
      "epoch": 4.540443420410156e-06,
      "step": 4761,
      "training_step_time": 17.006664514541626
    },
    {
      "epoch": 4.541397094726562e-06,
      "model_forward_time": 5.60104775428772,
      "step": 4762
    },
    {
      "epoch": 4.541397094726562e-06,
      "step": 4762,
      "training_step_time": 17.08117151260376
    },
    {
      "epoch": 4.542350769042969e-06,
      "model_forward_time": 5.608269929885864,
      "step": 4763
    },
    {
      "epoch": 4.542350769042969e-06,
      "step": 4763,
      "training_step_time": 17.08040976524353
    },
    {
      "epoch": 4.543304443359375e-06,
      "model_forward_time": 5.59700083732605,
      "step": 4764
    },
    {
      "epoch": 4.543304443359375e-06,
      "step": 4764,
      "training_step_time": 17.05724263191223
    },
    {
      "epoch": 4.544258117675781e-06,
      "model_forward_time": 5.963224172592163,
      "step": 4765
    },
    {
      "epoch": 4.544258117675781e-06,
      "step": 4765,
      "training_step_time": 17.4338219165802
    },
    {
      "epoch": 4.5452117919921875e-06,
      "model_forward_time": 5.591986417770386,
      "step": 4766
    },
    {
      "epoch": 4.5452117919921875e-06,
      "step": 4766,
      "training_step_time": 17.040446281433105
    },
    {
      "epoch": 4.546165466308594e-06,
      "model_forward_time": 5.597338914871216,
      "step": 4767
    },
    {
      "epoch": 4.546165466308594e-06,
      "step": 4767,
      "training_step_time": 17.060182571411133
    },
    {
      "epoch": 4.547119140625e-06,
      "model_forward_time": 5.596619129180908,
      "step": 4768
    },
    {
      "epoch": 4.547119140625e-06,
      "step": 4768,
      "training_step_time": 17.064604997634888
    },
    {
      "epoch": 4.548072814941406e-06,
      "model_forward_time": 5.608941078186035,
      "step": 4769
    },
    {
      "epoch": 4.548072814941406e-06,
      "step": 4769,
      "training_step_time": 17.06240725517273
    },
    {
      "epoch": 4.549026489257812e-06,
      "grad_norm": 1.3070025444030762,
      "learning_rate": 1e-05,
      "loss": 0.0774,
      "step": 4770
    },
    {
      "epoch": 4.549026489257812e-06,
      "model_forward_time": 5.965510129928589,
      "step": 4770
    },
    {
      "dynamics_loss_avg": 0.06595511473715306,
      "epoch": 4.549026489257812e-06,
      "step": 4770
    },
    {
      "action_loss_avg": 0.01948466831818223,
      "epoch": 4.549026489257812e-06,
      "step": 4770
    },
    {
      "epoch": 4.549026489257812e-06,
      "step": 4770,
      "training_step_time": 17.43535590171814
    },
    {
      "epoch": 4.549980163574219e-06,
      "model_forward_time": 5.598741769790649,
      "step": 4771
    },
    {
      "epoch": 4.549980163574219e-06,
      "step": 4771,
      "training_step_time": 17.051372528076172
    },
    {
      "epoch": 4.550933837890625e-06,
      "model_forward_time": 5.591736793518066,
      "step": 4772
    },
    {
      "epoch": 4.550933837890625e-06,
      "step": 4772,
      "training_step_time": 17.034038543701172
    },
    {
      "epoch": 4.551887512207031e-06,
      "model_forward_time": 5.5829994678497314,
      "step": 4773
    },
    {
      "epoch": 4.551887512207031e-06,
      "step": 4773,
      "training_step_time": 17.03801393508911
    },
    {
      "epoch": 4.5528411865234375e-06,
      "model_forward_time": 5.590565919876099,
      "step": 4774
    },
    {
      "epoch": 4.5528411865234375e-06,
      "step": 4774,
      "training_step_time": 17.053109884262085
    },
    {
      "epoch": 4.553794860839844e-06,
      "model_forward_time": 5.598711013793945,
      "step": 4775
    },
    {
      "epoch": 4.553794860839844e-06,
      "step": 4775,
      "training_step_time": 17.066266536712646
    },
    {
      "epoch": 4.55474853515625e-06,
      "model_forward_time": 5.977988958358765,
      "step": 4776
    },
    {
      "epoch": 4.55474853515625e-06,
      "step": 4776,
      "training_step_time": 17.436031579971313
    },
    {
      "epoch": 4.555702209472656e-06,
      "model_forward_time": 5.60391092300415,
      "step": 4777
    },
    {
      "epoch": 4.555702209472656e-06,
      "step": 4777,
      "training_step_time": 17.077720880508423
    },
    {
      "epoch": 4.556655883789062e-06,
      "model_forward_time": 5.595441579818726,
      "step": 4778
    },
    {
      "epoch": 4.556655883789062e-06,
      "step": 4778,
      "training_step_time": 17.066197633743286
    },
    {
      "epoch": 4.557609558105469e-06,
      "model_forward_time": 5.608481168746948,
      "step": 4779
    },
    {
      "epoch": 4.557609558105469e-06,
      "step": 4779,
      "training_step_time": 17.085132122039795
    },
    {
      "epoch": 4.558563232421875e-06,
      "grad_norm": 1.1875669956207275,
      "learning_rate": 1e-05,
      "loss": 0.0745,
      "step": 4780
    },
    {
      "epoch": 4.558563232421875e-06,
      "model_forward_time": 5.602154493331909,
      "step": 4780
    },
    {
      "dynamics_loss_avg": 0.054304996877908705,
      "epoch": 4.558563232421875e-06,
      "step": 4780
    },
    {
      "action_loss_avg": 0.013868287857621907,
      "epoch": 4.558563232421875e-06,
      "step": 4780
    },
    {
      "epoch": 4.558563232421875e-06,
      "step": 4780,
      "training_step_time": 17.06272315979004
    },
    {
      "epoch": 4.559516906738281e-06,
      "model_forward_time": 5.593274116516113,
      "step": 4781
    },
    {
      "epoch": 4.559516906738281e-06,
      "step": 4781,
      "training_step_time": 17.439684867858887
    },
    {
      "epoch": 4.5604705810546875e-06,
      "model_forward_time": 5.599588632583618,
      "step": 4782
    },
    {
      "epoch": 4.5604705810546875e-06,
      "step": 4782,
      "training_step_time": 17.062061071395874
    },
    {
      "epoch": 4.561424255371094e-06,
      "model_forward_time": 5.604040622711182,
      "step": 4783
    },
    {
      "epoch": 4.561424255371094e-06,
      "step": 4783,
      "training_step_time": 17.0598304271698
    },
    {
      "epoch": 4.5623779296875e-06,
      "model_forward_time": 5.58553671836853,
      "step": 4784
    },
    {
      "epoch": 4.5623779296875e-06,
      "step": 4784,
      "training_step_time": 17.026318550109863
    },
    {
      "epoch": 4.563331604003906e-06,
      "model_forward_time": 5.596968173980713,
      "step": 4785
    },
    {
      "epoch": 4.563331604003906e-06,
      "step": 4785,
      "training_step_time": 17.037643671035767
    },
    {
      "epoch": 4.564285278320312e-06,
      "model_forward_time": 5.587955713272095,
      "step": 4786
    },
    {
      "epoch": 4.564285278320312e-06,
      "step": 4786,
      "training_step_time": 17.064950704574585
    },
    {
      "epoch": 4.565238952636719e-06,
      "model_forward_time": 5.9695885181427,
      "step": 4787
    },
    {
      "epoch": 4.565238952636719e-06,
      "step": 4787,
      "training_step_time": 17.46906542778015
    },
    {
      "epoch": 4.566192626953125e-06,
      "model_forward_time": 5.616738557815552,
      "step": 4788
    },
    {
      "epoch": 4.566192626953125e-06,
      "step": 4788,
      "training_step_time": 17.08116340637207
    },
    {
      "epoch": 4.567146301269531e-06,
      "model_forward_time": 5.5875773429870605,
      "step": 4789
    },
    {
      "epoch": 4.567146301269531e-06,
      "step": 4789,
      "training_step_time": 17.02267837524414
    },
    {
      "epoch": 4.5680999755859375e-06,
      "grad_norm": 0.4396831691265106,
      "learning_rate": 1e-05,
      "loss": 0.0802,
      "step": 4790
    },
    {
      "epoch": 4.5680999755859375e-06,
      "model_forward_time": 5.583578109741211,
      "step": 4790
    },
    {
      "dynamics_loss_avg": 0.06649681851267815,
      "epoch": 4.5680999755859375e-06,
      "step": 4790
    },
    {
      "action_loss_avg": 0.013297087885439396,
      "epoch": 4.5680999755859375e-06,
      "step": 4790
    },
    {
      "epoch": 4.5680999755859375e-06,
      "step": 4790,
      "training_step_time": 17.006741762161255
    },
    {
      "epoch": 4.569053649902344e-06,
      "model_forward_time": 5.572555303573608,
      "step": 4791
    },
    {
      "epoch": 4.569053649902344e-06,
      "step": 4791,
      "training_step_time": 17.004615306854248
    },
    {
      "epoch": 4.57000732421875e-06,
      "model_forward_time": 5.943535804748535,
      "step": 4792
    },
    {
      "epoch": 4.57000732421875e-06,
      "step": 4792,
      "training_step_time": 17.37007999420166
    },
    {
      "epoch": 4.570960998535156e-06,
      "model_forward_time": 5.581915616989136,
      "step": 4793
    },
    {
      "epoch": 4.570960998535156e-06,
      "step": 4793,
      "training_step_time": 17.030959606170654
    },
    {
      "epoch": 4.571914672851562e-06,
      "model_forward_time": 5.599891662597656,
      "step": 4794
    },
    {
      "epoch": 4.571914672851562e-06,
      "step": 4794,
      "training_step_time": 17.074159622192383
    },
    {
      "epoch": 4.572868347167969e-06,
      "model_forward_time": 5.604101181030273,
      "step": 4795
    },
    {
      "epoch": 4.572868347167969e-06,
      "step": 4795,
      "training_step_time": 17.071539163589478
    },
    {
      "epoch": 4.573822021484375e-06,
      "model_forward_time": 5.596651792526245,
      "step": 4796
    },
    {
      "epoch": 4.573822021484375e-06,
      "step": 4796,
      "training_step_time": 17.054628133773804
    },
    {
      "epoch": 4.574775695800781e-06,
      "model_forward_time": 5.591447591781616,
      "step": 4797
    },
    {
      "epoch": 4.574775695800781e-06,
      "step": 4797,
      "training_step_time": 17.051661491394043
    },
    {
      "epoch": 4.5757293701171875e-06,
      "model_forward_time": 5.979002237319946,
      "step": 4798
    },
    {
      "epoch": 4.5757293701171875e-06,
      "step": 4798,
      "training_step_time": 17.4525146484375
    },
    {
      "epoch": 4.576683044433594e-06,
      "model_forward_time": 5.598641872406006,
      "step": 4799
    },
    {
      "epoch": 4.576683044433594e-06,
      "step": 4799,
      "training_step_time": 17.059645414352417
    },
    {
      "epoch": 4.57763671875e-06,
      "grad_norm": 0.8644358515739441,
      "learning_rate": 1e-05,
      "loss": 0.0555,
      "step": 4800
    },
    {
      "epoch": 4.57763671875e-06,
      "model_forward_time": 5.601144790649414,
      "step": 4800
    },
    {
      "dynamics_loss_avg": 0.04405174162238836,
      "epoch": 4.57763671875e-06,
      "step": 4800
    },
    {
      "action_loss_avg": 0.01283294390887022,
      "epoch": 4.57763671875e-06,
      "step": 4800
    },
    {
      "epoch": 4.57763671875e-06,
      "step": 4800,
      "training_step_time": 17.07013750076294
    },
    {
      "epoch": 4.578590393066406e-06,
      "model_forward_time": 5.604662179946899,
      "step": 4801
    },
    {
      "epoch": 4.578590393066406e-06,
      "step": 4801,
      "training_step_time": 17.063252449035645
    },
    {
      "epoch": 4.579544067382812e-06,
      "model_forward_time": 5.595069408416748,
      "step": 4802
    },
    {
      "epoch": 4.579544067382812e-06,
      "step": 4802,
      "training_step_time": 17.03984022140503
    },
    {
      "epoch": 4.580497741699219e-06,
      "model_forward_time": 5.947031259536743,
      "step": 4803
    },
    {
      "epoch": 4.580497741699219e-06,
      "step": 4803,
      "training_step_time": 17.38798451423645
    },
    {
      "epoch": 4.581451416015625e-06,
      "model_forward_time": 5.589046001434326,
      "step": 4804
    },
    {
      "epoch": 4.581451416015625e-06,
      "step": 4804,
      "training_step_time": 17.039889335632324
    },
    {
      "epoch": 4.582405090332031e-06,
      "model_forward_time": 5.599064588546753,
      "step": 4805
    },
    {
      "epoch": 4.582405090332031e-06,
      "step": 4805,
      "training_step_time": 17.079793691635132
    },
    {
      "epoch": 4.5833587646484375e-06,
      "model_forward_time": 5.599649667739868,
      "step": 4806
    },
    {
      "epoch": 4.5833587646484375e-06,
      "step": 4806,
      "training_step_time": 17.07164740562439
    },
    {
      "epoch": 4.584312438964844e-06,
      "model_forward_time": 5.602064371109009,
      "step": 4807
    },
    {
      "epoch": 4.584312438964844e-06,
      "step": 4807,
      "training_step_time": 17.052473545074463
    },
    {
      "epoch": 4.58526611328125e-06,
      "model_forward_time": 5.584688663482666,
      "step": 4808
    },
    {
      "epoch": 4.58526611328125e-06,
      "step": 4808,
      "training_step_time": 17.372349977493286
    },
    {
      "epoch": 4.586219787597656e-06,
      "model_forward_time": 5.594738483428955,
      "step": 4809
    },
    {
      "epoch": 4.586219787597656e-06,
      "step": 4809,
      "training_step_time": 17.044779300689697
    },
    {
      "epoch": 4.587173461914062e-06,
      "grad_norm": 0.4162711203098297,
      "learning_rate": 1e-05,
      "loss": 0.0752,
      "step": 4810
    },
    {
      "epoch": 4.587173461914062e-06,
      "model_forward_time": 5.59894585609436,
      "step": 4810
    },
    {
      "dynamics_loss_avg": 0.059418863244354726,
      "epoch": 4.587173461914062e-06,
      "step": 4810
    },
    {
      "action_loss_avg": 0.014436568645760418,
      "epoch": 4.587173461914062e-06,
      "step": 4810
    },
    {
      "epoch": 4.587173461914062e-06,
      "step": 4810,
      "training_step_time": 17.068307876586914
    },
    {
      "epoch": 4.588127136230469e-06,
      "model_forward_time": 5.605483770370483,
      "step": 4811
    },
    {
      "epoch": 4.588127136230469e-06,
      "step": 4811,
      "training_step_time": 17.09386157989502
    },
    {
      "epoch": 4.589080810546875e-06,
      "model_forward_time": 5.632185459136963,
      "step": 4812
    },
    {
      "epoch": 4.589080810546875e-06,
      "step": 4812,
      "training_step_time": 17.142157793045044
    },
    {
      "epoch": 4.590034484863281e-06,
      "model_forward_time": 5.613522291183472,
      "step": 4813
    },
    {
      "epoch": 4.590034484863281e-06,
      "step": 4813,
      "training_step_time": 17.113221883773804
    },
    {
      "epoch": 4.5909881591796876e-06,
      "model_forward_time": 5.965904235839844,
      "step": 4814
    },
    {
      "epoch": 4.5909881591796876e-06,
      "step": 4814,
      "training_step_time": 17.455816507339478
    },
    {
      "epoch": 4.591941833496094e-06,
      "model_forward_time": 5.611496210098267,
      "step": 4815
    },
    {
      "epoch": 4.591941833496094e-06,
      "step": 4815,
      "training_step_time": 17.09778356552124
    },
    {
      "epoch": 4.5928955078125e-06,
      "model_forward_time": 5.619596242904663,
      "step": 4816
    },
    {
      "epoch": 4.5928955078125e-06,
      "step": 4816,
      "training_step_time": 17.10023784637451
    },
    {
      "epoch": 4.593849182128906e-06,
      "model_forward_time": 5.591611385345459,
      "step": 4817
    },
    {
      "epoch": 4.593849182128906e-06,
      "step": 4817,
      "training_step_time": 17.028209686279297
    },
    {
      "epoch": 4.594802856445312e-06,
      "model_forward_time": 5.583963632583618,
      "step": 4818
    },
    {
      "epoch": 4.594802856445312e-06,
      "step": 4818,
      "training_step_time": 17.023951768875122
    },
    {
      "epoch": 4.595756530761719e-06,
      "model_forward_time": 5.588807582855225,
      "step": 4819
    },
    {
      "epoch": 4.595756530761719e-06,
      "step": 4819,
      "training_step_time": 17.393877029418945
    },
    {
      "epoch": 4.596710205078125e-06,
      "grad_norm": 1.1259446144104004,
      "learning_rate": 1e-05,
      "loss": 0.065,
      "step": 4820
    },
    {
      "epoch": 4.596710205078125e-06,
      "model_forward_time": 5.587916135787964,
      "step": 4820
    },
    {
      "dynamics_loss_avg": 0.05180875305086374,
      "epoch": 4.596710205078125e-06,
      "step": 4820
    },
    {
      "action_loss_avg": 0.013647024333477021,
      "epoch": 4.596710205078125e-06,
      "step": 4820
    },
    {
      "epoch": 4.596710205078125e-06,
      "step": 4820,
      "training_step_time": 17.026317358016968
    },
    {
      "epoch": 4.597663879394531e-06,
      "model_forward_time": 5.597706317901611,
      "step": 4821
    },
    {
      "epoch": 4.597663879394531e-06,
      "step": 4821,
      "training_step_time": 17.063485383987427
    },
    {
      "epoch": 4.5986175537109376e-06,
      "model_forward_time": 5.6060240268707275,
      "step": 4822
    },
    {
      "epoch": 4.5986175537109376e-06,
      "step": 4822,
      "training_step_time": 17.050183296203613
    },
    {
      "epoch": 4.599571228027344e-06,
      "model_forward_time": 5.595054626464844,
      "step": 4823
    },
    {
      "epoch": 4.599571228027344e-06,
      "step": 4823,
      "training_step_time": 17.04128646850586
    },
    {
      "epoch": 4.60052490234375e-06,
      "model_forward_time": 5.58828067779541,
      "step": 4824
    },
    {
      "epoch": 4.60052490234375e-06,
      "step": 4824,
      "training_step_time": 17.01621723175049
    },
    {
      "epoch": 4.601478576660156e-06,
      "model_forward_time": 5.931240558624268,
      "step": 4825
    },
    {
      "epoch": 4.601478576660156e-06,
      "step": 4825,
      "training_step_time": 17.404440879821777
    },
    {
      "epoch": 4.602432250976562e-06,
      "model_forward_time": 5.616062641143799,
      "step": 4826
    },
    {
      "epoch": 4.602432250976562e-06,
      "step": 4826,
      "training_step_time": 17.11094045639038
    },
    {
      "epoch": 4.603385925292969e-06,
      "model_forward_time": 5.625070095062256,
      "step": 4827
    },
    {
      "epoch": 4.603385925292969e-06,
      "step": 4827,
      "training_step_time": 17.10524868965149
    },
    {
      "epoch": 4.604339599609375e-06,
      "model_forward_time": 5.59520697593689,
      "step": 4828
    },
    {
      "epoch": 4.604339599609375e-06,
      "step": 4828,
      "training_step_time": 17.019338846206665
    },
    {
      "epoch": 4.605293273925781e-06,
      "model_forward_time": 5.5834550857543945,
      "step": 4829
    },
    {
      "epoch": 4.605293273925781e-06,
      "step": 4829,
      "training_step_time": 16.99759030342102
    },
    {
      "epoch": 4.6062469482421876e-06,
      "grad_norm": 0.8970654606819153,
      "learning_rate": 1e-05,
      "loss": 0.0747,
      "step": 4830
    },
    {
      "epoch": 4.6062469482421876e-06,
      "model_forward_time": 5.93794059753418,
      "step": 4830
    },
    {
      "dynamics_loss_avg": 0.05873066876083612,
      "epoch": 4.6062469482421876e-06,
      "step": 4830
    },
    {
      "action_loss_avg": 0.014973962120711804,
      "epoch": 4.6062469482421876e-06,
      "step": 4830
    },
    {
      "epoch": 4.6062469482421876e-06,
      "step": 4830,
      "training_step_time": 17.38034701347351
    },
    {
      "epoch": 4.607200622558594e-06,
      "model_forward_time": 5.5744147300720215,
      "step": 4831
    },
    {
      "epoch": 4.607200622558594e-06,
      "step": 4831,
      "training_step_time": 16.98029851913452
    },
    {
      "epoch": 4.608154296875e-06,
      "model_forward_time": 5.577041149139404,
      "step": 4832
    },
    {
      "epoch": 4.608154296875e-06,
      "step": 4832,
      "training_step_time": 17.001372575759888
    },
    {
      "epoch": 4.6091079711914064e-06,
      "model_forward_time": 5.588646173477173,
      "step": 4833
    },
    {
      "epoch": 4.6091079711914064e-06,
      "step": 4833,
      "training_step_time": 17.021496534347534
    },
    {
      "epoch": 4.610061645507812e-06,
      "model_forward_time": 5.575517654418945,
      "step": 4834
    },
    {
      "epoch": 4.610061645507812e-06,
      "step": 4834,
      "training_step_time": 17.025002002716064
    },
    {
      "epoch": 4.611015319824219e-06,
      "model_forward_time": 5.5908589363098145,
      "step": 4835
    },
    {
      "epoch": 4.611015319824219e-06,
      "step": 4835,
      "training_step_time": 17.3914897441864
    },
    {
      "epoch": 4.611968994140625e-06,
      "model_forward_time": 5.594709873199463,
      "step": 4836
    },
    {
      "epoch": 4.611968994140625e-06,
      "step": 4836,
      "training_step_time": 17.05846357345581
    },
    {
      "epoch": 4.612922668457031e-06,
      "model_forward_time": 5.602457761764526,
      "step": 4837
    },
    {
      "epoch": 4.612922668457031e-06,
      "step": 4837,
      "training_step_time": 17.06652545928955
    },
    {
      "epoch": 4.6138763427734376e-06,
      "model_forward_time": 5.594160795211792,
      "step": 4838
    },
    {
      "epoch": 4.6138763427734376e-06,
      "step": 4838,
      "training_step_time": 17.057661294937134
    },
    {
      "epoch": 4.614830017089844e-06,
      "model_forward_time": 5.603773832321167,
      "step": 4839
    },
    {
      "epoch": 4.614830017089844e-06,
      "step": 4839,
      "training_step_time": 17.061601400375366
    },
    {
      "epoch": 4.61578369140625e-06,
      "grad_norm": 0.9790908098220825,
      "learning_rate": 1e-05,
      "loss": 0.086,
      "step": 4840
    },
    {
      "epoch": 4.61578369140625e-06,
      "model_forward_time": 5.604790687561035,
      "step": 4840
    },
    {
      "dynamics_loss_avg": 0.06714208601042629,
      "epoch": 4.61578369140625e-06,
      "step": 4840
    },
    {
      "action_loss_avg": 0.018809015350416302,
      "epoch": 4.61578369140625e-06,
      "step": 4840
    },
    {
      "epoch": 4.61578369140625e-06,
      "step": 4840,
      "training_step_time": 17.06826424598694
    },
    {
      "epoch": 4.6167373657226564e-06,
      "model_forward_time": 5.962574243545532,
      "step": 4841
    },
    {
      "epoch": 4.6167373657226564e-06,
      "step": 4841,
      "training_step_time": 17.444638967514038
    },
    {
      "epoch": 4.617691040039062e-06,
      "model_forward_time": 5.6069746017456055,
      "step": 4842
    },
    {
      "epoch": 4.617691040039062e-06,
      "step": 4842,
      "training_step_time": 17.103469133377075
    },
    {
      "epoch": 4.618644714355469e-06,
      "model_forward_time": 5.618039131164551,
      "step": 4843
    },
    {
      "epoch": 4.618644714355469e-06,
      "step": 4843,
      "training_step_time": 17.11669945716858
    },
    {
      "epoch": 4.619598388671875e-06,
      "model_forward_time": 5.599706649780273,
      "step": 4844
    },
    {
      "epoch": 4.619598388671875e-06,
      "step": 4844,
      "training_step_time": 17.081634044647217
    },
    {
      "epoch": 4.620552062988281e-06,
      "model_forward_time": 5.606712818145752,
      "step": 4845
    },
    {
      "epoch": 4.620552062988281e-06,
      "step": 4845,
      "training_step_time": 17.04303765296936
    },
    {
      "epoch": 4.621505737304688e-06,
      "model_forward_time": 5.942075967788696,
      "step": 4846
    },
    {
      "epoch": 4.621505737304688e-06,
      "step": 4846,
      "training_step_time": 17.380935430526733
    },
    {
      "epoch": 4.622459411621094e-06,
      "model_forward_time": 5.591659307479858,
      "step": 4847
    },
    {
      "epoch": 4.622459411621094e-06,
      "step": 4847,
      "training_step_time": 17.041480541229248
    },
    {
      "epoch": 4.6234130859375e-06,
      "model_forward_time": 5.586026906967163,
      "step": 4848
    },
    {
      "epoch": 4.6234130859375e-06,
      "step": 4848,
      "training_step_time": 17.011428356170654
    },
    {
      "epoch": 4.6243667602539064e-06,
      "model_forward_time": 5.582654237747192,
      "step": 4849
    },
    {
      "epoch": 4.6243667602539064e-06,
      "step": 4849,
      "training_step_time": 17.02574110031128
    },
    {
      "epoch": 4.625320434570312e-06,
      "grad_norm": 2.1880245208740234,
      "learning_rate": 1e-05,
      "loss": 0.0731,
      "step": 4850
    },
    {
      "epoch": 4.625320434570312e-06,
      "model_forward_time": 5.588136196136475,
      "step": 4850
    },
    {
      "dynamics_loss_avg": 0.05807301588356495,
      "epoch": 4.625320434570312e-06,
      "step": 4850
    },
    {
      "action_loss_avg": 0.015024251583963633,
      "epoch": 4.625320434570312e-06,
      "step": 4850
    },
    {
      "epoch": 4.625320434570312e-06,
      "step": 4850,
      "training_step_time": 17.055079221725464
    },
    {
      "epoch": 4.626274108886719e-06,
      "model_forward_time": 5.590919733047485,
      "step": 4851
    },
    {
      "epoch": 4.626274108886719e-06,
      "step": 4851,
      "training_step_time": 17.052407264709473
    },
    {
      "epoch": 4.627227783203125e-06,
      "model_forward_time": 5.974963665008545,
      "step": 4852
    },
    {
      "epoch": 4.627227783203125e-06,
      "step": 4852,
      "training_step_time": 17.46564555168152
    },
    {
      "epoch": 4.628181457519531e-06,
      "model_forward_time": 5.611526012420654,
      "step": 4853
    },
    {
      "epoch": 4.628181457519531e-06,
      "step": 4853,
      "training_step_time": 17.07247304916382
    },
    {
      "epoch": 4.629135131835938e-06,
      "model_forward_time": 5.59599232673645,
      "step": 4854
    },
    {
      "epoch": 4.629135131835938e-06,
      "step": 4854,
      "training_step_time": 17.01322317123413
    },
    {
      "epoch": 4.630088806152344e-06,
      "model_forward_time": 5.577409505844116,
      "step": 4855
    },
    {
      "epoch": 4.630088806152344e-06,
      "step": 4855,
      "training_step_time": 17.00381565093994
    },
    {
      "epoch": 4.63104248046875e-06,
      "model_forward_time": 5.580721616744995,
      "step": 4856
    },
    {
      "epoch": 4.63104248046875e-06,
      "step": 4856,
      "training_step_time": 17.01550602912903
    },
    {
      "epoch": 4.6319961547851564e-06,
      "model_forward_time": 5.922935724258423,
      "step": 4857
    },
    {
      "epoch": 4.6319961547851564e-06,
      "step": 4857,
      "training_step_time": 17.377716541290283
    },
    {
      "epoch": 4.632949829101562e-06,
      "model_forward_time": 5.601989269256592,
      "step": 4858
    },
    {
      "epoch": 4.632949829101562e-06,
      "step": 4858,
      "training_step_time": 17.086204767227173
    },
    {
      "epoch": 4.633903503417969e-06,
      "model_forward_time": 5.606194257736206,
      "step": 4859
    },
    {
      "epoch": 4.633903503417969e-06,
      "step": 4859,
      "training_step_time": 17.09641695022583
    },
    {
      "epoch": 4.634857177734375e-06,
      "grad_norm": 0.577751636505127,
      "learning_rate": 1e-05,
      "loss": 0.0579,
      "step": 4860
    },
    {
      "epoch": 4.634857177734375e-06,
      "model_forward_time": 5.6101415157318115,
      "step": 4860
    },
    {
      "dynamics_loss_avg": 0.04397284602746367,
      "epoch": 4.634857177734375e-06,
      "step": 4860
    },
    {
      "action_loss_avg": 0.011636524990899488,
      "epoch": 4.634857177734375e-06,
      "step": 4860
    },
    {
      "epoch": 4.634857177734375e-06,
      "step": 4860,
      "training_step_time": 17.09562849998474
    },
    {
      "epoch": 4.635810852050781e-06,
      "model_forward_time": 5.613354206085205,
      "step": 4861
    },
    {
      "epoch": 4.635810852050781e-06,
      "step": 4861,
      "training_step_time": 17.116142511367798
    },
    {
      "epoch": 4.636764526367188e-06,
      "model_forward_time": 5.608023405075073,
      "step": 4862
    },
    {
      "epoch": 4.636764526367188e-06,
      "step": 4862,
      "training_step_time": 17.097909212112427
    },
    {
      "epoch": 4.637718200683594e-06,
      "model_forward_time": 5.970000743865967,
      "step": 4863
    },
    {
      "epoch": 4.637718200683594e-06,
      "step": 4863,
      "training_step_time": 17.45546793937683
    },
    {
      "epoch": 4.638671875e-06,
      "model_forward_time": 5.602869033813477,
      "step": 4864
    },
    {
      "epoch": 4.638671875e-06,
      "step": 4864,
      "training_step_time": 17.056798219680786
    },
    {
      "epoch": 4.6396255493164065e-06,
      "model_forward_time": 5.596928596496582,
      "step": 4865
    },
    {
      "epoch": 4.6396255493164065e-06,
      "step": 4865,
      "training_step_time": 17.066828966140747
    },
    {
      "epoch": 4.640579223632812e-06,
      "model_forward_time": 5.602246522903442,
      "step": 4866
    },
    {
      "epoch": 4.640579223632812e-06,
      "step": 4866,
      "training_step_time": 17.084468603134155
    },
    {
      "epoch": 4.641532897949219e-06,
      "model_forward_time": 5.603839159011841,
      "step": 4867
    },
    {
      "epoch": 4.641532897949219e-06,
      "step": 4867,
      "training_step_time": 17.05610752105713
    },
    {
      "epoch": 4.642486572265625e-06,
      "model_forward_time": 5.946774482727051,
      "step": 4868
    },
    {
      "epoch": 4.642486572265625e-06,
      "step": 4868,
      "training_step_time": 17.394908666610718
    },
    {
      "epoch": 4.643440246582031e-06,
      "model_forward_time": 5.591905832290649,
      "step": 4869
    },
    {
      "epoch": 4.643440246582031e-06,
      "step": 4869,
      "training_step_time": 17.041470527648926
    },
    {
      "epoch": 4.644393920898438e-06,
      "grad_norm": 1.6459249258041382,
      "learning_rate": 1e-05,
      "loss": 0.0733,
      "step": 4870
    },
    {
      "epoch": 4.644393920898438e-06,
      "model_forward_time": 5.589132785797119,
      "step": 4870
    },
    {
      "dynamics_loss_avg": 0.06417698375880718,
      "epoch": 4.644393920898438e-06,
      "step": 4870
    },
    {
      "action_loss_avg": 0.0152009024634026,
      "epoch": 4.644393920898438e-06,
      "step": 4870
    },
    {
      "epoch": 4.644393920898438e-06,
      "step": 4870,
      "training_step_time": 17.045180559158325
    },
    {
      "epoch": 4.645347595214844e-06,
      "model_forward_time": 5.588498115539551,
      "step": 4871
    },
    {
      "epoch": 4.645347595214844e-06,
      "step": 4871,
      "training_step_time": 17.024620532989502
    },
    {
      "epoch": 4.64630126953125e-06,
      "model_forward_time": 5.580471754074097,
      "step": 4872
    },
    {
      "epoch": 4.64630126953125e-06,
      "step": 4872,
      "training_step_time": 17.03006935119629
    },
    {
      "epoch": 4.6472549438476565e-06,
      "model_forward_time": 5.590961456298828,
      "step": 4873
    },
    {
      "epoch": 4.6472549438476565e-06,
      "step": 4873,
      "training_step_time": 17.391453981399536
    },
    {
      "epoch": 4.648208618164062e-06,
      "model_forward_time": 5.5970189571380615,
      "step": 4874
    },
    {
      "epoch": 4.648208618164062e-06,
      "step": 4874,
      "training_step_time": 17.058310747146606
    },
    {
      "epoch": 4.649162292480469e-06,
      "model_forward_time": 5.60618782043457,
      "step": 4875
    },
    {
      "epoch": 4.649162292480469e-06,
      "step": 4875,
      "training_step_time": 17.08995246887207
    },
    {
      "epoch": 4.650115966796875e-06,
      "model_forward_time": 5.609548568725586,
      "step": 4876
    },
    {
      "epoch": 4.650115966796875e-06,
      "step": 4876,
      "training_step_time": 17.08463144302368
    },
    {
      "epoch": 4.651069641113281e-06,
      "model_forward_time": 5.59208083152771,
      "step": 4877
    },
    {
      "epoch": 4.651069641113281e-06,
      "step": 4877,
      "training_step_time": 17.025623083114624
    },
    {
      "epoch": 4.652023315429688e-06,
      "model_forward_time": 5.574368476867676,
      "step": 4878
    },
    {
      "epoch": 4.652023315429688e-06,
      "step": 4878,
      "training_step_time": 16.98856496810913
    },
    {
      "epoch": 4.652976989746093e-06,
      "model_forward_time": 5.940598964691162,
      "step": 4879
    },
    {
      "epoch": 4.652976989746093e-06,
      "step": 4879,
      "training_step_time": 17.3668692111969
    },
    {
      "epoch": 4.6539306640625e-06,
      "grad_norm": 1.0899629592895508,
      "learning_rate": 1e-05,
      "loss": 0.0827,
      "step": 4880
    },
    {
      "epoch": 4.6539306640625e-06,
      "model_forward_time": 5.589104652404785,
      "step": 4880
    },
    {
      "dynamics_loss_avg": 0.06761252749711275,
      "epoch": 4.6539306640625e-06,
      "step": 4880
    },
    {
      "action_loss_avg": 0.016189712984487414,
      "epoch": 4.6539306640625e-06,
      "step": 4880
    },
    {
      "epoch": 4.6539306640625e-06,
      "step": 4880,
      "training_step_time": 17.037479877471924
    },
    {
      "epoch": 4.6548843383789065e-06,
      "model_forward_time": 5.602601766586304,
      "step": 4881
    },
    {
      "epoch": 4.6548843383789065e-06,
      "step": 4881,
      "training_step_time": 17.059894323349
    },
    {
      "epoch": 4.655838012695312e-06,
      "model_forward_time": 5.600944995880127,
      "step": 4882
    },
    {
      "epoch": 4.655838012695312e-06,
      "step": 4882,
      "training_step_time": 17.057915449142456
    },
    {
      "epoch": 4.656791687011719e-06,
      "model_forward_time": 5.594895362854004,
      "step": 4883
    },
    {
      "epoch": 4.656791687011719e-06,
      "step": 4883,
      "training_step_time": 17.052685260772705
    },
    {
      "epoch": 4.657745361328125e-06,
      "model_forward_time": 5.978151798248291,
      "step": 4884
    },
    {
      "epoch": 4.657745361328125e-06,
      "step": 4884,
      "training_step_time": 17.469200611114502
    },
    {
      "epoch": 4.658699035644531e-06,
      "model_forward_time": 5.620672225952148,
      "step": 4885
    },
    {
      "epoch": 4.658699035644531e-06,
      "step": 4885,
      "training_step_time": 17.1064715385437
    },
    {
      "epoch": 4.659652709960938e-06,
      "model_forward_time": 5.6054933071136475,
      "step": 4886
    },
    {
      "epoch": 4.659652709960938e-06,
      "step": 4886,
      "training_step_time": 17.080716371536255
    },
    {
      "epoch": 4.660606384277343e-06,
      "model_forward_time": 5.601698637008667,
      "step": 4887
    },
    {
      "epoch": 4.660606384277343e-06,
      "step": 4887,
      "training_step_time": 17.061516523361206
    },
    {
      "epoch": 4.66156005859375e-06,
      "model_forward_time": 5.594345808029175,
      "step": 4888
    },
    {
      "epoch": 4.66156005859375e-06,
      "step": 4888,
      "training_step_time": 17.00477886199951
    },
    {
      "epoch": 4.6625137329101565e-06,
      "model_forward_time": 5.5916428565979,
      "step": 4889
    },
    {
      "epoch": 4.6625137329101565e-06,
      "step": 4889,
      "training_step_time": 17.035929679870605
    },
    {
      "epoch": 4.663467407226562e-06,
      "grad_norm": 0.7075003385543823,
      "learning_rate": 1e-05,
      "loss": 0.0663,
      "step": 4890
    },
    {
      "epoch": 4.663467407226562e-06,
      "model_forward_time": 5.944888114929199,
      "step": 4890
    },
    {
      "dynamics_loss_avg": 0.04750509727746248,
      "epoch": 4.663467407226562e-06,
      "step": 4890
    },
    {
      "action_loss_avg": 0.01097551096463576,
      "epoch": 4.663467407226562e-06,
      "step": 4890
    },
    {
      "epoch": 4.663467407226562e-06,
      "step": 4890,
      "training_step_time": 17.43587636947632
    },
    {
      "epoch": 4.664421081542969e-06,
      "model_forward_time": 5.596450567245483,
      "step": 4891
    },
    {
      "epoch": 4.664421081542969e-06,
      "step": 4891,
      "training_step_time": 17.037418603897095
    },
    {
      "epoch": 4.665374755859375e-06,
      "model_forward_time": 5.597302436828613,
      "step": 4892
    },
    {
      "epoch": 4.665374755859375e-06,
      "step": 4892,
      "training_step_time": 17.077337503433228
    },
    {
      "epoch": 4.666328430175781e-06,
      "model_forward_time": 5.607269763946533,
      "step": 4893
    },
    {
      "epoch": 4.666328430175781e-06,
      "step": 4893,
      "training_step_time": 17.095940828323364
    },
    {
      "epoch": 4.667282104492188e-06,
      "model_forward_time": 5.606561183929443,
      "step": 4894
    },
    {
      "epoch": 4.667282104492188e-06,
      "step": 4894,
      "training_step_time": 17.07984948158264
    },
    {
      "epoch": 4.668235778808593e-06,
      "model_forward_time": 5.9861485958099365,
      "step": 4895
    },
    {
      "epoch": 4.668235778808593e-06,
      "step": 4895,
      "training_step_time": 17.483072757720947
    },
    {
      "epoch": 4.669189453125e-06,
      "model_forward_time": 5.612480878829956,
      "step": 4896
    },
    {
      "epoch": 4.669189453125e-06,
      "step": 4896,
      "training_step_time": 17.107767820358276
    },
    {
      "epoch": 4.6701431274414065e-06,
      "model_forward_time": 5.626652002334595,
      "step": 4897
    },
    {
      "epoch": 4.6701431274414065e-06,
      "step": 4897,
      "training_step_time": 17.1217360496521
    },
    {
      "epoch": 4.671096801757812e-06,
      "model_forward_time": 5.614489793777466,
      "step": 4898
    },
    {
      "epoch": 4.671096801757812e-06,
      "step": 4898,
      "training_step_time": 17.087708711624146
    },
    {
      "epoch": 4.672050476074219e-06,
      "model_forward_time": 5.600062608718872,
      "step": 4899
    },
    {
      "epoch": 4.672050476074219e-06,
      "step": 4899,
      "training_step_time": 17.07135510444641
    },
    {
      "epoch": 4.673004150390625e-06,
      "grad_norm": 2.348482370376587,
      "learning_rate": 1e-05,
      "loss": 0.0792,
      "step": 4900
    },
    {
      "epoch": 4.673004150390625e-06,
      "model_forward_time": 5.59552788734436,
      "step": 4900
    },
    {
      "dynamics_loss_avg": 0.06712561845779419,
      "epoch": 4.673004150390625e-06,
      "step": 4900
    },
    {
      "action_loss_avg": 0.01906875683926046,
      "epoch": 4.673004150390625e-06,
      "step": 4900
    },
    {
      "epoch": 4.673004150390625e-06,
      "step": 4900,
      "training_step_time": 17.055063009262085
    },
    {
      "epoch": 4.673957824707031e-06,
      "model_forward_time": 5.948147535324097,
      "step": 4901
    },
    {
      "epoch": 4.673957824707031e-06,
      "step": 4901,
      "training_step_time": 17.42050528526306
    },
    {
      "epoch": 4.674911499023438e-06,
      "model_forward_time": 5.606945991516113,
      "step": 4902
    },
    {
      "epoch": 4.674911499023438e-06,
      "step": 4902,
      "training_step_time": 17.095100164413452
    },
    {
      "epoch": 4.675865173339843e-06,
      "model_forward_time": 5.606907844543457,
      "step": 4903
    },
    {
      "epoch": 4.675865173339843e-06,
      "step": 4903,
      "training_step_time": 17.07845687866211
    },
    {
      "epoch": 4.67681884765625e-06,
      "model_forward_time": 5.600295543670654,
      "step": 4904
    },
    {
      "epoch": 4.67681884765625e-06,
      "step": 4904,
      "training_step_time": 17.03039789199829
    },
    {
      "epoch": 4.6777725219726565e-06,
      "model_forward_time": 5.579360246658325,
      "step": 4905
    },
    {
      "epoch": 4.6777725219726565e-06,
      "step": 4905,
      "training_step_time": 16.989237785339355
    },
    {
      "epoch": 4.678726196289062e-06,
      "model_forward_time": 5.947410583496094,
      "step": 4906
    },
    {
      "epoch": 4.678726196289062e-06,
      "step": 4906,
      "training_step_time": 17.394473552703857
    },
    {
      "epoch": 4.679679870605469e-06,
      "model_forward_time": 5.585334539413452,
      "step": 4907
    },
    {
      "epoch": 4.679679870605469e-06,
      "step": 4907,
      "training_step_time": 17.060744047164917
    },
    {
      "epoch": 4.680633544921875e-06,
      "model_forward_time": 5.607980728149414,
      "step": 4908
    },
    {
      "epoch": 4.680633544921875e-06,
      "step": 4908,
      "training_step_time": 17.09506607055664
    },
    {
      "epoch": 4.681587219238281e-06,
      "model_forward_time": 5.608469009399414,
      "step": 4909
    },
    {
      "epoch": 4.681587219238281e-06,
      "step": 4909,
      "training_step_time": 17.081091165542603
    },
    {
      "epoch": 4.682540893554688e-06,
      "grad_norm": 1.9874905347824097,
      "learning_rate": 1e-05,
      "loss": 0.0803,
      "step": 4910
    },
    {
      "epoch": 4.682540893554688e-06,
      "model_forward_time": 5.6049041748046875,
      "step": 4910
    },
    {
      "dynamics_loss_avg": 0.05984133370220661,
      "epoch": 4.682540893554688e-06,
      "step": 4910
    },
    {
      "action_loss_avg": 0.017708406178280713,
      "epoch": 4.682540893554688e-06,
      "step": 4910
    },
    {
      "epoch": 4.682540893554688e-06,
      "step": 4910,
      "training_step_time": 17.082221269607544
    },
    {
      "epoch": 4.683494567871093e-06,
      "model_forward_time": 5.6124842166900635,
      "step": 4911
    },
    {
      "epoch": 4.683494567871093e-06,
      "step": 4911,
      "training_step_time": 17.103241443634033
    },
    {
      "epoch": 4.6844482421875e-06,
      "model_forward_time": 5.951883316040039,
      "step": 4912
    },
    {
      "epoch": 4.6844482421875e-06,
      "step": 4912,
      "training_step_time": 17.440507888793945
    },
    {
      "epoch": 4.6854019165039065e-06,
      "model_forward_time": 5.611806154251099,
      "step": 4913
    },
    {
      "epoch": 4.6854019165039065e-06,
      "step": 4913,
      "training_step_time": 17.069993019104004
    },
    {
      "epoch": 4.686355590820312e-06,
      "model_forward_time": 5.597675085067749,
      "step": 4914
    },
    {
      "epoch": 4.686355590820312e-06,
      "step": 4914,
      "training_step_time": 17.07535719871521
    },
    {
      "epoch": 4.687309265136719e-06,
      "model_forward_time": 5.6069958209991455,
      "step": 4915
    },
    {
      "epoch": 4.687309265136719e-06,
      "step": 4915,
      "training_step_time": 17.07790780067444
    },
    {
      "epoch": 4.688262939453125e-06,
      "model_forward_time": 5.597931861877441,
      "step": 4916
    },
    {
      "epoch": 4.688262939453125e-06,
      "step": 4916,
      "training_step_time": 17.047401428222656
    },
    {
      "epoch": 4.689216613769531e-06,
      "model_forward_time": 5.970580816268921,
      "step": 4917
    },
    {
      "epoch": 4.689216613769531e-06,
      "step": 4917,
      "training_step_time": 17.401955604553223
    },
    {
      "epoch": 4.690170288085938e-06,
      "model_forward_time": 5.586036682128906,
      "step": 4918
    },
    {
      "epoch": 4.690170288085938e-06,
      "step": 4918,
      "training_step_time": 17.050836324691772
    },
    {
      "epoch": 4.691123962402343e-06,
      "model_forward_time": 5.594523191452026,
      "step": 4919
    },
    {
      "epoch": 4.691123962402343e-06,
      "step": 4919,
      "training_step_time": 17.045186519622803
    },
    {
      "epoch": 4.69207763671875e-06,
      "grad_norm": 1.0496253967285156,
      "learning_rate": 1e-05,
      "loss": 0.0621,
      "step": 4920
    },
    {
      "epoch": 4.69207763671875e-06,
      "model_forward_time": 5.593639373779297,
      "step": 4920
    },
    {
      "dynamics_loss_avg": 0.04790373295545578,
      "epoch": 4.69207763671875e-06,
      "step": 4920
    },
    {
      "action_loss_avg": 0.010755837475880981,
      "epoch": 4.69207763671875e-06,
      "step": 4920
    },
    {
      "epoch": 4.69207763671875e-06,
      "step": 4920,
      "training_step_time": 17.073455333709717
    },
    {
      "epoch": 4.6930313110351565e-06,
      "model_forward_time": 5.609557867050171,
      "step": 4921
    },
    {
      "epoch": 4.6930313110351565e-06,
      "step": 4921,
      "training_step_time": 17.099857568740845
    },
    {
      "epoch": 4.693984985351562e-06,
      "model_forward_time": 5.970269680023193,
      "step": 4922
    },
    {
      "epoch": 4.693984985351562e-06,
      "step": 4922,
      "training_step_time": 17.439253568649292
    },
    {
      "epoch": 4.694938659667969e-06,
      "model_forward_time": 5.591267108917236,
      "step": 4923
    },
    {
      "epoch": 4.694938659667969e-06,
      "step": 4923,
      "training_step_time": 17.008087635040283
    },
    {
      "epoch": 4.695892333984375e-06,
      "model_forward_time": 5.57296347618103,
      "step": 4924
    },
    {
      "epoch": 4.695892333984375e-06,
      "step": 4924,
      "training_step_time": 17.00187611579895
    },
    {
      "epoch": 4.696846008300781e-06,
      "model_forward_time": 5.5741047859191895,
      "step": 4925
    },
    {
      "epoch": 4.696846008300781e-06,
      "step": 4925,
      "training_step_time": 17.02360439300537
    },
    {
      "epoch": 4.697799682617188e-06,
      "model_forward_time": 5.576920747756958,
      "step": 4926
    },
    {
      "epoch": 4.697799682617188e-06,
      "step": 4926,
      "training_step_time": 17.003984212875366
    },
    {
      "epoch": 4.698753356933593e-06,
      "model_forward_time": 5.582220792770386,
      "step": 4927
    },
    {
      "epoch": 4.698753356933593e-06,
      "step": 4927,
      "training_step_time": 17.01171088218689
    },
    {
      "epoch": 4.69970703125e-06,
      "model_forward_time": 5.983840465545654,
      "step": 4928
    },
    {
      "epoch": 4.69970703125e-06,
      "step": 4928,
      "training_step_time": 17.45361566543579
    },
    {
      "epoch": 4.7006607055664065e-06,
      "model_forward_time": 5.607407331466675,
      "step": 4929
    },
    {
      "epoch": 4.7006607055664065e-06,
      "step": 4929,
      "training_step_time": 17.08033061027527
    },
    {
      "epoch": 4.701614379882812e-06,
      "grad_norm": 0.6323829889297485,
      "learning_rate": 1e-05,
      "loss": 0.0483,
      "step": 4930
    },
    {
      "epoch": 4.701614379882812e-06,
      "model_forward_time": 5.6003804206848145,
      "step": 4930
    },
    {
      "dynamics_loss_avg": 0.04367505759000778,
      "epoch": 4.701614379882812e-06,
      "step": 4930
    },
    {
      "action_loss_avg": 0.012671982706524431,
      "epoch": 4.701614379882812e-06,
      "step": 4930
    },
    {
      "epoch": 4.701614379882812e-06,
      "step": 4930,
      "training_step_time": 17.0549476146698
    },
    {
      "epoch": 4.702568054199219e-06,
      "model_forward_time": 5.599653959274292,
      "step": 4931
    },
    {
      "epoch": 4.702568054199219e-06,
      "step": 4931,
      "training_step_time": 17.069803476333618
    },
    {
      "epoch": 4.703521728515625e-06,
      "model_forward_time": 5.5976786613464355,
      "step": 4932
    },
    {
      "epoch": 4.703521728515625e-06,
      "step": 4932,
      "training_step_time": 17.070950746536255
    },
    {
      "epoch": 4.704475402832031e-06,
      "model_forward_time": 5.598895788192749,
      "step": 4933
    },
    {
      "epoch": 4.704475402832031e-06,
      "step": 4933,
      "training_step_time": 17.435611486434937
    },
    {
      "epoch": 4.705429077148438e-06,
      "model_forward_time": 5.603310823440552,
      "step": 4934
    },
    {
      "epoch": 4.705429077148438e-06,
      "step": 4934,
      "training_step_time": 17.099704265594482
    },
    {
      "epoch": 4.706382751464843e-06,
      "model_forward_time": 5.610991954803467,
      "step": 4935
    },
    {
      "epoch": 4.706382751464843e-06,
      "step": 4935,
      "training_step_time": 17.102513790130615
    },
    {
      "epoch": 4.70733642578125e-06,
      "model_forward_time": 5.608177900314331,
      "step": 4936
    },
    {
      "epoch": 4.70733642578125e-06,
      "step": 4936,
      "training_step_time": 17.0973961353302
    },
    {
      "epoch": 4.7082901000976565e-06,
      "model_forward_time": 5.604506731033325,
      "step": 4937
    },
    {
      "epoch": 4.7082901000976565e-06,
      "step": 4937,
      "training_step_time": 17.091696977615356
    },
    {
      "epoch": 4.709243774414062e-06,
      "model_forward_time": 5.60856556892395,
      "step": 4938
    },
    {
      "epoch": 4.709243774414062e-06,
      "step": 4938,
      "training_step_time": 17.06916642189026
    },
    {
      "epoch": 4.710197448730469e-06,
      "model_forward_time": 5.9642274379730225,
      "step": 4939
    },
    {
      "epoch": 4.710197448730469e-06,
      "step": 4939,
      "training_step_time": 17.373802661895752
    },
    {
      "epoch": 4.711151123046875e-06,
      "grad_norm": 2.03192138671875,
      "learning_rate": 1e-05,
      "loss": 0.0761,
      "step": 4940
    },
    {
      "epoch": 4.711151123046875e-06,
      "model_forward_time": 5.585983514785767,
      "step": 4940
    },
    {
      "dynamics_loss_avg": 0.06084775645285845,
      "epoch": 4.711151123046875e-06,
      "step": 4940
    },
    {
      "action_loss_avg": 0.0179177442798391,
      "epoch": 4.711151123046875e-06,
      "step": 4940
    },
    {
      "epoch": 4.711151123046875e-06,
      "step": 4940,
      "training_step_time": 17.02827739715576
    },
    {
      "epoch": 4.712104797363281e-06,
      "model_forward_time": 5.566432237625122,
      "step": 4941
    },
    {
      "epoch": 4.712104797363281e-06,
      "step": 4941,
      "training_step_time": 16.979948043823242
    },
    {
      "epoch": 4.713058471679688e-06,
      "model_forward_time": 5.578927755355835,
      "step": 4942
    },
    {
      "epoch": 4.713058471679688e-06,
      "step": 4942,
      "training_step_time": 17.011919736862183
    },
    {
      "epoch": 4.714012145996093e-06,
      "model_forward_time": 5.583843946456909,
      "step": 4943
    },
    {
      "epoch": 4.714012145996093e-06,
      "step": 4943,
      "training_step_time": 17.011029720306396
    },
    {
      "epoch": 4.7149658203125e-06,
      "model_forward_time": 5.5848963260650635,
      "step": 4944
    },
    {
      "epoch": 4.7149658203125e-06,
      "step": 4944,
      "training_step_time": 17.384791374206543
    },
    {
      "epoch": 4.7159194946289065e-06,
      "model_forward_time": 5.602546453475952,
      "step": 4945
    },
    {
      "epoch": 4.7159194946289065e-06,
      "step": 4945,
      "training_step_time": 17.113900661468506
    },
    {
      "epoch": 4.716873168945312e-06,
      "model_forward_time": 5.622694730758667,
      "step": 4946
    },
    {
      "epoch": 4.716873168945312e-06,
      "step": 4946,
      "training_step_time": 17.12033987045288
    },
    {
      "epoch": 4.717826843261719e-06,
      "model_forward_time": 5.61326789855957,
      "step": 4947
    },
    {
      "epoch": 4.717826843261719e-06,
      "step": 4947,
      "training_step_time": 17.089173555374146
    },
    {
      "epoch": 4.718780517578125e-06,
      "model_forward_time": 5.602346658706665,
      "step": 4948
    },
    {
      "epoch": 4.718780517578125e-06,
      "step": 4948,
      "training_step_time": 17.06799602508545
    },
    {
      "epoch": 4.719734191894531e-06,
      "model_forward_time": 5.612497091293335,
      "step": 4949
    },
    {
      "epoch": 4.719734191894531e-06,
      "step": 4949,
      "training_step_time": 17.07978367805481
    },
    {
      "epoch": 4.720687866210938e-06,
      "grad_norm": 0.7245944738388062,
      "learning_rate": 1e-05,
      "loss": 0.0916,
      "step": 4950
    },
    {
      "epoch": 4.720687866210938e-06,
      "model_forward_time": 5.975620746612549,
      "step": 4950
    },
    {
      "dynamics_loss_avg": 0.06614376548677683,
      "epoch": 4.720687866210938e-06,
      "step": 4950
    },
    {
      "action_loss_avg": 0.015503555396571755,
      "epoch": 4.720687866210938e-06,
      "step": 4950
    },
    {
      "epoch": 4.720687866210938e-06,
      "step": 4950,
      "training_step_time": 17.45168662071228
    },
    {
      "epoch": 4.721641540527343e-06,
      "model_forward_time": 5.598299503326416,
      "step": 4951
    },
    {
      "epoch": 4.721641540527343e-06,
      "step": 4951,
      "training_step_time": 17.050971508026123
    },
    {
      "epoch": 4.72259521484375e-06,
      "model_forward_time": 5.597050428390503,
      "step": 4952
    },
    {
      "epoch": 4.72259521484375e-06,
      "step": 4952,
      "training_step_time": 17.044963359832764
    },
    {
      "epoch": 4.7235488891601565e-06,
      "model_forward_time": 5.5902674198150635,
      "step": 4953
    },
    {
      "epoch": 4.7235488891601565e-06,
      "step": 4953,
      "training_step_time": 17.04961061477661
    },
    {
      "epoch": 4.724502563476562e-06,
      "model_forward_time": 5.610728025436401,
      "step": 4954
    },
    {
      "epoch": 4.724502563476562e-06,
      "step": 4954,
      "training_step_time": 17.11536931991577
    },
    {
      "epoch": 4.725456237792969e-06,
      "model_forward_time": 5.990591526031494,
      "step": 4955
    },
    {
      "epoch": 4.725456237792969e-06,
      "step": 4955,
      "training_step_time": 17.50719141960144
    },
    {
      "epoch": 4.726409912109375e-06,
      "model_forward_time": 5.6017210483551025,
      "step": 4956
    },
    {
      "epoch": 4.726409912109375e-06,
      "step": 4956,
      "training_step_time": 17.045578002929688
    },
    {
      "epoch": 4.727363586425781e-06,
      "model_forward_time": 5.589142799377441,
      "step": 4957
    },
    {
      "epoch": 4.727363586425781e-06,
      "step": 4957,
      "training_step_time": 16.997745990753174
    },
    {
      "epoch": 4.728317260742188e-06,
      "model_forward_time": 5.568831920623779,
      "step": 4958
    },
    {
      "epoch": 4.728317260742188e-06,
      "step": 4958,
      "training_step_time": 16.9958655834198
    },
    {
      "epoch": 4.729270935058593e-06,
      "model_forward_time": 5.5702574253082275,
      "step": 4959
    },
    {
      "epoch": 4.729270935058593e-06,
      "step": 4959,
      "training_step_time": 17.006301403045654
    },
    {
      "epoch": 4.730224609375e-06,
      "grad_norm": 0.9731631875038147,
      "learning_rate": 1e-05,
      "loss": 0.0505,
      "step": 4960
    },
    {
      "epoch": 4.730224609375e-06,
      "model_forward_time": 5.567998886108398,
      "step": 4960
    },
    {
      "dynamics_loss_avg": 0.04778903461992741,
      "epoch": 4.730224609375e-06,
      "step": 4960
    },
    {
      "action_loss_avg": 0.010641754139214754,
      "epoch": 4.730224609375e-06,
      "step": 4960
    },
    {
      "epoch": 4.730224609375e-06,
      "step": 4960,
      "training_step_time": 16.975791215896606
    },
    {
      "epoch": 4.7311782836914065e-06,
      "model_forward_time": 5.938513994216919,
      "step": 4961
    },
    {
      "epoch": 4.7311782836914065e-06,
      "step": 4961,
      "training_step_time": 17.357913732528687
    },
    {
      "epoch": 4.732131958007812e-06,
      "model_forward_time": 5.578163385391235,
      "step": 4962
    },
    {
      "epoch": 4.732131958007812e-06,
      "step": 4962,
      "training_step_time": 17.01921033859253
    },
    {
      "epoch": 4.733085632324219e-06,
      "model_forward_time": 5.598211050033569,
      "step": 4963
    },
    {
      "epoch": 4.733085632324219e-06,
      "step": 4963,
      "training_step_time": 17.05235505104065
    },
    {
      "epoch": 4.734039306640625e-06,
      "model_forward_time": 5.599417209625244,
      "step": 4964
    },
    {
      "epoch": 4.734039306640625e-06,
      "step": 4964,
      "training_step_time": 17.078333139419556
    },
    {
      "epoch": 4.734992980957031e-06,
      "model_forward_time": 5.6087353229522705,
      "step": 4965
    },
    {
      "epoch": 4.734992980957031e-06,
      "step": 4965,
      "training_step_time": 17.10943603515625
    },
    {
      "epoch": 4.735946655273438e-06,
      "model_forward_time": 5.60488748550415,
      "step": 4966
    },
    {
      "epoch": 4.735946655273438e-06,
      "step": 4966,
      "training_step_time": 17.447892904281616
    },
    {
      "epoch": 4.736900329589843e-06,
      "model_forward_time": 5.607456207275391,
      "step": 4967
    },
    {
      "epoch": 4.736900329589843e-06,
      "step": 4967,
      "training_step_time": 17.0866277217865
    },
    {
      "epoch": 4.73785400390625e-06,
      "model_forward_time": 5.599332809448242,
      "step": 4968
    },
    {
      "epoch": 4.73785400390625e-06,
      "step": 4968,
      "training_step_time": 17.077944040298462
    },
    {
      "epoch": 4.7388076782226565e-06,
      "model_forward_time": 5.600971221923828,
      "step": 4969
    },
    {
      "epoch": 4.7388076782226565e-06,
      "step": 4969,
      "training_step_time": 17.072227954864502
    },
    {
      "epoch": 4.739761352539062e-06,
      "grad_norm": 1.2455267906188965,
      "learning_rate": 1e-05,
      "loss": 0.085,
      "step": 4970
    },
    {
      "epoch": 4.739761352539062e-06,
      "model_forward_time": 5.6010026931762695,
      "step": 4970
    },
    {
      "dynamics_loss_avg": 0.06470998972654343,
      "epoch": 4.739761352539062e-06,
      "step": 4970
    },
    {
      "action_loss_avg": 0.01750158779323101,
      "epoch": 4.739761352539062e-06,
      "step": 4970
    },
    {
      "epoch": 4.739761352539062e-06,
      "step": 4970,
      "training_step_time": 17.06558060646057
    },
    {
      "epoch": 4.740715026855469e-06,
      "model_forward_time": 5.597705364227295,
      "step": 4971
    },
    {
      "epoch": 4.740715026855469e-06,
      "step": 4971,
      "training_step_time": 17.044344186782837
    },
    {
      "epoch": 4.741668701171875e-06,
      "model_forward_time": 5.976639986038208,
      "step": 4972
    },
    {
      "epoch": 4.741668701171875e-06,
      "step": 4972,
      "training_step_time": 17.442777395248413
    },
    {
      "epoch": 4.742622375488281e-06,
      "model_forward_time": 5.597761154174805,
      "step": 4973
    },
    {
      "epoch": 4.742622375488281e-06,
      "step": 4973,
      "training_step_time": 17.05610156059265
    },
    {
      "epoch": 4.743576049804688e-06,
      "model_forward_time": 5.597600936889648,
      "step": 4974
    },
    {
      "epoch": 4.743576049804688e-06,
      "step": 4974,
      "training_step_time": 17.030330181121826
    },
    {
      "epoch": 4.744529724121093e-06,
      "model_forward_time": 5.579715728759766,
      "step": 4975
    },
    {
      "epoch": 4.744529724121093e-06,
      "step": 4975,
      "training_step_time": 17.01307487487793
    },
    {
      "epoch": 4.7454833984375e-06,
      "model_forward_time": 5.58409571647644,
      "step": 4976
    },
    {
      "epoch": 4.7454833984375e-06,
      "step": 4976,
      "training_step_time": 17.00910520553589
    },
    {
      "epoch": 4.7464370727539065e-06,
      "model_forward_time": 5.582738161087036,
      "step": 4977
    },
    {
      "epoch": 4.7464370727539065e-06,
      "step": 4977,
      "training_step_time": 17.37882423400879
    },
    {
      "epoch": 4.747390747070312e-06,
      "model_forward_time": 5.596461296081543,
      "step": 4978
    },
    {
      "epoch": 4.747390747070312e-06,
      "step": 4978,
      "training_step_time": 17.042141914367676
    },
    {
      "epoch": 4.748344421386719e-06,
      "model_forward_time": 5.600389719009399,
      "step": 4979
    },
    {
      "epoch": 4.748344421386719e-06,
      "step": 4979,
      "training_step_time": 17.08053755760193
    },
    {
      "epoch": 4.749298095703125e-06,
      "grad_norm": 3.131661891937256,
      "learning_rate": 1e-05,
      "loss": 0.0824,
      "step": 4980
    },
    {
      "epoch": 4.749298095703125e-06,
      "model_forward_time": 5.6087377071380615,
      "step": 4980
    },
    {
      "dynamics_loss_avg": 0.06717907208949328,
      "epoch": 4.749298095703125e-06,
      "step": 4980
    },
    {
      "action_loss_avg": 0.0182533607352525,
      "epoch": 4.749298095703125e-06,
      "step": 4980
    },
    {
      "epoch": 4.749298095703125e-06,
      "step": 4980,
      "training_step_time": 17.083795070648193
    },
    {
      "epoch": 4.750251770019531e-06,
      "model_forward_time": 5.599314451217651,
      "step": 4981
    },
    {
      "epoch": 4.750251770019531e-06,
      "step": 4981,
      "training_step_time": 17.077497720718384
    },
    {
      "epoch": 4.751205444335938e-06,
      "model_forward_time": 5.609999179840088,
      "step": 4982
    },
    {
      "epoch": 4.751205444335938e-06,
      "step": 4982,
      "training_step_time": 17.0972101688385
    },
    {
      "epoch": 4.752159118652343e-06,
      "model_forward_time": 5.967405319213867,
      "step": 4983
    },
    {
      "epoch": 4.752159118652343e-06,
      "step": 4983,
      "training_step_time": 17.424103498458862
    },
    {
      "epoch": 4.75311279296875e-06,
      "model_forward_time": 5.596821546554565,
      "step": 4984
    },
    {
      "epoch": 4.75311279296875e-06,
      "step": 4984,
      "training_step_time": 17.034743309020996
    },
    {
      "epoch": 4.7540664672851566e-06,
      "model_forward_time": 5.587738275527954,
      "step": 4985
    },
    {
      "epoch": 4.7540664672851566e-06,
      "step": 4985,
      "training_step_time": 16.991877794265747
    },
    {
      "epoch": 4.755020141601562e-06,
      "model_forward_time": 5.580214738845825,
      "step": 4986
    },
    {
      "epoch": 4.755020141601562e-06,
      "step": 4986,
      "training_step_time": 17.02932906150818
    },
    {
      "epoch": 4.755973815917969e-06,
      "model_forward_time": 5.592528581619263,
      "step": 4987
    },
    {
      "epoch": 4.755973815917969e-06,
      "step": 4987,
      "training_step_time": 17.046068906784058
    },
    {
      "epoch": 4.756927490234375e-06,
      "model_forward_time": 5.949007272720337,
      "step": 4988
    },
    {
      "epoch": 4.756927490234375e-06,
      "step": 4988,
      "training_step_time": 17.40788745880127
    },
    {
      "epoch": 4.757881164550781e-06,
      "model_forward_time": 5.599036693572998,
      "step": 4989
    },
    {
      "epoch": 4.757881164550781e-06,
      "step": 4989,
      "training_step_time": 17.092392206192017
    },
    {
      "epoch": 4.758834838867188e-06,
      "grad_norm": 0.8189800381660461,
      "learning_rate": 1e-05,
      "loss": 0.0718,
      "step": 4990
    },
    {
      "epoch": 4.758834838867188e-06,
      "model_forward_time": 5.609582424163818,
      "step": 4990
    },
    {
      "dynamics_loss_avg": 0.05507998913526535,
      "epoch": 4.758834838867188e-06,
      "step": 4990
    },
    {
      "action_loss_avg": 0.01291734166443348,
      "epoch": 4.758834838867188e-06,
      "step": 4990
    },
    {
      "epoch": 4.758834838867188e-06,
      "step": 4990,
      "training_step_time": 17.07678723335266
    },
    {
      "epoch": 4.7597885131835934e-06,
      "model_forward_time": 5.593351125717163,
      "step": 4991
    },
    {
      "epoch": 4.7597885131835934e-06,
      "step": 4991,
      "training_step_time": 17.03659200668335
    },
    {
      "epoch": 4.7607421875e-06,
      "model_forward_time": 5.596556663513184,
      "step": 4992
    },
    {
      "epoch": 4.7607421875e-06,
      "step": 4992,
      "training_step_time": 17.055955171585083
    },
    {
      "epoch": 4.7616958618164066e-06,
      "model_forward_time": 5.589666128158569,
      "step": 4993
    },
    {
      "epoch": 4.7616958618164066e-06,
      "step": 4993,
      "training_step_time": 17.032485961914062
    },
    {
      "epoch": 4.762649536132812e-06,
      "model_forward_time": 5.9819419384002686,
      "step": 4994
    },
    {
      "epoch": 4.762649536132812e-06,
      "step": 4994,
      "training_step_time": 17.4514799118042
    },
    {
      "epoch": 4.763603210449219e-06,
      "model_forward_time": 5.636366367340088,
      "step": 4995
    },
    {
      "epoch": 4.763603210449219e-06,
      "step": 4995,
      "training_step_time": 17.119659185409546
    },
    {
      "epoch": 4.764556884765625e-06,
      "model_forward_time": 5.597094774246216,
      "step": 4996
    },
    {
      "epoch": 4.764556884765625e-06,
      "step": 4996,
      "training_step_time": 17.066418409347534
    },
    {
      "epoch": 4.765510559082031e-06,
      "model_forward_time": 5.604980945587158,
      "step": 4997
    },
    {
      "epoch": 4.765510559082031e-06,
      "step": 4997,
      "training_step_time": 17.03914713859558
    },
    {
      "epoch": 4.766464233398438e-06,
      "model_forward_time": 5.584441900253296,
      "step": 4998
    },
    {
      "epoch": 4.766464233398438e-06,
      "step": 4998,
      "training_step_time": 17.024625778198242
    },
    {
      "epoch": 4.7674179077148434e-06,
      "model_forward_time": 5.584259033203125,
      "step": 4999
    },
    {
      "epoch": 4.7674179077148434e-06,
      "step": 4999,
      "training_step_time": 17.380558729171753
    },
    {
      "epoch": 4.76837158203125e-06,
      "grad_norm": 0.8942216634750366,
      "learning_rate": 1e-05,
      "loss": 0.065,
      "step": 5000
    },
    {
      "epoch": 4.76837158203125e-06,
      "model_forward_time": 5.643874406814575,
      "step": 5000
    },
    {
      "dynamics_loss_avg": 0.051079174503684044,
      "epoch": 4.76837158203125e-06,
      "step": 5000
    },
    {
      "action_loss_avg": 0.011818192526698112,
      "epoch": 4.76837158203125e-06,
      "step": 5000
    },
    {
      "epoch": 4.76837158203125e-06,
      "step": 5000,
      "training_step_time": 17.104612112045288
    },
    {
      "epoch": 4.7693252563476566e-06,
      "model_forward_time": 5.6535484790802,
      "step": 5001
    },
    {
      "epoch": 4.7693252563476566e-06,
      "step": 5001,
      "training_step_time": 17.149889945983887
    },
    {
      "epoch": 4.770278930664062e-06,
      "model_forward_time": 5.663683176040649,
      "step": 5002
    },
    {
      "epoch": 4.770278930664062e-06,
      "step": 5002,
      "training_step_time": 17.191086053848267
    },
    {
      "epoch": 4.771232604980469e-06,
      "model_forward_time": 5.712050437927246,
      "step": 5003
    },
    {
      "epoch": 4.771232604980469e-06,
      "step": 5003,
      "training_step_time": 17.202503442764282
    },
    {
      "epoch": 4.772186279296875e-06,
      "model_forward_time": 5.593804121017456,
      "step": 5004
    },
    {
      "epoch": 4.772186279296875e-06,
      "step": 5004,
      "training_step_time": 17.018983364105225
    },
    {
      "epoch": 4.773139953613281e-06,
      "model_forward_time": 5.93223237991333,
      "step": 5005
    },
    {
      "epoch": 4.773139953613281e-06,
      "step": 5005,
      "training_step_time": 17.356547594070435
    },
    {
      "epoch": 4.774093627929688e-06,
      "model_forward_time": 5.571833610534668,
      "step": 5006
    },
    {
      "epoch": 4.774093627929688e-06,
      "step": 5006,
      "training_step_time": 16.978187799453735
    },
    {
      "epoch": 4.7750473022460934e-06,
      "model_forward_time": 5.5772950649261475,
      "step": 5007
    },
    {
      "epoch": 4.7750473022460934e-06,
      "step": 5007,
      "training_step_time": 16.986352682113647
    },
    {
      "epoch": 4.7760009765625e-06,
      "model_forward_time": 5.577723026275635,
      "step": 5008
    },
    {
      "epoch": 4.7760009765625e-06,
      "step": 5008,
      "training_step_time": 17.018598556518555
    },
    {
      "epoch": 4.7769546508789066e-06,
      "model_forward_time": 5.577831745147705,
      "step": 5009
    },
    {
      "epoch": 4.7769546508789066e-06,
      "step": 5009,
      "training_step_time": 16.994925498962402
    },
    {
      "epoch": 4.777908325195312e-06,
      "grad_norm": 1.1829758882522583,
      "learning_rate": 1e-05,
      "loss": 0.0592,
      "step": 5010
    },
    {
      "epoch": 4.777908325195312e-06,
      "model_forward_time": 5.960844039916992,
      "step": 5010
    },
    {
      "dynamics_loss_avg": 0.04127836413681507,
      "epoch": 4.777908325195312e-06,
      "step": 5010
    },
    {
      "action_loss_avg": 0.014046655048150569,
      "epoch": 4.777908325195312e-06,
      "step": 5010
    },
    {
      "epoch": 4.777908325195312e-06,
      "step": 5010,
      "training_step_time": 17.380997896194458
    },
    {
      "epoch": 4.778861999511719e-06,
      "model_forward_time": 5.588119029998779,
      "step": 5011
    },
    {
      "epoch": 4.778861999511719e-06,
      "step": 5011,
      "training_step_time": 17.05636477470398
    },
    {
      "epoch": 4.779815673828125e-06,
      "model_forward_time": 5.613756895065308,
      "step": 5012
    },
    {
      "epoch": 4.779815673828125e-06,
      "step": 5012,
      "training_step_time": 17.114885091781616
    },
    {
      "epoch": 4.780769348144531e-06,
      "model_forward_time": 5.617886066436768,
      "step": 5013
    },
    {
      "epoch": 4.780769348144531e-06,
      "step": 5013,
      "training_step_time": 17.09917116165161
    },
    {
      "epoch": 4.781723022460938e-06,
      "model_forward_time": 5.59639310836792,
      "step": 5014
    },
    {
      "epoch": 4.781723022460938e-06,
      "step": 5014,
      "training_step_time": 17.049845933914185
    },
    {
      "epoch": 4.7826766967773434e-06,
      "model_forward_time": 5.588357448577881,
      "step": 5015
    },
    {
      "epoch": 4.7826766967773434e-06,
      "step": 5015,
      "training_step_time": 17.392919063568115
    },
    {
      "epoch": 4.78363037109375e-06,
      "model_forward_time": 5.58462929725647,
      "step": 5016
    },
    {
      "epoch": 4.78363037109375e-06,
      "step": 5016,
      "training_step_time": 17.027593851089478
    },
    {
      "epoch": 4.784584045410157e-06,
      "model_forward_time": 5.599420547485352,
      "step": 5017
    },
    {
      "epoch": 4.784584045410157e-06,
      "step": 5017,
      "training_step_time": 17.05896520614624
    },
    {
      "epoch": 4.785537719726562e-06,
      "model_forward_time": 5.598253965377808,
      "step": 5018
    },
    {
      "epoch": 4.785537719726562e-06,
      "step": 5018,
      "training_step_time": 17.04619860649109
    },
    {
      "epoch": 4.786491394042969e-06,
      "model_forward_time": 5.594867944717407,
      "step": 5019
    },
    {
      "epoch": 4.786491394042969e-06,
      "step": 5019,
      "training_step_time": 17.041746139526367
    },
    {
      "epoch": 4.787445068359375e-06,
      "grad_norm": 0.9210015535354614,
      "learning_rate": 1e-05,
      "loss": 0.0741,
      "step": 5020
    },
    {
      "epoch": 4.787445068359375e-06,
      "model_forward_time": 5.604523658752441,
      "step": 5020
    },
    {
      "dynamics_loss_avg": 0.06237496752291918,
      "epoch": 4.787445068359375e-06,
      "step": 5020
    },
    {
      "action_loss_avg": 0.015618118969723583,
      "epoch": 4.787445068359375e-06,
      "step": 5020
    },
    {
      "epoch": 4.787445068359375e-06,
      "step": 5020,
      "training_step_time": 17.093470335006714
    },
    {
      "epoch": 4.788398742675781e-06,
      "model_forward_time": 5.972097873687744,
      "step": 5021
    },
    {
      "epoch": 4.788398742675781e-06,
      "step": 5021,
      "training_step_time": 17.437026262283325
    },
    {
      "epoch": 4.789352416992188e-06,
      "model_forward_time": 5.587146997451782,
      "step": 5022
    },
    {
      "epoch": 4.789352416992188e-06,
      "step": 5022,
      "training_step_time": 17.005841493606567
    },
    {
      "epoch": 4.7903060913085935e-06,
      "model_forward_time": 5.580598592758179,
      "step": 5023
    },
    {
      "epoch": 4.7903060913085935e-06,
      "step": 5023,
      "training_step_time": 17.01270365715027
    },
    {
      "epoch": 4.791259765625e-06,
      "model_forward_time": 5.569638729095459,
      "step": 5024
    },
    {
      "epoch": 4.791259765625e-06,
      "step": 5024,
      "training_step_time": 17.017143487930298
    },
    {
      "epoch": 4.792213439941407e-06,
      "model_forward_time": 5.571819067001343,
      "step": 5025
    },
    {
      "epoch": 4.792213439941407e-06,
      "step": 5025,
      "training_step_time": 16.981400728225708
    },
    {
      "epoch": 4.793167114257812e-06,
      "model_forward_time": 5.945605039596558,
      "step": 5026
    },
    {
      "epoch": 4.793167114257812e-06,
      "step": 5026,
      "training_step_time": 17.387205600738525
    },
    {
      "epoch": 4.794120788574219e-06,
      "model_forward_time": 5.596984624862671,
      "step": 5027
    },
    {
      "epoch": 4.794120788574219e-06,
      "step": 5027,
      "training_step_time": 17.034382343292236
    },
    {
      "epoch": 4.795074462890625e-06,
      "model_forward_time": 5.596941709518433,
      "step": 5028
    },
    {
      "epoch": 4.795074462890625e-06,
      "step": 5028,
      "training_step_time": 17.04204773902893
    },
    {
      "epoch": 4.796028137207031e-06,
      "model_forward_time": 5.587854623794556,
      "step": 5029
    },
    {
      "epoch": 4.796028137207031e-06,
      "step": 5029,
      "training_step_time": 17.06163191795349
    },
    {
      "epoch": 4.796981811523438e-06,
      "grad_norm": 0.6119184494018555,
      "learning_rate": 1e-05,
      "loss": 0.0917,
      "step": 5030
    },
    {
      "epoch": 4.796981811523438e-06,
      "model_forward_time": 5.60430383682251,
      "step": 5030
    },
    {
      "dynamics_loss_avg": 0.07086405623704195,
      "epoch": 4.796981811523438e-06,
      "step": 5030
    },
    {
      "action_loss_avg": 0.01784655936062336,
      "epoch": 4.796981811523438e-06,
      "step": 5030
    },
    {
      "epoch": 4.796981811523438e-06,
      "step": 5030,
      "training_step_time": 17.108799934387207
    },
    {
      "epoch": 4.7979354858398435e-06,
      "model_forward_time": 5.6149742603302,
      "step": 5031
    },
    {
      "epoch": 4.7979354858398435e-06,
      "step": 5031,
      "training_step_time": 17.13425898551941
    },
    {
      "epoch": 4.79888916015625e-06,
      "model_forward_time": 5.992478370666504,
      "step": 5032
    },
    {
      "epoch": 4.79888916015625e-06,
      "step": 5032,
      "training_step_time": 17.51026701927185
    },
    {
      "epoch": 4.799842834472657e-06,
      "model_forward_time": 5.615562200546265,
      "step": 5033
    },
    {
      "epoch": 4.799842834472657e-06,
      "step": 5033,
      "training_step_time": 17.123725175857544
    },
    {
      "epoch": 4.800796508789062e-06,
      "model_forward_time": 5.618135452270508,
      "step": 5034
    },
    {
      "epoch": 4.800796508789062e-06,
      "step": 5034,
      "training_step_time": 17.11715340614319
    },
    {
      "epoch": 4.801750183105469e-06,
      "model_forward_time": 5.617105960845947,
      "step": 5035
    },
    {
      "epoch": 4.801750183105469e-06,
      "step": 5035,
      "training_step_time": 17.110215425491333
    },
    {
      "epoch": 4.802703857421875e-06,
      "model_forward_time": 5.612759351730347,
      "step": 5036
    },
    {
      "epoch": 4.802703857421875e-06,
      "step": 5036,
      "training_step_time": 17.078442811965942
    },
    {
      "epoch": 4.803657531738281e-06,
      "model_forward_time": 5.938739061355591,
      "step": 5037
    },
    {
      "epoch": 4.803657531738281e-06,
      "step": 5037,
      "training_step_time": 17.343470811843872
    },
    {
      "epoch": 4.804611206054688e-06,
      "model_forward_time": 5.568142414093018,
      "step": 5038
    },
    {
      "epoch": 4.804611206054688e-06,
      "step": 5038,
      "training_step_time": 16.97257161140442
    },
    {
      "epoch": 4.8055648803710935e-06,
      "model_forward_time": 5.58379340171814,
      "step": 5039
    },
    {
      "epoch": 4.8055648803710935e-06,
      "step": 5039,
      "training_step_time": 17.034526109695435
    },
    {
      "epoch": 4.8065185546875e-06,
      "grad_norm": 1.0511219501495361,
      "learning_rate": 1e-05,
      "loss": 0.0895,
      "step": 5040
    },
    {
      "epoch": 4.8065185546875e-06,
      "model_forward_time": 5.589473009109497,
      "step": 5040
    },
    {
      "dynamics_loss_avg": 0.08101218938827515,
      "epoch": 4.8065185546875e-06,
      "step": 5040
    },
    {
      "action_loss_avg": 0.01835549445822835,
      "epoch": 4.8065185546875e-06,
      "step": 5040
    },
    {
      "epoch": 4.8065185546875e-06,
      "step": 5040,
      "training_step_time": 17.050922632217407
    },
    {
      "epoch": 4.807472229003907e-06,
      "model_forward_time": 5.592259168624878,
      "step": 5041
    },
    {
      "epoch": 4.807472229003907e-06,
      "step": 5041,
      "training_step_time": 17.040939331054688
    },
    {
      "epoch": 4.808425903320312e-06,
      "model_forward_time": 5.5873401165008545,
      "step": 5042
    },
    {
      "epoch": 4.808425903320312e-06,
      "step": 5042,
      "training_step_time": 17.03923225402832
    },
    {
      "epoch": 4.809379577636719e-06,
      "model_forward_time": 5.966088771820068,
      "step": 5043
    },
    {
      "epoch": 4.809379577636719e-06,
      "step": 5043,
      "training_step_time": 17.41335678100586
    },
    {
      "epoch": 4.810333251953125e-06,
      "model_forward_time": 5.592123746871948,
      "step": 5044
    },
    {
      "epoch": 4.810333251953125e-06,
      "step": 5044,
      "training_step_time": 17.05221915245056
    },
    {
      "epoch": 4.811286926269531e-06,
      "model_forward_time": 5.613693714141846,
      "step": 5045
    },
    {
      "epoch": 4.811286926269531e-06,
      "step": 5045,
      "training_step_time": 17.055728435516357
    },
    {
      "epoch": 4.812240600585938e-06,
      "model_forward_time": 5.593399286270142,
      "step": 5046
    },
    {
      "epoch": 4.812240600585938e-06,
      "step": 5046,
      "training_step_time": 17.02117919921875
    },
    {
      "epoch": 4.8131942749023435e-06,
      "model_forward_time": 5.5829079151153564,
      "step": 5047
    },
    {
      "epoch": 4.8131942749023435e-06,
      "step": 5047,
      "training_step_time": 17.009128093719482
    },
    {
      "epoch": 4.81414794921875e-06,
      "model_forward_time": 5.947138547897339,
      "step": 5048
    },
    {
      "epoch": 4.81414794921875e-06,
      "step": 5048,
      "training_step_time": 17.41530156135559
    },
    {
      "epoch": 4.815101623535157e-06,
      "model_forward_time": 5.613464117050171,
      "step": 5049
    },
    {
      "epoch": 4.815101623535157e-06,
      "step": 5049,
      "training_step_time": 17.124913692474365
    },
    {
      "epoch": 4.816055297851562e-06,
      "grad_norm": 0.5669054985046387,
      "learning_rate": 1e-05,
      "loss": 0.0666,
      "step": 5050
    },
    {
      "epoch": 4.816055297851562e-06,
      "model_forward_time": 5.623721122741699,
      "step": 5050
    },
    {
      "dynamics_loss_avg": 0.05190397500991821,
      "epoch": 4.816055297851562e-06,
      "step": 5050
    },
    {
      "action_loss_avg": 0.009771127672865986,
      "epoch": 4.816055297851562e-06,
      "step": 5050
    },
    {
      "epoch": 4.816055297851562e-06,
      "step": 5050,
      "training_step_time": 17.121487855911255
    },
    {
      "epoch": 4.817008972167969e-06,
      "model_forward_time": 5.616979360580444,
      "step": 5051
    },
    {
      "epoch": 4.817008972167969e-06,
      "step": 5051,
      "training_step_time": 17.115649938583374
    },
    {
      "epoch": 4.817962646484375e-06,
      "model_forward_time": 5.612857103347778,
      "step": 5052
    },
    {
      "epoch": 4.817962646484375e-06,
      "step": 5052,
      "training_step_time": 17.111183166503906
    },
    {
      "epoch": 4.818916320800781e-06,
      "model_forward_time": 5.6065685749053955,
      "step": 5053
    },
    {
      "epoch": 4.818916320800781e-06,
      "step": 5053,
      "training_step_time": 17.47629451751709
    },
    {
      "epoch": 4.819869995117188e-06,
      "model_forward_time": 5.613235235214233,
      "step": 5054
    },
    {
      "epoch": 4.819869995117188e-06,
      "step": 5054,
      "training_step_time": 17.100731134414673
    },
    {
      "epoch": 4.8208236694335935e-06,
      "model_forward_time": 5.598733186721802,
      "step": 5055
    },
    {
      "epoch": 4.8208236694335935e-06,
      "step": 5055,
      "training_step_time": 17.08925771713257
    },
    {
      "epoch": 4.82177734375e-06,
      "model_forward_time": 5.60611891746521,
      "step": 5056
    },
    {
      "epoch": 4.82177734375e-06,
      "step": 5056,
      "training_step_time": 17.09968376159668
    },
    {
      "epoch": 4.822731018066407e-06,
      "model_forward_time": 5.6129138469696045,
      "step": 5057
    },
    {
      "epoch": 4.822731018066407e-06,
      "step": 5057,
      "training_step_time": 17.101775646209717
    },
    {
      "epoch": 4.823684692382812e-06,
      "model_forward_time": 5.612120151519775,
      "step": 5058
    },
    {
      "epoch": 4.823684692382812e-06,
      "step": 5058,
      "training_step_time": 17.06755495071411
    },
    {
      "epoch": 4.824638366699219e-06,
      "model_forward_time": 5.933644533157349,
      "step": 5059
    },
    {
      "epoch": 4.824638366699219e-06,
      "step": 5059,
      "training_step_time": 17.370500087738037
    },
    {
      "epoch": 4.825592041015625e-06,
      "grad_norm": 0.7735134959220886,
      "learning_rate": 1e-05,
      "loss": 0.0593,
      "step": 5060
    },
    {
      "epoch": 4.825592041015625e-06,
      "model_forward_time": 5.585489988327026,
      "step": 5060
    },
    {
      "dynamics_loss_avg": 0.04683440085500479,
      "epoch": 4.825592041015625e-06,
      "step": 5060
    },
    {
      "action_loss_avg": 0.009812798723578453,
      "epoch": 4.825592041015625e-06,
      "step": 5060
    },
    {
      "epoch": 4.825592041015625e-06,
      "step": 5060,
      "training_step_time": 17.00728487968445
    },
    {
      "epoch": 4.826545715332031e-06,
      "model_forward_time": 5.595998764038086,
      "step": 5061
    },
    {
      "epoch": 4.826545715332031e-06,
      "step": 5061,
      "training_step_time": 17.04171109199524
    },
    {
      "epoch": 4.827499389648438e-06,
      "model_forward_time": 5.593343496322632,
      "step": 5062
    },
    {
      "epoch": 4.827499389648438e-06,
      "step": 5062,
      "training_step_time": 17.053180694580078
    },
    {
      "epoch": 4.8284530639648435e-06,
      "model_forward_time": 5.598565101623535,
      "step": 5063
    },
    {
      "epoch": 4.8284530639648435e-06,
      "step": 5063,
      "training_step_time": 17.069275617599487
    },
    {
      "epoch": 4.82940673828125e-06,
      "model_forward_time": 5.971495628356934,
      "step": 5064
    },
    {
      "epoch": 4.82940673828125e-06,
      "step": 5064,
      "training_step_time": 17.44282031059265
    },
    {
      "epoch": 4.830360412597657e-06,
      "model_forward_time": 5.611051797866821,
      "step": 5065
    },
    {
      "epoch": 4.830360412597657e-06,
      "step": 5065,
      "training_step_time": 17.100792169570923
    },
    {
      "epoch": 4.831314086914062e-06,
      "model_forward_time": 5.6149773597717285,
      "step": 5066
    },
    {
      "epoch": 4.831314086914062e-06,
      "step": 5066,
      "training_step_time": 17.124090671539307
    },
    {
      "epoch": 4.832267761230469e-06,
      "model_forward_time": 5.616253137588501,
      "step": 5067
    },
    {
      "epoch": 4.832267761230469e-06,
      "step": 5067,
      "training_step_time": 17.123430490493774
    },
    {
      "epoch": 4.833221435546875e-06,
      "model_forward_time": 5.623514175415039,
      "step": 5068
    },
    {
      "epoch": 4.833221435546875e-06,
      "step": 5068,
      "training_step_time": 17.123534679412842
    },
    {
      "epoch": 4.834175109863281e-06,
      "model_forward_time": 5.603751182556152,
      "step": 5069
    },
    {
      "epoch": 4.834175109863281e-06,
      "step": 5069,
      "training_step_time": 17.032168865203857
    },
    {
      "epoch": 4.835128784179688e-06,
      "grad_norm": 0.9035984873771667,
      "learning_rate": 1e-05,
      "loss": 0.0591,
      "step": 5070
    },
    {
      "epoch": 4.835128784179688e-06,
      "model_forward_time": 5.917599678039551,
      "step": 5070
    },
    {
      "dynamics_loss_avg": 0.04954371061176062,
      "epoch": 4.835128784179688e-06,
      "step": 5070
    },
    {
      "action_loss_avg": 0.010812146542593836,
      "epoch": 4.835128784179688e-06,
      "step": 5070
    },
    {
      "epoch": 4.835128784179688e-06,
      "step": 5070,
      "training_step_time": 17.327699422836304
    },
    {
      "epoch": 4.8360824584960935e-06,
      "model_forward_time": 5.567014932632446,
      "step": 5071
    },
    {
      "epoch": 4.8360824584960935e-06,
      "step": 5071,
      "training_step_time": 16.986892461776733
    },
    {
      "epoch": 4.8370361328125e-06,
      "model_forward_time": 5.575095176696777,
      "step": 5072
    },
    {
      "epoch": 4.8370361328125e-06,
      "step": 5072,
      "training_step_time": 17.00423526763916
    },
    {
      "epoch": 4.837989807128907e-06,
      "model_forward_time": 5.568437099456787,
      "step": 5073
    },
    {
      "epoch": 4.837989807128907e-06,
      "step": 5073,
      "training_step_time": 17.006155252456665
    },
    {
      "epoch": 4.838943481445312e-06,
      "model_forward_time": 5.572575569152832,
      "step": 5074
    },
    {
      "epoch": 4.838943481445312e-06,
      "step": 5074,
      "training_step_time": 16.989912509918213
    },
    {
      "epoch": 4.839897155761719e-06,
      "model_forward_time": 5.944766998291016,
      "step": 5075
    },
    {
      "epoch": 4.839897155761719e-06,
      "step": 5075,
      "training_step_time": 17.37522530555725
    },
    {
      "epoch": 4.840850830078125e-06,
      "model_forward_time": 5.621720790863037,
      "step": 5076
    },
    {
      "epoch": 4.840850830078125e-06,
      "step": 5076,
      "training_step_time": 17.115610122680664
    },
    {
      "epoch": 4.841804504394531e-06,
      "model_forward_time": 5.623660564422607,
      "step": 5077
    },
    {
      "epoch": 4.841804504394531e-06,
      "step": 5077,
      "training_step_time": 17.146177053451538
    },
    {
      "epoch": 4.842758178710938e-06,
      "model_forward_time": 5.621188640594482,
      "step": 5078
    },
    {
      "epoch": 4.842758178710938e-06,
      "step": 5078,
      "training_step_time": 17.129146337509155
    },
    {
      "epoch": 4.8437118530273435e-06,
      "model_forward_time": 5.621366262435913,
      "step": 5079
    },
    {
      "epoch": 4.8437118530273435e-06,
      "step": 5079,
      "training_step_time": 17.10818862915039
    },
    {
      "epoch": 4.84466552734375e-06,
      "grad_norm": 1.4464918375015259,
      "learning_rate": 1e-05,
      "loss": 0.0707,
      "step": 5080
    },
    {
      "epoch": 4.84466552734375e-06,
      "model_forward_time": 5.603652238845825,
      "step": 5080
    },
    {
      "dynamics_loss_avg": 0.05172584038227797,
      "epoch": 4.84466552734375e-06,
      "step": 5080
    },
    {
      "action_loss_avg": 0.015527275705244393,
      "epoch": 4.84466552734375e-06,
      "step": 5080
    },
    {
      "epoch": 4.84466552734375e-06,
      "step": 5080,
      "training_step_time": 17.08230447769165
    },
    {
      "epoch": 4.845619201660157e-06,
      "model_forward_time": 5.975894212722778,
      "step": 5081
    },
    {
      "epoch": 4.845619201660157e-06,
      "step": 5081,
      "training_step_time": 17.419028282165527
    },
    {
      "epoch": 4.846572875976562e-06,
      "model_forward_time": 5.595571041107178,
      "step": 5082
    },
    {
      "epoch": 4.846572875976562e-06,
      "step": 5082,
      "training_step_time": 17.033852577209473
    },
    {
      "epoch": 4.847526550292969e-06,
      "model_forward_time": 5.592265367507935,
      "step": 5083
    },
    {
      "epoch": 4.847526550292969e-06,
      "step": 5083,
      "training_step_time": 17.017075300216675
    },
    {
      "epoch": 4.848480224609375e-06,
      "model_forward_time": 5.583538770675659,
      "step": 5084
    },
    {
      "epoch": 4.848480224609375e-06,
      "step": 5084,
      "training_step_time": 16.998484134674072
    },
    {
      "epoch": 4.849433898925781e-06,
      "model_forward_time": 5.594924211502075,
      "step": 5085
    },
    {
      "epoch": 4.849433898925781e-06,
      "step": 5085,
      "training_step_time": 17.081665992736816
    },
    {
      "epoch": 4.850387573242188e-06,
      "model_forward_time": 5.99769926071167,
      "step": 5086
    },
    {
      "epoch": 4.850387573242188e-06,
      "step": 5086,
      "training_step_time": 17.50386667251587
    },
    {
      "epoch": 4.8513412475585935e-06,
      "model_forward_time": 5.613434553146362,
      "step": 5087
    },
    {
      "epoch": 4.8513412475585935e-06,
      "step": 5087,
      "training_step_time": 17.063465356826782
    },
    {
      "epoch": 4.852294921875e-06,
      "model_forward_time": 5.577334880828857,
      "step": 5088
    },
    {
      "epoch": 4.852294921875e-06,
      "step": 5088,
      "training_step_time": 17.00607657432556
    },
    {
      "epoch": 4.853248596191407e-06,
      "model_forward_time": 5.57300591468811,
      "step": 5089
    },
    {
      "epoch": 4.853248596191407e-06,
      "step": 5089,
      "training_step_time": 17.00892949104309
    },
    {
      "epoch": 4.854202270507812e-06,
      "grad_norm": 1.544064998626709,
      "learning_rate": 1e-05,
      "loss": 0.0744,
      "step": 5090
    },
    {
      "epoch": 4.854202270507812e-06,
      "model_forward_time": 5.578315019607544,
      "step": 5090
    },
    {
      "dynamics_loss_avg": 0.0636275965720415,
      "epoch": 4.854202270507812e-06,
      "step": 5090
    },
    {
      "action_loss_avg": 0.019980391906574368,
      "epoch": 4.854202270507812e-06,
      "step": 5090
    },
    {
      "epoch": 4.854202270507812e-06,
      "step": 5090,
      "training_step_time": 16.997668504714966
    },
    {
      "epoch": 4.855155944824219e-06,
      "model_forward_time": 5.576491117477417,
      "step": 5091
    },
    {
      "epoch": 4.855155944824219e-06,
      "step": 5091,
      "training_step_time": 16.989574432373047
    },
    {
      "epoch": 4.856109619140625e-06,
      "model_forward_time": 5.922644853591919,
      "step": 5092
    },
    {
      "epoch": 4.856109619140625e-06,
      "step": 5092,
      "training_step_time": 17.364402294158936
    },
    {
      "epoch": 4.857063293457031e-06,
      "model_forward_time": 5.580432415008545,
      "step": 5093
    },
    {
      "epoch": 4.857063293457031e-06,
      "step": 5093,
      "training_step_time": 16.994952917099
    },
    {
      "epoch": 4.858016967773438e-06,
      "model_forward_time": 5.579326391220093,
      "step": 5094
    },
    {
      "epoch": 4.858016967773438e-06,
      "step": 5094,
      "training_step_time": 17.00030827522278
    },
    {
      "epoch": 4.8589706420898435e-06,
      "model_forward_time": 5.5905609130859375,
      "step": 5095
    },
    {
      "epoch": 4.8589706420898435e-06,
      "step": 5095,
      "training_step_time": 17.03841996192932
    },
    {
      "epoch": 4.85992431640625e-06,
      "model_forward_time": 5.598206043243408,
      "step": 5096
    },
    {
      "epoch": 4.85992431640625e-06,
      "step": 5096,
      "training_step_time": 17.063082933425903
    },
    {
      "epoch": 4.860877990722657e-06,
      "model_forward_time": 5.965644836425781,
      "step": 5097
    },
    {
      "epoch": 4.860877990722657e-06,
      "step": 5097,
      "training_step_time": 17.462642192840576
    },
    {
      "epoch": 4.861831665039062e-06,
      "model_forward_time": 5.612056732177734,
      "step": 5098
    },
    {
      "epoch": 4.861831665039062e-06,
      "step": 5098,
      "training_step_time": 17.102643966674805
    },
    {
      "epoch": 4.862785339355469e-06,
      "model_forward_time": 5.603413343429565,
      "step": 5099
    },
    {
      "epoch": 4.862785339355469e-06,
      "step": 5099,
      "training_step_time": 17.10294795036316
    },
    {
      "epoch": 4.863739013671875e-06,
      "grad_norm": 0.7075799703598022,
      "learning_rate": 1e-05,
      "loss": 0.0646,
      "step": 5100
    },
    {
      "epoch": 4.863739013671875e-06,
      "model_forward_time": 5.613858461380005,
      "step": 5100
    },
    {
      "dynamics_loss_avg": 0.05811207108199597,
      "epoch": 4.863739013671875e-06,
      "step": 5100
    },
    {
      "action_loss_avg": 0.00857482322026044,
      "epoch": 4.863739013671875e-06,
      "step": 5100
    },
    {
      "epoch": 4.863739013671875e-06,
      "step": 5100,
      "training_step_time": 17.075212717056274
    },
    {
      "epoch": 4.864692687988281e-06,
      "model_forward_time": 5.604267835617065,
      "step": 5101
    },
    {
      "epoch": 4.864692687988281e-06,
      "step": 5101,
      "training_step_time": 17.07715344429016
    },
    {
      "epoch": 4.865646362304688e-06,
      "model_forward_time": 5.952087163925171,
      "step": 5102
    },
    {
      "epoch": 4.865646362304688e-06,
      "step": 5102,
      "training_step_time": 17.406389474868774
    },
    {
      "epoch": 4.8666000366210935e-06,
      "model_forward_time": 5.600237607955933,
      "step": 5103
    },
    {
      "epoch": 4.8666000366210935e-06,
      "step": 5103,
      "training_step_time": 17.05092453956604
    },
    {
      "epoch": 4.8675537109375e-06,
      "model_forward_time": 5.581114053726196,
      "step": 5104
    },
    {
      "epoch": 4.8675537109375e-06,
      "step": 5104,
      "training_step_time": 17.031750917434692
    },
    {
      "epoch": 4.868507385253907e-06,
      "model_forward_time": 5.597061395645142,
      "step": 5105
    },
    {
      "epoch": 4.868507385253907e-06,
      "step": 5105,
      "training_step_time": 17.078463554382324
    },
    {
      "epoch": 4.869461059570312e-06,
      "model_forward_time": 5.609590291976929,
      "step": 5106
    },
    {
      "epoch": 4.869461059570312e-06,
      "step": 5106,
      "training_step_time": 17.114772081375122
    },
    {
      "epoch": 4.870414733886719e-06,
      "model_forward_time": 5.616886854171753,
      "step": 5107
    },
    {
      "epoch": 4.870414733886719e-06,
      "step": 5107,
      "training_step_time": 17.10267925262451
    },
    {
      "epoch": 4.871368408203125e-06,
      "model_forward_time": 5.96495509147644,
      "step": 5108
    },
    {
      "epoch": 4.871368408203125e-06,
      "step": 5108,
      "training_step_time": 17.40820288658142
    },
    {
      "epoch": 4.872322082519531e-06,
      "model_forward_time": 5.575679540634155,
      "step": 5109
    },
    {
      "epoch": 4.872322082519531e-06,
      "step": 5109,
      "training_step_time": 16.997960805892944
    },
    {
      "epoch": 4.873275756835938e-06,
      "grad_norm": 1.0322750806808472,
      "learning_rate": 1e-05,
      "loss": 0.0881,
      "step": 5110
    },
    {
      "epoch": 4.873275756835938e-06,
      "model_forward_time": 5.580905914306641,
      "step": 5110
    },
    {
      "dynamics_loss_avg": 0.06246923692524433,
      "epoch": 4.873275756835938e-06,
      "step": 5110
    },
    {
      "action_loss_avg": 0.014143221033737063,
      "epoch": 4.873275756835938e-06,
      "step": 5110
    },
    {
      "epoch": 4.873275756835938e-06,
      "step": 5110,
      "training_step_time": 17.01304006576538
    },
    {
      "epoch": 4.8742294311523435e-06,
      "model_forward_time": 5.588872194290161,
      "step": 5111
    },
    {
      "epoch": 4.8742294311523435e-06,
      "step": 5111,
      "training_step_time": 17.048959732055664
    },
    {
      "epoch": 4.87518310546875e-06,
      "model_forward_time": 5.623136758804321,
      "step": 5112
    },
    {
      "epoch": 4.87518310546875e-06,
      "step": 5112,
      "training_step_time": 17.114757776260376
    },
    {
      "epoch": 4.876136779785157e-06,
      "model_forward_time": 5.59866738319397,
      "step": 5113
    },
    {
      "epoch": 4.876136779785157e-06,
      "step": 5113,
      "training_step_time": 17.419631958007812
    },
    {
      "epoch": 4.877090454101562e-06,
      "model_forward_time": 5.5908074378967285,
      "step": 5114
    },
    {
      "epoch": 4.877090454101562e-06,
      "step": 5114,
      "training_step_time": 17.02989387512207
    },
    {
      "epoch": 4.878044128417969e-06,
      "model_forward_time": 5.5940210819244385,
      "step": 5115
    },
    {
      "epoch": 4.878044128417969e-06,
      "step": 5115,
      "training_step_time": 17.04066753387451
    },
    {
      "epoch": 4.878997802734375e-06,
      "model_forward_time": 5.581331491470337,
      "step": 5116
    },
    {
      "epoch": 4.878997802734375e-06,
      "step": 5116,
      "training_step_time": 17.04030132293701
    },
    {
      "epoch": 4.879951477050781e-06,
      "model_forward_time": 5.593717336654663,
      "step": 5117
    },
    {
      "epoch": 4.879951477050781e-06,
      "step": 5117,
      "training_step_time": 17.061394453048706
    },
    {
      "epoch": 4.880905151367188e-06,
      "model_forward_time": 5.603939056396484,
      "step": 5118
    },
    {
      "epoch": 4.880905151367188e-06,
      "step": 5118,
      "training_step_time": 17.102845907211304
    },
    {
      "epoch": 4.8818588256835935e-06,
      "model_forward_time": 5.982333421707153,
      "step": 5119
    },
    {
      "epoch": 4.8818588256835935e-06,
      "step": 5119,
      "training_step_time": 17.477620601654053
    },
    {
      "epoch": 4.8828125e-06,
      "grad_norm": 0.6096917390823364,
      "learning_rate": 1e-05,
      "loss": 0.0792,
      "step": 5120
    },
    {
      "epoch": 4.8828125e-06,
      "model_forward_time": 5.611456394195557,
      "step": 5120
    },
    {
      "dynamics_loss_avg": 0.06846248246729374,
      "epoch": 4.8828125e-06,
      "step": 5120
    },
    {
      "action_loss_avg": 0.017272510332986712,
      "epoch": 4.8828125e-06,
      "step": 5120
    },
    {
      "epoch": 4.8828125e-06,
      "step": 5120,
      "training_step_time": 17.114007711410522
    },
    {
      "epoch": 4.883766174316407e-06,
      "model_forward_time": 5.618322372436523,
      "step": 5121
    },
    {
      "epoch": 4.883766174316407e-06,
      "step": 5121,
      "training_step_time": 17.115060329437256
    },
    {
      "epoch": 4.884719848632812e-06,
      "model_forward_time": 5.617149353027344,
      "step": 5122
    },
    {
      "epoch": 4.884719848632812e-06,
      "step": 5122,
      "training_step_time": 17.09393858909607
    },
    {
      "epoch": 4.885673522949219e-06,
      "model_forward_time": 5.577929496765137,
      "step": 5123
    },
    {
      "epoch": 4.885673522949219e-06,
      "step": 5123,
      "training_step_time": 17.022104263305664
    },
    {
      "epoch": 4.886627197265625e-06,
      "model_forward_time": 5.573827266693115,
      "step": 5124
    },
    {
      "epoch": 4.886627197265625e-06,
      "step": 5124,
      "training_step_time": 17.356797456741333
    },
    {
      "epoch": 4.887580871582031e-06,
      "model_forward_time": 5.57215428352356,
      "step": 5125
    },
    {
      "epoch": 4.887580871582031e-06,
      "step": 5125,
      "training_step_time": 16.978792905807495
    },
    {
      "epoch": 4.888534545898438e-06,
      "model_forward_time": 5.565740346908569,
      "step": 5126
    },
    {
      "epoch": 4.888534545898438e-06,
      "step": 5126,
      "training_step_time": 16.99771761894226
    },
    {
      "epoch": 4.8894882202148435e-06,
      "model_forward_time": 5.583043813705444,
      "step": 5127
    },
    {
      "epoch": 4.8894882202148435e-06,
      "step": 5127,
      "training_step_time": 17.00621795654297
    },
    {
      "epoch": 4.89044189453125e-06,
      "model_forward_time": 5.579161167144775,
      "step": 5128
    },
    {
      "epoch": 4.89044189453125e-06,
      "step": 5128,
      "training_step_time": 17.000503301620483
    },
    {
      "epoch": 4.891395568847656e-06,
      "model_forward_time": 5.58742618560791,
      "step": 5129
    },
    {
      "epoch": 4.891395568847656e-06,
      "step": 5129,
      "training_step_time": 17.002933979034424
    },
    {
      "epoch": 4.892349243164062e-06,
      "grad_norm": 1.023162603378296,
      "learning_rate": 1e-05,
      "loss": 0.0693,
      "step": 5130
    },
    {
      "epoch": 4.892349243164062e-06,
      "model_forward_time": 5.957438707351685,
      "step": 5130
    },
    {
      "dynamics_loss_avg": 0.05287207402288914,
      "epoch": 4.892349243164062e-06,
      "step": 5130
    },
    {
      "action_loss_avg": 0.011039072601124644,
      "epoch": 4.892349243164062e-06,
      "step": 5130
    },
    {
      "epoch": 4.892349243164062e-06,
      "step": 5130,
      "training_step_time": 17.395516395568848
    },
    {
      "epoch": 4.893302917480469e-06,
      "model_forward_time": 5.605548143386841,
      "step": 5131
    },
    {
      "epoch": 4.893302917480469e-06,
      "step": 5131,
      "training_step_time": 17.10843253135681
    },
    {
      "epoch": 4.894256591796875e-06,
      "model_forward_time": 5.618453502655029,
      "step": 5132
    },
    {
      "epoch": 4.894256591796875e-06,
      "step": 5132,
      "training_step_time": 17.134608030319214
    },
    {
      "epoch": 4.895210266113281e-06,
      "model_forward_time": 5.614565849304199,
      "step": 5133
    },
    {
      "epoch": 4.895210266113281e-06,
      "step": 5133,
      "training_step_time": 17.081180334091187
    },
    {
      "epoch": 4.896163940429688e-06,
      "model_forward_time": 5.583620548248291,
      "step": 5134
    },
    {
      "epoch": 4.896163940429688e-06,
      "step": 5134,
      "training_step_time": 17.018927574157715
    },
    {
      "epoch": 4.8971176147460935e-06,
      "model_forward_time": 5.9179441928863525,
      "step": 5135
    },
    {
      "epoch": 4.8971176147460935e-06,
      "step": 5135,
      "training_step_time": 17.33046865463257
    },
    {
      "epoch": 4.8980712890625e-06,
      "model_forward_time": 5.567392826080322,
      "step": 5136
    },
    {
      "epoch": 4.8980712890625e-06,
      "step": 5136,
      "training_step_time": 16.98869776725769
    },
    {
      "epoch": 4.899024963378906e-06,
      "model_forward_time": 5.561444044113159,
      "step": 5137
    },
    {
      "epoch": 4.899024963378906e-06,
      "step": 5137,
      "training_step_time": 16.987720727920532
    },
    {
      "epoch": 4.899978637695312e-06,
      "model_forward_time": 5.566317558288574,
      "step": 5138
    },
    {
      "epoch": 4.899978637695312e-06,
      "step": 5138,
      "training_step_time": 16.999387502670288
    },
    {
      "epoch": 4.900932312011719e-06,
      "model_forward_time": 5.579204797744751,
      "step": 5139
    },
    {
      "epoch": 4.900932312011719e-06,
      "step": 5139,
      "training_step_time": 16.998117685317993
    },
    {
      "epoch": 4.901885986328125e-06,
      "grad_norm": 0.2319028675556183,
      "learning_rate": 1e-05,
      "loss": 0.0649,
      "step": 5140
    },
    {
      "epoch": 4.901885986328125e-06,
      "model_forward_time": 5.5871312618255615,
      "step": 5140
    },
    {
      "dynamics_loss_avg": 0.04984937570989132,
      "epoch": 4.901885986328125e-06,
      "step": 5140
    },
    {
      "action_loss_avg": 0.016728770965710283,
      "epoch": 4.901885986328125e-06,
      "step": 5140
    },
    {
      "epoch": 4.901885986328125e-06,
      "step": 5140,
      "training_step_time": 17.009153604507446
    },
    {
      "epoch": 4.902839660644531e-06,
      "model_forward_time": 5.947953701019287,
      "step": 5141
    },
    {
      "epoch": 4.902839660644531e-06,
      "step": 5141,
      "training_step_time": 17.376618146896362
    },
    {
      "epoch": 4.903793334960938e-06,
      "model_forward_time": 5.602794408798218,
      "step": 5142
    },
    {
      "epoch": 4.903793334960938e-06,
      "step": 5142,
      "training_step_time": 17.054340600967407
    },
    {
      "epoch": 4.9047470092773436e-06,
      "model_forward_time": 5.602181673049927,
      "step": 5143
    },
    {
      "epoch": 4.9047470092773436e-06,
      "step": 5143,
      "training_step_time": 17.089839935302734
    },
    {
      "epoch": 4.90570068359375e-06,
      "model_forward_time": 5.610637664794922,
      "step": 5144
    },
    {
      "epoch": 4.90570068359375e-06,
      "step": 5144,
      "training_step_time": 17.12178063392639
    },
    {
      "epoch": 4.906654357910156e-06,
      "model_forward_time": 5.631977081298828,
      "step": 5145
    },
    {
      "epoch": 4.906654357910156e-06,
      "step": 5145,
      "training_step_time": 17.145546436309814
    },
    {
      "epoch": 4.907608032226562e-06,
      "model_forward_time": 5.614740371704102,
      "step": 5146
    },
    {
      "epoch": 4.907608032226562e-06,
      "step": 5146,
      "training_step_time": 17.470781803131104
    },
    {
      "epoch": 4.908561706542969e-06,
      "model_forward_time": 5.603461742401123,
      "step": 5147
    },
    {
      "epoch": 4.908561706542969e-06,
      "step": 5147,
      "training_step_time": 17.073797464370728
    },
    {
      "epoch": 4.909515380859375e-06,
      "model_forward_time": 5.6131346225738525,
      "step": 5148
    },
    {
      "epoch": 4.909515380859375e-06,
      "step": 5148,
      "training_step_time": 17.105628728866577
    },
    {
      "epoch": 4.910469055175781e-06,
      "model_forward_time": 5.60787034034729,
      "step": 5149
    },
    {
      "epoch": 4.910469055175781e-06,
      "step": 5149,
      "training_step_time": 17.087761640548706
    },
    {
      "epoch": 4.911422729492188e-06,
      "grad_norm": 1.1977136135101318,
      "learning_rate": 1e-05,
      "loss": 0.0624,
      "step": 5150
    },
    {
      "epoch": 4.911422729492188e-06,
      "model_forward_time": 5.599932432174683,
      "step": 5150
    },
    {
      "dynamics_loss_avg": 0.05344246942549944,
      "epoch": 4.911422729492188e-06,
      "step": 5150
    },
    {
      "action_loss_avg": 0.011324766231700778,
      "epoch": 4.911422729492188e-06,
      "step": 5150
    },
    {
      "epoch": 4.911422729492188e-06,
      "step": 5150,
      "training_step_time": 17.055426836013794
    },
    {
      "epoch": 4.9123764038085936e-06,
      "model_forward_time": 5.592911720275879,
      "step": 5151
    },
    {
      "epoch": 4.9123764038085936e-06,
      "step": 5151,
      "training_step_time": 17.021366357803345
    },
    {
      "epoch": 4.913330078125e-06,
      "model_forward_time": 5.96009087562561,
      "step": 5152
    },
    {
      "epoch": 4.913330078125e-06,
      "step": 5152,
      "training_step_time": 17.406407117843628
    },
    {
      "epoch": 4.914283752441406e-06,
      "model_forward_time": 5.59324836730957,
      "step": 5153
    },
    {
      "epoch": 4.914283752441406e-06,
      "step": 5153,
      "training_step_time": 17.058204889297485
    },
    {
      "epoch": 4.915237426757812e-06,
      "model_forward_time": 5.601041316986084,
      "step": 5154
    },
    {
      "epoch": 4.915237426757812e-06,
      "step": 5154,
      "training_step_time": 17.07309579849243
    },
    {
      "epoch": 4.916191101074219e-06,
      "model_forward_time": 5.613020658493042,
      "step": 5155
    },
    {
      "epoch": 4.916191101074219e-06,
      "step": 5155,
      "training_step_time": 17.088953018188477
    },
    {
      "epoch": 4.917144775390625e-06,
      "model_forward_time": 5.601679801940918,
      "step": 5156
    },
    {
      "epoch": 4.917144775390625e-06,
      "step": 5156,
      "training_step_time": 17.081656217575073
    },
    {
      "epoch": 4.918098449707031e-06,
      "model_forward_time": 5.607428312301636,
      "step": 5157
    },
    {
      "epoch": 4.918098449707031e-06,
      "step": 5157,
      "training_step_time": 17.45278024673462
    },
    {
      "epoch": 4.919052124023438e-06,
      "model_forward_time": 5.609959840774536,
      "step": 5158
    },
    {
      "epoch": 4.919052124023438e-06,
      "step": 5158,
      "training_step_time": 17.073415517807007
    },
    {
      "epoch": 4.9200057983398436e-06,
      "model_forward_time": 5.6016693115234375,
      "step": 5159
    },
    {
      "epoch": 4.9200057983398436e-06,
      "step": 5159,
      "training_step_time": 17.047179698944092
    },
    {
      "epoch": 4.92095947265625e-06,
      "grad_norm": 1.6884318590164185,
      "learning_rate": 1e-05,
      "loss": 0.0754,
      "step": 5160
    },
    {
      "epoch": 4.92095947265625e-06,
      "model_forward_time": 5.587618589401245,
      "step": 5160
    },
    {
      "dynamics_loss_avg": 0.062650316208601,
      "epoch": 4.92095947265625e-06,
      "step": 5160
    },
    {
      "action_loss_avg": 0.013774566072970628,
      "epoch": 4.92095947265625e-06,
      "step": 5160
    },
    {
      "epoch": 4.92095947265625e-06,
      "step": 5160,
      "training_step_time": 17.013051986694336
    },
    {
      "epoch": 4.921913146972656e-06,
      "model_forward_time": 5.577982187271118,
      "step": 5161
    },
    {
      "epoch": 4.921913146972656e-06,
      "step": 5161,
      "training_step_time": 17.019488096237183
    },
    {
      "epoch": 4.9228668212890624e-06,
      "model_forward_time": 5.5792927742004395,
      "step": 5162
    },
    {
      "epoch": 4.9228668212890624e-06,
      "step": 5162,
      "training_step_time": 17.02184247970581
    },
    {
      "epoch": 4.923820495605469e-06,
      "model_forward_time": 5.968492031097412,
      "step": 5163
    },
    {
      "epoch": 4.923820495605469e-06,
      "step": 5163,
      "training_step_time": 17.44148349761963
    },
    {
      "epoch": 4.924774169921875e-06,
      "model_forward_time": 5.612577199935913,
      "step": 5164
    },
    {
      "epoch": 4.924774169921875e-06,
      "step": 5164,
      "training_step_time": 17.104930877685547
    },
    {
      "epoch": 4.925727844238281e-06,
      "model_forward_time": 5.612282991409302,
      "step": 5165
    },
    {
      "epoch": 4.925727844238281e-06,
      "step": 5165,
      "training_step_time": 17.11125636100769
    },
    {
      "epoch": 4.926681518554688e-06,
      "model_forward_time": 5.616190195083618,
      "step": 5166
    },
    {
      "epoch": 4.926681518554688e-06,
      "step": 5166,
      "training_step_time": 17.10503101348877
    },
    {
      "epoch": 4.9276351928710936e-06,
      "model_forward_time": 5.602885484695435,
      "step": 5167
    },
    {
      "epoch": 4.9276351928710936e-06,
      "step": 5167,
      "training_step_time": 17.071565628051758
    },
    {
      "epoch": 4.9285888671875e-06,
      "model_forward_time": 5.947601795196533,
      "step": 5168
    },
    {
      "epoch": 4.9285888671875e-06,
      "step": 5168,
      "training_step_time": 17.388089656829834
    },
    {
      "epoch": 4.929542541503906e-06,
      "model_forward_time": 5.59513783454895,
      "step": 5169
    },
    {
      "epoch": 4.929542541503906e-06,
      "step": 5169,
      "training_step_time": 17.077460765838623
    },
    {
      "epoch": 4.9304962158203124e-06,
      "grad_norm": 0.5914214849472046,
      "learning_rate": 1e-05,
      "loss": 0.0904,
      "step": 5170
    },
    {
      "epoch": 4.9304962158203124e-06,
      "model_forward_time": 5.592893362045288,
      "step": 5170
    },
    {
      "dynamics_loss_avg": 0.07060779090970755,
      "epoch": 4.9304962158203124e-06,
      "step": 5170
    },
    {
      "action_loss_avg": 0.01728312112390995,
      "epoch": 4.9304962158203124e-06,
      "step": 5170
    },
    {
      "epoch": 4.9304962158203124e-06,
      "step": 5170,
      "training_step_time": 17.058411121368408
    },
    {
      "epoch": 4.931449890136719e-06,
      "model_forward_time": 5.602463483810425,
      "step": 5171
    },
    {
      "epoch": 4.931449890136719e-06,
      "step": 5171,
      "training_step_time": 17.064629793167114
    },
    {
      "epoch": 4.932403564453125e-06,
      "model_forward_time": 5.592321157455444,
      "step": 5172
    },
    {
      "epoch": 4.932403564453125e-06,
      "step": 5172,
      "training_step_time": 17.05902624130249
    },
    {
      "epoch": 4.933357238769531e-06,
      "model_forward_time": 5.593646049499512,
      "step": 5173
    },
    {
      "epoch": 4.933357238769531e-06,
      "step": 5173,
      "training_step_time": 17.046022653579712
    },
    {
      "epoch": 4.934310913085938e-06,
      "model_forward_time": 5.953160524368286,
      "step": 5174
    },
    {
      "epoch": 4.934310913085938e-06,
      "step": 5174,
      "training_step_time": 17.38616704940796
    },
    {
      "epoch": 4.935264587402344e-06,
      "model_forward_time": 5.594013214111328,
      "step": 5175
    },
    {
      "epoch": 4.935264587402344e-06,
      "step": 5175,
      "training_step_time": 17.052658081054688
    },
    {
      "epoch": 4.93621826171875e-06,
      "model_forward_time": 5.607303142547607,
      "step": 5176
    },
    {
      "epoch": 4.93621826171875e-06,
      "step": 5176,
      "training_step_time": 17.091075658798218
    },
    {
      "epoch": 4.937171936035156e-06,
      "model_forward_time": 5.611479043960571,
      "step": 5177
    },
    {
      "epoch": 4.937171936035156e-06,
      "step": 5177,
      "training_step_time": 17.12343168258667
    },
    {
      "epoch": 4.9381256103515624e-06,
      "model_forward_time": 5.616057395935059,
      "step": 5178
    },
    {
      "epoch": 4.9381256103515624e-06,
      "step": 5178,
      "training_step_time": 17.13387441635132
    },
    {
      "epoch": 4.939079284667969e-06,
      "model_forward_time": 5.619781970977783,
      "step": 5179
    },
    {
      "epoch": 4.939079284667969e-06,
      "step": 5179,
      "training_step_time": 17.491085529327393
    },
    {
      "epoch": 4.940032958984375e-06,
      "grad_norm": 1.2654922008514404,
      "learning_rate": 1e-05,
      "loss": 0.0675,
      "step": 5180
    },
    {
      "epoch": 4.940032958984375e-06,
      "model_forward_time": 5.617419719696045,
      "step": 5180
    },
    {
      "dynamics_loss_avg": 0.05475857006385922,
      "epoch": 4.940032958984375e-06,
      "step": 5180
    },
    {
      "action_loss_avg": 0.011274072714149952,
      "epoch": 4.940032958984375e-06,
      "step": 5180
    },
    {
      "epoch": 4.940032958984375e-06,
      "step": 5180,
      "training_step_time": 17.09487795829773
    },
    {
      "epoch": 4.940986633300781e-06,
      "model_forward_time": 5.600823640823364,
      "step": 5181
    },
    {
      "epoch": 4.940986633300781e-06,
      "step": 5181,
      "training_step_time": 17.05899453163147
    },
    {
      "epoch": 4.941940307617188e-06,
      "model_forward_time": 5.5963099002838135,
      "step": 5182
    },
    {
      "epoch": 4.941940307617188e-06,
      "step": 5182,
      "training_step_time": 17.04914665222168
    },
    {
      "epoch": 4.942893981933594e-06,
      "model_forward_time": 5.606977701187134,
      "step": 5183
    },
    {
      "epoch": 4.942893981933594e-06,
      "step": 5183,
      "training_step_time": 17.079759120941162
    },
    {
      "epoch": 4.94384765625e-06,
      "model_forward_time": 5.617425441741943,
      "step": 5184
    },
    {
      "epoch": 4.94384765625e-06,
      "step": 5184,
      "training_step_time": 17.117199420928955
    },
    {
      "epoch": 4.944801330566406e-06,
      "model_forward_time": 5.986071586608887,
      "step": 5185
    },
    {
      "epoch": 4.944801330566406e-06,
      "step": 5185,
      "training_step_time": 17.47279644012451
    },
    {
      "epoch": 4.9457550048828124e-06,
      "model_forward_time": 5.606670618057251,
      "step": 5186
    },
    {
      "epoch": 4.9457550048828124e-06,
      "step": 5186,
      "training_step_time": 17.07427167892456
    },
    {
      "epoch": 4.946708679199219e-06,
      "model_forward_time": 5.6019721031188965,
      "step": 5187
    },
    {
      "epoch": 4.946708679199219e-06,
      "step": 5187,
      "training_step_time": 17.032769918441772
    },
    {
      "epoch": 4.947662353515625e-06,
      "model_forward_time": 5.593580484390259,
      "step": 5188
    },
    {
      "epoch": 4.947662353515625e-06,
      "step": 5188,
      "training_step_time": 17.036593198776245
    },
    {
      "epoch": 4.948616027832031e-06,
      "model_forward_time": 5.593013286590576,
      "step": 5189
    },
    {
      "epoch": 4.948616027832031e-06,
      "step": 5189,
      "training_step_time": 17.053586721420288
    },
    {
      "epoch": 4.949569702148438e-06,
      "grad_norm": 0.19561725854873657,
      "learning_rate": 1e-05,
      "loss": 0.0704,
      "step": 5190
    },
    {
      "epoch": 4.949569702148438e-06,
      "model_forward_time": 5.606114387512207,
      "step": 5190
    },
    {
      "dynamics_loss_avg": 0.0564084243029356,
      "epoch": 4.949569702148438e-06,
      "step": 5190
    },
    {
      "action_loss_avg": 0.019937998231034725,
      "epoch": 4.949569702148438e-06,
      "step": 5190
    },
    {
      "epoch": 4.949569702148438e-06,
      "step": 5190,
      "training_step_time": 17.46962881088257
    },
    {
      "epoch": 4.950523376464844e-06,
      "model_forward_time": 5.608614683151245,
      "step": 5191
    },
    {
      "epoch": 4.950523376464844e-06,
      "step": 5191,
      "training_step_time": 17.109153509140015
    },
    {
      "epoch": 4.95147705078125e-06,
      "model_forward_time": 5.615094423294067,
      "step": 5192
    },
    {
      "epoch": 4.95147705078125e-06,
      "step": 5192,
      "training_step_time": 17.116265535354614
    },
    {
      "epoch": 4.952430725097656e-06,
      "model_forward_time": 5.607375860214233,
      "step": 5193
    },
    {
      "epoch": 4.952430725097656e-06,
      "step": 5193,
      "training_step_time": 17.09032702445984
    },
    {
      "epoch": 4.9533843994140625e-06,
      "model_forward_time": 5.6117706298828125,
      "step": 5194
    },
    {
      "epoch": 4.9533843994140625e-06,
      "step": 5194,
      "training_step_time": 17.099541664123535
    },
    {
      "epoch": 4.954338073730469e-06,
      "model_forward_time": 5.604357719421387,
      "step": 5195
    },
    {
      "epoch": 4.954338073730469e-06,
      "step": 5195,
      "training_step_time": 17.074665307998657
    },
    {
      "epoch": 4.955291748046875e-06,
      "model_forward_time": 5.964958906173706,
      "step": 5196
    },
    {
      "epoch": 4.955291748046875e-06,
      "step": 5196,
      "training_step_time": 17.433370351791382
    },
    {
      "epoch": 4.956245422363281e-06,
      "model_forward_time": 5.602180004119873,
      "step": 5197
    },
    {
      "epoch": 4.956245422363281e-06,
      "step": 5197,
      "training_step_time": 17.076682090759277
    },
    {
      "epoch": 4.957199096679688e-06,
      "model_forward_time": 5.5938990116119385,
      "step": 5198
    },
    {
      "epoch": 4.957199096679688e-06,
      "step": 5198,
      "training_step_time": 17.038967609405518
    },
    {
      "epoch": 4.958152770996094e-06,
      "model_forward_time": 5.582334041595459,
      "step": 5199
    },
    {
      "epoch": 4.958152770996094e-06,
      "step": 5199,
      "training_step_time": 17.02403450012207
    },
    {
      "epoch": 4.9591064453125e-06,
      "grad_norm": 0.3125835359096527,
      "learning_rate": 1e-05,
      "loss": 0.0892,
      "step": 5200
    },
    {
      "epoch": 4.9591064453125e-06,
      "model_forward_time": 5.581985712051392,
      "step": 5200
    },
    {
      "dynamics_loss_avg": 0.06636887118220329,
      "epoch": 4.9591064453125e-06,
      "step": 5200
    },
    {
      "action_loss_avg": 0.016834179079160093,
      "epoch": 4.9591064453125e-06,
      "step": 5200
    },
    {
      "epoch": 4.9591064453125e-06,
      "step": 5200,
      "training_step_time": 17.031616926193237
    },
    {
      "epoch": 4.960060119628906e-06,
      "model_forward_time": 5.960198163986206,
      "step": 5201
    },
    {
      "epoch": 4.960060119628906e-06,
      "step": 5201,
      "training_step_time": 17.431369066238403
    },
    {
      "epoch": 4.9610137939453125e-06,
      "model_forward_time": 5.6109185218811035,
      "step": 5202
    },
    {
      "epoch": 4.9610137939453125e-06,
      "step": 5202,
      "training_step_time": 17.098640203475952
    },
    {
      "epoch": 4.961967468261719e-06,
      "model_forward_time": 5.615347862243652,
      "step": 5203
    },
    {
      "epoch": 4.961967468261719e-06,
      "step": 5203,
      "training_step_time": 17.09715747833252
    },
    {
      "epoch": 4.962921142578125e-06,
      "model_forward_time": 5.601678848266602,
      "step": 5204
    },
    {
      "epoch": 4.962921142578125e-06,
      "step": 5204,
      "training_step_time": 17.07309579849243
    },
    {
      "epoch": 4.963874816894531e-06,
      "model_forward_time": 5.608283758163452,
      "step": 5205
    },
    {
      "epoch": 4.963874816894531e-06,
      "step": 5205,
      "training_step_time": 17.09625816345215
    },
    {
      "epoch": 4.964828491210938e-06,
      "model_forward_time": 5.605192422866821,
      "step": 5206
    },
    {
      "epoch": 4.964828491210938e-06,
      "step": 5206,
      "training_step_time": 17.096978425979614
    },
    {
      "epoch": 4.965782165527344e-06,
      "model_forward_time": 5.976611852645874,
      "step": 5207
    },
    {
      "epoch": 4.965782165527344e-06,
      "step": 5207,
      "training_step_time": 17.47946262359619
    },
    {
      "epoch": 4.96673583984375e-06,
      "model_forward_time": 5.613222360610962,
      "step": 5208
    },
    {
      "epoch": 4.96673583984375e-06,
      "step": 5208,
      "training_step_time": 17.115434169769287
    },
    {
      "epoch": 4.967689514160156e-06,
      "model_forward_time": 5.613817453384399,
      "step": 5209
    },
    {
      "epoch": 4.967689514160156e-06,
      "step": 5209,
      "training_step_time": 17.116305828094482
    },
    {
      "epoch": 4.9686431884765625e-06,
      "grad_norm": 2.4608380794525146,
      "learning_rate": 1e-05,
      "loss": 0.0841,
      "step": 5210
    },
    {
      "epoch": 4.9686431884765625e-06,
      "model_forward_time": 5.618962287902832,
      "step": 5210
    },
    {
      "dynamics_loss_avg": 0.06992146372795105,
      "epoch": 4.9686431884765625e-06,
      "step": 5210
    },
    {
      "action_loss_avg": 0.02014941070228815,
      "epoch": 4.9686431884765625e-06,
      "step": 5210
    },
    {
      "epoch": 4.9686431884765625e-06,
      "step": 5210,
      "training_step_time": 17.108774662017822
    },
    {
      "epoch": 4.969596862792969e-06,
      "model_forward_time": 5.6156158447265625,
      "step": 5211
    },
    {
      "epoch": 4.969596862792969e-06,
      "step": 5211,
      "training_step_time": 17.120219945907593
    },
    {
      "epoch": 4.970550537109375e-06,
      "model_forward_time": 5.608299016952515,
      "step": 5212
    },
    {
      "epoch": 4.970550537109375e-06,
      "step": 5212,
      "training_step_time": 17.442963361740112
    },
    {
      "epoch": 4.971504211425781e-06,
      "model_forward_time": 5.599076271057129,
      "step": 5213
    },
    {
      "epoch": 4.971504211425781e-06,
      "step": 5213,
      "training_step_time": 17.062543153762817
    },
    {
      "epoch": 4.972457885742188e-06,
      "model_forward_time": 5.596391916275024,
      "step": 5214
    },
    {
      "epoch": 4.972457885742188e-06,
      "step": 5214,
      "training_step_time": 17.032827377319336
    },
    {
      "epoch": 4.973411560058594e-06,
      "model_forward_time": 5.597520589828491,
      "step": 5215
    },
    {
      "epoch": 4.973411560058594e-06,
      "step": 5215,
      "training_step_time": 17.077377557754517
    },
    {
      "epoch": 4.974365234375e-06,
      "model_forward_time": 5.597683906555176,
      "step": 5216
    },
    {
      "epoch": 4.974365234375e-06,
      "step": 5216,
      "training_step_time": 17.061919927597046
    },
    {
      "epoch": 4.975318908691406e-06,
      "model_forward_time": 5.6006529331207275,
      "step": 5217
    },
    {
      "epoch": 4.975318908691406e-06,
      "step": 5217,
      "training_step_time": 17.055979013442993
    },
    {
      "epoch": 4.9762725830078125e-06,
      "model_forward_time": 5.973298072814941,
      "step": 5218
    },
    {
      "epoch": 4.9762725830078125e-06,
      "step": 5218,
      "training_step_time": 17.404768228530884
    },
    {
      "epoch": 4.977226257324219e-06,
      "model_forward_time": 5.594680070877075,
      "step": 5219
    },
    {
      "epoch": 4.977226257324219e-06,
      "step": 5219,
      "training_step_time": 17.02420210838318
    },
    {
      "epoch": 4.978179931640625e-06,
      "grad_norm": 0.9907560348510742,
      "learning_rate": 1e-05,
      "loss": 0.0582,
      "step": 5220
    },
    {
      "epoch": 4.978179931640625e-06,
      "model_forward_time": 5.591621398925781,
      "step": 5220
    },
    {
      "dynamics_loss_avg": 0.03893490824848413,
      "epoch": 4.978179931640625e-06,
      "step": 5220
    },
    {
      "action_loss_avg": 0.011337365908548235,
      "epoch": 4.978179931640625e-06,
      "step": 5220
    },
    {
      "epoch": 4.978179931640625e-06,
      "step": 5220,
      "training_step_time": 17.042336463928223
    },
    {
      "epoch": 4.979133605957031e-06,
      "model_forward_time": 5.595426559448242,
      "step": 5221
    },
    {
      "epoch": 4.979133605957031e-06,
      "step": 5221,
      "training_step_time": 17.030815601348877
    },
    {
      "epoch": 4.980087280273438e-06,
      "model_forward_time": 5.599198341369629,
      "step": 5222
    },
    {
      "epoch": 4.980087280273438e-06,
      "step": 5222,
      "training_step_time": 17.052731037139893
    },
    {
      "epoch": 4.981040954589844e-06,
      "model_forward_time": 5.598860025405884,
      "step": 5223
    },
    {
      "epoch": 4.981040954589844e-06,
      "step": 5223,
      "training_step_time": 17.42581605911255
    },
    {
      "epoch": 4.98199462890625e-06,
      "model_forward_time": 5.597550868988037,
      "step": 5224
    },
    {
      "epoch": 4.98199462890625e-06,
      "step": 5224,
      "training_step_time": 17.06166696548462
    },
    {
      "epoch": 4.982948303222656e-06,
      "model_forward_time": 5.602463483810425,
      "step": 5225
    },
    {
      "epoch": 4.982948303222656e-06,
      "step": 5225,
      "training_step_time": 17.05731177330017
    },
    {
      "epoch": 4.9839019775390625e-06,
      "model_forward_time": 5.655327796936035,
      "step": 5226
    },
    {
      "epoch": 4.9839019775390625e-06,
      "step": 5226,
      "training_step_time": 17.148746490478516
    },
    {
      "epoch": 4.984855651855469e-06,
      "model_forward_time": 5.601425886154175,
      "step": 5227
    },
    {
      "epoch": 4.984855651855469e-06,
      "step": 5227,
      "training_step_time": 17.078087091445923
    },
    {
      "epoch": 4.985809326171875e-06,
      "model_forward_time": 5.601069450378418,
      "step": 5228
    },
    {
      "epoch": 4.985809326171875e-06,
      "step": 5228,
      "training_step_time": 17.07024884223938
    },
    {
      "epoch": 4.986763000488281e-06,
      "model_forward_time": 5.976444959640503,
      "step": 5229
    },
    {
      "epoch": 4.986763000488281e-06,
      "step": 5229,
      "training_step_time": 17.43040704727173
    },
    {
      "epoch": 4.987716674804688e-06,
      "grad_norm": 1.2186837196350098,
      "learning_rate": 1e-05,
      "loss": 0.0792,
      "step": 5230
    },
    {
      "epoch": 4.987716674804688e-06,
      "model_forward_time": 5.587204933166504,
      "step": 5230
    },
    {
      "dynamics_loss_avg": 0.07230367250740528,
      "epoch": 4.987716674804688e-06,
      "step": 5230
    },
    {
      "action_loss_avg": 0.019176894682459533,
      "epoch": 4.987716674804688e-06,
      "step": 5230
    },
    {
      "epoch": 4.987716674804688e-06,
      "step": 5230,
      "training_step_time": 17.044103384017944
    },
    {
      "epoch": 4.988670349121094e-06,
      "model_forward_time": 5.596985101699829,
      "step": 5231
    },
    {
      "epoch": 4.988670349121094e-06,
      "step": 5231,
      "training_step_time": 17.06136202812195
    },
    {
      "epoch": 4.9896240234375e-06,
      "model_forward_time": 5.602111101150513,
      "step": 5232
    },
    {
      "epoch": 4.9896240234375e-06,
      "step": 5232,
      "training_step_time": 17.086768627166748
    },
    {
      "epoch": 4.990577697753906e-06,
      "model_forward_time": 5.6135008335113525,
      "step": 5233
    },
    {
      "epoch": 4.990577697753906e-06,
      "step": 5233,
      "training_step_time": 17.09855818748474
    },
    {
      "epoch": 4.9915313720703125e-06,
      "model_forward_time": 5.973113536834717,
      "step": 5234
    },
    {
      "epoch": 4.9915313720703125e-06,
      "step": 5234,
      "training_step_time": 17.464691162109375
    },
    {
      "epoch": 4.992485046386719e-06,
      "model_forward_time": 5.616791486740112,
      "step": 5235
    },
    {
      "epoch": 4.992485046386719e-06,
      "step": 5235,
      "training_step_time": 17.097898483276367
    },
    {
      "epoch": 4.993438720703125e-06,
      "model_forward_time": 5.5984580516815186,
      "step": 5236
    },
    {
      "epoch": 4.993438720703125e-06,
      "step": 5236,
      "training_step_time": 17.071305751800537
    },
    {
      "epoch": 4.994392395019531e-06,
      "model_forward_time": 5.5970778465271,
      "step": 5237
    },
    {
      "epoch": 4.994392395019531e-06,
      "step": 5237,
      "training_step_time": 17.037558555603027
    },
    {
      "epoch": 4.995346069335938e-06,
      "model_forward_time": 5.581543445587158,
      "step": 5238
    },
    {
      "epoch": 4.995346069335938e-06,
      "step": 5238,
      "training_step_time": 17.021206855773926
    },
    {
      "epoch": 4.996299743652344e-06,
      "model_forward_time": 5.606938362121582,
      "step": 5239
    },
    {
      "epoch": 4.996299743652344e-06,
      "step": 5239,
      "training_step_time": 17.086660861968994
    },
    {
      "epoch": 4.99725341796875e-06,
      "grad_norm": 2.2870521545410156,
      "learning_rate": 1e-05,
      "loss": 0.0733,
      "step": 5240
    },
    {
      "epoch": 4.99725341796875e-06,
      "model_forward_time": 5.975445747375488,
      "step": 5240
    },
    {
      "dynamics_loss_avg": 0.051932457648217675,
      "epoch": 4.99725341796875e-06,
      "step": 5240
    },
    {
      "action_loss_avg": 0.013757323403842748,
      "epoch": 4.99725341796875e-06,
      "step": 5240
    },
    {
      "epoch": 4.99725341796875e-06,
      "step": 5240,
      "training_step_time": 17.451854705810547
    },
    {
      "epoch": 4.998207092285156e-06,
      "model_forward_time": 5.601687908172607,
      "step": 5241
    },
    {
      "epoch": 4.998207092285156e-06,
      "step": 5241,
      "training_step_time": 17.076229572296143
    },
    {
      "epoch": 4.9991607666015625e-06,
      "model_forward_time": 5.600187301635742,
      "step": 5242
    },
    {
      "epoch": 4.9991607666015625e-06,
      "step": 5242,
      "training_step_time": 17.076467037200928
    },
    {
      "epoch": 5.000114440917969e-06,
      "model_forward_time": 5.6009581089019775,
      "step": 5243
    },
    {
      "epoch": 5.000114440917969e-06,
      "step": 5243,
      "training_step_time": 17.071325063705444
    },
    {
      "epoch": 5.001068115234375e-06,
      "model_forward_time": 5.601236820220947,
      "step": 5244
    },
    {
      "epoch": 5.001068115234375e-06,
      "step": 5244,
      "training_step_time": 17.07598114013672
    },
    {
      "epoch": 5.002021789550781e-06,
      "model_forward_time": 5.6028361320495605,
      "step": 5245
    },
    {
      "epoch": 5.002021789550781e-06,
      "step": 5245,
      "training_step_time": 17.53354501724243
    },
    {
      "epoch": 5.002975463867188e-06,
      "model_forward_time": 5.60599160194397,
      "step": 5246
    },
    {
      "epoch": 5.002975463867188e-06,
      "step": 5246,
      "training_step_time": 17.124930381774902
    },
    {
      "epoch": 5.003929138183594e-06,
      "model_forward_time": 5.624967813491821,
      "step": 5247
    },
    {
      "epoch": 5.003929138183594e-06,
      "step": 5247,
      "training_step_time": 17.1131591796875
    },
    {
      "epoch": 5.0048828125e-06,
      "model_forward_time": 5.5899810791015625,
      "step": 5248
    },
    {
      "epoch": 5.0048828125e-06,
      "step": 5248,
      "training_step_time": 17.019460678100586
    },
    {
      "epoch": 5.005836486816406e-06,
      "model_forward_time": 5.555594444274902,
      "step": 5249
    },
    {
      "epoch": 5.005836486816406e-06,
      "step": 5249,
      "training_step_time": 16.964660167694092
    },
    {
      "epoch": 5.0067901611328125e-06,
      "grad_norm": 0.9642002582550049,
      "learning_rate": 1e-05,
      "loss": 0.0739,
      "step": 5250
    },
    {
      "epoch": 5.0067901611328125e-06,
      "model_forward_time": 5.560817241668701,
      "step": 5250
    },
    {
      "dynamics_loss_avg": 0.05314127057790756,
      "epoch": 5.0067901611328125e-06,
      "step": 5250
    },
    {
      "action_loss_avg": 0.015621478017419577,
      "epoch": 5.0067901611328125e-06,
      "step": 5250
    },
    {
      "epoch": 5.0067901611328125e-06,
      "step": 5250,
      "training_step_time": 16.973304510116577
    },
    {
      "epoch": 5.007743835449219e-06,
      "model_forward_time": 5.921223878860474,
      "step": 5251
    },
    {
      "epoch": 5.007743835449219e-06,
      "step": 5251,
      "training_step_time": 17.332147359848022
    },
    {
      "epoch": 5.008697509765625e-06,
      "model_forward_time": 5.556480169296265,
      "step": 5252
    },
    {
      "epoch": 5.008697509765625e-06,
      "step": 5252,
      "training_step_time": 16.967466115951538
    },
    {
      "epoch": 5.009651184082031e-06,
      "model_forward_time": 5.557645559310913,
      "step": 5253
    },
    {
      "epoch": 5.009651184082031e-06,
      "step": 5253,
      "training_step_time": 16.978870391845703
    },
    {
      "epoch": 5.010604858398437e-06,
      "model_forward_time": 5.57152795791626,
      "step": 5254
    },
    {
      "epoch": 5.010604858398437e-06,
      "step": 5254,
      "training_step_time": 16.979113340377808
    },
    {
      "epoch": 5.011558532714844e-06,
      "model_forward_time": 5.584763526916504,
      "step": 5255
    },
    {
      "epoch": 5.011558532714844e-06,
      "step": 5255,
      "training_step_time": 17.008230209350586
    },
    {
      "epoch": 5.01251220703125e-06,
      "model_forward_time": 5.586001634597778,
      "step": 5256
    },
    {
      "epoch": 5.01251220703125e-06,
      "step": 5256,
      "training_step_time": 17.406394004821777
    },
    {
      "epoch": 5.013465881347656e-06,
      "model_forward_time": 5.592587947845459,
      "step": 5257
    },
    {
      "epoch": 5.013465881347656e-06,
      "step": 5257,
      "training_step_time": 17.064175128936768
    },
    {
      "epoch": 5.0144195556640625e-06,
      "model_forward_time": 5.6139092445373535,
      "step": 5258
    },
    {
      "epoch": 5.0144195556640625e-06,
      "step": 5258,
      "training_step_time": 17.100905895233154
    },
    {
      "epoch": 5.015373229980469e-06,
      "model_forward_time": 5.602260112762451,
      "step": 5259
    },
    {
      "epoch": 5.015373229980469e-06,
      "step": 5259,
      "training_step_time": 17.08113670349121
    },
    {
      "epoch": 5.016326904296875e-06,
      "grad_norm": 1.3630465269088745,
      "learning_rate": 1e-05,
      "loss": 0.0723,
      "step": 5260
    },
    {
      "epoch": 5.016326904296875e-06,
      "model_forward_time": 5.610983371734619,
      "step": 5260
    },
    {
      "dynamics_loss_avg": 0.06869422998279333,
      "epoch": 5.016326904296875e-06,
      "step": 5260
    },
    {
      "action_loss_avg": 0.01993251668754965,
      "epoch": 5.016326904296875e-06,
      "step": 5260
    },
    {
      "epoch": 5.016326904296875e-06,
      "step": 5260,
      "training_step_time": 17.07986283302307
    },
    {
      "epoch": 5.017280578613281e-06,
      "model_forward_time": 5.597379207611084,
      "step": 5261
    },
    {
      "epoch": 5.017280578613281e-06,
      "step": 5261,
      "training_step_time": 17.035828351974487
    },
    {
      "epoch": 5.018234252929687e-06,
      "model_forward_time": 5.955821752548218,
      "step": 5262
    },
    {
      "epoch": 5.018234252929687e-06,
      "step": 5262,
      "training_step_time": 17.37938141822815
    },
    {
      "epoch": 5.019187927246094e-06,
      "model_forward_time": 5.574280261993408,
      "step": 5263
    },
    {
      "epoch": 5.019187927246094e-06,
      "step": 5263,
      "training_step_time": 17.007471561431885
    },
    {
      "epoch": 5.0201416015625e-06,
      "model_forward_time": 5.591442823410034,
      "step": 5264
    },
    {
      "epoch": 5.0201416015625e-06,
      "step": 5264,
      "training_step_time": 17.043651580810547
    },
    {
      "epoch": 5.021095275878906e-06,
      "model_forward_time": 5.599062442779541,
      "step": 5265
    },
    {
      "epoch": 5.021095275878906e-06,
      "step": 5265,
      "training_step_time": 17.056977033615112
    },
    {
      "epoch": 5.0220489501953125e-06,
      "model_forward_time": 5.596412897109985,
      "step": 5266
    },
    {
      "epoch": 5.0220489501953125e-06,
      "step": 5266,
      "training_step_time": 17.05771827697754
    },
    {
      "epoch": 5.023002624511719e-06,
      "model_forward_time": 5.952547550201416,
      "step": 5267
    },
    {
      "epoch": 5.023002624511719e-06,
      "step": 5267,
      "training_step_time": 17.43717670440674
    },
    {
      "epoch": 5.023956298828125e-06,
      "model_forward_time": 5.613319396972656,
      "step": 5268
    },
    {
      "epoch": 5.023956298828125e-06,
      "step": 5268,
      "training_step_time": 17.11982536315918
    },
    {
      "epoch": 5.024909973144531e-06,
      "model_forward_time": 5.619877815246582,
      "step": 5269
    },
    {
      "epoch": 5.024909973144531e-06,
      "step": 5269,
      "training_step_time": 17.11833429336548
    },
    {
      "epoch": 5.025863647460937e-06,
      "grad_norm": 0.7913431525230408,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 5270
    },
    {
      "epoch": 5.025863647460937e-06,
      "model_forward_time": 5.612625360488892,
      "step": 5270
    },
    {
      "dynamics_loss_avg": 0.063365039229393,
      "epoch": 5.025863647460937e-06,
      "step": 5270
    },
    {
      "action_loss_avg": 0.015234603406861424,
      "epoch": 5.025863647460937e-06,
      "step": 5270
    },
    {
      "epoch": 5.025863647460937e-06,
      "step": 5270,
      "training_step_time": 17.099851608276367
    },
    {
      "epoch": 5.026817321777344e-06,
      "model_forward_time": 5.6185266971588135,
      "step": 5271
    },
    {
      "epoch": 5.026817321777344e-06,
      "step": 5271,
      "training_step_time": 17.102623224258423
    },
    {
      "epoch": 5.02777099609375e-06,
      "model_forward_time": 5.612198829650879,
      "step": 5272
    },
    {
      "epoch": 5.02777099609375e-06,
      "step": 5272,
      "training_step_time": 17.102088928222656
    },
    {
      "epoch": 5.028724670410156e-06,
      "model_forward_time": 5.973491668701172,
      "step": 5273
    },
    {
      "epoch": 5.028724670410156e-06,
      "step": 5273,
      "training_step_time": 17.463101625442505
    },
    {
      "epoch": 5.0296783447265625e-06,
      "model_forward_time": 5.614177227020264,
      "step": 5274
    },
    {
      "epoch": 5.0296783447265625e-06,
      "step": 5274,
      "training_step_time": 17.093416929244995
    },
    {
      "epoch": 5.030632019042969e-06,
      "model_forward_time": 5.611519813537598,
      "step": 5275
    },
    {
      "epoch": 5.030632019042969e-06,
      "step": 5275,
      "training_step_time": 17.10215973854065
    },
    {
      "epoch": 5.031585693359375e-06,
      "model_forward_time": 5.6099348068237305,
      "step": 5276
    },
    {
      "epoch": 5.031585693359375e-06,
      "step": 5276,
      "training_step_time": 17.0827317237854
    },
    {
      "epoch": 5.032539367675781e-06,
      "model_forward_time": 5.594243288040161,
      "step": 5277
    },
    {
      "epoch": 5.032539367675781e-06,
      "step": 5277,
      "training_step_time": 17.00326180458069
    },
    {
      "epoch": 5.033493041992187e-06,
      "model_forward_time": 5.5693747997283936,
      "step": 5278
    },
    {
      "epoch": 5.033493041992187e-06,
      "step": 5278,
      "training_step_time": 17.362177848815918
    },
    {
      "epoch": 5.034446716308594e-06,
      "model_forward_time": 5.575465202331543,
      "step": 5279
    },
    {
      "epoch": 5.034446716308594e-06,
      "step": 5279,
      "training_step_time": 17.003453731536865
    },
    {
      "epoch": 5.035400390625e-06,
      "grad_norm": 2.0666542053222656,
      "learning_rate": 1e-05,
      "loss": 0.0823,
      "step": 5280
    },
    {
      "epoch": 5.035400390625e-06,
      "model_forward_time": 5.568257570266724,
      "step": 5280
    },
    {
      "dynamics_loss_avg": 0.07147010527551174,
      "epoch": 5.035400390625e-06,
      "step": 5280
    },
    {
      "action_loss_avg": 0.01790247010067105,
      "epoch": 5.035400390625e-06,
      "step": 5280
    },
    {
      "epoch": 5.035400390625e-06,
      "step": 5280,
      "training_step_time": 16.985137939453125
    },
    {
      "epoch": 5.036354064941406e-06,
      "model_forward_time": 5.57748556137085,
      "step": 5281
    },
    {
      "epoch": 5.036354064941406e-06,
      "step": 5281,
      "training_step_time": 17.00053644180298
    },
    {
      "epoch": 5.0373077392578125e-06,
      "model_forward_time": 5.587546348571777,
      "step": 5282
    },
    {
      "epoch": 5.0373077392578125e-06,
      "step": 5282,
      "training_step_time": 17.01315426826477
    },
    {
      "epoch": 5.038261413574219e-06,
      "model_forward_time": 5.593611478805542,
      "step": 5283
    },
    {
      "epoch": 5.038261413574219e-06,
      "step": 5283,
      "training_step_time": 17.077967643737793
    },
    {
      "epoch": 5.039215087890625e-06,
      "model_forward_time": 5.99037504196167,
      "step": 5284
    },
    {
      "epoch": 5.039215087890625e-06,
      "step": 5284,
      "training_step_time": 17.51107358932495
    },
    {
      "epoch": 5.040168762207031e-06,
      "model_forward_time": 5.629408597946167,
      "step": 5285
    },
    {
      "epoch": 5.040168762207031e-06,
      "step": 5285,
      "training_step_time": 17.131348133087158
    },
    {
      "epoch": 5.041122436523437e-06,
      "model_forward_time": 5.602966547012329,
      "step": 5286
    },
    {
      "epoch": 5.041122436523437e-06,
      "step": 5286,
      "training_step_time": 17.086162090301514
    },
    {
      "epoch": 5.042076110839844e-06,
      "model_forward_time": 5.607361793518066,
      "step": 5287
    },
    {
      "epoch": 5.042076110839844e-06,
      "step": 5287,
      "training_step_time": 17.07391095161438
    },
    {
      "epoch": 5.04302978515625e-06,
      "model_forward_time": 5.59045147895813,
      "step": 5288
    },
    {
      "epoch": 5.04302978515625e-06,
      "step": 5288,
      "training_step_time": 17.048003673553467
    },
    {
      "epoch": 5.043983459472656e-06,
      "model_forward_time": 5.588670015335083,
      "step": 5289
    },
    {
      "epoch": 5.043983459472656e-06,
      "step": 5289,
      "training_step_time": 17.399542331695557
    },
    {
      "epoch": 5.0449371337890625e-06,
      "grad_norm": 1.7731866836547852,
      "learning_rate": 1e-05,
      "loss": 0.1,
      "step": 5290
    },
    {
      "epoch": 5.0449371337890625e-06,
      "model_forward_time": 5.586357355117798,
      "step": 5290
    },
    {
      "dynamics_loss_avg": 0.07242771591991186,
      "epoch": 5.0449371337890625e-06,
      "step": 5290
    },
    {
      "action_loss_avg": 0.022929798811674118,
      "epoch": 5.0449371337890625e-06,
      "step": 5290
    },
    {
      "epoch": 5.0449371337890625e-06,
      "step": 5290,
      "training_step_time": 17.030459880828857
    },
    {
      "epoch": 5.045890808105469e-06,
      "model_forward_time": 5.596330642700195,
      "step": 5291
    },
    {
      "epoch": 5.045890808105469e-06,
      "step": 5291,
      "training_step_time": 17.04556918144226
    },
    {
      "epoch": 5.046844482421875e-06,
      "model_forward_time": 5.596269130706787,
      "step": 5292
    },
    {
      "epoch": 5.046844482421875e-06,
      "step": 5292,
      "training_step_time": 17.04256844520569
    },
    {
      "epoch": 5.047798156738281e-06,
      "model_forward_time": 5.592723846435547,
      "step": 5293
    },
    {
      "epoch": 5.047798156738281e-06,
      "step": 5293,
      "training_step_time": 17.043577432632446
    },
    {
      "epoch": 5.048751831054687e-06,
      "model_forward_time": 5.59500527381897,
      "step": 5294
    },
    {
      "epoch": 5.048751831054687e-06,
      "step": 5294,
      "training_step_time": 17.040175437927246
    },
    {
      "epoch": 5.049705505371094e-06,
      "model_forward_time": 5.9659743309021,
      "step": 5295
    },
    {
      "epoch": 5.049705505371094e-06,
      "step": 5295,
      "training_step_time": 17.435395002365112
    },
    {
      "epoch": 5.0506591796875e-06,
      "model_forward_time": 5.598456382751465,
      "step": 5296
    },
    {
      "epoch": 5.0506591796875e-06,
      "step": 5296,
      "training_step_time": 17.064430236816406
    },
    {
      "epoch": 5.051612854003906e-06,
      "model_forward_time": 5.61380410194397,
      "step": 5297
    },
    {
      "epoch": 5.051612854003906e-06,
      "step": 5297,
      "training_step_time": 17.090404510498047
    },
    {
      "epoch": 5.0525665283203125e-06,
      "model_forward_time": 5.610885381698608,
      "step": 5298
    },
    {
      "epoch": 5.0525665283203125e-06,
      "step": 5298,
      "training_step_time": 17.080028533935547
    },
    {
      "epoch": 5.053520202636719e-06,
      "model_forward_time": 5.595353603363037,
      "step": 5299
    },
    {
      "epoch": 5.053520202636719e-06,
      "step": 5299,
      "training_step_time": 17.00268006324768
    },
    {
      "epoch": 5.054473876953125e-06,
      "grad_norm": 0.7881147265434265,
      "learning_rate": 1e-05,
      "loss": 0.0805,
      "step": 5300
    },
    {
      "epoch": 5.054473876953125e-06,
      "model_forward_time": 5.919777154922485,
      "step": 5300
    },
    {
      "dynamics_loss_avg": 0.055129573307931425,
      "epoch": 5.054473876953125e-06,
      "step": 5300
    },
    {
      "action_loss_avg": 0.020585214719176292,
      "epoch": 5.054473876953125e-06,
      "step": 5300
    },
    {
      "epoch": 5.054473876953125e-06,
      "step": 5300,
      "training_step_time": 17.34272050857544
    },
    {
      "epoch": 5.055427551269531e-06,
      "model_forward_time": 5.570311069488525,
      "step": 5301
    },
    {
      "epoch": 5.055427551269531e-06,
      "step": 5301,
      "training_step_time": 16.994020700454712
    },
    {
      "epoch": 5.056381225585937e-06,
      "model_forward_time": 5.562294006347656,
      "step": 5302
    },
    {
      "epoch": 5.056381225585937e-06,
      "step": 5302,
      "training_step_time": 16.97380304336548
    },
    {
      "epoch": 5.057334899902344e-06,
      "model_forward_time": 5.5755815505981445,
      "step": 5303
    },
    {
      "epoch": 5.057334899902344e-06,
      "step": 5303,
      "training_step_time": 17.001670360565186
    },
    {
      "epoch": 5.05828857421875e-06,
      "model_forward_time": 5.578466892242432,
      "step": 5304
    },
    {
      "epoch": 5.05828857421875e-06,
      "step": 5304,
      "training_step_time": 17.02136254310608
    },
    {
      "epoch": 5.059242248535156e-06,
      "model_forward_time": 5.596205472946167,
      "step": 5305
    },
    {
      "epoch": 5.059242248535156e-06,
      "step": 5305,
      "training_step_time": 17.039725065231323
    },
    {
      "epoch": 5.0601959228515625e-06,
      "model_forward_time": 5.986062526702881,
      "step": 5306
    },
    {
      "epoch": 5.0601959228515625e-06,
      "step": 5306,
      "training_step_time": 17.4554545879364
    },
    {
      "epoch": 5.061149597167969e-06,
      "model_forward_time": 5.6073079109191895,
      "step": 5307
    },
    {
      "epoch": 5.061149597167969e-06,
      "step": 5307,
      "training_step_time": 17.075541734695435
    },
    {
      "epoch": 5.062103271484375e-06,
      "model_forward_time": 5.593893766403198,
      "step": 5308
    },
    {
      "epoch": 5.062103271484375e-06,
      "step": 5308,
      "training_step_time": 17.040390253067017
    },
    {
      "epoch": 5.063056945800781e-06,
      "model_forward_time": 5.598421812057495,
      "step": 5309
    },
    {
      "epoch": 5.063056945800781e-06,
      "step": 5309,
      "training_step_time": 17.024821519851685
    },
    {
      "epoch": 5.064010620117187e-06,
      "grad_norm": 1.2102776765823364,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 5310
    },
    {
      "epoch": 5.064010620117187e-06,
      "model_forward_time": 5.594908237457275,
      "step": 5310
    },
    {
      "dynamics_loss_avg": 0.0533779539167881,
      "epoch": 5.064010620117187e-06,
      "step": 5310
    },
    {
      "action_loss_avg": 0.01096829967573285,
      "epoch": 5.064010620117187e-06,
      "step": 5310
    },
    {
      "epoch": 5.064010620117187e-06,
      "step": 5310,
      "training_step_time": 17.06840205192566
    },
    {
      "epoch": 5.064964294433594e-06,
      "model_forward_time": 5.5972301959991455,
      "step": 5311
    },
    {
      "epoch": 5.064964294433594e-06,
      "step": 5311,
      "training_step_time": 17.403594493865967
    },
    {
      "epoch": 5.06591796875e-06,
      "model_forward_time": 5.58448052406311,
      "step": 5312
    },
    {
      "epoch": 5.06591796875e-06,
      "step": 5312,
      "training_step_time": 17.02544617652893
    },
    {
      "epoch": 5.066871643066406e-06,
      "model_forward_time": 5.5808045864105225,
      "step": 5313
    },
    {
      "epoch": 5.066871643066406e-06,
      "step": 5313,
      "training_step_time": 17.03261113166809
    },
    {
      "epoch": 5.0678253173828126e-06,
      "model_forward_time": 5.579270839691162,
      "step": 5314
    },
    {
      "epoch": 5.0678253173828126e-06,
      "step": 5314,
      "training_step_time": 17.009538888931274
    },
    {
      "epoch": 5.068778991699219e-06,
      "model_forward_time": 5.597541570663452,
      "step": 5315
    },
    {
      "epoch": 5.068778991699219e-06,
      "step": 5315,
      "training_step_time": 17.066872119903564
    },
    {
      "epoch": 5.069732666015625e-06,
      "model_forward_time": 5.601364850997925,
      "step": 5316
    },
    {
      "epoch": 5.069732666015625e-06,
      "step": 5316,
      "training_step_time": 17.098673105239868
    },
    {
      "epoch": 5.070686340332031e-06,
      "model_forward_time": 5.99106240272522,
      "step": 5317
    },
    {
      "epoch": 5.070686340332031e-06,
      "step": 5317,
      "training_step_time": 17.46687388420105
    },
    {
      "epoch": 5.071640014648437e-06,
      "model_forward_time": 5.612495183944702,
      "step": 5318
    },
    {
      "epoch": 5.071640014648437e-06,
      "step": 5318,
      "training_step_time": 17.099007844924927
    },
    {
      "epoch": 5.072593688964844e-06,
      "model_forward_time": 5.602379560470581,
      "step": 5319
    },
    {
      "epoch": 5.072593688964844e-06,
      "step": 5319,
      "training_step_time": 17.074079990386963
    },
    {
      "epoch": 5.07354736328125e-06,
      "grad_norm": 0.27709972858428955,
      "learning_rate": 1e-05,
      "loss": 0.0751,
      "step": 5320
    },
    {
      "epoch": 5.07354736328125e-06,
      "model_forward_time": 5.596753120422363,
      "step": 5320
    },
    {
      "dynamics_loss_avg": 0.05837857909500599,
      "epoch": 5.07354736328125e-06,
      "step": 5320
    },
    {
      "action_loss_avg": 0.01500621581217274,
      "epoch": 5.07354736328125e-06,
      "step": 5320
    },
    {
      "epoch": 5.07354736328125e-06,
      "step": 5320,
      "training_step_time": 17.04943037033081
    },
    {
      "epoch": 5.074501037597656e-06,
      "model_forward_time": 5.5821897983551025,
      "step": 5321
    },
    {
      "epoch": 5.074501037597656e-06,
      "step": 5321,
      "training_step_time": 17.02142310142517
    },
    {
      "epoch": 5.0754547119140626e-06,
      "model_forward_time": 5.591254472732544,
      "step": 5322
    },
    {
      "epoch": 5.0754547119140626e-06,
      "step": 5322,
      "training_step_time": 17.41809344291687
    },
    {
      "epoch": 5.076408386230469e-06,
      "model_forward_time": 5.603376388549805,
      "step": 5323
    },
    {
      "epoch": 5.076408386230469e-06,
      "step": 5323,
      "training_step_time": 17.102041482925415
    },
    {
      "epoch": 5.077362060546875e-06,
      "model_forward_time": 5.612520217895508,
      "step": 5324
    },
    {
      "epoch": 5.077362060546875e-06,
      "step": 5324,
      "training_step_time": 17.10601496696472
    },
    {
      "epoch": 5.078315734863281e-06,
      "model_forward_time": 5.6070146560668945,
      "step": 5325
    },
    {
      "epoch": 5.078315734863281e-06,
      "step": 5325,
      "training_step_time": 17.086897373199463
    },
    {
      "epoch": 5.079269409179687e-06,
      "model_forward_time": 5.616134405136108,
      "step": 5326
    },
    {
      "epoch": 5.079269409179687e-06,
      "step": 5326,
      "training_step_time": 17.102713584899902
    },
    {
      "epoch": 5.080223083496094e-06,
      "model_forward_time": 5.612592935562134,
      "step": 5327
    },
    {
      "epoch": 5.080223083496094e-06,
      "step": 5327,
      "training_step_time": 17.09351897239685
    },
    {
      "epoch": 5.0811767578125e-06,
      "model_forward_time": 5.990685701370239,
      "step": 5328
    },
    {
      "epoch": 5.0811767578125e-06,
      "step": 5328,
      "training_step_time": 17.461998224258423
    },
    {
      "epoch": 5.082130432128906e-06,
      "model_forward_time": 5.59972882270813,
      "step": 5329
    },
    {
      "epoch": 5.082130432128906e-06,
      "step": 5329,
      "training_step_time": 17.071796894073486
    },
    {
      "epoch": 5.0830841064453126e-06,
      "grad_norm": 0.8946781158447266,
      "learning_rate": 1e-05,
      "loss": 0.0611,
      "step": 5330
    },
    {
      "epoch": 5.0830841064453126e-06,
      "model_forward_time": 5.591968774795532,
      "step": 5330
    },
    {
      "dynamics_loss_avg": 0.04618011955171823,
      "epoch": 5.0830841064453126e-06,
      "step": 5330
    },
    {
      "action_loss_avg": 0.013328968151472508,
      "epoch": 5.0830841064453126e-06,
      "step": 5330
    },
    {
      "epoch": 5.0830841064453126e-06,
      "step": 5330,
      "training_step_time": 17.029210329055786
    },
    {
      "epoch": 5.084037780761719e-06,
      "model_forward_time": 5.57650899887085,
      "step": 5331
    },
    {
      "epoch": 5.084037780761719e-06,
      "step": 5331,
      "training_step_time": 17.01414704322815
    },
    {
      "epoch": 5.084991455078125e-06,
      "model_forward_time": 5.597683668136597,
      "step": 5332
    },
    {
      "epoch": 5.084991455078125e-06,
      "step": 5332,
      "training_step_time": 17.086408615112305
    },
    {
      "epoch": 5.0859451293945314e-06,
      "model_forward_time": 5.971224069595337,
      "step": 5333
    },
    {
      "epoch": 5.0859451293945314e-06,
      "step": 5333,
      "training_step_time": 17.48463249206543
    },
    {
      "epoch": 5.086898803710937e-06,
      "model_forward_time": 5.626965761184692,
      "step": 5334
    },
    {
      "epoch": 5.086898803710937e-06,
      "step": 5334,
      "training_step_time": 17.15013575553894
    },
    {
      "epoch": 5.087852478027344e-06,
      "model_forward_time": 5.62463116645813,
      "step": 5335
    },
    {
      "epoch": 5.087852478027344e-06,
      "step": 5335,
      "training_step_time": 17.09166646003723
    },
    {
      "epoch": 5.08880615234375e-06,
      "model_forward_time": 5.5819549560546875,
      "step": 5336
    },
    {
      "epoch": 5.08880615234375e-06,
      "step": 5336,
      "training_step_time": 17.022183179855347
    },
    {
      "epoch": 5.089759826660156e-06,
      "model_forward_time": 5.574672222137451,
      "step": 5337
    },
    {
      "epoch": 5.089759826660156e-06,
      "step": 5337,
      "training_step_time": 16.97981333732605
    },
    {
      "epoch": 5.0907135009765626e-06,
      "model_forward_time": 5.570483684539795,
      "step": 5338
    },
    {
      "epoch": 5.0907135009765626e-06,
      "step": 5338,
      "training_step_time": 16.986838817596436
    },
    {
      "epoch": 5.091667175292969e-06,
      "model_forward_time": 5.956845760345459,
      "step": 5339
    },
    {
      "epoch": 5.091667175292969e-06,
      "step": 5339,
      "training_step_time": 17.377928495407104
    },
    {
      "epoch": 5.092620849609375e-06,
      "grad_norm": 0.9572001695632935,
      "learning_rate": 1e-05,
      "loss": 0.0594,
      "step": 5340
    },
    {
      "epoch": 5.092620849609375e-06,
      "model_forward_time": 5.581119060516357,
      "step": 5340
    },
    {
      "dynamics_loss_avg": 0.05369929131120443,
      "epoch": 5.092620849609375e-06,
      "step": 5340
    },
    {
      "action_loss_avg": 0.01137075824663043,
      "epoch": 5.092620849609375e-06,
      "step": 5340
    },
    {
      "epoch": 5.092620849609375e-06,
      "step": 5340,
      "training_step_time": 17.01357340812683
    },
    {
      "epoch": 5.0935745239257814e-06,
      "model_forward_time": 5.59254002571106,
      "step": 5341
    },
    {
      "epoch": 5.0935745239257814e-06,
      "step": 5341,
      "training_step_time": 17.04772186279297
    },
    {
      "epoch": 5.094528198242187e-06,
      "model_forward_time": 5.606051206588745,
      "step": 5342
    },
    {
      "epoch": 5.094528198242187e-06,
      "step": 5342,
      "training_step_time": 17.103764057159424
    },
    {
      "epoch": 5.095481872558594e-06,
      "model_forward_time": 5.613189458847046,
      "step": 5343
    },
    {
      "epoch": 5.095481872558594e-06,
      "step": 5343,
      "training_step_time": 17.08588480949402
    },
    {
      "epoch": 5.096435546875e-06,
      "model_forward_time": 5.5956337451934814,
      "step": 5344
    },
    {
      "epoch": 5.096435546875e-06,
      "step": 5344,
      "training_step_time": 17.40532374382019
    },
    {
      "epoch": 5.097389221191406e-06,
      "model_forward_time": 5.592236280441284,
      "step": 5345
    },
    {
      "epoch": 5.097389221191406e-06,
      "step": 5345,
      "training_step_time": 17.015238761901855
    },
    {
      "epoch": 5.098342895507813e-06,
      "model_forward_time": 5.580368995666504,
      "step": 5346
    },
    {
      "epoch": 5.098342895507813e-06,
      "step": 5346,
      "training_step_time": 17.00708818435669
    },
    {
      "epoch": 5.099296569824219e-06,
      "model_forward_time": 5.584726810455322,
      "step": 5347
    },
    {
      "epoch": 5.099296569824219e-06,
      "step": 5347,
      "training_step_time": 17.01855778694153
    },
    {
      "epoch": 5.100250244140625e-06,
      "model_forward_time": 5.577126502990723,
      "step": 5348
    },
    {
      "epoch": 5.100250244140625e-06,
      "step": 5348,
      "training_step_time": 16.999094009399414
    },
    {
      "epoch": 5.1012039184570314e-06,
      "model_forward_time": 5.584731817245483,
      "step": 5349
    },
    {
      "epoch": 5.1012039184570314e-06,
      "step": 5349,
      "training_step_time": 17.04927349090576
    },
    {
      "epoch": 5.102157592773437e-06,
      "grad_norm": 1.2023626565933228,
      "learning_rate": 1e-05,
      "loss": 0.0839,
      "step": 5350
    },
    {
      "epoch": 5.102157592773437e-06,
      "model_forward_time": 5.979130029678345,
      "step": 5350
    },
    {
      "dynamics_loss_avg": 0.06808699127286673,
      "epoch": 5.102157592773437e-06,
      "step": 5350
    },
    {
      "action_loss_avg": 0.014214851567521692,
      "epoch": 5.102157592773437e-06,
      "step": 5350
    },
    {
      "epoch": 5.102157592773437e-06,
      "step": 5350,
      "training_step_time": 17.46009063720703
    },
    {
      "epoch": 5.103111267089844e-06,
      "model_forward_time": 5.614544868469238,
      "step": 5351
    },
    {
      "epoch": 5.103111267089844e-06,
      "step": 5351,
      "training_step_time": 17.088708639144897
    },
    {
      "epoch": 5.10406494140625e-06,
      "model_forward_time": 5.610185623168945,
      "step": 5352
    },
    {
      "epoch": 5.10406494140625e-06,
      "step": 5352,
      "training_step_time": 17.10373091697693
    },
    {
      "epoch": 5.105018615722656e-06,
      "model_forward_time": 5.601641416549683,
      "step": 5353
    },
    {
      "epoch": 5.105018615722656e-06,
      "step": 5353,
      "training_step_time": 17.087859392166138
    },
    {
      "epoch": 5.105972290039063e-06,
      "model_forward_time": 5.6082799434661865,
      "step": 5354
    },
    {
      "epoch": 5.105972290039063e-06,
      "step": 5354,
      "training_step_time": 17.092716217041016
    },
    {
      "epoch": 5.106925964355469e-06,
      "model_forward_time": 5.971936941146851,
      "step": 5355
    },
    {
      "epoch": 5.106925964355469e-06,
      "step": 5355,
      "training_step_time": 17.456259727478027
    },
    {
      "epoch": 5.107879638671875e-06,
      "model_forward_time": 5.610764265060425,
      "step": 5356
    },
    {
      "epoch": 5.107879638671875e-06,
      "step": 5356,
      "training_step_time": 17.10653567314148
    },
    {
      "epoch": 5.1088333129882814e-06,
      "model_forward_time": 5.615337133407593,
      "step": 5357
    },
    {
      "epoch": 5.1088333129882814e-06,
      "step": 5357,
      "training_step_time": 17.094292402267456
    },
    {
      "epoch": 5.109786987304687e-06,
      "model_forward_time": 5.597104072570801,
      "step": 5358
    },
    {
      "epoch": 5.109786987304687e-06,
      "step": 5358,
      "training_step_time": 17.039121866226196
    },
    {
      "epoch": 5.110740661621094e-06,
      "model_forward_time": 5.583085536956787,
      "step": 5359
    },
    {
      "epoch": 5.110740661621094e-06,
      "step": 5359,
      "training_step_time": 17.016607522964478
    },
    {
      "epoch": 5.1116943359375e-06,
      "grad_norm": 1.2827297449111938,
      "learning_rate": 1e-05,
      "loss": 0.0782,
      "step": 5360
    },
    {
      "epoch": 5.1116943359375e-06,
      "model_forward_time": 5.589376926422119,
      "step": 5360
    },
    {
      "dynamics_loss_avg": 0.0646145548671484,
      "epoch": 5.1116943359375e-06,
      "step": 5360
    },
    {
      "action_loss_avg": 0.015458489209413529,
      "epoch": 5.1116943359375e-06,
      "step": 5360
    },
    {
      "epoch": 5.1116943359375e-06,
      "step": 5360,
      "training_step_time": 17.05908489227295
    },
    {
      "epoch": 5.112648010253906e-06,
      "model_forward_time": 5.97888708114624,
      "step": 5361
    },
    {
      "epoch": 5.112648010253906e-06,
      "step": 5361,
      "training_step_time": 17.44644784927368
    },
    {
      "epoch": 5.113601684570313e-06,
      "model_forward_time": 5.602752923965454,
      "step": 5362
    },
    {
      "epoch": 5.113601684570313e-06,
      "step": 5362,
      "training_step_time": 17.087902784347534
    },
    {
      "epoch": 5.114555358886719e-06,
      "model_forward_time": 5.6019721031188965,
      "step": 5363
    },
    {
      "epoch": 5.114555358886719e-06,
      "step": 5363,
      "training_step_time": 17.071599006652832
    },
    {
      "epoch": 5.115509033203125e-06,
      "model_forward_time": 5.602668046951294,
      "step": 5364
    },
    {
      "epoch": 5.115509033203125e-06,
      "step": 5364,
      "training_step_time": 17.08084511756897
    },
    {
      "epoch": 5.1164627075195315e-06,
      "model_forward_time": 5.601134777069092,
      "step": 5365
    },
    {
      "epoch": 5.1164627075195315e-06,
      "step": 5365,
      "training_step_time": 17.083129405975342
    },
    {
      "epoch": 5.117416381835937e-06,
      "model_forward_time": 5.953835964202881,
      "step": 5366
    },
    {
      "epoch": 5.117416381835937e-06,
      "step": 5366,
      "training_step_time": 17.41639018058777
    },
    {
      "epoch": 5.118370056152344e-06,
      "model_forward_time": 5.607645034790039,
      "step": 5367
    },
    {
      "epoch": 5.118370056152344e-06,
      "step": 5367,
      "training_step_time": 17.094579935073853
    },
    {
      "epoch": 5.11932373046875e-06,
      "model_forward_time": 5.611227750778198,
      "step": 5368
    },
    {
      "epoch": 5.11932373046875e-06,
      "step": 5368,
      "training_step_time": 17.104610443115234
    },
    {
      "epoch": 5.120277404785156e-06,
      "model_forward_time": 5.610797643661499,
      "step": 5369
    },
    {
      "epoch": 5.120277404785156e-06,
      "step": 5369,
      "training_step_time": 17.08864426612854
    },
    {
      "epoch": 5.121231079101563e-06,
      "grad_norm": 2.082237958908081,
      "learning_rate": 1e-05,
      "loss": 0.0851,
      "step": 5370
    },
    {
      "epoch": 5.121231079101563e-06,
      "model_forward_time": 5.5997936725616455,
      "step": 5370
    },
    {
      "dynamics_loss_avg": 0.0663670141249895,
      "epoch": 5.121231079101563e-06,
      "step": 5370
    },
    {
      "action_loss_avg": 0.016659711278043686,
      "epoch": 5.121231079101563e-06,
      "step": 5370
    },
    {
      "epoch": 5.121231079101563e-06,
      "step": 5370,
      "training_step_time": 17.069695234298706
    },
    {
      "epoch": 5.122184753417969e-06,
      "model_forward_time": 5.601258277893066,
      "step": 5371
    },
    {
      "epoch": 5.122184753417969e-06,
      "step": 5371,
      "training_step_time": 17.436098337173462
    },
    {
      "epoch": 5.123138427734375e-06,
      "model_forward_time": 5.586002826690674,
      "step": 5372
    },
    {
      "epoch": 5.123138427734375e-06,
      "step": 5372,
      "training_step_time": 17.026336431503296
    },
    {
      "epoch": 5.1240921020507815e-06,
      "model_forward_time": 5.587751388549805,
      "step": 5373
    },
    {
      "epoch": 5.1240921020507815e-06,
      "step": 5373,
      "training_step_time": 17.027508974075317
    },
    {
      "epoch": 5.125045776367187e-06,
      "model_forward_time": 5.600943326950073,
      "step": 5374
    },
    {
      "epoch": 5.125045776367187e-06,
      "step": 5374,
      "training_step_time": 17.08769464492798
    },
    {
      "epoch": 5.125999450683594e-06,
      "model_forward_time": 5.604097604751587,
      "step": 5375
    },
    {
      "epoch": 5.125999450683594e-06,
      "step": 5375,
      "training_step_time": 17.08782386779785
    },
    {
      "epoch": 5.126953125e-06,
      "model_forward_time": 5.5981810092926025,
      "step": 5376
    },
    {
      "epoch": 5.126953125e-06,
      "step": 5376,
      "training_step_time": 17.083951473236084
    },
    {
      "epoch": 5.127906799316406e-06,
      "model_forward_time": 5.965703964233398,
      "step": 5377
    },
    {
      "epoch": 5.127906799316406e-06,
      "step": 5377,
      "training_step_time": 17.454816818237305
    },
    {
      "epoch": 5.128860473632813e-06,
      "model_forward_time": 5.60869574546814,
      "step": 5378
    },
    {
      "epoch": 5.128860473632813e-06,
      "step": 5378,
      "training_step_time": 17.076509952545166
    },
    {
      "epoch": 5.129814147949218e-06,
      "model_forward_time": 5.603271961212158,
      "step": 5379
    },
    {
      "epoch": 5.129814147949218e-06,
      "step": 5379,
      "training_step_time": 17.062323093414307
    },
    {
      "epoch": 5.130767822265625e-06,
      "grad_norm": 1.500349521636963,
      "learning_rate": 1e-05,
      "loss": 0.0694,
      "step": 5380
    },
    {
      "epoch": 5.130767822265625e-06,
      "model_forward_time": 5.5914306640625,
      "step": 5380
    },
    {
      "dynamics_loss_avg": 0.056510680727660655,
      "epoch": 5.130767822265625e-06,
      "step": 5380
    },
    {
      "action_loss_avg": 0.013154914765618742,
      "epoch": 5.130767822265625e-06,
      "step": 5380
    },
    {
      "epoch": 5.130767822265625e-06,
      "step": 5380,
      "training_step_time": 17.049832820892334
    },
    {
      "epoch": 5.1317214965820315e-06,
      "model_forward_time": 5.596294164657593,
      "step": 5381
    },
    {
      "epoch": 5.1317214965820315e-06,
      "step": 5381,
      "training_step_time": 17.054609298706055
    },
    {
      "epoch": 5.132675170898437e-06,
      "model_forward_time": 5.600091457366943,
      "step": 5382
    },
    {
      "epoch": 5.132675170898437e-06,
      "step": 5382,
      "training_step_time": 17.44147539138794
    },
    {
      "epoch": 5.133628845214844e-06,
      "model_forward_time": 5.611169815063477,
      "step": 5383
    },
    {
      "epoch": 5.133628845214844e-06,
      "step": 5383,
      "training_step_time": 17.10578680038452
    },
    {
      "epoch": 5.13458251953125e-06,
      "model_forward_time": 5.624455451965332,
      "step": 5384
    },
    {
      "epoch": 5.13458251953125e-06,
      "step": 5384,
      "training_step_time": 17.093170166015625
    },
    {
      "epoch": 5.135536193847656e-06,
      "model_forward_time": 5.591006278991699,
      "step": 5385
    },
    {
      "epoch": 5.135536193847656e-06,
      "step": 5385,
      "training_step_time": 17.03791308403015
    },
    {
      "epoch": 5.136489868164063e-06,
      "model_forward_time": 5.581870794296265,
      "step": 5386
    },
    {
      "epoch": 5.136489868164063e-06,
      "step": 5386,
      "training_step_time": 17.000545501708984
    },
    {
      "epoch": 5.137443542480468e-06,
      "model_forward_time": 5.587080001831055,
      "step": 5387
    },
    {
      "epoch": 5.137443542480468e-06,
      "step": 5387,
      "training_step_time": 17.033949375152588
    },
    {
      "epoch": 5.138397216796875e-06,
      "model_forward_time": 5.9537060260772705,
      "step": 5388
    },
    {
      "epoch": 5.138397216796875e-06,
      "step": 5388,
      "training_step_time": 17.43302822113037
    },
    {
      "epoch": 5.1393508911132815e-06,
      "model_forward_time": 5.632382869720459,
      "step": 5389
    },
    {
      "epoch": 5.1393508911132815e-06,
      "step": 5389,
      "training_step_time": 17.137877464294434
    },
    {
      "epoch": 5.140304565429687e-06,
      "grad_norm": 0.6111040115356445,
      "learning_rate": 1e-05,
      "loss": 0.0613,
      "step": 5390
    },
    {
      "epoch": 5.140304565429687e-06,
      "model_forward_time": 5.623840808868408,
      "step": 5390
    },
    {
      "dynamics_loss_avg": 0.049092959798872474,
      "epoch": 5.140304565429687e-06,
      "step": 5390
    },
    {
      "action_loss_avg": 0.010505953920073807,
      "epoch": 5.140304565429687e-06,
      "step": 5390
    },
    {
      "epoch": 5.140304565429687e-06,
      "step": 5390,
      "training_step_time": 17.147266387939453
    },
    {
      "epoch": 5.141258239746094e-06,
      "model_forward_time": 5.626650810241699,
      "step": 5391
    },
    {
      "epoch": 5.141258239746094e-06,
      "step": 5391,
      "training_step_time": 17.120076417922974
    },
    {
      "epoch": 5.1422119140625e-06,
      "model_forward_time": 5.600179433822632,
      "step": 5392
    },
    {
      "epoch": 5.1422119140625e-06,
      "step": 5392,
      "training_step_time": 17.089346647262573
    },
    {
      "epoch": 5.143165588378906e-06,
      "model_forward_time": 5.973634481430054,
      "step": 5393
    },
    {
      "epoch": 5.143165588378906e-06,
      "step": 5393,
      "training_step_time": 17.442265510559082
    },
    {
      "epoch": 5.144119262695313e-06,
      "model_forward_time": 5.605688810348511,
      "step": 5394
    },
    {
      "epoch": 5.144119262695313e-06,
      "step": 5394,
      "training_step_time": 17.073238611221313
    },
    {
      "epoch": 5.145072937011718e-06,
      "model_forward_time": 5.589792728424072,
      "step": 5395
    },
    {
      "epoch": 5.145072937011718e-06,
      "step": 5395,
      "training_step_time": 17.01704454421997
    },
    {
      "epoch": 5.146026611328125e-06,
      "model_forward_time": 5.579955577850342,
      "step": 5396
    },
    {
      "epoch": 5.146026611328125e-06,
      "step": 5396,
      "training_step_time": 17.02088737487793
    },
    {
      "epoch": 5.1469802856445315e-06,
      "model_forward_time": 5.596071481704712,
      "step": 5397
    },
    {
      "epoch": 5.1469802856445315e-06,
      "step": 5397,
      "training_step_time": 17.038934230804443
    },
    {
      "epoch": 5.147933959960937e-06,
      "model_forward_time": 5.594731330871582,
      "step": 5398
    },
    {
      "epoch": 5.147933959960937e-06,
      "step": 5398,
      "training_step_time": 17.403417348861694
    },
    {
      "epoch": 5.148887634277344e-06,
      "model_forward_time": 5.594769477844238,
      "step": 5399
    },
    {
      "epoch": 5.148887634277344e-06,
      "step": 5399,
      "training_step_time": 17.05971932411194
    },
    {
      "epoch": 5.14984130859375e-06,
      "grad_norm": 0.2242174744606018,
      "learning_rate": 1e-05,
      "loss": 0.0717,
      "step": 5400
    },
    {
      "epoch": 5.14984130859375e-06,
      "model_forward_time": 5.587056636810303,
      "step": 5400
    },
    {
      "dynamics_loss_avg": 0.05860990788787603,
      "epoch": 5.14984130859375e-06,
      "step": 5400
    },
    {
      "action_loss_avg": 0.013876194693148136,
      "epoch": 5.14984130859375e-06,
      "step": 5400
    },
    {
      "epoch": 5.14984130859375e-06,
      "step": 5400,
      "training_step_time": 17.02532982826233
    },
    {
      "epoch": 5.150794982910156e-06,
      "model_forward_time": 5.579538583755493,
      "step": 5401
    },
    {
      "epoch": 5.150794982910156e-06,
      "step": 5401,
      "training_step_time": 17.009689807891846
    },
    {
      "epoch": 5.151748657226563e-06,
      "model_forward_time": 5.59537410736084,
      "step": 5402
    },
    {
      "epoch": 5.151748657226563e-06,
      "step": 5402,
      "training_step_time": 17.02364444732666
    },
    {
      "epoch": 5.152702331542968e-06,
      "model_forward_time": 5.590682506561279,
      "step": 5403
    },
    {
      "epoch": 5.152702331542968e-06,
      "step": 5403,
      "training_step_time": 17.058470010757446
    },
    {
      "epoch": 5.153656005859375e-06,
      "model_forward_time": 5.969908237457275,
      "step": 5404
    },
    {
      "epoch": 5.153656005859375e-06,
      "step": 5404,
      "training_step_time": 17.424126625061035
    },
    {
      "epoch": 5.1546096801757815e-06,
      "model_forward_time": 5.597168445587158,
      "step": 5405
    },
    {
      "epoch": 5.1546096801757815e-06,
      "step": 5405,
      "training_step_time": 17.07695508003235
    },
    {
      "epoch": 5.155563354492187e-06,
      "model_forward_time": 5.605222940444946,
      "step": 5406
    },
    {
      "epoch": 5.155563354492187e-06,
      "step": 5406,
      "training_step_time": 17.06720781326294
    },
    {
      "epoch": 5.156517028808594e-06,
      "model_forward_time": 5.600404977798462,
      "step": 5407
    },
    {
      "epoch": 5.156517028808594e-06,
      "step": 5407,
      "training_step_time": 17.038917541503906
    },
    {
      "epoch": 5.157470703125e-06,
      "model_forward_time": 5.594297647476196,
      "step": 5408
    },
    {
      "epoch": 5.157470703125e-06,
      "step": 5408,
      "training_step_time": 17.031900644302368
    },
    {
      "epoch": 5.158424377441406e-06,
      "model_forward_time": 5.940284729003906,
      "step": 5409
    },
    {
      "epoch": 5.158424377441406e-06,
      "step": 5409,
      "training_step_time": 17.372061252593994
    },
    {
      "epoch": 5.159378051757813e-06,
      "grad_norm": 0.25350236892700195,
      "learning_rate": 1e-05,
      "loss": 0.0774,
      "step": 5410
    },
    {
      "epoch": 5.159378051757813e-06,
      "model_forward_time": 5.583391189575195,
      "step": 5410
    },
    {
      "dynamics_loss_avg": 0.06280680969357491,
      "epoch": 5.159378051757813e-06,
      "step": 5410
    },
    {
      "action_loss_avg": 0.013696552184410394,
      "epoch": 5.159378051757813e-06,
      "step": 5410
    },
    {
      "epoch": 5.159378051757813e-06,
      "step": 5410,
      "training_step_time": 17.037209510803223
    },
    {
      "epoch": 5.160331726074218e-06,
      "model_forward_time": 5.601935386657715,
      "step": 5411
    },
    {
      "epoch": 5.160331726074218e-06,
      "step": 5411,
      "training_step_time": 17.107600688934326
    },
    {
      "epoch": 5.161285400390625e-06,
      "model_forward_time": 5.612827777862549,
      "step": 5412
    },
    {
      "epoch": 5.161285400390625e-06,
      "step": 5412,
      "training_step_time": 17.10902166366577
    },
    {
      "epoch": 5.1622390747070315e-06,
      "model_forward_time": 5.615081071853638,
      "step": 5413
    },
    {
      "epoch": 5.1622390747070315e-06,
      "step": 5413,
      "training_step_time": 17.087469816207886
    },
    {
      "epoch": 5.163192749023437e-06,
      "model_forward_time": 5.591084241867065,
      "step": 5414
    },
    {
      "epoch": 5.163192749023437e-06,
      "step": 5414,
      "training_step_time": 17.03371286392212
    },
    {
      "epoch": 5.164146423339844e-06,
      "model_forward_time": 5.945077896118164,
      "step": 5415
    },
    {
      "epoch": 5.164146423339844e-06,
      "step": 5415,
      "training_step_time": 17.38184905052185
    },
    {
      "epoch": 5.16510009765625e-06,
      "model_forward_time": 5.570366382598877,
      "step": 5416
    },
    {
      "epoch": 5.16510009765625e-06,
      "step": 5416,
      "training_step_time": 17.011243104934692
    },
    {
      "epoch": 5.166053771972656e-06,
      "model_forward_time": 5.572954416275024,
      "step": 5417
    },
    {
      "epoch": 5.166053771972656e-06,
      "step": 5417,
      "training_step_time": 16.998385190963745
    },
    {
      "epoch": 5.167007446289063e-06,
      "model_forward_time": 5.580389976501465,
      "step": 5418
    },
    {
      "epoch": 5.167007446289063e-06,
      "step": 5418,
      "training_step_time": 17.00302004814148
    },
    {
      "epoch": 5.167961120605468e-06,
      "model_forward_time": 5.5899505615234375,
      "step": 5419
    },
    {
      "epoch": 5.167961120605468e-06,
      "step": 5419,
      "training_step_time": 17.051605463027954
    },
    {
      "epoch": 5.168914794921875e-06,
      "grad_norm": 0.5930265188217163,
      "learning_rate": 1e-05,
      "loss": 0.073,
      "step": 5420
    },
    {
      "epoch": 5.168914794921875e-06,
      "model_forward_time": 5.956216335296631,
      "step": 5420
    },
    {
      "dynamics_loss_avg": 0.0621718930080533,
      "epoch": 5.168914794921875e-06,
      "step": 5420
    },
    {
      "action_loss_avg": 0.010953154601156711,
      "epoch": 5.168914794921875e-06,
      "step": 5420
    },
    {
      "epoch": 5.168914794921875e-06,
      "step": 5420,
      "training_step_time": 17.4367413520813
    },
    {
      "epoch": 5.1698684692382815e-06,
      "model_forward_time": 5.597201585769653,
      "step": 5421
    },
    {
      "epoch": 5.1698684692382815e-06,
      "step": 5421,
      "training_step_time": 17.069576263427734
    },
    {
      "epoch": 5.170822143554687e-06,
      "model_forward_time": 5.614405870437622,
      "step": 5422
    },
    {
      "epoch": 5.170822143554687e-06,
      "step": 5422,
      "training_step_time": 17.11131715774536
    },
    {
      "epoch": 5.171775817871094e-06,
      "model_forward_time": 5.617255210876465,
      "step": 5423
    },
    {
      "epoch": 5.171775817871094e-06,
      "step": 5423,
      "training_step_time": 17.124947786331177
    },
    {
      "epoch": 5.1727294921875e-06,
      "model_forward_time": 5.606484889984131,
      "step": 5424
    },
    {
      "epoch": 5.1727294921875e-06,
      "step": 5424,
      "training_step_time": 17.102462768554688
    },
    {
      "epoch": 5.173683166503906e-06,
      "model_forward_time": 5.603182315826416,
      "step": 5425
    },
    {
      "epoch": 5.173683166503906e-06,
      "step": 5425,
      "training_step_time": 17.08219003677368
    },
    {
      "epoch": 5.174636840820313e-06,
      "model_forward_time": 6.006501197814941,
      "step": 5426
    },
    {
      "epoch": 5.174636840820313e-06,
      "step": 5426,
      "training_step_time": 17.473228931427002
    },
    {
      "epoch": 5.175590515136718e-06,
      "model_forward_time": 5.59794807434082,
      "step": 5427
    },
    {
      "epoch": 5.175590515136718e-06,
      "step": 5427,
      "training_step_time": 17.056955575942993
    },
    {
      "epoch": 5.176544189453125e-06,
      "model_forward_time": 5.608829975128174,
      "step": 5428
    },
    {
      "epoch": 5.176544189453125e-06,
      "step": 5428,
      "training_step_time": 17.08384132385254
    },
    {
      "epoch": 5.1774978637695315e-06,
      "model_forward_time": 5.608211040496826,
      "step": 5429
    },
    {
      "epoch": 5.1774978637695315e-06,
      "step": 5429,
      "training_step_time": 17.113019943237305
    },
    {
      "epoch": 5.178451538085937e-06,
      "grad_norm": 0.5010243058204651,
      "learning_rate": 1e-05,
      "loss": 0.0661,
      "step": 5430
    },
    {
      "epoch": 5.178451538085937e-06,
      "model_forward_time": 5.608726978302002,
      "step": 5430
    },
    {
      "dynamics_loss_avg": 0.05312016382813454,
      "epoch": 5.178451538085937e-06,
      "step": 5430
    },
    {
      "action_loss_avg": 0.013786310609430075,
      "epoch": 5.178451538085937e-06,
      "step": 5430
    },
    {
      "epoch": 5.178451538085937e-06,
      "step": 5430,
      "training_step_time": 17.08790111541748
    },
    {
      "epoch": 5.179405212402344e-06,
      "model_forward_time": 5.959543228149414,
      "step": 5431
    },
    {
      "epoch": 5.179405212402344e-06,
      "step": 5431,
      "training_step_time": 17.39794111251831
    },
    {
      "epoch": 5.18035888671875e-06,
      "model_forward_time": 5.576159715652466,
      "step": 5432
    },
    {
      "epoch": 5.18035888671875e-06,
      "step": 5432,
      "training_step_time": 16.990295886993408
    },
    {
      "epoch": 5.181312561035156e-06,
      "model_forward_time": 5.582096815109253,
      "step": 5433
    },
    {
      "epoch": 5.181312561035156e-06,
      "step": 5433,
      "training_step_time": 17.027814626693726
    },
    {
      "epoch": 5.182266235351563e-06,
      "model_forward_time": 5.598968982696533,
      "step": 5434
    },
    {
      "epoch": 5.182266235351563e-06,
      "step": 5434,
      "training_step_time": 17.04998278617859
    },
    {
      "epoch": 5.183219909667968e-06,
      "model_forward_time": 5.5954906940460205,
      "step": 5435
    },
    {
      "epoch": 5.183219909667968e-06,
      "step": 5435,
      "training_step_time": 17.06093406677246
    },
    {
      "epoch": 5.184173583984375e-06,
      "model_forward_time": 5.594679117202759,
      "step": 5436
    },
    {
      "epoch": 5.184173583984375e-06,
      "step": 5436,
      "training_step_time": 17.42160964012146
    },
    {
      "epoch": 5.1851272583007815e-06,
      "model_forward_time": 5.610805034637451,
      "step": 5437
    },
    {
      "epoch": 5.1851272583007815e-06,
      "step": 5437,
      "training_step_time": 17.09964346885681
    },
    {
      "epoch": 5.186080932617187e-06,
      "model_forward_time": 5.613521099090576,
      "step": 5438
    },
    {
      "epoch": 5.186080932617187e-06,
      "step": 5438,
      "training_step_time": 17.097039937973022
    },
    {
      "epoch": 5.187034606933594e-06,
      "model_forward_time": 5.613220453262329,
      "step": 5439
    },
    {
      "epoch": 5.187034606933594e-06,
      "step": 5439,
      "training_step_time": 17.095840454101562
    },
    {
      "epoch": 5.18798828125e-06,
      "grad_norm": 0.5202146768569946,
      "learning_rate": 1e-05,
      "loss": 0.0601,
      "step": 5440
    },
    {
      "epoch": 5.18798828125e-06,
      "model_forward_time": 5.609982967376709,
      "step": 5440
    },
    {
      "dynamics_loss_avg": 0.050753405503928664,
      "epoch": 5.18798828125e-06,
      "step": 5440
    },
    {
      "action_loss_avg": 0.0075445934664458035,
      "epoch": 5.18798828125e-06,
      "step": 5440
    },
    {
      "epoch": 5.18798828125e-06,
      "step": 5440,
      "training_step_time": 17.09731435775757
    },
    {
      "epoch": 5.188941955566406e-06,
      "model_forward_time": 5.59775972366333,
      "step": 5441
    },
    {
      "epoch": 5.188941955566406e-06,
      "step": 5441,
      "training_step_time": 17.077665090560913
    },
    {
      "epoch": 5.189895629882813e-06,
      "model_forward_time": 5.954040765762329,
      "step": 5442
    },
    {
      "epoch": 5.189895629882813e-06,
      "step": 5442,
      "training_step_time": 17.44026494026184
    },
    {
      "epoch": 5.190849304199218e-06,
      "model_forward_time": 5.606785297393799,
      "step": 5443
    },
    {
      "epoch": 5.190849304199218e-06,
      "step": 5443,
      "training_step_time": 17.10856866836548
    },
    {
      "epoch": 5.191802978515625e-06,
      "model_forward_time": 5.622647047042847,
      "step": 5444
    },
    {
      "epoch": 5.191802978515625e-06,
      "step": 5444,
      "training_step_time": 17.128470420837402
    },
    {
      "epoch": 5.1927566528320315e-06,
      "model_forward_time": 5.623355388641357,
      "step": 5445
    },
    {
      "epoch": 5.1927566528320315e-06,
      "step": 5445,
      "training_step_time": 17.117530584335327
    },
    {
      "epoch": 5.193710327148437e-06,
      "model_forward_time": 5.616159200668335,
      "step": 5446
    },
    {
      "epoch": 5.193710327148437e-06,
      "step": 5446,
      "training_step_time": 17.111593008041382
    },
    {
      "epoch": 5.194664001464844e-06,
      "model_forward_time": 5.990901231765747,
      "step": 5447
    },
    {
      "epoch": 5.194664001464844e-06,
      "step": 5447,
      "training_step_time": 17.47034001350403
    },
    {
      "epoch": 5.19561767578125e-06,
      "model_forward_time": 5.592056512832642,
      "step": 5448
    },
    {
      "epoch": 5.19561767578125e-06,
      "step": 5448,
      "training_step_time": 17.0445396900177
    },
    {
      "epoch": 5.196571350097656e-06,
      "model_forward_time": 5.596413850784302,
      "step": 5449
    },
    {
      "epoch": 5.196571350097656e-06,
      "step": 5449,
      "training_step_time": 17.040863513946533
    },
    {
      "epoch": 5.197525024414063e-06,
      "grad_norm": 1.1903517246246338,
      "learning_rate": 1e-05,
      "loss": 0.0761,
      "step": 5450
    },
    {
      "epoch": 5.197525024414063e-06,
      "model_forward_time": 5.586028337478638,
      "step": 5450
    },
    {
      "dynamics_loss_avg": 0.06557210702449083,
      "epoch": 5.197525024414063e-06,
      "step": 5450
    },
    {
      "action_loss_avg": 0.017300589475780725,
      "epoch": 5.197525024414063e-06,
      "step": 5450
    },
    {
      "epoch": 5.197525024414063e-06,
      "step": 5450,
      "training_step_time": 17.037968158721924
    },
    {
      "epoch": 5.198478698730468e-06,
      "model_forward_time": 5.602802276611328,
      "step": 5451
    },
    {
      "epoch": 5.198478698730468e-06,
      "step": 5451,
      "training_step_time": 17.087799310684204
    },
    {
      "epoch": 5.199432373046875e-06,
      "model_forward_time": 5.606318235397339,
      "step": 5452
    },
    {
      "epoch": 5.199432373046875e-06,
      "step": 5452,
      "training_step_time": 17.10400938987732
    },
    {
      "epoch": 5.2003860473632815e-06,
      "model_forward_time": 5.953447341918945,
      "step": 5453
    },
    {
      "epoch": 5.2003860473632815e-06,
      "step": 5453,
      "training_step_time": 17.457908868789673
    },
    {
      "epoch": 5.201339721679687e-06,
      "model_forward_time": 5.615890979766846,
      "step": 5454
    },
    {
      "epoch": 5.201339721679687e-06,
      "step": 5454,
      "training_step_time": 17.10890007019043
    },
    {
      "epoch": 5.202293395996094e-06,
      "model_forward_time": 5.605130195617676,
      "step": 5455
    },
    {
      "epoch": 5.202293395996094e-06,
      "step": 5455,
      "training_step_time": 17.095146894454956
    },
    {
      "epoch": 5.2032470703125e-06,
      "model_forward_time": 5.652528285980225,
      "step": 5456
    },
    {
      "epoch": 5.2032470703125e-06,
      "step": 5456,
      "training_step_time": 17.158833980560303
    },
    {
      "epoch": 5.204200744628906e-06,
      "model_forward_time": 5.632514715194702,
      "step": 5457
    },
    {
      "epoch": 5.204200744628906e-06,
      "step": 5457,
      "training_step_time": 17.121484756469727
    },
    {
      "epoch": 5.205154418945313e-06,
      "model_forward_time": 6.0850067138671875,
      "step": 5458
    },
    {
      "epoch": 5.205154418945313e-06,
      "step": 5458,
      "training_step_time": 17.561859607696533
    },
    {
      "epoch": 5.206108093261718e-06,
      "model_forward_time": 5.593401670455933,
      "step": 5459
    },
    {
      "epoch": 5.206108093261718e-06,
      "step": 5459,
      "training_step_time": 17.04598617553711
    },
    {
      "epoch": 5.207061767578125e-06,
      "grad_norm": 0.7432354092597961,
      "learning_rate": 1e-05,
      "loss": 0.0733,
      "step": 5460
    },
    {
      "epoch": 5.207061767578125e-06,
      "model_forward_time": 5.598440408706665,
      "step": 5460
    },
    {
      "dynamics_loss_avg": 0.05184027589857578,
      "epoch": 5.207061767578125e-06,
      "step": 5460
    },
    {
      "action_loss_avg": 0.013344191515352577,
      "epoch": 5.207061767578125e-06,
      "step": 5460
    },
    {
      "epoch": 5.207061767578125e-06,
      "step": 5460,
      "training_step_time": 17.055798530578613
    },
    {
      "epoch": 5.2080154418945315e-06,
      "model_forward_time": 5.599998712539673,
      "step": 5461
    },
    {
      "epoch": 5.2080154418945315e-06,
      "step": 5461,
      "training_step_time": 17.069578170776367
    },
    {
      "epoch": 5.208969116210937e-06,
      "model_forward_time": 5.625661134719849,
      "step": 5462
    },
    {
      "epoch": 5.208969116210937e-06,
      "step": 5462,
      "training_step_time": 17.12986707687378
    },
    {
      "epoch": 5.209922790527344e-06,
      "model_forward_time": 5.6135594844818115,
      "step": 5463
    },
    {
      "epoch": 5.209922790527344e-06,
      "step": 5463,
      "training_step_time": 17.104225397109985
    },
    {
      "epoch": 5.21087646484375e-06,
      "model_forward_time": 5.955237627029419,
      "step": 5464
    },
    {
      "epoch": 5.21087646484375e-06,
      "step": 5464,
      "training_step_time": 17.441347122192383
    },
    {
      "epoch": 5.211830139160156e-06,
      "model_forward_time": 5.614147901535034,
      "step": 5465
    },
    {
      "epoch": 5.211830139160156e-06,
      "step": 5465,
      "training_step_time": 17.069077968597412
    },
    {
      "epoch": 5.212783813476563e-06,
      "model_forward_time": 5.609442234039307,
      "step": 5466
    },
    {
      "epoch": 5.212783813476563e-06,
      "step": 5466,
      "training_step_time": 17.109142065048218
    },
    {
      "epoch": 5.213737487792968e-06,
      "model_forward_time": 5.606675624847412,
      "step": 5467
    },
    {
      "epoch": 5.213737487792968e-06,
      "step": 5467,
      "training_step_time": 17.11290192604065
    },
    {
      "epoch": 5.214691162109375e-06,
      "model_forward_time": 5.610153913497925,
      "step": 5468
    },
    {
      "epoch": 5.214691162109375e-06,
      "step": 5468,
      "training_step_time": 17.107804536819458
    },
    {
      "epoch": 5.2156448364257815e-06,
      "model_forward_time": 5.980590343475342,
      "step": 5469
    },
    {
      "epoch": 5.2156448364257815e-06,
      "step": 5469,
      "training_step_time": 17.474436044692993
    },
    {
      "epoch": 5.216598510742187e-06,
      "grad_norm": 0.21107806265354156,
      "learning_rate": 1e-05,
      "loss": 0.059,
      "step": 5470
    },
    {
      "epoch": 5.216598510742187e-06,
      "model_forward_time": 5.604553461074829,
      "step": 5470
    },
    {
      "dynamics_loss_avg": 0.047997731156647204,
      "epoch": 5.216598510742187e-06,
      "step": 5470
    },
    {
      "action_loss_avg": 0.016753814835101365,
      "epoch": 5.216598510742187e-06,
      "step": 5470
    },
    {
      "epoch": 5.216598510742187e-06,
      "step": 5470,
      "training_step_time": 17.07144069671631
    },
    {
      "epoch": 5.217552185058594e-06,
      "model_forward_time": 5.591699600219727,
      "step": 5471
    },
    {
      "epoch": 5.217552185058594e-06,
      "step": 5471,
      "training_step_time": 17.052337646484375
    },
    {
      "epoch": 5.218505859375e-06,
      "model_forward_time": 5.594184160232544,
      "step": 5472
    },
    {
      "epoch": 5.218505859375e-06,
      "step": 5472,
      "training_step_time": 17.042604207992554
    },
    {
      "epoch": 5.219459533691406e-06,
      "model_forward_time": 5.6010472774505615,
      "step": 5473
    },
    {
      "epoch": 5.219459533691406e-06,
      "step": 5473,
      "training_step_time": 17.03511071205139
    },
    {
      "epoch": 5.220413208007813e-06,
      "model_forward_time": 5.596107721328735,
      "step": 5474
    },
    {
      "epoch": 5.220413208007813e-06,
      "step": 5474,
      "training_step_time": 17.039178609848022
    },
    {
      "epoch": 5.221366882324218e-06,
      "model_forward_time": 5.940337181091309,
      "step": 5475
    },
    {
      "epoch": 5.221366882324218e-06,
      "step": 5475,
      "training_step_time": 17.4072265625
    },
    {
      "epoch": 5.222320556640625e-06,
      "model_forward_time": 5.619653701782227,
      "step": 5476
    },
    {
      "epoch": 5.222320556640625e-06,
      "step": 5476,
      "training_step_time": 17.11182188987732
    },
    {
      "epoch": 5.2232742309570315e-06,
      "model_forward_time": 5.623290777206421,
      "step": 5477
    },
    {
      "epoch": 5.2232742309570315e-06,
      "step": 5477,
      "training_step_time": 17.103692531585693
    },
    {
      "epoch": 5.224227905273437e-06,
      "model_forward_time": 5.594265699386597,
      "step": 5478
    },
    {
      "epoch": 5.224227905273437e-06,
      "step": 5478,
      "training_step_time": 17.03834629058838
    },
    {
      "epoch": 5.225181579589844e-06,
      "model_forward_time": 5.580221176147461,
      "step": 5479
    },
    {
      "epoch": 5.225181579589844e-06,
      "step": 5479,
      "training_step_time": 16.993406534194946
    },
    {
      "epoch": 5.22613525390625e-06,
      "grad_norm": 1.647009015083313,
      "learning_rate": 1e-05,
      "loss": 0.065,
      "step": 5480
    },
    {
      "epoch": 5.22613525390625e-06,
      "model_forward_time": 5.945633888244629,
      "step": 5480
    },
    {
      "dynamics_loss_avg": 0.0593433728441596,
      "epoch": 5.22613525390625e-06,
      "step": 5480
    },
    {
      "action_loss_avg": 0.011946785403415561,
      "epoch": 5.22613525390625e-06,
      "step": 5480
    },
    {
      "epoch": 5.22613525390625e-06,
      "step": 5480,
      "training_step_time": 17.387457370758057
    },
    {
      "epoch": 5.227088928222656e-06,
      "model_forward_time": 5.56798243522644,
      "step": 5481
    },
    {
      "epoch": 5.227088928222656e-06,
      "step": 5481,
      "training_step_time": 16.972286224365234
    },
    {
      "epoch": 5.228042602539063e-06,
      "model_forward_time": 5.574949264526367,
      "step": 5482
    },
    {
      "epoch": 5.228042602539063e-06,
      "step": 5482,
      "training_step_time": 16.98697328567505
    },
    {
      "epoch": 5.228996276855468e-06,
      "model_forward_time": 5.577158212661743,
      "step": 5483
    },
    {
      "epoch": 5.228996276855468e-06,
      "step": 5483,
      "training_step_time": 17.003986597061157
    },
    {
      "epoch": 5.229949951171875e-06,
      "model_forward_time": 5.600577116012573,
      "step": 5484
    },
    {
      "epoch": 5.229949951171875e-06,
      "step": 5484,
      "training_step_time": 17.074281930923462
    },
    {
      "epoch": 5.2309036254882816e-06,
      "model_forward_time": 5.968798637390137,
      "step": 5485
    },
    {
      "epoch": 5.2309036254882816e-06,
      "step": 5485,
      "training_step_time": 17.455234050750732
    },
    {
      "epoch": 5.231857299804687e-06,
      "model_forward_time": 5.6012492179870605,
      "step": 5486
    },
    {
      "epoch": 5.231857299804687e-06,
      "step": 5486,
      "training_step_time": 17.056313514709473
    },
    {
      "epoch": 5.232810974121094e-06,
      "model_forward_time": 5.590672492980957,
      "step": 5487
    },
    {
      "epoch": 5.232810974121094e-06,
      "step": 5487,
      "training_step_time": 17.050285816192627
    },
    {
      "epoch": 5.2337646484375e-06,
      "model_forward_time": 5.596252679824829,
      "step": 5488
    },
    {
      "epoch": 5.2337646484375e-06,
      "step": 5488,
      "training_step_time": 17.0519859790802
    },
    {
      "epoch": 5.234718322753906e-06,
      "model_forward_time": 5.599998950958252,
      "step": 5489
    },
    {
      "epoch": 5.234718322753906e-06,
      "step": 5489,
      "training_step_time": 17.050766944885254
    },
    {
      "epoch": 5.235671997070313e-06,
      "grad_norm": 1.1845357418060303,
      "learning_rate": 1e-05,
      "loss": 0.0805,
      "step": 5490
    },
    {
      "epoch": 5.235671997070313e-06,
      "model_forward_time": 5.594896554946899,
      "step": 5490
    },
    {
      "dynamics_loss_avg": 0.060728159919381144,
      "epoch": 5.235671997070313e-06,
      "step": 5490
    },
    {
      "action_loss_avg": 0.014341340353712439,
      "epoch": 5.235671997070313e-06,
      "step": 5490
    },
    {
      "epoch": 5.235671997070313e-06,
      "step": 5490,
      "training_step_time": 17.031322956085205
    },
    {
      "epoch": 5.2366256713867184e-06,
      "model_forward_time": 5.958083868026733,
      "step": 5491
    },
    {
      "epoch": 5.2366256713867184e-06,
      "step": 5491,
      "training_step_time": 17.37541389465332
    },
    {
      "epoch": 5.237579345703125e-06,
      "model_forward_time": 5.601940631866455,
      "step": 5492
    },
    {
      "epoch": 5.237579345703125e-06,
      "step": 5492,
      "training_step_time": 17.055760622024536
    },
    {
      "epoch": 5.2385330200195316e-06,
      "model_forward_time": 5.606215000152588,
      "step": 5493
    },
    {
      "epoch": 5.2385330200195316e-06,
      "step": 5493,
      "training_step_time": 17.07897663116455
    },
    {
      "epoch": 5.239486694335937e-06,
      "model_forward_time": 5.603626728057861,
      "step": 5494
    },
    {
      "epoch": 5.239486694335937e-06,
      "step": 5494,
      "training_step_time": 17.070042610168457
    },
    {
      "epoch": 5.240440368652344e-06,
      "model_forward_time": 5.596517324447632,
      "step": 5495
    },
    {
      "epoch": 5.240440368652344e-06,
      "step": 5495,
      "training_step_time": 17.046116590499878
    },
    {
      "epoch": 5.24139404296875e-06,
      "model_forward_time": 5.586833477020264,
      "step": 5496
    },
    {
      "epoch": 5.24139404296875e-06,
      "step": 5496,
      "training_step_time": 17.419667720794678
    },
    {
      "epoch": 5.242347717285156e-06,
      "model_forward_time": 5.598061800003052,
      "step": 5497
    },
    {
      "epoch": 5.242347717285156e-06,
      "step": 5497,
      "training_step_time": 17.078645944595337
    },
    {
      "epoch": 5.243301391601563e-06,
      "model_forward_time": 5.609811544418335,
      "step": 5498
    },
    {
      "epoch": 5.243301391601563e-06,
      "step": 5498,
      "training_step_time": 17.112215995788574
    },
    {
      "epoch": 5.2442550659179684e-06,
      "model_forward_time": 5.60966157913208,
      "step": 5499
    },
    {
      "epoch": 5.2442550659179684e-06,
      "step": 5499,
      "training_step_time": 17.117191791534424
    },
    {
      "epoch": 5.245208740234375e-06,
      "grad_norm": 0.8800961971282959,
      "learning_rate": 1e-05,
      "loss": 0.0876,
      "step": 5500
    },
    {
      "epoch": 5.245208740234375e-06,
      "model_forward_time": 5.625783681869507,
      "step": 5500
    },
    {
      "dynamics_loss_avg": 0.07762257792055607,
      "epoch": 5.245208740234375e-06,
      "step": 5500
    },
    {
      "action_loss_avg": 0.019771521631628274,
      "epoch": 5.245208740234375e-06,
      "step": 5500
    },
    {
      "epoch": 5.245208740234375e-06,
      "step": 5500,
      "training_step_time": 17.09669280052185
    },
    {
      "epoch": 5.2461624145507816e-06,
      "model_forward_time": 5.580451488494873,
      "step": 5501
    },
    {
      "epoch": 5.2461624145507816e-06,
      "step": 5501,
      "training_step_time": 16.999496459960938
    },
    {
      "epoch": 5.247116088867187e-06,
      "model_forward_time": 5.930827379226685,
      "step": 5502
    },
    {
      "epoch": 5.247116088867187e-06,
      "step": 5502,
      "training_step_time": 17.341453313827515
    },
    {
      "epoch": 5.248069763183594e-06,
      "model_forward_time": 5.568107843399048,
      "step": 5503
    },
    {
      "epoch": 5.248069763183594e-06,
      "step": 5503,
      "training_step_time": 17.002573490142822
    },
    {
      "epoch": 5.2490234375e-06,
      "model_forward_time": 5.5735461711883545,
      "step": 5504
    },
    {
      "epoch": 5.2490234375e-06,
      "step": 5504,
      "training_step_time": 17.000115394592285
    },
    {
      "epoch": 5.249977111816406e-06,
      "model_forward_time": 5.585498571395874,
      "step": 5505
    },
    {
      "epoch": 5.249977111816406e-06,
      "step": 5505,
      "training_step_time": 16.99206829071045
    },
    {
      "epoch": 5.250930786132813e-06,
      "model_forward_time": 5.5792529582977295,
      "step": 5506
    },
    {
      "epoch": 5.250930786132813e-06,
      "step": 5506,
      "training_step_time": 17.020097732543945
    },
    {
      "epoch": 5.2518844604492184e-06,
      "model_forward_time": 5.590474843978882,
      "step": 5507
    },
    {
      "epoch": 5.2518844604492184e-06,
      "step": 5507,
      "training_step_time": 17.412718057632446
    },
    {
      "epoch": 5.252838134765625e-06,
      "model_forward_time": 5.580768823623657,
      "step": 5508
    },
    {
      "epoch": 5.252838134765625e-06,
      "step": 5508,
      "training_step_time": 17.019014596939087
    },
    {
      "epoch": 5.2537918090820316e-06,
      "model_forward_time": 5.615679979324341,
      "step": 5509
    },
    {
      "epoch": 5.2537918090820316e-06,
      "step": 5509,
      "training_step_time": 17.100440502166748
    },
    {
      "epoch": 5.254745483398437e-06,
      "grad_norm": 0.8931083679199219,
      "learning_rate": 1e-05,
      "loss": 0.1043,
      "step": 5510
    },
    {
      "epoch": 5.254745483398437e-06,
      "model_forward_time": 5.6506123542785645,
      "step": 5510
    },
    {
      "dynamics_loss_avg": 0.08095931373536587,
      "epoch": 5.254745483398437e-06,
      "step": 5510
    },
    {
      "action_loss_avg": 0.015825369441881777,
      "epoch": 5.254745483398437e-06,
      "step": 5510
    },
    {
      "epoch": 5.254745483398437e-06,
      "step": 5510,
      "training_step_time": 17.180009126663208
    },
    {
      "epoch": 5.255699157714844e-06,
      "model_forward_time": 5.616011619567871,
      "step": 5511
    },
    {
      "epoch": 5.255699157714844e-06,
      "step": 5511,
      "training_step_time": 17.119669914245605
    },
    {
      "epoch": 5.25665283203125e-06,
      "model_forward_time": 5.615009546279907,
      "step": 5512
    },
    {
      "epoch": 5.25665283203125e-06,
      "step": 5512,
      "training_step_time": 17.102723836898804
    },
    {
      "epoch": 5.257606506347656e-06,
      "model_forward_time": 5.987913131713867,
      "step": 5513
    },
    {
      "epoch": 5.257606506347656e-06,
      "step": 5513,
      "training_step_time": 17.462090253829956
    },
    {
      "epoch": 5.258560180664063e-06,
      "model_forward_time": 5.5984275341033936,
      "step": 5514
    },
    {
      "epoch": 5.258560180664063e-06,
      "step": 5514,
      "training_step_time": 17.056386709213257
    },
    {
      "epoch": 5.2595138549804684e-06,
      "model_forward_time": 5.586850643157959,
      "step": 5515
    },
    {
      "epoch": 5.2595138549804684e-06,
      "step": 5515,
      "training_step_time": 17.031627655029297
    },
    {
      "epoch": 5.260467529296875e-06,
      "model_forward_time": 5.589725017547607,
      "step": 5516
    },
    {
      "epoch": 5.260467529296875e-06,
      "step": 5516,
      "training_step_time": 17.023991346359253
    },
    {
      "epoch": 5.261421203613282e-06,
      "model_forward_time": 5.593002080917358,
      "step": 5517
    },
    {
      "epoch": 5.261421203613282e-06,
      "step": 5517,
      "training_step_time": 17.035247802734375
    },
    {
      "epoch": 5.262374877929687e-06,
      "model_forward_time": 5.954317331314087,
      "step": 5518
    },
    {
      "epoch": 5.262374877929687e-06,
      "step": 5518,
      "training_step_time": 17.41478657722473
    },
    {
      "epoch": 5.263328552246094e-06,
      "model_forward_time": 5.615926504135132,
      "step": 5519
    },
    {
      "epoch": 5.263328552246094e-06,
      "step": 5519,
      "training_step_time": 17.109790325164795
    },
    {
      "epoch": 5.2642822265625e-06,
      "grad_norm": 1.0070734024047852,
      "learning_rate": 1e-05,
      "loss": 0.0694,
      "step": 5520
    },
    {
      "epoch": 5.2642822265625e-06,
      "model_forward_time": 5.613872289657593,
      "step": 5520
    },
    {
      "dynamics_loss_avg": 0.053526073694229126,
      "epoch": 5.2642822265625e-06,
      "step": 5520
    },
    {
      "action_loss_avg": 0.014127405546605586,
      "epoch": 5.2642822265625e-06,
      "step": 5520
    },
    {
      "epoch": 5.2642822265625e-06,
      "step": 5520,
      "training_step_time": 17.119795560836792
    },
    {
      "epoch": 5.265235900878906e-06,
      "model_forward_time": 5.614424467086792,
      "step": 5521
    },
    {
      "epoch": 5.265235900878906e-06,
      "step": 5521,
      "training_step_time": 17.105012893676758
    },
    {
      "epoch": 5.266189575195313e-06,
      "model_forward_time": 5.607639789581299,
      "step": 5522
    },
    {
      "epoch": 5.266189575195313e-06,
      "step": 5522,
      "training_step_time": 17.060847997665405
    },
    {
      "epoch": 5.2671432495117185e-06,
      "model_forward_time": 5.59323787689209,
      "step": 5523
    },
    {
      "epoch": 5.2671432495117185e-06,
      "step": 5523,
      "training_step_time": 17.047847509384155
    },
    {
      "epoch": 5.268096923828125e-06,
      "model_forward_time": 5.973774433135986,
      "step": 5524
    },
    {
      "epoch": 5.268096923828125e-06,
      "step": 5524,
      "training_step_time": 17.439150094985962
    },
    {
      "epoch": 5.269050598144532e-06,
      "model_forward_time": 5.596474885940552,
      "step": 5525
    },
    {
      "epoch": 5.269050598144532e-06,
      "step": 5525,
      "training_step_time": 17.065112352371216
    },
    {
      "epoch": 5.270004272460937e-06,
      "model_forward_time": 5.60888409614563,
      "step": 5526
    },
    {
      "epoch": 5.270004272460937e-06,
      "step": 5526,
      "training_step_time": 17.08308243751526
    },
    {
      "epoch": 5.270957946777344e-06,
      "model_forward_time": 5.6160008907318115,
      "step": 5527
    },
    {
      "epoch": 5.270957946777344e-06,
      "step": 5527,
      "training_step_time": 17.108168840408325
    },
    {
      "epoch": 5.27191162109375e-06,
      "model_forward_time": 5.616391658782959,
      "step": 5528
    },
    {
      "epoch": 5.27191162109375e-06,
      "step": 5528,
      "training_step_time": 17.073842525482178
    },
    {
      "epoch": 5.272865295410156e-06,
      "model_forward_time": 5.590777635574341,
      "step": 5529
    },
    {
      "epoch": 5.272865295410156e-06,
      "step": 5529,
      "training_step_time": 17.376519918441772
    },
    {
      "epoch": 5.273818969726563e-06,
      "grad_norm": 0.36353617906570435,
      "learning_rate": 1e-05,
      "loss": 0.0676,
      "step": 5530
    },
    {
      "epoch": 5.273818969726563e-06,
      "model_forward_time": 5.570061445236206,
      "step": 5530
    },
    {
      "dynamics_loss_avg": 0.049907661601901056,
      "epoch": 5.273818969726563e-06,
      "step": 5530
    },
    {
      "action_loss_avg": 0.013994316710159183,
      "epoch": 5.273818969726563e-06,
      "step": 5530
    },
    {
      "epoch": 5.273818969726563e-06,
      "step": 5530,
      "training_step_time": 17.01146173477173
    },
    {
      "epoch": 5.2747726440429685e-06,
      "model_forward_time": 5.576715707778931,
      "step": 5531
    },
    {
      "epoch": 5.2747726440429685e-06,
      "step": 5531,
      "training_step_time": 16.996398448944092
    },
    {
      "epoch": 5.275726318359375e-06,
      "model_forward_time": 5.571119070053101,
      "step": 5532
    },
    {
      "epoch": 5.275726318359375e-06,
      "step": 5532,
      "training_step_time": 16.97633171081543
    },
    {
      "epoch": 5.276679992675782e-06,
      "model_forward_time": 5.583881378173828,
      "step": 5533
    },
    {
      "epoch": 5.276679992675782e-06,
      "step": 5533,
      "training_step_time": 17.018442630767822
    },
    {
      "epoch": 5.277633666992187e-06,
      "model_forward_time": 5.58527135848999,
      "step": 5534
    },
    {
      "epoch": 5.277633666992187e-06,
      "step": 5534,
      "training_step_time": 17.02047061920166
    },
    {
      "epoch": 5.278587341308594e-06,
      "model_forward_time": 5.966458320617676,
      "step": 5535
    },
    {
      "epoch": 5.278587341308594e-06,
      "step": 5535,
      "training_step_time": 17.37728476524353
    },
    {
      "epoch": 5.279541015625e-06,
      "model_forward_time": 5.594390153884888,
      "step": 5536
    },
    {
      "epoch": 5.279541015625e-06,
      "step": 5536,
      "training_step_time": 17.051127433776855
    },
    {
      "epoch": 5.280494689941406e-06,
      "model_forward_time": 5.599667310714722,
      "step": 5537
    },
    {
      "epoch": 5.280494689941406e-06,
      "step": 5537,
      "training_step_time": 17.09123706817627
    },
    {
      "epoch": 5.281448364257813e-06,
      "model_forward_time": 5.618117570877075,
      "step": 5538
    },
    {
      "epoch": 5.281448364257813e-06,
      "step": 5538,
      "training_step_time": 17.126635313034058
    },
    {
      "epoch": 5.2824020385742185e-06,
      "model_forward_time": 5.631754636764526,
      "step": 5539
    },
    {
      "epoch": 5.2824020385742185e-06,
      "step": 5539,
      "training_step_time": 17.13997173309326
    },
    {
      "epoch": 5.283355712890625e-06,
      "grad_norm": 0.4899904429912567,
      "learning_rate": 1e-05,
      "loss": 0.0653,
      "step": 5540
    },
    {
      "epoch": 5.283355712890625e-06,
      "model_forward_time": 5.602158308029175,
      "step": 5540
    },
    {
      "dynamics_loss_avg": 0.05590371936559677,
      "epoch": 5.283355712890625e-06,
      "step": 5540
    },
    {
      "action_loss_avg": 0.015701960073783994,
      "epoch": 5.283355712890625e-06,
      "step": 5540
    },
    {
      "epoch": 5.283355712890625e-06,
      "step": 5540,
      "training_step_time": 17.424253702163696
    },
    {
      "epoch": 5.284309387207032e-06,
      "model_forward_time": 5.59066367149353,
      "step": 5541
    },
    {
      "epoch": 5.284309387207032e-06,
      "step": 5541,
      "training_step_time": 17.033745288848877
    },
    {
      "epoch": 5.285263061523437e-06,
      "model_forward_time": 5.59121561050415,
      "step": 5542
    },
    {
      "epoch": 5.285263061523437e-06,
      "step": 5542,
      "training_step_time": 17.04278874397278
    },
    {
      "epoch": 5.286216735839844e-06,
      "model_forward_time": 5.593396902084351,
      "step": 5543
    },
    {
      "epoch": 5.286216735839844e-06,
      "step": 5543,
      "training_step_time": 17.056236743927002
    },
    {
      "epoch": 5.28717041015625e-06,
      "model_forward_time": 5.593092441558838,
      "step": 5544
    },
    {
      "epoch": 5.28717041015625e-06,
      "step": 5544,
      "training_step_time": 17.040908336639404
    },
    {
      "epoch": 5.288124084472656e-06,
      "model_forward_time": 5.590473175048828,
      "step": 5545
    },
    {
      "epoch": 5.288124084472656e-06,
      "step": 5545,
      "training_step_time": 17.049588918685913
    },
    {
      "epoch": 5.289077758789063e-06,
      "model_forward_time": 5.958047866821289,
      "step": 5546
    },
    {
      "epoch": 5.289077758789063e-06,
      "step": 5546,
      "training_step_time": 17.4273943901062
    },
    {
      "epoch": 5.2900314331054685e-06,
      "model_forward_time": 5.595364093780518,
      "step": 5547
    },
    {
      "epoch": 5.2900314331054685e-06,
      "step": 5547,
      "training_step_time": 17.064300537109375
    },
    {
      "epoch": 5.290985107421875e-06,
      "model_forward_time": 5.609280586242676,
      "step": 5548
    },
    {
      "epoch": 5.290985107421875e-06,
      "step": 5548,
      "training_step_time": 17.091614723205566
    },
    {
      "epoch": 5.291938781738282e-06,
      "model_forward_time": 5.614283323287964,
      "step": 5549
    },
    {
      "epoch": 5.291938781738282e-06,
      "step": 5549,
      "training_step_time": 17.07691979408264
    },
    {
      "epoch": 5.292892456054687e-06,
      "grad_norm": 1.3058348894119263,
      "learning_rate": 1e-05,
      "loss": 0.0795,
      "step": 5550
    },
    {
      "epoch": 5.292892456054687e-06,
      "model_forward_time": 5.601206541061401,
      "step": 5550
    },
    {
      "dynamics_loss_avg": 0.06228113453835249,
      "epoch": 5.292892456054687e-06,
      "step": 5550
    },
    {
      "action_loss_avg": 0.010917243780568242,
      "epoch": 5.292892456054687e-06,
      "step": 5550
    },
    {
      "epoch": 5.292892456054687e-06,
      "step": 5550,
      "training_step_time": 17.08173418045044
    },
    {
      "epoch": 5.293846130371094e-06,
      "model_forward_time": 5.9709391593933105,
      "step": 5551
    },
    {
      "epoch": 5.293846130371094e-06,
      "step": 5551,
      "training_step_time": 17.450026988983154
    },
    {
      "epoch": 5.2947998046875e-06,
      "model_forward_time": 5.601193189620972,
      "step": 5552
    },
    {
      "epoch": 5.2947998046875e-06,
      "step": 5552,
      "training_step_time": 17.06767177581787
    },
    {
      "epoch": 5.295753479003906e-06,
      "model_forward_time": 5.609145641326904,
      "step": 5553
    },
    {
      "epoch": 5.295753479003906e-06,
      "step": 5553,
      "training_step_time": 17.055410623550415
    },
    {
      "epoch": 5.296707153320313e-06,
      "model_forward_time": 5.583403825759888,
      "step": 5554
    },
    {
      "epoch": 5.296707153320313e-06,
      "step": 5554,
      "training_step_time": 17.015053510665894
    },
    {
      "epoch": 5.2976608276367185e-06,
      "model_forward_time": 5.588847637176514,
      "step": 5555
    },
    {
      "epoch": 5.2976608276367185e-06,
      "step": 5555,
      "training_step_time": 17.01306414604187
    },
    {
      "epoch": 5.298614501953125e-06,
      "model_forward_time": 5.597325086593628,
      "step": 5556
    },
    {
      "epoch": 5.298614501953125e-06,
      "step": 5556,
      "training_step_time": 17.037940502166748
    },
    {
      "epoch": 5.299568176269532e-06,
      "model_forward_time": 5.975832939147949,
      "step": 5557
    },
    {
      "epoch": 5.299568176269532e-06,
      "step": 5557,
      "training_step_time": 17.432279109954834
    },
    {
      "epoch": 5.300521850585937e-06,
      "model_forward_time": 5.607306241989136,
      "step": 5558
    },
    {
      "epoch": 5.300521850585937e-06,
      "step": 5558,
      "training_step_time": 17.084908962249756
    },
    {
      "epoch": 5.301475524902344e-06,
      "model_forward_time": 5.600994110107422,
      "step": 5559
    },
    {
      "epoch": 5.301475524902344e-06,
      "step": 5559,
      "training_step_time": 17.07740879058838
    },
    {
      "epoch": 5.30242919921875e-06,
      "grad_norm": 1.0578523874282837,
      "learning_rate": 1e-05,
      "loss": 0.0657,
      "step": 5560
    },
    {
      "epoch": 5.30242919921875e-06,
      "model_forward_time": 5.596433639526367,
      "step": 5560
    },
    {
      "dynamics_loss_avg": 0.057808449678123,
      "epoch": 5.30242919921875e-06,
      "step": 5560
    },
    {
      "action_loss_avg": 0.013191545382142067,
      "epoch": 5.30242919921875e-06,
      "step": 5560
    },
    {
      "epoch": 5.30242919921875e-06,
      "step": 5560,
      "training_step_time": 17.038928270339966
    },
    {
      "epoch": 5.303382873535156e-06,
      "model_forward_time": 5.580802917480469,
      "step": 5561
    },
    {
      "epoch": 5.303382873535156e-06,
      "step": 5561,
      "training_step_time": 17.00990343093872
    },
    {
      "epoch": 5.304336547851563e-06,
      "model_forward_time": 5.585906744003296,
      "step": 5562
    },
    {
      "epoch": 5.304336547851563e-06,
      "step": 5562,
      "training_step_time": 17.366533994674683
    },
    {
      "epoch": 5.3052902221679685e-06,
      "model_forward_time": 5.5915844440460205,
      "step": 5563
    },
    {
      "epoch": 5.3052902221679685e-06,
      "step": 5563,
      "training_step_time": 17.033969402313232
    },
    {
      "epoch": 5.306243896484375e-06,
      "model_forward_time": 5.600064039230347,
      "step": 5564
    },
    {
      "epoch": 5.306243896484375e-06,
      "step": 5564,
      "training_step_time": 17.072216749191284
    },
    {
      "epoch": 5.307197570800782e-06,
      "model_forward_time": 5.619051933288574,
      "step": 5565
    },
    {
      "epoch": 5.307197570800782e-06,
      "step": 5565,
      "training_step_time": 17.105509519577026
    },
    {
      "epoch": 5.308151245117187e-06,
      "model_forward_time": 5.596487045288086,
      "step": 5566
    },
    {
      "epoch": 5.308151245117187e-06,
      "step": 5566,
      "training_step_time": 17.032341241836548
    },
    {
      "epoch": 5.309104919433594e-06,
      "model_forward_time": 5.580438137054443,
      "step": 5567
    },
    {
      "epoch": 5.309104919433594e-06,
      "step": 5567,
      "training_step_time": 17.00548529624939
    },
    {
      "epoch": 5.31005859375e-06,
      "model_forward_time": 5.954297780990601,
      "step": 5568
    },
    {
      "epoch": 5.31005859375e-06,
      "step": 5568,
      "training_step_time": 17.381699323654175
    },
    {
      "epoch": 5.311012268066406e-06,
      "model_forward_time": 5.575171709060669,
      "step": 5569
    },
    {
      "epoch": 5.311012268066406e-06,
      "step": 5569,
      "training_step_time": 17.011616945266724
    },
    {
      "epoch": 5.311965942382813e-06,
      "grad_norm": 0.5770646333694458,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 5570
    },
    {
      "epoch": 5.311965942382813e-06,
      "model_forward_time": 5.590173006057739,
      "step": 5570
    },
    {
      "dynamics_loss_avg": 0.05223911441862583,
      "epoch": 5.311965942382813e-06,
      "step": 5570
    },
    {
      "action_loss_avg": 0.01174041242338717,
      "epoch": 5.311965942382813e-06,
      "step": 5570
    },
    {
      "epoch": 5.311965942382813e-06,
      "step": 5570,
      "training_step_time": 17.038172245025635
    },
    {
      "epoch": 5.3129196166992185e-06,
      "model_forward_time": 5.595590353012085,
      "step": 5571
    },
    {
      "epoch": 5.3129196166992185e-06,
      "step": 5571,
      "training_step_time": 17.046107053756714
    },
    {
      "epoch": 5.313873291015625e-06,
      "model_forward_time": 5.6021270751953125,
      "step": 5572
    },
    {
      "epoch": 5.313873291015625e-06,
      "step": 5572,
      "training_step_time": 17.08390188217163
    },
    {
      "epoch": 5.314826965332032e-06,
      "model_forward_time": 5.967789173126221,
      "step": 5573
    },
    {
      "epoch": 5.314826965332032e-06,
      "step": 5573,
      "training_step_time": 17.457302808761597
    },
    {
      "epoch": 5.315780639648437e-06,
      "model_forward_time": 5.6133294105529785,
      "step": 5574
    },
    {
      "epoch": 5.315780639648437e-06,
      "step": 5574,
      "training_step_time": 17.104302167892456
    },
    {
      "epoch": 5.316734313964844e-06,
      "model_forward_time": 5.622633457183838,
      "step": 5575
    },
    {
      "epoch": 5.316734313964844e-06,
      "step": 5575,
      "training_step_time": 17.119434118270874
    },
    {
      "epoch": 5.31768798828125e-06,
      "model_forward_time": 5.618934154510498,
      "step": 5576
    },
    {
      "epoch": 5.31768798828125e-06,
      "step": 5576,
      "training_step_time": 17.128979206085205
    },
    {
      "epoch": 5.318641662597656e-06,
      "model_forward_time": 5.623027086257935,
      "step": 5577
    },
    {
      "epoch": 5.318641662597656e-06,
      "step": 5577,
      "training_step_time": 17.131258726119995
    },
    {
      "epoch": 5.319595336914063e-06,
      "model_forward_time": 5.618074417114258,
      "step": 5578
    },
    {
      "epoch": 5.319595336914063e-06,
      "step": 5578,
      "training_step_time": 17.118330240249634
    },
    {
      "epoch": 5.3205490112304685e-06,
      "model_forward_time": 5.97206711769104,
      "step": 5579
    },
    {
      "epoch": 5.3205490112304685e-06,
      "step": 5579,
      "training_step_time": 17.457751035690308
    },
    {
      "epoch": 5.321502685546875e-06,
      "grad_norm": 0.38989078998565674,
      "learning_rate": 1e-05,
      "loss": 0.0692,
      "step": 5580
    },
    {
      "epoch": 5.321502685546875e-06,
      "model_forward_time": 5.609237432479858,
      "step": 5580
    },
    {
      "dynamics_loss_avg": 0.05595921576023102,
      "epoch": 5.321502685546875e-06,
      "step": 5580
    },
    {
      "action_loss_avg": 0.012817846261896193,
      "epoch": 5.321502685546875e-06,
      "step": 5580
    },
    {
      "epoch": 5.321502685546875e-06,
      "step": 5580,
      "training_step_time": 17.095659494400024
    },
    {
      "epoch": 5.322456359863282e-06,
      "model_forward_time": 5.610868692398071,
      "step": 5581
    },
    {
      "epoch": 5.322456359863282e-06,
      "step": 5581,
      "training_step_time": 17.106919765472412
    },
    {
      "epoch": 5.323410034179687e-06,
      "model_forward_time": 5.615478754043579,
      "step": 5582
    },
    {
      "epoch": 5.323410034179687e-06,
      "step": 5582,
      "training_step_time": 17.10615825653076
    },
    {
      "epoch": 5.324363708496094e-06,
      "model_forward_time": 5.610498905181885,
      "step": 5583
    },
    {
      "epoch": 5.324363708496094e-06,
      "step": 5583,
      "training_step_time": 17.096843242645264
    },
    {
      "epoch": 5.3253173828125e-06,
      "model_forward_time": 5.972618818283081,
      "step": 5584
    },
    {
      "epoch": 5.3253173828125e-06,
      "step": 5584,
      "training_step_time": 17.462881803512573
    },
    {
      "epoch": 5.326271057128906e-06,
      "model_forward_time": 5.610399484634399,
      "step": 5585
    },
    {
      "epoch": 5.326271057128906e-06,
      "step": 5585,
      "training_step_time": 17.09306263923645
    },
    {
      "epoch": 5.327224731445313e-06,
      "model_forward_time": 5.6031084060668945,
      "step": 5586
    },
    {
      "epoch": 5.327224731445313e-06,
      "step": 5586,
      "training_step_time": 17.062800645828247
    },
    {
      "epoch": 5.3281784057617185e-06,
      "model_forward_time": 5.598513603210449,
      "step": 5587
    },
    {
      "epoch": 5.3281784057617185e-06,
      "step": 5587,
      "training_step_time": 17.043753385543823
    },
    {
      "epoch": 5.329132080078125e-06,
      "model_forward_time": 5.594284296035767,
      "step": 5588
    },
    {
      "epoch": 5.329132080078125e-06,
      "step": 5588,
      "training_step_time": 17.03813099861145
    },
    {
      "epoch": 5.330085754394532e-06,
      "model_forward_time": 5.59318208694458,
      "step": 5589
    },
    {
      "epoch": 5.330085754394532e-06,
      "step": 5589,
      "training_step_time": 17.419098377227783
    },
    {
      "epoch": 5.331039428710937e-06,
      "grad_norm": 0.7504586577415466,
      "learning_rate": 1e-05,
      "loss": 0.0616,
      "step": 5590
    },
    {
      "epoch": 5.331039428710937e-06,
      "model_forward_time": 5.6016526222229,
      "step": 5590
    },
    {
      "dynamics_loss_avg": 0.04695949722081423,
      "epoch": 5.331039428710937e-06,
      "step": 5590
    },
    {
      "action_loss_avg": 0.01176790394820273,
      "epoch": 5.331039428710937e-06,
      "step": 5590
    },
    {
      "epoch": 5.331039428710937e-06,
      "step": 5590,
      "training_step_time": 17.05856943130493
    },
    {
      "epoch": 5.331993103027344e-06,
      "model_forward_time": 5.60209584236145,
      "step": 5591
    },
    {
      "epoch": 5.331993103027344e-06,
      "step": 5591,
      "training_step_time": 17.07889699935913
    },
    {
      "epoch": 5.33294677734375e-06,
      "model_forward_time": 5.595813989639282,
      "step": 5592
    },
    {
      "epoch": 5.33294677734375e-06,
      "step": 5592,
      "training_step_time": 17.054057359695435
    },
    {
      "epoch": 5.333900451660156e-06,
      "model_forward_time": 5.601501226425171,
      "step": 5593
    },
    {
      "epoch": 5.333900451660156e-06,
      "step": 5593,
      "training_step_time": 17.050705194473267
    },
    {
      "epoch": 5.334854125976563e-06,
      "model_forward_time": 5.590988397598267,
      "step": 5594
    },
    {
      "epoch": 5.334854125976563e-06,
      "step": 5594,
      "training_step_time": 17.01637887954712
    },
    {
      "epoch": 5.3358078002929685e-06,
      "model_forward_time": 5.934450626373291,
      "step": 5595
    },
    {
      "epoch": 5.3358078002929685e-06,
      "step": 5595,
      "training_step_time": 17.361222505569458
    },
    {
      "epoch": 5.336761474609375e-06,
      "model_forward_time": 5.596515655517578,
      "step": 5596
    },
    {
      "epoch": 5.336761474609375e-06,
      "step": 5596,
      "training_step_time": 17.06920862197876
    },
    {
      "epoch": 5.337715148925782e-06,
      "model_forward_time": 5.61278223991394,
      "step": 5597
    },
    {
      "epoch": 5.337715148925782e-06,
      "step": 5597,
      "training_step_time": 17.111369609832764
    },
    {
      "epoch": 5.338668823242187e-06,
      "model_forward_time": 5.6138317584991455,
      "step": 5598
    },
    {
      "epoch": 5.338668823242187e-06,
      "step": 5598,
      "training_step_time": 17.066627025604248
    },
    {
      "epoch": 5.339622497558594e-06,
      "model_forward_time": 5.5985026359558105,
      "step": 5599
    },
    {
      "epoch": 5.339622497558594e-06,
      "step": 5599,
      "training_step_time": 17.054486751556396
    },
    {
      "epoch": 5.340576171875e-06,
      "grad_norm": 1.517797827720642,
      "learning_rate": 1e-05,
      "loss": 0.0733,
      "step": 5600
    },
    {
      "epoch": 5.340576171875e-06,
      "model_forward_time": 5.592559099197388,
      "step": 5600
    },
    {
      "dynamics_loss_avg": 0.059995198249816896,
      "epoch": 5.340576171875e-06,
      "step": 5600
    },
    {
      "action_loss_avg": 0.016890165000222623,
      "epoch": 5.340576171875e-06,
      "step": 5600
    },
    {
      "epoch": 5.340576171875e-06,
      "step": 5600,
      "training_step_time": 17.38537621498108
    },
    {
      "epoch": 5.341529846191406e-06,
      "model_forward_time": 5.592172622680664,
      "step": 5601
    },
    {
      "epoch": 5.341529846191406e-06,
      "step": 5601,
      "training_step_time": 17.055041790008545
    },
    {
      "epoch": 5.342483520507813e-06,
      "model_forward_time": 5.600944757461548,
      "step": 5602
    },
    {
      "epoch": 5.342483520507813e-06,
      "step": 5602,
      "training_step_time": 17.08881187438965
    },
    {
      "epoch": 5.3434371948242185e-06,
      "model_forward_time": 5.608336448669434,
      "step": 5603
    },
    {
      "epoch": 5.3434371948242185e-06,
      "step": 5603,
      "training_step_time": 17.11053228378296
    },
    {
      "epoch": 5.344390869140625e-06,
      "model_forward_time": 5.610924005508423,
      "step": 5604
    },
    {
      "epoch": 5.344390869140625e-06,
      "step": 5604,
      "training_step_time": 17.1263484954834
    },
    {
      "epoch": 5.345344543457032e-06,
      "model_forward_time": 5.6148035526275635,
      "step": 5605
    },
    {
      "epoch": 5.345344543457032e-06,
      "step": 5605,
      "training_step_time": 17.114159107208252
    },
    {
      "epoch": 5.346298217773437e-06,
      "model_forward_time": 5.949531078338623,
      "step": 5606
    },
    {
      "epoch": 5.346298217773437e-06,
      "step": 5606,
      "training_step_time": 17.44146704673767
    },
    {
      "epoch": 5.347251892089844e-06,
      "model_forward_time": 5.605793714523315,
      "step": 5607
    },
    {
      "epoch": 5.347251892089844e-06,
      "step": 5607,
      "training_step_time": 17.101285457611084
    },
    {
      "epoch": 5.34820556640625e-06,
      "model_forward_time": 5.6032469272613525,
      "step": 5608
    },
    {
      "epoch": 5.34820556640625e-06,
      "step": 5608,
      "training_step_time": 17.091405868530273
    },
    {
      "epoch": 5.349159240722656e-06,
      "model_forward_time": 5.60810399055481,
      "step": 5609
    },
    {
      "epoch": 5.349159240722656e-06,
      "step": 5609,
      "training_step_time": 17.06710147857666
    },
    {
      "epoch": 5.350112915039063e-06,
      "grad_norm": 0.7263147234916687,
      "learning_rate": 1e-05,
      "loss": 0.0707,
      "step": 5610
    },
    {
      "epoch": 5.350112915039063e-06,
      "model_forward_time": 5.5763866901397705,
      "step": 5610
    },
    {
      "dynamics_loss_avg": 0.05912558697164059,
      "epoch": 5.350112915039063e-06,
      "step": 5610
    },
    {
      "action_loss_avg": 0.011623126454651356,
      "epoch": 5.350112915039063e-06,
      "step": 5610
    },
    {
      "epoch": 5.350112915039063e-06,
      "step": 5610,
      "training_step_time": 17.000019788742065
    },
    {
      "epoch": 5.3510665893554685e-06,
      "model_forward_time": 5.951531887054443,
      "step": 5611
    },
    {
      "epoch": 5.3510665893554685e-06,
      "step": 5611,
      "training_step_time": 17.366313457489014
    },
    {
      "epoch": 5.352020263671875e-06,
      "model_forward_time": 5.588425636291504,
      "step": 5612
    },
    {
      "epoch": 5.352020263671875e-06,
      "step": 5612,
      "training_step_time": 17.01103949546814
    },
    {
      "epoch": 5.352973937988282e-06,
      "model_forward_time": 5.573785066604614,
      "step": 5613
    },
    {
      "epoch": 5.352973937988282e-06,
      "step": 5613,
      "training_step_time": 17.005659103393555
    },
    {
      "epoch": 5.353927612304687e-06,
      "model_forward_time": 5.573275327682495,
      "step": 5614
    },
    {
      "epoch": 5.353927612304687e-06,
      "step": 5614,
      "training_step_time": 17.00598692893982
    },
    {
      "epoch": 5.354881286621094e-06,
      "model_forward_time": 5.581912040710449,
      "step": 5615
    },
    {
      "epoch": 5.354881286621094e-06,
      "step": 5615,
      "training_step_time": 16.992432594299316
    },
    {
      "epoch": 5.3558349609375e-06,
      "model_forward_time": 5.582743883132935,
      "step": 5616
    },
    {
      "epoch": 5.3558349609375e-06,
      "step": 5616,
      "training_step_time": 17.367186307907104
    },
    {
      "epoch": 5.356788635253906e-06,
      "model_forward_time": 5.581895351409912,
      "step": 5617
    },
    {
      "epoch": 5.356788635253906e-06,
      "step": 5617,
      "training_step_time": 17.0068039894104
    },
    {
      "epoch": 5.357742309570313e-06,
      "model_forward_time": 5.5902440547943115,
      "step": 5618
    },
    {
      "epoch": 5.357742309570313e-06,
      "step": 5618,
      "training_step_time": 17.069419622421265
    },
    {
      "epoch": 5.3586959838867185e-06,
      "model_forward_time": 5.599741458892822,
      "step": 5619
    },
    {
      "epoch": 5.3586959838867185e-06,
      "step": 5619,
      "training_step_time": 17.089595317840576
    },
    {
      "epoch": 5.359649658203125e-06,
      "grad_norm": 0.9648813009262085,
      "learning_rate": 1e-05,
      "loss": 0.0685,
      "step": 5620
    },
    {
      "epoch": 5.359649658203125e-06,
      "model_forward_time": 5.6093151569366455,
      "step": 5620
    },
    {
      "dynamics_loss_avg": 0.054917800985276696,
      "epoch": 5.359649658203125e-06,
      "step": 5620
    },
    {
      "action_loss_avg": 0.011383346933871507,
      "epoch": 5.359649658203125e-06,
      "step": 5620
    },
    {
      "epoch": 5.359649658203125e-06,
      "step": 5620,
      "training_step_time": 17.08916687965393
    },
    {
      "epoch": 5.360603332519532e-06,
      "model_forward_time": 5.593580961227417,
      "step": 5621
    },
    {
      "epoch": 5.360603332519532e-06,
      "step": 5621,
      "training_step_time": 17.02786350250244
    },
    {
      "epoch": 5.361557006835937e-06,
      "model_forward_time": 5.943840026855469,
      "step": 5622
    },
    {
      "epoch": 5.361557006835937e-06,
      "step": 5622,
      "training_step_time": 17.341638326644897
    },
    {
      "epoch": 5.362510681152344e-06,
      "model_forward_time": 5.577641248703003,
      "step": 5623
    },
    {
      "epoch": 5.362510681152344e-06,
      "step": 5623,
      "training_step_time": 16.938153505325317
    },
    {
      "epoch": 5.36346435546875e-06,
      "model_forward_time": 5.582083702087402,
      "step": 5624
    },
    {
      "epoch": 5.36346435546875e-06,
      "step": 5624,
      "training_step_time": 17.011247634887695
    },
    {
      "epoch": 5.364418029785156e-06,
      "model_forward_time": 5.5837578773498535,
      "step": 5625
    },
    {
      "epoch": 5.364418029785156e-06,
      "step": 5625,
      "training_step_time": 17.031524658203125
    },
    {
      "epoch": 5.365371704101563e-06,
      "model_forward_time": 5.583210229873657,
      "step": 5626
    },
    {
      "epoch": 5.365371704101563e-06,
      "step": 5626,
      "training_step_time": 17.04120707511902
    },
    {
      "epoch": 5.3663253784179685e-06,
      "model_forward_time": 5.962716817855835,
      "step": 5627
    },
    {
      "epoch": 5.3663253784179685e-06,
      "step": 5627,
      "training_step_time": 17.450021743774414
    },
    {
      "epoch": 5.367279052734375e-06,
      "model_forward_time": 5.602882385253906,
      "step": 5628
    },
    {
      "epoch": 5.367279052734375e-06,
      "step": 5628,
      "training_step_time": 17.0626437664032
    },
    {
      "epoch": 5.368232727050781e-06,
      "model_forward_time": 5.597810983657837,
      "step": 5629
    },
    {
      "epoch": 5.368232727050781e-06,
      "step": 5629,
      "training_step_time": 17.040526628494263
    },
    {
      "epoch": 5.369186401367187e-06,
      "grad_norm": 1.0320005416870117,
      "learning_rate": 1e-05,
      "loss": 0.0697,
      "step": 5630
    },
    {
      "epoch": 5.369186401367187e-06,
      "model_forward_time": 5.592546701431274,
      "step": 5630
    },
    {
      "dynamics_loss_avg": 0.0563947169110179,
      "epoch": 5.369186401367187e-06,
      "step": 5630
    },
    {
      "action_loss_avg": 0.015709132654592395,
      "epoch": 5.369186401367187e-06,
      "step": 5630
    },
    {
      "epoch": 5.369186401367187e-06,
      "step": 5630,
      "training_step_time": 17.015427827835083
    },
    {
      "epoch": 5.370140075683594e-06,
      "model_forward_time": 5.586202383041382,
      "step": 5631
    },
    {
      "epoch": 5.370140075683594e-06,
      "step": 5631,
      "training_step_time": 17.029329776763916
    },
    {
      "epoch": 5.37109375e-06,
      "model_forward_time": 5.605336427688599,
      "step": 5632
    },
    {
      "epoch": 5.37109375e-06,
      "step": 5632,
      "training_step_time": 17.050737380981445
    },
    {
      "epoch": 5.372047424316406e-06,
      "model_forward_time": 5.991884469985962,
      "step": 5633
    },
    {
      "epoch": 5.372047424316406e-06,
      "step": 5633,
      "training_step_time": 17.47227430343628
    },
    {
      "epoch": 5.373001098632813e-06,
      "model_forward_time": 5.5981035232543945,
      "step": 5634
    },
    {
      "epoch": 5.373001098632813e-06,
      "step": 5634,
      "training_step_time": 17.066815614700317
    },
    {
      "epoch": 5.3739547729492185e-06,
      "model_forward_time": 5.602964639663696,
      "step": 5635
    },
    {
      "epoch": 5.3739547729492185e-06,
      "step": 5635,
      "training_step_time": 17.059190034866333
    },
    {
      "epoch": 5.374908447265625e-06,
      "model_forward_time": 5.591088771820068,
      "step": 5636
    },
    {
      "epoch": 5.374908447265625e-06,
      "step": 5636,
      "training_step_time": 17.019603729248047
    },
    {
      "epoch": 5.375862121582031e-06,
      "model_forward_time": 5.58700966835022,
      "step": 5637
    },
    {
      "epoch": 5.375862121582031e-06,
      "step": 5637,
      "training_step_time": 17.010343074798584
    },
    {
      "epoch": 5.376815795898437e-06,
      "model_forward_time": 5.94734263420105,
      "step": 5638
    },
    {
      "epoch": 5.376815795898437e-06,
      "step": 5638,
      "training_step_time": 17.391279458999634
    },
    {
      "epoch": 5.377769470214844e-06,
      "model_forward_time": 5.596034526824951,
      "step": 5639
    },
    {
      "epoch": 5.377769470214844e-06,
      "step": 5639,
      "training_step_time": 17.046592950820923
    },
    {
      "epoch": 5.37872314453125e-06,
      "grad_norm": 1.1077404022216797,
      "learning_rate": 1e-05,
      "loss": 0.0715,
      "step": 5640
    },
    {
      "epoch": 5.37872314453125e-06,
      "model_forward_time": 5.609642028808594,
      "step": 5640
    },
    {
      "dynamics_loss_avg": 0.061013254709541796,
      "epoch": 5.37872314453125e-06,
      "step": 5640
    },
    {
      "action_loss_avg": 0.012085529579780995,
      "epoch": 5.37872314453125e-06,
      "step": 5640
    },
    {
      "epoch": 5.37872314453125e-06,
      "step": 5640,
      "training_step_time": 17.087287664413452
    },
    {
      "epoch": 5.379676818847656e-06,
      "model_forward_time": 5.6052632331848145,
      "step": 5641
    },
    {
      "epoch": 5.379676818847656e-06,
      "step": 5641,
      "training_step_time": 17.078437566757202
    },
    {
      "epoch": 5.380630493164063e-06,
      "model_forward_time": 5.5916266441345215,
      "step": 5642
    },
    {
      "epoch": 5.380630493164063e-06,
      "step": 5642,
      "training_step_time": 17.013094663619995
    },
    {
      "epoch": 5.3815841674804686e-06,
      "model_forward_time": 5.5749571323394775,
      "step": 5643
    },
    {
      "epoch": 5.3815841674804686e-06,
      "step": 5643,
      "training_step_time": 16.98759698867798
    },
    {
      "epoch": 5.382537841796875e-06,
      "model_forward_time": 5.930654287338257,
      "step": 5644
    },
    {
      "epoch": 5.382537841796875e-06,
      "step": 5644,
      "training_step_time": 17.367958068847656
    },
    {
      "epoch": 5.383491516113281e-06,
      "model_forward_time": 5.579692602157593,
      "step": 5645
    },
    {
      "epoch": 5.383491516113281e-06,
      "step": 5645,
      "training_step_time": 16.996718168258667
    },
    {
      "epoch": 5.384445190429687e-06,
      "model_forward_time": 5.5873191356658936,
      "step": 5646
    },
    {
      "epoch": 5.384445190429687e-06,
      "step": 5646,
      "training_step_time": 17.007593154907227
    },
    {
      "epoch": 5.385398864746094e-06,
      "model_forward_time": 5.5882439613342285,
      "step": 5647
    },
    {
      "epoch": 5.385398864746094e-06,
      "step": 5647,
      "training_step_time": 17.01697015762329
    },
    {
      "epoch": 5.3863525390625e-06,
      "model_forward_time": 5.596634864807129,
      "step": 5648
    },
    {
      "epoch": 5.3863525390625e-06,
      "step": 5648,
      "training_step_time": 17.078437328338623
    },
    {
      "epoch": 5.387306213378906e-06,
      "model_forward_time": 5.974024057388306,
      "step": 5649
    },
    {
      "epoch": 5.387306213378906e-06,
      "step": 5649,
      "training_step_time": 17.48693823814392
    },
    {
      "epoch": 5.388259887695313e-06,
      "grad_norm": 0.2184244692325592,
      "learning_rate": 1e-05,
      "loss": 0.069,
      "step": 5650
    },
    {
      "epoch": 5.388259887695313e-06,
      "model_forward_time": 5.6153883934021,
      "step": 5650
    },
    {
      "dynamics_loss_avg": 0.052724930178374055,
      "epoch": 5.388259887695313e-06,
      "step": 5650
    },
    {
      "action_loss_avg": 0.01140609837602824,
      "epoch": 5.388259887695313e-06,
      "step": 5650
    },
    {
      "epoch": 5.388259887695313e-06,
      "step": 5650,
      "training_step_time": 17.105075359344482
    },
    {
      "epoch": 5.3892135620117186e-06,
      "model_forward_time": 5.6269543170928955,
      "step": 5651
    },
    {
      "epoch": 5.3892135620117186e-06,
      "step": 5651,
      "training_step_time": 17.066174268722534
    },
    {
      "epoch": 5.390167236328125e-06,
      "model_forward_time": 5.613521575927734,
      "step": 5652
    },
    {
      "epoch": 5.390167236328125e-06,
      "step": 5652,
      "training_step_time": 17.05384087562561
    },
    {
      "epoch": 5.391120910644531e-06,
      "model_forward_time": 5.595320463180542,
      "step": 5653
    },
    {
      "epoch": 5.391120910644531e-06,
      "step": 5653,
      "training_step_time": 17.026588439941406
    },
    {
      "epoch": 5.392074584960937e-06,
      "model_forward_time": 5.589186191558838,
      "step": 5654
    },
    {
      "epoch": 5.392074584960937e-06,
      "step": 5654,
      "training_step_time": 17.398062229156494
    },
    {
      "epoch": 5.393028259277344e-06,
      "model_forward_time": 5.60440468788147,
      "step": 5655
    },
    {
      "epoch": 5.393028259277344e-06,
      "step": 5655,
      "training_step_time": 17.057562589645386
    },
    {
      "epoch": 5.39398193359375e-06,
      "model_forward_time": 5.601943492889404,
      "step": 5656
    },
    {
      "epoch": 5.39398193359375e-06,
      "step": 5656,
      "training_step_time": 17.052543878555298
    },
    {
      "epoch": 5.394935607910156e-06,
      "model_forward_time": 5.597541570663452,
      "step": 5657
    },
    {
      "epoch": 5.394935607910156e-06,
      "step": 5657,
      "training_step_time": 17.045177698135376
    },
    {
      "epoch": 5.395889282226563e-06,
      "model_forward_time": 5.5931713581085205,
      "step": 5658
    },
    {
      "epoch": 5.395889282226563e-06,
      "step": 5658,
      "training_step_time": 17.052343130111694
    },
    {
      "epoch": 5.3968429565429686e-06,
      "model_forward_time": 5.602099418640137,
      "step": 5659
    },
    {
      "epoch": 5.3968429565429686e-06,
      "step": 5659,
      "training_step_time": 17.092042207717896
    },
    {
      "epoch": 5.397796630859375e-06,
      "grad_norm": 1.407183289527893,
      "learning_rate": 1e-05,
      "loss": 0.0755,
      "step": 5660
    },
    {
      "epoch": 5.397796630859375e-06,
      "model_forward_time": 5.961057424545288,
      "step": 5660
    },
    {
      "dynamics_loss_avg": 0.0603515924885869,
      "epoch": 5.397796630859375e-06,
      "step": 5660
    },
    {
      "action_loss_avg": 0.017534794891253112,
      "epoch": 5.397796630859375e-06,
      "step": 5660
    },
    {
      "epoch": 5.397796630859375e-06,
      "step": 5660,
      "training_step_time": 17.455583333969116
    },
    {
      "epoch": 5.398750305175781e-06,
      "model_forward_time": 5.617302656173706,
      "step": 5661
    },
    {
      "epoch": 5.398750305175781e-06,
      "step": 5661,
      "training_step_time": 17.111418962478638
    },
    {
      "epoch": 5.3997039794921874e-06,
      "model_forward_time": 5.609659433364868,
      "step": 5662
    },
    {
      "epoch": 5.3997039794921874e-06,
      "step": 5662,
      "training_step_time": 17.101491689682007
    },
    {
      "epoch": 5.400657653808594e-06,
      "model_forward_time": 5.604676723480225,
      "step": 5663
    },
    {
      "epoch": 5.400657653808594e-06,
      "step": 5663,
      "training_step_time": 17.063135147094727
    },
    {
      "epoch": 5.401611328125e-06,
      "model_forward_time": 5.588497877120972,
      "step": 5664
    },
    {
      "epoch": 5.401611328125e-06,
      "step": 5664,
      "training_step_time": 17.014521598815918
    },
    {
      "epoch": 5.402565002441406e-06,
      "model_forward_time": 5.949957370758057,
      "step": 5665
    },
    {
      "epoch": 5.402565002441406e-06,
      "step": 5665,
      "training_step_time": 17.34598207473755
    },
    {
      "epoch": 5.403518676757813e-06,
      "model_forward_time": 5.578794956207275,
      "step": 5666
    },
    {
      "epoch": 5.403518676757813e-06,
      "step": 5666,
      "training_step_time": 16.99944257736206
    },
    {
      "epoch": 5.4044723510742186e-06,
      "model_forward_time": 5.582173585891724,
      "step": 5667
    },
    {
      "epoch": 5.4044723510742186e-06,
      "step": 5667,
      "training_step_time": 17.019104480743408
    },
    {
      "epoch": 5.405426025390625e-06,
      "model_forward_time": 5.57352614402771,
      "step": 5668
    },
    {
      "epoch": 5.405426025390625e-06,
      "step": 5668,
      "training_step_time": 17.00105118751526
    },
    {
      "epoch": 5.406379699707031e-06,
      "model_forward_time": 5.589762210845947,
      "step": 5669
    },
    {
      "epoch": 5.406379699707031e-06,
      "step": 5669,
      "training_step_time": 17.057188510894775
    },
    {
      "epoch": 5.4073333740234374e-06,
      "grad_norm": 1.1818405389785767,
      "learning_rate": 1e-05,
      "loss": 0.074,
      "step": 5670
    },
    {
      "epoch": 5.4073333740234374e-06,
      "model_forward_time": 5.608616590499878,
      "step": 5670
    },
    {
      "dynamics_loss_avg": 0.06618060115724803,
      "epoch": 5.4073333740234374e-06,
      "step": 5670
    },
    {
      "action_loss_avg": 0.014802042464725673,
      "epoch": 5.4073333740234374e-06,
      "step": 5670
    },
    {
      "epoch": 5.4073333740234374e-06,
      "step": 5670,
      "training_step_time": 17.078316926956177
    },
    {
      "epoch": 5.408287048339844e-06,
      "model_forward_time": 5.969468355178833,
      "step": 5671
    },
    {
      "epoch": 5.408287048339844e-06,
      "step": 5671,
      "training_step_time": 17.455435037612915
    },
    {
      "epoch": 5.40924072265625e-06,
      "model_forward_time": 5.600599527359009,
      "step": 5672
    },
    {
      "epoch": 5.40924072265625e-06,
      "step": 5672,
      "training_step_time": 17.081667184829712
    },
    {
      "epoch": 5.410194396972656e-06,
      "model_forward_time": 5.600946426391602,
      "step": 5673
    },
    {
      "epoch": 5.410194396972656e-06,
      "step": 5673,
      "training_step_time": 17.08163809776306
    },
    {
      "epoch": 5.411148071289063e-06,
      "model_forward_time": 5.60124135017395,
      "step": 5674
    },
    {
      "epoch": 5.411148071289063e-06,
      "step": 5674,
      "training_step_time": 17.05084729194641
    },
    {
      "epoch": 5.412101745605469e-06,
      "model_forward_time": 5.588004112243652,
      "step": 5675
    },
    {
      "epoch": 5.412101745605469e-06,
      "step": 5675,
      "training_step_time": 17.053862810134888
    },
    {
      "epoch": 5.413055419921875e-06,
      "model_forward_time": 5.961820363998413,
      "step": 5676
    },
    {
      "epoch": 5.413055419921875e-06,
      "step": 5676,
      "training_step_time": 17.415575742721558
    },
    {
      "epoch": 5.414009094238281e-06,
      "model_forward_time": 5.597488641738892,
      "step": 5677
    },
    {
      "epoch": 5.414009094238281e-06,
      "step": 5677,
      "training_step_time": 17.040396451950073
    },
    {
      "epoch": 5.4149627685546874e-06,
      "model_forward_time": 5.592698097229004,
      "step": 5678
    },
    {
      "epoch": 5.4149627685546874e-06,
      "step": 5678,
      "training_step_time": 17.03776264190674
    },
    {
      "epoch": 5.415916442871094e-06,
      "model_forward_time": 5.609122276306152,
      "step": 5679
    },
    {
      "epoch": 5.415916442871094e-06,
      "step": 5679,
      "training_step_time": 17.103491067886353
    },
    {
      "epoch": 5.4168701171875e-06,
      "grad_norm": 1.009999394416809,
      "learning_rate": 1e-05,
      "loss": 0.073,
      "step": 5680
    },
    {
      "epoch": 5.4168701171875e-06,
      "model_forward_time": 5.612539529800415,
      "step": 5680
    },
    {
      "dynamics_loss_avg": 0.05573304500430822,
      "epoch": 5.4168701171875e-06,
      "step": 5680
    },
    {
      "action_loss_avg": 0.013485764036886394,
      "epoch": 5.4168701171875e-06,
      "step": 5680
    },
    {
      "epoch": 5.4168701171875e-06,
      "step": 5680,
      "training_step_time": 17.11322283744812
    },
    {
      "epoch": 5.417823791503906e-06,
      "model_forward_time": 5.6039698123931885,
      "step": 5681
    },
    {
      "epoch": 5.417823791503906e-06,
      "step": 5681,
      "training_step_time": 17.102481842041016
    },
    {
      "epoch": 5.418777465820313e-06,
      "model_forward_time": 5.9702839851379395,
      "step": 5682
    },
    {
      "epoch": 5.418777465820313e-06,
      "step": 5682,
      "training_step_time": 17.44102144241333
    },
    {
      "epoch": 5.419731140136719e-06,
      "model_forward_time": 5.60384202003479,
      "step": 5683
    },
    {
      "epoch": 5.419731140136719e-06,
      "step": 5683,
      "training_step_time": 17.079448461532593
    },
    {
      "epoch": 5.420684814453125e-06,
      "model_forward_time": 5.604686498641968,
      "step": 5684
    },
    {
      "epoch": 5.420684814453125e-06,
      "step": 5684,
      "training_step_time": 17.091722011566162
    },
    {
      "epoch": 5.421638488769531e-06,
      "model_forward_time": 5.601218223571777,
      "step": 5685
    },
    {
      "epoch": 5.421638488769531e-06,
      "step": 5685,
      "training_step_time": 17.100900650024414
    },
    {
      "epoch": 5.4225921630859374e-06,
      "model_forward_time": 5.613890171051025,
      "step": 5686
    },
    {
      "epoch": 5.4225921630859374e-06,
      "step": 5686,
      "training_step_time": 17.10960555076599
    },
    {
      "epoch": 5.423545837402344e-06,
      "model_forward_time": 5.982413053512573,
      "step": 5687
    },
    {
      "epoch": 5.423545837402344e-06,
      "step": 5687,
      "training_step_time": 17.478548049926758
    },
    {
      "epoch": 5.42449951171875e-06,
      "model_forward_time": 5.601325273513794,
      "step": 5688
    },
    {
      "epoch": 5.42449951171875e-06,
      "step": 5688,
      "training_step_time": 17.085283279418945
    },
    {
      "epoch": 5.425453186035156e-06,
      "model_forward_time": 5.6034979820251465,
      "step": 5689
    },
    {
      "epoch": 5.425453186035156e-06,
      "step": 5689,
      "training_step_time": 17.071104288101196
    },
    {
      "epoch": 5.426406860351563e-06,
      "grad_norm": 0.4791994094848633,
      "learning_rate": 1e-05,
      "loss": 0.0792,
      "step": 5690
    },
    {
      "epoch": 5.426406860351563e-06,
      "model_forward_time": 5.5912535190582275,
      "step": 5690
    },
    {
      "dynamics_loss_avg": 0.0627409303560853,
      "epoch": 5.426406860351563e-06,
      "step": 5690
    },
    {
      "action_loss_avg": 0.016381546389311552,
      "epoch": 5.426406860351563e-06,
      "step": 5690
    },
    {
      "epoch": 5.426406860351563e-06,
      "step": 5690,
      "training_step_time": 17.04265069961548
    },
    {
      "epoch": 5.427360534667969e-06,
      "model_forward_time": 5.583457708358765,
      "step": 5691
    },
    {
      "epoch": 5.427360534667969e-06,
      "step": 5691,
      "training_step_time": 17.01828122138977
    },
    {
      "epoch": 5.428314208984375e-06,
      "model_forward_time": 5.579793691635132,
      "step": 5692
    },
    {
      "epoch": 5.428314208984375e-06,
      "step": 5692,
      "training_step_time": 17.02301001548767
    },
    {
      "epoch": 5.429267883300781e-06,
      "model_forward_time": 5.939746379852295,
      "step": 5693
    },
    {
      "epoch": 5.429267883300781e-06,
      "step": 5693,
      "training_step_time": 17.380703926086426
    },
    {
      "epoch": 5.4302215576171875e-06,
      "model_forward_time": 5.590715408325195,
      "step": 5694
    },
    {
      "epoch": 5.4302215576171875e-06,
      "step": 5694,
      "training_step_time": 17.062161684036255
    },
    {
      "epoch": 5.431175231933594e-06,
      "model_forward_time": 5.59946608543396,
      "step": 5695
    },
    {
      "epoch": 5.431175231933594e-06,
      "step": 5695,
      "training_step_time": 17.085553884506226
    },
    {
      "epoch": 5.43212890625e-06,
      "model_forward_time": 5.605578422546387,
      "step": 5696
    },
    {
      "epoch": 5.43212890625e-06,
      "step": 5696,
      "training_step_time": 17.10715913772583
    },
    {
      "epoch": 5.433082580566406e-06,
      "model_forward_time": 5.612760066986084,
      "step": 5697
    },
    {
      "epoch": 5.433082580566406e-06,
      "step": 5697,
      "training_step_time": 17.126884698867798
    },
    {
      "epoch": 5.434036254882813e-06,
      "model_forward_time": 5.99993371963501,
      "step": 5698
    },
    {
      "epoch": 5.434036254882813e-06,
      "step": 5698,
      "training_step_time": 17.48527765274048
    },
    {
      "epoch": 5.434989929199219e-06,
      "model_forward_time": 5.598919630050659,
      "step": 5699
    },
    {
      "epoch": 5.434989929199219e-06,
      "step": 5699,
      "training_step_time": 17.013218641281128
    },
    {
      "epoch": 5.435943603515625e-06,
      "grad_norm": 1.0308212041854858,
      "learning_rate": 1e-05,
      "loss": 0.0664,
      "step": 5700
    },
    {
      "epoch": 5.435943603515625e-06,
      "model_forward_time": 5.578866720199585,
      "step": 5700
    },
    {
      "dynamics_loss_avg": 0.051756215281784534,
      "epoch": 5.435943603515625e-06,
      "step": 5700
    },
    {
      "action_loss_avg": 0.012403533037286252,
      "epoch": 5.435943603515625e-06,
      "step": 5700
    },
    {
      "epoch": 5.435943603515625e-06,
      "step": 5700,
      "training_step_time": 17.001739263534546
    },
    {
      "epoch": 5.436897277832031e-06,
      "model_forward_time": 5.570908069610596,
      "step": 5701
    },
    {
      "epoch": 5.436897277832031e-06,
      "step": 5701,
      "training_step_time": 17.015406131744385
    },
    {
      "epoch": 5.4378509521484375e-06,
      "model_forward_time": 5.567570924758911,
      "step": 5702
    },
    {
      "epoch": 5.4378509521484375e-06,
      "step": 5702,
      "training_step_time": 16.9696946144104
    },
    {
      "epoch": 5.438804626464844e-06,
      "model_forward_time": 5.927351474761963,
      "step": 5703
    },
    {
      "epoch": 5.438804626464844e-06,
      "step": 5703,
      "training_step_time": 17.35144066810608
    },
    {
      "epoch": 5.43975830078125e-06,
      "model_forward_time": 5.589893817901611,
      "step": 5704
    },
    {
      "epoch": 5.43975830078125e-06,
      "step": 5704,
      "training_step_time": 17.032853841781616
    },
    {
      "epoch": 5.440711975097656e-06,
      "model_forward_time": 5.594213485717773,
      "step": 5705
    },
    {
      "epoch": 5.440711975097656e-06,
      "step": 5705,
      "training_step_time": 17.04975652694702
    },
    {
      "epoch": 5.441665649414063e-06,
      "model_forward_time": 5.606349468231201,
      "step": 5706
    },
    {
      "epoch": 5.441665649414063e-06,
      "step": 5706,
      "training_step_time": 17.076290607452393
    },
    {
      "epoch": 5.442619323730469e-06,
      "model_forward_time": 5.605071783065796,
      "step": 5707
    },
    {
      "epoch": 5.442619323730469e-06,
      "step": 5707,
      "training_step_time": 17.054478645324707
    },
    {
      "epoch": 5.443572998046875e-06,
      "model_forward_time": 5.592661142349243,
      "step": 5708
    },
    {
      "epoch": 5.443572998046875e-06,
      "step": 5708,
      "training_step_time": 17.029354572296143
    },
    {
      "epoch": 5.444526672363281e-06,
      "model_forward_time": 5.96167254447937,
      "step": 5709
    },
    {
      "epoch": 5.444526672363281e-06,
      "step": 5709,
      "training_step_time": 17.392738103866577
    },
    {
      "epoch": 5.4454803466796875e-06,
      "grad_norm": 0.819720983505249,
      "learning_rate": 1e-05,
      "loss": 0.0684,
      "step": 5710
    },
    {
      "epoch": 5.4454803466796875e-06,
      "model_forward_time": 5.58376407623291,
      "step": 5710
    },
    {
      "dynamics_loss_avg": 0.05410046223551035,
      "epoch": 5.4454803466796875e-06,
      "step": 5710
    },
    {
      "action_loss_avg": 0.013227763306349515,
      "epoch": 5.4454803466796875e-06,
      "step": 5710
    },
    {
      "epoch": 5.4454803466796875e-06,
      "step": 5710,
      "training_step_time": 17.010918617248535
    },
    {
      "epoch": 5.446434020996094e-06,
      "model_forward_time": 5.585787534713745,
      "step": 5711
    },
    {
      "epoch": 5.446434020996094e-06,
      "step": 5711,
      "training_step_time": 17.03838610649109
    },
    {
      "epoch": 5.4473876953125e-06,
      "model_forward_time": 5.598486661911011,
      "step": 5712
    },
    {
      "epoch": 5.4473876953125e-06,
      "step": 5712,
      "training_step_time": 17.071338891983032
    },
    {
      "epoch": 5.448341369628906e-06,
      "model_forward_time": 5.609144687652588,
      "step": 5713
    },
    {
      "epoch": 5.448341369628906e-06,
      "step": 5713,
      "training_step_time": 17.116509675979614
    },
    {
      "epoch": 5.449295043945313e-06,
      "model_forward_time": 5.6199634075164795,
      "step": 5714
    },
    {
      "epoch": 5.449295043945313e-06,
      "step": 5714,
      "training_step_time": 17.472052574157715
    },
    {
      "epoch": 5.450248718261719e-06,
      "model_forward_time": 5.593634605407715,
      "step": 5715
    },
    {
      "epoch": 5.450248718261719e-06,
      "step": 5715,
      "training_step_time": 17.04185390472412
    },
    {
      "epoch": 5.451202392578125e-06,
      "model_forward_time": 5.588815689086914,
      "step": 5716
    },
    {
      "epoch": 5.451202392578125e-06,
      "step": 5716,
      "training_step_time": 17.028000831604004
    },
    {
      "epoch": 5.452156066894531e-06,
      "model_forward_time": 5.576361179351807,
      "step": 5717
    },
    {
      "epoch": 5.452156066894531e-06,
      "step": 5717,
      "training_step_time": 16.996334552764893
    },
    {
      "epoch": 5.4531097412109375e-06,
      "model_forward_time": 5.587625503540039,
      "step": 5718
    },
    {
      "epoch": 5.4531097412109375e-06,
      "step": 5718,
      "training_step_time": 17.03575110435486
    },
    {
      "epoch": 5.454063415527344e-06,
      "model_forward_time": 5.594425678253174,
      "step": 5719
    },
    {
      "epoch": 5.454063415527344e-06,
      "step": 5719,
      "training_step_time": 17.05166792869568
    },
    {
      "epoch": 5.45501708984375e-06,
      "grad_norm": 0.1738428771495819,
      "learning_rate": 1e-05,
      "loss": 0.059,
      "step": 5720
    },
    {
      "epoch": 5.45501708984375e-06,
      "model_forward_time": 5.960706949234009,
      "step": 5720
    },
    {
      "dynamics_loss_avg": 0.04744714219123125,
      "epoch": 5.45501708984375e-06,
      "step": 5720
    },
    {
      "action_loss_avg": 0.010799856879748404,
      "epoch": 5.45501708984375e-06,
      "step": 5720
    },
    {
      "epoch": 5.45501708984375e-06,
      "step": 5720,
      "training_step_time": 17.420072078704834
    },
    {
      "epoch": 5.455970764160156e-06,
      "model_forward_time": 5.59175705909729,
      "step": 5721
    },
    {
      "epoch": 5.455970764160156e-06,
      "step": 5721,
      "training_step_time": 17.04308533668518
    },
    {
      "epoch": 5.456924438476563e-06,
      "model_forward_time": 5.588899612426758,
      "step": 5722
    },
    {
      "epoch": 5.456924438476563e-06,
      "step": 5722,
      "training_step_time": 17.04270911216736
    },
    {
      "epoch": 5.457878112792969e-06,
      "model_forward_time": 5.594386100769043,
      "step": 5723
    },
    {
      "epoch": 5.457878112792969e-06,
      "step": 5723,
      "training_step_time": 17.080914974212646
    },
    {
      "epoch": 5.458831787109375e-06,
      "model_forward_time": 5.6132848262786865,
      "step": 5724
    },
    {
      "epoch": 5.458831787109375e-06,
      "step": 5724,
      "training_step_time": 17.136717081069946
    },
    {
      "epoch": 5.459785461425781e-06,
      "model_forward_time": 5.6241295337677,
      "step": 5725
    },
    {
      "epoch": 5.459785461425781e-06,
      "step": 5725,
      "training_step_time": 17.466063022613525
    },
    {
      "epoch": 5.4607391357421875e-06,
      "model_forward_time": 5.588157653808594,
      "step": 5726
    },
    {
      "epoch": 5.4607391357421875e-06,
      "step": 5726,
      "training_step_time": 17.00676679611206
    },
    {
      "epoch": 5.461692810058594e-06,
      "model_forward_time": 5.562093496322632,
      "step": 5727
    },
    {
      "epoch": 5.461692810058594e-06,
      "step": 5727,
      "training_step_time": 16.980762720108032
    },
    {
      "epoch": 5.462646484375e-06,
      "model_forward_time": 5.553976774215698,
      "step": 5728
    },
    {
      "epoch": 5.462646484375e-06,
      "step": 5728,
      "training_step_time": 16.97538733482361
    },
    {
      "epoch": 5.463600158691406e-06,
      "model_forward_time": 5.555344343185425,
      "step": 5729
    },
    {
      "epoch": 5.463600158691406e-06,
      "step": 5729,
      "training_step_time": 16.961148500442505
    },
    {
      "epoch": 5.464553833007813e-06,
      "grad_norm": 1.0043752193450928,
      "learning_rate": 1e-05,
      "loss": 0.0744,
      "step": 5730
    },
    {
      "epoch": 5.464553833007813e-06,
      "model_forward_time": 5.558934450149536,
      "step": 5730
    },
    {
      "dynamics_loss_avg": 0.0613267108798027,
      "epoch": 5.464553833007813e-06,
      "step": 5730
    },
    {
      "action_loss_avg": 0.016944293398410082,
      "epoch": 5.464553833007813e-06,
      "step": 5730
    },
    {
      "epoch": 5.464553833007813e-06,
      "step": 5730,
      "training_step_time": 16.968432426452637
    },
    {
      "epoch": 5.465507507324219e-06,
      "model_forward_time": 5.9364964962005615,
      "step": 5731
    },
    {
      "epoch": 5.465507507324219e-06,
      "step": 5731,
      "training_step_time": 17.353901624679565
    },
    {
      "epoch": 5.466461181640625e-06,
      "model_forward_time": 5.565924882888794,
      "step": 5732
    },
    {
      "epoch": 5.466461181640625e-06,
      "step": 5732,
      "training_step_time": 16.978609323501587
    },
    {
      "epoch": 5.467414855957031e-06,
      "model_forward_time": 5.5840699672698975,
      "step": 5733
    },
    {
      "epoch": 5.467414855957031e-06,
      "step": 5733,
      "training_step_time": 16.99857258796692
    },
    {
      "epoch": 5.4683685302734375e-06,
      "model_forward_time": 5.575099945068359,
      "step": 5734
    },
    {
      "epoch": 5.4683685302734375e-06,
      "step": 5734,
      "training_step_time": 16.990687608718872
    },
    {
      "epoch": 5.469322204589844e-06,
      "model_forward_time": 5.597516775131226,
      "step": 5735
    },
    {
      "epoch": 5.469322204589844e-06,
      "step": 5735,
      "training_step_time": 17.072218656539917
    },
    {
      "epoch": 5.47027587890625e-06,
      "model_forward_time": 5.977573871612549,
      "step": 5736
    },
    {
      "epoch": 5.47027587890625e-06,
      "step": 5736,
      "training_step_time": 17.490339756011963
    },
    {
      "epoch": 5.471229553222656e-06,
      "model_forward_time": 5.611015558242798,
      "step": 5737
    },
    {
      "epoch": 5.471229553222656e-06,
      "step": 5737,
      "training_step_time": 17.066673278808594
    },
    {
      "epoch": 5.472183227539063e-06,
      "model_forward_time": 5.571476459503174,
      "step": 5738
    },
    {
      "epoch": 5.472183227539063e-06,
      "step": 5738,
      "training_step_time": 16.9844388961792
    },
    {
      "epoch": 5.473136901855469e-06,
      "model_forward_time": 5.547027111053467,
      "step": 5739
    },
    {
      "epoch": 5.473136901855469e-06,
      "step": 5739,
      "training_step_time": 16.96390199661255
    },
    {
      "epoch": 5.474090576171875e-06,
      "grad_norm": 0.605999767780304,
      "learning_rate": 1e-05,
      "loss": 0.072,
      "step": 5740
    },
    {
      "epoch": 5.474090576171875e-06,
      "model_forward_time": 5.5508131980896,
      "step": 5740
    },
    {
      "dynamics_loss_avg": 0.056184069626033305,
      "epoch": 5.474090576171875e-06,
      "step": 5740
    },
    {
      "action_loss_avg": 0.014029741287231445,
      "epoch": 5.474090576171875e-06,
      "step": 5740
    },
    {
      "epoch": 5.474090576171875e-06,
      "step": 5740,
      "training_step_time": 16.954697370529175
    },
    {
      "epoch": 5.475044250488281e-06,
      "model_forward_time": 5.547110557556152,
      "step": 5741
    },
    {
      "epoch": 5.475044250488281e-06,
      "step": 5741,
      "training_step_time": 16.952894687652588
    },
    {
      "epoch": 5.4759979248046875e-06,
      "model_forward_time": 5.923232793807983,
      "step": 5742
    },
    {
      "epoch": 5.4759979248046875e-06,
      "step": 5742,
      "training_step_time": 17.33430242538452
    },
    {
      "epoch": 5.476951599121094e-06,
      "model_forward_time": 5.558102130889893,
      "step": 5743
    },
    {
      "epoch": 5.476951599121094e-06,
      "step": 5743,
      "training_step_time": 16.972349643707275
    },
    {
      "epoch": 5.4779052734375e-06,
      "model_forward_time": 5.565601825714111,
      "step": 5744
    },
    {
      "epoch": 5.4779052734375e-06,
      "step": 5744,
      "training_step_time": 16.972577810287476
    },
    {
      "epoch": 5.478858947753906e-06,
      "model_forward_time": 5.5708699226379395,
      "step": 5745
    },
    {
      "epoch": 5.478858947753906e-06,
      "step": 5745,
      "training_step_time": 16.97852849960327
    },
    {
      "epoch": 5.479812622070313e-06,
      "model_forward_time": 5.575369119644165,
      "step": 5746
    },
    {
      "epoch": 5.479812622070313e-06,
      "step": 5746,
      "training_step_time": 16.9992253780365
    },
    {
      "epoch": 5.480766296386719e-06,
      "model_forward_time": 5.5892510414123535,
      "step": 5747
    },
    {
      "epoch": 5.480766296386719e-06,
      "step": 5747,
      "training_step_time": 17.418797492980957
    },
    {
      "epoch": 5.481719970703125e-06,
      "model_forward_time": 5.615826368331909,
      "step": 5748
    },
    {
      "epoch": 5.481719970703125e-06,
      "step": 5748,
      "training_step_time": 17.131481647491455
    },
    {
      "epoch": 5.482673645019531e-06,
      "model_forward_time": 5.674898624420166,
      "step": 5749
    },
    {
      "epoch": 5.482673645019531e-06,
      "step": 5749,
      "training_step_time": 17.184913635253906
    },
    {
      "epoch": 5.4836273193359375e-06,
      "grad_norm": 0.298467755317688,
      "learning_rate": 1e-05,
      "loss": 0.0614,
      "step": 5750
    },
    {
      "epoch": 5.4836273193359375e-06,
      "model_forward_time": 5.614456415176392,
      "step": 5750
    },
    {
      "dynamics_loss_avg": 0.04840863477438688,
      "epoch": 5.4836273193359375e-06,
      "step": 5750
    },
    {
      "action_loss_avg": 0.01074489471502602,
      "epoch": 5.4836273193359375e-06,
      "step": 5750
    },
    {
      "epoch": 5.4836273193359375e-06,
      "step": 5750,
      "training_step_time": 17.09754991531372
    },
    {
      "epoch": 5.484580993652344e-06,
      "model_forward_time": 5.607646226882935,
      "step": 5751
    },
    {
      "epoch": 5.484580993652344e-06,
      "step": 5751,
      "training_step_time": 17.090476751327515
    },
    {
      "epoch": 5.48553466796875e-06,
      "model_forward_time": 5.611401319503784,
      "step": 5752
    },
    {
      "epoch": 5.48553466796875e-06,
      "step": 5752,
      "training_step_time": 17.092461585998535
    },
    {
      "epoch": 5.486488342285156e-06,
      "model_forward_time": 5.972952604293823,
      "step": 5753
    },
    {
      "epoch": 5.486488342285156e-06,
      "step": 5753,
      "training_step_time": 17.45657730102539
    },
    {
      "epoch": 5.487442016601562e-06,
      "model_forward_time": 5.6035027503967285,
      "step": 5754
    },
    {
      "epoch": 5.487442016601562e-06,
      "step": 5754,
      "training_step_time": 17.09162735939026
    },
    {
      "epoch": 5.488395690917969e-06,
      "model_forward_time": 5.616300582885742,
      "step": 5755
    },
    {
      "epoch": 5.488395690917969e-06,
      "step": 5755,
      "training_step_time": 17.107144117355347
    },
    {
      "epoch": 5.489349365234375e-06,
      "model_forward_time": 5.599850177764893,
      "step": 5756
    },
    {
      "epoch": 5.489349365234375e-06,
      "step": 5756,
      "training_step_time": 17.090821743011475
    },
    {
      "epoch": 5.490303039550781e-06,
      "model_forward_time": 5.605544567108154,
      "step": 5757
    },
    {
      "epoch": 5.490303039550781e-06,
      "step": 5757,
      "training_step_time": 17.09510111808777
    },
    {
      "epoch": 5.4912567138671875e-06,
      "model_forward_time": 5.605482578277588,
      "step": 5758
    },
    {
      "epoch": 5.4912567138671875e-06,
      "step": 5758,
      "training_step_time": 17.45888113975525
    },
    {
      "epoch": 5.492210388183594e-06,
      "model_forward_time": 5.611394643783569,
      "step": 5759
    },
    {
      "epoch": 5.492210388183594e-06,
      "step": 5759,
      "training_step_time": 17.098806142807007
    },
    {
      "epoch": 5.4931640625e-06,
      "grad_norm": 1.0585094690322876,
      "learning_rate": 1e-05,
      "loss": 0.0481,
      "step": 5760
    },
    {
      "epoch": 5.4931640625e-06,
      "model_forward_time": 5.605625867843628,
      "step": 5760
    },
    {
      "dynamics_loss_avg": 0.040804208675399425,
      "epoch": 5.4931640625e-06,
      "step": 5760
    },
    {
      "action_loss_avg": 0.010788132320158184,
      "epoch": 5.4931640625e-06,
      "step": 5760
    },
    {
      "epoch": 5.4931640625e-06,
      "step": 5760,
      "training_step_time": 17.090094089508057
    },
    {
      "epoch": 5.494117736816406e-06,
      "model_forward_time": 5.601880073547363,
      "step": 5761
    },
    {
      "epoch": 5.494117736816406e-06,
      "step": 5761,
      "training_step_time": 17.053281545639038
    },
    {
      "epoch": 5.495071411132812e-06,
      "model_forward_time": 5.582491874694824,
      "step": 5762
    },
    {
      "epoch": 5.495071411132812e-06,
      "step": 5762,
      "training_step_time": 17.019915342330933
    },
    {
      "epoch": 5.496025085449219e-06,
      "model_forward_time": 5.581558704376221,
      "step": 5763
    },
    {
      "epoch": 5.496025085449219e-06,
      "step": 5763,
      "training_step_time": 16.997141122817993
    },
    {
      "epoch": 5.496978759765625e-06,
      "model_forward_time": 5.959210634231567,
      "step": 5764
    },
    {
      "epoch": 5.496978759765625e-06,
      "step": 5764,
      "training_step_time": 17.403888940811157
    },
    {
      "epoch": 5.497932434082031e-06,
      "model_forward_time": 5.624224901199341,
      "step": 5765
    },
    {
      "epoch": 5.497932434082031e-06,
      "step": 5765,
      "training_step_time": 17.116475820541382
    },
    {
      "epoch": 5.4988861083984375e-06,
      "model_forward_time": 5.624165058135986,
      "step": 5766
    },
    {
      "epoch": 5.4988861083984375e-06,
      "step": 5766,
      "training_step_time": 17.132852792739868
    },
    {
      "epoch": 5.499839782714844e-06,
      "model_forward_time": 5.612289190292358,
      "step": 5767
    },
    {
      "epoch": 5.499839782714844e-06,
      "step": 5767,
      "training_step_time": 17.11111307144165
    },
    {
      "epoch": 5.50079345703125e-06,
      "model_forward_time": 5.601284027099609,
      "step": 5768
    },
    {
      "epoch": 5.50079345703125e-06,
      "step": 5768,
      "training_step_time": 17.09098720550537
    },
    {
      "epoch": 5.501747131347656e-06,
      "model_forward_time": 5.966611862182617,
      "step": 5769
    },
    {
      "epoch": 5.501747131347656e-06,
      "step": 5769,
      "training_step_time": 17.4362850189209
    },
    {
      "epoch": 5.502700805664062e-06,
      "grad_norm": 0.8265737891197205,
      "learning_rate": 1e-05,
      "loss": 0.0676,
      "step": 5770
    },
    {
      "epoch": 5.502700805664062e-06,
      "model_forward_time": 5.617444753646851,
      "step": 5770
    },
    {
      "dynamics_loss_avg": 0.059753477945923804,
      "epoch": 5.502700805664062e-06,
      "step": 5770
    },
    {
      "action_loss_avg": 0.01351109128445387,
      "epoch": 5.502700805664062e-06,
      "step": 5770
    },
    {
      "epoch": 5.502700805664062e-06,
      "step": 5770,
      "training_step_time": 17.11985969543457
    },
    {
      "epoch": 5.503654479980469e-06,
      "model_forward_time": 5.612719535827637,
      "step": 5771
    },
    {
      "epoch": 5.503654479980469e-06,
      "step": 5771,
      "training_step_time": 17.124510526657104
    },
    {
      "epoch": 5.504608154296875e-06,
      "model_forward_time": 5.615004539489746,
      "step": 5772
    },
    {
      "epoch": 5.504608154296875e-06,
      "step": 5772,
      "training_step_time": 17.129347801208496
    },
    {
      "epoch": 5.505561828613281e-06,
      "model_forward_time": 5.620950937271118,
      "step": 5773
    },
    {
      "epoch": 5.505561828613281e-06,
      "step": 5773,
      "training_step_time": 17.126423835754395
    },
    {
      "epoch": 5.5065155029296875e-06,
      "model_forward_time": 5.616711616516113,
      "step": 5774
    },
    {
      "epoch": 5.5065155029296875e-06,
      "step": 5774,
      "training_step_time": 17.118173360824585
    },
    {
      "epoch": 5.507469177246094e-06,
      "model_forward_time": 5.985694646835327,
      "step": 5775
    },
    {
      "epoch": 5.507469177246094e-06,
      "step": 5775,
      "training_step_time": 17.46703600883484
    },
    {
      "epoch": 5.5084228515625e-06,
      "model_forward_time": 5.606297016143799,
      "step": 5776
    },
    {
      "epoch": 5.5084228515625e-06,
      "step": 5776,
      "training_step_time": 17.08841633796692
    },
    {
      "epoch": 5.509376525878906e-06,
      "model_forward_time": 5.603284120559692,
      "step": 5777
    },
    {
      "epoch": 5.509376525878906e-06,
      "step": 5777,
      "training_step_time": 17.07067894935608
    },
    {
      "epoch": 5.510330200195312e-06,
      "model_forward_time": 5.598889350891113,
      "step": 5778
    },
    {
      "epoch": 5.510330200195312e-06,
      "step": 5778,
      "training_step_time": 17.047730445861816
    },
    {
      "epoch": 5.511283874511719e-06,
      "model_forward_time": 5.605112552642822,
      "step": 5779
    },
    {
      "epoch": 5.511283874511719e-06,
      "step": 5779,
      "training_step_time": 17.060025215148926
    },
    {
      "epoch": 5.512237548828125e-06,
      "grad_norm": 0.6158220171928406,
      "learning_rate": 1e-05,
      "loss": 0.064,
      "step": 5780
    },
    {
      "epoch": 5.512237548828125e-06,
      "model_forward_time": 5.599032640457153,
      "step": 5780
    },
    {
      "dynamics_loss_avg": 0.04659372782334685,
      "epoch": 5.512237548828125e-06,
      "step": 5780
    },
    {
      "action_loss_avg": 0.012344100372865796,
      "epoch": 5.512237548828125e-06,
      "step": 5780
    },
    {
      "epoch": 5.512237548828125e-06,
      "step": 5780,
      "training_step_time": 17.41848587989807
    },
    {
      "epoch": 5.513191223144531e-06,
      "model_forward_time": 5.589252948760986,
      "step": 5781
    },
    {
      "epoch": 5.513191223144531e-06,
      "step": 5781,
      "training_step_time": 17.045092821121216
    },
    {
      "epoch": 5.5141448974609375e-06,
      "model_forward_time": 5.593743562698364,
      "step": 5782
    },
    {
      "epoch": 5.5141448974609375e-06,
      "step": 5782,
      "training_step_time": 17.082610845565796
    },
    {
      "epoch": 5.515098571777344e-06,
      "model_forward_time": 5.611979722976685,
      "step": 5783
    },
    {
      "epoch": 5.515098571777344e-06,
      "step": 5783,
      "training_step_time": 17.109450817108154
    },
    {
      "epoch": 5.51605224609375e-06,
      "model_forward_time": 5.613765478134155,
      "step": 5784
    },
    {
      "epoch": 5.51605224609375e-06,
      "step": 5784,
      "training_step_time": 17.118342876434326
    },
    {
      "epoch": 5.517005920410156e-06,
      "model_forward_time": 5.615316867828369,
      "step": 5785
    },
    {
      "epoch": 5.517005920410156e-06,
      "step": 5785,
      "training_step_time": 17.10022282600403
    },
    {
      "epoch": 5.517959594726562e-06,
      "model_forward_time": 6.004125595092773,
      "step": 5786
    },
    {
      "epoch": 5.517959594726562e-06,
      "step": 5786,
      "training_step_time": 17.48126220703125
    },
    {
      "epoch": 5.518913269042969e-06,
      "model_forward_time": 5.618362903594971,
      "step": 5787
    },
    {
      "epoch": 5.518913269042969e-06,
      "step": 5787,
      "training_step_time": 17.10824155807495
    },
    {
      "epoch": 5.519866943359375e-06,
      "model_forward_time": 5.609929084777832,
      "step": 5788
    },
    {
      "epoch": 5.519866943359375e-06,
      "step": 5788,
      "training_step_time": 17.089781522750854
    },
    {
      "epoch": 5.520820617675781e-06,
      "model_forward_time": 5.601412296295166,
      "step": 5789
    },
    {
      "epoch": 5.520820617675781e-06,
      "step": 5789,
      "training_step_time": 17.058894157409668
    },
    {
      "epoch": 5.5217742919921875e-06,
      "grad_norm": 0.5685917735099792,
      "learning_rate": 1e-05,
      "loss": 0.0798,
      "step": 5790
    },
    {
      "epoch": 5.5217742919921875e-06,
      "model_forward_time": 5.596499919891357,
      "step": 5790
    },
    {
      "dynamics_loss_avg": 0.06428675018250943,
      "epoch": 5.5217742919921875e-06,
      "step": 5790
    },
    {
      "action_loss_avg": 0.015422126092016698,
      "epoch": 5.5217742919921875e-06,
      "step": 5790
    },
    {
      "epoch": 5.5217742919921875e-06,
      "step": 5790,
      "training_step_time": 17.052497386932373
    },
    {
      "epoch": 5.522727966308594e-06,
      "model_forward_time": 5.58825159072876,
      "step": 5791
    },
    {
      "epoch": 5.522727966308594e-06,
      "step": 5791,
      "training_step_time": 17.396284341812134
    },
    {
      "epoch": 5.523681640625e-06,
      "model_forward_time": 5.6034018993377686,
      "step": 5792
    },
    {
      "epoch": 5.523681640625e-06,
      "step": 5792,
      "training_step_time": 17.069665670394897
    },
    {
      "epoch": 5.524635314941406e-06,
      "model_forward_time": 5.594825744628906,
      "step": 5793
    },
    {
      "epoch": 5.524635314941406e-06,
      "step": 5793,
      "training_step_time": 17.06343364715576
    },
    {
      "epoch": 5.525588989257812e-06,
      "model_forward_time": 5.600167274475098,
      "step": 5794
    },
    {
      "epoch": 5.525588989257812e-06,
      "step": 5794,
      "training_step_time": 17.093502283096313
    },
    {
      "epoch": 5.526542663574219e-06,
      "model_forward_time": 5.609380483627319,
      "step": 5795
    },
    {
      "epoch": 5.526542663574219e-06,
      "step": 5795,
      "training_step_time": 17.116313695907593
    },
    {
      "epoch": 5.527496337890625e-06,
      "model_forward_time": 5.611478328704834,
      "step": 5796
    },
    {
      "epoch": 5.527496337890625e-06,
      "step": 5796,
      "training_step_time": 17.078288316726685
    },
    {
      "epoch": 5.528450012207031e-06,
      "model_forward_time": 5.966621398925781,
      "step": 5797
    },
    {
      "epoch": 5.528450012207031e-06,
      "step": 5797,
      "training_step_time": 17.407536268234253
    },
    {
      "epoch": 5.5294036865234375e-06,
      "model_forward_time": 5.571876049041748,
      "step": 5798
    },
    {
      "epoch": 5.5294036865234375e-06,
      "step": 5798,
      "training_step_time": 16.982433557510376
    },
    {
      "epoch": 5.530357360839844e-06,
      "model_forward_time": 5.565849542617798,
      "step": 5799
    },
    {
      "epoch": 5.530357360839844e-06,
      "step": 5799,
      "training_step_time": 16.985470294952393
    },
    {
      "epoch": 5.53131103515625e-06,
      "grad_norm": 0.27308493852615356,
      "learning_rate": 1e-05,
      "loss": 0.0639,
      "step": 5800
    },
    {
      "epoch": 5.53131103515625e-06,
      "model_forward_time": 5.573877811431885,
      "step": 5800
    },
    {
      "dynamics_loss_avg": 0.057660356536507605,
      "epoch": 5.53131103515625e-06,
      "step": 5800
    },
    {
      "action_loss_avg": 0.01316199554130435,
      "epoch": 5.53131103515625e-06,
      "step": 5800
    },
    {
      "epoch": 5.53131103515625e-06,
      "step": 5800,
      "training_step_time": 17.00653600692749
    },
    {
      "epoch": 5.532264709472656e-06,
      "model_forward_time": 5.568575859069824,
      "step": 5801
    },
    {
      "epoch": 5.532264709472656e-06,
      "step": 5801,
      "training_step_time": 16.987722873687744
    },
    {
      "epoch": 5.533218383789062e-06,
      "model_forward_time": 5.943098068237305,
      "step": 5802
    },
    {
      "epoch": 5.533218383789062e-06,
      "step": 5802,
      "training_step_time": 17.382346153259277
    },
    {
      "epoch": 5.534172058105469e-06,
      "model_forward_time": 5.599475145339966,
      "step": 5803
    },
    {
      "epoch": 5.534172058105469e-06,
      "step": 5803,
      "training_step_time": 17.070979118347168
    },
    {
      "epoch": 5.535125732421875e-06,
      "model_forward_time": 5.606291770935059,
      "step": 5804
    },
    {
      "epoch": 5.535125732421875e-06,
      "step": 5804,
      "training_step_time": 17.059157848358154
    },
    {
      "epoch": 5.536079406738281e-06,
      "model_forward_time": 5.597705125808716,
      "step": 5805
    },
    {
      "epoch": 5.536079406738281e-06,
      "step": 5805,
      "training_step_time": 17.063270807266235
    },
    {
      "epoch": 5.5370330810546875e-06,
      "model_forward_time": 5.60242486000061,
      "step": 5806
    },
    {
      "epoch": 5.5370330810546875e-06,
      "step": 5806,
      "training_step_time": 17.07118821144104
    },
    {
      "epoch": 5.537986755371094e-06,
      "model_forward_time": 5.595870733261108,
      "step": 5807
    },
    {
      "epoch": 5.537986755371094e-06,
      "step": 5807,
      "training_step_time": 17.031315803527832
    },
    {
      "epoch": 5.5389404296875e-06,
      "model_forward_time": 5.952935457229614,
      "step": 5808
    },
    {
      "epoch": 5.5389404296875e-06,
      "step": 5808,
      "training_step_time": 17.393706560134888
    },
    {
      "epoch": 5.539894104003906e-06,
      "model_forward_time": 5.592845916748047,
      "step": 5809
    },
    {
      "epoch": 5.539894104003906e-06,
      "step": 5809,
      "training_step_time": 17.023066997528076
    },
    {
      "epoch": 5.540847778320312e-06,
      "grad_norm": 1.709260106086731,
      "learning_rate": 1e-05,
      "loss": 0.0855,
      "step": 5810
    },
    {
      "epoch": 5.540847778320312e-06,
      "model_forward_time": 5.590760707855225,
      "step": 5810
    },
    {
      "dynamics_loss_avg": 0.06261555310338736,
      "epoch": 5.540847778320312e-06,
      "step": 5810
    },
    {
      "action_loss_avg": 0.014206092548556625,
      "epoch": 5.540847778320312e-06,
      "step": 5810
    },
    {
      "epoch": 5.540847778320312e-06,
      "step": 5810,
      "training_step_time": 17.01356840133667
    },
    {
      "epoch": 5.541801452636719e-06,
      "model_forward_time": 5.587261199951172,
      "step": 5811
    },
    {
      "epoch": 5.541801452636719e-06,
      "step": 5811,
      "training_step_time": 17.022449016571045
    },
    {
      "epoch": 5.542755126953125e-06,
      "model_forward_time": 5.592435359954834,
      "step": 5812
    },
    {
      "epoch": 5.542755126953125e-06,
      "step": 5812,
      "training_step_time": 17.04437756538391
    },
    {
      "epoch": 5.543708801269531e-06,
      "model_forward_time": 5.597525596618652,
      "step": 5813
    },
    {
      "epoch": 5.543708801269531e-06,
      "step": 5813,
      "training_step_time": 17.428797006607056
    },
    {
      "epoch": 5.5446624755859376e-06,
      "model_forward_time": 5.608349561691284,
      "step": 5814
    },
    {
      "epoch": 5.5446624755859376e-06,
      "step": 5814,
      "training_step_time": 17.116628646850586
    },
    {
      "epoch": 5.545616149902344e-06,
      "model_forward_time": 5.625604629516602,
      "step": 5815
    },
    {
      "epoch": 5.545616149902344e-06,
      "step": 5815,
      "training_step_time": 17.14334225654602
    },
    {
      "epoch": 5.54656982421875e-06,
      "model_forward_time": 5.618594646453857,
      "step": 5816
    },
    {
      "epoch": 5.54656982421875e-06,
      "step": 5816,
      "training_step_time": 17.106101751327515
    },
    {
      "epoch": 5.547523498535156e-06,
      "model_forward_time": 5.6017069816589355,
      "step": 5817
    },
    {
      "epoch": 5.547523498535156e-06,
      "step": 5817,
      "training_step_time": 17.079397916793823
    },
    {
      "epoch": 5.548477172851562e-06,
      "model_forward_time": 5.6134033203125,
      "step": 5818
    },
    {
      "epoch": 5.548477172851562e-06,
      "step": 5818,
      "training_step_time": 17.08577060699463
    },
    {
      "epoch": 5.549430847167969e-06,
      "model_forward_time": 5.96117901802063,
      "step": 5819
    },
    {
      "epoch": 5.549430847167969e-06,
      "step": 5819,
      "training_step_time": 17.414624214172363
    },
    {
      "epoch": 5.550384521484375e-06,
      "grad_norm": 0.5566040277481079,
      "learning_rate": 1e-05,
      "loss": 0.0709,
      "step": 5820
    },
    {
      "epoch": 5.550384521484375e-06,
      "model_forward_time": 5.587637901306152,
      "step": 5820
    },
    {
      "dynamics_loss_avg": 0.0572948326356709,
      "epoch": 5.550384521484375e-06,
      "step": 5820
    },
    {
      "action_loss_avg": 0.014295001327991486,
      "epoch": 5.550384521484375e-06,
      "step": 5820
    },
    {
      "epoch": 5.550384521484375e-06,
      "step": 5820,
      "training_step_time": 17.037961721420288
    },
    {
      "epoch": 5.551338195800781e-06,
      "model_forward_time": 5.585008382797241,
      "step": 5821
    },
    {
      "epoch": 5.551338195800781e-06,
      "step": 5821,
      "training_step_time": 17.030391454696655
    },
    {
      "epoch": 5.5522918701171876e-06,
      "model_forward_time": 5.590862989425659,
      "step": 5822
    },
    {
      "epoch": 5.5522918701171876e-06,
      "step": 5822,
      "training_step_time": 17.044077157974243
    },
    {
      "epoch": 5.553245544433594e-06,
      "model_forward_time": 5.587815523147583,
      "step": 5823
    },
    {
      "epoch": 5.553245544433594e-06,
      "step": 5823,
      "training_step_time": 17.038897037506104
    },
    {
      "epoch": 5.55419921875e-06,
      "model_forward_time": 5.952616214752197,
      "step": 5824
    },
    {
      "epoch": 5.55419921875e-06,
      "step": 5824,
      "training_step_time": 17.4081289768219
    },
    {
      "epoch": 5.555152893066406e-06,
      "model_forward_time": 5.594238042831421,
      "step": 5825
    },
    {
      "epoch": 5.555152893066406e-06,
      "step": 5825,
      "training_step_time": 17.065373182296753
    },
    {
      "epoch": 5.556106567382812e-06,
      "model_forward_time": 5.61505389213562,
      "step": 5826
    },
    {
      "epoch": 5.556106567382812e-06,
      "step": 5826,
      "training_step_time": 17.095968008041382
    },
    {
      "epoch": 5.557060241699219e-06,
      "model_forward_time": 5.615428686141968,
      "step": 5827
    },
    {
      "epoch": 5.557060241699219e-06,
      "step": 5827,
      "training_step_time": 17.10945200920105
    },
    {
      "epoch": 5.558013916015625e-06,
      "model_forward_time": 5.604271411895752,
      "step": 5828
    },
    {
      "epoch": 5.558013916015625e-06,
      "step": 5828,
      "training_step_time": 17.064099550247192
    },
    {
      "epoch": 5.558967590332031e-06,
      "model_forward_time": 5.588607311248779,
      "step": 5829
    },
    {
      "epoch": 5.558967590332031e-06,
      "step": 5829,
      "training_step_time": 17.03180956840515
    },
    {
      "epoch": 5.5599212646484376e-06,
      "grad_norm": 1.5773710012435913,
      "learning_rate": 1e-05,
      "loss": 0.0493,
      "step": 5830
    },
    {
      "epoch": 5.5599212646484376e-06,
      "model_forward_time": 5.9646525382995605,
      "step": 5830
    },
    {
      "dynamics_loss_avg": 0.03840046934783459,
      "epoch": 5.5599212646484376e-06,
      "step": 5830
    },
    {
      "action_loss_avg": 0.007252553547732532,
      "epoch": 5.5599212646484376e-06,
      "step": 5830
    },
    {
      "epoch": 5.5599212646484376e-06,
      "step": 5830,
      "training_step_time": 17.382265329360962
    },
    {
      "epoch": 5.560874938964844e-06,
      "model_forward_time": 5.5907347202301025,
      "step": 5831
    },
    {
      "epoch": 5.560874938964844e-06,
      "step": 5831,
      "training_step_time": 17.051133155822754
    },
    {
      "epoch": 5.56182861328125e-06,
      "model_forward_time": 4.2865355014801025,
      "step": 5832
    },
    {
      "epoch": 5.56182861328125e-06,
      "step": 5832,
      "training_step_time": 12.816753387451172
    },
    {
      "epoch": 5.5627822875976564e-06,
      "model_forward_time": 5.62078857421875,
      "step": 5833
    },
    {
      "epoch": 5.5627822875976564e-06,
      "step": 5833,
      "training_step_time": 17.11885714530945
    },
    {
      "epoch": 5.563735961914062e-06,
      "model_forward_time": 5.615546464920044,
      "step": 5834
    },
    {
      "epoch": 5.563735961914062e-06,
      "step": 5834,
      "training_step_time": 17.103116989135742
    },
    {
      "epoch": 5.564689636230469e-06,
      "model_forward_time": 5.955630540847778,
      "step": 5835
    },
    {
      "epoch": 5.564689636230469e-06,
      "step": 5835,
      "training_step_time": 17.40920114517212
    },
    {
      "epoch": 5.565643310546875e-06,
      "model_forward_time": 5.588747024536133,
      "step": 5836
    },
    {
      "epoch": 5.565643310546875e-06,
      "step": 5836,
      "training_step_time": 17.052608251571655
    },
    {
      "epoch": 5.566596984863281e-06,
      "model_forward_time": 5.594860792160034,
      "step": 5837
    },
    {
      "epoch": 5.566596984863281e-06,
      "step": 5837,
      "training_step_time": 17.04773473739624
    },
    {
      "epoch": 5.5675506591796876e-06,
      "model_forward_time": 5.593300819396973,
      "step": 5838
    },
    {
      "epoch": 5.5675506591796876e-06,
      "step": 5838,
      "training_step_time": 17.030649423599243
    },
    {
      "epoch": 5.568504333496094e-06,
      "model_forward_time": 5.6025683879852295,
      "step": 5839
    },
    {
      "epoch": 5.568504333496094e-06,
      "step": 5839,
      "training_step_time": 17.07224464416504
    },
    {
      "epoch": 5.5694580078125e-06,
      "grad_norm": 2.266660451889038,
      "learning_rate": 1e-05,
      "loss": 0.0727,
      "step": 5840
    },
    {
      "epoch": 5.5694580078125e-06,
      "model_forward_time": 5.59878134727478,
      "step": 5840
    },
    {
      "dynamics_loss_avg": 0.05784985683858394,
      "epoch": 5.5694580078125e-06,
      "step": 5840
    },
    {
      "action_loss_avg": 0.015093043586239219,
      "epoch": 5.5694580078125e-06,
      "step": 5840
    },
    {
      "epoch": 5.5694580078125e-06,
      "step": 5840,
      "training_step_time": 17.061539888381958
    },
    {
      "epoch": 5.5704116821289064e-06,
      "model_forward_time": 5.961793422698975,
      "step": 5841
    },
    {
      "epoch": 5.5704116821289064e-06,
      "step": 5841,
      "training_step_time": 17.39223003387451
    },
    {
      "epoch": 5.571365356445312e-06,
      "model_forward_time": 5.594725131988525,
      "step": 5842
    },
    {
      "epoch": 5.571365356445312e-06,
      "step": 5842,
      "training_step_time": 17.031658172607422
    },
    {
      "epoch": 5.572319030761719e-06,
      "model_forward_time": 5.588068723678589,
      "step": 5843
    },
    {
      "epoch": 5.572319030761719e-06,
      "step": 5843,
      "training_step_time": 17.061075448989868
    },
    {
      "epoch": 5.573272705078125e-06,
      "model_forward_time": 5.612360000610352,
      "step": 5844
    },
    {
      "epoch": 5.573272705078125e-06,
      "step": 5844,
      "training_step_time": 17.126598119735718
    },
    {
      "epoch": 5.574226379394531e-06,
      "model_forward_time": 5.626446485519409,
      "step": 5845
    },
    {
      "epoch": 5.574226379394531e-06,
      "step": 5845,
      "training_step_time": 17.114855766296387
    },
    {
      "epoch": 5.575180053710938e-06,
      "model_forward_time": 5.946850299835205,
      "step": 5846
    },
    {
      "epoch": 5.575180053710938e-06,
      "step": 5846,
      "training_step_time": 17.397464752197266
    },
    {
      "epoch": 5.576133728027344e-06,
      "model_forward_time": 5.582313060760498,
      "step": 5847
    },
    {
      "epoch": 5.576133728027344e-06,
      "step": 5847,
      "training_step_time": 17.01396870613098
    },
    {
      "epoch": 5.57708740234375e-06,
      "model_forward_time": 5.575204610824585,
      "step": 5848
    },
    {
      "epoch": 5.57708740234375e-06,
      "step": 5848,
      "training_step_time": 16.995885848999023
    },
    {
      "epoch": 5.5780410766601564e-06,
      "model_forward_time": 5.578868865966797,
      "step": 5849
    },
    {
      "epoch": 5.5780410766601564e-06,
      "step": 5849,
      "training_step_time": 16.99550175666809
    },
    {
      "epoch": 5.578994750976562e-06,
      "grad_norm": 1.1151363849639893,
      "learning_rate": 1e-05,
      "loss": 0.0559,
      "step": 5850
    },
    {
      "epoch": 5.578994750976562e-06,
      "model_forward_time": 5.578798532485962,
      "step": 5850
    },
    {
      "dynamics_loss_avg": 0.045995454862713815,
      "epoch": 5.578994750976562e-06,
      "step": 5850
    },
    {
      "action_loss_avg": 0.00979156766552478,
      "epoch": 5.578994750976562e-06,
      "step": 5850
    },
    {
      "epoch": 5.578994750976562e-06,
      "step": 5850,
      "training_step_time": 17.012135982513428
    },
    {
      "epoch": 5.579948425292969e-06,
      "model_forward_time": 5.986408710479736,
      "step": 5851
    },
    {
      "epoch": 5.579948425292969e-06,
      "step": 5851,
      "training_step_time": 17.48287034034729
    },
    {
      "epoch": 5.580902099609375e-06,
      "model_forward_time": 5.610814094543457,
      "step": 5852
    },
    {
      "epoch": 5.580902099609375e-06,
      "step": 5852,
      "training_step_time": 17.125457763671875
    },
    {
      "epoch": 5.581855773925781e-06,
      "model_forward_time": 5.627070188522339,
      "step": 5853
    },
    {
      "epoch": 5.581855773925781e-06,
      "step": 5853,
      "training_step_time": 17.120919466018677
    },
    {
      "epoch": 5.582809448242188e-06,
      "model_forward_time": 5.621622562408447,
      "step": 5854
    },
    {
      "epoch": 5.582809448242188e-06,
      "step": 5854,
      "training_step_time": 17.111084699630737
    },
    {
      "epoch": 5.583763122558594e-06,
      "model_forward_time": 5.6048102378845215,
      "step": 5855
    },
    {
      "epoch": 5.583763122558594e-06,
      "step": 5855,
      "training_step_time": 17.10187864303589
    },
    {
      "epoch": 5.584716796875e-06,
      "model_forward_time": 5.59817099571228,
      "step": 5856
    },
    {
      "epoch": 5.584716796875e-06,
      "step": 5856,
      "training_step_time": 16.998385429382324
    },
    {
      "epoch": 5.5856704711914064e-06,
      "model_forward_time": 5.954344034194946,
      "step": 5857
    },
    {
      "epoch": 5.5856704711914064e-06,
      "step": 5857,
      "training_step_time": 17.421546697616577
    },
    {
      "epoch": 5.586624145507812e-06,
      "model_forward_time": 5.588014125823975,
      "step": 5858
    },
    {
      "epoch": 5.586624145507812e-06,
      "step": 5858,
      "training_step_time": 17.024211645126343
    },
    {
      "epoch": 5.587577819824219e-06,
      "model_forward_time": 5.583423137664795,
      "step": 5859
    },
    {
      "epoch": 5.587577819824219e-06,
      "step": 5859,
      "training_step_time": 17.018391132354736
    },
    {
      "epoch": 5.588531494140625e-06,
      "grad_norm": 1.4521377086639404,
      "learning_rate": 1e-05,
      "loss": 0.0727,
      "step": 5860
    },
    {
      "epoch": 5.588531494140625e-06,
      "model_forward_time": 5.579238414764404,
      "step": 5860
    },
    {
      "dynamics_loss_avg": 0.06310273576527833,
      "epoch": 5.588531494140625e-06,
      "step": 5860
    },
    {
      "action_loss_avg": 0.013333016354590655,
      "epoch": 5.588531494140625e-06,
      "step": 5860
    },
    {
      "epoch": 5.588531494140625e-06,
      "step": 5860,
      "training_step_time": 16.985666036605835
    },
    {
      "epoch": 5.589485168457031e-06,
      "model_forward_time": 5.591153144836426,
      "step": 5861
    },
    {
      "epoch": 5.589485168457031e-06,
      "step": 5861,
      "training_step_time": 17.035273551940918
    },
    {
      "epoch": 5.590438842773438e-06,
      "model_forward_time": 5.963288307189941,
      "step": 5862
    },
    {
      "epoch": 5.590438842773438e-06,
      "step": 5862,
      "training_step_time": 17.43233585357666
    },
    {
      "epoch": 5.591392517089844e-06,
      "model_forward_time": 5.5991597175598145,
      "step": 5863
    },
    {
      "epoch": 5.591392517089844e-06,
      "step": 5863,
      "training_step_time": 17.073381662368774
    },
    {
      "epoch": 5.59234619140625e-06,
      "model_forward_time": 5.610941171646118,
      "step": 5864
    },
    {
      "epoch": 5.59234619140625e-06,
      "step": 5864,
      "training_step_time": 17.077221632003784
    },
    {
      "epoch": 5.5932998657226565e-06,
      "model_forward_time": 5.593900442123413,
      "step": 5865
    },
    {
      "epoch": 5.5932998657226565e-06,
      "step": 5865,
      "training_step_time": 17.037328243255615
    },
    {
      "epoch": 5.594253540039062e-06,
      "model_forward_time": 5.598076820373535,
      "step": 5866
    },
    {
      "epoch": 5.594253540039062e-06,
      "step": 5866,
      "training_step_time": 17.04964566230774
    },
    {
      "epoch": 5.595207214355469e-06,
      "model_forward_time": 5.58537483215332,
      "step": 5867
    },
    {
      "epoch": 5.595207214355469e-06,
      "step": 5867,
      "training_step_time": 17.377328395843506
    },
    {
      "epoch": 5.596160888671875e-06,
      "model_forward_time": 5.590108871459961,
      "step": 5868
    },
    {
      "epoch": 5.596160888671875e-06,
      "step": 5868,
      "training_step_time": 17.05135726928711
    },
    {
      "epoch": 5.597114562988281e-06,
      "model_forward_time": 5.59697699546814,
      "step": 5869
    },
    {
      "epoch": 5.597114562988281e-06,
      "step": 5869,
      "training_step_time": 17.057199239730835
    },
    {
      "epoch": 5.598068237304688e-06,
      "grad_norm": 0.8797191977500916,
      "learning_rate": 1e-05,
      "loss": 0.0541,
      "step": 5870
    },
    {
      "epoch": 5.598068237304688e-06,
      "model_forward_time": 5.605799913406372,
      "step": 5870
    },
    {
      "dynamics_loss_avg": 0.04300035759806633,
      "epoch": 5.598068237304688e-06,
      "step": 5870
    },
    {
      "action_loss_avg": 0.01069547636434436,
      "epoch": 5.598068237304688e-06,
      "step": 5870
    },
    {
      "epoch": 5.598068237304688e-06,
      "step": 5870,
      "training_step_time": 17.093727827072144
    },
    {
      "epoch": 5.599021911621094e-06,
      "model_forward_time": 5.622671604156494,
      "step": 5871
    },
    {
      "epoch": 5.599021911621094e-06,
      "step": 5871,
      "training_step_time": 17.122112035751343
    },
    {
      "epoch": 5.5999755859375e-06,
      "model_forward_time": 5.613570928573608,
      "step": 5872
    },
    {
      "epoch": 5.5999755859375e-06,
      "step": 5872,
      "training_step_time": 17.100225925445557
    },
    {
      "epoch": 5.6009292602539065e-06,
      "model_forward_time": 5.9834840297698975,
      "step": 5873
    },
    {
      "epoch": 5.6009292602539065e-06,
      "step": 5873,
      "training_step_time": 17.465725660324097
    },
    {
      "epoch": 5.601882934570312e-06,
      "model_forward_time": 5.606513977050781,
      "step": 5874
    },
    {
      "epoch": 5.601882934570312e-06,
      "step": 5874,
      "training_step_time": 17.068899154663086
    },
    {
      "epoch": 5.602836608886719e-06,
      "model_forward_time": 5.597169637680054,
      "step": 5875
    },
    {
      "epoch": 5.602836608886719e-06,
      "step": 5875,
      "training_step_time": 17.058411836624146
    },
    {
      "epoch": 5.603790283203125e-06,
      "model_forward_time": 5.599576950073242,
      "step": 5876
    },
    {
      "epoch": 5.603790283203125e-06,
      "step": 5876,
      "training_step_time": 17.07509446144104
    },
    {
      "epoch": 5.604743957519531e-06,
      "model_forward_time": 5.593337297439575,
      "step": 5877
    },
    {
      "epoch": 5.604743957519531e-06,
      "step": 5877,
      "training_step_time": 17.059327602386475
    },
    {
      "epoch": 5.605697631835938e-06,
      "model_forward_time": 5.946090221405029,
      "step": 5878
    },
    {
      "epoch": 5.605697631835938e-06,
      "step": 5878,
      "training_step_time": 17.368458032608032
    },
    {
      "epoch": 5.606651306152343e-06,
      "model_forward_time": 5.591855525970459,
      "step": 5879
    },
    {
      "epoch": 5.606651306152343e-06,
      "step": 5879,
      "training_step_time": 17.054399013519287
    },
    {
      "epoch": 5.60760498046875e-06,
      "grad_norm": 1.3474076986312866,
      "learning_rate": 1e-05,
      "loss": 0.0748,
      "step": 5880
    },
    {
      "epoch": 5.60760498046875e-06,
      "model_forward_time": 5.597542762756348,
      "step": 5880
    },
    {
      "dynamics_loss_avg": 0.05632849633693695,
      "epoch": 5.60760498046875e-06,
      "step": 5880
    },
    {
      "action_loss_avg": 0.01795043982565403,
      "epoch": 5.60760498046875e-06,
      "step": 5880
    },
    {
      "epoch": 5.60760498046875e-06,
      "step": 5880,
      "training_step_time": 17.04912281036377
    },
    {
      "epoch": 5.6085586547851565e-06,
      "model_forward_time": 5.602618932723999,
      "step": 5881
    },
    {
      "epoch": 5.6085586547851565e-06,
      "step": 5881,
      "training_step_time": 17.075534105300903
    },
    {
      "epoch": 5.609512329101562e-06,
      "model_forward_time": 5.6074607372283936,
      "step": 5882
    },
    {
      "epoch": 5.609512329101562e-06,
      "step": 5882,
      "training_step_time": 17.092745304107666
    },
    {
      "epoch": 5.610466003417969e-06,
      "model_forward_time": 5.611116409301758,
      "step": 5883
    },
    {
      "epoch": 5.610466003417969e-06,
      "step": 5883,
      "training_step_time": 17.102922201156616
    },
    {
      "epoch": 5.611419677734375e-06,
      "model_forward_time": 5.9827985763549805,
      "step": 5884
    },
    {
      "epoch": 5.611419677734375e-06,
      "step": 5884,
      "training_step_time": 17.445895433425903
    },
    {
      "epoch": 5.612373352050781e-06,
      "model_forward_time": 5.591912269592285,
      "step": 5885
    },
    {
      "epoch": 5.612373352050781e-06,
      "step": 5885,
      "training_step_time": 17.00978183746338
    },
    {
      "epoch": 5.613327026367188e-06,
      "model_forward_time": 5.578077554702759,
      "step": 5886
    },
    {
      "epoch": 5.613327026367188e-06,
      "step": 5886,
      "training_step_time": 16.978506803512573
    },
    {
      "epoch": 5.614280700683593e-06,
      "model_forward_time": 5.576872825622559,
      "step": 5887
    },
    {
      "epoch": 5.614280700683593e-06,
      "step": 5887,
      "training_step_time": 17.00827956199646
    },
    {
      "epoch": 5.615234375e-06,
      "model_forward_time": 5.575467824935913,
      "step": 5888
    },
    {
      "epoch": 5.615234375e-06,
      "step": 5888,
      "training_step_time": 17.026991844177246
    },
    {
      "epoch": 5.6161880493164065e-06,
      "model_forward_time": 5.941177606582642,
      "step": 5889
    },
    {
      "epoch": 5.6161880493164065e-06,
      "step": 5889,
      "training_step_time": 17.367228746414185
    },
    {
      "epoch": 5.617141723632812e-06,
      "grad_norm": 1.2810993194580078,
      "learning_rate": 1e-05,
      "loss": 0.0745,
      "step": 5890
    },
    {
      "epoch": 5.617141723632812e-06,
      "model_forward_time": 5.583997964859009,
      "step": 5890
    },
    {
      "dynamics_loss_avg": 0.06524281539022922,
      "epoch": 5.617141723632812e-06,
      "step": 5890
    },
    {
      "action_loss_avg": 0.013032295880839229,
      "epoch": 5.617141723632812e-06,
      "step": 5890
    },
    {
      "epoch": 5.617141723632812e-06,
      "step": 5890,
      "training_step_time": 17.004501581192017
    },
    {
      "epoch": 5.618095397949219e-06,
      "model_forward_time": 5.586775064468384,
      "step": 5891
    },
    {
      "epoch": 5.618095397949219e-06,
      "step": 5891,
      "training_step_time": 17.02679705619812
    },
    {
      "epoch": 5.619049072265625e-06,
      "model_forward_time": 5.6001598834991455,
      "step": 5892
    },
    {
      "epoch": 5.619049072265625e-06,
      "step": 5892,
      "training_step_time": 17.08772563934326
    },
    {
      "epoch": 5.620002746582031e-06,
      "model_forward_time": 5.619112730026245,
      "step": 5893
    },
    {
      "epoch": 5.620002746582031e-06,
      "step": 5893,
      "training_step_time": 17.139097452163696
    },
    {
      "epoch": 5.620956420898438e-06,
      "model_forward_time": 5.630210638046265,
      "step": 5894
    },
    {
      "epoch": 5.620956420898438e-06,
      "step": 5894,
      "training_step_time": 17.149500608444214
    },
    {
      "epoch": 5.621910095214843e-06,
      "model_forward_time": 5.970105886459351,
      "step": 5895
    },
    {
      "epoch": 5.621910095214843e-06,
      "step": 5895,
      "training_step_time": 17.45308232307434
    },
    {
      "epoch": 5.62286376953125e-06,
      "model_forward_time": 5.598052024841309,
      "step": 5896
    },
    {
      "epoch": 5.62286376953125e-06,
      "step": 5896,
      "training_step_time": 17.061069011688232
    },
    {
      "epoch": 5.6238174438476565e-06,
      "model_forward_time": 5.595864295959473,
      "step": 5897
    },
    {
      "epoch": 5.6238174438476565e-06,
      "step": 5897,
      "training_step_time": 17.050693035125732
    },
    {
      "epoch": 5.624771118164062e-06,
      "model_forward_time": 5.593286037445068,
      "step": 5898
    },
    {
      "epoch": 5.624771118164062e-06,
      "step": 5898,
      "training_step_time": 17.05557942390442
    },
    {
      "epoch": 5.625724792480469e-06,
      "model_forward_time": 5.604575157165527,
      "step": 5899
    },
    {
      "epoch": 5.625724792480469e-06,
      "step": 5899,
      "training_step_time": 17.03570532798767
    },
    {
      "epoch": 5.626678466796875e-06,
      "grad_norm": 1.214586615562439,
      "learning_rate": 1e-05,
      "loss": 0.0931,
      "step": 5900
    },
    {
      "epoch": 5.626678466796875e-06,
      "model_forward_time": 5.932279825210571,
      "step": 5900
    },
    {
      "dynamics_loss_avg": 0.07970836162567138,
      "epoch": 5.626678466796875e-06,
      "step": 5900
    },
    {
      "action_loss_avg": 0.01507459320127964,
      "epoch": 5.626678466796875e-06,
      "step": 5900
    },
    {
      "epoch": 5.626678466796875e-06,
      "step": 5900,
      "training_step_time": 17.385870456695557
    },
    {
      "epoch": 5.627632141113281e-06,
      "model_forward_time": 5.599877834320068,
      "step": 5901
    },
    {
      "epoch": 5.627632141113281e-06,
      "step": 5901,
      "training_step_time": 17.090983152389526
    },
    {
      "epoch": 5.628585815429688e-06,
      "model_forward_time": 5.603510618209839,
      "step": 5902
    },
    {
      "epoch": 5.628585815429688e-06,
      "step": 5902,
      "training_step_time": 17.087502241134644
    },
    {
      "epoch": 5.629539489746093e-06,
      "model_forward_time": 5.599111557006836,
      "step": 5903
    },
    {
      "epoch": 5.629539489746093e-06,
      "step": 5903,
      "training_step_time": 17.034780025482178
    },
    {
      "epoch": 5.6304931640625e-06,
      "model_forward_time": 5.590975999832153,
      "step": 5904
    },
    {
      "epoch": 5.6304931640625e-06,
      "step": 5904,
      "training_step_time": 17.018408060073853
    },
    {
      "epoch": 5.6314468383789065e-06,
      "model_forward_time": 5.57293438911438,
      "step": 5905
    },
    {
      "epoch": 5.6314468383789065e-06,
      "step": 5905,
      "training_step_time": 17.34873914718628
    },
    {
      "epoch": 5.632400512695312e-06,
      "model_forward_time": 5.576150894165039,
      "step": 5906
    },
    {
      "epoch": 5.632400512695312e-06,
      "step": 5906,
      "training_step_time": 17.014917850494385
    },
    {
      "epoch": 5.633354187011719e-06,
      "model_forward_time": 5.592384099960327,
      "step": 5907
    },
    {
      "epoch": 5.633354187011719e-06,
      "step": 5907,
      "training_step_time": 17.068172216415405
    },
    {
      "epoch": 5.634307861328125e-06,
      "model_forward_time": 5.598491907119751,
      "step": 5908
    },
    {
      "epoch": 5.634307861328125e-06,
      "step": 5908,
      "training_step_time": 17.072500228881836
    },
    {
      "epoch": 5.635261535644531e-06,
      "model_forward_time": 5.601764440536499,
      "step": 5909
    },
    {
      "epoch": 5.635261535644531e-06,
      "step": 5909,
      "training_step_time": 17.038078784942627
    },
    {
      "epoch": 5.636215209960938e-06,
      "grad_norm": 1.0925610065460205,
      "learning_rate": 1e-05,
      "loss": 0.0868,
      "step": 5910
    },
    {
      "epoch": 5.636215209960938e-06,
      "model_forward_time": 5.583874225616455,
      "step": 5910
    },
    {
      "dynamics_loss_avg": 0.06546939946711064,
      "epoch": 5.636215209960938e-06,
      "step": 5910
    },
    {
      "action_loss_avg": 0.013786684139631689,
      "epoch": 5.636215209960938e-06,
      "step": 5910
    },
    {
      "epoch": 5.636215209960938e-06,
      "step": 5910,
      "training_step_time": 17.051429271697998
    },
    {
      "epoch": 5.637168884277343e-06,
      "model_forward_time": 5.9333720207214355,
      "step": 5911
    },
    {
      "epoch": 5.637168884277343e-06,
      "step": 5911,
      "training_step_time": 17.39621067047119
    },
    {
      "epoch": 5.63812255859375e-06,
      "model_forward_time": 5.596558094024658,
      "step": 5912
    },
    {
      "epoch": 5.63812255859375e-06,
      "step": 5912,
      "training_step_time": 17.064655542373657
    },
    {
      "epoch": 5.6390762329101565e-06,
      "model_forward_time": 5.603730916976929,
      "step": 5913
    },
    {
      "epoch": 5.6390762329101565e-06,
      "step": 5913,
      "training_step_time": 17.091013431549072
    },
    {
      "epoch": 5.640029907226562e-06,
      "model_forward_time": 5.612905502319336,
      "step": 5914
    },
    {
      "epoch": 5.640029907226562e-06,
      "step": 5914,
      "training_step_time": 17.09972071647644
    },
    {
      "epoch": 5.640983581542969e-06,
      "model_forward_time": 5.6134443283081055,
      "step": 5915
    },
    {
      "epoch": 5.640983581542969e-06,
      "step": 5915,
      "training_step_time": 17.104984521865845
    },
    {
      "epoch": 5.641937255859375e-06,
      "model_forward_time": 5.979262828826904,
      "step": 5916
    },
    {
      "epoch": 5.641937255859375e-06,
      "step": 5916,
      "training_step_time": 17.446462869644165
    },
    {
      "epoch": 5.642890930175781e-06,
      "model_forward_time": 5.598856210708618,
      "step": 5917
    },
    {
      "epoch": 5.642890930175781e-06,
      "step": 5917,
      "training_step_time": 17.069748163223267
    },
    {
      "epoch": 5.643844604492188e-06,
      "model_forward_time": 5.602471113204956,
      "step": 5918
    },
    {
      "epoch": 5.643844604492188e-06,
      "step": 5918,
      "training_step_time": 17.07983374595642
    },
    {
      "epoch": 5.644798278808593e-06,
      "model_forward_time": 5.6256489753723145,
      "step": 5919
    },
    {
      "epoch": 5.644798278808593e-06,
      "step": 5919,
      "training_step_time": 17.131049394607544
    },
    {
      "epoch": 5.645751953125e-06,
      "grad_norm": 1.5503864288330078,
      "learning_rate": 1e-05,
      "loss": 0.0665,
      "step": 5920
    },
    {
      "epoch": 5.645751953125e-06,
      "model_forward_time": 5.6158318519592285,
      "step": 5920
    },
    {
      "dynamics_loss_avg": 0.056663834303617475,
      "epoch": 5.645751953125e-06,
      "step": 5920
    },
    {
      "action_loss_avg": 0.018140785209834576,
      "epoch": 5.645751953125e-06,
      "step": 5920
    },
    {
      "epoch": 5.645751953125e-06,
      "step": 5920,
      "training_step_time": 17.115171432495117
    },
    {
      "epoch": 5.6467056274414065e-06,
      "model_forward_time": 5.601940393447876,
      "step": 5921
    },
    {
      "epoch": 5.6467056274414065e-06,
      "step": 5921,
      "training_step_time": 17.035764932632446
    },
    {
      "epoch": 5.647659301757812e-06,
      "model_forward_time": 5.924187183380127,
      "step": 5922
    },
    {
      "epoch": 5.647659301757812e-06,
      "step": 5922,
      "training_step_time": 17.366040468215942
    },
    {
      "epoch": 5.648612976074219e-06,
      "model_forward_time": 5.573163747787476,
      "step": 5923
    },
    {
      "epoch": 5.648612976074219e-06,
      "step": 5923,
      "training_step_time": 16.983872652053833
    },
    {
      "epoch": 5.649566650390625e-06,
      "model_forward_time": 5.578999757766724,
      "step": 5924
    },
    {
      "epoch": 5.649566650390625e-06,
      "step": 5924,
      "training_step_time": 16.990942001342773
    },
    {
      "epoch": 5.650520324707031e-06,
      "model_forward_time": 5.5773069858551025,
      "step": 5925
    },
    {
      "epoch": 5.650520324707031e-06,
      "step": 5925,
      "training_step_time": 16.998734712600708
    },
    {
      "epoch": 5.651473999023438e-06,
      "model_forward_time": 5.576993465423584,
      "step": 5926
    },
    {
      "epoch": 5.651473999023438e-06,
      "step": 5926,
      "training_step_time": 17.007297039031982
    },
    {
      "epoch": 5.652427673339843e-06,
      "model_forward_time": 5.944960832595825,
      "step": 5927
    },
    {
      "epoch": 5.652427673339843e-06,
      "step": 5927,
      "training_step_time": 17.36463975906372
    },
    {
      "epoch": 5.65338134765625e-06,
      "model_forward_time": 5.592365741729736,
      "step": 5928
    },
    {
      "epoch": 5.65338134765625e-06,
      "step": 5928,
      "training_step_time": 17.061445236206055
    },
    {
      "epoch": 5.6543350219726565e-06,
      "model_forward_time": 5.604417085647583,
      "step": 5929
    },
    {
      "epoch": 5.6543350219726565e-06,
      "step": 5929,
      "training_step_time": 17.10611844062805
    },
    {
      "epoch": 5.655288696289062e-06,
      "grad_norm": 0.6857835650444031,
      "learning_rate": 1e-05,
      "loss": 0.0763,
      "step": 5930
    },
    {
      "epoch": 5.655288696289062e-06,
      "model_forward_time": 5.629297733306885,
      "step": 5930
    },
    {
      "dynamics_loss_avg": 0.052578451484441756,
      "epoch": 5.655288696289062e-06,
      "step": 5930
    },
    {
      "action_loss_avg": 0.01495717391371727,
      "epoch": 5.655288696289062e-06,
      "step": 5930
    },
    {
      "epoch": 5.655288696289062e-06,
      "step": 5930,
      "training_step_time": 17.149847507476807
    },
    {
      "epoch": 5.656242370605469e-06,
      "model_forward_time": 5.62433648109436,
      "step": 5931
    },
    {
      "epoch": 5.656242370605469e-06,
      "step": 5931,
      "training_step_time": 17.133326530456543
    },
    {
      "epoch": 5.657196044921875e-06,
      "model_forward_time": 5.606414079666138,
      "step": 5932
    },
    {
      "epoch": 5.657196044921875e-06,
      "step": 5932,
      "training_step_time": 17.086970329284668
    },
    {
      "epoch": 5.658149719238281e-06,
      "model_forward_time": 5.949747562408447,
      "step": 5933
    },
    {
      "epoch": 5.658149719238281e-06,
      "step": 5933,
      "training_step_time": 17.418400287628174
    },
    {
      "epoch": 5.659103393554688e-06,
      "model_forward_time": 5.596273899078369,
      "step": 5934
    },
    {
      "epoch": 5.659103393554688e-06,
      "step": 5934,
      "training_step_time": 17.03241539001465
    },
    {
      "epoch": 5.660057067871093e-06,
      "model_forward_time": 5.587236404418945,
      "step": 5935
    },
    {
      "epoch": 5.660057067871093e-06,
      "step": 5935,
      "training_step_time": 17.036033868789673
    },
    {
      "epoch": 5.6610107421875e-06,
      "model_forward_time": 5.606878757476807,
      "step": 5936
    },
    {
      "epoch": 5.6610107421875e-06,
      "step": 5936,
      "training_step_time": 17.071393966674805
    },
    {
      "epoch": 5.6619644165039065e-06,
      "model_forward_time": 5.602270126342773,
      "step": 5937
    },
    {
      "epoch": 5.6619644165039065e-06,
      "step": 5937,
      "training_step_time": 17.041231632232666
    },
    {
      "epoch": 5.662918090820312e-06,
      "model_forward_time": 5.949303388595581,
      "step": 5938
    },
    {
      "epoch": 5.662918090820312e-06,
      "step": 5938,
      "training_step_time": 17.368163585662842
    },
    {
      "epoch": 5.663871765136719e-06,
      "model_forward_time": 5.581753253936768,
      "step": 5939
    },
    {
      "epoch": 5.663871765136719e-06,
      "step": 5939,
      "training_step_time": 17.02023720741272
    },
    {
      "epoch": 5.664825439453125e-06,
      "grad_norm": 0.7587686777114868,
      "learning_rate": 1e-05,
      "loss": 0.067,
      "step": 5940
    },
    {
      "epoch": 5.664825439453125e-06,
      "model_forward_time": 5.595051050186157,
      "step": 5940
    },
    {
      "dynamics_loss_avg": 0.05729129957035184,
      "epoch": 5.664825439453125e-06,
      "step": 5940
    },
    {
      "action_loss_avg": 0.012589811068028211,
      "epoch": 5.664825439453125e-06,
      "step": 5940
    },
    {
      "epoch": 5.664825439453125e-06,
      "step": 5940,
      "training_step_time": 17.076786041259766
    },
    {
      "epoch": 5.665779113769531e-06,
      "model_forward_time": 5.6065123081207275,
      "step": 5941
    },
    {
      "epoch": 5.665779113769531e-06,
      "step": 5941,
      "training_step_time": 17.082445859909058
    },
    {
      "epoch": 5.666732788085938e-06,
      "model_forward_time": 5.610743522644043,
      "step": 5942
    },
    {
      "epoch": 5.666732788085938e-06,
      "step": 5942,
      "training_step_time": 17.10650372505188
    },
    {
      "epoch": 5.667686462402343e-06,
      "model_forward_time": 5.623070955276489,
      "step": 5943
    },
    {
      "epoch": 5.667686462402343e-06,
      "step": 5943,
      "training_step_time": 17.13243269920349
    },
    {
      "epoch": 5.66864013671875e-06,
      "model_forward_time": 5.969020128250122,
      "step": 5944
    },
    {
      "epoch": 5.66864013671875e-06,
      "step": 5944,
      "training_step_time": 17.449848413467407
    },
    {
      "epoch": 5.6695938110351565e-06,
      "model_forward_time": 5.589672565460205,
      "step": 5945
    },
    {
      "epoch": 5.6695938110351565e-06,
      "step": 5945,
      "training_step_time": 17.046020030975342
    },
    {
      "epoch": 5.670547485351562e-06,
      "model_forward_time": 5.572966575622559,
      "step": 5946
    },
    {
      "epoch": 5.670547485351562e-06,
      "step": 5946,
      "training_step_time": 16.990323543548584
    },
    {
      "epoch": 5.671501159667969e-06,
      "model_forward_time": 5.566987037658691,
      "step": 5947
    },
    {
      "epoch": 5.671501159667969e-06,
      "step": 5947,
      "training_step_time": 16.978571891784668
    },
    {
      "epoch": 5.672454833984375e-06,
      "model_forward_time": 5.573123216629028,
      "step": 5948
    },
    {
      "epoch": 5.672454833984375e-06,
      "step": 5948,
      "training_step_time": 17.004518508911133
    },
    {
      "epoch": 5.673408508300781e-06,
      "model_forward_time": 5.934858322143555,
      "step": 5949
    },
    {
      "epoch": 5.673408508300781e-06,
      "step": 5949,
      "training_step_time": 17.387682676315308
    },
    {
      "epoch": 5.674362182617188e-06,
      "grad_norm": 0.6548844575881958,
      "learning_rate": 1e-05,
      "loss": 0.0503,
      "step": 5950
    },
    {
      "epoch": 5.674362182617188e-06,
      "model_forward_time": 5.583853721618652,
      "step": 5950
    },
    {
      "dynamics_loss_avg": 0.04548756554722786,
      "epoch": 5.674362182617188e-06,
      "step": 5950
    },
    {
      "action_loss_avg": 0.009686076920479536,
      "epoch": 5.674362182617188e-06,
      "step": 5950
    },
    {
      "epoch": 5.674362182617188e-06,
      "step": 5950,
      "training_step_time": 17.013625383377075
    },
    {
      "epoch": 5.675315856933593e-06,
      "model_forward_time": 5.588463544845581,
      "step": 5951
    },
    {
      "epoch": 5.675315856933593e-06,
      "step": 5951,
      "training_step_time": 17.0503933429718
    },
    {
      "epoch": 5.67626953125e-06,
      "model_forward_time": 5.6127660274505615,
      "step": 5952
    },
    {
      "epoch": 5.67626953125e-06,
      "step": 5952,
      "training_step_time": 17.112284421920776
    },
    {
      "epoch": 5.6772232055664065e-06,
      "model_forward_time": 5.618550062179565,
      "step": 5953
    },
    {
      "epoch": 5.6772232055664065e-06,
      "step": 5953,
      "training_step_time": 17.126006603240967
    },
    {
      "epoch": 5.678176879882812e-06,
      "model_forward_time": 5.971431493759155,
      "step": 5954
    },
    {
      "epoch": 5.678176879882812e-06,
      "step": 5954,
      "training_step_time": 17.4675395488739
    },
    {
      "epoch": 5.679130554199219e-06,
      "model_forward_time": 5.600025415420532,
      "step": 5955
    },
    {
      "epoch": 5.679130554199219e-06,
      "step": 5955,
      "training_step_time": 17.09170150756836
    },
    {
      "epoch": 5.680084228515625e-06,
      "model_forward_time": 5.610272407531738,
      "step": 5956
    },
    {
      "epoch": 5.680084228515625e-06,
      "step": 5956,
      "training_step_time": 17.09911799430847
    },
    {
      "epoch": 5.681037902832031e-06,
      "model_forward_time": 5.617877960205078,
      "step": 5957
    },
    {
      "epoch": 5.681037902832031e-06,
      "step": 5957,
      "training_step_time": 17.095635175704956
    },
    {
      "epoch": 5.681991577148438e-06,
      "model_forward_time": 5.603921413421631,
      "step": 5958
    },
    {
      "epoch": 5.681991577148438e-06,
      "step": 5958,
      "training_step_time": 17.097731351852417
    },
    {
      "epoch": 5.682945251464843e-06,
      "model_forward_time": 5.612314224243164,
      "step": 5959
    },
    {
      "epoch": 5.682945251464843e-06,
      "step": 5959,
      "training_step_time": 17.084872722625732
    },
    {
      "epoch": 5.68389892578125e-06,
      "grad_norm": 1.1080150604248047,
      "learning_rate": 1e-05,
      "loss": 0.0843,
      "step": 5960
    },
    {
      "epoch": 5.68389892578125e-06,
      "model_forward_time": 5.971623182296753,
      "step": 5960
    },
    {
      "dynamics_loss_avg": 0.06340217404067516,
      "epoch": 5.68389892578125e-06,
      "step": 5960
    },
    {
      "action_loss_avg": 0.019755216594785453,
      "epoch": 5.68389892578125e-06,
      "step": 5960
    },
    {
      "epoch": 5.68389892578125e-06,
      "step": 5960,
      "training_step_time": 17.432135105133057
    },
    {
      "epoch": 5.6848526000976565e-06,
      "model_forward_time": 5.594156742095947,
      "step": 5961
    },
    {
      "epoch": 5.6848526000976565e-06,
      "step": 5961,
      "training_step_time": 17.016400814056396
    },
    {
      "epoch": 5.685806274414062e-06,
      "model_forward_time": 5.581575870513916,
      "step": 5962
    },
    {
      "epoch": 5.685806274414062e-06,
      "step": 5962,
      "training_step_time": 16.996541023254395
    },
    {
      "epoch": 5.686759948730469e-06,
      "model_forward_time": 5.580117225646973,
      "step": 5963
    },
    {
      "epoch": 5.686759948730469e-06,
      "step": 5963,
      "training_step_time": 17.002955675125122
    },
    {
      "epoch": 5.687713623046875e-06,
      "model_forward_time": 5.568560600280762,
      "step": 5964
    },
    {
      "epoch": 5.687713623046875e-06,
      "step": 5964,
      "training_step_time": 17.003149271011353
    },
    {
      "epoch": 5.688667297363281e-06,
      "model_forward_time": 5.582407474517822,
      "step": 5965
    },
    {
      "epoch": 5.688667297363281e-06,
      "step": 5965,
      "training_step_time": 17.365489721298218
    },
    {
      "epoch": 5.689620971679688e-06,
      "model_forward_time": 5.582089424133301,
      "step": 5966
    },
    {
      "epoch": 5.689620971679688e-06,
      "step": 5966,
      "training_step_time": 17.025638103485107
    },
    {
      "epoch": 5.690574645996093e-06,
      "model_forward_time": 5.593531370162964,
      "step": 5967
    },
    {
      "epoch": 5.690574645996093e-06,
      "step": 5967,
      "training_step_time": 17.062663793563843
    },
    {
      "epoch": 5.6915283203125e-06,
      "model_forward_time": 5.6121156215667725,
      "step": 5968
    },
    {
      "epoch": 5.6915283203125e-06,
      "step": 5968,
      "training_step_time": 17.103541135787964
    },
    {
      "epoch": 5.6924819946289065e-06,
      "model_forward_time": 5.611332416534424,
      "step": 5969
    },
    {
      "epoch": 5.6924819946289065e-06,
      "step": 5969,
      "training_step_time": 17.114471912384033
    },
    {
      "epoch": 5.693435668945312e-06,
      "grad_norm": 0.8478977084159851,
      "learning_rate": 1e-05,
      "loss": 0.0666,
      "step": 5970
    },
    {
      "epoch": 5.693435668945312e-06,
      "model_forward_time": 5.60826563835144,
      "step": 5970
    },
    {
      "dynamics_loss_avg": 0.056275745294988154,
      "epoch": 5.693435668945312e-06,
      "step": 5970
    },
    {
      "action_loss_avg": 0.012151698721572757,
      "epoch": 5.693435668945312e-06,
      "step": 5970
    },
    {
      "epoch": 5.693435668945312e-06,
      "step": 5970,
      "training_step_time": 17.099132299423218
    },
    {
      "epoch": 5.694389343261719e-06,
      "model_forward_time": 5.973357439041138,
      "step": 5971
    },
    {
      "epoch": 5.694389343261719e-06,
      "step": 5971,
      "training_step_time": 17.433457612991333
    },
    {
      "epoch": 5.695343017578125e-06,
      "model_forward_time": 5.582009792327881,
      "step": 5972
    },
    {
      "epoch": 5.695343017578125e-06,
      "step": 5972,
      "training_step_time": 17.016482830047607
    },
    {
      "epoch": 5.696296691894531e-06,
      "model_forward_time": 5.584748983383179,
      "step": 5973
    },
    {
      "epoch": 5.696296691894531e-06,
      "step": 5973,
      "training_step_time": 17.026803731918335
    },
    {
      "epoch": 5.697250366210938e-06,
      "model_forward_time": 5.5883073806762695,
      "step": 5974
    },
    {
      "epoch": 5.697250366210938e-06,
      "step": 5974,
      "training_step_time": 17.05133891105652
    },
    {
      "epoch": 5.698204040527343e-06,
      "model_forward_time": 5.599975109100342,
      "step": 5975
    },
    {
      "epoch": 5.698204040527343e-06,
      "step": 5975,
      "training_step_time": 17.039936542510986
    },
    {
      "epoch": 5.69915771484375e-06,
      "model_forward_time": 5.587137460708618,
      "step": 5976
    },
    {
      "epoch": 5.69915771484375e-06,
      "step": 5976,
      "training_step_time": 17.40195608139038
    },
    {
      "epoch": 5.7001113891601565e-06,
      "model_forward_time": 5.591672658920288,
      "step": 5977
    },
    {
      "epoch": 5.7001113891601565e-06,
      "step": 5977,
      "training_step_time": 17.02781867980957
    },
    {
      "epoch": 5.701065063476562e-06,
      "model_forward_time": 5.587862730026245,
      "step": 5978
    },
    {
      "epoch": 5.701065063476562e-06,
      "step": 5978,
      "training_step_time": 17.039464712142944
    },
    {
      "epoch": 5.702018737792969e-06,
      "model_forward_time": 5.609201908111572,
      "step": 5979
    },
    {
      "epoch": 5.702018737792969e-06,
      "step": 5979,
      "training_step_time": 17.08925724029541
    },
    {
      "epoch": 5.702972412109375e-06,
      "grad_norm": 0.4738611578941345,
      "learning_rate": 1e-05,
      "loss": 0.0631,
      "step": 5980
    },
    {
      "epoch": 5.702972412109375e-06,
      "model_forward_time": 5.6151649951934814,
      "step": 5980
    },
    {
      "dynamics_loss_avg": 0.05051051434129476,
      "epoch": 5.702972412109375e-06,
      "step": 5980
    },
    {
      "action_loss_avg": 0.00931603650096804,
      "epoch": 5.702972412109375e-06,
      "step": 5980
    },
    {
      "epoch": 5.702972412109375e-06,
      "step": 5980,
      "training_step_time": 17.109982013702393
    },
    {
      "epoch": 5.703926086425781e-06,
      "model_forward_time": 5.621280193328857,
      "step": 5981
    },
    {
      "epoch": 5.703926086425781e-06,
      "step": 5981,
      "training_step_time": 17.128575325012207
    },
    {
      "epoch": 5.704879760742188e-06,
      "model_forward_time": 5.9892332553863525,
      "step": 5982
    },
    {
      "epoch": 5.704879760742188e-06,
      "step": 5982,
      "training_step_time": 17.49424386024475
    },
    {
      "epoch": 5.705833435058593e-06,
      "model_forward_time": 5.615568161010742,
      "step": 5983
    },
    {
      "epoch": 5.705833435058593e-06,
      "step": 5983,
      "training_step_time": 17.096397399902344
    },
    {
      "epoch": 5.706787109375e-06,
      "model_forward_time": 5.6007750034332275,
      "step": 5984
    },
    {
      "epoch": 5.706787109375e-06,
      "step": 5984,
      "training_step_time": 17.04892587661743
    },
    {
      "epoch": 5.7077407836914066e-06,
      "model_forward_time": 5.591612339019775,
      "step": 5985
    },
    {
      "epoch": 5.7077407836914066e-06,
      "step": 5985,
      "training_step_time": 17.05317997932434
    },
    {
      "epoch": 5.708694458007812e-06,
      "model_forward_time": 5.598932266235352,
      "step": 5986
    },
    {
      "epoch": 5.708694458007812e-06,
      "step": 5986,
      "training_step_time": 17.05723547935486
    },
    {
      "epoch": 5.709648132324219e-06,
      "model_forward_time": 5.95453143119812,
      "step": 5987
    },
    {
      "epoch": 5.709648132324219e-06,
      "step": 5987,
      "training_step_time": 17.410197734832764
    },
    {
      "epoch": 5.710601806640625e-06,
      "model_forward_time": 5.598376989364624,
      "step": 5988
    },
    {
      "epoch": 5.710601806640625e-06,
      "step": 5988,
      "training_step_time": 17.077253341674805
    },
    {
      "epoch": 5.711555480957031e-06,
      "model_forward_time": 5.60940408706665,
      "step": 5989
    },
    {
      "epoch": 5.711555480957031e-06,
      "step": 5989,
      "training_step_time": 17.107956647872925
    },
    {
      "epoch": 5.712509155273438e-06,
      "grad_norm": 1.4134986400604248,
      "learning_rate": 1e-05,
      "loss": 0.0686,
      "step": 5990
    },
    {
      "epoch": 5.712509155273438e-06,
      "model_forward_time": 5.594126224517822,
      "step": 5990
    },
    {
      "dynamics_loss_avg": 0.05240337997674942,
      "epoch": 5.712509155273438e-06,
      "step": 5990
    },
    {
      "action_loss_avg": 0.011375274672172964,
      "epoch": 5.712509155273438e-06,
      "step": 5990
    },
    {
      "epoch": 5.712509155273438e-06,
      "step": 5990,
      "training_step_time": 17.0496346950531
    },
    {
      "epoch": 5.7134628295898434e-06,
      "model_forward_time": 5.587153673171997,
      "step": 5991
    },
    {
      "epoch": 5.7134628295898434e-06,
      "step": 5991,
      "training_step_time": 17.01467251777649
    },
    {
      "epoch": 5.71441650390625e-06,
      "model_forward_time": 5.573962211608887,
      "step": 5992
    },
    {
      "epoch": 5.71441650390625e-06,
      "step": 5992,
      "training_step_time": 17.00066304206848
    },
    {
      "epoch": 5.7153701782226566e-06,
      "model_forward_time": 5.9604480266571045,
      "step": 5993
    },
    {
      "epoch": 5.7153701782226566e-06,
      "step": 5993,
      "training_step_time": 17.438144207000732
    },
    {
      "epoch": 5.716323852539062e-06,
      "model_forward_time": 5.609001636505127,
      "step": 5994
    },
    {
      "epoch": 5.716323852539062e-06,
      "step": 5994,
      "training_step_time": 17.08584690093994
    },
    {
      "epoch": 5.717277526855469e-06,
      "model_forward_time": 5.605413436889648,
      "step": 5995
    },
    {
      "epoch": 5.717277526855469e-06,
      "step": 5995,
      "training_step_time": 17.08588218688965
    },
    {
      "epoch": 5.718231201171875e-06,
      "model_forward_time": 5.597950458526611,
      "step": 5996
    },
    {
      "epoch": 5.718231201171875e-06,
      "step": 5996,
      "training_step_time": 17.05428123474121
    },
    {
      "epoch": 5.719184875488281e-06,
      "model_forward_time": 5.602314233779907,
      "step": 5997
    },
    {
      "epoch": 5.719184875488281e-06,
      "step": 5997,
      "training_step_time": 17.055599689483643
    },
    {
      "epoch": 5.720138549804688e-06,
      "model_forward_time": 5.598336935043335,
      "step": 5998
    },
    {
      "epoch": 5.720138549804688e-06,
      "step": 5998,
      "training_step_time": 17.427858114242554
    },
    {
      "epoch": 5.7210922241210934e-06,
      "model_forward_time": 5.585712671279907,
      "step": 5999
    },
    {
      "epoch": 5.7210922241210934e-06,
      "step": 5999,
      "training_step_time": 17.00876474380493
    },
    {
      "epoch": 5.7220458984375e-06,
      "grad_norm": 1.601792573928833,
      "learning_rate": 1e-05,
      "loss": 0.0724,
      "step": 6000
    },
    {
      "epoch": 5.7220458984375e-06,
      "model_forward_time": 5.655827522277832,
      "step": 6000
    },
    {
      "dynamics_loss_avg": 0.05987470857799053,
      "epoch": 5.7220458984375e-06,
      "step": 6000
    },
    {
      "action_loss_avg": 0.01793991243466735,
      "epoch": 5.7220458984375e-06,
      "step": 6000
    },
    {
      "epoch": 5.7220458984375e-06,
      "step": 6000,
      "training_step_time": 17.147786378860474
    },
    {
      "epoch": 5.7229995727539066e-06,
      "model_forward_time": 5.678247928619385,
      "step": 6001
    },
    {
      "epoch": 5.7229995727539066e-06,
      "step": 6001,
      "training_step_time": 17.184275150299072
    },
    {
      "epoch": 5.723953247070312e-06,
      "model_forward_time": 5.636698961257935,
      "step": 6002
    },
    {
      "epoch": 5.723953247070312e-06,
      "step": 6002,
      "training_step_time": 17.15208101272583
    },
    {
      "epoch": 5.724906921386719e-06,
      "model_forward_time": 5.700635194778442,
      "step": 6003
    },
    {
      "epoch": 5.724906921386719e-06,
      "step": 6003,
      "training_step_time": 17.195512056350708
    },
    {
      "epoch": 5.725860595703125e-06,
      "model_forward_time": 5.965747594833374,
      "step": 6004
    },
    {
      "epoch": 5.725860595703125e-06,
      "step": 6004,
      "training_step_time": 17.469438076019287
    },
    {
      "epoch": 5.726814270019531e-06,
      "model_forward_time": 5.597670078277588,
      "step": 6005
    },
    {
      "epoch": 5.726814270019531e-06,
      "step": 6005,
      "training_step_time": 17.05338406562805
    },
    {
      "epoch": 5.727767944335938e-06,
      "model_forward_time": 5.572144985198975,
      "step": 6006
    },
    {
      "epoch": 5.727767944335938e-06,
      "step": 6006,
      "training_step_time": 17.000438690185547
    },
    {
      "epoch": 5.7287216186523434e-06,
      "model_forward_time": 5.5832366943359375,
      "step": 6007
    },
    {
      "epoch": 5.7287216186523434e-06,
      "step": 6007,
      "training_step_time": 16.999743938446045
    },
    {
      "epoch": 5.72967529296875e-06,
      "model_forward_time": 5.577337026596069,
      "step": 6008
    },
    {
      "epoch": 5.72967529296875e-06,
      "step": 6008,
      "training_step_time": 17.018388986587524
    },
    {
      "epoch": 5.7306289672851566e-06,
      "model_forward_time": 5.972325563430786,
      "step": 6009
    },
    {
      "epoch": 5.7306289672851566e-06,
      "step": 6009,
      "training_step_time": 17.43029499053955
    },
    {
      "epoch": 5.731582641601562e-06,
      "grad_norm": 1.7602806091308594,
      "learning_rate": 1e-05,
      "loss": 0.1001,
      "step": 6010
    },
    {
      "epoch": 5.731582641601562e-06,
      "model_forward_time": 5.613152742385864,
      "step": 6010
    },
    {
      "dynamics_loss_avg": 0.07322389241307974,
      "epoch": 5.731582641601562e-06,
      "step": 6010
    },
    {
      "action_loss_avg": 0.025681708846241234,
      "epoch": 5.731582641601562e-06,
      "step": 6010
    },
    {
      "epoch": 5.731582641601562e-06,
      "step": 6010,
      "training_step_time": 17.093642950057983
    },
    {
      "epoch": 5.732536315917969e-06,
      "model_forward_time": 5.600713729858398,
      "step": 6011
    },
    {
      "epoch": 5.732536315917969e-06,
      "step": 6011,
      "training_step_time": 17.080567121505737
    },
    {
      "epoch": 5.733489990234375e-06,
      "model_forward_time": 5.604346036911011,
      "step": 6012
    },
    {
      "epoch": 5.733489990234375e-06,
      "step": 6012,
      "training_step_time": 17.082075834274292
    },
    {
      "epoch": 5.734443664550781e-06,
      "model_forward_time": 5.589061737060547,
      "step": 6013
    },
    {
      "epoch": 5.734443664550781e-06,
      "step": 6013,
      "training_step_time": 17.03614640235901
    },
    {
      "epoch": 5.735397338867188e-06,
      "model_forward_time": 5.572148561477661,
      "step": 6014
    },
    {
      "epoch": 5.735397338867188e-06,
      "step": 6014,
      "training_step_time": 16.979369163513184
    },
    {
      "epoch": 5.7363510131835934e-06,
      "model_forward_time": 5.93546724319458,
      "step": 6015
    },
    {
      "epoch": 5.7363510131835934e-06,
      "step": 6015,
      "training_step_time": 17.338355779647827
    },
    {
      "epoch": 5.7373046875e-06,
      "model_forward_time": 5.578840732574463,
      "step": 6016
    },
    {
      "epoch": 5.7373046875e-06,
      "step": 6016,
      "training_step_time": 17.004127740859985
    },
    {
      "epoch": 5.738258361816407e-06,
      "model_forward_time": 5.583213806152344,
      "step": 6017
    },
    {
      "epoch": 5.738258361816407e-06,
      "step": 6017,
      "training_step_time": 17.023693799972534
    },
    {
      "epoch": 5.739212036132812e-06,
      "model_forward_time": 5.595031499862671,
      "step": 6018
    },
    {
      "epoch": 5.739212036132812e-06,
      "step": 6018,
      "training_step_time": 17.052708864212036
    },
    {
      "epoch": 5.740165710449219e-06,
      "model_forward_time": 5.602891445159912,
      "step": 6019
    },
    {
      "epoch": 5.740165710449219e-06,
      "step": 6019,
      "training_step_time": 17.090083360671997
    },
    {
      "epoch": 5.741119384765625e-06,
      "grad_norm": 1.6042429208755493,
      "learning_rate": 1e-05,
      "loss": 0.0775,
      "step": 6020
    },
    {
      "epoch": 5.741119384765625e-06,
      "model_forward_time": 5.9423394203186035,
      "step": 6020
    },
    {
      "dynamics_loss_avg": 0.06151041109114885,
      "epoch": 5.741119384765625e-06,
      "step": 6020
    },
    {
      "action_loss_avg": 0.016919126070570202,
      "epoch": 5.741119384765625e-06,
      "step": 6020
    },
    {
      "epoch": 5.741119384765625e-06,
      "step": 6020,
      "training_step_time": 17.45072627067566
    },
    {
      "epoch": 5.742073059082031e-06,
      "model_forward_time": 5.625148773193359,
      "step": 6021
    },
    {
      "epoch": 5.742073059082031e-06,
      "step": 6021,
      "training_step_time": 17.123389959335327
    },
    {
      "epoch": 5.743026733398438e-06,
      "model_forward_time": 5.612282037734985,
      "step": 6022
    },
    {
      "epoch": 5.743026733398438e-06,
      "step": 6022,
      "training_step_time": 17.07525110244751
    },
    {
      "epoch": 5.7439804077148435e-06,
      "model_forward_time": 5.594120502471924,
      "step": 6023
    },
    {
      "epoch": 5.7439804077148435e-06,
      "step": 6023,
      "training_step_time": 16.994398832321167
    },
    {
      "epoch": 5.74493408203125e-06,
      "model_forward_time": 5.574771881103516,
      "step": 6024
    },
    {
      "epoch": 5.74493408203125e-06,
      "step": 6024,
      "training_step_time": 17.008333444595337
    },
    {
      "epoch": 5.745887756347657e-06,
      "model_forward_time": 5.951388120651245,
      "step": 6025
    },
    {
      "epoch": 5.745887756347657e-06,
      "step": 6025,
      "training_step_time": 17.38471269607544
    },
    {
      "epoch": 5.746841430664062e-06,
      "model_forward_time": 5.5785017013549805,
      "step": 6026
    },
    {
      "epoch": 5.746841430664062e-06,
      "step": 6026,
      "training_step_time": 16.991448640823364
    },
    {
      "epoch": 5.747795104980469e-06,
      "model_forward_time": 5.5724756717681885,
      "step": 6027
    },
    {
      "epoch": 5.747795104980469e-06,
      "step": 6027,
      "training_step_time": 16.988632440567017
    },
    {
      "epoch": 5.748748779296875e-06,
      "model_forward_time": 5.591859579086304,
      "step": 6028
    },
    {
      "epoch": 5.748748779296875e-06,
      "step": 6028,
      "training_step_time": 17.03036069869995
    },
    {
      "epoch": 5.749702453613281e-06,
      "model_forward_time": 5.59977388381958,
      "step": 6029
    },
    {
      "epoch": 5.749702453613281e-06,
      "step": 6029,
      "training_step_time": 17.074589490890503
    },
    {
      "epoch": 5.750656127929688e-06,
      "grad_norm": 1.3713533878326416,
      "learning_rate": 1e-05,
      "loss": 0.0719,
      "step": 6030
    },
    {
      "epoch": 5.750656127929688e-06,
      "model_forward_time": 5.6069793701171875,
      "step": 6030
    },
    {
      "dynamics_loss_avg": 0.05374857736751437,
      "epoch": 5.750656127929688e-06,
      "step": 6030
    },
    {
      "action_loss_avg": 0.0149175152531825,
      "epoch": 5.750656127929688e-06,
      "step": 6030
    },
    {
      "epoch": 5.750656127929688e-06,
      "step": 6030,
      "training_step_time": 17.102676153182983
    },
    {
      "epoch": 5.7516098022460935e-06,
      "model_forward_time": 5.965238332748413,
      "step": 6031
    },
    {
      "epoch": 5.7516098022460935e-06,
      "step": 6031,
      "training_step_time": 17.46269702911377
    },
    {
      "epoch": 5.7525634765625e-06,
      "model_forward_time": 5.6114232540130615,
      "step": 6032
    },
    {
      "epoch": 5.7525634765625e-06,
      "step": 6032,
      "training_step_time": 17.12155771255493
    },
    {
      "epoch": 5.753517150878907e-06,
      "model_forward_time": 5.619975805282593,
      "step": 6033
    },
    {
      "epoch": 5.753517150878907e-06,
      "step": 6033,
      "training_step_time": 17.113077878952026
    },
    {
      "epoch": 5.754470825195312e-06,
      "model_forward_time": 5.61519455909729,
      "step": 6034
    },
    {
      "epoch": 5.754470825195312e-06,
      "step": 6034,
      "training_step_time": 17.08981966972351
    },
    {
      "epoch": 5.755424499511719e-06,
      "model_forward_time": 5.592390537261963,
      "step": 6035
    },
    {
      "epoch": 5.755424499511719e-06,
      "step": 6035,
      "training_step_time": 17.019925355911255
    },
    {
      "epoch": 5.756378173828125e-06,
      "model_forward_time": 5.948685169219971,
      "step": 6036
    },
    {
      "epoch": 5.756378173828125e-06,
      "step": 6036,
      "training_step_time": 17.389376163482666
    },
    {
      "epoch": 5.757331848144531e-06,
      "model_forward_time": 5.596347808837891,
      "step": 6037
    },
    {
      "epoch": 5.757331848144531e-06,
      "step": 6037,
      "training_step_time": 17.05340027809143
    },
    {
      "epoch": 5.758285522460938e-06,
      "model_forward_time": 5.603541612625122,
      "step": 6038
    },
    {
      "epoch": 5.758285522460938e-06,
      "step": 6038,
      "training_step_time": 17.07536768913269
    },
    {
      "epoch": 5.7592391967773435e-06,
      "model_forward_time": 5.6155195236206055,
      "step": 6039
    },
    {
      "epoch": 5.7592391967773435e-06,
      "step": 6039,
      "training_step_time": 17.10821843147278
    },
    {
      "epoch": 5.76019287109375e-06,
      "grad_norm": 1.220276951789856,
      "learning_rate": 1e-05,
      "loss": 0.0846,
      "step": 6040
    },
    {
      "epoch": 5.76019287109375e-06,
      "model_forward_time": 5.61298680305481,
      "step": 6040
    },
    {
      "dynamics_loss_avg": 0.06863051503896714,
      "epoch": 5.76019287109375e-06,
      "step": 6040
    },
    {
      "action_loss_avg": 0.017337338207289575,
      "epoch": 5.76019287109375e-06,
      "step": 6040
    },
    {
      "epoch": 5.76019287109375e-06,
      "step": 6040,
      "training_step_time": 17.11743140220642
    },
    {
      "epoch": 5.761146545410157e-06,
      "model_forward_time": 5.625471115112305,
      "step": 6041
    },
    {
      "epoch": 5.761146545410157e-06,
      "step": 6041,
      "training_step_time": 17.459728240966797
    },
    {
      "epoch": 5.762100219726562e-06,
      "model_forward_time": 5.59702467918396,
      "step": 6042
    },
    {
      "epoch": 5.762100219726562e-06,
      "step": 6042,
      "training_step_time": 17.021658658981323
    },
    {
      "epoch": 5.763053894042969e-06,
      "model_forward_time": 5.593374729156494,
      "step": 6043
    },
    {
      "epoch": 5.763053894042969e-06,
      "step": 6043,
      "training_step_time": 17.025643825531006
    },
    {
      "epoch": 5.764007568359375e-06,
      "model_forward_time": 5.582154035568237,
      "step": 6044
    },
    {
      "epoch": 5.764007568359375e-06,
      "step": 6044,
      "training_step_time": 17.02750849723816
    },
    {
      "epoch": 5.764961242675781e-06,
      "model_forward_time": 5.597147226333618,
      "step": 6045
    },
    {
      "epoch": 5.764961242675781e-06,
      "step": 6045,
      "training_step_time": 17.07218885421753
    },
    {
      "epoch": 5.765914916992188e-06,
      "model_forward_time": 5.606848478317261,
      "step": 6046
    },
    {
      "epoch": 5.765914916992188e-06,
      "step": 6046,
      "training_step_time": 17.119077682495117
    },
    {
      "epoch": 5.7668685913085935e-06,
      "model_forward_time": 6.000171661376953,
      "step": 6047
    },
    {
      "epoch": 5.7668685913085935e-06,
      "step": 6047,
      "training_step_time": 17.502633333206177
    },
    {
      "epoch": 5.767822265625e-06,
      "model_forward_time": 5.629256725311279,
      "step": 6048
    },
    {
      "epoch": 5.767822265625e-06,
      "step": 6048,
      "training_step_time": 17.130289554595947
    },
    {
      "epoch": 5.768775939941407e-06,
      "model_forward_time": 5.636305570602417,
      "step": 6049
    },
    {
      "epoch": 5.768775939941407e-06,
      "step": 6049,
      "training_step_time": 17.122787952423096
    },
    {
      "epoch": 5.769729614257812e-06,
      "grad_norm": 1.2228331565856934,
      "learning_rate": 1e-05,
      "loss": 0.0742,
      "step": 6050
    },
    {
      "epoch": 5.769729614257812e-06,
      "model_forward_time": 5.618684768676758,
      "step": 6050
    },
    {
      "dynamics_loss_avg": 0.06558343730866908,
      "epoch": 5.769729614257812e-06,
      "step": 6050
    },
    {
      "action_loss_avg": 0.013617770955897867,
      "epoch": 5.769729614257812e-06,
      "step": 6050
    },
    {
      "epoch": 5.769729614257812e-06,
      "step": 6050,
      "training_step_time": 17.098435401916504
    },
    {
      "epoch": 5.770683288574219e-06,
      "model_forward_time": 5.607197046279907,
      "step": 6051
    },
    {
      "epoch": 5.770683288574219e-06,
      "step": 6051,
      "training_step_time": 17.082107543945312
    },
    {
      "epoch": 5.771636962890625e-06,
      "model_forward_time": 5.9799253940582275,
      "step": 6052
    },
    {
      "epoch": 5.771636962890625e-06,
      "step": 6052,
      "training_step_time": 17.46924614906311
    },
    {
      "epoch": 5.772590637207031e-06,
      "model_forward_time": 5.622534990310669,
      "step": 6053
    },
    {
      "epoch": 5.772590637207031e-06,
      "step": 6053,
      "training_step_time": 17.13547396659851
    },
    {
      "epoch": 5.773544311523438e-06,
      "model_forward_time": 5.618840932846069,
      "step": 6054
    },
    {
      "epoch": 5.773544311523438e-06,
      "step": 6054,
      "training_step_time": 17.12567639350891
    },
    {
      "epoch": 5.7744979858398435e-06,
      "model_forward_time": 5.611301898956299,
      "step": 6055
    },
    {
      "epoch": 5.7744979858398435e-06,
      "step": 6055,
      "training_step_time": 17.11377763748169
    },
    {
      "epoch": 5.77545166015625e-06,
      "model_forward_time": 5.609745264053345,
      "step": 6056
    },
    {
      "epoch": 5.77545166015625e-06,
      "step": 6056,
      "training_step_time": 17.088280200958252
    },
    {
      "epoch": 5.776405334472657e-06,
      "model_forward_time": 5.6002113819122314,
      "step": 6057
    },
    {
      "epoch": 5.776405334472657e-06,
      "step": 6057,
      "training_step_time": 17.032624006271362
    },
    {
      "epoch": 5.777359008789062e-06,
      "model_forward_time": 5.941502094268799,
      "step": 6058
    },
    {
      "epoch": 5.777359008789062e-06,
      "step": 6058,
      "training_step_time": 17.367106437683105
    },
    {
      "epoch": 5.778312683105469e-06,
      "model_forward_time": 5.567535877227783,
      "step": 6059
    },
    {
      "epoch": 5.778312683105469e-06,
      "step": 6059,
      "training_step_time": 17.013973236083984
    },
    {
      "epoch": 5.779266357421875e-06,
      "grad_norm": 2.238536834716797,
      "learning_rate": 1e-05,
      "loss": 0.0944,
      "step": 6060
    },
    {
      "epoch": 5.779266357421875e-06,
      "model_forward_time": 5.567192554473877,
      "step": 6060
    },
    {
      "dynamics_loss_avg": 0.07144837863743306,
      "epoch": 5.779266357421875e-06,
      "step": 6060
    },
    {
      "action_loss_avg": 0.018721370305866004,
      "epoch": 5.779266357421875e-06,
      "step": 6060
    },
    {
      "epoch": 5.779266357421875e-06,
      "step": 6060,
      "training_step_time": 16.972453594207764
    },
    {
      "epoch": 5.780220031738281e-06,
      "model_forward_time": 5.572617053985596,
      "step": 6061
    },
    {
      "epoch": 5.780220031738281e-06,
      "step": 6061,
      "training_step_time": 16.98891568183899
    },
    {
      "epoch": 5.781173706054688e-06,
      "model_forward_time": 5.577765703201294,
      "step": 6062
    },
    {
      "epoch": 5.781173706054688e-06,
      "step": 6062,
      "training_step_time": 17.01281237602234
    },
    {
      "epoch": 5.7821273803710935e-06,
      "model_forward_time": 5.9335808753967285,
      "step": 6063
    },
    {
      "epoch": 5.7821273803710935e-06,
      "step": 6063,
      "training_step_time": 17.36199450492859
    },
    {
      "epoch": 5.7830810546875e-06,
      "model_forward_time": 5.586297273635864,
      "step": 6064
    },
    {
      "epoch": 5.7830810546875e-06,
      "step": 6064,
      "training_step_time": 17.03218698501587
    },
    {
      "epoch": 5.784034729003907e-06,
      "model_forward_time": 5.601010322570801,
      "step": 6065
    },
    {
      "epoch": 5.784034729003907e-06,
      "step": 6065,
      "training_step_time": 17.073967218399048
    },
    {
      "epoch": 5.784988403320312e-06,
      "model_forward_time": 5.613511800765991,
      "step": 6066
    },
    {
      "epoch": 5.784988403320312e-06,
      "step": 6066,
      "training_step_time": 17.11224365234375
    },
    {
      "epoch": 5.785942077636719e-06,
      "model_forward_time": 5.611045837402344,
      "step": 6067
    },
    {
      "epoch": 5.785942077636719e-06,
      "step": 6067,
      "training_step_time": 17.068560361862183
    },
    {
      "epoch": 5.786895751953125e-06,
      "model_forward_time": 5.594607591629028,
      "step": 6068
    },
    {
      "epoch": 5.786895751953125e-06,
      "step": 6068,
      "training_step_time": 17.047536373138428
    },
    {
      "epoch": 5.787849426269531e-06,
      "model_forward_time": 5.95668888092041,
      "step": 6069
    },
    {
      "epoch": 5.787849426269531e-06,
      "step": 6069,
      "training_step_time": 17.42819857597351
    },
    {
      "epoch": 5.788803100585938e-06,
      "grad_norm": 0.15092246234416962,
      "learning_rate": 1e-05,
      "loss": 0.0587,
      "step": 6070
    },
    {
      "epoch": 5.788803100585938e-06,
      "model_forward_time": 5.601944923400879,
      "step": 6070
    },
    {
      "dynamics_loss_avg": 0.04944536332041025,
      "epoch": 5.788803100585938e-06,
      "step": 6070
    },
    {
      "action_loss_avg": 0.009561611514072865,
      "epoch": 5.788803100585938e-06,
      "step": 6070
    },
    {
      "epoch": 5.788803100585938e-06,
      "step": 6070,
      "training_step_time": 17.07914400100708
    },
    {
      "epoch": 5.7897567749023435e-06,
      "model_forward_time": 5.601638317108154,
      "step": 6071
    },
    {
      "epoch": 5.7897567749023435e-06,
      "step": 6071,
      "training_step_time": 17.07939910888672
    },
    {
      "epoch": 5.79071044921875e-06,
      "model_forward_time": 5.605647563934326,
      "step": 6072
    },
    {
      "epoch": 5.79071044921875e-06,
      "step": 6072,
      "training_step_time": 17.084972143173218
    },
    {
      "epoch": 5.791664123535157e-06,
      "model_forward_time": 5.6131439208984375,
      "step": 6073
    },
    {
      "epoch": 5.791664123535157e-06,
      "step": 6073,
      "training_step_time": 17.102885961532593
    },
    {
      "epoch": 5.792617797851562e-06,
      "model_forward_time": 5.9520204067230225,
      "step": 6074
    },
    {
      "epoch": 5.792617797851562e-06,
      "step": 6074,
      "training_step_time": 17.436452865600586
    },
    {
      "epoch": 5.793571472167969e-06,
      "model_forward_time": 5.607985019683838,
      "step": 6075
    },
    {
      "epoch": 5.793571472167969e-06,
      "step": 6075,
      "training_step_time": 17.088586568832397
    },
    {
      "epoch": 5.794525146484375e-06,
      "model_forward_time": 5.608338117599487,
      "step": 6076
    },
    {
      "epoch": 5.794525146484375e-06,
      "step": 6076,
      "training_step_time": 17.099809885025024
    },
    {
      "epoch": 5.795478820800781e-06,
      "model_forward_time": 5.603557825088501,
      "step": 6077
    },
    {
      "epoch": 5.795478820800781e-06,
      "step": 6077,
      "training_step_time": 17.073558807373047
    },
    {
      "epoch": 5.796432495117188e-06,
      "model_forward_time": 5.60538387298584,
      "step": 6078
    },
    {
      "epoch": 5.796432495117188e-06,
      "step": 6078,
      "training_step_time": 17.04028010368347
    },
    {
      "epoch": 5.7973861694335935e-06,
      "model_forward_time": 5.615368366241455,
      "step": 6079
    },
    {
      "epoch": 5.7973861694335935e-06,
      "step": 6079,
      "training_step_time": 17.398712396621704
    },
    {
      "epoch": 5.79833984375e-06,
      "grad_norm": 0.26922157406806946,
      "learning_rate": 1e-05,
      "loss": 0.0761,
      "step": 6080
    },
    {
      "epoch": 5.79833984375e-06,
      "model_forward_time": 5.580717325210571,
      "step": 6080
    },
    {
      "dynamics_loss_avg": 0.05945960059762001,
      "epoch": 5.79833984375e-06,
      "step": 6080
    },
    {
      "action_loss_avg": 0.012348165176808834,
      "epoch": 5.79833984375e-06,
      "step": 6080
    },
    {
      "epoch": 5.79833984375e-06,
      "step": 6080,
      "training_step_time": 17.00240921974182
    },
    {
      "epoch": 5.799293518066407e-06,
      "model_forward_time": 5.589656829833984,
      "step": 6081
    },
    {
      "epoch": 5.799293518066407e-06,
      "step": 6081,
      "training_step_time": 17.045274019241333
    },
    {
      "epoch": 5.800247192382812e-06,
      "model_forward_time": 5.598461151123047,
      "step": 6082
    },
    {
      "epoch": 5.800247192382812e-06,
      "step": 6082,
      "training_step_time": 17.07564687728882
    },
    {
      "epoch": 5.801200866699219e-06,
      "model_forward_time": 5.6223390102386475,
      "step": 6083
    },
    {
      "epoch": 5.801200866699219e-06,
      "step": 6083,
      "training_step_time": 17.12393832206726
    },
    {
      "epoch": 5.802154541015625e-06,
      "model_forward_time": 5.626060485839844,
      "step": 6084
    },
    {
      "epoch": 5.802154541015625e-06,
      "step": 6084,
      "training_step_time": 17.125949382781982
    },
    {
      "epoch": 5.803108215332031e-06,
      "model_forward_time": 5.975173473358154,
      "step": 6085
    },
    {
      "epoch": 5.803108215332031e-06,
      "step": 6085,
      "training_step_time": 17.439486742019653
    },
    {
      "epoch": 5.804061889648438e-06,
      "model_forward_time": 5.600358724594116,
      "step": 6086
    },
    {
      "epoch": 5.804061889648438e-06,
      "step": 6086,
      "training_step_time": 17.052917003631592
    },
    {
      "epoch": 5.8050155639648435e-06,
      "model_forward_time": 5.597971677780151,
      "step": 6087
    },
    {
      "epoch": 5.8050155639648435e-06,
      "step": 6087,
      "training_step_time": 17.06020712852478
    },
    {
      "epoch": 5.80596923828125e-06,
      "model_forward_time": 5.624351263046265,
      "step": 6088
    },
    {
      "epoch": 5.80596923828125e-06,
      "step": 6088,
      "training_step_time": 17.076776266098022
    },
    {
      "epoch": 5.806922912597657e-06,
      "model_forward_time": 5.593355417251587,
      "step": 6089
    },
    {
      "epoch": 5.806922912597657e-06,
      "step": 6089,
      "training_step_time": 17.02180576324463
    },
    {
      "epoch": 5.807876586914062e-06,
      "grad_norm": 0.27332592010498047,
      "learning_rate": 1e-05,
      "loss": 0.0707,
      "step": 6090
    },
    {
      "epoch": 5.807876586914062e-06,
      "model_forward_time": 5.973797082901001,
      "step": 6090
    },
    {
      "dynamics_loss_avg": 0.06088754665106535,
      "epoch": 5.807876586914062e-06,
      "step": 6090
    },
    {
      "action_loss_avg": 0.01247329656034708,
      "epoch": 5.807876586914062e-06,
      "step": 6090
    },
    {
      "epoch": 5.807876586914062e-06,
      "step": 6090,
      "training_step_time": 17.44336700439453
    },
    {
      "epoch": 5.808830261230469e-06,
      "model_forward_time": 5.6205360889434814,
      "step": 6091
    },
    {
      "epoch": 5.808830261230469e-06,
      "step": 6091,
      "training_step_time": 17.118895053863525
    },
    {
      "epoch": 5.809783935546875e-06,
      "model_forward_time": 5.622753858566284,
      "step": 6092
    },
    {
      "epoch": 5.809783935546875e-06,
      "step": 6092,
      "training_step_time": 17.12439489364624
    },
    {
      "epoch": 5.810737609863281e-06,
      "model_forward_time": 5.618081092834473,
      "step": 6093
    },
    {
      "epoch": 5.810737609863281e-06,
      "step": 6093,
      "training_step_time": 17.110710382461548
    },
    {
      "epoch": 5.811691284179688e-06,
      "model_forward_time": 5.622478485107422,
      "step": 6094
    },
    {
      "epoch": 5.811691284179688e-06,
      "step": 6094,
      "training_step_time": 17.120217084884644
    },
    {
      "epoch": 5.8126449584960935e-06,
      "model_forward_time": 5.622268438339233,
      "step": 6095
    },
    {
      "epoch": 5.8126449584960935e-06,
      "step": 6095,
      "training_step_time": 17.113070726394653
    },
    {
      "epoch": 5.8135986328125e-06,
      "model_forward_time": 5.958612680435181,
      "step": 6096
    },
    {
      "epoch": 5.8135986328125e-06,
      "step": 6096,
      "training_step_time": 17.445448637008667
    },
    {
      "epoch": 5.814552307128907e-06,
      "model_forward_time": 5.6224281787872314,
      "step": 6097
    },
    {
      "epoch": 5.814552307128907e-06,
      "step": 6097,
      "training_step_time": 17.11601448059082
    },
    {
      "epoch": 5.815505981445312e-06,
      "model_forward_time": 5.609024286270142,
      "step": 6098
    },
    {
      "epoch": 5.815505981445312e-06,
      "step": 6098,
      "training_step_time": 17.09728479385376
    },
    {
      "epoch": 5.816459655761719e-06,
      "model_forward_time": 5.614210844039917,
      "step": 6099
    },
    {
      "epoch": 5.816459655761719e-06,
      "step": 6099,
      "training_step_time": 17.095380067825317
    },
    {
      "epoch": 5.817413330078125e-06,
      "grad_norm": 0.7510960698127747,
      "learning_rate": 1e-05,
      "loss": 0.0925,
      "step": 6100
    },
    {
      "epoch": 5.817413330078125e-06,
      "model_forward_time": 5.619991064071655,
      "step": 6100
    },
    {
      "dynamics_loss_avg": 0.07620186507701873,
      "epoch": 5.817413330078125e-06,
      "step": 6100
    },
    {
      "action_loss_avg": 0.01900363522581756,
      "epoch": 5.817413330078125e-06,
      "step": 6100
    },
    {
      "epoch": 5.817413330078125e-06,
      "step": 6100,
      "training_step_time": 17.09891939163208
    },
    {
      "epoch": 5.818367004394531e-06,
      "model_forward_time": 5.974318027496338,
      "step": 6101
    },
    {
      "epoch": 5.818367004394531e-06,
      "step": 6101,
      "training_step_time": 17.4423246383667
    },
    {
      "epoch": 5.819320678710938e-06,
      "model_forward_time": 5.603651523590088,
      "step": 6102
    },
    {
      "epoch": 5.819320678710938e-06,
      "step": 6102,
      "training_step_time": 17.066834211349487
    },
    {
      "epoch": 5.8202743530273435e-06,
      "model_forward_time": 5.5982770919799805,
      "step": 6103
    },
    {
      "epoch": 5.8202743530273435e-06,
      "step": 6103,
      "training_step_time": 17.058122158050537
    },
    {
      "epoch": 5.82122802734375e-06,
      "model_forward_time": 5.598710536956787,
      "step": 6104
    },
    {
      "epoch": 5.82122802734375e-06,
      "step": 6104,
      "training_step_time": 17.033175468444824
    },
    {
      "epoch": 5.822181701660157e-06,
      "model_forward_time": 5.60084867477417,
      "step": 6105
    },
    {
      "epoch": 5.822181701660157e-06,
      "step": 6105,
      "training_step_time": 17.06701421737671
    },
    {
      "epoch": 5.823135375976562e-06,
      "model_forward_time": 5.602329254150391,
      "step": 6106
    },
    {
      "epoch": 5.823135375976562e-06,
      "step": 6106,
      "training_step_time": 17.0642409324646
    },
    {
      "epoch": 5.824089050292969e-06,
      "model_forward_time": 5.9386231899261475,
      "step": 6107
    },
    {
      "epoch": 5.824089050292969e-06,
      "step": 6107,
      "training_step_time": 17.386330604553223
    },
    {
      "epoch": 5.825042724609375e-06,
      "model_forward_time": 5.603294610977173,
      "step": 6108
    },
    {
      "epoch": 5.825042724609375e-06,
      "step": 6108,
      "training_step_time": 17.044101238250732
    },
    {
      "epoch": 5.825996398925781e-06,
      "model_forward_time": 5.591349124908447,
      "step": 6109
    },
    {
      "epoch": 5.825996398925781e-06,
      "step": 6109,
      "training_step_time": 17.03853464126587
    },
    {
      "epoch": 5.826950073242188e-06,
      "grad_norm": 1.0516000986099243,
      "learning_rate": 1e-05,
      "loss": 0.0642,
      "step": 6110
    },
    {
      "epoch": 5.826950073242188e-06,
      "model_forward_time": 5.602723121643066,
      "step": 6110
    },
    {
      "dynamics_loss_avg": 0.05667666420340538,
      "epoch": 5.826950073242188e-06,
      "step": 6110
    },
    {
      "action_loss_avg": 0.010996741545386612,
      "epoch": 5.826950073242188e-06,
      "step": 6110
    },
    {
      "epoch": 5.826950073242188e-06,
      "step": 6110,
      "training_step_time": 17.082794904708862
    },
    {
      "epoch": 5.8279037475585935e-06,
      "model_forward_time": 5.6014416217803955,
      "step": 6111
    },
    {
      "epoch": 5.8279037475585935e-06,
      "step": 6111,
      "training_step_time": 17.094802379608154
    },
    {
      "epoch": 5.828857421875e-06,
      "model_forward_time": 5.975698947906494,
      "step": 6112
    },
    {
      "epoch": 5.828857421875e-06,
      "step": 6112,
      "training_step_time": 17.449264526367188
    },
    {
      "epoch": 5.829811096191407e-06,
      "model_forward_time": 5.610936880111694,
      "step": 6113
    },
    {
      "epoch": 5.829811096191407e-06,
      "step": 6113,
      "training_step_time": 17.089937686920166
    },
    {
      "epoch": 5.830764770507812e-06,
      "model_forward_time": 5.604849338531494,
      "step": 6114
    },
    {
      "epoch": 5.830764770507812e-06,
      "step": 6114,
      "training_step_time": 17.079267978668213
    },
    {
      "epoch": 5.831718444824219e-06,
      "model_forward_time": 5.604305982589722,
      "step": 6115
    },
    {
      "epoch": 5.831718444824219e-06,
      "step": 6115,
      "training_step_time": 17.065099000930786
    },
    {
      "epoch": 5.832672119140625e-06,
      "model_forward_time": 5.587406873703003,
      "step": 6116
    },
    {
      "epoch": 5.832672119140625e-06,
      "step": 6116,
      "training_step_time": 17.011286735534668
    },
    {
      "epoch": 5.833625793457031e-06,
      "model_forward_time": 5.586501598358154,
      "step": 6117
    },
    {
      "epoch": 5.833625793457031e-06,
      "step": 6117,
      "training_step_time": 17.041805505752563
    },
    {
      "epoch": 5.834579467773438e-06,
      "model_forward_time": 5.94464111328125,
      "step": 6118
    },
    {
      "epoch": 5.834579467773438e-06,
      "step": 6118,
      "training_step_time": 17.411778688430786
    },
    {
      "epoch": 5.8355331420898435e-06,
      "model_forward_time": 5.5928332805633545,
      "step": 6119
    },
    {
      "epoch": 5.8355331420898435e-06,
      "step": 6119,
      "training_step_time": 17.04869270324707
    },
    {
      "epoch": 5.83648681640625e-06,
      "grad_norm": 1.7845538854599,
      "learning_rate": 1e-05,
      "loss": 0.0811,
      "step": 6120
    },
    {
      "epoch": 5.83648681640625e-06,
      "model_forward_time": 5.59031867980957,
      "step": 6120
    },
    {
      "dynamics_loss_avg": 0.06306962855160236,
      "epoch": 5.83648681640625e-06,
      "step": 6120
    },
    {
      "action_loss_avg": 0.012497411575168371,
      "epoch": 5.83648681640625e-06,
      "step": 6120
    },
    {
      "epoch": 5.83648681640625e-06,
      "step": 6120,
      "training_step_time": 17.03875231742859
    },
    {
      "epoch": 5.837440490722657e-06,
      "model_forward_time": 5.603759288787842,
      "step": 6121
    },
    {
      "epoch": 5.837440490722657e-06,
      "step": 6121,
      "training_step_time": 17.08968949317932
    },
    {
      "epoch": 5.838394165039062e-06,
      "model_forward_time": 5.61659049987793,
      "step": 6122
    },
    {
      "epoch": 5.838394165039062e-06,
      "step": 6122,
      "training_step_time": 17.127331495285034
    },
    {
      "epoch": 5.839347839355469e-06,
      "model_forward_time": 6.008120775222778,
      "step": 6123
    },
    {
      "epoch": 5.839347839355469e-06,
      "step": 6123,
      "training_step_time": 17.5025155544281
    },
    {
      "epoch": 5.840301513671875e-06,
      "model_forward_time": 5.6120686531066895,
      "step": 6124
    },
    {
      "epoch": 5.840301513671875e-06,
      "step": 6124,
      "training_step_time": 17.09135603904724
    },
    {
      "epoch": 5.841255187988281e-06,
      "model_forward_time": 5.6175031661987305,
      "step": 6125
    },
    {
      "epoch": 5.841255187988281e-06,
      "step": 6125,
      "training_step_time": 17.105554342269897
    },
    {
      "epoch": 5.842208862304688e-06,
      "model_forward_time": 5.603938341140747,
      "step": 6126
    },
    {
      "epoch": 5.842208862304688e-06,
      "step": 6126,
      "training_step_time": 17.065574169158936
    },
    {
      "epoch": 5.8431625366210935e-06,
      "model_forward_time": 5.597365379333496,
      "step": 6127
    },
    {
      "epoch": 5.8431625366210935e-06,
      "step": 6127,
      "training_step_time": 17.03728675842285
    },
    {
      "epoch": 5.8441162109375e-06,
      "model_forward_time": 5.9349822998046875,
      "step": 6128
    },
    {
      "epoch": 5.8441162109375e-06,
      "step": 6128,
      "training_step_time": 17.377188205718994
    },
    {
      "epoch": 5.845069885253906e-06,
      "model_forward_time": 5.595605134963989,
      "step": 6129
    },
    {
      "epoch": 5.845069885253906e-06,
      "step": 6129,
      "training_step_time": 17.055011510849
    },
    {
      "epoch": 5.846023559570312e-06,
      "grad_norm": 1.9016042947769165,
      "learning_rate": 1e-05,
      "loss": 0.1015,
      "step": 6130
    },
    {
      "epoch": 5.846023559570312e-06,
      "model_forward_time": 5.600285053253174,
      "step": 6130
    },
    {
      "dynamics_loss_avg": 0.07921121213585139,
      "epoch": 5.846023559570312e-06,
      "step": 6130
    },
    {
      "action_loss_avg": 0.01906962040811777,
      "epoch": 5.846023559570312e-06,
      "step": 6130
    },
    {
      "epoch": 5.846023559570312e-06,
      "step": 6130,
      "training_step_time": 17.080341815948486
    },
    {
      "epoch": 5.846977233886719e-06,
      "model_forward_time": 5.609635353088379,
      "step": 6131
    },
    {
      "epoch": 5.846977233886719e-06,
      "step": 6131,
      "training_step_time": 17.10896611213684
    },
    {
      "epoch": 5.847930908203125e-06,
      "model_forward_time": 5.610058784484863,
      "step": 6132
    },
    {
      "epoch": 5.847930908203125e-06,
      "step": 6132,
      "training_step_time": 17.099859476089478
    },
    {
      "epoch": 5.848884582519531e-06,
      "model_forward_time": 5.606493949890137,
      "step": 6133
    },
    {
      "epoch": 5.848884582519531e-06,
      "step": 6133,
      "training_step_time": 17.08120894432068
    },
    {
      "epoch": 5.849838256835938e-06,
      "model_forward_time": 5.970919609069824,
      "step": 6134
    },
    {
      "epoch": 5.849838256835938e-06,
      "step": 6134,
      "training_step_time": 17.404377698898315
    },
    {
      "epoch": 5.8507919311523435e-06,
      "model_forward_time": 5.592020034790039,
      "step": 6135
    },
    {
      "epoch": 5.8507919311523435e-06,
      "step": 6135,
      "training_step_time": 17.04105234146118
    },
    {
      "epoch": 5.85174560546875e-06,
      "model_forward_time": 5.5832860469818115,
      "step": 6136
    },
    {
      "epoch": 5.85174560546875e-06,
      "step": 6136,
      "training_step_time": 17.025546550750732
    },
    {
      "epoch": 5.852699279785156e-06,
      "model_forward_time": 5.589008569717407,
      "step": 6137
    },
    {
      "epoch": 5.852699279785156e-06,
      "step": 6137,
      "training_step_time": 17.040680408477783
    },
    {
      "epoch": 5.853652954101562e-06,
      "model_forward_time": 5.593506097793579,
      "step": 6138
    },
    {
      "epoch": 5.853652954101562e-06,
      "step": 6138,
      "training_step_time": 17.03802466392517
    },
    {
      "epoch": 5.854606628417969e-06,
      "model_forward_time": 5.6009228229522705,
      "step": 6139
    },
    {
      "epoch": 5.854606628417969e-06,
      "step": 6139,
      "training_step_time": 17.420921802520752
    },
    {
      "epoch": 5.855560302734375e-06,
      "grad_norm": 1.3818862438201904,
      "learning_rate": 1e-05,
      "loss": 0.0802,
      "step": 6140
    },
    {
      "epoch": 5.855560302734375e-06,
      "model_forward_time": 5.595222234725952,
      "step": 6140
    },
    {
      "dynamics_loss_avg": 0.06834099106490613,
      "epoch": 5.855560302734375e-06,
      "step": 6140
    },
    {
      "action_loss_avg": 0.01886034067720175,
      "epoch": 5.855560302734375e-06,
      "step": 6140
    },
    {
      "epoch": 5.855560302734375e-06,
      "step": 6140,
      "training_step_time": 17.056782484054565
    },
    {
      "epoch": 5.856513977050781e-06,
      "model_forward_time": 5.6048314571380615,
      "step": 6141
    },
    {
      "epoch": 5.856513977050781e-06,
      "step": 6141,
      "training_step_time": 17.075196981430054
    },
    {
      "epoch": 5.857467651367188e-06,
      "model_forward_time": 5.604325771331787,
      "step": 6142
    },
    {
      "epoch": 5.857467651367188e-06,
      "step": 6142,
      "training_step_time": 17.088171005249023
    },
    {
      "epoch": 5.8584213256835936e-06,
      "model_forward_time": 5.606899261474609,
      "step": 6143
    },
    {
      "epoch": 5.8584213256835936e-06,
      "step": 6143,
      "training_step_time": 17.10294461250305
    },
    {
      "epoch": 5.859375e-06,
      "model_forward_time": 5.5980119705200195,
      "step": 6144
    },
    {
      "epoch": 5.859375e-06,
      "step": 6144,
      "training_step_time": 17.08407688140869
    },
    {
      "epoch": 5.860328674316406e-06,
      "model_forward_time": 5.973182678222656,
      "step": 6145
    },
    {
      "epoch": 5.860328674316406e-06,
      "step": 6145,
      "training_step_time": 17.459776401519775
    },
    {
      "epoch": 5.861282348632812e-06,
      "model_forward_time": 5.609747409820557,
      "step": 6146
    },
    {
      "epoch": 5.861282348632812e-06,
      "step": 6146,
      "training_step_time": 17.090631246566772
    },
    {
      "epoch": 5.862236022949219e-06,
      "model_forward_time": 5.60786509513855,
      "step": 6147
    },
    {
      "epoch": 5.862236022949219e-06,
      "step": 6147,
      "training_step_time": 17.070573806762695
    },
    {
      "epoch": 5.863189697265625e-06,
      "model_forward_time": 5.595158576965332,
      "step": 6148
    },
    {
      "epoch": 5.863189697265625e-06,
      "step": 6148,
      "training_step_time": 17.015718460083008
    },
    {
      "epoch": 5.864143371582031e-06,
      "model_forward_time": 5.58127236366272,
      "step": 6149
    },
    {
      "epoch": 5.864143371582031e-06,
      "step": 6149,
      "training_step_time": 16.987102508544922
    },
    {
      "epoch": 5.865097045898438e-06,
      "grad_norm": 0.9086066484451294,
      "learning_rate": 1e-05,
      "loss": 0.0808,
      "step": 6150
    },
    {
      "epoch": 5.865097045898438e-06,
      "model_forward_time": 5.570757865905762,
      "step": 6150
    },
    {
      "dynamics_loss_avg": 0.06247862726449967,
      "epoch": 5.865097045898438e-06,
      "step": 6150
    },
    {
      "action_loss_avg": 0.013183547719381749,
      "epoch": 5.865097045898438e-06,
      "step": 6150
    },
    {
      "epoch": 5.865097045898438e-06,
      "step": 6150,
      "training_step_time": 17.374593257904053
    },
    {
      "epoch": 5.8660507202148436e-06,
      "model_forward_time": 5.577459335327148,
      "step": 6151
    },
    {
      "epoch": 5.8660507202148436e-06,
      "step": 6151,
      "training_step_time": 16.991013765335083
    },
    {
      "epoch": 5.86700439453125e-06,
      "model_forward_time": 5.574261665344238,
      "step": 6152
    },
    {
      "epoch": 5.86700439453125e-06,
      "step": 6152,
      "training_step_time": 16.975985288619995
    },
    {
      "epoch": 5.867958068847656e-06,
      "model_forward_time": 5.5786097049713135,
      "step": 6153
    },
    {
      "epoch": 5.867958068847656e-06,
      "step": 6153,
      "training_step_time": 16.998605012893677
    },
    {
      "epoch": 5.868911743164062e-06,
      "model_forward_time": 5.589188098907471,
      "step": 6154
    },
    {
      "epoch": 5.868911743164062e-06,
      "step": 6154,
      "training_step_time": 17.019346714019775
    },
    {
      "epoch": 5.869865417480469e-06,
      "model_forward_time": 5.591681480407715,
      "step": 6155
    },
    {
      "epoch": 5.869865417480469e-06,
      "step": 6155,
      "training_step_time": 17.06296420097351
    },
    {
      "epoch": 5.870819091796875e-06,
      "model_forward_time": 5.979404926300049,
      "step": 6156
    },
    {
      "epoch": 5.870819091796875e-06,
      "step": 6156,
      "training_step_time": 17.465648412704468
    },
    {
      "epoch": 5.871772766113281e-06,
      "model_forward_time": 5.599471092224121,
      "step": 6157
    },
    {
      "epoch": 5.871772766113281e-06,
      "step": 6157,
      "training_step_time": 17.06145715713501
    },
    {
      "epoch": 5.872726440429688e-06,
      "model_forward_time": 5.607612133026123,
      "step": 6158
    },
    {
      "epoch": 5.872726440429688e-06,
      "step": 6158,
      "training_step_time": 17.051841735839844
    },
    {
      "epoch": 5.8736801147460936e-06,
      "model_forward_time": 5.596505165100098,
      "step": 6159
    },
    {
      "epoch": 5.8736801147460936e-06,
      "step": 6159,
      "training_step_time": 17.060140371322632
    },
    {
      "epoch": 5.8746337890625e-06,
      "grad_norm": 1.240834355354309,
      "learning_rate": 1e-05,
      "loss": 0.0827,
      "step": 6160
    },
    {
      "epoch": 5.8746337890625e-06,
      "model_forward_time": 5.596169471740723,
      "step": 6160
    },
    {
      "dynamics_loss_avg": 0.07196050714701414,
      "epoch": 5.8746337890625e-06,
      "step": 6160
    },
    {
      "action_loss_avg": 0.01621390178333968,
      "epoch": 5.8746337890625e-06,
      "step": 6160
    },
    {
      "epoch": 5.8746337890625e-06,
      "step": 6160,
      "training_step_time": 17.06022334098816
    },
    {
      "epoch": 5.875587463378906e-06,
      "model_forward_time": 5.963008165359497,
      "step": 6161
    },
    {
      "epoch": 5.875587463378906e-06,
      "step": 6161,
      "training_step_time": 17.418906688690186
    },
    {
      "epoch": 5.8765411376953124e-06,
      "model_forward_time": 5.591120719909668,
      "step": 6162
    },
    {
      "epoch": 5.8765411376953124e-06,
      "step": 6162,
      "training_step_time": 17.0406014919281
    },
    {
      "epoch": 5.877494812011719e-06,
      "model_forward_time": 5.593832969665527,
      "step": 6163
    },
    {
      "epoch": 5.877494812011719e-06,
      "step": 6163,
      "training_step_time": 17.064067840576172
    },
    {
      "epoch": 5.878448486328125e-06,
      "model_forward_time": 5.610635757446289,
      "step": 6164
    },
    {
      "epoch": 5.878448486328125e-06,
      "step": 6164,
      "training_step_time": 17.07824158668518
    },
    {
      "epoch": 5.879402160644531e-06,
      "model_forward_time": 5.601528167724609,
      "step": 6165
    },
    {
      "epoch": 5.879402160644531e-06,
      "step": 6165,
      "training_step_time": 17.05071449279785
    },
    {
      "epoch": 5.880355834960938e-06,
      "model_forward_time": 5.597959995269775,
      "step": 6166
    },
    {
      "epoch": 5.880355834960938e-06,
      "step": 6166,
      "training_step_time": 17.05845856666565
    },
    {
      "epoch": 5.8813095092773436e-06,
      "model_forward_time": 5.949028968811035,
      "step": 6167
    },
    {
      "epoch": 5.8813095092773436e-06,
      "step": 6167,
      "training_step_time": 17.3818302154541
    },
    {
      "epoch": 5.88226318359375e-06,
      "model_forward_time": 5.578841686248779,
      "step": 6168
    },
    {
      "epoch": 5.88226318359375e-06,
      "step": 6168,
      "training_step_time": 17.009044408798218
    },
    {
      "epoch": 5.883216857910156e-06,
      "model_forward_time": 5.589612722396851,
      "step": 6169
    },
    {
      "epoch": 5.883216857910156e-06,
      "step": 6169,
      "training_step_time": 17.05316662788391
    },
    {
      "epoch": 5.8841705322265624e-06,
      "grad_norm": 0.7539054751396179,
      "learning_rate": 1e-05,
      "loss": 0.0697,
      "step": 6170
    },
    {
      "epoch": 5.8841705322265624e-06,
      "model_forward_time": 5.605614423751831,
      "step": 6170
    },
    {
      "dynamics_loss_avg": 0.05461762677878142,
      "epoch": 5.8841705322265624e-06,
      "step": 6170
    },
    {
      "action_loss_avg": 0.01372527542989701,
      "epoch": 5.8841705322265624e-06,
      "step": 6170
    },
    {
      "epoch": 5.8841705322265624e-06,
      "step": 6170,
      "training_step_time": 17.078442096710205
    },
    {
      "epoch": 5.885124206542969e-06,
      "model_forward_time": 5.596640586853027,
      "step": 6171
    },
    {
      "epoch": 5.885124206542969e-06,
      "step": 6171,
      "training_step_time": 17.062198877334595
    },
    {
      "epoch": 5.886077880859375e-06,
      "model_forward_time": 5.590367555618286,
      "step": 6172
    },
    {
      "epoch": 5.886077880859375e-06,
      "step": 6172,
      "training_step_time": 17.430784702301025
    },
    {
      "epoch": 5.887031555175781e-06,
      "model_forward_time": 5.614058017730713,
      "step": 6173
    },
    {
      "epoch": 5.887031555175781e-06,
      "step": 6173,
      "training_step_time": 17.108311891555786
    },
    {
      "epoch": 5.887985229492188e-06,
      "model_forward_time": 5.607007265090942,
      "step": 6174
    },
    {
      "epoch": 5.887985229492188e-06,
      "step": 6174,
      "training_step_time": 17.068229913711548
    },
    {
      "epoch": 5.888938903808594e-06,
      "model_forward_time": 5.5868752002716064,
      "step": 6175
    },
    {
      "epoch": 5.888938903808594e-06,
      "step": 6175,
      "training_step_time": 17.022221326828003
    },
    {
      "epoch": 5.889892578125e-06,
      "model_forward_time": 5.57868766784668,
      "step": 6176
    },
    {
      "epoch": 5.889892578125e-06,
      "step": 6176,
      "training_step_time": 17.00991916656494
    },
    {
      "epoch": 5.890846252441406e-06,
      "model_forward_time": 5.572162866592407,
      "step": 6177
    },
    {
      "epoch": 5.890846252441406e-06,
      "step": 6177,
      "training_step_time": 16.995567321777344
    },
    {
      "epoch": 5.8917999267578124e-06,
      "model_forward_time": 5.956762075424194,
      "step": 6178
    },
    {
      "epoch": 5.8917999267578124e-06,
      "step": 6178,
      "training_step_time": 17.376177549362183
    },
    {
      "epoch": 5.892753601074219e-06,
      "model_forward_time": 5.587657451629639,
      "step": 6179
    },
    {
      "epoch": 5.892753601074219e-06,
      "step": 6179,
      "training_step_time": 17.03839111328125
    },
    {
      "epoch": 5.893707275390625e-06,
      "grad_norm": 1.4118967056274414,
      "learning_rate": 1e-05,
      "loss": 0.0728,
      "step": 6180
    },
    {
      "epoch": 5.893707275390625e-06,
      "model_forward_time": 5.604156017303467,
      "step": 6180
    },
    {
      "dynamics_loss_avg": 0.055772301461547616,
      "epoch": 5.893707275390625e-06,
      "step": 6180
    },
    {
      "action_loss_avg": 0.01154243660857901,
      "epoch": 5.893707275390625e-06,
      "step": 6180
    },
    {
      "epoch": 5.893707275390625e-06,
      "step": 6180,
      "training_step_time": 17.08184790611267
    },
    {
      "epoch": 5.894660949707031e-06,
      "model_forward_time": 5.606366395950317,
      "step": 6181
    },
    {
      "epoch": 5.894660949707031e-06,
      "step": 6181,
      "training_step_time": 17.096532344818115
    },
    {
      "epoch": 5.895614624023438e-06,
      "model_forward_time": 5.620239734649658,
      "step": 6182
    },
    {
      "epoch": 5.895614624023438e-06,
      "step": 6182,
      "training_step_time": 17.09631323814392
    },
    {
      "epoch": 5.896568298339844e-06,
      "model_forward_time": 5.592472791671753,
      "step": 6183
    },
    {
      "epoch": 5.896568298339844e-06,
      "step": 6183,
      "training_step_time": 17.372772693634033
    },
    {
      "epoch": 5.89752197265625e-06,
      "model_forward_time": 5.572128057479858,
      "step": 6184
    },
    {
      "epoch": 5.89752197265625e-06,
      "step": 6184,
      "training_step_time": 17.022438764572144
    },
    {
      "epoch": 5.898475646972656e-06,
      "model_forward_time": 5.577326774597168,
      "step": 6185
    },
    {
      "epoch": 5.898475646972656e-06,
      "step": 6185,
      "training_step_time": 16.99449872970581
    },
    {
      "epoch": 5.8994293212890624e-06,
      "model_forward_time": 5.572526693344116,
      "step": 6186
    },
    {
      "epoch": 5.8994293212890624e-06,
      "step": 6186,
      "training_step_time": 16.991413593292236
    },
    {
      "epoch": 5.900382995605469e-06,
      "model_forward_time": 5.5824196338653564,
      "step": 6187
    },
    {
      "epoch": 5.900382995605469e-06,
      "step": 6187,
      "training_step_time": 17.012880563735962
    },
    {
      "epoch": 5.901336669921875e-06,
      "model_forward_time": 5.592016935348511,
      "step": 6188
    },
    {
      "epoch": 5.901336669921875e-06,
      "step": 6188,
      "training_step_time": 17.03995418548584
    },
    {
      "epoch": 5.902290344238281e-06,
      "model_forward_time": 5.975011587142944,
      "step": 6189
    },
    {
      "epoch": 5.902290344238281e-06,
      "step": 6189,
      "training_step_time": 17.45242166519165
    },
    {
      "epoch": 5.903244018554688e-06,
      "grad_norm": 0.4694768488407135,
      "learning_rate": 1e-05,
      "loss": 0.0805,
      "step": 6190
    },
    {
      "epoch": 5.903244018554688e-06,
      "model_forward_time": 5.595212936401367,
      "step": 6190
    },
    {
      "dynamics_loss_avg": 0.06484759319573641,
      "epoch": 5.903244018554688e-06,
      "step": 6190
    },
    {
      "action_loss_avg": 0.015093256602995098,
      "epoch": 5.903244018554688e-06,
      "step": 6190
    },
    {
      "epoch": 5.903244018554688e-06,
      "step": 6190,
      "training_step_time": 17.056888103485107
    },
    {
      "epoch": 5.904197692871094e-06,
      "model_forward_time": 5.591449022293091,
      "step": 6191
    },
    {
      "epoch": 5.904197692871094e-06,
      "step": 6191,
      "training_step_time": 17.02560830116272
    },
    {
      "epoch": 5.9051513671875e-06,
      "model_forward_time": 5.608974456787109,
      "step": 6192
    },
    {
      "epoch": 5.9051513671875e-06,
      "step": 6192,
      "training_step_time": 17.070364713668823
    },
    {
      "epoch": 5.906105041503906e-06,
      "model_forward_time": 5.615023612976074,
      "step": 6193
    },
    {
      "epoch": 5.906105041503906e-06,
      "step": 6193,
      "training_step_time": 17.094743967056274
    },
    {
      "epoch": 5.9070587158203125e-06,
      "model_forward_time": 5.961568593978882,
      "step": 6194
    },
    {
      "epoch": 5.9070587158203125e-06,
      "step": 6194,
      "training_step_time": 17.449271202087402
    },
    {
      "epoch": 5.908012390136719e-06,
      "model_forward_time": 5.620527029037476,
      "step": 6195
    },
    {
      "epoch": 5.908012390136719e-06,
      "step": 6195,
      "training_step_time": 17.102030038833618
    },
    {
      "epoch": 5.908966064453125e-06,
      "model_forward_time": 5.610116720199585,
      "step": 6196
    },
    {
      "epoch": 5.908966064453125e-06,
      "step": 6196,
      "training_step_time": 17.099351167678833
    },
    {
      "epoch": 5.909919738769531e-06,
      "model_forward_time": 5.598605155944824,
      "step": 6197
    },
    {
      "epoch": 5.909919738769531e-06,
      "step": 6197,
      "training_step_time": 17.057209968566895
    },
    {
      "epoch": 5.910873413085938e-06,
      "model_forward_time": 5.595285415649414,
      "step": 6198
    },
    {
      "epoch": 5.910873413085938e-06,
      "step": 6198,
      "training_step_time": 17.041455030441284
    },
    {
      "epoch": 5.911827087402344e-06,
      "model_forward_time": 5.598922491073608,
      "step": 6199
    },
    {
      "epoch": 5.911827087402344e-06,
      "step": 6199,
      "training_step_time": 17.025985717773438
    },
    {
      "epoch": 5.91278076171875e-06,
      "grad_norm": 0.2621934115886688,
      "learning_rate": 1e-05,
      "loss": 0.0622,
      "step": 6200
    },
    {
      "epoch": 5.91278076171875e-06,
      "model_forward_time": 5.958906412124634,
      "step": 6200
    },
    {
      "dynamics_loss_avg": 0.05416241586208344,
      "epoch": 5.91278076171875e-06,
      "step": 6200
    },
    {
      "action_loss_avg": 0.012867474881932139,
      "epoch": 5.91278076171875e-06,
      "step": 6200
    },
    {
      "epoch": 5.91278076171875e-06,
      "step": 6200,
      "training_step_time": 17.397933959960938
    },
    {
      "epoch": 5.913734436035156e-06,
      "model_forward_time": 5.597699165344238,
      "step": 6201
    },
    {
      "epoch": 5.913734436035156e-06,
      "step": 6201,
      "training_step_time": 17.082285165786743
    },
    {
      "epoch": 5.9146881103515625e-06,
      "model_forward_time": 5.603505849838257,
      "step": 6202
    },
    {
      "epoch": 5.9146881103515625e-06,
      "step": 6202,
      "training_step_time": 17.093398571014404
    },
    {
      "epoch": 5.915641784667969e-06,
      "model_forward_time": 5.619302749633789,
      "step": 6203
    },
    {
      "epoch": 5.915641784667969e-06,
      "step": 6203,
      "training_step_time": 17.092984676361084
    },
    {
      "epoch": 5.916595458984375e-06,
      "model_forward_time": 5.620713949203491,
      "step": 6204
    },
    {
      "epoch": 5.916595458984375e-06,
      "step": 6204,
      "training_step_time": 17.098552703857422
    },
    {
      "epoch": 5.917549133300781e-06,
      "model_forward_time": 5.596666574478149,
      "step": 6205
    },
    {
      "epoch": 5.917549133300781e-06,
      "step": 6205,
      "training_step_time": 17.424436807632446
    },
    {
      "epoch": 5.918502807617188e-06,
      "model_forward_time": 5.5965962409973145,
      "step": 6206
    },
    {
      "epoch": 5.918502807617188e-06,
      "step": 6206,
      "training_step_time": 17.036258459091187
    },
    {
      "epoch": 5.919456481933594e-06,
      "model_forward_time": 5.591458082199097,
      "step": 6207
    },
    {
      "epoch": 5.919456481933594e-06,
      "step": 6207,
      "training_step_time": 17.05495047569275
    },
    {
      "epoch": 5.92041015625e-06,
      "model_forward_time": 5.603158473968506,
      "step": 6208
    },
    {
      "epoch": 5.92041015625e-06,
      "step": 6208,
      "training_step_time": 17.07791519165039
    },
    {
      "epoch": 5.921363830566406e-06,
      "model_forward_time": 5.595354318618774,
      "step": 6209
    },
    {
      "epoch": 5.921363830566406e-06,
      "step": 6209,
      "training_step_time": 17.061789512634277
    },
    {
      "epoch": 5.9223175048828125e-06,
      "grad_norm": 1.4815560579299927,
      "learning_rate": 1e-05,
      "loss": 0.0743,
      "step": 6210
    },
    {
      "epoch": 5.9223175048828125e-06,
      "model_forward_time": 5.598613023757935,
      "step": 6210
    },
    {
      "dynamics_loss_avg": 0.05918528810143471,
      "epoch": 5.9223175048828125e-06,
      "step": 6210
    },
    {
      "action_loss_avg": 0.012142785731703043,
      "epoch": 5.9223175048828125e-06,
      "step": 6210
    },
    {
      "epoch": 5.9223175048828125e-06,
      "step": 6210,
      "training_step_time": 17.05085778236389
    },
    {
      "epoch": 5.923271179199219e-06,
      "model_forward_time": 5.9770402908325195,
      "step": 6211
    },
    {
      "epoch": 5.923271179199219e-06,
      "step": 6211,
      "training_step_time": 17.422852993011475
    },
    {
      "epoch": 5.924224853515625e-06,
      "model_forward_time": 5.583548545837402,
      "step": 6212
    },
    {
      "epoch": 5.924224853515625e-06,
      "step": 6212,
      "training_step_time": 17.005424976348877
    },
    {
      "epoch": 5.925178527832031e-06,
      "model_forward_time": 5.58933162689209,
      "step": 6213
    },
    {
      "epoch": 5.925178527832031e-06,
      "step": 6213,
      "training_step_time": 17.046417474746704
    },
    {
      "epoch": 5.926132202148438e-06,
      "model_forward_time": 5.596153259277344,
      "step": 6214
    },
    {
      "epoch": 5.926132202148438e-06,
      "step": 6214,
      "training_step_time": 17.0453679561615
    },
    {
      "epoch": 5.927085876464844e-06,
      "model_forward_time": 5.604331016540527,
      "step": 6215
    },
    {
      "epoch": 5.927085876464844e-06,
      "step": 6215,
      "training_step_time": 17.06623601913452
    },
    {
      "epoch": 5.92803955078125e-06,
      "model_forward_time": 5.596953868865967,
      "step": 6216
    },
    {
      "epoch": 5.92803955078125e-06,
      "step": 6216,
      "training_step_time": 17.389453887939453
    },
    {
      "epoch": 5.928993225097656e-06,
      "model_forward_time": 5.580336093902588,
      "step": 6217
    },
    {
      "epoch": 5.928993225097656e-06,
      "step": 6217,
      "training_step_time": 17.01779556274414
    },
    {
      "epoch": 5.9299468994140625e-06,
      "model_forward_time": 5.58733057975769,
      "step": 6218
    },
    {
      "epoch": 5.9299468994140625e-06,
      "step": 6218,
      "training_step_time": 17.031689643859863
    },
    {
      "epoch": 5.930900573730469e-06,
      "model_forward_time": 5.588284730911255,
      "step": 6219
    },
    {
      "epoch": 5.930900573730469e-06,
      "step": 6219,
      "training_step_time": 17.05737805366516
    },
    {
      "epoch": 5.931854248046875e-06,
      "grad_norm": 0.6445309519767761,
      "learning_rate": 1e-05,
      "loss": 0.0782,
      "step": 6220
    },
    {
      "epoch": 5.931854248046875e-06,
      "model_forward_time": 5.598082065582275,
      "step": 6220
    },
    {
      "dynamics_loss_avg": 0.06510753631591797,
      "epoch": 5.931854248046875e-06,
      "step": 6220
    },
    {
      "action_loss_avg": 0.012539742956869304,
      "epoch": 5.931854248046875e-06,
      "step": 6220
    },
    {
      "epoch": 5.931854248046875e-06,
      "step": 6220,
      "training_step_time": 17.084818601608276
    },
    {
      "epoch": 5.932807922363281e-06,
      "model_forward_time": 5.606966972351074,
      "step": 6221
    },
    {
      "epoch": 5.932807922363281e-06,
      "step": 6221,
      "training_step_time": 17.071914196014404
    },
    {
      "epoch": 5.933761596679688e-06,
      "model_forward_time": 5.9669342041015625,
      "step": 6222
    },
    {
      "epoch": 5.933761596679688e-06,
      "step": 6222,
      "training_step_time": 17.415311813354492
    },
    {
      "epoch": 5.934715270996094e-06,
      "model_forward_time": 5.5974485874176025,
      "step": 6223
    },
    {
      "epoch": 5.934715270996094e-06,
      "step": 6223,
      "training_step_time": 17.069132328033447
    },
    {
      "epoch": 5.9356689453125e-06,
      "model_forward_time": 5.5992114543914795,
      "step": 6224
    },
    {
      "epoch": 5.9356689453125e-06,
      "step": 6224,
      "training_step_time": 17.060750722885132
    },
    {
      "epoch": 5.936622619628906e-06,
      "model_forward_time": 5.595542907714844,
      "step": 6225
    },
    {
      "epoch": 5.936622619628906e-06,
      "step": 6225,
      "training_step_time": 17.05844283103943
    },
    {
      "epoch": 5.9375762939453125e-06,
      "model_forward_time": 5.599503993988037,
      "step": 6226
    },
    {
      "epoch": 5.9375762939453125e-06,
      "step": 6226,
      "training_step_time": 17.058544397354126
    },
    {
      "epoch": 5.938529968261719e-06,
      "model_forward_time": 5.956852912902832,
      "step": 6227
    },
    {
      "epoch": 5.938529968261719e-06,
      "step": 6227,
      "training_step_time": 17.448068618774414
    },
    {
      "epoch": 5.939483642578125e-06,
      "model_forward_time": 5.616085052490234,
      "step": 6228
    },
    {
      "epoch": 5.939483642578125e-06,
      "step": 6228,
      "training_step_time": 17.119216203689575
    },
    {
      "epoch": 5.940437316894531e-06,
      "model_forward_time": 5.597429990768433,
      "step": 6229
    },
    {
      "epoch": 5.940437316894531e-06,
      "step": 6229,
      "training_step_time": 17.040905237197876
    },
    {
      "epoch": 5.941390991210938e-06,
      "grad_norm": 1.8172619342803955,
      "learning_rate": 1e-05,
      "loss": 0.0891,
      "step": 6230
    },
    {
      "epoch": 5.941390991210938e-06,
      "model_forward_time": 5.574877023696899,
      "step": 6230
    },
    {
      "dynamics_loss_avg": 0.07132282964885235,
      "epoch": 5.941390991210938e-06,
      "step": 6230
    },
    {
      "action_loss_avg": 0.015858930302783847,
      "epoch": 5.941390991210938e-06,
      "step": 6230
    },
    {
      "epoch": 5.941390991210938e-06,
      "step": 6230,
      "training_step_time": 17.006407022476196
    },
    {
      "epoch": 5.942344665527344e-06,
      "model_forward_time": 5.57946515083313,
      "step": 6231
    },
    {
      "epoch": 5.942344665527344e-06,
      "step": 6231,
      "training_step_time": 16.99278736114502
    },
    {
      "epoch": 5.94329833984375e-06,
      "model_forward_time": 5.5673816204071045,
      "step": 6232
    },
    {
      "epoch": 5.94329833984375e-06,
      "step": 6232,
      "training_step_time": 16.983642101287842
    },
    {
      "epoch": 5.944252014160156e-06,
      "model_forward_time": 5.947797536849976,
      "step": 6233
    },
    {
      "epoch": 5.944252014160156e-06,
      "step": 6233,
      "training_step_time": 17.371793746948242
    },
    {
      "epoch": 5.9452056884765625e-06,
      "model_forward_time": 5.572177410125732,
      "step": 6234
    },
    {
      "epoch": 5.9452056884765625e-06,
      "step": 6234,
      "training_step_time": 17.00834822654724
    },
    {
      "epoch": 5.946159362792969e-06,
      "model_forward_time": 5.588272571563721,
      "step": 6235
    },
    {
      "epoch": 5.946159362792969e-06,
      "step": 6235,
      "training_step_time": 17.023671865463257
    },
    {
      "epoch": 5.947113037109375e-06,
      "model_forward_time": 5.592990875244141,
      "step": 6236
    },
    {
      "epoch": 5.947113037109375e-06,
      "step": 6236,
      "training_step_time": 17.071171760559082
    },
    {
      "epoch": 5.948066711425781e-06,
      "model_forward_time": 5.608154296875,
      "step": 6237
    },
    {
      "epoch": 5.948066711425781e-06,
      "step": 6237,
      "training_step_time": 17.111950397491455
    },
    {
      "epoch": 5.949020385742188e-06,
      "model_forward_time": 5.627094984054565,
      "step": 6238
    },
    {
      "epoch": 5.949020385742188e-06,
      "step": 6238,
      "training_step_time": 17.479568481445312
    },
    {
      "epoch": 5.949974060058594e-06,
      "model_forward_time": 5.6131885051727295,
      "step": 6239
    },
    {
      "epoch": 5.949974060058594e-06,
      "step": 6239,
      "training_step_time": 17.095905780792236
    },
    {
      "epoch": 5.950927734375e-06,
      "grad_norm": 0.8604800701141357,
      "learning_rate": 1e-05,
      "loss": 0.06,
      "step": 6240
    },
    {
      "epoch": 5.950927734375e-06,
      "model_forward_time": 5.599647521972656,
      "step": 6240
    },
    {
      "dynamics_loss_avg": 0.05728820338845253,
      "epoch": 5.950927734375e-06,
      "step": 6240
    },
    {
      "action_loss_avg": 0.009834899939596653,
      "epoch": 5.950927734375e-06,
      "step": 6240
    },
    {
      "epoch": 5.950927734375e-06,
      "step": 6240,
      "training_step_time": 17.067857265472412
    },
    {
      "epoch": 5.951881408691406e-06,
      "model_forward_time": 5.589447736740112,
      "step": 6241
    },
    {
      "epoch": 5.951881408691406e-06,
      "step": 6241,
      "training_step_time": 17.032808303833008
    },
    {
      "epoch": 5.9528350830078125e-06,
      "model_forward_time": 5.584622144699097,
      "step": 6242
    },
    {
      "epoch": 5.9528350830078125e-06,
      "step": 6242,
      "training_step_time": 17.00687885284424
    },
    {
      "epoch": 5.953788757324219e-06,
      "model_forward_time": 5.585269212722778,
      "step": 6243
    },
    {
      "epoch": 5.953788757324219e-06,
      "step": 6243,
      "training_step_time": 17.012816667556763
    },
    {
      "epoch": 5.954742431640625e-06,
      "model_forward_time": 5.964004039764404,
      "step": 6244
    },
    {
      "epoch": 5.954742431640625e-06,
      "step": 6244,
      "training_step_time": 17.421791553497314
    },
    {
      "epoch": 5.955696105957031e-06,
      "model_forward_time": 5.59581732749939,
      "step": 6245
    },
    {
      "epoch": 5.955696105957031e-06,
      "step": 6245,
      "training_step_time": 17.03112006187439
    },
    {
      "epoch": 5.956649780273438e-06,
      "model_forward_time": 5.583978176116943,
      "step": 6246
    },
    {
      "epoch": 5.956649780273438e-06,
      "step": 6246,
      "training_step_time": 17.02192711830139
    },
    {
      "epoch": 5.957603454589844e-06,
      "model_forward_time": 5.588127136230469,
      "step": 6247
    },
    {
      "epoch": 5.957603454589844e-06,
      "step": 6247,
      "training_step_time": 17.045847415924072
    },
    {
      "epoch": 5.95855712890625e-06,
      "model_forward_time": 5.599769830703735,
      "step": 6248
    },
    {
      "epoch": 5.95855712890625e-06,
      "step": 6248,
      "training_step_time": 17.066861391067505
    },
    {
      "epoch": 5.959510803222656e-06,
      "model_forward_time": 5.612591505050659,
      "step": 6249
    },
    {
      "epoch": 5.959510803222656e-06,
      "step": 6249,
      "training_step_time": 17.465403079986572
    },
    {
      "epoch": 5.9604644775390625e-06,
      "grad_norm": 0.5094828009605408,
      "learning_rate": 1e-05,
      "loss": 0.0818,
      "step": 6250
    },
    {
      "epoch": 5.9604644775390625e-06,
      "model_forward_time": 5.609662294387817,
      "step": 6250
    },
    {
      "dynamics_loss_avg": 0.06862590182572603,
      "epoch": 5.9604644775390625e-06,
      "step": 6250
    },
    {
      "action_loss_avg": 0.011087401630356907,
      "epoch": 5.9604644775390625e-06,
      "step": 6250
    },
    {
      "epoch": 5.9604644775390625e-06,
      "step": 6250,
      "training_step_time": 17.086469411849976
    },
    {
      "epoch": 5.961418151855469e-06,
      "model_forward_time": 5.600006341934204,
      "step": 6251
    },
    {
      "epoch": 5.961418151855469e-06,
      "step": 6251,
      "training_step_time": 17.05350637435913
    },
    {
      "epoch": 5.962371826171875e-06,
      "model_forward_time": 5.57744026184082,
      "step": 6252
    },
    {
      "epoch": 5.962371826171875e-06,
      "step": 6252,
      "training_step_time": 17.00809121131897
    },
    {
      "epoch": 5.963325500488281e-06,
      "model_forward_time": 5.584781169891357,
      "step": 6253
    },
    {
      "epoch": 5.963325500488281e-06,
      "step": 6253,
      "training_step_time": 17.03166890144348
    },
    {
      "epoch": 5.964279174804687e-06,
      "model_forward_time": 5.586602687835693,
      "step": 6254
    },
    {
      "epoch": 5.964279174804687e-06,
      "step": 6254,
      "training_step_time": 17.016667127609253
    },
    {
      "epoch": 5.965232849121094e-06,
      "model_forward_time": 5.954394340515137,
      "step": 6255
    },
    {
      "epoch": 5.965232849121094e-06,
      "step": 6255,
      "training_step_time": 17.420464277267456
    },
    {
      "epoch": 5.9661865234375e-06,
      "model_forward_time": 5.610072135925293,
      "step": 6256
    },
    {
      "epoch": 5.9661865234375e-06,
      "step": 6256,
      "training_step_time": 17.054813146591187
    },
    {
      "epoch": 5.967140197753906e-06,
      "model_forward_time": 5.5922157764434814,
      "step": 6257
    },
    {
      "epoch": 5.967140197753906e-06,
      "step": 6257,
      "training_step_time": 17.069578170776367
    },
    {
      "epoch": 5.9680938720703125e-06,
      "model_forward_time": 5.598616600036621,
      "step": 6258
    },
    {
      "epoch": 5.9680938720703125e-06,
      "step": 6258,
      "training_step_time": 17.0855872631073
    },
    {
      "epoch": 5.969047546386719e-06,
      "model_forward_time": 5.608290195465088,
      "step": 6259
    },
    {
      "epoch": 5.969047546386719e-06,
      "step": 6259,
      "training_step_time": 17.100656509399414
    },
    {
      "epoch": 5.970001220703125e-06,
      "grad_norm": 1.3064874410629272,
      "learning_rate": 1e-05,
      "loss": 0.0808,
      "step": 6260
    },
    {
      "epoch": 5.970001220703125e-06,
      "model_forward_time": 5.980159044265747,
      "step": 6260
    },
    {
      "dynamics_loss_avg": 0.06716181747615338,
      "epoch": 5.970001220703125e-06,
      "step": 6260
    },
    {
      "action_loss_avg": 0.013689272245392204,
      "epoch": 5.970001220703125e-06,
      "step": 6260
    },
    {
      "epoch": 5.970001220703125e-06,
      "step": 6260,
      "training_step_time": 17.46269941329956
    },
    {
      "epoch": 5.970954895019531e-06,
      "model_forward_time": 5.599941968917847,
      "step": 6261
    },
    {
      "epoch": 5.970954895019531e-06,
      "step": 6261,
      "training_step_time": 17.064292430877686
    },
    {
      "epoch": 5.971908569335937e-06,
      "model_forward_time": 5.597453832626343,
      "step": 6262
    },
    {
      "epoch": 5.971908569335937e-06,
      "step": 6262,
      "training_step_time": 17.059451580047607
    },
    {
      "epoch": 5.972862243652344e-06,
      "model_forward_time": 5.602293491363525,
      "step": 6263
    },
    {
      "epoch": 5.972862243652344e-06,
      "step": 6263,
      "training_step_time": 17.074113130569458
    },
    {
      "epoch": 5.97381591796875e-06,
      "model_forward_time": 5.604370355606079,
      "step": 6264
    },
    {
      "epoch": 5.97381591796875e-06,
      "step": 6264,
      "training_step_time": 17.069241762161255
    },
    {
      "epoch": 5.974769592285156e-06,
      "model_forward_time": 5.592572212219238,
      "step": 6265
    },
    {
      "epoch": 5.974769592285156e-06,
      "step": 6265,
      "training_step_time": 17.04788851737976
    },
    {
      "epoch": 5.9757232666015625e-06,
      "model_forward_time": 5.962218761444092,
      "step": 6266
    },
    {
      "epoch": 5.9757232666015625e-06,
      "step": 6266,
      "training_step_time": 17.39908742904663
    },
    {
      "epoch": 5.976676940917969e-06,
      "model_forward_time": 5.589961528778076,
      "step": 6267
    },
    {
      "epoch": 5.976676940917969e-06,
      "step": 6267,
      "training_step_time": 17.03923535346985
    },
    {
      "epoch": 5.977630615234375e-06,
      "model_forward_time": 5.607532262802124,
      "step": 6268
    },
    {
      "epoch": 5.977630615234375e-06,
      "step": 6268,
      "training_step_time": 17.07852578163147
    },
    {
      "epoch": 5.978584289550781e-06,
      "model_forward_time": 5.6030778884887695,
      "step": 6269
    },
    {
      "epoch": 5.978584289550781e-06,
      "step": 6269,
      "training_step_time": 17.069445610046387
    },
    {
      "epoch": 5.979537963867187e-06,
      "grad_norm": 0.18336012959480286,
      "learning_rate": 1e-05,
      "loss": 0.0584,
      "step": 6270
    },
    {
      "epoch": 5.979537963867187e-06,
      "model_forward_time": 5.593984365463257,
      "step": 6270
    },
    {
      "dynamics_loss_avg": 0.05133801233023405,
      "epoch": 5.979537963867187e-06,
      "step": 6270
    },
    {
      "action_loss_avg": 0.007775696401949972,
      "epoch": 5.979537963867187e-06,
      "step": 6270
    },
    {
      "epoch": 5.979537963867187e-06,
      "step": 6270,
      "training_step_time": 17.057592391967773
    },
    {
      "epoch": 5.980491638183594e-06,
      "model_forward_time": 5.601016283035278,
      "step": 6271
    },
    {
      "epoch": 5.980491638183594e-06,
      "step": 6271,
      "training_step_time": 17.446800231933594
    },
    {
      "epoch": 5.9814453125e-06,
      "model_forward_time": 5.597807884216309,
      "step": 6272
    },
    {
      "epoch": 5.9814453125e-06,
      "step": 6272,
      "training_step_time": 17.05742120742798
    },
    {
      "epoch": 5.982398986816406e-06,
      "model_forward_time": 5.600651979446411,
      "step": 6273
    },
    {
      "epoch": 5.982398986816406e-06,
      "step": 6273,
      "training_step_time": 17.05035400390625
    },
    {
      "epoch": 5.9833526611328125e-06,
      "model_forward_time": 5.592815160751343,
      "step": 6274
    },
    {
      "epoch": 5.9833526611328125e-06,
      "step": 6274,
      "training_step_time": 17.030319213867188
    },
    {
      "epoch": 5.984306335449219e-06,
      "model_forward_time": 5.578378200531006,
      "step": 6275
    },
    {
      "epoch": 5.984306335449219e-06,
      "step": 6275,
      "training_step_time": 17.018096446990967
    },
    {
      "epoch": 5.985260009765625e-06,
      "model_forward_time": 5.582439661026001,
      "step": 6276
    },
    {
      "epoch": 5.985260009765625e-06,
      "step": 6276,
      "training_step_time": 17.04393434524536
    },
    {
      "epoch": 5.986213684082031e-06,
      "model_forward_time": 5.978143692016602,
      "step": 6277
    },
    {
      "epoch": 5.986213684082031e-06,
      "step": 6277,
      "training_step_time": 17.465566873550415
    },
    {
      "epoch": 5.987167358398437e-06,
      "model_forward_time": 5.623463869094849,
      "step": 6278
    },
    {
      "epoch": 5.987167358398437e-06,
      "step": 6278,
      "training_step_time": 17.154513835906982
    },
    {
      "epoch": 5.988121032714844e-06,
      "model_forward_time": 5.619001865386963,
      "step": 6279
    },
    {
      "epoch": 5.988121032714844e-06,
      "step": 6279,
      "training_step_time": 17.112213850021362
    },
    {
      "epoch": 5.98907470703125e-06,
      "grad_norm": 0.1350271850824356,
      "learning_rate": 1e-05,
      "loss": 0.0706,
      "step": 6280
    },
    {
      "epoch": 5.98907470703125e-06,
      "model_forward_time": 5.623540878295898,
      "step": 6280
    },
    {
      "dynamics_loss_avg": 0.05555562060326338,
      "epoch": 5.98907470703125e-06,
      "step": 6280
    },
    {
      "action_loss_avg": 0.01010383467655629,
      "epoch": 5.98907470703125e-06,
      "step": 6280
    },
    {
      "epoch": 5.98907470703125e-06,
      "step": 6280,
      "training_step_time": 17.115200519561768
    },
    {
      "epoch": 5.990028381347656e-06,
      "model_forward_time": 5.609544277191162,
      "step": 6281
    },
    {
      "epoch": 5.990028381347656e-06,
      "step": 6281,
      "training_step_time": 17.102766036987305
    },
    {
      "epoch": 5.9909820556640625e-06,
      "model_forward_time": 5.60571026802063,
      "step": 6282
    },
    {
      "epoch": 5.9909820556640625e-06,
      "step": 6282,
      "training_step_time": 17.46265721321106
    },
    {
      "epoch": 5.991935729980469e-06,
      "model_forward_time": 5.606269359588623,
      "step": 6283
    },
    {
      "epoch": 5.991935729980469e-06,
      "step": 6283,
      "training_step_time": 17.090831518173218
    },
    {
      "epoch": 5.992889404296875e-06,
      "model_forward_time": 5.6040425300598145,
      "step": 6284
    },
    {
      "epoch": 5.992889404296875e-06,
      "step": 6284,
      "training_step_time": 17.085282802581787
    },
    {
      "epoch": 5.993843078613281e-06,
      "model_forward_time": 5.599431276321411,
      "step": 6285
    },
    {
      "epoch": 5.993843078613281e-06,
      "step": 6285,
      "training_step_time": 17.06829071044922
    },
    {
      "epoch": 5.994796752929687e-06,
      "model_forward_time": 5.597041845321655,
      "step": 6286
    },
    {
      "epoch": 5.994796752929687e-06,
      "step": 6286,
      "training_step_time": 17.03728985786438
    },
    {
      "epoch": 5.995750427246094e-06,
      "model_forward_time": 5.585264444351196,
      "step": 6287
    },
    {
      "epoch": 5.995750427246094e-06,
      "step": 6287,
      "training_step_time": 17.025949478149414
    },
    {
      "epoch": 5.9967041015625e-06,
      "model_forward_time": 5.9558868408203125,
      "step": 6288
    },
    {
      "epoch": 5.9967041015625e-06,
      "step": 6288,
      "training_step_time": 17.423141717910767
    },
    {
      "epoch": 5.997657775878906e-06,
      "model_forward_time": 5.596386194229126,
      "step": 6289
    },
    {
      "epoch": 5.997657775878906e-06,
      "step": 6289,
      "training_step_time": 17.04366374015808
    },
    {
      "epoch": 5.9986114501953125e-06,
      "grad_norm": 1.5274622440338135,
      "learning_rate": 1e-05,
      "loss": 0.088,
      "step": 6290
    },
    {
      "epoch": 5.9986114501953125e-06,
      "model_forward_time": 5.604965448379517,
      "step": 6290
    },
    {
      "dynamics_loss_avg": 0.07583754435181618,
      "epoch": 5.9986114501953125e-06,
      "step": 6290
    },
    {
      "action_loss_avg": 0.0174870029091835,
      "epoch": 5.9986114501953125e-06,
      "step": 6290
    },
    {
      "epoch": 5.9986114501953125e-06,
      "step": 6290,
      "training_step_time": 17.046261072158813
    },
    {
      "epoch": 5.999565124511719e-06,
      "model_forward_time": 5.592094898223877,
      "step": 6291
    },
    {
      "epoch": 5.999565124511719e-06,
      "step": 6291,
      "training_step_time": 17.042560338974
    },
    {
      "epoch": 6.000518798828125e-06,
      "model_forward_time": 5.596767425537109,
      "step": 6292
    },
    {
      "epoch": 6.000518798828125e-06,
      "step": 6292,
      "training_step_time": 17.03322958946228
    },
    {
      "epoch": 6.001472473144531e-06,
      "model_forward_time": 5.953703165054321,
      "step": 6293
    },
    {
      "epoch": 6.001472473144531e-06,
      "step": 6293,
      "training_step_time": 17.41931128501892
    },
    {
      "epoch": 6.002426147460937e-06,
      "model_forward_time": 5.6064817905426025,
      "step": 6294
    },
    {
      "epoch": 6.002426147460937e-06,
      "step": 6294,
      "training_step_time": 17.05228066444397
    },
    {
      "epoch": 6.003379821777344e-06,
      "model_forward_time": 5.600011587142944,
      "step": 6295
    },
    {
      "epoch": 6.003379821777344e-06,
      "step": 6295,
      "training_step_time": 17.069122314453125
    },
    {
      "epoch": 6.00433349609375e-06,
      "model_forward_time": 5.592069864273071,
      "step": 6296
    },
    {
      "epoch": 6.00433349609375e-06,
      "step": 6296,
      "training_step_time": 17.036478281021118
    },
    {
      "epoch": 6.005287170410156e-06,
      "model_forward_time": 5.602917909622192,
      "step": 6297
    },
    {
      "epoch": 6.005287170410156e-06,
      "step": 6297,
      "training_step_time": 17.039870738983154
    },
    {
      "epoch": 6.0062408447265625e-06,
      "model_forward_time": 5.599853038787842,
      "step": 6298
    },
    {
      "epoch": 6.0062408447265625e-06,
      "step": 6298,
      "training_step_time": 17.043956518173218
    },
    {
      "epoch": 6.007194519042969e-06,
      "model_forward_time": 5.955843925476074,
      "step": 6299
    },
    {
      "epoch": 6.007194519042969e-06,
      "step": 6299,
      "training_step_time": 17.409315586090088
    },
    {
      "epoch": 6.008148193359375e-06,
      "grad_norm": 0.6025993824005127,
      "learning_rate": 1e-05,
      "loss": 0.0635,
      "step": 6300
    },
    {
      "epoch": 6.008148193359375e-06,
      "model_forward_time": 5.595531940460205,
      "step": 6300
    },
    {
      "dynamics_loss_avg": 0.05273177344352007,
      "epoch": 6.008148193359375e-06,
      "step": 6300
    },
    {
      "action_loss_avg": 0.009980309382081031,
      "epoch": 6.008148193359375e-06,
      "step": 6300
    },
    {
      "epoch": 6.008148193359375e-06,
      "step": 6300,
      "training_step_time": 17.054121732711792
    },
    {
      "epoch": 6.009101867675781e-06,
      "model_forward_time": 5.596755504608154,
      "step": 6301
    },
    {
      "epoch": 6.009101867675781e-06,
      "step": 6301,
      "training_step_time": 17.053431510925293
    },
    {
      "epoch": 6.010055541992187e-06,
      "model_forward_time": 5.5947349071502686,
      "step": 6302
    },
    {
      "epoch": 6.010055541992187e-06,
      "step": 6302,
      "training_step_time": 17.047818899154663
    },
    {
      "epoch": 6.011009216308594e-06,
      "model_forward_time": 5.603359222412109,
      "step": 6303
    },
    {
      "epoch": 6.011009216308594e-06,
      "step": 6303,
      "training_step_time": 17.091672897338867
    },
    {
      "epoch": 6.011962890625e-06,
      "model_forward_time": 5.605634689331055,
      "step": 6304
    },
    {
      "epoch": 6.011962890625e-06,
      "step": 6304,
      "training_step_time": 17.475209951400757
    },
    {
      "epoch": 6.012916564941406e-06,
      "model_forward_time": 5.6013593673706055,
      "step": 6305
    },
    {
      "epoch": 6.012916564941406e-06,
      "step": 6305,
      "training_step_time": 17.0670108795166
    },
    {
      "epoch": 6.0138702392578125e-06,
      "model_forward_time": 5.604974031448364,
      "step": 6306
    },
    {
      "epoch": 6.0138702392578125e-06,
      "step": 6306,
      "training_step_time": 17.085585355758667
    },
    {
      "epoch": 6.014823913574219e-06,
      "model_forward_time": 5.605567693710327,
      "step": 6307
    },
    {
      "epoch": 6.014823913574219e-06,
      "step": 6307,
      "training_step_time": 17.08422303199768
    },
    {
      "epoch": 6.015777587890625e-06,
      "model_forward_time": 5.596174955368042,
      "step": 6308
    },
    {
      "epoch": 6.015777587890625e-06,
      "step": 6308,
      "training_step_time": 17.055169820785522
    },
    {
      "epoch": 6.016731262207031e-06,
      "model_forward_time": 5.595678091049194,
      "step": 6309
    },
    {
      "epoch": 6.016731262207031e-06,
      "step": 6309,
      "training_step_time": 17.04721450805664
    },
    {
      "epoch": 6.017684936523437e-06,
      "grad_norm": 0.30701160430908203,
      "learning_rate": 1e-05,
      "loss": 0.0702,
      "step": 6310
    },
    {
      "epoch": 6.017684936523437e-06,
      "model_forward_time": 5.9778196811676025,
      "step": 6310
    },
    {
      "dynamics_loss_avg": 0.05484496168792248,
      "epoch": 6.017684936523437e-06,
      "step": 6310
    },
    {
      "action_loss_avg": 0.01038051089271903,
      "epoch": 6.017684936523437e-06,
      "step": 6310
    },
    {
      "epoch": 6.017684936523437e-06,
      "step": 6310,
      "training_step_time": 17.415352821350098
    },
    {
      "epoch": 6.018638610839844e-06,
      "model_forward_time": 5.609189510345459,
      "step": 6311
    },
    {
      "epoch": 6.018638610839844e-06,
      "step": 6311,
      "training_step_time": 17.1031231880188
    },
    {
      "epoch": 6.01959228515625e-06,
      "model_forward_time": 5.625293254852295,
      "step": 6312
    },
    {
      "epoch": 6.01959228515625e-06,
      "step": 6312,
      "training_step_time": 17.10210084915161
    },
    {
      "epoch": 6.020545959472656e-06,
      "model_forward_time": 5.6022162437438965,
      "step": 6313
    },
    {
      "epoch": 6.020545959472656e-06,
      "step": 6313,
      "training_step_time": 17.06497883796692
    },
    {
      "epoch": 6.0214996337890626e-06,
      "model_forward_time": 5.6089067459106445,
      "step": 6314
    },
    {
      "epoch": 6.0214996337890626e-06,
      "step": 6314,
      "training_step_time": 17.101939916610718
    },
    {
      "epoch": 6.022453308105469e-06,
      "model_forward_time": 5.615539312362671,
      "step": 6315
    },
    {
      "epoch": 6.022453308105469e-06,
      "step": 6315,
      "training_step_time": 17.472261905670166
    },
    {
      "epoch": 6.023406982421875e-06,
      "model_forward_time": 5.614317417144775,
      "step": 6316
    },
    {
      "epoch": 6.023406982421875e-06,
      "step": 6316,
      "training_step_time": 17.08646512031555
    },
    {
      "epoch": 6.024360656738281e-06,
      "model_forward_time": 5.597023010253906,
      "step": 6317
    },
    {
      "epoch": 6.024360656738281e-06,
      "step": 6317,
      "training_step_time": 17.0394868850708
    },
    {
      "epoch": 6.025314331054687e-06,
      "model_forward_time": 5.5629541873931885,
      "step": 6318
    },
    {
      "epoch": 6.025314331054687e-06,
      "step": 6318,
      "training_step_time": 16.970301628112793
    },
    {
      "epoch": 6.026268005371094e-06,
      "model_forward_time": 5.582490921020508,
      "step": 6319
    },
    {
      "epoch": 6.026268005371094e-06,
      "step": 6319,
      "training_step_time": 16.99402666091919
    },
    {
      "epoch": 6.0272216796875e-06,
      "grad_norm": 0.36897119879722595,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 6320
    },
    {
      "epoch": 6.0272216796875e-06,
      "model_forward_time": 5.57793927192688,
      "step": 6320
    },
    {
      "dynamics_loss_avg": 0.06607655845582486,
      "epoch": 6.0272216796875e-06,
      "step": 6320
    },
    {
      "action_loss_avg": 0.013416403182782232,
      "epoch": 6.0272216796875e-06,
      "step": 6320
    },
    {
      "epoch": 6.0272216796875e-06,
      "step": 6320,
      "training_step_time": 16.994137287139893
    },
    {
      "epoch": 6.028175354003906e-06,
      "model_forward_time": 5.932075262069702,
      "step": 6321
    },
    {
      "epoch": 6.028175354003906e-06,
      "step": 6321,
      "training_step_time": 17.373513221740723
    },
    {
      "epoch": 6.0291290283203126e-06,
      "model_forward_time": 5.585054874420166,
      "step": 6322
    },
    {
      "epoch": 6.0291290283203126e-06,
      "step": 6322,
      "training_step_time": 17.02779197692871
    },
    {
      "epoch": 6.030082702636719e-06,
      "model_forward_time": 5.5985400676727295,
      "step": 6323
    },
    {
      "epoch": 6.030082702636719e-06,
      "step": 6323,
      "training_step_time": 17.058355569839478
    },
    {
      "epoch": 6.031036376953125e-06,
      "model_forward_time": 5.5971550941467285,
      "step": 6324
    },
    {
      "epoch": 6.031036376953125e-06,
      "step": 6324,
      "training_step_time": 17.079728364944458
    },
    {
      "epoch": 6.031990051269531e-06,
      "model_forward_time": 5.620155334472656,
      "step": 6325
    },
    {
      "epoch": 6.031990051269531e-06,
      "step": 6325,
      "training_step_time": 17.10236406326294
    },
    {
      "epoch": 6.032943725585937e-06,
      "model_forward_time": 5.978964805603027,
      "step": 6326
    },
    {
      "epoch": 6.032943725585937e-06,
      "step": 6326,
      "training_step_time": 17.471341133117676
    },
    {
      "epoch": 6.033897399902344e-06,
      "model_forward_time": 5.604907512664795,
      "step": 6327
    },
    {
      "epoch": 6.033897399902344e-06,
      "step": 6327,
      "training_step_time": 17.06743812561035
    },
    {
      "epoch": 6.03485107421875e-06,
      "model_forward_time": 5.58626651763916,
      "step": 6328
    },
    {
      "epoch": 6.03485107421875e-06,
      "step": 6328,
      "training_step_time": 17.006163120269775
    },
    {
      "epoch": 6.035804748535156e-06,
      "model_forward_time": 5.580756187438965,
      "step": 6329
    },
    {
      "epoch": 6.035804748535156e-06,
      "step": 6329,
      "training_step_time": 17.022156476974487
    },
    {
      "epoch": 6.0367584228515626e-06,
      "grad_norm": 0.6235396862030029,
      "learning_rate": 1e-05,
      "loss": 0.0795,
      "step": 6330
    },
    {
      "epoch": 6.0367584228515626e-06,
      "model_forward_time": 5.577225685119629,
      "step": 6330
    },
    {
      "dynamics_loss_avg": 0.06395407319068909,
      "epoch": 6.0367584228515626e-06,
      "step": 6330
    },
    {
      "action_loss_avg": 0.013438351871445774,
      "epoch": 6.0367584228515626e-06,
      "step": 6330
    },
    {
      "epoch": 6.0367584228515626e-06,
      "step": 6330,
      "training_step_time": 16.9953670501709
    },
    {
      "epoch": 6.037712097167969e-06,
      "model_forward_time": 5.588502407073975,
      "step": 6331
    },
    {
      "epoch": 6.037712097167969e-06,
      "step": 6331,
      "training_step_time": 16.99728226661682
    },
    {
      "epoch": 6.038665771484375e-06,
      "model_forward_time": 5.953099727630615,
      "step": 6332
    },
    {
      "epoch": 6.038665771484375e-06,
      "step": 6332,
      "training_step_time": 17.38422203063965
    },
    {
      "epoch": 6.0396194458007814e-06,
      "model_forward_time": 5.588139533996582,
      "step": 6333
    },
    {
      "epoch": 6.0396194458007814e-06,
      "step": 6333,
      "training_step_time": 17.045738220214844
    },
    {
      "epoch": 6.040573120117187e-06,
      "model_forward_time": 5.589461803436279,
      "step": 6334
    },
    {
      "epoch": 6.040573120117187e-06,
      "step": 6334,
      "training_step_time": 17.038942337036133
    },
    {
      "epoch": 6.041526794433594e-06,
      "model_forward_time": 5.604893445968628,
      "step": 6335
    },
    {
      "epoch": 6.041526794433594e-06,
      "step": 6335,
      "training_step_time": 17.09995198249817
    },
    {
      "epoch": 6.04248046875e-06,
      "model_forward_time": 5.614882946014404,
      "step": 6336
    },
    {
      "epoch": 6.04248046875e-06,
      "step": 6336,
      "training_step_time": 17.10384440422058
    },
    {
      "epoch": 6.043434143066406e-06,
      "model_forward_time": 5.621306896209717,
      "step": 6337
    },
    {
      "epoch": 6.043434143066406e-06,
      "step": 6337,
      "training_step_time": 17.452455759048462
    },
    {
      "epoch": 6.0443878173828126e-06,
      "model_forward_time": 5.590267658233643,
      "step": 6338
    },
    {
      "epoch": 6.0443878173828126e-06,
      "step": 6338,
      "training_step_time": 17.007831573486328
    },
    {
      "epoch": 6.045341491699219e-06,
      "model_forward_time": 5.574248790740967,
      "step": 6339
    },
    {
      "epoch": 6.045341491699219e-06,
      "step": 6339,
      "training_step_time": 16.985941648483276
    },
    {
      "epoch": 6.046295166015625e-06,
      "grad_norm": 0.8548149466514587,
      "learning_rate": 1e-05,
      "loss": 0.0859,
      "step": 6340
    },
    {
      "epoch": 6.046295166015625e-06,
      "model_forward_time": 5.568104028701782,
      "step": 6340
    },
    {
      "dynamics_loss_avg": 0.0689142256975174,
      "epoch": 6.046295166015625e-06,
      "step": 6340
    },
    {
      "action_loss_avg": 0.01420433702878654,
      "epoch": 6.046295166015625e-06,
      "step": 6340
    },
    {
      "epoch": 6.046295166015625e-06,
      "step": 6340,
      "training_step_time": 17.001493453979492
    },
    {
      "epoch": 6.0472488403320314e-06,
      "model_forward_time": 5.563086748123169,
      "step": 6341
    },
    {
      "epoch": 6.0472488403320314e-06,
      "step": 6341,
      "training_step_time": 16.97154712677002
    },
    {
      "epoch": 6.048202514648437e-06,
      "model_forward_time": 5.570260047912598,
      "step": 6342
    },
    {
      "epoch": 6.048202514648437e-06,
      "step": 6342,
      "training_step_time": 16.964914560317993
    },
    {
      "epoch": 6.049156188964844e-06,
      "model_forward_time": 5.9492738246917725,
      "step": 6343
    },
    {
      "epoch": 6.049156188964844e-06,
      "step": 6343,
      "training_step_time": 17.364712476730347
    },
    {
      "epoch": 6.05010986328125e-06,
      "model_forward_time": 5.576710224151611,
      "step": 6344
    },
    {
      "epoch": 6.05010986328125e-06,
      "step": 6344,
      "training_step_time": 17.01099157333374
    },
    {
      "epoch": 6.051063537597656e-06,
      "model_forward_time": 5.580729246139526,
      "step": 6345
    },
    {
      "epoch": 6.051063537597656e-06,
      "step": 6345,
      "training_step_time": 17.022164821624756
    },
    {
      "epoch": 6.052017211914063e-06,
      "model_forward_time": 5.593798875808716,
      "step": 6346
    },
    {
      "epoch": 6.052017211914063e-06,
      "step": 6346,
      "training_step_time": 17.06553316116333
    },
    {
      "epoch": 6.052970886230469e-06,
      "model_forward_time": 5.60337495803833,
      "step": 6347
    },
    {
      "epoch": 6.052970886230469e-06,
      "step": 6347,
      "training_step_time": 17.082948446273804
    },
    {
      "epoch": 6.053924560546875e-06,
      "model_forward_time": 5.61775803565979,
      "step": 6348
    },
    {
      "epoch": 6.053924560546875e-06,
      "step": 6348,
      "training_step_time": 17.47847270965576
    },
    {
      "epoch": 6.0548782348632814e-06,
      "model_forward_time": 5.605776786804199,
      "step": 6349
    },
    {
      "epoch": 6.0548782348632814e-06,
      "step": 6349,
      "training_step_time": 17.070976734161377
    },
    {
      "epoch": 6.055831909179687e-06,
      "grad_norm": 1.8932979106903076,
      "learning_rate": 1e-05,
      "loss": 0.0743,
      "step": 6350
    },
    {
      "epoch": 6.055831909179687e-06,
      "model_forward_time": 5.595670938491821,
      "step": 6350
    },
    {
      "dynamics_loss_avg": 0.05797600820660591,
      "epoch": 6.055831909179687e-06,
      "step": 6350
    },
    {
      "action_loss_avg": 0.012164699053391815,
      "epoch": 6.055831909179687e-06,
      "step": 6350
    },
    {
      "epoch": 6.055831909179687e-06,
      "step": 6350,
      "training_step_time": 17.0500431060791
    },
    {
      "epoch": 6.056785583496094e-06,
      "model_forward_time": 5.592172384262085,
      "step": 6351
    },
    {
      "epoch": 6.056785583496094e-06,
      "step": 6351,
      "training_step_time": 17.02534770965576
    },
    {
      "epoch": 6.0577392578125e-06,
      "model_forward_time": 5.581485271453857,
      "step": 6352
    },
    {
      "epoch": 6.0577392578125e-06,
      "step": 6352,
      "training_step_time": 17.001445770263672
    },
    {
      "epoch": 6.058692932128906e-06,
      "model_forward_time": 5.582095623016357,
      "step": 6353
    },
    {
      "epoch": 6.058692932128906e-06,
      "step": 6353,
      "training_step_time": 17.0435848236084
    },
    {
      "epoch": 6.059646606445313e-06,
      "model_forward_time": 5.979584693908691,
      "step": 6354
    },
    {
      "epoch": 6.059646606445313e-06,
      "step": 6354,
      "training_step_time": 17.46051549911499
    },
    {
      "epoch": 6.060600280761719e-06,
      "model_forward_time": 5.599708080291748,
      "step": 6355
    },
    {
      "epoch": 6.060600280761719e-06,
      "step": 6355,
      "training_step_time": 17.044285535812378
    },
    {
      "epoch": 6.061553955078125e-06,
      "model_forward_time": 5.598659038543701,
      "step": 6356
    },
    {
      "epoch": 6.061553955078125e-06,
      "step": 6356,
      "training_step_time": 17.041536569595337
    },
    {
      "epoch": 6.0625076293945314e-06,
      "model_forward_time": 5.5941290855407715,
      "step": 6357
    },
    {
      "epoch": 6.0625076293945314e-06,
      "step": 6357,
      "training_step_time": 17.029398679733276
    },
    {
      "epoch": 6.063461303710937e-06,
      "model_forward_time": 5.592113018035889,
      "step": 6358
    },
    {
      "epoch": 6.063461303710937e-06,
      "step": 6358,
      "training_step_time": 17.03497290611267
    },
    {
      "epoch": 6.064414978027344e-06,
      "model_forward_time": 5.951182126998901,
      "step": 6359
    },
    {
      "epoch": 6.064414978027344e-06,
      "step": 6359,
      "training_step_time": 17.4409396648407
    },
    {
      "epoch": 6.06536865234375e-06,
      "grad_norm": 0.7574607729911804,
      "learning_rate": 1e-05,
      "loss": 0.0785,
      "step": 6360
    },
    {
      "epoch": 6.06536865234375e-06,
      "model_forward_time": 5.617980480194092,
      "step": 6360
    },
    {
      "dynamics_loss_avg": 0.06403678562492132,
      "epoch": 6.06536865234375e-06,
      "step": 6360
    },
    {
      "action_loss_avg": 0.01679623262025416,
      "epoch": 6.06536865234375e-06,
      "step": 6360
    },
    {
      "epoch": 6.06536865234375e-06,
      "step": 6360,
      "training_step_time": 17.126961708068848
    },
    {
      "epoch": 6.066322326660156e-06,
      "model_forward_time": 5.602534770965576,
      "step": 6361
    },
    {
      "epoch": 6.066322326660156e-06,
      "step": 6361,
      "training_step_time": 17.06452965736389
    },
    {
      "epoch": 6.067276000976563e-06,
      "model_forward_time": 5.592589616775513,
      "step": 6362
    },
    {
      "epoch": 6.067276000976563e-06,
      "step": 6362,
      "training_step_time": 17.044074773788452
    },
    {
      "epoch": 6.068229675292969e-06,
      "model_forward_time": 5.584581136703491,
      "step": 6363
    },
    {
      "epoch": 6.068229675292969e-06,
      "step": 6363,
      "training_step_time": 17.033923387527466
    },
    {
      "epoch": 6.069183349609375e-06,
      "model_forward_time": 5.59040379524231,
      "step": 6364
    },
    {
      "epoch": 6.069183349609375e-06,
      "step": 6364,
      "training_step_time": 17.038688898086548
    },
    {
      "epoch": 6.0701370239257815e-06,
      "model_forward_time": 5.975808620452881,
      "step": 6365
    },
    {
      "epoch": 6.0701370239257815e-06,
      "step": 6365,
      "training_step_time": 17.390197277069092
    },
    {
      "epoch": 6.071090698242187e-06,
      "model_forward_time": 5.600461721420288,
      "step": 6366
    },
    {
      "epoch": 6.071090698242187e-06,
      "step": 6366,
      "training_step_time": 17.10137152671814
    },
    {
      "epoch": 6.072044372558594e-06,
      "model_forward_time": 5.621072292327881,
      "step": 6367
    },
    {
      "epoch": 6.072044372558594e-06,
      "step": 6367,
      "training_step_time": 17.100003480911255
    },
    {
      "epoch": 6.072998046875e-06,
      "model_forward_time": 5.599379777908325,
      "step": 6368
    },
    {
      "epoch": 6.072998046875e-06,
      "step": 6368,
      "training_step_time": 17.035093545913696
    },
    {
      "epoch": 6.073951721191406e-06,
      "model_forward_time": 5.579540729522705,
      "step": 6369
    },
    {
      "epoch": 6.073951721191406e-06,
      "step": 6369,
      "training_step_time": 16.99292802810669
    },
    {
      "epoch": 6.074905395507813e-06,
      "grad_norm": 1.0536168813705444,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 6370
    },
    {
      "epoch": 6.074905395507813e-06,
      "model_forward_time": 5.574307203292847,
      "step": 6370
    },
    {
      "dynamics_loss_avg": 0.06324988845735788,
      "epoch": 6.074905395507813e-06,
      "step": 6370
    },
    {
      "action_loss_avg": 0.013235940574668347,
      "epoch": 6.074905395507813e-06,
      "step": 6370
    },
    {
      "epoch": 6.074905395507813e-06,
      "step": 6370,
      "training_step_time": 17.366259336471558
    },
    {
      "epoch": 6.075859069824219e-06,
      "model_forward_time": 5.58020281791687,
      "step": 6371
    },
    {
      "epoch": 6.075859069824219e-06,
      "step": 6371,
      "training_step_time": 17.013705253601074
    },
    {
      "epoch": 6.076812744140625e-06,
      "model_forward_time": 5.588943243026733,
      "step": 6372
    },
    {
      "epoch": 6.076812744140625e-06,
      "step": 6372,
      "training_step_time": 17.02525019645691
    },
    {
      "epoch": 6.0777664184570315e-06,
      "model_forward_time": 5.600838661193848,
      "step": 6373
    },
    {
      "epoch": 6.0777664184570315e-06,
      "step": 6373,
      "training_step_time": 17.086639165878296
    },
    {
      "epoch": 6.078720092773437e-06,
      "model_forward_time": 5.615512132644653,
      "step": 6374
    },
    {
      "epoch": 6.078720092773437e-06,
      "step": 6374,
      "training_step_time": 17.106049299240112
    },
    {
      "epoch": 6.079673767089844e-06,
      "model_forward_time": 5.611457824707031,
      "step": 6375
    },
    {
      "epoch": 6.079673767089844e-06,
      "step": 6375,
      "training_step_time": 17.07572650909424
    },
    {
      "epoch": 6.08062744140625e-06,
      "model_forward_time": 5.981755256652832,
      "step": 6376
    },
    {
      "epoch": 6.08062744140625e-06,
      "step": 6376,
      "training_step_time": 17.455825567245483
    },
    {
      "epoch": 6.081581115722656e-06,
      "model_forward_time": 5.60029673576355,
      "step": 6377
    },
    {
      "epoch": 6.081581115722656e-06,
      "step": 6377,
      "training_step_time": 17.08354616165161
    },
    {
      "epoch": 6.082534790039063e-06,
      "model_forward_time": 5.614316463470459,
      "step": 6378
    },
    {
      "epoch": 6.082534790039063e-06,
      "step": 6378,
      "training_step_time": 17.098921060562134
    },
    {
      "epoch": 6.083488464355468e-06,
      "model_forward_time": 5.61151385307312,
      "step": 6379
    },
    {
      "epoch": 6.083488464355468e-06,
      "step": 6379,
      "training_step_time": 17.09212827682495
    },
    {
      "epoch": 6.084442138671875e-06,
      "grad_norm": 0.4619568884372711,
      "learning_rate": 1e-05,
      "loss": 0.0652,
      "step": 6380
    },
    {
      "epoch": 6.084442138671875e-06,
      "model_forward_time": 5.60735821723938,
      "step": 6380
    },
    {
      "dynamics_loss_avg": 0.05562090333551169,
      "epoch": 6.084442138671875e-06,
      "step": 6380
    },
    {
      "action_loss_avg": 0.008259443938732148,
      "epoch": 6.084442138671875e-06,
      "step": 6380
    },
    {
      "epoch": 6.084442138671875e-06,
      "step": 6380,
      "training_step_time": 17.096859455108643
    },
    {
      "epoch": 6.0853958129882815e-06,
      "model_forward_time": 5.611083984375,
      "step": 6381
    },
    {
      "epoch": 6.0853958129882815e-06,
      "step": 6381,
      "training_step_time": 17.456820011138916
    },
    {
      "epoch": 6.086349487304687e-06,
      "model_forward_time": 5.608707427978516,
      "step": 6382
    },
    {
      "epoch": 6.086349487304687e-06,
      "step": 6382,
      "training_step_time": 17.10498356819153
    },
    {
      "epoch": 6.087303161621094e-06,
      "model_forward_time": 5.609893798828125,
      "step": 6383
    },
    {
      "epoch": 6.087303161621094e-06,
      "step": 6383,
      "training_step_time": 17.09602975845337
    },
    {
      "epoch": 6.0882568359375e-06,
      "model_forward_time": 5.628286361694336,
      "step": 6384
    },
    {
      "epoch": 6.0882568359375e-06,
      "step": 6384,
      "training_step_time": 17.12958598136902
    },
    {
      "epoch": 6.089210510253906e-06,
      "model_forward_time": 5.613313674926758,
      "step": 6385
    },
    {
      "epoch": 6.089210510253906e-06,
      "step": 6385,
      "training_step_time": 17.096848964691162
    },
    {
      "epoch": 6.090164184570313e-06,
      "model_forward_time": 5.603819847106934,
      "step": 6386
    },
    {
      "epoch": 6.090164184570313e-06,
      "step": 6386,
      "training_step_time": 17.062551021575928
    },
    {
      "epoch": 6.091117858886718e-06,
      "model_forward_time": 5.96691107749939,
      "step": 6387
    },
    {
      "epoch": 6.091117858886718e-06,
      "step": 6387,
      "training_step_time": 17.437129497528076
    },
    {
      "epoch": 6.092071533203125e-06,
      "model_forward_time": 5.606133460998535,
      "step": 6388
    },
    {
      "epoch": 6.092071533203125e-06,
      "step": 6388,
      "training_step_time": 17.074389696121216
    },
    {
      "epoch": 6.0930252075195315e-06,
      "model_forward_time": 5.595871686935425,
      "step": 6389
    },
    {
      "epoch": 6.0930252075195315e-06,
      "step": 6389,
      "training_step_time": 17.03989887237549
    },
    {
      "epoch": 6.093978881835937e-06,
      "grad_norm": 1.0391662120819092,
      "learning_rate": 1e-05,
      "loss": 0.0762,
      "step": 6390
    },
    {
      "epoch": 6.093978881835937e-06,
      "model_forward_time": 5.582593679428101,
      "step": 6390
    },
    {
      "dynamics_loss_avg": 0.06916364151984453,
      "epoch": 6.093978881835937e-06,
      "step": 6390
    },
    {
      "action_loss_avg": 0.0120933523401618,
      "epoch": 6.093978881835937e-06,
      "step": 6390
    },
    {
      "epoch": 6.093978881835937e-06,
      "step": 6390,
      "training_step_time": 17.040379285812378
    },
    {
      "epoch": 6.094932556152344e-06,
      "model_forward_time": 5.599909782409668,
      "step": 6391
    },
    {
      "epoch": 6.094932556152344e-06,
      "step": 6391,
      "training_step_time": 17.05949306488037
    },
    {
      "epoch": 6.09588623046875e-06,
      "model_forward_time": 5.949738264083862,
      "step": 6392
    },
    {
      "epoch": 6.09588623046875e-06,
      "step": 6392,
      "training_step_time": 17.41112184524536
    },
    {
      "epoch": 6.096839904785156e-06,
      "model_forward_time": 5.604368448257446,
      "step": 6393
    },
    {
      "epoch": 6.096839904785156e-06,
      "step": 6393,
      "training_step_time": 17.076135396957397
    },
    {
      "epoch": 6.097793579101563e-06,
      "model_forward_time": 5.597613573074341,
      "step": 6394
    },
    {
      "epoch": 6.097793579101563e-06,
      "step": 6394,
      "training_step_time": 17.07748007774353
    },
    {
      "epoch": 6.098747253417968e-06,
      "model_forward_time": 5.601144552230835,
      "step": 6395
    },
    {
      "epoch": 6.098747253417968e-06,
      "step": 6395,
      "training_step_time": 17.077219247817993
    },
    {
      "epoch": 6.099700927734375e-06,
      "model_forward_time": 5.620480298995972,
      "step": 6396
    },
    {
      "epoch": 6.099700927734375e-06,
      "step": 6396,
      "training_step_time": 17.11006760597229
    },
    {
      "epoch": 6.1006546020507815e-06,
      "model_forward_time": 5.635723352432251,
      "step": 6397
    },
    {
      "epoch": 6.1006546020507815e-06,
      "step": 6397,
      "training_step_time": 17.09845209121704
    },
    {
      "epoch": 6.101608276367187e-06,
      "model_forward_time": 5.957048416137695,
      "step": 6398
    },
    {
      "epoch": 6.101608276367187e-06,
      "step": 6398,
      "training_step_time": 17.382659912109375
    },
    {
      "epoch": 6.102561950683594e-06,
      "model_forward_time": 5.58211350440979,
      "step": 6399
    },
    {
      "epoch": 6.102561950683594e-06,
      "step": 6399,
      "training_step_time": 16.996545791625977
    },
    {
      "epoch": 6.103515625e-06,
      "grad_norm": 0.4426625669002533,
      "learning_rate": 1e-05,
      "loss": 0.063,
      "step": 6400
    },
    {
      "epoch": 6.103515625e-06,
      "model_forward_time": 5.5831544399261475,
      "step": 6400
    },
    {
      "dynamics_loss_avg": 0.053697933442890644,
      "epoch": 6.103515625e-06,
      "step": 6400
    },
    {
      "action_loss_avg": 0.012150470539927483,
      "epoch": 6.103515625e-06,
      "step": 6400
    },
    {
      "epoch": 6.103515625e-06,
      "step": 6400,
      "training_step_time": 17.000264406204224
    },
    {
      "epoch": 6.104469299316406e-06,
      "model_forward_time": 5.580324411392212,
      "step": 6401
    },
    {
      "epoch": 6.104469299316406e-06,
      "step": 6401,
      "training_step_time": 17.020644903182983
    },
    {
      "epoch": 6.105422973632813e-06,
      "model_forward_time": 5.577967643737793,
      "step": 6402
    },
    {
      "epoch": 6.105422973632813e-06,
      "step": 6402,
      "training_step_time": 17.02442193031311
    },
    {
      "epoch": 6.106376647949218e-06,
      "model_forward_time": 5.595457315444946,
      "step": 6403
    },
    {
      "epoch": 6.106376647949218e-06,
      "step": 6403,
      "training_step_time": 17.405163288116455
    },
    {
      "epoch": 6.107330322265625e-06,
      "model_forward_time": 5.603715658187866,
      "step": 6404
    },
    {
      "epoch": 6.107330322265625e-06,
      "step": 6404,
      "training_step_time": 17.097176790237427
    },
    {
      "epoch": 6.1082839965820315e-06,
      "model_forward_time": 5.620351791381836,
      "step": 6405
    },
    {
      "epoch": 6.1082839965820315e-06,
      "step": 6405,
      "training_step_time": 17.136611461639404
    },
    {
      "epoch": 6.109237670898437e-06,
      "model_forward_time": 5.615200996398926,
      "step": 6406
    },
    {
      "epoch": 6.109237670898437e-06,
      "step": 6406,
      "training_step_time": 17.116480588912964
    },
    {
      "epoch": 6.110191345214844e-06,
      "model_forward_time": 5.622387886047363,
      "step": 6407
    },
    {
      "epoch": 6.110191345214844e-06,
      "step": 6407,
      "training_step_time": 17.111525535583496
    },
    {
      "epoch": 6.11114501953125e-06,
      "model_forward_time": 5.600786209106445,
      "step": 6408
    },
    {
      "epoch": 6.11114501953125e-06,
      "step": 6408,
      "training_step_time": 17.076154470443726
    },
    {
      "epoch": 6.112098693847656e-06,
      "model_forward_time": 5.972177267074585,
      "step": 6409
    },
    {
      "epoch": 6.112098693847656e-06,
      "step": 6409,
      "training_step_time": 17.446689128875732
    },
    {
      "epoch": 6.113052368164063e-06,
      "grad_norm": 0.693899929523468,
      "learning_rate": 1e-05,
      "loss": 0.0628,
      "step": 6410
    },
    {
      "epoch": 6.113052368164063e-06,
      "model_forward_time": 5.5975706577301025,
      "step": 6410
    },
    {
      "dynamics_loss_avg": 0.04568128753453493,
      "epoch": 6.113052368164063e-06,
      "step": 6410
    },
    {
      "action_loss_avg": 0.00793396751396358,
      "epoch": 6.113052368164063e-06,
      "step": 6410
    },
    {
      "epoch": 6.113052368164063e-06,
      "step": 6410,
      "training_step_time": 17.05189323425293
    },
    {
      "epoch": 6.114006042480468e-06,
      "model_forward_time": 5.589695692062378,
      "step": 6411
    },
    {
      "epoch": 6.114006042480468e-06,
      "step": 6411,
      "training_step_time": 17.040266036987305
    },
    {
      "epoch": 6.114959716796875e-06,
      "model_forward_time": 5.597123384475708,
      "step": 6412
    },
    {
      "epoch": 6.114959716796875e-06,
      "step": 6412,
      "training_step_time": 17.02760076522827
    },
    {
      "epoch": 6.1159133911132815e-06,
      "model_forward_time": 5.586108684539795,
      "step": 6413
    },
    {
      "epoch": 6.1159133911132815e-06,
      "step": 6413,
      "training_step_time": 17.038710832595825
    },
    {
      "epoch": 6.116867065429687e-06,
      "model_forward_time": 5.599363327026367,
      "step": 6414
    },
    {
      "epoch": 6.116867065429687e-06,
      "step": 6414,
      "training_step_time": 17.412229776382446
    },
    {
      "epoch": 6.117820739746094e-06,
      "model_forward_time": 5.602339744567871,
      "step": 6415
    },
    {
      "epoch": 6.117820739746094e-06,
      "step": 6415,
      "training_step_time": 17.078274488449097
    },
    {
      "epoch": 6.1187744140625e-06,
      "model_forward_time": 5.598871231079102,
      "step": 6416
    },
    {
      "epoch": 6.1187744140625e-06,
      "step": 6416,
      "training_step_time": 17.045163869857788
    },
    {
      "epoch": 6.119728088378906e-06,
      "model_forward_time": 5.594935655593872,
      "step": 6417
    },
    {
      "epoch": 6.119728088378906e-06,
      "step": 6417,
      "training_step_time": 17.055888414382935
    },
    {
      "epoch": 6.120681762695313e-06,
      "model_forward_time": 5.605083465576172,
      "step": 6418
    },
    {
      "epoch": 6.120681762695313e-06,
      "step": 6418,
      "training_step_time": 17.094154357910156
    },
    {
      "epoch": 6.121635437011718e-06,
      "model_forward_time": 5.609441757202148,
      "step": 6419
    },
    {
      "epoch": 6.121635437011718e-06,
      "step": 6419,
      "training_step_time": 17.10081696510315
    },
    {
      "epoch": 6.122589111328125e-06,
      "grad_norm": 0.8659738302230835,
      "learning_rate": 1e-05,
      "loss": 0.08,
      "step": 6420
    },
    {
      "epoch": 6.122589111328125e-06,
      "model_forward_time": 5.979691743850708,
      "step": 6420
    },
    {
      "dynamics_loss_avg": 0.07249966748058796,
      "epoch": 6.122589111328125e-06,
      "step": 6420
    },
    {
      "action_loss_avg": 0.015051889512687922,
      "epoch": 6.122589111328125e-06,
      "step": 6420
    },
    {
      "epoch": 6.122589111328125e-06,
      "step": 6420,
      "training_step_time": 17.44010901451111
    },
    {
      "epoch": 6.1235427856445315e-06,
      "model_forward_time": 5.583158731460571,
      "step": 6421
    },
    {
      "epoch": 6.1235427856445315e-06,
      "step": 6421,
      "training_step_time": 16.992499589920044
    },
    {
      "epoch": 6.124496459960937e-06,
      "model_forward_time": 5.573904514312744,
      "step": 6422
    },
    {
      "epoch": 6.124496459960937e-06,
      "step": 6422,
      "training_step_time": 16.989686012268066
    },
    {
      "epoch": 6.125450134277344e-06,
      "model_forward_time": 5.589260101318359,
      "step": 6423
    },
    {
      "epoch": 6.125450134277344e-06,
      "step": 6423,
      "training_step_time": 16.997928380966187
    },
    {
      "epoch": 6.12640380859375e-06,
      "model_forward_time": 5.557966232299805,
      "step": 6424
    },
    {
      "epoch": 6.12640380859375e-06,
      "step": 6424,
      "training_step_time": 16.98088574409485
    },
    {
      "epoch": 6.127357482910156e-06,
      "model_forward_time": 5.916656970977783,
      "step": 6425
    },
    {
      "epoch": 6.127357482910156e-06,
      "step": 6425,
      "training_step_time": 17.32620358467102
    },
    {
      "epoch": 6.128311157226563e-06,
      "model_forward_time": 5.571781635284424,
      "step": 6426
    },
    {
      "epoch": 6.128311157226563e-06,
      "step": 6426,
      "training_step_time": 16.992056846618652
    },
    {
      "epoch": 6.129264831542968e-06,
      "model_forward_time": 5.573351860046387,
      "step": 6427
    },
    {
      "epoch": 6.129264831542968e-06,
      "step": 6427,
      "training_step_time": 17.00590705871582
    },
    {
      "epoch": 6.130218505859375e-06,
      "model_forward_time": 5.581085681915283,
      "step": 6428
    },
    {
      "epoch": 6.130218505859375e-06,
      "step": 6428,
      "training_step_time": 17.045241832733154
    },
    {
      "epoch": 6.1311721801757815e-06,
      "model_forward_time": 5.603012800216675,
      "step": 6429
    },
    {
      "epoch": 6.1311721801757815e-06,
      "step": 6429,
      "training_step_time": 17.084222078323364
    },
    {
      "epoch": 6.132125854492187e-06,
      "grad_norm": 0.4546063542366028,
      "learning_rate": 1e-05,
      "loss": 0.0609,
      "step": 6430
    },
    {
      "epoch": 6.132125854492187e-06,
      "model_forward_time": 5.609604597091675,
      "step": 6430
    },
    {
      "dynamics_loss_avg": 0.04487419128417969,
      "epoch": 6.132125854492187e-06,
      "step": 6430
    },
    {
      "action_loss_avg": 0.007863501389510929,
      "epoch": 6.132125854492187e-06,
      "step": 6430
    },
    {
      "epoch": 6.132125854492187e-06,
      "step": 6430,
      "training_step_time": 17.108031272888184
    },
    {
      "epoch": 6.133079528808594e-06,
      "model_forward_time": 5.982029676437378,
      "step": 6431
    },
    {
      "epoch": 6.133079528808594e-06,
      "step": 6431,
      "training_step_time": 17.482306480407715
    },
    {
      "epoch": 6.134033203125e-06,
      "model_forward_time": 5.628010511398315,
      "step": 6432
    },
    {
      "epoch": 6.134033203125e-06,
      "step": 6432,
      "training_step_time": 17.137661695480347
    },
    {
      "epoch": 6.134986877441406e-06,
      "model_forward_time": 5.6176581382751465,
      "step": 6433
    },
    {
      "epoch": 6.134986877441406e-06,
      "step": 6433,
      "training_step_time": 17.102558135986328
    },
    {
      "epoch": 6.135940551757813e-06,
      "model_forward_time": 5.599721431732178,
      "step": 6434
    },
    {
      "epoch": 6.135940551757813e-06,
      "step": 6434,
      "training_step_time": 17.056283473968506
    },
    {
      "epoch": 6.136894226074218e-06,
      "model_forward_time": 5.595974683761597,
      "step": 6435
    },
    {
      "epoch": 6.136894226074218e-06,
      "step": 6435,
      "training_step_time": 17.034945964813232
    },
    {
      "epoch": 6.137847900390625e-06,
      "model_forward_time": 5.5801613330841064,
      "step": 6436
    },
    {
      "epoch": 6.137847900390625e-06,
      "step": 6436,
      "training_step_time": 17.375481605529785
    },
    {
      "epoch": 6.1388015747070315e-06,
      "model_forward_time": 5.584008693695068,
      "step": 6437
    },
    {
      "epoch": 6.1388015747070315e-06,
      "step": 6437,
      "training_step_time": 17.019675970077515
    },
    {
      "epoch": 6.139755249023437e-06,
      "model_forward_time": 5.594851493835449,
      "step": 6438
    },
    {
      "epoch": 6.139755249023437e-06,
      "step": 6438,
      "training_step_time": 17.06955862045288
    },
    {
      "epoch": 6.140708923339844e-06,
      "model_forward_time": 5.610596656799316,
      "step": 6439
    },
    {
      "epoch": 6.140708923339844e-06,
      "step": 6439,
      "training_step_time": 17.093302726745605
    },
    {
      "epoch": 6.14166259765625e-06,
      "grad_norm": 0.7843101620674133,
      "learning_rate": 1e-05,
      "loss": 0.0609,
      "step": 6440
    },
    {
      "epoch": 6.14166259765625e-06,
      "model_forward_time": 5.646404266357422,
      "step": 6440
    },
    {
      "dynamics_loss_avg": 0.05533306356519461,
      "epoch": 6.14166259765625e-06,
      "step": 6440
    },
    {
      "action_loss_avg": 0.008566322503611445,
      "epoch": 6.14166259765625e-06,
      "step": 6440
    },
    {
      "epoch": 6.14166259765625e-06,
      "step": 6440,
      "training_step_time": 17.125081300735474
    },
    {
      "epoch": 6.142616271972656e-06,
      "model_forward_time": 5.596308708190918,
      "step": 6441
    },
    {
      "epoch": 6.142616271972656e-06,
      "step": 6441,
      "training_step_time": 17.024255990982056
    },
    {
      "epoch": 6.143569946289063e-06,
      "model_forward_time": 5.934519529342651,
      "step": 6442
    },
    {
      "epoch": 6.143569946289063e-06,
      "step": 6442,
      "training_step_time": 17.34098482131958
    },
    {
      "epoch": 6.144523620605468e-06,
      "model_forward_time": 5.578430891036987,
      "step": 6443
    },
    {
      "epoch": 6.144523620605468e-06,
      "step": 6443,
      "training_step_time": 17.01358675956726
    },
    {
      "epoch": 6.145477294921875e-06,
      "model_forward_time": 5.581671953201294,
      "step": 6444
    },
    {
      "epoch": 6.145477294921875e-06,
      "step": 6444,
      "training_step_time": 16.987876415252686
    },
    {
      "epoch": 6.1464309692382815e-06,
      "model_forward_time": 5.580736875534058,
      "step": 6445
    },
    {
      "epoch": 6.1464309692382815e-06,
      "step": 6445,
      "training_step_time": 17.014490365982056
    },
    {
      "epoch": 6.147384643554687e-06,
      "model_forward_time": 5.59815239906311,
      "step": 6446
    },
    {
      "epoch": 6.147384643554687e-06,
      "step": 6446,
      "training_step_time": 17.098487615585327
    },
    {
      "epoch": 6.148338317871094e-06,
      "model_forward_time": 5.617267608642578,
      "step": 6447
    },
    {
      "epoch": 6.148338317871094e-06,
      "step": 6447,
      "training_step_time": 17.4902982711792
    },
    {
      "epoch": 6.1492919921875e-06,
      "model_forward_time": 5.615880250930786,
      "step": 6448
    },
    {
      "epoch": 6.1492919921875e-06,
      "step": 6448,
      "training_step_time": 17.101441621780396
    },
    {
      "epoch": 6.150245666503906e-06,
      "model_forward_time": 5.6032304763793945,
      "step": 6449
    },
    {
      "epoch": 6.150245666503906e-06,
      "step": 6449,
      "training_step_time": 17.094377279281616
    },
    {
      "epoch": 6.151199340820313e-06,
      "grad_norm": 0.6422467231750488,
      "learning_rate": 1e-05,
      "loss": 0.0699,
      "step": 6450
    },
    {
      "epoch": 6.151199340820313e-06,
      "model_forward_time": 5.599336385726929,
      "step": 6450
    },
    {
      "dynamics_loss_avg": 0.06218014508485794,
      "epoch": 6.151199340820313e-06,
      "step": 6450
    },
    {
      "action_loss_avg": 0.012493759347125887,
      "epoch": 6.151199340820313e-06,
      "step": 6450
    },
    {
      "epoch": 6.151199340820313e-06,
      "step": 6450,
      "training_step_time": 17.055715322494507
    },
    {
      "epoch": 6.152153015136718e-06,
      "model_forward_time": 5.6004345417022705,
      "step": 6451
    },
    {
      "epoch": 6.152153015136718e-06,
      "step": 6451,
      "training_step_time": 17.058732271194458
    },
    {
      "epoch": 6.153106689453125e-06,
      "model_forward_time": 5.589912176132202,
      "step": 6452
    },
    {
      "epoch": 6.153106689453125e-06,
      "step": 6452,
      "training_step_time": 17.02583932876587
    },
    {
      "epoch": 6.1540603637695315e-06,
      "model_forward_time": 5.957695245742798,
      "step": 6453
    },
    {
      "epoch": 6.1540603637695315e-06,
      "step": 6453,
      "training_step_time": 17.392300844192505
    },
    {
      "epoch": 6.155014038085937e-06,
      "model_forward_time": 5.597718715667725,
      "step": 6454
    },
    {
      "epoch": 6.155014038085937e-06,
      "step": 6454,
      "training_step_time": 17.08128809928894
    },
    {
      "epoch": 6.155967712402344e-06,
      "model_forward_time": 5.6223132610321045,
      "step": 6455
    },
    {
      "epoch": 6.155967712402344e-06,
      "step": 6455,
      "training_step_time": 17.13775086402893
    },
    {
      "epoch": 6.15692138671875e-06,
      "model_forward_time": 5.6145288944244385,
      "step": 6456
    },
    {
      "epoch": 6.15692138671875e-06,
      "step": 6456,
      "training_step_time": 17.116791486740112
    },
    {
      "epoch": 6.157875061035156e-06,
      "model_forward_time": 5.613236904144287,
      "step": 6457
    },
    {
      "epoch": 6.157875061035156e-06,
      "step": 6457,
      "training_step_time": 17.09184980392456
    },
    {
      "epoch": 6.158828735351563e-06,
      "model_forward_time": 5.965380430221558,
      "step": 6458
    },
    {
      "epoch": 6.158828735351563e-06,
      "step": 6458,
      "training_step_time": 17.435584783554077
    },
    {
      "epoch": 6.159782409667968e-06,
      "model_forward_time": 5.61359429359436,
      "step": 6459
    },
    {
      "epoch": 6.159782409667968e-06,
      "step": 6459,
      "training_step_time": 17.10162377357483
    },
    {
      "epoch": 6.160736083984375e-06,
      "grad_norm": 1.3796113729476929,
      "learning_rate": 1e-05,
      "loss": 0.0855,
      "step": 6460
    },
    {
      "epoch": 6.160736083984375e-06,
      "model_forward_time": 5.608935356140137,
      "step": 6460
    },
    {
      "dynamics_loss_avg": 0.06635295655578374,
      "epoch": 6.160736083984375e-06,
      "step": 6460
    },
    {
      "action_loss_avg": 0.012985803443007172,
      "epoch": 6.160736083984375e-06,
      "step": 6460
    },
    {
      "epoch": 6.160736083984375e-06,
      "step": 6460,
      "training_step_time": 17.107692003250122
    },
    {
      "epoch": 6.1616897583007815e-06,
      "model_forward_time": 5.611611366271973,
      "step": 6461
    },
    {
      "epoch": 6.1616897583007815e-06,
      "step": 6461,
      "training_step_time": 17.112918853759766
    },
    {
      "epoch": 6.162643432617187e-06,
      "model_forward_time": 5.612866640090942,
      "step": 6462
    },
    {
      "epoch": 6.162643432617187e-06,
      "step": 6462,
      "training_step_time": 17.12158966064453
    },
    {
      "epoch": 6.163597106933594e-06,
      "model_forward_time": 5.6114184856414795,
      "step": 6463
    },
    {
      "epoch": 6.163597106933594e-06,
      "step": 6463,
      "training_step_time": 17.124592065811157
    },
    {
      "epoch": 6.16455078125e-06,
      "model_forward_time": 5.976495742797852,
      "step": 6464
    },
    {
      "epoch": 6.16455078125e-06,
      "step": 6464,
      "training_step_time": 17.475626230239868
    },
    {
      "epoch": 6.165504455566406e-06,
      "model_forward_time": 5.609237194061279,
      "step": 6465
    },
    {
      "epoch": 6.165504455566406e-06,
      "step": 6465,
      "training_step_time": 17.063491344451904
    },
    {
      "epoch": 6.166458129882813e-06,
      "model_forward_time": 5.5942771434783936,
      "step": 6466
    },
    {
      "epoch": 6.166458129882813e-06,
      "step": 6466,
      "training_step_time": 17.030219316482544
    },
    {
      "epoch": 6.167411804199218e-06,
      "model_forward_time": 5.581066370010376,
      "step": 6467
    },
    {
      "epoch": 6.167411804199218e-06,
      "step": 6467,
      "training_step_time": 17.003387689590454
    },
    {
      "epoch": 6.168365478515625e-06,
      "model_forward_time": 5.573317289352417,
      "step": 6468
    },
    {
      "epoch": 6.168365478515625e-06,
      "step": 6468,
      "training_step_time": 17.00079035758972
    },
    {
      "epoch": 6.1693191528320315e-06,
      "model_forward_time": 5.595251798629761,
      "step": 6469
    },
    {
      "epoch": 6.1693191528320315e-06,
      "step": 6469,
      "training_step_time": 17.400736093521118
    },
    {
      "epoch": 6.170272827148437e-06,
      "grad_norm": 2.1690454483032227,
      "learning_rate": 1e-05,
      "loss": 0.0646,
      "step": 6470
    },
    {
      "epoch": 6.170272827148437e-06,
      "model_forward_time": 5.592530965805054,
      "step": 6470
    },
    {
      "dynamics_loss_avg": 0.0530655262991786,
      "epoch": 6.170272827148437e-06,
      "step": 6470
    },
    {
      "action_loss_avg": 0.01371398768387735,
      "epoch": 6.170272827148437e-06,
      "step": 6470
    },
    {
      "epoch": 6.170272827148437e-06,
      "step": 6470,
      "training_step_time": 17.055190801620483
    },
    {
      "epoch": 6.171226501464844e-06,
      "model_forward_time": 5.623982667922974,
      "step": 6471
    },
    {
      "epoch": 6.171226501464844e-06,
      "step": 6471,
      "training_step_time": 17.134515285491943
    },
    {
      "epoch": 6.17218017578125e-06,
      "model_forward_time": 5.620058059692383,
      "step": 6472
    },
    {
      "epoch": 6.17218017578125e-06,
      "step": 6472,
      "training_step_time": 17.118473291397095
    },
    {
      "epoch": 6.173133850097656e-06,
      "model_forward_time": 5.617745876312256,
      "step": 6473
    },
    {
      "epoch": 6.173133850097656e-06,
      "step": 6473,
      "training_step_time": 17.090704441070557
    },
    {
      "epoch": 6.174087524414063e-06,
      "model_forward_time": 5.600653648376465,
      "step": 6474
    },
    {
      "epoch": 6.174087524414063e-06,
      "step": 6474,
      "training_step_time": 17.085405588150024
    },
    {
      "epoch": 6.175041198730468e-06,
      "model_forward_time": 5.982656002044678,
      "step": 6475
    },
    {
      "epoch": 6.175041198730468e-06,
      "step": 6475,
      "training_step_time": 17.46096134185791
    },
    {
      "epoch": 6.175994873046875e-06,
      "model_forward_time": 5.603574752807617,
      "step": 6476
    },
    {
      "epoch": 6.175994873046875e-06,
      "step": 6476,
      "training_step_time": 17.074681758880615
    },
    {
      "epoch": 6.1769485473632815e-06,
      "model_forward_time": 5.606748342514038,
      "step": 6477
    },
    {
      "epoch": 6.1769485473632815e-06,
      "step": 6477,
      "training_step_time": 17.07505202293396
    },
    {
      "epoch": 6.177902221679687e-06,
      "model_forward_time": 5.596583366394043,
      "step": 6478
    },
    {
      "epoch": 6.177902221679687e-06,
      "step": 6478,
      "training_step_time": 17.063743829727173
    },
    {
      "epoch": 6.178855895996094e-06,
      "model_forward_time": 5.604658842086792,
      "step": 6479
    },
    {
      "epoch": 6.178855895996094e-06,
      "step": 6479,
      "training_step_time": 17.079678773880005
    },
    {
      "epoch": 6.1798095703125e-06,
      "grad_norm": 0.9703332185745239,
      "learning_rate": 1e-05,
      "loss": 0.0781,
      "step": 6480
    },
    {
      "epoch": 6.1798095703125e-06,
      "model_forward_time": 5.958416938781738,
      "step": 6480
    },
    {
      "dynamics_loss_avg": 0.06657586805522442,
      "epoch": 6.1798095703125e-06,
      "step": 6480
    },
    {
      "action_loss_avg": 0.014467142708599567,
      "epoch": 6.1798095703125e-06,
      "step": 6480
    },
    {
      "epoch": 6.1798095703125e-06,
      "step": 6480,
      "training_step_time": 17.461198568344116
    },
    {
      "epoch": 6.180763244628906e-06,
      "model_forward_time": 5.619262933731079,
      "step": 6481
    },
    {
      "epoch": 6.180763244628906e-06,
      "step": 6481,
      "training_step_time": 17.122808694839478
    },
    {
      "epoch": 6.181716918945313e-06,
      "model_forward_time": 5.609743356704712,
      "step": 6482
    },
    {
      "epoch": 6.181716918945313e-06,
      "step": 6482,
      "training_step_time": 17.126005172729492
    },
    {
      "epoch": 6.182670593261718e-06,
      "model_forward_time": 5.631301164627075,
      "step": 6483
    },
    {
      "epoch": 6.182670593261718e-06,
      "step": 6483,
      "training_step_time": 17.13300657272339
    },
    {
      "epoch": 6.183624267578125e-06,
      "model_forward_time": 5.6121978759765625,
      "step": 6484
    },
    {
      "epoch": 6.183624267578125e-06,
      "step": 6484,
      "training_step_time": 17.090599536895752
    },
    {
      "epoch": 6.1845779418945316e-06,
      "model_forward_time": 5.587242603302002,
      "step": 6485
    },
    {
      "epoch": 6.1845779418945316e-06,
      "step": 6485,
      "training_step_time": 17.031004667282104
    },
    {
      "epoch": 6.185531616210937e-06,
      "model_forward_time": 5.939700603485107,
      "step": 6486
    },
    {
      "epoch": 6.185531616210937e-06,
      "step": 6486,
      "training_step_time": 17.338704586029053
    },
    {
      "epoch": 6.186485290527344e-06,
      "model_forward_time": 5.58375883102417,
      "step": 6487
    },
    {
      "epoch": 6.186485290527344e-06,
      "step": 6487,
      "training_step_time": 17.018221855163574
    },
    {
      "epoch": 6.18743896484375e-06,
      "model_forward_time": 5.593543767929077,
      "step": 6488
    },
    {
      "epoch": 6.18743896484375e-06,
      "step": 6488,
      "training_step_time": 17.04669427871704
    },
    {
      "epoch": 6.188392639160156e-06,
      "model_forward_time": 5.608797073364258,
      "step": 6489
    },
    {
      "epoch": 6.188392639160156e-06,
      "step": 6489,
      "training_step_time": 17.09849739074707
    },
    {
      "epoch": 6.189346313476563e-06,
      "grad_norm": 0.9366269707679749,
      "learning_rate": 1e-05,
      "loss": 0.0746,
      "step": 6490
    },
    {
      "epoch": 6.189346313476563e-06,
      "model_forward_time": 5.598181962966919,
      "step": 6490
    },
    {
      "dynamics_loss_avg": 0.062234968319535254,
      "epoch": 6.189346313476563e-06,
      "step": 6490
    },
    {
      "action_loss_avg": 0.010206849733367562,
      "epoch": 6.189346313476563e-06,
      "step": 6490
    },
    {
      "epoch": 6.189346313476563e-06,
      "step": 6490,
      "training_step_time": 17.091398239135742
    },
    {
      "epoch": 6.1902999877929684e-06,
      "model_forward_time": 5.958669662475586,
      "step": 6491
    },
    {
      "epoch": 6.1902999877929684e-06,
      "step": 6491,
      "training_step_time": 17.444573163986206
    },
    {
      "epoch": 6.191253662109375e-06,
      "model_forward_time": 5.595152854919434,
      "step": 6492
    },
    {
      "epoch": 6.191253662109375e-06,
      "step": 6492,
      "training_step_time": 17.027442932128906
    },
    {
      "epoch": 6.1922073364257816e-06,
      "model_forward_time": 5.587682008743286,
      "step": 6493
    },
    {
      "epoch": 6.1922073364257816e-06,
      "step": 6493,
      "training_step_time": 17.034583806991577
    },
    {
      "epoch": 6.193161010742187e-06,
      "model_forward_time": 5.599571466445923,
      "step": 6494
    },
    {
      "epoch": 6.193161010742187e-06,
      "step": 6494,
      "training_step_time": 17.07402229309082
    },
    {
      "epoch": 6.194114685058594e-06,
      "model_forward_time": 5.599690675735474,
      "step": 6495
    },
    {
      "epoch": 6.194114685058594e-06,
      "step": 6495,
      "training_step_time": 17.045075178146362
    },
    {
      "epoch": 6.195068359375e-06,
      "model_forward_time": 5.58763313293457,
      "step": 6496
    },
    {
      "epoch": 6.195068359375e-06,
      "step": 6496,
      "training_step_time": 17.411813735961914
    },
    {
      "epoch": 6.196022033691406e-06,
      "model_forward_time": 5.592105865478516,
      "step": 6497
    },
    {
      "epoch": 6.196022033691406e-06,
      "step": 6497,
      "training_step_time": 17.022209644317627
    },
    {
      "epoch": 6.196975708007813e-06,
      "model_forward_time": 5.5977582931518555,
      "step": 6498
    },
    {
      "epoch": 6.196975708007813e-06,
      "step": 6498,
      "training_step_time": 17.085018634796143
    },
    {
      "epoch": 6.1979293823242184e-06,
      "model_forward_time": 5.611622095108032,
      "step": 6499
    },
    {
      "epoch": 6.1979293823242184e-06,
      "step": 6499,
      "training_step_time": 17.132399320602417
    },
    {
      "epoch": 6.198883056640625e-06,
      "grad_norm": 1.5803133249282837,
      "learning_rate": 1e-05,
      "loss": 0.0781,
      "step": 6500
    },
    {
      "epoch": 6.198883056640625e-06,
      "model_forward_time": 5.620293617248535,
      "step": 6500
    },
    {
      "dynamics_loss_avg": 0.06349794194102287,
      "epoch": 6.198883056640625e-06,
      "step": 6500
    },
    {
      "action_loss_avg": 0.009673458058387041,
      "epoch": 6.198883056640625e-06,
      "step": 6500
    },
    {
      "epoch": 6.198883056640625e-06,
      "step": 6500,
      "training_step_time": 17.123855590820312
    },
    {
      "epoch": 6.1998367309570316e-06,
      "model_forward_time": 5.618786573410034,
      "step": 6501
    },
    {
      "epoch": 6.1998367309570316e-06,
      "step": 6501,
      "training_step_time": 17.087859869003296
    },
    {
      "epoch": 6.200790405273437e-06,
      "model_forward_time": 5.942856311798096,
      "step": 6502
    },
    {
      "epoch": 6.200790405273437e-06,
      "step": 6502,
      "training_step_time": 17.351195335388184
    },
    {
      "epoch": 6.201744079589844e-06,
      "model_forward_time": 5.566573619842529,
      "step": 6503
    },
    {
      "epoch": 6.201744079589844e-06,
      "step": 6503,
      "training_step_time": 16.982846975326538
    },
    {
      "epoch": 6.20269775390625e-06,
      "model_forward_time": 5.558104038238525,
      "step": 6504
    },
    {
      "epoch": 6.20269775390625e-06,
      "step": 6504,
      "training_step_time": 17.002734661102295
    },
    {
      "epoch": 6.203651428222656e-06,
      "model_forward_time": 5.565532445907593,
      "step": 6505
    },
    {
      "epoch": 6.203651428222656e-06,
      "step": 6505,
      "training_step_time": 16.98085117340088
    },
    {
      "epoch": 6.204605102539063e-06,
      "model_forward_time": 5.5664801597595215,
      "step": 6506
    },
    {
      "epoch": 6.204605102539063e-06,
      "step": 6506,
      "training_step_time": 16.97718572616577
    },
    {
      "epoch": 6.2055587768554684e-06,
      "model_forward_time": 5.56755256652832,
      "step": 6507
    },
    {
      "epoch": 6.2055587768554684e-06,
      "step": 6507,
      "training_step_time": 17.364449739456177
    },
    {
      "epoch": 6.206512451171875e-06,
      "model_forward_time": 5.5987303256988525,
      "step": 6508
    },
    {
      "epoch": 6.206512451171875e-06,
      "step": 6508,
      "training_step_time": 17.0877947807312
    },
    {
      "epoch": 6.2074661254882816e-06,
      "model_forward_time": 5.627747297286987,
      "step": 6509
    },
    {
      "epoch": 6.2074661254882816e-06,
      "step": 6509,
      "training_step_time": 17.133708000183105
    },
    {
      "epoch": 6.208419799804687e-06,
      "grad_norm": 1.8621236085891724,
      "learning_rate": 1e-05,
      "loss": 0.0788,
      "step": 6510
    },
    {
      "epoch": 6.208419799804687e-06,
      "model_forward_time": 5.606565237045288,
      "step": 6510
    },
    {
      "dynamics_loss_avg": 0.0674124525859952,
      "epoch": 6.208419799804687e-06,
      "step": 6510
    },
    {
      "action_loss_avg": 0.019622982293367387,
      "epoch": 6.208419799804687e-06,
      "step": 6510
    },
    {
      "epoch": 6.208419799804687e-06,
      "step": 6510,
      "training_step_time": 17.074290990829468
    },
    {
      "epoch": 6.209373474121094e-06,
      "model_forward_time": 5.597816705703735,
      "step": 6511
    },
    {
      "epoch": 6.209373474121094e-06,
      "step": 6511,
      "training_step_time": 17.045596837997437
    },
    {
      "epoch": 6.2103271484375e-06,
      "model_forward_time": 5.590591669082642,
      "step": 6512
    },
    {
      "epoch": 6.2103271484375e-06,
      "step": 6512,
      "training_step_time": 17.04334044456482
    },
    {
      "epoch": 6.211280822753906e-06,
      "model_forward_time": 5.931105136871338,
      "step": 6513
    },
    {
      "epoch": 6.211280822753906e-06,
      "step": 6513,
      "training_step_time": 17.369488954544067
    },
    {
      "epoch": 6.212234497070313e-06,
      "model_forward_time": 5.593863487243652,
      "step": 6514
    },
    {
      "epoch": 6.212234497070313e-06,
      "step": 6514,
      "training_step_time": 17.071484088897705
    },
    {
      "epoch": 6.2131881713867184e-06,
      "model_forward_time": 5.613004684448242,
      "step": 6515
    },
    {
      "epoch": 6.2131881713867184e-06,
      "step": 6515,
      "training_step_time": 17.11222767829895
    },
    {
      "epoch": 6.214141845703125e-06,
      "model_forward_time": 5.627128839492798,
      "step": 6516
    },
    {
      "epoch": 6.214141845703125e-06,
      "step": 6516,
      "training_step_time": 17.134190797805786
    },
    {
      "epoch": 6.215095520019532e-06,
      "model_forward_time": 5.61543869972229,
      "step": 6517
    },
    {
      "epoch": 6.215095520019532e-06,
      "step": 6517,
      "training_step_time": 17.084149837493896
    },
    {
      "epoch": 6.216049194335937e-06,
      "model_forward_time": 5.962074518203735,
      "step": 6518
    },
    {
      "epoch": 6.216049194335937e-06,
      "step": 6518,
      "training_step_time": 17.3806471824646
    },
    {
      "epoch": 6.217002868652344e-06,
      "model_forward_time": 5.57365870475769,
      "step": 6519
    },
    {
      "epoch": 6.217002868652344e-06,
      "step": 6519,
      "training_step_time": 17.01436424255371
    },
    {
      "epoch": 6.21795654296875e-06,
      "grad_norm": 0.9970895648002625,
      "learning_rate": 1e-05,
      "loss": 0.0913,
      "step": 6520
    },
    {
      "epoch": 6.21795654296875e-06,
      "model_forward_time": 5.570287704467773,
      "step": 6520
    },
    {
      "dynamics_loss_avg": 0.07384983394294978,
      "epoch": 6.21795654296875e-06,
      "step": 6520
    },
    {
      "action_loss_avg": 0.014583888091146946,
      "epoch": 6.21795654296875e-06,
      "step": 6520
    },
    {
      "epoch": 6.21795654296875e-06,
      "step": 6520,
      "training_step_time": 16.98854899406433
    },
    {
      "epoch": 6.218910217285156e-06,
      "model_forward_time": 5.566948413848877,
      "step": 6521
    },
    {
      "epoch": 6.218910217285156e-06,
      "step": 6521,
      "training_step_time": 16.980505228042603
    },
    {
      "epoch": 6.219863891601563e-06,
      "model_forward_time": 5.571454048156738,
      "step": 6522
    },
    {
      "epoch": 6.219863891601563e-06,
      "step": 6522,
      "training_step_time": 16.98568844795227
    },
    {
      "epoch": 6.2208175659179685e-06,
      "model_forward_time": 5.579730749130249,
      "step": 6523
    },
    {
      "epoch": 6.2208175659179685e-06,
      "step": 6523,
      "training_step_time": 17.37535047531128
    },
    {
      "epoch": 6.221771240234375e-06,
      "model_forward_time": 5.5860676765441895,
      "step": 6524
    },
    {
      "epoch": 6.221771240234375e-06,
      "step": 6524,
      "training_step_time": 17.031099319458008
    },
    {
      "epoch": 6.222724914550782e-06,
      "model_forward_time": 5.59950590133667,
      "step": 6525
    },
    {
      "epoch": 6.222724914550782e-06,
      "step": 6525,
      "training_step_time": 17.077289581298828
    },
    {
      "epoch": 6.223678588867187e-06,
      "model_forward_time": 5.618106842041016,
      "step": 6526
    },
    {
      "epoch": 6.223678588867187e-06,
      "step": 6526,
      "training_step_time": 17.095936059951782
    },
    {
      "epoch": 6.224632263183594e-06,
      "model_forward_time": 5.602493762969971,
      "step": 6527
    },
    {
      "epoch": 6.224632263183594e-06,
      "step": 6527,
      "training_step_time": 17.080941438674927
    },
    {
      "epoch": 6.2255859375e-06,
      "model_forward_time": 5.600653886795044,
      "step": 6528
    },
    {
      "epoch": 6.2255859375e-06,
      "step": 6528,
      "training_step_time": 17.07183814048767
    },
    {
      "epoch": 6.226539611816406e-06,
      "model_forward_time": 5.96551513671875,
      "step": 6529
    },
    {
      "epoch": 6.226539611816406e-06,
      "step": 6529,
      "training_step_time": 17.417733669281006
    },
    {
      "epoch": 6.227493286132813e-06,
      "grad_norm": 1.344803810119629,
      "learning_rate": 1e-05,
      "loss": 0.0866,
      "step": 6530
    },
    {
      "epoch": 6.227493286132813e-06,
      "model_forward_time": 5.60033106803894,
      "step": 6530
    },
    {
      "dynamics_loss_avg": 0.07026465274393559,
      "epoch": 6.227493286132813e-06,
      "step": 6530
    },
    {
      "action_loss_avg": 0.015728848055005074,
      "epoch": 6.227493286132813e-06,
      "step": 6530
    },
    {
      "epoch": 6.227493286132813e-06,
      "step": 6530,
      "training_step_time": 17.03537917137146
    },
    {
      "epoch": 6.2284469604492185e-06,
      "model_forward_time": 5.579206228256226,
      "step": 6531
    },
    {
      "epoch": 6.2284469604492185e-06,
      "step": 6531,
      "training_step_time": 16.988167762756348
    },
    {
      "epoch": 6.229400634765625e-06,
      "model_forward_time": 5.579590320587158,
      "step": 6532
    },
    {
      "epoch": 6.229400634765625e-06,
      "step": 6532,
      "training_step_time": 16.996437788009644
    },
    {
      "epoch": 6.230354309082032e-06,
      "model_forward_time": 5.595757961273193,
      "step": 6533
    },
    {
      "epoch": 6.230354309082032e-06,
      "step": 6533,
      "training_step_time": 17.07084822654724
    },
    {
      "epoch": 6.231307983398437e-06,
      "model_forward_time": 5.96186637878418,
      "step": 6534
    },
    {
      "epoch": 6.231307983398437e-06,
      "step": 6534,
      "training_step_time": 17.44728112220764
    },
    {
      "epoch": 6.232261657714844e-06,
      "model_forward_time": 5.615312814712524,
      "step": 6535
    },
    {
      "epoch": 6.232261657714844e-06,
      "step": 6535,
      "training_step_time": 17.121477842330933
    },
    {
      "epoch": 6.23321533203125e-06,
      "model_forward_time": 5.619809627532959,
      "step": 6536
    },
    {
      "epoch": 6.23321533203125e-06,
      "step": 6536,
      "training_step_time": 17.122723817825317
    },
    {
      "epoch": 6.234169006347656e-06,
      "model_forward_time": 5.616684436798096,
      "step": 6537
    },
    {
      "epoch": 6.234169006347656e-06,
      "step": 6537,
      "training_step_time": 17.11411762237549
    },
    {
      "epoch": 6.235122680664063e-06,
      "model_forward_time": 5.624653577804565,
      "step": 6538
    },
    {
      "epoch": 6.235122680664063e-06,
      "step": 6538,
      "training_step_time": 17.12264060974121
    },
    {
      "epoch": 6.2360763549804685e-06,
      "model_forward_time": 5.619014024734497,
      "step": 6539
    },
    {
      "epoch": 6.2360763549804685e-06,
      "step": 6539,
      "training_step_time": 17.116336345672607
    },
    {
      "epoch": 6.237030029296875e-06,
      "grad_norm": 0.6840887665748596,
      "learning_rate": 1e-05,
      "loss": 0.0624,
      "step": 6540
    },
    {
      "epoch": 6.237030029296875e-06,
      "model_forward_time": 5.998043775558472,
      "step": 6540
    },
    {
      "dynamics_loss_avg": 0.05363886207342148,
      "epoch": 6.237030029296875e-06,
      "step": 6540
    },
    {
      "action_loss_avg": 0.010232346225529909,
      "epoch": 6.237030029296875e-06,
      "step": 6540
    },
    {
      "epoch": 6.237030029296875e-06,
      "step": 6540,
      "training_step_time": 17.461318969726562
    },
    {
      "epoch": 6.237983703613282e-06,
      "model_forward_time": 5.591881513595581,
      "step": 6541
    },
    {
      "epoch": 6.237983703613282e-06,
      "step": 6541,
      "training_step_time": 17.01072359085083
    },
    {
      "epoch": 6.238937377929687e-06,
      "model_forward_time": 5.578246593475342,
      "step": 6542
    },
    {
      "epoch": 6.238937377929687e-06,
      "step": 6542,
      "training_step_time": 17.0250403881073
    },
    {
      "epoch": 6.239891052246094e-06,
      "model_forward_time": 5.6172497272491455,
      "step": 6543
    },
    {
      "epoch": 6.239891052246094e-06,
      "step": 6543,
      "training_step_time": 17.105367183685303
    },
    {
      "epoch": 6.2408447265625e-06,
      "model_forward_time": 5.620915174484253,
      "step": 6544
    },
    {
      "epoch": 6.2408447265625e-06,
      "step": 6544,
      "training_step_time": 17.105048656463623
    },
    {
      "epoch": 6.241798400878906e-06,
      "model_forward_time": 5.951586723327637,
      "step": 6545
    },
    {
      "epoch": 6.241798400878906e-06,
      "step": 6545,
      "training_step_time": 17.386117696762085
    },
    {
      "epoch": 6.242752075195313e-06,
      "model_forward_time": 5.573312997817993,
      "step": 6546
    },
    {
      "epoch": 6.242752075195313e-06,
      "step": 6546,
      "training_step_time": 17.016262769699097
    },
    {
      "epoch": 6.2437057495117185e-06,
      "model_forward_time": 5.5785157680511475,
      "step": 6547
    },
    {
      "epoch": 6.2437057495117185e-06,
      "step": 6547,
      "training_step_time": 17.00426983833313
    },
    {
      "epoch": 6.244659423828125e-06,
      "model_forward_time": 5.590311050415039,
      "step": 6548
    },
    {
      "epoch": 6.244659423828125e-06,
      "step": 6548,
      "training_step_time": 17.050743103027344
    },
    {
      "epoch": 6.245613098144532e-06,
      "model_forward_time": 5.600991725921631,
      "step": 6549
    },
    {
      "epoch": 6.245613098144532e-06,
      "step": 6549,
      "training_step_time": 17.075705528259277
    },
    {
      "epoch": 6.246566772460937e-06,
      "grad_norm": 1.3985947370529175,
      "learning_rate": 1e-05,
      "loss": 0.0503,
      "step": 6550
    },
    {
      "epoch": 6.246566772460937e-06,
      "model_forward_time": 5.604284763336182,
      "step": 6550
    },
    {
      "dynamics_loss_avg": 0.04226522166281939,
      "epoch": 6.246566772460937e-06,
      "step": 6550
    },
    {
      "action_loss_avg": 0.006863870588131249,
      "epoch": 6.246566772460937e-06,
      "step": 6550
    },
    {
      "epoch": 6.246566772460937e-06,
      "step": 6550,
      "training_step_time": 17.07749319076538
    },
    {
      "epoch": 6.247520446777344e-06,
      "model_forward_time": 5.9938037395477295,
      "step": 6551
    },
    {
      "epoch": 6.247520446777344e-06,
      "step": 6551,
      "training_step_time": 17.456603527069092
    },
    {
      "epoch": 6.24847412109375e-06,
      "model_forward_time": 5.594572305679321,
      "step": 6552
    },
    {
      "epoch": 6.24847412109375e-06,
      "step": 6552,
      "training_step_time": 17.037500619888306
    },
    {
      "epoch": 6.249427795410156e-06,
      "model_forward_time": 5.5877814292907715,
      "step": 6553
    },
    {
      "epoch": 6.249427795410156e-06,
      "step": 6553,
      "training_step_time": 17.03255558013916
    },
    {
      "epoch": 6.250381469726563e-06,
      "model_forward_time": 5.5894787311553955,
      "step": 6554
    },
    {
      "epoch": 6.250381469726563e-06,
      "step": 6554,
      "training_step_time": 17.034259796142578
    },
    {
      "epoch": 6.2513351440429685e-06,
      "model_forward_time": 5.590362548828125,
      "step": 6555
    },
    {
      "epoch": 6.2513351440429685e-06,
      "step": 6555,
      "training_step_time": 17.05301547050476
    },
    {
      "epoch": 6.252288818359375e-06,
      "model_forward_time": 5.965955018997192,
      "step": 6556
    },
    {
      "epoch": 6.252288818359375e-06,
      "step": 6556,
      "training_step_time": 17.43287944793701
    },
    {
      "epoch": 6.253242492675782e-06,
      "model_forward_time": 5.597826242446899,
      "step": 6557
    },
    {
      "epoch": 6.253242492675782e-06,
      "step": 6557,
      "training_step_time": 17.043511867523193
    },
    {
      "epoch": 6.254196166992187e-06,
      "model_forward_time": 5.592063665390015,
      "step": 6558
    },
    {
      "epoch": 6.254196166992187e-06,
      "step": 6558,
      "training_step_time": 17.03466033935547
    },
    {
      "epoch": 6.255149841308594e-06,
      "model_forward_time": 5.592078447341919,
      "step": 6559
    },
    {
      "epoch": 6.255149841308594e-06,
      "step": 6559,
      "training_step_time": 17.052399158477783
    },
    {
      "epoch": 6.256103515625e-06,
      "grad_norm": 1.22905433177948,
      "learning_rate": 1e-05,
      "loss": 0.0746,
      "step": 6560
    },
    {
      "epoch": 6.256103515625e-06,
      "model_forward_time": 5.598232746124268,
      "step": 6560
    },
    {
      "dynamics_loss_avg": 0.06633863430470228,
      "epoch": 6.256103515625e-06,
      "step": 6560
    },
    {
      "action_loss_avg": 0.011825323128141463,
      "epoch": 6.256103515625e-06,
      "step": 6560
    },
    {
      "epoch": 6.256103515625e-06,
      "step": 6560,
      "training_step_time": 17.090248107910156
    },
    {
      "epoch": 6.257057189941406e-06,
      "model_forward_time": 5.617803335189819,
      "step": 6561
    },
    {
      "epoch": 6.257057189941406e-06,
      "step": 6561,
      "training_step_time": 17.48581862449646
    },
    {
      "epoch": 6.258010864257813e-06,
      "model_forward_time": 5.619644641876221,
      "step": 6562
    },
    {
      "epoch": 6.258010864257813e-06,
      "step": 6562,
      "training_step_time": 17.134108304977417
    },
    {
      "epoch": 6.2589645385742185e-06,
      "model_forward_time": 5.610286712646484,
      "step": 6563
    },
    {
      "epoch": 6.2589645385742185e-06,
      "step": 6563,
      "training_step_time": 17.07695960998535
    },
    {
      "epoch": 6.259918212890625e-06,
      "model_forward_time": 5.5886452198028564,
      "step": 6564
    },
    {
      "epoch": 6.259918212890625e-06,
      "step": 6564,
      "training_step_time": 17.018644332885742
    },
    {
      "epoch": 6.260871887207032e-06,
      "model_forward_time": 5.566894769668579,
      "step": 6565
    },
    {
      "epoch": 6.260871887207032e-06,
      "step": 6565,
      "training_step_time": 16.993120193481445
    },
    {
      "epoch": 6.261825561523437e-06,
      "model_forward_time": 5.57110071182251,
      "step": 6566
    },
    {
      "epoch": 6.261825561523437e-06,
      "step": 6566,
      "training_step_time": 16.97698211669922
    },
    {
      "epoch": 6.262779235839844e-06,
      "model_forward_time": 5.927720546722412,
      "step": 6567
    },
    {
      "epoch": 6.262779235839844e-06,
      "step": 6567,
      "training_step_time": 17.331541538238525
    },
    {
      "epoch": 6.26373291015625e-06,
      "model_forward_time": 5.560024261474609,
      "step": 6568
    },
    {
      "epoch": 6.26373291015625e-06,
      "step": 6568,
      "training_step_time": 16.997661352157593
    },
    {
      "epoch": 6.264686584472656e-06,
      "model_forward_time": 5.56804633140564,
      "step": 6569
    },
    {
      "epoch": 6.264686584472656e-06,
      "step": 6569,
      "training_step_time": 16.990432024002075
    },
    {
      "epoch": 6.265640258789063e-06,
      "grad_norm": 0.8065857887268066,
      "learning_rate": 1e-05,
      "loss": 0.0737,
      "step": 6570
    },
    {
      "epoch": 6.265640258789063e-06,
      "model_forward_time": 5.580267429351807,
      "step": 6570
    },
    {
      "dynamics_loss_avg": 0.0568612165749073,
      "epoch": 6.265640258789063e-06,
      "step": 6570
    },
    {
      "action_loss_avg": 0.009805615642108023,
      "epoch": 6.265640258789063e-06,
      "step": 6570
    },
    {
      "epoch": 6.265640258789063e-06,
      "step": 6570,
      "training_step_time": 16.992887020111084
    },
    {
      "epoch": 6.2665939331054685e-06,
      "model_forward_time": 5.588471412658691,
      "step": 6571
    },
    {
      "epoch": 6.2665939331054685e-06,
      "step": 6571,
      "training_step_time": 17.045433044433594
    },
    {
      "epoch": 6.267547607421875e-06,
      "model_forward_time": 5.9950034618377686,
      "step": 6572
    },
    {
      "epoch": 6.267547607421875e-06,
      "step": 6572,
      "training_step_time": 17.468677520751953
    },
    {
      "epoch": 6.268501281738282e-06,
      "model_forward_time": 5.615136623382568,
      "step": 6573
    },
    {
      "epoch": 6.268501281738282e-06,
      "step": 6573,
      "training_step_time": 17.10762858390808
    },
    {
      "epoch": 6.269454956054687e-06,
      "model_forward_time": 5.617065668106079,
      "step": 6574
    },
    {
      "epoch": 6.269454956054687e-06,
      "step": 6574,
      "training_step_time": 17.094533681869507
    },
    {
      "epoch": 6.270408630371094e-06,
      "model_forward_time": 5.589721441268921,
      "step": 6575
    },
    {
      "epoch": 6.270408630371094e-06,
      "step": 6575,
      "training_step_time": 17.003579139709473
    },
    {
      "epoch": 6.2713623046875e-06,
      "model_forward_time": 5.571784019470215,
      "step": 6576
    },
    {
      "epoch": 6.2713623046875e-06,
      "step": 6576,
      "training_step_time": 17.012434720993042
    },
    {
      "epoch": 6.272315979003906e-06,
      "model_forward_time": 5.56488561630249,
      "step": 6577
    },
    {
      "epoch": 6.272315979003906e-06,
      "step": 6577,
      "training_step_time": 16.973989963531494
    },
    {
      "epoch": 6.273269653320313e-06,
      "model_forward_time": 5.920650005340576,
      "step": 6578
    },
    {
      "epoch": 6.273269653320313e-06,
      "step": 6578,
      "training_step_time": 17.334410667419434
    },
    {
      "epoch": 6.2742233276367185e-06,
      "model_forward_time": 5.568386077880859,
      "step": 6579
    },
    {
      "epoch": 6.2742233276367185e-06,
      "step": 6579,
      "training_step_time": 16.985933303833008
    },
    {
      "epoch": 6.275177001953125e-06,
      "grad_norm": 1.1387213468551636,
      "learning_rate": 1e-05,
      "loss": 0.077,
      "step": 6580
    },
    {
      "epoch": 6.275177001953125e-06,
      "model_forward_time": 5.575533866882324,
      "step": 6580
    },
    {
      "dynamics_loss_avg": 0.06672863140702248,
      "epoch": 6.275177001953125e-06,
      "step": 6580
    },
    {
      "action_loss_avg": 0.01479598383884877,
      "epoch": 6.275177001953125e-06,
      "step": 6580
    },
    {
      "epoch": 6.275177001953125e-06,
      "step": 6580,
      "training_step_time": 17.0141282081604
    },
    {
      "epoch": 6.276130676269532e-06,
      "model_forward_time": 5.593626260757446,
      "step": 6581
    },
    {
      "epoch": 6.276130676269532e-06,
      "step": 6581,
      "training_step_time": 17.04677104949951
    },
    {
      "epoch": 6.277084350585937e-06,
      "model_forward_time": 5.600640058517456,
      "step": 6582
    },
    {
      "epoch": 6.277084350585937e-06,
      "step": 6582,
      "training_step_time": 17.062740564346313
    },
    {
      "epoch": 6.278038024902344e-06,
      "model_forward_time": 5.966175556182861,
      "step": 6583
    },
    {
      "epoch": 6.278038024902344e-06,
      "step": 6583,
      "training_step_time": 17.42927598953247
    },
    {
      "epoch": 6.27899169921875e-06,
      "model_forward_time": 5.600280284881592,
      "step": 6584
    },
    {
      "epoch": 6.27899169921875e-06,
      "step": 6584,
      "training_step_time": 17.059502601623535
    },
    {
      "epoch": 6.279945373535156e-06,
      "model_forward_time": 5.5957865715026855,
      "step": 6585
    },
    {
      "epoch": 6.279945373535156e-06,
      "step": 6585,
      "training_step_time": 17.074562311172485
    },
    {
      "epoch": 6.280899047851563e-06,
      "model_forward_time": 5.607125759124756,
      "step": 6586
    },
    {
      "epoch": 6.280899047851563e-06,
      "step": 6586,
      "training_step_time": 17.07269597053528
    },
    {
      "epoch": 6.2818527221679685e-06,
      "model_forward_time": 5.605121612548828,
      "step": 6587
    },
    {
      "epoch": 6.2818527221679685e-06,
      "step": 6587,
      "training_step_time": 17.074424266815186
    },
    {
      "epoch": 6.282806396484375e-06,
      "model_forward_time": 5.602739572525024,
      "step": 6588
    },
    {
      "epoch": 6.282806396484375e-06,
      "step": 6588,
      "training_step_time": 17.066864013671875
    },
    {
      "epoch": 6.283760070800782e-06,
      "model_forward_time": 5.936481952667236,
      "step": 6589
    },
    {
      "epoch": 6.283760070800782e-06,
      "step": 6589,
      "training_step_time": 17.37373423576355
    },
    {
      "epoch": 6.284713745117187e-06,
      "grad_norm": 1.216294527053833,
      "learning_rate": 1e-05,
      "loss": 0.0847,
      "step": 6590
    },
    {
      "epoch": 6.284713745117187e-06,
      "model_forward_time": 5.58293890953064,
      "step": 6590
    },
    {
      "dynamics_loss_avg": 0.06591331008821726,
      "epoch": 6.284713745117187e-06,
      "step": 6590
    },
    {
      "action_loss_avg": 0.01392014310695231,
      "epoch": 6.284713745117187e-06,
      "step": 6590
    },
    {
      "epoch": 6.284713745117187e-06,
      "step": 6590,
      "training_step_time": 17.039114952087402
    },
    {
      "epoch": 6.285667419433594e-06,
      "model_forward_time": 5.605291366577148,
      "step": 6591
    },
    {
      "epoch": 6.285667419433594e-06,
      "step": 6591,
      "training_step_time": 17.054774522781372
    },
    {
      "epoch": 6.28662109375e-06,
      "model_forward_time": 5.606502056121826,
      "step": 6592
    },
    {
      "epoch": 6.28662109375e-06,
      "step": 6592,
      "training_step_time": 17.100908756256104
    },
    {
      "epoch": 6.287574768066406e-06,
      "model_forward_time": 5.6132612228393555,
      "step": 6593
    },
    {
      "epoch": 6.287574768066406e-06,
      "step": 6593,
      "training_step_time": 17.09042453765869
    },
    {
      "epoch": 6.288528442382813e-06,
      "model_forward_time": 5.961470603942871,
      "step": 6594
    },
    {
      "epoch": 6.288528442382813e-06,
      "step": 6594,
      "training_step_time": 17.398503303527832
    },
    {
      "epoch": 6.2894821166992185e-06,
      "model_forward_time": 5.564555883407593,
      "step": 6595
    },
    {
      "epoch": 6.2894821166992185e-06,
      "step": 6595,
      "training_step_time": 16.994494915008545
    },
    {
      "epoch": 6.290435791015625e-06,
      "model_forward_time": 5.567860841751099,
      "step": 6596
    },
    {
      "epoch": 6.290435791015625e-06,
      "step": 6596,
      "training_step_time": 16.99037528038025
    },
    {
      "epoch": 6.291389465332032e-06,
      "model_forward_time": 5.564743995666504,
      "step": 6597
    },
    {
      "epoch": 6.291389465332032e-06,
      "step": 6597,
      "training_step_time": 16.973880767822266
    },
    {
      "epoch": 6.292343139648437e-06,
      "model_forward_time": 5.576826333999634,
      "step": 6598
    },
    {
      "epoch": 6.292343139648437e-06,
      "step": 6598,
      "training_step_time": 16.98940682411194
    },
    {
      "epoch": 6.293296813964844e-06,
      "model_forward_time": 5.588240623474121,
      "step": 6599
    },
    {
      "epoch": 6.293296813964844e-06,
      "step": 6599,
      "training_step_time": 17.02709174156189
    },
    {
      "epoch": 6.29425048828125e-06,
      "grad_norm": 1.0710824728012085,
      "learning_rate": 1e-05,
      "loss": 0.0673,
      "step": 6600
    },
    {
      "epoch": 6.29425048828125e-06,
      "model_forward_time": 5.9364519119262695,
      "step": 6600
    },
    {
      "dynamics_loss_avg": 0.06057551391422748,
      "epoch": 6.29425048828125e-06,
      "step": 6600
    },
    {
      "action_loss_avg": 0.013319756696000695,
      "epoch": 6.29425048828125e-06,
      "step": 6600
    },
    {
      "epoch": 6.29425048828125e-06,
      "step": 6600,
      "training_step_time": 17.38760471343994
    },
    {
      "epoch": 6.295204162597656e-06,
      "model_forward_time": 5.595983266830444,
      "step": 6601
    },
    {
      "epoch": 6.295204162597656e-06,
      "step": 6601,
      "training_step_time": 17.06491708755493
    },
    {
      "epoch": 6.296157836914063e-06,
      "model_forward_time": 5.598891496658325,
      "step": 6602
    },
    {
      "epoch": 6.296157836914063e-06,
      "step": 6602,
      "training_step_time": 17.058358430862427
    },
    {
      "epoch": 6.2971115112304685e-06,
      "model_forward_time": 5.605775833129883,
      "step": 6603
    },
    {
      "epoch": 6.2971115112304685e-06,
      "step": 6603,
      "training_step_time": 17.0807945728302
    },
    {
      "epoch": 6.298065185546875e-06,
      "model_forward_time": 5.606679439544678,
      "step": 6604
    },
    {
      "epoch": 6.298065185546875e-06,
      "step": 6604,
      "training_step_time": 17.097336292266846
    },
    {
      "epoch": 6.299018859863282e-06,
      "model_forward_time": 5.969993352890015,
      "step": 6605
    },
    {
      "epoch": 6.299018859863282e-06,
      "step": 6605,
      "training_step_time": 17.45641040802002
    },
    {
      "epoch": 6.299972534179687e-06,
      "model_forward_time": 5.602705240249634,
      "step": 6606
    },
    {
      "epoch": 6.299972534179687e-06,
      "step": 6606,
      "training_step_time": 17.07309126853943
    },
    {
      "epoch": 6.300926208496094e-06,
      "model_forward_time": 5.604870080947876,
      "step": 6607
    },
    {
      "epoch": 6.300926208496094e-06,
      "step": 6607,
      "training_step_time": 17.077017545700073
    },
    {
      "epoch": 6.3018798828125e-06,
      "model_forward_time": 5.592076539993286,
      "step": 6608
    },
    {
      "epoch": 6.3018798828125e-06,
      "step": 6608,
      "training_step_time": 17.0381760597229
    },
    {
      "epoch": 6.302833557128906e-06,
      "model_forward_time": 5.600346803665161,
      "step": 6609
    },
    {
      "epoch": 6.302833557128906e-06,
      "step": 6609,
      "training_step_time": 17.041179180145264
    },
    {
      "epoch": 6.303787231445313e-06,
      "grad_norm": 1.2975480556488037,
      "learning_rate": 1e-05,
      "loss": 0.0734,
      "step": 6610
    },
    {
      "epoch": 6.303787231445313e-06,
      "model_forward_time": 5.95203971862793,
      "step": 6610
    },
    {
      "dynamics_loss_avg": 0.05458833836019039,
      "epoch": 6.303787231445313e-06,
      "step": 6610
    },
    {
      "action_loss_avg": 0.014093370735645294,
      "epoch": 6.303787231445313e-06,
      "step": 6610
    },
    {
      "epoch": 6.303787231445313e-06,
      "step": 6610,
      "training_step_time": 17.37820816040039
    },
    {
      "epoch": 6.3047409057617185e-06,
      "model_forward_time": 5.611347913742065,
      "step": 6611
    },
    {
      "epoch": 6.3047409057617185e-06,
      "step": 6611,
      "training_step_time": 17.09892988204956
    },
    {
      "epoch": 6.305694580078125e-06,
      "model_forward_time": 5.625698089599609,
      "step": 6612
    },
    {
      "epoch": 6.305694580078125e-06,
      "step": 6612,
      "training_step_time": 17.134665727615356
    },
    {
      "epoch": 6.306648254394532e-06,
      "model_forward_time": 5.614624738693237,
      "step": 6613
    },
    {
      "epoch": 6.306648254394532e-06,
      "step": 6613,
      "training_step_time": 17.09471893310547
    },
    {
      "epoch": 6.307601928710937e-06,
      "model_forward_time": 5.609229564666748,
      "step": 6614
    },
    {
      "epoch": 6.307601928710937e-06,
      "step": 6614,
      "training_step_time": 17.06704330444336
    },
    {
      "epoch": 6.308555603027344e-06,
      "model_forward_time": 5.60050892829895,
      "step": 6615
    },
    {
      "epoch": 6.308555603027344e-06,
      "step": 6615,
      "training_step_time": 17.05027747154236
    },
    {
      "epoch": 6.30950927734375e-06,
      "model_forward_time": 5.957096576690674,
      "step": 6616
    },
    {
      "epoch": 6.30950927734375e-06,
      "step": 6616,
      "training_step_time": 17.414552927017212
    },
    {
      "epoch": 6.310462951660156e-06,
      "model_forward_time": 5.598247766494751,
      "step": 6617
    },
    {
      "epoch": 6.310462951660156e-06,
      "step": 6617,
      "training_step_time": 17.06067442893982
    },
    {
      "epoch": 6.311416625976563e-06,
      "model_forward_time": 5.606162071228027,
      "step": 6618
    },
    {
      "epoch": 6.311416625976563e-06,
      "step": 6618,
      "training_step_time": 17.10006093978882
    },
    {
      "epoch": 6.3123703002929685e-06,
      "model_forward_time": 5.611819505691528,
      "step": 6619
    },
    {
      "epoch": 6.3123703002929685e-06,
      "step": 6619,
      "training_step_time": 17.08941674232483
    },
    {
      "epoch": 6.313323974609375e-06,
      "grad_norm": 0.48968446254730225,
      "learning_rate": 1e-05,
      "loss": 0.058,
      "step": 6620
    },
    {
      "epoch": 6.313323974609375e-06,
      "model_forward_time": 5.599463939666748,
      "step": 6620
    },
    {
      "dynamics_loss_avg": 0.049501039832830426,
      "epoch": 6.313323974609375e-06,
      "step": 6620
    },
    {
      "action_loss_avg": 0.009540235484018922,
      "epoch": 6.313323974609375e-06,
      "step": 6620
    },
    {
      "epoch": 6.313323974609375e-06,
      "step": 6620,
      "training_step_time": 17.051583528518677
    },
    {
      "epoch": 6.314277648925782e-06,
      "model_forward_time": 5.596663951873779,
      "step": 6621
    },
    {
      "epoch": 6.314277648925782e-06,
      "step": 6621,
      "training_step_time": 17.415983200073242
    },
    {
      "epoch": 6.315231323242187e-06,
      "model_forward_time": 5.577401638031006,
      "step": 6622
    },
    {
      "epoch": 6.315231323242187e-06,
      "step": 6622,
      "training_step_time": 17.010635375976562
    },
    {
      "epoch": 6.316184997558594e-06,
      "model_forward_time": 5.586118936538696,
      "step": 6623
    },
    {
      "epoch": 6.316184997558594e-06,
      "step": 6623,
      "training_step_time": 17.033643007278442
    },
    {
      "epoch": 6.317138671875e-06,
      "model_forward_time": 5.597910642623901,
      "step": 6624
    },
    {
      "epoch": 6.317138671875e-06,
      "step": 6624,
      "training_step_time": 17.067272424697876
    },
    {
      "epoch": 6.318092346191406e-06,
      "model_forward_time": 5.619255065917969,
      "step": 6625
    },
    {
      "epoch": 6.318092346191406e-06,
      "step": 6625,
      "training_step_time": 17.128000259399414
    },
    {
      "epoch": 6.319046020507813e-06,
      "model_forward_time": 5.621774911880493,
      "step": 6626
    },
    {
      "epoch": 6.319046020507813e-06,
      "step": 6626,
      "training_step_time": 17.124932050704956
    },
    {
      "epoch": 6.3199996948242185e-06,
      "model_forward_time": 5.97359824180603,
      "step": 6627
    },
    {
      "epoch": 6.3199996948242185e-06,
      "step": 6627,
      "training_step_time": 17.450215339660645
    },
    {
      "epoch": 6.320953369140625e-06,
      "model_forward_time": 5.592901229858398,
      "step": 6628
    },
    {
      "epoch": 6.320953369140625e-06,
      "step": 6628,
      "training_step_time": 17.023741722106934
    },
    {
      "epoch": 6.321907043457031e-06,
      "model_forward_time": 5.587021589279175,
      "step": 6629
    },
    {
      "epoch": 6.321907043457031e-06,
      "step": 6629,
      "training_step_time": 17.021705865859985
    },
    {
      "epoch": 6.322860717773437e-06,
      "grad_norm": 1.3893840312957764,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 6630
    },
    {
      "epoch": 6.322860717773437e-06,
      "model_forward_time": 5.573912620544434,
      "step": 6630
    },
    {
      "dynamics_loss_avg": 0.059859991818666455,
      "epoch": 6.322860717773437e-06,
      "step": 6630
    },
    {
      "action_loss_avg": 0.009650041442364454,
      "epoch": 6.322860717773437e-06,
      "step": 6630
    },
    {
      "epoch": 6.322860717773437e-06,
      "step": 6630,
      "training_step_time": 17.002426385879517
    },
    {
      "epoch": 6.323814392089844e-06,
      "model_forward_time": 5.57437539100647,
      "step": 6631
    },
    {
      "epoch": 6.323814392089844e-06,
      "step": 6631,
      "training_step_time": 16.997886419296265
    },
    {
      "epoch": 6.32476806640625e-06,
      "model_forward_time": 5.586804389953613,
      "step": 6632
    },
    {
      "epoch": 6.32476806640625e-06,
      "step": 6632,
      "training_step_time": 17.387705326080322
    },
    {
      "epoch": 6.325721740722656e-06,
      "model_forward_time": 5.586342096328735,
      "step": 6633
    },
    {
      "epoch": 6.325721740722656e-06,
      "step": 6633,
      "training_step_time": 17.01840829849243
    },
    {
      "epoch": 6.326675415039063e-06,
      "model_forward_time": 5.594113111495972,
      "step": 6634
    },
    {
      "epoch": 6.326675415039063e-06,
      "step": 6634,
      "training_step_time": 17.06387209892273
    },
    {
      "epoch": 6.3276290893554685e-06,
      "model_forward_time": 5.593245983123779,
      "step": 6635
    },
    {
      "epoch": 6.3276290893554685e-06,
      "step": 6635,
      "training_step_time": 17.06702995300293
    },
    {
      "epoch": 6.328582763671875e-06,
      "model_forward_time": 5.618800401687622,
      "step": 6636
    },
    {
      "epoch": 6.328582763671875e-06,
      "step": 6636,
      "training_step_time": 17.11133122444153
    },
    {
      "epoch": 6.329536437988281e-06,
      "model_forward_time": 5.615599155426025,
      "step": 6637
    },
    {
      "epoch": 6.329536437988281e-06,
      "step": 6637,
      "training_step_time": 17.11974573135376
    },
    {
      "epoch": 6.330490112304687e-06,
      "model_forward_time": 5.981409311294556,
      "step": 6638
    },
    {
      "epoch": 6.330490112304687e-06,
      "step": 6638,
      "training_step_time": 17.47005534172058
    },
    {
      "epoch": 6.331443786621094e-06,
      "model_forward_time": 5.6078667640686035,
      "step": 6639
    },
    {
      "epoch": 6.331443786621094e-06,
      "step": 6639,
      "training_step_time": 17.056946516036987
    },
    {
      "epoch": 6.3323974609375e-06,
      "grad_norm": 0.8835594654083252,
      "learning_rate": 1e-05,
      "loss": 0.0786,
      "step": 6640
    },
    {
      "epoch": 6.3323974609375e-06,
      "model_forward_time": 5.592346429824829,
      "step": 6640
    },
    {
      "dynamics_loss_avg": 0.06245391555130482,
      "epoch": 6.3323974609375e-06,
      "step": 6640
    },
    {
      "action_loss_avg": 0.0135934978723526,
      "epoch": 6.3323974609375e-06,
      "step": 6640
    },
    {
      "epoch": 6.3323974609375e-06,
      "step": 6640,
      "training_step_time": 17.02989625930786
    },
    {
      "epoch": 6.333351135253906e-06,
      "model_forward_time": 5.57999062538147,
      "step": 6641
    },
    {
      "epoch": 6.333351135253906e-06,
      "step": 6641,
      "training_step_time": 17.00967264175415
    },
    {
      "epoch": 6.334304809570313e-06,
      "model_forward_time": 5.582850217819214,
      "step": 6642
    },
    {
      "epoch": 6.334304809570313e-06,
      "step": 6642,
      "training_step_time": 17.017449378967285
    },
    {
      "epoch": 6.3352584838867186e-06,
      "model_forward_time": 5.960559606552124,
      "step": 6643
    },
    {
      "epoch": 6.3352584838867186e-06,
      "step": 6643,
      "training_step_time": 17.430538654327393
    },
    {
      "epoch": 6.336212158203125e-06,
      "model_forward_time": 5.613295793533325,
      "step": 6644
    },
    {
      "epoch": 6.336212158203125e-06,
      "step": 6644,
      "training_step_time": 17.12359619140625
    },
    {
      "epoch": 6.337165832519531e-06,
      "model_forward_time": 5.621050119400024,
      "step": 6645
    },
    {
      "epoch": 6.337165832519531e-06,
      "step": 6645,
      "training_step_time": 17.140730381011963
    },
    {
      "epoch": 6.338119506835937e-06,
      "model_forward_time": 5.626641511917114,
      "step": 6646
    },
    {
      "epoch": 6.338119506835937e-06,
      "step": 6646,
      "training_step_time": 17.140300989151
    },
    {
      "epoch": 6.339073181152344e-06,
      "model_forward_time": 5.618341684341431,
      "step": 6647
    },
    {
      "epoch": 6.339073181152344e-06,
      "step": 6647,
      "training_step_time": 17.106809377670288
    },
    {
      "epoch": 6.34002685546875e-06,
      "model_forward_time": 5.616573810577393,
      "step": 6648
    },
    {
      "epoch": 6.34002685546875e-06,
      "step": 6648,
      "training_step_time": 17.09553050994873
    },
    {
      "epoch": 6.340980529785156e-06,
      "model_forward_time": 5.969178915023804,
      "step": 6649
    },
    {
      "epoch": 6.340980529785156e-06,
      "step": 6649,
      "training_step_time": 17.454792976379395
    },
    {
      "epoch": 6.341934204101563e-06,
      "grad_norm": 1.0151643753051758,
      "learning_rate": 1e-05,
      "loss": 0.0734,
      "step": 6650
    },
    {
      "epoch": 6.341934204101563e-06,
      "model_forward_time": 5.602538824081421,
      "step": 6650
    },
    {
      "dynamics_loss_avg": 0.0675686165690422,
      "epoch": 6.341934204101563e-06,
      "step": 6650
    },
    {
      "action_loss_avg": 0.012009056261740625,
      "epoch": 6.341934204101563e-06,
      "step": 6650
    },
    {
      "epoch": 6.341934204101563e-06,
      "step": 6650,
      "training_step_time": 17.073991537094116
    },
    {
      "epoch": 6.3428878784179686e-06,
      "model_forward_time": 5.603982210159302,
      "step": 6651
    },
    {
      "epoch": 6.3428878784179686e-06,
      "step": 6651,
      "training_step_time": 17.08060598373413
    },
    {
      "epoch": 6.343841552734375e-06,
      "model_forward_time": 5.5973801612854,
      "step": 6652
    },
    {
      "epoch": 6.343841552734375e-06,
      "step": 6652,
      "training_step_time": 17.047308206558228
    },
    {
      "epoch": 6.344795227050781e-06,
      "model_forward_time": 5.605701446533203,
      "step": 6653
    },
    {
      "epoch": 6.344795227050781e-06,
      "step": 6653,
      "training_step_time": 17.062408924102783
    },
    {
      "epoch": 6.345748901367187e-06,
      "model_forward_time": 5.595790147781372,
      "step": 6654
    },
    {
      "epoch": 6.345748901367187e-06,
      "step": 6654,
      "training_step_time": 17.417216300964355
    },
    {
      "epoch": 6.346702575683594e-06,
      "model_forward_time": 5.600416898727417,
      "step": 6655
    },
    {
      "epoch": 6.346702575683594e-06,
      "step": 6655,
      "training_step_time": 17.084514617919922
    },
    {
      "epoch": 6.34765625e-06,
      "model_forward_time": 5.607813358306885,
      "step": 6656
    },
    {
      "epoch": 6.34765625e-06,
      "step": 6656,
      "training_step_time": 17.08759355545044
    },
    {
      "epoch": 6.348609924316406e-06,
      "model_forward_time": 5.5974836349487305,
      "step": 6657
    },
    {
      "epoch": 6.348609924316406e-06,
      "step": 6657,
      "training_step_time": 17.006335735321045
    },
    {
      "epoch": 6.349563598632813e-06,
      "model_forward_time": 5.5807812213897705,
      "step": 6658
    },
    {
      "epoch": 6.349563598632813e-06,
      "step": 6658,
      "training_step_time": 16.980894327163696
    },
    {
      "epoch": 6.3505172729492186e-06,
      "model_forward_time": 5.594745397567749,
      "step": 6659
    },
    {
      "epoch": 6.3505172729492186e-06,
      "step": 6659,
      "training_step_time": 17.03807520866394
    },
    {
      "epoch": 6.351470947265625e-06,
      "grad_norm": 1.6172711849212646,
      "learning_rate": 1e-05,
      "loss": 0.0732,
      "step": 6660
    },
    {
      "epoch": 6.351470947265625e-06,
      "model_forward_time": 5.957392692565918,
      "step": 6660
    },
    {
      "dynamics_loss_avg": 0.05971393380314112,
      "epoch": 6.351470947265625e-06,
      "step": 6660
    },
    {
      "action_loss_avg": 0.011862150160595775,
      "epoch": 6.351470947265625e-06,
      "step": 6660
    },
    {
      "epoch": 6.351470947265625e-06,
      "step": 6660,
      "training_step_time": 17.402170419692993
    },
    {
      "epoch": 6.352424621582031e-06,
      "model_forward_time": 5.6059229373931885,
      "step": 6661
    },
    {
      "epoch": 6.352424621582031e-06,
      "step": 6661,
      "training_step_time": 17.101865530014038
    },
    {
      "epoch": 6.3533782958984374e-06,
      "model_forward_time": 5.616316080093384,
      "step": 6662
    },
    {
      "epoch": 6.3533782958984374e-06,
      "step": 6662,
      "training_step_time": 17.121020078659058
    },
    {
      "epoch": 6.354331970214844e-06,
      "model_forward_time": 5.595416307449341,
      "step": 6663
    },
    {
      "epoch": 6.354331970214844e-06,
      "step": 6663,
      "training_step_time": 17.05720281600952
    },
    {
      "epoch": 6.35528564453125e-06,
      "model_forward_time": 5.577770471572876,
      "step": 6664
    },
    {
      "epoch": 6.35528564453125e-06,
      "step": 6664,
      "training_step_time": 17.02491545677185
    },
    {
      "epoch": 6.356239318847656e-06,
      "model_forward_time": 5.578288316726685,
      "step": 6665
    },
    {
      "epoch": 6.356239318847656e-06,
      "step": 6665,
      "training_step_time": 17.369266271591187
    },
    {
      "epoch": 6.357192993164063e-06,
      "model_forward_time": 5.594038486480713,
      "step": 6666
    },
    {
      "epoch": 6.357192993164063e-06,
      "step": 6666,
      "training_step_time": 17.071403741836548
    },
    {
      "epoch": 6.3581466674804686e-06,
      "model_forward_time": 5.611487150192261,
      "step": 6667
    },
    {
      "epoch": 6.3581466674804686e-06,
      "step": 6667,
      "training_step_time": 17.11442232131958
    },
    {
      "epoch": 6.359100341796875e-06,
      "model_forward_time": 5.6205198764801025,
      "step": 6668
    },
    {
      "epoch": 6.359100341796875e-06,
      "step": 6668,
      "training_step_time": 17.118770837783813
    },
    {
      "epoch": 6.360054016113281e-06,
      "model_forward_time": 5.61545991897583,
      "step": 6669
    },
    {
      "epoch": 6.360054016113281e-06,
      "step": 6669,
      "training_step_time": 17.08376979827881
    },
    {
      "epoch": 6.3610076904296874e-06,
      "grad_norm": 0.4893956184387207,
      "learning_rate": 1e-05,
      "loss": 0.0705,
      "step": 6670
    },
    {
      "epoch": 6.3610076904296874e-06,
      "model_forward_time": 5.586358547210693,
      "step": 6670
    },
    {
      "dynamics_loss_avg": 0.06348251923918724,
      "epoch": 6.3610076904296874e-06,
      "step": 6670
    },
    {
      "action_loss_avg": 0.011499885702505708,
      "epoch": 6.3610076904296874e-06,
      "step": 6670
    },
    {
      "epoch": 6.3610076904296874e-06,
      "step": 6670,
      "training_step_time": 16.995025396347046
    },
    {
      "epoch": 6.361961364746094e-06,
      "model_forward_time": 5.938874006271362,
      "step": 6671
    },
    {
      "epoch": 6.361961364746094e-06,
      "step": 6671,
      "training_step_time": 17.36992049217224
    },
    {
      "epoch": 6.3629150390625e-06,
      "model_forward_time": 5.574785947799683,
      "step": 6672
    },
    {
      "epoch": 6.3629150390625e-06,
      "step": 6672,
      "training_step_time": 16.994597911834717
    },
    {
      "epoch": 6.363868713378906e-06,
      "model_forward_time": 5.5751683712005615,
      "step": 6673
    },
    {
      "epoch": 6.363868713378906e-06,
      "step": 6673,
      "training_step_time": 16.996405839920044
    },
    {
      "epoch": 6.364822387695313e-06,
      "model_forward_time": 5.580927848815918,
      "step": 6674
    },
    {
      "epoch": 6.364822387695313e-06,
      "step": 6674,
      "training_step_time": 17.0004780292511
    },
    {
      "epoch": 6.365776062011719e-06,
      "model_forward_time": 5.580259323120117,
      "step": 6675
    },
    {
      "epoch": 6.365776062011719e-06,
      "step": 6675,
      "training_step_time": 17.032430171966553
    },
    {
      "epoch": 6.366729736328125e-06,
      "model_forward_time": 5.9385035037994385,
      "step": 6676
    },
    {
      "epoch": 6.366729736328125e-06,
      "step": 6676,
      "training_step_time": 17.366010904312134
    },
    {
      "epoch": 6.367683410644531e-06,
      "model_forward_time": 5.586956739425659,
      "step": 6677
    },
    {
      "epoch": 6.367683410644531e-06,
      "step": 6677,
      "training_step_time": 17.058520555496216
    },
    {
      "epoch": 6.3686370849609374e-06,
      "model_forward_time": 5.6018171310424805,
      "step": 6678
    },
    {
      "epoch": 6.3686370849609374e-06,
      "step": 6678,
      "training_step_time": 17.092851638793945
    },
    {
      "epoch": 6.369590759277344e-06,
      "model_forward_time": 5.617454528808594,
      "step": 6679
    },
    {
      "epoch": 6.369590759277344e-06,
      "step": 6679,
      "training_step_time": 17.14001488685608
    },
    {
      "epoch": 6.37054443359375e-06,
      "grad_norm": 0.568904459476471,
      "learning_rate": 1e-05,
      "loss": 0.0754,
      "step": 6680
    },
    {
      "epoch": 6.37054443359375e-06,
      "model_forward_time": 5.627784013748169,
      "step": 6680
    },
    {
      "dynamics_loss_avg": 0.056401044502854346,
      "epoch": 6.37054443359375e-06,
      "step": 6680
    },
    {
      "action_loss_avg": 0.009959225496277212,
      "epoch": 6.37054443359375e-06,
      "step": 6680
    },
    {
      "epoch": 6.37054443359375e-06,
      "step": 6680,
      "training_step_time": 17.1171772480011
    },
    {
      "epoch": 6.371498107910156e-06,
      "model_forward_time": 5.593479156494141,
      "step": 6681
    },
    {
      "epoch": 6.371498107910156e-06,
      "step": 6681,
      "training_step_time": 17.00877857208252
    },
    {
      "epoch": 6.372451782226563e-06,
      "model_forward_time": 5.961463928222656,
      "step": 6682
    },
    {
      "epoch": 6.372451782226563e-06,
      "step": 6682,
      "training_step_time": 17.384170055389404
    },
    {
      "epoch": 6.373405456542969e-06,
      "model_forward_time": 5.571378469467163,
      "step": 6683
    },
    {
      "epoch": 6.373405456542969e-06,
      "step": 6683,
      "training_step_time": 16.999295949935913
    },
    {
      "epoch": 6.374359130859375e-06,
      "model_forward_time": 5.587658405303955,
      "step": 6684
    },
    {
      "epoch": 6.374359130859375e-06,
      "step": 6684,
      "training_step_time": 17.048609733581543
    },
    {
      "epoch": 6.375312805175781e-06,
      "model_forward_time": 5.6155736446380615,
      "step": 6685
    },
    {
      "epoch": 6.375312805175781e-06,
      "step": 6685,
      "training_step_time": 17.104305028915405
    },
    {
      "epoch": 6.3762664794921874e-06,
      "model_forward_time": 5.605323076248169,
      "step": 6686
    },
    {
      "epoch": 6.3762664794921874e-06,
      "step": 6686,
      "training_step_time": 17.061161756515503
    },
    {
      "epoch": 6.377220153808594e-06,
      "model_forward_time": 5.595555305480957,
      "step": 6687
    },
    {
      "epoch": 6.377220153808594e-06,
      "step": 6687,
      "training_step_time": 17.40584921836853
    },
    {
      "epoch": 6.378173828125e-06,
      "model_forward_time": 5.59683084487915,
      "step": 6688
    },
    {
      "epoch": 6.378173828125e-06,
      "step": 6688,
      "training_step_time": 17.062390327453613
    },
    {
      "epoch": 6.379127502441406e-06,
      "model_forward_time": 5.59990930557251,
      "step": 6689
    },
    {
      "epoch": 6.379127502441406e-06,
      "step": 6689,
      "training_step_time": 17.079445123672485
    },
    {
      "epoch": 6.380081176757813e-06,
      "grad_norm": 1.4653639793395996,
      "learning_rate": 1e-05,
      "loss": 0.0784,
      "step": 6690
    },
    {
      "epoch": 6.380081176757813e-06,
      "model_forward_time": 5.609155178070068,
      "step": 6690
    },
    {
      "dynamics_loss_avg": 0.06566784102469683,
      "epoch": 6.380081176757813e-06,
      "step": 6690
    },
    {
      "action_loss_avg": 0.01432282137684524,
      "epoch": 6.380081176757813e-06,
      "step": 6690
    },
    {
      "epoch": 6.380081176757813e-06,
      "step": 6690,
      "training_step_time": 17.09604001045227
    },
    {
      "epoch": 6.381034851074219e-06,
      "model_forward_time": 5.622255325317383,
      "step": 6691
    },
    {
      "epoch": 6.381034851074219e-06,
      "step": 6691,
      "training_step_time": 17.07224154472351
    },
    {
      "epoch": 6.381988525390625e-06,
      "model_forward_time": 5.609720468521118,
      "step": 6692
    },
    {
      "epoch": 6.381988525390625e-06,
      "step": 6692,
      "training_step_time": 17.10391926765442
    },
    {
      "epoch": 6.382942199707031e-06,
      "model_forward_time": 5.9785850048065186,
      "step": 6693
    },
    {
      "epoch": 6.382942199707031e-06,
      "step": 6693,
      "training_step_time": 17.4338641166687
    },
    {
      "epoch": 6.3838958740234375e-06,
      "model_forward_time": 5.602433919906616,
      "step": 6694
    },
    {
      "epoch": 6.3838958740234375e-06,
      "step": 6694,
      "training_step_time": 17.04475975036621
    },
    {
      "epoch": 6.384849548339844e-06,
      "model_forward_time": 5.597130060195923,
      "step": 6695
    },
    {
      "epoch": 6.384849548339844e-06,
      "step": 6695,
      "training_step_time": 17.059401273727417
    },
    {
      "epoch": 6.38580322265625e-06,
      "model_forward_time": 5.591644048690796,
      "step": 6696
    },
    {
      "epoch": 6.38580322265625e-06,
      "step": 6696,
      "training_step_time": 17.074492931365967
    },
    {
      "epoch": 6.386756896972656e-06,
      "model_forward_time": 5.599101305007935,
      "step": 6697
    },
    {
      "epoch": 6.386756896972656e-06,
      "step": 6697,
      "training_step_time": 17.07404851913452
    },
    {
      "epoch": 6.387710571289063e-06,
      "model_forward_time": 5.943883180618286,
      "step": 6698
    },
    {
      "epoch": 6.387710571289063e-06,
      "step": 6698,
      "training_step_time": 17.38958978652954
    },
    {
      "epoch": 6.388664245605469e-06,
      "model_forward_time": 5.594703197479248,
      "step": 6699
    },
    {
      "epoch": 6.388664245605469e-06,
      "step": 6699,
      "training_step_time": 17.010201692581177
    },
    {
      "epoch": 6.389617919921875e-06,
      "grad_norm": 1.0594820976257324,
      "learning_rate": 1e-05,
      "loss": 0.055,
      "step": 6700
    },
    {
      "epoch": 6.389617919921875e-06,
      "model_forward_time": 5.582638740539551,
      "step": 6700
    },
    {
      "dynamics_loss_avg": 0.04731449130922556,
      "epoch": 6.389617919921875e-06,
      "step": 6700
    },
    {
      "action_loss_avg": 0.008741401717998087,
      "epoch": 6.389617919921875e-06,
      "step": 6700
    },
    {
      "epoch": 6.389617919921875e-06,
      "step": 6700,
      "training_step_time": 17.02297282218933
    },
    {
      "epoch": 6.390571594238281e-06,
      "model_forward_time": 5.594921588897705,
      "step": 6701
    },
    {
      "epoch": 6.390571594238281e-06,
      "step": 6701,
      "training_step_time": 17.073919773101807
    },
    {
      "epoch": 6.3915252685546875e-06,
      "model_forward_time": 5.623803615570068,
      "step": 6702
    },
    {
      "epoch": 6.3915252685546875e-06,
      "step": 6702,
      "training_step_time": 17.14274549484253
    },
    {
      "epoch": 6.392478942871094e-06,
      "model_forward_time": 5.631206512451172,
      "step": 6703
    },
    {
      "epoch": 6.392478942871094e-06,
      "step": 6703,
      "training_step_time": 17.131133556365967
    },
    {
      "epoch": 6.3934326171875e-06,
      "model_forward_time": 5.966418027877808,
      "step": 6704
    },
    {
      "epoch": 6.3934326171875e-06,
      "step": 6704,
      "training_step_time": 17.442320823669434
    },
    {
      "epoch": 6.394386291503906e-06,
      "model_forward_time": 5.604536533355713,
      "step": 6705
    },
    {
      "epoch": 6.394386291503906e-06,
      "step": 6705,
      "training_step_time": 17.055313110351562
    },
    {
      "epoch": 6.395339965820313e-06,
      "model_forward_time": 5.606741428375244,
      "step": 6706
    },
    {
      "epoch": 6.395339965820313e-06,
      "step": 6706,
      "training_step_time": 17.07950782775879
    },
    {
      "epoch": 6.396293640136719e-06,
      "model_forward_time": 5.598520517349243,
      "step": 6707
    },
    {
      "epoch": 6.396293640136719e-06,
      "step": 6707,
      "training_step_time": 17.06225895881653
    },
    {
      "epoch": 6.397247314453125e-06,
      "model_forward_time": 5.5954649448394775,
      "step": 6708
    },
    {
      "epoch": 6.397247314453125e-06,
      "step": 6708,
      "training_step_time": 17.080483436584473
    },
    {
      "epoch": 6.398200988769531e-06,
      "model_forward_time": 5.9618775844573975,
      "step": 6709
    },
    {
      "epoch": 6.398200988769531e-06,
      "step": 6709,
      "training_step_time": 17.43958020210266
    },
    {
      "epoch": 6.3991546630859375e-06,
      "grad_norm": 0.5441535711288452,
      "learning_rate": 1e-05,
      "loss": 0.0728,
      "step": 6710
    },
    {
      "epoch": 6.3991546630859375e-06,
      "model_forward_time": 5.586163282394409,
      "step": 6710
    },
    {
      "dynamics_loss_avg": 0.062346691638231276,
      "epoch": 6.3991546630859375e-06,
      "step": 6710
    },
    {
      "action_loss_avg": 0.012345435842871667,
      "epoch": 6.3991546630859375e-06,
      "step": 6710
    },
    {
      "epoch": 6.3991546630859375e-06,
      "step": 6710,
      "training_step_time": 17.027976751327515
    },
    {
      "epoch": 6.400108337402344e-06,
      "model_forward_time": 5.580984115600586,
      "step": 6711
    },
    {
      "epoch": 6.400108337402344e-06,
      "step": 6711,
      "training_step_time": 17.025442361831665
    },
    {
      "epoch": 6.40106201171875e-06,
      "model_forward_time": 5.59783673286438,
      "step": 6712
    },
    {
      "epoch": 6.40106201171875e-06,
      "step": 6712,
      "training_step_time": 17.06363534927368
    },
    {
      "epoch": 6.402015686035156e-06,
      "model_forward_time": 5.608938217163086,
      "step": 6713
    },
    {
      "epoch": 6.402015686035156e-06,
      "step": 6713,
      "training_step_time": 17.073229789733887
    },
    {
      "epoch": 6.402969360351563e-06,
      "model_forward_time": 5.599631071090698,
      "step": 6714
    },
    {
      "epoch": 6.402969360351563e-06,
      "step": 6714,
      "training_step_time": 17.43409299850464
    },
    {
      "epoch": 6.403923034667969e-06,
      "model_forward_time": 5.601696014404297,
      "step": 6715
    },
    {
      "epoch": 6.403923034667969e-06,
      "step": 6715,
      "training_step_time": 17.10479187965393
    },
    {
      "epoch": 6.404876708984375e-06,
      "model_forward_time": 5.608264446258545,
      "step": 6716
    },
    {
      "epoch": 6.404876708984375e-06,
      "step": 6716,
      "training_step_time": 17.095372915267944
    },
    {
      "epoch": 6.405830383300781e-06,
      "model_forward_time": 5.61194109916687,
      "step": 6717
    },
    {
      "epoch": 6.405830383300781e-06,
      "step": 6717,
      "training_step_time": 17.085021495819092
    },
    {
      "epoch": 6.4067840576171875e-06,
      "model_forward_time": 5.608109951019287,
      "step": 6718
    },
    {
      "epoch": 6.4067840576171875e-06,
      "step": 6718,
      "training_step_time": 17.075141668319702
    },
    {
      "epoch": 6.407737731933594e-06,
      "model_forward_time": 5.593825578689575,
      "step": 6719
    },
    {
      "epoch": 6.407737731933594e-06,
      "step": 6719,
      "training_step_time": 17.01912808418274
    },
    {
      "epoch": 6.40869140625e-06,
      "grad_norm": 1.5921682119369507,
      "learning_rate": 1e-05,
      "loss": 0.0761,
      "step": 6720
    },
    {
      "epoch": 6.40869140625e-06,
      "model_forward_time": 5.922607421875,
      "step": 6720
    },
    {
      "dynamics_loss_avg": 0.05640008822083473,
      "epoch": 6.40869140625e-06,
      "step": 6720
    },
    {
      "action_loss_avg": 0.013524298928678036,
      "epoch": 6.40869140625e-06,
      "step": 6720
    },
    {
      "epoch": 6.40869140625e-06,
      "step": 6720,
      "training_step_time": 17.338423490524292
    },
    {
      "epoch": 6.409645080566406e-06,
      "model_forward_time": 5.586761474609375,
      "step": 6721
    },
    {
      "epoch": 6.409645080566406e-06,
      "step": 6721,
      "training_step_time": 17.015480995178223
    },
    {
      "epoch": 6.410598754882813e-06,
      "model_forward_time": 5.594663858413696,
      "step": 6722
    },
    {
      "epoch": 6.410598754882813e-06,
      "step": 6722,
      "training_step_time": 17.02845573425293
    },
    {
      "epoch": 6.411552429199219e-06,
      "model_forward_time": 5.58538818359375,
      "step": 6723
    },
    {
      "epoch": 6.411552429199219e-06,
      "step": 6723,
      "training_step_time": 17.040337800979614
    },
    {
      "epoch": 6.412506103515625e-06,
      "model_forward_time": 5.593864440917969,
      "step": 6724
    },
    {
      "epoch": 6.412506103515625e-06,
      "step": 6724,
      "training_step_time": 17.057772874832153
    },
    {
      "epoch": 6.413459777832031e-06,
      "model_forward_time": 5.6117095947265625,
      "step": 6725
    },
    {
      "epoch": 6.413459777832031e-06,
      "step": 6725,
      "training_step_time": 17.45113778114319
    },
    {
      "epoch": 6.4144134521484375e-06,
      "model_forward_time": 5.60819935798645,
      "step": 6726
    },
    {
      "epoch": 6.4144134521484375e-06,
      "step": 6726,
      "training_step_time": 17.105300188064575
    },
    {
      "epoch": 6.415367126464844e-06,
      "model_forward_time": 5.611392021179199,
      "step": 6727
    },
    {
      "epoch": 6.415367126464844e-06,
      "step": 6727,
      "training_step_time": 17.106385231018066
    },
    {
      "epoch": 6.41632080078125e-06,
      "model_forward_time": 5.608688831329346,
      "step": 6728
    },
    {
      "epoch": 6.41632080078125e-06,
      "step": 6728,
      "training_step_time": 17.075599431991577
    },
    {
      "epoch": 6.417274475097656e-06,
      "model_forward_time": 5.58117413520813,
      "step": 6729
    },
    {
      "epoch": 6.417274475097656e-06,
      "step": 6729,
      "training_step_time": 16.998548984527588
    },
    {
      "epoch": 6.418228149414063e-06,
      "grad_norm": 0.8331114053726196,
      "learning_rate": 1e-05,
      "loss": 0.0746,
      "step": 6730
    },
    {
      "epoch": 6.418228149414063e-06,
      "model_forward_time": 5.577620983123779,
      "step": 6730
    },
    {
      "dynamics_loss_avg": 0.06738923881202936,
      "epoch": 6.418228149414063e-06,
      "step": 6730
    },
    {
      "action_loss_avg": 0.015591937198769302,
      "epoch": 6.418228149414063e-06,
      "step": 6730
    },
    {
      "epoch": 6.418228149414063e-06,
      "step": 6730,
      "training_step_time": 16.993930101394653
    },
    {
      "epoch": 6.419181823730469e-06,
      "model_forward_time": 5.922613143920898,
      "step": 6731
    },
    {
      "epoch": 6.419181823730469e-06,
      "step": 6731,
      "training_step_time": 17.329190015792847
    },
    {
      "epoch": 6.420135498046875e-06,
      "model_forward_time": 5.589725494384766,
      "step": 6732
    },
    {
      "epoch": 6.420135498046875e-06,
      "step": 6732,
      "training_step_time": 17.032543182373047
    },
    {
      "epoch": 6.421089172363281e-06,
      "model_forward_time": 5.563717842102051,
      "step": 6733
    },
    {
      "epoch": 6.421089172363281e-06,
      "step": 6733,
      "training_step_time": 16.977073192596436
    },
    {
      "epoch": 6.4220428466796875e-06,
      "model_forward_time": 5.575334310531616,
      "step": 6734
    },
    {
      "epoch": 6.4220428466796875e-06,
      "step": 6734,
      "training_step_time": 16.983161211013794
    },
    {
      "epoch": 6.422996520996094e-06,
      "model_forward_time": 5.582967519760132,
      "step": 6735
    },
    {
      "epoch": 6.422996520996094e-06,
      "step": 6735,
      "training_step_time": 17.032301902770996
    },
    {
      "epoch": 6.4239501953125e-06,
      "model_forward_time": 5.9841601848602295,
      "step": 6736
    },
    {
      "epoch": 6.4239501953125e-06,
      "step": 6736,
      "training_step_time": 17.41547417640686
    },
    {
      "epoch": 6.424903869628906e-06,
      "model_forward_time": 5.6042516231536865,
      "step": 6737
    },
    {
      "epoch": 6.424903869628906e-06,
      "step": 6737,
      "training_step_time": 17.05216932296753
    },
    {
      "epoch": 6.425857543945313e-06,
      "model_forward_time": 5.607448577880859,
      "step": 6738
    },
    {
      "epoch": 6.425857543945313e-06,
      "step": 6738,
      "training_step_time": 17.044716835021973
    },
    {
      "epoch": 6.426811218261719e-06,
      "model_forward_time": 5.583425045013428,
      "step": 6739
    },
    {
      "epoch": 6.426811218261719e-06,
      "step": 6739,
      "training_step_time": 17.015239477157593
    },
    {
      "epoch": 6.427764892578125e-06,
      "grad_norm": 0.525905966758728,
      "learning_rate": 1e-05,
      "loss": 0.0911,
      "step": 6740
    },
    {
      "epoch": 6.427764892578125e-06,
      "model_forward_time": 5.5957114696502686,
      "step": 6740
    },
    {
      "dynamics_loss_avg": 0.0759669566527009,
      "epoch": 6.427764892578125e-06,
      "step": 6740
    },
    {
      "action_loss_avg": 0.013430748600512743,
      "epoch": 6.427764892578125e-06,
      "step": 6740
    },
    {
      "epoch": 6.427764892578125e-06,
      "step": 6740,
      "training_step_time": 17.050231456756592
    },
    {
      "epoch": 6.428718566894531e-06,
      "model_forward_time": 5.599209308624268,
      "step": 6741
    },
    {
      "epoch": 6.428718566894531e-06,
      "step": 6741,
      "training_step_time": 17.44092631340027
    },
    {
      "epoch": 6.4296722412109375e-06,
      "model_forward_time": 5.617715120315552,
      "step": 6742
    },
    {
      "epoch": 6.4296722412109375e-06,
      "step": 6742,
      "training_step_time": 17.135898113250732
    },
    {
      "epoch": 6.430625915527344e-06,
      "model_forward_time": 5.627557754516602,
      "step": 6743
    },
    {
      "epoch": 6.430625915527344e-06,
      "step": 6743,
      "training_step_time": 17.116529941558838
    },
    {
      "epoch": 6.43157958984375e-06,
      "model_forward_time": 5.581535577774048,
      "step": 6744
    },
    {
      "epoch": 6.43157958984375e-06,
      "step": 6744,
      "training_step_time": 17.01114296913147
    },
    {
      "epoch": 6.432533264160156e-06,
      "model_forward_time": 5.571968078613281,
      "step": 6745
    },
    {
      "epoch": 6.432533264160156e-06,
      "step": 6745,
      "training_step_time": 16.9804527759552
    },
    {
      "epoch": 6.433486938476563e-06,
      "model_forward_time": 5.562256097793579,
      "step": 6746
    },
    {
      "epoch": 6.433486938476563e-06,
      "step": 6746,
      "training_step_time": 16.98008966445923
    },
    {
      "epoch": 6.434440612792969e-06,
      "model_forward_time": 5.935925006866455,
      "step": 6747
    },
    {
      "epoch": 6.434440612792969e-06,
      "step": 6747,
      "training_step_time": 17.345327615737915
    },
    {
      "epoch": 6.435394287109375e-06,
      "model_forward_time": 5.561076879501343,
      "step": 6748
    },
    {
      "epoch": 6.435394287109375e-06,
      "step": 6748,
      "training_step_time": 16.980955600738525
    },
    {
      "epoch": 6.436347961425781e-06,
      "model_forward_time": 5.569065809249878,
      "step": 6749
    },
    {
      "epoch": 6.436347961425781e-06,
      "step": 6749,
      "training_step_time": 16.972097158432007
    },
    {
      "epoch": 6.4373016357421875e-06,
      "grad_norm": 0.9586470723152161,
      "learning_rate": 1e-05,
      "loss": 0.0654,
      "step": 6750
    },
    {
      "epoch": 6.4373016357421875e-06,
      "model_forward_time": 5.585708379745483,
      "step": 6750
    },
    {
      "dynamics_loss_avg": 0.051977393589913844,
      "epoch": 6.4373016357421875e-06,
      "step": 6750
    },
    {
      "action_loss_avg": 0.008629066764842718,
      "epoch": 6.4373016357421875e-06,
      "step": 6750
    },
    {
      "epoch": 6.4373016357421875e-06,
      "step": 6750,
      "training_step_time": 17.00504732131958
    },
    {
      "epoch": 6.438255310058594e-06,
      "model_forward_time": 5.577987909317017,
      "step": 6751
    },
    {
      "epoch": 6.438255310058594e-06,
      "step": 6751,
      "training_step_time": 17.021395921707153
    },
    {
      "epoch": 6.439208984375e-06,
      "model_forward_time": 5.9463372230529785,
      "step": 6752
    },
    {
      "epoch": 6.439208984375e-06,
      "step": 6752,
      "training_step_time": 17.42065739631653
    },
    {
      "epoch": 6.440162658691406e-06,
      "model_forward_time": 5.617140531539917,
      "step": 6753
    },
    {
      "epoch": 6.440162658691406e-06,
      "step": 6753,
      "training_step_time": 17.122403860092163
    },
    {
      "epoch": 6.441116333007812e-06,
      "model_forward_time": 5.60757303237915,
      "step": 6754
    },
    {
      "epoch": 6.441116333007812e-06,
      "step": 6754,
      "training_step_time": 17.078702449798584
    },
    {
      "epoch": 6.442070007324219e-06,
      "model_forward_time": 5.591200828552246,
      "step": 6755
    },
    {
      "epoch": 6.442070007324219e-06,
      "step": 6755,
      "training_step_time": 17.017914533615112
    },
    {
      "epoch": 6.443023681640625e-06,
      "model_forward_time": 5.568781137466431,
      "step": 6756
    },
    {
      "epoch": 6.443023681640625e-06,
      "step": 6756,
      "training_step_time": 16.991951942443848
    },
    {
      "epoch": 6.443977355957031e-06,
      "model_forward_time": 5.576157093048096,
      "step": 6757
    },
    {
      "epoch": 6.443977355957031e-06,
      "step": 6757,
      "training_step_time": 16.99700379371643
    },
    {
      "epoch": 6.4449310302734375e-06,
      "model_forward_time": 5.9469146728515625,
      "step": 6758
    },
    {
      "epoch": 6.4449310302734375e-06,
      "step": 6758,
      "training_step_time": 17.375534772872925
    },
    {
      "epoch": 6.445884704589844e-06,
      "model_forward_time": 5.565325021743774,
      "step": 6759
    },
    {
      "epoch": 6.445884704589844e-06,
      "step": 6759,
      "training_step_time": 16.990790367126465
    },
    {
      "epoch": 6.44683837890625e-06,
      "grad_norm": 1.3206045627593994,
      "learning_rate": 1e-05,
      "loss": 0.0617,
      "step": 6760
    },
    {
      "epoch": 6.44683837890625e-06,
      "model_forward_time": 5.577420234680176,
      "step": 6760
    },
    {
      "dynamics_loss_avg": 0.05832892255857587,
      "epoch": 6.44683837890625e-06,
      "step": 6760
    },
    {
      "action_loss_avg": 0.009660843852907419,
      "epoch": 6.44683837890625e-06,
      "step": 6760
    },
    {
      "epoch": 6.44683837890625e-06,
      "step": 6760,
      "training_step_time": 16.99405026435852
    },
    {
      "epoch": 6.447792053222656e-06,
      "model_forward_time": 5.5852978229522705,
      "step": 6761
    },
    {
      "epoch": 6.447792053222656e-06,
      "step": 6761,
      "training_step_time": 17.00394105911255
    },
    {
      "epoch": 6.448745727539062e-06,
      "model_forward_time": 5.595760107040405,
      "step": 6762
    },
    {
      "epoch": 6.448745727539062e-06,
      "step": 6762,
      "training_step_time": 17.076101541519165
    },
    {
      "epoch": 6.449699401855469e-06,
      "model_forward_time": 5.945353984832764,
      "step": 6763
    },
    {
      "epoch": 6.449699401855469e-06,
      "step": 6763,
      "training_step_time": 17.41992425918579
    },
    {
      "epoch": 6.450653076171875e-06,
      "model_forward_time": 5.596885681152344,
      "step": 6764
    },
    {
      "epoch": 6.450653076171875e-06,
      "step": 6764,
      "training_step_time": 17.062706470489502
    },
    {
      "epoch": 6.451606750488281e-06,
      "model_forward_time": 5.595422744750977,
      "step": 6765
    },
    {
      "epoch": 6.451606750488281e-06,
      "step": 6765,
      "training_step_time": 17.071325540542603
    },
    {
      "epoch": 6.4525604248046875e-06,
      "model_forward_time": 5.6070380210876465,
      "step": 6766
    },
    {
      "epoch": 6.4525604248046875e-06,
      "step": 6766,
      "training_step_time": 17.10099148750305
    },
    {
      "epoch": 6.453514099121094e-06,
      "model_forward_time": 5.616955757141113,
      "step": 6767
    },
    {
      "epoch": 6.453514099121094e-06,
      "step": 6767,
      "training_step_time": 17.097053050994873
    },
    {
      "epoch": 6.4544677734375e-06,
      "model_forward_time": 5.602591514587402,
      "step": 6768
    },
    {
      "epoch": 6.4544677734375e-06,
      "step": 6768,
      "training_step_time": 17.054147481918335
    },
    {
      "epoch": 6.455421447753906e-06,
      "model_forward_time": 5.961808204650879,
      "step": 6769
    },
    {
      "epoch": 6.455421447753906e-06,
      "step": 6769,
      "training_step_time": 17.386972188949585
    },
    {
      "epoch": 6.456375122070312e-06,
      "grad_norm": 0.9340012073516846,
      "learning_rate": 1e-05,
      "loss": 0.0763,
      "step": 6770
    },
    {
      "epoch": 6.456375122070312e-06,
      "model_forward_time": 5.583330392837524,
      "step": 6770
    },
    {
      "dynamics_loss_avg": 0.057038495503365996,
      "epoch": 6.456375122070312e-06,
      "step": 6770
    },
    {
      "action_loss_avg": 0.012984172999858856,
      "epoch": 6.456375122070312e-06,
      "step": 6770
    },
    {
      "epoch": 6.456375122070312e-06,
      "step": 6770,
      "training_step_time": 17.030625104904175
    },
    {
      "epoch": 6.457328796386719e-06,
      "model_forward_time": 5.585942268371582,
      "step": 6771
    },
    {
      "epoch": 6.457328796386719e-06,
      "step": 6771,
      "training_step_time": 17.0372314453125
    },
    {
      "epoch": 6.458282470703125e-06,
      "model_forward_time": 5.588453531265259,
      "step": 6772
    },
    {
      "epoch": 6.458282470703125e-06,
      "step": 6772,
      "training_step_time": 17.04980707168579
    },
    {
      "epoch": 6.459236145019531e-06,
      "model_forward_time": 5.607811689376831,
      "step": 6773
    },
    {
      "epoch": 6.459236145019531e-06,
      "step": 6773,
      "training_step_time": 17.09296441078186
    },
    {
      "epoch": 6.4601898193359375e-06,
      "model_forward_time": 5.9757304191589355,
      "step": 6774
    },
    {
      "epoch": 6.4601898193359375e-06,
      "step": 6774,
      "training_step_time": 17.472234964370728
    },
    {
      "epoch": 6.461143493652344e-06,
      "model_forward_time": 5.622589349746704,
      "step": 6775
    },
    {
      "epoch": 6.461143493652344e-06,
      "step": 6775,
      "training_step_time": 17.121715307235718
    },
    {
      "epoch": 6.46209716796875e-06,
      "model_forward_time": 5.602704763412476,
      "step": 6776
    },
    {
      "epoch": 6.46209716796875e-06,
      "step": 6776,
      "training_step_time": 17.06771445274353
    },
    {
      "epoch": 6.463050842285156e-06,
      "model_forward_time": 5.585953235626221,
      "step": 6777
    },
    {
      "epoch": 6.463050842285156e-06,
      "step": 6777,
      "training_step_time": 17.00961399078369
    },
    {
      "epoch": 6.464004516601562e-06,
      "model_forward_time": 5.576988697052002,
      "step": 6778
    },
    {
      "epoch": 6.464004516601562e-06,
      "step": 6778,
      "training_step_time": 17.00580620765686
    },
    {
      "epoch": 6.464958190917969e-06,
      "model_forward_time": 5.5752952098846436,
      "step": 6779
    },
    {
      "epoch": 6.464958190917969e-06,
      "step": 6779,
      "training_step_time": 17.353055477142334
    },
    {
      "epoch": 6.465911865234375e-06,
      "grad_norm": 1.007070779800415,
      "learning_rate": 1e-05,
      "loss": 0.0575,
      "step": 6780
    },
    {
      "epoch": 6.465911865234375e-06,
      "model_forward_time": 5.591593980789185,
      "step": 6780
    },
    {
      "dynamics_loss_avg": 0.049645808339118955,
      "epoch": 6.465911865234375e-06,
      "step": 6780
    },
    {
      "action_loss_avg": 0.00972118959762156,
      "epoch": 6.465911865234375e-06,
      "step": 6780
    },
    {
      "epoch": 6.465911865234375e-06,
      "step": 6780,
      "training_step_time": 17.036880254745483
    },
    {
      "epoch": 6.466865539550781e-06,
      "model_forward_time": 5.594985723495483,
      "step": 6781
    },
    {
      "epoch": 6.466865539550781e-06,
      "step": 6781,
      "training_step_time": 17.044912099838257
    },
    {
      "epoch": 6.4678192138671875e-06,
      "model_forward_time": 5.603275775909424,
      "step": 6782
    },
    {
      "epoch": 6.4678192138671875e-06,
      "step": 6782,
      "training_step_time": 17.087289094924927
    },
    {
      "epoch": 6.468772888183594e-06,
      "model_forward_time": 5.601302146911621,
      "step": 6783
    },
    {
      "epoch": 6.468772888183594e-06,
      "step": 6783,
      "training_step_time": 17.092323303222656
    },
    {
      "epoch": 6.4697265625e-06,
      "model_forward_time": 5.597143173217773,
      "step": 6784
    },
    {
      "epoch": 6.4697265625e-06,
      "step": 6784,
      "training_step_time": 17.03926706314087
    },
    {
      "epoch": 6.470680236816406e-06,
      "model_forward_time": 5.946936845779419,
      "step": 6785
    },
    {
      "epoch": 6.470680236816406e-06,
      "step": 6785,
      "training_step_time": 17.3694429397583
    },
    {
      "epoch": 6.471633911132812e-06,
      "model_forward_time": 5.58107852935791,
      "step": 6786
    },
    {
      "epoch": 6.471633911132812e-06,
      "step": 6786,
      "training_step_time": 17.023170709609985
    },
    {
      "epoch": 6.472587585449219e-06,
      "model_forward_time": 5.598644495010376,
      "step": 6787
    },
    {
      "epoch": 6.472587585449219e-06,
      "step": 6787,
      "training_step_time": 17.06569218635559
    },
    {
      "epoch": 6.473541259765625e-06,
      "model_forward_time": 5.6233439445495605,
      "step": 6788
    },
    {
      "epoch": 6.473541259765625e-06,
      "step": 6788,
      "training_step_time": 17.109408617019653
    },
    {
      "epoch": 6.474494934082031e-06,
      "model_forward_time": 5.615402460098267,
      "step": 6789
    },
    {
      "epoch": 6.474494934082031e-06,
      "step": 6789,
      "training_step_time": 17.099969387054443
    },
    {
      "epoch": 6.4754486083984375e-06,
      "grad_norm": 0.7142031788825989,
      "learning_rate": 1e-05,
      "loss": 0.0542,
      "step": 6790
    },
    {
      "epoch": 6.4754486083984375e-06,
      "model_forward_time": 5.984035491943359,
      "step": 6790
    },
    {
      "dynamics_loss_avg": 0.04663256797939539,
      "epoch": 6.4754486083984375e-06,
      "step": 6790
    },
    {
      "action_loss_avg": 0.006770774465985596,
      "epoch": 6.4754486083984375e-06,
      "step": 6790
    },
    {
      "epoch": 6.4754486083984375e-06,
      "step": 6790,
      "training_step_time": 17.44462275505066
    },
    {
      "epoch": 6.476402282714844e-06,
      "model_forward_time": 5.590092897415161,
      "step": 6791
    },
    {
      "epoch": 6.476402282714844e-06,
      "step": 6791,
      "training_step_time": 17.029064416885376
    },
    {
      "epoch": 6.47735595703125e-06,
      "model_forward_time": 5.578707695007324,
      "step": 6792
    },
    {
      "epoch": 6.47735595703125e-06,
      "step": 6792,
      "training_step_time": 17.009993076324463
    },
    {
      "epoch": 6.478309631347656e-06,
      "model_forward_time": 5.5838940143585205,
      "step": 6793
    },
    {
      "epoch": 6.478309631347656e-06,
      "step": 6793,
      "training_step_time": 17.035454988479614
    },
    {
      "epoch": 6.479263305664062e-06,
      "model_forward_time": 5.5965728759765625,
      "step": 6794
    },
    {
      "epoch": 6.479263305664062e-06,
      "step": 6794,
      "training_step_time": 17.055997133255005
    },
    {
      "epoch": 6.480216979980469e-06,
      "model_forward_time": 5.595699787139893,
      "step": 6795
    },
    {
      "epoch": 6.480216979980469e-06,
      "step": 6795,
      "training_step_time": 17.084322214126587
    },
    {
      "epoch": 6.481170654296875e-06,
      "model_forward_time": 5.961378574371338,
      "step": 6796
    },
    {
      "epoch": 6.481170654296875e-06,
      "step": 6796,
      "training_step_time": 17.460718154907227
    },
    {
      "epoch": 6.482124328613281e-06,
      "model_forward_time": 5.616740465164185,
      "step": 6797
    },
    {
      "epoch": 6.482124328613281e-06,
      "step": 6797,
      "training_step_time": 17.112770557403564
    },
    {
      "epoch": 6.4830780029296875e-06,
      "model_forward_time": 5.612551689147949,
      "step": 6798
    },
    {
      "epoch": 6.4830780029296875e-06,
      "step": 6798,
      "training_step_time": 17.104865789413452
    },
    {
      "epoch": 6.484031677246094e-06,
      "model_forward_time": 5.614951848983765,
      "step": 6799
    },
    {
      "epoch": 6.484031677246094e-06,
      "step": 6799,
      "training_step_time": 17.112506866455078
    },
    {
      "epoch": 6.4849853515625e-06,
      "grad_norm": 1.3053120374679565,
      "learning_rate": 1e-05,
      "loss": 0.0639,
      "step": 6800
    },
    {
      "epoch": 6.4849853515625e-06,
      "model_forward_time": 5.617099046707153,
      "step": 6800
    },
    {
      "dynamics_loss_avg": 0.05860453229397535,
      "epoch": 6.4849853515625e-06,
      "step": 6800
    },
    {
      "action_loss_avg": 0.01020271354354918,
      "epoch": 6.4849853515625e-06,
      "step": 6800
    },
    {
      "epoch": 6.4849853515625e-06,
      "step": 6800,
      "training_step_time": 17.106399536132812
    },
    {
      "epoch": 6.485939025878906e-06,
      "model_forward_time": 5.988430500030518,
      "step": 6801
    },
    {
      "epoch": 6.485939025878906e-06,
      "step": 6801,
      "training_step_time": 17.447702646255493
    },
    {
      "epoch": 6.486892700195312e-06,
      "model_forward_time": 5.605662107467651,
      "step": 6802
    },
    {
      "epoch": 6.486892700195312e-06,
      "step": 6802,
      "training_step_time": 17.07374358177185
    },
    {
      "epoch": 6.487846374511719e-06,
      "model_forward_time": 5.605489253997803,
      "step": 6803
    },
    {
      "epoch": 6.487846374511719e-06,
      "step": 6803,
      "training_step_time": 17.079668045043945
    },
    {
      "epoch": 6.488800048828125e-06,
      "model_forward_time": 5.598971605300903,
      "step": 6804
    },
    {
      "epoch": 6.488800048828125e-06,
      "step": 6804,
      "training_step_time": 17.064894914627075
    },
    {
      "epoch": 6.489753723144531e-06,
      "model_forward_time": 5.628710031509399,
      "step": 6805
    },
    {
      "epoch": 6.489753723144531e-06,
      "step": 6805,
      "training_step_time": 17.084981679916382
    },
    {
      "epoch": 6.4907073974609375e-06,
      "model_forward_time": 5.590590238571167,
      "step": 6806
    },
    {
      "epoch": 6.4907073974609375e-06,
      "step": 6806,
      "training_step_time": 17.03167748451233
    },
    {
      "epoch": 6.491661071777344e-06,
      "model_forward_time": 5.929523944854736,
      "step": 6807
    },
    {
      "epoch": 6.491661071777344e-06,
      "step": 6807,
      "training_step_time": 17.358319997787476
    },
    {
      "epoch": 6.49261474609375e-06,
      "model_forward_time": 5.581099987030029,
      "step": 6808
    },
    {
      "epoch": 6.49261474609375e-06,
      "step": 6808,
      "training_step_time": 17.017119884490967
    },
    {
      "epoch": 6.493568420410156e-06,
      "model_forward_time": 5.626493215560913,
      "step": 6809
    },
    {
      "epoch": 6.493568420410156e-06,
      "step": 6809,
      "training_step_time": 17.1063392162323
    },
    {
      "epoch": 6.494522094726562e-06,
      "grad_norm": 0.9548435807228088,
      "learning_rate": 1e-05,
      "loss": 0.0663,
      "step": 6810
    },
    {
      "epoch": 6.494522094726562e-06,
      "model_forward_time": 5.617377996444702,
      "step": 6810
    },
    {
      "dynamics_loss_avg": 0.05101943369954824,
      "epoch": 6.494522094726562e-06,
      "step": 6810
    },
    {
      "action_loss_avg": 0.010512051358819007,
      "epoch": 6.494522094726562e-06,
      "step": 6810
    },
    {
      "epoch": 6.494522094726562e-06,
      "step": 6810,
      "training_step_time": 17.104909420013428
    },
    {
      "epoch": 6.495475769042969e-06,
      "model_forward_time": 5.616028785705566,
      "step": 6811
    },
    {
      "epoch": 6.495475769042969e-06,
      "step": 6811,
      "training_step_time": 17.096270084381104
    },
    {
      "epoch": 6.496429443359375e-06,
      "model_forward_time": 5.974275827407837,
      "step": 6812
    },
    {
      "epoch": 6.496429443359375e-06,
      "step": 6812,
      "training_step_time": 17.44272232055664
    },
    {
      "epoch": 6.497383117675781e-06,
      "model_forward_time": 5.604989767074585,
      "step": 6813
    },
    {
      "epoch": 6.497383117675781e-06,
      "step": 6813,
      "training_step_time": 17.07646155357361
    },
    {
      "epoch": 6.4983367919921876e-06,
      "model_forward_time": 5.590484619140625,
      "step": 6814
    },
    {
      "epoch": 6.4983367919921876e-06,
      "step": 6814,
      "training_step_time": 17.047236680984497
    },
    {
      "epoch": 6.499290466308594e-06,
      "model_forward_time": 5.584670543670654,
      "step": 6815
    },
    {
      "epoch": 6.499290466308594e-06,
      "step": 6815,
      "training_step_time": 17.002007246017456
    },
    {
      "epoch": 6.500244140625e-06,
      "model_forward_time": 5.583272695541382,
      "step": 6816
    },
    {
      "epoch": 6.500244140625e-06,
      "step": 6816,
      "training_step_time": 17.035259008407593
    },
    {
      "epoch": 6.501197814941406e-06,
      "model_forward_time": 5.603145599365234,
      "step": 6817
    },
    {
      "epoch": 6.501197814941406e-06,
      "step": 6817,
      "training_step_time": 17.08976674079895
    },
    {
      "epoch": 6.502151489257812e-06,
      "model_forward_time": 5.967867136001587,
      "step": 6818
    },
    {
      "epoch": 6.502151489257812e-06,
      "step": 6818,
      "training_step_time": 17.48278570175171
    },
    {
      "epoch": 6.503105163574219e-06,
      "model_forward_time": 5.634293556213379,
      "step": 6819
    },
    {
      "epoch": 6.503105163574219e-06,
      "step": 6819,
      "training_step_time": 17.11211633682251
    },
    {
      "epoch": 6.504058837890625e-06,
      "grad_norm": 2.066868782043457,
      "learning_rate": 1e-05,
      "loss": 0.0792,
      "step": 6820
    },
    {
      "epoch": 6.504058837890625e-06,
      "model_forward_time": 5.596088886260986,
      "step": 6820
    },
    {
      "dynamics_loss_avg": 0.06980804540216923,
      "epoch": 6.504058837890625e-06,
      "step": 6820
    },
    {
      "action_loss_avg": 0.014498193049803376,
      "epoch": 6.504058837890625e-06,
      "step": 6820
    },
    {
      "epoch": 6.504058837890625e-06,
      "step": 6820,
      "training_step_time": 17.06687092781067
    },
    {
      "epoch": 6.505012512207031e-06,
      "model_forward_time": 5.572895765304565,
      "step": 6821
    },
    {
      "epoch": 6.505012512207031e-06,
      "step": 6821,
      "training_step_time": 16.979050636291504
    },
    {
      "epoch": 6.5059661865234376e-06,
      "model_forward_time": 5.581276893615723,
      "step": 6822
    },
    {
      "epoch": 6.5059661865234376e-06,
      "step": 6822,
      "training_step_time": 16.989890336990356
    },
    {
      "epoch": 6.506919860839844e-06,
      "model_forward_time": 5.941090822219849,
      "step": 6823
    },
    {
      "epoch": 6.506919860839844e-06,
      "step": 6823,
      "training_step_time": 17.389573335647583
    },
    {
      "epoch": 6.50787353515625e-06,
      "model_forward_time": 5.587853193283081,
      "step": 6824
    },
    {
      "epoch": 6.50787353515625e-06,
      "step": 6824,
      "training_step_time": 17.037777423858643
    },
    {
      "epoch": 6.508827209472656e-06,
      "model_forward_time": 5.5944132804870605,
      "step": 6825
    },
    {
      "epoch": 6.508827209472656e-06,
      "step": 6825,
      "training_step_time": 17.06973910331726
    },
    {
      "epoch": 6.509780883789062e-06,
      "model_forward_time": 5.609225034713745,
      "step": 6826
    },
    {
      "epoch": 6.509780883789062e-06,
      "step": 6826,
      "training_step_time": 17.10028862953186
    },
    {
      "epoch": 6.510734558105469e-06,
      "model_forward_time": 5.609283447265625,
      "step": 6827
    },
    {
      "epoch": 6.510734558105469e-06,
      "step": 6827,
      "training_step_time": 17.092601776123047
    },
    {
      "epoch": 6.511688232421875e-06,
      "model_forward_time": 5.9802656173706055,
      "step": 6828
    },
    {
      "epoch": 6.511688232421875e-06,
      "step": 6828,
      "training_step_time": 17.44262957572937
    },
    {
      "epoch": 6.512641906738281e-06,
      "model_forward_time": 5.586698532104492,
      "step": 6829
    },
    {
      "epoch": 6.512641906738281e-06,
      "step": 6829,
      "training_step_time": 16.99602198600769
    },
    {
      "epoch": 6.5135955810546876e-06,
      "grad_norm": 0.2928515076637268,
      "learning_rate": 1e-05,
      "loss": 0.0842,
      "step": 6830
    },
    {
      "epoch": 6.5135955810546876e-06,
      "model_forward_time": 5.577526330947876,
      "step": 6830
    },
    {
      "dynamics_loss_avg": 0.06897360272705555,
      "epoch": 6.5135955810546876e-06,
      "step": 6830
    },
    {
      "action_loss_avg": 0.012617792445234954,
      "epoch": 6.5135955810546876e-06,
      "step": 6830
    },
    {
      "epoch": 6.5135955810546876e-06,
      "step": 6830,
      "training_step_time": 16.99027967453003
    },
    {
      "epoch": 6.514549255371094e-06,
      "model_forward_time": 5.599365234375,
      "step": 6831
    },
    {
      "epoch": 6.514549255371094e-06,
      "step": 6831,
      "training_step_time": 17.04891061782837
    },
    {
      "epoch": 6.5155029296875e-06,
      "model_forward_time": 5.577997207641602,
      "step": 6832
    },
    {
      "epoch": 6.5155029296875e-06,
      "step": 6832,
      "training_step_time": 17.01774835586548
    },
    {
      "epoch": 6.5164566040039064e-06,
      "model_forward_time": 5.588420391082764,
      "step": 6833
    },
    {
      "epoch": 6.5164566040039064e-06,
      "step": 6833,
      "training_step_time": 17.0140323638916
    },
    {
      "epoch": 6.517410278320312e-06,
      "model_forward_time": 5.967989683151245,
      "step": 6834
    },
    {
      "epoch": 6.517410278320312e-06,
      "step": 6834,
      "training_step_time": 17.43933391571045
    },
    {
      "epoch": 6.518363952636719e-06,
      "model_forward_time": 5.606192111968994,
      "step": 6835
    },
    {
      "epoch": 6.518363952636719e-06,
      "step": 6835,
      "training_step_time": 17.06988835334778
    },
    {
      "epoch": 6.519317626953125e-06,
      "model_forward_time": 5.612372398376465,
      "step": 6836
    },
    {
      "epoch": 6.519317626953125e-06,
      "step": 6836,
      "training_step_time": 17.123454093933105
    },
    {
      "epoch": 6.520271301269531e-06,
      "model_forward_time": 5.60953164100647,
      "step": 6837
    },
    {
      "epoch": 6.520271301269531e-06,
      "step": 6837,
      "training_step_time": 17.113184928894043
    },
    {
      "epoch": 6.5212249755859376e-06,
      "model_forward_time": 5.629391431808472,
      "step": 6838
    },
    {
      "epoch": 6.5212249755859376e-06,
      "step": 6838,
      "training_step_time": 17.130155563354492
    },
    {
      "epoch": 6.522178649902344e-06,
      "model_forward_time": 5.612154960632324,
      "step": 6839
    },
    {
      "epoch": 6.522178649902344e-06,
      "step": 6839,
      "training_step_time": 17.458709239959717
    },
    {
      "epoch": 6.52313232421875e-06,
      "grad_norm": 1.543082356452942,
      "learning_rate": 1e-05,
      "loss": 0.0733,
      "step": 6840
    },
    {
      "epoch": 6.52313232421875e-06,
      "model_forward_time": 5.6074817180633545,
      "step": 6840
    },
    {
      "dynamics_loss_avg": 0.06361529547721148,
      "epoch": 6.52313232421875e-06,
      "step": 6840
    },
    {
      "action_loss_avg": 0.011531313974410295,
      "epoch": 6.52313232421875e-06,
      "step": 6840
    },
    {
      "epoch": 6.52313232421875e-06,
      "step": 6840,
      "training_step_time": 17.081730604171753
    },
    {
      "epoch": 6.5240859985351564e-06,
      "model_forward_time": 5.598237752914429,
      "step": 6841
    },
    {
      "epoch": 6.5240859985351564e-06,
      "step": 6841,
      "training_step_time": 17.047795295715332
    },
    {
      "epoch": 6.525039672851562e-06,
      "model_forward_time": 5.589974641799927,
      "step": 6842
    },
    {
      "epoch": 6.525039672851562e-06,
      "step": 6842,
      "training_step_time": 17.023704051971436
    },
    {
      "epoch": 6.525993347167969e-06,
      "model_forward_time": 5.595945358276367,
      "step": 6843
    },
    {
      "epoch": 6.525993347167969e-06,
      "step": 6843,
      "training_step_time": 17.066284656524658
    },
    {
      "epoch": 6.526947021484375e-06,
      "model_forward_time": 5.597853422164917,
      "step": 6844
    },
    {
      "epoch": 6.526947021484375e-06,
      "step": 6844,
      "training_step_time": 17.078412532806396
    },
    {
      "epoch": 6.527900695800781e-06,
      "model_forward_time": 5.981230735778809,
      "step": 6845
    },
    {
      "epoch": 6.527900695800781e-06,
      "step": 6845,
      "training_step_time": 17.48316478729248
    },
    {
      "epoch": 6.528854370117188e-06,
      "model_forward_time": 5.621142387390137,
      "step": 6846
    },
    {
      "epoch": 6.528854370117188e-06,
      "step": 6846,
      "training_step_time": 17.124652862548828
    },
    {
      "epoch": 6.529808044433594e-06,
      "model_forward_time": 5.619243621826172,
      "step": 6847
    },
    {
      "epoch": 6.529808044433594e-06,
      "step": 6847,
      "training_step_time": 17.12074065208435
    },
    {
      "epoch": 6.53076171875e-06,
      "model_forward_time": 5.6130101680755615,
      "step": 6848
    },
    {
      "epoch": 6.53076171875e-06,
      "step": 6848,
      "training_step_time": 17.109137773513794
    },
    {
      "epoch": 6.5317153930664064e-06,
      "model_forward_time": 5.611604690551758,
      "step": 6849
    },
    {
      "epoch": 6.5317153930664064e-06,
      "step": 6849,
      "training_step_time": 17.111189603805542
    },
    {
      "epoch": 6.532669067382812e-06,
      "grad_norm": 0.760903537273407,
      "learning_rate": 1e-05,
      "loss": 0.074,
      "step": 6850
    },
    {
      "epoch": 6.532669067382812e-06,
      "model_forward_time": 5.616502046585083,
      "step": 6850
    },
    {
      "dynamics_loss_avg": 0.05855737971141935,
      "epoch": 6.532669067382812e-06,
      "step": 6850
    },
    {
      "action_loss_avg": 0.007493532076478004,
      "epoch": 6.532669067382812e-06,
      "step": 6850
    },
    {
      "epoch": 6.532669067382812e-06,
      "step": 6850,
      "training_step_time": 17.452046871185303
    },
    {
      "epoch": 6.533622741699219e-06,
      "model_forward_time": 5.608633518218994,
      "step": 6851
    },
    {
      "epoch": 6.533622741699219e-06,
      "step": 6851,
      "training_step_time": 17.09425377845764
    },
    {
      "epoch": 6.534576416015625e-06,
      "model_forward_time": 5.608579158782959,
      "step": 6852
    },
    {
      "epoch": 6.534576416015625e-06,
      "step": 6852,
      "training_step_time": 17.072059869766235
    },
    {
      "epoch": 6.535530090332031e-06,
      "model_forward_time": 5.59853720664978,
      "step": 6853
    },
    {
      "epoch": 6.535530090332031e-06,
      "step": 6853,
      "training_step_time": 17.052066564559937
    },
    {
      "epoch": 6.536483764648438e-06,
      "model_forward_time": 5.585346460342407,
      "step": 6854
    },
    {
      "epoch": 6.536483764648438e-06,
      "step": 6854,
      "training_step_time": 17.0127272605896
    },
    {
      "epoch": 6.537437438964844e-06,
      "model_forward_time": 5.594056129455566,
      "step": 6855
    },
    {
      "epoch": 6.537437438964844e-06,
      "step": 6855,
      "training_step_time": 17.032011032104492
    },
    {
      "epoch": 6.53839111328125e-06,
      "model_forward_time": 5.965221881866455,
      "step": 6856
    },
    {
      "epoch": 6.53839111328125e-06,
      "step": 6856,
      "training_step_time": 17.443535089492798
    },
    {
      "epoch": 6.5393447875976564e-06,
      "model_forward_time": 5.612570524215698,
      "step": 6857
    },
    {
      "epoch": 6.5393447875976564e-06,
      "step": 6857,
      "training_step_time": 17.111026525497437
    },
    {
      "epoch": 6.540298461914062e-06,
      "model_forward_time": 5.619778871536255,
      "step": 6858
    },
    {
      "epoch": 6.540298461914062e-06,
      "step": 6858,
      "training_step_time": 17.081021308898926
    },
    {
      "epoch": 6.541252136230469e-06,
      "model_forward_time": 5.592491626739502,
      "step": 6859
    },
    {
      "epoch": 6.541252136230469e-06,
      "step": 6859,
      "training_step_time": 17.002975940704346
    },
    {
      "epoch": 6.542205810546875e-06,
      "grad_norm": 0.7524819374084473,
      "learning_rate": 1e-05,
      "loss": 0.0632,
      "step": 6860
    },
    {
      "epoch": 6.542205810546875e-06,
      "model_forward_time": 5.57417106628418,
      "step": 6860
    },
    {
      "dynamics_loss_avg": 0.05647336598485708,
      "epoch": 6.542205810546875e-06,
      "step": 6860
    },
    {
      "action_loss_avg": 0.010979152331128717,
      "epoch": 6.542205810546875e-06,
      "step": 6860
    },
    {
      "epoch": 6.542205810546875e-06,
      "step": 6860,
      "training_step_time": 17.000619649887085
    },
    {
      "epoch": 6.543159484863281e-06,
      "model_forward_time": 5.926056146621704,
      "step": 6861
    },
    {
      "epoch": 6.543159484863281e-06,
      "step": 6861,
      "training_step_time": 17.336888551712036
    },
    {
      "epoch": 6.544113159179688e-06,
      "model_forward_time": 5.566840171813965,
      "step": 6862
    },
    {
      "epoch": 6.544113159179688e-06,
      "step": 6862,
      "training_step_time": 16.998621463775635
    },
    {
      "epoch": 6.545066833496094e-06,
      "model_forward_time": 5.562310695648193,
      "step": 6863
    },
    {
      "epoch": 6.545066833496094e-06,
      "step": 6863,
      "training_step_time": 16.980716466903687
    },
    {
      "epoch": 6.5460205078125e-06,
      "model_forward_time": 5.59089732170105,
      "step": 6864
    },
    {
      "epoch": 6.5460205078125e-06,
      "step": 6864,
      "training_step_time": 16.9923312664032
    },
    {
      "epoch": 6.5469741821289065e-06,
      "model_forward_time": 5.574099540710449,
      "step": 6865
    },
    {
      "epoch": 6.5469741821289065e-06,
      "step": 6865,
      "training_step_time": 17.00496244430542
    },
    {
      "epoch": 6.547927856445312e-06,
      "model_forward_time": 5.597386121749878,
      "step": 6866
    },
    {
      "epoch": 6.547927856445312e-06,
      "step": 6866,
      "training_step_time": 17.075143337249756
    },
    {
      "epoch": 6.548881530761719e-06,
      "model_forward_time": 5.98760199546814,
      "step": 6867
    },
    {
      "epoch": 6.548881530761719e-06,
      "step": 6867,
      "training_step_time": 17.479701042175293
    },
    {
      "epoch": 6.549835205078125e-06,
      "model_forward_time": 5.609875202178955,
      "step": 6868
    },
    {
      "epoch": 6.549835205078125e-06,
      "step": 6868,
      "training_step_time": 17.073447465896606
    },
    {
      "epoch": 6.550788879394531e-06,
      "model_forward_time": 5.617373466491699,
      "step": 6869
    },
    {
      "epoch": 6.550788879394531e-06,
      "step": 6869,
      "training_step_time": 17.075243949890137
    },
    {
      "epoch": 6.551742553710938e-06,
      "grad_norm": 1.2226706743240356,
      "learning_rate": 1e-05,
      "loss": 0.0714,
      "step": 6870
    },
    {
      "epoch": 6.551742553710938e-06,
      "model_forward_time": 5.590741157531738,
      "step": 6870
    },
    {
      "dynamics_loss_avg": 0.06320814937353134,
      "epoch": 6.551742553710938e-06,
      "step": 6870
    },
    {
      "action_loss_avg": 0.014040688332170249,
      "epoch": 6.551742553710938e-06,
      "step": 6870
    },
    {
      "epoch": 6.551742553710938e-06,
      "step": 6870,
      "training_step_time": 17.05043125152588
    },
    {
      "epoch": 6.552696228027344e-06,
      "model_forward_time": 5.595445156097412,
      "step": 6871
    },
    {
      "epoch": 6.552696228027344e-06,
      "step": 6871,
      "training_step_time": 17.04873776435852
    },
    {
      "epoch": 6.55364990234375e-06,
      "model_forward_time": 5.596417665481567,
      "step": 6872
    },
    {
      "epoch": 6.55364990234375e-06,
      "step": 6872,
      "training_step_time": 17.421116590499878
    },
    {
      "epoch": 6.5546035766601565e-06,
      "model_forward_time": 5.609679222106934,
      "step": 6873
    },
    {
      "epoch": 6.5546035766601565e-06,
      "step": 6873,
      "training_step_time": 17.07578730583191
    },
    {
      "epoch": 6.555557250976562e-06,
      "model_forward_time": 5.592310190200806,
      "step": 6874
    },
    {
      "epoch": 6.555557250976562e-06,
      "step": 6874,
      "training_step_time": 17.040756464004517
    },
    {
      "epoch": 6.556510925292969e-06,
      "model_forward_time": 5.593535423278809,
      "step": 6875
    },
    {
      "epoch": 6.556510925292969e-06,
      "step": 6875,
      "training_step_time": 17.04049563407898
    },
    {
      "epoch": 6.557464599609375e-06,
      "model_forward_time": 5.5953779220581055,
      "step": 6876
    },
    {
      "epoch": 6.557464599609375e-06,
      "step": 6876,
      "training_step_time": 17.03103280067444
    },
    {
      "epoch": 6.558418273925781e-06,
      "model_forward_time": 5.600257396697998,
      "step": 6877
    },
    {
      "epoch": 6.558418273925781e-06,
      "step": 6877,
      "training_step_time": 17.06656503677368
    },
    {
      "epoch": 6.559371948242188e-06,
      "model_forward_time": 5.979008913040161,
      "step": 6878
    },
    {
      "epoch": 6.559371948242188e-06,
      "step": 6878,
      "training_step_time": 17.44919753074646
    },
    {
      "epoch": 6.560325622558593e-06,
      "model_forward_time": 5.610623121261597,
      "step": 6879
    },
    {
      "epoch": 6.560325622558593e-06,
      "step": 6879,
      "training_step_time": 17.08924388885498
    },
    {
      "epoch": 6.561279296875e-06,
      "grad_norm": 0.7764707803726196,
      "learning_rate": 1e-05,
      "loss": 0.0868,
      "step": 6880
    },
    {
      "epoch": 6.561279296875e-06,
      "model_forward_time": 5.608335018157959,
      "step": 6880
    },
    {
      "dynamics_loss_avg": 0.07225755788385868,
      "epoch": 6.561279296875e-06,
      "step": 6880
    },
    {
      "action_loss_avg": 0.011855848925188183,
      "epoch": 6.561279296875e-06,
      "step": 6880
    },
    {
      "epoch": 6.561279296875e-06,
      "step": 6880,
      "training_step_time": 17.113174200057983
    },
    {
      "epoch": 6.5622329711914065e-06,
      "model_forward_time": 5.612809658050537,
      "step": 6881
    },
    {
      "epoch": 6.5622329711914065e-06,
      "step": 6881,
      "training_step_time": 17.100322723388672
    },
    {
      "epoch": 6.563186645507812e-06,
      "model_forward_time": 5.603870153427124,
      "step": 6882
    },
    {
      "epoch": 6.563186645507812e-06,
      "step": 6882,
      "training_step_time": 17.097986221313477
    },
    {
      "epoch": 6.564140319824219e-06,
      "model_forward_time": 5.613147258758545,
      "step": 6883
    },
    {
      "epoch": 6.564140319824219e-06,
      "step": 6883,
      "training_step_time": 17.4457848072052
    },
    {
      "epoch": 6.565093994140625e-06,
      "model_forward_time": 5.604398250579834,
      "step": 6884
    },
    {
      "epoch": 6.565093994140625e-06,
      "step": 6884,
      "training_step_time": 17.0860595703125
    },
    {
      "epoch": 6.566047668457031e-06,
      "model_forward_time": 5.606031894683838,
      "step": 6885
    },
    {
      "epoch": 6.566047668457031e-06,
      "step": 6885,
      "training_step_time": 17.097805738449097
    },
    {
      "epoch": 6.567001342773438e-06,
      "model_forward_time": 5.613160133361816,
      "step": 6886
    },
    {
      "epoch": 6.567001342773438e-06,
      "step": 6886,
      "training_step_time": 17.113434314727783
    },
    {
      "epoch": 6.567955017089843e-06,
      "model_forward_time": 5.6115663051605225,
      "step": 6887
    },
    {
      "epoch": 6.567955017089843e-06,
      "step": 6887,
      "training_step_time": 17.108235120773315
    },
    {
      "epoch": 6.56890869140625e-06,
      "model_forward_time": 5.606571197509766,
      "step": 6888
    },
    {
      "epoch": 6.56890869140625e-06,
      "step": 6888,
      "training_step_time": 17.06824779510498
    },
    {
      "epoch": 6.5698623657226565e-06,
      "model_forward_time": 5.96764874458313,
      "step": 6889
    },
    {
      "epoch": 6.5698623657226565e-06,
      "step": 6889,
      "training_step_time": 17.392069816589355
    },
    {
      "epoch": 6.570816040039062e-06,
      "grad_norm": 1.8835352659225464,
      "learning_rate": 1e-05,
      "loss": 0.0789,
      "step": 6890
    },
    {
      "epoch": 6.570816040039062e-06,
      "model_forward_time": 5.575963020324707,
      "step": 6890
    },
    {
      "dynamics_loss_avg": 0.0626949418336153,
      "epoch": 6.570816040039062e-06,
      "step": 6890
    },
    {
      "action_loss_avg": 0.01433899300172925,
      "epoch": 6.570816040039062e-06,
      "step": 6890
    },
    {
      "epoch": 6.570816040039062e-06,
      "step": 6890,
      "training_step_time": 16.983502864837646
    },
    {
      "epoch": 6.571769714355469e-06,
      "model_forward_time": 5.571129083633423,
      "step": 6891
    },
    {
      "epoch": 6.571769714355469e-06,
      "step": 6891,
      "training_step_time": 16.98721671104431
    },
    {
      "epoch": 6.572723388671875e-06,
      "model_forward_time": 5.573482513427734,
      "step": 6892
    },
    {
      "epoch": 6.572723388671875e-06,
      "step": 6892,
      "training_step_time": 17.00989294052124
    },
    {
      "epoch": 6.573677062988281e-06,
      "model_forward_time": 5.596082925796509,
      "step": 6893
    },
    {
      "epoch": 6.573677062988281e-06,
      "step": 6893,
      "training_step_time": 17.056484699249268
    },
    {
      "epoch": 6.574630737304688e-06,
      "model_forward_time": 5.956146240234375,
      "step": 6894
    },
    {
      "epoch": 6.574630737304688e-06,
      "step": 6894,
      "training_step_time": 17.445969581604004
    },
    {
      "epoch": 6.575584411621093e-06,
      "model_forward_time": 5.61364221572876,
      "step": 6895
    },
    {
      "epoch": 6.575584411621093e-06,
      "step": 6895,
      "training_step_time": 17.109228134155273
    },
    {
      "epoch": 6.5765380859375e-06,
      "model_forward_time": 5.619814395904541,
      "step": 6896
    },
    {
      "epoch": 6.5765380859375e-06,
      "step": 6896,
      "training_step_time": 17.13146471977234
    },
    {
      "epoch": 6.5774917602539065e-06,
      "model_forward_time": 5.624914884567261,
      "step": 6897
    },
    {
      "epoch": 6.5774917602539065e-06,
      "step": 6897,
      "training_step_time": 17.130309104919434
    },
    {
      "epoch": 6.578445434570312e-06,
      "model_forward_time": 5.616239786148071,
      "step": 6898
    },
    {
      "epoch": 6.578445434570312e-06,
      "step": 6898,
      "training_step_time": 17.113458156585693
    },
    {
      "epoch": 6.579399108886719e-06,
      "model_forward_time": 5.610223770141602,
      "step": 6899
    },
    {
      "epoch": 6.579399108886719e-06,
      "step": 6899,
      "training_step_time": 17.07624101638794
    },
    {
      "epoch": 6.580352783203125e-06,
      "grad_norm": 0.8407197594642639,
      "learning_rate": 1e-05,
      "loss": 0.068,
      "step": 6900
    },
    {
      "epoch": 6.580352783203125e-06,
      "model_forward_time": 5.9670634269714355,
      "step": 6900
    },
    {
      "dynamics_loss_avg": 0.05701112169772386,
      "epoch": 6.580352783203125e-06,
      "step": 6900
    },
    {
      "action_loss_avg": 0.011468032281845808,
      "epoch": 6.580352783203125e-06,
      "step": 6900
    },
    {
      "epoch": 6.580352783203125e-06,
      "step": 6900,
      "training_step_time": 17.382596254348755
    },
    {
      "epoch": 6.581306457519531e-06,
      "model_forward_time": 5.5546979904174805,
      "step": 6901
    },
    {
      "epoch": 6.581306457519531e-06,
      "step": 6901,
      "training_step_time": 16.96906018257141
    },
    {
      "epoch": 6.582260131835938e-06,
      "model_forward_time": 5.568136930465698,
      "step": 6902
    },
    {
      "epoch": 6.582260131835938e-06,
      "step": 6902,
      "training_step_time": 16.984049081802368
    },
    {
      "epoch": 6.583213806152343e-06,
      "model_forward_time": 5.553339719772339,
      "step": 6903
    },
    {
      "epoch": 6.583213806152343e-06,
      "step": 6903,
      "training_step_time": 16.985920667648315
    },
    {
      "epoch": 6.58416748046875e-06,
      "model_forward_time": 5.55693793296814,
      "step": 6904
    },
    {
      "epoch": 6.58416748046875e-06,
      "step": 6904,
      "training_step_time": 16.9817054271698
    },
    {
      "epoch": 6.5851211547851565e-06,
      "model_forward_time": 5.5707995891571045,
      "step": 6905
    },
    {
      "epoch": 6.5851211547851565e-06,
      "step": 6905,
      "training_step_time": 17.33855414390564
    },
    {
      "epoch": 6.586074829101562e-06,
      "model_forward_time": 5.594467639923096,
      "step": 6906
    },
    {
      "epoch": 6.586074829101562e-06,
      "step": 6906,
      "training_step_time": 17.002676486968994
    },
    {
      "epoch": 6.587028503417969e-06,
      "model_forward_time": 5.5904858112335205,
      "step": 6907
    },
    {
      "epoch": 6.587028503417969e-06,
      "step": 6907,
      "training_step_time": 17.031764268875122
    },
    {
      "epoch": 6.587982177734375e-06,
      "model_forward_time": 5.60208535194397,
      "step": 6908
    },
    {
      "epoch": 6.587982177734375e-06,
      "step": 6908,
      "training_step_time": 17.07515859603882
    },
    {
      "epoch": 6.588935852050781e-06,
      "model_forward_time": 5.597370624542236,
      "step": 6909
    },
    {
      "epoch": 6.588935852050781e-06,
      "step": 6909,
      "training_step_time": 17.06002950668335
    },
    {
      "epoch": 6.589889526367188e-06,
      "grad_norm": 0.5994696617126465,
      "learning_rate": 1e-05,
      "loss": 0.0727,
      "step": 6910
    },
    {
      "epoch": 6.589889526367188e-06,
      "model_forward_time": 5.594595432281494,
      "step": 6910
    },
    {
      "dynamics_loss_avg": 0.06375272795557976,
      "epoch": 6.589889526367188e-06,
      "step": 6910
    },
    {
      "action_loss_avg": 0.00948772409465164,
      "epoch": 6.589889526367188e-06,
      "step": 6910
    },
    {
      "epoch": 6.589889526367188e-06,
      "step": 6910,
      "training_step_time": 17.03761649131775
    },
    {
      "epoch": 6.590843200683593e-06,
      "model_forward_time": 5.956763982772827,
      "step": 6911
    },
    {
      "epoch": 6.590843200683593e-06,
      "step": 6911,
      "training_step_time": 17.39580464363098
    },
    {
      "epoch": 6.591796875e-06,
      "model_forward_time": 5.587266206741333,
      "step": 6912
    },
    {
      "epoch": 6.591796875e-06,
      "step": 6912,
      "training_step_time": 17.026791095733643
    },
    {
      "epoch": 6.5927505493164065e-06,
      "model_forward_time": 5.594131708145142,
      "step": 6913
    },
    {
      "epoch": 6.5927505493164065e-06,
      "step": 6913,
      "training_step_time": 17.054516792297363
    },
    {
      "epoch": 6.593704223632812e-06,
      "model_forward_time": 5.610668420791626,
      "step": 6914
    },
    {
      "epoch": 6.593704223632812e-06,
      "step": 6914,
      "training_step_time": 17.092084884643555
    },
    {
      "epoch": 6.594657897949219e-06,
      "model_forward_time": 5.6287596225738525,
      "step": 6915
    },
    {
      "epoch": 6.594657897949219e-06,
      "step": 6915,
      "training_step_time": 17.107784032821655
    },
    {
      "epoch": 6.595611572265625e-06,
      "model_forward_time": 5.5992982387542725,
      "step": 6916
    },
    {
      "epoch": 6.595611572265625e-06,
      "step": 6916,
      "training_step_time": 17.402657985687256
    },
    {
      "epoch": 6.596565246582031e-06,
      "model_forward_time": 5.590715646743774,
      "step": 6917
    },
    {
      "epoch": 6.596565246582031e-06,
      "step": 6917,
      "training_step_time": 17.059309244155884
    },
    {
      "epoch": 6.597518920898438e-06,
      "model_forward_time": 5.594714879989624,
      "step": 6918
    },
    {
      "epoch": 6.597518920898438e-06,
      "step": 6918,
      "training_step_time": 17.045708417892456
    },
    {
      "epoch": 6.598472595214843e-06,
      "model_forward_time": 5.607639789581299,
      "step": 6919
    },
    {
      "epoch": 6.598472595214843e-06,
      "step": 6919,
      "training_step_time": 17.0898380279541
    },
    {
      "epoch": 6.59942626953125e-06,
      "grad_norm": 0.8830990791320801,
      "learning_rate": 1e-05,
      "loss": 0.0649,
      "step": 6920
    },
    {
      "epoch": 6.59942626953125e-06,
      "model_forward_time": 5.60806679725647,
      "step": 6920
    },
    {
      "dynamics_loss_avg": 0.05803310349583626,
      "epoch": 6.59942626953125e-06,
      "step": 6920
    },
    {
      "action_loss_avg": 0.009030172624625266,
      "epoch": 6.59942626953125e-06,
      "step": 6920
    },
    {
      "epoch": 6.59942626953125e-06,
      "step": 6920,
      "training_step_time": 17.099803686141968
    },
    {
      "epoch": 6.6003799438476565e-06,
      "model_forward_time": 5.613924741744995,
      "step": 6921
    },
    {
      "epoch": 6.6003799438476565e-06,
      "step": 6921,
      "training_step_time": 17.107635736465454
    },
    {
      "epoch": 6.601333618164062e-06,
      "model_forward_time": 5.98260235786438,
      "step": 6922
    },
    {
      "epoch": 6.601333618164062e-06,
      "step": 6922,
      "training_step_time": 17.46631622314453
    },
    {
      "epoch": 6.602287292480469e-06,
      "model_forward_time": 5.6219871044158936,
      "step": 6923
    },
    {
      "epoch": 6.602287292480469e-06,
      "step": 6923,
      "training_step_time": 17.11236000061035
    },
    {
      "epoch": 6.603240966796875e-06,
      "model_forward_time": 5.6104912757873535,
      "step": 6924
    },
    {
      "epoch": 6.603240966796875e-06,
      "step": 6924,
      "training_step_time": 17.10257387161255
    },
    {
      "epoch": 6.604194641113281e-06,
      "model_forward_time": 5.6076014041900635,
      "step": 6925
    },
    {
      "epoch": 6.604194641113281e-06,
      "step": 6925,
      "training_step_time": 17.122126579284668
    },
    {
      "epoch": 6.605148315429688e-06,
      "model_forward_time": 5.615438461303711,
      "step": 6926
    },
    {
      "epoch": 6.605148315429688e-06,
      "step": 6926,
      "training_step_time": 17.08616542816162
    },
    {
      "epoch": 6.606101989746093e-06,
      "model_forward_time": 5.947443723678589,
      "step": 6927
    },
    {
      "epoch": 6.606101989746093e-06,
      "step": 6927,
      "training_step_time": 17.362359762191772
    },
    {
      "epoch": 6.6070556640625e-06,
      "model_forward_time": 5.566635847091675,
      "step": 6928
    },
    {
      "epoch": 6.6070556640625e-06,
      "step": 6928,
      "training_step_time": 16.97468852996826
    },
    {
      "epoch": 6.6080093383789065e-06,
      "model_forward_time": 5.57010555267334,
      "step": 6929
    },
    {
      "epoch": 6.6080093383789065e-06,
      "step": 6929,
      "training_step_time": 16.989943742752075
    },
    {
      "epoch": 6.608963012695312e-06,
      "grad_norm": 0.6667521595954895,
      "learning_rate": 1e-05,
      "loss": 0.0809,
      "step": 6930
    },
    {
      "epoch": 6.608963012695312e-06,
      "model_forward_time": 5.567023515701294,
      "step": 6930
    },
    {
      "dynamics_loss_avg": 0.06486605890095234,
      "epoch": 6.608963012695312e-06,
      "step": 6930
    },
    {
      "action_loss_avg": 0.012669588602147997,
      "epoch": 6.608963012695312e-06,
      "step": 6930
    },
    {
      "epoch": 6.608963012695312e-06,
      "step": 6930,
      "training_step_time": 17.00380253791809
    },
    {
      "epoch": 6.609916687011719e-06,
      "model_forward_time": 5.58130955696106,
      "step": 6931
    },
    {
      "epoch": 6.609916687011719e-06,
      "step": 6931,
      "training_step_time": 17.000516891479492
    },
    {
      "epoch": 6.610870361328125e-06,
      "model_forward_time": 5.586129426956177,
      "step": 6932
    },
    {
      "epoch": 6.610870361328125e-06,
      "step": 6932,
      "training_step_time": 17.00982689857483
    },
    {
      "epoch": 6.611824035644531e-06,
      "model_forward_time": 5.947682619094849,
      "step": 6933
    },
    {
      "epoch": 6.611824035644531e-06,
      "step": 6933,
      "training_step_time": 17.38667058944702
    },
    {
      "epoch": 6.612777709960938e-06,
      "model_forward_time": 5.595254421234131,
      "step": 6934
    },
    {
      "epoch": 6.612777709960938e-06,
      "step": 6934,
      "training_step_time": 17.04887294769287
    },
    {
      "epoch": 6.613731384277343e-06,
      "model_forward_time": 5.607158422470093,
      "step": 6935
    },
    {
      "epoch": 6.613731384277343e-06,
      "step": 6935,
      "training_step_time": 17.111265182495117
    },
    {
      "epoch": 6.61468505859375e-06,
      "model_forward_time": 5.615716218948364,
      "step": 6936
    },
    {
      "epoch": 6.61468505859375e-06,
      "step": 6936,
      "training_step_time": 17.1347553730011
    },
    {
      "epoch": 6.6156387329101565e-06,
      "model_forward_time": 5.620245933532715,
      "step": 6937
    },
    {
      "epoch": 6.6156387329101565e-06,
      "step": 6937,
      "training_step_time": 17.126073837280273
    },
    {
      "epoch": 6.616592407226562e-06,
      "model_forward_time": 5.6140477657318115,
      "step": 6938
    },
    {
      "epoch": 6.616592407226562e-06,
      "step": 6938,
      "training_step_time": 17.480003833770752
    },
    {
      "epoch": 6.617546081542969e-06,
      "model_forward_time": 5.620484828948975,
      "step": 6939
    },
    {
      "epoch": 6.617546081542969e-06,
      "step": 6939,
      "training_step_time": 17.09431552886963
    },
    {
      "epoch": 6.618499755859375e-06,
      "grad_norm": 0.9319673180580139,
      "learning_rate": 1e-05,
      "loss": 0.0711,
      "step": 6940
    },
    {
      "epoch": 6.618499755859375e-06,
      "model_forward_time": 5.591240644454956,
      "step": 6940
    },
    {
      "dynamics_loss_avg": 0.05751752611249685,
      "epoch": 6.618499755859375e-06,
      "step": 6940
    },
    {
      "action_loss_avg": 0.012802145327441394,
      "epoch": 6.618499755859375e-06,
      "step": 6940
    },
    {
      "epoch": 6.618499755859375e-06,
      "step": 6940,
      "training_step_time": 17.02675223350525
    },
    {
      "epoch": 6.619453430175781e-06,
      "model_forward_time": 5.577426195144653,
      "step": 6941
    },
    {
      "epoch": 6.619453430175781e-06,
      "step": 6941,
      "training_step_time": 17.01374840736389
    },
    {
      "epoch": 6.620407104492188e-06,
      "model_forward_time": 5.576610326766968,
      "step": 6942
    },
    {
      "epoch": 6.620407104492188e-06,
      "step": 6942,
      "training_step_time": 17.012150287628174
    },
    {
      "epoch": 6.621360778808593e-06,
      "model_forward_time": 5.57924747467041,
      "step": 6943
    },
    {
      "epoch": 6.621360778808593e-06,
      "step": 6943,
      "training_step_time": 17.014817237854004
    },
    {
      "epoch": 6.622314453125e-06,
      "model_forward_time": 5.93927526473999,
      "step": 6944
    },
    {
      "epoch": 6.622314453125e-06,
      "step": 6944,
      "training_step_time": 17.342823266983032
    },
    {
      "epoch": 6.6232681274414065e-06,
      "model_forward_time": 5.584673643112183,
      "step": 6945
    },
    {
      "epoch": 6.6232681274414065e-06,
      "step": 6945,
      "training_step_time": 17.01912021636963
    },
    {
      "epoch": 6.624221801757812e-06,
      "model_forward_time": 5.581787586212158,
      "step": 6946
    },
    {
      "epoch": 6.624221801757812e-06,
      "step": 6946,
      "training_step_time": 17.031768321990967
    },
    {
      "epoch": 6.625175476074219e-06,
      "model_forward_time": 5.604092359542847,
      "step": 6947
    },
    {
      "epoch": 6.625175476074219e-06,
      "step": 6947,
      "training_step_time": 17.09509515762329
    },
    {
      "epoch": 6.626129150390625e-06,
      "model_forward_time": 5.621411561965942,
      "step": 6948
    },
    {
      "epoch": 6.626129150390625e-06,
      "step": 6948,
      "training_step_time": 17.12211036682129
    },
    {
      "epoch": 6.627082824707031e-06,
      "model_forward_time": 5.954416751861572,
      "step": 6949
    },
    {
      "epoch": 6.627082824707031e-06,
      "step": 6949,
      "training_step_time": 17.44444465637207
    },
    {
      "epoch": 6.628036499023438e-06,
      "grad_norm": 0.8895646929740906,
      "learning_rate": 1e-05,
      "loss": 0.0735,
      "step": 6950
    },
    {
      "epoch": 6.628036499023438e-06,
      "model_forward_time": 5.606877326965332,
      "step": 6950
    },
    {
      "dynamics_loss_avg": 0.05845788940787315,
      "epoch": 6.628036499023438e-06,
      "step": 6950
    },
    {
      "action_loss_avg": 0.013198880013078451,
      "epoch": 6.628036499023438e-06,
      "step": 6950
    },
    {
      "epoch": 6.628036499023438e-06,
      "step": 6950,
      "training_step_time": 17.08130717277527
    },
    {
      "epoch": 6.628990173339843e-06,
      "model_forward_time": 5.59575629234314,
      "step": 6951
    },
    {
      "epoch": 6.628990173339843e-06,
      "step": 6951,
      "training_step_time": 17.04952120780945
    },
    {
      "epoch": 6.62994384765625e-06,
      "model_forward_time": 5.5976176261901855,
      "step": 6952
    },
    {
      "epoch": 6.62994384765625e-06,
      "step": 6952,
      "training_step_time": 17.038710594177246
    },
    {
      "epoch": 6.6308975219726565e-06,
      "model_forward_time": 5.581238746643066,
      "step": 6953
    },
    {
      "epoch": 6.6308975219726565e-06,
      "step": 6953,
      "training_step_time": 17.01728844642639
    },
    {
      "epoch": 6.631851196289062e-06,
      "model_forward_time": 5.584220886230469,
      "step": 6954
    },
    {
      "epoch": 6.631851196289062e-06,
      "step": 6954,
      "training_step_time": 17.02241587638855
    },
    {
      "epoch": 6.632804870605469e-06,
      "model_forward_time": 5.943999528884888,
      "step": 6955
    },
    {
      "epoch": 6.632804870605469e-06,
      "step": 6955,
      "training_step_time": 17.41252112388611
    },
    {
      "epoch": 6.633758544921875e-06,
      "model_forward_time": 5.608407258987427,
      "step": 6956
    },
    {
      "epoch": 6.633758544921875e-06,
      "step": 6956,
      "training_step_time": 17.084777355194092
    },
    {
      "epoch": 6.634712219238281e-06,
      "model_forward_time": 5.610514163970947,
      "step": 6957
    },
    {
      "epoch": 6.634712219238281e-06,
      "step": 6957,
      "training_step_time": 17.078777313232422
    },
    {
      "epoch": 6.635665893554688e-06,
      "model_forward_time": 5.609582901000977,
      "step": 6958
    },
    {
      "epoch": 6.635665893554688e-06,
      "step": 6958,
      "training_step_time": 17.065435886383057
    },
    {
      "epoch": 6.636619567871093e-06,
      "model_forward_time": 5.600648880004883,
      "step": 6959
    },
    {
      "epoch": 6.636619567871093e-06,
      "step": 6959,
      "training_step_time": 17.03857970237732
    },
    {
      "epoch": 6.6375732421875e-06,
      "grad_norm": 2.1279003620147705,
      "learning_rate": 1e-05,
      "loss": 0.0786,
      "step": 6960
    },
    {
      "epoch": 6.6375732421875e-06,
      "model_forward_time": 5.935231447219849,
      "step": 6960
    },
    {
      "dynamics_loss_avg": 0.06949068866670131,
      "epoch": 6.6375732421875e-06,
      "step": 6960
    },
    {
      "action_loss_avg": 0.012949383188970386,
      "epoch": 6.6375732421875e-06,
      "step": 6960
    },
    {
      "epoch": 6.6375732421875e-06,
      "step": 6960,
      "training_step_time": 17.37094211578369
    },
    {
      "epoch": 6.6385269165039065e-06,
      "model_forward_time": 5.592016696929932,
      "step": 6961
    },
    {
      "epoch": 6.6385269165039065e-06,
      "step": 6961,
      "training_step_time": 17.05286979675293
    },
    {
      "epoch": 6.639480590820312e-06,
      "model_forward_time": 5.5869386196136475,
      "step": 6962
    },
    {
      "epoch": 6.639480590820312e-06,
      "step": 6962,
      "training_step_time": 17.042256116867065
    },
    {
      "epoch": 6.640434265136719e-06,
      "model_forward_time": 5.596315622329712,
      "step": 6963
    },
    {
      "epoch": 6.640434265136719e-06,
      "step": 6963,
      "training_step_time": 17.055734395980835
    },
    {
      "epoch": 6.641387939453125e-06,
      "model_forward_time": 5.590756416320801,
      "step": 6964
    },
    {
      "epoch": 6.641387939453125e-06,
      "step": 6964,
      "training_step_time": 17.02257251739502
    },
    {
      "epoch": 6.642341613769531e-06,
      "model_forward_time": 5.589824676513672,
      "step": 6965
    },
    {
      "epoch": 6.642341613769531e-06,
      "step": 6965,
      "training_step_time": 17.420555591583252
    },
    {
      "epoch": 6.643295288085938e-06,
      "model_forward_time": 5.611532926559448,
      "step": 6966
    },
    {
      "epoch": 6.643295288085938e-06,
      "step": 6966,
      "training_step_time": 17.06094789505005
    },
    {
      "epoch": 6.644248962402343e-06,
      "model_forward_time": 5.596651792526245,
      "step": 6967
    },
    {
      "epoch": 6.644248962402343e-06,
      "step": 6967,
      "training_step_time": 17.030909776687622
    },
    {
      "epoch": 6.64520263671875e-06,
      "model_forward_time": 5.5931010246276855,
      "step": 6968
    },
    {
      "epoch": 6.64520263671875e-06,
      "step": 6968,
      "training_step_time": 17.02725625038147
    },
    {
      "epoch": 6.6461563110351565e-06,
      "model_forward_time": 5.5886805057525635,
      "step": 6969
    },
    {
      "epoch": 6.6461563110351565e-06,
      "step": 6969,
      "training_step_time": 17.037352561950684
    },
    {
      "epoch": 6.647109985351562e-06,
      "grad_norm": 0.7951217293739319,
      "learning_rate": 1e-05,
      "loss": 0.0903,
      "step": 6970
    },
    {
      "epoch": 6.647109985351562e-06,
      "model_forward_time": 5.59326434135437,
      "step": 6970
    },
    {
      "dynamics_loss_avg": 0.0729281522333622,
      "epoch": 6.647109985351562e-06,
      "step": 6970
    },
    {
      "action_loss_avg": 0.01717584690777585,
      "epoch": 6.647109985351562e-06,
      "step": 6970
    },
    {
      "epoch": 6.647109985351562e-06,
      "step": 6970,
      "training_step_time": 17.04788041114807
    },
    {
      "epoch": 6.648063659667969e-06,
      "model_forward_time": 5.958625078201294,
      "step": 6971
    },
    {
      "epoch": 6.648063659667969e-06,
      "step": 6971,
      "training_step_time": 17.3970148563385
    },
    {
      "epoch": 6.649017333984375e-06,
      "model_forward_time": 5.593642950057983,
      "step": 6972
    },
    {
      "epoch": 6.649017333984375e-06,
      "step": 6972,
      "training_step_time": 17.033897638320923
    },
    {
      "epoch": 6.649971008300781e-06,
      "model_forward_time": 5.591490268707275,
      "step": 6973
    },
    {
      "epoch": 6.649971008300781e-06,
      "step": 6973,
      "training_step_time": 17.055264472961426
    },
    {
      "epoch": 6.650924682617188e-06,
      "model_forward_time": 5.602562665939331,
      "step": 6974
    },
    {
      "epoch": 6.650924682617188e-06,
      "step": 6974,
      "training_step_time": 17.095821619033813
    },
    {
      "epoch": 6.651878356933593e-06,
      "model_forward_time": 5.618604421615601,
      "step": 6975
    },
    {
      "epoch": 6.651878356933593e-06,
      "step": 6975,
      "training_step_time": 17.1278133392334
    },
    {
      "epoch": 6.65283203125e-06,
      "model_forward_time": 5.6006293296813965,
      "step": 6976
    },
    {
      "epoch": 6.65283203125e-06,
      "step": 6976,
      "training_step_time": 17.40257716178894
    },
    {
      "epoch": 6.6537857055664065e-06,
      "model_forward_time": 5.591760158538818,
      "step": 6977
    },
    {
      "epoch": 6.6537857055664065e-06,
      "step": 6977,
      "training_step_time": 17.035587072372437
    },
    {
      "epoch": 6.654739379882812e-06,
      "model_forward_time": 5.5853822231292725,
      "step": 6978
    },
    {
      "epoch": 6.654739379882812e-06,
      "step": 6978,
      "training_step_time": 17.040149688720703
    },
    {
      "epoch": 6.655693054199219e-06,
      "model_forward_time": 5.591003894805908,
      "step": 6979
    },
    {
      "epoch": 6.655693054199219e-06,
      "step": 6979,
      "training_step_time": 17.03155493736267
    },
    {
      "epoch": 6.656646728515625e-06,
      "grad_norm": 0.9855745434761047,
      "learning_rate": 1e-05,
      "loss": 0.0738,
      "step": 6980
    },
    {
      "epoch": 6.656646728515625e-06,
      "model_forward_time": 5.592103481292725,
      "step": 6980
    },
    {
      "dynamics_loss_avg": 0.06401802729815245,
      "epoch": 6.656646728515625e-06,
      "step": 6980
    },
    {
      "action_loss_avg": 0.0094689974328503,
      "epoch": 6.656646728515625e-06,
      "step": 6980
    },
    {
      "epoch": 6.656646728515625e-06,
      "step": 6980,
      "training_step_time": 17.077078580856323
    },
    {
      "epoch": 6.657600402832031e-06,
      "model_forward_time": 5.61515212059021,
      "step": 6981
    },
    {
      "epoch": 6.657600402832031e-06,
      "step": 6981,
      "training_step_time": 17.104145765304565
    },
    {
      "epoch": 6.658554077148438e-06,
      "model_forward_time": 5.954226970672607,
      "step": 6982
    },
    {
      "epoch": 6.658554077148438e-06,
      "step": 6982,
      "training_step_time": 17.40994143486023
    },
    {
      "epoch": 6.659507751464843e-06,
      "model_forward_time": 5.595926761627197,
      "step": 6983
    },
    {
      "epoch": 6.659507751464843e-06,
      "step": 6983,
      "training_step_time": 17.05816650390625
    },
    {
      "epoch": 6.66046142578125e-06,
      "model_forward_time": 5.591646432876587,
      "step": 6984
    },
    {
      "epoch": 6.66046142578125e-06,
      "step": 6984,
      "training_step_time": 17.029261589050293
    },
    {
      "epoch": 6.6614151000976566e-06,
      "model_forward_time": 5.596944570541382,
      "step": 6985
    },
    {
      "epoch": 6.6614151000976566e-06,
      "step": 6985,
      "training_step_time": 17.059253215789795
    },
    {
      "epoch": 6.662368774414062e-06,
      "model_forward_time": 5.591904401779175,
      "step": 6986
    },
    {
      "epoch": 6.662368774414062e-06,
      "step": 6986,
      "training_step_time": 17.05040192604065
    },
    {
      "epoch": 6.663322448730469e-06,
      "model_forward_time": 5.987319469451904,
      "step": 6987
    },
    {
      "epoch": 6.663322448730469e-06,
      "step": 6987,
      "training_step_time": 17.459112644195557
    },
    {
      "epoch": 6.664276123046875e-06,
      "model_forward_time": 5.615664720535278,
      "step": 6988
    },
    {
      "epoch": 6.664276123046875e-06,
      "step": 6988,
      "training_step_time": 17.0896737575531
    },
    {
      "epoch": 6.665229797363281e-06,
      "model_forward_time": 5.606130123138428,
      "step": 6989
    },
    {
      "epoch": 6.665229797363281e-06,
      "step": 6989,
      "training_step_time": 17.09262990951538
    },
    {
      "epoch": 6.666183471679688e-06,
      "grad_norm": 1.0597503185272217,
      "learning_rate": 1e-05,
      "loss": 0.0834,
      "step": 6990
    },
    {
      "epoch": 6.666183471679688e-06,
      "model_forward_time": 5.616685390472412,
      "step": 6990
    },
    {
      "dynamics_loss_avg": 0.06785382274538279,
      "epoch": 6.666183471679688e-06,
      "step": 6990
    },
    {
      "action_loss_avg": 0.0113352638669312,
      "epoch": 6.666183471679688e-06,
      "step": 6990
    },
    {
      "epoch": 6.666183471679688e-06,
      "step": 6990,
      "training_step_time": 17.124577045440674
    },
    {
      "epoch": 6.6671371459960934e-06,
      "model_forward_time": 5.612615585327148,
      "step": 6991
    },
    {
      "epoch": 6.6671371459960934e-06,
      "step": 6991,
      "training_step_time": 17.082271575927734
    },
    {
      "epoch": 6.6680908203125e-06,
      "model_forward_time": 5.614835262298584,
      "step": 6992
    },
    {
      "epoch": 6.6680908203125e-06,
      "step": 6992,
      "training_step_time": 17.46993660926819
    },
    {
      "epoch": 6.6690444946289066e-06,
      "model_forward_time": 5.59795069694519,
      "step": 6993
    },
    {
      "epoch": 6.6690444946289066e-06,
      "step": 6993,
      "training_step_time": 17.060485363006592
    },
    {
      "epoch": 6.669998168945312e-06,
      "model_forward_time": 5.594611406326294,
      "step": 6994
    },
    {
      "epoch": 6.669998168945312e-06,
      "step": 6994,
      "training_step_time": 17.019245624542236
    },
    {
      "epoch": 6.670951843261719e-06,
      "model_forward_time": 5.585568428039551,
      "step": 6995
    },
    {
      "epoch": 6.670951843261719e-06,
      "step": 6995,
      "training_step_time": 17.028045654296875
    },
    {
      "epoch": 6.671905517578125e-06,
      "model_forward_time": 5.601184844970703,
      "step": 6996
    },
    {
      "epoch": 6.671905517578125e-06,
      "step": 6996,
      "training_step_time": 17.06826949119568
    },
    {
      "epoch": 6.672859191894531e-06,
      "model_forward_time": 5.611783742904663,
      "step": 6997
    },
    {
      "epoch": 6.672859191894531e-06,
      "step": 6997,
      "training_step_time": 17.087711572647095
    },
    {
      "epoch": 6.673812866210938e-06,
      "model_forward_time": 5.980301856994629,
      "step": 6998
    },
    {
      "epoch": 6.673812866210938e-06,
      "step": 6998,
      "training_step_time": 17.458195447921753
    },
    {
      "epoch": 6.6747665405273434e-06,
      "model_forward_time": 5.6061110496521,
      "step": 6999
    },
    {
      "epoch": 6.6747665405273434e-06,
      "step": 6999,
      "training_step_time": 17.08259606361389
    },
    {
      "epoch": 6.67572021484375e-06,
      "grad_norm": 0.7085704803466797,
      "learning_rate": 1e-05,
      "loss": 0.074,
      "step": 7000
    },
    {
      "epoch": 6.67572021484375e-06,
      "model_forward_time": 5.676421403884888,
      "step": 7000
    },
    {
      "dynamics_loss_avg": 0.06976518072187901,
      "epoch": 6.67572021484375e-06,
      "step": 7000
    },
    {
      "action_loss_avg": 0.009688529232516885,
      "epoch": 6.67572021484375e-06,
      "step": 7000
    },
    {
      "epoch": 6.67572021484375e-06,
      "step": 7000,
      "training_step_time": 17.191051244735718
    },
    {
      "epoch": 6.6766738891601566e-06,
      "model_forward_time": 5.668732404708862,
      "step": 7001
    },
    {
      "epoch": 6.6766738891601566e-06,
      "step": 7001,
      "training_step_time": 17.177278995513916
    },
    {
      "epoch": 6.677627563476562e-06,
      "model_forward_time": 5.692777156829834,
      "step": 7002
    },
    {
      "epoch": 6.677627563476562e-06,
      "step": 7002,
      "training_step_time": 17.23520517349243
    },
    {
      "epoch": 6.678581237792969e-06,
      "model_forward_time": 6.07648491859436,
      "step": 7003
    },
    {
      "epoch": 6.678581237792969e-06,
      "step": 7003,
      "training_step_time": 17.61779499053955
    },
    {
      "epoch": 6.679534912109375e-06,
      "model_forward_time": 5.631959676742554,
      "step": 7004
    },
    {
      "epoch": 6.679534912109375e-06,
      "step": 7004,
      "training_step_time": 17.145719528198242
    },
    {
      "epoch": 6.680488586425781e-06,
      "model_forward_time": 5.616446495056152,
      "step": 7005
    },
    {
      "epoch": 6.680488586425781e-06,
      "step": 7005,
      "training_step_time": 17.12234401702881
    },
    {
      "epoch": 6.681442260742188e-06,
      "model_forward_time": 5.606817245483398,
      "step": 7006
    },
    {
      "epoch": 6.681442260742188e-06,
      "step": 7006,
      "training_step_time": 17.09851360321045
    },
    {
      "epoch": 6.6823959350585934e-06,
      "model_forward_time": 5.610984802246094,
      "step": 7007
    },
    {
      "epoch": 6.6823959350585934e-06,
      "step": 7007,
      "training_step_time": 17.09185004234314
    },
    {
      "epoch": 6.683349609375e-06,
      "model_forward_time": 5.6096179485321045,
      "step": 7008
    },
    {
      "epoch": 6.683349609375e-06,
      "step": 7008,
      "training_step_time": 17.09780502319336
    },
    {
      "epoch": 6.6843032836914066e-06,
      "model_forward_time": 5.9449498653411865,
      "step": 7009
    },
    {
      "epoch": 6.6843032836914066e-06,
      "step": 7009,
      "training_step_time": 17.38847517967224
    },
    {
      "epoch": 6.685256958007812e-06,
      "grad_norm": 0.9639753699302673,
      "learning_rate": 1e-05,
      "loss": 0.0743,
      "step": 7010
    },
    {
      "epoch": 6.685256958007812e-06,
      "model_forward_time": 5.596762418746948,
      "step": 7010
    },
    {
      "dynamics_loss_avg": 0.05953391268849373,
      "epoch": 6.685256958007812e-06,
      "step": 7010
    },
    {
      "action_loss_avg": 0.009765786794014275,
      "epoch": 6.685256958007812e-06,
      "step": 7010
    },
    {
      "epoch": 6.685256958007812e-06,
      "step": 7010,
      "training_step_time": 17.055691480636597
    },
    {
      "epoch": 6.686210632324219e-06,
      "model_forward_time": 5.59307861328125,
      "step": 7011
    },
    {
      "epoch": 6.686210632324219e-06,
      "step": 7011,
      "training_step_time": 17.03712511062622
    },
    {
      "epoch": 6.687164306640625e-06,
      "model_forward_time": 5.586118936538696,
      "step": 7012
    },
    {
      "epoch": 6.687164306640625e-06,
      "step": 7012,
      "training_step_time": 17.027827501296997
    },
    {
      "epoch": 6.688117980957031e-06,
      "model_forward_time": 5.5867273807525635,
      "step": 7013
    },
    {
      "epoch": 6.688117980957031e-06,
      "step": 7013,
      "training_step_time": 17.048540830612183
    },
    {
      "epoch": 6.689071655273438e-06,
      "model_forward_time": 5.980969667434692,
      "step": 7014
    },
    {
      "epoch": 6.689071655273438e-06,
      "step": 7014,
      "training_step_time": 17.48122525215149
    },
    {
      "epoch": 6.6900253295898434e-06,
      "model_forward_time": 5.626925230026245,
      "step": 7015
    },
    {
      "epoch": 6.6900253295898434e-06,
      "step": 7015,
      "training_step_time": 17.139409065246582
    },
    {
      "epoch": 6.69097900390625e-06,
      "model_forward_time": 5.6102454662323,
      "step": 7016
    },
    {
      "epoch": 6.69097900390625e-06,
      "step": 7016,
      "training_step_time": 17.113140106201172
    },
    {
      "epoch": 6.691932678222657e-06,
      "model_forward_time": 5.606743574142456,
      "step": 7017
    },
    {
      "epoch": 6.691932678222657e-06,
      "step": 7017,
      "training_step_time": 17.096285343170166
    },
    {
      "epoch": 6.692886352539062e-06,
      "model_forward_time": 5.6090781688690186,
      "step": 7018
    },
    {
      "epoch": 6.692886352539062e-06,
      "step": 7018,
      "training_step_time": 17.10503888130188
    },
    {
      "epoch": 6.693840026855469e-06,
      "model_forward_time": 5.618354320526123,
      "step": 7019
    },
    {
      "epoch": 6.693840026855469e-06,
      "step": 7019,
      "training_step_time": 17.483662366867065
    },
    {
      "epoch": 6.694793701171875e-06,
      "grad_norm": 0.39137619733810425,
      "learning_rate": 1e-05,
      "loss": 0.0765,
      "step": 7020
    },
    {
      "epoch": 6.694793701171875e-06,
      "model_forward_time": 5.6150290966033936,
      "step": 7020
    },
    {
      "dynamics_loss_avg": 0.06589713469147682,
      "epoch": 6.694793701171875e-06,
      "step": 7020
    },
    {
      "action_loss_avg": 0.012807236332446337,
      "epoch": 6.694793701171875e-06,
      "step": 7020
    },
    {
      "epoch": 6.694793701171875e-06,
      "step": 7020,
      "training_step_time": 17.09980082511902
    },
    {
      "epoch": 6.695747375488281e-06,
      "model_forward_time": 5.605965614318848,
      "step": 7021
    },
    {
      "epoch": 6.695747375488281e-06,
      "step": 7021,
      "training_step_time": 17.042420148849487
    },
    {
      "epoch": 6.696701049804688e-06,
      "model_forward_time": 5.601202964782715,
      "step": 7022
    },
    {
      "epoch": 6.696701049804688e-06,
      "step": 7022,
      "training_step_time": 17.066613912582397
    },
    {
      "epoch": 6.6976547241210935e-06,
      "model_forward_time": 5.592670917510986,
      "step": 7023
    },
    {
      "epoch": 6.6976547241210935e-06,
      "step": 7023,
      "training_step_time": 17.011903285980225
    },
    {
      "epoch": 6.6986083984375e-06,
      "model_forward_time": 5.590505599975586,
      "step": 7024
    },
    {
      "epoch": 6.6986083984375e-06,
      "step": 7024,
      "training_step_time": 17.027946710586548
    },
    {
      "epoch": 6.699562072753907e-06,
      "model_forward_time": 5.965728044509888,
      "step": 7025
    },
    {
      "epoch": 6.699562072753907e-06,
      "step": 7025,
      "training_step_time": 17.442108631134033
    },
    {
      "epoch": 6.700515747070312e-06,
      "model_forward_time": 5.601213216781616,
      "step": 7026
    },
    {
      "epoch": 6.700515747070312e-06,
      "step": 7026,
      "training_step_time": 17.087724924087524
    },
    {
      "epoch": 6.701469421386719e-06,
      "model_forward_time": 5.599350214004517,
      "step": 7027
    },
    {
      "epoch": 6.701469421386719e-06,
      "step": 7027,
      "training_step_time": 17.066596269607544
    },
    {
      "epoch": 6.702423095703125e-06,
      "model_forward_time": 5.5972113609313965,
      "step": 7028
    },
    {
      "epoch": 6.702423095703125e-06,
      "step": 7028,
      "training_step_time": 17.038815021514893
    },
    {
      "epoch": 6.703376770019531e-06,
      "model_forward_time": 5.599719524383545,
      "step": 7029
    },
    {
      "epoch": 6.703376770019531e-06,
      "step": 7029,
      "training_step_time": 17.04629135131836
    },
    {
      "epoch": 6.704330444335938e-06,
      "grad_norm": 0.9653849005699158,
      "learning_rate": 1e-05,
      "loss": 0.0682,
      "step": 7030
    },
    {
      "epoch": 6.704330444335938e-06,
      "model_forward_time": 5.596468210220337,
      "step": 7030
    },
    {
      "dynamics_loss_avg": 0.05675793699920177,
      "epoch": 6.704330444335938e-06,
      "step": 7030
    },
    {
      "action_loss_avg": 0.012909578217659146,
      "epoch": 6.704330444335938e-06,
      "step": 7030
    },
    {
      "epoch": 6.704330444335938e-06,
      "step": 7030,
      "training_step_time": 17.40203356742859
    },
    {
      "epoch": 6.7052841186523435e-06,
      "model_forward_time": 5.593846082687378,
      "step": 7031
    },
    {
      "epoch": 6.7052841186523435e-06,
      "step": 7031,
      "training_step_time": 17.039772748947144
    },
    {
      "epoch": 6.70623779296875e-06,
      "model_forward_time": 5.593783378601074,
      "step": 7032
    },
    {
      "epoch": 6.70623779296875e-06,
      "step": 7032,
      "training_step_time": 17.01858901977539
    },
    {
      "epoch": 6.707191467285157e-06,
      "model_forward_time": 5.592406988143921,
      "step": 7033
    },
    {
      "epoch": 6.707191467285157e-06,
      "step": 7033,
      "training_step_time": 17.068124055862427
    },
    {
      "epoch": 6.708145141601562e-06,
      "model_forward_time": 5.618609189987183,
      "step": 7034
    },
    {
      "epoch": 6.708145141601562e-06,
      "step": 7034,
      "training_step_time": 17.11924719810486
    },
    {
      "epoch": 6.709098815917969e-06,
      "model_forward_time": 5.613705396652222,
      "step": 7035
    },
    {
      "epoch": 6.709098815917969e-06,
      "step": 7035,
      "training_step_time": 17.108834505081177
    },
    {
      "epoch": 6.710052490234375e-06,
      "model_forward_time": 5.970555067062378,
      "step": 7036
    },
    {
      "epoch": 6.710052490234375e-06,
      "step": 7036,
      "training_step_time": 17.431560277938843
    },
    {
      "epoch": 6.711006164550781e-06,
      "model_forward_time": 5.606826305389404,
      "step": 7037
    },
    {
      "epoch": 6.711006164550781e-06,
      "step": 7037,
      "training_step_time": 17.06825017929077
    },
    {
      "epoch": 6.711959838867188e-06,
      "model_forward_time": 5.5955963134765625,
      "step": 7038
    },
    {
      "epoch": 6.711959838867188e-06,
      "step": 7038,
      "training_step_time": 17.045233488082886
    },
    {
      "epoch": 6.7129135131835935e-06,
      "model_forward_time": 5.596882343292236,
      "step": 7039
    },
    {
      "epoch": 6.7129135131835935e-06,
      "step": 7039,
      "training_step_time": 17.045990228652954
    },
    {
      "epoch": 6.7138671875e-06,
      "grad_norm": 1.0853416919708252,
      "learning_rate": 1e-05,
      "loss": 0.0711,
      "step": 7040
    },
    {
      "epoch": 6.7138671875e-06,
      "model_forward_time": 5.602084159851074,
      "step": 7040
    },
    {
      "dynamics_loss_avg": 0.061588986590504644,
      "epoch": 6.7138671875e-06,
      "step": 7040
    },
    {
      "action_loss_avg": 0.008998673828318714,
      "epoch": 6.7138671875e-06,
      "step": 7040
    },
    {
      "epoch": 6.7138671875e-06,
      "step": 7040,
      "training_step_time": 17.050272703170776
    },
    {
      "epoch": 6.714820861816407e-06,
      "model_forward_time": 5.951401948928833,
      "step": 7041
    },
    {
      "epoch": 6.714820861816407e-06,
      "step": 7041,
      "training_step_time": 17.396693229675293
    },
    {
      "epoch": 6.715774536132812e-06,
      "model_forward_time": 5.59637188911438,
      "step": 7042
    },
    {
      "epoch": 6.715774536132812e-06,
      "step": 7042,
      "training_step_time": 17.100196838378906
    },
    {
      "epoch": 6.716728210449219e-06,
      "model_forward_time": 5.623660564422607,
      "step": 7043
    },
    {
      "epoch": 6.716728210449219e-06,
      "step": 7043,
      "training_step_time": 17.136451721191406
    },
    {
      "epoch": 6.717681884765625e-06,
      "model_forward_time": 5.613268136978149,
      "step": 7044
    },
    {
      "epoch": 6.717681884765625e-06,
      "step": 7044,
      "training_step_time": 17.08441472053528
    },
    {
      "epoch": 6.718635559082031e-06,
      "model_forward_time": 5.597919464111328,
      "step": 7045
    },
    {
      "epoch": 6.718635559082031e-06,
      "step": 7045,
      "training_step_time": 17.057909965515137
    },
    {
      "epoch": 6.719589233398438e-06,
      "model_forward_time": 5.593520402908325,
      "step": 7046
    },
    {
      "epoch": 6.719589233398438e-06,
      "step": 7046,
      "training_step_time": 17.037900924682617
    },
    {
      "epoch": 6.7205429077148435e-06,
      "model_forward_time": 5.979432821273804,
      "step": 7047
    },
    {
      "epoch": 6.7205429077148435e-06,
      "step": 7047,
      "training_step_time": 17.404985189437866
    },
    {
      "epoch": 6.72149658203125e-06,
      "model_forward_time": 5.5996174812316895,
      "step": 7048
    },
    {
      "epoch": 6.72149658203125e-06,
      "step": 7048,
      "training_step_time": 17.045533895492554
    },
    {
      "epoch": 6.722450256347657e-06,
      "model_forward_time": 5.6046130657196045,
      "step": 7049
    },
    {
      "epoch": 6.722450256347657e-06,
      "step": 7049,
      "training_step_time": 17.069602489471436
    },
    {
      "epoch": 6.723403930664062e-06,
      "grad_norm": 1.623836874961853,
      "learning_rate": 1e-05,
      "loss": 0.0872,
      "step": 7050
    },
    {
      "epoch": 6.723403930664062e-06,
      "model_forward_time": 5.610652685165405,
      "step": 7050
    },
    {
      "dynamics_loss_avg": 0.07631092146039009,
      "epoch": 6.723403930664062e-06,
      "step": 7050
    },
    {
      "action_loss_avg": 0.012435972178354859,
      "epoch": 6.723403930664062e-06,
      "step": 7050
    },
    {
      "epoch": 6.723403930664062e-06,
      "step": 7050,
      "training_step_time": 17.074273109436035
    },
    {
      "epoch": 6.724357604980469e-06,
      "model_forward_time": 5.597001314163208,
      "step": 7051
    },
    {
      "epoch": 6.724357604980469e-06,
      "step": 7051,
      "training_step_time": 17.01271891593933
    },
    {
      "epoch": 6.725311279296875e-06,
      "model_forward_time": 5.591701030731201,
      "step": 7052
    },
    {
      "epoch": 6.725311279296875e-06,
      "step": 7052,
      "training_step_time": 17.42308020591736
    },
    {
      "epoch": 6.726264953613281e-06,
      "model_forward_time": 5.607812881469727,
      "step": 7053
    },
    {
      "epoch": 6.726264953613281e-06,
      "step": 7053,
      "training_step_time": 17.072645664215088
    },
    {
      "epoch": 6.727218627929688e-06,
      "model_forward_time": 5.602891445159912,
      "step": 7054
    },
    {
      "epoch": 6.727218627929688e-06,
      "step": 7054,
      "training_step_time": 17.065419912338257
    },
    {
      "epoch": 6.7281723022460935e-06,
      "model_forward_time": 5.587157726287842,
      "step": 7055
    },
    {
      "epoch": 6.7281723022460935e-06,
      "step": 7055,
      "training_step_time": 17.02307629585266
    },
    {
      "epoch": 6.7291259765625e-06,
      "model_forward_time": 5.574628591537476,
      "step": 7056
    },
    {
      "epoch": 6.7291259765625e-06,
      "step": 7056,
      "training_step_time": 17.01212978363037
    },
    {
      "epoch": 6.730079650878907e-06,
      "model_forward_time": 5.586954593658447,
      "step": 7057
    },
    {
      "epoch": 6.730079650878907e-06,
      "step": 7057,
      "training_step_time": 17.005417108535767
    },
    {
      "epoch": 6.731033325195312e-06,
      "model_forward_time": 5.939818859100342,
      "step": 7058
    },
    {
      "epoch": 6.731033325195312e-06,
      "step": 7058,
      "training_step_time": 17.382121324539185
    },
    {
      "epoch": 6.731986999511719e-06,
      "model_forward_time": 5.5922887325286865,
      "step": 7059
    },
    {
      "epoch": 6.731986999511719e-06,
      "step": 7059,
      "training_step_time": 17.051779747009277
    },
    {
      "epoch": 6.732940673828125e-06,
      "grad_norm": 1.2676939964294434,
      "learning_rate": 1e-05,
      "loss": 0.0754,
      "step": 7060
    },
    {
      "epoch": 6.732940673828125e-06,
      "model_forward_time": 5.604637861251831,
      "step": 7060
    },
    {
      "dynamics_loss_avg": 0.06241516135632992,
      "epoch": 6.732940673828125e-06,
      "step": 7060
    },
    {
      "action_loss_avg": 0.012341702356934547,
      "epoch": 6.732940673828125e-06,
      "step": 7060
    },
    {
      "epoch": 6.732940673828125e-06,
      "step": 7060,
      "training_step_time": 17.098426580429077
    },
    {
      "epoch": 6.733894348144531e-06,
      "model_forward_time": 5.615463733673096,
      "step": 7061
    },
    {
      "epoch": 6.733894348144531e-06,
      "step": 7061,
      "training_step_time": 17.09757351875305
    },
    {
      "epoch": 6.734848022460938e-06,
      "model_forward_time": 5.611105442047119,
      "step": 7062
    },
    {
      "epoch": 6.734848022460938e-06,
      "step": 7062,
      "training_step_time": 17.100455284118652
    },
    {
      "epoch": 6.7358016967773435e-06,
      "model_forward_time": 5.617013931274414,
      "step": 7063
    },
    {
      "epoch": 6.7358016967773435e-06,
      "step": 7063,
      "training_step_time": 17.48863935470581
    },
    {
      "epoch": 6.73675537109375e-06,
      "model_forward_time": 5.614403486251831,
      "step": 7064
    },
    {
      "epoch": 6.73675537109375e-06,
      "step": 7064,
      "training_step_time": 17.09969663619995
    },
    {
      "epoch": 6.737709045410157e-06,
      "model_forward_time": 5.586745500564575,
      "step": 7065
    },
    {
      "epoch": 6.737709045410157e-06,
      "step": 7065,
      "training_step_time": 16.99388599395752
    },
    {
      "epoch": 6.738662719726562e-06,
      "model_forward_time": 5.571854114532471,
      "step": 7066
    },
    {
      "epoch": 6.738662719726562e-06,
      "step": 7066,
      "training_step_time": 17.006120204925537
    },
    {
      "epoch": 6.739616394042969e-06,
      "model_forward_time": 5.566522598266602,
      "step": 7067
    },
    {
      "epoch": 6.739616394042969e-06,
      "step": 7067,
      "training_step_time": 16.99763774871826
    },
    {
      "epoch": 6.740570068359375e-06,
      "model_forward_time": 5.5737669467926025,
      "step": 7068
    },
    {
      "epoch": 6.740570068359375e-06,
      "step": 7068,
      "training_step_time": 16.988348484039307
    },
    {
      "epoch": 6.741523742675781e-06,
      "model_forward_time": 5.936058521270752,
      "step": 7069
    },
    {
      "epoch": 6.741523742675781e-06,
      "step": 7069,
      "training_step_time": 17.341410398483276
    },
    {
      "epoch": 6.742477416992188e-06,
      "grad_norm": 0.7335404753684998,
      "learning_rate": 1e-05,
      "loss": 0.0713,
      "step": 7070
    },
    {
      "epoch": 6.742477416992188e-06,
      "model_forward_time": 5.574612379074097,
      "step": 7070
    },
    {
      "dynamics_loss_avg": 0.06170596405863762,
      "epoch": 6.742477416992188e-06,
      "step": 7070
    },
    {
      "action_loss_avg": 0.01192312261555344,
      "epoch": 6.742477416992188e-06,
      "step": 7070
    },
    {
      "epoch": 6.742477416992188e-06,
      "step": 7070,
      "training_step_time": 17.01921820640564
    },
    {
      "epoch": 6.7434310913085935e-06,
      "model_forward_time": 5.58208966255188,
      "step": 7071
    },
    {
      "epoch": 6.7434310913085935e-06,
      "step": 7071,
      "training_step_time": 17.013617992401123
    },
    {
      "epoch": 6.744384765625e-06,
      "model_forward_time": 5.58429741859436,
      "step": 7072
    },
    {
      "epoch": 6.744384765625e-06,
      "step": 7072,
      "training_step_time": 17.023785829544067
    },
    {
      "epoch": 6.745338439941407e-06,
      "model_forward_time": 5.6002116203308105,
      "step": 7073
    },
    {
      "epoch": 6.745338439941407e-06,
      "step": 7073,
      "training_step_time": 17.057331562042236
    },
    {
      "epoch": 6.746292114257812e-06,
      "model_forward_time": 5.970980167388916,
      "step": 7074
    },
    {
      "epoch": 6.746292114257812e-06,
      "step": 7074,
      "training_step_time": 17.476978063583374
    },
    {
      "epoch": 6.747245788574219e-06,
      "model_forward_time": 5.610655069351196,
      "step": 7075
    },
    {
      "epoch": 6.747245788574219e-06,
      "step": 7075,
      "training_step_time": 17.06758141517639
    },
    {
      "epoch": 6.748199462890625e-06,
      "model_forward_time": 5.595267057418823,
      "step": 7076
    },
    {
      "epoch": 6.748199462890625e-06,
      "step": 7076,
      "training_step_time": 17.023308515548706
    },
    {
      "epoch": 6.749153137207031e-06,
      "model_forward_time": 5.589093446731567,
      "step": 7077
    },
    {
      "epoch": 6.749153137207031e-06,
      "step": 7077,
      "training_step_time": 17.001915216445923
    },
    {
      "epoch": 6.750106811523438e-06,
      "model_forward_time": 5.59011435508728,
      "step": 7078
    },
    {
      "epoch": 6.750106811523438e-06,
      "step": 7078,
      "training_step_time": 17.037973165512085
    },
    {
      "epoch": 6.7510604858398435e-06,
      "model_forward_time": 5.575797080993652,
      "step": 7079
    },
    {
      "epoch": 6.7510604858398435e-06,
      "step": 7079,
      "training_step_time": 17.00011682510376
    },
    {
      "epoch": 6.75201416015625e-06,
      "grad_norm": 0.6012057662010193,
      "learning_rate": 1e-05,
      "loss": 0.0713,
      "step": 7080
    },
    {
      "epoch": 6.75201416015625e-06,
      "model_forward_time": 5.95203161239624,
      "step": 7080
    },
    {
      "dynamics_loss_avg": 0.05747089348733425,
      "epoch": 6.75201416015625e-06,
      "step": 7080
    },
    {
      "action_loss_avg": 0.009045103588141501,
      "epoch": 6.75201416015625e-06,
      "step": 7080
    },
    {
      "epoch": 6.75201416015625e-06,
      "step": 7080,
      "training_step_time": 17.411487340927124
    },
    {
      "epoch": 6.752967834472657e-06,
      "model_forward_time": 5.605044841766357,
      "step": 7081
    },
    {
      "epoch": 6.752967834472657e-06,
      "step": 7081,
      "training_step_time": 17.096388816833496
    },
    {
      "epoch": 6.753921508789062e-06,
      "model_forward_time": 5.6084184646606445,
      "step": 7082
    },
    {
      "epoch": 6.753921508789062e-06,
      "step": 7082,
      "training_step_time": 17.098669052124023
    },
    {
      "epoch": 6.754875183105469e-06,
      "model_forward_time": 5.618242979049683,
      "step": 7083
    },
    {
      "epoch": 6.754875183105469e-06,
      "step": 7083,
      "training_step_time": 17.106950521469116
    },
    {
      "epoch": 6.755828857421875e-06,
      "model_forward_time": 5.619885206222534,
      "step": 7084
    },
    {
      "epoch": 6.755828857421875e-06,
      "step": 7084,
      "training_step_time": 17.10139775276184
    },
    {
      "epoch": 6.756782531738281e-06,
      "model_forward_time": 5.5977582931518555,
      "step": 7085
    },
    {
      "epoch": 6.756782531738281e-06,
      "step": 7085,
      "training_step_time": 17.411009550094604
    },
    {
      "epoch": 6.757736206054688e-06,
      "model_forward_time": 5.591676712036133,
      "step": 7086
    },
    {
      "epoch": 6.757736206054688e-06,
      "step": 7086,
      "training_step_time": 17.046295166015625
    },
    {
      "epoch": 6.7586898803710935e-06,
      "model_forward_time": 5.590900182723999,
      "step": 7087
    },
    {
      "epoch": 6.7586898803710935e-06,
      "step": 7087,
      "training_step_time": 17.06791639328003
    },
    {
      "epoch": 6.7596435546875e-06,
      "model_forward_time": 5.603707551956177,
      "step": 7088
    },
    {
      "epoch": 6.7596435546875e-06,
      "step": 7088,
      "training_step_time": 17.074859380722046
    },
    {
      "epoch": 6.760597229003907e-06,
      "model_forward_time": 5.604153394699097,
      "step": 7089
    },
    {
      "epoch": 6.760597229003907e-06,
      "step": 7089,
      "training_step_time": 17.093544244766235
    },
    {
      "epoch": 6.761550903320312e-06,
      "grad_norm": 1.0494251251220703,
      "learning_rate": 1e-05,
      "loss": 0.056,
      "step": 7090
    },
    {
      "epoch": 6.761550903320312e-06,
      "model_forward_time": 5.606354475021362,
      "step": 7090
    },
    {
      "dynamics_loss_avg": 0.049201965983957055,
      "epoch": 6.761550903320312e-06,
      "step": 7090
    },
    {
      "action_loss_avg": 0.0074096735683269795,
      "epoch": 6.761550903320312e-06,
      "step": 7090
    },
    {
      "epoch": 6.761550903320312e-06,
      "step": 7090,
      "training_step_time": 17.093727588653564
    },
    {
      "epoch": 6.762504577636719e-06,
      "model_forward_time": 5.9786694049835205,
      "step": 7091
    },
    {
      "epoch": 6.762504577636719e-06,
      "step": 7091,
      "training_step_time": 17.469257593154907
    },
    {
      "epoch": 6.763458251953125e-06,
      "model_forward_time": 5.6042211055755615,
      "step": 7092
    },
    {
      "epoch": 6.763458251953125e-06,
      "step": 7092,
      "training_step_time": 17.083056688308716
    },
    {
      "epoch": 6.764411926269531e-06,
      "model_forward_time": 5.607482433319092,
      "step": 7093
    },
    {
      "epoch": 6.764411926269531e-06,
      "step": 7093,
      "training_step_time": 17.09264850616455
    },
    {
      "epoch": 6.765365600585938e-06,
      "model_forward_time": 5.61226487159729,
      "step": 7094
    },
    {
      "epoch": 6.765365600585938e-06,
      "step": 7094,
      "training_step_time": 17.0914785861969
    },
    {
      "epoch": 6.7663192749023435e-06,
      "model_forward_time": 5.595757722854614,
      "step": 7095
    },
    {
      "epoch": 6.7663192749023435e-06,
      "step": 7095,
      "training_step_time": 17.046724796295166
    },
    {
      "epoch": 6.76727294921875e-06,
      "model_forward_time": 5.59483790397644,
      "step": 7096
    },
    {
      "epoch": 6.76727294921875e-06,
      "step": 7096,
      "training_step_time": 17.385971784591675
    },
    {
      "epoch": 6.768226623535157e-06,
      "model_forward_time": 5.601162433624268,
      "step": 7097
    },
    {
      "epoch": 6.768226623535157e-06,
      "step": 7097,
      "training_step_time": 17.068354606628418
    },
    {
      "epoch": 6.769180297851562e-06,
      "model_forward_time": 5.60979700088501,
      "step": 7098
    },
    {
      "epoch": 6.769180297851562e-06,
      "step": 7098,
      "training_step_time": 17.10701060295105
    },
    {
      "epoch": 6.770133972167969e-06,
      "model_forward_time": 5.622495174407959,
      "step": 7099
    },
    {
      "epoch": 6.770133972167969e-06,
      "step": 7099,
      "training_step_time": 17.124120473861694
    },
    {
      "epoch": 6.771087646484375e-06,
      "grad_norm": 0.3526739478111267,
      "learning_rate": 1e-05,
      "loss": 0.0705,
      "step": 7100
    },
    {
      "epoch": 6.771087646484375e-06,
      "model_forward_time": 5.61029052734375,
      "step": 7100
    },
    {
      "dynamics_loss_avg": 0.06754370518028736,
      "epoch": 6.771087646484375e-06,
      "step": 7100
    },
    {
      "action_loss_avg": 0.010405594622716308,
      "epoch": 6.771087646484375e-06,
      "step": 7100
    },
    {
      "epoch": 6.771087646484375e-06,
      "step": 7100,
      "training_step_time": 17.09018564224243
    },
    {
      "epoch": 6.772041320800781e-06,
      "model_forward_time": 5.613451957702637,
      "step": 7101
    },
    {
      "epoch": 6.772041320800781e-06,
      "step": 7101,
      "training_step_time": 17.072418689727783
    },
    {
      "epoch": 6.772994995117188e-06,
      "model_forward_time": 5.970587253570557,
      "step": 7102
    },
    {
      "epoch": 6.772994995117188e-06,
      "step": 7102,
      "training_step_time": 17.458287954330444
    },
    {
      "epoch": 6.7739486694335935e-06,
      "model_forward_time": 5.611417770385742,
      "step": 7103
    },
    {
      "epoch": 6.7739486694335935e-06,
      "step": 7103,
      "training_step_time": 17.07240653038025
    },
    {
      "epoch": 6.77490234375e-06,
      "model_forward_time": 5.595510244369507,
      "step": 7104
    },
    {
      "epoch": 6.77490234375e-06,
      "step": 7104,
      "training_step_time": 17.017802238464355
    },
    {
      "epoch": 6.775856018066407e-06,
      "model_forward_time": 5.591184616088867,
      "step": 7105
    },
    {
      "epoch": 6.775856018066407e-06,
      "step": 7105,
      "training_step_time": 17.020975589752197
    },
    {
      "epoch": 6.776809692382812e-06,
      "model_forward_time": 5.585421800613403,
      "step": 7106
    },
    {
      "epoch": 6.776809692382812e-06,
      "step": 7106,
      "training_step_time": 17.021662712097168
    },
    {
      "epoch": 6.777763366699219e-06,
      "model_forward_time": 5.931800127029419,
      "step": 7107
    },
    {
      "epoch": 6.777763366699219e-06,
      "step": 7107,
      "training_step_time": 17.35923194885254
    },
    {
      "epoch": 6.778717041015625e-06,
      "model_forward_time": 5.583713054656982,
      "step": 7108
    },
    {
      "epoch": 6.778717041015625e-06,
      "step": 7108,
      "training_step_time": 17.01857829093933
    },
    {
      "epoch": 6.779670715332031e-06,
      "model_forward_time": 5.597629070281982,
      "step": 7109
    },
    {
      "epoch": 6.779670715332031e-06,
      "step": 7109,
      "training_step_time": 17.0767765045166
    },
    {
      "epoch": 6.780624389648438e-06,
      "grad_norm": 0.7312943339347839,
      "learning_rate": 1e-05,
      "loss": 0.0767,
      "step": 7110
    },
    {
      "epoch": 6.780624389648438e-06,
      "model_forward_time": 5.6052000522613525,
      "step": 7110
    },
    {
      "dynamics_loss_avg": 0.0569414297118783,
      "epoch": 6.780624389648438e-06,
      "step": 7110
    },
    {
      "action_loss_avg": 0.008245512004941702,
      "epoch": 6.780624389648438e-06,
      "step": 7110
    },
    {
      "epoch": 6.780624389648438e-06,
      "step": 7110,
      "training_step_time": 17.070402145385742
    },
    {
      "epoch": 6.7815780639648435e-06,
      "model_forward_time": 5.5986902713775635,
      "step": 7111
    },
    {
      "epoch": 6.7815780639648435e-06,
      "step": 7111,
      "training_step_time": 17.0414822101593
    },
    {
      "epoch": 6.78253173828125e-06,
      "model_forward_time": 5.593137264251709,
      "step": 7112
    },
    {
      "epoch": 6.78253173828125e-06,
      "step": 7112,
      "training_step_time": 17.030396223068237
    },
    {
      "epoch": 6.783485412597657e-06,
      "model_forward_time": 5.950592994689941,
      "step": 7113
    },
    {
      "epoch": 6.783485412597657e-06,
      "step": 7113,
      "training_step_time": 17.389429330825806
    },
    {
      "epoch": 6.784439086914062e-06,
      "model_forward_time": 5.588691473007202,
      "step": 7114
    },
    {
      "epoch": 6.784439086914062e-06,
      "step": 7114,
      "training_step_time": 17.038638830184937
    },
    {
      "epoch": 6.785392761230469e-06,
      "model_forward_time": 5.600369691848755,
      "step": 7115
    },
    {
      "epoch": 6.785392761230469e-06,
      "step": 7115,
      "training_step_time": 17.07672381401062
    },
    {
      "epoch": 6.786346435546875e-06,
      "model_forward_time": 5.605625629425049,
      "step": 7116
    },
    {
      "epoch": 6.786346435546875e-06,
      "step": 7116,
      "training_step_time": 17.065169095993042
    },
    {
      "epoch": 6.787300109863281e-06,
      "model_forward_time": 5.624018430709839,
      "step": 7117
    },
    {
      "epoch": 6.787300109863281e-06,
      "step": 7117,
      "training_step_time": 17.12825345993042
    },
    {
      "epoch": 6.788253784179688e-06,
      "model_forward_time": 5.609468460083008,
      "step": 7118
    },
    {
      "epoch": 6.788253784179688e-06,
      "step": 7118,
      "training_step_time": 17.465259790420532
    },
    {
      "epoch": 6.7892074584960935e-06,
      "model_forward_time": 5.61242151260376,
      "step": 7119
    },
    {
      "epoch": 6.7892074584960935e-06,
      "step": 7119,
      "training_step_time": 17.102336168289185
    },
    {
      "epoch": 6.7901611328125e-06,
      "grad_norm": 0.5723062753677368,
      "learning_rate": 1e-05,
      "loss": 0.0537,
      "step": 7120
    },
    {
      "epoch": 6.7901611328125e-06,
      "model_forward_time": 5.609363555908203,
      "step": 7120
    },
    {
      "dynamics_loss_avg": 0.0550053708255291,
      "epoch": 6.7901611328125e-06,
      "step": 7120
    },
    {
      "action_loss_avg": 0.009685507952235639,
      "epoch": 6.7901611328125e-06,
      "step": 7120
    },
    {
      "epoch": 6.7901611328125e-06,
      "step": 7120,
      "training_step_time": 17.100219249725342
    },
    {
      "epoch": 6.791114807128907e-06,
      "model_forward_time": 5.596641302108765,
      "step": 7121
    },
    {
      "epoch": 6.791114807128907e-06,
      "step": 7121,
      "training_step_time": 17.03877592086792
    },
    {
      "epoch": 6.792068481445312e-06,
      "model_forward_time": 5.596966981887817,
      "step": 7122
    },
    {
      "epoch": 6.792068481445312e-06,
      "step": 7122,
      "training_step_time": 17.02875828742981
    },
    {
      "epoch": 6.793022155761719e-06,
      "model_forward_time": 5.583439588546753,
      "step": 7123
    },
    {
      "epoch": 6.793022155761719e-06,
      "step": 7123,
      "training_step_time": 16.993720531463623
    },
    {
      "epoch": 6.793975830078125e-06,
      "model_forward_time": 5.940045356750488,
      "step": 7124
    },
    {
      "epoch": 6.793975830078125e-06,
      "step": 7124,
      "training_step_time": 17.38446354866028
    },
    {
      "epoch": 6.794929504394531e-06,
      "model_forward_time": 5.578278303146362,
      "step": 7125
    },
    {
      "epoch": 6.794929504394531e-06,
      "step": 7125,
      "training_step_time": 16.98638606071472
    },
    {
      "epoch": 6.795883178710938e-06,
      "model_forward_time": 5.574364185333252,
      "step": 7126
    },
    {
      "epoch": 6.795883178710938e-06,
      "step": 7126,
      "training_step_time": 16.98505425453186
    },
    {
      "epoch": 6.7968368530273435e-06,
      "model_forward_time": 5.589123964309692,
      "step": 7127
    },
    {
      "epoch": 6.7968368530273435e-06,
      "step": 7127,
      "training_step_time": 17.027520895004272
    },
    {
      "epoch": 6.79779052734375e-06,
      "model_forward_time": 5.592012166976929,
      "step": 7128
    },
    {
      "epoch": 6.79779052734375e-06,
      "step": 7128,
      "training_step_time": 17.055110454559326
    },
    {
      "epoch": 6.798744201660156e-06,
      "model_forward_time": 5.598315238952637,
      "step": 7129
    },
    {
      "epoch": 6.798744201660156e-06,
      "step": 7129,
      "training_step_time": 17.40533947944641
    },
    {
      "epoch": 6.799697875976562e-06,
      "grad_norm": 1.2852394580841064,
      "learning_rate": 1e-05,
      "loss": 0.0834,
      "step": 7130
    },
    {
      "epoch": 6.799697875976562e-06,
      "model_forward_time": 5.6001245975494385,
      "step": 7130
    },
    {
      "dynamics_loss_avg": 0.07396294642239809,
      "epoch": 6.799697875976562e-06,
      "step": 7130
    },
    {
      "action_loss_avg": 0.011066403286531568,
      "epoch": 6.799697875976562e-06,
      "step": 7130
    },
    {
      "epoch": 6.799697875976562e-06,
      "step": 7130,
      "training_step_time": 17.08062982559204
    },
    {
      "epoch": 6.800651550292969e-06,
      "model_forward_time": 5.607038974761963,
      "step": 7131
    },
    {
      "epoch": 6.800651550292969e-06,
      "step": 7131,
      "training_step_time": 17.09680461883545
    },
    {
      "epoch": 6.801605224609375e-06,
      "model_forward_time": 5.610661506652832,
      "step": 7132
    },
    {
      "epoch": 6.801605224609375e-06,
      "step": 7132,
      "training_step_time": 17.11525845527649
    },
    {
      "epoch": 6.802558898925781e-06,
      "model_forward_time": 5.615907907485962,
      "step": 7133
    },
    {
      "epoch": 6.802558898925781e-06,
      "step": 7133,
      "training_step_time": 17.09695816040039
    },
    {
      "epoch": 6.803512573242188e-06,
      "model_forward_time": 5.592177629470825,
      "step": 7134
    },
    {
      "epoch": 6.803512573242188e-06,
      "step": 7134,
      "training_step_time": 17.050064086914062
    },
    {
      "epoch": 6.8044662475585935e-06,
      "model_forward_time": 5.959575176239014,
      "step": 7135
    },
    {
      "epoch": 6.8044662475585935e-06,
      "step": 7135,
      "training_step_time": 17.398916244506836
    },
    {
      "epoch": 6.805419921875e-06,
      "model_forward_time": 5.581251382827759,
      "step": 7136
    },
    {
      "epoch": 6.805419921875e-06,
      "step": 7136,
      "training_step_time": 17.027442455291748
    },
    {
      "epoch": 6.806373596191406e-06,
      "model_forward_time": 5.593700408935547,
      "step": 7137
    },
    {
      "epoch": 6.806373596191406e-06,
      "step": 7137,
      "training_step_time": 17.054968118667603
    },
    {
      "epoch": 6.807327270507812e-06,
      "model_forward_time": 5.612928628921509,
      "step": 7138
    },
    {
      "epoch": 6.807327270507812e-06,
      "step": 7138,
      "training_step_time": 17.09614086151123
    },
    {
      "epoch": 6.808280944824219e-06,
      "model_forward_time": 5.612987279891968,
      "step": 7139
    },
    {
      "epoch": 6.808280944824219e-06,
      "step": 7139,
      "training_step_time": 17.08086085319519
    },
    {
      "epoch": 6.809234619140625e-06,
      "grad_norm": 1.0018272399902344,
      "learning_rate": 1e-05,
      "loss": 0.0672,
      "step": 7140
    },
    {
      "epoch": 6.809234619140625e-06,
      "model_forward_time": 5.96477198600769,
      "step": 7140
    },
    {
      "dynamics_loss_avg": 0.05230078641325235,
      "epoch": 6.809234619140625e-06,
      "step": 7140
    },
    {
      "action_loss_avg": 0.00837355712428689,
      "epoch": 6.809234619140625e-06,
      "step": 7140
    },
    {
      "epoch": 6.809234619140625e-06,
      "step": 7140,
      "training_step_time": 17.41141939163208
    },
    {
      "epoch": 6.810188293457031e-06,
      "model_forward_time": 5.593319892883301,
      "step": 7141
    },
    {
      "epoch": 6.810188293457031e-06,
      "step": 7141,
      "training_step_time": 17.034499645233154
    },
    {
      "epoch": 6.811141967773438e-06,
      "model_forward_time": 5.595355272293091,
      "step": 7142
    },
    {
      "epoch": 6.811141967773438e-06,
      "step": 7142,
      "training_step_time": 17.041086196899414
    },
    {
      "epoch": 6.8120956420898436e-06,
      "model_forward_time": 5.593592405319214,
      "step": 7143
    },
    {
      "epoch": 6.8120956420898436e-06,
      "step": 7143,
      "training_step_time": 17.06770348548889
    },
    {
      "epoch": 6.81304931640625e-06,
      "model_forward_time": 5.610129117965698,
      "step": 7144
    },
    {
      "epoch": 6.81304931640625e-06,
      "step": 7144,
      "training_step_time": 17.069051027297974
    },
    {
      "epoch": 6.814002990722656e-06,
      "model_forward_time": 5.593043327331543,
      "step": 7145
    },
    {
      "epoch": 6.814002990722656e-06,
      "step": 7145,
      "training_step_time": 17.001542568206787
    },
    {
      "epoch": 6.814956665039062e-06,
      "model_forward_time": 5.942776441574097,
      "step": 7146
    },
    {
      "epoch": 6.814956665039062e-06,
      "step": 7146,
      "training_step_time": 17.367408990859985
    },
    {
      "epoch": 6.815910339355469e-06,
      "model_forward_time": 5.573668479919434,
      "step": 7147
    },
    {
      "epoch": 6.815910339355469e-06,
      "step": 7147,
      "training_step_time": 17.000544548034668
    },
    {
      "epoch": 6.816864013671875e-06,
      "model_forward_time": 5.578917026519775,
      "step": 7148
    },
    {
      "epoch": 6.816864013671875e-06,
      "step": 7148,
      "training_step_time": 17.000940561294556
    },
    {
      "epoch": 6.817817687988281e-06,
      "model_forward_time": 5.587622165679932,
      "step": 7149
    },
    {
      "epoch": 6.817817687988281e-06,
      "step": 7149,
      "training_step_time": 17.048696756362915
    },
    {
      "epoch": 6.818771362304688e-06,
      "grad_norm": 0.8409590721130371,
      "learning_rate": 1e-05,
      "loss": 0.0741,
      "step": 7150
    },
    {
      "epoch": 6.818771362304688e-06,
      "model_forward_time": 5.592453479766846,
      "step": 7150
    },
    {
      "dynamics_loss_avg": 0.06124101001769304,
      "epoch": 6.818771362304688e-06,
      "step": 7150
    },
    {
      "action_loss_avg": 0.010870879399590194,
      "epoch": 6.818771362304688e-06,
      "step": 7150
    },
    {
      "epoch": 6.818771362304688e-06,
      "step": 7150,
      "training_step_time": 17.0235857963562
    },
    {
      "epoch": 6.8197250366210936e-06,
      "model_forward_time": 5.605422019958496,
      "step": 7151
    },
    {
      "epoch": 6.8197250366210936e-06,
      "step": 7151,
      "training_step_time": 17.451330184936523
    },
    {
      "epoch": 6.8206787109375e-06,
      "model_forward_time": 5.597213268280029,
      "step": 7152
    },
    {
      "epoch": 6.8206787109375e-06,
      "step": 7152,
      "training_step_time": 17.027596950531006
    },
    {
      "epoch": 6.821632385253906e-06,
      "model_forward_time": 5.592922925949097,
      "step": 7153
    },
    {
      "epoch": 6.821632385253906e-06,
      "step": 7153,
      "training_step_time": 17.054784059524536
    },
    {
      "epoch": 6.822586059570312e-06,
      "model_forward_time": 5.618389844894409,
      "step": 7154
    },
    {
      "epoch": 6.822586059570312e-06,
      "step": 7154,
      "training_step_time": 17.077513694763184
    },
    {
      "epoch": 6.823539733886719e-06,
      "model_forward_time": 5.618222713470459,
      "step": 7155
    },
    {
      "epoch": 6.823539733886719e-06,
      "step": 7155,
      "training_step_time": 17.106899976730347
    },
    {
      "epoch": 6.824493408203125e-06,
      "model_forward_time": 5.6105170249938965,
      "step": 7156
    },
    {
      "epoch": 6.824493408203125e-06,
      "step": 7156,
      "training_step_time": 17.08624815940857
    },
    {
      "epoch": 6.825447082519531e-06,
      "model_forward_time": 5.991197824478149,
      "step": 7157
    },
    {
      "epoch": 6.825447082519531e-06,
      "step": 7157,
      "training_step_time": 17.472204446792603
    },
    {
      "epoch": 6.826400756835938e-06,
      "model_forward_time": 5.60187840461731,
      "step": 7158
    },
    {
      "epoch": 6.826400756835938e-06,
      "step": 7158,
      "training_step_time": 17.082147121429443
    },
    {
      "epoch": 6.8273544311523436e-06,
      "model_forward_time": 5.6178810596466064,
      "step": 7159
    },
    {
      "epoch": 6.8273544311523436e-06,
      "step": 7159,
      "training_step_time": 17.105299472808838
    },
    {
      "epoch": 6.82830810546875e-06,
      "grad_norm": 1.068466305732727,
      "learning_rate": 1e-05,
      "loss": 0.0644,
      "step": 7160
    },
    {
      "epoch": 6.82830810546875e-06,
      "model_forward_time": 5.614782810211182,
      "step": 7160
    },
    {
      "dynamics_loss_avg": 0.05562261007726192,
      "epoch": 6.82830810546875e-06,
      "step": 7160
    },
    {
      "action_loss_avg": 0.012606032541953027,
      "epoch": 6.82830810546875e-06,
      "step": 7160
    },
    {
      "epoch": 6.82830810546875e-06,
      "step": 7160,
      "training_step_time": 17.120850563049316
    },
    {
      "epoch": 6.829261779785156e-06,
      "model_forward_time": 5.612199783325195,
      "step": 7161
    },
    {
      "epoch": 6.829261779785156e-06,
      "step": 7161,
      "training_step_time": 17.12252640724182
    },
    {
      "epoch": 6.8302154541015624e-06,
      "model_forward_time": 5.618626832962036,
      "step": 7162
    },
    {
      "epoch": 6.8302154541015624e-06,
      "step": 7162,
      "training_step_time": 17.453511238098145
    },
    {
      "epoch": 6.831169128417969e-06,
      "model_forward_time": 5.593482732772827,
      "step": 7163
    },
    {
      "epoch": 6.831169128417969e-06,
      "step": 7163,
      "training_step_time": 17.026093244552612
    },
    {
      "epoch": 6.832122802734375e-06,
      "model_forward_time": 5.570154428482056,
      "step": 7164
    },
    {
      "epoch": 6.832122802734375e-06,
      "step": 7164,
      "training_step_time": 16.98227095603943
    },
    {
      "epoch": 6.833076477050781e-06,
      "model_forward_time": 5.563843011856079,
      "step": 7165
    },
    {
      "epoch": 6.833076477050781e-06,
      "step": 7165,
      "training_step_time": 16.987484216690063
    },
    {
      "epoch": 6.834030151367188e-06,
      "model_forward_time": 5.571955919265747,
      "step": 7166
    },
    {
      "epoch": 6.834030151367188e-06,
      "step": 7166,
      "training_step_time": 17.003329277038574
    },
    {
      "epoch": 6.8349838256835936e-06,
      "model_forward_time": 5.56684947013855,
      "step": 7167
    },
    {
      "epoch": 6.8349838256835936e-06,
      "step": 7167,
      "training_step_time": 16.987248420715332
    },
    {
      "epoch": 6.8359375e-06,
      "model_forward_time": 5.938877105712891,
      "step": 7168
    },
    {
      "epoch": 6.8359375e-06,
      "step": 7168,
      "training_step_time": 17.35750961303711
    },
    {
      "epoch": 6.836891174316406e-06,
      "model_forward_time": 5.5773890018463135,
      "step": 7169
    },
    {
      "epoch": 6.836891174316406e-06,
      "step": 7169,
      "training_step_time": 17.010802268981934
    },
    {
      "epoch": 6.8378448486328124e-06,
      "grad_norm": 0.9746838808059692,
      "learning_rate": 1e-05,
      "loss": 0.056,
      "step": 7170
    },
    {
      "epoch": 6.8378448486328124e-06,
      "model_forward_time": 5.584456205368042,
      "step": 7170
    },
    {
      "dynamics_loss_avg": 0.044695564173161985,
      "epoch": 6.8378448486328124e-06,
      "step": 7170
    },
    {
      "action_loss_avg": 0.007763001066632569,
      "epoch": 6.8378448486328124e-06,
      "step": 7170
    },
    {
      "epoch": 6.8378448486328124e-06,
      "step": 7170,
      "training_step_time": 17.017865896224976
    },
    {
      "epoch": 6.838798522949219e-06,
      "model_forward_time": 5.585784912109375,
      "step": 7171
    },
    {
      "epoch": 6.838798522949219e-06,
      "step": 7171,
      "training_step_time": 17.02700161933899
    },
    {
      "epoch": 6.839752197265625e-06,
      "model_forward_time": 5.598077058792114,
      "step": 7172
    },
    {
      "epoch": 6.839752197265625e-06,
      "step": 7172,
      "training_step_time": 17.082809448242188
    },
    {
      "epoch": 6.840705871582031e-06,
      "model_forward_time": 5.975910663604736,
      "step": 7173
    },
    {
      "epoch": 6.840705871582031e-06,
      "step": 7173,
      "training_step_time": 17.47786593437195
    },
    {
      "epoch": 6.841659545898438e-06,
      "model_forward_time": 5.6261372566223145,
      "step": 7174
    },
    {
      "epoch": 6.841659545898438e-06,
      "step": 7174,
      "training_step_time": 17.1507830619812
    },
    {
      "epoch": 6.842613220214844e-06,
      "model_forward_time": 5.630623817443848,
      "step": 7175
    },
    {
      "epoch": 6.842613220214844e-06,
      "step": 7175,
      "training_step_time": 17.147998571395874
    },
    {
      "epoch": 6.84356689453125e-06,
      "model_forward_time": 5.619464874267578,
      "step": 7176
    },
    {
      "epoch": 6.84356689453125e-06,
      "step": 7176,
      "training_step_time": 17.113991260528564
    },
    {
      "epoch": 6.844520568847656e-06,
      "model_forward_time": 5.608074903488159,
      "step": 7177
    },
    {
      "epoch": 6.844520568847656e-06,
      "step": 7177,
      "training_step_time": 17.05763006210327
    },
    {
      "epoch": 6.8454742431640624e-06,
      "model_forward_time": 5.600183486938477,
      "step": 7178
    },
    {
      "epoch": 6.8454742431640624e-06,
      "step": 7178,
      "training_step_time": 17.058644771575928
    },
    {
      "epoch": 6.846427917480469e-06,
      "model_forward_time": 5.963087558746338,
      "step": 7179
    },
    {
      "epoch": 6.846427917480469e-06,
      "step": 7179,
      "training_step_time": 17.391072034835815
    },
    {
      "epoch": 6.847381591796875e-06,
      "grad_norm": 0.4453452229499817,
      "learning_rate": 1e-05,
      "loss": 0.0689,
      "step": 7180
    },
    {
      "epoch": 6.847381591796875e-06,
      "model_forward_time": 5.575310230255127,
      "step": 7180
    },
    {
      "dynamics_loss_avg": 0.0601261492818594,
      "epoch": 6.847381591796875e-06,
      "step": 7180
    },
    {
      "action_loss_avg": 0.011496796971186996,
      "epoch": 6.847381591796875e-06,
      "step": 7180
    },
    {
      "epoch": 6.847381591796875e-06,
      "step": 7180,
      "training_step_time": 17.004027128219604
    },
    {
      "epoch": 6.848335266113281e-06,
      "model_forward_time": 5.580599069595337,
      "step": 7181
    },
    {
      "epoch": 6.848335266113281e-06,
      "step": 7181,
      "training_step_time": 17.020474672317505
    },
    {
      "epoch": 6.849288940429688e-06,
      "model_forward_time": 5.590699911117554,
      "step": 7182
    },
    {
      "epoch": 6.849288940429688e-06,
      "step": 7182,
      "training_step_time": 17.058056592941284
    },
    {
      "epoch": 6.850242614746094e-06,
      "model_forward_time": 5.609892845153809,
      "step": 7183
    },
    {
      "epoch": 6.850242614746094e-06,
      "step": 7183,
      "training_step_time": 17.118504524230957
    },
    {
      "epoch": 6.8511962890625e-06,
      "model_forward_time": 5.620312213897705,
      "step": 7184
    },
    {
      "epoch": 6.8511962890625e-06,
      "step": 7184,
      "training_step_time": 17.499626874923706
    },
    {
      "epoch": 6.852149963378906e-06,
      "model_forward_time": 5.618570804595947,
      "step": 7185
    },
    {
      "epoch": 6.852149963378906e-06,
      "step": 7185,
      "training_step_time": 17.130993843078613
    },
    {
      "epoch": 6.8531036376953124e-06,
      "model_forward_time": 5.621816158294678,
      "step": 7186
    },
    {
      "epoch": 6.8531036376953124e-06,
      "step": 7186,
      "training_step_time": 17.126216411590576
    },
    {
      "epoch": 6.854057312011719e-06,
      "model_forward_time": 5.61694073677063,
      "step": 7187
    },
    {
      "epoch": 6.854057312011719e-06,
      "step": 7187,
      "training_step_time": 17.108241319656372
    },
    {
      "epoch": 6.855010986328125e-06,
      "model_forward_time": 5.619570732116699,
      "step": 7188
    },
    {
      "epoch": 6.855010986328125e-06,
      "step": 7188,
      "training_step_time": 17.091254949569702
    },
    {
      "epoch": 6.855964660644531e-06,
      "model_forward_time": 5.589996337890625,
      "step": 7189
    },
    {
      "epoch": 6.855964660644531e-06,
      "step": 7189,
      "training_step_time": 17.02102279663086
    },
    {
      "epoch": 6.856918334960938e-06,
      "grad_norm": 1.227280855178833,
      "learning_rate": 1e-05,
      "loss": 0.0632,
      "step": 7190
    },
    {
      "epoch": 6.856918334960938e-06,
      "model_forward_time": 5.949535608291626,
      "step": 7190
    },
    {
      "dynamics_loss_avg": 0.057521221600472926,
      "epoch": 6.856918334960938e-06,
      "step": 7190
    },
    {
      "action_loss_avg": 0.007401834451593458,
      "epoch": 6.856918334960938e-06,
      "step": 7190
    },
    {
      "epoch": 6.856918334960938e-06,
      "step": 7190,
      "training_step_time": 17.352933406829834
    },
    {
      "epoch": 6.857872009277344e-06,
      "model_forward_time": 5.573040723800659,
      "step": 7191
    },
    {
      "epoch": 6.857872009277344e-06,
      "step": 7191,
      "training_step_time": 16.994271516799927
    },
    {
      "epoch": 6.85882568359375e-06,
      "model_forward_time": 5.59659218788147,
      "step": 7192
    },
    {
      "epoch": 6.85882568359375e-06,
      "step": 7192,
      "training_step_time": 17.038150310516357
    },
    {
      "epoch": 6.859779357910156e-06,
      "model_forward_time": 5.60528564453125,
      "step": 7193
    },
    {
      "epoch": 6.859779357910156e-06,
      "step": 7193,
      "training_step_time": 17.06766438484192
    },
    {
      "epoch": 6.8607330322265625e-06,
      "model_forward_time": 5.602426052093506,
      "step": 7194
    },
    {
      "epoch": 6.8607330322265625e-06,
      "step": 7194,
      "training_step_time": 17.066185235977173
    },
    {
      "epoch": 6.861686706542969e-06,
      "model_forward_time": 5.605957269668579,
      "step": 7195
    },
    {
      "epoch": 6.861686706542969e-06,
      "step": 7195,
      "training_step_time": 17.418482303619385
    },
    {
      "epoch": 6.862640380859375e-06,
      "model_forward_time": 5.596925497055054,
      "step": 7196
    },
    {
      "epoch": 6.862640380859375e-06,
      "step": 7196,
      "training_step_time": 17.023322820663452
    },
    {
      "epoch": 6.863594055175781e-06,
      "model_forward_time": 5.593897819519043,
      "step": 7197
    },
    {
      "epoch": 6.863594055175781e-06,
      "step": 7197,
      "training_step_time": 17.032390117645264
    },
    {
      "epoch": 6.864547729492188e-06,
      "model_forward_time": 5.592360734939575,
      "step": 7198
    },
    {
      "epoch": 6.864547729492188e-06,
      "step": 7198,
      "training_step_time": 17.062117338180542
    },
    {
      "epoch": 6.865501403808594e-06,
      "model_forward_time": 5.602924108505249,
      "step": 7199
    },
    {
      "epoch": 6.865501403808594e-06,
      "step": 7199,
      "training_step_time": 17.089126586914062
    },
    {
      "epoch": 6.866455078125e-06,
      "grad_norm": 0.5014627575874329,
      "learning_rate": 1e-05,
      "loss": 0.0894,
      "step": 7200
    },
    {
      "epoch": 6.866455078125e-06,
      "model_forward_time": 5.609764099121094,
      "step": 7200
    },
    {
      "dynamics_loss_avg": 0.07436857745051384,
      "epoch": 6.866455078125e-06,
      "step": 7200
    },
    {
      "action_loss_avg": 0.013567556021735072,
      "epoch": 6.866455078125e-06,
      "step": 7200
    },
    {
      "epoch": 6.866455078125e-06,
      "step": 7200,
      "training_step_time": 17.10596513748169
    },
    {
      "epoch": 6.867408752441406e-06,
      "model_forward_time": 5.979199409484863,
      "step": 7201
    },
    {
      "epoch": 6.867408752441406e-06,
      "step": 7201,
      "training_step_time": 17.482130527496338
    },
    {
      "epoch": 6.8683624267578125e-06,
      "model_forward_time": 5.6243979930877686,
      "step": 7202
    },
    {
      "epoch": 6.8683624267578125e-06,
      "step": 7202,
      "training_step_time": 17.111841678619385
    },
    {
      "epoch": 6.869316101074219e-06,
      "model_forward_time": 5.603121280670166,
      "step": 7203
    },
    {
      "epoch": 6.869316101074219e-06,
      "step": 7203,
      "training_step_time": 17.083163022994995
    },
    {
      "epoch": 6.870269775390625e-06,
      "model_forward_time": 5.603555202484131,
      "step": 7204
    },
    {
      "epoch": 6.870269775390625e-06,
      "step": 7204,
      "training_step_time": 17.05678629875183
    },
    {
      "epoch": 6.871223449707031e-06,
      "model_forward_time": 5.599980115890503,
      "step": 7205
    },
    {
      "epoch": 6.871223449707031e-06,
      "step": 7205,
      "training_step_time": 17.072465896606445
    },
    {
      "epoch": 6.872177124023438e-06,
      "model_forward_time": 5.956763029098511,
      "step": 7206
    },
    {
      "epoch": 6.872177124023438e-06,
      "step": 7206,
      "training_step_time": 17.429268836975098
    },
    {
      "epoch": 6.873130798339844e-06,
      "model_forward_time": 5.604028701782227,
      "step": 7207
    },
    {
      "epoch": 6.873130798339844e-06,
      "step": 7207,
      "training_step_time": 17.04682230949402
    },
    {
      "epoch": 6.87408447265625e-06,
      "model_forward_time": 5.6049110889434814,
      "step": 7208
    },
    {
      "epoch": 6.87408447265625e-06,
      "step": 7208,
      "training_step_time": 17.076746225357056
    },
    {
      "epoch": 6.875038146972656e-06,
      "model_forward_time": 5.598836183547974,
      "step": 7209
    },
    {
      "epoch": 6.875038146972656e-06,
      "step": 7209,
      "training_step_time": 17.047845125198364
    },
    {
      "epoch": 6.8759918212890625e-06,
      "grad_norm": 0.9926379323005676,
      "learning_rate": 1e-05,
      "loss": 0.0678,
      "step": 7210
    },
    {
      "epoch": 6.8759918212890625e-06,
      "model_forward_time": 5.599279403686523,
      "step": 7210
    },
    {
      "dynamics_loss_avg": 0.05621053371578455,
      "epoch": 6.8759918212890625e-06,
      "step": 7210
    },
    {
      "action_loss_avg": 0.009540623205248266,
      "epoch": 6.8759918212890625e-06,
      "step": 7210
    },
    {
      "epoch": 6.8759918212890625e-06,
      "step": 7210,
      "training_step_time": 17.033833980560303
    },
    {
      "epoch": 6.876945495605469e-06,
      "model_forward_time": 5.601731061935425,
      "step": 7211
    },
    {
      "epoch": 6.876945495605469e-06,
      "step": 7211,
      "training_step_time": 17.055185079574585
    },
    {
      "epoch": 6.877899169921875e-06,
      "model_forward_time": 5.9796459674835205,
      "step": 7212
    },
    {
      "epoch": 6.877899169921875e-06,
      "step": 7212,
      "training_step_time": 17.444822549819946
    },
    {
      "epoch": 6.878852844238281e-06,
      "model_forward_time": 5.612856149673462,
      "step": 7213
    },
    {
      "epoch": 6.878852844238281e-06,
      "step": 7213,
      "training_step_time": 17.080975770950317
    },
    {
      "epoch": 6.879806518554688e-06,
      "model_forward_time": 5.605482816696167,
      "step": 7214
    },
    {
      "epoch": 6.879806518554688e-06,
      "step": 7214,
      "training_step_time": 17.043458700180054
    },
    {
      "epoch": 6.880760192871094e-06,
      "model_forward_time": 5.6003663539886475,
      "step": 7215
    },
    {
      "epoch": 6.880760192871094e-06,
      "step": 7215,
      "training_step_time": 17.03907871246338
    },
    {
      "epoch": 6.8817138671875e-06,
      "model_forward_time": 5.6078550815582275,
      "step": 7216
    },
    {
      "epoch": 6.8817138671875e-06,
      "step": 7216,
      "training_step_time": 17.062943935394287
    },
    {
      "epoch": 6.882667541503906e-06,
      "model_forward_time": 5.602365493774414,
      "step": 7217
    },
    {
      "epoch": 6.882667541503906e-06,
      "step": 7217,
      "training_step_time": 17.43422794342041
    },
    {
      "epoch": 6.8836212158203125e-06,
      "model_forward_time": 5.600760221481323,
      "step": 7218
    },
    {
      "epoch": 6.8836212158203125e-06,
      "step": 7218,
      "training_step_time": 17.051156997680664
    },
    {
      "epoch": 6.884574890136719e-06,
      "model_forward_time": 5.589712619781494,
      "step": 7219
    },
    {
      "epoch": 6.884574890136719e-06,
      "step": 7219,
      "training_step_time": 17.017045974731445
    },
    {
      "epoch": 6.885528564453125e-06,
      "grad_norm": 1.5077108144760132,
      "learning_rate": 1e-05,
      "loss": 0.0664,
      "step": 7220
    },
    {
      "epoch": 6.885528564453125e-06,
      "model_forward_time": 5.595197677612305,
      "step": 7220
    },
    {
      "dynamics_loss_avg": 0.05742271095514297,
      "epoch": 6.885528564453125e-06,
      "step": 7220
    },
    {
      "action_loss_avg": 0.011380651779472829,
      "epoch": 6.885528564453125e-06,
      "step": 7220
    },
    {
      "epoch": 6.885528564453125e-06,
      "step": 7220,
      "training_step_time": 17.053206205368042
    },
    {
      "epoch": 6.886482238769531e-06,
      "model_forward_time": 5.594743728637695,
      "step": 7221
    },
    {
      "epoch": 6.886482238769531e-06,
      "step": 7221,
      "training_step_time": 17.039355278015137
    },
    {
      "epoch": 6.887435913085938e-06,
      "model_forward_time": 5.59491229057312,
      "step": 7222
    },
    {
      "epoch": 6.887435913085938e-06,
      "step": 7222,
      "training_step_time": 17.03617238998413
    },
    {
      "epoch": 6.888389587402344e-06,
      "model_forward_time": 5.964903831481934,
      "step": 7223
    },
    {
      "epoch": 6.888389587402344e-06,
      "step": 7223,
      "training_step_time": 17.40600562095642
    },
    {
      "epoch": 6.88934326171875e-06,
      "model_forward_time": 5.590143442153931,
      "step": 7224
    },
    {
      "epoch": 6.88934326171875e-06,
      "step": 7224,
      "training_step_time": 17.032114267349243
    },
    {
      "epoch": 6.890296936035156e-06,
      "model_forward_time": 5.600941181182861,
      "step": 7225
    },
    {
      "epoch": 6.890296936035156e-06,
      "step": 7225,
      "training_step_time": 17.08733582496643
    },
    {
      "epoch": 6.8912506103515625e-06,
      "model_forward_time": 5.606049537658691,
      "step": 7226
    },
    {
      "epoch": 6.8912506103515625e-06,
      "step": 7226,
      "training_step_time": 17.099836111068726
    },
    {
      "epoch": 6.892204284667969e-06,
      "model_forward_time": 5.610507249832153,
      "step": 7227
    },
    {
      "epoch": 6.892204284667969e-06,
      "step": 7227,
      "training_step_time": 17.11014485359192
    },
    {
      "epoch": 6.893157958984375e-06,
      "model_forward_time": 5.606436729431152,
      "step": 7228
    },
    {
      "epoch": 6.893157958984375e-06,
      "step": 7228,
      "training_step_time": 17.433977603912354
    },
    {
      "epoch": 6.894111633300781e-06,
      "model_forward_time": 5.587336301803589,
      "step": 7229
    },
    {
      "epoch": 6.894111633300781e-06,
      "step": 7229,
      "training_step_time": 16.997718572616577
    },
    {
      "epoch": 6.895065307617188e-06,
      "grad_norm": 0.7497448325157166,
      "learning_rate": 1e-05,
      "loss": 0.0798,
      "step": 7230
    },
    {
      "epoch": 6.895065307617188e-06,
      "model_forward_time": 5.570788860321045,
      "step": 7230
    },
    {
      "dynamics_loss_avg": 0.06644189078360796,
      "epoch": 6.895065307617188e-06,
      "step": 7230
    },
    {
      "action_loss_avg": 0.012003388837911189,
      "epoch": 6.895065307617188e-06,
      "step": 7230
    },
    {
      "epoch": 6.895065307617188e-06,
      "step": 7230,
      "training_step_time": 17.01701855659485
    },
    {
      "epoch": 6.896018981933594e-06,
      "model_forward_time": 5.563568830490112,
      "step": 7231
    },
    {
      "epoch": 6.896018981933594e-06,
      "step": 7231,
      "training_step_time": 16.975637912750244
    },
    {
      "epoch": 6.89697265625e-06,
      "model_forward_time": 5.565535306930542,
      "step": 7232
    },
    {
      "epoch": 6.89697265625e-06,
      "step": 7232,
      "training_step_time": 16.967350721359253
    },
    {
      "epoch": 6.897926330566406e-06,
      "model_forward_time": 5.5639729499816895,
      "step": 7233
    },
    {
      "epoch": 6.897926330566406e-06,
      "step": 7233,
      "training_step_time": 16.988162994384766
    },
    {
      "epoch": 6.8988800048828125e-06,
      "model_forward_time": 5.972397327423096,
      "step": 7234
    },
    {
      "epoch": 6.8988800048828125e-06,
      "step": 7234,
      "training_step_time": 17.40454912185669
    },
    {
      "epoch": 6.899833679199219e-06,
      "model_forward_time": 5.585965633392334,
      "step": 7235
    },
    {
      "epoch": 6.899833679199219e-06,
      "step": 7235,
      "training_step_time": 17.01106905937195
    },
    {
      "epoch": 6.900787353515625e-06,
      "model_forward_time": 5.5926830768585205,
      "step": 7236
    },
    {
      "epoch": 6.900787353515625e-06,
      "step": 7236,
      "training_step_time": 17.044847011566162
    },
    {
      "epoch": 6.901741027832031e-06,
      "model_forward_time": 5.604822635650635,
      "step": 7237
    },
    {
      "epoch": 6.901741027832031e-06,
      "step": 7237,
      "training_step_time": 17.08150863647461
    },
    {
      "epoch": 6.902694702148438e-06,
      "model_forward_time": 5.607733964920044,
      "step": 7238
    },
    {
      "epoch": 6.902694702148438e-06,
      "step": 7238,
      "training_step_time": 17.06338930130005
    },
    {
      "epoch": 6.903648376464844e-06,
      "model_forward_time": 5.960747718811035,
      "step": 7239
    },
    {
      "epoch": 6.903648376464844e-06,
      "step": 7239,
      "training_step_time": 17.4443097114563
    },
    {
      "epoch": 6.90460205078125e-06,
      "grad_norm": 0.6836972236633301,
      "learning_rate": 1e-05,
      "loss": 0.0677,
      "step": 7240
    },
    {
      "epoch": 6.90460205078125e-06,
      "model_forward_time": 5.635295152664185,
      "step": 7240
    },
    {
      "dynamics_loss_avg": 0.05382759720087051,
      "epoch": 6.90460205078125e-06,
      "step": 7240
    },
    {
      "action_loss_avg": 0.009679794777184725,
      "epoch": 6.90460205078125e-06,
      "step": 7240
    },
    {
      "epoch": 6.90460205078125e-06,
      "step": 7240,
      "training_step_time": 17.15726661682129
    },
    {
      "epoch": 6.905555725097656e-06,
      "model_forward_time": 5.611351490020752,
      "step": 7241
    },
    {
      "epoch": 6.905555725097656e-06,
      "step": 7241,
      "training_step_time": 17.07782292366028
    },
    {
      "epoch": 6.9065093994140625e-06,
      "model_forward_time": 5.58015775680542,
      "step": 7242
    },
    {
      "epoch": 6.9065093994140625e-06,
      "step": 7242,
      "training_step_time": 17.025264501571655
    },
    {
      "epoch": 6.907463073730469e-06,
      "model_forward_time": 5.572240114212036,
      "step": 7243
    },
    {
      "epoch": 6.907463073730469e-06,
      "step": 7243,
      "training_step_time": 16.99881625175476
    },
    {
      "epoch": 6.908416748046875e-06,
      "model_forward_time": 5.567430734634399,
      "step": 7244
    },
    {
      "epoch": 6.908416748046875e-06,
      "step": 7244,
      "training_step_time": 16.976834058761597
    },
    {
      "epoch": 6.909370422363281e-06,
      "model_forward_time": 5.922543287277222,
      "step": 7245
    },
    {
      "epoch": 6.909370422363281e-06,
      "step": 7245,
      "training_step_time": 17.352580070495605
    },
    {
      "epoch": 6.910324096679688e-06,
      "model_forward_time": 5.573575735092163,
      "step": 7246
    },
    {
      "epoch": 6.910324096679688e-06,
      "step": 7246,
      "training_step_time": 16.992339849472046
    },
    {
      "epoch": 6.911277770996094e-06,
      "model_forward_time": 5.573659658432007,
      "step": 7247
    },
    {
      "epoch": 6.911277770996094e-06,
      "step": 7247,
      "training_step_time": 16.982638359069824
    },
    {
      "epoch": 6.9122314453125e-06,
      "model_forward_time": 5.596518278121948,
      "step": 7248
    },
    {
      "epoch": 6.9122314453125e-06,
      "step": 7248,
      "training_step_time": 17.014123678207397
    },
    {
      "epoch": 6.913185119628906e-06,
      "model_forward_time": 5.579589366912842,
      "step": 7249
    },
    {
      "epoch": 6.913185119628906e-06,
      "step": 7249,
      "training_step_time": 17.024759769439697
    },
    {
      "epoch": 6.9141387939453125e-06,
      "grad_norm": 1.5826445817947388,
      "learning_rate": 1e-05,
      "loss": 0.0694,
      "step": 7250
    },
    {
      "epoch": 6.9141387939453125e-06,
      "model_forward_time": 5.581673622131348,
      "step": 7250
    },
    {
      "dynamics_loss_avg": 0.055371655896306035,
      "epoch": 6.9141387939453125e-06,
      "step": 7250
    },
    {
      "action_loss_avg": 0.013164791418239475,
      "epoch": 6.9141387939453125e-06,
      "step": 7250
    },
    {
      "epoch": 6.9141387939453125e-06,
      "step": 7250,
      "training_step_time": 17.39710545539856
    },
    {
      "epoch": 6.915092468261719e-06,
      "model_forward_time": 5.5941243171691895,
      "step": 7251
    },
    {
      "epoch": 6.915092468261719e-06,
      "step": 7251,
      "training_step_time": 17.074641704559326
    },
    {
      "epoch": 6.916046142578125e-06,
      "model_forward_time": 5.610213041305542,
      "step": 7252
    },
    {
      "epoch": 6.916046142578125e-06,
      "step": 7252,
      "training_step_time": 17.0977303981781
    },
    {
      "epoch": 6.916999816894531e-06,
      "model_forward_time": 5.604544401168823,
      "step": 7253
    },
    {
      "epoch": 6.916999816894531e-06,
      "step": 7253,
      "training_step_time": 17.099848985671997
    },
    {
      "epoch": 6.917953491210937e-06,
      "model_forward_time": 5.619762182235718,
      "step": 7254
    },
    {
      "epoch": 6.917953491210937e-06,
      "step": 7254,
      "training_step_time": 17.083841562271118
    },
    {
      "epoch": 6.918907165527344e-06,
      "model_forward_time": 5.588611125946045,
      "step": 7255
    },
    {
      "epoch": 6.918907165527344e-06,
      "step": 7255,
      "training_step_time": 17.0209903717041
    },
    {
      "epoch": 6.91986083984375e-06,
      "model_forward_time": 5.9528234004974365,
      "step": 7256
    },
    {
      "epoch": 6.91986083984375e-06,
      "step": 7256,
      "training_step_time": 17.36910343170166
    },
    {
      "epoch": 6.920814514160156e-06,
      "model_forward_time": 5.586815118789673,
      "step": 7257
    },
    {
      "epoch": 6.920814514160156e-06,
      "step": 7257,
      "training_step_time": 17.028228759765625
    },
    {
      "epoch": 6.9217681884765625e-06,
      "model_forward_time": 5.592280149459839,
      "step": 7258
    },
    {
      "epoch": 6.9217681884765625e-06,
      "step": 7258,
      "training_step_time": 17.013508081436157
    },
    {
      "epoch": 6.922721862792969e-06,
      "model_forward_time": 5.5905866622924805,
      "step": 7259
    },
    {
      "epoch": 6.922721862792969e-06,
      "step": 7259,
      "training_step_time": 17.00783085823059
    },
    {
      "epoch": 6.923675537109375e-06,
      "grad_norm": 1.5231820344924927,
      "learning_rate": 1e-05,
      "loss": 0.0748,
      "step": 7260
    },
    {
      "epoch": 6.923675537109375e-06,
      "model_forward_time": 5.588452339172363,
      "step": 7260
    },
    {
      "dynamics_loss_avg": 0.06328053474426269,
      "epoch": 6.923675537109375e-06,
      "step": 7260
    },
    {
      "action_loss_avg": 0.012674249522387982,
      "epoch": 6.923675537109375e-06,
      "step": 7260
    },
    {
      "epoch": 6.923675537109375e-06,
      "step": 7260,
      "training_step_time": 17.034371376037598
    },
    {
      "epoch": 6.924629211425781e-06,
      "model_forward_time": 5.945890188217163,
      "step": 7261
    },
    {
      "epoch": 6.924629211425781e-06,
      "step": 7261,
      "training_step_time": 17.41937017440796
    },
    {
      "epoch": 6.925582885742187e-06,
      "model_forward_time": 5.616384267807007,
      "step": 7262
    },
    {
      "epoch": 6.925582885742187e-06,
      "step": 7262,
      "training_step_time": 17.11872124671936
    },
    {
      "epoch": 6.926536560058594e-06,
      "model_forward_time": 5.612054824829102,
      "step": 7263
    },
    {
      "epoch": 6.926536560058594e-06,
      "step": 7263,
      "training_step_time": 17.129552125930786
    },
    {
      "epoch": 6.927490234375e-06,
      "model_forward_time": 5.611766576766968,
      "step": 7264
    },
    {
      "epoch": 6.927490234375e-06,
      "step": 7264,
      "training_step_time": 17.089706897735596
    },
    {
      "epoch": 6.928443908691406e-06,
      "model_forward_time": 5.608316659927368,
      "step": 7265
    },
    {
      "epoch": 6.928443908691406e-06,
      "step": 7265,
      "training_step_time": 17.094927310943604
    },
    {
      "epoch": 6.9293975830078125e-06,
      "model_forward_time": 5.6038572788238525,
      "step": 7266
    },
    {
      "epoch": 6.9293975830078125e-06,
      "step": 7266,
      "training_step_time": 17.08793592453003
    },
    {
      "epoch": 6.930351257324219e-06,
      "model_forward_time": 5.955345153808594,
      "step": 7267
    },
    {
      "epoch": 6.930351257324219e-06,
      "step": 7267,
      "training_step_time": 17.397872924804688
    },
    {
      "epoch": 6.931304931640625e-06,
      "model_forward_time": 5.602675199508667,
      "step": 7268
    },
    {
      "epoch": 6.931304931640625e-06,
      "step": 7268,
      "training_step_time": 17.04335904121399
    },
    {
      "epoch": 6.932258605957031e-06,
      "model_forward_time": 5.584079742431641,
      "step": 7269
    },
    {
      "epoch": 6.932258605957031e-06,
      "step": 7269,
      "training_step_time": 17.013739824295044
    },
    {
      "epoch": 6.933212280273437e-06,
      "grad_norm": 1.026630163192749,
      "learning_rate": 1e-05,
      "loss": 0.0668,
      "step": 7270
    },
    {
      "epoch": 6.933212280273437e-06,
      "model_forward_time": 5.595466136932373,
      "step": 7270
    },
    {
      "dynamics_loss_avg": 0.06005933713167906,
      "epoch": 6.933212280273437e-06,
      "step": 7270
    },
    {
      "action_loss_avg": 0.008682995988056063,
      "epoch": 6.933212280273437e-06,
      "step": 7270
    },
    {
      "epoch": 6.933212280273437e-06,
      "step": 7270,
      "training_step_time": 17.079663276672363
    },
    {
      "epoch": 6.934165954589844e-06,
      "model_forward_time": 5.618115186691284,
      "step": 7271
    },
    {
      "epoch": 6.934165954589844e-06,
      "step": 7271,
      "training_step_time": 17.118053436279297
    },
    {
      "epoch": 6.93511962890625e-06,
      "model_forward_time": 4.672467231750488,
      "step": 7272
    },
    {
      "epoch": 6.93511962890625e-06,
      "step": 7272,
      "training_step_time": 13.212444067001343
    },
    {
      "epoch": 6.936073303222656e-06,
      "model_forward_time": 5.633514642715454,
      "step": 7273
    },
    {
      "epoch": 6.936073303222656e-06,
      "step": 7273,
      "training_step_time": 17.125118017196655
    },
    {
      "epoch": 6.9370269775390625e-06,
      "model_forward_time": 5.629231929779053,
      "step": 7274
    },
    {
      "epoch": 6.9370269775390625e-06,
      "step": 7274,
      "training_step_time": 17.115768671035767
    },
    {
      "epoch": 6.937980651855469e-06,
      "model_forward_time": 5.590954542160034,
      "step": 7275
    },
    {
      "epoch": 6.937980651855469e-06,
      "step": 7275,
      "training_step_time": 17.002216577529907
    },
    {
      "epoch": 6.938934326171875e-06,
      "model_forward_time": 5.5689537525177,
      "step": 7276
    },
    {
      "epoch": 6.938934326171875e-06,
      "step": 7276,
      "training_step_time": 17.005704641342163
    },
    {
      "epoch": 6.939888000488281e-06,
      "model_forward_time": 5.571739196777344,
      "step": 7277
    },
    {
      "epoch": 6.939888000488281e-06,
      "step": 7277,
      "training_step_time": 17.345844507217407
    },
    {
      "epoch": 6.940841674804687e-06,
      "model_forward_time": 5.593092679977417,
      "step": 7278
    },
    {
      "epoch": 6.940841674804687e-06,
      "step": 7278,
      "training_step_time": 17.015657663345337
    },
    {
      "epoch": 6.941795349121094e-06,
      "model_forward_time": 5.573852062225342,
      "step": 7279
    },
    {
      "epoch": 6.941795349121094e-06,
      "step": 7279,
      "training_step_time": 16.997687578201294
    },
    {
      "epoch": 6.9427490234375e-06,
      "grad_norm": 1.8855115175247192,
      "learning_rate": 1e-05,
      "loss": 0.0871,
      "step": 7280
    },
    {
      "epoch": 6.9427490234375e-06,
      "model_forward_time": 5.575250864028931,
      "step": 7280
    },
    {
      "dynamics_loss_avg": 0.0731366328895092,
      "epoch": 6.9427490234375e-06,
      "step": 7280
    },
    {
      "action_loss_avg": 0.01305537314619869,
      "epoch": 6.9427490234375e-06,
      "step": 7280
    },
    {
      "epoch": 6.9427490234375e-06,
      "step": 7280,
      "training_step_time": 17.02630066871643
    },
    {
      "epoch": 6.943702697753906e-06,
      "model_forward_time": 5.587373733520508,
      "step": 7281
    },
    {
      "epoch": 6.943702697753906e-06,
      "step": 7281,
      "training_step_time": 17.060848474502563
    },
    {
      "epoch": 6.9446563720703125e-06,
      "model_forward_time": 5.6172637939453125,
      "step": 7282
    },
    {
      "epoch": 6.9446563720703125e-06,
      "step": 7282,
      "training_step_time": 17.12101674079895
    },
    {
      "epoch": 6.945610046386719e-06,
      "model_forward_time": 5.964473009109497,
      "step": 7283
    },
    {
      "epoch": 6.945610046386719e-06,
      "step": 7283,
      "training_step_time": 17.450916528701782
    },
    {
      "epoch": 6.946563720703125e-06,
      "model_forward_time": 5.604111671447754,
      "step": 7284
    },
    {
      "epoch": 6.946563720703125e-06,
      "step": 7284,
      "training_step_time": 17.05623960494995
    },
    {
      "epoch": 6.947517395019531e-06,
      "model_forward_time": 5.591456890106201,
      "step": 7285
    },
    {
      "epoch": 6.947517395019531e-06,
      "step": 7285,
      "training_step_time": 17.010700702667236
    },
    {
      "epoch": 6.948471069335937e-06,
      "model_forward_time": 5.581892967224121,
      "step": 7286
    },
    {
      "epoch": 6.948471069335937e-06,
      "step": 7286,
      "training_step_time": 17.004899740219116
    },
    {
      "epoch": 6.949424743652344e-06,
      "model_forward_time": 5.590538501739502,
      "step": 7287
    },
    {
      "epoch": 6.949424743652344e-06,
      "step": 7287,
      "training_step_time": 17.037333250045776
    },
    {
      "epoch": 6.95037841796875e-06,
      "model_forward_time": 5.976243495941162,
      "step": 7288
    },
    {
      "epoch": 6.95037841796875e-06,
      "step": 7288,
      "training_step_time": 17.4265456199646
    },
    {
      "epoch": 6.951332092285156e-06,
      "model_forward_time": 5.600797414779663,
      "step": 7289
    },
    {
      "epoch": 6.951332092285156e-06,
      "step": 7289,
      "training_step_time": 17.055872917175293
    },
    {
      "epoch": 6.9522857666015625e-06,
      "grad_norm": 0.6797547936439514,
      "learning_rate": 1e-05,
      "loss": 0.0672,
      "step": 7290
    },
    {
      "epoch": 6.9522857666015625e-06,
      "model_forward_time": 5.6001811027526855,
      "step": 7290
    },
    {
      "dynamics_loss_avg": 0.055715208500623704,
      "epoch": 6.9522857666015625e-06,
      "step": 7290
    },
    {
      "action_loss_avg": 0.01091987252002582,
      "epoch": 6.9522857666015625e-06,
      "step": 7290
    },
    {
      "epoch": 6.9522857666015625e-06,
      "step": 7290,
      "training_step_time": 17.065895080566406
    },
    {
      "epoch": 6.953239440917969e-06,
      "model_forward_time": 5.5957536697387695,
      "step": 7291
    },
    {
      "epoch": 6.953239440917969e-06,
      "step": 7291,
      "training_step_time": 17.043122053146362
    },
    {
      "epoch": 6.954193115234375e-06,
      "model_forward_time": 5.590916633605957,
      "step": 7292
    },
    {
      "epoch": 6.954193115234375e-06,
      "step": 7292,
      "training_step_time": 17.039575338363647
    },
    {
      "epoch": 6.955146789550781e-06,
      "model_forward_time": 5.592133045196533,
      "step": 7293
    },
    {
      "epoch": 6.955146789550781e-06,
      "step": 7293,
      "training_step_time": 17.03758144378662
    },
    {
      "epoch": 6.956100463867187e-06,
      "model_forward_time": 5.929667234420776,
      "step": 7294
    },
    {
      "epoch": 6.956100463867187e-06,
      "step": 7294,
      "training_step_time": 17.355833530426025
    },
    {
      "epoch": 6.957054138183594e-06,
      "model_forward_time": 5.610884189605713,
      "step": 7295
    },
    {
      "epoch": 6.957054138183594e-06,
      "step": 7295,
      "training_step_time": 17.05111575126648
    },
    {
      "epoch": 6.9580078125e-06,
      "model_forward_time": 5.614943742752075,
      "step": 7296
    },
    {
      "epoch": 6.9580078125e-06,
      "step": 7296,
      "training_step_time": 17.088533639907837
    },
    {
      "epoch": 6.958961486816406e-06,
      "model_forward_time": 5.607733488082886,
      "step": 7297
    },
    {
      "epoch": 6.958961486816406e-06,
      "step": 7297,
      "training_step_time": 17.095829725265503
    },
    {
      "epoch": 6.9599151611328125e-06,
      "model_forward_time": 5.608416557312012,
      "step": 7298
    },
    {
      "epoch": 6.9599151611328125e-06,
      "step": 7298,
      "training_step_time": 17.072531700134277
    },
    {
      "epoch": 6.960868835449219e-06,
      "model_forward_time": 5.959372282028198,
      "step": 7299
    },
    {
      "epoch": 6.960868835449219e-06,
      "step": 7299,
      "training_step_time": 17.391340732574463
    },
    {
      "epoch": 6.961822509765625e-06,
      "grad_norm": 0.4152035117149353,
      "learning_rate": 1e-05,
      "loss": 0.0726,
      "step": 7300
    },
    {
      "epoch": 6.961822509765625e-06,
      "model_forward_time": 5.58208441734314,
      "step": 7300
    },
    {
      "dynamics_loss_avg": 0.05975989382714033,
      "epoch": 6.961822509765625e-06,
      "step": 7300
    },
    {
      "action_loss_avg": 0.011911889887414873,
      "epoch": 6.961822509765625e-06,
      "step": 7300
    },
    {
      "epoch": 6.961822509765625e-06,
      "step": 7300,
      "training_step_time": 17.030973196029663
    },
    {
      "epoch": 6.962776184082031e-06,
      "model_forward_time": 5.588429927825928,
      "step": 7301
    },
    {
      "epoch": 6.962776184082031e-06,
      "step": 7301,
      "training_step_time": 17.04244065284729
    },
    {
      "epoch": 6.963729858398437e-06,
      "model_forward_time": 5.601093769073486,
      "step": 7302
    },
    {
      "epoch": 6.963729858398437e-06,
      "step": 7302,
      "training_step_time": 17.05716562271118
    },
    {
      "epoch": 6.964683532714844e-06,
      "model_forward_time": 5.5918145179748535,
      "step": 7303
    },
    {
      "epoch": 6.964683532714844e-06,
      "step": 7303,
      "training_step_time": 17.039703607559204
    },
    {
      "epoch": 6.96563720703125e-06,
      "model_forward_time": 5.589539527893066,
      "step": 7304
    },
    {
      "epoch": 6.96563720703125e-06,
      "step": 7304,
      "training_step_time": 17.385422229766846
    },
    {
      "epoch": 6.966590881347656e-06,
      "model_forward_time": 5.596095085144043,
      "step": 7305
    },
    {
      "epoch": 6.966590881347656e-06,
      "step": 7305,
      "training_step_time": 17.02837824821472
    },
    {
      "epoch": 6.9675445556640625e-06,
      "model_forward_time": 5.589035749435425,
      "step": 7306
    },
    {
      "epoch": 6.9675445556640625e-06,
      "step": 7306,
      "training_step_time": 17.042802095413208
    },
    {
      "epoch": 6.968498229980469e-06,
      "model_forward_time": 5.600708723068237,
      "step": 7307
    },
    {
      "epoch": 6.968498229980469e-06,
      "step": 7307,
      "training_step_time": 17.064780712127686
    },
    {
      "epoch": 6.969451904296875e-06,
      "model_forward_time": 5.587024450302124,
      "step": 7308
    },
    {
      "epoch": 6.969451904296875e-06,
      "step": 7308,
      "training_step_time": 17.027449131011963
    },
    {
      "epoch": 6.970405578613281e-06,
      "model_forward_time": 5.582846641540527,
      "step": 7309
    },
    {
      "epoch": 6.970405578613281e-06,
      "step": 7309,
      "training_step_time": 17.034374237060547
    },
    {
      "epoch": 6.971359252929687e-06,
      "grad_norm": 0.49914804100990295,
      "learning_rate": 1e-05,
      "loss": 0.0505,
      "step": 7310
    },
    {
      "epoch": 6.971359252929687e-06,
      "model_forward_time": 5.9471235275268555,
      "step": 7310
    },
    {
      "dynamics_loss_avg": 0.05152395684272051,
      "epoch": 6.971359252929687e-06,
      "step": 7310
    },
    {
      "action_loss_avg": 0.006695485301315784,
      "epoch": 6.971359252929687e-06,
      "step": 7310
    },
    {
      "epoch": 6.971359252929687e-06,
      "step": 7310,
      "training_step_time": 17.366599798202515
    },
    {
      "epoch": 6.972312927246094e-06,
      "model_forward_time": 5.589138984680176,
      "step": 7311
    },
    {
      "epoch": 6.972312927246094e-06,
      "step": 7311,
      "training_step_time": 17.06789517402649
    },
    {
      "epoch": 6.9732666015625e-06,
      "model_forward_time": 5.621172666549683,
      "step": 7312
    },
    {
      "epoch": 6.9732666015625e-06,
      "step": 7312,
      "training_step_time": 17.13360619544983
    },
    {
      "epoch": 6.974220275878906e-06,
      "model_forward_time": 5.620458126068115,
      "step": 7313
    },
    {
      "epoch": 6.974220275878906e-06,
      "step": 7313,
      "training_step_time": 17.125566482543945
    },
    {
      "epoch": 6.9751739501953126e-06,
      "model_forward_time": 5.6078290939331055,
      "step": 7314
    },
    {
      "epoch": 6.9751739501953126e-06,
      "step": 7314,
      "training_step_time": 17.089679956436157
    },
    {
      "epoch": 6.976127624511719e-06,
      "model_forward_time": 5.943902969360352,
      "step": 7315
    },
    {
      "epoch": 6.976127624511719e-06,
      "step": 7315,
      "training_step_time": 17.38685369491577
    },
    {
      "epoch": 6.977081298828125e-06,
      "model_forward_time": 5.593985557556152,
      "step": 7316
    },
    {
      "epoch": 6.977081298828125e-06,
      "step": 7316,
      "training_step_time": 17.02364468574524
    },
    {
      "epoch": 6.978034973144531e-06,
      "model_forward_time": 5.586144208908081,
      "step": 7317
    },
    {
      "epoch": 6.978034973144531e-06,
      "step": 7317,
      "training_step_time": 17.00043797492981
    },
    {
      "epoch": 6.978988647460937e-06,
      "model_forward_time": 5.594256639480591,
      "step": 7318
    },
    {
      "epoch": 6.978988647460937e-06,
      "step": 7318,
      "training_step_time": 17.043545484542847
    },
    {
      "epoch": 6.979942321777344e-06,
      "model_forward_time": 5.601640462875366,
      "step": 7319
    },
    {
      "epoch": 6.979942321777344e-06,
      "step": 7319,
      "training_step_time": 17.070430040359497
    },
    {
      "epoch": 6.98089599609375e-06,
      "grad_norm": 0.6545463800430298,
      "learning_rate": 1e-05,
      "loss": 0.0831,
      "step": 7320
    },
    {
      "epoch": 6.98089599609375e-06,
      "model_forward_time": 5.600825071334839,
      "step": 7320
    },
    {
      "dynamics_loss_avg": 0.06256331354379654,
      "epoch": 6.98089599609375e-06,
      "step": 7320
    },
    {
      "action_loss_avg": 0.011508869216777384,
      "epoch": 6.98089599609375e-06,
      "step": 7320
    },
    {
      "epoch": 6.98089599609375e-06,
      "step": 7320,
      "training_step_time": 17.06134271621704
    },
    {
      "epoch": 6.981849670410156e-06,
      "model_forward_time": 5.9602813720703125,
      "step": 7321
    },
    {
      "epoch": 6.981849670410156e-06,
      "step": 7321,
      "training_step_time": 17.38183856010437
    },
    {
      "epoch": 6.9828033447265626e-06,
      "model_forward_time": 5.577849388122559,
      "step": 7322
    },
    {
      "epoch": 6.9828033447265626e-06,
      "step": 7322,
      "training_step_time": 17.01892066001892
    },
    {
      "epoch": 6.983757019042969e-06,
      "model_forward_time": 5.583025693893433,
      "step": 7323
    },
    {
      "epoch": 6.983757019042969e-06,
      "step": 7323,
      "training_step_time": 17.035314321517944
    },
    {
      "epoch": 6.984710693359375e-06,
      "model_forward_time": 5.601047992706299,
      "step": 7324
    },
    {
      "epoch": 6.984710693359375e-06,
      "step": 7324,
      "training_step_time": 17.117044687271118
    },
    {
      "epoch": 6.985664367675781e-06,
      "model_forward_time": 5.629518270492554,
      "step": 7325
    },
    {
      "epoch": 6.985664367675781e-06,
      "step": 7325,
      "training_step_time": 17.135183572769165
    },
    {
      "epoch": 6.986618041992187e-06,
      "model_forward_time": 5.971107006072998,
      "step": 7326
    },
    {
      "epoch": 6.986618041992187e-06,
      "step": 7326,
      "training_step_time": 17.451112270355225
    },
    {
      "epoch": 6.987571716308594e-06,
      "model_forward_time": 5.607842445373535,
      "step": 7327
    },
    {
      "epoch": 6.987571716308594e-06,
      "step": 7327,
      "training_step_time": 17.094881057739258
    },
    {
      "epoch": 6.988525390625e-06,
      "model_forward_time": 5.602360963821411,
      "step": 7328
    },
    {
      "epoch": 6.988525390625e-06,
      "step": 7328,
      "training_step_time": 17.088818788528442
    },
    {
      "epoch": 6.989479064941406e-06,
      "model_forward_time": 5.601205825805664,
      "step": 7329
    },
    {
      "epoch": 6.989479064941406e-06,
      "step": 7329,
      "training_step_time": 17.06345510482788
    },
    {
      "epoch": 6.9904327392578126e-06,
      "grad_norm": 0.7741913199424744,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 7330
    },
    {
      "epoch": 6.9904327392578126e-06,
      "model_forward_time": 5.586786508560181,
      "step": 7330
    },
    {
      "dynamics_loss_avg": 0.0599641565233469,
      "epoch": 6.9904327392578126e-06,
      "step": 7330
    },
    {
      "action_loss_avg": 0.010904435394331813,
      "epoch": 6.9904327392578126e-06,
      "step": 7330
    },
    {
      "epoch": 6.9904327392578126e-06,
      "step": 7330,
      "training_step_time": 17.00061559677124
    },
    {
      "epoch": 6.991386413574219e-06,
      "model_forward_time": 5.581627607345581,
      "step": 7331
    },
    {
      "epoch": 6.991386413574219e-06,
      "step": 7331,
      "training_step_time": 16.996704578399658
    },
    {
      "epoch": 6.992340087890625e-06,
      "model_forward_time": 5.954617261886597,
      "step": 7332
    },
    {
      "epoch": 6.992340087890625e-06,
      "step": 7332,
      "training_step_time": 17.37278914451599
    },
    {
      "epoch": 6.9932937622070314e-06,
      "model_forward_time": 5.615982532501221,
      "step": 7333
    },
    {
      "epoch": 6.9932937622070314e-06,
      "step": 7333,
      "training_step_time": 17.07564115524292
    },
    {
      "epoch": 6.994247436523437e-06,
      "model_forward_time": 5.596127033233643,
      "step": 7334
    },
    {
      "epoch": 6.994247436523437e-06,
      "step": 7334,
      "training_step_time": 17.04983162879944
    },
    {
      "epoch": 6.995201110839844e-06,
      "model_forward_time": 5.593806982040405,
      "step": 7335
    },
    {
      "epoch": 6.995201110839844e-06,
      "step": 7335,
      "training_step_time": 17.038896083831787
    },
    {
      "epoch": 6.99615478515625e-06,
      "model_forward_time": 5.596988677978516,
      "step": 7336
    },
    {
      "epoch": 6.99615478515625e-06,
      "step": 7336,
      "training_step_time": 17.03713607788086
    },
    {
      "epoch": 6.997108459472656e-06,
      "model_forward_time": 5.9476234912872314,
      "step": 7337
    },
    {
      "epoch": 6.997108459472656e-06,
      "step": 7337,
      "training_step_time": 17.398648738861084
    },
    {
      "epoch": 6.9980621337890626e-06,
      "model_forward_time": 5.599203586578369,
      "step": 7338
    },
    {
      "epoch": 6.9980621337890626e-06,
      "step": 7338,
      "training_step_time": 17.092711448669434
    },
    {
      "epoch": 6.999015808105469e-06,
      "model_forward_time": 5.624380588531494,
      "step": 7339
    },
    {
      "epoch": 6.999015808105469e-06,
      "step": 7339,
      "training_step_time": 17.133901357650757
    },
    {
      "epoch": 6.999969482421875e-06,
      "grad_norm": 1.081994652748108,
      "learning_rate": 1e-05,
      "loss": 0.0662,
      "step": 7340
    },
    {
      "epoch": 6.999969482421875e-06,
      "model_forward_time": 5.630373239517212,
      "step": 7340
    },
    {
      "dynamics_loss_avg": 0.060542306303977965,
      "epoch": 6.999969482421875e-06,
      "step": 7340
    },
    {
      "action_loss_avg": 0.010795606160536408,
      "epoch": 6.999969482421875e-06,
      "step": 7340
    },
    {
      "epoch": 6.999969482421875e-06,
      "step": 7340,
      "training_step_time": 17.13645315170288
    },
    {
      "epoch": 7.0009231567382814e-06,
      "model_forward_time": 5.616441488265991,
      "step": 7341
    },
    {
      "epoch": 7.0009231567382814e-06,
      "step": 7341,
      "training_step_time": 17.12319326400757
    },
    {
      "epoch": 7.001876831054687e-06,
      "model_forward_time": 5.621251106262207,
      "step": 7342
    },
    {
      "epoch": 7.001876831054687e-06,
      "step": 7342,
      "training_step_time": 17.485734939575195
    },
    {
      "epoch": 7.002830505371094e-06,
      "model_forward_time": 5.61491322517395,
      "step": 7343
    },
    {
      "epoch": 7.002830505371094e-06,
      "step": 7343,
      "training_step_time": 17.089982271194458
    },
    {
      "epoch": 7.0037841796875e-06,
      "model_forward_time": 5.5900719165802,
      "step": 7344
    },
    {
      "epoch": 7.0037841796875e-06,
      "step": 7344,
      "training_step_time": 17.010680198669434
    },
    {
      "epoch": 7.004737854003906e-06,
      "model_forward_time": 5.574447870254517,
      "step": 7345
    },
    {
      "epoch": 7.004737854003906e-06,
      "step": 7345,
      "training_step_time": 17.014872074127197
    },
    {
      "epoch": 7.005691528320313e-06,
      "model_forward_time": 5.574493408203125,
      "step": 7346
    },
    {
      "epoch": 7.005691528320313e-06,
      "step": 7346,
      "training_step_time": 16.976792573928833
    },
    {
      "epoch": 7.006645202636719e-06,
      "model_forward_time": 5.575711488723755,
      "step": 7347
    },
    {
      "epoch": 7.006645202636719e-06,
      "step": 7347,
      "training_step_time": 16.97557020187378
    },
    {
      "epoch": 7.007598876953125e-06,
      "model_forward_time": 5.918606281280518,
      "step": 7348
    },
    {
      "epoch": 7.007598876953125e-06,
      "step": 7348,
      "training_step_time": 17.356155157089233
    },
    {
      "epoch": 7.0085525512695314e-06,
      "model_forward_time": 5.57516074180603,
      "step": 7349
    },
    {
      "epoch": 7.0085525512695314e-06,
      "step": 7349,
      "training_step_time": 16.99839949607849
    },
    {
      "epoch": 7.009506225585937e-06,
      "grad_norm": 1.5489226579666138,
      "learning_rate": 1e-05,
      "loss": 0.0812,
      "step": 7350
    },
    {
      "epoch": 7.009506225585937e-06,
      "model_forward_time": 5.5853307247161865,
      "step": 7350
    },
    {
      "dynamics_loss_avg": 0.07084157951176166,
      "epoch": 7.009506225585937e-06,
      "step": 7350
    },
    {
      "action_loss_avg": 0.014510917104780674,
      "epoch": 7.009506225585937e-06,
      "step": 7350
    },
    {
      "epoch": 7.009506225585937e-06,
      "step": 7350,
      "training_step_time": 17.03342318534851
    },
    {
      "epoch": 7.010459899902344e-06,
      "model_forward_time": 5.613284349441528,
      "step": 7351
    },
    {
      "epoch": 7.010459899902344e-06,
      "step": 7351,
      "training_step_time": 17.06909966468811
    },
    {
      "epoch": 7.01141357421875e-06,
      "model_forward_time": 5.60805606842041,
      "step": 7352
    },
    {
      "epoch": 7.01141357421875e-06,
      "step": 7352,
      "training_step_time": 17.09127163887024
    },
    {
      "epoch": 7.012367248535156e-06,
      "model_forward_time": 5.9868004322052,
      "step": 7353
    },
    {
      "epoch": 7.012367248535156e-06,
      "step": 7353,
      "training_step_time": 17.49390697479248
    },
    {
      "epoch": 7.013320922851563e-06,
      "model_forward_time": 5.614901781082153,
      "step": 7354
    },
    {
      "epoch": 7.013320922851563e-06,
      "step": 7354,
      "training_step_time": 17.13093614578247
    },
    {
      "epoch": 7.014274597167969e-06,
      "model_forward_time": 5.615166902542114,
      "step": 7355
    },
    {
      "epoch": 7.014274597167969e-06,
      "step": 7355,
      "training_step_time": 17.11873960494995
    },
    {
      "epoch": 7.015228271484375e-06,
      "model_forward_time": 5.608973503112793,
      "step": 7356
    },
    {
      "epoch": 7.015228271484375e-06,
      "step": 7356,
      "training_step_time": 17.10982084274292
    },
    {
      "epoch": 7.0161819458007814e-06,
      "model_forward_time": 5.624783039093018,
      "step": 7357
    },
    {
      "epoch": 7.0161819458007814e-06,
      "step": 7357,
      "training_step_time": 17.09480595588684
    },
    {
      "epoch": 7.017135620117187e-06,
      "model_forward_time": 5.615204334259033,
      "step": 7358
    },
    {
      "epoch": 7.017135620117187e-06,
      "step": 7358,
      "training_step_time": 17.097729444503784
    },
    {
      "epoch": 7.018089294433594e-06,
      "model_forward_time": 5.941102027893066,
      "step": 7359
    },
    {
      "epoch": 7.018089294433594e-06,
      "step": 7359,
      "training_step_time": 17.407413244247437
    },
    {
      "epoch": 7.01904296875e-06,
      "grad_norm": 1.0270122289657593,
      "learning_rate": 1e-05,
      "loss": 0.0859,
      "step": 7360
    },
    {
      "epoch": 7.01904296875e-06,
      "model_forward_time": 5.6001136302948,
      "step": 7360
    },
    {
      "dynamics_loss_avg": 0.07197732487693428,
      "epoch": 7.01904296875e-06,
      "step": 7360
    },
    {
      "action_loss_avg": 0.01184401661157608,
      "epoch": 7.01904296875e-06,
      "step": 7360
    },
    {
      "epoch": 7.01904296875e-06,
      "step": 7360,
      "training_step_time": 17.046677827835083
    },
    {
      "epoch": 7.019996643066406e-06,
      "model_forward_time": 5.5923240184783936,
      "step": 7361
    },
    {
      "epoch": 7.019996643066406e-06,
      "step": 7361,
      "training_step_time": 17.024795055389404
    },
    {
      "epoch": 7.020950317382813e-06,
      "model_forward_time": 5.595420837402344,
      "step": 7362
    },
    {
      "epoch": 7.020950317382813e-06,
      "step": 7362,
      "training_step_time": 17.058262825012207
    },
    {
      "epoch": 7.021903991699219e-06,
      "model_forward_time": 5.596951961517334,
      "step": 7363
    },
    {
      "epoch": 7.021903991699219e-06,
      "step": 7363,
      "training_step_time": 17.0481436252594
    },
    {
      "epoch": 7.022857666015625e-06,
      "model_forward_time": 5.950303792953491,
      "step": 7364
    },
    {
      "epoch": 7.022857666015625e-06,
      "step": 7364,
      "training_step_time": 17.40057134628296
    },
    {
      "epoch": 7.0238113403320315e-06,
      "model_forward_time": 5.5791401863098145,
      "step": 7365
    },
    {
      "epoch": 7.0238113403320315e-06,
      "step": 7365,
      "training_step_time": 16.990200757980347
    },
    {
      "epoch": 7.024765014648437e-06,
      "model_forward_time": 5.588147401809692,
      "step": 7366
    },
    {
      "epoch": 7.024765014648437e-06,
      "step": 7366,
      "training_step_time": 16.99254298210144
    },
    {
      "epoch": 7.025718688964844e-06,
      "model_forward_time": 5.578813314437866,
      "step": 7367
    },
    {
      "epoch": 7.025718688964844e-06,
      "step": 7367,
      "training_step_time": 17.00868010520935
    },
    {
      "epoch": 7.02667236328125e-06,
      "model_forward_time": 5.604609966278076,
      "step": 7368
    },
    {
      "epoch": 7.02667236328125e-06,
      "step": 7368,
      "training_step_time": 17.1043541431427
    },
    {
      "epoch": 7.027626037597656e-06,
      "model_forward_time": 5.624242067337036,
      "step": 7369
    },
    {
      "epoch": 7.027626037597656e-06,
      "step": 7369,
      "training_step_time": 17.136916875839233
    },
    {
      "epoch": 7.028579711914063e-06,
      "grad_norm": 0.5905443429946899,
      "learning_rate": 1e-05,
      "loss": 0.0662,
      "step": 7370
    },
    {
      "epoch": 7.028579711914063e-06,
      "model_forward_time": 5.962944269180298,
      "step": 7370
    },
    {
      "dynamics_loss_avg": 0.05320499986410141,
      "epoch": 7.028579711914063e-06,
      "step": 7370
    },
    {
      "action_loss_avg": 0.009130620351061224,
      "epoch": 7.028579711914063e-06,
      "step": 7370
    },
    {
      "epoch": 7.028579711914063e-06,
      "step": 7370,
      "training_step_time": 17.481962203979492
    },
    {
      "epoch": 7.029533386230469e-06,
      "model_forward_time": 5.6167097091674805,
      "step": 7371
    },
    {
      "epoch": 7.029533386230469e-06,
      "step": 7371,
      "training_step_time": 17.109236478805542
    },
    {
      "epoch": 7.030487060546875e-06,
      "model_forward_time": 5.6219260692596436,
      "step": 7372
    },
    {
      "epoch": 7.030487060546875e-06,
      "step": 7372,
      "training_step_time": 17.099287271499634
    },
    {
      "epoch": 7.0314407348632815e-06,
      "model_forward_time": 5.59409761428833,
      "step": 7373
    },
    {
      "epoch": 7.0314407348632815e-06,
      "step": 7373,
      "training_step_time": 17.003368854522705
    },
    {
      "epoch": 7.032394409179687e-06,
      "model_forward_time": 5.577022075653076,
      "step": 7374
    },
    {
      "epoch": 7.032394409179687e-06,
      "step": 7374,
      "training_step_time": 16.979517459869385
    },
    {
      "epoch": 7.033348083496094e-06,
      "model_forward_time": 5.939789772033691,
      "step": 7375
    },
    {
      "epoch": 7.033348083496094e-06,
      "step": 7375,
      "training_step_time": 17.381767988204956
    },
    {
      "epoch": 7.0343017578125e-06,
      "model_forward_time": 5.582365989685059,
      "step": 7376
    },
    {
      "epoch": 7.0343017578125e-06,
      "step": 7376,
      "training_step_time": 17.038445472717285
    },
    {
      "epoch": 7.035255432128906e-06,
      "model_forward_time": 5.5975165367126465,
      "step": 7377
    },
    {
      "epoch": 7.035255432128906e-06,
      "step": 7377,
      "training_step_time": 17.049687385559082
    },
    {
      "epoch": 7.036209106445313e-06,
      "model_forward_time": 5.59827184677124,
      "step": 7378
    },
    {
      "epoch": 7.036209106445313e-06,
      "step": 7378,
      "training_step_time": 17.054170846939087
    },
    {
      "epoch": 7.037162780761718e-06,
      "model_forward_time": 5.590812683105469,
      "step": 7379
    },
    {
      "epoch": 7.037162780761718e-06,
      "step": 7379,
      "training_step_time": 17.04311442375183
    },
    {
      "epoch": 7.038116455078125e-06,
      "grad_norm": 0.6413842439651489,
      "learning_rate": 1e-05,
      "loss": 0.0585,
      "step": 7380
    },
    {
      "epoch": 7.038116455078125e-06,
      "model_forward_time": 5.591650485992432,
      "step": 7380
    },
    {
      "dynamics_loss_avg": 0.05205904021859169,
      "epoch": 7.038116455078125e-06,
      "step": 7380
    },
    {
      "action_loss_avg": 0.009072322957217693,
      "epoch": 7.038116455078125e-06,
      "step": 7380
    },
    {
      "epoch": 7.038116455078125e-06,
      "step": 7380,
      "training_step_time": 17.027602672576904
    },
    {
      "epoch": 7.0390701293945315e-06,
      "model_forward_time": 5.942453145980835,
      "step": 7381
    },
    {
      "epoch": 7.0390701293945315e-06,
      "step": 7381,
      "training_step_time": 17.384174585342407
    },
    {
      "epoch": 7.040023803710937e-06,
      "model_forward_time": 5.598050355911255,
      "step": 7382
    },
    {
      "epoch": 7.040023803710937e-06,
      "step": 7382,
      "training_step_time": 17.05453658103943
    },
    {
      "epoch": 7.040977478027344e-06,
      "model_forward_time": 5.588843584060669,
      "step": 7383
    },
    {
      "epoch": 7.040977478027344e-06,
      "step": 7383,
      "training_step_time": 17.03345489501953
    },
    {
      "epoch": 7.04193115234375e-06,
      "model_forward_time": 5.603994607925415,
      "step": 7384
    },
    {
      "epoch": 7.04193115234375e-06,
      "step": 7384,
      "training_step_time": 17.07268714904785
    },
    {
      "epoch": 7.042884826660156e-06,
      "model_forward_time": 5.606274843215942,
      "step": 7385
    },
    {
      "epoch": 7.042884826660156e-06,
      "step": 7385,
      "training_step_time": 17.10091471672058
    },
    {
      "epoch": 7.043838500976563e-06,
      "model_forward_time": 5.980454206466675,
      "step": 7386
    },
    {
      "epoch": 7.043838500976563e-06,
      "step": 7386,
      "training_step_time": 17.484660387039185
    },
    {
      "epoch": 7.044792175292968e-06,
      "model_forward_time": 5.616173028945923,
      "step": 7387
    },
    {
      "epoch": 7.044792175292968e-06,
      "step": 7387,
      "training_step_time": 17.12280011177063
    },
    {
      "epoch": 7.045745849609375e-06,
      "model_forward_time": 5.615478038787842,
      "step": 7388
    },
    {
      "epoch": 7.045745849609375e-06,
      "step": 7388,
      "training_step_time": 17.07987880706787
    },
    {
      "epoch": 7.0466995239257815e-06,
      "model_forward_time": 5.596094369888306,
      "step": 7389
    },
    {
      "epoch": 7.0466995239257815e-06,
      "step": 7389,
      "training_step_time": 17.00455093383789
    },
    {
      "epoch": 7.047653198242187e-06,
      "grad_norm": 1.6297966241836548,
      "learning_rate": 1e-05,
      "loss": 0.0674,
      "step": 7390
    },
    {
      "epoch": 7.047653198242187e-06,
      "model_forward_time": 5.573340177536011,
      "step": 7390
    },
    {
      "dynamics_loss_avg": 0.0587672820314765,
      "epoch": 7.047653198242187e-06,
      "step": 7390
    },
    {
      "action_loss_avg": 0.009136007959023118,
      "epoch": 7.047653198242187e-06,
      "step": 7390
    },
    {
      "epoch": 7.047653198242187e-06,
      "step": 7390,
      "training_step_time": 17.01390027999878
    },
    {
      "epoch": 7.048606872558594e-06,
      "model_forward_time": 5.91788387298584,
      "step": 7391
    },
    {
      "epoch": 7.048606872558594e-06,
      "step": 7391,
      "training_step_time": 17.343087434768677
    },
    {
      "epoch": 7.049560546875e-06,
      "model_forward_time": 5.57801628112793,
      "step": 7392
    },
    {
      "epoch": 7.049560546875e-06,
      "step": 7392,
      "training_step_time": 16.992632627487183
    },
    {
      "epoch": 7.050514221191406e-06,
      "model_forward_time": 5.579166889190674,
      "step": 7393
    },
    {
      "epoch": 7.050514221191406e-06,
      "step": 7393,
      "training_step_time": 17.004331350326538
    },
    {
      "epoch": 7.051467895507813e-06,
      "model_forward_time": 5.585919618606567,
      "step": 7394
    },
    {
      "epoch": 7.051467895507813e-06,
      "step": 7394,
      "training_step_time": 17.041697025299072
    },
    {
      "epoch": 7.052421569824218e-06,
      "model_forward_time": 5.607147216796875,
      "step": 7395
    },
    {
      "epoch": 7.052421569824218e-06,
      "step": 7395,
      "training_step_time": 17.082573890686035
    },
    {
      "epoch": 7.053375244140625e-06,
      "model_forward_time": 5.6080992221832275,
      "step": 7396
    },
    {
      "epoch": 7.053375244140625e-06,
      "step": 7396,
      "training_step_time": 17.092153787612915
    },
    {
      "epoch": 7.0543289184570315e-06,
      "model_forward_time": 5.987527370452881,
      "step": 7397
    },
    {
      "epoch": 7.0543289184570315e-06,
      "step": 7397,
      "training_step_time": 17.47217893600464
    },
    {
      "epoch": 7.055282592773437e-06,
      "model_forward_time": 5.607992172241211,
      "step": 7398
    },
    {
      "epoch": 7.055282592773437e-06,
      "step": 7398,
      "training_step_time": 17.079014539718628
    },
    {
      "epoch": 7.056236267089844e-06,
      "model_forward_time": 5.597789287567139,
      "step": 7399
    },
    {
      "epoch": 7.056236267089844e-06,
      "step": 7399,
      "training_step_time": 17.025312662124634
    },
    {
      "epoch": 7.05718994140625e-06,
      "grad_norm": 1.6401841640472412,
      "learning_rate": 1e-05,
      "loss": 0.0834,
      "step": 7400
    },
    {
      "epoch": 7.05718994140625e-06,
      "model_forward_time": 5.58212947845459,
      "step": 7400
    },
    {
      "dynamics_loss_avg": 0.06621067523956299,
      "epoch": 7.05718994140625e-06,
      "step": 7400
    },
    {
      "action_loss_avg": 0.012165795871987939,
      "epoch": 7.05718994140625e-06,
      "step": 7400
    },
    {
      "epoch": 7.05718994140625e-06,
      "step": 7400,
      "training_step_time": 16.998958349227905
    },
    {
      "epoch": 7.058143615722656e-06,
      "model_forward_time": 5.590768814086914,
      "step": 7401
    },
    {
      "epoch": 7.058143615722656e-06,
      "step": 7401,
      "training_step_time": 17.027679443359375
    },
    {
      "epoch": 7.059097290039063e-06,
      "model_forward_time": 5.600165367126465,
      "step": 7402
    },
    {
      "epoch": 7.059097290039063e-06,
      "step": 7402,
      "training_step_time": 17.41417098045349
    },
    {
      "epoch": 7.060050964355468e-06,
      "model_forward_time": 5.596155166625977,
      "step": 7403
    },
    {
      "epoch": 7.060050964355468e-06,
      "step": 7403,
      "training_step_time": 17.07426428794861
    },
    {
      "epoch": 7.061004638671875e-06,
      "model_forward_time": 5.598021030426025,
      "step": 7404
    },
    {
      "epoch": 7.061004638671875e-06,
      "step": 7404,
      "training_step_time": 17.05186367034912
    },
    {
      "epoch": 7.0619583129882815e-06,
      "model_forward_time": 5.587267875671387,
      "step": 7405
    },
    {
      "epoch": 7.0619583129882815e-06,
      "step": 7405,
      "training_step_time": 17.009748935699463
    },
    {
      "epoch": 7.062911987304687e-06,
      "model_forward_time": 5.581758499145508,
      "step": 7406
    },
    {
      "epoch": 7.062911987304687e-06,
      "step": 7406,
      "training_step_time": 17.001381158828735
    },
    {
      "epoch": 7.063865661621094e-06,
      "model_forward_time": 5.598729610443115,
      "step": 7407
    },
    {
      "epoch": 7.063865661621094e-06,
      "step": 7407,
      "training_step_time": 17.018765926361084
    },
    {
      "epoch": 7.0648193359375e-06,
      "model_forward_time": 5.965080976486206,
      "step": 7408
    },
    {
      "epoch": 7.0648193359375e-06,
      "step": 7408,
      "training_step_time": 17.438615083694458
    },
    {
      "epoch": 7.065773010253906e-06,
      "model_forward_time": 5.635205268859863,
      "step": 7409
    },
    {
      "epoch": 7.065773010253906e-06,
      "step": 7409,
      "training_step_time": 17.094982862472534
    },
    {
      "epoch": 7.066726684570313e-06,
      "grad_norm": 1.6193956136703491,
      "learning_rate": 1e-05,
      "loss": 0.0736,
      "step": 7410
    },
    {
      "epoch": 7.066726684570313e-06,
      "model_forward_time": 5.6266443729400635,
      "step": 7410
    },
    {
      "dynamics_loss_avg": 0.060377926006913184,
      "epoch": 7.066726684570313e-06,
      "step": 7410
    },
    {
      "action_loss_avg": 0.01195737523958087,
      "epoch": 7.066726684570313e-06,
      "step": 7410
    },
    {
      "epoch": 7.066726684570313e-06,
      "step": 7410,
      "training_step_time": 17.127580165863037
    },
    {
      "epoch": 7.067680358886718e-06,
      "model_forward_time": 5.598377227783203,
      "step": 7411
    },
    {
      "epoch": 7.067680358886718e-06,
      "step": 7411,
      "training_step_time": 17.056474924087524
    },
    {
      "epoch": 7.068634033203125e-06,
      "model_forward_time": 5.582443952560425,
      "step": 7412
    },
    {
      "epoch": 7.068634033203125e-06,
      "step": 7412,
      "training_step_time": 16.98319673538208
    },
    {
      "epoch": 7.0695877075195315e-06,
      "model_forward_time": 5.5744757652282715,
      "step": 7413
    },
    {
      "epoch": 7.0695877075195315e-06,
      "step": 7413,
      "training_step_time": 17.377684593200684
    },
    {
      "epoch": 7.070541381835937e-06,
      "model_forward_time": 5.575159311294556,
      "step": 7414
    },
    {
      "epoch": 7.070541381835937e-06,
      "step": 7414,
      "training_step_time": 17.03818964958191
    },
    {
      "epoch": 7.071495056152344e-06,
      "model_forward_time": 5.602248430252075,
      "step": 7415
    },
    {
      "epoch": 7.071495056152344e-06,
      "step": 7415,
      "training_step_time": 17.090446949005127
    },
    {
      "epoch": 7.07244873046875e-06,
      "model_forward_time": 5.617671966552734,
      "step": 7416
    },
    {
      "epoch": 7.07244873046875e-06,
      "step": 7416,
      "training_step_time": 17.138093948364258
    },
    {
      "epoch": 7.073402404785156e-06,
      "model_forward_time": 5.618488311767578,
      "step": 7417
    },
    {
      "epoch": 7.073402404785156e-06,
      "step": 7417,
      "training_step_time": 17.087002515792847
    },
    {
      "epoch": 7.074356079101563e-06,
      "model_forward_time": 5.58305287361145,
      "step": 7418
    },
    {
      "epoch": 7.074356079101563e-06,
      "step": 7418,
      "training_step_time": 16.9938907623291
    },
    {
      "epoch": 7.075309753417968e-06,
      "model_forward_time": 5.94135046005249,
      "step": 7419
    },
    {
      "epoch": 7.075309753417968e-06,
      "step": 7419,
      "training_step_time": 17.36074137687683
    },
    {
      "epoch": 7.076263427734375e-06,
      "grad_norm": 1.503140926361084,
      "learning_rate": 1e-05,
      "loss": 0.0771,
      "step": 7420
    },
    {
      "epoch": 7.076263427734375e-06,
      "model_forward_time": 5.56231164932251,
      "step": 7420
    },
    {
      "dynamics_loss_avg": 0.07024164982140065,
      "epoch": 7.076263427734375e-06,
      "step": 7420
    },
    {
      "action_loss_avg": 0.011691262014210224,
      "epoch": 7.076263427734375e-06,
      "step": 7420
    },
    {
      "epoch": 7.076263427734375e-06,
      "step": 7420,
      "training_step_time": 16.98760223388672
    },
    {
      "epoch": 7.0772171020507815e-06,
      "model_forward_time": 5.574040651321411,
      "step": 7421
    },
    {
      "epoch": 7.0772171020507815e-06,
      "step": 7421,
      "training_step_time": 17.005990028381348
    },
    {
      "epoch": 7.078170776367187e-06,
      "model_forward_time": 5.562242269515991,
      "step": 7422
    },
    {
      "epoch": 7.078170776367187e-06,
      "step": 7422,
      "training_step_time": 16.975780248641968
    },
    {
      "epoch": 7.079124450683594e-06,
      "model_forward_time": 5.574362754821777,
      "step": 7423
    },
    {
      "epoch": 7.079124450683594e-06,
      "step": 7423,
      "training_step_time": 16.990281581878662
    },
    {
      "epoch": 7.080078125e-06,
      "model_forward_time": 5.9453535079956055,
      "step": 7424
    },
    {
      "epoch": 7.080078125e-06,
      "step": 7424,
      "training_step_time": 17.393081665039062
    },
    {
      "epoch": 7.081031799316406e-06,
      "model_forward_time": 5.584418535232544,
      "step": 7425
    },
    {
      "epoch": 7.081031799316406e-06,
      "step": 7425,
      "training_step_time": 17.044731616973877
    },
    {
      "epoch": 7.081985473632813e-06,
      "model_forward_time": 5.604316711425781,
      "step": 7426
    },
    {
      "epoch": 7.081985473632813e-06,
      "step": 7426,
      "training_step_time": 17.08847212791443
    },
    {
      "epoch": 7.082939147949218e-06,
      "model_forward_time": 5.5996599197387695,
      "step": 7427
    },
    {
      "epoch": 7.082939147949218e-06,
      "step": 7427,
      "training_step_time": 17.063103199005127
    },
    {
      "epoch": 7.083892822265625e-06,
      "model_forward_time": 5.5955281257629395,
      "step": 7428
    },
    {
      "epoch": 7.083892822265625e-06,
      "step": 7428,
      "training_step_time": 17.07968282699585
    },
    {
      "epoch": 7.0848464965820315e-06,
      "model_forward_time": 5.638387441635132,
      "step": 7429
    },
    {
      "epoch": 7.0848464965820315e-06,
      "step": 7429,
      "training_step_time": 17.133586168289185
    },
    {
      "epoch": 7.085800170898437e-06,
      "grad_norm": 0.6897432804107666,
      "learning_rate": 1e-05,
      "loss": 0.0725,
      "step": 7430
    },
    {
      "epoch": 7.085800170898437e-06,
      "model_forward_time": 5.972708463668823,
      "step": 7430
    },
    {
      "dynamics_loss_avg": 0.06182464547455311,
      "epoch": 7.085800170898437e-06,
      "step": 7430
    },
    {
      "action_loss_avg": 0.010497646289877594,
      "epoch": 7.085800170898437e-06,
      "step": 7430
    },
    {
      "epoch": 7.085800170898437e-06,
      "step": 7430,
      "training_step_time": 17.44873857498169
    },
    {
      "epoch": 7.086753845214844e-06,
      "model_forward_time": 5.589845418930054,
      "step": 7431
    },
    {
      "epoch": 7.086753845214844e-06,
      "step": 7431,
      "training_step_time": 17.01143455505371
    },
    {
      "epoch": 7.08770751953125e-06,
      "model_forward_time": 5.589640855789185,
      "step": 7432
    },
    {
      "epoch": 7.08770751953125e-06,
      "step": 7432,
      "training_step_time": 17.045471668243408
    },
    {
      "epoch": 7.088661193847656e-06,
      "model_forward_time": 5.599930286407471,
      "step": 7433
    },
    {
      "epoch": 7.088661193847656e-06,
      "step": 7433,
      "training_step_time": 17.00219702720642
    },
    {
      "epoch": 7.089614868164063e-06,
      "model_forward_time": 5.601343393325806,
      "step": 7434
    },
    {
      "epoch": 7.089614868164063e-06,
      "step": 7434,
      "training_step_time": 17.078529596328735
    },
    {
      "epoch": 7.090568542480468e-06,
      "model_forward_time": 5.611289978027344,
      "step": 7435
    },
    {
      "epoch": 7.090568542480468e-06,
      "step": 7435,
      "training_step_time": 17.462292671203613
    },
    {
      "epoch": 7.091522216796875e-06,
      "model_forward_time": 5.605042457580566,
      "step": 7436
    },
    {
      "epoch": 7.091522216796875e-06,
      "step": 7436,
      "training_step_time": 17.062031507492065
    },
    {
      "epoch": 7.0924758911132815e-06,
      "model_forward_time": 5.597214460372925,
      "step": 7437
    },
    {
      "epoch": 7.0924758911132815e-06,
      "step": 7437,
      "training_step_time": 17.044537782669067
    },
    {
      "epoch": 7.093429565429687e-06,
      "model_forward_time": 5.585094928741455,
      "step": 7438
    },
    {
      "epoch": 7.093429565429687e-06,
      "step": 7438,
      "training_step_time": 17.003557920455933
    },
    {
      "epoch": 7.094383239746094e-06,
      "model_forward_time": 5.576159477233887,
      "step": 7439
    },
    {
      "epoch": 7.094383239746094e-06,
      "step": 7439,
      "training_step_time": 16.996789693832397
    },
    {
      "epoch": 7.0953369140625e-06,
      "grad_norm": 0.9266824722290039,
      "learning_rate": 1e-05,
      "loss": 0.0726,
      "step": 7440
    },
    {
      "epoch": 7.0953369140625e-06,
      "model_forward_time": 5.583496570587158,
      "step": 7440
    },
    {
      "dynamics_loss_avg": 0.062491477467119694,
      "epoch": 7.0953369140625e-06,
      "step": 7440
    },
    {
      "action_loss_avg": 0.011152826435863971,
      "epoch": 7.0953369140625e-06,
      "step": 7440
    },
    {
      "epoch": 7.0953369140625e-06,
      "step": 7440,
      "training_step_time": 17.04246211051941
    },
    {
      "epoch": 7.096290588378906e-06,
      "model_forward_time": 5.973884582519531,
      "step": 7441
    },
    {
      "epoch": 7.096290588378906e-06,
      "step": 7441,
      "training_step_time": 17.480820178985596
    },
    {
      "epoch": 7.097244262695313e-06,
      "model_forward_time": 5.627586603164673,
      "step": 7442
    },
    {
      "epoch": 7.097244262695313e-06,
      "step": 7442,
      "training_step_time": 17.122002124786377
    },
    {
      "epoch": 7.098197937011718e-06,
      "model_forward_time": 5.598593235015869,
      "step": 7443
    },
    {
      "epoch": 7.098197937011718e-06,
      "step": 7443,
      "training_step_time": 17.06566286087036
    },
    {
      "epoch": 7.099151611328125e-06,
      "model_forward_time": 5.592599868774414,
      "step": 7444
    },
    {
      "epoch": 7.099151611328125e-06,
      "step": 7444,
      "training_step_time": 17.035913705825806
    },
    {
      "epoch": 7.1001052856445315e-06,
      "model_forward_time": 5.5799665451049805,
      "step": 7445
    },
    {
      "epoch": 7.1001052856445315e-06,
      "step": 7445,
      "training_step_time": 17.037816762924194
    },
    {
      "epoch": 7.101058959960937e-06,
      "model_forward_time": 5.61189341545105,
      "step": 7446
    },
    {
      "epoch": 7.101058959960937e-06,
      "step": 7446,
      "training_step_time": 17.451984643936157
    },
    {
      "epoch": 7.102012634277344e-06,
      "model_forward_time": 5.613220930099487,
      "step": 7447
    },
    {
      "epoch": 7.102012634277344e-06,
      "step": 7447,
      "training_step_time": 17.095531702041626
    },
    {
      "epoch": 7.10296630859375e-06,
      "model_forward_time": 5.588717222213745,
      "step": 7448
    },
    {
      "epoch": 7.10296630859375e-06,
      "step": 7448,
      "training_step_time": 17.060025691986084
    },
    {
      "epoch": 7.103919982910156e-06,
      "model_forward_time": 5.582508087158203,
      "step": 7449
    },
    {
      "epoch": 7.103919982910156e-06,
      "step": 7449,
      "training_step_time": 16.99632167816162
    },
    {
      "epoch": 7.104873657226563e-06,
      "grad_norm": 0.7996355295181274,
      "learning_rate": 1e-05,
      "loss": 0.0714,
      "step": 7450
    },
    {
      "epoch": 7.104873657226563e-06,
      "model_forward_time": 5.576805353164673,
      "step": 7450
    },
    {
      "dynamics_loss_avg": 0.05752787608653307,
      "epoch": 7.104873657226563e-06,
      "step": 7450
    },
    {
      "action_loss_avg": 0.009153356158640236,
      "epoch": 7.104873657226563e-06,
      "step": 7450
    },
    {
      "epoch": 7.104873657226563e-06,
      "step": 7450,
      "training_step_time": 17.003583431243896
    },
    {
      "epoch": 7.105827331542968e-06,
      "model_forward_time": 5.587587833404541,
      "step": 7451
    },
    {
      "epoch": 7.105827331542968e-06,
      "step": 7451,
      "training_step_time": 17.020578384399414
    },
    {
      "epoch": 7.106781005859375e-06,
      "model_forward_time": 5.970433235168457,
      "step": 7452
    },
    {
      "epoch": 7.106781005859375e-06,
      "step": 7452,
      "training_step_time": 17.468108892440796
    },
    {
      "epoch": 7.1077346801757815e-06,
      "model_forward_time": 5.618473291397095,
      "step": 7453
    },
    {
      "epoch": 7.1077346801757815e-06,
      "step": 7453,
      "training_step_time": 17.089583158493042
    },
    {
      "epoch": 7.108688354492187e-06,
      "model_forward_time": 5.5929248332977295,
      "step": 7454
    },
    {
      "epoch": 7.108688354492187e-06,
      "step": 7454,
      "training_step_time": 17.031776666641235
    },
    {
      "epoch": 7.109642028808594e-06,
      "model_forward_time": 5.573210716247559,
      "step": 7455
    },
    {
      "epoch": 7.109642028808594e-06,
      "step": 7455,
      "training_step_time": 17.014365673065186
    },
    {
      "epoch": 7.110595703125e-06,
      "model_forward_time": 5.5697901248931885,
      "step": 7456
    },
    {
      "epoch": 7.110595703125e-06,
      "step": 7456,
      "training_step_time": 16.98672127723694
    },
    {
      "epoch": 7.111549377441406e-06,
      "model_forward_time": 5.9330408573150635,
      "step": 7457
    },
    {
      "epoch": 7.111549377441406e-06,
      "step": 7457,
      "training_step_time": 17.34904456138611
    },
    {
      "epoch": 7.112503051757813e-06,
      "model_forward_time": 5.57835054397583,
      "step": 7458
    },
    {
      "epoch": 7.112503051757813e-06,
      "step": 7458,
      "training_step_time": 16.99099898338318
    },
    {
      "epoch": 7.113456726074218e-06,
      "model_forward_time": 5.5898261070251465,
      "step": 7459
    },
    {
      "epoch": 7.113456726074218e-06,
      "step": 7459,
      "training_step_time": 17.028124809265137
    },
    {
      "epoch": 7.114410400390625e-06,
      "grad_norm": 0.7092715501785278,
      "learning_rate": 1e-05,
      "loss": 0.0576,
      "step": 7460
    },
    {
      "epoch": 7.114410400390625e-06,
      "model_forward_time": 5.598689556121826,
      "step": 7460
    },
    {
      "dynamics_loss_avg": 0.05285308137536049,
      "epoch": 7.114410400390625e-06,
      "step": 7460
    },
    {
      "action_loss_avg": 0.008330461895093321,
      "epoch": 7.114410400390625e-06,
      "step": 7460
    },
    {
      "epoch": 7.114410400390625e-06,
      "step": 7460,
      "training_step_time": 17.054941415786743
    },
    {
      "epoch": 7.1153640747070315e-06,
      "model_forward_time": 5.593721866607666,
      "step": 7461
    },
    {
      "epoch": 7.1153640747070315e-06,
      "step": 7461,
      "training_step_time": 17.061708450317383
    },
    {
      "epoch": 7.116317749023437e-06,
      "model_forward_time": 5.596534967422485,
      "step": 7462
    },
    {
      "epoch": 7.116317749023437e-06,
      "step": 7462,
      "training_step_time": 17.04184317588806
    },
    {
      "epoch": 7.117271423339844e-06,
      "model_forward_time": 5.9471116065979,
      "step": 7463
    },
    {
      "epoch": 7.117271423339844e-06,
      "step": 7463,
      "training_step_time": 17.39055824279785
    },
    {
      "epoch": 7.11822509765625e-06,
      "model_forward_time": 5.594711542129517,
      "step": 7464
    },
    {
      "epoch": 7.11822509765625e-06,
      "step": 7464,
      "training_step_time": 17.047993183135986
    },
    {
      "epoch": 7.119178771972656e-06,
      "model_forward_time": 5.609837055206299,
      "step": 7465
    },
    {
      "epoch": 7.119178771972656e-06,
      "step": 7465,
      "training_step_time": 17.091285228729248
    },
    {
      "epoch": 7.120132446289063e-06,
      "model_forward_time": 5.610918760299683,
      "step": 7466
    },
    {
      "epoch": 7.120132446289063e-06,
      "step": 7466,
      "training_step_time": 17.091561794281006
    },
    {
      "epoch": 7.121086120605468e-06,
      "model_forward_time": 5.6171653270721436,
      "step": 7467
    },
    {
      "epoch": 7.121086120605468e-06,
      "step": 7467,
      "training_step_time": 17.113969564437866
    },
    {
      "epoch": 7.122039794921875e-06,
      "model_forward_time": 5.608539819717407,
      "step": 7468
    },
    {
      "epoch": 7.122039794921875e-06,
      "step": 7468,
      "training_step_time": 17.462299346923828
    },
    {
      "epoch": 7.1229934692382815e-06,
      "model_forward_time": 5.606252193450928,
      "step": 7469
    },
    {
      "epoch": 7.1229934692382815e-06,
      "step": 7469,
      "training_step_time": 17.087389707565308
    },
    {
      "epoch": 7.123947143554687e-06,
      "grad_norm": 1.1200430393218994,
      "learning_rate": 1e-05,
      "loss": 0.0767,
      "step": 7470
    },
    {
      "epoch": 7.123947143554687e-06,
      "model_forward_time": 5.603104591369629,
      "step": 7470
    },
    {
      "dynamics_loss_avg": 0.061761051416397095,
      "epoch": 7.123947143554687e-06,
      "step": 7470
    },
    {
      "action_loss_avg": 0.01165206260047853,
      "epoch": 7.123947143554687e-06,
      "step": 7470
    },
    {
      "epoch": 7.123947143554687e-06,
      "step": 7470,
      "training_step_time": 17.080909490585327
    },
    {
      "epoch": 7.124900817871094e-06,
      "model_forward_time": 5.598321199417114,
      "step": 7471
    },
    {
      "epoch": 7.124900817871094e-06,
      "step": 7471,
      "training_step_time": 17.07171607017517
    },
    {
      "epoch": 7.1258544921875e-06,
      "model_forward_time": 5.594899654388428,
      "step": 7472
    },
    {
      "epoch": 7.1258544921875e-06,
      "step": 7472,
      "training_step_time": 17.059009075164795
    },
    {
      "epoch": 7.126808166503906e-06,
      "model_forward_time": 5.596127033233643,
      "step": 7473
    },
    {
      "epoch": 7.126808166503906e-06,
      "step": 7473,
      "training_step_time": 17.072092294692993
    },
    {
      "epoch": 7.127761840820313e-06,
      "model_forward_time": 5.983253002166748,
      "step": 7474
    },
    {
      "epoch": 7.127761840820313e-06,
      "step": 7474,
      "training_step_time": 17.464677095413208
    },
    {
      "epoch": 7.128715515136718e-06,
      "model_forward_time": 5.6116392612457275,
      "step": 7475
    },
    {
      "epoch": 7.128715515136718e-06,
      "step": 7475,
      "training_step_time": 17.10801100730896
    },
    {
      "epoch": 7.129669189453125e-06,
      "model_forward_time": 5.630697250366211,
      "step": 7476
    },
    {
      "epoch": 7.129669189453125e-06,
      "step": 7476,
      "training_step_time": 17.149292707443237
    },
    {
      "epoch": 7.1306228637695315e-06,
      "model_forward_time": 5.624896049499512,
      "step": 7477
    },
    {
      "epoch": 7.1306228637695315e-06,
      "step": 7477,
      "training_step_time": 17.12245273590088
    },
    {
      "epoch": 7.131576538085937e-06,
      "model_forward_time": 5.606959104537964,
      "step": 7478
    },
    {
      "epoch": 7.131576538085937e-06,
      "step": 7478,
      "training_step_time": 17.06960415840149
    },
    {
      "epoch": 7.132530212402344e-06,
      "model_forward_time": 5.9444708824157715,
      "step": 7479
    },
    {
      "epoch": 7.132530212402344e-06,
      "step": 7479,
      "training_step_time": 17.401496410369873
    },
    {
      "epoch": 7.13348388671875e-06,
      "grad_norm": 1.2567652463912964,
      "learning_rate": 1e-05,
      "loss": 0.0714,
      "step": 7480
    },
    {
      "epoch": 7.13348388671875e-06,
      "model_forward_time": 5.60225772857666,
      "step": 7480
    },
    {
      "dynamics_loss_avg": 0.060826503485441205,
      "epoch": 7.13348388671875e-06,
      "step": 7480
    },
    {
      "action_loss_avg": 0.011142840213142335,
      "epoch": 7.13348388671875e-06,
      "step": 7480
    },
    {
      "epoch": 7.13348388671875e-06,
      "step": 7480,
      "training_step_time": 17.068318367004395
    },
    {
      "epoch": 7.134437561035156e-06,
      "model_forward_time": 5.599313497543335,
      "step": 7481
    },
    {
      "epoch": 7.134437561035156e-06,
      "step": 7481,
      "training_step_time": 17.074451208114624
    },
    {
      "epoch": 7.135391235351563e-06,
      "model_forward_time": 5.613422870635986,
      "step": 7482
    },
    {
      "epoch": 7.135391235351563e-06,
      "step": 7482,
      "training_step_time": 17.09597086906433
    },
    {
      "epoch": 7.136344909667968e-06,
      "model_forward_time": 5.6171815395355225,
      "step": 7483
    },
    {
      "epoch": 7.136344909667968e-06,
      "step": 7483,
      "training_step_time": 17.11717438697815
    },
    {
      "epoch": 7.137298583984375e-06,
      "model_forward_time": 5.6035239696502686,
      "step": 7484
    },
    {
      "epoch": 7.137298583984375e-06,
      "step": 7484,
      "training_step_time": 17.08765482902527
    },
    {
      "epoch": 7.1382522583007816e-06,
      "model_forward_time": 5.972970724105835,
      "step": 7485
    },
    {
      "epoch": 7.1382522583007816e-06,
      "step": 7485,
      "training_step_time": 17.471580505371094
    },
    {
      "epoch": 7.139205932617187e-06,
      "model_forward_time": 5.620640516281128,
      "step": 7486
    },
    {
      "epoch": 7.139205932617187e-06,
      "step": 7486,
      "training_step_time": 17.12644910812378
    },
    {
      "epoch": 7.140159606933594e-06,
      "model_forward_time": 5.618416786193848,
      "step": 7487
    },
    {
      "epoch": 7.140159606933594e-06,
      "step": 7487,
      "training_step_time": 17.11730122566223
    },
    {
      "epoch": 7.14111328125e-06,
      "model_forward_time": 5.621113538742065,
      "step": 7488
    },
    {
      "epoch": 7.14111328125e-06,
      "step": 7488,
      "training_step_time": 17.101436614990234
    },
    {
      "epoch": 7.142066955566406e-06,
      "model_forward_time": 5.600708723068237,
      "step": 7489
    },
    {
      "epoch": 7.142066955566406e-06,
      "step": 7489,
      "training_step_time": 17.034910678863525
    },
    {
      "epoch": 7.143020629882813e-06,
      "grad_norm": 0.5857908725738525,
      "learning_rate": 1e-05,
      "loss": 0.0619,
      "step": 7490
    },
    {
      "epoch": 7.143020629882813e-06,
      "model_forward_time": 5.923595905303955,
      "step": 7490
    },
    {
      "dynamics_loss_avg": 0.0555684395134449,
      "epoch": 7.143020629882813e-06,
      "step": 7490
    },
    {
      "action_loss_avg": 0.008848539879545569,
      "epoch": 7.143020629882813e-06,
      "step": 7490
    },
    {
      "epoch": 7.143020629882813e-06,
      "step": 7490,
      "training_step_time": 17.343335390090942
    },
    {
      "epoch": 7.1439743041992184e-06,
      "model_forward_time": 5.593063831329346,
      "step": 7491
    },
    {
      "epoch": 7.1439743041992184e-06,
      "step": 7491,
      "training_step_time": 17.030964374542236
    },
    {
      "epoch": 7.144927978515625e-06,
      "model_forward_time": 5.592989683151245,
      "step": 7492
    },
    {
      "epoch": 7.144927978515625e-06,
      "step": 7492,
      "training_step_time": 17.043004512786865
    },
    {
      "epoch": 7.1458816528320316e-06,
      "model_forward_time": 5.614423036575317,
      "step": 7493
    },
    {
      "epoch": 7.1458816528320316e-06,
      "step": 7493,
      "training_step_time": 17.05550241470337
    },
    {
      "epoch": 7.146835327148437e-06,
      "model_forward_time": 5.607038736343384,
      "step": 7494
    },
    {
      "epoch": 7.146835327148437e-06,
      "step": 7494,
      "training_step_time": 17.10444951057434
    },
    {
      "epoch": 7.147789001464844e-06,
      "model_forward_time": 5.61628270149231,
      "step": 7495
    },
    {
      "epoch": 7.147789001464844e-06,
      "step": 7495,
      "training_step_time": 17.48980402946472
    },
    {
      "epoch": 7.14874267578125e-06,
      "model_forward_time": 5.608306407928467,
      "step": 7496
    },
    {
      "epoch": 7.14874267578125e-06,
      "step": 7496,
      "training_step_time": 17.089404582977295
    },
    {
      "epoch": 7.149696350097656e-06,
      "model_forward_time": 5.606811046600342,
      "step": 7497
    },
    {
      "epoch": 7.149696350097656e-06,
      "step": 7497,
      "training_step_time": 17.05449604988098
    },
    {
      "epoch": 7.150650024414063e-06,
      "model_forward_time": 5.600185871124268,
      "step": 7498
    },
    {
      "epoch": 7.150650024414063e-06,
      "step": 7498,
      "training_step_time": 17.078693628311157
    },
    {
      "epoch": 7.1516036987304684e-06,
      "model_forward_time": 5.604638338088989,
      "step": 7499
    },
    {
      "epoch": 7.1516036987304684e-06,
      "step": 7499,
      "training_step_time": 17.081433057785034
    },
    {
      "epoch": 7.152557373046875e-06,
      "grad_norm": 1.0805939435958862,
      "learning_rate": 1e-05,
      "loss": 0.0755,
      "step": 7500
    },
    {
      "epoch": 7.152557373046875e-06,
      "model_forward_time": 5.601427793502808,
      "step": 7500
    },
    {
      "dynamics_loss_avg": 0.06453237496316433,
      "epoch": 7.152557373046875e-06,
      "step": 7500
    },
    {
      "action_loss_avg": 0.011951151816174389,
      "epoch": 7.152557373046875e-06,
      "step": 7500
    },
    {
      "epoch": 7.152557373046875e-06,
      "step": 7500,
      "training_step_time": 17.046038389205933
    },
    {
      "epoch": 7.1535110473632816e-06,
      "model_forward_time": 5.947328567504883,
      "step": 7501
    },
    {
      "epoch": 7.1535110473632816e-06,
      "step": 7501,
      "training_step_time": 17.388192415237427
    },
    {
      "epoch": 7.154464721679687e-06,
      "model_forward_time": 5.580620527267456,
      "step": 7502
    },
    {
      "epoch": 7.154464721679687e-06,
      "step": 7502,
      "training_step_time": 17.01145029067993
    },
    {
      "epoch": 7.155418395996094e-06,
      "model_forward_time": 5.582671403884888,
      "step": 7503
    },
    {
      "epoch": 7.155418395996094e-06,
      "step": 7503,
      "training_step_time": 17.057355403900146
    },
    {
      "epoch": 7.1563720703125e-06,
      "model_forward_time": 5.611918926239014,
      "step": 7504
    },
    {
      "epoch": 7.1563720703125e-06,
      "step": 7504,
      "training_step_time": 17.107682943344116
    },
    {
      "epoch": 7.157325744628906e-06,
      "model_forward_time": 5.62145209312439,
      "step": 7505
    },
    {
      "epoch": 7.157325744628906e-06,
      "step": 7505,
      "training_step_time": 17.118134021759033
    },
    {
      "epoch": 7.158279418945313e-06,
      "model_forward_time": 5.615026473999023,
      "step": 7506
    },
    {
      "epoch": 7.158279418945313e-06,
      "step": 7506,
      "training_step_time": 17.424989461898804
    },
    {
      "epoch": 7.1592330932617184e-06,
      "model_forward_time": 5.58786940574646,
      "step": 7507
    },
    {
      "epoch": 7.1592330932617184e-06,
      "step": 7507,
      "training_step_time": 16.99801516532898
    },
    {
      "epoch": 7.160186767578125e-06,
      "model_forward_time": 5.577646493911743,
      "step": 7508
    },
    {
      "epoch": 7.160186767578125e-06,
      "step": 7508,
      "training_step_time": 17.01553440093994
    },
    {
      "epoch": 7.1611404418945316e-06,
      "model_forward_time": 5.567846775054932,
      "step": 7509
    },
    {
      "epoch": 7.1611404418945316e-06,
      "step": 7509,
      "training_step_time": 16.98379945755005
    },
    {
      "epoch": 7.162094116210937e-06,
      "grad_norm": 0.21461886167526245,
      "learning_rate": 1e-05,
      "loss": 0.078,
      "step": 7510
    },
    {
      "epoch": 7.162094116210937e-06,
      "model_forward_time": 5.572009325027466,
      "step": 7510
    },
    {
      "dynamics_loss_avg": 0.06432656273245811,
      "epoch": 7.162094116210937e-06,
      "step": 7510
    },
    {
      "action_loss_avg": 0.01024399334564805,
      "epoch": 7.162094116210937e-06,
      "step": 7510
    },
    {
      "epoch": 7.162094116210937e-06,
      "step": 7510,
      "training_step_time": 16.99016785621643
    },
    {
      "epoch": 7.163047790527344e-06,
      "model_forward_time": 5.572264194488525,
      "step": 7511
    },
    {
      "epoch": 7.163047790527344e-06,
      "step": 7511,
      "training_step_time": 16.986146926879883
    },
    {
      "epoch": 7.16400146484375e-06,
      "model_forward_time": 5.920952558517456,
      "step": 7512
    },
    {
      "epoch": 7.16400146484375e-06,
      "step": 7512,
      "training_step_time": 17.35880446434021
    },
    {
      "epoch": 7.164955139160156e-06,
      "model_forward_time": 5.575134038925171,
      "step": 7513
    },
    {
      "epoch": 7.164955139160156e-06,
      "step": 7513,
      "training_step_time": 16.995915412902832
    },
    {
      "epoch": 7.165908813476563e-06,
      "model_forward_time": 5.583355903625488,
      "step": 7514
    },
    {
      "epoch": 7.165908813476563e-06,
      "step": 7514,
      "training_step_time": 17.017696380615234
    },
    {
      "epoch": 7.1668624877929684e-06,
      "model_forward_time": 5.592543125152588,
      "step": 7515
    },
    {
      "epoch": 7.1668624877929684e-06,
      "step": 7515,
      "training_step_time": 17.061286211013794
    },
    {
      "epoch": 7.167816162109375e-06,
      "model_forward_time": 5.611619234085083,
      "step": 7516
    },
    {
      "epoch": 7.167816162109375e-06,
      "step": 7516,
      "training_step_time": 17.082830905914307
    },
    {
      "epoch": 7.168769836425782e-06,
      "model_forward_time": 5.961878538131714,
      "step": 7517
    },
    {
      "epoch": 7.168769836425782e-06,
      "step": 7517,
      "training_step_time": 17.40921449661255
    },
    {
      "epoch": 7.169723510742187e-06,
      "model_forward_time": 5.5850372314453125,
      "step": 7518
    },
    {
      "epoch": 7.169723510742187e-06,
      "step": 7518,
      "training_step_time": 17.025099754333496
    },
    {
      "epoch": 7.170677185058594e-06,
      "model_forward_time": 5.590079307556152,
      "step": 7519
    },
    {
      "epoch": 7.170677185058594e-06,
      "step": 7519,
      "training_step_time": 17.0022189617157
    },
    {
      "epoch": 7.171630859375e-06,
      "grad_norm": 0.37528854608535767,
      "learning_rate": 1e-05,
      "loss": 0.0716,
      "step": 7520
    },
    {
      "epoch": 7.171630859375e-06,
      "model_forward_time": 5.586512088775635,
      "step": 7520
    },
    {
      "dynamics_loss_avg": 0.06416821535676717,
      "epoch": 7.171630859375e-06,
      "step": 7520
    },
    {
      "action_loss_avg": 0.011757074249908328,
      "epoch": 7.171630859375e-06,
      "step": 7520
    },
    {
      "epoch": 7.171630859375e-06,
      "step": 7520,
      "training_step_time": 17.046502828598022
    },
    {
      "epoch": 7.172584533691406e-06,
      "model_forward_time": 5.619109153747559,
      "step": 7521
    },
    {
      "epoch": 7.172584533691406e-06,
      "step": 7521,
      "training_step_time": 17.122005939483643
    },
    {
      "epoch": 7.173538208007813e-06,
      "model_forward_time": 5.622036933898926,
      "step": 7522
    },
    {
      "epoch": 7.173538208007813e-06,
      "step": 7522,
      "training_step_time": 17.456266164779663
    },
    {
      "epoch": 7.1744918823242185e-06,
      "model_forward_time": 5.598757266998291,
      "step": 7523
    },
    {
      "epoch": 7.1744918823242185e-06,
      "step": 7523,
      "training_step_time": 17.06508755683899
    },
    {
      "epoch": 7.175445556640625e-06,
      "model_forward_time": 5.604923963546753,
      "step": 7524
    },
    {
      "epoch": 7.175445556640625e-06,
      "step": 7524,
      "training_step_time": 17.065425395965576
    },
    {
      "epoch": 7.176399230957032e-06,
      "model_forward_time": 5.598066806793213,
      "step": 7525
    },
    {
      "epoch": 7.176399230957032e-06,
      "step": 7525,
      "training_step_time": 17.057978868484497
    },
    {
      "epoch": 7.177352905273437e-06,
      "model_forward_time": 5.593677282333374,
      "step": 7526
    },
    {
      "epoch": 7.177352905273437e-06,
      "step": 7526,
      "training_step_time": 17.034542322158813
    },
    {
      "epoch": 7.178306579589844e-06,
      "model_forward_time": 5.577831506729126,
      "step": 7527
    },
    {
      "epoch": 7.178306579589844e-06,
      "step": 7527,
      "training_step_time": 17.02431869506836
    },
    {
      "epoch": 7.17926025390625e-06,
      "model_forward_time": 5.9498450756073,
      "step": 7528
    },
    {
      "epoch": 7.17926025390625e-06,
      "step": 7528,
      "training_step_time": 17.425172328948975
    },
    {
      "epoch": 7.180213928222656e-06,
      "model_forward_time": 5.60249400138855,
      "step": 7529
    },
    {
      "epoch": 7.180213928222656e-06,
      "step": 7529,
      "training_step_time": 17.097270727157593
    },
    {
      "epoch": 7.181167602539063e-06,
      "grad_norm": 1.0056859254837036,
      "learning_rate": 1e-05,
      "loss": 0.0831,
      "step": 7530
    },
    {
      "epoch": 7.181167602539063e-06,
      "model_forward_time": 5.618341684341431,
      "step": 7530
    },
    {
      "dynamics_loss_avg": 0.06688165664672852,
      "epoch": 7.181167602539063e-06,
      "step": 7530
    },
    {
      "action_loss_avg": 0.012823436548933387,
      "epoch": 7.181167602539063e-06,
      "step": 7530
    },
    {
      "epoch": 7.181167602539063e-06,
      "step": 7530,
      "training_step_time": 17.09002685546875
    },
    {
      "epoch": 7.1821212768554685e-06,
      "model_forward_time": 5.587486267089844,
      "step": 7531
    },
    {
      "epoch": 7.1821212768554685e-06,
      "step": 7531,
      "training_step_time": 17.019967079162598
    },
    {
      "epoch": 7.183074951171875e-06,
      "model_forward_time": 5.575607538223267,
      "step": 7532
    },
    {
      "epoch": 7.183074951171875e-06,
      "step": 7532,
      "training_step_time": 16.98543691635132
    },
    {
      "epoch": 7.184028625488282e-06,
      "model_forward_time": 5.9323766231536865,
      "step": 7533
    },
    {
      "epoch": 7.184028625488282e-06,
      "step": 7533,
      "training_step_time": 17.33819341659546
    },
    {
      "epoch": 7.184982299804687e-06,
      "model_forward_time": 5.585617303848267,
      "step": 7534
    },
    {
      "epoch": 7.184982299804687e-06,
      "step": 7534,
      "training_step_time": 17.011996507644653
    },
    {
      "epoch": 7.185935974121094e-06,
      "model_forward_time": 5.589146614074707,
      "step": 7535
    },
    {
      "epoch": 7.185935974121094e-06,
      "step": 7535,
      "training_step_time": 17.041141271591187
    },
    {
      "epoch": 7.1868896484375e-06,
      "model_forward_time": 5.597872018814087,
      "step": 7536
    },
    {
      "epoch": 7.1868896484375e-06,
      "step": 7536,
      "training_step_time": 17.079575538635254
    },
    {
      "epoch": 7.187843322753906e-06,
      "model_forward_time": 5.596873998641968,
      "step": 7537
    },
    {
      "epoch": 7.187843322753906e-06,
      "step": 7537,
      "training_step_time": 17.05024814605713
    },
    {
      "epoch": 7.188796997070313e-06,
      "model_forward_time": 5.592510938644409,
      "step": 7538
    },
    {
      "epoch": 7.188796997070313e-06,
      "step": 7538,
      "training_step_time": 17.038589477539062
    },
    {
      "epoch": 7.1897506713867185e-06,
      "model_forward_time": 5.963643789291382,
      "step": 7539
    },
    {
      "epoch": 7.1897506713867185e-06,
      "step": 7539,
      "training_step_time": 17.42601227760315
    },
    {
      "epoch": 7.190704345703125e-06,
      "grad_norm": 1.599263072013855,
      "learning_rate": 1e-05,
      "loss": 0.0705,
      "step": 7540
    },
    {
      "epoch": 7.190704345703125e-06,
      "model_forward_time": 5.609813213348389,
      "step": 7540
    },
    {
      "dynamics_loss_avg": 0.05607538241893053,
      "epoch": 7.190704345703125e-06,
      "step": 7540
    },
    {
      "action_loss_avg": 0.011971315694972872,
      "epoch": 7.190704345703125e-06,
      "step": 7540
    },
    {
      "epoch": 7.190704345703125e-06,
      "step": 7540,
      "training_step_time": 17.10341477394104
    },
    {
      "epoch": 7.191658020019532e-06,
      "model_forward_time": 5.603549003601074,
      "step": 7541
    },
    {
      "epoch": 7.191658020019532e-06,
      "step": 7541,
      "training_step_time": 17.08873200416565
    },
    {
      "epoch": 7.192611694335937e-06,
      "model_forward_time": 5.600958585739136,
      "step": 7542
    },
    {
      "epoch": 7.192611694335937e-06,
      "step": 7542,
      "training_step_time": 17.052457571029663
    },
    {
      "epoch": 7.193565368652344e-06,
      "model_forward_time": 5.618743419647217,
      "step": 7543
    },
    {
      "epoch": 7.193565368652344e-06,
      "step": 7543,
      "training_step_time": 17.110591650009155
    },
    {
      "epoch": 7.19451904296875e-06,
      "model_forward_time": 5.949819326400757,
      "step": 7544
    },
    {
      "epoch": 7.19451904296875e-06,
      "step": 7544,
      "training_step_time": 17.405075311660767
    },
    {
      "epoch": 7.195472717285156e-06,
      "model_forward_time": 5.596315622329712,
      "step": 7545
    },
    {
      "epoch": 7.195472717285156e-06,
      "step": 7545,
      "training_step_time": 17.032862186431885
    },
    {
      "epoch": 7.196426391601563e-06,
      "model_forward_time": 5.595654010772705,
      "step": 7546
    },
    {
      "epoch": 7.196426391601563e-06,
      "step": 7546,
      "training_step_time": 17.034116983413696
    },
    {
      "epoch": 7.1973800659179685e-06,
      "model_forward_time": 5.592219591140747,
      "step": 7547
    },
    {
      "epoch": 7.1973800659179685e-06,
      "step": 7547,
      "training_step_time": 17.04971981048584
    },
    {
      "epoch": 7.198333740234375e-06,
      "model_forward_time": 5.597002744674683,
      "step": 7548
    },
    {
      "epoch": 7.198333740234375e-06,
      "step": 7548,
      "training_step_time": 17.057055234909058
    },
    {
      "epoch": 7.199287414550782e-06,
      "model_forward_time": 5.597605228424072,
      "step": 7549
    },
    {
      "epoch": 7.199287414550782e-06,
      "step": 7549,
      "training_step_time": 17.0682590007782
    },
    {
      "epoch": 7.200241088867187e-06,
      "grad_norm": 1.5056524276733398,
      "learning_rate": 1e-05,
      "loss": 0.064,
      "step": 7550
    },
    {
      "epoch": 7.200241088867187e-06,
      "model_forward_time": 6.006190061569214,
      "step": 7550
    },
    {
      "dynamics_loss_avg": 0.060571573115885256,
      "epoch": 7.200241088867187e-06,
      "step": 7550
    },
    {
      "action_loss_avg": 0.009197250613942743,
      "epoch": 7.200241088867187e-06,
      "step": 7550
    },
    {
      "epoch": 7.200241088867187e-06,
      "step": 7550,
      "training_step_time": 17.489290237426758
    },
    {
      "epoch": 7.201194763183594e-06,
      "model_forward_time": 5.615636110305786,
      "step": 7551
    },
    {
      "epoch": 7.201194763183594e-06,
      "step": 7551,
      "training_step_time": 17.120421171188354
    },
    {
      "epoch": 7.2021484375e-06,
      "model_forward_time": 5.645192623138428,
      "step": 7552
    },
    {
      "epoch": 7.2021484375e-06,
      "step": 7552,
      "training_step_time": 17.12915062904358
    },
    {
      "epoch": 7.203102111816406e-06,
      "model_forward_time": 5.596479177474976,
      "step": 7553
    },
    {
      "epoch": 7.203102111816406e-06,
      "step": 7553,
      "training_step_time": 17.019861698150635
    },
    {
      "epoch": 7.204055786132813e-06,
      "model_forward_time": 5.583148241043091,
      "step": 7554
    },
    {
      "epoch": 7.204055786132813e-06,
      "step": 7554,
      "training_step_time": 17.02373194694519
    },
    {
      "epoch": 7.2050094604492185e-06,
      "model_forward_time": 5.960026502609253,
      "step": 7555
    },
    {
      "epoch": 7.2050094604492185e-06,
      "step": 7555,
      "training_step_time": 17.410333156585693
    },
    {
      "epoch": 7.205963134765625e-06,
      "model_forward_time": 5.611093044281006,
      "step": 7556
    },
    {
      "epoch": 7.205963134765625e-06,
      "step": 7556,
      "training_step_time": 17.10867142677307
    },
    {
      "epoch": 7.206916809082032e-06,
      "model_forward_time": 4.291654109954834,
      "step": 7557
    },
    {
      "epoch": 7.206916809082032e-06,
      "step": 7557,
      "training_step_time": 12.844036102294922
    },
    {
      "epoch": 7.207870483398437e-06,
      "model_forward_time": 5.622830629348755,
      "step": 7558
    },
    {
      "epoch": 7.207870483398437e-06,
      "step": 7558,
      "training_step_time": 17.119903802871704
    },
    {
      "epoch": 7.208824157714844e-06,
      "model_forward_time": 5.620349884033203,
      "step": 7559
    },
    {
      "epoch": 7.208824157714844e-06,
      "step": 7559,
      "training_step_time": 17.084273099899292
    },
    {
      "epoch": 7.20977783203125e-06,
      "grad_norm": 1.1161823272705078,
      "learning_rate": 1e-05,
      "loss": 0.0724,
      "step": 7560
    },
    {
      "epoch": 7.20977783203125e-06,
      "model_forward_time": 5.590125560760498,
      "step": 7560
    },
    {
      "dynamics_loss_avg": 0.06101230792701244,
      "epoch": 7.20977783203125e-06,
      "step": 7560
    },
    {
      "action_loss_avg": 0.010985508048906922,
      "epoch": 7.20977783203125e-06,
      "step": 7560
    },
    {
      "epoch": 7.20977783203125e-06,
      "step": 7560,
      "training_step_time": 16.997187852859497
    },
    {
      "epoch": 7.210731506347656e-06,
      "model_forward_time": 5.961291790008545,
      "step": 7561
    },
    {
      "epoch": 7.210731506347656e-06,
      "step": 7561,
      "training_step_time": 17.40630006790161
    },
    {
      "epoch": 7.211685180664063e-06,
      "model_forward_time": 5.583087682723999,
      "step": 7562
    },
    {
      "epoch": 7.211685180664063e-06,
      "step": 7562,
      "training_step_time": 17.027172088623047
    },
    {
      "epoch": 7.2126388549804685e-06,
      "model_forward_time": 5.572284936904907,
      "step": 7563
    },
    {
      "epoch": 7.2126388549804685e-06,
      "step": 7563,
      "training_step_time": 17.001800775527954
    },
    {
      "epoch": 7.213592529296875e-06,
      "model_forward_time": 5.581587314605713,
      "step": 7564
    },
    {
      "epoch": 7.213592529296875e-06,
      "step": 7564,
      "training_step_time": 17.001941442489624
    },
    {
      "epoch": 7.214546203613282e-06,
      "model_forward_time": 5.596585035324097,
      "step": 7565
    },
    {
      "epoch": 7.214546203613282e-06,
      "step": 7565,
      "training_step_time": 17.02824592590332
    },
    {
      "epoch": 7.215499877929687e-06,
      "model_forward_time": 5.932307720184326,
      "step": 7566
    },
    {
      "epoch": 7.215499877929687e-06,
      "step": 7566,
      "training_step_time": 17.395575761795044
    },
    {
      "epoch": 7.216453552246094e-06,
      "model_forward_time": 5.605968952178955,
      "step": 7567
    },
    {
      "epoch": 7.216453552246094e-06,
      "step": 7567,
      "training_step_time": 17.09355664253235
    },
    {
      "epoch": 7.2174072265625e-06,
      "model_forward_time": 5.631196022033691,
      "step": 7568
    },
    {
      "epoch": 7.2174072265625e-06,
      "step": 7568,
      "training_step_time": 17.16620182991028
    },
    {
      "epoch": 7.218360900878906e-06,
      "model_forward_time": 5.622392892837524,
      "step": 7569
    },
    {
      "epoch": 7.218360900878906e-06,
      "step": 7569,
      "training_step_time": 17.092430353164673
    },
    {
      "epoch": 7.219314575195313e-06,
      "grad_norm": 1.6260796785354614,
      "learning_rate": 1e-05,
      "loss": 0.0758,
      "step": 7570
    },
    {
      "epoch": 7.219314575195313e-06,
      "model_forward_time": 5.585683107376099,
      "step": 7570
    },
    {
      "dynamics_loss_avg": 0.0586796659976244,
      "epoch": 7.219314575195313e-06,
      "step": 7570
    },
    {
      "action_loss_avg": 0.01105928337201476,
      "epoch": 7.219314575195313e-06,
      "step": 7570
    },
    {
      "epoch": 7.219314575195313e-06,
      "step": 7570,
      "training_step_time": 16.9958655834198
    },
    {
      "epoch": 7.2202682495117185e-06,
      "model_forward_time": 5.956105947494507,
      "step": 7571
    },
    {
      "epoch": 7.2202682495117185e-06,
      "step": 7571,
      "training_step_time": 17.36500096321106
    },
    {
      "epoch": 7.221221923828125e-06,
      "model_forward_time": 5.567285060882568,
      "step": 7572
    },
    {
      "epoch": 7.221221923828125e-06,
      "step": 7572,
      "training_step_time": 16.97459053993225
    },
    {
      "epoch": 7.222175598144532e-06,
      "model_forward_time": 5.56688928604126,
      "step": 7573
    },
    {
      "epoch": 7.222175598144532e-06,
      "step": 7573,
      "training_step_time": 17.000293731689453
    },
    {
      "epoch": 7.223129272460937e-06,
      "model_forward_time": 5.558651685714722,
      "step": 7574
    },
    {
      "epoch": 7.223129272460937e-06,
      "step": 7574,
      "training_step_time": 16.979653120040894
    },
    {
      "epoch": 7.224082946777344e-06,
      "model_forward_time": 5.582580804824829,
      "step": 7575
    },
    {
      "epoch": 7.224082946777344e-06,
      "step": 7575,
      "training_step_time": 17.011852741241455
    },
    {
      "epoch": 7.22503662109375e-06,
      "model_forward_time": 5.593170166015625,
      "step": 7576
    },
    {
      "epoch": 7.22503662109375e-06,
      "step": 7576,
      "training_step_time": 17.049843072891235
    },
    {
      "epoch": 7.225990295410156e-06,
      "model_forward_time": 5.957799434661865,
      "step": 7577
    },
    {
      "epoch": 7.225990295410156e-06,
      "step": 7577,
      "training_step_time": 17.411629915237427
    },
    {
      "epoch": 7.226943969726563e-06,
      "model_forward_time": 5.60834813117981,
      "step": 7578
    },
    {
      "epoch": 7.226943969726563e-06,
      "step": 7578,
      "training_step_time": 17.099836111068726
    },
    {
      "epoch": 7.2278976440429685e-06,
      "model_forward_time": 5.600233316421509,
      "step": 7579
    },
    {
      "epoch": 7.2278976440429685e-06,
      "step": 7579,
      "training_step_time": 17.053362369537354
    },
    {
      "epoch": 7.228851318359375e-06,
      "grad_norm": 0.5383172631263733,
      "learning_rate": 1e-05,
      "loss": 0.0681,
      "step": 7580
    },
    {
      "epoch": 7.228851318359375e-06,
      "model_forward_time": 5.595983982086182,
      "step": 7580
    },
    {
      "dynamics_loss_avg": 0.06146863140165806,
      "epoch": 7.228851318359375e-06,
      "step": 7580
    },
    {
      "action_loss_avg": 0.00883217379450798,
      "epoch": 7.228851318359375e-06,
      "step": 7580
    },
    {
      "epoch": 7.228851318359375e-06,
      "step": 7580,
      "training_step_time": 17.048957586288452
    },
    {
      "epoch": 7.229804992675782e-06,
      "model_forward_time": 5.587738513946533,
      "step": 7581
    },
    {
      "epoch": 7.229804992675782e-06,
      "step": 7581,
      "training_step_time": 17.023873329162598
    },
    {
      "epoch": 7.230758666992187e-06,
      "model_forward_time": 5.953428745269775,
      "step": 7582
    },
    {
      "epoch": 7.230758666992187e-06,
      "step": 7582,
      "training_step_time": 17.381497621536255
    },
    {
      "epoch": 7.231712341308594e-06,
      "model_forward_time": 5.584636688232422,
      "step": 7583
    },
    {
      "epoch": 7.231712341308594e-06,
      "step": 7583,
      "training_step_time": 17.045695781707764
    },
    {
      "epoch": 7.232666015625e-06,
      "model_forward_time": 5.609126329421997,
      "step": 7584
    },
    {
      "epoch": 7.232666015625e-06,
      "step": 7584,
      "training_step_time": 17.115980625152588
    },
    {
      "epoch": 7.233619689941406e-06,
      "model_forward_time": 5.619325876235962,
      "step": 7585
    },
    {
      "epoch": 7.233619689941406e-06,
      "step": 7585,
      "training_step_time": 17.118535041809082
    },
    {
      "epoch": 7.234573364257813e-06,
      "model_forward_time": 5.611164808273315,
      "step": 7586
    },
    {
      "epoch": 7.234573364257813e-06,
      "step": 7586,
      "training_step_time": 17.07797622680664
    },
    {
      "epoch": 7.2355270385742185e-06,
      "model_forward_time": 5.580387592315674,
      "step": 7587
    },
    {
      "epoch": 7.2355270385742185e-06,
      "step": 7587,
      "training_step_time": 17.019227027893066
    },
    {
      "epoch": 7.236480712890625e-06,
      "model_forward_time": 5.935352087020874,
      "step": 7588
    },
    {
      "epoch": 7.236480712890625e-06,
      "step": 7588,
      "training_step_time": 17.3767409324646
    },
    {
      "epoch": 7.237434387207032e-06,
      "model_forward_time": 5.587642192840576,
      "step": 7589
    },
    {
      "epoch": 7.237434387207032e-06,
      "step": 7589,
      "training_step_time": 17.04528021812439
    },
    {
      "epoch": 7.238388061523437e-06,
      "grad_norm": 1.0668456554412842,
      "learning_rate": 1e-05,
      "loss": 0.0639,
      "step": 7590
    },
    {
      "epoch": 7.238388061523437e-06,
      "model_forward_time": 5.605372190475464,
      "step": 7590
    },
    {
      "dynamics_loss_avg": 0.05870514288544655,
      "epoch": 7.238388061523437e-06,
      "step": 7590
    },
    {
      "action_loss_avg": 0.009362485096789896,
      "epoch": 7.238388061523437e-06,
      "step": 7590
    },
    {
      "epoch": 7.238388061523437e-06,
      "step": 7590,
      "training_step_time": 17.095880031585693
    },
    {
      "epoch": 7.239341735839844e-06,
      "model_forward_time": 5.601192474365234,
      "step": 7591
    },
    {
      "epoch": 7.239341735839844e-06,
      "step": 7591,
      "training_step_time": 17.085398197174072
    },
    {
      "epoch": 7.24029541015625e-06,
      "model_forward_time": 5.602391719818115,
      "step": 7592
    },
    {
      "epoch": 7.24029541015625e-06,
      "step": 7592,
      "training_step_time": 17.04311180114746
    },
    {
      "epoch": 7.241249084472656e-06,
      "model_forward_time": 5.978704214096069,
      "step": 7593
    },
    {
      "epoch": 7.241249084472656e-06,
      "step": 7593,
      "training_step_time": 17.440081119537354
    },
    {
      "epoch": 7.242202758789063e-06,
      "model_forward_time": 5.603530406951904,
      "step": 7594
    },
    {
      "epoch": 7.242202758789063e-06,
      "step": 7594,
      "training_step_time": 17.079514741897583
    },
    {
      "epoch": 7.2431564331054685e-06,
      "model_forward_time": 5.607251405715942,
      "step": 7595
    },
    {
      "epoch": 7.2431564331054685e-06,
      "step": 7595,
      "training_step_time": 17.10311794281006
    },
    {
      "epoch": 7.244110107421875e-06,
      "model_forward_time": 5.615654468536377,
      "step": 7596
    },
    {
      "epoch": 7.244110107421875e-06,
      "step": 7596,
      "training_step_time": 17.089295148849487
    },
    {
      "epoch": 7.245063781738282e-06,
      "model_forward_time": 5.618864059448242,
      "step": 7597
    },
    {
      "epoch": 7.245063781738282e-06,
      "step": 7597,
      "training_step_time": 17.1192626953125
    },
    {
      "epoch": 7.246017456054687e-06,
      "model_forward_time": 5.609632968902588,
      "step": 7598
    },
    {
      "epoch": 7.246017456054687e-06,
      "step": 7598,
      "training_step_time": 17.09799861907959
    },
    {
      "epoch": 7.246971130371094e-06,
      "model_forward_time": 5.949537515640259,
      "step": 7599
    },
    {
      "epoch": 7.246971130371094e-06,
      "step": 7599,
      "training_step_time": 17.43634867668152
    },
    {
      "epoch": 7.2479248046875e-06,
      "grad_norm": 1.4984935522079468,
      "learning_rate": 1e-05,
      "loss": 0.0757,
      "step": 7600
    },
    {
      "epoch": 7.2479248046875e-06,
      "model_forward_time": 5.608596086502075,
      "step": 7600
    },
    {
      "dynamics_loss_avg": 0.05813850574195385,
      "epoch": 7.2479248046875e-06,
      "step": 7600
    },
    {
      "action_loss_avg": 0.01243050447665155,
      "epoch": 7.2479248046875e-06,
      "step": 7600
    },
    {
      "epoch": 7.2479248046875e-06,
      "step": 7600,
      "training_step_time": 17.072412967681885
    },
    {
      "epoch": 7.248878479003906e-06,
      "model_forward_time": 5.598880290985107,
      "step": 7601
    },
    {
      "epoch": 7.248878479003906e-06,
      "step": 7601,
      "training_step_time": 17.048501014709473
    },
    {
      "epoch": 7.249832153320313e-06,
      "model_forward_time": 5.593437194824219,
      "step": 7602
    },
    {
      "epoch": 7.249832153320313e-06,
      "step": 7602,
      "training_step_time": 17.015514850616455
    },
    {
      "epoch": 7.2507858276367185e-06,
      "model_forward_time": 5.5873188972473145,
      "step": 7603
    },
    {
      "epoch": 7.2507858276367185e-06,
      "step": 7603,
      "training_step_time": 17.01217770576477
    },
    {
      "epoch": 7.251739501953125e-06,
      "model_forward_time": 5.958771705627441,
      "step": 7604
    },
    {
      "epoch": 7.251739501953125e-06,
      "step": 7604,
      "training_step_time": 17.423513650894165
    },
    {
      "epoch": 7.252693176269532e-06,
      "model_forward_time": 5.619280576705933,
      "step": 7605
    },
    {
      "epoch": 7.252693176269532e-06,
      "step": 7605,
      "training_step_time": 17.088627576828003
    },
    {
      "epoch": 7.253646850585937e-06,
      "model_forward_time": 5.599907159805298,
      "step": 7606
    },
    {
      "epoch": 7.253646850585937e-06,
      "step": 7606,
      "training_step_time": 17.06507134437561
    },
    {
      "epoch": 7.254600524902344e-06,
      "model_forward_time": 5.597914695739746,
      "step": 7607
    },
    {
      "epoch": 7.254600524902344e-06,
      "step": 7607,
      "training_step_time": 17.04686951637268
    },
    {
      "epoch": 7.25555419921875e-06,
      "model_forward_time": 5.615020990371704,
      "step": 7608
    },
    {
      "epoch": 7.25555419921875e-06,
      "step": 7608,
      "training_step_time": 17.081454038619995
    },
    {
      "epoch": 7.256507873535156e-06,
      "model_forward_time": 5.992748498916626,
      "step": 7609
    },
    {
      "epoch": 7.256507873535156e-06,
      "step": 7609,
      "training_step_time": 17.474116802215576
    },
    {
      "epoch": 7.257461547851563e-06,
      "grad_norm": 0.7547451257705688,
      "learning_rate": 1e-05,
      "loss": 0.0697,
      "step": 7610
    },
    {
      "epoch": 7.257461547851563e-06,
      "model_forward_time": 5.598981142044067,
      "step": 7610
    },
    {
      "dynamics_loss_avg": 0.061064792796969414,
      "epoch": 7.257461547851563e-06,
      "step": 7610
    },
    {
      "action_loss_avg": 0.011286866618320346,
      "epoch": 7.257461547851563e-06,
      "step": 7610
    },
    {
      "epoch": 7.257461547851563e-06,
      "step": 7610,
      "training_step_time": 17.05147385597229
    },
    {
      "epoch": 7.2584152221679685e-06,
      "model_forward_time": 5.611844778060913,
      "step": 7611
    },
    {
      "epoch": 7.2584152221679685e-06,
      "step": 7611,
      "training_step_time": 17.063098907470703
    },
    {
      "epoch": 7.259368896484375e-06,
      "model_forward_time": 5.581583261489868,
      "step": 7612
    },
    {
      "epoch": 7.259368896484375e-06,
      "step": 7612,
      "training_step_time": 17.018414735794067
    },
    {
      "epoch": 7.260322570800782e-06,
      "model_forward_time": 5.59002947807312,
      "step": 7613
    },
    {
      "epoch": 7.260322570800782e-06,
      "step": 7613,
      "training_step_time": 17.044352054595947
    },
    {
      "epoch": 7.261276245117187e-06,
      "model_forward_time": 5.602314472198486,
      "step": 7614
    },
    {
      "epoch": 7.261276245117187e-06,
      "step": 7614,
      "training_step_time": 17.06718420982361
    },
    {
      "epoch": 7.262229919433594e-06,
      "model_forward_time": 5.977275609970093,
      "step": 7615
    },
    {
      "epoch": 7.262229919433594e-06,
      "step": 7615,
      "training_step_time": 17.43688201904297
    },
    {
      "epoch": 7.26318359375e-06,
      "model_forward_time": 5.609289884567261,
      "step": 7616
    },
    {
      "epoch": 7.26318359375e-06,
      "step": 7616,
      "training_step_time": 17.083420276641846
    },
    {
      "epoch": 7.264137268066406e-06,
      "model_forward_time": 5.608958005905151,
      "step": 7617
    },
    {
      "epoch": 7.264137268066406e-06,
      "step": 7617,
      "training_step_time": 17.091382026672363
    },
    {
      "epoch": 7.265090942382813e-06,
      "model_forward_time": 5.602381229400635,
      "step": 7618
    },
    {
      "epoch": 7.265090942382813e-06,
      "step": 7618,
      "training_step_time": 17.066688776016235
    },
    {
      "epoch": 7.2660446166992185e-06,
      "model_forward_time": 5.588186025619507,
      "step": 7619
    },
    {
      "epoch": 7.2660446166992185e-06,
      "step": 7619,
      "training_step_time": 17.04196333885193
    },
    {
      "epoch": 7.266998291015625e-06,
      "grad_norm": 1.2211767435073853,
      "learning_rate": 1e-05,
      "loss": 0.0637,
      "step": 7620
    },
    {
      "epoch": 7.266998291015625e-06,
      "model_forward_time": 5.600497484207153,
      "step": 7620
    },
    {
      "dynamics_loss_avg": 0.05243146698921919,
      "epoch": 7.266998291015625e-06,
      "step": 7620
    },
    {
      "action_loss_avg": 0.00921466164290905,
      "epoch": 7.266998291015625e-06,
      "step": 7620
    },
    {
      "epoch": 7.266998291015625e-06,
      "step": 7620,
      "training_step_time": 17.427638053894043
    },
    {
      "epoch": 7.267951965332032e-06,
      "model_forward_time": 5.5972490310668945,
      "step": 7621
    },
    {
      "epoch": 7.267951965332032e-06,
      "step": 7621,
      "training_step_time": 17.07386541366577
    },
    {
      "epoch": 7.268905639648437e-06,
      "model_forward_time": 5.599526405334473,
      "step": 7622
    },
    {
      "epoch": 7.268905639648437e-06,
      "step": 7622,
      "training_step_time": 17.057602882385254
    },
    {
      "epoch": 7.269859313964844e-06,
      "model_forward_time": 5.591242074966431,
      "step": 7623
    },
    {
      "epoch": 7.269859313964844e-06,
      "step": 7623,
      "training_step_time": 17.04500460624695
    },
    {
      "epoch": 7.27081298828125e-06,
      "model_forward_time": 5.604458332061768,
      "step": 7624
    },
    {
      "epoch": 7.27081298828125e-06,
      "step": 7624,
      "training_step_time": 17.059898376464844
    },
    {
      "epoch": 7.271766662597656e-06,
      "model_forward_time": 5.6044206619262695,
      "step": 7625
    },
    {
      "epoch": 7.271766662597656e-06,
      "step": 7625,
      "training_step_time": 17.050612211227417
    },
    {
      "epoch": 7.272720336914063e-06,
      "model_forward_time": 5.975270509719849,
      "step": 7626
    },
    {
      "epoch": 7.272720336914063e-06,
      "step": 7626,
      "training_step_time": 17.412622690200806
    },
    {
      "epoch": 7.2736740112304685e-06,
      "model_forward_time": 5.602535009384155,
      "step": 7627
    },
    {
      "epoch": 7.2736740112304685e-06,
      "step": 7627,
      "training_step_time": 17.079614400863647
    },
    {
      "epoch": 7.274627685546875e-06,
      "model_forward_time": 5.603407144546509,
      "step": 7628
    },
    {
      "epoch": 7.274627685546875e-06,
      "step": 7628,
      "training_step_time": 17.092291593551636
    },
    {
      "epoch": 7.275581359863281e-06,
      "model_forward_time": 5.625363349914551,
      "step": 7629
    },
    {
      "epoch": 7.275581359863281e-06,
      "step": 7629,
      "training_step_time": 17.10284399986267
    },
    {
      "epoch": 7.276535034179687e-06,
      "grad_norm": 1.077221393585205,
      "learning_rate": 1e-05,
      "loss": 0.0819,
      "step": 7630
    },
    {
      "epoch": 7.276535034179687e-06,
      "model_forward_time": 5.604601860046387,
      "step": 7630
    },
    {
      "dynamics_loss_avg": 0.06901603676378727,
      "epoch": 7.276535034179687e-06,
      "step": 7630
    },
    {
      "action_loss_avg": 0.012736224685795605,
      "epoch": 7.276535034179687e-06,
      "step": 7630
    },
    {
      "epoch": 7.276535034179687e-06,
      "step": 7630,
      "training_step_time": 17.08635401725769
    },
    {
      "epoch": 7.277488708496094e-06,
      "model_forward_time": 5.61672306060791,
      "step": 7631
    },
    {
      "epoch": 7.277488708496094e-06,
      "step": 7631,
      "training_step_time": 17.443468809127808
    },
    {
      "epoch": 7.2784423828125e-06,
      "model_forward_time": 5.588310718536377,
      "step": 7632
    },
    {
      "epoch": 7.2784423828125e-06,
      "step": 7632,
      "training_step_time": 17.00168466567993
    },
    {
      "epoch": 7.279396057128906e-06,
      "model_forward_time": 5.568634510040283,
      "step": 7633
    },
    {
      "epoch": 7.279396057128906e-06,
      "step": 7633,
      "training_step_time": 16.98553705215454
    },
    {
      "epoch": 7.280349731445313e-06,
      "model_forward_time": 5.5653722286224365,
      "step": 7634
    },
    {
      "epoch": 7.280349731445313e-06,
      "step": 7634,
      "training_step_time": 17.003257989883423
    },
    {
      "epoch": 7.2813034057617185e-06,
      "model_forward_time": 5.5696117877960205,
      "step": 7635
    },
    {
      "epoch": 7.2813034057617185e-06,
      "step": 7635,
      "training_step_time": 16.9930682182312
    },
    {
      "epoch": 7.282257080078125e-06,
      "model_forward_time": 5.577820777893066,
      "step": 7636
    },
    {
      "epoch": 7.282257080078125e-06,
      "step": 7636,
      "training_step_time": 17.000431299209595
    },
    {
      "epoch": 7.283210754394531e-06,
      "model_forward_time": 5.950560092926025,
      "step": 7637
    },
    {
      "epoch": 7.283210754394531e-06,
      "step": 7637,
      "training_step_time": 17.374035120010376
    },
    {
      "epoch": 7.284164428710937e-06,
      "model_forward_time": 5.595900058746338,
      "step": 7638
    },
    {
      "epoch": 7.284164428710937e-06,
      "step": 7638,
      "training_step_time": 17.064024686813354
    },
    {
      "epoch": 7.285118103027344e-06,
      "model_forward_time": 5.612149238586426,
      "step": 7639
    },
    {
      "epoch": 7.285118103027344e-06,
      "step": 7639,
      "training_step_time": 17.08915138244629
    },
    {
      "epoch": 7.28607177734375e-06,
      "grad_norm": 1.1257696151733398,
      "learning_rate": 1e-05,
      "loss": 0.0553,
      "step": 7640
    },
    {
      "epoch": 7.28607177734375e-06,
      "model_forward_time": 5.603224754333496,
      "step": 7640
    },
    {
      "dynamics_loss_avg": 0.046788219409063456,
      "epoch": 7.28607177734375e-06,
      "step": 7640
    },
    {
      "action_loss_avg": 0.006828845851123333,
      "epoch": 7.28607177734375e-06,
      "step": 7640
    },
    {
      "epoch": 7.28607177734375e-06,
      "step": 7640,
      "training_step_time": 17.097570419311523
    },
    {
      "epoch": 7.287025451660156e-06,
      "model_forward_time": 5.60672664642334,
      "step": 7641
    },
    {
      "epoch": 7.287025451660156e-06,
      "step": 7641,
      "training_step_time": 17.080016136169434
    },
    {
      "epoch": 7.287979125976563e-06,
      "model_forward_time": 5.954042673110962,
      "step": 7642
    },
    {
      "epoch": 7.287979125976563e-06,
      "step": 7642,
      "training_step_time": 17.418638706207275
    },
    {
      "epoch": 7.2889328002929686e-06,
      "model_forward_time": 5.60631799697876,
      "step": 7643
    },
    {
      "epoch": 7.2889328002929686e-06,
      "step": 7643,
      "training_step_time": 17.063432455062866
    },
    {
      "epoch": 7.289886474609375e-06,
      "model_forward_time": 5.598843336105347,
      "step": 7644
    },
    {
      "epoch": 7.289886474609375e-06,
      "step": 7644,
      "training_step_time": 17.051178455352783
    },
    {
      "epoch": 7.290840148925781e-06,
      "model_forward_time": 5.593335151672363,
      "step": 7645
    },
    {
      "epoch": 7.290840148925781e-06,
      "step": 7645,
      "training_step_time": 17.04525637626648
    },
    {
      "epoch": 7.291793823242187e-06,
      "model_forward_time": 5.601461172103882,
      "step": 7646
    },
    {
      "epoch": 7.291793823242187e-06,
      "step": 7646,
      "training_step_time": 17.0761399269104
    },
    {
      "epoch": 7.292747497558594e-06,
      "model_forward_time": 5.597474098205566,
      "step": 7647
    },
    {
      "epoch": 7.292747497558594e-06,
      "step": 7647,
      "training_step_time": 17.052903413772583
    },
    {
      "epoch": 7.293701171875e-06,
      "model_forward_time": 5.991641998291016,
      "step": 7648
    },
    {
      "epoch": 7.293701171875e-06,
      "step": 7648,
      "training_step_time": 17.450026035308838
    },
    {
      "epoch": 7.294654846191406e-06,
      "model_forward_time": 5.601008176803589,
      "step": 7649
    },
    {
      "epoch": 7.294654846191406e-06,
      "step": 7649,
      "training_step_time": 17.0446035861969
    },
    {
      "epoch": 7.295608520507813e-06,
      "grad_norm": 0.5040600299835205,
      "learning_rate": 1e-05,
      "loss": 0.0713,
      "step": 7650
    },
    {
      "epoch": 7.295608520507813e-06,
      "model_forward_time": 5.599856853485107,
      "step": 7650
    },
    {
      "dynamics_loss_avg": 0.06425356529653073,
      "epoch": 7.295608520507813e-06,
      "step": 7650
    },
    {
      "action_loss_avg": 0.01248905323445797,
      "epoch": 7.295608520507813e-06,
      "step": 7650
    },
    {
      "epoch": 7.295608520507813e-06,
      "step": 7650,
      "training_step_time": 17.06681489944458
    },
    {
      "epoch": 7.2965621948242186e-06,
      "model_forward_time": 5.593759536743164,
      "step": 7651
    },
    {
      "epoch": 7.2965621948242186e-06,
      "step": 7651,
      "training_step_time": 17.051435708999634
    },
    {
      "epoch": 7.297515869140625e-06,
      "model_forward_time": 5.592807769775391,
      "step": 7652
    },
    {
      "epoch": 7.297515869140625e-06,
      "step": 7652,
      "training_step_time": 17.03215193748474
    },
    {
      "epoch": 7.298469543457031e-06,
      "model_forward_time": 5.589698314666748,
      "step": 7653
    },
    {
      "epoch": 7.298469543457031e-06,
      "step": 7653,
      "training_step_time": 17.380446195602417
    },
    {
      "epoch": 7.299423217773437e-06,
      "model_forward_time": 5.593427419662476,
      "step": 7654
    },
    {
      "epoch": 7.299423217773437e-06,
      "step": 7654,
      "training_step_time": 17.068053007125854
    },
    {
      "epoch": 7.300376892089844e-06,
      "model_forward_time": 5.614133596420288,
      "step": 7655
    },
    {
      "epoch": 7.300376892089844e-06,
      "step": 7655,
      "training_step_time": 17.09106731414795
    },
    {
      "epoch": 7.30133056640625e-06,
      "model_forward_time": 5.603194236755371,
      "step": 7656
    },
    {
      "epoch": 7.30133056640625e-06,
      "step": 7656,
      "training_step_time": 17.074797868728638
    },
    {
      "epoch": 7.302284240722656e-06,
      "model_forward_time": 5.61022424697876,
      "step": 7657
    },
    {
      "epoch": 7.302284240722656e-06,
      "step": 7657,
      "training_step_time": 17.078556776046753
    },
    {
      "epoch": 7.303237915039063e-06,
      "model_forward_time": 5.596560955047607,
      "step": 7658
    },
    {
      "epoch": 7.303237915039063e-06,
      "step": 7658,
      "training_step_time": 17.08600902557373
    },
    {
      "epoch": 7.3041915893554686e-06,
      "model_forward_time": 5.981475114822388,
      "step": 7659
    },
    {
      "epoch": 7.3041915893554686e-06,
      "step": 7659,
      "training_step_time": 17.4772891998291
    },
    {
      "epoch": 7.305145263671875e-06,
      "grad_norm": 0.8018122911453247,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 7660
    },
    {
      "epoch": 7.305145263671875e-06,
      "model_forward_time": 5.607834577560425,
      "step": 7660
    },
    {
      "dynamics_loss_avg": 0.06418155264109374,
      "epoch": 7.305145263671875e-06,
      "step": 7660
    },
    {
      "action_loss_avg": 0.012365989899262787,
      "epoch": 7.305145263671875e-06,
      "step": 7660
    },
    {
      "epoch": 7.305145263671875e-06,
      "step": 7660,
      "training_step_time": 17.098929405212402
    },
    {
      "epoch": 7.306098937988281e-06,
      "model_forward_time": 5.613523721694946,
      "step": 7661
    },
    {
      "epoch": 7.306098937988281e-06,
      "step": 7661,
      "training_step_time": 17.074694633483887
    },
    {
      "epoch": 7.3070526123046874e-06,
      "model_forward_time": 5.612238168716431,
      "step": 7662
    },
    {
      "epoch": 7.3070526123046874e-06,
      "step": 7662,
      "training_step_time": 17.07963252067566
    },
    {
      "epoch": 7.308006286621094e-06,
      "model_forward_time": 5.610025644302368,
      "step": 7663
    },
    {
      "epoch": 7.308006286621094e-06,
      "step": 7663,
      "training_step_time": 17.106107234954834
    },
    {
      "epoch": 7.3089599609375e-06,
      "model_forward_time": 5.612010955810547,
      "step": 7664
    },
    {
      "epoch": 7.3089599609375e-06,
      "step": 7664,
      "training_step_time": 17.457104444503784
    },
    {
      "epoch": 7.309913635253906e-06,
      "model_forward_time": 5.6312925815582275,
      "step": 7665
    },
    {
      "epoch": 7.309913635253906e-06,
      "step": 7665,
      "training_step_time": 17.09531855583191
    },
    {
      "epoch": 7.310867309570313e-06,
      "model_forward_time": 5.6003899574279785,
      "step": 7666
    },
    {
      "epoch": 7.310867309570313e-06,
      "step": 7666,
      "training_step_time": 17.0743408203125
    },
    {
      "epoch": 7.3118209838867186e-06,
      "model_forward_time": 5.607654809951782,
      "step": 7667
    },
    {
      "epoch": 7.3118209838867186e-06,
      "step": 7667,
      "training_step_time": 17.09022068977356
    },
    {
      "epoch": 7.312774658203125e-06,
      "model_forward_time": 5.612436771392822,
      "step": 7668
    },
    {
      "epoch": 7.312774658203125e-06,
      "step": 7668,
      "training_step_time": 17.086973428726196
    },
    {
      "epoch": 7.313728332519531e-06,
      "model_forward_time": 5.606692552566528,
      "step": 7669
    },
    {
      "epoch": 7.313728332519531e-06,
      "step": 7669,
      "training_step_time": 17.076485633850098
    },
    {
      "epoch": 7.3146820068359374e-06,
      "grad_norm": 0.08258164674043655,
      "learning_rate": 1e-05,
      "loss": 0.0824,
      "step": 7670
    },
    {
      "epoch": 7.3146820068359374e-06,
      "model_forward_time": 5.952472925186157,
      "step": 7670
    },
    {
      "dynamics_loss_avg": 0.07378793507814407,
      "epoch": 7.3146820068359374e-06,
      "step": 7670
    },
    {
      "action_loss_avg": 0.010265616315882653,
      "epoch": 7.3146820068359374e-06,
      "step": 7670
    },
    {
      "epoch": 7.3146820068359374e-06,
      "step": 7670,
      "training_step_time": 17.37365961074829
    },
    {
      "epoch": 7.315635681152344e-06,
      "model_forward_time": 5.566703796386719,
      "step": 7671
    },
    {
      "epoch": 7.315635681152344e-06,
      "step": 7671,
      "training_step_time": 16.978066444396973
    },
    {
      "epoch": 7.31658935546875e-06,
      "model_forward_time": 5.5708558559417725,
      "step": 7672
    },
    {
      "epoch": 7.31658935546875e-06,
      "step": 7672,
      "training_step_time": 17.014073610305786
    },
    {
      "epoch": 7.317543029785156e-06,
      "model_forward_time": 5.567091464996338,
      "step": 7673
    },
    {
      "epoch": 7.317543029785156e-06,
      "step": 7673,
      "training_step_time": 16.972678184509277
    },
    {
      "epoch": 7.318496704101563e-06,
      "model_forward_time": 5.58704948425293,
      "step": 7674
    },
    {
      "epoch": 7.318496704101563e-06,
      "step": 7674,
      "training_step_time": 17.048906326293945
    },
    {
      "epoch": 7.319450378417969e-06,
      "model_forward_time": 5.95783543586731,
      "step": 7675
    },
    {
      "epoch": 7.319450378417969e-06,
      "step": 7675,
      "training_step_time": 17.42886972427368
    },
    {
      "epoch": 7.320404052734375e-06,
      "model_forward_time": 5.608042240142822,
      "step": 7676
    },
    {
      "epoch": 7.320404052734375e-06,
      "step": 7676,
      "training_step_time": 17.067157745361328
    },
    {
      "epoch": 7.321357727050781e-06,
      "model_forward_time": 5.598800182342529,
      "step": 7677
    },
    {
      "epoch": 7.321357727050781e-06,
      "step": 7677,
      "training_step_time": 17.042372465133667
    },
    {
      "epoch": 7.3223114013671874e-06,
      "model_forward_time": 5.594542980194092,
      "step": 7678
    },
    {
      "epoch": 7.3223114013671874e-06,
      "step": 7678,
      "training_step_time": 17.02849268913269
    },
    {
      "epoch": 7.323265075683594e-06,
      "model_forward_time": 5.588923215866089,
      "step": 7679
    },
    {
      "epoch": 7.323265075683594e-06,
      "step": 7679,
      "training_step_time": 17.024667739868164
    },
    {
      "epoch": 7.32421875e-06,
      "grad_norm": 0.787921667098999,
      "learning_rate": 1e-05,
      "loss": 0.059,
      "step": 7680
    },
    {
      "epoch": 7.32421875e-06,
      "model_forward_time": 5.605695486068726,
      "step": 7680
    },
    {
      "dynamics_loss_avg": 0.052705968543887136,
      "epoch": 7.32421875e-06,
      "step": 7680
    },
    {
      "action_loss_avg": 0.008243288216181099,
      "epoch": 7.32421875e-06,
      "step": 7680
    },
    {
      "epoch": 7.32421875e-06,
      "step": 7680,
      "training_step_time": 17.072802782058716
    },
    {
      "epoch": 7.325172424316406e-06,
      "model_forward_time": 5.9912025928497314,
      "step": 7681
    },
    {
      "epoch": 7.325172424316406e-06,
      "step": 7681,
      "training_step_time": 17.48497748374939
    },
    {
      "epoch": 7.326126098632813e-06,
      "model_forward_time": 5.60211968421936,
      "step": 7682
    },
    {
      "epoch": 7.326126098632813e-06,
      "step": 7682,
      "training_step_time": 17.069123029708862
    },
    {
      "epoch": 7.327079772949219e-06,
      "model_forward_time": 5.60025429725647,
      "step": 7683
    },
    {
      "epoch": 7.327079772949219e-06,
      "step": 7683,
      "training_step_time": 17.03585124015808
    },
    {
      "epoch": 7.328033447265625e-06,
      "model_forward_time": 5.585505723953247,
      "step": 7684
    },
    {
      "epoch": 7.328033447265625e-06,
      "step": 7684,
      "training_step_time": 17.005056381225586
    },
    {
      "epoch": 7.328987121582031e-06,
      "model_forward_time": 5.5806145668029785,
      "step": 7685
    },
    {
      "epoch": 7.328987121582031e-06,
      "step": 7685,
      "training_step_time": 17.00147271156311
    },
    {
      "epoch": 7.3299407958984374e-06,
      "model_forward_time": 5.589020013809204,
      "step": 7686
    },
    {
      "epoch": 7.3299407958984374e-06,
      "step": 7686,
      "training_step_time": 17.41497015953064
    },
    {
      "epoch": 7.330894470214844e-06,
      "model_forward_time": 5.613870143890381,
      "step": 7687
    },
    {
      "epoch": 7.330894470214844e-06,
      "step": 7687,
      "training_step_time": 17.11050844192505
    },
    {
      "epoch": 7.33184814453125e-06,
      "model_forward_time": 5.615921258926392,
      "step": 7688
    },
    {
      "epoch": 7.33184814453125e-06,
      "step": 7688,
      "training_step_time": 17.09830594062805
    },
    {
      "epoch": 7.332801818847656e-06,
      "model_forward_time": 5.597043991088867,
      "step": 7689
    },
    {
      "epoch": 7.332801818847656e-06,
      "step": 7689,
      "training_step_time": 17.042895793914795
    },
    {
      "epoch": 7.333755493164063e-06,
      "grad_norm": 1.3740218877792358,
      "learning_rate": 1e-05,
      "loss": 0.0735,
      "step": 7690
    },
    {
      "epoch": 7.333755493164063e-06,
      "model_forward_time": 5.573972702026367,
      "step": 7690
    },
    {
      "dynamics_loss_avg": 0.05969831384718418,
      "epoch": 7.333755493164063e-06,
      "step": 7690
    },
    {
      "action_loss_avg": 0.009155416255816817,
      "epoch": 7.333755493164063e-06,
      "step": 7690
    },
    {
      "epoch": 7.333755493164063e-06,
      "step": 7690,
      "training_step_time": 16.993728637695312
    },
    {
      "epoch": 7.334709167480469e-06,
      "model_forward_time": 5.581908226013184,
      "step": 7691
    },
    {
      "epoch": 7.334709167480469e-06,
      "step": 7691,
      "training_step_time": 17.0238094329834
    },
    {
      "epoch": 7.335662841796875e-06,
      "model_forward_time": 5.948718070983887,
      "step": 7692
    },
    {
      "epoch": 7.335662841796875e-06,
      "step": 7692,
      "training_step_time": 17.389127016067505
    },
    {
      "epoch": 7.336616516113281e-06,
      "model_forward_time": 5.601816892623901,
      "step": 7693
    },
    {
      "epoch": 7.336616516113281e-06,
      "step": 7693,
      "training_step_time": 17.07209324836731
    },
    {
      "epoch": 7.3375701904296875e-06,
      "model_forward_time": 5.60068678855896,
      "step": 7694
    },
    {
      "epoch": 7.3375701904296875e-06,
      "step": 7694,
      "training_step_time": 17.068777561187744
    },
    {
      "epoch": 7.338523864746094e-06,
      "model_forward_time": 5.609329700469971,
      "step": 7695
    },
    {
      "epoch": 7.338523864746094e-06,
      "step": 7695,
      "training_step_time": 17.11005210876465
    },
    {
      "epoch": 7.3394775390625e-06,
      "model_forward_time": 5.617608308792114,
      "step": 7696
    },
    {
      "epoch": 7.3394775390625e-06,
      "step": 7696,
      "training_step_time": 17.096859216690063
    },
    {
      "epoch": 7.340431213378906e-06,
      "model_forward_time": 5.603356599807739,
      "step": 7697
    },
    {
      "epoch": 7.340431213378906e-06,
      "step": 7697,
      "training_step_time": 17.428496599197388
    },
    {
      "epoch": 7.341384887695313e-06,
      "model_forward_time": 5.593345880508423,
      "step": 7698
    },
    {
      "epoch": 7.341384887695313e-06,
      "step": 7698,
      "training_step_time": 17.02981948852539
    },
    {
      "epoch": 7.342338562011719e-06,
      "model_forward_time": 5.581587076187134,
      "step": 7699
    },
    {
      "epoch": 7.342338562011719e-06,
      "step": 7699,
      "training_step_time": 17.02356457710266
    },
    {
      "epoch": 7.343292236328125e-06,
      "grad_norm": 0.8583502769470215,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 7700
    },
    {
      "epoch": 7.343292236328125e-06,
      "model_forward_time": 5.581076622009277,
      "step": 7700
    },
    {
      "dynamics_loss_avg": 0.06449801046401263,
      "epoch": 7.343292236328125e-06,
      "step": 7700
    },
    {
      "action_loss_avg": 0.00886837299913168,
      "epoch": 7.343292236328125e-06,
      "step": 7700
    },
    {
      "epoch": 7.343292236328125e-06,
      "step": 7700,
      "training_step_time": 16.992764949798584
    },
    {
      "epoch": 7.344245910644531e-06,
      "model_forward_time": 5.580861568450928,
      "step": 7701
    },
    {
      "epoch": 7.344245910644531e-06,
      "step": 7701,
      "training_step_time": 17.006032705307007
    },
    {
      "epoch": 7.3451995849609375e-06,
      "model_forward_time": 5.597514867782593,
      "step": 7702
    },
    {
      "epoch": 7.3451995849609375e-06,
      "step": 7702,
      "training_step_time": 17.066550254821777
    },
    {
      "epoch": 7.346153259277344e-06,
      "model_forward_time": 5.9902215003967285,
      "step": 7703
    },
    {
      "epoch": 7.346153259277344e-06,
      "step": 7703,
      "training_step_time": 17.480726957321167
    },
    {
      "epoch": 7.34710693359375e-06,
      "model_forward_time": 5.603710889816284,
      "step": 7704
    },
    {
      "epoch": 7.34710693359375e-06,
      "step": 7704,
      "training_step_time": 17.095546007156372
    },
    {
      "epoch": 7.348060607910156e-06,
      "model_forward_time": 5.618546009063721,
      "step": 7705
    },
    {
      "epoch": 7.348060607910156e-06,
      "step": 7705,
      "training_step_time": 17.109966039657593
    },
    {
      "epoch": 7.349014282226563e-06,
      "model_forward_time": 5.607958555221558,
      "step": 7706
    },
    {
      "epoch": 7.349014282226563e-06,
      "step": 7706,
      "training_step_time": 17.095983028411865
    },
    {
      "epoch": 7.349967956542969e-06,
      "model_forward_time": 5.610738754272461,
      "step": 7707
    },
    {
      "epoch": 7.349967956542969e-06,
      "step": 7707,
      "training_step_time": 17.10777521133423
    },
    {
      "epoch": 7.350921630859375e-06,
      "model_forward_time": 5.953981161117554,
      "step": 7708
    },
    {
      "epoch": 7.350921630859375e-06,
      "step": 7708,
      "training_step_time": 17.417807579040527
    },
    {
      "epoch": 7.351875305175781e-06,
      "model_forward_time": 5.591699600219727,
      "step": 7709
    },
    {
      "epoch": 7.351875305175781e-06,
      "step": 7709,
      "training_step_time": 17.022106170654297
    },
    {
      "epoch": 7.3528289794921875e-06,
      "grad_norm": 0.44831159710884094,
      "learning_rate": 1e-05,
      "loss": 0.0553,
      "step": 7710
    },
    {
      "epoch": 7.3528289794921875e-06,
      "model_forward_time": 5.574178695678711,
      "step": 7710
    },
    {
      "dynamics_loss_avg": 0.04538757298141718,
      "epoch": 7.3528289794921875e-06,
      "step": 7710
    },
    {
      "action_loss_avg": 0.008254455868154764,
      "epoch": 7.3528289794921875e-06,
      "step": 7710
    },
    {
      "epoch": 7.3528289794921875e-06,
      "step": 7710,
      "training_step_time": 17.017776250839233
    },
    {
      "epoch": 7.353782653808594e-06,
      "model_forward_time": 5.57227349281311,
      "step": 7711
    },
    {
      "epoch": 7.353782653808594e-06,
      "step": 7711,
      "training_step_time": 16.988120794296265
    },
    {
      "epoch": 7.354736328125e-06,
      "model_forward_time": 5.570271253585815,
      "step": 7712
    },
    {
      "epoch": 7.354736328125e-06,
      "step": 7712,
      "training_step_time": 16.9987850189209
    },
    {
      "epoch": 7.355690002441406e-06,
      "model_forward_time": 5.570311069488525,
      "step": 7713
    },
    {
      "epoch": 7.355690002441406e-06,
      "step": 7713,
      "training_step_time": 16.993229389190674
    },
    {
      "epoch": 7.356643676757813e-06,
      "model_forward_time": 5.953811407089233,
      "step": 7714
    },
    {
      "epoch": 7.356643676757813e-06,
      "step": 7714,
      "training_step_time": 17.40151071548462
    },
    {
      "epoch": 7.357597351074219e-06,
      "model_forward_time": 5.5851991176605225,
      "step": 7715
    },
    {
      "epoch": 7.357597351074219e-06,
      "step": 7715,
      "training_step_time": 17.04906725883484
    },
    {
      "epoch": 7.358551025390625e-06,
      "model_forward_time": 5.613319635391235,
      "step": 7716
    },
    {
      "epoch": 7.358551025390625e-06,
      "step": 7716,
      "training_step_time": 17.111843585968018
    },
    {
      "epoch": 7.359504699707031e-06,
      "model_forward_time": 5.623275518417358,
      "step": 7717
    },
    {
      "epoch": 7.359504699707031e-06,
      "step": 7717,
      "training_step_time": 17.110987663269043
    },
    {
      "epoch": 7.3604583740234375e-06,
      "model_forward_time": 5.602253437042236,
      "step": 7718
    },
    {
      "epoch": 7.3604583740234375e-06,
      "step": 7718,
      "training_step_time": 17.047160148620605
    },
    {
      "epoch": 7.361412048339844e-06,
      "model_forward_time": 5.5851123332977295,
      "step": 7719
    },
    {
      "epoch": 7.361412048339844e-06,
      "step": 7719,
      "training_step_time": 17.38206458091736
    },
    {
      "epoch": 7.36236572265625e-06,
      "grad_norm": 1.4481005668640137,
      "learning_rate": 1e-05,
      "loss": 0.0765,
      "step": 7720
    },
    {
      "epoch": 7.36236572265625e-06,
      "model_forward_time": 5.582815408706665,
      "step": 7720
    },
    {
      "dynamics_loss_avg": 0.06553590279072523,
      "epoch": 7.36236572265625e-06,
      "step": 7720
    },
    {
      "action_loss_avg": 0.010189293016446754,
      "epoch": 7.36236572265625e-06,
      "step": 7720
    },
    {
      "epoch": 7.36236572265625e-06,
      "step": 7720,
      "training_step_time": 16.997009992599487
    },
    {
      "epoch": 7.363319396972656e-06,
      "model_forward_time": 5.58393931388855,
      "step": 7721
    },
    {
      "epoch": 7.363319396972656e-06,
      "step": 7721,
      "training_step_time": 17.017532110214233
    },
    {
      "epoch": 7.364273071289063e-06,
      "model_forward_time": 5.595012664794922,
      "step": 7722
    },
    {
      "epoch": 7.364273071289063e-06,
      "step": 7722,
      "training_step_time": 17.08030390739441
    },
    {
      "epoch": 7.365226745605469e-06,
      "model_forward_time": 5.610893249511719,
      "step": 7723
    },
    {
      "epoch": 7.365226745605469e-06,
      "step": 7723,
      "training_step_time": 17.104551076889038
    },
    {
      "epoch": 7.366180419921875e-06,
      "model_forward_time": 5.607731103897095,
      "step": 7724
    },
    {
      "epoch": 7.366180419921875e-06,
      "step": 7724,
      "training_step_time": 17.09263515472412
    },
    {
      "epoch": 7.367134094238281e-06,
      "model_forward_time": 5.959059715270996,
      "step": 7725
    },
    {
      "epoch": 7.367134094238281e-06,
      "step": 7725,
      "training_step_time": 17.398698806762695
    },
    {
      "epoch": 7.3680877685546875e-06,
      "model_forward_time": 5.57946515083313,
      "step": 7726
    },
    {
      "epoch": 7.3680877685546875e-06,
      "step": 7726,
      "training_step_time": 17.00197696685791
    },
    {
      "epoch": 7.369041442871094e-06,
      "model_forward_time": 5.585104942321777,
      "step": 7727
    },
    {
      "epoch": 7.369041442871094e-06,
      "step": 7727,
      "training_step_time": 17.011528491973877
    },
    {
      "epoch": 7.3699951171875e-06,
      "model_forward_time": 5.600091934204102,
      "step": 7728
    },
    {
      "epoch": 7.3699951171875e-06,
      "step": 7728,
      "training_step_time": 17.08656907081604
    },
    {
      "epoch": 7.370948791503906e-06,
      "model_forward_time": 5.610976934432983,
      "step": 7729
    },
    {
      "epoch": 7.370948791503906e-06,
      "step": 7729,
      "training_step_time": 17.105127573013306
    },
    {
      "epoch": 7.371902465820313e-06,
      "grad_norm": 0.6208978891372681,
      "learning_rate": 1e-05,
      "loss": 0.0665,
      "step": 7730
    },
    {
      "epoch": 7.371902465820313e-06,
      "model_forward_time": 5.964969635009766,
      "step": 7730
    },
    {
      "dynamics_loss_avg": 0.05559504199773073,
      "epoch": 7.371902465820313e-06,
      "step": 7730
    },
    {
      "action_loss_avg": 0.009697993146255612,
      "epoch": 7.371902465820313e-06,
      "step": 7730
    },
    {
      "epoch": 7.371902465820313e-06,
      "step": 7730,
      "training_step_time": 17.426432847976685
    },
    {
      "epoch": 7.372856140136719e-06,
      "model_forward_time": 5.594991683959961,
      "step": 7731
    },
    {
      "epoch": 7.372856140136719e-06,
      "step": 7731,
      "training_step_time": 17.004953861236572
    },
    {
      "epoch": 7.373809814453125e-06,
      "model_forward_time": 5.581059217453003,
      "step": 7732
    },
    {
      "epoch": 7.373809814453125e-06,
      "step": 7732,
      "training_step_time": 16.98638081550598
    },
    {
      "epoch": 7.374763488769531e-06,
      "model_forward_time": 5.570260286331177,
      "step": 7733
    },
    {
      "epoch": 7.374763488769531e-06,
      "step": 7733,
      "training_step_time": 17.00525212287903
    },
    {
      "epoch": 7.3757171630859375e-06,
      "model_forward_time": 5.5700907707214355,
      "step": 7734
    },
    {
      "epoch": 7.3757171630859375e-06,
      "step": 7734,
      "training_step_time": 16.984302759170532
    },
    {
      "epoch": 7.376670837402344e-06,
      "model_forward_time": 5.58976936340332,
      "step": 7735
    },
    {
      "epoch": 7.376670837402344e-06,
      "step": 7735,
      "training_step_time": 17.006697177886963
    },
    {
      "epoch": 7.37762451171875e-06,
      "model_forward_time": 5.960878372192383,
      "step": 7736
    },
    {
      "epoch": 7.37762451171875e-06,
      "step": 7736,
      "training_step_time": 17.393949270248413
    },
    {
      "epoch": 7.378578186035156e-06,
      "model_forward_time": 5.588811635971069,
      "step": 7737
    },
    {
      "epoch": 7.378578186035156e-06,
      "step": 7737,
      "training_step_time": 17.051581144332886
    },
    {
      "epoch": 7.379531860351563e-06,
      "model_forward_time": 5.604947805404663,
      "step": 7738
    },
    {
      "epoch": 7.379531860351563e-06,
      "step": 7738,
      "training_step_time": 17.112295627593994
    },
    {
      "epoch": 7.380485534667969e-06,
      "model_forward_time": 5.620508193969727,
      "step": 7739
    },
    {
      "epoch": 7.380485534667969e-06,
      "step": 7739,
      "training_step_time": 17.14118719100952
    },
    {
      "epoch": 7.381439208984375e-06,
      "grad_norm": 1.0102041959762573,
      "learning_rate": 1e-05,
      "loss": 0.0803,
      "step": 7740
    },
    {
      "epoch": 7.381439208984375e-06,
      "model_forward_time": 5.619332790374756,
      "step": 7740
    },
    {
      "dynamics_loss_avg": 0.07065666969865561,
      "epoch": 7.381439208984375e-06,
      "step": 7740
    },
    {
      "action_loss_avg": 0.013691772893071174,
      "epoch": 7.381439208984375e-06,
      "step": 7740
    },
    {
      "epoch": 7.381439208984375e-06,
      "step": 7740,
      "training_step_time": 17.133906602859497
    },
    {
      "epoch": 7.382392883300781e-06,
      "model_forward_time": 5.970746755599976,
      "step": 7741
    },
    {
      "epoch": 7.382392883300781e-06,
      "step": 7741,
      "training_step_time": 17.441255807876587
    },
    {
      "epoch": 7.3833465576171875e-06,
      "model_forward_time": 5.597686529159546,
      "step": 7742
    },
    {
      "epoch": 7.3833465576171875e-06,
      "step": 7742,
      "training_step_time": 17.050328969955444
    },
    {
      "epoch": 7.384300231933594e-06,
      "model_forward_time": 5.588751316070557,
      "step": 7743
    },
    {
      "epoch": 7.384300231933594e-06,
      "step": 7743,
      "training_step_time": 17.02235174179077
    },
    {
      "epoch": 7.38525390625e-06,
      "model_forward_time": 5.594704627990723,
      "step": 7744
    },
    {
      "epoch": 7.38525390625e-06,
      "step": 7744,
      "training_step_time": 17.04331398010254
    },
    {
      "epoch": 7.386207580566406e-06,
      "model_forward_time": 5.59091854095459,
      "step": 7745
    },
    {
      "epoch": 7.386207580566406e-06,
      "step": 7745,
      "training_step_time": 17.050156593322754
    },
    {
      "epoch": 7.387161254882813e-06,
      "model_forward_time": 5.597640752792358,
      "step": 7746
    },
    {
      "epoch": 7.387161254882813e-06,
      "step": 7746,
      "training_step_time": 17.454856157302856
    },
    {
      "epoch": 7.388114929199219e-06,
      "model_forward_time": 5.630362272262573,
      "step": 7747
    },
    {
      "epoch": 7.388114929199219e-06,
      "step": 7747,
      "training_step_time": 17.13319182395935
    },
    {
      "epoch": 7.389068603515625e-06,
      "model_forward_time": 5.602924823760986,
      "step": 7748
    },
    {
      "epoch": 7.389068603515625e-06,
      "step": 7748,
      "training_step_time": 17.051547288894653
    },
    {
      "epoch": 7.390022277832031e-06,
      "model_forward_time": 5.590261697769165,
      "step": 7749
    },
    {
      "epoch": 7.390022277832031e-06,
      "step": 7749,
      "training_step_time": 17.025912046432495
    },
    {
      "epoch": 7.3909759521484375e-06,
      "grad_norm": 0.7222974300384521,
      "learning_rate": 1e-05,
      "loss": 0.0735,
      "step": 7750
    },
    {
      "epoch": 7.3909759521484375e-06,
      "model_forward_time": 5.588414669036865,
      "step": 7750
    },
    {
      "dynamics_loss_avg": 0.06232543885707855,
      "epoch": 7.3909759521484375e-06,
      "step": 7750
    },
    {
      "action_loss_avg": 0.010553901572711766,
      "epoch": 7.3909759521484375e-06,
      "step": 7750
    },
    {
      "epoch": 7.3909759521484375e-06,
      "step": 7750,
      "training_step_time": 17.016343355178833
    },
    {
      "epoch": 7.391929626464844e-06,
      "model_forward_time": 5.595129013061523,
      "step": 7751
    },
    {
      "epoch": 7.391929626464844e-06,
      "step": 7751,
      "training_step_time": 17.02191662788391
    },
    {
      "epoch": 7.39288330078125e-06,
      "model_forward_time": 5.95678973197937,
      "step": 7752
    },
    {
      "epoch": 7.39288330078125e-06,
      "step": 7752,
      "training_step_time": 17.402326107025146
    },
    {
      "epoch": 7.393836975097656e-06,
      "model_forward_time": 5.603121042251587,
      "step": 7753
    },
    {
      "epoch": 7.393836975097656e-06,
      "step": 7753,
      "training_step_time": 17.107745885849
    },
    {
      "epoch": 7.394790649414062e-06,
      "model_forward_time": 5.620357513427734,
      "step": 7754
    },
    {
      "epoch": 7.394790649414062e-06,
      "step": 7754,
      "training_step_time": 17.142156839370728
    },
    {
      "epoch": 7.395744323730469e-06,
      "model_forward_time": 5.6313512325286865,
      "step": 7755
    },
    {
      "epoch": 7.395744323730469e-06,
      "step": 7755,
      "training_step_time": 17.097084283828735
    },
    {
      "epoch": 7.396697998046875e-06,
      "model_forward_time": 5.580115079879761,
      "step": 7756
    },
    {
      "epoch": 7.396697998046875e-06,
      "step": 7756,
      "training_step_time": 17.02330207824707
    },
    {
      "epoch": 7.397651672363281e-06,
      "model_forward_time": 5.562528371810913,
      "step": 7757
    },
    {
      "epoch": 7.397651672363281e-06,
      "step": 7757,
      "training_step_time": 17.33338737487793
    },
    {
      "epoch": 7.3986053466796875e-06,
      "model_forward_time": 5.55737566947937,
      "step": 7758
    },
    {
      "epoch": 7.3986053466796875e-06,
      "step": 7758,
      "training_step_time": 16.974985122680664
    },
    {
      "epoch": 7.399559020996094e-06,
      "model_forward_time": 5.55792498588562,
      "step": 7759
    },
    {
      "epoch": 7.399559020996094e-06,
      "step": 7759,
      "training_step_time": 16.98649501800537
    },
    {
      "epoch": 7.4005126953125e-06,
      "grad_norm": 0.6277420520782471,
      "learning_rate": 1e-05,
      "loss": 0.0663,
      "step": 7760
    },
    {
      "epoch": 7.4005126953125e-06,
      "model_forward_time": 5.564042568206787,
      "step": 7760
    },
    {
      "dynamics_loss_avg": 0.05864238329231739,
      "epoch": 7.4005126953125e-06,
      "step": 7760
    },
    {
      "action_loss_avg": 0.00782451949780807,
      "epoch": 7.4005126953125e-06,
      "step": 7760
    },
    {
      "epoch": 7.4005126953125e-06,
      "step": 7760,
      "training_step_time": 17.00456666946411
    },
    {
      "epoch": 7.401466369628906e-06,
      "model_forward_time": 5.612012624740601,
      "step": 7761
    },
    {
      "epoch": 7.401466369628906e-06,
      "step": 7761,
      "training_step_time": 17.05393409729004
    },
    {
      "epoch": 7.402420043945312e-06,
      "model_forward_time": 5.58722996711731,
      "step": 7762
    },
    {
      "epoch": 7.402420043945312e-06,
      "step": 7762,
      "training_step_time": 17.03593349456787
    },
    {
      "epoch": 7.403373718261719e-06,
      "model_forward_time": 5.946342706680298,
      "step": 7763
    },
    {
      "epoch": 7.403373718261719e-06,
      "step": 7763,
      "training_step_time": 17.39708185195923
    },
    {
      "epoch": 7.404327392578125e-06,
      "model_forward_time": 5.584006071090698,
      "step": 7764
    },
    {
      "epoch": 7.404327392578125e-06,
      "step": 7764,
      "training_step_time": 17.026597499847412
    },
    {
      "epoch": 7.405281066894531e-06,
      "model_forward_time": 5.595116853713989,
      "step": 7765
    },
    {
      "epoch": 7.405281066894531e-06,
      "step": 7765,
      "training_step_time": 17.07737135887146
    },
    {
      "epoch": 7.4062347412109375e-06,
      "model_forward_time": 5.605188369750977,
      "step": 7766
    },
    {
      "epoch": 7.4062347412109375e-06,
      "step": 7766,
      "training_step_time": 17.07908344268799
    },
    {
      "epoch": 7.407188415527344e-06,
      "model_forward_time": 5.602400779724121,
      "step": 7767
    },
    {
      "epoch": 7.407188415527344e-06,
      "step": 7767,
      "training_step_time": 17.04692506790161
    },
    {
      "epoch": 7.40814208984375e-06,
      "model_forward_time": 5.965795993804932,
      "step": 7768
    },
    {
      "epoch": 7.40814208984375e-06,
      "step": 7768,
      "training_step_time": 17.412766933441162
    },
    {
      "epoch": 7.409095764160156e-06,
      "model_forward_time": 5.611429929733276,
      "step": 7769
    },
    {
      "epoch": 7.409095764160156e-06,
      "step": 7769,
      "training_step_time": 17.075886011123657
    },
    {
      "epoch": 7.410049438476562e-06,
      "grad_norm": 1.0030932426452637,
      "learning_rate": 1e-05,
      "loss": 0.0896,
      "step": 7770
    },
    {
      "epoch": 7.410049438476562e-06,
      "model_forward_time": 5.601540565490723,
      "step": 7770
    },
    {
      "dynamics_loss_avg": 0.07809748835861682,
      "epoch": 7.410049438476562e-06,
      "step": 7770
    },
    {
      "action_loss_avg": 0.012793580908328295,
      "epoch": 7.410049438476562e-06,
      "step": 7770
    },
    {
      "epoch": 7.410049438476562e-06,
      "step": 7770,
      "training_step_time": 17.0958833694458
    },
    {
      "epoch": 7.411003112792969e-06,
      "model_forward_time": 5.6181628704071045,
      "step": 7771
    },
    {
      "epoch": 7.411003112792969e-06,
      "step": 7771,
      "training_step_time": 17.10308837890625
    },
    {
      "epoch": 7.411956787109375e-06,
      "model_forward_time": 5.611733436584473,
      "step": 7772
    },
    {
      "epoch": 7.411956787109375e-06,
      "step": 7772,
      "training_step_time": 17.070942878723145
    },
    {
      "epoch": 7.412910461425781e-06,
      "model_forward_time": 5.591906547546387,
      "step": 7773
    },
    {
      "epoch": 7.412910461425781e-06,
      "step": 7773,
      "training_step_time": 17.428361415863037
    },
    {
      "epoch": 7.4138641357421875e-06,
      "model_forward_time": 5.606808423995972,
      "step": 7774
    },
    {
      "epoch": 7.4138641357421875e-06,
      "step": 7774,
      "training_step_time": 17.106179237365723
    },
    {
      "epoch": 7.414817810058594e-06,
      "model_forward_time": 5.620840072631836,
      "step": 7775
    },
    {
      "epoch": 7.414817810058594e-06,
      "step": 7775,
      "training_step_time": 17.128984928131104
    },
    {
      "epoch": 7.415771484375e-06,
      "model_forward_time": 5.615403175354004,
      "step": 7776
    },
    {
      "epoch": 7.415771484375e-06,
      "step": 7776,
      "training_step_time": 17.129340410232544
    },
    {
      "epoch": 7.416725158691406e-06,
      "model_forward_time": 5.6147143840789795,
      "step": 7777
    },
    {
      "epoch": 7.416725158691406e-06,
      "step": 7777,
      "training_step_time": 17.113394498825073
    },
    {
      "epoch": 7.417678833007812e-06,
      "model_forward_time": 5.601842880249023,
      "step": 7778
    },
    {
      "epoch": 7.417678833007812e-06,
      "step": 7778,
      "training_step_time": 17.079430103302002
    },
    {
      "epoch": 7.418632507324219e-06,
      "model_forward_time": 5.971676588058472,
      "step": 7779
    },
    {
      "epoch": 7.418632507324219e-06,
      "step": 7779,
      "training_step_time": 17.438289642333984
    },
    {
      "epoch": 7.419586181640625e-06,
      "grad_norm": 1.520200252532959,
      "learning_rate": 1e-05,
      "loss": 0.0801,
      "step": 7780
    },
    {
      "epoch": 7.419586181640625e-06,
      "model_forward_time": 5.597759008407593,
      "step": 7780
    },
    {
      "dynamics_loss_avg": 0.06678374633193015,
      "epoch": 7.419586181640625e-06,
      "step": 7780
    },
    {
      "action_loss_avg": 0.01031041278038174,
      "epoch": 7.419586181640625e-06,
      "step": 7780
    },
    {
      "epoch": 7.419586181640625e-06,
      "step": 7780,
      "training_step_time": 17.04046130180359
    },
    {
      "epoch": 7.420539855957031e-06,
      "model_forward_time": 5.595229864120483,
      "step": 7781
    },
    {
      "epoch": 7.420539855957031e-06,
      "step": 7781,
      "training_step_time": 17.056084394454956
    },
    {
      "epoch": 7.4214935302734375e-06,
      "model_forward_time": 5.595095872879028,
      "step": 7782
    },
    {
      "epoch": 7.4214935302734375e-06,
      "step": 7782,
      "training_step_time": 17.027424097061157
    },
    {
      "epoch": 7.422447204589844e-06,
      "model_forward_time": 5.591951608657837,
      "step": 7783
    },
    {
      "epoch": 7.422447204589844e-06,
      "step": 7783,
      "training_step_time": 17.058698415756226
    },
    {
      "epoch": 7.42340087890625e-06,
      "model_forward_time": 5.9537787437438965,
      "step": 7784
    },
    {
      "epoch": 7.42340087890625e-06,
      "step": 7784,
      "training_step_time": 17.391955137252808
    },
    {
      "epoch": 7.424354553222656e-06,
      "model_forward_time": 5.588412523269653,
      "step": 7785
    },
    {
      "epoch": 7.424354553222656e-06,
      "step": 7785,
      "training_step_time": 17.01801371574402
    },
    {
      "epoch": 7.425308227539062e-06,
      "model_forward_time": 5.584511041641235,
      "step": 7786
    },
    {
      "epoch": 7.425308227539062e-06,
      "step": 7786,
      "training_step_time": 17.026344776153564
    },
    {
      "epoch": 7.426261901855469e-06,
      "model_forward_time": 5.5993993282318115,
      "step": 7787
    },
    {
      "epoch": 7.426261901855469e-06,
      "step": 7787,
      "training_step_time": 17.10159945487976
    },
    {
      "epoch": 7.427215576171875e-06,
      "model_forward_time": 5.620927333831787,
      "step": 7788
    },
    {
      "epoch": 7.427215576171875e-06,
      "step": 7788,
      "training_step_time": 17.15259337425232
    },
    {
      "epoch": 7.428169250488281e-06,
      "model_forward_time": 5.6287477016448975,
      "step": 7789
    },
    {
      "epoch": 7.428169250488281e-06,
      "step": 7789,
      "training_step_time": 17.126056909561157
    },
    {
      "epoch": 7.4291229248046875e-06,
      "grad_norm": 0.37634941935539246,
      "learning_rate": 1e-05,
      "loss": 0.0724,
      "step": 7790
    },
    {
      "epoch": 7.4291229248046875e-06,
      "model_forward_time": 5.9746081829071045,
      "step": 7790
    },
    {
      "dynamics_loss_avg": 0.060929268412292,
      "epoch": 7.4291229248046875e-06,
      "step": 7790
    },
    {
      "action_loss_avg": 0.012207089154981077,
      "epoch": 7.4291229248046875e-06,
      "step": 7790
    },
    {
      "epoch": 7.4291229248046875e-06,
      "step": 7790,
      "training_step_time": 17.418253183364868
    },
    {
      "epoch": 7.430076599121094e-06,
      "model_forward_time": 5.592512845993042,
      "step": 7791
    },
    {
      "epoch": 7.430076599121094e-06,
      "step": 7791,
      "training_step_time": 17.04413414001465
    },
    {
      "epoch": 7.4310302734375e-06,
      "model_forward_time": 5.595163106918335,
      "step": 7792
    },
    {
      "epoch": 7.4310302734375e-06,
      "step": 7792,
      "training_step_time": 17.041375398635864
    },
    {
      "epoch": 7.431983947753906e-06,
      "model_forward_time": 5.586547136306763,
      "step": 7793
    },
    {
      "epoch": 7.431983947753906e-06,
      "step": 7793,
      "training_step_time": 17.019071578979492
    },
    {
      "epoch": 7.432937622070312e-06,
      "model_forward_time": 5.591514348983765,
      "step": 7794
    },
    {
      "epoch": 7.432937622070312e-06,
      "step": 7794,
      "training_step_time": 17.032750129699707
    },
    {
      "epoch": 7.433891296386719e-06,
      "model_forward_time": 5.9493653774261475,
      "step": 7795
    },
    {
      "epoch": 7.433891296386719e-06,
      "step": 7795,
      "training_step_time": 17.418939352035522
    },
    {
      "epoch": 7.434844970703125e-06,
      "model_forward_time": 5.593411207199097,
      "step": 7796
    },
    {
      "epoch": 7.434844970703125e-06,
      "step": 7796,
      "training_step_time": 17.045778512954712
    },
    {
      "epoch": 7.435798645019531e-06,
      "model_forward_time": 5.592756032943726,
      "step": 7797
    },
    {
      "epoch": 7.435798645019531e-06,
      "step": 7797,
      "training_step_time": 17.051116704940796
    },
    {
      "epoch": 7.4367523193359375e-06,
      "model_forward_time": 5.592075824737549,
      "step": 7798
    },
    {
      "epoch": 7.4367523193359375e-06,
      "step": 7798,
      "training_step_time": 17.040635347366333
    },
    {
      "epoch": 7.437705993652344e-06,
      "model_forward_time": 5.594812631607056,
      "step": 7799
    },
    {
      "epoch": 7.437705993652344e-06,
      "step": 7799,
      "training_step_time": 17.033631563186646
    },
    {
      "epoch": 7.43865966796875e-06,
      "grad_norm": 0.6603389382362366,
      "learning_rate": 1e-05,
      "loss": 0.1008,
      "step": 7800
    },
    {
      "epoch": 7.43865966796875e-06,
      "model_forward_time": 5.5825934410095215,
      "step": 7800
    },
    {
      "dynamics_loss_avg": 0.08085401263087988,
      "epoch": 7.43865966796875e-06,
      "step": 7800
    },
    {
      "action_loss_avg": 0.015543501474894583,
      "epoch": 7.43865966796875e-06,
      "step": 7800
    },
    {
      "epoch": 7.43865966796875e-06,
      "step": 7800,
      "training_step_time": 17.02497434616089
    },
    {
      "epoch": 7.439613342285156e-06,
      "model_forward_time": 5.944084882736206,
      "step": 7801
    },
    {
      "epoch": 7.439613342285156e-06,
      "step": 7801,
      "training_step_time": 17.38945960998535
    },
    {
      "epoch": 7.440567016601562e-06,
      "model_forward_time": 5.58311915397644,
      "step": 7802
    },
    {
      "epoch": 7.440567016601562e-06,
      "step": 7802,
      "training_step_time": 17.04562211036682
    },
    {
      "epoch": 7.441520690917969e-06,
      "model_forward_time": 5.614483594894409,
      "step": 7803
    },
    {
      "epoch": 7.441520690917969e-06,
      "step": 7803,
      "training_step_time": 17.127046585083008
    },
    {
      "epoch": 7.442474365234375e-06,
      "model_forward_time": 5.629312515258789,
      "step": 7804
    },
    {
      "epoch": 7.442474365234375e-06,
      "step": 7804,
      "training_step_time": 17.146175146102905
    },
    {
      "epoch": 7.443428039550781e-06,
      "model_forward_time": 5.618980169296265,
      "step": 7805
    },
    {
      "epoch": 7.443428039550781e-06,
      "step": 7805,
      "training_step_time": 17.118271827697754
    },
    {
      "epoch": 7.4443817138671875e-06,
      "model_forward_time": 5.955493927001953,
      "step": 7806
    },
    {
      "epoch": 7.4443817138671875e-06,
      "step": 7806,
      "training_step_time": 17.4392671585083
    },
    {
      "epoch": 7.445335388183594e-06,
      "model_forward_time": 5.6080121994018555,
      "step": 7807
    },
    {
      "epoch": 7.445335388183594e-06,
      "step": 7807,
      "training_step_time": 17.09104871749878
    },
    {
      "epoch": 7.4462890625e-06,
      "model_forward_time": 5.6031272411346436,
      "step": 7808
    },
    {
      "epoch": 7.4462890625e-06,
      "step": 7808,
      "training_step_time": 17.063302040100098
    },
    {
      "epoch": 7.447242736816406e-06,
      "model_forward_time": 5.599602937698364,
      "step": 7809
    },
    {
      "epoch": 7.447242736816406e-06,
      "step": 7809,
      "training_step_time": 17.035054922103882
    },
    {
      "epoch": 7.448196411132812e-06,
      "grad_norm": 0.864706814289093,
      "learning_rate": 1e-05,
      "loss": 0.0643,
      "step": 7810
    },
    {
      "epoch": 7.448196411132812e-06,
      "model_forward_time": 5.586930513381958,
      "step": 7810
    },
    {
      "dynamics_loss_avg": 0.058769065886735916,
      "epoch": 7.448196411132812e-06,
      "step": 7810
    },
    {
      "action_loss_avg": 0.010649603605270386,
      "epoch": 7.448196411132812e-06,
      "step": 7810
    },
    {
      "epoch": 7.448196411132812e-06,
      "step": 7810,
      "training_step_time": 17.034093141555786
    },
    {
      "epoch": 7.449150085449219e-06,
      "model_forward_time": 5.5912253856658936,
      "step": 7811
    },
    {
      "epoch": 7.449150085449219e-06,
      "step": 7811,
      "training_step_time": 17.416006565093994
    },
    {
      "epoch": 7.450103759765625e-06,
      "model_forward_time": 5.60015869140625,
      "step": 7812
    },
    {
      "epoch": 7.450103759765625e-06,
      "step": 7812,
      "training_step_time": 17.061773538589478
    },
    {
      "epoch": 7.451057434082031e-06,
      "model_forward_time": 5.596017122268677,
      "step": 7813
    },
    {
      "epoch": 7.451057434082031e-06,
      "step": 7813,
      "training_step_time": 17.062381267547607
    },
    {
      "epoch": 7.4520111083984376e-06,
      "model_forward_time": 5.594624996185303,
      "step": 7814
    },
    {
      "epoch": 7.4520111083984376e-06,
      "step": 7814,
      "training_step_time": 17.042596340179443
    },
    {
      "epoch": 7.452964782714844e-06,
      "model_forward_time": 5.594781160354614,
      "step": 7815
    },
    {
      "epoch": 7.452964782714844e-06,
      "step": 7815,
      "training_step_time": 17.066590785980225
    },
    {
      "epoch": 7.45391845703125e-06,
      "model_forward_time": 5.615706920623779,
      "step": 7816
    },
    {
      "epoch": 7.45391845703125e-06,
      "step": 7816,
      "training_step_time": 17.109095811843872
    },
    {
      "epoch": 7.454872131347656e-06,
      "model_forward_time": 5.963357925415039,
      "step": 7817
    },
    {
      "epoch": 7.454872131347656e-06,
      "step": 7817,
      "training_step_time": 17.45281410217285
    },
    {
      "epoch": 7.455825805664062e-06,
      "model_forward_time": 5.612670421600342,
      "step": 7818
    },
    {
      "epoch": 7.455825805664062e-06,
      "step": 7818,
      "training_step_time": 17.086041927337646
    },
    {
      "epoch": 7.456779479980469e-06,
      "model_forward_time": 5.608957529067993,
      "step": 7819
    },
    {
      "epoch": 7.456779479980469e-06,
      "step": 7819,
      "training_step_time": 17.090399026870728
    },
    {
      "epoch": 7.457733154296875e-06,
      "grad_norm": 0.2520429193973541,
      "learning_rate": 1e-05,
      "loss": 0.0709,
      "step": 7820
    },
    {
      "epoch": 7.457733154296875e-06,
      "model_forward_time": 5.614640235900879,
      "step": 7820
    },
    {
      "dynamics_loss_avg": 0.056796207278966906,
      "epoch": 7.457733154296875e-06,
      "step": 7820
    },
    {
      "action_loss_avg": 0.008684418594930321,
      "epoch": 7.457733154296875e-06,
      "step": 7820
    },
    {
      "epoch": 7.457733154296875e-06,
      "step": 7820,
      "training_step_time": 17.076342344284058
    },
    {
      "epoch": 7.458686828613281e-06,
      "model_forward_time": 5.60210657119751,
      "step": 7821
    },
    {
      "epoch": 7.458686828613281e-06,
      "step": 7821,
      "training_step_time": 17.06280255317688
    },
    {
      "epoch": 7.4596405029296876e-06,
      "model_forward_time": 5.959992408752441,
      "step": 7822
    },
    {
      "epoch": 7.4596405029296876e-06,
      "step": 7822,
      "training_step_time": 17.417807579040527
    },
    {
      "epoch": 7.460594177246094e-06,
      "model_forward_time": 5.6028666496276855,
      "step": 7823
    },
    {
      "epoch": 7.460594177246094e-06,
      "step": 7823,
      "training_step_time": 17.073381662368774
    },
    {
      "epoch": 7.4615478515625e-06,
      "model_forward_time": 5.606743097305298,
      "step": 7824
    },
    {
      "epoch": 7.4615478515625e-06,
      "step": 7824,
      "training_step_time": 17.082956075668335
    },
    {
      "epoch": 7.462501525878906e-06,
      "model_forward_time": 5.609076023101807,
      "step": 7825
    },
    {
      "epoch": 7.462501525878906e-06,
      "step": 7825,
      "training_step_time": 17.101609706878662
    },
    {
      "epoch": 7.463455200195312e-06,
      "model_forward_time": 5.607979774475098,
      "step": 7826
    },
    {
      "epoch": 7.463455200195312e-06,
      "step": 7826,
      "training_step_time": 17.097207069396973
    },
    {
      "epoch": 7.464408874511719e-06,
      "model_forward_time": 5.637111663818359,
      "step": 7827
    },
    {
      "epoch": 7.464408874511719e-06,
      "step": 7827,
      "training_step_time": 17.114027500152588
    },
    {
      "epoch": 7.465362548828125e-06,
      "model_forward_time": 5.955489873886108,
      "step": 7828
    },
    {
      "epoch": 7.465362548828125e-06,
      "step": 7828,
      "training_step_time": 17.408225059509277
    },
    {
      "epoch": 7.466316223144531e-06,
      "model_forward_time": 5.615472078323364,
      "step": 7829
    },
    {
      "epoch": 7.466316223144531e-06,
      "step": 7829,
      "training_step_time": 17.06247067451477
    },
    {
      "epoch": 7.4672698974609376e-06,
      "grad_norm": 1.1467701196670532,
      "learning_rate": 1e-05,
      "loss": 0.07,
      "step": 7830
    },
    {
      "epoch": 7.4672698974609376e-06,
      "model_forward_time": 5.600261926651001,
      "step": 7830
    },
    {
      "dynamics_loss_avg": 0.06427048612385988,
      "epoch": 7.4672698974609376e-06,
      "step": 7830
    },
    {
      "action_loss_avg": 0.011100242449901997,
      "epoch": 7.4672698974609376e-06,
      "step": 7830
    },
    {
      "epoch": 7.4672698974609376e-06,
      "step": 7830,
      "training_step_time": 17.044235944747925
    },
    {
      "epoch": 7.468223571777344e-06,
      "model_forward_time": 5.597618818283081,
      "step": 7831
    },
    {
      "epoch": 7.468223571777344e-06,
      "step": 7831,
      "training_step_time": 17.071004390716553
    },
    {
      "epoch": 7.46917724609375e-06,
      "model_forward_time": 5.612401962280273,
      "step": 7832
    },
    {
      "epoch": 7.46917724609375e-06,
      "step": 7832,
      "training_step_time": 17.107301950454712
    },
    {
      "epoch": 7.4701309204101564e-06,
      "model_forward_time": 5.99968147277832,
      "step": 7833
    },
    {
      "epoch": 7.4701309204101564e-06,
      "step": 7833,
      "training_step_time": 17.49018907546997
    },
    {
      "epoch": 7.471084594726562e-06,
      "model_forward_time": 5.604768514633179,
      "step": 7834
    },
    {
      "epoch": 7.471084594726562e-06,
      "step": 7834,
      "training_step_time": 17.060784816741943
    },
    {
      "epoch": 7.472038269042969e-06,
      "model_forward_time": 5.58526349067688,
      "step": 7835
    },
    {
      "epoch": 7.472038269042969e-06,
      "step": 7835,
      "training_step_time": 17.009893894195557
    },
    {
      "epoch": 7.472991943359375e-06,
      "model_forward_time": 5.582756519317627,
      "step": 7836
    },
    {
      "epoch": 7.472991943359375e-06,
      "step": 7836,
      "training_step_time": 17.0189790725708
    },
    {
      "epoch": 7.473945617675781e-06,
      "model_forward_time": 5.5651726722717285,
      "step": 7837
    },
    {
      "epoch": 7.473945617675781e-06,
      "step": 7837,
      "training_step_time": 16.981731414794922
    },
    {
      "epoch": 7.4748992919921876e-06,
      "model_forward_time": 5.577941417694092,
      "step": 7838
    },
    {
      "epoch": 7.4748992919921876e-06,
      "step": 7838,
      "training_step_time": 16.984551429748535
    },
    {
      "epoch": 7.475852966308594e-06,
      "model_forward_time": 5.921781778335571,
      "step": 7839
    },
    {
      "epoch": 7.475852966308594e-06,
      "step": 7839,
      "training_step_time": 17.355981588363647
    },
    {
      "epoch": 7.476806640625e-06,
      "grad_norm": 0.22562865912914276,
      "learning_rate": 1e-05,
      "loss": 0.0522,
      "step": 7840
    },
    {
      "epoch": 7.476806640625e-06,
      "model_forward_time": 5.580360174179077,
      "step": 7840
    },
    {
      "dynamics_loss_avg": 0.04537389911711216,
      "epoch": 7.476806640625e-06,
      "step": 7840
    },
    {
      "action_loss_avg": 0.005270876071881503,
      "epoch": 7.476806640625e-06,
      "step": 7840
    },
    {
      "epoch": 7.476806640625e-06,
      "step": 7840,
      "training_step_time": 17.052192449569702
    },
    {
      "epoch": 7.4777603149414064e-06,
      "model_forward_time": 5.595882892608643,
      "step": 7841
    },
    {
      "epoch": 7.4777603149414064e-06,
      "step": 7841,
      "training_step_time": 17.06754231452942
    },
    {
      "epoch": 7.478713989257812e-06,
      "model_forward_time": 5.598917484283447,
      "step": 7842
    },
    {
      "epoch": 7.478713989257812e-06,
      "step": 7842,
      "training_step_time": 17.06883406639099
    },
    {
      "epoch": 7.479667663574219e-06,
      "model_forward_time": 5.602909326553345,
      "step": 7843
    },
    {
      "epoch": 7.479667663574219e-06,
      "step": 7843,
      "training_step_time": 17.04593539237976
    },
    {
      "epoch": 7.480621337890625e-06,
      "model_forward_time": 5.9602577686309814,
      "step": 7844
    },
    {
      "epoch": 7.480621337890625e-06,
      "step": 7844,
      "training_step_time": 17.439347505569458
    },
    {
      "epoch": 7.481575012207031e-06,
      "model_forward_time": 5.604637861251831,
      "step": 7845
    },
    {
      "epoch": 7.481575012207031e-06,
      "step": 7845,
      "training_step_time": 17.09780240058899
    },
    {
      "epoch": 7.482528686523438e-06,
      "model_forward_time": 5.617652654647827,
      "step": 7846
    },
    {
      "epoch": 7.482528686523438e-06,
      "step": 7846,
      "training_step_time": 17.1097993850708
    },
    {
      "epoch": 7.483482360839844e-06,
      "model_forward_time": 5.604970693588257,
      "step": 7847
    },
    {
      "epoch": 7.483482360839844e-06,
      "step": 7847,
      "training_step_time": 17.096418857574463
    },
    {
      "epoch": 7.48443603515625e-06,
      "model_forward_time": 5.6094276905059814,
      "step": 7848
    },
    {
      "epoch": 7.48443603515625e-06,
      "step": 7848,
      "training_step_time": 17.105738401412964
    },
    {
      "epoch": 7.4853897094726564e-06,
      "model_forward_time": 5.609588146209717,
      "step": 7849
    },
    {
      "epoch": 7.4853897094726564e-06,
      "step": 7849,
      "training_step_time": 17.087650775909424
    },
    {
      "epoch": 7.486343383789062e-06,
      "grad_norm": 1.0550615787506104,
      "learning_rate": 1e-05,
      "loss": 0.0625,
      "step": 7850
    },
    {
      "epoch": 7.486343383789062e-06,
      "model_forward_time": 5.95569109916687,
      "step": 7850
    },
    {
      "dynamics_loss_avg": 0.052840874530375004,
      "epoch": 7.486343383789062e-06,
      "step": 7850
    },
    {
      "action_loss_avg": 0.010415625001769513,
      "epoch": 7.486343383789062e-06,
      "step": 7850
    },
    {
      "epoch": 7.486343383789062e-06,
      "step": 7850,
      "training_step_time": 17.44872260093689
    },
    {
      "epoch": 7.487297058105469e-06,
      "model_forward_time": 5.61033034324646,
      "step": 7851
    },
    {
      "epoch": 7.487297058105469e-06,
      "step": 7851,
      "training_step_time": 17.11195397377014
    },
    {
      "epoch": 7.488250732421875e-06,
      "model_forward_time": 5.610284805297852,
      "step": 7852
    },
    {
      "epoch": 7.488250732421875e-06,
      "step": 7852,
      "training_step_time": 17.12147355079651
    },
    {
      "epoch": 7.489204406738281e-06,
      "model_forward_time": 5.613145351409912,
      "step": 7853
    },
    {
      "epoch": 7.489204406738281e-06,
      "step": 7853,
      "training_step_time": 17.113245248794556
    },
    {
      "epoch": 7.490158081054688e-06,
      "model_forward_time": 5.606341123580933,
      "step": 7854
    },
    {
      "epoch": 7.490158081054688e-06,
      "step": 7854,
      "training_step_time": 17.099339723587036
    },
    {
      "epoch": 7.491111755371094e-06,
      "model_forward_time": 5.974101305007935,
      "step": 7855
    },
    {
      "epoch": 7.491111755371094e-06,
      "step": 7855,
      "training_step_time": 17.46142864227295
    },
    {
      "epoch": 7.4920654296875e-06,
      "model_forward_time": 5.60788893699646,
      "step": 7856
    },
    {
      "epoch": 7.4920654296875e-06,
      "step": 7856,
      "training_step_time": 17.06784224510193
    },
    {
      "epoch": 7.4930191040039064e-06,
      "model_forward_time": 5.580291271209717,
      "step": 7857
    },
    {
      "epoch": 7.4930191040039064e-06,
      "step": 7857,
      "training_step_time": 17.004440784454346
    },
    {
      "epoch": 7.493972778320312e-06,
      "model_forward_time": 5.586643218994141,
      "step": 7858
    },
    {
      "epoch": 7.493972778320312e-06,
      "step": 7858,
      "training_step_time": 17.0311062335968
    },
    {
      "epoch": 7.494926452636719e-06,
      "model_forward_time": 5.606780529022217,
      "step": 7859
    },
    {
      "epoch": 7.494926452636719e-06,
      "step": 7859,
      "training_step_time": 17.055803775787354
    },
    {
      "epoch": 7.495880126953125e-06,
      "grad_norm": 0.5793201327323914,
      "learning_rate": 1e-05,
      "loss": 0.0734,
      "step": 7860
    },
    {
      "epoch": 7.495880126953125e-06,
      "model_forward_time": 6.113036632537842,
      "step": 7860
    },
    {
      "dynamics_loss_avg": 0.06433721398934722,
      "epoch": 7.495880126953125e-06,
      "step": 7860
    },
    {
      "action_loss_avg": 0.010161013595643453,
      "epoch": 7.495880126953125e-06,
      "step": 7860
    },
    {
      "epoch": 7.495880126953125e-06,
      "step": 7860,
      "training_step_time": 17.593932390213013
    },
    {
      "epoch": 7.496833801269531e-06,
      "model_forward_time": 5.608752012252808,
      "step": 7861
    },
    {
      "epoch": 7.496833801269531e-06,
      "step": 7861,
      "training_step_time": 17.10771155357361
    },
    {
      "epoch": 7.497787475585938e-06,
      "model_forward_time": 5.614719390869141,
      "step": 7862
    },
    {
      "epoch": 7.497787475585938e-06,
      "step": 7862,
      "training_step_time": 17.09465217590332
    },
    {
      "epoch": 7.498741149902344e-06,
      "model_forward_time": 5.615143775939941,
      "step": 7863
    },
    {
      "epoch": 7.498741149902344e-06,
      "step": 7863,
      "training_step_time": 17.115233659744263
    },
    {
      "epoch": 7.49969482421875e-06,
      "model_forward_time": 5.6151204109191895,
      "step": 7864
    },
    {
      "epoch": 7.49969482421875e-06,
      "step": 7864,
      "training_step_time": 17.095924377441406
    },
    {
      "epoch": 7.5006484985351565e-06,
      "model_forward_time": 5.589999198913574,
      "step": 7865
    },
    {
      "epoch": 7.5006484985351565e-06,
      "step": 7865,
      "training_step_time": 17.032215118408203
    },
    {
      "epoch": 7.501602172851562e-06,
      "model_forward_time": 6.014037609100342,
      "step": 7866
    },
    {
      "epoch": 7.501602172851562e-06,
      "step": 7866,
      "training_step_time": 17.453811645507812
    },
    {
      "epoch": 7.502555847167969e-06,
      "model_forward_time": 5.570533990859985,
      "step": 7867
    },
    {
      "epoch": 7.502555847167969e-06,
      "step": 7867,
      "training_step_time": 16.99161458015442
    },
    {
      "epoch": 7.503509521484375e-06,
      "model_forward_time": 5.564853191375732,
      "step": 7868
    },
    {
      "epoch": 7.503509521484375e-06,
      "step": 7868,
      "training_step_time": 16.98199200630188
    },
    {
      "epoch": 7.504463195800781e-06,
      "model_forward_time": 5.578096389770508,
      "step": 7869
    },
    {
      "epoch": 7.504463195800781e-06,
      "step": 7869,
      "training_step_time": 16.9912006855011
    },
    {
      "epoch": 7.505416870117188e-06,
      "grad_norm": 1.6485707759857178,
      "learning_rate": 1e-05,
      "loss": 0.0915,
      "step": 7870
    },
    {
      "epoch": 7.505416870117188e-06,
      "model_forward_time": 5.5709311962127686,
      "step": 7870
    },
    {
      "dynamics_loss_avg": 0.07617086060345173,
      "epoch": 7.505416870117188e-06,
      "step": 7870
    },
    {
      "action_loss_avg": 0.012760246731340885,
      "epoch": 7.505416870117188e-06,
      "step": 7870
    },
    {
      "epoch": 7.505416870117188e-06,
      "step": 7870,
      "training_step_time": 17.013680696487427
    },
    {
      "epoch": 7.506370544433594e-06,
      "model_forward_time": 5.586880922317505,
      "step": 7871
    },
    {
      "epoch": 7.506370544433594e-06,
      "step": 7871,
      "training_step_time": 17.417627573013306
    },
    {
      "epoch": 7.50732421875e-06,
      "model_forward_time": 5.588263750076294,
      "step": 7872
    },
    {
      "epoch": 7.50732421875e-06,
      "step": 7872,
      "training_step_time": 17.031510829925537
    },
    {
      "epoch": 7.5082778930664065e-06,
      "model_forward_time": 5.5959906578063965,
      "step": 7873
    },
    {
      "epoch": 7.5082778930664065e-06,
      "step": 7873,
      "training_step_time": 17.063881874084473
    },
    {
      "epoch": 7.509231567382812e-06,
      "model_forward_time": 5.606709241867065,
      "step": 7874
    },
    {
      "epoch": 7.509231567382812e-06,
      "step": 7874,
      "training_step_time": 17.0793879032135
    },
    {
      "epoch": 7.510185241699219e-06,
      "model_forward_time": 5.599143981933594,
      "step": 7875
    },
    {
      "epoch": 7.510185241699219e-06,
      "step": 7875,
      "training_step_time": 17.01436138153076
    },
    {
      "epoch": 7.511138916015625e-06,
      "model_forward_time": 5.617976188659668,
      "step": 7876
    },
    {
      "epoch": 7.511138916015625e-06,
      "step": 7876,
      "training_step_time": 17.070647954940796
    },
    {
      "epoch": 7.512092590332031e-06,
      "model_forward_time": 5.970236301422119,
      "step": 7877
    },
    {
      "epoch": 7.512092590332031e-06,
      "step": 7877,
      "training_step_time": 17.41761350631714
    },
    {
      "epoch": 7.513046264648438e-06,
      "model_forward_time": 5.600121259689331,
      "step": 7878
    },
    {
      "epoch": 7.513046264648438e-06,
      "step": 7878,
      "training_step_time": 17.075491428375244
    },
    {
      "epoch": 7.513999938964843e-06,
      "model_forward_time": 5.603784084320068,
      "step": 7879
    },
    {
      "epoch": 7.513999938964843e-06,
      "step": 7879,
      "training_step_time": 17.077478408813477
    },
    {
      "epoch": 7.51495361328125e-06,
      "grad_norm": 0.8977372050285339,
      "learning_rate": 1e-05,
      "loss": 0.0609,
      "step": 7880
    },
    {
      "epoch": 7.51495361328125e-06,
      "model_forward_time": 5.614173412322998,
      "step": 7880
    },
    {
      "dynamics_loss_avg": 0.05048226155340672,
      "epoch": 7.51495361328125e-06,
      "step": 7880
    },
    {
      "action_loss_avg": 0.00876286654965952,
      "epoch": 7.51495361328125e-06,
      "step": 7880
    },
    {
      "epoch": 7.51495361328125e-06,
      "step": 7880,
      "training_step_time": 17.092036485671997
    },
    {
      "epoch": 7.5159072875976565e-06,
      "model_forward_time": 5.602034330368042,
      "step": 7881
    },
    {
      "epoch": 7.5159072875976565e-06,
      "step": 7881,
      "training_step_time": 17.038672924041748
    },
    {
      "epoch": 7.516860961914062e-06,
      "model_forward_time": 5.59026575088501,
      "step": 7882
    },
    {
      "epoch": 7.516860961914062e-06,
      "step": 7882,
      "training_step_time": 17.412527799606323
    },
    {
      "epoch": 7.517814636230469e-06,
      "model_forward_time": 5.621238470077515,
      "step": 7883
    },
    {
      "epoch": 7.517814636230469e-06,
      "step": 7883,
      "training_step_time": 17.131311893463135
    },
    {
      "epoch": 7.518768310546875e-06,
      "model_forward_time": 5.610438823699951,
      "step": 7884
    },
    {
      "epoch": 7.518768310546875e-06,
      "step": 7884,
      "training_step_time": 17.120847702026367
    },
    {
      "epoch": 7.519721984863281e-06,
      "model_forward_time": 5.61534571647644,
      "step": 7885
    },
    {
      "epoch": 7.519721984863281e-06,
      "step": 7885,
      "training_step_time": 17.11138343811035
    },
    {
      "epoch": 7.520675659179688e-06,
      "model_forward_time": 5.605806589126587,
      "step": 7886
    },
    {
      "epoch": 7.520675659179688e-06,
      "step": 7886,
      "training_step_time": 17.06575298309326
    },
    {
      "epoch": 7.521629333496093e-06,
      "model_forward_time": 5.588922023773193,
      "step": 7887
    },
    {
      "epoch": 7.521629333496093e-06,
      "step": 7887,
      "training_step_time": 16.997901439666748
    },
    {
      "epoch": 7.5225830078125e-06,
      "model_forward_time": 5.937741994857788,
      "step": 7888
    },
    {
      "epoch": 7.5225830078125e-06,
      "step": 7888,
      "training_step_time": 17.34565281867981
    },
    {
      "epoch": 7.5235366821289065e-06,
      "model_forward_time": 5.5550537109375,
      "step": 7889
    },
    {
      "epoch": 7.5235366821289065e-06,
      "step": 7889,
      "training_step_time": 16.99359154701233
    },
    {
      "epoch": 7.524490356445312e-06,
      "grad_norm": 0.6618116497993469,
      "learning_rate": 1e-05,
      "loss": 0.0741,
      "step": 7890
    },
    {
      "epoch": 7.524490356445312e-06,
      "model_forward_time": 5.563311815261841,
      "step": 7890
    },
    {
      "dynamics_loss_avg": 0.06734169609844684,
      "epoch": 7.524490356445312e-06,
      "step": 7890
    },
    {
      "action_loss_avg": 0.009455313486978412,
      "epoch": 7.524490356445312e-06,
      "step": 7890
    },
    {
      "epoch": 7.524490356445312e-06,
      "step": 7890,
      "training_step_time": 16.984737396240234
    },
    {
      "epoch": 7.525444030761719e-06,
      "model_forward_time": 5.572066068649292,
      "step": 7891
    },
    {
      "epoch": 7.525444030761719e-06,
      "step": 7891,
      "training_step_time": 16.98004698753357
    },
    {
      "epoch": 7.526397705078125e-06,
      "model_forward_time": 5.574872970581055,
      "step": 7892
    },
    {
      "epoch": 7.526397705078125e-06,
      "step": 7892,
      "training_step_time": 17.011094331741333
    },
    {
      "epoch": 7.527351379394531e-06,
      "model_forward_time": 5.941782236099243,
      "step": 7893
    },
    {
      "epoch": 7.527351379394531e-06,
      "step": 7893,
      "training_step_time": 17.370734453201294
    },
    {
      "epoch": 7.528305053710938e-06,
      "model_forward_time": 5.5976269245147705,
      "step": 7894
    },
    {
      "epoch": 7.528305053710938e-06,
      "step": 7894,
      "training_step_time": 17.058922290802002
    },
    {
      "epoch": 7.529258728027343e-06,
      "model_forward_time": 5.601012468338013,
      "step": 7895
    },
    {
      "epoch": 7.529258728027343e-06,
      "step": 7895,
      "training_step_time": 17.079952478408813
    },
    {
      "epoch": 7.53021240234375e-06,
      "model_forward_time": 5.6104817390441895,
      "step": 7896
    },
    {
      "epoch": 7.53021240234375e-06,
      "step": 7896,
      "training_step_time": 17.096496105194092
    },
    {
      "epoch": 7.5311660766601565e-06,
      "model_forward_time": 5.618138551712036,
      "step": 7897
    },
    {
      "epoch": 7.5311660766601565e-06,
      "step": 7897,
      "training_step_time": 17.097026586532593
    },
    {
      "epoch": 7.532119750976562e-06,
      "model_forward_time": 5.60619044303894,
      "step": 7898
    },
    {
      "epoch": 7.532119750976562e-06,
      "step": 7898,
      "training_step_time": 17.05242133140564
    },
    {
      "epoch": 7.533073425292969e-06,
      "model_forward_time": 5.985367298126221,
      "step": 7899
    },
    {
      "epoch": 7.533073425292969e-06,
      "step": 7899,
      "training_step_time": 17.411832571029663
    },
    {
      "epoch": 7.534027099609375e-06,
      "grad_norm": 0.26332443952560425,
      "learning_rate": 1e-05,
      "loss": 0.0523,
      "step": 7900
    },
    {
      "epoch": 7.534027099609375e-06,
      "model_forward_time": 5.5838940143585205,
      "step": 7900
    },
    {
      "dynamics_loss_avg": 0.044768423587083814,
      "epoch": 7.534027099609375e-06,
      "step": 7900
    },
    {
      "action_loss_avg": 0.00668807802721858,
      "epoch": 7.534027099609375e-06,
      "step": 7900
    },
    {
      "epoch": 7.534027099609375e-06,
      "step": 7900,
      "training_step_time": 17.034554958343506
    },
    {
      "epoch": 7.534980773925781e-06,
      "model_forward_time": 5.583200931549072,
      "step": 7901
    },
    {
      "epoch": 7.534980773925781e-06,
      "step": 7901,
      "training_step_time": 17.040123462677002
    },
    {
      "epoch": 7.535934448242188e-06,
      "model_forward_time": 5.596116304397583,
      "step": 7902
    },
    {
      "epoch": 7.535934448242188e-06,
      "step": 7902,
      "training_step_time": 17.057403564453125
    },
    {
      "epoch": 7.536888122558593e-06,
      "model_forward_time": 5.601342678070068,
      "step": 7903
    },
    {
      "epoch": 7.536888122558593e-06,
      "step": 7903,
      "training_step_time": 17.065582752227783
    },
    {
      "epoch": 7.537841796875e-06,
      "model_forward_time": 5.598116636276245,
      "step": 7904
    },
    {
      "epoch": 7.537841796875e-06,
      "step": 7904,
      "training_step_time": 17.408036708831787
    },
    {
      "epoch": 7.5387954711914065e-06,
      "model_forward_time": 5.591264963150024,
      "step": 7905
    },
    {
      "epoch": 7.5387954711914065e-06,
      "step": 7905,
      "training_step_time": 17.01661491394043
    },
    {
      "epoch": 7.539749145507812e-06,
      "model_forward_time": 5.58887791633606,
      "step": 7906
    },
    {
      "epoch": 7.539749145507812e-06,
      "step": 7906,
      "training_step_time": 17.04465341567993
    },
    {
      "epoch": 7.540702819824219e-06,
      "model_forward_time": 5.595205068588257,
      "step": 7907
    },
    {
      "epoch": 7.540702819824219e-06,
      "step": 7907,
      "training_step_time": 17.0517258644104
    },
    {
      "epoch": 7.541656494140625e-06,
      "model_forward_time": 5.606076002120972,
      "step": 7908
    },
    {
      "epoch": 7.541656494140625e-06,
      "step": 7908,
      "training_step_time": 17.064693212509155
    },
    {
      "epoch": 7.542610168457031e-06,
      "model_forward_time": 5.610273122787476,
      "step": 7909
    },
    {
      "epoch": 7.542610168457031e-06,
      "step": 7909,
      "training_step_time": 17.100536823272705
    },
    {
      "epoch": 7.543563842773438e-06,
      "grad_norm": 0.7766961455345154,
      "learning_rate": 1e-05,
      "loss": 0.076,
      "step": 7910
    },
    {
      "epoch": 7.543563842773438e-06,
      "model_forward_time": 5.998952150344849,
      "step": 7910
    },
    {
      "dynamics_loss_avg": 0.06686263466253876,
      "epoch": 7.543563842773438e-06,
      "step": 7910
    },
    {
      "action_loss_avg": 0.01003685377072543,
      "epoch": 7.543563842773438e-06,
      "step": 7910
    },
    {
      "epoch": 7.543563842773438e-06,
      "step": 7910,
      "training_step_time": 17.495919227600098
    },
    {
      "epoch": 7.544517517089843e-06,
      "model_forward_time": 5.622207403182983,
      "step": 7911
    },
    {
      "epoch": 7.544517517089843e-06,
      "step": 7911,
      "training_step_time": 17.113959550857544
    },
    {
      "epoch": 7.54547119140625e-06,
      "model_forward_time": 5.593597888946533,
      "step": 7912
    },
    {
      "epoch": 7.54547119140625e-06,
      "step": 7912,
      "training_step_time": 17.025415182113647
    },
    {
      "epoch": 7.5464248657226565e-06,
      "model_forward_time": 5.579507112503052,
      "step": 7913
    },
    {
      "epoch": 7.5464248657226565e-06,
      "step": 7913,
      "training_step_time": 17.00324845314026
    },
    {
      "epoch": 7.547378540039062e-06,
      "model_forward_time": 5.571120262145996,
      "step": 7914
    },
    {
      "epoch": 7.547378540039062e-06,
      "step": 7914,
      "training_step_time": 16.98103165626526
    },
    {
      "epoch": 7.548332214355469e-06,
      "model_forward_time": 5.576449632644653,
      "step": 7915
    },
    {
      "epoch": 7.548332214355469e-06,
      "step": 7915,
      "training_step_time": 17.381715774536133
    },
    {
      "epoch": 7.549285888671875e-06,
      "model_forward_time": 5.57213568687439,
      "step": 7916
    },
    {
      "epoch": 7.549285888671875e-06,
      "step": 7916,
      "training_step_time": 16.99962329864502
    },
    {
      "epoch": 7.550239562988281e-06,
      "model_forward_time": 5.58246111869812,
      "step": 7917
    },
    {
      "epoch": 7.550239562988281e-06,
      "step": 7917,
      "training_step_time": 17.04101324081421
    },
    {
      "epoch": 7.551193237304688e-06,
      "model_forward_time": 5.5941643714904785,
      "step": 7918
    },
    {
      "epoch": 7.551193237304688e-06,
      "step": 7918,
      "training_step_time": 17.04742169380188
    },
    {
      "epoch": 7.552146911621093e-06,
      "model_forward_time": 5.597649335861206,
      "step": 7919
    },
    {
      "epoch": 7.552146911621093e-06,
      "step": 7919,
      "training_step_time": 17.043778896331787
    },
    {
      "epoch": 7.5531005859375e-06,
      "grad_norm": 2.207913875579834,
      "learning_rate": 1e-05,
      "loss": 0.0702,
      "step": 7920
    },
    {
      "epoch": 7.5531005859375e-06,
      "model_forward_time": 5.602264404296875,
      "step": 7920
    },
    {
      "dynamics_loss_avg": 0.05657059457153082,
      "epoch": 7.5531005859375e-06,
      "step": 7920
    },
    {
      "action_loss_avg": 0.010865030577406287,
      "epoch": 7.5531005859375e-06,
      "step": 7920
    },
    {
      "epoch": 7.5531005859375e-06,
      "step": 7920,
      "training_step_time": 17.0605309009552
    },
    {
      "epoch": 7.5540542602539065e-06,
      "model_forward_time": 5.98253321647644,
      "step": 7921
    },
    {
      "epoch": 7.5540542602539065e-06,
      "step": 7921,
      "training_step_time": 17.45058846473694
    },
    {
      "epoch": 7.555007934570312e-06,
      "model_forward_time": 5.597200632095337,
      "step": 7922
    },
    {
      "epoch": 7.555007934570312e-06,
      "step": 7922,
      "training_step_time": 17.0825355052948
    },
    {
      "epoch": 7.555961608886719e-06,
      "model_forward_time": 5.603454351425171,
      "step": 7923
    },
    {
      "epoch": 7.555961608886719e-06,
      "step": 7923,
      "training_step_time": 17.061079740524292
    },
    {
      "epoch": 7.556915283203125e-06,
      "model_forward_time": 5.606364965438843,
      "step": 7924
    },
    {
      "epoch": 7.556915283203125e-06,
      "step": 7924,
      "training_step_time": 17.073097467422485
    },
    {
      "epoch": 7.557868957519531e-06,
      "model_forward_time": 5.592093229293823,
      "step": 7925
    },
    {
      "epoch": 7.557868957519531e-06,
      "step": 7925,
      "training_step_time": 17.018389225006104
    },
    {
      "epoch": 7.558822631835938e-06,
      "model_forward_time": 5.9286208152771,
      "step": 7926
    },
    {
      "epoch": 7.558822631835938e-06,
      "step": 7926,
      "training_step_time": 17.36894989013672
    },
    {
      "epoch": 7.559776306152343e-06,
      "model_forward_time": 5.587239027023315,
      "step": 7927
    },
    {
      "epoch": 7.559776306152343e-06,
      "step": 7927,
      "training_step_time": 17.030798196792603
    },
    {
      "epoch": 7.56072998046875e-06,
      "model_forward_time": 5.593222379684448,
      "step": 7928
    },
    {
      "epoch": 7.56072998046875e-06,
      "step": 7928,
      "training_step_time": 17.056169748306274
    },
    {
      "epoch": 7.5616836547851565e-06,
      "model_forward_time": 5.598781585693359,
      "step": 7929
    },
    {
      "epoch": 7.5616836547851565e-06,
      "step": 7929,
      "training_step_time": 17.066429138183594
    },
    {
      "epoch": 7.562637329101562e-06,
      "grad_norm": 1.0655252933502197,
      "learning_rate": 1e-05,
      "loss": 0.0778,
      "step": 7930
    },
    {
      "epoch": 7.562637329101562e-06,
      "model_forward_time": 5.596294403076172,
      "step": 7930
    },
    {
      "dynamics_loss_avg": 0.06747983582317829,
      "epoch": 7.562637329101562e-06,
      "step": 7930
    },
    {
      "action_loss_avg": 0.013808809081092476,
      "epoch": 7.562637329101562e-06,
      "step": 7930
    },
    {
      "epoch": 7.562637329101562e-06,
      "step": 7930,
      "training_step_time": 17.050984621047974
    },
    {
      "epoch": 7.563591003417969e-06,
      "model_forward_time": 5.605638265609741,
      "step": 7931
    },
    {
      "epoch": 7.563591003417969e-06,
      "step": 7931,
      "training_step_time": 17.06833505630493
    },
    {
      "epoch": 7.564544677734375e-06,
      "model_forward_time": 5.977238416671753,
      "step": 7932
    },
    {
      "epoch": 7.564544677734375e-06,
      "step": 7932,
      "training_step_time": 17.404711723327637
    },
    {
      "epoch": 7.565498352050781e-06,
      "model_forward_time": 5.584740877151489,
      "step": 7933
    },
    {
      "epoch": 7.565498352050781e-06,
      "step": 7933,
      "training_step_time": 16.986900568008423
    },
    {
      "epoch": 7.566452026367188e-06,
      "model_forward_time": 5.585692405700684,
      "step": 7934
    },
    {
      "epoch": 7.566452026367188e-06,
      "step": 7934,
      "training_step_time": 17.027246475219727
    },
    {
      "epoch": 7.567405700683593e-06,
      "model_forward_time": 5.5778913497924805,
      "step": 7935
    },
    {
      "epoch": 7.567405700683593e-06,
      "step": 7935,
      "training_step_time": 17.03635334968567
    },
    {
      "epoch": 7.568359375e-06,
      "model_forward_time": 5.606146335601807,
      "step": 7936
    },
    {
      "epoch": 7.568359375e-06,
      "step": 7936,
      "training_step_time": 17.10494875907898
    },
    {
      "epoch": 7.5693130493164065e-06,
      "model_forward_time": 5.621877670288086,
      "step": 7937
    },
    {
      "epoch": 7.5693130493164065e-06,
      "step": 7937,
      "training_step_time": 17.490381717681885
    },
    {
      "epoch": 7.570266723632812e-06,
      "model_forward_time": 5.626151084899902,
      "step": 7938
    },
    {
      "epoch": 7.570266723632812e-06,
      "step": 7938,
      "training_step_time": 17.128071784973145
    },
    {
      "epoch": 7.571220397949219e-06,
      "model_forward_time": 5.620704650878906,
      "step": 7939
    },
    {
      "epoch": 7.571220397949219e-06,
      "step": 7939,
      "training_step_time": 17.13855290412903
    },
    {
      "epoch": 7.572174072265625e-06,
      "grad_norm": 0.6683104634284973,
      "learning_rate": 1e-05,
      "loss": 0.0666,
      "step": 7940
    },
    {
      "epoch": 7.572174072265625e-06,
      "model_forward_time": 5.634290456771851,
      "step": 7940
    },
    {
      "dynamics_loss_avg": 0.05831840243190527,
      "epoch": 7.572174072265625e-06,
      "step": 7940
    },
    {
      "action_loss_avg": 0.00848192316479981,
      "epoch": 7.572174072265625e-06,
      "step": 7940
    },
    {
      "epoch": 7.572174072265625e-06,
      "step": 7940,
      "training_step_time": 17.12585973739624
    },
    {
      "epoch": 7.573127746582031e-06,
      "model_forward_time": 5.608916759490967,
      "step": 7941
    },
    {
      "epoch": 7.573127746582031e-06,
      "step": 7941,
      "training_step_time": 17.073702335357666
    },
    {
      "epoch": 7.574081420898438e-06,
      "model_forward_time": 5.593423366546631,
      "step": 7942
    },
    {
      "epoch": 7.574081420898438e-06,
      "step": 7942,
      "training_step_time": 17.042360305786133
    },
    {
      "epoch": 7.575035095214843e-06,
      "model_forward_time": 5.966407537460327,
      "step": 7943
    },
    {
      "epoch": 7.575035095214843e-06,
      "step": 7943,
      "training_step_time": 17.39915657043457
    },
    {
      "epoch": 7.57598876953125e-06,
      "model_forward_time": 5.592279434204102,
      "step": 7944
    },
    {
      "epoch": 7.57598876953125e-06,
      "step": 7944,
      "training_step_time": 17.02228569984436
    },
    {
      "epoch": 7.5769424438476565e-06,
      "model_forward_time": 5.595749378204346,
      "step": 7945
    },
    {
      "epoch": 7.5769424438476565e-06,
      "step": 7945,
      "training_step_time": 17.07314395904541
    },
    {
      "epoch": 7.577896118164062e-06,
      "model_forward_time": 5.6153974533081055,
      "step": 7946
    },
    {
      "epoch": 7.577896118164062e-06,
      "step": 7946,
      "training_step_time": 17.084228515625
    },
    {
      "epoch": 7.578849792480469e-06,
      "model_forward_time": 5.598663568496704,
      "step": 7947
    },
    {
      "epoch": 7.578849792480469e-06,
      "step": 7947,
      "training_step_time": 17.041188955307007
    },
    {
      "epoch": 7.579803466796875e-06,
      "model_forward_time": 5.945220232009888,
      "step": 7948
    },
    {
      "epoch": 7.579803466796875e-06,
      "step": 7948,
      "training_step_time": 17.37881827354431
    },
    {
      "epoch": 7.580757141113281e-06,
      "model_forward_time": 5.586235523223877,
      "step": 7949
    },
    {
      "epoch": 7.580757141113281e-06,
      "step": 7949,
      "training_step_time": 17.00950598716736
    },
    {
      "epoch": 7.581710815429688e-06,
      "grad_norm": 0.7613717317581177,
      "learning_rate": 1e-05,
      "loss": 0.0735,
      "step": 7950
    },
    {
      "epoch": 7.581710815429688e-06,
      "model_forward_time": 5.571484088897705,
      "step": 7950
    },
    {
      "dynamics_loss_avg": 0.06449638158082963,
      "epoch": 7.581710815429688e-06,
      "step": 7950
    },
    {
      "action_loss_avg": 0.009316861582919955,
      "epoch": 7.581710815429688e-06,
      "step": 7950
    },
    {
      "epoch": 7.581710815429688e-06,
      "step": 7950,
      "training_step_time": 17.0116970539093
    },
    {
      "epoch": 7.582664489746093e-06,
      "model_forward_time": 5.584762334823608,
      "step": 7951
    },
    {
      "epoch": 7.582664489746093e-06,
      "step": 7951,
      "training_step_time": 17.033239364624023
    },
    {
      "epoch": 7.5836181640625e-06,
      "model_forward_time": 5.603454828262329,
      "step": 7952
    },
    {
      "epoch": 7.5836181640625e-06,
      "step": 7952,
      "training_step_time": 17.086769580841064
    },
    {
      "epoch": 7.5845718383789065e-06,
      "model_forward_time": 5.612071990966797,
      "step": 7953
    },
    {
      "epoch": 7.5845718383789065e-06,
      "step": 7953,
      "training_step_time": 17.102720975875854
    },
    {
      "epoch": 7.585525512695312e-06,
      "model_forward_time": 5.978175640106201,
      "step": 7954
    },
    {
      "epoch": 7.585525512695312e-06,
      "step": 7954,
      "training_step_time": 17.400343418121338
    },
    {
      "epoch": 7.586479187011719e-06,
      "model_forward_time": 5.605065107345581,
      "step": 7955
    },
    {
      "epoch": 7.586479187011719e-06,
      "step": 7955,
      "training_step_time": 17.06481170654297
    },
    {
      "epoch": 7.587432861328125e-06,
      "model_forward_time": 5.5951828956604,
      "step": 7956
    },
    {
      "epoch": 7.587432861328125e-06,
      "step": 7956,
      "training_step_time": 17.052982568740845
    },
    {
      "epoch": 7.588386535644531e-06,
      "model_forward_time": 5.5974907875061035,
      "step": 7957
    },
    {
      "epoch": 7.588386535644531e-06,
      "step": 7957,
      "training_step_time": 17.02903938293457
    },
    {
      "epoch": 7.589340209960938e-06,
      "model_forward_time": 5.601877689361572,
      "step": 7958
    },
    {
      "epoch": 7.589340209960938e-06,
      "step": 7958,
      "training_step_time": 17.068891286849976
    },
    {
      "epoch": 7.590293884277343e-06,
      "model_forward_time": 5.946510076522827,
      "step": 7959
    },
    {
      "epoch": 7.590293884277343e-06,
      "step": 7959,
      "training_step_time": 17.437774896621704
    },
    {
      "epoch": 7.59124755859375e-06,
      "grad_norm": 0.7445979118347168,
      "learning_rate": 1e-05,
      "loss": 0.0741,
      "step": 7960
    },
    {
      "epoch": 7.59124755859375e-06,
      "model_forward_time": 5.610893964767456,
      "step": 7960
    },
    {
      "dynamics_loss_avg": 0.0622429583221674,
      "epoch": 7.59124755859375e-06,
      "step": 7960
    },
    {
      "action_loss_avg": 0.009465142246335745,
      "epoch": 7.59124755859375e-06,
      "step": 7960
    },
    {
      "epoch": 7.59124755859375e-06,
      "step": 7960,
      "training_step_time": 17.08856701850891
    },
    {
      "epoch": 7.5922012329101565e-06,
      "model_forward_time": 5.58875846862793,
      "step": 7961
    },
    {
      "epoch": 7.5922012329101565e-06,
      "step": 7961,
      "training_step_time": 17.037494659423828
    },
    {
      "epoch": 7.593154907226562e-06,
      "model_forward_time": 5.572845458984375,
      "step": 7962
    },
    {
      "epoch": 7.593154907226562e-06,
      "step": 7962,
      "training_step_time": 16.980345249176025
    },
    {
      "epoch": 7.594108581542969e-06,
      "model_forward_time": 5.588693380355835,
      "step": 7963
    },
    {
      "epoch": 7.594108581542969e-06,
      "step": 7963,
      "training_step_time": 17.008897304534912
    },
    {
      "epoch": 7.595062255859375e-06,
      "model_forward_time": 5.576153755187988,
      "step": 7964
    },
    {
      "epoch": 7.595062255859375e-06,
      "step": 7964,
      "training_step_time": 17.355393648147583
    },
    {
      "epoch": 7.596015930175781e-06,
      "model_forward_time": 5.577139377593994,
      "step": 7965
    },
    {
      "epoch": 7.596015930175781e-06,
      "step": 7965,
      "training_step_time": 17.041461944580078
    },
    {
      "epoch": 7.596969604492188e-06,
      "model_forward_time": 5.600544691085815,
      "step": 7966
    },
    {
      "epoch": 7.596969604492188e-06,
      "step": 7966,
      "training_step_time": 17.08061909675598
    },
    {
      "epoch": 7.597923278808593e-06,
      "model_forward_time": 5.624105453491211,
      "step": 7967
    },
    {
      "epoch": 7.597923278808593e-06,
      "step": 7967,
      "training_step_time": 17.1191189289093
    },
    {
      "epoch": 7.598876953125e-06,
      "model_forward_time": 5.6099982261657715,
      "step": 7968
    },
    {
      "epoch": 7.598876953125e-06,
      "step": 7968,
      "training_step_time": 17.1176016330719
    },
    {
      "epoch": 7.5998306274414065e-06,
      "model_forward_time": 5.614670276641846,
      "step": 7969
    },
    {
      "epoch": 7.5998306274414065e-06,
      "step": 7969,
      "training_step_time": 17.103096961975098
    },
    {
      "epoch": 7.600784301757812e-06,
      "grad_norm": 0.908105194568634,
      "learning_rate": 1e-05,
      "loss": 0.0727,
      "step": 7970
    },
    {
      "epoch": 7.600784301757812e-06,
      "model_forward_time": 5.970222234725952,
      "step": 7970
    },
    {
      "dynamics_loss_avg": 0.06194509621709585,
      "epoch": 7.600784301757812e-06,
      "step": 7970
    },
    {
      "action_loss_avg": 0.010486369067803026,
      "epoch": 7.600784301757812e-06,
      "step": 7970
    },
    {
      "epoch": 7.600784301757812e-06,
      "step": 7970,
      "training_step_time": 17.450820922851562
    },
    {
      "epoch": 7.601737976074219e-06,
      "model_forward_time": 5.59926962852478,
      "step": 7971
    },
    {
      "epoch": 7.601737976074219e-06,
      "step": 7971,
      "training_step_time": 17.045145750045776
    },
    {
      "epoch": 7.602691650390625e-06,
      "model_forward_time": 5.592857360839844,
      "step": 7972
    },
    {
      "epoch": 7.602691650390625e-06,
      "step": 7972,
      "training_step_time": 17.008691787719727
    },
    {
      "epoch": 7.603645324707031e-06,
      "model_forward_time": 5.5976722240448,
      "step": 7973
    },
    {
      "epoch": 7.603645324707031e-06,
      "step": 7973,
      "training_step_time": 17.061332941055298
    },
    {
      "epoch": 7.604598999023438e-06,
      "model_forward_time": 5.601736545562744,
      "step": 7974
    },
    {
      "epoch": 7.604598999023438e-06,
      "step": 7974,
      "training_step_time": 17.056987047195435
    },
    {
      "epoch": 7.605552673339843e-06,
      "model_forward_time": 5.599979639053345,
      "step": 7975
    },
    {
      "epoch": 7.605552673339843e-06,
      "step": 7975,
      "training_step_time": 17.405872344970703
    },
    {
      "epoch": 7.60650634765625e-06,
      "model_forward_time": 5.591166257858276,
      "step": 7976
    },
    {
      "epoch": 7.60650634765625e-06,
      "step": 7976,
      "training_step_time": 17.02875852584839
    },
    {
      "epoch": 7.6074600219726565e-06,
      "model_forward_time": 5.585808992385864,
      "step": 7977
    },
    {
      "epoch": 7.6074600219726565e-06,
      "step": 7977,
      "training_step_time": 17.039570331573486
    },
    {
      "epoch": 7.608413696289062e-06,
      "model_forward_time": 5.597253084182739,
      "step": 7978
    },
    {
      "epoch": 7.608413696289062e-06,
      "step": 7978,
      "training_step_time": 17.063323736190796
    },
    {
      "epoch": 7.609367370605469e-06,
      "model_forward_time": 5.601975917816162,
      "step": 7979
    },
    {
      "epoch": 7.609367370605469e-06,
      "step": 7979,
      "training_step_time": 17.055806636810303
    },
    {
      "epoch": 7.610321044921875e-06,
      "grad_norm": 1.2351059913635254,
      "learning_rate": 1e-05,
      "loss": 0.0806,
      "step": 7980
    },
    {
      "epoch": 7.610321044921875e-06,
      "model_forward_time": 5.595007419586182,
      "step": 7980
    },
    {
      "dynamics_loss_avg": 0.06903050672262907,
      "epoch": 7.610321044921875e-06,
      "step": 7980
    },
    {
      "action_loss_avg": 0.01385819548740983,
      "epoch": 7.610321044921875e-06,
      "step": 7980
    },
    {
      "epoch": 7.610321044921875e-06,
      "step": 7980,
      "training_step_time": 17.0343496799469
    },
    {
      "epoch": 7.611274719238281e-06,
      "model_forward_time": 5.922967195510864,
      "step": 7981
    },
    {
      "epoch": 7.611274719238281e-06,
      "step": 7981,
      "training_step_time": 17.34562921524048
    },
    {
      "epoch": 7.612228393554688e-06,
      "model_forward_time": 5.596850395202637,
      "step": 7982
    },
    {
      "epoch": 7.612228393554688e-06,
      "step": 7982,
      "training_step_time": 17.02402663230896
    },
    {
      "epoch": 7.613182067871093e-06,
      "model_forward_time": 5.591926574707031,
      "step": 7983
    },
    {
      "epoch": 7.613182067871093e-06,
      "step": 7983,
      "training_step_time": 17.06054711341858
    },
    {
      "epoch": 7.6141357421875e-06,
      "model_forward_time": 5.608900308609009,
      "step": 7984
    },
    {
      "epoch": 7.6141357421875e-06,
      "step": 7984,
      "training_step_time": 17.065006971359253
    },
    {
      "epoch": 7.6150894165039066e-06,
      "model_forward_time": 5.610484838485718,
      "step": 7985
    },
    {
      "epoch": 7.6150894165039066e-06,
      "step": 7985,
      "training_step_time": 17.096415519714355
    },
    {
      "epoch": 7.616043090820312e-06,
      "model_forward_time": 5.976304769515991,
      "step": 7986
    },
    {
      "epoch": 7.616043090820312e-06,
      "step": 7986,
      "training_step_time": 17.428494691848755
    },
    {
      "epoch": 7.616996765136719e-06,
      "model_forward_time": 5.596853733062744,
      "step": 7987
    },
    {
      "epoch": 7.616996765136719e-06,
      "step": 7987,
      "training_step_time": 17.04542565345764
    },
    {
      "epoch": 7.617950439453125e-06,
      "model_forward_time": 5.6069176197052,
      "step": 7988
    },
    {
      "epoch": 7.617950439453125e-06,
      "step": 7988,
      "training_step_time": 17.08259105682373
    },
    {
      "epoch": 7.618904113769531e-06,
      "model_forward_time": 5.627422094345093,
      "step": 7989
    },
    {
      "epoch": 7.618904113769531e-06,
      "step": 7989,
      "training_step_time": 17.13862133026123
    },
    {
      "epoch": 7.619857788085938e-06,
      "grad_norm": 0.873655378818512,
      "learning_rate": 1e-05,
      "loss": 0.0693,
      "step": 7990
    },
    {
      "epoch": 7.619857788085938e-06,
      "model_forward_time": 5.620671033859253,
      "step": 7990
    },
    {
      "dynamics_loss_avg": 0.06611654199659825,
      "epoch": 7.619857788085938e-06,
      "step": 7990
    },
    {
      "action_loss_avg": 0.01272421337198466,
      "epoch": 7.619857788085938e-06,
      "step": 7990
    },
    {
      "epoch": 7.619857788085938e-06,
      "step": 7990,
      "training_step_time": 17.09838104248047
    },
    {
      "epoch": 7.6208114624023434e-06,
      "model_forward_time": 5.600107431411743,
      "step": 7991
    },
    {
      "epoch": 7.6208114624023434e-06,
      "step": 7991,
      "training_step_time": 17.40726637840271
    },
    {
      "epoch": 7.62176513671875e-06,
      "model_forward_time": 5.566725730895996,
      "step": 7992
    },
    {
      "epoch": 7.62176513671875e-06,
      "step": 7992,
      "training_step_time": 17.00248622894287
    },
    {
      "epoch": 7.6227188110351566e-06,
      "model_forward_time": 5.56742787361145,
      "step": 7993
    },
    {
      "epoch": 7.6227188110351566e-06,
      "step": 7993,
      "training_step_time": 16.984503507614136
    },
    {
      "epoch": 7.623672485351562e-06,
      "model_forward_time": 5.565223932266235,
      "step": 7994
    },
    {
      "epoch": 7.623672485351562e-06,
      "step": 7994,
      "training_step_time": 16.986430883407593
    },
    {
      "epoch": 7.624626159667969e-06,
      "model_forward_time": 5.574455976486206,
      "step": 7995
    },
    {
      "epoch": 7.624626159667969e-06,
      "step": 7995,
      "training_step_time": 17.009103536605835
    },
    {
      "epoch": 7.625579833984375e-06,
      "model_forward_time": 5.572293996810913,
      "step": 7996
    },
    {
      "epoch": 7.625579833984375e-06,
      "step": 7996,
      "training_step_time": 17.021310567855835
    },
    {
      "epoch": 7.626533508300781e-06,
      "model_forward_time": 5.976459264755249,
      "step": 7997
    },
    {
      "epoch": 7.626533508300781e-06,
      "step": 7997,
      "training_step_time": 17.45017695426941
    },
    {
      "epoch": 7.627487182617188e-06,
      "model_forward_time": 5.609001874923706,
      "step": 7998
    },
    {
      "epoch": 7.627487182617188e-06,
      "step": 7998,
      "training_step_time": 17.09942102432251
    },
    {
      "epoch": 7.6284408569335934e-06,
      "model_forward_time": 5.617289781570435,
      "step": 7999
    },
    {
      "epoch": 7.6284408569335934e-06,
      "step": 7999,
      "training_step_time": 17.100512742996216
    },
    {
      "epoch": 7.62939453125e-06,
      "grad_norm": 0.5753385424613953,
      "learning_rate": 1e-05,
      "loss": 0.0865,
      "step": 8000
    },
    {
      "epoch": 7.62939453125e-06,
      "model_forward_time": 5.660207271575928,
      "step": 8000
    },
    {
      "dynamics_loss_avg": 0.06518814601004123,
      "epoch": 7.62939453125e-06,
      "step": 8000
    },
    {
      "action_loss_avg": 0.011075844918377698,
      "epoch": 7.62939453125e-06,
      "step": 8000
    },
    {
      "epoch": 7.62939453125e-06,
      "step": 8000,
      "training_step_time": 17.18264651298523
    },
    {
      "epoch": 7.630348205566406e-06,
      "model_forward_time": 5.663910388946533,
      "step": 8001
    },
    {
      "epoch": 7.630348205566406e-06,
      "step": 8001,
      "training_step_time": 17.1685152053833
    },
    {
      "epoch": 7.631301879882813e-06,
      "model_forward_time": 6.0091235637664795,
      "step": 8002
    },
    {
      "epoch": 7.631301879882813e-06,
      "step": 8002,
      "training_step_time": 17.52887749671936
    },
    {
      "epoch": 7.632255554199219e-06,
      "model_forward_time": 5.678040027618408,
      "step": 8003
    },
    {
      "epoch": 7.632255554199219e-06,
      "step": 8003,
      "training_step_time": 17.222563982009888
    },
    {
      "epoch": 7.633209228515625e-06,
      "model_forward_time": 5.662025213241577,
      "step": 8004
    },
    {
      "epoch": 7.633209228515625e-06,
      "step": 8004,
      "training_step_time": 17.219820261001587
    },
    {
      "epoch": 7.634162902832032e-06,
      "model_forward_time": 5.627317905426025,
      "step": 8005
    },
    {
      "epoch": 7.634162902832032e-06,
      "step": 8005,
      "training_step_time": 17.14065432548523
    },
    {
      "epoch": 7.635116577148438e-06,
      "model_forward_time": 5.6168951988220215,
      "step": 8006
    },
    {
      "epoch": 7.635116577148438e-06,
      "step": 8006,
      "training_step_time": 17.084730625152588
    },
    {
      "epoch": 7.636070251464843e-06,
      "model_forward_time": 5.608329772949219,
      "step": 8007
    },
    {
      "epoch": 7.636070251464843e-06,
      "step": 8007,
      "training_step_time": 17.431214332580566
    },
    {
      "epoch": 7.63702392578125e-06,
      "model_forward_time": 5.598797559738159,
      "step": 8008
    },
    {
      "epoch": 7.63702392578125e-06,
      "step": 8008,
      "training_step_time": 17.058741569519043
    },
    {
      "epoch": 7.637977600097657e-06,
      "model_forward_time": 5.593708038330078,
      "step": 8009
    },
    {
      "epoch": 7.637977600097657e-06,
      "step": 8009,
      "training_step_time": 17.044563055038452
    },
    {
      "epoch": 7.638931274414062e-06,
      "grad_norm": 0.7589402198791504,
      "learning_rate": 1e-05,
      "loss": 0.0737,
      "step": 8010
    },
    {
      "epoch": 7.638931274414062e-06,
      "model_forward_time": 5.593734979629517,
      "step": 8010
    },
    {
      "dynamics_loss_avg": 0.060690832883119585,
      "epoch": 7.638931274414062e-06,
      "step": 8010
    },
    {
      "action_loss_avg": 0.010154717601835728,
      "epoch": 7.638931274414062e-06,
      "step": 8010
    },
    {
      "epoch": 7.638931274414062e-06,
      "step": 8010,
      "training_step_time": 17.07845401763916
    },
    {
      "epoch": 7.639884948730468e-06,
      "model_forward_time": 5.617226839065552,
      "step": 8011
    },
    {
      "epoch": 7.639884948730468e-06,
      "step": 8011,
      "training_step_time": 17.12679409980774
    },
    {
      "epoch": 7.640838623046875e-06,
      "model_forward_time": 5.62009072303772,
      "step": 8012
    },
    {
      "epoch": 7.640838623046875e-06,
      "step": 8012,
      "training_step_time": 17.11883568763733
    },
    {
      "epoch": 7.641792297363281e-06,
      "model_forward_time": 5.955951690673828,
      "step": 8013
    },
    {
      "epoch": 7.641792297363281e-06,
      "step": 8013,
      "training_step_time": 17.407844305038452
    },
    {
      "epoch": 7.642745971679687e-06,
      "model_forward_time": 5.5917112827301025,
      "step": 8014
    },
    {
      "epoch": 7.642745971679687e-06,
      "step": 8014,
      "training_step_time": 17.041247844696045
    },
    {
      "epoch": 7.643699645996094e-06,
      "model_forward_time": 5.587247371673584,
      "step": 8015
    },
    {
      "epoch": 7.643699645996094e-06,
      "step": 8015,
      "training_step_time": 16.996880292892456
    },
    {
      "epoch": 7.6446533203125e-06,
      "model_forward_time": 5.593554973602295,
      "step": 8016
    },
    {
      "epoch": 7.6446533203125e-06,
      "step": 8016,
      "training_step_time": 17.047104835510254
    },
    {
      "epoch": 7.645606994628906e-06,
      "model_forward_time": 5.59710431098938,
      "step": 8017
    },
    {
      "epoch": 7.645606994628906e-06,
      "step": 8017,
      "training_step_time": 17.069129943847656
    },
    {
      "epoch": 7.646560668945313e-06,
      "model_forward_time": 5.9818737506866455,
      "step": 8018
    },
    {
      "epoch": 7.646560668945313e-06,
      "step": 8018,
      "training_step_time": 17.445583820343018
    },
    {
      "epoch": 7.647514343261719e-06,
      "model_forward_time": 5.596299886703491,
      "step": 8019
    },
    {
      "epoch": 7.647514343261719e-06,
      "step": 8019,
      "training_step_time": 17.068088054656982
    },
    {
      "epoch": 7.648468017578125e-06,
      "grad_norm": 0.7655190229415894,
      "learning_rate": 1e-05,
      "loss": 0.0539,
      "step": 8020
    },
    {
      "epoch": 7.648468017578125e-06,
      "model_forward_time": 5.603021621704102,
      "step": 8020
    },
    {
      "dynamics_loss_avg": 0.05100900661200285,
      "epoch": 7.648468017578125e-06,
      "step": 8020
    },
    {
      "action_loss_avg": 0.006806421093642712,
      "epoch": 7.648468017578125e-06,
      "step": 8020
    },
    {
      "epoch": 7.648468017578125e-06,
      "step": 8020,
      "training_step_time": 17.093703508377075
    },
    {
      "epoch": 7.649421691894532e-06,
      "model_forward_time": 5.61188817024231,
      "step": 8021
    },
    {
      "epoch": 7.649421691894532e-06,
      "step": 8021,
      "training_step_time": 17.09124445915222
    },
    {
      "epoch": 7.650375366210938e-06,
      "model_forward_time": 5.608831882476807,
      "step": 8022
    },
    {
      "epoch": 7.650375366210938e-06,
      "step": 8022,
      "training_step_time": 17.06432271003723
    },
    {
      "epoch": 7.651329040527343e-06,
      "model_forward_time": 5.601390600204468,
      "step": 8023
    },
    {
      "epoch": 7.651329040527343e-06,
      "step": 8023,
      "training_step_time": 17.07388162612915
    },
    {
      "epoch": 7.65228271484375e-06,
      "model_forward_time": 5.938377618789673,
      "step": 8024
    },
    {
      "epoch": 7.65228271484375e-06,
      "step": 8024,
      "training_step_time": 17.408787488937378
    },
    {
      "epoch": 7.653236389160157e-06,
      "model_forward_time": 5.595324754714966,
      "step": 8025
    },
    {
      "epoch": 7.653236389160157e-06,
      "step": 8025,
      "training_step_time": 17.026718378067017
    },
    {
      "epoch": 7.654190063476562e-06,
      "model_forward_time": 5.5952489376068115,
      "step": 8026
    },
    {
      "epoch": 7.654190063476562e-06,
      "step": 8026,
      "training_step_time": 17.04221248626709
    },
    {
      "epoch": 7.655143737792968e-06,
      "model_forward_time": 5.595506906509399,
      "step": 8027
    },
    {
      "epoch": 7.655143737792968e-06,
      "step": 8027,
      "training_step_time": 17.032649993896484
    },
    {
      "epoch": 7.656097412109375e-06,
      "model_forward_time": 5.59160304069519,
      "step": 8028
    },
    {
      "epoch": 7.656097412109375e-06,
      "step": 8028,
      "training_step_time": 17.040902376174927
    },
    {
      "epoch": 7.657051086425781e-06,
      "model_forward_time": 5.968723773956299,
      "step": 8029
    },
    {
      "epoch": 7.657051086425781e-06,
      "step": 8029,
      "training_step_time": 17.41533875465393
    },
    {
      "epoch": 7.658004760742187e-06,
      "grad_norm": 0.7980818748474121,
      "learning_rate": 1e-05,
      "loss": 0.0558,
      "step": 8030
    },
    {
      "epoch": 7.658004760742187e-06,
      "model_forward_time": 5.591405630111694,
      "step": 8030
    },
    {
      "dynamics_loss_avg": 0.0448894165456295,
      "epoch": 7.658004760742187e-06,
      "step": 8030
    },
    {
      "action_loss_avg": 0.009038982610218228,
      "epoch": 7.658004760742187e-06,
      "step": 8030
    },
    {
      "epoch": 7.658004760742187e-06,
      "step": 8030,
      "training_step_time": 17.05374503135681
    },
    {
      "epoch": 7.658958435058594e-06,
      "model_forward_time": 5.600717067718506,
      "step": 8031
    },
    {
      "epoch": 7.658958435058594e-06,
      "step": 8031,
      "training_step_time": 17.058175802230835
    },
    {
      "epoch": 7.659912109375e-06,
      "model_forward_time": 5.599271297454834,
      "step": 8032
    },
    {
      "epoch": 7.659912109375e-06,
      "step": 8032,
      "training_step_time": 17.06779670715332
    },
    {
      "epoch": 7.660865783691406e-06,
      "model_forward_time": 5.601721286773682,
      "step": 8033
    },
    {
      "epoch": 7.660865783691406e-06,
      "step": 8033,
      "training_step_time": 17.063857793807983
    },
    {
      "epoch": 7.661819458007813e-06,
      "model_forward_time": 5.604533433914185,
      "step": 8034
    },
    {
      "epoch": 7.661819458007813e-06,
      "step": 8034,
      "training_step_time": 17.44142246246338
    },
    {
      "epoch": 7.662773132324219e-06,
      "model_forward_time": 5.599597215652466,
      "step": 8035
    },
    {
      "epoch": 7.662773132324219e-06,
      "step": 8035,
      "training_step_time": 17.07383704185486
    },
    {
      "epoch": 7.663726806640625e-06,
      "model_forward_time": 5.6016623973846436,
      "step": 8036
    },
    {
      "epoch": 7.663726806640625e-06,
      "step": 8036,
      "training_step_time": 17.076688289642334
    },
    {
      "epoch": 7.664680480957032e-06,
      "model_forward_time": 5.613493204116821,
      "step": 8037
    },
    {
      "epoch": 7.664680480957032e-06,
      "step": 8037,
      "training_step_time": 17.09118413925171
    },
    {
      "epoch": 7.665634155273438e-06,
      "model_forward_time": 5.609160423278809,
      "step": 8038
    },
    {
      "epoch": 7.665634155273438e-06,
      "step": 8038,
      "training_step_time": 17.067235231399536
    },
    {
      "epoch": 7.666587829589843e-06,
      "model_forward_time": 5.59578537940979,
      "step": 8039
    },
    {
      "epoch": 7.666587829589843e-06,
      "step": 8039,
      "training_step_time": 17.05245089530945
    },
    {
      "epoch": 7.66754150390625e-06,
      "grad_norm": 0.597881019115448,
      "learning_rate": 1e-05,
      "loss": 0.0729,
      "step": 8040
    },
    {
      "epoch": 7.66754150390625e-06,
      "model_forward_time": 5.964116096496582,
      "step": 8040
    },
    {
      "dynamics_loss_avg": 0.062280363962054254,
      "epoch": 7.66754150390625e-06,
      "step": 8040
    },
    {
      "action_loss_avg": 0.010432765260338784,
      "epoch": 7.66754150390625e-06,
      "step": 8040
    },
    {
      "epoch": 7.66754150390625e-06,
      "step": 8040,
      "training_step_time": 17.440491437911987
    },
    {
      "epoch": 7.668495178222657e-06,
      "model_forward_time": 5.612453937530518,
      "step": 8041
    },
    {
      "epoch": 7.668495178222657e-06,
      "step": 8041,
      "training_step_time": 17.070449113845825
    },
    {
      "epoch": 7.669448852539062e-06,
      "model_forward_time": 5.599765300750732,
      "step": 8042
    },
    {
      "epoch": 7.669448852539062e-06,
      "step": 8042,
      "training_step_time": 17.04845929145813
    },
    {
      "epoch": 7.670402526855468e-06,
      "model_forward_time": 5.594365119934082,
      "step": 8043
    },
    {
      "epoch": 7.670402526855468e-06,
      "step": 8043,
      "training_step_time": 17.049355506896973
    },
    {
      "epoch": 7.671356201171875e-06,
      "model_forward_time": 5.591164588928223,
      "step": 8044
    },
    {
      "epoch": 7.671356201171875e-06,
      "step": 8044,
      "training_step_time": 17.035200357437134
    },
    {
      "epoch": 7.672309875488281e-06,
      "model_forward_time": 5.959704875946045,
      "step": 8045
    },
    {
      "epoch": 7.672309875488281e-06,
      "step": 8045,
      "training_step_time": 17.423234462738037
    },
    {
      "epoch": 7.673263549804687e-06,
      "model_forward_time": 5.6078855991363525,
      "step": 8046
    },
    {
      "epoch": 7.673263549804687e-06,
      "step": 8046,
      "training_step_time": 17.11488938331604
    },
    {
      "epoch": 7.674217224121094e-06,
      "model_forward_time": 5.619912385940552,
      "step": 8047
    },
    {
      "epoch": 7.674217224121094e-06,
      "step": 8047,
      "training_step_time": 17.131943464279175
    },
    {
      "epoch": 7.6751708984375e-06,
      "model_forward_time": 5.61614727973938,
      "step": 8048
    },
    {
      "epoch": 7.6751708984375e-06,
      "step": 8048,
      "training_step_time": 17.119603395462036
    },
    {
      "epoch": 7.676124572753906e-06,
      "model_forward_time": 5.6160595417022705,
      "step": 8049
    },
    {
      "epoch": 7.676124572753906e-06,
      "step": 8049,
      "training_step_time": 17.117199659347534
    },
    {
      "epoch": 7.677078247070313e-06,
      "grad_norm": 0.29758355021476746,
      "learning_rate": 1e-05,
      "loss": 0.0615,
      "step": 8050
    },
    {
      "epoch": 7.677078247070313e-06,
      "model_forward_time": 5.608563423156738,
      "step": 8050
    },
    {
      "dynamics_loss_avg": 0.054092193767428395,
      "epoch": 7.677078247070313e-06,
      "step": 8050
    },
    {
      "action_loss_avg": 0.0077240023296326395,
      "epoch": 7.677078247070313e-06,
      "step": 8050
    },
    {
      "epoch": 7.677078247070313e-06,
      "step": 8050,
      "training_step_time": 17.063232898712158
    },
    {
      "epoch": 7.678031921386719e-06,
      "model_forward_time": 5.95858907699585,
      "step": 8051
    },
    {
      "epoch": 7.678031921386719e-06,
      "step": 8051,
      "training_step_time": 17.386586666107178
    },
    {
      "epoch": 7.678985595703125e-06,
      "model_forward_time": 5.582751512527466,
      "step": 8052
    },
    {
      "epoch": 7.678985595703125e-06,
      "step": 8052,
      "training_step_time": 17.024885654449463
    },
    {
      "epoch": 7.679939270019532e-06,
      "model_forward_time": 5.599453926086426,
      "step": 8053
    },
    {
      "epoch": 7.679939270019532e-06,
      "step": 8053,
      "training_step_time": 17.032807111740112
    },
    {
      "epoch": 7.680892944335938e-06,
      "model_forward_time": 5.625436544418335,
      "step": 8054
    },
    {
      "epoch": 7.680892944335938e-06,
      "step": 8054,
      "training_step_time": 17.05364990234375
    },
    {
      "epoch": 7.681846618652343e-06,
      "model_forward_time": 5.5991222858428955,
      "step": 8055
    },
    {
      "epoch": 7.681846618652343e-06,
      "step": 8055,
      "training_step_time": 17.033611297607422
    },
    {
      "epoch": 7.68280029296875e-06,
      "model_forward_time": 5.957429647445679,
      "step": 8056
    },
    {
      "epoch": 7.68280029296875e-06,
      "step": 8056,
      "training_step_time": 17.4050714969635
    },
    {
      "epoch": 7.683753967285157e-06,
      "model_forward_time": 5.589024305343628,
      "step": 8057
    },
    {
      "epoch": 7.683753967285157e-06,
      "step": 8057,
      "training_step_time": 17.037450551986694
    },
    {
      "epoch": 7.684707641601562e-06,
      "model_forward_time": 5.592547655105591,
      "step": 8058
    },
    {
      "epoch": 7.684707641601562e-06,
      "step": 8058,
      "training_step_time": 17.063297033309937
    },
    {
      "epoch": 7.685661315917968e-06,
      "model_forward_time": 5.606347560882568,
      "step": 8059
    },
    {
      "epoch": 7.685661315917968e-06,
      "step": 8059,
      "training_step_time": 17.07481336593628
    },
    {
      "epoch": 7.686614990234375e-06,
      "grad_norm": 0.8758652210235596,
      "learning_rate": 1e-05,
      "loss": 0.0758,
      "step": 8060
    },
    {
      "epoch": 7.686614990234375e-06,
      "model_forward_time": 5.604450225830078,
      "step": 8060
    },
    {
      "dynamics_loss_avg": 0.06233723517507315,
      "epoch": 7.686614990234375e-06,
      "step": 8060
    },
    {
      "action_loss_avg": 0.01205754061229527,
      "epoch": 7.686614990234375e-06,
      "step": 8060
    },
    {
      "epoch": 7.686614990234375e-06,
      "step": 8060,
      "training_step_time": 17.069929361343384
    },
    {
      "epoch": 7.687568664550781e-06,
      "model_forward_time": 5.598835706710815,
      "step": 8061
    },
    {
      "epoch": 7.687568664550781e-06,
      "step": 8061,
      "training_step_time": 17.042787551879883
    },
    {
      "epoch": 7.688522338867187e-06,
      "model_forward_time": 5.978960990905762,
      "step": 8062
    },
    {
      "epoch": 7.688522338867187e-06,
      "step": 8062,
      "training_step_time": 17.3866446018219
    },
    {
      "epoch": 7.689476013183594e-06,
      "model_forward_time": 5.620906352996826,
      "step": 8063
    },
    {
      "epoch": 7.689476013183594e-06,
      "step": 8063,
      "training_step_time": 17.075199127197266
    },
    {
      "epoch": 7.6904296875e-06,
      "model_forward_time": 5.613117218017578,
      "step": 8064
    },
    {
      "epoch": 7.6904296875e-06,
      "step": 8064,
      "training_step_time": 17.07324504852295
    },
    {
      "epoch": 7.691383361816406e-06,
      "model_forward_time": 5.620800495147705,
      "step": 8065
    },
    {
      "epoch": 7.691383361816406e-06,
      "step": 8065,
      "training_step_time": 17.13972282409668
    },
    {
      "epoch": 7.692337036132813e-06,
      "model_forward_time": 5.63502311706543,
      "step": 8066
    },
    {
      "epoch": 7.692337036132813e-06,
      "step": 8066,
      "training_step_time": 17.14431929588318
    },
    {
      "epoch": 7.693290710449219e-06,
      "model_forward_time": 5.992331266403198,
      "step": 8067
    },
    {
      "epoch": 7.693290710449219e-06,
      "step": 8067,
      "training_step_time": 17.479062795639038
    },
    {
      "epoch": 7.694244384765625e-06,
      "model_forward_time": 5.616103649139404,
      "step": 8068
    },
    {
      "epoch": 7.694244384765625e-06,
      "step": 8068,
      "training_step_time": 17.101619005203247
    },
    {
      "epoch": 7.695198059082032e-06,
      "model_forward_time": 5.606305122375488,
      "step": 8069
    },
    {
      "epoch": 7.695198059082032e-06,
      "step": 8069,
      "training_step_time": 17.0885808467865
    },
    {
      "epoch": 7.696151733398438e-06,
      "grad_norm": 0.8740540146827698,
      "learning_rate": 1e-05,
      "loss": 0.0682,
      "step": 8070
    },
    {
      "epoch": 7.696151733398438e-06,
      "model_forward_time": 5.599628925323486,
      "step": 8070
    },
    {
      "dynamics_loss_avg": 0.05982086323201656,
      "epoch": 7.696151733398438e-06,
      "step": 8070
    },
    {
      "action_loss_avg": 0.011967724096029997,
      "epoch": 7.696151733398438e-06,
      "step": 8070
    },
    {
      "epoch": 7.696151733398438e-06,
      "step": 8070,
      "training_step_time": 17.048051834106445
    },
    {
      "epoch": 7.697105407714843e-06,
      "model_forward_time": 5.6107213497161865,
      "step": 8071
    },
    {
      "epoch": 7.697105407714843e-06,
      "step": 8071,
      "training_step_time": 17.04112148284912
    },
    {
      "epoch": 7.69805908203125e-06,
      "model_forward_time": 5.603604316711426,
      "step": 8072
    },
    {
      "epoch": 7.69805908203125e-06,
      "step": 8072,
      "training_step_time": 17.440189599990845
    },
    {
      "epoch": 7.699012756347657e-06,
      "model_forward_time": 5.6140875816345215,
      "step": 8073
    },
    {
      "epoch": 7.699012756347657e-06,
      "step": 8073,
      "training_step_time": 17.131730794906616
    },
    {
      "epoch": 7.699966430664062e-06,
      "model_forward_time": 5.621689796447754,
      "step": 8074
    },
    {
      "epoch": 7.699966430664062e-06,
      "step": 8074,
      "training_step_time": 17.133493185043335
    },
    {
      "epoch": 7.700920104980468e-06,
      "model_forward_time": 5.625858783721924,
      "step": 8075
    },
    {
      "epoch": 7.700920104980468e-06,
      "step": 8075,
      "training_step_time": 17.114243745803833
    },
    {
      "epoch": 7.701873779296875e-06,
      "model_forward_time": 5.604113340377808,
      "step": 8076
    },
    {
      "epoch": 7.701873779296875e-06,
      "step": 8076,
      "training_step_time": 17.070671796798706
    },
    {
      "epoch": 7.702827453613281e-06,
      "model_forward_time": 5.575505495071411,
      "step": 8077
    },
    {
      "epoch": 7.702827453613281e-06,
      "step": 8077,
      "training_step_time": 16.987642288208008
    },
    {
      "epoch": 7.703781127929687e-06,
      "model_forward_time": 5.900385856628418,
      "step": 8078
    },
    {
      "epoch": 7.703781127929687e-06,
      "step": 8078,
      "training_step_time": 17.32560634613037
    },
    {
      "epoch": 7.704734802246094e-06,
      "model_forward_time": 5.570939064025879,
      "step": 8079
    },
    {
      "epoch": 7.704734802246094e-06,
      "step": 8079,
      "training_step_time": 17.00618290901184
    },
    {
      "epoch": 7.7056884765625e-06,
      "grad_norm": 1.6447196006774902,
      "learning_rate": 1e-05,
      "loss": 0.0691,
      "step": 8080
    },
    {
      "epoch": 7.7056884765625e-06,
      "model_forward_time": 5.568269491195679,
      "step": 8080
    },
    {
      "dynamics_loss_avg": 0.05429153051227331,
      "epoch": 7.7056884765625e-06,
      "step": 8080
    },
    {
      "action_loss_avg": 0.0094074449618347,
      "epoch": 7.7056884765625e-06,
      "step": 8080
    },
    {
      "epoch": 7.7056884765625e-06,
      "step": 8080,
      "training_step_time": 16.99662137031555
    },
    {
      "epoch": 7.706642150878906e-06,
      "model_forward_time": 5.573058843612671,
      "step": 8081
    },
    {
      "epoch": 7.706642150878906e-06,
      "step": 8081,
      "training_step_time": 16.981724739074707
    },
    {
      "epoch": 7.707595825195313e-06,
      "model_forward_time": 5.575188636779785,
      "step": 8082
    },
    {
      "epoch": 7.707595825195313e-06,
      "step": 8082,
      "training_step_time": 16.994447708129883
    },
    {
      "epoch": 7.708549499511719e-06,
      "model_forward_time": 5.5683159828186035,
      "step": 8083
    },
    {
      "epoch": 7.708549499511719e-06,
      "step": 8083,
      "training_step_time": 17.36871576309204
    },
    {
      "epoch": 7.709503173828125e-06,
      "model_forward_time": 5.576498746871948,
      "step": 8084
    },
    {
      "epoch": 7.709503173828125e-06,
      "step": 8084,
      "training_step_time": 17.035638570785522
    },
    {
      "epoch": 7.710456848144532e-06,
      "model_forward_time": 5.61793065071106,
      "step": 8085
    },
    {
      "epoch": 7.710456848144532e-06,
      "step": 8085,
      "training_step_time": 17.12444305419922
    },
    {
      "epoch": 7.711410522460938e-06,
      "model_forward_time": 5.613359451293945,
      "step": 8086
    },
    {
      "epoch": 7.711410522460938e-06,
      "step": 8086,
      "training_step_time": 17.107417821884155
    },
    {
      "epoch": 7.712364196777344e-06,
      "model_forward_time": 5.597057104110718,
      "step": 8087
    },
    {
      "epoch": 7.712364196777344e-06,
      "step": 8087,
      "training_step_time": 17.053566217422485
    },
    {
      "epoch": 7.71331787109375e-06,
      "model_forward_time": 5.584126710891724,
      "step": 8088
    },
    {
      "epoch": 7.71331787109375e-06,
      "step": 8088,
      "training_step_time": 17.02007508277893
    },
    {
      "epoch": 7.714271545410157e-06,
      "model_forward_time": 5.952899932861328,
      "step": 8089
    },
    {
      "epoch": 7.714271545410157e-06,
      "step": 8089,
      "training_step_time": 17.38299250602722
    },
    {
      "epoch": 7.715225219726562e-06,
      "grad_norm": 0.8477838039398193,
      "learning_rate": 1e-05,
      "loss": 0.0694,
      "step": 8090
    },
    {
      "epoch": 7.715225219726562e-06,
      "model_forward_time": 5.580855846405029,
      "step": 8090
    },
    {
      "dynamics_loss_avg": 0.06299004629254341,
      "epoch": 7.715225219726562e-06,
      "step": 8090
    },
    {
      "action_loss_avg": 0.00963995736092329,
      "epoch": 7.715225219726562e-06,
      "step": 8090
    },
    {
      "epoch": 7.715225219726562e-06,
      "step": 8090,
      "training_step_time": 17.039299726486206
    },
    {
      "epoch": 7.716178894042968e-06,
      "model_forward_time": 5.594076156616211,
      "step": 8091
    },
    {
      "epoch": 7.716178894042968e-06,
      "step": 8091,
      "training_step_time": 17.056944847106934
    },
    {
      "epoch": 7.717132568359376e-06,
      "model_forward_time": 5.604027509689331,
      "step": 8092
    },
    {
      "epoch": 7.717132568359376e-06,
      "step": 8092,
      "training_step_time": 17.076364517211914
    },
    {
      "epoch": 7.718086242675781e-06,
      "model_forward_time": 5.603079795837402,
      "step": 8093
    },
    {
      "epoch": 7.718086242675781e-06,
      "step": 8093,
      "training_step_time": 17.06706213951111
    },
    {
      "epoch": 7.719039916992187e-06,
      "model_forward_time": 5.969530344009399,
      "step": 8094
    },
    {
      "epoch": 7.719039916992187e-06,
      "step": 8094,
      "training_step_time": 17.412599802017212
    },
    {
      "epoch": 7.719993591308594e-06,
      "model_forward_time": 5.58644962310791,
      "step": 8095
    },
    {
      "epoch": 7.719993591308594e-06,
      "step": 8095,
      "training_step_time": 17.0439715385437
    },
    {
      "epoch": 7.720947265625e-06,
      "model_forward_time": 5.596909284591675,
      "step": 8096
    },
    {
      "epoch": 7.720947265625e-06,
      "step": 8096,
      "training_step_time": 17.084933519363403
    },
    {
      "epoch": 7.721900939941406e-06,
      "model_forward_time": 5.611004590988159,
      "step": 8097
    },
    {
      "epoch": 7.721900939941406e-06,
      "step": 8097,
      "training_step_time": 17.094602346420288
    },
    {
      "epoch": 7.722854614257813e-06,
      "model_forward_time": 5.617005348205566,
      "step": 8098
    },
    {
      "epoch": 7.722854614257813e-06,
      "step": 8098,
      "training_step_time": 17.104564428329468
    },
    {
      "epoch": 7.723808288574219e-06,
      "model_forward_time": 5.623296022415161,
      "step": 8099
    },
    {
      "epoch": 7.723808288574219e-06,
      "step": 8099,
      "training_step_time": 17.480138063430786
    },
    {
      "epoch": 7.724761962890625e-06,
      "grad_norm": 1.290488362312317,
      "learning_rate": 1e-05,
      "loss": 0.0702,
      "step": 8100
    },
    {
      "epoch": 7.724761962890625e-06,
      "model_forward_time": 5.6104583740234375,
      "step": 8100
    },
    {
      "dynamics_loss_avg": 0.05834991158917546,
      "epoch": 7.724761962890625e-06,
      "step": 8100
    },
    {
      "action_loss_avg": 0.009826450189575554,
      "epoch": 7.724761962890625e-06,
      "step": 8100
    },
    {
      "epoch": 7.724761962890625e-06,
      "step": 8100,
      "training_step_time": 17.09742546081543
    },
    {
      "epoch": 7.725715637207032e-06,
      "model_forward_time": 4.2743308544158936,
      "step": 8101
    },
    {
      "epoch": 7.725715637207032e-06,
      "step": 8101,
      "training_step_time": 12.80212664604187
    },
    {
      "epoch": 7.726669311523438e-06,
      "model_forward_time": 5.606801748275757,
      "step": 8102
    },
    {
      "epoch": 7.726669311523438e-06,
      "step": 8102,
      "training_step_time": 17.08534049987793
    },
    {
      "epoch": 7.727622985839844e-06,
      "model_forward_time": 5.611748695373535,
      "step": 8103
    },
    {
      "epoch": 7.727622985839844e-06,
      "step": 8103,
      "training_step_time": 17.091310739517212
    },
    {
      "epoch": 7.72857666015625e-06,
      "model_forward_time": 5.596386432647705,
      "step": 8104
    },
    {
      "epoch": 7.72857666015625e-06,
      "step": 8104,
      "training_step_time": 17.057023286819458
    },
    {
      "epoch": 7.729530334472657e-06,
      "model_forward_time": 5.974760055541992,
      "step": 8105
    },
    {
      "epoch": 7.729530334472657e-06,
      "step": 8105,
      "training_step_time": 17.418447256088257
    },
    {
      "epoch": 7.730484008789062e-06,
      "model_forward_time": 5.585316181182861,
      "step": 8106
    },
    {
      "epoch": 7.730484008789062e-06,
      "step": 8106,
      "training_step_time": 17.024866580963135
    },
    {
      "epoch": 7.731437683105468e-06,
      "model_forward_time": 5.5877978801727295,
      "step": 8107
    },
    {
      "epoch": 7.731437683105468e-06,
      "step": 8107,
      "training_step_time": 17.033892393112183
    },
    {
      "epoch": 7.732391357421876e-06,
      "model_forward_time": 5.594970226287842,
      "step": 8108
    },
    {
      "epoch": 7.732391357421876e-06,
      "step": 8108,
      "training_step_time": 17.053725242614746
    },
    {
      "epoch": 7.733345031738281e-06,
      "model_forward_time": 5.605709791183472,
      "step": 8109
    },
    {
      "epoch": 7.733345031738281e-06,
      "step": 8109,
      "training_step_time": 17.072947025299072
    },
    {
      "epoch": 7.734298706054687e-06,
      "grad_norm": 1.5481956005096436,
      "learning_rate": 1e-05,
      "loss": 0.0737,
      "step": 8110
    },
    {
      "epoch": 7.734298706054687e-06,
      "model_forward_time": 5.6149139404296875,
      "step": 8110
    },
    {
      "dynamics_loss_avg": 0.06631322875618935,
      "epoch": 7.734298706054687e-06,
      "step": 8110
    },
    {
      "action_loss_avg": 0.012490742467343807,
      "epoch": 7.734298706054687e-06,
      "step": 8110
    },
    {
      "epoch": 7.734298706054687e-06,
      "step": 8110,
      "training_step_time": 17.485881090164185
    },
    {
      "epoch": 7.735252380371094e-06,
      "model_forward_time": 5.608843088150024,
      "step": 8111
    },
    {
      "epoch": 7.735252380371094e-06,
      "step": 8111,
      "training_step_time": 17.112390756607056
    },
    {
      "epoch": 7.7362060546875e-06,
      "model_forward_time": 5.606093883514404,
      "step": 8112
    },
    {
      "epoch": 7.7362060546875e-06,
      "step": 8112,
      "training_step_time": 17.103338479995728
    },
    {
      "epoch": 7.737159729003906e-06,
      "model_forward_time": 5.611993074417114,
      "step": 8113
    },
    {
      "epoch": 7.737159729003906e-06,
      "step": 8113,
      "training_step_time": 17.104323148727417
    },
    {
      "epoch": 7.738113403320313e-06,
      "model_forward_time": 5.615405797958374,
      "step": 8114
    },
    {
      "epoch": 7.738113403320313e-06,
      "step": 8114,
      "training_step_time": 17.11481475830078
    },
    {
      "epoch": 7.739067077636719e-06,
      "model_forward_time": 5.604792833328247,
      "step": 8115
    },
    {
      "epoch": 7.739067077636719e-06,
      "step": 8115,
      "training_step_time": 17.092365503311157
    },
    {
      "epoch": 7.740020751953125e-06,
      "model_forward_time": 5.973546504974365,
      "step": 8116
    },
    {
      "epoch": 7.740020751953125e-06,
      "step": 8116,
      "training_step_time": 17.4372980594635
    },
    {
      "epoch": 7.740974426269532e-06,
      "model_forward_time": 5.602452039718628,
      "step": 8117
    },
    {
      "epoch": 7.740974426269532e-06,
      "step": 8117,
      "training_step_time": 17.084073066711426
    },
    {
      "epoch": 7.741928100585938e-06,
      "model_forward_time": 5.623672723770142,
      "step": 8118
    },
    {
      "epoch": 7.741928100585938e-06,
      "step": 8118,
      "training_step_time": 17.124751806259155
    },
    {
      "epoch": 7.742881774902344e-06,
      "model_forward_time": 5.6340155601501465,
      "step": 8119
    },
    {
      "epoch": 7.742881774902344e-06,
      "step": 8119,
      "training_step_time": 17.139882802963257
    },
    {
      "epoch": 7.74383544921875e-06,
      "grad_norm": 0.4638825058937073,
      "learning_rate": 1e-05,
      "loss": 0.088,
      "step": 8120
    },
    {
      "epoch": 7.74383544921875e-06,
      "model_forward_time": 5.621127367019653,
      "step": 8120
    },
    {
      "dynamics_loss_avg": 0.07168907355517148,
      "epoch": 7.74383544921875e-06,
      "step": 8120
    },
    {
      "action_loss_avg": 0.013535003759898245,
      "epoch": 7.74383544921875e-06,
      "step": 8120
    },
    {
      "epoch": 7.74383544921875e-06,
      "step": 8120,
      "training_step_time": 17.127089500427246
    },
    {
      "epoch": 7.744789123535157e-06,
      "model_forward_time": 5.981573104858398,
      "step": 8121
    },
    {
      "epoch": 7.744789123535157e-06,
      "step": 8121,
      "training_step_time": 17.446856260299683
    },
    {
      "epoch": 7.745742797851562e-06,
      "model_forward_time": 5.5824525356292725,
      "step": 8122
    },
    {
      "epoch": 7.745742797851562e-06,
      "step": 8122,
      "training_step_time": 17.011008977890015
    },
    {
      "epoch": 7.746696472167968e-06,
      "model_forward_time": 5.5784220695495605,
      "step": 8123
    },
    {
      "epoch": 7.746696472167968e-06,
      "step": 8123,
      "training_step_time": 17.00437593460083
    },
    {
      "epoch": 7.747650146484376e-06,
      "model_forward_time": 5.594559669494629,
      "step": 8124
    },
    {
      "epoch": 7.747650146484376e-06,
      "step": 8124,
      "training_step_time": 17.031755924224854
    },
    {
      "epoch": 7.748603820800781e-06,
      "model_forward_time": 5.576989412307739,
      "step": 8125
    },
    {
      "epoch": 7.748603820800781e-06,
      "step": 8125,
      "training_step_time": 17.026673078536987
    },
    {
      "epoch": 7.749557495117187e-06,
      "model_forward_time": 5.580475568771362,
      "step": 8126
    },
    {
      "epoch": 7.749557495117187e-06,
      "step": 8126,
      "training_step_time": 17.001893520355225
    },
    {
      "epoch": 7.750511169433594e-06,
      "model_forward_time": 5.95474100112915,
      "step": 8127
    },
    {
      "epoch": 7.750511169433594e-06,
      "step": 8127,
      "training_step_time": 17.35450553894043
    },
    {
      "epoch": 7.75146484375e-06,
      "model_forward_time": 5.58897852897644,
      "step": 8128
    },
    {
      "epoch": 7.75146484375e-06,
      "step": 8128,
      "training_step_time": 17.033508777618408
    },
    {
      "epoch": 7.752418518066406e-06,
      "model_forward_time": 5.596225261688232,
      "step": 8129
    },
    {
      "epoch": 7.752418518066406e-06,
      "step": 8129,
      "training_step_time": 17.05445122718811
    },
    {
      "epoch": 7.753372192382813e-06,
      "grad_norm": 0.826772928237915,
      "learning_rate": 1e-05,
      "loss": 0.0538,
      "step": 8130
    },
    {
      "epoch": 7.753372192382813e-06,
      "model_forward_time": 5.591668128967285,
      "step": 8130
    },
    {
      "dynamics_loss_avg": 0.04655317198485136,
      "epoch": 7.753372192382813e-06,
      "step": 8130
    },
    {
      "action_loss_avg": 0.008181737479753792,
      "epoch": 7.753372192382813e-06,
      "step": 8130
    },
    {
      "epoch": 7.753372192382813e-06,
      "step": 8130,
      "training_step_time": 17.05472731590271
    },
    {
      "epoch": 7.754325866699219e-06,
      "model_forward_time": 5.591525554656982,
      "step": 8131
    },
    {
      "epoch": 7.754325866699219e-06,
      "step": 8131,
      "training_step_time": 17.048622608184814
    },
    {
      "epoch": 7.755279541015625e-06,
      "model_forward_time": 5.965605735778809,
      "step": 8132
    },
    {
      "epoch": 7.755279541015625e-06,
      "step": 8132,
      "training_step_time": 17.43087649345398
    },
    {
      "epoch": 7.75623321533203e-06,
      "model_forward_time": 5.599805116653442,
      "step": 8133
    },
    {
      "epoch": 7.75623321533203e-06,
      "step": 8133,
      "training_step_time": 17.060693979263306
    },
    {
      "epoch": 7.757186889648438e-06,
      "model_forward_time": 5.600632429122925,
      "step": 8134
    },
    {
      "epoch": 7.757186889648438e-06,
      "step": 8134,
      "training_step_time": 17.03510570526123
    },
    {
      "epoch": 7.758140563964844e-06,
      "model_forward_time": 5.588985443115234,
      "step": 8135
    },
    {
      "epoch": 7.758140563964844e-06,
      "step": 8135,
      "training_step_time": 17.015427112579346
    },
    {
      "epoch": 7.75909423828125e-06,
      "model_forward_time": 5.588032960891724,
      "step": 8136
    },
    {
      "epoch": 7.75909423828125e-06,
      "step": 8136,
      "training_step_time": 17.021510124206543
    },
    {
      "epoch": 7.760047912597657e-06,
      "model_forward_time": 5.6032373905181885,
      "step": 8137
    },
    {
      "epoch": 7.760047912597657e-06,
      "step": 8137,
      "training_step_time": 17.439131498336792
    },
    {
      "epoch": 7.761001586914062e-06,
      "model_forward_time": 5.632972717285156,
      "step": 8138
    },
    {
      "epoch": 7.761001586914062e-06,
      "step": 8138,
      "training_step_time": 17.10114336013794
    },
    {
      "epoch": 7.761955261230468e-06,
      "model_forward_time": 5.589289426803589,
      "step": 8139
    },
    {
      "epoch": 7.761955261230468e-06,
      "step": 8139,
      "training_step_time": 17.042964696884155
    },
    {
      "epoch": 7.762908935546876e-06,
      "grad_norm": 0.9170346260070801,
      "learning_rate": 1e-05,
      "loss": 0.0729,
      "step": 8140
    },
    {
      "epoch": 7.762908935546876e-06,
      "model_forward_time": 5.5780723094940186,
      "step": 8140
    },
    {
      "dynamics_loss_avg": 0.06055812072008848,
      "epoch": 7.762908935546876e-06,
      "step": 8140
    },
    {
      "action_loss_avg": 0.009445713134482503,
      "epoch": 7.762908935546876e-06,
      "step": 8140
    },
    {
      "epoch": 7.762908935546876e-06,
      "step": 8140,
      "training_step_time": 17.028526306152344
    },
    {
      "epoch": 7.763862609863281e-06,
      "model_forward_time": 5.588698863983154,
      "step": 8141
    },
    {
      "epoch": 7.763862609863281e-06,
      "step": 8141,
      "training_step_time": 17.016093969345093
    },
    {
      "epoch": 7.764816284179687e-06,
      "model_forward_time": 5.595112085342407,
      "step": 8142
    },
    {
      "epoch": 7.764816284179687e-06,
      "step": 8142,
      "training_step_time": 17.04194402694702
    },
    {
      "epoch": 7.765769958496094e-06,
      "model_forward_time": 5.947290897369385,
      "step": 8143
    },
    {
      "epoch": 7.765769958496094e-06,
      "step": 8143,
      "training_step_time": 17.424118280410767
    },
    {
      "epoch": 7.7667236328125e-06,
      "model_forward_time": 5.611809253692627,
      "step": 8144
    },
    {
      "epoch": 7.7667236328125e-06,
      "step": 8144,
      "training_step_time": 17.066247701644897
    },
    {
      "epoch": 7.767677307128906e-06,
      "model_forward_time": 5.594402313232422,
      "step": 8145
    },
    {
      "epoch": 7.767677307128906e-06,
      "step": 8145,
      "training_step_time": 17.04015827178955
    },
    {
      "epoch": 7.768630981445313e-06,
      "model_forward_time": 5.588999032974243,
      "step": 8146
    },
    {
      "epoch": 7.768630981445313e-06,
      "step": 8146,
      "training_step_time": 17.007615089416504
    },
    {
      "epoch": 7.769584655761719e-06,
      "model_forward_time": 5.574697256088257,
      "step": 8147
    },
    {
      "epoch": 7.769584655761719e-06,
      "step": 8147,
      "training_step_time": 17.003798007965088
    },
    {
      "epoch": 7.770538330078125e-06,
      "model_forward_time": 5.95261549949646,
      "step": 8148
    },
    {
      "epoch": 7.770538330078125e-06,
      "step": 8148,
      "training_step_time": 17.393707752227783
    },
    {
      "epoch": 7.77149200439453e-06,
      "model_forward_time": 5.586930513381958,
      "step": 8149
    },
    {
      "epoch": 7.77149200439453e-06,
      "step": 8149,
      "training_step_time": 17.020017623901367
    },
    {
      "epoch": 7.772445678710938e-06,
      "grad_norm": 0.9512120485305786,
      "learning_rate": 1e-05,
      "loss": 0.066,
      "step": 8150
    },
    {
      "epoch": 7.772445678710938e-06,
      "model_forward_time": 5.5958571434021,
      "step": 8150
    },
    {
      "dynamics_loss_avg": 0.065069136954844,
      "epoch": 7.772445678710938e-06,
      "step": 8150
    },
    {
      "action_loss_avg": 0.009104555263184011,
      "epoch": 7.772445678710938e-06,
      "step": 8150
    },
    {
      "epoch": 7.772445678710938e-06,
      "step": 8150,
      "training_step_time": 17.07343816757202
    },
    {
      "epoch": 7.773399353027344e-06,
      "model_forward_time": 5.610142946243286,
      "step": 8151
    },
    {
      "epoch": 7.773399353027344e-06,
      "step": 8151,
      "training_step_time": 17.100743293762207
    },
    {
      "epoch": 7.77435302734375e-06,
      "model_forward_time": 5.599971771240234,
      "step": 8152
    },
    {
      "epoch": 7.77435302734375e-06,
      "step": 8152,
      "training_step_time": 17.069961071014404
    },
    {
      "epoch": 7.775306701660157e-06,
      "model_forward_time": 5.578306436538696,
      "step": 8153
    },
    {
      "epoch": 7.775306701660157e-06,
      "step": 8153,
      "training_step_time": 16.996341466903687
    },
    {
      "epoch": 7.776260375976562e-06,
      "model_forward_time": 5.907965660095215,
      "step": 8154
    },
    {
      "epoch": 7.776260375976562e-06,
      "step": 8154,
      "training_step_time": 17.327214002609253
    },
    {
      "epoch": 7.777214050292968e-06,
      "model_forward_time": 5.554631233215332,
      "step": 8155
    },
    {
      "epoch": 7.777214050292968e-06,
      "step": 8155,
      "training_step_time": 16.980640649795532
    },
    {
      "epoch": 7.778167724609376e-06,
      "model_forward_time": 5.565101385116577,
      "step": 8156
    },
    {
      "epoch": 7.778167724609376e-06,
      "step": 8156,
      "training_step_time": 16.987542390823364
    },
    {
      "epoch": 7.779121398925781e-06,
      "model_forward_time": 5.559596300125122,
      "step": 8157
    },
    {
      "epoch": 7.779121398925781e-06,
      "step": 8157,
      "training_step_time": 16.966394186019897
    },
    {
      "epoch": 7.780075073242187e-06,
      "model_forward_time": 5.573032855987549,
      "step": 8158
    },
    {
      "epoch": 7.780075073242187e-06,
      "step": 8158,
      "training_step_time": 16.99322485923767
    },
    {
      "epoch": 7.781028747558594e-06,
      "model_forward_time": 5.93705153465271,
      "step": 8159
    },
    {
      "epoch": 7.781028747558594e-06,
      "step": 8159,
      "training_step_time": 17.376771211624146
    },
    {
      "epoch": 7.781982421875e-06,
      "grad_norm": 1.106740951538086,
      "learning_rate": 1e-05,
      "loss": 0.0765,
      "step": 8160
    },
    {
      "epoch": 7.781982421875e-06,
      "model_forward_time": 5.57441782951355,
      "step": 8160
    },
    {
      "dynamics_loss_avg": 0.05836899671703577,
      "epoch": 7.781982421875e-06,
      "step": 8160
    },
    {
      "action_loss_avg": 0.010433627641759813,
      "epoch": 7.781982421875e-06,
      "step": 8160
    },
    {
      "epoch": 7.781982421875e-06,
      "step": 8160,
      "training_step_time": 16.982128381729126
    },
    {
      "epoch": 7.782936096191406e-06,
      "model_forward_time": 5.575986385345459,
      "step": 8161
    },
    {
      "epoch": 7.782936096191406e-06,
      "step": 8161,
      "training_step_time": 17.006422758102417
    },
    {
      "epoch": 7.783889770507813e-06,
      "model_forward_time": 5.584478378295898,
      "step": 8162
    },
    {
      "epoch": 7.783889770507813e-06,
      "step": 8162,
      "training_step_time": 17.03269910812378
    },
    {
      "epoch": 7.784843444824219e-06,
      "model_forward_time": 5.611106634140015,
      "step": 8163
    },
    {
      "epoch": 7.784843444824219e-06,
      "step": 8163,
      "training_step_time": 17.104527473449707
    },
    {
      "epoch": 7.785797119140625e-06,
      "model_forward_time": 5.625777006149292,
      "step": 8164
    },
    {
      "epoch": 7.785797119140625e-06,
      "step": 8164,
      "training_step_time": 17.130684852600098
    },
    {
      "epoch": 7.78675079345703e-06,
      "model_forward_time": 5.97210168838501,
      "step": 8165
    },
    {
      "epoch": 7.78675079345703e-06,
      "step": 8165,
      "training_step_time": 17.470181465148926
    },
    {
      "epoch": 7.787704467773438e-06,
      "model_forward_time": 5.616704940795898,
      "step": 8166
    },
    {
      "epoch": 7.787704467773438e-06,
      "step": 8166,
      "training_step_time": 17.116225242614746
    },
    {
      "epoch": 7.788658142089844e-06,
      "model_forward_time": 5.609633207321167,
      "step": 8167
    },
    {
      "epoch": 7.788658142089844e-06,
      "step": 8167,
      "training_step_time": 17.075262784957886
    },
    {
      "epoch": 7.78961181640625e-06,
      "model_forward_time": 5.589261293411255,
      "step": 8168
    },
    {
      "epoch": 7.78961181640625e-06,
      "step": 8168,
      "training_step_time": 17.006227016448975
    },
    {
      "epoch": 7.790565490722657e-06,
      "model_forward_time": 5.560815095901489,
      "step": 8169
    },
    {
      "epoch": 7.790565490722657e-06,
      "step": 8169,
      "training_step_time": 16.985976934432983
    },
    {
      "epoch": 7.791519165039062e-06,
      "grad_norm": 0.6348204612731934,
      "learning_rate": 1e-05,
      "loss": 0.0566,
      "step": 8170
    },
    {
      "epoch": 7.791519165039062e-06,
      "model_forward_time": 5.948517322540283,
      "step": 8170
    },
    {
      "dynamics_loss_avg": 0.04938683118671179,
      "epoch": 7.791519165039062e-06,
      "step": 8170
    },
    {
      "action_loss_avg": 0.009923769673332573,
      "epoch": 7.791519165039062e-06,
      "step": 8170
    },
    {
      "epoch": 7.791519165039062e-06,
      "step": 8170,
      "training_step_time": 17.374928951263428
    },
    {
      "epoch": 7.792472839355468e-06,
      "model_forward_time": 5.566954612731934,
      "step": 8171
    },
    {
      "epoch": 7.792472839355468e-06,
      "step": 8171,
      "training_step_time": 16.98288917541504
    },
    {
      "epoch": 7.793426513671876e-06,
      "model_forward_time": 5.567431926727295,
      "step": 8172
    },
    {
      "epoch": 7.793426513671876e-06,
      "step": 8172,
      "training_step_time": 16.98141598701477
    },
    {
      "epoch": 7.794380187988281e-06,
      "model_forward_time": 5.576550006866455,
      "step": 8173
    },
    {
      "epoch": 7.794380187988281e-06,
      "step": 8173,
      "training_step_time": 16.986246585845947
    },
    {
      "epoch": 7.795333862304687e-06,
      "model_forward_time": 5.576085567474365,
      "step": 8174
    },
    {
      "epoch": 7.795333862304687e-06,
      "step": 8174,
      "training_step_time": 17.018929958343506
    },
    {
      "epoch": 7.796287536621094e-06,
      "model_forward_time": 5.578447580337524,
      "step": 8175
    },
    {
      "epoch": 7.796287536621094e-06,
      "step": 8175,
      "training_step_time": 17.020332098007202
    },
    {
      "epoch": 7.7972412109375e-06,
      "model_forward_time": 5.957085132598877,
      "step": 8176
    },
    {
      "epoch": 7.7972412109375e-06,
      "step": 8176,
      "training_step_time": 17.475257635116577
    },
    {
      "epoch": 7.798194885253906e-06,
      "model_forward_time": 5.649635553359985,
      "step": 8177
    },
    {
      "epoch": 7.798194885253906e-06,
      "step": 8177,
      "training_step_time": 17.138855457305908
    },
    {
      "epoch": 7.799148559570313e-06,
      "model_forward_time": 5.605603933334351,
      "step": 8178
    },
    {
      "epoch": 7.799148559570313e-06,
      "step": 8178,
      "training_step_time": 17.05079698562622
    },
    {
      "epoch": 7.800102233886719e-06,
      "model_forward_time": 5.578738451004028,
      "step": 8179
    },
    {
      "epoch": 7.800102233886719e-06,
      "step": 8179,
      "training_step_time": 17.0001277923584
    },
    {
      "epoch": 7.801055908203125e-06,
      "grad_norm": 1.3285692930221558,
      "learning_rate": 1e-05,
      "loss": 0.0659,
      "step": 8180
    },
    {
      "epoch": 7.801055908203125e-06,
      "model_forward_time": 5.580855369567871,
      "step": 8180
    },
    {
      "dynamics_loss_avg": 0.053710109926760195,
      "epoch": 7.801055908203125e-06,
      "step": 8180
    },
    {
      "action_loss_avg": 0.00785934254527092,
      "epoch": 7.801055908203125e-06,
      "step": 8180
    },
    {
      "epoch": 7.801055908203125e-06,
      "step": 8180,
      "training_step_time": 16.992573022842407
    },
    {
      "epoch": 7.80200958251953e-06,
      "model_forward_time": 5.94934344291687,
      "step": 8181
    },
    {
      "epoch": 7.80200958251953e-06,
      "step": 8181,
      "training_step_time": 17.37557816505432
    },
    {
      "epoch": 7.802963256835938e-06,
      "model_forward_time": 5.5807201862335205,
      "step": 8182
    },
    {
      "epoch": 7.802963256835938e-06,
      "step": 8182,
      "training_step_time": 17.021986961364746
    },
    {
      "epoch": 7.803916931152344e-06,
      "model_forward_time": 5.631137371063232,
      "step": 8183
    },
    {
      "epoch": 7.803916931152344e-06,
      "step": 8183,
      "training_step_time": 17.091788291931152
    },
    {
      "epoch": 7.80487060546875e-06,
      "model_forward_time": 5.601884841918945,
      "step": 8184
    },
    {
      "epoch": 7.80487060546875e-06,
      "step": 8184,
      "training_step_time": 17.07148003578186
    },
    {
      "epoch": 7.805824279785157e-06,
      "model_forward_time": 5.604862451553345,
      "step": 8185
    },
    {
      "epoch": 7.805824279785157e-06,
      "step": 8185,
      "training_step_time": 17.081381797790527
    },
    {
      "epoch": 7.806777954101562e-06,
      "model_forward_time": 5.960920333862305,
      "step": 8186
    },
    {
      "epoch": 7.806777954101562e-06,
      "step": 8186,
      "training_step_time": 17.44231653213501
    },
    {
      "epoch": 7.807731628417968e-06,
      "model_forward_time": 5.604753255844116,
      "step": 8187
    },
    {
      "epoch": 7.807731628417968e-06,
      "step": 8187,
      "training_step_time": 17.05034589767456
    },
    {
      "epoch": 7.808685302734376e-06,
      "model_forward_time": 5.593963146209717,
      "step": 8188
    },
    {
      "epoch": 7.808685302734376e-06,
      "step": 8188,
      "training_step_time": 16.99928379058838
    },
    {
      "epoch": 7.809638977050781e-06,
      "model_forward_time": 5.583720445632935,
      "step": 8189
    },
    {
      "epoch": 7.809638977050781e-06,
      "step": 8189,
      "training_step_time": 17.026824951171875
    },
    {
      "epoch": 7.810592651367187e-06,
      "grad_norm": 0.32926368713378906,
      "learning_rate": 1e-05,
      "loss": 0.0665,
      "step": 8190
    },
    {
      "epoch": 7.810592651367187e-06,
      "model_forward_time": 5.580712795257568,
      "step": 8190
    },
    {
      "dynamics_loss_avg": 0.056750747561454776,
      "epoch": 7.810592651367187e-06,
      "step": 8190
    },
    {
      "action_loss_avg": 0.010440271790139377,
      "epoch": 7.810592651367187e-06,
      "step": 8190
    },
    {
      "epoch": 7.810592651367187e-06,
      "step": 8190,
      "training_step_time": 17.00940203666687
    },
    {
      "epoch": 7.811546325683594e-06,
      "model_forward_time": 5.58289361000061,
      "step": 8191
    },
    {
      "epoch": 7.811546325683594e-06,
      "step": 8191,
      "training_step_time": 16.99850869178772
    },
    {
      "epoch": 7.8125e-06,
      "model_forward_time": 5.962660789489746,
      "step": 8192
    },
    {
      "epoch": 7.8125e-06,
      "step": 8192,
      "training_step_time": 17.387853384017944
    },
    {
      "epoch": 7.813453674316406e-06,
      "model_forward_time": 5.591027021408081,
      "step": 8193
    },
    {
      "epoch": 7.813453674316406e-06,
      "step": 8193,
      "training_step_time": 17.04137873649597
    },
    {
      "epoch": 7.814407348632813e-06,
      "model_forward_time": 5.616621494293213,
      "step": 8194
    },
    {
      "epoch": 7.814407348632813e-06,
      "step": 8194,
      "training_step_time": 17.116692781448364
    },
    {
      "epoch": 7.815361022949219e-06,
      "model_forward_time": 5.609331130981445,
      "step": 8195
    },
    {
      "epoch": 7.815361022949219e-06,
      "step": 8195,
      "training_step_time": 17.111872911453247
    },
    {
      "epoch": 7.816314697265625e-06,
      "model_forward_time": 5.617298364639282,
      "step": 8196
    },
    {
      "epoch": 7.816314697265625e-06,
      "step": 8196,
      "training_step_time": 17.121593713760376
    },
    {
      "epoch": 7.81726837158203e-06,
      "model_forward_time": 5.618173122406006,
      "step": 8197
    },
    {
      "epoch": 7.81726837158203e-06,
      "step": 8197,
      "training_step_time": 17.463895082473755
    },
    {
      "epoch": 7.818222045898438e-06,
      "model_forward_time": 5.614698648452759,
      "step": 8198
    },
    {
      "epoch": 7.818222045898438e-06,
      "step": 8198,
      "training_step_time": 17.0937237739563
    },
    {
      "epoch": 7.819175720214844e-06,
      "model_forward_time": 5.6081743240356445,
      "step": 8199
    },
    {
      "epoch": 7.819175720214844e-06,
      "step": 8199,
      "training_step_time": 17.06454634666443
    },
    {
      "epoch": 7.82012939453125e-06,
      "grad_norm": 1.1708118915557861,
      "learning_rate": 1e-05,
      "loss": 0.0541,
      "step": 8200
    },
    {
      "epoch": 7.82012939453125e-06,
      "model_forward_time": 5.587725877761841,
      "step": 8200
    },
    {
      "dynamics_loss_avg": 0.04795197220519185,
      "epoch": 7.82012939453125e-06,
      "step": 8200
    },
    {
      "action_loss_avg": 0.007381712482310831,
      "epoch": 7.82012939453125e-06,
      "step": 8200
    },
    {
      "epoch": 7.82012939453125e-06,
      "step": 8200,
      "training_step_time": 16.99981951713562
    },
    {
      "epoch": 7.821083068847657e-06,
      "model_forward_time": 5.579980373382568,
      "step": 8201
    },
    {
      "epoch": 7.821083068847657e-06,
      "step": 8201,
      "training_step_time": 17.013506650924683
    },
    {
      "epoch": 7.822036743164062e-06,
      "model_forward_time": 5.579393625259399,
      "step": 8202
    },
    {
      "epoch": 7.822036743164062e-06,
      "step": 8202,
      "training_step_time": 17.004969358444214
    },
    {
      "epoch": 7.822990417480468e-06,
      "model_forward_time": 5.964322805404663,
      "step": 8203
    },
    {
      "epoch": 7.822990417480468e-06,
      "step": 8203,
      "training_step_time": 17.44704580307007
    },
    {
      "epoch": 7.823944091796876e-06,
      "model_forward_time": 5.60740327835083,
      "step": 8204
    },
    {
      "epoch": 7.823944091796876e-06,
      "step": 8204,
      "training_step_time": 17.109029531478882
    },
    {
      "epoch": 7.824897766113281e-06,
      "model_forward_time": 5.612265586853027,
      "step": 8205
    },
    {
      "epoch": 7.824897766113281e-06,
      "step": 8205,
      "training_step_time": 17.104892015457153
    },
    {
      "epoch": 7.825851440429687e-06,
      "model_forward_time": 5.611010551452637,
      "step": 8206
    },
    {
      "epoch": 7.825851440429687e-06,
      "step": 8206,
      "training_step_time": 17.09706449508667
    },
    {
      "epoch": 7.826805114746094e-06,
      "model_forward_time": 5.610247373580933,
      "step": 8207
    },
    {
      "epoch": 7.826805114746094e-06,
      "step": 8207,
      "training_step_time": 17.08786940574646
    },
    {
      "epoch": 7.8277587890625e-06,
      "model_forward_time": 5.603068113327026,
      "step": 8208
    },
    {
      "epoch": 7.8277587890625e-06,
      "step": 8208,
      "training_step_time": 17.412197828292847
    },
    {
      "epoch": 7.828712463378906e-06,
      "model_forward_time": 5.589227676391602,
      "step": 8209
    },
    {
      "epoch": 7.828712463378906e-06,
      "step": 8209,
      "training_step_time": 17.029993772506714
    },
    {
      "epoch": 7.829666137695313e-06,
      "grad_norm": 1.1743578910827637,
      "learning_rate": 1e-05,
      "loss": 0.0747,
      "step": 8210
    },
    {
      "epoch": 7.829666137695313e-06,
      "model_forward_time": 5.593007564544678,
      "step": 8210
    },
    {
      "dynamics_loss_avg": 0.0630555111914873,
      "epoch": 7.829666137695313e-06,
      "step": 8210
    },
    {
      "action_loss_avg": 0.012474497361108661,
      "epoch": 7.829666137695313e-06,
      "step": 8210
    },
    {
      "epoch": 7.829666137695313e-06,
      "step": 8210,
      "training_step_time": 17.051684856414795
    },
    {
      "epoch": 7.830619812011719e-06,
      "model_forward_time": 5.591163873672485,
      "step": 8211
    },
    {
      "epoch": 7.830619812011719e-06,
      "step": 8211,
      "training_step_time": 17.035956382751465
    },
    {
      "epoch": 7.831573486328125e-06,
      "model_forward_time": 5.590902328491211,
      "step": 8212
    },
    {
      "epoch": 7.831573486328125e-06,
      "step": 8212,
      "training_step_time": 17.036806106567383
    },
    {
      "epoch": 7.83252716064453e-06,
      "model_forward_time": 5.611189842224121,
      "step": 8213
    },
    {
      "epoch": 7.83252716064453e-06,
      "step": 8213,
      "training_step_time": 17.104065418243408
    },
    {
      "epoch": 7.833480834960938e-06,
      "model_forward_time": 5.986122131347656,
      "step": 8214
    },
    {
      "epoch": 7.833480834960938e-06,
      "step": 8214,
      "training_step_time": 17.47011947631836
    },
    {
      "epoch": 7.834434509277344e-06,
      "model_forward_time": 5.598222255706787,
      "step": 8215
    },
    {
      "epoch": 7.834434509277344e-06,
      "step": 8215,
      "training_step_time": 17.02380394935608
    },
    {
      "epoch": 7.83538818359375e-06,
      "model_forward_time": 5.5730507373809814,
      "step": 8216
    },
    {
      "epoch": 7.83538818359375e-06,
      "step": 8216,
      "training_step_time": 17.011319398880005
    },
    {
      "epoch": 7.836341857910157e-06,
      "model_forward_time": 5.573638677597046,
      "step": 8217
    },
    {
      "epoch": 7.836341857910157e-06,
      "step": 8217,
      "training_step_time": 17.02530837059021
    },
    {
      "epoch": 7.837295532226562e-06,
      "model_forward_time": 5.599165439605713,
      "step": 8218
    },
    {
      "epoch": 7.837295532226562e-06,
      "step": 8218,
      "training_step_time": 17.115625143051147
    },
    {
      "epoch": 7.838249206542968e-06,
      "model_forward_time": 5.983881950378418,
      "step": 8219
    },
    {
      "epoch": 7.838249206542968e-06,
      "step": 8219,
      "training_step_time": 17.49276828765869
    },
    {
      "epoch": 7.839202880859376e-06,
      "grad_norm": 0.8932891488075256,
      "learning_rate": 1e-05,
      "loss": 0.069,
      "step": 8220
    },
    {
      "epoch": 7.839202880859376e-06,
      "model_forward_time": 5.624222040176392,
      "step": 8220
    },
    {
      "dynamics_loss_avg": 0.06230276301503181,
      "epoch": 7.839202880859376e-06,
      "step": 8220
    },
    {
      "action_loss_avg": 0.00900860852561891,
      "epoch": 7.839202880859376e-06,
      "step": 8220
    },
    {
      "epoch": 7.839202880859376e-06,
      "step": 8220,
      "training_step_time": 17.12570333480835
    },
    {
      "epoch": 7.840156555175781e-06,
      "model_forward_time": 5.612406969070435,
      "step": 8221
    },
    {
      "epoch": 7.840156555175781e-06,
      "step": 8221,
      "training_step_time": 17.070659399032593
    },
    {
      "epoch": 7.841110229492187e-06,
      "model_forward_time": 5.579888582229614,
      "step": 8222
    },
    {
      "epoch": 7.841110229492187e-06,
      "step": 8222,
      "training_step_time": 17.00198221206665
    },
    {
      "epoch": 7.842063903808594e-06,
      "model_forward_time": 5.573612451553345,
      "step": 8223
    },
    {
      "epoch": 7.842063903808594e-06,
      "step": 8223,
      "training_step_time": 16.984388828277588
    },
    {
      "epoch": 7.843017578125e-06,
      "model_forward_time": 5.561034202575684,
      "step": 8224
    },
    {
      "epoch": 7.843017578125e-06,
      "step": 8224,
      "training_step_time": 16.980736255645752
    },
    {
      "epoch": 7.843971252441406e-06,
      "model_forward_time": 5.952125787734985,
      "step": 8225
    },
    {
      "epoch": 7.843971252441406e-06,
      "step": 8225,
      "training_step_time": 17.36415958404541
    },
    {
      "epoch": 7.844924926757813e-06,
      "model_forward_time": 5.562347888946533,
      "step": 8226
    },
    {
      "epoch": 7.844924926757813e-06,
      "step": 8226,
      "training_step_time": 16.9781813621521
    },
    {
      "epoch": 7.845878601074219e-06,
      "model_forward_time": 5.563865423202515,
      "step": 8227
    },
    {
      "epoch": 7.845878601074219e-06,
      "step": 8227,
      "training_step_time": 16.992366790771484
    },
    {
      "epoch": 7.846832275390625e-06,
      "model_forward_time": 5.563560247421265,
      "step": 8228
    },
    {
      "epoch": 7.846832275390625e-06,
      "step": 8228,
      "training_step_time": 16.991245985031128
    },
    {
      "epoch": 7.84778594970703e-06,
      "model_forward_time": 5.576096534729004,
      "step": 8229
    },
    {
      "epoch": 7.84778594970703e-06,
      "step": 8229,
      "training_step_time": 17.00825309753418
    },
    {
      "epoch": 7.848739624023438e-06,
      "grad_norm": 0.7188261151313782,
      "learning_rate": 1e-05,
      "loss": 0.0752,
      "step": 8230
    },
    {
      "epoch": 7.848739624023438e-06,
      "model_forward_time": 5.594162464141846,
      "step": 8230
    },
    {
      "dynamics_loss_avg": 0.06423904299736023,
      "epoch": 7.848739624023438e-06,
      "step": 8230
    },
    {
      "action_loss_avg": 0.011578673217445611,
      "epoch": 7.848739624023438e-06,
      "step": 8230
    },
    {
      "epoch": 7.848739624023438e-06,
      "step": 8230,
      "training_step_time": 17.41299271583557
    },
    {
      "epoch": 7.849693298339844e-06,
      "model_forward_time": 5.5990540981292725,
      "step": 8231
    },
    {
      "epoch": 7.849693298339844e-06,
      "step": 8231,
      "training_step_time": 17.09156036376953
    },
    {
      "epoch": 7.85064697265625e-06,
      "model_forward_time": 5.61193060874939,
      "step": 8232
    },
    {
      "epoch": 7.85064697265625e-06,
      "step": 8232,
      "training_step_time": 17.094300270080566
    },
    {
      "epoch": 7.851600646972657e-06,
      "model_forward_time": 5.610456943511963,
      "step": 8233
    },
    {
      "epoch": 7.851600646972657e-06,
      "step": 8233,
      "training_step_time": 17.112414360046387
    },
    {
      "epoch": 7.852554321289062e-06,
      "model_forward_time": 5.598863840103149,
      "step": 8234
    },
    {
      "epoch": 7.852554321289062e-06,
      "step": 8234,
      "training_step_time": 17.087061643600464
    },
    {
      "epoch": 7.853507995605468e-06,
      "model_forward_time": 5.610696792602539,
      "step": 8235
    },
    {
      "epoch": 7.853507995605468e-06,
      "step": 8235,
      "training_step_time": 17.07915735244751
    },
    {
      "epoch": 7.854461669921876e-06,
      "model_forward_time": 5.963142395019531,
      "step": 8236
    },
    {
      "epoch": 7.854461669921876e-06,
      "step": 8236,
      "training_step_time": 17.427138090133667
    },
    {
      "epoch": 7.855415344238281e-06,
      "model_forward_time": 5.59473443031311,
      "step": 8237
    },
    {
      "epoch": 7.855415344238281e-06,
      "step": 8237,
      "training_step_time": 17.046064853668213
    },
    {
      "epoch": 7.856369018554687e-06,
      "model_forward_time": 5.597476005554199,
      "step": 8238
    },
    {
      "epoch": 7.856369018554687e-06,
      "step": 8238,
      "training_step_time": 17.056375741958618
    },
    {
      "epoch": 7.857322692871094e-06,
      "model_forward_time": 5.594572305679321,
      "step": 8239
    },
    {
      "epoch": 7.857322692871094e-06,
      "step": 8239,
      "training_step_time": 17.03547954559326
    },
    {
      "epoch": 7.8582763671875e-06,
      "grad_norm": 0.7225179076194763,
      "learning_rate": 1e-05,
      "loss": 0.0798,
      "step": 8240
    },
    {
      "epoch": 7.8582763671875e-06,
      "model_forward_time": 5.588925123214722,
      "step": 8240
    },
    {
      "dynamics_loss_avg": 0.06782346591353416,
      "epoch": 7.8582763671875e-06,
      "step": 8240
    },
    {
      "action_loss_avg": 0.012369475490413607,
      "epoch": 7.8582763671875e-06,
      "step": 8240
    },
    {
      "epoch": 7.8582763671875e-06,
      "step": 8240,
      "training_step_time": 17.00851821899414
    },
    {
      "epoch": 7.859230041503906e-06,
      "model_forward_time": 5.597532272338867,
      "step": 8241
    },
    {
      "epoch": 7.859230041503906e-06,
      "step": 8241,
      "training_step_time": 17.430358171463013
    },
    {
      "epoch": 7.860183715820313e-06,
      "model_forward_time": 5.629642724990845,
      "step": 8242
    },
    {
      "epoch": 7.860183715820313e-06,
      "step": 8242,
      "training_step_time": 17.159580945968628
    },
    {
      "epoch": 7.861137390136719e-06,
      "model_forward_time": 5.622812509536743,
      "step": 8243
    },
    {
      "epoch": 7.861137390136719e-06,
      "step": 8243,
      "training_step_time": 17.159040689468384
    },
    {
      "epoch": 7.862091064453125e-06,
      "model_forward_time": 5.6170759201049805,
      "step": 8244
    },
    {
      "epoch": 7.862091064453125e-06,
      "step": 8244,
      "training_step_time": 17.097238302230835
    },
    {
      "epoch": 7.86304473876953e-06,
      "model_forward_time": 5.596502780914307,
      "step": 8245
    },
    {
      "epoch": 7.86304473876953e-06,
      "step": 8245,
      "training_step_time": 17.017828941345215
    },
    {
      "epoch": 7.863998413085938e-06,
      "model_forward_time": 5.57565975189209,
      "step": 8246
    },
    {
      "epoch": 7.863998413085938e-06,
      "step": 8246,
      "training_step_time": 17.016177654266357
    },
    {
      "epoch": 7.864952087402344e-06,
      "model_forward_time": 5.956935405731201,
      "step": 8247
    },
    {
      "epoch": 7.864952087402344e-06,
      "step": 8247,
      "training_step_time": 17.383150577545166
    },
    {
      "epoch": 7.86590576171875e-06,
      "model_forward_time": 5.572127342224121,
      "step": 8248
    },
    {
      "epoch": 7.86590576171875e-06,
      "step": 8248,
      "training_step_time": 16.981396675109863
    },
    {
      "epoch": 7.866859436035157e-06,
      "model_forward_time": 5.5920493602752686,
      "step": 8249
    },
    {
      "epoch": 7.866859436035157e-06,
      "step": 8249,
      "training_step_time": 17.04962658882141
    },
    {
      "epoch": 7.867813110351562e-06,
      "grad_norm": 0.9613591432571411,
      "learning_rate": 1e-05,
      "loss": 0.0567,
      "step": 8250
    },
    {
      "epoch": 7.867813110351562e-06,
      "model_forward_time": 5.60062575340271,
      "step": 8250
    },
    {
      "dynamics_loss_avg": 0.044156703352928164,
      "epoch": 7.867813110351562e-06,
      "step": 8250
    },
    {
      "action_loss_avg": 0.00860678981989622,
      "epoch": 7.867813110351562e-06,
      "step": 8250
    },
    {
      "epoch": 7.867813110351562e-06,
      "step": 8250,
      "training_step_time": 17.095258951187134
    },
    {
      "epoch": 7.868766784667968e-06,
      "model_forward_time": 5.624080657958984,
      "step": 8251
    },
    {
      "epoch": 7.868766784667968e-06,
      "step": 8251,
      "training_step_time": 17.159385442733765
    },
    {
      "epoch": 7.869720458984376e-06,
      "model_forward_time": 5.984254837036133,
      "step": 8252
    },
    {
      "epoch": 7.869720458984376e-06,
      "step": 8252,
      "training_step_time": 17.476502656936646
    },
    {
      "epoch": 7.870674133300781e-06,
      "model_forward_time": 5.609020709991455,
      "step": 8253
    },
    {
      "epoch": 7.870674133300781e-06,
      "step": 8253,
      "training_step_time": 17.085808038711548
    },
    {
      "epoch": 7.871627807617187e-06,
      "model_forward_time": 5.599740266799927,
      "step": 8254
    },
    {
      "epoch": 7.871627807617187e-06,
      "step": 8254,
      "training_step_time": 17.031412601470947
    },
    {
      "epoch": 7.872581481933594e-06,
      "model_forward_time": 5.572931528091431,
      "step": 8255
    },
    {
      "epoch": 7.872581481933594e-06,
      "step": 8255,
      "training_step_time": 16.99304437637329
    },
    {
      "epoch": 7.87353515625e-06,
      "model_forward_time": 5.569190740585327,
      "step": 8256
    },
    {
      "epoch": 7.87353515625e-06,
      "step": 8256,
      "training_step_time": 16.991849660873413
    },
    {
      "epoch": 7.874488830566406e-06,
      "model_forward_time": 5.572248220443726,
      "step": 8257
    },
    {
      "epoch": 7.874488830566406e-06,
      "step": 8257,
      "training_step_time": 16.97980499267578
    },
    {
      "epoch": 7.875442504882812e-06,
      "model_forward_time": 5.9433417320251465,
      "step": 8258
    },
    {
      "epoch": 7.875442504882812e-06,
      "step": 8258,
      "training_step_time": 17.39576482772827
    },
    {
      "epoch": 7.876396179199219e-06,
      "model_forward_time": 5.579678058624268,
      "step": 8259
    },
    {
      "epoch": 7.876396179199219e-06,
      "step": 8259,
      "training_step_time": 17.040287256240845
    },
    {
      "epoch": 7.877349853515625e-06,
      "grad_norm": 0.8040385842323303,
      "learning_rate": 1e-05,
      "loss": 0.0727,
      "step": 8260
    },
    {
      "epoch": 7.877349853515625e-06,
      "model_forward_time": 5.617911100387573,
      "step": 8260
    },
    {
      "dynamics_loss_avg": 0.06020093634724617,
      "epoch": 7.877349853515625e-06,
      "step": 8260
    },
    {
      "action_loss_avg": 0.013125110557302833,
      "epoch": 7.877349853515625e-06,
      "step": 8260
    },
    {
      "epoch": 7.877349853515625e-06,
      "step": 8260,
      "training_step_time": 17.119401216506958
    },
    {
      "epoch": 7.87830352783203e-06,
      "model_forward_time": 5.622182130813599,
      "step": 8261
    },
    {
      "epoch": 7.87830352783203e-06,
      "step": 8261,
      "training_step_time": 17.14008927345276
    },
    {
      "epoch": 7.879257202148438e-06,
      "model_forward_time": 5.614449977874756,
      "step": 8262
    },
    {
      "epoch": 7.879257202148438e-06,
      "step": 8262,
      "training_step_time": 17.076303482055664
    },
    {
      "epoch": 7.880210876464844e-06,
      "model_forward_time": 5.596489191055298,
      "step": 8263
    },
    {
      "epoch": 7.880210876464844e-06,
      "step": 8263,
      "training_step_time": 17.384430170059204
    },
    {
      "epoch": 7.88116455078125e-06,
      "model_forward_time": 5.566163063049316,
      "step": 8264
    },
    {
      "epoch": 7.88116455078125e-06,
      "step": 8264,
      "training_step_time": 16.965108394622803
    },
    {
      "epoch": 7.882118225097657e-06,
      "model_forward_time": 5.584137916564941,
      "step": 8265
    },
    {
      "epoch": 7.882118225097657e-06,
      "step": 8265,
      "training_step_time": 17.025407075881958
    },
    {
      "epoch": 7.883071899414063e-06,
      "model_forward_time": 5.58540415763855,
      "step": 8266
    },
    {
      "epoch": 7.883071899414063e-06,
      "step": 8266,
      "training_step_time": 17.026230573654175
    },
    {
      "epoch": 7.884025573730468e-06,
      "model_forward_time": 5.597851276397705,
      "step": 8267
    },
    {
      "epoch": 7.884025573730468e-06,
      "step": 8267,
      "training_step_time": 17.069434881210327
    },
    {
      "epoch": 7.884979248046876e-06,
      "model_forward_time": 5.617414951324463,
      "step": 8268
    },
    {
      "epoch": 7.884979248046876e-06,
      "step": 8268,
      "training_step_time": 17.118305683135986
    },
    {
      "epoch": 7.885932922363281e-06,
      "model_forward_time": 5.985128879547119,
      "step": 8269
    },
    {
      "epoch": 7.885932922363281e-06,
      "step": 8269,
      "training_step_time": 17.49634099006653
    },
    {
      "epoch": 7.886886596679687e-06,
      "grad_norm": 1.4698154926300049,
      "learning_rate": 1e-05,
      "loss": 0.0702,
      "step": 8270
    },
    {
      "epoch": 7.886886596679687e-06,
      "model_forward_time": 5.628839492797852,
      "step": 8270
    },
    {
      "dynamics_loss_avg": 0.060373281501233576,
      "epoch": 7.886886596679687e-06,
      "step": 8270
    },
    {
      "action_loss_avg": 0.010045018908567727,
      "epoch": 7.886886596679687e-06,
      "step": 8270
    },
    {
      "epoch": 7.886886596679687e-06,
      "step": 8270,
      "training_step_time": 17.139963388442993
    },
    {
      "epoch": 7.887840270996095e-06,
      "model_forward_time": 5.612564325332642,
      "step": 8271
    },
    {
      "epoch": 7.887840270996095e-06,
      "step": 8271,
      "training_step_time": 17.093478441238403
    },
    {
      "epoch": 7.8887939453125e-06,
      "model_forward_time": 5.6049089431762695,
      "step": 8272
    },
    {
      "epoch": 7.8887939453125e-06,
      "step": 8272,
      "training_step_time": 17.080395698547363
    },
    {
      "epoch": 7.889747619628906e-06,
      "model_forward_time": 5.602811813354492,
      "step": 8273
    },
    {
      "epoch": 7.889747619628906e-06,
      "step": 8273,
      "training_step_time": 17.08330750465393
    },
    {
      "epoch": 7.890701293945312e-06,
      "model_forward_time": 5.600770473480225,
      "step": 8274
    },
    {
      "epoch": 7.890701293945312e-06,
      "step": 8274,
      "training_step_time": 17.43892765045166
    },
    {
      "epoch": 7.891654968261719e-06,
      "model_forward_time": 5.597759962081909,
      "step": 8275
    },
    {
      "epoch": 7.891654968261719e-06,
      "step": 8275,
      "training_step_time": 17.067335844039917
    },
    {
      "epoch": 7.892608642578125e-06,
      "model_forward_time": 5.593720197677612,
      "step": 8276
    },
    {
      "epoch": 7.892608642578125e-06,
      "step": 8276,
      "training_step_time": 17.042445182800293
    },
    {
      "epoch": 7.89356231689453e-06,
      "model_forward_time": 5.594168424606323,
      "step": 8277
    },
    {
      "epoch": 7.89356231689453e-06,
      "step": 8277,
      "training_step_time": 17.059630870819092
    },
    {
      "epoch": 7.894515991210938e-06,
      "model_forward_time": 5.6012959480285645,
      "step": 8278
    },
    {
      "epoch": 7.894515991210938e-06,
      "step": 8278,
      "training_step_time": 17.080180406570435
    },
    {
      "epoch": 7.895469665527344e-06,
      "model_forward_time": 5.6061084270477295,
      "step": 8279
    },
    {
      "epoch": 7.895469665527344e-06,
      "step": 8279,
      "training_step_time": 17.09393310546875
    },
    {
      "epoch": 7.89642333984375e-06,
      "grad_norm": 0.8065987825393677,
      "learning_rate": 1e-05,
      "loss": 0.0684,
      "step": 8280
    },
    {
      "epoch": 7.89642333984375e-06,
      "model_forward_time": 5.977644920349121,
      "step": 8280
    },
    {
      "dynamics_loss_avg": 0.057876520417630674,
      "epoch": 7.89642333984375e-06,
      "step": 8280
    },
    {
      "action_loss_avg": 0.011195877380669118,
      "epoch": 7.89642333984375e-06,
      "step": 8280
    },
    {
      "epoch": 7.89642333984375e-06,
      "step": 8280,
      "training_step_time": 17.47812008857727
    },
    {
      "epoch": 7.897377014160157e-06,
      "model_forward_time": 5.617541790008545,
      "step": 8281
    },
    {
      "epoch": 7.897377014160157e-06,
      "step": 8281,
      "training_step_time": 17.070099353790283
    },
    {
      "epoch": 7.898330688476563e-06,
      "model_forward_time": 5.605518579483032,
      "step": 8282
    },
    {
      "epoch": 7.898330688476563e-06,
      "step": 8282,
      "training_step_time": 17.069014310836792
    },
    {
      "epoch": 7.899284362792968e-06,
      "model_forward_time": 5.598220586776733,
      "step": 8283
    },
    {
      "epoch": 7.899284362792968e-06,
      "step": 8283,
      "training_step_time": 17.039090394973755
    },
    {
      "epoch": 7.900238037109376e-06,
      "model_forward_time": 5.581039667129517,
      "step": 8284
    },
    {
      "epoch": 7.900238037109376e-06,
      "step": 8284,
      "training_step_time": 17.00399422645569
    },
    {
      "epoch": 7.901191711425781e-06,
      "model_forward_time": 5.93061375617981,
      "step": 8285
    },
    {
      "epoch": 7.901191711425781e-06,
      "step": 8285,
      "training_step_time": 17.378210306167603
    },
    {
      "epoch": 7.902145385742187e-06,
      "model_forward_time": 5.602128267288208,
      "step": 8286
    },
    {
      "epoch": 7.902145385742187e-06,
      "step": 8286,
      "training_step_time": 17.072210788726807
    },
    {
      "epoch": 7.903099060058595e-06,
      "model_forward_time": 5.6048195362091064,
      "step": 8287
    },
    {
      "epoch": 7.903099060058595e-06,
      "step": 8287,
      "training_step_time": 17.085204601287842
    },
    {
      "epoch": 7.904052734375e-06,
      "model_forward_time": 5.604615688323975,
      "step": 8288
    },
    {
      "epoch": 7.904052734375e-06,
      "step": 8288,
      "training_step_time": 17.080021858215332
    },
    {
      "epoch": 7.905006408691406e-06,
      "model_forward_time": 5.627427577972412,
      "step": 8289
    },
    {
      "epoch": 7.905006408691406e-06,
      "step": 8289,
      "training_step_time": 17.1102557182312
    },
    {
      "epoch": 7.905960083007812e-06,
      "grad_norm": 0.30275189876556396,
      "learning_rate": 1e-05,
      "loss": 0.0683,
      "step": 8290
    },
    {
      "epoch": 7.905960083007812e-06,
      "model_forward_time": 5.617265939712524,
      "step": 8290
    },
    {
      "dynamics_loss_avg": 0.06086275354027748,
      "epoch": 7.905960083007812e-06,
      "step": 8290
    },
    {
      "action_loss_avg": 0.010440583294257521,
      "epoch": 7.905960083007812e-06,
      "step": 8290
    },
    {
      "epoch": 7.905960083007812e-06,
      "step": 8290,
      "training_step_time": 17.09719181060791
    },
    {
      "epoch": 7.906913757324219e-06,
      "model_forward_time": 5.979973554611206,
      "step": 8291
    },
    {
      "epoch": 7.906913757324219e-06,
      "step": 8291,
      "training_step_time": 17.46102809906006
    },
    {
      "epoch": 7.907867431640625e-06,
      "model_forward_time": 5.598653078079224,
      "step": 8292
    },
    {
      "epoch": 7.907867431640625e-06,
      "step": 8292,
      "training_step_time": 17.04534602165222
    },
    {
      "epoch": 7.90882110595703e-06,
      "model_forward_time": 5.596668004989624,
      "step": 8293
    },
    {
      "epoch": 7.90882110595703e-06,
      "step": 8293,
      "training_step_time": 17.068431854248047
    },
    {
      "epoch": 7.909774780273438e-06,
      "model_forward_time": 5.5972394943237305,
      "step": 8294
    },
    {
      "epoch": 7.909774780273438e-06,
      "step": 8294,
      "training_step_time": 17.064185857772827
    },
    {
      "epoch": 7.910728454589844e-06,
      "model_forward_time": 5.59778094291687,
      "step": 8295
    },
    {
      "epoch": 7.910728454589844e-06,
      "step": 8295,
      "training_step_time": 17.051920175552368
    },
    {
      "epoch": 7.91168212890625e-06,
      "model_forward_time": 5.585808277130127,
      "step": 8296
    },
    {
      "epoch": 7.91168212890625e-06,
      "step": 8296,
      "training_step_time": 17.387957334518433
    },
    {
      "epoch": 7.912635803222657e-06,
      "model_forward_time": 5.6084020137786865,
      "step": 8297
    },
    {
      "epoch": 7.912635803222657e-06,
      "step": 8297,
      "training_step_time": 17.042964220046997
    },
    {
      "epoch": 7.913589477539063e-06,
      "model_forward_time": 5.583986520767212,
      "step": 8298
    },
    {
      "epoch": 7.913589477539063e-06,
      "step": 8298,
      "training_step_time": 17.01515007019043
    },
    {
      "epoch": 7.914543151855468e-06,
      "model_forward_time": 5.593075752258301,
      "step": 8299
    },
    {
      "epoch": 7.914543151855468e-06,
      "step": 8299,
      "training_step_time": 17.05774474143982
    },
    {
      "epoch": 7.915496826171876e-06,
      "grad_norm": 1.0066136121749878,
      "learning_rate": 1e-05,
      "loss": 0.0769,
      "step": 8300
    },
    {
      "epoch": 7.915496826171876e-06,
      "model_forward_time": 5.613150596618652,
      "step": 8300
    },
    {
      "dynamics_loss_avg": 0.06234748158603907,
      "epoch": 7.915496826171876e-06,
      "step": 8300
    },
    {
      "action_loss_avg": 0.011479074927046895,
      "epoch": 7.915496826171876e-06,
      "step": 8300
    },
    {
      "epoch": 7.915496826171876e-06,
      "step": 8300,
      "training_step_time": 17.12575078010559
    },
    {
      "epoch": 7.916450500488281e-06,
      "model_forward_time": 5.626270532608032,
      "step": 8301
    },
    {
      "epoch": 7.916450500488281e-06,
      "step": 8301,
      "training_step_time": 17.142276525497437
    },
    {
      "epoch": 7.917404174804687e-06,
      "model_forward_time": 5.996501684188843,
      "step": 8302
    },
    {
      "epoch": 7.917404174804687e-06,
      "step": 8302,
      "training_step_time": 17.49476957321167
    },
    {
      "epoch": 7.918357849121095e-06,
      "model_forward_time": 5.6107258796691895,
      "step": 8303
    },
    {
      "epoch": 7.918357849121095e-06,
      "step": 8303,
      "training_step_time": 17.07518196105957
    },
    {
      "epoch": 7.9193115234375e-06,
      "model_forward_time": 5.590691089630127,
      "step": 8304
    },
    {
      "epoch": 7.9193115234375e-06,
      "step": 8304,
      "training_step_time": 17.035627365112305
    },
    {
      "epoch": 7.920265197753906e-06,
      "model_forward_time": 5.58196759223938,
      "step": 8305
    },
    {
      "epoch": 7.920265197753906e-06,
      "step": 8305,
      "training_step_time": 17.006958961486816
    },
    {
      "epoch": 7.921218872070312e-06,
      "model_forward_time": 5.581523895263672,
      "step": 8306
    },
    {
      "epoch": 7.921218872070312e-06,
      "step": 8306,
      "training_step_time": 17.0002019405365
    },
    {
      "epoch": 7.922172546386719e-06,
      "model_forward_time": 5.585749864578247,
      "step": 8307
    },
    {
      "epoch": 7.922172546386719e-06,
      "step": 8307,
      "training_step_time": 17.39773392677307
    },
    {
      "epoch": 7.923126220703125e-06,
      "model_forward_time": 5.601080656051636,
      "step": 8308
    },
    {
      "epoch": 7.923126220703125e-06,
      "step": 8308,
      "training_step_time": 17.071593523025513
    },
    {
      "epoch": 7.92407989501953e-06,
      "model_forward_time": 5.5957183837890625,
      "step": 8309
    },
    {
      "epoch": 7.92407989501953e-06,
      "step": 8309,
      "training_step_time": 17.079217433929443
    },
    {
      "epoch": 7.925033569335938e-06,
      "grad_norm": 0.8804342746734619,
      "learning_rate": 1e-05,
      "loss": 0.0616,
      "step": 8310
    },
    {
      "epoch": 7.925033569335938e-06,
      "model_forward_time": 5.603836536407471,
      "step": 8310
    },
    {
      "dynamics_loss_avg": 0.053359837643802165,
      "epoch": 7.925033569335938e-06,
      "step": 8310
    },
    {
      "action_loss_avg": 0.008541268424596637,
      "epoch": 7.925033569335938e-06,
      "step": 8310
    },
    {
      "epoch": 7.925033569335938e-06,
      "step": 8310,
      "training_step_time": 17.087929010391235
    },
    {
      "epoch": 7.925987243652344e-06,
      "model_forward_time": 5.606371164321899,
      "step": 8311
    },
    {
      "epoch": 7.925987243652344e-06,
      "step": 8311,
      "training_step_time": 17.074800968170166
    },
    {
      "epoch": 7.92694091796875e-06,
      "model_forward_time": 5.601670265197754,
      "step": 8312
    },
    {
      "epoch": 7.92694091796875e-06,
      "step": 8312,
      "training_step_time": 17.08101987838745
    },
    {
      "epoch": 7.927894592285157e-06,
      "model_forward_time": 5.972275972366333,
      "step": 8313
    },
    {
      "epoch": 7.927894592285157e-06,
      "step": 8313,
      "training_step_time": 17.46262001991272
    },
    {
      "epoch": 7.928848266601563e-06,
      "model_forward_time": 5.608745574951172,
      "step": 8314
    },
    {
      "epoch": 7.928848266601563e-06,
      "step": 8314,
      "training_step_time": 17.098515510559082
    },
    {
      "epoch": 7.929801940917968e-06,
      "model_forward_time": 5.609376430511475,
      "step": 8315
    },
    {
      "epoch": 7.929801940917968e-06,
      "step": 8315,
      "training_step_time": 17.08350968360901
    },
    {
      "epoch": 7.930755615234376e-06,
      "model_forward_time": 5.612585544586182,
      "step": 8316
    },
    {
      "epoch": 7.930755615234376e-06,
      "step": 8316,
      "training_step_time": 17.098804473876953
    },
    {
      "epoch": 7.931709289550781e-06,
      "model_forward_time": 5.614369869232178,
      "step": 8317
    },
    {
      "epoch": 7.931709289550781e-06,
      "step": 8317,
      "training_step_time": 17.106584787368774
    },
    {
      "epoch": 7.932662963867187e-06,
      "model_forward_time": 5.973817586898804,
      "step": 8318
    },
    {
      "epoch": 7.932662963867187e-06,
      "step": 8318,
      "training_step_time": 17.43826198577881
    },
    {
      "epoch": 7.933616638183595e-06,
      "model_forward_time": 5.584482431411743,
      "step": 8319
    },
    {
      "epoch": 7.933616638183595e-06,
      "step": 8319,
      "training_step_time": 17.014775037765503
    },
    {
      "epoch": 7.9345703125e-06,
      "grad_norm": 1.5023314952850342,
      "learning_rate": 1e-05,
      "loss": 0.0717,
      "step": 8320
    },
    {
      "epoch": 7.9345703125e-06,
      "model_forward_time": 5.583481550216675,
      "step": 8320
    },
    {
      "dynamics_loss_avg": 0.057570628449320795,
      "epoch": 7.9345703125e-06,
      "step": 8320
    },
    {
      "action_loss_avg": 0.010005722753703594,
      "epoch": 7.9345703125e-06,
      "step": 8320
    },
    {
      "epoch": 7.9345703125e-06,
      "step": 8320,
      "training_step_time": 17.00846767425537
    },
    {
      "epoch": 7.935523986816406e-06,
      "model_forward_time": 5.598204851150513,
      "step": 8321
    },
    {
      "epoch": 7.935523986816406e-06,
      "step": 8321,
      "training_step_time": 17.09027647972107
    },
    {
      "epoch": 7.936477661132812e-06,
      "model_forward_time": 5.611008405685425,
      "step": 8322
    },
    {
      "epoch": 7.936477661132812e-06,
      "step": 8322,
      "training_step_time": 17.08579158782959
    },
    {
      "epoch": 7.937431335449219e-06,
      "model_forward_time": 5.599212408065796,
      "step": 8323
    },
    {
      "epoch": 7.937431335449219e-06,
      "step": 8323,
      "training_step_time": 17.047314643859863
    },
    {
      "epoch": 7.938385009765625e-06,
      "model_forward_time": 5.940540790557861,
      "step": 8324
    },
    {
      "epoch": 7.938385009765625e-06,
      "step": 8324,
      "training_step_time": 17.365618467330933
    },
    {
      "epoch": 7.93933868408203e-06,
      "model_forward_time": 5.581422328948975,
      "step": 8325
    },
    {
      "epoch": 7.93933868408203e-06,
      "step": 8325,
      "training_step_time": 16.999077320098877
    },
    {
      "epoch": 7.940292358398438e-06,
      "model_forward_time": 5.577738523483276,
      "step": 8326
    },
    {
      "epoch": 7.940292358398438e-06,
      "step": 8326,
      "training_step_time": 17.01304006576538
    },
    {
      "epoch": 7.941246032714844e-06,
      "model_forward_time": 5.589326858520508,
      "step": 8327
    },
    {
      "epoch": 7.941246032714844e-06,
      "step": 8327,
      "training_step_time": 17.060463666915894
    },
    {
      "epoch": 7.94219970703125e-06,
      "model_forward_time": 5.605817079544067,
      "step": 8328
    },
    {
      "epoch": 7.94219970703125e-06,
      "step": 8328,
      "training_step_time": 17.087228536605835
    },
    {
      "epoch": 7.943153381347657e-06,
      "model_forward_time": 5.61464262008667,
      "step": 8329
    },
    {
      "epoch": 7.943153381347657e-06,
      "step": 8329,
      "training_step_time": 17.457674026489258
    },
    {
      "epoch": 7.944107055664063e-06,
      "grad_norm": 1.1006523370742798,
      "learning_rate": 1e-05,
      "loss": 0.0498,
      "step": 8330
    },
    {
      "epoch": 7.944107055664063e-06,
      "model_forward_time": 5.58447003364563,
      "step": 8330
    },
    {
      "dynamics_loss_avg": 0.04891997333616018,
      "epoch": 7.944107055664063e-06,
      "step": 8330
    },
    {
      "action_loss_avg": 0.008023186516948045,
      "epoch": 7.944107055664063e-06,
      "step": 8330
    },
    {
      "epoch": 7.944107055664063e-06,
      "step": 8330,
      "training_step_time": 17.01987361907959
    },
    {
      "epoch": 7.945060729980468e-06,
      "model_forward_time": 5.578001499176025,
      "step": 8331
    },
    {
      "epoch": 7.945060729980468e-06,
      "step": 8331,
      "training_step_time": 17.00204825401306
    },
    {
      "epoch": 7.946014404296876e-06,
      "model_forward_time": 5.564006805419922,
      "step": 8332
    },
    {
      "epoch": 7.946014404296876e-06,
      "step": 8332,
      "training_step_time": 16.979039430618286
    },
    {
      "epoch": 7.946968078613281e-06,
      "model_forward_time": 5.566503286361694,
      "step": 8333
    },
    {
      "epoch": 7.946968078613281e-06,
      "step": 8333,
      "training_step_time": 16.978220462799072
    },
    {
      "epoch": 7.947921752929687e-06,
      "model_forward_time": 5.578896522521973,
      "step": 8334
    },
    {
      "epoch": 7.947921752929687e-06,
      "step": 8334,
      "training_step_time": 17.030439853668213
    },
    {
      "epoch": 7.948875427246095e-06,
      "model_forward_time": 5.956123352050781,
      "step": 8335
    },
    {
      "epoch": 7.948875427246095e-06,
      "step": 8335,
      "training_step_time": 17.38548231124878
    },
    {
      "epoch": 7.9498291015625e-06,
      "model_forward_time": 5.590349912643433,
      "step": 8336
    },
    {
      "epoch": 7.9498291015625e-06,
      "step": 8336,
      "training_step_time": 16.9904727935791
    },
    {
      "epoch": 7.950782775878906e-06,
      "model_forward_time": 5.584352493286133,
      "step": 8337
    },
    {
      "epoch": 7.950782775878906e-06,
      "step": 8337,
      "training_step_time": 17.02966833114624
    },
    {
      "epoch": 7.951736450195312e-06,
      "model_forward_time": 5.588663578033447,
      "step": 8338
    },
    {
      "epoch": 7.951736450195312e-06,
      "step": 8338,
      "training_step_time": 17.03542947769165
    },
    {
      "epoch": 7.952690124511719e-06,
      "model_forward_time": 5.599080801010132,
      "step": 8339
    },
    {
      "epoch": 7.952690124511719e-06,
      "step": 8339,
      "training_step_time": 17.084786891937256
    },
    {
      "epoch": 7.953643798828125e-06,
      "grad_norm": 0.4671964943408966,
      "learning_rate": 1e-05,
      "loss": 0.0777,
      "step": 8340
    },
    {
      "epoch": 7.953643798828125e-06,
      "model_forward_time": 5.611628293991089,
      "step": 8340
    },
    {
      "dynamics_loss_avg": 0.06087197074666619,
      "epoch": 7.953643798828125e-06,
      "step": 8340
    },
    {
      "action_loss_avg": 0.011618451355025173,
      "epoch": 7.953643798828125e-06,
      "step": 8340
    },
    {
      "epoch": 7.953643798828125e-06,
      "step": 8340,
      "training_step_time": 17.453277587890625
    },
    {
      "epoch": 7.95459747314453e-06,
      "model_forward_time": 5.599093675613403,
      "step": 8341
    },
    {
      "epoch": 7.95459747314453e-06,
      "step": 8341,
      "training_step_time": 17.022599935531616
    },
    {
      "epoch": 7.955551147460938e-06,
      "model_forward_time": 5.586301326751709,
      "step": 8342
    },
    {
      "epoch": 7.955551147460938e-06,
      "step": 8342,
      "training_step_time": 17.020040035247803
    },
    {
      "epoch": 7.956504821777344e-06,
      "model_forward_time": 5.596285104751587,
      "step": 8343
    },
    {
      "epoch": 7.956504821777344e-06,
      "step": 8343,
      "training_step_time": 17.040796279907227
    },
    {
      "epoch": 7.95745849609375e-06,
      "model_forward_time": 5.598129749298096,
      "step": 8344
    },
    {
      "epoch": 7.95745849609375e-06,
      "step": 8344,
      "training_step_time": 17.060730695724487
    },
    {
      "epoch": 7.958412170410157e-06,
      "model_forward_time": 5.6004087924957275,
      "step": 8345
    },
    {
      "epoch": 7.958412170410157e-06,
      "step": 8345,
      "training_step_time": 17.06774139404297
    },
    {
      "epoch": 7.959365844726563e-06,
      "model_forward_time": 5.985754013061523,
      "step": 8346
    },
    {
      "epoch": 7.959365844726563e-06,
      "step": 8346,
      "training_step_time": 17.44940757751465
    },
    {
      "epoch": 7.960319519042968e-06,
      "model_forward_time": 5.593164443969727,
      "step": 8347
    },
    {
      "epoch": 7.960319519042968e-06,
      "step": 8347,
      "training_step_time": 17.061950206756592
    },
    {
      "epoch": 7.961273193359376e-06,
      "model_forward_time": 5.602618455886841,
      "step": 8348
    },
    {
      "epoch": 7.961273193359376e-06,
      "step": 8348,
      "training_step_time": 17.072339057922363
    },
    {
      "epoch": 7.962226867675781e-06,
      "model_forward_time": 5.602761268615723,
      "step": 8349
    },
    {
      "epoch": 7.962226867675781e-06,
      "step": 8349,
      "training_step_time": 17.0493221282959
    },
    {
      "epoch": 7.963180541992187e-06,
      "grad_norm": 0.5078209042549133,
      "learning_rate": 1e-05,
      "loss": 0.0723,
      "step": 8350
    },
    {
      "epoch": 7.963180541992187e-06,
      "model_forward_time": 5.590482711791992,
      "step": 8350
    },
    {
      "dynamics_loss_avg": 0.05973908007144928,
      "epoch": 7.963180541992187e-06,
      "step": 8350
    },
    {
      "action_loss_avg": 0.010614189924672246,
      "epoch": 7.963180541992187e-06,
      "step": 8350
    },
    {
      "epoch": 7.963180541992187e-06,
      "step": 8350,
      "training_step_time": 17.03546905517578
    },
    {
      "epoch": 7.964134216308595e-06,
      "model_forward_time": 5.939187526702881,
      "step": 8351
    },
    {
      "epoch": 7.964134216308595e-06,
      "step": 8351,
      "training_step_time": 17.40054440498352
    },
    {
      "epoch": 7.965087890625e-06,
      "model_forward_time": 5.597941875457764,
      "step": 8352
    },
    {
      "epoch": 7.965087890625e-06,
      "step": 8352,
      "training_step_time": 17.088078022003174
    },
    {
      "epoch": 7.966041564941406e-06,
      "model_forward_time": 5.619579553604126,
      "step": 8353
    },
    {
      "epoch": 7.966041564941406e-06,
      "step": 8353,
      "training_step_time": 17.134308099746704
    },
    {
      "epoch": 7.966995239257812e-06,
      "model_forward_time": 5.617721796035767,
      "step": 8354
    },
    {
      "epoch": 7.966995239257812e-06,
      "step": 8354,
      "training_step_time": 17.1166672706604
    },
    {
      "epoch": 7.96794891357422e-06,
      "model_forward_time": 5.607809782028198,
      "step": 8355
    },
    {
      "epoch": 7.96794891357422e-06,
      "step": 8355,
      "training_step_time": 17.07789421081543
    },
    {
      "epoch": 7.968902587890625e-06,
      "model_forward_time": 5.587965250015259,
      "step": 8356
    },
    {
      "epoch": 7.968902587890625e-06,
      "step": 8356,
      "training_step_time": 17.00492763519287
    },
    {
      "epoch": 7.96985626220703e-06,
      "model_forward_time": 5.955792188644409,
      "step": 8357
    },
    {
      "epoch": 7.96985626220703e-06,
      "step": 8357,
      "training_step_time": 17.387598514556885
    },
    {
      "epoch": 7.970809936523438e-06,
      "model_forward_time": 5.561527490615845,
      "step": 8358
    },
    {
      "epoch": 7.970809936523438e-06,
      "step": 8358,
      "training_step_time": 16.974467277526855
    },
    {
      "epoch": 7.971763610839844e-06,
      "model_forward_time": 5.561120271682739,
      "step": 8359
    },
    {
      "epoch": 7.971763610839844e-06,
      "step": 8359,
      "training_step_time": 16.969850540161133
    },
    {
      "epoch": 7.97271728515625e-06,
      "grad_norm": 1.2895138263702393,
      "learning_rate": 1e-05,
      "loss": 0.0561,
      "step": 8360
    },
    {
      "epoch": 7.97271728515625e-06,
      "model_forward_time": 5.559123516082764,
      "step": 8360
    },
    {
      "dynamics_loss_avg": 0.04619602169841528,
      "epoch": 7.97271728515625e-06,
      "step": 8360
    },
    {
      "action_loss_avg": 0.008326399908401071,
      "epoch": 7.97271728515625e-06,
      "step": 8360
    },
    {
      "epoch": 7.97271728515625e-06,
      "step": 8360,
      "training_step_time": 16.978978872299194
    },
    {
      "epoch": 7.973670959472657e-06,
      "model_forward_time": 5.574607610702515,
      "step": 8361
    },
    {
      "epoch": 7.973670959472657e-06,
      "step": 8361,
      "training_step_time": 17.016529321670532
    },
    {
      "epoch": 7.974624633789063e-06,
      "model_forward_time": 5.581936359405518,
      "step": 8362
    },
    {
      "epoch": 7.974624633789063e-06,
      "step": 8362,
      "training_step_time": 17.34830141067505
    },
    {
      "epoch": 7.975578308105468e-06,
      "model_forward_time": 5.589223623275757,
      "step": 8363
    },
    {
      "epoch": 7.975578308105468e-06,
      "step": 8363,
      "training_step_time": 17.042782068252563
    },
    {
      "epoch": 7.976531982421876e-06,
      "model_forward_time": 5.605666875839233,
      "step": 8364
    },
    {
      "epoch": 7.976531982421876e-06,
      "step": 8364,
      "training_step_time": 17.085723638534546
    },
    {
      "epoch": 7.977485656738281e-06,
      "model_forward_time": 5.616479158401489,
      "step": 8365
    },
    {
      "epoch": 7.977485656738281e-06,
      "step": 8365,
      "training_step_time": 17.100266695022583
    },
    {
      "epoch": 7.978439331054687e-06,
      "model_forward_time": 5.634737014770508,
      "step": 8366
    },
    {
      "epoch": 7.978439331054687e-06,
      "step": 8366,
      "training_step_time": 17.087421655654907
    },
    {
      "epoch": 7.979393005371095e-06,
      "model_forward_time": 5.5838611125946045,
      "step": 8367
    },
    {
      "epoch": 7.979393005371095e-06,
      "step": 8367,
      "training_step_time": 17.015324592590332
    },
    {
      "epoch": 7.9803466796875e-06,
      "model_forward_time": 5.950690984725952,
      "step": 8368
    },
    {
      "epoch": 7.9803466796875e-06,
      "step": 8368,
      "training_step_time": 17.38145613670349
    },
    {
      "epoch": 7.981300354003906e-06,
      "model_forward_time": 5.625365972518921,
      "step": 8369
    },
    {
      "epoch": 7.981300354003906e-06,
      "step": 8369,
      "training_step_time": 17.052705764770508
    },
    {
      "epoch": 7.982254028320312e-06,
      "grad_norm": 0.9298490881919861,
      "learning_rate": 1e-05,
      "loss": 0.0568,
      "step": 8370
    },
    {
      "epoch": 7.982254028320312e-06,
      "model_forward_time": 5.593493223190308,
      "step": 8370
    },
    {
      "dynamics_loss_avg": 0.05391602776944637,
      "epoch": 7.982254028320312e-06,
      "step": 8370
    },
    {
      "action_loss_avg": 0.007561678579077124,
      "epoch": 7.982254028320312e-06,
      "step": 8370
    },
    {
      "epoch": 7.982254028320312e-06,
      "step": 8370,
      "training_step_time": 17.040037870407104
    },
    {
      "epoch": 7.98320770263672e-06,
      "model_forward_time": 5.597781181335449,
      "step": 8371
    },
    {
      "epoch": 7.98320770263672e-06,
      "step": 8371,
      "training_step_time": 17.056865453720093
    },
    {
      "epoch": 7.984161376953125e-06,
      "model_forward_time": 5.610166072845459,
      "step": 8372
    },
    {
      "epoch": 7.984161376953125e-06,
      "step": 8372,
      "training_step_time": 17.103670835494995
    },
    {
      "epoch": 7.98511505126953e-06,
      "model_forward_time": 5.609650135040283,
      "step": 8373
    },
    {
      "epoch": 7.98511505126953e-06,
      "step": 8373,
      "training_step_time": 17.48451280593872
    },
    {
      "epoch": 7.986068725585938e-06,
      "model_forward_time": 5.615097522735596,
      "step": 8374
    },
    {
      "epoch": 7.986068725585938e-06,
      "step": 8374,
      "training_step_time": 17.0865581035614
    },
    {
      "epoch": 7.987022399902344e-06,
      "model_forward_time": 5.5925211906433105,
      "step": 8375
    },
    {
      "epoch": 7.987022399902344e-06,
      "step": 8375,
      "training_step_time": 17.029491901397705
    },
    {
      "epoch": 7.98797607421875e-06,
      "model_forward_time": 5.584895133972168,
      "step": 8376
    },
    {
      "epoch": 7.98797607421875e-06,
      "step": 8376,
      "training_step_time": 17.035014152526855
    },
    {
      "epoch": 7.988929748535157e-06,
      "model_forward_time": 5.577125310897827,
      "step": 8377
    },
    {
      "epoch": 7.988929748535157e-06,
      "step": 8377,
      "training_step_time": 17.004632472991943
    },
    {
      "epoch": 7.989883422851563e-06,
      "model_forward_time": 5.59844970703125,
      "step": 8378
    },
    {
      "epoch": 7.989883422851563e-06,
      "step": 8378,
      "training_step_time": 17.066161155700684
    },
    {
      "epoch": 7.990837097167968e-06,
      "model_forward_time": 5.966115474700928,
      "step": 8379
    },
    {
      "epoch": 7.990837097167968e-06,
      "step": 8379,
      "training_step_time": 17.427541494369507
    },
    {
      "epoch": 7.991790771484376e-06,
      "grad_norm": 1.5820108652114868,
      "learning_rate": 1e-05,
      "loss": 0.0695,
      "step": 8380
    },
    {
      "epoch": 7.991790771484376e-06,
      "model_forward_time": 5.598662614822388,
      "step": 8380
    },
    {
      "dynamics_loss_avg": 0.0588196137920022,
      "epoch": 7.991790771484376e-06,
      "step": 8380
    },
    {
      "action_loss_avg": 0.009380248840898275,
      "epoch": 7.991790771484376e-06,
      "step": 8380
    },
    {
      "epoch": 7.991790771484376e-06,
      "step": 8380,
      "training_step_time": 17.06463098526001
    },
    {
      "epoch": 7.992744445800781e-06,
      "model_forward_time": 5.607178211212158,
      "step": 8381
    },
    {
      "epoch": 7.992744445800781e-06,
      "step": 8381,
      "training_step_time": 17.067901611328125
    },
    {
      "epoch": 7.993698120117187e-06,
      "model_forward_time": 5.596544027328491,
      "step": 8382
    },
    {
      "epoch": 7.993698120117187e-06,
      "step": 8382,
      "training_step_time": 17.03780770301819
    },
    {
      "epoch": 7.994651794433593e-06,
      "model_forward_time": 5.591089725494385,
      "step": 8383
    },
    {
      "epoch": 7.994651794433593e-06,
      "step": 8383,
      "training_step_time": 17.033637046813965
    },
    {
      "epoch": 7.99560546875e-06,
      "model_forward_time": 5.946422338485718,
      "step": 8384
    },
    {
      "epoch": 7.99560546875e-06,
      "step": 8384,
      "training_step_time": 17.38994336128235
    },
    {
      "epoch": 7.996559143066406e-06,
      "model_forward_time": 5.596189498901367,
      "step": 8385
    },
    {
      "epoch": 7.996559143066406e-06,
      "step": 8385,
      "training_step_time": 17.06822156906128
    },
    {
      "epoch": 7.997512817382812e-06,
      "model_forward_time": 5.601423263549805,
      "step": 8386
    },
    {
      "epoch": 7.997512817382812e-06,
      "step": 8386,
      "training_step_time": 17.081239938735962
    },
    {
      "epoch": 7.99846649169922e-06,
      "model_forward_time": 5.59745192527771,
      "step": 8387
    },
    {
      "epoch": 7.99846649169922e-06,
      "step": 8387,
      "training_step_time": 17.060285806655884
    },
    {
      "epoch": 7.999420166015625e-06,
      "model_forward_time": 5.604433536529541,
      "step": 8388
    },
    {
      "epoch": 7.999420166015625e-06,
      "step": 8388,
      "training_step_time": 17.093390226364136
    },
    {
      "epoch": 8.00037384033203e-06,
      "model_forward_time": 5.616566896438599,
      "step": 8389
    },
    {
      "epoch": 8.00037384033203e-06,
      "step": 8389,
      "training_step_time": 17.126914024353027
    },
    {
      "epoch": 8.001327514648438e-06,
      "grad_norm": 1.185515284538269,
      "learning_rate": 1e-05,
      "loss": 0.06,
      "step": 8390
    },
    {
      "epoch": 8.001327514648438e-06,
      "model_forward_time": 6.003487825393677,
      "step": 8390
    },
    {
      "dynamics_loss_avg": 0.052923839446157216,
      "epoch": 8.001327514648438e-06,
      "step": 8390
    },
    {
      "action_loss_avg": 0.009072203002870084,
      "epoch": 8.001327514648438e-06,
      "step": 8390
    },
    {
      "epoch": 8.001327514648438e-06,
      "step": 8390,
      "training_step_time": 17.484384536743164
    },
    {
      "epoch": 8.002281188964844e-06,
      "model_forward_time": 5.600939035415649,
      "step": 8391
    },
    {
      "epoch": 8.002281188964844e-06,
      "step": 8391,
      "training_step_time": 17.04727005958557
    },
    {
      "epoch": 8.00323486328125e-06,
      "model_forward_time": 5.583757638931274,
      "step": 8392
    },
    {
      "epoch": 8.00323486328125e-06,
      "step": 8392,
      "training_step_time": 17.01191735267639
    },
    {
      "epoch": 8.004188537597657e-06,
      "model_forward_time": 5.56677770614624,
      "step": 8393
    },
    {
      "epoch": 8.004188537597657e-06,
      "step": 8393,
      "training_step_time": 16.982835054397583
    },
    {
      "epoch": 8.005142211914063e-06,
      "model_forward_time": 5.56873631477356,
      "step": 8394
    },
    {
      "epoch": 8.005142211914063e-06,
      "step": 8394,
      "training_step_time": 16.972347259521484
    },
    {
      "epoch": 8.006095886230468e-06,
      "model_forward_time": 5.566728353500366,
      "step": 8395
    },
    {
      "epoch": 8.006095886230468e-06,
      "step": 8395,
      "training_step_time": 17.35879611968994
    },
    {
      "epoch": 8.007049560546876e-06,
      "model_forward_time": 5.570227146148682,
      "step": 8396
    },
    {
      "epoch": 8.007049560546876e-06,
      "step": 8396,
      "training_step_time": 16.989691257476807
    },
    {
      "epoch": 8.008003234863281e-06,
      "model_forward_time": 5.579218149185181,
      "step": 8397
    },
    {
      "epoch": 8.008003234863281e-06,
      "step": 8397,
      "training_step_time": 17.01203751564026
    },
    {
      "epoch": 8.008956909179687e-06,
      "model_forward_time": 5.592666149139404,
      "step": 8398
    },
    {
      "epoch": 8.008956909179687e-06,
      "step": 8398,
      "training_step_time": 17.058448314666748
    },
    {
      "epoch": 8.009910583496093e-06,
      "model_forward_time": 5.606470823287964,
      "step": 8399
    },
    {
      "epoch": 8.009910583496093e-06,
      "step": 8399,
      "training_step_time": 17.096393585205078
    },
    {
      "epoch": 8.0108642578125e-06,
      "grad_norm": 0.841385543346405,
      "learning_rate": 1e-05,
      "loss": 0.0607,
      "step": 8400
    },
    {
      "epoch": 8.0108642578125e-06,
      "model_forward_time": 5.618156909942627,
      "step": 8400
    },
    {
      "dynamics_loss_avg": 0.05000484716147184,
      "epoch": 8.0108642578125e-06,
      "step": 8400
    },
    {
      "action_loss_avg": 0.007002699794247747,
      "epoch": 8.0108642578125e-06,
      "step": 8400
    },
    {
      "epoch": 8.0108642578125e-06,
      "step": 8400,
      "training_step_time": 17.133620738983154
    },
    {
      "epoch": 8.011817932128906e-06,
      "model_forward_time": 5.992538213729858,
      "step": 8401
    },
    {
      "epoch": 8.011817932128906e-06,
      "step": 8401,
      "training_step_time": 17.467573165893555
    },
    {
      "epoch": 8.012771606445312e-06,
      "model_forward_time": 5.598139047622681,
      "step": 8402
    },
    {
      "epoch": 8.012771606445312e-06,
      "step": 8402,
      "training_step_time": 17.04549813270569
    },
    {
      "epoch": 8.01372528076172e-06,
      "model_forward_time": 5.573361873626709,
      "step": 8403
    },
    {
      "epoch": 8.01372528076172e-06,
      "step": 8403,
      "training_step_time": 17.00779390335083
    },
    {
      "epoch": 8.014678955078125e-06,
      "model_forward_time": 5.569549560546875,
      "step": 8404
    },
    {
      "epoch": 8.014678955078125e-06,
      "step": 8404,
      "training_step_time": 16.979491710662842
    },
    {
      "epoch": 8.01563262939453e-06,
      "model_forward_time": 5.580650806427002,
      "step": 8405
    },
    {
      "epoch": 8.01563262939453e-06,
      "step": 8405,
      "training_step_time": 16.989549160003662
    },
    {
      "epoch": 8.016586303710938e-06,
      "model_forward_time": 5.577857255935669,
      "step": 8406
    },
    {
      "epoch": 8.016586303710938e-06,
      "step": 8406,
      "training_step_time": 17.38684368133545
    },
    {
      "epoch": 8.017539978027344e-06,
      "model_forward_time": 5.5977184772491455,
      "step": 8407
    },
    {
      "epoch": 8.017539978027344e-06,
      "step": 8407,
      "training_step_time": 17.081926584243774
    },
    {
      "epoch": 8.01849365234375e-06,
      "model_forward_time": 5.622582674026489,
      "step": 8408
    },
    {
      "epoch": 8.01849365234375e-06,
      "step": 8408,
      "training_step_time": 17.110031366348267
    },
    {
      "epoch": 8.019447326660157e-06,
      "model_forward_time": 5.599980592727661,
      "step": 8409
    },
    {
      "epoch": 8.019447326660157e-06,
      "step": 8409,
      "training_step_time": 17.024533987045288
    },
    {
      "epoch": 8.020401000976563e-06,
      "grad_norm": 1.0817570686340332,
      "learning_rate": 1e-05,
      "loss": 0.0695,
      "step": 8410
    },
    {
      "epoch": 8.020401000976563e-06,
      "model_forward_time": 5.577952146530151,
      "step": 8410
    },
    {
      "dynamics_loss_avg": 0.06037726067006588,
      "epoch": 8.020401000976563e-06,
      "step": 8410
    },
    {
      "action_loss_avg": 0.009959002677351236,
      "epoch": 8.020401000976563e-06,
      "step": 8410
    },
    {
      "epoch": 8.020401000976563e-06,
      "step": 8410,
      "training_step_time": 17.010140895843506
    },
    {
      "epoch": 8.021354675292968e-06,
      "model_forward_time": 5.573744297027588,
      "step": 8411
    },
    {
      "epoch": 8.021354675292968e-06,
      "step": 8411,
      "training_step_time": 16.98642134666443
    },
    {
      "epoch": 8.022308349609376e-06,
      "model_forward_time": 5.947242021560669,
      "step": 8412
    },
    {
      "epoch": 8.022308349609376e-06,
      "step": 8412,
      "training_step_time": 17.374287366867065
    },
    {
      "epoch": 8.023262023925781e-06,
      "model_forward_time": 5.573090553283691,
      "step": 8413
    },
    {
      "epoch": 8.023262023925781e-06,
      "step": 8413,
      "training_step_time": 16.992198944091797
    },
    {
      "epoch": 8.024215698242187e-06,
      "model_forward_time": 5.595883131027222,
      "step": 8414
    },
    {
      "epoch": 8.024215698242187e-06,
      "step": 8414,
      "training_step_time": 17.053810358047485
    },
    {
      "epoch": 8.025169372558593e-06,
      "model_forward_time": 5.601950168609619,
      "step": 8415
    },
    {
      "epoch": 8.025169372558593e-06,
      "step": 8415,
      "training_step_time": 17.070231437683105
    },
    {
      "epoch": 8.026123046875e-06,
      "model_forward_time": 5.600595474243164,
      "step": 8416
    },
    {
      "epoch": 8.026123046875e-06,
      "step": 8416,
      "training_step_time": 17.075340270996094
    },
    {
      "epoch": 8.027076721191406e-06,
      "model_forward_time": 5.973422288894653,
      "step": 8417
    },
    {
      "epoch": 8.027076721191406e-06,
      "step": 8417,
      "training_step_time": 17.42691922187805
    },
    {
      "epoch": 8.028030395507812e-06,
      "model_forward_time": 5.592882394790649,
      "step": 8418
    },
    {
      "epoch": 8.028030395507812e-06,
      "step": 8418,
      "training_step_time": 17.033846378326416
    },
    {
      "epoch": 8.02898406982422e-06,
      "model_forward_time": 5.598198652267456,
      "step": 8419
    },
    {
      "epoch": 8.02898406982422e-06,
      "step": 8419,
      "training_step_time": 17.057878494262695
    },
    {
      "epoch": 8.029937744140625e-06,
      "grad_norm": 1.0741726160049438,
      "learning_rate": 1e-05,
      "loss": 0.059,
      "step": 8420
    },
    {
      "epoch": 8.029937744140625e-06,
      "model_forward_time": 5.6046459674835205,
      "step": 8420
    },
    {
      "dynamics_loss_avg": 0.0520798753015697,
      "epoch": 8.029937744140625e-06,
      "step": 8420
    },
    {
      "action_loss_avg": 0.007680729113053531,
      "epoch": 8.029937744140625e-06,
      "step": 8420
    },
    {
      "epoch": 8.029937744140625e-06,
      "step": 8420,
      "training_step_time": 17.089828491210938
    },
    {
      "epoch": 8.03089141845703e-06,
      "model_forward_time": 5.613182306289673,
      "step": 8421
    },
    {
      "epoch": 8.03089141845703e-06,
      "step": 8421,
      "training_step_time": 17.1028048992157
    },
    {
      "epoch": 8.031845092773438e-06,
      "model_forward_time": 5.610666036605835,
      "step": 8422
    },
    {
      "epoch": 8.031845092773438e-06,
      "step": 8422,
      "training_step_time": 17.105547666549683
    },
    {
      "epoch": 8.032798767089844e-06,
      "model_forward_time": 5.972620725631714,
      "step": 8423
    },
    {
      "epoch": 8.032798767089844e-06,
      "step": 8423,
      "training_step_time": 17.465683221817017
    },
    {
      "epoch": 8.03375244140625e-06,
      "model_forward_time": 5.60519552230835,
      "step": 8424
    },
    {
      "epoch": 8.03375244140625e-06,
      "step": 8424,
      "training_step_time": 17.079199075698853
    },
    {
      "epoch": 8.034706115722657e-06,
      "model_forward_time": 5.5994768142700195,
      "step": 8425
    },
    {
      "epoch": 8.034706115722657e-06,
      "step": 8425,
      "training_step_time": 17.04304265975952
    },
    {
      "epoch": 8.035659790039063e-06,
      "model_forward_time": 5.582979440689087,
      "step": 8426
    },
    {
      "epoch": 8.035659790039063e-06,
      "step": 8426,
      "training_step_time": 17.019031047821045
    },
    {
      "epoch": 8.036613464355468e-06,
      "model_forward_time": 5.596998691558838,
      "step": 8427
    },
    {
      "epoch": 8.036613464355468e-06,
      "step": 8427,
      "training_step_time": 17.06788396835327
    },
    {
      "epoch": 8.037567138671876e-06,
      "model_forward_time": 5.6032726764678955,
      "step": 8428
    },
    {
      "epoch": 8.037567138671876e-06,
      "step": 8428,
      "training_step_time": 17.454575538635254
    },
    {
      "epoch": 8.038520812988282e-06,
      "model_forward_time": 5.601375341415405,
      "step": 8429
    },
    {
      "epoch": 8.038520812988282e-06,
      "step": 8429,
      "training_step_time": 17.03041124343872
    },
    {
      "epoch": 8.039474487304687e-06,
      "grad_norm": 1.2312941551208496,
      "learning_rate": 1e-05,
      "loss": 0.0796,
      "step": 8430
    },
    {
      "epoch": 8.039474487304687e-06,
      "model_forward_time": 5.598611116409302,
      "step": 8430
    },
    {
      "dynamics_loss_avg": 0.07268036101013423,
      "epoch": 8.039474487304687e-06,
      "step": 8430
    },
    {
      "action_loss_avg": 0.01341919954866171,
      "epoch": 8.039474487304687e-06,
      "step": 8430
    },
    {
      "epoch": 8.039474487304687e-06,
      "step": 8430,
      "training_step_time": 17.03036856651306
    },
    {
      "epoch": 8.040428161621093e-06,
      "model_forward_time": 5.599353075027466,
      "step": 8431
    },
    {
      "epoch": 8.040428161621093e-06,
      "step": 8431,
      "training_step_time": 17.068334102630615
    },
    {
      "epoch": 8.0413818359375e-06,
      "model_forward_time": 5.606238603591919,
      "step": 8432
    },
    {
      "epoch": 8.0413818359375e-06,
      "step": 8432,
      "training_step_time": 17.088368892669678
    },
    {
      "epoch": 8.042335510253906e-06,
      "model_forward_time": 5.6169164180755615,
      "step": 8433
    },
    {
      "epoch": 8.042335510253906e-06,
      "step": 8433,
      "training_step_time": 17.12231683731079
    },
    {
      "epoch": 8.043289184570312e-06,
      "model_forward_time": 5.9988532066345215,
      "step": 8434
    },
    {
      "epoch": 8.043289184570312e-06,
      "step": 8434,
      "training_step_time": 17.4740092754364
    },
    {
      "epoch": 8.04424285888672e-06,
      "model_forward_time": 5.59716272354126,
      "step": 8435
    },
    {
      "epoch": 8.04424285888672e-06,
      "step": 8435,
      "training_step_time": 17.013261795043945
    },
    {
      "epoch": 8.045196533203125e-06,
      "model_forward_time": 5.573326110839844,
      "step": 8436
    },
    {
      "epoch": 8.045196533203125e-06,
      "step": 8436,
      "training_step_time": 16.986754894256592
    },
    {
      "epoch": 8.04615020751953e-06,
      "model_forward_time": 5.566420316696167,
      "step": 8437
    },
    {
      "epoch": 8.04615020751953e-06,
      "step": 8437,
      "training_step_time": 17.01390266418457
    },
    {
      "epoch": 8.047103881835938e-06,
      "model_forward_time": 5.581987619400024,
      "step": 8438
    },
    {
      "epoch": 8.047103881835938e-06,
      "step": 8438,
      "training_step_time": 16.996140956878662
    },
    {
      "epoch": 8.048057556152344e-06,
      "model_forward_time": 5.59075665473938,
      "step": 8439
    },
    {
      "epoch": 8.048057556152344e-06,
      "step": 8439,
      "training_step_time": 17.373226165771484
    },
    {
      "epoch": 8.04901123046875e-06,
      "grad_norm": 0.2851400375366211,
      "learning_rate": 1e-05,
      "loss": 0.0769,
      "step": 8440
    },
    {
      "epoch": 8.04901123046875e-06,
      "model_forward_time": 5.589345693588257,
      "step": 8440
    },
    {
      "dynamics_loss_avg": 0.06691259369254113,
      "epoch": 8.04901123046875e-06,
      "step": 8440
    },
    {
      "action_loss_avg": 0.010543913906440138,
      "epoch": 8.04901123046875e-06,
      "step": 8440
    },
    {
      "epoch": 8.04901123046875e-06,
      "step": 8440,
      "training_step_time": 17.04396080970764
    },
    {
      "epoch": 8.049964904785157e-06,
      "model_forward_time": 5.594430446624756,
      "step": 8441
    },
    {
      "epoch": 8.049964904785157e-06,
      "step": 8441,
      "training_step_time": 17.0917866230011
    },
    {
      "epoch": 8.050918579101563e-06,
      "model_forward_time": 5.620220899581909,
      "step": 8442
    },
    {
      "epoch": 8.050918579101563e-06,
      "step": 8442,
      "training_step_time": 17.128223657608032
    },
    {
      "epoch": 8.051872253417968e-06,
      "model_forward_time": 5.610789775848389,
      "step": 8443
    },
    {
      "epoch": 8.051872253417968e-06,
      "step": 8443,
      "training_step_time": 17.098894357681274
    },
    {
      "epoch": 8.052825927734376e-06,
      "model_forward_time": 5.6059792041778564,
      "step": 8444
    },
    {
      "epoch": 8.052825927734376e-06,
      "step": 8444,
      "training_step_time": 17.090083599090576
    },
    {
      "epoch": 8.053779602050782e-06,
      "model_forward_time": 5.969352722167969,
      "step": 8445
    },
    {
      "epoch": 8.053779602050782e-06,
      "step": 8445,
      "training_step_time": 17.44158911705017
    },
    {
      "epoch": 8.054733276367187e-06,
      "model_forward_time": 5.602094411849976,
      "step": 8446
    },
    {
      "epoch": 8.054733276367187e-06,
      "step": 8446,
      "training_step_time": 17.04484272003174
    },
    {
      "epoch": 8.055686950683593e-06,
      "model_forward_time": 5.5943989753723145,
      "step": 8447
    },
    {
      "epoch": 8.055686950683593e-06,
      "step": 8447,
      "training_step_time": 17.0611252784729
    },
    {
      "epoch": 8.056640625e-06,
      "model_forward_time": 5.603255033493042,
      "step": 8448
    },
    {
      "epoch": 8.056640625e-06,
      "step": 8448,
      "training_step_time": 17.058111667633057
    },
    {
      "epoch": 8.057594299316406e-06,
      "model_forward_time": 5.607542037963867,
      "step": 8449
    },
    {
      "epoch": 8.057594299316406e-06,
      "step": 8449,
      "training_step_time": 17.08630895614624
    },
    {
      "epoch": 8.058547973632812e-06,
      "grad_norm": 0.4974748492240906,
      "learning_rate": 1e-05,
      "loss": 0.0805,
      "step": 8450
    },
    {
      "epoch": 8.058547973632812e-06,
      "model_forward_time": 5.958703279495239,
      "step": 8450
    },
    {
      "dynamics_loss_avg": 0.06508465074002742,
      "epoch": 8.058547973632812e-06,
      "step": 8450
    },
    {
      "action_loss_avg": 0.01201807491015643,
      "epoch": 8.058547973632812e-06,
      "step": 8450
    },
    {
      "epoch": 8.058547973632812e-06,
      "step": 8450,
      "training_step_time": 17.44583225250244
    },
    {
      "epoch": 8.05950164794922e-06,
      "model_forward_time": 5.606914281845093,
      "step": 8451
    },
    {
      "epoch": 8.05950164794922e-06,
      "step": 8451,
      "training_step_time": 17.10570788383484
    },
    {
      "epoch": 8.060455322265625e-06,
      "model_forward_time": 5.611752033233643,
      "step": 8452
    },
    {
      "epoch": 8.060455322265625e-06,
      "step": 8452,
      "training_step_time": 17.098145008087158
    },
    {
      "epoch": 8.06140899658203e-06,
      "model_forward_time": 5.60437798500061,
      "step": 8453
    },
    {
      "epoch": 8.06140899658203e-06,
      "step": 8453,
      "training_step_time": 17.07969570159912
    },
    {
      "epoch": 8.062362670898438e-06,
      "model_forward_time": 5.593586683273315,
      "step": 8454
    },
    {
      "epoch": 8.062362670898438e-06,
      "step": 8454,
      "training_step_time": 17.055313110351562
    },
    {
      "epoch": 8.063316345214844e-06,
      "model_forward_time": 5.590164422988892,
      "step": 8455
    },
    {
      "epoch": 8.063316345214844e-06,
      "step": 8455,
      "training_step_time": 17.02526831626892
    },
    {
      "epoch": 8.06427001953125e-06,
      "model_forward_time": 5.9606873989105225,
      "step": 8456
    },
    {
      "epoch": 8.06427001953125e-06,
      "step": 8456,
      "training_step_time": 17.40731906890869
    },
    {
      "epoch": 8.065223693847657e-06,
      "model_forward_time": 5.593020915985107,
      "step": 8457
    },
    {
      "epoch": 8.065223693847657e-06,
      "step": 8457,
      "training_step_time": 17.033388137817383
    },
    {
      "epoch": 8.066177368164063e-06,
      "model_forward_time": 5.598348379135132,
      "step": 8458
    },
    {
      "epoch": 8.066177368164063e-06,
      "step": 8458,
      "training_step_time": 17.05251383781433
    },
    {
      "epoch": 8.067131042480468e-06,
      "model_forward_time": 5.5939953327178955,
      "step": 8459
    },
    {
      "epoch": 8.067131042480468e-06,
      "step": 8459,
      "training_step_time": 17.049636125564575
    },
    {
      "epoch": 8.068084716796876e-06,
      "grad_norm": 0.9138109683990479,
      "learning_rate": 1e-05,
      "loss": 0.0677,
      "step": 8460
    },
    {
      "epoch": 8.068084716796876e-06,
      "model_forward_time": 5.602889776229858,
      "step": 8460
    },
    {
      "dynamics_loss_avg": 0.05946339946240187,
      "epoch": 8.068084716796876e-06,
      "step": 8460
    },
    {
      "action_loss_avg": 0.00916672192979604,
      "epoch": 8.068084716796876e-06,
      "step": 8460
    },
    {
      "epoch": 8.068084716796876e-06,
      "step": 8460,
      "training_step_time": 17.068267583847046
    },
    {
      "epoch": 8.069038391113282e-06,
      "model_forward_time": 5.6242430210113525,
      "step": 8461
    },
    {
      "epoch": 8.069038391113282e-06,
      "step": 8461,
      "training_step_time": 17.49256134033203
    },
    {
      "epoch": 8.069992065429687e-06,
      "model_forward_time": 5.626333951950073,
      "step": 8462
    },
    {
      "epoch": 8.069992065429687e-06,
      "step": 8462,
      "training_step_time": 17.153969764709473
    },
    {
      "epoch": 8.070945739746093e-06,
      "model_forward_time": 5.627440929412842,
      "step": 8463
    },
    {
      "epoch": 8.070945739746093e-06,
      "step": 8463,
      "training_step_time": 17.1279776096344
    },
    {
      "epoch": 8.0718994140625e-06,
      "model_forward_time": 5.61947774887085,
      "step": 8464
    },
    {
      "epoch": 8.0718994140625e-06,
      "step": 8464,
      "training_step_time": 17.10248565673828
    },
    {
      "epoch": 8.072853088378906e-06,
      "model_forward_time": 5.596322536468506,
      "step": 8465
    },
    {
      "epoch": 8.072853088378906e-06,
      "step": 8465,
      "training_step_time": 16.997916221618652
    },
    {
      "epoch": 8.073806762695312e-06,
      "model_forward_time": 5.576452016830444,
      "step": 8466
    },
    {
      "epoch": 8.073806762695312e-06,
      "step": 8466,
      "training_step_time": 16.983470916748047
    },
    {
      "epoch": 8.07476043701172e-06,
      "model_forward_time": 5.947250127792358,
      "step": 8467
    },
    {
      "epoch": 8.07476043701172e-06,
      "step": 8467,
      "training_step_time": 17.366918802261353
    },
    {
      "epoch": 8.075714111328125e-06,
      "model_forward_time": 5.559872627258301,
      "step": 8468
    },
    {
      "epoch": 8.075714111328125e-06,
      "step": 8468,
      "training_step_time": 16.98096537590027
    },
    {
      "epoch": 8.07666778564453e-06,
      "model_forward_time": 5.564774751663208,
      "step": 8469
    },
    {
      "epoch": 8.07666778564453e-06,
      "step": 8469,
      "training_step_time": 16.98375415802002
    },
    {
      "epoch": 8.077621459960938e-06,
      "grad_norm": 0.6203500032424927,
      "learning_rate": 1e-05,
      "loss": 0.06,
      "step": 8470
    },
    {
      "epoch": 8.077621459960938e-06,
      "model_forward_time": 5.568885087966919,
      "step": 8470
    },
    {
      "dynamics_loss_avg": 0.051881869323551655,
      "epoch": 8.077621459960938e-06,
      "step": 8470
    },
    {
      "action_loss_avg": 0.007118790177628398,
      "epoch": 8.077621459960938e-06,
      "step": 8470
    },
    {
      "epoch": 8.077621459960938e-06,
      "step": 8470,
      "training_step_time": 16.977874517440796
    },
    {
      "epoch": 8.078575134277344e-06,
      "model_forward_time": 5.583250999450684,
      "step": 8471
    },
    {
      "epoch": 8.078575134277344e-06,
      "step": 8471,
      "training_step_time": 17.030157566070557
    },
    {
      "epoch": 8.07952880859375e-06,
      "model_forward_time": 5.587187767028809,
      "step": 8472
    },
    {
      "epoch": 8.07952880859375e-06,
      "step": 8472,
      "training_step_time": 17.436872243881226
    },
    {
      "epoch": 8.080482482910157e-06,
      "model_forward_time": 5.621821641921997,
      "step": 8473
    },
    {
      "epoch": 8.080482482910157e-06,
      "step": 8473,
      "training_step_time": 17.138071298599243
    },
    {
      "epoch": 8.081436157226563e-06,
      "model_forward_time": 5.615700006484985,
      "step": 8474
    },
    {
      "epoch": 8.081436157226563e-06,
      "step": 8474,
      "training_step_time": 17.09813666343689
    },
    {
      "epoch": 8.082389831542968e-06,
      "model_forward_time": 5.590556859970093,
      "step": 8475
    },
    {
      "epoch": 8.082389831542968e-06,
      "step": 8475,
      "training_step_time": 17.02942681312561
    },
    {
      "epoch": 8.083343505859376e-06,
      "model_forward_time": 5.565006971359253,
      "step": 8476
    },
    {
      "epoch": 8.083343505859376e-06,
      "step": 8476,
      "training_step_time": 16.98032784461975
    },
    {
      "epoch": 8.084297180175782e-06,
      "model_forward_time": 4.250668525695801,
      "step": 8477
    },
    {
      "epoch": 8.084297180175782e-06,
      "step": 8477,
      "training_step_time": 12.719360113143921
    },
    {
      "epoch": 8.085250854492187e-06,
      "model_forward_time": 5.928468942642212,
      "step": 8478
    },
    {
      "epoch": 8.085250854492187e-06,
      "step": 8478,
      "training_step_time": 17.346492767333984
    },
    {
      "epoch": 8.086204528808593e-06,
      "model_forward_time": 5.562687873840332,
      "step": 8479
    },
    {
      "epoch": 8.086204528808593e-06,
      "step": 8479,
      "training_step_time": 17.00175905227661
    },
    {
      "epoch": 8.087158203125e-06,
      "grad_norm": 1.104095697402954,
      "learning_rate": 1e-05,
      "loss": 0.0745,
      "step": 8480
    },
    {
      "epoch": 8.087158203125e-06,
      "model_forward_time": 5.580155611038208,
      "step": 8480
    },
    {
      "dynamics_loss_avg": 0.06278393790125847,
      "epoch": 8.087158203125e-06,
      "step": 8480
    },
    {
      "action_loss_avg": 0.008134770579636097,
      "epoch": 8.087158203125e-06,
      "step": 8480
    },
    {
      "epoch": 8.087158203125e-06,
      "step": 8480,
      "training_step_time": 16.98698902130127
    },
    {
      "epoch": 8.088111877441406e-06,
      "model_forward_time": 5.572713851928711,
      "step": 8481
    },
    {
      "epoch": 8.088111877441406e-06,
      "step": 8481,
      "training_step_time": 16.986305713653564
    },
    {
      "epoch": 8.089065551757812e-06,
      "model_forward_time": 5.585689544677734,
      "step": 8482
    },
    {
      "epoch": 8.089065551757812e-06,
      "step": 8482,
      "training_step_time": 17.004639625549316
    },
    {
      "epoch": 8.09001922607422e-06,
      "model_forward_time": 5.5887839794158936,
      "step": 8483
    },
    {
      "epoch": 8.09001922607422e-06,
      "step": 8483,
      "training_step_time": 17.411802530288696
    },
    {
      "epoch": 8.090972900390625e-06,
      "model_forward_time": 5.618645906448364,
      "step": 8484
    },
    {
      "epoch": 8.090972900390625e-06,
      "step": 8484,
      "training_step_time": 17.13082456588745
    },
    {
      "epoch": 8.09192657470703e-06,
      "model_forward_time": 5.619076728820801,
      "step": 8485
    },
    {
      "epoch": 8.09192657470703e-06,
      "step": 8485,
      "training_step_time": 17.109182834625244
    },
    {
      "epoch": 8.092880249023438e-06,
      "model_forward_time": 5.59971809387207,
      "step": 8486
    },
    {
      "epoch": 8.092880249023438e-06,
      "step": 8486,
      "training_step_time": 17.035717248916626
    },
    {
      "epoch": 8.093833923339844e-06,
      "model_forward_time": 5.5802531242370605,
      "step": 8487
    },
    {
      "epoch": 8.093833923339844e-06,
      "step": 8487,
      "training_step_time": 17.010794162750244
    },
    {
      "epoch": 8.09478759765625e-06,
      "model_forward_time": 5.5811779499053955,
      "step": 8488
    },
    {
      "epoch": 8.09478759765625e-06,
      "step": 8488,
      "training_step_time": 17.01156783103943
    },
    {
      "epoch": 8.095741271972657e-06,
      "model_forward_time": 5.972417116165161,
      "step": 8489
    },
    {
      "epoch": 8.095741271972657e-06,
      "step": 8489,
      "training_step_time": 17.417081594467163
    },
    {
      "epoch": 8.096694946289063e-06,
      "grad_norm": 0.9450604915618896,
      "learning_rate": 1e-05,
      "loss": 0.0518,
      "step": 8490
    },
    {
      "epoch": 8.096694946289063e-06,
      "model_forward_time": 5.6020495891571045,
      "step": 8490
    },
    {
      "dynamics_loss_avg": 0.04646002538502216,
      "epoch": 8.096694946289063e-06,
      "step": 8490
    },
    {
      "action_loss_avg": 0.007121229520998895,
      "epoch": 8.096694946289063e-06,
      "step": 8490
    },
    {
      "epoch": 8.096694946289063e-06,
      "step": 8490,
      "training_step_time": 17.059297561645508
    },
    {
      "epoch": 8.097648620605468e-06,
      "model_forward_time": 5.601651906967163,
      "step": 8491
    },
    {
      "epoch": 8.097648620605468e-06,
      "step": 8491,
      "training_step_time": 17.06274104118347
    },
    {
      "epoch": 8.098602294921876e-06,
      "model_forward_time": 5.5926430225372314,
      "step": 8492
    },
    {
      "epoch": 8.098602294921876e-06,
      "step": 8492,
      "training_step_time": 17.059279680252075
    },
    {
      "epoch": 8.099555969238282e-06,
      "model_forward_time": 5.597570419311523,
      "step": 8493
    },
    {
      "epoch": 8.099555969238282e-06,
      "step": 8493,
      "training_step_time": 17.077985763549805
    },
    {
      "epoch": 8.100509643554687e-06,
      "model_forward_time": 5.964164972305298,
      "step": 8494
    },
    {
      "epoch": 8.100509643554687e-06,
      "step": 8494,
      "training_step_time": 17.419836282730103
    },
    {
      "epoch": 8.101463317871093e-06,
      "model_forward_time": 5.602673292160034,
      "step": 8495
    },
    {
      "epoch": 8.101463317871093e-06,
      "step": 8495,
      "training_step_time": 17.082200288772583
    },
    {
      "epoch": 8.1024169921875e-06,
      "model_forward_time": 5.60853910446167,
      "step": 8496
    },
    {
      "epoch": 8.1024169921875e-06,
      "step": 8496,
      "training_step_time": 17.092297315597534
    },
    {
      "epoch": 8.103370666503906e-06,
      "model_forward_time": 5.6075215339660645,
      "step": 8497
    },
    {
      "epoch": 8.103370666503906e-06,
      "step": 8497,
      "training_step_time": 17.077462434768677
    },
    {
      "epoch": 8.104324340820312e-06,
      "model_forward_time": 5.584464073181152,
      "step": 8498
    },
    {
      "epoch": 8.104324340820312e-06,
      "step": 8498,
      "training_step_time": 17.032912254333496
    },
    {
      "epoch": 8.10527801513672e-06,
      "model_forward_time": 5.580858945846558,
      "step": 8499
    },
    {
      "epoch": 8.10527801513672e-06,
      "step": 8499,
      "training_step_time": 17.00736904144287
    },
    {
      "epoch": 8.106231689453125e-06,
      "grad_norm": 0.6487963795661926,
      "learning_rate": 1e-05,
      "loss": 0.0611,
      "step": 8500
    },
    {
      "epoch": 8.106231689453125e-06,
      "model_forward_time": 5.955061197280884,
      "step": 8500
    },
    {
      "dynamics_loss_avg": 0.049724008329212664,
      "epoch": 8.106231689453125e-06,
      "step": 8500
    },
    {
      "action_loss_avg": 0.008226476213894784,
      "epoch": 8.106231689453125e-06,
      "step": 8500
    },
    {
      "epoch": 8.106231689453125e-06,
      "step": 8500,
      "training_step_time": 17.406782150268555
    },
    {
      "epoch": 8.10718536376953e-06,
      "model_forward_time": 5.599960565567017,
      "step": 8501
    },
    {
      "epoch": 8.10718536376953e-06,
      "step": 8501,
      "training_step_time": 17.074963569641113
    },
    {
      "epoch": 8.108139038085938e-06,
      "model_forward_time": 5.606725454330444,
      "step": 8502
    },
    {
      "epoch": 8.108139038085938e-06,
      "step": 8502,
      "training_step_time": 17.08894681930542
    },
    {
      "epoch": 8.109092712402344e-06,
      "model_forward_time": 5.626960039138794,
      "step": 8503
    },
    {
      "epoch": 8.109092712402344e-06,
      "step": 8503,
      "training_step_time": 17.09593677520752
    },
    {
      "epoch": 8.11004638671875e-06,
      "model_forward_time": 5.593524217605591,
      "step": 8504
    },
    {
      "epoch": 8.11004638671875e-06,
      "step": 8504,
      "training_step_time": 17.035934686660767
    },
    {
      "epoch": 8.111000061035157e-06,
      "model_forward_time": 5.932965993881226,
      "step": 8505
    },
    {
      "epoch": 8.111000061035157e-06,
      "step": 8505,
      "training_step_time": 17.383294343948364
    },
    {
      "epoch": 8.111953735351563e-06,
      "model_forward_time": 5.5924036502838135,
      "step": 8506
    },
    {
      "epoch": 8.111953735351563e-06,
      "step": 8506,
      "training_step_time": 17.015936613082886
    },
    {
      "epoch": 8.112907409667968e-06,
      "model_forward_time": 5.597978830337524,
      "step": 8507
    },
    {
      "epoch": 8.112907409667968e-06,
      "step": 8507,
      "training_step_time": 17.040908336639404
    },
    {
      "epoch": 8.113861083984374e-06,
      "model_forward_time": 5.595790386199951,
      "step": 8508
    },
    {
      "epoch": 8.113861083984374e-06,
      "step": 8508,
      "training_step_time": 17.05918836593628
    },
    {
      "epoch": 8.114814758300782e-06,
      "model_forward_time": 5.598580598831177,
      "step": 8509
    },
    {
      "epoch": 8.114814758300782e-06,
      "step": 8509,
      "training_step_time": 17.065951108932495
    },
    {
      "epoch": 8.115768432617187e-06,
      "grad_norm": 0.37870004773139954,
      "learning_rate": 1e-05,
      "loss": 0.0587,
      "step": 8510
    },
    {
      "epoch": 8.115768432617187e-06,
      "model_forward_time": 5.598448753356934,
      "step": 8510
    },
    {
      "dynamics_loss_avg": 0.054962565936148164,
      "epoch": 8.115768432617187e-06,
      "step": 8510
    },
    {
      "action_loss_avg": 0.007900909218005837,
      "epoch": 8.115768432617187e-06,
      "step": 8510
    },
    {
      "epoch": 8.115768432617187e-06,
      "step": 8510,
      "training_step_time": 17.054349899291992
    },
    {
      "epoch": 8.116722106933593e-06,
      "model_forward_time": 5.98418664932251,
      "step": 8511
    },
    {
      "epoch": 8.116722106933593e-06,
      "step": 8511,
      "training_step_time": 17.468083381652832
    },
    {
      "epoch": 8.11767578125e-06,
      "model_forward_time": 5.649597406387329,
      "step": 8512
    },
    {
      "epoch": 8.11767578125e-06,
      "step": 8512,
      "training_step_time": 17.15560746192932
    },
    {
      "epoch": 8.118629455566406e-06,
      "model_forward_time": 5.618891954421997,
      "step": 8513
    },
    {
      "epoch": 8.118629455566406e-06,
      "step": 8513,
      "training_step_time": 17.1037757396698
    },
    {
      "epoch": 8.119583129882812e-06,
      "model_forward_time": 5.622481346130371,
      "step": 8514
    },
    {
      "epoch": 8.119583129882812e-06,
      "step": 8514,
      "training_step_time": 17.111743211746216
    },
    {
      "epoch": 8.12053680419922e-06,
      "model_forward_time": 5.618964195251465,
      "step": 8515
    },
    {
      "epoch": 8.12053680419922e-06,
      "step": 8515,
      "training_step_time": 17.10448431968689
    },
    {
      "epoch": 8.121490478515625e-06,
      "model_forward_time": 5.955759048461914,
      "step": 8516
    },
    {
      "epoch": 8.121490478515625e-06,
      "step": 8516,
      "training_step_time": 17.4013352394104
    },
    {
      "epoch": 8.12244415283203e-06,
      "model_forward_time": 5.591710805892944,
      "step": 8517
    },
    {
      "epoch": 8.12244415283203e-06,
      "step": 8517,
      "training_step_time": 17.04210352897644
    },
    {
      "epoch": 8.123397827148438e-06,
      "model_forward_time": 5.6109395027160645,
      "step": 8518
    },
    {
      "epoch": 8.123397827148438e-06,
      "step": 8518,
      "training_step_time": 17.050143241882324
    },
    {
      "epoch": 8.124351501464844e-06,
      "model_forward_time": 5.614334344863892,
      "step": 8519
    },
    {
      "epoch": 8.124351501464844e-06,
      "step": 8519,
      "training_step_time": 17.07086205482483
    },
    {
      "epoch": 8.12530517578125e-06,
      "grad_norm": 0.41703692078590393,
      "learning_rate": 1e-05,
      "loss": 0.0615,
      "step": 8520
    },
    {
      "epoch": 8.12530517578125e-06,
      "model_forward_time": 5.603177309036255,
      "step": 8520
    },
    {
      "dynamics_loss_avg": 0.052865827828645705,
      "epoch": 8.12530517578125e-06,
      "step": 8520
    },
    {
      "action_loss_avg": 0.008512448123656213,
      "epoch": 8.12530517578125e-06,
      "step": 8520
    },
    {
      "epoch": 8.12530517578125e-06,
      "step": 8520,
      "training_step_time": 17.086134433746338
    },
    {
      "epoch": 8.126258850097657e-06,
      "model_forward_time": 5.608959197998047,
      "step": 8521
    },
    {
      "epoch": 8.126258850097657e-06,
      "step": 8521,
      "training_step_time": 17.479896545410156
    },
    {
      "epoch": 8.127212524414063e-06,
      "model_forward_time": 5.61694598197937,
      "step": 8522
    },
    {
      "epoch": 8.127212524414063e-06,
      "step": 8522,
      "training_step_time": 17.094987869262695
    },
    {
      "epoch": 8.128166198730468e-06,
      "model_forward_time": 5.594466209411621,
      "step": 8523
    },
    {
      "epoch": 8.128166198730468e-06,
      "step": 8523,
      "training_step_time": 17.039944171905518
    },
    {
      "epoch": 8.129119873046874e-06,
      "model_forward_time": 5.584197044372559,
      "step": 8524
    },
    {
      "epoch": 8.129119873046874e-06,
      "step": 8524,
      "training_step_time": 17.01704239845276
    },
    {
      "epoch": 8.130073547363282e-06,
      "model_forward_time": 5.581481456756592,
      "step": 8525
    },
    {
      "epoch": 8.130073547363282e-06,
      "step": 8525,
      "training_step_time": 17.022786617279053
    },
    {
      "epoch": 8.131027221679687e-06,
      "model_forward_time": 5.5980284214019775,
      "step": 8526
    },
    {
      "epoch": 8.131027221679687e-06,
      "step": 8526,
      "training_step_time": 17.068883419036865
    },
    {
      "epoch": 8.131980895996093e-06,
      "model_forward_time": 5.957382917404175,
      "step": 8527
    },
    {
      "epoch": 8.131980895996093e-06,
      "step": 8527,
      "training_step_time": 17.419788360595703
    },
    {
      "epoch": 8.1329345703125e-06,
      "model_forward_time": 5.604519844055176,
      "step": 8528
    },
    {
      "epoch": 8.1329345703125e-06,
      "step": 8528,
      "training_step_time": 17.064108848571777
    },
    {
      "epoch": 8.133888244628906e-06,
      "model_forward_time": 5.597616195678711,
      "step": 8529
    },
    {
      "epoch": 8.133888244628906e-06,
      "step": 8529,
      "training_step_time": 17.0671648979187
    },
    {
      "epoch": 8.134841918945312e-06,
      "grad_norm": 0.9620324373245239,
      "learning_rate": 1e-05,
      "loss": 0.0655,
      "step": 8530
    },
    {
      "epoch": 8.134841918945312e-06,
      "model_forward_time": 5.596262216567993,
      "step": 8530
    },
    {
      "dynamics_loss_avg": 0.055499362759292126,
      "epoch": 8.134841918945312e-06,
      "step": 8530
    },
    {
      "action_loss_avg": 0.00972029655240476,
      "epoch": 8.134841918945312e-06,
      "step": 8530
    },
    {
      "epoch": 8.134841918945312e-06,
      "step": 8530,
      "training_step_time": 17.05555486679077
    },
    {
      "epoch": 8.13579559326172e-06,
      "model_forward_time": 5.593305349349976,
      "step": 8531
    },
    {
      "epoch": 8.13579559326172e-06,
      "step": 8531,
      "training_step_time": 17.043488025665283
    },
    {
      "epoch": 8.136749267578125e-06,
      "model_forward_time": 5.585696220397949,
      "step": 8532
    },
    {
      "epoch": 8.136749267578125e-06,
      "step": 8532,
      "training_step_time": 17.39930248260498
    },
    {
      "epoch": 8.13770294189453e-06,
      "model_forward_time": 5.598094701766968,
      "step": 8533
    },
    {
      "epoch": 8.13770294189453e-06,
      "step": 8533,
      "training_step_time": 17.070521116256714
    },
    {
      "epoch": 8.138656616210938e-06,
      "model_forward_time": 5.595864534378052,
      "step": 8534
    },
    {
      "epoch": 8.138656616210938e-06,
      "step": 8534,
      "training_step_time": 17.080517053604126
    },
    {
      "epoch": 8.139610290527344e-06,
      "model_forward_time": 5.612310886383057,
      "step": 8535
    },
    {
      "epoch": 8.139610290527344e-06,
      "step": 8535,
      "training_step_time": 17.106534004211426
    },
    {
      "epoch": 8.14056396484375e-06,
      "model_forward_time": 5.618250608444214,
      "step": 8536
    },
    {
      "epoch": 8.14056396484375e-06,
      "step": 8536,
      "training_step_time": 17.08700728416443
    },
    {
      "epoch": 8.141517639160157e-06,
      "model_forward_time": 5.589972257614136,
      "step": 8537
    },
    {
      "epoch": 8.141517639160157e-06,
      "step": 8537,
      "training_step_time": 17.00578546524048
    },
    {
      "epoch": 8.142471313476563e-06,
      "model_forward_time": 5.920238494873047,
      "step": 8538
    },
    {
      "epoch": 8.142471313476563e-06,
      "step": 8538,
      "training_step_time": 17.340882301330566
    },
    {
      "epoch": 8.143424987792968e-06,
      "model_forward_time": 5.577891111373901,
      "step": 8539
    },
    {
      "epoch": 8.143424987792968e-06,
      "step": 8539,
      "training_step_time": 17.000704526901245
    },
    {
      "epoch": 8.144378662109374e-06,
      "grad_norm": 1.2935363054275513,
      "learning_rate": 1e-05,
      "loss": 0.068,
      "step": 8540
    },
    {
      "epoch": 8.144378662109374e-06,
      "model_forward_time": 5.577549934387207,
      "step": 8540
    },
    {
      "dynamics_loss_avg": 0.05735751967877149,
      "epoch": 8.144378662109374e-06,
      "step": 8540
    },
    {
      "action_loss_avg": 0.00848169201053679,
      "epoch": 8.144378662109374e-06,
      "step": 8540
    },
    {
      "epoch": 8.144378662109374e-06,
      "step": 8540,
      "training_step_time": 17.028732776641846
    },
    {
      "epoch": 8.145332336425782e-06,
      "model_forward_time": 5.583873987197876,
      "step": 8541
    },
    {
      "epoch": 8.145332336425782e-06,
      "step": 8541,
      "training_step_time": 17.022454977035522
    },
    {
      "epoch": 8.146286010742187e-06,
      "model_forward_time": 5.6016435623168945,
      "step": 8542
    },
    {
      "epoch": 8.146286010742187e-06,
      "step": 8542,
      "training_step_time": 17.089502334594727
    },
    {
      "epoch": 8.147239685058593e-06,
      "model_forward_time": 5.985934495925903,
      "step": 8543
    },
    {
      "epoch": 8.147239685058593e-06,
      "step": 8543,
      "training_step_time": 17.470669984817505
    },
    {
      "epoch": 8.148193359375e-06,
      "model_forward_time": 5.608870029449463,
      "step": 8544
    },
    {
      "epoch": 8.148193359375e-06,
      "step": 8544,
      "training_step_time": 17.099122762680054
    },
    {
      "epoch": 8.149147033691406e-06,
      "model_forward_time": 5.631079912185669,
      "step": 8545
    },
    {
      "epoch": 8.149147033691406e-06,
      "step": 8545,
      "training_step_time": 17.13242530822754
    },
    {
      "epoch": 8.150100708007812e-06,
      "model_forward_time": 5.6090476512908936,
      "step": 8546
    },
    {
      "epoch": 8.150100708007812e-06,
      "step": 8546,
      "training_step_time": 17.079130172729492
    },
    {
      "epoch": 8.15105438232422e-06,
      "model_forward_time": 5.600250959396362,
      "step": 8547
    },
    {
      "epoch": 8.15105438232422e-06,
      "step": 8547,
      "training_step_time": 17.049755334854126
    },
    {
      "epoch": 8.152008056640625e-06,
      "model_forward_time": 5.596631288528442,
      "step": 8548
    },
    {
      "epoch": 8.152008056640625e-06,
      "step": 8548,
      "training_step_time": 17.42466115951538
    },
    {
      "epoch": 8.15296173095703e-06,
      "model_forward_time": 5.59351658821106,
      "step": 8549
    },
    {
      "epoch": 8.15296173095703e-06,
      "step": 8549,
      "training_step_time": 17.03299117088318
    },
    {
      "epoch": 8.153915405273438e-06,
      "grad_norm": 0.37784337997436523,
      "learning_rate": 1e-05,
      "loss": 0.0688,
      "step": 8550
    },
    {
      "epoch": 8.153915405273438e-06,
      "model_forward_time": 5.587994337081909,
      "step": 8550
    },
    {
      "dynamics_loss_avg": 0.06485911551862955,
      "epoch": 8.153915405273438e-06,
      "step": 8550
    },
    {
      "action_loss_avg": 0.009534080501180143,
      "epoch": 8.153915405273438e-06,
      "step": 8550
    },
    {
      "epoch": 8.153915405273438e-06,
      "step": 8550,
      "training_step_time": 17.031046867370605
    },
    {
      "epoch": 8.154869079589844e-06,
      "model_forward_time": 5.596617937088013,
      "step": 8551
    },
    {
      "epoch": 8.154869079589844e-06,
      "step": 8551,
      "training_step_time": 17.030481576919556
    },
    {
      "epoch": 8.15582275390625e-06,
      "model_forward_time": 5.59352445602417,
      "step": 8552
    },
    {
      "epoch": 8.15582275390625e-06,
      "step": 8552,
      "training_step_time": 17.04627299308777
    },
    {
      "epoch": 8.156776428222657e-06,
      "model_forward_time": 5.592141628265381,
      "step": 8553
    },
    {
      "epoch": 8.156776428222657e-06,
      "step": 8553,
      "training_step_time": 17.033365964889526
    },
    {
      "epoch": 8.157730102539063e-06,
      "model_forward_time": 5.96280312538147,
      "step": 8554
    },
    {
      "epoch": 8.157730102539063e-06,
      "step": 8554,
      "training_step_time": 17.39407730102539
    },
    {
      "epoch": 8.158683776855468e-06,
      "model_forward_time": 5.59555983543396,
      "step": 8555
    },
    {
      "epoch": 8.158683776855468e-06,
      "step": 8555,
      "training_step_time": 17.067609786987305
    },
    {
      "epoch": 8.159637451171874e-06,
      "model_forward_time": 5.616756200790405,
      "step": 8556
    },
    {
      "epoch": 8.159637451171874e-06,
      "step": 8556,
      "training_step_time": 17.133846521377563
    },
    {
      "epoch": 8.160591125488282e-06,
      "model_forward_time": 5.62294602394104,
      "step": 8557
    },
    {
      "epoch": 8.160591125488282e-06,
      "step": 8557,
      "training_step_time": 17.125564575195312
    },
    {
      "epoch": 8.161544799804687e-06,
      "model_forward_time": 5.619518280029297,
      "step": 8558
    },
    {
      "epoch": 8.161544799804687e-06,
      "step": 8558,
      "training_step_time": 17.100733041763306
    },
    {
      "epoch": 8.162498474121093e-06,
      "model_forward_time": 5.961393356323242,
      "step": 8559
    },
    {
      "epoch": 8.162498474121093e-06,
      "step": 8559,
      "training_step_time": 17.450758934020996
    },
    {
      "epoch": 8.1634521484375e-06,
      "grad_norm": 0.991624653339386,
      "learning_rate": 1e-05,
      "loss": 0.0565,
      "step": 8560
    },
    {
      "epoch": 8.1634521484375e-06,
      "model_forward_time": 5.611961364746094,
      "step": 8560
    },
    {
      "dynamics_loss_avg": 0.047103670239448545,
      "epoch": 8.1634521484375e-06,
      "step": 8560
    },
    {
      "action_loss_avg": 0.007090485445223749,
      "epoch": 8.1634521484375e-06,
      "step": 8560
    },
    {
      "epoch": 8.1634521484375e-06,
      "step": 8560,
      "training_step_time": 17.09808921813965
    },
    {
      "epoch": 8.164405822753906e-06,
      "model_forward_time": 5.61455512046814,
      "step": 8561
    },
    {
      "epoch": 8.164405822753906e-06,
      "step": 8561,
      "training_step_time": 17.085406064987183
    },
    {
      "epoch": 8.165359497070312e-06,
      "model_forward_time": 5.591908693313599,
      "step": 8562
    },
    {
      "epoch": 8.165359497070312e-06,
      "step": 8562,
      "training_step_time": 17.022403717041016
    },
    {
      "epoch": 8.16631317138672e-06,
      "model_forward_time": 5.570224285125732,
      "step": 8563
    },
    {
      "epoch": 8.16631317138672e-06,
      "step": 8563,
      "training_step_time": 17.006652355194092
    },
    {
      "epoch": 8.167266845703125e-06,
      "model_forward_time": 5.563012599945068,
      "step": 8564
    },
    {
      "epoch": 8.167266845703125e-06,
      "step": 8564,
      "training_step_time": 16.97570776939392
    },
    {
      "epoch": 8.16822052001953e-06,
      "model_forward_time": 5.933864116668701,
      "step": 8565
    },
    {
      "epoch": 8.16822052001953e-06,
      "step": 8565,
      "training_step_time": 17.344834327697754
    },
    {
      "epoch": 8.169174194335938e-06,
      "model_forward_time": 5.568188190460205,
      "step": 8566
    },
    {
      "epoch": 8.169174194335938e-06,
      "step": 8566,
      "training_step_time": 16.997148513793945
    },
    {
      "epoch": 8.170127868652344e-06,
      "model_forward_time": 5.579780101776123,
      "step": 8567
    },
    {
      "epoch": 8.170127868652344e-06,
      "step": 8567,
      "training_step_time": 17.012629985809326
    },
    {
      "epoch": 8.17108154296875e-06,
      "model_forward_time": 5.577488899230957,
      "step": 8568
    },
    {
      "epoch": 8.17108154296875e-06,
      "step": 8568,
      "training_step_time": 16.98701047897339
    },
    {
      "epoch": 8.172035217285157e-06,
      "model_forward_time": 5.5833940505981445,
      "step": 8569
    },
    {
      "epoch": 8.172035217285157e-06,
      "step": 8569,
      "training_step_time": 17.025546550750732
    },
    {
      "epoch": 8.172988891601563e-06,
      "grad_norm": 0.17313669621944427,
      "learning_rate": 1e-05,
      "loss": 0.0754,
      "step": 8570
    },
    {
      "epoch": 8.172988891601563e-06,
      "model_forward_time": 5.953042030334473,
      "step": 8570
    },
    {
      "dynamics_loss_avg": 0.057961118407547474,
      "epoch": 8.172988891601563e-06,
      "step": 8570
    },
    {
      "action_loss_avg": 0.010067356401123106,
      "epoch": 8.172988891601563e-06,
      "step": 8570
    },
    {
      "epoch": 8.172988891601563e-06,
      "step": 8570,
      "training_step_time": 17.427412509918213
    },
    {
      "epoch": 8.173942565917968e-06,
      "model_forward_time": 5.629945755004883,
      "step": 8571
    },
    {
      "epoch": 8.173942565917968e-06,
      "step": 8571,
      "training_step_time": 17.15537667274475
    },
    {
      "epoch": 8.174896240234374e-06,
      "model_forward_time": 5.625718593597412,
      "step": 8572
    },
    {
      "epoch": 8.174896240234374e-06,
      "step": 8572,
      "training_step_time": 17.137580394744873
    },
    {
      "epoch": 8.175849914550782e-06,
      "model_forward_time": 5.609350204467773,
      "step": 8573
    },
    {
      "epoch": 8.175849914550782e-06,
      "step": 8573,
      "training_step_time": 17.087470293045044
    },
    {
      "epoch": 8.176803588867187e-06,
      "model_forward_time": 5.593527555465698,
      "step": 8574
    },
    {
      "epoch": 8.176803588867187e-06,
      "step": 8574,
      "training_step_time": 17.041080474853516
    },
    {
      "epoch": 8.177757263183593e-06,
      "model_forward_time": 5.576102256774902,
      "step": 8575
    },
    {
      "epoch": 8.177757263183593e-06,
      "step": 8575,
      "training_step_time": 17.014018535614014
    },
    {
      "epoch": 8.1787109375e-06,
      "model_forward_time": 5.956808805465698,
      "step": 8576
    },
    {
      "epoch": 8.1787109375e-06,
      "step": 8576,
      "training_step_time": 17.432679891586304
    },
    {
      "epoch": 8.179664611816406e-06,
      "model_forward_time": 5.616759538650513,
      "step": 8577
    },
    {
      "epoch": 8.179664611816406e-06,
      "step": 8577,
      "training_step_time": 17.12014079093933
    },
    {
      "epoch": 8.180618286132812e-06,
      "model_forward_time": 5.6218976974487305,
      "step": 8578
    },
    {
      "epoch": 8.180618286132812e-06,
      "step": 8578,
      "training_step_time": 17.108822107315063
    },
    {
      "epoch": 8.18157196044922e-06,
      "model_forward_time": 5.598302602767944,
      "step": 8579
    },
    {
      "epoch": 8.18157196044922e-06,
      "step": 8579,
      "training_step_time": 17.035794258117676
    },
    {
      "epoch": 8.182525634765625e-06,
      "grad_norm": 1.2065865993499756,
      "learning_rate": 1e-05,
      "loss": 0.0554,
      "step": 8580
    },
    {
      "epoch": 8.182525634765625e-06,
      "model_forward_time": 5.5910866260528564,
      "step": 8580
    },
    {
      "dynamics_loss_avg": 0.0522043539211154,
      "epoch": 8.182525634765625e-06,
      "step": 8580
    },
    {
      "action_loss_avg": 0.007678142283111811,
      "epoch": 8.182525634765625e-06,
      "step": 8580
    },
    {
      "epoch": 8.182525634765625e-06,
      "step": 8580,
      "training_step_time": 17.061836004257202
    },
    {
      "epoch": 8.18347930908203e-06,
      "model_forward_time": 5.94257378578186,
      "step": 8581
    },
    {
      "epoch": 8.18347930908203e-06,
      "step": 8581,
      "training_step_time": 17.413784980773926
    },
    {
      "epoch": 8.184432983398438e-06,
      "model_forward_time": 5.614673852920532,
      "step": 8582
    },
    {
      "epoch": 8.184432983398438e-06,
      "step": 8582,
      "training_step_time": 17.101895093917847
    },
    {
      "epoch": 8.185386657714844e-06,
      "model_forward_time": 5.60404896736145,
      "step": 8583
    },
    {
      "epoch": 8.185386657714844e-06,
      "step": 8583,
      "training_step_time": 17.073418855667114
    },
    {
      "epoch": 8.18634033203125e-06,
      "model_forward_time": 5.5841593742370605,
      "step": 8584
    },
    {
      "epoch": 8.18634033203125e-06,
      "step": 8584,
      "training_step_time": 17.016289949417114
    },
    {
      "epoch": 8.187294006347657e-06,
      "model_forward_time": 5.5815277099609375,
      "step": 8585
    },
    {
      "epoch": 8.187294006347657e-06,
      "step": 8585,
      "training_step_time": 17.01303505897522
    },
    {
      "epoch": 8.188247680664063e-06,
      "model_forward_time": 5.574783086776733,
      "step": 8586
    },
    {
      "epoch": 8.188247680664063e-06,
      "step": 8586,
      "training_step_time": 17.38182520866394
    },
    {
      "epoch": 8.189201354980469e-06,
      "model_forward_time": 5.598727703094482,
      "step": 8587
    },
    {
      "epoch": 8.189201354980469e-06,
      "step": 8587,
      "training_step_time": 17.0483455657959
    },
    {
      "epoch": 8.190155029296874e-06,
      "model_forward_time": 5.601235866546631,
      "step": 8588
    },
    {
      "epoch": 8.190155029296874e-06,
      "step": 8588,
      "training_step_time": 17.07691264152527
    },
    {
      "epoch": 8.191108703613282e-06,
      "model_forward_time": 5.614496946334839,
      "step": 8589
    },
    {
      "epoch": 8.191108703613282e-06,
      "step": 8589,
      "training_step_time": 17.12455439567566
    },
    {
      "epoch": 8.192062377929687e-06,
      "grad_norm": 0.6622998118400574,
      "learning_rate": 1e-05,
      "loss": 0.0594,
      "step": 8590
    },
    {
      "epoch": 8.192062377929687e-06,
      "model_forward_time": 5.62500262260437,
      "step": 8590
    },
    {
      "dynamics_loss_avg": 0.0494084601290524,
      "epoch": 8.192062377929687e-06,
      "step": 8590
    },
    {
      "action_loss_avg": 0.008612803544383495,
      "epoch": 8.192062377929687e-06,
      "step": 8590
    },
    {
      "epoch": 8.192062377929687e-06,
      "step": 8590,
      "training_step_time": 17.128466606140137
    },
    {
      "epoch": 8.193016052246093e-06,
      "model_forward_time": 5.606937408447266,
      "step": 8591
    },
    {
      "epoch": 8.193016052246093e-06,
      "step": 8591,
      "training_step_time": 17.050586223602295
    },
    {
      "epoch": 8.1939697265625e-06,
      "model_forward_time": 5.926827430725098,
      "step": 8592
    },
    {
      "epoch": 8.1939697265625e-06,
      "step": 8592,
      "training_step_time": 17.35403537750244
    },
    {
      "epoch": 8.194923400878906e-06,
      "model_forward_time": 5.573090314865112,
      "step": 8593
    },
    {
      "epoch": 8.194923400878906e-06,
      "step": 8593,
      "training_step_time": 17.006304025650024
    },
    {
      "epoch": 8.195877075195312e-06,
      "model_forward_time": 5.565303325653076,
      "step": 8594
    },
    {
      "epoch": 8.195877075195312e-06,
      "step": 8594,
      "training_step_time": 16.990891218185425
    },
    {
      "epoch": 8.19683074951172e-06,
      "model_forward_time": 5.59742546081543,
      "step": 8595
    },
    {
      "epoch": 8.19683074951172e-06,
      "step": 8595,
      "training_step_time": 17.00775122642517
    },
    {
      "epoch": 8.197784423828125e-06,
      "model_forward_time": 5.584473371505737,
      "step": 8596
    },
    {
      "epoch": 8.197784423828125e-06,
      "step": 8596,
      "training_step_time": 17.032318115234375
    },
    {
      "epoch": 8.19873809814453e-06,
      "model_forward_time": 5.972898483276367,
      "step": 8597
    },
    {
      "epoch": 8.19873809814453e-06,
      "step": 8597,
      "training_step_time": 17.43579387664795
    },
    {
      "epoch": 8.199691772460938e-06,
      "model_forward_time": 5.624596118927002,
      "step": 8598
    },
    {
      "epoch": 8.199691772460938e-06,
      "step": 8598,
      "training_step_time": 17.136909246444702
    },
    {
      "epoch": 8.200645446777344e-06,
      "model_forward_time": 5.636472702026367,
      "step": 8599
    },
    {
      "epoch": 8.200645446777344e-06,
      "step": 8599,
      "training_step_time": 17.153706073760986
    },
    {
      "epoch": 8.20159912109375e-06,
      "grad_norm": 0.9149891138076782,
      "learning_rate": 1e-05,
      "loss": 0.081,
      "step": 8600
    },
    {
      "epoch": 8.20159912109375e-06,
      "model_forward_time": 5.6147589683532715,
      "step": 8600
    },
    {
      "dynamics_loss_avg": 0.07007722426205873,
      "epoch": 8.20159912109375e-06,
      "step": 8600
    },
    {
      "action_loss_avg": 0.012609202228486539,
      "epoch": 8.20159912109375e-06,
      "step": 8600
    },
    {
      "epoch": 8.20159912109375e-06,
      "step": 8600,
      "training_step_time": 17.094828128814697
    },
    {
      "epoch": 8.202552795410157e-06,
      "model_forward_time": 5.610092878341675,
      "step": 8601
    },
    {
      "epoch": 8.202552795410157e-06,
      "step": 8601,
      "training_step_time": 17.107682466506958
    },
    {
      "epoch": 8.203506469726563e-06,
      "model_forward_time": 5.610372066497803,
      "step": 8602
    },
    {
      "epoch": 8.203506469726563e-06,
      "step": 8602,
      "training_step_time": 17.104015827178955
    },
    {
      "epoch": 8.204460144042969e-06,
      "model_forward_time": 5.958344221115112,
      "step": 8603
    },
    {
      "epoch": 8.204460144042969e-06,
      "step": 8603,
      "training_step_time": 17.4416446685791
    },
    {
      "epoch": 8.205413818359374e-06,
      "model_forward_time": 5.608291387557983,
      "step": 8604
    },
    {
      "epoch": 8.205413818359374e-06,
      "step": 8604,
      "training_step_time": 17.096033811569214
    },
    {
      "epoch": 8.206367492675782e-06,
      "model_forward_time": 5.60712194442749,
      "step": 8605
    },
    {
      "epoch": 8.206367492675782e-06,
      "step": 8605,
      "training_step_time": 17.106049299240112
    },
    {
      "epoch": 8.207321166992187e-06,
      "model_forward_time": 5.612981557846069,
      "step": 8606
    },
    {
      "epoch": 8.207321166992187e-06,
      "step": 8606,
      "training_step_time": 17.10648274421692
    },
    {
      "epoch": 8.208274841308593e-06,
      "model_forward_time": 5.613681793212891,
      "step": 8607
    },
    {
      "epoch": 8.208274841308593e-06,
      "step": 8607,
      "training_step_time": 17.108179807662964
    },
    {
      "epoch": 8.209228515625e-06,
      "model_forward_time": 5.971760034561157,
      "step": 8608
    },
    {
      "epoch": 8.209228515625e-06,
      "step": 8608,
      "training_step_time": 17.447259187698364
    },
    {
      "epoch": 8.210182189941406e-06,
      "model_forward_time": 5.6086039543151855,
      "step": 8609
    },
    {
      "epoch": 8.210182189941406e-06,
      "step": 8609,
      "training_step_time": 17.102419137954712
    },
    {
      "epoch": 8.211135864257812e-06,
      "grad_norm": 1.2268816232681274,
      "learning_rate": 1e-05,
      "loss": 0.0538,
      "step": 8610
    },
    {
      "epoch": 8.211135864257812e-06,
      "model_forward_time": 5.60134220123291,
      "step": 8610
    },
    {
      "dynamics_loss_avg": 0.04535695649683476,
      "epoch": 8.211135864257812e-06,
      "step": 8610
    },
    {
      "action_loss_avg": 0.006922890478745103,
      "epoch": 8.211135864257812e-06,
      "step": 8610
    },
    {
      "epoch": 8.211135864257812e-06,
      "step": 8610,
      "training_step_time": 17.07086706161499
    },
    {
      "epoch": 8.21208953857422e-06,
      "model_forward_time": 5.594550848007202,
      "step": 8611
    },
    {
      "epoch": 8.21208953857422e-06,
      "step": 8611,
      "training_step_time": 17.01743173599243
    },
    {
      "epoch": 8.213043212890625e-06,
      "model_forward_time": 5.591876983642578,
      "step": 8612
    },
    {
      "epoch": 8.213043212890625e-06,
      "step": 8612,
      "training_step_time": 17.021984100341797
    },
    {
      "epoch": 8.21399688720703e-06,
      "model_forward_time": 5.610024929046631,
      "step": 8613
    },
    {
      "epoch": 8.21399688720703e-06,
      "step": 8613,
      "training_step_time": 17.084025621414185
    },
    {
      "epoch": 8.214950561523438e-06,
      "model_forward_time": 5.936578273773193,
      "step": 8614
    },
    {
      "epoch": 8.214950561523438e-06,
      "step": 8614,
      "training_step_time": 17.404151678085327
    },
    {
      "epoch": 8.215904235839844e-06,
      "model_forward_time": 5.590555191040039,
      "step": 8615
    },
    {
      "epoch": 8.215904235839844e-06,
      "step": 8615,
      "training_step_time": 17.020567893981934
    },
    {
      "epoch": 8.21685791015625e-06,
      "model_forward_time": 5.589014291763306,
      "step": 8616
    },
    {
      "epoch": 8.21685791015625e-06,
      "step": 8616,
      "training_step_time": 17.05611300468445
    },
    {
      "epoch": 8.217811584472657e-06,
      "model_forward_time": 5.598824501037598,
      "step": 8617
    },
    {
      "epoch": 8.217811584472657e-06,
      "step": 8617,
      "training_step_time": 17.063806772232056
    },
    {
      "epoch": 8.218765258789063e-06,
      "model_forward_time": 5.594843149185181,
      "step": 8618
    },
    {
      "epoch": 8.218765258789063e-06,
      "step": 8618,
      "training_step_time": 17.08296823501587
    },
    {
      "epoch": 8.219718933105469e-06,
      "model_forward_time": 5.979001522064209,
      "step": 8619
    },
    {
      "epoch": 8.219718933105469e-06,
      "step": 8619,
      "training_step_time": 17.435509204864502
    },
    {
      "epoch": 8.220672607421874e-06,
      "grad_norm": 1.3478127717971802,
      "learning_rate": 1e-05,
      "loss": 0.0646,
      "step": 8620
    },
    {
      "epoch": 8.220672607421874e-06,
      "model_forward_time": 5.610292673110962,
      "step": 8620
    },
    {
      "dynamics_loss_avg": 0.05638748276978731,
      "epoch": 8.220672607421874e-06,
      "step": 8620
    },
    {
      "action_loss_avg": 0.009530600137077272,
      "epoch": 8.220672607421874e-06,
      "step": 8620
    },
    {
      "epoch": 8.220672607421874e-06,
      "step": 8620,
      "training_step_time": 17.056234121322632
    },
    {
      "epoch": 8.221626281738282e-06,
      "model_forward_time": 5.6003618240356445,
      "step": 8621
    },
    {
      "epoch": 8.221626281738282e-06,
      "step": 8621,
      "training_step_time": 17.04570746421814
    },
    {
      "epoch": 8.222579956054687e-06,
      "model_forward_time": 5.592588663101196,
      "step": 8622
    },
    {
      "epoch": 8.222579956054687e-06,
      "step": 8622,
      "training_step_time": 17.01697540283203
    },
    {
      "epoch": 8.223533630371093e-06,
      "model_forward_time": 5.590181350708008,
      "step": 8623
    },
    {
      "epoch": 8.223533630371093e-06,
      "step": 8623,
      "training_step_time": 17.02463984489441
    },
    {
      "epoch": 8.2244873046875e-06,
      "model_forward_time": 5.6063292026519775,
      "step": 8624
    },
    {
      "epoch": 8.2244873046875e-06,
      "step": 8624,
      "training_step_time": 17.07618021965027
    },
    {
      "epoch": 8.225440979003906e-06,
      "model_forward_time": 5.948301076889038,
      "step": 8625
    },
    {
      "epoch": 8.225440979003906e-06,
      "step": 8625,
      "training_step_time": 17.432748556137085
    },
    {
      "epoch": 8.226394653320312e-06,
      "model_forward_time": 5.599217176437378,
      "step": 8626
    },
    {
      "epoch": 8.226394653320312e-06,
      "step": 8626,
      "training_step_time": 17.067771434783936
    },
    {
      "epoch": 8.22734832763672e-06,
      "model_forward_time": 5.5967466831207275,
      "step": 8627
    },
    {
      "epoch": 8.22734832763672e-06,
      "step": 8627,
      "training_step_time": 17.0561945438385
    },
    {
      "epoch": 8.228302001953125e-06,
      "model_forward_time": 5.591594934463501,
      "step": 8628
    },
    {
      "epoch": 8.228302001953125e-06,
      "step": 8628,
      "training_step_time": 17.039421319961548
    },
    {
      "epoch": 8.22925567626953e-06,
      "model_forward_time": 5.585577011108398,
      "step": 8629
    },
    {
      "epoch": 8.22925567626953e-06,
      "step": 8629,
      "training_step_time": 17.039220809936523
    },
    {
      "epoch": 8.230209350585938e-06,
      "grad_norm": 0.756191074848175,
      "learning_rate": 1e-05,
      "loss": 0.0757,
      "step": 8630
    },
    {
      "epoch": 8.230209350585938e-06,
      "model_forward_time": 5.957902431488037,
      "step": 8630
    },
    {
      "dynamics_loss_avg": 0.06620299071073532,
      "epoch": 8.230209350585938e-06,
      "step": 8630
    },
    {
      "action_loss_avg": 0.011982432520017027,
      "epoch": 8.230209350585938e-06,
      "step": 8630
    },
    {
      "epoch": 8.230209350585938e-06,
      "step": 8630,
      "training_step_time": 17.435720205307007
    },
    {
      "epoch": 8.231163024902344e-06,
      "model_forward_time": 5.608005523681641,
      "step": 8631
    },
    {
      "epoch": 8.231163024902344e-06,
      "step": 8631,
      "training_step_time": 17.068693161010742
    },
    {
      "epoch": 8.23211669921875e-06,
      "model_forward_time": 5.606885194778442,
      "step": 8632
    },
    {
      "epoch": 8.23211669921875e-06,
      "step": 8632,
      "training_step_time": 17.076145887374878
    },
    {
      "epoch": 8.233070373535155e-06,
      "model_forward_time": 5.600380897521973,
      "step": 8633
    },
    {
      "epoch": 8.233070373535155e-06,
      "step": 8633,
      "training_step_time": 17.082559823989868
    },
    {
      "epoch": 8.234024047851563e-06,
      "model_forward_time": 5.610815048217773,
      "step": 8634
    },
    {
      "epoch": 8.234024047851563e-06,
      "step": 8634,
      "training_step_time": 17.07022190093994
    },
    {
      "epoch": 8.234977722167969e-06,
      "model_forward_time": 5.972718000411987,
      "step": 8635
    },
    {
      "epoch": 8.234977722167969e-06,
      "step": 8635,
      "training_step_time": 17.41206932067871
    },
    {
      "epoch": 8.235931396484374e-06,
      "model_forward_time": 5.562839031219482,
      "step": 8636
    },
    {
      "epoch": 8.235931396484374e-06,
      "step": 8636,
      "training_step_time": 16.978878259658813
    },
    {
      "epoch": 8.236885070800782e-06,
      "model_forward_time": 5.5638182163238525,
      "step": 8637
    },
    {
      "epoch": 8.236885070800782e-06,
      "step": 8637,
      "training_step_time": 16.98074436187744
    },
    {
      "epoch": 8.237838745117187e-06,
      "model_forward_time": 5.571725368499756,
      "step": 8638
    },
    {
      "epoch": 8.237838745117187e-06,
      "step": 8638,
      "training_step_time": 17.003379583358765
    },
    {
      "epoch": 8.238792419433593e-06,
      "model_forward_time": 5.569859981536865,
      "step": 8639
    },
    {
      "epoch": 8.238792419433593e-06,
      "step": 8639,
      "training_step_time": 16.994942665100098
    },
    {
      "epoch": 8.23974609375e-06,
      "grad_norm": 0.812819242477417,
      "learning_rate": 1e-05,
      "loss": 0.072,
      "step": 8640
    },
    {
      "epoch": 8.23974609375e-06,
      "model_forward_time": 5.566317319869995,
      "step": 8640
    },
    {
      "dynamics_loss_avg": 0.060299486853182316,
      "epoch": 8.23974609375e-06,
      "step": 8640
    },
    {
      "action_loss_avg": 0.010465050768107175,
      "epoch": 8.23974609375e-06,
      "step": 8640
    },
    {
      "epoch": 8.23974609375e-06,
      "step": 8640,
      "training_step_time": 16.96695327758789
    },
    {
      "epoch": 8.240699768066406e-06,
      "model_forward_time": 5.938465118408203,
      "step": 8641
    },
    {
      "epoch": 8.240699768066406e-06,
      "step": 8641,
      "training_step_time": 17.346473217010498
    },
    {
      "epoch": 8.241653442382812e-06,
      "model_forward_time": 5.600651264190674,
      "step": 8642
    },
    {
      "epoch": 8.241653442382812e-06,
      "step": 8642,
      "training_step_time": 17.09771418571472
    },
    {
      "epoch": 8.24260711669922e-06,
      "model_forward_time": 5.626720905303955,
      "step": 8643
    },
    {
      "epoch": 8.24260711669922e-06,
      "step": 8643,
      "training_step_time": 17.119004487991333
    },
    {
      "epoch": 8.243560791015625e-06,
      "model_forward_time": 5.612068176269531,
      "step": 8644
    },
    {
      "epoch": 8.243560791015625e-06,
      "step": 8644,
      "training_step_time": 17.08045721054077
    },
    {
      "epoch": 8.24451446533203e-06,
      "model_forward_time": 5.5933678150177,
      "step": 8645
    },
    {
      "epoch": 8.24451446533203e-06,
      "step": 8645,
      "training_step_time": 17.03056263923645
    },
    {
      "epoch": 8.245468139648438e-06,
      "model_forward_time": 5.597155570983887,
      "step": 8646
    },
    {
      "epoch": 8.245468139648438e-06,
      "step": 8646,
      "training_step_time": 17.395955324172974
    },
    {
      "epoch": 8.246421813964844e-06,
      "model_forward_time": 5.583480358123779,
      "step": 8647
    },
    {
      "epoch": 8.246421813964844e-06,
      "step": 8647,
      "training_step_time": 17.036243677139282
    },
    {
      "epoch": 8.24737548828125e-06,
      "model_forward_time": 5.602985143661499,
      "step": 8648
    },
    {
      "epoch": 8.24737548828125e-06,
      "step": 8648,
      "training_step_time": 17.086966037750244
    },
    {
      "epoch": 8.248329162597655e-06,
      "model_forward_time": 5.60813307762146,
      "step": 8649
    },
    {
      "epoch": 8.248329162597655e-06,
      "step": 8649,
      "training_step_time": 17.068272590637207
    },
    {
      "epoch": 8.249282836914063e-06,
      "grad_norm": 1.0388686656951904,
      "learning_rate": 1e-05,
      "loss": 0.0858,
      "step": 8650
    },
    {
      "epoch": 8.249282836914063e-06,
      "model_forward_time": 5.592919111251831,
      "step": 8650
    },
    {
      "dynamics_loss_avg": 0.07323386017233133,
      "epoch": 8.249282836914063e-06,
      "step": 8650
    },
    {
      "action_loss_avg": 0.01292649400420487,
      "epoch": 8.249282836914063e-06,
      "step": 8650
    },
    {
      "epoch": 8.249282836914063e-06,
      "step": 8650,
      "training_step_time": 17.033114194869995
    },
    {
      "epoch": 8.250236511230469e-06,
      "model_forward_time": 5.578831434249878,
      "step": 8651
    },
    {
      "epoch": 8.250236511230469e-06,
      "step": 8651,
      "training_step_time": 16.998748064041138
    },
    {
      "epoch": 8.251190185546874e-06,
      "model_forward_time": 5.951453685760498,
      "step": 8652
    },
    {
      "epoch": 8.251190185546874e-06,
      "step": 8652,
      "training_step_time": 17.386788368225098
    },
    {
      "epoch": 8.252143859863282e-06,
      "model_forward_time": 5.587330102920532,
      "step": 8653
    },
    {
      "epoch": 8.252143859863282e-06,
      "step": 8653,
      "training_step_time": 17.033772945404053
    },
    {
      "epoch": 8.253097534179687e-06,
      "model_forward_time": 5.594009160995483,
      "step": 8654
    },
    {
      "epoch": 8.253097534179687e-06,
      "step": 8654,
      "training_step_time": 17.06234312057495
    },
    {
      "epoch": 8.254051208496093e-06,
      "model_forward_time": 5.599327564239502,
      "step": 8655
    },
    {
      "epoch": 8.254051208496093e-06,
      "step": 8655,
      "training_step_time": 17.058679580688477
    },
    {
      "epoch": 8.2550048828125e-06,
      "model_forward_time": 5.621274471282959,
      "step": 8656
    },
    {
      "epoch": 8.2550048828125e-06,
      "step": 8656,
      "training_step_time": 17.074925899505615
    },
    {
      "epoch": 8.255958557128906e-06,
      "model_forward_time": 5.649382829666138,
      "step": 8657
    },
    {
      "epoch": 8.255958557128906e-06,
      "step": 8657,
      "training_step_time": 17.497602224349976
    },
    {
      "epoch": 8.256912231445312e-06,
      "model_forward_time": 5.619831085205078,
      "step": 8658
    },
    {
      "epoch": 8.256912231445312e-06,
      "step": 8658,
      "training_step_time": 17.08065128326416
    },
    {
      "epoch": 8.25786590576172e-06,
      "model_forward_time": 5.6092329025268555,
      "step": 8659
    },
    {
      "epoch": 8.25786590576172e-06,
      "step": 8659,
      "training_step_time": 17.086185693740845
    },
    {
      "epoch": 8.258819580078125e-06,
      "grad_norm": 2.00069522857666,
      "learning_rate": 1e-05,
      "loss": 0.0735,
      "step": 8660
    },
    {
      "epoch": 8.258819580078125e-06,
      "model_forward_time": 5.609316825866699,
      "step": 8660
    },
    {
      "dynamics_loss_avg": 0.06266700271517038,
      "epoch": 8.258819580078125e-06,
      "step": 8660
    },
    {
      "action_loss_avg": 0.010368185304105281,
      "epoch": 8.258819580078125e-06,
      "step": 8660
    },
    {
      "epoch": 8.258819580078125e-06,
      "step": 8660,
      "training_step_time": 17.091388463974
    },
    {
      "epoch": 8.259773254394531e-06,
      "model_forward_time": 5.600295543670654,
      "step": 8661
    },
    {
      "epoch": 8.259773254394531e-06,
      "step": 8661,
      "training_step_time": 17.038503885269165
    },
    {
      "epoch": 8.260726928710938e-06,
      "model_forward_time": 5.57863974571228,
      "step": 8662
    },
    {
      "epoch": 8.260726928710938e-06,
      "step": 8662,
      "training_step_time": 17.01323413848877
    },
    {
      "epoch": 8.261680603027344e-06,
      "model_forward_time": 5.95865797996521,
      "step": 8663
    },
    {
      "epoch": 8.261680603027344e-06,
      "step": 8663,
      "training_step_time": 17.405315160751343
    },
    {
      "epoch": 8.26263427734375e-06,
      "model_forward_time": 5.599136114120483,
      "step": 8664
    },
    {
      "epoch": 8.26263427734375e-06,
      "step": 8664,
      "training_step_time": 17.086616039276123
    },
    {
      "epoch": 8.263587951660155e-06,
      "model_forward_time": 5.612540006637573,
      "step": 8665
    },
    {
      "epoch": 8.263587951660155e-06,
      "step": 8665,
      "training_step_time": 17.10068130493164
    },
    {
      "epoch": 8.264541625976563e-06,
      "model_forward_time": 5.611124038696289,
      "step": 8666
    },
    {
      "epoch": 8.264541625976563e-06,
      "step": 8666,
      "training_step_time": 17.125964879989624
    },
    {
      "epoch": 8.265495300292969e-06,
      "model_forward_time": 5.623513460159302,
      "step": 8667
    },
    {
      "epoch": 8.265495300292969e-06,
      "step": 8667,
      "training_step_time": 17.10078263282776
    },
    {
      "epoch": 8.266448974609374e-06,
      "model_forward_time": 5.961509466171265,
      "step": 8668
    },
    {
      "epoch": 8.266448974609374e-06,
      "step": 8668,
      "training_step_time": 17.415599584579468
    },
    {
      "epoch": 8.267402648925782e-06,
      "model_forward_time": 5.594784259796143,
      "step": 8669
    },
    {
      "epoch": 8.267402648925782e-06,
      "step": 8669,
      "training_step_time": 17.041806936264038
    },
    {
      "epoch": 8.268356323242187e-06,
      "grad_norm": 0.7961212992668152,
      "learning_rate": 1e-05,
      "loss": 0.0838,
      "step": 8670
    },
    {
      "epoch": 8.268356323242187e-06,
      "model_forward_time": 5.600067138671875,
      "step": 8670
    },
    {
      "dynamics_loss_avg": 0.07676571942865848,
      "epoch": 8.268356323242187e-06,
      "step": 8670
    },
    {
      "action_loss_avg": 0.011480185203254223,
      "epoch": 8.268356323242187e-06,
      "step": 8670
    },
    {
      "epoch": 8.268356323242187e-06,
      "step": 8670,
      "training_step_time": 17.095550298690796
    },
    {
      "epoch": 8.269309997558593e-06,
      "model_forward_time": 5.62409520149231,
      "step": 8671
    },
    {
      "epoch": 8.269309997558593e-06,
      "step": 8671,
      "training_step_time": 17.144129276275635
    },
    {
      "epoch": 8.270263671875e-06,
      "model_forward_time": 5.637516498565674,
      "step": 8672
    },
    {
      "epoch": 8.270263671875e-06,
      "step": 8672,
      "training_step_time": 17.127840518951416
    },
    {
      "epoch": 8.271217346191406e-06,
      "model_forward_time": 5.613146543502808,
      "step": 8673
    },
    {
      "epoch": 8.271217346191406e-06,
      "step": 8673,
      "training_step_time": 17.07020926475525
    },
    {
      "epoch": 8.272171020507812e-06,
      "model_forward_time": 5.951507806777954,
      "step": 8674
    },
    {
      "epoch": 8.272171020507812e-06,
      "step": 8674,
      "training_step_time": 17.381597995758057
    },
    {
      "epoch": 8.27312469482422e-06,
      "model_forward_time": 5.596278429031372,
      "step": 8675
    },
    {
      "epoch": 8.27312469482422e-06,
      "step": 8675,
      "training_step_time": 17.07856583595276
    },
    {
      "epoch": 8.274078369140625e-06,
      "model_forward_time": 5.6054089069366455,
      "step": 8676
    },
    {
      "epoch": 8.274078369140625e-06,
      "step": 8676,
      "training_step_time": 17.10336208343506
    },
    {
      "epoch": 8.275032043457031e-06,
      "model_forward_time": 5.633947849273682,
      "step": 8677
    },
    {
      "epoch": 8.275032043457031e-06,
      "step": 8677,
      "training_step_time": 17.108325719833374
    },
    {
      "epoch": 8.275985717773438e-06,
      "model_forward_time": 5.595963716506958,
      "step": 8678
    },
    {
      "epoch": 8.275985717773438e-06,
      "step": 8678,
      "training_step_time": 17.045307159423828
    },
    {
      "epoch": 8.276939392089844e-06,
      "model_forward_time": 5.573811054229736,
      "step": 8679
    },
    {
      "epoch": 8.276939392089844e-06,
      "step": 8679,
      "training_step_time": 17.353294610977173
    },
    {
      "epoch": 8.27789306640625e-06,
      "grad_norm": 1.3388649225234985,
      "learning_rate": 1e-05,
      "loss": 0.0597,
      "step": 8680
    },
    {
      "epoch": 8.27789306640625e-06,
      "model_forward_time": 5.560093402862549,
      "step": 8680
    },
    {
      "dynamics_loss_avg": 0.052086267806589606,
      "epoch": 8.27789306640625e-06,
      "step": 8680
    },
    {
      "action_loss_avg": 0.00709876308683306,
      "epoch": 8.27789306640625e-06,
      "step": 8680
    },
    {
      "epoch": 8.27789306640625e-06,
      "step": 8680,
      "training_step_time": 16.976643800735474
    },
    {
      "epoch": 8.278846740722655e-06,
      "model_forward_time": 5.565753221511841,
      "step": 8681
    },
    {
      "epoch": 8.278846740722655e-06,
      "step": 8681,
      "training_step_time": 17.00368094444275
    },
    {
      "epoch": 8.279800415039063e-06,
      "model_forward_time": 5.5767130851745605,
      "step": 8682
    },
    {
      "epoch": 8.279800415039063e-06,
      "step": 8682,
      "training_step_time": 17.00127100944519
    },
    {
      "epoch": 8.280754089355469e-06,
      "model_forward_time": 5.586125612258911,
      "step": 8683
    },
    {
      "epoch": 8.280754089355469e-06,
      "step": 8683,
      "training_step_time": 17.01547336578369
    },
    {
      "epoch": 8.281707763671874e-06,
      "model_forward_time": 5.588072299957275,
      "step": 8684
    },
    {
      "epoch": 8.281707763671874e-06,
      "step": 8684,
      "training_step_time": 17.03786039352417
    },
    {
      "epoch": 8.282661437988282e-06,
      "model_forward_time": 5.959752082824707,
      "step": 8685
    },
    {
      "epoch": 8.282661437988282e-06,
      "step": 8685,
      "training_step_time": 17.416234016418457
    },
    {
      "epoch": 8.283615112304687e-06,
      "model_forward_time": 5.601458787918091,
      "step": 8686
    },
    {
      "epoch": 8.283615112304687e-06,
      "step": 8686,
      "training_step_time": 17.090420961380005
    },
    {
      "epoch": 8.284568786621093e-06,
      "model_forward_time": 5.604886770248413,
      "step": 8687
    },
    {
      "epoch": 8.284568786621093e-06,
      "step": 8687,
      "training_step_time": 17.08353042602539
    },
    {
      "epoch": 8.2855224609375e-06,
      "model_forward_time": 5.608678817749023,
      "step": 8688
    },
    {
      "epoch": 8.2855224609375e-06,
      "step": 8688,
      "training_step_time": 17.081904411315918
    },
    {
      "epoch": 8.286476135253906e-06,
      "model_forward_time": 5.600611209869385,
      "step": 8689
    },
    {
      "epoch": 8.286476135253906e-06,
      "step": 8689,
      "training_step_time": 17.05282688140869
    },
    {
      "epoch": 8.287429809570312e-06,
      "grad_norm": 0.5583938360214233,
      "learning_rate": 1e-05,
      "loss": 0.0618,
      "step": 8690
    },
    {
      "epoch": 8.287429809570312e-06,
      "model_forward_time": 5.5977184772491455,
      "step": 8690
    },
    {
      "dynamics_loss_avg": 0.04990868270397186,
      "epoch": 8.287429809570312e-06,
      "step": 8690
    },
    {
      "action_loss_avg": 0.007668849965557456,
      "epoch": 8.287429809570312e-06,
      "step": 8690
    },
    {
      "epoch": 8.287429809570312e-06,
      "step": 8690,
      "training_step_time": 17.41007161140442
    },
    {
      "epoch": 8.28838348388672e-06,
      "model_forward_time": 5.597391843795776,
      "step": 8691
    },
    {
      "epoch": 8.28838348388672e-06,
      "step": 8691,
      "training_step_time": 17.052130699157715
    },
    {
      "epoch": 8.289337158203125e-06,
      "model_forward_time": 5.592768430709839,
      "step": 8692
    },
    {
      "epoch": 8.289337158203125e-06,
      "step": 8692,
      "training_step_time": 17.04451608657837
    },
    {
      "epoch": 8.290290832519531e-06,
      "model_forward_time": 5.597449779510498,
      "step": 8693
    },
    {
      "epoch": 8.290290832519531e-06,
      "step": 8693,
      "training_step_time": 17.061769723892212
    },
    {
      "epoch": 8.291244506835938e-06,
      "model_forward_time": 5.608414649963379,
      "step": 8694
    },
    {
      "epoch": 8.291244506835938e-06,
      "step": 8694,
      "training_step_time": 17.101566314697266
    },
    {
      "epoch": 8.292198181152344e-06,
      "model_forward_time": 5.607006072998047,
      "step": 8695
    },
    {
      "epoch": 8.292198181152344e-06,
      "step": 8695,
      "training_step_time": 17.107442378997803
    },
    {
      "epoch": 8.29315185546875e-06,
      "model_forward_time": 5.987209320068359,
      "step": 8696
    },
    {
      "epoch": 8.29315185546875e-06,
      "step": 8696,
      "training_step_time": 17.463299036026
    },
    {
      "epoch": 8.294105529785155e-06,
      "model_forward_time": 5.614689350128174,
      "step": 8697
    },
    {
      "epoch": 8.294105529785155e-06,
      "step": 8697,
      "training_step_time": 17.101131439208984
    },
    {
      "epoch": 8.295059204101563e-06,
      "model_forward_time": 5.60424542427063,
      "step": 8698
    },
    {
      "epoch": 8.295059204101563e-06,
      "step": 8698,
      "training_step_time": 17.083971977233887
    },
    {
      "epoch": 8.296012878417969e-06,
      "model_forward_time": 5.612869024276733,
      "step": 8699
    },
    {
      "epoch": 8.296012878417969e-06,
      "step": 8699,
      "training_step_time": 17.091495752334595
    },
    {
      "epoch": 8.296966552734374e-06,
      "grad_norm": 0.8886074423789978,
      "learning_rate": 1e-05,
      "loss": 0.0618,
      "step": 8700
    },
    {
      "epoch": 8.296966552734374e-06,
      "model_forward_time": 5.595188856124878,
      "step": 8700
    },
    {
      "dynamics_loss_avg": 0.05209971033036709,
      "epoch": 8.296966552734374e-06,
      "step": 8700
    },
    {
      "action_loss_avg": 0.00948849250562489,
      "epoch": 8.296966552734374e-06,
      "step": 8700
    },
    {
      "epoch": 8.296966552734374e-06,
      "step": 8700,
      "training_step_time": 17.03366780281067
    },
    {
      "epoch": 8.297920227050782e-06,
      "model_forward_time": 5.958622217178345,
      "step": 8701
    },
    {
      "epoch": 8.297920227050782e-06,
      "step": 8701,
      "training_step_time": 17.399895191192627
    },
    {
      "epoch": 8.298873901367187e-06,
      "model_forward_time": 5.590358734130859,
      "step": 8702
    },
    {
      "epoch": 8.298873901367187e-06,
      "step": 8702,
      "training_step_time": 17.0482439994812
    },
    {
      "epoch": 8.299827575683593e-06,
      "model_forward_time": 5.589059114456177,
      "step": 8703
    },
    {
      "epoch": 8.299827575683593e-06,
      "step": 8703,
      "training_step_time": 17.026031732559204
    },
    {
      "epoch": 8.30078125e-06,
      "model_forward_time": 5.596062898635864,
      "step": 8704
    },
    {
      "epoch": 8.30078125e-06,
      "step": 8704,
      "training_step_time": 17.020777463912964
    },
    {
      "epoch": 8.301734924316406e-06,
      "model_forward_time": 5.593656301498413,
      "step": 8705
    },
    {
      "epoch": 8.301734924316406e-06,
      "step": 8705,
      "training_step_time": 17.039840698242188
    },
    {
      "epoch": 8.302688598632812e-06,
      "model_forward_time": 5.592612028121948,
      "step": 8706
    },
    {
      "epoch": 8.302688598632812e-06,
      "step": 8706,
      "training_step_time": 17.05025577545166
    },
    {
      "epoch": 8.30364227294922e-06,
      "model_forward_time": 5.954177141189575,
      "step": 8707
    },
    {
      "epoch": 8.30364227294922e-06,
      "step": 8707,
      "training_step_time": 17.400017976760864
    },
    {
      "epoch": 8.304595947265625e-06,
      "model_forward_time": 5.600896120071411,
      "step": 8708
    },
    {
      "epoch": 8.304595947265625e-06,
      "step": 8708,
      "training_step_time": 17.074547290802002
    },
    {
      "epoch": 8.305549621582031e-06,
      "model_forward_time": 5.600255012512207,
      "step": 8709
    },
    {
      "epoch": 8.305549621582031e-06,
      "step": 8709,
      "training_step_time": 17.069663763046265
    },
    {
      "epoch": 8.306503295898438e-06,
      "grad_norm": 0.6963316798210144,
      "learning_rate": 1e-05,
      "loss": 0.0663,
      "step": 8710
    },
    {
      "epoch": 8.306503295898438e-06,
      "model_forward_time": 5.598681688308716,
      "step": 8710
    },
    {
      "dynamics_loss_avg": 0.057717140391469,
      "epoch": 8.306503295898438e-06,
      "step": 8710
    },
    {
      "action_loss_avg": 0.010535996523685753,
      "epoch": 8.306503295898438e-06,
      "step": 8710
    },
    {
      "epoch": 8.306503295898438e-06,
      "step": 8710,
      "training_step_time": 17.053001642227173
    },
    {
      "epoch": 8.307456970214844e-06,
      "model_forward_time": 5.591414451599121,
      "step": 8711
    },
    {
      "epoch": 8.307456970214844e-06,
      "step": 8711,
      "training_step_time": 17.059468269348145
    },
    {
      "epoch": 8.30841064453125e-06,
      "model_forward_time": 5.6081976890563965,
      "step": 8712
    },
    {
      "epoch": 8.30841064453125e-06,
      "step": 8712,
      "training_step_time": 17.45845079421997
    },
    {
      "epoch": 8.309364318847655e-06,
      "model_forward_time": 5.600618839263916,
      "step": 8713
    },
    {
      "epoch": 8.309364318847655e-06,
      "step": 8713,
      "training_step_time": 17.076146841049194
    },
    {
      "epoch": 8.310317993164063e-06,
      "model_forward_time": 5.606768608093262,
      "step": 8714
    },
    {
      "epoch": 8.310317993164063e-06,
      "step": 8714,
      "training_step_time": 17.07365322113037
    },
    {
      "epoch": 8.311271667480469e-06,
      "model_forward_time": 5.600066184997559,
      "step": 8715
    },
    {
      "epoch": 8.311271667480469e-06,
      "step": 8715,
      "training_step_time": 17.037904739379883
    },
    {
      "epoch": 8.312225341796874e-06,
      "model_forward_time": 5.58490777015686,
      "step": 8716
    },
    {
      "epoch": 8.312225341796874e-06,
      "step": 8716,
      "training_step_time": 17.025805234909058
    },
    {
      "epoch": 8.313179016113282e-06,
      "model_forward_time": 5.590355157852173,
      "step": 8717
    },
    {
      "epoch": 8.313179016113282e-06,
      "step": 8717,
      "training_step_time": 17.018569469451904
    },
    {
      "epoch": 8.314132690429687e-06,
      "model_forward_time": 5.953038454055786,
      "step": 8718
    },
    {
      "epoch": 8.314132690429687e-06,
      "step": 8718,
      "training_step_time": 17.417572498321533
    },
    {
      "epoch": 8.315086364746093e-06,
      "model_forward_time": 5.600747585296631,
      "step": 8719
    },
    {
      "epoch": 8.315086364746093e-06,
      "step": 8719,
      "training_step_time": 17.07726502418518
    },
    {
      "epoch": 8.3160400390625e-06,
      "grad_norm": 1.0466442108154297,
      "learning_rate": 1e-05,
      "loss": 0.0675,
      "step": 8720
    },
    {
      "epoch": 8.3160400390625e-06,
      "model_forward_time": 5.607909440994263,
      "step": 8720
    },
    {
      "dynamics_loss_avg": 0.055261217430233954,
      "epoch": 8.3160400390625e-06,
      "step": 8720
    },
    {
      "action_loss_avg": 0.008044085442088544,
      "epoch": 8.3160400390625e-06,
      "step": 8720
    },
    {
      "epoch": 8.3160400390625e-06,
      "step": 8720,
      "training_step_time": 17.085996389389038
    },
    {
      "epoch": 8.316993713378906e-06,
      "model_forward_time": 5.602849721908569,
      "step": 8721
    },
    {
      "epoch": 8.316993713378906e-06,
      "step": 8721,
      "training_step_time": 17.068124294281006
    },
    {
      "epoch": 8.317947387695312e-06,
      "model_forward_time": 5.603904485702515,
      "step": 8722
    },
    {
      "epoch": 8.317947387695312e-06,
      "step": 8722,
      "training_step_time": 17.047484159469604
    },
    {
      "epoch": 8.31890106201172e-06,
      "model_forward_time": 5.5855796337127686,
      "step": 8723
    },
    {
      "epoch": 8.31890106201172e-06,
      "step": 8723,
      "training_step_time": 17.384001970291138
    },
    {
      "epoch": 8.319854736328125e-06,
      "model_forward_time": 5.592553377151489,
      "step": 8724
    },
    {
      "epoch": 8.319854736328125e-06,
      "step": 8724,
      "training_step_time": 17.05005955696106
    },
    {
      "epoch": 8.320808410644531e-06,
      "model_forward_time": 5.615713119506836,
      "step": 8725
    },
    {
      "epoch": 8.320808410644531e-06,
      "step": 8725,
      "training_step_time": 17.0943820476532
    },
    {
      "epoch": 8.321762084960938e-06,
      "model_forward_time": 5.607238054275513,
      "step": 8726
    },
    {
      "epoch": 8.321762084960938e-06,
      "step": 8726,
      "training_step_time": 17.10046672821045
    },
    {
      "epoch": 8.322715759277344e-06,
      "model_forward_time": 5.618114471435547,
      "step": 8727
    },
    {
      "epoch": 8.322715759277344e-06,
      "step": 8727,
      "training_step_time": 17.1114022731781
    },
    {
      "epoch": 8.32366943359375e-06,
      "model_forward_time": 5.61569356918335,
      "step": 8728
    },
    {
      "epoch": 8.32366943359375e-06,
      "step": 8728,
      "training_step_time": 17.110579013824463
    },
    {
      "epoch": 8.324623107910155e-06,
      "model_forward_time": 5.979060888290405,
      "step": 8729
    },
    {
      "epoch": 8.324623107910155e-06,
      "step": 8729,
      "training_step_time": 17.473127603530884
    },
    {
      "epoch": 8.325576782226563e-06,
      "grad_norm": 0.7301215529441833,
      "learning_rate": 1e-05,
      "loss": 0.059,
      "step": 8730
    },
    {
      "epoch": 8.325576782226563e-06,
      "model_forward_time": 5.610628128051758,
      "step": 8730
    },
    {
      "dynamics_loss_avg": 0.05192612148821354,
      "epoch": 8.325576782226563e-06,
      "step": 8730
    },
    {
      "action_loss_avg": 0.007028412027284503,
      "epoch": 8.325576782226563e-06,
      "step": 8730
    },
    {
      "epoch": 8.325576782226563e-06,
      "step": 8730,
      "training_step_time": 17.0967538356781
    },
    {
      "epoch": 8.326530456542969e-06,
      "model_forward_time": 5.604424238204956,
      "step": 8731
    },
    {
      "epoch": 8.326530456542969e-06,
      "step": 8731,
      "training_step_time": 17.092769145965576
    },
    {
      "epoch": 8.327484130859374e-06,
      "model_forward_time": 5.613113164901733,
      "step": 8732
    },
    {
      "epoch": 8.327484130859374e-06,
      "step": 8732,
      "training_step_time": 17.096751928329468
    },
    {
      "epoch": 8.328437805175782e-06,
      "model_forward_time": 5.6076953411102295,
      "step": 8733
    },
    {
      "epoch": 8.328437805175782e-06,
      "step": 8733,
      "training_step_time": 17.0806667804718
    },
    {
      "epoch": 8.329391479492187e-06,
      "model_forward_time": 5.973634243011475,
      "step": 8734
    },
    {
      "epoch": 8.329391479492187e-06,
      "step": 8734,
      "training_step_time": 17.44073247909546
    },
    {
      "epoch": 8.330345153808593e-06,
      "model_forward_time": 5.607083797454834,
      "step": 8735
    },
    {
      "epoch": 8.330345153808593e-06,
      "step": 8735,
      "training_step_time": 17.0951931476593
    },
    {
      "epoch": 8.331298828125e-06,
      "model_forward_time": 5.613893270492554,
      "step": 8736
    },
    {
      "epoch": 8.331298828125e-06,
      "step": 8736,
      "training_step_time": 17.10488533973694
    },
    {
      "epoch": 8.332252502441406e-06,
      "model_forward_time": 5.605955600738525,
      "step": 8737
    },
    {
      "epoch": 8.332252502441406e-06,
      "step": 8737,
      "training_step_time": 17.109803676605225
    },
    {
      "epoch": 8.333206176757812e-06,
      "model_forward_time": 5.616274118423462,
      "step": 8738
    },
    {
      "epoch": 8.333206176757812e-06,
      "step": 8738,
      "training_step_time": 17.10240602493286
    },
    {
      "epoch": 8.33415985107422e-06,
      "model_forward_time": 5.6008687019348145,
      "step": 8739
    },
    {
      "epoch": 8.33415985107422e-06,
      "step": 8739,
      "training_step_time": 17.03835439682007
    },
    {
      "epoch": 8.335113525390625e-06,
      "grad_norm": 0.8187486529350281,
      "learning_rate": 1e-05,
      "loss": 0.0679,
      "step": 8740
    },
    {
      "epoch": 8.335113525390625e-06,
      "model_forward_time": 5.95419979095459,
      "step": 8740
    },
    {
      "dynamics_loss_avg": 0.0588810482993722,
      "epoch": 8.335113525390625e-06,
      "step": 8740
    },
    {
      "action_loss_avg": 0.009805896249599756,
      "epoch": 8.335113525390625e-06,
      "step": 8740
    },
    {
      "epoch": 8.335113525390625e-06,
      "step": 8740,
      "training_step_time": 17.361974716186523
    },
    {
      "epoch": 8.336067199707031e-06,
      "model_forward_time": 5.574468612670898,
      "step": 8741
    },
    {
      "epoch": 8.336067199707031e-06,
      "step": 8741,
      "training_step_time": 17.002755403518677
    },
    {
      "epoch": 8.337020874023438e-06,
      "model_forward_time": 5.57545804977417,
      "step": 8742
    },
    {
      "epoch": 8.337020874023438e-06,
      "step": 8742,
      "training_step_time": 16.998027324676514
    },
    {
      "epoch": 8.337974548339844e-06,
      "model_forward_time": 5.581047534942627,
      "step": 8743
    },
    {
      "epoch": 8.337974548339844e-06,
      "step": 8743,
      "training_step_time": 16.998369932174683
    },
    {
      "epoch": 8.33892822265625e-06,
      "model_forward_time": 5.570854187011719,
      "step": 8744
    },
    {
      "epoch": 8.33892822265625e-06,
      "step": 8744,
      "training_step_time": 17.008036851882935
    },
    {
      "epoch": 8.339881896972656e-06,
      "model_forward_time": 5.596385955810547,
      "step": 8745
    },
    {
      "epoch": 8.339881896972656e-06,
      "step": 8745,
      "training_step_time": 17.396531343460083
    },
    {
      "epoch": 8.340835571289063e-06,
      "model_forward_time": 5.599341869354248,
      "step": 8746
    },
    {
      "epoch": 8.340835571289063e-06,
      "step": 8746,
      "training_step_time": 17.073978424072266
    },
    {
      "epoch": 8.341789245605469e-06,
      "model_forward_time": 5.623422384262085,
      "step": 8747
    },
    {
      "epoch": 8.341789245605469e-06,
      "step": 8747,
      "training_step_time": 17.121284246444702
    },
    {
      "epoch": 8.342742919921874e-06,
      "model_forward_time": 5.6229846477508545,
      "step": 8748
    },
    {
      "epoch": 8.342742919921874e-06,
      "step": 8748,
      "training_step_time": 17.113072395324707
    },
    {
      "epoch": 8.343696594238282e-06,
      "model_forward_time": 5.604856729507446,
      "step": 8749
    },
    {
      "epoch": 8.343696594238282e-06,
      "step": 8749,
      "training_step_time": 17.034637928009033
    },
    {
      "epoch": 8.344650268554688e-06,
      "grad_norm": 0.5801685452461243,
      "learning_rate": 1e-05,
      "loss": 0.0763,
      "step": 8750
    },
    {
      "epoch": 8.344650268554688e-06,
      "model_forward_time": 5.594212293624878,
      "step": 8750
    },
    {
      "dynamics_loss_avg": 0.06237080506980419,
      "epoch": 8.344650268554688e-06,
      "step": 8750
    },
    {
      "action_loss_avg": 0.011770057468675077,
      "epoch": 8.344650268554688e-06,
      "step": 8750
    },
    {
      "epoch": 8.344650268554688e-06,
      "step": 8750,
      "training_step_time": 17.058865308761597
    },
    {
      "epoch": 8.345603942871093e-06,
      "model_forward_time": 5.967270374298096,
      "step": 8751
    },
    {
      "epoch": 8.345603942871093e-06,
      "step": 8751,
      "training_step_time": 17.40088391304016
    },
    {
      "epoch": 8.3465576171875e-06,
      "model_forward_time": 5.593080282211304,
      "step": 8752
    },
    {
      "epoch": 8.3465576171875e-06,
      "step": 8752,
      "training_step_time": 17.050698280334473
    },
    {
      "epoch": 8.347511291503906e-06,
      "model_forward_time": 5.5909423828125,
      "step": 8753
    },
    {
      "epoch": 8.347511291503906e-06,
      "step": 8753,
      "training_step_time": 17.0767023563385
    },
    {
      "epoch": 8.348464965820312e-06,
      "model_forward_time": 5.611358165740967,
      "step": 8754
    },
    {
      "epoch": 8.348464965820312e-06,
      "step": 8754,
      "training_step_time": 17.085999488830566
    },
    {
      "epoch": 8.34941864013672e-06,
      "model_forward_time": 5.601254224777222,
      "step": 8755
    },
    {
      "epoch": 8.34941864013672e-06,
      "step": 8755,
      "training_step_time": 17.02950882911682
    },
    {
      "epoch": 8.350372314453125e-06,
      "model_forward_time": 5.577353477478027,
      "step": 8756
    },
    {
      "epoch": 8.350372314453125e-06,
      "step": 8756,
      "training_step_time": 17.372418880462646
    },
    {
      "epoch": 8.351325988769531e-06,
      "model_forward_time": 5.5806708335876465,
      "step": 8757
    },
    {
      "epoch": 8.351325988769531e-06,
      "step": 8757,
      "training_step_time": 17.03058409690857
    },
    {
      "epoch": 8.352279663085937e-06,
      "model_forward_time": 5.601402997970581,
      "step": 8758
    },
    {
      "epoch": 8.352279663085937e-06,
      "step": 8758,
      "training_step_time": 17.069910526275635
    },
    {
      "epoch": 8.353233337402344e-06,
      "model_forward_time": 5.608303070068359,
      "step": 8759
    },
    {
      "epoch": 8.353233337402344e-06,
      "step": 8759,
      "training_step_time": 17.105858087539673
    },
    {
      "epoch": 8.35418701171875e-06,
      "grad_norm": 0.6125543117523193,
      "learning_rate": 1e-05,
      "loss": 0.0709,
      "step": 8760
    },
    {
      "epoch": 8.35418701171875e-06,
      "model_forward_time": 5.621549606323242,
      "step": 8760
    },
    {
      "dynamics_loss_avg": 0.06794611513614654,
      "epoch": 8.35418701171875e-06,
      "step": 8760
    },
    {
      "action_loss_avg": 0.009283821564167738,
      "epoch": 8.35418701171875e-06,
      "step": 8760
    },
    {
      "epoch": 8.35418701171875e-06,
      "step": 8760,
      "training_step_time": 17.110562324523926
    },
    {
      "epoch": 8.355140686035156e-06,
      "model_forward_time": 5.611172437667847,
      "step": 8761
    },
    {
      "epoch": 8.355140686035156e-06,
      "step": 8761,
      "training_step_time": 17.060285568237305
    },
    {
      "epoch": 8.356094360351563e-06,
      "model_forward_time": 5.963031530380249,
      "step": 8762
    },
    {
      "epoch": 8.356094360351563e-06,
      "step": 8762,
      "training_step_time": 17.392760276794434
    },
    {
      "epoch": 8.357048034667969e-06,
      "model_forward_time": 5.5772013664245605,
      "step": 8763
    },
    {
      "epoch": 8.357048034667969e-06,
      "step": 8763,
      "training_step_time": 16.983087062835693
    },
    {
      "epoch": 8.358001708984374e-06,
      "model_forward_time": 5.566189527511597,
      "step": 8764
    },
    {
      "epoch": 8.358001708984374e-06,
      "step": 8764,
      "training_step_time": 16.993845462799072
    },
    {
      "epoch": 8.358955383300782e-06,
      "model_forward_time": 5.57088303565979,
      "step": 8765
    },
    {
      "epoch": 8.358955383300782e-06,
      "step": 8765,
      "training_step_time": 16.997045040130615
    },
    {
      "epoch": 8.359909057617188e-06,
      "model_forward_time": 5.570659875869751,
      "step": 8766
    },
    {
      "epoch": 8.359909057617188e-06,
      "step": 8766,
      "training_step_time": 17.020756006240845
    },
    {
      "epoch": 8.360862731933593e-06,
      "model_forward_time": 5.939906120300293,
      "step": 8767
    },
    {
      "epoch": 8.360862731933593e-06,
      "step": 8767,
      "training_step_time": 17.375221729278564
    },
    {
      "epoch": 8.36181640625e-06,
      "model_forward_time": 5.6011176109313965,
      "step": 8768
    },
    {
      "epoch": 8.36181640625e-06,
      "step": 8768,
      "training_step_time": 17.066306352615356
    },
    {
      "epoch": 8.362770080566406e-06,
      "model_forward_time": 5.611661911010742,
      "step": 8769
    },
    {
      "epoch": 8.362770080566406e-06,
      "step": 8769,
      "training_step_time": 17.102964162826538
    },
    {
      "epoch": 8.363723754882812e-06,
      "grad_norm": 0.8971151113510132,
      "learning_rate": 1e-05,
      "loss": 0.06,
      "step": 8770
    },
    {
      "epoch": 8.363723754882812e-06,
      "model_forward_time": 5.606877326965332,
      "step": 8770
    },
    {
      "dynamics_loss_avg": 0.04773930162191391,
      "epoch": 8.363723754882812e-06,
      "step": 8770
    },
    {
      "action_loss_avg": 0.007831062143668532,
      "epoch": 8.363723754882812e-06,
      "step": 8770
    },
    {
      "epoch": 8.363723754882812e-06,
      "step": 8770,
      "training_step_time": 17.108355045318604
    },
    {
      "epoch": 8.36467742919922e-06,
      "model_forward_time": 5.612975120544434,
      "step": 8771
    },
    {
      "epoch": 8.36467742919922e-06,
      "step": 8771,
      "training_step_time": 17.11462664604187
    },
    {
      "epoch": 8.365631103515625e-06,
      "model_forward_time": 5.61075758934021,
      "step": 8772
    },
    {
      "epoch": 8.365631103515625e-06,
      "step": 8772,
      "training_step_time": 17.076658010482788
    },
    {
      "epoch": 8.366584777832031e-06,
      "model_forward_time": 5.978477716445923,
      "step": 8773
    },
    {
      "epoch": 8.366584777832031e-06,
      "step": 8773,
      "training_step_time": 17.398776054382324
    },
    {
      "epoch": 8.367538452148437e-06,
      "model_forward_time": 5.5770721435546875,
      "step": 8774
    },
    {
      "epoch": 8.367538452148437e-06,
      "step": 8774,
      "training_step_time": 16.98206377029419
    },
    {
      "epoch": 8.368492126464844e-06,
      "model_forward_time": 5.567978382110596,
      "step": 8775
    },
    {
      "epoch": 8.368492126464844e-06,
      "step": 8775,
      "training_step_time": 16.977149724960327
    },
    {
      "epoch": 8.36944580078125e-06,
      "model_forward_time": 5.568295240402222,
      "step": 8776
    },
    {
      "epoch": 8.36944580078125e-06,
      "step": 8776,
      "training_step_time": 17.007986307144165
    },
    {
      "epoch": 8.370399475097656e-06,
      "model_forward_time": 5.581176280975342,
      "step": 8777
    },
    {
      "epoch": 8.370399475097656e-06,
      "step": 8777,
      "training_step_time": 17.008666038513184
    },
    {
      "epoch": 8.371353149414063e-06,
      "model_forward_time": 5.606395721435547,
      "step": 8778
    },
    {
      "epoch": 8.371353149414063e-06,
      "step": 8778,
      "training_step_time": 17.46910524368286
    },
    {
      "epoch": 8.372306823730469e-06,
      "model_forward_time": 5.636818170547485,
      "step": 8779
    },
    {
      "epoch": 8.372306823730469e-06,
      "step": 8779,
      "training_step_time": 17.142213344573975
    },
    {
      "epoch": 8.373260498046874e-06,
      "grad_norm": 0.9967504143714905,
      "learning_rate": 1e-05,
      "loss": 0.0817,
      "step": 8780
    },
    {
      "epoch": 8.373260498046874e-06,
      "model_forward_time": 5.607652425765991,
      "step": 8780
    },
    {
      "dynamics_loss_avg": 0.07804641462862491,
      "epoch": 8.373260498046874e-06,
      "step": 8780
    },
    {
      "action_loss_avg": 0.013158458448015154,
      "epoch": 8.373260498046874e-06,
      "step": 8780
    },
    {
      "epoch": 8.373260498046874e-06,
      "step": 8780,
      "training_step_time": 17.05135726928711
    },
    {
      "epoch": 8.374214172363282e-06,
      "model_forward_time": 5.571503400802612,
      "step": 8781
    },
    {
      "epoch": 8.374214172363282e-06,
      "step": 8781,
      "training_step_time": 16.9852135181427
    },
    {
      "epoch": 8.375167846679688e-06,
      "model_forward_time": 5.561095476150513,
      "step": 8782
    },
    {
      "epoch": 8.375167846679688e-06,
      "step": 8782,
      "training_step_time": 16.973554849624634
    },
    {
      "epoch": 8.376121520996093e-06,
      "model_forward_time": 5.554284572601318,
      "step": 8783
    },
    {
      "epoch": 8.376121520996093e-06,
      "step": 8783,
      "training_step_time": 16.96760582923889
    },
    {
      "epoch": 8.3770751953125e-06,
      "model_forward_time": 5.9342942237854,
      "step": 8784
    },
    {
      "epoch": 8.3770751953125e-06,
      "step": 8784,
      "training_step_time": 17.335283994674683
    },
    {
      "epoch": 8.378028869628906e-06,
      "model_forward_time": 5.563016891479492,
      "step": 8785
    },
    {
      "epoch": 8.378028869628906e-06,
      "step": 8785,
      "training_step_time": 16.975572109222412
    },
    {
      "epoch": 8.378982543945312e-06,
      "model_forward_time": 5.5777740478515625,
      "step": 8786
    },
    {
      "epoch": 8.378982543945312e-06,
      "step": 8786,
      "training_step_time": 16.99682903289795
    },
    {
      "epoch": 8.37993621826172e-06,
      "model_forward_time": 5.574302434921265,
      "step": 8787
    },
    {
      "epoch": 8.37993621826172e-06,
      "step": 8787,
      "training_step_time": 17.01387047767639
    },
    {
      "epoch": 8.380889892578125e-06,
      "model_forward_time": 5.582726001739502,
      "step": 8788
    },
    {
      "epoch": 8.380889892578125e-06,
      "step": 8788,
      "training_step_time": 17.036478757858276
    },
    {
      "epoch": 8.381843566894531e-06,
      "model_forward_time": 5.60285496711731,
      "step": 8789
    },
    {
      "epoch": 8.381843566894531e-06,
      "step": 8789,
      "training_step_time": 17.4517719745636
    },
    {
      "epoch": 8.382797241210937e-06,
      "grad_norm": 0.5912920832633972,
      "learning_rate": 1e-05,
      "loss": 0.0666,
      "step": 8790
    },
    {
      "epoch": 8.382797241210937e-06,
      "model_forward_time": 5.620774984359741,
      "step": 8790
    },
    {
      "dynamics_loss_avg": 0.0497685369104147,
      "epoch": 8.382797241210937e-06,
      "step": 8790
    },
    {
      "action_loss_avg": 0.0060754838632419705,
      "epoch": 8.382797241210937e-06,
      "step": 8790
    },
    {
      "epoch": 8.382797241210937e-06,
      "step": 8790,
      "training_step_time": 17.14423680305481
    },
    {
      "epoch": 8.383750915527344e-06,
      "model_forward_time": 5.62766432762146,
      "step": 8791
    },
    {
      "epoch": 8.383750915527344e-06,
      "step": 8791,
      "training_step_time": 17.11503529548645
    },
    {
      "epoch": 8.38470458984375e-06,
      "model_forward_time": 5.597253084182739,
      "step": 8792
    },
    {
      "epoch": 8.38470458984375e-06,
      "step": 8792,
      "training_step_time": 17.048828125
    },
    {
      "epoch": 8.385658264160156e-06,
      "model_forward_time": 5.595019578933716,
      "step": 8793
    },
    {
      "epoch": 8.385658264160156e-06,
      "step": 8793,
      "training_step_time": 17.028048753738403
    },
    {
      "epoch": 8.386611938476563e-06,
      "model_forward_time": 5.583245515823364,
      "step": 8794
    },
    {
      "epoch": 8.386611938476563e-06,
      "step": 8794,
      "training_step_time": 16.992810249328613
    },
    {
      "epoch": 8.387565612792969e-06,
      "model_forward_time": 5.94661808013916,
      "step": 8795
    },
    {
      "epoch": 8.387565612792969e-06,
      "step": 8795,
      "training_step_time": 17.38508105278015
    },
    {
      "epoch": 8.388519287109374e-06,
      "model_forward_time": 5.614752292633057,
      "step": 8796
    },
    {
      "epoch": 8.388519287109374e-06,
      "step": 8796,
      "training_step_time": 17.10672926902771
    },
    {
      "epoch": 8.389472961425782e-06,
      "model_forward_time": 5.613287925720215,
      "step": 8797
    },
    {
      "epoch": 8.389472961425782e-06,
      "step": 8797,
      "training_step_time": 17.10438108444214
    },
    {
      "epoch": 8.390426635742188e-06,
      "model_forward_time": 5.598323822021484,
      "step": 8798
    },
    {
      "epoch": 8.390426635742188e-06,
      "step": 8798,
      "training_step_time": 17.04450511932373
    },
    {
      "epoch": 8.391380310058593e-06,
      "model_forward_time": 5.616990089416504,
      "step": 8799
    },
    {
      "epoch": 8.391380310058593e-06,
      "step": 8799,
      "training_step_time": 17.06618595123291
    },
    {
      "epoch": 8.392333984375e-06,
      "grad_norm": 1.4730749130249023,
      "learning_rate": 1e-05,
      "loss": 0.0638,
      "step": 8800
    },
    {
      "epoch": 8.392333984375e-06,
      "model_forward_time": 5.9470531940460205,
      "step": 8800
    },
    {
      "dynamics_loss_avg": 0.059254510700702666,
      "epoch": 8.392333984375e-06,
      "step": 8800
    },
    {
      "action_loss_avg": 0.007135756500065327,
      "epoch": 8.392333984375e-06,
      "step": 8800
    },
    {
      "epoch": 8.392333984375e-06,
      "step": 8800,
      "training_step_time": 17.378664016723633
    },
    {
      "epoch": 8.393287658691406e-06,
      "model_forward_time": 5.58230996131897,
      "step": 8801
    },
    {
      "epoch": 8.393287658691406e-06,
      "step": 8801,
      "training_step_time": 17.013602018356323
    },
    {
      "epoch": 8.394241333007812e-06,
      "model_forward_time": 5.6071741580963135,
      "step": 8802
    },
    {
      "epoch": 8.394241333007812e-06,
      "step": 8802,
      "training_step_time": 17.071845293045044
    },
    {
      "epoch": 8.39519500732422e-06,
      "model_forward_time": 5.607471466064453,
      "step": 8803
    },
    {
      "epoch": 8.39519500732422e-06,
      "step": 8803,
      "training_step_time": 17.107091903686523
    },
    {
      "epoch": 8.396148681640625e-06,
      "model_forward_time": 5.615411281585693,
      "step": 8804
    },
    {
      "epoch": 8.396148681640625e-06,
      "step": 8804,
      "training_step_time": 17.121523141860962
    },
    {
      "epoch": 8.397102355957031e-06,
      "model_forward_time": 5.614422082901001,
      "step": 8805
    },
    {
      "epoch": 8.397102355957031e-06,
      "step": 8805,
      "training_step_time": 17.111267566680908
    },
    {
      "epoch": 8.398056030273437e-06,
      "model_forward_time": 6.002739906311035,
      "step": 8806
    },
    {
      "epoch": 8.398056030273437e-06,
      "step": 8806,
      "training_step_time": 17.500202178955078
    },
    {
      "epoch": 8.399009704589844e-06,
      "model_forward_time": 5.612459182739258,
      "step": 8807
    },
    {
      "epoch": 8.399009704589844e-06,
      "step": 8807,
      "training_step_time": 17.109088897705078
    },
    {
      "epoch": 8.39996337890625e-06,
      "model_forward_time": 5.617947101593018,
      "step": 8808
    },
    {
      "epoch": 8.39996337890625e-06,
      "step": 8808,
      "training_step_time": 17.08521819114685
    },
    {
      "epoch": 8.400917053222656e-06,
      "model_forward_time": 5.592364549636841,
      "step": 8809
    },
    {
      "epoch": 8.400917053222656e-06,
      "step": 8809,
      "training_step_time": 17.037540197372437
    },
    {
      "epoch": 8.401870727539063e-06,
      "grad_norm": 1.105900526046753,
      "learning_rate": 1e-05,
      "loss": 0.0657,
      "step": 8810
    },
    {
      "epoch": 8.401870727539063e-06,
      "model_forward_time": 5.5900843143463135,
      "step": 8810
    },
    {
      "dynamics_loss_avg": 0.05943149272352457,
      "epoch": 8.401870727539063e-06,
      "step": 8810
    },
    {
      "action_loss_avg": 0.0077180241933092475,
      "epoch": 8.401870727539063e-06,
      "step": 8810
    },
    {
      "epoch": 8.401870727539063e-06,
      "step": 8810,
      "training_step_time": 17.01572060585022
    },
    {
      "epoch": 8.402824401855469e-06,
      "model_forward_time": 5.579402685165405,
      "step": 8811
    },
    {
      "epoch": 8.402824401855469e-06,
      "step": 8811,
      "training_step_time": 17.38816738128662
    },
    {
      "epoch": 8.403778076171874e-06,
      "model_forward_time": 5.597586393356323,
      "step": 8812
    },
    {
      "epoch": 8.403778076171874e-06,
      "step": 8812,
      "training_step_time": 17.065319776535034
    },
    {
      "epoch": 8.404731750488282e-06,
      "model_forward_time": 5.595659494400024,
      "step": 8813
    },
    {
      "epoch": 8.404731750488282e-06,
      "step": 8813,
      "training_step_time": 17.065601587295532
    },
    {
      "epoch": 8.405685424804688e-06,
      "model_forward_time": 5.6071083545684814,
      "step": 8814
    },
    {
      "epoch": 8.405685424804688e-06,
      "step": 8814,
      "training_step_time": 17.058817625045776
    },
    {
      "epoch": 8.406639099121093e-06,
      "model_forward_time": 5.608966112136841,
      "step": 8815
    },
    {
      "epoch": 8.406639099121093e-06,
      "step": 8815,
      "training_step_time": 17.096774339675903
    },
    {
      "epoch": 8.4075927734375e-06,
      "model_forward_time": 5.6199445724487305,
      "step": 8816
    },
    {
      "epoch": 8.4075927734375e-06,
      "step": 8816,
      "training_step_time": 17.12267518043518
    },
    {
      "epoch": 8.408546447753906e-06,
      "model_forward_time": 5.9890594482421875,
      "step": 8817
    },
    {
      "epoch": 8.408546447753906e-06,
      "step": 8817,
      "training_step_time": 17.484619140625
    },
    {
      "epoch": 8.409500122070312e-06,
      "model_forward_time": 5.610292673110962,
      "step": 8818
    },
    {
      "epoch": 8.409500122070312e-06,
      "step": 8818,
      "training_step_time": 17.071460247039795
    },
    {
      "epoch": 8.41045379638672e-06,
      "model_forward_time": 5.5925493240356445,
      "step": 8819
    },
    {
      "epoch": 8.41045379638672e-06,
      "step": 8819,
      "training_step_time": 17.023823261260986
    },
    {
      "epoch": 8.411407470703125e-06,
      "grad_norm": 1.9055472612380981,
      "learning_rate": 1e-05,
      "loss": 0.0708,
      "step": 8820
    },
    {
      "epoch": 8.411407470703125e-06,
      "model_forward_time": 5.574084997177124,
      "step": 8820
    },
    {
      "dynamics_loss_avg": 0.05873389206826687,
      "epoch": 8.411407470703125e-06,
      "step": 8820
    },
    {
      "action_loss_avg": 0.009927601017989219,
      "epoch": 8.411407470703125e-06,
      "step": 8820
    },
    {
      "epoch": 8.411407470703125e-06,
      "step": 8820,
      "training_step_time": 16.983999013900757
    },
    {
      "epoch": 8.412361145019531e-06,
      "model_forward_time": 5.581371068954468,
      "step": 8821
    },
    {
      "epoch": 8.412361145019531e-06,
      "step": 8821,
      "training_step_time": 17.014119148254395
    },
    {
      "epoch": 8.413314819335937e-06,
      "model_forward_time": 5.579623460769653,
      "step": 8822
    },
    {
      "epoch": 8.413314819335937e-06,
      "step": 8822,
      "training_step_time": 17.378127336502075
    },
    {
      "epoch": 8.414268493652344e-06,
      "model_forward_time": 5.586416721343994,
      "step": 8823
    },
    {
      "epoch": 8.414268493652344e-06,
      "step": 8823,
      "training_step_time": 17.003803253173828
    },
    {
      "epoch": 8.41522216796875e-06,
      "model_forward_time": 5.5807225704193115,
      "step": 8824
    },
    {
      "epoch": 8.41522216796875e-06,
      "step": 8824,
      "training_step_time": 17.00198769569397
    },
    {
      "epoch": 8.416175842285156e-06,
      "model_forward_time": 5.582642316818237,
      "step": 8825
    },
    {
      "epoch": 8.416175842285156e-06,
      "step": 8825,
      "training_step_time": 17.025121688842773
    },
    {
      "epoch": 8.417129516601563e-06,
      "model_forward_time": 5.5969038009643555,
      "step": 8826
    },
    {
      "epoch": 8.417129516601563e-06,
      "step": 8826,
      "training_step_time": 17.074912786483765
    },
    {
      "epoch": 8.418083190917969e-06,
      "model_forward_time": 5.598081111907959,
      "step": 8827
    },
    {
      "epoch": 8.418083190917969e-06,
      "step": 8827,
      "training_step_time": 17.0713050365448
    },
    {
      "epoch": 8.419036865234374e-06,
      "model_forward_time": 5.982656478881836,
      "step": 8828
    },
    {
      "epoch": 8.419036865234374e-06,
      "step": 8828,
      "training_step_time": 17.454660892486572
    },
    {
      "epoch": 8.419990539550782e-06,
      "model_forward_time": 5.601100206375122,
      "step": 8829
    },
    {
      "epoch": 8.419990539550782e-06,
      "step": 8829,
      "training_step_time": 17.044955492019653
    },
    {
      "epoch": 8.420944213867188e-06,
      "grad_norm": 0.6353341937065125,
      "learning_rate": 1e-05,
      "loss": 0.0642,
      "step": 8830
    },
    {
      "epoch": 8.420944213867188e-06,
      "model_forward_time": 5.58289361000061,
      "step": 8830
    },
    {
      "dynamics_loss_avg": 0.058712321892380716,
      "epoch": 8.420944213867188e-06,
      "step": 8830
    },
    {
      "action_loss_avg": 0.008029423654079437,
      "epoch": 8.420944213867188e-06,
      "step": 8830
    },
    {
      "epoch": 8.420944213867188e-06,
      "step": 8830,
      "training_step_time": 17.01218891143799
    },
    {
      "epoch": 8.421897888183593e-06,
      "model_forward_time": 5.585613489151001,
      "step": 8831
    },
    {
      "epoch": 8.421897888183593e-06,
      "step": 8831,
      "training_step_time": 16.994045972824097
    },
    {
      "epoch": 8.4228515625e-06,
      "model_forward_time": 5.578565835952759,
      "step": 8832
    },
    {
      "epoch": 8.4228515625e-06,
      "step": 8832,
      "training_step_time": 17.041381359100342
    },
    {
      "epoch": 8.423805236816406e-06,
      "model_forward_time": 5.971341848373413,
      "step": 8833
    },
    {
      "epoch": 8.423805236816406e-06,
      "step": 8833,
      "training_step_time": 17.43314838409424
    },
    {
      "epoch": 8.424758911132812e-06,
      "model_forward_time": 5.599408388137817,
      "step": 8834
    },
    {
      "epoch": 8.424758911132812e-06,
      "step": 8834,
      "training_step_time": 17.083709955215454
    },
    {
      "epoch": 8.42571258544922e-06,
      "model_forward_time": 5.620107412338257,
      "step": 8835
    },
    {
      "epoch": 8.42571258544922e-06,
      "step": 8835,
      "training_step_time": 17.12569522857666
    },
    {
      "epoch": 8.426666259765625e-06,
      "model_forward_time": 5.608239650726318,
      "step": 8836
    },
    {
      "epoch": 8.426666259765625e-06,
      "step": 8836,
      "training_step_time": 17.110163688659668
    },
    {
      "epoch": 8.427619934082031e-06,
      "model_forward_time": 5.610090255737305,
      "step": 8837
    },
    {
      "epoch": 8.427619934082031e-06,
      "step": 8837,
      "training_step_time": 17.099945306777954
    },
    {
      "epoch": 8.428573608398437e-06,
      "model_forward_time": 5.6111180782318115,
      "step": 8838
    },
    {
      "epoch": 8.428573608398437e-06,
      "step": 8838,
      "training_step_time": 17.09981417655945
    },
    {
      "epoch": 8.429527282714844e-06,
      "model_forward_time": 5.986227989196777,
      "step": 8839
    },
    {
      "epoch": 8.429527282714844e-06,
      "step": 8839,
      "training_step_time": 17.455642461776733
    },
    {
      "epoch": 8.43048095703125e-06,
      "grad_norm": 0.5325674414634705,
      "learning_rate": 1e-05,
      "loss": 0.0618,
      "step": 8840
    },
    {
      "epoch": 8.43048095703125e-06,
      "model_forward_time": 5.600984573364258,
      "step": 8840
    },
    {
      "dynamics_loss_avg": 0.050025160610675815,
      "epoch": 8.43048095703125e-06,
      "step": 8840
    },
    {
      "action_loss_avg": 0.00889003686606884,
      "epoch": 8.43048095703125e-06,
      "step": 8840
    },
    {
      "epoch": 8.43048095703125e-06,
      "step": 8840,
      "training_step_time": 17.051666975021362
    },
    {
      "epoch": 8.431434631347656e-06,
      "model_forward_time": 5.592016935348511,
      "step": 8841
    },
    {
      "epoch": 8.431434631347656e-06,
      "step": 8841,
      "training_step_time": 17.055504083633423
    },
    {
      "epoch": 8.432388305664063e-06,
      "model_forward_time": 5.596845626831055,
      "step": 8842
    },
    {
      "epoch": 8.432388305664063e-06,
      "step": 8842,
      "training_step_time": 17.063889980316162
    },
    {
      "epoch": 8.433341979980469e-06,
      "model_forward_time": 5.618909120559692,
      "step": 8843
    },
    {
      "epoch": 8.433341979980469e-06,
      "step": 8843,
      "training_step_time": 17.118879079818726
    },
    {
      "epoch": 8.434295654296874e-06,
      "model_forward_time": 5.611452102661133,
      "step": 8844
    },
    {
      "epoch": 8.434295654296874e-06,
      "step": 8844,
      "training_step_time": 17.46996545791626
    },
    {
      "epoch": 8.435249328613282e-06,
      "model_forward_time": 5.6210036277771,
      "step": 8845
    },
    {
      "epoch": 8.435249328613282e-06,
      "step": 8845,
      "training_step_time": 17.125558137893677
    },
    {
      "epoch": 8.436203002929688e-06,
      "model_forward_time": 5.6129233837127686,
      "step": 8846
    },
    {
      "epoch": 8.436203002929688e-06,
      "step": 8846,
      "training_step_time": 17.10096836090088
    },
    {
      "epoch": 8.437156677246093e-06,
      "model_forward_time": 5.619622468948364,
      "step": 8847
    },
    {
      "epoch": 8.437156677246093e-06,
      "step": 8847,
      "training_step_time": 17.12179708480835
    },
    {
      "epoch": 8.4381103515625e-06,
      "model_forward_time": 5.609933137893677,
      "step": 8848
    },
    {
      "epoch": 8.4381103515625e-06,
      "step": 8848,
      "training_step_time": 17.092323064804077
    },
    {
      "epoch": 8.439064025878906e-06,
      "model_forward_time": 5.597492456436157,
      "step": 8849
    },
    {
      "epoch": 8.439064025878906e-06,
      "step": 8849,
      "training_step_time": 17.056910514831543
    },
    {
      "epoch": 8.440017700195312e-06,
      "grad_norm": 0.45649874210357666,
      "learning_rate": 1e-05,
      "loss": 0.0613,
      "step": 8850
    },
    {
      "epoch": 8.440017700195312e-06,
      "model_forward_time": 5.954318523406982,
      "step": 8850
    },
    {
      "dynamics_loss_avg": 0.053929689526557925,
      "epoch": 8.440017700195312e-06,
      "step": 8850
    },
    {
      "action_loss_avg": 0.008091326383873821,
      "epoch": 8.440017700195312e-06,
      "step": 8850
    },
    {
      "epoch": 8.440017700195312e-06,
      "step": 8850,
      "training_step_time": 17.37378239631653
    },
    {
      "epoch": 8.44097137451172e-06,
      "model_forward_time": 5.584222316741943,
      "step": 8851
    },
    {
      "epoch": 8.44097137451172e-06,
      "step": 8851,
      "training_step_time": 17.01889181137085
    },
    {
      "epoch": 8.441925048828125e-06,
      "model_forward_time": 5.602552652359009,
      "step": 8852
    },
    {
      "epoch": 8.441925048828125e-06,
      "step": 8852,
      "training_step_time": 17.067375659942627
    },
    {
      "epoch": 8.442878723144531e-06,
      "model_forward_time": 5.619189977645874,
      "step": 8853
    },
    {
      "epoch": 8.442878723144531e-06,
      "step": 8853,
      "training_step_time": 17.114609003067017
    },
    {
      "epoch": 8.443832397460937e-06,
      "model_forward_time": 5.620440244674683,
      "step": 8854
    },
    {
      "epoch": 8.443832397460937e-06,
      "step": 8854,
      "training_step_time": 17.083831310272217
    },
    {
      "epoch": 8.444786071777344e-06,
      "model_forward_time": 5.930096626281738,
      "step": 8855
    },
    {
      "epoch": 8.444786071777344e-06,
      "step": 8855,
      "training_step_time": 17.35907769203186
    },
    {
      "epoch": 8.44573974609375e-06,
      "model_forward_time": 5.58478856086731,
      "step": 8856
    },
    {
      "epoch": 8.44573974609375e-06,
      "step": 8856,
      "training_step_time": 17.03284215927124
    },
    {
      "epoch": 8.446693420410156e-06,
      "model_forward_time": 5.589296340942383,
      "step": 8857
    },
    {
      "epoch": 8.446693420410156e-06,
      "step": 8857,
      "training_step_time": 17.02228856086731
    },
    {
      "epoch": 8.447647094726563e-06,
      "model_forward_time": 5.587883949279785,
      "step": 8858
    },
    {
      "epoch": 8.447647094726563e-06,
      "step": 8858,
      "training_step_time": 17.02206325531006
    },
    {
      "epoch": 8.448600769042969e-06,
      "model_forward_time": 5.5867390632629395,
      "step": 8859
    },
    {
      "epoch": 8.448600769042969e-06,
      "step": 8859,
      "training_step_time": 17.004836082458496
    },
    {
      "epoch": 8.449554443359374e-06,
      "grad_norm": 0.8781124949455261,
      "learning_rate": 1e-05,
      "loss": 0.058,
      "step": 8860
    },
    {
      "epoch": 8.449554443359374e-06,
      "model_forward_time": 5.588912725448608,
      "step": 8860
    },
    {
      "dynamics_loss_avg": 0.0501023193821311,
      "epoch": 8.449554443359374e-06,
      "step": 8860
    },
    {
      "action_loss_avg": 0.006267368257977068,
      "epoch": 8.449554443359374e-06,
      "step": 8860
    },
    {
      "epoch": 8.449554443359374e-06,
      "step": 8860,
      "training_step_time": 17.059165477752686
    },
    {
      "epoch": 8.450508117675782e-06,
      "model_forward_time": 5.955979347229004,
      "step": 8861
    },
    {
      "epoch": 8.450508117675782e-06,
      "step": 8861,
      "training_step_time": 17.449755430221558
    },
    {
      "epoch": 8.451461791992188e-06,
      "model_forward_time": 5.61655068397522,
      "step": 8862
    },
    {
      "epoch": 8.451461791992188e-06,
      "step": 8862,
      "training_step_time": 17.117501735687256
    },
    {
      "epoch": 8.452415466308593e-06,
      "model_forward_time": 5.618645668029785,
      "step": 8863
    },
    {
      "epoch": 8.452415466308593e-06,
      "step": 8863,
      "training_step_time": 17.11404776573181
    },
    {
      "epoch": 8.453369140625e-06,
      "model_forward_time": 5.623271942138672,
      "step": 8864
    },
    {
      "epoch": 8.453369140625e-06,
      "step": 8864,
      "training_step_time": 17.122354984283447
    },
    {
      "epoch": 8.454322814941406e-06,
      "model_forward_time": 5.616096019744873,
      "step": 8865
    },
    {
      "epoch": 8.454322814941406e-06,
      "step": 8865,
      "training_step_time": 17.116822481155396
    },
    {
      "epoch": 8.455276489257812e-06,
      "model_forward_time": 5.958150386810303,
      "step": 8866
    },
    {
      "epoch": 8.455276489257812e-06,
      "step": 8866,
      "training_step_time": 17.427926301956177
    },
    {
      "epoch": 8.45623016357422e-06,
      "model_forward_time": 5.598318576812744,
      "step": 8867
    },
    {
      "epoch": 8.45623016357422e-06,
      "step": 8867,
      "training_step_time": 17.033626794815063
    },
    {
      "epoch": 8.457183837890625e-06,
      "model_forward_time": 5.5909264087677,
      "step": 8868
    },
    {
      "epoch": 8.457183837890625e-06,
      "step": 8868,
      "training_step_time": 17.033985376358032
    },
    {
      "epoch": 8.458137512207031e-06,
      "model_forward_time": 5.5922417640686035,
      "step": 8869
    },
    {
      "epoch": 8.458137512207031e-06,
      "step": 8869,
      "training_step_time": 17.03632140159607
    },
    {
      "epoch": 8.459091186523437e-06,
      "grad_norm": 1.267784833908081,
      "learning_rate": 1e-05,
      "loss": 0.0766,
      "step": 8870
    },
    {
      "epoch": 8.459091186523437e-06,
      "model_forward_time": 5.597518444061279,
      "step": 8870
    },
    {
      "dynamics_loss_avg": 0.06786870397627354,
      "epoch": 8.459091186523437e-06,
      "step": 8870
    },
    {
      "action_loss_avg": 0.009388649393804372,
      "epoch": 8.459091186523437e-06,
      "step": 8870
    },
    {
      "epoch": 8.459091186523437e-06,
      "step": 8870,
      "training_step_time": 17.055054903030396
    },
    {
      "epoch": 8.460044860839844e-06,
      "model_forward_time": 5.607534646987915,
      "step": 8871
    },
    {
      "epoch": 8.460044860839844e-06,
      "step": 8871,
      "training_step_time": 17.450578212738037
    },
    {
      "epoch": 8.46099853515625e-06,
      "model_forward_time": 5.600961923599243,
      "step": 8872
    },
    {
      "epoch": 8.46099853515625e-06,
      "step": 8872,
      "training_step_time": 17.065062999725342
    },
    {
      "epoch": 8.461952209472656e-06,
      "model_forward_time": 5.598678112030029,
      "step": 8873
    },
    {
      "epoch": 8.461952209472656e-06,
      "step": 8873,
      "training_step_time": 17.041090965270996
    },
    {
      "epoch": 8.462905883789063e-06,
      "model_forward_time": 5.593288898468018,
      "step": 8874
    },
    {
      "epoch": 8.462905883789063e-06,
      "step": 8874,
      "training_step_time": 17.057373046875
    },
    {
      "epoch": 8.463859558105469e-06,
      "model_forward_time": 5.602416753768921,
      "step": 8875
    },
    {
      "epoch": 8.463859558105469e-06,
      "step": 8875,
      "training_step_time": 17.05950355529785
    },
    {
      "epoch": 8.464813232421874e-06,
      "model_forward_time": 5.59759783744812,
      "step": 8876
    },
    {
      "epoch": 8.464813232421874e-06,
      "step": 8876,
      "training_step_time": 17.056939125061035
    },
    {
      "epoch": 8.465766906738282e-06,
      "model_forward_time": 5.931617498397827,
      "step": 8877
    },
    {
      "epoch": 8.465766906738282e-06,
      "step": 8877,
      "training_step_time": 17.38250231742859
    },
    {
      "epoch": 8.466720581054688e-06,
      "model_forward_time": 5.595046281814575,
      "step": 8878
    },
    {
      "epoch": 8.466720581054688e-06,
      "step": 8878,
      "training_step_time": 17.045250177383423
    },
    {
      "epoch": 8.467674255371093e-06,
      "model_forward_time": 5.600350618362427,
      "step": 8879
    },
    {
      "epoch": 8.467674255371093e-06,
      "step": 8879,
      "training_step_time": 17.051115036010742
    },
    {
      "epoch": 8.4686279296875e-06,
      "grad_norm": 1.2320775985717773,
      "learning_rate": 1e-05,
      "loss": 0.0735,
      "step": 8880
    },
    {
      "epoch": 8.4686279296875e-06,
      "model_forward_time": 5.593076705932617,
      "step": 8880
    },
    {
      "dynamics_loss_avg": 0.06257897466421128,
      "epoch": 8.4686279296875e-06,
      "step": 8880
    },
    {
      "action_loss_avg": 0.010747747728601098,
      "epoch": 8.4686279296875e-06,
      "step": 8880
    },
    {
      "epoch": 8.4686279296875e-06,
      "step": 8880,
      "training_step_time": 17.039339065551758
    },
    {
      "epoch": 8.469581604003906e-06,
      "model_forward_time": 5.593504905700684,
      "step": 8881
    },
    {
      "epoch": 8.469581604003906e-06,
      "step": 8881,
      "training_step_time": 17.05092692375183
    },
    {
      "epoch": 8.470535278320312e-06,
      "model_forward_time": 5.599966287612915,
      "step": 8882
    },
    {
      "epoch": 8.470535278320312e-06,
      "step": 8882,
      "training_step_time": 17.419736623764038
    },
    {
      "epoch": 8.471488952636718e-06,
      "model_forward_time": 5.596421718597412,
      "step": 8883
    },
    {
      "epoch": 8.471488952636718e-06,
      "step": 8883,
      "training_step_time": 17.071271419525146
    },
    {
      "epoch": 8.472442626953125e-06,
      "model_forward_time": 5.607483863830566,
      "step": 8884
    },
    {
      "epoch": 8.472442626953125e-06,
      "step": 8884,
      "training_step_time": 17.10754942893982
    },
    {
      "epoch": 8.473396301269531e-06,
      "model_forward_time": 5.616175174713135,
      "step": 8885
    },
    {
      "epoch": 8.473396301269531e-06,
      "step": 8885,
      "training_step_time": 17.111320734024048
    },
    {
      "epoch": 8.474349975585937e-06,
      "model_forward_time": 5.614493370056152,
      "step": 8886
    },
    {
      "epoch": 8.474349975585937e-06,
      "step": 8886,
      "training_step_time": 17.089249849319458
    },
    {
      "epoch": 8.475303649902344e-06,
      "model_forward_time": 5.60317325592041,
      "step": 8887
    },
    {
      "epoch": 8.475303649902344e-06,
      "step": 8887,
      "training_step_time": 17.0619854927063
    },
    {
      "epoch": 8.47625732421875e-06,
      "model_forward_time": 5.940963268280029,
      "step": 8888
    },
    {
      "epoch": 8.47625732421875e-06,
      "step": 8888,
      "training_step_time": 17.404176235198975
    },
    {
      "epoch": 8.477210998535156e-06,
      "model_forward_time": 5.592844009399414,
      "step": 8889
    },
    {
      "epoch": 8.477210998535156e-06,
      "step": 8889,
      "training_step_time": 17.062679052352905
    },
    {
      "epoch": 8.478164672851563e-06,
      "grad_norm": 0.4180213510990143,
      "learning_rate": 1e-05,
      "loss": 0.0585,
      "step": 8890
    },
    {
      "epoch": 8.478164672851563e-06,
      "model_forward_time": 5.614058971405029,
      "step": 8890
    },
    {
      "dynamics_loss_avg": 0.04975290521979332,
      "epoch": 8.478164672851563e-06,
      "step": 8890
    },
    {
      "action_loss_avg": 0.0069994314573705195,
      "epoch": 8.478164672851563e-06,
      "step": 8890
    },
    {
      "epoch": 8.478164672851563e-06,
      "step": 8890,
      "training_step_time": 17.072948455810547
    },
    {
      "epoch": 8.479118347167969e-06,
      "model_forward_time": 5.5970165729522705,
      "step": 8891
    },
    {
      "epoch": 8.479118347167969e-06,
      "step": 8891,
      "training_step_time": 17.051506757736206
    },
    {
      "epoch": 8.480072021484374e-06,
      "model_forward_time": 5.596988677978516,
      "step": 8892
    },
    {
      "epoch": 8.480072021484374e-06,
      "step": 8892,
      "training_step_time": 17.06340479850769
    },
    {
      "epoch": 8.481025695800782e-06,
      "model_forward_time": 5.9698991775512695,
      "step": 8893
    },
    {
      "epoch": 8.481025695800782e-06,
      "step": 8893,
      "training_step_time": 17.43477439880371
    },
    {
      "epoch": 8.481979370117188e-06,
      "model_forward_time": 5.603956937789917,
      "step": 8894
    },
    {
      "epoch": 8.481979370117188e-06,
      "step": 8894,
      "training_step_time": 17.091416835784912
    },
    {
      "epoch": 8.482933044433593e-06,
      "model_forward_time": 5.619324445724487,
      "step": 8895
    },
    {
      "epoch": 8.482933044433593e-06,
      "step": 8895,
      "training_step_time": 17.10254979133606
    },
    {
      "epoch": 8.48388671875e-06,
      "model_forward_time": 5.593934774398804,
      "step": 8896
    },
    {
      "epoch": 8.48388671875e-06,
      "step": 8896,
      "training_step_time": 17.014445304870605
    },
    {
      "epoch": 8.484840393066406e-06,
      "model_forward_time": 5.578810930252075,
      "step": 8897
    },
    {
      "epoch": 8.484840393066406e-06,
      "step": 8897,
      "training_step_time": 17.000956058502197
    },
    {
      "epoch": 8.485794067382812e-06,
      "model_forward_time": 5.585436820983887,
      "step": 8898
    },
    {
      "epoch": 8.485794067382812e-06,
      "step": 8898,
      "training_step_time": 17.39873766899109
    },
    {
      "epoch": 8.486747741699218e-06,
      "model_forward_time": 5.587161064147949,
      "step": 8899
    },
    {
      "epoch": 8.486747741699218e-06,
      "step": 8899,
      "training_step_time": 17.023396492004395
    },
    {
      "epoch": 8.487701416015625e-06,
      "grad_norm": 0.7010958194732666,
      "learning_rate": 1e-05,
      "loss": 0.0675,
      "step": 8900
    },
    {
      "epoch": 8.487701416015625e-06,
      "model_forward_time": 5.587141752243042,
      "step": 8900
    },
    {
      "dynamics_loss_avg": 0.058097307570278645,
      "epoch": 8.487701416015625e-06,
      "step": 8900
    },
    {
      "action_loss_avg": 0.010397605039179325,
      "epoch": 8.487701416015625e-06,
      "step": 8900
    },
    {
      "epoch": 8.487701416015625e-06,
      "step": 8900,
      "training_step_time": 17.032063007354736
    },
    {
      "epoch": 8.488655090332031e-06,
      "model_forward_time": 5.60274863243103,
      "step": 8901
    },
    {
      "epoch": 8.488655090332031e-06,
      "step": 8901,
      "training_step_time": 17.076945543289185
    },
    {
      "epoch": 8.489608764648437e-06,
      "model_forward_time": 5.598293304443359,
      "step": 8902
    },
    {
      "epoch": 8.489608764648437e-06,
      "step": 8902,
      "training_step_time": 17.045997142791748
    },
    {
      "epoch": 8.490562438964844e-06,
      "model_forward_time": 5.597996234893799,
      "step": 8903
    },
    {
      "epoch": 8.490562438964844e-06,
      "step": 8903,
      "training_step_time": 17.04620337486267
    },
    {
      "epoch": 8.49151611328125e-06,
      "model_forward_time": 5.971837520599365,
      "step": 8904
    },
    {
      "epoch": 8.49151611328125e-06,
      "step": 8904,
      "training_step_time": 17.428616046905518
    },
    {
      "epoch": 8.492469787597656e-06,
      "model_forward_time": 5.601761817932129,
      "step": 8905
    },
    {
      "epoch": 8.492469787597656e-06,
      "step": 8905,
      "training_step_time": 17.087696075439453
    },
    {
      "epoch": 8.493423461914063e-06,
      "model_forward_time": 5.621661901473999,
      "step": 8906
    },
    {
      "epoch": 8.493423461914063e-06,
      "step": 8906,
      "training_step_time": 17.141750812530518
    },
    {
      "epoch": 8.494377136230469e-06,
      "model_forward_time": 5.625917673110962,
      "step": 8907
    },
    {
      "epoch": 8.494377136230469e-06,
      "step": 8907,
      "training_step_time": 17.12363886833191
    },
    {
      "epoch": 8.495330810546874e-06,
      "model_forward_time": 5.590299606323242,
      "step": 8908
    },
    {
      "epoch": 8.495330810546874e-06,
      "step": 8908,
      "training_step_time": 16.99377417564392
    },
    {
      "epoch": 8.496284484863282e-06,
      "model_forward_time": 5.923829078674316,
      "step": 8909
    },
    {
      "epoch": 8.496284484863282e-06,
      "step": 8909,
      "training_step_time": 17.35618305206299
    },
    {
      "epoch": 8.497238159179688e-06,
      "grad_norm": 0.2627715766429901,
      "learning_rate": 1e-05,
      "loss": 0.0489,
      "step": 8910
    },
    {
      "epoch": 8.497238159179688e-06,
      "model_forward_time": 5.570405006408691,
      "step": 8910
    },
    {
      "dynamics_loss_avg": 0.04163575302809477,
      "epoch": 8.497238159179688e-06,
      "step": 8910
    },
    {
      "action_loss_avg": 0.006837496883235872,
      "epoch": 8.497238159179688e-06,
      "step": 8910
    },
    {
      "epoch": 8.497238159179688e-06,
      "step": 8910,
      "training_step_time": 16.984036922454834
    },
    {
      "epoch": 8.498191833496093e-06,
      "model_forward_time": 5.570014953613281,
      "step": 8911
    },
    {
      "epoch": 8.498191833496093e-06,
      "step": 8911,
      "training_step_time": 16.979270935058594
    },
    {
      "epoch": 8.4991455078125e-06,
      "model_forward_time": 5.569914817810059,
      "step": 8912
    },
    {
      "epoch": 8.4991455078125e-06,
      "step": 8912,
      "training_step_time": 16.991034507751465
    },
    {
      "epoch": 8.500099182128906e-06,
      "model_forward_time": 5.576567649841309,
      "step": 8913
    },
    {
      "epoch": 8.500099182128906e-06,
      "step": 8913,
      "training_step_time": 17.014336109161377
    },
    {
      "epoch": 8.501052856445312e-06,
      "model_forward_time": 5.584291934967041,
      "step": 8914
    },
    {
      "epoch": 8.501052856445312e-06,
      "step": 8914,
      "training_step_time": 17.01188015937805
    },
    {
      "epoch": 8.502006530761718e-06,
      "model_forward_time": 5.946261882781982,
      "step": 8915
    },
    {
      "epoch": 8.502006530761718e-06,
      "step": 8915,
      "training_step_time": 17.407227039337158
    },
    {
      "epoch": 8.502960205078125e-06,
      "model_forward_time": 5.615041255950928,
      "step": 8916
    },
    {
      "epoch": 8.502960205078125e-06,
      "step": 8916,
      "training_step_time": 17.12553834915161
    },
    {
      "epoch": 8.503913879394531e-06,
      "model_forward_time": 5.6208436489105225,
      "step": 8917
    },
    {
      "epoch": 8.503913879394531e-06,
      "step": 8917,
      "training_step_time": 17.13462257385254
    },
    {
      "epoch": 8.504867553710937e-06,
      "model_forward_time": 5.612601280212402,
      "step": 8918
    },
    {
      "epoch": 8.504867553710937e-06,
      "step": 8918,
      "training_step_time": 17.082558155059814
    },
    {
      "epoch": 8.505821228027344e-06,
      "model_forward_time": 5.584737062454224,
      "step": 8919
    },
    {
      "epoch": 8.505821228027344e-06,
      "step": 8919,
      "training_step_time": 16.997462272644043
    },
    {
      "epoch": 8.50677490234375e-06,
      "grad_norm": 0.775238037109375,
      "learning_rate": 1e-05,
      "loss": 0.0729,
      "step": 8920
    },
    {
      "epoch": 8.50677490234375e-06,
      "model_forward_time": 5.928286075592041,
      "step": 8920
    },
    {
      "dynamics_loss_avg": 0.0652841366827488,
      "epoch": 8.50677490234375e-06,
      "step": 8920
    },
    {
      "action_loss_avg": 0.012396469525992871,
      "epoch": 8.50677490234375e-06,
      "step": 8920
    },
    {
      "epoch": 8.50677490234375e-06,
      "step": 8920,
      "training_step_time": 17.368211030960083
    },
    {
      "epoch": 8.507728576660156e-06,
      "model_forward_time": 5.558820724487305,
      "step": 8921
    },
    {
      "epoch": 8.507728576660156e-06,
      "step": 8921,
      "training_step_time": 16.98785662651062
    },
    {
      "epoch": 8.508682250976563e-06,
      "model_forward_time": 5.572972297668457,
      "step": 8922
    },
    {
      "epoch": 8.508682250976563e-06,
      "step": 8922,
      "training_step_time": 16.992205381393433
    },
    {
      "epoch": 8.509635925292969e-06,
      "model_forward_time": 5.574785232543945,
      "step": 8923
    },
    {
      "epoch": 8.509635925292969e-06,
      "step": 8923,
      "training_step_time": 17.000062465667725
    },
    {
      "epoch": 8.510589599609375e-06,
      "model_forward_time": 5.583231210708618,
      "step": 8924
    },
    {
      "epoch": 8.510589599609375e-06,
      "step": 8924,
      "training_step_time": 17.016434907913208
    },
    {
      "epoch": 8.511543273925782e-06,
      "model_forward_time": 5.577824592590332,
      "step": 8925
    },
    {
      "epoch": 8.511543273925782e-06,
      "step": 8925,
      "training_step_time": 16.997069358825684
    },
    {
      "epoch": 8.512496948242188e-06,
      "model_forward_time": 5.944113731384277,
      "step": 8926
    },
    {
      "epoch": 8.512496948242188e-06,
      "step": 8926,
      "training_step_time": 17.365012645721436
    },
    {
      "epoch": 8.513450622558593e-06,
      "model_forward_time": 5.588459014892578,
      "step": 8927
    },
    {
      "epoch": 8.513450622558593e-06,
      "step": 8927,
      "training_step_time": 17.043230295181274
    },
    {
      "epoch": 8.514404296875e-06,
      "model_forward_time": 5.599632501602173,
      "step": 8928
    },
    {
      "epoch": 8.514404296875e-06,
      "step": 8928,
      "training_step_time": 17.055935859680176
    },
    {
      "epoch": 8.515357971191407e-06,
      "model_forward_time": 5.5960118770599365,
      "step": 8929
    },
    {
      "epoch": 8.515357971191407e-06,
      "step": 8929,
      "training_step_time": 17.063877820968628
    },
    {
      "epoch": 8.516311645507812e-06,
      "grad_norm": 0.5243868827819824,
      "learning_rate": 1e-05,
      "loss": 0.069,
      "step": 8930
    },
    {
      "epoch": 8.516311645507812e-06,
      "model_forward_time": 5.599520921707153,
      "step": 8930
    },
    {
      "dynamics_loss_avg": 0.058222617767751214,
      "epoch": 8.516311645507812e-06,
      "step": 8930
    },
    {
      "action_loss_avg": 0.00843607229180634,
      "epoch": 8.516311645507812e-06,
      "step": 8930
    },
    {
      "epoch": 8.516311645507812e-06,
      "step": 8930,
      "training_step_time": 17.048582553863525
    },
    {
      "epoch": 8.517265319824218e-06,
      "model_forward_time": 5.956076622009277,
      "step": 8931
    },
    {
      "epoch": 8.517265319824218e-06,
      "step": 8931,
      "training_step_time": 17.408100366592407
    },
    {
      "epoch": 8.518218994140625e-06,
      "model_forward_time": 5.593487739562988,
      "step": 8932
    },
    {
      "epoch": 8.518218994140625e-06,
      "step": 8932,
      "training_step_time": 17.03477692604065
    },
    {
      "epoch": 8.519172668457031e-06,
      "model_forward_time": 5.60778546333313,
      "step": 8933
    },
    {
      "epoch": 8.519172668457031e-06,
      "step": 8933,
      "training_step_time": 17.109169721603394
    },
    {
      "epoch": 8.520126342773437e-06,
      "model_forward_time": 5.623777151107788,
      "step": 8934
    },
    {
      "epoch": 8.520126342773437e-06,
      "step": 8934,
      "training_step_time": 17.126526832580566
    },
    {
      "epoch": 8.521080017089844e-06,
      "model_forward_time": 5.623366594314575,
      "step": 8935
    },
    {
      "epoch": 8.521080017089844e-06,
      "step": 8935,
      "training_step_time": 17.12340998649597
    },
    {
      "epoch": 8.52203369140625e-06,
      "model_forward_time": 5.606583833694458,
      "step": 8936
    },
    {
      "epoch": 8.52203369140625e-06,
      "step": 8936,
      "training_step_time": 17.442521572113037
    },
    {
      "epoch": 8.522987365722656e-06,
      "model_forward_time": 5.617626428604126,
      "step": 8937
    },
    {
      "epoch": 8.522987365722656e-06,
      "step": 8937,
      "training_step_time": 17.112730741500854
    },
    {
      "epoch": 8.523941040039063e-06,
      "model_forward_time": 5.605357646942139,
      "step": 8938
    },
    {
      "epoch": 8.523941040039063e-06,
      "step": 8938,
      "training_step_time": 17.089447498321533
    },
    {
      "epoch": 8.524894714355469e-06,
      "model_forward_time": 5.609563589096069,
      "step": 8939
    },
    {
      "epoch": 8.524894714355469e-06,
      "step": 8939,
      "training_step_time": 17.106600284576416
    },
    {
      "epoch": 8.525848388671875e-06,
      "grad_norm": 0.4036024212837219,
      "learning_rate": 1e-05,
      "loss": 0.0587,
      "step": 8940
    },
    {
      "epoch": 8.525848388671875e-06,
      "model_forward_time": 5.616167068481445,
      "step": 8940
    },
    {
      "dynamics_loss_avg": 0.056476103514432906,
      "epoch": 8.525848388671875e-06,
      "step": 8940
    },
    {
      "action_loss_avg": 0.00887382000219077,
      "epoch": 8.525848388671875e-06,
      "step": 8940
    },
    {
      "epoch": 8.525848388671875e-06,
      "step": 8940,
      "training_step_time": 17.117070198059082
    },
    {
      "epoch": 8.526802062988282e-06,
      "model_forward_time": 5.62163782119751,
      "step": 8941
    },
    {
      "epoch": 8.526802062988282e-06,
      "step": 8941,
      "training_step_time": 17.11993432044983
    },
    {
      "epoch": 8.527755737304688e-06,
      "model_forward_time": 5.968910455703735,
      "step": 8942
    },
    {
      "epoch": 8.527755737304688e-06,
      "step": 8942,
      "training_step_time": 17.480527877807617
    },
    {
      "epoch": 8.528709411621093e-06,
      "model_forward_time": 5.642439603805542,
      "step": 8943
    },
    {
      "epoch": 8.528709411621093e-06,
      "step": 8943,
      "training_step_time": 17.160977840423584
    },
    {
      "epoch": 8.5296630859375e-06,
      "model_forward_time": 5.647906303405762,
      "step": 8944
    },
    {
      "epoch": 8.5296630859375e-06,
      "step": 8944,
      "training_step_time": 17.151917934417725
    },
    {
      "epoch": 8.530616760253907e-06,
      "model_forward_time": 5.621680974960327,
      "step": 8945
    },
    {
      "epoch": 8.530616760253907e-06,
      "step": 8945,
      "training_step_time": 17.12522530555725
    },
    {
      "epoch": 8.531570434570312e-06,
      "model_forward_time": 5.612513780593872,
      "step": 8946
    },
    {
      "epoch": 8.531570434570312e-06,
      "step": 8946,
      "training_step_time": 17.102644205093384
    },
    {
      "epoch": 8.532524108886718e-06,
      "model_forward_time": 5.980710983276367,
      "step": 8947
    },
    {
      "epoch": 8.532524108886718e-06,
      "step": 8947,
      "training_step_time": 17.46644616127014
    },
    {
      "epoch": 8.533477783203125e-06,
      "model_forward_time": 5.609334707260132,
      "step": 8948
    },
    {
      "epoch": 8.533477783203125e-06,
      "step": 8948,
      "training_step_time": 17.100701808929443
    },
    {
      "epoch": 8.534431457519531e-06,
      "model_forward_time": 5.615088939666748,
      "step": 8949
    },
    {
      "epoch": 8.534431457519531e-06,
      "step": 8949,
      "training_step_time": 17.110620498657227
    },
    {
      "epoch": 8.535385131835937e-06,
      "grad_norm": 1.9177396297454834,
      "learning_rate": 1e-05,
      "loss": 0.0692,
      "step": 8950
    },
    {
      "epoch": 8.535385131835937e-06,
      "model_forward_time": 5.611153602600098,
      "step": 8950
    },
    {
      "dynamics_loss_avg": 0.05275094090029597,
      "epoch": 8.535385131835937e-06,
      "step": 8950
    },
    {
      "action_loss_avg": 0.009563517046626658,
      "epoch": 8.535385131835937e-06,
      "step": 8950
    },
    {
      "epoch": 8.535385131835937e-06,
      "step": 8950,
      "training_step_time": 17.101016521453857
    },
    {
      "epoch": 8.536338806152344e-06,
      "model_forward_time": 5.605516672134399,
      "step": 8951
    },
    {
      "epoch": 8.536338806152344e-06,
      "step": 8951,
      "training_step_time": 17.0473952293396
    },
    {
      "epoch": 8.53729248046875e-06,
      "model_forward_time": 5.586919069290161,
      "step": 8952
    },
    {
      "epoch": 8.53729248046875e-06,
      "step": 8952,
      "training_step_time": 17.02893567085266
    },
    {
      "epoch": 8.538246154785156e-06,
      "model_forward_time": 5.95668363571167,
      "step": 8953
    },
    {
      "epoch": 8.538246154785156e-06,
      "step": 8953,
      "training_step_time": 17.40108323097229
    },
    {
      "epoch": 8.539199829101563e-06,
      "model_forward_time": 5.6051576137542725,
      "step": 8954
    },
    {
      "epoch": 8.539199829101563e-06,
      "step": 8954,
      "training_step_time": 17.074151039123535
    },
    {
      "epoch": 8.540153503417969e-06,
      "model_forward_time": 5.605718612670898,
      "step": 8955
    },
    {
      "epoch": 8.540153503417969e-06,
      "step": 8955,
      "training_step_time": 17.051028728485107
    },
    {
      "epoch": 8.541107177734375e-06,
      "model_forward_time": 5.596121311187744,
      "step": 8956
    },
    {
      "epoch": 8.541107177734375e-06,
      "step": 8956,
      "training_step_time": 17.054777145385742
    },
    {
      "epoch": 8.542060852050782e-06,
      "model_forward_time": 5.59672737121582,
      "step": 8957
    },
    {
      "epoch": 8.542060852050782e-06,
      "step": 8957,
      "training_step_time": 17.049776554107666
    },
    {
      "epoch": 8.543014526367188e-06,
      "model_forward_time": 5.98030161857605,
      "step": 8958
    },
    {
      "epoch": 8.543014526367188e-06,
      "step": 8958,
      "training_step_time": 17.439607858657837
    },
    {
      "epoch": 8.543968200683593e-06,
      "model_forward_time": 5.598245143890381,
      "step": 8959
    },
    {
      "epoch": 8.543968200683593e-06,
      "step": 8959,
      "training_step_time": 17.09676766395569
    },
    {
      "epoch": 8.544921875e-06,
      "grad_norm": 1.6676294803619385,
      "learning_rate": 1e-05,
      "loss": 0.0735,
      "step": 8960
    },
    {
      "epoch": 8.544921875e-06,
      "model_forward_time": 5.620064973831177,
      "step": 8960
    },
    {
      "dynamics_loss_avg": 0.06334132626652718,
      "epoch": 8.544921875e-06,
      "step": 8960
    },
    {
      "action_loss_avg": 0.009405003907158972,
      "epoch": 8.544921875e-06,
      "step": 8960
    },
    {
      "epoch": 8.544921875e-06,
      "step": 8960,
      "training_step_time": 17.12658190727234
    },
    {
      "epoch": 8.545875549316407e-06,
      "model_forward_time": 5.597822427749634,
      "step": 8961
    },
    {
      "epoch": 8.545875549316407e-06,
      "step": 8961,
      "training_step_time": 17.064833164215088
    },
    {
      "epoch": 8.546829223632812e-06,
      "model_forward_time": 5.590169668197632,
      "step": 8962
    },
    {
      "epoch": 8.546829223632812e-06,
      "step": 8962,
      "training_step_time": 17.029107332229614
    },
    {
      "epoch": 8.547782897949218e-06,
      "model_forward_time": 5.583259344100952,
      "step": 8963
    },
    {
      "epoch": 8.547782897949218e-06,
      "step": 8963,
      "training_step_time": 17.033426523208618
    },
    {
      "epoch": 8.548736572265625e-06,
      "model_forward_time": 5.9275500774383545,
      "step": 8964
    },
    {
      "epoch": 8.548736572265625e-06,
      "step": 8964,
      "training_step_time": 17.38187599182129
    },
    {
      "epoch": 8.549690246582031e-06,
      "model_forward_time": 5.59289813041687,
      "step": 8965
    },
    {
      "epoch": 8.549690246582031e-06,
      "step": 8965,
      "training_step_time": 17.067442893981934
    },
    {
      "epoch": 8.550643920898437e-06,
      "model_forward_time": 5.606442213058472,
      "step": 8966
    },
    {
      "epoch": 8.550643920898437e-06,
      "step": 8966,
      "training_step_time": 17.08398127555847
    },
    {
      "epoch": 8.551597595214844e-06,
      "model_forward_time": 5.608642101287842,
      "step": 8967
    },
    {
      "epoch": 8.551597595214844e-06,
      "step": 8967,
      "training_step_time": 17.0872905254364
    },
    {
      "epoch": 8.55255126953125e-06,
      "model_forward_time": 5.595864295959473,
      "step": 8968
    },
    {
      "epoch": 8.55255126953125e-06,
      "step": 8968,
      "training_step_time": 17.01280379295349
    },
    {
      "epoch": 8.553504943847656e-06,
      "model_forward_time": 5.9565489292144775,
      "step": 8969
    },
    {
      "epoch": 8.553504943847656e-06,
      "step": 8969,
      "training_step_time": 17.365275144577026
    },
    {
      "epoch": 8.554458618164063e-06,
      "grad_norm": 1.0597431659698486,
      "learning_rate": 1e-05,
      "loss": 0.0711,
      "step": 8970
    },
    {
      "epoch": 8.554458618164063e-06,
      "model_forward_time": 5.572211503982544,
      "step": 8970
    },
    {
      "dynamics_loss_avg": 0.06410357020795346,
      "epoch": 8.554458618164063e-06,
      "step": 8970
    },
    {
      "action_loss_avg": 0.010362204117700457,
      "epoch": 8.554458618164063e-06,
      "step": 8970
    },
    {
      "epoch": 8.554458618164063e-06,
      "step": 8970,
      "training_step_time": 17.02544903755188
    },
    {
      "epoch": 8.555412292480469e-06,
      "model_forward_time": 5.578582286834717,
      "step": 8971
    },
    {
      "epoch": 8.555412292480469e-06,
      "step": 8971,
      "training_step_time": 17.002074003219604
    },
    {
      "epoch": 8.556365966796875e-06,
      "model_forward_time": 5.584913969039917,
      "step": 8972
    },
    {
      "epoch": 8.556365966796875e-06,
      "step": 8972,
      "training_step_time": 16.995718240737915
    },
    {
      "epoch": 8.557319641113282e-06,
      "model_forward_time": 5.594066619873047,
      "step": 8973
    },
    {
      "epoch": 8.557319641113282e-06,
      "step": 8973,
      "training_step_time": 16.996031522750854
    },
    {
      "epoch": 8.558273315429688e-06,
      "model_forward_time": 5.58072829246521,
      "step": 8974
    },
    {
      "epoch": 8.558273315429688e-06,
      "step": 8974,
      "training_step_time": 17.028111457824707
    },
    {
      "epoch": 8.559226989746093e-06,
      "model_forward_time": 5.95007586479187,
      "step": 8975
    },
    {
      "epoch": 8.559226989746093e-06,
      "step": 8975,
      "training_step_time": 17.41671586036682
    },
    {
      "epoch": 8.5601806640625e-06,
      "model_forward_time": 5.5978968143463135,
      "step": 8976
    },
    {
      "epoch": 8.5601806640625e-06,
      "step": 8976,
      "training_step_time": 17.08786416053772
    },
    {
      "epoch": 8.561134338378907e-06,
      "model_forward_time": 5.6020872592926025,
      "step": 8977
    },
    {
      "epoch": 8.561134338378907e-06,
      "step": 8977,
      "training_step_time": 17.06795573234558
    },
    {
      "epoch": 8.562088012695312e-06,
      "model_forward_time": 5.598342180252075,
      "step": 8978
    },
    {
      "epoch": 8.562088012695312e-06,
      "step": 8978,
      "training_step_time": 17.062575817108154
    },
    {
      "epoch": 8.563041687011718e-06,
      "model_forward_time": 5.605810165405273,
      "step": 8979
    },
    {
      "epoch": 8.563041687011718e-06,
      "step": 8979,
      "training_step_time": 17.05803418159485
    },
    {
      "epoch": 8.563995361328125e-06,
      "grad_norm": 0.9243668913841248,
      "learning_rate": 1e-05,
      "loss": 0.0723,
      "step": 8980
    },
    {
      "epoch": 8.563995361328125e-06,
      "model_forward_time": 5.966990232467651,
      "step": 8980
    },
    {
      "dynamics_loss_avg": 0.05986527837812901,
      "epoch": 8.563995361328125e-06,
      "step": 8980
    },
    {
      "action_loss_avg": 0.009117613127455116,
      "epoch": 8.563995361328125e-06,
      "step": 8980
    },
    {
      "epoch": 8.563995361328125e-06,
      "step": 8980,
      "training_step_time": 17.39564824104309
    },
    {
      "epoch": 8.564949035644531e-06,
      "model_forward_time": 5.592276573181152,
      "step": 8981
    },
    {
      "epoch": 8.564949035644531e-06,
      "step": 8981,
      "training_step_time": 17.050480604171753
    },
    {
      "epoch": 8.565902709960937e-06,
      "model_forward_time": 5.620398759841919,
      "step": 8982
    },
    {
      "epoch": 8.565902709960937e-06,
      "step": 8982,
      "training_step_time": 17.102831602096558
    },
    {
      "epoch": 8.566856384277344e-06,
      "model_forward_time": 5.610609769821167,
      "step": 8983
    },
    {
      "epoch": 8.566856384277344e-06,
      "step": 8983,
      "training_step_time": 17.083614826202393
    },
    {
      "epoch": 8.56781005859375e-06,
      "model_forward_time": 5.593392610549927,
      "step": 8984
    },
    {
      "epoch": 8.56781005859375e-06,
      "step": 8984,
      "training_step_time": 17.019486665725708
    },
    {
      "epoch": 8.568763732910156e-06,
      "model_forward_time": 5.928348779678345,
      "step": 8985
    },
    {
      "epoch": 8.568763732910156e-06,
      "step": 8985,
      "training_step_time": 17.36069631576538
    },
    {
      "epoch": 8.569717407226563e-06,
      "model_forward_time": 5.585596561431885,
      "step": 8986
    },
    {
      "epoch": 8.569717407226563e-06,
      "step": 8986,
      "training_step_time": 17.019823789596558
    },
    {
      "epoch": 8.570671081542969e-06,
      "model_forward_time": 5.595916271209717,
      "step": 8987
    },
    {
      "epoch": 8.570671081542969e-06,
      "step": 8987,
      "training_step_time": 17.058722257614136
    },
    {
      "epoch": 8.571624755859375e-06,
      "model_forward_time": 5.600548982620239,
      "step": 8988
    },
    {
      "epoch": 8.571624755859375e-06,
      "step": 8988,
      "training_step_time": 17.05945062637329
    },
    {
      "epoch": 8.572578430175782e-06,
      "model_forward_time": 5.600275278091431,
      "step": 8989
    },
    {
      "epoch": 8.572578430175782e-06,
      "step": 8989,
      "training_step_time": 17.06683921813965
    },
    {
      "epoch": 8.573532104492188e-06,
      "grad_norm": 0.2809988856315613,
      "learning_rate": 1e-05,
      "loss": 0.0639,
      "step": 8990
    },
    {
      "epoch": 8.573532104492188e-06,
      "model_forward_time": 5.603879928588867,
      "step": 8990
    },
    {
      "dynamics_loss_avg": 0.05608661472797394,
      "epoch": 8.573532104492188e-06,
      "step": 8990
    },
    {
      "action_loss_avg": 0.009934904496185481,
      "epoch": 8.573532104492188e-06,
      "step": 8990
    },
    {
      "epoch": 8.573532104492188e-06,
      "step": 8990,
      "training_step_time": 17.102832317352295
    },
    {
      "epoch": 8.574485778808593e-06,
      "model_forward_time": 5.971635341644287,
      "step": 8991
    },
    {
      "epoch": 8.574485778808593e-06,
      "step": 8991,
      "training_step_time": 17.46750783920288
    },
    {
      "epoch": 8.575439453125e-06,
      "model_forward_time": 5.609019756317139,
      "step": 8992
    },
    {
      "epoch": 8.575439453125e-06,
      "step": 8992,
      "training_step_time": 17.085611820220947
    },
    {
      "epoch": 8.576393127441407e-06,
      "model_forward_time": 5.610032320022583,
      "step": 8993
    },
    {
      "epoch": 8.576393127441407e-06,
      "step": 8993,
      "training_step_time": 17.05504822731018
    },
    {
      "epoch": 8.577346801757812e-06,
      "model_forward_time": 5.597217798233032,
      "step": 8994
    },
    {
      "epoch": 8.577346801757812e-06,
      "step": 8994,
      "training_step_time": 17.05273985862732
    },
    {
      "epoch": 8.578300476074218e-06,
      "model_forward_time": 5.590284585952759,
      "step": 8995
    },
    {
      "epoch": 8.578300476074218e-06,
      "step": 8995,
      "training_step_time": 17.052926301956177
    },
    {
      "epoch": 8.579254150390625e-06,
      "model_forward_time": 5.596753835678101,
      "step": 8996
    },
    {
      "epoch": 8.579254150390625e-06,
      "step": 8996,
      "training_step_time": 17.431943655014038
    },
    {
      "epoch": 8.580207824707031e-06,
      "model_forward_time": 5.612253189086914,
      "step": 8997
    },
    {
      "epoch": 8.580207824707031e-06,
      "step": 8997,
      "training_step_time": 17.11777091026306
    },
    {
      "epoch": 8.581161499023437e-06,
      "model_forward_time": 5.627666234970093,
      "step": 8998
    },
    {
      "epoch": 8.581161499023437e-06,
      "step": 8998,
      "training_step_time": 17.13742756843567
    },
    {
      "epoch": 8.582115173339844e-06,
      "model_forward_time": 5.618672609329224,
      "step": 8999
    },
    {
      "epoch": 8.582115173339844e-06,
      "step": 8999,
      "training_step_time": 17.12069058418274
    },
    {
      "epoch": 8.58306884765625e-06,
      "grad_norm": 0.7919960021972656,
      "learning_rate": 1e-05,
      "loss": 0.0759,
      "step": 9000
    },
    {
      "epoch": 8.58306884765625e-06,
      "model_forward_time": 5.659205198287964,
      "step": 9000
    },
    {
      "dynamics_loss_avg": 0.06176130492240191,
      "epoch": 8.58306884765625e-06,
      "step": 9000
    },
    {
      "action_loss_avg": 0.009329581179190426,
      "epoch": 8.58306884765625e-06,
      "step": 9000
    },
    {
      "epoch": 8.58306884765625e-06,
      "step": 9000,
      "training_step_time": 17.147499799728394
    },
    {
      "epoch": 8.584022521972656e-06,
      "model_forward_time": 5.667737722396851,
      "step": 9001
    },
    {
      "epoch": 8.584022521972656e-06,
      "step": 9001,
      "training_step_time": 17.145298719406128
    },
    {
      "epoch": 8.584976196289063e-06,
      "model_forward_time": 6.14581561088562,
      "step": 9002
    },
    {
      "epoch": 8.584976196289063e-06,
      "step": 9002,
      "training_step_time": 17.659406900405884
    },
    {
      "epoch": 8.585929870605469e-06,
      "model_forward_time": 5.653857946395874,
      "step": 9003
    },
    {
      "epoch": 8.585929870605469e-06,
      "step": 9003,
      "training_step_time": 17.160902738571167
    },
    {
      "epoch": 8.586883544921875e-06,
      "model_forward_time": 5.62157416343689,
      "step": 9004
    },
    {
      "epoch": 8.586883544921875e-06,
      "step": 9004,
      "training_step_time": 17.114402770996094
    },
    {
      "epoch": 8.587837219238282e-06,
      "model_forward_time": 5.614687442779541,
      "step": 9005
    },
    {
      "epoch": 8.587837219238282e-06,
      "step": 9005,
      "training_step_time": 17.0947208404541
    },
    {
      "epoch": 8.588790893554688e-06,
      "model_forward_time": 5.605768918991089,
      "step": 9006
    },
    {
      "epoch": 8.588790893554688e-06,
      "step": 9006,
      "training_step_time": 17.082736492156982
    },
    {
      "epoch": 8.589744567871093e-06,
      "model_forward_time": 5.990762233734131,
      "step": 9007
    },
    {
      "epoch": 8.589744567871093e-06,
      "step": 9007,
      "training_step_time": 17.456241607666016
    },
    {
      "epoch": 8.5906982421875e-06,
      "model_forward_time": 5.594344615936279,
      "step": 9008
    },
    {
      "epoch": 8.5906982421875e-06,
      "step": 9008,
      "training_step_time": 17.021485090255737
    },
    {
      "epoch": 8.591651916503907e-06,
      "model_forward_time": 5.605033874511719,
      "step": 9009
    },
    {
      "epoch": 8.591651916503907e-06,
      "step": 9009,
      "training_step_time": 17.083892583847046
    },
    {
      "epoch": 8.592605590820312e-06,
      "grad_norm": 1.0172666311264038,
      "learning_rate": 1e-05,
      "loss": 0.0746,
      "step": 9010
    },
    {
      "epoch": 8.592605590820312e-06,
      "model_forward_time": 5.609557867050171,
      "step": 9010
    },
    {
      "dynamics_loss_avg": 0.07453068867325782,
      "epoch": 8.592605590820312e-06,
      "step": 9010
    },
    {
      "action_loss_avg": 0.010690653789788484,
      "epoch": 8.592605590820312e-06,
      "step": 9010
    },
    {
      "epoch": 8.592605590820312e-06,
      "step": 9010,
      "training_step_time": 17.106497049331665
    },
    {
      "epoch": 8.593559265136718e-06,
      "model_forward_time": 5.614880800247192,
      "step": 9011
    },
    {
      "epoch": 8.593559265136718e-06,
      "step": 9011,
      "training_step_time": 17.101277828216553
    },
    {
      "epoch": 8.594512939453125e-06,
      "model_forward_time": 5.613906383514404,
      "step": 9012
    },
    {
      "epoch": 8.594512939453125e-06,
      "step": 9012,
      "training_step_time": 17.11923336982727
    },
    {
      "epoch": 8.595466613769531e-06,
      "model_forward_time": 5.964354515075684,
      "step": 9013
    },
    {
      "epoch": 8.595466613769531e-06,
      "step": 9013,
      "training_step_time": 17.471912145614624
    },
    {
      "epoch": 8.596420288085937e-06,
      "model_forward_time": 5.608207941055298,
      "step": 9014
    },
    {
      "epoch": 8.596420288085937e-06,
      "step": 9014,
      "training_step_time": 17.094331979751587
    },
    {
      "epoch": 8.597373962402344e-06,
      "model_forward_time": 5.620894193649292,
      "step": 9015
    },
    {
      "epoch": 8.597373962402344e-06,
      "step": 9015,
      "training_step_time": 17.105830907821655
    },
    {
      "epoch": 8.59832763671875e-06,
      "model_forward_time": 5.620234251022339,
      "step": 9016
    },
    {
      "epoch": 8.59832763671875e-06,
      "step": 9016,
      "training_step_time": 17.090203523635864
    },
    {
      "epoch": 8.599281311035156e-06,
      "model_forward_time": 5.608226537704468,
      "step": 9017
    },
    {
      "epoch": 8.599281311035156e-06,
      "step": 9017,
      "training_step_time": 17.08370852470398
    },
    {
      "epoch": 8.600234985351563e-06,
      "model_forward_time": 5.971683025360107,
      "step": 9018
    },
    {
      "epoch": 8.600234985351563e-06,
      "step": 9018,
      "training_step_time": 17.423714876174927
    },
    {
      "epoch": 8.601188659667969e-06,
      "model_forward_time": 5.5899434089660645,
      "step": 9019
    },
    {
      "epoch": 8.601188659667969e-06,
      "step": 9019,
      "training_step_time": 17.031289100646973
    },
    {
      "epoch": 8.602142333984375e-06,
      "grad_norm": 0.8726795315742493,
      "learning_rate": 1e-05,
      "loss": 0.0666,
      "step": 9020
    },
    {
      "epoch": 8.602142333984375e-06,
      "model_forward_time": 5.577665090560913,
      "step": 9020
    },
    {
      "dynamics_loss_avg": 0.05150498878210783,
      "epoch": 8.602142333984375e-06,
      "step": 9020
    },
    {
      "action_loss_avg": 0.006644154153764248,
      "epoch": 8.602142333984375e-06,
      "step": 9020
    },
    {
      "epoch": 8.602142333984375e-06,
      "step": 9020,
      "training_step_time": 16.98243808746338
    },
    {
      "epoch": 8.603096008300782e-06,
      "model_forward_time": 5.6145546436309814,
      "step": 9021
    },
    {
      "epoch": 8.603096008300782e-06,
      "step": 9021,
      "training_step_time": 17.054280757904053
    },
    {
      "epoch": 8.604049682617188e-06,
      "model_forward_time": 5.589445352554321,
      "step": 9022
    },
    {
      "epoch": 8.604049682617188e-06,
      "step": 9022,
      "training_step_time": 17.025070667266846
    },
    {
      "epoch": 8.605003356933593e-06,
      "model_forward_time": 5.596208810806274,
      "step": 9023
    },
    {
      "epoch": 8.605003356933593e-06,
      "step": 9023,
      "training_step_time": 17.079902410507202
    },
    {
      "epoch": 8.60595703125e-06,
      "model_forward_time": 5.97478723526001,
      "step": 9024
    },
    {
      "epoch": 8.60595703125e-06,
      "step": 9024,
      "training_step_time": 17.495670318603516
    },
    {
      "epoch": 8.606910705566407e-06,
      "model_forward_time": 5.627472400665283,
      "step": 9025
    },
    {
      "epoch": 8.606910705566407e-06,
      "step": 9025,
      "training_step_time": 17.146066427230835
    },
    {
      "epoch": 8.607864379882812e-06,
      "model_forward_time": 5.617106676101685,
      "step": 9026
    },
    {
      "epoch": 8.607864379882812e-06,
      "step": 9026,
      "training_step_time": 17.122923612594604
    },
    {
      "epoch": 8.608818054199218e-06,
      "model_forward_time": 5.609661340713501,
      "step": 9027
    },
    {
      "epoch": 8.608818054199218e-06,
      "step": 9027,
      "training_step_time": 17.099310874938965
    },
    {
      "epoch": 8.609771728515625e-06,
      "model_forward_time": 5.607909679412842,
      "step": 9028
    },
    {
      "epoch": 8.609771728515625e-06,
      "step": 9028,
      "training_step_time": 17.08748722076416
    },
    {
      "epoch": 8.610725402832031e-06,
      "model_forward_time": 5.952336549758911,
      "step": 9029
    },
    {
      "epoch": 8.610725402832031e-06,
      "step": 9029,
      "training_step_time": 17.39868712425232
    },
    {
      "epoch": 8.611679077148437e-06,
      "grad_norm": 1.1110153198242188,
      "learning_rate": 1e-05,
      "loss": 0.0635,
      "step": 9030
    },
    {
      "epoch": 8.611679077148437e-06,
      "model_forward_time": 5.589744806289673,
      "step": 9030
    },
    {
      "dynamics_loss_avg": 0.05681218160316348,
      "epoch": 8.611679077148437e-06,
      "step": 9030
    },
    {
      "action_loss_avg": 0.009713302424643189,
      "epoch": 8.611679077148437e-06,
      "step": 9030
    },
    {
      "epoch": 8.611679077148437e-06,
      "step": 9030,
      "training_step_time": 17.01862359046936
    },
    {
      "epoch": 8.612632751464844e-06,
      "model_forward_time": 5.594007253646851,
      "step": 9031
    },
    {
      "epoch": 8.612632751464844e-06,
      "step": 9031,
      "training_step_time": 17.05279231071472
    },
    {
      "epoch": 8.61358642578125e-06,
      "model_forward_time": 5.602133274078369,
      "step": 9032
    },
    {
      "epoch": 8.61358642578125e-06,
      "step": 9032,
      "training_step_time": 17.088643550872803
    },
    {
      "epoch": 8.614540100097656e-06,
      "model_forward_time": 5.620175838470459,
      "step": 9033
    },
    {
      "epoch": 8.614540100097656e-06,
      "step": 9033,
      "training_step_time": 17.136150598526
    },
    {
      "epoch": 8.615493774414063e-06,
      "model_forward_time": 5.6130969524383545,
      "step": 9034
    },
    {
      "epoch": 8.615493774414063e-06,
      "step": 9034,
      "training_step_time": 17.476922273635864
    },
    {
      "epoch": 8.616447448730469e-06,
      "model_forward_time": 5.621349573135376,
      "step": 9035
    },
    {
      "epoch": 8.616447448730469e-06,
      "step": 9035,
      "training_step_time": 17.100507259368896
    },
    {
      "epoch": 8.617401123046875e-06,
      "model_forward_time": 5.5916523933410645,
      "step": 9036
    },
    {
      "epoch": 8.617401123046875e-06,
      "step": 9036,
      "training_step_time": 17.040420532226562
    },
    {
      "epoch": 8.618354797363282e-06,
      "model_forward_time": 5.587713718414307,
      "step": 9037
    },
    {
      "epoch": 8.618354797363282e-06,
      "step": 9037,
      "training_step_time": 17.045343160629272
    },
    {
      "epoch": 8.619308471679688e-06,
      "model_forward_time": 5.595669507980347,
      "step": 9038
    },
    {
      "epoch": 8.619308471679688e-06,
      "step": 9038,
      "training_step_time": 17.087754011154175
    },
    {
      "epoch": 8.620262145996093e-06,
      "model_forward_time": 5.6119301319122314,
      "step": 9039
    },
    {
      "epoch": 8.620262145996093e-06,
      "step": 9039,
      "training_step_time": 17.107770681381226
    },
    {
      "epoch": 8.6212158203125e-06,
      "grad_norm": 0.4613456130027771,
      "learning_rate": 1e-05,
      "loss": 0.0558,
      "step": 9040
    },
    {
      "epoch": 8.6212158203125e-06,
      "model_forward_time": 5.993391036987305,
      "step": 9040
    },
    {
      "dynamics_loss_avg": 0.05201516505330801,
      "epoch": 8.6212158203125e-06,
      "step": 9040
    },
    {
      "action_loss_avg": 0.006285740854218602,
      "epoch": 8.6212158203125e-06,
      "step": 9040
    },
    {
      "epoch": 8.6212158203125e-06,
      "step": 9040,
      "training_step_time": 17.492345571517944
    },
    {
      "epoch": 8.622169494628907e-06,
      "model_forward_time": 5.619285345077515,
      "step": 9041
    },
    {
      "epoch": 8.622169494628907e-06,
      "step": 9041,
      "training_step_time": 17.11978507041931
    },
    {
      "epoch": 8.623123168945312e-06,
      "model_forward_time": 5.615160703659058,
      "step": 9042
    },
    {
      "epoch": 8.623123168945312e-06,
      "step": 9042,
      "training_step_time": 17.06804847717285
    },
    {
      "epoch": 8.624076843261718e-06,
      "model_forward_time": 5.589123964309692,
      "step": 9043
    },
    {
      "epoch": 8.624076843261718e-06,
      "step": 9043,
      "training_step_time": 16.994288682937622
    },
    {
      "epoch": 8.625030517578125e-06,
      "model_forward_time": 5.57293176651001,
      "step": 9044
    },
    {
      "epoch": 8.625030517578125e-06,
      "step": 9044,
      "training_step_time": 16.998650074005127
    },
    {
      "epoch": 8.625984191894531e-06,
      "model_forward_time": 5.928899526596069,
      "step": 9045
    },
    {
      "epoch": 8.625984191894531e-06,
      "step": 9045,
      "training_step_time": 17.342422246932983
    },
    {
      "epoch": 8.626937866210937e-06,
      "model_forward_time": 5.5695576667785645,
      "step": 9046
    },
    {
      "epoch": 8.626937866210937e-06,
      "step": 9046,
      "training_step_time": 17.013511896133423
    },
    {
      "epoch": 8.627891540527344e-06,
      "model_forward_time": 5.569185495376587,
      "step": 9047
    },
    {
      "epoch": 8.627891540527344e-06,
      "step": 9047,
      "training_step_time": 16.985871076583862
    },
    {
      "epoch": 8.62884521484375e-06,
      "model_forward_time": 5.596576690673828,
      "step": 9048
    },
    {
      "epoch": 8.62884521484375e-06,
      "step": 9048,
      "training_step_time": 17.017707586288452
    },
    {
      "epoch": 8.629798889160156e-06,
      "model_forward_time": 5.581719875335693,
      "step": 9049
    },
    {
      "epoch": 8.629798889160156e-06,
      "step": 9049,
      "training_step_time": 17.010432958602905
    },
    {
      "epoch": 8.630752563476563e-06,
      "grad_norm": 1.119616985321045,
      "learning_rate": 1e-05,
      "loss": 0.074,
      "step": 9050
    },
    {
      "epoch": 8.630752563476563e-06,
      "model_forward_time": 5.580560207366943,
      "step": 9050
    },
    {
      "dynamics_loss_avg": 0.061639531329274175,
      "epoch": 8.630752563476563e-06,
      "step": 9050
    },
    {
      "action_loss_avg": 0.010548701859079301,
      "epoch": 8.630752563476563e-06,
      "step": 9050
    },
    {
      "epoch": 8.630752563476563e-06,
      "step": 9050,
      "training_step_time": 17.019659280776978
    },
    {
      "epoch": 8.631706237792969e-06,
      "model_forward_time": 5.948766469955444,
      "step": 9051
    },
    {
      "epoch": 8.631706237792969e-06,
      "step": 9051,
      "training_step_time": 17.38670301437378
    },
    {
      "epoch": 8.632659912109375e-06,
      "model_forward_time": 5.592577695846558,
      "step": 9052
    },
    {
      "epoch": 8.632659912109375e-06,
      "step": 9052,
      "training_step_time": 17.066063165664673
    },
    {
      "epoch": 8.633613586425782e-06,
      "model_forward_time": 5.6006996631622314,
      "step": 9053
    },
    {
      "epoch": 8.633613586425782e-06,
      "step": 9053,
      "training_step_time": 17.0626380443573
    },
    {
      "epoch": 8.634567260742188e-06,
      "model_forward_time": 5.614134311676025,
      "step": 9054
    },
    {
      "epoch": 8.634567260742188e-06,
      "step": 9054,
      "training_step_time": 17.08005976676941
    },
    {
      "epoch": 8.635520935058593e-06,
      "model_forward_time": 5.597081661224365,
      "step": 9055
    },
    {
      "epoch": 8.635520935058593e-06,
      "step": 9055,
      "training_step_time": 17.05982756614685
    },
    {
      "epoch": 8.636474609375e-06,
      "model_forward_time": 5.9490647315979,
      "step": 9056
    },
    {
      "epoch": 8.636474609375e-06,
      "step": 9056,
      "training_step_time": 17.379507780075073
    },
    {
      "epoch": 8.637428283691407e-06,
      "model_forward_time": 5.616026401519775,
      "step": 9057
    },
    {
      "epoch": 8.637428283691407e-06,
      "step": 9057,
      "training_step_time": 17.053887367248535
    },
    {
      "epoch": 8.638381958007812e-06,
      "model_forward_time": 5.593122959136963,
      "step": 9058
    },
    {
      "epoch": 8.638381958007812e-06,
      "step": 9058,
      "training_step_time": 17.061012744903564
    },
    {
      "epoch": 8.639335632324218e-06,
      "model_forward_time": 5.601136207580566,
      "step": 9059
    },
    {
      "epoch": 8.639335632324218e-06,
      "step": 9059,
      "training_step_time": 17.070976734161377
    },
    {
      "epoch": 8.640289306640625e-06,
      "grad_norm": 1.2461836338043213,
      "learning_rate": 1e-05,
      "loss": 0.0769,
      "step": 9060
    },
    {
      "epoch": 8.640289306640625e-06,
      "model_forward_time": 5.606233596801758,
      "step": 9060
    },
    {
      "dynamics_loss_avg": 0.06168291755020618,
      "epoch": 8.640289306640625e-06,
      "step": 9060
    },
    {
      "action_loss_avg": 0.012368237180635334,
      "epoch": 8.640289306640625e-06,
      "step": 9060
    },
    {
      "epoch": 8.640289306640625e-06,
      "step": 9060,
      "training_step_time": 17.06398892402649
    },
    {
      "epoch": 8.641242980957031e-06,
      "model_forward_time": 5.600845813751221,
      "step": 9061
    },
    {
      "epoch": 8.641242980957031e-06,
      "step": 9061,
      "training_step_time": 17.05211639404297
    },
    {
      "epoch": 8.642196655273437e-06,
      "model_forward_time": 5.9597978591918945,
      "step": 9062
    },
    {
      "epoch": 8.642196655273437e-06,
      "step": 9062,
      "training_step_time": 17.36918616294861
    },
    {
      "epoch": 8.643150329589844e-06,
      "model_forward_time": 5.6350414752960205,
      "step": 9063
    },
    {
      "epoch": 8.643150329589844e-06,
      "step": 9063,
      "training_step_time": 17.095452308654785
    },
    {
      "epoch": 8.64410400390625e-06,
      "model_forward_time": 5.609543561935425,
      "step": 9064
    },
    {
      "epoch": 8.64410400390625e-06,
      "step": 9064,
      "training_step_time": 17.098069667816162
    },
    {
      "epoch": 8.645057678222656e-06,
      "model_forward_time": 5.6100780963897705,
      "step": 9065
    },
    {
      "epoch": 8.645057678222656e-06,
      "step": 9065,
      "training_step_time": 17.11397671699524
    },
    {
      "epoch": 8.646011352539063e-06,
      "model_forward_time": 5.6463892459869385,
      "step": 9066
    },
    {
      "epoch": 8.646011352539063e-06,
      "step": 9066,
      "training_step_time": 17.1760995388031
    },
    {
      "epoch": 8.646965026855469e-06,
      "model_forward_time": 6.006287097930908,
      "step": 9067
    },
    {
      "epoch": 8.646965026855469e-06,
      "step": 9067,
      "training_step_time": 17.456966161727905
    },
    {
      "epoch": 8.647918701171875e-06,
      "model_forward_time": 5.611423015594482,
      "step": 9068
    },
    {
      "epoch": 8.647918701171875e-06,
      "step": 9068,
      "training_step_time": 17.03479504585266
    },
    {
      "epoch": 8.648872375488282e-06,
      "model_forward_time": 5.5947442054748535,
      "step": 9069
    },
    {
      "epoch": 8.648872375488282e-06,
      "step": 9069,
      "training_step_time": 17.034008264541626
    },
    {
      "epoch": 8.649826049804688e-06,
      "grad_norm": 1.5728156566619873,
      "learning_rate": 1e-05,
      "loss": 0.0694,
      "step": 9070
    },
    {
      "epoch": 8.649826049804688e-06,
      "model_forward_time": 5.6105475425720215,
      "step": 9070
    },
    {
      "dynamics_loss_avg": 0.060569139569997786,
      "epoch": 8.649826049804688e-06,
      "step": 9070
    },
    {
      "action_loss_avg": 0.01018640366382897,
      "epoch": 8.649826049804688e-06,
      "step": 9070
    },
    {
      "epoch": 8.649826049804688e-06,
      "step": 9070,
      "training_step_time": 17.05682373046875
    },
    {
      "epoch": 8.650779724121093e-06,
      "model_forward_time": 5.5797059535980225,
      "step": 9071
    },
    {
      "epoch": 8.650779724121093e-06,
      "step": 9071,
      "training_step_time": 17.014071702957153
    },
    {
      "epoch": 8.6517333984375e-06,
      "model_forward_time": 5.57635760307312,
      "step": 9072
    },
    {
      "epoch": 8.6517333984375e-06,
      "step": 9072,
      "training_step_time": 17.40360140800476
    },
    {
      "epoch": 8.652687072753907e-06,
      "model_forward_time": 5.622914552688599,
      "step": 9073
    },
    {
      "epoch": 8.652687072753907e-06,
      "step": 9073,
      "training_step_time": 17.114898920059204
    },
    {
      "epoch": 8.653640747070312e-06,
      "model_forward_time": 5.612284183502197,
      "step": 9074
    },
    {
      "epoch": 8.653640747070312e-06,
      "step": 9074,
      "training_step_time": 17.09643292427063
    },
    {
      "epoch": 8.654594421386718e-06,
      "model_forward_time": 5.605131387710571,
      "step": 9075
    },
    {
      "epoch": 8.654594421386718e-06,
      "step": 9075,
      "training_step_time": 17.088542699813843
    },
    {
      "epoch": 8.655548095703125e-06,
      "model_forward_time": 5.611622095108032,
      "step": 9076
    },
    {
      "epoch": 8.655548095703125e-06,
      "step": 9076,
      "training_step_time": 17.10302472114563
    },
    {
      "epoch": 8.656501770019531e-06,
      "model_forward_time": 5.6131603717803955,
      "step": 9077
    },
    {
      "epoch": 8.656501770019531e-06,
      "step": 9077,
      "training_step_time": 17.10803198814392
    },
    {
      "epoch": 8.657455444335937e-06,
      "model_forward_time": 5.976411581039429,
      "step": 9078
    },
    {
      "epoch": 8.657455444335937e-06,
      "step": 9078,
      "training_step_time": 17.45469641685486
    },
    {
      "epoch": 8.658409118652344e-06,
      "model_forward_time": 5.600422620773315,
      "step": 9079
    },
    {
      "epoch": 8.658409118652344e-06,
      "step": 9079,
      "training_step_time": 17.077717065811157
    },
    {
      "epoch": 8.65936279296875e-06,
      "grad_norm": 0.603899359703064,
      "learning_rate": 1e-05,
      "loss": 0.0529,
      "step": 9080
    },
    {
      "epoch": 8.65936279296875e-06,
      "model_forward_time": 5.596782684326172,
      "step": 9080
    },
    {
      "dynamics_loss_avg": 0.04572098292410374,
      "epoch": 8.65936279296875e-06,
      "step": 9080
    },
    {
      "action_loss_avg": 0.006705465819686651,
      "epoch": 8.65936279296875e-06,
      "step": 9080
    },
    {
      "epoch": 8.65936279296875e-06,
      "step": 9080,
      "training_step_time": 17.050929307937622
    },
    {
      "epoch": 8.660316467285156e-06,
      "model_forward_time": 5.607370138168335,
      "step": 9081
    },
    {
      "epoch": 8.660316467285156e-06,
      "step": 9081,
      "training_step_time": 17.101683378219604
    },
    {
      "epoch": 8.661270141601563e-06,
      "model_forward_time": 5.609935522079468,
      "step": 9082
    },
    {
      "epoch": 8.661270141601563e-06,
      "step": 9082,
      "training_step_time": 17.100842237472534
    },
    {
      "epoch": 8.662223815917969e-06,
      "model_forward_time": 5.611314535140991,
      "step": 9083
    },
    {
      "epoch": 8.662223815917969e-06,
      "step": 9083,
      "training_step_time": 17.462260723114014
    },
    {
      "epoch": 8.663177490234375e-06,
      "model_forward_time": 5.60664439201355,
      "step": 9084
    },
    {
      "epoch": 8.663177490234375e-06,
      "step": 9084,
      "training_step_time": 17.07588815689087
    },
    {
      "epoch": 8.664131164550782e-06,
      "model_forward_time": 5.606884956359863,
      "step": 9085
    },
    {
      "epoch": 8.664131164550782e-06,
      "step": 9085,
      "training_step_time": 17.10578942298889
    },
    {
      "epoch": 8.665084838867188e-06,
      "model_forward_time": 5.602153778076172,
      "step": 9086
    },
    {
      "epoch": 8.665084838867188e-06,
      "step": 9086,
      "training_step_time": 17.087017059326172
    },
    {
      "epoch": 8.666038513183594e-06,
      "model_forward_time": 5.604954957962036,
      "step": 9087
    },
    {
      "epoch": 8.666038513183594e-06,
      "step": 9087,
      "training_step_time": 17.086544275283813
    },
    {
      "epoch": 8.6669921875e-06,
      "model_forward_time": 5.610215663909912,
      "step": 9088
    },
    {
      "epoch": 8.6669921875e-06,
      "step": 9088,
      "training_step_time": 17.074134349822998
    },
    {
      "epoch": 8.667945861816407e-06,
      "model_forward_time": 5.953577756881714,
      "step": 9089
    },
    {
      "epoch": 8.667945861816407e-06,
      "step": 9089,
      "training_step_time": 17.440130949020386
    },
    {
      "epoch": 8.668899536132812e-06,
      "grad_norm": 0.7515230774879456,
      "learning_rate": 1e-05,
      "loss": 0.0604,
      "step": 9090
    },
    {
      "epoch": 8.668899536132812e-06,
      "model_forward_time": 5.5987324714660645,
      "step": 9090
    },
    {
      "dynamics_loss_avg": 0.05178646147251129,
      "epoch": 8.668899536132812e-06,
      "step": 9090
    },
    {
      "action_loss_avg": 0.007753071072511375,
      "epoch": 8.668899536132812e-06,
      "step": 9090
    },
    {
      "epoch": 8.668899536132812e-06,
      "step": 9090,
      "training_step_time": 17.06710433959961
    },
    {
      "epoch": 8.669853210449218e-06,
      "model_forward_time": 5.600528240203857,
      "step": 9091
    },
    {
      "epoch": 8.669853210449218e-06,
      "step": 9091,
      "training_step_time": 17.066582679748535
    },
    {
      "epoch": 8.670806884765626e-06,
      "model_forward_time": 5.603177785873413,
      "step": 9092
    },
    {
      "epoch": 8.670806884765626e-06,
      "step": 9092,
      "training_step_time": 17.061273336410522
    },
    {
      "epoch": 8.671760559082031e-06,
      "model_forward_time": 5.603568077087402,
      "step": 9093
    },
    {
      "epoch": 8.671760559082031e-06,
      "step": 9093,
      "training_step_time": 17.0742244720459
    },
    {
      "epoch": 8.672714233398437e-06,
      "model_forward_time": 5.969667911529541,
      "step": 9094
    },
    {
      "epoch": 8.672714233398437e-06,
      "step": 9094,
      "training_step_time": 17.450902462005615
    },
    {
      "epoch": 8.673667907714844e-06,
      "model_forward_time": 5.610987901687622,
      "step": 9095
    },
    {
      "epoch": 8.673667907714844e-06,
      "step": 9095,
      "training_step_time": 17.091747283935547
    },
    {
      "epoch": 8.67462158203125e-06,
      "model_forward_time": 5.612422704696655,
      "step": 9096
    },
    {
      "epoch": 8.67462158203125e-06,
      "step": 9096,
      "training_step_time": 17.08782196044922
    },
    {
      "epoch": 8.675575256347656e-06,
      "model_forward_time": 5.598127126693726,
      "step": 9097
    },
    {
      "epoch": 8.675575256347656e-06,
      "step": 9097,
      "training_step_time": 17.08274245262146
    },
    {
      "epoch": 8.676528930664063e-06,
      "model_forward_time": 5.613214015960693,
      "step": 9098
    },
    {
      "epoch": 8.676528930664063e-06,
      "step": 9098,
      "training_step_time": 17.087701082229614
    },
    {
      "epoch": 8.677482604980469e-06,
      "model_forward_time": 5.606882572174072,
      "step": 9099
    },
    {
      "epoch": 8.677482604980469e-06,
      "step": 9099,
      "training_step_time": 17.439615726470947
    },
    {
      "epoch": 8.678436279296875e-06,
      "grad_norm": 1.630374789237976,
      "learning_rate": 1e-05,
      "loss": 0.077,
      "step": 9100
    },
    {
      "epoch": 8.678436279296875e-06,
      "model_forward_time": 5.602915048599243,
      "step": 9100
    },
    {
      "dynamics_loss_avg": 0.06726573258638383,
      "epoch": 8.678436279296875e-06,
      "step": 9100
    },
    {
      "action_loss_avg": 0.010117577947676181,
      "epoch": 8.678436279296875e-06,
      "step": 9100
    },
    {
      "epoch": 8.678436279296875e-06,
      "step": 9100,
      "training_step_time": 17.073570013046265
    },
    {
      "epoch": 8.679389953613282e-06,
      "model_forward_time": 5.610374450683594,
      "step": 9101
    },
    {
      "epoch": 8.679389953613282e-06,
      "step": 9101,
      "training_step_time": 17.0939781665802
    },
    {
      "epoch": 8.680343627929688e-06,
      "model_forward_time": 5.607241868972778,
      "step": 9102
    },
    {
      "epoch": 8.680343627929688e-06,
      "step": 9102,
      "training_step_time": 17.092151880264282
    },
    {
      "epoch": 8.681297302246094e-06,
      "model_forward_time": 5.601731777191162,
      "step": 9103
    },
    {
      "epoch": 8.681297302246094e-06,
      "step": 9103,
      "training_step_time": 17.05107569694519
    },
    {
      "epoch": 8.6822509765625e-06,
      "model_forward_time": 5.595156669616699,
      "step": 9104
    },
    {
      "epoch": 8.6822509765625e-06,
      "step": 9104,
      "training_step_time": 17.044173002243042
    },
    {
      "epoch": 8.683204650878907e-06,
      "model_forward_time": 5.967851877212524,
      "step": 9105
    },
    {
      "epoch": 8.683204650878907e-06,
      "step": 9105,
      "training_step_time": 17.38940191268921
    },
    {
      "epoch": 8.684158325195312e-06,
      "model_forward_time": 5.59428334236145,
      "step": 9106
    },
    {
      "epoch": 8.684158325195312e-06,
      "step": 9106,
      "training_step_time": 17.070932388305664
    },
    {
      "epoch": 8.685111999511718e-06,
      "model_forward_time": 5.6073784828186035,
      "step": 9107
    },
    {
      "epoch": 8.685111999511718e-06,
      "step": 9107,
      "training_step_time": 17.101889610290527
    },
    {
      "epoch": 8.686065673828126e-06,
      "model_forward_time": 5.61912989616394,
      "step": 9108
    },
    {
      "epoch": 8.686065673828126e-06,
      "step": 9108,
      "training_step_time": 17.12065029144287
    },
    {
      "epoch": 8.687019348144531e-06,
      "model_forward_time": 5.622336149215698,
      "step": 9109
    },
    {
      "epoch": 8.687019348144531e-06,
      "step": 9109,
      "training_step_time": 17.108680725097656
    },
    {
      "epoch": 8.687973022460937e-06,
      "grad_norm": 0.9006580710411072,
      "learning_rate": 1e-05,
      "loss": 0.0725,
      "step": 9110
    },
    {
      "epoch": 8.687973022460937e-06,
      "model_forward_time": 5.9706244468688965,
      "step": 9110
    },
    {
      "dynamics_loss_avg": 0.061787852086126804,
      "epoch": 8.687973022460937e-06,
      "step": 9110
    },
    {
      "action_loss_avg": 0.013667925889603794,
      "epoch": 8.687973022460937e-06,
      "step": 9110
    },
    {
      "epoch": 8.687973022460937e-06,
      "step": 9110,
      "training_step_time": 17.453284740447998
    },
    {
      "epoch": 8.688926696777344e-06,
      "model_forward_time": 5.619897127151489,
      "step": 9111
    },
    {
      "epoch": 8.688926696777344e-06,
      "step": 9111,
      "training_step_time": 17.08093547821045
    },
    {
      "epoch": 8.68988037109375e-06,
      "model_forward_time": 5.600622892379761,
      "step": 9112
    },
    {
      "epoch": 8.68988037109375e-06,
      "step": 9112,
      "training_step_time": 17.01725149154663
    },
    {
      "epoch": 8.690834045410156e-06,
      "model_forward_time": 5.5801100730896,
      "step": 9113
    },
    {
      "epoch": 8.690834045410156e-06,
      "step": 9113,
      "training_step_time": 17.00125026702881
    },
    {
      "epoch": 8.691787719726563e-06,
      "model_forward_time": 5.565208196640015,
      "step": 9114
    },
    {
      "epoch": 8.691787719726563e-06,
      "step": 9114,
      "training_step_time": 16.995802402496338
    },
    {
      "epoch": 8.692741394042969e-06,
      "model_forward_time": 5.56816029548645,
      "step": 9115
    },
    {
      "epoch": 8.692741394042969e-06,
      "step": 9115,
      "training_step_time": 16.996968507766724
    },
    {
      "epoch": 8.693695068359375e-06,
      "model_forward_time": 5.956356525421143,
      "step": 9116
    },
    {
      "epoch": 8.693695068359375e-06,
      "step": 9116,
      "training_step_time": 17.366941452026367
    },
    {
      "epoch": 8.694648742675782e-06,
      "model_forward_time": 5.603983402252197,
      "step": 9117
    },
    {
      "epoch": 8.694648742675782e-06,
      "step": 9117,
      "training_step_time": 17.040433883666992
    },
    {
      "epoch": 8.695602416992188e-06,
      "model_forward_time": 5.588778257369995,
      "step": 9118
    },
    {
      "epoch": 8.695602416992188e-06,
      "step": 9118,
      "training_step_time": 17.021530389785767
    },
    {
      "epoch": 8.696556091308594e-06,
      "model_forward_time": 5.586652755737305,
      "step": 9119
    },
    {
      "epoch": 8.696556091308594e-06,
      "step": 9119,
      "training_step_time": 17.04026699066162
    },
    {
      "epoch": 8.697509765625e-06,
      "grad_norm": 0.9490432739257812,
      "learning_rate": 1e-05,
      "loss": 0.071,
      "step": 9120
    },
    {
      "epoch": 8.697509765625e-06,
      "model_forward_time": 5.594202518463135,
      "step": 9120
    },
    {
      "dynamics_loss_avg": 0.057481133565306666,
      "epoch": 8.697509765625e-06,
      "step": 9120
    },
    {
      "action_loss_avg": 0.009669673489406705,
      "epoch": 8.697509765625e-06,
      "step": 9120
    },
    {
      "epoch": 8.697509765625e-06,
      "step": 9120,
      "training_step_time": 17.06982660293579
    },
    {
      "epoch": 8.698463439941407e-06,
      "model_forward_time": 5.973085165023804,
      "step": 9121
    },
    {
      "epoch": 8.698463439941407e-06,
      "step": 9121,
      "training_step_time": 17.450453996658325
    },
    {
      "epoch": 8.699417114257812e-06,
      "model_forward_time": 5.60956335067749,
      "step": 9122
    },
    {
      "epoch": 8.699417114257812e-06,
      "step": 9122,
      "training_step_time": 17.063889741897583
    },
    {
      "epoch": 8.700370788574218e-06,
      "model_forward_time": 5.600051164627075,
      "step": 9123
    },
    {
      "epoch": 8.700370788574218e-06,
      "step": 9123,
      "training_step_time": 17.061293840408325
    },
    {
      "epoch": 8.701324462890626e-06,
      "model_forward_time": 5.5957348346710205,
      "step": 9124
    },
    {
      "epoch": 8.701324462890626e-06,
      "step": 9124,
      "training_step_time": 17.062407970428467
    },
    {
      "epoch": 8.702278137207031e-06,
      "model_forward_time": 5.595660448074341,
      "step": 9125
    },
    {
      "epoch": 8.702278137207031e-06,
      "step": 9125,
      "training_step_time": 17.055641412734985
    },
    {
      "epoch": 8.703231811523437e-06,
      "model_forward_time": 5.589484214782715,
      "step": 9126
    },
    {
      "epoch": 8.703231811523437e-06,
      "step": 9126,
      "training_step_time": 17.049880743026733
    },
    {
      "epoch": 8.704185485839844e-06,
      "model_forward_time": 5.966537237167358,
      "step": 9127
    },
    {
      "epoch": 8.704185485839844e-06,
      "step": 9127,
      "training_step_time": 17.427932262420654
    },
    {
      "epoch": 8.70513916015625e-06,
      "model_forward_time": 5.598019361495972,
      "step": 9128
    },
    {
      "epoch": 8.70513916015625e-06,
      "step": 9128,
      "training_step_time": 17.06570339202881
    },
    {
      "epoch": 8.706092834472656e-06,
      "model_forward_time": 5.599017858505249,
      "step": 9129
    },
    {
      "epoch": 8.706092834472656e-06,
      "step": 9129,
      "training_step_time": 17.051655530929565
    },
    {
      "epoch": 8.707046508789063e-06,
      "grad_norm": 1.2346409559249878,
      "learning_rate": 1e-05,
      "loss": 0.0951,
      "step": 9130
    },
    {
      "epoch": 8.707046508789063e-06,
      "model_forward_time": 5.590386152267456,
      "step": 9130
    },
    {
      "dynamics_loss_avg": 0.08150423346087336,
      "epoch": 8.707046508789063e-06,
      "step": 9130
    },
    {
      "action_loss_avg": 0.015123873727861791,
      "epoch": 8.707046508789063e-06,
      "step": 9130
    },
    {
      "epoch": 8.707046508789063e-06,
      "step": 9130,
      "training_step_time": 17.04073977470398
    },
    {
      "epoch": 8.708000183105469e-06,
      "model_forward_time": 5.595681428909302,
      "step": 9131
    },
    {
      "epoch": 8.708000183105469e-06,
      "step": 9131,
      "training_step_time": 17.054391622543335
    },
    {
      "epoch": 8.708953857421875e-06,
      "model_forward_time": 5.93964147567749,
      "step": 9132
    },
    {
      "epoch": 8.708953857421875e-06,
      "step": 9132,
      "training_step_time": 17.394052505493164
    },
    {
      "epoch": 8.70990753173828e-06,
      "model_forward_time": 5.589010715484619,
      "step": 9133
    },
    {
      "epoch": 8.70990753173828e-06,
      "step": 9133,
      "training_step_time": 17.035563707351685
    },
    {
      "epoch": 8.710861206054688e-06,
      "model_forward_time": 5.593736410140991,
      "step": 9134
    },
    {
      "epoch": 8.710861206054688e-06,
      "step": 9134,
      "training_step_time": 17.034398794174194
    },
    {
      "epoch": 8.711814880371094e-06,
      "model_forward_time": 5.588650703430176,
      "step": 9135
    },
    {
      "epoch": 8.711814880371094e-06,
      "step": 9135,
      "training_step_time": 17.054438591003418
    },
    {
      "epoch": 8.7127685546875e-06,
      "model_forward_time": 5.598301887512207,
      "step": 9136
    },
    {
      "epoch": 8.7127685546875e-06,
      "step": 9136,
      "training_step_time": 17.060765266418457
    },
    {
      "epoch": 8.713722229003907e-06,
      "model_forward_time": 5.604288578033447,
      "step": 9137
    },
    {
      "epoch": 8.713722229003907e-06,
      "step": 9137,
      "training_step_time": 17.440125942230225
    },
    {
      "epoch": 8.714675903320312e-06,
      "model_forward_time": 5.607952356338501,
      "step": 9138
    },
    {
      "epoch": 8.714675903320312e-06,
      "step": 9138,
      "training_step_time": 17.070688486099243
    },
    {
      "epoch": 8.715629577636718e-06,
      "model_forward_time": 5.591790437698364,
      "step": 9139
    },
    {
      "epoch": 8.715629577636718e-06,
      "step": 9139,
      "training_step_time": 17.032537698745728
    },
    {
      "epoch": 8.716583251953126e-06,
      "grad_norm": 1.1474777460098267,
      "learning_rate": 1e-05,
      "loss": 0.0859,
      "step": 9140
    },
    {
      "epoch": 8.716583251953126e-06,
      "model_forward_time": 5.593693733215332,
      "step": 9140
    },
    {
      "dynamics_loss_avg": 0.07547312937676906,
      "epoch": 8.716583251953126e-06,
      "step": 9140
    },
    {
      "action_loss_avg": 0.013238302571699024,
      "epoch": 8.716583251953126e-06,
      "step": 9140
    },
    {
      "epoch": 8.716583251953126e-06,
      "step": 9140,
      "training_step_time": 17.00795292854309
    },
    {
      "epoch": 8.717536926269531e-06,
      "model_forward_time": 5.582666635513306,
      "step": 9141
    },
    {
      "epoch": 8.717536926269531e-06,
      "step": 9141,
      "training_step_time": 17.030318021774292
    },
    {
      "epoch": 8.718490600585937e-06,
      "model_forward_time": 5.587922096252441,
      "step": 9142
    },
    {
      "epoch": 8.718490600585937e-06,
      "step": 9142,
      "training_step_time": 17.05155634880066
    },
    {
      "epoch": 8.719444274902344e-06,
      "model_forward_time": 5.952733755111694,
      "step": 9143
    },
    {
      "epoch": 8.719444274902344e-06,
      "step": 9143,
      "training_step_time": 17.44834303855896
    },
    {
      "epoch": 8.72039794921875e-06,
      "model_forward_time": 5.6156721115112305,
      "step": 9144
    },
    {
      "epoch": 8.72039794921875e-06,
      "step": 9144,
      "training_step_time": 17.11150884628296
    },
    {
      "epoch": 8.721351623535156e-06,
      "model_forward_time": 5.596888065338135,
      "step": 9145
    },
    {
      "epoch": 8.721351623535156e-06,
      "step": 9145,
      "training_step_time": 17.04249668121338
    },
    {
      "epoch": 8.722305297851563e-06,
      "model_forward_time": 5.600452899932861,
      "step": 9146
    },
    {
      "epoch": 8.722305297851563e-06,
      "step": 9146,
      "training_step_time": 17.04867386817932
    },
    {
      "epoch": 8.723258972167969e-06,
      "model_forward_time": 5.595090627670288,
      "step": 9147
    },
    {
      "epoch": 8.723258972167969e-06,
      "step": 9147,
      "training_step_time": 17.03028631210327
    },
    {
      "epoch": 8.724212646484375e-06,
      "model_forward_time": 5.962594270706177,
      "step": 9148
    },
    {
      "epoch": 8.724212646484375e-06,
      "step": 9148,
      "training_step_time": 17.41148090362549
    },
    {
      "epoch": 8.72516632080078e-06,
      "model_forward_time": 5.5879621505737305,
      "step": 9149
    },
    {
      "epoch": 8.72516632080078e-06,
      "step": 9149,
      "training_step_time": 17.052504777908325
    },
    {
      "epoch": 8.726119995117188e-06,
      "grad_norm": 1.6294586658477783,
      "learning_rate": 1e-05,
      "loss": 0.1015,
      "step": 9150
    },
    {
      "epoch": 8.726119995117188e-06,
      "model_forward_time": 5.598366975784302,
      "step": 9150
    },
    {
      "dynamics_loss_avg": 0.08220378495752811,
      "epoch": 8.726119995117188e-06,
      "step": 9150
    },
    {
      "action_loss_avg": 0.015998008428141475,
      "epoch": 8.726119995117188e-06,
      "step": 9150
    },
    {
      "epoch": 8.726119995117188e-06,
      "step": 9150,
      "training_step_time": 17.0896155834198
    },
    {
      "epoch": 8.727073669433594e-06,
      "model_forward_time": 5.620017051696777,
      "step": 9151
    },
    {
      "epoch": 8.727073669433594e-06,
      "step": 9151,
      "training_step_time": 17.133896112442017
    },
    {
      "epoch": 8.72802734375e-06,
      "model_forward_time": 5.611533164978027,
      "step": 9152
    },
    {
      "epoch": 8.72802734375e-06,
      "step": 9152,
      "training_step_time": 17.114856243133545
    },
    {
      "epoch": 8.728981018066407e-06,
      "model_forward_time": 5.615466117858887,
      "step": 9153
    },
    {
      "epoch": 8.728981018066407e-06,
      "step": 9153,
      "training_step_time": 17.081650972366333
    },
    {
      "epoch": 8.729934692382812e-06,
      "model_forward_time": 5.954511642456055,
      "step": 9154
    },
    {
      "epoch": 8.729934692382812e-06,
      "step": 9154,
      "training_step_time": 17.398546934127808
    },
    {
      "epoch": 8.730888366699218e-06,
      "model_forward_time": 5.579464435577393,
      "step": 9155
    },
    {
      "epoch": 8.730888366699218e-06,
      "step": 9155,
      "training_step_time": 17.003819227218628
    },
    {
      "epoch": 8.731842041015626e-06,
      "model_forward_time": 5.570341348648071,
      "step": 9156
    },
    {
      "epoch": 8.731842041015626e-06,
      "step": 9156,
      "training_step_time": 17.008720874786377
    },
    {
      "epoch": 8.732795715332031e-06,
      "model_forward_time": 5.579138994216919,
      "step": 9157
    },
    {
      "epoch": 8.732795715332031e-06,
      "step": 9157,
      "training_step_time": 17.009583950042725
    },
    {
      "epoch": 8.733749389648437e-06,
      "model_forward_time": 5.580816745758057,
      "step": 9158
    },
    {
      "epoch": 8.733749389648437e-06,
      "step": 9158,
      "training_step_time": 16.999927043914795
    },
    {
      "epoch": 8.734703063964844e-06,
      "model_forward_time": 5.965816259384155,
      "step": 9159
    },
    {
      "epoch": 8.734703063964844e-06,
      "step": 9159,
      "training_step_time": 17.430490016937256
    },
    {
      "epoch": 8.73565673828125e-06,
      "grad_norm": 0.41191771626472473,
      "learning_rate": 1e-05,
      "loss": 0.0661,
      "step": 9160
    },
    {
      "epoch": 8.73565673828125e-06,
      "model_forward_time": 5.6017725467681885,
      "step": 9160
    },
    {
      "dynamics_loss_avg": 0.05862775146961212,
      "epoch": 8.73565673828125e-06,
      "step": 9160
    },
    {
      "action_loss_avg": 0.006656686263158918,
      "epoch": 8.73565673828125e-06,
      "step": 9160
    },
    {
      "epoch": 8.73565673828125e-06,
      "step": 9160,
      "training_step_time": 17.08956789970398
    },
    {
      "epoch": 8.736610412597656e-06,
      "model_forward_time": 5.630559206008911,
      "step": 9161
    },
    {
      "epoch": 8.736610412597656e-06,
      "step": 9161,
      "training_step_time": 17.159185886383057
    },
    {
      "epoch": 8.737564086914063e-06,
      "model_forward_time": 5.623995065689087,
      "step": 9162
    },
    {
      "epoch": 8.737564086914063e-06,
      "step": 9162,
      "training_step_time": 17.100030183792114
    },
    {
      "epoch": 8.738517761230469e-06,
      "model_forward_time": 5.589054822921753,
      "step": 9163
    },
    {
      "epoch": 8.738517761230469e-06,
      "step": 9163,
      "training_step_time": 17.009233951568604
    },
    {
      "epoch": 8.739471435546875e-06,
      "model_forward_time": 5.569311618804932,
      "step": 9164
    },
    {
      "epoch": 8.739471435546875e-06,
      "step": 9164,
      "training_step_time": 17.01042628288269
    },
    {
      "epoch": 8.74042510986328e-06,
      "model_forward_time": 5.938558340072632,
      "step": 9165
    },
    {
      "epoch": 8.74042510986328e-06,
      "step": 9165,
      "training_step_time": 17.358537673950195
    },
    {
      "epoch": 8.741378784179688e-06,
      "model_forward_time": 5.598970413208008,
      "step": 9166
    },
    {
      "epoch": 8.741378784179688e-06,
      "step": 9166,
      "training_step_time": 17.035122871398926
    },
    {
      "epoch": 8.742332458496094e-06,
      "model_forward_time": 5.582111358642578,
      "step": 9167
    },
    {
      "epoch": 8.742332458496094e-06,
      "step": 9167,
      "training_step_time": 17.01339554786682
    },
    {
      "epoch": 8.7432861328125e-06,
      "model_forward_time": 5.602900266647339,
      "step": 9168
    },
    {
      "epoch": 8.7432861328125e-06,
      "step": 9168,
      "training_step_time": 17.033127546310425
    },
    {
      "epoch": 8.744239807128907e-06,
      "model_forward_time": 5.580101490020752,
      "step": 9169
    },
    {
      "epoch": 8.744239807128907e-06,
      "step": 9169,
      "training_step_time": 17.000689029693604
    },
    {
      "epoch": 8.745193481445312e-06,
      "grad_norm": 1.1443257331848145,
      "learning_rate": 1e-05,
      "loss": 0.0654,
      "step": 9170
    },
    {
      "epoch": 8.745193481445312e-06,
      "model_forward_time": 5.967897891998291,
      "step": 9170
    },
    {
      "dynamics_loss_avg": 0.06090954411774874,
      "epoch": 8.745193481445312e-06,
      "step": 9170
    },
    {
      "action_loss_avg": 0.009442709619179368,
      "epoch": 8.745193481445312e-06,
      "step": 9170
    },
    {
      "epoch": 8.745193481445312e-06,
      "step": 9170,
      "training_step_time": 17.41664147377014
    },
    {
      "epoch": 8.746147155761718e-06,
      "model_forward_time": 5.59031867980957,
      "step": 9171
    },
    {
      "epoch": 8.746147155761718e-06,
      "step": 9171,
      "training_step_time": 17.067911386489868
    },
    {
      "epoch": 8.747100830078126e-06,
      "model_forward_time": 5.6126954555511475,
      "step": 9172
    },
    {
      "epoch": 8.747100830078126e-06,
      "step": 9172,
      "training_step_time": 17.104249954223633
    },
    {
      "epoch": 8.748054504394531e-06,
      "model_forward_time": 5.621784925460815,
      "step": 9173
    },
    {
      "epoch": 8.748054504394531e-06,
      "step": 9173,
      "training_step_time": 17.103737592697144
    },
    {
      "epoch": 8.749008178710937e-06,
      "model_forward_time": 5.588278770446777,
      "step": 9174
    },
    {
      "epoch": 8.749008178710937e-06,
      "step": 9174,
      "training_step_time": 17.039911031723022
    },
    {
      "epoch": 8.749961853027344e-06,
      "model_forward_time": 5.574474096298218,
      "step": 9175
    },
    {
      "epoch": 8.749961853027344e-06,
      "step": 9175,
      "training_step_time": 17.011335134506226
    },
    {
      "epoch": 8.75091552734375e-06,
      "model_forward_time": 5.9146342277526855,
      "step": 9176
    },
    {
      "epoch": 8.75091552734375e-06,
      "step": 9176,
      "training_step_time": 17.329466581344604
    },
    {
      "epoch": 8.751869201660156e-06,
      "model_forward_time": 5.567975997924805,
      "step": 9177
    },
    {
      "epoch": 8.751869201660156e-06,
      "step": 9177,
      "training_step_time": 16.973738431930542
    },
    {
      "epoch": 8.752822875976563e-06,
      "model_forward_time": 5.57776403427124,
      "step": 9178
    },
    {
      "epoch": 8.752822875976563e-06,
      "step": 9178,
      "training_step_time": 16.9897940158844
    },
    {
      "epoch": 8.753776550292969e-06,
      "model_forward_time": 5.581939220428467,
      "step": 9179
    },
    {
      "epoch": 8.753776550292969e-06,
      "step": 9179,
      "training_step_time": 17.037801504135132
    },
    {
      "epoch": 8.754730224609375e-06,
      "grad_norm": 0.9759865999221802,
      "learning_rate": 1e-05,
      "loss": 0.0742,
      "step": 9180
    },
    {
      "epoch": 8.754730224609375e-06,
      "model_forward_time": 5.618969917297363,
      "step": 9180
    },
    {
      "dynamics_loss_avg": 0.062352574430406094,
      "epoch": 8.754730224609375e-06,
      "step": 9180
    },
    {
      "action_loss_avg": 0.009453334379941225,
      "epoch": 8.754730224609375e-06,
      "step": 9180
    },
    {
      "epoch": 8.754730224609375e-06,
      "step": 9180,
      "training_step_time": 17.105451583862305
    },
    {
      "epoch": 8.75568389892578e-06,
      "model_forward_time": 5.985959529876709,
      "step": 9181
    },
    {
      "epoch": 8.75568389892578e-06,
      "step": 9181,
      "training_step_time": 17.4682457447052
    },
    {
      "epoch": 8.756637573242188e-06,
      "model_forward_time": 5.603992700576782,
      "step": 9182
    },
    {
      "epoch": 8.756637573242188e-06,
      "step": 9182,
      "training_step_time": 17.078333139419556
    },
    {
      "epoch": 8.757591247558594e-06,
      "model_forward_time": 5.598362445831299,
      "step": 9183
    },
    {
      "epoch": 8.757591247558594e-06,
      "step": 9183,
      "training_step_time": 17.042421340942383
    },
    {
      "epoch": 8.758544921875e-06,
      "model_forward_time": 5.594165086746216,
      "step": 9184
    },
    {
      "epoch": 8.758544921875e-06,
      "step": 9184,
      "training_step_time": 17.001673698425293
    },
    {
      "epoch": 8.759498596191407e-06,
      "model_forward_time": 5.594034433364868,
      "step": 9185
    },
    {
      "epoch": 8.759498596191407e-06,
      "step": 9185,
      "training_step_time": 17.043450593948364
    },
    {
      "epoch": 8.760452270507812e-06,
      "model_forward_time": 5.949222564697266,
      "step": 9186
    },
    {
      "epoch": 8.760452270507812e-06,
      "step": 9186,
      "training_step_time": 17.407467365264893
    },
    {
      "epoch": 8.761405944824218e-06,
      "model_forward_time": 5.620802640914917,
      "step": 9187
    },
    {
      "epoch": 8.761405944824218e-06,
      "step": 9187,
      "training_step_time": 17.1070237159729
    },
    {
      "epoch": 8.762359619140626e-06,
      "model_forward_time": 5.616989612579346,
      "step": 9188
    },
    {
      "epoch": 8.762359619140626e-06,
      "step": 9188,
      "training_step_time": 17.10971212387085
    },
    {
      "epoch": 8.763313293457031e-06,
      "model_forward_time": 5.591888189315796,
      "step": 9189
    },
    {
      "epoch": 8.763313293457031e-06,
      "step": 9189,
      "training_step_time": 17.022225379943848
    },
    {
      "epoch": 8.764266967773437e-06,
      "grad_norm": 0.634699821472168,
      "learning_rate": 1e-05,
      "loss": 0.062,
      "step": 9190
    },
    {
      "epoch": 8.764266967773437e-06,
      "model_forward_time": 5.5782763957977295,
      "step": 9190
    },
    {
      "dynamics_loss_avg": 0.05168488025665283,
      "epoch": 8.764266967773437e-06,
      "step": 9190
    },
    {
      "action_loss_avg": 0.008845368400216103,
      "epoch": 8.764266967773437e-06,
      "step": 9190
    },
    {
      "epoch": 8.764266967773437e-06,
      "step": 9190,
      "training_step_time": 17.015663862228394
    },
    {
      "epoch": 8.765220642089844e-06,
      "model_forward_time": 5.584740877151489,
      "step": 9191
    },
    {
      "epoch": 8.765220642089844e-06,
      "step": 9191,
      "training_step_time": 17.017780780792236
    },
    {
      "epoch": 8.76617431640625e-06,
      "model_forward_time": 5.946662187576294,
      "step": 9192
    },
    {
      "epoch": 8.76617431640625e-06,
      "step": 9192,
      "training_step_time": 17.361225843429565
    },
    {
      "epoch": 8.767127990722656e-06,
      "model_forward_time": 5.587018966674805,
      "step": 9193
    },
    {
      "epoch": 8.767127990722656e-06,
      "step": 9193,
      "training_step_time": 17.051677703857422
    },
    {
      "epoch": 8.768081665039063e-06,
      "model_forward_time": 5.614243984222412,
      "step": 9194
    },
    {
      "epoch": 8.768081665039063e-06,
      "step": 9194,
      "training_step_time": 17.127254247665405
    },
    {
      "epoch": 8.769035339355469e-06,
      "model_forward_time": 5.630831718444824,
      "step": 9195
    },
    {
      "epoch": 8.769035339355469e-06,
      "step": 9195,
      "training_step_time": 17.14972424507141
    },
    {
      "epoch": 8.769989013671875e-06,
      "model_forward_time": 5.6166627407073975,
      "step": 9196
    },
    {
      "epoch": 8.769989013671875e-06,
      "step": 9196,
      "training_step_time": 17.1163547039032
    },
    {
      "epoch": 8.77094268798828e-06,
      "model_forward_time": 5.6058268547058105,
      "step": 9197
    },
    {
      "epoch": 8.77094268798828e-06,
      "step": 9197,
      "training_step_time": 17.453933477401733
    },
    {
      "epoch": 8.771896362304688e-06,
      "model_forward_time": 5.604854583740234,
      "step": 9198
    },
    {
      "epoch": 8.771896362304688e-06,
      "step": 9198,
      "training_step_time": 17.083770990371704
    },
    {
      "epoch": 8.772850036621094e-06,
      "model_forward_time": 5.613855361938477,
      "step": 9199
    },
    {
      "epoch": 8.772850036621094e-06,
      "step": 9199,
      "training_step_time": 17.102224111557007
    },
    {
      "epoch": 8.7738037109375e-06,
      "grad_norm": 0.5946109890937805,
      "learning_rate": 1e-05,
      "loss": 0.0735,
      "step": 9200
    },
    {
      "epoch": 8.7738037109375e-06,
      "model_forward_time": 5.614020109176636,
      "step": 9200
    },
    {
      "dynamics_loss_avg": 0.06366577744483948,
      "epoch": 8.7738037109375e-06,
      "step": 9200
    },
    {
      "action_loss_avg": 0.008511405391618609,
      "epoch": 8.7738037109375e-06,
      "step": 9200
    },
    {
      "epoch": 8.7738037109375e-06,
      "step": 9200,
      "training_step_time": 17.115275621414185
    },
    {
      "epoch": 8.774757385253907e-06,
      "model_forward_time": 5.615535497665405,
      "step": 9201
    },
    {
      "epoch": 8.774757385253907e-06,
      "step": 9201,
      "training_step_time": 17.117708683013916
    },
    {
      "epoch": 8.775711059570312e-06,
      "model_forward_time": 5.607733488082886,
      "step": 9202
    },
    {
      "epoch": 8.775711059570312e-06,
      "step": 9202,
      "training_step_time": 17.10679316520691
    },
    {
      "epoch": 8.776664733886718e-06,
      "model_forward_time": 6.003534555435181,
      "step": 9203
    },
    {
      "epoch": 8.776664733886718e-06,
      "step": 9203,
      "training_step_time": 17.495657920837402
    },
    {
      "epoch": 8.777618408203126e-06,
      "model_forward_time": 5.598843097686768,
      "step": 9204
    },
    {
      "epoch": 8.777618408203126e-06,
      "step": 9204,
      "training_step_time": 17.081949472427368
    },
    {
      "epoch": 8.778572082519531e-06,
      "model_forward_time": 5.607183456420898,
      "step": 9205
    },
    {
      "epoch": 8.778572082519531e-06,
      "step": 9205,
      "training_step_time": 17.093974351882935
    },
    {
      "epoch": 8.779525756835937e-06,
      "model_forward_time": 5.613885402679443,
      "step": 9206
    },
    {
      "epoch": 8.779525756835937e-06,
      "step": 9206,
      "training_step_time": 17.100183248519897
    },
    {
      "epoch": 8.780479431152344e-06,
      "model_forward_time": 5.616798162460327,
      "step": 9207
    },
    {
      "epoch": 8.780479431152344e-06,
      "step": 9207,
      "training_step_time": 17.11007571220398
    },
    {
      "epoch": 8.78143310546875e-06,
      "model_forward_time": 5.600884199142456,
      "step": 9208
    },
    {
      "epoch": 8.78143310546875e-06,
      "step": 9208,
      "training_step_time": 17.430793523788452
    },
    {
      "epoch": 8.782386779785156e-06,
      "model_forward_time": 5.595920562744141,
      "step": 9209
    },
    {
      "epoch": 8.782386779785156e-06,
      "step": 9209,
      "training_step_time": 17.047858715057373
    },
    {
      "epoch": 8.783340454101563e-06,
      "grad_norm": 0.9364034533500671,
      "learning_rate": 1e-05,
      "loss": 0.0724,
      "step": 9210
    },
    {
      "epoch": 8.783340454101563e-06,
      "model_forward_time": 5.580235481262207,
      "step": 9210
    },
    {
      "dynamics_loss_avg": 0.06510225292295217,
      "epoch": 8.783340454101563e-06,
      "step": 9210
    },
    {
      "action_loss_avg": 0.009301608544774354,
      "epoch": 8.783340454101563e-06,
      "step": 9210
    },
    {
      "epoch": 8.783340454101563e-06,
      "step": 9210,
      "training_step_time": 17.01921820640564
    },
    {
      "epoch": 8.784294128417969e-06,
      "model_forward_time": 5.584960460662842,
      "step": 9211
    },
    {
      "epoch": 8.784294128417969e-06,
      "step": 9211,
      "training_step_time": 17.009146451950073
    },
    {
      "epoch": 8.785247802734375e-06,
      "model_forward_time": 5.588097095489502,
      "step": 9212
    },
    {
      "epoch": 8.785247802734375e-06,
      "step": 9212,
      "training_step_time": 17.03235697746277
    },
    {
      "epoch": 8.78620147705078e-06,
      "model_forward_time": 5.600471019744873,
      "step": 9213
    },
    {
      "epoch": 8.78620147705078e-06,
      "step": 9213,
      "training_step_time": 17.08246088027954
    },
    {
      "epoch": 8.787155151367188e-06,
      "model_forward_time": 6.01904821395874,
      "step": 9214
    },
    {
      "epoch": 8.787155151367188e-06,
      "step": 9214,
      "training_step_time": 17.493186235427856
    },
    {
      "epoch": 8.788108825683594e-06,
      "model_forward_time": 5.646559953689575,
      "step": 9215
    },
    {
      "epoch": 8.788108825683594e-06,
      "step": 9215,
      "training_step_time": 17.180391788482666
    },
    {
      "epoch": 8.7890625e-06,
      "model_forward_time": 5.6177942752838135,
      "step": 9216
    },
    {
      "epoch": 8.7890625e-06,
      "step": 9216,
      "training_step_time": 17.10940718650818
    },
    {
      "epoch": 8.790016174316407e-06,
      "model_forward_time": 5.6099395751953125,
      "step": 9217
    },
    {
      "epoch": 8.790016174316407e-06,
      "step": 9217,
      "training_step_time": 17.092368364334106
    },
    {
      "epoch": 8.790969848632812e-06,
      "model_forward_time": 5.596842050552368,
      "step": 9218
    },
    {
      "epoch": 8.790969848632812e-06,
      "step": 9218,
      "training_step_time": 17.05541706085205
    },
    {
      "epoch": 8.791923522949218e-06,
      "model_forward_time": 5.963629245758057,
      "step": 9219
    },
    {
      "epoch": 8.791923522949218e-06,
      "step": 9219,
      "training_step_time": 17.426044702529907
    },
    {
      "epoch": 8.792877197265626e-06,
      "grad_norm": 0.7881561517715454,
      "learning_rate": 1e-05,
      "loss": 0.0694,
      "step": 9220
    },
    {
      "epoch": 8.792877197265626e-06,
      "model_forward_time": 5.612149477005005,
      "step": 9220
    },
    {
      "dynamics_loss_avg": 0.06168064083904028,
      "epoch": 8.792877197265626e-06,
      "step": 9220
    },
    {
      "action_loss_avg": 0.010751136392354966,
      "epoch": 8.792877197265626e-06,
      "step": 9220
    },
    {
      "epoch": 8.792877197265626e-06,
      "step": 9220,
      "training_step_time": 17.118728637695312
    },
    {
      "epoch": 8.793830871582031e-06,
      "model_forward_time": 5.632924318313599,
      "step": 9221
    },
    {
      "epoch": 8.793830871582031e-06,
      "step": 9221,
      "training_step_time": 17.143741607666016
    },
    {
      "epoch": 8.794784545898437e-06,
      "model_forward_time": 5.623580455780029,
      "step": 9222
    },
    {
      "epoch": 8.794784545898437e-06,
      "step": 9222,
      "training_step_time": 17.131548643112183
    },
    {
      "epoch": 8.795738220214844e-06,
      "model_forward_time": 5.621565818786621,
      "step": 9223
    },
    {
      "epoch": 8.795738220214844e-06,
      "step": 9223,
      "training_step_time": 17.121009588241577
    },
    {
      "epoch": 8.79669189453125e-06,
      "model_forward_time": 5.615569353103638,
      "step": 9224
    },
    {
      "epoch": 8.79669189453125e-06,
      "step": 9224,
      "training_step_time": 17.121830701828003
    },
    {
      "epoch": 8.797645568847656e-06,
      "model_forward_time": 6.000355005264282,
      "step": 9225
    },
    {
      "epoch": 8.797645568847656e-06,
      "step": 9225,
      "training_step_time": 17.487083196640015
    },
    {
      "epoch": 8.798599243164063e-06,
      "model_forward_time": 5.616842031478882,
      "step": 9226
    },
    {
      "epoch": 8.798599243164063e-06,
      "step": 9226,
      "training_step_time": 17.113685369491577
    },
    {
      "epoch": 8.799552917480469e-06,
      "model_forward_time": 5.615771055221558,
      "step": 9227
    },
    {
      "epoch": 8.799552917480469e-06,
      "step": 9227,
      "training_step_time": 17.11443567276001
    },
    {
      "epoch": 8.800506591796875e-06,
      "model_forward_time": 5.6110944747924805,
      "step": 9228
    },
    {
      "epoch": 8.800506591796875e-06,
      "step": 9228,
      "training_step_time": 17.117643356323242
    },
    {
      "epoch": 8.80146026611328e-06,
      "model_forward_time": 5.616419553756714,
      "step": 9229
    },
    {
      "epoch": 8.80146026611328e-06,
      "step": 9229,
      "training_step_time": 17.080576419830322
    },
    {
      "epoch": 8.802413940429688e-06,
      "grad_norm": 0.34946903586387634,
      "learning_rate": 1e-05,
      "loss": 0.0635,
      "step": 9230
    },
    {
      "epoch": 8.802413940429688e-06,
      "model_forward_time": 5.591784954071045,
      "step": 9230
    },
    {
      "dynamics_loss_avg": 0.05480791032314301,
      "epoch": 8.802413940429688e-06,
      "step": 9230
    },
    {
      "action_loss_avg": 0.0072995533235371115,
      "epoch": 8.802413940429688e-06,
      "step": 9230
    },
    {
      "epoch": 8.802413940429688e-06,
      "step": 9230,
      "training_step_time": 17.391199350357056
    },
    {
      "epoch": 8.803367614746094e-06,
      "model_forward_time": 5.588275194168091,
      "step": 9231
    },
    {
      "epoch": 8.803367614746094e-06,
      "step": 9231,
      "training_step_time": 17.030297994613647
    },
    {
      "epoch": 8.8043212890625e-06,
      "model_forward_time": 5.600977420806885,
      "step": 9232
    },
    {
      "epoch": 8.8043212890625e-06,
      "step": 9232,
      "training_step_time": 17.023004293441772
    },
    {
      "epoch": 8.805274963378907e-06,
      "model_forward_time": 5.591064214706421,
      "step": 9233
    },
    {
      "epoch": 8.805274963378907e-06,
      "step": 9233,
      "training_step_time": 17.055758476257324
    },
    {
      "epoch": 8.806228637695312e-06,
      "model_forward_time": 5.595832586288452,
      "step": 9234
    },
    {
      "epoch": 8.806228637695312e-06,
      "step": 9234,
      "training_step_time": 17.070027112960815
    },
    {
      "epoch": 8.807182312011718e-06,
      "model_forward_time": 5.60884428024292,
      "step": 9235
    },
    {
      "epoch": 8.807182312011718e-06,
      "step": 9235,
      "training_step_time": 17.104973793029785
    },
    {
      "epoch": 8.808135986328126e-06,
      "model_forward_time": 5.9817893505096436,
      "step": 9236
    },
    {
      "epoch": 8.808135986328126e-06,
      "step": 9236,
      "training_step_time": 17.459308385849
    },
    {
      "epoch": 8.809089660644531e-06,
      "model_forward_time": 5.604674816131592,
      "step": 9237
    },
    {
      "epoch": 8.809089660644531e-06,
      "step": 9237,
      "training_step_time": 17.078006982803345
    },
    {
      "epoch": 8.810043334960937e-06,
      "model_forward_time": 5.605442523956299,
      "step": 9238
    },
    {
      "epoch": 8.810043334960937e-06,
      "step": 9238,
      "training_step_time": 17.0599102973938
    },
    {
      "epoch": 8.810997009277344e-06,
      "model_forward_time": 5.590860366821289,
      "step": 9239
    },
    {
      "epoch": 8.810997009277344e-06,
      "step": 9239,
      "training_step_time": 17.037962436676025
    },
    {
      "epoch": 8.81195068359375e-06,
      "grad_norm": 0.8424144983291626,
      "learning_rate": 1e-05,
      "loss": 0.0796,
      "step": 9240
    },
    {
      "epoch": 8.81195068359375e-06,
      "model_forward_time": 5.613036870956421,
      "step": 9240
    },
    {
      "dynamics_loss_avg": 0.0661617636680603,
      "epoch": 8.81195068359375e-06,
      "step": 9240
    },
    {
      "action_loss_avg": 0.010902693076059223,
      "epoch": 8.81195068359375e-06,
      "step": 9240
    },
    {
      "epoch": 8.81195068359375e-06,
      "step": 9240,
      "training_step_time": 17.104310989379883
    },
    {
      "epoch": 8.812904357910156e-06,
      "model_forward_time": 5.615920305252075,
      "step": 9241
    },
    {
      "epoch": 8.812904357910156e-06,
      "step": 9241,
      "training_step_time": 17.48473620414734
    },
    {
      "epoch": 8.813858032226563e-06,
      "model_forward_time": 5.614926815032959,
      "step": 9242
    },
    {
      "epoch": 8.813858032226563e-06,
      "step": 9242,
      "training_step_time": 17.1146342754364
    },
    {
      "epoch": 8.814811706542969e-06,
      "model_forward_time": 5.615579605102539,
      "step": 9243
    },
    {
      "epoch": 8.814811706542969e-06,
      "step": 9243,
      "training_step_time": 17.08078408241272
    },
    {
      "epoch": 8.815765380859375e-06,
      "model_forward_time": 5.594500541687012,
      "step": 9244
    },
    {
      "epoch": 8.815765380859375e-06,
      "step": 9244,
      "training_step_time": 17.041969299316406
    },
    {
      "epoch": 8.81671905517578e-06,
      "model_forward_time": 5.59329628944397,
      "step": 9245
    },
    {
      "epoch": 8.81671905517578e-06,
      "step": 9245,
      "training_step_time": 17.027958869934082
    },
    {
      "epoch": 8.817672729492188e-06,
      "model_forward_time": 5.595472574234009,
      "step": 9246
    },
    {
      "epoch": 8.817672729492188e-06,
      "step": 9246,
      "training_step_time": 17.061893939971924
    },
    {
      "epoch": 8.818626403808594e-06,
      "model_forward_time": 5.968188524246216,
      "step": 9247
    },
    {
      "epoch": 8.818626403808594e-06,
      "step": 9247,
      "training_step_time": 17.436204433441162
    },
    {
      "epoch": 8.819580078125e-06,
      "model_forward_time": 5.595887660980225,
      "step": 9248
    },
    {
      "epoch": 8.819580078125e-06,
      "step": 9248,
      "training_step_time": 17.088507890701294
    },
    {
      "epoch": 8.820533752441407e-06,
      "model_forward_time": 5.606845378875732,
      "step": 9249
    },
    {
      "epoch": 8.820533752441407e-06,
      "step": 9249,
      "training_step_time": 17.097350120544434
    },
    {
      "epoch": 8.821487426757812e-06,
      "grad_norm": 1.17497718334198,
      "learning_rate": 1e-05,
      "loss": 0.0695,
      "step": 9250
    },
    {
      "epoch": 8.821487426757812e-06,
      "model_forward_time": 5.611337423324585,
      "step": 9250
    },
    {
      "dynamics_loss_avg": 0.060285691544413565,
      "epoch": 8.821487426757812e-06,
      "step": 9250
    },
    {
      "action_loss_avg": 0.009384659910574555,
      "epoch": 8.821487426757812e-06,
      "step": 9250
    },
    {
      "epoch": 8.821487426757812e-06,
      "step": 9250,
      "training_step_time": 17.097984313964844
    },
    {
      "epoch": 8.822441101074218e-06,
      "model_forward_time": 5.597073078155518,
      "step": 9251
    },
    {
      "epoch": 8.822441101074218e-06,
      "step": 9251,
      "training_step_time": 17.0642032623291
    },
    {
      "epoch": 8.823394775390626e-06,
      "model_forward_time": 5.966915845870972,
      "step": 9252
    },
    {
      "epoch": 8.823394775390626e-06,
      "step": 9252,
      "training_step_time": 17.43673872947693
    },
    {
      "epoch": 8.824348449707031e-06,
      "model_forward_time": 5.596762180328369,
      "step": 9253
    },
    {
      "epoch": 8.824348449707031e-06,
      "step": 9253,
      "training_step_time": 17.047895431518555
    },
    {
      "epoch": 8.825302124023437e-06,
      "model_forward_time": 5.5918920040130615,
      "step": 9254
    },
    {
      "epoch": 8.825302124023437e-06,
      "step": 9254,
      "training_step_time": 17.037648916244507
    },
    {
      "epoch": 8.826255798339844e-06,
      "model_forward_time": 5.6045451164245605,
      "step": 9255
    },
    {
      "epoch": 8.826255798339844e-06,
      "step": 9255,
      "training_step_time": 17.05948281288147
    },
    {
      "epoch": 8.82720947265625e-06,
      "model_forward_time": 5.598447561264038,
      "step": 9256
    },
    {
      "epoch": 8.82720947265625e-06,
      "step": 9256,
      "training_step_time": 17.096986293792725
    },
    {
      "epoch": 8.828163146972656e-06,
      "model_forward_time": 4.285128355026245,
      "step": 9257
    },
    {
      "epoch": 8.828163146972656e-06,
      "step": 9257,
      "training_step_time": 12.840967178344727
    },
    {
      "epoch": 8.829116821289062e-06,
      "model_forward_time": 6.000701189041138,
      "step": 9258
    },
    {
      "epoch": 8.829116821289062e-06,
      "step": 9258,
      "training_step_time": 17.50325894355774
    },
    {
      "epoch": 8.830070495605469e-06,
      "model_forward_time": 5.602964162826538,
      "step": 9259
    },
    {
      "epoch": 8.830070495605469e-06,
      "step": 9259,
      "training_step_time": 17.098925828933716
    },
    {
      "epoch": 8.831024169921875e-06,
      "grad_norm": 1.1689225435256958,
      "learning_rate": 1e-05,
      "loss": 0.0705,
      "step": 9260
    },
    {
      "epoch": 8.831024169921875e-06,
      "model_forward_time": 5.618263483047485,
      "step": 9260
    },
    {
      "dynamics_loss_avg": 0.06030820850282907,
      "epoch": 8.831024169921875e-06,
      "step": 9260
    },
    {
      "action_loss_avg": 0.008454929618164898,
      "epoch": 8.831024169921875e-06,
      "step": 9260
    },
    {
      "epoch": 8.831024169921875e-06,
      "step": 9260,
      "training_step_time": 17.112203121185303
    },
    {
      "epoch": 8.83197784423828e-06,
      "model_forward_time": 5.615000009536743,
      "step": 9261
    },
    {
      "epoch": 8.83197784423828e-06,
      "step": 9261,
      "training_step_time": 17.1016104221344
    },
    {
      "epoch": 8.832931518554688e-06,
      "model_forward_time": 5.602432727813721,
      "step": 9262
    },
    {
      "epoch": 8.832931518554688e-06,
      "step": 9262,
      "training_step_time": 17.083195209503174
    },
    {
      "epoch": 8.833885192871094e-06,
      "model_forward_time": 5.60706090927124,
      "step": 9263
    },
    {
      "epoch": 8.833885192871094e-06,
      "step": 9263,
      "training_step_time": 17.465330600738525
    },
    {
      "epoch": 8.8348388671875e-06,
      "model_forward_time": 5.610846281051636,
      "step": 9264
    },
    {
      "epoch": 8.8348388671875e-06,
      "step": 9264,
      "training_step_time": 17.102108001708984
    },
    {
      "epoch": 8.835792541503907e-06,
      "model_forward_time": 5.611666679382324,
      "step": 9265
    },
    {
      "epoch": 8.835792541503907e-06,
      "step": 9265,
      "training_step_time": 17.09168553352356
    },
    {
      "epoch": 8.836746215820313e-06,
      "model_forward_time": 5.603393077850342,
      "step": 9266
    },
    {
      "epoch": 8.836746215820313e-06,
      "step": 9266,
      "training_step_time": 17.036607027053833
    },
    {
      "epoch": 8.837699890136718e-06,
      "model_forward_time": 5.571500539779663,
      "step": 9267
    },
    {
      "epoch": 8.837699890136718e-06,
      "step": 9267,
      "training_step_time": 17.005534410476685
    },
    {
      "epoch": 8.838653564453126e-06,
      "model_forward_time": 5.576897382736206,
      "step": 9268
    },
    {
      "epoch": 8.838653564453126e-06,
      "step": 9268,
      "training_step_time": 16.999818563461304
    },
    {
      "epoch": 8.839607238769531e-06,
      "model_forward_time": 5.95646595954895,
      "step": 9269
    },
    {
      "epoch": 8.839607238769531e-06,
      "step": 9269,
      "training_step_time": 17.356093883514404
    },
    {
      "epoch": 8.840560913085937e-06,
      "grad_norm": 0.19458109140396118,
      "learning_rate": 1e-05,
      "loss": 0.0668,
      "step": 9270
    },
    {
      "epoch": 8.840560913085937e-06,
      "model_forward_time": 5.591980934143066,
      "step": 9270
    },
    {
      "dynamics_loss_avg": 0.057243352942168715,
      "epoch": 8.840560913085937e-06,
      "step": 9270
    },
    {
      "action_loss_avg": 0.00864528063684702,
      "epoch": 8.840560913085937e-06,
      "step": 9270
    },
    {
      "epoch": 8.840560913085937e-06,
      "step": 9270,
      "training_step_time": 17.031378984451294
    },
    {
      "epoch": 8.841514587402345e-06,
      "model_forward_time": 5.616041898727417,
      "step": 9271
    },
    {
      "epoch": 8.841514587402345e-06,
      "step": 9271,
      "training_step_time": 17.092169284820557
    },
    {
      "epoch": 8.84246826171875e-06,
      "model_forward_time": 5.603968858718872,
      "step": 9272
    },
    {
      "epoch": 8.84246826171875e-06,
      "step": 9272,
      "training_step_time": 17.09263515472412
    },
    {
      "epoch": 8.843421936035156e-06,
      "model_forward_time": 5.6491639614105225,
      "step": 9273
    },
    {
      "epoch": 8.843421936035156e-06,
      "step": 9273,
      "training_step_time": 17.159241676330566
    },
    {
      "epoch": 8.844375610351562e-06,
      "model_forward_time": 5.618295907974243,
      "step": 9274
    },
    {
      "epoch": 8.844375610351562e-06,
      "step": 9274,
      "training_step_time": 17.474605560302734
    },
    {
      "epoch": 8.845329284667969e-06,
      "model_forward_time": 5.618297576904297,
      "step": 9275
    },
    {
      "epoch": 8.845329284667969e-06,
      "step": 9275,
      "training_step_time": 17.094200372695923
    },
    {
      "epoch": 8.846282958984375e-06,
      "model_forward_time": 5.611831188201904,
      "step": 9276
    },
    {
      "epoch": 8.846282958984375e-06,
      "step": 9276,
      "training_step_time": 17.08323907852173
    },
    {
      "epoch": 8.84723663330078e-06,
      "model_forward_time": 5.617421865463257,
      "step": 9277
    },
    {
      "epoch": 8.84723663330078e-06,
      "step": 9277,
      "training_step_time": 17.07540988922119
    },
    {
      "epoch": 8.848190307617188e-06,
      "model_forward_time": 5.618941783905029,
      "step": 9278
    },
    {
      "epoch": 8.848190307617188e-06,
      "step": 9278,
      "training_step_time": 17.09553813934326
    },
    {
      "epoch": 8.849143981933594e-06,
      "model_forward_time": 5.610358238220215,
      "step": 9279
    },
    {
      "epoch": 8.849143981933594e-06,
      "step": 9279,
      "training_step_time": 17.081742763519287
    },
    {
      "epoch": 8.85009765625e-06,
      "grad_norm": 1.2393584251403809,
      "learning_rate": 1e-05,
      "loss": 0.0691,
      "step": 9280
    },
    {
      "epoch": 8.85009765625e-06,
      "model_forward_time": 6.0028235912323,
      "step": 9280
    },
    {
      "dynamics_loss_avg": 0.05995445903390646,
      "epoch": 8.85009765625e-06,
      "step": 9280
    },
    {
      "action_loss_avg": 0.008704461087472738,
      "epoch": 8.85009765625e-06,
      "step": 9280
    },
    {
      "epoch": 8.85009765625e-06,
      "step": 9280,
      "training_step_time": 17.463295698165894
    },
    {
      "epoch": 8.851051330566407e-06,
      "model_forward_time": 5.606568336486816,
      "step": 9281
    },
    {
      "epoch": 8.851051330566407e-06,
      "step": 9281,
      "training_step_time": 17.07662343978882
    },
    {
      "epoch": 8.852005004882813e-06,
      "model_forward_time": 5.5938074588775635,
      "step": 9282
    },
    {
      "epoch": 8.852005004882813e-06,
      "step": 9282,
      "training_step_time": 17.069262504577637
    },
    {
      "epoch": 8.852958679199218e-06,
      "model_forward_time": 5.595191240310669,
      "step": 9283
    },
    {
      "epoch": 8.852958679199218e-06,
      "step": 9283,
      "training_step_time": 17.04831600189209
    },
    {
      "epoch": 8.853912353515626e-06,
      "model_forward_time": 5.5956878662109375,
      "step": 9284
    },
    {
      "epoch": 8.853912353515626e-06,
      "step": 9284,
      "training_step_time": 17.03521180152893
    },
    {
      "epoch": 8.854866027832031e-06,
      "model_forward_time": 5.955831289291382,
      "step": 9285
    },
    {
      "epoch": 8.854866027832031e-06,
      "step": 9285,
      "training_step_time": 17.39831781387329
    },
    {
      "epoch": 8.855819702148437e-06,
      "model_forward_time": 5.5987958908081055,
      "step": 9286
    },
    {
      "epoch": 8.855819702148437e-06,
      "step": 9286,
      "training_step_time": 17.06977105140686
    },
    {
      "epoch": 8.856773376464845e-06,
      "model_forward_time": 5.606514930725098,
      "step": 9287
    },
    {
      "epoch": 8.856773376464845e-06,
      "step": 9287,
      "training_step_time": 17.098782300949097
    },
    {
      "epoch": 8.85772705078125e-06,
      "model_forward_time": 5.619562864303589,
      "step": 9288
    },
    {
      "epoch": 8.85772705078125e-06,
      "step": 9288,
      "training_step_time": 17.105023860931396
    },
    {
      "epoch": 8.858680725097656e-06,
      "model_forward_time": 5.60158109664917,
      "step": 9289
    },
    {
      "epoch": 8.858680725097656e-06,
      "step": 9289,
      "training_step_time": 17.055338621139526
    },
    {
      "epoch": 8.859634399414062e-06,
      "grad_norm": 0.7493025064468384,
      "learning_rate": 1e-05,
      "loss": 0.0631,
      "step": 9290
    },
    {
      "epoch": 8.859634399414062e-06,
      "model_forward_time": 5.588080406188965,
      "step": 9290
    },
    {
      "dynamics_loss_avg": 0.06128973858430982,
      "epoch": 8.859634399414062e-06,
      "step": 9290
    },
    {
      "action_loss_avg": 0.009216731617925689,
      "epoch": 8.859634399414062e-06,
      "step": 9290
    },
    {
      "epoch": 8.859634399414062e-06,
      "step": 9290,
      "training_step_time": 17.03189492225647
    },
    {
      "epoch": 8.860588073730469e-06,
      "model_forward_time": 5.970849275588989,
      "step": 9291
    },
    {
      "epoch": 8.860588073730469e-06,
      "step": 9291,
      "training_step_time": 17.433398246765137
    },
    {
      "epoch": 8.861541748046875e-06,
      "model_forward_time": 5.612875461578369,
      "step": 9292
    },
    {
      "epoch": 8.861541748046875e-06,
      "step": 9292,
      "training_step_time": 17.118056774139404
    },
    {
      "epoch": 8.86249542236328e-06,
      "model_forward_time": 5.629058599472046,
      "step": 9293
    },
    {
      "epoch": 8.86249542236328e-06,
      "step": 9293,
      "training_step_time": 17.1335391998291
    },
    {
      "epoch": 8.863449096679688e-06,
      "model_forward_time": 5.604329586029053,
      "step": 9294
    },
    {
      "epoch": 8.863449096679688e-06,
      "step": 9294,
      "training_step_time": 17.07870578765869
    },
    {
      "epoch": 8.864402770996094e-06,
      "model_forward_time": 5.591087102890015,
      "step": 9295
    },
    {
      "epoch": 8.864402770996094e-06,
      "step": 9295,
      "training_step_time": 17.031164169311523
    },
    {
      "epoch": 8.8653564453125e-06,
      "model_forward_time": 5.5886595249176025,
      "step": 9296
    },
    {
      "epoch": 8.8653564453125e-06,
      "step": 9296,
      "training_step_time": 17.36656928062439
    },
    {
      "epoch": 8.866310119628907e-06,
      "model_forward_time": 5.582072019577026,
      "step": 9297
    },
    {
      "epoch": 8.866310119628907e-06,
      "step": 9297,
      "training_step_time": 17.02784776687622
    },
    {
      "epoch": 8.867263793945313e-06,
      "model_forward_time": 5.584904193878174,
      "step": 9298
    },
    {
      "epoch": 8.867263793945313e-06,
      "step": 9298,
      "training_step_time": 17.027729034423828
    },
    {
      "epoch": 8.868217468261718e-06,
      "model_forward_time": 5.601861238479614,
      "step": 9299
    },
    {
      "epoch": 8.868217468261718e-06,
      "step": 9299,
      "training_step_time": 17.07760453224182
    },
    {
      "epoch": 8.869171142578126e-06,
      "grad_norm": 0.5532964468002319,
      "learning_rate": 1e-05,
      "loss": 0.0732,
      "step": 9300
    },
    {
      "epoch": 8.869171142578126e-06,
      "model_forward_time": 5.609853982925415,
      "step": 9300
    },
    {
      "dynamics_loss_avg": 0.06097451690584421,
      "epoch": 8.869171142578126e-06,
      "step": 9300
    },
    {
      "action_loss_avg": 0.00833472479134798,
      "epoch": 8.869171142578126e-06,
      "step": 9300
    },
    {
      "epoch": 8.869171142578126e-06,
      "step": 9300,
      "training_step_time": 17.091082334518433
    },
    {
      "epoch": 8.870124816894531e-06,
      "model_forward_time": 5.606464147567749,
      "step": 9301
    },
    {
      "epoch": 8.870124816894531e-06,
      "step": 9301,
      "training_step_time": 17.046170711517334
    },
    {
      "epoch": 8.871078491210937e-06,
      "model_forward_time": 5.9768877029418945,
      "step": 9302
    },
    {
      "epoch": 8.871078491210937e-06,
      "step": 9302,
      "training_step_time": 17.457908868789673
    },
    {
      "epoch": 8.872032165527345e-06,
      "model_forward_time": 5.595094680786133,
      "step": 9303
    },
    {
      "epoch": 8.872032165527345e-06,
      "step": 9303,
      "training_step_time": 17.057021379470825
    },
    {
      "epoch": 8.87298583984375e-06,
      "model_forward_time": 5.595515251159668,
      "step": 9304
    },
    {
      "epoch": 8.87298583984375e-06,
      "step": 9304,
      "training_step_time": 17.06867218017578
    },
    {
      "epoch": 8.873939514160156e-06,
      "model_forward_time": 5.611883163452148,
      "step": 9305
    },
    {
      "epoch": 8.873939514160156e-06,
      "step": 9305,
      "training_step_time": 17.080106258392334
    },
    {
      "epoch": 8.874893188476562e-06,
      "model_forward_time": 5.600843667984009,
      "step": 9306
    },
    {
      "epoch": 8.874893188476562e-06,
      "step": 9306,
      "training_step_time": 17.056157112121582
    },
    {
      "epoch": 8.875846862792969e-06,
      "model_forward_time": 5.589716196060181,
      "step": 9307
    },
    {
      "epoch": 8.875846862792969e-06,
      "step": 9307,
      "training_step_time": 17.41671919822693
    },
    {
      "epoch": 8.876800537109375e-06,
      "model_forward_time": 5.603254318237305,
      "step": 9308
    },
    {
      "epoch": 8.876800537109375e-06,
      "step": 9308,
      "training_step_time": 17.090193510055542
    },
    {
      "epoch": 8.87775421142578e-06,
      "model_forward_time": 5.611048936843872,
      "step": 9309
    },
    {
      "epoch": 8.87775421142578e-06,
      "step": 9309,
      "training_step_time": 17.113725900650024
    },
    {
      "epoch": 8.878707885742188e-06,
      "grad_norm": 1.5096317529678345,
      "learning_rate": 1e-05,
      "loss": 0.0758,
      "step": 9310
    },
    {
      "epoch": 8.878707885742188e-06,
      "model_forward_time": 5.603015422821045,
      "step": 9310
    },
    {
      "dynamics_loss_avg": 0.06691163033246994,
      "epoch": 8.878707885742188e-06,
      "step": 9310
    },
    {
      "action_loss_avg": 0.011027282057330012,
      "epoch": 8.878707885742188e-06,
      "step": 9310
    },
    {
      "epoch": 8.878707885742188e-06,
      "step": 9310,
      "training_step_time": 17.09876537322998
    },
    {
      "epoch": 8.879661560058594e-06,
      "model_forward_time": 5.6248228549957275,
      "step": 9311
    },
    {
      "epoch": 8.879661560058594e-06,
      "step": 9311,
      "training_step_time": 17.088448762893677
    },
    {
      "epoch": 8.880615234375e-06,
      "model_forward_time": 5.5904271602630615,
      "step": 9312
    },
    {
      "epoch": 8.880615234375e-06,
      "step": 9312,
      "training_step_time": 17.027309894561768
    },
    {
      "epoch": 8.881568908691407e-06,
      "model_forward_time": 5.936818361282349,
      "step": 9313
    },
    {
      "epoch": 8.881568908691407e-06,
      "step": 9313,
      "training_step_time": 17.362651586532593
    },
    {
      "epoch": 8.882522583007813e-06,
      "model_forward_time": 5.566659927368164,
      "step": 9314
    },
    {
      "epoch": 8.882522583007813e-06,
      "step": 9314,
      "training_step_time": 16.980255842208862
    },
    {
      "epoch": 8.883476257324218e-06,
      "model_forward_time": 5.573523044586182,
      "step": 9315
    },
    {
      "epoch": 8.883476257324218e-06,
      "step": 9315,
      "training_step_time": 16.994680881500244
    },
    {
      "epoch": 8.884429931640626e-06,
      "model_forward_time": 5.5677690505981445,
      "step": 9316
    },
    {
      "epoch": 8.884429931640626e-06,
      "step": 9316,
      "training_step_time": 16.99947738647461
    },
    {
      "epoch": 8.885383605957031e-06,
      "model_forward_time": 5.577798843383789,
      "step": 9317
    },
    {
      "epoch": 8.885383605957031e-06,
      "step": 9317,
      "training_step_time": 16.99497413635254
    },
    {
      "epoch": 8.886337280273437e-06,
      "model_forward_time": 5.942614316940308,
      "step": 9318
    },
    {
      "epoch": 8.886337280273437e-06,
      "step": 9318,
      "training_step_time": 17.349598169326782
    },
    {
      "epoch": 8.887290954589845e-06,
      "model_forward_time": 5.58410120010376,
      "step": 9319
    },
    {
      "epoch": 8.887290954589845e-06,
      "step": 9319,
      "training_step_time": 17.02818489074707
    },
    {
      "epoch": 8.88824462890625e-06,
      "grad_norm": 0.6895461678504944,
      "learning_rate": 1e-05,
      "loss": 0.0914,
      "step": 9320
    },
    {
      "epoch": 8.88824462890625e-06,
      "model_forward_time": 5.605890274047852,
      "step": 9320
    },
    {
      "dynamics_loss_avg": 0.07299504783004522,
      "epoch": 8.88824462890625e-06,
      "step": 9320
    },
    {
      "action_loss_avg": 0.012334174802526831,
      "epoch": 8.88824462890625e-06,
      "step": 9320
    },
    {
      "epoch": 8.88824462890625e-06,
      "step": 9320,
      "training_step_time": 17.091002941131592
    },
    {
      "epoch": 8.889198303222656e-06,
      "model_forward_time": 5.609422206878662,
      "step": 9321
    },
    {
      "epoch": 8.889198303222656e-06,
      "step": 9321,
      "training_step_time": 17.077486753463745
    },
    {
      "epoch": 8.890151977539062e-06,
      "model_forward_time": 5.597286701202393,
      "step": 9322
    },
    {
      "epoch": 8.890151977539062e-06,
      "step": 9322,
      "training_step_time": 17.087612628936768
    },
    {
      "epoch": 8.891105651855469e-06,
      "model_forward_time": 5.619234561920166,
      "step": 9323
    },
    {
      "epoch": 8.891105651855469e-06,
      "step": 9323,
      "training_step_time": 17.133537769317627
    },
    {
      "epoch": 8.892059326171875e-06,
      "model_forward_time": 5.998092174530029,
      "step": 9324
    },
    {
      "epoch": 8.892059326171875e-06,
      "step": 9324,
      "training_step_time": 17.51026749610901
    },
    {
      "epoch": 8.89301300048828e-06,
      "model_forward_time": 5.615014553070068,
      "step": 9325
    },
    {
      "epoch": 8.89301300048828e-06,
      "step": 9325,
      "training_step_time": 17.127336502075195
    },
    {
      "epoch": 8.893966674804688e-06,
      "model_forward_time": 5.632474184036255,
      "step": 9326
    },
    {
      "epoch": 8.893966674804688e-06,
      "step": 9326,
      "training_step_time": 17.163784742355347
    },
    {
      "epoch": 8.894920349121094e-06,
      "model_forward_time": 5.613587856292725,
      "step": 9327
    },
    {
      "epoch": 8.894920349121094e-06,
      "step": 9327,
      "training_step_time": 17.092051029205322
    },
    {
      "epoch": 8.8958740234375e-06,
      "model_forward_time": 5.577996015548706,
      "step": 9328
    },
    {
      "epoch": 8.8958740234375e-06,
      "step": 9328,
      "training_step_time": 17.02144193649292
    },
    {
      "epoch": 8.896827697753907e-06,
      "model_forward_time": 5.56658673286438,
      "step": 9329
    },
    {
      "epoch": 8.896827697753907e-06,
      "step": 9329,
      "training_step_time": 17.33644962310791
    },
    {
      "epoch": 8.897781372070313e-06,
      "grad_norm": 0.4652107357978821,
      "learning_rate": 1e-05,
      "loss": 0.0664,
      "step": 9330
    },
    {
      "epoch": 8.897781372070313e-06,
      "model_forward_time": 5.57067608833313,
      "step": 9330
    },
    {
      "dynamics_loss_avg": 0.06339203082025051,
      "epoch": 8.897781372070313e-06,
      "step": 9330
    },
    {
      "action_loss_avg": 0.00811849688179791,
      "epoch": 8.897781372070313e-06,
      "step": 9330
    },
    {
      "epoch": 8.897781372070313e-06,
      "step": 9330,
      "training_step_time": 16.98386788368225
    },
    {
      "epoch": 8.898735046386718e-06,
      "model_forward_time": 5.563364505767822,
      "step": 9331
    },
    {
      "epoch": 8.898735046386718e-06,
      "step": 9331,
      "training_step_time": 16.98981499671936
    },
    {
      "epoch": 8.899688720703126e-06,
      "model_forward_time": 5.552838563919067,
      "step": 9332
    },
    {
      "epoch": 8.899688720703126e-06,
      "step": 9332,
      "training_step_time": 16.981974363327026
    },
    {
      "epoch": 8.900642395019531e-06,
      "model_forward_time": 5.578667402267456,
      "step": 9333
    },
    {
      "epoch": 8.900642395019531e-06,
      "step": 9333,
      "training_step_time": 16.98921823501587
    },
    {
      "epoch": 8.901596069335937e-06,
      "model_forward_time": 5.579385042190552,
      "step": 9334
    },
    {
      "epoch": 8.901596069335937e-06,
      "step": 9334,
      "training_step_time": 16.987975120544434
    },
    {
      "epoch": 8.902549743652345e-06,
      "model_forward_time": 5.942164897918701,
      "step": 9335
    },
    {
      "epoch": 8.902549743652345e-06,
      "step": 9335,
      "training_step_time": 17.374688863754272
    },
    {
      "epoch": 8.90350341796875e-06,
      "model_forward_time": 5.584639549255371,
      "step": 9336
    },
    {
      "epoch": 8.90350341796875e-06,
      "step": 9336,
      "training_step_time": 17.02143621444702
    },
    {
      "epoch": 8.904457092285156e-06,
      "model_forward_time": 5.594165802001953,
      "step": 9337
    },
    {
      "epoch": 8.904457092285156e-06,
      "step": 9337,
      "training_step_time": 17.060264110565186
    },
    {
      "epoch": 8.905410766601562e-06,
      "model_forward_time": 5.5944178104400635,
      "step": 9338
    },
    {
      "epoch": 8.905410766601562e-06,
      "step": 9338,
      "training_step_time": 17.068013668060303
    },
    {
      "epoch": 8.906364440917969e-06,
      "model_forward_time": 5.610093116760254,
      "step": 9339
    },
    {
      "epoch": 8.906364440917969e-06,
      "step": 9339,
      "training_step_time": 17.10222578048706
    },
    {
      "epoch": 8.907318115234375e-06,
      "grad_norm": 0.6491045951843262,
      "learning_rate": 1e-05,
      "loss": 0.0662,
      "step": 9340
    },
    {
      "epoch": 8.907318115234375e-06,
      "model_forward_time": 5.6098291873931885,
      "step": 9340
    },
    {
      "dynamics_loss_avg": 0.052327371761202814,
      "epoch": 8.907318115234375e-06,
      "step": 9340
    },
    {
      "action_loss_avg": 0.009345389669761062,
      "epoch": 8.907318115234375e-06,
      "step": 9340
    },
    {
      "epoch": 8.907318115234375e-06,
      "step": 9340,
      "training_step_time": 17.453461408615112
    },
    {
      "epoch": 8.90827178955078e-06,
      "model_forward_time": 5.5874855518341064,
      "step": 9341
    },
    {
      "epoch": 8.90827178955078e-06,
      "step": 9341,
      "training_step_time": 17.010336875915527
    },
    {
      "epoch": 8.909225463867188e-06,
      "model_forward_time": 5.580992698669434,
      "step": 9342
    },
    {
      "epoch": 8.909225463867188e-06,
      "step": 9342,
      "training_step_time": 16.98654341697693
    },
    {
      "epoch": 8.910179138183594e-06,
      "model_forward_time": 5.580952882766724,
      "step": 9343
    },
    {
      "epoch": 8.910179138183594e-06,
      "step": 9343,
      "training_step_time": 17.02854561805725
    },
    {
      "epoch": 8.9111328125e-06,
      "model_forward_time": 5.595523357391357,
      "step": 9344
    },
    {
      "epoch": 8.9111328125e-06,
      "step": 9344,
      "training_step_time": 17.066004991531372
    },
    {
      "epoch": 8.912086486816407e-06,
      "model_forward_time": 5.60778021812439,
      "step": 9345
    },
    {
      "epoch": 8.912086486816407e-06,
      "step": 9345,
      "training_step_time": 17.081719636917114
    },
    {
      "epoch": 8.913040161132813e-06,
      "model_forward_time": 5.976025819778442,
      "step": 9346
    },
    {
      "epoch": 8.913040161132813e-06,
      "step": 9346,
      "training_step_time": 17.446632862091064
    },
    {
      "epoch": 8.913993835449218e-06,
      "model_forward_time": 5.61219048500061,
      "step": 9347
    },
    {
      "epoch": 8.913993835449218e-06,
      "step": 9347,
      "training_step_time": 17.097892999649048
    },
    {
      "epoch": 8.914947509765626e-06,
      "model_forward_time": 5.613937616348267,
      "step": 9348
    },
    {
      "epoch": 8.914947509765626e-06,
      "step": 9348,
      "training_step_time": 17.1083025932312
    },
    {
      "epoch": 8.915901184082031e-06,
      "model_forward_time": 5.616349935531616,
      "step": 9349
    },
    {
      "epoch": 8.915901184082031e-06,
      "step": 9349,
      "training_step_time": 17.106390714645386
    },
    {
      "epoch": 8.916854858398437e-06,
      "grad_norm": 0.8017641305923462,
      "learning_rate": 1e-05,
      "loss": 0.0642,
      "step": 9350
    },
    {
      "epoch": 8.916854858398437e-06,
      "model_forward_time": 5.6052021980285645,
      "step": 9350
    },
    {
      "dynamics_loss_avg": 0.06112874466925859,
      "epoch": 8.916854858398437e-06,
      "step": 9350
    },
    {
      "action_loss_avg": 0.007229709485545755,
      "epoch": 8.916854858398437e-06,
      "step": 9350
    },
    {
      "epoch": 8.916854858398437e-06,
      "step": 9350,
      "training_step_time": 17.097733974456787
    },
    {
      "epoch": 8.917808532714845e-06,
      "model_forward_time": 5.995734214782715,
      "step": 9351
    },
    {
      "epoch": 8.917808532714845e-06,
      "step": 9351,
      "training_step_time": 17.458215951919556
    },
    {
      "epoch": 8.91876220703125e-06,
      "model_forward_time": 5.587389945983887,
      "step": 9352
    },
    {
      "epoch": 8.91876220703125e-06,
      "step": 9352,
      "training_step_time": 17.02649974822998
    },
    {
      "epoch": 8.919715881347656e-06,
      "model_forward_time": 5.59570837020874,
      "step": 9353
    },
    {
      "epoch": 8.919715881347656e-06,
      "step": 9353,
      "training_step_time": 17.074984312057495
    },
    {
      "epoch": 8.920669555664062e-06,
      "model_forward_time": 5.611538887023926,
      "step": 9354
    },
    {
      "epoch": 8.920669555664062e-06,
      "step": 9354,
      "training_step_time": 17.108595371246338
    },
    {
      "epoch": 8.92162322998047e-06,
      "model_forward_time": 5.627418756484985,
      "step": 9355
    },
    {
      "epoch": 8.92162322998047e-06,
      "step": 9355,
      "training_step_time": 17.139124393463135
    },
    {
      "epoch": 8.922576904296875e-06,
      "model_forward_time": 5.628183841705322,
      "step": 9356
    },
    {
      "epoch": 8.922576904296875e-06,
      "step": 9356,
      "training_step_time": 17.136935710906982
    },
    {
      "epoch": 8.92353057861328e-06,
      "model_forward_time": 5.9801719188690186,
      "step": 9357
    },
    {
      "epoch": 8.92353057861328e-06,
      "step": 9357,
      "training_step_time": 17.4705069065094
    },
    {
      "epoch": 8.924484252929688e-06,
      "model_forward_time": 5.605907440185547,
      "step": 9358
    },
    {
      "epoch": 8.924484252929688e-06,
      "step": 9358,
      "training_step_time": 17.07620620727539
    },
    {
      "epoch": 8.925437927246094e-06,
      "model_forward_time": 5.613275527954102,
      "step": 9359
    },
    {
      "epoch": 8.925437927246094e-06,
      "step": 9359,
      "training_step_time": 17.10713791847229
    },
    {
      "epoch": 8.9263916015625e-06,
      "grad_norm": 0.887397050857544,
      "learning_rate": 1e-05,
      "loss": 0.0677,
      "step": 9360
    },
    {
      "epoch": 8.9263916015625e-06,
      "model_forward_time": 5.615416526794434,
      "step": 9360
    },
    {
      "dynamics_loss_avg": 0.05748120453208685,
      "epoch": 8.9263916015625e-06,
      "step": 9360
    },
    {
      "action_loss_avg": 0.009534575953148305,
      "epoch": 8.9263916015625e-06,
      "step": 9360
    },
    {
      "epoch": 8.9263916015625e-06,
      "step": 9360,
      "training_step_time": 17.075945615768433
    },
    {
      "epoch": 8.927345275878907e-06,
      "model_forward_time": 5.5954179763793945,
      "step": 9361
    },
    {
      "epoch": 8.927345275878907e-06,
      "step": 9361,
      "training_step_time": 17.076889038085938
    },
    {
      "epoch": 8.928298950195313e-06,
      "model_forward_time": 5.65057373046875,
      "step": 9362
    },
    {
      "epoch": 8.928298950195313e-06,
      "step": 9362,
      "training_step_time": 17.539531707763672
    },
    {
      "epoch": 8.929252624511718e-06,
      "model_forward_time": 5.661489248275757,
      "step": 9363
    },
    {
      "epoch": 8.929252624511718e-06,
      "step": 9363,
      "training_step_time": 17.17303156852722
    },
    {
      "epoch": 8.930206298828126e-06,
      "model_forward_time": 5.644225120544434,
      "step": 9364
    },
    {
      "epoch": 8.930206298828126e-06,
      "step": 9364,
      "training_step_time": 17.09726643562317
    },
    {
      "epoch": 8.931159973144531e-06,
      "model_forward_time": 5.5858354568481445,
      "step": 9365
    },
    {
      "epoch": 8.931159973144531e-06,
      "step": 9365,
      "training_step_time": 17.016366720199585
    },
    {
      "epoch": 8.932113647460937e-06,
      "model_forward_time": 5.584698915481567,
      "step": 9366
    },
    {
      "epoch": 8.932113647460937e-06,
      "step": 9366,
      "training_step_time": 17.043450355529785
    },
    {
      "epoch": 8.933067321777345e-06,
      "model_forward_time": 5.59749960899353,
      "step": 9367
    },
    {
      "epoch": 8.933067321777345e-06,
      "step": 9367,
      "training_step_time": 17.07143211364746
    },
    {
      "epoch": 8.93402099609375e-06,
      "model_forward_time": 5.999127149581909,
      "step": 9368
    },
    {
      "epoch": 8.93402099609375e-06,
      "step": 9368,
      "training_step_time": 17.472442865371704
    },
    {
      "epoch": 8.934974670410156e-06,
      "model_forward_time": 5.600835800170898,
      "step": 9369
    },
    {
      "epoch": 8.934974670410156e-06,
      "step": 9369,
      "training_step_time": 17.090412139892578
    },
    {
      "epoch": 8.935928344726562e-06,
      "grad_norm": 0.6747838258743286,
      "learning_rate": 1e-05,
      "loss": 0.0608,
      "step": 9370
    },
    {
      "epoch": 8.935928344726562e-06,
      "model_forward_time": 5.614565849304199,
      "step": 9370
    },
    {
      "dynamics_loss_avg": 0.05502837300300598,
      "epoch": 8.935928344726562e-06,
      "step": 9370
    },
    {
      "action_loss_avg": 0.007611097954213619,
      "epoch": 8.935928344726562e-06,
      "step": 9370
    },
    {
      "epoch": 8.935928344726562e-06,
      "step": 9370,
      "training_step_time": 17.111589908599854
    },
    {
      "epoch": 8.93688201904297e-06,
      "model_forward_time": 5.613606929779053,
      "step": 9371
    },
    {
      "epoch": 8.93688201904297e-06,
      "step": 9371,
      "training_step_time": 17.094361782073975
    },
    {
      "epoch": 8.937835693359375e-06,
      "model_forward_time": 5.60155987739563,
      "step": 9372
    },
    {
      "epoch": 8.937835693359375e-06,
      "step": 9372,
      "training_step_time": 17.05638027191162
    },
    {
      "epoch": 8.93878936767578e-06,
      "model_forward_time": 5.595872163772583,
      "step": 9373
    },
    {
      "epoch": 8.93878936767578e-06,
      "step": 9373,
      "training_step_time": 17.428409814834595
    },
    {
      "epoch": 8.939743041992188e-06,
      "model_forward_time": 5.5865843296051025,
      "step": 9374
    },
    {
      "epoch": 8.939743041992188e-06,
      "step": 9374,
      "training_step_time": 17.03976345062256
    },
    {
      "epoch": 8.940696716308594e-06,
      "model_forward_time": 5.596073150634766,
      "step": 9375
    },
    {
      "epoch": 8.940696716308594e-06,
      "step": 9375,
      "training_step_time": 17.0810866355896
    },
    {
      "epoch": 8.941650390625e-06,
      "model_forward_time": 5.625322341918945,
      "step": 9376
    },
    {
      "epoch": 8.941650390625e-06,
      "step": 9376,
      "training_step_time": 17.13174819946289
    },
    {
      "epoch": 8.942604064941407e-06,
      "model_forward_time": 5.621766805648804,
      "step": 9377
    },
    {
      "epoch": 8.942604064941407e-06,
      "step": 9377,
      "training_step_time": 17.120879888534546
    },
    {
      "epoch": 8.943557739257813e-06,
      "model_forward_time": 5.604510545730591,
      "step": 9378
    },
    {
      "epoch": 8.943557739257813e-06,
      "step": 9378,
      "training_step_time": 17.08312702178955
    },
    {
      "epoch": 8.944511413574218e-06,
      "model_forward_time": 5.964612722396851,
      "step": 9379
    },
    {
      "epoch": 8.944511413574218e-06,
      "step": 9379,
      "training_step_time": 17.430192470550537
    },
    {
      "epoch": 8.945465087890626e-06,
      "grad_norm": 1.4421827793121338,
      "learning_rate": 1e-05,
      "loss": 0.0686,
      "step": 9380
    },
    {
      "epoch": 8.945465087890626e-06,
      "model_forward_time": 5.598289489746094,
      "step": 9380
    },
    {
      "dynamics_loss_avg": 0.05709376074373722,
      "epoch": 8.945465087890626e-06,
      "step": 9380
    },
    {
      "action_loss_avg": 0.008930378430522978,
      "epoch": 8.945465087890626e-06,
      "step": 9380
    },
    {
      "epoch": 8.945465087890626e-06,
      "step": 9380,
      "training_step_time": 17.064794063568115
    },
    {
      "epoch": 8.946418762207031e-06,
      "model_forward_time": 5.604602098464966,
      "step": 9381
    },
    {
      "epoch": 8.946418762207031e-06,
      "step": 9381,
      "training_step_time": 17.039268732070923
    },
    {
      "epoch": 8.947372436523437e-06,
      "model_forward_time": 5.59128999710083,
      "step": 9382
    },
    {
      "epoch": 8.947372436523437e-06,
      "step": 9382,
      "training_step_time": 17.06105899810791
    },
    {
      "epoch": 8.948326110839843e-06,
      "model_forward_time": 5.600329399108887,
      "step": 9383
    },
    {
      "epoch": 8.948326110839843e-06,
      "step": 9383,
      "training_step_time": 17.085682153701782
    },
    {
      "epoch": 8.94927978515625e-06,
      "model_forward_time": 5.950911521911621,
      "step": 9384
    },
    {
      "epoch": 8.94927978515625e-06,
      "step": 9384,
      "training_step_time": 17.417448043823242
    },
    {
      "epoch": 8.950233459472656e-06,
      "model_forward_time": 5.609012842178345,
      "step": 9385
    },
    {
      "epoch": 8.950233459472656e-06,
      "step": 9385,
      "training_step_time": 17.066113233566284
    },
    {
      "epoch": 8.951187133789062e-06,
      "model_forward_time": 5.593981981277466,
      "step": 9386
    },
    {
      "epoch": 8.951187133789062e-06,
      "step": 9386,
      "training_step_time": 17.02513813972473
    },
    {
      "epoch": 8.95214080810547e-06,
      "model_forward_time": 5.591937065124512,
      "step": 9387
    },
    {
      "epoch": 8.95214080810547e-06,
      "step": 9387,
      "training_step_time": 17.056620121002197
    },
    {
      "epoch": 8.953094482421875e-06,
      "model_forward_time": 5.603722095489502,
      "step": 9388
    },
    {
      "epoch": 8.953094482421875e-06,
      "step": 9388,
      "training_step_time": 17.080971002578735
    },
    {
      "epoch": 8.95404815673828e-06,
      "model_forward_time": 5.614103555679321,
      "step": 9389
    },
    {
      "epoch": 8.95404815673828e-06,
      "step": 9389,
      "training_step_time": 17.113951921463013
    },
    {
      "epoch": 8.955001831054688e-06,
      "grad_norm": 0.7054892182350159,
      "learning_rate": 1e-05,
      "loss": 0.0721,
      "step": 9390
    },
    {
      "epoch": 8.955001831054688e-06,
      "model_forward_time": 5.978326320648193,
      "step": 9390
    },
    {
      "dynamics_loss_avg": 0.060022206231951716,
      "epoch": 8.955001831054688e-06,
      "step": 9390
    },
    {
      "action_loss_avg": 0.0114236562512815,
      "epoch": 8.955001831054688e-06,
      "step": 9390
    },
    {
      "epoch": 8.955001831054688e-06,
      "step": 9390,
      "training_step_time": 17.46709442138672
    },
    {
      "epoch": 8.955955505371094e-06,
      "model_forward_time": 5.60898232460022,
      "step": 9391
    },
    {
      "epoch": 8.955955505371094e-06,
      "step": 9391,
      "training_step_time": 17.091880798339844
    },
    {
      "epoch": 8.9569091796875e-06,
      "model_forward_time": 5.611110687255859,
      "step": 9392
    },
    {
      "epoch": 8.9569091796875e-06,
      "step": 9392,
      "training_step_time": 17.09915852546692
    },
    {
      "epoch": 8.957862854003907e-06,
      "model_forward_time": 5.6235671043396,
      "step": 9393
    },
    {
      "epoch": 8.957862854003907e-06,
      "step": 9393,
      "training_step_time": 17.112404584884644
    },
    {
      "epoch": 8.958816528320313e-06,
      "model_forward_time": 5.614197731018066,
      "step": 9394
    },
    {
      "epoch": 8.958816528320313e-06,
      "step": 9394,
      "training_step_time": 17.10780119895935
    },
    {
      "epoch": 8.959770202636718e-06,
      "model_forward_time": 5.633393049240112,
      "step": 9395
    },
    {
      "epoch": 8.959770202636718e-06,
      "step": 9395,
      "training_step_time": 17.464462757110596
    },
    {
      "epoch": 8.960723876953126e-06,
      "model_forward_time": 5.602738857269287,
      "step": 9396
    },
    {
      "epoch": 8.960723876953126e-06,
      "step": 9396,
      "training_step_time": 17.088141202926636
    },
    {
      "epoch": 8.961677551269531e-06,
      "model_forward_time": 5.603530406951904,
      "step": 9397
    },
    {
      "epoch": 8.961677551269531e-06,
      "step": 9397,
      "training_step_time": 17.069757223129272
    },
    {
      "epoch": 8.962631225585937e-06,
      "model_forward_time": 5.5900092124938965,
      "step": 9398
    },
    {
      "epoch": 8.962631225585937e-06,
      "step": 9398,
      "training_step_time": 17.030582189559937
    },
    {
      "epoch": 8.963584899902343e-06,
      "model_forward_time": 5.5879600048065186,
      "step": 9399
    },
    {
      "epoch": 8.963584899902343e-06,
      "step": 9399,
      "training_step_time": 17.04837918281555
    },
    {
      "epoch": 8.96453857421875e-06,
      "grad_norm": 0.950886070728302,
      "learning_rate": 1e-05,
      "loss": 0.0696,
      "step": 9400
    },
    {
      "epoch": 8.96453857421875e-06,
      "model_forward_time": 5.609508991241455,
      "step": 9400
    },
    {
      "dynamics_loss_avg": 0.06079757604748011,
      "epoch": 8.96453857421875e-06,
      "step": 9400
    },
    {
      "action_loss_avg": 0.0066034088376909494,
      "epoch": 8.96453857421875e-06,
      "step": 9400
    },
    {
      "epoch": 8.96453857421875e-06,
      "step": 9400,
      "training_step_time": 17.10285758972168
    },
    {
      "epoch": 8.965492248535156e-06,
      "model_forward_time": 5.976475954055786,
      "step": 9401
    },
    {
      "epoch": 8.965492248535156e-06,
      "step": 9401,
      "training_step_time": 17.468749046325684
    },
    {
      "epoch": 8.966445922851562e-06,
      "model_forward_time": 5.614690065383911,
      "step": 9402
    },
    {
      "epoch": 8.966445922851562e-06,
      "step": 9402,
      "training_step_time": 17.11417055130005
    },
    {
      "epoch": 8.96739959716797e-06,
      "model_forward_time": 5.612360715866089,
      "step": 9403
    },
    {
      "epoch": 8.96739959716797e-06,
      "step": 9403,
      "training_step_time": 17.10444164276123
    },
    {
      "epoch": 8.968353271484375e-06,
      "model_forward_time": 5.608273267745972,
      "step": 9404
    },
    {
      "epoch": 8.968353271484375e-06,
      "step": 9404,
      "training_step_time": 17.08397912979126
    },
    {
      "epoch": 8.96930694580078e-06,
      "model_forward_time": 5.600203275680542,
      "step": 9405
    },
    {
      "epoch": 8.96930694580078e-06,
      "step": 9405,
      "training_step_time": 17.07194209098816
    },
    {
      "epoch": 8.970260620117188e-06,
      "model_forward_time": 5.595022916793823,
      "step": 9406
    },
    {
      "epoch": 8.970260620117188e-06,
      "step": 9406,
      "training_step_time": 17.40628457069397
    },
    {
      "epoch": 8.971214294433594e-06,
      "model_forward_time": 5.597042560577393,
      "step": 9407
    },
    {
      "epoch": 8.971214294433594e-06,
      "step": 9407,
      "training_step_time": 17.04192066192627
    },
    {
      "epoch": 8.97216796875e-06,
      "model_forward_time": 4.272303819656372,
      "step": 9408
    },
    {
      "epoch": 8.97216796875e-06,
      "step": 9408,
      "training_step_time": 12.794790029525757
    },
    {
      "epoch": 8.973121643066407e-06,
      "model_forward_time": 5.611236810684204,
      "step": 9409
    },
    {
      "epoch": 8.973121643066407e-06,
      "step": 9409,
      "training_step_time": 17.104281902313232
    },
    {
      "epoch": 8.974075317382813e-06,
      "grad_norm": 0.6147035956382751,
      "learning_rate": 1e-05,
      "loss": 0.0518,
      "step": 9410
    },
    {
      "epoch": 8.974075317382813e-06,
      "model_forward_time": 5.60234522819519,
      "step": 9410
    },
    {
      "dynamics_loss_avg": 0.05074226446449757,
      "epoch": 8.974075317382813e-06,
      "step": 9410
    },
    {
      "action_loss_avg": 0.006699787091929465,
      "epoch": 8.974075317382813e-06,
      "step": 9410
    },
    {
      "epoch": 8.974075317382813e-06,
      "step": 9410,
      "training_step_time": 17.07636785507202
    },
    {
      "epoch": 8.975028991699218e-06,
      "model_forward_time": 5.601885557174683,
      "step": 9411
    },
    {
      "epoch": 8.975028991699218e-06,
      "step": 9411,
      "training_step_time": 17.085816621780396
    },
    {
      "epoch": 8.975982666015626e-06,
      "model_forward_time": 5.978433609008789,
      "step": 9412
    },
    {
      "epoch": 8.975982666015626e-06,
      "step": 9412,
      "training_step_time": 17.46353840827942
    },
    {
      "epoch": 8.976936340332031e-06,
      "model_forward_time": 5.612781047821045,
      "step": 9413
    },
    {
      "epoch": 8.976936340332031e-06,
      "step": 9413,
      "training_step_time": 17.091519832611084
    },
    {
      "epoch": 8.977890014648437e-06,
      "model_forward_time": 5.595040798187256,
      "step": 9414
    },
    {
      "epoch": 8.977890014648437e-06,
      "step": 9414,
      "training_step_time": 17.03404927253723
    },
    {
      "epoch": 8.978843688964843e-06,
      "model_forward_time": 5.588324308395386,
      "step": 9415
    },
    {
      "epoch": 8.978843688964843e-06,
      "step": 9415,
      "training_step_time": 17.038697719573975
    },
    {
      "epoch": 8.97979736328125e-06,
      "model_forward_time": 5.573753118515015,
      "step": 9416
    },
    {
      "epoch": 8.97979736328125e-06,
      "step": 9416,
      "training_step_time": 17.000242948532104
    },
    {
      "epoch": 8.980751037597656e-06,
      "model_forward_time": 5.601405382156372,
      "step": 9417
    },
    {
      "epoch": 8.980751037597656e-06,
      "step": 9417,
      "training_step_time": 17.38121485710144
    },
    {
      "epoch": 8.981704711914062e-06,
      "model_forward_time": 5.589022636413574,
      "step": 9418
    },
    {
      "epoch": 8.981704711914062e-06,
      "step": 9418,
      "training_step_time": 17.055272817611694
    },
    {
      "epoch": 8.98265838623047e-06,
      "model_forward_time": 5.6015944480896,
      "step": 9419
    },
    {
      "epoch": 8.98265838623047e-06,
      "step": 9419,
      "training_step_time": 17.093219757080078
    },
    {
      "epoch": 8.983612060546875e-06,
      "grad_norm": 1.8046627044677734,
      "learning_rate": 1e-05,
      "loss": 0.0696,
      "step": 9420
    },
    {
      "epoch": 8.983612060546875e-06,
      "model_forward_time": 5.613487005233765,
      "step": 9420
    },
    {
      "dynamics_loss_avg": 0.05719976015388965,
      "epoch": 8.983612060546875e-06,
      "step": 9420
    },
    {
      "action_loss_avg": 0.00985446865670383,
      "epoch": 8.983612060546875e-06,
      "step": 9420
    },
    {
      "epoch": 8.983612060546875e-06,
      "step": 9420,
      "training_step_time": 17.125904321670532
    },
    {
      "epoch": 8.98456573486328e-06,
      "model_forward_time": 5.6234681606292725,
      "step": 9421
    },
    {
      "epoch": 8.98456573486328e-06,
      "step": 9421,
      "training_step_time": 17.12121081352234
    },
    {
      "epoch": 8.985519409179688e-06,
      "model_forward_time": 5.600816249847412,
      "step": 9422
    },
    {
      "epoch": 8.985519409179688e-06,
      "step": 9422,
      "training_step_time": 17.053004026412964
    },
    {
      "epoch": 8.986473083496094e-06,
      "model_forward_time": 5.947423458099365,
      "step": 9423
    },
    {
      "epoch": 8.986473083496094e-06,
      "step": 9423,
      "training_step_time": 17.393645763397217
    },
    {
      "epoch": 8.9874267578125e-06,
      "model_forward_time": 5.577258825302124,
      "step": 9424
    },
    {
      "epoch": 8.9874267578125e-06,
      "step": 9424,
      "training_step_time": 17.011152029037476
    },
    {
      "epoch": 8.988380432128907e-06,
      "model_forward_time": 5.575690507888794,
      "step": 9425
    },
    {
      "epoch": 8.988380432128907e-06,
      "step": 9425,
      "training_step_time": 16.996463775634766
    },
    {
      "epoch": 8.989334106445313e-06,
      "model_forward_time": 5.585264682769775,
      "step": 9426
    },
    {
      "epoch": 8.989334106445313e-06,
      "step": 9426,
      "training_step_time": 17.05047607421875
    },
    {
      "epoch": 8.990287780761718e-06,
      "model_forward_time": 5.592987775802612,
      "step": 9427
    },
    {
      "epoch": 8.990287780761718e-06,
      "step": 9427,
      "training_step_time": 17.058351278305054
    },
    {
      "epoch": 8.991241455078126e-06,
      "model_forward_time": 5.952766418457031,
      "step": 9428
    },
    {
      "epoch": 8.991241455078126e-06,
      "step": 9428,
      "training_step_time": 17.432125329971313
    },
    {
      "epoch": 8.992195129394532e-06,
      "model_forward_time": 5.604086875915527,
      "step": 9429
    },
    {
      "epoch": 8.992195129394532e-06,
      "step": 9429,
      "training_step_time": 17.094255924224854
    },
    {
      "epoch": 8.993148803710937e-06,
      "grad_norm": 0.5338079333305359,
      "learning_rate": 1e-05,
      "loss": 0.0625,
      "step": 9430
    },
    {
      "epoch": 8.993148803710937e-06,
      "model_forward_time": 5.605456590652466,
      "step": 9430
    },
    {
      "dynamics_loss_avg": 0.05969277024269104,
      "epoch": 8.993148803710937e-06,
      "step": 9430
    },
    {
      "action_loss_avg": 0.008119688485749066,
      "epoch": 8.993148803710937e-06,
      "step": 9430
    },
    {
      "epoch": 8.993148803710937e-06,
      "step": 9430,
      "training_step_time": 17.051255702972412
    },
    {
      "epoch": 8.994102478027343e-06,
      "model_forward_time": 5.588903903961182,
      "step": 9431
    },
    {
      "epoch": 8.994102478027343e-06,
      "step": 9431,
      "training_step_time": 17.032031059265137
    },
    {
      "epoch": 8.99505615234375e-06,
      "model_forward_time": 5.583221197128296,
      "step": 9432
    },
    {
      "epoch": 8.99505615234375e-06,
      "step": 9432,
      "training_step_time": 16.984673976898193
    },
    {
      "epoch": 8.996009826660156e-06,
      "model_forward_time": 5.603798151016235,
      "step": 9433
    },
    {
      "epoch": 8.996009826660156e-06,
      "step": 9433,
      "training_step_time": 17.06985068321228
    },
    {
      "epoch": 8.996963500976562e-06,
      "model_forward_time": 5.963090419769287,
      "step": 9434
    },
    {
      "epoch": 8.996963500976562e-06,
      "step": 9434,
      "training_step_time": 17.452457427978516
    },
    {
      "epoch": 8.99791717529297e-06,
      "model_forward_time": 5.629708766937256,
      "step": 9435
    },
    {
      "epoch": 8.99791717529297e-06,
      "step": 9435,
      "training_step_time": 17.15098762512207
    },
    {
      "epoch": 8.998870849609375e-06,
      "model_forward_time": 5.632387161254883,
      "step": 9436
    },
    {
      "epoch": 8.998870849609375e-06,
      "step": 9436,
      "training_step_time": 17.15714430809021
    },
    {
      "epoch": 8.99982452392578e-06,
      "model_forward_time": 5.623053312301636,
      "step": 9437
    },
    {
      "epoch": 8.99982452392578e-06,
      "step": 9437,
      "training_step_time": 17.137202501296997
    },
    {
      "epoch": 9.000778198242188e-06,
      "model_forward_time": 5.6255457401275635,
      "step": 9438
    },
    {
      "epoch": 9.000778198242188e-06,
      "step": 9438,
      "training_step_time": 17.133567333221436
    },
    {
      "epoch": 9.001731872558594e-06,
      "model_forward_time": 5.975409030914307,
      "step": 9439
    },
    {
      "epoch": 9.001731872558594e-06,
      "step": 9439,
      "training_step_time": 17.48385524749756
    },
    {
      "epoch": 9.002685546875e-06,
      "grad_norm": 0.6209710240364075,
      "learning_rate": 1e-05,
      "loss": 0.0803,
      "step": 9440
    },
    {
      "epoch": 9.002685546875e-06,
      "model_forward_time": 5.6215431690216064,
      "step": 9440
    },
    {
      "dynamics_loss_avg": 0.06672382391989232,
      "epoch": 9.002685546875e-06,
      "step": 9440
    },
    {
      "action_loss_avg": 0.008354925271123647,
      "epoch": 9.002685546875e-06,
      "step": 9440
    },
    {
      "epoch": 9.002685546875e-06,
      "step": 9440,
      "training_step_time": 17.08369541168213
    },
    {
      "epoch": 9.003639221191407e-06,
      "model_forward_time": 5.576161623001099,
      "step": 9441
    },
    {
      "epoch": 9.003639221191407e-06,
      "step": 9441,
      "training_step_time": 16.999528169631958
    },
    {
      "epoch": 9.004592895507813e-06,
      "model_forward_time": 5.566413402557373,
      "step": 9442
    },
    {
      "epoch": 9.004592895507813e-06,
      "step": 9442,
      "training_step_time": 16.99841547012329
    },
    {
      "epoch": 9.005546569824218e-06,
      "model_forward_time": 5.561320066452026,
      "step": 9443
    },
    {
      "epoch": 9.005546569824218e-06,
      "step": 9443,
      "training_step_time": 16.97156834602356
    },
    {
      "epoch": 9.006500244140626e-06,
      "model_forward_time": 5.566457509994507,
      "step": 9444
    },
    {
      "epoch": 9.006500244140626e-06,
      "step": 9444,
      "training_step_time": 17.348588943481445
    },
    {
      "epoch": 9.007453918457032e-06,
      "model_forward_time": 5.566933870315552,
      "step": 9445
    },
    {
      "epoch": 9.007453918457032e-06,
      "step": 9445,
      "training_step_time": 16.97937321662903
    },
    {
      "epoch": 9.008407592773437e-06,
      "model_forward_time": 5.564895153045654,
      "step": 9446
    },
    {
      "epoch": 9.008407592773437e-06,
      "step": 9446,
      "training_step_time": 17.000898361206055
    },
    {
      "epoch": 9.009361267089843e-06,
      "model_forward_time": 5.570336818695068,
      "step": 9447
    },
    {
      "epoch": 9.009361267089843e-06,
      "step": 9447,
      "training_step_time": 16.982186317443848
    },
    {
      "epoch": 9.01031494140625e-06,
      "model_forward_time": 5.584501266479492,
      "step": 9448
    },
    {
      "epoch": 9.01031494140625e-06,
      "step": 9448,
      "training_step_time": 17.005512237548828
    },
    {
      "epoch": 9.011268615722656e-06,
      "model_forward_time": 5.584198951721191,
      "step": 9449
    },
    {
      "epoch": 9.011268615722656e-06,
      "step": 9449,
      "training_step_time": 17.02998399734497
    },
    {
      "epoch": 9.012222290039062e-06,
      "grad_norm": 1.3642191886901855,
      "learning_rate": 1e-05,
      "loss": 0.0568,
      "step": 9450
    },
    {
      "epoch": 9.012222290039062e-06,
      "model_forward_time": 5.960280179977417,
      "step": 9450
    },
    {
      "dynamics_loss_avg": 0.04682505652308464,
      "epoch": 9.012222290039062e-06,
      "step": 9450
    },
    {
      "action_loss_avg": 0.0071784227155148985,
      "epoch": 9.012222290039062e-06,
      "step": 9450
    },
    {
      "epoch": 9.012222290039062e-06,
      "step": 9450,
      "training_step_time": 17.439743757247925
    },
    {
      "epoch": 9.01317596435547e-06,
      "model_forward_time": 5.60589337348938,
      "step": 9451
    },
    {
      "epoch": 9.01317596435547e-06,
      "step": 9451,
      "training_step_time": 17.099595308303833
    },
    {
      "epoch": 9.014129638671875e-06,
      "model_forward_time": 5.60676383972168,
      "step": 9452
    },
    {
      "epoch": 9.014129638671875e-06,
      "step": 9452,
      "training_step_time": 17.11408758163452
    },
    {
      "epoch": 9.01508331298828e-06,
      "model_forward_time": 5.613877296447754,
      "step": 9453
    },
    {
      "epoch": 9.01508331298828e-06,
      "step": 9453,
      "training_step_time": 17.08972716331482
    },
    {
      "epoch": 9.016036987304688e-06,
      "model_forward_time": 5.581673860549927,
      "step": 9454
    },
    {
      "epoch": 9.016036987304688e-06,
      "step": 9454,
      "training_step_time": 17.01742649078369
    },
    {
      "epoch": 9.016990661621094e-06,
      "model_forward_time": 5.977924108505249,
      "step": 9455
    },
    {
      "epoch": 9.016990661621094e-06,
      "step": 9455,
      "training_step_time": 17.38939380645752
    },
    {
      "epoch": 9.0179443359375e-06,
      "model_forward_time": 5.575059652328491,
      "step": 9456
    },
    {
      "epoch": 9.0179443359375e-06,
      "step": 9456,
      "training_step_time": 17.013277053833008
    },
    {
      "epoch": 9.018898010253907e-06,
      "model_forward_time": 5.590975999832153,
      "step": 9457
    },
    {
      "epoch": 9.018898010253907e-06,
      "step": 9457,
      "training_step_time": 17.016382217407227
    },
    {
      "epoch": 9.019851684570313e-06,
      "model_forward_time": 5.584925889968872,
      "step": 9458
    },
    {
      "epoch": 9.019851684570313e-06,
      "step": 9458,
      "training_step_time": 17.03106665611267
    },
    {
      "epoch": 9.020805358886718e-06,
      "model_forward_time": 5.59285569190979,
      "step": 9459
    },
    {
      "epoch": 9.020805358886718e-06,
      "step": 9459,
      "training_step_time": 17.053499698638916
    },
    {
      "epoch": 9.021759033203126e-06,
      "grad_norm": 0.6613996028900146,
      "learning_rate": 1e-05,
      "loss": 0.049,
      "step": 9460
    },
    {
      "epoch": 9.021759033203126e-06,
      "model_forward_time": 5.602800607681274,
      "step": 9460
    },
    {
      "dynamics_loss_avg": 0.04539264533668756,
      "epoch": 9.021759033203126e-06,
      "step": 9460
    },
    {
      "action_loss_avg": 0.007547736633569002,
      "epoch": 9.021759033203126e-06,
      "step": 9460
    },
    {
      "epoch": 9.021759033203126e-06,
      "step": 9460,
      "training_step_time": 17.05732822418213
    },
    {
      "epoch": 9.022712707519532e-06,
      "model_forward_time": 5.958750486373901,
      "step": 9461
    },
    {
      "epoch": 9.022712707519532e-06,
      "step": 9461,
      "training_step_time": 17.396697282791138
    },
    {
      "epoch": 9.023666381835937e-06,
      "model_forward_time": 5.606086730957031,
      "step": 9462
    },
    {
      "epoch": 9.023666381835937e-06,
      "step": 9462,
      "training_step_time": 17.07648801803589
    },
    {
      "epoch": 9.024620056152343e-06,
      "model_forward_time": 5.608076095581055,
      "step": 9463
    },
    {
      "epoch": 9.024620056152343e-06,
      "step": 9463,
      "training_step_time": 17.06549644470215
    },
    {
      "epoch": 9.02557373046875e-06,
      "model_forward_time": 5.590147972106934,
      "step": 9464
    },
    {
      "epoch": 9.02557373046875e-06,
      "step": 9464,
      "training_step_time": 17.02893042564392
    },
    {
      "epoch": 9.026527404785156e-06,
      "model_forward_time": 5.590159177780151,
      "step": 9465
    },
    {
      "epoch": 9.026527404785156e-06,
      "step": 9465,
      "training_step_time": 17.028650522232056
    },
    {
      "epoch": 9.027481079101562e-06,
      "model_forward_time": 5.958788871765137,
      "step": 9466
    },
    {
      "epoch": 9.027481079101562e-06,
      "step": 9466,
      "training_step_time": 17.424290895462036
    },
    {
      "epoch": 9.02843475341797e-06,
      "model_forward_time": 5.608883380889893,
      "step": 9467
    },
    {
      "epoch": 9.02843475341797e-06,
      "step": 9467,
      "training_step_time": 17.111947059631348
    },
    {
      "epoch": 9.029388427734375e-06,
      "model_forward_time": 5.620355129241943,
      "step": 9468
    },
    {
      "epoch": 9.029388427734375e-06,
      "step": 9468,
      "training_step_time": 17.105692148208618
    },
    {
      "epoch": 9.03034210205078e-06,
      "model_forward_time": 5.598946809768677,
      "step": 9469
    },
    {
      "epoch": 9.03034210205078e-06,
      "step": 9469,
      "training_step_time": 17.02833914756775
    },
    {
      "epoch": 9.031295776367188e-06,
      "grad_norm": 0.6764870285987854,
      "learning_rate": 1e-05,
      "loss": 0.0648,
      "step": 9470
    },
    {
      "epoch": 9.031295776367188e-06,
      "model_forward_time": 5.586233615875244,
      "step": 9470
    },
    {
      "dynamics_loss_avg": 0.05318578388541937,
      "epoch": 9.031295776367188e-06,
      "step": 9470
    },
    {
      "action_loss_avg": 0.008334397606085986,
      "epoch": 9.031295776367188e-06,
      "step": 9470
    },
    {
      "epoch": 9.031295776367188e-06,
      "step": 9470,
      "training_step_time": 17.008289098739624
    },
    {
      "epoch": 9.032249450683594e-06,
      "model_forward_time": 5.577518463134766,
      "step": 9471
    },
    {
      "epoch": 9.032249450683594e-06,
      "step": 9471,
      "training_step_time": 17.347280502319336
    },
    {
      "epoch": 9.033203125e-06,
      "model_forward_time": 5.575691223144531,
      "step": 9472
    },
    {
      "epoch": 9.033203125e-06,
      "step": 9472,
      "training_step_time": 17.016968250274658
    },
    {
      "epoch": 9.034156799316407e-06,
      "model_forward_time": 5.574124574661255,
      "step": 9473
    },
    {
      "epoch": 9.034156799316407e-06,
      "step": 9473,
      "training_step_time": 17.006141901016235
    },
    {
      "epoch": 9.035110473632813e-06,
      "model_forward_time": 5.580774545669556,
      "step": 9474
    },
    {
      "epoch": 9.035110473632813e-06,
      "step": 9474,
      "training_step_time": 17.001566886901855
    },
    {
      "epoch": 9.036064147949218e-06,
      "model_forward_time": 5.589495658874512,
      "step": 9475
    },
    {
      "epoch": 9.036064147949218e-06,
      "step": 9475,
      "training_step_time": 17.004143238067627
    },
    {
      "epoch": 9.037017822265626e-06,
      "model_forward_time": 5.5976433753967285,
      "step": 9476
    },
    {
      "epoch": 9.037017822265626e-06,
      "step": 9476,
      "training_step_time": 17.037861824035645
    },
    {
      "epoch": 9.037971496582032e-06,
      "model_forward_time": 5.964426040649414,
      "step": 9477
    },
    {
      "epoch": 9.037971496582032e-06,
      "step": 9477,
      "training_step_time": 17.448597192764282
    },
    {
      "epoch": 9.038925170898437e-06,
      "model_forward_time": 5.606626272201538,
      "step": 9478
    },
    {
      "epoch": 9.038925170898437e-06,
      "step": 9478,
      "training_step_time": 17.077425479888916
    },
    {
      "epoch": 9.039878845214843e-06,
      "model_forward_time": 5.602866172790527,
      "step": 9479
    },
    {
      "epoch": 9.039878845214843e-06,
      "step": 9479,
      "training_step_time": 17.06571125984192
    },
    {
      "epoch": 9.04083251953125e-06,
      "grad_norm": 0.5443345308303833,
      "learning_rate": 1e-05,
      "loss": 0.0636,
      "step": 9480
    },
    {
      "epoch": 9.04083251953125e-06,
      "model_forward_time": 5.597557544708252,
      "step": 9480
    },
    {
      "dynamics_loss_avg": 0.055575528368353846,
      "epoch": 9.04083251953125e-06,
      "step": 9480
    },
    {
      "action_loss_avg": 0.008086696173995734,
      "epoch": 9.04083251953125e-06,
      "step": 9480
    },
    {
      "epoch": 9.04083251953125e-06,
      "step": 9480,
      "training_step_time": 17.061288833618164
    },
    {
      "epoch": 9.041786193847656e-06,
      "model_forward_time": 5.596735715866089,
      "step": 9481
    },
    {
      "epoch": 9.041786193847656e-06,
      "step": 9481,
      "training_step_time": 17.069814205169678
    },
    {
      "epoch": 9.042739868164062e-06,
      "model_forward_time": 5.970279932022095,
      "step": 9482
    },
    {
      "epoch": 9.042739868164062e-06,
      "step": 9482,
      "training_step_time": 17.45110011100769
    },
    {
      "epoch": 9.04369354248047e-06,
      "model_forward_time": 5.605306386947632,
      "step": 9483
    },
    {
      "epoch": 9.04369354248047e-06,
      "step": 9483,
      "training_step_time": 17.092703104019165
    },
    {
      "epoch": 9.044647216796875e-06,
      "model_forward_time": 5.611610651016235,
      "step": 9484
    },
    {
      "epoch": 9.044647216796875e-06,
      "step": 9484,
      "training_step_time": 17.079218864440918
    },
    {
      "epoch": 9.04560089111328e-06,
      "model_forward_time": 5.612135648727417,
      "step": 9485
    },
    {
      "epoch": 9.04560089111328e-06,
      "step": 9485,
      "training_step_time": 17.109323978424072
    },
    {
      "epoch": 9.046554565429688e-06,
      "model_forward_time": 5.608665704727173,
      "step": 9486
    },
    {
      "epoch": 9.046554565429688e-06,
      "step": 9486,
      "training_step_time": 17.103843212127686
    },
    {
      "epoch": 9.047508239746094e-06,
      "model_forward_time": 5.652747631072998,
      "step": 9487
    },
    {
      "epoch": 9.047508239746094e-06,
      "step": 9487,
      "training_step_time": 17.14238429069519
    },
    {
      "epoch": 9.0484619140625e-06,
      "model_forward_time": 6.175214767456055,
      "step": 9488
    },
    {
      "epoch": 9.0484619140625e-06,
      "step": 9488,
      "training_step_time": 17.671659469604492
    },
    {
      "epoch": 9.049415588378907e-06,
      "model_forward_time": 5.636711120605469,
      "step": 9489
    },
    {
      "epoch": 9.049415588378907e-06,
      "step": 9489,
      "training_step_time": 17.128246784210205
    },
    {
      "epoch": 9.050369262695313e-06,
      "grad_norm": 0.5599038600921631,
      "learning_rate": 1e-05,
      "loss": 0.0542,
      "step": 9490
    },
    {
      "epoch": 9.050369262695313e-06,
      "model_forward_time": 5.605641841888428,
      "step": 9490
    },
    {
      "dynamics_loss_avg": 0.04858755394816398,
      "epoch": 9.050369262695313e-06,
      "step": 9490
    },
    {
      "action_loss_avg": 0.0056969084776937965,
      "epoch": 9.050369262695313e-06,
      "step": 9490
    },
    {
      "epoch": 9.050369262695313e-06,
      "step": 9490,
      "training_step_time": 17.090943336486816
    },
    {
      "epoch": 9.051322937011718e-06,
      "model_forward_time": 5.613280534744263,
      "step": 9491
    },
    {
      "epoch": 9.051322937011718e-06,
      "step": 9491,
      "training_step_time": 17.137336492538452
    },
    {
      "epoch": 9.052276611328126e-06,
      "model_forward_time": 5.616546154022217,
      "step": 9492
    },
    {
      "epoch": 9.052276611328126e-06,
      "step": 9492,
      "training_step_time": 17.113901376724243
    },
    {
      "epoch": 9.053230285644532e-06,
      "model_forward_time": 5.610530138015747,
      "step": 9493
    },
    {
      "epoch": 9.053230285644532e-06,
      "step": 9493,
      "training_step_time": 17.518895864486694
    },
    {
      "epoch": 9.054183959960937e-06,
      "model_forward_time": 5.609563112258911,
      "step": 9494
    },
    {
      "epoch": 9.054183959960937e-06,
      "step": 9494,
      "training_step_time": 17.107128143310547
    },
    {
      "epoch": 9.055137634277343e-06,
      "model_forward_time": 5.6110522747039795,
      "step": 9495
    },
    {
      "epoch": 9.055137634277343e-06,
      "step": 9495,
      "training_step_time": 17.079818964004517
    },
    {
      "epoch": 9.05609130859375e-06,
      "model_forward_time": 5.605010986328125,
      "step": 9496
    },
    {
      "epoch": 9.05609130859375e-06,
      "step": 9496,
      "training_step_time": 17.061983823776245
    },
    {
      "epoch": 9.057044982910156e-06,
      "model_forward_time": 5.590548276901245,
      "step": 9497
    },
    {
      "epoch": 9.057044982910156e-06,
      "step": 9497,
      "training_step_time": 17.040260076522827
    },
    {
      "epoch": 9.057998657226562e-06,
      "model_forward_time": 5.597554445266724,
      "step": 9498
    },
    {
      "epoch": 9.057998657226562e-06,
      "step": 9498,
      "training_step_time": 17.087602853775024
    },
    {
      "epoch": 9.05895233154297e-06,
      "model_forward_time": 6.030447483062744,
      "step": 9499
    },
    {
      "epoch": 9.05895233154297e-06,
      "step": 9499,
      "training_step_time": 17.498342752456665
    },
    {
      "epoch": 9.059906005859375e-06,
      "grad_norm": 1.376450777053833,
      "learning_rate": 1e-05,
      "loss": 0.0718,
      "step": 9500
    },
    {
      "epoch": 9.059906005859375e-06,
      "model_forward_time": 5.59662652015686,
      "step": 9500
    },
    {
      "dynamics_loss_avg": 0.05999197605997324,
      "epoch": 9.059906005859375e-06,
      "step": 9500
    },
    {
      "action_loss_avg": 0.008394511218648404,
      "epoch": 9.059906005859375e-06,
      "step": 9500
    },
    {
      "epoch": 9.059906005859375e-06,
      "step": 9500,
      "training_step_time": 17.050044536590576
    },
    {
      "epoch": 9.06085968017578e-06,
      "model_forward_time": 5.587209224700928,
      "step": 9501
    },
    {
      "epoch": 9.06085968017578e-06,
      "step": 9501,
      "training_step_time": 17.01265788078308
    },
    {
      "epoch": 9.061813354492188e-06,
      "model_forward_time": 5.575135946273804,
      "step": 9502
    },
    {
      "epoch": 9.061813354492188e-06,
      "step": 9502,
      "training_step_time": 16.991153478622437
    },
    {
      "epoch": 9.062767028808594e-06,
      "model_forward_time": 5.573505163192749,
      "step": 9503
    },
    {
      "epoch": 9.062767028808594e-06,
      "step": 9503,
      "training_step_time": 17.023012161254883
    },
    {
      "epoch": 9.063720703125e-06,
      "model_forward_time": 6.093080043792725,
      "step": 9504
    },
    {
      "epoch": 9.063720703125e-06,
      "step": 9504,
      "training_step_time": 17.519654750823975
    },
    {
      "epoch": 9.064674377441407e-06,
      "model_forward_time": 5.593183755874634,
      "step": 9505
    },
    {
      "epoch": 9.064674377441407e-06,
      "step": 9505,
      "training_step_time": 17.040359020233154
    },
    {
      "epoch": 9.065628051757813e-06,
      "model_forward_time": 5.593476057052612,
      "step": 9506
    },
    {
      "epoch": 9.065628051757813e-06,
      "step": 9506,
      "training_step_time": 17.053175926208496
    },
    {
      "epoch": 9.066581726074218e-06,
      "model_forward_time": 5.623213052749634,
      "step": 9507
    },
    {
      "epoch": 9.066581726074218e-06,
      "step": 9507,
      "training_step_time": 17.0891752243042
    },
    {
      "epoch": 9.067535400390624e-06,
      "model_forward_time": 5.597713470458984,
      "step": 9508
    },
    {
      "epoch": 9.067535400390624e-06,
      "step": 9508,
      "training_step_time": 17.055023670196533
    },
    {
      "epoch": 9.068489074707032e-06,
      "model_forward_time": 5.588951110839844,
      "step": 9509
    },
    {
      "epoch": 9.068489074707032e-06,
      "step": 9509,
      "training_step_time": 17.03054690361023
    },
    {
      "epoch": 9.069442749023437e-06,
      "grad_norm": 0.5709346532821655,
      "learning_rate": 1e-05,
      "loss": 0.0671,
      "step": 9510
    },
    {
      "epoch": 9.069442749023437e-06,
      "model_forward_time": 5.985942125320435,
      "step": 9510
    },
    {
      "dynamics_loss_avg": 0.05930393766611815,
      "epoch": 9.069442749023437e-06,
      "step": 9510
    },
    {
      "action_loss_avg": 0.009543885872699321,
      "epoch": 9.069442749023437e-06,
      "step": 9510
    },
    {
      "epoch": 9.069442749023437e-06,
      "step": 9510,
      "training_step_time": 17.432108402252197
    },
    {
      "epoch": 9.070396423339843e-06,
      "model_forward_time": 5.590209722518921,
      "step": 9511
    },
    {
      "epoch": 9.070396423339843e-06,
      "step": 9511,
      "training_step_time": 17.044946432113647
    },
    {
      "epoch": 9.07135009765625e-06,
      "model_forward_time": 5.60728645324707,
      "step": 9512
    },
    {
      "epoch": 9.07135009765625e-06,
      "step": 9512,
      "training_step_time": 17.081028938293457
    },
    {
      "epoch": 9.072303771972656e-06,
      "model_forward_time": 5.610932350158691,
      "step": 9513
    },
    {
      "epoch": 9.072303771972656e-06,
      "step": 9513,
      "training_step_time": 17.1123046875
    },
    {
      "epoch": 9.073257446289062e-06,
      "model_forward_time": 5.621534109115601,
      "step": 9514
    },
    {
      "epoch": 9.073257446289062e-06,
      "step": 9514,
      "training_step_time": 17.130573272705078
    },
    {
      "epoch": 9.07421112060547e-06,
      "model_forward_time": 5.988642692565918,
      "step": 9515
    },
    {
      "epoch": 9.07421112060547e-06,
      "step": 9515,
      "training_step_time": 17.49221920967102
    },
    {
      "epoch": 9.075164794921875e-06,
      "model_forward_time": 5.594766139984131,
      "step": 9516
    },
    {
      "epoch": 9.075164794921875e-06,
      "step": 9516,
      "training_step_time": 17.049712419509888
    },
    {
      "epoch": 9.07611846923828e-06,
      "model_forward_time": 5.587888240814209,
      "step": 9517
    },
    {
      "epoch": 9.07611846923828e-06,
      "step": 9517,
      "training_step_time": 17.009650468826294
    },
    {
      "epoch": 9.077072143554688e-06,
      "model_forward_time": 5.576708078384399,
      "step": 9518
    },
    {
      "epoch": 9.077072143554688e-06,
      "step": 9518,
      "training_step_time": 17.006383657455444
    },
    {
      "epoch": 9.078025817871094e-06,
      "model_forward_time": 5.580068111419678,
      "step": 9519
    },
    {
      "epoch": 9.078025817871094e-06,
      "step": 9519,
      "training_step_time": 17.01437735557556
    },
    {
      "epoch": 9.0789794921875e-06,
      "grad_norm": 0.6257936358451843,
      "learning_rate": 1e-05,
      "loss": 0.0742,
      "step": 9520
    },
    {
      "epoch": 9.0789794921875e-06,
      "model_forward_time": 5.5917768478393555,
      "step": 9520
    },
    {
      "dynamics_loss_avg": 0.06987503282725811,
      "epoch": 9.0789794921875e-06,
      "step": 9520
    },
    {
      "action_loss_avg": 0.008029148308560252,
      "epoch": 9.0789794921875e-06,
      "step": 9520
    },
    {
      "epoch": 9.0789794921875e-06,
      "step": 9520,
      "training_step_time": 17.380059480667114
    },
    {
      "epoch": 9.079933166503907e-06,
      "model_forward_time": 5.59676718711853,
      "step": 9521
    },
    {
      "epoch": 9.079933166503907e-06,
      "step": 9521,
      "training_step_time": 17.043490409851074
    },
    {
      "epoch": 9.080886840820313e-06,
      "model_forward_time": 5.607524871826172,
      "step": 9522
    },
    {
      "epoch": 9.080886840820313e-06,
      "step": 9522,
      "training_step_time": 17.083507776260376
    },
    {
      "epoch": 9.081840515136718e-06,
      "model_forward_time": 5.601658821105957,
      "step": 9523
    },
    {
      "epoch": 9.081840515136718e-06,
      "step": 9523,
      "training_step_time": 17.084923267364502
    },
    {
      "epoch": 9.082794189453124e-06,
      "model_forward_time": 5.611072778701782,
      "step": 9524
    },
    {
      "epoch": 9.082794189453124e-06,
      "step": 9524,
      "training_step_time": 17.09915018081665
    },
    {
      "epoch": 9.083747863769532e-06,
      "model_forward_time": 5.611011743545532,
      "step": 9525
    },
    {
      "epoch": 9.083747863769532e-06,
      "step": 9525,
      "training_step_time": 17.09201669692993
    },
    {
      "epoch": 9.084701538085937e-06,
      "model_forward_time": 5.949601650238037,
      "step": 9526
    },
    {
      "epoch": 9.084701538085937e-06,
      "step": 9526,
      "training_step_time": 17.422646045684814
    },
    {
      "epoch": 9.085655212402343e-06,
      "model_forward_time": 5.600485324859619,
      "step": 9527
    },
    {
      "epoch": 9.085655212402343e-06,
      "step": 9527,
      "training_step_time": 17.07863163948059
    },
    {
      "epoch": 9.08660888671875e-06,
      "model_forward_time": 5.594845294952393,
      "step": 9528
    },
    {
      "epoch": 9.08660888671875e-06,
      "step": 9528,
      "training_step_time": 17.052177667617798
    },
    {
      "epoch": 9.087562561035156e-06,
      "model_forward_time": 5.585482835769653,
      "step": 9529
    },
    {
      "epoch": 9.087562561035156e-06,
      "step": 9529,
      "training_step_time": 17.017910718917847
    },
    {
      "epoch": 9.088516235351562e-06,
      "grad_norm": 0.4302450120449066,
      "learning_rate": 1e-05,
      "loss": 0.0662,
      "step": 9530
    },
    {
      "epoch": 9.088516235351562e-06,
      "model_forward_time": 5.5965752601623535,
      "step": 9530
    },
    {
      "dynamics_loss_avg": 0.053072788286954165,
      "epoch": 9.088516235351562e-06,
      "step": 9530
    },
    {
      "action_loss_avg": 0.008106088009662926,
      "epoch": 9.088516235351562e-06,
      "step": 9530
    },
    {
      "epoch": 9.088516235351562e-06,
      "step": 9530,
      "training_step_time": 17.078341722488403
    },
    {
      "epoch": 9.08946990966797e-06,
      "model_forward_time": 6.002676963806152,
      "step": 9531
    },
    {
      "epoch": 9.08946990966797e-06,
      "step": 9531,
      "training_step_time": 17.49982261657715
    },
    {
      "epoch": 9.090423583984375e-06,
      "model_forward_time": 5.612384796142578,
      "step": 9532
    },
    {
      "epoch": 9.090423583984375e-06,
      "step": 9532,
      "training_step_time": 17.11653995513916
    },
    {
      "epoch": 9.09137725830078e-06,
      "model_forward_time": 5.608761548995972,
      "step": 9533
    },
    {
      "epoch": 9.09137725830078e-06,
      "step": 9533,
      "training_step_time": 17.08305549621582
    },
    {
      "epoch": 9.092330932617188e-06,
      "model_forward_time": 5.601439952850342,
      "step": 9534
    },
    {
      "epoch": 9.092330932617188e-06,
      "step": 9534,
      "training_step_time": 17.05708384513855
    },
    {
      "epoch": 9.093284606933594e-06,
      "model_forward_time": 5.585942029953003,
      "step": 9535
    },
    {
      "epoch": 9.093284606933594e-06,
      "step": 9535,
      "training_step_time": 16.989126682281494
    },
    {
      "epoch": 9.09423828125e-06,
      "model_forward_time": 5.578742504119873,
      "step": 9536
    },
    {
      "epoch": 9.09423828125e-06,
      "step": 9536,
      "training_step_time": 17.020798206329346
    },
    {
      "epoch": 9.095191955566407e-06,
      "model_forward_time": 5.928951740264893,
      "step": 9537
    },
    {
      "epoch": 9.095191955566407e-06,
      "step": 9537,
      "training_step_time": 17.376588821411133
    },
    {
      "epoch": 9.096145629882813e-06,
      "model_forward_time": 5.603508234024048,
      "step": 9538
    },
    {
      "epoch": 9.096145629882813e-06,
      "step": 9538,
      "training_step_time": 17.080214977264404
    },
    {
      "epoch": 9.097099304199218e-06,
      "model_forward_time": 5.60495400428772,
      "step": 9539
    },
    {
      "epoch": 9.097099304199218e-06,
      "step": 9539,
      "training_step_time": 17.090895175933838
    },
    {
      "epoch": 9.098052978515624e-06,
      "grad_norm": 1.1101495027542114,
      "learning_rate": 1e-05,
      "loss": 0.0672,
      "step": 9540
    },
    {
      "epoch": 9.098052978515624e-06,
      "model_forward_time": 5.612752914428711,
      "step": 9540
    },
    {
      "dynamics_loss_avg": 0.0654333570972085,
      "epoch": 9.098052978515624e-06,
      "step": 9540
    },
    {
      "action_loss_avg": 0.011342798359692097,
      "epoch": 9.098052978515624e-06,
      "step": 9540
    },
    {
      "epoch": 9.098052978515624e-06,
      "step": 9540,
      "training_step_time": 17.10214924812317
    },
    {
      "epoch": 9.099006652832032e-06,
      "model_forward_time": 5.6131813526153564,
      "step": 9541
    },
    {
      "epoch": 9.099006652832032e-06,
      "step": 9541,
      "training_step_time": 17.114051342010498
    },
    {
      "epoch": 9.099960327148437e-06,
      "model_forward_time": 5.979979991912842,
      "step": 9542
    },
    {
      "epoch": 9.099960327148437e-06,
      "step": 9542,
      "training_step_time": 17.447129487991333
    },
    {
      "epoch": 9.100914001464843e-06,
      "model_forward_time": 5.6025354862213135,
      "step": 9543
    },
    {
      "epoch": 9.100914001464843e-06,
      "step": 9543,
      "training_step_time": 17.072171688079834
    },
    {
      "epoch": 9.10186767578125e-06,
      "model_forward_time": 5.603367567062378,
      "step": 9544
    },
    {
      "epoch": 9.10186767578125e-06,
      "step": 9544,
      "training_step_time": 17.054754734039307
    },
    {
      "epoch": 9.102821350097656e-06,
      "model_forward_time": 5.606670379638672,
      "step": 9545
    },
    {
      "epoch": 9.102821350097656e-06,
      "step": 9545,
      "training_step_time": 17.06962823867798
    },
    {
      "epoch": 9.103775024414062e-06,
      "model_forward_time": 5.605938673019409,
      "step": 9546
    },
    {
      "epoch": 9.103775024414062e-06,
      "step": 9546,
      "training_step_time": 17.080167055130005
    },
    {
      "epoch": 9.10472869873047e-06,
      "model_forward_time": 5.61419415473938,
      "step": 9547
    },
    {
      "epoch": 9.10472869873047e-06,
      "step": 9547,
      "training_step_time": 17.108314275741577
    },
    {
      "epoch": 9.105682373046875e-06,
      "model_forward_time": 5.944591522216797,
      "step": 9548
    },
    {
      "epoch": 9.105682373046875e-06,
      "step": 9548,
      "training_step_time": 17.426790952682495
    },
    {
      "epoch": 9.10663604736328e-06,
      "model_forward_time": 5.611699342727661,
      "step": 9549
    },
    {
      "epoch": 9.10663604736328e-06,
      "step": 9549,
      "training_step_time": 17.1060688495636
    },
    {
      "epoch": 9.107589721679688e-06,
      "grad_norm": 0.6703521609306335,
      "learning_rate": 1e-05,
      "loss": 0.075,
      "step": 9550
    },
    {
      "epoch": 9.107589721679688e-06,
      "model_forward_time": 5.613495588302612,
      "step": 9550
    },
    {
      "dynamics_loss_avg": 0.06054504476487636,
      "epoch": 9.107589721679688e-06,
      "step": 9550
    },
    {
      "action_loss_avg": 0.009152209199965,
      "epoch": 9.107589721679688e-06,
      "step": 9550
    },
    {
      "epoch": 9.107589721679688e-06,
      "step": 9550,
      "training_step_time": 17.108842372894287
    },
    {
      "epoch": 9.108543395996094e-06,
      "model_forward_time": 5.605722904205322,
      "step": 9551
    },
    {
      "epoch": 9.108543395996094e-06,
      "step": 9551,
      "training_step_time": 17.09761381149292
    },
    {
      "epoch": 9.1094970703125e-06,
      "model_forward_time": 5.600074529647827,
      "step": 9552
    },
    {
      "epoch": 9.1094970703125e-06,
      "step": 9552,
      "training_step_time": 17.051090717315674
    },
    {
      "epoch": 9.110450744628907e-06,
      "model_forward_time": 5.9522411823272705,
      "step": 9553
    },
    {
      "epoch": 9.110450744628907e-06,
      "step": 9553,
      "training_step_time": 17.38983654975891
    },
    {
      "epoch": 9.111404418945313e-06,
      "model_forward_time": 5.594975233078003,
      "step": 9554
    },
    {
      "epoch": 9.111404418945313e-06,
      "step": 9554,
      "training_step_time": 17.062726259231567
    },
    {
      "epoch": 9.112358093261718e-06,
      "model_forward_time": 5.605801343917847,
      "step": 9555
    },
    {
      "epoch": 9.112358093261718e-06,
      "step": 9555,
      "training_step_time": 17.089629888534546
    },
    {
      "epoch": 9.113311767578124e-06,
      "model_forward_time": 5.609699010848999,
      "step": 9556
    },
    {
      "epoch": 9.113311767578124e-06,
      "step": 9556,
      "training_step_time": 17.11291003227234
    },
    {
      "epoch": 9.114265441894532e-06,
      "model_forward_time": 5.61940336227417,
      "step": 9557
    },
    {
      "epoch": 9.114265441894532e-06,
      "step": 9557,
      "training_step_time": 17.132725715637207
    },
    {
      "epoch": 9.115219116210937e-06,
      "model_forward_time": 5.605451345443726,
      "step": 9558
    },
    {
      "epoch": 9.115219116210937e-06,
      "step": 9558,
      "training_step_time": 17.083812713623047
    },
    {
      "epoch": 9.116172790527343e-06,
      "model_forward_time": 5.926924467086792,
      "step": 9559
    },
    {
      "epoch": 9.116172790527343e-06,
      "step": 9559,
      "training_step_time": 17.368503093719482
    },
    {
      "epoch": 9.11712646484375e-06,
      "grad_norm": 1.138792634010315,
      "learning_rate": 1e-05,
      "loss": 0.0564,
      "step": 9560
    },
    {
      "epoch": 9.11712646484375e-06,
      "model_forward_time": 5.580852270126343,
      "step": 9560
    },
    {
      "dynamics_loss_avg": 0.047864732705056666,
      "epoch": 9.11712646484375e-06,
      "step": 9560
    },
    {
      "action_loss_avg": 0.008014820830430836,
      "epoch": 9.11712646484375e-06,
      "step": 9560
    },
    {
      "epoch": 9.11712646484375e-06,
      "step": 9560,
      "training_step_time": 17.024150133132935
    },
    {
      "epoch": 9.118080139160156e-06,
      "model_forward_time": 5.5879456996917725,
      "step": 9561
    },
    {
      "epoch": 9.118080139160156e-06,
      "step": 9561,
      "training_step_time": 17.02899956703186
    },
    {
      "epoch": 9.119033813476562e-06,
      "model_forward_time": 5.582115888595581,
      "step": 9562
    },
    {
      "epoch": 9.119033813476562e-06,
      "step": 9562,
      "training_step_time": 17.029035091400146
    },
    {
      "epoch": 9.11998748779297e-06,
      "model_forward_time": 5.581738233566284,
      "step": 9563
    },
    {
      "epoch": 9.11998748779297e-06,
      "step": 9563,
      "training_step_time": 17.005231142044067
    },
    {
      "epoch": 9.120941162109375e-06,
      "model_forward_time": 5.95451283454895,
      "step": 9564
    },
    {
      "epoch": 9.120941162109375e-06,
      "step": 9564,
      "training_step_time": 17.402349710464478
    },
    {
      "epoch": 9.12189483642578e-06,
      "model_forward_time": 5.578884840011597,
      "step": 9565
    },
    {
      "epoch": 9.12189483642578e-06,
      "step": 9565,
      "training_step_time": 17.045905590057373
    },
    {
      "epoch": 9.122848510742188e-06,
      "model_forward_time": 5.60771369934082,
      "step": 9566
    },
    {
      "epoch": 9.122848510742188e-06,
      "step": 9566,
      "training_step_time": 17.103297233581543
    },
    {
      "epoch": 9.123802185058594e-06,
      "model_forward_time": 5.632975339889526,
      "step": 9567
    },
    {
      "epoch": 9.123802185058594e-06,
      "step": 9567,
      "training_step_time": 17.175097227096558
    },
    {
      "epoch": 9.124755859375e-06,
      "model_forward_time": 5.630192041397095,
      "step": 9568
    },
    {
      "epoch": 9.124755859375e-06,
      "step": 9568,
      "training_step_time": 17.134273767471313
    },
    {
      "epoch": 9.125709533691407e-06,
      "model_forward_time": 5.987034797668457,
      "step": 9569
    },
    {
      "epoch": 9.125709533691407e-06,
      "step": 9569,
      "training_step_time": 17.470628261566162
    },
    {
      "epoch": 9.126663208007813e-06,
      "grad_norm": 1.2818244695663452,
      "learning_rate": 1e-05,
      "loss": 0.0765,
      "step": 9570
    },
    {
      "epoch": 9.126663208007813e-06,
      "model_forward_time": 5.601464748382568,
      "step": 9570
    },
    {
      "dynamics_loss_avg": 0.06696464084088802,
      "epoch": 9.126663208007813e-06,
      "step": 9570
    },
    {
      "action_loss_avg": 0.0099295053165406,
      "epoch": 9.126663208007813e-06,
      "step": 9570
    },
    {
      "epoch": 9.126663208007813e-06,
      "step": 9570,
      "training_step_time": 17.079832553863525
    },
    {
      "epoch": 9.127616882324218e-06,
      "model_forward_time": 5.6006739139556885,
      "step": 9571
    },
    {
      "epoch": 9.127616882324218e-06,
      "step": 9571,
      "training_step_time": 17.056519746780396
    },
    {
      "epoch": 9.128570556640624e-06,
      "model_forward_time": 5.582908630371094,
      "step": 9572
    },
    {
      "epoch": 9.128570556640624e-06,
      "step": 9572,
      "training_step_time": 17.0245041847229
    },
    {
      "epoch": 9.129524230957032e-06,
      "model_forward_time": 5.591946840286255,
      "step": 9573
    },
    {
      "epoch": 9.129524230957032e-06,
      "step": 9573,
      "training_step_time": 17.02225089073181
    },
    {
      "epoch": 9.130477905273437e-06,
      "model_forward_time": 5.591514825820923,
      "step": 9574
    },
    {
      "epoch": 9.130477905273437e-06,
      "step": 9574,
      "training_step_time": 17.040903091430664
    },
    {
      "epoch": 9.131431579589843e-06,
      "model_forward_time": 5.957860946655273,
      "step": 9575
    },
    {
      "epoch": 9.131431579589843e-06,
      "step": 9575,
      "training_step_time": 17.461820602416992
    },
    {
      "epoch": 9.13238525390625e-06,
      "model_forward_time": 5.626554489135742,
      "step": 9576
    },
    {
      "epoch": 9.13238525390625e-06,
      "step": 9576,
      "training_step_time": 17.14906334877014
    },
    {
      "epoch": 9.133338928222656e-06,
      "model_forward_time": 5.626570463180542,
      "step": 9577
    },
    {
      "epoch": 9.133338928222656e-06,
      "step": 9577,
      "training_step_time": 17.113744974136353
    },
    {
      "epoch": 9.134292602539062e-06,
      "model_forward_time": 5.591336011886597,
      "step": 9578
    },
    {
      "epoch": 9.134292602539062e-06,
      "step": 9578,
      "training_step_time": 17.01852297782898
    },
    {
      "epoch": 9.13524627685547e-06,
      "model_forward_time": 5.5769712924957275,
      "step": 9579
    },
    {
      "epoch": 9.13524627685547e-06,
      "step": 9579,
      "training_step_time": 17.021901607513428
    },
    {
      "epoch": 9.136199951171875e-06,
      "grad_norm": 0.655013382434845,
      "learning_rate": 1e-05,
      "loss": 0.054,
      "step": 9580
    },
    {
      "epoch": 9.136199951171875e-06,
      "model_forward_time": 5.594043016433716,
      "step": 9580
    },
    {
      "dynamics_loss_avg": 0.04532187413424253,
      "epoch": 9.136199951171875e-06,
      "step": 9580
    },
    {
      "action_loss_avg": 0.007281410531140864,
      "epoch": 9.136199951171875e-06,
      "step": 9580
    },
    {
      "epoch": 9.136199951171875e-06,
      "step": 9580,
      "training_step_time": 17.405576467514038
    },
    {
      "epoch": 9.13715362548828e-06,
      "model_forward_time": 5.605814218521118,
      "step": 9581
    },
    {
      "epoch": 9.13715362548828e-06,
      "step": 9581,
      "training_step_time": 17.067500591278076
    },
    {
      "epoch": 9.138107299804688e-06,
      "model_forward_time": 5.591525077819824,
      "step": 9582
    },
    {
      "epoch": 9.138107299804688e-06,
      "step": 9582,
      "training_step_time": 17.03032612800598
    },
    {
      "epoch": 9.139060974121094e-06,
      "model_forward_time": 5.59021782875061,
      "step": 9583
    },
    {
      "epoch": 9.139060974121094e-06,
      "step": 9583,
      "training_step_time": 17.02896785736084
    },
    {
      "epoch": 9.1400146484375e-06,
      "model_forward_time": 5.5956971645355225,
      "step": 9584
    },
    {
      "epoch": 9.1400146484375e-06,
      "step": 9584,
      "training_step_time": 17.074897050857544
    },
    {
      "epoch": 9.140968322753907e-06,
      "model_forward_time": 5.606444835662842,
      "step": 9585
    },
    {
      "epoch": 9.140968322753907e-06,
      "step": 9585,
      "training_step_time": 17.08435583114624
    },
    {
      "epoch": 9.141921997070313e-06,
      "model_forward_time": 5.98034930229187,
      "step": 9586
    },
    {
      "epoch": 9.141921997070313e-06,
      "step": 9586,
      "training_step_time": 17.428804636001587
    },
    {
      "epoch": 9.142875671386719e-06,
      "model_forward_time": 5.582290887832642,
      "step": 9587
    },
    {
      "epoch": 9.142875671386719e-06,
      "step": 9587,
      "training_step_time": 17.0097177028656
    },
    {
      "epoch": 9.143829345703124e-06,
      "model_forward_time": 5.595897197723389,
      "step": 9588
    },
    {
      "epoch": 9.143829345703124e-06,
      "step": 9588,
      "training_step_time": 17.050097227096558
    },
    {
      "epoch": 9.144783020019532e-06,
      "model_forward_time": 5.5959107875823975,
      "step": 9589
    },
    {
      "epoch": 9.144783020019532e-06,
      "step": 9589,
      "training_step_time": 17.03411602973938
    },
    {
      "epoch": 9.145736694335937e-06,
      "grad_norm": 0.5779603719711304,
      "learning_rate": 1e-05,
      "loss": 0.0606,
      "step": 9590
    },
    {
      "epoch": 9.145736694335937e-06,
      "model_forward_time": 5.59014105796814,
      "step": 9590
    },
    {
      "dynamics_loss_avg": 0.05510582104325294,
      "epoch": 9.145736694335937e-06,
      "step": 9590
    },
    {
      "action_loss_avg": 0.00827245502732694,
      "epoch": 9.145736694335937e-06,
      "step": 9590
    },
    {
      "epoch": 9.145736694335937e-06,
      "step": 9590,
      "training_step_time": 17.03382682800293
    },
    {
      "epoch": 9.146690368652343e-06,
      "model_forward_time": 5.613158226013184,
      "step": 9591
    },
    {
      "epoch": 9.146690368652343e-06,
      "step": 9591,
      "training_step_time": 17.46981143951416
    },
    {
      "epoch": 9.14764404296875e-06,
      "model_forward_time": 5.605199575424194,
      "step": 9592
    },
    {
      "epoch": 9.14764404296875e-06,
      "step": 9592,
      "training_step_time": 17.088902711868286
    },
    {
      "epoch": 9.148597717285156e-06,
      "model_forward_time": 5.62406325340271,
      "step": 9593
    },
    {
      "epoch": 9.148597717285156e-06,
      "step": 9593,
      "training_step_time": 17.096294164657593
    },
    {
      "epoch": 9.149551391601562e-06,
      "model_forward_time": 5.602445363998413,
      "step": 9594
    },
    {
      "epoch": 9.149551391601562e-06,
      "step": 9594,
      "training_step_time": 17.05028200149536
    },
    {
      "epoch": 9.15050506591797e-06,
      "model_forward_time": 5.596772193908691,
      "step": 9595
    },
    {
      "epoch": 9.15050506591797e-06,
      "step": 9595,
      "training_step_time": 17.039780855178833
    },
    {
      "epoch": 9.151458740234375e-06,
      "model_forward_time": 5.587062835693359,
      "step": 9596
    },
    {
      "epoch": 9.151458740234375e-06,
      "step": 9596,
      "training_step_time": 17.008829593658447
    },
    {
      "epoch": 9.15241241455078e-06,
      "model_forward_time": 5.958426475524902,
      "step": 9597
    },
    {
      "epoch": 9.15241241455078e-06,
      "step": 9597,
      "training_step_time": 17.409239530563354
    },
    {
      "epoch": 9.153366088867188e-06,
      "model_forward_time": 5.596269369125366,
      "step": 9598
    },
    {
      "epoch": 9.153366088867188e-06,
      "step": 9598,
      "training_step_time": 17.064165592193604
    },
    {
      "epoch": 9.154319763183594e-06,
      "model_forward_time": 5.607029914855957,
      "step": 9599
    },
    {
      "epoch": 9.154319763183594e-06,
      "step": 9599,
      "training_step_time": 17.1122567653656
    },
    {
      "epoch": 9.1552734375e-06,
      "grad_norm": 0.5649617314338684,
      "learning_rate": 1e-05,
      "loss": 0.0653,
      "step": 9600
    },
    {
      "epoch": 9.1552734375e-06,
      "model_forward_time": 5.623064756393433,
      "step": 9600
    },
    {
      "dynamics_loss_avg": 0.05631077326834202,
      "epoch": 9.1552734375e-06,
      "step": 9600
    },
    {
      "action_loss_avg": 0.00778955752030015,
      "epoch": 9.1552734375e-06,
      "step": 9600
    },
    {
      "epoch": 9.1552734375e-06,
      "step": 9600,
      "training_step_time": 17.126992225646973
    },
    {
      "epoch": 9.156227111816407e-06,
      "model_forward_time": 5.619083642959595,
      "step": 9601
    },
    {
      "epoch": 9.156227111816407e-06,
      "step": 9601,
      "training_step_time": 17.114359617233276
    },
    {
      "epoch": 9.157180786132813e-06,
      "model_forward_time": 5.958474397659302,
      "step": 9602
    },
    {
      "epoch": 9.157180786132813e-06,
      "step": 9602,
      "training_step_time": 17.45877957344055
    },
    {
      "epoch": 9.158134460449219e-06,
      "model_forward_time": 5.604273796081543,
      "step": 9603
    },
    {
      "epoch": 9.158134460449219e-06,
      "step": 9603,
      "training_step_time": 17.058279037475586
    },
    {
      "epoch": 9.159088134765624e-06,
      "model_forward_time": 5.592045545578003,
      "step": 9604
    },
    {
      "epoch": 9.159088134765624e-06,
      "step": 9604,
      "training_step_time": 17.03791356086731
    },
    {
      "epoch": 9.160041809082032e-06,
      "model_forward_time": 5.587821960449219,
      "step": 9605
    },
    {
      "epoch": 9.160041809082032e-06,
      "step": 9605,
      "training_step_time": 17.01474142074585
    },
    {
      "epoch": 9.160995483398437e-06,
      "model_forward_time": 5.584683179855347,
      "step": 9606
    },
    {
      "epoch": 9.160995483398437e-06,
      "step": 9606,
      "training_step_time": 17.03372573852539
    },
    {
      "epoch": 9.161949157714843e-06,
      "model_forward_time": 5.5988428592681885,
      "step": 9607
    },
    {
      "epoch": 9.161949157714843e-06,
      "step": 9607,
      "training_step_time": 17.081369876861572
    },
    {
      "epoch": 9.16290283203125e-06,
      "model_forward_time": 5.978825807571411,
      "step": 9608
    },
    {
      "epoch": 9.16290283203125e-06,
      "step": 9608,
      "training_step_time": 17.457752466201782
    },
    {
      "epoch": 9.163856506347656e-06,
      "model_forward_time": 5.598361492156982,
      "step": 9609
    },
    {
      "epoch": 9.163856506347656e-06,
      "step": 9609,
      "training_step_time": 17.037501096725464
    },
    {
      "epoch": 9.164810180664062e-06,
      "grad_norm": 1.4039862155914307,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 9610
    },
    {
      "epoch": 9.164810180664062e-06,
      "model_forward_time": 5.571507930755615,
      "step": 9610
    },
    {
      "dynamics_loss_avg": 0.07172717712819576,
      "epoch": 9.164810180664062e-06,
      "step": 9610
    },
    {
      "action_loss_avg": 0.01084206304512918,
      "epoch": 9.164810180664062e-06,
      "step": 9610
    },
    {
      "epoch": 9.164810180664062e-06,
      "step": 9610,
      "training_step_time": 16.998969316482544
    },
    {
      "epoch": 9.16576385498047e-06,
      "model_forward_time": 5.632727861404419,
      "step": 9611
    },
    {
      "epoch": 9.16576385498047e-06,
      "step": 9611,
      "training_step_time": 17.09574580192566
    },
    {
      "epoch": 9.166717529296875e-06,
      "model_forward_time": 5.63100266456604,
      "step": 9612
    },
    {
      "epoch": 9.166717529296875e-06,
      "step": 9612,
      "training_step_time": 17.128097534179688
    },
    {
      "epoch": 9.16767120361328e-06,
      "model_forward_time": 5.616873264312744,
      "step": 9613
    },
    {
      "epoch": 9.16767120361328e-06,
      "step": 9613,
      "training_step_time": 17.456384420394897
    },
    {
      "epoch": 9.168624877929688e-06,
      "model_forward_time": 5.625874280929565,
      "step": 9614
    },
    {
      "epoch": 9.168624877929688e-06,
      "step": 9614,
      "training_step_time": 17.126986742019653
    },
    {
      "epoch": 9.169578552246094e-06,
      "model_forward_time": 5.60615611076355,
      "step": 9615
    },
    {
      "epoch": 9.169578552246094e-06,
      "step": 9615,
      "training_step_time": 17.106794357299805
    },
    {
      "epoch": 9.1705322265625e-06,
      "model_forward_time": 5.609956741333008,
      "step": 9616
    },
    {
      "epoch": 9.1705322265625e-06,
      "step": 9616,
      "training_step_time": 17.079389572143555
    },
    {
      "epoch": 9.171485900878907e-06,
      "model_forward_time": 5.600188255310059,
      "step": 9617
    },
    {
      "epoch": 9.171485900878907e-06,
      "step": 9617,
      "training_step_time": 17.053278923034668
    },
    {
      "epoch": 9.172439575195313e-06,
      "model_forward_time": 5.579327344894409,
      "step": 9618
    },
    {
      "epoch": 9.172439575195313e-06,
      "step": 9618,
      "training_step_time": 16.99232578277588
    },
    {
      "epoch": 9.173393249511719e-06,
      "model_forward_time": 5.987982749938965,
      "step": 9619
    },
    {
      "epoch": 9.173393249511719e-06,
      "step": 9619,
      "training_step_time": 17.413573265075684
    },
    {
      "epoch": 9.174346923828124e-06,
      "grad_norm": 1.011237382888794,
      "learning_rate": 1e-05,
      "loss": 0.0724,
      "step": 9620
    },
    {
      "epoch": 9.174346923828124e-06,
      "model_forward_time": 5.592291593551636,
      "step": 9620
    },
    {
      "dynamics_loss_avg": 0.060919692181050776,
      "epoch": 9.174346923828124e-06,
      "step": 9620
    },
    {
      "action_loss_avg": 0.0076612226199358705,
      "epoch": 9.174346923828124e-06,
      "step": 9620
    },
    {
      "epoch": 9.174346923828124e-06,
      "step": 9620,
      "training_step_time": 17.057271718978882
    },
    {
      "epoch": 9.175300598144532e-06,
      "model_forward_time": 5.594851016998291,
      "step": 9621
    },
    {
      "epoch": 9.175300598144532e-06,
      "step": 9621,
      "training_step_time": 17.038552284240723
    },
    {
      "epoch": 9.176254272460937e-06,
      "model_forward_time": 5.592034101486206,
      "step": 9622
    },
    {
      "epoch": 9.176254272460937e-06,
      "step": 9622,
      "training_step_time": 17.055948734283447
    },
    {
      "epoch": 9.177207946777343e-06,
      "model_forward_time": 5.606413841247559,
      "step": 9623
    },
    {
      "epoch": 9.177207946777343e-06,
      "step": 9623,
      "training_step_time": 17.099284172058105
    },
    {
      "epoch": 9.17816162109375e-06,
      "model_forward_time": 5.659554958343506,
      "step": 9624
    },
    {
      "epoch": 9.17816162109375e-06,
      "step": 9624,
      "training_step_time": 17.51529049873352
    },
    {
      "epoch": 9.179115295410156e-06,
      "model_forward_time": 5.616344213485718,
      "step": 9625
    },
    {
      "epoch": 9.179115295410156e-06,
      "step": 9625,
      "training_step_time": 17.115767240524292
    },
    {
      "epoch": 9.180068969726562e-06,
      "model_forward_time": 5.610012531280518,
      "step": 9626
    },
    {
      "epoch": 9.180068969726562e-06,
      "step": 9626,
      "training_step_time": 17.114552974700928
    },
    {
      "epoch": 9.18102264404297e-06,
      "model_forward_time": 5.6217710971832275,
      "step": 9627
    },
    {
      "epoch": 9.18102264404297e-06,
      "step": 9627,
      "training_step_time": 17.111933946609497
    },
    {
      "epoch": 9.181976318359375e-06,
      "model_forward_time": 5.610073804855347,
      "step": 9628
    },
    {
      "epoch": 9.181976318359375e-06,
      "step": 9628,
      "training_step_time": 17.07225513458252
    },
    {
      "epoch": 9.18292999267578e-06,
      "model_forward_time": 5.608214855194092,
      "step": 9629
    },
    {
      "epoch": 9.18292999267578e-06,
      "step": 9629,
      "training_step_time": 17.09322690963745
    },
    {
      "epoch": 9.183883666992188e-06,
      "grad_norm": 0.6038967967033386,
      "learning_rate": 1e-05,
      "loss": 0.0719,
      "step": 9630
    },
    {
      "epoch": 9.183883666992188e-06,
      "model_forward_time": 5.971256256103516,
      "step": 9630
    },
    {
      "dynamics_loss_avg": 0.06596313100308179,
      "epoch": 9.183883666992188e-06,
      "step": 9630
    },
    {
      "action_loss_avg": 0.008285801112651824,
      "epoch": 9.183883666992188e-06,
      "step": 9630
    },
    {
      "epoch": 9.183883666992188e-06,
      "step": 9630,
      "training_step_time": 17.46862244606018
    },
    {
      "epoch": 9.184837341308594e-06,
      "model_forward_time": 5.613118886947632,
      "step": 9631
    },
    {
      "epoch": 9.184837341308594e-06,
      "step": 9631,
      "training_step_time": 17.11343026161194
    },
    {
      "epoch": 9.185791015625e-06,
      "model_forward_time": 5.616575717926025,
      "step": 9632
    },
    {
      "epoch": 9.185791015625e-06,
      "step": 9632,
      "training_step_time": 17.090018033981323
    },
    {
      "epoch": 9.186744689941405e-06,
      "model_forward_time": 5.607885837554932,
      "step": 9633
    },
    {
      "epoch": 9.186744689941405e-06,
      "step": 9633,
      "training_step_time": 17.068649768829346
    },
    {
      "epoch": 9.187698364257813e-06,
      "model_forward_time": 5.594614505767822,
      "step": 9634
    },
    {
      "epoch": 9.187698364257813e-06,
      "step": 9634,
      "training_step_time": 17.00519347190857
    },
    {
      "epoch": 9.188652038574219e-06,
      "model_forward_time": 5.940102577209473,
      "step": 9635
    },
    {
      "epoch": 9.188652038574219e-06,
      "step": 9635,
      "training_step_time": 17.368295669555664
    },
    {
      "epoch": 9.189605712890624e-06,
      "model_forward_time": 5.589000940322876,
      "step": 9636
    },
    {
      "epoch": 9.189605712890624e-06,
      "step": 9636,
      "training_step_time": 17.03279685974121
    },
    {
      "epoch": 9.190559387207032e-06,
      "model_forward_time": 5.601245403289795,
      "step": 9637
    },
    {
      "epoch": 9.190559387207032e-06,
      "step": 9637,
      "training_step_time": 17.077301263809204
    },
    {
      "epoch": 9.191513061523437e-06,
      "model_forward_time": 5.605145454406738,
      "step": 9638
    },
    {
      "epoch": 9.191513061523437e-06,
      "step": 9638,
      "training_step_time": 17.094993591308594
    },
    {
      "epoch": 9.192466735839843e-06,
      "model_forward_time": 5.612997770309448,
      "step": 9639
    },
    {
      "epoch": 9.192466735839843e-06,
      "step": 9639,
      "training_step_time": 17.121618270874023
    },
    {
      "epoch": 9.19342041015625e-06,
      "grad_norm": 1.3208377361297607,
      "learning_rate": 1e-05,
      "loss": 0.0683,
      "step": 9640
    },
    {
      "epoch": 9.19342041015625e-06,
      "model_forward_time": 5.62008810043335,
      "step": 9640
    },
    {
      "dynamics_loss_avg": 0.05550439991056919,
      "epoch": 9.19342041015625e-06,
      "step": 9640
    },
    {
      "action_loss_avg": 0.007296191831119359,
      "epoch": 9.19342041015625e-06,
      "step": 9640
    },
    {
      "epoch": 9.19342041015625e-06,
      "step": 9640,
      "training_step_time": 17.121533155441284
    },
    {
      "epoch": 9.194374084472656e-06,
      "model_forward_time": 6.0047338008880615,
      "step": 9641
    },
    {
      "epoch": 9.194374084472656e-06,
      "step": 9641,
      "training_step_time": 17.492632389068604
    },
    {
      "epoch": 9.195327758789062e-06,
      "model_forward_time": 5.602564096450806,
      "step": 9642
    },
    {
      "epoch": 9.195327758789062e-06,
      "step": 9642,
      "training_step_time": 17.088414430618286
    },
    {
      "epoch": 9.19628143310547e-06,
      "model_forward_time": 5.62164044380188,
      "step": 9643
    },
    {
      "epoch": 9.19628143310547e-06,
      "step": 9643,
      "training_step_time": 17.106101751327515
    },
    {
      "epoch": 9.197235107421875e-06,
      "model_forward_time": 5.611017227172852,
      "step": 9644
    },
    {
      "epoch": 9.197235107421875e-06,
      "step": 9644,
      "training_step_time": 17.08704972267151
    },
    {
      "epoch": 9.19818878173828e-06,
      "model_forward_time": 5.5971128940582275,
      "step": 9645
    },
    {
      "epoch": 9.19818878173828e-06,
      "step": 9645,
      "training_step_time": 17.066972494125366
    },
    {
      "epoch": 9.199142456054688e-06,
      "model_forward_time": 5.602831840515137,
      "step": 9646
    },
    {
      "epoch": 9.199142456054688e-06,
      "step": 9646,
      "training_step_time": 17.460456609725952
    },
    {
      "epoch": 9.200096130371094e-06,
      "model_forward_time": 5.598225355148315,
      "step": 9647
    },
    {
      "epoch": 9.200096130371094e-06,
      "step": 9647,
      "training_step_time": 17.045825481414795
    },
    {
      "epoch": 9.2010498046875e-06,
      "model_forward_time": 5.597710609436035,
      "step": 9648
    },
    {
      "epoch": 9.2010498046875e-06,
      "step": 9648,
      "training_step_time": 17.067121028900146
    },
    {
      "epoch": 9.202003479003905e-06,
      "model_forward_time": 5.592405557632446,
      "step": 9649
    },
    {
      "epoch": 9.202003479003905e-06,
      "step": 9649,
      "training_step_time": 17.045410633087158
    },
    {
      "epoch": 9.202957153320313e-06,
      "grad_norm": 0.39833012223243713,
      "learning_rate": 1e-05,
      "loss": 0.0538,
      "step": 9650
    },
    {
      "epoch": 9.202957153320313e-06,
      "model_forward_time": 5.598398685455322,
      "step": 9650
    },
    {
      "dynamics_loss_avg": 0.04920893032103777,
      "epoch": 9.202957153320313e-06,
      "step": 9650
    },
    {
      "action_loss_avg": 0.006556953815743327,
      "epoch": 9.202957153320313e-06,
      "step": 9650
    },
    {
      "epoch": 9.202957153320313e-06,
      "step": 9650,
      "training_step_time": 17.0774986743927
    },
    {
      "epoch": 9.203910827636719e-06,
      "model_forward_time": 5.602334022521973,
      "step": 9651
    },
    {
      "epoch": 9.203910827636719e-06,
      "step": 9651,
      "training_step_time": 17.056032180786133
    },
    {
      "epoch": 9.204864501953124e-06,
      "model_forward_time": 5.974912405014038,
      "step": 9652
    },
    {
      "epoch": 9.204864501953124e-06,
      "step": 9652,
      "training_step_time": 17.445289850234985
    },
    {
      "epoch": 9.205818176269532e-06,
      "model_forward_time": 5.603367805480957,
      "step": 9653
    },
    {
      "epoch": 9.205818176269532e-06,
      "step": 9653,
      "training_step_time": 17.069098949432373
    },
    {
      "epoch": 9.206771850585937e-06,
      "model_forward_time": 5.6000401973724365,
      "step": 9654
    },
    {
      "epoch": 9.206771850585937e-06,
      "step": 9654,
      "training_step_time": 17.048906564712524
    },
    {
      "epoch": 9.207725524902343e-06,
      "model_forward_time": 5.595957279205322,
      "step": 9655
    },
    {
      "epoch": 9.207725524902343e-06,
      "step": 9655,
      "training_step_time": 17.023799419403076
    },
    {
      "epoch": 9.20867919921875e-06,
      "model_forward_time": 5.592177391052246,
      "step": 9656
    },
    {
      "epoch": 9.20867919921875e-06,
      "step": 9656,
      "training_step_time": 17.02144193649292
    },
    {
      "epoch": 9.209632873535156e-06,
      "model_forward_time": 5.584100961685181,
      "step": 9657
    },
    {
      "epoch": 9.209632873535156e-06,
      "step": 9657,
      "training_step_time": 17.38640785217285
    },
    {
      "epoch": 9.210586547851562e-06,
      "model_forward_time": 5.58469033241272,
      "step": 9658
    },
    {
      "epoch": 9.210586547851562e-06,
      "step": 9658,
      "training_step_time": 17.0304913520813
    },
    {
      "epoch": 9.21154022216797e-06,
      "model_forward_time": 5.594184160232544,
      "step": 9659
    },
    {
      "epoch": 9.21154022216797e-06,
      "step": 9659,
      "training_step_time": 17.041781187057495
    },
    {
      "epoch": 9.212493896484375e-06,
      "grad_norm": 0.6966766715049744,
      "learning_rate": 1e-05,
      "loss": 0.0851,
      "step": 9660
    },
    {
      "epoch": 9.212493896484375e-06,
      "model_forward_time": 5.599835634231567,
      "step": 9660
    },
    {
      "dynamics_loss_avg": 0.07595601640641689,
      "epoch": 9.212493896484375e-06,
      "step": 9660
    },
    {
      "action_loss_avg": 0.010582342348061501,
      "epoch": 9.212493896484375e-06,
      "step": 9660
    },
    {
      "epoch": 9.212493896484375e-06,
      "step": 9660,
      "training_step_time": 17.070311784744263
    },
    {
      "epoch": 9.213447570800781e-06,
      "model_forward_time": 5.606481552124023,
      "step": 9661
    },
    {
      "epoch": 9.213447570800781e-06,
      "step": 9661,
      "training_step_time": 17.06661558151245
    },
    {
      "epoch": 9.214401245117188e-06,
      "model_forward_time": 5.595325946807861,
      "step": 9662
    },
    {
      "epoch": 9.214401245117188e-06,
      "step": 9662,
      "training_step_time": 17.05858039855957
    },
    {
      "epoch": 9.215354919433594e-06,
      "model_forward_time": 5.975146532058716,
      "step": 9663
    },
    {
      "epoch": 9.215354919433594e-06,
      "step": 9663,
      "training_step_time": 17.443117380142212
    },
    {
      "epoch": 9.21630859375e-06,
      "model_forward_time": 5.623010635375977,
      "step": 9664
    },
    {
      "epoch": 9.21630859375e-06,
      "step": 9664,
      "training_step_time": 17.117825269699097
    },
    {
      "epoch": 9.217262268066405e-06,
      "model_forward_time": 5.608517646789551,
      "step": 9665
    },
    {
      "epoch": 9.217262268066405e-06,
      "step": 9665,
      "training_step_time": 17.088292598724365
    },
    {
      "epoch": 9.218215942382813e-06,
      "model_forward_time": 5.6111836433410645,
      "step": 9666
    },
    {
      "epoch": 9.218215942382813e-06,
      "step": 9666,
      "training_step_time": 17.0966579914093
    },
    {
      "epoch": 9.219169616699219e-06,
      "model_forward_time": 5.599795818328857,
      "step": 9667
    },
    {
      "epoch": 9.219169616699219e-06,
      "step": 9667,
      "training_step_time": 17.080116987228394
    },
    {
      "epoch": 9.220123291015624e-06,
      "model_forward_time": 5.958066701889038,
      "step": 9668
    },
    {
      "epoch": 9.220123291015624e-06,
      "step": 9668,
      "training_step_time": 17.411872148513794
    },
    {
      "epoch": 9.221076965332032e-06,
      "model_forward_time": 5.580348968505859,
      "step": 9669
    },
    {
      "epoch": 9.221076965332032e-06,
      "step": 9669,
      "training_step_time": 17.000319957733154
    },
    {
      "epoch": 9.222030639648437e-06,
      "grad_norm": 0.9375202059745789,
      "learning_rate": 1e-05,
      "loss": 0.0561,
      "step": 9670
    },
    {
      "epoch": 9.222030639648437e-06,
      "model_forward_time": 5.575996160507202,
      "step": 9670
    },
    {
      "dynamics_loss_avg": 0.049926386401057246,
      "epoch": 9.222030639648437e-06,
      "step": 9670
    },
    {
      "action_loss_avg": 0.008964452799409628,
      "epoch": 9.222030639648437e-06,
      "step": 9670
    },
    {
      "epoch": 9.222030639648437e-06,
      "step": 9670,
      "training_step_time": 17.012794733047485
    },
    {
      "epoch": 9.222984313964843e-06,
      "model_forward_time": 5.585649490356445,
      "step": 9671
    },
    {
      "epoch": 9.222984313964843e-06,
      "step": 9671,
      "training_step_time": 17.022969722747803
    },
    {
      "epoch": 9.22393798828125e-06,
      "model_forward_time": 5.584665775299072,
      "step": 9672
    },
    {
      "epoch": 9.22393798828125e-06,
      "step": 9672,
      "training_step_time": 17.00395703315735
    },
    {
      "epoch": 9.224891662597656e-06,
      "model_forward_time": 5.579994201660156,
      "step": 9673
    },
    {
      "epoch": 9.224891662597656e-06,
      "step": 9673,
      "training_step_time": 17.038662672042847
    },
    {
      "epoch": 9.225845336914062e-06,
      "model_forward_time": 6.0037102699279785,
      "step": 9674
    },
    {
      "epoch": 9.225845336914062e-06,
      "step": 9674,
      "training_step_time": 17.493717908859253
    },
    {
      "epoch": 9.22679901123047e-06,
      "model_forward_time": 5.5946760177612305,
      "step": 9675
    },
    {
      "epoch": 9.22679901123047e-06,
      "step": 9675,
      "training_step_time": 17.068435668945312
    },
    {
      "epoch": 9.227752685546875e-06,
      "model_forward_time": 5.598534345626831,
      "step": 9676
    },
    {
      "epoch": 9.227752685546875e-06,
      "step": 9676,
      "training_step_time": 17.057138204574585
    },
    {
      "epoch": 9.228706359863281e-06,
      "model_forward_time": 5.588959455490112,
      "step": 9677
    },
    {
      "epoch": 9.228706359863281e-06,
      "step": 9677,
      "training_step_time": 17.045587062835693
    },
    {
      "epoch": 9.229660034179688e-06,
      "model_forward_time": 5.591579437255859,
      "step": 9678
    },
    {
      "epoch": 9.229660034179688e-06,
      "step": 9678,
      "training_step_time": 17.055105924606323
    },
    {
      "epoch": 9.230613708496094e-06,
      "model_forward_time": 5.605682849884033,
      "step": 9679
    },
    {
      "epoch": 9.230613708496094e-06,
      "step": 9679,
      "training_step_time": 17.4585018157959
    },
    {
      "epoch": 9.2315673828125e-06,
      "grad_norm": 0.4806993007659912,
      "learning_rate": 1e-05,
      "loss": 0.0692,
      "step": 9680
    },
    {
      "epoch": 9.2315673828125e-06,
      "model_forward_time": 5.602515697479248,
      "step": 9680
    },
    {
      "dynamics_loss_avg": 0.060345901548862456,
      "epoch": 9.2315673828125e-06,
      "step": 9680
    },
    {
      "action_loss_avg": 0.007998808915726841,
      "epoch": 9.2315673828125e-06,
      "step": 9680
    },
    {
      "epoch": 9.2315673828125e-06,
      "step": 9680,
      "training_step_time": 17.083086729049683
    },
    {
      "epoch": 9.232521057128905e-06,
      "model_forward_time": 5.611965179443359,
      "step": 9681
    },
    {
      "epoch": 9.232521057128905e-06,
      "step": 9681,
      "training_step_time": 17.100431203842163
    },
    {
      "epoch": 9.233474731445313e-06,
      "model_forward_time": 5.615141868591309,
      "step": 9682
    },
    {
      "epoch": 9.233474731445313e-06,
      "step": 9682,
      "training_step_time": 17.0911865234375
    },
    {
      "epoch": 9.234428405761719e-06,
      "model_forward_time": 5.614302158355713,
      "step": 9683
    },
    {
      "epoch": 9.234428405761719e-06,
      "step": 9683,
      "training_step_time": 17.12627863883972
    },
    {
      "epoch": 9.235382080078124e-06,
      "model_forward_time": 5.619929552078247,
      "step": 9684
    },
    {
      "epoch": 9.235382080078124e-06,
      "step": 9684,
      "training_step_time": 17.09321689605713
    },
    {
      "epoch": 9.236335754394532e-06,
      "model_forward_time": 5.975606203079224,
      "step": 9685
    },
    {
      "epoch": 9.236335754394532e-06,
      "step": 9685,
      "training_step_time": 17.405911684036255
    },
    {
      "epoch": 9.237289428710937e-06,
      "model_forward_time": 5.582351922988892,
      "step": 9686
    },
    {
      "epoch": 9.237289428710937e-06,
      "step": 9686,
      "training_step_time": 17.02483892440796
    },
    {
      "epoch": 9.238243103027343e-06,
      "model_forward_time": 5.609161138534546,
      "step": 9687
    },
    {
      "epoch": 9.238243103027343e-06,
      "step": 9687,
      "training_step_time": 17.079235553741455
    },
    {
      "epoch": 9.23919677734375e-06,
      "model_forward_time": 5.60674262046814,
      "step": 9688
    },
    {
      "epoch": 9.23919677734375e-06,
      "step": 9688,
      "training_step_time": 17.086776971817017
    },
    {
      "epoch": 9.240150451660156e-06,
      "model_forward_time": 5.603432655334473,
      "step": 9689
    },
    {
      "epoch": 9.240150451660156e-06,
      "step": 9689,
      "training_step_time": 17.07826018333435
    },
    {
      "epoch": 9.241104125976562e-06,
      "grad_norm": 0.7212816476821899,
      "learning_rate": 1e-05,
      "loss": 0.0685,
      "step": 9690
    },
    {
      "epoch": 9.241104125976562e-06,
      "model_forward_time": 5.598320245742798,
      "step": 9690
    },
    {
      "dynamics_loss_avg": 0.05947158876806498,
      "epoch": 9.241104125976562e-06,
      "step": 9690
    },
    {
      "action_loss_avg": 0.009595606522634626,
      "epoch": 9.241104125976562e-06,
      "step": 9690
    },
    {
      "epoch": 9.241104125976562e-06,
      "step": 9690,
      "training_step_time": 17.431577682495117
    },
    {
      "epoch": 9.24205780029297e-06,
      "model_forward_time": 5.5999462604522705,
      "step": 9691
    },
    {
      "epoch": 9.24205780029297e-06,
      "step": 9691,
      "training_step_time": 17.078052520751953
    },
    {
      "epoch": 9.243011474609375e-06,
      "model_forward_time": 5.608192682266235,
      "step": 9692
    },
    {
      "epoch": 9.243011474609375e-06,
      "step": 9692,
      "training_step_time": 17.088376760482788
    },
    {
      "epoch": 9.243965148925781e-06,
      "model_forward_time": 5.608597993850708,
      "step": 9693
    },
    {
      "epoch": 9.243965148925781e-06,
      "step": 9693,
      "training_step_time": 17.081467390060425
    },
    {
      "epoch": 9.244918823242188e-06,
      "model_forward_time": 5.604156732559204,
      "step": 9694
    },
    {
      "epoch": 9.244918823242188e-06,
      "step": 9694,
      "training_step_time": 17.070961475372314
    },
    {
      "epoch": 9.245872497558594e-06,
      "model_forward_time": 5.601322650909424,
      "step": 9695
    },
    {
      "epoch": 9.245872497558594e-06,
      "step": 9695,
      "training_step_time": 17.058496236801147
    },
    {
      "epoch": 9.246826171875e-06,
      "model_forward_time": 5.9655187129974365,
      "step": 9696
    },
    {
      "epoch": 9.246826171875e-06,
      "step": 9696,
      "training_step_time": 17.411507844924927
    },
    {
      "epoch": 9.247779846191405e-06,
      "model_forward_time": 5.591531991958618,
      "step": 9697
    },
    {
      "epoch": 9.247779846191405e-06,
      "step": 9697,
      "training_step_time": 17.044109582901
    },
    {
      "epoch": 9.248733520507813e-06,
      "model_forward_time": 5.589073657989502,
      "step": 9698
    },
    {
      "epoch": 9.248733520507813e-06,
      "step": 9698,
      "training_step_time": 17.025917530059814
    },
    {
      "epoch": 9.249687194824219e-06,
      "model_forward_time": 5.598093748092651,
      "step": 9699
    },
    {
      "epoch": 9.249687194824219e-06,
      "step": 9699,
      "training_step_time": 17.075176239013672
    },
    {
      "epoch": 9.250640869140624e-06,
      "grad_norm": 0.7816684246063232,
      "learning_rate": 1e-05,
      "loss": 0.0717,
      "step": 9700
    },
    {
      "epoch": 9.250640869140624e-06,
      "model_forward_time": 5.600473165512085,
      "step": 9700
    },
    {
      "dynamics_loss_avg": 0.05855779759585857,
      "epoch": 9.250640869140624e-06,
      "step": 9700
    },
    {
      "action_loss_avg": 0.01042504534125328,
      "epoch": 9.250640869140624e-06,
      "step": 9700
    },
    {
      "epoch": 9.250640869140624e-06,
      "step": 9700,
      "training_step_time": 17.076875686645508
    },
    {
      "epoch": 9.251594543457032e-06,
      "model_forward_time": 5.973080396652222,
      "step": 9701
    },
    {
      "epoch": 9.251594543457032e-06,
      "step": 9701,
      "training_step_time": 17.431349754333496
    },
    {
      "epoch": 9.252548217773437e-06,
      "model_forward_time": 5.606255769729614,
      "step": 9702
    },
    {
      "epoch": 9.252548217773437e-06,
      "step": 9702,
      "training_step_time": 17.055933713912964
    },
    {
      "epoch": 9.253501892089843e-06,
      "model_forward_time": 5.594432353973389,
      "step": 9703
    },
    {
      "epoch": 9.253501892089843e-06,
      "step": 9703,
      "training_step_time": 17.024133920669556
    },
    {
      "epoch": 9.25445556640625e-06,
      "model_forward_time": 5.588901519775391,
      "step": 9704
    },
    {
      "epoch": 9.25445556640625e-06,
      "step": 9704,
      "training_step_time": 17.045602083206177
    },
    {
      "epoch": 9.255409240722656e-06,
      "model_forward_time": 5.599184274673462,
      "step": 9705
    },
    {
      "epoch": 9.255409240722656e-06,
      "step": 9705,
      "training_step_time": 17.0810604095459
    },
    {
      "epoch": 9.256362915039062e-06,
      "model_forward_time": 5.608927965164185,
      "step": 9706
    },
    {
      "epoch": 9.256362915039062e-06,
      "step": 9706,
      "training_step_time": 17.099955558776855
    },
    {
      "epoch": 9.25731658935547e-06,
      "model_forward_time": 5.972120761871338,
      "step": 9707
    },
    {
      "epoch": 9.25731658935547e-06,
      "step": 9707,
      "training_step_time": 17.45339322090149
    },
    {
      "epoch": 9.258270263671875e-06,
      "model_forward_time": 5.604063510894775,
      "step": 9708
    },
    {
      "epoch": 9.258270263671875e-06,
      "step": 9708,
      "training_step_time": 17.08744978904724
    },
    {
      "epoch": 9.259223937988281e-06,
      "model_forward_time": 5.603072166442871,
      "step": 9709
    },
    {
      "epoch": 9.259223937988281e-06,
      "step": 9709,
      "training_step_time": 17.062243461608887
    },
    {
      "epoch": 9.260177612304688e-06,
      "grad_norm": 1.7010351419448853,
      "learning_rate": 1e-05,
      "loss": 0.0648,
      "step": 9710
    },
    {
      "epoch": 9.260177612304688e-06,
      "model_forward_time": 5.612876892089844,
      "step": 9710
    },
    {
      "dynamics_loss_avg": 0.056887786462903024,
      "epoch": 9.260177612304688e-06,
      "step": 9710
    },
    {
      "action_loss_avg": 0.00885749263688922,
      "epoch": 9.260177612304688e-06,
      "step": 9710
    },
    {
      "epoch": 9.260177612304688e-06,
      "step": 9710,
      "training_step_time": 17.088927030563354
    },
    {
      "epoch": 9.261131286621094e-06,
      "model_forward_time": 5.5970072746276855,
      "step": 9711
    },
    {
      "epoch": 9.261131286621094e-06,
      "step": 9711,
      "training_step_time": 17.06084966659546
    },
    {
      "epoch": 9.2620849609375e-06,
      "model_forward_time": 5.588832855224609,
      "step": 9712
    },
    {
      "epoch": 9.2620849609375e-06,
      "step": 9712,
      "training_step_time": 17.414615869522095
    },
    {
      "epoch": 9.263038635253905e-06,
      "model_forward_time": 5.588547468185425,
      "step": 9713
    },
    {
      "epoch": 9.263038635253905e-06,
      "step": 9713,
      "training_step_time": 17.015259504318237
    },
    {
      "epoch": 9.263992309570313e-06,
      "model_forward_time": 5.589238405227661,
      "step": 9714
    },
    {
      "epoch": 9.263992309570313e-06,
      "step": 9714,
      "training_step_time": 17.027897834777832
    },
    {
      "epoch": 9.264945983886719e-06,
      "model_forward_time": 5.59838604927063,
      "step": 9715
    },
    {
      "epoch": 9.264945983886719e-06,
      "step": 9715,
      "training_step_time": 17.06525182723999
    },
    {
      "epoch": 9.265899658203124e-06,
      "model_forward_time": 5.604124546051025,
      "step": 9716
    },
    {
      "epoch": 9.265899658203124e-06,
      "step": 9716,
      "training_step_time": 17.07284164428711
    },
    {
      "epoch": 9.266853332519532e-06,
      "model_forward_time": 5.611544609069824,
      "step": 9717
    },
    {
      "epoch": 9.266853332519532e-06,
      "step": 9717,
      "training_step_time": 17.102093935012817
    },
    {
      "epoch": 9.267807006835937e-06,
      "model_forward_time": 5.973712921142578,
      "step": 9718
    },
    {
      "epoch": 9.267807006835937e-06,
      "step": 9718,
      "training_step_time": 17.42254948616028
    },
    {
      "epoch": 9.268760681152343e-06,
      "model_forward_time": 5.59637713432312,
      "step": 9719
    },
    {
      "epoch": 9.268760681152343e-06,
      "step": 9719,
      "training_step_time": 17.035333156585693
    },
    {
      "epoch": 9.26971435546875e-06,
      "grad_norm": 0.7165825963020325,
      "learning_rate": 1e-05,
      "loss": 0.0755,
      "step": 9720
    },
    {
      "epoch": 9.26971435546875e-06,
      "model_forward_time": 5.586430788040161,
      "step": 9720
    },
    {
      "dynamics_loss_avg": 0.06662509553134441,
      "epoch": 9.26971435546875e-06,
      "step": 9720
    },
    {
      "action_loss_avg": 0.010579932294785977,
      "epoch": 9.26971435546875e-06,
      "step": 9720
    },
    {
      "epoch": 9.26971435546875e-06,
      "step": 9720,
      "training_step_time": 17.048850536346436
    },
    {
      "epoch": 9.270668029785156e-06,
      "model_forward_time": 5.607682704925537,
      "step": 9721
    },
    {
      "epoch": 9.270668029785156e-06,
      "step": 9721,
      "training_step_time": 17.114446878433228
    },
    {
      "epoch": 9.271621704101562e-06,
      "model_forward_time": 5.629589796066284,
      "step": 9722
    },
    {
      "epoch": 9.271621704101562e-06,
      "step": 9722,
      "training_step_time": 17.167147636413574
    },
    {
      "epoch": 9.27257537841797e-06,
      "model_forward_time": 5.622024297714233,
      "step": 9723
    },
    {
      "epoch": 9.27257537841797e-06,
      "step": 9723,
      "training_step_time": 17.48828411102295
    },
    {
      "epoch": 9.273529052734375e-06,
      "model_forward_time": 5.59756064414978,
      "step": 9724
    },
    {
      "epoch": 9.273529052734375e-06,
      "step": 9724,
      "training_step_time": 17.05926012992859
    },
    {
      "epoch": 9.274482727050781e-06,
      "model_forward_time": 5.583178758621216,
      "step": 9725
    },
    {
      "epoch": 9.274482727050781e-06,
      "step": 9725,
      "training_step_time": 17.012008905410767
    },
    {
      "epoch": 9.275436401367188e-06,
      "model_forward_time": 5.593688249588013,
      "step": 9726
    },
    {
      "epoch": 9.275436401367188e-06,
      "step": 9726,
      "training_step_time": 17.052201747894287
    },
    {
      "epoch": 9.276390075683594e-06,
      "model_forward_time": 5.5868916511535645,
      "step": 9727
    },
    {
      "epoch": 9.276390075683594e-06,
      "step": 9727,
      "training_step_time": 17.02627992630005
    },
    {
      "epoch": 9.27734375e-06,
      "model_forward_time": 5.596730470657349,
      "step": 9728
    },
    {
      "epoch": 9.27734375e-06,
      "step": 9728,
      "training_step_time": 17.050688982009888
    },
    {
      "epoch": 9.278297424316405e-06,
      "model_forward_time": 5.964190483093262,
      "step": 9729
    },
    {
      "epoch": 9.278297424316405e-06,
      "step": 9729,
      "training_step_time": 17.407114505767822
    },
    {
      "epoch": 9.279251098632813e-06,
      "grad_norm": 0.7314539551734924,
      "learning_rate": 1e-05,
      "loss": 0.0792,
      "step": 9730
    },
    {
      "epoch": 9.279251098632813e-06,
      "model_forward_time": 5.592771768569946,
      "step": 9730
    },
    {
      "dynamics_loss_avg": 0.0686785962432623,
      "epoch": 9.279251098632813e-06,
      "step": 9730
    },
    {
      "action_loss_avg": 0.012988883769139647,
      "epoch": 9.279251098632813e-06,
      "step": 9730
    },
    {
      "epoch": 9.279251098632813e-06,
      "step": 9730,
      "training_step_time": 17.05546259880066
    },
    {
      "epoch": 9.280204772949219e-06,
      "model_forward_time": 5.601140260696411,
      "step": 9731
    },
    {
      "epoch": 9.280204772949219e-06,
      "step": 9731,
      "training_step_time": 17.06182885169983
    },
    {
      "epoch": 9.281158447265624e-06,
      "model_forward_time": 5.602596998214722,
      "step": 9732
    },
    {
      "epoch": 9.281158447265624e-06,
      "step": 9732,
      "training_step_time": 17.038194179534912
    },
    {
      "epoch": 9.282112121582032e-06,
      "model_forward_time": 5.5971057415008545,
      "step": 9733
    },
    {
      "epoch": 9.282112121582032e-06,
      "step": 9733,
      "training_step_time": 17.02741551399231
    },
    {
      "epoch": 9.283065795898437e-06,
      "model_forward_time": 5.988224267959595,
      "step": 9734
    },
    {
      "epoch": 9.283065795898437e-06,
      "step": 9734,
      "training_step_time": 17.408761501312256
    },
    {
      "epoch": 9.284019470214843e-06,
      "model_forward_time": 5.605009078979492,
      "step": 9735
    },
    {
      "epoch": 9.284019470214843e-06,
      "step": 9735,
      "training_step_time": 17.06971526145935
    },
    {
      "epoch": 9.28497314453125e-06,
      "model_forward_time": 5.612225532531738,
      "step": 9736
    },
    {
      "epoch": 9.28497314453125e-06,
      "step": 9736,
      "training_step_time": 17.11087131500244
    },
    {
      "epoch": 9.285926818847656e-06,
      "model_forward_time": 5.613279581069946,
      "step": 9737
    },
    {
      "epoch": 9.285926818847656e-06,
      "step": 9737,
      "training_step_time": 17.108689069747925
    },
    {
      "epoch": 9.286880493164062e-06,
      "model_forward_time": 5.606172800064087,
      "step": 9738
    },
    {
      "epoch": 9.286880493164062e-06,
      "step": 9738,
      "training_step_time": 17.0858416557312
    },
    {
      "epoch": 9.28783416748047e-06,
      "model_forward_time": 5.598405838012695,
      "step": 9739
    },
    {
      "epoch": 9.28783416748047e-06,
      "step": 9739,
      "training_step_time": 17.048239707946777
    },
    {
      "epoch": 9.288787841796875e-06,
      "grad_norm": 0.6542540788650513,
      "learning_rate": 1e-05,
      "loss": 0.0679,
      "step": 9740
    },
    {
      "epoch": 9.288787841796875e-06,
      "model_forward_time": 5.976301193237305,
      "step": 9740
    },
    {
      "dynamics_loss_avg": 0.05990315191447735,
      "epoch": 9.288787841796875e-06,
      "step": 9740
    },
    {
      "action_loss_avg": 0.00948375160805881,
      "epoch": 9.288787841796875e-06,
      "step": 9740
    },
    {
      "epoch": 9.288787841796875e-06,
      "step": 9740,
      "training_step_time": 17.42356562614441
    },
    {
      "epoch": 9.289741516113281e-06,
      "model_forward_time": 5.599641799926758,
      "step": 9741
    },
    {
      "epoch": 9.289741516113281e-06,
      "step": 9741,
      "training_step_time": 17.076813459396362
    },
    {
      "epoch": 9.290695190429688e-06,
      "model_forward_time": 5.595564126968384,
      "step": 9742
    },
    {
      "epoch": 9.290695190429688e-06,
      "step": 9742,
      "training_step_time": 17.040011882781982
    },
    {
      "epoch": 9.291648864746094e-06,
      "model_forward_time": 5.61003565788269,
      "step": 9743
    },
    {
      "epoch": 9.291648864746094e-06,
      "step": 9743,
      "training_step_time": 17.073487997055054
    },
    {
      "epoch": 9.2926025390625e-06,
      "model_forward_time": 5.595029354095459,
      "step": 9744
    },
    {
      "epoch": 9.2926025390625e-06,
      "step": 9744,
      "training_step_time": 17.03630304336548
    },
    {
      "epoch": 9.293556213378906e-06,
      "model_forward_time": 5.598222255706787,
      "step": 9745
    },
    {
      "epoch": 9.293556213378906e-06,
      "step": 9745,
      "training_step_time": 17.43107843399048
    },
    {
      "epoch": 9.294509887695313e-06,
      "model_forward_time": 5.612823486328125,
      "step": 9746
    },
    {
      "epoch": 9.294509887695313e-06,
      "step": 9746,
      "training_step_time": 17.114450931549072
    },
    {
      "epoch": 9.295463562011719e-06,
      "model_forward_time": 5.616519927978516,
      "step": 9747
    },
    {
      "epoch": 9.295463562011719e-06,
      "step": 9747,
      "training_step_time": 17.08369469642639
    },
    {
      "epoch": 9.296417236328124e-06,
      "model_forward_time": 5.568579435348511,
      "step": 9748
    },
    {
      "epoch": 9.296417236328124e-06,
      "step": 9748,
      "training_step_time": 16.988818407058716
    },
    {
      "epoch": 9.297370910644532e-06,
      "model_forward_time": 5.568694591522217,
      "step": 9749
    },
    {
      "epoch": 9.297370910644532e-06,
      "step": 9749,
      "training_step_time": 16.977693557739258
    },
    {
      "epoch": 9.298324584960938e-06,
      "grad_norm": 0.9602096676826477,
      "learning_rate": 1e-05,
      "loss": 0.0881,
      "step": 9750
    },
    {
      "epoch": 9.298324584960938e-06,
      "model_forward_time": 5.564061880111694,
      "step": 9750
    },
    {
      "dynamics_loss_avg": 0.07284101881086827,
      "epoch": 9.298324584960938e-06,
      "step": 9750
    },
    {
      "action_loss_avg": 0.013313986361026764,
      "epoch": 9.298324584960938e-06,
      "step": 9750
    },
    {
      "epoch": 9.298324584960938e-06,
      "step": 9750,
      "training_step_time": 16.99020504951477
    },
    {
      "epoch": 9.299278259277343e-06,
      "model_forward_time": 5.940192699432373,
      "step": 9751
    },
    {
      "epoch": 9.299278259277343e-06,
      "step": 9751,
      "training_step_time": 17.3614284992218
    },
    {
      "epoch": 9.30023193359375e-06,
      "model_forward_time": 5.569772720336914,
      "step": 9752
    },
    {
      "epoch": 9.30023193359375e-06,
      "step": 9752,
      "training_step_time": 16.988523721694946
    },
    {
      "epoch": 9.301185607910156e-06,
      "model_forward_time": 5.570768356323242,
      "step": 9753
    },
    {
      "epoch": 9.301185607910156e-06,
      "step": 9753,
      "training_step_time": 17.00162672996521
    },
    {
      "epoch": 9.302139282226562e-06,
      "model_forward_time": 5.5887627601623535,
      "step": 9754
    },
    {
      "epoch": 9.302139282226562e-06,
      "step": 9754,
      "training_step_time": 17.03115940093994
    },
    {
      "epoch": 9.30309295654297e-06,
      "model_forward_time": 5.595419883728027,
      "step": 9755
    },
    {
      "epoch": 9.30309295654297e-06,
      "step": 9755,
      "training_step_time": 17.066137552261353
    },
    {
      "epoch": 9.304046630859375e-06,
      "model_forward_time": 5.608080148696899,
      "step": 9756
    },
    {
      "epoch": 9.304046630859375e-06,
      "step": 9756,
      "training_step_time": 17.458872318267822
    },
    {
      "epoch": 9.305000305175781e-06,
      "model_forward_time": 5.602234363555908,
      "step": 9757
    },
    {
      "epoch": 9.305000305175781e-06,
      "step": 9757,
      "training_step_time": 17.082245588302612
    },
    {
      "epoch": 9.305953979492187e-06,
      "model_forward_time": 5.600614547729492,
      "step": 9758
    },
    {
      "epoch": 9.305953979492187e-06,
      "step": 9758,
      "training_step_time": 17.045701265335083
    },
    {
      "epoch": 9.306907653808594e-06,
      "model_forward_time": 5.590461492538452,
      "step": 9759
    },
    {
      "epoch": 9.306907653808594e-06,
      "step": 9759,
      "training_step_time": 17.02103900909424
    },
    {
      "epoch": 9.307861328125e-06,
      "grad_norm": 1.4607288837432861,
      "learning_rate": 1e-05,
      "loss": 0.0772,
      "step": 9760
    },
    {
      "epoch": 9.307861328125e-06,
      "model_forward_time": 5.582365274429321,
      "step": 9760
    },
    {
      "dynamics_loss_avg": 0.06046708561480045,
      "epoch": 9.307861328125e-06,
      "step": 9760
    },
    {
      "action_loss_avg": 0.009432381624355913,
      "epoch": 9.307861328125e-06,
      "step": 9760
    },
    {
      "epoch": 9.307861328125e-06,
      "step": 9760,
      "training_step_time": 16.990911722183228
    },
    {
      "epoch": 9.308815002441406e-06,
      "model_forward_time": 5.587635040283203,
      "step": 9761
    },
    {
      "epoch": 9.308815002441406e-06,
      "step": 9761,
      "training_step_time": 17.030508279800415
    },
    {
      "epoch": 9.309768676757813e-06,
      "model_forward_time": 5.959507465362549,
      "step": 9762
    },
    {
      "epoch": 9.309768676757813e-06,
      "step": 9762,
      "training_step_time": 17.440256118774414
    },
    {
      "epoch": 9.310722351074219e-06,
      "model_forward_time": 5.607323408126831,
      "step": 9763
    },
    {
      "epoch": 9.310722351074219e-06,
      "step": 9763,
      "training_step_time": 17.094916105270386
    },
    {
      "epoch": 9.311676025390624e-06,
      "model_forward_time": 5.60273003578186,
      "step": 9764
    },
    {
      "epoch": 9.311676025390624e-06,
      "step": 9764,
      "training_step_time": 17.08043074607849
    },
    {
      "epoch": 9.312629699707032e-06,
      "model_forward_time": 5.605485439300537,
      "step": 9765
    },
    {
      "epoch": 9.312629699707032e-06,
      "step": 9765,
      "training_step_time": 17.074297666549683
    },
    {
      "epoch": 9.313583374023438e-06,
      "model_forward_time": 5.595676422119141,
      "step": 9766
    },
    {
      "epoch": 9.313583374023438e-06,
      "step": 9766,
      "training_step_time": 17.069939374923706
    },
    {
      "epoch": 9.314537048339843e-06,
      "model_forward_time": 5.952661037445068,
      "step": 9767
    },
    {
      "epoch": 9.314537048339843e-06,
      "step": 9767,
      "training_step_time": 17.392186164855957
    },
    {
      "epoch": 9.31549072265625e-06,
      "model_forward_time": 5.592286586761475,
      "step": 9768
    },
    {
      "epoch": 9.31549072265625e-06,
      "step": 9768,
      "training_step_time": 17.00909948348999
    },
    {
      "epoch": 9.316444396972656e-06,
      "model_forward_time": 5.593287467956543,
      "step": 9769
    },
    {
      "epoch": 9.316444396972656e-06,
      "step": 9769,
      "training_step_time": 17.042275190353394
    },
    {
      "epoch": 9.317398071289062e-06,
      "grad_norm": 0.554440975189209,
      "learning_rate": 1e-05,
      "loss": 0.0751,
      "step": 9770
    },
    {
      "epoch": 9.317398071289062e-06,
      "model_forward_time": 5.605723857879639,
      "step": 9770
    },
    {
      "dynamics_loss_avg": 0.06881758905947208,
      "epoch": 9.317398071289062e-06,
      "step": 9770
    },
    {
      "action_loss_avg": 0.012314881710335613,
      "epoch": 9.317398071289062e-06,
      "step": 9770
    },
    {
      "epoch": 9.317398071289062e-06,
      "step": 9770,
      "training_step_time": 17.088486433029175
    },
    {
      "epoch": 9.31835174560547e-06,
      "model_forward_time": 5.6184282302856445,
      "step": 9771
    },
    {
      "epoch": 9.31835174560547e-06,
      "step": 9771,
      "training_step_time": 17.091057300567627
    },
    {
      "epoch": 9.319305419921875e-06,
      "model_forward_time": 5.601592779159546,
      "step": 9772
    },
    {
      "epoch": 9.319305419921875e-06,
      "step": 9772,
      "training_step_time": 17.088325262069702
    },
    {
      "epoch": 9.320259094238281e-06,
      "model_forward_time": 5.969769239425659,
      "step": 9773
    },
    {
      "epoch": 9.320259094238281e-06,
      "step": 9773,
      "training_step_time": 17.444135904312134
    },
    {
      "epoch": 9.321212768554687e-06,
      "model_forward_time": 5.608545303344727,
      "step": 9774
    },
    {
      "epoch": 9.321212768554687e-06,
      "step": 9774,
      "training_step_time": 17.08511185646057
    },
    {
      "epoch": 9.322166442871094e-06,
      "model_forward_time": 5.5934553146362305,
      "step": 9775
    },
    {
      "epoch": 9.322166442871094e-06,
      "step": 9775,
      "training_step_time": 17.015932321548462
    },
    {
      "epoch": 9.3231201171875e-06,
      "model_forward_time": 5.580493927001953,
      "step": 9776
    },
    {
      "epoch": 9.3231201171875e-06,
      "step": 9776,
      "training_step_time": 17.005318641662598
    },
    {
      "epoch": 9.324073791503906e-06,
      "model_forward_time": 5.616011381149292,
      "step": 9777
    },
    {
      "epoch": 9.324073791503906e-06,
      "step": 9777,
      "training_step_time": 17.07141947746277
    },
    {
      "epoch": 9.325027465820313e-06,
      "model_forward_time": 5.591313123703003,
      "step": 9778
    },
    {
      "epoch": 9.325027465820313e-06,
      "step": 9778,
      "training_step_time": 17.425716876983643
    },
    {
      "epoch": 9.325981140136719e-06,
      "model_forward_time": 5.603966236114502,
      "step": 9779
    },
    {
      "epoch": 9.325981140136719e-06,
      "step": 9779,
      "training_step_time": 17.090519905090332
    },
    {
      "epoch": 9.326934814453124e-06,
      "grad_norm": 0.7590488791465759,
      "learning_rate": 1e-05,
      "loss": 0.0753,
      "step": 9780
    },
    {
      "epoch": 9.326934814453124e-06,
      "model_forward_time": 5.599447727203369,
      "step": 9780
    },
    {
      "dynamics_loss_avg": 0.06630502566695214,
      "epoch": 9.326934814453124e-06,
      "step": 9780
    },
    {
      "action_loss_avg": 0.009552904404699802,
      "epoch": 9.326934814453124e-06,
      "step": 9780
    },
    {
      "epoch": 9.326934814453124e-06,
      "step": 9780,
      "training_step_time": 17.062562704086304
    },
    {
      "epoch": 9.327888488769532e-06,
      "model_forward_time": 5.596401691436768,
      "step": 9781
    },
    {
      "epoch": 9.327888488769532e-06,
      "step": 9781,
      "training_step_time": 17.05736780166626
    },
    {
      "epoch": 9.328842163085938e-06,
      "model_forward_time": 5.614044904708862,
      "step": 9782
    },
    {
      "epoch": 9.328842163085938e-06,
      "step": 9782,
      "training_step_time": 17.0857515335083
    },
    {
      "epoch": 9.329795837402343e-06,
      "model_forward_time": 5.596631050109863,
      "step": 9783
    },
    {
      "epoch": 9.329795837402343e-06,
      "step": 9783,
      "training_step_time": 17.061890602111816
    },
    {
      "epoch": 9.33074951171875e-06,
      "model_forward_time": 5.966567277908325,
      "step": 9784
    },
    {
      "epoch": 9.33074951171875e-06,
      "step": 9784,
      "training_step_time": 17.415392875671387
    },
    {
      "epoch": 9.331703186035156e-06,
      "model_forward_time": 5.5999977588653564,
      "step": 9785
    },
    {
      "epoch": 9.331703186035156e-06,
      "step": 9785,
      "training_step_time": 17.075732231140137
    },
    {
      "epoch": 9.332656860351562e-06,
      "model_forward_time": 5.599697113037109,
      "step": 9786
    },
    {
      "epoch": 9.332656860351562e-06,
      "step": 9786,
      "training_step_time": 17.061063289642334
    },
    {
      "epoch": 9.33361053466797e-06,
      "model_forward_time": 5.598703384399414,
      "step": 9787
    },
    {
      "epoch": 9.33361053466797e-06,
      "step": 9787,
      "training_step_time": 17.061460494995117
    },
    {
      "epoch": 9.334564208984375e-06,
      "model_forward_time": 5.600585222244263,
      "step": 9788
    },
    {
      "epoch": 9.334564208984375e-06,
      "step": 9788,
      "training_step_time": 17.048789501190186
    },
    {
      "epoch": 9.335517883300781e-06,
      "model_forward_time": 5.5869832038879395,
      "step": 9789
    },
    {
      "epoch": 9.335517883300781e-06,
      "step": 9789,
      "training_step_time": 17.3892719745636
    },
    {
      "epoch": 9.336471557617187e-06,
      "grad_norm": 0.5421062111854553,
      "learning_rate": 1e-05,
      "loss": 0.0792,
      "step": 9790
    },
    {
      "epoch": 9.336471557617187e-06,
      "model_forward_time": 5.579801797866821,
      "step": 9790
    },
    {
      "dynamics_loss_avg": 0.0669987540692091,
      "epoch": 9.336471557617187e-06,
      "step": 9790
    },
    {
      "action_loss_avg": 0.009586264728568494,
      "epoch": 9.336471557617187e-06,
      "step": 9790
    },
    {
      "epoch": 9.336471557617187e-06,
      "step": 9790,
      "training_step_time": 16.98478603363037
    },
    {
      "epoch": 9.337425231933594e-06,
      "model_forward_time": 5.58868408203125,
      "step": 9791
    },
    {
      "epoch": 9.337425231933594e-06,
      "step": 9791,
      "training_step_time": 17.05889058113098
    },
    {
      "epoch": 9.33837890625e-06,
      "model_forward_time": 5.614600896835327,
      "step": 9792
    },
    {
      "epoch": 9.33837890625e-06,
      "step": 9792,
      "training_step_time": 17.078981637954712
    },
    {
      "epoch": 9.339332580566406e-06,
      "model_forward_time": 5.591979503631592,
      "step": 9793
    },
    {
      "epoch": 9.339332580566406e-06,
      "step": 9793,
      "training_step_time": 17.0322904586792
    },
    {
      "epoch": 9.340286254882813e-06,
      "model_forward_time": 5.592725515365601,
      "step": 9794
    },
    {
      "epoch": 9.340286254882813e-06,
      "step": 9794,
      "training_step_time": 17.017592430114746
    },
    {
      "epoch": 9.341239929199219e-06,
      "model_forward_time": 5.9631054401397705,
      "step": 9795
    },
    {
      "epoch": 9.341239929199219e-06,
      "step": 9795,
      "training_step_time": 17.422252416610718
    },
    {
      "epoch": 9.342193603515624e-06,
      "model_forward_time": 5.61318039894104,
      "step": 9796
    },
    {
      "epoch": 9.342193603515624e-06,
      "step": 9796,
      "training_step_time": 17.116481065750122
    },
    {
      "epoch": 9.343147277832032e-06,
      "model_forward_time": 5.61735987663269,
      "step": 9797
    },
    {
      "epoch": 9.343147277832032e-06,
      "step": 9797,
      "training_step_time": 17.11950397491455
    },
    {
      "epoch": 9.344100952148438e-06,
      "model_forward_time": 5.629103183746338,
      "step": 9798
    },
    {
      "epoch": 9.344100952148438e-06,
      "step": 9798,
      "training_step_time": 17.126988410949707
    },
    {
      "epoch": 9.345054626464843e-06,
      "model_forward_time": 5.619879961013794,
      "step": 9799
    },
    {
      "epoch": 9.345054626464843e-06,
      "step": 9799,
      "training_step_time": 17.112532377243042
    },
    {
      "epoch": 9.34600830078125e-06,
      "grad_norm": 1.0800586938858032,
      "learning_rate": 1e-05,
      "loss": 0.0797,
      "step": 9800
    },
    {
      "epoch": 9.34600830078125e-06,
      "model_forward_time": 5.970124959945679,
      "step": 9800
    },
    {
      "dynamics_loss_avg": 0.06687132306396962,
      "epoch": 9.34600830078125e-06,
      "step": 9800
    },
    {
      "action_loss_avg": 0.009877369506284595,
      "epoch": 9.34600830078125e-06,
      "step": 9800
    },
    {
      "epoch": 9.34600830078125e-06,
      "step": 9800,
      "training_step_time": 17.4636070728302
    },
    {
      "epoch": 9.346961975097656e-06,
      "model_forward_time": 5.6143434047698975,
      "step": 9801
    },
    {
      "epoch": 9.346961975097656e-06,
      "step": 9801,
      "training_step_time": 17.105146408081055
    },
    {
      "epoch": 9.347915649414062e-06,
      "model_forward_time": 5.616034269332886,
      "step": 9802
    },
    {
      "epoch": 9.347915649414062e-06,
      "step": 9802,
      "training_step_time": 17.10464644432068
    },
    {
      "epoch": 9.34886932373047e-06,
      "model_forward_time": 5.6109960079193115,
      "step": 9803
    },
    {
      "epoch": 9.34886932373047e-06,
      "step": 9803,
      "training_step_time": 17.10692596435547
    },
    {
      "epoch": 9.349822998046875e-06,
      "model_forward_time": 5.614271640777588,
      "step": 9804
    },
    {
      "epoch": 9.349822998046875e-06,
      "step": 9804,
      "training_step_time": 17.085495948791504
    },
    {
      "epoch": 9.350776672363281e-06,
      "model_forward_time": 5.594373941421509,
      "step": 9805
    },
    {
      "epoch": 9.350776672363281e-06,
      "step": 9805,
      "training_step_time": 17.007891178131104
    },
    {
      "epoch": 9.351730346679687e-06,
      "model_forward_time": 5.946199417114258,
      "step": 9806
    },
    {
      "epoch": 9.351730346679687e-06,
      "step": 9806,
      "training_step_time": 17.37560534477234
    },
    {
      "epoch": 9.352684020996094e-06,
      "model_forward_time": 5.5751471519470215,
      "step": 9807
    },
    {
      "epoch": 9.352684020996094e-06,
      "step": 9807,
      "training_step_time": 17.01827836036682
    },
    {
      "epoch": 9.3536376953125e-06,
      "model_forward_time": 5.576078176498413,
      "step": 9808
    },
    {
      "epoch": 9.3536376953125e-06,
      "step": 9808,
      "training_step_time": 17.000742197036743
    },
    {
      "epoch": 9.354591369628906e-06,
      "model_forward_time": 5.5788798332214355,
      "step": 9809
    },
    {
      "epoch": 9.354591369628906e-06,
      "step": 9809,
      "training_step_time": 17.020795583724976
    },
    {
      "epoch": 9.355545043945313e-06,
      "grad_norm": 0.4767707586288452,
      "learning_rate": 1e-05,
      "loss": 0.0607,
      "step": 9810
    },
    {
      "epoch": 9.355545043945313e-06,
      "model_forward_time": 5.590258359909058,
      "step": 9810
    },
    {
      "dynamics_loss_avg": 0.05293742436915636,
      "epoch": 9.355545043945313e-06,
      "step": 9810
    },
    {
      "action_loss_avg": 0.008452011551707984,
      "epoch": 9.355545043945313e-06,
      "step": 9810
    },
    {
      "epoch": 9.355545043945313e-06,
      "step": 9810,
      "training_step_time": 17.032267332077026
    },
    {
      "epoch": 9.356498718261719e-06,
      "model_forward_time": 5.6023876667022705,
      "step": 9811
    },
    {
      "epoch": 9.356498718261719e-06,
      "step": 9811,
      "training_step_time": 17.405237674713135
    },
    {
      "epoch": 9.357452392578124e-06,
      "model_forward_time": 5.606142520904541,
      "step": 9812
    },
    {
      "epoch": 9.357452392578124e-06,
      "step": 9812,
      "training_step_time": 17.1069393157959
    },
    {
      "epoch": 9.358406066894532e-06,
      "model_forward_time": 5.62075138092041,
      "step": 9813
    },
    {
      "epoch": 9.358406066894532e-06,
      "step": 9813,
      "training_step_time": 17.132031440734863
    },
    {
      "epoch": 9.359359741210938e-06,
      "model_forward_time": 5.625447034835815,
      "step": 9814
    },
    {
      "epoch": 9.359359741210938e-06,
      "step": 9814,
      "training_step_time": 17.111254692077637
    },
    {
      "epoch": 9.360313415527343e-06,
      "model_forward_time": 5.586332559585571,
      "step": 9815
    },
    {
      "epoch": 9.360313415527343e-06,
      "step": 9815,
      "training_step_time": 17.02547264099121
    },
    {
      "epoch": 9.36126708984375e-06,
      "model_forward_time": 5.5817766189575195,
      "step": 9816
    },
    {
      "epoch": 9.36126708984375e-06,
      "step": 9816,
      "training_step_time": 17.011064291000366
    },
    {
      "epoch": 9.362220764160156e-06,
      "model_forward_time": 5.952413082122803,
      "step": 9817
    },
    {
      "epoch": 9.362220764160156e-06,
      "step": 9817,
      "training_step_time": 17.383474588394165
    },
    {
      "epoch": 9.363174438476562e-06,
      "model_forward_time": 5.593292236328125,
      "step": 9818
    },
    {
      "epoch": 9.363174438476562e-06,
      "step": 9818,
      "training_step_time": 17.045008420944214
    },
    {
      "epoch": 9.36412811279297e-06,
      "model_forward_time": 5.59882378578186,
      "step": 9819
    },
    {
      "epoch": 9.36412811279297e-06,
      "step": 9819,
      "training_step_time": 17.081308364868164
    },
    {
      "epoch": 9.365081787109375e-06,
      "grad_norm": 0.3052254915237427,
      "learning_rate": 1e-05,
      "loss": 0.0756,
      "step": 9820
    },
    {
      "epoch": 9.365081787109375e-06,
      "model_forward_time": 5.624710321426392,
      "step": 9820
    },
    {
      "dynamics_loss_avg": 0.0650077573955059,
      "epoch": 9.365081787109375e-06,
      "step": 9820
    },
    {
      "action_loss_avg": 0.009085850836709142,
      "epoch": 9.365081787109375e-06,
      "step": 9820
    },
    {
      "epoch": 9.365081787109375e-06,
      "step": 9820,
      "training_step_time": 17.1221284866333
    },
    {
      "epoch": 9.366035461425781e-06,
      "model_forward_time": 5.600625991821289,
      "step": 9821
    },
    {
      "epoch": 9.366035461425781e-06,
      "step": 9821,
      "training_step_time": 17.079768419265747
    },
    {
      "epoch": 9.366989135742187e-06,
      "model_forward_time": 5.616097450256348,
      "step": 9822
    },
    {
      "epoch": 9.366989135742187e-06,
      "step": 9822,
      "training_step_time": 17.45186138153076
    },
    {
      "epoch": 9.367942810058594e-06,
      "model_forward_time": 5.59705376625061,
      "step": 9823
    },
    {
      "epoch": 9.367942810058594e-06,
      "step": 9823,
      "training_step_time": 17.054475784301758
    },
    {
      "epoch": 9.368896484375e-06,
      "model_forward_time": 5.591361999511719,
      "step": 9824
    },
    {
      "epoch": 9.368896484375e-06,
      "step": 9824,
      "training_step_time": 17.025413990020752
    },
    {
      "epoch": 9.369850158691406e-06,
      "model_forward_time": 5.591579437255859,
      "step": 9825
    },
    {
      "epoch": 9.369850158691406e-06,
      "step": 9825,
      "training_step_time": 17.042383670806885
    },
    {
      "epoch": 9.370803833007813e-06,
      "model_forward_time": 5.594698190689087,
      "step": 9826
    },
    {
      "epoch": 9.370803833007813e-06,
      "step": 9826,
      "training_step_time": 17.03642702102661
    },
    {
      "epoch": 9.371757507324219e-06,
      "model_forward_time": 5.590206146240234,
      "step": 9827
    },
    {
      "epoch": 9.371757507324219e-06,
      "step": 9827,
      "training_step_time": 17.04925847053528
    },
    {
      "epoch": 9.372711181640624e-06,
      "model_forward_time": 5.9645209312438965,
      "step": 9828
    },
    {
      "epoch": 9.372711181640624e-06,
      "step": 9828,
      "training_step_time": 17.443084239959717
    },
    {
      "epoch": 9.373664855957032e-06,
      "model_forward_time": 5.597872734069824,
      "step": 9829
    },
    {
      "epoch": 9.373664855957032e-06,
      "step": 9829,
      "training_step_time": 17.06013011932373
    },
    {
      "epoch": 9.374618530273438e-06,
      "grad_norm": 0.6552417278289795,
      "learning_rate": 1e-05,
      "loss": 0.0524,
      "step": 9830
    },
    {
      "epoch": 9.374618530273438e-06,
      "model_forward_time": 5.609548568725586,
      "step": 9830
    },
    {
      "dynamics_loss_avg": 0.053676326386630537,
      "epoch": 9.374618530273438e-06,
      "step": 9830
    },
    {
      "action_loss_avg": 0.006785267277155072,
      "epoch": 9.374618530273438e-06,
      "step": 9830
    },
    {
      "epoch": 9.374618530273438e-06,
      "step": 9830,
      "training_step_time": 17.107027053833008
    },
    {
      "epoch": 9.375572204589843e-06,
      "model_forward_time": 5.616935968399048,
      "step": 9831
    },
    {
      "epoch": 9.375572204589843e-06,
      "step": 9831,
      "training_step_time": 17.13025212287903
    },
    {
      "epoch": 9.37652587890625e-06,
      "model_forward_time": 5.6227312088012695,
      "step": 9832
    },
    {
      "epoch": 9.37652587890625e-06,
      "step": 9832,
      "training_step_time": 17.118473768234253
    },
    {
      "epoch": 9.377479553222656e-06,
      "model_forward_time": 5.9712302684783936,
      "step": 9833
    },
    {
      "epoch": 9.377479553222656e-06,
      "step": 9833,
      "training_step_time": 17.41870641708374
    },
    {
      "epoch": 9.378433227539062e-06,
      "model_forward_time": 5.585866928100586,
      "step": 9834
    },
    {
      "epoch": 9.378433227539062e-06,
      "step": 9834,
      "training_step_time": 17.034626245498657
    },
    {
      "epoch": 9.37938690185547e-06,
      "model_forward_time": 5.572701454162598,
      "step": 9835
    },
    {
      "epoch": 9.37938690185547e-06,
      "step": 9835,
      "training_step_time": 16.992374420166016
    },
    {
      "epoch": 9.380340576171875e-06,
      "model_forward_time": 5.576399803161621,
      "step": 9836
    },
    {
      "epoch": 9.380340576171875e-06,
      "step": 9836,
      "training_step_time": 16.995059967041016
    },
    {
      "epoch": 9.381294250488281e-06,
      "model_forward_time": 5.581236124038696,
      "step": 9837
    },
    {
      "epoch": 9.381294250488281e-06,
      "step": 9837,
      "training_step_time": 17.019062519073486
    },
    {
      "epoch": 9.382247924804687e-06,
      "model_forward_time": 5.578349590301514,
      "step": 9838
    },
    {
      "epoch": 9.382247924804687e-06,
      "step": 9838,
      "training_step_time": 17.019176483154297
    },
    {
      "epoch": 9.383201599121094e-06,
      "model_forward_time": 5.966871976852417,
      "step": 9839
    },
    {
      "epoch": 9.383201599121094e-06,
      "step": 9839,
      "training_step_time": 17.39730429649353
    },
    {
      "epoch": 9.3841552734375e-06,
      "grad_norm": 1.1382145881652832,
      "learning_rate": 1e-05,
      "loss": 0.0754,
      "step": 9840
    },
    {
      "epoch": 9.3841552734375e-06,
      "model_forward_time": 5.594136953353882,
      "step": 9840
    },
    {
      "dynamics_loss_avg": 0.0577029125764966,
      "epoch": 9.3841552734375e-06,
      "step": 9840
    },
    {
      "action_loss_avg": 0.0077924989629536865,
      "epoch": 9.3841552734375e-06,
      "step": 9840
    },
    {
      "epoch": 9.3841552734375e-06,
      "step": 9840,
      "training_step_time": 17.018694400787354
    },
    {
      "epoch": 9.385108947753906e-06,
      "model_forward_time": 5.600571393966675,
      "step": 9841
    },
    {
      "epoch": 9.385108947753906e-06,
      "step": 9841,
      "training_step_time": 17.034054040908813
    },
    {
      "epoch": 9.386062622070313e-06,
      "model_forward_time": 5.595118284225464,
      "step": 9842
    },
    {
      "epoch": 9.386062622070313e-06,
      "step": 9842,
      "training_step_time": 17.07270836830139
    },
    {
      "epoch": 9.387016296386719e-06,
      "model_forward_time": 5.599854469299316,
      "step": 9843
    },
    {
      "epoch": 9.387016296386719e-06,
      "step": 9843,
      "training_step_time": 17.07084608078003
    },
    {
      "epoch": 9.387969970703124e-06,
      "model_forward_time": 5.602087020874023,
      "step": 9844
    },
    {
      "epoch": 9.387969970703124e-06,
      "step": 9844,
      "training_step_time": 17.449373245239258
    },
    {
      "epoch": 9.388923645019532e-06,
      "model_forward_time": 5.59778356552124,
      "step": 9845
    },
    {
      "epoch": 9.388923645019532e-06,
      "step": 9845,
      "training_step_time": 17.039446353912354
    },
    {
      "epoch": 9.389877319335938e-06,
      "model_forward_time": 5.5859105587005615,
      "step": 9846
    },
    {
      "epoch": 9.389877319335938e-06,
      "step": 9846,
      "training_step_time": 17.033633708953857
    },
    {
      "epoch": 9.390830993652343e-06,
      "model_forward_time": 5.586706638336182,
      "step": 9847
    },
    {
      "epoch": 9.390830993652343e-06,
      "step": 9847,
      "training_step_time": 17.03906488418579
    },
    {
      "epoch": 9.39178466796875e-06,
      "model_forward_time": 5.595923900604248,
      "step": 9848
    },
    {
      "epoch": 9.39178466796875e-06,
      "step": 9848,
      "training_step_time": 17.06102204322815
    },
    {
      "epoch": 9.392738342285156e-06,
      "model_forward_time": 5.605509042739868,
      "step": 9849
    },
    {
      "epoch": 9.392738342285156e-06,
      "step": 9849,
      "training_step_time": 17.08653688430786
    },
    {
      "epoch": 9.393692016601562e-06,
      "grad_norm": 0.9412540793418884,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 9850
    },
    {
      "epoch": 9.393692016601562e-06,
      "model_forward_time": 5.989392995834351,
      "step": 9850
    },
    {
      "dynamics_loss_avg": 0.05779667068272829,
      "epoch": 9.393692016601562e-06,
      "step": 9850
    },
    {
      "action_loss_avg": 0.010892167780548334,
      "epoch": 9.393692016601562e-06,
      "step": 9850
    },
    {
      "epoch": 9.393692016601562e-06,
      "step": 9850,
      "training_step_time": 17.493240356445312
    },
    {
      "epoch": 9.39464569091797e-06,
      "model_forward_time": 5.624413728713989,
      "step": 9851
    },
    {
      "epoch": 9.39464569091797e-06,
      "step": 9851,
      "training_step_time": 17.108057975769043
    },
    {
      "epoch": 9.395599365234375e-06,
      "model_forward_time": 5.596513986587524,
      "step": 9852
    },
    {
      "epoch": 9.395599365234375e-06,
      "step": 9852,
      "training_step_time": 17.032923698425293
    },
    {
      "epoch": 9.396553039550781e-06,
      "model_forward_time": 5.588864326477051,
      "step": 9853
    },
    {
      "epoch": 9.396553039550781e-06,
      "step": 9853,
      "training_step_time": 17.038917064666748
    },
    {
      "epoch": 9.397506713867187e-06,
      "model_forward_time": 5.594820022583008,
      "step": 9854
    },
    {
      "epoch": 9.397506713867187e-06,
      "step": 9854,
      "training_step_time": 17.038070678710938
    },
    {
      "epoch": 9.398460388183594e-06,
      "model_forward_time": 5.941490888595581,
      "step": 9855
    },
    {
      "epoch": 9.398460388183594e-06,
      "step": 9855,
      "training_step_time": 17.408510208129883
    },
    {
      "epoch": 9.3994140625e-06,
      "model_forward_time": 5.597245931625366,
      "step": 9856
    },
    {
      "epoch": 9.3994140625e-06,
      "step": 9856,
      "training_step_time": 17.074652671813965
    },
    {
      "epoch": 9.400367736816406e-06,
      "model_forward_time": 5.603782892227173,
      "step": 9857
    },
    {
      "epoch": 9.400367736816406e-06,
      "step": 9857,
      "training_step_time": 17.038761138916016
    },
    {
      "epoch": 9.401321411132813e-06,
      "model_forward_time": 5.596062660217285,
      "step": 9858
    },
    {
      "epoch": 9.401321411132813e-06,
      "step": 9858,
      "training_step_time": 17.03590679168701
    },
    {
      "epoch": 9.402275085449219e-06,
      "model_forward_time": 5.5935773849487305,
      "step": 9859
    },
    {
      "epoch": 9.402275085449219e-06,
      "step": 9859,
      "training_step_time": 17.041123628616333
    },
    {
      "epoch": 9.403228759765624e-06,
      "grad_norm": 1.1506341695785522,
      "learning_rate": 1e-05,
      "loss": 0.0529,
      "step": 9860
    },
    {
      "epoch": 9.403228759765624e-06,
      "model_forward_time": 5.601348400115967,
      "step": 9860
    },
    {
      "dynamics_loss_avg": 0.05292475894093514,
      "epoch": 9.403228759765624e-06,
      "step": 9860
    },
    {
      "action_loss_avg": 0.00834130528382957,
      "epoch": 9.403228759765624e-06,
      "step": 9860
    },
    {
      "epoch": 9.403228759765624e-06,
      "step": 9860,
      "training_step_time": 17.09447479248047
    },
    {
      "epoch": 9.404182434082032e-06,
      "model_forward_time": 5.97728157043457,
      "step": 9861
    },
    {
      "epoch": 9.404182434082032e-06,
      "step": 9861,
      "training_step_time": 17.481393337249756
    },
    {
      "epoch": 9.405136108398438e-06,
      "model_forward_time": 5.6206629276275635,
      "step": 9862
    },
    {
      "epoch": 9.405136108398438e-06,
      "step": 9862,
      "training_step_time": 17.12277913093567
    },
    {
      "epoch": 9.406089782714843e-06,
      "model_forward_time": 5.619911193847656,
      "step": 9863
    },
    {
      "epoch": 9.406089782714843e-06,
      "step": 9863,
      "training_step_time": 17.100935459136963
    },
    {
      "epoch": 9.40704345703125e-06,
      "model_forward_time": 5.605888605117798,
      "step": 9864
    },
    {
      "epoch": 9.40704345703125e-06,
      "step": 9864,
      "training_step_time": 17.093127965927124
    },
    {
      "epoch": 9.407997131347656e-06,
      "model_forward_time": 5.614468812942505,
      "step": 9865
    },
    {
      "epoch": 9.407997131347656e-06,
      "step": 9865,
      "training_step_time": 17.111607789993286
    },
    {
      "epoch": 9.408950805664062e-06,
      "model_forward_time": 5.9633026123046875,
      "step": 9866
    },
    {
      "epoch": 9.408950805664062e-06,
      "step": 9866,
      "training_step_time": 17.45635175704956
    },
    {
      "epoch": 9.40990447998047e-06,
      "model_forward_time": 5.616392612457275,
      "step": 9867
    },
    {
      "epoch": 9.40990447998047e-06,
      "step": 9867,
      "training_step_time": 17.1176815032959
    },
    {
      "epoch": 9.410858154296875e-06,
      "model_forward_time": 5.602465629577637,
      "step": 9868
    },
    {
      "epoch": 9.410858154296875e-06,
      "step": 9868,
      "training_step_time": 17.060837745666504
    },
    {
      "epoch": 9.411811828613281e-06,
      "model_forward_time": 5.585781097412109,
      "step": 9869
    },
    {
      "epoch": 9.411811828613281e-06,
      "step": 9869,
      "training_step_time": 17.02348303794861
    },
    {
      "epoch": 9.412765502929687e-06,
      "grad_norm": 1.1044942140579224,
      "learning_rate": 1e-05,
      "loss": 0.0688,
      "step": 9870
    },
    {
      "epoch": 9.412765502929687e-06,
      "model_forward_time": 5.580246448516846,
      "step": 9870
    },
    {
      "dynamics_loss_avg": 0.053056991286575796,
      "epoch": 9.412765502929687e-06,
      "step": 9870
    },
    {
      "action_loss_avg": 0.008470511459745467,
      "epoch": 9.412765502929687e-06,
      "step": 9870
    },
    {
      "epoch": 9.412765502929687e-06,
      "step": 9870,
      "training_step_time": 16.985410690307617
    },
    {
      "epoch": 9.413719177246094e-06,
      "model_forward_time": 5.5846264362335205,
      "step": 9871
    },
    {
      "epoch": 9.413719177246094e-06,
      "step": 9871,
      "training_step_time": 17.379076719284058
    },
    {
      "epoch": 9.4146728515625e-06,
      "model_forward_time": 5.582340955734253,
      "step": 9872
    },
    {
      "epoch": 9.4146728515625e-06,
      "step": 9872,
      "training_step_time": 17.03006887435913
    },
    {
      "epoch": 9.415626525878906e-06,
      "model_forward_time": 5.60075306892395,
      "step": 9873
    },
    {
      "epoch": 9.415626525878906e-06,
      "step": 9873,
      "training_step_time": 17.07609486579895
    },
    {
      "epoch": 9.416580200195313e-06,
      "model_forward_time": 5.613978862762451,
      "step": 9874
    },
    {
      "epoch": 9.416580200195313e-06,
      "step": 9874,
      "training_step_time": 17.112292051315308
    },
    {
      "epoch": 9.417533874511719e-06,
      "model_forward_time": 5.608311891555786,
      "step": 9875
    },
    {
      "epoch": 9.417533874511719e-06,
      "step": 9875,
      "training_step_time": 17.07936716079712
    },
    {
      "epoch": 9.418487548828124e-06,
      "model_forward_time": 5.607903957366943,
      "step": 9876
    },
    {
      "epoch": 9.418487548828124e-06,
      "step": 9876,
      "training_step_time": 17.062725067138672
    },
    {
      "epoch": 9.419441223144532e-06,
      "model_forward_time": 5.949533224105835,
      "step": 9877
    },
    {
      "epoch": 9.419441223144532e-06,
      "step": 9877,
      "training_step_time": 17.42712664604187
    },
    {
      "epoch": 9.420394897460938e-06,
      "model_forward_time": 5.617119550704956,
      "step": 9878
    },
    {
      "epoch": 9.420394897460938e-06,
      "step": 9878,
      "training_step_time": 17.056455373764038
    },
    {
      "epoch": 9.421348571777343e-06,
      "model_forward_time": 5.614818572998047,
      "step": 9879
    },
    {
      "epoch": 9.421348571777343e-06,
      "step": 9879,
      "training_step_time": 17.100058794021606
    },
    {
      "epoch": 9.42230224609375e-06,
      "grad_norm": 0.7930171489715576,
      "learning_rate": 1e-05,
      "loss": 0.0779,
      "step": 9880
    },
    {
      "epoch": 9.42230224609375e-06,
      "model_forward_time": 5.621992349624634,
      "step": 9880
    },
    {
      "dynamics_loss_avg": 0.06505317706614733,
      "epoch": 9.42230224609375e-06,
      "step": 9880
    },
    {
      "action_loss_avg": 0.013167814258486033,
      "epoch": 9.42230224609375e-06,
      "step": 9880
    },
    {
      "epoch": 9.42230224609375e-06,
      "step": 9880,
      "training_step_time": 17.10019087791443
    },
    {
      "epoch": 9.423255920410156e-06,
      "model_forward_time": 5.604551315307617,
      "step": 9881
    },
    {
      "epoch": 9.423255920410156e-06,
      "step": 9881,
      "training_step_time": 17.09365940093994
    },
    {
      "epoch": 9.424209594726562e-06,
      "model_forward_time": 5.605879783630371,
      "step": 9882
    },
    {
      "epoch": 9.424209594726562e-06,
      "step": 9882,
      "training_step_time": 17.440492391586304
    },
    {
      "epoch": 9.425163269042968e-06,
      "model_forward_time": 5.6052024364471436,
      "step": 9883
    },
    {
      "epoch": 9.425163269042968e-06,
      "step": 9883,
      "training_step_time": 17.073533296585083
    },
    {
      "epoch": 9.426116943359375e-06,
      "model_forward_time": 5.59234094619751,
      "step": 9884
    },
    {
      "epoch": 9.426116943359375e-06,
      "step": 9884,
      "training_step_time": 17.057114601135254
    },
    {
      "epoch": 9.427070617675781e-06,
      "model_forward_time": 5.589829921722412,
      "step": 9885
    },
    {
      "epoch": 9.427070617675781e-06,
      "step": 9885,
      "training_step_time": 17.047781944274902
    },
    {
      "epoch": 9.428024291992187e-06,
      "model_forward_time": 5.593778371810913,
      "step": 9886
    },
    {
      "epoch": 9.428024291992187e-06,
      "step": 9886,
      "training_step_time": 17.064697265625
    },
    {
      "epoch": 9.428977966308594e-06,
      "model_forward_time": 5.598322153091431,
      "step": 9887
    },
    {
      "epoch": 9.428977966308594e-06,
      "step": 9887,
      "training_step_time": 17.082823514938354
    },
    {
      "epoch": 9.429931640625e-06,
      "model_forward_time": 5.966701984405518,
      "step": 9888
    },
    {
      "epoch": 9.429931640625e-06,
      "step": 9888,
      "training_step_time": 17.45851469039917
    },
    {
      "epoch": 9.430885314941406e-06,
      "model_forward_time": 5.6095030307769775,
      "step": 9889
    },
    {
      "epoch": 9.430885314941406e-06,
      "step": 9889,
      "training_step_time": 17.079315662384033
    },
    {
      "epoch": 9.431838989257813e-06,
      "grad_norm": 1.6221367120742798,
      "learning_rate": 1e-05,
      "loss": 0.0696,
      "step": 9890
    },
    {
      "epoch": 9.431838989257813e-06,
      "model_forward_time": 5.592235088348389,
      "step": 9890
    },
    {
      "dynamics_loss_avg": 0.061825820989906785,
      "epoch": 9.431838989257813e-06,
      "step": 9890
    },
    {
      "action_loss_avg": 0.008543814707081765,
      "epoch": 9.431838989257813e-06,
      "step": 9890
    },
    {
      "epoch": 9.431838989257813e-06,
      "step": 9890,
      "training_step_time": 17.01522397994995
    },
    {
      "epoch": 9.432792663574219e-06,
      "model_forward_time": 5.577333450317383,
      "step": 9891
    },
    {
      "epoch": 9.432792663574219e-06,
      "step": 9891,
      "training_step_time": 17.027231454849243
    },
    {
      "epoch": 9.433746337890624e-06,
      "model_forward_time": 5.579824447631836,
      "step": 9892
    },
    {
      "epoch": 9.433746337890624e-06,
      "step": 9892,
      "training_step_time": 17.0004563331604
    },
    {
      "epoch": 9.434700012207032e-06,
      "model_forward_time": 5.9841628074646,
      "step": 9893
    },
    {
      "epoch": 9.434700012207032e-06,
      "step": 9893,
      "training_step_time": 17.403497219085693
    },
    {
      "epoch": 9.435653686523438e-06,
      "model_forward_time": 5.631212949752808,
      "step": 9894
    },
    {
      "epoch": 9.435653686523438e-06,
      "step": 9894,
      "training_step_time": 17.11378502845764
    },
    {
      "epoch": 9.436607360839843e-06,
      "model_forward_time": 5.587016820907593,
      "step": 9895
    },
    {
      "epoch": 9.436607360839843e-06,
      "step": 9895,
      "training_step_time": 17.040964126586914
    },
    {
      "epoch": 9.43756103515625e-06,
      "model_forward_time": 5.613812446594238,
      "step": 9896
    },
    {
      "epoch": 9.43756103515625e-06,
      "step": 9896,
      "training_step_time": 17.095982313156128
    },
    {
      "epoch": 9.438514709472656e-06,
      "model_forward_time": 5.605076789855957,
      "step": 9897
    },
    {
      "epoch": 9.438514709472656e-06,
      "step": 9897,
      "training_step_time": 17.091483116149902
    },
    {
      "epoch": 9.439468383789062e-06,
      "model_forward_time": 5.599774360656738,
      "step": 9898
    },
    {
      "epoch": 9.439468383789062e-06,
      "step": 9898,
      "training_step_time": 17.410683631896973
    },
    {
      "epoch": 9.440422058105468e-06,
      "model_forward_time": 5.58764386177063,
      "step": 9899
    },
    {
      "epoch": 9.440422058105468e-06,
      "step": 9899,
      "training_step_time": 17.052008390426636
    },
    {
      "epoch": 9.441375732421875e-06,
      "grad_norm": 1.071824550628662,
      "learning_rate": 1e-05,
      "loss": 0.0796,
      "step": 9900
    },
    {
      "epoch": 9.441375732421875e-06,
      "model_forward_time": 5.5964674949646,
      "step": 9900
    },
    {
      "dynamics_loss_avg": 0.07057481594383716,
      "epoch": 9.441375732421875e-06,
      "step": 9900
    },
    {
      "action_loss_avg": 0.009240617998875678,
      "epoch": 9.441375732421875e-06,
      "step": 9900
    },
    {
      "epoch": 9.441375732421875e-06,
      "step": 9900,
      "training_step_time": 17.07326340675354
    },
    {
      "epoch": 9.442329406738281e-06,
      "model_forward_time": 5.606830596923828,
      "step": 9901
    },
    {
      "epoch": 9.442329406738281e-06,
      "step": 9901,
      "training_step_time": 17.08488941192627
    },
    {
      "epoch": 9.443283081054687e-06,
      "model_forward_time": 5.6104960441589355,
      "step": 9902
    },
    {
      "epoch": 9.443283081054687e-06,
      "step": 9902,
      "training_step_time": 17.11207365989685
    },
    {
      "epoch": 9.444236755371094e-06,
      "model_forward_time": 5.62315559387207,
      "step": 9903
    },
    {
      "epoch": 9.444236755371094e-06,
      "step": 9903,
      "training_step_time": 17.13624906539917
    },
    {
      "epoch": 9.4451904296875e-06,
      "model_forward_time": 5.994848728179932,
      "step": 9904
    },
    {
      "epoch": 9.4451904296875e-06,
      "step": 9904,
      "training_step_time": 17.50959825515747
    },
    {
      "epoch": 9.446144104003906e-06,
      "model_forward_time": 5.615774393081665,
      "step": 9905
    },
    {
      "epoch": 9.446144104003906e-06,
      "step": 9905,
      "training_step_time": 17.099257230758667
    },
    {
      "epoch": 9.447097778320313e-06,
      "model_forward_time": 5.601175546646118,
      "step": 9906
    },
    {
      "epoch": 9.447097778320313e-06,
      "step": 9906,
      "training_step_time": 17.043331146240234
    },
    {
      "epoch": 9.448051452636719e-06,
      "model_forward_time": 5.582440376281738,
      "step": 9907
    },
    {
      "epoch": 9.448051452636719e-06,
      "step": 9907,
      "training_step_time": 16.98736047744751
    },
    {
      "epoch": 9.449005126953124e-06,
      "model_forward_time": 5.587499618530273,
      "step": 9908
    },
    {
      "epoch": 9.449005126953124e-06,
      "step": 9908,
      "training_step_time": 16.995662450790405
    },
    {
      "epoch": 9.449958801269532e-06,
      "model_forward_time": 5.946527481079102,
      "step": 9909
    },
    {
      "epoch": 9.449958801269532e-06,
      "step": 9909,
      "training_step_time": 17.374505758285522
    },
    {
      "epoch": 9.450912475585938e-06,
      "grad_norm": 1.0524938106536865,
      "learning_rate": 1e-05,
      "loss": 0.0746,
      "step": 9910
    },
    {
      "epoch": 9.450912475585938e-06,
      "model_forward_time": 5.58235502243042,
      "step": 9910
    },
    {
      "dynamics_loss_avg": 0.061079572141170504,
      "epoch": 9.450912475585938e-06,
      "step": 9910
    },
    {
      "action_loss_avg": 0.011814035847783089,
      "epoch": 9.450912475585938e-06,
      "step": 9910
    },
    {
      "epoch": 9.450912475585938e-06,
      "step": 9910,
      "training_step_time": 17.024297952651978
    },
    {
      "epoch": 9.451866149902343e-06,
      "model_forward_time": 5.587352514266968,
      "step": 9911
    },
    {
      "epoch": 9.451866149902343e-06,
      "step": 9911,
      "training_step_time": 17.02364182472229
    },
    {
      "epoch": 9.45281982421875e-06,
      "model_forward_time": 5.59894061088562,
      "step": 9912
    },
    {
      "epoch": 9.45281982421875e-06,
      "step": 9912,
      "training_step_time": 17.029069423675537
    },
    {
      "epoch": 9.453773498535156e-06,
      "model_forward_time": 5.594654560089111,
      "step": 9913
    },
    {
      "epoch": 9.453773498535156e-06,
      "step": 9913,
      "training_step_time": 17.06231188774109
    },
    {
      "epoch": 9.454727172851562e-06,
      "model_forward_time": 5.605601787567139,
      "step": 9914
    },
    {
      "epoch": 9.454727172851562e-06,
      "step": 9914,
      "training_step_time": 17.079806804656982
    },
    {
      "epoch": 9.455680847167968e-06,
      "model_forward_time": 5.99094295501709,
      "step": 9915
    },
    {
      "epoch": 9.455680847167968e-06,
      "step": 9915,
      "training_step_time": 17.448984146118164
    },
    {
      "epoch": 9.456634521484375e-06,
      "model_forward_time": 5.594254732131958,
      "step": 9916
    },
    {
      "epoch": 9.456634521484375e-06,
      "step": 9916,
      "training_step_time": 17.04509973526001
    },
    {
      "epoch": 9.457588195800781e-06,
      "model_forward_time": 5.5951995849609375,
      "step": 9917
    },
    {
      "epoch": 9.457588195800781e-06,
      "step": 9917,
      "training_step_time": 17.045342922210693
    },
    {
      "epoch": 9.458541870117187e-06,
      "model_forward_time": 5.606328725814819,
      "step": 9918
    },
    {
      "epoch": 9.458541870117187e-06,
      "step": 9918,
      "training_step_time": 17.07783532142639
    },
    {
      "epoch": 9.459495544433594e-06,
      "model_forward_time": 5.594754695892334,
      "step": 9919
    },
    {
      "epoch": 9.459495544433594e-06,
      "step": 9919,
      "training_step_time": 17.075581550598145
    },
    {
      "epoch": 9.46044921875e-06,
      "grad_norm": 0.5563841462135315,
      "learning_rate": 1e-05,
      "loss": 0.0609,
      "step": 9920
    },
    {
      "epoch": 9.46044921875e-06,
      "model_forward_time": 5.973313570022583,
      "step": 9920
    },
    {
      "dynamics_loss_avg": 0.05173095967620611,
      "epoch": 9.46044921875e-06,
      "step": 9920
    },
    {
      "action_loss_avg": 0.00884083758573979,
      "epoch": 9.46044921875e-06,
      "step": 9920
    },
    {
      "epoch": 9.46044921875e-06,
      "step": 9920,
      "training_step_time": 17.45029854774475
    },
    {
      "epoch": 9.461402893066406e-06,
      "model_forward_time": 5.60423731803894,
      "step": 9921
    },
    {
      "epoch": 9.461402893066406e-06,
      "step": 9921,
      "training_step_time": 17.094563007354736
    },
    {
      "epoch": 9.462356567382813e-06,
      "model_forward_time": 5.6127471923828125,
      "step": 9922
    },
    {
      "epoch": 9.462356567382813e-06,
      "step": 9922,
      "training_step_time": 17.114554405212402
    },
    {
      "epoch": 9.463310241699219e-06,
      "model_forward_time": 5.623419523239136,
      "step": 9923
    },
    {
      "epoch": 9.463310241699219e-06,
      "step": 9923,
      "training_step_time": 17.11789584159851
    },
    {
      "epoch": 9.464263916015625e-06,
      "model_forward_time": 5.606866121292114,
      "step": 9924
    },
    {
      "epoch": 9.464263916015625e-06,
      "step": 9924,
      "training_step_time": 17.094391107559204
    },
    {
      "epoch": 9.465217590332032e-06,
      "model_forward_time": 5.617451429367065,
      "step": 9925
    },
    {
      "epoch": 9.465217590332032e-06,
      "step": 9925,
      "training_step_time": 17.092816829681396
    },
    {
      "epoch": 9.466171264648438e-06,
      "model_forward_time": 5.975957155227661,
      "step": 9926
    },
    {
      "epoch": 9.466171264648438e-06,
      "step": 9926,
      "training_step_time": 17.457415103912354
    },
    {
      "epoch": 9.467124938964843e-06,
      "model_forward_time": 5.6262781620025635,
      "step": 9927
    },
    {
      "epoch": 9.467124938964843e-06,
      "step": 9927,
      "training_step_time": 17.08131241798401
    },
    {
      "epoch": 9.46807861328125e-06,
      "model_forward_time": 5.597300052642822,
      "step": 9928
    },
    {
      "epoch": 9.46807861328125e-06,
      "step": 9928,
      "training_step_time": 17.059669971466064
    },
    {
      "epoch": 9.469032287597657e-06,
      "model_forward_time": 5.599578380584717,
      "step": 9929
    },
    {
      "epoch": 9.469032287597657e-06,
      "step": 9929,
      "training_step_time": 17.057493686676025
    },
    {
      "epoch": 9.469985961914062e-06,
      "grad_norm": 1.3824021816253662,
      "learning_rate": 1e-05,
      "loss": 0.0695,
      "step": 9930
    },
    {
      "epoch": 9.469985961914062e-06,
      "model_forward_time": 5.591497182846069,
      "step": 9930
    },
    {
      "dynamics_loss_avg": 0.05986604243516922,
      "epoch": 9.469985961914062e-06,
      "step": 9930
    },
    {
      "action_loss_avg": 0.008341576671227813,
      "epoch": 9.469985961914062e-06,
      "step": 9930
    },
    {
      "epoch": 9.469985961914062e-06,
      "step": 9930,
      "training_step_time": 17.08150339126587
    },
    {
      "epoch": 9.470939636230468e-06,
      "model_forward_time": 5.976622819900513,
      "step": 9931
    },
    {
      "epoch": 9.470939636230468e-06,
      "step": 9931,
      "training_step_time": 17.495720863342285
    },
    {
      "epoch": 9.471893310546875e-06,
      "model_forward_time": 5.614832162857056,
      "step": 9932
    },
    {
      "epoch": 9.471893310546875e-06,
      "step": 9932,
      "training_step_time": 17.141008853912354
    },
    {
      "epoch": 9.472846984863281e-06,
      "model_forward_time": 5.6323254108428955,
      "step": 9933
    },
    {
      "epoch": 9.472846984863281e-06,
      "step": 9933,
      "training_step_time": 17.14524245262146
    },
    {
      "epoch": 9.473800659179687e-06,
      "model_forward_time": 5.609583377838135,
      "step": 9934
    },
    {
      "epoch": 9.473800659179687e-06,
      "step": 9934,
      "training_step_time": 17.076873540878296
    },
    {
      "epoch": 9.474754333496094e-06,
      "model_forward_time": 5.589494705200195,
      "step": 9935
    },
    {
      "epoch": 9.474754333496094e-06,
      "step": 9935,
      "training_step_time": 17.005282402038574
    },
    {
      "epoch": 9.4757080078125e-06,
      "model_forward_time": 5.56967568397522,
      "step": 9936
    },
    {
      "epoch": 9.4757080078125e-06,
      "step": 9936,
      "training_step_time": 17.355398893356323
    },
    {
      "epoch": 9.476661682128906e-06,
      "model_forward_time": 5.567251443862915,
      "step": 9937
    },
    {
      "epoch": 9.476661682128906e-06,
      "step": 9937,
      "training_step_time": 16.99476146697998
    },
    {
      "epoch": 9.477615356445313e-06,
      "model_forward_time": 5.5610671043396,
      "step": 9938
    },
    {
      "epoch": 9.477615356445313e-06,
      "step": 9938,
      "training_step_time": 16.976519107818604
    },
    {
      "epoch": 9.478569030761719e-06,
      "model_forward_time": 5.586100101470947,
      "step": 9939
    },
    {
      "epoch": 9.478569030761719e-06,
      "step": 9939,
      "training_step_time": 17.00075125694275
    },
    {
      "epoch": 9.479522705078125e-06,
      "grad_norm": 0.39983415603637695,
      "learning_rate": 1e-05,
      "loss": 0.0639,
      "step": 9940
    },
    {
      "epoch": 9.479522705078125e-06,
      "model_forward_time": 5.5804362297058105,
      "step": 9940
    },
    {
      "dynamics_loss_avg": 0.057401510700583455,
      "epoch": 9.479522705078125e-06,
      "step": 9940
    },
    {
      "action_loss_avg": 0.008435429632663726,
      "epoch": 9.479522705078125e-06,
      "step": 9940
    },
    {
      "epoch": 9.479522705078125e-06,
      "step": 9940,
      "training_step_time": 17.02414894104004
    },
    {
      "epoch": 9.480476379394532e-06,
      "model_forward_time": 5.580966472625732,
      "step": 9941
    },
    {
      "epoch": 9.480476379394532e-06,
      "step": 9941,
      "training_step_time": 17.006468296051025
    },
    {
      "epoch": 9.481430053710938e-06,
      "model_forward_time": 5.930370092391968,
      "step": 9942
    },
    {
      "epoch": 9.481430053710938e-06,
      "step": 9942,
      "training_step_time": 17.364697217941284
    },
    {
      "epoch": 9.482383728027343e-06,
      "model_forward_time": 5.591668128967285,
      "step": 9943
    },
    {
      "epoch": 9.482383728027343e-06,
      "step": 9943,
      "training_step_time": 17.044684886932373
    },
    {
      "epoch": 9.48333740234375e-06,
      "model_forward_time": 5.609419822692871,
      "step": 9944
    },
    {
      "epoch": 9.48333740234375e-06,
      "step": 9944,
      "training_step_time": 17.086050033569336
    },
    {
      "epoch": 9.484291076660157e-06,
      "model_forward_time": 5.618678569793701,
      "step": 9945
    },
    {
      "epoch": 9.484291076660157e-06,
      "step": 9945,
      "training_step_time": 17.129554748535156
    },
    {
      "epoch": 9.485244750976562e-06,
      "model_forward_time": 5.629475116729736,
      "step": 9946
    },
    {
      "epoch": 9.485244750976562e-06,
      "step": 9946,
      "training_step_time": 17.13829016685486
    },
    {
      "epoch": 9.486198425292968e-06,
      "model_forward_time": 5.986626386642456,
      "step": 9947
    },
    {
      "epoch": 9.486198425292968e-06,
      "step": 9947,
      "training_step_time": 17.465718746185303
    },
    {
      "epoch": 9.487152099609375e-06,
      "model_forward_time": 5.60737419128418,
      "step": 9948
    },
    {
      "epoch": 9.487152099609375e-06,
      "step": 9948,
      "training_step_time": 17.0886070728302
    },
    {
      "epoch": 9.488105773925781e-06,
      "model_forward_time": 5.600054502487183,
      "step": 9949
    },
    {
      "epoch": 9.488105773925781e-06,
      "step": 9949,
      "training_step_time": 17.056498050689697
    },
    {
      "epoch": 9.489059448242187e-06,
      "grad_norm": 0.894644558429718,
      "learning_rate": 1e-05,
      "loss": 0.0765,
      "step": 9950
    },
    {
      "epoch": 9.489059448242187e-06,
      "model_forward_time": 5.592706203460693,
      "step": 9950
    },
    {
      "dynamics_loss_avg": 0.06734748482704163,
      "epoch": 9.489059448242187e-06,
      "step": 9950
    },
    {
      "action_loss_avg": 0.011914596054702997,
      "epoch": 9.489059448242187e-06,
      "step": 9950
    },
    {
      "epoch": 9.489059448242187e-06,
      "step": 9950,
      "training_step_time": 17.039575338363647
    },
    {
      "epoch": 9.490013122558594e-06,
      "model_forward_time": 5.590134620666504,
      "step": 9951
    },
    {
      "epoch": 9.490013122558594e-06,
      "step": 9951,
      "training_step_time": 17.0213041305542
    },
    {
      "epoch": 9.490966796875e-06,
      "model_forward_time": 5.584541320800781,
      "step": 9952
    },
    {
      "epoch": 9.490966796875e-06,
      "step": 9952,
      "training_step_time": 17.033758878707886
    },
    {
      "epoch": 9.491920471191406e-06,
      "model_forward_time": 5.963220119476318,
      "step": 9953
    },
    {
      "epoch": 9.491920471191406e-06,
      "step": 9953,
      "training_step_time": 17.428242683410645
    },
    {
      "epoch": 9.492874145507813e-06,
      "model_forward_time": 5.610981464385986,
      "step": 9954
    },
    {
      "epoch": 9.492874145507813e-06,
      "step": 9954,
      "training_step_time": 17.087584018707275
    },
    {
      "epoch": 9.493827819824219e-06,
      "model_forward_time": 5.600070238113403,
      "step": 9955
    },
    {
      "epoch": 9.493827819824219e-06,
      "step": 9955,
      "training_step_time": 17.0904438495636
    },
    {
      "epoch": 9.494781494140625e-06,
      "model_forward_time": 5.615187406539917,
      "step": 9956
    },
    {
      "epoch": 9.494781494140625e-06,
      "step": 9956,
      "training_step_time": 17.10178852081299
    },
    {
      "epoch": 9.495735168457032e-06,
      "model_forward_time": 5.613164663314819,
      "step": 9957
    },
    {
      "epoch": 9.495735168457032e-06,
      "step": 9957,
      "training_step_time": 17.059832334518433
    },
    {
      "epoch": 9.496688842773438e-06,
      "model_forward_time": 5.970370531082153,
      "step": 9958
    },
    {
      "epoch": 9.496688842773438e-06,
      "step": 9958,
      "training_step_time": 17.40648365020752
    },
    {
      "epoch": 9.497642517089843e-06,
      "model_forward_time": 5.5968101024627686,
      "step": 9959
    },
    {
      "epoch": 9.497642517089843e-06,
      "step": 9959,
      "training_step_time": 17.045534133911133
    },
    {
      "epoch": 9.49859619140625e-06,
      "grad_norm": 0.9882256984710693,
      "learning_rate": 1e-05,
      "loss": 0.0806,
      "step": 9960
    },
    {
      "epoch": 9.49859619140625e-06,
      "model_forward_time": 5.599756479263306,
      "step": 9960
    },
    {
      "dynamics_loss_avg": 0.07062863223254681,
      "epoch": 9.49859619140625e-06,
      "step": 9960
    },
    {
      "action_loss_avg": 0.011333528580144048,
      "epoch": 9.49859619140625e-06,
      "step": 9960
    },
    {
      "epoch": 9.49859619140625e-06,
      "step": 9960,
      "training_step_time": 17.083513021469116
    },
    {
      "epoch": 9.499549865722657e-06,
      "model_forward_time": 5.619405746459961,
      "step": 9961
    },
    {
      "epoch": 9.499549865722657e-06,
      "step": 9961,
      "training_step_time": 17.094380617141724
    },
    {
      "epoch": 9.500503540039062e-06,
      "model_forward_time": 5.6076819896698,
      "step": 9962
    },
    {
      "epoch": 9.500503540039062e-06,
      "step": 9962,
      "training_step_time": 17.07315707206726
    },
    {
      "epoch": 9.501457214355468e-06,
      "model_forward_time": 5.595071792602539,
      "step": 9963
    },
    {
      "epoch": 9.501457214355468e-06,
      "step": 9963,
      "training_step_time": 17.036272287368774
    },
    {
      "epoch": 9.502410888671875e-06,
      "model_forward_time": 5.936016798019409,
      "step": 9964
    },
    {
      "epoch": 9.502410888671875e-06,
      "step": 9964,
      "training_step_time": 17.36790895462036
    },
    {
      "epoch": 9.503364562988281e-06,
      "model_forward_time": 5.587677478790283,
      "step": 9965
    },
    {
      "epoch": 9.503364562988281e-06,
      "step": 9965,
      "training_step_time": 17.030092477798462
    },
    {
      "epoch": 9.504318237304687e-06,
      "model_forward_time": 5.595973253250122,
      "step": 9966
    },
    {
      "epoch": 9.504318237304687e-06,
      "step": 9966,
      "training_step_time": 17.05297827720642
    },
    {
      "epoch": 9.505271911621094e-06,
      "model_forward_time": 5.608711242675781,
      "step": 9967
    },
    {
      "epoch": 9.505271911621094e-06,
      "step": 9967,
      "training_step_time": 17.08344316482544
    },
    {
      "epoch": 9.5062255859375e-06,
      "model_forward_time": 5.599928140640259,
      "step": 9968
    },
    {
      "epoch": 9.5062255859375e-06,
      "step": 9968,
      "training_step_time": 17.05891251564026
    },
    {
      "epoch": 9.507179260253906e-06,
      "model_forward_time": 5.978595972061157,
      "step": 9969
    },
    {
      "epoch": 9.507179260253906e-06,
      "step": 9969,
      "training_step_time": 17.447917222976685
    },
    {
      "epoch": 9.508132934570313e-06,
      "grad_norm": 0.7218726277351379,
      "learning_rate": 1e-05,
      "loss": 0.0679,
      "step": 9970
    },
    {
      "epoch": 9.508132934570313e-06,
      "model_forward_time": 5.596991062164307,
      "step": 9970
    },
    {
      "dynamics_loss_avg": 0.0568948782980442,
      "epoch": 9.508132934570313e-06,
      "step": 9970
    },
    {
      "action_loss_avg": 0.008778695738874376,
      "epoch": 9.508132934570313e-06,
      "step": 9970
    },
    {
      "epoch": 9.508132934570313e-06,
      "step": 9970,
      "training_step_time": 17.06861662864685
    },
    {
      "epoch": 9.509086608886719e-06,
      "model_forward_time": 5.614132881164551,
      "step": 9971
    },
    {
      "epoch": 9.509086608886719e-06,
      "step": 9971,
      "training_step_time": 17.10009241104126
    },
    {
      "epoch": 9.510040283203125e-06,
      "model_forward_time": 5.617789268493652,
      "step": 9972
    },
    {
      "epoch": 9.510040283203125e-06,
      "step": 9972,
      "training_step_time": 17.130889892578125
    },
    {
      "epoch": 9.510993957519532e-06,
      "model_forward_time": 5.620010852813721,
      "step": 9973
    },
    {
      "epoch": 9.510993957519532e-06,
      "step": 9973,
      "training_step_time": 17.118748426437378
    },
    {
      "epoch": 9.511947631835938e-06,
      "model_forward_time": 5.597510099411011,
      "step": 9974
    },
    {
      "epoch": 9.511947631835938e-06,
      "step": 9974,
      "training_step_time": 17.036524057388306
    },
    {
      "epoch": 9.512901306152343e-06,
      "model_forward_time": 5.948055028915405,
      "step": 9975
    },
    {
      "epoch": 9.512901306152343e-06,
      "step": 9975,
      "training_step_time": 17.3742733001709
    },
    {
      "epoch": 9.51385498046875e-06,
      "model_forward_time": 5.580175161361694,
      "step": 9976
    },
    {
      "epoch": 9.51385498046875e-06,
      "step": 9976,
      "training_step_time": 16.987549781799316
    },
    {
      "epoch": 9.514808654785157e-06,
      "model_forward_time": 5.585484504699707,
      "step": 9977
    },
    {
      "epoch": 9.514808654785157e-06,
      "step": 9977,
      "training_step_time": 17.02690291404724
    },
    {
      "epoch": 9.515762329101562e-06,
      "model_forward_time": 5.601664066314697,
      "step": 9978
    },
    {
      "epoch": 9.515762329101562e-06,
      "step": 9978,
      "training_step_time": 17.085695028305054
    },
    {
      "epoch": 9.516716003417968e-06,
      "model_forward_time": 5.620016574859619,
      "step": 9979
    },
    {
      "epoch": 9.516716003417968e-06,
      "step": 9979,
      "training_step_time": 17.13360357284546
    },
    {
      "epoch": 9.517669677734375e-06,
      "grad_norm": 1.0633076429367065,
      "learning_rate": 1e-05,
      "loss": 0.0831,
      "step": 9980
    },
    {
      "epoch": 9.517669677734375e-06,
      "model_forward_time": 5.997506856918335,
      "step": 9980
    },
    {
      "dynamics_loss_avg": 0.06967744734138251,
      "epoch": 9.517669677734375e-06,
      "step": 9980
    },
    {
      "action_loss_avg": 0.01205583573319018,
      "epoch": 9.517669677734375e-06,
      "step": 9980
    },
    {
      "epoch": 9.517669677734375e-06,
      "step": 9980,
      "training_step_time": 17.517414093017578
    },
    {
      "epoch": 9.518623352050781e-06,
      "model_forward_time": 5.618204593658447,
      "step": 9981
    },
    {
      "epoch": 9.518623352050781e-06,
      "step": 9981,
      "training_step_time": 17.132338285446167
    },
    {
      "epoch": 9.519577026367187e-06,
      "model_forward_time": 5.627139091491699,
      "step": 9982
    },
    {
      "epoch": 9.519577026367187e-06,
      "step": 9982,
      "training_step_time": 17.126904249191284
    },
    {
      "epoch": 9.520530700683594e-06,
      "model_forward_time": 5.6183693408966064,
      "step": 9983
    },
    {
      "epoch": 9.520530700683594e-06,
      "step": 9983,
      "training_step_time": 17.110050201416016
    },
    {
      "epoch": 9.521484375e-06,
      "model_forward_time": 5.614171028137207,
      "step": 9984
    },
    {
      "epoch": 9.521484375e-06,
      "step": 9984,
      "training_step_time": 17.09182071685791
    },
    {
      "epoch": 9.522438049316406e-06,
      "model_forward_time": 5.96673583984375,
      "step": 9985
    },
    {
      "epoch": 9.522438049316406e-06,
      "step": 9985,
      "training_step_time": 17.451013326644897
    },
    {
      "epoch": 9.523391723632813e-06,
      "model_forward_time": 5.60992693901062,
      "step": 9986
    },
    {
      "epoch": 9.523391723632813e-06,
      "step": 9986,
      "training_step_time": 17.101633071899414
    },
    {
      "epoch": 9.524345397949219e-06,
      "model_forward_time": 5.609545469284058,
      "step": 9987
    },
    {
      "epoch": 9.524345397949219e-06,
      "step": 9987,
      "training_step_time": 17.108648777008057
    },
    {
      "epoch": 9.525299072265625e-06,
      "model_forward_time": 5.6127564907073975,
      "step": 9988
    },
    {
      "epoch": 9.525299072265625e-06,
      "step": 9988,
      "training_step_time": 17.100921869277954
    },
    {
      "epoch": 9.526252746582032e-06,
      "model_forward_time": 5.610222578048706,
      "step": 9989
    },
    {
      "epoch": 9.526252746582032e-06,
      "step": 9989,
      "training_step_time": 17.07864475250244
    },
    {
      "epoch": 9.527206420898438e-06,
      "grad_norm": 1.430283784866333,
      "learning_rate": 1e-05,
      "loss": 0.0659,
      "step": 9990
    },
    {
      "epoch": 9.527206420898438e-06,
      "model_forward_time": 5.612605810165405,
      "step": 9990
    },
    {
      "dynamics_loss_avg": 0.05617326833307743,
      "epoch": 9.527206420898438e-06,
      "step": 9990
    },
    {
      "action_loss_avg": 0.008215385396033526,
      "epoch": 9.527206420898438e-06,
      "step": 9990
    },
    {
      "epoch": 9.527206420898438e-06,
      "step": 9990,
      "training_step_time": 17.09060835838318
    },
    {
      "epoch": 9.528160095214843e-06,
      "model_forward_time": 5.9911675453186035,
      "step": 9991
    },
    {
      "epoch": 9.528160095214843e-06,
      "step": 9991,
      "training_step_time": 17.46539807319641
    },
    {
      "epoch": 9.52911376953125e-06,
      "model_forward_time": 5.601731538772583,
      "step": 9992
    },
    {
      "epoch": 9.52911376953125e-06,
      "step": 9992,
      "training_step_time": 17.083605766296387
    },
    {
      "epoch": 9.530067443847657e-06,
      "model_forward_time": 5.614915132522583,
      "step": 9993
    },
    {
      "epoch": 9.530067443847657e-06,
      "step": 9993,
      "training_step_time": 17.105555772781372
    },
    {
      "epoch": 9.531021118164062e-06,
      "model_forward_time": 5.601457118988037,
      "step": 9994
    },
    {
      "epoch": 9.531021118164062e-06,
      "step": 9994,
      "training_step_time": 17.05640959739685
    },
    {
      "epoch": 9.531974792480468e-06,
      "model_forward_time": 5.596137762069702,
      "step": 9995
    },
    {
      "epoch": 9.531974792480468e-06,
      "step": 9995,
      "training_step_time": 17.06766653060913
    },
    {
      "epoch": 9.532928466796875e-06,
      "model_forward_time": 5.601380348205566,
      "step": 9996
    },
    {
      "epoch": 9.532928466796875e-06,
      "step": 9996,
      "training_step_time": 17.44891047477722
    },
    {
      "epoch": 9.533882141113281e-06,
      "model_forward_time": 5.599016904830933,
      "step": 9997
    },
    {
      "epoch": 9.533882141113281e-06,
      "step": 9997,
      "training_step_time": 17.028817892074585
    },
    {
      "epoch": 9.534835815429687e-06,
      "model_forward_time": 5.5858728885650635,
      "step": 9998
    },
    {
      "epoch": 9.534835815429687e-06,
      "step": 9998,
      "training_step_time": 17.035388946533203
    },
    {
      "epoch": 9.535789489746094e-06,
      "model_forward_time": 5.576011896133423,
      "step": 9999
    },
    {
      "epoch": 9.535789489746094e-06,
      "step": 9999,
      "training_step_time": 17.002803564071655
    },
    {
      "epoch": 9.5367431640625e-06,
      "grad_norm": 1.1290736198425293,
      "learning_rate": 1e-05,
      "loss": 0.0675,
      "step": 10000
    },
    {
      "epoch": 9.5367431640625e-06,
      "model_forward_time": 5.797271490097046,
      "step": 10000
    },
    {
      "dynamics_loss_avg": 0.06088221855461597,
      "epoch": 9.5367431640625e-06,
      "step": 10000
    },
    {
      "action_loss_avg": 0.008675530389882623,
      "epoch": 9.5367431640625e-06,
      "step": 10000
    },
    {
      "epoch": 9.5367431640625e-06,
      "step": 10000,
      "training_step_time": 17.107303142547607
    },
    {
      "epoch": 9.537696838378906e-06,
      "model_forward_time": 6.026812314987183,
      "step": 10001
    },
    {
      "epoch": 9.537696838378906e-06,
      "step": 10001,
      "training_step_time": 17.335841417312622
    },
    {
      "epoch": 9.538650512695313e-06,
      "model_forward_time": 5.536050081253052,
      "step": 10002
    },
    {
      "epoch": 9.538650512695313e-06,
      "step": 10002,
      "training_step_time": 16.888596296310425
    },
    {
      "epoch": 9.539604187011719e-06,
      "model_forward_time": 5.560842037200928,
      "step": 10003
    },
    {
      "epoch": 9.539604187011719e-06,
      "step": 10003,
      "training_step_time": 16.950345993041992
    },
    {
      "epoch": 9.540557861328125e-06,
      "model_forward_time": 5.561765193939209,
      "step": 10004
    },
    {
      "epoch": 9.540557861328125e-06,
      "step": 10004,
      "training_step_time": 16.984756231307983
    },
    {
      "epoch": 9.541511535644532e-06,
      "model_forward_time": 5.563378810882568,
      "step": 10005
    },
    {
      "epoch": 9.541511535644532e-06,
      "step": 10005,
      "training_step_time": 16.984871864318848
    },
    {
      "epoch": 9.542465209960938e-06,
      "model_forward_time": 5.5831098556518555,
      "step": 10006
    },
    {
      "epoch": 9.542465209960938e-06,
      "step": 10006,
      "training_step_time": 17.400944471359253
    },
    {
      "epoch": 9.543418884277343e-06,
      "model_forward_time": 5.592383146286011,
      "step": 10007
    },
    {
      "epoch": 9.543418884277343e-06,
      "step": 10007,
      "training_step_time": 17.068078756332397
    },
    {
      "epoch": 9.54437255859375e-06,
      "model_forward_time": 5.616346836090088,
      "step": 10008
    },
    {
      "epoch": 9.54437255859375e-06,
      "step": 10008,
      "training_step_time": 17.1051287651062
    },
    {
      "epoch": 9.545326232910157e-06,
      "model_forward_time": 5.618687868118286,
      "step": 10009
    },
    {
      "epoch": 9.545326232910157e-06,
      "step": 10009,
      "training_step_time": 17.1109459400177
    },
    {
      "epoch": 9.546279907226562e-06,
      "grad_norm": 1.0597461462020874,
      "learning_rate": 1e-05,
      "loss": 0.0643,
      "step": 10010
    },
    {
      "epoch": 9.546279907226562e-06,
      "model_forward_time": 5.607135772705078,
      "step": 10010
    },
    {
      "dynamics_loss_avg": 0.055030678398907185,
      "epoch": 9.546279907226562e-06,
      "step": 10010
    },
    {
      "action_loss_avg": 0.008982847025617958,
      "epoch": 9.546279907226562e-06,
      "step": 10010
    },
    {
      "epoch": 9.546279907226562e-06,
      "step": 10010,
      "training_step_time": 17.082067012786865
    },
    {
      "epoch": 9.547233581542968e-06,
      "model_forward_time": 4.277986764907837,
      "step": 10011
    },
    {
      "epoch": 9.547233581542968e-06,
      "step": 10011,
      "training_step_time": 12.785128355026245
    },
    {
      "epoch": 9.548187255859375e-06,
      "model_forward_time": 5.945271015167236,
      "step": 10012
    },
    {
      "epoch": 9.548187255859375e-06,
      "step": 10012,
      "training_step_time": 17.39556097984314
    },
    {
      "epoch": 9.549140930175781e-06,
      "model_forward_time": 5.583221197128296,
      "step": 10013
    },
    {
      "epoch": 9.549140930175781e-06,
      "step": 10013,
      "training_step_time": 17.03390121459961
    },
    {
      "epoch": 9.550094604492187e-06,
      "model_forward_time": 5.602663278579712,
      "step": 10014
    },
    {
      "epoch": 9.550094604492187e-06,
      "step": 10014,
      "training_step_time": 17.073853969573975
    },
    {
      "epoch": 9.551048278808594e-06,
      "model_forward_time": 5.607161521911621,
      "step": 10015
    },
    {
      "epoch": 9.551048278808594e-06,
      "step": 10015,
      "training_step_time": 17.080815315246582
    },
    {
      "epoch": 9.552001953125e-06,
      "model_forward_time": 5.604160308837891,
      "step": 10016
    },
    {
      "epoch": 9.552001953125e-06,
      "step": 10016,
      "training_step_time": 17.08619713783264
    },
    {
      "epoch": 9.552955627441406e-06,
      "model_forward_time": 5.608242988586426,
      "step": 10017
    },
    {
      "epoch": 9.552955627441406e-06,
      "step": 10017,
      "training_step_time": 17.092021942138672
    },
    {
      "epoch": 9.553909301757813e-06,
      "model_forward_time": 5.966844320297241,
      "step": 10018
    },
    {
      "epoch": 9.553909301757813e-06,
      "step": 10018,
      "training_step_time": 17.441862106323242
    },
    {
      "epoch": 9.554862976074219e-06,
      "model_forward_time": 5.607144117355347,
      "step": 10019
    },
    {
      "epoch": 9.554862976074219e-06,
      "step": 10019,
      "training_step_time": 17.100899696350098
    },
    {
      "epoch": 9.555816650390625e-06,
      "grad_norm": 0.8236827254295349,
      "learning_rate": 1e-05,
      "loss": 0.0593,
      "step": 10020
    },
    {
      "epoch": 9.555816650390625e-06,
      "model_forward_time": 5.610392808914185,
      "step": 10020
    },
    {
      "dynamics_loss_avg": 0.051617061346769334,
      "epoch": 9.555816650390625e-06,
      "step": 10020
    },
    {
      "action_loss_avg": 0.007619616319425404,
      "epoch": 9.555816650390625e-06,
      "step": 10020
    },
    {
      "epoch": 9.555816650390625e-06,
      "step": 10020,
      "training_step_time": 17.083685874938965
    },
    {
      "epoch": 9.556770324707032e-06,
      "model_forward_time": 5.6040098667144775,
      "step": 10021
    },
    {
      "epoch": 9.556770324707032e-06,
      "step": 10021,
      "training_step_time": 17.061895608901978
    },
    {
      "epoch": 9.557723999023438e-06,
      "model_forward_time": 5.589165449142456,
      "step": 10022
    },
    {
      "epoch": 9.557723999023438e-06,
      "step": 10022,
      "training_step_time": 17.01557207107544
    },
    {
      "epoch": 9.558677673339843e-06,
      "model_forward_time": 5.928344964981079,
      "step": 10023
    },
    {
      "epoch": 9.558677673339843e-06,
      "step": 10023,
      "training_step_time": 17.35782766342163
    },
    {
      "epoch": 9.55963134765625e-06,
      "model_forward_time": 5.602314233779907,
      "step": 10024
    },
    {
      "epoch": 9.55963134765625e-06,
      "step": 10024,
      "training_step_time": 17.085307359695435
    },
    {
      "epoch": 9.560585021972657e-06,
      "model_forward_time": 5.625636100769043,
      "step": 10025
    },
    {
      "epoch": 9.560585021972657e-06,
      "step": 10025,
      "training_step_time": 17.14688992500305
    },
    {
      "epoch": 9.561538696289062e-06,
      "model_forward_time": 5.6195619106292725,
      "step": 10026
    },
    {
      "epoch": 9.561538696289062e-06,
      "step": 10026,
      "training_step_time": 17.10494351387024
    },
    {
      "epoch": 9.562492370605468e-06,
      "model_forward_time": 5.601167917251587,
      "step": 10027
    },
    {
      "epoch": 9.562492370605468e-06,
      "step": 10027,
      "training_step_time": 17.058277368545532
    },
    {
      "epoch": 9.563446044921875e-06,
      "model_forward_time": 5.972081184387207,
      "step": 10028
    },
    {
      "epoch": 9.563446044921875e-06,
      "step": 10028,
      "training_step_time": 17.40872049331665
    },
    {
      "epoch": 9.564399719238281e-06,
      "model_forward_time": 5.594813823699951,
      "step": 10029
    },
    {
      "epoch": 9.564399719238281e-06,
      "step": 10029,
      "training_step_time": 17.06119704246521
    },
    {
      "epoch": 9.565353393554687e-06,
      "grad_norm": 1.5534054040908813,
      "learning_rate": 1e-05,
      "loss": 0.0774,
      "step": 10030
    },
    {
      "epoch": 9.565353393554687e-06,
      "model_forward_time": 5.618318557739258,
      "step": 10030
    },
    {
      "dynamics_loss_avg": 0.06905497945845127,
      "epoch": 9.565353393554687e-06,
      "step": 10030
    },
    {
      "action_loss_avg": 0.009700806112959981,
      "epoch": 9.565353393554687e-06,
      "step": 10030
    },
    {
      "epoch": 9.565353393554687e-06,
      "step": 10030,
      "training_step_time": 17.114707708358765
    },
    {
      "epoch": 9.566307067871094e-06,
      "model_forward_time": 5.6105358600616455,
      "step": 10031
    },
    {
      "epoch": 9.566307067871094e-06,
      "step": 10031,
      "training_step_time": 17.098159313201904
    },
    {
      "epoch": 9.5672607421875e-06,
      "model_forward_time": 5.614098072052002,
      "step": 10032
    },
    {
      "epoch": 9.5672607421875e-06,
      "step": 10032,
      "training_step_time": 17.117353677749634
    },
    {
      "epoch": 9.568214416503906e-06,
      "model_forward_time": 5.611592531204224,
      "step": 10033
    },
    {
      "epoch": 9.568214416503906e-06,
      "step": 10033,
      "training_step_time": 17.1083664894104
    },
    {
      "epoch": 9.569168090820313e-06,
      "model_forward_time": 5.956136465072632,
      "step": 10034
    },
    {
      "epoch": 9.569168090820313e-06,
      "step": 10034,
      "training_step_time": 17.458786249160767
    },
    {
      "epoch": 9.570121765136719e-06,
      "model_forward_time": 5.614603757858276,
      "step": 10035
    },
    {
      "epoch": 9.570121765136719e-06,
      "step": 10035,
      "training_step_time": 17.09449338912964
    },
    {
      "epoch": 9.571075439453125e-06,
      "model_forward_time": 5.62922739982605,
      "step": 10036
    },
    {
      "epoch": 9.571075439453125e-06,
      "step": 10036,
      "training_step_time": 17.15866184234619
    },
    {
      "epoch": 9.572029113769532e-06,
      "model_forward_time": 5.619661092758179,
      "step": 10037
    },
    {
      "epoch": 9.572029113769532e-06,
      "step": 10037,
      "training_step_time": 17.097148895263672
    },
    {
      "epoch": 9.572982788085938e-06,
      "model_forward_time": 5.599352598190308,
      "step": 10038
    },
    {
      "epoch": 9.572982788085938e-06,
      "step": 10038,
      "training_step_time": 17.047733306884766
    },
    {
      "epoch": 9.573936462402343e-06,
      "model_forward_time": 5.593249320983887,
      "step": 10039
    },
    {
      "epoch": 9.573936462402343e-06,
      "step": 10039,
      "training_step_time": 17.41333532333374
    },
    {
      "epoch": 9.57489013671875e-06,
      "grad_norm": 0.8085618019104004,
      "learning_rate": 1e-05,
      "loss": 0.0676,
      "step": 10040
    },
    {
      "epoch": 9.57489013671875e-06,
      "model_forward_time": 5.5919201374053955,
      "step": 10040
    },
    {
      "dynamics_loss_avg": 0.05706507749855518,
      "epoch": 9.57489013671875e-06,
      "step": 10040
    },
    {
      "action_loss_avg": 0.007388372160494328,
      "epoch": 9.57489013671875e-06,
      "step": 10040
    },
    {
      "epoch": 9.57489013671875e-06,
      "step": 10040,
      "training_step_time": 17.045953273773193
    },
    {
      "epoch": 9.575843811035157e-06,
      "model_forward_time": 5.6048805713653564,
      "step": 10041
    },
    {
      "epoch": 9.575843811035157e-06,
      "step": 10041,
      "training_step_time": 17.07533574104309
    },
    {
      "epoch": 9.576797485351562e-06,
      "model_forward_time": 5.60363507270813,
      "step": 10042
    },
    {
      "epoch": 9.576797485351562e-06,
      "step": 10042,
      "training_step_time": 17.077993154525757
    },
    {
      "epoch": 9.577751159667968e-06,
      "model_forward_time": 5.606412410736084,
      "step": 10043
    },
    {
      "epoch": 9.577751159667968e-06,
      "step": 10043,
      "training_step_time": 17.08374285697937
    },
    {
      "epoch": 9.578704833984375e-06,
      "model_forward_time": 5.609682559967041,
      "step": 10044
    },
    {
      "epoch": 9.578704833984375e-06,
      "step": 10044,
      "training_step_time": 17.077912092208862
    },
    {
      "epoch": 9.579658508300781e-06,
      "model_forward_time": 5.989691734313965,
      "step": 10045
    },
    {
      "epoch": 9.579658508300781e-06,
      "step": 10045,
      "training_step_time": 17.45293688774109
    },
    {
      "epoch": 9.580612182617187e-06,
      "model_forward_time": 5.65786075592041,
      "step": 10046
    },
    {
      "epoch": 9.580612182617187e-06,
      "step": 10046,
      "training_step_time": 17.122053146362305
    },
    {
      "epoch": 9.581565856933594e-06,
      "model_forward_time": 5.5953075885772705,
      "step": 10047
    },
    {
      "epoch": 9.581565856933594e-06,
      "step": 10047,
      "training_step_time": 17.055156230926514
    },
    {
      "epoch": 9.58251953125e-06,
      "model_forward_time": 5.6048524379730225,
      "step": 10048
    },
    {
      "epoch": 9.58251953125e-06,
      "step": 10048,
      "training_step_time": 17.08257484436035
    },
    {
      "epoch": 9.583473205566406e-06,
      "model_forward_time": 5.61583137512207,
      "step": 10049
    },
    {
      "epoch": 9.583473205566406e-06,
      "step": 10049,
      "training_step_time": 17.114910125732422
    },
    {
      "epoch": 9.584426879882813e-06,
      "grad_norm": 0.47138795256614685,
      "learning_rate": 1e-05,
      "loss": 0.0694,
      "step": 10050
    },
    {
      "epoch": 9.584426879882813e-06,
      "model_forward_time": 5.62345027923584,
      "step": 10050
    },
    {
      "dynamics_loss_avg": 0.06649070512503386,
      "epoch": 9.584426879882813e-06,
      "step": 10050
    },
    {
      "action_loss_avg": 0.009702480235137045,
      "epoch": 9.584426879882813e-06,
      "step": 10050
    },
    {
      "epoch": 9.584426879882813e-06,
      "step": 10050,
      "training_step_time": 17.453490257263184
    },
    {
      "epoch": 9.585380554199219e-06,
      "model_forward_time": 5.598255395889282,
      "step": 10051
    },
    {
      "epoch": 9.585380554199219e-06,
      "step": 10051,
      "training_step_time": 17.040412664413452
    },
    {
      "epoch": 9.586334228515625e-06,
      "model_forward_time": 5.585655450820923,
      "step": 10052
    },
    {
      "epoch": 9.586334228515625e-06,
      "step": 10052,
      "training_step_time": 17.00076413154602
    },
    {
      "epoch": 9.587287902832032e-06,
      "model_forward_time": 5.5880842208862305,
      "step": 10053
    },
    {
      "epoch": 9.587287902832032e-06,
      "step": 10053,
      "training_step_time": 17.011990070343018
    },
    {
      "epoch": 9.588241577148438e-06,
      "model_forward_time": 5.575547218322754,
      "step": 10054
    },
    {
      "epoch": 9.588241577148438e-06,
      "step": 10054,
      "training_step_time": 17.029573440551758
    },
    {
      "epoch": 9.589195251464843e-06,
      "model_forward_time": 5.592762231826782,
      "step": 10055
    },
    {
      "epoch": 9.589195251464843e-06,
      "step": 10055,
      "training_step_time": 17.05692148208618
    },
    {
      "epoch": 9.59014892578125e-06,
      "model_forward_time": 5.965388298034668,
      "step": 10056
    },
    {
      "epoch": 9.59014892578125e-06,
      "step": 10056,
      "training_step_time": 17.46748685836792
    },
    {
      "epoch": 9.591102600097657e-06,
      "model_forward_time": 5.6156792640686035,
      "step": 10057
    },
    {
      "epoch": 9.591102600097657e-06,
      "step": 10057,
      "training_step_time": 17.12615418434143
    },
    {
      "epoch": 9.592056274414062e-06,
      "model_forward_time": 5.6241724491119385,
      "step": 10058
    },
    {
      "epoch": 9.592056274414062e-06,
      "step": 10058,
      "training_step_time": 17.14124298095703
    },
    {
      "epoch": 9.593009948730468e-06,
      "model_forward_time": 5.634029388427734,
      "step": 10059
    },
    {
      "epoch": 9.593009948730468e-06,
      "step": 10059,
      "training_step_time": 17.15417790412903
    },
    {
      "epoch": 9.593963623046875e-06,
      "grad_norm": 1.5555709600448608,
      "learning_rate": 1e-05,
      "loss": 0.0755,
      "step": 10060
    },
    {
      "epoch": 9.593963623046875e-06,
      "model_forward_time": 5.61693549156189,
      "step": 10060
    },
    {
      "dynamics_loss_avg": 0.06005469374358654,
      "epoch": 9.593963623046875e-06,
      "step": 10060
    },
    {
      "action_loss_avg": 0.010578188602812589,
      "epoch": 9.593963623046875e-06,
      "step": 10060
    },
    {
      "epoch": 9.593963623046875e-06,
      "step": 10060,
      "training_step_time": 17.11723780632019
    },
    {
      "epoch": 9.594917297363281e-06,
      "model_forward_time": 5.982100486755371,
      "step": 10061
    },
    {
      "epoch": 9.594917297363281e-06,
      "step": 10061,
      "training_step_time": 17.467096090316772
    },
    {
      "epoch": 9.595870971679687e-06,
      "model_forward_time": 5.613915681838989,
      "step": 10062
    },
    {
      "epoch": 9.595870971679687e-06,
      "step": 10062,
      "training_step_time": 17.093093395233154
    },
    {
      "epoch": 9.596824645996094e-06,
      "model_forward_time": 5.615131855010986,
      "step": 10063
    },
    {
      "epoch": 9.596824645996094e-06,
      "step": 10063,
      "training_step_time": 17.084739685058594
    },
    {
      "epoch": 9.5977783203125e-06,
      "model_forward_time": 5.609946250915527,
      "step": 10064
    },
    {
      "epoch": 9.5977783203125e-06,
      "step": 10064,
      "training_step_time": 17.065114498138428
    },
    {
      "epoch": 9.598731994628906e-06,
      "model_forward_time": 5.600459098815918,
      "step": 10065
    },
    {
      "epoch": 9.598731994628906e-06,
      "step": 10065,
      "training_step_time": 17.051153421401978
    },
    {
      "epoch": 9.599685668945313e-06,
      "model_forward_time": 5.588283061981201,
      "step": 10066
    },
    {
      "epoch": 9.599685668945313e-06,
      "step": 10066,
      "training_step_time": 17.050814628601074
    },
    {
      "epoch": 9.600639343261719e-06,
      "model_forward_time": 5.965269565582275,
      "step": 10067
    },
    {
      "epoch": 9.600639343261719e-06,
      "step": 10067,
      "training_step_time": 17.460550546646118
    },
    {
      "epoch": 9.601593017578125e-06,
      "model_forward_time": 5.622615814208984,
      "step": 10068
    },
    {
      "epoch": 9.601593017578125e-06,
      "step": 10068,
      "training_step_time": 17.133527517318726
    },
    {
      "epoch": 9.602546691894532e-06,
      "model_forward_time": 5.656808137893677,
      "step": 10069
    },
    {
      "epoch": 9.602546691894532e-06,
      "step": 10069,
      "training_step_time": 17.20412588119507
    },
    {
      "epoch": 9.603500366210938e-06,
      "grad_norm": 0.6857162117958069,
      "learning_rate": 1e-05,
      "loss": 0.0718,
      "step": 10070
    },
    {
      "epoch": 9.603500366210938e-06,
      "model_forward_time": 5.668781280517578,
      "step": 10070
    },
    {
      "dynamics_loss_avg": 0.060939946211874485,
      "epoch": 9.603500366210938e-06,
      "step": 10070
    },
    {
      "action_loss_avg": 0.008758006896823645,
      "epoch": 9.603500366210938e-06,
      "step": 10070
    },
    {
      "epoch": 9.603500366210938e-06,
      "step": 10070,
      "training_step_time": 17.1896653175354
    },
    {
      "epoch": 9.604454040527343e-06,
      "model_forward_time": 5.605055332183838,
      "step": 10071
    },
    {
      "epoch": 9.604454040527343e-06,
      "step": 10071,
      "training_step_time": 17.047850608825684
    },
    {
      "epoch": 9.60540771484375e-06,
      "model_forward_time": 5.986494779586792,
      "step": 10072
    },
    {
      "epoch": 9.60540771484375e-06,
      "step": 10072,
      "training_step_time": 17.44657850265503
    },
    {
      "epoch": 9.606361389160157e-06,
      "model_forward_time": 5.6078877449035645,
      "step": 10073
    },
    {
      "epoch": 9.606361389160157e-06,
      "step": 10073,
      "training_step_time": 17.09096884727478
    },
    {
      "epoch": 9.607315063476562e-06,
      "model_forward_time": 5.625411033630371,
      "step": 10074
    },
    {
      "epoch": 9.607315063476562e-06,
      "step": 10074,
      "training_step_time": 17.137795448303223
    },
    {
      "epoch": 9.608268737792968e-06,
      "model_forward_time": 5.60699987411499,
      "step": 10075
    },
    {
      "epoch": 9.608268737792968e-06,
      "step": 10075,
      "training_step_time": 17.077366590499878
    },
    {
      "epoch": 9.609222412109375e-06,
      "model_forward_time": 5.599898815155029,
      "step": 10076
    },
    {
      "epoch": 9.609222412109375e-06,
      "step": 10076,
      "training_step_time": 17.04431462287903
    },
    {
      "epoch": 9.610176086425781e-06,
      "model_forward_time": 5.588820934295654,
      "step": 10077
    },
    {
      "epoch": 9.610176086425781e-06,
      "step": 10077,
      "training_step_time": 17.03813600540161
    },
    {
      "epoch": 9.611129760742187e-06,
      "model_forward_time": 5.952071189880371,
      "step": 10078
    },
    {
      "epoch": 9.611129760742187e-06,
      "step": 10078,
      "training_step_time": 17.417783975601196
    },
    {
      "epoch": 9.612083435058594e-06,
      "model_forward_time": 5.611505508422852,
      "step": 10079
    },
    {
      "epoch": 9.612083435058594e-06,
      "step": 10079,
      "training_step_time": 17.11407160758972
    },
    {
      "epoch": 9.613037109375e-06,
      "grad_norm": 1.4991854429244995,
      "learning_rate": 1e-05,
      "loss": 0.051,
      "step": 10080
    },
    {
      "epoch": 9.613037109375e-06,
      "model_forward_time": 5.624043941497803,
      "step": 10080
    },
    {
      "dynamics_loss_avg": 0.04864102452993393,
      "epoch": 9.613037109375e-06,
      "step": 10080
    },
    {
      "action_loss_avg": 0.006551673426292837,
      "epoch": 9.613037109375e-06,
      "step": 10080
    },
    {
      "epoch": 9.613037109375e-06,
      "step": 10080,
      "training_step_time": 17.139840364456177
    },
    {
      "epoch": 9.613990783691406e-06,
      "model_forward_time": 5.6320960521698,
      "step": 10081
    },
    {
      "epoch": 9.613990783691406e-06,
      "step": 10081,
      "training_step_time": 17.12798237800598
    },
    {
      "epoch": 9.614944458007813e-06,
      "model_forward_time": 5.602834701538086,
      "step": 10082
    },
    {
      "epoch": 9.614944458007813e-06,
      "step": 10082,
      "training_step_time": 17.074323177337646
    },
    {
      "epoch": 9.615898132324219e-06,
      "model_forward_time": 5.967478275299072,
      "step": 10083
    },
    {
      "epoch": 9.615898132324219e-06,
      "step": 10083,
      "training_step_time": 17.422343015670776
    },
    {
      "epoch": 9.616851806640625e-06,
      "model_forward_time": 5.606337308883667,
      "step": 10084
    },
    {
      "epoch": 9.616851806640625e-06,
      "step": 10084,
      "training_step_time": 17.078676462173462
    },
    {
      "epoch": 9.617805480957032e-06,
      "model_forward_time": 5.623788833618164,
      "step": 10085
    },
    {
      "epoch": 9.617805480957032e-06,
      "step": 10085,
      "training_step_time": 17.096610069274902
    },
    {
      "epoch": 9.618759155273438e-06,
      "model_forward_time": 5.597932577133179,
      "step": 10086
    },
    {
      "epoch": 9.618759155273438e-06,
      "step": 10086,
      "training_step_time": 17.04937243461609
    },
    {
      "epoch": 9.619712829589844e-06,
      "model_forward_time": 5.596699237823486,
      "step": 10087
    },
    {
      "epoch": 9.619712829589844e-06,
      "step": 10087,
      "training_step_time": 17.04937171936035
    },
    {
      "epoch": 9.62066650390625e-06,
      "model_forward_time": 5.95102596282959,
      "step": 10088
    },
    {
      "epoch": 9.62066650390625e-06,
      "step": 10088,
      "training_step_time": 17.413086414337158
    },
    {
      "epoch": 9.621620178222657e-06,
      "model_forward_time": 5.601561069488525,
      "step": 10089
    },
    {
      "epoch": 9.621620178222657e-06,
      "step": 10089,
      "training_step_time": 17.081992149353027
    },
    {
      "epoch": 9.622573852539062e-06,
      "grad_norm": 0.5574042797088623,
      "learning_rate": 1e-05,
      "loss": 0.0684,
      "step": 10090
    },
    {
      "epoch": 9.622573852539062e-06,
      "model_forward_time": 5.603999376296997,
      "step": 10090
    },
    {
      "dynamics_loss_avg": 0.05732959676533937,
      "epoch": 9.622573852539062e-06,
      "step": 10090
    },
    {
      "action_loss_avg": 0.008588488749228417,
      "epoch": 9.622573852539062e-06,
      "step": 10090
    },
    {
      "epoch": 9.622573852539062e-06,
      "step": 10090,
      "training_step_time": 17.09460163116455
    },
    {
      "epoch": 9.623527526855468e-06,
      "model_forward_time": 5.61368989944458,
      "step": 10091
    },
    {
      "epoch": 9.623527526855468e-06,
      "step": 10091,
      "training_step_time": 17.10722851753235
    },
    {
      "epoch": 9.624481201171876e-06,
      "model_forward_time": 5.621994256973267,
      "step": 10092
    },
    {
      "epoch": 9.624481201171876e-06,
      "step": 10092,
      "training_step_time": 17.089025259017944
    },
    {
      "epoch": 9.625434875488281e-06,
      "model_forward_time": 5.584718227386475,
      "step": 10093
    },
    {
      "epoch": 9.625434875488281e-06,
      "step": 10093,
      "training_step_time": 17.018662452697754
    },
    {
      "epoch": 9.626388549804687e-06,
      "model_forward_time": 5.935039281845093,
      "step": 10094
    },
    {
      "epoch": 9.626388549804687e-06,
      "step": 10094,
      "training_step_time": 17.34006404876709
    },
    {
      "epoch": 9.627342224121094e-06,
      "model_forward_time": 5.568314552307129,
      "step": 10095
    },
    {
      "epoch": 9.627342224121094e-06,
      "step": 10095,
      "training_step_time": 16.997973203659058
    },
    {
      "epoch": 9.6282958984375e-06,
      "model_forward_time": 5.572188854217529,
      "step": 10096
    },
    {
      "epoch": 9.6282958984375e-06,
      "step": 10096,
      "training_step_time": 17.003711223602295
    },
    {
      "epoch": 9.629249572753906e-06,
      "model_forward_time": 5.571053981781006,
      "step": 10097
    },
    {
      "epoch": 9.629249572753906e-06,
      "step": 10097,
      "training_step_time": 16.980973482131958
    },
    {
      "epoch": 9.630203247070313e-06,
      "model_forward_time": 5.589033126831055,
      "step": 10098
    },
    {
      "epoch": 9.630203247070313e-06,
      "step": 10098,
      "training_step_time": 17.01152539253235
    },
    {
      "epoch": 9.631156921386719e-06,
      "model_forward_time": 5.972584247589111,
      "step": 10099
    },
    {
      "epoch": 9.631156921386719e-06,
      "step": 10099,
      "training_step_time": 17.421164512634277
    },
    {
      "epoch": 9.632110595703125e-06,
      "grad_norm": 0.943069338798523,
      "learning_rate": 1e-05,
      "loss": 0.0751,
      "step": 10100
    },
    {
      "epoch": 9.632110595703125e-06,
      "model_forward_time": 5.61180567741394,
      "step": 10100
    },
    {
      "dynamics_loss_avg": 0.0684009239077568,
      "epoch": 9.632110595703125e-06,
      "step": 10100
    },
    {
      "action_loss_avg": 0.010948381060734392,
      "epoch": 9.632110595703125e-06,
      "step": 10100
    },
    {
      "epoch": 9.632110595703125e-06,
      "step": 10100,
      "training_step_time": 17.09110164642334
    },
    {
      "epoch": 9.633064270019532e-06,
      "model_forward_time": 5.6098313331604,
      "step": 10101
    },
    {
      "epoch": 9.633064270019532e-06,
      "step": 10101,
      "training_step_time": 17.088646411895752
    },
    {
      "epoch": 9.634017944335938e-06,
      "model_forward_time": 5.607262372970581,
      "step": 10102
    },
    {
      "epoch": 9.634017944335938e-06,
      "step": 10102,
      "training_step_time": 17.081128120422363
    },
    {
      "epoch": 9.634971618652344e-06,
      "model_forward_time": 5.588578939437866,
      "step": 10103
    },
    {
      "epoch": 9.634971618652344e-06,
      "step": 10103,
      "training_step_time": 17.02587628364563
    },
    {
      "epoch": 9.63592529296875e-06,
      "model_forward_time": 5.599114179611206,
      "step": 10104
    },
    {
      "epoch": 9.63592529296875e-06,
      "step": 10104,
      "training_step_time": 17.088223934173584
    },
    {
      "epoch": 9.636878967285157e-06,
      "model_forward_time": 6.006461143493652,
      "step": 10105
    },
    {
      "epoch": 9.636878967285157e-06,
      "step": 10105,
      "training_step_time": 17.491527557373047
    },
    {
      "epoch": 9.637832641601562e-06,
      "model_forward_time": 5.606791257858276,
      "step": 10106
    },
    {
      "epoch": 9.637832641601562e-06,
      "step": 10106,
      "training_step_time": 17.090481519699097
    },
    {
      "epoch": 9.638786315917968e-06,
      "model_forward_time": 5.621732234954834,
      "step": 10107
    },
    {
      "epoch": 9.638786315917968e-06,
      "step": 10107,
      "training_step_time": 17.07821226119995
    },
    {
      "epoch": 9.639739990234376e-06,
      "model_forward_time": 5.586346387863159,
      "step": 10108
    },
    {
      "epoch": 9.639739990234376e-06,
      "step": 10108,
      "training_step_time": 17.011430978775024
    },
    {
      "epoch": 9.640693664550781e-06,
      "model_forward_time": 5.597528457641602,
      "step": 10109
    },
    {
      "epoch": 9.640693664550781e-06,
      "step": 10109,
      "training_step_time": 17.026196002960205
    },
    {
      "epoch": 9.641647338867187e-06,
      "grad_norm": 0.4769572615623474,
      "learning_rate": 1e-05,
      "loss": 0.0643,
      "step": 10110
    },
    {
      "epoch": 9.641647338867187e-06,
      "model_forward_time": 5.595022439956665,
      "step": 10110
    },
    {
      "dynamics_loss_avg": 0.055513005889952186,
      "epoch": 9.641647338867187e-06,
      "step": 10110
    },
    {
      "action_loss_avg": 0.007515493198297918,
      "epoch": 9.641647338867187e-06,
      "step": 10110
    },
    {
      "epoch": 9.641647338867187e-06,
      "step": 10110,
      "training_step_time": 17.42110252380371
    },
    {
      "epoch": 9.642601013183594e-06,
      "model_forward_time": 5.600906848907471,
      "step": 10111
    },
    {
      "epoch": 9.642601013183594e-06,
      "step": 10111,
      "training_step_time": 17.078962087631226
    },
    {
      "epoch": 9.6435546875e-06,
      "model_forward_time": 5.602809190750122,
      "step": 10112
    },
    {
      "epoch": 9.6435546875e-06,
      "step": 10112,
      "training_step_time": 17.078065872192383
    },
    {
      "epoch": 9.644508361816406e-06,
      "model_forward_time": 5.616740703582764,
      "step": 10113
    },
    {
      "epoch": 9.644508361816406e-06,
      "step": 10113,
      "training_step_time": 17.082911014556885
    },
    {
      "epoch": 9.645462036132813e-06,
      "model_forward_time": 5.612311363220215,
      "step": 10114
    },
    {
      "epoch": 9.645462036132813e-06,
      "step": 10114,
      "training_step_time": 17.056736707687378
    },
    {
      "epoch": 9.646415710449219e-06,
      "model_forward_time": 5.600358486175537,
      "step": 10115
    },
    {
      "epoch": 9.646415710449219e-06,
      "step": 10115,
      "training_step_time": 17.070969104766846
    },
    {
      "epoch": 9.647369384765625e-06,
      "model_forward_time": 5.978104114532471,
      "step": 10116
    },
    {
      "epoch": 9.647369384765625e-06,
      "step": 10116,
      "training_step_time": 17.441313982009888
    },
    {
      "epoch": 9.648323059082032e-06,
      "model_forward_time": 5.60336709022522,
      "step": 10117
    },
    {
      "epoch": 9.648323059082032e-06,
      "step": 10117,
      "training_step_time": 17.075319051742554
    },
    {
      "epoch": 9.649276733398438e-06,
      "model_forward_time": 5.6074018478393555,
      "step": 10118
    },
    {
      "epoch": 9.649276733398438e-06,
      "step": 10118,
      "training_step_time": 17.083878755569458
    },
    {
      "epoch": 9.650230407714844e-06,
      "model_forward_time": 5.619621753692627,
      "step": 10119
    },
    {
      "epoch": 9.650230407714844e-06,
      "step": 10119,
      "training_step_time": 17.102173805236816
    },
    {
      "epoch": 9.65118408203125e-06,
      "grad_norm": 0.8550860285758972,
      "learning_rate": 1e-05,
      "loss": 0.0592,
      "step": 10120
    },
    {
      "epoch": 9.65118408203125e-06,
      "model_forward_time": 5.62235164642334,
      "step": 10120
    },
    {
      "dynamics_loss_avg": 0.0468209957703948,
      "epoch": 9.65118408203125e-06,
      "step": 10120
    },
    {
      "action_loss_avg": 0.00756820784881711,
      "epoch": 9.65118408203125e-06,
      "step": 10120
    },
    {
      "epoch": 9.65118408203125e-06,
      "step": 10120,
      "training_step_time": 17.13096284866333
    },
    {
      "epoch": 9.652137756347657e-06,
      "model_forward_time": 5.964430570602417,
      "step": 10121
    },
    {
      "epoch": 9.652137756347657e-06,
      "step": 10121,
      "training_step_time": 17.4487202167511
    },
    {
      "epoch": 9.653091430664062e-06,
      "model_forward_time": 5.630977630615234,
      "step": 10122
    },
    {
      "epoch": 9.653091430664062e-06,
      "step": 10122,
      "training_step_time": 17.11216974258423
    },
    {
      "epoch": 9.654045104980468e-06,
      "model_forward_time": 5.61120343208313,
      "step": 10123
    },
    {
      "epoch": 9.654045104980468e-06,
      "step": 10123,
      "training_step_time": 17.09881019592285
    },
    {
      "epoch": 9.654998779296876e-06,
      "model_forward_time": 5.605746507644653,
      "step": 10124
    },
    {
      "epoch": 9.654998779296876e-06,
      "step": 10124,
      "training_step_time": 17.078362226486206
    },
    {
      "epoch": 9.655952453613281e-06,
      "model_forward_time": 5.621487140655518,
      "step": 10125
    },
    {
      "epoch": 9.655952453613281e-06,
      "step": 10125,
      "training_step_time": 17.116283655166626
    },
    {
      "epoch": 9.656906127929687e-06,
      "model_forward_time": 5.603602409362793,
      "step": 10126
    },
    {
      "epoch": 9.656906127929687e-06,
      "step": 10126,
      "training_step_time": 17.100364685058594
    },
    {
      "epoch": 9.657859802246094e-06,
      "model_forward_time": 5.9974095821380615,
      "step": 10127
    },
    {
      "epoch": 9.657859802246094e-06,
      "step": 10127,
      "training_step_time": 17.492084741592407
    },
    {
      "epoch": 9.6588134765625e-06,
      "model_forward_time": 5.608222723007202,
      "step": 10128
    },
    {
      "epoch": 9.6588134765625e-06,
      "step": 10128,
      "training_step_time": 17.07624912261963
    },
    {
      "epoch": 9.659767150878906e-06,
      "model_forward_time": 5.597492694854736,
      "step": 10129
    },
    {
      "epoch": 9.659767150878906e-06,
      "step": 10129,
      "training_step_time": 17.056884765625
    },
    {
      "epoch": 9.660720825195313e-06,
      "grad_norm": 0.3550109565258026,
      "learning_rate": 1e-05,
      "loss": 0.0744,
      "step": 10130
    },
    {
      "epoch": 9.660720825195313e-06,
      "model_forward_time": 5.601311206817627,
      "step": 10130
    },
    {
      "dynamics_loss_avg": 0.06626985110342502,
      "epoch": 9.660720825195313e-06,
      "step": 10130
    },
    {
      "action_loss_avg": 0.01036724103614688,
      "epoch": 9.660720825195313e-06,
      "step": 10130
    },
    {
      "epoch": 9.660720825195313e-06,
      "step": 10130,
      "training_step_time": 17.03971838951111
    },
    {
      "epoch": 9.661674499511719e-06,
      "model_forward_time": 5.600811958312988,
      "step": 10131
    },
    {
      "epoch": 9.661674499511719e-06,
      "step": 10131,
      "training_step_time": 17.067220211029053
    },
    {
      "epoch": 9.662628173828125e-06,
      "model_forward_time": 5.94743013381958,
      "step": 10132
    },
    {
      "epoch": 9.662628173828125e-06,
      "step": 10132,
      "training_step_time": 17.405159950256348
    },
    {
      "epoch": 9.66358184814453e-06,
      "model_forward_time": 5.5978498458862305,
      "step": 10133
    },
    {
      "epoch": 9.66358184814453e-06,
      "step": 10133,
      "training_step_time": 17.0353581905365
    },
    {
      "epoch": 9.664535522460938e-06,
      "model_forward_time": 5.586400508880615,
      "step": 10134
    },
    {
      "epoch": 9.664535522460938e-06,
      "step": 10134,
      "training_step_time": 17.0145046710968
    },
    {
      "epoch": 9.665489196777344e-06,
      "model_forward_time": 5.589456796646118,
      "step": 10135
    },
    {
      "epoch": 9.665489196777344e-06,
      "step": 10135,
      "training_step_time": 17.053823232650757
    },
    {
      "epoch": 9.66644287109375e-06,
      "model_forward_time": 5.595120906829834,
      "step": 10136
    },
    {
      "epoch": 9.66644287109375e-06,
      "step": 10136,
      "training_step_time": 17.07508897781372
    },
    {
      "epoch": 9.667396545410157e-06,
      "model_forward_time": 5.605659484863281,
      "step": 10137
    },
    {
      "epoch": 9.667396545410157e-06,
      "step": 10137,
      "training_step_time": 17.075627088546753
    },
    {
      "epoch": 9.668350219726562e-06,
      "model_forward_time": 5.9747021198272705,
      "step": 10138
    },
    {
      "epoch": 9.668350219726562e-06,
      "step": 10138,
      "training_step_time": 17.47841787338257
    },
    {
      "epoch": 9.669303894042968e-06,
      "model_forward_time": 5.619074583053589,
      "step": 10139
    },
    {
      "epoch": 9.669303894042968e-06,
      "step": 10139,
      "training_step_time": 17.12830662727356
    },
    {
      "epoch": 9.670257568359376e-06,
      "grad_norm": 0.8135103583335876,
      "learning_rate": 1e-05,
      "loss": 0.065,
      "step": 10140
    },
    {
      "epoch": 9.670257568359376e-06,
      "model_forward_time": 5.625085353851318,
      "step": 10140
    },
    {
      "dynamics_loss_avg": 0.054087925143539904,
      "epoch": 9.670257568359376e-06,
      "step": 10140
    },
    {
      "action_loss_avg": 0.00915752670262009,
      "epoch": 9.670257568359376e-06,
      "step": 10140
    },
    {
      "epoch": 9.670257568359376e-06,
      "step": 10140,
      "training_step_time": 17.135099411010742
    },
    {
      "epoch": 9.671211242675781e-06,
      "model_forward_time": 5.6201958656311035,
      "step": 10141
    },
    {
      "epoch": 9.671211242675781e-06,
      "step": 10141,
      "training_step_time": 17.124499797821045
    },
    {
      "epoch": 9.672164916992187e-06,
      "model_forward_time": 5.606826305389404,
      "step": 10142
    },
    {
      "epoch": 9.672164916992187e-06,
      "step": 10142,
      "training_step_time": 17.10996913909912
    },
    {
      "epoch": 9.673118591308594e-06,
      "model_forward_time": 5.619725465774536,
      "step": 10143
    },
    {
      "epoch": 9.673118591308594e-06,
      "step": 10143,
      "training_step_time": 17.49906277656555
    },
    {
      "epoch": 9.674072265625e-06,
      "model_forward_time": 5.608443260192871,
      "step": 10144
    },
    {
      "epoch": 9.674072265625e-06,
      "step": 10144,
      "training_step_time": 17.085832834243774
    },
    {
      "epoch": 9.675025939941406e-06,
      "model_forward_time": 5.60416054725647,
      "step": 10145
    },
    {
      "epoch": 9.675025939941406e-06,
      "step": 10145,
      "training_step_time": 17.0643208026886
    },
    {
      "epoch": 9.675979614257813e-06,
      "model_forward_time": 5.589094400405884,
      "step": 10146
    },
    {
      "epoch": 9.675979614257813e-06,
      "step": 10146,
      "training_step_time": 17.051788568496704
    },
    {
      "epoch": 9.676933288574219e-06,
      "model_forward_time": 5.603291273117065,
      "step": 10147
    },
    {
      "epoch": 9.676933288574219e-06,
      "step": 10147,
      "training_step_time": 17.069243669509888
    },
    {
      "epoch": 9.677886962890625e-06,
      "model_forward_time": 5.606307029724121,
      "step": 10148
    },
    {
      "epoch": 9.677886962890625e-06,
      "step": 10148,
      "training_step_time": 17.080618381500244
    },
    {
      "epoch": 9.67884063720703e-06,
      "model_forward_time": 5.967299222946167,
      "step": 10149
    },
    {
      "epoch": 9.67884063720703e-06,
      "step": 10149,
      "training_step_time": 17.436452627182007
    },
    {
      "epoch": 9.679794311523438e-06,
      "grad_norm": 1.3834880590438843,
      "learning_rate": 1e-05,
      "loss": 0.0682,
      "step": 10150
    },
    {
      "epoch": 9.679794311523438e-06,
      "model_forward_time": 5.595946550369263,
      "step": 10150
    },
    {
      "dynamics_loss_avg": 0.06443198621273041,
      "epoch": 9.679794311523438e-06,
      "step": 10150
    },
    {
      "action_loss_avg": 0.008832616033032537,
      "epoch": 9.679794311523438e-06,
      "step": 10150
    },
    {
      "epoch": 9.679794311523438e-06,
      "step": 10150,
      "training_step_time": 17.058041095733643
    },
    {
      "epoch": 9.680747985839844e-06,
      "model_forward_time": 5.606035470962524,
      "step": 10151
    },
    {
      "epoch": 9.680747985839844e-06,
      "step": 10151,
      "training_step_time": 17.04346776008606
    },
    {
      "epoch": 9.68170166015625e-06,
      "model_forward_time": 5.596553325653076,
      "step": 10152
    },
    {
      "epoch": 9.68170166015625e-06,
      "step": 10152,
      "training_step_time": 17.03618860244751
    },
    {
      "epoch": 9.682655334472657e-06,
      "model_forward_time": 5.583255767822266,
      "step": 10153
    },
    {
      "epoch": 9.682655334472657e-06,
      "step": 10153,
      "training_step_time": 17.025147199630737
    },
    {
      "epoch": 9.683609008789062e-06,
      "model_forward_time": 5.96168851852417,
      "step": 10154
    },
    {
      "epoch": 9.683609008789062e-06,
      "step": 10154,
      "training_step_time": 17.41132879257202
    },
    {
      "epoch": 9.684562683105468e-06,
      "model_forward_time": 5.587858438491821,
      "step": 10155
    },
    {
      "epoch": 9.684562683105468e-06,
      "step": 10155,
      "training_step_time": 17.052762746810913
    },
    {
      "epoch": 9.685516357421876e-06,
      "model_forward_time": 5.59949803352356,
      "step": 10156
    },
    {
      "epoch": 9.685516357421876e-06,
      "step": 10156,
      "training_step_time": 17.054697275161743
    },
    {
      "epoch": 9.686470031738281e-06,
      "model_forward_time": 5.610113859176636,
      "step": 10157
    },
    {
      "epoch": 9.686470031738281e-06,
      "step": 10157,
      "training_step_time": 17.09757161140442
    },
    {
      "epoch": 9.687423706054687e-06,
      "model_forward_time": 5.605653762817383,
      "step": 10158
    },
    {
      "epoch": 9.687423706054687e-06,
      "step": 10158,
      "training_step_time": 17.085362672805786
    },
    {
      "epoch": 9.688377380371094e-06,
      "model_forward_time": 5.603877067565918,
      "step": 10159
    },
    {
      "epoch": 9.688377380371094e-06,
      "step": 10159,
      "training_step_time": 17.07774043083191
    },
    {
      "epoch": 9.6893310546875e-06,
      "grad_norm": 0.4419768154621124,
      "learning_rate": 1e-05,
      "loss": 0.0614,
      "step": 10160
    },
    {
      "epoch": 9.6893310546875e-06,
      "model_forward_time": 5.975204944610596,
      "step": 10160
    },
    {
      "dynamics_loss_avg": 0.05398489199578762,
      "epoch": 9.6893310546875e-06,
      "step": 10160
    },
    {
      "action_loss_avg": 0.007913437020033599,
      "epoch": 9.6893310546875e-06,
      "step": 10160
    },
    {
      "epoch": 9.6893310546875e-06,
      "step": 10160,
      "training_step_time": 17.41457772254944
    },
    {
      "epoch": 9.690284729003906e-06,
      "model_forward_time": 5.580738544464111,
      "step": 10161
    },
    {
      "epoch": 9.690284729003906e-06,
      "step": 10161,
      "training_step_time": 17.019106149673462
    },
    {
      "epoch": 9.691238403320313e-06,
      "model_forward_time": 5.6023595333099365,
      "step": 10162
    },
    {
      "epoch": 9.691238403320313e-06,
      "step": 10162,
      "training_step_time": 17.06466054916382
    },
    {
      "epoch": 9.692192077636719e-06,
      "model_forward_time": 5.597729921340942,
      "step": 10163
    },
    {
      "epoch": 9.692192077636719e-06,
      "step": 10163,
      "training_step_time": 17.07551860809326
    },
    {
      "epoch": 9.693145751953125e-06,
      "model_forward_time": 5.617363929748535,
      "step": 10164
    },
    {
      "epoch": 9.693145751953125e-06,
      "step": 10164,
      "training_step_time": 17.124605178833008
    },
    {
      "epoch": 9.69409942626953e-06,
      "model_forward_time": 5.969594955444336,
      "step": 10165
    },
    {
      "epoch": 9.69409942626953e-06,
      "step": 10165,
      "training_step_time": 17.47108030319214
    },
    {
      "epoch": 9.695053100585938e-06,
      "model_forward_time": 5.623478412628174,
      "step": 10166
    },
    {
      "epoch": 9.695053100585938e-06,
      "step": 10166,
      "training_step_time": 17.117829084396362
    },
    {
      "epoch": 9.696006774902344e-06,
      "model_forward_time": 5.6127660274505615,
      "step": 10167
    },
    {
      "epoch": 9.696006774902344e-06,
      "step": 10167,
      "training_step_time": 17.100063800811768
    },
    {
      "epoch": 9.69696044921875e-06,
      "model_forward_time": 5.613209009170532,
      "step": 10168
    },
    {
      "epoch": 9.69696044921875e-06,
      "step": 10168,
      "training_step_time": 17.115412712097168
    },
    {
      "epoch": 9.697914123535157e-06,
      "model_forward_time": 5.611288547515869,
      "step": 10169
    },
    {
      "epoch": 9.697914123535157e-06,
      "step": 10169,
      "training_step_time": 17.10149908065796
    },
    {
      "epoch": 9.698867797851562e-06,
      "grad_norm": 1.508394718170166,
      "learning_rate": 1e-05,
      "loss": 0.066,
      "step": 10170
    },
    {
      "epoch": 9.698867797851562e-06,
      "model_forward_time": 5.61129093170166,
      "step": 10170
    },
    {
      "dynamics_loss_avg": 0.055149278976023196,
      "epoch": 9.698867797851562e-06,
      "step": 10170
    },
    {
      "action_loss_avg": 0.009610241372138261,
      "epoch": 9.698867797851562e-06,
      "step": 10170
    },
    {
      "epoch": 9.698867797851562e-06,
      "step": 10170,
      "training_step_time": 17.074458122253418
    },
    {
      "epoch": 9.699821472167968e-06,
      "model_forward_time": 5.972863435745239,
      "step": 10171
    },
    {
      "epoch": 9.699821472167968e-06,
      "step": 10171,
      "training_step_time": 17.40930461883545
    },
    {
      "epoch": 9.700775146484376e-06,
      "model_forward_time": 5.616978883743286,
      "step": 10172
    },
    {
      "epoch": 9.700775146484376e-06,
      "step": 10172,
      "training_step_time": 17.1178777217865
    },
    {
      "epoch": 9.701728820800781e-06,
      "model_forward_time": 5.6317689418792725,
      "step": 10173
    },
    {
      "epoch": 9.701728820800781e-06,
      "step": 10173,
      "training_step_time": 17.127046823501587
    },
    {
      "epoch": 9.702682495117187e-06,
      "model_forward_time": 5.601659774780273,
      "step": 10174
    },
    {
      "epoch": 9.702682495117187e-06,
      "step": 10174,
      "training_step_time": 17.037880420684814
    },
    {
      "epoch": 9.703636169433594e-06,
      "model_forward_time": 5.582159519195557,
      "step": 10175
    },
    {
      "epoch": 9.703636169433594e-06,
      "step": 10175,
      "training_step_time": 17.01988196372986
    },
    {
      "epoch": 9.70458984375e-06,
      "model_forward_time": 5.560532808303833,
      "step": 10176
    },
    {
      "epoch": 9.70458984375e-06,
      "step": 10176,
      "training_step_time": 17.34224557876587
    },
    {
      "epoch": 9.705543518066406e-06,
      "model_forward_time": 5.570590972900391,
      "step": 10177
    },
    {
      "epoch": 9.705543518066406e-06,
      "step": 10177,
      "training_step_time": 16.973838806152344
    },
    {
      "epoch": 9.706497192382813e-06,
      "model_forward_time": 5.573753356933594,
      "step": 10178
    },
    {
      "epoch": 9.706497192382813e-06,
      "step": 10178,
      "training_step_time": 16.99731469154358
    },
    {
      "epoch": 9.707450866699219e-06,
      "model_forward_time": 4.262701749801636,
      "step": 10179
    },
    {
      "epoch": 9.707450866699219e-06,
      "step": 10179,
      "training_step_time": 12.752115249633789
    },
    {
      "epoch": 9.708404541015625e-06,
      "grad_norm": 1.655692219734192,
      "learning_rate": 1e-05,
      "loss": 0.0658,
      "step": 10180
    },
    {
      "epoch": 9.708404541015625e-06,
      "model_forward_time": 5.584279298782349,
      "step": 10180
    },
    {
      "dynamics_loss_avg": 0.05607644431293011,
      "epoch": 9.708404541015625e-06,
      "step": 10180
    },
    {
      "action_loss_avg": 0.008259844360873103,
      "epoch": 9.708404541015625e-06,
      "step": 10180
    },
    {
      "epoch": 9.708404541015625e-06,
      "step": 10180,
      "training_step_time": 17.03317642211914
    },
    {
      "epoch": 9.70935821533203e-06,
      "model_forward_time": 5.604938268661499,
      "step": 10181
    },
    {
      "epoch": 9.70935821533203e-06,
      "step": 10181,
      "training_step_time": 17.076195240020752
    },
    {
      "epoch": 9.710311889648438e-06,
      "model_forward_time": 5.973664999008179,
      "step": 10182
    },
    {
      "epoch": 9.710311889648438e-06,
      "step": 10182,
      "training_step_time": 17.451585292816162
    },
    {
      "epoch": 9.711265563964844e-06,
      "model_forward_time": 5.610552072525024,
      "step": 10183
    },
    {
      "epoch": 9.711265563964844e-06,
      "step": 10183,
      "training_step_time": 17.07559108734131
    },
    {
      "epoch": 9.71221923828125e-06,
      "model_forward_time": 5.590679883956909,
      "step": 10184
    },
    {
      "epoch": 9.71221923828125e-06,
      "step": 10184,
      "training_step_time": 17.02603030204773
    },
    {
      "epoch": 9.713172912597657e-06,
      "model_forward_time": 5.595815181732178,
      "step": 10185
    },
    {
      "epoch": 9.713172912597657e-06,
      "step": 10185,
      "training_step_time": 17.01408052444458
    },
    {
      "epoch": 9.714126586914062e-06,
      "model_forward_time": 5.586129903793335,
      "step": 10186
    },
    {
      "epoch": 9.714126586914062e-06,
      "step": 10186,
      "training_step_time": 17.007551670074463
    },
    {
      "epoch": 9.715080261230468e-06,
      "model_forward_time": 5.600439548492432,
      "step": 10187
    },
    {
      "epoch": 9.715080261230468e-06,
      "step": 10187,
      "training_step_time": 17.424372673034668
    },
    {
      "epoch": 9.716033935546876e-06,
      "model_forward_time": 5.6187074184417725,
      "step": 10188
    },
    {
      "epoch": 9.716033935546876e-06,
      "step": 10188,
      "training_step_time": 17.116591453552246
    },
    {
      "epoch": 9.716987609863281e-06,
      "model_forward_time": 5.603095054626465,
      "step": 10189
    },
    {
      "epoch": 9.716987609863281e-06,
      "step": 10189,
      "training_step_time": 17.080236434936523
    },
    {
      "epoch": 9.717941284179687e-06,
      "grad_norm": 1.0552430152893066,
      "learning_rate": 1e-05,
      "loss": 0.0748,
      "step": 10190
    },
    {
      "epoch": 9.717941284179687e-06,
      "model_forward_time": 5.5953264236450195,
      "step": 10190
    },
    {
      "dynamics_loss_avg": 0.06658946387469769,
      "epoch": 9.717941284179687e-06,
      "step": 10190
    },
    {
      "action_loss_avg": 0.010297192633152008,
      "epoch": 9.717941284179687e-06,
      "step": 10190
    },
    {
      "epoch": 9.717941284179687e-06,
      "step": 10190,
      "training_step_time": 17.024735927581787
    },
    {
      "epoch": 9.718894958496094e-06,
      "model_forward_time": 5.603381872177124,
      "step": 10191
    },
    {
      "epoch": 9.718894958496094e-06,
      "step": 10191,
      "training_step_time": 17.05689525604248
    },
    {
      "epoch": 9.7198486328125e-06,
      "model_forward_time": 5.597131967544556,
      "step": 10192
    },
    {
      "epoch": 9.7198486328125e-06,
      "step": 10192,
      "training_step_time": 17.06444263458252
    },
    {
      "epoch": 9.720802307128906e-06,
      "model_forward_time": 5.976624965667725,
      "step": 10193
    },
    {
      "epoch": 9.720802307128906e-06,
      "step": 10193,
      "training_step_time": 17.46330952644348
    },
    {
      "epoch": 9.721755981445313e-06,
      "model_forward_time": 5.6096532344818115,
      "step": 10194
    },
    {
      "epoch": 9.721755981445313e-06,
      "step": 10194,
      "training_step_time": 17.100132942199707
    },
    {
      "epoch": 9.722709655761719e-06,
      "model_forward_time": 5.621806621551514,
      "step": 10195
    },
    {
      "epoch": 9.722709655761719e-06,
      "step": 10195,
      "training_step_time": 17.117783308029175
    },
    {
      "epoch": 9.723663330078125e-06,
      "model_forward_time": 5.619954824447632,
      "step": 10196
    },
    {
      "epoch": 9.723663330078125e-06,
      "step": 10196,
      "training_step_time": 17.12749409675598
    },
    {
      "epoch": 9.72461700439453e-06,
      "model_forward_time": 5.6085474491119385,
      "step": 10197
    },
    {
      "epoch": 9.72461700439453e-06,
      "step": 10197,
      "training_step_time": 17.108776092529297
    },
    {
      "epoch": 9.725570678710938e-06,
      "model_forward_time": 5.9654529094696045,
      "step": 10198
    },
    {
      "epoch": 9.725570678710938e-06,
      "step": 10198,
      "training_step_time": 17.44205927848816
    },
    {
      "epoch": 9.726524353027344e-06,
      "model_forward_time": 5.621577262878418,
      "step": 10199
    },
    {
      "epoch": 9.726524353027344e-06,
      "step": 10199,
      "training_step_time": 17.11485004425049
    },
    {
      "epoch": 9.72747802734375e-06,
      "grad_norm": 0.7027688026428223,
      "learning_rate": 1e-05,
      "loss": 0.0674,
      "step": 10200
    },
    {
      "epoch": 9.72747802734375e-06,
      "model_forward_time": 5.604891300201416,
      "step": 10200
    },
    {
      "dynamics_loss_avg": 0.059671522863209245,
      "epoch": 9.72747802734375e-06,
      "step": 10200
    },
    {
      "action_loss_avg": 0.007799041597172618,
      "epoch": 9.72747802734375e-06,
      "step": 10200
    },
    {
      "epoch": 9.72747802734375e-06,
      "step": 10200,
      "training_step_time": 17.07241463661194
    },
    {
      "epoch": 9.728431701660157e-06,
      "model_forward_time": 5.594988107681274,
      "step": 10201
    },
    {
      "epoch": 9.728431701660157e-06,
      "step": 10201,
      "training_step_time": 17.029969692230225
    },
    {
      "epoch": 9.729385375976562e-06,
      "model_forward_time": 5.57170295715332,
      "step": 10202
    },
    {
      "epoch": 9.729385375976562e-06,
      "step": 10202,
      "training_step_time": 17.02430009841919
    },
    {
      "epoch": 9.730339050292968e-06,
      "model_forward_time": 5.5869996547698975,
      "step": 10203
    },
    {
      "epoch": 9.730339050292968e-06,
      "step": 10203,
      "training_step_time": 17.014006853103638
    },
    {
      "epoch": 9.731292724609376e-06,
      "model_forward_time": 5.94212532043457,
      "step": 10204
    },
    {
      "epoch": 9.731292724609376e-06,
      "step": 10204,
      "training_step_time": 17.376713514328003
    },
    {
      "epoch": 9.732246398925781e-06,
      "model_forward_time": 5.598332166671753,
      "step": 10205
    },
    {
      "epoch": 9.732246398925781e-06,
      "step": 10205,
      "training_step_time": 17.04108190536499
    },
    {
      "epoch": 9.733200073242187e-06,
      "model_forward_time": 5.60739278793335,
      "step": 10206
    },
    {
      "epoch": 9.733200073242187e-06,
      "step": 10206,
      "training_step_time": 17.098222494125366
    },
    {
      "epoch": 9.734153747558594e-06,
      "model_forward_time": 5.629461288452148,
      "step": 10207
    },
    {
      "epoch": 9.734153747558594e-06,
      "step": 10207,
      "training_step_time": 17.152072429656982
    },
    {
      "epoch": 9.735107421875e-06,
      "model_forward_time": 5.630312919616699,
      "step": 10208
    },
    {
      "epoch": 9.735107421875e-06,
      "step": 10208,
      "training_step_time": 17.12104606628418
    },
    {
      "epoch": 9.736061096191406e-06,
      "model_forward_time": 5.961740970611572,
      "step": 10209
    },
    {
      "epoch": 9.736061096191406e-06,
      "step": 10209,
      "training_step_time": 17.40773105621338
    },
    {
      "epoch": 9.737014770507813e-06,
      "grad_norm": 0.8842027187347412,
      "learning_rate": 1e-05,
      "loss": 0.0791,
      "step": 10210
    },
    {
      "epoch": 9.737014770507813e-06,
      "model_forward_time": 5.589439392089844,
      "step": 10210
    },
    {
      "dynamics_loss_avg": 0.06592961419373751,
      "epoch": 9.737014770507813e-06,
      "step": 10210
    },
    {
      "action_loss_avg": 0.008844253746792674,
      "epoch": 9.737014770507813e-06,
      "step": 10210
    },
    {
      "epoch": 9.737014770507813e-06,
      "step": 10210,
      "training_step_time": 17.030447006225586
    },
    {
      "epoch": 9.737968444824219e-06,
      "model_forward_time": 5.63587212562561,
      "step": 10211
    },
    {
      "epoch": 9.737968444824219e-06,
      "step": 10211,
      "training_step_time": 17.151390314102173
    },
    {
      "epoch": 9.738922119140625e-06,
      "model_forward_time": 5.600564956665039,
      "step": 10212
    },
    {
      "epoch": 9.738922119140625e-06,
      "step": 10212,
      "training_step_time": 17.05839490890503
    },
    {
      "epoch": 9.73987579345703e-06,
      "model_forward_time": 5.602113962173462,
      "step": 10213
    },
    {
      "epoch": 9.73987579345703e-06,
      "step": 10213,
      "training_step_time": 17.051497220993042
    },
    {
      "epoch": 9.740829467773438e-06,
      "model_forward_time": 5.618488311767578,
      "step": 10214
    },
    {
      "epoch": 9.740829467773438e-06,
      "step": 10214,
      "training_step_time": 17.09114694595337
    },
    {
      "epoch": 9.741783142089844e-06,
      "model_forward_time": 6.001126289367676,
      "step": 10215
    },
    {
      "epoch": 9.741783142089844e-06,
      "step": 10215,
      "training_step_time": 17.48197340965271
    },
    {
      "epoch": 9.74273681640625e-06,
      "model_forward_time": 5.639760971069336,
      "step": 10216
    },
    {
      "epoch": 9.74273681640625e-06,
      "step": 10216,
      "training_step_time": 17.14276957511902
    },
    {
      "epoch": 9.743690490722657e-06,
      "model_forward_time": 5.624872922897339,
      "step": 10217
    },
    {
      "epoch": 9.743690490722657e-06,
      "step": 10217,
      "training_step_time": 17.098347663879395
    },
    {
      "epoch": 9.744644165039062e-06,
      "model_forward_time": 5.616062164306641,
      "step": 10218
    },
    {
      "epoch": 9.744644165039062e-06,
      "step": 10218,
      "training_step_time": 17.100799798965454
    },
    {
      "epoch": 9.745597839355468e-06,
      "model_forward_time": 5.615602254867554,
      "step": 10219
    },
    {
      "epoch": 9.745597839355468e-06,
      "step": 10219,
      "training_step_time": 17.11024284362793
    },
    {
      "epoch": 9.746551513671876e-06,
      "grad_norm": 0.4127914011478424,
      "learning_rate": 1e-05,
      "loss": 0.0595,
      "step": 10220
    },
    {
      "epoch": 9.746551513671876e-06,
      "model_forward_time": 5.972119569778442,
      "step": 10220
    },
    {
      "dynamics_loss_avg": 0.051904861070215705,
      "epoch": 9.746551513671876e-06,
      "step": 10220
    },
    {
      "action_loss_avg": 0.007082197081763298,
      "epoch": 9.746551513671876e-06,
      "step": 10220
    },
    {
      "epoch": 9.746551513671876e-06,
      "step": 10220,
      "training_step_time": 17.450145483016968
    },
    {
      "epoch": 9.747505187988281e-06,
      "model_forward_time": 5.617214918136597,
      "step": 10221
    },
    {
      "epoch": 9.747505187988281e-06,
      "step": 10221,
      "training_step_time": 17.05660581588745
    },
    {
      "epoch": 9.748458862304687e-06,
      "model_forward_time": 5.59166145324707,
      "step": 10222
    },
    {
      "epoch": 9.748458862304687e-06,
      "step": 10222,
      "training_step_time": 17.03100323677063
    },
    {
      "epoch": 9.749412536621094e-06,
      "model_forward_time": 5.590206146240234,
      "step": 10223
    },
    {
      "epoch": 9.749412536621094e-06,
      "step": 10223,
      "training_step_time": 17.048128128051758
    },
    {
      "epoch": 9.7503662109375e-06,
      "model_forward_time": 5.601112604141235,
      "step": 10224
    },
    {
      "epoch": 9.7503662109375e-06,
      "step": 10224,
      "training_step_time": 17.081331491470337
    },
    {
      "epoch": 9.751319885253906e-06,
      "model_forward_time": 5.61168646812439,
      "step": 10225
    },
    {
      "epoch": 9.751319885253906e-06,
      "step": 10225,
      "training_step_time": 17.10009741783142
    },
    {
      "epoch": 9.752273559570313e-06,
      "model_forward_time": 5.993390083312988,
      "step": 10226
    },
    {
      "epoch": 9.752273559570313e-06,
      "step": 10226,
      "training_step_time": 17.49302649497986
    },
    {
      "epoch": 9.753227233886719e-06,
      "model_forward_time": 5.616979360580444,
      "step": 10227
    },
    {
      "epoch": 9.753227233886719e-06,
      "step": 10227,
      "training_step_time": 17.10647749900818
    },
    {
      "epoch": 9.754180908203125e-06,
      "model_forward_time": 5.599512100219727,
      "step": 10228
    },
    {
      "epoch": 9.754180908203125e-06,
      "step": 10228,
      "training_step_time": 17.05651354789734
    },
    {
      "epoch": 9.75513458251953e-06,
      "model_forward_time": 5.60803747177124,
      "step": 10229
    },
    {
      "epoch": 9.75513458251953e-06,
      "step": 10229,
      "training_step_time": 17.077672958374023
    },
    {
      "epoch": 9.756088256835938e-06,
      "grad_norm": 1.4547338485717773,
      "learning_rate": 1e-05,
      "loss": 0.0506,
      "step": 10230
    },
    {
      "epoch": 9.756088256835938e-06,
      "model_forward_time": 5.596033811569214,
      "step": 10230
    },
    {
      "dynamics_loss_avg": 0.0465404037386179,
      "epoch": 9.756088256835938e-06,
      "step": 10230
    },
    {
      "action_loss_avg": 0.008655361272394657,
      "epoch": 9.756088256835938e-06,
      "step": 10230
    },
    {
      "epoch": 9.756088256835938e-06,
      "step": 10230,
      "training_step_time": 17.063401222229004
    },
    {
      "epoch": 9.757041931152344e-06,
      "model_forward_time": 5.959562301635742,
      "step": 10231
    },
    {
      "epoch": 9.757041931152344e-06,
      "step": 10231,
      "training_step_time": 17.440319538116455
    },
    {
      "epoch": 9.75799560546875e-06,
      "model_forward_time": 5.617427349090576,
      "step": 10232
    },
    {
      "epoch": 9.75799560546875e-06,
      "step": 10232,
      "training_step_time": 17.10040855407715
    },
    {
      "epoch": 9.758949279785157e-06,
      "model_forward_time": 5.60413384437561,
      "step": 10233
    },
    {
      "epoch": 9.758949279785157e-06,
      "step": 10233,
      "training_step_time": 17.080122232437134
    },
    {
      "epoch": 9.759902954101562e-06,
      "model_forward_time": 5.610167503356934,
      "step": 10234
    },
    {
      "epoch": 9.759902954101562e-06,
      "step": 10234,
      "training_step_time": 17.056506633758545
    },
    {
      "epoch": 9.760856628417968e-06,
      "model_forward_time": 5.595132827758789,
      "step": 10235
    },
    {
      "epoch": 9.760856628417968e-06,
      "step": 10235,
      "training_step_time": 17.013813734054565
    },
    {
      "epoch": 9.761810302734376e-06,
      "model_forward_time": 5.590881586074829,
      "step": 10236
    },
    {
      "epoch": 9.761810302734376e-06,
      "step": 10236,
      "training_step_time": 17.029547929763794
    },
    {
      "epoch": 9.762763977050781e-06,
      "model_forward_time": 5.963630199432373,
      "step": 10237
    },
    {
      "epoch": 9.762763977050781e-06,
      "step": 10237,
      "training_step_time": 17.412189722061157
    },
    {
      "epoch": 9.763717651367187e-06,
      "model_forward_time": 5.596650838851929,
      "step": 10238
    },
    {
      "epoch": 9.763717651367187e-06,
      "step": 10238,
      "training_step_time": 17.055750370025635
    },
    {
      "epoch": 9.764671325683594e-06,
      "model_forward_time": 5.596676349639893,
      "step": 10239
    },
    {
      "epoch": 9.764671325683594e-06,
      "step": 10239,
      "training_step_time": 17.0617938041687
    },
    {
      "epoch": 9.765625e-06,
      "grad_norm": 1.0338472127914429,
      "learning_rate": 1e-05,
      "loss": 0.0661,
      "step": 10240
    },
    {
      "epoch": 9.765625e-06,
      "model_forward_time": 5.594154596328735,
      "step": 10240
    },
    {
      "dynamics_loss_avg": 0.057889188453555104,
      "epoch": 9.765625e-06,
      "step": 10240
    },
    {
      "action_loss_avg": 0.009475363092496992,
      "epoch": 9.765625e-06,
      "step": 10240
    },
    {
      "epoch": 9.765625e-06,
      "step": 10240,
      "training_step_time": 17.032093048095703
    },
    {
      "epoch": 9.766578674316406e-06,
      "model_forward_time": 5.60139536857605,
      "step": 10241
    },
    {
      "epoch": 9.766578674316406e-06,
      "step": 10241,
      "training_step_time": 17.078830003738403
    },
    {
      "epoch": 9.767532348632813e-06,
      "model_forward_time": 5.602676630020142,
      "step": 10242
    },
    {
      "epoch": 9.767532348632813e-06,
      "step": 10242,
      "training_step_time": 17.464092254638672
    },
    {
      "epoch": 9.768486022949219e-06,
      "model_forward_time": 5.619638442993164,
      "step": 10243
    },
    {
      "epoch": 9.768486022949219e-06,
      "step": 10243,
      "training_step_time": 17.130841732025146
    },
    {
      "epoch": 9.769439697265625e-06,
      "model_forward_time": 5.622716426849365,
      "step": 10244
    },
    {
      "epoch": 9.769439697265625e-06,
      "step": 10244,
      "training_step_time": 17.140358209609985
    },
    {
      "epoch": 9.77039337158203e-06,
      "model_forward_time": 5.622358798980713,
      "step": 10245
    },
    {
      "epoch": 9.77039337158203e-06,
      "step": 10245,
      "training_step_time": 17.129053115844727
    },
    {
      "epoch": 9.771347045898438e-06,
      "model_forward_time": 5.613940477371216,
      "step": 10246
    },
    {
      "epoch": 9.771347045898438e-06,
      "step": 10246,
      "training_step_time": 17.11047625541687
    },
    {
      "epoch": 9.772300720214844e-06,
      "model_forward_time": 5.605801582336426,
      "step": 10247
    },
    {
      "epoch": 9.772300720214844e-06,
      "step": 10247,
      "training_step_time": 17.055943727493286
    },
    {
      "epoch": 9.77325439453125e-06,
      "model_forward_time": 6.002249479293823,
      "step": 10248
    },
    {
      "epoch": 9.77325439453125e-06,
      "step": 10248,
      "training_step_time": 17.466898918151855
    },
    {
      "epoch": 9.774208068847657e-06,
      "model_forward_time": 5.605825662612915,
      "step": 10249
    },
    {
      "epoch": 9.774208068847657e-06,
      "step": 10249,
      "training_step_time": 17.10333752632141
    },
    {
      "epoch": 9.775161743164062e-06,
      "grad_norm": 0.5635053515434265,
      "learning_rate": 1e-05,
      "loss": 0.0668,
      "step": 10250
    },
    {
      "epoch": 9.775161743164062e-06,
      "model_forward_time": 5.627237558364868,
      "step": 10250
    },
    {
      "dynamics_loss_avg": 0.06400258429348468,
      "epoch": 9.775161743164062e-06,
      "step": 10250
    },
    {
      "action_loss_avg": 0.008786366856656968,
      "epoch": 9.775161743164062e-06,
      "step": 10250
    },
    {
      "epoch": 9.775161743164062e-06,
      "step": 10250,
      "training_step_time": 17.160271167755127
    },
    {
      "epoch": 9.776115417480468e-06,
      "model_forward_time": 5.621204614639282,
      "step": 10251
    },
    {
      "epoch": 9.776115417480468e-06,
      "step": 10251,
      "training_step_time": 17.15071439743042
    },
    {
      "epoch": 9.777069091796876e-06,
      "model_forward_time": 5.621537923812866,
      "step": 10252
    },
    {
      "epoch": 9.777069091796876e-06,
      "step": 10252,
      "training_step_time": 17.098517417907715
    },
    {
      "epoch": 9.778022766113281e-06,
      "model_forward_time": 5.943982839584351,
      "step": 10253
    },
    {
      "epoch": 9.778022766113281e-06,
      "step": 10253,
      "training_step_time": 17.367960214614868
    },
    {
      "epoch": 9.778976440429687e-06,
      "model_forward_time": 5.566145658493042,
      "step": 10254
    },
    {
      "epoch": 9.778976440429687e-06,
      "step": 10254,
      "training_step_time": 16.986068964004517
    },
    {
      "epoch": 9.779930114746094e-06,
      "model_forward_time": 5.564302206039429,
      "step": 10255
    },
    {
      "epoch": 9.779930114746094e-06,
      "step": 10255,
      "training_step_time": 16.988953113555908
    },
    {
      "epoch": 9.7808837890625e-06,
      "model_forward_time": 5.556306600570679,
      "step": 10256
    },
    {
      "epoch": 9.7808837890625e-06,
      "step": 10256,
      "training_step_time": 16.968330144882202
    },
    {
      "epoch": 9.781837463378906e-06,
      "model_forward_time": 5.562429189682007,
      "step": 10257
    },
    {
      "epoch": 9.781837463378906e-06,
      "step": 10257,
      "training_step_time": 16.952086448669434
    },
    {
      "epoch": 9.782791137695312e-06,
      "model_forward_time": 5.576319217681885,
      "step": 10258
    },
    {
      "epoch": 9.782791137695312e-06,
      "step": 10258,
      "training_step_time": 17.00421643257141
    },
    {
      "epoch": 9.783744812011719e-06,
      "model_forward_time": 5.951371908187866,
      "step": 10259
    },
    {
      "epoch": 9.783744812011719e-06,
      "step": 10259,
      "training_step_time": 17.3815495967865
    },
    {
      "epoch": 9.784698486328125e-06,
      "grad_norm": 0.5581098198890686,
      "learning_rate": 1e-05,
      "loss": 0.0719,
      "step": 10260
    },
    {
      "epoch": 9.784698486328125e-06,
      "model_forward_time": 5.581876516342163,
      "step": 10260
    },
    {
      "dynamics_loss_avg": 0.055553235672414304,
      "epoch": 9.784698486328125e-06,
      "step": 10260
    },
    {
      "action_loss_avg": 0.008532402757555246,
      "epoch": 9.784698486328125e-06,
      "step": 10260
    },
    {
      "epoch": 9.784698486328125e-06,
      "step": 10260,
      "training_step_time": 17.03252863883972
    },
    {
      "epoch": 9.78565216064453e-06,
      "model_forward_time": 5.595242023468018,
      "step": 10261
    },
    {
      "epoch": 9.78565216064453e-06,
      "step": 10261,
      "training_step_time": 17.052671909332275
    },
    {
      "epoch": 9.786605834960938e-06,
      "model_forward_time": 5.602509260177612,
      "step": 10262
    },
    {
      "epoch": 9.786605834960938e-06,
      "step": 10262,
      "training_step_time": 17.074406147003174
    },
    {
      "epoch": 9.787559509277344e-06,
      "model_forward_time": 5.619041681289673,
      "step": 10263
    },
    {
      "epoch": 9.787559509277344e-06,
      "step": 10263,
      "training_step_time": 17.133729219436646
    },
    {
      "epoch": 9.78851318359375e-06,
      "model_forward_time": 6.053045034408569,
      "step": 10264
    },
    {
      "epoch": 9.78851318359375e-06,
      "step": 10264,
      "training_step_time": 17.56584358215332
    },
    {
      "epoch": 9.789466857910157e-06,
      "model_forward_time": 5.617748022079468,
      "step": 10265
    },
    {
      "epoch": 9.789466857910157e-06,
      "step": 10265,
      "training_step_time": 17.117120265960693
    },
    {
      "epoch": 9.790420532226563e-06,
      "model_forward_time": 5.616663217544556,
      "step": 10266
    },
    {
      "epoch": 9.790420532226563e-06,
      "step": 10266,
      "training_step_time": 17.089174270629883
    },
    {
      "epoch": 9.791374206542968e-06,
      "model_forward_time": 5.644224643707275,
      "step": 10267
    },
    {
      "epoch": 9.791374206542968e-06,
      "step": 10267,
      "training_step_time": 17.120718955993652
    },
    {
      "epoch": 9.792327880859376e-06,
      "model_forward_time": 5.637484550476074,
      "step": 10268
    },
    {
      "epoch": 9.792327880859376e-06,
      "step": 10268,
      "training_step_time": 17.138546466827393
    },
    {
      "epoch": 9.793281555175781e-06,
      "model_forward_time": 5.612246036529541,
      "step": 10269
    },
    {
      "epoch": 9.793281555175781e-06,
      "step": 10269,
      "training_step_time": 17.088794469833374
    },
    {
      "epoch": 9.794235229492187e-06,
      "grad_norm": 0.5057154893875122,
      "learning_rate": 1e-05,
      "loss": 0.0737,
      "step": 10270
    },
    {
      "epoch": 9.794235229492187e-06,
      "model_forward_time": 5.969868898391724,
      "step": 10270
    },
    {
      "dynamics_loss_avg": 0.06143159903585911,
      "epoch": 9.794235229492187e-06,
      "step": 10270
    },
    {
      "action_loss_avg": 0.00849062092602253,
      "epoch": 9.794235229492187e-06,
      "step": 10270
    },
    {
      "epoch": 9.794235229492187e-06,
      "step": 10270,
      "training_step_time": 17.421189546585083
    },
    {
      "epoch": 9.795188903808595e-06,
      "model_forward_time": 5.572547674179077,
      "step": 10271
    },
    {
      "epoch": 9.795188903808595e-06,
      "step": 10271,
      "training_step_time": 16.978554487228394
    },
    {
      "epoch": 9.796142578125e-06,
      "model_forward_time": 5.580935955047607,
      "step": 10272
    },
    {
      "epoch": 9.796142578125e-06,
      "step": 10272,
      "training_step_time": 16.998472213745117
    },
    {
      "epoch": 9.797096252441406e-06,
      "model_forward_time": 4.256890296936035,
      "step": 10273
    },
    {
      "epoch": 9.797096252441406e-06,
      "step": 10273,
      "training_step_time": 12.75872015953064
    },
    {
      "epoch": 9.798049926757812e-06,
      "model_forward_time": 5.584199905395508,
      "step": 10274
    },
    {
      "epoch": 9.798049926757812e-06,
      "step": 10274,
      "training_step_time": 17.04962396621704
    },
    {
      "epoch": 9.799003601074219e-06,
      "model_forward_time": 5.620200872421265,
      "step": 10275
    },
    {
      "epoch": 9.799003601074219e-06,
      "step": 10275,
      "training_step_time": 17.117372751235962
    },
    {
      "epoch": 9.799957275390625e-06,
      "model_forward_time": 5.971984148025513,
      "step": 10276
    },
    {
      "epoch": 9.799957275390625e-06,
      "step": 10276,
      "training_step_time": 17.485785722732544
    },
    {
      "epoch": 9.80091094970703e-06,
      "model_forward_time": 5.631410598754883,
      "step": 10277
    },
    {
      "epoch": 9.80091094970703e-06,
      "step": 10277,
      "training_step_time": 17.14937472343445
    },
    {
      "epoch": 9.801864624023438e-06,
      "model_forward_time": 4.3006250858306885,
      "step": 10278
    },
    {
      "epoch": 9.801864624023438e-06,
      "step": 10278,
      "training_step_time": 12.848254442214966
    },
    {
      "epoch": 9.802818298339844e-06,
      "model_forward_time": 5.614612579345703,
      "step": 10279
    },
    {
      "epoch": 9.802818298339844e-06,
      "step": 10279,
      "training_step_time": 17.069130182266235
    },
    {
      "epoch": 9.80377197265625e-06,
      "grad_norm": 0.781545877456665,
      "learning_rate": 1e-05,
      "loss": 0.0581,
      "step": 10280
    },
    {
      "epoch": 9.80377197265625e-06,
      "model_forward_time": 5.588172912597656,
      "step": 10280
    },
    {
      "dynamics_loss_avg": 0.053199582174420355,
      "epoch": 9.80377197265625e-06,
      "step": 10280
    },
    {
      "action_loss_avg": 0.00712819907348603,
      "epoch": 9.80377197265625e-06,
      "step": 10280
    },
    {
      "epoch": 9.80377197265625e-06,
      "step": 10280,
      "training_step_time": 17.011863231658936
    },
    {
      "epoch": 9.804725646972657e-06,
      "model_forward_time": 5.952401161193848,
      "step": 10281
    },
    {
      "epoch": 9.804725646972657e-06,
      "step": 10281,
      "training_step_time": 17.3787522315979
    },
    {
      "epoch": 9.805679321289063e-06,
      "model_forward_time": 5.605676889419556,
      "step": 10282
    },
    {
      "epoch": 9.805679321289063e-06,
      "step": 10282,
      "training_step_time": 17.070866346359253
    },
    {
      "epoch": 9.806632995605468e-06,
      "model_forward_time": 5.617821931838989,
      "step": 10283
    },
    {
      "epoch": 9.806632995605468e-06,
      "step": 10283,
      "training_step_time": 17.12274742126465
    },
    {
      "epoch": 9.807586669921876e-06,
      "model_forward_time": 5.618719100952148,
      "step": 10284
    },
    {
      "epoch": 9.807586669921876e-06,
      "step": 10284,
      "training_step_time": 17.13998031616211
    },
    {
      "epoch": 9.808540344238281e-06,
      "model_forward_time": 5.6191322803497314,
      "step": 10285
    },
    {
      "epoch": 9.808540344238281e-06,
      "step": 10285,
      "training_step_time": 17.123512506484985
    },
    {
      "epoch": 9.809494018554687e-06,
      "model_forward_time": 5.629391670227051,
      "step": 10286
    },
    {
      "epoch": 9.809494018554687e-06,
      "step": 10286,
      "training_step_time": 17.49856448173523
    },
    {
      "epoch": 9.810447692871095e-06,
      "model_forward_time": 5.611191987991333,
      "step": 10287
    },
    {
      "epoch": 9.810447692871095e-06,
      "step": 10287,
      "training_step_time": 17.077678203582764
    },
    {
      "epoch": 9.8114013671875e-06,
      "model_forward_time": 5.602572202682495,
      "step": 10288
    },
    {
      "epoch": 9.8114013671875e-06,
      "step": 10288,
      "training_step_time": 17.071431875228882
    },
    {
      "epoch": 9.812355041503906e-06,
      "model_forward_time": 5.604480028152466,
      "step": 10289
    },
    {
      "epoch": 9.812355041503906e-06,
      "step": 10289,
      "training_step_time": 17.085477590560913
    },
    {
      "epoch": 9.813308715820312e-06,
      "grad_norm": 0.786956250667572,
      "learning_rate": 1e-05,
      "loss": 0.0769,
      "step": 10290
    },
    {
      "epoch": 9.813308715820312e-06,
      "model_forward_time": 5.616778135299683,
      "step": 10290
    },
    {
      "dynamics_loss_avg": 0.0679010946303606,
      "epoch": 9.813308715820312e-06,
      "step": 10290
    },
    {
      "action_loss_avg": 0.01125103197991848,
      "epoch": 9.813308715820312e-06,
      "step": 10290
    },
    {
      "epoch": 9.813308715820312e-06,
      "step": 10290,
      "training_step_time": 17.115601778030396
    },
    {
      "epoch": 9.814262390136719e-06,
      "model_forward_time": 5.60007643699646,
      "step": 10291
    },
    {
      "epoch": 9.814262390136719e-06,
      "step": 10291,
      "training_step_time": 17.062736988067627
    },
    {
      "epoch": 9.815216064453125e-06,
      "model_forward_time": 5.96796727180481,
      "step": 10292
    },
    {
      "epoch": 9.815216064453125e-06,
      "step": 10292,
      "training_step_time": 17.3939208984375
    },
    {
      "epoch": 9.81616973876953e-06,
      "model_forward_time": 5.599761724472046,
      "step": 10293
    },
    {
      "epoch": 9.81616973876953e-06,
      "step": 10293,
      "training_step_time": 17.03497624397278
    },
    {
      "epoch": 9.817123413085938e-06,
      "model_forward_time": 5.595103025436401,
      "step": 10294
    },
    {
      "epoch": 9.817123413085938e-06,
      "step": 10294,
      "training_step_time": 17.078625917434692
    },
    {
      "epoch": 9.818077087402344e-06,
      "model_forward_time": 5.618107557296753,
      "step": 10295
    },
    {
      "epoch": 9.818077087402344e-06,
      "step": 10295,
      "training_step_time": 17.11204767227173
    },
    {
      "epoch": 9.81903076171875e-06,
      "model_forward_time": 5.6189775466918945,
      "step": 10296
    },
    {
      "epoch": 9.81903076171875e-06,
      "step": 10296,
      "training_step_time": 17.087199926376343
    },
    {
      "epoch": 9.819984436035157e-06,
      "model_forward_time": 5.975623607635498,
      "step": 10297
    },
    {
      "epoch": 9.819984436035157e-06,
      "step": 10297,
      "training_step_time": 17.423052310943604
    },
    {
      "epoch": 9.820938110351563e-06,
      "model_forward_time": 5.613939046859741,
      "step": 10298
    },
    {
      "epoch": 9.820938110351563e-06,
      "step": 10298,
      "training_step_time": 17.10304880142212
    },
    {
      "epoch": 9.821891784667968e-06,
      "model_forward_time": 5.610383033752441,
      "step": 10299
    },
    {
      "epoch": 9.821891784667968e-06,
      "step": 10299,
      "training_step_time": 17.119911670684814
    },
    {
      "epoch": 9.822845458984376e-06,
      "grad_norm": 1.5509471893310547,
      "learning_rate": 1e-05,
      "loss": 0.0858,
      "step": 10300
    },
    {
      "epoch": 9.822845458984376e-06,
      "model_forward_time": 5.622821092605591,
      "step": 10300
    },
    {
      "dynamics_loss_avg": 0.07615221180021763,
      "epoch": 9.822845458984376e-06,
      "step": 10300
    },
    {
      "action_loss_avg": 0.00970628389623016,
      "epoch": 9.822845458984376e-06,
      "step": 10300
    },
    {
      "epoch": 9.822845458984376e-06,
      "step": 10300,
      "training_step_time": 17.138477325439453
    },
    {
      "epoch": 9.823799133300781e-06,
      "model_forward_time": 5.622864246368408,
      "step": 10301
    },
    {
      "epoch": 9.823799133300781e-06,
      "step": 10301,
      "training_step_time": 17.142505407333374
    },
    {
      "epoch": 9.824752807617187e-06,
      "model_forward_time": 5.623484373092651,
      "step": 10302
    },
    {
      "epoch": 9.824752807617187e-06,
      "step": 10302,
      "training_step_time": 17.120105266571045
    },
    {
      "epoch": 9.825706481933595e-06,
      "model_forward_time": 5.979199171066284,
      "step": 10303
    },
    {
      "epoch": 9.825706481933595e-06,
      "step": 10303,
      "training_step_time": 17.455861568450928
    },
    {
      "epoch": 9.82666015625e-06,
      "model_forward_time": 5.597490072250366,
      "step": 10304
    },
    {
      "epoch": 9.82666015625e-06,
      "step": 10304,
      "training_step_time": 17.041693210601807
    },
    {
      "epoch": 9.827613830566406e-06,
      "model_forward_time": 5.593436002731323,
      "step": 10305
    },
    {
      "epoch": 9.827613830566406e-06,
      "step": 10305,
      "training_step_time": 17.046152591705322
    },
    {
      "epoch": 9.828567504882812e-06,
      "model_forward_time": 5.610924005508423,
      "step": 10306
    },
    {
      "epoch": 9.828567504882812e-06,
      "step": 10306,
      "training_step_time": 17.09394097328186
    },
    {
      "epoch": 9.829521179199219e-06,
      "model_forward_time": 5.61559796333313,
      "step": 10307
    },
    {
      "epoch": 9.829521179199219e-06,
      "step": 10307,
      "training_step_time": 17.114155292510986
    },
    {
      "epoch": 9.830474853515625e-06,
      "model_forward_time": 5.972755432128906,
      "step": 10308
    },
    {
      "epoch": 9.830474853515625e-06,
      "step": 10308,
      "training_step_time": 17.466737270355225
    },
    {
      "epoch": 9.83142852783203e-06,
      "model_forward_time": 5.600119590759277,
      "step": 10309
    },
    {
      "epoch": 9.83142852783203e-06,
      "step": 10309,
      "training_step_time": 17.052212476730347
    },
    {
      "epoch": 9.832382202148438e-06,
      "grad_norm": 0.6797165274620056,
      "learning_rate": 1e-05,
      "loss": 0.0695,
      "step": 10310
    },
    {
      "epoch": 9.832382202148438e-06,
      "model_forward_time": 5.577739715576172,
      "step": 10310
    },
    {
      "dynamics_loss_avg": 0.05597008764743805,
      "epoch": 9.832382202148438e-06,
      "step": 10310
    },
    {
      "action_loss_avg": 0.008510695514269173,
      "epoch": 9.832382202148438e-06,
      "step": 10310
    },
    {
      "epoch": 9.832382202148438e-06,
      "step": 10310,
      "training_step_time": 16.989819765090942
    },
    {
      "epoch": 9.833335876464844e-06,
      "model_forward_time": 5.5780298709869385,
      "step": 10311
    },
    {
      "epoch": 9.833335876464844e-06,
      "step": 10311,
      "training_step_time": 17.008445978164673
    },
    {
      "epoch": 9.83428955078125e-06,
      "model_forward_time": 5.57758355140686,
      "step": 10312
    },
    {
      "epoch": 9.83428955078125e-06,
      "step": 10312,
      "training_step_time": 17.02576780319214
    },
    {
      "epoch": 9.835243225097657e-06,
      "model_forward_time": 5.608278512954712,
      "step": 10313
    },
    {
      "epoch": 9.835243225097657e-06,
      "step": 10313,
      "training_step_time": 17.092111587524414
    },
    {
      "epoch": 9.836196899414063e-06,
      "model_forward_time": 6.015371561050415,
      "step": 10314
    },
    {
      "epoch": 9.836196899414063e-06,
      "step": 10314,
      "training_step_time": 17.530977964401245
    },
    {
      "epoch": 9.837150573730468e-06,
      "model_forward_time": 5.6107869148254395,
      "step": 10315
    },
    {
      "epoch": 9.837150573730468e-06,
      "step": 10315,
      "training_step_time": 17.069040775299072
    },
    {
      "epoch": 9.838104248046876e-06,
      "model_forward_time": 5.601041316986084,
      "step": 10316
    },
    {
      "epoch": 9.838104248046876e-06,
      "step": 10316,
      "training_step_time": 17.044177532196045
    },
    {
      "epoch": 9.839057922363281e-06,
      "model_forward_time": 5.585049152374268,
      "step": 10317
    },
    {
      "epoch": 9.839057922363281e-06,
      "step": 10317,
      "training_step_time": 16.993780851364136
    },
    {
      "epoch": 9.840011596679687e-06,
      "model_forward_time": 5.573358058929443,
      "step": 10318
    },
    {
      "epoch": 9.840011596679687e-06,
      "step": 10318,
      "training_step_time": 16.99796986579895
    },
    {
      "epoch": 9.840965270996095e-06,
      "model_forward_time": 5.9201624393463135,
      "step": 10319
    },
    {
      "epoch": 9.840965270996095e-06,
      "step": 10319,
      "training_step_time": 17.35692524909973
    },
    {
      "epoch": 9.8419189453125e-06,
      "grad_norm": 1.0398005247116089,
      "learning_rate": 1e-05,
      "loss": 0.0738,
      "step": 10320
    },
    {
      "epoch": 9.8419189453125e-06,
      "model_forward_time": 5.577199459075928,
      "step": 10320
    },
    {
      "dynamics_loss_avg": 0.06885480657219886,
      "epoch": 9.8419189453125e-06,
      "step": 10320
    },
    {
      "action_loss_avg": 0.010258249007165432,
      "epoch": 9.8419189453125e-06,
      "step": 10320
    },
    {
      "epoch": 9.8419189453125e-06,
      "step": 10320,
      "training_step_time": 17.017632007598877
    },
    {
      "epoch": 9.842872619628906e-06,
      "model_forward_time": 5.576285362243652,
      "step": 10321
    },
    {
      "epoch": 9.842872619628906e-06,
      "step": 10321,
      "training_step_time": 17.00313448905945
    },
    {
      "epoch": 9.843826293945312e-06,
      "model_forward_time": 5.588473558425903,
      "step": 10322
    },
    {
      "epoch": 9.843826293945312e-06,
      "step": 10322,
      "training_step_time": 17.06273627281189
    },
    {
      "epoch": 9.844779968261719e-06,
      "model_forward_time": 5.608654022216797,
      "step": 10323
    },
    {
      "epoch": 9.844779968261719e-06,
      "step": 10323,
      "training_step_time": 17.130824089050293
    },
    {
      "epoch": 9.845733642578125e-06,
      "model_forward_time": 5.619987487792969,
      "step": 10324
    },
    {
      "epoch": 9.845733642578125e-06,
      "step": 10324,
      "training_step_time": 17.12715172767639
    },
    {
      "epoch": 9.84668731689453e-06,
      "model_forward_time": 5.98662805557251,
      "step": 10325
    },
    {
      "epoch": 9.84668731689453e-06,
      "step": 10325,
      "training_step_time": 17.429543495178223
    },
    {
      "epoch": 9.847640991210938e-06,
      "model_forward_time": 5.5804665088653564,
      "step": 10326
    },
    {
      "epoch": 9.847640991210938e-06,
      "step": 10326,
      "training_step_time": 16.999939680099487
    },
    {
      "epoch": 9.848594665527344e-06,
      "model_forward_time": 5.582178115844727,
      "step": 10327
    },
    {
      "epoch": 9.848594665527344e-06,
      "step": 10327,
      "training_step_time": 17.010805368423462
    },
    {
      "epoch": 9.84954833984375e-06,
      "model_forward_time": 5.564819574356079,
      "step": 10328
    },
    {
      "epoch": 9.84954833984375e-06,
      "step": 10328,
      "training_step_time": 16.98720932006836
    },
    {
      "epoch": 9.850502014160157e-06,
      "model_forward_time": 5.584231853485107,
      "step": 10329
    },
    {
      "epoch": 9.850502014160157e-06,
      "step": 10329,
      "training_step_time": 16.997480869293213
    },
    {
      "epoch": 9.851455688476563e-06,
      "grad_norm": 0.6328449845314026,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 10330
    },
    {
      "epoch": 9.851455688476563e-06,
      "model_forward_time": 5.9314494132995605,
      "step": 10330
    },
    {
      "dynamics_loss_avg": 0.05636353455483913,
      "epoch": 9.851455688476563e-06,
      "step": 10330
    },
    {
      "action_loss_avg": 0.008040594263002276,
      "epoch": 9.851455688476563e-06,
      "step": 10330
    },
    {
      "epoch": 9.851455688476563e-06,
      "step": 10330,
      "training_step_time": 17.36633062362671
    },
    {
      "epoch": 9.852409362792968e-06,
      "model_forward_time": 5.58794641494751,
      "step": 10331
    },
    {
      "epoch": 9.852409362792968e-06,
      "step": 10331,
      "training_step_time": 17.036011695861816
    },
    {
      "epoch": 9.853363037109376e-06,
      "model_forward_time": 4.264855146408081,
      "step": 10332
    },
    {
      "epoch": 9.853363037109376e-06,
      "step": 10332,
      "training_step_time": 12.779109716415405
    },
    {
      "epoch": 9.854316711425781e-06,
      "model_forward_time": 5.623155117034912,
      "step": 10333
    },
    {
      "epoch": 9.854316711425781e-06,
      "step": 10333,
      "training_step_time": 17.10135841369629
    },
    {
      "epoch": 9.855270385742187e-06,
      "model_forward_time": 5.611624479293823,
      "step": 10334
    },
    {
      "epoch": 9.855270385742187e-06,
      "step": 10334,
      "training_step_time": 17.10641312599182
    },
    {
      "epoch": 9.856224060058595e-06,
      "model_forward_time": 5.608320236206055,
      "step": 10335
    },
    {
      "epoch": 9.856224060058595e-06,
      "step": 10335,
      "training_step_time": 17.07429027557373
    },
    {
      "epoch": 9.857177734375e-06,
      "model_forward_time": 5.963104724884033,
      "step": 10336
    },
    {
      "epoch": 9.857177734375e-06,
      "step": 10336,
      "training_step_time": 17.390647172927856
    },
    {
      "epoch": 9.858131408691406e-06,
      "model_forward_time": 5.5768938064575195,
      "step": 10337
    },
    {
      "epoch": 9.858131408691406e-06,
      "step": 10337,
      "training_step_time": 16.998109817504883
    },
    {
      "epoch": 9.859085083007812e-06,
      "model_forward_time": 5.5819971561431885,
      "step": 10338
    },
    {
      "epoch": 9.859085083007812e-06,
      "step": 10338,
      "training_step_time": 17.016085386276245
    },
    {
      "epoch": 9.860038757324219e-06,
      "model_forward_time": 5.593035459518433,
      "step": 10339
    },
    {
      "epoch": 9.860038757324219e-06,
      "step": 10339,
      "training_step_time": 17.042449474334717
    },
    {
      "epoch": 9.860992431640625e-06,
      "grad_norm": 1.3461028337478638,
      "learning_rate": 1e-05,
      "loss": 0.087,
      "step": 10340
    },
    {
      "epoch": 9.860992431640625e-06,
      "model_forward_time": 5.604855537414551,
      "step": 10340
    },
    {
      "dynamics_loss_avg": 0.07299985475838185,
      "epoch": 9.860992431640625e-06,
      "step": 10340
    },
    {
      "action_loss_avg": 0.012491630110889673,
      "epoch": 9.860992431640625e-06,
      "step": 10340
    },
    {
      "epoch": 9.860992431640625e-06,
      "step": 10340,
      "training_step_time": 17.078373432159424
    },
    {
      "epoch": 9.86194610595703e-06,
      "model_forward_time": 5.953383207321167,
      "step": 10341
    },
    {
      "epoch": 9.86194610595703e-06,
      "step": 10341,
      "training_step_time": 17.415667057037354
    },
    {
      "epoch": 9.862899780273438e-06,
      "model_forward_time": 5.595001459121704,
      "step": 10342
    },
    {
      "epoch": 9.862899780273438e-06,
      "step": 10342,
      "training_step_time": 17.032485008239746
    },
    {
      "epoch": 9.863853454589844e-06,
      "model_forward_time": 5.594794750213623,
      "step": 10343
    },
    {
      "epoch": 9.863853454589844e-06,
      "step": 10343,
      "training_step_time": 17.03869938850403
    },
    {
      "epoch": 9.86480712890625e-06,
      "model_forward_time": 5.596177339553833,
      "step": 10344
    },
    {
      "epoch": 9.86480712890625e-06,
      "step": 10344,
      "training_step_time": 17.047698497772217
    },
    {
      "epoch": 9.865760803222657e-06,
      "model_forward_time": 5.595608949661255,
      "step": 10345
    },
    {
      "epoch": 9.865760803222657e-06,
      "step": 10345,
      "training_step_time": 17.051637649536133
    },
    {
      "epoch": 9.866714477539063e-06,
      "model_forward_time": 5.982510328292847,
      "step": 10346
    },
    {
      "epoch": 9.866714477539063e-06,
      "step": 10346,
      "training_step_time": 17.41732406616211
    },
    {
      "epoch": 9.867668151855468e-06,
      "model_forward_time": 5.5919106006622314,
      "step": 10347
    },
    {
      "epoch": 9.867668151855468e-06,
      "step": 10347,
      "training_step_time": 17.04551100730896
    },
    {
      "epoch": 9.868621826171876e-06,
      "model_forward_time": 5.602635145187378,
      "step": 10348
    },
    {
      "epoch": 9.868621826171876e-06,
      "step": 10348,
      "training_step_time": 17.064720392227173
    },
    {
      "epoch": 9.869575500488281e-06,
      "model_forward_time": 5.601892471313477,
      "step": 10349
    },
    {
      "epoch": 9.869575500488281e-06,
      "step": 10349,
      "training_step_time": 17.07572627067566
    },
    {
      "epoch": 9.870529174804687e-06,
      "grad_norm": 0.6539949178695679,
      "learning_rate": 1e-05,
      "loss": 0.0572,
      "step": 10350
    },
    {
      "epoch": 9.870529174804687e-06,
      "model_forward_time": 5.608263254165649,
      "step": 10350
    },
    {
      "dynamics_loss_avg": 0.0487543111667037,
      "epoch": 9.870529174804687e-06,
      "step": 10350
    },
    {
      "action_loss_avg": 0.008003383199684322,
      "epoch": 9.870529174804687e-06,
      "step": 10350
    },
    {
      "epoch": 9.870529174804687e-06,
      "step": 10350,
      "training_step_time": 17.081644773483276
    },
    {
      "epoch": 9.871482849121095e-06,
      "model_forward_time": 5.616907119750977,
      "step": 10351
    },
    {
      "epoch": 9.871482849121095e-06,
      "step": 10351,
      "training_step_time": 17.12279486656189
    },
    {
      "epoch": 9.8724365234375e-06,
      "model_forward_time": 5.9738547801971436,
      "step": 10352
    },
    {
      "epoch": 9.8724365234375e-06,
      "step": 10352,
      "training_step_time": 17.446739435195923
    },
    {
      "epoch": 9.873390197753906e-06,
      "model_forward_time": 5.583232164382935,
      "step": 10353
    },
    {
      "epoch": 9.873390197753906e-06,
      "step": 10353,
      "training_step_time": 16.997131824493408
    },
    {
      "epoch": 9.874343872070312e-06,
      "model_forward_time": 5.6062095165252686,
      "step": 10354
    },
    {
      "epoch": 9.874343872070312e-06,
      "step": 10354,
      "training_step_time": 17.062183380126953
    },
    {
      "epoch": 9.87529754638672e-06,
      "model_forward_time": 5.57529091835022,
      "step": 10355
    },
    {
      "epoch": 9.87529754638672e-06,
      "step": 10355,
      "training_step_time": 16.986708879470825
    },
    {
      "epoch": 9.876251220703125e-06,
      "model_forward_time": 5.567673206329346,
      "step": 10356
    },
    {
      "epoch": 9.876251220703125e-06,
      "step": 10356,
      "training_step_time": 16.986103534698486
    },
    {
      "epoch": 9.87720489501953e-06,
      "model_forward_time": 5.951169490814209,
      "step": 10357
    },
    {
      "epoch": 9.87720489501953e-06,
      "step": 10357,
      "training_step_time": 17.368303298950195
    },
    {
      "epoch": 9.878158569335938e-06,
      "model_forward_time": 5.62932825088501,
      "step": 10358
    },
    {
      "epoch": 9.878158569335938e-06,
      "step": 10358,
      "training_step_time": 17.076053857803345
    },
    {
      "epoch": 9.879112243652344e-06,
      "model_forward_time": 5.586240291595459,
      "step": 10359
    },
    {
      "epoch": 9.879112243652344e-06,
      "step": 10359,
      "training_step_time": 17.01523447036743
    },
    {
      "epoch": 9.88006591796875e-06,
      "grad_norm": 0.839489758014679,
      "learning_rate": 1e-05,
      "loss": 0.0708,
      "step": 10360
    },
    {
      "epoch": 9.88006591796875e-06,
      "model_forward_time": 5.6034321784973145,
      "step": 10360
    },
    {
      "dynamics_loss_avg": 0.06359410583972931,
      "epoch": 9.88006591796875e-06,
      "step": 10360
    },
    {
      "action_loss_avg": 0.008661832939833403,
      "epoch": 9.88006591796875e-06,
      "step": 10360
    },
    {
      "epoch": 9.88006591796875e-06,
      "step": 10360,
      "training_step_time": 17.06873607635498
    },
    {
      "epoch": 9.881019592285157e-06,
      "model_forward_time": 5.621313095092773,
      "step": 10361
    },
    {
      "epoch": 9.881019592285157e-06,
      "step": 10361,
      "training_step_time": 17.07589817047119
    },
    {
      "epoch": 9.881973266601563e-06,
      "model_forward_time": 5.608433485031128,
      "step": 10362
    },
    {
      "epoch": 9.881973266601563e-06,
      "step": 10362,
      "training_step_time": 17.092825412750244
    },
    {
      "epoch": 9.882926940917968e-06,
      "model_forward_time": 5.957066535949707,
      "step": 10363
    },
    {
      "epoch": 9.882926940917968e-06,
      "step": 10363,
      "training_step_time": 17.414875507354736
    },
    {
      "epoch": 9.883880615234376e-06,
      "model_forward_time": 5.591645240783691,
      "step": 10364
    },
    {
      "epoch": 9.883880615234376e-06,
      "step": 10364,
      "training_step_time": 17.017158269882202
    },
    {
      "epoch": 9.884834289550781e-06,
      "model_forward_time": 5.583439826965332,
      "step": 10365
    },
    {
      "epoch": 9.884834289550781e-06,
      "step": 10365,
      "training_step_time": 17.005057096481323
    },
    {
      "epoch": 9.885787963867187e-06,
      "model_forward_time": 5.5900349617004395,
      "step": 10366
    },
    {
      "epoch": 9.885787963867187e-06,
      "step": 10366,
      "training_step_time": 17.06666398048401
    },
    {
      "epoch": 9.886741638183595e-06,
      "model_forward_time": 5.605072021484375,
      "step": 10367
    },
    {
      "epoch": 9.886741638183595e-06,
      "step": 10367,
      "training_step_time": 17.088032245635986
    },
    {
      "epoch": 9.8876953125e-06,
      "model_forward_time": 5.9827680587768555,
      "step": 10368
    },
    {
      "epoch": 9.8876953125e-06,
      "step": 10368,
      "training_step_time": 17.460450649261475
    },
    {
      "epoch": 9.888648986816406e-06,
      "model_forward_time": 5.609413146972656,
      "step": 10369
    },
    {
      "epoch": 9.888648986816406e-06,
      "step": 10369,
      "training_step_time": 17.09734058380127
    },
    {
      "epoch": 9.889602661132812e-06,
      "grad_norm": 0.339920312166214,
      "learning_rate": 1e-05,
      "loss": 0.0631,
      "step": 10370
    },
    {
      "epoch": 9.889602661132812e-06,
      "model_forward_time": 5.619944334030151,
      "step": 10370
    },
    {
      "dynamics_loss_avg": 0.05708938948810101,
      "epoch": 9.889602661132812e-06,
      "step": 10370
    },
    {
      "action_loss_avg": 0.007101597171276808,
      "epoch": 9.889602661132812e-06,
      "step": 10370
    },
    {
      "epoch": 9.889602661132812e-06,
      "step": 10370,
      "training_step_time": 17.09220004081726
    },
    {
      "epoch": 9.89055633544922e-06,
      "model_forward_time": 5.6046624183654785,
      "step": 10371
    },
    {
      "epoch": 9.89055633544922e-06,
      "step": 10371,
      "training_step_time": 17.075056791305542
    },
    {
      "epoch": 9.891510009765625e-06,
      "model_forward_time": 5.593181848526001,
      "step": 10372
    },
    {
      "epoch": 9.891510009765625e-06,
      "step": 10372,
      "training_step_time": 17.051865100860596
    },
    {
      "epoch": 9.89246368408203e-06,
      "model_forward_time": 5.597214698791504,
      "step": 10373
    },
    {
      "epoch": 9.89246368408203e-06,
      "step": 10373,
      "training_step_time": 17.071031093597412
    },
    {
      "epoch": 9.893417358398438e-06,
      "model_forward_time": 5.9553210735321045,
      "step": 10374
    },
    {
      "epoch": 9.893417358398438e-06,
      "step": 10374,
      "training_step_time": 17.450371980667114
    },
    {
      "epoch": 9.894371032714844e-06,
      "model_forward_time": 5.6256797313690186,
      "step": 10375
    },
    {
      "epoch": 9.894371032714844e-06,
      "step": 10375,
      "training_step_time": 17.12925696372986
    },
    {
      "epoch": 9.89532470703125e-06,
      "model_forward_time": 5.605825424194336,
      "step": 10376
    },
    {
      "epoch": 9.89532470703125e-06,
      "step": 10376,
      "training_step_time": 17.062500953674316
    },
    {
      "epoch": 9.896278381347657e-06,
      "model_forward_time": 5.587687253952026,
      "step": 10377
    },
    {
      "epoch": 9.896278381347657e-06,
      "step": 10377,
      "training_step_time": 17.030179977416992
    },
    {
      "epoch": 9.897232055664063e-06,
      "model_forward_time": 5.597555875778198,
      "step": 10378
    },
    {
      "epoch": 9.897232055664063e-06,
      "step": 10378,
      "training_step_time": 17.068259477615356
    },
    {
      "epoch": 9.898185729980468e-06,
      "model_forward_time": 5.978975772857666,
      "step": 10379
    },
    {
      "epoch": 9.898185729980468e-06,
      "step": 10379,
      "training_step_time": 17.47847580909729
    },
    {
      "epoch": 9.899139404296876e-06,
      "grad_norm": 0.20945771038532257,
      "learning_rate": 1e-05,
      "loss": 0.0523,
      "step": 10380
    },
    {
      "epoch": 9.899139404296876e-06,
      "model_forward_time": 5.599375009536743,
      "step": 10380
    },
    {
      "dynamics_loss_avg": 0.05057316720485687,
      "epoch": 9.899139404296876e-06,
      "step": 10380
    },
    {
      "action_loss_avg": 0.007049023546278477,
      "epoch": 9.899139404296876e-06,
      "step": 10380
    },
    {
      "epoch": 9.899139404296876e-06,
      "step": 10380,
      "training_step_time": 17.06482195854187
    },
    {
      "epoch": 9.900093078613281e-06,
      "model_forward_time": 5.607100486755371,
      "step": 10381
    },
    {
      "epoch": 9.900093078613281e-06,
      "step": 10381,
      "training_step_time": 17.069162130355835
    },
    {
      "epoch": 9.901046752929687e-06,
      "model_forward_time": 5.5987708568573,
      "step": 10382
    },
    {
      "epoch": 9.901046752929687e-06,
      "step": 10382,
      "training_step_time": 17.04836916923523
    },
    {
      "epoch": 9.902000427246093e-06,
      "model_forward_time": 5.5962748527526855,
      "step": 10383
    },
    {
      "epoch": 9.902000427246093e-06,
      "step": 10383,
      "training_step_time": 17.06050682067871
    },
    {
      "epoch": 9.9029541015625e-06,
      "model_forward_time": 5.953381061553955,
      "step": 10384
    },
    {
      "epoch": 9.9029541015625e-06,
      "step": 10384,
      "training_step_time": 17.392687797546387
    },
    {
      "epoch": 9.903907775878906e-06,
      "model_forward_time": 5.595030784606934,
      "step": 10385
    },
    {
      "epoch": 9.903907775878906e-06,
      "step": 10385,
      "training_step_time": 17.043182373046875
    },
    {
      "epoch": 9.904861450195312e-06,
      "model_forward_time": 5.597162485122681,
      "step": 10386
    },
    {
      "epoch": 9.904861450195312e-06,
      "step": 10386,
      "training_step_time": 17.07478666305542
    },
    {
      "epoch": 9.90581512451172e-06,
      "model_forward_time": 5.605114459991455,
      "step": 10387
    },
    {
      "epoch": 9.90581512451172e-06,
      "step": 10387,
      "training_step_time": 17.092331647872925
    },
    {
      "epoch": 9.906768798828125e-06,
      "model_forward_time": 5.604345321655273,
      "step": 10388
    },
    {
      "epoch": 9.906768798828125e-06,
      "step": 10388,
      "training_step_time": 17.065154314041138
    },
    {
      "epoch": 9.90772247314453e-06,
      "model_forward_time": 5.606897830963135,
      "step": 10389
    },
    {
      "epoch": 9.90772247314453e-06,
      "step": 10389,
      "training_step_time": 17.077110528945923
    },
    {
      "epoch": 9.908676147460938e-06,
      "grad_norm": 0.3037554621696472,
      "learning_rate": 1e-05,
      "loss": 0.0586,
      "step": 10390
    },
    {
      "epoch": 9.908676147460938e-06,
      "model_forward_time": 5.9668378829956055,
      "step": 10390
    },
    {
      "dynamics_loss_avg": 0.04239078080281615,
      "epoch": 9.908676147460938e-06,
      "step": 10390
    },
    {
      "action_loss_avg": 0.007506846135947853,
      "epoch": 9.908676147460938e-06,
      "step": 10390
    },
    {
      "epoch": 9.908676147460938e-06,
      "step": 10390,
      "training_step_time": 17.415843725204468
    },
    {
      "epoch": 9.909629821777344e-06,
      "model_forward_time": 5.598902702331543,
      "step": 10391
    },
    {
      "epoch": 9.909629821777344e-06,
      "step": 10391,
      "training_step_time": 17.091871738433838
    },
    {
      "epoch": 9.91058349609375e-06,
      "model_forward_time": 5.616496562957764,
      "step": 10392
    },
    {
      "epoch": 9.91058349609375e-06,
      "step": 10392,
      "training_step_time": 17.140902042388916
    },
    {
      "epoch": 9.911537170410157e-06,
      "model_forward_time": 5.67988133430481,
      "step": 10393
    },
    {
      "epoch": 9.911537170410157e-06,
      "step": 10393,
      "training_step_time": 17.19697666168213
    },
    {
      "epoch": 9.912490844726563e-06,
      "model_forward_time": 5.621194124221802,
      "step": 10394
    },
    {
      "epoch": 9.912490844726563e-06,
      "step": 10394,
      "training_step_time": 17.06786322593689
    },
    {
      "epoch": 9.913444519042968e-06,
      "model_forward_time": 5.589539527893066,
      "step": 10395
    },
    {
      "epoch": 9.913444519042968e-06,
      "step": 10395,
      "training_step_time": 17.392361164093018
    },
    {
      "epoch": 9.914398193359376e-06,
      "model_forward_time": 5.591056823730469,
      "step": 10396
    },
    {
      "epoch": 9.914398193359376e-06,
      "step": 10396,
      "training_step_time": 17.043786764144897
    },
    {
      "epoch": 9.915351867675781e-06,
      "model_forward_time": 5.574341773986816,
      "step": 10397
    },
    {
      "epoch": 9.915351867675781e-06,
      "step": 10397,
      "training_step_time": 16.99528479576111
    },
    {
      "epoch": 9.916305541992187e-06,
      "model_forward_time": 5.606164216995239,
      "step": 10398
    },
    {
      "epoch": 9.916305541992187e-06,
      "step": 10398,
      "training_step_time": 17.033513069152832
    },
    {
      "epoch": 9.917259216308593e-06,
      "model_forward_time": 5.591775178909302,
      "step": 10399
    },
    {
      "epoch": 9.917259216308593e-06,
      "step": 10399,
      "training_step_time": 17.039748668670654
    },
    {
      "epoch": 9.918212890625e-06,
      "grad_norm": 0.7468281388282776,
      "learning_rate": 1e-05,
      "loss": 0.0513,
      "step": 10400
    },
    {
      "epoch": 9.918212890625e-06,
      "model_forward_time": 5.5980401039123535,
      "step": 10400
    },
    {
      "dynamics_loss_avg": 0.047127819061279295,
      "epoch": 9.918212890625e-06,
      "step": 10400
    },
    {
      "action_loss_avg": 0.007329965056851506,
      "epoch": 9.918212890625e-06,
      "step": 10400
    },
    {
      "epoch": 9.918212890625e-06,
      "step": 10400,
      "training_step_time": 17.061001777648926
    },
    {
      "epoch": 9.919166564941406e-06,
      "model_forward_time": 5.9760215282440186,
      "step": 10401
    },
    {
      "epoch": 9.919166564941406e-06,
      "step": 10401,
      "training_step_time": 17.45030379295349
    },
    {
      "epoch": 9.920120239257812e-06,
      "model_forward_time": 5.612456798553467,
      "step": 10402
    },
    {
      "epoch": 9.920120239257812e-06,
      "step": 10402,
      "training_step_time": 17.113242864608765
    },
    {
      "epoch": 9.92107391357422e-06,
      "model_forward_time": 5.676122665405273,
      "step": 10403
    },
    {
      "epoch": 9.92107391357422e-06,
      "step": 10403,
      "training_step_time": 17.18963050842285
    },
    {
      "epoch": 9.922027587890625e-06,
      "model_forward_time": 5.599135875701904,
      "step": 10404
    },
    {
      "epoch": 9.922027587890625e-06,
      "step": 10404,
      "training_step_time": 17.050286054611206
    },
    {
      "epoch": 9.92298126220703e-06,
      "model_forward_time": 5.595586776733398,
      "step": 10405
    },
    {
      "epoch": 9.92298126220703e-06,
      "step": 10405,
      "training_step_time": 17.03122854232788
    },
    {
      "epoch": 9.923934936523438e-06,
      "model_forward_time": 5.582634925842285,
      "step": 10406
    },
    {
      "epoch": 9.923934936523438e-06,
      "step": 10406,
      "training_step_time": 17.385701894760132
    },
    {
      "epoch": 9.924888610839844e-06,
      "model_forward_time": 5.584451913833618,
      "step": 10407
    },
    {
      "epoch": 9.924888610839844e-06,
      "step": 10407,
      "training_step_time": 17.029485940933228
    },
    {
      "epoch": 9.92584228515625e-06,
      "model_forward_time": 5.59033203125,
      "step": 10408
    },
    {
      "epoch": 9.92584228515625e-06,
      "step": 10408,
      "training_step_time": 17.0311541557312
    },
    {
      "epoch": 9.926795959472657e-06,
      "model_forward_time": 4.270864963531494,
      "step": 10409
    },
    {
      "epoch": 9.926795959472657e-06,
      "step": 10409,
      "training_step_time": 12.77566409111023
    },
    {
      "epoch": 9.927749633789063e-06,
      "grad_norm": 0.5367735624313354,
      "learning_rate": 1e-05,
      "loss": 0.0661,
      "step": 10410
    },
    {
      "epoch": 9.927749633789063e-06,
      "model_forward_time": 5.588457107543945,
      "step": 10410
    },
    {
      "dynamics_loss_avg": 0.06201150668784976,
      "epoch": 9.927749633789063e-06,
      "step": 10410
    },
    {
      "action_loss_avg": 0.010005380702205002,
      "epoch": 9.927749633789063e-06,
      "step": 10410
    },
    {
      "epoch": 9.927749633789063e-06,
      "step": 10410,
      "training_step_time": 17.03962206840515
    },
    {
      "epoch": 9.928703308105468e-06,
      "model_forward_time": 5.602672815322876,
      "step": 10411
    },
    {
      "epoch": 9.928703308105468e-06,
      "step": 10411,
      "training_step_time": 17.07676386833191
    },
    {
      "epoch": 9.929656982421876e-06,
      "model_forward_time": 5.986586570739746,
      "step": 10412
    },
    {
      "epoch": 9.929656982421876e-06,
      "step": 10412,
      "training_step_time": 17.46799945831299
    },
    {
      "epoch": 9.930610656738281e-06,
      "model_forward_time": 5.610486268997192,
      "step": 10413
    },
    {
      "epoch": 9.930610656738281e-06,
      "step": 10413,
      "training_step_time": 17.097803831100464
    },
    {
      "epoch": 9.931564331054687e-06,
      "model_forward_time": 5.612875938415527,
      "step": 10414
    },
    {
      "epoch": 9.931564331054687e-06,
      "step": 10414,
      "training_step_time": 17.094366550445557
    },
    {
      "epoch": 9.932518005371093e-06,
      "model_forward_time": 5.602603912353516,
      "step": 10415
    },
    {
      "epoch": 9.932518005371093e-06,
      "step": 10415,
      "training_step_time": 17.06439971923828
    },
    {
      "epoch": 9.9334716796875e-06,
      "model_forward_time": 5.598226308822632,
      "step": 10416
    },
    {
      "epoch": 9.9334716796875e-06,
      "step": 10416,
      "training_step_time": 17.048144340515137
    },
    {
      "epoch": 9.934425354003906e-06,
      "model_forward_time": 5.5974156856536865,
      "step": 10417
    },
    {
      "epoch": 9.934425354003906e-06,
      "step": 10417,
      "training_step_time": 17.400083303451538
    },
    {
      "epoch": 9.935379028320312e-06,
      "model_forward_time": 5.595301866531372,
      "step": 10418
    },
    {
      "epoch": 9.935379028320312e-06,
      "step": 10418,
      "training_step_time": 17.093311548233032
    },
    {
      "epoch": 9.93633270263672e-06,
      "model_forward_time": 5.640786170959473,
      "step": 10419
    },
    {
      "epoch": 9.93633270263672e-06,
      "step": 10419,
      "training_step_time": 17.185248374938965
    },
    {
      "epoch": 9.937286376953125e-06,
      "grad_norm": 0.5916573405265808,
      "learning_rate": 1e-05,
      "loss": 0.0656,
      "step": 10420
    },
    {
      "epoch": 9.937286376953125e-06,
      "model_forward_time": 5.628174781799316,
      "step": 10420
    },
    {
      "dynamics_loss_avg": 0.05719385603442788,
      "epoch": 9.937286376953125e-06,
      "step": 10420
    },
    {
      "action_loss_avg": 0.007484596822177991,
      "epoch": 9.937286376953125e-06,
      "step": 10420
    },
    {
      "epoch": 9.937286376953125e-06,
      "step": 10420,
      "training_step_time": 17.117255926132202
    },
    {
      "epoch": 9.93824005126953e-06,
      "model_forward_time": 5.606948137283325,
      "step": 10421
    },
    {
      "epoch": 9.93824005126953e-06,
      "step": 10421,
      "training_step_time": 17.046772718429565
    },
    {
      "epoch": 9.939193725585938e-06,
      "model_forward_time": 5.5878989696502686,
      "step": 10422
    },
    {
      "epoch": 9.939193725585938e-06,
      "step": 10422,
      "training_step_time": 17.0287926197052
    },
    {
      "epoch": 9.940147399902344e-06,
      "model_forward_time": 5.945169687271118,
      "step": 10423
    },
    {
      "epoch": 9.940147399902344e-06,
      "step": 10423,
      "training_step_time": 17.35957980155945
    },
    {
      "epoch": 9.94110107421875e-06,
      "model_forward_time": 5.5906007289886475,
      "step": 10424
    },
    {
      "epoch": 9.94110107421875e-06,
      "step": 10424,
      "training_step_time": 17.024855852127075
    },
    {
      "epoch": 9.942054748535157e-06,
      "model_forward_time": 5.587435007095337,
      "step": 10425
    },
    {
      "epoch": 9.942054748535157e-06,
      "step": 10425,
      "training_step_time": 17.038971185684204
    },
    {
      "epoch": 9.943008422851563e-06,
      "model_forward_time": 5.595102548599243,
      "step": 10426
    },
    {
      "epoch": 9.943008422851563e-06,
      "step": 10426,
      "training_step_time": 17.045961141586304
    },
    {
      "epoch": 9.943962097167968e-06,
      "model_forward_time": 5.598321914672852,
      "step": 10427
    },
    {
      "epoch": 9.943962097167968e-06,
      "step": 10427,
      "training_step_time": 17.076009035110474
    },
    {
      "epoch": 9.944915771484376e-06,
      "model_forward_time": 5.974167346954346,
      "step": 10428
    },
    {
      "epoch": 9.944915771484376e-06,
      "step": 10428,
      "training_step_time": 17.463813543319702
    },
    {
      "epoch": 9.945869445800782e-06,
      "model_forward_time": 5.623178482055664,
      "step": 10429
    },
    {
      "epoch": 9.945869445800782e-06,
      "step": 10429,
      "training_step_time": 17.13483238220215
    },
    {
      "epoch": 9.946823120117187e-06,
      "grad_norm": 0.7693153023719788,
      "learning_rate": 1e-05,
      "loss": 0.0614,
      "step": 10430
    },
    {
      "epoch": 9.946823120117187e-06,
      "model_forward_time": 5.6162190437316895,
      "step": 10430
    },
    {
      "dynamics_loss_avg": 0.04669892527163029,
      "epoch": 9.946823120117187e-06,
      "step": 10430
    },
    {
      "action_loss_avg": 0.0050723109277896585,
      "epoch": 9.946823120117187e-06,
      "step": 10430
    },
    {
      "epoch": 9.946823120117187e-06,
      "step": 10430,
      "training_step_time": 17.096721172332764
    },
    {
      "epoch": 9.947776794433593e-06,
      "model_forward_time": 5.603524684906006,
      "step": 10431
    },
    {
      "epoch": 9.947776794433593e-06,
      "step": 10431,
      "training_step_time": 17.03209662437439
    },
    {
      "epoch": 9.94873046875e-06,
      "model_forward_time": 5.5735228061676025,
      "step": 10432
    },
    {
      "epoch": 9.94873046875e-06,
      "step": 10432,
      "training_step_time": 17.005012273788452
    },
    {
      "epoch": 9.949684143066406e-06,
      "model_forward_time": 5.573214769363403,
      "step": 10433
    },
    {
      "epoch": 9.949684143066406e-06,
      "step": 10433,
      "training_step_time": 16.980260610580444
    },
    {
      "epoch": 9.950637817382812e-06,
      "model_forward_time": 5.933109998703003,
      "step": 10434
    },
    {
      "epoch": 9.950637817382812e-06,
      "step": 10434,
      "training_step_time": 17.373700380325317
    },
    {
      "epoch": 9.95159149169922e-06,
      "model_forward_time": 5.570363521575928,
      "step": 10435
    },
    {
      "epoch": 9.95159149169922e-06,
      "step": 10435,
      "training_step_time": 16.9826021194458
    },
    {
      "epoch": 9.952545166015625e-06,
      "model_forward_time": 5.591625690460205,
      "step": 10436
    },
    {
      "epoch": 9.952545166015625e-06,
      "step": 10436,
      "training_step_time": 17.01532793045044
    },
    {
      "epoch": 9.95349884033203e-06,
      "model_forward_time": 5.591531991958618,
      "step": 10437
    },
    {
      "epoch": 9.95349884033203e-06,
      "step": 10437,
      "training_step_time": 17.036800384521484
    },
    {
      "epoch": 9.954452514648438e-06,
      "model_forward_time": 5.580304861068726,
      "step": 10438
    },
    {
      "epoch": 9.954452514648438e-06,
      "step": 10438,
      "training_step_time": 17.021342515945435
    },
    {
      "epoch": 9.955406188964844e-06,
      "model_forward_time": 5.948384523391724,
      "step": 10439
    },
    {
      "epoch": 9.955406188964844e-06,
      "step": 10439,
      "training_step_time": 17.40188956260681
    },
    {
      "epoch": 9.95635986328125e-06,
      "grad_norm": 0.6910807490348816,
      "learning_rate": 1e-05,
      "loss": 0.061,
      "step": 10440
    },
    {
      "epoch": 9.95635986328125e-06,
      "model_forward_time": 5.596055746078491,
      "step": 10440
    },
    {
      "dynamics_loss_avg": 0.0559011435136199,
      "epoch": 9.95635986328125e-06,
      "step": 10440
    },
    {
      "action_loss_avg": 0.007930788025259972,
      "epoch": 9.95635986328125e-06,
      "step": 10440
    },
    {
      "epoch": 9.95635986328125e-06,
      "step": 10440,
      "training_step_time": 17.058403253555298
    },
    {
      "epoch": 9.957313537597657e-06,
      "model_forward_time": 5.605664968490601,
      "step": 10441
    },
    {
      "epoch": 9.957313537597657e-06,
      "step": 10441,
      "training_step_time": 17.081294536590576
    },
    {
      "epoch": 9.958267211914063e-06,
      "model_forward_time": 5.591914415359497,
      "step": 10442
    },
    {
      "epoch": 9.958267211914063e-06,
      "step": 10442,
      "training_step_time": 17.053940773010254
    },
    {
      "epoch": 9.959220886230468e-06,
      "model_forward_time": 5.594619989395142,
      "step": 10443
    },
    {
      "epoch": 9.959220886230468e-06,
      "step": 10443,
      "training_step_time": 17.029082775115967
    },
    {
      "epoch": 9.960174560546876e-06,
      "model_forward_time": 5.589961290359497,
      "step": 10444
    },
    {
      "epoch": 9.960174560546876e-06,
      "step": 10444,
      "training_step_time": 17.042500495910645
    },
    {
      "epoch": 9.961128234863282e-06,
      "model_forward_time": 5.980977535247803,
      "step": 10445
    },
    {
      "epoch": 9.961128234863282e-06,
      "step": 10445,
      "training_step_time": 17.43818187713623
    },
    {
      "epoch": 9.962081909179687e-06,
      "model_forward_time": 5.606526136398315,
      "step": 10446
    },
    {
      "epoch": 9.962081909179687e-06,
      "step": 10446,
      "training_step_time": 17.102142810821533
    },
    {
      "epoch": 9.963035583496093e-06,
      "model_forward_time": 5.623676061630249,
      "step": 10447
    },
    {
      "epoch": 9.963035583496093e-06,
      "step": 10447,
      "training_step_time": 17.135056018829346
    },
    {
      "epoch": 9.9639892578125e-06,
      "model_forward_time": 5.626377820968628,
      "step": 10448
    },
    {
      "epoch": 9.9639892578125e-06,
      "step": 10448,
      "training_step_time": 17.12179398536682
    },
    {
      "epoch": 9.964942932128906e-06,
      "model_forward_time": 5.60874605178833,
      "step": 10449
    },
    {
      "epoch": 9.964942932128906e-06,
      "step": 10449,
      "training_step_time": 17.05726194381714
    },
    {
      "epoch": 9.965896606445312e-06,
      "grad_norm": 0.5162062048912048,
      "learning_rate": 1e-05,
      "loss": 0.0583,
      "step": 10450
    },
    {
      "epoch": 9.965896606445312e-06,
      "model_forward_time": 5.936077356338501,
      "step": 10450
    },
    {
      "dynamics_loss_avg": 0.05099106300622225,
      "epoch": 9.965896606445312e-06,
      "step": 10450
    },
    {
      "action_loss_avg": 0.006500378297641873,
      "epoch": 9.965896606445312e-06,
      "step": 10450
    },
    {
      "epoch": 9.965896606445312e-06,
      "step": 10450,
      "training_step_time": 17.358143091201782
    },
    {
      "epoch": 9.96685028076172e-06,
      "model_forward_time": 5.584859371185303,
      "step": 10451
    },
    {
      "epoch": 9.96685028076172e-06,
      "step": 10451,
      "training_step_time": 17.024585723876953
    },
    {
      "epoch": 9.967803955078125e-06,
      "model_forward_time": 5.593169689178467,
      "step": 10452
    },
    {
      "epoch": 9.967803955078125e-06,
      "step": 10452,
      "training_step_time": 17.027490615844727
    },
    {
      "epoch": 9.96875762939453e-06,
      "model_forward_time": 5.589725494384766,
      "step": 10453
    },
    {
      "epoch": 9.96875762939453e-06,
      "step": 10453,
      "training_step_time": 17.031571865081787
    },
    {
      "epoch": 9.969711303710938e-06,
      "model_forward_time": 5.596360206604004,
      "step": 10454
    },
    {
      "epoch": 9.969711303710938e-06,
      "step": 10454,
      "training_step_time": 17.090889930725098
    },
    {
      "epoch": 9.970664978027344e-06,
      "model_forward_time": 5.628540515899658,
      "step": 10455
    },
    {
      "epoch": 9.970664978027344e-06,
      "step": 10455,
      "training_step_time": 17.52544140815735
    },
    {
      "epoch": 9.97161865234375e-06,
      "model_forward_time": 5.637300252914429,
      "step": 10456
    },
    {
      "epoch": 9.97161865234375e-06,
      "step": 10456,
      "training_step_time": 17.14567255973816
    },
    {
      "epoch": 9.972572326660157e-06,
      "model_forward_time": 5.616008043289185,
      "step": 10457
    },
    {
      "epoch": 9.972572326660157e-06,
      "step": 10457,
      "training_step_time": 17.06660747528076
    },
    {
      "epoch": 9.973526000976563e-06,
      "model_forward_time": 5.5937933921813965,
      "step": 10458
    },
    {
      "epoch": 9.973526000976563e-06,
      "step": 10458,
      "training_step_time": 17.00797462463379
    },
    {
      "epoch": 9.974479675292968e-06,
      "model_forward_time": 5.564644813537598,
      "step": 10459
    },
    {
      "epoch": 9.974479675292968e-06,
      "step": 10459,
      "training_step_time": 16.974592447280884
    },
    {
      "epoch": 9.975433349609376e-06,
      "grad_norm": 0.7189124226570129,
      "learning_rate": 1e-05,
      "loss": 0.0656,
      "step": 10460
    },
    {
      "epoch": 9.975433349609376e-06,
      "model_forward_time": 5.556207895278931,
      "step": 10460
    },
    {
      "dynamics_loss_avg": 0.06639494434930385,
      "epoch": 9.975433349609376e-06,
      "step": 10460
    },
    {
      "action_loss_avg": 0.008373192365979776,
      "epoch": 9.975433349609376e-06,
      "step": 10460
    },
    {
      "epoch": 9.975433349609376e-06,
      "step": 10460,
      "training_step_time": 16.979766130447388
    },
    {
      "epoch": 9.976387023925782e-06,
      "model_forward_time": 5.93939733505249,
      "step": 10461
    },
    {
      "epoch": 9.976387023925782e-06,
      "step": 10461,
      "training_step_time": 17.35973858833313
    },
    {
      "epoch": 9.977340698242187e-06,
      "model_forward_time": 5.566223621368408,
      "step": 10462
    },
    {
      "epoch": 9.977340698242187e-06,
      "step": 10462,
      "training_step_time": 16.993247509002686
    },
    {
      "epoch": 9.978294372558593e-06,
      "model_forward_time": 5.582143545150757,
      "step": 10463
    },
    {
      "epoch": 9.978294372558593e-06,
      "step": 10463,
      "training_step_time": 16.996005535125732
    },
    {
      "epoch": 9.979248046875e-06,
      "model_forward_time": 5.622395753860474,
      "step": 10464
    },
    {
      "epoch": 9.979248046875e-06,
      "step": 10464,
      "training_step_time": 17.059755086898804
    },
    {
      "epoch": 9.980201721191406e-06,
      "model_forward_time": 5.617809057235718,
      "step": 10465
    },
    {
      "epoch": 9.980201721191406e-06,
      "step": 10465,
      "training_step_time": 17.087694883346558
    },
    {
      "epoch": 9.981155395507812e-06,
      "model_forward_time": 5.642596960067749,
      "step": 10466
    },
    {
      "epoch": 9.981155395507812e-06,
      "step": 10466,
      "training_step_time": 17.49559497833252
    },
    {
      "epoch": 9.98210906982422e-06,
      "model_forward_time": 5.62164306640625,
      "step": 10467
    },
    {
      "epoch": 9.98210906982422e-06,
      "step": 10467,
      "training_step_time": 17.12651252746582
    },
    {
      "epoch": 9.983062744140625e-06,
      "model_forward_time": 5.613765001296997,
      "step": 10468
    },
    {
      "epoch": 9.983062744140625e-06,
      "step": 10468,
      "training_step_time": 17.116672039031982
    },
    {
      "epoch": 9.98401641845703e-06,
      "model_forward_time": 5.632174253463745,
      "step": 10469
    },
    {
      "epoch": 9.98401641845703e-06,
      "step": 10469,
      "training_step_time": 17.133371591567993
    },
    {
      "epoch": 9.984970092773438e-06,
      "grad_norm": 0.6460592150688171,
      "learning_rate": 1e-05,
      "loss": 0.0725,
      "step": 10470
    },
    {
      "epoch": 9.984970092773438e-06,
      "model_forward_time": 5.612727880477905,
      "step": 10470
    },
    {
      "dynamics_loss_avg": 0.054444731771945955,
      "epoch": 9.984970092773438e-06,
      "step": 10470
    },
    {
      "action_loss_avg": 0.007951727486215532,
      "epoch": 9.984970092773438e-06,
      "step": 10470
    },
    {
      "epoch": 9.984970092773438e-06,
      "step": 10470,
      "training_step_time": 17.092885971069336
    },
    {
      "epoch": 9.985923767089844e-06,
      "model_forward_time": 5.609086275100708,
      "step": 10471
    },
    {
      "epoch": 9.985923767089844e-06,
      "step": 10471,
      "training_step_time": 17.082761526107788
    },
    {
      "epoch": 9.98687744140625e-06,
      "model_forward_time": 5.942978858947754,
      "step": 10472
    },
    {
      "epoch": 9.98687744140625e-06,
      "step": 10472,
      "training_step_time": 17.413832664489746
    },
    {
      "epoch": 9.987831115722657e-06,
      "model_forward_time": 5.612222909927368,
      "step": 10473
    },
    {
      "epoch": 9.987831115722657e-06,
      "step": 10473,
      "training_step_time": 17.111984729766846
    },
    {
      "epoch": 9.988784790039063e-06,
      "model_forward_time": 5.634942293167114,
      "step": 10474
    },
    {
      "epoch": 9.988784790039063e-06,
      "step": 10474,
      "training_step_time": 17.16358208656311
    },
    {
      "epoch": 9.989738464355468e-06,
      "model_forward_time": 5.631124019622803,
      "step": 10475
    },
    {
      "epoch": 9.989738464355468e-06,
      "step": 10475,
      "training_step_time": 17.142247200012207
    },
    {
      "epoch": 9.990692138671876e-06,
      "model_forward_time": 5.598088979721069,
      "step": 10476
    },
    {
      "epoch": 9.990692138671876e-06,
      "step": 10476,
      "training_step_time": 17.047330379486084
    },
    {
      "epoch": 9.991645812988282e-06,
      "model_forward_time": 6.112205743789673,
      "step": 10477
    },
    {
      "epoch": 9.991645812988282e-06,
      "step": 10477,
      "training_step_time": 17.564213514328003
    },
    {
      "epoch": 9.992599487304687e-06,
      "model_forward_time": 5.618038177490234,
      "step": 10478
    },
    {
      "epoch": 9.992599487304687e-06,
      "step": 10478,
      "training_step_time": 17.036513805389404
    },
    {
      "epoch": 9.993553161621093e-06,
      "model_forward_time": 5.596006870269775,
      "step": 10479
    },
    {
      "epoch": 9.993553161621093e-06,
      "step": 10479,
      "training_step_time": 17.047335386276245
    },
    {
      "epoch": 9.9945068359375e-06,
      "grad_norm": 1.0523288249969482,
      "learning_rate": 1e-05,
      "loss": 0.0679,
      "step": 10480
    },
    {
      "epoch": 9.9945068359375e-06,
      "model_forward_time": 5.604609489440918,
      "step": 10480
    },
    {
      "dynamics_loss_avg": 0.06373933311551809,
      "epoch": 9.9945068359375e-06,
      "step": 10480
    },
    {
      "action_loss_avg": 0.007903120701666922,
      "epoch": 9.9945068359375e-06,
      "step": 10480
    },
    {
      "epoch": 9.9945068359375e-06,
      "step": 10480,
      "training_step_time": 17.064318656921387
    },
    {
      "epoch": 9.995460510253906e-06,
      "model_forward_time": 5.604001522064209,
      "step": 10481
    },
    {
      "epoch": 9.995460510253906e-06,
      "step": 10481,
      "training_step_time": 17.077802896499634
    },
    {
      "epoch": 9.996414184570312e-06,
      "model_forward_time": 5.614043235778809,
      "step": 10482
    },
    {
      "epoch": 9.996414184570312e-06,
      "step": 10482,
      "training_step_time": 17.096805572509766
    },
    {
      "epoch": 9.99736785888672e-06,
      "model_forward_time": 6.030508995056152,
      "step": 10483
    },
    {
      "epoch": 9.99736785888672e-06,
      "step": 10483,
      "training_step_time": 17.51540732383728
    },
    {
      "epoch": 9.998321533203125e-06,
      "model_forward_time": 5.614156723022461,
      "step": 10484
    },
    {
      "epoch": 9.998321533203125e-06,
      "step": 10484,
      "training_step_time": 17.12179946899414
    },
    {
      "epoch": 9.99927520751953e-06,
      "model_forward_time": 5.623634338378906,
      "step": 10485
    },
    {
      "epoch": 9.99927520751953e-06,
      "step": 10485,
      "training_step_time": 17.12153196334839
    },
    {
      "epoch": 1.0000228881835938e-05,
      "model_forward_time": 5.601849555969238,
      "step": 10486
    },
    {
      "epoch": 1.0000228881835938e-05,
      "step": 10486,
      "training_step_time": 17.091543912887573
    },
    {
      "epoch": 1.0001182556152344e-05,
      "model_forward_time": 5.605065584182739,
      "step": 10487
    },
    {
      "epoch": 1.0001182556152344e-05,
      "step": 10487,
      "training_step_time": 17.053731203079224
    },
    {
      "epoch": 1.000213623046875e-05,
      "model_forward_time": 5.9844160079956055,
      "step": 10488
    },
    {
      "epoch": 1.000213623046875e-05,
      "step": 10488,
      "training_step_time": 17.461829662322998
    },
    {
      "epoch": 1.0003089904785157e-05,
      "model_forward_time": 5.620526552200317,
      "step": 10489
    },
    {
      "epoch": 1.0003089904785157e-05,
      "step": 10489,
      "training_step_time": 17.090131282806396
    },
    {
      "epoch": 1.0004043579101563e-05,
      "grad_norm": 1.2897425889968872,
      "learning_rate": 1e-05,
      "loss": 0.0697,
      "step": 10490
    },
    {
      "epoch": 1.0004043579101563e-05,
      "model_forward_time": 5.598847389221191,
      "step": 10490
    },
    {
      "dynamics_loss_avg": 0.062423847150057554,
      "epoch": 1.0004043579101563e-05,
      "step": 10490
    },
    {
      "action_loss_avg": 0.011610749055398628,
      "epoch": 1.0004043579101563e-05,
      "step": 10490
    },
    {
      "epoch": 1.0004043579101563e-05,
      "step": 10490,
      "training_step_time": 17.08158040046692
    },
    {
      "epoch": 1.0004997253417968e-05,
      "model_forward_time": 5.603156089782715,
      "step": 10491
    },
    {
      "epoch": 1.0004997253417968e-05,
      "step": 10491,
      "training_step_time": 17.082026958465576
    },
    {
      "epoch": 1.0005950927734376e-05,
      "model_forward_time": 5.6063525676727295,
      "step": 10492
    },
    {
      "epoch": 1.0005950927734376e-05,
      "step": 10492,
      "training_step_time": 17.084938526153564
    },
    {
      "epoch": 1.0006904602050782e-05,
      "model_forward_time": 5.6137425899505615,
      "step": 10493
    },
    {
      "epoch": 1.0006904602050782e-05,
      "step": 10493,
      "training_step_time": 17.079589366912842
    },
    {
      "epoch": 1.0007858276367187e-05,
      "model_forward_time": 5.948075532913208,
      "step": 10494
    },
    {
      "epoch": 1.0007858276367187e-05,
      "step": 10494,
      "training_step_time": 17.43309259414673
    },
    {
      "epoch": 1.0008811950683593e-05,
      "model_forward_time": 5.600899696350098,
      "step": 10495
    },
    {
      "epoch": 1.0008811950683593e-05,
      "step": 10495,
      "training_step_time": 17.071476697921753
    },
    {
      "epoch": 1.0009765625e-05,
      "model_forward_time": 5.603562116622925,
      "step": 10496
    },
    {
      "epoch": 1.0009765625e-05,
      "step": 10496,
      "training_step_time": 17.065455675125122
    },
    {
      "epoch": 1.0010719299316406e-05,
      "model_forward_time": 5.5928637981414795,
      "step": 10497
    },
    {
      "epoch": 1.0010719299316406e-05,
      "step": 10497,
      "training_step_time": 17.044503927230835
    },
    {
      "epoch": 1.0011672973632812e-05,
      "model_forward_time": 5.591677665710449,
      "step": 10498
    },
    {
      "epoch": 1.0011672973632812e-05,
      "step": 10498,
      "training_step_time": 17.08190107345581
    },
    {
      "epoch": 1.001262664794922e-05,
      "model_forward_time": 5.987969875335693,
      "step": 10499
    },
    {
      "epoch": 1.001262664794922e-05,
      "step": 10499,
      "training_step_time": 17.49456262588501
    },
    {
      "epoch": 1.0013580322265625e-05,
      "grad_norm": 0.6151379346847534,
      "learning_rate": 1e-05,
      "loss": 0.0633,
      "step": 10500
    },
    {
      "epoch": 1.0013580322265625e-05,
      "model_forward_time": 5.636342287063599,
      "step": 10500
    },
    {
      "dynamics_loss_avg": 0.054354666545987126,
      "epoch": 1.0013580322265625e-05,
      "step": 10500
    },
    {
      "action_loss_avg": 0.0052453703479841355,
      "epoch": 1.0013580322265625e-05,
      "step": 10500
    },
    {
      "epoch": 1.0013580322265625e-05,
      "step": 10500,
      "training_step_time": 17.151506185531616
    },
    {
      "epoch": 1.001453399658203e-05,
      "model_forward_time": 5.62346076965332,
      "step": 10501
    },
    {
      "epoch": 1.001453399658203e-05,
      "step": 10501,
      "training_step_time": 17.12744379043579
    },
    {
      "epoch": 1.0015487670898438e-05,
      "model_forward_time": 5.605212688446045,
      "step": 10502
    },
    {
      "epoch": 1.0015487670898438e-05,
      "step": 10502,
      "training_step_time": 17.059212923049927
    },
    {
      "epoch": 1.0016441345214844e-05,
      "model_forward_time": 5.579739093780518,
      "step": 10503
    },
    {
      "epoch": 1.0016441345214844e-05,
      "step": 10503,
      "training_step_time": 16.998945951461792
    },
    {
      "epoch": 1.001739501953125e-05,
      "model_forward_time": 5.945800542831421,
      "step": 10504
    },
    {
      "epoch": 1.001739501953125e-05,
      "step": 10504,
      "training_step_time": 17.397823572158813
    },
    {
      "epoch": 1.0018348693847657e-05,
      "model_forward_time": 5.601484537124634,
      "step": 10505
    },
    {
      "epoch": 1.0018348693847657e-05,
      "step": 10505,
      "training_step_time": 17.09811234474182
    },
    {
      "epoch": 1.0019302368164063e-05,
      "model_forward_time": 5.62565016746521,
      "step": 10506
    },
    {
      "epoch": 1.0019302368164063e-05,
      "step": 10506,
      "training_step_time": 17.12563419342041
    },
    {
      "epoch": 1.0020256042480468e-05,
      "model_forward_time": 5.6115288734436035,
      "step": 10507
    },
    {
      "epoch": 1.0020256042480468e-05,
      "step": 10507,
      "training_step_time": 17.089600563049316
    },
    {
      "epoch": 1.0021209716796874e-05,
      "model_forward_time": 5.5788445472717285,
      "step": 10508
    },
    {
      "epoch": 1.0021209716796874e-05,
      "step": 10508,
      "training_step_time": 16.999980449676514
    },
    {
      "epoch": 1.0022163391113282e-05,
      "model_forward_time": 5.575751066207886,
      "step": 10509
    },
    {
      "epoch": 1.0022163391113282e-05,
      "step": 10509,
      "training_step_time": 17.016592025756836
    },
    {
      "epoch": 1.0023117065429687e-05,
      "grad_norm": 0.6112471222877502,
      "learning_rate": 1e-05,
      "loss": 0.073,
      "step": 10510
    },
    {
      "epoch": 1.0023117065429687e-05,
      "model_forward_time": 5.944937705993652,
      "step": 10510
    },
    {
      "dynamics_loss_avg": 0.06078009307384491,
      "epoch": 1.0023117065429687e-05,
      "step": 10510
    },
    {
      "action_loss_avg": 0.010003728745505213,
      "epoch": 1.0023117065429687e-05,
      "step": 10510
    },
    {
      "epoch": 1.0023117065429687e-05,
      "step": 10510,
      "training_step_time": 17.381227493286133
    },
    {
      "epoch": 1.0024070739746093e-05,
      "model_forward_time": 5.591063022613525,
      "step": 10511
    },
    {
      "epoch": 1.0024070739746093e-05,
      "step": 10511,
      "training_step_time": 17.009596824645996
    },
    {
      "epoch": 1.00250244140625e-05,
      "model_forward_time": 5.583845853805542,
      "step": 10512
    },
    {
      "epoch": 1.00250244140625e-05,
      "step": 10512,
      "training_step_time": 17.034067153930664
    },
    {
      "epoch": 1.0025978088378906e-05,
      "model_forward_time": 5.597628116607666,
      "step": 10513
    },
    {
      "epoch": 1.0025978088378906e-05,
      "step": 10513,
      "training_step_time": 17.05281090736389
    },
    {
      "epoch": 1.0026931762695312e-05,
      "model_forward_time": 5.601500034332275,
      "step": 10514
    },
    {
      "epoch": 1.0026931762695312e-05,
      "step": 10514,
      "training_step_time": 17.083935022354126
    },
    {
      "epoch": 1.002788543701172e-05,
      "model_forward_time": 5.9566333293914795,
      "step": 10515
    },
    {
      "epoch": 1.002788543701172e-05,
      "step": 10515,
      "training_step_time": 17.428346157073975
    },
    {
      "epoch": 1.0028839111328125e-05,
      "model_forward_time": 5.64534068107605,
      "step": 10516
    },
    {
      "epoch": 1.0028839111328125e-05,
      "step": 10516,
      "training_step_time": 17.15597152709961
    },
    {
      "epoch": 1.002979278564453e-05,
      "model_forward_time": 5.611641883850098,
      "step": 10517
    },
    {
      "epoch": 1.002979278564453e-05,
      "step": 10517,
      "training_step_time": 17.088555574417114
    },
    {
      "epoch": 1.0030746459960938e-05,
      "model_forward_time": 5.616703033447266,
      "step": 10518
    },
    {
      "epoch": 1.0030746459960938e-05,
      "step": 10518,
      "training_step_time": 17.107622146606445
    },
    {
      "epoch": 1.0031700134277344e-05,
      "model_forward_time": 5.622269630432129,
      "step": 10519
    },
    {
      "epoch": 1.0031700134277344e-05,
      "step": 10519,
      "training_step_time": 17.114610195159912
    },
    {
      "epoch": 1.003265380859375e-05,
      "grad_norm": 0.5956209897994995,
      "learning_rate": 1e-05,
      "loss": 0.0692,
      "step": 10520
    },
    {
      "epoch": 1.003265380859375e-05,
      "model_forward_time": 5.6076366901397705,
      "step": 10520
    },
    {
      "dynamics_loss_avg": 0.06335753723978996,
      "epoch": 1.003265380859375e-05,
      "step": 10520
    },
    {
      "action_loss_avg": 0.00863841746468097,
      "epoch": 1.003265380859375e-05,
      "step": 10520
    },
    {
      "epoch": 1.003265380859375e-05,
      "step": 10520,
      "training_step_time": 17.44137740135193
    },
    {
      "epoch": 1.0033607482910157e-05,
      "model_forward_time": 5.594178676605225,
      "step": 10521
    },
    {
      "epoch": 1.0033607482910157e-05,
      "step": 10521,
      "training_step_time": 17.035555362701416
    },
    {
      "epoch": 1.0034561157226563e-05,
      "model_forward_time": 5.593942165374756,
      "step": 10522
    },
    {
      "epoch": 1.0034561157226563e-05,
      "step": 10522,
      "training_step_time": 17.06196427345276
    },
    {
      "epoch": 1.0035514831542968e-05,
      "model_forward_time": 5.601741552352905,
      "step": 10523
    },
    {
      "epoch": 1.0035514831542968e-05,
      "step": 10523,
      "training_step_time": 17.075992345809937
    },
    {
      "epoch": 1.0036468505859374e-05,
      "model_forward_time": 5.6070921421051025,
      "step": 10524
    },
    {
      "epoch": 1.0036468505859374e-05,
      "step": 10524,
      "training_step_time": 17.10572910308838
    },
    {
      "epoch": 1.0037422180175782e-05,
      "model_forward_time": 5.615238904953003,
      "step": 10525
    },
    {
      "epoch": 1.0037422180175782e-05,
      "step": 10525,
      "training_step_time": 17.11758780479431
    },
    {
      "epoch": 1.0038375854492187e-05,
      "model_forward_time": 5.982219219207764,
      "step": 10526
    },
    {
      "epoch": 1.0038375854492187e-05,
      "step": 10526,
      "training_step_time": 17.486604690551758
    },
    {
      "epoch": 1.0039329528808593e-05,
      "model_forward_time": 5.612933874130249,
      "step": 10527
    },
    {
      "epoch": 1.0039329528808593e-05,
      "step": 10527,
      "training_step_time": 17.0962917804718
    },
    {
      "epoch": 1.0040283203125e-05,
      "model_forward_time": 5.607091903686523,
      "step": 10528
    },
    {
      "epoch": 1.0040283203125e-05,
      "step": 10528,
      "training_step_time": 17.088035106658936
    },
    {
      "epoch": 1.0041236877441406e-05,
      "model_forward_time": 5.607830286026001,
      "step": 10529
    },
    {
      "epoch": 1.0041236877441406e-05,
      "step": 10529,
      "training_step_time": 17.096572875976562
    },
    {
      "epoch": 1.0042190551757812e-05,
      "grad_norm": 0.4699918031692505,
      "learning_rate": 1e-05,
      "loss": 0.0587,
      "step": 10530
    },
    {
      "epoch": 1.0042190551757812e-05,
      "model_forward_time": 5.591864347457886,
      "step": 10530
    },
    {
      "dynamics_loss_avg": 0.04668958280235529,
      "epoch": 1.0042190551757812e-05,
      "step": 10530
    },
    {
      "action_loss_avg": 0.006885754014365375,
      "epoch": 1.0042190551757812e-05,
      "step": 10530
    },
    {
      "epoch": 1.0042190551757812e-05,
      "step": 10530,
      "training_step_time": 17.03948712348938
    },
    {
      "epoch": 1.004314422607422e-05,
      "model_forward_time": 5.996719121932983,
      "step": 10531
    },
    {
      "epoch": 1.004314422607422e-05,
      "step": 10531,
      "training_step_time": 17.400652170181274
    },
    {
      "epoch": 1.0044097900390625e-05,
      "model_forward_time": 5.590116739273071,
      "step": 10532
    },
    {
      "epoch": 1.0044097900390625e-05,
      "step": 10532,
      "training_step_time": 17.042260885238647
    },
    {
      "epoch": 1.004505157470703e-05,
      "model_forward_time": 5.590152025222778,
      "step": 10533
    },
    {
      "epoch": 1.004505157470703e-05,
      "step": 10533,
      "training_step_time": 17.05365228652954
    },
    {
      "epoch": 1.0046005249023438e-05,
      "model_forward_time": 5.611787796020508,
      "step": 10534
    },
    {
      "epoch": 1.0046005249023438e-05,
      "step": 10534,
      "training_step_time": 17.09531331062317
    },
    {
      "epoch": 1.0046958923339844e-05,
      "model_forward_time": 5.6113057136535645,
      "step": 10535
    },
    {
      "epoch": 1.0046958923339844e-05,
      "step": 10535,
      "training_step_time": 17.111764192581177
    },
    {
      "epoch": 1.004791259765625e-05,
      "model_forward_time": 5.627729177474976,
      "step": 10536
    },
    {
      "epoch": 1.004791259765625e-05,
      "step": 10536,
      "training_step_time": 17.087028980255127
    },
    {
      "epoch": 1.0048866271972657e-05,
      "model_forward_time": 5.93556022644043,
      "step": 10537
    },
    {
      "epoch": 1.0048866271972657e-05,
      "step": 10537,
      "training_step_time": 17.370319843292236
    },
    {
      "epoch": 1.0049819946289063e-05,
      "model_forward_time": 5.5800158977508545,
      "step": 10538
    },
    {
      "epoch": 1.0049819946289063e-05,
      "step": 10538,
      "training_step_time": 17.009552478790283
    },
    {
      "epoch": 1.0050773620605468e-05,
      "model_forward_time": 5.594870328903198,
      "step": 10539
    },
    {
      "epoch": 1.0050773620605468e-05,
      "step": 10539,
      "training_step_time": 17.075382709503174
    },
    {
      "epoch": 1.0051727294921874e-05,
      "grad_norm": 0.695402979850769,
      "learning_rate": 1e-05,
      "loss": 0.0643,
      "step": 10540
    },
    {
      "epoch": 1.0051727294921874e-05,
      "model_forward_time": 5.622154235839844,
      "step": 10540
    },
    {
      "dynamics_loss_avg": 0.059998422116041186,
      "epoch": 1.0051727294921874e-05,
      "step": 10540
    },
    {
      "action_loss_avg": 0.010528599517419935,
      "epoch": 1.0051727294921874e-05,
      "step": 10540
    },
    {
      "epoch": 1.0051727294921874e-05,
      "step": 10540,
      "training_step_time": 17.140124082565308
    },
    {
      "epoch": 1.0052680969238282e-05,
      "model_forward_time": 5.64669132232666,
      "step": 10541
    },
    {
      "epoch": 1.0052680969238282e-05,
      "step": 10541,
      "training_step_time": 17.155370712280273
    },
    {
      "epoch": 1.0053634643554687e-05,
      "model_forward_time": 5.977470636367798,
      "step": 10542
    },
    {
      "epoch": 1.0053634643554687e-05,
      "step": 10542,
      "training_step_time": 17.435198307037354
    },
    {
      "epoch": 1.0054588317871093e-05,
      "model_forward_time": 5.623986721038818,
      "step": 10543
    },
    {
      "epoch": 1.0054588317871093e-05,
      "step": 10543,
      "training_step_time": 17.099984645843506
    },
    {
      "epoch": 1.00555419921875e-05,
      "model_forward_time": 5.575425386428833,
      "step": 10544
    },
    {
      "epoch": 1.00555419921875e-05,
      "step": 10544,
      "training_step_time": 17.023932695388794
    },
    {
      "epoch": 1.0056495666503906e-05,
      "model_forward_time": 5.587778806686401,
      "step": 10545
    },
    {
      "epoch": 1.0056495666503906e-05,
      "step": 10545,
      "training_step_time": 17.01620078086853
    },
    {
      "epoch": 1.0057449340820312e-05,
      "model_forward_time": 5.591934442520142,
      "step": 10546
    },
    {
      "epoch": 1.0057449340820312e-05,
      "step": 10546,
      "training_step_time": 17.0204861164093
    },
    {
      "epoch": 1.005840301513672e-05,
      "model_forward_time": 5.586572885513306,
      "step": 10547
    },
    {
      "epoch": 1.005840301513672e-05,
      "step": 10547,
      "training_step_time": 17.031296491622925
    },
    {
      "epoch": 1.0059356689453125e-05,
      "model_forward_time": 5.941708087921143,
      "step": 10548
    },
    {
      "epoch": 1.0059356689453125e-05,
      "step": 10548,
      "training_step_time": 17.404869556427002
    },
    {
      "epoch": 1.006031036376953e-05,
      "model_forward_time": 5.603828191757202,
      "step": 10549
    },
    {
      "epoch": 1.006031036376953e-05,
      "step": 10549,
      "training_step_time": 17.08322811126709
    },
    {
      "epoch": 1.0061264038085938e-05,
      "grad_norm": 0.3242855966091156,
      "learning_rate": 1e-05,
      "loss": 0.0776,
      "step": 10550
    },
    {
      "epoch": 1.0061264038085938e-05,
      "model_forward_time": 5.629722356796265,
      "step": 10550
    },
    {
      "dynamics_loss_avg": 0.06689646765589714,
      "epoch": 1.0061264038085938e-05,
      "step": 10550
    },
    {
      "action_loss_avg": 0.011449761223047972,
      "epoch": 1.0061264038085938e-05,
      "step": 10550
    },
    {
      "epoch": 1.0061264038085938e-05,
      "step": 10550,
      "training_step_time": 17.15960144996643
    },
    {
      "epoch": 1.0062217712402344e-05,
      "model_forward_time": 5.653737545013428,
      "step": 10551
    },
    {
      "epoch": 1.0062217712402344e-05,
      "step": 10551,
      "training_step_time": 17.12008023262024
    },
    {
      "epoch": 1.006317138671875e-05,
      "model_forward_time": 5.6162192821502686,
      "step": 10552
    },
    {
      "epoch": 1.006317138671875e-05,
      "step": 10552,
      "training_step_time": 17.095107793807983
    },
    {
      "epoch": 1.0064125061035157e-05,
      "model_forward_time": 6.1361541748046875,
      "step": 10553
    },
    {
      "epoch": 1.0064125061035157e-05,
      "step": 10553,
      "training_step_time": 17.620705366134644
    },
    {
      "epoch": 1.0065078735351563e-05,
      "model_forward_time": 5.609878301620483,
      "step": 10554
    },
    {
      "epoch": 1.0065078735351563e-05,
      "step": 10554,
      "training_step_time": 17.091515064239502
    },
    {
      "epoch": 1.0066032409667968e-05,
      "model_forward_time": 5.614327669143677,
      "step": 10555
    },
    {
      "epoch": 1.0066032409667968e-05,
      "step": 10555,
      "training_step_time": 17.094910860061646
    },
    {
      "epoch": 1.0066986083984374e-05,
      "model_forward_time": 5.600642919540405,
      "step": 10556
    },
    {
      "epoch": 1.0066986083984374e-05,
      "step": 10556,
      "training_step_time": 17.06322431564331
    },
    {
      "epoch": 1.0067939758300782e-05,
      "model_forward_time": 5.6299803256988525,
      "step": 10557
    },
    {
      "epoch": 1.0067939758300782e-05,
      "step": 10557,
      "training_step_time": 17.12095856666565
    },
    {
      "epoch": 1.0068893432617187e-05,
      "model_forward_time": 5.621783494949341,
      "step": 10558
    },
    {
      "epoch": 1.0068893432617187e-05,
      "step": 10558,
      "training_step_time": 17.11780595779419
    },
    {
      "epoch": 1.0069847106933593e-05,
      "model_forward_time": 5.983396291732788,
      "step": 10559
    },
    {
      "epoch": 1.0069847106933593e-05,
      "step": 10559,
      "training_step_time": 17.484609365463257
    },
    {
      "epoch": 1.007080078125e-05,
      "grad_norm": 1.0733722448349,
      "learning_rate": 1e-05,
      "loss": 0.0786,
      "step": 10560
    },
    {
      "epoch": 1.007080078125e-05,
      "model_forward_time": 5.6056153774261475,
      "step": 10560
    },
    {
      "dynamics_loss_avg": 0.06293667480349541,
      "epoch": 1.007080078125e-05,
      "step": 10560
    },
    {
      "action_loss_avg": 0.009472799045033753,
      "epoch": 1.007080078125e-05,
      "step": 10560
    },
    {
      "epoch": 1.007080078125e-05,
      "step": 10560,
      "training_step_time": 17.10668683052063
    },
    {
      "epoch": 1.0071754455566406e-05,
      "model_forward_time": 5.635226726531982,
      "step": 10561
    },
    {
      "epoch": 1.0071754455566406e-05,
      "step": 10561,
      "training_step_time": 17.133569478988647
    },
    {
      "epoch": 1.0072708129882812e-05,
      "model_forward_time": 5.62645697593689,
      "step": 10562
    },
    {
      "epoch": 1.0072708129882812e-05,
      "step": 10562,
      "training_step_time": 17.127298831939697
    },
    {
      "epoch": 1.007366180419922e-05,
      "model_forward_time": 5.614952325820923,
      "step": 10563
    },
    {
      "epoch": 1.007366180419922e-05,
      "step": 10563,
      "training_step_time": 17.09475588798523
    },
    {
      "epoch": 1.0074615478515625e-05,
      "model_forward_time": 6.034287929534912,
      "step": 10564
    },
    {
      "epoch": 1.0074615478515625e-05,
      "step": 10564,
      "training_step_time": 17.4993577003479
    },
    {
      "epoch": 1.007556915283203e-05,
      "model_forward_time": 5.586408615112305,
      "step": 10565
    },
    {
      "epoch": 1.007556915283203e-05,
      "step": 10565,
      "training_step_time": 17.013352155685425
    },
    {
      "epoch": 1.0076522827148438e-05,
      "model_forward_time": 5.587041854858398,
      "step": 10566
    },
    {
      "epoch": 1.0076522827148438e-05,
      "step": 10566,
      "training_step_time": 17.039336442947388
    },
    {
      "epoch": 1.0077476501464844e-05,
      "model_forward_time": 5.604784250259399,
      "step": 10567
    },
    {
      "epoch": 1.0077476501464844e-05,
      "step": 10567,
      "training_step_time": 17.09148335456848
    },
    {
      "epoch": 1.007843017578125e-05,
      "model_forward_time": 5.620731592178345,
      "step": 10568
    },
    {
      "epoch": 1.007843017578125e-05,
      "step": 10568,
      "training_step_time": 17.15149188041687
    },
    {
      "epoch": 1.0079383850097657e-05,
      "model_forward_time": 5.974835634231567,
      "step": 10569
    },
    {
      "epoch": 1.0079383850097657e-05,
      "step": 10569,
      "training_step_time": 17.47246479988098
    },
    {
      "epoch": 1.0080337524414063e-05,
      "grad_norm": 1.6167001724243164,
      "learning_rate": 1e-05,
      "loss": 0.0614,
      "step": 10570
    },
    {
      "epoch": 1.0080337524414063e-05,
      "model_forward_time": 5.6148693561553955,
      "step": 10570
    },
    {
      "dynamics_loss_avg": 0.05672039482742548,
      "epoch": 1.0080337524414063e-05,
      "step": 10570
    },
    {
      "action_loss_avg": 0.008689043193589896,
      "epoch": 1.0080337524414063e-05,
      "step": 10570
    },
    {
      "epoch": 1.0080337524414063e-05,
      "step": 10570,
      "training_step_time": 17.11116313934326
    },
    {
      "epoch": 1.0081291198730468e-05,
      "model_forward_time": 5.610652923583984,
      "step": 10571
    },
    {
      "epoch": 1.0081291198730468e-05,
      "step": 10571,
      "training_step_time": 17.107760906219482
    },
    {
      "epoch": 1.0082244873046874e-05,
      "model_forward_time": 5.606904983520508,
      "step": 10572
    },
    {
      "epoch": 1.0082244873046874e-05,
      "step": 10572,
      "training_step_time": 17.093149662017822
    },
    {
      "epoch": 1.0083198547363282e-05,
      "model_forward_time": 5.6066694259643555,
      "step": 10573
    },
    {
      "epoch": 1.0083198547363282e-05,
      "step": 10573,
      "training_step_time": 17.077391624450684
    },
    {
      "epoch": 1.0084152221679687e-05,
      "model_forward_time": 5.609549283981323,
      "step": 10574
    },
    {
      "epoch": 1.0084152221679687e-05,
      "step": 10574,
      "training_step_time": 17.03789186477661
    },
    {
      "epoch": 1.0085105895996093e-05,
      "model_forward_time": 5.956557512283325,
      "step": 10575
    },
    {
      "epoch": 1.0085105895996093e-05,
      "step": 10575,
      "training_step_time": 17.4208722114563
    },
    {
      "epoch": 1.00860595703125e-05,
      "model_forward_time": 5.628965377807617,
      "step": 10576
    },
    {
      "epoch": 1.00860595703125e-05,
      "step": 10576,
      "training_step_time": 17.136043310165405
    },
    {
      "epoch": 1.0087013244628906e-05,
      "model_forward_time": 5.63592791557312,
      "step": 10577
    },
    {
      "epoch": 1.0087013244628906e-05,
      "step": 10577,
      "training_step_time": 17.152588844299316
    },
    {
      "epoch": 1.0087966918945312e-05,
      "model_forward_time": 5.619201898574829,
      "step": 10578
    },
    {
      "epoch": 1.0087966918945312e-05,
      "step": 10578,
      "training_step_time": 17.102238416671753
    },
    {
      "epoch": 1.008892059326172e-05,
      "model_forward_time": 5.603412389755249,
      "step": 10579
    },
    {
      "epoch": 1.008892059326172e-05,
      "step": 10579,
      "training_step_time": 17.05186176300049
    },
    {
      "epoch": 1.0089874267578125e-05,
      "grad_norm": 0.541877031326294,
      "learning_rate": 1e-05,
      "loss": 0.0567,
      "step": 10580
    },
    {
      "epoch": 1.0089874267578125e-05,
      "model_forward_time": 5.600106477737427,
      "step": 10580
    },
    {
      "dynamics_loss_avg": 0.04923999831080437,
      "epoch": 1.0089874267578125e-05,
      "step": 10580
    },
    {
      "action_loss_avg": 0.006760284933261573,
      "epoch": 1.0089874267578125e-05,
      "step": 10580
    },
    {
      "epoch": 1.0089874267578125e-05,
      "step": 10580,
      "training_step_time": 17.407732486724854
    },
    {
      "epoch": 1.009082794189453e-05,
      "model_forward_time": 5.590759515762329,
      "step": 10581
    },
    {
      "epoch": 1.009082794189453e-05,
      "step": 10581,
      "training_step_time": 17.023595571517944
    },
    {
      "epoch": 1.0091781616210938e-05,
      "model_forward_time": 5.594166278839111,
      "step": 10582
    },
    {
      "epoch": 1.0091781616210938e-05,
      "step": 10582,
      "training_step_time": 17.068858861923218
    },
    {
      "epoch": 1.0092735290527344e-05,
      "model_forward_time": 5.610165596008301,
      "step": 10583
    },
    {
      "epoch": 1.0092735290527344e-05,
      "step": 10583,
      "training_step_time": 17.084575653076172
    },
    {
      "epoch": 1.009368896484375e-05,
      "model_forward_time": 5.60309100151062,
      "step": 10584
    },
    {
      "epoch": 1.009368896484375e-05,
      "step": 10584,
      "training_step_time": 17.083909511566162
    },
    {
      "epoch": 1.0094642639160157e-05,
      "model_forward_time": 5.6129679679870605,
      "step": 10585
    },
    {
      "epoch": 1.0094642639160157e-05,
      "step": 10585,
      "training_step_time": 17.081297159194946
    },
    {
      "epoch": 1.0095596313476563e-05,
      "model_forward_time": 5.9608964920043945,
      "step": 10586
    },
    {
      "epoch": 1.0095596313476563e-05,
      "step": 10586,
      "training_step_time": 17.403194189071655
    },
    {
      "epoch": 1.0096549987792969e-05,
      "model_forward_time": 5.590043067932129,
      "step": 10587
    },
    {
      "epoch": 1.0096549987792969e-05,
      "step": 10587,
      "training_step_time": 17.055906295776367
    },
    {
      "epoch": 1.0097503662109374e-05,
      "model_forward_time": 5.600215673446655,
      "step": 10588
    },
    {
      "epoch": 1.0097503662109374e-05,
      "step": 10588,
      "training_step_time": 17.056814432144165
    },
    {
      "epoch": 1.0098457336425782e-05,
      "model_forward_time": 5.606021404266357,
      "step": 10589
    },
    {
      "epoch": 1.0098457336425782e-05,
      "step": 10589,
      "training_step_time": 17.06393027305603
    },
    {
      "epoch": 1.0099411010742187e-05,
      "grad_norm": 0.9675894975662231,
      "learning_rate": 1e-05,
      "loss": 0.0826,
      "step": 10590
    },
    {
      "epoch": 1.0099411010742187e-05,
      "model_forward_time": 5.606659650802612,
      "step": 10590
    },
    {
      "dynamics_loss_avg": 0.07261306457221509,
      "epoch": 1.0099411010742187e-05,
      "step": 10590
    },
    {
      "action_loss_avg": 0.009350014640949667,
      "epoch": 1.0099411010742187e-05,
      "step": 10590
    },
    {
      "epoch": 1.0099411010742187e-05,
      "step": 10590,
      "training_step_time": 17.09241771697998
    },
    {
      "epoch": 1.0100364685058593e-05,
      "model_forward_time": 5.61203145980835,
      "step": 10591
    },
    {
      "epoch": 1.0100364685058593e-05,
      "step": 10591,
      "training_step_time": 17.4882709980011
    },
    {
      "epoch": 1.0101318359375e-05,
      "model_forward_time": 5.598957061767578,
      "step": 10592
    },
    {
      "epoch": 1.0101318359375e-05,
      "step": 10592,
      "training_step_time": 17.057459592819214
    },
    {
      "epoch": 1.0102272033691406e-05,
      "model_forward_time": 5.59212851524353,
      "step": 10593
    },
    {
      "epoch": 1.0102272033691406e-05,
      "step": 10593,
      "training_step_time": 17.0500271320343
    },
    {
      "epoch": 1.0103225708007812e-05,
      "model_forward_time": 5.577005863189697,
      "step": 10594
    },
    {
      "epoch": 1.0103225708007812e-05,
      "step": 10594,
      "training_step_time": 17.00678586959839
    },
    {
      "epoch": 1.010417938232422e-05,
      "model_forward_time": 5.591870069503784,
      "step": 10595
    },
    {
      "epoch": 1.010417938232422e-05,
      "step": 10595,
      "training_step_time": 17.065207481384277
    },
    {
      "epoch": 1.0105133056640625e-05,
      "model_forward_time": 5.609743118286133,
      "step": 10596
    },
    {
      "epoch": 1.0105133056640625e-05,
      "step": 10596,
      "training_step_time": 17.05840277671814
    },
    {
      "epoch": 1.010608673095703e-05,
      "model_forward_time": 4.6449174880981445,
      "step": 10597
    },
    {
      "epoch": 1.010608673095703e-05,
      "step": 10597,
      "training_step_time": 13.159924983978271
    },
    {
      "epoch": 1.0107040405273438e-05,
      "model_forward_time": 5.606856822967529,
      "step": 10598
    },
    {
      "epoch": 1.0107040405273438e-05,
      "step": 10598,
      "training_step_time": 17.088518619537354
    },
    {
      "epoch": 1.0107994079589844e-05,
      "model_forward_time": 5.624716758728027,
      "step": 10599
    },
    {
      "epoch": 1.0107994079589844e-05,
      "step": 10599,
      "training_step_time": 17.060585498809814
    },
    {
      "epoch": 1.010894775390625e-05,
      "grad_norm": 0.24661025404930115,
      "learning_rate": 1e-05,
      "loss": 0.0587,
      "step": 10600
    },
    {
      "epoch": 1.010894775390625e-05,
      "model_forward_time": 5.594056844711304,
      "step": 10600
    },
    {
      "dynamics_loss_avg": 0.049118523485958576,
      "epoch": 1.010894775390625e-05,
      "step": 10600
    },
    {
      "action_loss_avg": 0.0070652288617566224,
      "epoch": 1.010894775390625e-05,
      "step": 10600
    },
    {
      "epoch": 1.010894775390625e-05,
      "step": 10600,
      "training_step_time": 17.033021926879883
    },
    {
      "epoch": 1.0109901428222657e-05,
      "model_forward_time": 5.580533504486084,
      "step": 10601
    },
    {
      "epoch": 1.0109901428222657e-05,
      "step": 10601,
      "training_step_time": 17.024799585342407
    },
    {
      "epoch": 1.0110855102539063e-05,
      "model_forward_time": 5.578700542449951,
      "step": 10602
    },
    {
      "epoch": 1.0110855102539063e-05,
      "step": 10602,
      "training_step_time": 17.355855703353882
    },
    {
      "epoch": 1.0111808776855469e-05,
      "model_forward_time": 5.575500249862671,
      "step": 10603
    },
    {
      "epoch": 1.0111808776855469e-05,
      "step": 10603,
      "training_step_time": 17.00642466545105
    },
    {
      "epoch": 1.0112762451171874e-05,
      "model_forward_time": 5.597381114959717,
      "step": 10604
    },
    {
      "epoch": 1.0112762451171874e-05,
      "step": 10604,
      "training_step_time": 17.04465341567993
    },
    {
      "epoch": 1.0113716125488282e-05,
      "model_forward_time": 5.60167384147644,
      "step": 10605
    },
    {
      "epoch": 1.0113716125488282e-05,
      "step": 10605,
      "training_step_time": 17.08380389213562
    },
    {
      "epoch": 1.0114669799804687e-05,
      "model_forward_time": 5.618867635726929,
      "step": 10606
    },
    {
      "epoch": 1.0114669799804687e-05,
      "step": 10606,
      "training_step_time": 17.094671487808228
    },
    {
      "epoch": 1.0115623474121093e-05,
      "model_forward_time": 5.597581386566162,
      "step": 10607
    },
    {
      "epoch": 1.0115623474121093e-05,
      "step": 10607,
      "training_step_time": 17.04428195953369
    },
    {
      "epoch": 1.01165771484375e-05,
      "model_forward_time": 5.9714035987854,
      "step": 10608
    },
    {
      "epoch": 1.01165771484375e-05,
      "step": 10608,
      "training_step_time": 17.40781545639038
    },
    {
      "epoch": 1.0117530822753906e-05,
      "model_forward_time": 5.585163593292236,
      "step": 10609
    },
    {
      "epoch": 1.0117530822753906e-05,
      "step": 10609,
      "training_step_time": 17.009729146957397
    },
    {
      "epoch": 1.0118484497070312e-05,
      "grad_norm": 0.7123680114746094,
      "learning_rate": 1e-05,
      "loss": 0.0602,
      "step": 10610
    },
    {
      "epoch": 1.0118484497070312e-05,
      "model_forward_time": 5.584038019180298,
      "step": 10610
    },
    {
      "dynamics_loss_avg": 0.05314143970608711,
      "epoch": 1.0118484497070312e-05,
      "step": 10610
    },
    {
      "action_loss_avg": 0.006191394245252013,
      "epoch": 1.0118484497070312e-05,
      "step": 10610
    },
    {
      "epoch": 1.0118484497070312e-05,
      "step": 10610,
      "training_step_time": 17.023020029067993
    },
    {
      "epoch": 1.011943817138672e-05,
      "model_forward_time": 5.599603652954102,
      "step": 10611
    },
    {
      "epoch": 1.011943817138672e-05,
      "step": 10611,
      "training_step_time": 17.0484459400177
    },
    {
      "epoch": 1.0120391845703125e-05,
      "model_forward_time": 5.597732067108154,
      "step": 10612
    },
    {
      "epoch": 1.0120391845703125e-05,
      "step": 10612,
      "training_step_time": 17.060970783233643
    },
    {
      "epoch": 1.012134552001953e-05,
      "model_forward_time": 5.60526442527771,
      "step": 10613
    },
    {
      "epoch": 1.012134552001953e-05,
      "step": 10613,
      "training_step_time": 17.435973405838013
    },
    {
      "epoch": 1.0122299194335938e-05,
      "model_forward_time": 5.59962797164917,
      "step": 10614
    },
    {
      "epoch": 1.0122299194335938e-05,
      "step": 10614,
      "training_step_time": 17.07632613182068
    },
    {
      "epoch": 1.0123252868652344e-05,
      "model_forward_time": 5.597294569015503,
      "step": 10615
    },
    {
      "epoch": 1.0123252868652344e-05,
      "step": 10615,
      "training_step_time": 17.06713891029358
    },
    {
      "epoch": 1.012420654296875e-05,
      "model_forward_time": 5.597235202789307,
      "step": 10616
    },
    {
      "epoch": 1.012420654296875e-05,
      "step": 10616,
      "training_step_time": 17.05695128440857
    },
    {
      "epoch": 1.0125160217285157e-05,
      "model_forward_time": 5.608785390853882,
      "step": 10617
    },
    {
      "epoch": 1.0125160217285157e-05,
      "step": 10617,
      "training_step_time": 17.10858130455017
    },
    {
      "epoch": 1.0126113891601563e-05,
      "model_forward_time": 5.604885816574097,
      "step": 10618
    },
    {
      "epoch": 1.0126113891601563e-05,
      "step": 10618,
      "training_step_time": 17.098902225494385
    },
    {
      "epoch": 1.0127067565917969e-05,
      "model_forward_time": 6.022555112838745,
      "step": 10619
    },
    {
      "epoch": 1.0127067565917969e-05,
      "step": 10619,
      "training_step_time": 17.497549057006836
    },
    {
      "epoch": 1.0128021240234374e-05,
      "grad_norm": 0.6759490966796875,
      "learning_rate": 1e-05,
      "loss": 0.0627,
      "step": 10620
    },
    {
      "epoch": 1.0128021240234374e-05,
      "model_forward_time": 5.604504823684692,
      "step": 10620
    },
    {
      "dynamics_loss_avg": 0.05658417195081711,
      "epoch": 1.0128021240234374e-05,
      "step": 10620
    },
    {
      "action_loss_avg": 0.008179186098277568,
      "epoch": 1.0128021240234374e-05,
      "step": 10620
    },
    {
      "epoch": 1.0128021240234374e-05,
      "step": 10620,
      "training_step_time": 17.09090280532837
    },
    {
      "epoch": 1.0128974914550782e-05,
      "model_forward_time": 5.612086296081543,
      "step": 10621
    },
    {
      "epoch": 1.0128974914550782e-05,
      "step": 10621,
      "training_step_time": 17.087738752365112
    },
    {
      "epoch": 1.0129928588867187e-05,
      "model_forward_time": 5.601693391799927,
      "step": 10622
    },
    {
      "epoch": 1.0129928588867187e-05,
      "step": 10622,
      "training_step_time": 17.072189807891846
    },
    {
      "epoch": 1.0130882263183593e-05,
      "model_forward_time": 5.6000800132751465,
      "step": 10623
    },
    {
      "epoch": 1.0130882263183593e-05,
      "step": 10623,
      "training_step_time": 17.06351661682129
    },
    {
      "epoch": 1.01318359375e-05,
      "model_forward_time": 5.593868732452393,
      "step": 10624
    },
    {
      "epoch": 1.01318359375e-05,
      "step": 10624,
      "training_step_time": 17.407437086105347
    },
    {
      "epoch": 1.0132789611816406e-05,
      "model_forward_time": 5.597459316253662,
      "step": 10625
    },
    {
      "epoch": 1.0132789611816406e-05,
      "step": 10625,
      "training_step_time": 17.045852661132812
    },
    {
      "epoch": 1.0133743286132812e-05,
      "model_forward_time": 5.591040134429932,
      "step": 10626
    },
    {
      "epoch": 1.0133743286132812e-05,
      "step": 10626,
      "training_step_time": 17.016477823257446
    },
    {
      "epoch": 1.013469696044922e-05,
      "model_forward_time": 5.590776205062866,
      "step": 10627
    },
    {
      "epoch": 1.013469696044922e-05,
      "step": 10627,
      "training_step_time": 17.02974033355713
    },
    {
      "epoch": 1.0135650634765625e-05,
      "model_forward_time": 5.610264301300049,
      "step": 10628
    },
    {
      "epoch": 1.0135650634765625e-05,
      "step": 10628,
      "training_step_time": 17.121873140335083
    },
    {
      "epoch": 1.013660430908203e-05,
      "model_forward_time": 5.61711573600769,
      "step": 10629
    },
    {
      "epoch": 1.013660430908203e-05,
      "step": 10629,
      "training_step_time": 17.119609355926514
    },
    {
      "epoch": 1.0137557983398438e-05,
      "grad_norm": 0.742874264717102,
      "learning_rate": 1e-05,
      "loss": 0.0837,
      "step": 10630
    },
    {
      "epoch": 1.0137557983398438e-05,
      "model_forward_time": 5.975187540054321,
      "step": 10630
    },
    {
      "dynamics_loss_avg": 0.07360212532803416,
      "epoch": 1.0137557983398438e-05,
      "step": 10630
    },
    {
      "action_loss_avg": 0.014052842021919787,
      "epoch": 1.0137557983398438e-05,
      "step": 10630
    },
    {
      "epoch": 1.0137557983398438e-05,
      "step": 10630,
      "training_step_time": 17.451656818389893
    },
    {
      "epoch": 1.0138511657714844e-05,
      "model_forward_time": 5.610620737075806,
      "step": 10631
    },
    {
      "epoch": 1.0138511657714844e-05,
      "step": 10631,
      "training_step_time": 17.067234992980957
    },
    {
      "epoch": 1.013946533203125e-05,
      "model_forward_time": 5.622025012969971,
      "step": 10632
    },
    {
      "epoch": 1.013946533203125e-05,
      "step": 10632,
      "training_step_time": 17.06924605369568
    },
    {
      "epoch": 1.0140419006347655e-05,
      "model_forward_time": 5.594818115234375,
      "step": 10633
    },
    {
      "epoch": 1.0140419006347655e-05,
      "step": 10633,
      "training_step_time": 17.021650314331055
    },
    {
      "epoch": 1.0141372680664063e-05,
      "model_forward_time": 5.593062162399292,
      "step": 10634
    },
    {
      "epoch": 1.0141372680664063e-05,
      "step": 10634,
      "training_step_time": 17.0129234790802
    },
    {
      "epoch": 1.0142326354980469e-05,
      "model_forward_time": 5.957791566848755,
      "step": 10635
    },
    {
      "epoch": 1.0142326354980469e-05,
      "step": 10635,
      "training_step_time": 17.421016454696655
    },
    {
      "epoch": 1.0143280029296874e-05,
      "model_forward_time": 5.5982091426849365,
      "step": 10636
    },
    {
      "epoch": 1.0143280029296874e-05,
      "step": 10636,
      "training_step_time": 17.063860416412354
    },
    {
      "epoch": 1.0144233703613282e-05,
      "model_forward_time": 5.610975980758667,
      "step": 10637
    },
    {
      "epoch": 1.0144233703613282e-05,
      "step": 10637,
      "training_step_time": 17.09742307662964
    },
    {
      "epoch": 1.0145187377929687e-05,
      "model_forward_time": 5.609618902206421,
      "step": 10638
    },
    {
      "epoch": 1.0145187377929687e-05,
      "step": 10638,
      "training_step_time": 17.077346086502075
    },
    {
      "epoch": 1.0146141052246093e-05,
      "model_forward_time": 5.601954936981201,
      "step": 10639
    },
    {
      "epoch": 1.0146141052246093e-05,
      "step": 10639,
      "training_step_time": 17.075227975845337
    },
    {
      "epoch": 1.01470947265625e-05,
      "grad_norm": 0.45212122797966003,
      "learning_rate": 1e-05,
      "loss": 0.0551,
      "step": 10640
    },
    {
      "epoch": 1.01470947265625e-05,
      "model_forward_time": 5.602703332901001,
      "step": 10640
    },
    {
      "dynamics_loss_avg": 0.04895722828805447,
      "epoch": 1.01470947265625e-05,
      "step": 10640
    },
    {
      "action_loss_avg": 0.005536298966035247,
      "epoch": 1.01470947265625e-05,
      "step": 10640
    },
    {
      "epoch": 1.01470947265625e-05,
      "step": 10640,
      "training_step_time": 17.080829858779907
    },
    {
      "epoch": 1.0148048400878906e-05,
      "model_forward_time": 5.983694553375244,
      "step": 10641
    },
    {
      "epoch": 1.0148048400878906e-05,
      "step": 10641,
      "training_step_time": 17.454991579055786
    },
    {
      "epoch": 1.0149002075195312e-05,
      "model_forward_time": 5.600627899169922,
      "step": 10642
    },
    {
      "epoch": 1.0149002075195312e-05,
      "step": 10642,
      "training_step_time": 17.035342693328857
    },
    {
      "epoch": 1.014995574951172e-05,
      "model_forward_time": 5.59189248085022,
      "step": 10643
    },
    {
      "epoch": 1.014995574951172e-05,
      "step": 10643,
      "training_step_time": 17.04150414466858
    },
    {
      "epoch": 1.0150909423828125e-05,
      "model_forward_time": 5.5923686027526855,
      "step": 10644
    },
    {
      "epoch": 1.0150909423828125e-05,
      "step": 10644,
      "training_step_time": 17.027194023132324
    },
    {
      "epoch": 1.015186309814453e-05,
      "model_forward_time": 5.584492921829224,
      "step": 10645
    },
    {
      "epoch": 1.015186309814453e-05,
      "step": 10645,
      "training_step_time": 17.040704011917114
    },
    {
      "epoch": 1.0152816772460938e-05,
      "model_forward_time": 5.6023337841033936,
      "step": 10646
    },
    {
      "epoch": 1.0152816772460938e-05,
      "step": 10646,
      "training_step_time": 17.431200981140137
    },
    {
      "epoch": 1.0153770446777344e-05,
      "model_forward_time": 5.610236644744873,
      "step": 10647
    },
    {
      "epoch": 1.0153770446777344e-05,
      "step": 10647,
      "training_step_time": 17.105773210525513
    },
    {
      "epoch": 1.015472412109375e-05,
      "model_forward_time": 5.626831769943237,
      "step": 10648
    },
    {
      "epoch": 1.015472412109375e-05,
      "step": 10648,
      "training_step_time": 17.145299911499023
    },
    {
      "epoch": 1.0155677795410155e-05,
      "model_forward_time": 5.625962972640991,
      "step": 10649
    },
    {
      "epoch": 1.0155677795410155e-05,
      "step": 10649,
      "training_step_time": 17.13328790664673
    },
    {
      "epoch": 1.0156631469726563e-05,
      "grad_norm": 0.7455525994300842,
      "learning_rate": 1e-05,
      "loss": 0.0594,
      "step": 10650
    },
    {
      "epoch": 1.0156631469726563e-05,
      "model_forward_time": 5.613715171813965,
      "step": 10650
    },
    {
      "dynamics_loss_avg": 0.052270450815558436,
      "epoch": 1.0156631469726563e-05,
      "step": 10650
    },
    {
      "action_loss_avg": 0.005032294848933816,
      "epoch": 1.0156631469726563e-05,
      "step": 10650
    },
    {
      "epoch": 1.0156631469726563e-05,
      "step": 10650,
      "training_step_time": 17.121849060058594
    },
    {
      "epoch": 1.0157585144042969e-05,
      "model_forward_time": 5.627213716506958,
      "step": 10651
    },
    {
      "epoch": 1.0157585144042969e-05,
      "step": 10651,
      "training_step_time": 17.125537872314453
    },
    {
      "epoch": 1.0158538818359374e-05,
      "model_forward_time": 6.007117509841919,
      "step": 10652
    },
    {
      "epoch": 1.0158538818359374e-05,
      "step": 10652,
      "training_step_time": 17.493213653564453
    },
    {
      "epoch": 1.0159492492675782e-05,
      "model_forward_time": 5.591264247894287,
      "step": 10653
    },
    {
      "epoch": 1.0159492492675782e-05,
      "step": 10653,
      "training_step_time": 17.001369953155518
    },
    {
      "epoch": 1.0160446166992187e-05,
      "model_forward_time": 5.5795838832855225,
      "step": 10654
    },
    {
      "epoch": 1.0160446166992187e-05,
      "step": 10654,
      "training_step_time": 17.010870933532715
    },
    {
      "epoch": 1.0161399841308593e-05,
      "model_forward_time": 5.583573579788208,
      "step": 10655
    },
    {
      "epoch": 1.0161399841308593e-05,
      "step": 10655,
      "training_step_time": 16.988196849822998
    },
    {
      "epoch": 1.0162353515625e-05,
      "model_forward_time": 5.589067697525024,
      "step": 10656
    },
    {
      "epoch": 1.0162353515625e-05,
      "step": 10656,
      "training_step_time": 17.008817434310913
    },
    {
      "epoch": 1.0163307189941406e-05,
      "model_forward_time": 5.583081483840942,
      "step": 10657
    },
    {
      "epoch": 1.0163307189941406e-05,
      "step": 10657,
      "training_step_time": 17.38890838623047
    },
    {
      "epoch": 1.0164260864257812e-05,
      "model_forward_time": 5.584250450134277,
      "step": 10658
    },
    {
      "epoch": 1.0164260864257812e-05,
      "step": 10658,
      "training_step_time": 17.039987087249756
    },
    {
      "epoch": 1.016521453857422e-05,
      "model_forward_time": 5.591428995132446,
      "step": 10659
    },
    {
      "epoch": 1.016521453857422e-05,
      "step": 10659,
      "training_step_time": 17.02371644973755
    },
    {
      "epoch": 1.0166168212890625e-05,
      "grad_norm": 0.6829612255096436,
      "learning_rate": 1e-05,
      "loss": 0.0689,
      "step": 10660
    },
    {
      "epoch": 1.0166168212890625e-05,
      "model_forward_time": 5.579123020172119,
      "step": 10660
    },
    {
      "dynamics_loss_avg": 0.061773056909441945,
      "epoch": 1.0166168212890625e-05,
      "step": 10660
    },
    {
      "action_loss_avg": 0.008587802690453828,
      "epoch": 1.0166168212890625e-05,
      "step": 10660
    },
    {
      "epoch": 1.0166168212890625e-05,
      "step": 10660,
      "training_step_time": 17.018890142440796
    },
    {
      "epoch": 1.0167121887207031e-05,
      "model_forward_time": 5.603437423706055,
      "step": 10661
    },
    {
      "epoch": 1.0167121887207031e-05,
      "step": 10661,
      "training_step_time": 17.073630332946777
    },
    {
      "epoch": 1.0168075561523438e-05,
      "model_forward_time": 5.614257574081421,
      "step": 10662
    },
    {
      "epoch": 1.0168075561523438e-05,
      "step": 10662,
      "training_step_time": 17.11651039123535
    },
    {
      "epoch": 1.0169029235839844e-05,
      "model_forward_time": 5.989908218383789,
      "step": 10663
    },
    {
      "epoch": 1.0169029235839844e-05,
      "step": 10663,
      "training_step_time": 17.45906376838684
    },
    {
      "epoch": 1.016998291015625e-05,
      "model_forward_time": 5.592688798904419,
      "step": 10664
    },
    {
      "epoch": 1.016998291015625e-05,
      "step": 10664,
      "training_step_time": 17.029082775115967
    },
    {
      "epoch": 1.0170936584472655e-05,
      "model_forward_time": 5.588149070739746,
      "step": 10665
    },
    {
      "epoch": 1.0170936584472655e-05,
      "step": 10665,
      "training_step_time": 17.035855293273926
    },
    {
      "epoch": 1.0171890258789063e-05,
      "model_forward_time": 5.574450731277466,
      "step": 10666
    },
    {
      "epoch": 1.0171890258789063e-05,
      "step": 10666,
      "training_step_time": 16.968746662139893
    },
    {
      "epoch": 1.0172843933105469e-05,
      "model_forward_time": 5.585487604141235,
      "step": 10667
    },
    {
      "epoch": 1.0172843933105469e-05,
      "step": 10667,
      "training_step_time": 16.98962092399597
    },
    {
      "epoch": 1.0173797607421874e-05,
      "model_forward_time": 5.948565721511841,
      "step": 10668
    },
    {
      "epoch": 1.0173797607421874e-05,
      "step": 10668,
      "training_step_time": 17.37777352333069
    },
    {
      "epoch": 1.0174751281738282e-05,
      "model_forward_time": 5.583754539489746,
      "step": 10669
    },
    {
      "epoch": 1.0174751281738282e-05,
      "step": 10669,
      "training_step_time": 17.042739629745483
    },
    {
      "epoch": 1.0175704956054687e-05,
      "grad_norm": 0.42280080914497375,
      "learning_rate": 1e-05,
      "loss": 0.066,
      "step": 10670
    },
    {
      "epoch": 1.0175704956054687e-05,
      "model_forward_time": 5.5955328941345215,
      "step": 10670
    },
    {
      "dynamics_loss_avg": 0.05596985444426537,
      "epoch": 1.0175704956054687e-05,
      "step": 10670
    },
    {
      "action_loss_avg": 0.0073403507936745886,
      "epoch": 1.0175704956054687e-05,
      "step": 10670
    },
    {
      "epoch": 1.0175704956054687e-05,
      "step": 10670,
      "training_step_time": 17.072386264801025
    },
    {
      "epoch": 1.0176658630371093e-05,
      "model_forward_time": 5.614231824874878,
      "step": 10671
    },
    {
      "epoch": 1.0176658630371093e-05,
      "step": 10671,
      "training_step_time": 17.10686707496643
    },
    {
      "epoch": 1.01776123046875e-05,
      "model_forward_time": 5.608864068984985,
      "step": 10672
    },
    {
      "epoch": 1.01776123046875e-05,
      "step": 10672,
      "training_step_time": 17.107300281524658
    },
    {
      "epoch": 1.0178565979003906e-05,
      "model_forward_time": 5.614847421646118,
      "step": 10673
    },
    {
      "epoch": 1.0178565979003906e-05,
      "step": 10673,
      "training_step_time": 17.10253596305847
    },
    {
      "epoch": 1.0179519653320312e-05,
      "model_forward_time": 5.988657474517822,
      "step": 10674
    },
    {
      "epoch": 1.0179519653320312e-05,
      "step": 10674,
      "training_step_time": 17.46898102760315
    },
    {
      "epoch": 1.018047332763672e-05,
      "model_forward_time": 5.6046462059021,
      "step": 10675
    },
    {
      "epoch": 1.018047332763672e-05,
      "step": 10675,
      "training_step_time": 17.058027744293213
    },
    {
      "epoch": 1.0181427001953125e-05,
      "model_forward_time": 5.597524404525757,
      "step": 10676
    },
    {
      "epoch": 1.0181427001953125e-05,
      "step": 10676,
      "training_step_time": 17.031617403030396
    },
    {
      "epoch": 1.0182380676269531e-05,
      "model_forward_time": 5.586174964904785,
      "step": 10677
    },
    {
      "epoch": 1.0182380676269531e-05,
      "step": 10677,
      "training_step_time": 17.069887161254883
    },
    {
      "epoch": 1.0183334350585938e-05,
      "model_forward_time": 5.617274522781372,
      "step": 10678
    },
    {
      "epoch": 1.0183334350585938e-05,
      "step": 10678,
      "training_step_time": 17.131842613220215
    },
    {
      "epoch": 1.0184288024902344e-05,
      "model_forward_time": 5.623734951019287,
      "step": 10679
    },
    {
      "epoch": 1.0184288024902344e-05,
      "step": 10679,
      "training_step_time": 17.48011565208435
    },
    {
      "epoch": 1.018524169921875e-05,
      "grad_norm": 1.2020809650421143,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 10680
    },
    {
      "epoch": 1.018524169921875e-05,
      "model_forward_time": 5.592392683029175,
      "step": 10680
    },
    {
      "dynamics_loss_avg": 0.06397547032684088,
      "epoch": 1.018524169921875e-05,
      "step": 10680
    },
    {
      "action_loss_avg": 0.012041710910852998,
      "epoch": 1.018524169921875e-05,
      "step": 10680
    },
    {
      "epoch": 1.018524169921875e-05,
      "step": 10680,
      "training_step_time": 17.022854566574097
    },
    {
      "epoch": 1.0186195373535155e-05,
      "model_forward_time": 5.565549850463867,
      "step": 10681
    },
    {
      "epoch": 1.0186195373535155e-05,
      "step": 10681,
      "training_step_time": 16.98802900314331
    },
    {
      "epoch": 1.0187149047851563e-05,
      "model_forward_time": 5.555343866348267,
      "step": 10682
    },
    {
      "epoch": 1.0187149047851563e-05,
      "step": 10682,
      "training_step_time": 16.97390866279602
    },
    {
      "epoch": 1.0188102722167969e-05,
      "model_forward_time": 5.561911582946777,
      "step": 10683
    },
    {
      "epoch": 1.0188102722167969e-05,
      "step": 10683,
      "training_step_time": 16.974433660507202
    },
    {
      "epoch": 1.0189056396484374e-05,
      "model_forward_time": 5.559174537658691,
      "step": 10684
    },
    {
      "epoch": 1.0189056396484374e-05,
      "step": 10684,
      "training_step_time": 16.997701406478882
    },
    {
      "epoch": 1.0190010070800782e-05,
      "model_forward_time": 5.9394142627716064,
      "step": 10685
    },
    {
      "epoch": 1.0190010070800782e-05,
      "step": 10685,
      "training_step_time": 17.353294610977173
    },
    {
      "epoch": 1.0190963745117187e-05,
      "model_forward_time": 5.5934484004974365,
      "step": 10686
    },
    {
      "epoch": 1.0190963745117187e-05,
      "step": 10686,
      "training_step_time": 17.00286602973938
    },
    {
      "epoch": 1.0191917419433593e-05,
      "model_forward_time": 5.596283674240112,
      "step": 10687
    },
    {
      "epoch": 1.0191917419433593e-05,
      "step": 10687,
      "training_step_time": 17.01638436317444
    },
    {
      "epoch": 1.019287109375e-05,
      "model_forward_time": 5.578323125839233,
      "step": 10688
    },
    {
      "epoch": 1.019287109375e-05,
      "step": 10688,
      "training_step_time": 17.023472785949707
    },
    {
      "epoch": 1.0193824768066406e-05,
      "model_forward_time": 5.604271411895752,
      "step": 10689
    },
    {
      "epoch": 1.0193824768066406e-05,
      "step": 10689,
      "training_step_time": 17.074256420135498
    },
    {
      "epoch": 1.0194778442382812e-05,
      "grad_norm": 1.1445772647857666,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 10690
    },
    {
      "epoch": 1.0194778442382812e-05,
      "model_forward_time": 5.607903957366943,
      "step": 10690
    },
    {
      "dynamics_loss_avg": 0.06209415104240179,
      "epoch": 1.0194778442382812e-05,
      "step": 10690
    },
    {
      "action_loss_avg": 0.009187985747121274,
      "epoch": 1.0194778442382812e-05,
      "step": 10690
    },
    {
      "epoch": 1.0194778442382812e-05,
      "step": 10690,
      "training_step_time": 17.453426361083984
    },
    {
      "epoch": 1.019573211669922e-05,
      "model_forward_time": 5.60189414024353,
      "step": 10691
    },
    {
      "epoch": 1.019573211669922e-05,
      "step": 10691,
      "training_step_time": 17.06145405769348
    },
    {
      "epoch": 1.0196685791015625e-05,
      "model_forward_time": 5.600995779037476,
      "step": 10692
    },
    {
      "epoch": 1.0196685791015625e-05,
      "step": 10692,
      "training_step_time": 17.045247554779053
    },
    {
      "epoch": 1.0197639465332031e-05,
      "model_forward_time": 5.593242168426514,
      "step": 10693
    },
    {
      "epoch": 1.0197639465332031e-05,
      "step": 10693,
      "training_step_time": 17.04288101196289
    },
    {
      "epoch": 1.0198593139648438e-05,
      "model_forward_time": 5.596947193145752,
      "step": 10694
    },
    {
      "epoch": 1.0198593139648438e-05,
      "step": 10694,
      "training_step_time": 17.04022526741028
    },
    {
      "epoch": 1.0199546813964844e-05,
      "model_forward_time": 5.589054107666016,
      "step": 10695
    },
    {
      "epoch": 1.0199546813964844e-05,
      "step": 10695,
      "training_step_time": 17.027621507644653
    },
    {
      "epoch": 1.020050048828125e-05,
      "model_forward_time": 5.981184005737305,
      "step": 10696
    },
    {
      "epoch": 1.020050048828125e-05,
      "step": 10696,
      "training_step_time": 17.45587968826294
    },
    {
      "epoch": 1.0201454162597655e-05,
      "model_forward_time": 5.599177122116089,
      "step": 10697
    },
    {
      "epoch": 1.0201454162597655e-05,
      "step": 10697,
      "training_step_time": 17.05749225616455
    },
    {
      "epoch": 1.0202407836914063e-05,
      "model_forward_time": 5.600049257278442,
      "step": 10698
    },
    {
      "epoch": 1.0202407836914063e-05,
      "step": 10698,
      "training_step_time": 17.074430227279663
    },
    {
      "epoch": 1.0203361511230469e-05,
      "model_forward_time": 5.604985952377319,
      "step": 10699
    },
    {
      "epoch": 1.0203361511230469e-05,
      "step": 10699,
      "training_step_time": 17.100597381591797
    },
    {
      "epoch": 1.0204315185546874e-05,
      "grad_norm": 1.7202978134155273,
      "learning_rate": 1e-05,
      "loss": 0.0749,
      "step": 10700
    },
    {
      "epoch": 1.0204315185546874e-05,
      "model_forward_time": 5.620265245437622,
      "step": 10700
    },
    {
      "dynamics_loss_avg": 0.07010719794780015,
      "epoch": 1.0204315185546874e-05,
      "step": 10700
    },
    {
      "action_loss_avg": 0.008822707028593868,
      "epoch": 1.0204315185546874e-05,
      "step": 10700
    },
    {
      "epoch": 1.0204315185546874e-05,
      "step": 10700,
      "training_step_time": 17.100223302841187
    },
    {
      "epoch": 1.0205268859863282e-05,
      "model_forward_time": 5.958128452301025,
      "step": 10701
    },
    {
      "epoch": 1.0205268859863282e-05,
      "step": 10701,
      "training_step_time": 17.392629146575928
    },
    {
      "epoch": 1.0206222534179687e-05,
      "model_forward_time": 5.591542720794678,
      "step": 10702
    },
    {
      "epoch": 1.0206222534179687e-05,
      "step": 10702,
      "training_step_time": 17.03559899330139
    },
    {
      "epoch": 1.0207176208496093e-05,
      "model_forward_time": 5.598310947418213,
      "step": 10703
    },
    {
      "epoch": 1.0207176208496093e-05,
      "step": 10703,
      "training_step_time": 17.065842390060425
    },
    {
      "epoch": 1.02081298828125e-05,
      "model_forward_time": 5.6120264530181885,
      "step": 10704
    },
    {
      "epoch": 1.02081298828125e-05,
      "step": 10704,
      "training_step_time": 17.110138654708862
    },
    {
      "epoch": 1.0209083557128906e-05,
      "model_forward_time": 5.617154121398926,
      "step": 10705
    },
    {
      "epoch": 1.0209083557128906e-05,
      "step": 10705,
      "training_step_time": 17.11269450187683
    },
    {
      "epoch": 1.0210037231445312e-05,
      "model_forward_time": 5.619009256362915,
      "step": 10706
    },
    {
      "epoch": 1.0210037231445312e-05,
      "step": 10706,
      "training_step_time": 17.09913682937622
    },
    {
      "epoch": 1.021099090576172e-05,
      "model_forward_time": 5.968273639678955,
      "step": 10707
    },
    {
      "epoch": 1.021099090576172e-05,
      "step": 10707,
      "training_step_time": 17.425100564956665
    },
    {
      "epoch": 1.0211944580078125e-05,
      "model_forward_time": 5.595479965209961,
      "step": 10708
    },
    {
      "epoch": 1.0211944580078125e-05,
      "step": 10708,
      "training_step_time": 17.010415077209473
    },
    {
      "epoch": 1.0212898254394531e-05,
      "model_forward_time": 5.587262392044067,
      "step": 10709
    },
    {
      "epoch": 1.0212898254394531e-05,
      "step": 10709,
      "training_step_time": 16.996094465255737
    },
    {
      "epoch": 1.0213851928710938e-05,
      "grad_norm": 1.0780361890792847,
      "learning_rate": 1e-05,
      "loss": 0.0828,
      "step": 10710
    },
    {
      "epoch": 1.0213851928710938e-05,
      "model_forward_time": 5.5860512256622314,
      "step": 10710
    },
    {
      "dynamics_loss_avg": 0.06723004579544067,
      "epoch": 1.0213851928710938e-05,
      "step": 10710
    },
    {
      "action_loss_avg": 0.009186246700119227,
      "epoch": 1.0213851928710938e-05,
      "step": 10710
    },
    {
      "epoch": 1.0213851928710938e-05,
      "step": 10710,
      "training_step_time": 17.02233576774597
    },
    {
      "epoch": 1.0214805603027344e-05,
      "model_forward_time": 5.588575124740601,
      "step": 10711
    },
    {
      "epoch": 1.0214805603027344e-05,
      "step": 10711,
      "training_step_time": 17.010672569274902
    },
    {
      "epoch": 1.021575927734375e-05,
      "model_forward_time": 5.581700801849365,
      "step": 10712
    },
    {
      "epoch": 1.021575927734375e-05,
      "step": 10712,
      "training_step_time": 17.383739948272705
    },
    {
      "epoch": 1.0216712951660155e-05,
      "model_forward_time": 5.5866804122924805,
      "step": 10713
    },
    {
      "epoch": 1.0216712951660155e-05,
      "step": 10713,
      "training_step_time": 17.01962924003601
    },
    {
      "epoch": 1.0217666625976563e-05,
      "model_forward_time": 5.586886167526245,
      "step": 10714
    },
    {
      "epoch": 1.0217666625976563e-05,
      "step": 10714,
      "training_step_time": 17.0707528591156
    },
    {
      "epoch": 1.0218620300292969e-05,
      "model_forward_time": 5.596001386642456,
      "step": 10715
    },
    {
      "epoch": 1.0218620300292969e-05,
      "step": 10715,
      "training_step_time": 17.074514150619507
    },
    {
      "epoch": 1.0219573974609374e-05,
      "model_forward_time": 5.601006746292114,
      "step": 10716
    },
    {
      "epoch": 1.0219573974609374e-05,
      "step": 10716,
      "training_step_time": 17.075624704360962
    },
    {
      "epoch": 1.0220527648925782e-05,
      "model_forward_time": 5.593991994857788,
      "step": 10717
    },
    {
      "epoch": 1.0220527648925782e-05,
      "step": 10717,
      "training_step_time": 17.052918434143066
    },
    {
      "epoch": 1.0221481323242187e-05,
      "model_forward_time": 5.983592510223389,
      "step": 10718
    },
    {
      "epoch": 1.0221481323242187e-05,
      "step": 10718,
      "training_step_time": 17.477859497070312
    },
    {
      "epoch": 1.0222434997558593e-05,
      "model_forward_time": 5.625237703323364,
      "step": 10719
    },
    {
      "epoch": 1.0222434997558593e-05,
      "step": 10719,
      "training_step_time": 17.081326961517334
    },
    {
      "epoch": 1.0223388671875e-05,
      "grad_norm": 1.20169198513031,
      "learning_rate": 1e-05,
      "loss": 0.0825,
      "step": 10720
    },
    {
      "epoch": 1.0223388671875e-05,
      "model_forward_time": 5.5941481590271,
      "step": 10720
    },
    {
      "dynamics_loss_avg": 0.08291584774851798,
      "epoch": 1.0223388671875e-05,
      "step": 10720
    },
    {
      "action_loss_avg": 0.00925570004619658,
      "epoch": 1.0223388671875e-05,
      "step": 10720
    },
    {
      "epoch": 1.0223388671875e-05,
      "step": 10720,
      "training_step_time": 17.00137186050415
    },
    {
      "epoch": 1.0224342346191406e-05,
      "model_forward_time": 5.5795323848724365,
      "step": 10721
    },
    {
      "epoch": 1.0224342346191406e-05,
      "step": 10721,
      "training_step_time": 17.00874924659729
    },
    {
      "epoch": 1.0225296020507812e-05,
      "model_forward_time": 5.580357789993286,
      "step": 10722
    },
    {
      "epoch": 1.0225296020507812e-05,
      "step": 10722,
      "training_step_time": 17.03123188018799
    },
    {
      "epoch": 1.022624969482422e-05,
      "model_forward_time": 5.588063716888428,
      "step": 10723
    },
    {
      "epoch": 1.022624969482422e-05,
      "step": 10723,
      "training_step_time": 17.390477418899536
    },
    {
      "epoch": 1.0227203369140625e-05,
      "model_forward_time": 5.596022367477417,
      "step": 10724
    },
    {
      "epoch": 1.0227203369140625e-05,
      "step": 10724,
      "training_step_time": 17.05411410331726
    },
    {
      "epoch": 1.0228157043457031e-05,
      "model_forward_time": 5.60124397277832,
      "step": 10725
    },
    {
      "epoch": 1.0228157043457031e-05,
      "step": 10725,
      "training_step_time": 17.104247570037842
    },
    {
      "epoch": 1.0229110717773438e-05,
      "model_forward_time": 5.6214280128479,
      "step": 10726
    },
    {
      "epoch": 1.0229110717773438e-05,
      "step": 10726,
      "training_step_time": 17.1261146068573
    },
    {
      "epoch": 1.0230064392089844e-05,
      "model_forward_time": 5.606874227523804,
      "step": 10727
    },
    {
      "epoch": 1.0230064392089844e-05,
      "step": 10727,
      "training_step_time": 17.021029233932495
    },
    {
      "epoch": 1.023101806640625e-05,
      "model_forward_time": 5.605741739273071,
      "step": 10728
    },
    {
      "epoch": 1.023101806640625e-05,
      "step": 10728,
      "training_step_time": 17.0467586517334
    },
    {
      "epoch": 1.0231971740722655e-05,
      "model_forward_time": 5.955204248428345,
      "step": 10729
    },
    {
      "epoch": 1.0231971740722655e-05,
      "step": 10729,
      "training_step_time": 17.370115280151367
    },
    {
      "epoch": 1.0232925415039063e-05,
      "grad_norm": 0.757116973400116,
      "learning_rate": 1e-05,
      "loss": 0.0851,
      "step": 10730
    },
    {
      "epoch": 1.0232925415039063e-05,
      "model_forward_time": 5.5961713790893555,
      "step": 10730
    },
    {
      "dynamics_loss_avg": 0.06841178182512522,
      "epoch": 1.0232925415039063e-05,
      "step": 10730
    },
    {
      "action_loss_avg": 0.008345017139799894,
      "epoch": 1.0232925415039063e-05,
      "step": 10730
    },
    {
      "epoch": 1.0232925415039063e-05,
      "step": 10730,
      "training_step_time": 17.08371067047119
    },
    {
      "epoch": 1.0233879089355469e-05,
      "model_forward_time": 5.614836692810059,
      "step": 10731
    },
    {
      "epoch": 1.0233879089355469e-05,
      "step": 10731,
      "training_step_time": 17.14074420928955
    },
    {
      "epoch": 1.0234832763671874e-05,
      "model_forward_time": 5.6245739459991455,
      "step": 10732
    },
    {
      "epoch": 1.0234832763671874e-05,
      "step": 10732,
      "training_step_time": 17.12429404258728
    },
    {
      "epoch": 1.0235786437988282e-05,
      "model_forward_time": 5.613962650299072,
      "step": 10733
    },
    {
      "epoch": 1.0235786437988282e-05,
      "step": 10733,
      "training_step_time": 17.049877166748047
    },
    {
      "epoch": 1.0236740112304687e-05,
      "model_forward_time": 5.952380180358887,
      "step": 10734
    },
    {
      "epoch": 1.0236740112304687e-05,
      "step": 10734,
      "training_step_time": 17.38235831260681
    },
    {
      "epoch": 1.0237693786621093e-05,
      "model_forward_time": 5.59192681312561,
      "step": 10735
    },
    {
      "epoch": 1.0237693786621093e-05,
      "step": 10735,
      "training_step_time": 17.003878116607666
    },
    {
      "epoch": 1.02386474609375e-05,
      "model_forward_time": 5.616476058959961,
      "step": 10736
    },
    {
      "epoch": 1.02386474609375e-05,
      "step": 10736,
      "training_step_time": 17.088362455368042
    },
    {
      "epoch": 1.0239601135253906e-05,
      "model_forward_time": 5.6332621574401855,
      "step": 10737
    },
    {
      "epoch": 1.0239601135253906e-05,
      "step": 10737,
      "training_step_time": 17.128061056137085
    },
    {
      "epoch": 1.0240554809570312e-05,
      "model_forward_time": 5.631535530090332,
      "step": 10738
    },
    {
      "epoch": 1.0240554809570312e-05,
      "step": 10738,
      "training_step_time": 17.166231155395508
    },
    {
      "epoch": 1.024150848388672e-05,
      "model_forward_time": 5.627807140350342,
      "step": 10739
    },
    {
      "epoch": 1.024150848388672e-05,
      "step": 10739,
      "training_step_time": 17.162721395492554
    },
    {
      "epoch": 1.0242462158203125e-05,
      "grad_norm": 0.8559670448303223,
      "learning_rate": 1e-05,
      "loss": 0.0573,
      "step": 10740
    },
    {
      "epoch": 1.0242462158203125e-05,
      "model_forward_time": 6.012844800949097,
      "step": 10740
    },
    {
      "dynamics_loss_avg": 0.050794520787894724,
      "epoch": 1.0242462158203125e-05,
      "step": 10740
    },
    {
      "action_loss_avg": 0.007014051009900868,
      "epoch": 1.0242462158203125e-05,
      "step": 10740
    },
    {
      "epoch": 1.0242462158203125e-05,
      "step": 10740,
      "training_step_time": 17.51590657234192
    },
    {
      "epoch": 1.0243415832519531e-05,
      "model_forward_time": 5.602390289306641,
      "step": 10741
    },
    {
      "epoch": 1.0243415832519531e-05,
      "step": 10741,
      "training_step_time": 17.043335914611816
    },
    {
      "epoch": 1.0244369506835938e-05,
      "model_forward_time": 5.602407217025757,
      "step": 10742
    },
    {
      "epoch": 1.0244369506835938e-05,
      "step": 10742,
      "training_step_time": 17.064309120178223
    },
    {
      "epoch": 1.0245323181152344e-05,
      "model_forward_time": 5.599451303482056,
      "step": 10743
    },
    {
      "epoch": 1.0245323181152344e-05,
      "step": 10743,
      "training_step_time": 17.053478479385376
    },
    {
      "epoch": 1.024627685546875e-05,
      "model_forward_time": 5.600236415863037,
      "step": 10744
    },
    {
      "epoch": 1.024627685546875e-05,
      "step": 10744,
      "training_step_time": 17.028496503829956
    },
    {
      "epoch": 1.0247230529785156e-05,
      "model_forward_time": 5.609782934188843,
      "step": 10745
    },
    {
      "epoch": 1.0247230529785156e-05,
      "step": 10745,
      "training_step_time": 17.440940856933594
    },
    {
      "epoch": 1.0248184204101563e-05,
      "model_forward_time": 5.596068859100342,
      "step": 10746
    },
    {
      "epoch": 1.0248184204101563e-05,
      "step": 10746,
      "training_step_time": 17.07901096343994
    },
    {
      "epoch": 1.0249137878417969e-05,
      "model_forward_time": 5.606398820877075,
      "step": 10747
    },
    {
      "epoch": 1.0249137878417969e-05,
      "step": 10747,
      "training_step_time": 17.09962773323059
    },
    {
      "epoch": 1.0250091552734374e-05,
      "model_forward_time": 5.614875555038452,
      "step": 10748
    },
    {
      "epoch": 1.0250091552734374e-05,
      "step": 10748,
      "training_step_time": 17.111384868621826
    },
    {
      "epoch": 1.0251045227050782e-05,
      "model_forward_time": 5.605568170547485,
      "step": 10749
    },
    {
      "epoch": 1.0251045227050782e-05,
      "step": 10749,
      "training_step_time": 17.090381383895874
    },
    {
      "epoch": 1.0251998901367188e-05,
      "grad_norm": 0.9467220902442932,
      "learning_rate": 1e-05,
      "loss": 0.0642,
      "step": 10750
    },
    {
      "epoch": 1.0251998901367188e-05,
      "model_forward_time": 5.606751918792725,
      "step": 10750
    },
    {
      "dynamics_loss_avg": 0.057044159062206744,
      "epoch": 1.0251998901367188e-05,
      "step": 10750
    },
    {
      "action_loss_avg": 0.00896193995140493,
      "epoch": 1.0251998901367188e-05,
      "step": 10750
    },
    {
      "epoch": 1.0251998901367188e-05,
      "step": 10750,
      "training_step_time": 17.094358682632446
    },
    {
      "epoch": 1.0252952575683593e-05,
      "model_forward_time": 5.994475603103638,
      "step": 10751
    },
    {
      "epoch": 1.0252952575683593e-05,
      "step": 10751,
      "training_step_time": 17.479538202285767
    },
    {
      "epoch": 1.025390625e-05,
      "model_forward_time": 5.607311964035034,
      "step": 10752
    },
    {
      "epoch": 1.025390625e-05,
      "step": 10752,
      "training_step_time": 17.105273485183716
    },
    {
      "epoch": 1.0254859924316406e-05,
      "model_forward_time": 5.609325408935547,
      "step": 10753
    },
    {
      "epoch": 1.0254859924316406e-05,
      "step": 10753,
      "training_step_time": 17.09155511856079
    },
    {
      "epoch": 1.0255813598632812e-05,
      "model_forward_time": 5.626493215560913,
      "step": 10754
    },
    {
      "epoch": 1.0255813598632812e-05,
      "step": 10754,
      "training_step_time": 17.1158127784729
    },
    {
      "epoch": 1.025676727294922e-05,
      "model_forward_time": 5.62011981010437,
      "step": 10755
    },
    {
      "epoch": 1.025676727294922e-05,
      "step": 10755,
      "training_step_time": 17.105270862579346
    },
    {
      "epoch": 1.0257720947265625e-05,
      "model_forward_time": 5.620784282684326,
      "step": 10756
    },
    {
      "epoch": 1.0257720947265625e-05,
      "step": 10756,
      "training_step_time": 17.478246927261353
    },
    {
      "epoch": 1.0258674621582031e-05,
      "model_forward_time": 5.589639902114868,
      "step": 10757
    },
    {
      "epoch": 1.0258674621582031e-05,
      "step": 10757,
      "training_step_time": 17.025120496749878
    },
    {
      "epoch": 1.0259628295898437e-05,
      "model_forward_time": 5.588147401809692,
      "step": 10758
    },
    {
      "epoch": 1.0259628295898437e-05,
      "step": 10758,
      "training_step_time": 16.994272470474243
    },
    {
      "epoch": 1.0260581970214844e-05,
      "model_forward_time": 5.572459697723389,
      "step": 10759
    },
    {
      "epoch": 1.0260581970214844e-05,
      "step": 10759,
      "training_step_time": 16.994938373565674
    },
    {
      "epoch": 1.026153564453125e-05,
      "grad_norm": 0.615865170955658,
      "learning_rate": 1e-05,
      "loss": 0.0825,
      "step": 10760
    },
    {
      "epoch": 1.026153564453125e-05,
      "model_forward_time": 5.569976568222046,
      "step": 10760
    },
    {
      "dynamics_loss_avg": 0.06544623468071223,
      "epoch": 1.026153564453125e-05,
      "step": 10760
    },
    {
      "action_loss_avg": 0.012242467887699605,
      "epoch": 1.026153564453125e-05,
      "step": 10760
    },
    {
      "epoch": 1.026153564453125e-05,
      "step": 10760,
      "training_step_time": 17.006927251815796
    },
    {
      "epoch": 1.0262489318847656e-05,
      "model_forward_time": 5.569241285324097,
      "step": 10761
    },
    {
      "epoch": 1.0262489318847656e-05,
      "step": 10761,
      "training_step_time": 16.972760915756226
    },
    {
      "epoch": 1.0263442993164063e-05,
      "model_forward_time": 5.946475267410278,
      "step": 10762
    },
    {
      "epoch": 1.0263442993164063e-05,
      "step": 10762,
      "training_step_time": 17.367101669311523
    },
    {
      "epoch": 1.0264396667480469e-05,
      "model_forward_time": 5.586541414260864,
      "step": 10763
    },
    {
      "epoch": 1.0264396667480469e-05,
      "step": 10763,
      "training_step_time": 17.032175302505493
    },
    {
      "epoch": 1.0265350341796874e-05,
      "model_forward_time": 5.584780693054199,
      "step": 10764
    },
    {
      "epoch": 1.0265350341796874e-05,
      "step": 10764,
      "training_step_time": 17.013781547546387
    },
    {
      "epoch": 1.0266304016113282e-05,
      "model_forward_time": 5.587949275970459,
      "step": 10765
    },
    {
      "epoch": 1.0266304016113282e-05,
      "step": 10765,
      "training_step_time": 17.067931175231934
    },
    {
      "epoch": 1.0267257690429688e-05,
      "model_forward_time": 5.615057706832886,
      "step": 10766
    },
    {
      "epoch": 1.0267257690429688e-05,
      "step": 10766,
      "training_step_time": 17.132705211639404
    },
    {
      "epoch": 1.0268211364746093e-05,
      "model_forward_time": 5.989318132400513,
      "step": 10767
    },
    {
      "epoch": 1.0268211364746093e-05,
      "step": 10767,
      "training_step_time": 17.47564458847046
    },
    {
      "epoch": 1.02691650390625e-05,
      "model_forward_time": 5.600195646286011,
      "step": 10768
    },
    {
      "epoch": 1.02691650390625e-05,
      "step": 10768,
      "training_step_time": 17.066691160202026
    },
    {
      "epoch": 1.0270118713378906e-05,
      "model_forward_time": 5.598236322402954,
      "step": 10769
    },
    {
      "epoch": 1.0270118713378906e-05,
      "step": 10769,
      "training_step_time": 17.056694269180298
    },
    {
      "epoch": 1.0271072387695312e-05,
      "grad_norm": 0.549225389957428,
      "learning_rate": 1e-05,
      "loss": 0.0647,
      "step": 10770
    },
    {
      "epoch": 1.0271072387695312e-05,
      "model_forward_time": 5.596708536148071,
      "step": 10770
    },
    {
      "dynamics_loss_avg": 0.056281177792698145,
      "epoch": 1.0271072387695312e-05,
      "step": 10770
    },
    {
      "action_loss_avg": 0.009462042537052184,
      "epoch": 1.0271072387695312e-05,
      "step": 10770
    },
    {
      "epoch": 1.0271072387695312e-05,
      "step": 10770,
      "training_step_time": 17.07838749885559
    },
    {
      "epoch": 1.027202606201172e-05,
      "model_forward_time": 5.613795280456543,
      "step": 10771
    },
    {
      "epoch": 1.027202606201172e-05,
      "step": 10771,
      "training_step_time": 17.115401029586792
    },
    {
      "epoch": 1.0272979736328125e-05,
      "model_forward_time": 5.613992691040039,
      "step": 10772
    },
    {
      "epoch": 1.0272979736328125e-05,
      "step": 10772,
      "training_step_time": 17.131227254867554
    },
    {
      "epoch": 1.0273933410644531e-05,
      "model_forward_time": 5.9999799728393555,
      "step": 10773
    },
    {
      "epoch": 1.0273933410644531e-05,
      "step": 10773,
      "training_step_time": 17.50609827041626
    },
    {
      "epoch": 1.0274887084960937e-05,
      "model_forward_time": 5.615377426147461,
      "step": 10774
    },
    {
      "epoch": 1.0274887084960937e-05,
      "step": 10774,
      "training_step_time": 17.088083744049072
    },
    {
      "epoch": 1.0275840759277344e-05,
      "model_forward_time": 5.592044115066528,
      "step": 10775
    },
    {
      "epoch": 1.0275840759277344e-05,
      "step": 10775,
      "training_step_time": 17.023908376693726
    },
    {
      "epoch": 1.027679443359375e-05,
      "model_forward_time": 5.582392454147339,
      "step": 10776
    },
    {
      "epoch": 1.027679443359375e-05,
      "step": 10776,
      "training_step_time": 17.00598406791687
    },
    {
      "epoch": 1.0277748107910156e-05,
      "model_forward_time": 5.568808317184448,
      "step": 10777
    },
    {
      "epoch": 1.0277748107910156e-05,
      "step": 10777,
      "training_step_time": 16.98030662536621
    },
    {
      "epoch": 1.0278701782226563e-05,
      "model_forward_time": 5.564860582351685,
      "step": 10778
    },
    {
      "epoch": 1.0278701782226563e-05,
      "step": 10778,
      "training_step_time": 17.362581968307495
    },
    {
      "epoch": 1.0279655456542969e-05,
      "model_forward_time": 5.5791099071502686,
      "step": 10779
    },
    {
      "epoch": 1.0279655456542969e-05,
      "step": 10779,
      "training_step_time": 17.005844116210938
    },
    {
      "epoch": 1.0280609130859374e-05,
      "grad_norm": 0.7430670857429504,
      "learning_rate": 1e-05,
      "loss": 0.0654,
      "step": 10780
    },
    {
      "epoch": 1.0280609130859374e-05,
      "model_forward_time": 5.575392484664917,
      "step": 10780
    },
    {
      "dynamics_loss_avg": 0.06746426448225976,
      "epoch": 1.0280609130859374e-05,
      "step": 10780
    },
    {
      "action_loss_avg": 0.010394578566774726,
      "epoch": 1.0280609130859374e-05,
      "step": 10780
    },
    {
      "epoch": 1.0280609130859374e-05,
      "step": 10780,
      "training_step_time": 16.987871885299683
    },
    {
      "epoch": 1.0281562805175782e-05,
      "model_forward_time": 5.584927320480347,
      "step": 10781
    },
    {
      "epoch": 1.0281562805175782e-05,
      "step": 10781,
      "training_step_time": 16.994842290878296
    },
    {
      "epoch": 1.0282516479492188e-05,
      "model_forward_time": 5.589399814605713,
      "step": 10782
    },
    {
      "epoch": 1.0282516479492188e-05,
      "step": 10782,
      "training_step_time": 17.031462907791138
    },
    {
      "epoch": 1.0283470153808593e-05,
      "model_forward_time": 5.586867570877075,
      "step": 10783
    },
    {
      "epoch": 1.0283470153808593e-05,
      "step": 10783,
      "training_step_time": 17.047022104263306
    },
    {
      "epoch": 1.0284423828125e-05,
      "model_forward_time": 6.001371145248413,
      "step": 10784
    },
    {
      "epoch": 1.0284423828125e-05,
      "step": 10784,
      "training_step_time": 17.478456258773804
    },
    {
      "epoch": 1.0285377502441406e-05,
      "model_forward_time": 5.604367017745972,
      "step": 10785
    },
    {
      "epoch": 1.0285377502441406e-05,
      "step": 10785,
      "training_step_time": 17.08436870574951
    },
    {
      "epoch": 1.0286331176757812e-05,
      "model_forward_time": 5.608452081680298,
      "step": 10786
    },
    {
      "epoch": 1.0286331176757812e-05,
      "step": 10786,
      "training_step_time": 17.10270404815674
    },
    {
      "epoch": 1.028728485107422e-05,
      "model_forward_time": 5.6232640743255615,
      "step": 10787
    },
    {
      "epoch": 1.028728485107422e-05,
      "step": 10787,
      "training_step_time": 17.133012533187866
    },
    {
      "epoch": 1.0288238525390625e-05,
      "model_forward_time": 5.618815183639526,
      "step": 10788
    },
    {
      "epoch": 1.0288238525390625e-05,
      "step": 10788,
      "training_step_time": 17.13397765159607
    },
    {
      "epoch": 1.0289192199707031e-05,
      "model_forward_time": 5.621692895889282,
      "step": 10789
    },
    {
      "epoch": 1.0289192199707031e-05,
      "step": 10789,
      "training_step_time": 17.48962378501892
    },
    {
      "epoch": 1.0290145874023437e-05,
      "grad_norm": 1.0565937757492065,
      "learning_rate": 1e-05,
      "loss": 0.0754,
      "step": 10790
    },
    {
      "epoch": 1.0290145874023437e-05,
      "model_forward_time": 5.592079401016235,
      "step": 10790
    },
    {
      "dynamics_loss_avg": 0.054023153707385065,
      "epoch": 1.0290145874023437e-05,
      "step": 10790
    },
    {
      "action_loss_avg": 0.008408167911693454,
      "epoch": 1.0290145874023437e-05,
      "step": 10790
    },
    {
      "epoch": 1.0290145874023437e-05,
      "step": 10790,
      "training_step_time": 17.02569270133972
    },
    {
      "epoch": 1.0291099548339844e-05,
      "model_forward_time": 5.557234764099121,
      "step": 10791
    },
    {
      "epoch": 1.0291099548339844e-05,
      "step": 10791,
      "training_step_time": 16.96055841445923
    },
    {
      "epoch": 1.029205322265625e-05,
      "model_forward_time": 5.559309244155884,
      "step": 10792
    },
    {
      "epoch": 1.029205322265625e-05,
      "step": 10792,
      "training_step_time": 16.96466875076294
    },
    {
      "epoch": 1.0293006896972656e-05,
      "model_forward_time": 5.550616979598999,
      "step": 10793
    },
    {
      "epoch": 1.0293006896972656e-05,
      "step": 10793,
      "training_step_time": 16.954158544540405
    },
    {
      "epoch": 1.0293960571289063e-05,
      "model_forward_time": 5.56048583984375,
      "step": 10794
    },
    {
      "epoch": 1.0293960571289063e-05,
      "step": 10794,
      "training_step_time": 16.990888595581055
    },
    {
      "epoch": 1.0294914245605469e-05,
      "model_forward_time": 5.929056406021118,
      "step": 10795
    },
    {
      "epoch": 1.0294914245605469e-05,
      "step": 10795,
      "training_step_time": 17.333749294281006
    },
    {
      "epoch": 1.0295867919921874e-05,
      "model_forward_time": 5.563727140426636,
      "step": 10796
    },
    {
      "epoch": 1.0295867919921874e-05,
      "step": 10796,
      "training_step_time": 16.9759361743927
    },
    {
      "epoch": 1.0296821594238282e-05,
      "model_forward_time": 5.564208030700684,
      "step": 10797
    },
    {
      "epoch": 1.0296821594238282e-05,
      "step": 10797,
      "training_step_time": 16.99839735031128
    },
    {
      "epoch": 1.0297775268554688e-05,
      "model_forward_time": 5.564614295959473,
      "step": 10798
    },
    {
      "epoch": 1.0297775268554688e-05,
      "step": 10798,
      "training_step_time": 17.008028745651245
    },
    {
      "epoch": 1.0298728942871093e-05,
      "model_forward_time": 5.58123254776001,
      "step": 10799
    },
    {
      "epoch": 1.0298728942871093e-05,
      "step": 10799,
      "training_step_time": 17.01274847984314
    },
    {
      "epoch": 1.02996826171875e-05,
      "grad_norm": 0.6613636612892151,
      "learning_rate": 1e-05,
      "loss": 0.0591,
      "step": 10800
    },
    {
      "epoch": 1.02996826171875e-05,
      "model_forward_time": 5.951982736587524,
      "step": 10800
    },
    {
      "dynamics_loss_avg": 0.05289044845849276,
      "epoch": 1.02996826171875e-05,
      "step": 10800
    },
    {
      "action_loss_avg": 0.006454885553102941,
      "epoch": 1.02996826171875e-05,
      "step": 10800
    },
    {
      "epoch": 1.02996826171875e-05,
      "step": 10800,
      "training_step_time": 17.39896821975708
    },
    {
      "epoch": 1.0300636291503906e-05,
      "model_forward_time": 5.604631185531616,
      "step": 10801
    },
    {
      "epoch": 1.0300636291503906e-05,
      "step": 10801,
      "training_step_time": 17.070729732513428
    },
    {
      "epoch": 1.0301589965820312e-05,
      "model_forward_time": 5.613491535186768,
      "step": 10802
    },
    {
      "epoch": 1.0301589965820312e-05,
      "step": 10802,
      "training_step_time": 17.095062732696533
    },
    {
      "epoch": 1.030254364013672e-05,
      "model_forward_time": 5.602646112442017,
      "step": 10803
    },
    {
      "epoch": 1.030254364013672e-05,
      "step": 10803,
      "training_step_time": 17.06734275817871
    },
    {
      "epoch": 1.0303497314453125e-05,
      "model_forward_time": 5.592994213104248,
      "step": 10804
    },
    {
      "epoch": 1.0303497314453125e-05,
      "step": 10804,
      "training_step_time": 17.043400526046753
    },
    {
      "epoch": 1.0304450988769531e-05,
      "model_forward_time": 5.592745065689087,
      "step": 10805
    },
    {
      "epoch": 1.0304450988769531e-05,
      "step": 10805,
      "training_step_time": 17.030837297439575
    },
    {
      "epoch": 1.0305404663085937e-05,
      "model_forward_time": 5.9518303871154785,
      "step": 10806
    },
    {
      "epoch": 1.0305404663085937e-05,
      "step": 10806,
      "training_step_time": 17.42381978034973
    },
    {
      "epoch": 1.0306358337402344e-05,
      "model_forward_time": 5.597776412963867,
      "step": 10807
    },
    {
      "epoch": 1.0306358337402344e-05,
      "step": 10807,
      "training_step_time": 17.094806671142578
    },
    {
      "epoch": 1.030731201171875e-05,
      "model_forward_time": 5.636845111846924,
      "step": 10808
    },
    {
      "epoch": 1.030731201171875e-05,
      "step": 10808,
      "training_step_time": 17.157238006591797
    },
    {
      "epoch": 1.0308265686035156e-05,
      "model_forward_time": 5.620298862457275,
      "step": 10809
    },
    {
      "epoch": 1.0308265686035156e-05,
      "step": 10809,
      "training_step_time": 17.103623628616333
    },
    {
      "epoch": 1.0309219360351563e-05,
      "grad_norm": 0.6166481375694275,
      "learning_rate": 1e-05,
      "loss": 0.0779,
      "step": 10810
    },
    {
      "epoch": 1.0309219360351563e-05,
      "model_forward_time": 5.613161087036133,
      "step": 10810
    },
    {
      "dynamics_loss_avg": 0.0701051790267229,
      "epoch": 1.0309219360351563e-05,
      "step": 10810
    },
    {
      "action_loss_avg": 0.008557363552972675,
      "epoch": 1.0309219360351563e-05,
      "step": 10810
    },
    {
      "epoch": 1.0309219360351563e-05,
      "step": 10810,
      "training_step_time": 17.060630559921265
    },
    {
      "epoch": 1.0310173034667969e-05,
      "model_forward_time": 5.584515333175659,
      "step": 10811
    },
    {
      "epoch": 1.0310173034667969e-05,
      "step": 10811,
      "training_step_time": 17.39220356941223
    },
    {
      "epoch": 1.0311126708984374e-05,
      "model_forward_time": 5.599361181259155,
      "step": 10812
    },
    {
      "epoch": 1.0311126708984374e-05,
      "step": 10812,
      "training_step_time": 17.06346559524536
    },
    {
      "epoch": 1.0312080383300782e-05,
      "model_forward_time": 5.601832151412964,
      "step": 10813
    },
    {
      "epoch": 1.0312080383300782e-05,
      "step": 10813,
      "training_step_time": 17.079832315444946
    },
    {
      "epoch": 1.0313034057617188e-05,
      "model_forward_time": 5.5926806926727295,
      "step": 10814
    },
    {
      "epoch": 1.0313034057617188e-05,
      "step": 10814,
      "training_step_time": 17.061026573181152
    },
    {
      "epoch": 1.0313987731933593e-05,
      "model_forward_time": 5.606487274169922,
      "step": 10815
    },
    {
      "epoch": 1.0313987731933593e-05,
      "step": 10815,
      "training_step_time": 17.073697090148926
    },
    {
      "epoch": 1.031494140625e-05,
      "model_forward_time": 5.58624005317688,
      "step": 10816
    },
    {
      "epoch": 1.031494140625e-05,
      "step": 10816,
      "training_step_time": 17.045823574066162
    },
    {
      "epoch": 1.0315895080566406e-05,
      "model_forward_time": 5.980614185333252,
      "step": 10817
    },
    {
      "epoch": 1.0315895080566406e-05,
      "step": 10817,
      "training_step_time": 17.439033269882202
    },
    {
      "epoch": 1.0316848754882812e-05,
      "model_forward_time": 5.598301649093628,
      "step": 10818
    },
    {
      "epoch": 1.0316848754882812e-05,
      "step": 10818,
      "training_step_time": 17.062135696411133
    },
    {
      "epoch": 1.031780242919922e-05,
      "model_forward_time": 5.5994672775268555,
      "step": 10819
    },
    {
      "epoch": 1.031780242919922e-05,
      "step": 10819,
      "training_step_time": 17.076400995254517
    },
    {
      "epoch": 1.0318756103515625e-05,
      "grad_norm": 0.3615129888057709,
      "learning_rate": 1e-05,
      "loss": 0.0573,
      "step": 10820
    },
    {
      "epoch": 1.0318756103515625e-05,
      "model_forward_time": 5.617172002792358,
      "step": 10820
    },
    {
      "dynamics_loss_avg": 0.05595693830400705,
      "epoch": 1.0318756103515625e-05,
      "step": 10820
    },
    {
      "action_loss_avg": 0.006890540709719062,
      "epoch": 1.0318756103515625e-05,
      "step": 10820
    },
    {
      "epoch": 1.0318756103515625e-05,
      "step": 10820,
      "training_step_time": 17.10718870162964
    },
    {
      "epoch": 1.0319709777832031e-05,
      "model_forward_time": 5.614269018173218,
      "step": 10821
    },
    {
      "epoch": 1.0319709777832031e-05,
      "step": 10821,
      "training_step_time": 17.096505165100098
    },
    {
      "epoch": 1.0320663452148437e-05,
      "model_forward_time": 5.594438314437866,
      "step": 10822
    },
    {
      "epoch": 1.0320663452148437e-05,
      "step": 10822,
      "training_step_time": 17.431334257125854
    },
    {
      "epoch": 1.0321617126464844e-05,
      "model_forward_time": 5.587296009063721,
      "step": 10823
    },
    {
      "epoch": 1.0321617126464844e-05,
      "step": 10823,
      "training_step_time": 17.037659406661987
    },
    {
      "epoch": 1.032257080078125e-05,
      "model_forward_time": 5.595757722854614,
      "step": 10824
    },
    {
      "epoch": 1.032257080078125e-05,
      "step": 10824,
      "training_step_time": 17.042798042297363
    },
    {
      "epoch": 1.0323524475097656e-05,
      "model_forward_time": 5.596563100814819,
      "step": 10825
    },
    {
      "epoch": 1.0323524475097656e-05,
      "step": 10825,
      "training_step_time": 17.088253259658813
    },
    {
      "epoch": 1.0324478149414063e-05,
      "model_forward_time": 5.613322734832764,
      "step": 10826
    },
    {
      "epoch": 1.0324478149414063e-05,
      "step": 10826,
      "training_step_time": 17.088853120803833
    },
    {
      "epoch": 1.0325431823730469e-05,
      "model_forward_time": 5.592138767242432,
      "step": 10827
    },
    {
      "epoch": 1.0325431823730469e-05,
      "step": 10827,
      "training_step_time": 17.012404441833496
    },
    {
      "epoch": 1.0326385498046874e-05,
      "model_forward_time": 5.947054624557495,
      "step": 10828
    },
    {
      "epoch": 1.0326385498046874e-05,
      "step": 10828,
      "training_step_time": 17.399582862854004
    },
    {
      "epoch": 1.0327339172363282e-05,
      "model_forward_time": 5.581038951873779,
      "step": 10829
    },
    {
      "epoch": 1.0327339172363282e-05,
      "step": 10829,
      "training_step_time": 16.986804723739624
    },
    {
      "epoch": 1.0328292846679688e-05,
      "grad_norm": 0.9727718830108643,
      "learning_rate": 1e-05,
      "loss": 0.0622,
      "step": 10830
    },
    {
      "epoch": 1.0328292846679688e-05,
      "model_forward_time": 5.578567981719971,
      "step": 10830
    },
    {
      "dynamics_loss_avg": 0.05114625710994005,
      "epoch": 1.0328292846679688e-05,
      "step": 10830
    },
    {
      "action_loss_avg": 0.005571521795354783,
      "epoch": 1.0328292846679688e-05,
      "step": 10830
    },
    {
      "epoch": 1.0328292846679688e-05,
      "step": 10830,
      "training_step_time": 17.002087354660034
    },
    {
      "epoch": 1.0329246520996093e-05,
      "model_forward_time": 5.590903997421265,
      "step": 10831
    },
    {
      "epoch": 1.0329246520996093e-05,
      "step": 10831,
      "training_step_time": 17.011603593826294
    },
    {
      "epoch": 1.03302001953125e-05,
      "model_forward_time": 5.5977091789245605,
      "step": 10832
    },
    {
      "epoch": 1.03302001953125e-05,
      "step": 10832,
      "training_step_time": 17.080916166305542
    },
    {
      "epoch": 1.0331153869628906e-05,
      "model_forward_time": 5.9721856117248535,
      "step": 10833
    },
    {
      "epoch": 1.0331153869628906e-05,
      "step": 10833,
      "training_step_time": 17.460713386535645
    },
    {
      "epoch": 1.0332107543945312e-05,
      "model_forward_time": 5.611953258514404,
      "step": 10834
    },
    {
      "epoch": 1.0332107543945312e-05,
      "step": 10834,
      "training_step_time": 17.106698751449585
    },
    {
      "epoch": 1.033306121826172e-05,
      "model_forward_time": 5.62717604637146,
      "step": 10835
    },
    {
      "epoch": 1.033306121826172e-05,
      "step": 10835,
      "training_step_time": 17.118164539337158
    },
    {
      "epoch": 1.0334014892578125e-05,
      "model_forward_time": 5.616771221160889,
      "step": 10836
    },
    {
      "epoch": 1.0334014892578125e-05,
      "step": 10836,
      "training_step_time": 17.101404905319214
    },
    {
      "epoch": 1.0334968566894531e-05,
      "model_forward_time": 5.589066505432129,
      "step": 10837
    },
    {
      "epoch": 1.0334968566894531e-05,
      "step": 10837,
      "training_step_time": 17.021624088287354
    },
    {
      "epoch": 1.0335922241210937e-05,
      "model_forward_time": 5.585724830627441,
      "step": 10838
    },
    {
      "epoch": 1.0335922241210937e-05,
      "step": 10838,
      "training_step_time": 16.99232316017151
    },
    {
      "epoch": 1.0336875915527344e-05,
      "model_forward_time": 5.946069002151489,
      "step": 10839
    },
    {
      "epoch": 1.0336875915527344e-05,
      "step": 10839,
      "training_step_time": 17.390069007873535
    },
    {
      "epoch": 1.033782958984375e-05,
      "grad_norm": 0.7726563215255737,
      "learning_rate": 1e-05,
      "loss": 0.0619,
      "step": 10840
    },
    {
      "epoch": 1.033782958984375e-05,
      "model_forward_time": 5.580442428588867,
      "step": 10840
    },
    {
      "dynamics_loss_avg": 0.05573264025151729,
      "epoch": 1.033782958984375e-05,
      "step": 10840
    },
    {
      "action_loss_avg": 0.008406323613598943,
      "epoch": 1.033782958984375e-05,
      "step": 10840
    },
    {
      "epoch": 1.033782958984375e-05,
      "step": 10840,
      "training_step_time": 16.985098361968994
    },
    {
      "epoch": 1.0338783264160156e-05,
      "model_forward_time": 5.588145971298218,
      "step": 10841
    },
    {
      "epoch": 1.0338783264160156e-05,
      "step": 10841,
      "training_step_time": 17.06318426132202
    },
    {
      "epoch": 1.0339736938476563e-05,
      "model_forward_time": 5.603766441345215,
      "step": 10842
    },
    {
      "epoch": 1.0339736938476563e-05,
      "step": 10842,
      "training_step_time": 17.103073120117188
    },
    {
      "epoch": 1.0340690612792969e-05,
      "model_forward_time": 5.614134311676025,
      "step": 10843
    },
    {
      "epoch": 1.0340690612792969e-05,
      "step": 10843,
      "training_step_time": 17.084769010543823
    },
    {
      "epoch": 1.0341644287109374e-05,
      "model_forward_time": 5.588162660598755,
      "step": 10844
    },
    {
      "epoch": 1.0341644287109374e-05,
      "step": 10844,
      "training_step_time": 17.378627061843872
    },
    {
      "epoch": 1.0342597961425782e-05,
      "model_forward_time": 5.587079048156738,
      "step": 10845
    },
    {
      "epoch": 1.0342597961425782e-05,
      "step": 10845,
      "training_step_time": 17.012104272842407
    },
    {
      "epoch": 1.0343551635742188e-05,
      "model_forward_time": 5.586636781692505,
      "step": 10846
    },
    {
      "epoch": 1.0343551635742188e-05,
      "step": 10846,
      "training_step_time": 17.02390956878662
    },
    {
      "epoch": 1.0344505310058593e-05,
      "model_forward_time": 5.600226640701294,
      "step": 10847
    },
    {
      "epoch": 1.0344505310058593e-05,
      "step": 10847,
      "training_step_time": 17.09162187576294
    },
    {
      "epoch": 1.0345458984375e-05,
      "model_forward_time": 5.615547180175781,
      "step": 10848
    },
    {
      "epoch": 1.0345458984375e-05,
      "step": 10848,
      "training_step_time": 17.11817717552185
    },
    {
      "epoch": 1.0346412658691406e-05,
      "model_forward_time": 5.605618476867676,
      "step": 10849
    },
    {
      "epoch": 1.0346412658691406e-05,
      "step": 10849,
      "training_step_time": 17.088748693466187
    },
    {
      "epoch": 1.0347366333007812e-05,
      "grad_norm": 1.6097750663757324,
      "learning_rate": 1e-05,
      "loss": 0.0532,
      "step": 10850
    },
    {
      "epoch": 1.0347366333007812e-05,
      "model_forward_time": 5.975940704345703,
      "step": 10850
    },
    {
      "dynamics_loss_avg": 0.04764405246824026,
      "epoch": 1.0347366333007812e-05,
      "step": 10850
    },
    {
      "action_loss_avg": 0.006565137079451233,
      "epoch": 1.0347366333007812e-05,
      "step": 10850
    },
    {
      "epoch": 1.0347366333007812e-05,
      "step": 10850,
      "training_step_time": 17.44575309753418
    },
    {
      "epoch": 1.034832000732422e-05,
      "model_forward_time": 5.614574909210205,
      "step": 10851
    },
    {
      "epoch": 1.034832000732422e-05,
      "step": 10851,
      "training_step_time": 17.097766876220703
    },
    {
      "epoch": 1.0349273681640625e-05,
      "model_forward_time": 5.601216077804565,
      "step": 10852
    },
    {
      "epoch": 1.0349273681640625e-05,
      "step": 10852,
      "training_step_time": 17.08058261871338
    },
    {
      "epoch": 1.0350227355957031e-05,
      "model_forward_time": 5.589356184005737,
      "step": 10853
    },
    {
      "epoch": 1.0350227355957031e-05,
      "step": 10853,
      "training_step_time": 17.028443336486816
    },
    {
      "epoch": 1.0351181030273437e-05,
      "model_forward_time": 5.585461854934692,
      "step": 10854
    },
    {
      "epoch": 1.0351181030273437e-05,
      "step": 10854,
      "training_step_time": 17.022139072418213
    },
    {
      "epoch": 1.0352134704589844e-05,
      "model_forward_time": 5.5869975090026855,
      "step": 10855
    },
    {
      "epoch": 1.0352134704589844e-05,
      "step": 10855,
      "training_step_time": 17.408515214920044
    },
    {
      "epoch": 1.035308837890625e-05,
      "model_forward_time": 5.620120525360107,
      "step": 10856
    },
    {
      "epoch": 1.035308837890625e-05,
      "step": 10856,
      "training_step_time": 17.14785861968994
    },
    {
      "epoch": 1.0354042053222656e-05,
      "model_forward_time": 5.634202241897583,
      "step": 10857
    },
    {
      "epoch": 1.0354042053222656e-05,
      "step": 10857,
      "training_step_time": 17.137459754943848
    },
    {
      "epoch": 1.0354995727539063e-05,
      "model_forward_time": 5.605533599853516,
      "step": 10858
    },
    {
      "epoch": 1.0354995727539063e-05,
      "step": 10858,
      "training_step_time": 17.038362979888916
    },
    {
      "epoch": 1.0355949401855469e-05,
      "model_forward_time": 5.587371349334717,
      "step": 10859
    },
    {
      "epoch": 1.0355949401855469e-05,
      "step": 10859,
      "training_step_time": 17.028625965118408
    },
    {
      "epoch": 1.0356903076171874e-05,
      "grad_norm": 1.0579091310501099,
      "learning_rate": 1e-05,
      "loss": 0.0668,
      "step": 10860
    },
    {
      "epoch": 1.0356903076171874e-05,
      "model_forward_time": 5.588334798812866,
      "step": 10860
    },
    {
      "dynamics_loss_avg": 0.055458524636924265,
      "epoch": 1.0356903076171874e-05,
      "step": 10860
    },
    {
      "action_loss_avg": 0.007836793176829815,
      "epoch": 1.0356903076171874e-05,
      "step": 10860
    },
    {
      "epoch": 1.0356903076171874e-05,
      "step": 10860,
      "training_step_time": 17.017014265060425
    },
    {
      "epoch": 1.0357856750488282e-05,
      "model_forward_time": 5.960535287857056,
      "step": 10861
    },
    {
      "epoch": 1.0357856750488282e-05,
      "step": 10861,
      "training_step_time": 17.43833637237549
    },
    {
      "epoch": 1.0358810424804688e-05,
      "model_forward_time": 5.612370252609253,
      "step": 10862
    },
    {
      "epoch": 1.0358810424804688e-05,
      "step": 10862,
      "training_step_time": 17.080342292785645
    },
    {
      "epoch": 1.0359764099121093e-05,
      "model_forward_time": 5.594890356063843,
      "step": 10863
    },
    {
      "epoch": 1.0359764099121093e-05,
      "step": 10863,
      "training_step_time": 17.071857452392578
    },
    {
      "epoch": 1.03607177734375e-05,
      "model_forward_time": 5.609312534332275,
      "step": 10864
    },
    {
      "epoch": 1.03607177734375e-05,
      "step": 10864,
      "training_step_time": 17.094115495681763
    },
    {
      "epoch": 1.0361671447753906e-05,
      "model_forward_time": 5.611804246902466,
      "step": 10865
    },
    {
      "epoch": 1.0361671447753906e-05,
      "step": 10865,
      "training_step_time": 17.123181104660034
    },
    {
      "epoch": 1.0362625122070312e-05,
      "model_forward_time": 5.98267674446106,
      "step": 10866
    },
    {
      "epoch": 1.0362625122070312e-05,
      "step": 10866,
      "training_step_time": 17.46523690223694
    },
    {
      "epoch": 1.036357879638672e-05,
      "model_forward_time": 5.608415126800537,
      "step": 10867
    },
    {
      "epoch": 1.036357879638672e-05,
      "step": 10867,
      "training_step_time": 17.07349920272827
    },
    {
      "epoch": 1.0364532470703125e-05,
      "model_forward_time": 5.641139030456543,
      "step": 10868
    },
    {
      "epoch": 1.0364532470703125e-05,
      "step": 10868,
      "training_step_time": 17.106582403182983
    },
    {
      "epoch": 1.0365486145019531e-05,
      "model_forward_time": 5.591657638549805,
      "step": 10869
    },
    {
      "epoch": 1.0365486145019531e-05,
      "step": 10869,
      "training_step_time": 17.04253602027893
    },
    {
      "epoch": 1.0366439819335937e-05,
      "grad_norm": 1.5403295755386353,
      "learning_rate": 1e-05,
      "loss": 0.0723,
      "step": 10870
    },
    {
      "epoch": 1.0366439819335937e-05,
      "model_forward_time": 5.595502853393555,
      "step": 10870
    },
    {
      "dynamics_loss_avg": 0.07089835368096828,
      "epoch": 1.0366439819335937e-05,
      "step": 10870
    },
    {
      "action_loss_avg": 0.008034419873729349,
      "epoch": 1.0366439819335937e-05,
      "step": 10870
    },
    {
      "epoch": 1.0366439819335937e-05,
      "step": 10870,
      "training_step_time": 17.046707153320312
    },
    {
      "epoch": 1.0367393493652344e-05,
      "model_forward_time": 5.594558238983154,
      "step": 10871
    },
    {
      "epoch": 1.0367393493652344e-05,
      "step": 10871,
      "training_step_time": 17.066396713256836
    },
    {
      "epoch": 1.036834716796875e-05,
      "model_forward_time": 5.985610485076904,
      "step": 10872
    },
    {
      "epoch": 1.036834716796875e-05,
      "step": 10872,
      "training_step_time": 17.47675609588623
    },
    {
      "epoch": 1.0369300842285156e-05,
      "model_forward_time": 5.6157004833221436,
      "step": 10873
    },
    {
      "epoch": 1.0369300842285156e-05,
      "step": 10873,
      "training_step_time": 17.10141682624817
    },
    {
      "epoch": 1.0370254516601563e-05,
      "model_forward_time": 5.619764804840088,
      "step": 10874
    },
    {
      "epoch": 1.0370254516601563e-05,
      "step": 10874,
      "training_step_time": 17.105645895004272
    },
    {
      "epoch": 1.0371208190917969e-05,
      "model_forward_time": 5.619863033294678,
      "step": 10875
    },
    {
      "epoch": 1.0371208190917969e-05,
      "step": 10875,
      "training_step_time": 17.093189001083374
    },
    {
      "epoch": 1.0372161865234374e-05,
      "model_forward_time": 5.594134092330933,
      "step": 10876
    },
    {
      "epoch": 1.0372161865234374e-05,
      "step": 10876,
      "training_step_time": 17.006200075149536
    },
    {
      "epoch": 1.0373115539550782e-05,
      "model_forward_time": 5.58868145942688,
      "step": 10877
    },
    {
      "epoch": 1.0373115539550782e-05,
      "step": 10877,
      "training_step_time": 17.388036727905273
    },
    {
      "epoch": 1.0374069213867188e-05,
      "model_forward_time": 5.57423734664917,
      "step": 10878
    },
    {
      "epoch": 1.0374069213867188e-05,
      "step": 10878,
      "training_step_time": 16.988778591156006
    },
    {
      "epoch": 1.0375022888183593e-05,
      "model_forward_time": 5.586621522903442,
      "step": 10879
    },
    {
      "epoch": 1.0375022888183593e-05,
      "step": 10879,
      "training_step_time": 16.995944499969482
    },
    {
      "epoch": 1.03759765625e-05,
      "grad_norm": 0.6437689065933228,
      "learning_rate": 1e-05,
      "loss": 0.0641,
      "step": 10880
    },
    {
      "epoch": 1.03759765625e-05,
      "model_forward_time": 5.588716745376587,
      "step": 10880
    },
    {
      "dynamics_loss_avg": 0.04880734048783779,
      "epoch": 1.03759765625e-05,
      "step": 10880
    },
    {
      "action_loss_avg": 0.00822335509583354,
      "epoch": 1.03759765625e-05,
      "step": 10880
    },
    {
      "epoch": 1.03759765625e-05,
      "step": 10880,
      "training_step_time": 17.01577877998352
    },
    {
      "epoch": 1.0376930236816406e-05,
      "model_forward_time": 5.58237624168396,
      "step": 10881
    },
    {
      "epoch": 1.0376930236816406e-05,
      "step": 10881,
      "training_step_time": 17.033063888549805
    },
    {
      "epoch": 1.0377883911132812e-05,
      "model_forward_time": 5.604549169540405,
      "step": 10882
    },
    {
      "epoch": 1.0377883911132812e-05,
      "step": 10882,
      "training_step_time": 17.04592800140381
    },
    {
      "epoch": 1.0378837585449218e-05,
      "model_forward_time": 5.973999261856079,
      "step": 10883
    },
    {
      "epoch": 1.0378837585449218e-05,
      "step": 10883,
      "training_step_time": 17.430402040481567
    },
    {
      "epoch": 1.0379791259765625e-05,
      "model_forward_time": 5.609437465667725,
      "step": 10884
    },
    {
      "epoch": 1.0379791259765625e-05,
      "step": 10884,
      "training_step_time": 17.09285044670105
    },
    {
      "epoch": 1.0380744934082031e-05,
      "model_forward_time": 5.617497444152832,
      "step": 10885
    },
    {
      "epoch": 1.0380744934082031e-05,
      "step": 10885,
      "training_step_time": 17.12557101249695
    },
    {
      "epoch": 1.0381698608398437e-05,
      "model_forward_time": 5.619325399398804,
      "step": 10886
    },
    {
      "epoch": 1.0381698608398437e-05,
      "step": 10886,
      "training_step_time": 17.076998949050903
    },
    {
      "epoch": 1.0382652282714844e-05,
      "model_forward_time": 5.585533857345581,
      "step": 10887
    },
    {
      "epoch": 1.0382652282714844e-05,
      "step": 10887,
      "training_step_time": 17.01170825958252
    },
    {
      "epoch": 1.038360595703125e-05,
      "model_forward_time": 5.587043523788452,
      "step": 10888
    },
    {
      "epoch": 1.038360595703125e-05,
      "step": 10888,
      "training_step_time": 17.39117980003357
    },
    {
      "epoch": 1.0384559631347656e-05,
      "model_forward_time": 5.583472013473511,
      "step": 10889
    },
    {
      "epoch": 1.0384559631347656e-05,
      "step": 10889,
      "training_step_time": 17.024779558181763
    },
    {
      "epoch": 1.0385513305664063e-05,
      "grad_norm": 0.5825853943824768,
      "learning_rate": 1e-05,
      "loss": 0.0709,
      "step": 10890
    },
    {
      "epoch": 1.0385513305664063e-05,
      "model_forward_time": 5.584728002548218,
      "step": 10890
    },
    {
      "dynamics_loss_avg": 0.06869004704058171,
      "epoch": 1.0385513305664063e-05,
      "step": 10890
    },
    {
      "action_loss_avg": 0.009689360507763922,
      "epoch": 1.0385513305664063e-05,
      "step": 10890
    },
    {
      "epoch": 1.0385513305664063e-05,
      "step": 10890,
      "training_step_time": 17.00803518295288
    },
    {
      "epoch": 1.0386466979980469e-05,
      "model_forward_time": 5.590491771697998,
      "step": 10891
    },
    {
      "epoch": 1.0386466979980469e-05,
      "step": 10891,
      "training_step_time": 17.009800910949707
    },
    {
      "epoch": 1.0387420654296874e-05,
      "model_forward_time": 5.581110715866089,
      "step": 10892
    },
    {
      "epoch": 1.0387420654296874e-05,
      "step": 10892,
      "training_step_time": 17.02459478378296
    },
    {
      "epoch": 1.0388374328613282e-05,
      "model_forward_time": 5.586369752883911,
      "step": 10893
    },
    {
      "epoch": 1.0388374328613282e-05,
      "step": 10893,
      "training_step_time": 17.055249214172363
    },
    {
      "epoch": 1.0389328002929688e-05,
      "model_forward_time": 5.9755754470825195,
      "step": 10894
    },
    {
      "epoch": 1.0389328002929688e-05,
      "step": 10894,
      "training_step_time": 17.48334574699402
    },
    {
      "epoch": 1.0390281677246093e-05,
      "model_forward_time": 5.609250545501709,
      "step": 10895
    },
    {
      "epoch": 1.0390281677246093e-05,
      "step": 10895,
      "training_step_time": 17.059499263763428
    },
    {
      "epoch": 1.03912353515625e-05,
      "model_forward_time": 5.585385799407959,
      "step": 10896
    },
    {
      "epoch": 1.03912353515625e-05,
      "step": 10896,
      "training_step_time": 17.026336431503296
    },
    {
      "epoch": 1.0392189025878906e-05,
      "model_forward_time": 5.5705742835998535,
      "step": 10897
    },
    {
      "epoch": 1.0392189025878906e-05,
      "step": 10897,
      "training_step_time": 16.978755474090576
    },
    {
      "epoch": 1.0393142700195312e-05,
      "model_forward_time": 5.569666385650635,
      "step": 10898
    },
    {
      "epoch": 1.0393142700195312e-05,
      "step": 10898,
      "training_step_time": 16.982619047164917
    },
    {
      "epoch": 1.0394096374511718e-05,
      "model_forward_time": 5.926194190979004,
      "step": 10899
    },
    {
      "epoch": 1.0394096374511718e-05,
      "step": 10899,
      "training_step_time": 17.33969283103943
    },
    {
      "epoch": 1.0395050048828125e-05,
      "grad_norm": 1.400097370147705,
      "learning_rate": 1e-05,
      "loss": 0.078,
      "step": 10900
    },
    {
      "epoch": 1.0395050048828125e-05,
      "model_forward_time": 5.578952789306641,
      "step": 10900
    },
    {
      "dynamics_loss_avg": 0.06378489304333926,
      "epoch": 1.0395050048828125e-05,
      "step": 10900
    },
    {
      "action_loss_avg": 0.008788810088299214,
      "epoch": 1.0395050048828125e-05,
      "step": 10900
    },
    {
      "epoch": 1.0395050048828125e-05,
      "step": 10900,
      "training_step_time": 17.020005464553833
    },
    {
      "epoch": 1.0396003723144531e-05,
      "model_forward_time": 5.589845657348633,
      "step": 10901
    },
    {
      "epoch": 1.0396003723144531e-05,
      "step": 10901,
      "training_step_time": 17.0277681350708
    },
    {
      "epoch": 1.0396957397460937e-05,
      "model_forward_time": 5.587165594100952,
      "step": 10902
    },
    {
      "epoch": 1.0396957397460937e-05,
      "step": 10902,
      "training_step_time": 17.050759315490723
    },
    {
      "epoch": 1.0397911071777344e-05,
      "model_forward_time": 5.601135015487671,
      "step": 10903
    },
    {
      "epoch": 1.0397911071777344e-05,
      "step": 10903,
      "training_step_time": 17.072426080703735
    },
    {
      "epoch": 1.039886474609375e-05,
      "model_forward_time": 5.606713056564331,
      "step": 10904
    },
    {
      "epoch": 1.039886474609375e-05,
      "step": 10904,
      "training_step_time": 17.104533910751343
    },
    {
      "epoch": 1.0399818420410156e-05,
      "model_forward_time": 6.0054450035095215,
      "step": 10905
    },
    {
      "epoch": 1.0399818420410156e-05,
      "step": 10905,
      "training_step_time": 17.51269555091858
    },
    {
      "epoch": 1.0400772094726563e-05,
      "model_forward_time": 5.620238304138184,
      "step": 10906
    },
    {
      "epoch": 1.0400772094726563e-05,
      "step": 10906,
      "training_step_time": 17.112869262695312
    },
    {
      "epoch": 1.0401725769042969e-05,
      "model_forward_time": 5.650335311889648,
      "step": 10907
    },
    {
      "epoch": 1.0401725769042969e-05,
      "step": 10907,
      "training_step_time": 17.172571897506714
    },
    {
      "epoch": 1.0402679443359374e-05,
      "model_forward_time": 5.624170541763306,
      "step": 10908
    },
    {
      "epoch": 1.0402679443359374e-05,
      "step": 10908,
      "training_step_time": 17.115883111953735
    },
    {
      "epoch": 1.0403633117675782e-05,
      "model_forward_time": 5.613545179367065,
      "step": 10909
    },
    {
      "epoch": 1.0403633117675782e-05,
      "step": 10909,
      "training_step_time": 17.079981803894043
    },
    {
      "epoch": 1.0404586791992188e-05,
      "grad_norm": 1.6923143863677979,
      "learning_rate": 1e-05,
      "loss": 0.0673,
      "step": 10910
    },
    {
      "epoch": 1.0404586791992188e-05,
      "model_forward_time": 5.593175649642944,
      "step": 10910
    },
    {
      "dynamics_loss_avg": 0.05760765187442303,
      "epoch": 1.0404586791992188e-05,
      "step": 10910
    },
    {
      "action_loss_avg": 0.010241892887279391,
      "epoch": 1.0404586791992188e-05,
      "step": 10910
    },
    {
      "epoch": 1.0404586791992188e-05,
      "step": 10910,
      "training_step_time": 17.39577841758728
    },
    {
      "epoch": 1.0405540466308593e-05,
      "model_forward_time": 5.601362943649292,
      "step": 10911
    },
    {
      "epoch": 1.0405540466308593e-05,
      "step": 10911,
      "training_step_time": 17.069049835205078
    },
    {
      "epoch": 1.0406494140625e-05,
      "model_forward_time": 5.6229541301727295,
      "step": 10912
    },
    {
      "epoch": 1.0406494140625e-05,
      "step": 10912,
      "training_step_time": 17.110466241836548
    },
    {
      "epoch": 1.0407447814941406e-05,
      "model_forward_time": 5.611993312835693,
      "step": 10913
    },
    {
      "epoch": 1.0407447814941406e-05,
      "step": 10913,
      "training_step_time": 17.101342916488647
    },
    {
      "epoch": 1.0408401489257812e-05,
      "model_forward_time": 5.603085517883301,
      "step": 10914
    },
    {
      "epoch": 1.0408401489257812e-05,
      "step": 10914,
      "training_step_time": 17.07975935935974
    },
    {
      "epoch": 1.0409355163574218e-05,
      "model_forward_time": 5.606426239013672,
      "step": 10915
    },
    {
      "epoch": 1.0409355163574218e-05,
      "step": 10915,
      "training_step_time": 17.062245845794678
    },
    {
      "epoch": 1.0410308837890625e-05,
      "model_forward_time": 5.985217809677124,
      "step": 10916
    },
    {
      "epoch": 1.0410308837890625e-05,
      "step": 10916,
      "training_step_time": 17.45746397972107
    },
    {
      "epoch": 1.0411262512207031e-05,
      "model_forward_time": 5.629739999771118,
      "step": 10917
    },
    {
      "epoch": 1.0411262512207031e-05,
      "step": 10917,
      "training_step_time": 17.15408444404602
    },
    {
      "epoch": 1.0412216186523437e-05,
      "model_forward_time": 5.627583742141724,
      "step": 10918
    },
    {
      "epoch": 1.0412216186523437e-05,
      "step": 10918,
      "training_step_time": 17.13620638847351
    },
    {
      "epoch": 1.0413169860839844e-05,
      "model_forward_time": 5.610940217971802,
      "step": 10919
    },
    {
      "epoch": 1.0413169860839844e-05,
      "step": 10919,
      "training_step_time": 17.060120820999146
    },
    {
      "epoch": 1.041412353515625e-05,
      "grad_norm": 1.2343801259994507,
      "learning_rate": 1e-05,
      "loss": 0.064,
      "step": 10920
    },
    {
      "epoch": 1.041412353515625e-05,
      "model_forward_time": 5.586009740829468,
      "step": 10920
    },
    {
      "dynamics_loss_avg": 0.05447965767234564,
      "epoch": 1.041412353515625e-05,
      "step": 10920
    },
    {
      "action_loss_avg": 0.00989297687774524,
      "epoch": 1.041412353515625e-05,
      "step": 10920
    },
    {
      "epoch": 1.041412353515625e-05,
      "step": 10920,
      "training_step_time": 17.016899347305298
    },
    {
      "epoch": 1.0415077209472656e-05,
      "model_forward_time": 5.59917426109314,
      "step": 10921
    },
    {
      "epoch": 1.0415077209472656e-05,
      "step": 10921,
      "training_step_time": 17.423609972000122
    },
    {
      "epoch": 1.0416030883789063e-05,
      "model_forward_time": 5.596050977706909,
      "step": 10922
    },
    {
      "epoch": 1.0416030883789063e-05,
      "step": 10922,
      "training_step_time": 17.064496517181396
    },
    {
      "epoch": 1.0416984558105469e-05,
      "model_forward_time": 5.617067098617554,
      "step": 10923
    },
    {
      "epoch": 1.0416984558105469e-05,
      "step": 10923,
      "training_step_time": 17.11551308631897
    },
    {
      "epoch": 1.0417938232421875e-05,
      "model_forward_time": 5.62444806098938,
      "step": 10924
    },
    {
      "epoch": 1.0417938232421875e-05,
      "step": 10924,
      "training_step_time": 17.114283084869385
    },
    {
      "epoch": 1.0418891906738282e-05,
      "model_forward_time": 5.606858015060425,
      "step": 10925
    },
    {
      "epoch": 1.0418891906738282e-05,
      "step": 10925,
      "training_step_time": 17.05653190612793
    },
    {
      "epoch": 1.0419845581054688e-05,
      "model_forward_time": 5.584216833114624,
      "step": 10926
    },
    {
      "epoch": 1.0419845581054688e-05,
      "step": 10926,
      "training_step_time": 17.042345762252808
    },
    {
      "epoch": 1.0420799255371093e-05,
      "model_forward_time": 5.975537538528442,
      "step": 10927
    },
    {
      "epoch": 1.0420799255371093e-05,
      "step": 10927,
      "training_step_time": 17.45028042793274
    },
    {
      "epoch": 1.04217529296875e-05,
      "model_forward_time": 5.601647853851318,
      "step": 10928
    },
    {
      "epoch": 1.04217529296875e-05,
      "step": 10928,
      "training_step_time": 17.076454162597656
    },
    {
      "epoch": 1.0422706604003907e-05,
      "model_forward_time": 5.594340085983276,
      "step": 10929
    },
    {
      "epoch": 1.0422706604003907e-05,
      "step": 10929,
      "training_step_time": 17.063050746917725
    },
    {
      "epoch": 1.0423660278320312e-05,
      "grad_norm": 0.31149938702583313,
      "learning_rate": 1e-05,
      "loss": 0.056,
      "step": 10930
    },
    {
      "epoch": 1.0423660278320312e-05,
      "model_forward_time": 5.605790138244629,
      "step": 10930
    },
    {
      "dynamics_loss_avg": 0.04823150262236595,
      "epoch": 1.0423660278320312e-05,
      "step": 10930
    },
    {
      "action_loss_avg": 0.006059398059733212,
      "epoch": 1.0423660278320312e-05,
      "step": 10930
    },
    {
      "epoch": 1.0423660278320312e-05,
      "step": 10930,
      "training_step_time": 17.042579650878906
    },
    {
      "epoch": 1.0424613952636718e-05,
      "model_forward_time": 5.606904029846191,
      "step": 10931
    },
    {
      "epoch": 1.0424613952636718e-05,
      "step": 10931,
      "training_step_time": 17.071380615234375
    },
    {
      "epoch": 1.0425567626953125e-05,
      "model_forward_time": 5.987032175064087,
      "step": 10932
    },
    {
      "epoch": 1.0425567626953125e-05,
      "step": 10932,
      "training_step_time": 17.46160364151001
    },
    {
      "epoch": 1.0426521301269531e-05,
      "model_forward_time": 5.602436542510986,
      "step": 10933
    },
    {
      "epoch": 1.0426521301269531e-05,
      "step": 10933,
      "training_step_time": 17.065233945846558
    },
    {
      "epoch": 1.0427474975585937e-05,
      "model_forward_time": 5.5898661613464355,
      "step": 10934
    },
    {
      "epoch": 1.0427474975585937e-05,
      "step": 10934,
      "training_step_time": 17.020070791244507
    },
    {
      "epoch": 1.0428428649902344e-05,
      "model_forward_time": 5.5862696170806885,
      "step": 10935
    },
    {
      "epoch": 1.0428428649902344e-05,
      "step": 10935,
      "training_step_time": 17.024741649627686
    },
    {
      "epoch": 1.042938232421875e-05,
      "model_forward_time": 5.5893988609313965,
      "step": 10936
    },
    {
      "epoch": 1.042938232421875e-05,
      "step": 10936,
      "training_step_time": 17.046339750289917
    },
    {
      "epoch": 1.0430335998535156e-05,
      "model_forward_time": 5.593171834945679,
      "step": 10937
    },
    {
      "epoch": 1.0430335998535156e-05,
      "step": 10937,
      "training_step_time": 17.048824071884155
    },
    {
      "epoch": 1.0431289672851563e-05,
      "model_forward_time": 5.975947380065918,
      "step": 10938
    },
    {
      "epoch": 1.0431289672851563e-05,
      "step": 10938,
      "training_step_time": 17.451406478881836
    },
    {
      "epoch": 1.0432243347167969e-05,
      "model_forward_time": 5.599933862686157,
      "step": 10939
    },
    {
      "epoch": 1.0432243347167969e-05,
      "step": 10939,
      "training_step_time": 17.086498975753784
    },
    {
      "epoch": 1.0433197021484375e-05,
      "grad_norm": 1.100515365600586,
      "learning_rate": 1e-05,
      "loss": 0.0647,
      "step": 10940
    },
    {
      "epoch": 1.0433197021484375e-05,
      "model_forward_time": 5.61022686958313,
      "step": 10940
    },
    {
      "dynamics_loss_avg": 0.058097480051219466,
      "epoch": 1.0433197021484375e-05,
      "step": 10940
    },
    {
      "action_loss_avg": 0.00830445282626897,
      "epoch": 1.0433197021484375e-05,
      "step": 10940
    },
    {
      "epoch": 1.0433197021484375e-05,
      "step": 10940,
      "training_step_time": 17.09702205657959
    },
    {
      "epoch": 1.0434150695800782e-05,
      "model_forward_time": 5.610234498977661,
      "step": 10941
    },
    {
      "epoch": 1.0434150695800782e-05,
      "step": 10941,
      "training_step_time": 17.089466094970703
    },
    {
      "epoch": 1.0435104370117188e-05,
      "model_forward_time": 5.600274562835693,
      "step": 10942
    },
    {
      "epoch": 1.0435104370117188e-05,
      "step": 10942,
      "training_step_time": 17.033642053604126
    },
    {
      "epoch": 1.0436058044433593e-05,
      "model_forward_time": 5.592558860778809,
      "step": 10943
    },
    {
      "epoch": 1.0436058044433593e-05,
      "step": 10943,
      "training_step_time": 17.37727689743042
    },
    {
      "epoch": 1.043701171875e-05,
      "model_forward_time": 5.579999923706055,
      "step": 10944
    },
    {
      "epoch": 1.043701171875e-05,
      "step": 10944,
      "training_step_time": 17.039371013641357
    },
    {
      "epoch": 1.0437965393066407e-05,
      "model_forward_time": 5.593954563140869,
      "step": 10945
    },
    {
      "epoch": 1.0437965393066407e-05,
      "step": 10945,
      "training_step_time": 17.072086095809937
    },
    {
      "epoch": 1.0438919067382812e-05,
      "model_forward_time": 5.610524654388428,
      "step": 10946
    },
    {
      "epoch": 1.0438919067382812e-05,
      "step": 10946,
      "training_step_time": 17.106486082077026
    },
    {
      "epoch": 1.0439872741699218e-05,
      "model_forward_time": 5.612692832946777,
      "step": 10947
    },
    {
      "epoch": 1.0439872741699218e-05,
      "step": 10947,
      "training_step_time": 17.123831748962402
    },
    {
      "epoch": 1.0440826416015625e-05,
      "model_forward_time": 5.612388610839844,
      "step": 10948
    },
    {
      "epoch": 1.0440826416015625e-05,
      "step": 10948,
      "training_step_time": 17.087292909622192
    },
    {
      "epoch": 1.0441780090332031e-05,
      "model_forward_time": 5.96164083480835,
      "step": 10949
    },
    {
      "epoch": 1.0441780090332031e-05,
      "step": 10949,
      "training_step_time": 17.40349006652832
    },
    {
      "epoch": 1.0442733764648437e-05,
      "grad_norm": 2.163637638092041,
      "learning_rate": 1e-05,
      "loss": 0.058,
      "step": 10950
    },
    {
      "epoch": 1.0442733764648437e-05,
      "model_forward_time": 5.569239854812622,
      "step": 10950
    },
    {
      "dynamics_loss_avg": 0.048046443797647956,
      "epoch": 1.0442733764648437e-05,
      "step": 10950
    },
    {
      "action_loss_avg": 0.00747352777980268,
      "epoch": 1.0442733764648437e-05,
      "step": 10950
    },
    {
      "epoch": 1.0442733764648437e-05,
      "step": 10950,
      "training_step_time": 16.97206974029541
    },
    {
      "epoch": 1.0443687438964844e-05,
      "model_forward_time": 5.570110321044922,
      "step": 10951
    },
    {
      "epoch": 1.0443687438964844e-05,
      "step": 10951,
      "training_step_time": 16.988689661026
    },
    {
      "epoch": 1.044464111328125e-05,
      "model_forward_time": 5.576385498046875,
      "step": 10952
    },
    {
      "epoch": 1.044464111328125e-05,
      "step": 10952,
      "training_step_time": 16.99203372001648
    },
    {
      "epoch": 1.0445594787597656e-05,
      "model_forward_time": 5.576004981994629,
      "step": 10953
    },
    {
      "epoch": 1.0445594787597656e-05,
      "step": 10953,
      "training_step_time": 17.015315055847168
    },
    {
      "epoch": 1.0446548461914063e-05,
      "model_forward_time": 5.571869373321533,
      "step": 10954
    },
    {
      "epoch": 1.0446548461914063e-05,
      "step": 10954,
      "training_step_time": 17.365077257156372
    },
    {
      "epoch": 1.0447502136230469e-05,
      "model_forward_time": 5.586100816726685,
      "step": 10955
    },
    {
      "epoch": 1.0447502136230469e-05,
      "step": 10955,
      "training_step_time": 17.024555921554565
    },
    {
      "epoch": 1.0448455810546875e-05,
      "model_forward_time": 5.577655792236328,
      "step": 10956
    },
    {
      "epoch": 1.0448455810546875e-05,
      "step": 10956,
      "training_step_time": 16.997698068618774
    },
    {
      "epoch": 1.0449409484863282e-05,
      "model_forward_time": 5.59507155418396,
      "step": 10957
    },
    {
      "epoch": 1.0449409484863282e-05,
      "step": 10957,
      "training_step_time": 17.082769632339478
    },
    {
      "epoch": 1.0450363159179688e-05,
      "model_forward_time": 5.607820749282837,
      "step": 10958
    },
    {
      "epoch": 1.0450363159179688e-05,
      "step": 10958,
      "training_step_time": 17.129602670669556
    },
    {
      "epoch": 1.0451316833496093e-05,
      "model_forward_time": 5.623769760131836,
      "step": 10959
    },
    {
      "epoch": 1.0451316833496093e-05,
      "step": 10959,
      "training_step_time": 17.13995623588562
    },
    {
      "epoch": 1.04522705078125e-05,
      "grad_norm": 1.212107539176941,
      "learning_rate": 1e-05,
      "loss": 0.0788,
      "step": 10960
    },
    {
      "epoch": 1.04522705078125e-05,
      "model_forward_time": 6.000421047210693,
      "step": 10960
    },
    {
      "dynamics_loss_avg": 0.07083406019955873,
      "epoch": 1.04522705078125e-05,
      "step": 10960
    },
    {
      "action_loss_avg": 0.009612420981284231,
      "epoch": 1.04522705078125e-05,
      "step": 10960
    },
    {
      "epoch": 1.04522705078125e-05,
      "step": 10960,
      "training_step_time": 17.47661280632019
    },
    {
      "epoch": 1.0453224182128907e-05,
      "model_forward_time": 5.590242862701416,
      "step": 10961
    },
    {
      "epoch": 1.0453224182128907e-05,
      "step": 10961,
      "training_step_time": 17.02406597137451
    },
    {
      "epoch": 1.0454177856445312e-05,
      "model_forward_time": 5.5689377784729,
      "step": 10962
    },
    {
      "epoch": 1.0454177856445312e-05,
      "step": 10962,
      "training_step_time": 16.982455253601074
    },
    {
      "epoch": 1.0455131530761718e-05,
      "model_forward_time": 5.573614597320557,
      "step": 10963
    },
    {
      "epoch": 1.0455131530761718e-05,
      "step": 10963,
      "training_step_time": 16.987799406051636
    },
    {
      "epoch": 1.0456085205078125e-05,
      "model_forward_time": 5.576622486114502,
      "step": 10964
    },
    {
      "epoch": 1.0456085205078125e-05,
      "step": 10964,
      "training_step_time": 17.004172563552856
    },
    {
      "epoch": 1.0457038879394531e-05,
      "model_forward_time": 5.9220921993255615,
      "step": 10965
    },
    {
      "epoch": 1.0457038879394531e-05,
      "step": 10965,
      "training_step_time": 17.346447467803955
    },
    {
      "epoch": 1.0457992553710937e-05,
      "model_forward_time": 5.574681043624878,
      "step": 10966
    },
    {
      "epoch": 1.0457992553710937e-05,
      "step": 10966,
      "training_step_time": 16.987314462661743
    },
    {
      "epoch": 1.0458946228027344e-05,
      "model_forward_time": 5.581483602523804,
      "step": 10967
    },
    {
      "epoch": 1.0458946228027344e-05,
      "step": 10967,
      "training_step_time": 16.990771532058716
    },
    {
      "epoch": 1.045989990234375e-05,
      "model_forward_time": 5.587444067001343,
      "step": 10968
    },
    {
      "epoch": 1.045989990234375e-05,
      "step": 10968,
      "training_step_time": 17.037128925323486
    },
    {
      "epoch": 1.0460853576660156e-05,
      "model_forward_time": 5.586999416351318,
      "step": 10969
    },
    {
      "epoch": 1.0460853576660156e-05,
      "step": 10969,
      "training_step_time": 17.065425872802734
    },
    {
      "epoch": 1.0461807250976563e-05,
      "grad_norm": 0.3756641447544098,
      "learning_rate": 1e-05,
      "loss": 0.0641,
      "step": 10970
    },
    {
      "epoch": 1.0461807250976563e-05,
      "model_forward_time": 5.613704204559326,
      "step": 10970
    },
    {
      "dynamics_loss_avg": 0.057309073954820634,
      "epoch": 1.0461807250976563e-05,
      "step": 10970
    },
    {
      "action_loss_avg": 0.007944186718668788,
      "epoch": 1.0461807250976563e-05,
      "step": 10970
    },
    {
      "epoch": 1.0461807250976563e-05,
      "step": 10970,
      "training_step_time": 17.106228590011597
    },
    {
      "epoch": 1.0462760925292969e-05,
      "model_forward_time": 6.000659942626953,
      "step": 10971
    },
    {
      "epoch": 1.0462760925292969e-05,
      "step": 10971,
      "training_step_time": 17.45643973350525
    },
    {
      "epoch": 1.0463714599609375e-05,
      "model_forward_time": 5.597755432128906,
      "step": 10972
    },
    {
      "epoch": 1.0463714599609375e-05,
      "step": 10972,
      "training_step_time": 17.046228885650635
    },
    {
      "epoch": 1.0464668273925782e-05,
      "model_forward_time": 5.582179307937622,
      "step": 10973
    },
    {
      "epoch": 1.0464668273925782e-05,
      "step": 10973,
      "training_step_time": 17.004717111587524
    },
    {
      "epoch": 1.0465621948242188e-05,
      "model_forward_time": 5.568537712097168,
      "step": 10974
    },
    {
      "epoch": 1.0465621948242188e-05,
      "step": 10974,
      "training_step_time": 16.983672857284546
    },
    {
      "epoch": 1.0466575622558593e-05,
      "model_forward_time": 5.579374551773071,
      "step": 10975
    },
    {
      "epoch": 1.0466575622558593e-05,
      "step": 10975,
      "training_step_time": 17.00125479698181
    },
    {
      "epoch": 1.0467529296875e-05,
      "model_forward_time": 5.56389045715332,
      "step": 10976
    },
    {
      "epoch": 1.0467529296875e-05,
      "step": 10976,
      "training_step_time": 17.369091510772705
    },
    {
      "epoch": 1.0468482971191407e-05,
      "model_forward_time": 5.590672492980957,
      "step": 10977
    },
    {
      "epoch": 1.0468482971191407e-05,
      "step": 10977,
      "training_step_time": 17.003361225128174
    },
    {
      "epoch": 1.0469436645507812e-05,
      "model_forward_time": 5.584759950637817,
      "step": 10978
    },
    {
      "epoch": 1.0469436645507812e-05,
      "step": 10978,
      "training_step_time": 17.04614806175232
    },
    {
      "epoch": 1.0470390319824218e-05,
      "model_forward_time": 5.6033101081848145,
      "step": 10979
    },
    {
      "epoch": 1.0470390319824218e-05,
      "step": 10979,
      "training_step_time": 17.09395456314087
    },
    {
      "epoch": 1.0471343994140625e-05,
      "grad_norm": 0.5030131936073303,
      "learning_rate": 1e-05,
      "loss": 0.0673,
      "step": 10980
    },
    {
      "epoch": 1.0471343994140625e-05,
      "model_forward_time": 5.5991129875183105,
      "step": 10980
    },
    {
      "dynamics_loss_avg": 0.057784291356801985,
      "epoch": 1.0471343994140625e-05,
      "step": 10980
    },
    {
      "action_loss_avg": 0.008864022325724363,
      "epoch": 1.0471343994140625e-05,
      "step": 10980
    },
    {
      "epoch": 1.0471343994140625e-05,
      "step": 10980,
      "training_step_time": 17.04693627357483
    },
    {
      "epoch": 1.0472297668457031e-05,
      "model_forward_time": 5.59771203994751,
      "step": 10981
    },
    {
      "epoch": 1.0472297668457031e-05,
      "step": 10981,
      "training_step_time": 17.045486211776733
    },
    {
      "epoch": 1.0473251342773437e-05,
      "model_forward_time": 5.957891225814819,
      "step": 10982
    },
    {
      "epoch": 1.0473251342773437e-05,
      "step": 10982,
      "training_step_time": 17.40307068824768
    },
    {
      "epoch": 1.0474205017089844e-05,
      "model_forward_time": 5.581518888473511,
      "step": 10983
    },
    {
      "epoch": 1.0474205017089844e-05,
      "step": 10983,
      "training_step_time": 17.02834916114807
    },
    {
      "epoch": 1.047515869140625e-05,
      "model_forward_time": 5.5936079025268555,
      "step": 10984
    },
    {
      "epoch": 1.047515869140625e-05,
      "step": 10984,
      "training_step_time": 17.06514883041382
    },
    {
      "epoch": 1.0476112365722656e-05,
      "model_forward_time": 5.598475456237793,
      "step": 10985
    },
    {
      "epoch": 1.0476112365722656e-05,
      "step": 10985,
      "training_step_time": 17.067628145217896
    },
    {
      "epoch": 1.0477066040039063e-05,
      "model_forward_time": 5.598110914230347,
      "step": 10986
    },
    {
      "epoch": 1.0477066040039063e-05,
      "step": 10986,
      "training_step_time": 17.0724515914917
    },
    {
      "epoch": 1.0478019714355469e-05,
      "model_forward_time": 5.613025188446045,
      "step": 10987
    },
    {
      "epoch": 1.0478019714355469e-05,
      "step": 10987,
      "training_step_time": 17.436920881271362
    },
    {
      "epoch": 1.0478973388671875e-05,
      "model_forward_time": 5.605465412139893,
      "step": 10988
    },
    {
      "epoch": 1.0478973388671875e-05,
      "step": 10988,
      "training_step_time": 17.072213172912598
    },
    {
      "epoch": 1.0479927062988282e-05,
      "model_forward_time": 5.600261211395264,
      "step": 10989
    },
    {
      "epoch": 1.0479927062988282e-05,
      "step": 10989,
      "training_step_time": 17.053558111190796
    },
    {
      "epoch": 1.0480880737304688e-05,
      "grad_norm": 0.9558208584785461,
      "learning_rate": 1e-05,
      "loss": 0.0651,
      "step": 10990
    },
    {
      "epoch": 1.0480880737304688e-05,
      "model_forward_time": 5.590153455734253,
      "step": 10990
    },
    {
      "dynamics_loss_avg": 0.06192174982279539,
      "epoch": 1.0480880737304688e-05,
      "step": 10990
    },
    {
      "action_loss_avg": 0.008656546659767627,
      "epoch": 1.0480880737304688e-05,
      "step": 10990
    },
    {
      "epoch": 1.0480880737304688e-05,
      "step": 10990,
      "training_step_time": 17.024410247802734
    },
    {
      "epoch": 1.0481834411621093e-05,
      "model_forward_time": 5.583863735198975,
      "step": 10991
    },
    {
      "epoch": 1.0481834411621093e-05,
      "step": 10991,
      "training_step_time": 17.035684823989868
    },
    {
      "epoch": 1.04827880859375e-05,
      "model_forward_time": 5.601755142211914,
      "step": 10992
    },
    {
      "epoch": 1.04827880859375e-05,
      "step": 10992,
      "training_step_time": 17.086209535598755
    },
    {
      "epoch": 1.0483741760253907e-05,
      "model_forward_time": 5.992692470550537,
      "step": 10993
    },
    {
      "epoch": 1.0483741760253907e-05,
      "step": 10993,
      "training_step_time": 17.51723051071167
    },
    {
      "epoch": 1.0484695434570312e-05,
      "model_forward_time": 5.635217666625977,
      "step": 10994
    },
    {
      "epoch": 1.0484695434570312e-05,
      "step": 10994,
      "training_step_time": 17.16293239593506
    },
    {
      "epoch": 1.0485649108886718e-05,
      "model_forward_time": 5.624480247497559,
      "step": 10995
    },
    {
      "epoch": 1.0485649108886718e-05,
      "step": 10995,
      "training_step_time": 17.109978675842285
    },
    {
      "epoch": 1.0486602783203125e-05,
      "model_forward_time": 5.595865488052368,
      "step": 10996
    },
    {
      "epoch": 1.0486602783203125e-05,
      "step": 10996,
      "training_step_time": 17.009655237197876
    },
    {
      "epoch": 1.0487556457519531e-05,
      "model_forward_time": 5.590271711349487,
      "step": 10997
    },
    {
      "epoch": 1.0487556457519531e-05,
      "step": 10997,
      "training_step_time": 17.00091314315796
    },
    {
      "epoch": 1.0488510131835937e-05,
      "model_forward_time": 5.9415788650512695,
      "step": 10998
    },
    {
      "epoch": 1.0488510131835937e-05,
      "step": 10998,
      "training_step_time": 17.377259731292725
    },
    {
      "epoch": 1.0489463806152344e-05,
      "model_forward_time": 5.598583221435547,
      "step": 10999
    },
    {
      "epoch": 1.0489463806152344e-05,
      "step": 10999,
      "training_step_time": 17.044466495513916
    },
    {
      "epoch": 1.049041748046875e-05,
      "grad_norm": 1.1182879209518433,
      "learning_rate": 1e-05,
      "loss": 0.0713,
      "step": 11000
    },
    {
      "epoch": 1.049041748046875e-05,
      "model_forward_time": 5.651577472686768,
      "step": 11000
    },
    {
      "dynamics_loss_avg": 0.059294745698571204,
      "epoch": 1.049041748046875e-05,
      "step": 11000
    },
    {
      "action_loss_avg": 0.00598478481406346,
      "epoch": 1.049041748046875e-05,
      "step": 11000
    },
    {
      "epoch": 1.049041748046875e-05,
      "step": 11000,
      "training_step_time": 17.127257585525513
    },
    {
      "epoch": 1.0491371154785156e-05,
      "model_forward_time": 5.6622655391693115,
      "step": 11001
    },
    {
      "epoch": 1.0491371154785156e-05,
      "step": 11001,
      "training_step_time": 17.163180589675903
    },
    {
      "epoch": 1.0492324829101563e-05,
      "model_forward_time": 5.643877744674683,
      "step": 11002
    },
    {
      "epoch": 1.0492324829101563e-05,
      "step": 11002,
      "training_step_time": 17.15596079826355
    },
    {
      "epoch": 1.0493278503417969e-05,
      "model_forward_time": 5.68438196182251,
      "step": 11003
    },
    {
      "epoch": 1.0493278503417969e-05,
      "step": 11003,
      "training_step_time": 17.180165767669678
    },
    {
      "epoch": 1.0494232177734375e-05,
      "model_forward_time": 6.134809970855713,
      "step": 11004
    },
    {
      "epoch": 1.0494232177734375e-05,
      "step": 11004,
      "training_step_time": 17.638121604919434
    },
    {
      "epoch": 1.0495185852050782e-05,
      "model_forward_time": 5.653081178665161,
      "step": 11005
    },
    {
      "epoch": 1.0495185852050782e-05,
      "step": 11005,
      "training_step_time": 17.146501779556274
    },
    {
      "epoch": 1.0496139526367188e-05,
      "model_forward_time": 5.610150575637817,
      "step": 11006
    },
    {
      "epoch": 1.0496139526367188e-05,
      "step": 11006,
      "training_step_time": 17.084662914276123
    },
    {
      "epoch": 1.0497093200683593e-05,
      "model_forward_time": 5.616239070892334,
      "step": 11007
    },
    {
      "epoch": 1.0497093200683593e-05,
      "step": 11007,
      "training_step_time": 17.113704442977905
    },
    {
      "epoch": 1.0498046875e-05,
      "model_forward_time": 5.613162040710449,
      "step": 11008
    },
    {
      "epoch": 1.0498046875e-05,
      "step": 11008,
      "training_step_time": 17.121767044067383
    },
    {
      "epoch": 1.0499000549316407e-05,
      "model_forward_time": 5.607711553573608,
      "step": 11009
    },
    {
      "epoch": 1.0499000549316407e-05,
      "step": 11009,
      "training_step_time": 17.525765657424927
    },
    {
      "epoch": 1.0499954223632812e-05,
      "grad_norm": 0.4892207086086273,
      "learning_rate": 1e-05,
      "loss": 0.0535,
      "step": 11010
    },
    {
      "epoch": 1.0499954223632812e-05,
      "model_forward_time": 5.613380193710327,
      "step": 11010
    },
    {
      "dynamics_loss_avg": 0.04600144578143954,
      "epoch": 1.0499954223632812e-05,
      "step": 11010
    },
    {
      "action_loss_avg": 0.005966583685949445,
      "epoch": 1.0499954223632812e-05,
      "step": 11010
    },
    {
      "epoch": 1.0499954223632812e-05,
      "step": 11010,
      "training_step_time": 17.119935512542725
    },
    {
      "epoch": 1.0500907897949218e-05,
      "model_forward_time": 5.624004602432251,
      "step": 11011
    },
    {
      "epoch": 1.0500907897949218e-05,
      "step": 11011,
      "training_step_time": 17.104881525039673
    },
    {
      "epoch": 1.0501861572265625e-05,
      "model_forward_time": 5.5938825607299805,
      "step": 11012
    },
    {
      "epoch": 1.0501861572265625e-05,
      "step": 11012,
      "training_step_time": 17.015485525131226
    },
    {
      "epoch": 1.0502815246582031e-05,
      "model_forward_time": 5.573909282684326,
      "step": 11013
    },
    {
      "epoch": 1.0502815246582031e-05,
      "step": 11013,
      "training_step_time": 17.01299476623535
    },
    {
      "epoch": 1.0503768920898437e-05,
      "model_forward_time": 5.56227445602417,
      "step": 11014
    },
    {
      "epoch": 1.0503768920898437e-05,
      "step": 11014,
      "training_step_time": 16.982291221618652
    },
    {
      "epoch": 1.0504722595214844e-05,
      "model_forward_time": 5.973975896835327,
      "step": 11015
    },
    {
      "epoch": 1.0504722595214844e-05,
      "step": 11015,
      "training_step_time": 17.39085865020752
    },
    {
      "epoch": 1.050567626953125e-05,
      "model_forward_time": 5.585052251815796,
      "step": 11016
    },
    {
      "epoch": 1.050567626953125e-05,
      "step": 11016,
      "training_step_time": 17.001863956451416
    },
    {
      "epoch": 1.0506629943847656e-05,
      "model_forward_time": 5.5806310176849365,
      "step": 11017
    },
    {
      "epoch": 1.0506629943847656e-05,
      "step": 11017,
      "training_step_time": 17.02672266960144
    },
    {
      "epoch": 1.0507583618164063e-05,
      "model_forward_time": 5.585148572921753,
      "step": 11018
    },
    {
      "epoch": 1.0507583618164063e-05,
      "step": 11018,
      "training_step_time": 17.043060064315796
    },
    {
      "epoch": 1.0508537292480469e-05,
      "model_forward_time": 5.588964462280273,
      "step": 11019
    },
    {
      "epoch": 1.0508537292480469e-05,
      "step": 11019,
      "training_step_time": 17.054467916488647
    },
    {
      "epoch": 1.0509490966796875e-05,
      "grad_norm": 1.0695592164993286,
      "learning_rate": 1e-05,
      "loss": 0.0649,
      "step": 11020
    },
    {
      "epoch": 1.0509490966796875e-05,
      "model_forward_time": 5.95562744140625,
      "step": 11020
    },
    {
      "dynamics_loss_avg": 0.05829626005142927,
      "epoch": 1.0509490966796875e-05,
      "step": 11020
    },
    {
      "action_loss_avg": 0.008088880637660622,
      "epoch": 1.0509490966796875e-05,
      "step": 11020
    },
    {
      "epoch": 1.0509490966796875e-05,
      "step": 11020,
      "training_step_time": 17.43782067298889
    },
    {
      "epoch": 1.0510444641113282e-05,
      "model_forward_time": 5.613059759140015,
      "step": 11021
    },
    {
      "epoch": 1.0510444641113282e-05,
      "step": 11021,
      "training_step_time": 17.1395001411438
    },
    {
      "epoch": 1.0511398315429688e-05,
      "model_forward_time": 5.616523027420044,
      "step": 11022
    },
    {
      "epoch": 1.0511398315429688e-05,
      "step": 11022,
      "training_step_time": 17.126741886138916
    },
    {
      "epoch": 1.0512351989746093e-05,
      "model_forward_time": 5.620154857635498,
      "step": 11023
    },
    {
      "epoch": 1.0512351989746093e-05,
      "step": 11023,
      "training_step_time": 17.100208520889282
    },
    {
      "epoch": 1.05133056640625e-05,
      "model_forward_time": 5.594265460968018,
      "step": 11024
    },
    {
      "epoch": 1.05133056640625e-05,
      "step": 11024,
      "training_step_time": 17.019882678985596
    },
    {
      "epoch": 1.0514259338378907e-05,
      "model_forward_time": 5.563885927200317,
      "step": 11025
    },
    {
      "epoch": 1.0514259338378907e-05,
      "step": 11025,
      "training_step_time": 16.99481964111328
    },
    {
      "epoch": 1.0515213012695312e-05,
      "model_forward_time": 5.9306111335754395,
      "step": 11026
    },
    {
      "epoch": 1.0515213012695312e-05,
      "step": 11026,
      "training_step_time": 17.34445595741272
    },
    {
      "epoch": 1.0516166687011718e-05,
      "model_forward_time": 5.57239294052124,
      "step": 11027
    },
    {
      "epoch": 1.0516166687011718e-05,
      "step": 11027,
      "training_step_time": 16.979459047317505
    },
    {
      "epoch": 1.0517120361328125e-05,
      "model_forward_time": 5.5802037715911865,
      "step": 11028
    },
    {
      "epoch": 1.0517120361328125e-05,
      "step": 11028,
      "training_step_time": 17.023507833480835
    },
    {
      "epoch": 1.0518074035644531e-05,
      "model_forward_time": 5.569382905960083,
      "step": 11029
    },
    {
      "epoch": 1.0518074035644531e-05,
      "step": 11029,
      "training_step_time": 17.008740425109863
    },
    {
      "epoch": 1.0519027709960937e-05,
      "grad_norm": 0.7349356412887573,
      "learning_rate": 1e-05,
      "loss": 0.0759,
      "step": 11030
    },
    {
      "epoch": 1.0519027709960937e-05,
      "model_forward_time": 5.581902742385864,
      "step": 11030
    },
    {
      "dynamics_loss_avg": 0.06253307946026325,
      "epoch": 1.0519027709960937e-05,
      "step": 11030
    },
    {
      "action_loss_avg": 0.010626574163325131,
      "epoch": 1.0519027709960937e-05,
      "step": 11030
    },
    {
      "epoch": 1.0519027709960937e-05,
      "step": 11030,
      "training_step_time": 16.993637561798096
    },
    {
      "epoch": 1.0519981384277344e-05,
      "model_forward_time": 5.594010591506958,
      "step": 11031
    },
    {
      "epoch": 1.0519981384277344e-05,
      "step": 11031,
      "training_step_time": 17.425601959228516
    },
    {
      "epoch": 1.052093505859375e-05,
      "model_forward_time": 5.603351831436157,
      "step": 11032
    },
    {
      "epoch": 1.052093505859375e-05,
      "step": 11032,
      "training_step_time": 17.08895754814148
    },
    {
      "epoch": 1.0521888732910156e-05,
      "model_forward_time": 5.621601581573486,
      "step": 11033
    },
    {
      "epoch": 1.0521888732910156e-05,
      "step": 11033,
      "training_step_time": 17.13060426712036
    },
    {
      "epoch": 1.0522842407226563e-05,
      "model_forward_time": 5.617343425750732,
      "step": 11034
    },
    {
      "epoch": 1.0522842407226563e-05,
      "step": 11034,
      "training_step_time": 17.130997896194458
    },
    {
      "epoch": 1.0523796081542969e-05,
      "model_forward_time": 5.609486818313599,
      "step": 11035
    },
    {
      "epoch": 1.0523796081542969e-05,
      "step": 11035,
      "training_step_time": 17.10065460205078
    },
    {
      "epoch": 1.0524749755859375e-05,
      "model_forward_time": 5.610825538635254,
      "step": 11036
    },
    {
      "epoch": 1.0524749755859375e-05,
      "step": 11036,
      "training_step_time": 17.091986179351807
    },
    {
      "epoch": 1.0525703430175782e-05,
      "model_forward_time": 5.97674822807312,
      "step": 11037
    },
    {
      "epoch": 1.0525703430175782e-05,
      "step": 11037,
      "training_step_time": 17.45717215538025
    },
    {
      "epoch": 1.0526657104492188e-05,
      "model_forward_time": 5.615897178649902,
      "step": 11038
    },
    {
      "epoch": 1.0526657104492188e-05,
      "step": 11038,
      "training_step_time": 17.10613441467285
    },
    {
      "epoch": 1.0527610778808593e-05,
      "model_forward_time": 5.608356475830078,
      "step": 11039
    },
    {
      "epoch": 1.0527610778808593e-05,
      "step": 11039,
      "training_step_time": 17.08925986289978
    },
    {
      "epoch": 1.0528564453125e-05,
      "grad_norm": 1.205261468887329,
      "learning_rate": 1e-05,
      "loss": 0.0508,
      "step": 11040
    },
    {
      "epoch": 1.0528564453125e-05,
      "model_forward_time": 5.61159086227417,
      "step": 11040
    },
    {
      "dynamics_loss_avg": 0.050120612047612666,
      "epoch": 1.0528564453125e-05,
      "step": 11040
    },
    {
      "action_loss_avg": 0.008582739648409188,
      "epoch": 1.0528564453125e-05,
      "step": 11040
    },
    {
      "epoch": 1.0528564453125e-05,
      "step": 11040,
      "training_step_time": 17.080363988876343
    },
    {
      "epoch": 1.0529518127441407e-05,
      "model_forward_time": 5.589542388916016,
      "step": 11041
    },
    {
      "epoch": 1.0529518127441407e-05,
      "step": 11041,
      "training_step_time": 17.050405025482178
    },
    {
      "epoch": 1.0530471801757812e-05,
      "model_forward_time": 5.938404083251953,
      "step": 11042
    },
    {
      "epoch": 1.0530471801757812e-05,
      "step": 11042,
      "training_step_time": 17.358861207962036
    },
    {
      "epoch": 1.0531425476074218e-05,
      "model_forward_time": 5.594697952270508,
      "step": 11043
    },
    {
      "epoch": 1.0531425476074218e-05,
      "step": 11043,
      "training_step_time": 17.026816368103027
    },
    {
      "epoch": 1.0532379150390625e-05,
      "model_forward_time": 5.589211702346802,
      "step": 11044
    },
    {
      "epoch": 1.0532379150390625e-05,
      "step": 11044,
      "training_step_time": 17.02940607070923
    },
    {
      "epoch": 1.0533332824707031e-05,
      "model_forward_time": 5.5889928340911865,
      "step": 11045
    },
    {
      "epoch": 1.0533332824707031e-05,
      "step": 11045,
      "training_step_time": 17.04957604408264
    },
    {
      "epoch": 1.0534286499023437e-05,
      "model_forward_time": 5.598952531814575,
      "step": 11046
    },
    {
      "epoch": 1.0534286499023437e-05,
      "step": 11046,
      "training_step_time": 17.062036991119385
    },
    {
      "epoch": 1.0535240173339844e-05,
      "model_forward_time": 5.6026291847229,
      "step": 11047
    },
    {
      "epoch": 1.0535240173339844e-05,
      "step": 11047,
      "training_step_time": 17.05654263496399
    },
    {
      "epoch": 1.053619384765625e-05,
      "model_forward_time": 5.9479286670684814,
      "step": 11048
    },
    {
      "epoch": 1.053619384765625e-05,
      "step": 11048,
      "training_step_time": 17.411008834838867
    },
    {
      "epoch": 1.0537147521972656e-05,
      "model_forward_time": 5.6259660720825195,
      "step": 11049
    },
    {
      "epoch": 1.0537147521972656e-05,
      "step": 11049,
      "training_step_time": 17.124788761138916
    },
    {
      "epoch": 1.0538101196289063e-05,
      "grad_norm": 0.8208821415901184,
      "learning_rate": 1e-05,
      "loss": 0.0637,
      "step": 11050
    },
    {
      "epoch": 1.0538101196289063e-05,
      "model_forward_time": 5.62442421913147,
      "step": 11050
    },
    {
      "dynamics_loss_avg": 0.05831255409866572,
      "epoch": 1.0538101196289063e-05,
      "step": 11050
    },
    {
      "action_loss_avg": 0.008623676863498986,
      "epoch": 1.0538101196289063e-05,
      "step": 11050
    },
    {
      "epoch": 1.0538101196289063e-05,
      "step": 11050,
      "training_step_time": 17.144551277160645
    },
    {
      "epoch": 1.0539054870605469e-05,
      "model_forward_time": 5.637170791625977,
      "step": 11051
    },
    {
      "epoch": 1.0539054870605469e-05,
      "step": 11051,
      "training_step_time": 17.134071350097656
    },
    {
      "epoch": 1.0540008544921875e-05,
      "model_forward_time": 5.625349760055542,
      "step": 11052
    },
    {
      "epoch": 1.0540008544921875e-05,
      "step": 11052,
      "training_step_time": 17.1108238697052
    },
    {
      "epoch": 1.0540962219238282e-05,
      "model_forward_time": 5.966104984283447,
      "step": 11053
    },
    {
      "epoch": 1.0540962219238282e-05,
      "step": 11053,
      "training_step_time": 17.43211007118225
    },
    {
      "epoch": 1.0541915893554688e-05,
      "model_forward_time": 5.596861124038696,
      "step": 11054
    },
    {
      "epoch": 1.0541915893554688e-05,
      "step": 11054,
      "training_step_time": 17.03354001045227
    },
    {
      "epoch": 1.0542869567871093e-05,
      "model_forward_time": 5.589894771575928,
      "step": 11055
    },
    {
      "epoch": 1.0542869567871093e-05,
      "step": 11055,
      "training_step_time": 17.029578685760498
    },
    {
      "epoch": 1.05438232421875e-05,
      "model_forward_time": 5.603082180023193,
      "step": 11056
    },
    {
      "epoch": 1.05438232421875e-05,
      "step": 11056,
      "training_step_time": 17.064757823944092
    },
    {
      "epoch": 1.0544776916503907e-05,
      "model_forward_time": 5.606680393218994,
      "step": 11057
    },
    {
      "epoch": 1.0544776916503907e-05,
      "step": 11057,
      "training_step_time": 17.087270259857178
    },
    {
      "epoch": 1.0545730590820312e-05,
      "model_forward_time": 5.6008923053741455,
      "step": 11058
    },
    {
      "epoch": 1.0545730590820312e-05,
      "step": 11058,
      "training_step_time": 17.432531595230103
    },
    {
      "epoch": 1.0546684265136718e-05,
      "model_forward_time": 5.601786136627197,
      "step": 11059
    },
    {
      "epoch": 1.0546684265136718e-05,
      "step": 11059,
      "training_step_time": 17.074712991714478
    },
    {
      "epoch": 1.0547637939453125e-05,
      "grad_norm": 1.1613233089447021,
      "learning_rate": 1e-05,
      "loss": 0.0598,
      "step": 11060
    },
    {
      "epoch": 1.0547637939453125e-05,
      "model_forward_time": 5.600295305252075,
      "step": 11060
    },
    {
      "dynamics_loss_avg": 0.044820203073322774,
      "epoch": 1.0547637939453125e-05,
      "step": 11060
    },
    {
      "action_loss_avg": 0.006265969423111528,
      "epoch": 1.0547637939453125e-05,
      "step": 11060
    },
    {
      "epoch": 1.0547637939453125e-05,
      "step": 11060,
      "training_step_time": 17.04181170463562
    },
    {
      "epoch": 1.0548591613769531e-05,
      "model_forward_time": 5.593887567520142,
      "step": 11061
    },
    {
      "epoch": 1.0548591613769531e-05,
      "step": 11061,
      "training_step_time": 17.052037954330444
    },
    {
      "epoch": 1.0549545288085937e-05,
      "model_forward_time": 5.584806680679321,
      "step": 11062
    },
    {
      "epoch": 1.0549545288085937e-05,
      "step": 11062,
      "training_step_time": 17.01247477531433
    },
    {
      "epoch": 1.0550498962402344e-05,
      "model_forward_time": 5.588009595870972,
      "step": 11063
    },
    {
      "epoch": 1.0550498962402344e-05,
      "step": 11063,
      "training_step_time": 17.04365563392639
    },
    {
      "epoch": 1.055145263671875e-05,
      "model_forward_time": 5.942732095718384,
      "step": 11064
    },
    {
      "epoch": 1.055145263671875e-05,
      "step": 11064,
      "training_step_time": 17.415555238723755
    },
    {
      "epoch": 1.0552406311035156e-05,
      "model_forward_time": 5.612082481384277,
      "step": 11065
    },
    {
      "epoch": 1.0552406311035156e-05,
      "step": 11065,
      "training_step_time": 17.124430179595947
    },
    {
      "epoch": 1.0553359985351563e-05,
      "model_forward_time": 5.620664596557617,
      "step": 11066
    },
    {
      "epoch": 1.0553359985351563e-05,
      "step": 11066,
      "training_step_time": 17.120999574661255
    },
    {
      "epoch": 1.0554313659667969e-05,
      "model_forward_time": 5.611363410949707,
      "step": 11067
    },
    {
      "epoch": 1.0554313659667969e-05,
      "step": 11067,
      "training_step_time": 17.07996416091919
    },
    {
      "epoch": 1.0555267333984375e-05,
      "model_forward_time": 5.584025621414185,
      "step": 11068
    },
    {
      "epoch": 1.0555267333984375e-05,
      "step": 11068,
      "training_step_time": 17.016323566436768
    },
    {
      "epoch": 1.0556221008300782e-05,
      "model_forward_time": 5.967503309249878,
      "step": 11069
    },
    {
      "epoch": 1.0556221008300782e-05,
      "step": 11069,
      "training_step_time": 17.40737533569336
    },
    {
      "epoch": 1.0557174682617188e-05,
      "grad_norm": 1.168769359588623,
      "learning_rate": 1e-05,
      "loss": 0.067,
      "step": 11070
    },
    {
      "epoch": 1.0557174682617188e-05,
      "model_forward_time": 5.595580577850342,
      "step": 11070
    },
    {
      "dynamics_loss_avg": 0.0580366987735033,
      "epoch": 1.0557174682617188e-05,
      "step": 11070
    },
    {
      "action_loss_avg": 0.008687912044115365,
      "epoch": 1.0557174682617188e-05,
      "step": 11070
    },
    {
      "epoch": 1.0557174682617188e-05,
      "step": 11070,
      "training_step_time": 17.07128667831421
    },
    {
      "epoch": 1.0558128356933593e-05,
      "model_forward_time": 5.617807149887085,
      "step": 11071
    },
    {
      "epoch": 1.0558128356933593e-05,
      "step": 11071,
      "training_step_time": 17.11703896522522
    },
    {
      "epoch": 1.055908203125e-05,
      "model_forward_time": 5.611566781997681,
      "step": 11072
    },
    {
      "epoch": 1.055908203125e-05,
      "step": 11072,
      "training_step_time": 17.119442224502563
    },
    {
      "epoch": 1.0560035705566407e-05,
      "model_forward_time": 5.619584321975708,
      "step": 11073
    },
    {
      "epoch": 1.0560035705566407e-05,
      "step": 11073,
      "training_step_time": 17.087620973587036
    },
    {
      "epoch": 1.0560989379882812e-05,
      "model_forward_time": 5.589545965194702,
      "step": 11074
    },
    {
      "epoch": 1.0560989379882812e-05,
      "step": 11074,
      "training_step_time": 17.0294132232666
    },
    {
      "epoch": 1.0561943054199218e-05,
      "model_forward_time": 5.927452564239502,
      "step": 11075
    },
    {
      "epoch": 1.0561943054199218e-05,
      "step": 11075,
      "training_step_time": 17.34701681137085
    },
    {
      "epoch": 1.0562896728515625e-05,
      "model_forward_time": 5.573064088821411,
      "step": 11076
    },
    {
      "epoch": 1.0562896728515625e-05,
      "step": 11076,
      "training_step_time": 16.984598875045776
    },
    {
      "epoch": 1.0563850402832031e-05,
      "model_forward_time": 5.560649633407593,
      "step": 11077
    },
    {
      "epoch": 1.0563850402832031e-05,
      "step": 11077,
      "training_step_time": 16.96999430656433
    },
    {
      "epoch": 1.0564804077148437e-05,
      "model_forward_time": 5.567285060882568,
      "step": 11078
    },
    {
      "epoch": 1.0564804077148437e-05,
      "step": 11078,
      "training_step_time": 17.00683856010437
    },
    {
      "epoch": 1.0565757751464844e-05,
      "model_forward_time": 5.56924843788147,
      "step": 11079
    },
    {
      "epoch": 1.0565757751464844e-05,
      "step": 11079,
      "training_step_time": 16.986511707305908
    },
    {
      "epoch": 1.056671142578125e-05,
      "grad_norm": 1.007129192352295,
      "learning_rate": 1e-05,
      "loss": 0.0671,
      "step": 11080
    },
    {
      "epoch": 1.056671142578125e-05,
      "model_forward_time": 5.954603672027588,
      "step": 11080
    },
    {
      "dynamics_loss_avg": 0.06266740374267102,
      "epoch": 1.056671142578125e-05,
      "step": 11080
    },
    {
      "action_loss_avg": 0.0065647070994600655,
      "epoch": 1.056671142578125e-05,
      "step": 11080
    },
    {
      "epoch": 1.056671142578125e-05,
      "step": 11080,
      "training_step_time": 17.366691827774048
    },
    {
      "epoch": 1.0567665100097656e-05,
      "model_forward_time": 5.592615127563477,
      "step": 11081
    },
    {
      "epoch": 1.0567665100097656e-05,
      "step": 11081,
      "training_step_time": 17.01277995109558
    },
    {
      "epoch": 1.0568618774414063e-05,
      "model_forward_time": 5.590067386627197,
      "step": 11082
    },
    {
      "epoch": 1.0568618774414063e-05,
      "step": 11082,
      "training_step_time": 17.04411768913269
    },
    {
      "epoch": 1.0569572448730469e-05,
      "model_forward_time": 5.610504627227783,
      "step": 11083
    },
    {
      "epoch": 1.0569572448730469e-05,
      "step": 11083,
      "training_step_time": 17.116928339004517
    },
    {
      "epoch": 1.0570526123046875e-05,
      "model_forward_time": 5.6239800453186035,
      "step": 11084
    },
    {
      "epoch": 1.0570526123046875e-05,
      "step": 11084,
      "training_step_time": 17.139415502548218
    },
    {
      "epoch": 1.0571479797363282e-05,
      "model_forward_time": 5.626951217651367,
      "step": 11085
    },
    {
      "epoch": 1.0571479797363282e-05,
      "step": 11085,
      "training_step_time": 17.14384627342224
    },
    {
      "epoch": 1.0572433471679688e-05,
      "model_forward_time": 5.955868721008301,
      "step": 11086
    },
    {
      "epoch": 1.0572433471679688e-05,
      "step": 11086,
      "training_step_time": 17.474853038787842
    },
    {
      "epoch": 1.0573387145996094e-05,
      "model_forward_time": 5.616088390350342,
      "step": 11087
    },
    {
      "epoch": 1.0573387145996094e-05,
      "step": 11087,
      "training_step_time": 17.116904258728027
    },
    {
      "epoch": 1.05743408203125e-05,
      "model_forward_time": 5.606003284454346,
      "step": 11088
    },
    {
      "epoch": 1.05743408203125e-05,
      "step": 11088,
      "training_step_time": 17.086090087890625
    },
    {
      "epoch": 1.0575294494628907e-05,
      "model_forward_time": 5.606111526489258,
      "step": 11089
    },
    {
      "epoch": 1.0575294494628907e-05,
      "step": 11089,
      "training_step_time": 17.057722568511963
    },
    {
      "epoch": 1.0576248168945312e-05,
      "grad_norm": 1.1660292148590088,
      "learning_rate": 1e-05,
      "loss": 0.0653,
      "step": 11090
    },
    {
      "epoch": 1.0576248168945312e-05,
      "model_forward_time": 5.615489482879639,
      "step": 11090
    },
    {
      "dynamics_loss_avg": 0.052874731831252575,
      "epoch": 1.0576248168945312e-05,
      "step": 11090
    },
    {
      "action_loss_avg": 0.00812955186702311,
      "epoch": 1.0576248168945312e-05,
      "step": 11090
    },
    {
      "epoch": 1.0576248168945312e-05,
      "step": 11090,
      "training_step_time": 17.104271411895752
    },
    {
      "epoch": 1.0577201843261718e-05,
      "model_forward_time": 5.602629899978638,
      "step": 11091
    },
    {
      "epoch": 1.0577201843261718e-05,
      "step": 11091,
      "training_step_time": 17.427265644073486
    },
    {
      "epoch": 1.0578155517578126e-05,
      "model_forward_time": 5.595945835113525,
      "step": 11092
    },
    {
      "epoch": 1.0578155517578126e-05,
      "step": 11092,
      "training_step_time": 17.047611474990845
    },
    {
      "epoch": 1.0579109191894531e-05,
      "model_forward_time": 5.593320608139038,
      "step": 11093
    },
    {
      "epoch": 1.0579109191894531e-05,
      "step": 11093,
      "training_step_time": 17.02763080596924
    },
    {
      "epoch": 1.0580062866210937e-05,
      "model_forward_time": 5.590227842330933,
      "step": 11094
    },
    {
      "epoch": 1.0580062866210937e-05,
      "step": 11094,
      "training_step_time": 17.028566360473633
    },
    {
      "epoch": 1.0581016540527344e-05,
      "model_forward_time": 5.591608762741089,
      "step": 11095
    },
    {
      "epoch": 1.0581016540527344e-05,
      "step": 11095,
      "training_step_time": 17.016796112060547
    },
    {
      "epoch": 1.058197021484375e-05,
      "model_forward_time": 5.590220928192139,
      "step": 11096
    },
    {
      "epoch": 1.058197021484375e-05,
      "step": 11096,
      "training_step_time": 17.028754949569702
    },
    {
      "epoch": 1.0582923889160156e-05,
      "model_forward_time": 5.938180446624756,
      "step": 11097
    },
    {
      "epoch": 1.0582923889160156e-05,
      "step": 11097,
      "training_step_time": 17.399778127670288
    },
    {
      "epoch": 1.0583877563476563e-05,
      "model_forward_time": 5.604707956314087,
      "step": 11098
    },
    {
      "epoch": 1.0583877563476563e-05,
      "step": 11098,
      "training_step_time": 17.097334146499634
    },
    {
      "epoch": 1.0584831237792969e-05,
      "model_forward_time": 5.609522342681885,
      "step": 11099
    },
    {
      "epoch": 1.0584831237792969e-05,
      "step": 11099,
      "training_step_time": 17.10334062576294
    },
    {
      "epoch": 1.0585784912109375e-05,
      "grad_norm": 1.143397569656372,
      "learning_rate": 1e-05,
      "loss": 0.0523,
      "step": 11100
    },
    {
      "epoch": 1.0585784912109375e-05,
      "model_forward_time": 5.608327388763428,
      "step": 11100
    },
    {
      "dynamics_loss_avg": 0.055859409645199774,
      "epoch": 1.0585784912109375e-05,
      "step": 11100
    },
    {
      "action_loss_avg": 0.008237762097269296,
      "epoch": 1.0585784912109375e-05,
      "step": 11100
    },
    {
      "epoch": 1.0585784912109375e-05,
      "step": 11100,
      "training_step_time": 17.075795888900757
    },
    {
      "epoch": 1.0586738586425782e-05,
      "model_forward_time": 5.599596738815308,
      "step": 11101
    },
    {
      "epoch": 1.0586738586425782e-05,
      "step": 11101,
      "training_step_time": 17.032598733901978
    },
    {
      "epoch": 1.0587692260742188e-05,
      "model_forward_time": 5.975675106048584,
      "step": 11102
    },
    {
      "epoch": 1.0587692260742188e-05,
      "step": 11102,
      "training_step_time": 17.4193434715271
    },
    {
      "epoch": 1.0588645935058594e-05,
      "model_forward_time": 5.584604978561401,
      "step": 11103
    },
    {
      "epoch": 1.0588645935058594e-05,
      "step": 11103,
      "training_step_time": 17.012187719345093
    },
    {
      "epoch": 1.0589599609375e-05,
      "model_forward_time": 5.591245889663696,
      "step": 11104
    },
    {
      "epoch": 1.0589599609375e-05,
      "step": 11104,
      "training_step_time": 17.053253889083862
    },
    {
      "epoch": 1.0590553283691407e-05,
      "model_forward_time": 5.61167573928833,
      "step": 11105
    },
    {
      "epoch": 1.0590553283691407e-05,
      "step": 11105,
      "training_step_time": 17.10740613937378
    },
    {
      "epoch": 1.0591506958007812e-05,
      "model_forward_time": 5.607203245162964,
      "step": 11106
    },
    {
      "epoch": 1.0591506958007812e-05,
      "step": 11106,
      "training_step_time": 17.10343098640442
    },
    {
      "epoch": 1.0592460632324218e-05,
      "model_forward_time": 5.614089250564575,
      "step": 11107
    },
    {
      "epoch": 1.0592460632324218e-05,
      "step": 11107,
      "training_step_time": 17.099522352218628
    },
    {
      "epoch": 1.0593414306640626e-05,
      "model_forward_time": 5.96375298500061,
      "step": 11108
    },
    {
      "epoch": 1.0593414306640626e-05,
      "step": 11108,
      "training_step_time": 17.44369125366211
    },
    {
      "epoch": 1.0594367980957031e-05,
      "model_forward_time": 5.60674262046814,
      "step": 11109
    },
    {
      "epoch": 1.0594367980957031e-05,
      "step": 11109,
      "training_step_time": 17.09094738960266
    },
    {
      "epoch": 1.0595321655273437e-05,
      "grad_norm": 2.01934814453125,
      "learning_rate": 1e-05,
      "loss": 0.0788,
      "step": 11110
    },
    {
      "epoch": 1.0595321655273437e-05,
      "model_forward_time": 5.611189603805542,
      "step": 11110
    },
    {
      "dynamics_loss_avg": 0.05747959855943918,
      "epoch": 1.0595321655273437e-05,
      "step": 11110
    },
    {
      "action_loss_avg": 0.010003148973919452,
      "epoch": 1.0595321655273437e-05,
      "step": 11110
    },
    {
      "epoch": 1.0595321655273437e-05,
      "step": 11110,
      "training_step_time": 17.11111807823181
    },
    {
      "epoch": 1.0596275329589844e-05,
      "model_forward_time": 4.288990259170532,
      "step": 11111
    },
    {
      "epoch": 1.0596275329589844e-05,
      "step": 11111,
      "training_step_time": 12.818801403045654
    },
    {
      "epoch": 1.059722900390625e-05,
      "model_forward_time": 5.6054277420043945,
      "step": 11112
    },
    {
      "epoch": 1.059722900390625e-05,
      "step": 11112,
      "training_step_time": 17.05666708946228
    },
    {
      "epoch": 1.0598182678222656e-05,
      "model_forward_time": 5.582850694656372,
      "step": 11113
    },
    {
      "epoch": 1.0598182678222656e-05,
      "step": 11113,
      "training_step_time": 17.37200403213501
    },
    {
      "epoch": 1.0599136352539063e-05,
      "model_forward_time": 5.581345796585083,
      "step": 11114
    },
    {
      "epoch": 1.0599136352539063e-05,
      "step": 11114,
      "training_step_time": 16.996238470077515
    },
    {
      "epoch": 1.0600090026855469e-05,
      "model_forward_time": 5.584471225738525,
      "step": 11115
    },
    {
      "epoch": 1.0600090026855469e-05,
      "step": 11115,
      "training_step_time": 17.016740083694458
    },
    {
      "epoch": 1.0601043701171875e-05,
      "model_forward_time": 5.596920728683472,
      "step": 11116
    },
    {
      "epoch": 1.0601043701171875e-05,
      "step": 11116,
      "training_step_time": 17.073415517807007
    },
    {
      "epoch": 1.0601997375488282e-05,
      "model_forward_time": 5.61401891708374,
      "step": 11117
    },
    {
      "epoch": 1.0601997375488282e-05,
      "step": 11117,
      "training_step_time": 17.114643812179565
    },
    {
      "epoch": 1.0602951049804688e-05,
      "model_forward_time": 5.622913360595703,
      "step": 11118
    },
    {
      "epoch": 1.0602951049804688e-05,
      "step": 11118,
      "training_step_time": 17.1196072101593
    },
    {
      "epoch": 1.0603904724121094e-05,
      "model_forward_time": 5.944030523300171,
      "step": 11119
    },
    {
      "epoch": 1.0603904724121094e-05,
      "step": 11119,
      "training_step_time": 17.395028352737427
    },
    {
      "epoch": 1.06048583984375e-05,
      "grad_norm": 0.7425031661987305,
      "learning_rate": 1e-05,
      "loss": 0.0737,
      "step": 11120
    },
    {
      "epoch": 1.06048583984375e-05,
      "model_forward_time": 5.575766086578369,
      "step": 11120
    },
    {
      "dynamics_loss_avg": 0.06327595412731171,
      "epoch": 1.06048583984375e-05,
      "step": 11120
    },
    {
      "action_loss_avg": 0.011398413777351379,
      "epoch": 1.06048583984375e-05,
      "step": 11120
    },
    {
      "epoch": 1.06048583984375e-05,
      "step": 11120,
      "training_step_time": 17.00823974609375
    },
    {
      "epoch": 1.0605812072753907e-05,
      "model_forward_time": 5.567699432373047,
      "step": 11121
    },
    {
      "epoch": 1.0605812072753907e-05,
      "step": 11121,
      "training_step_time": 16.97931146621704
    },
    {
      "epoch": 1.0606765747070312e-05,
      "model_forward_time": 5.575847864151001,
      "step": 11122
    },
    {
      "epoch": 1.0606765747070312e-05,
      "step": 11122,
      "training_step_time": 16.98245358467102
    },
    {
      "epoch": 1.0607719421386718e-05,
      "model_forward_time": 5.587766170501709,
      "step": 11123
    },
    {
      "epoch": 1.0607719421386718e-05,
      "step": 11123,
      "training_step_time": 17.03233766555786
    },
    {
      "epoch": 1.0608673095703126e-05,
      "model_forward_time": 5.984676122665405,
      "step": 11124
    },
    {
      "epoch": 1.0608673095703126e-05,
      "step": 11124,
      "training_step_time": 17.435283184051514
    },
    {
      "epoch": 1.0609626770019531e-05,
      "model_forward_time": 5.603405237197876,
      "step": 11125
    },
    {
      "epoch": 1.0609626770019531e-05,
      "step": 11125,
      "training_step_time": 17.09871244430542
    },
    {
      "epoch": 1.0610580444335937e-05,
      "model_forward_time": 5.6235785484313965,
      "step": 11126
    },
    {
      "epoch": 1.0610580444335937e-05,
      "step": 11126,
      "training_step_time": 17.120150327682495
    },
    {
      "epoch": 1.0611534118652344e-05,
      "model_forward_time": 5.609247446060181,
      "step": 11127
    },
    {
      "epoch": 1.0611534118652344e-05,
      "step": 11127,
      "training_step_time": 17.074977159500122
    },
    {
      "epoch": 1.061248779296875e-05,
      "model_forward_time": 5.613860130310059,
      "step": 11128
    },
    {
      "epoch": 1.061248779296875e-05,
      "step": 11128,
      "training_step_time": 17.073286294937134
    },
    {
      "epoch": 1.0613441467285156e-05,
      "model_forward_time": 5.591563940048218,
      "step": 11129
    },
    {
      "epoch": 1.0613441467285156e-05,
      "step": 11129,
      "training_step_time": 17.03127646446228
    },
    {
      "epoch": 1.0614395141601563e-05,
      "grad_norm": 1.798188328742981,
      "learning_rate": 1e-05,
      "loss": 0.0869,
      "step": 11130
    },
    {
      "epoch": 1.0614395141601563e-05,
      "model_forward_time": 5.934545278549194,
      "step": 11130
    },
    {
      "dynamics_loss_avg": 0.07806578204035759,
      "epoch": 1.0614395141601563e-05,
      "step": 11130
    },
    {
      "action_loss_avg": 0.011897333897650242,
      "epoch": 1.0614395141601563e-05,
      "step": 11130
    },
    {
      "epoch": 1.0614395141601563e-05,
      "step": 11130,
      "training_step_time": 17.377952814102173
    },
    {
      "epoch": 1.0615348815917969e-05,
      "model_forward_time": 5.598479270935059,
      "step": 11131
    },
    {
      "epoch": 1.0615348815917969e-05,
      "step": 11131,
      "training_step_time": 17.03226113319397
    },
    {
      "epoch": 1.0616302490234375e-05,
      "model_forward_time": 5.619275331497192,
      "step": 11132
    },
    {
      "epoch": 1.0616302490234375e-05,
      "step": 11132,
      "training_step_time": 17.13684320449829
    },
    {
      "epoch": 1.061725616455078e-05,
      "model_forward_time": 5.6349570751190186,
      "step": 11133
    },
    {
      "epoch": 1.061725616455078e-05,
      "step": 11133,
      "training_step_time": 17.1642849445343
    },
    {
      "epoch": 1.0618209838867188e-05,
      "model_forward_time": 5.628035306930542,
      "step": 11134
    },
    {
      "epoch": 1.0618209838867188e-05,
      "step": 11134,
      "training_step_time": 17.107306003570557
    },
    {
      "epoch": 1.0619163513183594e-05,
      "model_forward_time": 5.965815544128418,
      "step": 11135
    },
    {
      "epoch": 1.0619163513183594e-05,
      "step": 11135,
      "training_step_time": 17.404499053955078
    },
    {
      "epoch": 1.06201171875e-05,
      "model_forward_time": 5.5733582973480225,
      "step": 11136
    },
    {
      "epoch": 1.06201171875e-05,
      "step": 11136,
      "training_step_time": 16.980705738067627
    },
    {
      "epoch": 1.0621070861816407e-05,
      "model_forward_time": 5.573824644088745,
      "step": 11137
    },
    {
      "epoch": 1.0621070861816407e-05,
      "step": 11137,
      "training_step_time": 16.995075464248657
    },
    {
      "epoch": 1.0622024536132812e-05,
      "model_forward_time": 5.572323322296143,
      "step": 11138
    },
    {
      "epoch": 1.0622024536132812e-05,
      "step": 11138,
      "training_step_time": 16.988158464431763
    },
    {
      "epoch": 1.0622978210449218e-05,
      "model_forward_time": 5.566459655761719,
      "step": 11139
    },
    {
      "epoch": 1.0622978210449218e-05,
      "step": 11139,
      "training_step_time": 17.014107942581177
    },
    {
      "epoch": 1.0623931884765626e-05,
      "grad_norm": 1.400433897972107,
      "learning_rate": 1e-05,
      "loss": 0.0594,
      "step": 11140
    },
    {
      "epoch": 1.0623931884765626e-05,
      "model_forward_time": 5.579261064529419,
      "step": 11140
    },
    {
      "dynamics_loss_avg": 0.047314022481441495,
      "epoch": 1.0623931884765626e-05,
      "step": 11140
    },
    {
      "action_loss_avg": 0.007742567360401154,
      "epoch": 1.0623931884765626e-05,
      "step": 11140
    },
    {
      "epoch": 1.0623931884765626e-05,
      "step": 11140,
      "training_step_time": 17.354867458343506
    },
    {
      "epoch": 1.0624885559082031e-05,
      "model_forward_time": 5.589196443557739,
      "step": 11141
    },
    {
      "epoch": 1.0624885559082031e-05,
      "step": 11141,
      "training_step_time": 17.016380786895752
    },
    {
      "epoch": 1.0625839233398437e-05,
      "model_forward_time": 5.575798988342285,
      "step": 11142
    },
    {
      "epoch": 1.0625839233398437e-05,
      "step": 11142,
      "training_step_time": 17.010555028915405
    },
    {
      "epoch": 1.0626792907714844e-05,
      "model_forward_time": 5.592022657394409,
      "step": 11143
    },
    {
      "epoch": 1.0626792907714844e-05,
      "step": 11143,
      "training_step_time": 17.028179168701172
    },
    {
      "epoch": 1.062774658203125e-05,
      "model_forward_time": 5.600144624710083,
      "step": 11144
    },
    {
      "epoch": 1.062774658203125e-05,
      "step": 11144,
      "training_step_time": 17.06718945503235
    },
    {
      "epoch": 1.0628700256347656e-05,
      "model_forward_time": 5.599598169326782,
      "step": 11145
    },
    {
      "epoch": 1.0628700256347656e-05,
      "step": 11145,
      "training_step_time": 17.059932947158813
    },
    {
      "epoch": 1.0629653930664063e-05,
      "model_forward_time": 5.97732138633728,
      "step": 11146
    },
    {
      "epoch": 1.0629653930664063e-05,
      "step": 11146,
      "training_step_time": 17.44449019432068
    },
    {
      "epoch": 1.0630607604980469e-05,
      "model_forward_time": 5.614189863204956,
      "step": 11147
    },
    {
      "epoch": 1.0630607604980469e-05,
      "step": 11147,
      "training_step_time": 17.11976909637451
    },
    {
      "epoch": 1.0631561279296875e-05,
      "model_forward_time": 5.613641262054443,
      "step": 11148
    },
    {
      "epoch": 1.0631561279296875e-05,
      "step": 11148,
      "training_step_time": 17.104265689849854
    },
    {
      "epoch": 1.063251495361328e-05,
      "model_forward_time": 5.637570858001709,
      "step": 11149
    },
    {
      "epoch": 1.063251495361328e-05,
      "step": 11149,
      "training_step_time": 17.144355535507202
    },
    {
      "epoch": 1.0633468627929688e-05,
      "grad_norm": 0.35099855065345764,
      "learning_rate": 1e-05,
      "loss": 0.0594,
      "step": 11150
    },
    {
      "epoch": 1.0633468627929688e-05,
      "model_forward_time": 5.613722085952759,
      "step": 11150
    },
    {
      "dynamics_loss_avg": 0.05704166051000357,
      "epoch": 1.0633468627929688e-05,
      "step": 11150
    },
    {
      "action_loss_avg": 0.007349583762697875,
      "epoch": 1.0633468627929688e-05,
      "step": 11150
    },
    {
      "epoch": 1.0633468627929688e-05,
      "step": 11150,
      "training_step_time": 17.110774755477905
    },
    {
      "epoch": 1.0634422302246094e-05,
      "model_forward_time": 5.980955123901367,
      "step": 11151
    },
    {
      "epoch": 1.0634422302246094e-05,
      "step": 11151,
      "training_step_time": 17.450439929962158
    },
    {
      "epoch": 1.06353759765625e-05,
      "model_forward_time": 5.593463182449341,
      "step": 11152
    },
    {
      "epoch": 1.06353759765625e-05,
      "step": 11152,
      "training_step_time": 17.038999319076538
    },
    {
      "epoch": 1.0636329650878907e-05,
      "model_forward_time": 5.597577095031738,
      "step": 11153
    },
    {
      "epoch": 1.0636329650878907e-05,
      "step": 11153,
      "training_step_time": 17.034847021102905
    },
    {
      "epoch": 1.0637283325195312e-05,
      "model_forward_time": 5.590995788574219,
      "step": 11154
    },
    {
      "epoch": 1.0637283325195312e-05,
      "step": 11154,
      "training_step_time": 17.024970769882202
    },
    {
      "epoch": 1.0638236999511718e-05,
      "model_forward_time": 5.602043867111206,
      "step": 11155
    },
    {
      "epoch": 1.0638236999511718e-05,
      "step": 11155,
      "training_step_time": 17.07735776901245
    },
    {
      "epoch": 1.0639190673828126e-05,
      "model_forward_time": 5.611605644226074,
      "step": 11156
    },
    {
      "epoch": 1.0639190673828126e-05,
      "step": 11156,
      "training_step_time": 17.099002599716187
    },
    {
      "epoch": 1.0640144348144531e-05,
      "model_forward_time": 5.972633123397827,
      "step": 11157
    },
    {
      "epoch": 1.0640144348144531e-05,
      "step": 11157,
      "training_step_time": 17.436188220977783
    },
    {
      "epoch": 1.0641098022460937e-05,
      "model_forward_time": 5.610365152359009,
      "step": 11158
    },
    {
      "epoch": 1.0641098022460937e-05,
      "step": 11158,
      "training_step_time": 17.058502435684204
    },
    {
      "epoch": 1.0642051696777344e-05,
      "model_forward_time": 5.59647011756897,
      "step": 11159
    },
    {
      "epoch": 1.0642051696777344e-05,
      "step": 11159,
      "training_step_time": 17.04033613204956
    },
    {
      "epoch": 1.064300537109375e-05,
      "grad_norm": 0.2933829128742218,
      "learning_rate": 1e-05,
      "loss": 0.056,
      "step": 11160
    },
    {
      "epoch": 1.064300537109375e-05,
      "model_forward_time": 5.592773675918579,
      "step": 11160
    },
    {
      "dynamics_loss_avg": 0.046773898601531985,
      "epoch": 1.064300537109375e-05,
      "step": 11160
    },
    {
      "action_loss_avg": 0.0074727983679622415,
      "epoch": 1.064300537109375e-05,
      "step": 11160
    },
    {
      "epoch": 1.064300537109375e-05,
      "step": 11160,
      "training_step_time": 17.05704641342163
    },
    {
      "epoch": 1.0643959045410156e-05,
      "model_forward_time": 5.5960564613342285,
      "step": 11161
    },
    {
      "epoch": 1.0643959045410156e-05,
      "step": 11161,
      "training_step_time": 17.085060834884644
    },
    {
      "epoch": 1.0644912719726563e-05,
      "model_forward_time": 5.982305288314819,
      "step": 11162
    },
    {
      "epoch": 1.0644912719726563e-05,
      "step": 11162,
      "training_step_time": 17.46235728263855
    },
    {
      "epoch": 1.0645866394042969e-05,
      "model_forward_time": 5.605701923370361,
      "step": 11163
    },
    {
      "epoch": 1.0645866394042969e-05,
      "step": 11163,
      "training_step_time": 17.08292293548584
    },
    {
      "epoch": 1.0646820068359375e-05,
      "model_forward_time": 5.604225397109985,
      "step": 11164
    },
    {
      "epoch": 1.0646820068359375e-05,
      "step": 11164,
      "training_step_time": 17.0841383934021
    },
    {
      "epoch": 1.064777374267578e-05,
      "model_forward_time": 5.60009765625,
      "step": 11165
    },
    {
      "epoch": 1.064777374267578e-05,
      "step": 11165,
      "training_step_time": 17.072468519210815
    },
    {
      "epoch": 1.0648727416992188e-05,
      "model_forward_time": 5.602216005325317,
      "step": 11166
    },
    {
      "epoch": 1.0648727416992188e-05,
      "step": 11166,
      "training_step_time": 17.08820080757141
    },
    {
      "epoch": 1.0649681091308594e-05,
      "model_forward_time": 5.611536026000977,
      "step": 11167
    },
    {
      "epoch": 1.0649681091308594e-05,
      "step": 11167,
      "training_step_time": 17.097907304763794
    },
    {
      "epoch": 1.0650634765625e-05,
      "model_forward_time": 5.977512359619141,
      "step": 11168
    },
    {
      "epoch": 1.0650634765625e-05,
      "step": 11168,
      "training_step_time": 17.474008798599243
    },
    {
      "epoch": 1.0651588439941407e-05,
      "model_forward_time": 5.611694812774658,
      "step": 11169
    },
    {
      "epoch": 1.0651588439941407e-05,
      "step": 11169,
      "training_step_time": 17.104912757873535
    },
    {
      "epoch": 1.0652542114257812e-05,
      "grad_norm": 0.20614904165267944,
      "learning_rate": 1e-05,
      "loss": 0.062,
      "step": 11170
    },
    {
      "epoch": 1.0652542114257812e-05,
      "model_forward_time": 5.6146674156188965,
      "step": 11170
    },
    {
      "dynamics_loss_avg": 0.05337759647518396,
      "epoch": 1.0652542114257812e-05,
      "step": 11170
    },
    {
      "action_loss_avg": 0.007930398709140718,
      "epoch": 1.0652542114257812e-05,
      "step": 11170
    },
    {
      "epoch": 1.0652542114257812e-05,
      "step": 11170,
      "training_step_time": 17.103864908218384
    },
    {
      "epoch": 1.0653495788574218e-05,
      "model_forward_time": 5.618952035903931,
      "step": 11171
    },
    {
      "epoch": 1.0653495788574218e-05,
      "step": 11171,
      "training_step_time": 17.114171504974365
    },
    {
      "epoch": 1.0654449462890626e-05,
      "model_forward_time": 5.610734224319458,
      "step": 11172
    },
    {
      "epoch": 1.0654449462890626e-05,
      "step": 11172,
      "training_step_time": 17.1082124710083
    },
    {
      "epoch": 1.0655403137207031e-05,
      "model_forward_time": 5.9672276973724365,
      "step": 11173
    },
    {
      "epoch": 1.0655403137207031e-05,
      "step": 11173,
      "training_step_time": 17.434439659118652
    },
    {
      "epoch": 1.0656356811523437e-05,
      "model_forward_time": 5.603588104248047,
      "step": 11174
    },
    {
      "epoch": 1.0656356811523437e-05,
      "step": 11174,
      "training_step_time": 17.09481644630432
    },
    {
      "epoch": 1.0657310485839844e-05,
      "model_forward_time": 5.612534284591675,
      "step": 11175
    },
    {
      "epoch": 1.0657310485839844e-05,
      "step": 11175,
      "training_step_time": 17.08217740058899
    },
    {
      "epoch": 1.065826416015625e-05,
      "model_forward_time": 5.597792387008667,
      "step": 11176
    },
    {
      "epoch": 1.065826416015625e-05,
      "step": 11176,
      "training_step_time": 17.04138445854187
    },
    {
      "epoch": 1.0659217834472656e-05,
      "model_forward_time": 5.586657762527466,
      "step": 11177
    },
    {
      "epoch": 1.0659217834472656e-05,
      "step": 11177,
      "training_step_time": 17.020034790039062
    },
    {
      "epoch": 1.0660171508789063e-05,
      "model_forward_time": 5.594509124755859,
      "step": 11178
    },
    {
      "epoch": 1.0660171508789063e-05,
      "step": 11178,
      "training_step_time": 17.419873476028442
    },
    {
      "epoch": 1.0661125183105469e-05,
      "model_forward_time": 5.601710796356201,
      "step": 11179
    },
    {
      "epoch": 1.0661125183105469e-05,
      "step": 11179,
      "training_step_time": 17.077672481536865
    },
    {
      "epoch": 1.0662078857421875e-05,
      "grad_norm": 0.2905416488647461,
      "learning_rate": 1e-05,
      "loss": 0.0732,
      "step": 11180
    },
    {
      "epoch": 1.0662078857421875e-05,
      "model_forward_time": 5.608096122741699,
      "step": 11180
    },
    {
      "dynamics_loss_avg": 0.06501946188509464,
      "epoch": 1.0662078857421875e-05,
      "step": 11180
    },
    {
      "action_loss_avg": 0.00861779081169516,
      "epoch": 1.0662078857421875e-05,
      "step": 11180
    },
    {
      "epoch": 1.0662078857421875e-05,
      "step": 11180,
      "training_step_time": 17.070573329925537
    },
    {
      "epoch": 1.066303253173828e-05,
      "model_forward_time": 5.598869562149048,
      "step": 11181
    },
    {
      "epoch": 1.066303253173828e-05,
      "step": 11181,
      "training_step_time": 17.057950735092163
    },
    {
      "epoch": 1.0663986206054688e-05,
      "model_forward_time": 5.598842620849609,
      "step": 11182
    },
    {
      "epoch": 1.0663986206054688e-05,
      "step": 11182,
      "training_step_time": 17.04342269897461
    },
    {
      "epoch": 1.0664939880371094e-05,
      "model_forward_time": 5.594890594482422,
      "step": 11183
    },
    {
      "epoch": 1.0664939880371094e-05,
      "step": 11183,
      "training_step_time": 17.040741682052612
    },
    {
      "epoch": 1.06658935546875e-05,
      "model_forward_time": 5.938445806503296,
      "step": 11184
    },
    {
      "epoch": 1.06658935546875e-05,
      "step": 11184,
      "training_step_time": 17.393565893173218
    },
    {
      "epoch": 1.0666847229003907e-05,
      "model_forward_time": 5.601836442947388,
      "step": 11185
    },
    {
      "epoch": 1.0666847229003907e-05,
      "step": 11185,
      "training_step_time": 17.096016883850098
    },
    {
      "epoch": 1.0667800903320312e-05,
      "model_forward_time": 5.619814395904541,
      "step": 11186
    },
    {
      "epoch": 1.0667800903320312e-05,
      "step": 11186,
      "training_step_time": 17.105216026306152
    },
    {
      "epoch": 1.0668754577636718e-05,
      "model_forward_time": 5.603782653808594,
      "step": 11187
    },
    {
      "epoch": 1.0668754577636718e-05,
      "step": 11187,
      "training_step_time": 17.078073740005493
    },
    {
      "epoch": 1.0669708251953126e-05,
      "model_forward_time": 5.606114864349365,
      "step": 11188
    },
    {
      "epoch": 1.0669708251953126e-05,
      "step": 11188,
      "training_step_time": 17.065511465072632
    },
    {
      "epoch": 1.0670661926269531e-05,
      "model_forward_time": 5.604548931121826,
      "step": 11189
    },
    {
      "epoch": 1.0670661926269531e-05,
      "step": 11189,
      "training_step_time": 17.42495608329773
    },
    {
      "epoch": 1.0671615600585937e-05,
      "grad_norm": 0.3857620656490326,
      "learning_rate": 1e-05,
      "loss": 0.0618,
      "step": 11190
    },
    {
      "epoch": 1.0671615600585937e-05,
      "model_forward_time": 5.605244874954224,
      "step": 11190
    },
    {
      "dynamics_loss_avg": 0.05460230018943548,
      "epoch": 1.0671615600585937e-05,
      "step": 11190
    },
    {
      "action_loss_avg": 0.007632189569994808,
      "epoch": 1.0671615600585937e-05,
      "step": 11190
    },
    {
      "epoch": 1.0671615600585937e-05,
      "step": 11190,
      "training_step_time": 17.085789680480957
    },
    {
      "epoch": 1.0672569274902344e-05,
      "model_forward_time": 5.602569341659546,
      "step": 11191
    },
    {
      "epoch": 1.0672569274902344e-05,
      "step": 11191,
      "training_step_time": 17.0658221244812
    },
    {
      "epoch": 1.067352294921875e-05,
      "model_forward_time": 5.592220067977905,
      "step": 11192
    },
    {
      "epoch": 1.067352294921875e-05,
      "step": 11192,
      "training_step_time": 17.05621075630188
    },
    {
      "epoch": 1.0674476623535156e-05,
      "model_forward_time": 5.609151363372803,
      "step": 11193
    },
    {
      "epoch": 1.0674476623535156e-05,
      "step": 11193,
      "training_step_time": 17.11710524559021
    },
    {
      "epoch": 1.0675430297851563e-05,
      "model_forward_time": 5.630344390869141,
      "step": 11194
    },
    {
      "epoch": 1.0675430297851563e-05,
      "step": 11194,
      "training_step_time": 17.15350103378296
    },
    {
      "epoch": 1.0676383972167969e-05,
      "model_forward_time": 5.968814849853516,
      "step": 11195
    },
    {
      "epoch": 1.0676383972167969e-05,
      "step": 11195,
      "training_step_time": 17.465301275253296
    },
    {
      "epoch": 1.0677337646484375e-05,
      "model_forward_time": 5.6136155128479,
      "step": 11196
    },
    {
      "epoch": 1.0677337646484375e-05,
      "step": 11196,
      "training_step_time": 17.05091166496277
    },
    {
      "epoch": 1.067829132080078e-05,
      "model_forward_time": 5.594738006591797,
      "step": 11197
    },
    {
      "epoch": 1.067829132080078e-05,
      "step": 11197,
      "training_step_time": 17.014779567718506
    },
    {
      "epoch": 1.0679244995117188e-05,
      "model_forward_time": 5.566685438156128,
      "step": 11198
    },
    {
      "epoch": 1.0679244995117188e-05,
      "step": 11198,
      "training_step_time": 16.98128914833069
    },
    {
      "epoch": 1.0680198669433594e-05,
      "model_forward_time": 5.580547332763672,
      "step": 11199
    },
    {
      "epoch": 1.0680198669433594e-05,
      "step": 11199,
      "training_step_time": 17.00479006767273
    },
    {
      "epoch": 1.068115234375e-05,
      "grad_norm": 0.6698305606842041,
      "learning_rate": 1e-05,
      "loss": 0.0695,
      "step": 11200
    },
    {
      "epoch": 1.068115234375e-05,
      "model_forward_time": 5.939844369888306,
      "step": 11200
    },
    {
      "dynamics_loss_avg": 0.06452773362398148,
      "epoch": 1.068115234375e-05,
      "step": 11200
    },
    {
      "action_loss_avg": 0.009285740181803703,
      "epoch": 1.068115234375e-05,
      "step": 11200
    },
    {
      "epoch": 1.068115234375e-05,
      "step": 11200,
      "training_step_time": 17.365745544433594
    },
    {
      "epoch": 1.0682106018066407e-05,
      "model_forward_time": 5.573148488998413,
      "step": 11201
    },
    {
      "epoch": 1.0682106018066407e-05,
      "step": 11201,
      "training_step_time": 16.99178910255432
    },
    {
      "epoch": 1.0683059692382812e-05,
      "model_forward_time": 5.573396682739258,
      "step": 11202
    },
    {
      "epoch": 1.0683059692382812e-05,
      "step": 11202,
      "training_step_time": 16.99637222290039
    },
    {
      "epoch": 1.0684013366699218e-05,
      "model_forward_time": 5.587466478347778,
      "step": 11203
    },
    {
      "epoch": 1.0684013366699218e-05,
      "step": 11203,
      "training_step_time": 17.040985107421875
    },
    {
      "epoch": 1.0684967041015626e-05,
      "model_forward_time": 5.580000638961792,
      "step": 11204
    },
    {
      "epoch": 1.0684967041015626e-05,
      "step": 11204,
      "training_step_time": 17.018234968185425
    },
    {
      "epoch": 1.0685920715332031e-05,
      "model_forward_time": 5.590931415557861,
      "step": 11205
    },
    {
      "epoch": 1.0685920715332031e-05,
      "step": 11205,
      "training_step_time": 17.0553560256958
    },
    {
      "epoch": 1.0686874389648437e-05,
      "model_forward_time": 5.960624694824219,
      "step": 11206
    },
    {
      "epoch": 1.0686874389648437e-05,
      "step": 11206,
      "training_step_time": 17.46042037010193
    },
    {
      "epoch": 1.0687828063964844e-05,
      "model_forward_time": 5.616773843765259,
      "step": 11207
    },
    {
      "epoch": 1.0687828063964844e-05,
      "step": 11207,
      "training_step_time": 17.09389901161194
    },
    {
      "epoch": 1.068878173828125e-05,
      "model_forward_time": 5.597901344299316,
      "step": 11208
    },
    {
      "epoch": 1.068878173828125e-05,
      "step": 11208,
      "training_step_time": 17.083852767944336
    },
    {
      "epoch": 1.0689735412597656e-05,
      "model_forward_time": 5.6108927726745605,
      "step": 11209
    },
    {
      "epoch": 1.0689735412597656e-05,
      "step": 11209,
      "training_step_time": 17.102744340896606
    },
    {
      "epoch": 1.0690689086914063e-05,
      "grad_norm": 1.2772117853164673,
      "learning_rate": 1e-05,
      "loss": 0.0713,
      "step": 11210
    },
    {
      "epoch": 1.0690689086914063e-05,
      "model_forward_time": 5.613389253616333,
      "step": 11210
    },
    {
      "dynamics_loss_avg": 0.05929490197449923,
      "epoch": 1.0690689086914063e-05,
      "step": 11210
    },
    {
      "action_loss_avg": 0.006150334840640426,
      "epoch": 1.0690689086914063e-05,
      "step": 11210
    },
    {
      "epoch": 1.0690689086914063e-05,
      "step": 11210,
      "training_step_time": 17.104147911071777
    },
    {
      "epoch": 1.0691642761230469e-05,
      "model_forward_time": 5.979901075363159,
      "step": 11211
    },
    {
      "epoch": 1.0691642761230469e-05,
      "step": 11211,
      "training_step_time": 17.44655466079712
    },
    {
      "epoch": 1.0692596435546875e-05,
      "model_forward_time": 5.60852837562561,
      "step": 11212
    },
    {
      "epoch": 1.0692596435546875e-05,
      "step": 11212,
      "training_step_time": 17.08969497680664
    },
    {
      "epoch": 1.069355010986328e-05,
      "model_forward_time": 5.604491233825684,
      "step": 11213
    },
    {
      "epoch": 1.069355010986328e-05,
      "step": 11213,
      "training_step_time": 17.084449291229248
    },
    {
      "epoch": 1.0694503784179688e-05,
      "model_forward_time": 5.615386486053467,
      "step": 11214
    },
    {
      "epoch": 1.0694503784179688e-05,
      "step": 11214,
      "training_step_time": 17.088895797729492
    },
    {
      "epoch": 1.0695457458496094e-05,
      "model_forward_time": 5.640527963638306,
      "step": 11215
    },
    {
      "epoch": 1.0695457458496094e-05,
      "step": 11215,
      "training_step_time": 17.123401403427124
    },
    {
      "epoch": 1.06964111328125e-05,
      "model_forward_time": 5.606835603713989,
      "step": 11216
    },
    {
      "epoch": 1.06964111328125e-05,
      "step": 11216,
      "training_step_time": 17.092283487319946
    },
    {
      "epoch": 1.0697364807128907e-05,
      "model_forward_time": 5.949018955230713,
      "step": 11217
    },
    {
      "epoch": 1.0697364807128907e-05,
      "step": 11217,
      "training_step_time": 17.429129600524902
    },
    {
      "epoch": 1.0698318481445312e-05,
      "model_forward_time": 5.620833158493042,
      "step": 11218
    },
    {
      "epoch": 1.0698318481445312e-05,
      "step": 11218,
      "training_step_time": 17.089958906173706
    },
    {
      "epoch": 1.0699272155761718e-05,
      "model_forward_time": 5.614215135574341,
      "step": 11219
    },
    {
      "epoch": 1.0699272155761718e-05,
      "step": 11219,
      "training_step_time": 17.098755359649658
    },
    {
      "epoch": 1.0700225830078126e-05,
      "grad_norm": 1.0205010175704956,
      "learning_rate": 1e-05,
      "loss": 0.0522,
      "step": 11220
    },
    {
      "epoch": 1.0700225830078126e-05,
      "model_forward_time": 5.609762191772461,
      "step": 11220
    },
    {
      "dynamics_loss_avg": 0.04931627083569765,
      "epoch": 1.0700225830078126e-05,
      "step": 11220
    },
    {
      "action_loss_avg": 0.00860006925649941,
      "epoch": 1.0700225830078126e-05,
      "step": 11220
    },
    {
      "epoch": 1.0700225830078126e-05,
      "step": 11220,
      "training_step_time": 17.11741280555725
    },
    {
      "epoch": 1.0701179504394531e-05,
      "model_forward_time": 5.635922431945801,
      "step": 11221
    },
    {
      "epoch": 1.0701179504394531e-05,
      "step": 11221,
      "training_step_time": 17.12674045562744
    },
    {
      "epoch": 1.0702133178710937e-05,
      "model_forward_time": 5.994046926498413,
      "step": 11222
    },
    {
      "epoch": 1.0702133178710937e-05,
      "step": 11222,
      "training_step_time": 17.49799609184265
    },
    {
      "epoch": 1.0703086853027344e-05,
      "model_forward_time": 5.619839906692505,
      "step": 11223
    },
    {
      "epoch": 1.0703086853027344e-05,
      "step": 11223,
      "training_step_time": 17.11056923866272
    },
    {
      "epoch": 1.070404052734375e-05,
      "model_forward_time": 5.626709699630737,
      "step": 11224
    },
    {
      "epoch": 1.070404052734375e-05,
      "step": 11224,
      "training_step_time": 17.12409496307373
    },
    {
      "epoch": 1.0704994201660156e-05,
      "model_forward_time": 5.6161558628082275,
      "step": 11225
    },
    {
      "epoch": 1.0704994201660156e-05,
      "step": 11225,
      "training_step_time": 17.118220567703247
    },
    {
      "epoch": 1.0705947875976563e-05,
      "model_forward_time": 5.614089250564575,
      "step": 11226
    },
    {
      "epoch": 1.0705947875976563e-05,
      "step": 11226,
      "training_step_time": 17.08233094215393
    },
    {
      "epoch": 1.0706901550292969e-05,
      "model_forward_time": 5.946450710296631,
      "step": 11227
    },
    {
      "epoch": 1.0706901550292969e-05,
      "step": 11227,
      "training_step_time": 17.384265899658203
    },
    {
      "epoch": 1.0707855224609375e-05,
      "model_forward_time": 5.581287384033203,
      "step": 11228
    },
    {
      "epoch": 1.0707855224609375e-05,
      "step": 11228,
      "training_step_time": 16.98848032951355
    },
    {
      "epoch": 1.070880889892578e-05,
      "model_forward_time": 5.578277111053467,
      "step": 11229
    },
    {
      "epoch": 1.070880889892578e-05,
      "step": 11229,
      "training_step_time": 16.998712301254272
    },
    {
      "epoch": 1.0709762573242188e-05,
      "grad_norm": 0.8318625688552856,
      "learning_rate": 1e-05,
      "loss": 0.0809,
      "step": 11230
    },
    {
      "epoch": 1.0709762573242188e-05,
      "model_forward_time": 5.575747489929199,
      "step": 11230
    },
    {
      "dynamics_loss_avg": 0.06426943466067314,
      "epoch": 1.0709762573242188e-05,
      "step": 11230
    },
    {
      "action_loss_avg": 0.008350214152596892,
      "epoch": 1.0709762573242188e-05,
      "step": 11230
    },
    {
      "epoch": 1.0709762573242188e-05,
      "step": 11230,
      "training_step_time": 17.023311138153076
    },
    {
      "epoch": 1.0710716247558594e-05,
      "model_forward_time": 5.580498218536377,
      "step": 11231
    },
    {
      "epoch": 1.0710716247558594e-05,
      "step": 11231,
      "training_step_time": 16.997689962387085
    },
    {
      "epoch": 1.0711669921875e-05,
      "model_forward_time": 5.5814368724823,
      "step": 11232
    },
    {
      "epoch": 1.0711669921875e-05,
      "step": 11232,
      "training_step_time": 17.011704206466675
    },
    {
      "epoch": 1.0712623596191407e-05,
      "model_forward_time": 5.9485955238342285,
      "step": 11233
    },
    {
      "epoch": 1.0712623596191407e-05,
      "step": 11233,
      "training_step_time": 17.381852626800537
    },
    {
      "epoch": 1.0713577270507812e-05,
      "model_forward_time": 5.588974475860596,
      "step": 11234
    },
    {
      "epoch": 1.0713577270507812e-05,
      "step": 11234,
      "training_step_time": 17.038174629211426
    },
    {
      "epoch": 1.0714530944824218e-05,
      "model_forward_time": 5.603832006454468,
      "step": 11235
    },
    {
      "epoch": 1.0714530944824218e-05,
      "step": 11235,
      "training_step_time": 17.083847761154175
    },
    {
      "epoch": 1.0715484619140626e-05,
      "model_forward_time": 5.605791091918945,
      "step": 11236
    },
    {
      "epoch": 1.0715484619140626e-05,
      "step": 11236,
      "training_step_time": 17.098729610443115
    },
    {
      "epoch": 1.0716438293457031e-05,
      "model_forward_time": 5.621893882751465,
      "step": 11237
    },
    {
      "epoch": 1.0716438293457031e-05,
      "step": 11237,
      "training_step_time": 17.09286093711853
    },
    {
      "epoch": 1.0717391967773437e-05,
      "model_forward_time": 5.941894054412842,
      "step": 11238
    },
    {
      "epoch": 1.0717391967773437e-05,
      "step": 11238,
      "training_step_time": 17.380348443984985
    },
    {
      "epoch": 1.0718345642089844e-05,
      "model_forward_time": 5.575193643569946,
      "step": 11239
    },
    {
      "epoch": 1.0718345642089844e-05,
      "step": 11239,
      "training_step_time": 16.98945689201355
    },
    {
      "epoch": 1.071929931640625e-05,
      "grad_norm": 0.6007328033447266,
      "learning_rate": 1e-05,
      "loss": 0.0753,
      "step": 11240
    },
    {
      "epoch": 1.071929931640625e-05,
      "model_forward_time": 5.568749666213989,
      "step": 11240
    },
    {
      "dynamics_loss_avg": 0.07427851259708404,
      "epoch": 1.071929931640625e-05,
      "step": 11240
    },
    {
      "action_loss_avg": 0.010156457824632526,
      "epoch": 1.071929931640625e-05,
      "step": 11240
    },
    {
      "epoch": 1.071929931640625e-05,
      "step": 11240,
      "training_step_time": 16.988531351089478
    },
    {
      "epoch": 1.0720252990722656e-05,
      "model_forward_time": 5.582762002944946,
      "step": 11241
    },
    {
      "epoch": 1.0720252990722656e-05,
      "step": 11241,
      "training_step_time": 17.014588117599487
    },
    {
      "epoch": 1.0721206665039063e-05,
      "model_forward_time": 5.582568407058716,
      "step": 11242
    },
    {
      "epoch": 1.0721206665039063e-05,
      "step": 11242,
      "training_step_time": 17.00361132621765
    },
    {
      "epoch": 1.0722160339355469e-05,
      "model_forward_time": 5.586643934249878,
      "step": 11243
    },
    {
      "epoch": 1.0722160339355469e-05,
      "step": 11243,
      "training_step_time": 17.369890451431274
    },
    {
      "epoch": 1.0723114013671875e-05,
      "model_forward_time": 5.5783984661102295,
      "step": 11244
    },
    {
      "epoch": 1.0723114013671875e-05,
      "step": 11244,
      "training_step_time": 17.003005981445312
    },
    {
      "epoch": 1.072406768798828e-05,
      "model_forward_time": 5.58667778968811,
      "step": 11245
    },
    {
      "epoch": 1.072406768798828e-05,
      "step": 11245,
      "training_step_time": 17.055787801742554
    },
    {
      "epoch": 1.0725021362304688e-05,
      "model_forward_time": 5.6063573360443115,
      "step": 11246
    },
    {
      "epoch": 1.0725021362304688e-05,
      "step": 11246,
      "training_step_time": 17.09721302986145
    },
    {
      "epoch": 1.0725975036621094e-05,
      "model_forward_time": 5.615242958068848,
      "step": 11247
    },
    {
      "epoch": 1.0725975036621094e-05,
      "step": 11247,
      "training_step_time": 17.131684064865112
    },
    {
      "epoch": 1.07269287109375e-05,
      "model_forward_time": 5.619170427322388,
      "step": 11248
    },
    {
      "epoch": 1.07269287109375e-05,
      "step": 11248,
      "training_step_time": 17.131441116333008
    },
    {
      "epoch": 1.0727882385253907e-05,
      "model_forward_time": 5.96974778175354,
      "step": 11249
    },
    {
      "epoch": 1.0727882385253907e-05,
      "step": 11249,
      "training_step_time": 17.460882425308228
    },
    {
      "epoch": 1.0728836059570312e-05,
      "grad_norm": 0.8909705877304077,
      "learning_rate": 1e-05,
      "loss": 0.0747,
      "step": 11250
    },
    {
      "epoch": 1.0728836059570312e-05,
      "model_forward_time": 5.600550889968872,
      "step": 11250
    },
    {
      "dynamics_loss_avg": 0.06374447215348482,
      "epoch": 1.0728836059570312e-05,
      "step": 11250
    },
    {
      "action_loss_avg": 0.008023840840905905,
      "epoch": 1.0728836059570312e-05,
      "step": 11250
    },
    {
      "epoch": 1.0728836059570312e-05,
      "step": 11250,
      "training_step_time": 17.057377815246582
    },
    {
      "epoch": 1.0729789733886718e-05,
      "model_forward_time": 5.587463140487671,
      "step": 11251
    },
    {
      "epoch": 1.0729789733886718e-05,
      "step": 11251,
      "training_step_time": 16.99994993209839
    },
    {
      "epoch": 1.0730743408203126e-05,
      "model_forward_time": 5.585862398147583,
      "step": 11252
    },
    {
      "epoch": 1.0730743408203126e-05,
      "step": 11252,
      "training_step_time": 17.01097273826599
    },
    {
      "epoch": 1.0731697082519531e-05,
      "model_forward_time": 5.58461856842041,
      "step": 11253
    },
    {
      "epoch": 1.0731697082519531e-05,
      "step": 11253,
      "training_step_time": 17.029245376586914
    },
    {
      "epoch": 1.0732650756835937e-05,
      "model_forward_time": 5.977031707763672,
      "step": 11254
    },
    {
      "epoch": 1.0732650756835937e-05,
      "step": 11254,
      "training_step_time": 17.475555658340454
    },
    {
      "epoch": 1.0733604431152344e-05,
      "model_forward_time": 5.614234209060669,
      "step": 11255
    },
    {
      "epoch": 1.0733604431152344e-05,
      "step": 11255,
      "training_step_time": 17.089319229125977
    },
    {
      "epoch": 1.073455810546875e-05,
      "model_forward_time": 5.617857217788696,
      "step": 11256
    },
    {
      "epoch": 1.073455810546875e-05,
      "step": 11256,
      "training_step_time": 17.136879205703735
    },
    {
      "epoch": 1.0735511779785156e-05,
      "model_forward_time": 5.621376037597656,
      "step": 11257
    },
    {
      "epoch": 1.0735511779785156e-05,
      "step": 11257,
      "training_step_time": 17.11924982070923
    },
    {
      "epoch": 1.0736465454101562e-05,
      "model_forward_time": 5.6182639598846436,
      "step": 11258
    },
    {
      "epoch": 1.0736465454101562e-05,
      "step": 11258,
      "training_step_time": 17.097822666168213
    },
    {
      "epoch": 1.0737419128417969e-05,
      "model_forward_time": 5.613227844238281,
      "step": 11259
    },
    {
      "epoch": 1.0737419128417969e-05,
      "step": 11259,
      "training_step_time": 17.080173015594482
    },
    {
      "epoch": 1.0738372802734375e-05,
      "grad_norm": 0.47170138359069824,
      "learning_rate": 1e-05,
      "loss": 0.0617,
      "step": 11260
    },
    {
      "epoch": 1.0738372802734375e-05,
      "model_forward_time": 5.956483602523804,
      "step": 11260
    },
    {
      "dynamics_loss_avg": 0.05394240990281105,
      "epoch": 1.0738372802734375e-05,
      "step": 11260
    },
    {
      "action_loss_avg": 0.006659439019858837,
      "epoch": 1.0738372802734375e-05,
      "step": 11260
    },
    {
      "epoch": 1.0738372802734375e-05,
      "step": 11260,
      "training_step_time": 17.41539764404297
    },
    {
      "epoch": 1.073932647705078e-05,
      "model_forward_time": 5.595178842544556,
      "step": 11261
    },
    {
      "epoch": 1.073932647705078e-05,
      "step": 11261,
      "training_step_time": 17.051022052764893
    },
    {
      "epoch": 1.0740280151367188e-05,
      "model_forward_time": 5.594877243041992,
      "step": 11262
    },
    {
      "epoch": 1.0740280151367188e-05,
      "step": 11262,
      "training_step_time": 17.037806034088135
    },
    {
      "epoch": 1.0741233825683594e-05,
      "model_forward_time": 5.604664325714111,
      "step": 11263
    },
    {
      "epoch": 1.0741233825683594e-05,
      "step": 11263,
      "training_step_time": 17.078460931777954
    },
    {
      "epoch": 1.07421875e-05,
      "model_forward_time": 5.610578298568726,
      "step": 11264
    },
    {
      "epoch": 1.07421875e-05,
      "step": 11264,
      "training_step_time": 17.084363222122192
    },
    {
      "epoch": 1.0743141174316407e-05,
      "model_forward_time": 5.991919040679932,
      "step": 11265
    },
    {
      "epoch": 1.0743141174316407e-05,
      "step": 11265,
      "training_step_time": 17.44292449951172
    },
    {
      "epoch": 1.0744094848632813e-05,
      "model_forward_time": 5.5835349559783936,
      "step": 11266
    },
    {
      "epoch": 1.0744094848632813e-05,
      "step": 11266,
      "training_step_time": 17.005892515182495
    },
    {
      "epoch": 1.0745048522949218e-05,
      "model_forward_time": 5.591072082519531,
      "step": 11267
    },
    {
      "epoch": 1.0745048522949218e-05,
      "step": 11267,
      "training_step_time": 17.03942060470581
    },
    {
      "epoch": 1.0746002197265626e-05,
      "model_forward_time": 5.595718860626221,
      "step": 11268
    },
    {
      "epoch": 1.0746002197265626e-05,
      "step": 11268,
      "training_step_time": 17.033154726028442
    },
    {
      "epoch": 1.0746955871582031e-05,
      "model_forward_time": 5.60054087638855,
      "step": 11269
    },
    {
      "epoch": 1.0746955871582031e-05,
      "step": 11269,
      "training_step_time": 17.08356547355652
    },
    {
      "epoch": 1.0747909545898437e-05,
      "grad_norm": 0.19031214714050293,
      "learning_rate": 1e-05,
      "loss": 0.0528,
      "step": 11270
    },
    {
      "epoch": 1.0747909545898437e-05,
      "model_forward_time": 5.617509365081787,
      "step": 11270
    },
    {
      "dynamics_loss_avg": 0.052212285716086625,
      "epoch": 1.0747909545898437e-05,
      "step": 11270
    },
    {
      "action_loss_avg": 0.005881640233565122,
      "epoch": 1.0747909545898437e-05,
      "step": 11270
    },
    {
      "epoch": 1.0747909545898437e-05,
      "step": 11270,
      "training_step_time": 17.123197317123413
    },
    {
      "epoch": 1.0748863220214845e-05,
      "model_forward_time": 5.975882053375244,
      "step": 11271
    },
    {
      "epoch": 1.0748863220214845e-05,
      "step": 11271,
      "training_step_time": 17.45753788948059
    },
    {
      "epoch": 1.074981689453125e-05,
      "model_forward_time": 5.584100961685181,
      "step": 11272
    },
    {
      "epoch": 1.074981689453125e-05,
      "step": 11272,
      "training_step_time": 17.018951177597046
    },
    {
      "epoch": 1.0750770568847656e-05,
      "model_forward_time": 5.583679676055908,
      "step": 11273
    },
    {
      "epoch": 1.0750770568847656e-05,
      "step": 11273,
      "training_step_time": 16.989761352539062
    },
    {
      "epoch": 1.0751724243164062e-05,
      "model_forward_time": 5.574803113937378,
      "step": 11274
    },
    {
      "epoch": 1.0751724243164062e-05,
      "step": 11274,
      "training_step_time": 16.998151540756226
    },
    {
      "epoch": 1.0752677917480469e-05,
      "model_forward_time": 5.58299708366394,
      "step": 11275
    },
    {
      "epoch": 1.0752677917480469e-05,
      "step": 11275,
      "training_step_time": 17.013842344284058
    },
    {
      "epoch": 1.0753631591796875e-05,
      "model_forward_time": 5.590716361999512,
      "step": 11276
    },
    {
      "epoch": 1.0753631591796875e-05,
      "step": 11276,
      "training_step_time": 17.4062442779541
    },
    {
      "epoch": 1.075458526611328e-05,
      "model_forward_time": 5.600921869277954,
      "step": 11277
    },
    {
      "epoch": 1.075458526611328e-05,
      "step": 11277,
      "training_step_time": 17.074808835983276
    },
    {
      "epoch": 1.0755538940429688e-05,
      "model_forward_time": 5.601070880889893,
      "step": 11278
    },
    {
      "epoch": 1.0755538940429688e-05,
      "step": 11278,
      "training_step_time": 17.069502115249634
    },
    {
      "epoch": 1.0756492614746094e-05,
      "model_forward_time": 5.593655109405518,
      "step": 11279
    },
    {
      "epoch": 1.0756492614746094e-05,
      "step": 11279,
      "training_step_time": 17.046669960021973
    },
    {
      "epoch": 1.07574462890625e-05,
      "grad_norm": 0.38136130571365356,
      "learning_rate": 1e-05,
      "loss": 0.051,
      "step": 11280
    },
    {
      "epoch": 1.07574462890625e-05,
      "model_forward_time": 5.585389614105225,
      "step": 11280
    },
    {
      "dynamics_loss_avg": 0.040149320289492606,
      "epoch": 1.07574462890625e-05,
      "step": 11280
    },
    {
      "action_loss_avg": 0.005161230464000255,
      "epoch": 1.07574462890625e-05,
      "step": 11280
    },
    {
      "epoch": 1.07574462890625e-05,
      "step": 11280,
      "training_step_time": 17.024953365325928
    },
    {
      "epoch": 1.0758399963378907e-05,
      "model_forward_time": 5.589106798171997,
      "step": 11281
    },
    {
      "epoch": 1.0758399963378907e-05,
      "step": 11281,
      "training_step_time": 17.03522825241089
    },
    {
      "epoch": 1.0759353637695313e-05,
      "model_forward_time": 5.941896200180054,
      "step": 11282
    },
    {
      "epoch": 1.0759353637695313e-05,
      "step": 11282,
      "training_step_time": 17.42393684387207
    },
    {
      "epoch": 1.0760307312011718e-05,
      "model_forward_time": 5.606095314025879,
      "step": 11283
    },
    {
      "epoch": 1.0760307312011718e-05,
      "step": 11283,
      "training_step_time": 17.1058030128479
    },
    {
      "epoch": 1.0761260986328126e-05,
      "model_forward_time": 5.613003969192505,
      "step": 11284
    },
    {
      "epoch": 1.0761260986328126e-05,
      "step": 11284,
      "training_step_time": 17.12247633934021
    },
    {
      "epoch": 1.0762214660644531e-05,
      "model_forward_time": 5.623589992523193,
      "step": 11285
    },
    {
      "epoch": 1.0762214660644531e-05,
      "step": 11285,
      "training_step_time": 17.147019624710083
    },
    {
      "epoch": 1.0763168334960937e-05,
      "model_forward_time": 5.630172967910767,
      "step": 11286
    },
    {
      "epoch": 1.0763168334960937e-05,
      "step": 11286,
      "training_step_time": 17.131239652633667
    },
    {
      "epoch": 1.0764122009277345e-05,
      "model_forward_time": 5.997169017791748,
      "step": 11287
    },
    {
      "epoch": 1.0764122009277345e-05,
      "step": 11287,
      "training_step_time": 17.483454704284668
    },
    {
      "epoch": 1.076507568359375e-05,
      "model_forward_time": 5.6177592277526855,
      "step": 11288
    },
    {
      "epoch": 1.076507568359375e-05,
      "step": 11288,
      "training_step_time": 17.110459089279175
    },
    {
      "epoch": 1.0766029357910156e-05,
      "model_forward_time": 5.609189987182617,
      "step": 11289
    },
    {
      "epoch": 1.0766029357910156e-05,
      "step": 11289,
      "training_step_time": 17.096659898757935
    },
    {
      "epoch": 1.0766983032226562e-05,
      "grad_norm": 0.4377562701702118,
      "learning_rate": 1e-05,
      "loss": 0.0752,
      "step": 11290
    },
    {
      "epoch": 1.0766983032226562e-05,
      "model_forward_time": 5.614159107208252,
      "step": 11290
    },
    {
      "dynamics_loss_avg": 0.061571041122078896,
      "epoch": 1.0766983032226562e-05,
      "step": 11290
    },
    {
      "action_loss_avg": 0.012077313428744674,
      "epoch": 1.0766983032226562e-05,
      "step": 11290
    },
    {
      "epoch": 1.0766983032226562e-05,
      "step": 11290,
      "training_step_time": 17.104918241500854
    },
    {
      "epoch": 1.0767936706542969e-05,
      "model_forward_time": 5.62097954750061,
      "step": 11291
    },
    {
      "epoch": 1.0767936706542969e-05,
      "step": 11291,
      "training_step_time": 17.10548210144043
    },
    {
      "epoch": 1.0768890380859375e-05,
      "model_forward_time": 5.607217311859131,
      "step": 11292
    },
    {
      "epoch": 1.0768890380859375e-05,
      "step": 11292,
      "training_step_time": 17.08279848098755
    },
    {
      "epoch": 1.076984405517578e-05,
      "model_forward_time": 5.953909873962402,
      "step": 11293
    },
    {
      "epoch": 1.076984405517578e-05,
      "step": 11293,
      "training_step_time": 17.390971183776855
    },
    {
      "epoch": 1.0770797729492188e-05,
      "model_forward_time": 5.5801122188568115,
      "step": 11294
    },
    {
      "epoch": 1.0770797729492188e-05,
      "step": 11294,
      "training_step_time": 17.010855436325073
    },
    {
      "epoch": 1.0771751403808594e-05,
      "model_forward_time": 5.578103303909302,
      "step": 11295
    },
    {
      "epoch": 1.0771751403808594e-05,
      "step": 11295,
      "training_step_time": 17.041476726531982
    },
    {
      "epoch": 1.0772705078125e-05,
      "model_forward_time": 5.599095344543457,
      "step": 11296
    },
    {
      "epoch": 1.0772705078125e-05,
      "step": 11296,
      "training_step_time": 17.08602786064148
    },
    {
      "epoch": 1.0773658752441407e-05,
      "model_forward_time": 5.617805004119873,
      "step": 11297
    },
    {
      "epoch": 1.0773658752441407e-05,
      "step": 11297,
      "training_step_time": 17.126899003982544
    },
    {
      "epoch": 1.0774612426757813e-05,
      "model_forward_time": 6.004897356033325,
      "step": 11298
    },
    {
      "epoch": 1.0774612426757813e-05,
      "step": 11298,
      "training_step_time": 17.51023554801941
    },
    {
      "epoch": 1.0775566101074218e-05,
      "model_forward_time": 5.611000299453735,
      "step": 11299
    },
    {
      "epoch": 1.0775566101074218e-05,
      "step": 11299,
      "training_step_time": 17.071475744247437
    },
    {
      "epoch": 1.0776519775390626e-05,
      "grad_norm": 0.4516100585460663,
      "learning_rate": 1e-05,
      "loss": 0.0641,
      "step": 11300
    },
    {
      "epoch": 1.0776519775390626e-05,
      "model_forward_time": 5.59179425239563,
      "step": 11300
    },
    {
      "dynamics_loss_avg": 0.05504901222884655,
      "epoch": 1.0776519775390626e-05,
      "step": 11300
    },
    {
      "action_loss_avg": 0.0076036267448216675,
      "epoch": 1.0776519775390626e-05,
      "step": 11300
    },
    {
      "epoch": 1.0776519775390626e-05,
      "step": 11300,
      "training_step_time": 17.019629955291748
    },
    {
      "epoch": 1.0777473449707031e-05,
      "model_forward_time": 5.586891174316406,
      "step": 11301
    },
    {
      "epoch": 1.0777473449707031e-05,
      "step": 11301,
      "training_step_time": 17.02615451812744
    },
    {
      "epoch": 1.0778427124023437e-05,
      "model_forward_time": 5.594534635543823,
      "step": 11302
    },
    {
      "epoch": 1.0778427124023437e-05,
      "step": 11302,
      "training_step_time": 17.081700563430786
    },
    {
      "epoch": 1.0779380798339845e-05,
      "model_forward_time": 5.6159913539886475,
      "step": 11303
    },
    {
      "epoch": 1.0779380798339845e-05,
      "step": 11303,
      "training_step_time": 17.094850540161133
    },
    {
      "epoch": 1.078033447265625e-05,
      "model_forward_time": 5.950963497161865,
      "step": 11304
    },
    {
      "epoch": 1.078033447265625e-05,
      "step": 11304,
      "training_step_time": 17.44012999534607
    },
    {
      "epoch": 1.0781288146972656e-05,
      "model_forward_time": 5.6249918937683105,
      "step": 11305
    },
    {
      "epoch": 1.0781288146972656e-05,
      "step": 11305,
      "training_step_time": 17.118407249450684
    },
    {
      "epoch": 1.0782241821289062e-05,
      "model_forward_time": 5.613253355026245,
      "step": 11306
    },
    {
      "epoch": 1.0782241821289062e-05,
      "step": 11306,
      "training_step_time": 17.078885793685913
    },
    {
      "epoch": 1.0783195495605469e-05,
      "model_forward_time": 5.591435194015503,
      "step": 11307
    },
    {
      "epoch": 1.0783195495605469e-05,
      "step": 11307,
      "training_step_time": 17.01139998435974
    },
    {
      "epoch": 1.0784149169921875e-05,
      "model_forward_time": 5.5710768699646,
      "step": 11308
    },
    {
      "epoch": 1.0784149169921875e-05,
      "step": 11308,
      "training_step_time": 16.983699083328247
    },
    {
      "epoch": 1.078510284423828e-05,
      "model_forward_time": 5.576401710510254,
      "step": 11309
    },
    {
      "epoch": 1.078510284423828e-05,
      "step": 11309,
      "training_step_time": 17.37138867378235
    },
    {
      "epoch": 1.0786056518554688e-05,
      "grad_norm": 1.0575655698776245,
      "learning_rate": 1e-05,
      "loss": 0.0603,
      "step": 11310
    },
    {
      "epoch": 1.0786056518554688e-05,
      "model_forward_time": 5.577393054962158,
      "step": 11310
    },
    {
      "dynamics_loss_avg": 0.06003292053937912,
      "epoch": 1.0786056518554688e-05,
      "step": 11310
    },
    {
      "action_loss_avg": 0.008007536199875176,
      "epoch": 1.0786056518554688e-05,
      "step": 11310
    },
    {
      "epoch": 1.0786056518554688e-05,
      "step": 11310,
      "training_step_time": 17.001630067825317
    },
    {
      "epoch": 1.0787010192871094e-05,
      "model_forward_time": 5.582086086273193,
      "step": 11311
    },
    {
      "epoch": 1.0787010192871094e-05,
      "step": 11311,
      "training_step_time": 16.99456810951233
    },
    {
      "epoch": 1.07879638671875e-05,
      "model_forward_time": 5.582170248031616,
      "step": 11312
    },
    {
      "epoch": 1.07879638671875e-05,
      "step": 11312,
      "training_step_time": 17.00689172744751
    },
    {
      "epoch": 1.0788917541503907e-05,
      "model_forward_time": 5.587257146835327,
      "step": 11313
    },
    {
      "epoch": 1.0788917541503907e-05,
      "step": 11313,
      "training_step_time": 17.083778619766235
    },
    {
      "epoch": 1.0789871215820313e-05,
      "model_forward_time": 5.621806859970093,
      "step": 11314
    },
    {
      "epoch": 1.0789871215820313e-05,
      "step": 11314,
      "training_step_time": 17.14313840866089
    },
    {
      "epoch": 1.0790824890136718e-05,
      "model_forward_time": 5.998078346252441,
      "step": 11315
    },
    {
      "epoch": 1.0790824890136718e-05,
      "step": 11315,
      "training_step_time": 17.50784683227539
    },
    {
      "epoch": 1.0791778564453126e-05,
      "model_forward_time": 5.607205152511597,
      "step": 11316
    },
    {
      "epoch": 1.0791778564453126e-05,
      "step": 11316,
      "training_step_time": 17.10025405883789
    },
    {
      "epoch": 1.0792732238769531e-05,
      "model_forward_time": 5.613351345062256,
      "step": 11317
    },
    {
      "epoch": 1.0792732238769531e-05,
      "step": 11317,
      "training_step_time": 17.10266613960266
    },
    {
      "epoch": 1.0793685913085937e-05,
      "model_forward_time": 5.609359264373779,
      "step": 11318
    },
    {
      "epoch": 1.0793685913085937e-05,
      "step": 11318,
      "training_step_time": 17.09929633140564
    },
    {
      "epoch": 1.0794639587402345e-05,
      "model_forward_time": 5.607346057891846,
      "step": 11319
    },
    {
      "epoch": 1.0794639587402345e-05,
      "step": 11319,
      "training_step_time": 17.069831609725952
    },
    {
      "epoch": 1.079559326171875e-05,
      "grad_norm": 0.8310322165489197,
      "learning_rate": 1e-05,
      "loss": 0.0874,
      "step": 11320
    },
    {
      "epoch": 1.079559326171875e-05,
      "model_forward_time": 5.981420516967773,
      "step": 11320
    },
    {
      "dynamics_loss_avg": 0.0701898492872715,
      "epoch": 1.079559326171875e-05,
      "step": 11320
    },
    {
      "action_loss_avg": 0.011466591665521265,
      "epoch": 1.079559326171875e-05,
      "step": 11320
    },
    {
      "epoch": 1.079559326171875e-05,
      "step": 11320,
      "training_step_time": 17.430854082107544
    },
    {
      "epoch": 1.0796546936035156e-05,
      "model_forward_time": 5.576127767562866,
      "step": 11321
    },
    {
      "epoch": 1.0796546936035156e-05,
      "step": 11321,
      "training_step_time": 17.02424693107605
    },
    {
      "epoch": 1.0797500610351562e-05,
      "model_forward_time": 5.578721761703491,
      "step": 11322
    },
    {
      "epoch": 1.0797500610351562e-05,
      "step": 11322,
      "training_step_time": 16.99666690826416
    },
    {
      "epoch": 1.0798454284667969e-05,
      "model_forward_time": 5.583645820617676,
      "step": 11323
    },
    {
      "epoch": 1.0798454284667969e-05,
      "step": 11323,
      "training_step_time": 17.023839712142944
    },
    {
      "epoch": 1.0799407958984375e-05,
      "model_forward_time": 5.598783254623413,
      "step": 11324
    },
    {
      "epoch": 1.0799407958984375e-05,
      "step": 11324,
      "training_step_time": 17.08984136581421
    },
    {
      "epoch": 1.080036163330078e-05,
      "model_forward_time": 5.613115549087524,
      "step": 11325
    },
    {
      "epoch": 1.080036163330078e-05,
      "step": 11325,
      "training_step_time": 17.103198766708374
    },
    {
      "epoch": 1.0801315307617188e-05,
      "model_forward_time": 5.9401233196258545,
      "step": 11326
    },
    {
      "epoch": 1.0801315307617188e-05,
      "step": 11326,
      "training_step_time": 17.390504598617554
    },
    {
      "epoch": 1.0802268981933594e-05,
      "model_forward_time": 5.596603870391846,
      "step": 11327
    },
    {
      "epoch": 1.0802268981933594e-05,
      "step": 11327,
      "training_step_time": 17.030630350112915
    },
    {
      "epoch": 1.080322265625e-05,
      "model_forward_time": 5.586533546447754,
      "step": 11328
    },
    {
      "epoch": 1.080322265625e-05,
      "step": 11328,
      "training_step_time": 17.039228677749634
    },
    {
      "epoch": 1.0804176330566407e-05,
      "model_forward_time": 5.595370054244995,
      "step": 11329
    },
    {
      "epoch": 1.0804176330566407e-05,
      "step": 11329,
      "training_step_time": 17.019042491912842
    },
    {
      "epoch": 1.0805130004882813e-05,
      "grad_norm": 0.5792567133903503,
      "learning_rate": 1e-05,
      "loss": 0.0642,
      "step": 11330
    },
    {
      "epoch": 1.0805130004882813e-05,
      "model_forward_time": 5.610095024108887,
      "step": 11330
    },
    {
      "dynamics_loss_avg": 0.05329721681773662,
      "epoch": 1.0805130004882813e-05,
      "step": 11330
    },
    {
      "action_loss_avg": 0.007269311137497425,
      "epoch": 1.0805130004882813e-05,
      "step": 11330
    },
    {
      "epoch": 1.0805130004882813e-05,
      "step": 11330,
      "training_step_time": 17.07000756263733
    },
    {
      "epoch": 1.0806083679199218e-05,
      "model_forward_time": 5.979087591171265,
      "step": 11331
    },
    {
      "epoch": 1.0806083679199218e-05,
      "step": 11331,
      "training_step_time": 17.439533233642578
    },
    {
      "epoch": 1.0807037353515626e-05,
      "model_forward_time": 5.606335163116455,
      "step": 11332
    },
    {
      "epoch": 1.0807037353515626e-05,
      "step": 11332,
      "training_step_time": 17.075669765472412
    },
    {
      "epoch": 1.0807991027832031e-05,
      "model_forward_time": 5.60630464553833,
      "step": 11333
    },
    {
      "epoch": 1.0807991027832031e-05,
      "step": 11333,
      "training_step_time": 17.083832263946533
    },
    {
      "epoch": 1.0808944702148437e-05,
      "model_forward_time": 5.599284887313843,
      "step": 11334
    },
    {
      "epoch": 1.0808944702148437e-05,
      "step": 11334,
      "training_step_time": 17.091182470321655
    },
    {
      "epoch": 1.0809898376464845e-05,
      "model_forward_time": 5.613744258880615,
      "step": 11335
    },
    {
      "epoch": 1.0809898376464845e-05,
      "step": 11335,
      "training_step_time": 17.111047506332397
    },
    {
      "epoch": 1.081085205078125e-05,
      "model_forward_time": 5.617952585220337,
      "step": 11336
    },
    {
      "epoch": 1.081085205078125e-05,
      "step": 11336,
      "training_step_time": 17.103084087371826
    },
    {
      "epoch": 1.0811805725097656e-05,
      "model_forward_time": 5.978941440582275,
      "step": 11337
    },
    {
      "epoch": 1.0811805725097656e-05,
      "step": 11337,
      "training_step_time": 17.46796441078186
    },
    {
      "epoch": 1.0812759399414062e-05,
      "model_forward_time": 5.60885214805603,
      "step": 11338
    },
    {
      "epoch": 1.0812759399414062e-05,
      "step": 11338,
      "training_step_time": 17.071902751922607
    },
    {
      "epoch": 1.0813713073730469e-05,
      "model_forward_time": 5.592876672744751,
      "step": 11339
    },
    {
      "epoch": 1.0813713073730469e-05,
      "step": 11339,
      "training_step_time": 17.050659894943237
    },
    {
      "epoch": 1.0814666748046875e-05,
      "grad_norm": 0.37296104431152344,
      "learning_rate": 1e-05,
      "loss": 0.0545,
      "step": 11340
    },
    {
      "epoch": 1.0814666748046875e-05,
      "model_forward_time": 5.591956615447998,
      "step": 11340
    },
    {
      "dynamics_loss_avg": 0.05477562751621008,
      "epoch": 1.0814666748046875e-05,
      "step": 11340
    },
    {
      "action_loss_avg": 0.0060440860223025085,
      "epoch": 1.0814666748046875e-05,
      "step": 11340
    },
    {
      "epoch": 1.0814666748046875e-05,
      "step": 11340,
      "training_step_time": 17.028963327407837
    },
    {
      "epoch": 1.081562042236328e-05,
      "model_forward_time": 5.5901830196380615,
      "step": 11341
    },
    {
      "epoch": 1.081562042236328e-05,
      "step": 11341,
      "training_step_time": 17.040810346603394
    },
    {
      "epoch": 1.0816574096679688e-05,
      "model_forward_time": 5.599745988845825,
      "step": 11342
    },
    {
      "epoch": 1.0816574096679688e-05,
      "step": 11342,
      "training_step_time": 17.41777205467224
    },
    {
      "epoch": 1.0817527770996094e-05,
      "model_forward_time": 5.602656364440918,
      "step": 11343
    },
    {
      "epoch": 1.0817527770996094e-05,
      "step": 11343,
      "training_step_time": 17.06198024749756
    },
    {
      "epoch": 1.08184814453125e-05,
      "model_forward_time": 5.596652507781982,
      "step": 11344
    },
    {
      "epoch": 1.08184814453125e-05,
      "step": 11344,
      "training_step_time": 17.047988414764404
    },
    {
      "epoch": 1.0819435119628907e-05,
      "model_forward_time": 5.593684911727905,
      "step": 11345
    },
    {
      "epoch": 1.0819435119628907e-05,
      "step": 11345,
      "training_step_time": 17.024449348449707
    },
    {
      "epoch": 1.0820388793945313e-05,
      "model_forward_time": 5.587629318237305,
      "step": 11346
    },
    {
      "epoch": 1.0820388793945313e-05,
      "step": 11346,
      "training_step_time": 17.02451515197754
    },
    {
      "epoch": 1.0821342468261718e-05,
      "model_forward_time": 5.600200653076172,
      "step": 11347
    },
    {
      "epoch": 1.0821342468261718e-05,
      "step": 11347,
      "training_step_time": 17.061512231826782
    },
    {
      "epoch": 1.0822296142578126e-05,
      "model_forward_time": 5.951189756393433,
      "step": 11348
    },
    {
      "epoch": 1.0822296142578126e-05,
      "step": 11348,
      "training_step_time": 17.431884765625
    },
    {
      "epoch": 1.0823249816894531e-05,
      "model_forward_time": 5.607114553451538,
      "step": 11349
    },
    {
      "epoch": 1.0823249816894531e-05,
      "step": 11349,
      "training_step_time": 17.097950220108032
    },
    {
      "epoch": 1.0824203491210937e-05,
      "grad_norm": 0.6869427561759949,
      "learning_rate": 1e-05,
      "loss": 0.0577,
      "step": 11350
    },
    {
      "epoch": 1.0824203491210937e-05,
      "model_forward_time": 5.614166975021362,
      "step": 11350
    },
    {
      "dynamics_loss_avg": 0.049003233574330805,
      "epoch": 1.0824203491210937e-05,
      "step": 11350
    },
    {
      "action_loss_avg": 0.00680825951276347,
      "epoch": 1.0824203491210937e-05,
      "step": 11350
    },
    {
      "epoch": 1.0824203491210937e-05,
      "step": 11350,
      "training_step_time": 17.116347789764404
    },
    {
      "epoch": 1.0825157165527345e-05,
      "model_forward_time": 5.613557815551758,
      "step": 11351
    },
    {
      "epoch": 1.0825157165527345e-05,
      "step": 11351,
      "training_step_time": 17.112960815429688
    },
    {
      "epoch": 1.082611083984375e-05,
      "model_forward_time": 5.609942436218262,
      "step": 11352
    },
    {
      "epoch": 1.082611083984375e-05,
      "step": 11352,
      "training_step_time": 17.109621047973633
    },
    {
      "epoch": 1.0827064514160156e-05,
      "model_forward_time": 6.002702474594116,
      "step": 11353
    },
    {
      "epoch": 1.0827064514160156e-05,
      "step": 11353,
      "training_step_time": 17.50868582725525
    },
    {
      "epoch": 1.0828018188476562e-05,
      "model_forward_time": 5.614928245544434,
      "step": 11354
    },
    {
      "epoch": 1.0828018188476562e-05,
      "step": 11354,
      "training_step_time": 17.091878652572632
    },
    {
      "epoch": 1.082897186279297e-05,
      "model_forward_time": 5.587731599807739,
      "step": 11355
    },
    {
      "epoch": 1.082897186279297e-05,
      "step": 11355,
      "training_step_time": 17.022769451141357
    },
    {
      "epoch": 1.0829925537109375e-05,
      "model_forward_time": 5.563528060913086,
      "step": 11356
    },
    {
      "epoch": 1.0829925537109375e-05,
      "step": 11356,
      "training_step_time": 16.97027277946472
    },
    {
      "epoch": 1.083087921142578e-05,
      "model_forward_time": 5.569029808044434,
      "step": 11357
    },
    {
      "epoch": 1.083087921142578e-05,
      "step": 11357,
      "training_step_time": 16.977678537368774
    },
    {
      "epoch": 1.0831832885742188e-05,
      "model_forward_time": 5.5696120262146,
      "step": 11358
    },
    {
      "epoch": 1.0831832885742188e-05,
      "step": 11358,
      "training_step_time": 16.98851251602173
    },
    {
      "epoch": 1.0832786560058594e-05,
      "model_forward_time": 5.910232067108154,
      "step": 11359
    },
    {
      "epoch": 1.0832786560058594e-05,
      "step": 11359,
      "training_step_time": 17.355759620666504
    },
    {
      "epoch": 1.0833740234375e-05,
      "grad_norm": 0.4484405517578125,
      "learning_rate": 1e-05,
      "loss": 0.0678,
      "step": 11360
    },
    {
      "epoch": 1.0833740234375e-05,
      "model_forward_time": 5.573593854904175,
      "step": 11360
    },
    {
      "dynamics_loss_avg": 0.060071174241602424,
      "epoch": 1.0833740234375e-05,
      "step": 11360
    },
    {
      "action_loss_avg": 0.008464835514314473,
      "epoch": 1.0833740234375e-05,
      "step": 11360
    },
    {
      "epoch": 1.0833740234375e-05,
      "step": 11360,
      "training_step_time": 16.994367122650146
    },
    {
      "epoch": 1.0834693908691407e-05,
      "model_forward_time": 5.5869667530059814,
      "step": 11361
    },
    {
      "epoch": 1.0834693908691407e-05,
      "step": 11361,
      "training_step_time": 16.996928215026855
    },
    {
      "epoch": 1.0835647583007813e-05,
      "model_forward_time": 5.580981492996216,
      "step": 11362
    },
    {
      "epoch": 1.0835647583007813e-05,
      "step": 11362,
      "training_step_time": 17.03509211540222
    },
    {
      "epoch": 1.0836601257324218e-05,
      "model_forward_time": 5.589904069900513,
      "step": 11363
    },
    {
      "epoch": 1.0836601257324218e-05,
      "step": 11363,
      "training_step_time": 17.050381422042847
    },
    {
      "epoch": 1.0837554931640626e-05,
      "model_forward_time": 5.99080753326416,
      "step": 11364
    },
    {
      "epoch": 1.0837554931640626e-05,
      "step": 11364,
      "training_step_time": 17.448289394378662
    },
    {
      "epoch": 1.0838508605957031e-05,
      "model_forward_time": 5.601548671722412,
      "step": 11365
    },
    {
      "epoch": 1.0838508605957031e-05,
      "step": 11365,
      "training_step_time": 17.06151533126831
    },
    {
      "epoch": 1.0839462280273437e-05,
      "model_forward_time": 5.598493576049805,
      "step": 11366
    },
    {
      "epoch": 1.0839462280273437e-05,
      "step": 11366,
      "training_step_time": 17.041267156600952
    },
    {
      "epoch": 1.0840415954589845e-05,
      "model_forward_time": 5.585536479949951,
      "step": 11367
    },
    {
      "epoch": 1.0840415954589845e-05,
      "step": 11367,
      "training_step_time": 17.017823696136475
    },
    {
      "epoch": 1.084136962890625e-05,
      "model_forward_time": 4.270268678665161,
      "step": 11368
    },
    {
      "epoch": 1.084136962890625e-05,
      "step": 11368,
      "training_step_time": 12.781776666641235
    },
    {
      "epoch": 1.0842323303222656e-05,
      "model_forward_time": 5.601125955581665,
      "step": 11369
    },
    {
      "epoch": 1.0842323303222656e-05,
      "step": 11369,
      "training_step_time": 17.092334985733032
    },
    {
      "epoch": 1.0843276977539062e-05,
      "grad_norm": 0.9319976568222046,
      "learning_rate": 1e-05,
      "loss": 0.0659,
      "step": 11370
    },
    {
      "epoch": 1.0843276977539062e-05,
      "model_forward_time": 5.980333089828491,
      "step": 11370
    },
    {
      "dynamics_loss_avg": 0.05312685091048479,
      "epoch": 1.0843276977539062e-05,
      "step": 11370
    },
    {
      "action_loss_avg": 0.008769730501808227,
      "epoch": 1.0843276977539062e-05,
      "step": 11370
    },
    {
      "epoch": 1.0843276977539062e-05,
      "step": 11370,
      "training_step_time": 17.498863697052002
    },
    {
      "epoch": 1.084423065185547e-05,
      "model_forward_time": 5.617983102798462,
      "step": 11371
    },
    {
      "epoch": 1.084423065185547e-05,
      "step": 11371,
      "training_step_time": 17.116341829299927
    },
    {
      "epoch": 1.0845184326171875e-05,
      "model_forward_time": 5.621591091156006,
      "step": 11372
    },
    {
      "epoch": 1.0845184326171875e-05,
      "step": 11372,
      "training_step_time": 17.098773956298828
    },
    {
      "epoch": 1.084613800048828e-05,
      "model_forward_time": 5.6020801067352295,
      "step": 11373
    },
    {
      "epoch": 1.084613800048828e-05,
      "step": 11373,
      "training_step_time": 17.090219259262085
    },
    {
      "epoch": 1.0847091674804688e-05,
      "model_forward_time": 5.608767986297607,
      "step": 11374
    },
    {
      "epoch": 1.0847091674804688e-05,
      "step": 11374,
      "training_step_time": 17.111168146133423
    },
    {
      "epoch": 1.0848045349121094e-05,
      "model_forward_time": 5.616994380950928,
      "step": 11375
    },
    {
      "epoch": 1.0848045349121094e-05,
      "step": 11375,
      "training_step_time": 17.463697910308838
    },
    {
      "epoch": 1.08489990234375e-05,
      "model_forward_time": 5.596967935562134,
      "step": 11376
    },
    {
      "epoch": 1.08489990234375e-05,
      "step": 11376,
      "training_step_time": 17.044196367263794
    },
    {
      "epoch": 1.0849952697753907e-05,
      "model_forward_time": 5.593705892562866,
      "step": 11377
    },
    {
      "epoch": 1.0849952697753907e-05,
      "step": 11377,
      "training_step_time": 17.03234314918518
    },
    {
      "epoch": 1.0850906372070313e-05,
      "model_forward_time": 5.601083040237427,
      "step": 11378
    },
    {
      "epoch": 1.0850906372070313e-05,
      "step": 11378,
      "training_step_time": 17.043092012405396
    },
    {
      "epoch": 1.0851860046386718e-05,
      "model_forward_time": 5.591902494430542,
      "step": 11379
    },
    {
      "epoch": 1.0851860046386718e-05,
      "step": 11379,
      "training_step_time": 17.0430588722229
    },
    {
      "epoch": 1.0852813720703126e-05,
      "grad_norm": 0.32110777497291565,
      "learning_rate": 1e-05,
      "loss": 0.0559,
      "step": 11380
    },
    {
      "epoch": 1.0852813720703126e-05,
      "model_forward_time": 5.602003574371338,
      "step": 11380
    },
    {
      "dynamics_loss_avg": 0.05473283343017101,
      "epoch": 1.0852813720703126e-05,
      "step": 11380
    },
    {
      "action_loss_avg": 0.005451614828780293,
      "epoch": 1.0852813720703126e-05,
      "step": 11380
    },
    {
      "epoch": 1.0852813720703126e-05,
      "step": 11380,
      "training_step_time": 17.083221673965454
    },
    {
      "epoch": 1.0853767395019531e-05,
      "model_forward_time": 5.960591554641724,
      "step": 11381
    },
    {
      "epoch": 1.0853767395019531e-05,
      "step": 11381,
      "training_step_time": 17.446521043777466
    },
    {
      "epoch": 1.0854721069335937e-05,
      "model_forward_time": 5.601579427719116,
      "step": 11382
    },
    {
      "epoch": 1.0854721069335937e-05,
      "step": 11382,
      "training_step_time": 17.065637588500977
    },
    {
      "epoch": 1.0855674743652343e-05,
      "model_forward_time": 5.593714237213135,
      "step": 11383
    },
    {
      "epoch": 1.0855674743652343e-05,
      "step": 11383,
      "training_step_time": 17.027102947235107
    },
    {
      "epoch": 1.085662841796875e-05,
      "model_forward_time": 5.598191261291504,
      "step": 11384
    },
    {
      "epoch": 1.085662841796875e-05,
      "step": 11384,
      "training_step_time": 17.063487768173218
    },
    {
      "epoch": 1.0857582092285156e-05,
      "model_forward_time": 5.597070217132568,
      "step": 11385
    },
    {
      "epoch": 1.0857582092285156e-05,
      "step": 11385,
      "training_step_time": 17.06375002861023
    },
    {
      "epoch": 1.0858535766601562e-05,
      "model_forward_time": 5.974311113357544,
      "step": 11386
    },
    {
      "epoch": 1.0858535766601562e-05,
      "step": 11386,
      "training_step_time": 17.43916153907776
    },
    {
      "epoch": 1.085948944091797e-05,
      "model_forward_time": 5.596367359161377,
      "step": 11387
    },
    {
      "epoch": 1.085948944091797e-05,
      "step": 11387,
      "training_step_time": 17.030203104019165
    },
    {
      "epoch": 1.0860443115234375e-05,
      "model_forward_time": 5.596044301986694,
      "step": 11388
    },
    {
      "epoch": 1.0860443115234375e-05,
      "step": 11388,
      "training_step_time": 17.077978134155273
    },
    {
      "epoch": 1.086139678955078e-05,
      "model_forward_time": 5.6100239753723145,
      "step": 11389
    },
    {
      "epoch": 1.086139678955078e-05,
      "step": 11389,
      "training_step_time": 17.105562925338745
    },
    {
      "epoch": 1.0862350463867188e-05,
      "grad_norm": 0.4058774709701538,
      "learning_rate": 1e-05,
      "loss": 0.0515,
      "step": 11390
    },
    {
      "epoch": 1.0862350463867188e-05,
      "model_forward_time": 5.617117404937744,
      "step": 11390
    },
    {
      "dynamics_loss_avg": 0.04175413902848959,
      "epoch": 1.0862350463867188e-05,
      "step": 11390
    },
    {
      "action_loss_avg": 0.006142867077142,
      "epoch": 1.0862350463867188e-05,
      "step": 11390
    },
    {
      "epoch": 1.0862350463867188e-05,
      "step": 11390,
      "training_step_time": 17.12533974647522
    },
    {
      "epoch": 1.0863304138183594e-05,
      "model_forward_time": 5.62120795249939,
      "step": 11391
    },
    {
      "epoch": 1.0863304138183594e-05,
      "step": 11391,
      "training_step_time": 17.120079040527344
    },
    {
      "epoch": 1.08642578125e-05,
      "model_forward_time": 5.972850561141968,
      "step": 11392
    },
    {
      "epoch": 1.08642578125e-05,
      "step": 11392,
      "training_step_time": 17.464284420013428
    },
    {
      "epoch": 1.0865211486816407e-05,
      "model_forward_time": 5.62253999710083,
      "step": 11393
    },
    {
      "epoch": 1.0865211486816407e-05,
      "step": 11393,
      "training_step_time": 17.11305046081543
    },
    {
      "epoch": 1.0866165161132813e-05,
      "model_forward_time": 5.612615346908569,
      "step": 11394
    },
    {
      "epoch": 1.0866165161132813e-05,
      "step": 11394,
      "training_step_time": 17.10881519317627
    },
    {
      "epoch": 1.0867118835449218e-05,
      "model_forward_time": 5.603947162628174,
      "step": 11395
    },
    {
      "epoch": 1.0867118835449218e-05,
      "step": 11395,
      "training_step_time": 17.04157042503357
    },
    {
      "epoch": 1.0868072509765626e-05,
      "model_forward_time": 5.582653045654297,
      "step": 11396
    },
    {
      "epoch": 1.0868072509765626e-05,
      "step": 11396,
      "training_step_time": 17.011438608169556
    },
    {
      "epoch": 1.0869026184082031e-05,
      "model_forward_time": 5.968512535095215,
      "step": 11397
    },
    {
      "epoch": 1.0869026184082031e-05,
      "step": 11397,
      "training_step_time": 17.416663885116577
    },
    {
      "epoch": 1.0869979858398437e-05,
      "model_forward_time": 5.584151983261108,
      "step": 11398
    },
    {
      "epoch": 1.0869979858398437e-05,
      "step": 11398,
      "training_step_time": 17.001003742218018
    },
    {
      "epoch": 1.0870933532714843e-05,
      "model_forward_time": 5.582123756408691,
      "step": 11399
    },
    {
      "epoch": 1.0870933532714843e-05,
      "step": 11399,
      "training_step_time": 17.025475025177002
    },
    {
      "epoch": 1.087188720703125e-05,
      "grad_norm": 0.9321232438087463,
      "learning_rate": 1e-05,
      "loss": 0.069,
      "step": 11400
    },
    {
      "epoch": 1.087188720703125e-05,
      "model_forward_time": 5.609184741973877,
      "step": 11400
    },
    {
      "dynamics_loss_avg": 0.060629957169294355,
      "epoch": 1.087188720703125e-05,
      "step": 11400
    },
    {
      "action_loss_avg": 0.006914188724476844,
      "epoch": 1.087188720703125e-05,
      "step": 11400
    },
    {
      "epoch": 1.087188720703125e-05,
      "step": 11400,
      "training_step_time": 17.088634967803955
    },
    {
      "epoch": 1.0872840881347656e-05,
      "model_forward_time": 5.609671592712402,
      "step": 11401
    },
    {
      "epoch": 1.0872840881347656e-05,
      "step": 11401,
      "training_step_time": 17.110835552215576
    },
    {
      "epoch": 1.0873794555664062e-05,
      "model_forward_time": 5.620422601699829,
      "step": 11402
    },
    {
      "epoch": 1.0873794555664062e-05,
      "step": 11402,
      "training_step_time": 17.12899088859558
    },
    {
      "epoch": 1.087474822998047e-05,
      "model_forward_time": 5.970155239105225,
      "step": 11403
    },
    {
      "epoch": 1.087474822998047e-05,
      "step": 11403,
      "training_step_time": 17.474303007125854
    },
    {
      "epoch": 1.0875701904296875e-05,
      "model_forward_time": 5.63018798828125,
      "step": 11404
    },
    {
      "epoch": 1.0875701904296875e-05,
      "step": 11404,
      "training_step_time": 17.133188009262085
    },
    {
      "epoch": 1.087665557861328e-05,
      "model_forward_time": 5.614349842071533,
      "step": 11405
    },
    {
      "epoch": 1.087665557861328e-05,
      "step": 11405,
      "training_step_time": 17.114328622817993
    },
    {
      "epoch": 1.0877609252929688e-05,
      "model_forward_time": 5.621016502380371,
      "step": 11406
    },
    {
      "epoch": 1.0877609252929688e-05,
      "step": 11406,
      "training_step_time": 17.106337785720825
    },
    {
      "epoch": 1.0878562927246094e-05,
      "model_forward_time": 5.617221355438232,
      "step": 11407
    },
    {
      "epoch": 1.0878562927246094e-05,
      "step": 11407,
      "training_step_time": 17.10596537590027
    },
    {
      "epoch": 1.08795166015625e-05,
      "model_forward_time": 5.60075306892395,
      "step": 11408
    },
    {
      "epoch": 1.08795166015625e-05,
      "step": 11408,
      "training_step_time": 17.40384578704834
    },
    {
      "epoch": 1.0880470275878907e-05,
      "model_forward_time": 5.5839056968688965,
      "step": 11409
    },
    {
      "epoch": 1.0880470275878907e-05,
      "step": 11409,
      "training_step_time": 17.03960657119751
    },
    {
      "epoch": 1.0881423950195313e-05,
      "grad_norm": 0.7153940796852112,
      "learning_rate": 1e-05,
      "loss": 0.0485,
      "step": 11410
    },
    {
      "epoch": 1.0881423950195313e-05,
      "model_forward_time": 5.5979719161987305,
      "step": 11410
    },
    {
      "dynamics_loss_avg": 0.05270636244677007,
      "epoch": 1.0881423950195313e-05,
      "step": 11410
    },
    {
      "action_loss_avg": 0.007434930477757007,
      "epoch": 1.0881423950195313e-05,
      "step": 11410
    },
    {
      "epoch": 1.0881423950195313e-05,
      "step": 11410,
      "training_step_time": 17.062073230743408
    },
    {
      "epoch": 1.0882377624511718e-05,
      "model_forward_time": 5.601700067520142,
      "step": 11411
    },
    {
      "epoch": 1.0882377624511718e-05,
      "step": 11411,
      "training_step_time": 17.065001964569092
    },
    {
      "epoch": 1.0883331298828126e-05,
      "model_forward_time": 5.593843460083008,
      "step": 11412
    },
    {
      "epoch": 1.0883331298828126e-05,
      "step": 11412,
      "training_step_time": 17.05014133453369
    },
    {
      "epoch": 1.0884284973144531e-05,
      "model_forward_time": 5.5940635204315186,
      "step": 11413
    },
    {
      "epoch": 1.0884284973144531e-05,
      "step": 11413,
      "training_step_time": 17.05801796913147
    },
    {
      "epoch": 1.0885238647460937e-05,
      "model_forward_time": 5.956270933151245,
      "step": 11414
    },
    {
      "epoch": 1.0885238647460937e-05,
      "step": 11414,
      "training_step_time": 17.415245532989502
    },
    {
      "epoch": 1.0886192321777343e-05,
      "model_forward_time": 5.596755504608154,
      "step": 11415
    },
    {
      "epoch": 1.0886192321777343e-05,
      "step": 11415,
      "training_step_time": 17.05822515487671
    },
    {
      "epoch": 1.088714599609375e-05,
      "model_forward_time": 5.594149589538574,
      "step": 11416
    },
    {
      "epoch": 1.088714599609375e-05,
      "step": 11416,
      "training_step_time": 17.03213405609131
    },
    {
      "epoch": 1.0888099670410156e-05,
      "model_forward_time": 5.598567724227905,
      "step": 11417
    },
    {
      "epoch": 1.0888099670410156e-05,
      "step": 11417,
      "training_step_time": 17.041180849075317
    },
    {
      "epoch": 1.0889053344726562e-05,
      "model_forward_time": 5.597071647644043,
      "step": 11418
    },
    {
      "epoch": 1.0889053344726562e-05,
      "step": 11418,
      "training_step_time": 16.99722123146057
    },
    {
      "epoch": 1.089000701904297e-05,
      "model_forward_time": 6.009852409362793,
      "step": 11419
    },
    {
      "epoch": 1.089000701904297e-05,
      "step": 11419,
      "training_step_time": 17.505441904067993
    },
    {
      "epoch": 1.0890960693359375e-05,
      "grad_norm": 1.3836889266967773,
      "learning_rate": 1e-05,
      "loss": 0.0905,
      "step": 11420
    },
    {
      "epoch": 1.0890960693359375e-05,
      "model_forward_time": 5.623793363571167,
      "step": 11420
    },
    {
      "dynamics_loss_avg": 0.07323617991060019,
      "epoch": 1.0890960693359375e-05,
      "step": 11420
    },
    {
      "action_loss_avg": 0.009744097734801472,
      "epoch": 1.0890960693359375e-05,
      "step": 11420
    },
    {
      "epoch": 1.0890960693359375e-05,
      "step": 11420,
      "training_step_time": 17.13935160636902
    },
    {
      "epoch": 1.089191436767578e-05,
      "model_forward_time": 5.627002000808716,
      "step": 11421
    },
    {
      "epoch": 1.089191436767578e-05,
      "step": 11421,
      "training_step_time": 17.137904405593872
    },
    {
      "epoch": 1.0892868041992188e-05,
      "model_forward_time": 5.628497362136841,
      "step": 11422
    },
    {
      "epoch": 1.0892868041992188e-05,
      "step": 11422,
      "training_step_time": 17.11031723022461
    },
    {
      "epoch": 1.0893821716308594e-05,
      "model_forward_time": 5.606497764587402,
      "step": 11423
    },
    {
      "epoch": 1.0893821716308594e-05,
      "step": 11423,
      "training_step_time": 17.051717281341553
    },
    {
      "epoch": 1.0894775390625e-05,
      "model_forward_time": 5.589118003845215,
      "step": 11424
    },
    {
      "epoch": 1.0894775390625e-05,
      "step": 11424,
      "training_step_time": 17.043266534805298
    },
    {
      "epoch": 1.0895729064941407e-05,
      "model_forward_time": 5.931866884231567,
      "step": 11425
    },
    {
      "epoch": 1.0895729064941407e-05,
      "step": 11425,
      "training_step_time": 17.405824422836304
    },
    {
      "epoch": 1.0896682739257813e-05,
      "model_forward_time": 5.606120824813843,
      "step": 11426
    },
    {
      "epoch": 1.0896682739257813e-05,
      "step": 11426,
      "training_step_time": 17.105279684066772
    },
    {
      "epoch": 1.0897636413574218e-05,
      "model_forward_time": 5.625475645065308,
      "step": 11427
    },
    {
      "epoch": 1.0897636413574218e-05,
      "step": 11427,
      "training_step_time": 17.131370067596436
    },
    {
      "epoch": 1.0898590087890626e-05,
      "model_forward_time": 5.628264665603638,
      "step": 11428
    },
    {
      "epoch": 1.0898590087890626e-05,
      "step": 11428,
      "training_step_time": 17.139631748199463
    },
    {
      "epoch": 1.0899543762207032e-05,
      "model_forward_time": 5.625231027603149,
      "step": 11429
    },
    {
      "epoch": 1.0899543762207032e-05,
      "step": 11429,
      "training_step_time": 17.12700653076172
    },
    {
      "epoch": 1.0900497436523437e-05,
      "grad_norm": 0.5790239572525024,
      "learning_rate": 1e-05,
      "loss": 0.0685,
      "step": 11430
    },
    {
      "epoch": 1.0900497436523437e-05,
      "model_forward_time": 5.9992921352386475,
      "step": 11430
    },
    {
      "dynamics_loss_avg": 0.059100464545190336,
      "epoch": 1.0900497436523437e-05,
      "step": 11430
    },
    {
      "action_loss_avg": 0.007621410395950079,
      "epoch": 1.0900497436523437e-05,
      "step": 11430
    },
    {
      "epoch": 1.0900497436523437e-05,
      "step": 11430,
      "training_step_time": 17.495371341705322
    },
    {
      "epoch": 1.0901451110839843e-05,
      "model_forward_time": 5.606267690658569,
      "step": 11431
    },
    {
      "epoch": 1.0901451110839843e-05,
      "step": 11431,
      "training_step_time": 17.10135817527771
    },
    {
      "epoch": 1.090240478515625e-05,
      "model_forward_time": 5.613290071487427,
      "step": 11432
    },
    {
      "epoch": 1.090240478515625e-05,
      "step": 11432,
      "training_step_time": 17.113800764083862
    },
    {
      "epoch": 1.0903358459472656e-05,
      "model_forward_time": 5.646501064300537,
      "step": 11433
    },
    {
      "epoch": 1.0903358459472656e-05,
      "step": 11433,
      "training_step_time": 17.164525747299194
    },
    {
      "epoch": 1.0904312133789062e-05,
      "model_forward_time": 5.641679763793945,
      "step": 11434
    },
    {
      "epoch": 1.0904312133789062e-05,
      "step": 11434,
      "training_step_time": 17.151386260986328
    },
    {
      "epoch": 1.090526580810547e-05,
      "model_forward_time": 5.638442039489746,
      "step": 11435
    },
    {
      "epoch": 1.090526580810547e-05,
      "step": 11435,
      "training_step_time": 17.088826417922974
    },
    {
      "epoch": 1.0906219482421875e-05,
      "model_forward_time": 5.946707248687744,
      "step": 11436
    },
    {
      "epoch": 1.0906219482421875e-05,
      "step": 11436,
      "training_step_time": 17.392061948776245
    },
    {
      "epoch": 1.090717315673828e-05,
      "model_forward_time": 5.593418121337891,
      "step": 11437
    },
    {
      "epoch": 1.090717315673828e-05,
      "step": 11437,
      "training_step_time": 17.033236742019653
    },
    {
      "epoch": 1.0908126831054688e-05,
      "model_forward_time": 5.604593992233276,
      "step": 11438
    },
    {
      "epoch": 1.0908126831054688e-05,
      "step": 11438,
      "training_step_time": 17.076092958450317
    },
    {
      "epoch": 1.0909080505371094e-05,
      "model_forward_time": 5.61648964881897,
      "step": 11439
    },
    {
      "epoch": 1.0909080505371094e-05,
      "step": 11439,
      "training_step_time": 17.108222484588623
    },
    {
      "epoch": 1.09100341796875e-05,
      "grad_norm": 0.6444231867790222,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 11440
    },
    {
      "epoch": 1.09100341796875e-05,
      "model_forward_time": 5.627910852432251,
      "step": 11440
    },
    {
      "dynamics_loss_avg": 0.0729547806084156,
      "epoch": 1.09100341796875e-05,
      "step": 11440
    },
    {
      "action_loss_avg": 0.009145964309573174,
      "epoch": 1.09100341796875e-05,
      "step": 11440
    },
    {
      "epoch": 1.09100341796875e-05,
      "step": 11440,
      "training_step_time": 17.142754077911377
    },
    {
      "epoch": 1.0910987854003907e-05,
      "model_forward_time": 5.621453523635864,
      "step": 11441
    },
    {
      "epoch": 1.0910987854003907e-05,
      "step": 11441,
      "training_step_time": 17.47413444519043
    },
    {
      "epoch": 1.0911941528320313e-05,
      "model_forward_time": 5.603304386138916,
      "step": 11442
    },
    {
      "epoch": 1.0911941528320313e-05,
      "step": 11442,
      "training_step_time": 17.046269416809082
    },
    {
      "epoch": 1.0912895202636718e-05,
      "model_forward_time": 5.5854785442352295,
      "step": 11443
    },
    {
      "epoch": 1.0912895202636718e-05,
      "step": 11443,
      "training_step_time": 17.021600484848022
    },
    {
      "epoch": 1.0913848876953126e-05,
      "model_forward_time": 5.596855163574219,
      "step": 11444
    },
    {
      "epoch": 1.0913848876953126e-05,
      "step": 11444,
      "training_step_time": 17.06375551223755
    },
    {
      "epoch": 1.0914802551269532e-05,
      "model_forward_time": 5.622599840164185,
      "step": 11445
    },
    {
      "epoch": 1.0914802551269532e-05,
      "step": 11445,
      "training_step_time": 17.119457960128784
    },
    {
      "epoch": 1.0915756225585937e-05,
      "model_forward_time": 5.61626672744751,
      "step": 11446
    },
    {
      "epoch": 1.0915756225585937e-05,
      "step": 11446,
      "training_step_time": 17.134434938430786
    },
    {
      "epoch": 1.0916709899902343e-05,
      "model_forward_time": 5.981716632843018,
      "step": 11447
    },
    {
      "epoch": 1.0916709899902343e-05,
      "step": 11447,
      "training_step_time": 17.463745594024658
    },
    {
      "epoch": 1.091766357421875e-05,
      "model_forward_time": 5.598376512527466,
      "step": 11448
    },
    {
      "epoch": 1.091766357421875e-05,
      "step": 11448,
      "training_step_time": 17.036882162094116
    },
    {
      "epoch": 1.0918617248535156e-05,
      "model_forward_time": 5.584754705429077,
      "step": 11449
    },
    {
      "epoch": 1.0918617248535156e-05,
      "step": 11449,
      "training_step_time": 17.00338387489319
    },
    {
      "epoch": 1.0919570922851562e-05,
      "grad_norm": 1.12021005153656,
      "learning_rate": 1e-05,
      "loss": 0.0675,
      "step": 11450
    },
    {
      "epoch": 1.0919570922851562e-05,
      "model_forward_time": 5.582083702087402,
      "step": 11450
    },
    {
      "dynamics_loss_avg": 0.06043846346437931,
      "epoch": 1.0919570922851562e-05,
      "step": 11450
    },
    {
      "action_loss_avg": 0.007283155550248921,
      "epoch": 1.0919570922851562e-05,
      "step": 11450
    },
    {
      "epoch": 1.0919570922851562e-05,
      "step": 11450,
      "training_step_time": 17.0233633518219
    },
    {
      "epoch": 1.092052459716797e-05,
      "model_forward_time": 5.584096431732178,
      "step": 11451
    },
    {
      "epoch": 1.092052459716797e-05,
      "step": 11451,
      "training_step_time": 17.046562671661377
    },
    {
      "epoch": 1.0921478271484375e-05,
      "model_forward_time": 5.98839807510376,
      "step": 11452
    },
    {
      "epoch": 1.0921478271484375e-05,
      "step": 11452,
      "training_step_time": 17.47797417640686
    },
    {
      "epoch": 1.092243194580078e-05,
      "model_forward_time": 5.617885589599609,
      "step": 11453
    },
    {
      "epoch": 1.092243194580078e-05,
      "step": 11453,
      "training_step_time": 17.096039533615112
    },
    {
      "epoch": 1.0923385620117188e-05,
      "model_forward_time": 5.603090524673462,
      "step": 11454
    },
    {
      "epoch": 1.0923385620117188e-05,
      "step": 11454,
      "training_step_time": 17.075114250183105
    },
    {
      "epoch": 1.0924339294433594e-05,
      "model_forward_time": 5.576601266860962,
      "step": 11455
    },
    {
      "epoch": 1.0924339294433594e-05,
      "step": 11455,
      "training_step_time": 16.999250173568726
    },
    {
      "epoch": 1.092529296875e-05,
      "model_forward_time": 5.584742069244385,
      "step": 11456
    },
    {
      "epoch": 1.092529296875e-05,
      "step": 11456,
      "training_step_time": 17.019936084747314
    },
    {
      "epoch": 1.0926246643066407e-05,
      "model_forward_time": 5.594327449798584,
      "step": 11457
    },
    {
      "epoch": 1.0926246643066407e-05,
      "step": 11457,
      "training_step_time": 17.052513360977173
    },
    {
      "epoch": 1.0927200317382813e-05,
      "model_forward_time": 5.938751459121704,
      "step": 11458
    },
    {
      "epoch": 1.0927200317382813e-05,
      "step": 11458,
      "training_step_time": 17.393551349639893
    },
    {
      "epoch": 1.0928153991699218e-05,
      "model_forward_time": 5.6056952476501465,
      "step": 11459
    },
    {
      "epoch": 1.0928153991699218e-05,
      "step": 11459,
      "training_step_time": 17.080119371414185
    },
    {
      "epoch": 1.0929107666015626e-05,
      "grad_norm": 0.9389939308166504,
      "learning_rate": 1e-05,
      "loss": 0.0589,
      "step": 11460
    },
    {
      "epoch": 1.0929107666015626e-05,
      "model_forward_time": 5.603126287460327,
      "step": 11460
    },
    {
      "dynamics_loss_avg": 0.053955217450857164,
      "epoch": 1.0929107666015626e-05,
      "step": 11460
    },
    {
      "action_loss_avg": 0.006648530322127044,
      "epoch": 1.0929107666015626e-05,
      "step": 11460
    },
    {
      "epoch": 1.0929107666015626e-05,
      "step": 11460,
      "training_step_time": 17.06084132194519
    },
    {
      "epoch": 1.0930061340332032e-05,
      "model_forward_time": 5.5926477909088135,
      "step": 11461
    },
    {
      "epoch": 1.0930061340332032e-05,
      "step": 11461,
      "training_step_time": 17.02210831642151
    },
    {
      "epoch": 1.0931015014648437e-05,
      "model_forward_time": 5.583050012588501,
      "step": 11462
    },
    {
      "epoch": 1.0931015014648437e-05,
      "step": 11462,
      "training_step_time": 17.034384965896606
    },
    {
      "epoch": 1.0931968688964843e-05,
      "model_forward_time": 5.989628076553345,
      "step": 11463
    },
    {
      "epoch": 1.0931968688964843e-05,
      "step": 11463,
      "training_step_time": 17.48543119430542
    },
    {
      "epoch": 1.093292236328125e-05,
      "model_forward_time": 5.629738092422485,
      "step": 11464
    },
    {
      "epoch": 1.093292236328125e-05,
      "step": 11464,
      "training_step_time": 17.123512029647827
    },
    {
      "epoch": 1.0933876037597656e-05,
      "model_forward_time": 5.607824325561523,
      "step": 11465
    },
    {
      "epoch": 1.0933876037597656e-05,
      "step": 11465,
      "training_step_time": 17.04796028137207
    },
    {
      "epoch": 1.0934829711914062e-05,
      "model_forward_time": 5.596027612686157,
      "step": 11466
    },
    {
      "epoch": 1.0934829711914062e-05,
      "step": 11466,
      "training_step_time": 17.028535842895508
    },
    {
      "epoch": 1.093578338623047e-05,
      "model_forward_time": 5.583625555038452,
      "step": 11467
    },
    {
      "epoch": 1.093578338623047e-05,
      "step": 11467,
      "training_step_time": 17.003379344940186
    },
    {
      "epoch": 1.0936737060546875e-05,
      "model_forward_time": 5.579714298248291,
      "step": 11468
    },
    {
      "epoch": 1.0936737060546875e-05,
      "step": 11468,
      "training_step_time": 16.993695974349976
    },
    {
      "epoch": 1.093769073486328e-05,
      "model_forward_time": 5.9350457191467285,
      "step": 11469
    },
    {
      "epoch": 1.093769073486328e-05,
      "step": 11469,
      "training_step_time": 17.39970564842224
    },
    {
      "epoch": 1.0938644409179688e-05,
      "grad_norm": 0.972350001335144,
      "learning_rate": 1e-05,
      "loss": 0.0636,
      "step": 11470
    },
    {
      "epoch": 1.0938644409179688e-05,
      "model_forward_time": 5.608448266983032,
      "step": 11470
    },
    {
      "dynamics_loss_avg": 0.05455939471721649,
      "epoch": 1.0938644409179688e-05,
      "step": 11470
    },
    {
      "action_loss_avg": 0.00806599692441523,
      "epoch": 1.0938644409179688e-05,
      "step": 11470
    },
    {
      "epoch": 1.0938644409179688e-05,
      "step": 11470,
      "training_step_time": 17.086378574371338
    },
    {
      "epoch": 1.0939598083496094e-05,
      "model_forward_time": 5.606056213378906,
      "step": 11471
    },
    {
      "epoch": 1.0939598083496094e-05,
      "step": 11471,
      "training_step_time": 17.077301740646362
    },
    {
      "epoch": 1.09405517578125e-05,
      "model_forward_time": 5.598860025405884,
      "step": 11472
    },
    {
      "epoch": 1.09405517578125e-05,
      "step": 11472,
      "training_step_time": 17.055083751678467
    },
    {
      "epoch": 1.0941505432128907e-05,
      "model_forward_time": 5.597352504730225,
      "step": 11473
    },
    {
      "epoch": 1.0941505432128907e-05,
      "step": 11473,
      "training_step_time": 17.040746927261353
    },
    {
      "epoch": 1.0942459106445313e-05,
      "model_forward_time": 5.599491357803345,
      "step": 11474
    },
    {
      "epoch": 1.0942459106445313e-05,
      "step": 11474,
      "training_step_time": 17.43271279335022
    },
    {
      "epoch": 1.0943412780761718e-05,
      "model_forward_time": 5.6043806076049805,
      "step": 11475
    },
    {
      "epoch": 1.0943412780761718e-05,
      "step": 11475,
      "training_step_time": 17.09526515007019
    },
    {
      "epoch": 1.0944366455078126e-05,
      "model_forward_time": 5.61923623085022,
      "step": 11476
    },
    {
      "epoch": 1.0944366455078126e-05,
      "step": 11476,
      "training_step_time": 17.13187861442566
    },
    {
      "epoch": 1.0945320129394532e-05,
      "model_forward_time": 5.614058494567871,
      "step": 11477
    },
    {
      "epoch": 1.0945320129394532e-05,
      "step": 11477,
      "training_step_time": 17.097408771514893
    },
    {
      "epoch": 1.0946273803710937e-05,
      "model_forward_time": 5.595526933670044,
      "step": 11478
    },
    {
      "epoch": 1.0946273803710937e-05,
      "step": 11478,
      "training_step_time": 17.03205680847168
    },
    {
      "epoch": 1.0947227478027343e-05,
      "model_forward_time": 5.576842546463013,
      "step": 11479
    },
    {
      "epoch": 1.0947227478027343e-05,
      "step": 11479,
      "training_step_time": 16.981469869613647
    },
    {
      "epoch": 1.094818115234375e-05,
      "grad_norm": 0.3211230933666229,
      "learning_rate": 1e-05,
      "loss": 0.0811,
      "step": 11480
    },
    {
      "epoch": 1.094818115234375e-05,
      "model_forward_time": 5.926478624343872,
      "step": 11480
    },
    {
      "dynamics_loss_avg": 0.07450335174798965,
      "epoch": 1.094818115234375e-05,
      "step": 11480
    },
    {
      "action_loss_avg": 0.00957873915322125,
      "epoch": 1.094818115234375e-05,
      "step": 11480
    },
    {
      "epoch": 1.094818115234375e-05,
      "step": 11480,
      "training_step_time": 17.351033449172974
    },
    {
      "epoch": 1.0949134826660156e-05,
      "model_forward_time": 5.563611745834351,
      "step": 11481
    },
    {
      "epoch": 1.0949134826660156e-05,
      "step": 11481,
      "training_step_time": 16.975834131240845
    },
    {
      "epoch": 1.0950088500976562e-05,
      "model_forward_time": 5.572055816650391,
      "step": 11482
    },
    {
      "epoch": 1.0950088500976562e-05,
      "step": 11482,
      "training_step_time": 16.986101150512695
    },
    {
      "epoch": 1.095104217529297e-05,
      "model_forward_time": 5.5874316692352295,
      "step": 11483
    },
    {
      "epoch": 1.095104217529297e-05,
      "step": 11483,
      "training_step_time": 17.041544437408447
    },
    {
      "epoch": 1.0951995849609375e-05,
      "model_forward_time": 5.605343341827393,
      "step": 11484
    },
    {
      "epoch": 1.0951995849609375e-05,
      "step": 11484,
      "training_step_time": 17.090506553649902
    },
    {
      "epoch": 1.095294952392578e-05,
      "model_forward_time": 5.993286371231079,
      "step": 11485
    },
    {
      "epoch": 1.095294952392578e-05,
      "step": 11485,
      "training_step_time": 17.47870945930481
    },
    {
      "epoch": 1.0953903198242188e-05,
      "model_forward_time": 5.596850395202637,
      "step": 11486
    },
    {
      "epoch": 1.0953903198242188e-05,
      "step": 11486,
      "training_step_time": 17.075923681259155
    },
    {
      "epoch": 1.0954856872558594e-05,
      "model_forward_time": 5.602989912033081,
      "step": 11487
    },
    {
      "epoch": 1.0954856872558594e-05,
      "step": 11487,
      "training_step_time": 17.092109203338623
    },
    {
      "epoch": 1.0955810546875e-05,
      "model_forward_time": 5.607431173324585,
      "step": 11488
    },
    {
      "epoch": 1.0955810546875e-05,
      "step": 11488,
      "training_step_time": 17.097978591918945
    },
    {
      "epoch": 1.0956764221191407e-05,
      "model_forward_time": 5.6090943813323975,
      "step": 11489
    },
    {
      "epoch": 1.0956764221191407e-05,
      "step": 11489,
      "training_step_time": 17.083066940307617
    },
    {
      "epoch": 1.0957717895507813e-05,
      "grad_norm": 0.9625788331031799,
      "learning_rate": 1e-05,
      "loss": 0.0657,
      "step": 11490
    },
    {
      "epoch": 1.0957717895507813e-05,
      "model_forward_time": 5.602645397186279,
      "step": 11490
    },
    {
      "dynamics_loss_avg": 0.05154898017644882,
      "epoch": 1.0957717895507813e-05,
      "step": 11490
    },
    {
      "action_loss_avg": 0.006460384745150804,
      "epoch": 1.0957717895507813e-05,
      "step": 11490
    },
    {
      "epoch": 1.0957717895507813e-05,
      "step": 11490,
      "training_step_time": 17.07618999481201
    },
    {
      "epoch": 1.0958671569824218e-05,
      "model_forward_time": 5.9516894817352295,
      "step": 11491
    },
    {
      "epoch": 1.0958671569824218e-05,
      "step": 11491,
      "training_step_time": 17.39211392402649
    },
    {
      "epoch": 1.0959625244140626e-05,
      "model_forward_time": 5.590209722518921,
      "step": 11492
    },
    {
      "epoch": 1.0959625244140626e-05,
      "step": 11492,
      "training_step_time": 17.023966073989868
    },
    {
      "epoch": 1.0960578918457032e-05,
      "model_forward_time": 4.268468379974365,
      "step": 11493
    },
    {
      "epoch": 1.0960578918457032e-05,
      "step": 11493,
      "training_step_time": 12.766889095306396
    },
    {
      "epoch": 1.0961532592773437e-05,
      "model_forward_time": 5.586740493774414,
      "step": 11494
    },
    {
      "epoch": 1.0961532592773437e-05,
      "step": 11494,
      "training_step_time": 17.0301673412323
    },
    {
      "epoch": 1.0962486267089843e-05,
      "model_forward_time": 5.596198797225952,
      "step": 11495
    },
    {
      "epoch": 1.0962486267089843e-05,
      "step": 11495,
      "training_step_time": 17.090368270874023
    },
    {
      "epoch": 1.096343994140625e-05,
      "model_forward_time": 5.631419658660889,
      "step": 11496
    },
    {
      "epoch": 1.096343994140625e-05,
      "step": 11496,
      "training_step_time": 17.48975706100464
    },
    {
      "epoch": 1.0964393615722656e-05,
      "model_forward_time": 5.6119983196258545,
      "step": 11497
    },
    {
      "epoch": 1.0964393615722656e-05,
      "step": 11497,
      "training_step_time": 17.096599102020264
    },
    {
      "epoch": 1.0965347290039062e-05,
      "model_forward_time": 5.612792015075684,
      "step": 11498
    },
    {
      "epoch": 1.0965347290039062e-05,
      "step": 11498,
      "training_step_time": 17.08948826789856
    },
    {
      "epoch": 1.096630096435547e-05,
      "model_forward_time": 5.597609519958496,
      "step": 11499
    },
    {
      "epoch": 1.096630096435547e-05,
      "step": 11499,
      "training_step_time": 17.05658459663391
    },
    {
      "epoch": 1.0967254638671875e-05,
      "grad_norm": 0.46938779950141907,
      "learning_rate": 1e-05,
      "loss": 0.0573,
      "step": 11500
    },
    {
      "epoch": 1.0967254638671875e-05,
      "model_forward_time": 5.592019319534302,
      "step": 11500
    },
    {
      "dynamics_loss_avg": 0.04894373137503862,
      "epoch": 1.0967254638671875e-05,
      "step": 11500
    },
    {
      "action_loss_avg": 0.0064632494002580644,
      "epoch": 1.0967254638671875e-05,
      "step": 11500
    },
    {
      "epoch": 1.0967254638671875e-05,
      "step": 11500,
      "training_step_time": 17.04754614830017
    },
    {
      "epoch": 1.096820831298828e-05,
      "model_forward_time": 5.595762729644775,
      "step": 11501
    },
    {
      "epoch": 1.096820831298828e-05,
      "step": 11501,
      "training_step_time": 17.05922508239746
    },
    {
      "epoch": 1.0969161987304688e-05,
      "model_forward_time": 5.956468820571899,
      "step": 11502
    },
    {
      "epoch": 1.0969161987304688e-05,
      "step": 11502,
      "training_step_time": 17.433791160583496
    },
    {
      "epoch": 1.0970115661621094e-05,
      "model_forward_time": 5.601302623748779,
      "step": 11503
    },
    {
      "epoch": 1.0970115661621094e-05,
      "step": 11503,
      "training_step_time": 17.039418697357178
    },
    {
      "epoch": 1.09710693359375e-05,
      "model_forward_time": 5.593723773956299,
      "step": 11504
    },
    {
      "epoch": 1.09710693359375e-05,
      "step": 11504,
      "training_step_time": 17.02676820755005
    },
    {
      "epoch": 1.0972023010253907e-05,
      "model_forward_time": 5.589906930923462,
      "step": 11505
    },
    {
      "epoch": 1.0972023010253907e-05,
      "step": 11505,
      "training_step_time": 17.023796558380127
    },
    {
      "epoch": 1.0972976684570313e-05,
      "model_forward_time": 5.594982862472534,
      "step": 11506
    },
    {
      "epoch": 1.0972976684570313e-05,
      "step": 11506,
      "training_step_time": 17.056227207183838
    },
    {
      "epoch": 1.0973930358886718e-05,
      "model_forward_time": 5.97579288482666,
      "step": 11507
    },
    {
      "epoch": 1.0973930358886718e-05,
      "step": 11507,
      "training_step_time": 17.436909914016724
    },
    {
      "epoch": 1.0974884033203124e-05,
      "model_forward_time": 5.6216254234313965,
      "step": 11508
    },
    {
      "epoch": 1.0974884033203124e-05,
      "step": 11508,
      "training_step_time": 17.124947547912598
    },
    {
      "epoch": 1.0975837707519532e-05,
      "model_forward_time": 5.611802101135254,
      "step": 11509
    },
    {
      "epoch": 1.0975837707519532e-05,
      "step": 11509,
      "training_step_time": 17.0866916179657
    },
    {
      "epoch": 1.0976791381835937e-05,
      "grad_norm": 0.9864418506622314,
      "learning_rate": 1e-05,
      "loss": 0.0527,
      "step": 11510
    },
    {
      "epoch": 1.0976791381835937e-05,
      "model_forward_time": 5.599546909332275,
      "step": 11510
    },
    {
      "dynamics_loss_avg": 0.05002453848719597,
      "epoch": 1.0976791381835937e-05,
      "step": 11510
    },
    {
      "action_loss_avg": 0.008643487142398954,
      "epoch": 1.0976791381835937e-05,
      "step": 11510
    },
    {
      "epoch": 1.0976791381835937e-05,
      "step": 11510,
      "training_step_time": 17.01002025604248
    },
    {
      "epoch": 1.0977745056152343e-05,
      "model_forward_time": 5.566661357879639,
      "step": 11511
    },
    {
      "epoch": 1.0977745056152343e-05,
      "step": 11511,
      "training_step_time": 16.984405994415283
    },
    {
      "epoch": 1.097869873046875e-05,
      "model_forward_time": 5.557286500930786,
      "step": 11512
    },
    {
      "epoch": 1.097869873046875e-05,
      "step": 11512,
      "training_step_time": 16.976503133773804
    },
    {
      "epoch": 1.0979652404785156e-05,
      "model_forward_time": 5.903035402297974,
      "step": 11513
    },
    {
      "epoch": 1.0979652404785156e-05,
      "step": 11513,
      "training_step_time": 17.31631636619568
    },
    {
      "epoch": 1.0980606079101562e-05,
      "model_forward_time": 5.564021825790405,
      "step": 11514
    },
    {
      "epoch": 1.0980606079101562e-05,
      "step": 11514,
      "training_step_time": 16.976746082305908
    },
    {
      "epoch": 1.098155975341797e-05,
      "model_forward_time": 5.5763139724731445,
      "step": 11515
    },
    {
      "epoch": 1.098155975341797e-05,
      "step": 11515,
      "training_step_time": 17.017111778259277
    },
    {
      "epoch": 1.0982513427734375e-05,
      "model_forward_time": 5.577796459197998,
      "step": 11516
    },
    {
      "epoch": 1.0982513427734375e-05,
      "step": 11516,
      "training_step_time": 17.00699019432068
    },
    {
      "epoch": 1.098346710205078e-05,
      "model_forward_time": 5.580908536911011,
      "step": 11517
    },
    {
      "epoch": 1.098346710205078e-05,
      "step": 11517,
      "training_step_time": 17.02312397956848
    },
    {
      "epoch": 1.0984420776367188e-05,
      "model_forward_time": 5.958676815032959,
      "step": 11518
    },
    {
      "epoch": 1.0984420776367188e-05,
      "step": 11518,
      "training_step_time": 17.411543369293213
    },
    {
      "epoch": 1.0985374450683594e-05,
      "model_forward_time": 5.597385883331299,
      "step": 11519
    },
    {
      "epoch": 1.0985374450683594e-05,
      "step": 11519,
      "training_step_time": 17.055421113967896
    },
    {
      "epoch": 1.0986328125e-05,
      "grad_norm": 0.5864443778991699,
      "learning_rate": 1e-05,
      "loss": 0.0634,
      "step": 11520
    },
    {
      "epoch": 1.0986328125e-05,
      "model_forward_time": 5.602090120315552,
      "step": 11520
    },
    {
      "dynamics_loss_avg": 0.05886124633252621,
      "epoch": 1.0986328125e-05,
      "step": 11520
    },
    {
      "action_loss_avg": 0.006900893128477037,
      "epoch": 1.0986328125e-05,
      "step": 11520
    },
    {
      "epoch": 1.0986328125e-05,
      "step": 11520,
      "training_step_time": 17.08246397972107
    },
    {
      "epoch": 1.0987281799316407e-05,
      "model_forward_time": 5.602166414260864,
      "step": 11521
    },
    {
      "epoch": 1.0987281799316407e-05,
      "step": 11521,
      "training_step_time": 17.064955234527588
    },
    {
      "epoch": 1.0988235473632813e-05,
      "model_forward_time": 5.608453035354614,
      "step": 11522
    },
    {
      "epoch": 1.0988235473632813e-05,
      "step": 11522,
      "training_step_time": 17.037257194519043
    },
    {
      "epoch": 1.0989189147949218e-05,
      "model_forward_time": 5.591793775558472,
      "step": 11523
    },
    {
      "epoch": 1.0989189147949218e-05,
      "step": 11523,
      "training_step_time": 17.39463710784912
    },
    {
      "epoch": 1.0990142822265624e-05,
      "model_forward_time": 5.583633899688721,
      "step": 11524
    },
    {
      "epoch": 1.0990142822265624e-05,
      "step": 11524,
      "training_step_time": 16.988780736923218
    },
    {
      "epoch": 1.0991096496582032e-05,
      "model_forward_time": 5.592329740524292,
      "step": 11525
    },
    {
      "epoch": 1.0991096496582032e-05,
      "step": 11525,
      "training_step_time": 17.00166082382202
    },
    {
      "epoch": 1.0992050170898437e-05,
      "model_forward_time": 5.598647832870483,
      "step": 11526
    },
    {
      "epoch": 1.0992050170898437e-05,
      "step": 11526,
      "training_step_time": 17.054444313049316
    },
    {
      "epoch": 1.0993003845214843e-05,
      "model_forward_time": 5.58986759185791,
      "step": 11527
    },
    {
      "epoch": 1.0993003845214843e-05,
      "step": 11527,
      "training_step_time": 17.06044602394104
    },
    {
      "epoch": 1.099395751953125e-05,
      "model_forward_time": 5.599859237670898,
      "step": 11528
    },
    {
      "epoch": 1.099395751953125e-05,
      "step": 11528,
      "training_step_time": 17.035106420516968
    },
    {
      "epoch": 1.0994911193847656e-05,
      "model_forward_time": 4.599456310272217,
      "step": 11529
    },
    {
      "epoch": 1.0994911193847656e-05,
      "step": 11529,
      "training_step_time": 13.096219062805176
    },
    {
      "epoch": 1.0995864868164062e-05,
      "grad_norm": 0.6464616656303406,
      "learning_rate": 1e-05,
      "loss": 0.0638,
      "step": 11530
    },
    {
      "epoch": 1.0995864868164062e-05,
      "model_forward_time": 5.595762014389038,
      "step": 11530
    },
    {
      "dynamics_loss_avg": 0.04991056211292744,
      "epoch": 1.0995864868164062e-05,
      "step": 11530
    },
    {
      "action_loss_avg": 0.005855805950704962,
      "epoch": 1.0995864868164062e-05,
      "step": 11530
    },
    {
      "epoch": 1.0995864868164062e-05,
      "step": 11530,
      "training_step_time": 17.07041645050049
    },
    {
      "epoch": 1.099681854248047e-05,
      "model_forward_time": 5.601795434951782,
      "step": 11531
    },
    {
      "epoch": 1.099681854248047e-05,
      "step": 11531,
      "training_step_time": 17.08497643470764
    },
    {
      "epoch": 1.0997772216796875e-05,
      "model_forward_time": 5.603448390960693,
      "step": 11532
    },
    {
      "epoch": 1.0997772216796875e-05,
      "step": 11532,
      "training_step_time": 17.12513017654419
    },
    {
      "epoch": 1.099872589111328e-05,
      "model_forward_time": 5.618695974349976,
      "step": 11533
    },
    {
      "epoch": 1.099872589111328e-05,
      "step": 11533,
      "training_step_time": 17.096482753753662
    },
    {
      "epoch": 1.0999679565429688e-05,
      "model_forward_time": 5.592099905014038,
      "step": 11534
    },
    {
      "epoch": 1.0999679565429688e-05,
      "step": 11534,
      "training_step_time": 17.393521785736084
    },
    {
      "epoch": 1.1000633239746094e-05,
      "model_forward_time": 5.56359601020813,
      "step": 11535
    },
    {
      "epoch": 1.1000633239746094e-05,
      "step": 11535,
      "training_step_time": 16.98762583732605
    },
    {
      "epoch": 1.10015869140625e-05,
      "model_forward_time": 5.567862033843994,
      "step": 11536
    },
    {
      "epoch": 1.10015869140625e-05,
      "step": 11536,
      "training_step_time": 16.978479623794556
    },
    {
      "epoch": 1.1002540588378907e-05,
      "model_forward_time": 5.565059185028076,
      "step": 11537
    },
    {
      "epoch": 1.1002540588378907e-05,
      "step": 11537,
      "training_step_time": 16.98073935508728
    },
    {
      "epoch": 1.1003494262695313e-05,
      "model_forward_time": 5.57388973236084,
      "step": 11538
    },
    {
      "epoch": 1.1003494262695313e-05,
      "step": 11538,
      "training_step_time": 17.015632390975952
    },
    {
      "epoch": 1.1004447937011718e-05,
      "model_forward_time": 5.567603349685669,
      "step": 11539
    },
    {
      "epoch": 1.1004447937011718e-05,
      "step": 11539,
      "training_step_time": 16.977133989334106
    },
    {
      "epoch": 1.1005401611328124e-05,
      "grad_norm": 0.5783703923225403,
      "learning_rate": 1e-05,
      "loss": 0.0513,
      "step": 11540
    },
    {
      "epoch": 1.1005401611328124e-05,
      "model_forward_time": 5.962292432785034,
      "step": 11540
    },
    {
      "dynamics_loss_avg": 0.0449396600946784,
      "epoch": 1.1005401611328124e-05,
      "step": 11540
    },
    {
      "action_loss_avg": 0.006341789779253304,
      "epoch": 1.1005401611328124e-05,
      "step": 11540
    },
    {
      "epoch": 1.1005401611328124e-05,
      "step": 11540,
      "training_step_time": 17.380157470703125
    },
    {
      "epoch": 1.1006355285644532e-05,
      "model_forward_time": 5.581628799438477,
      "step": 11541
    },
    {
      "epoch": 1.1006355285644532e-05,
      "step": 11541,
      "training_step_time": 16.996841430664062
    },
    {
      "epoch": 1.1007308959960937e-05,
      "model_forward_time": 5.606965065002441,
      "step": 11542
    },
    {
      "epoch": 1.1007308959960937e-05,
      "step": 11542,
      "training_step_time": 17.050597190856934
    },
    {
      "epoch": 1.1008262634277343e-05,
      "model_forward_time": 5.60128927230835,
      "step": 11543
    },
    {
      "epoch": 1.1008262634277343e-05,
      "step": 11543,
      "training_step_time": 17.078771114349365
    },
    {
      "epoch": 1.100921630859375e-05,
      "model_forward_time": 5.613564968109131,
      "step": 11544
    },
    {
      "epoch": 1.100921630859375e-05,
      "step": 11544,
      "training_step_time": 17.099682331085205
    },
    {
      "epoch": 1.1010169982910156e-05,
      "model_forward_time": 5.597784042358398,
      "step": 11545
    },
    {
      "epoch": 1.1010169982910156e-05,
      "step": 11545,
      "training_step_time": 17.4184730052948
    },
    {
      "epoch": 1.1011123657226562e-05,
      "model_forward_time": 5.593980073928833,
      "step": 11546
    },
    {
      "epoch": 1.1011123657226562e-05,
      "step": 11546,
      "training_step_time": 17.05417275428772
    },
    {
      "epoch": 1.101207733154297e-05,
      "model_forward_time": 5.586254119873047,
      "step": 11547
    },
    {
      "epoch": 1.101207733154297e-05,
      "step": 11547,
      "training_step_time": 17.034766674041748
    },
    {
      "epoch": 1.1013031005859375e-05,
      "model_forward_time": 5.58948826789856,
      "step": 11548
    },
    {
      "epoch": 1.1013031005859375e-05,
      "step": 11548,
      "training_step_time": 17.04745841026306
    },
    {
      "epoch": 1.101398468017578e-05,
      "model_forward_time": 5.597126007080078,
      "step": 11549
    },
    {
      "epoch": 1.101398468017578e-05,
      "step": 11549,
      "training_step_time": 17.03726029396057
    },
    {
      "epoch": 1.1014938354492188e-05,
      "grad_norm": 0.945947527885437,
      "learning_rate": 1e-05,
      "loss": 0.0564,
      "step": 11550
    },
    {
      "epoch": 1.1014938354492188e-05,
      "model_forward_time": 5.595292568206787,
      "step": 11550
    },
    {
      "dynamics_loss_avg": 0.05215760972350836,
      "epoch": 1.1014938354492188e-05,
      "step": 11550
    },
    {
      "action_loss_avg": 0.006808071420527995,
      "epoch": 1.1014938354492188e-05,
      "step": 11550
    },
    {
      "epoch": 1.1014938354492188e-05,
      "step": 11550,
      "training_step_time": 17.063631296157837
    },
    {
      "epoch": 1.1015892028808594e-05,
      "model_forward_time": 4.6189892292022705,
      "step": 11551
    },
    {
      "epoch": 1.1015892028808594e-05,
      "step": 11551,
      "training_step_time": 13.14146375656128
    },
    {
      "epoch": 1.1016845703125e-05,
      "model_forward_time": 5.592739820480347,
      "step": 11552
    },
    {
      "epoch": 1.1016845703125e-05,
      "step": 11552,
      "training_step_time": 17.070064783096313
    },
    {
      "epoch": 1.1017799377441407e-05,
      "model_forward_time": 5.5997772216796875,
      "step": 11553
    },
    {
      "epoch": 1.1017799377441407e-05,
      "step": 11553,
      "training_step_time": 17.057408809661865
    },
    {
      "epoch": 1.1018753051757813e-05,
      "model_forward_time": 5.594785213470459,
      "step": 11554
    },
    {
      "epoch": 1.1018753051757813e-05,
      "step": 11554,
      "training_step_time": 17.05063819885254
    },
    {
      "epoch": 1.1019706726074218e-05,
      "model_forward_time": 5.591891050338745,
      "step": 11555
    },
    {
      "epoch": 1.1019706726074218e-05,
      "step": 11555,
      "training_step_time": 17.023748636245728
    },
    {
      "epoch": 1.1020660400390624e-05,
      "model_forward_time": 5.595474720001221,
      "step": 11556
    },
    {
      "epoch": 1.1020660400390624e-05,
      "step": 11556,
      "training_step_time": 17.383439540863037
    },
    {
      "epoch": 1.1021614074707032e-05,
      "model_forward_time": 5.602550268173218,
      "step": 11557
    },
    {
      "epoch": 1.1021614074707032e-05,
      "step": 11557,
      "training_step_time": 17.072772979736328
    },
    {
      "epoch": 1.1022567749023437e-05,
      "model_forward_time": 5.61290431022644,
      "step": 11558
    },
    {
      "epoch": 1.1022567749023437e-05,
      "step": 11558,
      "training_step_time": 17.084689617156982
    },
    {
      "epoch": 1.1023521423339843e-05,
      "model_forward_time": 5.608666181564331,
      "step": 11559
    },
    {
      "epoch": 1.1023521423339843e-05,
      "step": 11559,
      "training_step_time": 17.074665546417236
    },
    {
      "epoch": 1.102447509765625e-05,
      "grad_norm": 1.076684832572937,
      "learning_rate": 1e-05,
      "loss": 0.0695,
      "step": 11560
    },
    {
      "epoch": 1.102447509765625e-05,
      "model_forward_time": 5.595670938491821,
      "step": 11560
    },
    {
      "dynamics_loss_avg": 0.05717079993337393,
      "epoch": 1.102447509765625e-05,
      "step": 11560
    },
    {
      "action_loss_avg": 0.009342059073969722,
      "epoch": 1.102447509765625e-05,
      "step": 11560
    },
    {
      "epoch": 1.102447509765625e-05,
      "step": 11560,
      "training_step_time": 17.06765604019165
    },
    {
      "epoch": 1.1025428771972656e-05,
      "model_forward_time": 5.605058670043945,
      "step": 11561
    },
    {
      "epoch": 1.1025428771972656e-05,
      "step": 11561,
      "training_step_time": 17.045329809188843
    },
    {
      "epoch": 1.1026382446289062e-05,
      "model_forward_time": 5.960056781768799,
      "step": 11562
    },
    {
      "epoch": 1.1026382446289062e-05,
      "step": 11562,
      "training_step_time": 17.40245032310486
    },
    {
      "epoch": 1.102733612060547e-05,
      "model_forward_time": 5.588332176208496,
      "step": 11563
    },
    {
      "epoch": 1.102733612060547e-05,
      "step": 11563,
      "training_step_time": 17.02644157409668
    },
    {
      "epoch": 1.1028289794921875e-05,
      "model_forward_time": 5.5992751121521,
      "step": 11564
    },
    {
      "epoch": 1.1028289794921875e-05,
      "step": 11564,
      "training_step_time": 17.0606107711792
    },
    {
      "epoch": 1.102924346923828e-05,
      "model_forward_time": 5.606319904327393,
      "step": 11565
    },
    {
      "epoch": 1.102924346923828e-05,
      "step": 11565,
      "training_step_time": 17.09690833091736
    },
    {
      "epoch": 1.1030197143554688e-05,
      "model_forward_time": 5.617754220962524,
      "step": 11566
    },
    {
      "epoch": 1.1030197143554688e-05,
      "step": 11566,
      "training_step_time": 17.093218326568604
    },
    {
      "epoch": 1.1031150817871094e-05,
      "model_forward_time": 5.962863445281982,
      "step": 11567
    },
    {
      "epoch": 1.1031150817871094e-05,
      "step": 11567,
      "training_step_time": 17.399258852005005
    },
    {
      "epoch": 1.10321044921875e-05,
      "model_forward_time": 5.577356338500977,
      "step": 11568
    },
    {
      "epoch": 1.10321044921875e-05,
      "step": 11568,
      "training_step_time": 17.004990577697754
    },
    {
      "epoch": 1.1033058166503907e-05,
      "model_forward_time": 5.582156181335449,
      "step": 11569
    },
    {
      "epoch": 1.1033058166503907e-05,
      "step": 11569,
      "training_step_time": 17.008721351623535
    },
    {
      "epoch": 1.1034011840820313e-05,
      "grad_norm": 1.0133870840072632,
      "learning_rate": 1e-05,
      "loss": 0.0545,
      "step": 11570
    },
    {
      "epoch": 1.1034011840820313e-05,
      "model_forward_time": 5.593899726867676,
      "step": 11570
    },
    {
      "dynamics_loss_avg": 0.057021685875952244,
      "epoch": 1.1034011840820313e-05,
      "step": 11570
    },
    {
      "action_loss_avg": 0.009591687843203545,
      "epoch": 1.1034011840820313e-05,
      "step": 11570
    },
    {
      "epoch": 1.1034011840820313e-05,
      "step": 11570,
      "training_step_time": 17.04796576499939
    },
    {
      "epoch": 1.1034965515136718e-05,
      "model_forward_time": 5.594365358352661,
      "step": 11571
    },
    {
      "epoch": 1.1034965515136718e-05,
      "step": 11571,
      "training_step_time": 17.051104068756104
    },
    {
      "epoch": 1.1035919189453124e-05,
      "model_forward_time": 5.6002209186553955,
      "step": 11572
    },
    {
      "epoch": 1.1035919189453124e-05,
      "step": 11572,
      "training_step_time": 17.066755294799805
    },
    {
      "epoch": 1.1036872863769532e-05,
      "model_forward_time": 5.960426568984985,
      "step": 11573
    },
    {
      "epoch": 1.1036872863769532e-05,
      "step": 11573,
      "training_step_time": 17.41563844680786
    },
    {
      "epoch": 1.1037826538085937e-05,
      "model_forward_time": 5.602305173873901,
      "step": 11574
    },
    {
      "epoch": 1.1037826538085937e-05,
      "step": 11574,
      "training_step_time": 17.083888292312622
    },
    {
      "epoch": 1.1038780212402343e-05,
      "model_forward_time": 5.604342222213745,
      "step": 11575
    },
    {
      "epoch": 1.1038780212402343e-05,
      "step": 11575,
      "training_step_time": 17.1099693775177
    },
    {
      "epoch": 1.103973388671875e-05,
      "model_forward_time": 5.611055850982666,
      "step": 11576
    },
    {
      "epoch": 1.103973388671875e-05,
      "step": 11576,
      "training_step_time": 17.087566137313843
    },
    {
      "epoch": 1.1040687561035156e-05,
      "model_forward_time": 5.6354663372039795,
      "step": 11577
    },
    {
      "epoch": 1.1040687561035156e-05,
      "step": 11577,
      "training_step_time": 17.12290048599243
    },
    {
      "epoch": 1.1041641235351562e-05,
      "model_forward_time": 5.967127084732056,
      "step": 11578
    },
    {
      "epoch": 1.1041641235351562e-05,
      "step": 11578,
      "training_step_time": 17.422913074493408
    },
    {
      "epoch": 1.104259490966797e-05,
      "model_forward_time": 5.593475580215454,
      "step": 11579
    },
    {
      "epoch": 1.104259490966797e-05,
      "step": 11579,
      "training_step_time": 17.012685537338257
    },
    {
      "epoch": 1.1043548583984375e-05,
      "grad_norm": 1.081376552581787,
      "learning_rate": 1e-05,
      "loss": 0.0828,
      "step": 11580
    },
    {
      "epoch": 1.1043548583984375e-05,
      "model_forward_time": 5.577847480773926,
      "step": 11580
    },
    {
      "dynamics_loss_avg": 0.06492430064827204,
      "epoch": 1.1043548583984375e-05,
      "step": 11580
    },
    {
      "action_loss_avg": 0.009309595020022243,
      "epoch": 1.1043548583984375e-05,
      "step": 11580
    },
    {
      "epoch": 1.1043548583984375e-05,
      "step": 11580,
      "training_step_time": 17.022080183029175
    },
    {
      "epoch": 1.104450225830078e-05,
      "model_forward_time": 5.586486339569092,
      "step": 11581
    },
    {
      "epoch": 1.104450225830078e-05,
      "step": 11581,
      "training_step_time": 16.99977397918701
    },
    {
      "epoch": 1.1045455932617188e-05,
      "model_forward_time": 5.579965829849243,
      "step": 11582
    },
    {
      "epoch": 1.1045455932617188e-05,
      "step": 11582,
      "training_step_time": 17.022396326065063
    },
    {
      "epoch": 1.1046409606933594e-05,
      "model_forward_time": 5.592056751251221,
      "step": 11583
    },
    {
      "epoch": 1.1046409606933594e-05,
      "step": 11583,
      "training_step_time": 17.060317039489746
    },
    {
      "epoch": 1.104736328125e-05,
      "model_forward_time": 5.989634037017822,
      "step": 11584
    },
    {
      "epoch": 1.104736328125e-05,
      "step": 11584,
      "training_step_time": 17.503018856048584
    },
    {
      "epoch": 1.1048316955566407e-05,
      "model_forward_time": 5.610758066177368,
      "step": 11585
    },
    {
      "epoch": 1.1048316955566407e-05,
      "step": 11585,
      "training_step_time": 17.099485874176025
    },
    {
      "epoch": 1.1049270629882813e-05,
      "model_forward_time": 5.599653482437134,
      "step": 11586
    },
    {
      "epoch": 1.1049270629882813e-05,
      "step": 11586,
      "training_step_time": 17.051133632659912
    },
    {
      "epoch": 1.1050224304199219e-05,
      "model_forward_time": 5.590181112289429,
      "step": 11587
    },
    {
      "epoch": 1.1050224304199219e-05,
      "step": 11587,
      "training_step_time": 17.037129163742065
    },
    {
      "epoch": 1.1051177978515624e-05,
      "model_forward_time": 5.596747159957886,
      "step": 11588
    },
    {
      "epoch": 1.1051177978515624e-05,
      "step": 11588,
      "training_step_time": 17.07268977165222
    },
    {
      "epoch": 1.1052131652832032e-05,
      "model_forward_time": 5.956039667129517,
      "step": 11589
    },
    {
      "epoch": 1.1052131652832032e-05,
      "step": 11589,
      "training_step_time": 17.44049382209778
    },
    {
      "epoch": 1.1053085327148437e-05,
      "grad_norm": 0.9049102663993835,
      "learning_rate": 1e-05,
      "loss": 0.0683,
      "step": 11590
    },
    {
      "epoch": 1.1053085327148437e-05,
      "model_forward_time": 5.618650674819946,
      "step": 11590
    },
    {
      "dynamics_loss_avg": 0.059655226022005084,
      "epoch": 1.1053085327148437e-05,
      "step": 11590
    },
    {
      "action_loss_avg": 0.006587603664956987,
      "epoch": 1.1053085327148437e-05,
      "step": 11590
    },
    {
      "epoch": 1.1053085327148437e-05,
      "step": 11590,
      "training_step_time": 17.124414443969727
    },
    {
      "epoch": 1.1054039001464843e-05,
      "model_forward_time": 5.62671160697937,
      "step": 11591
    },
    {
      "epoch": 1.1054039001464843e-05,
      "step": 11591,
      "training_step_time": 17.099313259124756
    },
    {
      "epoch": 1.105499267578125e-05,
      "model_forward_time": 5.592844247817993,
      "step": 11592
    },
    {
      "epoch": 1.105499267578125e-05,
      "step": 11592,
      "training_step_time": 17.068276405334473
    },
    {
      "epoch": 1.1055946350097656e-05,
      "model_forward_time": 5.592721700668335,
      "step": 11593
    },
    {
      "epoch": 1.1055946350097656e-05,
      "step": 11593,
      "training_step_time": 17.0509614944458
    },
    {
      "epoch": 1.1056900024414062e-05,
      "model_forward_time": 5.6020965576171875,
      "step": 11594
    },
    {
      "epoch": 1.1056900024414062e-05,
      "step": 11594,
      "training_step_time": 17.456620931625366
    },
    {
      "epoch": 1.105785369873047e-05,
      "model_forward_time": 5.5923216342926025,
      "step": 11595
    },
    {
      "epoch": 1.105785369873047e-05,
      "step": 11595,
      "training_step_time": 17.05437135696411
    },
    {
      "epoch": 1.1058807373046875e-05,
      "model_forward_time": 5.599464416503906,
      "step": 11596
    },
    {
      "epoch": 1.1058807373046875e-05,
      "step": 11596,
      "training_step_time": 17.075427055358887
    },
    {
      "epoch": 1.105976104736328e-05,
      "model_forward_time": 5.608388662338257,
      "step": 11597
    },
    {
      "epoch": 1.105976104736328e-05,
      "step": 11597,
      "training_step_time": 17.087292432785034
    },
    {
      "epoch": 1.1060714721679688e-05,
      "model_forward_time": 5.612366199493408,
      "step": 11598
    },
    {
      "epoch": 1.1060714721679688e-05,
      "step": 11598,
      "training_step_time": 17.11416983604431
    },
    {
      "epoch": 1.1061668395996094e-05,
      "model_forward_time": 5.618267059326172,
      "step": 11599
    },
    {
      "epoch": 1.1061668395996094e-05,
      "step": 11599,
      "training_step_time": 17.11969566345215
    },
    {
      "epoch": 1.10626220703125e-05,
      "grad_norm": 0.9439313411712646,
      "learning_rate": 1e-05,
      "loss": 0.0602,
      "step": 11600
    },
    {
      "epoch": 1.10626220703125e-05,
      "model_forward_time": 5.967594385147095,
      "step": 11600
    },
    {
      "dynamics_loss_avg": 0.05955542773008347,
      "epoch": 1.10626220703125e-05,
      "step": 11600
    },
    {
      "action_loss_avg": 0.007245433237403631,
      "epoch": 1.10626220703125e-05,
      "step": 11600
    },
    {
      "epoch": 1.10626220703125e-05,
      "step": 11600,
      "training_step_time": 17.449897289276123
    },
    {
      "epoch": 1.1063575744628907e-05,
      "model_forward_time": 5.594648599624634,
      "step": 11601
    },
    {
      "epoch": 1.1063575744628907e-05,
      "step": 11601,
      "training_step_time": 17.022708892822266
    },
    {
      "epoch": 1.1064529418945313e-05,
      "model_forward_time": 5.577741861343384,
      "step": 11602
    },
    {
      "epoch": 1.1064529418945313e-05,
      "step": 11602,
      "training_step_time": 17.005712747573853
    },
    {
      "epoch": 1.1065483093261719e-05,
      "model_forward_time": 5.57649564743042,
      "step": 11603
    },
    {
      "epoch": 1.1065483093261719e-05,
      "step": 11603,
      "training_step_time": 17.0115065574646
    },
    {
      "epoch": 1.1066436767578124e-05,
      "model_forward_time": 5.590635776519775,
      "step": 11604
    },
    {
      "epoch": 1.1066436767578124e-05,
      "step": 11604,
      "training_step_time": 17.021047592163086
    },
    {
      "epoch": 1.1067390441894532e-05,
      "model_forward_time": 5.586406469345093,
      "step": 11605
    },
    {
      "epoch": 1.1067390441894532e-05,
      "step": 11605,
      "training_step_time": 17.41137671470642
    },
    {
      "epoch": 1.1068344116210937e-05,
      "model_forward_time": 5.609349012374878,
      "step": 11606
    },
    {
      "epoch": 1.1068344116210937e-05,
      "step": 11606,
      "training_step_time": 17.102057695388794
    },
    {
      "epoch": 1.1069297790527343e-05,
      "model_forward_time": 5.624769926071167,
      "step": 11607
    },
    {
      "epoch": 1.1069297790527343e-05,
      "step": 11607,
      "training_step_time": 17.133148908615112
    },
    {
      "epoch": 1.107025146484375e-05,
      "model_forward_time": 5.605335235595703,
      "step": 11608
    },
    {
      "epoch": 1.107025146484375e-05,
      "step": 11608,
      "training_step_time": 17.06257677078247
    },
    {
      "epoch": 1.1071205139160156e-05,
      "model_forward_time": 5.590906143188477,
      "step": 11609
    },
    {
      "epoch": 1.1071205139160156e-05,
      "step": 11609,
      "training_step_time": 17.017935276031494
    },
    {
      "epoch": 1.1072158813476562e-05,
      "grad_norm": 0.6511536836624146,
      "learning_rate": 1e-05,
      "loss": 0.0673,
      "step": 11610
    },
    {
      "epoch": 1.1072158813476562e-05,
      "model_forward_time": 5.579435586929321,
      "step": 11610
    },
    {
      "dynamics_loss_avg": 0.05309704579412937,
      "epoch": 1.1072158813476562e-05,
      "step": 11610
    },
    {
      "action_loss_avg": 0.006582786410581321,
      "epoch": 1.1072158813476562e-05,
      "step": 11610
    },
    {
      "epoch": 1.1072158813476562e-05,
      "step": 11610,
      "training_step_time": 17.014132738113403
    },
    {
      "epoch": 1.107311248779297e-05,
      "model_forward_time": 5.928390026092529,
      "step": 11611
    },
    {
      "epoch": 1.107311248779297e-05,
      "step": 11611,
      "training_step_time": 17.35842728614807
    },
    {
      "epoch": 1.1074066162109375e-05,
      "model_forward_time": 5.594751358032227,
      "step": 11612
    },
    {
      "epoch": 1.1074066162109375e-05,
      "step": 11612,
      "training_step_time": 17.063223361968994
    },
    {
      "epoch": 1.107501983642578e-05,
      "model_forward_time": 5.606450796127319,
      "step": 11613
    },
    {
      "epoch": 1.107501983642578e-05,
      "step": 11613,
      "training_step_time": 17.10310697555542
    },
    {
      "epoch": 1.1075973510742188e-05,
      "model_forward_time": 5.629706144332886,
      "step": 11614
    },
    {
      "epoch": 1.1075973510742188e-05,
      "step": 11614,
      "training_step_time": 17.148207902908325
    },
    {
      "epoch": 1.1076927185058594e-05,
      "model_forward_time": 5.622529983520508,
      "step": 11615
    },
    {
      "epoch": 1.1076927185058594e-05,
      "step": 11615,
      "training_step_time": 17.11275815963745
    },
    {
      "epoch": 1.1077880859375e-05,
      "model_forward_time": 5.9750587940216064,
      "step": 11616
    },
    {
      "epoch": 1.1077880859375e-05,
      "step": 11616,
      "training_step_time": 17.391330242156982
    },
    {
      "epoch": 1.1078834533691407e-05,
      "model_forward_time": 5.564729928970337,
      "step": 11617
    },
    {
      "epoch": 1.1078834533691407e-05,
      "step": 11617,
      "training_step_time": 16.99031949043274
    },
    {
      "epoch": 1.1079788208007813e-05,
      "model_forward_time": 5.557892084121704,
      "step": 11618
    },
    {
      "epoch": 1.1079788208007813e-05,
      "step": 11618,
      "training_step_time": 16.976465940475464
    },
    {
      "epoch": 1.1080741882324219e-05,
      "model_forward_time": 5.553938865661621,
      "step": 11619
    },
    {
      "epoch": 1.1080741882324219e-05,
      "step": 11619,
      "training_step_time": 16.971619129180908
    },
    {
      "epoch": 1.1081695556640624e-05,
      "grad_norm": 0.3190177083015442,
      "learning_rate": 1e-05,
      "loss": 0.06,
      "step": 11620
    },
    {
      "epoch": 1.1081695556640624e-05,
      "model_forward_time": 5.553197860717773,
      "step": 11620
    },
    {
      "dynamics_loss_avg": 0.05589582454413176,
      "epoch": 1.1081695556640624e-05,
      "step": 11620
    },
    {
      "action_loss_avg": 0.006398838432505727,
      "epoch": 1.1081695556640624e-05,
      "step": 11620
    },
    {
      "epoch": 1.1081695556640624e-05,
      "step": 11620,
      "training_step_time": 16.968072414398193
    },
    {
      "epoch": 1.1082649230957032e-05,
      "model_forward_time": 5.575209617614746,
      "step": 11621
    },
    {
      "epoch": 1.1082649230957032e-05,
      "step": 11621,
      "training_step_time": 16.994033575057983
    },
    {
      "epoch": 1.1083602905273437e-05,
      "model_forward_time": 5.909272909164429,
      "step": 11622
    },
    {
      "epoch": 1.1083602905273437e-05,
      "step": 11622,
      "training_step_time": 17.33701181411743
    },
    {
      "epoch": 1.1084556579589843e-05,
      "model_forward_time": 5.5786755084991455,
      "step": 11623
    },
    {
      "epoch": 1.1084556579589843e-05,
      "step": 11623,
      "training_step_time": 16.98465371131897
    },
    {
      "epoch": 1.108551025390625e-05,
      "model_forward_time": 5.5839080810546875,
      "step": 11624
    },
    {
      "epoch": 1.108551025390625e-05,
      "step": 11624,
      "training_step_time": 17.020506143569946
    },
    {
      "epoch": 1.1086463928222656e-05,
      "model_forward_time": 5.601255416870117,
      "step": 11625
    },
    {
      "epoch": 1.1086463928222656e-05,
      "step": 11625,
      "training_step_time": 17.070154666900635
    },
    {
      "epoch": 1.1087417602539062e-05,
      "model_forward_time": 5.617984771728516,
      "step": 11626
    },
    {
      "epoch": 1.1087417602539062e-05,
      "step": 11626,
      "training_step_time": 17.127353191375732
    },
    {
      "epoch": 1.108837127685547e-05,
      "model_forward_time": 6.0078489780426025,
      "step": 11627
    },
    {
      "epoch": 1.108837127685547e-05,
      "step": 11627,
      "training_step_time": 17.487098217010498
    },
    {
      "epoch": 1.1089324951171875e-05,
      "model_forward_time": 5.596958637237549,
      "step": 11628
    },
    {
      "epoch": 1.1089324951171875e-05,
      "step": 11628,
      "training_step_time": 17.033762216567993
    },
    {
      "epoch": 1.109027862548828e-05,
      "model_forward_time": 5.576563835144043,
      "step": 11629
    },
    {
      "epoch": 1.109027862548828e-05,
      "step": 11629,
      "training_step_time": 17.021889686584473
    },
    {
      "epoch": 1.1091232299804688e-05,
      "grad_norm": 0.8840788006782532,
      "learning_rate": 1e-05,
      "loss": 0.0592,
      "step": 11630
    },
    {
      "epoch": 1.1091232299804688e-05,
      "model_forward_time": 5.577691555023193,
      "step": 11630
    },
    {
      "dynamics_loss_avg": 0.05162388626486063,
      "epoch": 1.1091232299804688e-05,
      "step": 11630
    },
    {
      "action_loss_avg": 0.008320317813195288,
      "epoch": 1.1091232299804688e-05,
      "step": 11630
    },
    {
      "epoch": 1.1091232299804688e-05,
      "step": 11630,
      "training_step_time": 16.98669719696045
    },
    {
      "epoch": 1.1092185974121094e-05,
      "model_forward_time": 5.591503381729126,
      "step": 11631
    },
    {
      "epoch": 1.1092185974121094e-05,
      "step": 11631,
      "training_step_time": 17.03771710395813
    },
    {
      "epoch": 1.10931396484375e-05,
      "model_forward_time": 5.603579759597778,
      "step": 11632
    },
    {
      "epoch": 1.10931396484375e-05,
      "step": 11632,
      "training_step_time": 17.085058212280273
    },
    {
      "epoch": 1.1094093322753905e-05,
      "model_forward_time": 5.968330383300781,
      "step": 11633
    },
    {
      "epoch": 1.1094093322753905e-05,
      "step": 11633,
      "training_step_time": 17.45997166633606
    },
    {
      "epoch": 1.1095046997070313e-05,
      "model_forward_time": 5.627254247665405,
      "step": 11634
    },
    {
      "epoch": 1.1095046997070313e-05,
      "step": 11634,
      "training_step_time": 17.133231163024902
    },
    {
      "epoch": 1.1096000671386719e-05,
      "model_forward_time": 5.619659185409546,
      "step": 11635
    },
    {
      "epoch": 1.1096000671386719e-05,
      "step": 11635,
      "training_step_time": 17.124496936798096
    },
    {
      "epoch": 1.1096954345703124e-05,
      "model_forward_time": 5.611079216003418,
      "step": 11636
    },
    {
      "epoch": 1.1096954345703124e-05,
      "step": 11636,
      "training_step_time": 17.102412939071655
    },
    {
      "epoch": 1.1097908020019532e-05,
      "model_forward_time": 5.6108903884887695,
      "step": 11637
    },
    {
      "epoch": 1.1097908020019532e-05,
      "step": 11637,
      "training_step_time": 17.096233367919922
    },
    {
      "epoch": 1.1098861694335937e-05,
      "model_forward_time": 5.979378700256348,
      "step": 11638
    },
    {
      "epoch": 1.1098861694335937e-05,
      "step": 11638,
      "training_step_time": 17.468891859054565
    },
    {
      "epoch": 1.1099815368652343e-05,
      "model_forward_time": 5.611331462860107,
      "step": 11639
    },
    {
      "epoch": 1.1099815368652343e-05,
      "step": 11639,
      "training_step_time": 17.094706773757935
    },
    {
      "epoch": 1.110076904296875e-05,
      "grad_norm": 0.6166830658912659,
      "learning_rate": 1e-05,
      "loss": 0.0667,
      "step": 11640
    },
    {
      "epoch": 1.110076904296875e-05,
      "model_forward_time": 5.5966637134552,
      "step": 11640
    },
    {
      "dynamics_loss_avg": 0.06080052703619003,
      "epoch": 1.110076904296875e-05,
      "step": 11640
    },
    {
      "action_loss_avg": 0.007248336495831609,
      "epoch": 1.110076904296875e-05,
      "step": 11640
    },
    {
      "epoch": 1.110076904296875e-05,
      "step": 11640,
      "training_step_time": 17.046111822128296
    },
    {
      "epoch": 1.1101722717285156e-05,
      "model_forward_time": 5.582823753356934,
      "step": 11641
    },
    {
      "epoch": 1.1101722717285156e-05,
      "step": 11641,
      "training_step_time": 17.016961336135864
    },
    {
      "epoch": 1.1102676391601562e-05,
      "model_forward_time": 5.578044414520264,
      "step": 11642
    },
    {
      "epoch": 1.1102676391601562e-05,
      "step": 11642,
      "training_step_time": 16.989389657974243
    },
    {
      "epoch": 1.110363006591797e-05,
      "model_forward_time": 5.928949356079102,
      "step": 11643
    },
    {
      "epoch": 1.110363006591797e-05,
      "step": 11643,
      "training_step_time": 17.341806650161743
    },
    {
      "epoch": 1.1104583740234375e-05,
      "model_forward_time": 5.588329315185547,
      "step": 11644
    },
    {
      "epoch": 1.1104583740234375e-05,
      "step": 11644,
      "training_step_time": 17.031426191329956
    },
    {
      "epoch": 1.110553741455078e-05,
      "model_forward_time": 5.603801488876343,
      "step": 11645
    },
    {
      "epoch": 1.110553741455078e-05,
      "step": 11645,
      "training_step_time": 17.092467546463013
    },
    {
      "epoch": 1.1106491088867188e-05,
      "model_forward_time": 5.600050687789917,
      "step": 11646
    },
    {
      "epoch": 1.1106491088867188e-05,
      "step": 11646,
      "training_step_time": 17.085169076919556
    },
    {
      "epoch": 1.1107444763183594e-05,
      "model_forward_time": 5.61651611328125,
      "step": 11647
    },
    {
      "epoch": 1.1107444763183594e-05,
      "step": 11647,
      "training_step_time": 17.097089052200317
    },
    {
      "epoch": 1.11083984375e-05,
      "model_forward_time": 5.608576059341431,
      "step": 11648
    },
    {
      "epoch": 1.11083984375e-05,
      "step": 11648,
      "training_step_time": 17.08735179901123
    },
    {
      "epoch": 1.1109352111816405e-05,
      "model_forward_time": 5.99093222618103,
      "step": 11649
    },
    {
      "epoch": 1.1109352111816405e-05,
      "step": 11649,
      "training_step_time": 17.478320121765137
    },
    {
      "epoch": 1.1110305786132813e-05,
      "grad_norm": 1.0608943700790405,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 11650
    },
    {
      "epoch": 1.1110305786132813e-05,
      "model_forward_time": 5.613786935806274,
      "step": 11650
    },
    {
      "dynamics_loss_avg": 0.058735033869743346,
      "epoch": 1.1110305786132813e-05,
      "step": 11650
    },
    {
      "action_loss_avg": 0.008149843325372785,
      "epoch": 1.1110305786132813e-05,
      "step": 11650
    },
    {
      "epoch": 1.1110305786132813e-05,
      "step": 11650,
      "training_step_time": 17.077678680419922
    },
    {
      "epoch": 1.1111259460449219e-05,
      "model_forward_time": 5.6009202003479,
      "step": 11651
    },
    {
      "epoch": 1.1111259460449219e-05,
      "step": 11651,
      "training_step_time": 17.065682888031006
    },
    {
      "epoch": 1.1112213134765624e-05,
      "model_forward_time": 5.606546878814697,
      "step": 11652
    },
    {
      "epoch": 1.1112213134765624e-05,
      "step": 11652,
      "training_step_time": 17.061922073364258
    },
    {
      "epoch": 1.1113166809082032e-05,
      "model_forward_time": 5.601875066757202,
      "step": 11653
    },
    {
      "epoch": 1.1113166809082032e-05,
      "step": 11653,
      "training_step_time": 17.076151847839355
    },
    {
      "epoch": 1.1114120483398437e-05,
      "model_forward_time": 5.948385238647461,
      "step": 11654
    },
    {
      "epoch": 1.1114120483398437e-05,
      "step": 11654,
      "training_step_time": 17.388765811920166
    },
    {
      "epoch": 1.1115074157714843e-05,
      "model_forward_time": 5.613323450088501,
      "step": 11655
    },
    {
      "epoch": 1.1115074157714843e-05,
      "step": 11655,
      "training_step_time": 17.064078092575073
    },
    {
      "epoch": 1.111602783203125e-05,
      "model_forward_time": 5.612748384475708,
      "step": 11656
    },
    {
      "epoch": 1.111602783203125e-05,
      "step": 11656,
      "training_step_time": 17.122010707855225
    },
    {
      "epoch": 1.1116981506347656e-05,
      "model_forward_time": 5.624596834182739,
      "step": 11657
    },
    {
      "epoch": 1.1116981506347656e-05,
      "step": 11657,
      "training_step_time": 17.136168479919434
    },
    {
      "epoch": 1.1117935180664062e-05,
      "model_forward_time": 5.618930816650391,
      "step": 11658
    },
    {
      "epoch": 1.1117935180664062e-05,
      "step": 11658,
      "training_step_time": 17.123712301254272
    },
    {
      "epoch": 1.111888885498047e-05,
      "model_forward_time": 5.6157615184783936,
      "step": 11659
    },
    {
      "epoch": 1.111888885498047e-05,
      "step": 11659,
      "training_step_time": 17.094727039337158
    },
    {
      "epoch": 1.1119842529296875e-05,
      "grad_norm": 0.6695065498352051,
      "learning_rate": 1e-05,
      "loss": 0.0527,
      "step": 11660
    },
    {
      "epoch": 1.1119842529296875e-05,
      "model_forward_time": 5.9860429763793945,
      "step": 11660
    },
    {
      "dynamics_loss_avg": 0.044214756786823274,
      "epoch": 1.1119842529296875e-05,
      "step": 11660
    },
    {
      "action_loss_avg": 0.004840448743198067,
      "epoch": 1.1119842529296875e-05,
      "step": 11660
    },
    {
      "epoch": 1.1119842529296875e-05,
      "step": 11660,
      "training_step_time": 17.45674753189087
    },
    {
      "epoch": 1.1120796203613281e-05,
      "model_forward_time": 5.609455108642578,
      "step": 11661
    },
    {
      "epoch": 1.1120796203613281e-05,
      "step": 11661,
      "training_step_time": 17.077526569366455
    },
    {
      "epoch": 1.1121749877929688e-05,
      "model_forward_time": 5.604865074157715,
      "step": 11662
    },
    {
      "epoch": 1.1121749877929688e-05,
      "step": 11662,
      "training_step_time": 17.05762028694153
    },
    {
      "epoch": 1.1122703552246094e-05,
      "model_forward_time": 5.599102020263672,
      "step": 11663
    },
    {
      "epoch": 1.1122703552246094e-05,
      "step": 11663,
      "training_step_time": 17.045781135559082
    },
    {
      "epoch": 1.11236572265625e-05,
      "model_forward_time": 5.5953688621521,
      "step": 11664
    },
    {
      "epoch": 1.11236572265625e-05,
      "step": 11664,
      "training_step_time": 17.049077033996582
    },
    {
      "epoch": 1.1124610900878905e-05,
      "model_forward_time": 5.596723318099976,
      "step": 11665
    },
    {
      "epoch": 1.1124610900878905e-05,
      "step": 11665,
      "training_step_time": 17.403337478637695
    },
    {
      "epoch": 1.1125564575195313e-05,
      "model_forward_time": 5.602672100067139,
      "step": 11666
    },
    {
      "epoch": 1.1125564575195313e-05,
      "step": 11666,
      "training_step_time": 17.10113525390625
    },
    {
      "epoch": 1.1126518249511719e-05,
      "model_forward_time": 5.625893592834473,
      "step": 11667
    },
    {
      "epoch": 1.1126518249511719e-05,
      "step": 11667,
      "training_step_time": 17.145534992218018
    },
    {
      "epoch": 1.1127471923828124e-05,
      "model_forward_time": 5.630160331726074,
      "step": 11668
    },
    {
      "epoch": 1.1127471923828124e-05,
      "step": 11668,
      "training_step_time": 17.15234088897705
    },
    {
      "epoch": 1.1128425598144532e-05,
      "model_forward_time": 5.620559215545654,
      "step": 11669
    },
    {
      "epoch": 1.1128425598144532e-05,
      "step": 11669,
      "training_step_time": 17.105016231536865
    },
    {
      "epoch": 1.1129379272460937e-05,
      "grad_norm": 0.7324264049530029,
      "learning_rate": 1e-05,
      "loss": 0.051,
      "step": 11670
    },
    {
      "epoch": 1.1129379272460937e-05,
      "model_forward_time": 5.606413841247559,
      "step": 11670
    },
    {
      "dynamics_loss_avg": 0.04878844246268273,
      "epoch": 1.1129379272460937e-05,
      "step": 11670
    },
    {
      "action_loss_avg": 0.005657235626131296,
      "epoch": 1.1129379272460937e-05,
      "step": 11670
    },
    {
      "epoch": 1.1129379272460937e-05,
      "step": 11670,
      "training_step_time": 17.103774547576904
    },
    {
      "epoch": 1.1130332946777343e-05,
      "model_forward_time": 5.992534875869751,
      "step": 11671
    },
    {
      "epoch": 1.1130332946777343e-05,
      "step": 11671,
      "training_step_time": 17.48731780052185
    },
    {
      "epoch": 1.113128662109375e-05,
      "model_forward_time": 5.614573240280151,
      "step": 11672
    },
    {
      "epoch": 1.113128662109375e-05,
      "step": 11672,
      "training_step_time": 17.119649171829224
    },
    {
      "epoch": 1.1132240295410156e-05,
      "model_forward_time": 5.618963718414307,
      "step": 11673
    },
    {
      "epoch": 1.1132240295410156e-05,
      "step": 11673,
      "training_step_time": 17.12284755706787
    },
    {
      "epoch": 1.1133193969726562e-05,
      "model_forward_time": 5.606426239013672,
      "step": 11674
    },
    {
      "epoch": 1.1133193969726562e-05,
      "step": 11674,
      "training_step_time": 17.099146604537964
    },
    {
      "epoch": 1.113414764404297e-05,
      "model_forward_time": 5.616431713104248,
      "step": 11675
    },
    {
      "epoch": 1.113414764404297e-05,
      "step": 11675,
      "training_step_time": 17.093515634536743
    },
    {
      "epoch": 1.1135101318359375e-05,
      "model_forward_time": 5.971363544464111,
      "step": 11676
    },
    {
      "epoch": 1.1135101318359375e-05,
      "step": 11676,
      "training_step_time": 17.448528289794922
    },
    {
      "epoch": 1.1136054992675781e-05,
      "model_forward_time": 5.599447250366211,
      "step": 11677
    },
    {
      "epoch": 1.1136054992675781e-05,
      "step": 11677,
      "training_step_time": 17.0748507976532
    },
    {
      "epoch": 1.1137008666992188e-05,
      "model_forward_time": 5.6161699295043945,
      "step": 11678
    },
    {
      "epoch": 1.1137008666992188e-05,
      "step": 11678,
      "training_step_time": 17.09246516227722
    },
    {
      "epoch": 1.1137962341308594e-05,
      "model_forward_time": 5.621965169906616,
      "step": 11679
    },
    {
      "epoch": 1.1137962341308594e-05,
      "step": 11679,
      "training_step_time": 17.0979220867157
    },
    {
      "epoch": 1.1138916015625e-05,
      "grad_norm": 0.6121503114700317,
      "learning_rate": 1e-05,
      "loss": 0.0799,
      "step": 11680
    },
    {
      "epoch": 1.1138916015625e-05,
      "model_forward_time": 5.618526458740234,
      "step": 11680
    },
    {
      "dynamics_loss_avg": 0.07065444551408291,
      "epoch": 1.1138916015625e-05,
      "step": 11680
    },
    {
      "action_loss_avg": 0.010315257892943918,
      "epoch": 1.1138916015625e-05,
      "step": 11680
    },
    {
      "epoch": 1.1138916015625e-05,
      "step": 11680,
      "training_step_time": 17.100573301315308
    },
    {
      "epoch": 1.1139869689941405e-05,
      "model_forward_time": 5.612110137939453,
      "step": 11681
    },
    {
      "epoch": 1.1139869689941405e-05,
      "step": 11681,
      "training_step_time": 17.101436138153076
    },
    {
      "epoch": 1.1140823364257813e-05,
      "model_forward_time": 5.990948915481567,
      "step": 11682
    },
    {
      "epoch": 1.1140823364257813e-05,
      "step": 11682,
      "training_step_time": 17.46784257888794
    },
    {
      "epoch": 1.1141777038574219e-05,
      "model_forward_time": 5.605915784835815,
      "step": 11683
    },
    {
      "epoch": 1.1141777038574219e-05,
      "step": 11683,
      "training_step_time": 17.069209575653076
    },
    {
      "epoch": 1.1142730712890624e-05,
      "model_forward_time": 5.593121767044067,
      "step": 11684
    },
    {
      "epoch": 1.1142730712890624e-05,
      "step": 11684,
      "training_step_time": 17.042933702468872
    },
    {
      "epoch": 1.1143684387207032e-05,
      "model_forward_time": 5.593326091766357,
      "step": 11685
    },
    {
      "epoch": 1.1143684387207032e-05,
      "step": 11685,
      "training_step_time": 17.053504705429077
    },
    {
      "epoch": 1.1144638061523437e-05,
      "model_forward_time": 5.605551719665527,
      "step": 11686
    },
    {
      "epoch": 1.1144638061523437e-05,
      "step": 11686,
      "training_step_time": 17.047021865844727
    },
    {
      "epoch": 1.1145591735839843e-05,
      "model_forward_time": 5.98490047454834,
      "step": 11687
    },
    {
      "epoch": 1.1145591735839843e-05,
      "step": 11687,
      "training_step_time": 17.48316526412964
    },
    {
      "epoch": 1.114654541015625e-05,
      "model_forward_time": 5.608953475952148,
      "step": 11688
    },
    {
      "epoch": 1.114654541015625e-05,
      "step": 11688,
      "training_step_time": 17.08280658721924
    },
    {
      "epoch": 1.1147499084472656e-05,
      "model_forward_time": 5.592971324920654,
      "step": 11689
    },
    {
      "epoch": 1.1147499084472656e-05,
      "step": 11689,
      "training_step_time": 17.009167194366455
    },
    {
      "epoch": 1.1148452758789062e-05,
      "grad_norm": 0.5100674629211426,
      "learning_rate": 1e-05,
      "loss": 0.0776,
      "step": 11690
    },
    {
      "epoch": 1.1148452758789062e-05,
      "model_forward_time": 5.574767112731934,
      "step": 11690
    },
    {
      "dynamics_loss_avg": 0.07095162514597178,
      "epoch": 1.1148452758789062e-05,
      "step": 11690
    },
    {
      "action_loss_avg": 0.00887646961491555,
      "epoch": 1.1148452758789062e-05,
      "step": 11690
    },
    {
      "epoch": 1.1148452758789062e-05,
      "step": 11690,
      "training_step_time": 17.01810121536255
    },
    {
      "epoch": 1.114940643310547e-05,
      "model_forward_time": 5.577200651168823,
      "step": 11691
    },
    {
      "epoch": 1.114940643310547e-05,
      "step": 11691,
      "training_step_time": 16.990365505218506
    },
    {
      "epoch": 1.1150360107421875e-05,
      "model_forward_time": 5.581442356109619,
      "step": 11692
    },
    {
      "epoch": 1.1150360107421875e-05,
      "step": 11692,
      "training_step_time": 17.4179368019104
    },
    {
      "epoch": 1.1151313781738281e-05,
      "model_forward_time": 5.613247632980347,
      "step": 11693
    },
    {
      "epoch": 1.1151313781738281e-05,
      "step": 11693,
      "training_step_time": 17.11549210548401
    },
    {
      "epoch": 1.1152267456054688e-05,
      "model_forward_time": 5.623716354370117,
      "step": 11694
    },
    {
      "epoch": 1.1152267456054688e-05,
      "step": 11694,
      "training_step_time": 17.098049640655518
    },
    {
      "epoch": 1.1153221130371094e-05,
      "model_forward_time": 5.599425554275513,
      "step": 11695
    },
    {
      "epoch": 1.1153221130371094e-05,
      "step": 11695,
      "training_step_time": 17.050021648406982
    },
    {
      "epoch": 1.11541748046875e-05,
      "model_forward_time": 5.582854986190796,
      "step": 11696
    },
    {
      "epoch": 1.11541748046875e-05,
      "step": 11696,
      "training_step_time": 16.98528289794922
    },
    {
      "epoch": 1.1155128479003905e-05,
      "model_forward_time": 5.570424795150757,
      "step": 11697
    },
    {
      "epoch": 1.1155128479003905e-05,
      "step": 11697,
      "training_step_time": 17.00628161430359
    },
    {
      "epoch": 1.1156082153320313e-05,
      "model_forward_time": 5.933197736740112,
      "step": 11698
    },
    {
      "epoch": 1.1156082153320313e-05,
      "step": 11698,
      "training_step_time": 17.35206437110901
    },
    {
      "epoch": 1.1157035827636719e-05,
      "model_forward_time": 5.5910210609436035,
      "step": 11699
    },
    {
      "epoch": 1.1157035827636719e-05,
      "step": 11699,
      "training_step_time": 17.02206587791443
    },
    {
      "epoch": 1.1157989501953124e-05,
      "grad_norm": 0.7773701548576355,
      "learning_rate": 1e-05,
      "loss": 0.0759,
      "step": 11700
    },
    {
      "epoch": 1.1157989501953124e-05,
      "model_forward_time": 5.599616527557373,
      "step": 11700
    },
    {
      "dynamics_loss_avg": 0.06255739033222199,
      "epoch": 1.1157989501953124e-05,
      "step": 11700
    },
    {
      "action_loss_avg": 0.008707963582128286,
      "epoch": 1.1157989501953124e-05,
      "step": 11700
    },
    {
      "epoch": 1.1157989501953124e-05,
      "step": 11700,
      "training_step_time": 17.069364070892334
    },
    {
      "epoch": 1.1158943176269532e-05,
      "model_forward_time": 5.648360729217529,
      "step": 11701
    },
    {
      "epoch": 1.1158943176269532e-05,
      "step": 11701,
      "training_step_time": 17.157848358154297
    },
    {
      "epoch": 1.1159896850585937e-05,
      "model_forward_time": 5.650561094284058,
      "step": 11702
    },
    {
      "epoch": 1.1159896850585937e-05,
      "step": 11702,
      "training_step_time": 17.152485609054565
    },
    {
      "epoch": 1.1160850524902343e-05,
      "model_forward_time": 5.986316919326782,
      "step": 11703
    },
    {
      "epoch": 1.1160850524902343e-05,
      "step": 11703,
      "training_step_time": 17.458895444869995
    },
    {
      "epoch": 1.116180419921875e-05,
      "model_forward_time": 5.5921361446380615,
      "step": 11704
    },
    {
      "epoch": 1.116180419921875e-05,
      "step": 11704,
      "training_step_time": 17.024834632873535
    },
    {
      "epoch": 1.1162757873535156e-05,
      "model_forward_time": 5.580965518951416,
      "step": 11705
    },
    {
      "epoch": 1.1162757873535156e-05,
      "step": 11705,
      "training_step_time": 17.016728401184082
    },
    {
      "epoch": 1.1163711547851562e-05,
      "model_forward_time": 5.584507465362549,
      "step": 11706
    },
    {
      "epoch": 1.1163711547851562e-05,
      "step": 11706,
      "training_step_time": 17.002054691314697
    },
    {
      "epoch": 1.116466522216797e-05,
      "model_forward_time": 5.5802905559539795,
      "step": 11707
    },
    {
      "epoch": 1.116466522216797e-05,
      "step": 11707,
      "training_step_time": 17.01145839691162
    },
    {
      "epoch": 1.1165618896484375e-05,
      "model_forward_time": 5.585640907287598,
      "step": 11708
    },
    {
      "epoch": 1.1165618896484375e-05,
      "step": 11708,
      "training_step_time": 17.02600932121277
    },
    {
      "epoch": 1.1166572570800781e-05,
      "model_forward_time": 5.940513849258423,
      "step": 11709
    },
    {
      "epoch": 1.1166572570800781e-05,
      "step": 11709,
      "training_step_time": 17.382651567459106
    },
    {
      "epoch": 1.1167526245117188e-05,
      "grad_norm": 0.37458062171936035,
      "learning_rate": 1e-05,
      "loss": 0.0465,
      "step": 11710
    },
    {
      "epoch": 1.1167526245117188e-05,
      "model_forward_time": 5.60303807258606,
      "step": 11710
    },
    {
      "dynamics_loss_avg": 0.04799207616597414,
      "epoch": 1.1167526245117188e-05,
      "step": 11710
    },
    {
      "action_loss_avg": 0.004638371779583395,
      "epoch": 1.1167526245117188e-05,
      "step": 11710
    },
    {
      "epoch": 1.1167526245117188e-05,
      "step": 11710,
      "training_step_time": 17.101811408996582
    },
    {
      "epoch": 1.1168479919433594e-05,
      "model_forward_time": 5.637712240219116,
      "step": 11711
    },
    {
      "epoch": 1.1168479919433594e-05,
      "step": 11711,
      "training_step_time": 17.177860260009766
    },
    {
      "epoch": 1.116943359375e-05,
      "model_forward_time": 5.621523380279541,
      "step": 11712
    },
    {
      "epoch": 1.116943359375e-05,
      "step": 11712,
      "training_step_time": 17.12045669555664
    },
    {
      "epoch": 1.1170387268066405e-05,
      "model_forward_time": 5.611668109893799,
      "step": 11713
    },
    {
      "epoch": 1.1170387268066405e-05,
      "step": 11713,
      "training_step_time": 17.070916891098022
    },
    {
      "epoch": 1.1171340942382813e-05,
      "model_forward_time": 5.974724769592285,
      "step": 11714
    },
    {
      "epoch": 1.1171340942382813e-05,
      "step": 11714,
      "training_step_time": 17.449951887130737
    },
    {
      "epoch": 1.1172294616699219e-05,
      "model_forward_time": 5.617201566696167,
      "step": 11715
    },
    {
      "epoch": 1.1172294616699219e-05,
      "step": 11715,
      "training_step_time": 17.11816930770874
    },
    {
      "epoch": 1.1173248291015624e-05,
      "model_forward_time": 5.6202428340911865,
      "step": 11716
    },
    {
      "epoch": 1.1173248291015624e-05,
      "step": 11716,
      "training_step_time": 17.094587802886963
    },
    {
      "epoch": 1.1174201965332032e-05,
      "model_forward_time": 5.58514928817749,
      "step": 11717
    },
    {
      "epoch": 1.1174201965332032e-05,
      "step": 11717,
      "training_step_time": 17.013421535491943
    },
    {
      "epoch": 1.1175155639648437e-05,
      "model_forward_time": 5.58072566986084,
      "step": 11718
    },
    {
      "epoch": 1.1175155639648437e-05,
      "step": 11718,
      "training_step_time": 16.991178035736084
    },
    {
      "epoch": 1.1176109313964843e-05,
      "model_forward_time": 5.574732303619385,
      "step": 11719
    },
    {
      "epoch": 1.1176109313964843e-05,
      "step": 11719,
      "training_step_time": 16.98877787590027
    },
    {
      "epoch": 1.117706298828125e-05,
      "grad_norm": 1.0058375597000122,
      "learning_rate": 1e-05,
      "loss": 0.0834,
      "step": 11720
    },
    {
      "epoch": 1.117706298828125e-05,
      "model_forward_time": 5.91403341293335,
      "step": 11720
    },
    {
      "dynamics_loss_avg": 0.06758721843361855,
      "epoch": 1.117706298828125e-05,
      "step": 11720
    },
    {
      "action_loss_avg": 0.009263358917087317,
      "epoch": 1.117706298828125e-05,
      "step": 11720
    },
    {
      "epoch": 1.117706298828125e-05,
      "step": 11720,
      "training_step_time": 17.35456085205078
    },
    {
      "epoch": 1.1178016662597656e-05,
      "model_forward_time": 5.585314035415649,
      "step": 11721
    },
    {
      "epoch": 1.1178016662597656e-05,
      "step": 11721,
      "training_step_time": 17.017509698867798
    },
    {
      "epoch": 1.1178970336914062e-05,
      "model_forward_time": 5.592181921005249,
      "step": 11722
    },
    {
      "epoch": 1.1178970336914062e-05,
      "step": 11722,
      "training_step_time": 17.00905752182007
    },
    {
      "epoch": 1.117992401123047e-05,
      "model_forward_time": 5.578714609146118,
      "step": 11723
    },
    {
      "epoch": 1.117992401123047e-05,
      "step": 11723,
      "training_step_time": 17.013750791549683
    },
    {
      "epoch": 1.1180877685546875e-05,
      "model_forward_time": 5.598436117172241,
      "step": 11724
    },
    {
      "epoch": 1.1180877685546875e-05,
      "step": 11724,
      "training_step_time": 17.06360149383545
    },
    {
      "epoch": 1.1181831359863281e-05,
      "model_forward_time": 5.61356782913208,
      "step": 11725
    },
    {
      "epoch": 1.1181831359863281e-05,
      "step": 11725,
      "training_step_time": 17.48521614074707
    },
    {
      "epoch": 1.1182785034179688e-05,
      "model_forward_time": 5.622762441635132,
      "step": 11726
    },
    {
      "epoch": 1.1182785034179688e-05,
      "step": 11726,
      "training_step_time": 17.11054253578186
    },
    {
      "epoch": 1.1183738708496094e-05,
      "model_forward_time": 5.609059810638428,
      "step": 11727
    },
    {
      "epoch": 1.1183738708496094e-05,
      "step": 11727,
      "training_step_time": 17.081071853637695
    },
    {
      "epoch": 1.11846923828125e-05,
      "model_forward_time": 5.6033854484558105,
      "step": 11728
    },
    {
      "epoch": 1.11846923828125e-05,
      "step": 11728,
      "training_step_time": 17.036008596420288
    },
    {
      "epoch": 1.1185646057128905e-05,
      "model_forward_time": 5.5939812660217285,
      "step": 11729
    },
    {
      "epoch": 1.1185646057128905e-05,
      "step": 11729,
      "training_step_time": 17.028536558151245
    },
    {
      "epoch": 1.1186599731445313e-05,
      "grad_norm": 0.721878707408905,
      "learning_rate": 1e-05,
      "loss": 0.0798,
      "step": 11730
    },
    {
      "epoch": 1.1186599731445313e-05,
      "model_forward_time": 5.594350099563599,
      "step": 11730
    },
    {
      "dynamics_loss_avg": 0.07791991047561168,
      "epoch": 1.1186599731445313e-05,
      "step": 11730
    },
    {
      "action_loss_avg": 0.01076097646728158,
      "epoch": 1.1186599731445313e-05,
      "step": 11730
    },
    {
      "epoch": 1.1186599731445313e-05,
      "step": 11730,
      "training_step_time": 17.055229902267456
    },
    {
      "epoch": 1.1187553405761719e-05,
      "model_forward_time": 5.963247537612915,
      "step": 11731
    },
    {
      "epoch": 1.1187553405761719e-05,
      "step": 11731,
      "training_step_time": 17.437066316604614
    },
    {
      "epoch": 1.1188507080078124e-05,
      "model_forward_time": 5.607786417007446,
      "step": 11732
    },
    {
      "epoch": 1.1188507080078124e-05,
      "step": 11732,
      "training_step_time": 17.06748127937317
    },
    {
      "epoch": 1.1189460754394532e-05,
      "model_forward_time": 5.602711915969849,
      "step": 11733
    },
    {
      "epoch": 1.1189460754394532e-05,
      "step": 11733,
      "training_step_time": 17.08131194114685
    },
    {
      "epoch": 1.1190414428710937e-05,
      "model_forward_time": 5.5972325801849365,
      "step": 11734
    },
    {
      "epoch": 1.1190414428710937e-05,
      "step": 11734,
      "training_step_time": 17.053980588912964
    },
    {
      "epoch": 1.1191368103027343e-05,
      "model_forward_time": 5.5880560874938965,
      "step": 11735
    },
    {
      "epoch": 1.1191368103027343e-05,
      "step": 11735,
      "training_step_time": 17.040143966674805
    },
    {
      "epoch": 1.119232177734375e-05,
      "model_forward_time": 5.96893048286438,
      "step": 11736
    },
    {
      "epoch": 1.119232177734375e-05,
      "step": 11736,
      "training_step_time": 17.413334608078003
    },
    {
      "epoch": 1.1193275451660156e-05,
      "model_forward_time": 5.5899035930633545,
      "step": 11737
    },
    {
      "epoch": 1.1193275451660156e-05,
      "step": 11737,
      "training_step_time": 17.048035860061646
    },
    {
      "epoch": 1.1194229125976562e-05,
      "model_forward_time": 5.604208946228027,
      "step": 11738
    },
    {
      "epoch": 1.1194229125976562e-05,
      "step": 11738,
      "training_step_time": 17.098503351211548
    },
    {
      "epoch": 1.119518280029297e-05,
      "model_forward_time": 5.613252639770508,
      "step": 11739
    },
    {
      "epoch": 1.119518280029297e-05,
      "step": 11739,
      "training_step_time": 17.09512734413147
    },
    {
      "epoch": 1.1196136474609375e-05,
      "grad_norm": 0.6730413436889648,
      "learning_rate": 1e-05,
      "loss": 0.0678,
      "step": 11740
    },
    {
      "epoch": 1.1196136474609375e-05,
      "model_forward_time": 5.614371299743652,
      "step": 11740
    },
    {
      "dynamics_loss_avg": 0.052058884873986244,
      "epoch": 1.1196136474609375e-05,
      "step": 11740
    },
    {
      "action_loss_avg": 0.005050797539297491,
      "epoch": 1.1196136474609375e-05,
      "step": 11740
    },
    {
      "epoch": 1.1196136474609375e-05,
      "step": 11740,
      "training_step_time": 17.115262031555176
    },
    {
      "epoch": 1.1197090148925781e-05,
      "model_forward_time": 5.613421678543091,
      "step": 11741
    },
    {
      "epoch": 1.1197090148925781e-05,
      "step": 11741,
      "training_step_time": 17.092591285705566
    },
    {
      "epoch": 1.1198043823242188e-05,
      "model_forward_time": 5.951907396316528,
      "step": 11742
    },
    {
      "epoch": 1.1198043823242188e-05,
      "step": 11742,
      "training_step_time": 17.426398515701294
    },
    {
      "epoch": 1.1198997497558594e-05,
      "model_forward_time": 5.59802508354187,
      "step": 11743
    },
    {
      "epoch": 1.1198997497558594e-05,
      "step": 11743,
      "training_step_time": 17.048638105392456
    },
    {
      "epoch": 1.1199951171875e-05,
      "model_forward_time": 5.58598256111145,
      "step": 11744
    },
    {
      "epoch": 1.1199951171875e-05,
      "step": 11744,
      "training_step_time": 17.012171268463135
    },
    {
      "epoch": 1.1200904846191406e-05,
      "model_forward_time": 5.589265823364258,
      "step": 11745
    },
    {
      "epoch": 1.1200904846191406e-05,
      "step": 11745,
      "training_step_time": 17.041290760040283
    },
    {
      "epoch": 1.1201858520507813e-05,
      "model_forward_time": 5.592039585113525,
      "step": 11746
    },
    {
      "epoch": 1.1201858520507813e-05,
      "step": 11746,
      "training_step_time": 17.06037473678589
    },
    {
      "epoch": 1.1202812194824219e-05,
      "model_forward_time": 5.9829723834991455,
      "step": 11747
    },
    {
      "epoch": 1.1202812194824219e-05,
      "step": 11747,
      "training_step_time": 17.41968297958374
    },
    {
      "epoch": 1.1203765869140624e-05,
      "model_forward_time": 5.599164247512817,
      "step": 11748
    },
    {
      "epoch": 1.1203765869140624e-05,
      "step": 11748,
      "training_step_time": 17.03956127166748
    },
    {
      "epoch": 1.1204719543457032e-05,
      "model_forward_time": 5.599316358566284,
      "step": 11749
    },
    {
      "epoch": 1.1204719543457032e-05,
      "step": 11749,
      "training_step_time": 17.080716848373413
    },
    {
      "epoch": 1.1205673217773438e-05,
      "grad_norm": 0.3707599639892578,
      "learning_rate": 1e-05,
      "loss": 0.0722,
      "step": 11750
    },
    {
      "epoch": 1.1205673217773438e-05,
      "model_forward_time": 5.612210988998413,
      "step": 11750
    },
    {
      "dynamics_loss_avg": 0.06350671462714672,
      "epoch": 1.1205673217773438e-05,
      "step": 11750
    },
    {
      "action_loss_avg": 0.010457044350914657,
      "epoch": 1.1205673217773438e-05,
      "step": 11750
    },
    {
      "epoch": 1.1205673217773438e-05,
      "step": 11750,
      "training_step_time": 17.12410283088684
    },
    {
      "epoch": 1.1206626892089843e-05,
      "model_forward_time": 5.628549814224243,
      "step": 11751
    },
    {
      "epoch": 1.1206626892089843e-05,
      "step": 11751,
      "training_step_time": 17.144511699676514
    },
    {
      "epoch": 1.120758056640625e-05,
      "model_forward_time": 5.615706205368042,
      "step": 11752
    },
    {
      "epoch": 1.120758056640625e-05,
      "step": 11752,
      "training_step_time": 17.115681886672974
    },
    {
      "epoch": 1.1208534240722656e-05,
      "model_forward_time": 5.953892707824707,
      "step": 11753
    },
    {
      "epoch": 1.1208534240722656e-05,
      "step": 11753,
      "training_step_time": 17.434499502182007
    },
    {
      "epoch": 1.1209487915039062e-05,
      "model_forward_time": 5.604703187942505,
      "step": 11754
    },
    {
      "epoch": 1.1209487915039062e-05,
      "step": 11754,
      "training_step_time": 17.04186749458313
    },
    {
      "epoch": 1.121044158935547e-05,
      "model_forward_time": 5.592216491699219,
      "step": 11755
    },
    {
      "epoch": 1.121044158935547e-05,
      "step": 11755,
      "training_step_time": 17.050220727920532
    },
    {
      "epoch": 1.1211395263671875e-05,
      "model_forward_time": 5.607319593429565,
      "step": 11756
    },
    {
      "epoch": 1.1211395263671875e-05,
      "step": 11756,
      "training_step_time": 17.06111216545105
    },
    {
      "epoch": 1.1212348937988281e-05,
      "model_forward_time": 5.601565361022949,
      "step": 11757
    },
    {
      "epoch": 1.1212348937988281e-05,
      "step": 11757,
      "training_step_time": 17.070873022079468
    },
    {
      "epoch": 1.1213302612304687e-05,
      "model_forward_time": 5.60433554649353,
      "step": 11758
    },
    {
      "epoch": 1.1213302612304687e-05,
      "step": 11758,
      "training_step_time": 17.4655339717865
    },
    {
      "epoch": 1.1214256286621094e-05,
      "model_forward_time": 5.603358507156372,
      "step": 11759
    },
    {
      "epoch": 1.1214256286621094e-05,
      "step": 11759,
      "training_step_time": 17.06838607788086
    },
    {
      "epoch": 1.12152099609375e-05,
      "grad_norm": 0.9316847920417786,
      "learning_rate": 1e-05,
      "loss": 0.0703,
      "step": 11760
    },
    {
      "epoch": 1.12152099609375e-05,
      "model_forward_time": 5.606653213500977,
      "step": 11760
    },
    {
      "dynamics_loss_avg": 0.06023177169263363,
      "epoch": 1.12152099609375e-05,
      "step": 11760
    },
    {
      "action_loss_avg": 0.010045819729566575,
      "epoch": 1.12152099609375e-05,
      "step": 11760
    },
    {
      "epoch": 1.12152099609375e-05,
      "step": 11760,
      "training_step_time": 17.094130277633667
    },
    {
      "epoch": 1.1216163635253906e-05,
      "model_forward_time": 5.619412183761597,
      "step": 11761
    },
    {
      "epoch": 1.1216163635253906e-05,
      "step": 11761,
      "training_step_time": 17.10901927947998
    },
    {
      "epoch": 1.1217117309570313e-05,
      "model_forward_time": 5.612948656082153,
      "step": 11762
    },
    {
      "epoch": 1.1217117309570313e-05,
      "step": 11762,
      "training_step_time": 17.100736141204834
    },
    {
      "epoch": 1.1218070983886719e-05,
      "model_forward_time": 5.603765487670898,
      "step": 11763
    },
    {
      "epoch": 1.1218070983886719e-05,
      "step": 11763,
      "training_step_time": 17.09599256515503
    },
    {
      "epoch": 1.1219024658203124e-05,
      "model_forward_time": 5.955302715301514,
      "step": 11764
    },
    {
      "epoch": 1.1219024658203124e-05,
      "step": 11764,
      "training_step_time": 17.455010414123535
    },
    {
      "epoch": 1.1219978332519532e-05,
      "model_forward_time": 5.614595174789429,
      "step": 11765
    },
    {
      "epoch": 1.1219978332519532e-05,
      "step": 11765,
      "training_step_time": 17.09783387184143
    },
    {
      "epoch": 1.1220932006835938e-05,
      "model_forward_time": 5.609752655029297,
      "step": 11766
    },
    {
      "epoch": 1.1220932006835938e-05,
      "step": 11766,
      "training_step_time": 17.10335922241211
    },
    {
      "epoch": 1.1221885681152343e-05,
      "model_forward_time": 5.609914064407349,
      "step": 11767
    },
    {
      "epoch": 1.1221885681152343e-05,
      "step": 11767,
      "training_step_time": 17.08483386039734
    },
    {
      "epoch": 1.122283935546875e-05,
      "model_forward_time": 5.590025186538696,
      "step": 11768
    },
    {
      "epoch": 1.122283935546875e-05,
      "step": 11768,
      "training_step_time": 17.03322958946228
    },
    {
      "epoch": 1.1223793029785156e-05,
      "model_forward_time": 5.970489025115967,
      "step": 11769
    },
    {
      "epoch": 1.1223793029785156e-05,
      "step": 11769,
      "training_step_time": 17.412898063659668
    },
    {
      "epoch": 1.1224746704101562e-05,
      "grad_norm": 0.35239148139953613,
      "learning_rate": 1e-05,
      "loss": 0.0646,
      "step": 11770
    },
    {
      "epoch": 1.1224746704101562e-05,
      "model_forward_time": 5.588230848312378,
      "step": 11770
    },
    {
      "dynamics_loss_avg": 0.05679712072014809,
      "epoch": 1.1224746704101562e-05,
      "step": 11770
    },
    {
      "action_loss_avg": 0.006708619557321072,
      "epoch": 1.1224746704101562e-05,
      "step": 11770
    },
    {
      "epoch": 1.1224746704101562e-05,
      "step": 11770,
      "training_step_time": 17.038756132125854
    },
    {
      "epoch": 1.122570037841797e-05,
      "model_forward_time": 5.607078552246094,
      "step": 11771
    },
    {
      "epoch": 1.122570037841797e-05,
      "step": 11771,
      "training_step_time": 17.082092761993408
    },
    {
      "epoch": 1.1226654052734375e-05,
      "model_forward_time": 5.610834360122681,
      "step": 11772
    },
    {
      "epoch": 1.1226654052734375e-05,
      "step": 11772,
      "training_step_time": 17.091558694839478
    },
    {
      "epoch": 1.1227607727050781e-05,
      "model_forward_time": 5.6033031940460205,
      "step": 11773
    },
    {
      "epoch": 1.1227607727050781e-05,
      "step": 11773,
      "training_step_time": 17.063499212265015
    },
    {
      "epoch": 1.1228561401367187e-05,
      "model_forward_time": 5.6013360023498535,
      "step": 11774
    },
    {
      "epoch": 1.1228561401367187e-05,
      "step": 11774,
      "training_step_time": 17.066598415374756
    },
    {
      "epoch": 1.1229515075683594e-05,
      "model_forward_time": 5.966630220413208,
      "step": 11775
    },
    {
      "epoch": 1.1229515075683594e-05,
      "step": 11775,
      "training_step_time": 17.41963505744934
    },
    {
      "epoch": 1.123046875e-05,
      "model_forward_time": 5.594839096069336,
      "step": 11776
    },
    {
      "epoch": 1.123046875e-05,
      "step": 11776,
      "training_step_time": 17.047483682632446
    },
    {
      "epoch": 1.1231422424316406e-05,
      "model_forward_time": 5.596872806549072,
      "step": 11777
    },
    {
      "epoch": 1.1231422424316406e-05,
      "step": 11777,
      "training_step_time": 17.04188346862793
    },
    {
      "epoch": 1.1232376098632813e-05,
      "model_forward_time": 5.581523418426514,
      "step": 11778
    },
    {
      "epoch": 1.1232376098632813e-05,
      "step": 11778,
      "training_step_time": 17.029757261276245
    },
    {
      "epoch": 1.1233329772949219e-05,
      "model_forward_time": 5.611928224563599,
      "step": 11779
    },
    {
      "epoch": 1.1233329772949219e-05,
      "step": 11779,
      "training_step_time": 17.110411882400513
    },
    {
      "epoch": 1.1234283447265624e-05,
      "grad_norm": 0.7214230298995972,
      "learning_rate": 1e-05,
      "loss": 0.0589,
      "step": 11780
    },
    {
      "epoch": 1.1234283447265624e-05,
      "model_forward_time": 6.002525806427002,
      "step": 11780
    },
    {
      "dynamics_loss_avg": 0.05154905691742897,
      "epoch": 1.1234283447265624e-05,
      "step": 11780
    },
    {
      "action_loss_avg": 0.008173911017365754,
      "epoch": 1.1234283447265624e-05,
      "step": 11780
    },
    {
      "epoch": 1.1234283447265624e-05,
      "step": 11780,
      "training_step_time": 17.530956983566284
    },
    {
      "epoch": 1.1235237121582032e-05,
      "model_forward_time": 5.621192932128906,
      "step": 11781
    },
    {
      "epoch": 1.1235237121582032e-05,
      "step": 11781,
      "training_step_time": 17.07249927520752
    },
    {
      "epoch": 1.1236190795898438e-05,
      "model_forward_time": 5.601624011993408,
      "step": 11782
    },
    {
      "epoch": 1.1236190795898438e-05,
      "step": 11782,
      "training_step_time": 17.070315837860107
    },
    {
      "epoch": 1.1237144470214843e-05,
      "model_forward_time": 5.598909378051758,
      "step": 11783
    },
    {
      "epoch": 1.1237144470214843e-05,
      "step": 11783,
      "training_step_time": 17.06498670578003
    },
    {
      "epoch": 1.123809814453125e-05,
      "model_forward_time": 5.603261470794678,
      "step": 11784
    },
    {
      "epoch": 1.123809814453125e-05,
      "step": 11784,
      "training_step_time": 17.05123233795166
    },
    {
      "epoch": 1.1239051818847656e-05,
      "model_forward_time": 5.592488765716553,
      "step": 11785
    },
    {
      "epoch": 1.1239051818847656e-05,
      "step": 11785,
      "training_step_time": 17.046452045440674
    },
    {
      "epoch": 1.1240005493164062e-05,
      "model_forward_time": 5.933899164199829,
      "step": 11786
    },
    {
      "epoch": 1.1240005493164062e-05,
      "step": 11786,
      "training_step_time": 17.395297050476074
    },
    {
      "epoch": 1.124095916748047e-05,
      "model_forward_time": 5.607569456100464,
      "step": 11787
    },
    {
      "epoch": 1.124095916748047e-05,
      "step": 11787,
      "training_step_time": 17.098803520202637
    },
    {
      "epoch": 1.1241912841796875e-05,
      "model_forward_time": 5.627665281295776,
      "step": 11788
    },
    {
      "epoch": 1.1241912841796875e-05,
      "step": 11788,
      "training_step_time": 17.11568856239319
    },
    {
      "epoch": 1.1242866516113281e-05,
      "model_forward_time": 5.598131895065308,
      "step": 11789
    },
    {
      "epoch": 1.1242866516113281e-05,
      "step": 11789,
      "training_step_time": 17.052411794662476
    },
    {
      "epoch": 1.1243820190429687e-05,
      "grad_norm": 1.118326187133789,
      "learning_rate": 1e-05,
      "loss": 0.077,
      "step": 11790
    },
    {
      "epoch": 1.1243820190429687e-05,
      "model_forward_time": 5.59996223449707,
      "step": 11790
    },
    {
      "dynamics_loss_avg": 0.07027382533997298,
      "epoch": 1.1243820190429687e-05,
      "step": 11790
    },
    {
      "action_loss_avg": 0.009855131525546313,
      "epoch": 1.1243820190429687e-05,
      "step": 11790
    },
    {
      "epoch": 1.1243820190429687e-05,
      "step": 11790,
      "training_step_time": 17.04454731941223
    },
    {
      "epoch": 1.1244773864746094e-05,
      "model_forward_time": 5.58983039855957,
      "step": 11791
    },
    {
      "epoch": 1.1244773864746094e-05,
      "step": 11791,
      "training_step_time": 17.38273334503174
    },
    {
      "epoch": 1.12457275390625e-05,
      "model_forward_time": 5.595710039138794,
      "step": 11792
    },
    {
      "epoch": 1.12457275390625e-05,
      "step": 11792,
      "training_step_time": 17.042135000228882
    },
    {
      "epoch": 1.1246681213378906e-05,
      "model_forward_time": 5.599237442016602,
      "step": 11793
    },
    {
      "epoch": 1.1246681213378906e-05,
      "step": 11793,
      "training_step_time": 17.049623727798462
    },
    {
      "epoch": 1.1247634887695313e-05,
      "model_forward_time": 5.595724821090698,
      "step": 11794
    },
    {
      "epoch": 1.1247634887695313e-05,
      "step": 11794,
      "training_step_time": 17.070021390914917
    },
    {
      "epoch": 1.1248588562011719e-05,
      "model_forward_time": 5.616780519485474,
      "step": 11795
    },
    {
      "epoch": 1.1248588562011719e-05,
      "step": 11795,
      "training_step_time": 17.1253399848938
    },
    {
      "epoch": 1.1249542236328124e-05,
      "model_forward_time": 5.619587659835815,
      "step": 11796
    },
    {
      "epoch": 1.1249542236328124e-05,
      "step": 11796,
      "training_step_time": 17.096716165542603
    },
    {
      "epoch": 1.1250495910644532e-05,
      "model_forward_time": 5.946892976760864,
      "step": 11797
    },
    {
      "epoch": 1.1250495910644532e-05,
      "step": 11797,
      "training_step_time": 17.369208812713623
    },
    {
      "epoch": 1.1251449584960938e-05,
      "model_forward_time": 5.578961133956909,
      "step": 11798
    },
    {
      "epoch": 1.1251449584960938e-05,
      "step": 11798,
      "training_step_time": 16.98006820678711
    },
    {
      "epoch": 1.1252403259277343e-05,
      "model_forward_time": 5.584324359893799,
      "step": 11799
    },
    {
      "epoch": 1.1252403259277343e-05,
      "step": 11799,
      "training_step_time": 16.99316954612732
    },
    {
      "epoch": 1.125335693359375e-05,
      "grad_norm": 0.8391980528831482,
      "learning_rate": 1e-05,
      "loss": 0.0777,
      "step": 11800
    },
    {
      "epoch": 1.125335693359375e-05,
      "model_forward_time": 5.601034879684448,
      "step": 11800
    },
    {
      "dynamics_loss_avg": 0.06371572129428386,
      "epoch": 1.125335693359375e-05,
      "step": 11800
    },
    {
      "action_loss_avg": 0.009787431568838656,
      "epoch": 1.125335693359375e-05,
      "step": 11800
    },
    {
      "epoch": 1.125335693359375e-05,
      "step": 11800,
      "training_step_time": 17.05458116531372
    },
    {
      "epoch": 1.1254310607910156e-05,
      "model_forward_time": 5.597769737243652,
      "step": 11801
    },
    {
      "epoch": 1.1254310607910156e-05,
      "step": 11801,
      "training_step_time": 17.053333520889282
    },
    {
      "epoch": 1.1255264282226562e-05,
      "model_forward_time": 5.995190620422363,
      "step": 11802
    },
    {
      "epoch": 1.1255264282226562e-05,
      "step": 11802,
      "training_step_time": 17.453668355941772
    },
    {
      "epoch": 1.125621795654297e-05,
      "model_forward_time": 5.59746527671814,
      "step": 11803
    },
    {
      "epoch": 1.125621795654297e-05,
      "step": 11803,
      "training_step_time": 17.037883520126343
    },
    {
      "epoch": 1.1257171630859375e-05,
      "model_forward_time": 5.601629257202148,
      "step": 11804
    },
    {
      "epoch": 1.1257171630859375e-05,
      "step": 11804,
      "training_step_time": 17.068713426589966
    },
    {
      "epoch": 1.1258125305175781e-05,
      "model_forward_time": 5.602481126785278,
      "step": 11805
    },
    {
      "epoch": 1.1258125305175781e-05,
      "step": 11805,
      "training_step_time": 17.092777729034424
    },
    {
      "epoch": 1.1259078979492187e-05,
      "model_forward_time": 5.621453523635864,
      "step": 11806
    },
    {
      "epoch": 1.1259078979492187e-05,
      "step": 11806,
      "training_step_time": 17.116516590118408
    },
    {
      "epoch": 1.1260032653808594e-05,
      "model_forward_time": 5.618617057800293,
      "step": 11807
    },
    {
      "epoch": 1.1260032653808594e-05,
      "step": 11807,
      "training_step_time": 17.100911378860474
    },
    {
      "epoch": 1.1260986328125e-05,
      "model_forward_time": 5.964866876602173,
      "step": 11808
    },
    {
      "epoch": 1.1260986328125e-05,
      "step": 11808,
      "training_step_time": 17.445358276367188
    },
    {
      "epoch": 1.1261940002441406e-05,
      "model_forward_time": 5.596932411193848,
      "step": 11809
    },
    {
      "epoch": 1.1261940002441406e-05,
      "step": 11809,
      "training_step_time": 17.05016803741455
    },
    {
      "epoch": 1.1262893676757813e-05,
      "grad_norm": 0.6438797116279602,
      "learning_rate": 1e-05,
      "loss": 0.0618,
      "step": 11810
    },
    {
      "epoch": 1.1262893676757813e-05,
      "model_forward_time": 5.594477891921997,
      "step": 11810
    },
    {
      "dynamics_loss_avg": 0.05991906160488725,
      "epoch": 1.1262893676757813e-05,
      "step": 11810
    },
    {
      "action_loss_avg": 0.007647747104056179,
      "epoch": 1.1262893676757813e-05,
      "step": 11810
    },
    {
      "epoch": 1.1262893676757813e-05,
      "step": 11810,
      "training_step_time": 17.03883671760559
    },
    {
      "epoch": 1.1263847351074219e-05,
      "model_forward_time": 5.597419261932373,
      "step": 11811
    },
    {
      "epoch": 1.1263847351074219e-05,
      "step": 11811,
      "training_step_time": 17.036123752593994
    },
    {
      "epoch": 1.1264801025390624e-05,
      "model_forward_time": 5.5894856452941895,
      "step": 11812
    },
    {
      "epoch": 1.1264801025390624e-05,
      "step": 11812,
      "training_step_time": 17.036036491394043
    },
    {
      "epoch": 1.1265754699707032e-05,
      "model_forward_time": 5.990621328353882,
      "step": 11813
    },
    {
      "epoch": 1.1265754699707032e-05,
      "step": 11813,
      "training_step_time": 17.44737482070923
    },
    {
      "epoch": 1.1266708374023438e-05,
      "model_forward_time": 5.605528354644775,
      "step": 11814
    },
    {
      "epoch": 1.1266708374023438e-05,
      "step": 11814,
      "training_step_time": 17.096291542053223
    },
    {
      "epoch": 1.1267662048339843e-05,
      "model_forward_time": 5.622231483459473,
      "step": 11815
    },
    {
      "epoch": 1.1267662048339843e-05,
      "step": 11815,
      "training_step_time": 17.113825798034668
    },
    {
      "epoch": 1.126861572265625e-05,
      "model_forward_time": 5.6115803718566895,
      "step": 11816
    },
    {
      "epoch": 1.126861572265625e-05,
      "step": 11816,
      "training_step_time": 17.086809396743774
    },
    {
      "epoch": 1.1269569396972656e-05,
      "model_forward_time": 5.59619665145874,
      "step": 11817
    },
    {
      "epoch": 1.1269569396972656e-05,
      "step": 11817,
      "training_step_time": 17.022175073623657
    },
    {
      "epoch": 1.1270523071289062e-05,
      "model_forward_time": 5.6319169998168945,
      "step": 11818
    },
    {
      "epoch": 1.1270523071289062e-05,
      "step": 11818,
      "training_step_time": 17.064876317977905
    },
    {
      "epoch": 1.127147674560547e-05,
      "model_forward_time": 4.629394769668579,
      "step": 11819
    },
    {
      "epoch": 1.127147674560547e-05,
      "step": 11819,
      "training_step_time": 13.113468885421753
    },
    {
      "epoch": 1.1272430419921875e-05,
      "grad_norm": 1.0509687662124634,
      "learning_rate": 1e-05,
      "loss": 0.0844,
      "step": 11820
    },
    {
      "epoch": 1.1272430419921875e-05,
      "model_forward_time": 5.583616256713867,
      "step": 11820
    },
    {
      "dynamics_loss_avg": 0.07188615687191487,
      "epoch": 1.1272430419921875e-05,
      "step": 11820
    },
    {
      "action_loss_avg": 0.009676407184451818,
      "epoch": 1.1272430419921875e-05,
      "step": 11820
    },
    {
      "epoch": 1.1272430419921875e-05,
      "step": 11820,
      "training_step_time": 17.00216269493103
    },
    {
      "epoch": 1.1273384094238281e-05,
      "model_forward_time": 5.593213796615601,
      "step": 11821
    },
    {
      "epoch": 1.1273384094238281e-05,
      "step": 11821,
      "training_step_time": 17.05594754219055
    },
    {
      "epoch": 1.1274337768554687e-05,
      "model_forward_time": 5.609683275222778,
      "step": 11822
    },
    {
      "epoch": 1.1274337768554687e-05,
      "step": 11822,
      "training_step_time": 17.111454486846924
    },
    {
      "epoch": 1.1275291442871094e-05,
      "model_forward_time": 5.62147331237793,
      "step": 11823
    },
    {
      "epoch": 1.1275291442871094e-05,
      "step": 11823,
      "training_step_time": 17.12860345840454
    },
    {
      "epoch": 1.12762451171875e-05,
      "model_forward_time": 5.623278617858887,
      "step": 11824
    },
    {
      "epoch": 1.12762451171875e-05,
      "step": 11824,
      "training_step_time": 17.107197284698486
    },
    {
      "epoch": 1.1277198791503906e-05,
      "model_forward_time": 5.965279817581177,
      "step": 11825
    },
    {
      "epoch": 1.1277198791503906e-05,
      "step": 11825,
      "training_step_time": 17.387134313583374
    },
    {
      "epoch": 1.1278152465820313e-05,
      "model_forward_time": 5.5854651927948,
      "step": 11826
    },
    {
      "epoch": 1.1278152465820313e-05,
      "step": 11826,
      "training_step_time": 17.020707368850708
    },
    {
      "epoch": 1.1279106140136719e-05,
      "model_forward_time": 5.575624942779541,
      "step": 11827
    },
    {
      "epoch": 1.1279106140136719e-05,
      "step": 11827,
      "training_step_time": 17.01777672767639
    },
    {
      "epoch": 1.1280059814453124e-05,
      "model_forward_time": 5.578822135925293,
      "step": 11828
    },
    {
      "epoch": 1.1280059814453124e-05,
      "step": 11828,
      "training_step_time": 17.013350009918213
    },
    {
      "epoch": 1.1281013488769532e-05,
      "model_forward_time": 5.58656907081604,
      "step": 11829
    },
    {
      "epoch": 1.1281013488769532e-05,
      "step": 11829,
      "training_step_time": 17.02644944190979
    },
    {
      "epoch": 1.1281967163085938e-05,
      "grad_norm": 0.4126558303833008,
      "learning_rate": 1e-05,
      "loss": 0.0804,
      "step": 11830
    },
    {
      "epoch": 1.1281967163085938e-05,
      "model_forward_time": 5.947547435760498,
      "step": 11830
    },
    {
      "dynamics_loss_avg": 0.07162454277276993,
      "epoch": 1.1281967163085938e-05,
      "step": 11830
    },
    {
      "action_loss_avg": 0.011248782486654818,
      "epoch": 1.1281967163085938e-05,
      "step": 11830
    },
    {
      "epoch": 1.1281967163085938e-05,
      "step": 11830,
      "training_step_time": 17.378007173538208
    },
    {
      "epoch": 1.1282920837402343e-05,
      "model_forward_time": 5.603231906890869,
      "step": 11831
    },
    {
      "epoch": 1.1282920837402343e-05,
      "step": 11831,
      "training_step_time": 17.06975269317627
    },
    {
      "epoch": 1.128387451171875e-05,
      "model_forward_time": 5.602602481842041,
      "step": 11832
    },
    {
      "epoch": 1.128387451171875e-05,
      "step": 11832,
      "training_step_time": 17.06393575668335
    },
    {
      "epoch": 1.1284828186035156e-05,
      "model_forward_time": 5.599677801132202,
      "step": 11833
    },
    {
      "epoch": 1.1284828186035156e-05,
      "step": 11833,
      "training_step_time": 17.051504135131836
    },
    {
      "epoch": 1.1285781860351562e-05,
      "model_forward_time": 5.5878143310546875,
      "step": 11834
    },
    {
      "epoch": 1.1285781860351562e-05,
      "step": 11834,
      "training_step_time": 17.026062488555908
    },
    {
      "epoch": 1.128673553466797e-05,
      "model_forward_time": 5.981400966644287,
      "step": 11835
    },
    {
      "epoch": 1.128673553466797e-05,
      "step": 11835,
      "training_step_time": 17.4440975189209
    },
    {
      "epoch": 1.1287689208984375e-05,
      "model_forward_time": 5.597630262374878,
      "step": 11836
    },
    {
      "epoch": 1.1287689208984375e-05,
      "step": 11836,
      "training_step_time": 17.05981135368347
    },
    {
      "epoch": 1.1288642883300781e-05,
      "model_forward_time": 5.594665050506592,
      "step": 11837
    },
    {
      "epoch": 1.1288642883300781e-05,
      "step": 11837,
      "training_step_time": 17.054856061935425
    },
    {
      "epoch": 1.1289596557617187e-05,
      "model_forward_time": 5.606736660003662,
      "step": 11838
    },
    {
      "epoch": 1.1289596557617187e-05,
      "step": 11838,
      "training_step_time": 17.081562519073486
    },
    {
      "epoch": 1.1290550231933594e-05,
      "model_forward_time": 5.616212368011475,
      "step": 11839
    },
    {
      "epoch": 1.1290550231933594e-05,
      "step": 11839,
      "training_step_time": 17.104462146759033
    },
    {
      "epoch": 1.129150390625e-05,
      "grad_norm": 1.1810641288757324,
      "learning_rate": 1e-05,
      "loss": 0.0933,
      "step": 11840
    },
    {
      "epoch": 1.129150390625e-05,
      "model_forward_time": 5.606902360916138,
      "step": 11840
    },
    {
      "dynamics_loss_avg": 0.07758756428956985,
      "epoch": 1.129150390625e-05,
      "step": 11840
    },
    {
      "action_loss_avg": 0.011151447240263224,
      "epoch": 1.129150390625e-05,
      "step": 11840
    },
    {
      "epoch": 1.129150390625e-05,
      "step": 11840,
      "training_step_time": 17.116127967834473
    },
    {
      "epoch": 1.1292457580566406e-05,
      "model_forward_time": 5.984185695648193,
      "step": 11841
    },
    {
      "epoch": 1.1292457580566406e-05,
      "step": 11841,
      "training_step_time": 17.48552370071411
    },
    {
      "epoch": 1.1293411254882813e-05,
      "model_forward_time": 5.6161699295043945,
      "step": 11842
    },
    {
      "epoch": 1.1293411254882813e-05,
      "step": 11842,
      "training_step_time": 17.12055253982544
    },
    {
      "epoch": 1.1294364929199219e-05,
      "model_forward_time": 5.627813816070557,
      "step": 11843
    },
    {
      "epoch": 1.1294364929199219e-05,
      "step": 11843,
      "training_step_time": 17.13154435157776
    },
    {
      "epoch": 1.1295318603515624e-05,
      "model_forward_time": 5.616267681121826,
      "step": 11844
    },
    {
      "epoch": 1.1295318603515624e-05,
      "step": 11844,
      "training_step_time": 17.107239246368408
    },
    {
      "epoch": 1.1296272277832032e-05,
      "model_forward_time": 5.612744092941284,
      "step": 11845
    },
    {
      "epoch": 1.1296272277832032e-05,
      "step": 11845,
      "training_step_time": 17.093276977539062
    },
    {
      "epoch": 1.1297225952148438e-05,
      "model_forward_time": 5.998266696929932,
      "step": 11846
    },
    {
      "epoch": 1.1297225952148438e-05,
      "step": 11846,
      "training_step_time": 17.46836566925049
    },
    {
      "epoch": 1.1298179626464843e-05,
      "model_forward_time": 5.611108779907227,
      "step": 11847
    },
    {
      "epoch": 1.1298179626464843e-05,
      "step": 11847,
      "training_step_time": 17.09148144721985
    },
    {
      "epoch": 1.129913330078125e-05,
      "model_forward_time": 5.62108039855957,
      "step": 11848
    },
    {
      "epoch": 1.129913330078125e-05,
      "step": 11848,
      "training_step_time": 17.120909690856934
    },
    {
      "epoch": 1.1300086975097656e-05,
      "model_forward_time": 5.622373580932617,
      "step": 11849
    },
    {
      "epoch": 1.1300086975097656e-05,
      "step": 11849,
      "training_step_time": 17.11973762512207
    },
    {
      "epoch": 1.1301040649414062e-05,
      "grad_norm": 0.8577178120613098,
      "learning_rate": 1e-05,
      "loss": 0.0894,
      "step": 11850
    },
    {
      "epoch": 1.1301040649414062e-05,
      "model_forward_time": 5.6201066970825195,
      "step": 11850
    },
    {
      "dynamics_loss_avg": 0.0824596133083105,
      "epoch": 1.1301040649414062e-05,
      "step": 11850
    },
    {
      "action_loss_avg": 0.011312800040468574,
      "epoch": 1.1301040649414062e-05,
      "step": 11850
    },
    {
      "epoch": 1.1301040649414062e-05,
      "step": 11850,
      "training_step_time": 17.120387077331543
    },
    {
      "epoch": 1.130199432373047e-05,
      "model_forward_time": 5.639843225479126,
      "step": 11851
    },
    {
      "epoch": 1.130199432373047e-05,
      "step": 11851,
      "training_step_time": 17.137571811676025
    },
    {
      "epoch": 1.1302947998046875e-05,
      "model_forward_time": 5.965365886688232,
      "step": 11852
    },
    {
      "epoch": 1.1302947998046875e-05,
      "step": 11852,
      "training_step_time": 17.460460424423218
    },
    {
      "epoch": 1.1303901672363281e-05,
      "model_forward_time": 5.609117031097412,
      "step": 11853
    },
    {
      "epoch": 1.1303901672363281e-05,
      "step": 11853,
      "training_step_time": 17.097652435302734
    },
    {
      "epoch": 1.1304855346679687e-05,
      "model_forward_time": 5.612409591674805,
      "step": 11854
    },
    {
      "epoch": 1.1304855346679687e-05,
      "step": 11854,
      "training_step_time": 17.08357286453247
    },
    {
      "epoch": 1.1305809020996094e-05,
      "model_forward_time": 5.586590528488159,
      "step": 11855
    },
    {
      "epoch": 1.1305809020996094e-05,
      "step": 11855,
      "training_step_time": 16.999738693237305
    },
    {
      "epoch": 1.13067626953125e-05,
      "model_forward_time": 5.578322649002075,
      "step": 11856
    },
    {
      "epoch": 1.13067626953125e-05,
      "step": 11856,
      "training_step_time": 17.006904363632202
    },
    {
      "epoch": 1.1307716369628906e-05,
      "model_forward_time": 5.562142372131348,
      "step": 11857
    },
    {
      "epoch": 1.1307716369628906e-05,
      "step": 11857,
      "training_step_time": 17.360703706741333
    },
    {
      "epoch": 1.1308670043945313e-05,
      "model_forward_time": 5.57019829750061,
      "step": 11858
    },
    {
      "epoch": 1.1308670043945313e-05,
      "step": 11858,
      "training_step_time": 16.98657536506653
    },
    {
      "epoch": 1.1309623718261719e-05,
      "model_forward_time": 5.5643532276153564,
      "step": 11859
    },
    {
      "epoch": 1.1309623718261719e-05,
      "step": 11859,
      "training_step_time": 16.980668783187866
    },
    {
      "epoch": 1.1310577392578124e-05,
      "grad_norm": 1.0989863872528076,
      "learning_rate": 1e-05,
      "loss": 0.0703,
      "step": 11860
    },
    {
      "epoch": 1.1310577392578124e-05,
      "model_forward_time": 5.575484752655029,
      "step": 11860
    },
    {
      "dynamics_loss_avg": 0.05924023687839508,
      "epoch": 1.1310577392578124e-05,
      "step": 11860
    },
    {
      "action_loss_avg": 0.009005459793843328,
      "epoch": 1.1310577392578124e-05,
      "step": 11860
    },
    {
      "epoch": 1.1310577392578124e-05,
      "step": 11860,
      "training_step_time": 16.99915909767151
    },
    {
      "epoch": 1.1311531066894532e-05,
      "model_forward_time": 5.573652505874634,
      "step": 11861
    },
    {
      "epoch": 1.1311531066894532e-05,
      "step": 11861,
      "training_step_time": 17.016462326049805
    },
    {
      "epoch": 1.1312484741210938e-05,
      "model_forward_time": 5.576826095581055,
      "step": 11862
    },
    {
      "epoch": 1.1312484741210938e-05,
      "step": 11862,
      "training_step_time": 17.012834548950195
    },
    {
      "epoch": 1.1313438415527343e-05,
      "model_forward_time": 5.948466777801514,
      "step": 11863
    },
    {
      "epoch": 1.1313438415527343e-05,
      "step": 11863,
      "training_step_time": 17.420177936553955
    },
    {
      "epoch": 1.131439208984375e-05,
      "model_forward_time": 5.608535528182983,
      "step": 11864
    },
    {
      "epoch": 1.131439208984375e-05,
      "step": 11864,
      "training_step_time": 17.076834440231323
    },
    {
      "epoch": 1.1315345764160156e-05,
      "model_forward_time": 5.598896265029907,
      "step": 11865
    },
    {
      "epoch": 1.1315345764160156e-05,
      "step": 11865,
      "training_step_time": 17.076274394989014
    },
    {
      "epoch": 1.1316299438476562e-05,
      "model_forward_time": 5.600100755691528,
      "step": 11866
    },
    {
      "epoch": 1.1316299438476562e-05,
      "step": 11866,
      "training_step_time": 17.03544282913208
    },
    {
      "epoch": 1.131725311279297e-05,
      "model_forward_time": 5.609532594680786,
      "step": 11867
    },
    {
      "epoch": 1.131725311279297e-05,
      "step": 11867,
      "training_step_time": 17.036201238632202
    },
    {
      "epoch": 1.1318206787109375e-05,
      "model_forward_time": 5.976211786270142,
      "step": 11868
    },
    {
      "epoch": 1.1318206787109375e-05,
      "step": 11868,
      "training_step_time": 17.421303749084473
    },
    {
      "epoch": 1.1319160461425781e-05,
      "model_forward_time": 5.611706495285034,
      "step": 11869
    },
    {
      "epoch": 1.1319160461425781e-05,
      "step": 11869,
      "training_step_time": 17.125219583511353
    },
    {
      "epoch": 1.1320114135742187e-05,
      "grad_norm": 0.5040239095687866,
      "learning_rate": 1e-05,
      "loss": 0.0523,
      "step": 11870
    },
    {
      "epoch": 1.1320114135742187e-05,
      "model_forward_time": 5.630151748657227,
      "step": 11870
    },
    {
      "dynamics_loss_avg": 0.043298795260488986,
      "epoch": 1.1320114135742187e-05,
      "step": 11870
    },
    {
      "action_loss_avg": 0.005069563048891723,
      "epoch": 1.1320114135742187e-05,
      "step": 11870
    },
    {
      "epoch": 1.1320114135742187e-05,
      "step": 11870,
      "training_step_time": 17.14849281311035
    },
    {
      "epoch": 1.1321067810058594e-05,
      "model_forward_time": 5.615683078765869,
      "step": 11871
    },
    {
      "epoch": 1.1321067810058594e-05,
      "step": 11871,
      "training_step_time": 17.112305641174316
    },
    {
      "epoch": 1.1322021484375e-05,
      "model_forward_time": 5.6091649532318115,
      "step": 11872
    },
    {
      "epoch": 1.1322021484375e-05,
      "step": 11872,
      "training_step_time": 17.080184936523438
    },
    {
      "epoch": 1.1322975158691406e-05,
      "model_forward_time": 5.605414390563965,
      "step": 11873
    },
    {
      "epoch": 1.1322975158691406e-05,
      "step": 11873,
      "training_step_time": 17.072961568832397
    },
    {
      "epoch": 1.1323928833007813e-05,
      "model_forward_time": 5.947575569152832,
      "step": 11874
    },
    {
      "epoch": 1.1323928833007813e-05,
      "step": 11874,
      "training_step_time": 17.4156551361084
    },
    {
      "epoch": 1.1324882507324219e-05,
      "model_forward_time": 5.60033655166626,
      "step": 11875
    },
    {
      "epoch": 1.1324882507324219e-05,
      "step": 11875,
      "training_step_time": 17.065277576446533
    },
    {
      "epoch": 1.1325836181640624e-05,
      "model_forward_time": 5.590027093887329,
      "step": 11876
    },
    {
      "epoch": 1.1325836181640624e-05,
      "step": 11876,
      "training_step_time": 17.04764175415039
    },
    {
      "epoch": 1.1326789855957032e-05,
      "model_forward_time": 5.597966909408569,
      "step": 11877
    },
    {
      "epoch": 1.1326789855957032e-05,
      "step": 11877,
      "training_step_time": 17.058730602264404
    },
    {
      "epoch": 1.1327743530273438e-05,
      "model_forward_time": 5.598068475723267,
      "step": 11878
    },
    {
      "epoch": 1.1327743530273438e-05,
      "step": 11878,
      "training_step_time": 17.09177327156067
    },
    {
      "epoch": 1.1328697204589843e-05,
      "model_forward_time": 5.993295431137085,
      "step": 11879
    },
    {
      "epoch": 1.1328697204589843e-05,
      "step": 11879,
      "training_step_time": 17.458401679992676
    },
    {
      "epoch": 1.132965087890625e-05,
      "grad_norm": 0.7180252075195312,
      "learning_rate": 1e-05,
      "loss": 0.057,
      "step": 11880
    },
    {
      "epoch": 1.132965087890625e-05,
      "model_forward_time": 5.597289562225342,
      "step": 11880
    },
    {
      "dynamics_loss_avg": 0.05418132729828358,
      "epoch": 1.132965087890625e-05,
      "step": 11880
    },
    {
      "action_loss_avg": 0.008173862565308809,
      "epoch": 1.132965087890625e-05,
      "step": 11880
    },
    {
      "epoch": 1.132965087890625e-05,
      "step": 11880,
      "training_step_time": 17.0452663898468
    },
    {
      "epoch": 1.1330604553222656e-05,
      "model_forward_time": 5.5948991775512695,
      "step": 11881
    },
    {
      "epoch": 1.1330604553222656e-05,
      "step": 11881,
      "training_step_time": 17.042738437652588
    },
    {
      "epoch": 1.1331558227539062e-05,
      "model_forward_time": 5.5942702293396,
      "step": 11882
    },
    {
      "epoch": 1.1331558227539062e-05,
      "step": 11882,
      "training_step_time": 17.02968692779541
    },
    {
      "epoch": 1.1332511901855468e-05,
      "model_forward_time": 5.60156512260437,
      "step": 11883
    },
    {
      "epoch": 1.1332511901855468e-05,
      "step": 11883,
      "training_step_time": 17.029682636260986
    },
    {
      "epoch": 1.1333465576171875e-05,
      "model_forward_time": 5.602711200714111,
      "step": 11884
    },
    {
      "epoch": 1.1333465576171875e-05,
      "step": 11884,
      "training_step_time": 17.094088792800903
    },
    {
      "epoch": 1.1334419250488281e-05,
      "model_forward_time": 5.975074529647827,
      "step": 11885
    },
    {
      "epoch": 1.1334419250488281e-05,
      "step": 11885,
      "training_step_time": 17.469635248184204
    },
    {
      "epoch": 1.1335372924804687e-05,
      "model_forward_time": 5.614778518676758,
      "step": 11886
    },
    {
      "epoch": 1.1335372924804687e-05,
      "step": 11886,
      "training_step_time": 17.11726403236389
    },
    {
      "epoch": 1.1336326599121094e-05,
      "model_forward_time": 5.635533094406128,
      "step": 11887
    },
    {
      "epoch": 1.1336326599121094e-05,
      "step": 11887,
      "training_step_time": 17.13651990890503
    },
    {
      "epoch": 1.13372802734375e-05,
      "model_forward_time": 5.617652416229248,
      "step": 11888
    },
    {
      "epoch": 1.13372802734375e-05,
      "step": 11888,
      "training_step_time": 17.106668949127197
    },
    {
      "epoch": 1.1338233947753906e-05,
      "model_forward_time": 5.610550165176392,
      "step": 11889
    },
    {
      "epoch": 1.1338233947753906e-05,
      "step": 11889,
      "training_step_time": 17.116680145263672
    },
    {
      "epoch": 1.1339187622070313e-05,
      "grad_norm": 0.9557488560676575,
      "learning_rate": 1e-05,
      "loss": 0.0618,
      "step": 11890
    },
    {
      "epoch": 1.1339187622070313e-05,
      "model_forward_time": 5.617410898208618,
      "step": 11890
    },
    {
      "dynamics_loss_avg": 0.05016512461006641,
      "epoch": 1.1339187622070313e-05,
      "step": 11890
    },
    {
      "action_loss_avg": 0.005591931147500872,
      "epoch": 1.1339187622070313e-05,
      "step": 11890
    },
    {
      "epoch": 1.1339187622070313e-05,
      "step": 11890,
      "training_step_time": 17.482893705368042
    },
    {
      "epoch": 1.1340141296386719e-05,
      "model_forward_time": 5.623014450073242,
      "step": 11891
    },
    {
      "epoch": 1.1340141296386719e-05,
      "step": 11891,
      "training_step_time": 17.12133479118347
    },
    {
      "epoch": 1.1341094970703124e-05,
      "model_forward_time": 5.618680477142334,
      "step": 11892
    },
    {
      "epoch": 1.1341094970703124e-05,
      "step": 11892,
      "training_step_time": 17.122387170791626
    },
    {
      "epoch": 1.1342048645019532e-05,
      "model_forward_time": 5.609685897827148,
      "step": 11893
    },
    {
      "epoch": 1.1342048645019532e-05,
      "step": 11893,
      "training_step_time": 17.08562421798706
    },
    {
      "epoch": 1.1343002319335938e-05,
      "model_forward_time": 5.588809013366699,
      "step": 11894
    },
    {
      "epoch": 1.1343002319335938e-05,
      "step": 11894,
      "training_step_time": 17.002305269241333
    },
    {
      "epoch": 1.1343955993652343e-05,
      "model_forward_time": 5.569302082061768,
      "step": 11895
    },
    {
      "epoch": 1.1343955993652343e-05,
      "step": 11895,
      "training_step_time": 17.01776099205017
    },
    {
      "epoch": 1.134490966796875e-05,
      "model_forward_time": 5.907509088516235,
      "step": 11896
    },
    {
      "epoch": 1.134490966796875e-05,
      "step": 11896,
      "training_step_time": 17.324089527130127
    },
    {
      "epoch": 1.1345863342285156e-05,
      "model_forward_time": 5.575992107391357,
      "step": 11897
    },
    {
      "epoch": 1.1345863342285156e-05,
      "step": 11897,
      "training_step_time": 16.995546579360962
    },
    {
      "epoch": 1.1346817016601562e-05,
      "model_forward_time": 5.5823822021484375,
      "step": 11898
    },
    {
      "epoch": 1.1346817016601562e-05,
      "step": 11898,
      "training_step_time": 17.01692247390747
    },
    {
      "epoch": 1.1347770690917968e-05,
      "model_forward_time": 5.593888998031616,
      "step": 11899
    },
    {
      "epoch": 1.1347770690917968e-05,
      "step": 11899,
      "training_step_time": 17.04094958305359
    },
    {
      "epoch": 1.1348724365234375e-05,
      "grad_norm": 1.0899213552474976,
      "learning_rate": 1e-05,
      "loss": 0.0503,
      "step": 11900
    },
    {
      "epoch": 1.1348724365234375e-05,
      "model_forward_time": 5.5946431159973145,
      "step": 11900
    },
    {
      "dynamics_loss_avg": 0.04661761876195669,
      "epoch": 1.1348724365234375e-05,
      "step": 11900
    },
    {
      "action_loss_avg": 0.006686676223762333,
      "epoch": 1.1348724365234375e-05,
      "step": 11900
    },
    {
      "epoch": 1.1348724365234375e-05,
      "step": 11900,
      "training_step_time": 17.071032524108887
    },
    {
      "epoch": 1.1349678039550781e-05,
      "model_forward_time": 5.981418132781982,
      "step": 11901
    },
    {
      "epoch": 1.1349678039550781e-05,
      "step": 11901,
      "training_step_time": 17.472058296203613
    },
    {
      "epoch": 1.1350631713867187e-05,
      "model_forward_time": 5.614553928375244,
      "step": 11902
    },
    {
      "epoch": 1.1350631713867187e-05,
      "step": 11902,
      "training_step_time": 17.124890565872192
    },
    {
      "epoch": 1.1351585388183594e-05,
      "model_forward_time": 5.613691806793213,
      "step": 11903
    },
    {
      "epoch": 1.1351585388183594e-05,
      "step": 11903,
      "training_step_time": 17.100603818893433
    },
    {
      "epoch": 1.13525390625e-05,
      "model_forward_time": 5.61153244972229,
      "step": 11904
    },
    {
      "epoch": 1.13525390625e-05,
      "step": 11904,
      "training_step_time": 17.096999645233154
    },
    {
      "epoch": 1.1353492736816406e-05,
      "model_forward_time": 5.6122143268585205,
      "step": 11905
    },
    {
      "epoch": 1.1353492736816406e-05,
      "step": 11905,
      "training_step_time": 17.10521173477173
    },
    {
      "epoch": 1.1354446411132813e-05,
      "model_forward_time": 5.615300416946411,
      "step": 11906
    },
    {
      "epoch": 1.1354446411132813e-05,
      "step": 11906,
      "training_step_time": 17.105259656906128
    },
    {
      "epoch": 1.1355400085449219e-05,
      "model_forward_time": 5.979454040527344,
      "step": 11907
    },
    {
      "epoch": 1.1355400085449219e-05,
      "step": 11907,
      "training_step_time": 17.446881771087646
    },
    {
      "epoch": 1.1356353759765624e-05,
      "model_forward_time": 5.5866358280181885,
      "step": 11908
    },
    {
      "epoch": 1.1356353759765624e-05,
      "step": 11908,
      "training_step_time": 17.002307415008545
    },
    {
      "epoch": 1.1357307434082032e-05,
      "model_forward_time": 5.576867341995239,
      "step": 11909
    },
    {
      "epoch": 1.1357307434082032e-05,
      "step": 11909,
      "training_step_time": 16.992366075515747
    },
    {
      "epoch": 1.1358261108398438e-05,
      "grad_norm": 0.6421798467636108,
      "learning_rate": 1e-05,
      "loss": 0.0742,
      "step": 11910
    },
    {
      "epoch": 1.1358261108398438e-05,
      "model_forward_time": 5.56130051612854,
      "step": 11910
    },
    {
      "dynamics_loss_avg": 0.06711283661425113,
      "epoch": 1.1358261108398438e-05,
      "step": 11910
    },
    {
      "action_loss_avg": 0.009674824215471745,
      "epoch": 1.1358261108398438e-05,
      "step": 11910
    },
    {
      "epoch": 1.1358261108398438e-05,
      "step": 11910,
      "training_step_time": 16.998048543930054
    },
    {
      "epoch": 1.1359214782714843e-05,
      "model_forward_time": 5.562115669250488,
      "step": 11911
    },
    {
      "epoch": 1.1359214782714843e-05,
      "step": 11911,
      "training_step_time": 16.967711687088013
    },
    {
      "epoch": 1.136016845703125e-05,
      "model_forward_time": 5.92869758605957,
      "step": 11912
    },
    {
      "epoch": 1.136016845703125e-05,
      "step": 11912,
      "training_step_time": 17.34927225112915
    },
    {
      "epoch": 1.1361122131347656e-05,
      "model_forward_time": 5.569979667663574,
      "step": 11913
    },
    {
      "epoch": 1.1361122131347656e-05,
      "step": 11913,
      "training_step_time": 16.997124671936035
    },
    {
      "epoch": 1.1362075805664062e-05,
      "model_forward_time": 5.574281930923462,
      "step": 11914
    },
    {
      "epoch": 1.1362075805664062e-05,
      "step": 11914,
      "training_step_time": 17.020015001296997
    },
    {
      "epoch": 1.1363029479980468e-05,
      "model_forward_time": 5.576144695281982,
      "step": 11915
    },
    {
      "epoch": 1.1363029479980468e-05,
      "step": 11915,
      "training_step_time": 16.98318600654602
    },
    {
      "epoch": 1.1363983154296875e-05,
      "model_forward_time": 5.588441371917725,
      "step": 11916
    },
    {
      "epoch": 1.1363983154296875e-05,
      "step": 11916,
      "training_step_time": 17.07079839706421
    },
    {
      "epoch": 1.1364936828613281e-05,
      "model_forward_time": 5.625112533569336,
      "step": 11917
    },
    {
      "epoch": 1.1364936828613281e-05,
      "step": 11917,
      "training_step_time": 17.13638925552368
    },
    {
      "epoch": 1.1365890502929687e-05,
      "model_forward_time": 5.969689607620239,
      "step": 11918
    },
    {
      "epoch": 1.1365890502929687e-05,
      "step": 11918,
      "training_step_time": 17.46151113510132
    },
    {
      "epoch": 1.1366844177246094e-05,
      "model_forward_time": 5.612607002258301,
      "step": 11919
    },
    {
      "epoch": 1.1366844177246094e-05,
      "step": 11919,
      "training_step_time": 17.106200456619263
    },
    {
      "epoch": 1.13677978515625e-05,
      "grad_norm": 1.558064341545105,
      "learning_rate": 1e-05,
      "loss": 0.0793,
      "step": 11920
    },
    {
      "epoch": 1.13677978515625e-05,
      "model_forward_time": 5.609164714813232,
      "step": 11920
    },
    {
      "dynamics_loss_avg": 0.06628273371607066,
      "epoch": 1.13677978515625e-05,
      "step": 11920
    },
    {
      "action_loss_avg": 0.00887394689489156,
      "epoch": 1.13677978515625e-05,
      "step": 11920
    },
    {
      "epoch": 1.13677978515625e-05,
      "step": 11920,
      "training_step_time": 17.117798805236816
    },
    {
      "epoch": 1.1368751525878906e-05,
      "model_forward_time": 5.618299961090088,
      "step": 11921
    },
    {
      "epoch": 1.1368751525878906e-05,
      "step": 11921,
      "training_step_time": 17.10324192047119
    },
    {
      "epoch": 1.1369705200195313e-05,
      "model_forward_time": 5.61444354057312,
      "step": 11922
    },
    {
      "epoch": 1.1369705200195313e-05,
      "step": 11922,
      "training_step_time": 17.097471475601196
    },
    {
      "epoch": 1.1370658874511719e-05,
      "model_forward_time": 5.618384838104248,
      "step": 11923
    },
    {
      "epoch": 1.1370658874511719e-05,
      "step": 11923,
      "training_step_time": 17.50881004333496
    },
    {
      "epoch": 1.1371612548828125e-05,
      "model_forward_time": 5.61090350151062,
      "step": 11924
    },
    {
      "epoch": 1.1371612548828125e-05,
      "step": 11924,
      "training_step_time": 17.06967782974243
    },
    {
      "epoch": 1.1372566223144532e-05,
      "model_forward_time": 5.5906548500061035,
      "step": 11925
    },
    {
      "epoch": 1.1372566223144532e-05,
      "step": 11925,
      "training_step_time": 17.037477731704712
    },
    {
      "epoch": 1.1373519897460938e-05,
      "model_forward_time": 5.608950614929199,
      "step": 11926
    },
    {
      "epoch": 1.1373519897460938e-05,
      "step": 11926,
      "training_step_time": 17.111246585845947
    },
    {
      "epoch": 1.1374473571777343e-05,
      "model_forward_time": 5.622641086578369,
      "step": 11927
    },
    {
      "epoch": 1.1374473571777343e-05,
      "step": 11927,
      "training_step_time": 17.120089054107666
    },
    {
      "epoch": 1.137542724609375e-05,
      "model_forward_time": 5.601589679718018,
      "step": 11928
    },
    {
      "epoch": 1.137542724609375e-05,
      "step": 11928,
      "training_step_time": 17.0309898853302
    },
    {
      "epoch": 1.1376380920410157e-05,
      "model_forward_time": 5.932860851287842,
      "step": 11929
    },
    {
      "epoch": 1.1376380920410157e-05,
      "step": 11929,
      "training_step_time": 17.36904239654541
    },
    {
      "epoch": 1.1377334594726562e-05,
      "grad_norm": 0.40731772780418396,
      "learning_rate": 1e-05,
      "loss": 0.0534,
      "step": 11930
    },
    {
      "epoch": 1.1377334594726562e-05,
      "model_forward_time": 5.581947088241577,
      "step": 11930
    },
    {
      "dynamics_loss_avg": 0.05098283607512712,
      "epoch": 1.1377334594726562e-05,
      "step": 11930
    },
    {
      "action_loss_avg": 0.005346279963850975,
      "epoch": 1.1377334594726562e-05,
      "step": 11930
    },
    {
      "epoch": 1.1377334594726562e-05,
      "step": 11930,
      "training_step_time": 17.024364233016968
    },
    {
      "epoch": 1.1378288269042968e-05,
      "model_forward_time": 5.599485158920288,
      "step": 11931
    },
    {
      "epoch": 1.1378288269042968e-05,
      "step": 11931,
      "training_step_time": 17.077980756759644
    },
    {
      "epoch": 1.1379241943359375e-05,
      "model_forward_time": 5.60850715637207,
      "step": 11932
    },
    {
      "epoch": 1.1379241943359375e-05,
      "step": 11932,
      "training_step_time": 17.10132384300232
    },
    {
      "epoch": 1.1380195617675781e-05,
      "model_forward_time": 5.612345457077026,
      "step": 11933
    },
    {
      "epoch": 1.1380195617675781e-05,
      "step": 11933,
      "training_step_time": 17.109291791915894
    },
    {
      "epoch": 1.1381149291992187e-05,
      "model_forward_time": 6.0177507400512695,
      "step": 11934
    },
    {
      "epoch": 1.1381149291992187e-05,
      "step": 11934,
      "training_step_time": 17.495800733566284
    },
    {
      "epoch": 1.1382102966308594e-05,
      "model_forward_time": 5.600597858428955,
      "step": 11935
    },
    {
      "epoch": 1.1382102966308594e-05,
      "step": 11935,
      "training_step_time": 17.061553239822388
    },
    {
      "epoch": 1.1383056640625e-05,
      "model_forward_time": 5.593749284744263,
      "step": 11936
    },
    {
      "epoch": 1.1383056640625e-05,
      "step": 11936,
      "training_step_time": 17.034918308258057
    },
    {
      "epoch": 1.1384010314941406e-05,
      "model_forward_time": 5.591703414916992,
      "step": 11937
    },
    {
      "epoch": 1.1384010314941406e-05,
      "step": 11937,
      "training_step_time": 17.046267986297607
    },
    {
      "epoch": 1.1384963989257813e-05,
      "model_forward_time": 5.591466188430786,
      "step": 11938
    },
    {
      "epoch": 1.1384963989257813e-05,
      "step": 11938,
      "training_step_time": 17.064688205718994
    },
    {
      "epoch": 1.1385917663574219e-05,
      "model_forward_time": 5.60223650932312,
      "step": 11939
    },
    {
      "epoch": 1.1385917663574219e-05,
      "step": 11939,
      "training_step_time": 17.09139370918274
    },
    {
      "epoch": 1.1386871337890625e-05,
      "grad_norm": 0.6108039021492004,
      "learning_rate": 1e-05,
      "loss": 0.0498,
      "step": 11940
    },
    {
      "epoch": 1.1386871337890625e-05,
      "model_forward_time": 5.96699070930481,
      "step": 11940
    },
    {
      "dynamics_loss_avg": 0.03973482381552458,
      "epoch": 1.1386871337890625e-05,
      "step": 11940
    },
    {
      "action_loss_avg": 0.004349666682537645,
      "epoch": 1.1386871337890625e-05,
      "step": 11940
    },
    {
      "epoch": 1.1386871337890625e-05,
      "step": 11940,
      "training_step_time": 17.46297860145569
    },
    {
      "epoch": 1.1387825012207032e-05,
      "model_forward_time": 5.614429473876953,
      "step": 11941
    },
    {
      "epoch": 1.1387825012207032e-05,
      "step": 11941,
      "training_step_time": 17.118653297424316
    },
    {
      "epoch": 1.1388778686523438e-05,
      "model_forward_time": 5.6171910762786865,
      "step": 11942
    },
    {
      "epoch": 1.1388778686523438e-05,
      "step": 11942,
      "training_step_time": 17.11277174949646
    },
    {
      "epoch": 1.1389732360839843e-05,
      "model_forward_time": 5.610997438430786,
      "step": 11943
    },
    {
      "epoch": 1.1389732360839843e-05,
      "step": 11943,
      "training_step_time": 17.095526695251465
    },
    {
      "epoch": 1.139068603515625e-05,
      "model_forward_time": 5.608995676040649,
      "step": 11944
    },
    {
      "epoch": 1.139068603515625e-05,
      "step": 11944,
      "training_step_time": 17.097769498825073
    },
    {
      "epoch": 1.1391639709472657e-05,
      "model_forward_time": 6.005781173706055,
      "step": 11945
    },
    {
      "epoch": 1.1391639709472657e-05,
      "step": 11945,
      "training_step_time": 17.50384259223938
    },
    {
      "epoch": 1.1392593383789062e-05,
      "model_forward_time": 5.610092639923096,
      "step": 11946
    },
    {
      "epoch": 1.1392593383789062e-05,
      "step": 11946,
      "training_step_time": 17.090553522109985
    },
    {
      "epoch": 1.1393547058105468e-05,
      "model_forward_time": 5.591443777084351,
      "step": 11947
    },
    {
      "epoch": 1.1393547058105468e-05,
      "step": 11947,
      "training_step_time": 17.01994490623474
    },
    {
      "epoch": 1.1394500732421875e-05,
      "model_forward_time": 5.572556972503662,
      "step": 11948
    },
    {
      "epoch": 1.1394500732421875e-05,
      "step": 11948,
      "training_step_time": 17.009398460388184
    },
    {
      "epoch": 1.1395454406738281e-05,
      "model_forward_time": 5.584493637084961,
      "step": 11949
    },
    {
      "epoch": 1.1395454406738281e-05,
      "step": 11949,
      "training_step_time": 17.00893235206604
    },
    {
      "epoch": 1.1396408081054687e-05,
      "grad_norm": 0.6484856605529785,
      "learning_rate": 1e-05,
      "loss": 0.0646,
      "step": 11950
    },
    {
      "epoch": 1.1396408081054687e-05,
      "model_forward_time": 5.569615364074707,
      "step": 11950
    },
    {
      "dynamics_loss_avg": 0.059074209071695805,
      "epoch": 1.1396408081054687e-05,
      "step": 11950
    },
    {
      "action_loss_avg": 0.008916838164441287,
      "epoch": 1.1396408081054687e-05,
      "step": 11950
    },
    {
      "epoch": 1.1396408081054687e-05,
      "step": 11950,
      "training_step_time": 16.993572235107422
    },
    {
      "epoch": 1.1397361755371094e-05,
      "model_forward_time": 5.93314003944397,
      "step": 11951
    },
    {
      "epoch": 1.1397361755371094e-05,
      "step": 11951,
      "training_step_time": 17.36702799797058
    },
    {
      "epoch": 1.13983154296875e-05,
      "model_forward_time": 5.575128555297852,
      "step": 11952
    },
    {
      "epoch": 1.13983154296875e-05,
      "step": 11952,
      "training_step_time": 17.034096240997314
    },
    {
      "epoch": 1.1399269104003906e-05,
      "model_forward_time": 5.58925199508667,
      "step": 11953
    },
    {
      "epoch": 1.1399269104003906e-05,
      "step": 11953,
      "training_step_time": 17.037160396575928
    },
    {
      "epoch": 1.1400222778320313e-05,
      "model_forward_time": 5.598472356796265,
      "step": 11954
    },
    {
      "epoch": 1.1400222778320313e-05,
      "step": 11954,
      "training_step_time": 17.070924043655396
    },
    {
      "epoch": 1.1401176452636719e-05,
      "model_forward_time": 5.625875234603882,
      "step": 11955
    },
    {
      "epoch": 1.1401176452636719e-05,
      "step": 11955,
      "training_step_time": 17.123406410217285
    },
    {
      "epoch": 1.1402130126953125e-05,
      "model_forward_time": 5.606116533279419,
      "step": 11956
    },
    {
      "epoch": 1.1402130126953125e-05,
      "step": 11956,
      "training_step_time": 17.434091567993164
    },
    {
      "epoch": 1.1403083801269532e-05,
      "model_forward_time": 5.609867334365845,
      "step": 11957
    },
    {
      "epoch": 1.1403083801269532e-05,
      "step": 11957,
      "training_step_time": 17.094077825546265
    },
    {
      "epoch": 1.1404037475585938e-05,
      "model_forward_time": 5.602094888687134,
      "step": 11958
    },
    {
      "epoch": 1.1404037475585938e-05,
      "step": 11958,
      "training_step_time": 17.085430145263672
    },
    {
      "epoch": 1.1404991149902343e-05,
      "model_forward_time": 5.603799104690552,
      "step": 11959
    },
    {
      "epoch": 1.1404991149902343e-05,
      "step": 11959,
      "training_step_time": 17.087294340133667
    },
    {
      "epoch": 1.140594482421875e-05,
      "grad_norm": 0.7731156945228577,
      "learning_rate": 1e-05,
      "loss": 0.0752,
      "step": 11960
    },
    {
      "epoch": 1.140594482421875e-05,
      "model_forward_time": 5.608152866363525,
      "step": 11960
    },
    {
      "dynamics_loss_avg": 0.06993741318583488,
      "epoch": 1.140594482421875e-05,
      "step": 11960
    },
    {
      "action_loss_avg": 0.010151275619864463,
      "epoch": 1.140594482421875e-05,
      "step": 11960
    },
    {
      "epoch": 1.140594482421875e-05,
      "step": 11960,
      "training_step_time": 17.09473443031311
    },
    {
      "epoch": 1.1406898498535157e-05,
      "model_forward_time": 5.612240314483643,
      "step": 11961
    },
    {
      "epoch": 1.1406898498535157e-05,
      "step": 11961,
      "training_step_time": 17.082221508026123
    },
    {
      "epoch": 1.1407852172851562e-05,
      "model_forward_time": 5.9530298709869385,
      "step": 11962
    },
    {
      "epoch": 1.1407852172851562e-05,
      "step": 11962,
      "training_step_time": 17.444315671920776
    },
    {
      "epoch": 1.1408805847167968e-05,
      "model_forward_time": 5.608672142028809,
      "step": 11963
    },
    {
      "epoch": 1.1408805847167968e-05,
      "step": 11963,
      "training_step_time": 17.102355241775513
    },
    {
      "epoch": 1.1409759521484375e-05,
      "model_forward_time": 5.609434366226196,
      "step": 11964
    },
    {
      "epoch": 1.1409759521484375e-05,
      "step": 11964,
      "training_step_time": 17.071805477142334
    },
    {
      "epoch": 1.1410713195800781e-05,
      "model_forward_time": 5.5936667919158936,
      "step": 11965
    },
    {
      "epoch": 1.1410713195800781e-05,
      "step": 11965,
      "training_step_time": 17.010591745376587
    },
    {
      "epoch": 1.1411666870117187e-05,
      "model_forward_time": 5.582744359970093,
      "step": 11966
    },
    {
      "epoch": 1.1411666870117187e-05,
      "step": 11966,
      "training_step_time": 17.00609827041626
    },
    {
      "epoch": 1.1412620544433594e-05,
      "model_forward_time": 5.9442994594573975,
      "step": 11967
    },
    {
      "epoch": 1.1412620544433594e-05,
      "step": 11967,
      "training_step_time": 17.383681297302246
    },
    {
      "epoch": 1.141357421875e-05,
      "model_forward_time": 5.564439535140991,
      "step": 11968
    },
    {
      "epoch": 1.141357421875e-05,
      "step": 11968,
      "training_step_time": 16.975009441375732
    },
    {
      "epoch": 1.1414527893066406e-05,
      "model_forward_time": 5.579198837280273,
      "step": 11969
    },
    {
      "epoch": 1.1414527893066406e-05,
      "step": 11969,
      "training_step_time": 16.994779586791992
    },
    {
      "epoch": 1.1415481567382813e-05,
      "grad_norm": 0.8765515685081482,
      "learning_rate": 1e-05,
      "loss": 0.0564,
      "step": 11970
    },
    {
      "epoch": 1.1415481567382813e-05,
      "model_forward_time": 5.603220701217651,
      "step": 11970
    },
    {
      "dynamics_loss_avg": 0.04547494286671281,
      "epoch": 1.1415481567382813e-05,
      "step": 11970
    },
    {
      "action_loss_avg": 0.00743888805154711,
      "epoch": 1.1415481567382813e-05,
      "step": 11970
    },
    {
      "epoch": 1.1415481567382813e-05,
      "step": 11970,
      "training_step_time": 17.047842741012573
    },
    {
      "epoch": 1.1416435241699219e-05,
      "model_forward_time": 5.5907227993011475,
      "step": 11971
    },
    {
      "epoch": 1.1416435241699219e-05,
      "step": 11971,
      "training_step_time": 17.042155742645264
    },
    {
      "epoch": 1.1417388916015625e-05,
      "model_forward_time": 5.602789402008057,
      "step": 11972
    },
    {
      "epoch": 1.1417388916015625e-05,
      "step": 11972,
      "training_step_time": 17.0717830657959
    },
    {
      "epoch": 1.1418342590332032e-05,
      "model_forward_time": 5.953207731246948,
      "step": 11973
    },
    {
      "epoch": 1.1418342590332032e-05,
      "step": 11973,
      "training_step_time": 17.422154426574707
    },
    {
      "epoch": 1.1419296264648438e-05,
      "model_forward_time": 5.599315643310547,
      "step": 11974
    },
    {
      "epoch": 1.1419296264648438e-05,
      "step": 11974,
      "training_step_time": 17.0271954536438
    },
    {
      "epoch": 1.1420249938964843e-05,
      "model_forward_time": 5.588052272796631,
      "step": 11975
    },
    {
      "epoch": 1.1420249938964843e-05,
      "step": 11975,
      "training_step_time": 17.02503514289856
    },
    {
      "epoch": 1.142120361328125e-05,
      "model_forward_time": 5.591996192932129,
      "step": 11976
    },
    {
      "epoch": 1.142120361328125e-05,
      "step": 11976,
      "training_step_time": 17.035115957260132
    },
    {
      "epoch": 1.1422157287597657e-05,
      "model_forward_time": 5.595602750778198,
      "step": 11977
    },
    {
      "epoch": 1.1422157287597657e-05,
      "step": 11977,
      "training_step_time": 17.057119607925415
    },
    {
      "epoch": 1.1423110961914062e-05,
      "model_forward_time": 5.982205152511597,
      "step": 11978
    },
    {
      "epoch": 1.1423110961914062e-05,
      "step": 11978,
      "training_step_time": 17.416027069091797
    },
    {
      "epoch": 1.1424064636230468e-05,
      "model_forward_time": 5.595477104187012,
      "step": 11979
    },
    {
      "epoch": 1.1424064636230468e-05,
      "step": 11979,
      "training_step_time": 17.037391662597656
    },
    {
      "epoch": 1.1425018310546875e-05,
      "grad_norm": 1.3401668071746826,
      "learning_rate": 1e-05,
      "loss": 0.0871,
      "step": 11980
    },
    {
      "epoch": 1.1425018310546875e-05,
      "model_forward_time": 5.579856634140015,
      "step": 11980
    },
    {
      "dynamics_loss_avg": 0.08316213004291058,
      "epoch": 1.1425018310546875e-05,
      "step": 11980
    },
    {
      "action_loss_avg": 0.014139598119072616,
      "epoch": 1.1425018310546875e-05,
      "step": 11980
    },
    {
      "epoch": 1.1425018310546875e-05,
      "step": 11980,
      "training_step_time": 17.007624626159668
    },
    {
      "epoch": 1.1425971984863281e-05,
      "model_forward_time": 5.590859413146973,
      "step": 11981
    },
    {
      "epoch": 1.1425971984863281e-05,
      "step": 11981,
      "training_step_time": 17.064543962478638
    },
    {
      "epoch": 1.1426925659179687e-05,
      "model_forward_time": 5.616708993911743,
      "step": 11982
    },
    {
      "epoch": 1.1426925659179687e-05,
      "step": 11982,
      "training_step_time": 17.118701934814453
    },
    {
      "epoch": 1.1427879333496094e-05,
      "model_forward_time": 5.619216442108154,
      "step": 11983
    },
    {
      "epoch": 1.1427879333496094e-05,
      "step": 11983,
      "training_step_time": 17.0957350730896
    },
    {
      "epoch": 1.14288330078125e-05,
      "model_forward_time": 5.957777976989746,
      "step": 11984
    },
    {
      "epoch": 1.14288330078125e-05,
      "step": 11984,
      "training_step_time": 17.385612964630127
    },
    {
      "epoch": 1.1429786682128906e-05,
      "model_forward_time": 5.585810422897339,
      "step": 11985
    },
    {
      "epoch": 1.1429786682128906e-05,
      "step": 11985,
      "training_step_time": 17.011857271194458
    },
    {
      "epoch": 1.1430740356445313e-05,
      "model_forward_time": 5.569899082183838,
      "step": 11986
    },
    {
      "epoch": 1.1430740356445313e-05,
      "step": 11986,
      "training_step_time": 17.01624631881714
    },
    {
      "epoch": 1.1431694030761719e-05,
      "model_forward_time": 5.587163686752319,
      "step": 11987
    },
    {
      "epoch": 1.1431694030761719e-05,
      "step": 11987,
      "training_step_time": 17.010481119155884
    },
    {
      "epoch": 1.1432647705078125e-05,
      "model_forward_time": 5.5940868854522705,
      "step": 11988
    },
    {
      "epoch": 1.1432647705078125e-05,
      "step": 11988,
      "training_step_time": 17.035922288894653
    },
    {
      "epoch": 1.1433601379394532e-05,
      "model_forward_time": 5.599667549133301,
      "step": 11989
    },
    {
      "epoch": 1.1433601379394532e-05,
      "step": 11989,
      "training_step_time": 17.438616275787354
    },
    {
      "epoch": 1.1434555053710938e-05,
      "grad_norm": 0.5155997276306152,
      "learning_rate": 1e-05,
      "loss": 0.0659,
      "step": 11990
    },
    {
      "epoch": 1.1434555053710938e-05,
      "model_forward_time": 5.618156671524048,
      "step": 11990
    },
    {
      "dynamics_loss_avg": 0.05134355053305626,
      "epoch": 1.1434555053710938e-05,
      "step": 11990
    },
    {
      "action_loss_avg": 0.008167119743302465,
      "epoch": 1.1434555053710938e-05,
      "step": 11990
    },
    {
      "epoch": 1.1434555053710938e-05,
      "step": 11990,
      "training_step_time": 17.14283776283264
    },
    {
      "epoch": 1.1435508728027343e-05,
      "model_forward_time": 5.641350746154785,
      "step": 11991
    },
    {
      "epoch": 1.1435508728027343e-05,
      "step": 11991,
      "training_step_time": 17.125412940979004
    },
    {
      "epoch": 1.143646240234375e-05,
      "model_forward_time": 5.591904878616333,
      "step": 11992
    },
    {
      "epoch": 1.143646240234375e-05,
      "step": 11992,
      "training_step_time": 17.03457999229431
    },
    {
      "epoch": 1.1437416076660157e-05,
      "model_forward_time": 5.576842784881592,
      "step": 11993
    },
    {
      "epoch": 1.1437416076660157e-05,
      "step": 11993,
      "training_step_time": 17.014020442962646
    },
    {
      "epoch": 1.1438369750976562e-05,
      "model_forward_time": 5.567147254943848,
      "step": 11994
    },
    {
      "epoch": 1.1438369750976562e-05,
      "step": 11994,
      "training_step_time": 16.99075937271118
    },
    {
      "epoch": 1.1439323425292968e-05,
      "model_forward_time": 5.915754079818726,
      "step": 11995
    },
    {
      "epoch": 1.1439323425292968e-05,
      "step": 11995,
      "training_step_time": 17.32496213912964
    },
    {
      "epoch": 1.1440277099609375e-05,
      "model_forward_time": 5.56915807723999,
      "step": 11996
    },
    {
      "epoch": 1.1440277099609375e-05,
      "step": 11996,
      "training_step_time": 16.98743224143982
    },
    {
      "epoch": 1.1441230773925781e-05,
      "model_forward_time": 5.584345579147339,
      "step": 11997
    },
    {
      "epoch": 1.1441230773925781e-05,
      "step": 11997,
      "training_step_time": 17.0180823802948
    },
    {
      "epoch": 1.1442184448242187e-05,
      "model_forward_time": 5.627254486083984,
      "step": 11998
    },
    {
      "epoch": 1.1442184448242187e-05,
      "step": 11998,
      "training_step_time": 17.060593605041504
    },
    {
      "epoch": 1.1443138122558594e-05,
      "model_forward_time": 5.5967419147491455,
      "step": 11999
    },
    {
      "epoch": 1.1443138122558594e-05,
      "step": 11999,
      "training_step_time": 17.034898042678833
    },
    {
      "epoch": 1.1444091796875e-05,
      "grad_norm": 0.9286913275718689,
      "learning_rate": 1e-05,
      "loss": 0.0765,
      "step": 12000
    },
    {
      "epoch": 1.1444091796875e-05,
      "model_forward_time": 6.025777816772461,
      "step": 12000
    },
    {
      "dynamics_loss_avg": 0.06509322002530098,
      "epoch": 1.1444091796875e-05,
      "step": 12000
    },
    {
      "action_loss_avg": 0.009132992778904735,
      "epoch": 1.1444091796875e-05,
      "step": 12000
    },
    {
      "epoch": 1.1444091796875e-05,
      "step": 12000,
      "training_step_time": 17.52627658843994
    },
    {
      "epoch": 1.1445045471191406e-05,
      "model_forward_time": 5.668353796005249,
      "step": 12001
    },
    {
      "epoch": 1.1445045471191406e-05,
      "step": 12001,
      "training_step_time": 17.201419353485107
    },
    {
      "epoch": 1.1445999145507813e-05,
      "model_forward_time": 5.632122755050659,
      "step": 12002
    },
    {
      "epoch": 1.1445999145507813e-05,
      "step": 12002,
      "training_step_time": 17.15213108062744
    },
    {
      "epoch": 1.1446952819824219e-05,
      "model_forward_time": 5.672877311706543,
      "step": 12003
    },
    {
      "epoch": 1.1446952819824219e-05,
      "step": 12003,
      "training_step_time": 17.153068780899048
    },
    {
      "epoch": 1.1447906494140625e-05,
      "model_forward_time": 5.599443435668945,
      "step": 12004
    },
    {
      "epoch": 1.1447906494140625e-05,
      "step": 12004,
      "training_step_time": 17.027769327163696
    },
    {
      "epoch": 1.1448860168457032e-05,
      "model_forward_time": 5.5901079177856445,
      "step": 12005
    },
    {
      "epoch": 1.1448860168457032e-05,
      "step": 12005,
      "training_step_time": 17.030990839004517
    },
    {
      "epoch": 1.1449813842773438e-05,
      "model_forward_time": 5.9841039180755615,
      "step": 12006
    },
    {
      "epoch": 1.1449813842773438e-05,
      "step": 12006,
      "training_step_time": 17.469536781311035
    },
    {
      "epoch": 1.1450767517089843e-05,
      "model_forward_time": 5.6154491901397705,
      "step": 12007
    },
    {
      "epoch": 1.1450767517089843e-05,
      "step": 12007,
      "training_step_time": 17.105756998062134
    },
    {
      "epoch": 1.145172119140625e-05,
      "model_forward_time": 5.600176572799683,
      "step": 12008
    },
    {
      "epoch": 1.145172119140625e-05,
      "step": 12008,
      "training_step_time": 17.024302005767822
    },
    {
      "epoch": 1.1452674865722657e-05,
      "model_forward_time": 5.5838282108306885,
      "step": 12009
    },
    {
      "epoch": 1.1452674865722657e-05,
      "step": 12009,
      "training_step_time": 17.021058320999146
    },
    {
      "epoch": 1.1453628540039062e-05,
      "grad_norm": 0.6311708092689514,
      "learning_rate": 1e-05,
      "loss": 0.0653,
      "step": 12010
    },
    {
      "epoch": 1.1453628540039062e-05,
      "model_forward_time": 5.587390661239624,
      "step": 12010
    },
    {
      "dynamics_loss_avg": 0.05893594603985548,
      "epoch": 1.1453628540039062e-05,
      "step": 12010
    },
    {
      "action_loss_avg": 0.006308184214867651,
      "epoch": 1.1453628540039062e-05,
      "step": 12010
    },
    {
      "epoch": 1.1453628540039062e-05,
      "step": 12010,
      "training_step_time": 17.020731449127197
    },
    {
      "epoch": 1.1454582214355468e-05,
      "model_forward_time": 5.937028646469116,
      "step": 12011
    },
    {
      "epoch": 1.1454582214355468e-05,
      "step": 12011,
      "training_step_time": 17.372975826263428
    },
    {
      "epoch": 1.1455535888671875e-05,
      "model_forward_time": 5.589341402053833,
      "step": 12012
    },
    {
      "epoch": 1.1455535888671875e-05,
      "step": 12012,
      "training_step_time": 17.031298398971558
    },
    {
      "epoch": 1.1456489562988281e-05,
      "model_forward_time": 5.605271339416504,
      "step": 12013
    },
    {
      "epoch": 1.1456489562988281e-05,
      "step": 12013,
      "training_step_time": 17.0587477684021
    },
    {
      "epoch": 1.1457443237304687e-05,
      "model_forward_time": 5.603314161300659,
      "step": 12014
    },
    {
      "epoch": 1.1457443237304687e-05,
      "step": 12014,
      "training_step_time": 17.077313899993896
    },
    {
      "epoch": 1.1458396911621094e-05,
      "model_forward_time": 5.613848924636841,
      "step": 12015
    },
    {
      "epoch": 1.1458396911621094e-05,
      "step": 12015,
      "training_step_time": 17.0753653049469
    },
    {
      "epoch": 1.14593505859375e-05,
      "model_forward_time": 5.595608949661255,
      "step": 12016
    },
    {
      "epoch": 1.14593505859375e-05,
      "step": 12016,
      "training_step_time": 17.406291007995605
    },
    {
      "epoch": 1.1460304260253906e-05,
      "model_forward_time": 5.583731412887573,
      "step": 12017
    },
    {
      "epoch": 1.1460304260253906e-05,
      "step": 12017,
      "training_step_time": 17.032834768295288
    },
    {
      "epoch": 1.1461257934570313e-05,
      "model_forward_time": 5.601047515869141,
      "step": 12018
    },
    {
      "epoch": 1.1461257934570313e-05,
      "step": 12018,
      "training_step_time": 17.075579166412354
    },
    {
      "epoch": 1.1462211608886719e-05,
      "model_forward_time": 5.609230041503906,
      "step": 12019
    },
    {
      "epoch": 1.1462211608886719e-05,
      "step": 12019,
      "training_step_time": 17.107345581054688
    },
    {
      "epoch": 1.1463165283203125e-05,
      "grad_norm": 0.9435408115386963,
      "learning_rate": 1e-05,
      "loss": 0.0806,
      "step": 12020
    },
    {
      "epoch": 1.1463165283203125e-05,
      "model_forward_time": 5.6148176193237305,
      "step": 12020
    },
    {
      "dynamics_loss_avg": 0.06839667018502951,
      "epoch": 1.1463165283203125e-05,
      "step": 12020
    },
    {
      "action_loss_avg": 0.008875198871828615,
      "epoch": 1.1463165283203125e-05,
      "step": 12020
    },
    {
      "epoch": 1.1463165283203125e-05,
      "step": 12020,
      "training_step_time": 17.11654233932495
    },
    {
      "epoch": 1.1464118957519532e-05,
      "model_forward_time": 5.6264448165893555,
      "step": 12021
    },
    {
      "epoch": 1.1464118957519532e-05,
      "step": 12021,
      "training_step_time": 17.100771188735962
    },
    {
      "epoch": 1.1465072631835938e-05,
      "model_forward_time": 5.945473909378052,
      "step": 12022
    },
    {
      "epoch": 1.1465072631835938e-05,
      "step": 12022,
      "training_step_time": 17.41482138633728
    },
    {
      "epoch": 1.1466026306152343e-05,
      "model_forward_time": 5.603957653045654,
      "step": 12023
    },
    {
      "epoch": 1.1466026306152343e-05,
      "step": 12023,
      "training_step_time": 17.092089414596558
    },
    {
      "epoch": 1.146697998046875e-05,
      "model_forward_time": 5.609439849853516,
      "step": 12024
    },
    {
      "epoch": 1.146697998046875e-05,
      "step": 12024,
      "training_step_time": 17.088085889816284
    },
    {
      "epoch": 1.1467933654785157e-05,
      "model_forward_time": 5.615101099014282,
      "step": 12025
    },
    {
      "epoch": 1.1467933654785157e-05,
      "step": 12025,
      "training_step_time": 17.064196348190308
    },
    {
      "epoch": 1.1468887329101562e-05,
      "model_forward_time": 5.593741416931152,
      "step": 12026
    },
    {
      "epoch": 1.1468887329101562e-05,
      "step": 12026,
      "training_step_time": 17.01569104194641
    },
    {
      "epoch": 1.1469841003417968e-05,
      "model_forward_time": 5.966575622558594,
      "step": 12027
    },
    {
      "epoch": 1.1469841003417968e-05,
      "step": 12027,
      "training_step_time": 17.436663389205933
    },
    {
      "epoch": 1.1470794677734375e-05,
      "model_forward_time": 5.600895404815674,
      "step": 12028
    },
    {
      "epoch": 1.1470794677734375e-05,
      "step": 12028,
      "training_step_time": 17.08673906326294
    },
    {
      "epoch": 1.1471748352050781e-05,
      "model_forward_time": 5.611279010772705,
      "step": 12029
    },
    {
      "epoch": 1.1471748352050781e-05,
      "step": 12029,
      "training_step_time": 17.128531455993652
    },
    {
      "epoch": 1.1472702026367187e-05,
      "grad_norm": 0.7539824843406677,
      "learning_rate": 1e-05,
      "loss": 0.0794,
      "step": 12030
    },
    {
      "epoch": 1.1472702026367187e-05,
      "model_forward_time": 5.620775938034058,
      "step": 12030
    },
    {
      "dynamics_loss_avg": 0.06888831220567226,
      "epoch": 1.1472702026367187e-05,
      "step": 12030
    },
    {
      "action_loss_avg": 0.010580979334190489,
      "epoch": 1.1472702026367187e-05,
      "step": 12030
    },
    {
      "epoch": 1.1472702026367187e-05,
      "step": 12030,
      "training_step_time": 17.138859510421753
    },
    {
      "epoch": 1.1473655700683594e-05,
      "model_forward_time": 5.6240174770355225,
      "step": 12031
    },
    {
      "epoch": 1.1473655700683594e-05,
      "step": 12031,
      "training_step_time": 17.12694239616394
    },
    {
      "epoch": 1.1474609375e-05,
      "model_forward_time": 5.616705417633057,
      "step": 12032
    },
    {
      "epoch": 1.1474609375e-05,
      "step": 12032,
      "training_step_time": 17.11890983581543
    },
    {
      "epoch": 1.1475563049316406e-05,
      "model_forward_time": 5.978257417678833,
      "step": 12033
    },
    {
      "epoch": 1.1475563049316406e-05,
      "step": 12033,
      "training_step_time": 17.45401692390442
    },
    {
      "epoch": 1.1476516723632813e-05,
      "model_forward_time": 5.608640670776367,
      "step": 12034
    },
    {
      "epoch": 1.1476516723632813e-05,
      "step": 12034,
      "training_step_time": 17.09398627281189
    },
    {
      "epoch": 1.1477470397949219e-05,
      "model_forward_time": 5.610762596130371,
      "step": 12035
    },
    {
      "epoch": 1.1477470397949219e-05,
      "step": 12035,
      "training_step_time": 17.09180235862732
    },
    {
      "epoch": 1.1478424072265625e-05,
      "model_forward_time": 5.604509592056274,
      "step": 12036
    },
    {
      "epoch": 1.1478424072265625e-05,
      "step": 12036,
      "training_step_time": 17.06857419013977
    },
    {
      "epoch": 1.1479377746582032e-05,
      "model_forward_time": 5.599832534790039,
      "step": 12037
    },
    {
      "epoch": 1.1479377746582032e-05,
      "step": 12037,
      "training_step_time": 17.04626750946045
    },
    {
      "epoch": 1.1480331420898438e-05,
      "model_forward_time": 5.969803810119629,
      "step": 12038
    },
    {
      "epoch": 1.1480331420898438e-05,
      "step": 12038,
      "training_step_time": 17.40457272529602
    },
    {
      "epoch": 1.1481285095214843e-05,
      "model_forward_time": 5.611398696899414,
      "step": 12039
    },
    {
      "epoch": 1.1481285095214843e-05,
      "step": 12039,
      "training_step_time": 17.08618998527527
    },
    {
      "epoch": 1.148223876953125e-05,
      "grad_norm": 0.5406190752983093,
      "learning_rate": 1e-05,
      "loss": 0.0634,
      "step": 12040
    },
    {
      "epoch": 1.148223876953125e-05,
      "model_forward_time": 5.603569507598877,
      "step": 12040
    },
    {
      "dynamics_loss_avg": 0.0607775628566742,
      "epoch": 1.148223876953125e-05,
      "step": 12040
    },
    {
      "action_loss_avg": 0.007950365729629993,
      "epoch": 1.148223876953125e-05,
      "step": 12040
    },
    {
      "epoch": 1.148223876953125e-05,
      "step": 12040,
      "training_step_time": 17.102697610855103
    },
    {
      "epoch": 1.1483192443847657e-05,
      "model_forward_time": 5.614925384521484,
      "step": 12041
    },
    {
      "epoch": 1.1483192443847657e-05,
      "step": 12041,
      "training_step_time": 17.10817289352417
    },
    {
      "epoch": 1.1484146118164062e-05,
      "model_forward_time": 5.615796089172363,
      "step": 12042
    },
    {
      "epoch": 1.1484146118164062e-05,
      "step": 12042,
      "training_step_time": 17.112591981887817
    },
    {
      "epoch": 1.1485099792480468e-05,
      "model_forward_time": 5.6227991580963135,
      "step": 12043
    },
    {
      "epoch": 1.1485099792480468e-05,
      "step": 12043,
      "training_step_time": 17.096898555755615
    },
    {
      "epoch": 1.1486053466796875e-05,
      "model_forward_time": 5.936322450637817,
      "step": 12044
    },
    {
      "epoch": 1.1486053466796875e-05,
      "step": 12044,
      "training_step_time": 17.35814118385315
    },
    {
      "epoch": 1.1487007141113281e-05,
      "model_forward_time": 5.588952302932739,
      "step": 12045
    },
    {
      "epoch": 1.1487007141113281e-05,
      "step": 12045,
      "training_step_time": 17.00531578063965
    },
    {
      "epoch": 1.1487960815429687e-05,
      "model_forward_time": 5.568513870239258,
      "step": 12046
    },
    {
      "epoch": 1.1487960815429687e-05,
      "step": 12046,
      "training_step_time": 16.997420072555542
    },
    {
      "epoch": 1.1488914489746094e-05,
      "model_forward_time": 5.562510251998901,
      "step": 12047
    },
    {
      "epoch": 1.1488914489746094e-05,
      "step": 12047,
      "training_step_time": 16.98340678215027
    },
    {
      "epoch": 1.14898681640625e-05,
      "model_forward_time": 5.5668110847473145,
      "step": 12048
    },
    {
      "epoch": 1.14898681640625e-05,
      "step": 12048,
      "training_step_time": 16.987200498580933
    },
    {
      "epoch": 1.1490821838378906e-05,
      "model_forward_time": 5.951063871383667,
      "step": 12049
    },
    {
      "epoch": 1.1490821838378906e-05,
      "step": 12049,
      "training_step_time": 17.365737915039062
    },
    {
      "epoch": 1.1491775512695313e-05,
      "grad_norm": 0.9681240916252136,
      "learning_rate": 1e-05,
      "loss": 0.0691,
      "step": 12050
    },
    {
      "epoch": 1.1491775512695313e-05,
      "model_forward_time": 5.591870069503784,
      "step": 12050
    },
    {
      "dynamics_loss_avg": 0.056582878530025485,
      "epoch": 1.1491775512695313e-05,
      "step": 12050
    },
    {
      "action_loss_avg": 0.007039028930012137,
      "epoch": 1.1491775512695313e-05,
      "step": 12050
    },
    {
      "epoch": 1.1491775512695313e-05,
      "step": 12050,
      "training_step_time": 17.071660041809082
    },
    {
      "epoch": 1.1492729187011719e-05,
      "model_forward_time": 5.618772029876709,
      "step": 12051
    },
    {
      "epoch": 1.1492729187011719e-05,
      "step": 12051,
      "training_step_time": 17.14216423034668
    },
    {
      "epoch": 1.1493682861328125e-05,
      "model_forward_time": 5.627103567123413,
      "step": 12052
    },
    {
      "epoch": 1.1493682861328125e-05,
      "step": 12052,
      "training_step_time": 17.14153289794922
    },
    {
      "epoch": 1.1494636535644532e-05,
      "model_forward_time": 5.614718437194824,
      "step": 12053
    },
    {
      "epoch": 1.1494636535644532e-05,
      "step": 12053,
      "training_step_time": 17.12556529045105
    },
    {
      "epoch": 1.1495590209960938e-05,
      "model_forward_time": 5.9966278076171875,
      "step": 12054
    },
    {
      "epoch": 1.1495590209960938e-05,
      "step": 12054,
      "training_step_time": 17.46595788002014
    },
    {
      "epoch": 1.1496543884277343e-05,
      "model_forward_time": 5.605583429336548,
      "step": 12055
    },
    {
      "epoch": 1.1496543884277343e-05,
      "step": 12055,
      "training_step_time": 17.078097581863403
    },
    {
      "epoch": 1.149749755859375e-05,
      "model_forward_time": 5.592437982559204,
      "step": 12056
    },
    {
      "epoch": 1.149749755859375e-05,
      "step": 12056,
      "training_step_time": 17.022829294204712
    },
    {
      "epoch": 1.1498451232910157e-05,
      "model_forward_time": 5.612577199935913,
      "step": 12057
    },
    {
      "epoch": 1.1498451232910157e-05,
      "step": 12057,
      "training_step_time": 17.055594205856323
    },
    {
      "epoch": 1.1499404907226562e-05,
      "model_forward_time": 5.603795289993286,
      "step": 12058
    },
    {
      "epoch": 1.1499404907226562e-05,
      "step": 12058,
      "training_step_time": 17.05780577659607
    },
    {
      "epoch": 1.1500358581542968e-05,
      "model_forward_time": 5.610055208206177,
      "step": 12059
    },
    {
      "epoch": 1.1500358581542968e-05,
      "step": 12059,
      "training_step_time": 17.09360980987549
    },
    {
      "epoch": 1.1501312255859375e-05,
      "grad_norm": 1.2019667625427246,
      "learning_rate": 1e-05,
      "loss": 0.0609,
      "step": 12060
    },
    {
      "epoch": 1.1501312255859375e-05,
      "model_forward_time": 5.997543096542358,
      "step": 12060
    },
    {
      "dynamics_loss_avg": 0.05566624496132135,
      "epoch": 1.1501312255859375e-05,
      "step": 12060
    },
    {
      "action_loss_avg": 0.007306655426509679,
      "epoch": 1.1501312255859375e-05,
      "step": 12060
    },
    {
      "epoch": 1.1501312255859375e-05,
      "step": 12060,
      "training_step_time": 17.506314754486084
    },
    {
      "epoch": 1.1502265930175781e-05,
      "model_forward_time": 5.6149256229400635,
      "step": 12061
    },
    {
      "epoch": 1.1502265930175781e-05,
      "step": 12061,
      "training_step_time": 17.101796627044678
    },
    {
      "epoch": 1.1503219604492187e-05,
      "model_forward_time": 5.6229119300842285,
      "step": 12062
    },
    {
      "epoch": 1.1503219604492187e-05,
      "step": 12062,
      "training_step_time": 17.088444471359253
    },
    {
      "epoch": 1.1504173278808594e-05,
      "model_forward_time": 5.596855401992798,
      "step": 12063
    },
    {
      "epoch": 1.1504173278808594e-05,
      "step": 12063,
      "training_step_time": 17.026216745376587
    },
    {
      "epoch": 1.1505126953125e-05,
      "model_forward_time": 5.594196081161499,
      "step": 12064
    },
    {
      "epoch": 1.1505126953125e-05,
      "step": 12064,
      "training_step_time": 16.996917724609375
    },
    {
      "epoch": 1.1506080627441406e-05,
      "model_forward_time": 5.5834949016571045,
      "step": 12065
    },
    {
      "epoch": 1.1506080627441406e-05,
      "step": 12065,
      "training_step_time": 17.388355255126953
    },
    {
      "epoch": 1.1507034301757813e-05,
      "model_forward_time": 5.5726001262664795,
      "step": 12066
    },
    {
      "epoch": 1.1507034301757813e-05,
      "step": 12066,
      "training_step_time": 16.98656988143921
    },
    {
      "epoch": 1.1507987976074219e-05,
      "model_forward_time": 5.593905210494995,
      "step": 12067
    },
    {
      "epoch": 1.1507987976074219e-05,
      "step": 12067,
      "training_step_time": 17.033770084381104
    },
    {
      "epoch": 1.1508941650390625e-05,
      "model_forward_time": 5.594027280807495,
      "step": 12068
    },
    {
      "epoch": 1.1508941650390625e-05,
      "step": 12068,
      "training_step_time": 17.037654399871826
    },
    {
      "epoch": 1.1509895324707032e-05,
      "model_forward_time": 5.600707054138184,
      "step": 12069
    },
    {
      "epoch": 1.1509895324707032e-05,
      "step": 12069,
      "training_step_time": 17.033161401748657
    },
    {
      "epoch": 1.1510848999023438e-05,
      "grad_norm": 0.611539900302887,
      "learning_rate": 1e-05,
      "loss": 0.0824,
      "step": 12070
    },
    {
      "epoch": 1.1510848999023438e-05,
      "model_forward_time": 5.599398851394653,
      "step": 12070
    },
    {
      "dynamics_loss_avg": 0.07276450395584107,
      "epoch": 1.1510848999023438e-05,
      "step": 12070
    },
    {
      "action_loss_avg": 0.009062366560101509,
      "epoch": 1.1510848999023438e-05,
      "step": 12070
    },
    {
      "epoch": 1.1510848999023438e-05,
      "step": 12070,
      "training_step_time": 17.059407234191895
    },
    {
      "epoch": 1.1511802673339843e-05,
      "model_forward_time": 5.966553449630737,
      "step": 12071
    },
    {
      "epoch": 1.1511802673339843e-05,
      "step": 12071,
      "training_step_time": 17.41827964782715
    },
    {
      "epoch": 1.151275634765625e-05,
      "model_forward_time": 5.59987998008728,
      "step": 12072
    },
    {
      "epoch": 1.151275634765625e-05,
      "step": 12072,
      "training_step_time": 17.06398105621338
    },
    {
      "epoch": 1.1513710021972657e-05,
      "model_forward_time": 5.597981929779053,
      "step": 12073
    },
    {
      "epoch": 1.1513710021972657e-05,
      "step": 12073,
      "training_step_time": 17.063075065612793
    },
    {
      "epoch": 1.1514663696289062e-05,
      "model_forward_time": 5.604783773422241,
      "step": 12074
    },
    {
      "epoch": 1.1514663696289062e-05,
      "step": 12074,
      "training_step_time": 17.1047101020813
    },
    {
      "epoch": 1.1515617370605468e-05,
      "model_forward_time": 4.291704177856445,
      "step": 12075
    },
    {
      "epoch": 1.1515617370605468e-05,
      "step": 12075,
      "training_step_time": 12.845651865005493
    },
    {
      "epoch": 1.1516571044921875e-05,
      "model_forward_time": 5.632159471511841,
      "step": 12076
    },
    {
      "epoch": 1.1516571044921875e-05,
      "step": 12076,
      "training_step_time": 17.131541967391968
    },
    {
      "epoch": 1.1517524719238281e-05,
      "model_forward_time": 5.978313207626343,
      "step": 12077
    },
    {
      "epoch": 1.1517524719238281e-05,
      "step": 12077,
      "training_step_time": 17.467448234558105
    },
    {
      "epoch": 1.1518478393554687e-05,
      "model_forward_time": 5.5917088985443115,
      "step": 12078
    },
    {
      "epoch": 1.1518478393554687e-05,
      "step": 12078,
      "training_step_time": 17.04309320449829
    },
    {
      "epoch": 1.1519432067871094e-05,
      "model_forward_time": 5.573940753936768,
      "step": 12079
    },
    {
      "epoch": 1.1519432067871094e-05,
      "step": 12079,
      "training_step_time": 16.99054479598999
    },
    {
      "epoch": 1.15203857421875e-05,
      "grad_norm": 0.5304568409919739,
      "learning_rate": 1e-05,
      "loss": 0.0706,
      "step": 12080
    },
    {
      "epoch": 1.15203857421875e-05,
      "model_forward_time": 5.578171014785767,
      "step": 12080
    },
    {
      "dynamics_loss_avg": 0.06499767303466797,
      "epoch": 1.15203857421875e-05,
      "step": 12080
    },
    {
      "action_loss_avg": 0.008409292041324078,
      "epoch": 1.15203857421875e-05,
      "step": 12080
    },
    {
      "epoch": 1.15203857421875e-05,
      "step": 12080,
      "training_step_time": 16.988360166549683
    },
    {
      "epoch": 1.1521339416503906e-05,
      "model_forward_time": 5.58327841758728,
      "step": 12081
    },
    {
      "epoch": 1.1521339416503906e-05,
      "step": 12081,
      "training_step_time": 17.02609872817993
    },
    {
      "epoch": 1.1522293090820313e-05,
      "model_forward_time": 5.972136735916138,
      "step": 12082
    },
    {
      "epoch": 1.1522293090820313e-05,
      "step": 12082,
      "training_step_time": 17.40803050994873
    },
    {
      "epoch": 1.1523246765136719e-05,
      "model_forward_time": 5.585736274719238,
      "step": 12083
    },
    {
      "epoch": 1.1523246765136719e-05,
      "step": 12083,
      "training_step_time": 17.031269550323486
    },
    {
      "epoch": 1.1524200439453125e-05,
      "model_forward_time": 5.5926947593688965,
      "step": 12084
    },
    {
      "epoch": 1.1524200439453125e-05,
      "step": 12084,
      "training_step_time": 17.061967134475708
    },
    {
      "epoch": 1.1525154113769532e-05,
      "model_forward_time": 5.615940809249878,
      "step": 12085
    },
    {
      "epoch": 1.1525154113769532e-05,
      "step": 12085,
      "training_step_time": 17.107612133026123
    },
    {
      "epoch": 1.1526107788085938e-05,
      "model_forward_time": 5.623582601547241,
      "step": 12086
    },
    {
      "epoch": 1.1526107788085938e-05,
      "step": 12086,
      "training_step_time": 17.13467025756836
    },
    {
      "epoch": 1.1527061462402344e-05,
      "model_forward_time": 5.983687162399292,
      "step": 12087
    },
    {
      "epoch": 1.1527061462402344e-05,
      "step": 12087,
      "training_step_time": 17.484126091003418
    },
    {
      "epoch": 1.152801513671875e-05,
      "model_forward_time": 5.621414661407471,
      "step": 12088
    },
    {
      "epoch": 1.152801513671875e-05,
      "step": 12088,
      "training_step_time": 17.12617516517639
    },
    {
      "epoch": 1.1528968811035157e-05,
      "model_forward_time": 5.6156675815582275,
      "step": 12089
    },
    {
      "epoch": 1.1528968811035157e-05,
      "step": 12089,
      "training_step_time": 17.116191387176514
    },
    {
      "epoch": 1.1529922485351562e-05,
      "grad_norm": 0.6323933005332947,
      "learning_rate": 1e-05,
      "loss": 0.0623,
      "step": 12090
    },
    {
      "epoch": 1.1529922485351562e-05,
      "model_forward_time": 5.605440139770508,
      "step": 12090
    },
    {
      "dynamics_loss_avg": 0.04984549935907125,
      "epoch": 1.1529922485351562e-05,
      "step": 12090
    },
    {
      "action_loss_avg": 0.006366375950165093,
      "epoch": 1.1529922485351562e-05,
      "step": 12090
    },
    {
      "epoch": 1.1529922485351562e-05,
      "step": 12090,
      "training_step_time": 17.103271484375
    },
    {
      "epoch": 1.1530876159667968e-05,
      "model_forward_time": 5.61393666267395,
      "step": 12091
    },
    {
      "epoch": 1.1530876159667968e-05,
      "step": 12091,
      "training_step_time": 17.122250080108643
    },
    {
      "epoch": 1.1531829833984376e-05,
      "model_forward_time": 5.609805583953857,
      "step": 12092
    },
    {
      "epoch": 1.1531829833984376e-05,
      "step": 12092,
      "training_step_time": 17.097894430160522
    },
    {
      "epoch": 1.1532783508300781e-05,
      "model_forward_time": 5.985939264297485,
      "step": 12093
    },
    {
      "epoch": 1.1532783508300781e-05,
      "step": 12093,
      "training_step_time": 17.48266100883484
    },
    {
      "epoch": 1.1533737182617187e-05,
      "model_forward_time": 5.616811037063599,
      "step": 12094
    },
    {
      "epoch": 1.1533737182617187e-05,
      "step": 12094,
      "training_step_time": 17.10556435585022
    },
    {
      "epoch": 1.1534690856933594e-05,
      "model_forward_time": 5.599435329437256,
      "step": 12095
    },
    {
      "epoch": 1.1534690856933594e-05,
      "step": 12095,
      "training_step_time": 17.037238359451294
    },
    {
      "epoch": 1.153564453125e-05,
      "model_forward_time": 5.587778329849243,
      "step": 12096
    },
    {
      "epoch": 1.153564453125e-05,
      "step": 12096,
      "training_step_time": 17.046428680419922
    },
    {
      "epoch": 1.1536598205566406e-05,
      "model_forward_time": 5.593137502670288,
      "step": 12097
    },
    {
      "epoch": 1.1536598205566406e-05,
      "step": 12097,
      "training_step_time": 17.010624885559082
    },
    {
      "epoch": 1.1537551879882813e-05,
      "model_forward_time": 5.593393802642822,
      "step": 12098
    },
    {
      "epoch": 1.1537551879882813e-05,
      "step": 12098,
      "training_step_time": 17.390141248703003
    },
    {
      "epoch": 1.1538505554199219e-05,
      "model_forward_time": 5.632949352264404,
      "step": 12099
    },
    {
      "epoch": 1.1538505554199219e-05,
      "step": 12099,
      "training_step_time": 17.05504560470581
    },
    {
      "epoch": 1.1539459228515625e-05,
      "grad_norm": 0.4325519800186157,
      "learning_rate": 1e-05,
      "loss": 0.0551,
      "step": 12100
    },
    {
      "epoch": 1.1539459228515625e-05,
      "model_forward_time": 5.584040641784668,
      "step": 12100
    },
    {
      "dynamics_loss_avg": 0.05114476904273033,
      "epoch": 1.1539459228515625e-05,
      "step": 12100
    },
    {
      "action_loss_avg": 0.0062297428725287315,
      "epoch": 1.1539459228515625e-05,
      "step": 12100
    },
    {
      "epoch": 1.1539459228515625e-05,
      "step": 12100,
      "training_step_time": 17.035825729370117
    },
    {
      "epoch": 1.1540412902832032e-05,
      "model_forward_time": 5.640391826629639,
      "step": 12101
    },
    {
      "epoch": 1.1540412902832032e-05,
      "step": 12101,
      "training_step_time": 17.14551615715027
    },
    {
      "epoch": 1.1541366577148438e-05,
      "model_forward_time": 5.61447811126709,
      "step": 12102
    },
    {
      "epoch": 1.1541366577148438e-05,
      "step": 12102,
      "training_step_time": 17.12887716293335
    },
    {
      "epoch": 1.1542320251464844e-05,
      "model_forward_time": 5.62490701675415,
      "step": 12103
    },
    {
      "epoch": 1.1542320251464844e-05,
      "step": 12103,
      "training_step_time": 17.127444744110107
    },
    {
      "epoch": 1.154327392578125e-05,
      "model_forward_time": 5.976296663284302,
      "step": 12104
    },
    {
      "epoch": 1.154327392578125e-05,
      "step": 12104,
      "training_step_time": 17.430577754974365
    },
    {
      "epoch": 1.1544227600097657e-05,
      "model_forward_time": 5.583751916885376,
      "step": 12105
    },
    {
      "epoch": 1.1544227600097657e-05,
      "step": 12105,
      "training_step_time": 17.00093960762024
    },
    {
      "epoch": 1.1545181274414062e-05,
      "model_forward_time": 5.594128608703613,
      "step": 12106
    },
    {
      "epoch": 1.1545181274414062e-05,
      "step": 12106,
      "training_step_time": 17.0104718208313
    },
    {
      "epoch": 1.1546134948730468e-05,
      "model_forward_time": 5.589168548583984,
      "step": 12107
    },
    {
      "epoch": 1.1546134948730468e-05,
      "step": 12107,
      "training_step_time": 17.025532007217407
    },
    {
      "epoch": 1.1547088623046876e-05,
      "model_forward_time": 5.594845771789551,
      "step": 12108
    },
    {
      "epoch": 1.1547088623046876e-05,
      "step": 12108,
      "training_step_time": 17.023223161697388
    },
    {
      "epoch": 1.1548042297363281e-05,
      "model_forward_time": 5.584027051925659,
      "step": 12109
    },
    {
      "epoch": 1.1548042297363281e-05,
      "step": 12109,
      "training_step_time": 17.39861249923706
    },
    {
      "epoch": 1.1548995971679687e-05,
      "grad_norm": 0.8958353400230408,
      "learning_rate": 1e-05,
      "loss": 0.0544,
      "step": 12110
    },
    {
      "epoch": 1.1548995971679687e-05,
      "model_forward_time": 5.5982725620269775,
      "step": 12110
    },
    {
      "dynamics_loss_avg": 0.047928054630756375,
      "epoch": 1.1548995971679687e-05,
      "step": 12110
    },
    {
      "action_loss_avg": 0.005423421901650727,
      "epoch": 1.1548995971679687e-05,
      "step": 12110
    },
    {
      "epoch": 1.1548995971679687e-05,
      "step": 12110,
      "training_step_time": 17.07629895210266
    },
    {
      "epoch": 1.1549949645996094e-05,
      "model_forward_time": 5.614461660385132,
      "step": 12111
    },
    {
      "epoch": 1.1549949645996094e-05,
      "step": 12111,
      "training_step_time": 17.12792468070984
    },
    {
      "epoch": 1.15509033203125e-05,
      "model_forward_time": 5.619360685348511,
      "step": 12112
    },
    {
      "epoch": 1.15509033203125e-05,
      "step": 12112,
      "training_step_time": 17.130512475967407
    },
    {
      "epoch": 1.1551856994628906e-05,
      "model_forward_time": 5.626216173171997,
      "step": 12113
    },
    {
      "epoch": 1.1551856994628906e-05,
      "step": 12113,
      "training_step_time": 17.134986639022827
    },
    {
      "epoch": 1.1552810668945313e-05,
      "model_forward_time": 5.616555213928223,
      "step": 12114
    },
    {
      "epoch": 1.1552810668945313e-05,
      "step": 12114,
      "training_step_time": 17.120920419692993
    },
    {
      "epoch": 1.1553764343261719e-05,
      "model_forward_time": 5.999240398406982,
      "step": 12115
    },
    {
      "epoch": 1.1553764343261719e-05,
      "step": 12115,
      "training_step_time": 17.488500118255615
    },
    {
      "epoch": 1.1554718017578125e-05,
      "model_forward_time": 5.606389284133911,
      "step": 12116
    },
    {
      "epoch": 1.1554718017578125e-05,
      "step": 12116,
      "training_step_time": 17.081570386886597
    },
    {
      "epoch": 1.1555671691894532e-05,
      "model_forward_time": 5.603106260299683,
      "step": 12117
    },
    {
      "epoch": 1.1555671691894532e-05,
      "step": 12117,
      "training_step_time": 17.050853729248047
    },
    {
      "epoch": 1.1556625366210938e-05,
      "model_forward_time": 5.6007912158966064,
      "step": 12118
    },
    {
      "epoch": 1.1556625366210938e-05,
      "step": 12118,
      "training_step_time": 17.059735774993896
    },
    {
      "epoch": 1.1557579040527344e-05,
      "model_forward_time": 5.5924928188323975,
      "step": 12119
    },
    {
      "epoch": 1.1557579040527344e-05,
      "step": 12119,
      "training_step_time": 17.079927682876587
    },
    {
      "epoch": 1.155853271484375e-05,
      "grad_norm": 0.6341103315353394,
      "learning_rate": 1e-05,
      "loss": 0.0593,
      "step": 12120
    },
    {
      "epoch": 1.155853271484375e-05,
      "model_forward_time": 5.967491865158081,
      "step": 12120
    },
    {
      "dynamics_loss_avg": 0.05979851838201285,
      "epoch": 1.155853271484375e-05,
      "step": 12120
    },
    {
      "action_loss_avg": 0.008915466303005815,
      "epoch": 1.155853271484375e-05,
      "step": 12120
    },
    {
      "epoch": 1.155853271484375e-05,
      "step": 12120,
      "training_step_time": 17.447311639785767
    },
    {
      "epoch": 1.1559486389160157e-05,
      "model_forward_time": 5.621387004852295,
      "step": 12121
    },
    {
      "epoch": 1.1559486389160157e-05,
      "step": 12121,
      "training_step_time": 17.12458562850952
    },
    {
      "epoch": 1.1560440063476562e-05,
      "model_forward_time": 5.608173847198486,
      "step": 12122
    },
    {
      "epoch": 1.1560440063476562e-05,
      "step": 12122,
      "training_step_time": 17.051351308822632
    },
    {
      "epoch": 1.1561393737792968e-05,
      "model_forward_time": 5.584091663360596,
      "step": 12123
    },
    {
      "epoch": 1.1561393737792968e-05,
      "step": 12123,
      "training_step_time": 17.020830631256104
    },
    {
      "epoch": 1.1562347412109376e-05,
      "model_forward_time": 5.583285808563232,
      "step": 12124
    },
    {
      "epoch": 1.1562347412109376e-05,
      "step": 12124,
      "training_step_time": 17.02330231666565
    },
    {
      "epoch": 1.1563301086425781e-05,
      "model_forward_time": 5.5890092849731445,
      "step": 12125
    },
    {
      "epoch": 1.1563301086425781e-05,
      "step": 12125,
      "training_step_time": 17.024142742156982
    },
    {
      "epoch": 1.1564254760742187e-05,
      "model_forward_time": 5.9471435546875,
      "step": 12126
    },
    {
      "epoch": 1.1564254760742187e-05,
      "step": 12126,
      "training_step_time": 17.41062021255493
    },
    {
      "epoch": 1.1565208435058594e-05,
      "model_forward_time": 5.60992169380188,
      "step": 12127
    },
    {
      "epoch": 1.1565208435058594e-05,
      "step": 12127,
      "training_step_time": 17.100343465805054
    },
    {
      "epoch": 1.1566162109375e-05,
      "model_forward_time": 5.6176111698150635,
      "step": 12128
    },
    {
      "epoch": 1.1566162109375e-05,
      "step": 12128,
      "training_step_time": 17.116394758224487
    },
    {
      "epoch": 1.1567115783691406e-05,
      "model_forward_time": 5.602879047393799,
      "step": 12129
    },
    {
      "epoch": 1.1567115783691406e-05,
      "step": 12129,
      "training_step_time": 17.046558141708374
    },
    {
      "epoch": 1.1568069458007813e-05,
      "grad_norm": 1.5060093402862549,
      "learning_rate": 1e-05,
      "loss": 0.0796,
      "step": 12130
    },
    {
      "epoch": 1.1568069458007813e-05,
      "model_forward_time": 5.5796942710876465,
      "step": 12130
    },
    {
      "dynamics_loss_avg": 0.06386810280382633,
      "epoch": 1.1568069458007813e-05,
      "step": 12130
    },
    {
      "action_loss_avg": 0.00846090572886169,
      "epoch": 1.1568069458007813e-05,
      "step": 12130
    },
    {
      "epoch": 1.1568069458007813e-05,
      "step": 12130,
      "training_step_time": 17.026142358779907
    },
    {
      "epoch": 1.1569023132324219e-05,
      "model_forward_time": 5.587937831878662,
      "step": 12131
    },
    {
      "epoch": 1.1569023132324219e-05,
      "step": 12131,
      "training_step_time": 17.377291202545166
    },
    {
      "epoch": 1.1569976806640625e-05,
      "model_forward_time": 5.591907978057861,
      "step": 12132
    },
    {
      "epoch": 1.1569976806640625e-05,
      "step": 12132,
      "training_step_time": 17.040998458862305
    },
    {
      "epoch": 1.157093048095703e-05,
      "model_forward_time": 5.593021869659424,
      "step": 12133
    },
    {
      "epoch": 1.157093048095703e-05,
      "step": 12133,
      "training_step_time": 17.059951543807983
    },
    {
      "epoch": 1.1571884155273438e-05,
      "model_forward_time": 5.609585285186768,
      "step": 12134
    },
    {
      "epoch": 1.1571884155273438e-05,
      "step": 12134,
      "training_step_time": 17.051790952682495
    },
    {
      "epoch": 1.1572837829589844e-05,
      "model_forward_time": 5.605052709579468,
      "step": 12135
    },
    {
      "epoch": 1.1572837829589844e-05,
      "step": 12135,
      "training_step_time": 17.075135231018066
    },
    {
      "epoch": 1.157379150390625e-05,
      "model_forward_time": 5.613495588302612,
      "step": 12136
    },
    {
      "epoch": 1.157379150390625e-05,
      "step": 12136,
      "training_step_time": 17.11207628250122
    },
    {
      "epoch": 1.1574745178222657e-05,
      "model_forward_time": 6.002565860748291,
      "step": 12137
    },
    {
      "epoch": 1.1574745178222657e-05,
      "step": 12137,
      "training_step_time": 17.46994400024414
    },
    {
      "epoch": 1.1575698852539062e-05,
      "model_forward_time": 5.591534614562988,
      "step": 12138
    },
    {
      "epoch": 1.1575698852539062e-05,
      "step": 12138,
      "training_step_time": 17.020729780197144
    },
    {
      "epoch": 1.1576652526855468e-05,
      "model_forward_time": 5.589877605438232,
      "step": 12139
    },
    {
      "epoch": 1.1576652526855468e-05,
      "step": 12139,
      "training_step_time": 17.005312204360962
    },
    {
      "epoch": 1.1577606201171876e-05,
      "grad_norm": 0.6107447743415833,
      "learning_rate": 1e-05,
      "loss": 0.0673,
      "step": 12140
    },
    {
      "epoch": 1.1577606201171876e-05,
      "model_forward_time": 5.567054748535156,
      "step": 12140
    },
    {
      "dynamics_loss_avg": 0.05605864636600018,
      "epoch": 1.1577606201171876e-05,
      "step": 12140
    },
    {
      "action_loss_avg": 0.009249572502449155,
      "epoch": 1.1577606201171876e-05,
      "step": 12140
    },
    {
      "epoch": 1.1577606201171876e-05,
      "step": 12140,
      "training_step_time": 16.98052144050598
    },
    {
      "epoch": 1.1578559875488281e-05,
      "model_forward_time": 5.57677960395813,
      "step": 12141
    },
    {
      "epoch": 1.1578559875488281e-05,
      "step": 12141,
      "training_step_time": 17.01092505455017
    },
    {
      "epoch": 1.1579513549804687e-05,
      "model_forward_time": 5.572342872619629,
      "step": 12142
    },
    {
      "epoch": 1.1579513549804687e-05,
      "step": 12142,
      "training_step_time": 17.391080617904663
    },
    {
      "epoch": 1.1580467224121094e-05,
      "model_forward_time": 5.583215236663818,
      "step": 12143
    },
    {
      "epoch": 1.1580467224121094e-05,
      "step": 12143,
      "training_step_time": 16.988609790802002
    },
    {
      "epoch": 1.15814208984375e-05,
      "model_forward_time": 5.593870401382446,
      "step": 12144
    },
    {
      "epoch": 1.15814208984375e-05,
      "step": 12144,
      "training_step_time": 17.003278017044067
    },
    {
      "epoch": 1.1582374572753906e-05,
      "model_forward_time": 5.588221549987793,
      "step": 12145
    },
    {
      "epoch": 1.1582374572753906e-05,
      "step": 12145,
      "training_step_time": 17.035330057144165
    },
    {
      "epoch": 1.1583328247070313e-05,
      "model_forward_time": 5.592807769775391,
      "step": 12146
    },
    {
      "epoch": 1.1583328247070313e-05,
      "step": 12146,
      "training_step_time": 17.04966402053833
    },
    {
      "epoch": 1.1584281921386719e-05,
      "model_forward_time": 5.596908330917358,
      "step": 12147
    },
    {
      "epoch": 1.1584281921386719e-05,
      "step": 12147,
      "training_step_time": 17.05574369430542
    },
    {
      "epoch": 1.1585235595703125e-05,
      "model_forward_time": 5.977276086807251,
      "step": 12148
    },
    {
      "epoch": 1.1585235595703125e-05,
      "step": 12148,
      "training_step_time": 17.45226764678955
    },
    {
      "epoch": 1.158618927001953e-05,
      "model_forward_time": 5.595318078994751,
      "step": 12149
    },
    {
      "epoch": 1.158618927001953e-05,
      "step": 12149,
      "training_step_time": 17.030661582946777
    },
    {
      "epoch": 1.1587142944335938e-05,
      "grad_norm": 0.7129268050193787,
      "learning_rate": 1e-05,
      "loss": 0.0643,
      "step": 12150
    },
    {
      "epoch": 1.1587142944335938e-05,
      "model_forward_time": 5.587144136428833,
      "step": 12150
    },
    {
      "dynamics_loss_avg": 0.05818071737885475,
      "epoch": 1.1587142944335938e-05,
      "step": 12150
    },
    {
      "action_loss_avg": 0.006388725573197007,
      "epoch": 1.1587142944335938e-05,
      "step": 12150
    },
    {
      "epoch": 1.1587142944335938e-05,
      "step": 12150,
      "training_step_time": 17.037517786026
    },
    {
      "epoch": 1.1588096618652344e-05,
      "model_forward_time": 5.593862056732178,
      "step": 12151
    },
    {
      "epoch": 1.1588096618652344e-05,
      "step": 12151,
      "training_step_time": 17.049399375915527
    },
    {
      "epoch": 1.158905029296875e-05,
      "model_forward_time": 5.598751783370972,
      "step": 12152
    },
    {
      "epoch": 1.158905029296875e-05,
      "step": 12152,
      "training_step_time": 17.07089328765869
    },
    {
      "epoch": 1.1590003967285157e-05,
      "model_forward_time": 5.9892191886901855,
      "step": 12153
    },
    {
      "epoch": 1.1590003967285157e-05,
      "step": 12153,
      "training_step_time": 17.47387981414795
    },
    {
      "epoch": 1.1590957641601562e-05,
      "model_forward_time": 5.60400652885437,
      "step": 12154
    },
    {
      "epoch": 1.1590957641601562e-05,
      "step": 12154,
      "training_step_time": 17.093259572982788
    },
    {
      "epoch": 1.1591911315917968e-05,
      "model_forward_time": 5.6205971240997314,
      "step": 12155
    },
    {
      "epoch": 1.1591911315917968e-05,
      "step": 12155,
      "training_step_time": 17.11117720603943
    },
    {
      "epoch": 1.1592864990234376e-05,
      "model_forward_time": 5.611192941665649,
      "step": 12156
    },
    {
      "epoch": 1.1592864990234376e-05,
      "step": 12156,
      "training_step_time": 17.11337184906006
    },
    {
      "epoch": 1.1593818664550781e-05,
      "model_forward_time": 5.618115425109863,
      "step": 12157
    },
    {
      "epoch": 1.1593818664550781e-05,
      "step": 12157,
      "training_step_time": 17.111737728118896
    },
    {
      "epoch": 1.1594772338867187e-05,
      "model_forward_time": 5.610889196395874,
      "step": 12158
    },
    {
      "epoch": 1.1594772338867187e-05,
      "step": 12158,
      "training_step_time": 17.108766317367554
    },
    {
      "epoch": 1.1595726013183594e-05,
      "model_forward_time": 5.981036424636841,
      "step": 12159
    },
    {
      "epoch": 1.1595726013183594e-05,
      "step": 12159,
      "training_step_time": 17.468169450759888
    },
    {
      "epoch": 1.15966796875e-05,
      "grad_norm": 0.5144110321998596,
      "learning_rate": 1e-05,
      "loss": 0.0471,
      "step": 12160
    },
    {
      "epoch": 1.15966796875e-05,
      "model_forward_time": 5.609347820281982,
      "step": 12160
    },
    {
      "dynamics_loss_avg": 0.042957826517522335,
      "epoch": 1.15966796875e-05,
      "step": 12160
    },
    {
      "action_loss_avg": 0.004421251197345555,
      "epoch": 1.15966796875e-05,
      "step": 12160
    },
    {
      "epoch": 1.15966796875e-05,
      "step": 12160,
      "training_step_time": 17.07595944404602
    },
    {
      "epoch": 1.1597633361816406e-05,
      "model_forward_time": 5.588467836380005,
      "step": 12161
    },
    {
      "epoch": 1.1597633361816406e-05,
      "step": 12161,
      "training_step_time": 17.010667324066162
    },
    {
      "epoch": 1.1598587036132813e-05,
      "model_forward_time": 5.592777490615845,
      "step": 12162
    },
    {
      "epoch": 1.1598587036132813e-05,
      "step": 12162,
      "training_step_time": 17.050451278686523
    },
    {
      "epoch": 1.1599540710449219e-05,
      "model_forward_time": 5.605879068374634,
      "step": 12163
    },
    {
      "epoch": 1.1599540710449219e-05,
      "step": 12163,
      "training_step_time": 17.11003589630127
    },
    {
      "epoch": 1.1600494384765625e-05,
      "model_forward_time": 5.620624542236328,
      "step": 12164
    },
    {
      "epoch": 1.1600494384765625e-05,
      "step": 12164,
      "training_step_time": 17.487931489944458
    },
    {
      "epoch": 1.160144805908203e-05,
      "model_forward_time": 5.6151440143585205,
      "step": 12165
    },
    {
      "epoch": 1.160144805908203e-05,
      "step": 12165,
      "training_step_time": 17.11707639694214
    },
    {
      "epoch": 1.1602401733398438e-05,
      "model_forward_time": 5.615189790725708,
      "step": 12166
    },
    {
      "epoch": 1.1602401733398438e-05,
      "step": 12166,
      "training_step_time": 17.120501041412354
    },
    {
      "epoch": 1.1603355407714844e-05,
      "model_forward_time": 5.6196043491363525,
      "step": 12167
    },
    {
      "epoch": 1.1603355407714844e-05,
      "step": 12167,
      "training_step_time": 17.12084722518921
    },
    {
      "epoch": 1.160430908203125e-05,
      "model_forward_time": 5.618835210800171,
      "step": 12168
    },
    {
      "epoch": 1.160430908203125e-05,
      "step": 12168,
      "training_step_time": 17.112589120864868
    },
    {
      "epoch": 1.1605262756347657e-05,
      "model_forward_time": 5.618298292160034,
      "step": 12169
    },
    {
      "epoch": 1.1605262756347657e-05,
      "step": 12169,
      "training_step_time": 17.12338399887085
    },
    {
      "epoch": 1.1606216430664062e-05,
      "grad_norm": 0.31271985173225403,
      "learning_rate": 1e-05,
      "loss": 0.0555,
      "step": 12170
    },
    {
      "epoch": 1.1606216430664062e-05,
      "model_forward_time": 5.990283489227295,
      "step": 12170
    },
    {
      "dynamics_loss_avg": 0.04618514478206635,
      "epoch": 1.1606216430664062e-05,
      "step": 12170
    },
    {
      "action_loss_avg": 0.007150427903980017,
      "epoch": 1.1606216430664062e-05,
      "step": 12170
    },
    {
      "epoch": 1.1606216430664062e-05,
      "step": 12170,
      "training_step_time": 17.487507820129395
    },
    {
      "epoch": 1.1607170104980468e-05,
      "model_forward_time": 5.606208086013794,
      "step": 12171
    },
    {
      "epoch": 1.1607170104980468e-05,
      "step": 12171,
      "training_step_time": 17.08055281639099
    },
    {
      "epoch": 1.1608123779296876e-05,
      "model_forward_time": 5.60955548286438,
      "step": 12172
    },
    {
      "epoch": 1.1608123779296876e-05,
      "step": 12172,
      "training_step_time": 17.09236741065979
    },
    {
      "epoch": 1.1609077453613281e-05,
      "model_forward_time": 5.604891777038574,
      "step": 12173
    },
    {
      "epoch": 1.1609077453613281e-05,
      "step": 12173,
      "training_step_time": 17.10849165916443
    },
    {
      "epoch": 1.1610031127929687e-05,
      "model_forward_time": 5.601912498474121,
      "step": 12174
    },
    {
      "epoch": 1.1610031127929687e-05,
      "step": 12174,
      "training_step_time": 17.087212562561035
    },
    {
      "epoch": 1.1610984802246094e-05,
      "model_forward_time": 5.622795343399048,
      "step": 12175
    },
    {
      "epoch": 1.1610984802246094e-05,
      "step": 12175,
      "training_step_time": 17.47152614593506
    },
    {
      "epoch": 1.16119384765625e-05,
      "model_forward_time": 5.608821153640747,
      "step": 12176
    },
    {
      "epoch": 1.16119384765625e-05,
      "step": 12176,
      "training_step_time": 17.07602548599243
    },
    {
      "epoch": 1.1612892150878906e-05,
      "model_forward_time": 5.602308988571167,
      "step": 12177
    },
    {
      "epoch": 1.1612892150878906e-05,
      "step": 12177,
      "training_step_time": 17.06501007080078
    },
    {
      "epoch": 1.1613845825195313e-05,
      "model_forward_time": 5.586655855178833,
      "step": 12178
    },
    {
      "epoch": 1.1613845825195313e-05,
      "step": 12178,
      "training_step_time": 17.03470516204834
    },
    {
      "epoch": 1.1614799499511719e-05,
      "model_forward_time": 5.59527587890625,
      "step": 12179
    },
    {
      "epoch": 1.1614799499511719e-05,
      "step": 12179,
      "training_step_time": 17.020341396331787
    },
    {
      "epoch": 1.1615753173828125e-05,
      "grad_norm": 0.7736672163009644,
      "learning_rate": 1e-05,
      "loss": 0.0693,
      "step": 12180
    },
    {
      "epoch": 1.1615753173828125e-05,
      "model_forward_time": 5.59675931930542,
      "step": 12180
    },
    {
      "dynamics_loss_avg": 0.06750341029837728,
      "epoch": 1.1615753173828125e-05,
      "step": 12180
    },
    {
      "action_loss_avg": 0.007924937328789383,
      "epoch": 1.1615753173828125e-05,
      "step": 12180
    },
    {
      "epoch": 1.1615753173828125e-05,
      "step": 12180,
      "training_step_time": 17.054215908050537
    },
    {
      "epoch": 1.161670684814453e-05,
      "model_forward_time": 5.976529121398926,
      "step": 12181
    },
    {
      "epoch": 1.161670684814453e-05,
      "step": 12181,
      "training_step_time": 17.444856643676758
    },
    {
      "epoch": 1.1617660522460938e-05,
      "model_forward_time": 5.6095733642578125,
      "step": 12182
    },
    {
      "epoch": 1.1617660522460938e-05,
      "step": 12182,
      "training_step_time": 17.065162897109985
    },
    {
      "epoch": 1.1618614196777344e-05,
      "model_forward_time": 5.5929906368255615,
      "step": 12183
    },
    {
      "epoch": 1.1618614196777344e-05,
      "step": 12183,
      "training_step_time": 17.027851581573486
    },
    {
      "epoch": 1.161956787109375e-05,
      "model_forward_time": 5.584082365036011,
      "step": 12184
    },
    {
      "epoch": 1.161956787109375e-05,
      "step": 12184,
      "training_step_time": 17.02515435218811
    },
    {
      "epoch": 1.1620521545410157e-05,
      "model_forward_time": 5.593038082122803,
      "step": 12185
    },
    {
      "epoch": 1.1620521545410157e-05,
      "step": 12185,
      "training_step_time": 17.062885761260986
    },
    {
      "epoch": 1.1621475219726562e-05,
      "model_forward_time": 5.981855630874634,
      "step": 12186
    },
    {
      "epoch": 1.1621475219726562e-05,
      "step": 12186,
      "training_step_time": 17.478252410888672
    },
    {
      "epoch": 1.1622428894042968e-05,
      "model_forward_time": 5.607074737548828,
      "step": 12187
    },
    {
      "epoch": 1.1622428894042968e-05,
      "step": 12187,
      "training_step_time": 17.077627658843994
    },
    {
      "epoch": 1.1623382568359376e-05,
      "model_forward_time": 5.6277618408203125,
      "step": 12188
    },
    {
      "epoch": 1.1623382568359376e-05,
      "step": 12188,
      "training_step_time": 17.12495231628418
    },
    {
      "epoch": 1.1624336242675781e-05,
      "model_forward_time": 5.6070075035095215,
      "step": 12189
    },
    {
      "epoch": 1.1624336242675781e-05,
      "step": 12189,
      "training_step_time": 17.106478691101074
    },
    {
      "epoch": 1.1625289916992187e-05,
      "grad_norm": 0.9762757420539856,
      "learning_rate": 1e-05,
      "loss": 0.0699,
      "step": 12190
    },
    {
      "epoch": 1.1625289916992187e-05,
      "model_forward_time": 5.6142213344573975,
      "step": 12190
    },
    {
      "dynamics_loss_avg": 0.05428523998707533,
      "epoch": 1.1625289916992187e-05,
      "step": 12190
    },
    {
      "action_loss_avg": 0.009350674937013536,
      "epoch": 1.1625289916992187e-05,
      "step": 12190
    },
    {
      "epoch": 1.1625289916992187e-05,
      "step": 12190,
      "training_step_time": 17.114842176437378
    },
    {
      "epoch": 1.1626243591308594e-05,
      "model_forward_time": 5.613706827163696,
      "step": 12191
    },
    {
      "epoch": 1.1626243591308594e-05,
      "step": 12191,
      "training_step_time": 17.12373995780945
    },
    {
      "epoch": 1.1627197265625e-05,
      "model_forward_time": 5.981544733047485,
      "step": 12192
    },
    {
      "epoch": 1.1627197265625e-05,
      "step": 12192,
      "training_step_time": 17.460986137390137
    },
    {
      "epoch": 1.1628150939941406e-05,
      "model_forward_time": 5.595997095108032,
      "step": 12193
    },
    {
      "epoch": 1.1628150939941406e-05,
      "step": 12193,
      "training_step_time": 17.008188247680664
    },
    {
      "epoch": 1.1629104614257813e-05,
      "model_forward_time": 5.578150272369385,
      "step": 12194
    },
    {
      "epoch": 1.1629104614257813e-05,
      "step": 12194,
      "training_step_time": 17.004262924194336
    },
    {
      "epoch": 1.1630058288574219e-05,
      "model_forward_time": 5.571108102798462,
      "step": 12195
    },
    {
      "epoch": 1.1630058288574219e-05,
      "step": 12195,
      "training_step_time": 16.99829888343811
    },
    {
      "epoch": 1.1631011962890625e-05,
      "model_forward_time": 5.570344924926758,
      "step": 12196
    },
    {
      "epoch": 1.1631011962890625e-05,
      "step": 12196,
      "training_step_time": 16.98784589767456
    },
    {
      "epoch": 1.163196563720703e-05,
      "model_forward_time": 5.586358070373535,
      "step": 12197
    },
    {
      "epoch": 1.163196563720703e-05,
      "step": 12197,
      "training_step_time": 17.386796474456787
    },
    {
      "epoch": 1.1632919311523438e-05,
      "model_forward_time": 5.581998586654663,
      "step": 12198
    },
    {
      "epoch": 1.1632919311523438e-05,
      "step": 12198,
      "training_step_time": 17.01909351348877
    },
    {
      "epoch": 1.1633872985839844e-05,
      "model_forward_time": 5.5758209228515625,
      "step": 12199
    },
    {
      "epoch": 1.1633872985839844e-05,
      "step": 12199,
      "training_step_time": 17.00650429725647
    },
    {
      "epoch": 1.163482666015625e-05,
      "grad_norm": 0.5516944527626038,
      "learning_rate": 1e-05,
      "loss": 0.0497,
      "step": 12200
    },
    {
      "epoch": 1.163482666015625e-05,
      "model_forward_time": 5.586897134780884,
      "step": 12200
    },
    {
      "dynamics_loss_avg": 0.05063339788466692,
      "epoch": 1.163482666015625e-05,
      "step": 12200
    },
    {
      "action_loss_avg": 0.005235644429922104,
      "epoch": 1.163482666015625e-05,
      "step": 12200
    },
    {
      "epoch": 1.163482666015625e-05,
      "step": 12200,
      "training_step_time": 16.998112440109253
    },
    {
      "epoch": 1.1635780334472657e-05,
      "model_forward_time": 5.60141921043396,
      "step": 12201
    },
    {
      "epoch": 1.1635780334472657e-05,
      "step": 12201,
      "training_step_time": 17.06314992904663
    },
    {
      "epoch": 1.1636734008789062e-05,
      "model_forward_time": 5.623307228088379,
      "step": 12202
    },
    {
      "epoch": 1.1636734008789062e-05,
      "step": 12202,
      "training_step_time": 17.12216877937317
    },
    {
      "epoch": 1.1637687683105468e-05,
      "model_forward_time": 6.002504348754883,
      "step": 12203
    },
    {
      "epoch": 1.1637687683105468e-05,
      "step": 12203,
      "training_step_time": 17.48955273628235
    },
    {
      "epoch": 1.1638641357421876e-05,
      "model_forward_time": 5.590397596359253,
      "step": 12204
    },
    {
      "epoch": 1.1638641357421876e-05,
      "step": 12204,
      "training_step_time": 17.00692057609558
    },
    {
      "epoch": 1.1639595031738281e-05,
      "model_forward_time": 5.584234952926636,
      "step": 12205
    },
    {
      "epoch": 1.1639595031738281e-05,
      "step": 12205,
      "training_step_time": 17.002342462539673
    },
    {
      "epoch": 1.1640548706054687e-05,
      "model_forward_time": 5.5792975425720215,
      "step": 12206
    },
    {
      "epoch": 1.1640548706054687e-05,
      "step": 12206,
      "training_step_time": 17.021268844604492
    },
    {
      "epoch": 1.1641502380371094e-05,
      "model_forward_time": 5.567315101623535,
      "step": 12207
    },
    {
      "epoch": 1.1641502380371094e-05,
      "step": 12207,
      "training_step_time": 16.973748207092285
    },
    {
      "epoch": 1.16424560546875e-05,
      "model_forward_time": 5.579253435134888,
      "step": 12208
    },
    {
      "epoch": 1.16424560546875e-05,
      "step": 12208,
      "training_step_time": 17.37181830406189
    },
    {
      "epoch": 1.1643409729003906e-05,
      "model_forward_time": 5.577132940292358,
      "step": 12209
    },
    {
      "epoch": 1.1643409729003906e-05,
      "step": 12209,
      "training_step_time": 16.99891710281372
    },
    {
      "epoch": 1.1644363403320313e-05,
      "grad_norm": 0.7908344864845276,
      "learning_rate": 1e-05,
      "loss": 0.059,
      "step": 12210
    },
    {
      "epoch": 1.1644363403320313e-05,
      "model_forward_time": 5.583799362182617,
      "step": 12210
    },
    {
      "dynamics_loss_avg": 0.050241968408226964,
      "epoch": 1.1644363403320313e-05,
      "step": 12210
    },
    {
      "action_loss_avg": 0.006848978326888755,
      "epoch": 1.1644363403320313e-05,
      "step": 12210
    },
    {
      "epoch": 1.1644363403320313e-05,
      "step": 12210,
      "training_step_time": 17.028414964675903
    },
    {
      "epoch": 1.1645317077636719e-05,
      "model_forward_time": 5.59900426864624,
      "step": 12211
    },
    {
      "epoch": 1.1645317077636719e-05,
      "step": 12211,
      "training_step_time": 17.080569982528687
    },
    {
      "epoch": 1.1646270751953125e-05,
      "model_forward_time": 5.604477167129517,
      "step": 12212
    },
    {
      "epoch": 1.1646270751953125e-05,
      "step": 12212,
      "training_step_time": 17.102718353271484
    },
    {
      "epoch": 1.164722442626953e-05,
      "model_forward_time": 5.607390642166138,
      "step": 12213
    },
    {
      "epoch": 1.164722442626953e-05,
      "step": 12213,
      "training_step_time": 17.089827299118042
    },
    {
      "epoch": 1.1648178100585938e-05,
      "model_forward_time": 5.99787974357605,
      "step": 12214
    },
    {
      "epoch": 1.1648178100585938e-05,
      "step": 12214,
      "training_step_time": 17.499932289123535
    },
    {
      "epoch": 1.1649131774902344e-05,
      "model_forward_time": 5.619647026062012,
      "step": 12215
    },
    {
      "epoch": 1.1649131774902344e-05,
      "step": 12215,
      "training_step_time": 17.1075382232666
    },
    {
      "epoch": 1.165008544921875e-05,
      "model_forward_time": 5.61655592918396,
      "step": 12216
    },
    {
      "epoch": 1.165008544921875e-05,
      "step": 12216,
      "training_step_time": 17.103776454925537
    },
    {
      "epoch": 1.1651039123535157e-05,
      "model_forward_time": 5.610347270965576,
      "step": 12217
    },
    {
      "epoch": 1.1651039123535157e-05,
      "step": 12217,
      "training_step_time": 17.094971895217896
    },
    {
      "epoch": 1.1651992797851562e-05,
      "model_forward_time": 5.612019062042236,
      "step": 12218
    },
    {
      "epoch": 1.1651992797851562e-05,
      "step": 12218,
      "training_step_time": 17.08950710296631
    },
    {
      "epoch": 1.1652946472167968e-05,
      "model_forward_time": 5.963012933731079,
      "step": 12219
    },
    {
      "epoch": 1.1652946472167968e-05,
      "step": 12219,
      "training_step_time": 17.43412208557129
    },
    {
      "epoch": 1.1653900146484376e-05,
      "grad_norm": 1.1107255220413208,
      "learning_rate": 1e-05,
      "loss": 0.0852,
      "step": 12220
    },
    {
      "epoch": 1.1653900146484376e-05,
      "model_forward_time": 5.610772371292114,
      "step": 12220
    },
    {
      "dynamics_loss_avg": 0.0741242628544569,
      "epoch": 1.1653900146484376e-05,
      "step": 12220
    },
    {
      "action_loss_avg": 0.007898234692402185,
      "epoch": 1.1653900146484376e-05,
      "step": 12220
    },
    {
      "epoch": 1.1653900146484376e-05,
      "step": 12220,
      "training_step_time": 17.09277105331421
    },
    {
      "epoch": 1.1654853820800781e-05,
      "model_forward_time": 5.6069464683532715,
      "step": 12221
    },
    {
      "epoch": 1.1654853820800781e-05,
      "step": 12221,
      "training_step_time": 17.075584173202515
    },
    {
      "epoch": 1.1655807495117187e-05,
      "model_forward_time": 5.602243661880493,
      "step": 12222
    },
    {
      "epoch": 1.1655807495117187e-05,
      "step": 12222,
      "training_step_time": 17.081329107284546
    },
    {
      "epoch": 1.1656761169433594e-05,
      "model_forward_time": 5.6092848777771,
      "step": 12223
    },
    {
      "epoch": 1.1656761169433594e-05,
      "step": 12223,
      "training_step_time": 17.08522129058838
    },
    {
      "epoch": 1.165771484375e-05,
      "model_forward_time": 5.606686353683472,
      "step": 12224
    },
    {
      "epoch": 1.165771484375e-05,
      "step": 12224,
      "training_step_time": 17.10054636001587
    },
    {
      "epoch": 1.1658668518066406e-05,
      "model_forward_time": 5.987475872039795,
      "step": 12225
    },
    {
      "epoch": 1.1658668518066406e-05,
      "step": 12225,
      "training_step_time": 17.454954147338867
    },
    {
      "epoch": 1.1659622192382813e-05,
      "model_forward_time": 5.599803686141968,
      "step": 12226
    },
    {
      "epoch": 1.1659622192382813e-05,
      "step": 12226,
      "training_step_time": 17.092963933944702
    },
    {
      "epoch": 1.1660575866699219e-05,
      "model_forward_time": 5.5981340408325195,
      "step": 12227
    },
    {
      "epoch": 1.1660575866699219e-05,
      "step": 12227,
      "training_step_time": 17.07516646385193
    },
    {
      "epoch": 1.1661529541015625e-05,
      "model_forward_time": 5.600300073623657,
      "step": 12228
    },
    {
      "epoch": 1.1661529541015625e-05,
      "step": 12228,
      "training_step_time": 17.081955432891846
    },
    {
      "epoch": 1.166248321533203e-05,
      "model_forward_time": 5.631357669830322,
      "step": 12229
    },
    {
      "epoch": 1.166248321533203e-05,
      "step": 12229,
      "training_step_time": 17.06891369819641
    },
    {
      "epoch": 1.1663436889648438e-05,
      "grad_norm": 0.8967233896255493,
      "learning_rate": 1e-05,
      "loss": 0.0682,
      "step": 12230
    },
    {
      "epoch": 1.1663436889648438e-05,
      "model_forward_time": 5.596464157104492,
      "step": 12230
    },
    {
      "dynamics_loss_avg": 0.062265141680836675,
      "epoch": 1.1663436889648438e-05,
      "step": 12230
    },
    {
      "action_loss_avg": 0.009164935001172126,
      "epoch": 1.1663436889648438e-05,
      "step": 12230
    },
    {
      "epoch": 1.1663436889648438e-05,
      "step": 12230,
      "training_step_time": 17.442232847213745
    },
    {
      "epoch": 1.1664390563964844e-05,
      "model_forward_time": 5.598923683166504,
      "step": 12231
    },
    {
      "epoch": 1.1664390563964844e-05,
      "step": 12231,
      "training_step_time": 17.053426265716553
    },
    {
      "epoch": 1.166534423828125e-05,
      "model_forward_time": 5.594361066818237,
      "step": 12232
    },
    {
      "epoch": 1.166534423828125e-05,
      "step": 12232,
      "training_step_time": 17.048360586166382
    },
    {
      "epoch": 1.1666297912597657e-05,
      "model_forward_time": 5.590767860412598,
      "step": 12233
    },
    {
      "epoch": 1.1666297912597657e-05,
      "step": 12233,
      "training_step_time": 17.045183897018433
    },
    {
      "epoch": 1.1667251586914062e-05,
      "model_forward_time": 5.600398540496826,
      "step": 12234
    },
    {
      "epoch": 1.1667251586914062e-05,
      "step": 12234,
      "training_step_time": 17.070789337158203
    },
    {
      "epoch": 1.1668205261230468e-05,
      "model_forward_time": 5.596613883972168,
      "step": 12235
    },
    {
      "epoch": 1.1668205261230468e-05,
      "step": 12235,
      "training_step_time": 17.075993061065674
    },
    {
      "epoch": 1.1669158935546876e-05,
      "model_forward_time": 5.973239183425903,
      "step": 12236
    },
    {
      "epoch": 1.1669158935546876e-05,
      "step": 12236,
      "training_step_time": 17.43266797065735
    },
    {
      "epoch": 1.1670112609863281e-05,
      "model_forward_time": 5.616005182266235,
      "step": 12237
    },
    {
      "epoch": 1.1670112609863281e-05,
      "step": 12237,
      "training_step_time": 17.097150325775146
    },
    {
      "epoch": 1.1671066284179687e-05,
      "model_forward_time": 5.611460208892822,
      "step": 12238
    },
    {
      "epoch": 1.1671066284179687e-05,
      "step": 12238,
      "training_step_time": 17.090839624404907
    },
    {
      "epoch": 1.1672019958496094e-05,
      "model_forward_time": 5.601186752319336,
      "step": 12239
    },
    {
      "epoch": 1.1672019958496094e-05,
      "step": 12239,
      "training_step_time": 17.075101375579834
    },
    {
      "epoch": 1.16729736328125e-05,
      "grad_norm": 0.6847999095916748,
      "learning_rate": 1e-05,
      "loss": 0.0679,
      "step": 12240
    },
    {
      "epoch": 1.16729736328125e-05,
      "model_forward_time": 5.607653856277466,
      "step": 12240
    },
    {
      "dynamics_loss_avg": 0.05839758403599262,
      "epoch": 1.16729736328125e-05,
      "step": 12240
    },
    {
      "action_loss_avg": 0.008051504369359464,
      "epoch": 1.16729736328125e-05,
      "step": 12240
    },
    {
      "epoch": 1.16729736328125e-05,
      "step": 12240,
      "training_step_time": 17.040590524673462
    },
    {
      "epoch": 1.1673927307128906e-05,
      "model_forward_time": 5.587203025817871,
      "step": 12241
    },
    {
      "epoch": 1.1673927307128906e-05,
      "step": 12241,
      "training_step_time": 17.39106321334839
    },
    {
      "epoch": 1.1674880981445313e-05,
      "model_forward_time": 5.5898237228393555,
      "step": 12242
    },
    {
      "epoch": 1.1674880981445313e-05,
      "step": 12242,
      "training_step_time": 17.01790976524353
    },
    {
      "epoch": 1.1675834655761719e-05,
      "model_forward_time": 5.598172903060913,
      "step": 12243
    },
    {
      "epoch": 1.1675834655761719e-05,
      "step": 12243,
      "training_step_time": 17.079725980758667
    },
    {
      "epoch": 1.1676788330078125e-05,
      "model_forward_time": 5.619188070297241,
      "step": 12244
    },
    {
      "epoch": 1.1676788330078125e-05,
      "step": 12244,
      "training_step_time": 17.149436473846436
    },
    {
      "epoch": 1.167774200439453e-05,
      "model_forward_time": 5.624716758728027,
      "step": 12245
    },
    {
      "epoch": 1.167774200439453e-05,
      "step": 12245,
      "training_step_time": 17.122816562652588
    },
    {
      "epoch": 1.1678695678710938e-05,
      "model_forward_time": 5.6160078048706055,
      "step": 12246
    },
    {
      "epoch": 1.1678695678710938e-05,
      "step": 12246,
      "training_step_time": 17.09866762161255
    },
    {
      "epoch": 1.1679649353027344e-05,
      "model_forward_time": 5.980591773986816,
      "step": 12247
    },
    {
      "epoch": 1.1679649353027344e-05,
      "step": 12247,
      "training_step_time": 17.459097623825073
    },
    {
      "epoch": 1.168060302734375e-05,
      "model_forward_time": 5.61417818069458,
      "step": 12248
    },
    {
      "epoch": 1.168060302734375e-05,
      "step": 12248,
      "training_step_time": 17.102173805236816
    },
    {
      "epoch": 1.1681556701660157e-05,
      "model_forward_time": 5.6178505420684814,
      "step": 12249
    },
    {
      "epoch": 1.1681556701660157e-05,
      "step": 12249,
      "training_step_time": 17.110015153884888
    },
    {
      "epoch": 1.1682510375976562e-05,
      "grad_norm": 0.6069909930229187,
      "learning_rate": 1e-05,
      "loss": 0.0675,
      "step": 12250
    },
    {
      "epoch": 1.1682510375976562e-05,
      "model_forward_time": 5.609457015991211,
      "step": 12250
    },
    {
      "dynamics_loss_avg": 0.06205735839903355,
      "epoch": 1.1682510375976562e-05,
      "step": 12250
    },
    {
      "action_loss_avg": 0.007899409101810306,
      "epoch": 1.1682510375976562e-05,
      "step": 12250
    },
    {
      "epoch": 1.1682510375976562e-05,
      "step": 12250,
      "training_step_time": 17.09616494178772
    },
    {
      "epoch": 1.1683464050292968e-05,
      "model_forward_time": 5.6123247146606445,
      "step": 12251
    },
    {
      "epoch": 1.1683464050292968e-05,
      "step": 12251,
      "training_step_time": 17.097649574279785
    },
    {
      "epoch": 1.1684417724609376e-05,
      "model_forward_time": 5.971908807754517,
      "step": 12252
    },
    {
      "epoch": 1.1684417724609376e-05,
      "step": 12252,
      "training_step_time": 17.455883026123047
    },
    {
      "epoch": 1.1685371398925781e-05,
      "model_forward_time": 5.613456964492798,
      "step": 12253
    },
    {
      "epoch": 1.1685371398925781e-05,
      "step": 12253,
      "training_step_time": 17.101056814193726
    },
    {
      "epoch": 1.1686325073242187e-05,
      "model_forward_time": 5.618026256561279,
      "step": 12254
    },
    {
      "epoch": 1.1686325073242187e-05,
      "step": 12254,
      "training_step_time": 17.123270511627197
    },
    {
      "epoch": 1.1687278747558594e-05,
      "model_forward_time": 5.614475250244141,
      "step": 12255
    },
    {
      "epoch": 1.1687278747558594e-05,
      "step": 12255,
      "training_step_time": 17.082756519317627
    },
    {
      "epoch": 1.1688232421875e-05,
      "model_forward_time": 5.592580318450928,
      "step": 12256
    },
    {
      "epoch": 1.1688232421875e-05,
      "step": 12256,
      "training_step_time": 17.022456169128418
    },
    {
      "epoch": 1.1689186096191406e-05,
      "model_forward_time": 5.574324369430542,
      "step": 12257
    },
    {
      "epoch": 1.1689186096191406e-05,
      "step": 12257,
      "training_step_time": 16.97967529296875
    },
    {
      "epoch": 1.1690139770507812e-05,
      "model_forward_time": 5.958743572235107,
      "step": 12258
    },
    {
      "epoch": 1.1690139770507812e-05,
      "step": 12258,
      "training_step_time": 17.36851930618286
    },
    {
      "epoch": 1.1691093444824219e-05,
      "model_forward_time": 5.5780017375946045,
      "step": 12259
    },
    {
      "epoch": 1.1691093444824219e-05,
      "step": 12259,
      "training_step_time": 17.02186679840088
    },
    {
      "epoch": 1.1692047119140625e-05,
      "grad_norm": 0.3323127031326294,
      "learning_rate": 1e-05,
      "loss": 0.0683,
      "step": 12260
    },
    {
      "epoch": 1.1692047119140625e-05,
      "model_forward_time": 5.582614898681641,
      "step": 12260
    },
    {
      "dynamics_loss_avg": 0.06012984961271286,
      "epoch": 1.1692047119140625e-05,
      "step": 12260
    },
    {
      "action_loss_avg": 0.007873285096138716,
      "epoch": 1.1692047119140625e-05,
      "step": 12260
    },
    {
      "epoch": 1.1692047119140625e-05,
      "step": 12260,
      "training_step_time": 17.033343076705933
    },
    {
      "epoch": 1.169300079345703e-05,
      "model_forward_time": 5.591879844665527,
      "step": 12261
    },
    {
      "epoch": 1.169300079345703e-05,
      "step": 12261,
      "training_step_time": 17.087644338607788
    },
    {
      "epoch": 1.1693954467773438e-05,
      "model_forward_time": 5.600390672683716,
      "step": 12262
    },
    {
      "epoch": 1.1693954467773438e-05,
      "step": 12262,
      "training_step_time": 17.071695804595947
    },
    {
      "epoch": 1.1694908142089844e-05,
      "model_forward_time": 5.618812322616577,
      "step": 12263
    },
    {
      "epoch": 1.1694908142089844e-05,
      "step": 12263,
      "training_step_time": 17.45729684829712
    },
    {
      "epoch": 1.169586181640625e-05,
      "model_forward_time": 5.629368782043457,
      "step": 12264
    },
    {
      "epoch": 1.169586181640625e-05,
      "step": 12264,
      "training_step_time": 17.091716289520264
    },
    {
      "epoch": 1.1696815490722657e-05,
      "model_forward_time": 5.593496799468994,
      "step": 12265
    },
    {
      "epoch": 1.1696815490722657e-05,
      "step": 12265,
      "training_step_time": 17.016433715820312
    },
    {
      "epoch": 1.1697769165039063e-05,
      "model_forward_time": 5.5787575244903564,
      "step": 12266
    },
    {
      "epoch": 1.1697769165039063e-05,
      "step": 12266,
      "training_step_time": 17.0140643119812
    },
    {
      "epoch": 1.1698722839355468e-05,
      "model_forward_time": 5.57231330871582,
      "step": 12267
    },
    {
      "epoch": 1.1698722839355468e-05,
      "step": 12267,
      "training_step_time": 17.019670009613037
    },
    {
      "epoch": 1.1699676513671876e-05,
      "model_forward_time": 5.5794806480407715,
      "step": 12268
    },
    {
      "epoch": 1.1699676513671876e-05,
      "step": 12268,
      "training_step_time": 16.99113965034485
    },
    {
      "epoch": 1.1700630187988281e-05,
      "model_forward_time": 5.94427227973938,
      "step": 12269
    },
    {
      "epoch": 1.1700630187988281e-05,
      "step": 12269,
      "training_step_time": 17.352762699127197
    },
    {
      "epoch": 1.1701583862304687e-05,
      "grad_norm": 0.9876415729522705,
      "learning_rate": 1e-05,
      "loss": 0.0574,
      "step": 12270
    },
    {
      "epoch": 1.1701583862304687e-05,
      "model_forward_time": 5.586210012435913,
      "step": 12270
    },
    {
      "dynamics_loss_avg": 0.04913710933178663,
      "epoch": 1.1701583862304687e-05,
      "step": 12270
    },
    {
      "action_loss_avg": 0.006019729748368263,
      "epoch": 1.1701583862304687e-05,
      "step": 12270
    },
    {
      "epoch": 1.1701583862304687e-05,
      "step": 12270,
      "training_step_time": 17.019184112548828
    },
    {
      "epoch": 1.1702537536621095e-05,
      "model_forward_time": 5.601381063461304,
      "step": 12271
    },
    {
      "epoch": 1.1702537536621095e-05,
      "step": 12271,
      "training_step_time": 17.065892219543457
    },
    {
      "epoch": 1.17034912109375e-05,
      "model_forward_time": 5.60212516784668,
      "step": 12272
    },
    {
      "epoch": 1.17034912109375e-05,
      "step": 12272,
      "training_step_time": 17.087751626968384
    },
    {
      "epoch": 1.1704444885253906e-05,
      "model_forward_time": 5.6108808517456055,
      "step": 12273
    },
    {
      "epoch": 1.1704444885253906e-05,
      "step": 12273,
      "training_step_time": 17.085471868515015
    },
    {
      "epoch": 1.1705398559570312e-05,
      "model_forward_time": 5.601473569869995,
      "step": 12274
    },
    {
      "epoch": 1.1705398559570312e-05,
      "step": 12274,
      "training_step_time": 17.42899465560913
    },
    {
      "epoch": 1.1706352233886719e-05,
      "model_forward_time": 5.596280574798584,
      "step": 12275
    },
    {
      "epoch": 1.1706352233886719e-05,
      "step": 12275,
      "training_step_time": 17.059458017349243
    },
    {
      "epoch": 1.1707305908203125e-05,
      "model_forward_time": 5.600239276885986,
      "step": 12276
    },
    {
      "epoch": 1.1707305908203125e-05,
      "step": 12276,
      "training_step_time": 17.089760303497314
    },
    {
      "epoch": 1.170825958251953e-05,
      "model_forward_time": 5.61758828163147,
      "step": 12277
    },
    {
      "epoch": 1.170825958251953e-05,
      "step": 12277,
      "training_step_time": 17.11293649673462
    },
    {
      "epoch": 1.1709213256835938e-05,
      "model_forward_time": 5.609018087387085,
      "step": 12278
    },
    {
      "epoch": 1.1709213256835938e-05,
      "step": 12278,
      "training_step_time": 17.10724449157715
    },
    {
      "epoch": 1.1710166931152344e-05,
      "model_forward_time": 5.6057820320129395,
      "step": 12279
    },
    {
      "epoch": 1.1710166931152344e-05,
      "step": 12279,
      "training_step_time": 17.105956554412842
    },
    {
      "epoch": 1.171112060546875e-05,
      "grad_norm": 0.4899178743362427,
      "learning_rate": 1e-05,
      "loss": 0.062,
      "step": 12280
    },
    {
      "epoch": 1.171112060546875e-05,
      "model_forward_time": 5.98813533782959,
      "step": 12280
    },
    {
      "dynamics_loss_avg": 0.05913163386285305,
      "epoch": 1.171112060546875e-05,
      "step": 12280
    },
    {
      "action_loss_avg": 0.0075986267649568616,
      "epoch": 1.171112060546875e-05,
      "step": 12280
    },
    {
      "epoch": 1.171112060546875e-05,
      "step": 12280,
      "training_step_time": 17.4644775390625
    },
    {
      "epoch": 1.1712074279785157e-05,
      "model_forward_time": 5.59384298324585,
      "step": 12281
    },
    {
      "epoch": 1.1712074279785157e-05,
      "step": 12281,
      "training_step_time": 17.052229404449463
    },
    {
      "epoch": 1.1713027954101563e-05,
      "model_forward_time": 5.589724779129028,
      "step": 12282
    },
    {
      "epoch": 1.1713027954101563e-05,
      "step": 12282,
      "training_step_time": 17.033198356628418
    },
    {
      "epoch": 1.1713981628417968e-05,
      "model_forward_time": 5.606338024139404,
      "step": 12283
    },
    {
      "epoch": 1.1713981628417968e-05,
      "step": 12283,
      "training_step_time": 17.07091188430786
    },
    {
      "epoch": 1.1714935302734376e-05,
      "model_forward_time": 5.60582971572876,
      "step": 12284
    },
    {
      "epoch": 1.1714935302734376e-05,
      "step": 12284,
      "training_step_time": 17.09442663192749
    },
    {
      "epoch": 1.1715888977050781e-05,
      "model_forward_time": 5.9613213539123535,
      "step": 12285
    },
    {
      "epoch": 1.1715888977050781e-05,
      "step": 12285,
      "training_step_time": 17.46377468109131
    },
    {
      "epoch": 1.1716842651367187e-05,
      "model_forward_time": 5.606796979904175,
      "step": 12286
    },
    {
      "epoch": 1.1716842651367187e-05,
      "step": 12286,
      "training_step_time": 17.108213424682617
    },
    {
      "epoch": 1.1717796325683595e-05,
      "model_forward_time": 5.612344026565552,
      "step": 12287
    },
    {
      "epoch": 1.1717796325683595e-05,
      "step": 12287,
      "training_step_time": 17.10580062866211
    },
    {
      "epoch": 1.171875e-05,
      "model_forward_time": 5.608627557754517,
      "step": 12288
    },
    {
      "epoch": 1.171875e-05,
      "step": 12288,
      "training_step_time": 17.104402542114258
    },
    {
      "epoch": 1.1719703674316406e-05,
      "model_forward_time": 5.612311363220215,
      "step": 12289
    },
    {
      "epoch": 1.1719703674316406e-05,
      "step": 12289,
      "training_step_time": 17.087501764297485
    },
    {
      "epoch": 1.1720657348632812e-05,
      "grad_norm": 0.8579356074333191,
      "learning_rate": 1e-05,
      "loss": 0.0722,
      "step": 12290
    },
    {
      "epoch": 1.1720657348632812e-05,
      "model_forward_time": 5.594166040420532,
      "step": 12290
    },
    {
      "dynamics_loss_avg": 0.0639540895819664,
      "epoch": 1.1720657348632812e-05,
      "step": 12290
    },
    {
      "action_loss_avg": 0.007059690554160625,
      "epoch": 1.1720657348632812e-05,
      "step": 12290
    },
    {
      "epoch": 1.1720657348632812e-05,
      "step": 12290,
      "training_step_time": 17.01185154914856
    },
    {
      "epoch": 1.1721611022949219e-05,
      "model_forward_time": 5.944589853286743,
      "step": 12291
    },
    {
      "epoch": 1.1721611022949219e-05,
      "step": 12291,
      "training_step_time": 17.353036165237427
    },
    {
      "epoch": 1.1722564697265625e-05,
      "model_forward_time": 5.5794854164123535,
      "step": 12292
    },
    {
      "epoch": 1.1722564697265625e-05,
      "step": 12292,
      "training_step_time": 16.991931438446045
    },
    {
      "epoch": 1.172351837158203e-05,
      "model_forward_time": 5.575361251831055,
      "step": 12293
    },
    {
      "epoch": 1.172351837158203e-05,
      "step": 12293,
      "training_step_time": 17.014914989471436
    },
    {
      "epoch": 1.1724472045898438e-05,
      "model_forward_time": 5.585543155670166,
      "step": 12294
    },
    {
      "epoch": 1.1724472045898438e-05,
      "step": 12294,
      "training_step_time": 17.039570331573486
    },
    {
      "epoch": 1.1725425720214844e-05,
      "model_forward_time": 5.599599123001099,
      "step": 12295
    },
    {
      "epoch": 1.1725425720214844e-05,
      "step": 12295,
      "training_step_time": 17.084691762924194
    },
    {
      "epoch": 1.172637939453125e-05,
      "model_forward_time": 5.614613771438599,
      "step": 12296
    },
    {
      "epoch": 1.172637939453125e-05,
      "step": 12296,
      "training_step_time": 17.506290435791016
    },
    {
      "epoch": 1.1727333068847657e-05,
      "model_forward_time": 5.615385293960571,
      "step": 12297
    },
    {
      "epoch": 1.1727333068847657e-05,
      "step": 12297,
      "training_step_time": 17.1041202545166
    },
    {
      "epoch": 1.1728286743164063e-05,
      "model_forward_time": 5.618213415145874,
      "step": 12298
    },
    {
      "epoch": 1.1728286743164063e-05,
      "step": 12298,
      "training_step_time": 17.113805770874023
    },
    {
      "epoch": 1.1729240417480468e-05,
      "model_forward_time": 5.624332666397095,
      "step": 12299
    },
    {
      "epoch": 1.1729240417480468e-05,
      "step": 12299,
      "training_step_time": 17.120638132095337
    },
    {
      "epoch": 1.1730194091796876e-05,
      "grad_norm": 0.706771731376648,
      "learning_rate": 1e-05,
      "loss": 0.0523,
      "step": 12300
    },
    {
      "epoch": 1.1730194091796876e-05,
      "model_forward_time": 5.61662220954895,
      "step": 12300
    },
    {
      "dynamics_loss_avg": 0.04332233611494303,
      "epoch": 1.1730194091796876e-05,
      "step": 12300
    },
    {
      "action_loss_avg": 0.0043424917850643395,
      "epoch": 1.1730194091796876e-05,
      "step": 12300
    },
    {
      "epoch": 1.1730194091796876e-05,
      "step": 12300,
      "training_step_time": 17.093966245651245
    },
    {
      "epoch": 1.1731147766113281e-05,
      "model_forward_time": 5.606050968170166,
      "step": 12301
    },
    {
      "epoch": 1.1731147766113281e-05,
      "step": 12301,
      "training_step_time": 17.060950994491577
    },
    {
      "epoch": 1.1732101440429687e-05,
      "model_forward_time": 5.979788303375244,
      "step": 12302
    },
    {
      "epoch": 1.1732101440429687e-05,
      "step": 12302,
      "training_step_time": 17.452306985855103
    },
    {
      "epoch": 1.1733055114746095e-05,
      "model_forward_time": 5.610021114349365,
      "step": 12303
    },
    {
      "epoch": 1.1733055114746095e-05,
      "step": 12303,
      "training_step_time": 17.081326246261597
    },
    {
      "epoch": 1.17340087890625e-05,
      "model_forward_time": 5.598333835601807,
      "step": 12304
    },
    {
      "epoch": 1.17340087890625e-05,
      "step": 12304,
      "training_step_time": 17.051713466644287
    },
    {
      "epoch": 1.1734962463378906e-05,
      "model_forward_time": 5.590129137039185,
      "step": 12305
    },
    {
      "epoch": 1.1734962463378906e-05,
      "step": 12305,
      "training_step_time": 17.054709434509277
    },
    {
      "epoch": 1.1735916137695312e-05,
      "model_forward_time": 5.604534149169922,
      "step": 12306
    },
    {
      "epoch": 1.1735916137695312e-05,
      "step": 12306,
      "training_step_time": 17.082144260406494
    },
    {
      "epoch": 1.1736869812011719e-05,
      "model_forward_time": 5.610921621322632,
      "step": 12307
    },
    {
      "epoch": 1.1736869812011719e-05,
      "step": 12307,
      "training_step_time": 17.465643405914307
    },
    {
      "epoch": 1.1737823486328125e-05,
      "model_forward_time": 5.605283498764038,
      "step": 12308
    },
    {
      "epoch": 1.1737823486328125e-05,
      "step": 12308,
      "training_step_time": 17.088419198989868
    },
    {
      "epoch": 1.173877716064453e-05,
      "model_forward_time": 5.602505445480347,
      "step": 12309
    },
    {
      "epoch": 1.173877716064453e-05,
      "step": 12309,
      "training_step_time": 17.08074688911438
    },
    {
      "epoch": 1.1739730834960938e-05,
      "grad_norm": 0.4144183397293091,
      "learning_rate": 1e-05,
      "loss": 0.0601,
      "step": 12310
    },
    {
      "epoch": 1.1739730834960938e-05,
      "model_forward_time": 5.592494487762451,
      "step": 12310
    },
    {
      "dynamics_loss_avg": 0.052813163865357636,
      "epoch": 1.1739730834960938e-05,
      "step": 12310
    },
    {
      "action_loss_avg": 0.00658828328596428,
      "epoch": 1.1739730834960938e-05,
      "step": 12310
    },
    {
      "epoch": 1.1739730834960938e-05,
      "step": 12310,
      "training_step_time": 17.046000003814697
    },
    {
      "epoch": 1.1740684509277344e-05,
      "model_forward_time": 5.5983664989471436,
      "step": 12311
    },
    {
      "epoch": 1.1740684509277344e-05,
      "step": 12311,
      "training_step_time": 17.05826449394226
    },
    {
      "epoch": 1.174163818359375e-05,
      "model_forward_time": 5.606425523757935,
      "step": 12312
    },
    {
      "epoch": 1.174163818359375e-05,
      "step": 12312,
      "training_step_time": 17.038904666900635
    },
    {
      "epoch": 1.1742591857910157e-05,
      "model_forward_time": 5.987726449966431,
      "step": 12313
    },
    {
      "epoch": 1.1742591857910157e-05,
      "step": 12313,
      "training_step_time": 17.42882537841797
    },
    {
      "epoch": 1.1743545532226563e-05,
      "model_forward_time": 5.59652853012085,
      "step": 12314
    },
    {
      "epoch": 1.1743545532226563e-05,
      "step": 12314,
      "training_step_time": 17.018888235092163
    },
    {
      "epoch": 1.1744499206542968e-05,
      "model_forward_time": 5.582168102264404,
      "step": 12315
    },
    {
      "epoch": 1.1744499206542968e-05,
      "step": 12315,
      "training_step_time": 17.021628618240356
    },
    {
      "epoch": 1.1745452880859376e-05,
      "model_forward_time": 5.5944249629974365,
      "step": 12316
    },
    {
      "epoch": 1.1745452880859376e-05,
      "step": 12316,
      "training_step_time": 17.039977550506592
    },
    {
      "epoch": 1.1746406555175781e-05,
      "model_forward_time": 5.583942174911499,
      "step": 12317
    },
    {
      "epoch": 1.1746406555175781e-05,
      "step": 12317,
      "training_step_time": 17.04478406906128
    },
    {
      "epoch": 1.1747360229492187e-05,
      "model_forward_time": 5.973834037780762,
      "step": 12318
    },
    {
      "epoch": 1.1747360229492187e-05,
      "step": 12318,
      "training_step_time": 17.447028875350952
    },
    {
      "epoch": 1.1748313903808595e-05,
      "model_forward_time": 5.615015029907227,
      "step": 12319
    },
    {
      "epoch": 1.1748313903808595e-05,
      "step": 12319,
      "training_step_time": 17.09634280204773
    },
    {
      "epoch": 1.1749267578125e-05,
      "grad_norm": 1.2037991285324097,
      "learning_rate": 1e-05,
      "loss": 0.0656,
      "step": 12320
    },
    {
      "epoch": 1.1749267578125e-05,
      "model_forward_time": 5.607433080673218,
      "step": 12320
    },
    {
      "dynamics_loss_avg": 0.061387256160378455,
      "epoch": 1.1749267578125e-05,
      "step": 12320
    },
    {
      "action_loss_avg": 0.009199266333598643,
      "epoch": 1.1749267578125e-05,
      "step": 12320
    },
    {
      "epoch": 1.1749267578125e-05,
      "step": 12320,
      "training_step_time": 17.106457233428955
    },
    {
      "epoch": 1.1750221252441406e-05,
      "model_forward_time": 5.628245830535889,
      "step": 12321
    },
    {
      "epoch": 1.1750221252441406e-05,
      "step": 12321,
      "training_step_time": 17.114285707473755
    },
    {
      "epoch": 1.1751174926757812e-05,
      "model_forward_time": 5.5966737270355225,
      "step": 12322
    },
    {
      "epoch": 1.1751174926757812e-05,
      "step": 12322,
      "training_step_time": 17.00570011138916
    },
    {
      "epoch": 1.1752128601074219e-05,
      "model_forward_time": 5.578283071517944,
      "step": 12323
    },
    {
      "epoch": 1.1752128601074219e-05,
      "step": 12323,
      "training_step_time": 17.01077175140381
    },
    {
      "epoch": 1.1753082275390625e-05,
      "model_forward_time": 5.960616588592529,
      "step": 12324
    },
    {
      "epoch": 1.1753082275390625e-05,
      "step": 12324,
      "training_step_time": 17.386102437973022
    },
    {
      "epoch": 1.175403594970703e-05,
      "model_forward_time": 5.588741064071655,
      "step": 12325
    },
    {
      "epoch": 1.175403594970703e-05,
      "step": 12325,
      "training_step_time": 17.048336029052734
    },
    {
      "epoch": 1.1754989624023438e-05,
      "model_forward_time": 5.601993083953857,
      "step": 12326
    },
    {
      "epoch": 1.1754989624023438e-05,
      "step": 12326,
      "training_step_time": 17.098273515701294
    },
    {
      "epoch": 1.1755943298339844e-05,
      "model_forward_time": 5.624582767486572,
      "step": 12327
    },
    {
      "epoch": 1.1755943298339844e-05,
      "step": 12327,
      "training_step_time": 17.119425296783447
    },
    {
      "epoch": 1.175689697265625e-05,
      "model_forward_time": 5.620022535324097,
      "step": 12328
    },
    {
      "epoch": 1.175689697265625e-05,
      "step": 12328,
      "training_step_time": 17.129703760147095
    },
    {
      "epoch": 1.1757850646972657e-05,
      "model_forward_time": 5.620093822479248,
      "step": 12329
    },
    {
      "epoch": 1.1757850646972657e-05,
      "step": 12329,
      "training_step_time": 17.477065086364746
    },
    {
      "epoch": 1.1758804321289063e-05,
      "grad_norm": 1.166096568107605,
      "learning_rate": 1e-05,
      "loss": 0.0818,
      "step": 12330
    },
    {
      "epoch": 1.1758804321289063e-05,
      "model_forward_time": 5.618545293807983,
      "step": 12330
    },
    {
      "dynamics_loss_avg": 0.07087104804813862,
      "epoch": 1.1758804321289063e-05,
      "step": 12330
    },
    {
      "action_loss_avg": 0.008926045894622803,
      "epoch": 1.1758804321289063e-05,
      "step": 12330
    },
    {
      "epoch": 1.1758804321289063e-05,
      "step": 12330,
      "training_step_time": 17.118332862854004
    },
    {
      "epoch": 1.1759757995605468e-05,
      "model_forward_time": 5.614439010620117,
      "step": 12331
    },
    {
      "epoch": 1.1759757995605468e-05,
      "step": 12331,
      "training_step_time": 17.100743055343628
    },
    {
      "epoch": 1.1760711669921876e-05,
      "model_forward_time": 5.60677695274353,
      "step": 12332
    },
    {
      "epoch": 1.1760711669921876e-05,
      "step": 12332,
      "training_step_time": 17.092970848083496
    },
    {
      "epoch": 1.1761665344238281e-05,
      "model_forward_time": 5.597655773162842,
      "step": 12333
    },
    {
      "epoch": 1.1761665344238281e-05,
      "step": 12333,
      "training_step_time": 17.047824382781982
    },
    {
      "epoch": 1.1762619018554687e-05,
      "model_forward_time": 5.595952272415161,
      "step": 12334
    },
    {
      "epoch": 1.1762619018554687e-05,
      "step": 12334,
      "training_step_time": 17.01132893562317
    },
    {
      "epoch": 1.1763572692871095e-05,
      "model_forward_time": 5.977160930633545,
      "step": 12335
    },
    {
      "epoch": 1.1763572692871095e-05,
      "step": 12335,
      "training_step_time": 17.411283254623413
    },
    {
      "epoch": 1.17645263671875e-05,
      "model_forward_time": 5.5986692905426025,
      "step": 12336
    },
    {
      "epoch": 1.17645263671875e-05,
      "step": 12336,
      "training_step_time": 17.07566475868225
    },
    {
      "epoch": 1.1765480041503906e-05,
      "model_forward_time": 5.612547397613525,
      "step": 12337
    },
    {
      "epoch": 1.1765480041503906e-05,
      "step": 12337,
      "training_step_time": 17.106645584106445
    },
    {
      "epoch": 1.1766433715820312e-05,
      "model_forward_time": 5.612972974777222,
      "step": 12338
    },
    {
      "epoch": 1.1766433715820312e-05,
      "step": 12338,
      "training_step_time": 17.09683895111084
    },
    {
      "epoch": 1.1767387390136719e-05,
      "model_forward_time": 5.610468864440918,
      "step": 12339
    },
    {
      "epoch": 1.1767387390136719e-05,
      "step": 12339,
      "training_step_time": 17.08265495300293
    },
    {
      "epoch": 1.1768341064453125e-05,
      "grad_norm": 0.6600992679595947,
      "learning_rate": 1e-05,
      "loss": 0.065,
      "step": 12340
    },
    {
      "epoch": 1.1768341064453125e-05,
      "model_forward_time": 5.597351312637329,
      "step": 12340
    },
    {
      "dynamics_loss_avg": 0.05809012548997998,
      "epoch": 1.1768341064453125e-05,
      "step": 12340
    },
    {
      "action_loss_avg": 0.0078377305297181,
      "epoch": 1.1768341064453125e-05,
      "step": 12340
    },
    {
      "epoch": 1.1768341064453125e-05,
      "step": 12340,
      "training_step_time": 17.432652235031128
    },
    {
      "epoch": 1.176929473876953e-05,
      "model_forward_time": 5.593157052993774,
      "step": 12341
    },
    {
      "epoch": 1.176929473876953e-05,
      "step": 12341,
      "training_step_time": 17.0392849445343
    },
    {
      "epoch": 1.1770248413085938e-05,
      "model_forward_time": 5.591844081878662,
      "step": 12342
    },
    {
      "epoch": 1.1770248413085938e-05,
      "step": 12342,
      "training_step_time": 17.023759841918945
    },
    {
      "epoch": 1.1771202087402344e-05,
      "model_forward_time": 5.620219469070435,
      "step": 12343
    },
    {
      "epoch": 1.1771202087402344e-05,
      "step": 12343,
      "training_step_time": 17.09821915626526
    },
    {
      "epoch": 1.177215576171875e-05,
      "model_forward_time": 5.60842752456665,
      "step": 12344
    },
    {
      "epoch": 1.177215576171875e-05,
      "step": 12344,
      "training_step_time": 17.104647159576416
    },
    {
      "epoch": 1.1773109436035157e-05,
      "model_forward_time": 5.62846565246582,
      "step": 12345
    },
    {
      "epoch": 1.1773109436035157e-05,
      "step": 12345,
      "training_step_time": 17.134693384170532
    },
    {
      "epoch": 1.1774063110351563e-05,
      "model_forward_time": 5.988859176635742,
      "step": 12346
    },
    {
      "epoch": 1.1774063110351563e-05,
      "step": 12346,
      "training_step_time": 17.489109992980957
    },
    {
      "epoch": 1.1775016784667968e-05,
      "model_forward_time": 5.594887018203735,
      "step": 12347
    },
    {
      "epoch": 1.1775016784667968e-05,
      "step": 12347,
      "training_step_time": 17.05734348297119
    },
    {
      "epoch": 1.1775970458984376e-05,
      "model_forward_time": 4.267827033996582,
      "step": 12348
    },
    {
      "epoch": 1.1775970458984376e-05,
      "step": 12348,
      "training_step_time": 12.791042566299438
    },
    {
      "epoch": 1.1776924133300781e-05,
      "model_forward_time": 5.587864398956299,
      "step": 12349
    },
    {
      "epoch": 1.1776924133300781e-05,
      "step": 12349,
      "training_step_time": 17.035709619522095
    },
    {
      "epoch": 1.1777877807617187e-05,
      "grad_norm": 0.364326536655426,
      "learning_rate": 1e-05,
      "loss": 0.065,
      "step": 12350
    },
    {
      "epoch": 1.1777877807617187e-05,
      "model_forward_time": 5.594721078872681,
      "step": 12350
    },
    {
      "dynamics_loss_avg": 0.05418893620371819,
      "epoch": 1.1777877807617187e-05,
      "step": 12350
    },
    {
      "action_loss_avg": 0.008048275113105774,
      "epoch": 1.1777877807617187e-05,
      "step": 12350
    },
    {
      "epoch": 1.1777877807617187e-05,
      "step": 12350,
      "training_step_time": 17.023178815841675
    },
    {
      "epoch": 1.1778831481933595e-05,
      "model_forward_time": 5.606662273406982,
      "step": 12351
    },
    {
      "epoch": 1.1778831481933595e-05,
      "step": 12351,
      "training_step_time": 17.448449850082397
    },
    {
      "epoch": 1.177978515625e-05,
      "model_forward_time": 5.60680079460144,
      "step": 12352
    },
    {
      "epoch": 1.177978515625e-05,
      "step": 12352,
      "training_step_time": 17.101409196853638
    },
    {
      "epoch": 1.1780738830566406e-05,
      "model_forward_time": 5.608355522155762,
      "step": 12353
    },
    {
      "epoch": 1.1780738830566406e-05,
      "step": 12353,
      "training_step_time": 17.109864711761475
    },
    {
      "epoch": 1.1781692504882812e-05,
      "model_forward_time": 5.617278099060059,
      "step": 12354
    },
    {
      "epoch": 1.1781692504882812e-05,
      "step": 12354,
      "training_step_time": 17.10728621482849
    },
    {
      "epoch": 1.178264617919922e-05,
      "model_forward_time": 5.606152296066284,
      "step": 12355
    },
    {
      "epoch": 1.178264617919922e-05,
      "step": 12355,
      "training_step_time": 17.091308116912842
    },
    {
      "epoch": 1.1783599853515625e-05,
      "model_forward_time": 5.618926525115967,
      "step": 12356
    },
    {
      "epoch": 1.1783599853515625e-05,
      "step": 12356,
      "training_step_time": 17.10560655593872
    },
    {
      "epoch": 1.178455352783203e-05,
      "model_forward_time": 5.992457389831543,
      "step": 12357
    },
    {
      "epoch": 1.178455352783203e-05,
      "step": 12357,
      "training_step_time": 17.470521450042725
    },
    {
      "epoch": 1.1785507202148438e-05,
      "model_forward_time": 5.605747938156128,
      "step": 12358
    },
    {
      "epoch": 1.1785507202148438e-05,
      "step": 12358,
      "training_step_time": 17.076313495635986
    },
    {
      "epoch": 1.1786460876464844e-05,
      "model_forward_time": 5.611880540847778,
      "step": 12359
    },
    {
      "epoch": 1.1786460876464844e-05,
      "step": 12359,
      "training_step_time": 17.098910570144653
    },
    {
      "epoch": 1.178741455078125e-05,
      "grad_norm": 0.6072645783424377,
      "learning_rate": 1e-05,
      "loss": 0.082,
      "step": 12360
    },
    {
      "epoch": 1.178741455078125e-05,
      "model_forward_time": 5.601634979248047,
      "step": 12360
    },
    {
      "dynamics_loss_avg": 0.07439720891416073,
      "epoch": 1.178741455078125e-05,
      "step": 12360
    },
    {
      "action_loss_avg": 0.009836945100687444,
      "epoch": 1.178741455078125e-05,
      "step": 12360
    },
    {
      "epoch": 1.178741455078125e-05,
      "step": 12360,
      "training_step_time": 17.081242561340332
    },
    {
      "epoch": 1.1788368225097657e-05,
      "model_forward_time": 5.602333307266235,
      "step": 12361
    },
    {
      "epoch": 1.1788368225097657e-05,
      "step": 12361,
      "training_step_time": 17.088462352752686
    },
    {
      "epoch": 1.1789321899414063e-05,
      "model_forward_time": 5.611710071563721,
      "step": 12362
    },
    {
      "epoch": 1.1789321899414063e-05,
      "step": 12362,
      "training_step_time": 17.425823211669922
    },
    {
      "epoch": 1.1790275573730468e-05,
      "model_forward_time": 5.597986698150635,
      "step": 12363
    },
    {
      "epoch": 1.1790275573730468e-05,
      "step": 12363,
      "training_step_time": 17.05699110031128
    },
    {
      "epoch": 1.1791229248046876e-05,
      "model_forward_time": 5.598184585571289,
      "step": 12364
    },
    {
      "epoch": 1.1791229248046876e-05,
      "step": 12364,
      "training_step_time": 17.051273584365845
    },
    {
      "epoch": 1.1792182922363281e-05,
      "model_forward_time": 5.580684185028076,
      "step": 12365
    },
    {
      "epoch": 1.1792182922363281e-05,
      "step": 12365,
      "training_step_time": 17.006956577301025
    },
    {
      "epoch": 1.1793136596679687e-05,
      "model_forward_time": 5.581632852554321,
      "step": 12366
    },
    {
      "epoch": 1.1793136596679687e-05,
      "step": 12366,
      "training_step_time": 17.024591207504272
    },
    {
      "epoch": 1.1794090270996095e-05,
      "model_forward_time": 5.581592321395874,
      "step": 12367
    },
    {
      "epoch": 1.1794090270996095e-05,
      "step": 12367,
      "training_step_time": 17.02306842803955
    },
    {
      "epoch": 1.17950439453125e-05,
      "model_forward_time": 5.972837448120117,
      "step": 12368
    },
    {
      "epoch": 1.17950439453125e-05,
      "step": 12368,
      "training_step_time": 17.429749727249146
    },
    {
      "epoch": 1.1795997619628906e-05,
      "model_forward_time": 5.613868474960327,
      "step": 12369
    },
    {
      "epoch": 1.1795997619628906e-05,
      "step": 12369,
      "training_step_time": 17.109392881393433
    },
    {
      "epoch": 1.1796951293945312e-05,
      "grad_norm": 0.5424386858940125,
      "learning_rate": 1e-05,
      "loss": 0.0627,
      "step": 12370
    },
    {
      "epoch": 1.1796951293945312e-05,
      "model_forward_time": 5.635249853134155,
      "step": 12370
    },
    {
      "dynamics_loss_avg": 0.05456973426043987,
      "epoch": 1.1796951293945312e-05,
      "step": 12370
    },
    {
      "action_loss_avg": 0.006500245095230639,
      "epoch": 1.1796951293945312e-05,
      "step": 12370
    },
    {
      "epoch": 1.1796951293945312e-05,
      "step": 12370,
      "training_step_time": 17.162571907043457
    },
    {
      "epoch": 1.179790496826172e-05,
      "model_forward_time": 5.624892711639404,
      "step": 12371
    },
    {
      "epoch": 1.179790496826172e-05,
      "step": 12371,
      "training_step_time": 17.128432035446167
    },
    {
      "epoch": 1.1798858642578125e-05,
      "model_forward_time": 5.613450765609741,
      "step": 12372
    },
    {
      "epoch": 1.1798858642578125e-05,
      "step": 12372,
      "training_step_time": 17.0910005569458
    },
    {
      "epoch": 1.179981231689453e-05,
      "model_forward_time": 5.605316638946533,
      "step": 12373
    },
    {
      "epoch": 1.179981231689453e-05,
      "step": 12373,
      "training_step_time": 17.434697151184082
    },
    {
      "epoch": 1.1800765991210938e-05,
      "model_forward_time": 5.60595703125,
      "step": 12374
    },
    {
      "epoch": 1.1800765991210938e-05,
      "step": 12374,
      "training_step_time": 17.056668281555176
    },
    {
      "epoch": 1.1801719665527344e-05,
      "model_forward_time": 5.606445074081421,
      "step": 12375
    },
    {
      "epoch": 1.1801719665527344e-05,
      "step": 12375,
      "training_step_time": 17.056711196899414
    },
    {
      "epoch": 1.180267333984375e-05,
      "model_forward_time": 5.598299026489258,
      "step": 12376
    },
    {
      "epoch": 1.180267333984375e-05,
      "step": 12376,
      "training_step_time": 17.085965394973755
    },
    {
      "epoch": 1.1803627014160157e-05,
      "model_forward_time": 5.608000993728638,
      "step": 12377
    },
    {
      "epoch": 1.1803627014160157e-05,
      "step": 12377,
      "training_step_time": 17.080101490020752
    },
    {
      "epoch": 1.1804580688476563e-05,
      "model_forward_time": 5.613449811935425,
      "step": 12378
    },
    {
      "epoch": 1.1804580688476563e-05,
      "step": 12378,
      "training_step_time": 17.118496894836426
    },
    {
      "epoch": 1.1805534362792968e-05,
      "model_forward_time": 5.982754468917847,
      "step": 12379
    },
    {
      "epoch": 1.1805534362792968e-05,
      "step": 12379,
      "training_step_time": 17.457315683364868
    },
    {
      "epoch": 1.1806488037109376e-05,
      "grad_norm": 0.5473757386207581,
      "learning_rate": 1e-05,
      "loss": 0.0612,
      "step": 12380
    },
    {
      "epoch": 1.1806488037109376e-05,
      "model_forward_time": 5.601005792617798,
      "step": 12380
    },
    {
      "dynamics_loss_avg": 0.061646273173391816,
      "epoch": 1.1806488037109376e-05,
      "step": 12380
    },
    {
      "action_loss_avg": 0.008483395865187048,
      "epoch": 1.1806488037109376e-05,
      "step": 12380
    },
    {
      "epoch": 1.1806488037109376e-05,
      "step": 12380,
      "training_step_time": 17.08346390724182
    },
    {
      "epoch": 1.1807441711425781e-05,
      "model_forward_time": 5.6075050830841064,
      "step": 12381
    },
    {
      "epoch": 1.1807441711425781e-05,
      "step": 12381,
      "training_step_time": 17.07502317428589
    },
    {
      "epoch": 1.1808395385742187e-05,
      "model_forward_time": 5.607388257980347,
      "step": 12382
    },
    {
      "epoch": 1.1808395385742187e-05,
      "step": 12382,
      "training_step_time": 17.08351993560791
    },
    {
      "epoch": 1.1809349060058593e-05,
      "model_forward_time": 5.612882614135742,
      "step": 12383
    },
    {
      "epoch": 1.1809349060058593e-05,
      "step": 12383,
      "training_step_time": 17.09253716468811
    },
    {
      "epoch": 1.1810302734375e-05,
      "model_forward_time": 5.953835725784302,
      "step": 12384
    },
    {
      "epoch": 1.1810302734375e-05,
      "step": 12384,
      "training_step_time": 17.399364233016968
    },
    {
      "epoch": 1.1811256408691406e-05,
      "model_forward_time": 5.581304311752319,
      "step": 12385
    },
    {
      "epoch": 1.1811256408691406e-05,
      "step": 12385,
      "training_step_time": 17.02185082435608
    },
    {
      "epoch": 1.1812210083007812e-05,
      "model_forward_time": 5.575419187545776,
      "step": 12386
    },
    {
      "epoch": 1.1812210083007812e-05,
      "step": 12386,
      "training_step_time": 16.995712757110596
    },
    {
      "epoch": 1.181316375732422e-05,
      "model_forward_time": 5.5826733112335205,
      "step": 12387
    },
    {
      "epoch": 1.181316375732422e-05,
      "step": 12387,
      "training_step_time": 16.995434045791626
    },
    {
      "epoch": 1.1814117431640625e-05,
      "model_forward_time": 5.582752227783203,
      "step": 12388
    },
    {
      "epoch": 1.1814117431640625e-05,
      "step": 12388,
      "training_step_time": 17.02021074295044
    },
    {
      "epoch": 1.181507110595703e-05,
      "model_forward_time": 5.591586351394653,
      "step": 12389
    },
    {
      "epoch": 1.181507110595703e-05,
      "step": 12389,
      "training_step_time": 17.022709131240845
    },
    {
      "epoch": 1.1816024780273438e-05,
      "grad_norm": 0.6180837154388428,
      "learning_rate": 1e-05,
      "loss": 0.0776,
      "step": 12390
    },
    {
      "epoch": 1.1816024780273438e-05,
      "model_forward_time": 5.99210262298584,
      "step": 12390
    },
    {
      "dynamics_loss_avg": 0.060275617986917496,
      "epoch": 1.1816024780273438e-05,
      "step": 12390
    },
    {
      "action_loss_avg": 0.006847637402825057,
      "epoch": 1.1816024780273438e-05,
      "step": 12390
    },
    {
      "epoch": 1.1816024780273438e-05,
      "step": 12390,
      "training_step_time": 17.49475932121277
    },
    {
      "epoch": 1.1816978454589844e-05,
      "model_forward_time": 5.623715400695801,
      "step": 12391
    },
    {
      "epoch": 1.1816978454589844e-05,
      "step": 12391,
      "training_step_time": 17.14481258392334
    },
    {
      "epoch": 1.181793212890625e-05,
      "model_forward_time": 5.632784128189087,
      "step": 12392
    },
    {
      "epoch": 1.181793212890625e-05,
      "step": 12392,
      "training_step_time": 17.130435943603516
    },
    {
      "epoch": 1.1818885803222657e-05,
      "model_forward_time": 5.607462167739868,
      "step": 12393
    },
    {
      "epoch": 1.1818885803222657e-05,
      "step": 12393,
      "training_step_time": 17.09263825416565
    },
    {
      "epoch": 1.1819839477539063e-05,
      "model_forward_time": 5.61181640625,
      "step": 12394
    },
    {
      "epoch": 1.1819839477539063e-05,
      "step": 12394,
      "training_step_time": 17.109935522079468
    },
    {
      "epoch": 1.1820793151855468e-05,
      "model_forward_time": 5.611340522766113,
      "step": 12395
    },
    {
      "epoch": 1.1820793151855468e-05,
      "step": 12395,
      "training_step_time": 17.44195008277893
    },
    {
      "epoch": 1.1821746826171876e-05,
      "model_forward_time": 5.5968852043151855,
      "step": 12396
    },
    {
      "epoch": 1.1821746826171876e-05,
      "step": 12396,
      "training_step_time": 17.04938268661499
    },
    {
      "epoch": 1.1822700500488281e-05,
      "model_forward_time": 5.578899145126343,
      "step": 12397
    },
    {
      "epoch": 1.1822700500488281e-05,
      "step": 12397,
      "training_step_time": 16.994881868362427
    },
    {
      "epoch": 1.1823654174804687e-05,
      "model_forward_time": 5.56784200668335,
      "step": 12398
    },
    {
      "epoch": 1.1823654174804687e-05,
      "step": 12398,
      "training_step_time": 16.98059344291687
    },
    {
      "epoch": 1.1824607849121093e-05,
      "model_forward_time": 5.561892986297607,
      "step": 12399
    },
    {
      "epoch": 1.1824607849121093e-05,
      "step": 12399,
      "training_step_time": 16.990852117538452
    },
    {
      "epoch": 1.18255615234375e-05,
      "grad_norm": 0.8750392198562622,
      "learning_rate": 1e-05,
      "loss": 0.0565,
      "step": 12400
    },
    {
      "epoch": 1.18255615234375e-05,
      "model_forward_time": 5.5752294063568115,
      "step": 12400
    },
    {
      "dynamics_loss_avg": 0.053265557065606116,
      "epoch": 1.18255615234375e-05,
      "step": 12400
    },
    {
      "action_loss_avg": 0.005450050439685583,
      "epoch": 1.18255615234375e-05,
      "step": 12400
    },
    {
      "epoch": 1.18255615234375e-05,
      "step": 12400,
      "training_step_time": 17.013291835784912
    },
    {
      "epoch": 1.1826515197753906e-05,
      "model_forward_time": 5.958542823791504,
      "step": 12401
    },
    {
      "epoch": 1.1826515197753906e-05,
      "step": 12401,
      "training_step_time": 17.36348032951355
    },
    {
      "epoch": 1.1827468872070312e-05,
      "model_forward_time": 5.579532861709595,
      "step": 12402
    },
    {
      "epoch": 1.1827468872070312e-05,
      "step": 12402,
      "training_step_time": 17.017964601516724
    },
    {
      "epoch": 1.182842254638672e-05,
      "model_forward_time": 5.596643924713135,
      "step": 12403
    },
    {
      "epoch": 1.182842254638672e-05,
      "step": 12403,
      "training_step_time": 17.065731525421143
    },
    {
      "epoch": 1.1829376220703125e-05,
      "model_forward_time": 5.608244895935059,
      "step": 12404
    },
    {
      "epoch": 1.1829376220703125e-05,
      "step": 12404,
      "training_step_time": 17.101654052734375
    },
    {
      "epoch": 1.183032989501953e-05,
      "model_forward_time": 5.615376949310303,
      "step": 12405
    },
    {
      "epoch": 1.183032989501953e-05,
      "step": 12405,
      "training_step_time": 17.111790895462036
    },
    {
      "epoch": 1.1831283569335938e-05,
      "model_forward_time": 5.6161134243011475,
      "step": 12406
    },
    {
      "epoch": 1.1831283569335938e-05,
      "step": 12406,
      "training_step_time": 17.458340167999268
    },
    {
      "epoch": 1.1832237243652344e-05,
      "model_forward_time": 5.59397292137146,
      "step": 12407
    },
    {
      "epoch": 1.1832237243652344e-05,
      "step": 12407,
      "training_step_time": 17.026745796203613
    },
    {
      "epoch": 1.183319091796875e-05,
      "model_forward_time": 5.58739972114563,
      "step": 12408
    },
    {
      "epoch": 1.183319091796875e-05,
      "step": 12408,
      "training_step_time": 17.00286054611206
    },
    {
      "epoch": 1.1834144592285157e-05,
      "model_forward_time": 5.582799673080444,
      "step": 12409
    },
    {
      "epoch": 1.1834144592285157e-05,
      "step": 12409,
      "training_step_time": 16.98959732055664
    },
    {
      "epoch": 1.1835098266601563e-05,
      "grad_norm": 0.6962046027183533,
      "learning_rate": 1e-05,
      "loss": 0.0606,
      "step": 12410
    },
    {
      "epoch": 1.1835098266601563e-05,
      "model_forward_time": 5.592564582824707,
      "step": 12410
    },
    {
      "dynamics_loss_avg": 0.055477744340896605,
      "epoch": 1.1835098266601563e-05,
      "step": 12410
    },
    {
      "action_loss_avg": 0.0055933316936716436,
      "epoch": 1.1835098266601563e-05,
      "step": 12410
    },
    {
      "epoch": 1.1835098266601563e-05,
      "step": 12410,
      "training_step_time": 17.04733657836914
    },
    {
      "epoch": 1.1836051940917968e-05,
      "model_forward_time": 5.585921049118042,
      "step": 12411
    },
    {
      "epoch": 1.1836051940917968e-05,
      "step": 12411,
      "training_step_time": 17.018540143966675
    },
    {
      "epoch": 1.1837005615234376e-05,
      "model_forward_time": 5.973835468292236,
      "step": 12412
    },
    {
      "epoch": 1.1837005615234376e-05,
      "step": 12412,
      "training_step_time": 17.429508447647095
    },
    {
      "epoch": 1.1837959289550781e-05,
      "model_forward_time": 5.603636980056763,
      "step": 12413
    },
    {
      "epoch": 1.1837959289550781e-05,
      "step": 12413,
      "training_step_time": 17.0953586101532
    },
    {
      "epoch": 1.1838912963867187e-05,
      "model_forward_time": 5.608030080795288,
      "step": 12414
    },
    {
      "epoch": 1.1838912963867187e-05,
      "step": 12414,
      "training_step_time": 17.10906434059143
    },
    {
      "epoch": 1.1839866638183593e-05,
      "model_forward_time": 5.633635520935059,
      "step": 12415
    },
    {
      "epoch": 1.1839866638183593e-05,
      "step": 12415,
      "training_step_time": 17.09731149673462
    },
    {
      "epoch": 1.18408203125e-05,
      "model_forward_time": 5.601796388626099,
      "step": 12416
    },
    {
      "epoch": 1.18408203125e-05,
      "step": 12416,
      "training_step_time": 17.068545818328857
    },
    {
      "epoch": 1.1841773986816406e-05,
      "model_forward_time": 5.965697288513184,
      "step": 12417
    },
    {
      "epoch": 1.1841773986816406e-05,
      "step": 12417,
      "training_step_time": 17.443979501724243
    },
    {
      "epoch": 1.1842727661132812e-05,
      "model_forward_time": 5.608903646469116,
      "step": 12418
    },
    {
      "epoch": 1.1842727661132812e-05,
      "step": 12418,
      "training_step_time": 17.11205267906189
    },
    {
      "epoch": 1.184368133544922e-05,
      "model_forward_time": 5.621120929718018,
      "step": 12419
    },
    {
      "epoch": 1.184368133544922e-05,
      "step": 12419,
      "training_step_time": 17.143159866333008
    },
    {
      "epoch": 1.1844635009765625e-05,
      "grad_norm": 0.460750937461853,
      "learning_rate": 1e-05,
      "loss": 0.0614,
      "step": 12420
    },
    {
      "epoch": 1.1844635009765625e-05,
      "model_forward_time": 5.623417377471924,
      "step": 12420
    },
    {
      "dynamics_loss_avg": 0.05228774314746261,
      "epoch": 1.1844635009765625e-05,
      "step": 12420
    },
    {
      "action_loss_avg": 0.007093823462491855,
      "epoch": 1.1844635009765625e-05,
      "step": 12420
    },
    {
      "epoch": 1.1844635009765625e-05,
      "step": 12420,
      "training_step_time": 17.129449367523193
    },
    {
      "epoch": 1.184558868408203e-05,
      "model_forward_time": 5.608581304550171,
      "step": 12421
    },
    {
      "epoch": 1.184558868408203e-05,
      "step": 12421,
      "training_step_time": 17.1001877784729
    },
    {
      "epoch": 1.1846542358398438e-05,
      "model_forward_time": 5.596033334732056,
      "step": 12422
    },
    {
      "epoch": 1.1846542358398438e-05,
      "step": 12422,
      "training_step_time": 17.045079946517944
    },
    {
      "epoch": 1.1847496032714844e-05,
      "model_forward_time": 5.949705600738525,
      "step": 12423
    },
    {
      "epoch": 1.1847496032714844e-05,
      "step": 12423,
      "training_step_time": 17.40220832824707
    },
    {
      "epoch": 1.184844970703125e-05,
      "model_forward_time": 4.26428484916687,
      "step": 12424
    },
    {
      "epoch": 1.184844970703125e-05,
      "step": 12424,
      "training_step_time": 12.790688514709473
    },
    {
      "epoch": 1.1849403381347657e-05,
      "model_forward_time": 5.596247911453247,
      "step": 12425
    },
    {
      "epoch": 1.1849403381347657e-05,
      "step": 12425,
      "training_step_time": 17.084546327590942
    },
    {
      "epoch": 1.1850357055664063e-05,
      "model_forward_time": 5.605445623397827,
      "step": 12426
    },
    {
      "epoch": 1.1850357055664063e-05,
      "step": 12426,
      "training_step_time": 17.068019151687622
    },
    {
      "epoch": 1.1851310729980468e-05,
      "model_forward_time": 5.598259210586548,
      "step": 12427
    },
    {
      "epoch": 1.1851310729980468e-05,
      "step": 12427,
      "training_step_time": 17.059611558914185
    },
    {
      "epoch": 1.1852264404296876e-05,
      "model_forward_time": 5.606484413146973,
      "step": 12428
    },
    {
      "epoch": 1.1852264404296876e-05,
      "step": 12428,
      "training_step_time": 17.46498394012451
    },
    {
      "epoch": 1.1853218078613282e-05,
      "model_forward_time": 5.601378679275513,
      "step": 12429
    },
    {
      "epoch": 1.1853218078613282e-05,
      "step": 12429,
      "training_step_time": 17.077738523483276
    },
    {
      "epoch": 1.1854171752929687e-05,
      "grad_norm": 0.794371485710144,
      "learning_rate": 1e-05,
      "loss": 0.0622,
      "step": 12430
    },
    {
      "epoch": 1.1854171752929687e-05,
      "model_forward_time": 5.609013795852661,
      "step": 12430
    },
    {
      "dynamics_loss_avg": 0.05347339008003473,
      "epoch": 1.1854171752929687e-05,
      "step": 12430
    },
    {
      "action_loss_avg": 0.007236566348001361,
      "epoch": 1.1854171752929687e-05,
      "step": 12430
    },
    {
      "epoch": 1.1854171752929687e-05,
      "step": 12430,
      "training_step_time": 17.07241916656494
    },
    {
      "epoch": 1.1855125427246093e-05,
      "model_forward_time": 5.596341609954834,
      "step": 12431
    },
    {
      "epoch": 1.1855125427246093e-05,
      "step": 12431,
      "training_step_time": 17.064502477645874
    },
    {
      "epoch": 1.18560791015625e-05,
      "model_forward_time": 5.613147020339966,
      "step": 12432
    },
    {
      "epoch": 1.18560791015625e-05,
      "step": 12432,
      "training_step_time": 17.115432500839233
    },
    {
      "epoch": 1.1857032775878906e-05,
      "model_forward_time": 5.608990430831909,
      "step": 12433
    },
    {
      "epoch": 1.1857032775878906e-05,
      "step": 12433,
      "training_step_time": 17.11234712600708
    },
    {
      "epoch": 1.1857986450195312e-05,
      "model_forward_time": 5.981334209442139,
      "step": 12434
    },
    {
      "epoch": 1.1857986450195312e-05,
      "step": 12434,
      "training_step_time": 17.451900005340576
    },
    {
      "epoch": 1.185894012451172e-05,
      "model_forward_time": 5.601088285446167,
      "step": 12435
    },
    {
      "epoch": 1.185894012451172e-05,
      "step": 12435,
      "training_step_time": 17.06911849975586
    },
    {
      "epoch": 1.1859893798828125e-05,
      "model_forward_time": 5.598330497741699,
      "step": 12436
    },
    {
      "epoch": 1.1859893798828125e-05,
      "step": 12436,
      "training_step_time": 17.080429315567017
    },
    {
      "epoch": 1.186084747314453e-05,
      "model_forward_time": 5.6066296100616455,
      "step": 12437
    },
    {
      "epoch": 1.186084747314453e-05,
      "step": 12437,
      "training_step_time": 17.086884021759033
    },
    {
      "epoch": 1.1861801147460938e-05,
      "model_forward_time": 5.604336500167847,
      "step": 12438
    },
    {
      "epoch": 1.1861801147460938e-05,
      "step": 12438,
      "training_step_time": 17.051425457000732
    },
    {
      "epoch": 1.1862754821777344e-05,
      "model_forward_time": 5.959632635116577,
      "step": 12439
    },
    {
      "epoch": 1.1862754821777344e-05,
      "step": 12439,
      "training_step_time": 17.41767168045044
    },
    {
      "epoch": 1.186370849609375e-05,
      "grad_norm": 0.7911350727081299,
      "learning_rate": 1e-05,
      "loss": 0.0681,
      "step": 12440
    },
    {
      "epoch": 1.186370849609375e-05,
      "model_forward_time": 5.599759817123413,
      "step": 12440
    },
    {
      "dynamics_loss_avg": 0.0613199008628726,
      "epoch": 1.186370849609375e-05,
      "step": 12440
    },
    {
      "action_loss_avg": 0.006507668038830161,
      "epoch": 1.186370849609375e-05,
      "step": 12440
    },
    {
      "epoch": 1.186370849609375e-05,
      "step": 12440,
      "training_step_time": 17.0661883354187
    },
    {
      "epoch": 1.1864662170410157e-05,
      "model_forward_time": 5.624143600463867,
      "step": 12441
    },
    {
      "epoch": 1.1864662170410157e-05,
      "step": 12441,
      "training_step_time": 17.098914623260498
    },
    {
      "epoch": 1.1865615844726563e-05,
      "model_forward_time": 5.601803779602051,
      "step": 12442
    },
    {
      "epoch": 1.1865615844726563e-05,
      "step": 12442,
      "training_step_time": 17.062617778778076
    },
    {
      "epoch": 1.1866569519042968e-05,
      "model_forward_time": 5.601036548614502,
      "step": 12443
    },
    {
      "epoch": 1.1866569519042968e-05,
      "step": 12443,
      "training_step_time": 17.08155345916748
    },
    {
      "epoch": 1.1867523193359376e-05,
      "model_forward_time": 5.604466676712036,
      "step": 12444
    },
    {
      "epoch": 1.1867523193359376e-05,
      "step": 12444,
      "training_step_time": 17.06552505493164
    },
    {
      "epoch": 1.1868476867675782e-05,
      "model_forward_time": 5.965280771255493,
      "step": 12445
    },
    {
      "epoch": 1.1868476867675782e-05,
      "step": 12445,
      "training_step_time": 17.414541006088257
    },
    {
      "epoch": 1.1869430541992187e-05,
      "model_forward_time": 5.5911219120025635,
      "step": 12446
    },
    {
      "epoch": 1.1869430541992187e-05,
      "step": 12446,
      "training_step_time": 17.038955450057983
    },
    {
      "epoch": 1.1870384216308593e-05,
      "model_forward_time": 5.594837427139282,
      "step": 12447
    },
    {
      "epoch": 1.1870384216308593e-05,
      "step": 12447,
      "training_step_time": 17.050463438034058
    },
    {
      "epoch": 1.1871337890625e-05,
      "model_forward_time": 5.594733238220215,
      "step": 12448
    },
    {
      "epoch": 1.1871337890625e-05,
      "step": 12448,
      "training_step_time": 17.05150842666626
    },
    {
      "epoch": 1.1872291564941406e-05,
      "model_forward_time": 5.592524766921997,
      "step": 12449
    },
    {
      "epoch": 1.1872291564941406e-05,
      "step": 12449,
      "training_step_time": 17.024367094039917
    },
    {
      "epoch": 1.1873245239257812e-05,
      "grad_norm": 0.5796422958374023,
      "learning_rate": 1e-05,
      "loss": 0.0588,
      "step": 12450
    },
    {
      "epoch": 1.1873245239257812e-05,
      "model_forward_time": 5.951153993606567,
      "step": 12450
    },
    {
      "dynamics_loss_avg": 0.05467451419681311,
      "epoch": 1.1873245239257812e-05,
      "step": 12450
    },
    {
      "action_loss_avg": 0.005116482498124242,
      "epoch": 1.1873245239257812e-05,
      "step": 12450
    },
    {
      "epoch": 1.1873245239257812e-05,
      "step": 12450,
      "training_step_time": 17.419025659561157
    },
    {
      "epoch": 1.187419891357422e-05,
      "model_forward_time": 5.6006269454956055,
      "step": 12451
    },
    {
      "epoch": 1.187419891357422e-05,
      "step": 12451,
      "training_step_time": 17.04794979095459
    },
    {
      "epoch": 1.1875152587890625e-05,
      "model_forward_time": 5.593765735626221,
      "step": 12452
    },
    {
      "epoch": 1.1875152587890625e-05,
      "step": 12452,
      "training_step_time": 17.06423807144165
    },
    {
      "epoch": 1.187610626220703e-05,
      "model_forward_time": 5.602746963500977,
      "step": 12453
    },
    {
      "epoch": 1.187610626220703e-05,
      "step": 12453,
      "training_step_time": 17.095295667648315
    },
    {
      "epoch": 1.1877059936523438e-05,
      "model_forward_time": 5.615964651107788,
      "step": 12454
    },
    {
      "epoch": 1.1877059936523438e-05,
      "step": 12454,
      "training_step_time": 17.124051094055176
    },
    {
      "epoch": 1.1878013610839844e-05,
      "model_forward_time": 5.612848997116089,
      "step": 12455
    },
    {
      "epoch": 1.1878013610839844e-05,
      "step": 12455,
      "training_step_time": 17.465999841690063
    },
    {
      "epoch": 1.187896728515625e-05,
      "model_forward_time": 5.596937656402588,
      "step": 12456
    },
    {
      "epoch": 1.187896728515625e-05,
      "step": 12456,
      "training_step_time": 17.037134885787964
    },
    {
      "epoch": 1.1879920959472657e-05,
      "model_forward_time": 5.601397752761841,
      "step": 12457
    },
    {
      "epoch": 1.1879920959472657e-05,
      "step": 12457,
      "training_step_time": 17.05630087852478
    },
    {
      "epoch": 1.1880874633789063e-05,
      "model_forward_time": 5.607701063156128,
      "step": 12458
    },
    {
      "epoch": 1.1880874633789063e-05,
      "step": 12458,
      "training_step_time": 17.065775156021118
    },
    {
      "epoch": 1.1881828308105468e-05,
      "model_forward_time": 5.598769426345825,
      "step": 12459
    },
    {
      "epoch": 1.1881828308105468e-05,
      "step": 12459,
      "training_step_time": 17.074440002441406
    },
    {
      "epoch": 1.1882781982421876e-05,
      "grad_norm": 1.196527361869812,
      "learning_rate": 1e-05,
      "loss": 0.0654,
      "step": 12460
    },
    {
      "epoch": 1.1882781982421876e-05,
      "model_forward_time": 5.61263370513916,
      "step": 12460
    },
    {
      "dynamics_loss_avg": 0.06042951792478561,
      "epoch": 1.1882781982421876e-05,
      "step": 12460
    },
    {
      "action_loss_avg": 0.007766971754608676,
      "epoch": 1.1882781982421876e-05,
      "step": 12460
    },
    {
      "epoch": 1.1882781982421876e-05,
      "step": 12460,
      "training_step_time": 17.10815453529358
    },
    {
      "epoch": 1.1883735656738282e-05,
      "model_forward_time": 5.981477737426758,
      "step": 12461
    },
    {
      "epoch": 1.1883735656738282e-05,
      "step": 12461,
      "training_step_time": 17.454314470291138
    },
    {
      "epoch": 1.1884689331054687e-05,
      "model_forward_time": 5.611785173416138,
      "step": 12462
    },
    {
      "epoch": 1.1884689331054687e-05,
      "step": 12462,
      "training_step_time": 17.07945990562439
    },
    {
      "epoch": 1.1885643005371093e-05,
      "model_forward_time": 5.59857177734375,
      "step": 12463
    },
    {
      "epoch": 1.1885643005371093e-05,
      "step": 12463,
      "training_step_time": 17.068219423294067
    },
    {
      "epoch": 1.18865966796875e-05,
      "model_forward_time": 5.613483428955078,
      "step": 12464
    },
    {
      "epoch": 1.18865966796875e-05,
      "step": 12464,
      "training_step_time": 17.08308243751526
    },
    {
      "epoch": 1.1887550354003906e-05,
      "model_forward_time": 5.600771903991699,
      "step": 12465
    },
    {
      "epoch": 1.1887550354003906e-05,
      "step": 12465,
      "training_step_time": 17.074321269989014
    },
    {
      "epoch": 1.1888504028320312e-05,
      "model_forward_time": 5.597359657287598,
      "step": 12466
    },
    {
      "epoch": 1.1888504028320312e-05,
      "step": 12466,
      "training_step_time": 17.41710138320923
    },
    {
      "epoch": 1.188945770263672e-05,
      "model_forward_time": 5.590529203414917,
      "step": 12467
    },
    {
      "epoch": 1.188945770263672e-05,
      "step": 12467,
      "training_step_time": 17.03788471221924
    },
    {
      "epoch": 1.1890411376953125e-05,
      "model_forward_time": 5.60131573677063,
      "step": 12468
    },
    {
      "epoch": 1.1890411376953125e-05,
      "step": 12468,
      "training_step_time": 17.062662839889526
    },
    {
      "epoch": 1.189136505126953e-05,
      "model_forward_time": 5.606974124908447,
      "step": 12469
    },
    {
      "epoch": 1.189136505126953e-05,
      "step": 12469,
      "training_step_time": 17.070545434951782
    },
    {
      "epoch": 1.1892318725585938e-05,
      "grad_norm": 1.052719235420227,
      "learning_rate": 1e-05,
      "loss": 0.0707,
      "step": 12470
    },
    {
      "epoch": 1.1892318725585938e-05,
      "model_forward_time": 5.6001951694488525,
      "step": 12470
    },
    {
      "dynamics_loss_avg": 0.061313527077436446,
      "epoch": 1.1892318725585938e-05,
      "step": 12470
    },
    {
      "action_loss_avg": 0.006424725288525224,
      "epoch": 1.1892318725585938e-05,
      "step": 12470
    },
    {
      "epoch": 1.1892318725585938e-05,
      "step": 12470,
      "training_step_time": 17.07189393043518
    },
    {
      "epoch": 1.1893272399902344e-05,
      "model_forward_time": 5.602842807769775,
      "step": 12471
    },
    {
      "epoch": 1.1893272399902344e-05,
      "step": 12471,
      "training_step_time": 17.07879877090454
    },
    {
      "epoch": 1.189422607421875e-05,
      "model_forward_time": 5.9789958000183105,
      "step": 12472
    },
    {
      "epoch": 1.189422607421875e-05,
      "step": 12472,
      "training_step_time": 17.454272747039795
    },
    {
      "epoch": 1.1895179748535157e-05,
      "model_forward_time": 5.60737419128418,
      "step": 12473
    },
    {
      "epoch": 1.1895179748535157e-05,
      "step": 12473,
      "training_step_time": 17.0817928314209
    },
    {
      "epoch": 1.1896133422851563e-05,
      "model_forward_time": 5.594234466552734,
      "step": 12474
    },
    {
      "epoch": 1.1896133422851563e-05,
      "step": 12474,
      "training_step_time": 17.043367624282837
    },
    {
      "epoch": 1.1897087097167968e-05,
      "model_forward_time": 5.595104217529297,
      "step": 12475
    },
    {
      "epoch": 1.1897087097167968e-05,
      "step": 12475,
      "training_step_time": 17.02300500869751
    },
    {
      "epoch": 1.1898040771484376e-05,
      "model_forward_time": 5.593576192855835,
      "step": 12476
    },
    {
      "epoch": 1.1898040771484376e-05,
      "step": 12476,
      "training_step_time": 17.058361768722534
    },
    {
      "epoch": 1.1898994445800782e-05,
      "model_forward_time": 5.983376979827881,
      "step": 12477
    },
    {
      "epoch": 1.1898994445800782e-05,
      "step": 12477,
      "training_step_time": 17.461360454559326
    },
    {
      "epoch": 1.1899948120117187e-05,
      "model_forward_time": 4.286682367324829,
      "step": 12478
    },
    {
      "epoch": 1.1899948120117187e-05,
      "step": 12478,
      "training_step_time": 12.839745283126831
    },
    {
      "epoch": 1.1900901794433593e-05,
      "model_forward_time": 5.624175071716309,
      "step": 12479
    },
    {
      "epoch": 1.1900901794433593e-05,
      "step": 12479,
      "training_step_time": 17.147531986236572
    },
    {
      "epoch": 1.190185546875e-05,
      "grad_norm": 0.6504266262054443,
      "learning_rate": 1e-05,
      "loss": 0.0518,
      "step": 12480
    },
    {
      "epoch": 1.190185546875e-05,
      "model_forward_time": 5.628330230712891,
      "step": 12480
    },
    {
      "dynamics_loss_avg": 0.05800690958276391,
      "epoch": 1.190185546875e-05,
      "step": 12480
    },
    {
      "action_loss_avg": 0.00810632836073637,
      "epoch": 1.190185546875e-05,
      "step": 12480
    },
    {
      "epoch": 1.190185546875e-05,
      "step": 12480,
      "training_step_time": 17.08386254310608
    },
    {
      "epoch": 1.1902809143066406e-05,
      "model_forward_time": 5.590870380401611,
      "step": 12481
    },
    {
      "epoch": 1.1902809143066406e-05,
      "step": 12481,
      "training_step_time": 17.02717900276184
    },
    {
      "epoch": 1.1903762817382812e-05,
      "model_forward_time": 5.579146385192871,
      "step": 12482
    },
    {
      "epoch": 1.1903762817382812e-05,
      "step": 12482,
      "training_step_time": 17.013660430908203
    },
    {
      "epoch": 1.190471649169922e-05,
      "model_forward_time": 5.952224016189575,
      "step": 12483
    },
    {
      "epoch": 1.190471649169922e-05,
      "step": 12483,
      "training_step_time": 17.417247772216797
    },
    {
      "epoch": 1.1905670166015625e-05,
      "model_forward_time": 5.607267379760742,
      "step": 12484
    },
    {
      "epoch": 1.1905670166015625e-05,
      "step": 12484,
      "training_step_time": 17.112396240234375
    },
    {
      "epoch": 1.190662384033203e-05,
      "model_forward_time": 5.62836480140686,
      "step": 12485
    },
    {
      "epoch": 1.190662384033203e-05,
      "step": 12485,
      "training_step_time": 17.140823364257812
    },
    {
      "epoch": 1.1907577514648438e-05,
      "model_forward_time": 5.628596782684326,
      "step": 12486
    },
    {
      "epoch": 1.1907577514648438e-05,
      "step": 12486,
      "training_step_time": 17.14400291442871
    },
    {
      "epoch": 1.1908531188964844e-05,
      "model_forward_time": 5.628891229629517,
      "step": 12487
    },
    {
      "epoch": 1.1908531188964844e-05,
      "step": 12487,
      "training_step_time": 17.13388466835022
    },
    {
      "epoch": 1.190948486328125e-05,
      "model_forward_time": 5.998820543289185,
      "step": 12488
    },
    {
      "epoch": 1.190948486328125e-05,
      "step": 12488,
      "training_step_time": 17.50979495048523
    },
    {
      "epoch": 1.1910438537597657e-05,
      "model_forward_time": 5.619041681289673,
      "step": 12489
    },
    {
      "epoch": 1.1910438537597657e-05,
      "step": 12489,
      "training_step_time": 17.083571672439575
    },
    {
      "epoch": 1.1911392211914063e-05,
      "grad_norm": 0.7959701418876648,
      "learning_rate": 1e-05,
      "loss": 0.0775,
      "step": 12490
    },
    {
      "epoch": 1.1911392211914063e-05,
      "model_forward_time": 5.623245000839233,
      "step": 12490
    },
    {
      "dynamics_loss_avg": 0.05455060750246048,
      "epoch": 1.1911392211914063e-05,
      "step": 12490
    },
    {
      "action_loss_avg": 0.0077089938335120674,
      "epoch": 1.1911392211914063e-05,
      "step": 12490
    },
    {
      "epoch": 1.1911392211914063e-05,
      "step": 12490,
      "training_step_time": 17.04428243637085
    },
    {
      "epoch": 1.1912345886230468e-05,
      "model_forward_time": 5.569690704345703,
      "step": 12491
    },
    {
      "epoch": 1.1912345886230468e-05,
      "step": 12491,
      "training_step_time": 16.991490364074707
    },
    {
      "epoch": 1.1913299560546876e-05,
      "model_forward_time": 5.557900905609131,
      "step": 12492
    },
    {
      "epoch": 1.1913299560546876e-05,
      "step": 12492,
      "training_step_time": 16.98647379875183
    },
    {
      "epoch": 1.1914253234863282e-05,
      "model_forward_time": 5.573766231536865,
      "step": 12493
    },
    {
      "epoch": 1.1914253234863282e-05,
      "step": 12493,
      "training_step_time": 16.981849670410156
    },
    {
      "epoch": 1.1915206909179687e-05,
      "model_forward_time": 5.941867828369141,
      "step": 12494
    },
    {
      "epoch": 1.1915206909179687e-05,
      "step": 12494,
      "training_step_time": 17.34475874900818
    },
    {
      "epoch": 1.1916160583496093e-05,
      "model_forward_time": 5.569005727767944,
      "step": 12495
    },
    {
      "epoch": 1.1916160583496093e-05,
      "step": 12495,
      "training_step_time": 17.00755476951599
    },
    {
      "epoch": 1.19171142578125e-05,
      "model_forward_time": 5.584906816482544,
      "step": 12496
    },
    {
      "epoch": 1.19171142578125e-05,
      "step": 12496,
      "training_step_time": 16.998419523239136
    },
    {
      "epoch": 1.1918067932128906e-05,
      "model_forward_time": 5.582773923873901,
      "step": 12497
    },
    {
      "epoch": 1.1918067932128906e-05,
      "step": 12497,
      "training_step_time": 17.009740352630615
    },
    {
      "epoch": 1.1919021606445312e-05,
      "model_forward_time": 5.60825252532959,
      "step": 12498
    },
    {
      "epoch": 1.1919021606445312e-05,
      "step": 12498,
      "training_step_time": 17.056519031524658
    },
    {
      "epoch": 1.191997528076172e-05,
      "model_forward_time": 5.960410118103027,
      "step": 12499
    },
    {
      "epoch": 1.191997528076172e-05,
      "step": 12499,
      "training_step_time": 17.44923758506775
    },
    {
      "epoch": 1.1920928955078125e-05,
      "grad_norm": 0.7289741635322571,
      "learning_rate": 1e-05,
      "loss": 0.0565,
      "step": 12500
    },
    {
      "epoch": 1.1920928955078125e-05,
      "model_forward_time": 5.613956689834595,
      "step": 12500
    },
    {
      "dynamics_loss_avg": 0.059442630223929885,
      "epoch": 1.1920928955078125e-05,
      "step": 12500
    },
    {
      "action_loss_avg": 0.005787495849654078,
      "epoch": 1.1920928955078125e-05,
      "step": 12500
    },
    {
      "epoch": 1.1920928955078125e-05,
      "step": 12500,
      "training_step_time": 17.12136745452881
    },
    {
      "epoch": 1.192188262939453e-05,
      "model_forward_time": 5.629607200622559,
      "step": 12501
    },
    {
      "epoch": 1.192188262939453e-05,
      "step": 12501,
      "training_step_time": 17.132905960083008
    },
    {
      "epoch": 1.1922836303710938e-05,
      "model_forward_time": 5.610621929168701,
      "step": 12502
    },
    {
      "epoch": 1.1922836303710938e-05,
      "step": 12502,
      "training_step_time": 17.08182430267334
    },
    {
      "epoch": 1.1923789978027344e-05,
      "model_forward_time": 5.616986274719238,
      "step": 12503
    },
    {
      "epoch": 1.1923789978027344e-05,
      "step": 12503,
      "training_step_time": 17.05791401863098
    },
    {
      "epoch": 1.192474365234375e-05,
      "model_forward_time": 5.596550941467285,
      "step": 12504
    },
    {
      "epoch": 1.192474365234375e-05,
      "step": 12504,
      "training_step_time": 17.032326698303223
    },
    {
      "epoch": 1.1925697326660157e-05,
      "model_forward_time": 5.938011169433594,
      "step": 12505
    },
    {
      "epoch": 1.1925697326660157e-05,
      "step": 12505,
      "training_step_time": 17.38983964920044
    },
    {
      "epoch": 1.1926651000976563e-05,
      "model_forward_time": 5.592643976211548,
      "step": 12506
    },
    {
      "epoch": 1.1926651000976563e-05,
      "step": 12506,
      "training_step_time": 17.049108266830444
    },
    {
      "epoch": 1.1927604675292968e-05,
      "model_forward_time": 5.6175079345703125,
      "step": 12507
    },
    {
      "epoch": 1.1927604675292968e-05,
      "step": 12507,
      "training_step_time": 17.102023124694824
    },
    {
      "epoch": 1.1928558349609374e-05,
      "model_forward_time": 5.607928037643433,
      "step": 12508
    },
    {
      "epoch": 1.1928558349609374e-05,
      "step": 12508,
      "training_step_time": 17.105509757995605
    },
    {
      "epoch": 1.1929512023925782e-05,
      "model_forward_time": 5.604371547698975,
      "step": 12509
    },
    {
      "epoch": 1.1929512023925782e-05,
      "step": 12509,
      "training_step_time": 17.08259391784668
    },
    {
      "epoch": 1.1930465698242187e-05,
      "grad_norm": 1.0090712308883667,
      "learning_rate": 1e-05,
      "loss": 0.0706,
      "step": 12510
    },
    {
      "epoch": 1.1930465698242187e-05,
      "model_forward_time": 5.98053765296936,
      "step": 12510
    },
    {
      "dynamics_loss_avg": 0.05487505923956633,
      "epoch": 1.1930465698242187e-05,
      "step": 12510
    },
    {
      "action_loss_avg": 0.007582359970547259,
      "epoch": 1.1930465698242187e-05,
      "step": 12510
    },
    {
      "epoch": 1.1930465698242187e-05,
      "step": 12510,
      "training_step_time": 17.45452618598938
    },
    {
      "epoch": 1.1931419372558593e-05,
      "model_forward_time": 5.59547233581543,
      "step": 12511
    },
    {
      "epoch": 1.1931419372558593e-05,
      "step": 12511,
      "training_step_time": 17.059326887130737
    },
    {
      "epoch": 1.1932373046875e-05,
      "model_forward_time": 5.6098949909210205,
      "step": 12512
    },
    {
      "epoch": 1.1932373046875e-05,
      "step": 12512,
      "training_step_time": 17.073922634124756
    },
    {
      "epoch": 1.1933326721191406e-05,
      "model_forward_time": 5.600151538848877,
      "step": 12513
    },
    {
      "epoch": 1.1933326721191406e-05,
      "step": 12513,
      "training_step_time": 17.067298650741577
    },
    {
      "epoch": 1.1934280395507812e-05,
      "model_forward_time": 5.603601932525635,
      "step": 12514
    },
    {
      "epoch": 1.1934280395507812e-05,
      "step": 12514,
      "training_step_time": 17.064462900161743
    },
    {
      "epoch": 1.193523406982422e-05,
      "model_forward_time": 5.943975925445557,
      "step": 12515
    },
    {
      "epoch": 1.193523406982422e-05,
      "step": 12515,
      "training_step_time": 17.40468144416809
    },
    {
      "epoch": 1.1936187744140625e-05,
      "model_forward_time": 5.591291666030884,
      "step": 12516
    },
    {
      "epoch": 1.1936187744140625e-05,
      "step": 12516,
      "training_step_time": 17.017061233520508
    },
    {
      "epoch": 1.193714141845703e-05,
      "model_forward_time": 5.594501733779907,
      "step": 12517
    },
    {
      "epoch": 1.193714141845703e-05,
      "step": 12517,
      "training_step_time": 17.032625913619995
    },
    {
      "epoch": 1.1938095092773438e-05,
      "model_forward_time": 5.605509042739868,
      "step": 12518
    },
    {
      "epoch": 1.1938095092773438e-05,
      "step": 12518,
      "training_step_time": 17.074862718582153
    },
    {
      "epoch": 1.1939048767089844e-05,
      "model_forward_time": 5.605304002761841,
      "step": 12519
    },
    {
      "epoch": 1.1939048767089844e-05,
      "step": 12519,
      "training_step_time": 17.107515335083008
    },
    {
      "epoch": 1.194000244140625e-05,
      "grad_norm": 0.7068595290184021,
      "learning_rate": 1e-05,
      "loss": 0.0611,
      "step": 12520
    },
    {
      "epoch": 1.194000244140625e-05,
      "model_forward_time": 5.629859924316406,
      "step": 12520
    },
    {
      "dynamics_loss_avg": 0.06129700653254986,
      "epoch": 1.194000244140625e-05,
      "step": 12520
    },
    {
      "action_loss_avg": 0.009381116821896286,
      "epoch": 1.194000244140625e-05,
      "step": 12520
    },
    {
      "epoch": 1.194000244140625e-05,
      "step": 12520,
      "training_step_time": 17.151872634887695
    },
    {
      "epoch": 1.1940956115722657e-05,
      "model_forward_time": 5.986126899719238,
      "step": 12521
    },
    {
      "epoch": 1.1940956115722657e-05,
      "step": 12521,
      "training_step_time": 17.48957061767578
    },
    {
      "epoch": 1.1941909790039063e-05,
      "model_forward_time": 5.618085861206055,
      "step": 12522
    },
    {
      "epoch": 1.1941909790039063e-05,
      "step": 12522,
      "training_step_time": 17.127026557922363
    },
    {
      "epoch": 1.1942863464355468e-05,
      "model_forward_time": 5.621885538101196,
      "step": 12523
    },
    {
      "epoch": 1.1942863464355468e-05,
      "step": 12523,
      "training_step_time": 17.124539852142334
    },
    {
      "epoch": 1.1943817138671874e-05,
      "model_forward_time": 5.618179559707642,
      "step": 12524
    },
    {
      "epoch": 1.1943817138671874e-05,
      "step": 12524,
      "training_step_time": 17.1110680103302
    },
    {
      "epoch": 1.1944770812988282e-05,
      "model_forward_time": 5.6050004959106445,
      "step": 12525
    },
    {
      "epoch": 1.1944770812988282e-05,
      "step": 12525,
      "training_step_time": 17.056509971618652
    },
    {
      "epoch": 1.1945724487304687e-05,
      "model_forward_time": 5.602770090103149,
      "step": 12526
    },
    {
      "epoch": 1.1945724487304687e-05,
      "step": 12526,
      "training_step_time": 17.43896722793579
    },
    {
      "epoch": 1.1946678161621093e-05,
      "model_forward_time": 5.599190711975098,
      "step": 12527
    },
    {
      "epoch": 1.1946678161621093e-05,
      "step": 12527,
      "training_step_time": 17.091270923614502
    },
    {
      "epoch": 1.19476318359375e-05,
      "model_forward_time": 5.604460954666138,
      "step": 12528
    },
    {
      "epoch": 1.19476318359375e-05,
      "step": 12528,
      "training_step_time": 17.10612392425537
    },
    {
      "epoch": 1.1948585510253906e-05,
      "model_forward_time": 5.6210362911224365,
      "step": 12529
    },
    {
      "epoch": 1.1948585510253906e-05,
      "step": 12529,
      "training_step_time": 17.111687183380127
    },
    {
      "epoch": 1.1949539184570312e-05,
      "grad_norm": 1.02607262134552,
      "learning_rate": 1e-05,
      "loss": 0.072,
      "step": 12530
    },
    {
      "epoch": 1.1949539184570312e-05,
      "model_forward_time": 5.626404762268066,
      "step": 12530
    },
    {
      "dynamics_loss_avg": 0.05558984111994505,
      "epoch": 1.1949539184570312e-05,
      "step": 12530
    },
    {
      "action_loss_avg": 0.008034253760706633,
      "epoch": 1.1949539184570312e-05,
      "step": 12530
    },
    {
      "epoch": 1.1949539184570312e-05,
      "step": 12530,
      "training_step_time": 17.11716079711914
    },
    {
      "epoch": 1.195049285888672e-05,
      "model_forward_time": 5.600008249282837,
      "step": 12531
    },
    {
      "epoch": 1.195049285888672e-05,
      "step": 12531,
      "training_step_time": 17.01930856704712
    },
    {
      "epoch": 1.1951446533203125e-05,
      "model_forward_time": 5.959605693817139,
      "step": 12532
    },
    {
      "epoch": 1.1951446533203125e-05,
      "step": 12532,
      "training_step_time": 17.37592840194702
    },
    {
      "epoch": 1.195240020751953e-05,
      "model_forward_time": 5.578641653060913,
      "step": 12533
    },
    {
      "epoch": 1.195240020751953e-05,
      "step": 12533,
      "training_step_time": 17.01809549331665
    },
    {
      "epoch": 1.1953353881835938e-05,
      "model_forward_time": 5.575075626373291,
      "step": 12534
    },
    {
      "epoch": 1.1953353881835938e-05,
      "step": 12534,
      "training_step_time": 16.992405891418457
    },
    {
      "epoch": 1.1954307556152344e-05,
      "model_forward_time": 5.573982000350952,
      "step": 12535
    },
    {
      "epoch": 1.1954307556152344e-05,
      "step": 12535,
      "training_step_time": 16.988807201385498
    },
    {
      "epoch": 1.195526123046875e-05,
      "model_forward_time": 5.576578855514526,
      "step": 12536
    },
    {
      "epoch": 1.195526123046875e-05,
      "step": 12536,
      "training_step_time": 17.00230383872986
    },
    {
      "epoch": 1.1956214904785157e-05,
      "model_forward_time": 5.9334917068481445,
      "step": 12537
    },
    {
      "epoch": 1.1956214904785157e-05,
      "step": 12537,
      "training_step_time": 17.383930683135986
    },
    {
      "epoch": 1.1957168579101563e-05,
      "model_forward_time": 5.591183185577393,
      "step": 12538
    },
    {
      "epoch": 1.1957168579101563e-05,
      "step": 12538,
      "training_step_time": 17.025824308395386
    },
    {
      "epoch": 1.1958122253417968e-05,
      "model_forward_time": 5.5936572551727295,
      "step": 12539
    },
    {
      "epoch": 1.1958122253417968e-05,
      "step": 12539,
      "training_step_time": 17.056002616882324
    },
    {
      "epoch": 1.1959075927734374e-05,
      "grad_norm": 0.430411159992218,
      "learning_rate": 1e-05,
      "loss": 0.0641,
      "step": 12540
    },
    {
      "epoch": 1.1959075927734374e-05,
      "model_forward_time": 5.602931976318359,
      "step": 12540
    },
    {
      "dynamics_loss_avg": 0.05545930610969663,
      "epoch": 1.1959075927734374e-05,
      "step": 12540
    },
    {
      "action_loss_avg": 0.00570812706137076,
      "epoch": 1.1959075927734374e-05,
      "step": 12540
    },
    {
      "epoch": 1.1959075927734374e-05,
      "step": 12540,
      "training_step_time": 17.087596893310547
    },
    {
      "epoch": 1.1960029602050782e-05,
      "model_forward_time": 5.611436128616333,
      "step": 12541
    },
    {
      "epoch": 1.1960029602050782e-05,
      "step": 12541,
      "training_step_time": 17.130526542663574
    },
    {
      "epoch": 1.1960983276367187e-05,
      "model_forward_time": 5.6185925006866455,
      "step": 12542
    },
    {
      "epoch": 1.1960983276367187e-05,
      "step": 12542,
      "training_step_time": 17.103590726852417
    },
    {
      "epoch": 1.1961936950683593e-05,
      "model_forward_time": 5.983966827392578,
      "step": 12543
    },
    {
      "epoch": 1.1961936950683593e-05,
      "step": 12543,
      "training_step_time": 17.450344800949097
    },
    {
      "epoch": 1.1962890625e-05,
      "model_forward_time": 5.60737943649292,
      "step": 12544
    },
    {
      "epoch": 1.1962890625e-05,
      "step": 12544,
      "training_step_time": 17.10831618309021
    },
    {
      "epoch": 1.1963844299316406e-05,
      "model_forward_time": 5.61229133605957,
      "step": 12545
    },
    {
      "epoch": 1.1963844299316406e-05,
      "step": 12545,
      "training_step_time": 17.11985754966736
    },
    {
      "epoch": 1.1964797973632812e-05,
      "model_forward_time": 5.623044729232788,
      "step": 12546
    },
    {
      "epoch": 1.1964797973632812e-05,
      "step": 12546,
      "training_step_time": 17.131381273269653
    },
    {
      "epoch": 1.196575164794922e-05,
      "model_forward_time": 5.61887264251709,
      "step": 12547
    },
    {
      "epoch": 1.196575164794922e-05,
      "step": 12547,
      "training_step_time": 17.121135234832764
    },
    {
      "epoch": 1.1966705322265625e-05,
      "model_forward_time": 5.96512770652771,
      "step": 12548
    },
    {
      "epoch": 1.1966705322265625e-05,
      "step": 12548,
      "training_step_time": 17.46266007423401
    },
    {
      "epoch": 1.196765899658203e-05,
      "model_forward_time": 5.61244535446167,
      "step": 12549
    },
    {
      "epoch": 1.196765899658203e-05,
      "step": 12549,
      "training_step_time": 17.09774875640869
    },
    {
      "epoch": 1.1968612670898438e-05,
      "grad_norm": 0.6002059578895569,
      "learning_rate": 1e-05,
      "loss": 0.0567,
      "step": 12550
    },
    {
      "epoch": 1.1968612670898438e-05,
      "model_forward_time": 5.615484237670898,
      "step": 12550
    },
    {
      "dynamics_loss_avg": 0.05822325050830841,
      "epoch": 1.1968612670898438e-05,
      "step": 12550
    },
    {
      "action_loss_avg": 0.00592302322620526,
      "epoch": 1.1968612670898438e-05,
      "step": 12550
    },
    {
      "epoch": 1.1968612670898438e-05,
      "step": 12550,
      "training_step_time": 17.08573007583618
    },
    {
      "epoch": 1.1969566345214844e-05,
      "model_forward_time": 5.599446535110474,
      "step": 12551
    },
    {
      "epoch": 1.1969566345214844e-05,
      "step": 12551,
      "training_step_time": 17.04669499397278
    },
    {
      "epoch": 1.197052001953125e-05,
      "model_forward_time": 5.594205617904663,
      "step": 12552
    },
    {
      "epoch": 1.197052001953125e-05,
      "step": 12552,
      "training_step_time": 17.042861700057983
    },
    {
      "epoch": 1.1971473693847657e-05,
      "model_forward_time": 5.58157753944397,
      "step": 12553
    },
    {
      "epoch": 1.1971473693847657e-05,
      "step": 12553,
      "training_step_time": 17.398441076278687
    },
    {
      "epoch": 1.1972427368164063e-05,
      "model_forward_time": 5.594170570373535,
      "step": 12554
    },
    {
      "epoch": 1.1972427368164063e-05,
      "step": 12554,
      "training_step_time": 17.024208784103394
    },
    {
      "epoch": 1.1973381042480468e-05,
      "model_forward_time": 5.593623876571655,
      "step": 12555
    },
    {
      "epoch": 1.1973381042480468e-05,
      "step": 12555,
      "training_step_time": 17.062437295913696
    },
    {
      "epoch": 1.1974334716796874e-05,
      "model_forward_time": 5.611783027648926,
      "step": 12556
    },
    {
      "epoch": 1.1974334716796874e-05,
      "step": 12556,
      "training_step_time": 17.093886375427246
    },
    {
      "epoch": 1.1975288391113282e-05,
      "model_forward_time": 5.6404173374176025,
      "step": 12557
    },
    {
      "epoch": 1.1975288391113282e-05,
      "step": 12557,
      "training_step_time": 17.185014247894287
    },
    {
      "epoch": 1.1976242065429687e-05,
      "model_forward_time": 5.652904033660889,
      "step": 12558
    },
    {
      "epoch": 1.1976242065429687e-05,
      "step": 12558,
      "training_step_time": 17.15536069869995
    },
    {
      "epoch": 1.1977195739746093e-05,
      "model_forward_time": 6.11673641204834,
      "step": 12559
    },
    {
      "epoch": 1.1977195739746093e-05,
      "step": 12559,
      "training_step_time": 17.587198972702026
    },
    {
      "epoch": 1.19781494140625e-05,
      "grad_norm": 0.2750544548034668,
      "learning_rate": 1e-05,
      "loss": 0.0551,
      "step": 12560
    },
    {
      "epoch": 1.19781494140625e-05,
      "model_forward_time": 5.6159608364105225,
      "step": 12560
    },
    {
      "dynamics_loss_avg": 0.04411969557404518,
      "epoch": 1.19781494140625e-05,
      "step": 12560
    },
    {
      "action_loss_avg": 0.005372950155287981,
      "epoch": 1.19781494140625e-05,
      "step": 12560
    },
    {
      "epoch": 1.19781494140625e-05,
      "step": 12560,
      "training_step_time": 17.086433172225952
    },
    {
      "epoch": 1.1979103088378906e-05,
      "model_forward_time": 5.609594106674194,
      "step": 12561
    },
    {
      "epoch": 1.1979103088378906e-05,
      "step": 12561,
      "training_step_time": 17.106877088546753
    },
    {
      "epoch": 1.1980056762695312e-05,
      "model_forward_time": 5.617151975631714,
      "step": 12562
    },
    {
      "epoch": 1.1980056762695312e-05,
      "step": 12562,
      "training_step_time": 17.110023021697998
    },
    {
      "epoch": 1.198101043701172e-05,
      "model_forward_time": 5.623228311538696,
      "step": 12563
    },
    {
      "epoch": 1.198101043701172e-05,
      "step": 12563,
      "training_step_time": 17.110368013381958
    },
    {
      "epoch": 1.1981964111328125e-05,
      "model_forward_time": 5.998813152313232,
      "step": 12564
    },
    {
      "epoch": 1.1981964111328125e-05,
      "step": 12564,
      "training_step_time": 17.49568796157837
    },
    {
      "epoch": 1.198291778564453e-05,
      "model_forward_time": 5.621247291564941,
      "step": 12565
    },
    {
      "epoch": 1.198291778564453e-05,
      "step": 12565,
      "training_step_time": 17.09760546684265
    },
    {
      "epoch": 1.1983871459960938e-05,
      "model_forward_time": 5.594625949859619,
      "step": 12566
    },
    {
      "epoch": 1.1983871459960938e-05,
      "step": 12566,
      "training_step_time": 17.026030778884888
    },
    {
      "epoch": 1.1984825134277344e-05,
      "model_forward_time": 5.588656425476074,
      "step": 12567
    },
    {
      "epoch": 1.1984825134277344e-05,
      "step": 12567,
      "training_step_time": 17.024701595306396
    },
    {
      "epoch": 1.198577880859375e-05,
      "model_forward_time": 5.586152076721191,
      "step": 12568
    },
    {
      "epoch": 1.198577880859375e-05,
      "step": 12568,
      "training_step_time": 17.018173694610596
    },
    {
      "epoch": 1.1986732482910157e-05,
      "model_forward_time": 5.590311288833618,
      "step": 12569
    },
    {
      "epoch": 1.1986732482910157e-05,
      "step": 12569,
      "training_step_time": 17.026893854141235
    },
    {
      "epoch": 1.1987686157226563e-05,
      "grad_norm": 0.6186017990112305,
      "learning_rate": 1e-05,
      "loss": 0.0564,
      "step": 12570
    },
    {
      "epoch": 1.1987686157226563e-05,
      "model_forward_time": 6.014100551605225,
      "step": 12570
    },
    {
      "dynamics_loss_avg": 0.0519060829654336,
      "epoch": 1.1987686157226563e-05,
      "step": 12570
    },
    {
      "action_loss_avg": 0.006489088386297226,
      "epoch": 1.1987686157226563e-05,
      "step": 12570
    },
    {
      "epoch": 1.1987686157226563e-05,
      "step": 12570,
      "training_step_time": 17.43678593635559
    },
    {
      "epoch": 1.1988639831542968e-05,
      "model_forward_time": 5.5995330810546875,
      "step": 12571
    },
    {
      "epoch": 1.1988639831542968e-05,
      "step": 12571,
      "training_step_time": 17.05534338951111
    },
    {
      "epoch": 1.1989593505859374e-05,
      "model_forward_time": 5.6111249923706055,
      "step": 12572
    },
    {
      "epoch": 1.1989593505859374e-05,
      "step": 12572,
      "training_step_time": 17.09436058998108
    },
    {
      "epoch": 1.1990547180175782e-05,
      "model_forward_time": 5.615384340286255,
      "step": 12573
    },
    {
      "epoch": 1.1990547180175782e-05,
      "step": 12573,
      "training_step_time": 17.117499828338623
    },
    {
      "epoch": 1.1991500854492187e-05,
      "model_forward_time": 5.611637592315674,
      "step": 12574
    },
    {
      "epoch": 1.1991500854492187e-05,
      "step": 12574,
      "training_step_time": 17.121734619140625
    },
    {
      "epoch": 1.1992454528808593e-05,
      "model_forward_time": 6.006389617919922,
      "step": 12575
    },
    {
      "epoch": 1.1992454528808593e-05,
      "step": 12575,
      "training_step_time": 17.503224849700928
    },
    {
      "epoch": 1.1993408203125e-05,
      "model_forward_time": 5.613941431045532,
      "step": 12576
    },
    {
      "epoch": 1.1993408203125e-05,
      "step": 12576,
      "training_step_time": 17.114722728729248
    },
    {
      "epoch": 1.1994361877441406e-05,
      "model_forward_time": 5.611109733581543,
      "step": 12577
    },
    {
      "epoch": 1.1994361877441406e-05,
      "step": 12577,
      "training_step_time": 17.10810375213623
    },
    {
      "epoch": 1.1995315551757812e-05,
      "model_forward_time": 5.616780519485474,
      "step": 12578
    },
    {
      "epoch": 1.1995315551757812e-05,
      "step": 12578,
      "training_step_time": 17.094746828079224
    },
    {
      "epoch": 1.199626922607422e-05,
      "model_forward_time": 5.59549355506897,
      "step": 12579
    },
    {
      "epoch": 1.199626922607422e-05,
      "step": 12579,
      "training_step_time": 17.033029556274414
    },
    {
      "epoch": 1.1997222900390625e-05,
      "grad_norm": 1.4965555667877197,
      "learning_rate": 1e-05,
      "loss": 0.0516,
      "step": 12580
    },
    {
      "epoch": 1.1997222900390625e-05,
      "model_forward_time": 5.5915820598602295,
      "step": 12580
    },
    {
      "dynamics_loss_avg": 0.046415743231773374,
      "epoch": 1.1997222900390625e-05,
      "step": 12580
    },
    {
      "action_loss_avg": 0.005115582328289747,
      "epoch": 1.1997222900390625e-05,
      "step": 12580
    },
    {
      "epoch": 1.1997222900390625e-05,
      "step": 12580,
      "training_step_time": 17.015254735946655
    },
    {
      "epoch": 1.199817657470703e-05,
      "model_forward_time": 5.94840407371521,
      "step": 12581
    },
    {
      "epoch": 1.199817657470703e-05,
      "step": 12581,
      "training_step_time": 17.408494234085083
    },
    {
      "epoch": 1.1999130249023438e-05,
      "model_forward_time": 5.602409601211548,
      "step": 12582
    },
    {
      "epoch": 1.1999130249023438e-05,
      "step": 12582,
      "training_step_time": 17.034862995147705
    },
    {
      "epoch": 1.2000083923339844e-05,
      "model_forward_time": 5.598239421844482,
      "step": 12583
    },
    {
      "epoch": 1.2000083923339844e-05,
      "step": 12583,
      "training_step_time": 17.0924551486969
    },
    {
      "epoch": 1.200103759765625e-05,
      "model_forward_time": 5.6191933155059814,
      "step": 12584
    },
    {
      "epoch": 1.200103759765625e-05,
      "step": 12584,
      "training_step_time": 17.127893209457397
    },
    {
      "epoch": 1.2001991271972657e-05,
      "model_forward_time": 5.614582300186157,
      "step": 12585
    },
    {
      "epoch": 1.2001991271972657e-05,
      "step": 12585,
      "training_step_time": 17.095364332199097
    },
    {
      "epoch": 1.2002944946289063e-05,
      "model_forward_time": 5.983713150024414,
      "step": 12586
    },
    {
      "epoch": 1.2002944946289063e-05,
      "step": 12586,
      "training_step_time": 17.43628764152527
    },
    {
      "epoch": 1.2003898620605469e-05,
      "model_forward_time": 5.578921556472778,
      "step": 12587
    },
    {
      "epoch": 1.2003898620605469e-05,
      "step": 12587,
      "training_step_time": 17.004822492599487
    },
    {
      "epoch": 1.2004852294921874e-05,
      "model_forward_time": 5.584072828292847,
      "step": 12588
    },
    {
      "epoch": 1.2004852294921874e-05,
      "step": 12588,
      "training_step_time": 17.007347583770752
    },
    {
      "epoch": 1.2005805969238282e-05,
      "model_forward_time": 5.592225790023804,
      "step": 12589
    },
    {
      "epoch": 1.2005805969238282e-05,
      "step": 12589,
      "training_step_time": 17.00770092010498
    },
    {
      "epoch": 1.2006759643554687e-05,
      "grad_norm": 0.48794907331466675,
      "learning_rate": 1e-05,
      "loss": 0.06,
      "step": 12590
    },
    {
      "epoch": 1.2006759643554687e-05,
      "model_forward_time": 5.596616506576538,
      "step": 12590
    },
    {
      "dynamics_loss_avg": 0.05205981731414795,
      "epoch": 1.2006759643554687e-05,
      "step": 12590
    },
    {
      "action_loss_avg": 0.00661279559135437,
      "epoch": 1.2006759643554687e-05,
      "step": 12590
    },
    {
      "epoch": 1.2006759643554687e-05,
      "step": 12590,
      "training_step_time": 17.05412268638611
    },
    {
      "epoch": 1.2007713317871093e-05,
      "model_forward_time": 5.597339391708374,
      "step": 12591
    },
    {
      "epoch": 1.2007713317871093e-05,
      "step": 12591,
      "training_step_time": 17.04041314125061
    },
    {
      "epoch": 1.20086669921875e-05,
      "model_forward_time": 5.953725576400757,
      "step": 12592
    },
    {
      "epoch": 1.20086669921875e-05,
      "step": 12592,
      "training_step_time": 17.431295156478882
    },
    {
      "epoch": 1.2009620666503906e-05,
      "model_forward_time": 5.602583646774292,
      "step": 12593
    },
    {
      "epoch": 1.2009620666503906e-05,
      "step": 12593,
      "training_step_time": 17.08723521232605
    },
    {
      "epoch": 1.2010574340820312e-05,
      "model_forward_time": 5.610480308532715,
      "step": 12594
    },
    {
      "epoch": 1.2010574340820312e-05,
      "step": 12594,
      "training_step_time": 17.079962253570557
    },
    {
      "epoch": 1.201152801513672e-05,
      "model_forward_time": 5.600352764129639,
      "step": 12595
    },
    {
      "epoch": 1.201152801513672e-05,
      "step": 12595,
      "training_step_time": 17.03787875175476
    },
    {
      "epoch": 1.2012481689453125e-05,
      "model_forward_time": 5.59445595741272,
      "step": 12596
    },
    {
      "epoch": 1.2012481689453125e-05,
      "step": 12596,
      "training_step_time": 17.040587902069092
    },
    {
      "epoch": 1.201343536376953e-05,
      "model_forward_time": 5.971985101699829,
      "step": 12597
    },
    {
      "epoch": 1.201343536376953e-05,
      "step": 12597,
      "training_step_time": 17.396466493606567
    },
    {
      "epoch": 1.2014389038085938e-05,
      "model_forward_time": 5.606518745422363,
      "step": 12598
    },
    {
      "epoch": 1.2014389038085938e-05,
      "step": 12598,
      "training_step_time": 17.0783429145813
    },
    {
      "epoch": 1.2015342712402344e-05,
      "model_forward_time": 5.6202404499053955,
      "step": 12599
    },
    {
      "epoch": 1.2015342712402344e-05,
      "step": 12599,
      "training_step_time": 17.124052047729492
    },
    {
      "epoch": 1.201629638671875e-05,
      "grad_norm": 0.28542429208755493,
      "learning_rate": 1e-05,
      "loss": 0.0643,
      "step": 12600
    },
    {
      "epoch": 1.201629638671875e-05,
      "model_forward_time": 5.622540473937988,
      "step": 12600
    },
    {
      "dynamics_loss_avg": 0.06147201359272003,
      "epoch": 1.201629638671875e-05,
      "step": 12600
    },
    {
      "action_loss_avg": 0.008080730400979518,
      "epoch": 1.201629638671875e-05,
      "step": 12600
    },
    {
      "epoch": 1.201629638671875e-05,
      "step": 12600,
      "training_step_time": 17.10323405265808
    },
    {
      "epoch": 1.2017250061035157e-05,
      "model_forward_time": 5.599770545959473,
      "step": 12601
    },
    {
      "epoch": 1.2017250061035157e-05,
      "step": 12601,
      "training_step_time": 17.04332685470581
    },
    {
      "epoch": 1.2018203735351563e-05,
      "model_forward_time": 5.954850912094116,
      "step": 12602
    },
    {
      "epoch": 1.2018203735351563e-05,
      "step": 12602,
      "training_step_time": 17.388773441314697
    },
    {
      "epoch": 1.2019157409667969e-05,
      "model_forward_time": 5.589606761932373,
      "step": 12603
    },
    {
      "epoch": 1.2019157409667969e-05,
      "step": 12603,
      "training_step_time": 17.006648302078247
    },
    {
      "epoch": 1.2020111083984374e-05,
      "model_forward_time": 5.598253965377808,
      "step": 12604
    },
    {
      "epoch": 1.2020111083984374e-05,
      "step": 12604,
      "training_step_time": 17.048895359039307
    },
    {
      "epoch": 1.2021064758300782e-05,
      "model_forward_time": 5.595530033111572,
      "step": 12605
    },
    {
      "epoch": 1.2021064758300782e-05,
      "step": 12605,
      "training_step_time": 17.051944494247437
    },
    {
      "epoch": 1.2022018432617187e-05,
      "model_forward_time": 5.6043689250946045,
      "step": 12606
    },
    {
      "epoch": 1.2022018432617187e-05,
      "step": 12606,
      "training_step_time": 17.07551097869873
    },
    {
      "epoch": 1.2022972106933593e-05,
      "model_forward_time": 5.6058349609375,
      "step": 12607
    },
    {
      "epoch": 1.2022972106933593e-05,
      "step": 12607,
      "training_step_time": 17.094947338104248
    },
    {
      "epoch": 1.202392578125e-05,
      "model_forward_time": 6.00982928276062,
      "step": 12608
    },
    {
      "epoch": 1.202392578125e-05,
      "step": 12608,
      "training_step_time": 17.51027512550354
    },
    {
      "epoch": 1.2024879455566406e-05,
      "model_forward_time": 5.62428879737854,
      "step": 12609
    },
    {
      "epoch": 1.2024879455566406e-05,
      "step": 12609,
      "training_step_time": 17.06392240524292
    },
    {
      "epoch": 1.2025833129882812e-05,
      "grad_norm": 0.8012644648551941,
      "learning_rate": 1e-05,
      "loss": 0.0735,
      "step": 12610
    },
    {
      "epoch": 1.2025833129882812e-05,
      "model_forward_time": 5.610119581222534,
      "step": 12610
    },
    {
      "dynamics_loss_avg": 0.05986166149377823,
      "epoch": 1.2025833129882812e-05,
      "step": 12610
    },
    {
      "action_loss_avg": 0.0067119334009476,
      "epoch": 1.2025833129882812e-05,
      "step": 12610
    },
    {
      "epoch": 1.2025833129882812e-05,
      "step": 12610,
      "training_step_time": 17.104125261306763
    },
    {
      "epoch": 1.202678680419922e-05,
      "model_forward_time": 5.609377384185791,
      "step": 12611
    },
    {
      "epoch": 1.202678680419922e-05,
      "step": 12611,
      "training_step_time": 17.091233730316162
    },
    {
      "epoch": 1.2027740478515625e-05,
      "model_forward_time": 5.607331037521362,
      "step": 12612
    },
    {
      "epoch": 1.2027740478515625e-05,
      "step": 12612,
      "training_step_time": 17.090659141540527
    },
    {
      "epoch": 1.202869415283203e-05,
      "model_forward_time": 5.601433753967285,
      "step": 12613
    },
    {
      "epoch": 1.202869415283203e-05,
      "step": 12613,
      "training_step_time": 17.44362998008728
    },
    {
      "epoch": 1.2029647827148438e-05,
      "model_forward_time": 5.594788074493408,
      "step": 12614
    },
    {
      "epoch": 1.2029647827148438e-05,
      "step": 12614,
      "training_step_time": 17.049288988113403
    },
    {
      "epoch": 1.2030601501464844e-05,
      "model_forward_time": 5.586772441864014,
      "step": 12615
    },
    {
      "epoch": 1.2030601501464844e-05,
      "step": 12615,
      "training_step_time": 17.034924745559692
    },
    {
      "epoch": 1.203155517578125e-05,
      "model_forward_time": 5.602477550506592,
      "step": 12616
    },
    {
      "epoch": 1.203155517578125e-05,
      "step": 12616,
      "training_step_time": 17.08329963684082
    },
    {
      "epoch": 1.2032508850097657e-05,
      "model_forward_time": 5.617462635040283,
      "step": 12617
    },
    {
      "epoch": 1.2032508850097657e-05,
      "step": 12617,
      "training_step_time": 17.103365421295166
    },
    {
      "epoch": 1.2033462524414063e-05,
      "model_forward_time": 5.599613666534424,
      "step": 12618
    },
    {
      "epoch": 1.2033462524414063e-05,
      "step": 12618,
      "training_step_time": 17.075727701187134
    },
    {
      "epoch": 1.2034416198730469e-05,
      "model_forward_time": 5.966536998748779,
      "step": 12619
    },
    {
      "epoch": 1.2034416198730469e-05,
      "step": 12619,
      "training_step_time": 17.421035051345825
    },
    {
      "epoch": 1.2035369873046874e-05,
      "grad_norm": 0.34126174449920654,
      "learning_rate": 1e-05,
      "loss": 0.0627,
      "step": 12620
    },
    {
      "epoch": 1.2035369873046874e-05,
      "model_forward_time": 5.5906081199646,
      "step": 12620
    },
    {
      "dynamics_loss_avg": 0.062027806974947454,
      "epoch": 1.2035369873046874e-05,
      "step": 12620
    },
    {
      "action_loss_avg": 0.007292486680671573,
      "epoch": 1.2035369873046874e-05,
      "step": 12620
    },
    {
      "epoch": 1.2035369873046874e-05,
      "step": 12620,
      "training_step_time": 17.018798351287842
    },
    {
      "epoch": 1.2036323547363282e-05,
      "model_forward_time": 5.585554838180542,
      "step": 12621
    },
    {
      "epoch": 1.2036323547363282e-05,
      "step": 12621,
      "training_step_time": 17.020210027694702
    },
    {
      "epoch": 1.2037277221679687e-05,
      "model_forward_time": 5.589583158493042,
      "step": 12622
    },
    {
      "epoch": 1.2037277221679687e-05,
      "step": 12622,
      "training_step_time": 17.047050714492798
    },
    {
      "epoch": 1.2038230895996093e-05,
      "model_forward_time": 5.6030542850494385,
      "step": 12623
    },
    {
      "epoch": 1.2038230895996093e-05,
      "step": 12623,
      "training_step_time": 17.088703393936157
    },
    {
      "epoch": 1.20391845703125e-05,
      "model_forward_time": 5.60951566696167,
      "step": 12624
    },
    {
      "epoch": 1.20391845703125e-05,
      "step": 12624,
      "training_step_time": 17.44927740097046
    },
    {
      "epoch": 1.2040138244628906e-05,
      "model_forward_time": 5.605880975723267,
      "step": 12625
    },
    {
      "epoch": 1.2040138244628906e-05,
      "step": 12625,
      "training_step_time": 17.05973744392395
    },
    {
      "epoch": 1.2041091918945312e-05,
      "model_forward_time": 5.58697247505188,
      "step": 12626
    },
    {
      "epoch": 1.2041091918945312e-05,
      "step": 12626,
      "training_step_time": 17.01475238800049
    },
    {
      "epoch": 1.204204559326172e-05,
      "model_forward_time": 5.593602418899536,
      "step": 12627
    },
    {
      "epoch": 1.204204559326172e-05,
      "step": 12627,
      "training_step_time": 17.028920888900757
    },
    {
      "epoch": 1.2042999267578125e-05,
      "model_forward_time": 5.599992990493774,
      "step": 12628
    },
    {
      "epoch": 1.2042999267578125e-05,
      "step": 12628,
      "training_step_time": 17.056625366210938
    },
    {
      "epoch": 1.204395294189453e-05,
      "model_forward_time": 5.6030433177948,
      "step": 12629
    },
    {
      "epoch": 1.204395294189453e-05,
      "step": 12629,
      "training_step_time": 17.090479850769043
    },
    {
      "epoch": 1.2044906616210938e-05,
      "grad_norm": 0.5594682097434998,
      "learning_rate": 1e-05,
      "loss": 0.065,
      "step": 12630
    },
    {
      "epoch": 1.2044906616210938e-05,
      "model_forward_time": 5.989826440811157,
      "step": 12630
    },
    {
      "dynamics_loss_avg": 0.05414848756045103,
      "epoch": 1.2044906616210938e-05,
      "step": 12630
    },
    {
      "action_loss_avg": 0.007563531841151416,
      "epoch": 1.2044906616210938e-05,
      "step": 12630
    },
    {
      "epoch": 1.2044906616210938e-05,
      "step": 12630,
      "training_step_time": 17.486635208129883
    },
    {
      "epoch": 1.2045860290527344e-05,
      "model_forward_time": 5.632026433944702,
      "step": 12631
    },
    {
      "epoch": 1.2045860290527344e-05,
      "step": 12631,
      "training_step_time": 17.102265119552612
    },
    {
      "epoch": 1.204681396484375e-05,
      "model_forward_time": 5.611112117767334,
      "step": 12632
    },
    {
      "epoch": 1.204681396484375e-05,
      "step": 12632,
      "training_step_time": 17.088725090026855
    },
    {
      "epoch": 1.2047767639160155e-05,
      "model_forward_time": 5.611974000930786,
      "step": 12633
    },
    {
      "epoch": 1.2047767639160155e-05,
      "step": 12633,
      "training_step_time": 17.109034299850464
    },
    {
      "epoch": 1.2048721313476563e-05,
      "model_forward_time": 5.600656986236572,
      "step": 12634
    },
    {
      "epoch": 1.2048721313476563e-05,
      "step": 12634,
      "training_step_time": 17.083516359329224
    },
    {
      "epoch": 1.2049674987792969e-05,
      "model_forward_time": 5.9773781299591064,
      "step": 12635
    },
    {
      "epoch": 1.2049674987792969e-05,
      "step": 12635,
      "training_step_time": 17.430939435958862
    },
    {
      "epoch": 1.2050628662109374e-05,
      "model_forward_time": 5.599787950515747,
      "step": 12636
    },
    {
      "epoch": 1.2050628662109374e-05,
      "step": 12636,
      "training_step_time": 17.041160345077515
    },
    {
      "epoch": 1.2051582336425782e-05,
      "model_forward_time": 5.588976621627808,
      "step": 12637
    },
    {
      "epoch": 1.2051582336425782e-05,
      "step": 12637,
      "training_step_time": 17.06385612487793
    },
    {
      "epoch": 1.2052536010742187e-05,
      "model_forward_time": 5.60413122177124,
      "step": 12638
    },
    {
      "epoch": 1.2052536010742187e-05,
      "step": 12638,
      "training_step_time": 17.095858335494995
    },
    {
      "epoch": 1.2053489685058593e-05,
      "model_forward_time": 5.623856067657471,
      "step": 12639
    },
    {
      "epoch": 1.2053489685058593e-05,
      "step": 12639,
      "training_step_time": 17.143112421035767
    },
    {
      "epoch": 1.2054443359375e-05,
      "grad_norm": 1.0696251392364502,
      "learning_rate": 1e-05,
      "loss": 0.0792,
      "step": 12640
    },
    {
      "epoch": 1.2054443359375e-05,
      "model_forward_time": 5.620406866073608,
      "step": 12640
    },
    {
      "dynamics_loss_avg": 0.0749854702502489,
      "epoch": 1.2054443359375e-05,
      "step": 12640
    },
    {
      "action_loss_avg": 0.011961508775129914,
      "epoch": 1.2054443359375e-05,
      "step": 12640
    },
    {
      "epoch": 1.2054443359375e-05,
      "step": 12640,
      "training_step_time": 17.126357078552246
    },
    {
      "epoch": 1.2055397033691406e-05,
      "model_forward_time": 6.003243923187256,
      "step": 12641
    },
    {
      "epoch": 1.2055397033691406e-05,
      "step": 12641,
      "training_step_time": 17.491219997406006
    },
    {
      "epoch": 1.2056350708007812e-05,
      "model_forward_time": 5.602067232131958,
      "step": 12642
    },
    {
      "epoch": 1.2056350708007812e-05,
      "step": 12642,
      "training_step_time": 17.08781862258911
    },
    {
      "epoch": 1.205730438232422e-05,
      "model_forward_time": 5.608262062072754,
      "step": 12643
    },
    {
      "epoch": 1.205730438232422e-05,
      "step": 12643,
      "training_step_time": 17.07465958595276
    },
    {
      "epoch": 1.2058258056640625e-05,
      "model_forward_time": 5.606244087219238,
      "step": 12644
    },
    {
      "epoch": 1.2058258056640625e-05,
      "step": 12644,
      "training_step_time": 17.062084436416626
    },
    {
      "epoch": 1.205921173095703e-05,
      "model_forward_time": 5.5943896770477295,
      "step": 12645
    },
    {
      "epoch": 1.205921173095703e-05,
      "step": 12645,
      "training_step_time": 17.040419578552246
    },
    {
      "epoch": 1.2060165405273438e-05,
      "model_forward_time": 5.618152379989624,
      "step": 12646
    },
    {
      "epoch": 1.2060165405273438e-05,
      "step": 12646,
      "training_step_time": 17.446714878082275
    },
    {
      "epoch": 1.2061119079589844e-05,
      "model_forward_time": 5.587026357650757,
      "step": 12647
    },
    {
      "epoch": 1.2061119079589844e-05,
      "step": 12647,
      "training_step_time": 17.029552936553955
    },
    {
      "epoch": 1.206207275390625e-05,
      "model_forward_time": 5.611881494522095,
      "step": 12648
    },
    {
      "epoch": 1.206207275390625e-05,
      "step": 12648,
      "training_step_time": 17.09707999229431
    },
    {
      "epoch": 1.2063026428222655e-05,
      "model_forward_time": 5.621970891952515,
      "step": 12649
    },
    {
      "epoch": 1.2063026428222655e-05,
      "step": 12649,
      "training_step_time": 17.155859231948853
    },
    {
      "epoch": 1.2063980102539063e-05,
      "grad_norm": 0.42869535088539124,
      "learning_rate": 1e-05,
      "loss": 0.075,
      "step": 12650
    },
    {
      "epoch": 1.2063980102539063e-05,
      "model_forward_time": 5.619771957397461,
      "step": 12650
    },
    {
      "dynamics_loss_avg": 0.05983835849910975,
      "epoch": 1.2063980102539063e-05,
      "step": 12650
    },
    {
      "action_loss_avg": 0.010191214410588146,
      "epoch": 1.2063980102539063e-05,
      "step": 12650
    },
    {
      "epoch": 1.2063980102539063e-05,
      "step": 12650,
      "training_step_time": 17.112314701080322
    },
    {
      "epoch": 1.2064933776855469e-05,
      "model_forward_time": 5.615742921829224,
      "step": 12651
    },
    {
      "epoch": 1.2064933776855469e-05,
      "step": 12651,
      "training_step_time": 17.07908272743225
    },
    {
      "epoch": 1.2065887451171874e-05,
      "model_forward_time": 5.961003541946411,
      "step": 12652
    },
    {
      "epoch": 1.2065887451171874e-05,
      "step": 12652,
      "training_step_time": 17.415691614151
    },
    {
      "epoch": 1.2066841125488282e-05,
      "model_forward_time": 5.602968692779541,
      "step": 12653
    },
    {
      "epoch": 1.2066841125488282e-05,
      "step": 12653,
      "training_step_time": 17.045411586761475
    },
    {
      "epoch": 1.2067794799804687e-05,
      "model_forward_time": 5.586236476898193,
      "step": 12654
    },
    {
      "epoch": 1.2067794799804687e-05,
      "step": 12654,
      "training_step_time": 17.021544694900513
    },
    {
      "epoch": 1.2068748474121093e-05,
      "model_forward_time": 5.599207639694214,
      "step": 12655
    },
    {
      "epoch": 1.2068748474121093e-05,
      "step": 12655,
      "training_step_time": 17.061490058898926
    },
    {
      "epoch": 1.20697021484375e-05,
      "model_forward_time": 5.610538005828857,
      "step": 12656
    },
    {
      "epoch": 1.20697021484375e-05,
      "step": 12656,
      "training_step_time": 17.102155208587646
    },
    {
      "epoch": 1.2070655822753906e-05,
      "model_forward_time": 5.614861726760864,
      "step": 12657
    },
    {
      "epoch": 1.2070655822753906e-05,
      "step": 12657,
      "training_step_time": 17.459722757339478
    },
    {
      "epoch": 1.2071609497070312e-05,
      "model_forward_time": 5.60112738609314,
      "step": 12658
    },
    {
      "epoch": 1.2071609497070312e-05,
      "step": 12658,
      "training_step_time": 17.052752017974854
    },
    {
      "epoch": 1.207256317138672e-05,
      "model_forward_time": 5.598371744155884,
      "step": 12659
    },
    {
      "epoch": 1.207256317138672e-05,
      "step": 12659,
      "training_step_time": 17.062224626541138
    },
    {
      "epoch": 1.2073516845703125e-05,
      "grad_norm": 0.6535451412200928,
      "learning_rate": 1e-05,
      "loss": 0.0549,
      "step": 12660
    },
    {
      "epoch": 1.2073516845703125e-05,
      "model_forward_time": 5.606318473815918,
      "step": 12660
    },
    {
      "dynamics_loss_avg": 0.048007174767553805,
      "epoch": 1.2073516845703125e-05,
      "step": 12660
    },
    {
      "action_loss_avg": 0.006864119169767946,
      "epoch": 1.2073516845703125e-05,
      "step": 12660
    },
    {
      "epoch": 1.2073516845703125e-05,
      "step": 12660,
      "training_step_time": 17.08543610572815
    },
    {
      "epoch": 1.2074470520019531e-05,
      "model_forward_time": 5.615612983703613,
      "step": 12661
    },
    {
      "epoch": 1.2074470520019531e-05,
      "step": 12661,
      "training_step_time": 17.067238330841064
    },
    {
      "epoch": 1.2075424194335938e-05,
      "model_forward_time": 5.590624809265137,
      "step": 12662
    },
    {
      "epoch": 1.2075424194335938e-05,
      "step": 12662,
      "training_step_time": 17.0420823097229
    },
    {
      "epoch": 1.2076377868652344e-05,
      "model_forward_time": 5.970178127288818,
      "step": 12663
    },
    {
      "epoch": 1.2076377868652344e-05,
      "step": 12663,
      "training_step_time": 17.416404247283936
    },
    {
      "epoch": 1.207733154296875e-05,
      "model_forward_time": 5.610480546951294,
      "step": 12664
    },
    {
      "epoch": 1.207733154296875e-05,
      "step": 12664,
      "training_step_time": 17.128557205200195
    },
    {
      "epoch": 1.2078285217285155e-05,
      "model_forward_time": 5.625649690628052,
      "step": 12665
    },
    {
      "epoch": 1.2078285217285155e-05,
      "step": 12665,
      "training_step_time": 17.124649047851562
    },
    {
      "epoch": 1.2079238891601563e-05,
      "model_forward_time": 5.6097846031188965,
      "step": 12666
    },
    {
      "epoch": 1.2079238891601563e-05,
      "step": 12666,
      "training_step_time": 17.075833320617676
    },
    {
      "epoch": 1.2080192565917969e-05,
      "model_forward_time": 5.585154056549072,
      "step": 12667
    },
    {
      "epoch": 1.2080192565917969e-05,
      "step": 12667,
      "training_step_time": 17.024824857711792
    },
    {
      "epoch": 1.2081146240234374e-05,
      "model_forward_time": 5.93919038772583,
      "step": 12668
    },
    {
      "epoch": 1.2081146240234374e-05,
      "step": 12668,
      "training_step_time": 17.373466968536377
    },
    {
      "epoch": 1.2082099914550782e-05,
      "model_forward_time": 5.594291687011719,
      "step": 12669
    },
    {
      "epoch": 1.2082099914550782e-05,
      "step": 12669,
      "training_step_time": 17.018068075180054
    },
    {
      "epoch": 1.2083053588867187e-05,
      "grad_norm": 1.3340368270874023,
      "learning_rate": 1e-05,
      "loss": 0.0744,
      "step": 12670
    },
    {
      "epoch": 1.2083053588867187e-05,
      "model_forward_time": 5.600590705871582,
      "step": 12670
    },
    {
      "dynamics_loss_avg": 0.06287287902086973,
      "epoch": 1.2083053588867187e-05,
      "step": 12670
    },
    {
      "action_loss_avg": 0.010087376914452762,
      "epoch": 1.2083053588867187e-05,
      "step": 12670
    },
    {
      "epoch": 1.2083053588867187e-05,
      "step": 12670,
      "training_step_time": 17.070096731185913
    },
    {
      "epoch": 1.2084007263183593e-05,
      "model_forward_time": 5.610692739486694,
      "step": 12671
    },
    {
      "epoch": 1.2084007263183593e-05,
      "step": 12671,
      "training_step_time": 17.10749125480652
    },
    {
      "epoch": 1.20849609375e-05,
      "model_forward_time": 5.613109350204468,
      "step": 12672
    },
    {
      "epoch": 1.20849609375e-05,
      "step": 12672,
      "training_step_time": 17.095923900604248
    },
    {
      "epoch": 1.2085914611816406e-05,
      "model_forward_time": 5.597778081893921,
      "step": 12673
    },
    {
      "epoch": 1.2085914611816406e-05,
      "step": 12673,
      "training_step_time": 17.03481650352478
    },
    {
      "epoch": 1.2086868286132812e-05,
      "model_forward_time": 5.949129819869995,
      "step": 12674
    },
    {
      "epoch": 1.2086868286132812e-05,
      "step": 12674,
      "training_step_time": 17.376367807388306
    },
    {
      "epoch": 1.208782196044922e-05,
      "model_forward_time": 5.584250211715698,
      "step": 12675
    },
    {
      "epoch": 1.208782196044922e-05,
      "step": 12675,
      "training_step_time": 16.998002290725708
    },
    {
      "epoch": 1.2088775634765625e-05,
      "model_forward_time": 5.5864973068237305,
      "step": 12676
    },
    {
      "epoch": 1.2088775634765625e-05,
      "step": 12676,
      "training_step_time": 16.99984359741211
    },
    {
      "epoch": 1.2089729309082031e-05,
      "model_forward_time": 5.585367679595947,
      "step": 12677
    },
    {
      "epoch": 1.2089729309082031e-05,
      "step": 12677,
      "training_step_time": 17.018199682235718
    },
    {
      "epoch": 1.2090682983398438e-05,
      "model_forward_time": 5.574107646942139,
      "step": 12678
    },
    {
      "epoch": 1.2090682983398438e-05,
      "step": 12678,
      "training_step_time": 17.01745104789734
    },
    {
      "epoch": 1.2091636657714844e-05,
      "model_forward_time": 5.590080261230469,
      "step": 12679
    },
    {
      "epoch": 1.2091636657714844e-05,
      "step": 12679,
      "training_step_time": 17.417888641357422
    },
    {
      "epoch": 1.209259033203125e-05,
      "grad_norm": 0.7740684747695923,
      "learning_rate": 1e-05,
      "loss": 0.062,
      "step": 12680
    },
    {
      "epoch": 1.209259033203125e-05,
      "model_forward_time": 5.595070123672485,
      "step": 12680
    },
    {
      "dynamics_loss_avg": 0.05110404044389725,
      "epoch": 1.209259033203125e-05,
      "step": 12680
    },
    {
      "action_loss_avg": 0.010161321330815553,
      "epoch": 1.209259033203125e-05,
      "step": 12680
    },
    {
      "epoch": 1.209259033203125e-05,
      "step": 12680,
      "training_step_time": 17.074751615524292
    },
    {
      "epoch": 1.2093544006347655e-05,
      "model_forward_time": 5.605916261672974,
      "step": 12681
    },
    {
      "epoch": 1.2093544006347655e-05,
      "step": 12681,
      "training_step_time": 17.06331443786621
    },
    {
      "epoch": 1.2094497680664063e-05,
      "model_forward_time": 5.607712268829346,
      "step": 12682
    },
    {
      "epoch": 1.2094497680664063e-05,
      "step": 12682,
      "training_step_time": 17.057696104049683
    },
    {
      "epoch": 1.2095451354980469e-05,
      "model_forward_time": 5.590969085693359,
      "step": 12683
    },
    {
      "epoch": 1.2095451354980469e-05,
      "step": 12683,
      "training_step_time": 17.03577995300293
    },
    {
      "epoch": 1.2096405029296874e-05,
      "model_forward_time": 5.586958646774292,
      "step": 12684
    },
    {
      "epoch": 1.2096405029296874e-05,
      "step": 12684,
      "training_step_time": 17.015820741653442
    },
    {
      "epoch": 1.2097358703613282e-05,
      "model_forward_time": 5.955970764160156,
      "step": 12685
    },
    {
      "epoch": 1.2097358703613282e-05,
      "step": 12685,
      "training_step_time": 17.40850019454956
    },
    {
      "epoch": 1.2098312377929687e-05,
      "model_forward_time": 5.601281404495239,
      "step": 12686
    },
    {
      "epoch": 1.2098312377929687e-05,
      "step": 12686,
      "training_step_time": 17.095530033111572
    },
    {
      "epoch": 1.2099266052246093e-05,
      "model_forward_time": 5.620800256729126,
      "step": 12687
    },
    {
      "epoch": 1.2099266052246093e-05,
      "step": 12687,
      "training_step_time": 17.141724586486816
    },
    {
      "epoch": 1.21002197265625e-05,
      "model_forward_time": 5.624986886978149,
      "step": 12688
    },
    {
      "epoch": 1.21002197265625e-05,
      "step": 12688,
      "training_step_time": 17.115921020507812
    },
    {
      "epoch": 1.2101173400878906e-05,
      "model_forward_time": 5.601477861404419,
      "step": 12689
    },
    {
      "epoch": 1.2101173400878906e-05,
      "step": 12689,
      "training_step_time": 17.04876947402954
    },
    {
      "epoch": 1.2102127075195312e-05,
      "grad_norm": 0.6153872609138489,
      "learning_rate": 1e-05,
      "loss": 0.0739,
      "step": 12690
    },
    {
      "epoch": 1.2102127075195312e-05,
      "model_forward_time": 5.58270001411438,
      "step": 12690
    },
    {
      "dynamics_loss_avg": 0.06537835821509361,
      "epoch": 1.2102127075195312e-05,
      "step": 12690
    },
    {
      "action_loss_avg": 0.008870627777650953,
      "epoch": 1.2102127075195312e-05,
      "step": 12690
    },
    {
      "epoch": 1.2102127075195312e-05,
      "step": 12690,
      "training_step_time": 17.37798309326172
    },
    {
      "epoch": 1.210308074951172e-05,
      "model_forward_time": 5.587085723876953,
      "step": 12691
    },
    {
      "epoch": 1.210308074951172e-05,
      "step": 12691,
      "training_step_time": 16.996881246566772
    },
    {
      "epoch": 1.2104034423828125e-05,
      "model_forward_time": 5.587852239608765,
      "step": 12692
    },
    {
      "epoch": 1.2104034423828125e-05,
      "step": 12692,
      "training_step_time": 17.036346912384033
    },
    {
      "epoch": 1.2104988098144531e-05,
      "model_forward_time": 5.595848321914673,
      "step": 12693
    },
    {
      "epoch": 1.2104988098144531e-05,
      "step": 12693,
      "training_step_time": 17.070706605911255
    },
    {
      "epoch": 1.2105941772460938e-05,
      "model_forward_time": 5.604584217071533,
      "step": 12694
    },
    {
      "epoch": 1.2105941772460938e-05,
      "step": 12694,
      "training_step_time": 17.077582597732544
    },
    {
      "epoch": 1.2106895446777344e-05,
      "model_forward_time": 5.596937656402588,
      "step": 12695
    },
    {
      "epoch": 1.2106895446777344e-05,
      "step": 12695,
      "training_step_time": 17.07642126083374
    },
    {
      "epoch": 1.210784912109375e-05,
      "model_forward_time": 5.99501633644104,
      "step": 12696
    },
    {
      "epoch": 1.210784912109375e-05,
      "step": 12696,
      "training_step_time": 17.477325677871704
    },
    {
      "epoch": 1.2108802795410155e-05,
      "model_forward_time": 5.619192838668823,
      "step": 12697
    },
    {
      "epoch": 1.2108802795410155e-05,
      "step": 12697,
      "training_step_time": 17.100674867630005
    },
    {
      "epoch": 1.2109756469726563e-05,
      "model_forward_time": 5.611047267913818,
      "step": 12698
    },
    {
      "epoch": 1.2109756469726563e-05,
      "step": 12698,
      "training_step_time": 17.098111629486084
    },
    {
      "epoch": 1.2110710144042969e-05,
      "model_forward_time": 5.620561122894287,
      "step": 12699
    },
    {
      "epoch": 1.2110710144042969e-05,
      "step": 12699,
      "training_step_time": 17.109923601150513
    },
    {
      "epoch": 1.2111663818359374e-05,
      "grad_norm": 1.6318528652191162,
      "learning_rate": 1e-05,
      "loss": 0.0781,
      "step": 12700
    },
    {
      "epoch": 1.2111663818359374e-05,
      "model_forward_time": 5.611915826797485,
      "step": 12700
    },
    {
      "dynamics_loss_avg": 0.06915084160864353,
      "epoch": 1.2111663818359374e-05,
      "step": 12700
    },
    {
      "action_loss_avg": 0.012520182435400784,
      "epoch": 1.2111663818359374e-05,
      "step": 12700
    },
    {
      "epoch": 1.2111663818359374e-05,
      "step": 12700,
      "training_step_time": 17.108102321624756
    },
    {
      "epoch": 1.2112617492675782e-05,
      "model_forward_time": 5.961364984512329,
      "step": 12701
    },
    {
      "epoch": 1.2112617492675782e-05,
      "step": 12701,
      "training_step_time": 17.451739072799683
    },
    {
      "epoch": 1.2113571166992187e-05,
      "model_forward_time": 5.614729404449463,
      "step": 12702
    },
    {
      "epoch": 1.2113571166992187e-05,
      "step": 12702,
      "training_step_time": 17.134476900100708
    },
    {
      "epoch": 1.2114524841308593e-05,
      "model_forward_time": 5.6278557777404785,
      "step": 12703
    },
    {
      "epoch": 1.2114524841308593e-05,
      "step": 12703,
      "training_step_time": 17.13076162338257
    },
    {
      "epoch": 1.2115478515625e-05,
      "model_forward_time": 5.616342306137085,
      "step": 12704
    },
    {
      "epoch": 1.2115478515625e-05,
      "step": 12704,
      "training_step_time": 17.110831260681152
    },
    {
      "epoch": 1.2116432189941406e-05,
      "model_forward_time": 5.63791561126709,
      "step": 12705
    },
    {
      "epoch": 1.2116432189941406e-05,
      "step": 12705,
      "training_step_time": 17.136571884155273
    },
    {
      "epoch": 1.2117385864257812e-05,
      "model_forward_time": 5.622478008270264,
      "step": 12706
    },
    {
      "epoch": 1.2117385864257812e-05,
      "step": 12706,
      "training_step_time": 17.117533445358276
    },
    {
      "epoch": 1.211833953857422e-05,
      "model_forward_time": 5.986093044281006,
      "step": 12707
    },
    {
      "epoch": 1.211833953857422e-05,
      "step": 12707,
      "training_step_time": 17.45862102508545
    },
    {
      "epoch": 1.2119293212890625e-05,
      "model_forward_time": 5.595601558685303,
      "step": 12708
    },
    {
      "epoch": 1.2119293212890625e-05,
      "step": 12708,
      "training_step_time": 17.07955265045166
    },
    {
      "epoch": 1.2120246887207031e-05,
      "model_forward_time": 5.609194040298462,
      "step": 12709
    },
    {
      "epoch": 1.2120246887207031e-05,
      "step": 12709,
      "training_step_time": 17.104650497436523
    },
    {
      "epoch": 1.2121200561523438e-05,
      "grad_norm": 0.8061568140983582,
      "learning_rate": 1e-05,
      "loss": 0.0666,
      "step": 12710
    },
    {
      "epoch": 1.2121200561523438e-05,
      "model_forward_time": 5.615241527557373,
      "step": 12710
    },
    {
      "dynamics_loss_avg": 0.05638068243861198,
      "epoch": 1.2121200561523438e-05,
      "step": 12710
    },
    {
      "action_loss_avg": 0.006854716350790113,
      "epoch": 1.2121200561523438e-05,
      "step": 12710
    },
    {
      "epoch": 1.2121200561523438e-05,
      "step": 12710,
      "training_step_time": 17.08394432067871
    },
    {
      "epoch": 1.2122154235839844e-05,
      "model_forward_time": 5.595585107803345,
      "step": 12711
    },
    {
      "epoch": 1.2122154235839844e-05,
      "step": 12711,
      "training_step_time": 17.03088665008545
    },
    {
      "epoch": 1.212310791015625e-05,
      "model_forward_time": 5.577648401260376,
      "step": 12712
    },
    {
      "epoch": 1.212310791015625e-05,
      "step": 12712,
      "training_step_time": 17.399973154067993
    },
    {
      "epoch": 1.2124061584472655e-05,
      "model_forward_time": 5.605341911315918,
      "step": 12713
    },
    {
      "epoch": 1.2124061584472655e-05,
      "step": 12713,
      "training_step_time": 17.105828523635864
    },
    {
      "epoch": 1.2125015258789063e-05,
      "model_forward_time": 5.616213798522949,
      "step": 12714
    },
    {
      "epoch": 1.2125015258789063e-05,
      "step": 12714,
      "training_step_time": 17.129308700561523
    },
    {
      "epoch": 1.2125968933105469e-05,
      "model_forward_time": 5.616146564483643,
      "step": 12715
    },
    {
      "epoch": 1.2125968933105469e-05,
      "step": 12715,
      "training_step_time": 17.126779317855835
    },
    {
      "epoch": 1.2126922607421874e-05,
      "model_forward_time": 5.617700815200806,
      "step": 12716
    },
    {
      "epoch": 1.2126922607421874e-05,
      "step": 12716,
      "training_step_time": 17.114232063293457
    },
    {
      "epoch": 1.2127876281738282e-05,
      "model_forward_time": 5.604083299636841,
      "step": 12717
    },
    {
      "epoch": 1.2127876281738282e-05,
      "step": 12717,
      "training_step_time": 17.09103488922119
    },
    {
      "epoch": 1.2128829956054687e-05,
      "model_forward_time": 5.966628789901733,
      "step": 12718
    },
    {
      "epoch": 1.2128829956054687e-05,
      "step": 12718,
      "training_step_time": 17.447015047073364
    },
    {
      "epoch": 1.2129783630371093e-05,
      "model_forward_time": 5.608024835586548,
      "step": 12719
    },
    {
      "epoch": 1.2129783630371093e-05,
      "step": 12719,
      "training_step_time": 17.05763864517212
    },
    {
      "epoch": 1.21307373046875e-05,
      "grad_norm": 0.4896514117717743,
      "learning_rate": 1e-05,
      "loss": 0.0622,
      "step": 12720
    },
    {
      "epoch": 1.21307373046875e-05,
      "model_forward_time": 5.589717626571655,
      "step": 12720
    },
    {
      "dynamics_loss_avg": 0.05192223619669676,
      "epoch": 1.21307373046875e-05,
      "step": 12720
    },
    {
      "action_loss_avg": 0.005705952737480402,
      "epoch": 1.21307373046875e-05,
      "step": 12720
    },
    {
      "epoch": 1.21307373046875e-05,
      "step": 12720,
      "training_step_time": 17.043182134628296
    },
    {
      "epoch": 1.2131690979003906e-05,
      "model_forward_time": 5.606581926345825,
      "step": 12721
    },
    {
      "epoch": 1.2131690979003906e-05,
      "step": 12721,
      "training_step_time": 17.094660997390747
    },
    {
      "epoch": 1.2132644653320312e-05,
      "model_forward_time": 5.606506109237671,
      "step": 12722
    },
    {
      "epoch": 1.2132644653320312e-05,
      "step": 12722,
      "training_step_time": 17.09091854095459
    },
    {
      "epoch": 1.213359832763672e-05,
      "model_forward_time": 5.594886779785156,
      "step": 12723
    },
    {
      "epoch": 1.213359832763672e-05,
      "step": 12723,
      "training_step_time": 17.41281294822693
    },
    {
      "epoch": 1.2134552001953125e-05,
      "model_forward_time": 5.599026441574097,
      "step": 12724
    },
    {
      "epoch": 1.2134552001953125e-05,
      "step": 12724,
      "training_step_time": 17.078044652938843
    },
    {
      "epoch": 1.2135505676269531e-05,
      "model_forward_time": 5.610456705093384,
      "step": 12725
    },
    {
      "epoch": 1.2135505676269531e-05,
      "step": 12725,
      "training_step_time": 17.097975492477417
    },
    {
      "epoch": 1.2136459350585938e-05,
      "model_forward_time": 5.605987071990967,
      "step": 12726
    },
    {
      "epoch": 1.2136459350585938e-05,
      "step": 12726,
      "training_step_time": 17.074317455291748
    },
    {
      "epoch": 1.2137413024902344e-05,
      "model_forward_time": 5.598394393920898,
      "step": 12727
    },
    {
      "epoch": 1.2137413024902344e-05,
      "step": 12727,
      "training_step_time": 17.063964366912842
    },
    {
      "epoch": 1.213836669921875e-05,
      "model_forward_time": 5.607604503631592,
      "step": 12728
    },
    {
      "epoch": 1.213836669921875e-05,
      "step": 12728,
      "training_step_time": 17.08847737312317
    },
    {
      "epoch": 1.2139320373535155e-05,
      "model_forward_time": 5.983020067214966,
      "step": 12729
    },
    {
      "epoch": 1.2139320373535155e-05,
      "step": 12729,
      "training_step_time": 17.443735122680664
    },
    {
      "epoch": 1.2140274047851563e-05,
      "grad_norm": 1.2834460735321045,
      "learning_rate": 1e-05,
      "loss": 0.0758,
      "step": 12730
    },
    {
      "epoch": 1.2140274047851563e-05,
      "model_forward_time": 5.600304365158081,
      "step": 12730
    },
    {
      "dynamics_loss_avg": 0.06914523970335722,
      "epoch": 1.2140274047851563e-05,
      "step": 12730
    },
    {
      "action_loss_avg": 0.008381419873330742,
      "epoch": 1.2140274047851563e-05,
      "step": 12730
    },
    {
      "epoch": 1.2140274047851563e-05,
      "step": 12730,
      "training_step_time": 17.032776594161987
    },
    {
      "epoch": 1.2141227722167969e-05,
      "model_forward_time": 5.586121320724487,
      "step": 12731
    },
    {
      "epoch": 1.2141227722167969e-05,
      "step": 12731,
      "training_step_time": 17.03450345993042
    },
    {
      "epoch": 1.2142181396484374e-05,
      "model_forward_time": 5.591145038604736,
      "step": 12732
    },
    {
      "epoch": 1.2142181396484374e-05,
      "step": 12732,
      "training_step_time": 17.06468439102173
    },
    {
      "epoch": 1.2143135070800782e-05,
      "model_forward_time": 5.607902765274048,
      "step": 12733
    },
    {
      "epoch": 1.2143135070800782e-05,
      "step": 12733,
      "training_step_time": 17.099568367004395
    },
    {
      "epoch": 1.2144088745117187e-05,
      "model_forward_time": 5.97286581993103,
      "step": 12734
    },
    {
      "epoch": 1.2144088745117187e-05,
      "step": 12734,
      "training_step_time": 17.463090181350708
    },
    {
      "epoch": 1.2145042419433593e-05,
      "model_forward_time": 5.609338283538818,
      "step": 12735
    },
    {
      "epoch": 1.2145042419433593e-05,
      "step": 12735,
      "training_step_time": 17.075273513793945
    },
    {
      "epoch": 1.214599609375e-05,
      "model_forward_time": 5.6059253215789795,
      "step": 12736
    },
    {
      "epoch": 1.214599609375e-05,
      "step": 12736,
      "training_step_time": 17.059842824935913
    },
    {
      "epoch": 1.2146949768066406e-05,
      "model_forward_time": 5.598330974578857,
      "step": 12737
    },
    {
      "epoch": 1.2146949768066406e-05,
      "step": 12737,
      "training_step_time": 17.05346918106079
    },
    {
      "epoch": 1.2147903442382812e-05,
      "model_forward_time": 5.608251571655273,
      "step": 12738
    },
    {
      "epoch": 1.2147903442382812e-05,
      "step": 12738,
      "training_step_time": 17.075012922286987
    },
    {
      "epoch": 1.214885711669922e-05,
      "model_forward_time": 5.604703664779663,
      "step": 12739
    },
    {
      "epoch": 1.214885711669922e-05,
      "step": 12739,
      "training_step_time": 17.07993984222412
    },
    {
      "epoch": 1.2149810791015625e-05,
      "grad_norm": 1.0959621667861938,
      "learning_rate": 1e-05,
      "loss": 0.0703,
      "step": 12740
    },
    {
      "epoch": 1.2149810791015625e-05,
      "model_forward_time": 5.978435277938843,
      "step": 12740
    },
    {
      "dynamics_loss_avg": 0.06367783360183239,
      "epoch": 1.2149810791015625e-05,
      "step": 12740
    },
    {
      "action_loss_avg": 0.008736829296685756,
      "epoch": 1.2149810791015625e-05,
      "step": 12740
    },
    {
      "epoch": 1.2149810791015625e-05,
      "step": 12740,
      "training_step_time": 17.461698055267334
    },
    {
      "epoch": 1.2150764465332031e-05,
      "model_forward_time": 5.609994173049927,
      "step": 12741
    },
    {
      "epoch": 1.2150764465332031e-05,
      "step": 12741,
      "training_step_time": 17.093746662139893
    },
    {
      "epoch": 1.2151718139648438e-05,
      "model_forward_time": 5.612680912017822,
      "step": 12742
    },
    {
      "epoch": 1.2151718139648438e-05,
      "step": 12742,
      "training_step_time": 17.109846591949463
    },
    {
      "epoch": 1.2152671813964844e-05,
      "model_forward_time": 5.611455678939819,
      "step": 12743
    },
    {
      "epoch": 1.2152671813964844e-05,
      "step": 12743,
      "training_step_time": 17.080448150634766
    },
    {
      "epoch": 1.215362548828125e-05,
      "model_forward_time": 5.597715616226196,
      "step": 12744
    },
    {
      "epoch": 1.215362548828125e-05,
      "step": 12744,
      "training_step_time": 17.056646823883057
    },
    {
      "epoch": 1.2154579162597656e-05,
      "model_forward_time": 5.59010648727417,
      "step": 12745
    },
    {
      "epoch": 1.2154579162597656e-05,
      "step": 12745,
      "training_step_time": 17.389272212982178
    },
    {
      "epoch": 1.2155532836914063e-05,
      "model_forward_time": 5.578634977340698,
      "step": 12746
    },
    {
      "epoch": 1.2155532836914063e-05,
      "step": 12746,
      "training_step_time": 17.024715423583984
    },
    {
      "epoch": 1.2156486511230469e-05,
      "model_forward_time": 5.5910325050354,
      "step": 12747
    },
    {
      "epoch": 1.2156486511230469e-05,
      "step": 12747,
      "training_step_time": 17.05223774909973
    },
    {
      "epoch": 1.2157440185546874e-05,
      "model_forward_time": 5.599548578262329,
      "step": 12748
    },
    {
      "epoch": 1.2157440185546874e-05,
      "step": 12748,
      "training_step_time": 17.05790114402771
    },
    {
      "epoch": 1.2158393859863282e-05,
      "model_forward_time": 5.602424383163452,
      "step": 12749
    },
    {
      "epoch": 1.2158393859863282e-05,
      "step": 12749,
      "training_step_time": 17.059980630874634
    },
    {
      "epoch": 1.2159347534179688e-05,
      "grad_norm": 0.41265028715133667,
      "learning_rate": 1e-05,
      "loss": 0.0678,
      "step": 12750
    },
    {
      "epoch": 1.2159347534179688e-05,
      "model_forward_time": 5.588066816329956,
      "step": 12750
    },
    {
      "dynamics_loss_avg": 0.06878991797566414,
      "epoch": 1.2159347534179688e-05,
      "step": 12750
    },
    {
      "action_loss_avg": 0.008377359947189689,
      "epoch": 1.2159347534179688e-05,
      "step": 12750
    },
    {
      "epoch": 1.2159347534179688e-05,
      "step": 12750,
      "training_step_time": 17.035003662109375
    },
    {
      "epoch": 1.2160301208496093e-05,
      "model_forward_time": 5.947843313217163,
      "step": 12751
    },
    {
      "epoch": 1.2160301208496093e-05,
      "step": 12751,
      "training_step_time": 17.3888156414032
    },
    {
      "epoch": 1.21612548828125e-05,
      "model_forward_time": 5.602869987487793,
      "step": 12752
    },
    {
      "epoch": 1.21612548828125e-05,
      "step": 12752,
      "training_step_time": 17.101911544799805
    },
    {
      "epoch": 1.2162208557128906e-05,
      "model_forward_time": 5.621635675430298,
      "step": 12753
    },
    {
      "epoch": 1.2162208557128906e-05,
      "step": 12753,
      "training_step_time": 17.12094807624817
    },
    {
      "epoch": 1.2163162231445312e-05,
      "model_forward_time": 5.615228891372681,
      "step": 12754
    },
    {
      "epoch": 1.2163162231445312e-05,
      "step": 12754,
      "training_step_time": 17.110944032669067
    },
    {
      "epoch": 1.216411590576172e-05,
      "model_forward_time": 5.605578422546387,
      "step": 12755
    },
    {
      "epoch": 1.216411590576172e-05,
      "step": 12755,
      "training_step_time": 17.087586641311646
    },
    {
      "epoch": 1.2165069580078125e-05,
      "model_forward_time": 5.6001410484313965,
      "step": 12756
    },
    {
      "epoch": 1.2165069580078125e-05,
      "step": 12756,
      "training_step_time": 17.4379620552063
    },
    {
      "epoch": 1.2166023254394531e-05,
      "model_forward_time": 5.598034143447876,
      "step": 12757
    },
    {
      "epoch": 1.2166023254394531e-05,
      "step": 12757,
      "training_step_time": 17.040672540664673
    },
    {
      "epoch": 1.2166976928710937e-05,
      "model_forward_time": 5.590494394302368,
      "step": 12758
    },
    {
      "epoch": 1.2166976928710937e-05,
      "step": 12758,
      "training_step_time": 17.024434089660645
    },
    {
      "epoch": 1.2167930603027344e-05,
      "model_forward_time": 5.595354318618774,
      "step": 12759
    },
    {
      "epoch": 1.2167930603027344e-05,
      "step": 12759,
      "training_step_time": 17.06984782218933
    },
    {
      "epoch": 1.216888427734375e-05,
      "grad_norm": 0.8578259348869324,
      "learning_rate": 1e-05,
      "loss": 0.0795,
      "step": 12760
    },
    {
      "epoch": 1.216888427734375e-05,
      "model_forward_time": 5.6165611743927,
      "step": 12760
    },
    {
      "dynamics_loss_avg": 0.05935315955430269,
      "epoch": 1.216888427734375e-05,
      "step": 12760
    },
    {
      "action_loss_avg": 0.007630811375565827,
      "epoch": 1.216888427734375e-05,
      "step": 12760
    },
    {
      "epoch": 1.216888427734375e-05,
      "step": 12760,
      "training_step_time": 17.12025547027588
    },
    {
      "epoch": 1.2169837951660156e-05,
      "model_forward_time": 5.613349199295044,
      "step": 12761
    },
    {
      "epoch": 1.2169837951660156e-05,
      "step": 12761,
      "training_step_time": 17.14084792137146
    },
    {
      "epoch": 1.2170791625976563e-05,
      "model_forward_time": 5.984266519546509,
      "step": 12762
    },
    {
      "epoch": 1.2170791625976563e-05,
      "step": 12762,
      "training_step_time": 17.48856258392334
    },
    {
      "epoch": 1.2171745300292969e-05,
      "model_forward_time": 5.618864297866821,
      "step": 12763
    },
    {
      "epoch": 1.2171745300292969e-05,
      "step": 12763,
      "training_step_time": 17.104992389678955
    },
    {
      "epoch": 1.2172698974609374e-05,
      "model_forward_time": 5.609759330749512,
      "step": 12764
    },
    {
      "epoch": 1.2172698974609374e-05,
      "step": 12764,
      "training_step_time": 17.071617126464844
    },
    {
      "epoch": 1.2173652648925782e-05,
      "model_forward_time": 5.602340936660767,
      "step": 12765
    },
    {
      "epoch": 1.2173652648925782e-05,
      "step": 12765,
      "training_step_time": 17.073405742645264
    },
    {
      "epoch": 1.2174606323242188e-05,
      "model_forward_time": 5.593816518783569,
      "step": 12766
    },
    {
      "epoch": 1.2174606323242188e-05,
      "step": 12766,
      "training_step_time": 17.04388666152954
    },
    {
      "epoch": 1.2175559997558593e-05,
      "model_forward_time": 5.95997428894043,
      "step": 12767
    },
    {
      "epoch": 1.2175559997558593e-05,
      "step": 12767,
      "training_step_time": 17.42558479309082
    },
    {
      "epoch": 1.2176513671875e-05,
      "model_forward_time": 5.599122524261475,
      "step": 12768
    },
    {
      "epoch": 1.2176513671875e-05,
      "step": 12768,
      "training_step_time": 17.056589365005493
    },
    {
      "epoch": 1.2177467346191406e-05,
      "model_forward_time": 5.593697547912598,
      "step": 12769
    },
    {
      "epoch": 1.2177467346191406e-05,
      "step": 12769,
      "training_step_time": 17.05691933631897
    },
    {
      "epoch": 1.2178421020507812e-05,
      "grad_norm": 0.6923879981040955,
      "learning_rate": 1e-05,
      "loss": 0.0571,
      "step": 12770
    },
    {
      "epoch": 1.2178421020507812e-05,
      "model_forward_time": 5.597267150878906,
      "step": 12770
    },
    {
      "dynamics_loss_avg": 0.05229638908058405,
      "epoch": 1.2178421020507812e-05,
      "step": 12770
    },
    {
      "action_loss_avg": 0.0064500410691834984,
      "epoch": 1.2178421020507812e-05,
      "step": 12770
    },
    {
      "epoch": 1.2178421020507812e-05,
      "step": 12770,
      "training_step_time": 17.05727505683899
    },
    {
      "epoch": 1.217937469482422e-05,
      "model_forward_time": 5.598755121231079,
      "step": 12771
    },
    {
      "epoch": 1.217937469482422e-05,
      "step": 12771,
      "training_step_time": 17.060163736343384
    },
    {
      "epoch": 1.2180328369140625e-05,
      "model_forward_time": 5.594797134399414,
      "step": 12772
    },
    {
      "epoch": 1.2180328369140625e-05,
      "step": 12772,
      "training_step_time": 17.065351724624634
    },
    {
      "epoch": 1.2181282043457031e-05,
      "model_forward_time": 6.005379915237427,
      "step": 12773
    },
    {
      "epoch": 1.2181282043457031e-05,
      "step": 12773,
      "training_step_time": 17.47610092163086
    },
    {
      "epoch": 1.2182235717773437e-05,
      "model_forward_time": 5.588621377944946,
      "step": 12774
    },
    {
      "epoch": 1.2182235717773437e-05,
      "step": 12774,
      "training_step_time": 17.0187349319458
    },
    {
      "epoch": 1.2183189392089844e-05,
      "model_forward_time": 5.584250211715698,
      "step": 12775
    },
    {
      "epoch": 1.2183189392089844e-05,
      "step": 12775,
      "training_step_time": 17.02832055091858
    },
    {
      "epoch": 1.218414306640625e-05,
      "model_forward_time": 5.597444295883179,
      "step": 12776
    },
    {
      "epoch": 1.218414306640625e-05,
      "step": 12776,
      "training_step_time": 17.058369398117065
    },
    {
      "epoch": 1.2185096740722656e-05,
      "model_forward_time": 5.602913856506348,
      "step": 12777
    },
    {
      "epoch": 1.2185096740722656e-05,
      "step": 12777,
      "training_step_time": 17.06466007232666
    },
    {
      "epoch": 1.2186050415039063e-05,
      "model_forward_time": 5.599146127700806,
      "step": 12778
    },
    {
      "epoch": 1.2186050415039063e-05,
      "step": 12778,
      "training_step_time": 17.43183660507202
    },
    {
      "epoch": 1.2187004089355469e-05,
      "model_forward_time": 5.595570087432861,
      "step": 12779
    },
    {
      "epoch": 1.2187004089355469e-05,
      "step": 12779,
      "training_step_time": 17.05143451690674
    },
    {
      "epoch": 1.2187957763671874e-05,
      "grad_norm": 0.5224649310112,
      "learning_rate": 1e-05,
      "loss": 0.054,
      "step": 12780
    },
    {
      "epoch": 1.2187957763671874e-05,
      "model_forward_time": 5.598072052001953,
      "step": 12780
    },
    {
      "dynamics_loss_avg": 0.050556722283363345,
      "epoch": 1.2187957763671874e-05,
      "step": 12780
    },
    {
      "action_loss_avg": 0.00763948995154351,
      "epoch": 1.2187957763671874e-05,
      "step": 12780
    },
    {
      "epoch": 1.2187957763671874e-05,
      "step": 12780,
      "training_step_time": 17.0343234539032
    },
    {
      "epoch": 1.2188911437988282e-05,
      "model_forward_time": 5.5989298820495605,
      "step": 12781
    },
    {
      "epoch": 1.2188911437988282e-05,
      "step": 12781,
      "training_step_time": 17.046981811523438
    },
    {
      "epoch": 1.2189865112304688e-05,
      "model_forward_time": 5.592785835266113,
      "step": 12782
    },
    {
      "epoch": 1.2189865112304688e-05,
      "step": 12782,
      "training_step_time": 17.06614089012146
    },
    {
      "epoch": 1.2190818786621093e-05,
      "model_forward_time": 5.604885101318359,
      "step": 12783
    },
    {
      "epoch": 1.2190818786621093e-05,
      "step": 12783,
      "training_step_time": 17.08481764793396
    },
    {
      "epoch": 1.21917724609375e-05,
      "model_forward_time": 5.972558259963989,
      "step": 12784
    },
    {
      "epoch": 1.21917724609375e-05,
      "step": 12784,
      "training_step_time": 17.441096544265747
    },
    {
      "epoch": 1.2192726135253906e-05,
      "model_forward_time": 5.596844673156738,
      "step": 12785
    },
    {
      "epoch": 1.2192726135253906e-05,
      "step": 12785,
      "training_step_time": 17.05072855949402
    },
    {
      "epoch": 1.2193679809570312e-05,
      "model_forward_time": 5.5942864418029785,
      "step": 12786
    },
    {
      "epoch": 1.2193679809570312e-05,
      "step": 12786,
      "training_step_time": 17.054293632507324
    },
    {
      "epoch": 1.219463348388672e-05,
      "model_forward_time": 5.61171817779541,
      "step": 12787
    },
    {
      "epoch": 1.219463348388672e-05,
      "step": 12787,
      "training_step_time": 17.063531398773193
    },
    {
      "epoch": 1.2195587158203125e-05,
      "model_forward_time": 5.587408065795898,
      "step": 12788
    },
    {
      "epoch": 1.2195587158203125e-05,
      "step": 12788,
      "training_step_time": 17.036340713500977
    },
    {
      "epoch": 1.2196540832519531e-05,
      "model_forward_time": 5.614840984344482,
      "step": 12789
    },
    {
      "epoch": 1.2196540832519531e-05,
      "step": 12789,
      "training_step_time": 17.472203969955444
    },
    {
      "epoch": 1.2197494506835937e-05,
      "grad_norm": 0.9222002625465393,
      "learning_rate": 1e-05,
      "loss": 0.0594,
      "step": 12790
    },
    {
      "epoch": 1.2197494506835937e-05,
      "model_forward_time": 5.610052585601807,
      "step": 12790
    },
    {
      "dynamics_loss_avg": 0.048534626886248586,
      "epoch": 1.2197494506835937e-05,
      "step": 12790
    },
    {
      "action_loss_avg": 0.006618091370910406,
      "epoch": 1.2197494506835937e-05,
      "step": 12790
    },
    {
      "epoch": 1.2197494506835937e-05,
      "step": 12790,
      "training_step_time": 17.097668170928955
    },
    {
      "epoch": 1.2198448181152344e-05,
      "model_forward_time": 5.589369297027588,
      "step": 12791
    },
    {
      "epoch": 1.2198448181152344e-05,
      "step": 12791,
      "training_step_time": 17.045899868011475
    },
    {
      "epoch": 1.219940185546875e-05,
      "model_forward_time": 5.583894729614258,
      "step": 12792
    },
    {
      "epoch": 1.219940185546875e-05,
      "step": 12792,
      "training_step_time": 17.021353244781494
    },
    {
      "epoch": 1.2200355529785156e-05,
      "model_forward_time": 5.5799736976623535,
      "step": 12793
    },
    {
      "epoch": 1.2200355529785156e-05,
      "step": 12793,
      "training_step_time": 17.027810096740723
    },
    {
      "epoch": 1.2201309204101563e-05,
      "model_forward_time": 5.6053385734558105,
      "step": 12794
    },
    {
      "epoch": 1.2201309204101563e-05,
      "step": 12794,
      "training_step_time": 17.0896999835968
    },
    {
      "epoch": 1.2202262878417969e-05,
      "model_forward_time": 5.978201389312744,
      "step": 12795
    },
    {
      "epoch": 1.2202262878417969e-05,
      "step": 12795,
      "training_step_time": 17.4770770072937
    },
    {
      "epoch": 1.2203216552734374e-05,
      "model_forward_time": 5.6115193367004395,
      "step": 12796
    },
    {
      "epoch": 1.2203216552734374e-05,
      "step": 12796,
      "training_step_time": 17.083999633789062
    },
    {
      "epoch": 1.2204170227050782e-05,
      "model_forward_time": 5.59453821182251,
      "step": 12797
    },
    {
      "epoch": 1.2204170227050782e-05,
      "step": 12797,
      "training_step_time": 17.050326585769653
    },
    {
      "epoch": 1.2205123901367188e-05,
      "model_forward_time": 5.5868823528289795,
      "step": 12798
    },
    {
      "epoch": 1.2205123901367188e-05,
      "step": 12798,
      "training_step_time": 17.025741577148438
    },
    {
      "epoch": 1.2206077575683593e-05,
      "model_forward_time": 5.590763330459595,
      "step": 12799
    },
    {
      "epoch": 1.2206077575683593e-05,
      "step": 12799,
      "training_step_time": 17.032343864440918
    },
    {
      "epoch": 1.220703125e-05,
      "grad_norm": 0.3891640901565552,
      "learning_rate": 1e-05,
      "loss": 0.0706,
      "step": 12800
    },
    {
      "epoch": 1.220703125e-05,
      "model_forward_time": 5.969141721725464,
      "step": 12800
    },
    {
      "dynamics_loss_avg": 0.0648765480145812,
      "epoch": 1.220703125e-05,
      "step": 12800
    },
    {
      "action_loss_avg": 0.008937748079188169,
      "epoch": 1.220703125e-05,
      "step": 12800
    },
    {
      "epoch": 1.220703125e-05,
      "step": 12800,
      "training_step_time": 17.41893458366394
    },
    {
      "epoch": 1.2207984924316406e-05,
      "model_forward_time": 5.6001200675964355,
      "step": 12801
    },
    {
      "epoch": 1.2207984924316406e-05,
      "step": 12801,
      "training_step_time": 17.05561375617981
    },
    {
      "epoch": 1.2208938598632812e-05,
      "model_forward_time": 5.60436749458313,
      "step": 12802
    },
    {
      "epoch": 1.2208938598632812e-05,
      "step": 12802,
      "training_step_time": 17.07182025909424
    },
    {
      "epoch": 1.220989227294922e-05,
      "model_forward_time": 5.6033172607421875,
      "step": 12803
    },
    {
      "epoch": 1.220989227294922e-05,
      "step": 12803,
      "training_step_time": 17.068480968475342
    },
    {
      "epoch": 1.2210845947265625e-05,
      "model_forward_time": 5.59312629699707,
      "step": 12804
    },
    {
      "epoch": 1.2210845947265625e-05,
      "step": 12804,
      "training_step_time": 17.057660579681396
    },
    {
      "epoch": 1.2211799621582031e-05,
      "model_forward_time": 5.600112438201904,
      "step": 12805
    },
    {
      "epoch": 1.2211799621582031e-05,
      "step": 12805,
      "training_step_time": 17.066290140151978
    },
    {
      "epoch": 1.2212753295898437e-05,
      "model_forward_time": 5.957467794418335,
      "step": 12806
    },
    {
      "epoch": 1.2212753295898437e-05,
      "step": 12806,
      "training_step_time": 17.396040201187134
    },
    {
      "epoch": 1.2213706970214844e-05,
      "model_forward_time": 5.58240532875061,
      "step": 12807
    },
    {
      "epoch": 1.2213706970214844e-05,
      "step": 12807,
      "training_step_time": 17.013264894485474
    },
    {
      "epoch": 1.221466064453125e-05,
      "model_forward_time": 5.592653751373291,
      "step": 12808
    },
    {
      "epoch": 1.221466064453125e-05,
      "step": 12808,
      "training_step_time": 17.019599199295044
    },
    {
      "epoch": 1.2215614318847656e-05,
      "model_forward_time": 5.5928404331207275,
      "step": 12809
    },
    {
      "epoch": 1.2215614318847656e-05,
      "step": 12809,
      "training_step_time": 17.076545238494873
    },
    {
      "epoch": 1.2216567993164063e-05,
      "grad_norm": 1.506259799003601,
      "learning_rate": 1e-05,
      "loss": 0.0649,
      "step": 12810
    },
    {
      "epoch": 1.2216567993164063e-05,
      "model_forward_time": 5.61738657951355,
      "step": 12810
    },
    {
      "dynamics_loss_avg": 0.05385501366108656,
      "epoch": 1.2216567993164063e-05,
      "step": 12810
    },
    {
      "action_loss_avg": 0.008960565808229149,
      "epoch": 1.2216567993164063e-05,
      "step": 12810
    },
    {
      "epoch": 1.2216567993164063e-05,
      "step": 12810,
      "training_step_time": 17.14292025566101
    },
    {
      "epoch": 1.2217521667480469e-05,
      "model_forward_time": 5.631150722503662,
      "step": 12811
    },
    {
      "epoch": 1.2217521667480469e-05,
      "step": 12811,
      "training_step_time": 17.48054552078247
    },
    {
      "epoch": 1.2218475341796874e-05,
      "model_forward_time": 5.58995246887207,
      "step": 12812
    },
    {
      "epoch": 1.2218475341796874e-05,
      "step": 12812,
      "training_step_time": 17.009581565856934
    },
    {
      "epoch": 1.2219429016113282e-05,
      "model_forward_time": 5.560298442840576,
      "step": 12813
    },
    {
      "epoch": 1.2219429016113282e-05,
      "step": 12813,
      "training_step_time": 16.974541187286377
    },
    {
      "epoch": 1.2220382690429688e-05,
      "model_forward_time": 5.556999921798706,
      "step": 12814
    },
    {
      "epoch": 1.2220382690429688e-05,
      "step": 12814,
      "training_step_time": 16.977033853530884
    },
    {
      "epoch": 1.2221336364746093e-05,
      "model_forward_time": 5.559592008590698,
      "step": 12815
    },
    {
      "epoch": 1.2221336364746093e-05,
      "step": 12815,
      "training_step_time": 16.970797538757324
    },
    {
      "epoch": 1.22222900390625e-05,
      "model_forward_time": 5.559107303619385,
      "step": 12816
    },
    {
      "epoch": 1.22222900390625e-05,
      "step": 12816,
      "training_step_time": 16.9609375
    },
    {
      "epoch": 1.2223243713378906e-05,
      "model_forward_time": 5.9290452003479,
      "step": 12817
    },
    {
      "epoch": 1.2223243713378906e-05,
      "step": 12817,
      "training_step_time": 17.344991445541382
    },
    {
      "epoch": 1.2224197387695312e-05,
      "model_forward_time": 5.584939241409302,
      "step": 12818
    },
    {
      "epoch": 1.2224197387695312e-05,
      "step": 12818,
      "training_step_time": 17.019829511642456
    },
    {
      "epoch": 1.222515106201172e-05,
      "model_forward_time": 5.572228670120239,
      "step": 12819
    },
    {
      "epoch": 1.222515106201172e-05,
      "step": 12819,
      "training_step_time": 16.984227895736694
    },
    {
      "epoch": 1.2226104736328125e-05,
      "grad_norm": 0.5502747893333435,
      "learning_rate": 1e-05,
      "loss": 0.062,
      "step": 12820
    },
    {
      "epoch": 1.2226104736328125e-05,
      "model_forward_time": 5.5777177810668945,
      "step": 12820
    },
    {
      "dynamics_loss_avg": 0.053105642274022105,
      "epoch": 1.2226104736328125e-05,
      "step": 12820
    },
    {
      "action_loss_avg": 0.005162805231520906,
      "epoch": 1.2226104736328125e-05,
      "step": 12820
    },
    {
      "epoch": 1.2226104736328125e-05,
      "step": 12820,
      "training_step_time": 17.005202293395996
    },
    {
      "epoch": 1.2227058410644531e-05,
      "model_forward_time": 5.584577322006226,
      "step": 12821
    },
    {
      "epoch": 1.2227058410644531e-05,
      "step": 12821,
      "training_step_time": 17.012229919433594
    },
    {
      "epoch": 1.2228012084960937e-05,
      "model_forward_time": 5.590793132781982,
      "step": 12822
    },
    {
      "epoch": 1.2228012084960937e-05,
      "step": 12822,
      "training_step_time": 17.426575183868408
    },
    {
      "epoch": 1.2228965759277344e-05,
      "model_forward_time": 5.608388185501099,
      "step": 12823
    },
    {
      "epoch": 1.2228965759277344e-05,
      "step": 12823,
      "training_step_time": 17.101257801055908
    },
    {
      "epoch": 1.222991943359375e-05,
      "model_forward_time": 5.595370054244995,
      "step": 12824
    },
    {
      "epoch": 1.222991943359375e-05,
      "step": 12824,
      "training_step_time": 17.05755305290222
    },
    {
      "epoch": 1.2230873107910156e-05,
      "model_forward_time": 5.589414596557617,
      "step": 12825
    },
    {
      "epoch": 1.2230873107910156e-05,
      "step": 12825,
      "training_step_time": 17.029518365859985
    },
    {
      "epoch": 1.2231826782226563e-05,
      "model_forward_time": 5.577138900756836,
      "step": 12826
    },
    {
      "epoch": 1.2231826782226563e-05,
      "step": 12826,
      "training_step_time": 16.982898950576782
    },
    {
      "epoch": 1.2232780456542969e-05,
      "model_forward_time": 5.577908039093018,
      "step": 12827
    },
    {
      "epoch": 1.2232780456542969e-05,
      "step": 12827,
      "training_step_time": 17.02092671394348
    },
    {
      "epoch": 1.2233734130859374e-05,
      "model_forward_time": 5.969410419464111,
      "step": 12828
    },
    {
      "epoch": 1.2233734130859374e-05,
      "step": 12828,
      "training_step_time": 17.441682815551758
    },
    {
      "epoch": 1.2234687805175782e-05,
      "model_forward_time": 5.600733280181885,
      "step": 12829
    },
    {
      "epoch": 1.2234687805175782e-05,
      "step": 12829,
      "training_step_time": 17.061910390853882
    },
    {
      "epoch": 1.2235641479492188e-05,
      "grad_norm": 0.4294660985469818,
      "learning_rate": 1e-05,
      "loss": 0.0706,
      "step": 12830
    },
    {
      "epoch": 1.2235641479492188e-05,
      "model_forward_time": 5.604179382324219,
      "step": 12830
    },
    {
      "dynamics_loss_avg": 0.06559247262775898,
      "epoch": 1.2235641479492188e-05,
      "step": 12830
    },
    {
      "action_loss_avg": 0.008907749503850936,
      "epoch": 1.2235641479492188e-05,
      "step": 12830
    },
    {
      "epoch": 1.2235641479492188e-05,
      "step": 12830,
      "training_step_time": 17.05745530128479
    },
    {
      "epoch": 1.2236595153808593e-05,
      "model_forward_time": 5.592702150344849,
      "step": 12831
    },
    {
      "epoch": 1.2236595153808593e-05,
      "step": 12831,
      "training_step_time": 17.01202654838562
    },
    {
      "epoch": 1.2237548828125e-05,
      "model_forward_time": 5.589955806732178,
      "step": 12832
    },
    {
      "epoch": 1.2237548828125e-05,
      "step": 12832,
      "training_step_time": 17.005434274673462
    },
    {
      "epoch": 1.2238502502441406e-05,
      "model_forward_time": 5.95874285697937,
      "step": 12833
    },
    {
      "epoch": 1.2238502502441406e-05,
      "step": 12833,
      "training_step_time": 17.41396927833557
    },
    {
      "epoch": 1.2239456176757812e-05,
      "model_forward_time": 5.591400623321533,
      "step": 12834
    },
    {
      "epoch": 1.2239456176757812e-05,
      "step": 12834,
      "training_step_time": 17.068320989608765
    },
    {
      "epoch": 1.224040985107422e-05,
      "model_forward_time": 5.6197593212127686,
      "step": 12835
    },
    {
      "epoch": 1.224040985107422e-05,
      "step": 12835,
      "training_step_time": 17.129535675048828
    },
    {
      "epoch": 1.2241363525390625e-05,
      "model_forward_time": 5.618127107620239,
      "step": 12836
    },
    {
      "epoch": 1.2241363525390625e-05,
      "step": 12836,
      "training_step_time": 17.108091831207275
    },
    {
      "epoch": 1.2242317199707031e-05,
      "model_forward_time": 5.617461681365967,
      "step": 12837
    },
    {
      "epoch": 1.2242317199707031e-05,
      "step": 12837,
      "training_step_time": 17.092047691345215
    },
    {
      "epoch": 1.2243270874023437e-05,
      "model_forward_time": 5.5996105670928955,
      "step": 12838
    },
    {
      "epoch": 1.2243270874023437e-05,
      "step": 12838,
      "training_step_time": 17.074735164642334
    },
    {
      "epoch": 1.2244224548339844e-05,
      "model_forward_time": 5.979129076004028,
      "step": 12839
    },
    {
      "epoch": 1.2244224548339844e-05,
      "step": 12839,
      "training_step_time": 17.43749189376831
    },
    {
      "epoch": 1.224517822265625e-05,
      "grad_norm": 0.2462652176618576,
      "learning_rate": 1e-05,
      "loss": 0.0731,
      "step": 12840
    },
    {
      "epoch": 1.224517822265625e-05,
      "model_forward_time": 5.60112738609314,
      "step": 12840
    },
    {
      "dynamics_loss_avg": 0.06452646553516388,
      "epoch": 1.224517822265625e-05,
      "step": 12840
    },
    {
      "action_loss_avg": 0.007529604888986796,
      "epoch": 1.224517822265625e-05,
      "step": 12840
    },
    {
      "epoch": 1.224517822265625e-05,
      "step": 12840,
      "training_step_time": 17.049463272094727
    },
    {
      "epoch": 1.2246131896972656e-05,
      "model_forward_time": 5.594829082489014,
      "step": 12841
    },
    {
      "epoch": 1.2246131896972656e-05,
      "step": 12841,
      "training_step_time": 17.049057722091675
    },
    {
      "epoch": 1.2247085571289063e-05,
      "model_forward_time": 5.5961689949035645,
      "step": 12842
    },
    {
      "epoch": 1.2247085571289063e-05,
      "step": 12842,
      "training_step_time": 17.04490041732788
    },
    {
      "epoch": 1.2248039245605469e-05,
      "model_forward_time": 5.602619886398315,
      "step": 12843
    },
    {
      "epoch": 1.2248039245605469e-05,
      "step": 12843,
      "training_step_time": 17.09741473197937
    },
    {
      "epoch": 1.2248992919921874e-05,
      "model_forward_time": 5.6190102100372314,
      "step": 12844
    },
    {
      "epoch": 1.2248992919921874e-05,
      "step": 12844,
      "training_step_time": 17.478103637695312
    },
    {
      "epoch": 1.2249946594238282e-05,
      "model_forward_time": 5.620450258255005,
      "step": 12845
    },
    {
      "epoch": 1.2249946594238282e-05,
      "step": 12845,
      "training_step_time": 17.102604150772095
    },
    {
      "epoch": 1.2250900268554688e-05,
      "model_forward_time": 5.597283124923706,
      "step": 12846
    },
    {
      "epoch": 1.2250900268554688e-05,
      "step": 12846,
      "training_step_time": 17.074669361114502
    },
    {
      "epoch": 1.2251853942871093e-05,
      "model_forward_time": 5.604276180267334,
      "step": 12847
    },
    {
      "epoch": 1.2251853942871093e-05,
      "step": 12847,
      "training_step_time": 17.084794759750366
    },
    {
      "epoch": 1.22528076171875e-05,
      "model_forward_time": 5.602597951889038,
      "step": 12848
    },
    {
      "epoch": 1.22528076171875e-05,
      "step": 12848,
      "training_step_time": 17.074324131011963
    },
    {
      "epoch": 1.2253761291503906e-05,
      "model_forward_time": 5.605366468429565,
      "step": 12849
    },
    {
      "epoch": 1.2253761291503906e-05,
      "step": 12849,
      "training_step_time": 17.084449768066406
    },
    {
      "epoch": 1.2254714965820312e-05,
      "grad_norm": 0.2987581789493561,
      "learning_rate": 1e-05,
      "loss": 0.0569,
      "step": 12850
    },
    {
      "epoch": 1.2254714965820312e-05,
      "model_forward_time": 5.978436470031738,
      "step": 12850
    },
    {
      "dynamics_loss_avg": 0.05164974806830287,
      "epoch": 1.2254714965820312e-05,
      "step": 12850
    },
    {
      "action_loss_avg": 0.00479409946128726,
      "epoch": 1.2254714965820312e-05,
      "step": 12850
    },
    {
      "epoch": 1.2254714965820312e-05,
      "step": 12850,
      "training_step_time": 17.452311992645264
    },
    {
      "epoch": 1.225566864013672e-05,
      "model_forward_time": 5.626385688781738,
      "step": 12851
    },
    {
      "epoch": 1.225566864013672e-05,
      "step": 12851,
      "training_step_time": 17.068182706832886
    },
    {
      "epoch": 1.2256622314453125e-05,
      "model_forward_time": 5.59226393699646,
      "step": 12852
    },
    {
      "epoch": 1.2256622314453125e-05,
      "step": 12852,
      "training_step_time": 17.026494026184082
    },
    {
      "epoch": 1.2257575988769531e-05,
      "model_forward_time": 5.595524787902832,
      "step": 12853
    },
    {
      "epoch": 1.2257575988769531e-05,
      "step": 12853,
      "training_step_time": 17.062658071517944
    },
    {
      "epoch": 1.2258529663085937e-05,
      "model_forward_time": 5.642571449279785,
      "step": 12854
    },
    {
      "epoch": 1.2258529663085937e-05,
      "step": 12854,
      "training_step_time": 17.16927933692932
    },
    {
      "epoch": 1.2259483337402344e-05,
      "model_forward_time": 5.62151575088501,
      "step": 12855
    },
    {
      "epoch": 1.2259483337402344e-05,
      "step": 12855,
      "training_step_time": 17.459778785705566
    },
    {
      "epoch": 1.226043701171875e-05,
      "model_forward_time": 5.604307651519775,
      "step": 12856
    },
    {
      "epoch": 1.226043701171875e-05,
      "step": 12856,
      "training_step_time": 17.07195019721985
    },
    {
      "epoch": 1.2261390686035156e-05,
      "model_forward_time": 5.598958730697632,
      "step": 12857
    },
    {
      "epoch": 1.2261390686035156e-05,
      "step": 12857,
      "training_step_time": 17.050209522247314
    },
    {
      "epoch": 1.2262344360351563e-05,
      "model_forward_time": 5.59764552116394,
      "step": 12858
    },
    {
      "epoch": 1.2262344360351563e-05,
      "step": 12858,
      "training_step_time": 17.044757604599
    },
    {
      "epoch": 1.2263298034667969e-05,
      "model_forward_time": 5.590972423553467,
      "step": 12859
    },
    {
      "epoch": 1.2263298034667969e-05,
      "step": 12859,
      "training_step_time": 17.035479068756104
    },
    {
      "epoch": 1.2264251708984374e-05,
      "grad_norm": 1.0070663690567017,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 12860
    },
    {
      "epoch": 1.2264251708984374e-05,
      "model_forward_time": 5.589945554733276,
      "step": 12860
    },
    {
      "dynamics_loss_avg": 0.0642965892329812,
      "epoch": 1.2264251708984374e-05,
      "step": 12860
    },
    {
      "action_loss_avg": 0.006421659514307976,
      "epoch": 1.2264251708984374e-05,
      "step": 12860
    },
    {
      "epoch": 1.2264251708984374e-05,
      "step": 12860,
      "training_step_time": 17.026219129562378
    },
    {
      "epoch": 1.2265205383300782e-05,
      "model_forward_time": 5.975754022598267,
      "step": 12861
    },
    {
      "epoch": 1.2265205383300782e-05,
      "step": 12861,
      "training_step_time": 17.42726969718933
    },
    {
      "epoch": 1.2266159057617188e-05,
      "model_forward_time": 5.6020307540893555,
      "step": 12862
    },
    {
      "epoch": 1.2266159057617188e-05,
      "step": 12862,
      "training_step_time": 17.031063079833984
    },
    {
      "epoch": 1.2267112731933593e-05,
      "model_forward_time": 5.599432706832886,
      "step": 12863
    },
    {
      "epoch": 1.2267112731933593e-05,
      "step": 12863,
      "training_step_time": 17.072805643081665
    },
    {
      "epoch": 1.226806640625e-05,
      "model_forward_time": 5.605936050415039,
      "step": 12864
    },
    {
      "epoch": 1.226806640625e-05,
      "step": 12864,
      "training_step_time": 17.110766410827637
    },
    {
      "epoch": 1.2269020080566406e-05,
      "model_forward_time": 5.612011909484863,
      "step": 12865
    },
    {
      "epoch": 1.2269020080566406e-05,
      "step": 12865,
      "training_step_time": 17.083210945129395
    },
    {
      "epoch": 1.2269973754882812e-05,
      "model_forward_time": 5.9533326625823975,
      "step": 12866
    },
    {
      "epoch": 1.2269973754882812e-05,
      "step": 12866,
      "training_step_time": 17.374156951904297
    },
    {
      "epoch": 1.227092742919922e-05,
      "model_forward_time": 5.606241464614868,
      "step": 12867
    },
    {
      "epoch": 1.227092742919922e-05,
      "step": 12867,
      "training_step_time": 17.0507755279541
    },
    {
      "epoch": 1.2271881103515625e-05,
      "model_forward_time": 5.583618640899658,
      "step": 12868
    },
    {
      "epoch": 1.2271881103515625e-05,
      "step": 12868,
      "training_step_time": 17.013736724853516
    },
    {
      "epoch": 1.2272834777832031e-05,
      "model_forward_time": 5.584013938903809,
      "step": 12869
    },
    {
      "epoch": 1.2272834777832031e-05,
      "step": 12869,
      "training_step_time": 17.008463382720947
    },
    {
      "epoch": 1.2273788452148437e-05,
      "grad_norm": 1.2719722986221313,
      "learning_rate": 1e-05,
      "loss": 0.071,
      "step": 12870
    },
    {
      "epoch": 1.2273788452148437e-05,
      "model_forward_time": 5.594974040985107,
      "step": 12870
    },
    {
      "dynamics_loss_avg": 0.05883403904736042,
      "epoch": 1.2273788452148437e-05,
      "step": 12870
    },
    {
      "action_loss_avg": 0.007778864807914942,
      "epoch": 1.2273788452148437e-05,
      "step": 12870
    },
    {
      "epoch": 1.2273788452148437e-05,
      "step": 12870,
      "training_step_time": 17.036235094070435
    },
    {
      "epoch": 1.2274742126464844e-05,
      "model_forward_time": 5.598230361938477,
      "step": 12871
    },
    {
      "epoch": 1.2274742126464844e-05,
      "step": 12871,
      "training_step_time": 17.06587314605713
    },
    {
      "epoch": 1.227569580078125e-05,
      "model_forward_time": 5.978589296340942,
      "step": 12872
    },
    {
      "epoch": 1.227569580078125e-05,
      "step": 12872,
      "training_step_time": 17.453402042388916
    },
    {
      "epoch": 1.2276649475097656e-05,
      "model_forward_time": 5.621885299682617,
      "step": 12873
    },
    {
      "epoch": 1.2276649475097656e-05,
      "step": 12873,
      "training_step_time": 17.103353261947632
    },
    {
      "epoch": 1.2277603149414063e-05,
      "model_forward_time": 5.607544660568237,
      "step": 12874
    },
    {
      "epoch": 1.2277603149414063e-05,
      "step": 12874,
      "training_step_time": 17.102967500686646
    },
    {
      "epoch": 1.2278556823730469e-05,
      "model_forward_time": 5.615164041519165,
      "step": 12875
    },
    {
      "epoch": 1.2278556823730469e-05,
      "step": 12875,
      "training_step_time": 17.12047266960144
    },
    {
      "epoch": 1.2279510498046874e-05,
      "model_forward_time": 5.62730860710144,
      "step": 12876
    },
    {
      "epoch": 1.2279510498046874e-05,
      "step": 12876,
      "training_step_time": 17.100104331970215
    },
    {
      "epoch": 1.2280464172363282e-05,
      "model_forward_time": 5.583283424377441,
      "step": 12877
    },
    {
      "epoch": 1.2280464172363282e-05,
      "step": 12877,
      "training_step_time": 17.372586250305176
    },
    {
      "epoch": 1.2281417846679688e-05,
      "model_forward_time": 5.57917857170105,
      "step": 12878
    },
    {
      "epoch": 1.2281417846679688e-05,
      "step": 12878,
      "training_step_time": 17.017786264419556
    },
    {
      "epoch": 1.2282371520996093e-05,
      "model_forward_time": 5.578258991241455,
      "step": 12879
    },
    {
      "epoch": 1.2282371520996093e-05,
      "step": 12879,
      "training_step_time": 16.99448537826538
    },
    {
      "epoch": 1.22833251953125e-05,
      "grad_norm": 0.7562565803527832,
      "learning_rate": 1e-05,
      "loss": 0.0671,
      "step": 12880
    },
    {
      "epoch": 1.22833251953125e-05,
      "model_forward_time": 5.582671403884888,
      "step": 12880
    },
    {
      "dynamics_loss_avg": 0.05907453987747431,
      "epoch": 1.22833251953125e-05,
      "step": 12880
    },
    {
      "action_loss_avg": 0.006855051871389151,
      "epoch": 1.22833251953125e-05,
      "step": 12880
    },
    {
      "epoch": 1.22833251953125e-05,
      "step": 12880,
      "training_step_time": 16.987967252731323
    },
    {
      "epoch": 1.2284278869628906e-05,
      "model_forward_time": 5.580601692199707,
      "step": 12881
    },
    {
      "epoch": 1.2284278869628906e-05,
      "step": 12881,
      "training_step_time": 17.01309299468994
    },
    {
      "epoch": 1.2285232543945312e-05,
      "model_forward_time": 5.594581842422485,
      "step": 12882
    },
    {
      "epoch": 1.2285232543945312e-05,
      "step": 12882,
      "training_step_time": 17.048431158065796
    },
    {
      "epoch": 1.2286186218261718e-05,
      "model_forward_time": 5.977662563323975,
      "step": 12883
    },
    {
      "epoch": 1.2286186218261718e-05,
      "step": 12883,
      "training_step_time": 17.456050634384155
    },
    {
      "epoch": 1.2287139892578125e-05,
      "model_forward_time": 5.608180284500122,
      "step": 12884
    },
    {
      "epoch": 1.2287139892578125e-05,
      "step": 12884,
      "training_step_time": 17.067245483398438
    },
    {
      "epoch": 1.2288093566894531e-05,
      "model_forward_time": 5.592455863952637,
      "step": 12885
    },
    {
      "epoch": 1.2288093566894531e-05,
      "step": 12885,
      "training_step_time": 17.030585527420044
    },
    {
      "epoch": 1.2289047241210937e-05,
      "model_forward_time": 5.583733797073364,
      "step": 12886
    },
    {
      "epoch": 1.2289047241210937e-05,
      "step": 12886,
      "training_step_time": 17.03516721725464
    },
    {
      "epoch": 1.2290000915527344e-05,
      "model_forward_time": 5.602290391921997,
      "step": 12887
    },
    {
      "epoch": 1.2290000915527344e-05,
      "step": 12887,
      "training_step_time": 17.076387643814087
    },
    {
      "epoch": 1.229095458984375e-05,
      "model_forward_time": 5.6049957275390625,
      "step": 12888
    },
    {
      "epoch": 1.229095458984375e-05,
      "step": 12888,
      "training_step_time": 17.457228183746338
    },
    {
      "epoch": 1.2291908264160156e-05,
      "model_forward_time": 5.602473497390747,
      "step": 12889
    },
    {
      "epoch": 1.2291908264160156e-05,
      "step": 12889,
      "training_step_time": 17.061070680618286
    },
    {
      "epoch": 1.2292861938476563e-05,
      "grad_norm": 0.24317389726638794,
      "learning_rate": 1e-05,
      "loss": 0.0713,
      "step": 12890
    },
    {
      "epoch": 1.2292861938476563e-05,
      "model_forward_time": 5.591381311416626,
      "step": 12890
    },
    {
      "dynamics_loss_avg": 0.0642895213328302,
      "epoch": 1.2292861938476563e-05,
      "step": 12890
    },
    {
      "action_loss_avg": 0.006987884198315442,
      "epoch": 1.2292861938476563e-05,
      "step": 12890
    },
    {
      "epoch": 1.2292861938476563e-05,
      "step": 12890,
      "training_step_time": 16.95871329307556
    },
    {
      "epoch": 1.2293815612792969e-05,
      "model_forward_time": 5.597038745880127,
      "step": 12891
    },
    {
      "epoch": 1.2293815612792969e-05,
      "step": 12891,
      "training_step_time": 17.032037496566772
    },
    {
      "epoch": 1.2294769287109374e-05,
      "model_forward_time": 5.600753545761108,
      "step": 12892
    },
    {
      "epoch": 1.2294769287109374e-05,
      "step": 12892,
      "training_step_time": 17.06739616394043
    },
    {
      "epoch": 1.2295722961425782e-05,
      "model_forward_time": 5.610360860824585,
      "step": 12893
    },
    {
      "epoch": 1.2295722961425782e-05,
      "step": 12893,
      "training_step_time": 17.100658655166626
    },
    {
      "epoch": 1.2296676635742188e-05,
      "model_forward_time": 5.988913536071777,
      "step": 12894
    },
    {
      "epoch": 1.2296676635742188e-05,
      "step": 12894,
      "training_step_time": 17.49547553062439
    },
    {
      "epoch": 1.2297630310058593e-05,
      "model_forward_time": 5.620104789733887,
      "step": 12895
    },
    {
      "epoch": 1.2297630310058593e-05,
      "step": 12895,
      "training_step_time": 17.122310876846313
    },
    {
      "epoch": 1.2298583984375e-05,
      "model_forward_time": 5.622058868408203,
      "step": 12896
    },
    {
      "epoch": 1.2298583984375e-05,
      "step": 12896,
      "training_step_time": 17.120824575424194
    },
    {
      "epoch": 1.2299537658691406e-05,
      "model_forward_time": 5.616580963134766,
      "step": 12897
    },
    {
      "epoch": 1.2299537658691406e-05,
      "step": 12897,
      "training_step_time": 17.104504585266113
    },
    {
      "epoch": 1.2300491333007812e-05,
      "model_forward_time": 5.618380546569824,
      "step": 12898
    },
    {
      "epoch": 1.2300491333007812e-05,
      "step": 12898,
      "training_step_time": 17.11889338493347
    },
    {
      "epoch": 1.2301445007324218e-05,
      "model_forward_time": 5.97156286239624,
      "step": 12899
    },
    {
      "epoch": 1.2301445007324218e-05,
      "step": 12899,
      "training_step_time": 17.453287601470947
    },
    {
      "epoch": 1.2302398681640625e-05,
      "grad_norm": 0.40487703680992126,
      "learning_rate": 1e-05,
      "loss": 0.0564,
      "step": 12900
    },
    {
      "epoch": 1.2302398681640625e-05,
      "model_forward_time": 5.6196815967559814,
      "step": 12900
    },
    {
      "dynamics_loss_avg": 0.05399495773017406,
      "epoch": 1.2302398681640625e-05,
      "step": 12900
    },
    {
      "action_loss_avg": 0.006993933627381921,
      "epoch": 1.2302398681640625e-05,
      "step": 12900
    },
    {
      "epoch": 1.2302398681640625e-05,
      "step": 12900,
      "training_step_time": 17.109873056411743
    },
    {
      "epoch": 1.2303352355957031e-05,
      "model_forward_time": 5.610410213470459,
      "step": 12901
    },
    {
      "epoch": 1.2303352355957031e-05,
      "step": 12901,
      "training_step_time": 17.0989727973938
    },
    {
      "epoch": 1.2304306030273437e-05,
      "model_forward_time": 5.6139490604400635,
      "step": 12902
    },
    {
      "epoch": 1.2304306030273437e-05,
      "step": 12902,
      "training_step_time": 17.120535373687744
    },
    {
      "epoch": 1.2305259704589844e-05,
      "model_forward_time": 5.613569259643555,
      "step": 12903
    },
    {
      "epoch": 1.2305259704589844e-05,
      "step": 12903,
      "training_step_time": 17.087027311325073
    },
    {
      "epoch": 1.230621337890625e-05,
      "model_forward_time": 5.593170166015625,
      "step": 12904
    },
    {
      "epoch": 1.230621337890625e-05,
      "step": 12904,
      "training_step_time": 17.01275658607483
    },
    {
      "epoch": 1.2307167053222656e-05,
      "model_forward_time": 5.9955315589904785,
      "step": 12905
    },
    {
      "epoch": 1.2307167053222656e-05,
      "step": 12905,
      "training_step_time": 17.435741901397705
    },
    {
      "epoch": 1.2308120727539063e-05,
      "model_forward_time": 5.572579860687256,
      "step": 12906
    },
    {
      "epoch": 1.2308120727539063e-05,
      "step": 12906,
      "training_step_time": 16.988129138946533
    },
    {
      "epoch": 1.2309074401855469e-05,
      "model_forward_time": 5.572265863418579,
      "step": 12907
    },
    {
      "epoch": 1.2309074401855469e-05,
      "step": 12907,
      "training_step_time": 16.981764793395996
    },
    {
      "epoch": 1.2310028076171874e-05,
      "model_forward_time": 5.575322866439819,
      "step": 12908
    },
    {
      "epoch": 1.2310028076171874e-05,
      "step": 12908,
      "training_step_time": 17.002241373062134
    },
    {
      "epoch": 1.2310981750488282e-05,
      "model_forward_time": 5.587444543838501,
      "step": 12909
    },
    {
      "epoch": 1.2310981750488282e-05,
      "step": 12909,
      "training_step_time": 17.037784337997437
    },
    {
      "epoch": 1.2311935424804688e-05,
      "grad_norm": 0.691394567489624,
      "learning_rate": 1e-05,
      "loss": 0.0698,
      "step": 12910
    },
    {
      "epoch": 1.2311935424804688e-05,
      "model_forward_time": 5.603424310684204,
      "step": 12910
    },
    {
      "dynamics_loss_avg": 0.06540819481015206,
      "epoch": 1.2311935424804688e-05,
      "step": 12910
    },
    {
      "action_loss_avg": 0.008781649451702834,
      "epoch": 1.2311935424804688e-05,
      "step": 12910
    },
    {
      "epoch": 1.2311935424804688e-05,
      "step": 12910,
      "training_step_time": 17.461724281311035
    },
    {
      "epoch": 1.2312889099121093e-05,
      "model_forward_time": 5.618134498596191,
      "step": 12911
    },
    {
      "epoch": 1.2312889099121093e-05,
      "step": 12911,
      "training_step_time": 17.13954997062683
    },
    {
      "epoch": 1.23138427734375e-05,
      "model_forward_time": 5.6316444873809814,
      "step": 12912
    },
    {
      "epoch": 1.23138427734375e-05,
      "step": 12912,
      "training_step_time": 17.126219511032104
    },
    {
      "epoch": 1.2314796447753906e-05,
      "model_forward_time": 5.609705686569214,
      "step": 12913
    },
    {
      "epoch": 1.2314796447753906e-05,
      "step": 12913,
      "training_step_time": 17.073467016220093
    },
    {
      "epoch": 1.2315750122070312e-05,
      "model_forward_time": 5.58843469619751,
      "step": 12914
    },
    {
      "epoch": 1.2315750122070312e-05,
      "step": 12914,
      "training_step_time": 17.00904369354248
    },
    {
      "epoch": 1.2316703796386718e-05,
      "model_forward_time": 5.5950517654418945,
      "step": 12915
    },
    {
      "epoch": 1.2316703796386718e-05,
      "step": 12915,
      "training_step_time": 17.071114778518677
    },
    {
      "epoch": 1.2317657470703125e-05,
      "model_forward_time": 6.005401372909546,
      "step": 12916
    },
    {
      "epoch": 1.2317657470703125e-05,
      "step": 12916,
      "training_step_time": 17.518625259399414
    },
    {
      "epoch": 1.2318611145019531e-05,
      "model_forward_time": 5.624638795852661,
      "step": 12917
    },
    {
      "epoch": 1.2318611145019531e-05,
      "step": 12917,
      "training_step_time": 17.11415195465088
    },
    {
      "epoch": 1.2319564819335937e-05,
      "model_forward_time": 5.595566749572754,
      "step": 12918
    },
    {
      "epoch": 1.2319564819335937e-05,
      "step": 12918,
      "training_step_time": 17.036668062210083
    },
    {
      "epoch": 1.2320518493652344e-05,
      "model_forward_time": 5.589077711105347,
      "step": 12919
    },
    {
      "epoch": 1.2320518493652344e-05,
      "step": 12919,
      "training_step_time": 17.004385232925415
    },
    {
      "epoch": 1.232147216796875e-05,
      "grad_norm": 0.47452646493911743,
      "learning_rate": 1e-05,
      "loss": 0.0722,
      "step": 12920
    },
    {
      "epoch": 1.232147216796875e-05,
      "model_forward_time": 5.589249849319458,
      "step": 12920
    },
    {
      "dynamics_loss_avg": 0.060342921130359176,
      "epoch": 1.232147216796875e-05,
      "step": 12920
    },
    {
      "action_loss_avg": 0.00672684699529782,
      "epoch": 1.232147216796875e-05,
      "step": 12920
    },
    {
      "epoch": 1.232147216796875e-05,
      "step": 12920,
      "training_step_time": 17.021488428115845
    },
    {
      "epoch": 1.2322425842285156e-05,
      "model_forward_time": 5.585310459136963,
      "step": 12921
    },
    {
      "epoch": 1.2322425842285156e-05,
      "step": 12921,
      "training_step_time": 17.402682781219482
    },
    {
      "epoch": 1.2323379516601563e-05,
      "model_forward_time": 5.611891508102417,
      "step": 12922
    },
    {
      "epoch": 1.2323379516601563e-05,
      "step": 12922,
      "training_step_time": 17.099673986434937
    },
    {
      "epoch": 1.2324333190917969e-05,
      "model_forward_time": 5.6091015338897705,
      "step": 12923
    },
    {
      "epoch": 1.2324333190917969e-05,
      "step": 12923,
      "training_step_time": 17.083613634109497
    },
    {
      "epoch": 1.2325286865234375e-05,
      "model_forward_time": 5.609605550765991,
      "step": 12924
    },
    {
      "epoch": 1.2325286865234375e-05,
      "step": 12924,
      "training_step_time": 17.099397659301758
    },
    {
      "epoch": 1.2326240539550782e-05,
      "model_forward_time": 5.6111040115356445,
      "step": 12925
    },
    {
      "epoch": 1.2326240539550782e-05,
      "step": 12925,
      "training_step_time": 17.09864616394043
    },
    {
      "epoch": 1.2327194213867188e-05,
      "model_forward_time": 5.609188795089722,
      "step": 12926
    },
    {
      "epoch": 1.2327194213867188e-05,
      "step": 12926,
      "training_step_time": 17.085931301116943
    },
    {
      "epoch": 1.2328147888183593e-05,
      "model_forward_time": 5.986872673034668,
      "step": 12927
    },
    {
      "epoch": 1.2328147888183593e-05,
      "step": 12927,
      "training_step_time": 17.44336462020874
    },
    {
      "epoch": 1.23291015625e-05,
      "model_forward_time": 5.590268135070801,
      "step": 12928
    },
    {
      "epoch": 1.23291015625e-05,
      "step": 12928,
      "training_step_time": 17.01985216140747
    },
    {
      "epoch": 1.2330055236816407e-05,
      "model_forward_time": 5.6008288860321045,
      "step": 12929
    },
    {
      "epoch": 1.2330055236816407e-05,
      "step": 12929,
      "training_step_time": 17.0666983127594
    },
    {
      "epoch": 1.2331008911132812e-05,
      "grad_norm": 0.7908021807670593,
      "learning_rate": 1e-05,
      "loss": 0.0617,
      "step": 12930
    },
    {
      "epoch": 1.2331008911132812e-05,
      "model_forward_time": 5.60633397102356,
      "step": 12930
    },
    {
      "dynamics_loss_avg": 0.052515242248773575,
      "epoch": 1.2331008911132812e-05,
      "step": 12930
    },
    {
      "action_loss_avg": 0.006826355564408004,
      "epoch": 1.2331008911132812e-05,
      "step": 12930
    },
    {
      "epoch": 1.2331008911132812e-05,
      "step": 12930,
      "training_step_time": 17.090614318847656
    },
    {
      "epoch": 1.2331962585449218e-05,
      "model_forward_time": 5.612122297286987,
      "step": 12931
    },
    {
      "epoch": 1.2331962585449218e-05,
      "step": 12931,
      "training_step_time": 17.0741765499115
    },
    {
      "epoch": 1.2332916259765625e-05,
      "model_forward_time": 5.965843915939331,
      "step": 12932
    },
    {
      "epoch": 1.2332916259765625e-05,
      "step": 12932,
      "training_step_time": 17.431531190872192
    },
    {
      "epoch": 1.2333869934082031e-05,
      "model_forward_time": 5.594693899154663,
      "step": 12933
    },
    {
      "epoch": 1.2333869934082031e-05,
      "step": 12933,
      "training_step_time": 17.034226179122925
    },
    {
      "epoch": 1.2334823608398437e-05,
      "model_forward_time": 5.595633029937744,
      "step": 12934
    },
    {
      "epoch": 1.2334823608398437e-05,
      "step": 12934,
      "training_step_time": 17.056031703948975
    },
    {
      "epoch": 1.2335777282714844e-05,
      "model_forward_time": 5.6047422885894775,
      "step": 12935
    },
    {
      "epoch": 1.2335777282714844e-05,
      "step": 12935,
      "training_step_time": 17.089481353759766
    },
    {
      "epoch": 1.233673095703125e-05,
      "model_forward_time": 5.619255542755127,
      "step": 12936
    },
    {
      "epoch": 1.233673095703125e-05,
      "step": 12936,
      "training_step_time": 17.130021572113037
    },
    {
      "epoch": 1.2337684631347656e-05,
      "model_forward_time": 5.629643440246582,
      "step": 12937
    },
    {
      "epoch": 1.2337684631347656e-05,
      "step": 12937,
      "training_step_time": 17.142859935760498
    },
    {
      "epoch": 1.2338638305664063e-05,
      "model_forward_time": 6.002323627471924,
      "step": 12938
    },
    {
      "epoch": 1.2338638305664063e-05,
      "step": 12938,
      "training_step_time": 17.494309425354004
    },
    {
      "epoch": 1.2339591979980469e-05,
      "model_forward_time": 5.591451644897461,
      "step": 12939
    },
    {
      "epoch": 1.2339591979980469e-05,
      "step": 12939,
      "training_step_time": 17.047199964523315
    },
    {
      "epoch": 1.2340545654296875e-05,
      "grad_norm": 0.5628030300140381,
      "learning_rate": 1e-05,
      "loss": 0.0599,
      "step": 12940
    },
    {
      "epoch": 1.2340545654296875e-05,
      "model_forward_time": 5.5812623500823975,
      "step": 12940
    },
    {
      "dynamics_loss_avg": 0.055153119890019295,
      "epoch": 1.2340545654296875e-05,
      "step": 12940
    },
    {
      "action_loss_avg": 0.007820101641118526,
      "epoch": 1.2340545654296875e-05,
      "step": 12940
    },
    {
      "epoch": 1.2340545654296875e-05,
      "step": 12940,
      "training_step_time": 17.00011897087097
    },
    {
      "epoch": 1.2341499328613282e-05,
      "model_forward_time": 5.582522630691528,
      "step": 12941
    },
    {
      "epoch": 1.2341499328613282e-05,
      "step": 12941,
      "training_step_time": 17.015965461730957
    },
    {
      "epoch": 1.2342453002929688e-05,
      "model_forward_time": 5.5930564403533936,
      "step": 12942
    },
    {
      "epoch": 1.2342453002929688e-05,
      "step": 12942,
      "training_step_time": 17.064777851104736
    },
    {
      "epoch": 1.2343406677246093e-05,
      "model_forward_time": 5.605154037475586,
      "step": 12943
    },
    {
      "epoch": 1.2343406677246093e-05,
      "step": 12943,
      "training_step_time": 17.431921005249023
    },
    {
      "epoch": 1.23443603515625e-05,
      "model_forward_time": 5.599014043807983,
      "step": 12944
    },
    {
      "epoch": 1.23443603515625e-05,
      "step": 12944,
      "training_step_time": 17.082599878311157
    },
    {
      "epoch": 1.2345314025878907e-05,
      "model_forward_time": 5.60223650932312,
      "step": 12945
    },
    {
      "epoch": 1.2345314025878907e-05,
      "step": 12945,
      "training_step_time": 17.04944157600403
    },
    {
      "epoch": 1.2346267700195312e-05,
      "model_forward_time": 5.582905292510986,
      "step": 12946
    },
    {
      "epoch": 1.2346267700195312e-05,
      "step": 12946,
      "training_step_time": 17.004948377609253
    },
    {
      "epoch": 1.2347221374511718e-05,
      "model_forward_time": 5.590749025344849,
      "step": 12947
    },
    {
      "epoch": 1.2347221374511718e-05,
      "step": 12947,
      "training_step_time": 17.026973724365234
    },
    {
      "epoch": 1.2348175048828125e-05,
      "model_forward_time": 5.598544120788574,
      "step": 12948
    },
    {
      "epoch": 1.2348175048828125e-05,
      "step": 12948,
      "training_step_time": 17.09077548980713
    },
    {
      "epoch": 1.2349128723144531e-05,
      "model_forward_time": 6.002611875534058,
      "step": 12949
    },
    {
      "epoch": 1.2349128723144531e-05,
      "step": 12949,
      "training_step_time": 17.494983196258545
    },
    {
      "epoch": 1.2350082397460937e-05,
      "grad_norm": 0.8284071087837219,
      "learning_rate": 1e-05,
      "loss": 0.0627,
      "step": 12950
    },
    {
      "epoch": 1.2350082397460937e-05,
      "model_forward_time": 5.606727361679077,
      "step": 12950
    },
    {
      "dynamics_loss_avg": 0.05340253058820963,
      "epoch": 1.2350082397460937e-05,
      "step": 12950
    },
    {
      "action_loss_avg": 0.005613280949182809,
      "epoch": 1.2350082397460937e-05,
      "step": 12950
    },
    {
      "epoch": 1.2350082397460937e-05,
      "step": 12950,
      "training_step_time": 17.072630167007446
    },
    {
      "epoch": 1.2351036071777344e-05,
      "model_forward_time": 5.602970600128174,
      "step": 12951
    },
    {
      "epoch": 1.2351036071777344e-05,
      "step": 12951,
      "training_step_time": 17.082274198532104
    },
    {
      "epoch": 1.235198974609375e-05,
      "model_forward_time": 5.609144687652588,
      "step": 12952
    },
    {
      "epoch": 1.235198974609375e-05,
      "step": 12952,
      "training_step_time": 17.08204674720764
    },
    {
      "epoch": 1.2352943420410156e-05,
      "model_forward_time": 5.601228952407837,
      "step": 12953
    },
    {
      "epoch": 1.2352943420410156e-05,
      "step": 12953,
      "training_step_time": 17.052798986434937
    },
    {
      "epoch": 1.2353897094726563e-05,
      "model_forward_time": 5.5882415771484375,
      "step": 12954
    },
    {
      "epoch": 1.2353897094726563e-05,
      "step": 12954,
      "training_step_time": 17.391669034957886
    },
    {
      "epoch": 1.2354850769042969e-05,
      "model_forward_time": 5.58165979385376,
      "step": 12955
    },
    {
      "epoch": 1.2354850769042969e-05,
      "step": 12955,
      "training_step_time": 17.006457090377808
    },
    {
      "epoch": 1.2355804443359375e-05,
      "model_forward_time": 5.59741997718811,
      "step": 12956
    },
    {
      "epoch": 1.2355804443359375e-05,
      "step": 12956,
      "training_step_time": 17.062541723251343
    },
    {
      "epoch": 1.2356758117675782e-05,
      "model_forward_time": 5.602643966674805,
      "step": 12957
    },
    {
      "epoch": 1.2356758117675782e-05,
      "step": 12957,
      "training_step_time": 17.068746328353882
    },
    {
      "epoch": 1.2357711791992188e-05,
      "model_forward_time": 5.606025695800781,
      "step": 12958
    },
    {
      "epoch": 1.2357711791992188e-05,
      "step": 12958,
      "training_step_time": 17.04123091697693
    },
    {
      "epoch": 1.2358665466308593e-05,
      "model_forward_time": 5.600924491882324,
      "step": 12959
    },
    {
      "epoch": 1.2358665466308593e-05,
      "step": 12959,
      "training_step_time": 17.0655574798584
    },
    {
      "epoch": 1.2359619140625e-05,
      "grad_norm": 0.5070855021476746,
      "learning_rate": 1e-05,
      "loss": 0.0721,
      "step": 12960
    },
    {
      "epoch": 1.2359619140625e-05,
      "model_forward_time": 5.96290135383606,
      "step": 12960
    },
    {
      "dynamics_loss_avg": 0.0655812731012702,
      "epoch": 1.2359619140625e-05,
      "step": 12960
    },
    {
      "action_loss_avg": 0.008371900138445198,
      "epoch": 1.2359619140625e-05,
      "step": 12960
    },
    {
      "epoch": 1.2359619140625e-05,
      "step": 12960,
      "training_step_time": 17.404846668243408
    },
    {
      "epoch": 1.2360572814941407e-05,
      "model_forward_time": 5.585692882537842,
      "step": 12961
    },
    {
      "epoch": 1.2360572814941407e-05,
      "step": 12961,
      "training_step_time": 17.019222497940063
    },
    {
      "epoch": 1.2361526489257812e-05,
      "model_forward_time": 5.587235927581787,
      "step": 12962
    },
    {
      "epoch": 1.2361526489257812e-05,
      "step": 12962,
      "training_step_time": 17.057058095932007
    },
    {
      "epoch": 1.2362480163574218e-05,
      "model_forward_time": 4.277239561080933,
      "step": 12963
    },
    {
      "epoch": 1.2362480163574218e-05,
      "step": 12963,
      "training_step_time": 12.808000326156616
    },
    {
      "epoch": 1.2363433837890625e-05,
      "model_forward_time": 5.607143878936768,
      "step": 12964
    },
    {
      "epoch": 1.2363433837890625e-05,
      "step": 12964,
      "training_step_time": 17.12262988090515
    },
    {
      "epoch": 1.2364387512207031e-05,
      "model_forward_time": 5.606080055236816,
      "step": 12965
    },
    {
      "epoch": 1.2364387512207031e-05,
      "step": 12965,
      "training_step_time": 17.454861640930176
    },
    {
      "epoch": 1.2365341186523437e-05,
      "model_forward_time": 5.614478349685669,
      "step": 12966
    },
    {
      "epoch": 1.2365341186523437e-05,
      "step": 12966,
      "training_step_time": 17.120551824569702
    },
    {
      "epoch": 1.2366294860839844e-05,
      "model_forward_time": 5.613245964050293,
      "step": 12967
    },
    {
      "epoch": 1.2366294860839844e-05,
      "step": 12967,
      "training_step_time": 17.110803365707397
    },
    {
      "epoch": 1.236724853515625e-05,
      "model_forward_time": 5.621252536773682,
      "step": 12968
    },
    {
      "epoch": 1.236724853515625e-05,
      "step": 12968,
      "training_step_time": 17.115313053131104
    },
    {
      "epoch": 1.2368202209472656e-05,
      "model_forward_time": 5.6116721630096436,
      "step": 12969
    },
    {
      "epoch": 1.2368202209472656e-05,
      "step": 12969,
      "training_step_time": 17.090310096740723
    },
    {
      "epoch": 1.2369155883789063e-05,
      "grad_norm": 1.4713075160980225,
      "learning_rate": 1e-05,
      "loss": 0.0788,
      "step": 12970
    },
    {
      "epoch": 1.2369155883789063e-05,
      "model_forward_time": 5.6167168617248535,
      "step": 12970
    },
    {
      "dynamics_loss_avg": 0.06936103105545044,
      "epoch": 1.2369155883789063e-05,
      "step": 12970
    },
    {
      "action_loss_avg": 0.009151448518969119,
      "epoch": 1.2369155883789063e-05,
      "step": 12970
    },
    {
      "epoch": 1.2369155883789063e-05,
      "step": 12970,
      "training_step_time": 17.09834909439087
    },
    {
      "epoch": 1.2370109558105469e-05,
      "model_forward_time": 5.978704452514648,
      "step": 12971
    },
    {
      "epoch": 1.2370109558105469e-05,
      "step": 12971,
      "training_step_time": 17.468058109283447
    },
    {
      "epoch": 1.2371063232421875e-05,
      "model_forward_time": 5.610280990600586,
      "step": 12972
    },
    {
      "epoch": 1.2371063232421875e-05,
      "step": 12972,
      "training_step_time": 17.098546743392944
    },
    {
      "epoch": 1.2372016906738282e-05,
      "model_forward_time": 5.601556777954102,
      "step": 12973
    },
    {
      "epoch": 1.2372016906738282e-05,
      "step": 12973,
      "training_step_time": 17.05151128768921
    },
    {
      "epoch": 1.2372970581054688e-05,
      "model_forward_time": 5.610580205917358,
      "step": 12974
    },
    {
      "epoch": 1.2372970581054688e-05,
      "step": 12974,
      "training_step_time": 17.111170768737793
    },
    {
      "epoch": 1.2373924255371093e-05,
      "model_forward_time": 5.612922668457031,
      "step": 12975
    },
    {
      "epoch": 1.2373924255371093e-05,
      "step": 12975,
      "training_step_time": 17.111103773117065
    },
    {
      "epoch": 1.23748779296875e-05,
      "model_forward_time": 5.622070074081421,
      "step": 12976
    },
    {
      "epoch": 1.23748779296875e-05,
      "step": 12976,
      "training_step_time": 17.448925495147705
    },
    {
      "epoch": 1.2375831604003907e-05,
      "model_forward_time": 5.588461637496948,
      "step": 12977
    },
    {
      "epoch": 1.2375831604003907e-05,
      "step": 12977,
      "training_step_time": 17.0224666595459
    },
    {
      "epoch": 1.2376785278320312e-05,
      "model_forward_time": 5.558842658996582,
      "step": 12978
    },
    {
      "epoch": 1.2376785278320312e-05,
      "step": 12978,
      "training_step_time": 16.979397296905518
    },
    {
      "epoch": 1.2377738952636718e-05,
      "model_forward_time": 5.562281608581543,
      "step": 12979
    },
    {
      "epoch": 1.2377738952636718e-05,
      "step": 12979,
      "training_step_time": 16.975897312164307
    },
    {
      "epoch": 1.2378692626953125e-05,
      "grad_norm": 1.175264596939087,
      "learning_rate": 1e-05,
      "loss": 0.0557,
      "step": 12980
    },
    {
      "epoch": 1.2378692626953125e-05,
      "model_forward_time": 5.562883377075195,
      "step": 12980
    },
    {
      "dynamics_loss_avg": 0.05182206705212593,
      "epoch": 1.2378692626953125e-05,
      "step": 12980
    },
    {
      "action_loss_avg": 0.004140181140974164,
      "epoch": 1.2378692626953125e-05,
      "step": 12980
    },
    {
      "epoch": 1.2378692626953125e-05,
      "step": 12980,
      "training_step_time": 16.973763704299927
    },
    {
      "epoch": 1.2379646301269531e-05,
      "model_forward_time": 5.563173532485962,
      "step": 12981
    },
    {
      "epoch": 1.2379646301269531e-05,
      "step": 12981,
      "training_step_time": 16.982698917388916
    },
    {
      "epoch": 1.2380599975585937e-05,
      "model_forward_time": 5.933973789215088,
      "step": 12982
    },
    {
      "epoch": 1.2380599975585937e-05,
      "step": 12982,
      "training_step_time": 17.349313259124756
    },
    {
      "epoch": 1.2381553649902344e-05,
      "model_forward_time": 5.570989608764648,
      "step": 12983
    },
    {
      "epoch": 1.2381553649902344e-05,
      "step": 12983,
      "training_step_time": 16.979166746139526
    },
    {
      "epoch": 1.238250732421875e-05,
      "model_forward_time": 5.59111213684082,
      "step": 12984
    },
    {
      "epoch": 1.238250732421875e-05,
      "step": 12984,
      "training_step_time": 17.01710820198059
    },
    {
      "epoch": 1.2383460998535156e-05,
      "model_forward_time": 5.587050199508667,
      "step": 12985
    },
    {
      "epoch": 1.2383460998535156e-05,
      "step": 12985,
      "training_step_time": 17.042185068130493
    },
    {
      "epoch": 1.2384414672851563e-05,
      "model_forward_time": 5.603043556213379,
      "step": 12986
    },
    {
      "epoch": 1.2384414672851563e-05,
      "step": 12986,
      "training_step_time": 17.08928918838501
    },
    {
      "epoch": 1.2385368347167969e-05,
      "model_forward_time": 5.612491607666016,
      "step": 12987
    },
    {
      "epoch": 1.2385368347167969e-05,
      "step": 12987,
      "training_step_time": 17.498044967651367
    },
    {
      "epoch": 1.2386322021484375e-05,
      "model_forward_time": 5.620389938354492,
      "step": 12988
    },
    {
      "epoch": 1.2386322021484375e-05,
      "step": 12988,
      "training_step_time": 17.10937809944153
    },
    {
      "epoch": 1.2387275695800782e-05,
      "model_forward_time": 5.5993146896362305,
      "step": 12989
    },
    {
      "epoch": 1.2387275695800782e-05,
      "step": 12989,
      "training_step_time": 17.03341817855835
    },
    {
      "epoch": 1.2388229370117188e-05,
      "grad_norm": 1.6313291788101196,
      "learning_rate": 1e-05,
      "loss": 0.0762,
      "step": 12990
    },
    {
      "epoch": 1.2388229370117188e-05,
      "model_forward_time": 5.585526943206787,
      "step": 12990
    },
    {
      "dynamics_loss_avg": 0.06568976435810328,
      "epoch": 1.2388229370117188e-05,
      "step": 12990
    },
    {
      "action_loss_avg": 0.00918171787634492,
      "epoch": 1.2388229370117188e-05,
      "step": 12990
    },
    {
      "epoch": 1.2388229370117188e-05,
      "step": 12990,
      "training_step_time": 17.018101692199707
    },
    {
      "epoch": 1.2389183044433593e-05,
      "model_forward_time": 5.578155517578125,
      "step": 12991
    },
    {
      "epoch": 1.2389183044433593e-05,
      "step": 12991,
      "training_step_time": 16.99365544319153
    },
    {
      "epoch": 1.239013671875e-05,
      "model_forward_time": 5.597763299942017,
      "step": 12992
    },
    {
      "epoch": 1.239013671875e-05,
      "step": 12992,
      "training_step_time": 17.037553548812866
    },
    {
      "epoch": 1.2391090393066407e-05,
      "model_forward_time": 5.9656572341918945,
      "step": 12993
    },
    {
      "epoch": 1.2391090393066407e-05,
      "step": 12993,
      "training_step_time": 17.454431295394897
    },
    {
      "epoch": 1.2392044067382812e-05,
      "model_forward_time": 5.626678705215454,
      "step": 12994
    },
    {
      "epoch": 1.2392044067382812e-05,
      "step": 12994,
      "training_step_time": 17.15128493309021
    },
    {
      "epoch": 1.2392997741699218e-05,
      "model_forward_time": 5.633912801742554,
      "step": 12995
    },
    {
      "epoch": 1.2392997741699218e-05,
      "step": 12995,
      "training_step_time": 17.1122944355011
    },
    {
      "epoch": 1.2393951416015625e-05,
      "model_forward_time": 5.589744329452515,
      "step": 12996
    },
    {
      "epoch": 1.2393951416015625e-05,
      "step": 12996,
      "training_step_time": 17.02615261077881
    },
    {
      "epoch": 1.2394905090332031e-05,
      "model_forward_time": 5.587246894836426,
      "step": 12997
    },
    {
      "epoch": 1.2394905090332031e-05,
      "step": 12997,
      "training_step_time": 17.009297370910645
    },
    {
      "epoch": 1.2395858764648437e-05,
      "model_forward_time": 5.931161880493164,
      "step": 12998
    },
    {
      "epoch": 1.2395858764648437e-05,
      "step": 12998,
      "training_step_time": 17.339760780334473
    },
    {
      "epoch": 1.2396812438964844e-05,
      "model_forward_time": 5.564918041229248,
      "step": 12999
    },
    {
      "epoch": 1.2396812438964844e-05,
      "step": 12999,
      "training_step_time": 16.976917266845703
    },
    {
      "epoch": 1.239776611328125e-05,
      "grad_norm": 1.113978385925293,
      "learning_rate": 1e-05,
      "loss": 0.0617,
      "step": 13000
    },
    {
      "epoch": 1.239776611328125e-05,
      "model_forward_time": 5.619059324264526,
      "step": 13000
    },
    {
      "dynamics_loss_avg": 0.05417374633252621,
      "epoch": 1.239776611328125e-05,
      "step": 13000
    },
    {
      "action_loss_avg": 0.007624473702162504,
      "epoch": 1.239776611328125e-05,
      "step": 13000
    },
    {
      "epoch": 1.239776611328125e-05,
      "step": 13000,
      "training_step_time": 17.06873393058777
    },
    {
      "epoch": 1.2398719787597656e-05,
      "model_forward_time": 5.635536432266235,
      "step": 13001
    },
    {
      "epoch": 1.2398719787597656e-05,
      "step": 13001,
      "training_step_time": 17.03781747817993
    },
    {
      "epoch": 1.2399673461914063e-05,
      "model_forward_time": 5.631278991699219,
      "step": 13002
    },
    {
      "epoch": 1.2399673461914063e-05,
      "step": 13002,
      "training_step_time": 17.067237615585327
    },
    {
      "epoch": 1.2400627136230469e-05,
      "model_forward_time": 5.577719211578369,
      "step": 13003
    },
    {
      "epoch": 1.2400627136230469e-05,
      "step": 13003,
      "training_step_time": 17.00861120223999
    },
    {
      "epoch": 1.2401580810546875e-05,
      "model_forward_time": 5.949874401092529,
      "step": 13004
    },
    {
      "epoch": 1.2401580810546875e-05,
      "step": 13004,
      "training_step_time": 17.400206804275513
    },
    {
      "epoch": 1.2402534484863282e-05,
      "model_forward_time": 5.612138986587524,
      "step": 13005
    },
    {
      "epoch": 1.2402534484863282e-05,
      "step": 13005,
      "training_step_time": 17.10893702507019
    },
    {
      "epoch": 1.2403488159179688e-05,
      "model_forward_time": 5.626987457275391,
      "step": 13006
    },
    {
      "epoch": 1.2403488159179688e-05,
      "step": 13006,
      "training_step_time": 17.130767822265625
    },
    {
      "epoch": 1.2404441833496093e-05,
      "model_forward_time": 5.612387657165527,
      "step": 13007
    },
    {
      "epoch": 1.2404441833496093e-05,
      "step": 13007,
      "training_step_time": 17.09654402732849
    },
    {
      "epoch": 1.24053955078125e-05,
      "model_forward_time": 5.609550476074219,
      "step": 13008
    },
    {
      "epoch": 1.24053955078125e-05,
      "step": 13008,
      "training_step_time": 17.08965826034546
    },
    {
      "epoch": 1.2406349182128907e-05,
      "model_forward_time": 5.610976696014404,
      "step": 13009
    },
    {
      "epoch": 1.2406349182128907e-05,
      "step": 13009,
      "training_step_time": 17.440686225891113
    },
    {
      "epoch": 1.2407302856445312e-05,
      "grad_norm": 0.6331391334533691,
      "learning_rate": 1e-05,
      "loss": 0.0645,
      "step": 13010
    },
    {
      "epoch": 1.2407302856445312e-05,
      "model_forward_time": 5.6072564125061035,
      "step": 13010
    },
    {
      "dynamics_loss_avg": 0.06248832866549492,
      "epoch": 1.2407302856445312e-05,
      "step": 13010
    },
    {
      "action_loss_avg": 0.009140273835510016,
      "epoch": 1.2407302856445312e-05,
      "step": 13010
    },
    {
      "epoch": 1.2407302856445312e-05,
      "step": 13010,
      "training_step_time": 17.059166193008423
    },
    {
      "epoch": 1.2408256530761718e-05,
      "model_forward_time": 5.598129749298096,
      "step": 13011
    },
    {
      "epoch": 1.2408256530761718e-05,
      "step": 13011,
      "training_step_time": 17.023762226104736
    },
    {
      "epoch": 1.2409210205078125e-05,
      "model_forward_time": 5.585042238235474,
      "step": 13012
    },
    {
      "epoch": 1.2409210205078125e-05,
      "step": 13012,
      "training_step_time": 17.059139728546143
    },
    {
      "epoch": 1.2410163879394531e-05,
      "model_forward_time": 5.607269048690796,
      "step": 13013
    },
    {
      "epoch": 1.2410163879394531e-05,
      "step": 13013,
      "training_step_time": 17.082468271255493
    },
    {
      "epoch": 1.2411117553710937e-05,
      "model_forward_time": 5.609970569610596,
      "step": 13014
    },
    {
      "epoch": 1.2411117553710937e-05,
      "step": 13014,
      "training_step_time": 17.120338916778564
    },
    {
      "epoch": 1.2412071228027344e-05,
      "model_forward_time": 5.986495733261108,
      "step": 13015
    },
    {
      "epoch": 1.2412071228027344e-05,
      "step": 13015,
      "training_step_time": 17.477201461791992
    },
    {
      "epoch": 1.241302490234375e-05,
      "model_forward_time": 5.600480794906616,
      "step": 13016
    },
    {
      "epoch": 1.241302490234375e-05,
      "step": 13016,
      "training_step_time": 17.051108360290527
    },
    {
      "epoch": 1.2413978576660156e-05,
      "model_forward_time": 5.59331488609314,
      "step": 13017
    },
    {
      "epoch": 1.2413978576660156e-05,
      "step": 13017,
      "training_step_time": 17.036035299301147
    },
    {
      "epoch": 1.2414932250976563e-05,
      "model_forward_time": 5.600691080093384,
      "step": 13018
    },
    {
      "epoch": 1.2414932250976563e-05,
      "step": 13018,
      "training_step_time": 17.05204153060913
    },
    {
      "epoch": 1.2415885925292969e-05,
      "model_forward_time": 5.5989415645599365,
      "step": 13019
    },
    {
      "epoch": 1.2415885925292969e-05,
      "step": 13019,
      "training_step_time": 17.05416464805603
    },
    {
      "epoch": 1.2416839599609375e-05,
      "grad_norm": 0.9471895098686218,
      "learning_rate": 1e-05,
      "loss": 0.0676,
      "step": 13020
    },
    {
      "epoch": 1.2416839599609375e-05,
      "model_forward_time": 5.958103656768799,
      "step": 13020
    },
    {
      "dynamics_loss_avg": 0.05595017857849598,
      "epoch": 1.2416839599609375e-05,
      "step": 13020
    },
    {
      "action_loss_avg": 0.007319142832420767,
      "epoch": 1.2416839599609375e-05,
      "step": 13020
    },
    {
      "epoch": 1.2416839599609375e-05,
      "step": 13020,
      "training_step_time": 17.435916423797607
    },
    {
      "epoch": 1.2417793273925782e-05,
      "model_forward_time": 5.6052329540252686,
      "step": 13021
    },
    {
      "epoch": 1.2417793273925782e-05,
      "step": 13021,
      "training_step_time": 17.09480929374695
    },
    {
      "epoch": 1.2418746948242188e-05,
      "model_forward_time": 5.616439342498779,
      "step": 13022
    },
    {
      "epoch": 1.2418746948242188e-05,
      "step": 13022,
      "training_step_time": 17.11477780342102
    },
    {
      "epoch": 1.2419700622558593e-05,
      "model_forward_time": 5.616337776184082,
      "step": 13023
    },
    {
      "epoch": 1.2419700622558593e-05,
      "step": 13023,
      "training_step_time": 17.13387417793274
    },
    {
      "epoch": 1.2420654296875e-05,
      "model_forward_time": 5.621107339859009,
      "step": 13024
    },
    {
      "epoch": 1.2420654296875e-05,
      "step": 13024,
      "training_step_time": 17.13919162750244
    },
    {
      "epoch": 1.2421607971191407e-05,
      "model_forward_time": 5.620588541030884,
      "step": 13025
    },
    {
      "epoch": 1.2421607971191407e-05,
      "step": 13025,
      "training_step_time": 17.130841970443726
    },
    {
      "epoch": 1.2422561645507812e-05,
      "model_forward_time": 5.998631238937378,
      "step": 13026
    },
    {
      "epoch": 1.2422561645507812e-05,
      "step": 13026,
      "training_step_time": 17.488099575042725
    },
    {
      "epoch": 1.2423515319824218e-05,
      "model_forward_time": 5.617628574371338,
      "step": 13027
    },
    {
      "epoch": 1.2423515319824218e-05,
      "step": 13027,
      "training_step_time": 17.126858949661255
    },
    {
      "epoch": 1.2424468994140625e-05,
      "model_forward_time": 5.608532190322876,
      "step": 13028
    },
    {
      "epoch": 1.2424468994140625e-05,
      "step": 13028,
      "training_step_time": 17.093652486801147
    },
    {
      "epoch": 1.2425422668457031e-05,
      "model_forward_time": 5.607061147689819,
      "step": 13029
    },
    {
      "epoch": 1.2425422668457031e-05,
      "step": 13029,
      "training_step_time": 17.084269285202026
    },
    {
      "epoch": 1.2426376342773437e-05,
      "grad_norm": 0.3834614157676697,
      "learning_rate": 1e-05,
      "loss": 0.0404,
      "step": 13030
    },
    {
      "epoch": 1.2426376342773437e-05,
      "model_forward_time": 5.599704027175903,
      "step": 13030
    },
    {
      "dynamics_loss_avg": 0.03448454290628433,
      "epoch": 1.2426376342773437e-05,
      "step": 13030
    },
    {
      "action_loss_avg": 0.003478933166479692,
      "epoch": 1.2426376342773437e-05,
      "step": 13030
    },
    {
      "epoch": 1.2426376342773437e-05,
      "step": 13030,
      "training_step_time": 17.035762548446655
    },
    {
      "epoch": 1.2427330017089844e-05,
      "model_forward_time": 5.600412845611572,
      "step": 13031
    },
    {
      "epoch": 1.2427330017089844e-05,
      "step": 13031,
      "training_step_time": 17.437227725982666
    },
    {
      "epoch": 1.242828369140625e-05,
      "model_forward_time": 5.594598293304443,
      "step": 13032
    },
    {
      "epoch": 1.242828369140625e-05,
      "step": 13032,
      "training_step_time": 17.062166929244995
    },
    {
      "epoch": 1.2429237365722656e-05,
      "model_forward_time": 5.619680166244507,
      "step": 13033
    },
    {
      "epoch": 1.2429237365722656e-05,
      "step": 13033,
      "training_step_time": 17.118922233581543
    },
    {
      "epoch": 1.2430191040039063e-05,
      "model_forward_time": 5.620302200317383,
      "step": 13034
    },
    {
      "epoch": 1.2430191040039063e-05,
      "step": 13034,
      "training_step_time": 17.09413480758667
    },
    {
      "epoch": 1.2431144714355469e-05,
      "model_forward_time": 5.592945575714111,
      "step": 13035
    },
    {
      "epoch": 1.2431144714355469e-05,
      "step": 13035,
      "training_step_time": 17.008384466171265
    },
    {
      "epoch": 1.2432098388671875e-05,
      "model_forward_time": 5.565430402755737,
      "step": 13036
    },
    {
      "epoch": 1.2432098388671875e-05,
      "step": 13036,
      "training_step_time": 16.9806170463562
    },
    {
      "epoch": 1.2433052062988282e-05,
      "model_forward_time": 5.948582649230957,
      "step": 13037
    },
    {
      "epoch": 1.2433052062988282e-05,
      "step": 13037,
      "training_step_time": 17.35465359687805
    },
    {
      "epoch": 1.2434005737304688e-05,
      "model_forward_time": 5.57621955871582,
      "step": 13038
    },
    {
      "epoch": 1.2434005737304688e-05,
      "step": 13038,
      "training_step_time": 17.01947593688965
    },
    {
      "epoch": 1.2434959411621093e-05,
      "model_forward_time": 5.5721986293792725,
      "step": 13039
    },
    {
      "epoch": 1.2434959411621093e-05,
      "step": 13039,
      "training_step_time": 16.986623525619507
    },
    {
      "epoch": 1.24359130859375e-05,
      "grad_norm": 1.7446491718292236,
      "learning_rate": 1e-05,
      "loss": 0.0728,
      "step": 13040
    },
    {
      "epoch": 1.24359130859375e-05,
      "model_forward_time": 5.577470302581787,
      "step": 13040
    },
    {
      "dynamics_loss_avg": 0.062198017910122874,
      "epoch": 1.24359130859375e-05,
      "step": 13040
    },
    {
      "action_loss_avg": 0.009160166350193321,
      "epoch": 1.24359130859375e-05,
      "step": 13040
    },
    {
      "epoch": 1.24359130859375e-05,
      "step": 13040,
      "training_step_time": 16.993365049362183
    },
    {
      "epoch": 1.2436866760253907e-05,
      "model_forward_time": 5.586219072341919,
      "step": 13041
    },
    {
      "epoch": 1.2436866760253907e-05,
      "step": 13041,
      "training_step_time": 17.039820432662964
    },
    {
      "epoch": 1.2437820434570312e-05,
      "model_forward_time": 5.584611654281616,
      "step": 13042
    },
    {
      "epoch": 1.2437820434570312e-05,
      "step": 13042,
      "training_step_time": 17.407562971115112
    },
    {
      "epoch": 1.2438774108886718e-05,
      "model_forward_time": 5.603758811950684,
      "step": 13043
    },
    {
      "epoch": 1.2438774108886718e-05,
      "step": 13043,
      "training_step_time": 17.092087507247925
    },
    {
      "epoch": 1.2439727783203125e-05,
      "model_forward_time": 5.605399131774902,
      "step": 13044
    },
    {
      "epoch": 1.2439727783203125e-05,
      "step": 13044,
      "training_step_time": 17.07826519012451
    },
    {
      "epoch": 1.2440681457519531e-05,
      "model_forward_time": 5.597370624542236,
      "step": 13045
    },
    {
      "epoch": 1.2440681457519531e-05,
      "step": 13045,
      "training_step_time": 17.056128978729248
    },
    {
      "epoch": 1.2441635131835937e-05,
      "model_forward_time": 5.584460973739624,
      "step": 13046
    },
    {
      "epoch": 1.2441635131835937e-05,
      "step": 13046,
      "training_step_time": 17.009155988693237
    },
    {
      "epoch": 1.2442588806152344e-05,
      "model_forward_time": 5.585670232772827,
      "step": 13047
    },
    {
      "epoch": 1.2442588806152344e-05,
      "step": 13047,
      "training_step_time": 17.03009843826294
    },
    {
      "epoch": 1.244354248046875e-05,
      "model_forward_time": 5.977706432342529,
      "step": 13048
    },
    {
      "epoch": 1.244354248046875e-05,
      "step": 13048,
      "training_step_time": 17.45021390914917
    },
    {
      "epoch": 1.2444496154785156e-05,
      "model_forward_time": 5.620265483856201,
      "step": 13049
    },
    {
      "epoch": 1.2444496154785156e-05,
      "step": 13049,
      "training_step_time": 17.132184743881226
    },
    {
      "epoch": 1.2445449829101563e-05,
      "grad_norm": 0.9042597413063049,
      "learning_rate": 1e-05,
      "loss": 0.0459,
      "step": 13050
    },
    {
      "epoch": 1.2445449829101563e-05,
      "model_forward_time": 5.629481077194214,
      "step": 13050
    },
    {
      "dynamics_loss_avg": 0.04370062574744225,
      "epoch": 1.2445449829101563e-05,
      "step": 13050
    },
    {
      "action_loss_avg": 0.005003971268888563,
      "epoch": 1.2445449829101563e-05,
      "step": 13050
    },
    {
      "epoch": 1.2445449829101563e-05,
      "step": 13050,
      "training_step_time": 17.14711332321167
    },
    {
      "epoch": 1.2446403503417969e-05,
      "model_forward_time": 5.623858451843262,
      "step": 13051
    },
    {
      "epoch": 1.2446403503417969e-05,
      "step": 13051,
      "training_step_time": 17.122800588607788
    },
    {
      "epoch": 1.2447357177734375e-05,
      "model_forward_time": 5.639045715332031,
      "step": 13052
    },
    {
      "epoch": 1.2447357177734375e-05,
      "step": 13052,
      "training_step_time": 17.156672954559326
    },
    {
      "epoch": 1.2448310852050782e-05,
      "model_forward_time": 5.986577272415161,
      "step": 13053
    },
    {
      "epoch": 1.2448310852050782e-05,
      "step": 13053,
      "training_step_time": 17.447587728500366
    },
    {
      "epoch": 1.2449264526367188e-05,
      "model_forward_time": 5.599275588989258,
      "step": 13054
    },
    {
      "epoch": 1.2449264526367188e-05,
      "step": 13054,
      "training_step_time": 17.07537841796875
    },
    {
      "epoch": 1.2450218200683593e-05,
      "model_forward_time": 5.604573965072632,
      "step": 13055
    },
    {
      "epoch": 1.2450218200683593e-05,
      "step": 13055,
      "training_step_time": 17.047458171844482
    },
    {
      "epoch": 1.2451171875e-05,
      "model_forward_time": 5.594140291213989,
      "step": 13056
    },
    {
      "epoch": 1.2451171875e-05,
      "step": 13056,
      "training_step_time": 17.027392148971558
    },
    {
      "epoch": 1.2452125549316407e-05,
      "model_forward_time": 5.598948240280151,
      "step": 13057
    },
    {
      "epoch": 1.2452125549316407e-05,
      "step": 13057,
      "training_step_time": 17.049683332443237
    },
    {
      "epoch": 1.2453079223632812e-05,
      "model_forward_time": 5.600444793701172,
      "step": 13058
    },
    {
      "epoch": 1.2453079223632812e-05,
      "step": 13058,
      "training_step_time": 17.065832376480103
    },
    {
      "epoch": 1.2454032897949218e-05,
      "model_forward_time": 5.982769250869751,
      "step": 13059
    },
    {
      "epoch": 1.2454032897949218e-05,
      "step": 13059,
      "training_step_time": 17.41443371772766
    },
    {
      "epoch": 1.2454986572265625e-05,
      "grad_norm": 0.22308672964572906,
      "learning_rate": 1e-05,
      "loss": 0.0562,
      "step": 13060
    },
    {
      "epoch": 1.2454986572265625e-05,
      "model_forward_time": 5.590205430984497,
      "step": 13060
    },
    {
      "dynamics_loss_avg": 0.055574094876646996,
      "epoch": 1.2454986572265625e-05,
      "step": 13060
    },
    {
      "action_loss_avg": 0.006669230584520847,
      "epoch": 1.2454986572265625e-05,
      "step": 13060
    },
    {
      "epoch": 1.2454986572265625e-05,
      "step": 13060,
      "training_step_time": 17.0380916595459
    },
    {
      "epoch": 1.2455940246582031e-05,
      "model_forward_time": 5.587882995605469,
      "step": 13061
    },
    {
      "epoch": 1.2455940246582031e-05,
      "step": 13061,
      "training_step_time": 17.048221349716187
    },
    {
      "epoch": 1.2456893920898437e-05,
      "model_forward_time": 5.59878945350647,
      "step": 13062
    },
    {
      "epoch": 1.2456893920898437e-05,
      "step": 13062,
      "training_step_time": 17.06358551979065
    },
    {
      "epoch": 1.2457847595214844e-05,
      "model_forward_time": 5.602268934249878,
      "step": 13063
    },
    {
      "epoch": 1.2457847595214844e-05,
      "step": 13063,
      "training_step_time": 17.0703387260437
    },
    {
      "epoch": 1.245880126953125e-05,
      "model_forward_time": 5.601510047912598,
      "step": 13064
    },
    {
      "epoch": 1.245880126953125e-05,
      "step": 13064,
      "training_step_time": 17.415416955947876
    },
    {
      "epoch": 1.2459754943847656e-05,
      "model_forward_time": 5.5872650146484375,
      "step": 13065
    },
    {
      "epoch": 1.2459754943847656e-05,
      "step": 13065,
      "training_step_time": 17.026360034942627
    },
    {
      "epoch": 1.2460708618164063e-05,
      "model_forward_time": 5.592196226119995,
      "step": 13066
    },
    {
      "epoch": 1.2460708618164063e-05,
      "step": 13066,
      "training_step_time": 17.031795263290405
    },
    {
      "epoch": 1.2461662292480469e-05,
      "model_forward_time": 5.593514442443848,
      "step": 13067
    },
    {
      "epoch": 1.2461662292480469e-05,
      "step": 13067,
      "training_step_time": 17.05559015274048
    },
    {
      "epoch": 1.2462615966796875e-05,
      "model_forward_time": 5.597909927368164,
      "step": 13068
    },
    {
      "epoch": 1.2462615966796875e-05,
      "step": 13068,
      "training_step_time": 17.068796396255493
    },
    {
      "epoch": 1.2463569641113282e-05,
      "model_forward_time": 5.588750600814819,
      "step": 13069
    },
    {
      "epoch": 1.2463569641113282e-05,
      "step": 13069,
      "training_step_time": 17.03254795074463
    },
    {
      "epoch": 1.2464523315429688e-05,
      "grad_norm": 1.67012357711792,
      "learning_rate": 1e-05,
      "loss": 0.0643,
      "step": 13070
    },
    {
      "epoch": 1.2464523315429688e-05,
      "model_forward_time": 5.964908599853516,
      "step": 13070
    },
    {
      "dynamics_loss_avg": 0.05350648667663336,
      "epoch": 1.2464523315429688e-05,
      "step": 13070
    },
    {
      "action_loss_avg": 0.006771101453341544,
      "epoch": 1.2464523315429688e-05,
      "step": 13070
    },
    {
      "epoch": 1.2464523315429688e-05,
      "step": 13070,
      "training_step_time": 17.407588958740234
    },
    {
      "epoch": 1.2465476989746093e-05,
      "model_forward_time": 5.596472263336182,
      "step": 13071
    },
    {
      "epoch": 1.2465476989746093e-05,
      "step": 13071,
      "training_step_time": 17.091054677963257
    },
    {
      "epoch": 1.24664306640625e-05,
      "model_forward_time": 5.6204674243927,
      "step": 13072
    },
    {
      "epoch": 1.24664306640625e-05,
      "step": 13072,
      "training_step_time": 17.141406774520874
    },
    {
      "epoch": 1.2467384338378907e-05,
      "model_forward_time": 5.634340524673462,
      "step": 13073
    },
    {
      "epoch": 1.2467384338378907e-05,
      "step": 13073,
      "training_step_time": 17.129413604736328
    },
    {
      "epoch": 1.2468338012695312e-05,
      "model_forward_time": 5.608062744140625,
      "step": 13074
    },
    {
      "epoch": 1.2468338012695312e-05,
      "step": 13074,
      "training_step_time": 17.079871892929077
    },
    {
      "epoch": 1.2469291687011718e-05,
      "model_forward_time": 5.627474069595337,
      "step": 13075
    },
    {
      "epoch": 1.2469291687011718e-05,
      "step": 13075,
      "training_step_time": 17.475627422332764
    },
    {
      "epoch": 1.2470245361328125e-05,
      "model_forward_time": 5.601248264312744,
      "step": 13076
    },
    {
      "epoch": 1.2470245361328125e-05,
      "step": 13076,
      "training_step_time": 17.057862281799316
    },
    {
      "epoch": 1.2471199035644531e-05,
      "model_forward_time": 5.604387521743774,
      "step": 13077
    },
    {
      "epoch": 1.2471199035644531e-05,
      "step": 13077,
      "training_step_time": 17.047663927078247
    },
    {
      "epoch": 1.2472152709960937e-05,
      "model_forward_time": 5.581259727478027,
      "step": 13078
    },
    {
      "epoch": 1.2472152709960937e-05,
      "step": 13078,
      "training_step_time": 17.022282600402832
    },
    {
      "epoch": 1.2473106384277344e-05,
      "model_forward_time": 5.591507196426392,
      "step": 13079
    },
    {
      "epoch": 1.2473106384277344e-05,
      "step": 13079,
      "training_step_time": 17.052472591400146
    },
    {
      "epoch": 1.247406005859375e-05,
      "grad_norm": 1.1612755060195923,
      "learning_rate": 1e-05,
      "loss": 0.0619,
      "step": 13080
    },
    {
      "epoch": 1.247406005859375e-05,
      "model_forward_time": 5.602076530456543,
      "step": 13080
    },
    {
      "dynamics_loss_avg": 0.05202805288136005,
      "epoch": 1.247406005859375e-05,
      "step": 13080
    },
    {
      "action_loss_avg": 0.006516220606863498,
      "epoch": 1.247406005859375e-05,
      "step": 13080
    },
    {
      "epoch": 1.247406005859375e-05,
      "step": 13080,
      "training_step_time": 17.061060190200806
    },
    {
      "epoch": 1.2475013732910156e-05,
      "model_forward_time": 5.9680016040802,
      "step": 13081
    },
    {
      "epoch": 1.2475013732910156e-05,
      "step": 13081,
      "training_step_time": 17.45022201538086
    },
    {
      "epoch": 1.2475967407226563e-05,
      "model_forward_time": 5.598007678985596,
      "step": 13082
    },
    {
      "epoch": 1.2475967407226563e-05,
      "step": 13082,
      "training_step_time": 17.0529146194458
    },
    {
      "epoch": 1.2476921081542969e-05,
      "model_forward_time": 5.591145038604736,
      "step": 13083
    },
    {
      "epoch": 1.2476921081542969e-05,
      "step": 13083,
      "training_step_time": 17.008938312530518
    },
    {
      "epoch": 1.2477874755859375e-05,
      "model_forward_time": 5.581002235412598,
      "step": 13084
    },
    {
      "epoch": 1.2477874755859375e-05,
      "step": 13084,
      "training_step_time": 17.035231351852417
    },
    {
      "epoch": 1.2478828430175782e-05,
      "model_forward_time": 5.597636938095093,
      "step": 13085
    },
    {
      "epoch": 1.2478828430175782e-05,
      "step": 13085,
      "training_step_time": 17.054081439971924
    },
    {
      "epoch": 1.2479782104492188e-05,
      "model_forward_time": 5.981065034866333,
      "step": 13086
    },
    {
      "epoch": 1.2479782104492188e-05,
      "step": 13086,
      "training_step_time": 17.443539142608643
    },
    {
      "epoch": 1.2480735778808594e-05,
      "model_forward_time": 5.6151652336120605,
      "step": 13087
    },
    {
      "epoch": 1.2480735778808594e-05,
      "step": 13087,
      "training_step_time": 17.060450792312622
    },
    {
      "epoch": 1.2481689453125e-05,
      "model_forward_time": 5.5978474617004395,
      "step": 13088
    },
    {
      "epoch": 1.2481689453125e-05,
      "step": 13088,
      "training_step_time": 17.0531108379364
    },
    {
      "epoch": 1.2482643127441407e-05,
      "model_forward_time": 5.593122959136963,
      "step": 13089
    },
    {
      "epoch": 1.2482643127441407e-05,
      "step": 13089,
      "training_step_time": 17.005987405776978
    },
    {
      "epoch": 1.2483596801757812e-05,
      "grad_norm": 1.7579864263534546,
      "learning_rate": 1e-05,
      "loss": 0.067,
      "step": 13090
    },
    {
      "epoch": 1.2483596801757812e-05,
      "model_forward_time": 5.592970609664917,
      "step": 13090
    },
    {
      "dynamics_loss_avg": 0.060926768369972704,
      "epoch": 1.2483596801757812e-05,
      "step": 13090
    },
    {
      "action_loss_avg": 0.008292637136764824,
      "epoch": 1.2483596801757812e-05,
      "step": 13090
    },
    {
      "epoch": 1.2483596801757812e-05,
      "step": 13090,
      "training_step_time": 17.05443263053894
    },
    {
      "epoch": 1.2484550476074218e-05,
      "model_forward_time": 5.600205659866333,
      "step": 13091
    },
    {
      "epoch": 1.2484550476074218e-05,
      "step": 13091,
      "training_step_time": 17.09078550338745
    },
    {
      "epoch": 1.2485504150390626e-05,
      "model_forward_time": 5.9907567501068115,
      "step": 13092
    },
    {
      "epoch": 1.2485504150390626e-05,
      "step": 13092,
      "training_step_time": 17.51716136932373
    },
    {
      "epoch": 1.2486457824707031e-05,
      "model_forward_time": 5.612199306488037,
      "step": 13093
    },
    {
      "epoch": 1.2486457824707031e-05,
      "step": 13093,
      "training_step_time": 17.10663366317749
    },
    {
      "epoch": 1.2487411499023437e-05,
      "model_forward_time": 5.61110782623291,
      "step": 13094
    },
    {
      "epoch": 1.2487411499023437e-05,
      "step": 13094,
      "training_step_time": 17.099058866500854
    },
    {
      "epoch": 1.2488365173339844e-05,
      "model_forward_time": 5.610116481781006,
      "step": 13095
    },
    {
      "epoch": 1.2488365173339844e-05,
      "step": 13095,
      "training_step_time": 17.097848415374756
    },
    {
      "epoch": 1.248931884765625e-05,
      "model_forward_time": 5.613448619842529,
      "step": 13096
    },
    {
      "epoch": 1.248931884765625e-05,
      "step": 13096,
      "training_step_time": 17.114542722702026
    },
    {
      "epoch": 1.2490272521972656e-05,
      "model_forward_time": 5.616618871688843,
      "step": 13097
    },
    {
      "epoch": 1.2490272521972656e-05,
      "step": 13097,
      "training_step_time": 17.47133469581604
    },
    {
      "epoch": 1.2491226196289063e-05,
      "model_forward_time": 5.593854904174805,
      "step": 13098
    },
    {
      "epoch": 1.2491226196289063e-05,
      "step": 13098,
      "training_step_time": 17.04648447036743
    },
    {
      "epoch": 1.2492179870605469e-05,
      "model_forward_time": 5.589743137359619,
      "step": 13099
    },
    {
      "epoch": 1.2492179870605469e-05,
      "step": 13099,
      "training_step_time": 17.049798011779785
    },
    {
      "epoch": 1.2493133544921875e-05,
      "grad_norm": 0.6780087351799011,
      "learning_rate": 1e-05,
      "loss": 0.0696,
      "step": 13100
    },
    {
      "epoch": 1.2493133544921875e-05,
      "model_forward_time": 5.606569290161133,
      "step": 13100
    },
    {
      "dynamics_loss_avg": 0.06161263808608055,
      "epoch": 1.2493133544921875e-05,
      "step": 13100
    },
    {
      "action_loss_avg": 0.00684354524128139,
      "epoch": 1.2493133544921875e-05,
      "step": 13100
    },
    {
      "epoch": 1.2493133544921875e-05,
      "step": 13100,
      "training_step_time": 17.084840536117554
    },
    {
      "epoch": 1.2494087219238282e-05,
      "model_forward_time": 5.631436586380005,
      "step": 13101
    },
    {
      "epoch": 1.2494087219238282e-05,
      "step": 13101,
      "training_step_time": 17.08225917816162
    },
    {
      "epoch": 1.2495040893554688e-05,
      "model_forward_time": 5.585771560668945,
      "step": 13102
    },
    {
      "epoch": 1.2495040893554688e-05,
      "step": 13102,
      "training_step_time": 17.02492332458496
    },
    {
      "epoch": 1.2495994567871094e-05,
      "model_forward_time": 5.943116903305054,
      "step": 13103
    },
    {
      "epoch": 1.2495994567871094e-05,
      "step": 13103,
      "training_step_time": 17.38368034362793
    },
    {
      "epoch": 1.24969482421875e-05,
      "model_forward_time": 5.591368198394775,
      "step": 13104
    },
    {
      "epoch": 1.24969482421875e-05,
      "step": 13104,
      "training_step_time": 17.053945779800415
    },
    {
      "epoch": 1.2497901916503907e-05,
      "model_forward_time": 5.60093879699707,
      "step": 13105
    },
    {
      "epoch": 1.2497901916503907e-05,
      "step": 13105,
      "training_step_time": 17.07205367088318
    },
    {
      "epoch": 1.2498855590820312e-05,
      "model_forward_time": 5.599363565444946,
      "step": 13106
    },
    {
      "epoch": 1.2498855590820312e-05,
      "step": 13106,
      "training_step_time": 17.077352285385132
    },
    {
      "epoch": 1.2499809265136718e-05,
      "model_forward_time": 5.618361949920654,
      "step": 13107
    },
    {
      "epoch": 1.2499809265136718e-05,
      "step": 13107,
      "training_step_time": 17.103456497192383
    },
    {
      "epoch": 1.2500762939453126e-05,
      "model_forward_time": 5.612075328826904,
      "step": 13108
    },
    {
      "epoch": 1.2500762939453126e-05,
      "step": 13108,
      "training_step_time": 17.46252703666687
    },
    {
      "epoch": 1.2501716613769531e-05,
      "model_forward_time": 5.606011390686035,
      "step": 13109
    },
    {
      "epoch": 1.2501716613769531e-05,
      "step": 13109,
      "training_step_time": 17.07834529876709
    },
    {
      "epoch": 1.2502670288085937e-05,
      "grad_norm": 1.2374519109725952,
      "learning_rate": 1e-05,
      "loss": 0.0611,
      "step": 13110
    },
    {
      "epoch": 1.2502670288085937e-05,
      "model_forward_time": 5.593966007232666,
      "step": 13110
    },
    {
      "dynamics_loss_avg": 0.05530136805027723,
      "epoch": 1.2502670288085937e-05,
      "step": 13110
    },
    {
      "action_loss_avg": 0.00644436851143837,
      "epoch": 1.2502670288085937e-05,
      "step": 13110
    },
    {
      "epoch": 1.2502670288085937e-05,
      "step": 13110,
      "training_step_time": 17.025962352752686
    },
    {
      "epoch": 1.2503623962402344e-05,
      "model_forward_time": 5.604719877243042,
      "step": 13111
    },
    {
      "epoch": 1.2503623962402344e-05,
      "step": 13111,
      "training_step_time": 17.093626737594604
    },
    {
      "epoch": 1.250457763671875e-05,
      "model_forward_time": 5.607554912567139,
      "step": 13112
    },
    {
      "epoch": 1.250457763671875e-05,
      "step": 13112,
      "training_step_time": 17.123329401016235
    },
    {
      "epoch": 1.2505531311035156e-05,
      "model_forward_time": 5.608785152435303,
      "step": 13113
    },
    {
      "epoch": 1.2505531311035156e-05,
      "step": 13113,
      "training_step_time": 17.079946517944336
    },
    {
      "epoch": 1.2506484985351563e-05,
      "model_forward_time": 5.9753687381744385,
      "step": 13114
    },
    {
      "epoch": 1.2506484985351563e-05,
      "step": 13114,
      "training_step_time": 17.415292501449585
    },
    {
      "epoch": 1.2507438659667969e-05,
      "model_forward_time": 5.578043460845947,
      "step": 13115
    },
    {
      "epoch": 1.2507438659667969e-05,
      "step": 13115,
      "training_step_time": 17.008186101913452
    },
    {
      "epoch": 1.2508392333984375e-05,
      "model_forward_time": 5.578283786773682,
      "step": 13116
    },
    {
      "epoch": 1.2508392333984375e-05,
      "step": 13116,
      "training_step_time": 16.99771785736084
    },
    {
      "epoch": 1.2509346008300782e-05,
      "model_forward_time": 5.5940186977386475,
      "step": 13117
    },
    {
      "epoch": 1.2509346008300782e-05,
      "step": 13117,
      "training_step_time": 17.039175033569336
    },
    {
      "epoch": 1.2510299682617188e-05,
      "model_forward_time": 5.607522249221802,
      "step": 13118
    },
    {
      "epoch": 1.2510299682617188e-05,
      "step": 13118,
      "training_step_time": 17.080628871917725
    },
    {
      "epoch": 1.2511253356933594e-05,
      "model_forward_time": 5.983551263809204,
      "step": 13119
    },
    {
      "epoch": 1.2511253356933594e-05,
      "step": 13119,
      "training_step_time": 17.471577167510986
    },
    {
      "epoch": 1.251220703125e-05,
      "grad_norm": 0.6194005012512207,
      "learning_rate": 1e-05,
      "loss": 0.0686,
      "step": 13120
    },
    {
      "epoch": 1.251220703125e-05,
      "model_forward_time": 5.602294921875,
      "step": 13120
    },
    {
      "dynamics_loss_avg": 0.06470889672636986,
      "epoch": 1.251220703125e-05,
      "step": 13120
    },
    {
      "action_loss_avg": 0.006628830684348941,
      "epoch": 1.251220703125e-05,
      "step": 13120
    },
    {
      "epoch": 1.251220703125e-05,
      "step": 13120,
      "training_step_time": 17.060745000839233
    },
    {
      "epoch": 1.2513160705566407e-05,
      "model_forward_time": 5.592199087142944,
      "step": 13121
    },
    {
      "epoch": 1.2513160705566407e-05,
      "step": 13121,
      "training_step_time": 17.043753147125244
    },
    {
      "epoch": 1.2514114379882812e-05,
      "model_forward_time": 5.589746475219727,
      "step": 13122
    },
    {
      "epoch": 1.2514114379882812e-05,
      "step": 13122,
      "training_step_time": 17.02161407470703
    },
    {
      "epoch": 1.2515068054199218e-05,
      "model_forward_time": 5.586854934692383,
      "step": 13123
    },
    {
      "epoch": 1.2515068054199218e-05,
      "step": 13123,
      "training_step_time": 17.04529905319214
    },
    {
      "epoch": 1.2516021728515626e-05,
      "model_forward_time": 4.273570775985718,
      "step": 13124
    },
    {
      "epoch": 1.2516021728515626e-05,
      "step": 13124,
      "training_step_time": 12.802753210067749
    },
    {
      "epoch": 1.2516975402832031e-05,
      "model_forward_time": 5.946669816970825,
      "step": 13125
    },
    {
      "epoch": 1.2516975402832031e-05,
      "step": 13125,
      "training_step_time": 17.408573150634766
    },
    {
      "epoch": 1.2517929077148437e-05,
      "model_forward_time": 5.5938544273376465,
      "step": 13126
    },
    {
      "epoch": 1.2517929077148437e-05,
      "step": 13126,
      "training_step_time": 17.039242267608643
    },
    {
      "epoch": 1.2518882751464844e-05,
      "model_forward_time": 5.587301254272461,
      "step": 13127
    },
    {
      "epoch": 1.2518882751464844e-05,
      "step": 13127,
      "training_step_time": 17.052897691726685
    },
    {
      "epoch": 1.251983642578125e-05,
      "model_forward_time": 5.600750684738159,
      "step": 13128
    },
    {
      "epoch": 1.251983642578125e-05,
      "step": 13128,
      "training_step_time": 17.066529035568237
    },
    {
      "epoch": 1.2520790100097656e-05,
      "model_forward_time": 5.598779201507568,
      "step": 13129
    },
    {
      "epoch": 1.2520790100097656e-05,
      "step": 13129,
      "training_step_time": 17.036535263061523
    },
    {
      "epoch": 1.2521743774414063e-05,
      "grad_norm": 1.0730165243148804,
      "learning_rate": 1e-05,
      "loss": 0.0609,
      "step": 13130
    },
    {
      "epoch": 1.2521743774414063e-05,
      "model_forward_time": 5.587953567504883,
      "step": 13130
    },
    {
      "dynamics_loss_avg": 0.048374610207974914,
      "epoch": 1.2521743774414063e-05,
      "step": 13130
    },
    {
      "action_loss_avg": 0.006240933132357895,
      "epoch": 1.2521743774414063e-05,
      "step": 13130
    },
    {
      "epoch": 1.2521743774414063e-05,
      "step": 13130,
      "training_step_time": 17.41357731819153
    },
    {
      "epoch": 1.2522697448730469e-05,
      "model_forward_time": 5.6043877601623535,
      "step": 13131
    },
    {
      "epoch": 1.2522697448730469e-05,
      "step": 13131,
      "training_step_time": 17.090284824371338
    },
    {
      "epoch": 1.2523651123046875e-05,
      "model_forward_time": 5.606889009475708,
      "step": 13132
    },
    {
      "epoch": 1.2523651123046875e-05,
      "step": 13132,
      "training_step_time": 17.101907968521118
    },
    {
      "epoch": 1.252460479736328e-05,
      "model_forward_time": 5.611218452453613,
      "step": 13133
    },
    {
      "epoch": 1.252460479736328e-05,
      "step": 13133,
      "training_step_time": 17.09631109237671
    },
    {
      "epoch": 1.2525558471679688e-05,
      "model_forward_time": 5.6186347007751465,
      "step": 13134
    },
    {
      "epoch": 1.2525558471679688e-05,
      "step": 13134,
      "training_step_time": 17.10682773590088
    },
    {
      "epoch": 1.2526512145996094e-05,
      "model_forward_time": 5.611905336380005,
      "step": 13135
    },
    {
      "epoch": 1.2526512145996094e-05,
      "step": 13135,
      "training_step_time": 17.107219219207764
    },
    {
      "epoch": 1.25274658203125e-05,
      "model_forward_time": 5.957935571670532,
      "step": 13136
    },
    {
      "epoch": 1.25274658203125e-05,
      "step": 13136,
      "training_step_time": 17.447162628173828
    },
    {
      "epoch": 1.2528419494628907e-05,
      "model_forward_time": 5.610421657562256,
      "step": 13137
    },
    {
      "epoch": 1.2528419494628907e-05,
      "step": 13137,
      "training_step_time": 17.10502052307129
    },
    {
      "epoch": 1.2529373168945312e-05,
      "model_forward_time": 5.612236499786377,
      "step": 13138
    },
    {
      "epoch": 1.2529373168945312e-05,
      "step": 13138,
      "training_step_time": 17.103534936904907
    },
    {
      "epoch": 1.2530326843261718e-05,
      "model_forward_time": 5.605725049972534,
      "step": 13139
    },
    {
      "epoch": 1.2530326843261718e-05,
      "step": 13139,
      "training_step_time": 17.06471562385559
    },
    {
      "epoch": 1.2531280517578126e-05,
      "grad_norm": 0.6036927700042725,
      "learning_rate": 1e-05,
      "loss": 0.0459,
      "step": 13140
    },
    {
      "epoch": 1.2531280517578126e-05,
      "model_forward_time": 5.590373277664185,
      "step": 13140
    },
    {
      "dynamics_loss_avg": 0.04515186343342066,
      "epoch": 1.2531280517578126e-05,
      "step": 13140
    },
    {
      "action_loss_avg": 0.007365607644896954,
      "epoch": 1.2531280517578126e-05,
      "step": 13140
    },
    {
      "epoch": 1.2531280517578126e-05,
      "step": 13140,
      "training_step_time": 17.023231267929077
    },
    {
      "epoch": 1.2532234191894531e-05,
      "model_forward_time": 5.9720563888549805,
      "step": 13141
    },
    {
      "epoch": 1.2532234191894531e-05,
      "step": 13141,
      "training_step_time": 17.4047372341156
    },
    {
      "epoch": 1.2533187866210937e-05,
      "model_forward_time": 5.582250356674194,
      "step": 13142
    },
    {
      "epoch": 1.2533187866210937e-05,
      "step": 13142,
      "training_step_time": 17.036404848098755
    },
    {
      "epoch": 1.2534141540527344e-05,
      "model_forward_time": 5.5961034297943115,
      "step": 13143
    },
    {
      "epoch": 1.2534141540527344e-05,
      "step": 13143,
      "training_step_time": 17.066417455673218
    },
    {
      "epoch": 1.253509521484375e-05,
      "model_forward_time": 5.608491659164429,
      "step": 13144
    },
    {
      "epoch": 1.253509521484375e-05,
      "step": 13144,
      "training_step_time": 17.09983229637146
    },
    {
      "epoch": 1.2536048889160156e-05,
      "model_forward_time": 5.617909669876099,
      "step": 13145
    },
    {
      "epoch": 1.2536048889160156e-05,
      "step": 13145,
      "training_step_time": 17.12516188621521
    },
    {
      "epoch": 1.2537002563476563e-05,
      "model_forward_time": 5.637133598327637,
      "step": 13146
    },
    {
      "epoch": 1.2537002563476563e-05,
      "step": 13146,
      "training_step_time": 17.160334825515747
    },
    {
      "epoch": 1.2537956237792969e-05,
      "model_forward_time": 5.98024845123291,
      "step": 13147
    },
    {
      "epoch": 1.2537956237792969e-05,
      "step": 13147,
      "training_step_time": 17.462627172470093
    },
    {
      "epoch": 1.2538909912109375e-05,
      "model_forward_time": 5.6085288524627686,
      "step": 13148
    },
    {
      "epoch": 1.2538909912109375e-05,
      "step": 13148,
      "training_step_time": 17.064361333847046
    },
    {
      "epoch": 1.253986358642578e-05,
      "model_forward_time": 5.576290130615234,
      "step": 13149
    },
    {
      "epoch": 1.253986358642578e-05,
      "step": 13149,
      "training_step_time": 16.9887797832489
    },
    {
      "epoch": 1.2540817260742188e-05,
      "grad_norm": 0.4841981828212738,
      "learning_rate": 1e-05,
      "loss": 0.067,
      "step": 13150
    },
    {
      "epoch": 1.2540817260742188e-05,
      "model_forward_time": 5.585006952285767,
      "step": 13150
    },
    {
      "dynamics_loss_avg": 0.054196385107934474,
      "epoch": 1.2540817260742188e-05,
      "step": 13150
    },
    {
      "action_loss_avg": 0.007640158291906118,
      "epoch": 1.2540817260742188e-05,
      "step": 13150
    },
    {
      "epoch": 1.2540817260742188e-05,
      "step": 13150,
      "training_step_time": 16.99620819091797
    },
    {
      "epoch": 1.2541770935058594e-05,
      "model_forward_time": 5.5859527587890625,
      "step": 13151
    },
    {
      "epoch": 1.2541770935058594e-05,
      "step": 13151,
      "training_step_time": 17.01161003112793
    },
    {
      "epoch": 1.2542724609375e-05,
      "model_forward_time": 5.968888521194458,
      "step": 13152
    },
    {
      "epoch": 1.2542724609375e-05,
      "step": 13152,
      "training_step_time": 17.442007064819336
    },
    {
      "epoch": 1.2543678283691407e-05,
      "model_forward_time": 5.617286443710327,
      "step": 13153
    },
    {
      "epoch": 1.2543678283691407e-05,
      "step": 13153,
      "training_step_time": 17.14503264427185
    },
    {
      "epoch": 1.2544631958007812e-05,
      "model_forward_time": 5.631405830383301,
      "step": 13154
    },
    {
      "epoch": 1.2544631958007812e-05,
      "step": 13154,
      "training_step_time": 17.152039051055908
    },
    {
      "epoch": 1.2545585632324218e-05,
      "model_forward_time": 5.635281324386597,
      "step": 13155
    },
    {
      "epoch": 1.2545585632324218e-05,
      "step": 13155,
      "training_step_time": 17.124621152877808
    },
    {
      "epoch": 1.2546539306640626e-05,
      "model_forward_time": 5.592799186706543,
      "step": 13156
    },
    {
      "epoch": 1.2546539306640626e-05,
      "step": 13156,
      "training_step_time": 17.040826082229614
    },
    {
      "epoch": 1.2547492980957031e-05,
      "model_forward_time": 5.588357448577881,
      "step": 13157
    },
    {
      "epoch": 1.2547492980957031e-05,
      "step": 13157,
      "training_step_time": 17.01782512664795
    },
    {
      "epoch": 1.2548446655273437e-05,
      "model_forward_time": 5.94205117225647,
      "step": 13158
    },
    {
      "epoch": 1.2548446655273437e-05,
      "step": 13158,
      "training_step_time": 17.391130208969116
    },
    {
      "epoch": 1.2549400329589844e-05,
      "model_forward_time": 5.593372821807861,
      "step": 13159
    },
    {
      "epoch": 1.2549400329589844e-05,
      "step": 13159,
      "training_step_time": 17.043540000915527
    },
    {
      "epoch": 1.255035400390625e-05,
      "grad_norm": 0.4877711832523346,
      "learning_rate": 1e-05,
      "loss": 0.0529,
      "step": 13160
    },
    {
      "epoch": 1.255035400390625e-05,
      "model_forward_time": 5.597237825393677,
      "step": 13160
    },
    {
      "dynamics_loss_avg": 0.0482065673917532,
      "epoch": 1.255035400390625e-05,
      "step": 13160
    },
    {
      "action_loss_avg": 0.005077166319824755,
      "epoch": 1.255035400390625e-05,
      "step": 13160
    },
    {
      "epoch": 1.255035400390625e-05,
      "step": 13160,
      "training_step_time": 17.05510902404785
    },
    {
      "epoch": 1.2551307678222656e-05,
      "model_forward_time": 5.603115558624268,
      "step": 13161
    },
    {
      "epoch": 1.2551307678222656e-05,
      "step": 13161,
      "training_step_time": 17.07534408569336
    },
    {
      "epoch": 1.2552261352539063e-05,
      "model_forward_time": 5.601636171340942,
      "step": 13162
    },
    {
      "epoch": 1.2552261352539063e-05,
      "step": 13162,
      "training_step_time": 17.06986713409424
    },
    {
      "epoch": 1.2553215026855469e-05,
      "model_forward_time": 5.6041553020477295,
      "step": 13163
    },
    {
      "epoch": 1.2553215026855469e-05,
      "step": 13163,
      "training_step_time": 17.4120876789093
    },
    {
      "epoch": 1.2554168701171875e-05,
      "model_forward_time": 5.586796998977661,
      "step": 13164
    },
    {
      "epoch": 1.2554168701171875e-05,
      "step": 13164,
      "training_step_time": 17.049837827682495
    },
    {
      "epoch": 1.255512237548828e-05,
      "model_forward_time": 5.605373859405518,
      "step": 13165
    },
    {
      "epoch": 1.255512237548828e-05,
      "step": 13165,
      "training_step_time": 17.072882413864136
    },
    {
      "epoch": 1.2556076049804688e-05,
      "model_forward_time": 5.607518434524536,
      "step": 13166
    },
    {
      "epoch": 1.2556076049804688e-05,
      "step": 13166,
      "training_step_time": 17.099390029907227
    },
    {
      "epoch": 1.2557029724121094e-05,
      "model_forward_time": 5.603738069534302,
      "step": 13167
    },
    {
      "epoch": 1.2557029724121094e-05,
      "step": 13167,
      "training_step_time": 17.100643157958984
    },
    {
      "epoch": 1.25579833984375e-05,
      "model_forward_time": 5.629865884780884,
      "step": 13168
    },
    {
      "epoch": 1.25579833984375e-05,
      "step": 13168,
      "training_step_time": 17.136768579483032
    },
    {
      "epoch": 1.2558937072753907e-05,
      "model_forward_time": 5.976532936096191,
      "step": 13169
    },
    {
      "epoch": 1.2558937072753907e-05,
      "step": 13169,
      "training_step_time": 17.488993406295776
    },
    {
      "epoch": 1.2559890747070312e-05,
      "grad_norm": 0.48333871364593506,
      "learning_rate": 1e-05,
      "loss": 0.0755,
      "step": 13170
    },
    {
      "epoch": 1.2559890747070312e-05,
      "model_forward_time": 5.63288688659668,
      "step": 13170
    },
    {
      "dynamics_loss_avg": 0.07241702452301979,
      "epoch": 1.2559890747070312e-05,
      "step": 13170
    },
    {
      "action_loss_avg": 0.008607255015522241,
      "epoch": 1.2559890747070312e-05,
      "step": 13170
    },
    {
      "epoch": 1.2559890747070312e-05,
      "step": 13170,
      "training_step_time": 17.133482217788696
    },
    {
      "epoch": 1.2560844421386718e-05,
      "model_forward_time": 5.6453025341033936,
      "step": 13171
    },
    {
      "epoch": 1.2560844421386718e-05,
      "step": 13171,
      "training_step_time": 17.128470420837402
    },
    {
      "epoch": 1.2561798095703126e-05,
      "model_forward_time": 5.6223533153533936,
      "step": 13172
    },
    {
      "epoch": 1.2561798095703126e-05,
      "step": 13172,
      "training_step_time": 17.11745810508728
    },
    {
      "epoch": 1.2562751770019531e-05,
      "model_forward_time": 5.6117472648620605,
      "step": 13173
    },
    {
      "epoch": 1.2562751770019531e-05,
      "step": 13173,
      "training_step_time": 17.105059146881104
    },
    {
      "epoch": 1.2563705444335937e-05,
      "model_forward_time": 5.998339653015137,
      "step": 13174
    },
    {
      "epoch": 1.2563705444335937e-05,
      "step": 13174,
      "training_step_time": 17.423237085342407
    },
    {
      "epoch": 1.2564659118652344e-05,
      "model_forward_time": 5.6069910526275635,
      "step": 13175
    },
    {
      "epoch": 1.2564659118652344e-05,
      "step": 13175,
      "training_step_time": 17.081559658050537
    },
    {
      "epoch": 1.256561279296875e-05,
      "model_forward_time": 5.5984649658203125,
      "step": 13176
    },
    {
      "epoch": 1.256561279296875e-05,
      "step": 13176,
      "training_step_time": 17.06926727294922
    },
    {
      "epoch": 1.2566566467285156e-05,
      "model_forward_time": 5.599175453186035,
      "step": 13177
    },
    {
      "epoch": 1.2566566467285156e-05,
      "step": 13177,
      "training_step_time": 17.069957494735718
    },
    {
      "epoch": 1.2567520141601563e-05,
      "model_forward_time": 5.6151282787323,
      "step": 13178
    },
    {
      "epoch": 1.2567520141601563e-05,
      "step": 13178,
      "training_step_time": 17.035194158554077
    },
    {
      "epoch": 1.2568473815917969e-05,
      "model_forward_time": 5.611856460571289,
      "step": 13179
    },
    {
      "epoch": 1.2568473815917969e-05,
      "step": 13179,
      "training_step_time": 17.05492329597473
    },
    {
      "epoch": 1.2569427490234375e-05,
      "grad_norm": 0.6632546186447144,
      "learning_rate": 1e-05,
      "loss": 0.0882,
      "step": 13180
    },
    {
      "epoch": 1.2569427490234375e-05,
      "model_forward_time": 5.945946931838989,
      "step": 13180
    },
    {
      "dynamics_loss_avg": 0.07498965561389923,
      "epoch": 1.2569427490234375e-05,
      "step": 13180
    },
    {
      "action_loss_avg": 0.00917989290319383,
      "epoch": 1.2569427490234375e-05,
      "step": 13180
    },
    {
      "epoch": 1.2569427490234375e-05,
      "step": 13180,
      "training_step_time": 17.42005157470703
    },
    {
      "epoch": 1.257038116455078e-05,
      "model_forward_time": 5.627588272094727,
      "step": 13181
    },
    {
      "epoch": 1.257038116455078e-05,
      "step": 13181,
      "training_step_time": 17.150847673416138
    },
    {
      "epoch": 1.2571334838867188e-05,
      "model_forward_time": 5.634955406188965,
      "step": 13182
    },
    {
      "epoch": 1.2571334838867188e-05,
      "step": 13182,
      "training_step_time": 17.152337551116943
    },
    {
      "epoch": 1.2572288513183594e-05,
      "model_forward_time": 5.620725154876709,
      "step": 13183
    },
    {
      "epoch": 1.2572288513183594e-05,
      "step": 13183,
      "training_step_time": 17.097574472427368
    },
    {
      "epoch": 1.25732421875e-05,
      "model_forward_time": 5.584331274032593,
      "step": 13184
    },
    {
      "epoch": 1.25732421875e-05,
      "step": 13184,
      "training_step_time": 17.016311168670654
    },
    {
      "epoch": 1.2574195861816407e-05,
      "model_forward_time": 5.983365774154663,
      "step": 13185
    },
    {
      "epoch": 1.2574195861816407e-05,
      "step": 13185,
      "training_step_time": 17.420318841934204
    },
    {
      "epoch": 1.2575149536132812e-05,
      "model_forward_time": 5.595310926437378,
      "step": 13186
    },
    {
      "epoch": 1.2575149536132812e-05,
      "step": 13186,
      "training_step_time": 17.051135778427124
    },
    {
      "epoch": 1.2576103210449218e-05,
      "model_forward_time": 5.6044816970825195,
      "step": 13187
    },
    {
      "epoch": 1.2576103210449218e-05,
      "step": 13187,
      "training_step_time": 17.11041021347046
    },
    {
      "epoch": 1.2577056884765626e-05,
      "model_forward_time": 5.615221977233887,
      "step": 13188
    },
    {
      "epoch": 1.2577056884765626e-05,
      "step": 13188,
      "training_step_time": 17.085225105285645
    },
    {
      "epoch": 1.2578010559082031e-05,
      "model_forward_time": 5.5954999923706055,
      "step": 13189
    },
    {
      "epoch": 1.2578010559082031e-05,
      "step": 13189,
      "training_step_time": 17.011578798294067
    },
    {
      "epoch": 1.2578964233398437e-05,
      "grad_norm": 0.3599824607372284,
      "learning_rate": 1e-05,
      "loss": 0.067,
      "step": 13190
    },
    {
      "epoch": 1.2578964233398437e-05,
      "model_forward_time": 5.571812868118286,
      "step": 13190
    },
    {
      "dynamics_loss_avg": 0.05912703480571509,
      "epoch": 1.2578964233398437e-05,
      "step": 13190
    },
    {
      "action_loss_avg": 0.006850800546817482,
      "epoch": 1.2578964233398437e-05,
      "step": 13190
    },
    {
      "epoch": 1.2578964233398437e-05,
      "step": 13190,
      "training_step_time": 17.002518892288208
    },
    {
      "epoch": 1.2579917907714844e-05,
      "model_forward_time": 5.926266670227051,
      "step": 13191
    },
    {
      "epoch": 1.2579917907714844e-05,
      "step": 13191,
      "training_step_time": 17.351690769195557
    },
    {
      "epoch": 1.258087158203125e-05,
      "model_forward_time": 5.578866481781006,
      "step": 13192
    },
    {
      "epoch": 1.258087158203125e-05,
      "step": 13192,
      "training_step_time": 16.98112392425537
    },
    {
      "epoch": 1.2581825256347656e-05,
      "model_forward_time": 5.588660717010498,
      "step": 13193
    },
    {
      "epoch": 1.2581825256347656e-05,
      "step": 13193,
      "training_step_time": 17.017778396606445
    },
    {
      "epoch": 1.2582778930664063e-05,
      "model_forward_time": 5.579231023788452,
      "step": 13194
    },
    {
      "epoch": 1.2582778930664063e-05,
      "step": 13194,
      "training_step_time": 17.0272159576416
    },
    {
      "epoch": 1.2583732604980469e-05,
      "model_forward_time": 5.589611768722534,
      "step": 13195
    },
    {
      "epoch": 1.2583732604980469e-05,
      "step": 13195,
      "training_step_time": 17.064038038253784
    },
    {
      "epoch": 1.2584686279296875e-05,
      "model_forward_time": 5.6214399337768555,
      "step": 13196
    },
    {
      "epoch": 1.2584686279296875e-05,
      "step": 13196,
      "training_step_time": 17.495768785476685
    },
    {
      "epoch": 1.258563995361328e-05,
      "model_forward_time": 5.622080564498901,
      "step": 13197
    },
    {
      "epoch": 1.258563995361328e-05,
      "step": 13197,
      "training_step_time": 17.106782913208008
    },
    {
      "epoch": 1.2586593627929688e-05,
      "model_forward_time": 5.595737457275391,
      "step": 13198
    },
    {
      "epoch": 1.2586593627929688e-05,
      "step": 13198,
      "training_step_time": 17.03476047515869
    },
    {
      "epoch": 1.2587547302246094e-05,
      "model_forward_time": 5.570353984832764,
      "step": 13199
    },
    {
      "epoch": 1.2587547302246094e-05,
      "step": 13199,
      "training_step_time": 16.998836040496826
    },
    {
      "epoch": 1.25885009765625e-05,
      "grad_norm": 0.3094647526741028,
      "learning_rate": 1e-05,
      "loss": 0.0612,
      "step": 13200
    },
    {
      "epoch": 1.25885009765625e-05,
      "model_forward_time": 5.5814104080200195,
      "step": 13200
    },
    {
      "dynamics_loss_avg": 0.05335055012255907,
      "epoch": 1.25885009765625e-05,
      "step": 13200
    },
    {
      "action_loss_avg": 0.006513238628394901,
      "epoch": 1.25885009765625e-05,
      "step": 13200
    },
    {
      "epoch": 1.25885009765625e-05,
      "step": 13200,
      "training_step_time": 16.99282717704773
    },
    {
      "epoch": 1.2589454650878907e-05,
      "model_forward_time": 5.571962356567383,
      "step": 13201
    },
    {
      "epoch": 1.2589454650878907e-05,
      "step": 13201,
      "training_step_time": 16.993590831756592
    },
    {
      "epoch": 1.2590408325195312e-05,
      "model_forward_time": 5.926976203918457,
      "step": 13202
    },
    {
      "epoch": 1.2590408325195312e-05,
      "step": 13202,
      "training_step_time": 17.356379985809326
    },
    {
      "epoch": 1.2591361999511718e-05,
      "model_forward_time": 5.6141016483306885,
      "step": 13203
    },
    {
      "epoch": 1.2591361999511718e-05,
      "step": 13203,
      "training_step_time": 17.039766788482666
    },
    {
      "epoch": 1.2592315673828126e-05,
      "model_forward_time": 5.5964696407318115,
      "step": 13204
    },
    {
      "epoch": 1.2592315673828126e-05,
      "step": 13204,
      "training_step_time": 17.055201292037964
    },
    {
      "epoch": 1.2593269348144531e-05,
      "model_forward_time": 5.608311414718628,
      "step": 13205
    },
    {
      "epoch": 1.2593269348144531e-05,
      "step": 13205,
      "training_step_time": 17.080633640289307
    },
    {
      "epoch": 1.2594223022460937e-05,
      "model_forward_time": 5.612446546554565,
      "step": 13206
    },
    {
      "epoch": 1.2594223022460937e-05,
      "step": 13206,
      "training_step_time": 17.11026382446289
    },
    {
      "epoch": 1.2595176696777344e-05,
      "model_forward_time": 6.00725531578064,
      "step": 13207
    },
    {
      "epoch": 1.2595176696777344e-05,
      "step": 13207,
      "training_step_time": 17.489438772201538
    },
    {
      "epoch": 1.259613037109375e-05,
      "model_forward_time": 5.618067264556885,
      "step": 13208
    },
    {
      "epoch": 1.259613037109375e-05,
      "step": 13208,
      "training_step_time": 17.09207820892334
    },
    {
      "epoch": 1.2597084045410156e-05,
      "model_forward_time": 5.614228248596191,
      "step": 13209
    },
    {
      "epoch": 1.2597084045410156e-05,
      "step": 13209,
      "training_step_time": 17.09465456008911
    },
    {
      "epoch": 1.2598037719726563e-05,
      "grad_norm": 1.097699522972107,
      "learning_rate": 1e-05,
      "loss": 0.0665,
      "step": 13210
    },
    {
      "epoch": 1.2598037719726563e-05,
      "model_forward_time": 5.6213884353637695,
      "step": 13210
    },
    {
      "dynamics_loss_avg": 0.05835138931870461,
      "epoch": 1.2598037719726563e-05,
      "step": 13210
    },
    {
      "action_loss_avg": 0.007629966968670488,
      "epoch": 1.2598037719726563e-05,
      "step": 13210
    },
    {
      "epoch": 1.2598037719726563e-05,
      "step": 13210,
      "training_step_time": 17.126067399978638
    },
    {
      "epoch": 1.2598991394042969e-05,
      "model_forward_time": 5.609756946563721,
      "step": 13211
    },
    {
      "epoch": 1.2598991394042969e-05,
      "step": 13211,
      "training_step_time": 17.086254119873047
    },
    {
      "epoch": 1.2599945068359375e-05,
      "model_forward_time": 5.603595972061157,
      "step": 13212
    },
    {
      "epoch": 1.2599945068359375e-05,
      "step": 13212,
      "training_step_time": 17.069810152053833
    },
    {
      "epoch": 1.260089874267578e-05,
      "model_forward_time": 5.946230888366699,
      "step": 13213
    },
    {
      "epoch": 1.260089874267578e-05,
      "step": 13213,
      "training_step_time": 17.39582633972168
    },
    {
      "epoch": 1.2601852416992188e-05,
      "model_forward_time": 5.625420570373535,
      "step": 13214
    },
    {
      "epoch": 1.2601852416992188e-05,
      "step": 13214,
      "training_step_time": 17.093499422073364
    },
    {
      "epoch": 1.2602806091308594e-05,
      "model_forward_time": 5.604179859161377,
      "step": 13215
    },
    {
      "epoch": 1.2602806091308594e-05,
      "step": 13215,
      "training_step_time": 17.07830047607422
    },
    {
      "epoch": 1.2603759765625e-05,
      "model_forward_time": 5.641739845275879,
      "step": 13216
    },
    {
      "epoch": 1.2603759765625e-05,
      "step": 13216,
      "training_step_time": 17.098799228668213
    },
    {
      "epoch": 1.2604713439941407e-05,
      "model_forward_time": 5.630350351333618,
      "step": 13217
    },
    {
      "epoch": 1.2604713439941407e-05,
      "step": 13217,
      "training_step_time": 17.108724117279053
    },
    {
      "epoch": 1.2605667114257812e-05,
      "model_forward_time": 5.9776387214660645,
      "step": 13218
    },
    {
      "epoch": 1.2605667114257812e-05,
      "step": 13218,
      "training_step_time": 17.423035860061646
    },
    {
      "epoch": 1.2606620788574218e-05,
      "model_forward_time": 5.63334059715271,
      "step": 13219
    },
    {
      "epoch": 1.2606620788574218e-05,
      "step": 13219,
      "training_step_time": 17.153891801834106
    },
    {
      "epoch": 1.2607574462890626e-05,
      "grad_norm": 0.5047913193702698,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 13220
    },
    {
      "epoch": 1.2607574462890626e-05,
      "model_forward_time": 5.628155708312988,
      "step": 13220
    },
    {
      "dynamics_loss_avg": 0.0626819035038352,
      "epoch": 1.2607574462890626e-05,
      "step": 13220
    },
    {
      "action_loss_avg": 0.007131344568915665,
      "epoch": 1.2607574462890626e-05,
      "step": 13220
    },
    {
      "epoch": 1.2607574462890626e-05,
      "step": 13220,
      "training_step_time": 17.142332792282104
    },
    {
      "epoch": 1.2608528137207031e-05,
      "model_forward_time": 5.6232805252075195,
      "step": 13221
    },
    {
      "epoch": 1.2608528137207031e-05,
      "step": 13221,
      "training_step_time": 17.119093418121338
    },
    {
      "epoch": 1.2609481811523437e-05,
      "model_forward_time": 5.612987995147705,
      "step": 13222
    },
    {
      "epoch": 1.2609481811523437e-05,
      "step": 13222,
      "training_step_time": 17.11102533340454
    },
    {
      "epoch": 1.2610435485839844e-05,
      "model_forward_time": 5.602194786071777,
      "step": 13223
    },
    {
      "epoch": 1.2610435485839844e-05,
      "step": 13223,
      "training_step_time": 17.084270238876343
    },
    {
      "epoch": 1.261138916015625e-05,
      "model_forward_time": 5.9587647914886475,
      "step": 13224
    },
    {
      "epoch": 1.261138916015625e-05,
      "step": 13224,
      "training_step_time": 17.448543071746826
    },
    {
      "epoch": 1.2612342834472656e-05,
      "model_forward_time": 5.616567611694336,
      "step": 13225
    },
    {
      "epoch": 1.2612342834472656e-05,
      "step": 13225,
      "training_step_time": 17.114569902420044
    },
    {
      "epoch": 1.2613296508789063e-05,
      "model_forward_time": 5.629871129989624,
      "step": 13226
    },
    {
      "epoch": 1.2613296508789063e-05,
      "step": 13226,
      "training_step_time": 17.112842082977295
    },
    {
      "epoch": 1.2614250183105469e-05,
      "model_forward_time": 5.5954203605651855,
      "step": 13227
    },
    {
      "epoch": 1.2614250183105469e-05,
      "step": 13227,
      "training_step_time": 17.0334050655365
    },
    {
      "epoch": 1.2615203857421875e-05,
      "model_forward_time": 5.590504884719849,
      "step": 13228
    },
    {
      "epoch": 1.2615203857421875e-05,
      "step": 13228,
      "training_step_time": 17.032763242721558
    },
    {
      "epoch": 1.261615753173828e-05,
      "model_forward_time": 5.57559061050415,
      "step": 13229
    },
    {
      "epoch": 1.261615753173828e-05,
      "step": 13229,
      "training_step_time": 17.371275663375854
    },
    {
      "epoch": 1.2617111206054688e-05,
      "grad_norm": 0.5814314484596252,
      "learning_rate": 1e-05,
      "loss": 0.0703,
      "step": 13230
    },
    {
      "epoch": 1.2617111206054688e-05,
      "model_forward_time": 5.590441703796387,
      "step": 13230
    },
    {
      "dynamics_loss_avg": 0.06739578247070313,
      "epoch": 1.2617111206054688e-05,
      "step": 13230
    },
    {
      "action_loss_avg": 0.007774304016493261,
      "epoch": 1.2617111206054688e-05,
      "step": 13230
    },
    {
      "epoch": 1.2617111206054688e-05,
      "step": 13230,
      "training_step_time": 17.003598928451538
    },
    {
      "epoch": 1.2618064880371094e-05,
      "model_forward_time": 5.579376220703125,
      "step": 13231
    },
    {
      "epoch": 1.2618064880371094e-05,
      "step": 13231,
      "training_step_time": 16.991631269454956
    },
    {
      "epoch": 1.26190185546875e-05,
      "model_forward_time": 5.593193531036377,
      "step": 13232
    },
    {
      "epoch": 1.26190185546875e-05,
      "step": 13232,
      "training_step_time": 17.04605722427368
    },
    {
      "epoch": 1.2619972229003907e-05,
      "model_forward_time": 5.5970001220703125,
      "step": 13233
    },
    {
      "epoch": 1.2619972229003907e-05,
      "step": 13233,
      "training_step_time": 17.022916555404663
    },
    {
      "epoch": 1.2620925903320312e-05,
      "model_forward_time": 5.593068838119507,
      "step": 13234
    },
    {
      "epoch": 1.2620925903320312e-05,
      "step": 13234,
      "training_step_time": 17.067610502243042
    },
    {
      "epoch": 1.2621879577636718e-05,
      "model_forward_time": 5.973723649978638,
      "step": 13235
    },
    {
      "epoch": 1.2621879577636718e-05,
      "step": 13235,
      "training_step_time": 17.44501256942749
    },
    {
      "epoch": 1.2622833251953126e-05,
      "model_forward_time": 5.630212783813477,
      "step": 13236
    },
    {
      "epoch": 1.2622833251953126e-05,
      "step": 13236,
      "training_step_time": 17.149460554122925
    },
    {
      "epoch": 1.2623786926269531e-05,
      "model_forward_time": 5.619089603424072,
      "step": 13237
    },
    {
      "epoch": 1.2623786926269531e-05,
      "step": 13237,
      "training_step_time": 17.1092848777771
    },
    {
      "epoch": 1.2624740600585937e-05,
      "model_forward_time": 5.610208511352539,
      "step": 13238
    },
    {
      "epoch": 1.2624740600585937e-05,
      "step": 13238,
      "training_step_time": 17.110388040542603
    },
    {
      "epoch": 1.2625694274902344e-05,
      "model_forward_time": 5.611658811569214,
      "step": 13239
    },
    {
      "epoch": 1.2625694274902344e-05,
      "step": 13239,
      "training_step_time": 17.11327815055847
    },
    {
      "epoch": 1.262664794921875e-05,
      "grad_norm": 0.31030523777008057,
      "learning_rate": 1e-05,
      "loss": 0.0474,
      "step": 13240
    },
    {
      "epoch": 1.262664794921875e-05,
      "model_forward_time": 5.98696231842041,
      "step": 13240
    },
    {
      "dynamics_loss_avg": 0.03924181945621967,
      "epoch": 1.262664794921875e-05,
      "step": 13240
    },
    {
      "action_loss_avg": 0.004384222277440131,
      "epoch": 1.262664794921875e-05,
      "step": 13240
    },
    {
      "epoch": 1.262664794921875e-05,
      "step": 13240,
      "training_step_time": 17.47044801712036
    },
    {
      "epoch": 1.2627601623535156e-05,
      "model_forward_time": 5.611884355545044,
      "step": 13241
    },
    {
      "epoch": 1.2627601623535156e-05,
      "step": 13241,
      "training_step_time": 17.0884428024292
    },
    {
      "epoch": 1.2628555297851563e-05,
      "model_forward_time": 5.608609676361084,
      "step": 13242
    },
    {
      "epoch": 1.2628555297851563e-05,
      "step": 13242,
      "training_step_time": 17.09480857849121
    },
    {
      "epoch": 1.2629508972167969e-05,
      "model_forward_time": 5.613703966140747,
      "step": 13243
    },
    {
      "epoch": 1.2629508972167969e-05,
      "step": 13243,
      "training_step_time": 17.122493267059326
    },
    {
      "epoch": 1.2630462646484375e-05,
      "model_forward_time": 5.6212921142578125,
      "step": 13244
    },
    {
      "epoch": 1.2630462646484375e-05,
      "step": 13244,
      "training_step_time": 17.127504587173462
    },
    {
      "epoch": 1.263141632080078e-05,
      "model_forward_time": 5.624976873397827,
      "step": 13245
    },
    {
      "epoch": 1.263141632080078e-05,
      "step": 13245,
      "training_step_time": 17.13678526878357
    },
    {
      "epoch": 1.2632369995117188e-05,
      "model_forward_time": 5.980027437210083,
      "step": 13246
    },
    {
      "epoch": 1.2632369995117188e-05,
      "step": 13246,
      "training_step_time": 17.47802495956421
    },
    {
      "epoch": 1.2633323669433594e-05,
      "model_forward_time": 5.600545883178711,
      "step": 13247
    },
    {
      "epoch": 1.2633323669433594e-05,
      "step": 13247,
      "training_step_time": 17.044535875320435
    },
    {
      "epoch": 1.263427734375e-05,
      "model_forward_time": 5.588333606719971,
      "step": 13248
    },
    {
      "epoch": 1.263427734375e-05,
      "step": 13248,
      "training_step_time": 17.015451192855835
    },
    {
      "epoch": 1.2635231018066407e-05,
      "model_forward_time": 5.581678152084351,
      "step": 13249
    },
    {
      "epoch": 1.2635231018066407e-05,
      "step": 13249,
      "training_step_time": 17.019652843475342
    },
    {
      "epoch": 1.2636184692382812e-05,
      "grad_norm": 1.2941144704818726,
      "learning_rate": 1e-05,
      "loss": 0.0718,
      "step": 13250
    },
    {
      "epoch": 1.2636184692382812e-05,
      "model_forward_time": 5.591838359832764,
      "step": 13250
    },
    {
      "dynamics_loss_avg": 0.06955150663852691,
      "epoch": 1.2636184692382812e-05,
      "step": 13250
    },
    {
      "action_loss_avg": 0.009177414583973587,
      "epoch": 1.2636184692382812e-05,
      "step": 13250
    },
    {
      "epoch": 1.2636184692382812e-05,
      "step": 13250,
      "training_step_time": 17.03041958808899
    },
    {
      "epoch": 1.2637138366699218e-05,
      "model_forward_time": 5.974431276321411,
      "step": 13251
    },
    {
      "epoch": 1.2637138366699218e-05,
      "step": 13251,
      "training_step_time": 17.4286208152771
    },
    {
      "epoch": 1.2638092041015626e-05,
      "model_forward_time": 5.593911170959473,
      "step": 13252
    },
    {
      "epoch": 1.2638092041015626e-05,
      "step": 13252,
      "training_step_time": 17.06687068939209
    },
    {
      "epoch": 1.2639045715332031e-05,
      "model_forward_time": 5.603276014328003,
      "step": 13253
    },
    {
      "epoch": 1.2639045715332031e-05,
      "step": 13253,
      "training_step_time": 17.05668878555298
    },
    {
      "epoch": 1.2639999389648437e-05,
      "model_forward_time": 5.594442844390869,
      "step": 13254
    },
    {
      "epoch": 1.2639999389648437e-05,
      "step": 13254,
      "training_step_time": 17.054546356201172
    },
    {
      "epoch": 1.2640953063964844e-05,
      "model_forward_time": 5.605175733566284,
      "step": 13255
    },
    {
      "epoch": 1.2640953063964844e-05,
      "step": 13255,
      "training_step_time": 17.048431396484375
    },
    {
      "epoch": 1.264190673828125e-05,
      "model_forward_time": 5.592773914337158,
      "step": 13256
    },
    {
      "epoch": 1.264190673828125e-05,
      "step": 13256,
      "training_step_time": 17.055638074874878
    },
    {
      "epoch": 1.2642860412597656e-05,
      "model_forward_time": 5.964312314987183,
      "step": 13257
    },
    {
      "epoch": 1.2642860412597656e-05,
      "step": 13257,
      "training_step_time": 17.456434965133667
    },
    {
      "epoch": 1.2643814086914062e-05,
      "model_forward_time": 5.617652654647827,
      "step": 13258
    },
    {
      "epoch": 1.2643814086914062e-05,
      "step": 13258,
      "training_step_time": 17.088953495025635
    },
    {
      "epoch": 1.2644767761230469e-05,
      "model_forward_time": 5.585622549057007,
      "step": 13259
    },
    {
      "epoch": 1.2644767761230469e-05,
      "step": 13259,
      "training_step_time": 17.015495538711548
    },
    {
      "epoch": 1.2645721435546875e-05,
      "grad_norm": 0.3014581501483917,
      "learning_rate": 1e-05,
      "loss": 0.0706,
      "step": 13260
    },
    {
      "epoch": 1.2645721435546875e-05,
      "model_forward_time": 5.580756902694702,
      "step": 13260
    },
    {
      "dynamics_loss_avg": 0.053312179446220395,
      "epoch": 1.2645721435546875e-05,
      "step": 13260
    },
    {
      "action_loss_avg": 0.008942161104641855,
      "epoch": 1.2645721435546875e-05,
      "step": 13260
    },
    {
      "epoch": 1.2645721435546875e-05,
      "step": 13260,
      "training_step_time": 17.022931575775146
    },
    {
      "epoch": 1.264667510986328e-05,
      "model_forward_time": 5.58697509765625,
      "step": 13261
    },
    {
      "epoch": 1.264667510986328e-05,
      "step": 13261,
      "training_step_time": 16.995805501937866
    },
    {
      "epoch": 1.2647628784179688e-05,
      "model_forward_time": 5.589250564575195,
      "step": 13262
    },
    {
      "epoch": 1.2647628784179688e-05,
      "step": 13262,
      "training_step_time": 17.396221160888672
    },
    {
      "epoch": 1.2648582458496094e-05,
      "model_forward_time": 5.584899663925171,
      "step": 13263
    },
    {
      "epoch": 1.2648582458496094e-05,
      "step": 13263,
      "training_step_time": 17.047974348068237
    },
    {
      "epoch": 1.26495361328125e-05,
      "model_forward_time": 5.594793796539307,
      "step": 13264
    },
    {
      "epoch": 1.26495361328125e-05,
      "step": 13264,
      "training_step_time": 17.076778411865234
    },
    {
      "epoch": 1.2650489807128907e-05,
      "model_forward_time": 5.614168643951416,
      "step": 13265
    },
    {
      "epoch": 1.2650489807128907e-05,
      "step": 13265,
      "training_step_time": 17.12883448600769
    },
    {
      "epoch": 1.2651443481445313e-05,
      "model_forward_time": 5.629674434661865,
      "step": 13266
    },
    {
      "epoch": 1.2651443481445313e-05,
      "step": 13266,
      "training_step_time": 17.143235445022583
    },
    {
      "epoch": 1.2652397155761718e-05,
      "model_forward_time": 5.599032163619995,
      "step": 13267
    },
    {
      "epoch": 1.2652397155761718e-05,
      "step": 13267,
      "training_step_time": 17.06542730331421
    },
    {
      "epoch": 1.2653350830078126e-05,
      "model_forward_time": 5.957474231719971,
      "step": 13268
    },
    {
      "epoch": 1.2653350830078126e-05,
      "step": 13268,
      "training_step_time": 17.393694639205933
    },
    {
      "epoch": 1.2654304504394531e-05,
      "model_forward_time": 5.589548587799072,
      "step": 13269
    },
    {
      "epoch": 1.2654304504394531e-05,
      "step": 13269,
      "training_step_time": 17.01460289955139
    },
    {
      "epoch": 1.2655258178710937e-05,
      "grad_norm": 0.9983702301979065,
      "learning_rate": 1e-05,
      "loss": 0.0605,
      "step": 13270
    },
    {
      "epoch": 1.2655258178710937e-05,
      "model_forward_time": 5.585399389266968,
      "step": 13270
    },
    {
      "dynamics_loss_avg": 0.05140843894332647,
      "epoch": 1.2655258178710937e-05,
      "step": 13270
    },
    {
      "action_loss_avg": 0.009240366658195854,
      "epoch": 1.2655258178710937e-05,
      "step": 13270
    },
    {
      "epoch": 1.2655258178710937e-05,
      "step": 13270,
      "training_step_time": 17.040410041809082
    },
    {
      "epoch": 1.2656211853027345e-05,
      "model_forward_time": 5.59416651725769,
      "step": 13271
    },
    {
      "epoch": 1.2656211853027345e-05,
      "step": 13271,
      "training_step_time": 17.035609483718872
    },
    {
      "epoch": 1.265716552734375e-05,
      "model_forward_time": 5.587682485580444,
      "step": 13272
    },
    {
      "epoch": 1.265716552734375e-05,
      "step": 13272,
      "training_step_time": 17.03893232345581
    },
    {
      "epoch": 1.2658119201660156e-05,
      "model_forward_time": 5.983876705169678,
      "step": 13273
    },
    {
      "epoch": 1.2658119201660156e-05,
      "step": 13273,
      "training_step_time": 17.443628311157227
    },
    {
      "epoch": 1.2659072875976562e-05,
      "model_forward_time": 5.596304416656494,
      "step": 13274
    },
    {
      "epoch": 1.2659072875976562e-05,
      "step": 13274,
      "training_step_time": 17.073020458221436
    },
    {
      "epoch": 1.2660026550292969e-05,
      "model_forward_time": 5.6025049686431885,
      "step": 13275
    },
    {
      "epoch": 1.2660026550292969e-05,
      "step": 13275,
      "training_step_time": 17.07232666015625
    },
    {
      "epoch": 1.2660980224609375e-05,
      "model_forward_time": 5.595985412597656,
      "step": 13276
    },
    {
      "epoch": 1.2660980224609375e-05,
      "step": 13276,
      "training_step_time": 17.044501304626465
    },
    {
      "epoch": 1.266193389892578e-05,
      "model_forward_time": 5.593947172164917,
      "step": 13277
    },
    {
      "epoch": 1.266193389892578e-05,
      "step": 13277,
      "training_step_time": 17.047350645065308
    },
    {
      "epoch": 1.2662887573242188e-05,
      "model_forward_time": 5.601521730422974,
      "step": 13278
    },
    {
      "epoch": 1.2662887573242188e-05,
      "step": 13278,
      "training_step_time": 17.103562116622925
    },
    {
      "epoch": 1.2663841247558594e-05,
      "model_forward_time": 5.967738151550293,
      "step": 13279
    },
    {
      "epoch": 1.2663841247558594e-05,
      "step": 13279,
      "training_step_time": 17.47374391555786
    },
    {
      "epoch": 1.2664794921875e-05,
      "grad_norm": 1.0733444690704346,
      "learning_rate": 1e-05,
      "loss": 0.0562,
      "step": 13280
    },
    {
      "epoch": 1.2664794921875e-05,
      "model_forward_time": 5.621443510055542,
      "step": 13280
    },
    {
      "dynamics_loss_avg": 0.049365826323628424,
      "epoch": 1.2664794921875e-05,
      "step": 13280
    },
    {
      "action_loss_avg": 0.0066749553196132185,
      "epoch": 1.2664794921875e-05,
      "step": 13280
    },
    {
      "epoch": 1.2664794921875e-05,
      "step": 13280,
      "training_step_time": 17.137173414230347
    },
    {
      "epoch": 1.2665748596191407e-05,
      "model_forward_time": 5.6152849197387695,
      "step": 13281
    },
    {
      "epoch": 1.2665748596191407e-05,
      "step": 13281,
      "training_step_time": 17.095964431762695
    },
    {
      "epoch": 1.2666702270507813e-05,
      "model_forward_time": 5.587463617324829,
      "step": 13282
    },
    {
      "epoch": 1.2666702270507813e-05,
      "step": 13282,
      "training_step_time": 17.018798351287842
    },
    {
      "epoch": 1.2667655944824218e-05,
      "model_forward_time": 5.5860583782196045,
      "step": 13283
    },
    {
      "epoch": 1.2667655944824218e-05,
      "step": 13283,
      "training_step_time": 17.006275177001953
    },
    {
      "epoch": 1.2668609619140626e-05,
      "model_forward_time": 5.943872690200806,
      "step": 13284
    },
    {
      "epoch": 1.2668609619140626e-05,
      "step": 13284,
      "training_step_time": 17.36844038963318
    },
    {
      "epoch": 1.2669563293457031e-05,
      "model_forward_time": 5.579582691192627,
      "step": 13285
    },
    {
      "epoch": 1.2669563293457031e-05,
      "step": 13285,
      "training_step_time": 17.0114483833313
    },
    {
      "epoch": 1.2670516967773437e-05,
      "model_forward_time": 5.581125259399414,
      "step": 13286
    },
    {
      "epoch": 1.2670516967773437e-05,
      "step": 13286,
      "training_step_time": 17.008487939834595
    },
    {
      "epoch": 1.2671470642089845e-05,
      "model_forward_time": 5.5977418422698975,
      "step": 13287
    },
    {
      "epoch": 1.2671470642089845e-05,
      "step": 13287,
      "training_step_time": 17.057377338409424
    },
    {
      "epoch": 1.267242431640625e-05,
      "model_forward_time": 5.60143780708313,
      "step": 13288
    },
    {
      "epoch": 1.267242431640625e-05,
      "step": 13288,
      "training_step_time": 17.073446035385132
    },
    {
      "epoch": 1.2673377990722656e-05,
      "model_forward_time": 5.6149561405181885,
      "step": 13289
    },
    {
      "epoch": 1.2673377990722656e-05,
      "step": 13289,
      "training_step_time": 17.092032432556152
    },
    {
      "epoch": 1.2674331665039062e-05,
      "grad_norm": 0.8089906573295593,
      "learning_rate": 1e-05,
      "loss": 0.0499,
      "step": 13290
    },
    {
      "epoch": 1.2674331665039062e-05,
      "model_forward_time": 5.961992025375366,
      "step": 13290
    },
    {
      "dynamics_loss_avg": 0.05044747292995453,
      "epoch": 1.2674331665039062e-05,
      "step": 13290
    },
    {
      "action_loss_avg": 0.00652175882132724,
      "epoch": 1.2674331665039062e-05,
      "step": 13290
    },
    {
      "epoch": 1.2674331665039062e-05,
      "step": 13290,
      "training_step_time": 17.446889638900757
    },
    {
      "epoch": 1.2675285339355469e-05,
      "model_forward_time": 5.5968918800354,
      "step": 13291
    },
    {
      "epoch": 1.2675285339355469e-05,
      "step": 13291,
      "training_step_time": 17.0427188873291
    },
    {
      "epoch": 1.2676239013671875e-05,
      "model_forward_time": 5.587462902069092,
      "step": 13292
    },
    {
      "epoch": 1.2676239013671875e-05,
      "step": 13292,
      "training_step_time": 17.025171518325806
    },
    {
      "epoch": 1.267719268798828e-05,
      "model_forward_time": 5.584527969360352,
      "step": 13293
    },
    {
      "epoch": 1.267719268798828e-05,
      "step": 13293,
      "training_step_time": 17.04689598083496
    },
    {
      "epoch": 1.2678146362304688e-05,
      "model_forward_time": 5.6189446449279785,
      "step": 13294
    },
    {
      "epoch": 1.2678146362304688e-05,
      "step": 13294,
      "training_step_time": 17.118612051010132
    },
    {
      "epoch": 1.2679100036621094e-05,
      "model_forward_time": 5.618301630020142,
      "step": 13295
    },
    {
      "epoch": 1.2679100036621094e-05,
      "step": 13295,
      "training_step_time": 17.508065462112427
    },
    {
      "epoch": 1.26800537109375e-05,
      "model_forward_time": 5.611150503158569,
      "step": 13296
    },
    {
      "epoch": 1.26800537109375e-05,
      "step": 13296,
      "training_step_time": 17.07685136795044
    },
    {
      "epoch": 1.2681007385253907e-05,
      "model_forward_time": 5.611397743225098,
      "step": 13297
    },
    {
      "epoch": 1.2681007385253907e-05,
      "step": 13297,
      "training_step_time": 17.065831899642944
    },
    {
      "epoch": 1.2681961059570313e-05,
      "model_forward_time": 5.60392951965332,
      "step": 13298
    },
    {
      "epoch": 1.2681961059570313e-05,
      "step": 13298,
      "training_step_time": 17.063067197799683
    },
    {
      "epoch": 1.2682914733886718e-05,
      "model_forward_time": 5.5911054611206055,
      "step": 13299
    },
    {
      "epoch": 1.2682914733886718e-05,
      "step": 13299,
      "training_step_time": 17.01607394218445
    },
    {
      "epoch": 1.2683868408203126e-05,
      "grad_norm": 1.6999624967575073,
      "learning_rate": 1e-05,
      "loss": 0.0723,
      "step": 13300
    },
    {
      "epoch": 1.2683868408203126e-05,
      "model_forward_time": 5.585188865661621,
      "step": 13300
    },
    {
      "dynamics_loss_avg": 0.061938327923417094,
      "epoch": 1.2683868408203126e-05,
      "step": 13300
    },
    {
      "action_loss_avg": 0.008122915728017687,
      "epoch": 1.2683868408203126e-05,
      "step": 13300
    },
    {
      "epoch": 1.2683868408203126e-05,
      "step": 13300,
      "training_step_time": 17.01469111442566
    },
    {
      "epoch": 1.2684822082519531e-05,
      "model_forward_time": 5.956002712249756,
      "step": 13301
    },
    {
      "epoch": 1.2684822082519531e-05,
      "step": 13301,
      "training_step_time": 17.4277982711792
    },
    {
      "epoch": 1.2685775756835937e-05,
      "model_forward_time": 5.603813648223877,
      "step": 13302
    },
    {
      "epoch": 1.2685775756835937e-05,
      "step": 13302,
      "training_step_time": 17.09853982925415
    },
    {
      "epoch": 1.2686729431152345e-05,
      "model_forward_time": 5.625528573989868,
      "step": 13303
    },
    {
      "epoch": 1.2686729431152345e-05,
      "step": 13303,
      "training_step_time": 17.119913339614868
    },
    {
      "epoch": 1.268768310546875e-05,
      "model_forward_time": 5.604217290878296,
      "step": 13304
    },
    {
      "epoch": 1.268768310546875e-05,
      "step": 13304,
      "training_step_time": 17.047935962677002
    },
    {
      "epoch": 1.2688636779785156e-05,
      "model_forward_time": 5.583396911621094,
      "step": 13305
    },
    {
      "epoch": 1.2688636779785156e-05,
      "step": 13305,
      "training_step_time": 17.030516624450684
    },
    {
      "epoch": 1.2689590454101562e-05,
      "model_forward_time": 5.976910352706909,
      "step": 13306
    },
    {
      "epoch": 1.2689590454101562e-05,
      "step": 13306,
      "training_step_time": 17.458664417266846
    },
    {
      "epoch": 1.2690544128417969e-05,
      "model_forward_time": 5.616023540496826,
      "step": 13307
    },
    {
      "epoch": 1.2690544128417969e-05,
      "step": 13307,
      "training_step_time": 17.127519607543945
    },
    {
      "epoch": 1.2691497802734375e-05,
      "model_forward_time": 5.614586353302002,
      "step": 13308
    },
    {
      "epoch": 1.2691497802734375e-05,
      "step": 13308,
      "training_step_time": 17.11070203781128
    },
    {
      "epoch": 1.269245147705078e-05,
      "model_forward_time": 5.618066072463989,
      "step": 13309
    },
    {
      "epoch": 1.269245147705078e-05,
      "step": 13309,
      "training_step_time": 17.128679990768433
    },
    {
      "epoch": 1.2693405151367188e-05,
      "grad_norm": 0.7061273455619812,
      "learning_rate": 1e-05,
      "loss": 0.0682,
      "step": 13310
    },
    {
      "epoch": 1.2693405151367188e-05,
      "model_forward_time": 5.6150641441345215,
      "step": 13310
    },
    {
      "dynamics_loss_avg": 0.05892911422997713,
      "epoch": 1.2693405151367188e-05,
      "step": 13310
    },
    {
      "action_loss_avg": 0.007037797686643898,
      "epoch": 1.2693405151367188e-05,
      "step": 13310
    },
    {
      "epoch": 1.2693405151367188e-05,
      "step": 13310,
      "training_step_time": 17.115490913391113
    },
    {
      "epoch": 1.2694358825683594e-05,
      "model_forward_time": 5.610842943191528,
      "step": 13311
    },
    {
      "epoch": 1.2694358825683594e-05,
      "step": 13311,
      "training_step_time": 17.114147901535034
    },
    {
      "epoch": 1.26953125e-05,
      "model_forward_time": 5.974878787994385,
      "step": 13312
    },
    {
      "epoch": 1.26953125e-05,
      "step": 13312,
      "training_step_time": 17.444178342819214
    },
    {
      "epoch": 1.2696266174316407e-05,
      "model_forward_time": 5.609447240829468,
      "step": 13313
    },
    {
      "epoch": 1.2696266174316407e-05,
      "step": 13313,
      "training_step_time": 17.105146884918213
    },
    {
      "epoch": 1.2697219848632813e-05,
      "model_forward_time": 5.609896421432495,
      "step": 13314
    },
    {
      "epoch": 1.2697219848632813e-05,
      "step": 13314,
      "training_step_time": 17.100938320159912
    },
    {
      "epoch": 1.2698173522949218e-05,
      "model_forward_time": 5.611295223236084,
      "step": 13315
    },
    {
      "epoch": 1.2698173522949218e-05,
      "step": 13315,
      "training_step_time": 17.096815586090088
    },
    {
      "epoch": 1.2699127197265626e-05,
      "model_forward_time": 5.601802349090576,
      "step": 13316
    },
    {
      "epoch": 1.2699127197265626e-05,
      "step": 13316,
      "training_step_time": 17.093252420425415
    },
    {
      "epoch": 1.2700080871582031e-05,
      "model_forward_time": 5.991087913513184,
      "step": 13317
    },
    {
      "epoch": 1.2700080871582031e-05,
      "step": 13317,
      "training_step_time": 17.495336055755615
    },
    {
      "epoch": 1.2701034545898437e-05,
      "model_forward_time": 5.62294864654541,
      "step": 13318
    },
    {
      "epoch": 1.2701034545898437e-05,
      "step": 13318,
      "training_step_time": 17.125682830810547
    },
    {
      "epoch": 1.2701988220214845e-05,
      "model_forward_time": 5.6201159954071045,
      "step": 13319
    },
    {
      "epoch": 1.2701988220214845e-05,
      "step": 13319,
      "training_step_time": 17.10744547843933
    },
    {
      "epoch": 1.270294189453125e-05,
      "grad_norm": 0.30690324306488037,
      "learning_rate": 1e-05,
      "loss": 0.0609,
      "step": 13320
    },
    {
      "epoch": 1.270294189453125e-05,
      "model_forward_time": 5.609057426452637,
      "step": 13320
    },
    {
      "dynamics_loss_avg": 0.057464790716767314,
      "epoch": 1.270294189453125e-05,
      "step": 13320
    },
    {
      "action_loss_avg": 0.0066978704999201,
      "epoch": 1.270294189453125e-05,
      "step": 13320
    },
    {
      "epoch": 1.270294189453125e-05,
      "step": 13320,
      "training_step_time": 17.093925714492798
    },
    {
      "epoch": 1.2703895568847656e-05,
      "model_forward_time": 5.606515884399414,
      "step": 13321
    },
    {
      "epoch": 1.2703895568847656e-05,
      "step": 13321,
      "training_step_time": 17.091099739074707
    },
    {
      "epoch": 1.2704849243164062e-05,
      "model_forward_time": 5.615650415420532,
      "step": 13322
    },
    {
      "epoch": 1.2704849243164062e-05,
      "step": 13322,
      "training_step_time": 17.101096391677856
    },
    {
      "epoch": 1.2705802917480469e-05,
      "model_forward_time": 5.980838298797607,
      "step": 13323
    },
    {
      "epoch": 1.2705802917480469e-05,
      "step": 13323,
      "training_step_time": 17.48859667778015
    },
    {
      "epoch": 1.2706756591796875e-05,
      "model_forward_time": 5.624232769012451,
      "step": 13324
    },
    {
      "epoch": 1.2706756591796875e-05,
      "step": 13324,
      "training_step_time": 17.130834579467773
    },
    {
      "epoch": 1.270771026611328e-05,
      "model_forward_time": 5.609656572341919,
      "step": 13325
    },
    {
      "epoch": 1.270771026611328e-05,
      "step": 13325,
      "training_step_time": 17.110107421875
    },
    {
      "epoch": 1.2708663940429688e-05,
      "model_forward_time": 5.610071420669556,
      "step": 13326
    },
    {
      "epoch": 1.2708663940429688e-05,
      "step": 13326,
      "training_step_time": 17.105363368988037
    },
    {
      "epoch": 1.2709617614746094e-05,
      "model_forward_time": 5.619791746139526,
      "step": 13327
    },
    {
      "epoch": 1.2709617614746094e-05,
      "step": 13327,
      "training_step_time": 17.100338459014893
    },
    {
      "epoch": 1.27105712890625e-05,
      "model_forward_time": 5.622786998748779,
      "step": 13328
    },
    {
      "epoch": 1.27105712890625e-05,
      "step": 13328,
      "training_step_time": 17.4861741065979
    },
    {
      "epoch": 1.2711524963378907e-05,
      "model_forward_time": 5.605517387390137,
      "step": 13329
    },
    {
      "epoch": 1.2711524963378907e-05,
      "step": 13329,
      "training_step_time": 17.101001977920532
    },
    {
      "epoch": 1.2712478637695313e-05,
      "grad_norm": 0.5944759845733643,
      "learning_rate": 1e-05,
      "loss": 0.0671,
      "step": 13330
    },
    {
      "epoch": 1.2712478637695313e-05,
      "model_forward_time": 5.6077611446380615,
      "step": 13330
    },
    {
      "dynamics_loss_avg": 0.05937865059822798,
      "epoch": 1.2712478637695313e-05,
      "step": 13330
    },
    {
      "action_loss_avg": 0.009762325114570558,
      "epoch": 1.2712478637695313e-05,
      "step": 13330
    },
    {
      "epoch": 1.2712478637695313e-05,
      "step": 13330,
      "training_step_time": 17.08336615562439
    },
    {
      "epoch": 1.2713432312011718e-05,
      "model_forward_time": 5.60638689994812,
      "step": 13331
    },
    {
      "epoch": 1.2713432312011718e-05,
      "step": 13331,
      "training_step_time": 17.049639225006104
    },
    {
      "epoch": 1.2714385986328126e-05,
      "model_forward_time": 5.594840049743652,
      "step": 13332
    },
    {
      "epoch": 1.2714385986328126e-05,
      "step": 13332,
      "training_step_time": 17.026223182678223
    },
    {
      "epoch": 1.2715339660644531e-05,
      "model_forward_time": 5.593143463134766,
      "step": 13333
    },
    {
      "epoch": 1.2715339660644531e-05,
      "step": 13333,
      "training_step_time": 17.014586448669434
    },
    {
      "epoch": 1.2716293334960937e-05,
      "model_forward_time": 5.954704761505127,
      "step": 13334
    },
    {
      "epoch": 1.2716293334960937e-05,
      "step": 13334,
      "training_step_time": 17.394644260406494
    },
    {
      "epoch": 1.2717247009277345e-05,
      "model_forward_time": 5.61142110824585,
      "step": 13335
    },
    {
      "epoch": 1.2717247009277345e-05,
      "step": 13335,
      "training_step_time": 17.091832637786865
    },
    {
      "epoch": 1.271820068359375e-05,
      "model_forward_time": 5.617392063140869,
      "step": 13336
    },
    {
      "epoch": 1.271820068359375e-05,
      "step": 13336,
      "training_step_time": 17.116033792495728
    },
    {
      "epoch": 1.2719154357910156e-05,
      "model_forward_time": 5.610145568847656,
      "step": 13337
    },
    {
      "epoch": 1.2719154357910156e-05,
      "step": 13337,
      "training_step_time": 17.058307647705078
    },
    {
      "epoch": 1.2720108032226562e-05,
      "model_forward_time": 5.584090232849121,
      "step": 13338
    },
    {
      "epoch": 1.2720108032226562e-05,
      "step": 13338,
      "training_step_time": 17.032232522964478
    },
    {
      "epoch": 1.2721061706542969e-05,
      "model_forward_time": 5.970073461532593,
      "step": 13339
    },
    {
      "epoch": 1.2721061706542969e-05,
      "step": 13339,
      "training_step_time": 17.426332473754883
    },
    {
      "epoch": 1.2722015380859375e-05,
      "grad_norm": 0.41871145367622375,
      "learning_rate": 1e-05,
      "loss": 0.0614,
      "step": 13340
    },
    {
      "epoch": 1.2722015380859375e-05,
      "model_forward_time": 5.594029426574707,
      "step": 13340
    },
    {
      "dynamics_loss_avg": 0.05264871120452881,
      "epoch": 1.2722015380859375e-05,
      "step": 13340
    },
    {
      "action_loss_avg": 0.007158511783927679,
      "epoch": 1.2722015380859375e-05,
      "step": 13340
    },
    {
      "epoch": 1.2722015380859375e-05,
      "step": 13340,
      "training_step_time": 17.05155086517334
    },
    {
      "epoch": 1.272296905517578e-05,
      "model_forward_time": 5.603409767150879,
      "step": 13341
    },
    {
      "epoch": 1.272296905517578e-05,
      "step": 13341,
      "training_step_time": 17.07194185256958
    },
    {
      "epoch": 1.2723922729492188e-05,
      "model_forward_time": 5.601459980010986,
      "step": 13342
    },
    {
      "epoch": 1.2723922729492188e-05,
      "step": 13342,
      "training_step_time": 17.08339786529541
    },
    {
      "epoch": 1.2724876403808594e-05,
      "model_forward_time": 5.615348815917969,
      "step": 13343
    },
    {
      "epoch": 1.2724876403808594e-05,
      "step": 13343,
      "training_step_time": 17.09386396408081
    },
    {
      "epoch": 1.2725830078125e-05,
      "model_forward_time": 5.60749077796936,
      "step": 13344
    },
    {
      "epoch": 1.2725830078125e-05,
      "step": 13344,
      "training_step_time": 17.072264909744263
    },
    {
      "epoch": 1.2726783752441407e-05,
      "model_forward_time": 5.949425935745239,
      "step": 13345
    },
    {
      "epoch": 1.2726783752441407e-05,
      "step": 13345,
      "training_step_time": 17.410173654556274
    },
    {
      "epoch": 1.2727737426757813e-05,
      "model_forward_time": 5.6087915897369385,
      "step": 13346
    },
    {
      "epoch": 1.2727737426757813e-05,
      "step": 13346,
      "training_step_time": 17.055938005447388
    },
    {
      "epoch": 1.2728691101074218e-05,
      "model_forward_time": 5.5998969078063965,
      "step": 13347
    },
    {
      "epoch": 1.2728691101074218e-05,
      "step": 13347,
      "training_step_time": 17.076687812805176
    },
    {
      "epoch": 1.2729644775390626e-05,
      "model_forward_time": 5.602291584014893,
      "step": 13348
    },
    {
      "epoch": 1.2729644775390626e-05,
      "step": 13348,
      "training_step_time": 17.06889796257019
    },
    {
      "epoch": 1.2730598449707031e-05,
      "model_forward_time": 5.597525358200073,
      "step": 13349
    },
    {
      "epoch": 1.2730598449707031e-05,
      "step": 13349,
      "training_step_time": 17.046098947525024
    },
    {
      "epoch": 1.2731552124023437e-05,
      "grad_norm": 0.5587963461875916,
      "learning_rate": 1e-05,
      "loss": 0.0607,
      "step": 13350
    },
    {
      "epoch": 1.2731552124023437e-05,
      "model_forward_time": 5.984620571136475,
      "step": 13350
    },
    {
      "dynamics_loss_avg": 0.05614528451114893,
      "epoch": 1.2731552124023437e-05,
      "step": 13350
    },
    {
      "action_loss_avg": 0.0072486544726416465,
      "epoch": 1.2731552124023437e-05,
      "step": 13350
    },
    {
      "epoch": 1.2731552124023437e-05,
      "step": 13350,
      "training_step_time": 17.45130467414856
    },
    {
      "epoch": 1.2732505798339845e-05,
      "model_forward_time": 5.640439987182617,
      "step": 13351
    },
    {
      "epoch": 1.2732505798339845e-05,
      "step": 13351,
      "training_step_time": 17.148813724517822
    },
    {
      "epoch": 1.273345947265625e-05,
      "model_forward_time": 5.617362022399902,
      "step": 13352
    },
    {
      "epoch": 1.273345947265625e-05,
      "step": 13352,
      "training_step_time": 17.123751878738403
    },
    {
      "epoch": 1.2734413146972656e-05,
      "model_forward_time": 5.619800090789795,
      "step": 13353
    },
    {
      "epoch": 1.2734413146972656e-05,
      "step": 13353,
      "training_step_time": 17.08484983444214
    },
    {
      "epoch": 1.2735366821289062e-05,
      "model_forward_time": 5.586731910705566,
      "step": 13354
    },
    {
      "epoch": 1.2735366821289062e-05,
      "step": 13354,
      "training_step_time": 17.019389390945435
    },
    {
      "epoch": 1.273632049560547e-05,
      "model_forward_time": 5.59720516204834,
      "step": 13355
    },
    {
      "epoch": 1.273632049560547e-05,
      "step": 13355,
      "training_step_time": 17.019670486450195
    },
    {
      "epoch": 1.2737274169921875e-05,
      "model_forward_time": 5.935349464416504,
      "step": 13356
    },
    {
      "epoch": 1.2737274169921875e-05,
      "step": 13356,
      "training_step_time": 17.350212574005127
    },
    {
      "epoch": 1.273822784423828e-05,
      "model_forward_time": 5.579919099807739,
      "step": 13357
    },
    {
      "epoch": 1.273822784423828e-05,
      "step": 13357,
      "training_step_time": 17.01658821105957
    },
    {
      "epoch": 1.2739181518554688e-05,
      "model_forward_time": 5.575926065444946,
      "step": 13358
    },
    {
      "epoch": 1.2739181518554688e-05,
      "step": 13358,
      "training_step_time": 16.984142303466797
    },
    {
      "epoch": 1.2740135192871094e-05,
      "model_forward_time": 5.595329999923706,
      "step": 13359
    },
    {
      "epoch": 1.2740135192871094e-05,
      "step": 13359,
      "training_step_time": 17.03642773628235
    },
    {
      "epoch": 1.27410888671875e-05,
      "grad_norm": 1.8019896745681763,
      "learning_rate": 1e-05,
      "loss": 0.0799,
      "step": 13360
    },
    {
      "epoch": 1.27410888671875e-05,
      "model_forward_time": 5.595741033554077,
      "step": 13360
    },
    {
      "dynamics_loss_avg": 0.06589749082922935,
      "epoch": 1.27410888671875e-05,
      "step": 13360
    },
    {
      "action_loss_avg": 0.010686474572867154,
      "epoch": 1.27410888671875e-05,
      "step": 13360
    },
    {
      "epoch": 1.27410888671875e-05,
      "step": 13360,
      "training_step_time": 17.047890424728394
    },
    {
      "epoch": 1.2742042541503907e-05,
      "model_forward_time": 5.598472356796265,
      "step": 13361
    },
    {
      "epoch": 1.2742042541503907e-05,
      "step": 13361,
      "training_step_time": 17.42653465270996
    },
    {
      "epoch": 1.2742996215820313e-05,
      "model_forward_time": 5.608457565307617,
      "step": 13362
    },
    {
      "epoch": 1.2742996215820313e-05,
      "step": 13362,
      "training_step_time": 17.091753244400024
    },
    {
      "epoch": 1.2743949890136718e-05,
      "model_forward_time": 5.601691484451294,
      "step": 13363
    },
    {
      "epoch": 1.2743949890136718e-05,
      "step": 13363,
      "training_step_time": 17.076666593551636
    },
    {
      "epoch": 1.2744903564453126e-05,
      "model_forward_time": 5.606616973876953,
      "step": 13364
    },
    {
      "epoch": 1.2744903564453126e-05,
      "step": 13364,
      "training_step_time": 17.08418846130371
    },
    {
      "epoch": 1.2745857238769531e-05,
      "model_forward_time": 5.6176769733428955,
      "step": 13365
    },
    {
      "epoch": 1.2745857238769531e-05,
      "step": 13365,
      "training_step_time": 17.103370189666748
    },
    {
      "epoch": 1.2746810913085937e-05,
      "model_forward_time": 5.621001482009888,
      "step": 13366
    },
    {
      "epoch": 1.2746810913085937e-05,
      "step": 13366,
      "training_step_time": 17.12503957748413
    },
    {
      "epoch": 1.2747764587402345e-05,
      "model_forward_time": 6.012397766113281,
      "step": 13367
    },
    {
      "epoch": 1.2747764587402345e-05,
      "step": 13367,
      "training_step_time": 17.519683122634888
    },
    {
      "epoch": 1.274871826171875e-05,
      "model_forward_time": 5.616893529891968,
      "step": 13368
    },
    {
      "epoch": 1.274871826171875e-05,
      "step": 13368,
      "training_step_time": 17.114538431167603
    },
    {
      "epoch": 1.2749671936035156e-05,
      "model_forward_time": 5.607670307159424,
      "step": 13369
    },
    {
      "epoch": 1.2749671936035156e-05,
      "step": 13369,
      "training_step_time": 17.09107804298401
    },
    {
      "epoch": 1.2750625610351562e-05,
      "grad_norm": 0.7744681239128113,
      "learning_rate": 1e-05,
      "loss": 0.068,
      "step": 13370
    },
    {
      "epoch": 1.2750625610351562e-05,
      "model_forward_time": 5.610873460769653,
      "step": 13370
    },
    {
      "dynamics_loss_avg": 0.05866897124797106,
      "epoch": 1.2750625610351562e-05,
      "step": 13370
    },
    {
      "action_loss_avg": 0.008723800559528172,
      "epoch": 1.2750625610351562e-05,
      "step": 13370
    },
    {
      "epoch": 1.2750625610351562e-05,
      "step": 13370,
      "training_step_time": 17.093896865844727
    },
    {
      "epoch": 1.275157928466797e-05,
      "model_forward_time": 5.6156227588653564,
      "step": 13371
    },
    {
      "epoch": 1.275157928466797e-05,
      "step": 13371,
      "training_step_time": 17.073153018951416
    },
    {
      "epoch": 1.2752532958984375e-05,
      "model_forward_time": 5.95247745513916,
      "step": 13372
    },
    {
      "epoch": 1.2752532958984375e-05,
      "step": 13372,
      "training_step_time": 17.38515567779541
    },
    {
      "epoch": 1.275348663330078e-05,
      "model_forward_time": 5.593756675720215,
      "step": 13373
    },
    {
      "epoch": 1.275348663330078e-05,
      "step": 13373,
      "training_step_time": 17.039557933807373
    },
    {
      "epoch": 1.2754440307617188e-05,
      "model_forward_time": 5.591381549835205,
      "step": 13374
    },
    {
      "epoch": 1.2754440307617188e-05,
      "step": 13374,
      "training_step_time": 17.064351797103882
    },
    {
      "epoch": 1.2755393981933594e-05,
      "model_forward_time": 5.609787225723267,
      "step": 13375
    },
    {
      "epoch": 1.2755393981933594e-05,
      "step": 13375,
      "training_step_time": 17.109766960144043
    },
    {
      "epoch": 1.275634765625e-05,
      "model_forward_time": 5.621682167053223,
      "step": 13376
    },
    {
      "epoch": 1.275634765625e-05,
      "step": 13376,
      "training_step_time": 17.13151216506958
    },
    {
      "epoch": 1.2757301330566407e-05,
      "model_forward_time": 5.626916885375977,
      "step": 13377
    },
    {
      "epoch": 1.2757301330566407e-05,
      "step": 13377,
      "training_step_time": 17.122509717941284
    },
    {
      "epoch": 1.2758255004882813e-05,
      "model_forward_time": 5.955060720443726,
      "step": 13378
    },
    {
      "epoch": 1.2758255004882813e-05,
      "step": 13378,
      "training_step_time": 17.423193216323853
    },
    {
      "epoch": 1.2759208679199218e-05,
      "model_forward_time": 5.5951292514801025,
      "step": 13379
    },
    {
      "epoch": 1.2759208679199218e-05,
      "step": 13379,
      "training_step_time": 17.058595895767212
    },
    {
      "epoch": 1.2760162353515626e-05,
      "grad_norm": 0.6328979730606079,
      "learning_rate": 1e-05,
      "loss": 0.0498,
      "step": 13380
    },
    {
      "epoch": 1.2760162353515626e-05,
      "model_forward_time": 5.599569797515869,
      "step": 13380
    },
    {
      "dynamics_loss_avg": 0.04141966942697763,
      "epoch": 1.2760162353515626e-05,
      "step": 13380
    },
    {
      "action_loss_avg": 0.004609315516427159,
      "epoch": 1.2760162353515626e-05,
      "step": 13380
    },
    {
      "epoch": 1.2760162353515626e-05,
      "step": 13380,
      "training_step_time": 17.074522495269775
    },
    {
      "epoch": 1.2761116027832031e-05,
      "model_forward_time": 5.609774589538574,
      "step": 13381
    },
    {
      "epoch": 1.2761116027832031e-05,
      "step": 13381,
      "training_step_time": 17.098516941070557
    },
    {
      "epoch": 1.2762069702148437e-05,
      "model_forward_time": 5.6216278076171875,
      "step": 13382
    },
    {
      "epoch": 1.2762069702148437e-05,
      "step": 13382,
      "training_step_time": 17.13645052909851
    },
    {
      "epoch": 1.2763023376464843e-05,
      "model_forward_time": 6.011666297912598,
      "step": 13383
    },
    {
      "epoch": 1.2763023376464843e-05,
      "step": 13383,
      "training_step_time": 17.535147190093994
    },
    {
      "epoch": 1.276397705078125e-05,
      "model_forward_time": 5.6168718338012695,
      "step": 13384
    },
    {
      "epoch": 1.276397705078125e-05,
      "step": 13384,
      "training_step_time": 17.13735270500183
    },
    {
      "epoch": 1.2764930725097656e-05,
      "model_forward_time": 5.624910116195679,
      "step": 13385
    },
    {
      "epoch": 1.2764930725097656e-05,
      "step": 13385,
      "training_step_time": 17.10123562812805
    },
    {
      "epoch": 1.2765884399414062e-05,
      "model_forward_time": 5.59364128112793,
      "step": 13386
    },
    {
      "epoch": 1.2765884399414062e-05,
      "step": 13386,
      "training_step_time": 17.002741813659668
    },
    {
      "epoch": 1.276683807373047e-05,
      "model_forward_time": 5.595783233642578,
      "step": 13387
    },
    {
      "epoch": 1.276683807373047e-05,
      "step": 13387,
      "training_step_time": 17.02672243118286
    },
    {
      "epoch": 1.2767791748046875e-05,
      "model_forward_time": 5.574016332626343,
      "step": 13388
    },
    {
      "epoch": 1.2767791748046875e-05,
      "step": 13388,
      "training_step_time": 17.021400451660156
    },
    {
      "epoch": 1.276874542236328e-05,
      "model_forward_time": 5.928972005844116,
      "step": 13389
    },
    {
      "epoch": 1.276874542236328e-05,
      "step": 13389,
      "training_step_time": 17.348447799682617
    },
    {
      "epoch": 1.2769699096679688e-05,
      "grad_norm": 0.998734176158905,
      "learning_rate": 1e-05,
      "loss": 0.0516,
      "step": 13390
    },
    {
      "epoch": 1.2769699096679688e-05,
      "model_forward_time": 5.590169668197632,
      "step": 13390
    },
    {
      "dynamics_loss_avg": 0.04878794886171818,
      "epoch": 1.2769699096679688e-05,
      "step": 13390
    },
    {
      "action_loss_avg": 0.00583913178415969,
      "epoch": 1.2769699096679688e-05,
      "step": 13390
    },
    {
      "epoch": 1.2769699096679688e-05,
      "step": 13390,
      "training_step_time": 17.025516510009766
    },
    {
      "epoch": 1.2770652770996094e-05,
      "model_forward_time": 5.581551790237427,
      "step": 13391
    },
    {
      "epoch": 1.2770652770996094e-05,
      "step": 13391,
      "training_step_time": 17.006542205810547
    },
    {
      "epoch": 1.27716064453125e-05,
      "model_forward_time": 5.590493202209473,
      "step": 13392
    },
    {
      "epoch": 1.27716064453125e-05,
      "step": 13392,
      "training_step_time": 17.049113512039185
    },
    {
      "epoch": 1.2772560119628907e-05,
      "model_forward_time": 5.618239164352417,
      "step": 13393
    },
    {
      "epoch": 1.2772560119628907e-05,
      "step": 13393,
      "training_step_time": 17.113642930984497
    },
    {
      "epoch": 1.2773513793945313e-05,
      "model_forward_time": 5.623852252960205,
      "step": 13394
    },
    {
      "epoch": 1.2773513793945313e-05,
      "step": 13394,
      "training_step_time": 17.512294054031372
    },
    {
      "epoch": 1.2774467468261718e-05,
      "model_forward_time": 5.626326322555542,
      "step": 13395
    },
    {
      "epoch": 1.2774467468261718e-05,
      "step": 13395,
      "training_step_time": 17.14232301712036
    },
    {
      "epoch": 1.2775421142578126e-05,
      "model_forward_time": 5.621518850326538,
      "step": 13396
    },
    {
      "epoch": 1.2775421142578126e-05,
      "step": 13396,
      "training_step_time": 17.11660599708557
    },
    {
      "epoch": 1.2776374816894531e-05,
      "model_forward_time": 5.607865810394287,
      "step": 13397
    },
    {
      "epoch": 1.2776374816894531e-05,
      "step": 13397,
      "training_step_time": 17.05955171585083
    },
    {
      "epoch": 1.2777328491210937e-05,
      "model_forward_time": 5.590234041213989,
      "step": 13398
    },
    {
      "epoch": 1.2777328491210937e-05,
      "step": 13398,
      "training_step_time": 17.008673429489136
    },
    {
      "epoch": 1.2778282165527343e-05,
      "model_forward_time": 5.584304332733154,
      "step": 13399
    },
    {
      "epoch": 1.2778282165527343e-05,
      "step": 13399,
      "training_step_time": 17.03246784210205
    },
    {
      "epoch": 1.277923583984375e-05,
      "grad_norm": 0.37950336933135986,
      "learning_rate": 1e-05,
      "loss": 0.0622,
      "step": 13400
    },
    {
      "epoch": 1.277923583984375e-05,
      "model_forward_time": 5.937988758087158,
      "step": 13400
    },
    {
      "dynamics_loss_avg": 0.0586225088685751,
      "epoch": 1.277923583984375e-05,
      "step": 13400
    },
    {
      "action_loss_avg": 0.008513306197710336,
      "epoch": 1.277923583984375e-05,
      "step": 13400
    },
    {
      "epoch": 1.277923583984375e-05,
      "step": 13400,
      "training_step_time": 17.38228178024292
    },
    {
      "epoch": 1.2780189514160156e-05,
      "model_forward_time": 5.603631019592285,
      "step": 13401
    },
    {
      "epoch": 1.2780189514160156e-05,
      "step": 13401,
      "training_step_time": 17.062642097473145
    },
    {
      "epoch": 1.2781143188476562e-05,
      "model_forward_time": 5.619571924209595,
      "step": 13402
    },
    {
      "epoch": 1.2781143188476562e-05,
      "step": 13402,
      "training_step_time": 17.10205364227295
    },
    {
      "epoch": 1.278209686279297e-05,
      "model_forward_time": 5.60998797416687,
      "step": 13403
    },
    {
      "epoch": 1.278209686279297e-05,
      "step": 13403,
      "training_step_time": 17.093466758728027
    },
    {
      "epoch": 1.2783050537109375e-05,
      "model_forward_time": 5.612973213195801,
      "step": 13404
    },
    {
      "epoch": 1.2783050537109375e-05,
      "step": 13404,
      "training_step_time": 17.117490530014038
    },
    {
      "epoch": 1.278400421142578e-05,
      "model_forward_time": 6.002642631530762,
      "step": 13405
    },
    {
      "epoch": 1.278400421142578e-05,
      "step": 13405,
      "training_step_time": 17.508832216262817
    },
    {
      "epoch": 1.2784957885742188e-05,
      "model_forward_time": 5.597102403640747,
      "step": 13406
    },
    {
      "epoch": 1.2784957885742188e-05,
      "step": 13406,
      "training_step_time": 17.05046820640564
    },
    {
      "epoch": 1.2785911560058594e-05,
      "model_forward_time": 5.599810361862183,
      "step": 13407
    },
    {
      "epoch": 1.2785911560058594e-05,
      "step": 13407,
      "training_step_time": 17.03619933128357
    },
    {
      "epoch": 1.2786865234375e-05,
      "model_forward_time": 5.592258930206299,
      "step": 13408
    },
    {
      "epoch": 1.2786865234375e-05,
      "step": 13408,
      "training_step_time": 17.04944896697998
    },
    {
      "epoch": 1.2787818908691407e-05,
      "model_forward_time": 5.6096086502075195,
      "step": 13409
    },
    {
      "epoch": 1.2787818908691407e-05,
      "step": 13409,
      "training_step_time": 17.09262466430664
    },
    {
      "epoch": 1.2788772583007813e-05,
      "grad_norm": 0.28838226199150085,
      "learning_rate": 1e-05,
      "loss": 0.0604,
      "step": 13410
    },
    {
      "epoch": 1.2788772583007813e-05,
      "model_forward_time": 5.621469497680664,
      "step": 13410
    },
    {
      "dynamics_loss_avg": 0.049187590926885606,
      "epoch": 1.2788772583007813e-05,
      "step": 13410
    },
    {
      "action_loss_avg": 0.00448011226253584,
      "epoch": 1.2788772583007813e-05,
      "step": 13410
    },
    {
      "epoch": 1.2788772583007813e-05,
      "step": 13410,
      "training_step_time": 17.129419088363647
    },
    {
      "epoch": 1.2789726257324218e-05,
      "model_forward_time": 5.956520080566406,
      "step": 13411
    },
    {
      "epoch": 1.2789726257324218e-05,
      "step": 13411,
      "training_step_time": 17.447301626205444
    },
    {
      "epoch": 1.2790679931640626e-05,
      "model_forward_time": 5.616894245147705,
      "step": 13412
    },
    {
      "epoch": 1.2790679931640626e-05,
      "step": 13412,
      "training_step_time": 17.105234622955322
    },
    {
      "epoch": 1.2791633605957031e-05,
      "model_forward_time": 5.602349042892456,
      "step": 13413
    },
    {
      "epoch": 1.2791633605957031e-05,
      "step": 13413,
      "training_step_time": 17.076658964157104
    },
    {
      "epoch": 1.2792587280273437e-05,
      "model_forward_time": 5.6092164516448975,
      "step": 13414
    },
    {
      "epoch": 1.2792587280273437e-05,
      "step": 13414,
      "training_step_time": 17.058589220046997
    },
    {
      "epoch": 1.2793540954589843e-05,
      "model_forward_time": 5.596941709518433,
      "step": 13415
    },
    {
      "epoch": 1.2793540954589843e-05,
      "step": 13415,
      "training_step_time": 17.036844968795776
    },
    {
      "epoch": 1.279449462890625e-05,
      "model_forward_time": 5.962292432785034,
      "step": 13416
    },
    {
      "epoch": 1.279449462890625e-05,
      "step": 13416,
      "training_step_time": 17.402878761291504
    },
    {
      "epoch": 1.2795448303222656e-05,
      "model_forward_time": 5.598302364349365,
      "step": 13417
    },
    {
      "epoch": 1.2795448303222656e-05,
      "step": 13417,
      "training_step_time": 17.076021194458008
    },
    {
      "epoch": 1.2796401977539062e-05,
      "model_forward_time": 5.621067523956299,
      "step": 13418
    },
    {
      "epoch": 1.2796401977539062e-05,
      "step": 13418,
      "training_step_time": 17.134043216705322
    },
    {
      "epoch": 1.279735565185547e-05,
      "model_forward_time": 5.632851839065552,
      "step": 13419
    },
    {
      "epoch": 1.279735565185547e-05,
      "step": 13419,
      "training_step_time": 17.155539751052856
    },
    {
      "epoch": 1.2798309326171875e-05,
      "grad_norm": 0.6634788513183594,
      "learning_rate": 1e-05,
      "loss": 0.0552,
      "step": 13420
    },
    {
      "epoch": 1.2798309326171875e-05,
      "model_forward_time": 5.630166530609131,
      "step": 13420
    },
    {
      "dynamics_loss_avg": 0.04831613264977932,
      "epoch": 1.2798309326171875e-05,
      "step": 13420
    },
    {
      "action_loss_avg": 0.00588302641408518,
      "epoch": 1.2798309326171875e-05,
      "step": 13420
    },
    {
      "epoch": 1.2798309326171875e-05,
      "step": 13420,
      "training_step_time": 17.149429082870483
    },
    {
      "epoch": 1.279926300048828e-05,
      "model_forward_time": 5.623677015304565,
      "step": 13421
    },
    {
      "epoch": 1.279926300048828e-05,
      "step": 13421,
      "training_step_time": 17.134097576141357
    },
    {
      "epoch": 1.2800216674804688e-05,
      "model_forward_time": 5.9648566246032715,
      "step": 13422
    },
    {
      "epoch": 1.2800216674804688e-05,
      "step": 13422,
      "training_step_time": 17.43851900100708
    },
    {
      "epoch": 1.2801170349121094e-05,
      "model_forward_time": 5.591002941131592,
      "step": 13423
    },
    {
      "epoch": 1.2801170349121094e-05,
      "step": 13423,
      "training_step_time": 17.02156114578247
    },
    {
      "epoch": 1.28021240234375e-05,
      "model_forward_time": 5.612118244171143,
      "step": 13424
    },
    {
      "epoch": 1.28021240234375e-05,
      "step": 13424,
      "training_step_time": 17.04264545440674
    },
    {
      "epoch": 1.2803077697753907e-05,
      "model_forward_time": 5.6172919273376465,
      "step": 13425
    },
    {
      "epoch": 1.2803077697753907e-05,
      "step": 13425,
      "training_step_time": 17.090752124786377
    },
    {
      "epoch": 1.2804031372070313e-05,
      "model_forward_time": 5.615242004394531,
      "step": 13426
    },
    {
      "epoch": 1.2804031372070313e-05,
      "step": 13426,
      "training_step_time": 17.08412456512451
    },
    {
      "epoch": 1.2804985046386718e-05,
      "model_forward_time": 5.599444389343262,
      "step": 13427
    },
    {
      "epoch": 1.2804985046386718e-05,
      "step": 13427,
      "training_step_time": 17.412556409835815
    },
    {
      "epoch": 1.2805938720703126e-05,
      "model_forward_time": 5.590398550033569,
      "step": 13428
    },
    {
      "epoch": 1.2805938720703126e-05,
      "step": 13428,
      "training_step_time": 17.056071043014526
    },
    {
      "epoch": 1.2806892395019532e-05,
      "model_forward_time": 5.598376512527466,
      "step": 13429
    },
    {
      "epoch": 1.2806892395019532e-05,
      "step": 13429,
      "training_step_time": 17.102645874023438
    },
    {
      "epoch": 1.2807846069335937e-05,
      "grad_norm": 0.6860260963439941,
      "learning_rate": 1e-05,
      "loss": 0.0511,
      "step": 13430
    },
    {
      "epoch": 1.2807846069335937e-05,
      "model_forward_time": 5.616735219955444,
      "step": 13430
    },
    {
      "dynamics_loss_avg": 0.04802335929125547,
      "epoch": 1.2807846069335937e-05,
      "step": 13430
    },
    {
      "action_loss_avg": 0.00504419319331646,
      "epoch": 1.2807846069335937e-05,
      "step": 13430
    },
    {
      "epoch": 1.2807846069335937e-05,
      "step": 13430,
      "training_step_time": 17.147408962249756
    },
    {
      "epoch": 1.2808799743652343e-05,
      "model_forward_time": 5.618128061294556,
      "step": 13431
    },
    {
      "epoch": 1.2808799743652343e-05,
      "step": 13431,
      "training_step_time": 17.11675477027893
    },
    {
      "epoch": 1.280975341796875e-05,
      "model_forward_time": 5.608327865600586,
      "step": 13432
    },
    {
      "epoch": 1.280975341796875e-05,
      "step": 13432,
      "training_step_time": 17.082773447036743
    },
    {
      "epoch": 1.2810707092285156e-05,
      "model_forward_time": 5.9640538692474365,
      "step": 13433
    },
    {
      "epoch": 1.2810707092285156e-05,
      "step": 13433,
      "training_step_time": 17.414295434951782
    },
    {
      "epoch": 1.2811660766601562e-05,
      "model_forward_time": 5.595768690109253,
      "step": 13434
    },
    {
      "epoch": 1.2811660766601562e-05,
      "step": 13434,
      "training_step_time": 17.048389673233032
    },
    {
      "epoch": 1.281261444091797e-05,
      "model_forward_time": 5.594161510467529,
      "step": 13435
    },
    {
      "epoch": 1.281261444091797e-05,
      "step": 13435,
      "training_step_time": 17.03717541694641
    },
    {
      "epoch": 1.2813568115234375e-05,
      "model_forward_time": 5.589732646942139,
      "step": 13436
    },
    {
      "epoch": 1.2813568115234375e-05,
      "step": 13436,
      "training_step_time": 17.02024817466736
    },
    {
      "epoch": 1.281452178955078e-05,
      "model_forward_time": 5.5853447914123535,
      "step": 13437
    },
    {
      "epoch": 1.281452178955078e-05,
      "step": 13437,
      "training_step_time": 17.056922674179077
    },
    {
      "epoch": 1.2815475463867188e-05,
      "model_forward_time": 5.98872447013855,
      "step": 13438
    },
    {
      "epoch": 1.2815475463867188e-05,
      "step": 13438,
      "training_step_time": 17.466979503631592
    },
    {
      "epoch": 1.2816429138183594e-05,
      "model_forward_time": 5.619624853134155,
      "step": 13439
    },
    {
      "epoch": 1.2816429138183594e-05,
      "step": 13439,
      "training_step_time": 17.139049530029297
    },
    {
      "epoch": 1.28173828125e-05,
      "grad_norm": 0.7091267108917236,
      "learning_rate": 1e-05,
      "loss": 0.0744,
      "step": 13440
    },
    {
      "epoch": 1.28173828125e-05,
      "model_forward_time": 5.62749719619751,
      "step": 13440
    },
    {
      "dynamics_loss_avg": 0.06662240289151669,
      "epoch": 1.28173828125e-05,
      "step": 13440
    },
    {
      "action_loss_avg": 0.010226265201345086,
      "epoch": 1.28173828125e-05,
      "step": 13440
    },
    {
      "epoch": 1.28173828125e-05,
      "step": 13440,
      "training_step_time": 17.11780548095703
    },
    {
      "epoch": 1.2818336486816407e-05,
      "model_forward_time": 5.606042385101318,
      "step": 13441
    },
    {
      "epoch": 1.2818336486816407e-05,
      "step": 13441,
      "training_step_time": 17.092941761016846
    },
    {
      "epoch": 1.2819290161132813e-05,
      "model_forward_time": 5.6110148429870605,
      "step": 13442
    },
    {
      "epoch": 1.2819290161132813e-05,
      "step": 13442,
      "training_step_time": 17.099228858947754
    },
    {
      "epoch": 1.2820243835449218e-05,
      "model_forward_time": 5.616028785705566,
      "step": 13443
    },
    {
      "epoch": 1.2820243835449218e-05,
      "step": 13443,
      "training_step_time": 17.102071523666382
    },
    {
      "epoch": 1.2821197509765626e-05,
      "model_forward_time": 5.9622979164123535,
      "step": 13444
    },
    {
      "epoch": 1.2821197509765626e-05,
      "step": 13444,
      "training_step_time": 17.433731079101562
    },
    {
      "epoch": 1.2822151184082032e-05,
      "model_forward_time": 5.608416795730591,
      "step": 13445
    },
    {
      "epoch": 1.2822151184082032e-05,
      "step": 13445,
      "training_step_time": 17.081640243530273
    },
    {
      "epoch": 1.2823104858398437e-05,
      "model_forward_time": 5.611711740493774,
      "step": 13446
    },
    {
      "epoch": 1.2823104858398437e-05,
      "step": 13446,
      "training_step_time": 17.097975492477417
    },
    {
      "epoch": 1.2824058532714843e-05,
      "model_forward_time": 5.627722978591919,
      "step": 13447
    },
    {
      "epoch": 1.2824058532714843e-05,
      "step": 13447,
      "training_step_time": 17.12005043029785
    },
    {
      "epoch": 1.282501220703125e-05,
      "model_forward_time": 5.607082366943359,
      "step": 13448
    },
    {
      "epoch": 1.282501220703125e-05,
      "step": 13448,
      "training_step_time": 17.07658314704895
    },
    {
      "epoch": 1.2825965881347656e-05,
      "model_forward_time": 5.982890844345093,
      "step": 13449
    },
    {
      "epoch": 1.2825965881347656e-05,
      "step": 13449,
      "training_step_time": 17.45844602584839
    },
    {
      "epoch": 1.2826919555664062e-05,
      "grad_norm": 0.6894009113311768,
      "learning_rate": 1e-05,
      "loss": 0.0697,
      "step": 13450
    },
    {
      "epoch": 1.2826919555664062e-05,
      "model_forward_time": 5.598881959915161,
      "step": 13450
    },
    {
      "dynamics_loss_avg": 0.06106631364673376,
      "epoch": 1.2826919555664062e-05,
      "step": 13450
    },
    {
      "action_loss_avg": 0.006985744368284941,
      "epoch": 1.2826919555664062e-05,
      "step": 13450
    },
    {
      "epoch": 1.2826919555664062e-05,
      "step": 13450,
      "training_step_time": 17.052507162094116
    },
    {
      "epoch": 1.282787322998047e-05,
      "model_forward_time": 5.588455677032471,
      "step": 13451
    },
    {
      "epoch": 1.282787322998047e-05,
      "step": 13451,
      "training_step_time": 17.04324960708618
    },
    {
      "epoch": 1.2828826904296875e-05,
      "model_forward_time": 5.5994837284088135,
      "step": 13452
    },
    {
      "epoch": 1.2828826904296875e-05,
      "step": 13452,
      "training_step_time": 17.05325412750244
    },
    {
      "epoch": 1.282978057861328e-05,
      "model_forward_time": 5.594884157180786,
      "step": 13453
    },
    {
      "epoch": 1.282978057861328e-05,
      "step": 13453,
      "training_step_time": 17.06406021118164
    },
    {
      "epoch": 1.2830734252929688e-05,
      "model_forward_time": 5.609642267227173,
      "step": 13454
    },
    {
      "epoch": 1.2830734252929688e-05,
      "step": 13454,
      "training_step_time": 17.11542844772339
    },
    {
      "epoch": 1.2831687927246094e-05,
      "model_forward_time": 5.950887680053711,
      "step": 13455
    },
    {
      "epoch": 1.2831687927246094e-05,
      "step": 13455,
      "training_step_time": 17.44769811630249
    },
    {
      "epoch": 1.28326416015625e-05,
      "model_forward_time": 5.610414505004883,
      "step": 13456
    },
    {
      "epoch": 1.28326416015625e-05,
      "step": 13456,
      "training_step_time": 17.120126008987427
    },
    {
      "epoch": 1.2833595275878907e-05,
      "model_forward_time": 5.611480236053467,
      "step": 13457
    },
    {
      "epoch": 1.2833595275878907e-05,
      "step": 13457,
      "training_step_time": 17.105170249938965
    },
    {
      "epoch": 1.2834548950195313e-05,
      "model_forward_time": 5.614339113235474,
      "step": 13458
    },
    {
      "epoch": 1.2834548950195313e-05,
      "step": 13458,
      "training_step_time": 17.10773992538452
    },
    {
      "epoch": 1.2835502624511718e-05,
      "model_forward_time": 5.609338998794556,
      "step": 13459
    },
    {
      "epoch": 1.2835502624511718e-05,
      "step": 13459,
      "training_step_time": 17.104085445404053
    },
    {
      "epoch": 1.2836456298828126e-05,
      "grad_norm": 0.35234999656677246,
      "learning_rate": 1e-05,
      "loss": 0.0666,
      "step": 13460
    },
    {
      "epoch": 1.2836456298828126e-05,
      "model_forward_time": 5.981403350830078,
      "step": 13460
    },
    {
      "dynamics_loss_avg": 0.058104560151696204,
      "epoch": 1.2836456298828126e-05,
      "step": 13460
    },
    {
      "action_loss_avg": 0.007620589155703783,
      "epoch": 1.2836456298828126e-05,
      "step": 13460
    },
    {
      "epoch": 1.2836456298828126e-05,
      "step": 13460,
      "training_step_time": 17.462183237075806
    },
    {
      "epoch": 1.2837409973144532e-05,
      "model_forward_time": 5.598629713058472,
      "step": 13461
    },
    {
      "epoch": 1.2837409973144532e-05,
      "step": 13461,
      "training_step_time": 17.015331268310547
    },
    {
      "epoch": 1.2838363647460937e-05,
      "model_forward_time": 5.581652879714966,
      "step": 13462
    },
    {
      "epoch": 1.2838363647460937e-05,
      "step": 13462,
      "training_step_time": 17.021848440170288
    },
    {
      "epoch": 1.2839317321777343e-05,
      "model_forward_time": 5.595892667770386,
      "step": 13463
    },
    {
      "epoch": 1.2839317321777343e-05,
      "step": 13463,
      "training_step_time": 17.043250560760498
    },
    {
      "epoch": 1.284027099609375e-05,
      "model_forward_time": 5.597886562347412,
      "step": 13464
    },
    {
      "epoch": 1.284027099609375e-05,
      "step": 13464,
      "training_step_time": 17.044235706329346
    },
    {
      "epoch": 1.2841224670410156e-05,
      "model_forward_time": 5.58121657371521,
      "step": 13465
    },
    {
      "epoch": 1.2841224670410156e-05,
      "step": 13465,
      "training_step_time": 16.987274885177612
    },
    {
      "epoch": 1.2842178344726562e-05,
      "model_forward_time": 5.9413275718688965,
      "step": 13466
    },
    {
      "epoch": 1.2842178344726562e-05,
      "step": 13466,
      "training_step_time": 17.367758750915527
    },
    {
      "epoch": 1.284313201904297e-05,
      "model_forward_time": 5.5808210372924805,
      "step": 13467
    },
    {
      "epoch": 1.284313201904297e-05,
      "step": 13467,
      "training_step_time": 17.004194974899292
    },
    {
      "epoch": 1.2844085693359375e-05,
      "model_forward_time": 5.59097146987915,
      "step": 13468
    },
    {
      "epoch": 1.2844085693359375e-05,
      "step": 13468,
      "training_step_time": 17.060450077056885
    },
    {
      "epoch": 1.284503936767578e-05,
      "model_forward_time": 5.602851152420044,
      "step": 13469
    },
    {
      "epoch": 1.284503936767578e-05,
      "step": 13469,
      "training_step_time": 17.09863018989563
    },
    {
      "epoch": 1.2845993041992188e-05,
      "grad_norm": 0.6290714144706726,
      "learning_rate": 1e-05,
      "loss": 0.0537,
      "step": 13470
    },
    {
      "epoch": 1.2845993041992188e-05,
      "model_forward_time": 5.622409820556641,
      "step": 13470
    },
    {
      "dynamics_loss_avg": 0.049727586656808855,
      "epoch": 1.2845993041992188e-05,
      "step": 13470
    },
    {
      "action_loss_avg": 0.0065441280021332204,
      "epoch": 1.2845993041992188e-05,
      "step": 13470
    },
    {
      "epoch": 1.2845993041992188e-05,
      "step": 13470,
      "training_step_time": 17.11545968055725
    },
    {
      "epoch": 1.2846946716308594e-05,
      "model_forward_time": 6.001488208770752,
      "step": 13471
    },
    {
      "epoch": 1.2846946716308594e-05,
      "step": 13471,
      "training_step_time": 17.486498832702637
    },
    {
      "epoch": 1.2847900390625e-05,
      "model_forward_time": 5.60774040222168,
      "step": 13472
    },
    {
      "epoch": 1.2847900390625e-05,
      "step": 13472,
      "training_step_time": 17.0869779586792
    },
    {
      "epoch": 1.2848854064941407e-05,
      "model_forward_time": 5.6068031787872314,
      "step": 13473
    },
    {
      "epoch": 1.2848854064941407e-05,
      "step": 13473,
      "training_step_time": 17.065314054489136
    },
    {
      "epoch": 1.2849807739257813e-05,
      "model_forward_time": 5.59383225440979,
      "step": 13474
    },
    {
      "epoch": 1.2849807739257813e-05,
      "step": 13474,
      "training_step_time": 17.029478788375854
    },
    {
      "epoch": 1.2850761413574218e-05,
      "model_forward_time": 5.593941688537598,
      "step": 13475
    },
    {
      "epoch": 1.2850761413574218e-05,
      "step": 13475,
      "training_step_time": 17.026591062545776
    },
    {
      "epoch": 1.2851715087890626e-05,
      "model_forward_time": 5.960713148117065,
      "step": 13476
    },
    {
      "epoch": 1.2851715087890626e-05,
      "step": 13476,
      "training_step_time": 17.409656286239624
    },
    {
      "epoch": 1.2852668762207032e-05,
      "model_forward_time": 5.595769882202148,
      "step": 13477
    },
    {
      "epoch": 1.2852668762207032e-05,
      "step": 13477,
      "training_step_time": 17.06544256210327
    },
    {
      "epoch": 1.2853622436523437e-05,
      "model_forward_time": 5.603224992752075,
      "step": 13478
    },
    {
      "epoch": 1.2853622436523437e-05,
      "step": 13478,
      "training_step_time": 17.078421115875244
    },
    {
      "epoch": 1.2854576110839843e-05,
      "model_forward_time": 5.6296586990356445,
      "step": 13479
    },
    {
      "epoch": 1.2854576110839843e-05,
      "step": 13479,
      "training_step_time": 17.15852928161621
    },
    {
      "epoch": 1.285552978515625e-05,
      "grad_norm": 0.7050377726554871,
      "learning_rate": 1e-05,
      "loss": 0.071,
      "step": 13480
    },
    {
      "epoch": 1.285552978515625e-05,
      "model_forward_time": 5.636531114578247,
      "step": 13480
    },
    {
      "dynamics_loss_avg": 0.05982260201126337,
      "epoch": 1.285552978515625e-05,
      "step": 13480
    },
    {
      "action_loss_avg": 0.009598830109462142,
      "epoch": 1.285552978515625e-05,
      "step": 13480
    },
    {
      "epoch": 1.285552978515625e-05,
      "step": 13480,
      "training_step_time": 17.129862308502197
    },
    {
      "epoch": 1.2856483459472656e-05,
      "model_forward_time": 5.620697498321533,
      "step": 13481
    },
    {
      "epoch": 1.2856483459472656e-05,
      "step": 13481,
      "training_step_time": 17.119160652160645
    },
    {
      "epoch": 1.2857437133789062e-05,
      "model_forward_time": 6.0018699169158936,
      "step": 13482
    },
    {
      "epoch": 1.2857437133789062e-05,
      "step": 13482,
      "training_step_time": 17.481070518493652
    },
    {
      "epoch": 1.285839080810547e-05,
      "model_forward_time": 5.606546878814697,
      "step": 13483
    },
    {
      "epoch": 1.285839080810547e-05,
      "step": 13483,
      "training_step_time": 17.06374192237854
    },
    {
      "epoch": 1.2859344482421875e-05,
      "model_forward_time": 5.60343599319458,
      "step": 13484
    },
    {
      "epoch": 1.2859344482421875e-05,
      "step": 13484,
      "training_step_time": 17.078020811080933
    },
    {
      "epoch": 1.286029815673828e-05,
      "model_forward_time": 5.608452796936035,
      "step": 13485
    },
    {
      "epoch": 1.286029815673828e-05,
      "step": 13485,
      "training_step_time": 17.09054446220398
    },
    {
      "epoch": 1.2861251831054688e-05,
      "model_forward_time": 5.604635238647461,
      "step": 13486
    },
    {
      "epoch": 1.2861251831054688e-05,
      "step": 13486,
      "training_step_time": 17.071077823638916
    },
    {
      "epoch": 1.2862205505371094e-05,
      "model_forward_time": 5.601099252700806,
      "step": 13487
    },
    {
      "epoch": 1.2862205505371094e-05,
      "step": 13487,
      "training_step_time": 17.44926619529724
    },
    {
      "epoch": 1.28631591796875e-05,
      "model_forward_time": 5.607954502105713,
      "step": 13488
    },
    {
      "epoch": 1.28631591796875e-05,
      "step": 13488,
      "training_step_time": 17.10020923614502
    },
    {
      "epoch": 1.2864112854003907e-05,
      "model_forward_time": 5.609381914138794,
      "step": 13489
    },
    {
      "epoch": 1.2864112854003907e-05,
      "step": 13489,
      "training_step_time": 17.089075326919556
    },
    {
      "epoch": 1.2865066528320313e-05,
      "grad_norm": 0.9518016576766968,
      "learning_rate": 1e-05,
      "loss": 0.0751,
      "step": 13490
    },
    {
      "epoch": 1.2865066528320313e-05,
      "model_forward_time": 5.608428001403809,
      "step": 13490
    },
    {
      "dynamics_loss_avg": 0.07009266391396522,
      "epoch": 1.2865066528320313e-05,
      "step": 13490
    },
    {
      "action_loss_avg": 0.007815409149043263,
      "epoch": 1.2865066528320313e-05,
      "step": 13490
    },
    {
      "epoch": 1.2865066528320313e-05,
      "step": 13490,
      "training_step_time": 17.046852588653564
    },
    {
      "epoch": 1.2866020202636718e-05,
      "model_forward_time": 5.583684921264648,
      "step": 13491
    },
    {
      "epoch": 1.2866020202636718e-05,
      "step": 13491,
      "training_step_time": 17.04163908958435
    },
    {
      "epoch": 1.2866973876953126e-05,
      "model_forward_time": 5.598351240158081,
      "step": 13492
    },
    {
      "epoch": 1.2866973876953126e-05,
      "step": 13492,
      "training_step_time": 17.078365087509155
    },
    {
      "epoch": 1.2867927551269532e-05,
      "model_forward_time": 5.980320930480957,
      "step": 13493
    },
    {
      "epoch": 1.2867927551269532e-05,
      "step": 13493,
      "training_step_time": 17.45791983604431
    },
    {
      "epoch": 1.2868881225585937e-05,
      "model_forward_time": 5.61470890045166,
      "step": 13494
    },
    {
      "epoch": 1.2868881225585937e-05,
      "step": 13494,
      "training_step_time": 17.107539892196655
    },
    {
      "epoch": 1.2869834899902343e-05,
      "model_forward_time": 5.608191013336182,
      "step": 13495
    },
    {
      "epoch": 1.2869834899902343e-05,
      "step": 13495,
      "training_step_time": 17.09671926498413
    },
    {
      "epoch": 1.287078857421875e-05,
      "model_forward_time": 5.613141298294067,
      "step": 13496
    },
    {
      "epoch": 1.287078857421875e-05,
      "step": 13496,
      "training_step_time": 17.114492654800415
    },
    {
      "epoch": 1.2871742248535156e-05,
      "model_forward_time": 5.653172254562378,
      "step": 13497
    },
    {
      "epoch": 1.2871742248535156e-05,
      "step": 13497,
      "training_step_time": 17.162336111068726
    },
    {
      "epoch": 1.2872695922851562e-05,
      "model_forward_time": 5.6255645751953125,
      "step": 13498
    },
    {
      "epoch": 1.2872695922851562e-05,
      "step": 13498,
      "training_step_time": 17.485328435897827
    },
    {
      "epoch": 1.287364959716797e-05,
      "model_forward_time": 5.6304030418396,
      "step": 13499
    },
    {
      "epoch": 1.287364959716797e-05,
      "step": 13499,
      "training_step_time": 17.169369220733643
    },
    {
      "epoch": 1.2874603271484375e-05,
      "grad_norm": 0.47066789865493774,
      "learning_rate": 1e-05,
      "loss": 0.0633,
      "step": 13500
    },
    {
      "epoch": 1.2874603271484375e-05,
      "model_forward_time": 5.648780345916748,
      "step": 13500
    },
    {
      "dynamics_loss_avg": 0.05205837171524763,
      "epoch": 1.2874603271484375e-05,
      "step": 13500
    },
    {
      "action_loss_avg": 0.005790960066951812,
      "epoch": 1.2874603271484375e-05,
      "step": 13500
    },
    {
      "epoch": 1.2874603271484375e-05,
      "step": 13500,
      "training_step_time": 17.213906049728394
    },
    {
      "epoch": 1.287555694580078e-05,
      "model_forward_time": 5.625790357589722,
      "step": 13501
    },
    {
      "epoch": 1.287555694580078e-05,
      "step": 13501,
      "training_step_time": 17.132941007614136
    },
    {
      "epoch": 1.2876510620117188e-05,
      "model_forward_time": 5.6185832023620605,
      "step": 13502
    },
    {
      "epoch": 1.2876510620117188e-05,
      "step": 13502,
      "training_step_time": 17.100013971328735
    },
    {
      "epoch": 1.2877464294433594e-05,
      "model_forward_time": 5.586818695068359,
      "step": 13503
    },
    {
      "epoch": 1.2877464294433594e-05,
      "step": 13503,
      "training_step_time": 17.01607894897461
    },
    {
      "epoch": 1.287841796875e-05,
      "model_forward_time": 5.940288305282593,
      "step": 13504
    },
    {
      "epoch": 1.287841796875e-05,
      "step": 13504,
      "training_step_time": 17.35142469406128
    },
    {
      "epoch": 1.2879371643066407e-05,
      "model_forward_time": 5.573577165603638,
      "step": 13505
    },
    {
      "epoch": 1.2879371643066407e-05,
      "step": 13505,
      "training_step_time": 17.005582332611084
    },
    {
      "epoch": 1.2880325317382813e-05,
      "model_forward_time": 5.578867197036743,
      "step": 13506
    },
    {
      "epoch": 1.2880325317382813e-05,
      "step": 13506,
      "training_step_time": 17.001661777496338
    },
    {
      "epoch": 1.2881278991699218e-05,
      "model_forward_time": 5.577800035476685,
      "step": 13507
    },
    {
      "epoch": 1.2881278991699218e-05,
      "step": 13507,
      "training_step_time": 16.993970155715942
    },
    {
      "epoch": 1.2882232666015624e-05,
      "model_forward_time": 5.5923919677734375,
      "step": 13508
    },
    {
      "epoch": 1.2882232666015624e-05,
      "step": 13508,
      "training_step_time": 17.00450348854065
    },
    {
      "epoch": 1.2883186340332032e-05,
      "model_forward_time": 5.944164037704468,
      "step": 13509
    },
    {
      "epoch": 1.2883186340332032e-05,
      "step": 13509,
      "training_step_time": 17.39059543609619
    },
    {
      "epoch": 1.2884140014648437e-05,
      "grad_norm": 0.24266144633293152,
      "learning_rate": 1e-05,
      "loss": 0.0567,
      "step": 13510
    },
    {
      "epoch": 1.2884140014648437e-05,
      "model_forward_time": 5.576995849609375,
      "step": 13510
    },
    {
      "dynamics_loss_avg": 0.04774860460311174,
      "epoch": 1.2884140014648437e-05,
      "step": 13510
    },
    {
      "action_loss_avg": 0.0075524861458688974,
      "epoch": 1.2884140014648437e-05,
      "step": 13510
    },
    {
      "epoch": 1.2884140014648437e-05,
      "step": 13510,
      "training_step_time": 17.01807141304016
    },
    {
      "epoch": 1.2885093688964843e-05,
      "model_forward_time": 5.614661455154419,
      "step": 13511
    },
    {
      "epoch": 1.2885093688964843e-05,
      "step": 13511,
      "training_step_time": 17.069686889648438
    },
    {
      "epoch": 1.288604736328125e-05,
      "model_forward_time": 5.598722219467163,
      "step": 13512
    },
    {
      "epoch": 1.288604736328125e-05,
      "step": 13512,
      "training_step_time": 17.07753896713257
    },
    {
      "epoch": 1.2887001037597656e-05,
      "model_forward_time": 5.607094764709473,
      "step": 13513
    },
    {
      "epoch": 1.2887001037597656e-05,
      "step": 13513,
      "training_step_time": 17.098204135894775
    },
    {
      "epoch": 1.2887954711914062e-05,
      "model_forward_time": 5.61507773399353,
      "step": 13514
    },
    {
      "epoch": 1.2887954711914062e-05,
      "step": 13514,
      "training_step_time": 17.122735738754272
    },
    {
      "epoch": 1.288890838623047e-05,
      "model_forward_time": 5.998072624206543,
      "step": 13515
    },
    {
      "epoch": 1.288890838623047e-05,
      "step": 13515,
      "training_step_time": 17.476513624191284
    },
    {
      "epoch": 1.2889862060546875e-05,
      "model_forward_time": 5.597120761871338,
      "step": 13516
    },
    {
      "epoch": 1.2889862060546875e-05,
      "step": 13516,
      "training_step_time": 17.016159057617188
    },
    {
      "epoch": 1.289081573486328e-05,
      "model_forward_time": 5.5850958824157715,
      "step": 13517
    },
    {
      "epoch": 1.289081573486328e-05,
      "step": 13517,
      "training_step_time": 17.0468692779541
    },
    {
      "epoch": 1.2891769409179688e-05,
      "model_forward_time": 5.571640729904175,
      "step": 13518
    },
    {
      "epoch": 1.2891769409179688e-05,
      "step": 13518,
      "training_step_time": 16.981038093566895
    },
    {
      "epoch": 1.2892723083496094e-05,
      "model_forward_time": 5.588509798049927,
      "step": 13519
    },
    {
      "epoch": 1.2892723083496094e-05,
      "step": 13519,
      "training_step_time": 17.031434774398804
    },
    {
      "epoch": 1.28936767578125e-05,
      "grad_norm": 1.007001519203186,
      "learning_rate": 1e-05,
      "loss": 0.0682,
      "step": 13520
    },
    {
      "epoch": 1.28936767578125e-05,
      "model_forward_time": 5.601186275482178,
      "step": 13520
    },
    {
      "dynamics_loss_avg": 0.06526126619428396,
      "epoch": 1.28936767578125e-05,
      "step": 13520
    },
    {
      "action_loss_avg": 0.009732118615647777,
      "epoch": 1.28936767578125e-05,
      "step": 13520
    },
    {
      "epoch": 1.28936767578125e-05,
      "step": 13520,
      "training_step_time": 17.441096305847168
    },
    {
      "epoch": 1.2894630432128907e-05,
      "model_forward_time": 5.615650415420532,
      "step": 13521
    },
    {
      "epoch": 1.2894630432128907e-05,
      "step": 13521,
      "training_step_time": 17.134220838546753
    },
    {
      "epoch": 1.2895584106445313e-05,
      "model_forward_time": 5.630619525909424,
      "step": 13522
    },
    {
      "epoch": 1.2895584106445313e-05,
      "step": 13522,
      "training_step_time": 17.139214992523193
    },
    {
      "epoch": 1.2896537780761718e-05,
      "model_forward_time": 5.614912986755371,
      "step": 13523
    },
    {
      "epoch": 1.2896537780761718e-05,
      "step": 13523,
      "training_step_time": 17.096335649490356
    },
    {
      "epoch": 1.2897491455078124e-05,
      "model_forward_time": 5.607180595397949,
      "step": 13524
    },
    {
      "epoch": 1.2897491455078124e-05,
      "step": 13524,
      "training_step_time": 17.094449996948242
    },
    {
      "epoch": 1.2898445129394532e-05,
      "model_forward_time": 5.60564923286438,
      "step": 13525
    },
    {
      "epoch": 1.2898445129394532e-05,
      "step": 13525,
      "training_step_time": 17.074626445770264
    },
    {
      "epoch": 1.2899398803710937e-05,
      "model_forward_time": 5.973064661026001,
      "step": 13526
    },
    {
      "epoch": 1.2899398803710937e-05,
      "step": 13526,
      "training_step_time": 17.426132917404175
    },
    {
      "epoch": 1.2900352478027343e-05,
      "model_forward_time": 5.597684144973755,
      "step": 13527
    },
    {
      "epoch": 1.2900352478027343e-05,
      "step": 13527,
      "training_step_time": 17.039977312088013
    },
    {
      "epoch": 1.290130615234375e-05,
      "model_forward_time": 5.598275899887085,
      "step": 13528
    },
    {
      "epoch": 1.290130615234375e-05,
      "step": 13528,
      "training_step_time": 17.040752410888672
    },
    {
      "epoch": 1.2902259826660156e-05,
      "model_forward_time": 5.602932691574097,
      "step": 13529
    },
    {
      "epoch": 1.2902259826660156e-05,
      "step": 13529,
      "training_step_time": 17.074517726898193
    },
    {
      "epoch": 1.2903213500976562e-05,
      "grad_norm": 0.9364714026451111,
      "learning_rate": 1e-05,
      "loss": 0.068,
      "step": 13530
    },
    {
      "epoch": 1.2903213500976562e-05,
      "model_forward_time": 5.598657846450806,
      "step": 13530
    },
    {
      "dynamics_loss_avg": 0.05784764029085636,
      "epoch": 1.2903213500976562e-05,
      "step": 13530
    },
    {
      "action_loss_avg": 0.007624449394643307,
      "epoch": 1.2903213500976562e-05,
      "step": 13530
    },
    {
      "epoch": 1.2903213500976562e-05,
      "step": 13530,
      "training_step_time": 17.069663524627686
    },
    {
      "epoch": 1.290416717529297e-05,
      "model_forward_time": 5.600870132446289,
      "step": 13531
    },
    {
      "epoch": 1.290416717529297e-05,
      "step": 13531,
      "training_step_time": 17.44960331916809
    },
    {
      "epoch": 1.2905120849609375e-05,
      "model_forward_time": 5.592255353927612,
      "step": 13532
    },
    {
      "epoch": 1.2905120849609375e-05,
      "step": 13532,
      "training_step_time": 17.028372526168823
    },
    {
      "epoch": 1.290607452392578e-05,
      "model_forward_time": 5.608268737792969,
      "step": 13533
    },
    {
      "epoch": 1.290607452392578e-05,
      "step": 13533,
      "training_step_time": 17.088117599487305
    },
    {
      "epoch": 1.2907028198242188e-05,
      "model_forward_time": 5.603683948516846,
      "step": 13534
    },
    {
      "epoch": 1.2907028198242188e-05,
      "step": 13534,
      "training_step_time": 17.092775583267212
    },
    {
      "epoch": 1.2907981872558594e-05,
      "model_forward_time": 5.616260051727295,
      "step": 13535
    },
    {
      "epoch": 1.2907981872558594e-05,
      "step": 13535,
      "training_step_time": 17.106456756591797
    },
    {
      "epoch": 1.2908935546875e-05,
      "model_forward_time": 5.613883972167969,
      "step": 13536
    },
    {
      "epoch": 1.2908935546875e-05,
      "step": 13536,
      "training_step_time": 17.114003658294678
    },
    {
      "epoch": 1.2909889221191407e-05,
      "model_forward_time": 5.984462261199951,
      "step": 13537
    },
    {
      "epoch": 1.2909889221191407e-05,
      "step": 13537,
      "training_step_time": 17.467792510986328
    },
    {
      "epoch": 1.2910842895507813e-05,
      "model_forward_time": 5.607078552246094,
      "step": 13538
    },
    {
      "epoch": 1.2910842895507813e-05,
      "step": 13538,
      "training_step_time": 17.08712935447693
    },
    {
      "epoch": 1.2911796569824218e-05,
      "model_forward_time": 5.593383312225342,
      "step": 13539
    },
    {
      "epoch": 1.2911796569824218e-05,
      "step": 13539,
      "training_step_time": 17.031410932540894
    },
    {
      "epoch": 1.2912750244140624e-05,
      "grad_norm": 1.0053613185882568,
      "learning_rate": 1e-05,
      "loss": 0.0594,
      "step": 13540
    },
    {
      "epoch": 1.2912750244140624e-05,
      "model_forward_time": 5.591810703277588,
      "step": 13540
    },
    {
      "dynamics_loss_avg": 0.052215521782636644,
      "epoch": 1.2912750244140624e-05,
      "step": 13540
    },
    {
      "action_loss_avg": 0.008465504716150462,
      "epoch": 1.2912750244140624e-05,
      "step": 13540
    },
    {
      "epoch": 1.2912750244140624e-05,
      "step": 13540,
      "training_step_time": 17.032913208007812
    },
    {
      "epoch": 1.2913703918457032e-05,
      "model_forward_time": 5.597715854644775,
      "step": 13541
    },
    {
      "epoch": 1.2913703918457032e-05,
      "step": 13541,
      "training_step_time": 17.038925886154175
    },
    {
      "epoch": 1.2914657592773437e-05,
      "model_forward_time": 5.958404541015625,
      "step": 13542
    },
    {
      "epoch": 1.2914657592773437e-05,
      "step": 13542,
      "training_step_time": 17.43535590171814
    },
    {
      "epoch": 1.2915611267089843e-05,
      "model_forward_time": 5.611119031906128,
      "step": 13543
    },
    {
      "epoch": 1.2915611267089843e-05,
      "step": 13543,
      "training_step_time": 17.09415030479431
    },
    {
      "epoch": 1.291656494140625e-05,
      "model_forward_time": 5.616862773895264,
      "step": 13544
    },
    {
      "epoch": 1.291656494140625e-05,
      "step": 13544,
      "training_step_time": 17.1153621673584
    },
    {
      "epoch": 1.2917518615722656e-05,
      "model_forward_time": 5.60956597328186,
      "step": 13545
    },
    {
      "epoch": 1.2917518615722656e-05,
      "step": 13545,
      "training_step_time": 17.10584807395935
    },
    {
      "epoch": 1.2918472290039062e-05,
      "model_forward_time": 5.612455368041992,
      "step": 13546
    },
    {
      "epoch": 1.2918472290039062e-05,
      "step": 13546,
      "training_step_time": 17.087930917739868
    },
    {
      "epoch": 1.291942596435547e-05,
      "model_forward_time": 5.5982818603515625,
      "step": 13547
    },
    {
      "epoch": 1.291942596435547e-05,
      "step": 13547,
      "training_step_time": 17.035373210906982
    },
    {
      "epoch": 1.2920379638671875e-05,
      "model_forward_time": 5.954437971115112,
      "step": 13548
    },
    {
      "epoch": 1.2920379638671875e-05,
      "step": 13548,
      "training_step_time": 17.397809743881226
    },
    {
      "epoch": 1.292133331298828e-05,
      "model_forward_time": 5.592231512069702,
      "step": 13549
    },
    {
      "epoch": 1.292133331298828e-05,
      "step": 13549,
      "training_step_time": 17.049824237823486
    },
    {
      "epoch": 1.2922286987304688e-05,
      "grad_norm": 1.0005121231079102,
      "learning_rate": 1e-05,
      "loss": 0.0627,
      "step": 13550
    },
    {
      "epoch": 1.2922286987304688e-05,
      "model_forward_time": 5.643351793289185,
      "step": 13550
    },
    {
      "dynamics_loss_avg": 0.05413620099425316,
      "epoch": 1.2922286987304688e-05,
      "step": 13550
    },
    {
      "action_loss_avg": 0.007771775382570923,
      "epoch": 1.2922286987304688e-05,
      "step": 13550
    },
    {
      "epoch": 1.2922286987304688e-05,
      "step": 13550,
      "training_step_time": 17.1720712184906
    },
    {
      "epoch": 1.2923240661621094e-05,
      "model_forward_time": 5.623483180999756,
      "step": 13551
    },
    {
      "epoch": 1.2923240661621094e-05,
      "step": 13551,
      "training_step_time": 17.101834774017334
    },
    {
      "epoch": 1.29241943359375e-05,
      "model_forward_time": 5.6045026779174805,
      "step": 13552
    },
    {
      "epoch": 1.29241943359375e-05,
      "step": 13552,
      "training_step_time": 17.076380491256714
    },
    {
      "epoch": 1.2925148010253907e-05,
      "model_forward_time": 5.5985753536224365,
      "step": 13553
    },
    {
      "epoch": 1.2925148010253907e-05,
      "step": 13553,
      "training_step_time": 17.4120614528656
    },
    {
      "epoch": 1.2926101684570313e-05,
      "model_forward_time": 5.592658519744873,
      "step": 13554
    },
    {
      "epoch": 1.2926101684570313e-05,
      "step": 13554,
      "training_step_time": 17.033507108688354
    },
    {
      "epoch": 1.2927055358886718e-05,
      "model_forward_time": 5.598058462142944,
      "step": 13555
    },
    {
      "epoch": 1.2927055358886718e-05,
      "step": 13555,
      "training_step_time": 17.053058862686157
    },
    {
      "epoch": 1.2928009033203124e-05,
      "model_forward_time": 5.610691547393799,
      "step": 13556
    },
    {
      "epoch": 1.2928009033203124e-05,
      "step": 13556,
      "training_step_time": 17.07739543914795
    },
    {
      "epoch": 1.2928962707519532e-05,
      "model_forward_time": 5.605384349822998,
      "step": 13557
    },
    {
      "epoch": 1.2928962707519532e-05,
      "step": 13557,
      "training_step_time": 17.109367609024048
    },
    {
      "epoch": 1.2929916381835937e-05,
      "model_forward_time": 5.6187520027160645,
      "step": 13558
    },
    {
      "epoch": 1.2929916381835937e-05,
      "step": 13558,
      "training_step_time": 17.117058277130127
    },
    {
      "epoch": 1.2930870056152343e-05,
      "model_forward_time": 6.020008563995361,
      "step": 13559
    },
    {
      "epoch": 1.2930870056152343e-05,
      "step": 13559,
      "training_step_time": 17.4961576461792
    },
    {
      "epoch": 1.293182373046875e-05,
      "grad_norm": 1.1564325094223022,
      "learning_rate": 1e-05,
      "loss": 0.0592,
      "step": 13560
    },
    {
      "epoch": 1.293182373046875e-05,
      "model_forward_time": 5.5992279052734375,
      "step": 13560
    },
    {
      "dynamics_loss_avg": 0.05260828342288733,
      "epoch": 1.293182373046875e-05,
      "step": 13560
    },
    {
      "action_loss_avg": 0.006406333576887846,
      "epoch": 1.293182373046875e-05,
      "step": 13560
    },
    {
      "epoch": 1.293182373046875e-05,
      "step": 13560,
      "training_step_time": 17.06469488143921
    },
    {
      "epoch": 1.2932777404785156e-05,
      "model_forward_time": 5.601854562759399,
      "step": 13561
    },
    {
      "epoch": 1.2932777404785156e-05,
      "step": 13561,
      "training_step_time": 17.068174123764038
    },
    {
      "epoch": 1.2933731079101562e-05,
      "model_forward_time": 5.605762004852295,
      "step": 13562
    },
    {
      "epoch": 1.2933731079101562e-05,
      "step": 13562,
      "training_step_time": 17.03652310371399
    },
    {
      "epoch": 1.293468475341797e-05,
      "model_forward_time": 5.59689736366272,
      "step": 13563
    },
    {
      "epoch": 1.293468475341797e-05,
      "step": 13563,
      "training_step_time": 17.080541133880615
    },
    {
      "epoch": 1.2935638427734375e-05,
      "model_forward_time": 5.614467620849609,
      "step": 13564
    },
    {
      "epoch": 1.2935638427734375e-05,
      "step": 13564,
      "training_step_time": 17.46671152114868
    },
    {
      "epoch": 1.293659210205078e-05,
      "model_forward_time": 5.60837459564209,
      "step": 13565
    },
    {
      "epoch": 1.293659210205078e-05,
      "step": 13565,
      "training_step_time": 17.081075191497803
    },
    {
      "epoch": 1.2937545776367188e-05,
      "model_forward_time": 5.599433183670044,
      "step": 13566
    },
    {
      "epoch": 1.2937545776367188e-05,
      "step": 13566,
      "training_step_time": 17.052244663238525
    },
    {
      "epoch": 1.2938499450683594e-05,
      "model_forward_time": 5.58925461769104,
      "step": 13567
    },
    {
      "epoch": 1.2938499450683594e-05,
      "step": 13567,
      "training_step_time": 17.023133277893066
    },
    {
      "epoch": 1.2939453125e-05,
      "model_forward_time": 5.584903240203857,
      "step": 13568
    },
    {
      "epoch": 1.2939453125e-05,
      "step": 13568,
      "training_step_time": 17.01058840751648
    },
    {
      "epoch": 1.2940406799316407e-05,
      "model_forward_time": 5.591851234436035,
      "step": 13569
    },
    {
      "epoch": 1.2940406799316407e-05,
      "step": 13569,
      "training_step_time": 17.03080153465271
    },
    {
      "epoch": 1.2941360473632813e-05,
      "grad_norm": 1.135870337486267,
      "learning_rate": 1e-05,
      "loss": 0.0623,
      "step": 13570
    },
    {
      "epoch": 1.2941360473632813e-05,
      "model_forward_time": 5.976649284362793,
      "step": 13570
    },
    {
      "dynamics_loss_avg": 0.05343005135655403,
      "epoch": 1.2941360473632813e-05,
      "step": 13570
    },
    {
      "action_loss_avg": 0.0057705304818227885,
      "epoch": 1.2941360473632813e-05,
      "step": 13570
    },
    {
      "epoch": 1.2941360473632813e-05,
      "step": 13570,
      "training_step_time": 17.43530797958374
    },
    {
      "epoch": 1.2942314147949218e-05,
      "model_forward_time": 5.614617824554443,
      "step": 13571
    },
    {
      "epoch": 1.2942314147949218e-05,
      "step": 13571,
      "training_step_time": 17.13182258605957
    },
    {
      "epoch": 1.2943267822265624e-05,
      "model_forward_time": 5.621610879898071,
      "step": 13572
    },
    {
      "epoch": 1.2943267822265624e-05,
      "step": 13572,
      "training_step_time": 17.14639687538147
    },
    {
      "epoch": 1.2944221496582032e-05,
      "model_forward_time": 5.612559080123901,
      "step": 13573
    },
    {
      "epoch": 1.2944221496582032e-05,
      "step": 13573,
      "training_step_time": 17.110977172851562
    },
    {
      "epoch": 1.2945175170898437e-05,
      "model_forward_time": 5.6105992794036865,
      "step": 13574
    },
    {
      "epoch": 1.2945175170898437e-05,
      "step": 13574,
      "training_step_time": 17.098759651184082
    },
    {
      "epoch": 1.2946128845214843e-05,
      "model_forward_time": 5.965709447860718,
      "step": 13575
    },
    {
      "epoch": 1.2946128845214843e-05,
      "step": 13575,
      "training_step_time": 17.43986940383911
    },
    {
      "epoch": 1.294708251953125e-05,
      "model_forward_time": 5.592509984970093,
      "step": 13576
    },
    {
      "epoch": 1.294708251953125e-05,
      "step": 13576,
      "training_step_time": 17.043914794921875
    },
    {
      "epoch": 1.2948036193847656e-05,
      "model_forward_time": 5.591697931289673,
      "step": 13577
    },
    {
      "epoch": 1.2948036193847656e-05,
      "step": 13577,
      "training_step_time": 17.04808735847473
    },
    {
      "epoch": 1.2948989868164062e-05,
      "model_forward_time": 5.599080562591553,
      "step": 13578
    },
    {
      "epoch": 1.2948989868164062e-05,
      "step": 13578,
      "training_step_time": 17.06363868713379
    },
    {
      "epoch": 1.294994354248047e-05,
      "model_forward_time": 5.602275133132935,
      "step": 13579
    },
    {
      "epoch": 1.294994354248047e-05,
      "step": 13579,
      "training_step_time": 17.08558964729309
    },
    {
      "epoch": 1.2950897216796875e-05,
      "grad_norm": 1.0392986536026,
      "learning_rate": 1e-05,
      "loss": 0.0599,
      "step": 13580
    },
    {
      "epoch": 1.2950897216796875e-05,
      "model_forward_time": 5.614173889160156,
      "step": 13580
    },
    {
      "dynamics_loss_avg": 0.05190397799015045,
      "epoch": 1.2950897216796875e-05,
      "step": 13580
    },
    {
      "action_loss_avg": 0.007662325247656554,
      "epoch": 1.2950897216796875e-05,
      "step": 13580
    },
    {
      "epoch": 1.2950897216796875e-05,
      "step": 13580,
      "training_step_time": 17.129433155059814
    },
    {
      "epoch": 1.295185089111328e-05,
      "model_forward_time": 6.012141466140747,
      "step": 13581
    },
    {
      "epoch": 1.295185089111328e-05,
      "step": 13581,
      "training_step_time": 17.49966597557068
    },
    {
      "epoch": 1.2952804565429688e-05,
      "model_forward_time": 5.59912896156311,
      "step": 13582
    },
    {
      "epoch": 1.2952804565429688e-05,
      "step": 13582,
      "training_step_time": 17.063504934310913
    },
    {
      "epoch": 1.2953758239746094e-05,
      "model_forward_time": 5.582870244979858,
      "step": 13583
    },
    {
      "epoch": 1.2953758239746094e-05,
      "step": 13583,
      "training_step_time": 17.001853466033936
    },
    {
      "epoch": 1.29547119140625e-05,
      "model_forward_time": 5.587862014770508,
      "step": 13584
    },
    {
      "epoch": 1.29547119140625e-05,
      "step": 13584,
      "training_step_time": 17.023727416992188
    },
    {
      "epoch": 1.2955665588378907e-05,
      "model_forward_time": 5.605486869812012,
      "step": 13585
    },
    {
      "epoch": 1.2955665588378907e-05,
      "step": 13585,
      "training_step_time": 17.05529236793518
    },
    {
      "epoch": 1.2956619262695313e-05,
      "model_forward_time": 5.594450235366821,
      "step": 13586
    },
    {
      "epoch": 1.2956619262695313e-05,
      "step": 13586,
      "training_step_time": 17.430694818496704
    },
    {
      "epoch": 1.2957572937011719e-05,
      "model_forward_time": 5.6032493114471436,
      "step": 13587
    },
    {
      "epoch": 1.2957572937011719e-05,
      "step": 13587,
      "training_step_time": 17.094546794891357
    },
    {
      "epoch": 1.2958526611328124e-05,
      "model_forward_time": 5.604930877685547,
      "step": 13588
    },
    {
      "epoch": 1.2958526611328124e-05,
      "step": 13588,
      "training_step_time": 17.079429388046265
    },
    {
      "epoch": 1.2959480285644532e-05,
      "model_forward_time": 5.6071202754974365,
      "step": 13589
    },
    {
      "epoch": 1.2959480285644532e-05,
      "step": 13589,
      "training_step_time": 17.062448263168335
    },
    {
      "epoch": 1.2960433959960937e-05,
      "grad_norm": 0.5588803887367249,
      "learning_rate": 1e-05,
      "loss": 0.0569,
      "step": 13590
    },
    {
      "epoch": 1.2960433959960937e-05,
      "model_forward_time": 5.600364923477173,
      "step": 13590
    },
    {
      "dynamics_loss_avg": 0.054492123425006866,
      "epoch": 1.2960433959960937e-05,
      "step": 13590
    },
    {
      "action_loss_avg": 0.005849951994605362,
      "epoch": 1.2960433959960937e-05,
      "step": 13590
    },
    {
      "epoch": 1.2960433959960937e-05,
      "step": 13590,
      "training_step_time": 17.044352054595947
    },
    {
      "epoch": 1.2961387634277343e-05,
      "model_forward_time": 5.592819929122925,
      "step": 13591
    },
    {
      "epoch": 1.2961387634277343e-05,
      "step": 13591,
      "training_step_time": 17.027486085891724
    },
    {
      "epoch": 1.296234130859375e-05,
      "model_forward_time": 5.964141368865967,
      "step": 13592
    },
    {
      "epoch": 1.296234130859375e-05,
      "step": 13592,
      "training_step_time": 17.39140558242798
    },
    {
      "epoch": 1.2963294982910156e-05,
      "model_forward_time": 5.601040840148926,
      "step": 13593
    },
    {
      "epoch": 1.2963294982910156e-05,
      "step": 13593,
      "training_step_time": 17.077006101608276
    },
    {
      "epoch": 1.2964248657226562e-05,
      "model_forward_time": 5.618431568145752,
      "step": 13594
    },
    {
      "epoch": 1.2964248657226562e-05,
      "step": 13594,
      "training_step_time": 17.142472505569458
    },
    {
      "epoch": 1.296520233154297e-05,
      "model_forward_time": 5.621584415435791,
      "step": 13595
    },
    {
      "epoch": 1.296520233154297e-05,
      "step": 13595,
      "training_step_time": 17.132912397384644
    },
    {
      "epoch": 1.2966156005859375e-05,
      "model_forward_time": 5.620564222335815,
      "step": 13596
    },
    {
      "epoch": 1.2966156005859375e-05,
      "step": 13596,
      "training_step_time": 17.111241102218628
    },
    {
      "epoch": 1.296710968017578e-05,
      "model_forward_time": 5.604192018508911,
      "step": 13597
    },
    {
      "epoch": 1.296710968017578e-05,
      "step": 13597,
      "training_step_time": 17.4118070602417
    },
    {
      "epoch": 1.2968063354492188e-05,
      "model_forward_time": 5.591858148574829,
      "step": 13598
    },
    {
      "epoch": 1.2968063354492188e-05,
      "step": 13598,
      "training_step_time": 17.0115864276886
    },
    {
      "epoch": 1.2969017028808594e-05,
      "model_forward_time": 5.594601631164551,
      "step": 13599
    },
    {
      "epoch": 1.2969017028808594e-05,
      "step": 13599,
      "training_step_time": 17.040781497955322
    },
    {
      "epoch": 1.2969970703125e-05,
      "grad_norm": 1.6526967287063599,
      "learning_rate": 1e-05,
      "loss": 0.0853,
      "step": 13600
    },
    {
      "epoch": 1.2969970703125e-05,
      "model_forward_time": 5.606766700744629,
      "step": 13600
    },
    {
      "dynamics_loss_avg": 0.07541160192340612,
      "epoch": 1.2969970703125e-05,
      "step": 13600
    },
    {
      "action_loss_avg": 0.010602960200048982,
      "epoch": 1.2969970703125e-05,
      "step": 13600
    },
    {
      "epoch": 1.2969970703125e-05,
      "step": 13600,
      "training_step_time": 17.05894422531128
    },
    {
      "epoch": 1.2970924377441407e-05,
      "model_forward_time": 5.606067895889282,
      "step": 13601
    },
    {
      "epoch": 1.2970924377441407e-05,
      "step": 13601,
      "training_step_time": 17.063368558883667
    },
    {
      "epoch": 1.2971878051757813e-05,
      "model_forward_time": 5.596128940582275,
      "step": 13602
    },
    {
      "epoch": 1.2971878051757813e-05,
      "step": 13602,
      "training_step_time": 17.047349452972412
    },
    {
      "epoch": 1.2972831726074219e-05,
      "model_forward_time": 5.963423252105713,
      "step": 13603
    },
    {
      "epoch": 1.2972831726074219e-05,
      "step": 13603,
      "training_step_time": 17.41202688217163
    },
    {
      "epoch": 1.2973785400390624e-05,
      "model_forward_time": 5.6111180782318115,
      "step": 13604
    },
    {
      "epoch": 1.2973785400390624e-05,
      "step": 13604,
      "training_step_time": 17.095999240875244
    },
    {
      "epoch": 1.2974739074707032e-05,
      "model_forward_time": 5.6171793937683105,
      "step": 13605
    },
    {
      "epoch": 1.2974739074707032e-05,
      "step": 13605,
      "training_step_time": 17.101633310317993
    },
    {
      "epoch": 1.2975692749023437e-05,
      "model_forward_time": 5.597901105880737,
      "step": 13606
    },
    {
      "epoch": 1.2975692749023437e-05,
      "step": 13606,
      "training_step_time": 17.056994676589966
    },
    {
      "epoch": 1.2976646423339843e-05,
      "model_forward_time": 5.594707250595093,
      "step": 13607
    },
    {
      "epoch": 1.2976646423339843e-05,
      "step": 13607,
      "training_step_time": 17.014854192733765
    },
    {
      "epoch": 1.297760009765625e-05,
      "model_forward_time": 5.938031435012817,
      "step": 13608
    },
    {
      "epoch": 1.297760009765625e-05,
      "step": 13608,
      "training_step_time": 17.379936456680298
    },
    {
      "epoch": 1.2978553771972656e-05,
      "model_forward_time": 5.57676887512207,
      "step": 13609
    },
    {
      "epoch": 1.2978553771972656e-05,
      "step": 13609,
      "training_step_time": 17.011357307434082
    },
    {
      "epoch": 1.2979507446289062e-05,
      "grad_norm": 0.8370424509048462,
      "learning_rate": 1e-05,
      "loss": 0.0645,
      "step": 13610
    },
    {
      "epoch": 1.2979507446289062e-05,
      "model_forward_time": 5.591360330581665,
      "step": 13610
    },
    {
      "dynamics_loss_avg": 0.05224221255630255,
      "epoch": 1.2979507446289062e-05,
      "step": 13610
    },
    {
      "action_loss_avg": 0.0064050598535686735,
      "epoch": 1.2979507446289062e-05,
      "step": 13610
    },
    {
      "epoch": 1.2979507446289062e-05,
      "step": 13610,
      "training_step_time": 17.069720029830933
    },
    {
      "epoch": 1.298046112060547e-05,
      "model_forward_time": 5.617369174957275,
      "step": 13611
    },
    {
      "epoch": 1.298046112060547e-05,
      "step": 13611,
      "training_step_time": 17.1244056224823
    },
    {
      "epoch": 1.2981414794921875e-05,
      "model_forward_time": 5.628809928894043,
      "step": 13612
    },
    {
      "epoch": 1.2981414794921875e-05,
      "step": 13612,
      "training_step_time": 17.105514764785767
    },
    {
      "epoch": 1.298236846923828e-05,
      "model_forward_time": 5.618074893951416,
      "step": 13613
    },
    {
      "epoch": 1.298236846923828e-05,
      "step": 13613,
      "training_step_time": 17.10653066635132
    },
    {
      "epoch": 1.2983322143554688e-05,
      "model_forward_time": 5.982579469680786,
      "step": 13614
    },
    {
      "epoch": 1.2983322143554688e-05,
      "step": 13614,
      "training_step_time": 17.454604864120483
    },
    {
      "epoch": 1.2984275817871094e-05,
      "model_forward_time": 5.611806869506836,
      "step": 13615
    },
    {
      "epoch": 1.2984275817871094e-05,
      "step": 13615,
      "training_step_time": 17.080692052841187
    },
    {
      "epoch": 1.29852294921875e-05,
      "model_forward_time": 5.5944273471832275,
      "step": 13616
    },
    {
      "epoch": 1.29852294921875e-05,
      "step": 13616,
      "training_step_time": 17.045716762542725
    },
    {
      "epoch": 1.2986183166503907e-05,
      "model_forward_time": 5.596256256103516,
      "step": 13617
    },
    {
      "epoch": 1.2986183166503907e-05,
      "step": 13617,
      "training_step_time": 17.051496744155884
    },
    {
      "epoch": 1.2987136840820313e-05,
      "model_forward_time": 5.616374254226685,
      "step": 13618
    },
    {
      "epoch": 1.2987136840820313e-05,
      "step": 13618,
      "training_step_time": 17.079344272613525
    },
    {
      "epoch": 1.2988090515136719e-05,
      "model_forward_time": 5.616229295730591,
      "step": 13619
    },
    {
      "epoch": 1.2988090515136719e-05,
      "step": 13619,
      "training_step_time": 17.476343631744385
    },
    {
      "epoch": 1.2989044189453124e-05,
      "grad_norm": 0.7457884550094604,
      "learning_rate": 1e-05,
      "loss": 0.0631,
      "step": 13620
    },
    {
      "epoch": 1.2989044189453124e-05,
      "model_forward_time": 5.631844997406006,
      "step": 13620
    },
    {
      "dynamics_loss_avg": 0.05880378466099501,
      "epoch": 1.2989044189453124e-05,
      "step": 13620
    },
    {
      "action_loss_avg": 0.0077086323872208595,
      "epoch": 1.2989044189453124e-05,
      "step": 13620
    },
    {
      "epoch": 1.2989044189453124e-05,
      "step": 13620,
      "training_step_time": 17.130048274993896
    },
    {
      "epoch": 1.2989997863769532e-05,
      "model_forward_time": 5.611025094985962,
      "step": 13621
    },
    {
      "epoch": 1.2989997863769532e-05,
      "step": 13621,
      "training_step_time": 17.097576141357422
    },
    {
      "epoch": 1.2990951538085937e-05,
      "model_forward_time": 5.6156487464904785,
      "step": 13622
    },
    {
      "epoch": 1.2990951538085937e-05,
      "step": 13622,
      "training_step_time": 17.106032133102417
    },
    {
      "epoch": 1.2991905212402343e-05,
      "model_forward_time": 5.609738111495972,
      "step": 13623
    },
    {
      "epoch": 1.2991905212402343e-05,
      "step": 13623,
      "training_step_time": 17.101583242416382
    },
    {
      "epoch": 1.299285888671875e-05,
      "model_forward_time": 5.6072752475738525,
      "step": 13624
    },
    {
      "epoch": 1.299285888671875e-05,
      "step": 13624,
      "training_step_time": 17.076191186904907
    },
    {
      "epoch": 1.2993812561035156e-05,
      "model_forward_time": 5.9698710441589355,
      "step": 13625
    },
    {
      "epoch": 1.2993812561035156e-05,
      "step": 13625,
      "training_step_time": 17.3962504863739
    },
    {
      "epoch": 1.2994766235351562e-05,
      "model_forward_time": 5.579345941543579,
      "step": 13626
    },
    {
      "epoch": 1.2994766235351562e-05,
      "step": 13626,
      "training_step_time": 17.003320693969727
    },
    {
      "epoch": 1.299571990966797e-05,
      "model_forward_time": 5.577496767044067,
      "step": 13627
    },
    {
      "epoch": 1.299571990966797e-05,
      "step": 13627,
      "training_step_time": 17.012877464294434
    },
    {
      "epoch": 1.2996673583984375e-05,
      "model_forward_time": 5.596847057342529,
      "step": 13628
    },
    {
      "epoch": 1.2996673583984375e-05,
      "step": 13628,
      "training_step_time": 17.03692126274109
    },
    {
      "epoch": 1.299762725830078e-05,
      "model_forward_time": 5.595689535140991,
      "step": 13629
    },
    {
      "epoch": 1.299762725830078e-05,
      "step": 13629,
      "training_step_time": 17.028510332107544
    },
    {
      "epoch": 1.2998580932617188e-05,
      "grad_norm": 1.2085238695144653,
      "learning_rate": 1e-05,
      "loss": 0.0528,
      "step": 13630
    },
    {
      "epoch": 1.2998580932617188e-05,
      "model_forward_time": 5.59244966506958,
      "step": 13630
    },
    {
      "dynamics_loss_avg": 0.043901503831148145,
      "epoch": 1.2998580932617188e-05,
      "step": 13630
    },
    {
      "action_loss_avg": 0.005475264694541693,
      "epoch": 1.2998580932617188e-05,
      "step": 13630
    },
    {
      "epoch": 1.2998580932617188e-05,
      "step": 13630,
      "training_step_time": 17.42102813720703
    },
    {
      "epoch": 1.2999534606933594e-05,
      "model_forward_time": 5.601376533508301,
      "step": 13631
    },
    {
      "epoch": 1.2999534606933594e-05,
      "step": 13631,
      "training_step_time": 17.101749658584595
    },
    {
      "epoch": 1.300048828125e-05,
      "model_forward_time": 5.628859519958496,
      "step": 13632
    },
    {
      "epoch": 1.300048828125e-05,
      "step": 13632,
      "training_step_time": 17.16696286201477
    },
    {
      "epoch": 1.3001441955566405e-05,
      "model_forward_time": 5.6273193359375,
      "step": 13633
    },
    {
      "epoch": 1.3001441955566405e-05,
      "step": 13633,
      "training_step_time": 17.10722517967224
    },
    {
      "epoch": 1.3002395629882813e-05,
      "model_forward_time": 5.58527398109436,
      "step": 13634
    },
    {
      "epoch": 1.3002395629882813e-05,
      "step": 13634,
      "training_step_time": 17.013150215148926
    },
    {
      "epoch": 1.3003349304199219e-05,
      "model_forward_time": 5.570058822631836,
      "step": 13635
    },
    {
      "epoch": 1.3003349304199219e-05,
      "step": 13635,
      "training_step_time": 16.989556312561035
    },
    {
      "epoch": 1.3004302978515624e-05,
      "model_forward_time": 5.929867506027222,
      "step": 13636
    },
    {
      "epoch": 1.3004302978515624e-05,
      "step": 13636,
      "training_step_time": 17.352664470672607
    },
    {
      "epoch": 1.3005256652832032e-05,
      "model_forward_time": 5.562251806259155,
      "step": 13637
    },
    {
      "epoch": 1.3005256652832032e-05,
      "step": 13637,
      "training_step_time": 16.96469283103943
    },
    {
      "epoch": 1.3006210327148437e-05,
      "model_forward_time": 5.586026668548584,
      "step": 13638
    },
    {
      "epoch": 1.3006210327148437e-05,
      "step": 13638,
      "training_step_time": 17.0203959941864
    },
    {
      "epoch": 1.3007164001464843e-05,
      "model_forward_time": 5.574795722961426,
      "step": 13639
    },
    {
      "epoch": 1.3007164001464843e-05,
      "step": 13639,
      "training_step_time": 16.99936604499817
    },
    {
      "epoch": 1.300811767578125e-05,
      "grad_norm": 0.42583921551704407,
      "learning_rate": 1e-05,
      "loss": 0.069,
      "step": 13640
    },
    {
      "epoch": 1.300811767578125e-05,
      "model_forward_time": 5.603433847427368,
      "step": 13640
    },
    {
      "dynamics_loss_avg": 0.06568687148392201,
      "epoch": 1.300811767578125e-05,
      "step": 13640
    },
    {
      "action_loss_avg": 0.007877622335217892,
      "epoch": 1.300811767578125e-05,
      "step": 13640
    },
    {
      "epoch": 1.300811767578125e-05,
      "step": 13640,
      "training_step_time": 17.013944149017334
    },
    {
      "epoch": 1.3009071350097656e-05,
      "model_forward_time": 5.931116342544556,
      "step": 13641
    },
    {
      "epoch": 1.3009071350097656e-05,
      "step": 13641,
      "training_step_time": 17.394291400909424
    },
    {
      "epoch": 1.3010025024414062e-05,
      "model_forward_time": 5.627689361572266,
      "step": 13642
    },
    {
      "epoch": 1.3010025024414062e-05,
      "step": 13642,
      "training_step_time": 17.142690658569336
    },
    {
      "epoch": 1.301097869873047e-05,
      "model_forward_time": 5.614835739135742,
      "step": 13643
    },
    {
      "epoch": 1.301097869873047e-05,
      "step": 13643,
      "training_step_time": 17.113656044006348
    },
    {
      "epoch": 1.3011932373046875e-05,
      "model_forward_time": 5.613756418228149,
      "step": 13644
    },
    {
      "epoch": 1.3011932373046875e-05,
      "step": 13644,
      "training_step_time": 17.106563568115234
    },
    {
      "epoch": 1.301288604736328e-05,
      "model_forward_time": 5.602576732635498,
      "step": 13645
    },
    {
      "epoch": 1.301288604736328e-05,
      "step": 13645,
      "training_step_time": 17.082075357437134
    },
    {
      "epoch": 1.3013839721679688e-05,
      "model_forward_time": 5.608092546463013,
      "step": 13646
    },
    {
      "epoch": 1.3013839721679688e-05,
      "step": 13646,
      "training_step_time": 17.08537244796753
    },
    {
      "epoch": 1.3014793395996094e-05,
      "model_forward_time": 5.979259014129639,
      "step": 13647
    },
    {
      "epoch": 1.3014793395996094e-05,
      "step": 13647,
      "training_step_time": 17.455007314682007
    },
    {
      "epoch": 1.30157470703125e-05,
      "model_forward_time": 5.613527059555054,
      "step": 13648
    },
    {
      "epoch": 1.30157470703125e-05,
      "step": 13648,
      "training_step_time": 17.095083951950073
    },
    {
      "epoch": 1.3016700744628905e-05,
      "model_forward_time": 5.600644588470459,
      "step": 13649
    },
    {
      "epoch": 1.3016700744628905e-05,
      "step": 13649,
      "training_step_time": 17.053473711013794
    },
    {
      "epoch": 1.3017654418945313e-05,
      "grad_norm": 0.6275531649589539,
      "learning_rate": 1e-05,
      "loss": 0.0661,
      "step": 13650
    },
    {
      "epoch": 1.3017654418945313e-05,
      "model_forward_time": 5.591054201126099,
      "step": 13650
    },
    {
      "dynamics_loss_avg": 0.05600669048726559,
      "epoch": 1.3017654418945313e-05,
      "step": 13650
    },
    {
      "action_loss_avg": 0.006859837821684778,
      "epoch": 1.3017654418945313e-05,
      "step": 13650
    },
    {
      "epoch": 1.3017654418945313e-05,
      "step": 13650,
      "training_step_time": 17.045552253723145
    },
    {
      "epoch": 1.3018608093261719e-05,
      "model_forward_time": 5.582124471664429,
      "step": 13651
    },
    {
      "epoch": 1.3018608093261719e-05,
      "step": 13651,
      "training_step_time": 16.997825145721436
    },
    {
      "epoch": 1.3019561767578124e-05,
      "model_forward_time": 5.579556941986084,
      "step": 13652
    },
    {
      "epoch": 1.3019561767578124e-05,
      "step": 13652,
      "training_step_time": 17.370052576065063
    },
    {
      "epoch": 1.3020515441894532e-05,
      "model_forward_time": 5.586784362792969,
      "step": 13653
    },
    {
      "epoch": 1.3020515441894532e-05,
      "step": 13653,
      "training_step_time": 17.032522678375244
    },
    {
      "epoch": 1.3021469116210937e-05,
      "model_forward_time": 5.596010446548462,
      "step": 13654
    },
    {
      "epoch": 1.3021469116210937e-05,
      "step": 13654,
      "training_step_time": 17.0782732963562
    },
    {
      "epoch": 1.3022422790527343e-05,
      "model_forward_time": 5.604018926620483,
      "step": 13655
    },
    {
      "epoch": 1.3022422790527343e-05,
      "step": 13655,
      "training_step_time": 17.065070867538452
    },
    {
      "epoch": 1.302337646484375e-05,
      "model_forward_time": 5.596099138259888,
      "step": 13656
    },
    {
      "epoch": 1.302337646484375e-05,
      "step": 13656,
      "training_step_time": 17.008941888809204
    },
    {
      "epoch": 1.3024330139160156e-05,
      "model_forward_time": 5.569722414016724,
      "step": 13657
    },
    {
      "epoch": 1.3024330139160156e-05,
      "step": 13657,
      "training_step_time": 16.99912714958191
    },
    {
      "epoch": 1.3025283813476562e-05,
      "model_forward_time": 5.9363462924957275,
      "step": 13658
    },
    {
      "epoch": 1.3025283813476562e-05,
      "step": 13658,
      "training_step_time": 17.339411973953247
    },
    {
      "epoch": 1.302623748779297e-05,
      "model_forward_time": 5.576000690460205,
      "step": 13659
    },
    {
      "epoch": 1.302623748779297e-05,
      "step": 13659,
      "training_step_time": 16.983452320098877
    },
    {
      "epoch": 1.3027191162109375e-05,
      "grad_norm": 0.9383096694946289,
      "learning_rate": 1e-05,
      "loss": 0.0477,
      "step": 13660
    },
    {
      "epoch": 1.3027191162109375e-05,
      "model_forward_time": 5.580543756484985,
      "step": 13660
    },
    {
      "dynamics_loss_avg": 0.045624487474560736,
      "epoch": 1.3027191162109375e-05,
      "step": 13660
    },
    {
      "action_loss_avg": 0.005976036516949534,
      "epoch": 1.3027191162109375e-05,
      "step": 13660
    },
    {
      "epoch": 1.3027191162109375e-05,
      "step": 13660,
      "training_step_time": 16.9896457195282
    },
    {
      "epoch": 1.3028144836425781e-05,
      "model_forward_time": 5.5809125900268555,
      "step": 13661
    },
    {
      "epoch": 1.3028144836425781e-05,
      "step": 13661,
      "training_step_time": 17.011173248291016
    },
    {
      "epoch": 1.3029098510742188e-05,
      "model_forward_time": 5.590052843093872,
      "step": 13662
    },
    {
      "epoch": 1.3029098510742188e-05,
      "step": 13662,
      "training_step_time": 17.041900873184204
    },
    {
      "epoch": 1.3030052185058594e-05,
      "model_forward_time": 5.598774433135986,
      "step": 13663
    },
    {
      "epoch": 1.3030052185058594e-05,
      "step": 13663,
      "training_step_time": 17.458632946014404
    },
    {
      "epoch": 1.3031005859375e-05,
      "model_forward_time": 5.625485897064209,
      "step": 13664
    },
    {
      "epoch": 1.3031005859375e-05,
      "step": 13664,
      "training_step_time": 17.122446298599243
    },
    {
      "epoch": 1.3031959533691405e-05,
      "model_forward_time": 5.617292881011963,
      "step": 13665
    },
    {
      "epoch": 1.3031959533691405e-05,
      "step": 13665,
      "training_step_time": 17.095951795578003
    },
    {
      "epoch": 1.3032913208007813e-05,
      "model_forward_time": 5.59646463394165,
      "step": 13666
    },
    {
      "epoch": 1.3032913208007813e-05,
      "step": 13666,
      "training_step_time": 17.012380123138428
    },
    {
      "epoch": 1.3033866882324219e-05,
      "model_forward_time": 5.569867134094238,
      "step": 13667
    },
    {
      "epoch": 1.3033866882324219e-05,
      "step": 13667,
      "training_step_time": 16.99041485786438
    },
    {
      "epoch": 1.3034820556640624e-05,
      "model_forward_time": 5.57671332359314,
      "step": 13668
    },
    {
      "epoch": 1.3034820556640624e-05,
      "step": 13668,
      "training_step_time": 17.005020141601562
    },
    {
      "epoch": 1.3035774230957032e-05,
      "model_forward_time": 5.947157144546509,
      "step": 13669
    },
    {
      "epoch": 1.3035774230957032e-05,
      "step": 13669,
      "training_step_time": 17.381887435913086
    },
    {
      "epoch": 1.3036727905273437e-05,
      "grad_norm": 0.8253230452537537,
      "learning_rate": 1e-05,
      "loss": 0.0665,
      "step": 13670
    },
    {
      "epoch": 1.3036727905273437e-05,
      "model_forward_time": 5.575526237487793,
      "step": 13670
    },
    {
      "dynamics_loss_avg": 0.05742759183049202,
      "epoch": 1.3036727905273437e-05,
      "step": 13670
    },
    {
      "action_loss_avg": 0.0069218732882291075,
      "epoch": 1.3036727905273437e-05,
      "step": 13670
    },
    {
      "epoch": 1.3036727905273437e-05,
      "step": 13670,
      "training_step_time": 17.00145721435547
    },
    {
      "epoch": 1.3037681579589843e-05,
      "model_forward_time": 5.580707550048828,
      "step": 13671
    },
    {
      "epoch": 1.3037681579589843e-05,
      "step": 13671,
      "training_step_time": 17.01471495628357
    },
    {
      "epoch": 1.303863525390625e-05,
      "model_forward_time": 5.582583665847778,
      "step": 13672
    },
    {
      "epoch": 1.303863525390625e-05,
      "step": 13672,
      "training_step_time": 17.01131272315979
    },
    {
      "epoch": 1.3039588928222656e-05,
      "model_forward_time": 5.587653398513794,
      "step": 13673
    },
    {
      "epoch": 1.3039588928222656e-05,
      "step": 13673,
      "training_step_time": 17.044771432876587
    },
    {
      "epoch": 1.3040542602539062e-05,
      "model_forward_time": 5.9554524421691895,
      "step": 13674
    },
    {
      "epoch": 1.3040542602539062e-05,
      "step": 13674,
      "training_step_time": 17.435568809509277
    },
    {
      "epoch": 1.304149627685547e-05,
      "model_forward_time": 5.6183366775512695,
      "step": 13675
    },
    {
      "epoch": 1.304149627685547e-05,
      "step": 13675,
      "training_step_time": 17.110857248306274
    },
    {
      "epoch": 1.3042449951171875e-05,
      "model_forward_time": 5.605545520782471,
      "step": 13676
    },
    {
      "epoch": 1.3042449951171875e-05,
      "step": 13676,
      "training_step_time": 17.05413317680359
    },
    {
      "epoch": 1.3043403625488281e-05,
      "model_forward_time": 5.579221725463867,
      "step": 13677
    },
    {
      "epoch": 1.3043403625488281e-05,
      "step": 13677,
      "training_step_time": 17.003546237945557
    },
    {
      "epoch": 1.3044357299804688e-05,
      "model_forward_time": 5.573450088500977,
      "step": 13678
    },
    {
      "epoch": 1.3044357299804688e-05,
      "step": 13678,
      "training_step_time": 16.97780132293701
    },
    {
      "epoch": 1.3045310974121094e-05,
      "model_forward_time": 5.588766574859619,
      "step": 13679
    },
    {
      "epoch": 1.3045310974121094e-05,
      "step": 13679,
      "training_step_time": 17.007270336151123
    },
    {
      "epoch": 1.30462646484375e-05,
      "grad_norm": 0.4495971202850342,
      "learning_rate": 1e-05,
      "loss": 0.0606,
      "step": 13680
    },
    {
      "epoch": 1.30462646484375e-05,
      "model_forward_time": 5.944380044937134,
      "step": 13680
    },
    {
      "dynamics_loss_avg": 0.05541248433291912,
      "epoch": 1.30462646484375e-05,
      "step": 13680
    },
    {
      "action_loss_avg": 0.008487764932215213,
      "epoch": 1.30462646484375e-05,
      "step": 13680
    },
    {
      "epoch": 1.30462646484375e-05,
      "step": 13680,
      "training_step_time": 17.39444375038147
    },
    {
      "epoch": 1.3047218322753905e-05,
      "model_forward_time": 5.5799641609191895,
      "step": 13681
    },
    {
      "epoch": 1.3047218322753905e-05,
      "step": 13681,
      "training_step_time": 17.006171464920044
    },
    {
      "epoch": 1.3048171997070313e-05,
      "model_forward_time": 5.59163498878479,
      "step": 13682
    },
    {
      "epoch": 1.3048171997070313e-05,
      "step": 13682,
      "training_step_time": 17.050678253173828
    },
    {
      "epoch": 1.3049125671386719e-05,
      "model_forward_time": 5.614113807678223,
      "step": 13683
    },
    {
      "epoch": 1.3049125671386719e-05,
      "step": 13683,
      "training_step_time": 17.12461018562317
    },
    {
      "epoch": 1.3050079345703124e-05,
      "model_forward_time": 5.629420280456543,
      "step": 13684
    },
    {
      "epoch": 1.3050079345703124e-05,
      "step": 13684,
      "training_step_time": 17.156174659729004
    },
    {
      "epoch": 1.3051033020019532e-05,
      "model_forward_time": 5.626443386077881,
      "step": 13685
    },
    {
      "epoch": 1.3051033020019532e-05,
      "step": 13685,
      "training_step_time": 17.525063037872314
    },
    {
      "epoch": 1.3051986694335937e-05,
      "model_forward_time": 5.592841625213623,
      "step": 13686
    },
    {
      "epoch": 1.3051986694335937e-05,
      "step": 13686,
      "training_step_time": 17.01375675201416
    },
    {
      "epoch": 1.3052940368652343e-05,
      "model_forward_time": 5.559131145477295,
      "step": 13687
    },
    {
      "epoch": 1.3052940368652343e-05,
      "step": 13687,
      "training_step_time": 16.972261667251587
    },
    {
      "epoch": 1.305389404296875e-05,
      "model_forward_time": 5.559956789016724,
      "step": 13688
    },
    {
      "epoch": 1.305389404296875e-05,
      "step": 13688,
      "training_step_time": 16.976686000823975
    },
    {
      "epoch": 1.3054847717285156e-05,
      "model_forward_time": 5.563058853149414,
      "step": 13689
    },
    {
      "epoch": 1.3054847717285156e-05,
      "step": 13689,
      "training_step_time": 16.984107494354248
    },
    {
      "epoch": 1.3055801391601562e-05,
      "grad_norm": 1.1298069953918457,
      "learning_rate": 1e-05,
      "loss": 0.0791,
      "step": 13690
    },
    {
      "epoch": 1.3055801391601562e-05,
      "model_forward_time": 5.57454776763916,
      "step": 13690
    },
    {
      "dynamics_loss_avg": 0.06976619604974985,
      "epoch": 1.3055801391601562e-05,
      "step": 13690
    },
    {
      "action_loss_avg": 0.008176473318599164,
      "epoch": 1.3055801391601562e-05,
      "step": 13690
    },
    {
      "epoch": 1.3055801391601562e-05,
      "step": 13690,
      "training_step_time": 16.996015787124634
    },
    {
      "epoch": 1.305675506591797e-05,
      "model_forward_time": 5.946352005004883,
      "step": 13691
    },
    {
      "epoch": 1.305675506591797e-05,
      "step": 13691,
      "training_step_time": 17.3820481300354
    },
    {
      "epoch": 1.3057708740234375e-05,
      "model_forward_time": 5.562857151031494,
      "step": 13692
    },
    {
      "epoch": 1.3057708740234375e-05,
      "step": 13692,
      "training_step_time": 16.996942281723022
    },
    {
      "epoch": 1.3058662414550781e-05,
      "model_forward_time": 5.568570613861084,
      "step": 13693
    },
    {
      "epoch": 1.3058662414550781e-05,
      "step": 13693,
      "training_step_time": 16.996524572372437
    },
    {
      "epoch": 1.3059616088867188e-05,
      "model_forward_time": 5.586118221282959,
      "step": 13694
    },
    {
      "epoch": 1.3059616088867188e-05,
      "step": 13694,
      "training_step_time": 17.00976252555847
    },
    {
      "epoch": 1.3060569763183594e-05,
      "model_forward_time": 5.5928380489349365,
      "step": 13695
    },
    {
      "epoch": 1.3060569763183594e-05,
      "step": 13695,
      "training_step_time": 17.072817087173462
    },
    {
      "epoch": 1.30615234375e-05,
      "model_forward_time": 5.606014251708984,
      "step": 13696
    },
    {
      "epoch": 1.30615234375e-05,
      "step": 13696,
      "training_step_time": 17.470215320587158
    },
    {
      "epoch": 1.3062477111816405e-05,
      "model_forward_time": 5.599780321121216,
      "step": 13697
    },
    {
      "epoch": 1.3062477111816405e-05,
      "step": 13697,
      "training_step_time": 17.045904636383057
    },
    {
      "epoch": 1.3063430786132813e-05,
      "model_forward_time": 5.583809852600098,
      "step": 13698
    },
    {
      "epoch": 1.3063430786132813e-05,
      "step": 13698,
      "training_step_time": 17.02318525314331
    },
    {
      "epoch": 1.3064384460449219e-05,
      "model_forward_time": 5.58801531791687,
      "step": 13699
    },
    {
      "epoch": 1.3064384460449219e-05,
      "step": 13699,
      "training_step_time": 17.03704571723938
    },
    {
      "epoch": 1.3065338134765624e-05,
      "grad_norm": 0.8026484847068787,
      "learning_rate": 1e-05,
      "loss": 0.0606,
      "step": 13700
    },
    {
      "epoch": 1.3065338134765624e-05,
      "model_forward_time": 5.586089134216309,
      "step": 13700
    },
    {
      "dynamics_loss_avg": 0.057589429803192614,
      "epoch": 1.3065338134765624e-05,
      "step": 13700
    },
    {
      "action_loss_avg": 0.007652366440743208,
      "epoch": 1.3065338134765624e-05,
      "step": 13700
    },
    {
      "epoch": 1.3065338134765624e-05,
      "step": 13700,
      "training_step_time": 17.05524492263794
    },
    {
      "epoch": 1.3066291809082032e-05,
      "model_forward_time": 5.611366510391235,
      "step": 13701
    },
    {
      "epoch": 1.3066291809082032e-05,
      "step": 13701,
      "training_step_time": 17.119176387786865
    },
    {
      "epoch": 1.3067245483398437e-05,
      "model_forward_time": 5.995726585388184,
      "step": 13702
    },
    {
      "epoch": 1.3067245483398437e-05,
      "step": 13702,
      "training_step_time": 17.514594316482544
    },
    {
      "epoch": 1.3068199157714843e-05,
      "model_forward_time": 5.6089794635772705,
      "step": 13703
    },
    {
      "epoch": 1.3068199157714843e-05,
      "step": 13703,
      "training_step_time": 17.091292142868042
    },
    {
      "epoch": 1.306915283203125e-05,
      "model_forward_time": 5.6105797290802,
      "step": 13704
    },
    {
      "epoch": 1.306915283203125e-05,
      "step": 13704,
      "training_step_time": 17.107600450515747
    },
    {
      "epoch": 1.3070106506347656e-05,
      "model_forward_time": 5.6090052127838135,
      "step": 13705
    },
    {
      "epoch": 1.3070106506347656e-05,
      "step": 13705,
      "training_step_time": 17.094395399093628
    },
    {
      "epoch": 1.3071060180664062e-05,
      "model_forward_time": 5.607000112533569,
      "step": 13706
    },
    {
      "epoch": 1.3071060180664062e-05,
      "step": 13706,
      "training_step_time": 17.05349063873291
    },
    {
      "epoch": 1.307201385498047e-05,
      "model_forward_time": 5.934582948684692,
      "step": 13707
    },
    {
      "epoch": 1.307201385498047e-05,
      "step": 13707,
      "training_step_time": 17.374485969543457
    },
    {
      "epoch": 1.3072967529296875e-05,
      "model_forward_time": 5.565054416656494,
      "step": 13708
    },
    {
      "epoch": 1.3072967529296875e-05,
      "step": 13708,
      "training_step_time": 16.97977352142334
    },
    {
      "epoch": 1.3073921203613281e-05,
      "model_forward_time": 5.567535638809204,
      "step": 13709
    },
    {
      "epoch": 1.3073921203613281e-05,
      "step": 13709,
      "training_step_time": 16.9699604511261
    },
    {
      "epoch": 1.3074874877929688e-05,
      "grad_norm": 0.3122875988483429,
      "learning_rate": 1e-05,
      "loss": 0.054,
      "step": 13710
    },
    {
      "epoch": 1.3074874877929688e-05,
      "model_forward_time": 5.576629400253296,
      "step": 13710
    },
    {
      "dynamics_loss_avg": 0.04689689073711634,
      "epoch": 1.3074874877929688e-05,
      "step": 13710
    },
    {
      "action_loss_avg": 0.005818476108834148,
      "epoch": 1.3074874877929688e-05,
      "step": 13710
    },
    {
      "epoch": 1.3074874877929688e-05,
      "step": 13710,
      "training_step_time": 17.009947061538696
    },
    {
      "epoch": 1.3075828552246094e-05,
      "model_forward_time": 5.581525087356567,
      "step": 13711
    },
    {
      "epoch": 1.3075828552246094e-05,
      "step": 13711,
      "training_step_time": 17.02540349960327
    },
    {
      "epoch": 1.30767822265625e-05,
      "model_forward_time": 5.580313205718994,
      "step": 13712
    },
    {
      "epoch": 1.30767822265625e-05,
      "step": 13712,
      "training_step_time": 17.024856090545654
    },
    {
      "epoch": 1.3077735900878905e-05,
      "model_forward_time": 5.989536285400391,
      "step": 13713
    },
    {
      "epoch": 1.3077735900878905e-05,
      "step": 13713,
      "training_step_time": 17.469624757766724
    },
    {
      "epoch": 1.3078689575195313e-05,
      "model_forward_time": 5.602056503295898,
      "step": 13714
    },
    {
      "epoch": 1.3078689575195313e-05,
      "step": 13714,
      "training_step_time": 17.053953647613525
    },
    {
      "epoch": 1.3079643249511719e-05,
      "model_forward_time": 5.597972869873047,
      "step": 13715
    },
    {
      "epoch": 1.3079643249511719e-05,
      "step": 13715,
      "training_step_time": 17.057948112487793
    },
    {
      "epoch": 1.3080596923828124e-05,
      "model_forward_time": 5.610360622406006,
      "step": 13716
    },
    {
      "epoch": 1.3080596923828124e-05,
      "step": 13716,
      "training_step_time": 17.07674551010132
    },
    {
      "epoch": 1.3081550598144532e-05,
      "model_forward_time": 5.605689525604248,
      "step": 13717
    },
    {
      "epoch": 1.3081550598144532e-05,
      "step": 13717,
      "training_step_time": 17.081348419189453
    },
    {
      "epoch": 1.3082504272460937e-05,
      "model_forward_time": 5.604345083236694,
      "step": 13718
    },
    {
      "epoch": 1.3082504272460937e-05,
      "step": 13718,
      "training_step_time": 17.46871280670166
    },
    {
      "epoch": 1.3083457946777343e-05,
      "model_forward_time": 5.626526355743408,
      "step": 13719
    },
    {
      "epoch": 1.3083457946777343e-05,
      "step": 13719,
      "training_step_time": 17.126355171203613
    },
    {
      "epoch": 1.308441162109375e-05,
      "grad_norm": 1.0664228200912476,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 13720
    },
    {
      "epoch": 1.308441162109375e-05,
      "model_forward_time": 5.60157036781311,
      "step": 13720
    },
    {
      "dynamics_loss_avg": 0.05808775760233402,
      "epoch": 1.308441162109375e-05,
      "step": 13720
    },
    {
      "action_loss_avg": 0.0065647543407976626,
      "epoch": 1.308441162109375e-05,
      "step": 13720
    },
    {
      "epoch": 1.308441162109375e-05,
      "step": 13720,
      "training_step_time": 17.10278058052063
    },
    {
      "epoch": 1.3085365295410156e-05,
      "model_forward_time": 5.608865737915039,
      "step": 13721
    },
    {
      "epoch": 1.3085365295410156e-05,
      "step": 13721,
      "training_step_time": 17.095909357070923
    },
    {
      "epoch": 1.3086318969726562e-05,
      "model_forward_time": 5.605201005935669,
      "step": 13722
    },
    {
      "epoch": 1.3086318969726562e-05,
      "step": 13722,
      "training_step_time": 17.08707880973816
    },
    {
      "epoch": 1.308727264404297e-05,
      "model_forward_time": 5.604954957962036,
      "step": 13723
    },
    {
      "epoch": 1.308727264404297e-05,
      "step": 13723,
      "training_step_time": 17.096808910369873
    },
    {
      "epoch": 1.3088226318359375e-05,
      "model_forward_time": 6.005144834518433,
      "step": 13724
    },
    {
      "epoch": 1.3088226318359375e-05,
      "step": 13724,
      "training_step_time": 17.48725438117981
    },
    {
      "epoch": 1.3089179992675781e-05,
      "model_forward_time": 5.5995800495147705,
      "step": 13725
    },
    {
      "epoch": 1.3089179992675781e-05,
      "step": 13725,
      "training_step_time": 17.081939935684204
    },
    {
      "epoch": 1.3090133666992188e-05,
      "model_forward_time": 5.6010191440582275,
      "step": 13726
    },
    {
      "epoch": 1.3090133666992188e-05,
      "step": 13726,
      "training_step_time": 17.054211378097534
    },
    {
      "epoch": 1.3091087341308594e-05,
      "model_forward_time": 5.5986714363098145,
      "step": 13727
    },
    {
      "epoch": 1.3091087341308594e-05,
      "step": 13727,
      "training_step_time": 17.06594944000244
    },
    {
      "epoch": 1.3092041015625e-05,
      "model_forward_time": 5.604310989379883,
      "step": 13728
    },
    {
      "epoch": 1.3092041015625e-05,
      "step": 13728,
      "training_step_time": 17.06671690940857
    },
    {
      "epoch": 1.3092994689941405e-05,
      "model_forward_time": 5.6048688888549805,
      "step": 13729
    },
    {
      "epoch": 1.3092994689941405e-05,
      "step": 13729,
      "training_step_time": 17.426251888275146
    },
    {
      "epoch": 1.3093948364257813e-05,
      "grad_norm": 0.6704140305519104,
      "learning_rate": 1e-05,
      "loss": 0.0755,
      "step": 13730
    },
    {
      "epoch": 1.3093948364257813e-05,
      "model_forward_time": 5.601586818695068,
      "step": 13730
    },
    {
      "dynamics_loss_avg": 0.061319997534155846,
      "epoch": 1.3093948364257813e-05,
      "step": 13730
    },
    {
      "action_loss_avg": 0.009504902898333967,
      "epoch": 1.3093948364257813e-05,
      "step": 13730
    },
    {
      "epoch": 1.3093948364257813e-05,
      "step": 13730,
      "training_step_time": 17.05097007751465
    },
    {
      "epoch": 1.3094902038574219e-05,
      "model_forward_time": 5.58695912361145,
      "step": 13731
    },
    {
      "epoch": 1.3094902038574219e-05,
      "step": 13731,
      "training_step_time": 17.04510188102722
    },
    {
      "epoch": 1.3095855712890624e-05,
      "model_forward_time": 5.602772235870361,
      "step": 13732
    },
    {
      "epoch": 1.3095855712890624e-05,
      "step": 13732,
      "training_step_time": 17.085829496383667
    },
    {
      "epoch": 1.3096809387207032e-05,
      "model_forward_time": 5.616986989974976,
      "step": 13733
    },
    {
      "epoch": 1.3096809387207032e-05,
      "step": 13733,
      "training_step_time": 17.10585594177246
    },
    {
      "epoch": 1.3097763061523437e-05,
      "model_forward_time": 5.618361473083496,
      "step": 13734
    },
    {
      "epoch": 1.3097763061523437e-05,
      "step": 13734,
      "training_step_time": 17.084983587265015
    },
    {
      "epoch": 1.3098716735839843e-05,
      "model_forward_time": 5.953079700469971,
      "step": 13735
    },
    {
      "epoch": 1.3098716735839843e-05,
      "step": 13735,
      "training_step_time": 17.377557039260864
    },
    {
      "epoch": 1.309967041015625e-05,
      "model_forward_time": 5.577573776245117,
      "step": 13736
    },
    {
      "epoch": 1.309967041015625e-05,
      "step": 13736,
      "training_step_time": 16.99551248550415
    },
    {
      "epoch": 1.3100624084472656e-05,
      "model_forward_time": 5.572045803070068,
      "step": 13737
    },
    {
      "epoch": 1.3100624084472656e-05,
      "step": 13737,
      "training_step_time": 17.009679555892944
    },
    {
      "epoch": 1.3101577758789062e-05,
      "model_forward_time": 5.568553447723389,
      "step": 13738
    },
    {
      "epoch": 1.3101577758789062e-05,
      "step": 13738,
      "training_step_time": 16.983506441116333
    },
    {
      "epoch": 1.310253143310547e-05,
      "model_forward_time": 5.575725317001343,
      "step": 13739
    },
    {
      "epoch": 1.310253143310547e-05,
      "step": 13739,
      "training_step_time": 16.99550223350525
    },
    {
      "epoch": 1.3103485107421875e-05,
      "grad_norm": 1.1272695064544678,
      "learning_rate": 1e-05,
      "loss": 0.0615,
      "step": 13740
    },
    {
      "epoch": 1.3103485107421875e-05,
      "model_forward_time": 5.963104963302612,
      "step": 13740
    },
    {
      "dynamics_loss_avg": 0.055538121797144416,
      "epoch": 1.3103485107421875e-05,
      "step": 13740
    },
    {
      "action_loss_avg": 0.007585107209160924,
      "epoch": 1.3103485107421875e-05,
      "step": 13740
    },
    {
      "epoch": 1.3103485107421875e-05,
      "step": 13740,
      "training_step_time": 17.397093296051025
    },
    {
      "epoch": 1.3104438781738281e-05,
      "model_forward_time": 5.590808629989624,
      "step": 13741
    },
    {
      "epoch": 1.3104438781738281e-05,
      "step": 13741,
      "training_step_time": 17.06516194343567
    },
    {
      "epoch": 1.3105392456054688e-05,
      "model_forward_time": 5.604780197143555,
      "step": 13742
    },
    {
      "epoch": 1.3105392456054688e-05,
      "step": 13742,
      "training_step_time": 17.077470541000366
    },
    {
      "epoch": 1.3106346130371094e-05,
      "model_forward_time": 5.600949287414551,
      "step": 13743
    },
    {
      "epoch": 1.3106346130371094e-05,
      "step": 13743,
      "training_step_time": 17.08339810371399
    },
    {
      "epoch": 1.31072998046875e-05,
      "model_forward_time": 5.6175196170806885,
      "step": 13744
    },
    {
      "epoch": 1.31072998046875e-05,
      "step": 13744,
      "training_step_time": 17.099430799484253
    },
    {
      "epoch": 1.3108253479003906e-05,
      "model_forward_time": 5.610597372055054,
      "step": 13745
    },
    {
      "epoch": 1.3108253479003906e-05,
      "step": 13745,
      "training_step_time": 17.061173677444458
    },
    {
      "epoch": 1.3109207153320313e-05,
      "model_forward_time": 5.956754446029663,
      "step": 13746
    },
    {
      "epoch": 1.3109207153320313e-05,
      "step": 13746,
      "training_step_time": 17.401479721069336
    },
    {
      "epoch": 1.3110160827636719e-05,
      "model_forward_time": 5.594323635101318,
      "step": 13747
    },
    {
      "epoch": 1.3110160827636719e-05,
      "step": 13747,
      "training_step_time": 17.04553723335266
    },
    {
      "epoch": 1.3111114501953124e-05,
      "model_forward_time": 5.594184875488281,
      "step": 13748
    },
    {
      "epoch": 1.3111114501953124e-05,
      "step": 13748,
      "training_step_time": 17.046315908432007
    },
    {
      "epoch": 1.3112068176269532e-05,
      "model_forward_time": 5.59796142578125,
      "step": 13749
    },
    {
      "epoch": 1.3112068176269532e-05,
      "step": 13749,
      "training_step_time": 17.062799215316772
    },
    {
      "epoch": 1.3113021850585938e-05,
      "grad_norm": 0.8927490711212158,
      "learning_rate": 1e-05,
      "loss": 0.0719,
      "step": 13750
    },
    {
      "epoch": 1.3113021850585938e-05,
      "model_forward_time": 5.598839521408081,
      "step": 13750
    },
    {
      "dynamics_loss_avg": 0.06807681545615196,
      "epoch": 1.3113021850585938e-05,
      "step": 13750
    },
    {
      "action_loss_avg": 0.009467345010489226,
      "epoch": 1.3113021850585938e-05,
      "step": 13750
    },
    {
      "epoch": 1.3113021850585938e-05,
      "step": 13750,
      "training_step_time": 17.061116695404053
    },
    {
      "epoch": 1.3113975524902343e-05,
      "model_forward_time": 5.602033615112305,
      "step": 13751
    },
    {
      "epoch": 1.3113975524902343e-05,
      "step": 13751,
      "training_step_time": 17.418185234069824
    },
    {
      "epoch": 1.311492919921875e-05,
      "model_forward_time": 5.591647624969482,
      "step": 13752
    },
    {
      "epoch": 1.311492919921875e-05,
      "step": 13752,
      "training_step_time": 17.029473543167114
    },
    {
      "epoch": 1.3115882873535156e-05,
      "model_forward_time": 5.590126991271973,
      "step": 13753
    },
    {
      "epoch": 1.3115882873535156e-05,
      "step": 13753,
      "training_step_time": 17.010498762130737
    },
    {
      "epoch": 1.3116836547851562e-05,
      "model_forward_time": 5.5901453495025635,
      "step": 13754
    },
    {
      "epoch": 1.3116836547851562e-05,
      "step": 13754,
      "training_step_time": 17.040930032730103
    },
    {
      "epoch": 1.311779022216797e-05,
      "model_forward_time": 5.593867778778076,
      "step": 13755
    },
    {
      "epoch": 1.311779022216797e-05,
      "step": 13755,
      "training_step_time": 17.076388835906982
    },
    {
      "epoch": 1.3118743896484375e-05,
      "model_forward_time": 5.614776372909546,
      "step": 13756
    },
    {
      "epoch": 1.3118743896484375e-05,
      "step": 13756,
      "training_step_time": 17.047508001327515
    },
    {
      "epoch": 1.3119697570800781e-05,
      "model_forward_time": 5.970186233520508,
      "step": 13757
    },
    {
      "epoch": 1.3119697570800781e-05,
      "step": 13757,
      "training_step_time": 17.416234016418457
    },
    {
      "epoch": 1.3120651245117187e-05,
      "model_forward_time": 5.5955610275268555,
      "step": 13758
    },
    {
      "epoch": 1.3120651245117187e-05,
      "step": 13758,
      "training_step_time": 17.048869371414185
    },
    {
      "epoch": 1.3121604919433594e-05,
      "model_forward_time": 5.59469199180603,
      "step": 13759
    },
    {
      "epoch": 1.3121604919433594e-05,
      "step": 13759,
      "training_step_time": 17.06672716140747
    },
    {
      "epoch": 1.312255859375e-05,
      "grad_norm": 0.7956851720809937,
      "learning_rate": 1e-05,
      "loss": 0.0628,
      "step": 13760
    },
    {
      "epoch": 1.312255859375e-05,
      "model_forward_time": 5.6001136302948,
      "step": 13760
    },
    {
      "dynamics_loss_avg": 0.05407025907188654,
      "epoch": 1.312255859375e-05,
      "step": 13760
    },
    {
      "action_loss_avg": 0.005831548199057579,
      "epoch": 1.312255859375e-05,
      "step": 13760
    },
    {
      "epoch": 1.312255859375e-05,
      "step": 13760,
      "training_step_time": 17.06361436843872
    },
    {
      "epoch": 1.3123512268066406e-05,
      "model_forward_time": 5.597844123840332,
      "step": 13761
    },
    {
      "epoch": 1.3123512268066406e-05,
      "step": 13761,
      "training_step_time": 17.05776286125183
    },
    {
      "epoch": 1.3124465942382813e-05,
      "model_forward_time": 5.61060094833374,
      "step": 13762
    },
    {
      "epoch": 1.3124465942382813e-05,
      "step": 13762,
      "training_step_time": 17.429075956344604
    },
    {
      "epoch": 1.3125419616699219e-05,
      "model_forward_time": 5.602222442626953,
      "step": 13763
    },
    {
      "epoch": 1.3125419616699219e-05,
      "step": 13763,
      "training_step_time": 17.062775373458862
    },
    {
      "epoch": 1.3126373291015624e-05,
      "model_forward_time": 5.5962910652160645,
      "step": 13764
    },
    {
      "epoch": 1.3126373291015624e-05,
      "step": 13764,
      "training_step_time": 17.04991888999939
    },
    {
      "epoch": 1.3127326965332032e-05,
      "model_forward_time": 5.590275049209595,
      "step": 13765
    },
    {
      "epoch": 1.3127326965332032e-05,
      "step": 13765,
      "training_step_time": 17.04917812347412
    },
    {
      "epoch": 1.3128280639648438e-05,
      "model_forward_time": 5.595873832702637,
      "step": 13766
    },
    {
      "epoch": 1.3128280639648438e-05,
      "step": 13766,
      "training_step_time": 17.065478801727295
    },
    {
      "epoch": 1.3129234313964843e-05,
      "model_forward_time": 5.602078437805176,
      "step": 13767
    },
    {
      "epoch": 1.3129234313964843e-05,
      "step": 13767,
      "training_step_time": 17.07958483695984
    },
    {
      "epoch": 1.313018798828125e-05,
      "model_forward_time": 5.989729404449463,
      "step": 13768
    },
    {
      "epoch": 1.313018798828125e-05,
      "step": 13768,
      "training_step_time": 17.454994201660156
    },
    {
      "epoch": 1.3131141662597656e-05,
      "model_forward_time": 5.602159023284912,
      "step": 13769
    },
    {
      "epoch": 1.3131141662597656e-05,
      "step": 13769,
      "training_step_time": 17.067034006118774
    },
    {
      "epoch": 1.3132095336914062e-05,
      "grad_norm": 0.6830335259437561,
      "learning_rate": 1e-05,
      "loss": 0.0768,
      "step": 13770
    },
    {
      "epoch": 1.3132095336914062e-05,
      "model_forward_time": 5.598613739013672,
      "step": 13770
    },
    {
      "dynamics_loss_avg": 0.06523536797612906,
      "epoch": 1.3132095336914062e-05,
      "step": 13770
    },
    {
      "action_loss_avg": 0.00970953218638897,
      "epoch": 1.3132095336914062e-05,
      "step": 13770
    },
    {
      "epoch": 1.3132095336914062e-05,
      "step": 13770,
      "training_step_time": 17.063512802124023
    },
    {
      "epoch": 1.313304901123047e-05,
      "model_forward_time": 5.599672794342041,
      "step": 13771
    },
    {
      "epoch": 1.313304901123047e-05,
      "step": 13771,
      "training_step_time": 17.057633876800537
    },
    {
      "epoch": 1.3134002685546875e-05,
      "model_forward_time": 5.597847938537598,
      "step": 13772
    },
    {
      "epoch": 1.3134002685546875e-05,
      "step": 13772,
      "training_step_time": 17.072025299072266
    },
    {
      "epoch": 1.3134956359863281e-05,
      "model_forward_time": 5.959767580032349,
      "step": 13773
    },
    {
      "epoch": 1.3134956359863281e-05,
      "step": 13773,
      "training_step_time": 17.387621879577637
    },
    {
      "epoch": 1.3135910034179687e-05,
      "model_forward_time": 5.587906360626221,
      "step": 13774
    },
    {
      "epoch": 1.3135910034179687e-05,
      "step": 13774,
      "training_step_time": 17.04804754257202
    },
    {
      "epoch": 1.3136863708496094e-05,
      "model_forward_time": 5.598456859588623,
      "step": 13775
    },
    {
      "epoch": 1.3136863708496094e-05,
      "step": 13775,
      "training_step_time": 17.091935634613037
    },
    {
      "epoch": 1.31378173828125e-05,
      "model_forward_time": 5.617908716201782,
      "step": 13776
    },
    {
      "epoch": 1.31378173828125e-05,
      "step": 13776,
      "training_step_time": 17.12078356742859
    },
    {
      "epoch": 1.3138771057128906e-05,
      "model_forward_time": 5.619466304779053,
      "step": 13777
    },
    {
      "epoch": 1.3138771057128906e-05,
      "step": 13777,
      "training_step_time": 17.123568773269653
    },
    {
      "epoch": 1.3139724731445313e-05,
      "model_forward_time": 5.624006748199463,
      "step": 13778
    },
    {
      "epoch": 1.3139724731445313e-05,
      "step": 13778,
      "training_step_time": 17.12418842315674
    },
    {
      "epoch": 1.3140678405761719e-05,
      "model_forward_time": 5.9988367557525635,
      "step": 13779
    },
    {
      "epoch": 1.3140678405761719e-05,
      "step": 13779,
      "training_step_time": 17.483087062835693
    },
    {
      "epoch": 1.3141632080078124e-05,
      "grad_norm": 0.9088059067726135,
      "learning_rate": 1e-05,
      "loss": 0.063,
      "step": 13780
    },
    {
      "epoch": 1.3141632080078124e-05,
      "model_forward_time": 5.616484642028809,
      "step": 13780
    },
    {
      "dynamics_loss_avg": 0.05897023305296898,
      "epoch": 1.3141632080078124e-05,
      "step": 13780
    },
    {
      "action_loss_avg": 0.010337269771844149,
      "epoch": 1.3141632080078124e-05,
      "step": 13780
    },
    {
      "epoch": 1.3141632080078124e-05,
      "step": 13780,
      "training_step_time": 17.09929585456848
    },
    {
      "epoch": 1.3142585754394532e-05,
      "model_forward_time": 5.613213539123535,
      "step": 13781
    },
    {
      "epoch": 1.3142585754394532e-05,
      "step": 13781,
      "training_step_time": 17.114823579788208
    },
    {
      "epoch": 1.3143539428710938e-05,
      "model_forward_time": 5.612663507461548,
      "step": 13782
    },
    {
      "epoch": 1.3143539428710938e-05,
      "step": 13782,
      "training_step_time": 17.10214328765869
    },
    {
      "epoch": 1.3144493103027343e-05,
      "model_forward_time": 5.615587472915649,
      "step": 13783
    },
    {
      "epoch": 1.3144493103027343e-05,
      "step": 13783,
      "training_step_time": 17.107163429260254
    },
    {
      "epoch": 1.314544677734375e-05,
      "model_forward_time": 5.629640340805054,
      "step": 13784
    },
    {
      "epoch": 1.314544677734375e-05,
      "step": 13784,
      "training_step_time": 17.418959379196167
    },
    {
      "epoch": 1.3146400451660156e-05,
      "model_forward_time": 5.586009502410889,
      "step": 13785
    },
    {
      "epoch": 1.3146400451660156e-05,
      "step": 13785,
      "training_step_time": 17.024768829345703
    },
    {
      "epoch": 1.3147354125976562e-05,
      "model_forward_time": 5.578804016113281,
      "step": 13786
    },
    {
      "epoch": 1.3147354125976562e-05,
      "step": 13786,
      "training_step_time": 17.015671730041504
    },
    {
      "epoch": 1.314830780029297e-05,
      "model_forward_time": 5.586443901062012,
      "step": 13787
    },
    {
      "epoch": 1.314830780029297e-05,
      "step": 13787,
      "training_step_time": 17.04152798652649
    },
    {
      "epoch": 1.3149261474609375e-05,
      "model_forward_time": 5.591327667236328,
      "step": 13788
    },
    {
      "epoch": 1.3149261474609375e-05,
      "step": 13788,
      "training_step_time": 17.041584730148315
    },
    {
      "epoch": 1.3150215148925781e-05,
      "model_forward_time": 5.601943016052246,
      "step": 13789
    },
    {
      "epoch": 1.3150215148925781e-05,
      "step": 13789,
      "training_step_time": 17.08406949043274
    },
    {
      "epoch": 1.3151168823242187e-05,
      "grad_norm": 0.6639204621315002,
      "learning_rate": 1e-05,
      "loss": 0.0755,
      "step": 13790
    },
    {
      "epoch": 1.3151168823242187e-05,
      "model_forward_time": 5.982924699783325,
      "step": 13790
    },
    {
      "dynamics_loss_avg": 0.05789553858339787,
      "epoch": 1.3151168823242187e-05,
      "step": 13790
    },
    {
      "action_loss_avg": 0.007564540370367467,
      "epoch": 1.3151168823242187e-05,
      "step": 13790
    },
    {
      "epoch": 1.3151168823242187e-05,
      "step": 13790,
      "training_step_time": 17.44434881210327
    },
    {
      "epoch": 1.3152122497558594e-05,
      "model_forward_time": 5.601618051528931,
      "step": 13791
    },
    {
      "epoch": 1.3152122497558594e-05,
      "step": 13791,
      "training_step_time": 17.055105686187744
    },
    {
      "epoch": 1.3153076171875e-05,
      "model_forward_time": 5.596469879150391,
      "step": 13792
    },
    {
      "epoch": 1.3153076171875e-05,
      "step": 13792,
      "training_step_time": 17.052549600601196
    },
    {
      "epoch": 1.3154029846191406e-05,
      "model_forward_time": 5.5910937786102295,
      "step": 13793
    },
    {
      "epoch": 1.3154029846191406e-05,
      "step": 13793,
      "training_step_time": 17.0280818939209
    },
    {
      "epoch": 1.3154983520507813e-05,
      "model_forward_time": 5.583994626998901,
      "step": 13794
    },
    {
      "epoch": 1.3154983520507813e-05,
      "step": 13794,
      "training_step_time": 17.020259618759155
    },
    {
      "epoch": 1.3155937194824219e-05,
      "model_forward_time": 5.598655939102173,
      "step": 13795
    },
    {
      "epoch": 1.3155937194824219e-05,
      "step": 13795,
      "training_step_time": 17.41828179359436
    },
    {
      "epoch": 1.3156890869140624e-05,
      "model_forward_time": 5.599031686782837,
      "step": 13796
    },
    {
      "epoch": 1.3156890869140624e-05,
      "step": 13796,
      "training_step_time": 17.068488121032715
    },
    {
      "epoch": 1.3157844543457032e-05,
      "model_forward_time": 5.605353593826294,
      "step": 13797
    },
    {
      "epoch": 1.3157844543457032e-05,
      "step": 13797,
      "training_step_time": 17.081461668014526
    },
    {
      "epoch": 1.3158798217773438e-05,
      "model_forward_time": 5.596580982208252,
      "step": 13798
    },
    {
      "epoch": 1.3158798217773438e-05,
      "step": 13798,
      "training_step_time": 17.06213068962097
    },
    {
      "epoch": 1.3159751892089843e-05,
      "model_forward_time": 5.6161394119262695,
      "step": 13799
    },
    {
      "epoch": 1.3159751892089843e-05,
      "step": 13799,
      "training_step_time": 17.098395109176636
    },
    {
      "epoch": 1.316070556640625e-05,
      "grad_norm": 0.47741755843162537,
      "learning_rate": 1e-05,
      "loss": 0.0734,
      "step": 13800
    },
    {
      "epoch": 1.316070556640625e-05,
      "model_forward_time": 5.604851722717285,
      "step": 13800
    },
    {
      "dynamics_loss_avg": 0.06566395349800587,
      "epoch": 1.316070556640625e-05,
      "step": 13800
    },
    {
      "action_loss_avg": 0.011476444290019571,
      "epoch": 1.316070556640625e-05,
      "step": 13800
    },
    {
      "epoch": 1.316070556640625e-05,
      "step": 13800,
      "training_step_time": 17.10086989402771
    },
    {
      "epoch": 1.3161659240722656e-05,
      "model_forward_time": 5.980253458023071,
      "step": 13801
    },
    {
      "epoch": 1.3161659240722656e-05,
      "step": 13801,
      "training_step_time": 17.459287881851196
    },
    {
      "epoch": 1.3162612915039062e-05,
      "model_forward_time": 5.604090213775635,
      "step": 13802
    },
    {
      "epoch": 1.3162612915039062e-05,
      "step": 13802,
      "training_step_time": 17.072482347488403
    },
    {
      "epoch": 1.316356658935547e-05,
      "model_forward_time": 5.6097636222839355,
      "step": 13803
    },
    {
      "epoch": 1.316356658935547e-05,
      "step": 13803,
      "training_step_time": 17.03875422477722
    },
    {
      "epoch": 1.3164520263671875e-05,
      "model_forward_time": 5.580132722854614,
      "step": 13804
    },
    {
      "epoch": 1.3164520263671875e-05,
      "step": 13804,
      "training_step_time": 16.98908519744873
    },
    {
      "epoch": 1.3165473937988281e-05,
      "model_forward_time": 5.575549840927124,
      "step": 13805
    },
    {
      "epoch": 1.3165473937988281e-05,
      "step": 13805,
      "training_step_time": 17.01802706718445
    },
    {
      "epoch": 1.3166427612304687e-05,
      "model_forward_time": 5.944946765899658,
      "step": 13806
    },
    {
      "epoch": 1.3166427612304687e-05,
      "step": 13806,
      "training_step_time": 17.383856534957886
    },
    {
      "epoch": 1.3167381286621094e-05,
      "model_forward_time": 5.586667776107788,
      "step": 13807
    },
    {
      "epoch": 1.3167381286621094e-05,
      "step": 13807,
      "training_step_time": 17.045859575271606
    },
    {
      "epoch": 1.31683349609375e-05,
      "model_forward_time": 5.605244159698486,
      "step": 13808
    },
    {
      "epoch": 1.31683349609375e-05,
      "step": 13808,
      "training_step_time": 17.090802907943726
    },
    {
      "epoch": 1.3169288635253906e-05,
      "model_forward_time": 5.6141884326934814,
      "step": 13809
    },
    {
      "epoch": 1.3169288635253906e-05,
      "step": 13809,
      "training_step_time": 17.133130073547363
    },
    {
      "epoch": 1.3170242309570313e-05,
      "grad_norm": 0.6180357336997986,
      "learning_rate": 1e-05,
      "loss": 0.06,
      "step": 13810
    },
    {
      "epoch": 1.3170242309570313e-05,
      "model_forward_time": 5.6278722286224365,
      "step": 13810
    },
    {
      "dynamics_loss_avg": 0.058280695043504235,
      "epoch": 1.3170242309570313e-05,
      "step": 13810
    },
    {
      "action_loss_avg": 0.008148907567374408,
      "epoch": 1.3170242309570313e-05,
      "step": 13810
    },
    {
      "epoch": 1.3170242309570313e-05,
      "step": 13810,
      "training_step_time": 17.1307532787323
    },
    {
      "epoch": 1.3171195983886719e-05,
      "model_forward_time": 5.602679252624512,
      "step": 13811
    },
    {
      "epoch": 1.3171195983886719e-05,
      "step": 13811,
      "training_step_time": 17.06654381752014
    },
    {
      "epoch": 1.3172149658203124e-05,
      "model_forward_time": 5.972370147705078,
      "step": 13812
    },
    {
      "epoch": 1.3172149658203124e-05,
      "step": 13812,
      "training_step_time": 17.40832781791687
    },
    {
      "epoch": 1.3173103332519532e-05,
      "model_forward_time": 5.586702108383179,
      "step": 13813
    },
    {
      "epoch": 1.3173103332519532e-05,
      "step": 13813,
      "training_step_time": 17.03543496131897
    },
    {
      "epoch": 1.3174057006835938e-05,
      "model_forward_time": 5.592413663864136,
      "step": 13814
    },
    {
      "epoch": 1.3174057006835938e-05,
      "step": 13814,
      "training_step_time": 17.05604124069214
    },
    {
      "epoch": 1.3175010681152343e-05,
      "model_forward_time": 5.6020121574401855,
      "step": 13815
    },
    {
      "epoch": 1.3175010681152343e-05,
      "step": 13815,
      "training_step_time": 17.082571744918823
    },
    {
      "epoch": 1.317596435546875e-05,
      "model_forward_time": 5.604226350784302,
      "step": 13816
    },
    {
      "epoch": 1.317596435546875e-05,
      "step": 13816,
      "training_step_time": 17.083677768707275
    },
    {
      "epoch": 1.3176918029785156e-05,
      "model_forward_time": 5.607110500335693,
      "step": 13817
    },
    {
      "epoch": 1.3176918029785156e-05,
      "step": 13817,
      "training_step_time": 17.442936658859253
    },
    {
      "epoch": 1.3177871704101562e-05,
      "model_forward_time": 5.604332685470581,
      "step": 13818
    },
    {
      "epoch": 1.3177871704101562e-05,
      "step": 13818,
      "training_step_time": 17.076338529586792
    },
    {
      "epoch": 1.317882537841797e-05,
      "model_forward_time": 5.59845757484436,
      "step": 13819
    },
    {
      "epoch": 1.317882537841797e-05,
      "step": 13819,
      "training_step_time": 17.050567626953125
    },
    {
      "epoch": 1.3179779052734375e-05,
      "grad_norm": 1.2994564771652222,
      "learning_rate": 1e-05,
      "loss": 0.0784,
      "step": 13820
    },
    {
      "epoch": 1.3179779052734375e-05,
      "model_forward_time": 5.60590386390686,
      "step": 13820
    },
    {
      "dynamics_loss_avg": 0.05965173095464706,
      "epoch": 1.3179779052734375e-05,
      "step": 13820
    },
    {
      "action_loss_avg": 0.008809353597462178,
      "epoch": 1.3179779052734375e-05,
      "step": 13820
    },
    {
      "epoch": 1.3179779052734375e-05,
      "step": 13820,
      "training_step_time": 17.053699254989624
    },
    {
      "epoch": 1.3180732727050781e-05,
      "model_forward_time": 5.602370262145996,
      "step": 13821
    },
    {
      "epoch": 1.3180732727050781e-05,
      "step": 13821,
      "training_step_time": 17.05451011657715
    },
    {
      "epoch": 1.3181686401367187e-05,
      "model_forward_time": 5.5862226486206055,
      "step": 13822
    },
    {
      "epoch": 1.3181686401367187e-05,
      "step": 13822,
      "training_step_time": 17.019542932510376
    },
    {
      "epoch": 1.3182640075683594e-05,
      "model_forward_time": 5.973333120346069,
      "step": 13823
    },
    {
      "epoch": 1.3182640075683594e-05,
      "step": 13823,
      "training_step_time": 17.4243426322937
    },
    {
      "epoch": 1.318359375e-05,
      "model_forward_time": 5.60008978843689,
      "step": 13824
    },
    {
      "epoch": 1.318359375e-05,
      "step": 13824,
      "training_step_time": 17.076383352279663
    },
    {
      "epoch": 1.3184547424316406e-05,
      "model_forward_time": 5.598963737487793,
      "step": 13825
    },
    {
      "epoch": 1.3184547424316406e-05,
      "step": 13825,
      "training_step_time": 17.059677600860596
    },
    {
      "epoch": 1.3185501098632813e-05,
      "model_forward_time": 5.606092691421509,
      "step": 13826
    },
    {
      "epoch": 1.3185501098632813e-05,
      "step": 13826,
      "training_step_time": 17.093563079833984
    },
    {
      "epoch": 1.3186454772949219e-05,
      "model_forward_time": 5.615449905395508,
      "step": 13827
    },
    {
      "epoch": 1.3186454772949219e-05,
      "step": 13827,
      "training_step_time": 17.09953022003174
    },
    {
      "epoch": 1.3187408447265624e-05,
      "model_forward_time": 5.610415458679199,
      "step": 13828
    },
    {
      "epoch": 1.3187408447265624e-05,
      "step": 13828,
      "training_step_time": 17.45662021636963
    },
    {
      "epoch": 1.3188362121582032e-05,
      "model_forward_time": 5.603724956512451,
      "step": 13829
    },
    {
      "epoch": 1.3188362121582032e-05,
      "step": 13829,
      "training_step_time": 17.087536573410034
    },
    {
      "epoch": 1.3189315795898438e-05,
      "grad_norm": 0.9741904139518738,
      "learning_rate": 1e-05,
      "loss": 0.0555,
      "step": 13830
    },
    {
      "epoch": 1.3189315795898438e-05,
      "model_forward_time": 5.599793195724487,
      "step": 13830
    },
    {
      "dynamics_loss_avg": 0.051563454978168013,
      "epoch": 1.3189315795898438e-05,
      "step": 13830
    },
    {
      "action_loss_avg": 0.0065176607575267555,
      "epoch": 1.3189315795898438e-05,
      "step": 13830
    },
    {
      "epoch": 1.3189315795898438e-05,
      "step": 13830,
      "training_step_time": 17.062072038650513
    },
    {
      "epoch": 1.3190269470214843e-05,
      "model_forward_time": 5.600154399871826,
      "step": 13831
    },
    {
      "epoch": 1.3190269470214843e-05,
      "step": 13831,
      "training_step_time": 17.057899713516235
    },
    {
      "epoch": 1.319122314453125e-05,
      "model_forward_time": 5.59126353263855,
      "step": 13832
    },
    {
      "epoch": 1.319122314453125e-05,
      "step": 13832,
      "training_step_time": 17.04584550857544
    },
    {
      "epoch": 1.3192176818847656e-05,
      "model_forward_time": 5.601553201675415,
      "step": 13833
    },
    {
      "epoch": 1.3192176818847656e-05,
      "step": 13833,
      "training_step_time": 17.08857274055481
    },
    {
      "epoch": 1.3193130493164062e-05,
      "model_forward_time": 5.992572784423828,
      "step": 13834
    },
    {
      "epoch": 1.3193130493164062e-05,
      "step": 13834,
      "training_step_time": 17.480384826660156
    },
    {
      "epoch": 1.319408416748047e-05,
      "model_forward_time": 5.606348991394043,
      "step": 13835
    },
    {
      "epoch": 1.319408416748047e-05,
      "step": 13835,
      "training_step_time": 17.10877537727356
    },
    {
      "epoch": 1.3195037841796875e-05,
      "model_forward_time": 5.610318660736084,
      "step": 13836
    },
    {
      "epoch": 1.3195037841796875e-05,
      "step": 13836,
      "training_step_time": 17.076625108718872
    },
    {
      "epoch": 1.3195991516113281e-05,
      "model_forward_time": 5.608025074005127,
      "step": 13837
    },
    {
      "epoch": 1.3195991516113281e-05,
      "step": 13837,
      "training_step_time": 17.100462198257446
    },
    {
      "epoch": 1.3196945190429687e-05,
      "model_forward_time": 5.613590478897095,
      "step": 13838
    },
    {
      "epoch": 1.3196945190429687e-05,
      "step": 13838,
      "training_step_time": 17.111189365386963
    },
    {
      "epoch": 1.3197898864746094e-05,
      "model_forward_time": 5.978204250335693,
      "step": 13839
    },
    {
      "epoch": 1.3197898864746094e-05,
      "step": 13839,
      "training_step_time": 17.460416078567505
    },
    {
      "epoch": 1.31988525390625e-05,
      "grad_norm": 0.6702579259872437,
      "learning_rate": 1e-05,
      "loss": 0.0587,
      "step": 13840
    },
    {
      "epoch": 1.31988525390625e-05,
      "model_forward_time": 5.613149404525757,
      "step": 13840
    },
    {
      "dynamics_loss_avg": 0.05467911101877689,
      "epoch": 1.31988525390625e-05,
      "step": 13840
    },
    {
      "action_loss_avg": 0.007314118673093617,
      "epoch": 1.31988525390625e-05,
      "step": 13840
    },
    {
      "epoch": 1.31988525390625e-05,
      "step": 13840,
      "training_step_time": 17.06880235671997
    },
    {
      "epoch": 1.3199806213378906e-05,
      "model_forward_time": 5.596698999404907,
      "step": 13841
    },
    {
      "epoch": 1.3199806213378906e-05,
      "step": 13841,
      "training_step_time": 17.05975103378296
    },
    {
      "epoch": 1.3200759887695313e-05,
      "model_forward_time": 5.604772329330444,
      "step": 13842
    },
    {
      "epoch": 1.3200759887695313e-05,
      "step": 13842,
      "training_step_time": 17.077603578567505
    },
    {
      "epoch": 1.3201713562011719e-05,
      "model_forward_time": 5.594825506210327,
      "step": 13843
    },
    {
      "epoch": 1.3201713562011719e-05,
      "step": 13843,
      "training_step_time": 17.06910991668701
    },
    {
      "epoch": 1.3202667236328124e-05,
      "model_forward_time": 5.602962493896484,
      "step": 13844
    },
    {
      "epoch": 1.3202667236328124e-05,
      "step": 13844,
      "training_step_time": 17.089274168014526
    },
    {
      "epoch": 1.3203620910644532e-05,
      "model_forward_time": 5.982588052749634,
      "step": 13845
    },
    {
      "epoch": 1.3203620910644532e-05,
      "step": 13845,
      "training_step_time": 17.49039936065674
    },
    {
      "epoch": 1.3204574584960938e-05,
      "model_forward_time": 5.6048338413238525,
      "step": 13846
    },
    {
      "epoch": 1.3204574584960938e-05,
      "step": 13846,
      "training_step_time": 17.101123809814453
    },
    {
      "epoch": 1.3205528259277343e-05,
      "model_forward_time": 5.6073527336120605,
      "step": 13847
    },
    {
      "epoch": 1.3205528259277343e-05,
      "step": 13847,
      "training_step_time": 17.114482879638672
    },
    {
      "epoch": 1.320648193359375e-05,
      "model_forward_time": 5.620161533355713,
      "step": 13848
    },
    {
      "epoch": 1.320648193359375e-05,
      "step": 13848,
      "training_step_time": 17.097725868225098
    },
    {
      "epoch": 1.3207435607910156e-05,
      "model_forward_time": 5.594923496246338,
      "step": 13849
    },
    {
      "epoch": 1.3207435607910156e-05,
      "step": 13849,
      "training_step_time": 17.011040449142456
    },
    {
      "epoch": 1.3208389282226562e-05,
      "grad_norm": 0.5771730542182922,
      "learning_rate": 1e-05,
      "loss": 0.0776,
      "step": 13850
    },
    {
      "epoch": 1.3208389282226562e-05,
      "model_forward_time": 5.585244417190552,
      "step": 13850
    },
    {
      "dynamics_loss_avg": 0.0704154098406434,
      "epoch": 1.3208389282226562e-05,
      "step": 13850
    },
    {
      "action_loss_avg": 0.009032101533375681,
      "epoch": 1.3208389282226562e-05,
      "step": 13850
    },
    {
      "epoch": 1.3208389282226562e-05,
      "step": 13850,
      "training_step_time": 17.381133794784546
    },
    {
      "epoch": 1.320934295654297e-05,
      "model_forward_time": 5.582081079483032,
      "step": 13851
    },
    {
      "epoch": 1.320934295654297e-05,
      "step": 13851,
      "training_step_time": 17.02692723274231
    },
    {
      "epoch": 1.3210296630859375e-05,
      "model_forward_time": 5.571993350982666,
      "step": 13852
    },
    {
      "epoch": 1.3210296630859375e-05,
      "step": 13852,
      "training_step_time": 16.986257076263428
    },
    {
      "epoch": 1.3211250305175781e-05,
      "model_forward_time": 5.58326268196106,
      "step": 13853
    },
    {
      "epoch": 1.3211250305175781e-05,
      "step": 13853,
      "training_step_time": 17.05198097229004
    },
    {
      "epoch": 1.3212203979492187e-05,
      "model_forward_time": 5.610077619552612,
      "step": 13854
    },
    {
      "epoch": 1.3212203979492187e-05,
      "step": 13854,
      "training_step_time": 17.104538440704346
    },
    {
      "epoch": 1.3213157653808594e-05,
      "model_forward_time": 5.611603736877441,
      "step": 13855
    },
    {
      "epoch": 1.3213157653808594e-05,
      "step": 13855,
      "training_step_time": 17.093138694763184
    },
    {
      "epoch": 1.3214111328125e-05,
      "model_forward_time": 5.972151517868042,
      "step": 13856
    },
    {
      "epoch": 1.3214111328125e-05,
      "step": 13856,
      "training_step_time": 17.425809860229492
    },
    {
      "epoch": 1.3215065002441406e-05,
      "model_forward_time": 5.621248960494995,
      "step": 13857
    },
    {
      "epoch": 1.3215065002441406e-05,
      "step": 13857,
      "training_step_time": 17.102843523025513
    },
    {
      "epoch": 1.3216018676757813e-05,
      "model_forward_time": 5.602016925811768,
      "step": 13858
    },
    {
      "epoch": 1.3216018676757813e-05,
      "step": 13858,
      "training_step_time": 17.110517263412476
    },
    {
      "epoch": 1.3216972351074219e-05,
      "model_forward_time": 5.612914800643921,
      "step": 13859
    },
    {
      "epoch": 1.3216972351074219e-05,
      "step": 13859,
      "training_step_time": 17.05888533592224
    },
    {
      "epoch": 1.3217926025390624e-05,
      "grad_norm": 1.520493984222412,
      "learning_rate": 1e-05,
      "loss": 0.0738,
      "step": 13860
    },
    {
      "epoch": 1.3217926025390624e-05,
      "model_forward_time": 5.6008758544921875,
      "step": 13860
    },
    {
      "dynamics_loss_avg": 0.06490526590496301,
      "epoch": 1.3217926025390624e-05,
      "step": 13860
    },
    {
      "action_loss_avg": 0.006935309059917927,
      "epoch": 1.3217926025390624e-05,
      "step": 13860
    },
    {
      "epoch": 1.3217926025390624e-05,
      "step": 13860,
      "training_step_time": 17.07874035835266
    },
    {
      "epoch": 1.3218879699707032e-05,
      "model_forward_time": 5.599468231201172,
      "step": 13861
    },
    {
      "epoch": 1.3218879699707032e-05,
      "step": 13861,
      "training_step_time": 17.426411390304565
    },
    {
      "epoch": 1.3219833374023438e-05,
      "model_forward_time": 5.5923683643341064,
      "step": 13862
    },
    {
      "epoch": 1.3219833374023438e-05,
      "step": 13862,
      "training_step_time": 17.028435230255127
    },
    {
      "epoch": 1.3220787048339843e-05,
      "model_forward_time": 5.600872039794922,
      "step": 13863
    },
    {
      "epoch": 1.3220787048339843e-05,
      "step": 13863,
      "training_step_time": 17.05751872062683
    },
    {
      "epoch": 1.322174072265625e-05,
      "model_forward_time": 5.60636568069458,
      "step": 13864
    },
    {
      "epoch": 1.322174072265625e-05,
      "step": 13864,
      "training_step_time": 17.08438467979431
    },
    {
      "epoch": 1.3222694396972656e-05,
      "model_forward_time": 5.616374254226685,
      "step": 13865
    },
    {
      "epoch": 1.3222694396972656e-05,
      "step": 13865,
      "training_step_time": 17.118609189987183
    },
    {
      "epoch": 1.3223648071289062e-05,
      "model_forward_time": 5.620956659317017,
      "step": 13866
    },
    {
      "epoch": 1.3223648071289062e-05,
      "step": 13866,
      "training_step_time": 17.140323400497437
    },
    {
      "epoch": 1.322460174560547e-05,
      "model_forward_time": 6.012071371078491,
      "step": 13867
    },
    {
      "epoch": 1.322460174560547e-05,
      "step": 13867,
      "training_step_time": 17.504934072494507
    },
    {
      "epoch": 1.3225555419921875e-05,
      "model_forward_time": 5.604545831680298,
      "step": 13868
    },
    {
      "epoch": 1.3225555419921875e-05,
      "step": 13868,
      "training_step_time": 17.055103063583374
    },
    {
      "epoch": 1.3226509094238281e-05,
      "model_forward_time": 5.582215309143066,
      "step": 13869
    },
    {
      "epoch": 1.3226509094238281e-05,
      "step": 13869,
      "training_step_time": 16.99514889717102
    },
    {
      "epoch": 1.3227462768554687e-05,
      "grad_norm": 0.3089308440685272,
      "learning_rate": 1e-05,
      "loss": 0.0739,
      "step": 13870
    },
    {
      "epoch": 1.3227462768554687e-05,
      "model_forward_time": 5.589593410491943,
      "step": 13870
    },
    {
      "dynamics_loss_avg": 0.06363723687827587,
      "epoch": 1.3227462768554687e-05,
      "step": 13870
    },
    {
      "action_loss_avg": 0.006481827609241009,
      "epoch": 1.3227462768554687e-05,
      "step": 13870
    },
    {
      "epoch": 1.3227462768554687e-05,
      "step": 13870,
      "training_step_time": 17.04237675666809
    },
    {
      "epoch": 1.3228416442871094e-05,
      "model_forward_time": 5.581622362136841,
      "step": 13871
    },
    {
      "epoch": 1.3228416442871094e-05,
      "step": 13871,
      "training_step_time": 16.9908344745636
    },
    {
      "epoch": 1.32293701171875e-05,
      "model_forward_time": 5.936474084854126,
      "step": 13872
    },
    {
      "epoch": 1.32293701171875e-05,
      "step": 13872,
      "training_step_time": 17.380534410476685
    },
    {
      "epoch": 1.3230323791503906e-05,
      "model_forward_time": 5.599891424179077,
      "step": 13873
    },
    {
      "epoch": 1.3230323791503906e-05,
      "step": 13873,
      "training_step_time": 17.079761028289795
    },
    {
      "epoch": 1.3231277465820313e-05,
      "model_forward_time": 5.607175588607788,
      "step": 13874
    },
    {
      "epoch": 1.3231277465820313e-05,
      "step": 13874,
      "training_step_time": 17.09947681427002
    },
    {
      "epoch": 1.3232231140136719e-05,
      "model_forward_time": 5.622641324996948,
      "step": 13875
    },
    {
      "epoch": 1.3232231140136719e-05,
      "step": 13875,
      "training_step_time": 17.146833181381226
    },
    {
      "epoch": 1.3233184814453124e-05,
      "model_forward_time": 5.631751298904419,
      "step": 13876
    },
    {
      "epoch": 1.3233184814453124e-05,
      "step": 13876,
      "training_step_time": 17.140862941741943
    },
    {
      "epoch": 1.3234138488769532e-05,
      "model_forward_time": 5.601696014404297,
      "step": 13877
    },
    {
      "epoch": 1.3234138488769532e-05,
      "step": 13877,
      "training_step_time": 17.036970853805542
    },
    {
      "epoch": 1.3235092163085938e-05,
      "model_forward_time": 5.943989038467407,
      "step": 13878
    },
    {
      "epoch": 1.3235092163085938e-05,
      "step": 13878,
      "training_step_time": 17.396808624267578
    },
    {
      "epoch": 1.3236045837402343e-05,
      "model_forward_time": 5.5699357986450195,
      "step": 13879
    },
    {
      "epoch": 1.3236045837402343e-05,
      "step": 13879,
      "training_step_time": 16.980836629867554
    },
    {
      "epoch": 1.323699951171875e-05,
      "grad_norm": 0.47451546788215637,
      "learning_rate": 1e-05,
      "loss": 0.0664,
      "step": 13880
    },
    {
      "epoch": 1.323699951171875e-05,
      "model_forward_time": 5.574761629104614,
      "step": 13880
    },
    {
      "dynamics_loss_avg": 0.06790307015180588,
      "epoch": 1.323699951171875e-05,
      "step": 13880
    },
    {
      "action_loss_avg": 0.009656789619475603,
      "epoch": 1.323699951171875e-05,
      "step": 13880
    },
    {
      "epoch": 1.323699951171875e-05,
      "step": 13880,
      "training_step_time": 16.985467195510864
    },
    {
      "epoch": 1.3237953186035156e-05,
      "model_forward_time": 5.574082851409912,
      "step": 13881
    },
    {
      "epoch": 1.3237953186035156e-05,
      "step": 13881,
      "training_step_time": 17.012101650238037
    },
    {
      "epoch": 1.3238906860351562e-05,
      "model_forward_time": 5.585058212280273,
      "step": 13882
    },
    {
      "epoch": 1.3238906860351562e-05,
      "step": 13882,
      "training_step_time": 17.031039237976074
    },
    {
      "epoch": 1.3239860534667968e-05,
      "model_forward_time": 5.594959735870361,
      "step": 13883
    },
    {
      "epoch": 1.3239860534667968e-05,
      "step": 13883,
      "training_step_time": 17.443527698516846
    },
    {
      "epoch": 1.3240814208984375e-05,
      "model_forward_time": 5.5947489738464355,
      "step": 13884
    },
    {
      "epoch": 1.3240814208984375e-05,
      "step": 13884,
      "training_step_time": 17.07871103286743
    },
    {
      "epoch": 1.3241767883300781e-05,
      "model_forward_time": 5.623058080673218,
      "step": 13885
    },
    {
      "epoch": 1.3241767883300781e-05,
      "step": 13885,
      "training_step_time": 17.123076915740967
    },
    {
      "epoch": 1.3242721557617187e-05,
      "model_forward_time": 5.61852765083313,
      "step": 13886
    },
    {
      "epoch": 1.3242721557617187e-05,
      "step": 13886,
      "training_step_time": 17.10213041305542
    },
    {
      "epoch": 1.3243675231933594e-05,
      "model_forward_time": 5.589259624481201,
      "step": 13887
    },
    {
      "epoch": 1.3243675231933594e-05,
      "step": 13887,
      "training_step_time": 17.040363550186157
    },
    {
      "epoch": 1.324462890625e-05,
      "model_forward_time": 5.585331439971924,
      "step": 13888
    },
    {
      "epoch": 1.324462890625e-05,
      "step": 13888,
      "training_step_time": 17.01326608657837
    },
    {
      "epoch": 1.3245582580566406e-05,
      "model_forward_time": 5.954786777496338,
      "step": 13889
    },
    {
      "epoch": 1.3245582580566406e-05,
      "step": 13889,
      "training_step_time": 17.393751859664917
    },
    {
      "epoch": 1.3246536254882813e-05,
      "grad_norm": 0.8583061695098877,
      "learning_rate": 1e-05,
      "loss": 0.082,
      "step": 13890
    },
    {
      "epoch": 1.3246536254882813e-05,
      "model_forward_time": 5.5981364250183105,
      "step": 13890
    },
    {
      "dynamics_loss_avg": 0.06625334229320287,
      "epoch": 1.3246536254882813e-05,
      "step": 13890
    },
    {
      "action_loss_avg": 0.007866058847866952,
      "epoch": 1.3246536254882813e-05,
      "step": 13890
    },
    {
      "epoch": 1.3246536254882813e-05,
      "step": 13890,
      "training_step_time": 17.062641620635986
    },
    {
      "epoch": 1.3247489929199219e-05,
      "model_forward_time": 5.6054511070251465,
      "step": 13891
    },
    {
      "epoch": 1.3247489929199219e-05,
      "step": 13891,
      "training_step_time": 17.08462691307068
    },
    {
      "epoch": 1.3248443603515624e-05,
      "model_forward_time": 5.603617191314697,
      "step": 13892
    },
    {
      "epoch": 1.3248443603515624e-05,
      "step": 13892,
      "training_step_time": 17.066585302352905
    },
    {
      "epoch": 1.3249397277832032e-05,
      "model_forward_time": 5.590801954269409,
      "step": 13893
    },
    {
      "epoch": 1.3249397277832032e-05,
      "step": 13893,
      "training_step_time": 17.036550283432007
    },
    {
      "epoch": 1.3250350952148438e-05,
      "model_forward_time": 5.58052659034729,
      "step": 13894
    },
    {
      "epoch": 1.3250350952148438e-05,
      "step": 13894,
      "training_step_time": 17.397862911224365
    },
    {
      "epoch": 1.3251304626464843e-05,
      "model_forward_time": 5.586486101150513,
      "step": 13895
    },
    {
      "epoch": 1.3251304626464843e-05,
      "step": 13895,
      "training_step_time": 17.016652584075928
    },
    {
      "epoch": 1.325225830078125e-05,
      "model_forward_time": 5.601731061935425,
      "step": 13896
    },
    {
      "epoch": 1.325225830078125e-05,
      "step": 13896,
      "training_step_time": 17.09306240081787
    },
    {
      "epoch": 1.3253211975097656e-05,
      "model_forward_time": 5.616492509841919,
      "step": 13897
    },
    {
      "epoch": 1.3253211975097656e-05,
      "step": 13897,
      "training_step_time": 17.142781496047974
    },
    {
      "epoch": 1.3254165649414062e-05,
      "model_forward_time": 5.626177787780762,
      "step": 13898
    },
    {
      "epoch": 1.3254165649414062e-05,
      "step": 13898,
      "training_step_time": 17.14112162590027
    },
    {
      "epoch": 1.3255119323730468e-05,
      "model_forward_time": 5.613924026489258,
      "step": 13899
    },
    {
      "epoch": 1.3255119323730468e-05,
      "step": 13899,
      "training_step_time": 17.077864170074463
    },
    {
      "epoch": 1.3256072998046875e-05,
      "grad_norm": 0.5342065691947937,
      "learning_rate": 1e-05,
      "loss": 0.0843,
      "step": 13900
    },
    {
      "epoch": 1.3256072998046875e-05,
      "model_forward_time": 5.962597370147705,
      "step": 13900
    },
    {
      "dynamics_loss_avg": 0.0721916738897562,
      "epoch": 1.3256072998046875e-05,
      "step": 13900
    },
    {
      "action_loss_avg": 0.008315718104131519,
      "epoch": 1.3256072998046875e-05,
      "step": 13900
    },
    {
      "epoch": 1.3256072998046875e-05,
      "step": 13900,
      "training_step_time": 17.400954961776733
    },
    {
      "epoch": 1.3257026672363281e-05,
      "model_forward_time": 5.587642431259155,
      "step": 13901
    },
    {
      "epoch": 1.3257026672363281e-05,
      "step": 13901,
      "training_step_time": 17.01736879348755
    },
    {
      "epoch": 1.3257980346679687e-05,
      "model_forward_time": 5.572955369949341,
      "step": 13902
    },
    {
      "epoch": 1.3257980346679687e-05,
      "step": 13902,
      "training_step_time": 16.994072198867798
    },
    {
      "epoch": 1.3258934020996094e-05,
      "model_forward_time": 5.583704948425293,
      "step": 13903
    },
    {
      "epoch": 1.3258934020996094e-05,
      "step": 13903,
      "training_step_time": 16.994102001190186
    },
    {
      "epoch": 1.32598876953125e-05,
      "model_forward_time": 5.58838152885437,
      "step": 13904
    },
    {
      "epoch": 1.32598876953125e-05,
      "step": 13904,
      "training_step_time": 17.026890993118286
    },
    {
      "epoch": 1.3260841369628906e-05,
      "model_forward_time": 5.952268362045288,
      "step": 13905
    },
    {
      "epoch": 1.3260841369628906e-05,
      "step": 13905,
      "training_step_time": 17.422653675079346
    },
    {
      "epoch": 1.3261795043945313e-05,
      "model_forward_time": 5.615293979644775,
      "step": 13906
    },
    {
      "epoch": 1.3261795043945313e-05,
      "step": 13906,
      "training_step_time": 17.127086877822876
    },
    {
      "epoch": 1.3262748718261719e-05,
      "model_forward_time": 5.617292404174805,
      "step": 13907
    },
    {
      "epoch": 1.3262748718261719e-05,
      "step": 13907,
      "training_step_time": 17.131795644760132
    },
    {
      "epoch": 1.3263702392578124e-05,
      "model_forward_time": 5.611673355102539,
      "step": 13908
    },
    {
      "epoch": 1.3263702392578124e-05,
      "step": 13908,
      "training_step_time": 17.068332195281982
    },
    {
      "epoch": 1.3264656066894532e-05,
      "model_forward_time": 5.5847063064575195,
      "step": 13909
    },
    {
      "epoch": 1.3264656066894532e-05,
      "step": 13909,
      "training_step_time": 17.006792306900024
    },
    {
      "epoch": 1.3265609741210938e-05,
      "grad_norm": 0.4118805527687073,
      "learning_rate": 1e-05,
      "loss": 0.0479,
      "step": 13910
    },
    {
      "epoch": 1.3265609741210938e-05,
      "model_forward_time": 5.57256293296814,
      "step": 13910
    },
    {
      "dynamics_loss_avg": 0.04539826102554798,
      "epoch": 1.3265609741210938e-05,
      "step": 13910
    },
    {
      "action_loss_avg": 0.0049297184683382515,
      "epoch": 1.3265609741210938e-05,
      "step": 13910
    },
    {
      "epoch": 1.3265609741210938e-05,
      "step": 13910,
      "training_step_time": 16.983007431030273
    },
    {
      "epoch": 1.3266563415527343e-05,
      "model_forward_time": 5.929175138473511,
      "step": 13911
    },
    {
      "epoch": 1.3266563415527343e-05,
      "step": 13911,
      "training_step_time": 17.345909595489502
    },
    {
      "epoch": 1.326751708984375e-05,
      "model_forward_time": 5.598463296890259,
      "step": 13912
    },
    {
      "epoch": 1.326751708984375e-05,
      "step": 13912,
      "training_step_time": 17.04407572746277
    },
    {
      "epoch": 1.3268470764160156e-05,
      "model_forward_time": 5.578662633895874,
      "step": 13913
    },
    {
      "epoch": 1.3268470764160156e-05,
      "step": 13913,
      "training_step_time": 17.005968809127808
    },
    {
      "epoch": 1.3269424438476562e-05,
      "model_forward_time": 5.579220294952393,
      "step": 13914
    },
    {
      "epoch": 1.3269424438476562e-05,
      "step": 13914,
      "training_step_time": 17.017566680908203
    },
    {
      "epoch": 1.3270378112792968e-05,
      "model_forward_time": 5.582676887512207,
      "step": 13915
    },
    {
      "epoch": 1.3270378112792968e-05,
      "step": 13915,
      "training_step_time": 17.023020267486572
    },
    {
      "epoch": 1.3271331787109375e-05,
      "model_forward_time": 5.594985485076904,
      "step": 13916
    },
    {
      "epoch": 1.3271331787109375e-05,
      "step": 13916,
      "training_step_time": 17.421015977859497
    },
    {
      "epoch": 1.3272285461425781e-05,
      "model_forward_time": 5.616566896438599,
      "step": 13917
    },
    {
      "epoch": 1.3272285461425781e-05,
      "step": 13917,
      "training_step_time": 17.13908553123474
    },
    {
      "epoch": 1.3273239135742187e-05,
      "model_forward_time": 5.615568399429321,
      "step": 13918
    },
    {
      "epoch": 1.3273239135742187e-05,
      "step": 13918,
      "training_step_time": 17.10801601409912
    },
    {
      "epoch": 1.3274192810058594e-05,
      "model_forward_time": 5.5929176807403564,
      "step": 13919
    },
    {
      "epoch": 1.3274192810058594e-05,
      "step": 13919,
      "training_step_time": 17.038536548614502
    },
    {
      "epoch": 1.3275146484375e-05,
      "grad_norm": 1.0016999244689941,
      "learning_rate": 1e-05,
      "loss": 0.0774,
      "step": 13920
    },
    {
      "epoch": 1.3275146484375e-05,
      "model_forward_time": 5.583233118057251,
      "step": 13920
    },
    {
      "dynamics_loss_avg": 0.06511751059442758,
      "epoch": 1.3275146484375e-05,
      "step": 13920
    },
    {
      "action_loss_avg": 0.009044344490393996,
      "epoch": 1.3275146484375e-05,
      "step": 13920
    },
    {
      "epoch": 1.3275146484375e-05,
      "step": 13920,
      "training_step_time": 17.0365047454834
    },
    {
      "epoch": 1.3276100158691406e-05,
      "model_forward_time": 5.593600034713745,
      "step": 13921
    },
    {
      "epoch": 1.3276100158691406e-05,
      "step": 13921,
      "training_step_time": 17.061004638671875
    },
    {
      "epoch": 1.3277053833007813e-05,
      "model_forward_time": 5.973301887512207,
      "step": 13922
    },
    {
      "epoch": 1.3277053833007813e-05,
      "step": 13922,
      "training_step_time": 17.45036220550537
    },
    {
      "epoch": 1.3278007507324219e-05,
      "model_forward_time": 5.598979711532593,
      "step": 13923
    },
    {
      "epoch": 1.3278007507324219e-05,
      "step": 13923,
      "training_step_time": 17.062875509262085
    },
    {
      "epoch": 1.3278961181640625e-05,
      "model_forward_time": 5.592008113861084,
      "step": 13924
    },
    {
      "epoch": 1.3278961181640625e-05,
      "step": 13924,
      "training_step_time": 17.053755521774292
    },
    {
      "epoch": 1.3279914855957032e-05,
      "model_forward_time": 5.585564851760864,
      "step": 13925
    },
    {
      "epoch": 1.3279914855957032e-05,
      "step": 13925,
      "training_step_time": 17.044134378433228
    },
    {
      "epoch": 1.3280868530273438e-05,
      "model_forward_time": 5.595060348510742,
      "step": 13926
    },
    {
      "epoch": 1.3280868530273438e-05,
      "step": 13926,
      "training_step_time": 17.045171976089478
    },
    {
      "epoch": 1.3281822204589843e-05,
      "model_forward_time": 5.600348472595215,
      "step": 13927
    },
    {
      "epoch": 1.3281822204589843e-05,
      "step": 13927,
      "training_step_time": 17.47071099281311
    },
    {
      "epoch": 1.328277587890625e-05,
      "model_forward_time": 5.620974063873291,
      "step": 13928
    },
    {
      "epoch": 1.328277587890625e-05,
      "step": 13928,
      "training_step_time": 17.161890268325806
    },
    {
      "epoch": 1.3283729553222657e-05,
      "model_forward_time": 5.634905576705933,
      "step": 13929
    },
    {
      "epoch": 1.3283729553222657e-05,
      "step": 13929,
      "training_step_time": 17.154847145080566
    },
    {
      "epoch": 1.3284683227539062e-05,
      "grad_norm": 0.7223707437515259,
      "learning_rate": 1e-05,
      "loss": 0.0699,
      "step": 13930
    },
    {
      "epoch": 1.3284683227539062e-05,
      "model_forward_time": 5.620126724243164,
      "step": 13930
    },
    {
      "dynamics_loss_avg": 0.06407997459173202,
      "epoch": 1.3284683227539062e-05,
      "step": 13930
    },
    {
      "action_loss_avg": 0.006458584824576974,
      "epoch": 1.3284683227539062e-05,
      "step": 13930
    },
    {
      "epoch": 1.3284683227539062e-05,
      "step": 13930,
      "training_step_time": 17.107213497161865
    },
    {
      "epoch": 1.3285636901855468e-05,
      "model_forward_time": 5.6115171909332275,
      "step": 13931
    },
    {
      "epoch": 1.3285636901855468e-05,
      "step": 13931,
      "training_step_time": 17.08828568458557
    },
    {
      "epoch": 1.3286590576171875e-05,
      "model_forward_time": 5.609925270080566,
      "step": 13932
    },
    {
      "epoch": 1.3286590576171875e-05,
      "step": 13932,
      "training_step_time": 17.076575756072998
    },
    {
      "epoch": 1.3287544250488281e-05,
      "model_forward_time": 5.965641975402832,
      "step": 13933
    },
    {
      "epoch": 1.3287544250488281e-05,
      "step": 13933,
      "training_step_time": 17.435254335403442
    },
    {
      "epoch": 1.3288497924804687e-05,
      "model_forward_time": 5.60648250579834,
      "step": 13934
    },
    {
      "epoch": 1.3288497924804687e-05,
      "step": 13934,
      "training_step_time": 17.087042808532715
    },
    {
      "epoch": 1.3289451599121094e-05,
      "model_forward_time": 5.604733228683472,
      "step": 13935
    },
    {
      "epoch": 1.3289451599121094e-05,
      "step": 13935,
      "training_step_time": 17.06510353088379
    },
    {
      "epoch": 1.32904052734375e-05,
      "model_forward_time": 5.598473072052002,
      "step": 13936
    },
    {
      "epoch": 1.32904052734375e-05,
      "step": 13936,
      "training_step_time": 17.078596591949463
    },
    {
      "epoch": 1.3291358947753906e-05,
      "model_forward_time": 5.6103715896606445,
      "step": 13937
    },
    {
      "epoch": 1.3291358947753906e-05,
      "step": 13937,
      "training_step_time": 17.0872642993927
    },
    {
      "epoch": 1.3292312622070313e-05,
      "model_forward_time": 5.989352464675903,
      "step": 13938
    },
    {
      "epoch": 1.3292312622070313e-05,
      "step": 13938,
      "training_step_time": 17.461549520492554
    },
    {
      "epoch": 1.3293266296386719e-05,
      "model_forward_time": 5.61137318611145,
      "step": 13939
    },
    {
      "epoch": 1.3293266296386719e-05,
      "step": 13939,
      "training_step_time": 17.085803747177124
    },
    {
      "epoch": 1.3294219970703125e-05,
      "grad_norm": 0.6580496430397034,
      "learning_rate": 1e-05,
      "loss": 0.0554,
      "step": 13940
    },
    {
      "epoch": 1.3294219970703125e-05,
      "model_forward_time": 5.605223894119263,
      "step": 13940
    },
    {
      "dynamics_loss_avg": 0.04932236969470978,
      "epoch": 1.3294219970703125e-05,
      "step": 13940
    },
    {
      "action_loss_avg": 0.00684354871045798,
      "epoch": 1.3294219970703125e-05,
      "step": 13940
    },
    {
      "epoch": 1.3294219970703125e-05,
      "step": 13940,
      "training_step_time": 17.073094844818115
    },
    {
      "epoch": 1.3295173645019532e-05,
      "model_forward_time": 5.60125732421875,
      "step": 13941
    },
    {
      "epoch": 1.3295173645019532e-05,
      "step": 13941,
      "training_step_time": 17.060920238494873
    },
    {
      "epoch": 1.3296127319335938e-05,
      "model_forward_time": 5.595096826553345,
      "step": 13942
    },
    {
      "epoch": 1.3296127319335938e-05,
      "step": 13942,
      "training_step_time": 17.063754320144653
    },
    {
      "epoch": 1.3297080993652343e-05,
      "model_forward_time": 5.5861594676971436,
      "step": 13943
    },
    {
      "epoch": 1.3297080993652343e-05,
      "step": 13943,
      "training_step_time": 17.053635120391846
    },
    {
      "epoch": 1.329803466796875e-05,
      "model_forward_time": 5.982763767242432,
      "step": 13944
    },
    {
      "epoch": 1.329803466796875e-05,
      "step": 13944,
      "training_step_time": 17.463069200515747
    },
    {
      "epoch": 1.3298988342285157e-05,
      "model_forward_time": 5.6025590896606445,
      "step": 13945
    },
    {
      "epoch": 1.3298988342285157e-05,
      "step": 13945,
      "training_step_time": 17.09055471420288
    },
    {
      "epoch": 1.3299942016601562e-05,
      "model_forward_time": 5.612546920776367,
      "step": 13946
    },
    {
      "epoch": 1.3299942016601562e-05,
      "step": 13946,
      "training_step_time": 17.056769847869873
    },
    {
      "epoch": 1.3300895690917968e-05,
      "model_forward_time": 5.590641736984253,
      "step": 13947
    },
    {
      "epoch": 1.3300895690917968e-05,
      "step": 13947,
      "training_step_time": 17.052812814712524
    },
    {
      "epoch": 1.3301849365234375e-05,
      "model_forward_time": 5.610659837722778,
      "step": 13948
    },
    {
      "epoch": 1.3301849365234375e-05,
      "step": 13948,
      "training_step_time": 17.103138208389282
    },
    {
      "epoch": 1.3302803039550781e-05,
      "model_forward_time": 5.6176440715789795,
      "step": 13949
    },
    {
      "epoch": 1.3302803039550781e-05,
      "step": 13949,
      "training_step_time": 17.51891040802002
    },
    {
      "epoch": 1.3303756713867187e-05,
      "grad_norm": 1.0333912372589111,
      "learning_rate": 1e-05,
      "loss": 0.0584,
      "step": 13950
    },
    {
      "epoch": 1.3303756713867187e-05,
      "model_forward_time": 5.630143165588379,
      "step": 13950
    },
    {
      "dynamics_loss_avg": 0.05697936899960041,
      "epoch": 1.3303756713867187e-05,
      "step": 13950
    },
    {
      "action_loss_avg": 0.007195375999435783,
      "epoch": 1.3303756713867187e-05,
      "step": 13950
    },
    {
      "epoch": 1.3303756713867187e-05,
      "step": 13950,
      "training_step_time": 17.151454210281372
    },
    {
      "epoch": 1.3304710388183594e-05,
      "model_forward_time": 5.612695693969727,
      "step": 13951
    },
    {
      "epoch": 1.3304710388183594e-05,
      "step": 13951,
      "training_step_time": 17.10883378982544
    },
    {
      "epoch": 1.33056640625e-05,
      "model_forward_time": 5.605486631393433,
      "step": 13952
    },
    {
      "epoch": 1.33056640625e-05,
      "step": 13952,
      "training_step_time": 17.10326385498047
    },
    {
      "epoch": 1.3306617736816406e-05,
      "model_forward_time": 5.6122636795043945,
      "step": 13953
    },
    {
      "epoch": 1.3306617736816406e-05,
      "step": 13953,
      "training_step_time": 17.103960752487183
    },
    {
      "epoch": 1.3307571411132813e-05,
      "model_forward_time": 5.609511137008667,
      "step": 13954
    },
    {
      "epoch": 1.3307571411132813e-05,
      "step": 13954,
      "training_step_time": 17.074262380599976
    },
    {
      "epoch": 1.3308525085449219e-05,
      "model_forward_time": 5.986475229263306,
      "step": 13955
    },
    {
      "epoch": 1.3308525085449219e-05,
      "step": 13955,
      "training_step_time": 17.48546862602234
    },
    {
      "epoch": 1.3309478759765625e-05,
      "model_forward_time": 5.604108810424805,
      "step": 13956
    },
    {
      "epoch": 1.3309478759765625e-05,
      "step": 13956,
      "training_step_time": 17.079525232315063
    },
    {
      "epoch": 1.3310432434082032e-05,
      "model_forward_time": 5.618284702301025,
      "step": 13957
    },
    {
      "epoch": 1.3310432434082032e-05,
      "step": 13957,
      "training_step_time": 17.11423897743225
    },
    {
      "epoch": 1.3311386108398438e-05,
      "model_forward_time": 5.615867376327515,
      "step": 13958
    },
    {
      "epoch": 1.3311386108398438e-05,
      "step": 13958,
      "training_step_time": 17.091233015060425
    },
    {
      "epoch": 1.3312339782714843e-05,
      "model_forward_time": 5.595076084136963,
      "step": 13959
    },
    {
      "epoch": 1.3312339782714843e-05,
      "step": 13959,
      "training_step_time": 17.034806966781616
    },
    {
      "epoch": 1.331329345703125e-05,
      "grad_norm": 0.7584285140037537,
      "learning_rate": 1e-05,
      "loss": 0.0646,
      "step": 13960
    },
    {
      "epoch": 1.331329345703125e-05,
      "model_forward_time": 5.594841718673706,
      "step": 13960
    },
    {
      "dynamics_loss_avg": 0.05488881133496761,
      "epoch": 1.331329345703125e-05,
      "step": 13960
    },
    {
      "action_loss_avg": 0.006794818863272667,
      "epoch": 1.331329345703125e-05,
      "step": 13960
    },
    {
      "epoch": 1.331329345703125e-05,
      "step": 13960,
      "training_step_time": 17.367825984954834
    },
    {
      "epoch": 1.3314247131347657e-05,
      "model_forward_time": 5.5751447677612305,
      "step": 13961
    },
    {
      "epoch": 1.3314247131347657e-05,
      "step": 13961,
      "training_step_time": 17.014976501464844
    },
    {
      "epoch": 1.3315200805664062e-05,
      "model_forward_time": 5.583768129348755,
      "step": 13962
    },
    {
      "epoch": 1.3315200805664062e-05,
      "step": 13962,
      "training_step_time": 17.008766174316406
    },
    {
      "epoch": 1.3316154479980468e-05,
      "model_forward_time": 5.584132671356201,
      "step": 13963
    },
    {
      "epoch": 1.3316154479980468e-05,
      "step": 13963,
      "training_step_time": 17.01646852493286
    },
    {
      "epoch": 1.3317108154296875e-05,
      "model_forward_time": 5.586266040802002,
      "step": 13964
    },
    {
      "epoch": 1.3317108154296875e-05,
      "step": 13964,
      "training_step_time": 17.02414870262146
    },
    {
      "epoch": 1.3318061828613281e-05,
      "model_forward_time": 5.5955445766448975,
      "step": 13965
    },
    {
      "epoch": 1.3318061828613281e-05,
      "step": 13965,
      "training_step_time": 17.04126739501953
    },
    {
      "epoch": 1.3319015502929687e-05,
      "model_forward_time": 6.008753538131714,
      "step": 13966
    },
    {
      "epoch": 1.3319015502929687e-05,
      "step": 13966,
      "training_step_time": 17.495712757110596
    },
    {
      "epoch": 1.3319969177246094e-05,
      "model_forward_time": 5.606921434402466,
      "step": 13967
    },
    {
      "epoch": 1.3319969177246094e-05,
      "step": 13967,
      "training_step_time": 17.094813346862793
    },
    {
      "epoch": 1.33209228515625e-05,
      "model_forward_time": 5.603744983673096,
      "step": 13968
    },
    {
      "epoch": 1.33209228515625e-05,
      "step": 13968,
      "training_step_time": 17.08699917793274
    },
    {
      "epoch": 1.3321876525878906e-05,
      "model_forward_time": 5.611665725708008,
      "step": 13969
    },
    {
      "epoch": 1.3321876525878906e-05,
      "step": 13969,
      "training_step_time": 17.121896505355835
    },
    {
      "epoch": 1.3322830200195313e-05,
      "grad_norm": 0.5902244448661804,
      "learning_rate": 1e-05,
      "loss": 0.0595,
      "step": 13970
    },
    {
      "epoch": 1.3322830200195313e-05,
      "model_forward_time": 5.624584197998047,
      "step": 13970
    },
    {
      "dynamics_loss_avg": 0.050155071169137956,
      "epoch": 1.3322830200195313e-05,
      "step": 13970
    },
    {
      "action_loss_avg": 0.006138121313415468,
      "epoch": 1.3322830200195313e-05,
      "step": 13970
    },
    {
      "epoch": 1.3322830200195313e-05,
      "step": 13970,
      "training_step_time": 17.124425411224365
    },
    {
      "epoch": 1.3323783874511719e-05,
      "model_forward_time": 5.98285984992981,
      "step": 13971
    },
    {
      "epoch": 1.3323783874511719e-05,
      "step": 13971,
      "training_step_time": 17.447101354599
    },
    {
      "epoch": 1.3324737548828125e-05,
      "model_forward_time": 5.594525337219238,
      "step": 13972
    },
    {
      "epoch": 1.3324737548828125e-05,
      "step": 13972,
      "training_step_time": 17.031052589416504
    },
    {
      "epoch": 1.3325691223144532e-05,
      "model_forward_time": 5.585525751113892,
      "step": 13973
    },
    {
      "epoch": 1.3325691223144532e-05,
      "step": 13973,
      "training_step_time": 17.032505750656128
    },
    {
      "epoch": 1.3326644897460938e-05,
      "model_forward_time": 5.584228992462158,
      "step": 13974
    },
    {
      "epoch": 1.3326644897460938e-05,
      "step": 13974,
      "training_step_time": 17.015142917633057
    },
    {
      "epoch": 1.3327598571777343e-05,
      "model_forward_time": 5.606668710708618,
      "step": 13975
    },
    {
      "epoch": 1.3327598571777343e-05,
      "step": 13975,
      "training_step_time": 17.066283226013184
    },
    {
      "epoch": 1.332855224609375e-05,
      "model_forward_time": 5.6007208824157715,
      "step": 13976
    },
    {
      "epoch": 1.332855224609375e-05,
      "step": 13976,
      "training_step_time": 17.046863317489624
    },
    {
      "epoch": 1.3329505920410157e-05,
      "model_forward_time": 5.966068267822266,
      "step": 13977
    },
    {
      "epoch": 1.3329505920410157e-05,
      "step": 13977,
      "training_step_time": 17.410780429840088
    },
    {
      "epoch": 1.3330459594726562e-05,
      "model_forward_time": 5.5969250202178955,
      "step": 13978
    },
    {
      "epoch": 1.3330459594726562e-05,
      "step": 13978,
      "training_step_time": 17.046281814575195
    },
    {
      "epoch": 1.3331413269042968e-05,
      "model_forward_time": 5.599537134170532,
      "step": 13979
    },
    {
      "epoch": 1.3331413269042968e-05,
      "step": 13979,
      "training_step_time": 17.038288354873657
    },
    {
      "epoch": 1.3332366943359375e-05,
      "grad_norm": 0.47299855947494507,
      "learning_rate": 1e-05,
      "loss": 0.0572,
      "step": 13980
    },
    {
      "epoch": 1.3332366943359375e-05,
      "model_forward_time": 5.594244956970215,
      "step": 13980
    },
    {
      "dynamics_loss_avg": 0.056883110851049426,
      "epoch": 1.3332366943359375e-05,
      "step": 13980
    },
    {
      "action_loss_avg": 0.006560442736372352,
      "epoch": 1.3332366943359375e-05,
      "step": 13980
    },
    {
      "epoch": 1.3332366943359375e-05,
      "step": 13980,
      "training_step_time": 17.02854895591736
    },
    {
      "epoch": 1.3333320617675781e-05,
      "model_forward_time": 5.60442328453064,
      "step": 13981
    },
    {
      "epoch": 1.3333320617675781e-05,
      "step": 13981,
      "training_step_time": 17.065460205078125
    },
    {
      "epoch": 1.3334274291992187e-05,
      "model_forward_time": 5.5977747440338135,
      "step": 13982
    },
    {
      "epoch": 1.3334274291992187e-05,
      "step": 13982,
      "training_step_time": 17.41505718231201
    },
    {
      "epoch": 1.3335227966308594e-05,
      "model_forward_time": 5.59324312210083,
      "step": 13983
    },
    {
      "epoch": 1.3335227966308594e-05,
      "step": 13983,
      "training_step_time": 17.035013914108276
    },
    {
      "epoch": 1.3336181640625e-05,
      "model_forward_time": 5.586504936218262,
      "step": 13984
    },
    {
      "epoch": 1.3336181640625e-05,
      "step": 13984,
      "training_step_time": 17.023985624313354
    },
    {
      "epoch": 1.3337135314941406e-05,
      "model_forward_time": 5.588207721710205,
      "step": 13985
    },
    {
      "epoch": 1.3337135314941406e-05,
      "step": 13985,
      "training_step_time": 17.030045986175537
    },
    {
      "epoch": 1.3338088989257813e-05,
      "model_forward_time": 5.584937334060669,
      "step": 13986
    },
    {
      "epoch": 1.3338088989257813e-05,
      "step": 13986,
      "training_step_time": 17.046441078186035
    },
    {
      "epoch": 1.3339042663574219e-05,
      "model_forward_time": 5.600901365280151,
      "step": 13987
    },
    {
      "epoch": 1.3339042663574219e-05,
      "step": 13987,
      "training_step_time": 17.075661420822144
    },
    {
      "epoch": 1.3339996337890625e-05,
      "model_forward_time": 5.9915337562561035,
      "step": 13988
    },
    {
      "epoch": 1.3339996337890625e-05,
      "step": 13988,
      "training_step_time": 17.49399733543396
    },
    {
      "epoch": 1.3340950012207032e-05,
      "model_forward_time": 5.624998569488525,
      "step": 13989
    },
    {
      "epoch": 1.3340950012207032e-05,
      "step": 13989,
      "training_step_time": 17.126754760742188
    },
    {
      "epoch": 1.3341903686523438e-05,
      "grad_norm": 1.415266990661621,
      "learning_rate": 1e-05,
      "loss": 0.0823,
      "step": 13990
    },
    {
      "epoch": 1.3341903686523438e-05,
      "model_forward_time": 5.610281229019165,
      "step": 13990
    },
    {
      "dynamics_loss_avg": 0.06831398643553258,
      "epoch": 1.3341903686523438e-05,
      "step": 13990
    },
    {
      "action_loss_avg": 0.008972615376114846,
      "epoch": 1.3341903686523438e-05,
      "step": 13990
    },
    {
      "epoch": 1.3341903686523438e-05,
      "step": 13990,
      "training_step_time": 17.117356777191162
    },
    {
      "epoch": 1.3342857360839843e-05,
      "model_forward_time": 5.612622499465942,
      "step": 13991
    },
    {
      "epoch": 1.3342857360839843e-05,
      "step": 13991,
      "training_step_time": 17.108835458755493
    },
    {
      "epoch": 1.334381103515625e-05,
      "model_forward_time": 5.615890979766846,
      "step": 13992
    },
    {
      "epoch": 1.334381103515625e-05,
      "step": 13992,
      "training_step_time": 17.09070324897766
    },
    {
      "epoch": 1.3344764709472657e-05,
      "model_forward_time": 5.602733373641968,
      "step": 13993
    },
    {
      "epoch": 1.3344764709472657e-05,
      "step": 13993,
      "training_step_time": 17.43642830848694
    },
    {
      "epoch": 1.3345718383789062e-05,
      "model_forward_time": 5.605410575866699,
      "step": 13994
    },
    {
      "epoch": 1.3345718383789062e-05,
      "step": 13994,
      "training_step_time": 17.09581732749939
    },
    {
      "epoch": 1.3346672058105468e-05,
      "model_forward_time": 5.608718633651733,
      "step": 13995
    },
    {
      "epoch": 1.3346672058105468e-05,
      "step": 13995,
      "training_step_time": 17.094290494918823
    },
    {
      "epoch": 1.3347625732421875e-05,
      "model_forward_time": 5.621348142623901,
      "step": 13996
    },
    {
      "epoch": 1.3347625732421875e-05,
      "step": 13996,
      "training_step_time": 17.109588861465454
    },
    {
      "epoch": 1.3348579406738281e-05,
      "model_forward_time": 5.611103773117065,
      "step": 13997
    },
    {
      "epoch": 1.3348579406738281e-05,
      "step": 13997,
      "training_step_time": 17.10291337966919
    },
    {
      "epoch": 1.3349533081054687e-05,
      "model_forward_time": 5.5936384201049805,
      "step": 13998
    },
    {
      "epoch": 1.3349533081054687e-05,
      "step": 13998,
      "training_step_time": 17.0583655834198
    },
    {
      "epoch": 1.3350486755371094e-05,
      "model_forward_time": 5.961128234863281,
      "step": 13999
    },
    {
      "epoch": 1.3350486755371094e-05,
      "step": 13999,
      "training_step_time": 17.41092586517334
    },
    {
      "epoch": 1.33514404296875e-05,
      "grad_norm": 0.6036731004714966,
      "learning_rate": 1e-05,
      "loss": 0.0654,
      "step": 14000
    },
    {
      "epoch": 1.33514404296875e-05,
      "model_forward_time": 5.635270118713379,
      "step": 14000
    },
    {
      "dynamics_loss_avg": 0.05855939388275146,
      "epoch": 1.33514404296875e-05,
      "step": 14000
    },
    {
      "action_loss_avg": 0.0074906687717884776,
      "epoch": 1.33514404296875e-05,
      "step": 14000
    },
    {
      "epoch": 1.33514404296875e-05,
      "step": 14000,
      "training_step_time": 17.038919925689697
    },
    {
      "epoch": 1.3352394104003906e-05,
      "model_forward_time": 5.630267858505249,
      "step": 14001
    },
    {
      "epoch": 1.3352394104003906e-05,
      "step": 14001,
      "training_step_time": 17.08094573020935
    },
    {
      "epoch": 1.3353347778320313e-05,
      "model_forward_time": 5.640811443328857,
      "step": 14002
    },
    {
      "epoch": 1.3353347778320313e-05,
      "step": 14002,
      "training_step_time": 17.112894773483276
    },
    {
      "epoch": 1.3354301452636719e-05,
      "model_forward_time": 5.6579718589782715,
      "step": 14003
    },
    {
      "epoch": 1.3354301452636719e-05,
      "step": 14003,
      "training_step_time": 17.156712770462036
    },
    {
      "epoch": 1.3355255126953125e-05,
      "model_forward_time": 5.642450332641602,
      "step": 14004
    },
    {
      "epoch": 1.3355255126953125e-05,
      "step": 14004,
      "training_step_time": 17.504183530807495
    },
    {
      "epoch": 1.3356208801269532e-05,
      "model_forward_time": 5.620618104934692,
      "step": 14005
    },
    {
      "epoch": 1.3356208801269532e-05,
      "step": 14005,
      "training_step_time": 17.130632638931274
    },
    {
      "epoch": 1.3357162475585938e-05,
      "model_forward_time": 5.615832090377808,
      "step": 14006
    },
    {
      "epoch": 1.3357162475585938e-05,
      "step": 14006,
      "training_step_time": 17.101289987564087
    },
    {
      "epoch": 1.3358116149902343e-05,
      "model_forward_time": 5.608549118041992,
      "step": 14007
    },
    {
      "epoch": 1.3358116149902343e-05,
      "step": 14007,
      "training_step_time": 17.056368827819824
    },
    {
      "epoch": 1.335906982421875e-05,
      "model_forward_time": 5.6153059005737305,
      "step": 14008
    },
    {
      "epoch": 1.335906982421875e-05,
      "step": 14008,
      "training_step_time": 17.0784649848938
    },
    {
      "epoch": 1.3360023498535157e-05,
      "model_forward_time": 5.593830347061157,
      "step": 14009
    },
    {
      "epoch": 1.3360023498535157e-05,
      "step": 14009,
      "training_step_time": 17.04386067390442
    },
    {
      "epoch": 1.3360977172851562e-05,
      "grad_norm": 0.5211693644523621,
      "learning_rate": 1e-05,
      "loss": 0.0761,
      "step": 14010
    },
    {
      "epoch": 1.3360977172851562e-05,
      "model_forward_time": 5.978561878204346,
      "step": 14010
    },
    {
      "dynamics_loss_avg": 0.06750009059906006,
      "epoch": 1.3360977172851562e-05,
      "step": 14010
    },
    {
      "action_loss_avg": 0.009222737490199506,
      "epoch": 1.3360977172851562e-05,
      "step": 14010
    },
    {
      "epoch": 1.3360977172851562e-05,
      "step": 14010,
      "training_step_time": 17.438195943832397
    },
    {
      "epoch": 1.3361930847167968e-05,
      "model_forward_time": 5.601574420928955,
      "step": 14011
    },
    {
      "epoch": 1.3361930847167968e-05,
      "step": 14011,
      "training_step_time": 17.08568811416626
    },
    {
      "epoch": 1.3362884521484375e-05,
      "model_forward_time": 5.603753328323364,
      "step": 14012
    },
    {
      "epoch": 1.3362884521484375e-05,
      "step": 14012,
      "training_step_time": 17.077982187271118
    },
    {
      "epoch": 1.3363838195800781e-05,
      "model_forward_time": 5.614314317703247,
      "step": 14013
    },
    {
      "epoch": 1.3363838195800781e-05,
      "step": 14013,
      "training_step_time": 17.115973711013794
    },
    {
      "epoch": 1.3364791870117187e-05,
      "model_forward_time": 5.614729642868042,
      "step": 14014
    },
    {
      "epoch": 1.3364791870117187e-05,
      "step": 14014,
      "training_step_time": 17.088658332824707
    },
    {
      "epoch": 1.3365745544433594e-05,
      "model_forward_time": 5.982483863830566,
      "step": 14015
    },
    {
      "epoch": 1.3365745544433594e-05,
      "step": 14015,
      "training_step_time": 17.483649492263794
    },
    {
      "epoch": 1.336669921875e-05,
      "model_forward_time": 5.62328314781189,
      "step": 14016
    },
    {
      "epoch": 1.336669921875e-05,
      "step": 14016,
      "training_step_time": 17.129958391189575
    },
    {
      "epoch": 1.3367652893066406e-05,
      "model_forward_time": 5.620134115219116,
      "step": 14017
    },
    {
      "epoch": 1.3367652893066406e-05,
      "step": 14017,
      "training_step_time": 17.124690771102905
    },
    {
      "epoch": 1.3368606567382813e-05,
      "model_forward_time": 5.618766784667969,
      "step": 14018
    },
    {
      "epoch": 1.3368606567382813e-05,
      "step": 14018,
      "training_step_time": 17.09995675086975
    },
    {
      "epoch": 1.3369560241699219e-05,
      "model_forward_time": 5.616631746292114,
      "step": 14019
    },
    {
      "epoch": 1.3369560241699219e-05,
      "step": 14019,
      "training_step_time": 17.114558696746826
    },
    {
      "epoch": 1.3370513916015625e-05,
      "grad_norm": 0.6986159682273865,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 14020
    },
    {
      "epoch": 1.3370513916015625e-05,
      "model_forward_time": 5.613857746124268,
      "step": 14020
    },
    {
      "dynamics_loss_avg": 0.06434971503913403,
      "epoch": 1.3370513916015625e-05,
      "step": 14020
    },
    {
      "action_loss_avg": 0.006566811329685151,
      "epoch": 1.3370513916015625e-05,
      "step": 14020
    },
    {
      "epoch": 1.3370513916015625e-05,
      "step": 14020,
      "training_step_time": 17.11425256729126
    },
    {
      "epoch": 1.3371467590332032e-05,
      "model_forward_time": 5.988516569137573,
      "step": 14021
    },
    {
      "epoch": 1.3371467590332032e-05,
      "step": 14021,
      "training_step_time": 17.4721839427948
    },
    {
      "epoch": 1.3372421264648438e-05,
      "model_forward_time": 5.605285167694092,
      "step": 14022
    },
    {
      "epoch": 1.3372421264648438e-05,
      "step": 14022,
      "training_step_time": 17.083994388580322
    },
    {
      "epoch": 1.3373374938964843e-05,
      "model_forward_time": 5.602859735488892,
      "step": 14023
    },
    {
      "epoch": 1.3373374938964843e-05,
      "step": 14023,
      "training_step_time": 17.070826292037964
    },
    {
      "epoch": 1.337432861328125e-05,
      "model_forward_time": 5.6067116260528564,
      "step": 14024
    },
    {
      "epoch": 1.337432861328125e-05,
      "step": 14024,
      "training_step_time": 17.086978435516357
    },
    {
      "epoch": 1.3375282287597657e-05,
      "model_forward_time": 5.602899789810181,
      "step": 14025
    },
    {
      "epoch": 1.3375282287597657e-05,
      "step": 14025,
      "training_step_time": 17.074493169784546
    },
    {
      "epoch": 1.3376235961914062e-05,
      "model_forward_time": 5.964226245880127,
      "step": 14026
    },
    {
      "epoch": 1.3376235961914062e-05,
      "step": 14026,
      "training_step_time": 17.427255153656006
    },
    {
      "epoch": 1.3377189636230468e-05,
      "model_forward_time": 5.60141921043396,
      "step": 14027
    },
    {
      "epoch": 1.3377189636230468e-05,
      "step": 14027,
      "training_step_time": 17.08889365196228
    },
    {
      "epoch": 1.3378143310546875e-05,
      "model_forward_time": 5.603623628616333,
      "step": 14028
    },
    {
      "epoch": 1.3378143310546875e-05,
      "step": 14028,
      "training_step_time": 17.1043860912323
    },
    {
      "epoch": 1.3379096984863281e-05,
      "model_forward_time": 5.617789030075073,
      "step": 14029
    },
    {
      "epoch": 1.3379096984863281e-05,
      "step": 14029,
      "training_step_time": 17.122131824493408
    },
    {
      "epoch": 1.3380050659179687e-05,
      "grad_norm": 0.2174735963344574,
      "learning_rate": 1e-05,
      "loss": 0.0764,
      "step": 14030
    },
    {
      "epoch": 1.3380050659179687e-05,
      "model_forward_time": 5.628749370574951,
      "step": 14030
    },
    {
      "dynamics_loss_avg": 0.06718310881406069,
      "epoch": 1.3380050659179687e-05,
      "step": 14030
    },
    {
      "action_loss_avg": 0.007895364821888507,
      "epoch": 1.3380050659179687e-05,
      "step": 14030
    },
    {
      "epoch": 1.3380050659179687e-05,
      "step": 14030,
      "training_step_time": 17.15135622024536
    },
    {
      "epoch": 1.3381004333496094e-05,
      "model_forward_time": 5.625057697296143,
      "step": 14031
    },
    {
      "epoch": 1.3381004333496094e-05,
      "step": 14031,
      "training_step_time": 17.14149761199951
    },
    {
      "epoch": 1.33819580078125e-05,
      "model_forward_time": 5.998157501220703,
      "step": 14032
    },
    {
      "epoch": 1.33819580078125e-05,
      "step": 14032,
      "training_step_time": 17.510342597961426
    },
    {
      "epoch": 1.3382911682128906e-05,
      "model_forward_time": 5.614690542221069,
      "step": 14033
    },
    {
      "epoch": 1.3382911682128906e-05,
      "step": 14033,
      "training_step_time": 17.11253523826599
    },
    {
      "epoch": 1.3383865356445313e-05,
      "model_forward_time": 5.609912633895874,
      "step": 14034
    },
    {
      "epoch": 1.3383865356445313e-05,
      "step": 14034,
      "training_step_time": 17.09969472885132
    },
    {
      "epoch": 1.3384819030761719e-05,
      "model_forward_time": 5.611385822296143,
      "step": 14035
    },
    {
      "epoch": 1.3384819030761719e-05,
      "step": 14035,
      "training_step_time": 17.10927152633667
    },
    {
      "epoch": 1.3385772705078125e-05,
      "model_forward_time": 5.618335008621216,
      "step": 14036
    },
    {
      "epoch": 1.3385772705078125e-05,
      "step": 14036,
      "training_step_time": 17.11041235923767
    },
    {
      "epoch": 1.3386726379394532e-05,
      "model_forward_time": 5.611786365509033,
      "step": 14037
    },
    {
      "epoch": 1.3386726379394532e-05,
      "step": 14037,
      "training_step_time": 17.45064902305603
    },
    {
      "epoch": 1.3387680053710938e-05,
      "model_forward_time": 5.586559534072876,
      "step": 14038
    },
    {
      "epoch": 1.3387680053710938e-05,
      "step": 14038,
      "training_step_time": 17.04038405418396
    },
    {
      "epoch": 1.3388633728027343e-05,
      "model_forward_time": 5.597030162811279,
      "step": 14039
    },
    {
      "epoch": 1.3388633728027343e-05,
      "step": 14039,
      "training_step_time": 17.02775263786316
    },
    {
      "epoch": 1.338958740234375e-05,
      "grad_norm": 1.2313907146453857,
      "learning_rate": 1e-05,
      "loss": 0.0485,
      "step": 14040
    },
    {
      "epoch": 1.338958740234375e-05,
      "model_forward_time": 5.604104995727539,
      "step": 14040
    },
    {
      "dynamics_loss_avg": 0.0461041247472167,
      "epoch": 1.338958740234375e-05,
      "step": 14040
    },
    {
      "action_loss_avg": 0.006970747699961066,
      "epoch": 1.338958740234375e-05,
      "step": 14040
    },
    {
      "epoch": 1.338958740234375e-05,
      "step": 14040,
      "training_step_time": 17.07548761367798
    },
    {
      "epoch": 1.3390541076660157e-05,
      "model_forward_time": 5.616656541824341,
      "step": 14041
    },
    {
      "epoch": 1.3390541076660157e-05,
      "step": 14041,
      "training_step_time": 17.1019766330719
    },
    {
      "epoch": 1.3391494750976562e-05,
      "model_forward_time": 5.617328882217407,
      "step": 14042
    },
    {
      "epoch": 1.3391494750976562e-05,
      "step": 14042,
      "training_step_time": 17.103710651397705
    },
    {
      "epoch": 1.3392448425292968e-05,
      "model_forward_time": 6.010896444320679,
      "step": 14043
    },
    {
      "epoch": 1.3392448425292968e-05,
      "step": 14043,
      "training_step_time": 17.523540258407593
    },
    {
      "epoch": 1.3393402099609375e-05,
      "model_forward_time": 5.636813163757324,
      "step": 14044
    },
    {
      "epoch": 1.3393402099609375e-05,
      "step": 14044,
      "training_step_time": 17.159562349319458
    },
    {
      "epoch": 1.3394355773925781e-05,
      "model_forward_time": 5.626211643218994,
      "step": 14045
    },
    {
      "epoch": 1.3394355773925781e-05,
      "step": 14045,
      "training_step_time": 17.136478900909424
    },
    {
      "epoch": 1.3395309448242187e-05,
      "model_forward_time": 5.63040566444397,
      "step": 14046
    },
    {
      "epoch": 1.3395309448242187e-05,
      "step": 14046,
      "training_step_time": 17.120754957199097
    },
    {
      "epoch": 1.3396263122558594e-05,
      "model_forward_time": 5.6431334018707275,
      "step": 14047
    },
    {
      "epoch": 1.3396263122558594e-05,
      "step": 14047,
      "training_step_time": 17.16920781135559
    },
    {
      "epoch": 1.3397216796875e-05,
      "model_forward_time": 5.983180046081543,
      "step": 14048
    },
    {
      "epoch": 1.3397216796875e-05,
      "step": 14048,
      "training_step_time": 17.470903396606445
    },
    {
      "epoch": 1.3398170471191406e-05,
      "model_forward_time": 5.608022212982178,
      "step": 14049
    },
    {
      "epoch": 1.3398170471191406e-05,
      "step": 14049,
      "training_step_time": 17.07276177406311
    },
    {
      "epoch": 1.3399124145507813e-05,
      "grad_norm": 0.8560112118721008,
      "learning_rate": 1e-05,
      "loss": 0.0689,
      "step": 14050
    },
    {
      "epoch": 1.3399124145507813e-05,
      "model_forward_time": 5.59066915512085,
      "step": 14050
    },
    {
      "dynamics_loss_avg": 0.053323529474437234,
      "epoch": 1.3399124145507813e-05,
      "step": 14050
    },
    {
      "action_loss_avg": 0.00795822071377188,
      "epoch": 1.3399124145507813e-05,
      "step": 14050
    },
    {
      "epoch": 1.3399124145507813e-05,
      "step": 14050,
      "training_step_time": 17.04852557182312
    },
    {
      "epoch": 1.3400077819824219e-05,
      "model_forward_time": 5.590821266174316,
      "step": 14051
    },
    {
      "epoch": 1.3400077819824219e-05,
      "step": 14051,
      "training_step_time": 17.0282883644104
    },
    {
      "epoch": 1.3401031494140625e-05,
      "model_forward_time": 5.595302104949951,
      "step": 14052
    },
    {
      "epoch": 1.3401031494140625e-05,
      "step": 14052,
      "training_step_time": 17.039803504943848
    },
    {
      "epoch": 1.3401985168457032e-05,
      "model_forward_time": 5.606988430023193,
      "step": 14053
    },
    {
      "epoch": 1.3401985168457032e-05,
      "step": 14053,
      "training_step_time": 17.087647676467896
    },
    {
      "epoch": 1.3402938842773438e-05,
      "model_forward_time": 5.984818935394287,
      "step": 14054
    },
    {
      "epoch": 1.3402938842773438e-05,
      "step": 14054,
      "training_step_time": 17.473822832107544
    },
    {
      "epoch": 1.3403892517089843e-05,
      "model_forward_time": 5.6142332553863525,
      "step": 14055
    },
    {
      "epoch": 1.3403892517089843e-05,
      "step": 14055,
      "training_step_time": 17.121451139450073
    },
    {
      "epoch": 1.340484619140625e-05,
      "model_forward_time": 5.634281873703003,
      "step": 14056
    },
    {
      "epoch": 1.340484619140625e-05,
      "step": 14056,
      "training_step_time": 17.12706708908081
    },
    {
      "epoch": 1.3405799865722657e-05,
      "model_forward_time": 5.627250909805298,
      "step": 14057
    },
    {
      "epoch": 1.3405799865722657e-05,
      "step": 14057,
      "training_step_time": 17.135666847229004
    },
    {
      "epoch": 1.3406753540039062e-05,
      "model_forward_time": 5.619267702102661,
      "step": 14058
    },
    {
      "epoch": 1.3406753540039062e-05,
      "step": 14058,
      "training_step_time": 17.09190583229065
    },
    {
      "epoch": 1.3407707214355468e-05,
      "model_forward_time": 5.966628074645996,
      "step": 14059
    },
    {
      "epoch": 1.3407707214355468e-05,
      "step": 14059,
      "training_step_time": 17.392082691192627
    },
    {
      "epoch": 1.3408660888671875e-05,
      "grad_norm": 1.4318541288375854,
      "learning_rate": 1e-05,
      "loss": 0.062,
      "step": 14060
    },
    {
      "epoch": 1.3408660888671875e-05,
      "model_forward_time": 5.574565172195435,
      "step": 14060
    },
    {
      "dynamics_loss_avg": 0.05805689338594675,
      "epoch": 1.3408660888671875e-05,
      "step": 14060
    },
    {
      "action_loss_avg": 0.009273428982123733,
      "epoch": 1.3408660888671875e-05,
      "step": 14060
    },
    {
      "epoch": 1.3408660888671875e-05,
      "step": 14060,
      "training_step_time": 17.01322293281555
    },
    {
      "epoch": 1.3409614562988281e-05,
      "model_forward_time": 5.569904327392578,
      "step": 14061
    },
    {
      "epoch": 1.3409614562988281e-05,
      "step": 14061,
      "training_step_time": 16.990423679351807
    },
    {
      "epoch": 1.3410568237304687e-05,
      "model_forward_time": 5.588587999343872,
      "step": 14062
    },
    {
      "epoch": 1.3410568237304687e-05,
      "step": 14062,
      "training_step_time": 16.998119831085205
    },
    {
      "epoch": 1.3411521911621094e-05,
      "model_forward_time": 5.575881004333496,
      "step": 14063
    },
    {
      "epoch": 1.3411521911621094e-05,
      "step": 14063,
      "training_step_time": 17.00809335708618
    },
    {
      "epoch": 1.34124755859375e-05,
      "model_forward_time": 5.582434177398682,
      "step": 14064
    },
    {
      "epoch": 1.34124755859375e-05,
      "step": 14064,
      "training_step_time": 17.032085418701172
    },
    {
      "epoch": 1.3413429260253906e-05,
      "model_forward_time": 5.96452260017395,
      "step": 14065
    },
    {
      "epoch": 1.3413429260253906e-05,
      "step": 14065,
      "training_step_time": 17.416565418243408
    },
    {
      "epoch": 1.3414382934570313e-05,
      "model_forward_time": 5.599203586578369,
      "step": 14066
    },
    {
      "epoch": 1.3414382934570313e-05,
      "step": 14066,
      "training_step_time": 17.081403255462646
    },
    {
      "epoch": 1.3415336608886719e-05,
      "model_forward_time": 5.607186317443848,
      "step": 14067
    },
    {
      "epoch": 1.3415336608886719e-05,
      "step": 14067,
      "training_step_time": 17.06381893157959
    },
    {
      "epoch": 1.3416290283203125e-05,
      "model_forward_time": 5.602737188339233,
      "step": 14068
    },
    {
      "epoch": 1.3416290283203125e-05,
      "step": 14068,
      "training_step_time": 17.069644927978516
    },
    {
      "epoch": 1.3417243957519532e-05,
      "model_forward_time": 5.599246501922607,
      "step": 14069
    },
    {
      "epoch": 1.3417243957519532e-05,
      "step": 14069,
      "training_step_time": 17.062274932861328
    },
    {
      "epoch": 1.3418197631835938e-05,
      "grad_norm": 0.9982307553291321,
      "learning_rate": 1e-05,
      "loss": 0.0656,
      "step": 14070
    },
    {
      "epoch": 1.3418197631835938e-05,
      "model_forward_time": 5.590442657470703,
      "step": 14070
    },
    {
      "dynamics_loss_avg": 0.05680065415799618,
      "epoch": 1.3418197631835938e-05,
      "step": 14070
    },
    {
      "action_loss_avg": 0.008347591059282422,
      "epoch": 1.3418197631835938e-05,
      "step": 14070
    },
    {
      "epoch": 1.3418197631835938e-05,
      "step": 14070,
      "training_step_time": 17.400773286819458
    },
    {
      "epoch": 1.3419151306152343e-05,
      "model_forward_time": 5.595262765884399,
      "step": 14071
    },
    {
      "epoch": 1.3419151306152343e-05,
      "step": 14071,
      "training_step_time": 17.038772583007812
    },
    {
      "epoch": 1.342010498046875e-05,
      "model_forward_time": 5.589956283569336,
      "step": 14072
    },
    {
      "epoch": 1.342010498046875e-05,
      "step": 14072,
      "training_step_time": 17.08138108253479
    },
    {
      "epoch": 1.3421058654785157e-05,
      "model_forward_time": 5.613213300704956,
      "step": 14073
    },
    {
      "epoch": 1.3421058654785157e-05,
      "step": 14073,
      "training_step_time": 17.12920570373535
    },
    {
      "epoch": 1.3422012329101562e-05,
      "model_forward_time": 5.627797603607178,
      "step": 14074
    },
    {
      "epoch": 1.3422012329101562e-05,
      "step": 14074,
      "training_step_time": 17.13322877883911
    },
    {
      "epoch": 1.3422966003417968e-05,
      "model_forward_time": 5.615638732910156,
      "step": 14075
    },
    {
      "epoch": 1.3422966003417968e-05,
      "step": 14075,
      "training_step_time": 17.084309816360474
    },
    {
      "epoch": 1.3423919677734375e-05,
      "model_forward_time": 5.963005542755127,
      "step": 14076
    },
    {
      "epoch": 1.3423919677734375e-05,
      "step": 14076,
      "training_step_time": 17.397080659866333
    },
    {
      "epoch": 1.3424873352050781e-05,
      "model_forward_time": 5.582215309143066,
      "step": 14077
    },
    {
      "epoch": 1.3424873352050781e-05,
      "step": 14077,
      "training_step_time": 16.990203619003296
    },
    {
      "epoch": 1.3425827026367187e-05,
      "model_forward_time": 5.576004981994629,
      "step": 14078
    },
    {
      "epoch": 1.3425827026367187e-05,
      "step": 14078,
      "training_step_time": 17.000367641448975
    },
    {
      "epoch": 1.3426780700683594e-05,
      "model_forward_time": 5.582317113876343,
      "step": 14079
    },
    {
      "epoch": 1.3426780700683594e-05,
      "step": 14079,
      "training_step_time": 17.031607627868652
    },
    {
      "epoch": 1.3427734375e-05,
      "grad_norm": 0.8936977982521057,
      "learning_rate": 1e-05,
      "loss": 0.0672,
      "step": 14080
    },
    {
      "epoch": 1.3427734375e-05,
      "model_forward_time": 5.588896989822388,
      "step": 14080
    },
    {
      "dynamics_loss_avg": 0.056501269713044165,
      "epoch": 1.3427734375e-05,
      "step": 14080
    },
    {
      "action_loss_avg": 0.007913420093245804,
      "epoch": 1.3427734375e-05,
      "step": 14080
    },
    {
      "epoch": 1.3427734375e-05,
      "step": 14080,
      "training_step_time": 17.0025851726532
    },
    {
      "epoch": 1.3428688049316406e-05,
      "model_forward_time": 5.953551769256592,
      "step": 14081
    },
    {
      "epoch": 1.3428688049316406e-05,
      "step": 14081,
      "training_step_time": 17.383014678955078
    },
    {
      "epoch": 1.3429641723632813e-05,
      "model_forward_time": 5.59617805480957,
      "step": 14082
    },
    {
      "epoch": 1.3429641723632813e-05,
      "step": 14082,
      "training_step_time": 17.032508611679077
    },
    {
      "epoch": 1.3430595397949219e-05,
      "model_forward_time": 5.5904834270477295,
      "step": 14083
    },
    {
      "epoch": 1.3430595397949219e-05,
      "step": 14083,
      "training_step_time": 17.041555881500244
    },
    {
      "epoch": 1.3431549072265625e-05,
      "model_forward_time": 5.59438157081604,
      "step": 14084
    },
    {
      "epoch": 1.3431549072265625e-05,
      "step": 14084,
      "training_step_time": 17.06128215789795
    },
    {
      "epoch": 1.3432502746582032e-05,
      "model_forward_time": 5.599483966827393,
      "step": 14085
    },
    {
      "epoch": 1.3432502746582032e-05,
      "step": 14085,
      "training_step_time": 17.068856954574585
    },
    {
      "epoch": 1.3433456420898438e-05,
      "model_forward_time": 5.608281373977661,
      "step": 14086
    },
    {
      "epoch": 1.3433456420898438e-05,
      "step": 14086,
      "training_step_time": 17.092060565948486
    },
    {
      "epoch": 1.3434410095214844e-05,
      "model_forward_time": 5.960299015045166,
      "step": 14087
    },
    {
      "epoch": 1.3434410095214844e-05,
      "step": 14087,
      "training_step_time": 17.418611764907837
    },
    {
      "epoch": 1.343536376953125e-05,
      "model_forward_time": 5.594818115234375,
      "step": 14088
    },
    {
      "epoch": 1.343536376953125e-05,
      "step": 14088,
      "training_step_time": 17.07141661643982
    },
    {
      "epoch": 1.3436317443847657e-05,
      "model_forward_time": 5.619433641433716,
      "step": 14089
    },
    {
      "epoch": 1.3436317443847657e-05,
      "step": 14089,
      "training_step_time": 17.11465287208557
    },
    {
      "epoch": 1.3437271118164062e-05,
      "grad_norm": 0.8759326934814453,
      "learning_rate": 1e-05,
      "loss": 0.0674,
      "step": 14090
    },
    {
      "epoch": 1.3437271118164062e-05,
      "model_forward_time": 5.623977899551392,
      "step": 14090
    },
    {
      "dynamics_loss_avg": 0.05687385052442551,
      "epoch": 1.3437271118164062e-05,
      "step": 14090
    },
    {
      "action_loss_avg": 0.008006467623636127,
      "epoch": 1.3437271118164062e-05,
      "step": 14090
    },
    {
      "epoch": 1.3437271118164062e-05,
      "step": 14090,
      "training_step_time": 17.119091749191284
    },
    {
      "epoch": 1.3438224792480468e-05,
      "model_forward_time": 5.623752117156982,
      "step": 14091
    },
    {
      "epoch": 1.3438224792480468e-05,
      "step": 14091,
      "training_step_time": 17.12929058074951
    },
    {
      "epoch": 1.3439178466796876e-05,
      "model_forward_time": 5.981405258178711,
      "step": 14092
    },
    {
      "epoch": 1.3439178466796876e-05,
      "step": 14092,
      "training_step_time": 17.498353004455566
    },
    {
      "epoch": 1.3440132141113281e-05,
      "model_forward_time": 5.614667177200317,
      "step": 14093
    },
    {
      "epoch": 1.3440132141113281e-05,
      "step": 14093,
      "training_step_time": 17.112221240997314
    },
    {
      "epoch": 1.3441085815429687e-05,
      "model_forward_time": 5.612099885940552,
      "step": 14094
    },
    {
      "epoch": 1.3441085815429687e-05,
      "step": 14094,
      "training_step_time": 17.098976612091064
    },
    {
      "epoch": 1.3442039489746094e-05,
      "model_forward_time": 5.61665940284729,
      "step": 14095
    },
    {
      "epoch": 1.3442039489746094e-05,
      "step": 14095,
      "training_step_time": 17.116358518600464
    },
    {
      "epoch": 1.34429931640625e-05,
      "model_forward_time": 5.6089887619018555,
      "step": 14096
    },
    {
      "epoch": 1.34429931640625e-05,
      "step": 14096,
      "training_step_time": 17.098690509796143
    },
    {
      "epoch": 1.3443946838378906e-05,
      "model_forward_time": 5.604400157928467,
      "step": 14097
    },
    {
      "epoch": 1.3443946838378906e-05,
      "step": 14097,
      "training_step_time": 17.102110147476196
    },
    {
      "epoch": 1.3444900512695313e-05,
      "model_forward_time": 5.982055187225342,
      "step": 14098
    },
    {
      "epoch": 1.3444900512695313e-05,
      "step": 14098,
      "training_step_time": 17.46087336540222
    },
    {
      "epoch": 1.3445854187011719e-05,
      "model_forward_time": 5.6056296825408936,
      "step": 14099
    },
    {
      "epoch": 1.3445854187011719e-05,
      "step": 14099,
      "training_step_time": 17.090293169021606
    },
    {
      "epoch": 1.3446807861328125e-05,
      "grad_norm": 0.5274976491928101,
      "learning_rate": 1e-05,
      "loss": 0.0565,
      "step": 14100
    },
    {
      "epoch": 1.3446807861328125e-05,
      "model_forward_time": 5.606087923049927,
      "step": 14100
    },
    {
      "dynamics_loss_avg": 0.052361696399748324,
      "epoch": 1.3446807861328125e-05,
      "step": 14100
    },
    {
      "action_loss_avg": 0.005621552839875222,
      "epoch": 1.3446807861328125e-05,
      "step": 14100
    },
    {
      "epoch": 1.3446807861328125e-05,
      "step": 14100,
      "training_step_time": 17.07803773880005
    },
    {
      "epoch": 1.3447761535644532e-05,
      "model_forward_time": 5.599587678909302,
      "step": 14101
    },
    {
      "epoch": 1.3447761535644532e-05,
      "step": 14101,
      "training_step_time": 17.0453200340271
    },
    {
      "epoch": 1.3448715209960938e-05,
      "model_forward_time": 5.595702886581421,
      "step": 14102
    },
    {
      "epoch": 1.3448715209960938e-05,
      "step": 14102,
      "training_step_time": 17.048056840896606
    },
    {
      "epoch": 1.3449668884277344e-05,
      "model_forward_time": 5.5993406772613525,
      "step": 14103
    },
    {
      "epoch": 1.3449668884277344e-05,
      "step": 14103,
      "training_step_time": 17.45463466644287
    },
    {
      "epoch": 1.345062255859375e-05,
      "model_forward_time": 5.608691930770874,
      "step": 14104
    },
    {
      "epoch": 1.345062255859375e-05,
      "step": 14104,
      "training_step_time": 17.115700006484985
    },
    {
      "epoch": 1.3451576232910157e-05,
      "model_forward_time": 5.608154773712158,
      "step": 14105
    },
    {
      "epoch": 1.3451576232910157e-05,
      "step": 14105,
      "training_step_time": 17.072612285614014
    },
    {
      "epoch": 1.3452529907226562e-05,
      "model_forward_time": 5.596961975097656,
      "step": 14106
    },
    {
      "epoch": 1.3452529907226562e-05,
      "step": 14106,
      "training_step_time": 17.03971266746521
    },
    {
      "epoch": 1.3453483581542968e-05,
      "model_forward_time": 5.591269493103027,
      "step": 14107
    },
    {
      "epoch": 1.3453483581542968e-05,
      "step": 14107,
      "training_step_time": 17.02070903778076
    },
    {
      "epoch": 1.3454437255859376e-05,
      "model_forward_time": 5.588470935821533,
      "step": 14108
    },
    {
      "epoch": 1.3454437255859376e-05,
      "step": 14108,
      "training_step_time": 17.038439750671387
    },
    {
      "epoch": 1.3455390930175781e-05,
      "model_forward_time": 5.986067295074463,
      "step": 14109
    },
    {
      "epoch": 1.3455390930175781e-05,
      "step": 14109,
      "training_step_time": 17.429399013519287
    },
    {
      "epoch": 1.3456344604492187e-05,
      "grad_norm": 0.6797956228256226,
      "learning_rate": 1e-05,
      "loss": 0.0578,
      "step": 14110
    },
    {
      "epoch": 1.3456344604492187e-05,
      "model_forward_time": 5.596034288406372,
      "step": 14110
    },
    {
      "dynamics_loss_avg": 0.05138506386429072,
      "epoch": 1.3456344604492187e-05,
      "step": 14110
    },
    {
      "action_loss_avg": 0.006638110749190673,
      "epoch": 1.3456344604492187e-05,
      "step": 14110
    },
    {
      "epoch": 1.3456344604492187e-05,
      "step": 14110,
      "training_step_time": 17.065337657928467
    },
    {
      "epoch": 1.3457298278808594e-05,
      "model_forward_time": 5.606274127960205,
      "step": 14111
    },
    {
      "epoch": 1.3457298278808594e-05,
      "step": 14111,
      "training_step_time": 17.074098587036133
    },
    {
      "epoch": 1.3458251953125e-05,
      "model_forward_time": 5.612209320068359,
      "step": 14112
    },
    {
      "epoch": 1.3458251953125e-05,
      "step": 14112,
      "training_step_time": 17.092275381088257
    },
    {
      "epoch": 1.3459205627441406e-05,
      "model_forward_time": 5.605503797531128,
      "step": 14113
    },
    {
      "epoch": 1.3459205627441406e-05,
      "step": 14113,
      "training_step_time": 17.082712650299072
    },
    {
      "epoch": 1.3460159301757813e-05,
      "model_forward_time": 6.013944149017334,
      "step": 14114
    },
    {
      "epoch": 1.3460159301757813e-05,
      "step": 14114,
      "training_step_time": 17.474337816238403
    },
    {
      "epoch": 1.3461112976074219e-05,
      "model_forward_time": 5.596472978591919,
      "step": 14115
    },
    {
      "epoch": 1.3461112976074219e-05,
      "step": 14115,
      "training_step_time": 17.031764268875122
    },
    {
      "epoch": 1.3462066650390625e-05,
      "model_forward_time": 5.592212677001953,
      "step": 14116
    },
    {
      "epoch": 1.3462066650390625e-05,
      "step": 14116,
      "training_step_time": 17.022559881210327
    },
    {
      "epoch": 1.3463020324707032e-05,
      "model_forward_time": 5.6164186000823975,
      "step": 14117
    },
    {
      "epoch": 1.3463020324707032e-05,
      "step": 14117,
      "training_step_time": 17.080719470977783
    },
    {
      "epoch": 1.3463973999023438e-05,
      "model_forward_time": 5.601756811141968,
      "step": 14118
    },
    {
      "epoch": 1.3463973999023438e-05,
      "step": 14118,
      "training_step_time": 17.082942724227905
    },
    {
      "epoch": 1.3464927673339844e-05,
      "model_forward_time": 5.596453428268433,
      "step": 14119
    },
    {
      "epoch": 1.3464927673339844e-05,
      "step": 14119,
      "training_step_time": 17.08680272102356
    },
    {
      "epoch": 1.346588134765625e-05,
      "grad_norm": 0.9715408682823181,
      "learning_rate": 1e-05,
      "loss": 0.0682,
      "step": 14120
    },
    {
      "epoch": 1.346588134765625e-05,
      "model_forward_time": 6.011262893676758,
      "step": 14120
    },
    {
      "dynamics_loss_avg": 0.060104367509484294,
      "epoch": 1.346588134765625e-05,
      "step": 14120
    },
    {
      "action_loss_avg": 0.008725726278498769,
      "epoch": 1.346588134765625e-05,
      "step": 14120
    },
    {
      "epoch": 1.346588134765625e-05,
      "step": 14120,
      "training_step_time": 17.493117809295654
    },
    {
      "epoch": 1.3466835021972657e-05,
      "model_forward_time": 5.612602472305298,
      "step": 14121
    },
    {
      "epoch": 1.3466835021972657e-05,
      "step": 14121,
      "training_step_time": 17.093498706817627
    },
    {
      "epoch": 1.3467788696289062e-05,
      "model_forward_time": 5.599663019180298,
      "step": 14122
    },
    {
      "epoch": 1.3467788696289062e-05,
      "step": 14122,
      "training_step_time": 17.049299240112305
    },
    {
      "epoch": 1.3468742370605468e-05,
      "model_forward_time": 5.594750881195068,
      "step": 14123
    },
    {
      "epoch": 1.3468742370605468e-05,
      "step": 14123,
      "training_step_time": 17.04663062095642
    },
    {
      "epoch": 1.3469696044921876e-05,
      "model_forward_time": 5.618223667144775,
      "step": 14124
    },
    {
      "epoch": 1.3469696044921876e-05,
      "step": 14124,
      "training_step_time": 17.093912601470947
    },
    {
      "epoch": 1.3470649719238281e-05,
      "model_forward_time": 5.968117952346802,
      "step": 14125
    },
    {
      "epoch": 1.3470649719238281e-05,
      "step": 14125,
      "training_step_time": 17.46278738975525
    },
    {
      "epoch": 1.3471603393554687e-05,
      "model_forward_time": 5.619524002075195,
      "step": 14126
    },
    {
      "epoch": 1.3471603393554687e-05,
      "step": 14126,
      "training_step_time": 17.129907608032227
    },
    {
      "epoch": 1.3472557067871094e-05,
      "model_forward_time": 5.629188776016235,
      "step": 14127
    },
    {
      "epoch": 1.3472557067871094e-05,
      "step": 14127,
      "training_step_time": 17.1299090385437
    },
    {
      "epoch": 1.34735107421875e-05,
      "model_forward_time": 5.6175291538238525,
      "step": 14128
    },
    {
      "epoch": 1.34735107421875e-05,
      "step": 14128,
      "training_step_time": 17.112629175186157
    },
    {
      "epoch": 1.3474464416503906e-05,
      "model_forward_time": 5.637740612030029,
      "step": 14129
    },
    {
      "epoch": 1.3474464416503906e-05,
      "step": 14129,
      "training_step_time": 17.135188102722168
    },
    {
      "epoch": 1.3475418090820313e-05,
      "grad_norm": 0.5569465756416321,
      "learning_rate": 1e-05,
      "loss": 0.0537,
      "step": 14130
    },
    {
      "epoch": 1.3475418090820313e-05,
      "model_forward_time": 5.616580963134766,
      "step": 14130
    },
    {
      "dynamics_loss_avg": 0.05081505216658115,
      "epoch": 1.3475418090820313e-05,
      "step": 14130
    },
    {
      "action_loss_avg": 0.004706024448387325,
      "epoch": 1.3475418090820313e-05,
      "step": 14130
    },
    {
      "epoch": 1.3475418090820313e-05,
      "step": 14130,
      "training_step_time": 17.121505975723267
    },
    {
      "epoch": 1.3476371765136719e-05,
      "model_forward_time": 5.987574100494385,
      "step": 14131
    },
    {
      "epoch": 1.3476371765136719e-05,
      "step": 14131,
      "training_step_time": 17.49741816520691
    },
    {
      "epoch": 1.3477325439453125e-05,
      "model_forward_time": 5.6207029819488525,
      "step": 14132
    },
    {
      "epoch": 1.3477325439453125e-05,
      "step": 14132,
      "training_step_time": 17.129112720489502
    },
    {
      "epoch": 1.347827911376953e-05,
      "model_forward_time": 5.626365900039673,
      "step": 14133
    },
    {
      "epoch": 1.347827911376953e-05,
      "step": 14133,
      "training_step_time": 17.131826877593994
    },
    {
      "epoch": 1.3479232788085938e-05,
      "model_forward_time": 5.61782169342041,
      "step": 14134
    },
    {
      "epoch": 1.3479232788085938e-05,
      "step": 14134,
      "training_step_time": 17.140409231185913
    },
    {
      "epoch": 1.3480186462402344e-05,
      "model_forward_time": 5.613016843795776,
      "step": 14135
    },
    {
      "epoch": 1.3480186462402344e-05,
      "step": 14135,
      "training_step_time": 17.085193872451782
    },
    {
      "epoch": 1.348114013671875e-05,
      "model_forward_time": 5.593207836151123,
      "step": 14136
    },
    {
      "epoch": 1.348114013671875e-05,
      "step": 14136,
      "training_step_time": 17.419456005096436
    },
    {
      "epoch": 1.3482093811035157e-05,
      "model_forward_time": 5.585512638092041,
      "step": 14137
    },
    {
      "epoch": 1.3482093811035157e-05,
      "step": 14137,
      "training_step_time": 16.989111185073853
    },
    {
      "epoch": 1.3483047485351562e-05,
      "model_forward_time": 5.5783469676971436,
      "step": 14138
    },
    {
      "epoch": 1.3483047485351562e-05,
      "step": 14138,
      "training_step_time": 16.997922897338867
    },
    {
      "epoch": 1.3484001159667968e-05,
      "model_forward_time": 5.588161945343018,
      "step": 14139
    },
    {
      "epoch": 1.3484001159667968e-05,
      "step": 14139,
      "training_step_time": 17.016172885894775
    },
    {
      "epoch": 1.3484954833984376e-05,
      "grad_norm": 0.8524659276008606,
      "learning_rate": 1e-05,
      "loss": 0.0815,
      "step": 14140
    },
    {
      "epoch": 1.3484954833984376e-05,
      "model_forward_time": 5.579586744308472,
      "step": 14140
    },
    {
      "dynamics_loss_avg": 0.0717071533203125,
      "epoch": 1.3484954833984376e-05,
      "step": 14140
    },
    {
      "action_loss_avg": 0.009976161550730467,
      "epoch": 1.3484954833984376e-05,
      "step": 14140
    },
    {
      "epoch": 1.3484954833984376e-05,
      "step": 14140,
      "training_step_time": 17.004123210906982
    },
    {
      "epoch": 1.3485908508300781e-05,
      "model_forward_time": 5.586740016937256,
      "step": 14141
    },
    {
      "epoch": 1.3485908508300781e-05,
      "step": 14141,
      "training_step_time": 17.02803111076355
    },
    {
      "epoch": 1.3486862182617187e-05,
      "model_forward_time": 5.976969480514526,
      "step": 14142
    },
    {
      "epoch": 1.3486862182617187e-05,
      "step": 14142,
      "training_step_time": 17.416738510131836
    },
    {
      "epoch": 1.3487815856933594e-05,
      "model_forward_time": 5.598679304122925,
      "step": 14143
    },
    {
      "epoch": 1.3487815856933594e-05,
      "step": 14143,
      "training_step_time": 17.057559728622437
    },
    {
      "epoch": 1.348876953125e-05,
      "model_forward_time": 5.6179938316345215,
      "step": 14144
    },
    {
      "epoch": 1.348876953125e-05,
      "step": 14144,
      "training_step_time": 17.124569177627563
    },
    {
      "epoch": 1.3489723205566406e-05,
      "model_forward_time": 5.624017000198364,
      "step": 14145
    },
    {
      "epoch": 1.3489723205566406e-05,
      "step": 14145,
      "training_step_time": 17.106297969818115
    },
    {
      "epoch": 1.3490676879882813e-05,
      "model_forward_time": 5.5992043018341064,
      "step": 14146
    },
    {
      "epoch": 1.3490676879882813e-05,
      "step": 14146,
      "training_step_time": 17.04489827156067
    },
    {
      "epoch": 1.3491630554199219e-05,
      "model_forward_time": 5.962690353393555,
      "step": 14147
    },
    {
      "epoch": 1.3491630554199219e-05,
      "step": 14147,
      "training_step_time": 17.40717339515686
    },
    {
      "epoch": 1.3492584228515625e-05,
      "model_forward_time": 5.5826592445373535,
      "step": 14148
    },
    {
      "epoch": 1.3492584228515625e-05,
      "step": 14148,
      "training_step_time": 17.02690577507019
    },
    {
      "epoch": 1.349353790283203e-05,
      "model_forward_time": 5.586022615432739,
      "step": 14149
    },
    {
      "epoch": 1.349353790283203e-05,
      "step": 14149,
      "training_step_time": 17.040183544158936
    },
    {
      "epoch": 1.3494491577148438e-05,
      "grad_norm": 0.8420480489730835,
      "learning_rate": 1e-05,
      "loss": 0.0671,
      "step": 14150
    },
    {
      "epoch": 1.3494491577148438e-05,
      "model_forward_time": 5.600008964538574,
      "step": 14150
    },
    {
      "dynamics_loss_avg": 0.060954540595412256,
      "epoch": 1.3494491577148438e-05,
      "step": 14150
    },
    {
      "action_loss_avg": 0.010911301383748651,
      "epoch": 1.3494491577148438e-05,
      "step": 14150
    },
    {
      "epoch": 1.3494491577148438e-05,
      "step": 14150,
      "training_step_time": 17.071605443954468
    },
    {
      "epoch": 1.3495445251464844e-05,
      "model_forward_time": 5.611102819442749,
      "step": 14151
    },
    {
      "epoch": 1.3495445251464844e-05,
      "step": 14151,
      "training_step_time": 17.088910341262817
    },
    {
      "epoch": 1.349639892578125e-05,
      "model_forward_time": 5.609241247177124,
      "step": 14152
    },
    {
      "epoch": 1.349639892578125e-05,
      "step": 14152,
      "training_step_time": 17.070823669433594
    },
    {
      "epoch": 1.3497352600097657e-05,
      "model_forward_time": 5.966094493865967,
      "step": 14153
    },
    {
      "epoch": 1.3497352600097657e-05,
      "step": 14153,
      "training_step_time": 17.39861273765564
    },
    {
      "epoch": 1.3498306274414062e-05,
      "model_forward_time": 5.593645095825195,
      "step": 14154
    },
    {
      "epoch": 1.3498306274414062e-05,
      "step": 14154,
      "training_step_time": 17.019890308380127
    },
    {
      "epoch": 1.3499259948730468e-05,
      "model_forward_time": 5.584401607513428,
      "step": 14155
    },
    {
      "epoch": 1.3499259948730468e-05,
      "step": 14155,
      "training_step_time": 17.028858423233032
    },
    {
      "epoch": 1.3500213623046876e-05,
      "model_forward_time": 5.589657783508301,
      "step": 14156
    },
    {
      "epoch": 1.3500213623046876e-05,
      "step": 14156,
      "training_step_time": 17.058662176132202
    },
    {
      "epoch": 1.3501167297363281e-05,
      "model_forward_time": 5.596523284912109,
      "step": 14157
    },
    {
      "epoch": 1.3501167297363281e-05,
      "step": 14157,
      "training_step_time": 17.090309143066406
    },
    {
      "epoch": 1.3502120971679687e-05,
      "model_forward_time": 5.980631589889526,
      "step": 14158
    },
    {
      "epoch": 1.3502120971679687e-05,
      "step": 14158,
      "training_step_time": 17.493648767471313
    },
    {
      "epoch": 1.3503074645996094e-05,
      "model_forward_time": 5.627233028411865,
      "step": 14159
    },
    {
      "epoch": 1.3503074645996094e-05,
      "step": 14159,
      "training_step_time": 17.13285493850708
    },
    {
      "epoch": 1.35040283203125e-05,
      "grad_norm": 0.32239866256713867,
      "learning_rate": 1e-05,
      "loss": 0.0594,
      "step": 14160
    },
    {
      "epoch": 1.35040283203125e-05,
      "model_forward_time": 5.614047527313232,
      "step": 14160
    },
    {
      "dynamics_loss_avg": 0.04552200585603714,
      "epoch": 1.35040283203125e-05,
      "step": 14160
    },
    {
      "action_loss_avg": 0.005142315686680376,
      "epoch": 1.35040283203125e-05,
      "step": 14160
    },
    {
      "epoch": 1.35040283203125e-05,
      "step": 14160,
      "training_step_time": 17.103753328323364
    },
    {
      "epoch": 1.3504981994628906e-05,
      "model_forward_time": 5.61522364616394,
      "step": 14161
    },
    {
      "epoch": 1.3504981994628906e-05,
      "step": 14161,
      "training_step_time": 17.09663677215576
    },
    {
      "epoch": 1.3505935668945313e-05,
      "model_forward_time": 5.603608131408691,
      "step": 14162
    },
    {
      "epoch": 1.3505935668945313e-05,
      "step": 14162,
      "training_step_time": 17.078629970550537
    },
    {
      "epoch": 1.3506889343261719e-05,
      "model_forward_time": 5.598254919052124,
      "step": 14163
    },
    {
      "epoch": 1.3506889343261719e-05,
      "step": 14163,
      "training_step_time": 17.061782598495483
    },
    {
      "epoch": 1.3507843017578125e-05,
      "model_forward_time": 5.971556901931763,
      "step": 14164
    },
    {
      "epoch": 1.3507843017578125e-05,
      "step": 14164,
      "training_step_time": 17.433278799057007
    },
    {
      "epoch": 1.350879669189453e-05,
      "model_forward_time": 5.62433934211731,
      "step": 14165
    },
    {
      "epoch": 1.350879669189453e-05,
      "step": 14165,
      "training_step_time": 17.098881483078003
    },
    {
      "epoch": 1.3509750366210938e-05,
      "model_forward_time": 5.592495918273926,
      "step": 14166
    },
    {
      "epoch": 1.3509750366210938e-05,
      "step": 14166,
      "training_step_time": 17.042885780334473
    },
    {
      "epoch": 1.3510704040527344e-05,
      "model_forward_time": 5.6037468910217285,
      "step": 14167
    },
    {
      "epoch": 1.3510704040527344e-05,
      "step": 14167,
      "training_step_time": 17.097883224487305
    },
    {
      "epoch": 1.351165771484375e-05,
      "model_forward_time": 5.609505653381348,
      "step": 14168
    },
    {
      "epoch": 1.351165771484375e-05,
      "step": 14168,
      "training_step_time": 17.11310863494873
    },
    {
      "epoch": 1.3512611389160157e-05,
      "model_forward_time": 5.6102166175842285,
      "step": 14169
    },
    {
      "epoch": 1.3512611389160157e-05,
      "step": 14169,
      "training_step_time": 17.461078643798828
    },
    {
      "epoch": 1.3513565063476562e-05,
      "grad_norm": 1.099137783050537,
      "learning_rate": 1e-05,
      "loss": 0.0513,
      "step": 14170
    },
    {
      "epoch": 1.3513565063476562e-05,
      "model_forward_time": 5.611146450042725,
      "step": 14170
    },
    {
      "dynamics_loss_avg": 0.045704581588506696,
      "epoch": 1.3513565063476562e-05,
      "step": 14170
    },
    {
      "action_loss_avg": 0.005010526976548135,
      "epoch": 1.3513565063476562e-05,
      "step": 14170
    },
    {
      "epoch": 1.3513565063476562e-05,
      "step": 14170,
      "training_step_time": 17.097328424453735
    },
    {
      "epoch": 1.3514518737792968e-05,
      "model_forward_time": 5.604232311248779,
      "step": 14171
    },
    {
      "epoch": 1.3514518737792968e-05,
      "step": 14171,
      "training_step_time": 17.079861879348755
    },
    {
      "epoch": 1.3515472412109376e-05,
      "model_forward_time": 5.593551874160767,
      "step": 14172
    },
    {
      "epoch": 1.3515472412109376e-05,
      "step": 14172,
      "training_step_time": 17.064085006713867
    },
    {
      "epoch": 1.3516426086425781e-05,
      "model_forward_time": 5.604577541351318,
      "step": 14173
    },
    {
      "epoch": 1.3516426086425781e-05,
      "step": 14173,
      "training_step_time": 17.078410625457764
    },
    {
      "epoch": 1.3517379760742187e-05,
      "model_forward_time": 5.621359586715698,
      "step": 14174
    },
    {
      "epoch": 1.3517379760742187e-05,
      "step": 14174,
      "training_step_time": 17.11064338684082
    },
    {
      "epoch": 1.3518333435058594e-05,
      "model_forward_time": 5.996309757232666,
      "step": 14175
    },
    {
      "epoch": 1.3518333435058594e-05,
      "step": 14175,
      "training_step_time": 17.50492000579834
    },
    {
      "epoch": 1.3519287109375e-05,
      "model_forward_time": 5.622600793838501,
      "step": 14176
    },
    {
      "epoch": 1.3519287109375e-05,
      "step": 14176,
      "training_step_time": 17.115264892578125
    },
    {
      "epoch": 1.3520240783691406e-05,
      "model_forward_time": 5.6040871143341064,
      "step": 14177
    },
    {
      "epoch": 1.3520240783691406e-05,
      "step": 14177,
      "training_step_time": 17.06523633003235
    },
    {
      "epoch": 1.3521194458007813e-05,
      "model_forward_time": 5.5905921459198,
      "step": 14178
    },
    {
      "epoch": 1.3521194458007813e-05,
      "step": 14178,
      "training_step_time": 17.028335571289062
    },
    {
      "epoch": 1.3522148132324219e-05,
      "model_forward_time": 5.587554931640625,
      "step": 14179
    },
    {
      "epoch": 1.3522148132324219e-05,
      "step": 14179,
      "training_step_time": 17.030219316482544
    },
    {
      "epoch": 1.3523101806640625e-05,
      "grad_norm": 0.7674040198326111,
      "learning_rate": 1e-05,
      "loss": 0.0617,
      "step": 14180
    },
    {
      "epoch": 1.3523101806640625e-05,
      "model_forward_time": 5.9593870639801025,
      "step": 14180
    },
    {
      "dynamics_loss_avg": 0.059159682132303715,
      "epoch": 1.3523101806640625e-05,
      "step": 14180
    },
    {
      "action_loss_avg": 0.007528574741445482,
      "epoch": 1.3523101806640625e-05,
      "step": 14180
    },
    {
      "epoch": 1.3523101806640625e-05,
      "step": 14180,
      "training_step_time": 17.413580417633057
    },
    {
      "epoch": 1.352405548095703e-05,
      "model_forward_time": 5.598956823348999,
      "step": 14181
    },
    {
      "epoch": 1.352405548095703e-05,
      "step": 14181,
      "training_step_time": 17.08878993988037
    },
    {
      "epoch": 1.3525009155273438e-05,
      "model_forward_time": 5.625232696533203,
      "step": 14182
    },
    {
      "epoch": 1.3525009155273438e-05,
      "step": 14182,
      "training_step_time": 17.147747039794922
    },
    {
      "epoch": 1.3525962829589844e-05,
      "model_forward_time": 5.6311023235321045,
      "step": 14183
    },
    {
      "epoch": 1.3525962829589844e-05,
      "step": 14183,
      "training_step_time": 17.15340495109558
    },
    {
      "epoch": 1.352691650390625e-05,
      "model_forward_time": 5.623562574386597,
      "step": 14184
    },
    {
      "epoch": 1.352691650390625e-05,
      "step": 14184,
      "training_step_time": 17.103518962860107
    },
    {
      "epoch": 1.3527870178222657e-05,
      "model_forward_time": 5.594238758087158,
      "step": 14185
    },
    {
      "epoch": 1.3527870178222657e-05,
      "step": 14185,
      "training_step_time": 17.03755521774292
    },
    {
      "epoch": 1.3528823852539062e-05,
      "model_forward_time": 5.967733144760132,
      "step": 14186
    },
    {
      "epoch": 1.3528823852539062e-05,
      "step": 14186,
      "training_step_time": 17.387407541275024
    },
    {
      "epoch": 1.3529777526855468e-05,
      "model_forward_time": 5.585041761398315,
      "step": 14187
    },
    {
      "epoch": 1.3529777526855468e-05,
      "step": 14187,
      "training_step_time": 17.009481191635132
    },
    {
      "epoch": 1.3530731201171876e-05,
      "model_forward_time": 5.5774078369140625,
      "step": 14188
    },
    {
      "epoch": 1.3530731201171876e-05,
      "step": 14188,
      "training_step_time": 17.01628828048706
    },
    {
      "epoch": 1.3531684875488281e-05,
      "model_forward_time": 5.590984344482422,
      "step": 14189
    },
    {
      "epoch": 1.3531684875488281e-05,
      "step": 14189,
      "training_step_time": 17.04954957962036
    },
    {
      "epoch": 1.3532638549804687e-05,
      "grad_norm": 0.6951048970222473,
      "learning_rate": 1e-05,
      "loss": 0.0586,
      "step": 14190
    },
    {
      "epoch": 1.3532638549804687e-05,
      "model_forward_time": 5.5867063999176025,
      "step": 14190
    },
    {
      "dynamics_loss_avg": 0.04911974761635065,
      "epoch": 1.3532638549804687e-05,
      "step": 14190
    },
    {
      "action_loss_avg": 0.0059014720842242244,
      "epoch": 1.3532638549804687e-05,
      "step": 14190
    },
    {
      "epoch": 1.3532638549804687e-05,
      "step": 14190,
      "training_step_time": 17.023796319961548
    },
    {
      "epoch": 1.3533592224121094e-05,
      "model_forward_time": 5.951341867446899,
      "step": 14191
    },
    {
      "epoch": 1.3533592224121094e-05,
      "step": 14191,
      "training_step_time": 17.41189217567444
    },
    {
      "epoch": 1.35345458984375e-05,
      "model_forward_time": 5.595093727111816,
      "step": 14192
    },
    {
      "epoch": 1.35345458984375e-05,
      "step": 14192,
      "training_step_time": 17.07362985610962
    },
    {
      "epoch": 1.3535499572753906e-05,
      "model_forward_time": 5.604507207870483,
      "step": 14193
    },
    {
      "epoch": 1.3535499572753906e-05,
      "step": 14193,
      "training_step_time": 17.092856645584106
    },
    {
      "epoch": 1.3536453247070313e-05,
      "model_forward_time": 5.611492872238159,
      "step": 14194
    },
    {
      "epoch": 1.3536453247070313e-05,
      "step": 14194,
      "training_step_time": 17.11409831047058
    },
    {
      "epoch": 1.3537406921386719e-05,
      "model_forward_time": 5.61850118637085,
      "step": 14195
    },
    {
      "epoch": 1.3537406921386719e-05,
      "step": 14195,
      "training_step_time": 17.11880326271057
    },
    {
      "epoch": 1.3538360595703125e-05,
      "model_forward_time": 5.62192177772522,
      "step": 14196
    },
    {
      "epoch": 1.3538360595703125e-05,
      "step": 14196,
      "training_step_time": 17.114577531814575
    },
    {
      "epoch": 1.353931427001953e-05,
      "model_forward_time": 5.990568399429321,
      "step": 14197
    },
    {
      "epoch": 1.353931427001953e-05,
      "step": 14197,
      "training_step_time": 17.470182180404663
    },
    {
      "epoch": 1.3540267944335938e-05,
      "model_forward_time": 5.614399671554565,
      "step": 14198
    },
    {
      "epoch": 1.3540267944335938e-05,
      "step": 14198,
      "training_step_time": 17.10994529724121
    },
    {
      "epoch": 1.3541221618652344e-05,
      "model_forward_time": 5.615540266036987,
      "step": 14199
    },
    {
      "epoch": 1.3541221618652344e-05,
      "step": 14199,
      "training_step_time": 17.094979763031006
    },
    {
      "epoch": 1.354217529296875e-05,
      "grad_norm": 0.6666563153266907,
      "learning_rate": 1e-05,
      "loss": 0.0588,
      "step": 14200
    },
    {
      "epoch": 1.354217529296875e-05,
      "model_forward_time": 5.607344388961792,
      "step": 14200
    },
    {
      "dynamics_loss_avg": 0.0570441858842969,
      "epoch": 1.354217529296875e-05,
      "step": 14200
    },
    {
      "action_loss_avg": 0.007245987839996815,
      "epoch": 1.354217529296875e-05,
      "step": 14200
    },
    {
      "epoch": 1.354217529296875e-05,
      "step": 14200,
      "training_step_time": 17.09180760383606
    },
    {
      "epoch": 1.3543128967285157e-05,
      "model_forward_time": 5.617696762084961,
      "step": 14201
    },
    {
      "epoch": 1.3543128967285157e-05,
      "step": 14201,
      "training_step_time": 17.104923009872437
    },
    {
      "epoch": 1.3544082641601562e-05,
      "model_forward_time": 5.612368822097778,
      "step": 14202
    },
    {
      "epoch": 1.3544082641601562e-05,
      "step": 14202,
      "training_step_time": 17.483134746551514
    },
    {
      "epoch": 1.3545036315917968e-05,
      "model_forward_time": 5.61271858215332,
      "step": 14203
    },
    {
      "epoch": 1.3545036315917968e-05,
      "step": 14203,
      "training_step_time": 17.119168758392334
    },
    {
      "epoch": 1.3545989990234376e-05,
      "model_forward_time": 5.6154704093933105,
      "step": 14204
    },
    {
      "epoch": 1.3545989990234376e-05,
      "step": 14204,
      "training_step_time": 17.123477458953857
    },
    {
      "epoch": 1.3546943664550781e-05,
      "model_forward_time": 5.610256195068359,
      "step": 14205
    },
    {
      "epoch": 1.3546943664550781e-05,
      "step": 14205,
      "training_step_time": 17.07981562614441
    },
    {
      "epoch": 1.3547897338867187e-05,
      "model_forward_time": 5.58699893951416,
      "step": 14206
    },
    {
      "epoch": 1.3547897338867187e-05,
      "step": 14206,
      "training_step_time": 16.999210357666016
    },
    {
      "epoch": 1.3548851013183594e-05,
      "model_forward_time": 5.582557201385498,
      "step": 14207
    },
    {
      "epoch": 1.3548851013183594e-05,
      "step": 14207,
      "training_step_time": 16.99978518486023
    },
    {
      "epoch": 1.35498046875e-05,
      "model_forward_time": 5.971746921539307,
      "step": 14208
    },
    {
      "epoch": 1.35498046875e-05,
      "step": 14208,
      "training_step_time": 17.417962312698364
    },
    {
      "epoch": 1.3550758361816406e-05,
      "model_forward_time": 5.5878987312316895,
      "step": 14209
    },
    {
      "epoch": 1.3550758361816406e-05,
      "step": 14209,
      "training_step_time": 17.008060216903687
    },
    {
      "epoch": 1.3551712036132813e-05,
      "grad_norm": 0.9402945637702942,
      "learning_rate": 1e-05,
      "loss": 0.0762,
      "step": 14210
    },
    {
      "epoch": 1.3551712036132813e-05,
      "model_forward_time": 5.5895609855651855,
      "step": 14210
    },
    {
      "dynamics_loss_avg": 0.06290588416159153,
      "epoch": 1.3551712036132813e-05,
      "step": 14210
    },
    {
      "action_loss_avg": 0.008730873512104154,
      "epoch": 1.3551712036132813e-05,
      "step": 14210
    },
    {
      "epoch": 1.3551712036132813e-05,
      "step": 14210,
      "training_step_time": 17.013619661331177
    },
    {
      "epoch": 1.3552665710449219e-05,
      "model_forward_time": 5.593779802322388,
      "step": 14211
    },
    {
      "epoch": 1.3552665710449219e-05,
      "step": 14211,
      "training_step_time": 17.03521490097046
    },
    {
      "epoch": 1.3553619384765625e-05,
      "model_forward_time": 5.5948309898376465,
      "step": 14212
    },
    {
      "epoch": 1.3553619384765625e-05,
      "step": 14212,
      "training_step_time": 17.04050612449646
    },
    {
      "epoch": 1.355457305908203e-05,
      "model_forward_time": 5.953080654144287,
      "step": 14213
    },
    {
      "epoch": 1.355457305908203e-05,
      "step": 14213,
      "training_step_time": 17.4278302192688
    },
    {
      "epoch": 1.3555526733398438e-05,
      "model_forward_time": 5.626272201538086,
      "step": 14214
    },
    {
      "epoch": 1.3555526733398438e-05,
      "step": 14214,
      "training_step_time": 17.14350390434265
    },
    {
      "epoch": 1.3556480407714844e-05,
      "model_forward_time": 5.640331029891968,
      "step": 14215
    },
    {
      "epoch": 1.3556480407714844e-05,
      "step": 14215,
      "training_step_time": 17.16915798187256
    },
    {
      "epoch": 1.355743408203125e-05,
      "model_forward_time": 5.631408929824829,
      "step": 14216
    },
    {
      "epoch": 1.355743408203125e-05,
      "step": 14216,
      "training_step_time": 17.124110221862793
    },
    {
      "epoch": 1.3558387756347657e-05,
      "model_forward_time": 5.60887885093689,
      "step": 14217
    },
    {
      "epoch": 1.3558387756347657e-05,
      "step": 14217,
      "training_step_time": 17.08900475502014
    },
    {
      "epoch": 1.3559341430664062e-05,
      "model_forward_time": 5.612202882766724,
      "step": 14218
    },
    {
      "epoch": 1.3559341430664062e-05,
      "step": 14218,
      "training_step_time": 17.096803188323975
    },
    {
      "epoch": 1.3560295104980468e-05,
      "model_forward_time": 5.97922420501709,
      "step": 14219
    },
    {
      "epoch": 1.3560295104980468e-05,
      "step": 14219,
      "training_step_time": 17.436798572540283
    },
    {
      "epoch": 1.3561248779296876e-05,
      "grad_norm": 0.8423939943313599,
      "learning_rate": 1e-05,
      "loss": 0.0754,
      "step": 14220
    },
    {
      "epoch": 1.3561248779296876e-05,
      "model_forward_time": 5.587184429168701,
      "step": 14220
    },
    {
      "dynamics_loss_avg": 0.06401004139333963,
      "epoch": 1.3561248779296876e-05,
      "step": 14220
    },
    {
      "action_loss_avg": 0.008692856994457543,
      "epoch": 1.3561248779296876e-05,
      "step": 14220
    },
    {
      "epoch": 1.3561248779296876e-05,
      "step": 14220,
      "training_step_time": 17.041057586669922
    },
    {
      "epoch": 1.3562202453613281e-05,
      "model_forward_time": 5.598305702209473,
      "step": 14221
    },
    {
      "epoch": 1.3562202453613281e-05,
      "step": 14221,
      "training_step_time": 17.04434061050415
    },
    {
      "epoch": 1.3563156127929687e-05,
      "model_forward_time": 5.598430395126343,
      "step": 14222
    },
    {
      "epoch": 1.3563156127929687e-05,
      "step": 14222,
      "training_step_time": 17.058570861816406
    },
    {
      "epoch": 1.3564109802246094e-05,
      "model_forward_time": 5.606760740280151,
      "step": 14223
    },
    {
      "epoch": 1.3564109802246094e-05,
      "step": 14223,
      "training_step_time": 17.073861360549927
    },
    {
      "epoch": 1.35650634765625e-05,
      "model_forward_time": 5.977985143661499,
      "step": 14224
    },
    {
      "epoch": 1.35650634765625e-05,
      "step": 14224,
      "training_step_time": 17.43902564048767
    },
    {
      "epoch": 1.3566017150878906e-05,
      "model_forward_time": 5.599951505661011,
      "step": 14225
    },
    {
      "epoch": 1.3566017150878906e-05,
      "step": 14225,
      "training_step_time": 17.084226608276367
    },
    {
      "epoch": 1.3566970825195313e-05,
      "model_forward_time": 5.621757984161377,
      "step": 14226
    },
    {
      "epoch": 1.3566970825195313e-05,
      "step": 14226,
      "training_step_time": 17.132014751434326
    },
    {
      "epoch": 1.3567924499511719e-05,
      "model_forward_time": 5.618137836456299,
      "step": 14227
    },
    {
      "epoch": 1.3567924499511719e-05,
      "step": 14227,
      "training_step_time": 17.12329411506653
    },
    {
      "epoch": 1.3568878173828125e-05,
      "model_forward_time": 5.615586996078491,
      "step": 14228
    },
    {
      "epoch": 1.3568878173828125e-05,
      "step": 14228,
      "training_step_time": 17.103017568588257
    },
    {
      "epoch": 1.356983184814453e-05,
      "model_forward_time": 4.309097051620483,
      "step": 14229
    },
    {
      "epoch": 1.356983184814453e-05,
      "step": 14229,
      "training_step_time": 12.85636019706726
    },
    {
      "epoch": 1.3570785522460938e-05,
      "grad_norm": 1.9262701272964478,
      "learning_rate": 1e-05,
      "loss": 0.0537,
      "step": 14230
    },
    {
      "epoch": 1.3570785522460938e-05,
      "model_forward_time": 5.980027198791504,
      "step": 14230
    },
    {
      "dynamics_loss_avg": 0.04989960826933384,
      "epoch": 1.3570785522460938e-05,
      "step": 14230
    },
    {
      "action_loss_avg": 0.0076127864187583326,
      "epoch": 1.3570785522460938e-05,
      "step": 14230
    },
    {
      "epoch": 1.3570785522460938e-05,
      "step": 14230,
      "training_step_time": 17.46619749069214
    },
    {
      "epoch": 1.3571739196777344e-05,
      "model_forward_time": 5.610813617706299,
      "step": 14231
    },
    {
      "epoch": 1.3571739196777344e-05,
      "step": 14231,
      "training_step_time": 17.061633825302124
    },
    {
      "epoch": 1.357269287109375e-05,
      "model_forward_time": 5.611359596252441,
      "step": 14232
    },
    {
      "epoch": 1.357269287109375e-05,
      "step": 14232,
      "training_step_time": 17.09246277809143
    },
    {
      "epoch": 1.3573646545410157e-05,
      "model_forward_time": 5.604631185531616,
      "step": 14233
    },
    {
      "epoch": 1.3573646545410157e-05,
      "step": 14233,
      "training_step_time": 17.09710717201233
    },
    {
      "epoch": 1.3574600219726562e-05,
      "model_forward_time": 5.608030319213867,
      "step": 14234
    },
    {
      "epoch": 1.3574600219726562e-05,
      "step": 14234,
      "training_step_time": 17.0975124835968
    },
    {
      "epoch": 1.3575553894042968e-05,
      "model_forward_time": 5.620280981063843,
      "step": 14235
    },
    {
      "epoch": 1.3575553894042968e-05,
      "step": 14235,
      "training_step_time": 17.127656936645508
    },
    {
      "epoch": 1.3576507568359376e-05,
      "model_forward_time": 5.979060649871826,
      "step": 14236
    },
    {
      "epoch": 1.3576507568359376e-05,
      "step": 14236,
      "training_step_time": 17.469075202941895
    },
    {
      "epoch": 1.3577461242675781e-05,
      "model_forward_time": 5.59672212600708,
      "step": 14237
    },
    {
      "epoch": 1.3577461242675781e-05,
      "step": 14237,
      "training_step_time": 17.009037017822266
    },
    {
      "epoch": 1.3578414916992187e-05,
      "model_forward_time": 5.5834667682647705,
      "step": 14238
    },
    {
      "epoch": 1.3578414916992187e-05,
      "step": 14238,
      "training_step_time": 17.007264375686646
    },
    {
      "epoch": 1.3579368591308594e-05,
      "model_forward_time": 5.575857639312744,
      "step": 14239
    },
    {
      "epoch": 1.3579368591308594e-05,
      "step": 14239,
      "training_step_time": 16.9949312210083
    },
    {
      "epoch": 1.3580322265625e-05,
      "grad_norm": 1.0246922969818115,
      "learning_rate": 1e-05,
      "loss": 0.0654,
      "step": 14240
    },
    {
      "epoch": 1.3580322265625e-05,
      "model_forward_time": 5.575762987136841,
      "step": 14240
    },
    {
      "dynamics_loss_avg": 0.059521767497062686,
      "epoch": 1.3580322265625e-05,
      "step": 14240
    },
    {
      "action_loss_avg": 0.006300995638594031,
      "epoch": 1.3580322265625e-05,
      "step": 14240
    },
    {
      "epoch": 1.3580322265625e-05,
      "step": 14240,
      "training_step_time": 16.9892315864563
    },
    {
      "epoch": 1.3581275939941406e-05,
      "model_forward_time": 5.9568140506744385,
      "step": 14241
    },
    {
      "epoch": 1.3581275939941406e-05,
      "step": 14241,
      "training_step_time": 17.38983988761902
    },
    {
      "epoch": 1.3582229614257813e-05,
      "model_forward_time": 5.583559513092041,
      "step": 14242
    },
    {
      "epoch": 1.3582229614257813e-05,
      "step": 14242,
      "training_step_time": 17.028034210205078
    },
    {
      "epoch": 1.3583183288574219e-05,
      "model_forward_time": 5.589735507965088,
      "step": 14243
    },
    {
      "epoch": 1.3583183288574219e-05,
      "step": 14243,
      "training_step_time": 17.04735493659973
    },
    {
      "epoch": 1.3584136962890625e-05,
      "model_forward_time": 5.583688259124756,
      "step": 14244
    },
    {
      "epoch": 1.3584136962890625e-05,
      "step": 14244,
      "training_step_time": 17.017590522766113
    },
    {
      "epoch": 1.358509063720703e-05,
      "model_forward_time": 5.59819769859314,
      "step": 14245
    },
    {
      "epoch": 1.358509063720703e-05,
      "step": 14245,
      "training_step_time": 17.079193115234375
    },
    {
      "epoch": 1.3586044311523438e-05,
      "model_forward_time": 5.973310470581055,
      "step": 14246
    },
    {
      "epoch": 1.3586044311523438e-05,
      "step": 14246,
      "training_step_time": 17.44316601753235
    },
    {
      "epoch": 1.3586997985839844e-05,
      "model_forward_time": 5.606220006942749,
      "step": 14247
    },
    {
      "epoch": 1.3586997985839844e-05,
      "step": 14247,
      "training_step_time": 17.080384016036987
    },
    {
      "epoch": 1.358795166015625e-05,
      "model_forward_time": 5.609788417816162,
      "step": 14248
    },
    {
      "epoch": 1.358795166015625e-05,
      "step": 14248,
      "training_step_time": 17.090856790542603
    },
    {
      "epoch": 1.3588905334472657e-05,
      "model_forward_time": 5.606693267822266,
      "step": 14249
    },
    {
      "epoch": 1.3588905334472657e-05,
      "step": 14249,
      "training_step_time": 17.0746009349823
    },
    {
      "epoch": 1.3589859008789062e-05,
      "grad_norm": 0.8820679783821106,
      "learning_rate": 1e-05,
      "loss": 0.0633,
      "step": 14250
    },
    {
      "epoch": 1.3589859008789062e-05,
      "model_forward_time": 5.59977650642395,
      "step": 14250
    },
    {
      "dynamics_loss_avg": 0.05376146184280515,
      "epoch": 1.3589859008789062e-05,
      "step": 14250
    },
    {
      "action_loss_avg": 0.008063632517587394,
      "epoch": 1.3589859008789062e-05,
      "step": 14250
    },
    {
      "epoch": 1.3589859008789062e-05,
      "step": 14250,
      "training_step_time": 17.081979274749756
    },
    {
      "epoch": 1.3590812683105468e-05,
      "model_forward_time": 5.618359804153442,
      "step": 14251
    },
    {
      "epoch": 1.3590812683105468e-05,
      "step": 14251,
      "training_step_time": 17.114611625671387
    },
    {
      "epoch": 1.3591766357421876e-05,
      "model_forward_time": 6.009839773178101,
      "step": 14252
    },
    {
      "epoch": 1.3591766357421876e-05,
      "step": 14252,
      "training_step_time": 17.501027584075928
    },
    {
      "epoch": 1.3592720031738281e-05,
      "model_forward_time": 5.616251230239868,
      "step": 14253
    },
    {
      "epoch": 1.3592720031738281e-05,
      "step": 14253,
      "training_step_time": 17.08651041984558
    },
    {
      "epoch": 1.3593673706054687e-05,
      "model_forward_time": 5.609996318817139,
      "step": 14254
    },
    {
      "epoch": 1.3593673706054687e-05,
      "step": 14254,
      "training_step_time": 17.07052445411682
    },
    {
      "epoch": 1.3594627380371094e-05,
      "model_forward_time": 5.6187355518341064,
      "step": 14255
    },
    {
      "epoch": 1.3594627380371094e-05,
      "step": 14255,
      "training_step_time": 17.098267555236816
    },
    {
      "epoch": 1.35955810546875e-05,
      "model_forward_time": 5.6000096797943115,
      "step": 14256
    },
    {
      "epoch": 1.35955810546875e-05,
      "step": 14256,
      "training_step_time": 17.05952787399292
    },
    {
      "epoch": 1.3596534729003906e-05,
      "model_forward_time": 5.962103366851807,
      "step": 14257
    },
    {
      "epoch": 1.3596534729003906e-05,
      "step": 14257,
      "training_step_time": 17.41731333732605
    },
    {
      "epoch": 1.3597488403320312e-05,
      "model_forward_time": 5.603403329849243,
      "step": 14258
    },
    {
      "epoch": 1.3597488403320312e-05,
      "step": 14258,
      "training_step_time": 17.08416223526001
    },
    {
      "epoch": 1.3598442077636719e-05,
      "model_forward_time": 5.601752281188965,
      "step": 14259
    },
    {
      "epoch": 1.3598442077636719e-05,
      "step": 14259,
      "training_step_time": 17.08952522277832
    },
    {
      "epoch": 1.3599395751953125e-05,
      "grad_norm": 1.6380228996276855,
      "learning_rate": 1e-05,
      "loss": 0.0686,
      "step": 14260
    },
    {
      "epoch": 1.3599395751953125e-05,
      "model_forward_time": 5.609720468521118,
      "step": 14260
    },
    {
      "dynamics_loss_avg": 0.05889336206018925,
      "epoch": 1.3599395751953125e-05,
      "step": 14260
    },
    {
      "action_loss_avg": 0.008781165140680969,
      "epoch": 1.3599395751953125e-05,
      "step": 14260
    },
    {
      "epoch": 1.3599395751953125e-05,
      "step": 14260,
      "training_step_time": 17.090590238571167
    },
    {
      "epoch": 1.360034942626953e-05,
      "model_forward_time": 5.605206489562988,
      "step": 14261
    },
    {
      "epoch": 1.360034942626953e-05,
      "step": 14261,
      "training_step_time": 17.07075524330139
    },
    {
      "epoch": 1.3601303100585938e-05,
      "model_forward_time": 5.59694504737854,
      "step": 14262
    },
    {
      "epoch": 1.3601303100585938e-05,
      "step": 14262,
      "training_step_time": 17.084001779556274
    },
    {
      "epoch": 1.3602256774902344e-05,
      "model_forward_time": 5.969287395477295,
      "step": 14263
    },
    {
      "epoch": 1.3602256774902344e-05,
      "step": 14263,
      "training_step_time": 17.427188634872437
    },
    {
      "epoch": 1.360321044921875e-05,
      "model_forward_time": 5.598272800445557,
      "step": 14264
    },
    {
      "epoch": 1.360321044921875e-05,
      "step": 14264,
      "training_step_time": 17.05104422569275
    },
    {
      "epoch": 1.3604164123535157e-05,
      "model_forward_time": 5.607312917709351,
      "step": 14265
    },
    {
      "epoch": 1.3604164123535157e-05,
      "step": 14265,
      "training_step_time": 17.067945957183838
    },
    {
      "epoch": 1.3605117797851563e-05,
      "model_forward_time": 5.608633756637573,
      "step": 14266
    },
    {
      "epoch": 1.3605117797851563e-05,
      "step": 14266,
      "training_step_time": 17.092592239379883
    },
    {
      "epoch": 1.3606071472167968e-05,
      "model_forward_time": 5.60483193397522,
      "step": 14267
    },
    {
      "epoch": 1.3606071472167968e-05,
      "step": 14267,
      "training_step_time": 17.055644035339355
    },
    {
      "epoch": 1.3607025146484376e-05,
      "model_forward_time": 5.59922456741333,
      "step": 14268
    },
    {
      "epoch": 1.3607025146484376e-05,
      "step": 14268,
      "training_step_time": 17.403128147125244
    },
    {
      "epoch": 1.3607978820800781e-05,
      "model_forward_time": 5.586174726486206,
      "step": 14269
    },
    {
      "epoch": 1.3607978820800781e-05,
      "step": 14269,
      "training_step_time": 16.994067668914795
    },
    {
      "epoch": 1.3608932495117187e-05,
      "grad_norm": 0.709560215473175,
      "learning_rate": 1e-05,
      "loss": 0.065,
      "step": 14270
    },
    {
      "epoch": 1.3608932495117187e-05,
      "model_forward_time": 5.589102268218994,
      "step": 14270
    },
    {
      "dynamics_loss_avg": 0.07042822651565075,
      "epoch": 1.3608932495117187e-05,
      "step": 14270
    },
    {
      "action_loss_avg": 0.008858978073112667,
      "epoch": 1.3608932495117187e-05,
      "step": 14270
    },
    {
      "epoch": 1.3608932495117187e-05,
      "step": 14270,
      "training_step_time": 17.02238965034485
    },
    {
      "epoch": 1.3609886169433595e-05,
      "model_forward_time": 5.587717533111572,
      "step": 14271
    },
    {
      "epoch": 1.3609886169433595e-05,
      "step": 14271,
      "training_step_time": 17.02199649810791
    },
    {
      "epoch": 1.361083984375e-05,
      "model_forward_time": 5.591414451599121,
      "step": 14272
    },
    {
      "epoch": 1.361083984375e-05,
      "step": 14272,
      "training_step_time": 17.02302622795105
    },
    {
      "epoch": 1.3611793518066406e-05,
      "model_forward_time": 5.588669300079346,
      "step": 14273
    },
    {
      "epoch": 1.3611793518066406e-05,
      "step": 14273,
      "training_step_time": 17.05938982963562
    },
    {
      "epoch": 1.3612747192382812e-05,
      "model_forward_time": 5.989946365356445,
      "step": 14274
    },
    {
      "epoch": 1.3612747192382812e-05,
      "step": 14274,
      "training_step_time": 17.477580547332764
    },
    {
      "epoch": 1.3613700866699219e-05,
      "model_forward_time": 5.625325441360474,
      "step": 14275
    },
    {
      "epoch": 1.3613700866699219e-05,
      "step": 14275,
      "training_step_time": 17.130436420440674
    },
    {
      "epoch": 1.3614654541015625e-05,
      "model_forward_time": 5.6240458488464355,
      "step": 14276
    },
    {
      "epoch": 1.3614654541015625e-05,
      "step": 14276,
      "training_step_time": 17.13603186607361
    },
    {
      "epoch": 1.361560821533203e-05,
      "model_forward_time": 5.620231628417969,
      "step": 14277
    },
    {
      "epoch": 1.361560821533203e-05,
      "step": 14277,
      "training_step_time": 17.110007524490356
    },
    {
      "epoch": 1.3616561889648438e-05,
      "model_forward_time": 5.612029790878296,
      "step": 14278
    },
    {
      "epoch": 1.3616561889648438e-05,
      "step": 14278,
      "training_step_time": 17.10944104194641
    },
    {
      "epoch": 1.3617515563964844e-05,
      "model_forward_time": 6.005148410797119,
      "step": 14279
    },
    {
      "epoch": 1.3617515563964844e-05,
      "step": 14279,
      "training_step_time": 17.49560523033142
    },
    {
      "epoch": 1.361846923828125e-05,
      "grad_norm": 1.4527509212493896,
      "learning_rate": 1e-05,
      "loss": 0.0973,
      "step": 14280
    },
    {
      "epoch": 1.361846923828125e-05,
      "model_forward_time": 5.612120866775513,
      "step": 14280
    },
    {
      "dynamics_loss_avg": 0.07416062746196986,
      "epoch": 1.361846923828125e-05,
      "step": 14280
    },
    {
      "action_loss_avg": 0.0095379248727113,
      "epoch": 1.361846923828125e-05,
      "step": 14280
    },
    {
      "epoch": 1.361846923828125e-05,
      "step": 14280,
      "training_step_time": 17.10772204399109
    },
    {
      "epoch": 1.3619422912597657e-05,
      "model_forward_time": 5.621257543563843,
      "step": 14281
    },
    {
      "epoch": 1.3619422912597657e-05,
      "step": 14281,
      "training_step_time": 17.115126371383667
    },
    {
      "epoch": 1.3620376586914063e-05,
      "model_forward_time": 5.611361026763916,
      "step": 14282
    },
    {
      "epoch": 1.3620376586914063e-05,
      "step": 14282,
      "training_step_time": 17.100279569625854
    },
    {
      "epoch": 1.3621330261230468e-05,
      "model_forward_time": 5.602964162826538,
      "step": 14283
    },
    {
      "epoch": 1.3621330261230468e-05,
      "step": 14283,
      "training_step_time": 17.054789066314697
    },
    {
      "epoch": 1.3622283935546876e-05,
      "model_forward_time": 5.600099563598633,
      "step": 14284
    },
    {
      "epoch": 1.3622283935546876e-05,
      "step": 14284,
      "training_step_time": 17.05114245414734
    },
    {
      "epoch": 1.3623237609863281e-05,
      "model_forward_time": 5.976064682006836,
      "step": 14285
    },
    {
      "epoch": 1.3623237609863281e-05,
      "step": 14285,
      "training_step_time": 17.41288995742798
    },
    {
      "epoch": 1.3624191284179687e-05,
      "model_forward_time": 5.586196422576904,
      "step": 14286
    },
    {
      "epoch": 1.3624191284179687e-05,
      "step": 14286,
      "training_step_time": 16.99144196510315
    },
    {
      "epoch": 1.3625144958496095e-05,
      "model_forward_time": 5.591300010681152,
      "step": 14287
    },
    {
      "epoch": 1.3625144958496095e-05,
      "step": 14287,
      "training_step_time": 17.039758443832397
    },
    {
      "epoch": 1.36260986328125e-05,
      "model_forward_time": 5.604215860366821,
      "step": 14288
    },
    {
      "epoch": 1.36260986328125e-05,
      "step": 14288,
      "training_step_time": 17.074817419052124
    },
    {
      "epoch": 1.3627052307128906e-05,
      "model_forward_time": 5.6072728633880615,
      "step": 14289
    },
    {
      "epoch": 1.3627052307128906e-05,
      "step": 14289,
      "training_step_time": 17.0988826751709
    },
    {
      "epoch": 1.3628005981445312e-05,
      "grad_norm": 0.7040669322013855,
      "learning_rate": 1e-05,
      "loss": 0.0481,
      "step": 14290
    },
    {
      "epoch": 1.3628005981445312e-05,
      "model_forward_time": 5.974145412445068,
      "step": 14290
    },
    {
      "dynamics_loss_avg": 0.048715687915682794,
      "epoch": 1.3628005981445312e-05,
      "step": 14290
    },
    {
      "action_loss_avg": 0.005475230095908046,
      "epoch": 1.3628005981445312e-05,
      "step": 14290
    },
    {
      "epoch": 1.3628005981445312e-05,
      "step": 14290,
      "training_step_time": 17.45113492012024
    },
    {
      "epoch": 1.3628959655761719e-05,
      "model_forward_time": 5.593528747558594,
      "step": 14291
    },
    {
      "epoch": 1.3628959655761719e-05,
      "step": 14291,
      "training_step_time": 17.03317642211914
    },
    {
      "epoch": 1.3629913330078125e-05,
      "model_forward_time": 5.596253395080566,
      "step": 14292
    },
    {
      "epoch": 1.3629913330078125e-05,
      "step": 14292,
      "training_step_time": 17.09161686897278
    },
    {
      "epoch": 1.363086700439453e-05,
      "model_forward_time": 5.614811420440674,
      "step": 14293
    },
    {
      "epoch": 1.363086700439453e-05,
      "step": 14293,
      "training_step_time": 17.139336824417114
    },
    {
      "epoch": 1.3631820678710938e-05,
      "model_forward_time": 5.6326494216918945,
      "step": 14294
    },
    {
      "epoch": 1.3631820678710938e-05,
      "step": 14294,
      "training_step_time": 17.149322748184204
    },
    {
      "epoch": 1.3632774353027344e-05,
      "model_forward_time": 5.631517171859741,
      "step": 14295
    },
    {
      "epoch": 1.3632774353027344e-05,
      "step": 14295,
      "training_step_time": 17.135255098342896
    },
    {
      "epoch": 1.363372802734375e-05,
      "model_forward_time": 6.00196099281311,
      "step": 14296
    },
    {
      "epoch": 1.363372802734375e-05,
      "step": 14296,
      "training_step_time": 17.47854471206665
    },
    {
      "epoch": 1.3634681701660157e-05,
      "model_forward_time": 5.616132020950317,
      "step": 14297
    },
    {
      "epoch": 1.3634681701660157e-05,
      "step": 14297,
      "training_step_time": 17.04907989501953
    },
    {
      "epoch": 1.3635635375976563e-05,
      "model_forward_time": 5.579097747802734,
      "step": 14298
    },
    {
      "epoch": 1.3635635375976563e-05,
      "step": 14298,
      "training_step_time": 16.990473985671997
    },
    {
      "epoch": 1.3636589050292968e-05,
      "model_forward_time": 5.5878684520721436,
      "step": 14299
    },
    {
      "epoch": 1.3636589050292968e-05,
      "step": 14299,
      "training_step_time": 17.042791604995728
    },
    {
      "epoch": 1.3637542724609376e-05,
      "grad_norm": 0.6677570939064026,
      "learning_rate": 1e-05,
      "loss": 0.0626,
      "step": 14300
    },
    {
      "epoch": 1.3637542724609376e-05,
      "model_forward_time": 5.593662738800049,
      "step": 14300
    },
    {
      "dynamics_loss_avg": 0.05060199089348316,
      "epoch": 1.3637542724609376e-05,
      "step": 14300
    },
    {
      "action_loss_avg": 0.007451104233041406,
      "epoch": 1.3637542724609376e-05,
      "step": 14300
    },
    {
      "epoch": 1.3637542724609376e-05,
      "step": 14300,
      "training_step_time": 17.037386178970337
    },
    {
      "epoch": 1.3638496398925781e-05,
      "model_forward_time": 5.594682693481445,
      "step": 14301
    },
    {
      "epoch": 1.3638496398925781e-05,
      "step": 14301,
      "training_step_time": 17.455620050430298
    },
    {
      "epoch": 1.3639450073242187e-05,
      "model_forward_time": 5.630715370178223,
      "step": 14302
    },
    {
      "epoch": 1.3639450073242187e-05,
      "step": 14302,
      "training_step_time": 17.160112142562866
    },
    {
      "epoch": 1.3640403747558595e-05,
      "model_forward_time": 5.6281349658966064,
      "step": 14303
    },
    {
      "epoch": 1.3640403747558595e-05,
      "step": 14303,
      "training_step_time": 17.13835048675537
    },
    {
      "epoch": 1.3641357421875e-05,
      "model_forward_time": 5.601168394088745,
      "step": 14304
    },
    {
      "epoch": 1.3641357421875e-05,
      "step": 14304,
      "training_step_time": 17.036649227142334
    },
    {
      "epoch": 1.3642311096191406e-05,
      "model_forward_time": 5.577605485916138,
      "step": 14305
    },
    {
      "epoch": 1.3642311096191406e-05,
      "step": 14305,
      "training_step_time": 16.98280692100525
    },
    {
      "epoch": 1.3643264770507812e-05,
      "model_forward_time": 5.5774314403533936,
      "step": 14306
    },
    {
      "epoch": 1.3643264770507812e-05,
      "step": 14306,
      "training_step_time": 16.9902184009552
    },
    {
      "epoch": 1.3644218444824219e-05,
      "model_forward_time": 5.9429097175598145,
      "step": 14307
    },
    {
      "epoch": 1.3644218444824219e-05,
      "step": 14307,
      "training_step_time": 17.3897705078125
    },
    {
      "epoch": 1.3645172119140625e-05,
      "model_forward_time": 5.57484245300293,
      "step": 14308
    },
    {
      "epoch": 1.3645172119140625e-05,
      "step": 14308,
      "training_step_time": 16.99686074256897
    },
    {
      "epoch": 1.364612579345703e-05,
      "model_forward_time": 5.594841003417969,
      "step": 14309
    },
    {
      "epoch": 1.364612579345703e-05,
      "step": 14309,
      "training_step_time": 17.021657466888428
    },
    {
      "epoch": 1.3647079467773438e-05,
      "grad_norm": 1.1190277338027954,
      "learning_rate": 1e-05,
      "loss": 0.0639,
      "step": 14310
    },
    {
      "epoch": 1.3647079467773438e-05,
      "model_forward_time": 5.592405080795288,
      "step": 14310
    },
    {
      "dynamics_loss_avg": 0.05402205251157284,
      "epoch": 1.3647079467773438e-05,
      "step": 14310
    },
    {
      "action_loss_avg": 0.007430473016574979,
      "epoch": 1.3647079467773438e-05,
      "step": 14310
    },
    {
      "epoch": 1.3647079467773438e-05,
      "step": 14310,
      "training_step_time": 17.040069580078125
    },
    {
      "epoch": 1.3648033142089844e-05,
      "model_forward_time": 5.6110005378723145,
      "step": 14311
    },
    {
      "epoch": 1.3648033142089844e-05,
      "step": 14311,
      "training_step_time": 17.08772039413452
    },
    {
      "epoch": 1.364898681640625e-05,
      "model_forward_time": 5.962010383605957,
      "step": 14312
    },
    {
      "epoch": 1.364898681640625e-05,
      "step": 14312,
      "training_step_time": 17.467108488082886
    },
    {
      "epoch": 1.3649940490722657e-05,
      "model_forward_time": 5.6036155223846436,
      "step": 14313
    },
    {
      "epoch": 1.3649940490722657e-05,
      "step": 14313,
      "training_step_time": 17.076359272003174
    },
    {
      "epoch": 1.3650894165039063e-05,
      "model_forward_time": 5.600886106491089,
      "step": 14314
    },
    {
      "epoch": 1.3650894165039063e-05,
      "step": 14314,
      "training_step_time": 17.047553777694702
    },
    {
      "epoch": 1.3651847839355468e-05,
      "model_forward_time": 5.582182168960571,
      "step": 14315
    },
    {
      "epoch": 1.3651847839355468e-05,
      "step": 14315,
      "training_step_time": 17.042089462280273
    },
    {
      "epoch": 1.3652801513671876e-05,
      "model_forward_time": 5.594542741775513,
      "step": 14316
    },
    {
      "epoch": 1.3652801513671876e-05,
      "step": 14316,
      "training_step_time": 17.037546396255493
    },
    {
      "epoch": 1.3653755187988281e-05,
      "model_forward_time": 5.637190103530884,
      "step": 14317
    },
    {
      "epoch": 1.3653755187988281e-05,
      "step": 14317,
      "training_step_time": 17.13370990753174
    },
    {
      "epoch": 1.3654708862304687e-05,
      "model_forward_time": 6.1351318359375,
      "step": 14318
    },
    {
      "epoch": 1.3654708862304687e-05,
      "step": 14318,
      "training_step_time": 17.6438045501709
    },
    {
      "epoch": 1.3655662536621095e-05,
      "model_forward_time": 5.613271713256836,
      "step": 14319
    },
    {
      "epoch": 1.3655662536621095e-05,
      "step": 14319,
      "training_step_time": 17.100978136062622
    },
    {
      "epoch": 1.36566162109375e-05,
      "grad_norm": 0.3884246349334717,
      "learning_rate": 1e-05,
      "loss": 0.0478,
      "step": 14320
    },
    {
      "epoch": 1.36566162109375e-05,
      "model_forward_time": 5.603639125823975,
      "step": 14320
    },
    {
      "dynamics_loss_avg": 0.04632788375020027,
      "epoch": 1.36566162109375e-05,
      "step": 14320
    },
    {
      "action_loss_avg": 0.005052106897346675,
      "epoch": 1.36566162109375e-05,
      "step": 14320
    },
    {
      "epoch": 1.36566162109375e-05,
      "step": 14320,
      "training_step_time": 17.07790517807007
    },
    {
      "epoch": 1.3657569885253906e-05,
      "model_forward_time": 5.6026763916015625,
      "step": 14321
    },
    {
      "epoch": 1.3657569885253906e-05,
      "step": 14321,
      "training_step_time": 17.06204342842102
    },
    {
      "epoch": 1.3658523559570312e-05,
      "model_forward_time": 5.604656934738159,
      "step": 14322
    },
    {
      "epoch": 1.3658523559570312e-05,
      "step": 14322,
      "training_step_time": 17.051292419433594
    },
    {
      "epoch": 1.3659477233886719e-05,
      "model_forward_time": 6.026233673095703,
      "step": 14323
    },
    {
      "epoch": 1.3659477233886719e-05,
      "step": 14323,
      "training_step_time": 17.497681856155396
    },
    {
      "epoch": 1.3660430908203125e-05,
      "model_forward_time": 5.6000752449035645,
      "step": 14324
    },
    {
      "epoch": 1.3660430908203125e-05,
      "step": 14324,
      "training_step_time": 17.06517481803894
    },
    {
      "epoch": 1.366138458251953e-05,
      "model_forward_time": 5.594457149505615,
      "step": 14325
    },
    {
      "epoch": 1.366138458251953e-05,
      "step": 14325,
      "training_step_time": 17.072510719299316
    },
    {
      "epoch": 1.3662338256835938e-05,
      "model_forward_time": 5.617948055267334,
      "step": 14326
    },
    {
      "epoch": 1.3662338256835938e-05,
      "step": 14326,
      "training_step_time": 17.11252236366272
    },
    {
      "epoch": 1.3663291931152344e-05,
      "model_forward_time": 5.621112823486328,
      "step": 14327
    },
    {
      "epoch": 1.3663291931152344e-05,
      "step": 14327,
      "training_step_time": 17.133281707763672
    },
    {
      "epoch": 1.366424560546875e-05,
      "model_forward_time": 5.621616363525391,
      "step": 14328
    },
    {
      "epoch": 1.366424560546875e-05,
      "step": 14328,
      "training_step_time": 17.13122034072876
    },
    {
      "epoch": 1.3665199279785157e-05,
      "model_forward_time": 6.02914834022522,
      "step": 14329
    },
    {
      "epoch": 1.3665199279785157e-05,
      "step": 14329,
      "training_step_time": 17.51814603805542
    },
    {
      "epoch": 1.3666152954101563e-05,
      "grad_norm": 0.6583316326141357,
      "learning_rate": 1e-05,
      "loss": 0.0569,
      "step": 14330
    },
    {
      "epoch": 1.3666152954101563e-05,
      "model_forward_time": 5.622905492782593,
      "step": 14330
    },
    {
      "dynamics_loss_avg": 0.04702792149037123,
      "epoch": 1.3666152954101563e-05,
      "step": 14330
    },
    {
      "action_loss_avg": 0.005900076800025999,
      "epoch": 1.3666152954101563e-05,
      "step": 14330
    },
    {
      "epoch": 1.3666152954101563e-05,
      "step": 14330,
      "training_step_time": 17.121569871902466
    },
    {
      "epoch": 1.3667106628417968e-05,
      "model_forward_time": 5.6157386302948,
      "step": 14331
    },
    {
      "epoch": 1.3667106628417968e-05,
      "step": 14331,
      "training_step_time": 17.11257290840149
    },
    {
      "epoch": 1.3668060302734376e-05,
      "model_forward_time": 5.6157755851745605,
      "step": 14332
    },
    {
      "epoch": 1.3668060302734376e-05,
      "step": 14332,
      "training_step_time": 17.109060764312744
    },
    {
      "epoch": 1.3669013977050781e-05,
      "model_forward_time": 5.60941743850708,
      "step": 14333
    },
    {
      "epoch": 1.3669013977050781e-05,
      "step": 14333,
      "training_step_time": 17.096970319747925
    },
    {
      "epoch": 1.3669967651367187e-05,
      "model_forward_time": 5.60369348526001,
      "step": 14334
    },
    {
      "epoch": 1.3669967651367187e-05,
      "step": 14334,
      "training_step_time": 17.45811939239502
    },
    {
      "epoch": 1.3670921325683595e-05,
      "model_forward_time": 5.605788469314575,
      "step": 14335
    },
    {
      "epoch": 1.3670921325683595e-05,
      "step": 14335,
      "training_step_time": 17.088266611099243
    },
    {
      "epoch": 1.3671875e-05,
      "model_forward_time": 5.602693796157837,
      "step": 14336
    },
    {
      "epoch": 1.3671875e-05,
      "step": 14336,
      "training_step_time": 17.07345151901245
    },
    {
      "epoch": 1.3672828674316406e-05,
      "model_forward_time": 5.592628240585327,
      "step": 14337
    },
    {
      "epoch": 1.3672828674316406e-05,
      "step": 14337,
      "training_step_time": 17.020135164260864
    },
    {
      "epoch": 1.3673782348632812e-05,
      "model_forward_time": 5.58869481086731,
      "step": 14338
    },
    {
      "epoch": 1.3673782348632812e-05,
      "step": 14338,
      "training_step_time": 17.020145654678345
    },
    {
      "epoch": 1.3674736022949219e-05,
      "model_forward_time": 5.591732978820801,
      "step": 14339
    },
    {
      "epoch": 1.3674736022949219e-05,
      "step": 14339,
      "training_step_time": 17.069523572921753
    },
    {
      "epoch": 1.3675689697265625e-05,
      "grad_norm": 0.0856005996465683,
      "learning_rate": 1e-05,
      "loss": 0.0592,
      "step": 14340
    },
    {
      "epoch": 1.3675689697265625e-05,
      "model_forward_time": 5.980132341384888,
      "step": 14340
    },
    {
      "dynamics_loss_avg": 0.05360476225614548,
      "epoch": 1.3675689697265625e-05,
      "step": 14340
    },
    {
      "action_loss_avg": 0.006123605836182833,
      "epoch": 1.3675689697265625e-05,
      "step": 14340
    },
    {
      "epoch": 1.3675689697265625e-05,
      "step": 14340,
      "training_step_time": 17.478538274765015
    },
    {
      "epoch": 1.367664337158203e-05,
      "model_forward_time": 5.615462064743042,
      "step": 14341
    },
    {
      "epoch": 1.367664337158203e-05,
      "step": 14341,
      "training_step_time": 17.097907066345215
    },
    {
      "epoch": 1.3677597045898438e-05,
      "model_forward_time": 5.598337411880493,
      "step": 14342
    },
    {
      "epoch": 1.3677597045898438e-05,
      "step": 14342,
      "training_step_time": 17.03484559059143
    },
    {
      "epoch": 1.3678550720214844e-05,
      "model_forward_time": 5.592743873596191,
      "step": 14343
    },
    {
      "epoch": 1.3678550720214844e-05,
      "step": 14343,
      "training_step_time": 17.03375005722046
    },
    {
      "epoch": 1.367950439453125e-05,
      "model_forward_time": 5.59747052192688,
      "step": 14344
    },
    {
      "epoch": 1.367950439453125e-05,
      "step": 14344,
      "training_step_time": 17.064205169677734
    },
    {
      "epoch": 1.3680458068847657e-05,
      "model_forward_time": 5.994697093963623,
      "step": 14345
    },
    {
      "epoch": 1.3680458068847657e-05,
      "step": 14345,
      "training_step_time": 17.490482807159424
    },
    {
      "epoch": 1.3681411743164063e-05,
      "model_forward_time": 5.617982625961304,
      "step": 14346
    },
    {
      "epoch": 1.3681411743164063e-05,
      "step": 14346,
      "training_step_time": 17.104535341262817
    },
    {
      "epoch": 1.3682365417480468e-05,
      "model_forward_time": 5.602398872375488,
      "step": 14347
    },
    {
      "epoch": 1.3682365417480468e-05,
      "step": 14347,
      "training_step_time": 17.052592039108276
    },
    {
      "epoch": 1.3683319091796876e-05,
      "model_forward_time": 5.593252420425415,
      "step": 14348
    },
    {
      "epoch": 1.3683319091796876e-05,
      "step": 14348,
      "training_step_time": 17.034403085708618
    },
    {
      "epoch": 1.3684272766113281e-05,
      "model_forward_time": 5.588854789733887,
      "step": 14349
    },
    {
      "epoch": 1.3684272766113281e-05,
      "step": 14349,
      "training_step_time": 17.06242799758911
    },
    {
      "epoch": 1.3685226440429687e-05,
      "grad_norm": 0.6879026889801025,
      "learning_rate": 1e-05,
      "loss": 0.0628,
      "step": 14350
    },
    {
      "epoch": 1.3685226440429687e-05,
      "model_forward_time": 5.609700679779053,
      "step": 14350
    },
    {
      "dynamics_loss_avg": 0.056100411154329774,
      "epoch": 1.3685226440429687e-05,
      "step": 14350
    },
    {
      "action_loss_avg": 0.007418826292268932,
      "epoch": 1.3685226440429687e-05,
      "step": 14350
    },
    {
      "epoch": 1.3685226440429687e-05,
      "step": 14350,
      "training_step_time": 17.10585308074951
    },
    {
      "epoch": 1.3686180114746095e-05,
      "model_forward_time": 5.990325927734375,
      "step": 14351
    },
    {
      "epoch": 1.3686180114746095e-05,
      "step": 14351,
      "training_step_time": 17.47020697593689
    },
    {
      "epoch": 1.36871337890625e-05,
      "model_forward_time": 5.596155166625977,
      "step": 14352
    },
    {
      "epoch": 1.36871337890625e-05,
      "step": 14352,
      "training_step_time": 17.04435157775879
    },
    {
      "epoch": 1.3688087463378906e-05,
      "model_forward_time": 5.58254599571228,
      "step": 14353
    },
    {
      "epoch": 1.3688087463378906e-05,
      "step": 14353,
      "training_step_time": 17.007699728012085
    },
    {
      "epoch": 1.3689041137695312e-05,
      "model_forward_time": 5.580995559692383,
      "step": 14354
    },
    {
      "epoch": 1.3689041137695312e-05,
      "step": 14354,
      "training_step_time": 17.002543210983276
    },
    {
      "epoch": 1.368999481201172e-05,
      "model_forward_time": 5.584566593170166,
      "step": 14355
    },
    {
      "epoch": 1.368999481201172e-05,
      "step": 14355,
      "training_step_time": 16.996521949768066
    },
    {
      "epoch": 1.3690948486328125e-05,
      "model_forward_time": 5.9665210247039795,
      "step": 14356
    },
    {
      "epoch": 1.3690948486328125e-05,
      "step": 14356,
      "training_step_time": 17.413084268569946
    },
    {
      "epoch": 1.369190216064453e-05,
      "model_forward_time": 5.588010311126709,
      "step": 14357
    },
    {
      "epoch": 1.369190216064453e-05,
      "step": 14357,
      "training_step_time": 17.053953886032104
    },
    {
      "epoch": 1.3692855834960938e-05,
      "model_forward_time": 5.612057685852051,
      "step": 14358
    },
    {
      "epoch": 1.3692855834960938e-05,
      "step": 14358,
      "training_step_time": 17.109649181365967
    },
    {
      "epoch": 1.3693809509277344e-05,
      "model_forward_time": 5.6267900466918945,
      "step": 14359
    },
    {
      "epoch": 1.3693809509277344e-05,
      "step": 14359,
      "training_step_time": 17.12298011779785
    },
    {
      "epoch": 1.369476318359375e-05,
      "grad_norm": 0.38371512293815613,
      "learning_rate": 1e-05,
      "loss": 0.047,
      "step": 14360
    },
    {
      "epoch": 1.369476318359375e-05,
      "model_forward_time": 5.6158154010772705,
      "step": 14360
    },
    {
      "dynamics_loss_avg": 0.046280468441545965,
      "epoch": 1.369476318359375e-05,
      "step": 14360
    },
    {
      "action_loss_avg": 0.0064309667330235245,
      "epoch": 1.369476318359375e-05,
      "step": 14360
    },
    {
      "epoch": 1.369476318359375e-05,
      "step": 14360,
      "training_step_time": 17.09238052368164
    },
    {
      "epoch": 1.3695716857910157e-05,
      "model_forward_time": 5.623450994491577,
      "step": 14361
    },
    {
      "epoch": 1.3695716857910157e-05,
      "step": 14361,
      "training_step_time": 17.112205505371094
    },
    {
      "epoch": 1.3696670532226563e-05,
      "model_forward_time": 6.001844167709351,
      "step": 14362
    },
    {
      "epoch": 1.3696670532226563e-05,
      "step": 14362,
      "training_step_time": 17.46219277381897
    },
    {
      "epoch": 1.3697624206542968e-05,
      "model_forward_time": 5.596304416656494,
      "step": 14363
    },
    {
      "epoch": 1.3697624206542968e-05,
      "step": 14363,
      "training_step_time": 17.026360273361206
    },
    {
      "epoch": 1.3698577880859376e-05,
      "model_forward_time": 5.592800617218018,
      "step": 14364
    },
    {
      "epoch": 1.3698577880859376e-05,
      "step": 14364,
      "training_step_time": 17.046130895614624
    },
    {
      "epoch": 1.3699531555175781e-05,
      "model_forward_time": 5.595982313156128,
      "step": 14365
    },
    {
      "epoch": 1.3699531555175781e-05,
      "step": 14365,
      "training_step_time": 17.068516731262207
    },
    {
      "epoch": 1.3700485229492187e-05,
      "model_forward_time": 5.606485366821289,
      "step": 14366
    },
    {
      "epoch": 1.3700485229492187e-05,
      "step": 14366,
      "training_step_time": 17.118196964263916
    },
    {
      "epoch": 1.3701438903808595e-05,
      "model_forward_time": 5.628547430038452,
      "step": 14367
    },
    {
      "epoch": 1.3701438903808595e-05,
      "step": 14367,
      "training_step_time": 17.4865665435791
    },
    {
      "epoch": 1.3702392578125e-05,
      "model_forward_time": 5.6152660846710205,
      "step": 14368
    },
    {
      "epoch": 1.3702392578125e-05,
      "step": 14368,
      "training_step_time": 17.08371376991272
    },
    {
      "epoch": 1.3703346252441406e-05,
      "model_forward_time": 5.592866659164429,
      "step": 14369
    },
    {
      "epoch": 1.3703346252441406e-05,
      "step": 14369,
      "training_step_time": 17.000170469284058
    },
    {
      "epoch": 1.3704299926757812e-05,
      "grad_norm": 0.2639509439468384,
      "learning_rate": 1e-05,
      "loss": 0.0687,
      "step": 14370
    },
    {
      "epoch": 1.3704299926757812e-05,
      "model_forward_time": 5.585495471954346,
      "step": 14370
    },
    {
      "dynamics_loss_avg": 0.05734190978109836,
      "epoch": 1.3704299926757812e-05,
      "step": 14370
    },
    {
      "action_loss_avg": 0.008567188610322773,
      "epoch": 1.3704299926757812e-05,
      "step": 14370
    },
    {
      "epoch": 1.3704299926757812e-05,
      "step": 14370,
      "training_step_time": 17.0162296295166
    },
    {
      "epoch": 1.370525360107422e-05,
      "model_forward_time": 5.586388111114502,
      "step": 14371
    },
    {
      "epoch": 1.370525360107422e-05,
      "step": 14371,
      "training_step_time": 17.01906132698059
    },
    {
      "epoch": 1.3706207275390625e-05,
      "model_forward_time": 5.592845439910889,
      "step": 14372
    },
    {
      "epoch": 1.3706207275390625e-05,
      "step": 14372,
      "training_step_time": 17.077476978302002
    },
    {
      "epoch": 1.370716094970703e-05,
      "model_forward_time": 5.9914867877960205,
      "step": 14373
    },
    {
      "epoch": 1.370716094970703e-05,
      "step": 14373,
      "training_step_time": 17.46336007118225
    },
    {
      "epoch": 1.3708114624023438e-05,
      "model_forward_time": 5.595783948898315,
      "step": 14374
    },
    {
      "epoch": 1.3708114624023438e-05,
      "step": 14374,
      "training_step_time": 17.058264017105103
    },
    {
      "epoch": 1.3709068298339844e-05,
      "model_forward_time": 5.5968945026397705,
      "step": 14375
    },
    {
      "epoch": 1.3709068298339844e-05,
      "step": 14375,
      "training_step_time": 17.034061908721924
    },
    {
      "epoch": 1.371002197265625e-05,
      "model_forward_time": 5.591131925582886,
      "step": 14376
    },
    {
      "epoch": 1.371002197265625e-05,
      "step": 14376,
      "training_step_time": 17.060667276382446
    },
    {
      "epoch": 1.3710975646972657e-05,
      "model_forward_time": 5.604018449783325,
      "step": 14377
    },
    {
      "epoch": 1.3710975646972657e-05,
      "step": 14377,
      "training_step_time": 17.070133686065674
    },
    {
      "epoch": 1.3711929321289063e-05,
      "model_forward_time": 5.961401462554932,
      "step": 14378
    },
    {
      "epoch": 1.3711929321289063e-05,
      "step": 14378,
      "training_step_time": 17.41895031929016
    },
    {
      "epoch": 1.3712882995605468e-05,
      "model_forward_time": 5.596059322357178,
      "step": 14379
    },
    {
      "epoch": 1.3712882995605468e-05,
      "step": 14379,
      "training_step_time": 17.041672229766846
    },
    {
      "epoch": 1.3713836669921876e-05,
      "grad_norm": 0.8076273202896118,
      "learning_rate": 1e-05,
      "loss": 0.0702,
      "step": 14380
    },
    {
      "epoch": 1.3713836669921876e-05,
      "model_forward_time": 5.589154958724976,
      "step": 14380
    },
    {
      "dynamics_loss_avg": 0.0634176829829812,
      "epoch": 1.3713836669921876e-05,
      "step": 14380
    },
    {
      "action_loss_avg": 0.006810144300106913,
      "epoch": 1.3713836669921876e-05,
      "step": 14380
    },
    {
      "epoch": 1.3713836669921876e-05,
      "step": 14380,
      "training_step_time": 17.011151552200317
    },
    {
      "epoch": 1.3714790344238281e-05,
      "model_forward_time": 5.591429710388184,
      "step": 14381
    },
    {
      "epoch": 1.3714790344238281e-05,
      "step": 14381,
      "training_step_time": 17.053680181503296
    },
    {
      "epoch": 1.3715744018554687e-05,
      "model_forward_time": 5.599507093429565,
      "step": 14382
    },
    {
      "epoch": 1.3715744018554687e-05,
      "step": 14382,
      "training_step_time": 17.07695984840393
    },
    {
      "epoch": 1.3716697692871093e-05,
      "model_forward_time": 5.606149196624756,
      "step": 14383
    },
    {
      "epoch": 1.3716697692871093e-05,
      "step": 14383,
      "training_step_time": 17.0556001663208
    },
    {
      "epoch": 1.37176513671875e-05,
      "model_forward_time": 6.003518104553223,
      "step": 14384
    },
    {
      "epoch": 1.37176513671875e-05,
      "step": 14384,
      "training_step_time": 17.47257137298584
    },
    {
      "epoch": 1.3718605041503906e-05,
      "model_forward_time": 5.603402853012085,
      "step": 14385
    },
    {
      "epoch": 1.3718605041503906e-05,
      "step": 14385,
      "training_step_time": 17.058514833450317
    },
    {
      "epoch": 1.3719558715820312e-05,
      "model_forward_time": 5.596482515335083,
      "step": 14386
    },
    {
      "epoch": 1.3719558715820312e-05,
      "step": 14386,
      "training_step_time": 17.055327892303467
    },
    {
      "epoch": 1.372051239013672e-05,
      "model_forward_time": 5.608642101287842,
      "step": 14387
    },
    {
      "epoch": 1.372051239013672e-05,
      "step": 14387,
      "training_step_time": 17.077495574951172
    },
    {
      "epoch": 1.3721466064453125e-05,
      "model_forward_time": 5.611725568771362,
      "step": 14388
    },
    {
      "epoch": 1.3721466064453125e-05,
      "step": 14388,
      "training_step_time": 17.080549001693726
    },
    {
      "epoch": 1.372241973876953e-05,
      "model_forward_time": 5.966768741607666,
      "step": 14389
    },
    {
      "epoch": 1.372241973876953e-05,
      "step": 14389,
      "training_step_time": 17.412956714630127
    },
    {
      "epoch": 1.3723373413085938e-05,
      "grad_norm": 1.0326080322265625,
      "learning_rate": 1e-05,
      "loss": 0.0641,
      "step": 14390
    },
    {
      "epoch": 1.3723373413085938e-05,
      "model_forward_time": 5.604731798171997,
      "step": 14390
    },
    {
      "dynamics_loss_avg": 0.05377388559281826,
      "epoch": 1.3723373413085938e-05,
      "step": 14390
    },
    {
      "action_loss_avg": 0.005720708100125193,
      "epoch": 1.3723373413085938e-05,
      "step": 14390
    },
    {
      "epoch": 1.3723373413085938e-05,
      "step": 14390,
      "training_step_time": 17.0835542678833
    },
    {
      "epoch": 1.3724327087402344e-05,
      "model_forward_time": 5.6057586669921875,
      "step": 14391
    },
    {
      "epoch": 1.3724327087402344e-05,
      "step": 14391,
      "training_step_time": 17.06434941291809
    },
    {
      "epoch": 1.372528076171875e-05,
      "model_forward_time": 5.599604368209839,
      "step": 14392
    },
    {
      "epoch": 1.372528076171875e-05,
      "step": 14392,
      "training_step_time": 17.042020797729492
    },
    {
      "epoch": 1.3726234436035157e-05,
      "model_forward_time": 5.590146780014038,
      "step": 14393
    },
    {
      "epoch": 1.3726234436035157e-05,
      "step": 14393,
      "training_step_time": 17.033828735351562
    },
    {
      "epoch": 1.3727188110351563e-05,
      "model_forward_time": 5.586086273193359,
      "step": 14394
    },
    {
      "epoch": 1.3727188110351563e-05,
      "step": 14394,
      "training_step_time": 17.039998292922974
    },
    {
      "epoch": 1.3728141784667968e-05,
      "model_forward_time": 5.999861478805542,
      "step": 14395
    },
    {
      "epoch": 1.3728141784667968e-05,
      "step": 14395,
      "training_step_time": 17.460909843444824
    },
    {
      "epoch": 1.3729095458984376e-05,
      "model_forward_time": 5.6000213623046875,
      "step": 14396
    },
    {
      "epoch": 1.3729095458984376e-05,
      "step": 14396,
      "training_step_time": 17.07086730003357
    },
    {
      "epoch": 1.3730049133300781e-05,
      "model_forward_time": 5.609067916870117,
      "step": 14397
    },
    {
      "epoch": 1.3730049133300781e-05,
      "step": 14397,
      "training_step_time": 17.067457914352417
    },
    {
      "epoch": 1.3731002807617187e-05,
      "model_forward_time": 5.600938558578491,
      "step": 14398
    },
    {
      "epoch": 1.3731002807617187e-05,
      "step": 14398,
      "training_step_time": 17.05967926979065
    },
    {
      "epoch": 1.3731956481933593e-05,
      "model_forward_time": 5.5894834995269775,
      "step": 14399
    },
    {
      "epoch": 1.3731956481933593e-05,
      "step": 14399,
      "training_step_time": 17.016427516937256
    },
    {
      "epoch": 1.373291015625e-05,
      "grad_norm": 1.371535062789917,
      "learning_rate": 1e-05,
      "loss": 0.0551,
      "step": 14400
    },
    {
      "epoch": 1.373291015625e-05,
      "model_forward_time": 5.5940587520599365,
      "step": 14400
    },
    {
      "dynamics_loss_avg": 0.04847479127347469,
      "epoch": 1.373291015625e-05,
      "step": 14400
    },
    {
      "action_loss_avg": 0.006061895145103335,
      "epoch": 1.373291015625e-05,
      "step": 14400
    },
    {
      "epoch": 1.373291015625e-05,
      "step": 14400,
      "training_step_time": 17.41961669921875
    },
    {
      "epoch": 1.3733863830566406e-05,
      "model_forward_time": 5.610166311264038,
      "step": 14401
    },
    {
      "epoch": 1.3733863830566406e-05,
      "step": 14401,
      "training_step_time": 17.101807832717896
    },
    {
      "epoch": 1.3734817504882812e-05,
      "model_forward_time": 5.625784158706665,
      "step": 14402
    },
    {
      "epoch": 1.3734817504882812e-05,
      "step": 14402,
      "training_step_time": 17.12836790084839
    },
    {
      "epoch": 1.373577117919922e-05,
      "model_forward_time": 5.629072427749634,
      "step": 14403
    },
    {
      "epoch": 1.373577117919922e-05,
      "step": 14403,
      "training_step_time": 17.1347873210907
    },
    {
      "epoch": 1.3736724853515625e-05,
      "model_forward_time": 5.600446462631226,
      "step": 14404
    },
    {
      "epoch": 1.3736724853515625e-05,
      "step": 14404,
      "training_step_time": 17.038063049316406
    },
    {
      "epoch": 1.373767852783203e-05,
      "model_forward_time": 5.591217756271362,
      "step": 14405
    },
    {
      "epoch": 1.373767852783203e-05,
      "step": 14405,
      "training_step_time": 17.021323680877686
    },
    {
      "epoch": 1.3738632202148438e-05,
      "model_forward_time": 5.962397575378418,
      "step": 14406
    },
    {
      "epoch": 1.3738632202148438e-05,
      "step": 14406,
      "training_step_time": 17.39187002182007
    },
    {
      "epoch": 1.3739585876464844e-05,
      "model_forward_time": 5.590080738067627,
      "step": 14407
    },
    {
      "epoch": 1.3739585876464844e-05,
      "step": 14407,
      "training_step_time": 17.035008192062378
    },
    {
      "epoch": 1.374053955078125e-05,
      "model_forward_time": 5.597116708755493,
      "step": 14408
    },
    {
      "epoch": 1.374053955078125e-05,
      "step": 14408,
      "training_step_time": 17.085633516311646
    },
    {
      "epoch": 1.3741493225097657e-05,
      "model_forward_time": 5.6269237995147705,
      "step": 14409
    },
    {
      "epoch": 1.3741493225097657e-05,
      "step": 14409,
      "training_step_time": 17.13901114463806
    },
    {
      "epoch": 1.3742446899414063e-05,
      "grad_norm": 0.5090615153312683,
      "learning_rate": 1e-05,
      "loss": 0.0569,
      "step": 14410
    },
    {
      "epoch": 1.3742446899414063e-05,
      "model_forward_time": 5.632955312728882,
      "step": 14410
    },
    {
      "dynamics_loss_avg": 0.05238884408026934,
      "epoch": 1.3742446899414063e-05,
      "step": 14410
    },
    {
      "action_loss_avg": 0.005623883556108922,
      "epoch": 1.3742446899414063e-05,
      "step": 14410
    },
    {
      "epoch": 1.3742446899414063e-05,
      "step": 14410,
      "training_step_time": 17.138092279434204
    },
    {
      "epoch": 1.3743400573730468e-05,
      "model_forward_time": 5.954901456832886,
      "step": 14411
    },
    {
      "epoch": 1.3743400573730468e-05,
      "step": 14411,
      "training_step_time": 17.385891437530518
    },
    {
      "epoch": 1.3744354248046876e-05,
      "model_forward_time": 5.563804388046265,
      "step": 14412
    },
    {
      "epoch": 1.3744354248046876e-05,
      "step": 14412,
      "training_step_time": 16.992019176483154
    },
    {
      "epoch": 1.3745307922363281e-05,
      "model_forward_time": 5.564964771270752,
      "step": 14413
    },
    {
      "epoch": 1.3745307922363281e-05,
      "step": 14413,
      "training_step_time": 16.98629403114319
    },
    {
      "epoch": 1.3746261596679687e-05,
      "model_forward_time": 5.5751142501831055,
      "step": 14414
    },
    {
      "epoch": 1.3746261596679687e-05,
      "step": 14414,
      "training_step_time": 16.991938591003418
    },
    {
      "epoch": 1.3747215270996093e-05,
      "model_forward_time": 5.565153121948242,
      "step": 14415
    },
    {
      "epoch": 1.3747215270996093e-05,
      "step": 14415,
      "training_step_time": 16.981797695159912
    },
    {
      "epoch": 1.37481689453125e-05,
      "model_forward_time": 5.583572626113892,
      "step": 14416
    },
    {
      "epoch": 1.37481689453125e-05,
      "step": 14416,
      "training_step_time": 17.01270842552185
    },
    {
      "epoch": 1.3749122619628906e-05,
      "model_forward_time": 5.957031726837158,
      "step": 14417
    },
    {
      "epoch": 1.3749122619628906e-05,
      "step": 14417,
      "training_step_time": 17.405314922332764
    },
    {
      "epoch": 1.3750076293945312e-05,
      "model_forward_time": 5.577829599380493,
      "step": 14418
    },
    {
      "epoch": 1.3750076293945312e-05,
      "step": 14418,
      "training_step_time": 17.02424645423889
    },
    {
      "epoch": 1.375102996826172e-05,
      "model_forward_time": 5.612290859222412,
      "step": 14419
    },
    {
      "epoch": 1.375102996826172e-05,
      "step": 14419,
      "training_step_time": 17.090818881988525
    },
    {
      "epoch": 1.3751983642578125e-05,
      "grad_norm": 0.35755571722984314,
      "learning_rate": 1e-05,
      "loss": 0.0661,
      "step": 14420
    },
    {
      "epoch": 1.3751983642578125e-05,
      "model_forward_time": 5.615676403045654,
      "step": 14420
    },
    {
      "dynamics_loss_avg": 0.06055255346000195,
      "epoch": 1.3751983642578125e-05,
      "step": 14420
    },
    {
      "action_loss_avg": 0.008056305302307009,
      "epoch": 1.3751983642578125e-05,
      "step": 14420
    },
    {
      "epoch": 1.3751983642578125e-05,
      "step": 14420,
      "training_step_time": 17.11083674430847
    },
    {
      "epoch": 1.375293731689453e-05,
      "model_forward_time": 5.632201671600342,
      "step": 14421
    },
    {
      "epoch": 1.375293731689453e-05,
      "step": 14421,
      "training_step_time": 17.12307047843933
    },
    {
      "epoch": 1.3753890991210938e-05,
      "model_forward_time": 5.969958305358887,
      "step": 14422
    },
    {
      "epoch": 1.3753890991210938e-05,
      "step": 14422,
      "training_step_time": 17.441240549087524
    },
    {
      "epoch": 1.3754844665527344e-05,
      "model_forward_time": 5.636698246002197,
      "step": 14423
    },
    {
      "epoch": 1.3754844665527344e-05,
      "step": 14423,
      "training_step_time": 17.10062527656555
    },
    {
      "epoch": 1.375579833984375e-05,
      "model_forward_time": 5.598620414733887,
      "step": 14424
    },
    {
      "epoch": 1.375579833984375e-05,
      "step": 14424,
      "training_step_time": 17.040592670440674
    },
    {
      "epoch": 1.3756752014160157e-05,
      "model_forward_time": 5.606717348098755,
      "step": 14425
    },
    {
      "epoch": 1.3756752014160157e-05,
      "step": 14425,
      "training_step_time": 17.097567319869995
    },
    {
      "epoch": 1.3757705688476563e-05,
      "model_forward_time": 5.6212317943573,
      "step": 14426
    },
    {
      "epoch": 1.3757705688476563e-05,
      "step": 14426,
      "training_step_time": 17.145637035369873
    },
    {
      "epoch": 1.3758659362792968e-05,
      "model_forward_time": 5.63513708114624,
      "step": 14427
    },
    {
      "epoch": 1.3758659362792968e-05,
      "step": 14427,
      "training_step_time": 17.527438640594482
    },
    {
      "epoch": 1.3759613037109376e-05,
      "model_forward_time": 5.624029874801636,
      "step": 14428
    },
    {
      "epoch": 1.3759613037109376e-05,
      "step": 14428,
      "training_step_time": 17.13162875175476
    },
    {
      "epoch": 1.3760566711425782e-05,
      "model_forward_time": 5.621097087860107,
      "step": 14429
    },
    {
      "epoch": 1.3760566711425782e-05,
      "step": 14429,
      "training_step_time": 17.123603582382202
    },
    {
      "epoch": 1.3761520385742187e-05,
      "grad_norm": 0.27346667647361755,
      "learning_rate": 1e-05,
      "loss": 0.0565,
      "step": 14430
    },
    {
      "epoch": 1.3761520385742187e-05,
      "model_forward_time": 5.6528966426849365,
      "step": 14430
    },
    {
      "dynamics_loss_avg": 0.054469450004398824,
      "epoch": 1.3761520385742187e-05,
      "step": 14430
    },
    {
      "action_loss_avg": 0.0060177361010573804,
      "epoch": 1.3761520385742187e-05,
      "step": 14430
    },
    {
      "epoch": 1.3761520385742187e-05,
      "step": 14430,
      "training_step_time": 17.14849305152893
    },
    {
      "epoch": 1.3762474060058593e-05,
      "model_forward_time": 5.615232467651367,
      "step": 14431
    },
    {
      "epoch": 1.3762474060058593e-05,
      "step": 14431,
      "training_step_time": 17.09654188156128
    },
    {
      "epoch": 1.3763427734375e-05,
      "model_forward_time": 5.606444835662842,
      "step": 14432
    },
    {
      "epoch": 1.3763427734375e-05,
      "step": 14432,
      "training_step_time": 17.04805016517639
    },
    {
      "epoch": 1.3764381408691406e-05,
      "model_forward_time": 5.971193552017212,
      "step": 14433
    },
    {
      "epoch": 1.3764381408691406e-05,
      "step": 14433,
      "training_step_time": 17.407947301864624
    },
    {
      "epoch": 1.3765335083007812e-05,
      "model_forward_time": 5.5899481773376465,
      "step": 14434
    },
    {
      "epoch": 1.3765335083007812e-05,
      "step": 14434,
      "training_step_time": 17.024240732192993
    },
    {
      "epoch": 1.376628875732422e-05,
      "model_forward_time": 5.587097406387329,
      "step": 14435
    },
    {
      "epoch": 1.376628875732422e-05,
      "step": 14435,
      "training_step_time": 17.04920744895935
    },
    {
      "epoch": 1.3767242431640625e-05,
      "model_forward_time": 5.598886013031006,
      "step": 14436
    },
    {
      "epoch": 1.3767242431640625e-05,
      "step": 14436,
      "training_step_time": 17.071266889572144
    },
    {
      "epoch": 1.376819610595703e-05,
      "model_forward_time": 5.610081434249878,
      "step": 14437
    },
    {
      "epoch": 1.376819610595703e-05,
      "step": 14437,
      "training_step_time": 17.090708017349243
    },
    {
      "epoch": 1.3769149780273438e-05,
      "model_forward_time": 5.991600275039673,
      "step": 14438
    },
    {
      "epoch": 1.3769149780273438e-05,
      "step": 14438,
      "training_step_time": 17.46300959587097
    },
    {
      "epoch": 1.3770103454589844e-05,
      "model_forward_time": 5.613351583480835,
      "step": 14439
    },
    {
      "epoch": 1.3770103454589844e-05,
      "step": 14439,
      "training_step_time": 17.10790514945984
    },
    {
      "epoch": 1.377105712890625e-05,
      "grad_norm": 0.6223240494728088,
      "learning_rate": 1e-05,
      "loss": 0.0571,
      "step": 14440
    },
    {
      "epoch": 1.377105712890625e-05,
      "model_forward_time": 5.611227750778198,
      "step": 14440
    },
    {
      "dynamics_loss_avg": 0.050396572798490524,
      "epoch": 1.377105712890625e-05,
      "step": 14440
    },
    {
      "action_loss_avg": 0.005277466052211821,
      "epoch": 1.377105712890625e-05,
      "step": 14440
    },
    {
      "epoch": 1.377105712890625e-05,
      "step": 14440,
      "training_step_time": 17.10175085067749
    },
    {
      "epoch": 1.3772010803222657e-05,
      "model_forward_time": 5.61728572845459,
      "step": 14441
    },
    {
      "epoch": 1.3772010803222657e-05,
      "step": 14441,
      "training_step_time": 17.12172222137451
    },
    {
      "epoch": 1.3772964477539063e-05,
      "model_forward_time": 5.614288091659546,
      "step": 14442
    },
    {
      "epoch": 1.3772964477539063e-05,
      "step": 14442,
      "training_step_time": 17.121108055114746
    },
    {
      "epoch": 1.3773918151855468e-05,
      "model_forward_time": 5.6242594718933105,
      "step": 14443
    },
    {
      "epoch": 1.3773918151855468e-05,
      "step": 14443,
      "training_step_time": 17.130489349365234
    },
    {
      "epoch": 1.3774871826171876e-05,
      "model_forward_time": 5.982120990753174,
      "step": 14444
    },
    {
      "epoch": 1.3774871826171876e-05,
      "step": 14444,
      "training_step_time": 17.476054430007935
    },
    {
      "epoch": 1.3775825500488282e-05,
      "model_forward_time": 5.599607944488525,
      "step": 14445
    },
    {
      "epoch": 1.3775825500488282e-05,
      "step": 14445,
      "training_step_time": 17.072136878967285
    },
    {
      "epoch": 1.3776779174804687e-05,
      "model_forward_time": 5.60150146484375,
      "step": 14446
    },
    {
      "epoch": 1.3776779174804687e-05,
      "step": 14446,
      "training_step_time": 17.054855346679688
    },
    {
      "epoch": 1.3777732849121093e-05,
      "model_forward_time": 5.601849555969238,
      "step": 14447
    },
    {
      "epoch": 1.3777732849121093e-05,
      "step": 14447,
      "training_step_time": 17.03471088409424
    },
    {
      "epoch": 1.37786865234375e-05,
      "model_forward_time": 5.607562065124512,
      "step": 14448
    },
    {
      "epoch": 1.37786865234375e-05,
      "step": 14448,
      "training_step_time": 17.048009872436523
    },
    {
      "epoch": 1.3779640197753906e-05,
      "model_forward_time": 5.965038537979126,
      "step": 14449
    },
    {
      "epoch": 1.3779640197753906e-05,
      "step": 14449,
      "training_step_time": 17.42573857307434
    },
    {
      "epoch": 1.3780593872070312e-05,
      "grad_norm": 1.191508412361145,
      "learning_rate": 1e-05,
      "loss": 0.0768,
      "step": 14450
    },
    {
      "epoch": 1.3780593872070312e-05,
      "model_forward_time": 5.641573190689087,
      "step": 14450
    },
    {
      "dynamics_loss_avg": 0.06620760355144739,
      "epoch": 1.3780593872070312e-05,
      "step": 14450
    },
    {
      "action_loss_avg": 0.00870571241248399,
      "epoch": 1.3780593872070312e-05,
      "step": 14450
    },
    {
      "epoch": 1.3780593872070312e-05,
      "step": 14450,
      "training_step_time": 17.149208545684814
    },
    {
      "epoch": 1.378154754638672e-05,
      "model_forward_time": 5.625269651412964,
      "step": 14451
    },
    {
      "epoch": 1.378154754638672e-05,
      "step": 14451,
      "training_step_time": 17.128295183181763
    },
    {
      "epoch": 1.3782501220703125e-05,
      "model_forward_time": 5.610140085220337,
      "step": 14452
    },
    {
      "epoch": 1.3782501220703125e-05,
      "step": 14452,
      "training_step_time": 17.094481945037842
    },
    {
      "epoch": 1.378345489501953e-05,
      "model_forward_time": 5.607152462005615,
      "step": 14453
    },
    {
      "epoch": 1.378345489501953e-05,
      "step": 14453,
      "training_step_time": 17.096086978912354
    },
    {
      "epoch": 1.3784408569335938e-05,
      "model_forward_time": 5.605909109115601,
      "step": 14454
    },
    {
      "epoch": 1.3784408569335938e-05,
      "step": 14454,
      "training_step_time": 17.07870364189148
    },
    {
      "epoch": 1.3785362243652344e-05,
      "model_forward_time": 5.957200050354004,
      "step": 14455
    },
    {
      "epoch": 1.3785362243652344e-05,
      "step": 14455,
      "training_step_time": 17.40433692932129
    },
    {
      "epoch": 1.378631591796875e-05,
      "model_forward_time": 5.599195718765259,
      "step": 14456
    },
    {
      "epoch": 1.378631591796875e-05,
      "step": 14456,
      "training_step_time": 17.042640924453735
    },
    {
      "epoch": 1.3787269592285157e-05,
      "model_forward_time": 5.5972983837127686,
      "step": 14457
    },
    {
      "epoch": 1.3787269592285157e-05,
      "step": 14457,
      "training_step_time": 17.028569221496582
    },
    {
      "epoch": 1.3788223266601563e-05,
      "model_forward_time": 5.586340427398682,
      "step": 14458
    },
    {
      "epoch": 1.3788223266601563e-05,
      "step": 14458,
      "training_step_time": 17.04154086112976
    },
    {
      "epoch": 1.3789176940917968e-05,
      "model_forward_time": 5.601840019226074,
      "step": 14459
    },
    {
      "epoch": 1.3789176940917968e-05,
      "step": 14459,
      "training_step_time": 17.070327043533325
    },
    {
      "epoch": 1.3790130615234376e-05,
      "grad_norm": 0.46422484517097473,
      "learning_rate": 1e-05,
      "loss": 0.0676,
      "step": 14460
    },
    {
      "epoch": 1.3790130615234376e-05,
      "model_forward_time": 6.001569032669067,
      "step": 14460
    },
    {
      "dynamics_loss_avg": 0.06077877841889858,
      "epoch": 1.3790130615234376e-05,
      "step": 14460
    },
    {
      "action_loss_avg": 0.007583477860316634,
      "epoch": 1.3790130615234376e-05,
      "step": 14460
    },
    {
      "epoch": 1.3790130615234376e-05,
      "step": 14460,
      "training_step_time": 17.499451875686646
    },
    {
      "epoch": 1.3791084289550782e-05,
      "model_forward_time": 5.618043422698975,
      "step": 14461
    },
    {
      "epoch": 1.3791084289550782e-05,
      "step": 14461,
      "training_step_time": 17.11186456680298
    },
    {
      "epoch": 1.3792037963867187e-05,
      "model_forward_time": 5.620271921157837,
      "step": 14462
    },
    {
      "epoch": 1.3792037963867187e-05,
      "step": 14462,
      "training_step_time": 17.114408493041992
    },
    {
      "epoch": 1.3792991638183593e-05,
      "model_forward_time": 5.609077453613281,
      "step": 14463
    },
    {
      "epoch": 1.3792991638183593e-05,
      "step": 14463,
      "training_step_time": 17.099286317825317
    },
    {
      "epoch": 1.37939453125e-05,
      "model_forward_time": 5.613011360168457,
      "step": 14464
    },
    {
      "epoch": 1.37939453125e-05,
      "step": 14464,
      "training_step_time": 17.086057901382446
    },
    {
      "epoch": 1.3794898986816406e-05,
      "model_forward_time": 5.5917932987213135,
      "step": 14465
    },
    {
      "epoch": 1.3794898986816406e-05,
      "step": 14465,
      "training_step_time": 17.037493228912354
    },
    {
      "epoch": 1.3795852661132812e-05,
      "model_forward_time": 5.962436199188232,
      "step": 14466
    },
    {
      "epoch": 1.3795852661132812e-05,
      "step": 14466,
      "training_step_time": 17.407586812973022
    },
    {
      "epoch": 1.379680633544922e-05,
      "model_forward_time": 5.592295408248901,
      "step": 14467
    },
    {
      "epoch": 1.379680633544922e-05,
      "step": 14467,
      "training_step_time": 17.044062614440918
    },
    {
      "epoch": 1.3797760009765625e-05,
      "model_forward_time": 5.599591255187988,
      "step": 14468
    },
    {
      "epoch": 1.3797760009765625e-05,
      "step": 14468,
      "training_step_time": 17.085011959075928
    },
    {
      "epoch": 1.379871368408203e-05,
      "model_forward_time": 5.628355979919434,
      "step": 14469
    },
    {
      "epoch": 1.379871368408203e-05,
      "step": 14469,
      "training_step_time": 17.109049320220947
    },
    {
      "epoch": 1.3799667358398438e-05,
      "grad_norm": 0.3101288378238678,
      "learning_rate": 1e-05,
      "loss": 0.0459,
      "step": 14470
    },
    {
      "epoch": 1.3799667358398438e-05,
      "model_forward_time": 5.604656457901001,
      "step": 14470
    },
    {
      "dynamics_loss_avg": 0.04013978932052851,
      "epoch": 1.3799667358398438e-05,
      "step": 14470
    },
    {
      "action_loss_avg": 0.004923504043836147,
      "epoch": 1.3799667358398438e-05,
      "step": 14470
    },
    {
      "epoch": 1.3799667358398438e-05,
      "step": 14470,
      "training_step_time": 17.092386722564697
    },
    {
      "epoch": 1.3800621032714844e-05,
      "model_forward_time": 6.002874135971069,
      "step": 14471
    },
    {
      "epoch": 1.3800621032714844e-05,
      "step": 14471,
      "training_step_time": 17.477982997894287
    },
    {
      "epoch": 1.380157470703125e-05,
      "model_forward_time": 5.603698015213013,
      "step": 14472
    },
    {
      "epoch": 1.380157470703125e-05,
      "step": 14472,
      "training_step_time": 17.036953687667847
    },
    {
      "epoch": 1.3802528381347657e-05,
      "model_forward_time": 5.601402521133423,
      "step": 14473
    },
    {
      "epoch": 1.3802528381347657e-05,
      "step": 14473,
      "training_step_time": 17.07658052444458
    },
    {
      "epoch": 1.3803482055664063e-05,
      "model_forward_time": 5.6096227169036865,
      "step": 14474
    },
    {
      "epoch": 1.3803482055664063e-05,
      "step": 14474,
      "training_step_time": 17.07530188560486
    },
    {
      "epoch": 1.3804435729980468e-05,
      "model_forward_time": 5.617271184921265,
      "step": 14475
    },
    {
      "epoch": 1.3804435729980468e-05,
      "step": 14475,
      "training_step_time": 17.128130197525024
    },
    {
      "epoch": 1.3805389404296876e-05,
      "model_forward_time": 6.004715442657471,
      "step": 14476
    },
    {
      "epoch": 1.3805389404296876e-05,
      "step": 14476,
      "training_step_time": 17.497341632843018
    },
    {
      "epoch": 1.3806343078613282e-05,
      "model_forward_time": 5.612696647644043,
      "step": 14477
    },
    {
      "epoch": 1.3806343078613282e-05,
      "step": 14477,
      "training_step_time": 17.1050283908844
    },
    {
      "epoch": 1.3807296752929687e-05,
      "model_forward_time": 5.614270448684692,
      "step": 14478
    },
    {
      "epoch": 1.3807296752929687e-05,
      "step": 14478,
      "training_step_time": 17.09407901763916
    },
    {
      "epoch": 1.3808250427246093e-05,
      "model_forward_time": 5.612998008728027,
      "step": 14479
    },
    {
      "epoch": 1.3808250427246093e-05,
      "step": 14479,
      "training_step_time": 17.094773292541504
    },
    {
      "epoch": 1.38092041015625e-05,
      "grad_norm": 0.8037290573120117,
      "learning_rate": 1e-05,
      "loss": 0.0708,
      "step": 14480
    },
    {
      "epoch": 1.38092041015625e-05,
      "model_forward_time": 5.596961975097656,
      "step": 14480
    },
    {
      "dynamics_loss_avg": 0.058881590515375136,
      "epoch": 1.38092041015625e-05,
      "step": 14480
    },
    {
      "action_loss_avg": 0.008747685607522726,
      "epoch": 1.38092041015625e-05,
      "step": 14480
    },
    {
      "epoch": 1.38092041015625e-05,
      "step": 14480,
      "training_step_time": 17.059491395950317
    },
    {
      "epoch": 1.3810157775878906e-05,
      "model_forward_time": 4.264455080032349,
      "step": 14481
    },
    {
      "epoch": 1.3810157775878906e-05,
      "step": 14481,
      "training_step_time": 12.769685745239258
    },
    {
      "epoch": 1.3811111450195312e-05,
      "model_forward_time": 5.975915193557739,
      "step": 14482
    },
    {
      "epoch": 1.3811111450195312e-05,
      "step": 14482,
      "training_step_time": 17.43092966079712
    },
    {
      "epoch": 1.381206512451172e-05,
      "model_forward_time": 5.6092774868011475,
      "step": 14483
    },
    {
      "epoch": 1.381206512451172e-05,
      "step": 14483,
      "training_step_time": 17.10012722015381
    },
    {
      "epoch": 1.3813018798828125e-05,
      "model_forward_time": 5.608270168304443,
      "step": 14484
    },
    {
      "epoch": 1.3813018798828125e-05,
      "step": 14484,
      "training_step_time": 17.1365647315979
    },
    {
      "epoch": 1.381397247314453e-05,
      "model_forward_time": 5.613252878189087,
      "step": 14485
    },
    {
      "epoch": 1.381397247314453e-05,
      "step": 14485,
      "training_step_time": 17.08859157562256
    },
    {
      "epoch": 1.3814926147460938e-05,
      "model_forward_time": 5.595521688461304,
      "step": 14486
    },
    {
      "epoch": 1.3814926147460938e-05,
      "step": 14486,
      "training_step_time": 17.046244859695435
    },
    {
      "epoch": 1.3815879821777344e-05,
      "model_forward_time": 5.582167625427246,
      "step": 14487
    },
    {
      "epoch": 1.3815879821777344e-05,
      "step": 14487,
      "training_step_time": 16.997466564178467
    },
    {
      "epoch": 1.381683349609375e-05,
      "model_forward_time": 5.931545972824097,
      "step": 14488
    },
    {
      "epoch": 1.381683349609375e-05,
      "step": 14488,
      "training_step_time": 17.348163604736328
    },
    {
      "epoch": 1.3817787170410157e-05,
      "model_forward_time": 5.5823493003845215,
      "step": 14489
    },
    {
      "epoch": 1.3817787170410157e-05,
      "step": 14489,
      "training_step_time": 17.022473335266113
    },
    {
      "epoch": 1.3818740844726563e-05,
      "grad_norm": 0.9924879670143127,
      "learning_rate": 1e-05,
      "loss": 0.0728,
      "step": 14490
    },
    {
      "epoch": 1.3818740844726563e-05,
      "model_forward_time": 5.591964960098267,
      "step": 14490
    },
    {
      "dynamics_loss_avg": 0.06416546031832696,
      "epoch": 1.3818740844726563e-05,
      "step": 14490
    },
    {
      "action_loss_avg": 0.008666148758493364,
      "epoch": 1.3818740844726563e-05,
      "step": 14490
    },
    {
      "epoch": 1.3818740844726563e-05,
      "step": 14490,
      "training_step_time": 17.045508861541748
    },
    {
      "epoch": 1.3819694519042968e-05,
      "model_forward_time": 5.607855319976807,
      "step": 14491
    },
    {
      "epoch": 1.3819694519042968e-05,
      "step": 14491,
      "training_step_time": 17.094969987869263
    },
    {
      "epoch": 1.3820648193359376e-05,
      "model_forward_time": 5.620303630828857,
      "step": 14492
    },
    {
      "epoch": 1.3820648193359376e-05,
      "step": 14492,
      "training_step_time": 17.128878831863403
    },
    {
      "epoch": 1.3821601867675782e-05,
      "model_forward_time": 6.007492780685425,
      "step": 14493
    },
    {
      "epoch": 1.3821601867675782e-05,
      "step": 14493,
      "training_step_time": 17.48664402961731
    },
    {
      "epoch": 1.3822555541992187e-05,
      "model_forward_time": 5.60866117477417,
      "step": 14494
    },
    {
      "epoch": 1.3822555541992187e-05,
      "step": 14494,
      "training_step_time": 17.061598539352417
    },
    {
      "epoch": 1.3823509216308593e-05,
      "model_forward_time": 5.597350835800171,
      "step": 14495
    },
    {
      "epoch": 1.3823509216308593e-05,
      "step": 14495,
      "training_step_time": 17.011635303497314
    },
    {
      "epoch": 1.3824462890625e-05,
      "model_forward_time": 5.584353923797607,
      "step": 14496
    },
    {
      "epoch": 1.3824462890625e-05,
      "step": 14496,
      "training_step_time": 17.002090454101562
    },
    {
      "epoch": 1.3825416564941406e-05,
      "model_forward_time": 5.588030815124512,
      "step": 14497
    },
    {
      "epoch": 1.3825416564941406e-05,
      "step": 14497,
      "training_step_time": 17.042637825012207
    },
    {
      "epoch": 1.3826370239257812e-05,
      "model_forward_time": 5.603813648223877,
      "step": 14498
    },
    {
      "epoch": 1.3826370239257812e-05,
      "step": 14498,
      "training_step_time": 17.45626163482666
    },
    {
      "epoch": 1.382732391357422e-05,
      "model_forward_time": 5.620958566665649,
      "step": 14499
    },
    {
      "epoch": 1.382732391357422e-05,
      "step": 14499,
      "training_step_time": 17.11623764038086
    },
    {
      "epoch": 1.3828277587890625e-05,
      "grad_norm": 0.3602399230003357,
      "learning_rate": 1e-05,
      "loss": 0.0501,
      "step": 14500
    },
    {
      "epoch": 1.3828277587890625e-05,
      "model_forward_time": 5.619590997695923,
      "step": 14500
    },
    {
      "dynamics_loss_avg": 0.04831451363861561,
      "epoch": 1.3828277587890625e-05,
      "step": 14500
    },
    {
      "action_loss_avg": 0.005673034279607236,
      "epoch": 1.3828277587890625e-05,
      "step": 14500
    },
    {
      "epoch": 1.3828277587890625e-05,
      "step": 14500,
      "training_step_time": 17.109092712402344
    },
    {
      "epoch": 1.382923126220703e-05,
      "model_forward_time": 5.608968734741211,
      "step": 14501
    },
    {
      "epoch": 1.382923126220703e-05,
      "step": 14501,
      "training_step_time": 17.08025622367859
    },
    {
      "epoch": 1.3830184936523438e-05,
      "model_forward_time": 5.602372169494629,
      "step": 14502
    },
    {
      "epoch": 1.3830184936523438e-05,
      "step": 14502,
      "training_step_time": 17.041303873062134
    },
    {
      "epoch": 1.3831138610839844e-05,
      "model_forward_time": 5.588974475860596,
      "step": 14503
    },
    {
      "epoch": 1.3831138610839844e-05,
      "step": 14503,
      "training_step_time": 17.044556379318237
    },
    {
      "epoch": 1.383209228515625e-05,
      "model_forward_time": 5.992341756820679,
      "step": 14504
    },
    {
      "epoch": 1.383209228515625e-05,
      "step": 14504,
      "training_step_time": 17.45456838607788
    },
    {
      "epoch": 1.3833045959472657e-05,
      "model_forward_time": 5.605394601821899,
      "step": 14505
    },
    {
      "epoch": 1.3833045959472657e-05,
      "step": 14505,
      "training_step_time": 17.07671308517456
    },
    {
      "epoch": 1.3833999633789063e-05,
      "model_forward_time": 5.606762886047363,
      "step": 14506
    },
    {
      "epoch": 1.3833999633789063e-05,
      "step": 14506,
      "training_step_time": 17.051865816116333
    },
    {
      "epoch": 1.3834953308105468e-05,
      "model_forward_time": 5.592841148376465,
      "step": 14507
    },
    {
      "epoch": 1.3834953308105468e-05,
      "step": 14507,
      "training_step_time": 17.010048389434814
    },
    {
      "epoch": 1.3835906982421874e-05,
      "model_forward_time": 5.596561908721924,
      "step": 14508
    },
    {
      "epoch": 1.3835906982421874e-05,
      "step": 14508,
      "training_step_time": 17.034705877304077
    },
    {
      "epoch": 1.3836860656738282e-05,
      "model_forward_time": 5.9703123569488525,
      "step": 14509
    },
    {
      "epoch": 1.3836860656738282e-05,
      "step": 14509,
      "training_step_time": 17.46219515800476
    },
    {
      "epoch": 1.3837814331054687e-05,
      "grad_norm": 0.9072951674461365,
      "learning_rate": 1e-05,
      "loss": 0.0628,
      "step": 14510
    },
    {
      "epoch": 1.3837814331054687e-05,
      "model_forward_time": 5.627366781234741,
      "step": 14510
    },
    {
      "dynamics_loss_avg": 0.05092886369675398,
      "epoch": 1.3837814331054687e-05,
      "step": 14510
    },
    {
      "action_loss_avg": 0.008481648331508041,
      "epoch": 1.3837814331054687e-05,
      "step": 14510
    },
    {
      "epoch": 1.3837814331054687e-05,
      "step": 14510,
      "training_step_time": 17.157078981399536
    },
    {
      "epoch": 1.3838768005371093e-05,
      "model_forward_time": 5.634909391403198,
      "step": 14511
    },
    {
      "epoch": 1.3838768005371093e-05,
      "step": 14511,
      "training_step_time": 17.130245447158813
    },
    {
      "epoch": 1.38397216796875e-05,
      "model_forward_time": 5.594488620758057,
      "step": 14512
    },
    {
      "epoch": 1.38397216796875e-05,
      "step": 14512,
      "training_step_time": 17.035277605056763
    },
    {
      "epoch": 1.3840675354003906e-05,
      "model_forward_time": 5.561372995376587,
      "step": 14513
    },
    {
      "epoch": 1.3840675354003906e-05,
      "step": 14513,
      "training_step_time": 16.980034589767456
    },
    {
      "epoch": 1.3841629028320312e-05,
      "model_forward_time": 5.559411287307739,
      "step": 14514
    },
    {
      "epoch": 1.3841629028320312e-05,
      "step": 14514,
      "training_step_time": 16.968007564544678
    },
    {
      "epoch": 1.384258270263672e-05,
      "model_forward_time": 5.939455509185791,
      "step": 14515
    },
    {
      "epoch": 1.384258270263672e-05,
      "step": 14515,
      "training_step_time": 17.359963417053223
    },
    {
      "epoch": 1.3843536376953125e-05,
      "model_forward_time": 5.572376728057861,
      "step": 14516
    },
    {
      "epoch": 1.3843536376953125e-05,
      "step": 14516,
      "training_step_time": 17.008074283599854
    },
    {
      "epoch": 1.384449005126953e-05,
      "model_forward_time": 5.566485404968262,
      "step": 14517
    },
    {
      "epoch": 1.384449005126953e-05,
      "step": 14517,
      "training_step_time": 16.973288536071777
    },
    {
      "epoch": 1.3845443725585938e-05,
      "model_forward_time": 5.576831817626953,
      "step": 14518
    },
    {
      "epoch": 1.3845443725585938e-05,
      "step": 14518,
      "training_step_time": 17.001004457473755
    },
    {
      "epoch": 1.3846397399902344e-05,
      "model_forward_time": 5.587635278701782,
      "step": 14519
    },
    {
      "epoch": 1.3846397399902344e-05,
      "step": 14519,
      "training_step_time": 17.023754119873047
    },
    {
      "epoch": 1.384735107421875e-05,
      "grad_norm": 1.4532378911972046,
      "learning_rate": 1e-05,
      "loss": 0.0699,
      "step": 14520
    },
    {
      "epoch": 1.384735107421875e-05,
      "model_forward_time": 5.579663515090942,
      "step": 14520
    },
    {
      "dynamics_loss_avg": 0.06655518598854542,
      "epoch": 1.384735107421875e-05,
      "step": 14520
    },
    {
      "action_loss_avg": 0.010722970846109093,
      "epoch": 1.384735107421875e-05,
      "step": 14520
    },
    {
      "epoch": 1.384735107421875e-05,
      "step": 14520,
      "training_step_time": 17.386236429214478
    },
    {
      "epoch": 1.3848304748535157e-05,
      "model_forward_time": 5.590770721435547,
      "step": 14521
    },
    {
      "epoch": 1.3848304748535157e-05,
      "step": 14521,
      "training_step_time": 17.02679467201233
    },
    {
      "epoch": 1.3849258422851563e-05,
      "model_forward_time": 5.596840858459473,
      "step": 14522
    },
    {
      "epoch": 1.3849258422851563e-05,
      "step": 14522,
      "training_step_time": 17.06988835334778
    },
    {
      "epoch": 1.3850212097167968e-05,
      "model_forward_time": 5.6198413372039795,
      "step": 14523
    },
    {
      "epoch": 1.3850212097167968e-05,
      "step": 14523,
      "training_step_time": 17.138429164886475
    },
    {
      "epoch": 1.3851165771484374e-05,
      "model_forward_time": 5.6193201541900635,
      "step": 14524
    },
    {
      "epoch": 1.3851165771484374e-05,
      "step": 14524,
      "training_step_time": 17.14166569709778
    },
    {
      "epoch": 1.3852119445800782e-05,
      "model_forward_time": 5.626946210861206,
      "step": 14525
    },
    {
      "epoch": 1.3852119445800782e-05,
      "step": 14525,
      "training_step_time": 17.144945859909058
    },
    {
      "epoch": 1.3853073120117187e-05,
      "model_forward_time": 5.989690780639648,
      "step": 14526
    },
    {
      "epoch": 1.3853073120117187e-05,
      "step": 14526,
      "training_step_time": 17.493253707885742
    },
    {
      "epoch": 1.3854026794433593e-05,
      "model_forward_time": 5.619020462036133,
      "step": 14527
    },
    {
      "epoch": 1.3854026794433593e-05,
      "step": 14527,
      "training_step_time": 17.113730907440186
    },
    {
      "epoch": 1.385498046875e-05,
      "model_forward_time": 5.610664367675781,
      "step": 14528
    },
    {
      "epoch": 1.385498046875e-05,
      "step": 14528,
      "training_step_time": 17.10729718208313
    },
    {
      "epoch": 1.3855934143066406e-05,
      "model_forward_time": 5.624597787857056,
      "step": 14529
    },
    {
      "epoch": 1.3855934143066406e-05,
      "step": 14529,
      "training_step_time": 17.119547605514526
    },
    {
      "epoch": 1.3856887817382812e-05,
      "grad_norm": 0.49164289236068726,
      "learning_rate": 1e-05,
      "loss": 0.0726,
      "step": 14530
    },
    {
      "epoch": 1.3856887817382812e-05,
      "model_forward_time": 5.609531402587891,
      "step": 14530
    },
    {
      "dynamics_loss_avg": 0.056418031454086304,
      "epoch": 1.3856887817382812e-05,
      "step": 14530
    },
    {
      "action_loss_avg": 0.00951266842894256,
      "epoch": 1.3856887817382812e-05,
      "step": 14530
    },
    {
      "epoch": 1.3856887817382812e-05,
      "step": 14530,
      "training_step_time": 17.072499752044678
    },
    {
      "epoch": 1.385784149169922e-05,
      "model_forward_time": 5.591487407684326,
      "step": 14531
    },
    {
      "epoch": 1.385784149169922e-05,
      "step": 14531,
      "training_step_time": 17.41033148765564
    },
    {
      "epoch": 1.3858795166015625e-05,
      "model_forward_time": 5.596256732940674,
      "step": 14532
    },
    {
      "epoch": 1.3858795166015625e-05,
      "step": 14532,
      "training_step_time": 17.006662607192993
    },
    {
      "epoch": 1.385974884033203e-05,
      "model_forward_time": 5.584409236907959,
      "step": 14533
    },
    {
      "epoch": 1.385974884033203e-05,
      "step": 14533,
      "training_step_time": 17.02301526069641
    },
    {
      "epoch": 1.3860702514648438e-05,
      "model_forward_time": 5.595524072647095,
      "step": 14534
    },
    {
      "epoch": 1.3860702514648438e-05,
      "step": 14534,
      "training_step_time": 17.06181001663208
    },
    {
      "epoch": 1.3861656188964844e-05,
      "model_forward_time": 5.605121374130249,
      "step": 14535
    },
    {
      "epoch": 1.3861656188964844e-05,
      "step": 14535,
      "training_step_time": 17.10393190383911
    },
    {
      "epoch": 1.386260986328125e-05,
      "model_forward_time": 5.606777191162109,
      "step": 14536
    },
    {
      "epoch": 1.386260986328125e-05,
      "step": 14536,
      "training_step_time": 17.093716621398926
    },
    {
      "epoch": 1.3863563537597657e-05,
      "model_forward_time": 5.970335483551025,
      "step": 14537
    },
    {
      "epoch": 1.3863563537597657e-05,
      "step": 14537,
      "training_step_time": 17.45064377784729
    },
    {
      "epoch": 1.3864517211914063e-05,
      "model_forward_time": 5.615697622299194,
      "step": 14538
    },
    {
      "epoch": 1.3864517211914063e-05,
      "step": 14538,
      "training_step_time": 17.092846393585205
    },
    {
      "epoch": 1.3865470886230468e-05,
      "model_forward_time": 5.609581470489502,
      "step": 14539
    },
    {
      "epoch": 1.3865470886230468e-05,
      "step": 14539,
      "training_step_time": 17.08667016029358
    },
    {
      "epoch": 1.3866424560546874e-05,
      "grad_norm": 0.5372170209884644,
      "learning_rate": 1e-05,
      "loss": 0.0561,
      "step": 14540
    },
    {
      "epoch": 1.3866424560546874e-05,
      "model_forward_time": 5.602313756942749,
      "step": 14540
    },
    {
      "dynamics_loss_avg": 0.05182918161153793,
      "epoch": 1.3866424560546874e-05,
      "step": 14540
    },
    {
      "action_loss_avg": 0.008209105324931442,
      "epoch": 1.3866424560546874e-05,
      "step": 14540
    },
    {
      "epoch": 1.3866424560546874e-05,
      "step": 14540,
      "training_step_time": 17.069908618927002
    },
    {
      "epoch": 1.3867378234863282e-05,
      "model_forward_time": 5.604528903961182,
      "step": 14541
    },
    {
      "epoch": 1.3867378234863282e-05,
      "step": 14541,
      "training_step_time": 17.065239191055298
    },
    {
      "epoch": 1.3868331909179687e-05,
      "model_forward_time": 5.969002962112427,
      "step": 14542
    },
    {
      "epoch": 1.3868331909179687e-05,
      "step": 14542,
      "training_step_time": 17.459033489227295
    },
    {
      "epoch": 1.3869285583496093e-05,
      "model_forward_time": 5.613245964050293,
      "step": 14543
    },
    {
      "epoch": 1.3869285583496093e-05,
      "step": 14543,
      "training_step_time": 17.08794331550598
    },
    {
      "epoch": 1.38702392578125e-05,
      "model_forward_time": 5.604553937911987,
      "step": 14544
    },
    {
      "epoch": 1.38702392578125e-05,
      "step": 14544,
      "training_step_time": 17.0910484790802
    },
    {
      "epoch": 1.3871192932128906e-05,
      "model_forward_time": 5.611563682556152,
      "step": 14545
    },
    {
      "epoch": 1.3871192932128906e-05,
      "step": 14545,
      "training_step_time": 17.11409330368042
    },
    {
      "epoch": 1.3872146606445312e-05,
      "model_forward_time": 5.611971139907837,
      "step": 14546
    },
    {
      "epoch": 1.3872146606445312e-05,
      "step": 14546,
      "training_step_time": 17.114766120910645
    },
    {
      "epoch": 1.387310028076172e-05,
      "model_forward_time": 5.617593765258789,
      "step": 14547
    },
    {
      "epoch": 1.387310028076172e-05,
      "step": 14547,
      "training_step_time": 17.120063304901123
    },
    {
      "epoch": 1.3874053955078125e-05,
      "model_forward_time": 5.991612434387207,
      "step": 14548
    },
    {
      "epoch": 1.3874053955078125e-05,
      "step": 14548,
      "training_step_time": 17.469544887542725
    },
    {
      "epoch": 1.387500762939453e-05,
      "model_forward_time": 5.612976312637329,
      "step": 14549
    },
    {
      "epoch": 1.387500762939453e-05,
      "step": 14549,
      "training_step_time": 17.102136373519897
    },
    {
      "epoch": 1.3875961303710938e-05,
      "grad_norm": 0.3848918676376343,
      "learning_rate": 1e-05,
      "loss": 0.0618,
      "step": 14550
    },
    {
      "epoch": 1.3875961303710938e-05,
      "model_forward_time": 5.610583066940308,
      "step": 14550
    },
    {
      "dynamics_loss_avg": 0.051611901447176936,
      "epoch": 1.3875961303710938e-05,
      "step": 14550
    },
    {
      "action_loss_avg": 0.007432133029215038,
      "epoch": 1.3875961303710938e-05,
      "step": 14550
    },
    {
      "epoch": 1.3875961303710938e-05,
      "step": 14550,
      "training_step_time": 17.102909803390503
    },
    {
      "epoch": 1.3876914978027344e-05,
      "model_forward_time": 5.61424708366394,
      "step": 14551
    },
    {
      "epoch": 1.3876914978027344e-05,
      "step": 14551,
      "training_step_time": 17.11095142364502
    },
    {
      "epoch": 1.387786865234375e-05,
      "model_forward_time": 5.6249706745147705,
      "step": 14552
    },
    {
      "epoch": 1.387786865234375e-05,
      "step": 14552,
      "training_step_time": 17.13306713104248
    },
    {
      "epoch": 1.3878822326660157e-05,
      "model_forward_time": 5.619765758514404,
      "step": 14553
    },
    {
      "epoch": 1.3878822326660157e-05,
      "step": 14553,
      "training_step_time": 17.492619037628174
    },
    {
      "epoch": 1.3879776000976563e-05,
      "model_forward_time": 5.614601373672485,
      "step": 14554
    },
    {
      "epoch": 1.3879776000976563e-05,
      "step": 14554,
      "training_step_time": 17.098549365997314
    },
    {
      "epoch": 1.3880729675292968e-05,
      "model_forward_time": 5.6148154735565186,
      "step": 14555
    },
    {
      "epoch": 1.3880729675292968e-05,
      "step": 14555,
      "training_step_time": 17.103423357009888
    },
    {
      "epoch": 1.3881683349609374e-05,
      "model_forward_time": 5.62788200378418,
      "step": 14556
    },
    {
      "epoch": 1.3881683349609374e-05,
      "step": 14556,
      "training_step_time": 17.112937211990356
    },
    {
      "epoch": 1.3882637023925782e-05,
      "model_forward_time": 5.609323978424072,
      "step": 14557
    },
    {
      "epoch": 1.3882637023925782e-05,
      "step": 14557,
      "training_step_time": 17.072132110595703
    },
    {
      "epoch": 1.3883590698242187e-05,
      "model_forward_time": 5.590528249740601,
      "step": 14558
    },
    {
      "epoch": 1.3883590698242187e-05,
      "step": 14558,
      "training_step_time": 17.047974586486816
    },
    {
      "epoch": 1.3884544372558593e-05,
      "model_forward_time": 6.002676010131836,
      "step": 14559
    },
    {
      "epoch": 1.3884544372558593e-05,
      "step": 14559,
      "training_step_time": 17.503780841827393
    },
    {
      "epoch": 1.3885498046875e-05,
      "grad_norm": 0.766291081905365,
      "learning_rate": 1e-05,
      "loss": 0.0609,
      "step": 14560
    },
    {
      "epoch": 1.3885498046875e-05,
      "model_forward_time": 5.6249918937683105,
      "step": 14560
    },
    {
      "dynamics_loss_avg": 0.055052966251969336,
      "epoch": 1.3885498046875e-05,
      "step": 14560
    },
    {
      "action_loss_avg": 0.00764179506804794,
      "epoch": 1.3885498046875e-05,
      "step": 14560
    },
    {
      "epoch": 1.3885498046875e-05,
      "step": 14560,
      "training_step_time": 17.119524478912354
    },
    {
      "epoch": 1.3886451721191406e-05,
      "model_forward_time": 5.598146438598633,
      "step": 14561
    },
    {
      "epoch": 1.3886451721191406e-05,
      "step": 14561,
      "training_step_time": 17.069989442825317
    },
    {
      "epoch": 1.3887405395507812e-05,
      "model_forward_time": 5.587348699569702,
      "step": 14562
    },
    {
      "epoch": 1.3887405395507812e-05,
      "step": 14562,
      "training_step_time": 17.025683641433716
    },
    {
      "epoch": 1.388835906982422e-05,
      "model_forward_time": 5.591288328170776,
      "step": 14563
    },
    {
      "epoch": 1.388835906982422e-05,
      "step": 14563,
      "training_step_time": 17.005841970443726
    },
    {
      "epoch": 1.3889312744140625e-05,
      "model_forward_time": 5.597062826156616,
      "step": 14564
    },
    {
      "epoch": 1.3889312744140625e-05,
      "step": 14564,
      "training_step_time": 17.444549798965454
    },
    {
      "epoch": 1.389026641845703e-05,
      "model_forward_time": 5.614226341247559,
      "step": 14565
    },
    {
      "epoch": 1.389026641845703e-05,
      "step": 14565,
      "training_step_time": 17.10811710357666
    },
    {
      "epoch": 1.3891220092773438e-05,
      "model_forward_time": 5.61151123046875,
      "step": 14566
    },
    {
      "epoch": 1.3891220092773438e-05,
      "step": 14566,
      "training_step_time": 17.10317826271057
    },
    {
      "epoch": 1.3892173767089844e-05,
      "model_forward_time": 5.614041090011597,
      "step": 14567
    },
    {
      "epoch": 1.3892173767089844e-05,
      "step": 14567,
      "training_step_time": 17.127028465270996
    },
    {
      "epoch": 1.389312744140625e-05,
      "model_forward_time": 5.62172794342041,
      "step": 14568
    },
    {
      "epoch": 1.389312744140625e-05,
      "step": 14568,
      "training_step_time": 17.1363422870636
    },
    {
      "epoch": 1.3894081115722657e-05,
      "model_forward_time": 5.607887029647827,
      "step": 14569
    },
    {
      "epoch": 1.3894081115722657e-05,
      "step": 14569,
      "training_step_time": 17.12572693824768
    },
    {
      "epoch": 1.3895034790039063e-05,
      "grad_norm": 1.0929653644561768,
      "learning_rate": 1e-05,
      "loss": 0.0655,
      "step": 14570
    },
    {
      "epoch": 1.3895034790039063e-05,
      "model_forward_time": 6.011485815048218,
      "step": 14570
    },
    {
      "dynamics_loss_avg": 0.05791613757610321,
      "epoch": 1.3895034790039063e-05,
      "step": 14570
    },
    {
      "action_loss_avg": 0.009031845058780163,
      "epoch": 1.3895034790039063e-05,
      "step": 14570
    },
    {
      "epoch": 1.3895034790039063e-05,
      "step": 14570,
      "training_step_time": 17.489500522613525
    },
    {
      "epoch": 1.3895988464355468e-05,
      "model_forward_time": 5.598470449447632,
      "step": 14571
    },
    {
      "epoch": 1.3895988464355468e-05,
      "step": 14571,
      "training_step_time": 17.030503749847412
    },
    {
      "epoch": 1.3896942138671874e-05,
      "model_forward_time": 5.57230544090271,
      "step": 14572
    },
    {
      "epoch": 1.3896942138671874e-05,
      "step": 14572,
      "training_step_time": 16.981434106826782
    },
    {
      "epoch": 1.3897895812988282e-05,
      "model_forward_time": 5.566832542419434,
      "step": 14573
    },
    {
      "epoch": 1.3897895812988282e-05,
      "step": 14573,
      "training_step_time": 17.007910013198853
    },
    {
      "epoch": 1.3898849487304687e-05,
      "model_forward_time": 5.569134473800659,
      "step": 14574
    },
    {
      "epoch": 1.3898849487304687e-05,
      "step": 14574,
      "training_step_time": 16.994390726089478
    },
    {
      "epoch": 1.3899803161621093e-05,
      "model_forward_time": 5.946849822998047,
      "step": 14575
    },
    {
      "epoch": 1.3899803161621093e-05,
      "step": 14575,
      "training_step_time": 17.396948099136353
    },
    {
      "epoch": 1.39007568359375e-05,
      "model_forward_time": 5.591838598251343,
      "step": 14576
    },
    {
      "epoch": 1.39007568359375e-05,
      "step": 14576,
      "training_step_time": 17.04945707321167
    },
    {
      "epoch": 1.3901710510253906e-05,
      "model_forward_time": 5.609295129776001,
      "step": 14577
    },
    {
      "epoch": 1.3901710510253906e-05,
      "step": 14577,
      "training_step_time": 17.10053539276123
    },
    {
      "epoch": 1.3902664184570312e-05,
      "model_forward_time": 5.687040567398071,
      "step": 14578
    },
    {
      "epoch": 1.3902664184570312e-05,
      "step": 14578,
      "training_step_time": 17.212199687957764
    },
    {
      "epoch": 1.390361785888672e-05,
      "model_forward_time": 5.635898590087891,
      "step": 14579
    },
    {
      "epoch": 1.390361785888672e-05,
      "step": 14579,
      "training_step_time": 17.151451110839844
    },
    {
      "epoch": 1.3904571533203125e-05,
      "grad_norm": 1.3303958177566528,
      "learning_rate": 1e-05,
      "loss": 0.0705,
      "step": 14580
    },
    {
      "epoch": 1.3904571533203125e-05,
      "model_forward_time": 5.620906114578247,
      "step": 14580
    },
    {
      "dynamics_loss_avg": 0.0601924017071724,
      "epoch": 1.3904571533203125e-05,
      "step": 14580
    },
    {
      "action_loss_avg": 0.007553044753149152,
      "epoch": 1.3904571533203125e-05,
      "step": 14580
    },
    {
      "epoch": 1.3904571533203125e-05,
      "step": 14580,
      "training_step_time": 17.135311603546143
    },
    {
      "epoch": 1.390552520751953e-05,
      "model_forward_time": 5.996216535568237,
      "step": 14581
    },
    {
      "epoch": 1.390552520751953e-05,
      "step": 14581,
      "training_step_time": 17.477851629257202
    },
    {
      "epoch": 1.3906478881835938e-05,
      "model_forward_time": 5.61301589012146,
      "step": 14582
    },
    {
      "epoch": 1.3906478881835938e-05,
      "step": 14582,
      "training_step_time": 17.092825651168823
    },
    {
      "epoch": 1.3907432556152344e-05,
      "model_forward_time": 5.60043478012085,
      "step": 14583
    },
    {
      "epoch": 1.3907432556152344e-05,
      "step": 14583,
      "training_step_time": 17.087137460708618
    },
    {
      "epoch": 1.390838623046875e-05,
      "model_forward_time": 5.598740577697754,
      "step": 14584
    },
    {
      "epoch": 1.390838623046875e-05,
      "step": 14584,
      "training_step_time": 17.03841280937195
    },
    {
      "epoch": 1.3909339904785157e-05,
      "model_forward_time": 5.607097387313843,
      "step": 14585
    },
    {
      "epoch": 1.3909339904785157e-05,
      "step": 14585,
      "training_step_time": 17.058351039886475
    },
    {
      "epoch": 1.3910293579101563e-05,
      "model_forward_time": 5.583685398101807,
      "step": 14586
    },
    {
      "epoch": 1.3910293579101563e-05,
      "step": 14586,
      "training_step_time": 17.389928579330444
    },
    {
      "epoch": 1.3911247253417969e-05,
      "model_forward_time": 5.587226152420044,
      "step": 14587
    },
    {
      "epoch": 1.3911247253417969e-05,
      "step": 14587,
      "training_step_time": 17.046653270721436
    },
    {
      "epoch": 1.3912200927734374e-05,
      "model_forward_time": 5.604273319244385,
      "step": 14588
    },
    {
      "epoch": 1.3912200927734374e-05,
      "step": 14588,
      "training_step_time": 17.07408356666565
    },
    {
      "epoch": 1.3913154602050782e-05,
      "model_forward_time": 5.602107048034668,
      "step": 14589
    },
    {
      "epoch": 1.3913154602050782e-05,
      "step": 14589,
      "training_step_time": 17.09594964981079
    },
    {
      "epoch": 1.3914108276367187e-05,
      "grad_norm": 0.8725435733795166,
      "learning_rate": 1e-05,
      "loss": 0.0696,
      "step": 14590
    },
    {
      "epoch": 1.3914108276367187e-05,
      "model_forward_time": 5.615621089935303,
      "step": 14590
    },
    {
      "dynamics_loss_avg": 0.06082397401332855,
      "epoch": 1.3914108276367187e-05,
      "step": 14590
    },
    {
      "action_loss_avg": 0.006396203814074397,
      "epoch": 1.3914108276367187e-05,
      "step": 14590
    },
    {
      "epoch": 1.3914108276367187e-05,
      "step": 14590,
      "training_step_time": 17.134334325790405
    },
    {
      "epoch": 1.3915061950683593e-05,
      "model_forward_time": 5.614372253417969,
      "step": 14591
    },
    {
      "epoch": 1.3915061950683593e-05,
      "step": 14591,
      "training_step_time": 17.096216917037964
    },
    {
      "epoch": 1.3916015625e-05,
      "model_forward_time": 5.979205131530762,
      "step": 14592
    },
    {
      "epoch": 1.3916015625e-05,
      "step": 14592,
      "training_step_time": 17.41098952293396
    },
    {
      "epoch": 1.3916969299316406e-05,
      "model_forward_time": 5.586626768112183,
      "step": 14593
    },
    {
      "epoch": 1.3916969299316406e-05,
      "step": 14593,
      "training_step_time": 17.024488925933838
    },
    {
      "epoch": 1.3917922973632812e-05,
      "model_forward_time": 5.576383829116821,
      "step": 14594
    },
    {
      "epoch": 1.3917922973632812e-05,
      "step": 14594,
      "training_step_time": 17.00090456008911
    },
    {
      "epoch": 1.391887664794922e-05,
      "model_forward_time": 5.583773851394653,
      "step": 14595
    },
    {
      "epoch": 1.391887664794922e-05,
      "step": 14595,
      "training_step_time": 17.015561819076538
    },
    {
      "epoch": 1.3919830322265625e-05,
      "model_forward_time": 5.582247734069824,
      "step": 14596
    },
    {
      "epoch": 1.3919830322265625e-05,
      "step": 14596,
      "training_step_time": 17.038949251174927
    },
    {
      "epoch": 1.392078399658203e-05,
      "model_forward_time": 5.5947229862213135,
      "step": 14597
    },
    {
      "epoch": 1.392078399658203e-05,
      "step": 14597,
      "training_step_time": 17.410792350769043
    },
    {
      "epoch": 1.3921737670898438e-05,
      "model_forward_time": 5.6006317138671875,
      "step": 14598
    },
    {
      "epoch": 1.3921737670898438e-05,
      "step": 14598,
      "training_step_time": 17.035000562667847
    },
    {
      "epoch": 1.3922691345214844e-05,
      "model_forward_time": 5.605788707733154,
      "step": 14599
    },
    {
      "epoch": 1.3922691345214844e-05,
      "step": 14599,
      "training_step_time": 17.06984257698059
    },
    {
      "epoch": 1.392364501953125e-05,
      "grad_norm": 1.268355131149292,
      "learning_rate": 1e-05,
      "loss": 0.0551,
      "step": 14600
    },
    {
      "epoch": 1.392364501953125e-05,
      "model_forward_time": 5.60108494758606,
      "step": 14600
    },
    {
      "dynamics_loss_avg": 0.05162136536091566,
      "epoch": 1.392364501953125e-05,
      "step": 14600
    },
    {
      "action_loss_avg": 0.005431759334169328,
      "epoch": 1.392364501953125e-05,
      "step": 14600
    },
    {
      "epoch": 1.392364501953125e-05,
      "step": 14600,
      "training_step_time": 17.084763050079346
    },
    {
      "epoch": 1.3924598693847657e-05,
      "model_forward_time": 5.618729114532471,
      "step": 14601
    },
    {
      "epoch": 1.3924598693847657e-05,
      "step": 14601,
      "training_step_time": 17.09342360496521
    },
    {
      "epoch": 1.3925552368164063e-05,
      "model_forward_time": 5.600336313247681,
      "step": 14602
    },
    {
      "epoch": 1.3925552368164063e-05,
      "step": 14602,
      "training_step_time": 17.03904938697815
    },
    {
      "epoch": 1.3926506042480469e-05,
      "model_forward_time": 5.963829040527344,
      "step": 14603
    },
    {
      "epoch": 1.3926506042480469e-05,
      "step": 14603,
      "training_step_time": 17.394434690475464
    },
    {
      "epoch": 1.3927459716796874e-05,
      "model_forward_time": 5.5791215896606445,
      "step": 14604
    },
    {
      "epoch": 1.3927459716796874e-05,
      "step": 14604,
      "training_step_time": 17.009310483932495
    },
    {
      "epoch": 1.3928413391113282e-05,
      "model_forward_time": 5.588031768798828,
      "step": 14605
    },
    {
      "epoch": 1.3928413391113282e-05,
      "step": 14605,
      "training_step_time": 17.004706621170044
    },
    {
      "epoch": 1.3929367065429687e-05,
      "model_forward_time": 5.591036081314087,
      "step": 14606
    },
    {
      "epoch": 1.3929367065429687e-05,
      "step": 14606,
      "training_step_time": 17.010995864868164
    },
    {
      "epoch": 1.3930320739746093e-05,
      "model_forward_time": 5.627073526382446,
      "step": 14607
    },
    {
      "epoch": 1.3930320739746093e-05,
      "step": 14607,
      "training_step_time": 17.116658210754395
    },
    {
      "epoch": 1.39312744140625e-05,
      "model_forward_time": 6.145134210586548,
      "step": 14608
    },
    {
      "epoch": 1.39312744140625e-05,
      "step": 14608,
      "training_step_time": 17.63736653327942
    },
    {
      "epoch": 1.3932228088378906e-05,
      "model_forward_time": 5.655540704727173,
      "step": 14609
    },
    {
      "epoch": 1.3932228088378906e-05,
      "step": 14609,
      "training_step_time": 17.14007329940796
    },
    {
      "epoch": 1.3933181762695312e-05,
      "grad_norm": 0.5735297203063965,
      "learning_rate": 1e-05,
      "loss": 0.0552,
      "step": 14610
    },
    {
      "epoch": 1.3933181762695312e-05,
      "model_forward_time": 5.609492063522339,
      "step": 14610
    },
    {
      "dynamics_loss_avg": 0.047423635795712474,
      "epoch": 1.3933181762695312e-05,
      "step": 14610
    },
    {
      "action_loss_avg": 0.0053305378649383785,
      "epoch": 1.3933181762695312e-05,
      "step": 14610
    },
    {
      "epoch": 1.3933181762695312e-05,
      "step": 14610,
      "training_step_time": 17.08025598526001
    },
    {
      "epoch": 1.393413543701172e-05,
      "model_forward_time": 5.609636068344116,
      "step": 14611
    },
    {
      "epoch": 1.393413543701172e-05,
      "step": 14611,
      "training_step_time": 17.093810558319092
    },
    {
      "epoch": 1.3935089111328125e-05,
      "model_forward_time": 5.621847629547119,
      "step": 14612
    },
    {
      "epoch": 1.3935089111328125e-05,
      "step": 14612,
      "training_step_time": 17.116459608078003
    },
    {
      "epoch": 1.393604278564453e-05,
      "model_forward_time": 5.62337064743042,
      "step": 14613
    },
    {
      "epoch": 1.393604278564453e-05,
      "step": 14613,
      "training_step_time": 17.134329557418823
    },
    {
      "epoch": 1.3936996459960938e-05,
      "model_forward_time": 6.030547857284546,
      "step": 14614
    },
    {
      "epoch": 1.3936996459960938e-05,
      "step": 14614,
      "training_step_time": 17.52785587310791
    },
    {
      "epoch": 1.3937950134277344e-05,
      "model_forward_time": 5.615459203720093,
      "step": 14615
    },
    {
      "epoch": 1.3937950134277344e-05,
      "step": 14615,
      "training_step_time": 17.11636734008789
    },
    {
      "epoch": 1.393890380859375e-05,
      "model_forward_time": 5.613476514816284,
      "step": 14616
    },
    {
      "epoch": 1.393890380859375e-05,
      "step": 14616,
      "training_step_time": 17.118021965026855
    },
    {
      "epoch": 1.3939857482910157e-05,
      "model_forward_time": 5.624964475631714,
      "step": 14617
    },
    {
      "epoch": 1.3939857482910157e-05,
      "step": 14617,
      "training_step_time": 17.122766733169556
    },
    {
      "epoch": 1.3940811157226563e-05,
      "model_forward_time": 5.6186065673828125,
      "step": 14618
    },
    {
      "epoch": 1.3940811157226563e-05,
      "step": 14618,
      "training_step_time": 17.0908842086792
    },
    {
      "epoch": 1.3941764831542969e-05,
      "model_forward_time": 5.606299638748169,
      "step": 14619
    },
    {
      "epoch": 1.3941764831542969e-05,
      "step": 14619,
      "training_step_time": 17.489309549331665
    },
    {
      "epoch": 1.3942718505859374e-05,
      "grad_norm": 0.7777065634727478,
      "learning_rate": 1e-05,
      "loss": 0.0645,
      "step": 14620
    },
    {
      "epoch": 1.3942718505859374e-05,
      "model_forward_time": 5.586811304092407,
      "step": 14620
    },
    {
      "dynamics_loss_avg": 0.05629484765231609,
      "epoch": 1.3942718505859374e-05,
      "step": 14620
    },
    {
      "action_loss_avg": 0.007511760666966438,
      "epoch": 1.3942718505859374e-05,
      "step": 14620
    },
    {
      "epoch": 1.3942718505859374e-05,
      "step": 14620,
      "training_step_time": 17.035545825958252
    },
    {
      "epoch": 1.3943672180175782e-05,
      "model_forward_time": 5.596232652664185,
      "step": 14621
    },
    {
      "epoch": 1.3943672180175782e-05,
      "step": 14621,
      "training_step_time": 17.052905797958374
    },
    {
      "epoch": 1.3944625854492187e-05,
      "model_forward_time": 5.602073907852173,
      "step": 14622
    },
    {
      "epoch": 1.3944625854492187e-05,
      "step": 14622,
      "training_step_time": 17.060460090637207
    },
    {
      "epoch": 1.3945579528808593e-05,
      "model_forward_time": 5.5991106033325195,
      "step": 14623
    },
    {
      "epoch": 1.3945579528808593e-05,
      "step": 14623,
      "training_step_time": 17.076157093048096
    },
    {
      "epoch": 1.3946533203125e-05,
      "model_forward_time": 5.600696563720703,
      "step": 14624
    },
    {
      "epoch": 1.3946533203125e-05,
      "step": 14624,
      "training_step_time": 17.07981300354004
    },
    {
      "epoch": 1.3947486877441406e-05,
      "model_forward_time": 5.996671438217163,
      "step": 14625
    },
    {
      "epoch": 1.3947486877441406e-05,
      "step": 14625,
      "training_step_time": 17.476474285125732
    },
    {
      "epoch": 1.3948440551757812e-05,
      "model_forward_time": 5.597392797470093,
      "step": 14626
    },
    {
      "epoch": 1.3948440551757812e-05,
      "step": 14626,
      "training_step_time": 17.03882670402527
    },
    {
      "epoch": 1.394939422607422e-05,
      "model_forward_time": 5.58599328994751,
      "step": 14627
    },
    {
      "epoch": 1.394939422607422e-05,
      "step": 14627,
      "training_step_time": 17.04415464401245
    },
    {
      "epoch": 1.3950347900390625e-05,
      "model_forward_time": 5.594553709030151,
      "step": 14628
    },
    {
      "epoch": 1.3950347900390625e-05,
      "step": 14628,
      "training_step_time": 17.057454824447632
    },
    {
      "epoch": 1.395130157470703e-05,
      "model_forward_time": 5.631651401519775,
      "step": 14629
    },
    {
      "epoch": 1.395130157470703e-05,
      "step": 14629,
      "training_step_time": 17.10574698448181
    },
    {
      "epoch": 1.3952255249023438e-05,
      "grad_norm": 0.47699812054634094,
      "learning_rate": 1e-05,
      "loss": 0.0551,
      "step": 14630
    },
    {
      "epoch": 1.3952255249023438e-05,
      "model_forward_time": 5.5920538902282715,
      "step": 14630
    },
    {
      "dynamics_loss_avg": 0.04904317483305931,
      "epoch": 1.3952255249023438e-05,
      "step": 14630
    },
    {
      "action_loss_avg": 0.006289462663698941,
      "epoch": 1.3952255249023438e-05,
      "step": 14630
    },
    {
      "epoch": 1.3952255249023438e-05,
      "step": 14630,
      "training_step_time": 17.412519454956055
    },
    {
      "epoch": 1.3953208923339844e-05,
      "model_forward_time": 5.6009392738342285,
      "step": 14631
    },
    {
      "epoch": 1.3953208923339844e-05,
      "step": 14631,
      "training_step_time": 17.081685543060303
    },
    {
      "epoch": 1.395416259765625e-05,
      "model_forward_time": 5.5990989208221436,
      "step": 14632
    },
    {
      "epoch": 1.395416259765625e-05,
      "step": 14632,
      "training_step_time": 17.047536849975586
    },
    {
      "epoch": 1.3955116271972655e-05,
      "model_forward_time": 5.59949517250061,
      "step": 14633
    },
    {
      "epoch": 1.3955116271972655e-05,
      "step": 14633,
      "training_step_time": 17.02815532684326
    },
    {
      "epoch": 1.3956069946289063e-05,
      "model_forward_time": 5.589171648025513,
      "step": 14634
    },
    {
      "epoch": 1.3956069946289063e-05,
      "step": 14634,
      "training_step_time": 17.052958488464355
    },
    {
      "epoch": 1.3957023620605469e-05,
      "model_forward_time": 5.5988287925720215,
      "step": 14635
    },
    {
      "epoch": 1.3957023620605469e-05,
      "step": 14635,
      "training_step_time": 17.054514169692993
    },
    {
      "epoch": 1.3957977294921874e-05,
      "model_forward_time": 5.974966049194336,
      "step": 14636
    },
    {
      "epoch": 1.3957977294921874e-05,
      "step": 14636,
      "training_step_time": 17.434027433395386
    },
    {
      "epoch": 1.3958930969238282e-05,
      "model_forward_time": 5.60161280632019,
      "step": 14637
    },
    {
      "epoch": 1.3958930969238282e-05,
      "step": 14637,
      "training_step_time": 17.05740523338318
    },
    {
      "epoch": 1.3959884643554687e-05,
      "model_forward_time": 5.595629692077637,
      "step": 14638
    },
    {
      "epoch": 1.3959884643554687e-05,
      "step": 14638,
      "training_step_time": 17.0560200214386
    },
    {
      "epoch": 1.3960838317871093e-05,
      "model_forward_time": 5.592882394790649,
      "step": 14639
    },
    {
      "epoch": 1.3960838317871093e-05,
      "step": 14639,
      "training_step_time": 17.06487226486206
    },
    {
      "epoch": 1.39617919921875e-05,
      "grad_norm": 0.5963380932807922,
      "learning_rate": 1e-05,
      "loss": 0.0529,
      "step": 14640
    },
    {
      "epoch": 1.39617919921875e-05,
      "model_forward_time": 5.60966157913208,
      "step": 14640
    },
    {
      "dynamics_loss_avg": 0.047388223186135295,
      "epoch": 1.39617919921875e-05,
      "step": 14640
    },
    {
      "action_loss_avg": 0.005438057845458389,
      "epoch": 1.39617919921875e-05,
      "step": 14640
    },
    {
      "epoch": 1.39617919921875e-05,
      "step": 14640,
      "training_step_time": 17.087687015533447
    },
    {
      "epoch": 1.3962745666503906e-05,
      "model_forward_time": 5.968020677566528,
      "step": 14641
    },
    {
      "epoch": 1.3962745666503906e-05,
      "step": 14641,
      "training_step_time": 17.436339616775513
    },
    {
      "epoch": 1.3963699340820312e-05,
      "model_forward_time": 5.6101953983306885,
      "step": 14642
    },
    {
      "epoch": 1.3963699340820312e-05,
      "step": 14642,
      "training_step_time": 17.08113193511963
    },
    {
      "epoch": 1.396465301513672e-05,
      "model_forward_time": 5.604823350906372,
      "step": 14643
    },
    {
      "epoch": 1.396465301513672e-05,
      "step": 14643,
      "training_step_time": 17.084299087524414
    },
    {
      "epoch": 1.3965606689453125e-05,
      "model_forward_time": 5.601656436920166,
      "step": 14644
    },
    {
      "epoch": 1.3965606689453125e-05,
      "step": 14644,
      "training_step_time": 17.07608389854431
    },
    {
      "epoch": 1.396656036376953e-05,
      "model_forward_time": 5.6042046546936035,
      "step": 14645
    },
    {
      "epoch": 1.396656036376953e-05,
      "step": 14645,
      "training_step_time": 17.05563521385193
    },
    {
      "epoch": 1.3967514038085938e-05,
      "model_forward_time": 5.594481945037842,
      "step": 14646
    },
    {
      "epoch": 1.3967514038085938e-05,
      "step": 14646,
      "training_step_time": 17.048262357711792
    },
    {
      "epoch": 1.3968467712402344e-05,
      "model_forward_time": 5.984862804412842,
      "step": 14647
    },
    {
      "epoch": 1.3968467712402344e-05,
      "step": 14647,
      "training_step_time": 17.456995725631714
    },
    {
      "epoch": 1.396942138671875e-05,
      "model_forward_time": 5.608362197875977,
      "step": 14648
    },
    {
      "epoch": 1.396942138671875e-05,
      "step": 14648,
      "training_step_time": 17.109156370162964
    },
    {
      "epoch": 1.3970375061035155e-05,
      "model_forward_time": 5.611746311187744,
      "step": 14649
    },
    {
      "epoch": 1.3970375061035155e-05,
      "step": 14649,
      "training_step_time": 17.099304914474487
    },
    {
      "epoch": 1.3971328735351563e-05,
      "grad_norm": 0.6656908988952637,
      "learning_rate": 1e-05,
      "loss": 0.0543,
      "step": 14650
    },
    {
      "epoch": 1.3971328735351563e-05,
      "model_forward_time": 5.614465236663818,
      "step": 14650
    },
    {
      "dynamics_loss_avg": 0.04737152475863695,
      "epoch": 1.3971328735351563e-05,
      "step": 14650
    },
    {
      "action_loss_avg": 0.005434449855238199,
      "epoch": 1.3971328735351563e-05,
      "step": 14650
    },
    {
      "epoch": 1.3971328735351563e-05,
      "step": 14650,
      "training_step_time": 17.11210536956787
    },
    {
      "epoch": 1.3972282409667969e-05,
      "model_forward_time": 5.612688779830933,
      "step": 14651
    },
    {
      "epoch": 1.3972282409667969e-05,
      "step": 14651,
      "training_step_time": 17.110472917556763
    },
    {
      "epoch": 1.3973236083984374e-05,
      "model_forward_time": 5.609503746032715,
      "step": 14652
    },
    {
      "epoch": 1.3973236083984374e-05,
      "step": 14652,
      "training_step_time": 17.471629858016968
    },
    {
      "epoch": 1.3974189758300782e-05,
      "model_forward_time": 5.619312286376953,
      "step": 14653
    },
    {
      "epoch": 1.3974189758300782e-05,
      "step": 14653,
      "training_step_time": 17.099708795547485
    },
    {
      "epoch": 1.3975143432617187e-05,
      "model_forward_time": 5.6011457443237305,
      "step": 14654
    },
    {
      "epoch": 1.3975143432617187e-05,
      "step": 14654,
      "training_step_time": 17.069110870361328
    },
    {
      "epoch": 1.3976097106933593e-05,
      "model_forward_time": 5.603418827056885,
      "step": 14655
    },
    {
      "epoch": 1.3976097106933593e-05,
      "step": 14655,
      "training_step_time": 17.066670417785645
    },
    {
      "epoch": 1.397705078125e-05,
      "model_forward_time": 5.5975048542022705,
      "step": 14656
    },
    {
      "epoch": 1.397705078125e-05,
      "step": 14656,
      "training_step_time": 17.040435791015625
    },
    {
      "epoch": 1.3978004455566406e-05,
      "model_forward_time": 5.579852342605591,
      "step": 14657
    },
    {
      "epoch": 1.3978004455566406e-05,
      "step": 14657,
      "training_step_time": 17.018810033798218
    },
    {
      "epoch": 1.3978958129882812e-05,
      "model_forward_time": 5.955175399780273,
      "step": 14658
    },
    {
      "epoch": 1.3978958129882812e-05,
      "step": 14658,
      "training_step_time": 17.418840646743774
    },
    {
      "epoch": 1.397991180419922e-05,
      "model_forward_time": 5.617830514907837,
      "step": 14659
    },
    {
      "epoch": 1.397991180419922e-05,
      "step": 14659,
      "training_step_time": 17.1277859210968
    },
    {
      "epoch": 1.3980865478515625e-05,
      "grad_norm": 0.7841933369636536,
      "learning_rate": 1e-05,
      "loss": 0.0633,
      "step": 14660
    },
    {
      "epoch": 1.3980865478515625e-05,
      "model_forward_time": 5.646965026855469,
      "step": 14660
    },
    {
      "dynamics_loss_avg": 0.05823062993586063,
      "epoch": 1.3980865478515625e-05,
      "step": 14660
    },
    {
      "action_loss_avg": 0.008237126260064542,
      "epoch": 1.3980865478515625e-05,
      "step": 14660
    },
    {
      "epoch": 1.3980865478515625e-05,
      "step": 14660,
      "training_step_time": 17.15704846382141
    },
    {
      "epoch": 1.3981819152832031e-05,
      "model_forward_time": 5.5998382568359375,
      "step": 14661
    },
    {
      "epoch": 1.3981819152832031e-05,
      "step": 14661,
      "training_step_time": 17.05915403366089
    },
    {
      "epoch": 1.3982772827148438e-05,
      "model_forward_time": 5.578750848770142,
      "step": 14662
    },
    {
      "epoch": 1.3982772827148438e-05,
      "step": 14662,
      "training_step_time": 17.00482439994812
    },
    {
      "epoch": 1.3983726501464844e-05,
      "model_forward_time": 5.589615821838379,
      "step": 14663
    },
    {
      "epoch": 1.3983726501464844e-05,
      "step": 14663,
      "training_step_time": 17.384589433670044
    },
    {
      "epoch": 1.398468017578125e-05,
      "model_forward_time": 5.58789849281311,
      "step": 14664
    },
    {
      "epoch": 1.398468017578125e-05,
      "step": 14664,
      "training_step_time": 17.037500143051147
    },
    {
      "epoch": 1.3985633850097655e-05,
      "model_forward_time": 5.610059976577759,
      "step": 14665
    },
    {
      "epoch": 1.3985633850097655e-05,
      "step": 14665,
      "training_step_time": 17.087095737457275
    },
    {
      "epoch": 1.3986587524414063e-05,
      "model_forward_time": 5.617339611053467,
      "step": 14666
    },
    {
      "epoch": 1.3986587524414063e-05,
      "step": 14666,
      "training_step_time": 17.135332584381104
    },
    {
      "epoch": 1.3987541198730469e-05,
      "model_forward_time": 5.614686965942383,
      "step": 14667
    },
    {
      "epoch": 1.3987541198730469e-05,
      "step": 14667,
      "training_step_time": 17.117635488510132
    },
    {
      "epoch": 1.3988494873046874e-05,
      "model_forward_time": 5.607460021972656,
      "step": 14668
    },
    {
      "epoch": 1.3988494873046874e-05,
      "step": 14668,
      "training_step_time": 17.06533908843994
    },
    {
      "epoch": 1.3989448547363282e-05,
      "model_forward_time": 5.961817741394043,
      "step": 14669
    },
    {
      "epoch": 1.3989448547363282e-05,
      "step": 14669,
      "training_step_time": 17.372142553329468
    },
    {
      "epoch": 1.3990402221679687e-05,
      "grad_norm": 0.13623595237731934,
      "learning_rate": 1e-05,
      "loss": 0.0478,
      "step": 14670
    },
    {
      "epoch": 1.3990402221679687e-05,
      "model_forward_time": 5.5663111209869385,
      "step": 14670
    },
    {
      "dynamics_loss_avg": 0.04550761342979968,
      "epoch": 1.3990402221679687e-05,
      "step": 14670
    },
    {
      "action_loss_avg": 0.005418927659047767,
      "epoch": 1.3990402221679687e-05,
      "step": 14670
    },
    {
      "epoch": 1.3990402221679687e-05,
      "step": 14670,
      "training_step_time": 16.973605394363403
    },
    {
      "epoch": 1.3991355895996093e-05,
      "model_forward_time": 5.577099561691284,
      "step": 14671
    },
    {
      "epoch": 1.3991355895996093e-05,
      "step": 14671,
      "training_step_time": 16.97935676574707
    },
    {
      "epoch": 1.39923095703125e-05,
      "model_forward_time": 5.58236837387085,
      "step": 14672
    },
    {
      "epoch": 1.39923095703125e-05,
      "step": 14672,
      "training_step_time": 17.035701990127563
    },
    {
      "epoch": 1.3993263244628906e-05,
      "model_forward_time": 5.578218936920166,
      "step": 14673
    },
    {
      "epoch": 1.3993263244628906e-05,
      "step": 14673,
      "training_step_time": 17.003387212753296
    },
    {
      "epoch": 1.3994216918945312e-05,
      "model_forward_time": 5.954624652862549,
      "step": 14674
    },
    {
      "epoch": 1.3994216918945312e-05,
      "step": 14674,
      "training_step_time": 17.376241207122803
    },
    {
      "epoch": 1.399517059326172e-05,
      "model_forward_time": 5.593050241470337,
      "step": 14675
    },
    {
      "epoch": 1.399517059326172e-05,
      "step": 14675,
      "training_step_time": 17.050191640853882
    },
    {
      "epoch": 1.3996124267578125e-05,
      "model_forward_time": 5.626263856887817,
      "step": 14676
    },
    {
      "epoch": 1.3996124267578125e-05,
      "step": 14676,
      "training_step_time": 17.13947033882141
    },
    {
      "epoch": 1.3997077941894531e-05,
      "model_forward_time": 5.6515212059021,
      "step": 14677
    },
    {
      "epoch": 1.3997077941894531e-05,
      "step": 14677,
      "training_step_time": 17.18079161643982
    },
    {
      "epoch": 1.3998031616210938e-05,
      "model_forward_time": 5.617906093597412,
      "step": 14678
    },
    {
      "epoch": 1.3998031616210938e-05,
      "step": 14678,
      "training_step_time": 17.108458757400513
    },
    {
      "epoch": 1.3998985290527344e-05,
      "model_forward_time": 5.607447862625122,
      "step": 14679
    },
    {
      "epoch": 1.3998985290527344e-05,
      "step": 14679,
      "training_step_time": 17.085986375808716
    },
    {
      "epoch": 1.399993896484375e-05,
      "grad_norm": 0.8713400959968567,
      "learning_rate": 1e-05,
      "loss": 0.0662,
      "step": 14680
    },
    {
      "epoch": 1.399993896484375e-05,
      "model_forward_time": 5.99169659614563,
      "step": 14680
    },
    {
      "dynamics_loss_avg": 0.056149017065763474,
      "epoch": 1.399993896484375e-05,
      "step": 14680
    },
    {
      "action_loss_avg": 0.006698869680985808,
      "epoch": 1.399993896484375e-05,
      "step": 14680
    },
    {
      "epoch": 1.399993896484375e-05,
      "step": 14680,
      "training_step_time": 17.458686351776123
    },
    {
      "epoch": 1.4000892639160155e-05,
      "model_forward_time": 5.5980939865112305,
      "step": 14681
    },
    {
      "epoch": 1.4000892639160155e-05,
      "step": 14681,
      "training_step_time": 17.052751779556274
    },
    {
      "epoch": 1.4001846313476563e-05,
      "model_forward_time": 5.592600584030151,
      "step": 14682
    },
    {
      "epoch": 1.4001846313476563e-05,
      "step": 14682,
      "training_step_time": 17.05238699913025
    },
    {
      "epoch": 1.4002799987792969e-05,
      "model_forward_time": 5.602847337722778,
      "step": 14683
    },
    {
      "epoch": 1.4002799987792969e-05,
      "step": 14683,
      "training_step_time": 17.065439224243164
    },
    {
      "epoch": 1.4003753662109374e-05,
      "model_forward_time": 5.610629081726074,
      "step": 14684
    },
    {
      "epoch": 1.4003753662109374e-05,
      "step": 14684,
      "training_step_time": 17.07968306541443
    },
    {
      "epoch": 1.4004707336425782e-05,
      "model_forward_time": 5.6007325649261475,
      "step": 14685
    },
    {
      "epoch": 1.4004707336425782e-05,
      "step": 14685,
      "training_step_time": 17.429155588150024
    },
    {
      "epoch": 1.4005661010742187e-05,
      "model_forward_time": 5.591804027557373,
      "step": 14686
    },
    {
      "epoch": 1.4005661010742187e-05,
      "step": 14686,
      "training_step_time": 17.051758527755737
    },
    {
      "epoch": 1.4006614685058593e-05,
      "model_forward_time": 5.609341144561768,
      "step": 14687
    },
    {
      "epoch": 1.4006614685058593e-05,
      "step": 14687,
      "training_step_time": 17.0804386138916
    },
    {
      "epoch": 1.4007568359375e-05,
      "model_forward_time": 5.606609106063843,
      "step": 14688
    },
    {
      "epoch": 1.4007568359375e-05,
      "step": 14688,
      "training_step_time": 17.072678327560425
    },
    {
      "epoch": 1.4008522033691406e-05,
      "model_forward_time": 5.606354236602783,
      "step": 14689
    },
    {
      "epoch": 1.4008522033691406e-05,
      "step": 14689,
      "training_step_time": 17.06817126274109
    },
    {
      "epoch": 1.4009475708007812e-05,
      "grad_norm": 0.8644900321960449,
      "learning_rate": 1e-05,
      "loss": 0.0634,
      "step": 14690
    },
    {
      "epoch": 1.4009475708007812e-05,
      "model_forward_time": 5.6029839515686035,
      "step": 14690
    },
    {
      "dynamics_loss_avg": 0.05647587925195694,
      "epoch": 1.4009475708007812e-05,
      "step": 14690
    },
    {
      "action_loss_avg": 0.008715340658091009,
      "epoch": 1.4009475708007812e-05,
      "step": 14690
    },
    {
      "epoch": 1.4009475708007812e-05,
      "step": 14690,
      "training_step_time": 17.07739496231079
    },
    {
      "epoch": 1.401042938232422e-05,
      "model_forward_time": 5.990036249160767,
      "step": 14691
    },
    {
      "epoch": 1.401042938232422e-05,
      "step": 14691,
      "training_step_time": 17.45570397377014
    },
    {
      "epoch": 1.4011383056640625e-05,
      "model_forward_time": 5.597079277038574,
      "step": 14692
    },
    {
      "epoch": 1.4011383056640625e-05,
      "step": 14692,
      "training_step_time": 17.03906011581421
    },
    {
      "epoch": 1.4012336730957031e-05,
      "model_forward_time": 5.592050552368164,
      "step": 14693
    },
    {
      "epoch": 1.4012336730957031e-05,
      "step": 14693,
      "training_step_time": 17.018413066864014
    },
    {
      "epoch": 1.4013290405273438e-05,
      "model_forward_time": 5.588194131851196,
      "step": 14694
    },
    {
      "epoch": 1.4013290405273438e-05,
      "step": 14694,
      "training_step_time": 17.021528482437134
    },
    {
      "epoch": 1.4014244079589844e-05,
      "model_forward_time": 5.593543767929077,
      "step": 14695
    },
    {
      "epoch": 1.4014244079589844e-05,
      "step": 14695,
      "training_step_time": 17.059385776519775
    },
    {
      "epoch": 1.401519775390625e-05,
      "model_forward_time": 5.621139764785767,
      "step": 14696
    },
    {
      "epoch": 1.401519775390625e-05,
      "step": 14696,
      "training_step_time": 17.45449733734131
    },
    {
      "epoch": 1.4016151428222655e-05,
      "model_forward_time": 5.6190185546875,
      "step": 14697
    },
    {
      "epoch": 1.4016151428222655e-05,
      "step": 14697,
      "training_step_time": 17.121079206466675
    },
    {
      "epoch": 1.4017105102539063e-05,
      "model_forward_time": 5.61366605758667,
      "step": 14698
    },
    {
      "epoch": 1.4017105102539063e-05,
      "step": 14698,
      "training_step_time": 17.096394777297974
    },
    {
      "epoch": 1.4018058776855469e-05,
      "model_forward_time": 5.6033337116241455,
      "step": 14699
    },
    {
      "epoch": 1.4018058776855469e-05,
      "step": 14699,
      "training_step_time": 17.076653242111206
    },
    {
      "epoch": 1.4019012451171874e-05,
      "grad_norm": 0.6349931359291077,
      "learning_rate": 1e-05,
      "loss": 0.0733,
      "step": 14700
    },
    {
      "epoch": 1.4019012451171874e-05,
      "model_forward_time": 5.596810579299927,
      "step": 14700
    },
    {
      "dynamics_loss_avg": 0.0630003210157156,
      "epoch": 1.4019012451171874e-05,
      "step": 14700
    },
    {
      "action_loss_avg": 0.007564344303682446,
      "epoch": 1.4019012451171874e-05,
      "step": 14700
    },
    {
      "epoch": 1.4019012451171874e-05,
      "step": 14700,
      "training_step_time": 17.073402643203735
    },
    {
      "epoch": 1.4019966125488282e-05,
      "model_forward_time": 5.612476825714111,
      "step": 14701
    },
    {
      "epoch": 1.4019966125488282e-05,
      "step": 14701,
      "training_step_time": 17.088038206100464
    },
    {
      "epoch": 1.4020919799804687e-05,
      "model_forward_time": 5.973199129104614,
      "step": 14702
    },
    {
      "epoch": 1.4020919799804687e-05,
      "step": 14702,
      "training_step_time": 17.41843295097351
    },
    {
      "epoch": 1.4021873474121093e-05,
      "model_forward_time": 5.596733808517456,
      "step": 14703
    },
    {
      "epoch": 1.4021873474121093e-05,
      "step": 14703,
      "training_step_time": 17.040228843688965
    },
    {
      "epoch": 1.40228271484375e-05,
      "model_forward_time": 5.593952894210815,
      "step": 14704
    },
    {
      "epoch": 1.40228271484375e-05,
      "step": 14704,
      "training_step_time": 17.036413192749023
    },
    {
      "epoch": 1.4023780822753906e-05,
      "model_forward_time": 5.593277454376221,
      "step": 14705
    },
    {
      "epoch": 1.4023780822753906e-05,
      "step": 14705,
      "training_step_time": 17.031192302703857
    },
    {
      "epoch": 1.4024734497070312e-05,
      "model_forward_time": 5.589282512664795,
      "step": 14706
    },
    {
      "epoch": 1.4024734497070312e-05,
      "step": 14706,
      "training_step_time": 17.021607398986816
    },
    {
      "epoch": 1.402568817138672e-05,
      "model_forward_time": 5.9633262157440186,
      "step": 14707
    },
    {
      "epoch": 1.402568817138672e-05,
      "step": 14707,
      "training_step_time": 17.394246816635132
    },
    {
      "epoch": 1.4026641845703125e-05,
      "model_forward_time": 5.59378719329834,
      "step": 14708
    },
    {
      "epoch": 1.4026641845703125e-05,
      "step": 14708,
      "training_step_time": 17.05741786956787
    },
    {
      "epoch": 1.4027595520019531e-05,
      "model_forward_time": 5.603964328765869,
      "step": 14709
    },
    {
      "epoch": 1.4027595520019531e-05,
      "step": 14709,
      "training_step_time": 17.0755615234375
    },
    {
      "epoch": 1.4028549194335938e-05,
      "grad_norm": 0.40502113103866577,
      "learning_rate": 1e-05,
      "loss": 0.0629,
      "step": 14710
    },
    {
      "epoch": 1.4028549194335938e-05,
      "model_forward_time": 5.610929250717163,
      "step": 14710
    },
    {
      "dynamics_loss_avg": 0.058391038421541454,
      "epoch": 1.4028549194335938e-05,
      "step": 14710
    },
    {
      "action_loss_avg": 0.007227649888955056,
      "epoch": 1.4028549194335938e-05,
      "step": 14710
    },
    {
      "epoch": 1.4028549194335938e-05,
      "step": 14710,
      "training_step_time": 17.10387110710144
    },
    {
      "epoch": 1.4029502868652344e-05,
      "model_forward_time": 5.645259141921997,
      "step": 14711
    },
    {
      "epoch": 1.4029502868652344e-05,
      "step": 14711,
      "training_step_time": 17.123117685317993
    },
    {
      "epoch": 1.403045654296875e-05,
      "model_forward_time": 5.601590394973755,
      "step": 14712
    },
    {
      "epoch": 1.403045654296875e-05,
      "step": 14712,
      "training_step_time": 17.035618782043457
    },
    {
      "epoch": 1.4031410217285155e-05,
      "model_forward_time": 5.939056158065796,
      "step": 14713
    },
    {
      "epoch": 1.4031410217285155e-05,
      "step": 14713,
      "training_step_time": 17.368628978729248
    },
    {
      "epoch": 1.4032363891601563e-05,
      "model_forward_time": 5.568497896194458,
      "step": 14714
    },
    {
      "epoch": 1.4032363891601563e-05,
      "step": 14714,
      "training_step_time": 17.00292706489563
    },
    {
      "epoch": 1.4033317565917969e-05,
      "model_forward_time": 5.569439888000488,
      "step": 14715
    },
    {
      "epoch": 1.4033317565917969e-05,
      "step": 14715,
      "training_step_time": 16.983795404434204
    },
    {
      "epoch": 1.4034271240234374e-05,
      "model_forward_time": 5.571255445480347,
      "step": 14716
    },
    {
      "epoch": 1.4034271240234374e-05,
      "step": 14716,
      "training_step_time": 16.990427255630493
    },
    {
      "epoch": 1.4035224914550782e-05,
      "model_forward_time": 5.570331811904907,
      "step": 14717
    },
    {
      "epoch": 1.4035224914550782e-05,
      "step": 14717,
      "training_step_time": 17.00607943534851
    },
    {
      "epoch": 1.4036178588867187e-05,
      "model_forward_time": 5.5855677127838135,
      "step": 14718
    },
    {
      "epoch": 1.4036178588867187e-05,
      "step": 14718,
      "training_step_time": 17.39062523841858
    },
    {
      "epoch": 1.4037132263183593e-05,
      "model_forward_time": 5.5899224281311035,
      "step": 14719
    },
    {
      "epoch": 1.4037132263183593e-05,
      "step": 14719,
      "training_step_time": 17.062753438949585
    },
    {
      "epoch": 1.40380859375e-05,
      "grad_norm": 0.8455188274383545,
      "learning_rate": 1e-05,
      "loss": 0.0696,
      "step": 14720
    },
    {
      "epoch": 1.40380859375e-05,
      "model_forward_time": 5.609012126922607,
      "step": 14720
    },
    {
      "dynamics_loss_avg": 0.06132359653711319,
      "epoch": 1.40380859375e-05,
      "step": 14720
    },
    {
      "action_loss_avg": 0.007901356182992458,
      "epoch": 1.40380859375e-05,
      "step": 14720
    },
    {
      "epoch": 1.40380859375e-05,
      "step": 14720,
      "training_step_time": 17.09297013282776
    },
    {
      "epoch": 1.4039039611816406e-05,
      "model_forward_time": 5.622163772583008,
      "step": 14721
    },
    {
      "epoch": 1.4039039611816406e-05,
      "step": 14721,
      "training_step_time": 17.12359118461609
    },
    {
      "epoch": 1.4039993286132812e-05,
      "model_forward_time": 5.607522487640381,
      "step": 14722
    },
    {
      "epoch": 1.4039993286132812e-05,
      "step": 14722,
      "training_step_time": 17.072784900665283
    },
    {
      "epoch": 1.404094696044922e-05,
      "model_forward_time": 5.597714185714722,
      "step": 14723
    },
    {
      "epoch": 1.404094696044922e-05,
      "step": 14723,
      "training_step_time": 17.040332794189453
    },
    {
      "epoch": 1.4041900634765625e-05,
      "model_forward_time": 5.947527885437012,
      "step": 14724
    },
    {
      "epoch": 1.4041900634765625e-05,
      "step": 14724,
      "training_step_time": 17.382030248641968
    },
    {
      "epoch": 1.4042854309082031e-05,
      "model_forward_time": 5.58903694152832,
      "step": 14725
    },
    {
      "epoch": 1.4042854309082031e-05,
      "step": 14725,
      "training_step_time": 17.04585576057434
    },
    {
      "epoch": 1.4043807983398438e-05,
      "model_forward_time": 5.593447923660278,
      "step": 14726
    },
    {
      "epoch": 1.4043807983398438e-05,
      "step": 14726,
      "training_step_time": 17.06757926940918
    },
    {
      "epoch": 1.4044761657714844e-05,
      "model_forward_time": 5.612630367279053,
      "step": 14727
    },
    {
      "epoch": 1.4044761657714844e-05,
      "step": 14727,
      "training_step_time": 17.11345934867859
    },
    {
      "epoch": 1.404571533203125e-05,
      "model_forward_time": 5.608854293823242,
      "step": 14728
    },
    {
      "epoch": 1.404571533203125e-05,
      "step": 14728,
      "training_step_time": 17.075007677078247
    },
    {
      "epoch": 1.4046669006347655e-05,
      "model_forward_time": 5.61137843132019,
      "step": 14729
    },
    {
      "epoch": 1.4046669006347655e-05,
      "step": 14729,
      "training_step_time": 17.432466506958008
    },
    {
      "epoch": 1.4047622680664063e-05,
      "grad_norm": 0.5269656777381897,
      "learning_rate": 1e-05,
      "loss": 0.0617,
      "step": 14730
    },
    {
      "epoch": 1.4047622680664063e-05,
      "model_forward_time": 5.595416069030762,
      "step": 14730
    },
    {
      "dynamics_loss_avg": 0.053772860765457155,
      "epoch": 1.4047622680664063e-05,
      "step": 14730
    },
    {
      "action_loss_avg": 0.006420726189389825,
      "epoch": 1.4047622680664063e-05,
      "step": 14730
    },
    {
      "epoch": 1.4047622680664063e-05,
      "step": 14730,
      "training_step_time": 17.035395860671997
    },
    {
      "epoch": 1.4048576354980469e-05,
      "model_forward_time": 5.583719968795776,
      "step": 14731
    },
    {
      "epoch": 1.4048576354980469e-05,
      "step": 14731,
      "training_step_time": 17.011823654174805
    },
    {
      "epoch": 1.4049530029296874e-05,
      "model_forward_time": 5.588104486465454,
      "step": 14732
    },
    {
      "epoch": 1.4049530029296874e-05,
      "step": 14732,
      "training_step_time": 17.036261796951294
    },
    {
      "epoch": 1.4050483703613282e-05,
      "model_forward_time": 5.584211111068726,
      "step": 14733
    },
    {
      "epoch": 1.4050483703613282e-05,
      "step": 14733,
      "training_step_time": 17.034684896469116
    },
    {
      "epoch": 1.4051437377929687e-05,
      "model_forward_time": 5.605165481567383,
      "step": 14734
    },
    {
      "epoch": 1.4051437377929687e-05,
      "step": 14734,
      "training_step_time": 17.070611000061035
    },
    {
      "epoch": 1.4052391052246093e-05,
      "model_forward_time": 5.982225179672241,
      "step": 14735
    },
    {
      "epoch": 1.4052391052246093e-05,
      "step": 14735,
      "training_step_time": 17.45951509475708
    },
    {
      "epoch": 1.40533447265625e-05,
      "model_forward_time": 5.605526924133301,
      "step": 14736
    },
    {
      "epoch": 1.40533447265625e-05,
      "step": 14736,
      "training_step_time": 17.076282739639282
    },
    {
      "epoch": 1.4054298400878906e-05,
      "model_forward_time": 5.608384132385254,
      "step": 14737
    },
    {
      "epoch": 1.4054298400878906e-05,
      "step": 14737,
      "training_step_time": 17.11386203765869
    },
    {
      "epoch": 1.4055252075195312e-05,
      "model_forward_time": 5.604764938354492,
      "step": 14738
    },
    {
      "epoch": 1.4055252075195312e-05,
      "step": 14738,
      "training_step_time": 17.090763092041016
    },
    {
      "epoch": 1.405620574951172e-05,
      "model_forward_time": 5.6034815311431885,
      "step": 14739
    },
    {
      "epoch": 1.405620574951172e-05,
      "step": 14739,
      "training_step_time": 17.056630849838257
    },
    {
      "epoch": 1.4057159423828125e-05,
      "grad_norm": 0.23584435880184174,
      "learning_rate": 1e-05,
      "loss": 0.0498,
      "step": 14740
    },
    {
      "epoch": 1.4057159423828125e-05,
      "model_forward_time": 5.940810918807983,
      "step": 14740
    },
    {
      "dynamics_loss_avg": 0.04656901452690363,
      "epoch": 1.4057159423828125e-05,
      "step": 14740
    },
    {
      "action_loss_avg": 0.005023315583821386,
      "epoch": 1.4057159423828125e-05,
      "step": 14740
    },
    {
      "epoch": 1.4057159423828125e-05,
      "step": 14740,
      "training_step_time": 17.39016604423523
    },
    {
      "epoch": 1.4058113098144531e-05,
      "model_forward_time": 5.586151599884033,
      "step": 14741
    },
    {
      "epoch": 1.4058113098144531e-05,
      "step": 14741,
      "training_step_time": 17.035439252853394
    },
    {
      "epoch": 1.4059066772460938e-05,
      "model_forward_time": 5.591547250747681,
      "step": 14742
    },
    {
      "epoch": 1.4059066772460938e-05,
      "step": 14742,
      "training_step_time": 17.04160189628601
    },
    {
      "epoch": 1.4060020446777344e-05,
      "model_forward_time": 5.599756240844727,
      "step": 14743
    },
    {
      "epoch": 1.4060020446777344e-05,
      "step": 14743,
      "training_step_time": 17.088115692138672
    },
    {
      "epoch": 1.406097412109375e-05,
      "model_forward_time": 5.6266374588012695,
      "step": 14744
    },
    {
      "epoch": 1.406097412109375e-05,
      "step": 14744,
      "training_step_time": 17.136845111846924
    },
    {
      "epoch": 1.4061927795410156e-05,
      "model_forward_time": 5.620501756668091,
      "step": 14745
    },
    {
      "epoch": 1.4061927795410156e-05,
      "step": 14745,
      "training_step_time": 17.11501383781433
    },
    {
      "epoch": 1.4062881469726563e-05,
      "model_forward_time": 5.99765682220459,
      "step": 14746
    },
    {
      "epoch": 1.4062881469726563e-05,
      "step": 14746,
      "training_step_time": 17.487070322036743
    },
    {
      "epoch": 1.4063835144042969e-05,
      "model_forward_time": 5.599177598953247,
      "step": 14747
    },
    {
      "epoch": 1.4063835144042969e-05,
      "step": 14747,
      "training_step_time": 17.046998500823975
    },
    {
      "epoch": 1.4064788818359374e-05,
      "model_forward_time": 5.6001996994018555,
      "step": 14748
    },
    {
      "epoch": 1.4064788818359374e-05,
      "step": 14748,
      "training_step_time": 17.054821252822876
    },
    {
      "epoch": 1.4065742492675782e-05,
      "model_forward_time": 5.59413480758667,
      "step": 14749
    },
    {
      "epoch": 1.4065742492675782e-05,
      "step": 14749,
      "training_step_time": 17.064658164978027
    },
    {
      "epoch": 1.4066696166992188e-05,
      "grad_norm": 0.9765462279319763,
      "learning_rate": 1e-05,
      "loss": 0.0661,
      "step": 14750
    },
    {
      "epoch": 1.4066696166992188e-05,
      "model_forward_time": 5.599323034286499,
      "step": 14750
    },
    {
      "dynamics_loss_avg": 0.05907311309129,
      "epoch": 1.4066696166992188e-05,
      "step": 14750
    },
    {
      "action_loss_avg": 0.008296755026094615,
      "epoch": 1.4066696166992188e-05,
      "step": 14750
    },
    {
      "epoch": 1.4066696166992188e-05,
      "step": 14750,
      "training_step_time": 17.069667100906372
    },
    {
      "epoch": 1.4067649841308593e-05,
      "model_forward_time": 5.614877462387085,
      "step": 14751
    },
    {
      "epoch": 1.4067649841308593e-05,
      "step": 14751,
      "training_step_time": 17.4599928855896
    },
    {
      "epoch": 1.4068603515625e-05,
      "model_forward_time": 5.603253602981567,
      "step": 14752
    },
    {
      "epoch": 1.4068603515625e-05,
      "step": 14752,
      "training_step_time": 17.078493356704712
    },
    {
      "epoch": 1.4069557189941406e-05,
      "model_forward_time": 5.610211610794067,
      "step": 14753
    },
    {
      "epoch": 1.4069557189941406e-05,
      "step": 14753,
      "training_step_time": 17.099120140075684
    },
    {
      "epoch": 1.4070510864257812e-05,
      "model_forward_time": 5.614054441452026,
      "step": 14754
    },
    {
      "epoch": 1.4070510864257812e-05,
      "step": 14754,
      "training_step_time": 17.10635542869568
    },
    {
      "epoch": 1.407146453857422e-05,
      "model_forward_time": 5.615103721618652,
      "step": 14755
    },
    {
      "epoch": 1.407146453857422e-05,
      "step": 14755,
      "training_step_time": 17.102096557617188
    },
    {
      "epoch": 1.4072418212890625e-05,
      "model_forward_time": 4.279782056808472,
      "step": 14756
    },
    {
      "epoch": 1.4072418212890625e-05,
      "step": 14756,
      "training_step_time": 12.81734299659729
    },
    {
      "epoch": 1.4073371887207031e-05,
      "model_forward_time": 5.9901511669158936,
      "step": 14757
    },
    {
      "epoch": 1.4073371887207031e-05,
      "step": 14757,
      "training_step_time": 17.48177409172058
    },
    {
      "epoch": 1.4074325561523437e-05,
      "model_forward_time": 5.6020214557647705,
      "step": 14758
    },
    {
      "epoch": 1.4074325561523437e-05,
      "step": 14758,
      "training_step_time": 17.07184648513794
    },
    {
      "epoch": 1.4075279235839844e-05,
      "model_forward_time": 5.599883317947388,
      "step": 14759
    },
    {
      "epoch": 1.4075279235839844e-05,
      "step": 14759,
      "training_step_time": 17.07522487640381
    },
    {
      "epoch": 1.407623291015625e-05,
      "grad_norm": 1.5154098272323608,
      "learning_rate": 1e-05,
      "loss": 0.0833,
      "step": 14760
    },
    {
      "epoch": 1.407623291015625e-05,
      "model_forward_time": 5.606239318847656,
      "step": 14760
    },
    {
      "dynamics_loss_avg": 0.06692906673997641,
      "epoch": 1.407623291015625e-05,
      "step": 14760
    },
    {
      "action_loss_avg": 0.011022764001972973,
      "epoch": 1.407623291015625e-05,
      "step": 14760
    },
    {
      "epoch": 1.407623291015625e-05,
      "step": 14760,
      "training_step_time": 17.088343143463135
    },
    {
      "epoch": 1.4077186584472656e-05,
      "model_forward_time": 5.591045618057251,
      "step": 14761
    },
    {
      "epoch": 1.4077186584472656e-05,
      "step": 14761,
      "training_step_time": 17.038264513015747
    },
    {
      "epoch": 1.4078140258789063e-05,
      "model_forward_time": 5.589594841003418,
      "step": 14762
    },
    {
      "epoch": 1.4078140258789063e-05,
      "step": 14762,
      "training_step_time": 17.401524305343628
    },
    {
      "epoch": 1.4079093933105469e-05,
      "model_forward_time": 5.593859672546387,
      "step": 14763
    },
    {
      "epoch": 1.4079093933105469e-05,
      "step": 14763,
      "training_step_time": 17.037956953048706
    },
    {
      "epoch": 1.4080047607421874e-05,
      "model_forward_time": 5.585982084274292,
      "step": 14764
    },
    {
      "epoch": 1.4080047607421874e-05,
      "step": 14764,
      "training_step_time": 17.018190145492554
    },
    {
      "epoch": 1.4081001281738282e-05,
      "model_forward_time": 5.595356702804565,
      "step": 14765
    },
    {
      "epoch": 1.4081001281738282e-05,
      "step": 14765,
      "training_step_time": 17.04378867149353
    },
    {
      "epoch": 1.4081954956054688e-05,
      "model_forward_time": 5.6036436557769775,
      "step": 14766
    },
    {
      "epoch": 1.4081954956054688e-05,
      "step": 14766,
      "training_step_time": 17.08308506011963
    },
    {
      "epoch": 1.4082908630371093e-05,
      "model_forward_time": 5.616176605224609,
      "step": 14767
    },
    {
      "epoch": 1.4082908630371093e-05,
      "step": 14767,
      "training_step_time": 17.10662603378296
    },
    {
      "epoch": 1.40838623046875e-05,
      "model_forward_time": 5.996069669723511,
      "step": 14768
    },
    {
      "epoch": 1.40838623046875e-05,
      "step": 14768,
      "training_step_time": 17.489027976989746
    },
    {
      "epoch": 1.4084815979003906e-05,
      "model_forward_time": 5.617563962936401,
      "step": 14769
    },
    {
      "epoch": 1.4084815979003906e-05,
      "step": 14769,
      "training_step_time": 17.10129737854004
    },
    {
      "epoch": 1.4085769653320312e-05,
      "grad_norm": 0.34902578592300415,
      "learning_rate": 1e-05,
      "loss": 0.0487,
      "step": 14770
    },
    {
      "epoch": 1.4085769653320312e-05,
      "model_forward_time": 5.61592960357666,
      "step": 14770
    },
    {
      "dynamics_loss_avg": 0.0480433726683259,
      "epoch": 1.4085769653320312e-05,
      "step": 14770
    },
    {
      "action_loss_avg": 0.004689248511567712,
      "epoch": 1.4085769653320312e-05,
      "step": 14770
    },
    {
      "epoch": 1.4085769653320312e-05,
      "step": 14770,
      "training_step_time": 17.104708671569824
    },
    {
      "epoch": 1.408672332763672e-05,
      "model_forward_time": 5.612858533859253,
      "step": 14771
    },
    {
      "epoch": 1.408672332763672e-05,
      "step": 14771,
      "training_step_time": 17.094697952270508
    },
    {
      "epoch": 1.4087677001953125e-05,
      "model_forward_time": 5.6048760414123535,
      "step": 14772
    },
    {
      "epoch": 1.4087677001953125e-05,
      "step": 14772,
      "training_step_time": 17.089850425720215
    },
    {
      "epoch": 1.4088630676269531e-05,
      "model_forward_time": 5.9506754875183105,
      "step": 14773
    },
    {
      "epoch": 1.4088630676269531e-05,
      "step": 14773,
      "training_step_time": 17.420111894607544
    },
    {
      "epoch": 1.4089584350585937e-05,
      "model_forward_time": 5.601967811584473,
      "step": 14774
    },
    {
      "epoch": 1.4089584350585937e-05,
      "step": 14774,
      "training_step_time": 17.050714254379272
    },
    {
      "epoch": 1.4090538024902344e-05,
      "model_forward_time": 5.603949546813965,
      "step": 14775
    },
    {
      "epoch": 1.4090538024902344e-05,
      "step": 14775,
      "training_step_time": 17.084805488586426
    },
    {
      "epoch": 1.409149169921875e-05,
      "model_forward_time": 5.599908828735352,
      "step": 14776
    },
    {
      "epoch": 1.409149169921875e-05,
      "step": 14776,
      "training_step_time": 17.057053327560425
    },
    {
      "epoch": 1.4092445373535156e-05,
      "model_forward_time": 5.602228879928589,
      "step": 14777
    },
    {
      "epoch": 1.4092445373535156e-05,
      "step": 14777,
      "training_step_time": 17.06709861755371
    },
    {
      "epoch": 1.4093399047851563e-05,
      "model_forward_time": 5.598829984664917,
      "step": 14778
    },
    {
      "epoch": 1.4093399047851563e-05,
      "step": 14778,
      "training_step_time": 17.046197175979614
    },
    {
      "epoch": 1.4094352722167969e-05,
      "model_forward_time": 5.954730272293091,
      "step": 14779
    },
    {
      "epoch": 1.4094352722167969e-05,
      "step": 14779,
      "training_step_time": 17.39338731765747
    },
    {
      "epoch": 1.4095306396484374e-05,
      "grad_norm": 0.5200579762458801,
      "learning_rate": 1e-05,
      "loss": 0.0541,
      "step": 14780
    },
    {
      "epoch": 1.4095306396484374e-05,
      "model_forward_time": 5.59255051612854,
      "step": 14780
    },
    {
      "dynamics_loss_avg": 0.04855602364987135,
      "epoch": 1.4095306396484374e-05,
      "step": 14780
    },
    {
      "action_loss_avg": 0.005252412054687738,
      "epoch": 1.4095306396484374e-05,
      "step": 14780
    },
    {
      "epoch": 1.4095306396484374e-05,
      "step": 14780,
      "training_step_time": 17.04915976524353
    },
    {
      "epoch": 1.4096260070800782e-05,
      "model_forward_time": 5.598642349243164,
      "step": 14781
    },
    {
      "epoch": 1.4096260070800782e-05,
      "step": 14781,
      "training_step_time": 17.0542094707489
    },
    {
      "epoch": 1.4097213745117188e-05,
      "model_forward_time": 5.6021013259887695,
      "step": 14782
    },
    {
      "epoch": 1.4097213745117188e-05,
      "step": 14782,
      "training_step_time": 17.078924894332886
    },
    {
      "epoch": 1.4098167419433593e-05,
      "model_forward_time": 5.622344493865967,
      "step": 14783
    },
    {
      "epoch": 1.4098167419433593e-05,
      "step": 14783,
      "training_step_time": 17.132006883621216
    },
    {
      "epoch": 1.409912109375e-05,
      "model_forward_time": 5.621805191040039,
      "step": 14784
    },
    {
      "epoch": 1.409912109375e-05,
      "step": 14784,
      "training_step_time": 17.487175464630127
    },
    {
      "epoch": 1.4100074768066406e-05,
      "model_forward_time": 5.590975522994995,
      "step": 14785
    },
    {
      "epoch": 1.4100074768066406e-05,
      "step": 14785,
      "training_step_time": 17.041806936264038
    },
    {
      "epoch": 1.4101028442382812e-05,
      "model_forward_time": 5.585707902908325,
      "step": 14786
    },
    {
      "epoch": 1.4101028442382812e-05,
      "step": 14786,
      "training_step_time": 17.038345336914062
    },
    {
      "epoch": 1.410198211669922e-05,
      "model_forward_time": 5.591789722442627,
      "step": 14787
    },
    {
      "epoch": 1.410198211669922e-05,
      "step": 14787,
      "training_step_time": 17.04127597808838
    },
    {
      "epoch": 1.4102935791015625e-05,
      "model_forward_time": 5.6045496463775635,
      "step": 14788
    },
    {
      "epoch": 1.4102935791015625e-05,
      "step": 14788,
      "training_step_time": 17.075289487838745
    },
    {
      "epoch": 1.4103889465332031e-05,
      "model_forward_time": 5.606570243835449,
      "step": 14789
    },
    {
      "epoch": 1.4103889465332031e-05,
      "step": 14789,
      "training_step_time": 17.083959102630615
    },
    {
      "epoch": 1.4104843139648437e-05,
      "grad_norm": 0.3438259959220886,
      "learning_rate": 1e-05,
      "loss": 0.0446,
      "step": 14790
    },
    {
      "epoch": 1.4104843139648437e-05,
      "model_forward_time": 5.989443778991699,
      "step": 14790
    },
    {
      "dynamics_loss_avg": 0.03733373917639256,
      "epoch": 1.4104843139648437e-05,
      "step": 14790
    },
    {
      "action_loss_avg": 0.004480076278559864,
      "epoch": 1.4104843139648437e-05,
      "step": 14790
    },
    {
      "epoch": 1.4104843139648437e-05,
      "step": 14790,
      "training_step_time": 17.449711084365845
    },
    {
      "epoch": 1.4105796813964844e-05,
      "model_forward_time": 5.591925859451294,
      "step": 14791
    },
    {
      "epoch": 1.4105796813964844e-05,
      "step": 14791,
      "training_step_time": 17.007770776748657
    },
    {
      "epoch": 1.410675048828125e-05,
      "model_forward_time": 5.585805654525757,
      "step": 14792
    },
    {
      "epoch": 1.410675048828125e-05,
      "step": 14792,
      "training_step_time": 17.028425455093384
    },
    {
      "epoch": 1.4107704162597656e-05,
      "model_forward_time": 5.5826029777526855,
      "step": 14793
    },
    {
      "epoch": 1.4107704162597656e-05,
      "step": 14793,
      "training_step_time": 16.997402906417847
    },
    {
      "epoch": 1.4108657836914063e-05,
      "model_forward_time": 5.594447612762451,
      "step": 14794
    },
    {
      "epoch": 1.4108657836914063e-05,
      "step": 14794,
      "training_step_time": 17.046794414520264
    },
    {
      "epoch": 1.4109611511230469e-05,
      "model_forward_time": 5.596609830856323,
      "step": 14795
    },
    {
      "epoch": 1.4109611511230469e-05,
      "step": 14795,
      "training_step_time": 17.426121711730957
    },
    {
      "epoch": 1.4110565185546874e-05,
      "model_forward_time": 5.603546380996704,
      "step": 14796
    },
    {
      "epoch": 1.4110565185546874e-05,
      "step": 14796,
      "training_step_time": 17.068373441696167
    },
    {
      "epoch": 1.4111518859863282e-05,
      "model_forward_time": 5.609328746795654,
      "step": 14797
    },
    {
      "epoch": 1.4111518859863282e-05,
      "step": 14797,
      "training_step_time": 17.06356382369995
    },
    {
      "epoch": 1.4112472534179688e-05,
      "model_forward_time": 5.607799053192139,
      "step": 14798
    },
    {
      "epoch": 1.4112472534179688e-05,
      "step": 14798,
      "training_step_time": 17.097341537475586
    },
    {
      "epoch": 1.4113426208496093e-05,
      "model_forward_time": 5.612395763397217,
      "step": 14799
    },
    {
      "epoch": 1.4113426208496093e-05,
      "step": 14799,
      "training_step_time": 17.114290475845337
    },
    {
      "epoch": 1.41143798828125e-05,
      "grad_norm": 0.6074876189231873,
      "learning_rate": 1e-05,
      "loss": 0.041,
      "step": 14800
    },
    {
      "epoch": 1.41143798828125e-05,
      "model_forward_time": 5.61506199836731,
      "step": 14800
    },
    {
      "dynamics_loss_avg": 0.03876754306256771,
      "epoch": 1.41143798828125e-05,
      "step": 14800
    },
    {
      "action_loss_avg": 0.004583697929047048,
      "epoch": 1.41143798828125e-05,
      "step": 14800
    },
    {
      "epoch": 1.41143798828125e-05,
      "step": 14800,
      "training_step_time": 17.096908569335938
    },
    {
      "epoch": 1.4115333557128906e-05,
      "model_forward_time": 5.991724967956543,
      "step": 14801
    },
    {
      "epoch": 1.4115333557128906e-05,
      "step": 14801,
      "training_step_time": 17.4784095287323
    },
    {
      "epoch": 1.4116287231445312e-05,
      "model_forward_time": 5.61133337020874,
      "step": 14802
    },
    {
      "epoch": 1.4116287231445312e-05,
      "step": 14802,
      "training_step_time": 17.11768102645874
    },
    {
      "epoch": 1.411724090576172e-05,
      "model_forward_time": 5.605967283248901,
      "step": 14803
    },
    {
      "epoch": 1.411724090576172e-05,
      "step": 14803,
      "training_step_time": 17.06790280342102
    },
    {
      "epoch": 1.4118194580078125e-05,
      "model_forward_time": 5.584825038909912,
      "step": 14804
    },
    {
      "epoch": 1.4118194580078125e-05,
      "step": 14804,
      "training_step_time": 17.004130125045776
    },
    {
      "epoch": 1.4119148254394531e-05,
      "model_forward_time": 5.590005159378052,
      "step": 14805
    },
    {
      "epoch": 1.4119148254394531e-05,
      "step": 14805,
      "training_step_time": 17.0256085395813
    },
    {
      "epoch": 1.4120101928710937e-05,
      "model_forward_time": 5.951420783996582,
      "step": 14806
    },
    {
      "epoch": 1.4120101928710937e-05,
      "step": 14806,
      "training_step_time": 17.375977754592896
    },
    {
      "epoch": 1.4121055603027344e-05,
      "model_forward_time": 5.581674575805664,
      "step": 14807
    },
    {
      "epoch": 1.4121055603027344e-05,
      "step": 14807,
      "training_step_time": 17.01209807395935
    },
    {
      "epoch": 1.412200927734375e-05,
      "model_forward_time": 5.595848321914673,
      "step": 14808
    },
    {
      "epoch": 1.412200927734375e-05,
      "step": 14808,
      "training_step_time": 17.06612253189087
    },
    {
      "epoch": 1.4122962951660156e-05,
      "model_forward_time": 5.617231607437134,
      "step": 14809
    },
    {
      "epoch": 1.4122962951660156e-05,
      "step": 14809,
      "training_step_time": 17.13103985786438
    },
    {
      "epoch": 1.4123916625976563e-05,
      "grad_norm": 0.18145857751369476,
      "learning_rate": 1e-05,
      "loss": 0.0636,
      "step": 14810
    },
    {
      "epoch": 1.4123916625976563e-05,
      "model_forward_time": 5.637771844863892,
      "step": 14810
    },
    {
      "dynamics_loss_avg": 0.05776250902563333,
      "epoch": 1.4123916625976563e-05,
      "step": 14810
    },
    {
      "action_loss_avg": 0.006501452525844798,
      "epoch": 1.4123916625976563e-05,
      "step": 14810
    },
    {
      "epoch": 1.4123916625976563e-05,
      "step": 14810,
      "training_step_time": 17.154126405715942
    },
    {
      "epoch": 1.4124870300292969e-05,
      "model_forward_time": 5.616343975067139,
      "step": 14811
    },
    {
      "epoch": 1.4124870300292969e-05,
      "step": 14811,
      "training_step_time": 17.09428644180298
    },
    {
      "epoch": 1.4125823974609374e-05,
      "model_forward_time": 5.98830246925354,
      "step": 14812
    },
    {
      "epoch": 1.4125823974609374e-05,
      "step": 14812,
      "training_step_time": 17.41601061820984
    },
    {
      "epoch": 1.4126777648925782e-05,
      "model_forward_time": 5.58959698677063,
      "step": 14813
    },
    {
      "epoch": 1.4126777648925782e-05,
      "step": 14813,
      "training_step_time": 17.01837182044983
    },
    {
      "epoch": 1.4127731323242188e-05,
      "model_forward_time": 5.588058233261108,
      "step": 14814
    },
    {
      "epoch": 1.4127731323242188e-05,
      "step": 14814,
      "training_step_time": 17.01940631866455
    },
    {
      "epoch": 1.4128684997558593e-05,
      "model_forward_time": 5.593336343765259,
      "step": 14815
    },
    {
      "epoch": 1.4128684997558593e-05,
      "step": 14815,
      "training_step_time": 17.053449630737305
    },
    {
      "epoch": 1.4129638671875e-05,
      "model_forward_time": 5.60870623588562,
      "step": 14816
    },
    {
      "epoch": 1.4129638671875e-05,
      "step": 14816,
      "training_step_time": 17.08939528465271
    },
    {
      "epoch": 1.4130592346191406e-05,
      "model_forward_time": 5.615589380264282,
      "step": 14817
    },
    {
      "epoch": 1.4130592346191406e-05,
      "step": 14817,
      "training_step_time": 17.506205081939697
    },
    {
      "epoch": 1.4131546020507812e-05,
      "model_forward_time": 5.626275300979614,
      "step": 14818
    },
    {
      "epoch": 1.4131546020507812e-05,
      "step": 14818,
      "training_step_time": 17.136940002441406
    },
    {
      "epoch": 1.413249969482422e-05,
      "model_forward_time": 5.609087228775024,
      "step": 14819
    },
    {
      "epoch": 1.413249969482422e-05,
      "step": 14819,
      "training_step_time": 17.092140674591064
    },
    {
      "epoch": 1.4133453369140625e-05,
      "grad_norm": 1.0265058279037476,
      "learning_rate": 1e-05,
      "loss": 0.0602,
      "step": 14820
    },
    {
      "epoch": 1.4133453369140625e-05,
      "model_forward_time": 5.593115568161011,
      "step": 14820
    },
    {
      "dynamics_loss_avg": 0.05360101126134396,
      "epoch": 1.4133453369140625e-05,
      "step": 14820
    },
    {
      "action_loss_avg": 0.00534848147071898,
      "epoch": 1.4133453369140625e-05,
      "step": 14820
    },
    {
      "epoch": 1.4133453369140625e-05,
      "step": 14820,
      "training_step_time": 17.031747817993164
    },
    {
      "epoch": 1.4134407043457031e-05,
      "model_forward_time": 5.5949625968933105,
      "step": 14821
    },
    {
      "epoch": 1.4134407043457031e-05,
      "step": 14821,
      "training_step_time": 17.03439235687256
    },
    {
      "epoch": 1.4135360717773437e-05,
      "model_forward_time": 5.582206964492798,
      "step": 14822
    },
    {
      "epoch": 1.4135360717773437e-05,
      "step": 14822,
      "training_step_time": 17.00557231903076
    },
    {
      "epoch": 1.4136314392089844e-05,
      "model_forward_time": 5.96677827835083,
      "step": 14823
    },
    {
      "epoch": 1.4136314392089844e-05,
      "step": 14823,
      "training_step_time": 17.433547735214233
    },
    {
      "epoch": 1.413726806640625e-05,
      "model_forward_time": 5.625618934631348,
      "step": 14824
    },
    {
      "epoch": 1.413726806640625e-05,
      "step": 14824,
      "training_step_time": 17.110562801361084
    },
    {
      "epoch": 1.4138221740722656e-05,
      "model_forward_time": 5.609819650650024,
      "step": 14825
    },
    {
      "epoch": 1.4138221740722656e-05,
      "step": 14825,
      "training_step_time": 17.11101746559143
    },
    {
      "epoch": 1.4139175415039063e-05,
      "model_forward_time": 5.612351179122925,
      "step": 14826
    },
    {
      "epoch": 1.4139175415039063e-05,
      "step": 14826,
      "training_step_time": 17.10665535926819
    },
    {
      "epoch": 1.4140129089355469e-05,
      "model_forward_time": 5.604811906814575,
      "step": 14827
    },
    {
      "epoch": 1.4140129089355469e-05,
      "step": 14827,
      "training_step_time": 17.090239763259888
    },
    {
      "epoch": 1.4141082763671874e-05,
      "model_forward_time": 5.60857892036438,
      "step": 14828
    },
    {
      "epoch": 1.4141082763671874e-05,
      "step": 14828,
      "training_step_time": 17.43449568748474
    },
    {
      "epoch": 1.4142036437988282e-05,
      "model_forward_time": 5.608077764511108,
      "step": 14829
    },
    {
      "epoch": 1.4142036437988282e-05,
      "step": 14829,
      "training_step_time": 17.072433948516846
    },
    {
      "epoch": 1.4142990112304688e-05,
      "grad_norm": 0.536260724067688,
      "learning_rate": 1e-05,
      "loss": 0.0525,
      "step": 14830
    },
    {
      "epoch": 1.4142990112304688e-05,
      "model_forward_time": 5.611196517944336,
      "step": 14830
    },
    {
      "dynamics_loss_avg": 0.05337386215105653,
      "epoch": 1.4142990112304688e-05,
      "step": 14830
    },
    {
      "action_loss_avg": 0.005205456586554647,
      "epoch": 1.4142990112304688e-05,
      "step": 14830
    },
    {
      "epoch": 1.4142990112304688e-05,
      "step": 14830,
      "training_step_time": 17.081323385238647
    },
    {
      "epoch": 1.4143943786621093e-05,
      "model_forward_time": 5.5955986976623535,
      "step": 14831
    },
    {
      "epoch": 1.4143943786621093e-05,
      "step": 14831,
      "training_step_time": 17.06946611404419
    },
    {
      "epoch": 1.41448974609375e-05,
      "model_forward_time": 5.6049933433532715,
      "step": 14832
    },
    {
      "epoch": 1.41448974609375e-05,
      "step": 14832,
      "training_step_time": 17.075514554977417
    },
    {
      "epoch": 1.4145851135253906e-05,
      "model_forward_time": 5.601423025131226,
      "step": 14833
    },
    {
      "epoch": 1.4145851135253906e-05,
      "step": 14833,
      "training_step_time": 17.097331762313843
    },
    {
      "epoch": 1.4146804809570312e-05,
      "model_forward_time": 5.9883034229278564,
      "step": 14834
    },
    {
      "epoch": 1.4146804809570312e-05,
      "step": 14834,
      "training_step_time": 17.484673023223877
    },
    {
      "epoch": 1.414775848388672e-05,
      "model_forward_time": 5.615863561630249,
      "step": 14835
    },
    {
      "epoch": 1.414775848388672e-05,
      "step": 14835,
      "training_step_time": 17.09793448448181
    },
    {
      "epoch": 1.4148712158203125e-05,
      "model_forward_time": 5.609773635864258,
      "step": 14836
    },
    {
      "epoch": 1.4148712158203125e-05,
      "step": 14836,
      "training_step_time": 17.084937572479248
    },
    {
      "epoch": 1.4149665832519531e-05,
      "model_forward_time": 5.614490509033203,
      "step": 14837
    },
    {
      "epoch": 1.4149665832519531e-05,
      "step": 14837,
      "training_step_time": 17.10904812812805
    },
    {
      "epoch": 1.4150619506835937e-05,
      "model_forward_time": 5.612417936325073,
      "step": 14838
    },
    {
      "epoch": 1.4150619506835937e-05,
      "step": 14838,
      "training_step_time": 17.109670639038086
    },
    {
      "epoch": 1.4151573181152344e-05,
      "model_forward_time": 5.962673187255859,
      "step": 14839
    },
    {
      "epoch": 1.4151573181152344e-05,
      "step": 14839,
      "training_step_time": 17.437434911727905
    },
    {
      "epoch": 1.415252685546875e-05,
      "grad_norm": 1.1541509628295898,
      "learning_rate": 1e-05,
      "loss": 0.0552,
      "step": 14840
    },
    {
      "epoch": 1.415252685546875e-05,
      "model_forward_time": 5.614324331283569,
      "step": 14840
    },
    {
      "dynamics_loss_avg": 0.04702960997819901,
      "epoch": 1.415252685546875e-05,
      "step": 14840
    },
    {
      "action_loss_avg": 0.004807562846690416,
      "epoch": 1.415252685546875e-05,
      "step": 14840
    },
    {
      "epoch": 1.415252685546875e-05,
      "step": 14840,
      "training_step_time": 17.080923080444336
    },
    {
      "epoch": 1.4153480529785156e-05,
      "model_forward_time": 5.599323987960815,
      "step": 14841
    },
    {
      "epoch": 1.4153480529785156e-05,
      "step": 14841,
      "training_step_time": 17.041497230529785
    },
    {
      "epoch": 1.4154434204101563e-05,
      "model_forward_time": 5.588628053665161,
      "step": 14842
    },
    {
      "epoch": 1.4154434204101563e-05,
      "step": 14842,
      "training_step_time": 17.019638776779175
    },
    {
      "epoch": 1.4155387878417969e-05,
      "model_forward_time": 5.583235025405884,
      "step": 14843
    },
    {
      "epoch": 1.4155387878417969e-05,
      "step": 14843,
      "training_step_time": 17.032567024230957
    },
    {
      "epoch": 1.4156341552734374e-05,
      "model_forward_time": 5.602444410324097,
      "step": 14844
    },
    {
      "epoch": 1.4156341552734374e-05,
      "step": 14844,
      "training_step_time": 17.08665108680725
    },
    {
      "epoch": 1.4157295227050782e-05,
      "model_forward_time": 6.0026936531066895,
      "step": 14845
    },
    {
      "epoch": 1.4157295227050782e-05,
      "step": 14845,
      "training_step_time": 17.51190423965454
    },
    {
      "epoch": 1.4158248901367188e-05,
      "model_forward_time": 5.63164758682251,
      "step": 14846
    },
    {
      "epoch": 1.4158248901367188e-05,
      "step": 14846,
      "training_step_time": 17.1527316570282
    },
    {
      "epoch": 1.4159202575683593e-05,
      "model_forward_time": 5.608245849609375,
      "step": 14847
    },
    {
      "epoch": 1.4159202575683593e-05,
      "step": 14847,
      "training_step_time": 17.100346088409424
    },
    {
      "epoch": 1.416015625e-05,
      "model_forward_time": 5.605787515640259,
      "step": 14848
    },
    {
      "epoch": 1.416015625e-05,
      "step": 14848,
      "training_step_time": 17.074068069458008
    },
    {
      "epoch": 1.4161109924316406e-05,
      "model_forward_time": 5.605618238449097,
      "step": 14849
    },
    {
      "epoch": 1.4161109924316406e-05,
      "step": 14849,
      "training_step_time": 17.073639631271362
    },
    {
      "epoch": 1.4162063598632812e-05,
      "grad_norm": 0.8101246953010559,
      "learning_rate": 1e-05,
      "loss": 0.0601,
      "step": 14850
    },
    {
      "epoch": 1.4162063598632812e-05,
      "model_forward_time": 5.600576162338257,
      "step": 14850
    },
    {
      "dynamics_loss_avg": 0.05008528381586075,
      "epoch": 1.4162063598632812e-05,
      "step": 14850
    },
    {
      "action_loss_avg": 0.006595124234445393,
      "epoch": 1.4162063598632812e-05,
      "step": 14850
    },
    {
      "epoch": 1.4162063598632812e-05,
      "step": 14850,
      "training_step_time": 17.424866437911987
    },
    {
      "epoch": 1.416301727294922e-05,
      "model_forward_time": 5.5918309688568115,
      "step": 14851
    },
    {
      "epoch": 1.416301727294922e-05,
      "step": 14851,
      "training_step_time": 17.037293910980225
    },
    {
      "epoch": 1.4163970947265625e-05,
      "model_forward_time": 5.586695194244385,
      "step": 14852
    },
    {
      "epoch": 1.4163970947265625e-05,
      "step": 14852,
      "training_step_time": 17.05252695083618
    },
    {
      "epoch": 1.4164924621582031e-05,
      "model_forward_time": 5.6092610359191895,
      "step": 14853
    },
    {
      "epoch": 1.4164924621582031e-05,
      "step": 14853,
      "training_step_time": 17.109114408493042
    },
    {
      "epoch": 1.4165878295898437e-05,
      "model_forward_time": 5.621825695037842,
      "step": 14854
    },
    {
      "epoch": 1.4165878295898437e-05,
      "step": 14854,
      "training_step_time": 17.12285876274109
    },
    {
      "epoch": 1.4166831970214844e-05,
      "model_forward_time": 5.611627578735352,
      "step": 14855
    },
    {
      "epoch": 1.4166831970214844e-05,
      "step": 14855,
      "training_step_time": 17.09097981452942
    },
    {
      "epoch": 1.416778564453125e-05,
      "model_forward_time": 6.011088848114014,
      "step": 14856
    },
    {
      "epoch": 1.416778564453125e-05,
      "step": 14856,
      "training_step_time": 17.441338062286377
    },
    {
      "epoch": 1.4168739318847656e-05,
      "model_forward_time": 5.584040880203247,
      "step": 14857
    },
    {
      "epoch": 1.4168739318847656e-05,
      "step": 14857,
      "training_step_time": 17.021079778671265
    },
    {
      "epoch": 1.4169692993164063e-05,
      "model_forward_time": 5.598703384399414,
      "step": 14858
    },
    {
      "epoch": 1.4169692993164063e-05,
      "step": 14858,
      "training_step_time": 17.05994987487793
    },
    {
      "epoch": 1.4170646667480469e-05,
      "model_forward_time": 5.612773418426514,
      "step": 14859
    },
    {
      "epoch": 1.4170646667480469e-05,
      "step": 14859,
      "training_step_time": 17.11057186126709
    },
    {
      "epoch": 1.4171600341796874e-05,
      "grad_norm": 0.9626468420028687,
      "learning_rate": 1e-05,
      "loss": 0.0602,
      "step": 14860
    },
    {
      "epoch": 1.4171600341796874e-05,
      "model_forward_time": 5.614694833755493,
      "step": 14860
    },
    {
      "dynamics_loss_avg": 0.05366103360429406,
      "epoch": 1.4171600341796874e-05,
      "step": 14860
    },
    {
      "action_loss_avg": 0.007053417706629261,
      "epoch": 1.4171600341796874e-05,
      "step": 14860
    },
    {
      "epoch": 1.4171600341796874e-05,
      "step": 14860,
      "training_step_time": 17.097676515579224
    },
    {
      "epoch": 1.4172554016113282e-05,
      "model_forward_time": 5.59460973739624,
      "step": 14861
    },
    {
      "epoch": 1.4172554016113282e-05,
      "step": 14861,
      "training_step_time": 17.42576503753662
    },
    {
      "epoch": 1.4173507690429688e-05,
      "model_forward_time": 5.5843400955200195,
      "step": 14862
    },
    {
      "epoch": 1.4173507690429688e-05,
      "step": 14862,
      "training_step_time": 17.03074598312378
    },
    {
      "epoch": 1.4174461364746093e-05,
      "model_forward_time": 5.58479905128479,
      "step": 14863
    },
    {
      "epoch": 1.4174461364746093e-05,
      "step": 14863,
      "training_step_time": 17.051422357559204
    },
    {
      "epoch": 1.41754150390625e-05,
      "model_forward_time": 5.6086952686309814,
      "step": 14864
    },
    {
      "epoch": 1.41754150390625e-05,
      "step": 14864,
      "training_step_time": 17.117725372314453
    },
    {
      "epoch": 1.4176368713378906e-05,
      "model_forward_time": 5.632927179336548,
      "step": 14865
    },
    {
      "epoch": 1.4176368713378906e-05,
      "step": 14865,
      "training_step_time": 17.14440393447876
    },
    {
      "epoch": 1.4177322387695312e-05,
      "model_forward_time": 5.620036840438843,
      "step": 14866
    },
    {
      "epoch": 1.4177322387695312e-05,
      "step": 14866,
      "training_step_time": 17.100860118865967
    },
    {
      "epoch": 1.417827606201172e-05,
      "model_forward_time": 5.997344493865967,
      "step": 14867
    },
    {
      "epoch": 1.417827606201172e-05,
      "step": 14867,
      "training_step_time": 17.480252981185913
    },
    {
      "epoch": 1.4179229736328125e-05,
      "model_forward_time": 5.622156381607056,
      "step": 14868
    },
    {
      "epoch": 1.4179229736328125e-05,
      "step": 14868,
      "training_step_time": 17.094515085220337
    },
    {
      "epoch": 1.4180183410644531e-05,
      "model_forward_time": 5.601603031158447,
      "step": 14869
    },
    {
      "epoch": 1.4180183410644531e-05,
      "step": 14869,
      "training_step_time": 17.074107885360718
    },
    {
      "epoch": 1.4181137084960937e-05,
      "grad_norm": 0.3788698613643646,
      "learning_rate": 1e-05,
      "loss": 0.0515,
      "step": 14870
    },
    {
      "epoch": 1.4181137084960937e-05,
      "model_forward_time": 5.60349440574646,
      "step": 14870
    },
    {
      "dynamics_loss_avg": 0.04726300463080406,
      "epoch": 1.4181137084960937e-05,
      "step": 14870
    },
    {
      "action_loss_avg": 0.005814726417884231,
      "epoch": 1.4181137084960937e-05,
      "step": 14870
    },
    {
      "epoch": 1.4181137084960937e-05,
      "step": 14870,
      "training_step_time": 17.053006649017334
    },
    {
      "epoch": 1.4182090759277344e-05,
      "model_forward_time": 5.5875244140625,
      "step": 14871
    },
    {
      "epoch": 1.4182090759277344e-05,
      "step": 14871,
      "training_step_time": 17.03765320777893
    },
    {
      "epoch": 1.418304443359375e-05,
      "model_forward_time": 5.963380575180054,
      "step": 14872
    },
    {
      "epoch": 1.418304443359375e-05,
      "step": 14872,
      "training_step_time": 17.463280200958252
    },
    {
      "epoch": 1.4183998107910156e-05,
      "model_forward_time": 5.619950771331787,
      "step": 14873
    },
    {
      "epoch": 1.4183998107910156e-05,
      "step": 14873,
      "training_step_time": 17.14209008216858
    },
    {
      "epoch": 1.4184951782226563e-05,
      "model_forward_time": 5.628809928894043,
      "step": 14874
    },
    {
      "epoch": 1.4184951782226563e-05,
      "step": 14874,
      "training_step_time": 17.148088932037354
    },
    {
      "epoch": 1.4185905456542969e-05,
      "model_forward_time": 5.612499952316284,
      "step": 14875
    },
    {
      "epoch": 1.4185905456542969e-05,
      "step": 14875,
      "training_step_time": 17.08454990386963
    },
    {
      "epoch": 1.4186859130859374e-05,
      "model_forward_time": 5.60226035118103,
      "step": 14876
    },
    {
      "epoch": 1.4186859130859374e-05,
      "step": 14876,
      "training_step_time": 17.05136799812317
    },
    {
      "epoch": 1.4187812805175782e-05,
      "model_forward_time": 5.595295429229736,
      "step": 14877
    },
    {
      "epoch": 1.4187812805175782e-05,
      "step": 14877,
      "training_step_time": 17.030338048934937
    },
    {
      "epoch": 1.4188766479492188e-05,
      "model_forward_time": 5.973291397094727,
      "step": 14878
    },
    {
      "epoch": 1.4188766479492188e-05,
      "step": 14878,
      "training_step_time": 17.452490091323853
    },
    {
      "epoch": 1.4189720153808593e-05,
      "model_forward_time": 5.607813835144043,
      "step": 14879
    },
    {
      "epoch": 1.4189720153808593e-05,
      "step": 14879,
      "training_step_time": 17.11128044128418
    },
    {
      "epoch": 1.4190673828125e-05,
      "grad_norm": 1.0655912160873413,
      "learning_rate": 1e-05,
      "loss": 0.0678,
      "step": 14880
    },
    {
      "epoch": 1.4190673828125e-05,
      "model_forward_time": 5.631201505661011,
      "step": 14880
    },
    {
      "dynamics_loss_avg": 0.05855861641466618,
      "epoch": 1.4190673828125e-05,
      "step": 14880
    },
    {
      "action_loss_avg": 0.008306835335679353,
      "epoch": 1.4190673828125e-05,
      "step": 14880
    },
    {
      "epoch": 1.4190673828125e-05,
      "step": 14880,
      "training_step_time": 17.126076698303223
    },
    {
      "epoch": 1.4191627502441406e-05,
      "model_forward_time": 5.5913848876953125,
      "step": 14881
    },
    {
      "epoch": 1.4191627502441406e-05,
      "step": 14881,
      "training_step_time": 17.03951382637024
    },
    {
      "epoch": 1.4192581176757812e-05,
      "model_forward_time": 5.577773094177246,
      "step": 14882
    },
    {
      "epoch": 1.4192581176757812e-05,
      "step": 14882,
      "training_step_time": 17.00109839439392
    },
    {
      "epoch": 1.4193534851074218e-05,
      "model_forward_time": 5.576844930648804,
      "step": 14883
    },
    {
      "epoch": 1.4193534851074218e-05,
      "step": 14883,
      "training_step_time": 17.347105741500854
    },
    {
      "epoch": 1.4194488525390625e-05,
      "model_forward_time": 5.57738184928894,
      "step": 14884
    },
    {
      "epoch": 1.4194488525390625e-05,
      "step": 14884,
      "training_step_time": 16.986637830734253
    },
    {
      "epoch": 1.4195442199707031e-05,
      "model_forward_time": 5.5749006271362305,
      "step": 14885
    },
    {
      "epoch": 1.4195442199707031e-05,
      "step": 14885,
      "training_step_time": 17.02446937561035
    },
    {
      "epoch": 1.4196395874023437e-05,
      "model_forward_time": 5.585253477096558,
      "step": 14886
    },
    {
      "epoch": 1.4196395874023437e-05,
      "step": 14886,
      "training_step_time": 17.017252683639526
    },
    {
      "epoch": 1.4197349548339844e-05,
      "model_forward_time": 5.587182283401489,
      "step": 14887
    },
    {
      "epoch": 1.4197349548339844e-05,
      "step": 14887,
      "training_step_time": 17.018654346466064
    },
    {
      "epoch": 1.419830322265625e-05,
      "model_forward_time": 5.597639083862305,
      "step": 14888
    },
    {
      "epoch": 1.419830322265625e-05,
      "step": 14888,
      "training_step_time": 17.08084225654602
    },
    {
      "epoch": 1.4199256896972656e-05,
      "model_forward_time": 6.00825572013855,
      "step": 14889
    },
    {
      "epoch": 1.4199256896972656e-05,
      "step": 14889,
      "training_step_time": 17.52744746208191
    },
    {
      "epoch": 1.4200210571289063e-05,
      "grad_norm": 0.8137275576591492,
      "learning_rate": 1e-05,
      "loss": 0.0537,
      "step": 14890
    },
    {
      "epoch": 1.4200210571289063e-05,
      "model_forward_time": 5.626216888427734,
      "step": 14890
    },
    {
      "dynamics_loss_avg": 0.05250636022537947,
      "epoch": 1.4200210571289063e-05,
      "step": 14890
    },
    {
      "action_loss_avg": 0.0068662874167785045,
      "epoch": 1.4200210571289063e-05,
      "step": 14890
    },
    {
      "epoch": 1.4200210571289063e-05,
      "step": 14890,
      "training_step_time": 17.117278575897217
    },
    {
      "epoch": 1.4201164245605469e-05,
      "model_forward_time": 5.602087497711182,
      "step": 14891
    },
    {
      "epoch": 1.4201164245605469e-05,
      "step": 14891,
      "training_step_time": 17.068345546722412
    },
    {
      "epoch": 1.4202117919921874e-05,
      "model_forward_time": 5.583737373352051,
      "step": 14892
    },
    {
      "epoch": 1.4202117919921874e-05,
      "step": 14892,
      "training_step_time": 17.021515369415283
    },
    {
      "epoch": 1.4203071594238282e-05,
      "model_forward_time": 5.586331129074097,
      "step": 14893
    },
    {
      "epoch": 1.4203071594238282e-05,
      "step": 14893,
      "training_step_time": 17.031461715698242
    },
    {
      "epoch": 1.4204025268554688e-05,
      "model_forward_time": 5.58996319770813,
      "step": 14894
    },
    {
      "epoch": 1.4204025268554688e-05,
      "step": 14894,
      "training_step_time": 17.40350842475891
    },
    {
      "epoch": 1.4204978942871093e-05,
      "model_forward_time": 5.597890853881836,
      "step": 14895
    },
    {
      "epoch": 1.4204978942871093e-05,
      "step": 14895,
      "training_step_time": 17.05797052383423
    },
    {
      "epoch": 1.42059326171875e-05,
      "model_forward_time": 5.595044851303101,
      "step": 14896
    },
    {
      "epoch": 1.42059326171875e-05,
      "step": 14896,
      "training_step_time": 17.02567768096924
    },
    {
      "epoch": 1.4206886291503906e-05,
      "model_forward_time": 5.599879741668701,
      "step": 14897
    },
    {
      "epoch": 1.4206886291503906e-05,
      "step": 14897,
      "training_step_time": 17.07470178604126
    },
    {
      "epoch": 1.4207839965820312e-05,
      "model_forward_time": 5.599940538406372,
      "step": 14898
    },
    {
      "epoch": 1.4207839965820312e-05,
      "step": 14898,
      "training_step_time": 17.072715044021606
    },
    {
      "epoch": 1.4208793640136718e-05,
      "model_forward_time": 5.611925840377808,
      "step": 14899
    },
    {
      "epoch": 1.4208793640136718e-05,
      "step": 14899,
      "training_step_time": 17.103995084762573
    },
    {
      "epoch": 1.4209747314453125e-05,
      "grad_norm": 1.7704609632492065,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 14900
    },
    {
      "epoch": 1.4209747314453125e-05,
      "model_forward_time": 6.000070095062256,
      "step": 14900
    },
    {
      "dynamics_loss_avg": 0.054970862343907356,
      "epoch": 1.4209747314453125e-05,
      "step": 14900
    },
    {
      "action_loss_avg": 0.006588115799240768,
      "epoch": 1.4209747314453125e-05,
      "step": 14900
    },
    {
      "epoch": 1.4209747314453125e-05,
      "step": 14900,
      "training_step_time": 17.48158860206604
    },
    {
      "epoch": 1.4210700988769531e-05,
      "model_forward_time": 5.602757930755615,
      "step": 14901
    },
    {
      "epoch": 1.4210700988769531e-05,
      "step": 14901,
      "training_step_time": 17.05836248397827
    },
    {
      "epoch": 1.4211654663085937e-05,
      "model_forward_time": 5.594112157821655,
      "step": 14902
    },
    {
      "epoch": 1.4211654663085937e-05,
      "step": 14902,
      "training_step_time": 17.07420802116394
    },
    {
      "epoch": 1.4212608337402344e-05,
      "model_forward_time": 5.600930690765381,
      "step": 14903
    },
    {
      "epoch": 1.4212608337402344e-05,
      "step": 14903,
      "training_step_time": 17.089072942733765
    },
    {
      "epoch": 1.421356201171875e-05,
      "model_forward_time": 5.605908393859863,
      "step": 14904
    },
    {
      "epoch": 1.421356201171875e-05,
      "step": 14904,
      "training_step_time": 17.06343126296997
    },
    {
      "epoch": 1.4214515686035156e-05,
      "model_forward_time": 5.954969882965088,
      "step": 14905
    },
    {
      "epoch": 1.4214515686035156e-05,
      "step": 14905,
      "training_step_time": 17.409008264541626
    },
    {
      "epoch": 1.4215469360351563e-05,
      "model_forward_time": 5.600549221038818,
      "step": 14906
    },
    {
      "epoch": 1.4215469360351563e-05,
      "step": 14906,
      "training_step_time": 17.074820518493652
    },
    {
      "epoch": 1.4216423034667969e-05,
      "model_forward_time": 5.599790811538696,
      "step": 14907
    },
    {
      "epoch": 1.4216423034667969e-05,
      "step": 14907,
      "training_step_time": 17.05675172805786
    },
    {
      "epoch": 1.4217376708984374e-05,
      "model_forward_time": 5.6069817543029785,
      "step": 14908
    },
    {
      "epoch": 1.4217376708984374e-05,
      "step": 14908,
      "training_step_time": 17.054688930511475
    },
    {
      "epoch": 1.4218330383300782e-05,
      "model_forward_time": 5.598718166351318,
      "step": 14909
    },
    {
      "epoch": 1.4218330383300782e-05,
      "step": 14909,
      "training_step_time": 17.072182416915894
    },
    {
      "epoch": 1.4219284057617188e-05,
      "grad_norm": 0.5706639885902405,
      "learning_rate": 1e-05,
      "loss": 0.0462,
      "step": 14910
    },
    {
      "epoch": 1.4219284057617188e-05,
      "model_forward_time": 5.603033781051636,
      "step": 14910
    },
    {
      "dynamics_loss_avg": 0.04666897170245647,
      "epoch": 1.4219284057617188e-05,
      "step": 14910
    },
    {
      "action_loss_avg": 0.004923098371364176,
      "epoch": 1.4219284057617188e-05,
      "step": 14910
    },
    {
      "epoch": 1.4219284057617188e-05,
      "step": 14910,
      "training_step_time": 17.088820934295654
    },
    {
      "epoch": 1.4220237731933593e-05,
      "model_forward_time": 5.976231098175049,
      "step": 14911
    },
    {
      "epoch": 1.4220237731933593e-05,
      "step": 14911,
      "training_step_time": 17.452425241470337
    },
    {
      "epoch": 1.422119140625e-05,
      "model_forward_time": 5.599725246429443,
      "step": 14912
    },
    {
      "epoch": 1.422119140625e-05,
      "step": 14912,
      "training_step_time": 17.054298162460327
    },
    {
      "epoch": 1.4222145080566406e-05,
      "model_forward_time": 5.5850560665130615,
      "step": 14913
    },
    {
      "epoch": 1.4222145080566406e-05,
      "step": 14913,
      "training_step_time": 17.016512155532837
    },
    {
      "epoch": 1.4223098754882812e-05,
      "model_forward_time": 5.595896005630493,
      "step": 14914
    },
    {
      "epoch": 1.4223098754882812e-05,
      "step": 14914,
      "training_step_time": 17.059342861175537
    },
    {
      "epoch": 1.4224052429199218e-05,
      "model_forward_time": 5.60066294670105,
      "step": 14915
    },
    {
      "epoch": 1.4224052429199218e-05,
      "step": 14915,
      "training_step_time": 17.06477189064026
    },
    {
      "epoch": 1.4225006103515625e-05,
      "model_forward_time": 5.603837251663208,
      "step": 14916
    },
    {
      "epoch": 1.4225006103515625e-05,
      "step": 14916,
      "training_step_time": 17.45817756652832
    },
    {
      "epoch": 1.4225959777832031e-05,
      "model_forward_time": 5.6086695194244385,
      "step": 14917
    },
    {
      "epoch": 1.4225959777832031e-05,
      "step": 14917,
      "training_step_time": 17.084200859069824
    },
    {
      "epoch": 1.4226913452148437e-05,
      "model_forward_time": 5.595300197601318,
      "step": 14918
    },
    {
      "epoch": 1.4226913452148437e-05,
      "step": 14918,
      "training_step_time": 17.036832332611084
    },
    {
      "epoch": 1.4227867126464844e-05,
      "model_forward_time": 5.589509725570679,
      "step": 14919
    },
    {
      "epoch": 1.4227867126464844e-05,
      "step": 14919,
      "training_step_time": 17.027791023254395
    },
    {
      "epoch": 1.422882080078125e-05,
      "grad_norm": 0.28395378589630127,
      "learning_rate": 1e-05,
      "loss": 0.0571,
      "step": 14920
    },
    {
      "epoch": 1.422882080078125e-05,
      "model_forward_time": 5.572602987289429,
      "step": 14920
    },
    {
      "dynamics_loss_avg": 0.05396074242889881,
      "epoch": 1.422882080078125e-05,
      "step": 14920
    },
    {
      "action_loss_avg": 0.00641749199712649,
      "epoch": 1.422882080078125e-05,
      "step": 14920
    },
    {
      "epoch": 1.422882080078125e-05,
      "step": 14920,
      "training_step_time": 16.979112148284912
    },
    {
      "epoch": 1.4229774475097656e-05,
      "model_forward_time": 5.580057144165039,
      "step": 14921
    },
    {
      "epoch": 1.4229774475097656e-05,
      "step": 14921,
      "training_step_time": 17.028718948364258
    },
    {
      "epoch": 1.4230728149414063e-05,
      "model_forward_time": 5.980603456497192,
      "step": 14922
    },
    {
      "epoch": 1.4230728149414063e-05,
      "step": 14922,
      "training_step_time": 17.4339759349823
    },
    {
      "epoch": 1.4231681823730469e-05,
      "model_forward_time": 5.609574317932129,
      "step": 14923
    },
    {
      "epoch": 1.4231681823730469e-05,
      "step": 14923,
      "training_step_time": 17.074528217315674
    },
    {
      "epoch": 1.4232635498046875e-05,
      "model_forward_time": 5.609379768371582,
      "step": 14924
    },
    {
      "epoch": 1.4232635498046875e-05,
      "step": 14924,
      "training_step_time": 17.106394290924072
    },
    {
      "epoch": 1.4233589172363282e-05,
      "model_forward_time": 5.619095325469971,
      "step": 14925
    },
    {
      "epoch": 1.4233589172363282e-05,
      "step": 14925,
      "training_step_time": 17.123785972595215
    },
    {
      "epoch": 1.4234542846679688e-05,
      "model_forward_time": 5.608983278274536,
      "step": 14926
    },
    {
      "epoch": 1.4234542846679688e-05,
      "step": 14926,
      "training_step_time": 17.1004159450531
    },
    {
      "epoch": 1.4235496520996093e-05,
      "model_forward_time": 5.612417221069336,
      "step": 14927
    },
    {
      "epoch": 1.4235496520996093e-05,
      "step": 14927,
      "training_step_time": 17.470735549926758
    },
    {
      "epoch": 1.42364501953125e-05,
      "model_forward_time": 5.6105101108551025,
      "step": 14928
    },
    {
      "epoch": 1.42364501953125e-05,
      "step": 14928,
      "training_step_time": 17.114654302597046
    },
    {
      "epoch": 1.4237403869628907e-05,
      "model_forward_time": 5.611782550811768,
      "step": 14929
    },
    {
      "epoch": 1.4237403869628907e-05,
      "step": 14929,
      "training_step_time": 17.09412455558777
    },
    {
      "epoch": 1.4238357543945312e-05,
      "grad_norm": 1.065645456314087,
      "learning_rate": 1e-05,
      "loss": 0.0675,
      "step": 14930
    },
    {
      "epoch": 1.4238357543945312e-05,
      "model_forward_time": 5.6184401512146,
      "step": 14930
    },
    {
      "dynamics_loss_avg": 0.054956112056970596,
      "epoch": 1.4238357543945312e-05,
      "step": 14930
    },
    {
      "action_loss_avg": 0.008771605836227537,
      "epoch": 1.4238357543945312e-05,
      "step": 14930
    },
    {
      "epoch": 1.4238357543945312e-05,
      "step": 14930,
      "training_step_time": 17.110470294952393
    },
    {
      "epoch": 1.4239311218261718e-05,
      "model_forward_time": 5.606368780136108,
      "step": 14931
    },
    {
      "epoch": 1.4239311218261718e-05,
      "step": 14931,
      "training_step_time": 17.0805561542511
    },
    {
      "epoch": 1.4240264892578125e-05,
      "model_forward_time": 5.608503103256226,
      "step": 14932
    },
    {
      "epoch": 1.4240264892578125e-05,
      "step": 14932,
      "training_step_time": 17.10713529586792
    },
    {
      "epoch": 1.4241218566894531e-05,
      "model_forward_time": 5.984596252441406,
      "step": 14933
    },
    {
      "epoch": 1.4241218566894531e-05,
      "step": 14933,
      "training_step_time": 17.473185300827026
    },
    {
      "epoch": 1.4242172241210937e-05,
      "model_forward_time": 5.620413541793823,
      "step": 14934
    },
    {
      "epoch": 1.4242172241210937e-05,
      "step": 14934,
      "training_step_time": 17.119911432266235
    },
    {
      "epoch": 1.4243125915527344e-05,
      "model_forward_time": 5.622478723526001,
      "step": 14935
    },
    {
      "epoch": 1.4243125915527344e-05,
      "step": 14935,
      "training_step_time": 17.13138747215271
    },
    {
      "epoch": 1.424407958984375e-05,
      "model_forward_time": 5.6172356605529785,
      "step": 14936
    },
    {
      "epoch": 1.424407958984375e-05,
      "step": 14936,
      "training_step_time": 17.09614109992981
    },
    {
      "epoch": 1.4245033264160156e-05,
      "model_forward_time": 5.600031137466431,
      "step": 14937
    },
    {
      "epoch": 1.4245033264160156e-05,
      "step": 14937,
      "training_step_time": 17.0346896648407
    },
    {
      "epoch": 1.4245986938476563e-05,
      "model_forward_time": 5.950675010681152,
      "step": 14938
    },
    {
      "epoch": 1.4245986938476563e-05,
      "step": 14938,
      "training_step_time": 17.39342451095581
    },
    {
      "epoch": 1.4246940612792969e-05,
      "model_forward_time": 5.584085941314697,
      "step": 14939
    },
    {
      "epoch": 1.4246940612792969e-05,
      "step": 14939,
      "training_step_time": 17.038544416427612
    },
    {
      "epoch": 1.4247894287109375e-05,
      "grad_norm": 1.1243624687194824,
      "learning_rate": 1e-05,
      "loss": 0.0731,
      "step": 14940
    },
    {
      "epoch": 1.4247894287109375e-05,
      "model_forward_time": 5.5994343757629395,
      "step": 14940
    },
    {
      "dynamics_loss_avg": 0.06783895138651133,
      "epoch": 1.4247894287109375e-05,
      "step": 14940
    },
    {
      "action_loss_avg": 0.009277523623313754,
      "epoch": 1.4247894287109375e-05,
      "step": 14940
    },
    {
      "epoch": 1.4247894287109375e-05,
      "step": 14940,
      "training_step_time": 17.06363606452942
    },
    {
      "epoch": 1.4248847961425782e-05,
      "model_forward_time": 5.602829694747925,
      "step": 14941
    },
    {
      "epoch": 1.4248847961425782e-05,
      "step": 14941,
      "training_step_time": 17.059837102890015
    },
    {
      "epoch": 1.4249801635742188e-05,
      "model_forward_time": 5.6040802001953125,
      "step": 14942
    },
    {
      "epoch": 1.4249801635742188e-05,
      "step": 14942,
      "training_step_time": 17.06782102584839
    },
    {
      "epoch": 1.4250755310058593e-05,
      "model_forward_time": 5.607000112533569,
      "step": 14943
    },
    {
      "epoch": 1.4250755310058593e-05,
      "step": 14943,
      "training_step_time": 17.077627897262573
    },
    {
      "epoch": 1.4251708984375e-05,
      "model_forward_time": 5.999396085739136,
      "step": 14944
    },
    {
      "epoch": 1.4251708984375e-05,
      "step": 14944,
      "training_step_time": 17.46731972694397
    },
    {
      "epoch": 1.4252662658691407e-05,
      "model_forward_time": 5.5985987186431885,
      "step": 14945
    },
    {
      "epoch": 1.4252662658691407e-05,
      "step": 14945,
      "training_step_time": 17.041645526885986
    },
    {
      "epoch": 1.4253616333007812e-05,
      "model_forward_time": 5.5892674922943115,
      "step": 14946
    },
    {
      "epoch": 1.4253616333007812e-05,
      "step": 14946,
      "training_step_time": 17.02235221862793
    },
    {
      "epoch": 1.4254570007324218e-05,
      "model_forward_time": 5.599985361099243,
      "step": 14947
    },
    {
      "epoch": 1.4254570007324218e-05,
      "step": 14947,
      "training_step_time": 17.065099477767944
    },
    {
      "epoch": 1.4255523681640625e-05,
      "model_forward_time": 5.601178884506226,
      "step": 14948
    },
    {
      "epoch": 1.4255523681640625e-05,
      "step": 14948,
      "training_step_time": 17.060115337371826
    },
    {
      "epoch": 1.4256477355957031e-05,
      "model_forward_time": 5.605303525924683,
      "step": 14949
    },
    {
      "epoch": 1.4256477355957031e-05,
      "step": 14949,
      "training_step_time": 17.443126440048218
    },
    {
      "epoch": 1.4257431030273437e-05,
      "grad_norm": 0.7336200475692749,
      "learning_rate": 1e-05,
      "loss": 0.0711,
      "step": 14950
    },
    {
      "epoch": 1.4257431030273437e-05,
      "model_forward_time": 5.603941440582275,
      "step": 14950
    },
    {
      "dynamics_loss_avg": 0.05904489923268556,
      "epoch": 1.4257431030273437e-05,
      "step": 14950
    },
    {
      "action_loss_avg": 0.00868268650956452,
      "epoch": 1.4257431030273437e-05,
      "step": 14950
    },
    {
      "epoch": 1.4257431030273437e-05,
      "step": 14950,
      "training_step_time": 17.07452154159546
    },
    {
      "epoch": 1.4258384704589844e-05,
      "model_forward_time": 5.599377155303955,
      "step": 14951
    },
    {
      "epoch": 1.4258384704589844e-05,
      "step": 14951,
      "training_step_time": 17.019267082214355
    },
    {
      "epoch": 1.425933837890625e-05,
      "model_forward_time": 5.5945868492126465,
      "step": 14952
    },
    {
      "epoch": 1.425933837890625e-05,
      "step": 14952,
      "training_step_time": 17.036255359649658
    },
    {
      "epoch": 1.4260292053222656e-05,
      "model_forward_time": 5.594365119934082,
      "step": 14953
    },
    {
      "epoch": 1.4260292053222656e-05,
      "step": 14953,
      "training_step_time": 17.055219888687134
    },
    {
      "epoch": 1.4261245727539063e-05,
      "model_forward_time": 5.603747367858887,
      "step": 14954
    },
    {
      "epoch": 1.4261245727539063e-05,
      "step": 14954,
      "training_step_time": 17.06775736808777
    },
    {
      "epoch": 1.4262199401855469e-05,
      "model_forward_time": 6.006563425064087,
      "step": 14955
    },
    {
      "epoch": 1.4262199401855469e-05,
      "step": 14955,
      "training_step_time": 17.472034692764282
    },
    {
      "epoch": 1.4263153076171875e-05,
      "model_forward_time": 5.606454133987427,
      "step": 14956
    },
    {
      "epoch": 1.4263153076171875e-05,
      "step": 14956,
      "training_step_time": 17.066501140594482
    },
    {
      "epoch": 1.4264106750488282e-05,
      "model_forward_time": 5.61112117767334,
      "step": 14957
    },
    {
      "epoch": 1.4264106750488282e-05,
      "step": 14957,
      "training_step_time": 17.087424993515015
    },
    {
      "epoch": 1.4265060424804688e-05,
      "model_forward_time": 5.6224541664123535,
      "step": 14958
    },
    {
      "epoch": 1.4265060424804688e-05,
      "step": 14958,
      "training_step_time": 17.111770153045654
    },
    {
      "epoch": 1.4266014099121093e-05,
      "model_forward_time": 5.61677098274231,
      "step": 14959
    },
    {
      "epoch": 1.4266014099121093e-05,
      "step": 14959,
      "training_step_time": 17.10991382598877
    },
    {
      "epoch": 1.42669677734375e-05,
      "grad_norm": 0.7913819551467896,
      "learning_rate": 1e-05,
      "loss": 0.0819,
      "step": 14960
    },
    {
      "epoch": 1.42669677734375e-05,
      "model_forward_time": 5.6070122718811035,
      "step": 14960
    },
    {
      "dynamics_loss_avg": 0.07088255956768989,
      "epoch": 1.42669677734375e-05,
      "step": 14960
    },
    {
      "action_loss_avg": 0.010399318835698067,
      "epoch": 1.42669677734375e-05,
      "step": 14960
    },
    {
      "epoch": 1.42669677734375e-05,
      "step": 14960,
      "training_step_time": 17.464813947677612
    },
    {
      "epoch": 1.4267921447753907e-05,
      "model_forward_time": 5.621882438659668,
      "step": 14961
    },
    {
      "epoch": 1.4267921447753907e-05,
      "step": 14961,
      "training_step_time": 17.095101356506348
    },
    {
      "epoch": 1.4268875122070312e-05,
      "model_forward_time": 5.606786012649536,
      "step": 14962
    },
    {
      "epoch": 1.4268875122070312e-05,
      "step": 14962,
      "training_step_time": 17.08947253227234
    },
    {
      "epoch": 1.4269828796386718e-05,
      "model_forward_time": 5.600310802459717,
      "step": 14963
    },
    {
      "epoch": 1.4269828796386718e-05,
      "step": 14963,
      "training_step_time": 17.061357975006104
    },
    {
      "epoch": 1.4270782470703125e-05,
      "model_forward_time": 5.608963966369629,
      "step": 14964
    },
    {
      "epoch": 1.4270782470703125e-05,
      "step": 14964,
      "training_step_time": 17.07124900817871
    },
    {
      "epoch": 1.4271736145019531e-05,
      "model_forward_time": 5.603795051574707,
      "step": 14965
    },
    {
      "epoch": 1.4271736145019531e-05,
      "step": 14965,
      "training_step_time": 17.060883283615112
    },
    {
      "epoch": 1.4272689819335937e-05,
      "model_forward_time": 5.990638017654419,
      "step": 14966
    },
    {
      "epoch": 1.4272689819335937e-05,
      "step": 14966,
      "training_step_time": 17.471757411956787
    },
    {
      "epoch": 1.4273643493652344e-05,
      "model_forward_time": 5.5984437465667725,
      "step": 14967
    },
    {
      "epoch": 1.4273643493652344e-05,
      "step": 14967,
      "training_step_time": 17.06764507293701
    },
    {
      "epoch": 1.427459716796875e-05,
      "model_forward_time": 5.602518320083618,
      "step": 14968
    },
    {
      "epoch": 1.427459716796875e-05,
      "step": 14968,
      "training_step_time": 17.07626748085022
    },
    {
      "epoch": 1.4275550842285156e-05,
      "model_forward_time": 5.607883930206299,
      "step": 14969
    },
    {
      "epoch": 1.4275550842285156e-05,
      "step": 14969,
      "training_step_time": 17.059886693954468
    },
    {
      "epoch": 1.4276504516601563e-05,
      "grad_norm": 1.1813136339187622,
      "learning_rate": 1e-05,
      "loss": 0.0703,
      "step": 14970
    },
    {
      "epoch": 1.4276504516601563e-05,
      "model_forward_time": 5.5956621170043945,
      "step": 14970
    },
    {
      "dynamics_loss_avg": 0.05804577749222517,
      "epoch": 1.4276504516601563e-05,
      "step": 14970
    },
    {
      "action_loss_avg": 0.007073700870387256,
      "epoch": 1.4276504516601563e-05,
      "step": 14970
    },
    {
      "epoch": 1.4276504516601563e-05,
      "step": 14970,
      "training_step_time": 17.04491424560547
    },
    {
      "epoch": 1.4277458190917969e-05,
      "model_forward_time": 5.953091382980347,
      "step": 14971
    },
    {
      "epoch": 1.4277458190917969e-05,
      "step": 14971,
      "training_step_time": 17.377633333206177
    },
    {
      "epoch": 1.4278411865234375e-05,
      "model_forward_time": 5.594023942947388,
      "step": 14972
    },
    {
      "epoch": 1.4278411865234375e-05,
      "step": 14972,
      "training_step_time": 17.04093599319458
    },
    {
      "epoch": 1.4279365539550782e-05,
      "model_forward_time": 5.610302686691284,
      "step": 14973
    },
    {
      "epoch": 1.4279365539550782e-05,
      "step": 14973,
      "training_step_time": 17.12330651283264
    },
    {
      "epoch": 1.4280319213867188e-05,
      "model_forward_time": 5.622864484786987,
      "step": 14974
    },
    {
      "epoch": 1.4280319213867188e-05,
      "step": 14974,
      "training_step_time": 17.117143869400024
    },
    {
      "epoch": 1.4281272888183593e-05,
      "model_forward_time": 5.5992255210876465,
      "step": 14975
    },
    {
      "epoch": 1.4281272888183593e-05,
      "step": 14975,
      "training_step_time": 17.06068778038025
    },
    {
      "epoch": 1.42822265625e-05,
      "model_forward_time": 5.5969398021698,
      "step": 14976
    },
    {
      "epoch": 1.42822265625e-05,
      "step": 14976,
      "training_step_time": 17.062251567840576
    },
    {
      "epoch": 1.4283180236816407e-05,
      "model_forward_time": 5.976460218429565,
      "step": 14977
    },
    {
      "epoch": 1.4283180236816407e-05,
      "step": 14977,
      "training_step_time": 17.43832564353943
    },
    {
      "epoch": 1.4284133911132812e-05,
      "model_forward_time": 5.602363586425781,
      "step": 14978
    },
    {
      "epoch": 1.4284133911132812e-05,
      "step": 14978,
      "training_step_time": 17.05587363243103
    },
    {
      "epoch": 1.4285087585449218e-05,
      "model_forward_time": 5.600565433502197,
      "step": 14979
    },
    {
      "epoch": 1.4285087585449218e-05,
      "step": 14979,
      "training_step_time": 17.075717210769653
    },
    {
      "epoch": 1.4286041259765625e-05,
      "grad_norm": 0.9429332613945007,
      "learning_rate": 1e-05,
      "loss": 0.063,
      "step": 14980
    },
    {
      "epoch": 1.4286041259765625e-05,
      "model_forward_time": 5.606427192687988,
      "step": 14980
    },
    {
      "dynamics_loss_avg": 0.06008395254611969,
      "epoch": 1.4286041259765625e-05,
      "step": 14980
    },
    {
      "action_loss_avg": 0.008051873021759092,
      "epoch": 1.4286041259765625e-05,
      "step": 14980
    },
    {
      "epoch": 1.4286041259765625e-05,
      "step": 14980,
      "training_step_time": 17.072305917739868
    },
    {
      "epoch": 1.4286994934082031e-05,
      "model_forward_time": 5.601836442947388,
      "step": 14981
    },
    {
      "epoch": 1.4286994934082031e-05,
      "step": 14981,
      "training_step_time": 17.087283611297607
    },
    {
      "epoch": 1.4287948608398437e-05,
      "model_forward_time": 5.610388278961182,
      "step": 14982
    },
    {
      "epoch": 1.4287948608398437e-05,
      "step": 14982,
      "training_step_time": 17.457350730895996
    },
    {
      "epoch": 1.4288902282714844e-05,
      "model_forward_time": 5.6042492389678955,
      "step": 14983
    },
    {
      "epoch": 1.4288902282714844e-05,
      "step": 14983,
      "training_step_time": 17.104989767074585
    },
    {
      "epoch": 1.428985595703125e-05,
      "model_forward_time": 5.617991924285889,
      "step": 14984
    },
    {
      "epoch": 1.428985595703125e-05,
      "step": 14984,
      "training_step_time": 17.128276824951172
    },
    {
      "epoch": 1.4290809631347656e-05,
      "model_forward_time": 5.6300177574157715,
      "step": 14985
    },
    {
      "epoch": 1.4290809631347656e-05,
      "step": 14985,
      "training_step_time": 17.134713649749756
    },
    {
      "epoch": 1.4291763305664063e-05,
      "model_forward_time": 5.615815162658691,
      "step": 14986
    },
    {
      "epoch": 1.4291763305664063e-05,
      "step": 14986,
      "training_step_time": 17.104084491729736
    },
    {
      "epoch": 1.4292716979980469e-05,
      "model_forward_time": 5.60617995262146,
      "step": 14987
    },
    {
      "epoch": 1.4292716979980469e-05,
      "step": 14987,
      "training_step_time": 17.039917707443237
    },
    {
      "epoch": 1.4293670654296875e-05,
      "model_forward_time": 5.961245059967041,
      "step": 14988
    },
    {
      "epoch": 1.4293670654296875e-05,
      "step": 14988,
      "training_step_time": 17.399731636047363
    },
    {
      "epoch": 1.4294624328613282e-05,
      "model_forward_time": 5.569790840148926,
      "step": 14989
    },
    {
      "epoch": 1.4294624328613282e-05,
      "step": 14989,
      "training_step_time": 16.9831223487854
    },
    {
      "epoch": 1.4295578002929688e-05,
      "grad_norm": 1.6317875385284424,
      "learning_rate": 1e-05,
      "loss": 0.0914,
      "step": 14990
    },
    {
      "epoch": 1.4295578002929688e-05,
      "model_forward_time": 5.583189010620117,
      "step": 14990
    },
    {
      "dynamics_loss_avg": 0.07578160502016544,
      "epoch": 1.4295578002929688e-05,
      "step": 14990
    },
    {
      "action_loss_avg": 0.012400460615754127,
      "epoch": 1.4295578002929688e-05,
      "step": 14990
    },
    {
      "epoch": 1.4295578002929688e-05,
      "step": 14990,
      "training_step_time": 16.98957133293152
    },
    {
      "epoch": 1.4296531677246093e-05,
      "model_forward_time": 5.586076498031616,
      "step": 14991
    },
    {
      "epoch": 1.4296531677246093e-05,
      "step": 14991,
      "training_step_time": 17.030270099639893
    },
    {
      "epoch": 1.42974853515625e-05,
      "model_forward_time": 5.590058088302612,
      "step": 14992
    },
    {
      "epoch": 1.42974853515625e-05,
      "step": 14992,
      "training_step_time": 17.035706996917725
    },
    {
      "epoch": 1.4298439025878907e-05,
      "model_forward_time": 5.598425626754761,
      "step": 14993
    },
    {
      "epoch": 1.4298439025878907e-05,
      "step": 14993,
      "training_step_time": 17.437572717666626
    },
    {
      "epoch": 1.4299392700195312e-05,
      "model_forward_time": 5.6170759201049805,
      "step": 14994
    },
    {
      "epoch": 1.4299392700195312e-05,
      "step": 14994,
      "training_step_time": 17.113879680633545
    },
    {
      "epoch": 1.4300346374511718e-05,
      "model_forward_time": 5.6256103515625,
      "step": 14995
    },
    {
      "epoch": 1.4300346374511718e-05,
      "step": 14995,
      "training_step_time": 17.12556505203247
    },
    {
      "epoch": 1.4301300048828125e-05,
      "model_forward_time": 5.591583490371704,
      "step": 14996
    },
    {
      "epoch": 1.4301300048828125e-05,
      "step": 14996,
      "training_step_time": 17.0357027053833
    },
    {
      "epoch": 1.4302253723144531e-05,
      "model_forward_time": 5.586193084716797,
      "step": 14997
    },
    {
      "epoch": 1.4302253723144531e-05,
      "step": 14997,
      "training_step_time": 17.041717767715454
    },
    {
      "epoch": 1.4303207397460937e-05,
      "model_forward_time": 5.598021507263184,
      "step": 14998
    },
    {
      "epoch": 1.4303207397460937e-05,
      "step": 14998,
      "training_step_time": 17.07051920890808
    },
    {
      "epoch": 1.4304161071777344e-05,
      "model_forward_time": 5.98615288734436,
      "step": 14999
    },
    {
      "epoch": 1.4304161071777344e-05,
      "step": 14999,
      "training_step_time": 17.456411123275757
    },
    {
      "epoch": 1.430511474609375e-05,
      "grad_norm": 0.40090540051460266,
      "learning_rate": 1e-05,
      "loss": 0.0657,
      "step": 15000
    },
    {
      "epoch": 1.430511474609375e-05,
      "model_forward_time": 5.642765283584595,
      "step": 15000
    },
    {
      "dynamics_loss_avg": 0.05894103012979031,
      "epoch": 1.430511474609375e-05,
      "step": 15000
    },
    {
      "action_loss_avg": 0.009058604249730706,
      "epoch": 1.430511474609375e-05,
      "step": 15000
    },
    {
      "epoch": 1.430511474609375e-05,
      "step": 15000,
      "training_step_time": 17.120006799697876
    },
    {
      "epoch": 1.4306068420410156e-05,
      "model_forward_time": 5.665880918502808,
      "step": 15001
    },
    {
      "epoch": 1.4306068420410156e-05,
      "step": 15001,
      "training_step_time": 17.18466877937317
    },
    {
      "epoch": 1.4307022094726563e-05,
      "model_forward_time": 5.669745922088623,
      "step": 15002
    },
    {
      "epoch": 1.4307022094726563e-05,
      "step": 15002,
      "training_step_time": 17.20617938041687
    },
    {
      "epoch": 1.4307975769042969e-05,
      "model_forward_time": 5.661988019943237,
      "step": 15003
    },
    {
      "epoch": 1.4307975769042969e-05,
      "step": 15003,
      "training_step_time": 17.159258365631104
    },
    {
      "epoch": 1.4308929443359375e-05,
      "model_forward_time": 5.6901397705078125,
      "step": 15004
    },
    {
      "epoch": 1.4308929443359375e-05,
      "step": 15004,
      "training_step_time": 17.554807662963867
    },
    {
      "epoch": 1.4309883117675782e-05,
      "model_forward_time": 5.628474712371826,
      "step": 15005
    },
    {
      "epoch": 1.4309883117675782e-05,
      "step": 15005,
      "training_step_time": 17.142423391342163
    },
    {
      "epoch": 1.4310836791992188e-05,
      "model_forward_time": 5.632715940475464,
      "step": 15006
    },
    {
      "epoch": 1.4310836791992188e-05,
      "step": 15006,
      "training_step_time": 17.1064555644989
    },
    {
      "epoch": 1.4311790466308593e-05,
      "model_forward_time": 5.595532178878784,
      "step": 15007
    },
    {
      "epoch": 1.4311790466308593e-05,
      "step": 15007,
      "training_step_time": 17.02941918373108
    },
    {
      "epoch": 1.4312744140625e-05,
      "model_forward_time": 5.589586019515991,
      "step": 15008
    },
    {
      "epoch": 1.4312744140625e-05,
      "step": 15008,
      "training_step_time": 16.997533082962036
    },
    {
      "epoch": 1.4313697814941407e-05,
      "model_forward_time": 5.5784783363342285,
      "step": 15009
    },
    {
      "epoch": 1.4313697814941407e-05,
      "step": 15009,
      "training_step_time": 16.986183643341064
    },
    {
      "epoch": 1.4314651489257812e-05,
      "grad_norm": 0.6788131594657898,
      "learning_rate": 1e-05,
      "loss": 0.0592,
      "step": 15010
    },
    {
      "epoch": 1.4314651489257812e-05,
      "model_forward_time": 5.939463138580322,
      "step": 15010
    },
    {
      "dynamics_loss_avg": 0.053621497191488746,
      "epoch": 1.4314651489257812e-05,
      "step": 15010
    },
    {
      "action_loss_avg": 0.009247682429850102,
      "epoch": 1.4314651489257812e-05,
      "step": 15010
    },
    {
      "epoch": 1.4314651489257812e-05,
      "step": 15010,
      "training_step_time": 17.379818201065063
    },
    {
      "epoch": 1.4315605163574218e-05,
      "model_forward_time": 5.574799060821533,
      "step": 15011
    },
    {
      "epoch": 1.4315605163574218e-05,
      "step": 15011,
      "training_step_time": 16.990116596221924
    },
    {
      "epoch": 1.4316558837890625e-05,
      "model_forward_time": 5.5843071937561035,
      "step": 15012
    },
    {
      "epoch": 1.4316558837890625e-05,
      "step": 15012,
      "training_step_time": 17.02627158164978
    },
    {
      "epoch": 1.4317512512207031e-05,
      "model_forward_time": 5.595426321029663,
      "step": 15013
    },
    {
      "epoch": 1.4317512512207031e-05,
      "step": 15013,
      "training_step_time": 17.041183948516846
    },
    {
      "epoch": 1.4318466186523437e-05,
      "model_forward_time": 5.585889101028442,
      "step": 15014
    },
    {
      "epoch": 1.4318466186523437e-05,
      "step": 15014,
      "training_step_time": 17.034966468811035
    },
    {
      "epoch": 1.4319419860839844e-05,
      "model_forward_time": 5.601228475570679,
      "step": 15015
    },
    {
      "epoch": 1.4319419860839844e-05,
      "step": 15015,
      "training_step_time": 17.44670343399048
    },
    {
      "epoch": 1.432037353515625e-05,
      "model_forward_time": 5.59624171257019,
      "step": 15016
    },
    {
      "epoch": 1.432037353515625e-05,
      "step": 15016,
      "training_step_time": 17.05266284942627
    },
    {
      "epoch": 1.4321327209472656e-05,
      "model_forward_time": 5.6000354290008545,
      "step": 15017
    },
    {
      "epoch": 1.4321327209472656e-05,
      "step": 15017,
      "training_step_time": 17.04187297821045
    },
    {
      "epoch": 1.4322280883789063e-05,
      "model_forward_time": 5.602571249008179,
      "step": 15018
    },
    {
      "epoch": 1.4322280883789063e-05,
      "step": 15018,
      "training_step_time": 17.07023024559021
    },
    {
      "epoch": 1.4323234558105469e-05,
      "model_forward_time": 5.613229036331177,
      "step": 15019
    },
    {
      "epoch": 1.4323234558105469e-05,
      "step": 15019,
      "training_step_time": 17.111461639404297
    },
    {
      "epoch": 1.4324188232421875e-05,
      "grad_norm": 1.230377197265625,
      "learning_rate": 1e-05,
      "loss": 0.0619,
      "step": 15020
    },
    {
      "epoch": 1.4324188232421875e-05,
      "model_forward_time": 5.619690418243408,
      "step": 15020
    },
    {
      "dynamics_loss_avg": 0.04593341378495097,
      "epoch": 1.4324188232421875e-05,
      "step": 15020
    },
    {
      "action_loss_avg": 0.008225815091282129,
      "epoch": 1.4324188232421875e-05,
      "step": 15020
    },
    {
      "epoch": 1.4324188232421875e-05,
      "step": 15020,
      "training_step_time": 17.111119985580444
    },
    {
      "epoch": 1.4325141906738282e-05,
      "model_forward_time": 5.970607042312622,
      "step": 15021
    },
    {
      "epoch": 1.4325141906738282e-05,
      "step": 15021,
      "training_step_time": 17.465611219406128
    },
    {
      "epoch": 1.4326095581054688e-05,
      "model_forward_time": 5.613624572753906,
      "step": 15022
    },
    {
      "epoch": 1.4326095581054688e-05,
      "step": 15022,
      "training_step_time": 17.08988881111145
    },
    {
      "epoch": 1.4327049255371093e-05,
      "model_forward_time": 5.591172933578491,
      "step": 15023
    },
    {
      "epoch": 1.4327049255371093e-05,
      "step": 15023,
      "training_step_time": 17.02363085746765
    },
    {
      "epoch": 1.43280029296875e-05,
      "model_forward_time": 5.592182159423828,
      "step": 15024
    },
    {
      "epoch": 1.43280029296875e-05,
      "step": 15024,
      "training_step_time": 17.015127897262573
    },
    {
      "epoch": 1.4328956604003907e-05,
      "model_forward_time": 5.581431150436401,
      "step": 15025
    },
    {
      "epoch": 1.4328956604003907e-05,
      "step": 15025,
      "training_step_time": 17.011672258377075
    },
    {
      "epoch": 1.4329910278320312e-05,
      "model_forward_time": 5.968532562255859,
      "step": 15026
    },
    {
      "epoch": 1.4329910278320312e-05,
      "step": 15026,
      "training_step_time": 17.394632577896118
    },
    {
      "epoch": 1.4330863952636718e-05,
      "model_forward_time": 5.597239971160889,
      "step": 15027
    },
    {
      "epoch": 1.4330863952636718e-05,
      "step": 15027,
      "training_step_time": 17.021119356155396
    },
    {
      "epoch": 1.4331817626953125e-05,
      "model_forward_time": 5.589800119400024,
      "step": 15028
    },
    {
      "epoch": 1.4331817626953125e-05,
      "step": 15028,
      "training_step_time": 17.04006814956665
    },
    {
      "epoch": 1.4332771301269531e-05,
      "model_forward_time": 5.603084325790405,
      "step": 15029
    },
    {
      "epoch": 1.4332771301269531e-05,
      "step": 15029,
      "training_step_time": 17.07369637489319
    },
    {
      "epoch": 1.4333724975585937e-05,
      "grad_norm": 1.6246532201766968,
      "learning_rate": 1e-05,
      "loss": 0.0767,
      "step": 15030
    },
    {
      "epoch": 1.4333724975585937e-05,
      "model_forward_time": 5.6078033447265625,
      "step": 15030
    },
    {
      "dynamics_loss_avg": 0.07045216653496027,
      "epoch": 1.4333724975585937e-05,
      "step": 15030
    },
    {
      "action_loss_avg": 0.013396164472214877,
      "epoch": 1.4333724975585937e-05,
      "step": 15030
    },
    {
      "epoch": 1.4333724975585937e-05,
      "step": 15030,
      "training_step_time": 17.09259605407715
    },
    {
      "epoch": 1.4334678649902344e-05,
      "model_forward_time": 5.5993735790252686,
      "step": 15031
    },
    {
      "epoch": 1.4334678649902344e-05,
      "step": 15031,
      "training_step_time": 17.079870462417603
    },
    {
      "epoch": 1.433563232421875e-05,
      "model_forward_time": 5.935820579528809,
      "step": 15032
    },
    {
      "epoch": 1.433563232421875e-05,
      "step": 15032,
      "training_step_time": 17.393014669418335
    },
    {
      "epoch": 1.4336585998535156e-05,
      "model_forward_time": 5.604696035385132,
      "step": 15033
    },
    {
      "epoch": 1.4336585998535156e-05,
      "step": 15033,
      "training_step_time": 17.077057361602783
    },
    {
      "epoch": 1.4337539672851563e-05,
      "model_forward_time": 5.606435775756836,
      "step": 15034
    },
    {
      "epoch": 1.4337539672851563e-05,
      "step": 15034,
      "training_step_time": 17.079902172088623
    },
    {
      "epoch": 1.4338493347167969e-05,
      "model_forward_time": 5.613419055938721,
      "step": 15035
    },
    {
      "epoch": 1.4338493347167969e-05,
      "step": 15035,
      "training_step_time": 17.1058292388916
    },
    {
      "epoch": 1.4339447021484375e-05,
      "model_forward_time": 5.620332956314087,
      "step": 15036
    },
    {
      "epoch": 1.4339447021484375e-05,
      "step": 15036,
      "training_step_time": 17.113667249679565
    },
    {
      "epoch": 1.4340400695800782e-05,
      "model_forward_time": 5.979499816894531,
      "step": 15037
    },
    {
      "epoch": 1.4340400695800782e-05,
      "step": 15037,
      "training_step_time": 17.482158184051514
    },
    {
      "epoch": 1.4341354370117188e-05,
      "model_forward_time": 5.615859270095825,
      "step": 15038
    },
    {
      "epoch": 1.4341354370117188e-05,
      "step": 15038,
      "training_step_time": 17.11442995071411
    },
    {
      "epoch": 1.4342308044433593e-05,
      "model_forward_time": 5.6125710010528564,
      "step": 15039
    },
    {
      "epoch": 1.4342308044433593e-05,
      "step": 15039,
      "training_step_time": 17.098241090774536
    },
    {
      "epoch": 1.434326171875e-05,
      "grad_norm": 1.1454048156738281,
      "learning_rate": 1e-05,
      "loss": 0.07,
      "step": 15040
    },
    {
      "epoch": 1.434326171875e-05,
      "model_forward_time": 5.6204352378845215,
      "step": 15040
    },
    {
      "dynamics_loss_avg": 0.05714189782738686,
      "epoch": 1.434326171875e-05,
      "step": 15040
    },
    {
      "action_loss_avg": 0.009877849603071809,
      "epoch": 1.434326171875e-05,
      "step": 15040
    },
    {
      "epoch": 1.434326171875e-05,
      "step": 15040,
      "training_step_time": 17.092379093170166
    },
    {
      "epoch": 1.4344215393066407e-05,
      "model_forward_time": 5.626142978668213,
      "step": 15041
    },
    {
      "epoch": 1.4344215393066407e-05,
      "step": 15041,
      "training_step_time": 17.10598373413086
    },
    {
      "epoch": 1.4345169067382812e-05,
      "model_forward_time": 5.615866422653198,
      "step": 15042
    },
    {
      "epoch": 1.4345169067382812e-05,
      "step": 15042,
      "training_step_time": 17.090339422225952
    },
    {
      "epoch": 1.4346122741699218e-05,
      "model_forward_time": 5.979278564453125,
      "step": 15043
    },
    {
      "epoch": 1.4346122741699218e-05,
      "step": 15043,
      "training_step_time": 17.45917057991028
    },
    {
      "epoch": 1.4347076416015625e-05,
      "model_forward_time": 5.6198201179504395,
      "step": 15044
    },
    {
      "epoch": 1.4347076416015625e-05,
      "step": 15044,
      "training_step_time": 17.09239649772644
    },
    {
      "epoch": 1.4348030090332031e-05,
      "model_forward_time": 5.608778476715088,
      "step": 15045
    },
    {
      "epoch": 1.4348030090332031e-05,
      "step": 15045,
      "training_step_time": 17.09326171875
    },
    {
      "epoch": 1.4348983764648437e-05,
      "model_forward_time": 5.610812425613403,
      "step": 15046
    },
    {
      "epoch": 1.4348983764648437e-05,
      "step": 15046,
      "training_step_time": 17.12380337715149
    },
    {
      "epoch": 1.4349937438964844e-05,
      "model_forward_time": 5.627312183380127,
      "step": 15047
    },
    {
      "epoch": 1.4349937438964844e-05,
      "step": 15047,
      "training_step_time": 17.13912868499756
    },
    {
      "epoch": 1.435089111328125e-05,
      "model_forward_time": 5.979918479919434,
      "step": 15048
    },
    {
      "epoch": 1.435089111328125e-05,
      "step": 15048,
      "training_step_time": 17.448687314987183
    },
    {
      "epoch": 1.4351844787597656e-05,
      "model_forward_time": 5.602274179458618,
      "step": 15049
    },
    {
      "epoch": 1.4351844787597656e-05,
      "step": 15049,
      "training_step_time": 17.053929805755615
    },
    {
      "epoch": 1.4352798461914063e-05,
      "grad_norm": 0.25736188888549805,
      "learning_rate": 1e-05,
      "loss": 0.0537,
      "step": 15050
    },
    {
      "epoch": 1.4352798461914063e-05,
      "model_forward_time": 5.5863196849823,
      "step": 15050
    },
    {
      "dynamics_loss_avg": 0.04479120448231697,
      "epoch": 1.4352798461914063e-05,
      "step": 15050
    },
    {
      "action_loss_avg": 0.005829707335215062,
      "epoch": 1.4352798461914063e-05,
      "step": 15050
    },
    {
      "epoch": 1.4352798461914063e-05,
      "step": 15050,
      "training_step_time": 17.034833669662476
    },
    {
      "epoch": 1.4353752136230469e-05,
      "model_forward_time": 5.598006963729858,
      "step": 15051
    },
    {
      "epoch": 1.4353752136230469e-05,
      "step": 15051,
      "training_step_time": 17.056241035461426
    },
    {
      "epoch": 1.4354705810546875e-05,
      "model_forward_time": 5.592406272888184,
      "step": 15052
    },
    {
      "epoch": 1.4354705810546875e-05,
      "step": 15052,
      "training_step_time": 17.033111572265625
    },
    {
      "epoch": 1.4355659484863282e-05,
      "model_forward_time": 5.967973947525024,
      "step": 15053
    },
    {
      "epoch": 1.4355659484863282e-05,
      "step": 15053,
      "training_step_time": 17.442314863204956
    },
    {
      "epoch": 1.4356613159179688e-05,
      "model_forward_time": 5.605229616165161,
      "step": 15054
    },
    {
      "epoch": 1.4356613159179688e-05,
      "step": 15054,
      "training_step_time": 17.089635848999023
    },
    {
      "epoch": 1.4357566833496093e-05,
      "model_forward_time": 5.607595920562744,
      "step": 15055
    },
    {
      "epoch": 1.4357566833496093e-05,
      "step": 15055,
      "training_step_time": 17.089622974395752
    },
    {
      "epoch": 1.43585205078125e-05,
      "model_forward_time": 5.615270614624023,
      "step": 15056
    },
    {
      "epoch": 1.43585205078125e-05,
      "step": 15056,
      "training_step_time": 17.095345973968506
    },
    {
      "epoch": 1.4359474182128907e-05,
      "model_forward_time": 5.610168218612671,
      "step": 15057
    },
    {
      "epoch": 1.4359474182128907e-05,
      "step": 15057,
      "training_step_time": 17.094472646713257
    },
    {
      "epoch": 1.4360427856445312e-05,
      "model_forward_time": 5.617501497268677,
      "step": 15058
    },
    {
      "epoch": 1.4360427856445312e-05,
      "step": 15058,
      "training_step_time": 17.09747290611267
    },
    {
      "epoch": 1.4361381530761718e-05,
      "model_forward_time": 5.9796130657196045,
      "step": 15059
    },
    {
      "epoch": 1.4361381530761718e-05,
      "step": 15059,
      "training_step_time": 17.408733367919922
    },
    {
      "epoch": 1.4362335205078125e-05,
      "grad_norm": 1.3172259330749512,
      "learning_rate": 1e-05,
      "loss": 0.0541,
      "step": 15060
    },
    {
      "epoch": 1.4362335205078125e-05,
      "model_forward_time": 5.592262506484985,
      "step": 15060
    },
    {
      "dynamics_loss_avg": 0.051154385507106784,
      "epoch": 1.4362335205078125e-05,
      "step": 15060
    },
    {
      "action_loss_avg": 0.008378857420757413,
      "epoch": 1.4362335205078125e-05,
      "step": 15060
    },
    {
      "epoch": 1.4362335205078125e-05,
      "step": 15060,
      "training_step_time": 17.046676874160767
    },
    {
      "epoch": 1.4363288879394531e-05,
      "model_forward_time": 5.595394134521484,
      "step": 15061
    },
    {
      "epoch": 1.4363288879394531e-05,
      "step": 15061,
      "training_step_time": 17.021410703659058
    },
    {
      "epoch": 1.4364242553710937e-05,
      "model_forward_time": 5.588353633880615,
      "step": 15062
    },
    {
      "epoch": 1.4364242553710937e-05,
      "step": 15062,
      "training_step_time": 17.03633165359497
    },
    {
      "epoch": 1.4365196228027344e-05,
      "model_forward_time": 5.597360372543335,
      "step": 15063
    },
    {
      "epoch": 1.4365196228027344e-05,
      "step": 15063,
      "training_step_time": 17.024222135543823
    },
    {
      "epoch": 1.436614990234375e-05,
      "model_forward_time": 5.946139335632324,
      "step": 15064
    },
    {
      "epoch": 1.436614990234375e-05,
      "step": 15064,
      "training_step_time": 17.389575004577637
    },
    {
      "epoch": 1.4367103576660156e-05,
      "model_forward_time": 5.59559965133667,
      "step": 15065
    },
    {
      "epoch": 1.4367103576660156e-05,
      "step": 15065,
      "training_step_time": 17.06816864013672
    },
    {
      "epoch": 1.4368057250976563e-05,
      "model_forward_time": 5.605926513671875,
      "step": 15066
    },
    {
      "epoch": 1.4368057250976563e-05,
      "step": 15066,
      "training_step_time": 17.09357237815857
    },
    {
      "epoch": 1.4369010925292969e-05,
      "model_forward_time": 5.617656230926514,
      "step": 15067
    },
    {
      "epoch": 1.4369010925292969e-05,
      "step": 15067,
      "training_step_time": 17.10339665412903
    },
    {
      "epoch": 1.4369964599609375e-05,
      "model_forward_time": 5.609898567199707,
      "step": 15068
    },
    {
      "epoch": 1.4369964599609375e-05,
      "step": 15068,
      "training_step_time": 17.094184160232544
    },
    {
      "epoch": 1.4370918273925782e-05,
      "model_forward_time": 5.598598957061768,
      "step": 15069
    },
    {
      "epoch": 1.4370918273925782e-05,
      "step": 15069,
      "training_step_time": 17.07628083229065
    },
    {
      "epoch": 1.4371871948242188e-05,
      "grad_norm": 1.1017637252807617,
      "learning_rate": 1e-05,
      "loss": 0.0855,
      "step": 15070
    },
    {
      "epoch": 1.4371871948242188e-05,
      "model_forward_time": 5.967414379119873,
      "step": 15070
    },
    {
      "dynamics_loss_avg": 0.06967993341386318,
      "epoch": 1.4371871948242188e-05,
      "step": 15070
    },
    {
      "action_loss_avg": 0.017281597189139575,
      "epoch": 1.4371871948242188e-05,
      "step": 15070
    },
    {
      "epoch": 1.4371871948242188e-05,
      "step": 15070,
      "training_step_time": 17.410189867019653
    },
    {
      "epoch": 1.4372825622558593e-05,
      "model_forward_time": 5.581112623214722,
      "step": 15071
    },
    {
      "epoch": 1.4372825622558593e-05,
      "step": 15071,
      "training_step_time": 17.03335189819336
    },
    {
      "epoch": 1.4373779296875e-05,
      "model_forward_time": 5.594172239303589,
      "step": 15072
    },
    {
      "epoch": 1.4373779296875e-05,
      "step": 15072,
      "training_step_time": 17.048075199127197
    },
    {
      "epoch": 1.4374732971191407e-05,
      "model_forward_time": 5.607927560806274,
      "step": 15073
    },
    {
      "epoch": 1.4374732971191407e-05,
      "step": 15073,
      "training_step_time": 17.062849283218384
    },
    {
      "epoch": 1.4375686645507812e-05,
      "model_forward_time": 5.6058995723724365,
      "step": 15074
    },
    {
      "epoch": 1.4375686645507812e-05,
      "step": 15074,
      "training_step_time": 17.068259477615356
    },
    {
      "epoch": 1.4376640319824218e-05,
      "model_forward_time": 5.6025614738464355,
      "step": 15075
    },
    {
      "epoch": 1.4376640319824218e-05,
      "step": 15075,
      "training_step_time": 17.413302659988403
    },
    {
      "epoch": 1.4377593994140625e-05,
      "model_forward_time": 5.598043918609619,
      "step": 15076
    },
    {
      "epoch": 1.4377593994140625e-05,
      "step": 15076,
      "training_step_time": 17.080921173095703
    },
    {
      "epoch": 1.4378547668457031e-05,
      "model_forward_time": 5.606338262557983,
      "step": 15077
    },
    {
      "epoch": 1.4378547668457031e-05,
      "step": 15077,
      "training_step_time": 17.094966888427734
    },
    {
      "epoch": 1.4379501342773437e-05,
      "model_forward_time": 5.629921913146973,
      "step": 15078
    },
    {
      "epoch": 1.4379501342773437e-05,
      "step": 15078,
      "training_step_time": 17.127920150756836
    },
    {
      "epoch": 1.4380455017089844e-05,
      "model_forward_time": 5.617681980133057,
      "step": 15079
    },
    {
      "epoch": 1.4380455017089844e-05,
      "step": 15079,
      "training_step_time": 17.119173288345337
    },
    {
      "epoch": 1.438140869140625e-05,
      "grad_norm": 1.5012823343276978,
      "learning_rate": 1e-05,
      "loss": 0.0687,
      "step": 15080
    },
    {
      "epoch": 1.438140869140625e-05,
      "model_forward_time": 5.614979028701782,
      "step": 15080
    },
    {
      "dynamics_loss_avg": 0.05398009549826384,
      "epoch": 1.438140869140625e-05,
      "step": 15080
    },
    {
      "action_loss_avg": 0.00867302876431495,
      "epoch": 1.438140869140625e-05,
      "step": 15080
    },
    {
      "epoch": 1.438140869140625e-05,
      "step": 15080,
      "training_step_time": 17.115736722946167
    },
    {
      "epoch": 1.4382362365722656e-05,
      "model_forward_time": 5.9854395389556885,
      "step": 15081
    },
    {
      "epoch": 1.4382362365722656e-05,
      "step": 15081,
      "training_step_time": 17.46722984313965
    },
    {
      "epoch": 1.4383316040039063e-05,
      "model_forward_time": 5.613545179367065,
      "step": 15082
    },
    {
      "epoch": 1.4383316040039063e-05,
      "step": 15082,
      "training_step_time": 17.065163373947144
    },
    {
      "epoch": 1.4384269714355469e-05,
      "model_forward_time": 5.585486173629761,
      "step": 15083
    },
    {
      "epoch": 1.4384269714355469e-05,
      "step": 15083,
      "training_step_time": 17.019764184951782
    },
    {
      "epoch": 1.4385223388671875e-05,
      "model_forward_time": 5.592320203781128,
      "step": 15084
    },
    {
      "epoch": 1.4385223388671875e-05,
      "step": 15084,
      "training_step_time": 17.040568351745605
    },
    {
      "epoch": 1.4386177062988282e-05,
      "model_forward_time": 5.596345663070679,
      "step": 15085
    },
    {
      "epoch": 1.4386177062988282e-05,
      "step": 15085,
      "training_step_time": 17.049677848815918
    },
    {
      "epoch": 1.4387130737304688e-05,
      "model_forward_time": 5.97062611579895,
      "step": 15086
    },
    {
      "epoch": 1.4387130737304688e-05,
      "step": 15086,
      "training_step_time": 17.430671453475952
    },
    {
      "epoch": 1.4388084411621094e-05,
      "model_forward_time": 5.603145360946655,
      "step": 15087
    },
    {
      "epoch": 1.4388084411621094e-05,
      "step": 15087,
      "training_step_time": 17.087697744369507
    },
    {
      "epoch": 1.43890380859375e-05,
      "model_forward_time": 5.607964515686035,
      "step": 15088
    },
    {
      "epoch": 1.43890380859375e-05,
      "step": 15088,
      "training_step_time": 17.08355975151062
    },
    {
      "epoch": 1.4389991760253907e-05,
      "model_forward_time": 5.59520697593689,
      "step": 15089
    },
    {
      "epoch": 1.4389991760253907e-05,
      "step": 15089,
      "training_step_time": 17.045033931732178
    },
    {
      "epoch": 1.4390945434570312e-05,
      "grad_norm": 1.2305707931518555,
      "learning_rate": 1e-05,
      "loss": 0.0632,
      "step": 15090
    },
    {
      "epoch": 1.4390945434570312e-05,
      "model_forward_time": 5.585117816925049,
      "step": 15090
    },
    {
      "dynamics_loss_avg": 0.05496967751532793,
      "epoch": 1.4390945434570312e-05,
      "step": 15090
    },
    {
      "action_loss_avg": 0.008663390786387027,
      "epoch": 1.4390945434570312e-05,
      "step": 15090
    },
    {
      "epoch": 1.4390945434570312e-05,
      "step": 15090,
      "training_step_time": 17.04277801513672
    },
    {
      "epoch": 1.4391899108886718e-05,
      "model_forward_time": 5.604681968688965,
      "step": 15091
    },
    {
      "epoch": 1.4391899108886718e-05,
      "step": 15091,
      "training_step_time": 17.084516048431396
    },
    {
      "epoch": 1.4392852783203126e-05,
      "model_forward_time": 5.985044002532959,
      "step": 15092
    },
    {
      "epoch": 1.4392852783203126e-05,
      "step": 15092,
      "training_step_time": 17.485658407211304
    },
    {
      "epoch": 1.4393806457519531e-05,
      "model_forward_time": 5.606342315673828,
      "step": 15093
    },
    {
      "epoch": 1.4393806457519531e-05,
      "step": 15093,
      "training_step_time": 17.10192084312439
    },
    {
      "epoch": 1.4394760131835937e-05,
      "model_forward_time": 5.625407457351685,
      "step": 15094
    },
    {
      "epoch": 1.4394760131835937e-05,
      "step": 15094,
      "training_step_time": 17.123287200927734
    },
    {
      "epoch": 1.4395713806152344e-05,
      "model_forward_time": 5.608017206192017,
      "step": 15095
    },
    {
      "epoch": 1.4395713806152344e-05,
      "step": 15095,
      "training_step_time": 17.10416841506958
    },
    {
      "epoch": 1.439666748046875e-05,
      "model_forward_time": 5.61291766166687,
      "step": 15096
    },
    {
      "epoch": 1.439666748046875e-05,
      "step": 15096,
      "training_step_time": 17.112256288528442
    },
    {
      "epoch": 1.4397621154785156e-05,
      "model_forward_time": 5.961684226989746,
      "step": 15097
    },
    {
      "epoch": 1.4397621154785156e-05,
      "step": 15097,
      "training_step_time": 17.430367946624756
    },
    {
      "epoch": 1.4398574829101563e-05,
      "model_forward_time": 5.609205722808838,
      "step": 15098
    },
    {
      "epoch": 1.4398574829101563e-05,
      "step": 15098,
      "training_step_time": 17.08264660835266
    },
    {
      "epoch": 1.4399528503417969e-05,
      "model_forward_time": 5.613827228546143,
      "step": 15099
    },
    {
      "epoch": 1.4399528503417969e-05,
      "step": 15099,
      "training_step_time": 17.109922170639038
    },
    {
      "epoch": 1.4400482177734375e-05,
      "grad_norm": 0.40946024656295776,
      "learning_rate": 1e-05,
      "loss": 0.0565,
      "step": 15100
    },
    {
      "epoch": 1.4400482177734375e-05,
      "model_forward_time": 5.618869781494141,
      "step": 15100
    },
    {
      "dynamics_loss_avg": 0.0487182205542922,
      "epoch": 1.4400482177734375e-05,
      "step": 15100
    },
    {
      "action_loss_avg": 0.00686756162904203,
      "epoch": 1.4400482177734375e-05,
      "step": 15100
    },
    {
      "epoch": 1.4400482177734375e-05,
      "step": 15100,
      "training_step_time": 17.121185541152954
    },
    {
      "epoch": 1.4401435852050782e-05,
      "model_forward_time": 5.617573499679565,
      "step": 15101
    },
    {
      "epoch": 1.4401435852050782e-05,
      "step": 15101,
      "training_step_time": 17.098536491394043
    },
    {
      "epoch": 1.4402389526367188e-05,
      "model_forward_time": 5.605900049209595,
      "step": 15102
    },
    {
      "epoch": 1.4402389526367188e-05,
      "step": 15102,
      "training_step_time": 17.086726665496826
    },
    {
      "epoch": 1.4403343200683594e-05,
      "model_forward_time": 5.980672359466553,
      "step": 15103
    },
    {
      "epoch": 1.4403343200683594e-05,
      "step": 15103,
      "training_step_time": 17.448696851730347
    },
    {
      "epoch": 1.4404296875e-05,
      "model_forward_time": 5.604891300201416,
      "step": 15104
    },
    {
      "epoch": 1.4404296875e-05,
      "step": 15104,
      "training_step_time": 17.075038194656372
    },
    {
      "epoch": 1.4405250549316407e-05,
      "model_forward_time": 5.591603517532349,
      "step": 15105
    },
    {
      "epoch": 1.4405250549316407e-05,
      "step": 15105,
      "training_step_time": 17.029850244522095
    },
    {
      "epoch": 1.4406204223632812e-05,
      "model_forward_time": 5.602160692214966,
      "step": 15106
    },
    {
      "epoch": 1.4406204223632812e-05,
      "step": 15106,
      "training_step_time": 17.058438777923584
    },
    {
      "epoch": 1.4407157897949218e-05,
      "model_forward_time": 5.605224847793579,
      "step": 15107
    },
    {
      "epoch": 1.4407157897949218e-05,
      "step": 15107,
      "training_step_time": 17.11145043373108
    },
    {
      "epoch": 1.4408111572265626e-05,
      "model_forward_time": 5.6125500202178955,
      "step": 15108
    },
    {
      "epoch": 1.4408111572265626e-05,
      "step": 15108,
      "training_step_time": 17.46016573905945
    },
    {
      "epoch": 1.4409065246582031e-05,
      "model_forward_time": 5.59255576133728,
      "step": 15109
    },
    {
      "epoch": 1.4409065246582031e-05,
      "step": 15109,
      "training_step_time": 17.01584482192993
    },
    {
      "epoch": 1.4410018920898437e-05,
      "grad_norm": 0.610749363899231,
      "learning_rate": 1e-05,
      "loss": 0.0644,
      "step": 15110
    },
    {
      "epoch": 1.4410018920898437e-05,
      "model_forward_time": 5.596634387969971,
      "step": 15110
    },
    {
      "dynamics_loss_avg": 0.05275706928223371,
      "epoch": 1.4410018920898437e-05,
      "step": 15110
    },
    {
      "action_loss_avg": 0.009728316310793161,
      "epoch": 1.4410018920898437e-05,
      "step": 15110
    },
    {
      "epoch": 1.4410018920898437e-05,
      "step": 15110,
      "training_step_time": 17.033608198165894
    },
    {
      "epoch": 1.4410972595214844e-05,
      "model_forward_time": 5.586410284042358,
      "step": 15111
    },
    {
      "epoch": 1.4410972595214844e-05,
      "step": 15111,
      "training_step_time": 17.022210359573364
    },
    {
      "epoch": 1.441192626953125e-05,
      "model_forward_time": 5.593332767486572,
      "step": 15112
    },
    {
      "epoch": 1.441192626953125e-05,
      "step": 15112,
      "training_step_time": 17.039376497268677
    },
    {
      "epoch": 1.4412879943847656e-05,
      "model_forward_time": 5.597678184509277,
      "step": 15113
    },
    {
      "epoch": 1.4412879943847656e-05,
      "step": 15113,
      "training_step_time": 17.074358701705933
    },
    {
      "epoch": 1.4413833618164063e-05,
      "model_forward_time": 5.981121301651001,
      "step": 15114
    },
    {
      "epoch": 1.4413833618164063e-05,
      "step": 15114,
      "training_step_time": 17.483242988586426
    },
    {
      "epoch": 1.4414787292480469e-05,
      "model_forward_time": 5.615732669830322,
      "step": 15115
    },
    {
      "epoch": 1.4414787292480469e-05,
      "step": 15115,
      "training_step_time": 17.10623598098755
    },
    {
      "epoch": 1.4415740966796875e-05,
      "model_forward_time": 5.617191791534424,
      "step": 15116
    },
    {
      "epoch": 1.4415740966796875e-05,
      "step": 15116,
      "training_step_time": 17.11447763442993
    },
    {
      "epoch": 1.4416694641113282e-05,
      "model_forward_time": 5.611464500427246,
      "step": 15117
    },
    {
      "epoch": 1.4416694641113282e-05,
      "step": 15117,
      "training_step_time": 17.1040198802948
    },
    {
      "epoch": 1.4417648315429688e-05,
      "model_forward_time": 5.606764793395996,
      "step": 15118
    },
    {
      "epoch": 1.4417648315429688e-05,
      "step": 15118,
      "training_step_time": 17.105280876159668
    },
    {
      "epoch": 1.4418601989746094e-05,
      "model_forward_time": 5.984034061431885,
      "step": 15119
    },
    {
      "epoch": 1.4418601989746094e-05,
      "step": 15119,
      "training_step_time": 17.474022150039673
    },
    {
      "epoch": 1.44195556640625e-05,
      "grad_norm": 0.9334301352500916,
      "learning_rate": 1e-05,
      "loss": 0.0552,
      "step": 15120
    },
    {
      "epoch": 1.44195556640625e-05,
      "model_forward_time": 5.604931592941284,
      "step": 15120
    },
    {
      "dynamics_loss_avg": 0.053631798923015596,
      "epoch": 1.44195556640625e-05,
      "step": 15120
    },
    {
      "action_loss_avg": 0.009806671843398363,
      "epoch": 1.44195556640625e-05,
      "step": 15120
    },
    {
      "epoch": 1.44195556640625e-05,
      "step": 15120,
      "training_step_time": 17.067144632339478
    },
    {
      "epoch": 1.4420509338378907e-05,
      "model_forward_time": 5.6076014041900635,
      "step": 15121
    },
    {
      "epoch": 1.4420509338378907e-05,
      "step": 15121,
      "training_step_time": 17.086531400680542
    },
    {
      "epoch": 1.4421463012695312e-05,
      "model_forward_time": 5.599851131439209,
      "step": 15122
    },
    {
      "epoch": 1.4421463012695312e-05,
      "step": 15122,
      "training_step_time": 17.092698097229004
    },
    {
      "epoch": 1.4422416687011718e-05,
      "model_forward_time": 5.617512941360474,
      "step": 15123
    },
    {
      "epoch": 1.4422416687011718e-05,
      "step": 15123,
      "training_step_time": 17.118712186813354
    },
    {
      "epoch": 1.4423370361328126e-05,
      "model_forward_time": 5.625201940536499,
      "step": 15124
    },
    {
      "epoch": 1.4423370361328126e-05,
      "step": 15124,
      "training_step_time": 17.103602647781372
    },
    {
      "epoch": 1.4424324035644531e-05,
      "model_forward_time": 5.978943109512329,
      "step": 15125
    },
    {
      "epoch": 1.4424324035644531e-05,
      "step": 15125,
      "training_step_time": 17.412221670150757
    },
    {
      "epoch": 1.4425277709960937e-05,
      "model_forward_time": 5.586341619491577,
      "step": 15126
    },
    {
      "epoch": 1.4425277709960937e-05,
      "step": 15126,
      "training_step_time": 16.9986355304718
    },
    {
      "epoch": 1.4426231384277344e-05,
      "model_forward_time": 5.594600677490234,
      "step": 15127
    },
    {
      "epoch": 1.4426231384277344e-05,
      "step": 15127,
      "training_step_time": 17.04602336883545
    },
    {
      "epoch": 1.442718505859375e-05,
      "model_forward_time": 5.615538835525513,
      "step": 15128
    },
    {
      "epoch": 1.442718505859375e-05,
      "step": 15128,
      "training_step_time": 17.116368532180786
    },
    {
      "epoch": 1.4428138732910156e-05,
      "model_forward_time": 5.634545564651489,
      "step": 15129
    },
    {
      "epoch": 1.4428138732910156e-05,
      "step": 15129,
      "training_step_time": 17.134039640426636
    },
    {
      "epoch": 1.4429092407226563e-05,
      "grad_norm": 0.32991617918014526,
      "learning_rate": 1e-05,
      "loss": 0.0696,
      "step": 15130
    },
    {
      "epoch": 1.4429092407226563e-05,
      "model_forward_time": 5.96443510055542,
      "step": 15130
    },
    {
      "dynamics_loss_avg": 0.057338247634470464,
      "epoch": 1.4429092407226563e-05,
      "step": 15130
    },
    {
      "action_loss_avg": 0.006951088085770607,
      "epoch": 1.4429092407226563e-05,
      "step": 15130
    },
    {
      "epoch": 1.4429092407226563e-05,
      "step": 15130,
      "training_step_time": 17.41995596885681
    },
    {
      "epoch": 1.4430046081542969e-05,
      "model_forward_time": 5.590867519378662,
      "step": 15131
    },
    {
      "epoch": 1.4430046081542969e-05,
      "step": 15131,
      "training_step_time": 17.036505222320557
    },
    {
      "epoch": 1.4430999755859375e-05,
      "model_forward_time": 5.5745697021484375,
      "step": 15132
    },
    {
      "epoch": 1.4430999755859375e-05,
      "step": 15132,
      "training_step_time": 17.002612352371216
    },
    {
      "epoch": 1.443195343017578e-05,
      "model_forward_time": 5.583894491195679,
      "step": 15133
    },
    {
      "epoch": 1.443195343017578e-05,
      "step": 15133,
      "training_step_time": 17.017391204833984
    },
    {
      "epoch": 1.4432907104492188e-05,
      "model_forward_time": 5.58823823928833,
      "step": 15134
    },
    {
      "epoch": 1.4432907104492188e-05,
      "step": 15134,
      "training_step_time": 17.014397859573364
    },
    {
      "epoch": 1.4433860778808594e-05,
      "model_forward_time": 5.594441175460815,
      "step": 15135
    },
    {
      "epoch": 1.4433860778808594e-05,
      "step": 15135,
      "training_step_time": 17.04471731185913
    },
    {
      "epoch": 1.4434814453125e-05,
      "model_forward_time": 5.96295428276062,
      "step": 15136
    },
    {
      "epoch": 1.4434814453125e-05,
      "step": 15136,
      "training_step_time": 17.437973499298096
    },
    {
      "epoch": 1.4435768127441407e-05,
      "model_forward_time": 5.618306875228882,
      "step": 15137
    },
    {
      "epoch": 1.4435768127441407e-05,
      "step": 15137,
      "training_step_time": 17.140045166015625
    },
    {
      "epoch": 1.4436721801757812e-05,
      "model_forward_time": 5.639207124710083,
      "step": 15138
    },
    {
      "epoch": 1.4436721801757812e-05,
      "step": 15138,
      "training_step_time": 17.150805473327637
    },
    {
      "epoch": 1.4437675476074218e-05,
      "model_forward_time": 5.610393047332764,
      "step": 15139
    },
    {
      "epoch": 1.4437675476074218e-05,
      "step": 15139,
      "training_step_time": 17.08866047859192
    },
    {
      "epoch": 1.4438629150390626e-05,
      "grad_norm": 0.7267324328422546,
      "learning_rate": 1e-05,
      "loss": 0.0565,
      "step": 15140
    },
    {
      "epoch": 1.4438629150390626e-05,
      "model_forward_time": 5.611627101898193,
      "step": 15140
    },
    {
      "dynamics_loss_avg": 0.04789523370563984,
      "epoch": 1.4438629150390626e-05,
      "step": 15140
    },
    {
      "action_loss_avg": 0.0077116345637477934,
      "epoch": 1.4438629150390626e-05,
      "step": 15140
    },
    {
      "epoch": 1.4438629150390626e-05,
      "step": 15140,
      "training_step_time": 17.090384244918823
    },
    {
      "epoch": 1.4439582824707031e-05,
      "model_forward_time": 5.600810527801514,
      "step": 15141
    },
    {
      "epoch": 1.4439582824707031e-05,
      "step": 15141,
      "training_step_time": 17.45082974433899
    },
    {
      "epoch": 1.4440536499023437e-05,
      "model_forward_time": 5.5946128368377686,
      "step": 15142
    },
    {
      "epoch": 1.4440536499023437e-05,
      "step": 15142,
      "training_step_time": 17.041882276535034
    },
    {
      "epoch": 1.4441490173339844e-05,
      "model_forward_time": 5.5886311531066895,
      "step": 15143
    },
    {
      "epoch": 1.4441490173339844e-05,
      "step": 15143,
      "training_step_time": 17.033119916915894
    },
    {
      "epoch": 1.444244384765625e-05,
      "model_forward_time": 5.584911346435547,
      "step": 15144
    },
    {
      "epoch": 1.444244384765625e-05,
      "step": 15144,
      "training_step_time": 17.038512229919434
    },
    {
      "epoch": 1.4443397521972656e-05,
      "model_forward_time": 5.610435962677002,
      "step": 15145
    },
    {
      "epoch": 1.4443397521972656e-05,
      "step": 15145,
      "training_step_time": 17.125522136688232
    },
    {
      "epoch": 1.4444351196289063e-05,
      "model_forward_time": 5.627391815185547,
      "step": 15146
    },
    {
      "epoch": 1.4444351196289063e-05,
      "step": 15146,
      "training_step_time": 17.13277292251587
    },
    {
      "epoch": 1.4445304870605469e-05,
      "model_forward_time": 5.989231109619141,
      "step": 15147
    },
    {
      "epoch": 1.4445304870605469e-05,
      "step": 15147,
      "training_step_time": 17.464372158050537
    },
    {
      "epoch": 1.4446258544921875e-05,
      "model_forward_time": 5.611110210418701,
      "step": 15148
    },
    {
      "epoch": 1.4446258544921875e-05,
      "step": 15148,
      "training_step_time": 17.097465753555298
    },
    {
      "epoch": 1.444721221923828e-05,
      "model_forward_time": 5.606032133102417,
      "step": 15149
    },
    {
      "epoch": 1.444721221923828e-05,
      "step": 15149,
      "training_step_time": 17.067185163497925
    },
    {
      "epoch": 1.4448165893554688e-05,
      "grad_norm": 1.2319854497909546,
      "learning_rate": 1e-05,
      "loss": 0.0756,
      "step": 15150
    },
    {
      "epoch": 1.4448165893554688e-05,
      "model_forward_time": 5.595518589019775,
      "step": 15150
    },
    {
      "dynamics_loss_avg": 0.06546230725944042,
      "epoch": 1.4448165893554688e-05,
      "step": 15150
    },
    {
      "action_loss_avg": 0.011391194630414247,
      "epoch": 1.4448165893554688e-05,
      "step": 15150
    },
    {
      "epoch": 1.4448165893554688e-05,
      "step": 15150,
      "training_step_time": 17.04065489768982
    },
    {
      "epoch": 1.4449119567871094e-05,
      "model_forward_time": 5.581527233123779,
      "step": 15151
    },
    {
      "epoch": 1.4449119567871094e-05,
      "step": 15151,
      "training_step_time": 17.029602766036987
    },
    {
      "epoch": 1.44500732421875e-05,
      "model_forward_time": 5.9573283195495605,
      "step": 15152
    },
    {
      "epoch": 1.44500732421875e-05,
      "step": 15152,
      "training_step_time": 17.405061960220337
    },
    {
      "epoch": 1.4451026916503907e-05,
      "model_forward_time": 5.59572434425354,
      "step": 15153
    },
    {
      "epoch": 1.4451026916503907e-05,
      "step": 15153,
      "training_step_time": 17.06335425376892
    },
    {
      "epoch": 1.4451980590820312e-05,
      "model_forward_time": 5.614746332168579,
      "step": 15154
    },
    {
      "epoch": 1.4451980590820312e-05,
      "step": 15154,
      "training_step_time": 17.112653493881226
    },
    {
      "epoch": 1.4452934265136718e-05,
      "model_forward_time": 5.610728979110718,
      "step": 15155
    },
    {
      "epoch": 1.4452934265136718e-05,
      "step": 15155,
      "training_step_time": 17.111642599105835
    },
    {
      "epoch": 1.4453887939453126e-05,
      "model_forward_time": 5.616957902908325,
      "step": 15156
    },
    {
      "epoch": 1.4453887939453126e-05,
      "step": 15156,
      "training_step_time": 17.084811210632324
    },
    {
      "epoch": 1.4454841613769531e-05,
      "model_forward_time": 5.60329008102417,
      "step": 15157
    },
    {
      "epoch": 1.4454841613769531e-05,
      "step": 15157,
      "training_step_time": 17.051420211791992
    },
    {
      "epoch": 1.4455795288085937e-05,
      "model_forward_time": 5.962535619735718,
      "step": 15158
    },
    {
      "epoch": 1.4455795288085937e-05,
      "step": 15158,
      "training_step_time": 17.413536310195923
    },
    {
      "epoch": 1.4456748962402344e-05,
      "model_forward_time": 5.586079835891724,
      "step": 15159
    },
    {
      "epoch": 1.4456748962402344e-05,
      "step": 15159,
      "training_step_time": 17.033664226531982
    },
    {
      "epoch": 1.445770263671875e-05,
      "grad_norm": 0.8023310303688049,
      "learning_rate": 1e-05,
      "loss": 0.0731,
      "step": 15160
    },
    {
      "epoch": 1.445770263671875e-05,
      "model_forward_time": 5.599149942398071,
      "step": 15160
    },
    {
      "dynamics_loss_avg": 0.06286787455901503,
      "epoch": 1.445770263671875e-05,
      "step": 15160
    },
    {
      "action_loss_avg": 0.01052292394451797,
      "epoch": 1.445770263671875e-05,
      "step": 15160
    },
    {
      "epoch": 1.445770263671875e-05,
      "step": 15160,
      "training_step_time": 17.072278261184692
    },
    {
      "epoch": 1.4458656311035156e-05,
      "model_forward_time": 5.594759702682495,
      "step": 15161
    },
    {
      "epoch": 1.4458656311035156e-05,
      "step": 15161,
      "training_step_time": 17.06444525718689
    },
    {
      "epoch": 1.4459609985351563e-05,
      "model_forward_time": 5.607241153717041,
      "step": 15162
    },
    {
      "epoch": 1.4459609985351563e-05,
      "step": 15162,
      "training_step_time": 17.04928183555603
    },
    {
      "epoch": 1.4460563659667969e-05,
      "model_forward_time": 5.967788219451904,
      "step": 15163
    },
    {
      "epoch": 1.4460563659667969e-05,
      "step": 15163,
      "training_step_time": 17.42996096611023
    },
    {
      "epoch": 1.4461517333984375e-05,
      "model_forward_time": 5.601172208786011,
      "step": 15164
    },
    {
      "epoch": 1.4461517333984375e-05,
      "step": 15164,
      "training_step_time": 17.09516930580139
    },
    {
      "epoch": 1.446247100830078e-05,
      "model_forward_time": 5.623420000076294,
      "step": 15165
    },
    {
      "epoch": 1.446247100830078e-05,
      "step": 15165,
      "training_step_time": 17.137654066085815
    },
    {
      "epoch": 1.4463424682617188e-05,
      "model_forward_time": 5.617829322814941,
      "step": 15166
    },
    {
      "epoch": 1.4463424682617188e-05,
      "step": 15166,
      "training_step_time": 17.127686738967896
    },
    {
      "epoch": 1.4464378356933594e-05,
      "model_forward_time": 5.600256443023682,
      "step": 15167
    },
    {
      "epoch": 1.4464378356933594e-05,
      "step": 15167,
      "training_step_time": 17.08155059814453
    },
    {
      "epoch": 1.446533203125e-05,
      "model_forward_time": 5.614549875259399,
      "step": 15168
    },
    {
      "epoch": 1.446533203125e-05,
      "step": 15168,
      "training_step_time": 17.084808111190796
    },
    {
      "epoch": 1.4466285705566407e-05,
      "model_forward_time": 5.987132787704468,
      "step": 15169
    },
    {
      "epoch": 1.4466285705566407e-05,
      "step": 15169,
      "training_step_time": 17.453033447265625
    },
    {
      "epoch": 1.4467239379882812e-05,
      "grad_norm": 0.8394421339035034,
      "learning_rate": 1e-05,
      "loss": 0.0543,
      "step": 15170
    },
    {
      "epoch": 1.4467239379882812e-05,
      "model_forward_time": 5.614169120788574,
      "step": 15170
    },
    {
      "dynamics_loss_avg": 0.04608154697343707,
      "epoch": 1.4467239379882812e-05,
      "step": 15170
    },
    {
      "action_loss_avg": 0.006487122317776084,
      "epoch": 1.4467239379882812e-05,
      "step": 15170
    },
    {
      "epoch": 1.4467239379882812e-05,
      "step": 15170,
      "training_step_time": 17.116016149520874
    },
    {
      "epoch": 1.4468193054199218e-05,
      "model_forward_time": 5.622999906539917,
      "step": 15171
    },
    {
      "epoch": 1.4468193054199218e-05,
      "step": 15171,
      "training_step_time": 17.121208429336548
    },
    {
      "epoch": 1.4469146728515626e-05,
      "model_forward_time": 5.614720821380615,
      "step": 15172
    },
    {
      "epoch": 1.4469146728515626e-05,
      "step": 15172,
      "training_step_time": 17.115572929382324
    },
    {
      "epoch": 1.4470100402832031e-05,
      "model_forward_time": 5.6181721687316895,
      "step": 15173
    },
    {
      "epoch": 1.4470100402832031e-05,
      "step": 15173,
      "training_step_time": 17.089123010635376
    },
    {
      "epoch": 1.4471054077148437e-05,
      "model_forward_time": 5.613606214523315,
      "step": 15174
    },
    {
      "epoch": 1.4471054077148437e-05,
      "step": 15174,
      "training_step_time": 17.455836534500122
    },
    {
      "epoch": 1.4472007751464844e-05,
      "model_forward_time": 5.6013922691345215,
      "step": 15175
    },
    {
      "epoch": 1.4472007751464844e-05,
      "step": 15175,
      "training_step_time": 17.085920333862305
    },
    {
      "epoch": 1.447296142578125e-05,
      "model_forward_time": 5.598220586776733,
      "step": 15176
    },
    {
      "epoch": 1.447296142578125e-05,
      "step": 15176,
      "training_step_time": 17.0412175655365
    },
    {
      "epoch": 1.4473915100097656e-05,
      "model_forward_time": 5.580688714981079,
      "step": 15177
    },
    {
      "epoch": 1.4473915100097656e-05,
      "step": 15177,
      "training_step_time": 17.035779237747192
    },
    {
      "epoch": 1.4474868774414063e-05,
      "model_forward_time": 5.603419780731201,
      "step": 15178
    },
    {
      "epoch": 1.4474868774414063e-05,
      "step": 15178,
      "training_step_time": 17.065335988998413
    },
    {
      "epoch": 1.4475822448730469e-05,
      "model_forward_time": 5.60228967666626,
      "step": 15179
    },
    {
      "epoch": 1.4475822448730469e-05,
      "step": 15179,
      "training_step_time": 17.081827402114868
    },
    {
      "epoch": 1.4476776123046875e-05,
      "grad_norm": 0.5896385312080383,
      "learning_rate": 1e-05,
      "loss": 0.0421,
      "step": 15180
    },
    {
      "epoch": 1.4476776123046875e-05,
      "model_forward_time": 5.996984004974365,
      "step": 15180
    },
    {
      "dynamics_loss_avg": 0.0346259206533432,
      "epoch": 1.4476776123046875e-05,
      "step": 15180
    },
    {
      "action_loss_avg": 0.005338529171422124,
      "epoch": 1.4476776123046875e-05,
      "step": 15180
    },
    {
      "epoch": 1.4476776123046875e-05,
      "step": 15180,
      "training_step_time": 17.50125503540039
    },
    {
      "epoch": 1.447772979736328e-05,
      "model_forward_time": 5.611945152282715,
      "step": 15181
    },
    {
      "epoch": 1.447772979736328e-05,
      "step": 15181,
      "training_step_time": 17.08823037147522
    },
    {
      "epoch": 1.4478683471679688e-05,
      "model_forward_time": 5.610219717025757,
      "step": 15182
    },
    {
      "epoch": 1.4478683471679688e-05,
      "step": 15182,
      "training_step_time": 17.07891821861267
    },
    {
      "epoch": 1.4479637145996094e-05,
      "model_forward_time": 5.5971550941467285,
      "step": 15183
    },
    {
      "epoch": 1.4479637145996094e-05,
      "step": 15183,
      "training_step_time": 17.04505729675293
    },
    {
      "epoch": 1.44805908203125e-05,
      "model_forward_time": 5.5931432247161865,
      "step": 15184
    },
    {
      "epoch": 1.44805908203125e-05,
      "step": 15184,
      "training_step_time": 17.040419578552246
    },
    {
      "epoch": 1.4481544494628907e-05,
      "model_forward_time": 5.973231077194214,
      "step": 15185
    },
    {
      "epoch": 1.4481544494628907e-05,
      "step": 15185,
      "training_step_time": 17.44547128677368
    },
    {
      "epoch": 1.4482498168945312e-05,
      "model_forward_time": 5.59781813621521,
      "step": 15186
    },
    {
      "epoch": 1.4482498168945312e-05,
      "step": 15186,
      "training_step_time": 17.062821865081787
    },
    {
      "epoch": 1.4483451843261718e-05,
      "model_forward_time": 5.593484163284302,
      "step": 15187
    },
    {
      "epoch": 1.4483451843261718e-05,
      "step": 15187,
      "training_step_time": 17.066222429275513
    },
    {
      "epoch": 1.4484405517578126e-05,
      "model_forward_time": 5.607973337173462,
      "step": 15188
    },
    {
      "epoch": 1.4484405517578126e-05,
      "step": 15188,
      "training_step_time": 17.099745988845825
    },
    {
      "epoch": 1.4485359191894531e-05,
      "model_forward_time": 5.619896411895752,
      "step": 15189
    },
    {
      "epoch": 1.4485359191894531e-05,
      "step": 15189,
      "training_step_time": 17.12598752975464
    },
    {
      "epoch": 1.4486312866210937e-05,
      "grad_norm": 0.4601859450340271,
      "learning_rate": 1e-05,
      "loss": 0.0465,
      "step": 15190
    },
    {
      "epoch": 1.4486312866210937e-05,
      "model_forward_time": 5.620638132095337,
      "step": 15190
    },
    {
      "dynamics_loss_avg": 0.045818698778748515,
      "epoch": 1.4486312866210937e-05,
      "step": 15190
    },
    {
      "action_loss_avg": 0.0058940171264112,
      "epoch": 1.4486312866210937e-05,
      "step": 15190
    },
    {
      "epoch": 1.4486312866210937e-05,
      "step": 15190,
      "training_step_time": 17.15224838256836
    },
    {
      "epoch": 1.4487266540527344e-05,
      "model_forward_time": 5.993769884109497,
      "step": 15191
    },
    {
      "epoch": 1.4487266540527344e-05,
      "step": 15191,
      "training_step_time": 17.492840051651
    },
    {
      "epoch": 1.448822021484375e-05,
      "model_forward_time": 5.623552322387695,
      "step": 15192
    },
    {
      "epoch": 1.448822021484375e-05,
      "step": 15192,
      "training_step_time": 17.105307579040527
    },
    {
      "epoch": 1.4489173889160156e-05,
      "model_forward_time": 5.608067989349365,
      "step": 15193
    },
    {
      "epoch": 1.4489173889160156e-05,
      "step": 15193,
      "training_step_time": 17.09282422065735
    },
    {
      "epoch": 1.4490127563476563e-05,
      "model_forward_time": 5.607054710388184,
      "step": 15194
    },
    {
      "epoch": 1.4490127563476563e-05,
      "step": 15194,
      "training_step_time": 17.071204900741577
    },
    {
      "epoch": 1.4491081237792969e-05,
      "model_forward_time": 5.598862886428833,
      "step": 15195
    },
    {
      "epoch": 1.4491081237792969e-05,
      "step": 15195,
      "training_step_time": 17.028722524642944
    },
    {
      "epoch": 1.4492034912109375e-05,
      "model_forward_time": 5.957032680511475,
      "step": 15196
    },
    {
      "epoch": 1.4492034912109375e-05,
      "step": 15196,
      "training_step_time": 17.397988080978394
    },
    {
      "epoch": 1.449298858642578e-05,
      "model_forward_time": 5.593069314956665,
      "step": 15197
    },
    {
      "epoch": 1.449298858642578e-05,
      "step": 15197,
      "training_step_time": 17.03507399559021
    },
    {
      "epoch": 1.4493942260742188e-05,
      "model_forward_time": 5.5980544090271,
      "step": 15198
    },
    {
      "epoch": 1.4493942260742188e-05,
      "step": 15198,
      "training_step_time": 17.071051597595215
    },
    {
      "epoch": 1.4494895935058594e-05,
      "model_forward_time": 5.6091694831848145,
      "step": 15199
    },
    {
      "epoch": 1.4494895935058594e-05,
      "step": 15199,
      "training_step_time": 17.101648092269897
    },
    {
      "epoch": 1.4495849609375e-05,
      "grad_norm": 0.42742621898651123,
      "learning_rate": 1e-05,
      "loss": 0.0602,
      "step": 15200
    },
    {
      "epoch": 1.4495849609375e-05,
      "model_forward_time": 5.6188342571258545,
      "step": 15200
    },
    {
      "dynamics_loss_avg": 0.05086830798536539,
      "epoch": 1.4495849609375e-05,
      "step": 15200
    },
    {
      "action_loss_avg": 0.008416688977740705,
      "epoch": 1.4495849609375e-05,
      "step": 15200
    },
    {
      "epoch": 1.4495849609375e-05,
      "step": 15200,
      "training_step_time": 17.122897624969482
    },
    {
      "epoch": 1.4496803283691407e-05,
      "model_forward_time": 5.609287977218628,
      "step": 15201
    },
    {
      "epoch": 1.4496803283691407e-05,
      "step": 15201,
      "training_step_time": 17.0740647315979
    },
    {
      "epoch": 1.4497756958007812e-05,
      "model_forward_time": 5.952450275421143,
      "step": 15202
    },
    {
      "epoch": 1.4497756958007812e-05,
      "step": 15202,
      "training_step_time": 17.37231945991516
    },
    {
      "epoch": 1.4498710632324218e-05,
      "model_forward_time": 5.566939115524292,
      "step": 15203
    },
    {
      "epoch": 1.4498710632324218e-05,
      "step": 15203,
      "training_step_time": 17.003941535949707
    },
    {
      "epoch": 1.4499664306640626e-05,
      "model_forward_time": 5.565096139907837,
      "step": 15204
    },
    {
      "epoch": 1.4499664306640626e-05,
      "step": 15204,
      "training_step_time": 17.000199794769287
    },
    {
      "epoch": 1.4500617980957031e-05,
      "model_forward_time": 5.566339492797852,
      "step": 15205
    },
    {
      "epoch": 1.4500617980957031e-05,
      "step": 15205,
      "training_step_time": 16.96767807006836
    },
    {
      "epoch": 1.4501571655273437e-05,
      "model_forward_time": 5.57195782661438,
      "step": 15206
    },
    {
      "epoch": 1.4501571655273437e-05,
      "step": 15206,
      "training_step_time": 16.98974585533142
    },
    {
      "epoch": 1.4502525329589844e-05,
      "model_forward_time": 5.586199998855591,
      "step": 15207
    },
    {
      "epoch": 1.4502525329589844e-05,
      "step": 15207,
      "training_step_time": 17.38371992111206
    },
    {
      "epoch": 1.450347900390625e-05,
      "model_forward_time": 5.571703195571899,
      "step": 15208
    },
    {
      "epoch": 1.450347900390625e-05,
      "step": 15208,
      "training_step_time": 17.003037929534912
    },
    {
      "epoch": 1.4504432678222656e-05,
      "model_forward_time": 5.580884695053101,
      "step": 15209
    },
    {
      "epoch": 1.4504432678222656e-05,
      "step": 15209,
      "training_step_time": 16.997395992279053
    },
    {
      "epoch": 1.4505386352539063e-05,
      "grad_norm": 0.5632864832878113,
      "learning_rate": 1e-05,
      "loss": 0.0707,
      "step": 15210
    },
    {
      "epoch": 1.4505386352539063e-05,
      "model_forward_time": 5.586892127990723,
      "step": 15210
    },
    {
      "dynamics_loss_avg": 0.059616444259881975,
      "epoch": 1.4505386352539063e-05,
      "step": 15210
    },
    {
      "action_loss_avg": 0.009637093544006348,
      "epoch": 1.4505386352539063e-05,
      "step": 15210
    },
    {
      "epoch": 1.4505386352539063e-05,
      "step": 15210,
      "training_step_time": 17.04203152656555
    },
    {
      "epoch": 1.4506340026855469e-05,
      "model_forward_time": 5.59433388710022,
      "step": 15211
    },
    {
      "epoch": 1.4506340026855469e-05,
      "step": 15211,
      "training_step_time": 17.030741214752197
    },
    {
      "epoch": 1.4507293701171875e-05,
      "model_forward_time": 5.608144521713257,
      "step": 15212
    },
    {
      "epoch": 1.4507293701171875e-05,
      "step": 15212,
      "training_step_time": 17.087968826293945
    },
    {
      "epoch": 1.450824737548828e-05,
      "model_forward_time": 5.9946019649505615,
      "step": 15213
    },
    {
      "epoch": 1.450824737548828e-05,
      "step": 15213,
      "training_step_time": 17.44817328453064
    },
    {
      "epoch": 1.4509201049804688e-05,
      "model_forward_time": 5.593497037887573,
      "step": 15214
    },
    {
      "epoch": 1.4509201049804688e-05,
      "step": 15214,
      "training_step_time": 17.028045654296875
    },
    {
      "epoch": 1.4510154724121094e-05,
      "model_forward_time": 5.584126949310303,
      "step": 15215
    },
    {
      "epoch": 1.4510154724121094e-05,
      "step": 15215,
      "training_step_time": 17.022390365600586
    },
    {
      "epoch": 1.45111083984375e-05,
      "model_forward_time": 5.582803249359131,
      "step": 15216
    },
    {
      "epoch": 1.45111083984375e-05,
      "step": 15216,
      "training_step_time": 17.032209396362305
    },
    {
      "epoch": 1.4512062072753907e-05,
      "model_forward_time": 5.598448038101196,
      "step": 15217
    },
    {
      "epoch": 1.4512062072753907e-05,
      "step": 15217,
      "training_step_time": 17.080137491226196
    },
    {
      "epoch": 1.4513015747070312e-05,
      "model_forward_time": 5.96624493598938,
      "step": 15218
    },
    {
      "epoch": 1.4513015747070312e-05,
      "step": 15218,
      "training_step_time": 17.44916534423828
    },
    {
      "epoch": 1.4513969421386718e-05,
      "model_forward_time": 5.606735467910767,
      "step": 15219
    },
    {
      "epoch": 1.4513969421386718e-05,
      "step": 15219,
      "training_step_time": 17.082948446273804
    },
    {
      "epoch": 1.4514923095703126e-05,
      "grad_norm": 0.6608652472496033,
      "learning_rate": 1e-05,
      "loss": 0.0579,
      "step": 15220
    },
    {
      "epoch": 1.4514923095703126e-05,
      "model_forward_time": 5.610371828079224,
      "step": 15220
    },
    {
      "dynamics_loss_avg": 0.05087581053376198,
      "epoch": 1.4514923095703126e-05,
      "step": 15220
    },
    {
      "action_loss_avg": 0.0063221438671462234,
      "epoch": 1.4514923095703126e-05,
      "step": 15220
    },
    {
      "epoch": 1.4514923095703126e-05,
      "step": 15220,
      "training_step_time": 17.09816598892212
    },
    {
      "epoch": 1.4515876770019531e-05,
      "model_forward_time": 5.609850883483887,
      "step": 15221
    },
    {
      "epoch": 1.4515876770019531e-05,
      "step": 15221,
      "training_step_time": 17.097469091415405
    },
    {
      "epoch": 1.4516830444335937e-05,
      "model_forward_time": 5.6105875968933105,
      "step": 15222
    },
    {
      "epoch": 1.4516830444335937e-05,
      "step": 15222,
      "training_step_time": 17.092861652374268
    },
    {
      "epoch": 1.4517784118652344e-05,
      "model_forward_time": 5.606916427612305,
      "step": 15223
    },
    {
      "epoch": 1.4517784118652344e-05,
      "step": 15223,
      "training_step_time": 17.08620595932007
    },
    {
      "epoch": 1.451873779296875e-05,
      "model_forward_time": 5.982690811157227,
      "step": 15224
    },
    {
      "epoch": 1.451873779296875e-05,
      "step": 15224,
      "training_step_time": 17.451743364334106
    },
    {
      "epoch": 1.4519691467285156e-05,
      "model_forward_time": 5.6027631759643555,
      "step": 15225
    },
    {
      "epoch": 1.4519691467285156e-05,
      "step": 15225,
      "training_step_time": 17.08141255378723
    },
    {
      "epoch": 1.4520645141601563e-05,
      "model_forward_time": 5.59450101852417,
      "step": 15226
    },
    {
      "epoch": 1.4520645141601563e-05,
      "step": 15226,
      "training_step_time": 17.030614852905273
    },
    {
      "epoch": 1.4521598815917969e-05,
      "model_forward_time": 5.596486330032349,
      "step": 15227
    },
    {
      "epoch": 1.4521598815917969e-05,
      "step": 15227,
      "training_step_time": 17.039711475372314
    },
    {
      "epoch": 1.4522552490234375e-05,
      "model_forward_time": 5.595116853713989,
      "step": 15228
    },
    {
      "epoch": 1.4522552490234375e-05,
      "step": 15228,
      "training_step_time": 17.044548273086548
    },
    {
      "epoch": 1.452350616455078e-05,
      "model_forward_time": 5.947019100189209,
      "step": 15229
    },
    {
      "epoch": 1.452350616455078e-05,
      "step": 15229,
      "training_step_time": 17.389154195785522
    },
    {
      "epoch": 1.4524459838867188e-05,
      "grad_norm": 1.1286404132843018,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 15230
    },
    {
      "epoch": 1.4524459838867188e-05,
      "model_forward_time": 5.600013732910156,
      "step": 15230
    },
    {
      "dynamics_loss_avg": 0.062445553578436376,
      "epoch": 1.4524459838867188e-05,
      "step": 15230
    },
    {
      "action_loss_avg": 0.008788228128105402,
      "epoch": 1.4524459838867188e-05,
      "step": 15230
    },
    {
      "epoch": 1.4524459838867188e-05,
      "step": 15230,
      "training_step_time": 17.08307695388794
    },
    {
      "epoch": 1.4525413513183594e-05,
      "model_forward_time": 5.610121011734009,
      "step": 15231
    },
    {
      "epoch": 1.4525413513183594e-05,
      "step": 15231,
      "training_step_time": 17.112644910812378
    },
    {
      "epoch": 1.45263671875e-05,
      "model_forward_time": 5.61612868309021,
      "step": 15232
    },
    {
      "epoch": 1.45263671875e-05,
      "step": 15232,
      "training_step_time": 17.127174615859985
    },
    {
      "epoch": 1.4527320861816407e-05,
      "model_forward_time": 5.610230445861816,
      "step": 15233
    },
    {
      "epoch": 1.4527320861816407e-05,
      "step": 15233,
      "training_step_time": 17.082696437835693
    },
    {
      "epoch": 1.4528274536132812e-05,
      "model_forward_time": 5.588904142379761,
      "step": 15234
    },
    {
      "epoch": 1.4528274536132812e-05,
      "step": 15234,
      "training_step_time": 17.031335592269897
    },
    {
      "epoch": 1.4529228210449218e-05,
      "model_forward_time": 5.951288223266602,
      "step": 15235
    },
    {
      "epoch": 1.4529228210449218e-05,
      "step": 15235,
      "training_step_time": 17.367666482925415
    },
    {
      "epoch": 1.4530181884765626e-05,
      "model_forward_time": 5.566397666931152,
      "step": 15236
    },
    {
      "epoch": 1.4530181884765626e-05,
      "step": 15236,
      "training_step_time": 16.985187530517578
    },
    {
      "epoch": 1.4531135559082031e-05,
      "model_forward_time": 5.567206859588623,
      "step": 15237
    },
    {
      "epoch": 1.4531135559082031e-05,
      "step": 15237,
      "training_step_time": 16.999974966049194
    },
    {
      "epoch": 1.4532089233398437e-05,
      "model_forward_time": 5.579720497131348,
      "step": 15238
    },
    {
      "epoch": 1.4532089233398437e-05,
      "step": 15238,
      "training_step_time": 17.033293962478638
    },
    {
      "epoch": 1.4533042907714844e-05,
      "model_forward_time": 5.572054147720337,
      "step": 15239
    },
    {
      "epoch": 1.4533042907714844e-05,
      "step": 15239,
      "training_step_time": 16.98945689201355
    },
    {
      "epoch": 1.453399658203125e-05,
      "grad_norm": 0.8347634077072144,
      "learning_rate": 1e-05,
      "loss": 0.069,
      "step": 15240
    },
    {
      "epoch": 1.453399658203125e-05,
      "model_forward_time": 5.600121021270752,
      "step": 15240
    },
    {
      "dynamics_loss_avg": 0.05951152518391609,
      "epoch": 1.453399658203125e-05,
      "step": 15240
    },
    {
      "action_loss_avg": 0.008995413687080145,
      "epoch": 1.453399658203125e-05,
      "step": 15240
    },
    {
      "epoch": 1.453399658203125e-05,
      "step": 15240,
      "training_step_time": 17.435128927230835
    },
    {
      "epoch": 1.4534950256347656e-05,
      "model_forward_time": 5.6156229972839355,
      "step": 15241
    },
    {
      "epoch": 1.4534950256347656e-05,
      "step": 15241,
      "training_step_time": 17.116624116897583
    },
    {
      "epoch": 1.4535903930664063e-05,
      "model_forward_time": 5.6208202838897705,
      "step": 15242
    },
    {
      "epoch": 1.4535903930664063e-05,
      "step": 15242,
      "training_step_time": 17.13199472427368
    },
    {
      "epoch": 1.4536857604980469e-05,
      "model_forward_time": 5.622958183288574,
      "step": 15243
    },
    {
      "epoch": 1.4536857604980469e-05,
      "step": 15243,
      "training_step_time": 17.12753391265869
    },
    {
      "epoch": 1.4537811279296875e-05,
      "model_forward_time": 5.594088077545166,
      "step": 15244
    },
    {
      "epoch": 1.4537811279296875e-05,
      "step": 15244,
      "training_step_time": 17.0288143157959
    },
    {
      "epoch": 1.453876495361328e-05,
      "model_forward_time": 5.582260608673096,
      "step": 15245
    },
    {
      "epoch": 1.453876495361328e-05,
      "step": 15245,
      "training_step_time": 17.021121740341187
    },
    {
      "epoch": 1.4539718627929688e-05,
      "model_forward_time": 5.947533130645752,
      "step": 15246
    },
    {
      "epoch": 1.4539718627929688e-05,
      "step": 15246,
      "training_step_time": 17.377514839172363
    },
    {
      "epoch": 1.4540672302246094e-05,
      "model_forward_time": 5.5812718868255615,
      "step": 15247
    },
    {
      "epoch": 1.4540672302246094e-05,
      "step": 15247,
      "training_step_time": 16.99727487564087
    },
    {
      "epoch": 1.45416259765625e-05,
      "model_forward_time": 5.5829176902771,
      "step": 15248
    },
    {
      "epoch": 1.45416259765625e-05,
      "step": 15248,
      "training_step_time": 17.030914545059204
    },
    {
      "epoch": 1.4542579650878907e-05,
      "model_forward_time": 5.604517698287964,
      "step": 15249
    },
    {
      "epoch": 1.4542579650878907e-05,
      "step": 15249,
      "training_step_time": 17.06668210029602
    },
    {
      "epoch": 1.4543533325195312e-05,
      "grad_norm": 0.6938426494598389,
      "learning_rate": 1e-05,
      "loss": 0.054,
      "step": 15250
    },
    {
      "epoch": 1.4543533325195312e-05,
      "model_forward_time": 5.605069637298584,
      "step": 15250
    },
    {
      "dynamics_loss_avg": 0.04767227955162525,
      "epoch": 1.4543533325195312e-05,
      "step": 15250
    },
    {
      "action_loss_avg": 0.007311788690276444,
      "epoch": 1.4543533325195312e-05,
      "step": 15250
    },
    {
      "epoch": 1.4543533325195312e-05,
      "step": 15250,
      "training_step_time": 17.093652725219727
    },
    {
      "epoch": 1.4544486999511718e-05,
      "model_forward_time": 5.980765104293823,
      "step": 15251
    },
    {
      "epoch": 1.4544486999511718e-05,
      "step": 15251,
      "training_step_time": 17.48153066635132
    },
    {
      "epoch": 1.4545440673828126e-05,
      "model_forward_time": 5.653198480606079,
      "step": 15252
    },
    {
      "epoch": 1.4545440673828126e-05,
      "step": 15252,
      "training_step_time": 17.20202660560608
    },
    {
      "epoch": 1.4546394348144531e-05,
      "model_forward_time": 5.614515542984009,
      "step": 15253
    },
    {
      "epoch": 1.4546394348144531e-05,
      "step": 15253,
      "training_step_time": 17.1225528717041
    },
    {
      "epoch": 1.4547348022460937e-05,
      "model_forward_time": 5.625678300857544,
      "step": 15254
    },
    {
      "epoch": 1.4547348022460937e-05,
      "step": 15254,
      "training_step_time": 17.10791039466858
    },
    {
      "epoch": 1.4548301696777344e-05,
      "model_forward_time": 5.588165044784546,
      "step": 15255
    },
    {
      "epoch": 1.4548301696777344e-05,
      "step": 15255,
      "training_step_time": 17.008522987365723
    },
    {
      "epoch": 1.454925537109375e-05,
      "model_forward_time": 5.562232732772827,
      "step": 15256
    },
    {
      "epoch": 1.454925537109375e-05,
      "step": 15256,
      "training_step_time": 16.982066869735718
    },
    {
      "epoch": 1.4550209045410156e-05,
      "model_forward_time": 5.9464099407196045,
      "step": 15257
    },
    {
      "epoch": 1.4550209045410156e-05,
      "step": 15257,
      "training_step_time": 17.375397205352783
    },
    {
      "epoch": 1.4551162719726562e-05,
      "model_forward_time": 5.57505989074707,
      "step": 15258
    },
    {
      "epoch": 1.4551162719726562e-05,
      "step": 15258,
      "training_step_time": 16.993057012557983
    },
    {
      "epoch": 1.4552116394042969e-05,
      "model_forward_time": 5.574229955673218,
      "step": 15259
    },
    {
      "epoch": 1.4552116394042969e-05,
      "step": 15259,
      "training_step_time": 16.990223169326782
    },
    {
      "epoch": 1.4553070068359375e-05,
      "grad_norm": 0.7678762674331665,
      "learning_rate": 1e-05,
      "loss": 0.0597,
      "step": 15260
    },
    {
      "epoch": 1.4553070068359375e-05,
      "model_forward_time": 5.567692041397095,
      "step": 15260
    },
    {
      "dynamics_loss_avg": 0.053280357457697394,
      "epoch": 1.4553070068359375e-05,
      "step": 15260
    },
    {
      "action_loss_avg": 0.008754466520622372,
      "epoch": 1.4553070068359375e-05,
      "step": 15260
    },
    {
      "epoch": 1.4553070068359375e-05,
      "step": 15260,
      "training_step_time": 17.004435777664185
    },
    {
      "epoch": 1.455402374267578e-05,
      "model_forward_time": 5.571161508560181,
      "step": 15261
    },
    {
      "epoch": 1.455402374267578e-05,
      "step": 15261,
      "training_step_time": 17.00464367866516
    },
    {
      "epoch": 1.4554977416992188e-05,
      "model_forward_time": 5.932361364364624,
      "step": 15262
    },
    {
      "epoch": 1.4554977416992188e-05,
      "step": 15262,
      "training_step_time": 17.334526300430298
    },
    {
      "epoch": 1.4555931091308594e-05,
      "model_forward_time": 5.590902090072632,
      "step": 15263
    },
    {
      "epoch": 1.4555931091308594e-05,
      "step": 15263,
      "training_step_time": 17.04629945755005
    },
    {
      "epoch": 1.4556884765625e-05,
      "model_forward_time": 5.601153135299683,
      "step": 15264
    },
    {
      "epoch": 1.4556884765625e-05,
      "step": 15264,
      "training_step_time": 17.07880139350891
    },
    {
      "epoch": 1.4557838439941407e-05,
      "model_forward_time": 5.611047029495239,
      "step": 15265
    },
    {
      "epoch": 1.4557838439941407e-05,
      "step": 15265,
      "training_step_time": 17.10398817062378
    },
    {
      "epoch": 1.4558792114257813e-05,
      "model_forward_time": 5.598066806793213,
      "step": 15266
    },
    {
      "epoch": 1.4558792114257813e-05,
      "step": 15266,
      "training_step_time": 17.048368453979492
    },
    {
      "epoch": 1.4559745788574218e-05,
      "model_forward_time": 5.5788657665252686,
      "step": 15267
    },
    {
      "epoch": 1.4559745788574218e-05,
      "step": 15267,
      "training_step_time": 16.986705541610718
    },
    {
      "epoch": 1.4560699462890626e-05,
      "model_forward_time": 5.942527770996094,
      "step": 15268
    },
    {
      "epoch": 1.4560699462890626e-05,
      "step": 15268,
      "training_step_time": 17.38050150871277
    },
    {
      "epoch": 1.4561653137207031e-05,
      "model_forward_time": 5.569839954376221,
      "step": 15269
    },
    {
      "epoch": 1.4561653137207031e-05,
      "step": 15269,
      "training_step_time": 16.98291325569153
    },
    {
      "epoch": 1.4562606811523437e-05,
      "grad_norm": 0.794279158115387,
      "learning_rate": 1e-05,
      "loss": 0.0569,
      "step": 15270
    },
    {
      "epoch": 1.4562606811523437e-05,
      "model_forward_time": 5.569190502166748,
      "step": 15270
    },
    {
      "dynamics_loss_avg": 0.04497104156762362,
      "epoch": 1.4562606811523437e-05,
      "step": 15270
    },
    {
      "action_loss_avg": 0.009621129313018173,
      "epoch": 1.4562606811523437e-05,
      "step": 15270
    },
    {
      "epoch": 1.4562606811523437e-05,
      "step": 15270,
      "training_step_time": 16.981403827667236
    },
    {
      "epoch": 1.4563560485839845e-05,
      "model_forward_time": 5.588108777999878,
      "step": 15271
    },
    {
      "epoch": 1.4563560485839845e-05,
      "step": 15271,
      "training_step_time": 17.03313660621643
    },
    {
      "epoch": 1.456451416015625e-05,
      "model_forward_time": 5.6058549880981445,
      "step": 15272
    },
    {
      "epoch": 1.456451416015625e-05,
      "step": 15272,
      "training_step_time": 17.088582515716553
    },
    {
      "epoch": 1.4565467834472656e-05,
      "model_forward_time": 5.615342617034912,
      "step": 15273
    },
    {
      "epoch": 1.4565467834472656e-05,
      "step": 15273,
      "training_step_time": 17.465195655822754
    },
    {
      "epoch": 1.4566421508789062e-05,
      "model_forward_time": 5.625648021697998,
      "step": 15274
    },
    {
      "epoch": 1.4566421508789062e-05,
      "step": 15274,
      "training_step_time": 17.133328914642334
    },
    {
      "epoch": 1.4567375183105469e-05,
      "model_forward_time": 5.60967493057251,
      "step": 15275
    },
    {
      "epoch": 1.4567375183105469e-05,
      "step": 15275,
      "training_step_time": 17.113673210144043
    },
    {
      "epoch": 1.4568328857421875e-05,
      "model_forward_time": 5.614876985549927,
      "step": 15276
    },
    {
      "epoch": 1.4568328857421875e-05,
      "step": 15276,
      "training_step_time": 17.10944366455078
    },
    {
      "epoch": 1.456928253173828e-05,
      "model_forward_time": 5.61859130859375,
      "step": 15277
    },
    {
      "epoch": 1.456928253173828e-05,
      "step": 15277,
      "training_step_time": 17.11387038230896
    },
    {
      "epoch": 1.4570236206054688e-05,
      "model_forward_time": 5.608607292175293,
      "step": 15278
    },
    {
      "epoch": 1.4570236206054688e-05,
      "step": 15278,
      "training_step_time": 17.096457481384277
    },
    {
      "epoch": 1.4571189880371094e-05,
      "model_forward_time": 5.991725444793701,
      "step": 15279
    },
    {
      "epoch": 1.4571189880371094e-05,
      "step": 15279,
      "training_step_time": 17.46642279624939
    },
    {
      "epoch": 1.45721435546875e-05,
      "grad_norm": 0.5760214924812317,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 15280
    },
    {
      "epoch": 1.45721435546875e-05,
      "model_forward_time": 5.615643501281738,
      "step": 15280
    },
    {
      "dynamics_loss_avg": 0.0628344502300024,
      "epoch": 1.45721435546875e-05,
      "step": 15280
    },
    {
      "action_loss_avg": 0.00913855736143887,
      "epoch": 1.45721435546875e-05,
      "step": 15280
    },
    {
      "epoch": 1.45721435546875e-05,
      "step": 15280,
      "training_step_time": 17.10863447189331
    },
    {
      "epoch": 1.4573097229003907e-05,
      "model_forward_time": 5.5984814167022705,
      "step": 15281
    },
    {
      "epoch": 1.4573097229003907e-05,
      "step": 15281,
      "training_step_time": 17.08839201927185
    },
    {
      "epoch": 1.4574050903320313e-05,
      "model_forward_time": 5.6107707023620605,
      "step": 15282
    },
    {
      "epoch": 1.4574050903320313e-05,
      "step": 15282,
      "training_step_time": 17.109822273254395
    },
    {
      "epoch": 1.4575004577636718e-05,
      "model_forward_time": 5.614155292510986,
      "step": 15283
    },
    {
      "epoch": 1.4575004577636718e-05,
      "step": 15283,
      "training_step_time": 17.12153387069702
    },
    {
      "epoch": 1.4575958251953126e-05,
      "model_forward_time": 5.9762091636657715,
      "step": 15284
    },
    {
      "epoch": 1.4575958251953126e-05,
      "step": 15284,
      "training_step_time": 17.45105814933777
    },
    {
      "epoch": 1.4576911926269531e-05,
      "model_forward_time": 5.602057456970215,
      "step": 15285
    },
    {
      "epoch": 1.4576911926269531e-05,
      "step": 15285,
      "training_step_time": 17.080453395843506
    },
    {
      "epoch": 1.4577865600585937e-05,
      "model_forward_time": 5.609231948852539,
      "step": 15286
    },
    {
      "epoch": 1.4577865600585937e-05,
      "step": 15286,
      "training_step_time": 17.078375577926636
    },
    {
      "epoch": 1.4578819274902345e-05,
      "model_forward_time": 5.601596355438232,
      "step": 15287
    },
    {
      "epoch": 1.4578819274902345e-05,
      "step": 15287,
      "training_step_time": 17.04864501953125
    },
    {
      "epoch": 1.457977294921875e-05,
      "model_forward_time": 5.5917809009552,
      "step": 15288
    },
    {
      "epoch": 1.457977294921875e-05,
      "step": 15288,
      "training_step_time": 17.040971755981445
    },
    {
      "epoch": 1.4580726623535156e-05,
      "model_forward_time": 5.581071376800537,
      "step": 15289
    },
    {
      "epoch": 1.4580726623535156e-05,
      "step": 15289,
      "training_step_time": 17.00392723083496
    },
    {
      "epoch": 1.4581680297851562e-05,
      "grad_norm": 0.5373398065567017,
      "learning_rate": 1e-05,
      "loss": 0.0671,
      "step": 15290
    },
    {
      "epoch": 1.4581680297851562e-05,
      "model_forward_time": 5.9608845710754395,
      "step": 15290
    },
    {
      "dynamics_loss_avg": 0.05437921490520239,
      "epoch": 1.4581680297851562e-05,
      "step": 15290
    },
    {
      "action_loss_avg": 0.006379633909091353,
      "epoch": 1.4581680297851562e-05,
      "step": 15290
    },
    {
      "epoch": 1.4581680297851562e-05,
      "step": 15290,
      "training_step_time": 17.418946266174316
    },
    {
      "epoch": 1.4582633972167969e-05,
      "model_forward_time": 5.61034893989563,
      "step": 15291
    },
    {
      "epoch": 1.4582633972167969e-05,
      "step": 15291,
      "training_step_time": 17.10493516921997
    },
    {
      "epoch": 1.4583587646484375e-05,
      "model_forward_time": 5.619113206863403,
      "step": 15292
    },
    {
      "epoch": 1.4583587646484375e-05,
      "step": 15292,
      "training_step_time": 17.1361403465271
    },
    {
      "epoch": 1.458454132080078e-05,
      "model_forward_time": 5.620203018188477,
      "step": 15293
    },
    {
      "epoch": 1.458454132080078e-05,
      "step": 15293,
      "training_step_time": 17.1330463886261
    },
    {
      "epoch": 1.4585494995117188e-05,
      "model_forward_time": 5.625395774841309,
      "step": 15294
    },
    {
      "epoch": 1.4585494995117188e-05,
      "step": 15294,
      "training_step_time": 17.114843130111694
    },
    {
      "epoch": 1.4586448669433594e-05,
      "model_forward_time": 5.953488349914551,
      "step": 15295
    },
    {
      "epoch": 1.4586448669433594e-05,
      "step": 15295,
      "training_step_time": 17.389721870422363
    },
    {
      "epoch": 1.458740234375e-05,
      "model_forward_time": 5.5845701694488525,
      "step": 15296
    },
    {
      "epoch": 1.458740234375e-05,
      "step": 15296,
      "training_step_time": 17.006678104400635
    },
    {
      "epoch": 1.4588356018066407e-05,
      "model_forward_time": 5.574597597122192,
      "step": 15297
    },
    {
      "epoch": 1.4588356018066407e-05,
      "step": 15297,
      "training_step_time": 16.996013641357422
    },
    {
      "epoch": 1.4589309692382813e-05,
      "model_forward_time": 5.575415134429932,
      "step": 15298
    },
    {
      "epoch": 1.4589309692382813e-05,
      "step": 15298,
      "training_step_time": 17.000526666641235
    },
    {
      "epoch": 1.4590263366699218e-05,
      "model_forward_time": 5.5779852867126465,
      "step": 15299
    },
    {
      "epoch": 1.4590263366699218e-05,
      "step": 15299,
      "training_step_time": 17.011753797531128
    },
    {
      "epoch": 1.4591217041015626e-05,
      "grad_norm": 0.5832412242889404,
      "learning_rate": 1e-05,
      "loss": 0.0465,
      "step": 15300
    },
    {
      "epoch": 1.4591217041015626e-05,
      "model_forward_time": 5.587401390075684,
      "step": 15300
    },
    {
      "dynamics_loss_avg": 0.04555051140487194,
      "epoch": 1.4591217041015626e-05,
      "step": 15300
    },
    {
      "action_loss_avg": 0.005379875004291534,
      "epoch": 1.4591217041015626e-05,
      "step": 15300
    },
    {
      "epoch": 1.4591217041015626e-05,
      "step": 15300,
      "training_step_time": 17.003023862838745
    },
    {
      "epoch": 1.4592170715332031e-05,
      "model_forward_time": 5.952931642532349,
      "step": 15301
    },
    {
      "epoch": 1.4592170715332031e-05,
      "step": 15301,
      "training_step_time": 17.441105842590332
    },
    {
      "epoch": 1.4593124389648437e-05,
      "model_forward_time": 5.622192144393921,
      "step": 15302
    },
    {
      "epoch": 1.4593124389648437e-05,
      "step": 15302,
      "training_step_time": 17.140246629714966
    },
    {
      "epoch": 1.4594078063964845e-05,
      "model_forward_time": 4.294513702392578,
      "step": 15303
    },
    {
      "epoch": 1.4594078063964845e-05,
      "step": 15303,
      "training_step_time": 12.841078519821167
    },
    {
      "epoch": 1.459503173828125e-05,
      "model_forward_time": 5.608836889266968,
      "step": 15304
    },
    {
      "epoch": 1.459503173828125e-05,
      "step": 15304,
      "training_step_time": 17.10442352294922
    },
    {
      "epoch": 1.4595985412597656e-05,
      "model_forward_time": 5.611961841583252,
      "step": 15305
    },
    {
      "epoch": 1.4595985412597656e-05,
      "step": 15305,
      "training_step_time": 17.099881649017334
    },
    {
      "epoch": 1.4596939086914062e-05,
      "model_forward_time": 5.6093809604644775,
      "step": 15306
    },
    {
      "epoch": 1.4596939086914062e-05,
      "step": 15306,
      "training_step_time": 17.081446409225464
    },
    {
      "epoch": 1.4597892761230469e-05,
      "model_forward_time": 5.969799757003784,
      "step": 15307
    },
    {
      "epoch": 1.4597892761230469e-05,
      "step": 15307,
      "training_step_time": 17.465097904205322
    },
    {
      "epoch": 1.4598846435546875e-05,
      "model_forward_time": 5.613532543182373,
      "step": 15308
    },
    {
      "epoch": 1.4598846435546875e-05,
      "step": 15308,
      "training_step_time": 17.113056659698486
    },
    {
      "epoch": 1.459980010986328e-05,
      "model_forward_time": 5.621262073516846,
      "step": 15309
    },
    {
      "epoch": 1.459980010986328e-05,
      "step": 15309,
      "training_step_time": 17.127795457839966
    },
    {
      "epoch": 1.4600753784179688e-05,
      "grad_norm": 0.1803809255361557,
      "learning_rate": 1e-05,
      "loss": 0.0461,
      "step": 15310
    },
    {
      "epoch": 1.4600753784179688e-05,
      "model_forward_time": 4.294518232345581,
      "step": 15310
    },
    {
      "dynamics_loss_avg": 0.04312491538003087,
      "epoch": 1.4600753784179688e-05,
      "step": 15310
    },
    {
      "action_loss_avg": 0.00753904776647687,
      "epoch": 1.4600753784179688e-05,
      "step": 15310
    },
    {
      "epoch": 1.4600753784179688e-05,
      "step": 15310,
      "training_step_time": 12.837903261184692
    },
    {
      "epoch": 1.4601707458496094e-05,
      "model_forward_time": 5.627963542938232,
      "step": 15311
    },
    {
      "epoch": 1.4601707458496094e-05,
      "step": 15311,
      "training_step_time": 17.125211000442505
    },
    {
      "epoch": 1.46026611328125e-05,
      "model_forward_time": 5.973189353942871,
      "step": 15312
    },
    {
      "epoch": 1.46026611328125e-05,
      "step": 15312,
      "training_step_time": 17.409661293029785
    },
    {
      "epoch": 1.4603614807128907e-05,
      "model_forward_time": 5.571640253067017,
      "step": 15313
    },
    {
      "epoch": 1.4603614807128907e-05,
      "step": 15313,
      "training_step_time": 16.993286609649658
    },
    {
      "epoch": 1.4604568481445313e-05,
      "model_forward_time": 5.5809924602508545,
      "step": 15314
    },
    {
      "epoch": 1.4604568481445313e-05,
      "step": 15314,
      "training_step_time": 17.021100282669067
    },
    {
      "epoch": 1.4605522155761718e-05,
      "model_forward_time": 5.5871899127960205,
      "step": 15315
    },
    {
      "epoch": 1.4605522155761718e-05,
      "step": 15315,
      "training_step_time": 17.03411889076233
    },
    {
      "epoch": 1.4606475830078126e-05,
      "model_forward_time": 5.589584112167358,
      "step": 15316
    },
    {
      "epoch": 1.4606475830078126e-05,
      "step": 15316,
      "training_step_time": 17.04213833808899
    },
    {
      "epoch": 1.4607429504394531e-05,
      "model_forward_time": 5.597604274749756,
      "step": 15317
    },
    {
      "epoch": 1.4607429504394531e-05,
      "step": 15317,
      "training_step_time": 17.428600788116455
    },
    {
      "epoch": 1.4608383178710937e-05,
      "model_forward_time": 5.6031248569488525,
      "step": 15318
    },
    {
      "epoch": 1.4608383178710937e-05,
      "step": 15318,
      "training_step_time": 17.085481643676758
    },
    {
      "epoch": 1.4609336853027345e-05,
      "model_forward_time": 5.613996982574463,
      "step": 15319
    },
    {
      "epoch": 1.4609336853027345e-05,
      "step": 15319,
      "training_step_time": 17.097906827926636
    },
    {
      "epoch": 1.461029052734375e-05,
      "grad_norm": 0.9511690139770508,
      "learning_rate": 1e-05,
      "loss": 0.0706,
      "step": 15320
    },
    {
      "epoch": 1.461029052734375e-05,
      "model_forward_time": 5.606314420700073,
      "step": 15320
    },
    {
      "dynamics_loss_avg": 0.05654968414455652,
      "epoch": 1.461029052734375e-05,
      "step": 15320
    },
    {
      "action_loss_avg": 0.00943494790699333,
      "epoch": 1.461029052734375e-05,
      "step": 15320
    },
    {
      "epoch": 1.461029052734375e-05,
      "step": 15320,
      "training_step_time": 17.103414297103882
    },
    {
      "epoch": 1.4611244201660156e-05,
      "model_forward_time": 5.612241744995117,
      "step": 15321
    },
    {
      "epoch": 1.4611244201660156e-05,
      "step": 15321,
      "training_step_time": 17.111557722091675
    },
    {
      "epoch": 1.4612197875976562e-05,
      "model_forward_time": 5.624080419540405,
      "step": 15322
    },
    {
      "epoch": 1.4612197875976562e-05,
      "step": 15322,
      "training_step_time": 17.07937455177307
    },
    {
      "epoch": 1.4613151550292969e-05,
      "model_forward_time": 5.9695258140563965,
      "step": 15323
    },
    {
      "epoch": 1.4613151550292969e-05,
      "step": 15323,
      "training_step_time": 17.40094256401062
    },
    {
      "epoch": 1.4614105224609375e-05,
      "model_forward_time": 5.582640171051025,
      "step": 15324
    },
    {
      "epoch": 1.4614105224609375e-05,
      "step": 15324,
      "training_step_time": 17.013224601745605
    },
    {
      "epoch": 1.461505889892578e-05,
      "model_forward_time": 5.589422941207886,
      "step": 15325
    },
    {
      "epoch": 1.461505889892578e-05,
      "step": 15325,
      "training_step_time": 17.017680406570435
    },
    {
      "epoch": 1.4616012573242188e-05,
      "model_forward_time": 5.581907510757446,
      "step": 15326
    },
    {
      "epoch": 1.4616012573242188e-05,
      "step": 15326,
      "training_step_time": 17.02401089668274
    },
    {
      "epoch": 1.4616966247558594e-05,
      "model_forward_time": 5.593799114227295,
      "step": 15327
    },
    {
      "epoch": 1.4616966247558594e-05,
      "step": 15327,
      "training_step_time": 17.058273315429688
    },
    {
      "epoch": 1.4617919921875e-05,
      "model_forward_time": 5.986256837844849,
      "step": 15328
    },
    {
      "epoch": 1.4617919921875e-05,
      "step": 15328,
      "training_step_time": 17.479202032089233
    },
    {
      "epoch": 1.4618873596191407e-05,
      "model_forward_time": 5.626366138458252,
      "step": 15329
    },
    {
      "epoch": 1.4618873596191407e-05,
      "step": 15329,
      "training_step_time": 17.099711656570435
    },
    {
      "epoch": 1.4619827270507813e-05,
      "grad_norm": 1.2541028261184692,
      "learning_rate": 1e-05,
      "loss": 0.0812,
      "step": 15330
    },
    {
      "epoch": 1.4619827270507813e-05,
      "model_forward_time": 5.5896382331848145,
      "step": 15330
    },
    {
      "dynamics_loss_avg": 0.06720820106565953,
      "epoch": 1.4619827270507813e-05,
      "step": 15330
    },
    {
      "action_loss_avg": 0.009932435769587755,
      "epoch": 1.4619827270507813e-05,
      "step": 15330
    },
    {
      "epoch": 1.4619827270507813e-05,
      "step": 15330,
      "training_step_time": 17.019169330596924
    },
    {
      "epoch": 1.4620780944824218e-05,
      "model_forward_time": 5.580170392990112,
      "step": 15331
    },
    {
      "epoch": 1.4620780944824218e-05,
      "step": 15331,
      "training_step_time": 16.99518346786499
    },
    {
      "epoch": 1.4621734619140626e-05,
      "model_forward_time": 5.5622878074646,
      "step": 15332
    },
    {
      "epoch": 1.4621734619140626e-05,
      "step": 15332,
      "training_step_time": 16.967111110687256
    },
    {
      "epoch": 1.4622688293457031e-05,
      "model_forward_time": 5.563727378845215,
      "step": 15333
    },
    {
      "epoch": 1.4622688293457031e-05,
      "step": 15333,
      "training_step_time": 17.003530979156494
    },
    {
      "epoch": 1.4623641967773437e-05,
      "model_forward_time": 5.950946807861328,
      "step": 15334
    },
    {
      "epoch": 1.4623641967773437e-05,
      "step": 15334,
      "training_step_time": 17.37010669708252
    },
    {
      "epoch": 1.4624595642089845e-05,
      "model_forward_time": 5.593709468841553,
      "step": 15335
    },
    {
      "epoch": 1.4624595642089845e-05,
      "step": 15335,
      "training_step_time": 17.061918258666992
    },
    {
      "epoch": 1.462554931640625e-05,
      "model_forward_time": 5.606304168701172,
      "step": 15336
    },
    {
      "epoch": 1.462554931640625e-05,
      "step": 15336,
      "training_step_time": 17.067370891571045
    },
    {
      "epoch": 1.4626502990722656e-05,
      "model_forward_time": 5.600402116775513,
      "step": 15337
    },
    {
      "epoch": 1.4626502990722656e-05,
      "step": 15337,
      "training_step_time": 17.07967972755432
    },
    {
      "epoch": 1.4627456665039062e-05,
      "model_forward_time": 5.594882011413574,
      "step": 15338
    },
    {
      "epoch": 1.4627456665039062e-05,
      "step": 15338,
      "training_step_time": 17.049826622009277
    },
    {
      "epoch": 1.4628410339355469e-05,
      "model_forward_time": 5.600313186645508,
      "step": 15339
    },
    {
      "epoch": 1.4628410339355469e-05,
      "step": 15339,
      "training_step_time": 17.4409601688385
    },
    {
      "epoch": 1.4629364013671875e-05,
      "grad_norm": 1.3149256706237793,
      "learning_rate": 1e-05,
      "loss": 0.0682,
      "step": 15340
    },
    {
      "epoch": 1.4629364013671875e-05,
      "model_forward_time": 5.607161283493042,
      "step": 15340
    },
    {
      "dynamics_loss_avg": 0.06062564887106418,
      "epoch": 1.4629364013671875e-05,
      "step": 15340
    },
    {
      "action_loss_avg": 0.009100663056597114,
      "epoch": 1.4629364013671875e-05,
      "step": 15340
    },
    {
      "epoch": 1.4629364013671875e-05,
      "step": 15340,
      "training_step_time": 17.077224016189575
    },
    {
      "epoch": 1.463031768798828e-05,
      "model_forward_time": 5.606264591217041,
      "step": 15341
    },
    {
      "epoch": 1.463031768798828e-05,
      "step": 15341,
      "training_step_time": 17.063089847564697
    },
    {
      "epoch": 1.4631271362304688e-05,
      "model_forward_time": 5.601093769073486,
      "step": 15342
    },
    {
      "epoch": 1.4631271362304688e-05,
      "step": 15342,
      "training_step_time": 17.069985389709473
    },
    {
      "epoch": 1.4632225036621094e-05,
      "model_forward_time": 5.604394435882568,
      "step": 15343
    },
    {
      "epoch": 1.4632225036621094e-05,
      "step": 15343,
      "training_step_time": 17.08281683921814
    },
    {
      "epoch": 1.46331787109375e-05,
      "model_forward_time": 5.603567838668823,
      "step": 15344
    },
    {
      "epoch": 1.46331787109375e-05,
      "step": 15344,
      "training_step_time": 17.086390495300293
    },
    {
      "epoch": 1.4634132385253907e-05,
      "model_forward_time": 5.9886181354522705,
      "step": 15345
    },
    {
      "epoch": 1.4634132385253907e-05,
      "step": 15345,
      "training_step_time": 17.47936248779297
    },
    {
      "epoch": 1.4635086059570313e-05,
      "model_forward_time": 5.612886667251587,
      "step": 15346
    },
    {
      "epoch": 1.4635086059570313e-05,
      "step": 15346,
      "training_step_time": 17.064103364944458
    },
    {
      "epoch": 1.4636039733886718e-05,
      "model_forward_time": 5.58982253074646,
      "step": 15347
    },
    {
      "epoch": 1.4636039733886718e-05,
      "step": 15347,
      "training_step_time": 17.01453948020935
    },
    {
      "epoch": 1.4636993408203126e-05,
      "model_forward_time": 5.591928720474243,
      "step": 15348
    },
    {
      "epoch": 1.4636993408203126e-05,
      "step": 15348,
      "training_step_time": 17.017801761627197
    },
    {
      "epoch": 1.4637947082519531e-05,
      "model_forward_time": 5.595369100570679,
      "step": 15349
    },
    {
      "epoch": 1.4637947082519531e-05,
      "step": 15349,
      "training_step_time": 17.043847799301147
    },
    {
      "epoch": 1.4638900756835937e-05,
      "grad_norm": 0.9320590496063232,
      "learning_rate": 1e-05,
      "loss": 0.0686,
      "step": 15350
    },
    {
      "epoch": 1.4638900756835937e-05,
      "model_forward_time": 5.950870037078857,
      "step": 15350
    },
    {
      "dynamics_loss_avg": 0.062008231319487093,
      "epoch": 1.4638900756835937e-05,
      "step": 15350
    },
    {
      "action_loss_avg": 0.00790092897368595,
      "epoch": 1.4638900756835937e-05,
      "step": 15350
    },
    {
      "epoch": 1.4638900756835937e-05,
      "step": 15350,
      "training_step_time": 17.395689487457275
    },
    {
      "epoch": 1.4639854431152345e-05,
      "model_forward_time": 5.598517656326294,
      "step": 15351
    },
    {
      "epoch": 1.4639854431152345e-05,
      "step": 15351,
      "training_step_time": 17.051294088363647
    },
    {
      "epoch": 1.464080810546875e-05,
      "model_forward_time": 5.600409507751465,
      "step": 15352
    },
    {
      "epoch": 1.464080810546875e-05,
      "step": 15352,
      "training_step_time": 17.0848548412323
    },
    {
      "epoch": 1.4641761779785156e-05,
      "model_forward_time": 5.6216044425964355,
      "step": 15353
    },
    {
      "epoch": 1.4641761779785156e-05,
      "step": 15353,
      "training_step_time": 17.13303279876709
    },
    {
      "epoch": 1.4642715454101562e-05,
      "model_forward_time": 5.620466709136963,
      "step": 15354
    },
    {
      "epoch": 1.4642715454101562e-05,
      "step": 15354,
      "training_step_time": 17.10874104499817
    },
    {
      "epoch": 1.464366912841797e-05,
      "model_forward_time": 5.598545551300049,
      "step": 15355
    },
    {
      "epoch": 1.464366912841797e-05,
      "step": 15355,
      "training_step_time": 17.050021171569824
    },
    {
      "epoch": 1.4644622802734375e-05,
      "model_forward_time": 5.952081203460693,
      "step": 15356
    },
    {
      "epoch": 1.4644622802734375e-05,
      "step": 15356,
      "training_step_time": 17.39323091506958
    },
    {
      "epoch": 1.464557647705078e-05,
      "model_forward_time": 5.585296392440796,
      "step": 15357
    },
    {
      "epoch": 1.464557647705078e-05,
      "step": 15357,
      "training_step_time": 17.047690391540527
    },
    {
      "epoch": 1.4646530151367188e-05,
      "model_forward_time": 5.6221604347229,
      "step": 15358
    },
    {
      "epoch": 1.4646530151367188e-05,
      "step": 15358,
      "training_step_time": 17.117979526519775
    },
    {
      "epoch": 1.4647483825683594e-05,
      "model_forward_time": 5.622380018234253,
      "step": 15359
    },
    {
      "epoch": 1.4647483825683594e-05,
      "step": 15359,
      "training_step_time": 17.11287808418274
    },
    {
      "epoch": 1.46484375e-05,
      "grad_norm": 0.4378328323364258,
      "learning_rate": 1e-05,
      "loss": 0.0465,
      "step": 15360
    },
    {
      "epoch": 1.46484375e-05,
      "model_forward_time": 5.593483924865723,
      "step": 15360
    },
    {
      "dynamics_loss_avg": 0.03954527620226145,
      "epoch": 1.46484375e-05,
      "step": 15360
    },
    {
      "action_loss_avg": 0.004202889348380268,
      "epoch": 1.46484375e-05,
      "step": 15360
    },
    {
      "epoch": 1.46484375e-05,
      "step": 15360,
      "training_step_time": 17.04401922225952
    },
    {
      "epoch": 1.4649391174316407e-05,
      "model_forward_time": 5.942461729049683,
      "step": 15361
    },
    {
      "epoch": 1.4649391174316407e-05,
      "step": 15361,
      "training_step_time": 17.370049238204956
    },
    {
      "epoch": 1.4650344848632813e-05,
      "model_forward_time": 5.5867180824279785,
      "step": 15362
    },
    {
      "epoch": 1.4650344848632813e-05,
      "step": 15362,
      "training_step_time": 16.996058225631714
    },
    {
      "epoch": 1.4651298522949218e-05,
      "model_forward_time": 5.5768654346466064,
      "step": 15363
    },
    {
      "epoch": 1.4651298522949218e-05,
      "step": 15363,
      "training_step_time": 16.995593786239624
    },
    {
      "epoch": 1.4652252197265626e-05,
      "model_forward_time": 5.585536003112793,
      "step": 15364
    },
    {
      "epoch": 1.4652252197265626e-05,
      "step": 15364,
      "training_step_time": 17.026184558868408
    },
    {
      "epoch": 1.4653205871582031e-05,
      "model_forward_time": 5.584799766540527,
      "step": 15365
    },
    {
      "epoch": 1.4653205871582031e-05,
      "step": 15365,
      "training_step_time": 17.02460765838623
    },
    {
      "epoch": 1.4654159545898437e-05,
      "model_forward_time": 5.5938825607299805,
      "step": 15366
    },
    {
      "epoch": 1.4654159545898437e-05,
      "step": 15366,
      "training_step_time": 17.394744873046875
    },
    {
      "epoch": 1.4655113220214845e-05,
      "model_forward_time": 5.592249870300293,
      "step": 15367
    },
    {
      "epoch": 1.4655113220214845e-05,
      "step": 15367,
      "training_step_time": 17.037797927856445
    },
    {
      "epoch": 1.465606689453125e-05,
      "model_forward_time": 5.5903847217559814,
      "step": 15368
    },
    {
      "epoch": 1.465606689453125e-05,
      "step": 15368,
      "training_step_time": 17.04782271385193
    },
    {
      "epoch": 1.4657020568847656e-05,
      "model_forward_time": 5.59642767906189,
      "step": 15369
    },
    {
      "epoch": 1.4657020568847656e-05,
      "step": 15369,
      "training_step_time": 17.0618634223938
    },
    {
      "epoch": 1.4657974243164062e-05,
      "grad_norm": 0.790825366973877,
      "learning_rate": 1e-05,
      "loss": 0.0625,
      "step": 15370
    },
    {
      "epoch": 1.4657974243164062e-05,
      "model_forward_time": 5.604176044464111,
      "step": 15370
    },
    {
      "dynamics_loss_avg": 0.05453127464279532,
      "epoch": 1.4657974243164062e-05,
      "step": 15370
    },
    {
      "action_loss_avg": 0.007448611198924482,
      "epoch": 1.4657974243164062e-05,
      "step": 15370
    },
    {
      "epoch": 1.4657974243164062e-05,
      "step": 15370,
      "training_step_time": 17.081916093826294
    },
    {
      "epoch": 1.465892791748047e-05,
      "model_forward_time": 5.615794658660889,
      "step": 15371
    },
    {
      "epoch": 1.465892791748047e-05,
      "step": 15371,
      "training_step_time": 17.076855421066284
    },
    {
      "epoch": 1.4659881591796875e-05,
      "model_forward_time": 5.963382720947266,
      "step": 15372
    },
    {
      "epoch": 1.4659881591796875e-05,
      "step": 15372,
      "training_step_time": 17.469399452209473
    },
    {
      "epoch": 1.466083526611328e-05,
      "model_forward_time": 5.611062526702881,
      "step": 15373
    },
    {
      "epoch": 1.466083526611328e-05,
      "step": 15373,
      "training_step_time": 17.084556102752686
    },
    {
      "epoch": 1.4661788940429688e-05,
      "model_forward_time": 5.600529193878174,
      "step": 15374
    },
    {
      "epoch": 1.4661788940429688e-05,
      "step": 15374,
      "training_step_time": 17.059950828552246
    },
    {
      "epoch": 1.4662742614746094e-05,
      "model_forward_time": 5.6008875370025635,
      "step": 15375
    },
    {
      "epoch": 1.4662742614746094e-05,
      "step": 15375,
      "training_step_time": 17.064774990081787
    },
    {
      "epoch": 1.46636962890625e-05,
      "model_forward_time": 5.601355314254761,
      "step": 15376
    },
    {
      "epoch": 1.46636962890625e-05,
      "step": 15376,
      "training_step_time": 17.07353401184082
    },
    {
      "epoch": 1.4664649963378907e-05,
      "model_forward_time": 5.9885032176971436,
      "step": 15377
    },
    {
      "epoch": 1.4664649963378907e-05,
      "step": 15377,
      "training_step_time": 17.45076823234558
    },
    {
      "epoch": 1.4665603637695313e-05,
      "model_forward_time": 5.600796699523926,
      "step": 15378
    },
    {
      "epoch": 1.4665603637695313e-05,
      "step": 15378,
      "training_step_time": 17.089078187942505
    },
    {
      "epoch": 1.4666557312011718e-05,
      "model_forward_time": 5.6139655113220215,
      "step": 15379
    },
    {
      "epoch": 1.4666557312011718e-05,
      "step": 15379,
      "training_step_time": 17.123571634292603
    },
    {
      "epoch": 1.4667510986328126e-05,
      "grad_norm": 1.4282830953598022,
      "learning_rate": 1e-05,
      "loss": 0.0628,
      "step": 15380
    },
    {
      "epoch": 1.4667510986328126e-05,
      "model_forward_time": 5.624736785888672,
      "step": 15380
    },
    {
      "dynamics_loss_avg": 0.0528687996789813,
      "epoch": 1.4667510986328126e-05,
      "step": 15380
    },
    {
      "action_loss_avg": 0.007820913125760854,
      "epoch": 1.4667510986328126e-05,
      "step": 15380
    },
    {
      "epoch": 1.4667510986328126e-05,
      "step": 15380,
      "training_step_time": 17.14937973022461
    },
    {
      "epoch": 1.4668464660644531e-05,
      "model_forward_time": 5.627772331237793,
      "step": 15381
    },
    {
      "epoch": 1.4668464660644531e-05,
      "step": 15381,
      "training_step_time": 17.114537715911865
    },
    {
      "epoch": 1.4669418334960937e-05,
      "model_forward_time": 5.597545146942139,
      "step": 15382
    },
    {
      "epoch": 1.4669418334960937e-05,
      "step": 15382,
      "training_step_time": 17.052566289901733
    },
    {
      "epoch": 1.4670372009277343e-05,
      "model_forward_time": 5.9393229484558105,
      "step": 15383
    },
    {
      "epoch": 1.4670372009277343e-05,
      "step": 15383,
      "training_step_time": 17.3841335773468
    },
    {
      "epoch": 1.467132568359375e-05,
      "model_forward_time": 5.5887439250946045,
      "step": 15384
    },
    {
      "epoch": 1.467132568359375e-05,
      "step": 15384,
      "training_step_time": 17.042356967926025
    },
    {
      "epoch": 1.4672279357910156e-05,
      "model_forward_time": 5.608155965805054,
      "step": 15385
    },
    {
      "epoch": 1.4672279357910156e-05,
      "step": 15385,
      "training_step_time": 17.09104871749878
    },
    {
      "epoch": 1.4673233032226562e-05,
      "model_forward_time": 5.607868671417236,
      "step": 15386
    },
    {
      "epoch": 1.4673233032226562e-05,
      "step": 15386,
      "training_step_time": 17.095494508743286
    },
    {
      "epoch": 1.467418670654297e-05,
      "model_forward_time": 5.611920595169067,
      "step": 15387
    },
    {
      "epoch": 1.467418670654297e-05,
      "step": 15387,
      "training_step_time": 17.117248058319092
    },
    {
      "epoch": 1.4675140380859375e-05,
      "model_forward_time": 5.984339952468872,
      "step": 15388
    },
    {
      "epoch": 1.4675140380859375e-05,
      "step": 15388,
      "training_step_time": 17.468581199645996
    },
    {
      "epoch": 1.467609405517578e-05,
      "model_forward_time": 5.603209495544434,
      "step": 15389
    },
    {
      "epoch": 1.467609405517578e-05,
      "step": 15389,
      "training_step_time": 17.07045316696167
    },
    {
      "epoch": 1.4677047729492188e-05,
      "grad_norm": 0.43925046920776367,
      "learning_rate": 1e-05,
      "loss": 0.0636,
      "step": 15390
    },
    {
      "epoch": 1.4677047729492188e-05,
      "model_forward_time": 5.596454381942749,
      "step": 15390
    },
    {
      "dynamics_loss_avg": 0.06136165540665388,
      "epoch": 1.4677047729492188e-05,
      "step": 15390
    },
    {
      "action_loss_avg": 0.008615688886493444,
      "epoch": 1.4677047729492188e-05,
      "step": 15390
    },
    {
      "epoch": 1.4677047729492188e-05,
      "step": 15390,
      "training_step_time": 17.033252477645874
    },
    {
      "epoch": 1.4678001403808594e-05,
      "model_forward_time": 5.591009855270386,
      "step": 15391
    },
    {
      "epoch": 1.4678001403808594e-05,
      "step": 15391,
      "training_step_time": 17.02564525604248
    },
    {
      "epoch": 1.4678955078125e-05,
      "model_forward_time": 5.602634429931641,
      "step": 15392
    },
    {
      "epoch": 1.4678955078125e-05,
      "step": 15392,
      "training_step_time": 17.05807638168335
    },
    {
      "epoch": 1.4679908752441407e-05,
      "model_forward_time": 5.597057342529297,
      "step": 15393
    },
    {
      "epoch": 1.4679908752441407e-05,
      "step": 15393,
      "training_step_time": 17.091357707977295
    },
    {
      "epoch": 1.4680862426757813e-05,
      "model_forward_time": 5.98168683052063,
      "step": 15394
    },
    {
      "epoch": 1.4680862426757813e-05,
      "step": 15394,
      "training_step_time": 17.493948221206665
    },
    {
      "epoch": 1.4681816101074218e-05,
      "model_forward_time": 5.6100075244903564,
      "step": 15395
    },
    {
      "epoch": 1.4681816101074218e-05,
      "step": 15395,
      "training_step_time": 17.072641611099243
    },
    {
      "epoch": 1.4682769775390626e-05,
      "model_forward_time": 5.589047431945801,
      "step": 15396
    },
    {
      "epoch": 1.4682769775390626e-05,
      "step": 15396,
      "training_step_time": 17.02423858642578
    },
    {
      "epoch": 1.4683723449707031e-05,
      "model_forward_time": 5.582867860794067,
      "step": 15397
    },
    {
      "epoch": 1.4683723449707031e-05,
      "step": 15397,
      "training_step_time": 17.021320819854736
    },
    {
      "epoch": 1.4684677124023437e-05,
      "model_forward_time": 5.592007160186768,
      "step": 15398
    },
    {
      "epoch": 1.4684677124023437e-05,
      "step": 15398,
      "training_step_time": 17.0661563873291
    },
    {
      "epoch": 1.4685630798339843e-05,
      "model_forward_time": 5.633408546447754,
      "step": 15399
    },
    {
      "epoch": 1.4685630798339843e-05,
      "step": 15399,
      "training_step_time": 17.527443408966064
    },
    {
      "epoch": 1.468658447265625e-05,
      "grad_norm": 0.6758062839508057,
      "learning_rate": 1e-05,
      "loss": 0.0548,
      "step": 15400
    },
    {
      "epoch": 1.468658447265625e-05,
      "model_forward_time": 5.624252080917358,
      "step": 15400
    },
    {
      "dynamics_loss_avg": 0.04481961242854595,
      "epoch": 1.468658447265625e-05,
      "step": 15400
    },
    {
      "action_loss_avg": 0.006850264233071357,
      "epoch": 1.468658447265625e-05,
      "step": 15400
    },
    {
      "epoch": 1.468658447265625e-05,
      "step": 15400,
      "training_step_time": 17.130106687545776
    },
    {
      "epoch": 1.4687538146972656e-05,
      "model_forward_time": 5.615135669708252,
      "step": 15401
    },
    {
      "epoch": 1.4687538146972656e-05,
      "step": 15401,
      "training_step_time": 17.10368537902832
    },
    {
      "epoch": 1.4688491821289062e-05,
      "model_forward_time": 5.610396385192871,
      "step": 15402
    },
    {
      "epoch": 1.4688491821289062e-05,
      "step": 15402,
      "training_step_time": 17.07291889190674
    },
    {
      "epoch": 1.468944549560547e-05,
      "model_forward_time": 5.605538368225098,
      "step": 15403
    },
    {
      "epoch": 1.468944549560547e-05,
      "step": 15403,
      "training_step_time": 17.06755542755127
    },
    {
      "epoch": 1.4690399169921875e-05,
      "model_forward_time": 5.59779167175293,
      "step": 15404
    },
    {
      "epoch": 1.4690399169921875e-05,
      "step": 15404,
      "training_step_time": 17.057311534881592
    },
    {
      "epoch": 1.469135284423828e-05,
      "model_forward_time": 5.966164827346802,
      "step": 15405
    },
    {
      "epoch": 1.469135284423828e-05,
      "step": 15405,
      "training_step_time": 17.407788515090942
    },
    {
      "epoch": 1.4692306518554688e-05,
      "model_forward_time": 5.597508430480957,
      "step": 15406
    },
    {
      "epoch": 1.4692306518554688e-05,
      "step": 15406,
      "training_step_time": 17.045138359069824
    },
    {
      "epoch": 1.4693260192871094e-05,
      "model_forward_time": 5.597177743911743,
      "step": 15407
    },
    {
      "epoch": 1.4693260192871094e-05,
      "step": 15407,
      "training_step_time": 17.052634239196777
    },
    {
      "epoch": 1.46942138671875e-05,
      "model_forward_time": 5.596771717071533,
      "step": 15408
    },
    {
      "epoch": 1.46942138671875e-05,
      "step": 15408,
      "training_step_time": 17.062158823013306
    },
    {
      "epoch": 1.4695167541503907e-05,
      "model_forward_time": 5.6024134159088135,
      "step": 15409
    },
    {
      "epoch": 1.4695167541503907e-05,
      "step": 15409,
      "training_step_time": 17.0568904876709
    },
    {
      "epoch": 1.4696121215820313e-05,
      "grad_norm": 0.4048219323158264,
      "learning_rate": 1e-05,
      "loss": 0.0551,
      "step": 15410
    },
    {
      "epoch": 1.4696121215820313e-05,
      "model_forward_time": 5.9797163009643555,
      "step": 15410
    },
    {
      "dynamics_loss_avg": 0.0495373573154211,
      "epoch": 1.4696121215820313e-05,
      "step": 15410
    },
    {
      "action_loss_avg": 0.005737158795818686,
      "epoch": 1.4696121215820313e-05,
      "step": 15410
    },
    {
      "epoch": 1.4696121215820313e-05,
      "step": 15410,
      "training_step_time": 17.423259496688843
    },
    {
      "epoch": 1.4697074890136718e-05,
      "model_forward_time": 5.588814973831177,
      "step": 15411
    },
    {
      "epoch": 1.4697074890136718e-05,
      "step": 15411,
      "training_step_time": 17.01530885696411
    },
    {
      "epoch": 1.4698028564453126e-05,
      "model_forward_time": 5.593147277832031,
      "step": 15412
    },
    {
      "epoch": 1.4698028564453126e-05,
      "step": 15412,
      "training_step_time": 17.03918957710266
    },
    {
      "epoch": 1.4698982238769531e-05,
      "model_forward_time": 5.6023852825164795,
      "step": 15413
    },
    {
      "epoch": 1.4698982238769531e-05,
      "step": 15413,
      "training_step_time": 17.09308624267578
    },
    {
      "epoch": 1.4699935913085937e-05,
      "model_forward_time": 5.618566274642944,
      "step": 15414
    },
    {
      "epoch": 1.4699935913085937e-05,
      "step": 15414,
      "training_step_time": 17.10696244239807
    },
    {
      "epoch": 1.4700889587402343e-05,
      "model_forward_time": 5.616378545761108,
      "step": 15415
    },
    {
      "epoch": 1.4700889587402343e-05,
      "step": 15415,
      "training_step_time": 17.110193490982056
    },
    {
      "epoch": 1.470184326171875e-05,
      "model_forward_time": 5.9653589725494385,
      "step": 15416
    },
    {
      "epoch": 1.470184326171875e-05,
      "step": 15416,
      "training_step_time": 17.464780569076538
    },
    {
      "epoch": 1.4702796936035156e-05,
      "model_forward_time": 5.6152660846710205,
      "step": 15417
    },
    {
      "epoch": 1.4702796936035156e-05,
      "step": 15417,
      "training_step_time": 17.099341869354248
    },
    {
      "epoch": 1.4703750610351562e-05,
      "model_forward_time": 5.615393161773682,
      "step": 15418
    },
    {
      "epoch": 1.4703750610351562e-05,
      "step": 15418,
      "training_step_time": 17.106456518173218
    },
    {
      "epoch": 1.470470428466797e-05,
      "model_forward_time": 5.611976385116577,
      "step": 15419
    },
    {
      "epoch": 1.470470428466797e-05,
      "step": 15419,
      "training_step_time": 17.102837562561035
    },
    {
      "epoch": 1.4705657958984375e-05,
      "grad_norm": 0.3798360824584961,
      "learning_rate": 1e-05,
      "loss": 0.0573,
      "step": 15420
    },
    {
      "epoch": 1.4705657958984375e-05,
      "model_forward_time": 5.605682373046875,
      "step": 15420
    },
    {
      "dynamics_loss_avg": 0.05092835333198309,
      "epoch": 1.4705657958984375e-05,
      "step": 15420
    },
    {
      "action_loss_avg": 0.005623131990432739,
      "epoch": 1.4705657958984375e-05,
      "step": 15420
    },
    {
      "epoch": 1.4705657958984375e-05,
      "step": 15420,
      "training_step_time": 17.094487190246582
    },
    {
      "epoch": 1.470661163330078e-05,
      "model_forward_time": 5.9935526847839355,
      "step": 15421
    },
    {
      "epoch": 1.470661163330078e-05,
      "step": 15421,
      "training_step_time": 17.469412088394165
    },
    {
      "epoch": 1.4707565307617188e-05,
      "model_forward_time": 5.599402666091919,
      "step": 15422
    },
    {
      "epoch": 1.4707565307617188e-05,
      "step": 15422,
      "training_step_time": 17.070175170898438
    },
    {
      "epoch": 1.4708518981933594e-05,
      "model_forward_time": 5.600792646408081,
      "step": 15423
    },
    {
      "epoch": 1.4708518981933594e-05,
      "step": 15423,
      "training_step_time": 17.06519055366516
    },
    {
      "epoch": 1.470947265625e-05,
      "model_forward_time": 5.6011412143707275,
      "step": 15424
    },
    {
      "epoch": 1.470947265625e-05,
      "step": 15424,
      "training_step_time": 17.033272743225098
    },
    {
      "epoch": 1.4710426330566407e-05,
      "model_forward_time": 5.595421075820923,
      "step": 15425
    },
    {
      "epoch": 1.4710426330566407e-05,
      "step": 15425,
      "training_step_time": 17.04440212249756
    },
    {
      "epoch": 1.4711380004882813e-05,
      "model_forward_time": 5.606500625610352,
      "step": 15426
    },
    {
      "epoch": 1.4711380004882813e-05,
      "step": 15426,
      "training_step_time": 17.10263681411743
    },
    {
      "epoch": 1.4712333679199218e-05,
      "model_forward_time": 5.969375133514404,
      "step": 15427
    },
    {
      "epoch": 1.4712333679199218e-05,
      "step": 15427,
      "training_step_time": 17.485698223114014
    },
    {
      "epoch": 1.4713287353515626e-05,
      "model_forward_time": 5.631300449371338,
      "step": 15428
    },
    {
      "epoch": 1.4713287353515626e-05,
      "step": 15428,
      "training_step_time": 17.12768268585205
    },
    {
      "epoch": 1.4714241027832032e-05,
      "model_forward_time": 5.622913360595703,
      "step": 15429
    },
    {
      "epoch": 1.4714241027832032e-05,
      "step": 15429,
      "training_step_time": 17.08243703842163
    },
    {
      "epoch": 1.4715194702148437e-05,
      "grad_norm": 0.8477534055709839,
      "learning_rate": 1e-05,
      "loss": 0.0684,
      "step": 15430
    },
    {
      "epoch": 1.4715194702148437e-05,
      "model_forward_time": 5.583364009857178,
      "step": 15430
    },
    {
      "dynamics_loss_avg": 0.06526880748569966,
      "epoch": 1.4715194702148437e-05,
      "step": 15430
    },
    {
      "action_loss_avg": 0.010192987928166985,
      "epoch": 1.4715194702148437e-05,
      "step": 15430
    },
    {
      "epoch": 1.4715194702148437e-05,
      "step": 15430,
      "training_step_time": 16.989075422286987
    },
    {
      "epoch": 1.4716148376464843e-05,
      "model_forward_time": 5.586735486984253,
      "step": 15431
    },
    {
      "epoch": 1.4716148376464843e-05,
      "step": 15431,
      "training_step_time": 16.99676251411438
    },
    {
      "epoch": 1.471710205078125e-05,
      "model_forward_time": 5.578215837478638,
      "step": 15432
    },
    {
      "epoch": 1.471710205078125e-05,
      "step": 15432,
      "training_step_time": 17.3722243309021
    },
    {
      "epoch": 1.4718055725097656e-05,
      "model_forward_time": 5.568310737609863,
      "step": 15433
    },
    {
      "epoch": 1.4718055725097656e-05,
      "step": 15433,
      "training_step_time": 16.99589967727661
    },
    {
      "epoch": 1.4719009399414062e-05,
      "model_forward_time": 5.587465763092041,
      "step": 15434
    },
    {
      "epoch": 1.4719009399414062e-05,
      "step": 15434,
      "training_step_time": 16.99435305595398
    },
    {
      "epoch": 1.471996307373047e-05,
      "model_forward_time": 5.583505630493164,
      "step": 15435
    },
    {
      "epoch": 1.471996307373047e-05,
      "step": 15435,
      "training_step_time": 17.007020950317383
    },
    {
      "epoch": 1.4720916748046875e-05,
      "model_forward_time": 5.577748775482178,
      "step": 15436
    },
    {
      "epoch": 1.4720916748046875e-05,
      "step": 15436,
      "training_step_time": 17.02303957939148
    },
    {
      "epoch": 1.472187042236328e-05,
      "model_forward_time": 5.574553728103638,
      "step": 15437
    },
    {
      "epoch": 1.472187042236328e-05,
      "step": 15437,
      "training_step_time": 17.00481152534485
    },
    {
      "epoch": 1.4722824096679688e-05,
      "model_forward_time": 5.960876226425171,
      "step": 15438
    },
    {
      "epoch": 1.4722824096679688e-05,
      "step": 15438,
      "training_step_time": 17.39784812927246
    },
    {
      "epoch": 1.4723777770996094e-05,
      "model_forward_time": 5.589133977890015,
      "step": 15439
    },
    {
      "epoch": 1.4723777770996094e-05,
      "step": 15439,
      "training_step_time": 17.0718412399292
    },
    {
      "epoch": 1.47247314453125e-05,
      "grad_norm": 0.9160251617431641,
      "learning_rate": 1e-05,
      "loss": 0.0697,
      "step": 15440
    },
    {
      "epoch": 1.47247314453125e-05,
      "model_forward_time": 5.614742279052734,
      "step": 15440
    },
    {
      "dynamics_loss_avg": 0.059073792025446895,
      "epoch": 1.47247314453125e-05,
      "step": 15440
    },
    {
      "action_loss_avg": 0.00609813800547272,
      "epoch": 1.47247314453125e-05,
      "step": 15440
    },
    {
      "epoch": 1.47247314453125e-05,
      "step": 15440,
      "training_step_time": 17.122877836227417
    },
    {
      "epoch": 1.4725685119628907e-05,
      "model_forward_time": 5.62161660194397,
      "step": 15441
    },
    {
      "epoch": 1.4725685119628907e-05,
      "step": 15441,
      "training_step_time": 17.101751804351807
    },
    {
      "epoch": 1.4726638793945313e-05,
      "model_forward_time": 5.593519926071167,
      "step": 15442
    },
    {
      "epoch": 1.4726638793945313e-05,
      "step": 15442,
      "training_step_time": 17.037367343902588
    },
    {
      "epoch": 1.4727592468261718e-05,
      "model_forward_time": 5.97606897354126,
      "step": 15443
    },
    {
      "epoch": 1.4727592468261718e-05,
      "step": 15443,
      "training_step_time": 17.410534381866455
    },
    {
      "epoch": 1.4728546142578126e-05,
      "model_forward_time": 5.57937216758728,
      "step": 15444
    },
    {
      "epoch": 1.4728546142578126e-05,
      "step": 15444,
      "training_step_time": 17.00510048866272
    },
    {
      "epoch": 1.4729499816894532e-05,
      "model_forward_time": 5.592520475387573,
      "step": 15445
    },
    {
      "epoch": 1.4729499816894532e-05,
      "step": 15445,
      "training_step_time": 17.048574924468994
    },
    {
      "epoch": 1.4730453491210937e-05,
      "model_forward_time": 5.606989145278931,
      "step": 15446
    },
    {
      "epoch": 1.4730453491210937e-05,
      "step": 15446,
      "training_step_time": 17.090542554855347
    },
    {
      "epoch": 1.4731407165527343e-05,
      "model_forward_time": 5.614779949188232,
      "step": 15447
    },
    {
      "epoch": 1.4731407165527343e-05,
      "step": 15447,
      "training_step_time": 17.12684917449951
    },
    {
      "epoch": 1.473236083984375e-05,
      "model_forward_time": 5.623610496520996,
      "step": 15448
    },
    {
      "epoch": 1.473236083984375e-05,
      "step": 15448,
      "training_step_time": 17.144190549850464
    },
    {
      "epoch": 1.4733314514160156e-05,
      "model_forward_time": 5.966946363449097,
      "step": 15449
    },
    {
      "epoch": 1.4733314514160156e-05,
      "step": 15449,
      "training_step_time": 17.48364496231079
    },
    {
      "epoch": 1.4734268188476562e-05,
      "grad_norm": 0.9451330304145813,
      "learning_rate": 1e-05,
      "loss": 0.073,
      "step": 15450
    },
    {
      "epoch": 1.4734268188476562e-05,
      "model_forward_time": 5.623533010482788,
      "step": 15450
    },
    {
      "dynamics_loss_avg": 0.06253068931400776,
      "epoch": 1.4734268188476562e-05,
      "step": 15450
    },
    {
      "action_loss_avg": 0.009523493191227316,
      "epoch": 1.4734268188476562e-05,
      "step": 15450
    },
    {
      "epoch": 1.4734268188476562e-05,
      "step": 15450,
      "training_step_time": 17.136019945144653
    },
    {
      "epoch": 1.473522186279297e-05,
      "model_forward_time": 5.612445116043091,
      "step": 15451
    },
    {
      "epoch": 1.473522186279297e-05,
      "step": 15451,
      "training_step_time": 17.07908320426941
    },
    {
      "epoch": 1.4736175537109375e-05,
      "model_forward_time": 5.59527850151062,
      "step": 15452
    },
    {
      "epoch": 1.4736175537109375e-05,
      "step": 15452,
      "training_step_time": 17.036664962768555
    },
    {
      "epoch": 1.473712921142578e-05,
      "model_forward_time": 5.573538780212402,
      "step": 15453
    },
    {
      "epoch": 1.473712921142578e-05,
      "step": 15453,
      "training_step_time": 16.98199987411499
    },
    {
      "epoch": 1.4738082885742188e-05,
      "model_forward_time": 5.957117795944214,
      "step": 15454
    },
    {
      "epoch": 1.4738082885742188e-05,
      "step": 15454,
      "training_step_time": 17.36465573310852
    },
    {
      "epoch": 1.4739036560058594e-05,
      "model_forward_time": 5.576488018035889,
      "step": 15455
    },
    {
      "epoch": 1.4739036560058594e-05,
      "step": 15455,
      "training_step_time": 17.01529574394226
    },
    {
      "epoch": 1.4739990234375e-05,
      "model_forward_time": 5.583766460418701,
      "step": 15456
    },
    {
      "epoch": 1.4739990234375e-05,
      "step": 15456,
      "training_step_time": 16.988383293151855
    },
    {
      "epoch": 1.4740943908691407e-05,
      "model_forward_time": 5.57862114906311,
      "step": 15457
    },
    {
      "epoch": 1.4740943908691407e-05,
      "step": 15457,
      "training_step_time": 17.017991065979004
    },
    {
      "epoch": 1.4741897583007813e-05,
      "model_forward_time": 5.5842530727386475,
      "step": 15458
    },
    {
      "epoch": 1.4741897583007813e-05,
      "step": 15458,
      "training_step_time": 17.0267493724823
    },
    {
      "epoch": 1.4742851257324218e-05,
      "model_forward_time": 5.5843346118927,
      "step": 15459
    },
    {
      "epoch": 1.4742851257324218e-05,
      "step": 15459,
      "training_step_time": 17.017520427703857
    },
    {
      "epoch": 1.4743804931640626e-05,
      "grad_norm": 1.5143392086029053,
      "learning_rate": 1e-05,
      "loss": 0.0588,
      "step": 15460
    },
    {
      "epoch": 1.4743804931640626e-05,
      "model_forward_time": 5.950561285018921,
      "step": 15460
    },
    {
      "dynamics_loss_avg": 0.04852876290678978,
      "epoch": 1.4743804931640626e-05,
      "step": 15460
    },
    {
      "action_loss_avg": 0.006933084828779102,
      "epoch": 1.4743804931640626e-05,
      "step": 15460
    },
    {
      "epoch": 1.4743804931640626e-05,
      "step": 15460,
      "training_step_time": 17.436652183532715
    },
    {
      "epoch": 1.4744758605957032e-05,
      "model_forward_time": 5.601765155792236,
      "step": 15461
    },
    {
      "epoch": 1.4744758605957032e-05,
      "step": 15461,
      "training_step_time": 17.099926471710205
    },
    {
      "epoch": 1.4745712280273437e-05,
      "model_forward_time": 5.607959985733032,
      "step": 15462
    },
    {
      "epoch": 1.4745712280273437e-05,
      "step": 15462,
      "training_step_time": 17.058143615722656
    },
    {
      "epoch": 1.4746665954589843e-05,
      "model_forward_time": 5.616967439651489,
      "step": 15463
    },
    {
      "epoch": 1.4746665954589843e-05,
      "step": 15463,
      "training_step_time": 17.10308289527893
    },
    {
      "epoch": 1.474761962890625e-05,
      "model_forward_time": 5.600645542144775,
      "step": 15464
    },
    {
      "epoch": 1.474761962890625e-05,
      "step": 15464,
      "training_step_time": 17.07278823852539
    },
    {
      "epoch": 1.4748573303222656e-05,
      "model_forward_time": 5.600764989852905,
      "step": 15465
    },
    {
      "epoch": 1.4748573303222656e-05,
      "step": 15465,
      "training_step_time": 17.441213607788086
    },
    {
      "epoch": 1.4749526977539062e-05,
      "model_forward_time": 5.592755317687988,
      "step": 15466
    },
    {
      "epoch": 1.4749526977539062e-05,
      "step": 15466,
      "training_step_time": 17.041982650756836
    },
    {
      "epoch": 1.475048065185547e-05,
      "model_forward_time": 5.605429410934448,
      "step": 15467
    },
    {
      "epoch": 1.475048065185547e-05,
      "step": 15467,
      "training_step_time": 17.09547996520996
    },
    {
      "epoch": 1.4751434326171875e-05,
      "model_forward_time": 5.6147847175598145,
      "step": 15468
    },
    {
      "epoch": 1.4751434326171875e-05,
      "step": 15468,
      "training_step_time": 17.09731650352478
    },
    {
      "epoch": 1.475238800048828e-05,
      "model_forward_time": 5.604870319366455,
      "step": 15469
    },
    {
      "epoch": 1.475238800048828e-05,
      "step": 15469,
      "training_step_time": 17.080094575881958
    },
    {
      "epoch": 1.4753341674804688e-05,
      "grad_norm": 0.8248659372329712,
      "learning_rate": 1e-05,
      "loss": 0.0692,
      "step": 15470
    },
    {
      "epoch": 1.4753341674804688e-05,
      "model_forward_time": 5.616971015930176,
      "step": 15470
    },
    {
      "dynamics_loss_avg": 0.06305187698453665,
      "epoch": 1.4753341674804688e-05,
      "step": 15470
    },
    {
      "action_loss_avg": 0.008587060030549765,
      "epoch": 1.4753341674804688e-05,
      "step": 15470
    },
    {
      "epoch": 1.4753341674804688e-05,
      "step": 15470,
      "training_step_time": 17.08962655067444
    },
    {
      "epoch": 1.4754295349121094e-05,
      "model_forward_time": 6.126691818237305,
      "step": 15471
    },
    {
      "epoch": 1.4754295349121094e-05,
      "step": 15471,
      "training_step_time": 17.609111070632935
    },
    {
      "epoch": 1.47552490234375e-05,
      "model_forward_time": 5.609174728393555,
      "step": 15472
    },
    {
      "epoch": 1.47552490234375e-05,
      "step": 15472,
      "training_step_time": 17.07056951522827
    },
    {
      "epoch": 1.4756202697753907e-05,
      "model_forward_time": 5.605376243591309,
      "step": 15473
    },
    {
      "epoch": 1.4756202697753907e-05,
      "step": 15473,
      "training_step_time": 17.083898544311523
    },
    {
      "epoch": 1.4757156372070313e-05,
      "model_forward_time": 5.6046366691589355,
      "step": 15474
    },
    {
      "epoch": 1.4757156372070313e-05,
      "step": 15474,
      "training_step_time": 17.04690909385681
    },
    {
      "epoch": 1.4758110046386718e-05,
      "model_forward_time": 5.602650880813599,
      "step": 15475
    },
    {
      "epoch": 1.4758110046386718e-05,
      "step": 15475,
      "training_step_time": 17.0619637966156
    },
    {
      "epoch": 1.4759063720703126e-05,
      "model_forward_time": 6.046109676361084,
      "step": 15476
    },
    {
      "epoch": 1.4759063720703126e-05,
      "step": 15476,
      "training_step_time": 17.505032062530518
    },
    {
      "epoch": 1.4760017395019532e-05,
      "model_forward_time": 5.600704193115234,
      "step": 15477
    },
    {
      "epoch": 1.4760017395019532e-05,
      "step": 15477,
      "training_step_time": 17.078758001327515
    },
    {
      "epoch": 1.4760971069335937e-05,
      "model_forward_time": 5.614678382873535,
      "step": 15478
    },
    {
      "epoch": 1.4760971069335937e-05,
      "step": 15478,
      "training_step_time": 17.11781668663025
    },
    {
      "epoch": 1.4761924743652343e-05,
      "model_forward_time": 5.624826908111572,
      "step": 15479
    },
    {
      "epoch": 1.4761924743652343e-05,
      "step": 15479,
      "training_step_time": 17.130789756774902
    },
    {
      "epoch": 1.476287841796875e-05,
      "grad_norm": 1.549177885055542,
      "learning_rate": 1e-05,
      "loss": 0.0593,
      "step": 15480
    },
    {
      "epoch": 1.476287841796875e-05,
      "model_forward_time": 5.626622915267944,
      "step": 15480
    },
    {
      "dynamics_loss_avg": 0.06247395593672991,
      "epoch": 1.476287841796875e-05,
      "step": 15480
    },
    {
      "action_loss_avg": 0.0074636000324971976,
      "epoch": 1.476287841796875e-05,
      "step": 15480
    },
    {
      "epoch": 1.476287841796875e-05,
      "step": 15480,
      "training_step_time": 17.1454918384552
    },
    {
      "epoch": 1.4763832092285156e-05,
      "model_forward_time": 5.622703790664673,
      "step": 15481
    },
    {
      "epoch": 1.4763832092285156e-05,
      "step": 15481,
      "training_step_time": 17.129143714904785
    },
    {
      "epoch": 1.4764785766601562e-05,
      "model_forward_time": 5.969371795654297,
      "step": 15482
    },
    {
      "epoch": 1.4764785766601562e-05,
      "step": 15482,
      "training_step_time": 17.452288389205933
    },
    {
      "epoch": 1.476573944091797e-05,
      "model_forward_time": 5.6012043952941895,
      "step": 15483
    },
    {
      "epoch": 1.476573944091797e-05,
      "step": 15483,
      "training_step_time": 17.02514123916626
    },
    {
      "epoch": 1.4766693115234375e-05,
      "model_forward_time": 5.569592237472534,
      "step": 15484
    },
    {
      "epoch": 1.4766693115234375e-05,
      "step": 15484,
      "training_step_time": 16.98249626159668
    },
    {
      "epoch": 1.476764678955078e-05,
      "model_forward_time": 5.565329551696777,
      "step": 15485
    },
    {
      "epoch": 1.476764678955078e-05,
      "step": 15485,
      "training_step_time": 16.99866247177124
    },
    {
      "epoch": 1.4768600463867188e-05,
      "model_forward_time": 5.563449382781982,
      "step": 15486
    },
    {
      "epoch": 1.4768600463867188e-05,
      "step": 15486,
      "training_step_time": 16.986522436141968
    },
    {
      "epoch": 1.4769554138183594e-05,
      "model_forward_time": 5.9640491008758545,
      "step": 15487
    },
    {
      "epoch": 1.4769554138183594e-05,
      "step": 15487,
      "training_step_time": 17.370612859725952
    },
    {
      "epoch": 1.47705078125e-05,
      "model_forward_time": 5.568774461746216,
      "step": 15488
    },
    {
      "epoch": 1.47705078125e-05,
      "step": 15488,
      "training_step_time": 16.986780166625977
    },
    {
      "epoch": 1.4771461486816407e-05,
      "model_forward_time": 5.576407194137573,
      "step": 15489
    },
    {
      "epoch": 1.4771461486816407e-05,
      "step": 15489,
      "training_step_time": 17.011855125427246
    },
    {
      "epoch": 1.4772415161132813e-05,
      "grad_norm": 0.6663709878921509,
      "learning_rate": 1e-05,
      "loss": 0.069,
      "step": 15490
    },
    {
      "epoch": 1.4772415161132813e-05,
      "model_forward_time": 5.591484546661377,
      "step": 15490
    },
    {
      "dynamics_loss_avg": 0.056837166845798495,
      "epoch": 1.4772415161132813e-05,
      "step": 15490
    },
    {
      "action_loss_avg": 0.00643036097753793,
      "epoch": 1.4772415161132813e-05,
      "step": 15490
    },
    {
      "epoch": 1.4772415161132813e-05,
      "step": 15490,
      "training_step_time": 17.014307975769043
    },
    {
      "epoch": 1.4773368835449218e-05,
      "model_forward_time": 5.5964696407318115,
      "step": 15491
    },
    {
      "epoch": 1.4773368835449218e-05,
      "step": 15491,
      "training_step_time": 17.088871955871582
    },
    {
      "epoch": 1.4774322509765626e-05,
      "model_forward_time": 5.605708122253418,
      "step": 15492
    },
    {
      "epoch": 1.4774322509765626e-05,
      "step": 15492,
      "training_step_time": 17.104636907577515
    },
    {
      "epoch": 1.4775276184082032e-05,
      "model_forward_time": 5.982913017272949,
      "step": 15493
    },
    {
      "epoch": 1.4775276184082032e-05,
      "step": 15493,
      "training_step_time": 17.49786949157715
    },
    {
      "epoch": 1.4776229858398437e-05,
      "model_forward_time": 5.622747898101807,
      "step": 15494
    },
    {
      "epoch": 1.4776229858398437e-05,
      "step": 15494,
      "training_step_time": 17.136655807495117
    },
    {
      "epoch": 1.4777183532714843e-05,
      "model_forward_time": 5.6165502071380615,
      "step": 15495
    },
    {
      "epoch": 1.4777183532714843e-05,
      "step": 15495,
      "training_step_time": 17.134094953536987
    },
    {
      "epoch": 1.477813720703125e-05,
      "model_forward_time": 5.618642568588257,
      "step": 15496
    },
    {
      "epoch": 1.477813720703125e-05,
      "step": 15496,
      "training_step_time": 17.131909132003784
    },
    {
      "epoch": 1.4779090881347656e-05,
      "model_forward_time": 5.630069017410278,
      "step": 15497
    },
    {
      "epoch": 1.4779090881347656e-05,
      "step": 15497,
      "training_step_time": 17.121579885482788
    },
    {
      "epoch": 1.4780044555664062e-05,
      "model_forward_time": 5.604930400848389,
      "step": 15498
    },
    {
      "epoch": 1.4780044555664062e-05,
      "step": 15498,
      "training_step_time": 17.43090796470642
    },
    {
      "epoch": 1.478099822998047e-05,
      "model_forward_time": 5.599260091781616,
      "step": 15499
    },
    {
      "epoch": 1.478099822998047e-05,
      "step": 15499,
      "training_step_time": 17.0810227394104
    },
    {
      "epoch": 1.4781951904296875e-05,
      "grad_norm": 0.9044815301895142,
      "learning_rate": 1e-05,
      "loss": 0.0576,
      "step": 15500
    },
    {
      "epoch": 1.4781951904296875e-05,
      "model_forward_time": 5.601769924163818,
      "step": 15500
    },
    {
      "dynamics_loss_avg": 0.04553852919489145,
      "epoch": 1.4781951904296875e-05,
      "step": 15500
    },
    {
      "action_loss_avg": 0.004931749508250505,
      "epoch": 1.4781951904296875e-05,
      "step": 15500
    },
    {
      "epoch": 1.4781951904296875e-05,
      "step": 15500,
      "training_step_time": 17.07775592803955
    },
    {
      "epoch": 1.478290557861328e-05,
      "model_forward_time": 5.612197160720825,
      "step": 15501
    },
    {
      "epoch": 1.478290557861328e-05,
      "step": 15501,
      "training_step_time": 17.06857466697693
    },
    {
      "epoch": 1.4783859252929688e-05,
      "model_forward_time": 5.6012938022613525,
      "step": 15502
    },
    {
      "epoch": 1.4783859252929688e-05,
      "step": 15502,
      "training_step_time": 17.058842420578003
    },
    {
      "epoch": 1.4784812927246094e-05,
      "model_forward_time": 5.592360734939575,
      "step": 15503
    },
    {
      "epoch": 1.4784812927246094e-05,
      "step": 15503,
      "training_step_time": 17.0202579498291
    },
    {
      "epoch": 1.47857666015625e-05,
      "model_forward_time": 5.94402003288269,
      "step": 15504
    },
    {
      "epoch": 1.47857666015625e-05,
      "step": 15504,
      "training_step_time": 17.390652894973755
    },
    {
      "epoch": 1.4786720275878907e-05,
      "model_forward_time": 5.59335470199585,
      "step": 15505
    },
    {
      "epoch": 1.4786720275878907e-05,
      "step": 15505,
      "training_step_time": 17.067872047424316
    },
    {
      "epoch": 1.4787673950195313e-05,
      "model_forward_time": 5.603026866912842,
      "step": 15506
    },
    {
      "epoch": 1.4787673950195313e-05,
      "step": 15506,
      "training_step_time": 17.07507085800171
    },
    {
      "epoch": 1.4788627624511718e-05,
      "model_forward_time": 5.597628116607666,
      "step": 15507
    },
    {
      "epoch": 1.4788627624511718e-05,
      "step": 15507,
      "training_step_time": 17.057910680770874
    },
    {
      "epoch": 1.4789581298828124e-05,
      "model_forward_time": 5.598980665206909,
      "step": 15508
    },
    {
      "epoch": 1.4789581298828124e-05,
      "step": 15508,
      "training_step_time": 17.053918600082397
    },
    {
      "epoch": 1.4790534973144532e-05,
      "model_forward_time": 6.035077095031738,
      "step": 15509
    },
    {
      "epoch": 1.4790534973144532e-05,
      "step": 15509,
      "training_step_time": 17.514119625091553
    },
    {
      "epoch": 1.4791488647460937e-05,
      "grad_norm": 0.5207697749137878,
      "learning_rate": 1e-05,
      "loss": 0.0608,
      "step": 15510
    },
    {
      "epoch": 1.4791488647460937e-05,
      "model_forward_time": 5.608060359954834,
      "step": 15510
    },
    {
      "dynamics_loss_avg": 0.05692025404423475,
      "epoch": 1.4791488647460937e-05,
      "step": 15510
    },
    {
      "action_loss_avg": 0.008288061060011388,
      "epoch": 1.4791488647460937e-05,
      "step": 15510
    },
    {
      "epoch": 1.4791488647460937e-05,
      "step": 15510,
      "training_step_time": 17.079922914505005
    },
    {
      "epoch": 1.4792442321777343e-05,
      "model_forward_time": 5.598215579986572,
      "step": 15511
    },
    {
      "epoch": 1.4792442321777343e-05,
      "step": 15511,
      "training_step_time": 17.06836771965027
    },
    {
      "epoch": 1.479339599609375e-05,
      "model_forward_time": 5.592691659927368,
      "step": 15512
    },
    {
      "epoch": 1.479339599609375e-05,
      "step": 15512,
      "training_step_time": 17.040643453598022
    },
    {
      "epoch": 1.4794349670410156e-05,
      "model_forward_time": 5.600008964538574,
      "step": 15513
    },
    {
      "epoch": 1.4794349670410156e-05,
      "step": 15513,
      "training_step_time": 17.0809063911438
    },
    {
      "epoch": 1.4795303344726562e-05,
      "model_forward_time": 5.607738018035889,
      "step": 15514
    },
    {
      "epoch": 1.4795303344726562e-05,
      "step": 15514,
      "training_step_time": 17.09493327140808
    },
    {
      "epoch": 1.479625701904297e-05,
      "model_forward_time": 5.963244438171387,
      "step": 15515
    },
    {
      "epoch": 1.479625701904297e-05,
      "step": 15515,
      "training_step_time": 17.42761993408203
    },
    {
      "epoch": 1.4797210693359375e-05,
      "model_forward_time": 4.266271352767944,
      "step": 15516
    },
    {
      "epoch": 1.4797210693359375e-05,
      "step": 15516,
      "training_step_time": 12.769566774368286
    },
    {
      "epoch": 1.479816436767578e-05,
      "model_forward_time": 5.582877159118652,
      "step": 15517
    },
    {
      "epoch": 1.479816436767578e-05,
      "step": 15517,
      "training_step_time": 17.028916835784912
    },
    {
      "epoch": 1.4799118041992188e-05,
      "model_forward_time": 5.594778299331665,
      "step": 15518
    },
    {
      "epoch": 1.4799118041992188e-05,
      "step": 15518,
      "training_step_time": 17.085784673690796
    },
    {
      "epoch": 1.4800071716308594e-05,
      "model_forward_time": 5.621748447418213,
      "step": 15519
    },
    {
      "epoch": 1.4800071716308594e-05,
      "step": 15519,
      "training_step_time": 17.107671976089478
    },
    {
      "epoch": 1.4801025390625e-05,
      "grad_norm": 1.3693125247955322,
      "learning_rate": 1e-05,
      "loss": 0.0631,
      "step": 15520
    },
    {
      "epoch": 1.4801025390625e-05,
      "model_forward_time": 5.6099021434783936,
      "step": 15520
    },
    {
      "dynamics_loss_avg": 0.048463355377316476,
      "epoch": 1.4801025390625e-05,
      "step": 15520
    },
    {
      "action_loss_avg": 0.01016447793226689,
      "epoch": 1.4801025390625e-05,
      "step": 15520
    },
    {
      "epoch": 1.4801025390625e-05,
      "step": 15520,
      "training_step_time": 17.480077028274536
    },
    {
      "epoch": 1.4801979064941407e-05,
      "model_forward_time": 5.5989251136779785,
      "step": 15521
    },
    {
      "epoch": 1.4801979064941407e-05,
      "step": 15521,
      "training_step_time": 17.07690167427063
    },
    {
      "epoch": 1.4802932739257813e-05,
      "model_forward_time": 5.615522623062134,
      "step": 15522
    },
    {
      "epoch": 1.4802932739257813e-05,
      "step": 15522,
      "training_step_time": 17.10559892654419
    },
    {
      "epoch": 1.4803886413574218e-05,
      "model_forward_time": 5.605009078979492,
      "step": 15523
    },
    {
      "epoch": 1.4803886413574218e-05,
      "step": 15523,
      "training_step_time": 17.09002375602722
    },
    {
      "epoch": 1.4804840087890624e-05,
      "model_forward_time": 5.606435537338257,
      "step": 15524
    },
    {
      "epoch": 1.4804840087890624e-05,
      "step": 15524,
      "training_step_time": 17.086944103240967
    },
    {
      "epoch": 1.4805793762207032e-05,
      "model_forward_time": 5.614000082015991,
      "step": 15525
    },
    {
      "epoch": 1.4805793762207032e-05,
      "step": 15525,
      "training_step_time": 17.109139919281006
    },
    {
      "epoch": 1.4806747436523437e-05,
      "model_forward_time": 5.964233875274658,
      "step": 15526
    },
    {
      "epoch": 1.4806747436523437e-05,
      "step": 15526,
      "training_step_time": 17.4466712474823
    },
    {
      "epoch": 1.4807701110839843e-05,
      "model_forward_time": 5.599494457244873,
      "step": 15527
    },
    {
      "epoch": 1.4807701110839843e-05,
      "step": 15527,
      "training_step_time": 17.08103108406067
    },
    {
      "epoch": 1.480865478515625e-05,
      "model_forward_time": 5.594188928604126,
      "step": 15528
    },
    {
      "epoch": 1.480865478515625e-05,
      "step": 15528,
      "training_step_time": 17.0634343624115
    },
    {
      "epoch": 1.4809608459472656e-05,
      "model_forward_time": 5.604877471923828,
      "step": 15529
    },
    {
      "epoch": 1.4809608459472656e-05,
      "step": 15529,
      "training_step_time": 17.092705726623535
    },
    {
      "epoch": 1.4810562133789062e-05,
      "grad_norm": 0.9827640652656555,
      "learning_rate": 1e-05,
      "loss": 0.0702,
      "step": 15530
    },
    {
      "epoch": 1.4810562133789062e-05,
      "model_forward_time": 5.609038591384888,
      "step": 15530
    },
    {
      "dynamics_loss_avg": 0.06280098091810941,
      "epoch": 1.4810562133789062e-05,
      "step": 15530
    },
    {
      "action_loss_avg": 0.008628075616434217,
      "epoch": 1.4810562133789062e-05,
      "step": 15530
    },
    {
      "epoch": 1.4810562133789062e-05,
      "step": 15530,
      "training_step_time": 17.08859372138977
    },
    {
      "epoch": 1.481151580810547e-05,
      "model_forward_time": 5.972445249557495,
      "step": 15531
    },
    {
      "epoch": 1.481151580810547e-05,
      "step": 15531,
      "training_step_time": 17.413954973220825
    },
    {
      "epoch": 1.4812469482421875e-05,
      "model_forward_time": 5.6008734703063965,
      "step": 15532
    },
    {
      "epoch": 1.4812469482421875e-05,
      "step": 15532,
      "training_step_time": 17.07449436187744
    },
    {
      "epoch": 1.481342315673828e-05,
      "model_forward_time": 5.62757420539856,
      "step": 15533
    },
    {
      "epoch": 1.481342315673828e-05,
      "step": 15533,
      "training_step_time": 17.123563289642334
    },
    {
      "epoch": 1.4814376831054688e-05,
      "model_forward_time": 5.602487802505493,
      "step": 15534
    },
    {
      "epoch": 1.4814376831054688e-05,
      "step": 15534,
      "training_step_time": 17.053044080734253
    },
    {
      "epoch": 1.4815330505371094e-05,
      "model_forward_time": 5.588116645812988,
      "step": 15535
    },
    {
      "epoch": 1.4815330505371094e-05,
      "step": 15535,
      "training_step_time": 17.02151131629944
    },
    {
      "epoch": 1.48162841796875e-05,
      "model_forward_time": 5.583221912384033,
      "step": 15536
    },
    {
      "epoch": 1.48162841796875e-05,
      "step": 15536,
      "training_step_time": 17.03158688545227
    },
    {
      "epoch": 1.4817237854003907e-05,
      "model_forward_time": 5.949021816253662,
      "step": 15537
    },
    {
      "epoch": 1.4817237854003907e-05,
      "step": 15537,
      "training_step_time": 17.449576139450073
    },
    {
      "epoch": 1.4818191528320313e-05,
      "model_forward_time": 5.6223530769348145,
      "step": 15538
    },
    {
      "epoch": 1.4818191528320313e-05,
      "step": 15538,
      "training_step_time": 17.139002323150635
    },
    {
      "epoch": 1.4819145202636718e-05,
      "model_forward_time": 5.631007194519043,
      "step": 15539
    },
    {
      "epoch": 1.4819145202636718e-05,
      "step": 15539,
      "training_step_time": 17.12020778656006
    },
    {
      "epoch": 1.4820098876953124e-05,
      "grad_norm": 0.6270980834960938,
      "learning_rate": 1e-05,
      "loss": 0.0646,
      "step": 15540
    },
    {
      "epoch": 1.4820098876953124e-05,
      "model_forward_time": 5.611598968505859,
      "step": 15540
    },
    {
      "dynamics_loss_avg": 0.06099271811544895,
      "epoch": 1.4820098876953124e-05,
      "step": 15540
    },
    {
      "action_loss_avg": 0.008382002171128989,
      "epoch": 1.4820098876953124e-05,
      "step": 15540
    },
    {
      "epoch": 1.4820098876953124e-05,
      "step": 15540,
      "training_step_time": 17.07012963294983
    },
    {
      "epoch": 1.4821052551269532e-05,
      "model_forward_time": 5.593576192855835,
      "step": 15541
    },
    {
      "epoch": 1.4821052551269532e-05,
      "step": 15541,
      "training_step_time": 17.00741481781006
    },
    {
      "epoch": 1.4822006225585937e-05,
      "model_forward_time": 5.963221549987793,
      "step": 15542
    },
    {
      "epoch": 1.4822006225585937e-05,
      "step": 15542,
      "training_step_time": 17.408658742904663
    },
    {
      "epoch": 1.4822959899902343e-05,
      "model_forward_time": 5.597827434539795,
      "step": 15543
    },
    {
      "epoch": 1.4822959899902343e-05,
      "step": 15543,
      "training_step_time": 17.04973030090332
    },
    {
      "epoch": 1.482391357421875e-05,
      "model_forward_time": 5.603539705276489,
      "step": 15544
    },
    {
      "epoch": 1.482391357421875e-05,
      "step": 15544,
      "training_step_time": 17.06475591659546
    },
    {
      "epoch": 1.4824867248535156e-05,
      "model_forward_time": 5.610992908477783,
      "step": 15545
    },
    {
      "epoch": 1.4824867248535156e-05,
      "step": 15545,
      "training_step_time": 17.080515384674072
    },
    {
      "epoch": 1.4825820922851562e-05,
      "model_forward_time": 5.609071493148804,
      "step": 15546
    },
    {
      "epoch": 1.4825820922851562e-05,
      "step": 15546,
      "training_step_time": 17.08223581314087
    },
    {
      "epoch": 1.482677459716797e-05,
      "model_forward_time": 5.604682683944702,
      "step": 15547
    },
    {
      "epoch": 1.482677459716797e-05,
      "step": 15547,
      "training_step_time": 17.464219331741333
    },
    {
      "epoch": 1.4827728271484375e-05,
      "model_forward_time": 5.597833633422852,
      "step": 15548
    },
    {
      "epoch": 1.4827728271484375e-05,
      "step": 15548,
      "training_step_time": 17.078702926635742
    },
    {
      "epoch": 1.482868194580078e-05,
      "model_forward_time": 5.5968756675720215,
      "step": 15549
    },
    {
      "epoch": 1.482868194580078e-05,
      "step": 15549,
      "training_step_time": 17.07292866706848
    },
    {
      "epoch": 1.4829635620117188e-05,
      "grad_norm": 0.31718695163726807,
      "learning_rate": 1e-05,
      "loss": 0.066,
      "step": 15550
    },
    {
      "epoch": 1.4829635620117188e-05,
      "model_forward_time": 5.598693132400513,
      "step": 15550
    },
    {
      "dynamics_loss_avg": 0.054177777096629146,
      "epoch": 1.4829635620117188e-05,
      "step": 15550
    },
    {
      "action_loss_avg": 0.008308081352151931,
      "epoch": 1.4829635620117188e-05,
      "step": 15550
    },
    {
      "epoch": 1.4829635620117188e-05,
      "step": 15550,
      "training_step_time": 17.051729440689087
    },
    {
      "epoch": 1.4830589294433594e-05,
      "model_forward_time": 5.597841024398804,
      "step": 15551
    },
    {
      "epoch": 1.4830589294433594e-05,
      "step": 15551,
      "training_step_time": 17.08360481262207
    },
    {
      "epoch": 1.483154296875e-05,
      "model_forward_time": 5.610262632369995,
      "step": 15552
    },
    {
      "epoch": 1.483154296875e-05,
      "step": 15552,
      "training_step_time": 17.09825348854065
    },
    {
      "epoch": 1.4832496643066407e-05,
      "model_forward_time": 5.983488082885742,
      "step": 15553
    },
    {
      "epoch": 1.4832496643066407e-05,
      "step": 15553,
      "training_step_time": 17.47960877418518
    },
    {
      "epoch": 1.4833450317382813e-05,
      "model_forward_time": 5.611979007720947,
      "step": 15554
    },
    {
      "epoch": 1.4833450317382813e-05,
      "step": 15554,
      "training_step_time": 17.113030910491943
    },
    {
      "epoch": 1.4834403991699218e-05,
      "model_forward_time": 5.609002351760864,
      "step": 15555
    },
    {
      "epoch": 1.4834403991699218e-05,
      "step": 15555,
      "training_step_time": 17.091562747955322
    },
    {
      "epoch": 1.4835357666015624e-05,
      "model_forward_time": 5.5917418003082275,
      "step": 15556
    },
    {
      "epoch": 1.4835357666015624e-05,
      "step": 15556,
      "training_step_time": 17.032394409179688
    },
    {
      "epoch": 1.4836311340332032e-05,
      "model_forward_time": 5.583964824676514,
      "step": 15557
    },
    {
      "epoch": 1.4836311340332032e-05,
      "step": 15557,
      "training_step_time": 17.004316568374634
    },
    {
      "epoch": 1.4837265014648437e-05,
      "model_forward_time": 5.956166744232178,
      "step": 15558
    },
    {
      "epoch": 1.4837265014648437e-05,
      "step": 15558,
      "training_step_time": 17.40869665145874
    },
    {
      "epoch": 1.4838218688964843e-05,
      "model_forward_time": 5.614162445068359,
      "step": 15559
    },
    {
      "epoch": 1.4838218688964843e-05,
      "step": 15559,
      "training_step_time": 17.098153114318848
    },
    {
      "epoch": 1.483917236328125e-05,
      "grad_norm": 0.7941223382949829,
      "learning_rate": 1e-05,
      "loss": 0.0511,
      "step": 15560
    },
    {
      "epoch": 1.483917236328125e-05,
      "model_forward_time": 5.628307819366455,
      "step": 15560
    },
    {
      "dynamics_loss_avg": 0.04412240581586957,
      "epoch": 1.483917236328125e-05,
      "step": 15560
    },
    {
      "action_loss_avg": 0.005779827025253326,
      "epoch": 1.483917236328125e-05,
      "step": 15560
    },
    {
      "epoch": 1.483917236328125e-05,
      "step": 15560,
      "training_step_time": 17.12205219268799
    },
    {
      "epoch": 1.4840126037597656e-05,
      "model_forward_time": 5.595306873321533,
      "step": 15561
    },
    {
      "epoch": 1.4840126037597656e-05,
      "step": 15561,
      "training_step_time": 17.06807589530945
    },
    {
      "epoch": 1.4841079711914062e-05,
      "model_forward_time": 5.606935262680054,
      "step": 15562
    },
    {
      "epoch": 1.4841079711914062e-05,
      "step": 15562,
      "training_step_time": 17.058788299560547
    },
    {
      "epoch": 1.484203338623047e-05,
      "model_forward_time": 5.592000246047974,
      "step": 15563
    },
    {
      "epoch": 1.484203338623047e-05,
      "step": 15563,
      "training_step_time": 17.061420440673828
    },
    {
      "epoch": 1.4842987060546875e-05,
      "model_forward_time": 5.98963737487793,
      "step": 15564
    },
    {
      "epoch": 1.4842987060546875e-05,
      "step": 15564,
      "training_step_time": 17.47414493560791
    },
    {
      "epoch": 1.484394073486328e-05,
      "model_forward_time": 5.600417375564575,
      "step": 15565
    },
    {
      "epoch": 1.484394073486328e-05,
      "step": 15565,
      "training_step_time": 17.073350191116333
    },
    {
      "epoch": 1.4844894409179688e-05,
      "model_forward_time": 5.606219291687012,
      "step": 15566
    },
    {
      "epoch": 1.4844894409179688e-05,
      "step": 15566,
      "training_step_time": 17.08696436882019
    },
    {
      "epoch": 1.4845848083496094e-05,
      "model_forward_time": 5.601288795471191,
      "step": 15567
    },
    {
      "epoch": 1.4845848083496094e-05,
      "step": 15567,
      "training_step_time": 17.060304164886475
    },
    {
      "epoch": 1.48468017578125e-05,
      "model_forward_time": 5.603964328765869,
      "step": 15568
    },
    {
      "epoch": 1.48468017578125e-05,
      "step": 15568,
      "training_step_time": 17.067550659179688
    },
    {
      "epoch": 1.4847755432128907e-05,
      "model_forward_time": 5.968164920806885,
      "step": 15569
    },
    {
      "epoch": 1.4847755432128907e-05,
      "step": 15569,
      "training_step_time": 17.405048370361328
    },
    {
      "epoch": 1.4848709106445313e-05,
      "grad_norm": 0.2703861594200134,
      "learning_rate": 1e-05,
      "loss": 0.0659,
      "step": 15570
    },
    {
      "epoch": 1.4848709106445313e-05,
      "model_forward_time": 5.599926471710205,
      "step": 15570
    },
    {
      "dynamics_loss_avg": 0.058784068748354915,
      "epoch": 1.4848709106445313e-05,
      "step": 15570
    },
    {
      "action_loss_avg": 0.008492013497743756,
      "epoch": 1.4848709106445313e-05,
      "step": 15570
    },
    {
      "epoch": 1.4848709106445313e-05,
      "step": 15570,
      "training_step_time": 17.08925199508667
    },
    {
      "epoch": 1.4849662780761718e-05,
      "model_forward_time": 5.6073620319366455,
      "step": 15571
    },
    {
      "epoch": 1.4849662780761718e-05,
      "step": 15571,
      "training_step_time": 17.087891101837158
    },
    {
      "epoch": 1.4850616455078124e-05,
      "model_forward_time": 5.597059726715088,
      "step": 15572
    },
    {
      "epoch": 1.4850616455078124e-05,
      "step": 15572,
      "training_step_time": 17.08892011642456
    },
    {
      "epoch": 1.4851570129394532e-05,
      "model_forward_time": 5.61288046836853,
      "step": 15573
    },
    {
      "epoch": 1.4851570129394532e-05,
      "step": 15573,
      "training_step_time": 17.11472511291504
    },
    {
      "epoch": 1.4852523803710937e-05,
      "model_forward_time": 5.630541086196899,
      "step": 15574
    },
    {
      "epoch": 1.4852523803710937e-05,
      "step": 15574,
      "training_step_time": 17.14651322364807
    },
    {
      "epoch": 1.4853477478027343e-05,
      "model_forward_time": 5.982459306716919,
      "step": 15575
    },
    {
      "epoch": 1.4853477478027343e-05,
      "step": 15575,
      "training_step_time": 17.463558197021484
    },
    {
      "epoch": 1.485443115234375e-05,
      "model_forward_time": 5.596266508102417,
      "step": 15576
    },
    {
      "epoch": 1.485443115234375e-05,
      "step": 15576,
      "training_step_time": 17.03183674812317
    },
    {
      "epoch": 1.4855384826660156e-05,
      "model_forward_time": 5.585123538970947,
      "step": 15577
    },
    {
      "epoch": 1.4855384826660156e-05,
      "step": 15577,
      "training_step_time": 17.032830476760864
    },
    {
      "epoch": 1.4856338500976562e-05,
      "model_forward_time": 5.591719150543213,
      "step": 15578
    },
    {
      "epoch": 1.4856338500976562e-05,
      "step": 15578,
      "training_step_time": 17.067979097366333
    },
    {
      "epoch": 1.485729217529297e-05,
      "model_forward_time": 5.602757692337036,
      "step": 15579
    },
    {
      "epoch": 1.485729217529297e-05,
      "step": 15579,
      "training_step_time": 17.08644962310791
    },
    {
      "epoch": 1.4858245849609375e-05,
      "grad_norm": 1.0365962982177734,
      "learning_rate": 1e-05,
      "loss": 0.0601,
      "step": 15580
    },
    {
      "epoch": 1.4858245849609375e-05,
      "model_forward_time": 5.9748780727386475,
      "step": 15580
    },
    {
      "dynamics_loss_avg": 0.05218730848282575,
      "epoch": 1.4858245849609375e-05,
      "step": 15580
    },
    {
      "action_loss_avg": 0.006930402643047273,
      "epoch": 1.4858245849609375e-05,
      "step": 15580
    },
    {
      "epoch": 1.4858245849609375e-05,
      "step": 15580,
      "training_step_time": 17.476707935333252
    },
    {
      "epoch": 1.485919952392578e-05,
      "model_forward_time": 5.618061065673828,
      "step": 15581
    },
    {
      "epoch": 1.485919952392578e-05,
      "step": 15581,
      "training_step_time": 17.136054039001465
    },
    {
      "epoch": 1.4860153198242188e-05,
      "model_forward_time": 5.623895645141602,
      "step": 15582
    },
    {
      "epoch": 1.4860153198242188e-05,
      "step": 15582,
      "training_step_time": 17.125106811523438
    },
    {
      "epoch": 1.4861106872558594e-05,
      "model_forward_time": 5.613547325134277,
      "step": 15583
    },
    {
      "epoch": 1.4861106872558594e-05,
      "step": 15583,
      "training_step_time": 17.107260942459106
    },
    {
      "epoch": 1.4862060546875e-05,
      "model_forward_time": 5.614682912826538,
      "step": 15584
    },
    {
      "epoch": 1.4862060546875e-05,
      "step": 15584,
      "training_step_time": 17.09053659439087
    },
    {
      "epoch": 1.4863014221191407e-05,
      "model_forward_time": 5.604864597320557,
      "step": 15585
    },
    {
      "epoch": 1.4863014221191407e-05,
      "step": 15585,
      "training_step_time": 17.468777894973755
    },
    {
      "epoch": 1.4863967895507813e-05,
      "model_forward_time": 5.614535808563232,
      "step": 15586
    },
    {
      "epoch": 1.4863967895507813e-05,
      "step": 15586,
      "training_step_time": 17.10961627960205
    },
    {
      "epoch": 1.4864921569824219e-05,
      "model_forward_time": 5.605607509613037,
      "step": 15587
    },
    {
      "epoch": 1.4864921569824219e-05,
      "step": 15587,
      "training_step_time": 17.088327646255493
    },
    {
      "epoch": 1.4865875244140624e-05,
      "model_forward_time": 5.605676889419556,
      "step": 15588
    },
    {
      "epoch": 1.4865875244140624e-05,
      "step": 15588,
      "training_step_time": 17.09499716758728
    },
    {
      "epoch": 1.4866828918457032e-05,
      "model_forward_time": 5.60809850692749,
      "step": 15589
    },
    {
      "epoch": 1.4866828918457032e-05,
      "step": 15589,
      "training_step_time": 17.08446955680847
    },
    {
      "epoch": 1.4867782592773437e-05,
      "grad_norm": 0.6189407110214233,
      "learning_rate": 1e-05,
      "loss": 0.0617,
      "step": 15590
    },
    {
      "epoch": 1.4867782592773437e-05,
      "model_forward_time": 5.597443342208862,
      "step": 15590
    },
    {
      "dynamics_loss_avg": 0.06020852215588093,
      "epoch": 1.4867782592773437e-05,
      "step": 15590
    },
    {
      "action_loss_avg": 0.007024909439496696,
      "epoch": 1.4867782592773437e-05,
      "step": 15590
    },
    {
      "epoch": 1.4867782592773437e-05,
      "step": 15590,
      "training_step_time": 17.024019956588745
    },
    {
      "epoch": 1.4868736267089843e-05,
      "model_forward_time": 5.956615447998047,
      "step": 15591
    },
    {
      "epoch": 1.4868736267089843e-05,
      "step": 15591,
      "training_step_time": 17.377200603485107
    },
    {
      "epoch": 1.486968994140625e-05,
      "model_forward_time": 5.58045768737793,
      "step": 15592
    },
    {
      "epoch": 1.486968994140625e-05,
      "step": 15592,
      "training_step_time": 17.022684335708618
    },
    {
      "epoch": 1.4870643615722656e-05,
      "model_forward_time": 5.5787131786346436,
      "step": 15593
    },
    {
      "epoch": 1.4870643615722656e-05,
      "step": 15593,
      "training_step_time": 17.0119469165802
    },
    {
      "epoch": 1.4871597290039062e-05,
      "model_forward_time": 5.588171720504761,
      "step": 15594
    },
    {
      "epoch": 1.4871597290039062e-05,
      "step": 15594,
      "training_step_time": 17.01956343650818
    },
    {
      "epoch": 1.487255096435547e-05,
      "model_forward_time": 5.5958850383758545,
      "step": 15595
    },
    {
      "epoch": 1.487255096435547e-05,
      "step": 15595,
      "training_step_time": 17.038947582244873
    },
    {
      "epoch": 1.4873504638671875e-05,
      "model_forward_time": 5.592106819152832,
      "step": 15596
    },
    {
      "epoch": 1.4873504638671875e-05,
      "step": 15596,
      "training_step_time": 17.452613353729248
    },
    {
      "epoch": 1.487445831298828e-05,
      "model_forward_time": 5.605467319488525,
      "step": 15597
    },
    {
      "epoch": 1.487445831298828e-05,
      "step": 15597,
      "training_step_time": 17.10175061225891
    },
    {
      "epoch": 1.4875411987304688e-05,
      "model_forward_time": 5.616712331771851,
      "step": 15598
    },
    {
      "epoch": 1.4875411987304688e-05,
      "step": 15598,
      "training_step_time": 17.09689712524414
    },
    {
      "epoch": 1.4876365661621094e-05,
      "model_forward_time": 5.5973451137542725,
      "step": 15599
    },
    {
      "epoch": 1.4876365661621094e-05,
      "step": 15599,
      "training_step_time": 17.053004026412964
    },
    {
      "epoch": 1.48773193359375e-05,
      "grad_norm": 1.2502623796463013,
      "learning_rate": 1e-05,
      "loss": 0.0691,
      "step": 15600
    },
    {
      "epoch": 1.48773193359375e-05,
      "model_forward_time": 5.583027601242065,
      "step": 15600
    },
    {
      "dynamics_loss_avg": 0.05603010356426239,
      "epoch": 1.48773193359375e-05,
      "step": 15600
    },
    {
      "action_loss_avg": 0.008715743001084774,
      "epoch": 1.48773193359375e-05,
      "step": 15600
    },
    {
      "epoch": 1.48773193359375e-05,
      "step": 15600,
      "training_step_time": 17.0380437374115
    },
    {
      "epoch": 1.4878273010253907e-05,
      "model_forward_time": 5.59992241859436,
      "step": 15601
    },
    {
      "epoch": 1.4878273010253907e-05,
      "step": 15601,
      "training_step_time": 17.059144496917725
    },
    {
      "epoch": 1.4879226684570313e-05,
      "model_forward_time": 5.94917368888855,
      "step": 15602
    },
    {
      "epoch": 1.4879226684570313e-05,
      "step": 15602,
      "training_step_time": 17.41214156150818
    },
    {
      "epoch": 1.4880180358886719e-05,
      "model_forward_time": 5.615847826004028,
      "step": 15603
    },
    {
      "epoch": 1.4880180358886719e-05,
      "step": 15603,
      "training_step_time": 17.123608589172363
    },
    {
      "epoch": 1.4881134033203124e-05,
      "model_forward_time": 5.631548881530762,
      "step": 15604
    },
    {
      "epoch": 1.4881134033203124e-05,
      "step": 15604,
      "training_step_time": 17.124666452407837
    },
    {
      "epoch": 1.4882087707519532e-05,
      "model_forward_time": 5.621798753738403,
      "step": 15605
    },
    {
      "epoch": 1.4882087707519532e-05,
      "step": 15605,
      "training_step_time": 17.08973455429077
    },
    {
      "epoch": 1.4883041381835937e-05,
      "model_forward_time": 5.622706890106201,
      "step": 15606
    },
    {
      "epoch": 1.4883041381835937e-05,
      "step": 15606,
      "training_step_time": 17.058990716934204
    },
    {
      "epoch": 1.4883995056152343e-05,
      "model_forward_time": 5.979547023773193,
      "step": 15607
    },
    {
      "epoch": 1.4883995056152343e-05,
      "step": 15607,
      "training_step_time": 17.43166995048523
    },
    {
      "epoch": 1.488494873046875e-05,
      "model_forward_time": 5.587776184082031,
      "step": 15608
    },
    {
      "epoch": 1.488494873046875e-05,
      "step": 15608,
      "training_step_time": 17.03899884223938
    },
    {
      "epoch": 1.4885902404785156e-05,
      "model_forward_time": 5.589627742767334,
      "step": 15609
    },
    {
      "epoch": 1.4885902404785156e-05,
      "step": 15609,
      "training_step_time": 17.025683164596558
    },
    {
      "epoch": 1.4886856079101562e-05,
      "grad_norm": 1.5348166227340698,
      "learning_rate": 1e-05,
      "loss": 0.0633,
      "step": 15610
    },
    {
      "epoch": 1.4886856079101562e-05,
      "model_forward_time": 5.606525421142578,
      "step": 15610
    },
    {
      "dynamics_loss_avg": 0.06194948516786099,
      "epoch": 1.4886856079101562e-05,
      "step": 15610
    },
    {
      "action_loss_avg": 0.007326798094436526,
      "epoch": 1.4886856079101562e-05,
      "step": 15610
    },
    {
      "epoch": 1.4886856079101562e-05,
      "step": 15610,
      "training_step_time": 17.057274103164673
    },
    {
      "epoch": 1.488780975341797e-05,
      "model_forward_time": 5.591972351074219,
      "step": 15611
    },
    {
      "epoch": 1.488780975341797e-05,
      "step": 15611,
      "training_step_time": 17.05456304550171
    },
    {
      "epoch": 1.4888763427734375e-05,
      "model_forward_time": 5.6028828620910645,
      "step": 15612
    },
    {
      "epoch": 1.4888763427734375e-05,
      "step": 15612,
      "training_step_time": 17.070759296417236
    },
    {
      "epoch": 1.488971710205078e-05,
      "model_forward_time": 5.976547718048096,
      "step": 15613
    },
    {
      "epoch": 1.488971710205078e-05,
      "step": 15613,
      "training_step_time": 17.457094192504883
    },
    {
      "epoch": 1.4890670776367188e-05,
      "model_forward_time": 5.591963768005371,
      "step": 15614
    },
    {
      "epoch": 1.4890670776367188e-05,
      "step": 15614,
      "training_step_time": 17.056352615356445
    },
    {
      "epoch": 1.4891624450683594e-05,
      "model_forward_time": 5.584179401397705,
      "step": 15615
    },
    {
      "epoch": 1.4891624450683594e-05,
      "step": 15615,
      "training_step_time": 17.044180870056152
    },
    {
      "epoch": 1.4892578125e-05,
      "model_forward_time": 5.596345663070679,
      "step": 15616
    },
    {
      "epoch": 1.4892578125e-05,
      "step": 15616,
      "training_step_time": 17.076168537139893
    },
    {
      "epoch": 1.4893531799316407e-05,
      "model_forward_time": 5.620930433273315,
      "step": 15617
    },
    {
      "epoch": 1.4893531799316407e-05,
      "step": 15617,
      "training_step_time": 17.14476227760315
    },
    {
      "epoch": 1.4894485473632813e-05,
      "model_forward_time": 6.001836538314819,
      "step": 15618
    },
    {
      "epoch": 1.4894485473632813e-05,
      "step": 15618,
      "training_step_time": 17.502782106399536
    },
    {
      "epoch": 1.4895439147949219e-05,
      "model_forward_time": 5.6161558628082275,
      "step": 15619
    },
    {
      "epoch": 1.4895439147949219e-05,
      "step": 15619,
      "training_step_time": 17.089462280273438
    },
    {
      "epoch": 1.4896392822265624e-05,
      "grad_norm": 1.05659818649292,
      "learning_rate": 1e-05,
      "loss": 0.0677,
      "step": 15620
    },
    {
      "epoch": 1.4896392822265624e-05,
      "model_forward_time": 5.6174561977386475,
      "step": 15620
    },
    {
      "dynamics_loss_avg": 0.055747007951140404,
      "epoch": 1.4896392822265624e-05,
      "step": 15620
    },
    {
      "action_loss_avg": 0.0078516052570194,
      "epoch": 1.4896392822265624e-05,
      "step": 15620
    },
    {
      "epoch": 1.4896392822265624e-05,
      "step": 15620,
      "training_step_time": 17.102845668792725
    },
    {
      "epoch": 1.4897346496582032e-05,
      "model_forward_time": 5.615577936172485,
      "step": 15621
    },
    {
      "epoch": 1.4897346496582032e-05,
      "step": 15621,
      "training_step_time": 17.12118649482727
    },
    {
      "epoch": 1.4898300170898437e-05,
      "model_forward_time": 5.614567995071411,
      "step": 15622
    },
    {
      "epoch": 1.4898300170898437e-05,
      "step": 15622,
      "training_step_time": 17.118881702423096
    },
    {
      "epoch": 1.4899253845214843e-05,
      "model_forward_time": 5.626006364822388,
      "step": 15623
    },
    {
      "epoch": 1.4899253845214843e-05,
      "step": 15623,
      "training_step_time": 17.1242995262146
    },
    {
      "epoch": 1.490020751953125e-05,
      "model_forward_time": 5.967162132263184,
      "step": 15624
    },
    {
      "epoch": 1.490020751953125e-05,
      "step": 15624,
      "training_step_time": 17.438507795333862
    },
    {
      "epoch": 1.4901161193847656e-05,
      "model_forward_time": 5.603971481323242,
      "step": 15625
    },
    {
      "epoch": 1.4901161193847656e-05,
      "step": 15625,
      "training_step_time": 17.062278747558594
    },
    {
      "epoch": 1.4902114868164062e-05,
      "model_forward_time": 5.604394197463989,
      "step": 15626
    },
    {
      "epoch": 1.4902114868164062e-05,
      "step": 15626,
      "training_step_time": 17.04793930053711
    },
    {
      "epoch": 1.490306854248047e-05,
      "model_forward_time": 5.594876527786255,
      "step": 15627
    },
    {
      "epoch": 1.490306854248047e-05,
      "step": 15627,
      "training_step_time": 17.104995727539062
    },
    {
      "epoch": 1.4904022216796875e-05,
      "model_forward_time": 5.61830472946167,
      "step": 15628
    },
    {
      "epoch": 1.4904022216796875e-05,
      "step": 15628,
      "training_step_time": 17.133057832717896
    },
    {
      "epoch": 1.490497589111328e-05,
      "model_forward_time": 6.003153324127197,
      "step": 15629
    },
    {
      "epoch": 1.490497589111328e-05,
      "step": 15629,
      "training_step_time": 17.501650094985962
    },
    {
      "epoch": 1.4905929565429688e-05,
      "grad_norm": 0.9475592970848083,
      "learning_rate": 1e-05,
      "loss": 0.0729,
      "step": 15630
    },
    {
      "epoch": 1.4905929565429688e-05,
      "model_forward_time": 5.608686208724976,
      "step": 15630
    },
    {
      "dynamics_loss_avg": 0.0635846134275198,
      "epoch": 1.4905929565429688e-05,
      "step": 15630
    },
    {
      "action_loss_avg": 0.009959601820446551,
      "epoch": 1.4905929565429688e-05,
      "step": 15630
    },
    {
      "epoch": 1.4905929565429688e-05,
      "step": 15630,
      "training_step_time": 17.05880570411682
    },
    {
      "epoch": 1.4906883239746094e-05,
      "model_forward_time": 5.583191156387329,
      "step": 15631
    },
    {
      "epoch": 1.4906883239746094e-05,
      "step": 15631,
      "training_step_time": 17.00819206237793
    },
    {
      "epoch": 1.49078369140625e-05,
      "model_forward_time": 5.589134693145752,
      "step": 15632
    },
    {
      "epoch": 1.49078369140625e-05,
      "step": 15632,
      "training_step_time": 17.016124486923218
    },
    {
      "epoch": 1.4908790588378905e-05,
      "model_forward_time": 5.58519983291626,
      "step": 15633
    },
    {
      "epoch": 1.4908790588378905e-05,
      "step": 15633,
      "training_step_time": 17.013237476348877
    },
    {
      "epoch": 1.4909744262695313e-05,
      "model_forward_time": 5.9382264614105225,
      "step": 15634
    },
    {
      "epoch": 1.4909744262695313e-05,
      "step": 15634,
      "training_step_time": 17.381867170333862
    },
    {
      "epoch": 1.4910697937011719e-05,
      "model_forward_time": 5.592371463775635,
      "step": 15635
    },
    {
      "epoch": 1.4910697937011719e-05,
      "step": 15635,
      "training_step_time": 17.06344985961914
    },
    {
      "epoch": 1.4911651611328124e-05,
      "model_forward_time": 5.601114273071289,
      "step": 15636
    },
    {
      "epoch": 1.4911651611328124e-05,
      "step": 15636,
      "training_step_time": 17.106722593307495
    },
    {
      "epoch": 1.4912605285644532e-05,
      "model_forward_time": 5.619335412979126,
      "step": 15637
    },
    {
      "epoch": 1.4912605285644532e-05,
      "step": 15637,
      "training_step_time": 17.141562938690186
    },
    {
      "epoch": 1.4913558959960937e-05,
      "model_forward_time": 5.620337247848511,
      "step": 15638
    },
    {
      "epoch": 1.4913558959960937e-05,
      "step": 15638,
      "training_step_time": 17.12739586830139
    },
    {
      "epoch": 1.4914512634277343e-05,
      "model_forward_time": 5.614979982376099,
      "step": 15639
    },
    {
      "epoch": 1.4914512634277343e-05,
      "step": 15639,
      "training_step_time": 17.107561111450195
    },
    {
      "epoch": 1.491546630859375e-05,
      "grad_norm": 1.1042320728302002,
      "learning_rate": 1e-05,
      "loss": 0.0584,
      "step": 15640
    },
    {
      "epoch": 1.491546630859375e-05,
      "model_forward_time": 5.971634864807129,
      "step": 15640
    },
    {
      "dynamics_loss_avg": 0.046568705886602405,
      "epoch": 1.491546630859375e-05,
      "step": 15640
    },
    {
      "action_loss_avg": 0.007542017474770546,
      "epoch": 1.491546630859375e-05,
      "step": 15640
    },
    {
      "epoch": 1.491546630859375e-05,
      "step": 15640,
      "training_step_time": 17.42049479484558
    },
    {
      "epoch": 1.4916419982910156e-05,
      "model_forward_time": 5.5976972579956055,
      "step": 15641
    },
    {
      "epoch": 1.4916419982910156e-05,
      "step": 15641,
      "training_step_time": 17.056487321853638
    },
    {
      "epoch": 1.4917373657226562e-05,
      "model_forward_time": 5.6046693325042725,
      "step": 15642
    },
    {
      "epoch": 1.4917373657226562e-05,
      "step": 15642,
      "training_step_time": 17.062697649002075
    },
    {
      "epoch": 1.491832733154297e-05,
      "model_forward_time": 5.609357833862305,
      "step": 15643
    },
    {
      "epoch": 1.491832733154297e-05,
      "step": 15643,
      "training_step_time": 17.10970163345337
    },
    {
      "epoch": 1.4919281005859375e-05,
      "model_forward_time": 5.627278804779053,
      "step": 15644
    },
    {
      "epoch": 1.4919281005859375e-05,
      "step": 15644,
      "training_step_time": 17.134818077087402
    },
    {
      "epoch": 1.492023468017578e-05,
      "model_forward_time": 5.961807012557983,
      "step": 15645
    },
    {
      "epoch": 1.492023468017578e-05,
      "step": 15645,
      "training_step_time": 17.45396113395691
    },
    {
      "epoch": 1.4921188354492188e-05,
      "model_forward_time": 5.6230080127716064,
      "step": 15646
    },
    {
      "epoch": 1.4921188354492188e-05,
      "step": 15646,
      "training_step_time": 17.115291595458984
    },
    {
      "epoch": 1.4922142028808594e-05,
      "model_forward_time": 5.613130569458008,
      "step": 15647
    },
    {
      "epoch": 1.4922142028808594e-05,
      "step": 15647,
      "training_step_time": 17.106873273849487
    },
    {
      "epoch": 1.4923095703125e-05,
      "model_forward_time": 5.620002269744873,
      "step": 15648
    },
    {
      "epoch": 1.4923095703125e-05,
      "step": 15648,
      "training_step_time": 17.1066415309906
    },
    {
      "epoch": 1.4924049377441405e-05,
      "model_forward_time": 5.617477893829346,
      "step": 15649
    },
    {
      "epoch": 1.4924049377441405e-05,
      "step": 15649,
      "training_step_time": 17.080141305923462
    },
    {
      "epoch": 1.4925003051757813e-05,
      "grad_norm": 0.5626869201660156,
      "learning_rate": 1e-05,
      "loss": 0.0661,
      "step": 15650
    },
    {
      "epoch": 1.4925003051757813e-05,
      "model_forward_time": 5.601085662841797,
      "step": 15650
    },
    {
      "dynamics_loss_avg": 0.058227797783911225,
      "epoch": 1.4925003051757813e-05,
      "step": 15650
    },
    {
      "action_loss_avg": 0.00854521170258522,
      "epoch": 1.4925003051757813e-05,
      "step": 15650
    },
    {
      "epoch": 1.4925003051757813e-05,
      "step": 15650,
      "training_step_time": 17.417726039886475
    },
    {
      "epoch": 1.4925956726074219e-05,
      "model_forward_time": 5.596017599105835,
      "step": 15651
    },
    {
      "epoch": 1.4925956726074219e-05,
      "step": 15651,
      "training_step_time": 17.05027198791504
    },
    {
      "epoch": 1.4926910400390624e-05,
      "model_forward_time": 5.595755338668823,
      "step": 15652
    },
    {
      "epoch": 1.4926910400390624e-05,
      "step": 15652,
      "training_step_time": 17.08385133743286
    },
    {
      "epoch": 1.4927864074707032e-05,
      "model_forward_time": 5.608318567276001,
      "step": 15653
    },
    {
      "epoch": 1.4927864074707032e-05,
      "step": 15653,
      "training_step_time": 17.089402437210083
    },
    {
      "epoch": 1.4928817749023437e-05,
      "model_forward_time": 5.611116886138916,
      "step": 15654
    },
    {
      "epoch": 1.4928817749023437e-05,
      "step": 15654,
      "training_step_time": 17.1050865650177
    },
    {
      "epoch": 1.4929771423339843e-05,
      "model_forward_time": 5.6093292236328125,
      "step": 15655
    },
    {
      "epoch": 1.4929771423339843e-05,
      "step": 15655,
      "training_step_time": 17.089269399642944
    },
    {
      "epoch": 1.493072509765625e-05,
      "model_forward_time": 5.947410583496094,
      "step": 15656
    },
    {
      "epoch": 1.493072509765625e-05,
      "step": 15656,
      "training_step_time": 17.412639379501343
    },
    {
      "epoch": 1.4931678771972656e-05,
      "model_forward_time": 5.616209983825684,
      "step": 15657
    },
    {
      "epoch": 1.4931678771972656e-05,
      "step": 15657,
      "training_step_time": 17.12550187110901
    },
    {
      "epoch": 1.4932632446289062e-05,
      "model_forward_time": 5.623473882675171,
      "step": 15658
    },
    {
      "epoch": 1.4932632446289062e-05,
      "step": 15658,
      "training_step_time": 17.132435083389282
    },
    {
      "epoch": 1.493358612060547e-05,
      "model_forward_time": 5.62026834487915,
      "step": 15659
    },
    {
      "epoch": 1.493358612060547e-05,
      "step": 15659,
      "training_step_time": 17.134575843811035
    },
    {
      "epoch": 1.4934539794921875e-05,
      "grad_norm": 0.5042721033096313,
      "learning_rate": 1e-05,
      "loss": 0.0485,
      "step": 15660
    },
    {
      "epoch": 1.4934539794921875e-05,
      "model_forward_time": 5.617831230163574,
      "step": 15660
    },
    {
      "dynamics_loss_avg": 0.04117172807455063,
      "epoch": 1.4934539794921875e-05,
      "step": 15660
    },
    {
      "action_loss_avg": 0.006305294460617006,
      "epoch": 1.4934539794921875e-05,
      "step": 15660
    },
    {
      "epoch": 1.4934539794921875e-05,
      "step": 15660,
      "training_step_time": 17.129153728485107
    },
    {
      "epoch": 1.4935493469238281e-05,
      "model_forward_time": 5.99127984046936,
      "step": 15661
    },
    {
      "epoch": 1.4935493469238281e-05,
      "step": 15661,
      "training_step_time": 17.488139629364014
    },
    {
      "epoch": 1.4936447143554688e-05,
      "model_forward_time": 5.606335401535034,
      "step": 15662
    },
    {
      "epoch": 1.4936447143554688e-05,
      "step": 15662,
      "training_step_time": 17.073602199554443
    },
    {
      "epoch": 1.4937400817871094e-05,
      "model_forward_time": 5.5956196784973145,
      "step": 15663
    },
    {
      "epoch": 1.4937400817871094e-05,
      "step": 15663,
      "training_step_time": 17.058053493499756
    },
    {
      "epoch": 1.49383544921875e-05,
      "model_forward_time": 5.592719554901123,
      "step": 15664
    },
    {
      "epoch": 1.49383544921875e-05,
      "step": 15664,
      "training_step_time": 17.04330539703369
    },
    {
      "epoch": 1.4939308166503905e-05,
      "model_forward_time": 5.603931665420532,
      "step": 15665
    },
    {
      "epoch": 1.4939308166503905e-05,
      "step": 15665,
      "training_step_time": 17.09497833251953
    },
    {
      "epoch": 1.4940261840820313e-05,
      "model_forward_time": 5.620382070541382,
      "step": 15666
    },
    {
      "epoch": 1.4940261840820313e-05,
      "step": 15666,
      "training_step_time": 17.134246349334717
    },
    {
      "epoch": 1.4941215515136719e-05,
      "model_forward_time": 5.995458602905273,
      "step": 15667
    },
    {
      "epoch": 1.4941215515136719e-05,
      "step": 15667,
      "training_step_time": 17.507671356201172
    },
    {
      "epoch": 1.4942169189453124e-05,
      "model_forward_time": 5.6123175621032715,
      "step": 15668
    },
    {
      "epoch": 1.4942169189453124e-05,
      "step": 15668,
      "training_step_time": 17.110792636871338
    },
    {
      "epoch": 1.4943122863769532e-05,
      "model_forward_time": 5.614626407623291,
      "step": 15669
    },
    {
      "epoch": 1.4943122863769532e-05,
      "step": 15669,
      "training_step_time": 17.103792667388916
    },
    {
      "epoch": 1.4944076538085937e-05,
      "grad_norm": 0.708745539188385,
      "learning_rate": 1e-05,
      "loss": 0.0578,
      "step": 15670
    },
    {
      "epoch": 1.4944076538085937e-05,
      "model_forward_time": 5.594696044921875,
      "step": 15670
    },
    {
      "dynamics_loss_avg": 0.049596007168293,
      "epoch": 1.4944076538085937e-05,
      "step": 15670
    },
    {
      "action_loss_avg": 0.007977719185873866,
      "epoch": 1.4944076538085937e-05,
      "step": 15670
    },
    {
      "epoch": 1.4944076538085937e-05,
      "step": 15670,
      "training_step_time": 17.029566049575806
    },
    {
      "epoch": 1.4945030212402343e-05,
      "model_forward_time": 5.574065923690796,
      "step": 15671
    },
    {
      "epoch": 1.4945030212402343e-05,
      "step": 15671,
      "training_step_time": 16.998703956604004
    },
    {
      "epoch": 1.494598388671875e-05,
      "model_forward_time": 5.977336645126343,
      "step": 15672
    },
    {
      "epoch": 1.494598388671875e-05,
      "step": 15672,
      "training_step_time": 17.418351650238037
    },
    {
      "epoch": 1.4946937561035156e-05,
      "model_forward_time": 5.579558849334717,
      "step": 15673
    },
    {
      "epoch": 1.4946937561035156e-05,
      "step": 15673,
      "training_step_time": 16.99837613105774
    },
    {
      "epoch": 1.4947891235351562e-05,
      "model_forward_time": 5.576918125152588,
      "step": 15674
    },
    {
      "epoch": 1.4947891235351562e-05,
      "step": 15674,
      "training_step_time": 16.998304843902588
    },
    {
      "epoch": 1.494884490966797e-05,
      "model_forward_time": 5.587506294250488,
      "step": 15675
    },
    {
      "epoch": 1.494884490966797e-05,
      "step": 15675,
      "training_step_time": 17.0462908744812
    },
    {
      "epoch": 1.4949798583984375e-05,
      "model_forward_time": 5.602763652801514,
      "step": 15676
    },
    {
      "epoch": 1.4949798583984375e-05,
      "step": 15676,
      "training_step_time": 17.080974102020264
    },
    {
      "epoch": 1.4950752258300781e-05,
      "model_forward_time": 5.610614776611328,
      "step": 15677
    },
    {
      "epoch": 1.4950752258300781e-05,
      "step": 15677,
      "training_step_time": 17.087210178375244
    },
    {
      "epoch": 1.4951705932617188e-05,
      "model_forward_time": 5.959396123886108,
      "step": 15678
    },
    {
      "epoch": 1.4951705932617188e-05,
      "step": 15678,
      "training_step_time": 17.41043519973755
    },
    {
      "epoch": 1.4952659606933594e-05,
      "model_forward_time": 5.60283350944519,
      "step": 15679
    },
    {
      "epoch": 1.4952659606933594e-05,
      "step": 15679,
      "training_step_time": 17.0687358379364
    },
    {
      "epoch": 1.495361328125e-05,
      "grad_norm": 0.9348145723342896,
      "learning_rate": 1e-05,
      "loss": 0.0599,
      "step": 15680
    },
    {
      "epoch": 1.495361328125e-05,
      "model_forward_time": 5.6326069831848145,
      "step": 15680
    },
    {
      "dynamics_loss_avg": 0.05213593207299709,
      "epoch": 1.495361328125e-05,
      "step": 15680
    },
    {
      "action_loss_avg": 0.007960910513065755,
      "epoch": 1.495361328125e-05,
      "step": 15680
    },
    {
      "epoch": 1.495361328125e-05,
      "step": 15680,
      "training_step_time": 17.144551515579224
    },
    {
      "epoch": 1.4954566955566405e-05,
      "model_forward_time": 5.615991830825806,
      "step": 15681
    },
    {
      "epoch": 1.4954566955566405e-05,
      "step": 15681,
      "training_step_time": 17.12493085861206
    },
    {
      "epoch": 1.4955520629882813e-05,
      "model_forward_time": 5.613588809967041,
      "step": 15682
    },
    {
      "epoch": 1.4955520629882813e-05,
      "step": 15682,
      "training_step_time": 17.11293363571167
    },
    {
      "epoch": 1.4956474304199219e-05,
      "model_forward_time": 5.6118323802948,
      "step": 15683
    },
    {
      "epoch": 1.4956474304199219e-05,
      "step": 15683,
      "training_step_time": 17.47461986541748
    },
    {
      "epoch": 1.4957427978515624e-05,
      "model_forward_time": 5.607674598693848,
      "step": 15684
    },
    {
      "epoch": 1.4957427978515624e-05,
      "step": 15684,
      "training_step_time": 17.089678287506104
    },
    {
      "epoch": 1.4958381652832032e-05,
      "model_forward_time": 5.609346151351929,
      "step": 15685
    },
    {
      "epoch": 1.4958381652832032e-05,
      "step": 15685,
      "training_step_time": 17.068406581878662
    },
    {
      "epoch": 1.4959335327148437e-05,
      "model_forward_time": 5.5926783084869385,
      "step": 15686
    },
    {
      "epoch": 1.4959335327148437e-05,
      "step": 15686,
      "training_step_time": 17.04323959350586
    },
    {
      "epoch": 1.4960289001464843e-05,
      "model_forward_time": 5.600250005722046,
      "step": 15687
    },
    {
      "epoch": 1.4960289001464843e-05,
      "step": 15687,
      "training_step_time": 17.04140329360962
    },
    {
      "epoch": 1.496124267578125e-05,
      "model_forward_time": 5.6031365394592285,
      "step": 15688
    },
    {
      "epoch": 1.496124267578125e-05,
      "step": 15688,
      "training_step_time": 17.06666135787964
    },
    {
      "epoch": 1.4962196350097656e-05,
      "model_forward_time": 5.945343255996704,
      "step": 15689
    },
    {
      "epoch": 1.4962196350097656e-05,
      "step": 15689,
      "training_step_time": 17.39600396156311
    },
    {
      "epoch": 1.4963150024414062e-05,
      "grad_norm": 0.733371913433075,
      "learning_rate": 1e-05,
      "loss": 0.0392,
      "step": 15690
    },
    {
      "epoch": 1.4963150024414062e-05,
      "model_forward_time": 5.60703444480896,
      "step": 15690
    },
    {
      "dynamics_loss_avg": 0.03537561511620879,
      "epoch": 1.4963150024414062e-05,
      "step": 15690
    },
    {
      "action_loss_avg": 0.004158487753011286,
      "epoch": 1.4963150024414062e-05,
      "step": 15690
    },
    {
      "epoch": 1.4963150024414062e-05,
      "step": 15690,
      "training_step_time": 17.066710710525513
    },
    {
      "epoch": 1.496410369873047e-05,
      "model_forward_time": 5.606364965438843,
      "step": 15691
    },
    {
      "epoch": 1.496410369873047e-05,
      "step": 15691,
      "training_step_time": 17.083688974380493
    },
    {
      "epoch": 1.4965057373046875e-05,
      "model_forward_time": 5.623138904571533,
      "step": 15692
    },
    {
      "epoch": 1.4965057373046875e-05,
      "step": 15692,
      "training_step_time": 17.1359806060791
    },
    {
      "epoch": 1.4966011047363281e-05,
      "model_forward_time": 5.630365371704102,
      "step": 15693
    },
    {
      "epoch": 1.4966011047363281e-05,
      "step": 15693,
      "training_step_time": 17.14204740524292
    },
    {
      "epoch": 1.4966964721679688e-05,
      "model_forward_time": 6.008541584014893,
      "step": 15694
    },
    {
      "epoch": 1.4966964721679688e-05,
      "step": 15694,
      "training_step_time": 17.4872567653656
    },
    {
      "epoch": 1.4967918395996094e-05,
      "model_forward_time": 5.6101908683776855,
      "step": 15695
    },
    {
      "epoch": 1.4967918395996094e-05,
      "step": 15695,
      "training_step_time": 17.103938341140747
    },
    {
      "epoch": 1.49688720703125e-05,
      "model_forward_time": 5.606357574462891,
      "step": 15696
    },
    {
      "epoch": 1.49688720703125e-05,
      "step": 15696,
      "training_step_time": 17.020294189453125
    },
    {
      "epoch": 1.4969825744628905e-05,
      "model_forward_time": 5.609517812728882,
      "step": 15697
    },
    {
      "epoch": 1.4969825744628905e-05,
      "step": 15697,
      "training_step_time": 17.08340883255005
    },
    {
      "epoch": 1.4970779418945313e-05,
      "model_forward_time": 5.6003687381744385,
      "step": 15698
    },
    {
      "epoch": 1.4970779418945313e-05,
      "step": 15698,
      "training_step_time": 17.06151533126831
    },
    {
      "epoch": 1.4971733093261719e-05,
      "model_forward_time": 5.5961573123931885,
      "step": 15699
    },
    {
      "epoch": 1.4971733093261719e-05,
      "step": 15699,
      "training_step_time": 17.07675313949585
    },
    {
      "epoch": 1.4972686767578124e-05,
      "grad_norm": 1.0398586988449097,
      "learning_rate": 1e-05,
      "loss": 0.0589,
      "step": 15700
    },
    {
      "epoch": 1.4972686767578124e-05,
      "model_forward_time": 5.98371434211731,
      "step": 15700
    },
    {
      "dynamics_loss_avg": 0.0537619836628437,
      "epoch": 1.4972686767578124e-05,
      "step": 15700
    },
    {
      "action_loss_avg": 0.008418137859553098,
      "epoch": 1.4972686767578124e-05,
      "step": 15700
    },
    {
      "epoch": 1.4972686767578124e-05,
      "step": 15700,
      "training_step_time": 17.473350286483765
    },
    {
      "epoch": 1.4973640441894532e-05,
      "model_forward_time": 5.617108583450317,
      "step": 15701
    },
    {
      "epoch": 1.4973640441894532e-05,
      "step": 15701,
      "training_step_time": 17.10700535774231
    },
    {
      "epoch": 1.4974594116210937e-05,
      "model_forward_time": 5.627237558364868,
      "step": 15702
    },
    {
      "epoch": 1.4974594116210937e-05,
      "step": 15702,
      "training_step_time": 17.080292463302612
    },
    {
      "epoch": 1.4975547790527343e-05,
      "model_forward_time": 5.600289344787598,
      "step": 15703
    },
    {
      "epoch": 1.4975547790527343e-05,
      "step": 15703,
      "training_step_time": 17.068356037139893
    },
    {
      "epoch": 1.497650146484375e-05,
      "model_forward_time": 5.5969274044036865,
      "step": 15704
    },
    {
      "epoch": 1.497650146484375e-05,
      "step": 15704,
      "training_step_time": 17.024892330169678
    },
    {
      "epoch": 1.4977455139160156e-05,
      "model_forward_time": 5.9675681591033936,
      "step": 15705
    },
    {
      "epoch": 1.4977455139160156e-05,
      "step": 15705,
      "training_step_time": 17.430046319961548
    },
    {
      "epoch": 1.4978408813476562e-05,
      "model_forward_time": 5.590311288833618,
      "step": 15706
    },
    {
      "epoch": 1.4978408813476562e-05,
      "step": 15706,
      "training_step_time": 17.020609617233276
    },
    {
      "epoch": 1.497936248779297e-05,
      "model_forward_time": 5.592915058135986,
      "step": 15707
    },
    {
      "epoch": 1.497936248779297e-05,
      "step": 15707,
      "training_step_time": 17.052474975585938
    },
    {
      "epoch": 1.4980316162109375e-05,
      "model_forward_time": 5.602138996124268,
      "step": 15708
    },
    {
      "epoch": 1.4980316162109375e-05,
      "step": 15708,
      "training_step_time": 17.077926635742188
    },
    {
      "epoch": 1.4981269836425781e-05,
      "model_forward_time": 5.607480764389038,
      "step": 15709
    },
    {
      "epoch": 1.4981269836425781e-05,
      "step": 15709,
      "training_step_time": 17.083969354629517
    },
    {
      "epoch": 1.4982223510742188e-05,
      "grad_norm": 0.5594856142997742,
      "learning_rate": 1e-05,
      "loss": 0.0545,
      "step": 15710
    },
    {
      "epoch": 1.4982223510742188e-05,
      "model_forward_time": 5.603814125061035,
      "step": 15710
    },
    {
      "dynamics_loss_avg": 0.048879236169159415,
      "epoch": 1.4982223510742188e-05,
      "step": 15710
    },
    {
      "action_loss_avg": 0.005566409835591912,
      "epoch": 1.4982223510742188e-05,
      "step": 15710
    },
    {
      "epoch": 1.4982223510742188e-05,
      "step": 15710,
      "training_step_time": 17.050592184066772
    },
    {
      "epoch": 1.4983177185058594e-05,
      "model_forward_time": 5.95513916015625,
      "step": 15711
    },
    {
      "epoch": 1.4983177185058594e-05,
      "step": 15711,
      "training_step_time": 17.435781955718994
    },
    {
      "epoch": 1.4984130859375e-05,
      "model_forward_time": 5.610823154449463,
      "step": 15712
    },
    {
      "epoch": 1.4984130859375e-05,
      "step": 15712,
      "training_step_time": 17.09786033630371
    },
    {
      "epoch": 1.4985084533691405e-05,
      "model_forward_time": 5.612765789031982,
      "step": 15713
    },
    {
      "epoch": 1.4985084533691405e-05,
      "step": 15713,
      "training_step_time": 17.092408180236816
    },
    {
      "epoch": 1.4986038208007813e-05,
      "model_forward_time": 5.60697603225708,
      "step": 15714
    },
    {
      "epoch": 1.4986038208007813e-05,
      "step": 15714,
      "training_step_time": 17.071436882019043
    },
    {
      "epoch": 1.4986991882324219e-05,
      "model_forward_time": 5.597564697265625,
      "step": 15715
    },
    {
      "epoch": 1.4986991882324219e-05,
      "step": 15715,
      "training_step_time": 17.055908918380737
    },
    {
      "epoch": 1.4987945556640624e-05,
      "model_forward_time": 5.590516567230225,
      "step": 15716
    },
    {
      "epoch": 1.4987945556640624e-05,
      "step": 15716,
      "training_step_time": 17.40673041343689
    },
    {
      "epoch": 1.4988899230957032e-05,
      "model_forward_time": 5.585216522216797,
      "step": 15717
    },
    {
      "epoch": 1.4988899230957032e-05,
      "step": 15717,
      "training_step_time": 17.025521278381348
    },
    {
      "epoch": 1.4989852905273437e-05,
      "model_forward_time": 5.590299129486084,
      "step": 15718
    },
    {
      "epoch": 1.4989852905273437e-05,
      "step": 15718,
      "training_step_time": 17.038023710250854
    },
    {
      "epoch": 1.4990806579589843e-05,
      "model_forward_time": 5.60564923286438,
      "step": 15719
    },
    {
      "epoch": 1.4990806579589843e-05,
      "step": 15719,
      "training_step_time": 17.072972059249878
    },
    {
      "epoch": 1.499176025390625e-05,
      "grad_norm": 0.45875900983810425,
      "learning_rate": 1e-05,
      "loss": 0.0613,
      "step": 15720
    },
    {
      "epoch": 1.499176025390625e-05,
      "model_forward_time": 5.59394907951355,
      "step": 15720
    },
    {
      "dynamics_loss_avg": 0.05198064371943474,
      "epoch": 1.499176025390625e-05,
      "step": 15720
    },
    {
      "action_loss_avg": 0.007033121166750788,
      "epoch": 1.499176025390625e-05,
      "step": 15720
    },
    {
      "epoch": 1.499176025390625e-05,
      "step": 15720,
      "training_step_time": 17.07185459136963
    },
    {
      "epoch": 1.4992713928222656e-05,
      "model_forward_time": 5.60106897354126,
      "step": 15721
    },
    {
      "epoch": 1.4992713928222656e-05,
      "step": 15721,
      "training_step_time": 17.04494857788086
    },
    {
      "epoch": 1.4993667602539062e-05,
      "model_forward_time": 5.9713990688323975,
      "step": 15722
    },
    {
      "epoch": 1.4993667602539062e-05,
      "step": 15722,
      "training_step_time": 17.452702045440674
    },
    {
      "epoch": 1.499462127685547e-05,
      "model_forward_time": 5.612210512161255,
      "step": 15723
    },
    {
      "epoch": 1.499462127685547e-05,
      "step": 15723,
      "training_step_time": 17.088061094284058
    },
    {
      "epoch": 1.4995574951171875e-05,
      "model_forward_time": 5.603576898574829,
      "step": 15724
    },
    {
      "epoch": 1.4995574951171875e-05,
      "step": 15724,
      "training_step_time": 17.06431007385254
    },
    {
      "epoch": 1.4996528625488281e-05,
      "model_forward_time": 5.590782642364502,
      "step": 15725
    },
    {
      "epoch": 1.4996528625488281e-05,
      "step": 15725,
      "training_step_time": 17.043424367904663
    },
    {
      "epoch": 1.4997482299804688e-05,
      "model_forward_time": 5.5831193923950195,
      "step": 15726
    },
    {
      "epoch": 1.4997482299804688e-05,
      "step": 15726,
      "training_step_time": 17.042465925216675
    },
    {
      "epoch": 1.4998435974121094e-05,
      "model_forward_time": 5.98469352722168,
      "step": 15727
    },
    {
      "epoch": 1.4998435974121094e-05,
      "step": 15727,
      "training_step_time": 17.44846510887146
    },
    {
      "epoch": 1.49993896484375e-05,
      "model_forward_time": 5.605276107788086,
      "step": 15728
    },
    {
      "epoch": 1.49993896484375e-05,
      "step": 15728,
      "training_step_time": 17.076178789138794
    },
    {
      "epoch": 1.5000343322753905e-05,
      "model_forward_time": 5.609157562255859,
      "step": 15729
    },
    {
      "epoch": 1.5000343322753905e-05,
      "step": 15729,
      "training_step_time": 17.09267544746399
    },
    {
      "epoch": 1.5001296997070313e-05,
      "grad_norm": 1.0555959939956665,
      "learning_rate": 1e-05,
      "loss": 0.0655,
      "step": 15730
    },
    {
      "epoch": 1.5001296997070313e-05,
      "model_forward_time": 5.61243748664856,
      "step": 15730
    },
    {
      "dynamics_loss_avg": 0.05743605233728886,
      "epoch": 1.5001296997070313e-05,
      "step": 15730
    },
    {
      "action_loss_avg": 0.009777145716361702,
      "epoch": 1.5001296997070313e-05,
      "step": 15730
    },
    {
      "epoch": 1.5001296997070313e-05,
      "step": 15730,
      "training_step_time": 17.097256183624268
    },
    {
      "epoch": 1.5002250671386719e-05,
      "model_forward_time": 5.603586673736572,
      "step": 15731
    },
    {
      "epoch": 1.5002250671386719e-05,
      "step": 15731,
      "training_step_time": 17.051881074905396
    },
    {
      "epoch": 1.5003204345703124e-05,
      "model_forward_time": 5.577679634094238,
      "step": 15732
    },
    {
      "epoch": 1.5003204345703124e-05,
      "step": 15732,
      "training_step_time": 17.017239332199097
    },
    {
      "epoch": 1.5004158020019532e-05,
      "model_forward_time": 5.937361240386963,
      "step": 15733
    },
    {
      "epoch": 1.5004158020019532e-05,
      "step": 15733,
      "training_step_time": 17.38290023803711
    },
    {
      "epoch": 1.5005111694335937e-05,
      "model_forward_time": 5.58478569984436,
      "step": 15734
    },
    {
      "epoch": 1.5005111694335937e-05,
      "step": 15734,
      "training_step_time": 17.038318395614624
    },
    {
      "epoch": 1.5006065368652343e-05,
      "model_forward_time": 5.608281135559082,
      "step": 15735
    },
    {
      "epoch": 1.5006065368652343e-05,
      "step": 15735,
      "training_step_time": 17.103105783462524
    },
    {
      "epoch": 1.500701904296875e-05,
      "model_forward_time": 5.617123365402222,
      "step": 15736
    },
    {
      "epoch": 1.500701904296875e-05,
      "step": 15736,
      "training_step_time": 17.142975568771362
    },
    {
      "epoch": 1.5007972717285156e-05,
      "model_forward_time": 5.631467580795288,
      "step": 15737
    },
    {
      "epoch": 1.5007972717285156e-05,
      "step": 15737,
      "training_step_time": 17.13314437866211
    },
    {
      "epoch": 1.5008926391601562e-05,
      "model_forward_time": 6.010179042816162,
      "step": 15738
    },
    {
      "epoch": 1.5008926391601562e-05,
      "step": 15738,
      "training_step_time": 17.495976209640503
    },
    {
      "epoch": 1.500988006591797e-05,
      "model_forward_time": 5.609648942947388,
      "step": 15739
    },
    {
      "epoch": 1.500988006591797e-05,
      "step": 15739,
      "training_step_time": 17.101811408996582
    },
    {
      "epoch": 1.5010833740234375e-05,
      "grad_norm": 1.036354422569275,
      "learning_rate": 1e-05,
      "loss": 0.0774,
      "step": 15740
    },
    {
      "epoch": 1.5010833740234375e-05,
      "model_forward_time": 5.603068828582764,
      "step": 15740
    },
    {
      "dynamics_loss_avg": 0.06679279748350382,
      "epoch": 1.5010833740234375e-05,
      "step": 15740
    },
    {
      "action_loss_avg": 0.009198404010385275,
      "epoch": 1.5010833740234375e-05,
      "step": 15740
    },
    {
      "epoch": 1.5010833740234375e-05,
      "step": 15740,
      "training_step_time": 17.085744619369507
    },
    {
      "epoch": 1.5011787414550781e-05,
      "model_forward_time": 5.616581916809082,
      "step": 15741
    },
    {
      "epoch": 1.5011787414550781e-05,
      "step": 15741,
      "training_step_time": 17.109069108963013
    },
    {
      "epoch": 1.5012741088867188e-05,
      "model_forward_time": 5.608983039855957,
      "step": 15742
    },
    {
      "epoch": 1.5012741088867188e-05,
      "step": 15742,
      "training_step_time": 17.05268406867981
    },
    {
      "epoch": 1.5013694763183594e-05,
      "model_forward_time": 5.601078033447266,
      "step": 15743
    },
    {
      "epoch": 1.5013694763183594e-05,
      "step": 15743,
      "training_step_time": 17.07020378112793
    },
    {
      "epoch": 1.50146484375e-05,
      "model_forward_time": 5.965206623077393,
      "step": 15744
    },
    {
      "epoch": 1.50146484375e-05,
      "step": 15744,
      "training_step_time": 17.43396544456482
    },
    {
      "epoch": 1.5015602111816406e-05,
      "model_forward_time": 5.621744632720947,
      "step": 15745
    },
    {
      "epoch": 1.5015602111816406e-05,
      "step": 15745,
      "training_step_time": 17.14302158355713
    },
    {
      "epoch": 1.5016555786132813e-05,
      "model_forward_time": 5.627550363540649,
      "step": 15746
    },
    {
      "epoch": 1.5016555786132813e-05,
      "step": 15746,
      "training_step_time": 17.140627145767212
    },
    {
      "epoch": 1.5017509460449219e-05,
      "model_forward_time": 5.6236889362335205,
      "step": 15747
    },
    {
      "epoch": 1.5017509460449219e-05,
      "step": 15747,
      "training_step_time": 17.1338152885437
    },
    {
      "epoch": 1.5018463134765624e-05,
      "model_forward_time": 5.61243462562561,
      "step": 15748
    },
    {
      "epoch": 1.5018463134765624e-05,
      "step": 15748,
      "training_step_time": 17.10943341255188
    },
    {
      "epoch": 1.5019416809082032e-05,
      "model_forward_time": 5.606321811676025,
      "step": 15749
    },
    {
      "epoch": 1.5019416809082032e-05,
      "step": 15749,
      "training_step_time": 17.450371980667114
    },
    {
      "epoch": 1.5020370483398438e-05,
      "grad_norm": 0.7739643454551697,
      "learning_rate": 1e-05,
      "loss": 0.049,
      "step": 15750
    },
    {
      "epoch": 1.5020370483398438e-05,
      "model_forward_time": 5.6193928718566895,
      "step": 15750
    },
    {
      "dynamics_loss_avg": 0.041754201613366605,
      "epoch": 1.5020370483398438e-05,
      "step": 15750
    },
    {
      "action_loss_avg": 0.006883227871730924,
      "epoch": 1.5020370483398438e-05,
      "step": 15750
    },
    {
      "epoch": 1.5020370483398438e-05,
      "step": 15750,
      "training_step_time": 17.11752676963806
    },
    {
      "epoch": 1.5021324157714843e-05,
      "model_forward_time": 5.600797891616821,
      "step": 15751
    },
    {
      "epoch": 1.5021324157714843e-05,
      "step": 15751,
      "training_step_time": 17.063706159591675
    },
    {
      "epoch": 1.502227783203125e-05,
      "model_forward_time": 5.599217176437378,
      "step": 15752
    },
    {
      "epoch": 1.502227783203125e-05,
      "step": 15752,
      "training_step_time": 17.054515838623047
    },
    {
      "epoch": 1.5023231506347656e-05,
      "model_forward_time": 5.5873308181762695,
      "step": 15753
    },
    {
      "epoch": 1.5023231506347656e-05,
      "step": 15753,
      "training_step_time": 17.03881812095642
    },
    {
      "epoch": 1.5024185180664062e-05,
      "model_forward_time": 5.5877392292022705,
      "step": 15754
    },
    {
      "epoch": 1.5024185180664062e-05,
      "step": 15754,
      "training_step_time": 17.045044422149658
    },
    {
      "epoch": 1.502513885498047e-05,
      "model_forward_time": 5.980532884597778,
      "step": 15755
    },
    {
      "epoch": 1.502513885498047e-05,
      "step": 15755,
      "training_step_time": 17.458740234375
    },
    {
      "epoch": 1.5026092529296875e-05,
      "model_forward_time": 5.615900039672852,
      "step": 15756
    },
    {
      "epoch": 1.5026092529296875e-05,
      "step": 15756,
      "training_step_time": 17.122124195098877
    },
    {
      "epoch": 1.5027046203613281e-05,
      "model_forward_time": 5.623940467834473,
      "step": 15757
    },
    {
      "epoch": 1.5027046203613281e-05,
      "step": 15757,
      "training_step_time": 17.142446517944336
    },
    {
      "epoch": 1.5027999877929687e-05,
      "model_forward_time": 5.6261146068573,
      "step": 15758
    },
    {
      "epoch": 1.5027999877929687e-05,
      "step": 15758,
      "training_step_time": 17.113202571868896
    },
    {
      "epoch": 1.5028953552246094e-05,
      "model_forward_time": 5.609496593475342,
      "step": 15759
    },
    {
      "epoch": 1.5028953552246094e-05,
      "step": 15759,
      "training_step_time": 17.096047401428223
    },
    {
      "epoch": 1.50299072265625e-05,
      "grad_norm": 0.2639009654521942,
      "learning_rate": 1e-05,
      "loss": 0.0572,
      "step": 15760
    },
    {
      "epoch": 1.50299072265625e-05,
      "model_forward_time": 5.996410369873047,
      "step": 15760
    },
    {
      "dynamics_loss_avg": 0.049978071451187135,
      "epoch": 1.50299072265625e-05,
      "step": 15760
    },
    {
      "action_loss_avg": 0.006447669374756515,
      "epoch": 1.50299072265625e-05,
      "step": 15760
    },
    {
      "epoch": 1.50299072265625e-05,
      "step": 15760,
      "training_step_time": 17.477096796035767
    },
    {
      "epoch": 1.5030860900878906e-05,
      "model_forward_time": 5.597848176956177,
      "step": 15761
    },
    {
      "epoch": 1.5030860900878906e-05,
      "step": 15761,
      "training_step_time": 17.056331872940063
    },
    {
      "epoch": 1.5031814575195313e-05,
      "model_forward_time": 5.590668439865112,
      "step": 15762
    },
    {
      "epoch": 1.5031814575195313e-05,
      "step": 15762,
      "training_step_time": 17.03709125518799
    },
    {
      "epoch": 1.5032768249511719e-05,
      "model_forward_time": 5.592405080795288,
      "step": 15763
    },
    {
      "epoch": 1.5032768249511719e-05,
      "step": 15763,
      "training_step_time": 17.04404306411743
    },
    {
      "epoch": 1.5033721923828124e-05,
      "model_forward_time": 5.588602304458618,
      "step": 15764
    },
    {
      "epoch": 1.5033721923828124e-05,
      "step": 15764,
      "training_step_time": 17.064083337783813
    },
    {
      "epoch": 1.5034675598144532e-05,
      "model_forward_time": 5.615019798278809,
      "step": 15765
    },
    {
      "epoch": 1.5034675598144532e-05,
      "step": 15765,
      "training_step_time": 17.12077522277832
    },
    {
      "epoch": 1.5035629272460938e-05,
      "model_forward_time": 5.977309465408325,
      "step": 15766
    },
    {
      "epoch": 1.5035629272460938e-05,
      "step": 15766,
      "training_step_time": 17.493733406066895
    },
    {
      "epoch": 1.5036582946777343e-05,
      "model_forward_time": 5.620894908905029,
      "step": 15767
    },
    {
      "epoch": 1.5036582946777343e-05,
      "step": 15767,
      "training_step_time": 17.139286756515503
    },
    {
      "epoch": 1.503753662109375e-05,
      "model_forward_time": 5.618104934692383,
      "step": 15768
    },
    {
      "epoch": 1.503753662109375e-05,
      "step": 15768,
      "training_step_time": 17.125856161117554
    },
    {
      "epoch": 1.5038490295410156e-05,
      "model_forward_time": 5.612063646316528,
      "step": 15769
    },
    {
      "epoch": 1.5038490295410156e-05,
      "step": 15769,
      "training_step_time": 17.092146158218384
    },
    {
      "epoch": 1.5039443969726562e-05,
      "grad_norm": 0.7925341129302979,
      "learning_rate": 1e-05,
      "loss": 0.0713,
      "step": 15770
    },
    {
      "epoch": 1.5039443969726562e-05,
      "model_forward_time": 5.620316028594971,
      "step": 15770
    },
    {
      "dynamics_loss_avg": 0.06653008759021758,
      "epoch": 1.5039443969726562e-05,
      "step": 15770
    },
    {
      "action_loss_avg": 0.010959906131029129,
      "epoch": 1.5039443969726562e-05,
      "step": 15770
    },
    {
      "epoch": 1.5039443969726562e-05,
      "step": 15770,
      "training_step_time": 17.095369815826416
    },
    {
      "epoch": 1.504039764404297e-05,
      "model_forward_time": 5.981675148010254,
      "step": 15771
    },
    {
      "epoch": 1.504039764404297e-05,
      "step": 15771,
      "training_step_time": 17.45115041732788
    },
    {
      "epoch": 1.5041351318359375e-05,
      "model_forward_time": 5.601277112960815,
      "step": 15772
    },
    {
      "epoch": 1.5041351318359375e-05,
      "step": 15772,
      "training_step_time": 17.07079553604126
    },
    {
      "epoch": 1.5042304992675781e-05,
      "model_forward_time": 5.60549521446228,
      "step": 15773
    },
    {
      "epoch": 1.5042304992675781e-05,
      "step": 15773,
      "training_step_time": 17.07662582397461
    },
    {
      "epoch": 1.5043258666992187e-05,
      "model_forward_time": 5.600996255874634,
      "step": 15774
    },
    {
      "epoch": 1.5043258666992187e-05,
      "step": 15774,
      "training_step_time": 17.064269304275513
    },
    {
      "epoch": 1.5044212341308594e-05,
      "model_forward_time": 5.603580474853516,
      "step": 15775
    },
    {
      "epoch": 1.5044212341308594e-05,
      "step": 15775,
      "training_step_time": 17.082646369934082
    },
    {
      "epoch": 1.5045166015625e-05,
      "model_forward_time": 5.6173906326293945,
      "step": 15776
    },
    {
      "epoch": 1.5045166015625e-05,
      "step": 15776,
      "training_step_time": 17.121417760849
    },
    {
      "epoch": 1.5046119689941406e-05,
      "model_forward_time": 5.975462198257446,
      "step": 15777
    },
    {
      "epoch": 1.5046119689941406e-05,
      "step": 15777,
      "training_step_time": 17.469883918762207
    },
    {
      "epoch": 1.5047073364257813e-05,
      "model_forward_time": 5.608755826950073,
      "step": 15778
    },
    {
      "epoch": 1.5047073364257813e-05,
      "step": 15778,
      "training_step_time": 17.100760459899902
    },
    {
      "epoch": 1.5048027038574219e-05,
      "model_forward_time": 5.597251892089844,
      "step": 15779
    },
    {
      "epoch": 1.5048027038574219e-05,
      "step": 15779,
      "training_step_time": 17.047666788101196
    },
    {
      "epoch": 1.5048980712890624e-05,
      "grad_norm": 0.6628607511520386,
      "learning_rate": 1e-05,
      "loss": 0.0664,
      "step": 15780
    },
    {
      "epoch": 1.5048980712890624e-05,
      "model_forward_time": 5.595024108886719,
      "step": 15780
    },
    {
      "dynamics_loss_avg": 0.05056588454172015,
      "epoch": 1.5048980712890624e-05,
      "step": 15780
    },
    {
      "action_loss_avg": 0.006669100397266448,
      "epoch": 1.5048980712890624e-05,
      "step": 15780
    },
    {
      "epoch": 1.5048980712890624e-05,
      "step": 15780,
      "training_step_time": 17.027604818344116
    },
    {
      "epoch": 1.5049934387207032e-05,
      "model_forward_time": 5.578266382217407,
      "step": 15781
    },
    {
      "epoch": 1.5049934387207032e-05,
      "step": 15781,
      "training_step_time": 17.008223056793213
    },
    {
      "epoch": 1.5050888061523438e-05,
      "model_forward_time": 5.59283447265625,
      "step": 15782
    },
    {
      "epoch": 1.5050888061523438e-05,
      "step": 15782,
      "training_step_time": 17.397855043411255
    },
    {
      "epoch": 1.5051841735839843e-05,
      "model_forward_time": 5.588702201843262,
      "step": 15783
    },
    {
      "epoch": 1.5051841735839843e-05,
      "step": 15783,
      "training_step_time": 17.05862045288086
    },
    {
      "epoch": 1.505279541015625e-05,
      "model_forward_time": 5.597409248352051,
      "step": 15784
    },
    {
      "epoch": 1.505279541015625e-05,
      "step": 15784,
      "training_step_time": 17.1119384765625
    },
    {
      "epoch": 1.5053749084472656e-05,
      "model_forward_time": 5.6181800365448,
      "step": 15785
    },
    {
      "epoch": 1.5053749084472656e-05,
      "step": 15785,
      "training_step_time": 17.11966347694397
    },
    {
      "epoch": 1.5054702758789062e-05,
      "model_forward_time": 5.6125102043151855,
      "step": 15786
    },
    {
      "epoch": 1.5054702758789062e-05,
      "step": 15786,
      "training_step_time": 17.10281991958618
    },
    {
      "epoch": 1.505565643310547e-05,
      "model_forward_time": 5.59222149848938,
      "step": 15787
    },
    {
      "epoch": 1.505565643310547e-05,
      "step": 15787,
      "training_step_time": 17.011536359786987
    },
    {
      "epoch": 1.5056610107421875e-05,
      "model_forward_time": 5.930798053741455,
      "step": 15788
    },
    {
      "epoch": 1.5056610107421875e-05,
      "step": 15788,
      "training_step_time": 17.336519479751587
    },
    {
      "epoch": 1.5057563781738281e-05,
      "model_forward_time": 5.572263240814209,
      "step": 15789
    },
    {
      "epoch": 1.5057563781738281e-05,
      "step": 15789,
      "training_step_time": 17.013116121292114
    },
    {
      "epoch": 1.5058517456054687e-05,
      "grad_norm": 0.658757746219635,
      "learning_rate": 1e-05,
      "loss": 0.0504,
      "step": 15790
    },
    {
      "epoch": 1.5058517456054687e-05,
      "model_forward_time": 5.5682268142700195,
      "step": 15790
    },
    {
      "dynamics_loss_avg": 0.04946814887225628,
      "epoch": 1.5058517456054687e-05,
      "step": 15790
    },
    {
      "action_loss_avg": 0.005677260598167777,
      "epoch": 1.5058517456054687e-05,
      "step": 15790
    },
    {
      "epoch": 1.5058517456054687e-05,
      "step": 15790,
      "training_step_time": 17.004658460617065
    },
    {
      "epoch": 1.5059471130371094e-05,
      "model_forward_time": 5.573027849197388,
      "step": 15791
    },
    {
      "epoch": 1.5059471130371094e-05,
      "step": 15791,
      "training_step_time": 16.994951248168945
    },
    {
      "epoch": 1.50604248046875e-05,
      "model_forward_time": 5.5732197761535645,
      "step": 15792
    },
    {
      "epoch": 1.50604248046875e-05,
      "step": 15792,
      "training_step_time": 16.99497628211975
    },
    {
      "epoch": 1.5061378479003906e-05,
      "model_forward_time": 5.960516452789307,
      "step": 15793
    },
    {
      "epoch": 1.5061378479003906e-05,
      "step": 15793,
      "training_step_time": 17.38610005378723
    },
    {
      "epoch": 1.5062332153320313e-05,
      "model_forward_time": 5.584432125091553,
      "step": 15794
    },
    {
      "epoch": 1.5062332153320313e-05,
      "step": 15794,
      "training_step_time": 16.996745347976685
    },
    {
      "epoch": 1.5063285827636719e-05,
      "model_forward_time": 5.600279092788696,
      "step": 15795
    },
    {
      "epoch": 1.5063285827636719e-05,
      "step": 15795,
      "training_step_time": 17.072659969329834
    },
    {
      "epoch": 1.5064239501953124e-05,
      "model_forward_time": 5.609983921051025,
      "step": 15796
    },
    {
      "epoch": 1.5064239501953124e-05,
      "step": 15796,
      "training_step_time": 17.116271257400513
    },
    {
      "epoch": 1.5065193176269532e-05,
      "model_forward_time": 5.648089408874512,
      "step": 15797
    },
    {
      "epoch": 1.5065193176269532e-05,
      "step": 15797,
      "training_step_time": 17.126935243606567
    },
    {
      "epoch": 1.5066146850585938e-05,
      "model_forward_time": 5.601428747177124,
      "step": 15798
    },
    {
      "epoch": 1.5066146850585938e-05,
      "step": 15798,
      "training_step_time": 17.060874462127686
    },
    {
      "epoch": 1.5067100524902343e-05,
      "model_forward_time": 5.948362350463867,
      "step": 15799
    },
    {
      "epoch": 1.5067100524902343e-05,
      "step": 15799,
      "training_step_time": 17.36239004135132
    },
    {
      "epoch": 1.506805419921875e-05,
      "grad_norm": 0.9786838889122009,
      "learning_rate": 1e-05,
      "loss": 0.0625,
      "step": 15800
    },
    {
      "epoch": 1.506805419921875e-05,
      "model_forward_time": 5.588603734970093,
      "step": 15800
    },
    {
      "dynamics_loss_avg": 0.05769082643091679,
      "epoch": 1.506805419921875e-05,
      "step": 15800
    },
    {
      "action_loss_avg": 0.00771327530965209,
      "epoch": 1.506805419921875e-05,
      "step": 15800
    },
    {
      "epoch": 1.506805419921875e-05,
      "step": 15800,
      "training_step_time": 16.99980926513672
    },
    {
      "epoch": 1.5069007873535156e-05,
      "model_forward_time": 5.572769641876221,
      "step": 15801
    },
    {
      "epoch": 1.5069007873535156e-05,
      "step": 15801,
      "training_step_time": 17.02288818359375
    },
    {
      "epoch": 1.5069961547851562e-05,
      "model_forward_time": 5.58915114402771,
      "step": 15802
    },
    {
      "epoch": 1.5069961547851562e-05,
      "step": 15802,
      "training_step_time": 17.01298475265503
    },
    {
      "epoch": 1.507091522216797e-05,
      "model_forward_time": 5.585963487625122,
      "step": 15803
    },
    {
      "epoch": 1.507091522216797e-05,
      "step": 15803,
      "training_step_time": 17.019774436950684
    },
    {
      "epoch": 1.5071868896484375e-05,
      "model_forward_time": 5.985489130020142,
      "step": 15804
    },
    {
      "epoch": 1.5071868896484375e-05,
      "step": 15804,
      "training_step_time": 17.45016121864319
    },
    {
      "epoch": 1.5072822570800781e-05,
      "model_forward_time": 5.600446701049805,
      "step": 15805
    },
    {
      "epoch": 1.5072822570800781e-05,
      "step": 15805,
      "training_step_time": 17.07559370994568
    },
    {
      "epoch": 1.5073776245117187e-05,
      "model_forward_time": 5.6055991649627686,
      "step": 15806
    },
    {
      "epoch": 1.5073776245117187e-05,
      "step": 15806,
      "training_step_time": 17.081274271011353
    },
    {
      "epoch": 1.5074729919433594e-05,
      "model_forward_time": 5.607479810714722,
      "step": 15807
    },
    {
      "epoch": 1.5074729919433594e-05,
      "step": 15807,
      "training_step_time": 17.074246644973755
    },
    {
      "epoch": 1.507568359375e-05,
      "model_forward_time": 5.59882116317749,
      "step": 15808
    },
    {
      "epoch": 1.507568359375e-05,
      "step": 15808,
      "training_step_time": 17.04269313812256
    },
    {
      "epoch": 1.5076637268066406e-05,
      "model_forward_time": 5.592451572418213,
      "step": 15809
    },
    {
      "epoch": 1.5076637268066406e-05,
      "step": 15809,
      "training_step_time": 17.070549726486206
    },
    {
      "epoch": 1.5077590942382813e-05,
      "grad_norm": 0.8088247776031494,
      "learning_rate": 1e-05,
      "loss": 0.0657,
      "step": 15810
    },
    {
      "epoch": 1.5077590942382813e-05,
      "model_forward_time": 5.958871126174927,
      "step": 15810
    },
    {
      "dynamics_loss_avg": 0.0549273032695055,
      "epoch": 1.5077590942382813e-05,
      "step": 15810
    },
    {
      "action_loss_avg": 0.006330667203292251,
      "epoch": 1.5077590942382813e-05,
      "step": 15810
    },
    {
      "epoch": 1.5077590942382813e-05,
      "step": 15810,
      "training_step_time": 17.438241243362427
    },
    {
      "epoch": 1.5078544616699219e-05,
      "model_forward_time": 5.612458944320679,
      "step": 15811
    },
    {
      "epoch": 1.5078544616699219e-05,
      "step": 15811,
      "training_step_time": 17.13735318183899
    },
    {
      "epoch": 1.5079498291015624e-05,
      "model_forward_time": 5.622199535369873,
      "step": 15812
    },
    {
      "epoch": 1.5079498291015624e-05,
      "step": 15812,
      "training_step_time": 17.129687309265137
    },
    {
      "epoch": 1.5080451965332032e-05,
      "model_forward_time": 5.609327793121338,
      "step": 15813
    },
    {
      "epoch": 1.5080451965332032e-05,
      "step": 15813,
      "training_step_time": 17.11209726333618
    },
    {
      "epoch": 1.5081405639648438e-05,
      "model_forward_time": 5.609468221664429,
      "step": 15814
    },
    {
      "epoch": 1.5081405639648438e-05,
      "step": 15814,
      "training_step_time": 17.108565092086792
    },
    {
      "epoch": 1.5082359313964843e-05,
      "model_forward_time": 5.621548414230347,
      "step": 15815
    },
    {
      "epoch": 1.5082359313964843e-05,
      "step": 15815,
      "training_step_time": 17.49179196357727
    },
    {
      "epoch": 1.508331298828125e-05,
      "model_forward_time": 5.6232829093933105,
      "step": 15816
    },
    {
      "epoch": 1.508331298828125e-05,
      "step": 15816,
      "training_step_time": 17.09879207611084
    },
    {
      "epoch": 1.5084266662597656e-05,
      "model_forward_time": 5.5911431312561035,
      "step": 15817
    },
    {
      "epoch": 1.5084266662597656e-05,
      "step": 15817,
      "training_step_time": 17.04320216178894
    },
    {
      "epoch": 1.5085220336914062e-05,
      "model_forward_time": 5.580111503601074,
      "step": 15818
    },
    {
      "epoch": 1.5085220336914062e-05,
      "step": 15818,
      "training_step_time": 16.991364240646362
    },
    {
      "epoch": 1.508617401123047e-05,
      "model_forward_time": 5.570868492126465,
      "step": 15819
    },
    {
      "epoch": 1.508617401123047e-05,
      "step": 15819,
      "training_step_time": 16.98132348060608
    },
    {
      "epoch": 1.5087127685546875e-05,
      "grad_norm": 1.6997865438461304,
      "learning_rate": 1e-05,
      "loss": 0.0562,
      "step": 15820
    },
    {
      "epoch": 1.5087127685546875e-05,
      "model_forward_time": 5.57474160194397,
      "step": 15820
    },
    {
      "dynamics_loss_avg": 0.04737854115664959,
      "epoch": 1.5087127685546875e-05,
      "step": 15820
    },
    {
      "action_loss_avg": 0.0069981310050934555,
      "epoch": 1.5087127685546875e-05,
      "step": 15820
    },
    {
      "epoch": 1.5087127685546875e-05,
      "step": 15820,
      "training_step_time": 17.02053475379944
    },
    {
      "epoch": 1.5088081359863281e-05,
      "model_forward_time": 5.936713218688965,
      "step": 15821
    },
    {
      "epoch": 1.5088081359863281e-05,
      "step": 15821,
      "training_step_time": 17.34448480606079
    },
    {
      "epoch": 1.5089035034179687e-05,
      "model_forward_time": 5.578722238540649,
      "step": 15822
    },
    {
      "epoch": 1.5089035034179687e-05,
      "step": 15822,
      "training_step_time": 16.996564149856567
    },
    {
      "epoch": 1.5089988708496094e-05,
      "model_forward_time": 5.5814714431762695,
      "step": 15823
    },
    {
      "epoch": 1.5089988708496094e-05,
      "step": 15823,
      "training_step_time": 16.995875120162964
    },
    {
      "epoch": 1.50909423828125e-05,
      "model_forward_time": 5.578073263168335,
      "step": 15824
    },
    {
      "epoch": 1.50909423828125e-05,
      "step": 15824,
      "training_step_time": 17.036670446395874
    },
    {
      "epoch": 1.5091896057128906e-05,
      "model_forward_time": 5.592885732650757,
      "step": 15825
    },
    {
      "epoch": 1.5091896057128906e-05,
      "step": 15825,
      "training_step_time": 17.05653715133667
    },
    {
      "epoch": 1.5092849731445313e-05,
      "model_forward_time": 5.981198310852051,
      "step": 15826
    },
    {
      "epoch": 1.5092849731445313e-05,
      "step": 15826,
      "training_step_time": 17.451911687850952
    },
    {
      "epoch": 1.5093803405761719e-05,
      "model_forward_time": 5.609463453292847,
      "step": 15827
    },
    {
      "epoch": 1.5093803405761719e-05,
      "step": 15827,
      "training_step_time": 17.08675742149353
    },
    {
      "epoch": 1.5094757080078124e-05,
      "model_forward_time": 5.601208925247192,
      "step": 15828
    },
    {
      "epoch": 1.5094757080078124e-05,
      "step": 15828,
      "training_step_time": 17.064953088760376
    },
    {
      "epoch": 1.5095710754394532e-05,
      "model_forward_time": 5.606597900390625,
      "step": 15829
    },
    {
      "epoch": 1.5095710754394532e-05,
      "step": 15829,
      "training_step_time": 17.083279371261597
    },
    {
      "epoch": 1.5096664428710938e-05,
      "grad_norm": 0.7976444363594055,
      "learning_rate": 1e-05,
      "loss": 0.0782,
      "step": 15830
    },
    {
      "epoch": 1.5096664428710938e-05,
      "model_forward_time": 5.609560012817383,
      "step": 15830
    },
    {
      "dynamics_loss_avg": 0.06932523921132087,
      "epoch": 1.5096664428710938e-05,
      "step": 15830
    },
    {
      "action_loss_avg": 0.010545052704401314,
      "epoch": 1.5096664428710938e-05,
      "step": 15830
    },
    {
      "epoch": 1.5096664428710938e-05,
      "step": 15830,
      "training_step_time": 17.082648277282715
    },
    {
      "epoch": 1.5097618103027343e-05,
      "model_forward_time": 5.615228891372681,
      "step": 15831
    },
    {
      "epoch": 1.5097618103027343e-05,
      "step": 15831,
      "training_step_time": 17.0731999874115
    },
    {
      "epoch": 1.509857177734375e-05,
      "model_forward_time": 5.951815366744995,
      "step": 15832
    },
    {
      "epoch": 1.509857177734375e-05,
      "step": 15832,
      "training_step_time": 17.397645473480225
    },
    {
      "epoch": 1.5099525451660156e-05,
      "model_forward_time": 5.603038787841797,
      "step": 15833
    },
    {
      "epoch": 1.5099525451660156e-05,
      "step": 15833,
      "training_step_time": 17.102442502975464
    },
    {
      "epoch": 1.5100479125976562e-05,
      "model_forward_time": 5.62755823135376,
      "step": 15834
    },
    {
      "epoch": 1.5100479125976562e-05,
      "step": 15834,
      "training_step_time": 17.146355390548706
    },
    {
      "epoch": 1.510143280029297e-05,
      "model_forward_time": 5.620586395263672,
      "step": 15835
    },
    {
      "epoch": 1.510143280029297e-05,
      "step": 15835,
      "training_step_time": 17.0999972820282
    },
    {
      "epoch": 1.5102386474609375e-05,
      "model_forward_time": 5.592942714691162,
      "step": 15836
    },
    {
      "epoch": 1.5102386474609375e-05,
      "step": 15836,
      "training_step_time": 17.048248291015625
    },
    {
      "epoch": 1.5103340148925781e-05,
      "model_forward_time": 5.976542949676514,
      "step": 15837
    },
    {
      "epoch": 1.5103340148925781e-05,
      "step": 15837,
      "training_step_time": 17.418686866760254
    },
    {
      "epoch": 1.5104293823242187e-05,
      "model_forward_time": 5.603544235229492,
      "step": 15838
    },
    {
      "epoch": 1.5104293823242187e-05,
      "step": 15838,
      "training_step_time": 17.072479963302612
    },
    {
      "epoch": 1.5105247497558594e-05,
      "model_forward_time": 5.604989767074585,
      "step": 15839
    },
    {
      "epoch": 1.5105247497558594e-05,
      "step": 15839,
      "training_step_time": 17.075438022613525
    },
    {
      "epoch": 1.5106201171875e-05,
      "grad_norm": 0.31502971053123474,
      "learning_rate": 1e-05,
      "loss": 0.0614,
      "step": 15840
    },
    {
      "epoch": 1.5106201171875e-05,
      "model_forward_time": 5.602042198181152,
      "step": 15840
    },
    {
      "dynamics_loss_avg": 0.0571563521400094,
      "epoch": 1.5106201171875e-05,
      "step": 15840
    },
    {
      "action_loss_avg": 0.007017273572273552,
      "epoch": 1.5106201171875e-05,
      "step": 15840
    },
    {
      "epoch": 1.5106201171875e-05,
      "step": 15840,
      "training_step_time": 17.051273584365845
    },
    {
      "epoch": 1.5107154846191406e-05,
      "model_forward_time": 5.596336364746094,
      "step": 15841
    },
    {
      "epoch": 1.5107154846191406e-05,
      "step": 15841,
      "training_step_time": 17.050950288772583
    },
    {
      "epoch": 1.5108108520507813e-05,
      "model_forward_time": 5.595046281814575,
      "step": 15842
    },
    {
      "epoch": 1.5108108520507813e-05,
      "step": 15842,
      "training_step_time": 17.037660121917725
    },
    {
      "epoch": 1.5109062194824219e-05,
      "model_forward_time": 5.9330503940582275,
      "step": 15843
    },
    {
      "epoch": 1.5109062194824219e-05,
      "step": 15843,
      "training_step_time": 17.388524532318115
    },
    {
      "epoch": 1.5110015869140624e-05,
      "model_forward_time": 5.595983028411865,
      "step": 15844
    },
    {
      "epoch": 1.5110015869140624e-05,
      "step": 15844,
      "training_step_time": 17.07334613800049
    },
    {
      "epoch": 1.5110969543457032e-05,
      "model_forward_time": 5.597431898117065,
      "step": 15845
    },
    {
      "epoch": 1.5110969543457032e-05,
      "step": 15845,
      "training_step_time": 17.057163953781128
    },
    {
      "epoch": 1.5111923217773438e-05,
      "model_forward_time": 5.599032878875732,
      "step": 15846
    },
    {
      "epoch": 1.5111923217773438e-05,
      "step": 15846,
      "training_step_time": 17.04025435447693
    },
    {
      "epoch": 1.5112876892089843e-05,
      "model_forward_time": 5.587472915649414,
      "step": 15847
    },
    {
      "epoch": 1.5112876892089843e-05,
      "step": 15847,
      "training_step_time": 17.043888807296753
    },
    {
      "epoch": 1.511383056640625e-05,
      "model_forward_time": 5.6009180545806885,
      "step": 15848
    },
    {
      "epoch": 1.511383056640625e-05,
      "step": 15848,
      "training_step_time": 17.443979024887085
    },
    {
      "epoch": 1.5114784240722656e-05,
      "model_forward_time": 5.606543064117432,
      "step": 15849
    },
    {
      "epoch": 1.5114784240722656e-05,
      "step": 15849,
      "training_step_time": 17.092321634292603
    },
    {
      "epoch": 1.5115737915039062e-05,
      "grad_norm": 0.7472816109657288,
      "learning_rate": 1e-05,
      "loss": 0.0604,
      "step": 15850
    },
    {
      "epoch": 1.5115737915039062e-05,
      "model_forward_time": 5.609112501144409,
      "step": 15850
    },
    {
      "dynamics_loss_avg": 0.053091739676892756,
      "epoch": 1.5115737915039062e-05,
      "step": 15850
    },
    {
      "action_loss_avg": 0.008394762920215727,
      "epoch": 1.5115737915039062e-05,
      "step": 15850
    },
    {
      "epoch": 1.5115737915039062e-05,
      "step": 15850,
      "training_step_time": 17.087333917617798
    },
    {
      "epoch": 1.511669158935547e-05,
      "model_forward_time": 5.614134788513184,
      "step": 15851
    },
    {
      "epoch": 1.511669158935547e-05,
      "step": 15851,
      "training_step_time": 17.09576940536499
    },
    {
      "epoch": 1.5117645263671875e-05,
      "model_forward_time": 5.605065107345581,
      "step": 15852
    },
    {
      "epoch": 1.5117645263671875e-05,
      "step": 15852,
      "training_step_time": 17.101709365844727
    },
    {
      "epoch": 1.5118598937988281e-05,
      "model_forward_time": 5.614908933639526,
      "step": 15853
    },
    {
      "epoch": 1.5118598937988281e-05,
      "step": 15853,
      "training_step_time": 17.102292776107788
    },
    {
      "epoch": 1.5119552612304687e-05,
      "model_forward_time": 5.980125665664673,
      "step": 15854
    },
    {
      "epoch": 1.5119552612304687e-05,
      "step": 15854,
      "training_step_time": 17.46214771270752
    },
    {
      "epoch": 1.5120506286621094e-05,
      "model_forward_time": 5.604079484939575,
      "step": 15855
    },
    {
      "epoch": 1.5120506286621094e-05,
      "step": 15855,
      "training_step_time": 17.08894181251526
    },
    {
      "epoch": 1.51214599609375e-05,
      "model_forward_time": 5.612932205200195,
      "step": 15856
    },
    {
      "epoch": 1.51214599609375e-05,
      "step": 15856,
      "training_step_time": 17.112130165100098
    },
    {
      "epoch": 1.5122413635253906e-05,
      "model_forward_time": 5.600012540817261,
      "step": 15857
    },
    {
      "epoch": 1.5122413635253906e-05,
      "step": 15857,
      "training_step_time": 17.08633279800415
    },
    {
      "epoch": 1.5123367309570313e-05,
      "model_forward_time": 5.584674596786499,
      "step": 15858
    },
    {
      "epoch": 1.5123367309570313e-05,
      "step": 15858,
      "training_step_time": 17.02192711830139
    },
    {
      "epoch": 1.5124320983886719e-05,
      "model_forward_time": 5.964063882827759,
      "step": 15859
    },
    {
      "epoch": 1.5124320983886719e-05,
      "step": 15859,
      "training_step_time": 17.3899028301239
    },
    {
      "epoch": 1.5125274658203124e-05,
      "grad_norm": 0.8699634075164795,
      "learning_rate": 1e-05,
      "loss": 0.069,
      "step": 15860
    },
    {
      "epoch": 1.5125274658203124e-05,
      "model_forward_time": 5.592776536941528,
      "step": 15860
    },
    {
      "dynamics_loss_avg": 0.05679243057966232,
      "epoch": 1.5125274658203124e-05,
      "step": 15860
    },
    {
      "action_loss_avg": 0.010040988423861563,
      "epoch": 1.5125274658203124e-05,
      "step": 15860
    },
    {
      "epoch": 1.5125274658203124e-05,
      "step": 15860,
      "training_step_time": 17.00754141807556
    },
    {
      "epoch": 1.5126228332519532e-05,
      "model_forward_time": 5.5719895362854,
      "step": 15861
    },
    {
      "epoch": 1.5126228332519532e-05,
      "step": 15861,
      "training_step_time": 17.008066177368164
    },
    {
      "epoch": 1.5127182006835938e-05,
      "model_forward_time": 5.575281620025635,
      "step": 15862
    },
    {
      "epoch": 1.5127182006835938e-05,
      "step": 15862,
      "training_step_time": 17.021392107009888
    },
    {
      "epoch": 1.5128135681152343e-05,
      "model_forward_time": 5.596859693527222,
      "step": 15863
    },
    {
      "epoch": 1.5128135681152343e-05,
      "step": 15863,
      "training_step_time": 17.032851696014404
    },
    {
      "epoch": 1.512908935546875e-05,
      "model_forward_time": 5.589856863021851,
      "step": 15864
    },
    {
      "epoch": 1.512908935546875e-05,
      "step": 15864,
      "training_step_time": 17.059044361114502
    },
    {
      "epoch": 1.5130043029785156e-05,
      "model_forward_time": 5.950137138366699,
      "step": 15865
    },
    {
      "epoch": 1.5130043029785156e-05,
      "step": 15865,
      "training_step_time": 17.439753532409668
    },
    {
      "epoch": 1.5130996704101562e-05,
      "model_forward_time": 5.615489959716797,
      "step": 15866
    },
    {
      "epoch": 1.5130996704101562e-05,
      "step": 15866,
      "training_step_time": 17.10533905029297
    },
    {
      "epoch": 1.513195037841797e-05,
      "model_forward_time": 5.613274812698364,
      "step": 15867
    },
    {
      "epoch": 1.513195037841797e-05,
      "step": 15867,
      "training_step_time": 17.128854036331177
    },
    {
      "epoch": 1.5132904052734375e-05,
      "model_forward_time": 5.626295328140259,
      "step": 15868
    },
    {
      "epoch": 1.5132904052734375e-05,
      "step": 15868,
      "training_step_time": 17.119751930236816
    },
    {
      "epoch": 1.5133857727050781e-05,
      "model_forward_time": 5.591198205947876,
      "step": 15869
    },
    {
      "epoch": 1.5133857727050781e-05,
      "step": 15869,
      "training_step_time": 17.057750940322876
    },
    {
      "epoch": 1.5134811401367187e-05,
      "grad_norm": 0.9758484363555908,
      "learning_rate": 1e-05,
      "loss": 0.0712,
      "step": 15870
    },
    {
      "epoch": 1.5134811401367187e-05,
      "model_forward_time": 5.965010404586792,
      "step": 15870
    },
    {
      "dynamics_loss_avg": 0.061218945868313314,
      "epoch": 1.5134811401367187e-05,
      "step": 15870
    },
    {
      "action_loss_avg": 0.0076624209526926276,
      "epoch": 1.5134811401367187e-05,
      "step": 15870
    },
    {
      "epoch": 1.5134811401367187e-05,
      "step": 15870,
      "training_step_time": 17.389767169952393
    },
    {
      "epoch": 1.5135765075683594e-05,
      "model_forward_time": 5.592175722122192,
      "step": 15871
    },
    {
      "epoch": 1.5135765075683594e-05,
      "step": 15871,
      "training_step_time": 17.04346513748169
    },
    {
      "epoch": 1.513671875e-05,
      "model_forward_time": 5.609715461730957,
      "step": 15872
    },
    {
      "epoch": 1.513671875e-05,
      "step": 15872,
      "training_step_time": 17.075973510742188
    },
    {
      "epoch": 1.5137672424316406e-05,
      "model_forward_time": 5.6019203662872314,
      "step": 15873
    },
    {
      "epoch": 1.5137672424316406e-05,
      "step": 15873,
      "training_step_time": 17.071011781692505
    },
    {
      "epoch": 1.5138626098632813e-05,
      "model_forward_time": 5.617446660995483,
      "step": 15874
    },
    {
      "epoch": 1.5138626098632813e-05,
      "step": 15874,
      "training_step_time": 17.018240213394165
    },
    {
      "epoch": 1.5139579772949219e-05,
      "model_forward_time": 5.6265363693237305,
      "step": 15875
    },
    {
      "epoch": 1.5139579772949219e-05,
      "step": 15875,
      "training_step_time": 17.129003763198853
    },
    {
      "epoch": 1.5140533447265624e-05,
      "model_forward_time": 5.959840536117554,
      "step": 15876
    },
    {
      "epoch": 1.5140533447265624e-05,
      "step": 15876,
      "training_step_time": 17.427291870117188
    },
    {
      "epoch": 1.5141487121582032e-05,
      "model_forward_time": 5.600078105926514,
      "step": 15877
    },
    {
      "epoch": 1.5141487121582032e-05,
      "step": 15877,
      "training_step_time": 17.056103706359863
    },
    {
      "epoch": 1.5142440795898438e-05,
      "model_forward_time": 5.642653703689575,
      "step": 15878
    },
    {
      "epoch": 1.5142440795898438e-05,
      "step": 15878,
      "training_step_time": 17.14113736152649
    },
    {
      "epoch": 1.5143394470214843e-05,
      "model_forward_time": 4.288058280944824,
      "step": 15879
    },
    {
      "epoch": 1.5143394470214843e-05,
      "step": 15879,
      "training_step_time": 12.846406936645508
    },
    {
      "epoch": 1.514434814453125e-05,
      "grad_norm": 0.8458303213119507,
      "learning_rate": 1e-05,
      "loss": 0.0566,
      "step": 15880
    },
    {
      "epoch": 1.514434814453125e-05,
      "model_forward_time": 5.622551679611206,
      "step": 15880
    },
    {
      "dynamics_loss_avg": 0.05315524749457836,
      "epoch": 1.514434814453125e-05,
      "step": 15880
    },
    {
      "action_loss_avg": 0.00698144193738699,
      "epoch": 1.514434814453125e-05,
      "step": 15880
    },
    {
      "epoch": 1.514434814453125e-05,
      "step": 15880,
      "training_step_time": 17.12786030769348
    },
    {
      "epoch": 1.5145301818847656e-05,
      "model_forward_time": 5.615413188934326,
      "step": 15881
    },
    {
      "epoch": 1.5145301818847656e-05,
      "step": 15881,
      "training_step_time": 17.107216119766235
    },
    {
      "epoch": 1.5146255493164062e-05,
      "model_forward_time": 5.980920314788818,
      "step": 15882
    },
    {
      "epoch": 1.5146255493164062e-05,
      "step": 15882,
      "training_step_time": 17.47146773338318
    },
    {
      "epoch": 1.5147209167480468e-05,
      "model_forward_time": 5.6127283573150635,
      "step": 15883
    },
    {
      "epoch": 1.5147209167480468e-05,
      "step": 15883,
      "training_step_time": 17.111701011657715
    },
    {
      "epoch": 1.5148162841796875e-05,
      "model_forward_time": 5.61227822303772,
      "step": 15884
    },
    {
      "epoch": 1.5148162841796875e-05,
      "step": 15884,
      "training_step_time": 17.11106252670288
    },
    {
      "epoch": 1.5149116516113281e-05,
      "model_forward_time": 5.616726875305176,
      "step": 15885
    },
    {
      "epoch": 1.5149116516113281e-05,
      "step": 15885,
      "training_step_time": 17.079187631607056
    },
    {
      "epoch": 1.5150070190429687e-05,
      "model_forward_time": 5.605811595916748,
      "step": 15886
    },
    {
      "epoch": 1.5150070190429687e-05,
      "step": 15886,
      "training_step_time": 17.028130531311035
    },
    {
      "epoch": 1.5151023864746094e-05,
      "model_forward_time": 5.938485145568848,
      "step": 15887
    },
    {
      "epoch": 1.5151023864746094e-05,
      "step": 15887,
      "training_step_time": 17.352426290512085
    },
    {
      "epoch": 1.51519775390625e-05,
      "model_forward_time": 5.588647365570068,
      "step": 15888
    },
    {
      "epoch": 1.51519775390625e-05,
      "step": 15888,
      "training_step_time": 17.033500909805298
    },
    {
      "epoch": 1.5152931213378906e-05,
      "model_forward_time": 5.580587387084961,
      "step": 15889
    },
    {
      "epoch": 1.5152931213378906e-05,
      "step": 15889,
      "training_step_time": 17.014378309249878
    },
    {
      "epoch": 1.5153884887695313e-05,
      "grad_norm": 0.9353124499320984,
      "learning_rate": 1e-05,
      "loss": 0.0622,
      "step": 15890
    },
    {
      "epoch": 1.5153884887695313e-05,
      "model_forward_time": 5.585438966751099,
      "step": 15890
    },
    {
      "dynamics_loss_avg": 0.05270132385194302,
      "epoch": 1.5153884887695313e-05,
      "step": 15890
    },
    {
      "action_loss_avg": 0.007343041140120477,
      "epoch": 1.5153884887695313e-05,
      "step": 15890
    },
    {
      "epoch": 1.5153884887695313e-05,
      "step": 15890,
      "training_step_time": 17.07817769050598
    },
    {
      "epoch": 1.5154838562011719e-05,
      "model_forward_time": 5.622333765029907,
      "step": 15891
    },
    {
      "epoch": 1.5154838562011719e-05,
      "step": 15891,
      "training_step_time": 17.09289813041687
    },
    {
      "epoch": 1.5155792236328124e-05,
      "model_forward_time": 6.014928817749023,
      "step": 15892
    },
    {
      "epoch": 1.5155792236328124e-05,
      "step": 15892,
      "training_step_time": 17.531798839569092
    },
    {
      "epoch": 1.5156745910644532e-05,
      "model_forward_time": 5.622345924377441,
      "step": 15893
    },
    {
      "epoch": 1.5156745910644532e-05,
      "step": 15893,
      "training_step_time": 17.12190556526184
    },
    {
      "epoch": 1.5157699584960938e-05,
      "model_forward_time": 5.615018844604492,
      "step": 15894
    },
    {
      "epoch": 1.5157699584960938e-05,
      "step": 15894,
      "training_step_time": 17.11896586418152
    },
    {
      "epoch": 1.5158653259277343e-05,
      "model_forward_time": 5.6220009326934814,
      "step": 15895
    },
    {
      "epoch": 1.5158653259277343e-05,
      "step": 15895,
      "training_step_time": 17.116883039474487
    },
    {
      "epoch": 1.515960693359375e-05,
      "model_forward_time": 5.6109702587127686,
      "step": 15896
    },
    {
      "epoch": 1.515960693359375e-05,
      "step": 15896,
      "training_step_time": 17.068368434906006
    },
    {
      "epoch": 1.5160560607910156e-05,
      "model_forward_time": 5.601000785827637,
      "step": 15897
    },
    {
      "epoch": 1.5160560607910156e-05,
      "step": 15897,
      "training_step_time": 17.019809007644653
    },
    {
      "epoch": 1.5161514282226562e-05,
      "model_forward_time": 5.9593305587768555,
      "step": 15898
    },
    {
      "epoch": 1.5161514282226562e-05,
      "step": 15898,
      "training_step_time": 17.4302818775177
    },
    {
      "epoch": 1.5162467956542968e-05,
      "model_forward_time": 5.595431327819824,
      "step": 15899
    },
    {
      "epoch": 1.5162467956542968e-05,
      "step": 15899,
      "training_step_time": 17.04955768585205
    },
    {
      "epoch": 1.5163421630859375e-05,
      "grad_norm": 0.7395570278167725,
      "learning_rate": 1e-05,
      "loss": 0.0567,
      "step": 15900
    },
    {
      "epoch": 1.5163421630859375e-05,
      "model_forward_time": 5.583747863769531,
      "step": 15900
    },
    {
      "dynamics_loss_avg": 0.04872992765158415,
      "epoch": 1.5163421630859375e-05,
      "step": 15900
    },
    {
      "action_loss_avg": 0.006423467677086592,
      "epoch": 1.5163421630859375e-05,
      "step": 15900
    },
    {
      "epoch": 1.5163421630859375e-05,
      "step": 15900,
      "training_step_time": 17.0229811668396
    },
    {
      "epoch": 1.5164375305175781e-05,
      "model_forward_time": 5.601318836212158,
      "step": 15901
    },
    {
      "epoch": 1.5164375305175781e-05,
      "step": 15901,
      "training_step_time": 17.06924796104431
    },
    {
      "epoch": 1.5165328979492187e-05,
      "model_forward_time": 5.603442430496216,
      "step": 15902
    },
    {
      "epoch": 1.5165328979492187e-05,
      "step": 15902,
      "training_step_time": 17.082477569580078
    },
    {
      "epoch": 1.5166282653808594e-05,
      "model_forward_time": 5.98007345199585,
      "step": 15903
    },
    {
      "epoch": 1.5166282653808594e-05,
      "step": 15903,
      "training_step_time": 17.408133268356323
    },
    {
      "epoch": 1.5167236328125e-05,
      "model_forward_time": 5.587669610977173,
      "step": 15904
    },
    {
      "epoch": 1.5167236328125e-05,
      "step": 15904,
      "training_step_time": 17.04626441001892
    },
    {
      "epoch": 1.5168190002441406e-05,
      "model_forward_time": 5.593474626541138,
      "step": 15905
    },
    {
      "epoch": 1.5168190002441406e-05,
      "step": 15905,
      "training_step_time": 17.06911063194275
    },
    {
      "epoch": 1.5169143676757813e-05,
      "model_forward_time": 5.611175060272217,
      "step": 15906
    },
    {
      "epoch": 1.5169143676757813e-05,
      "step": 15906,
      "training_step_time": 17.129106760025024
    },
    {
      "epoch": 1.5170097351074219e-05,
      "model_forward_time": 5.626711130142212,
      "step": 15907
    },
    {
      "epoch": 1.5170097351074219e-05,
      "step": 15907,
      "training_step_time": 17.10819673538208
    },
    {
      "epoch": 1.5171051025390624e-05,
      "model_forward_time": 5.593403100967407,
      "step": 15908
    },
    {
      "epoch": 1.5171051025390624e-05,
      "step": 15908,
      "training_step_time": 17.033183574676514
    },
    {
      "epoch": 1.5172004699707032e-05,
      "model_forward_time": 5.919120788574219,
      "step": 15909
    },
    {
      "epoch": 1.5172004699707032e-05,
      "step": 15909,
      "training_step_time": 17.335164546966553
    },
    {
      "epoch": 1.5172958374023438e-05,
      "grad_norm": 1.0113705396652222,
      "learning_rate": 1e-05,
      "loss": 0.0546,
      "step": 15910
    },
    {
      "epoch": 1.5172958374023438e-05,
      "model_forward_time": 5.563782453536987,
      "step": 15910
    },
    {
      "dynamics_loss_avg": 0.050131565891206264,
      "epoch": 1.5172958374023438e-05,
      "step": 15910
    },
    {
      "action_loss_avg": 0.008092528930865229,
      "epoch": 1.5172958374023438e-05,
      "step": 15910
    },
    {
      "epoch": 1.5172958374023438e-05,
      "step": 15910,
      "training_step_time": 16.975438356399536
    },
    {
      "epoch": 1.5173912048339843e-05,
      "model_forward_time": 5.564480543136597,
      "step": 15911
    },
    {
      "epoch": 1.5173912048339843e-05,
      "step": 15911,
      "training_step_time": 16.982626914978027
    },
    {
      "epoch": 1.517486572265625e-05,
      "model_forward_time": 5.5613179206848145,
      "step": 15912
    },
    {
      "epoch": 1.517486572265625e-05,
      "step": 15912,
      "training_step_time": 16.988994598388672
    },
    {
      "epoch": 1.5175819396972656e-05,
      "model_forward_time": 5.573080062866211,
      "step": 15913
    },
    {
      "epoch": 1.5175819396972656e-05,
      "step": 15913,
      "training_step_time": 16.9822998046875
    },
    {
      "epoch": 1.5176773071289062e-05,
      "model_forward_time": 5.582370758056641,
      "step": 15914
    },
    {
      "epoch": 1.5176773071289062e-05,
      "step": 15914,
      "training_step_time": 17.35839533805847
    },
    {
      "epoch": 1.5177726745605468e-05,
      "model_forward_time": 5.571852445602417,
      "step": 15915
    },
    {
      "epoch": 1.5177726745605468e-05,
      "step": 15915,
      "training_step_time": 17.011776208877563
    },
    {
      "epoch": 1.5178680419921875e-05,
      "model_forward_time": 5.5791356563568115,
      "step": 15916
    },
    {
      "epoch": 1.5178680419921875e-05,
      "step": 15916,
      "training_step_time": 17.03294062614441
    },
    {
      "epoch": 1.5179634094238281e-05,
      "model_forward_time": 5.597243309020996,
      "step": 15917
    },
    {
      "epoch": 1.5179634094238281e-05,
      "step": 15917,
      "training_step_time": 17.065062761306763
    },
    {
      "epoch": 1.5180587768554687e-05,
      "model_forward_time": 5.596017599105835,
      "step": 15918
    },
    {
      "epoch": 1.5180587768554687e-05,
      "step": 15918,
      "training_step_time": 17.078201055526733
    },
    {
      "epoch": 1.5181541442871094e-05,
      "model_forward_time": 5.612384080886841,
      "step": 15919
    },
    {
      "epoch": 1.5181541442871094e-05,
      "step": 15919,
      "training_step_time": 17.077850580215454
    },
    {
      "epoch": 1.51824951171875e-05,
      "grad_norm": 0.9602242708206177,
      "learning_rate": 1e-05,
      "loss": 0.0569,
      "step": 15920
    },
    {
      "epoch": 1.51824951171875e-05,
      "model_forward_time": 5.97730827331543,
      "step": 15920
    },
    {
      "dynamics_loss_avg": 0.046617540344595906,
      "epoch": 1.51824951171875e-05,
      "step": 15920
    },
    {
      "action_loss_avg": 0.00636341639328748,
      "epoch": 1.51824951171875e-05,
      "step": 15920
    },
    {
      "epoch": 1.51824951171875e-05,
      "step": 15920,
      "training_step_time": 17.453837156295776
    },
    {
      "epoch": 1.5183448791503906e-05,
      "model_forward_time": 5.606200695037842,
      "step": 15921
    },
    {
      "epoch": 1.5183448791503906e-05,
      "step": 15921,
      "training_step_time": 17.085814476013184
    },
    {
      "epoch": 1.5184402465820313e-05,
      "model_forward_time": 5.610897064208984,
      "step": 15922
    },
    {
      "epoch": 1.5184402465820313e-05,
      "step": 15922,
      "training_step_time": 17.102502822875977
    },
    {
      "epoch": 1.5185356140136719e-05,
      "model_forward_time": 5.611264944076538,
      "step": 15923
    },
    {
      "epoch": 1.5185356140136719e-05,
      "step": 15923,
      "training_step_time": 17.085344076156616
    },
    {
      "epoch": 1.5186309814453125e-05,
      "model_forward_time": 5.607771158218384,
      "step": 15924
    },
    {
      "epoch": 1.5186309814453125e-05,
      "step": 15924,
      "training_step_time": 17.086730003356934
    },
    {
      "epoch": 1.5187263488769532e-05,
      "model_forward_time": 5.969959259033203,
      "step": 15925
    },
    {
      "epoch": 1.5187263488769532e-05,
      "step": 15925,
      "training_step_time": 17.40125870704651
    },
    {
      "epoch": 1.5188217163085938e-05,
      "model_forward_time": 5.58738899230957,
      "step": 15926
    },
    {
      "epoch": 1.5188217163085938e-05,
      "step": 15926,
      "training_step_time": 17.035094261169434
    },
    {
      "epoch": 1.5189170837402343e-05,
      "model_forward_time": 5.589648485183716,
      "step": 15927
    },
    {
      "epoch": 1.5189170837402343e-05,
      "step": 15927,
      "training_step_time": 17.040085792541504
    },
    {
      "epoch": 1.519012451171875e-05,
      "model_forward_time": 5.60213828086853,
      "step": 15928
    },
    {
      "epoch": 1.519012451171875e-05,
      "step": 15928,
      "training_step_time": 17.090274572372437
    },
    {
      "epoch": 1.5191078186035157e-05,
      "model_forward_time": 5.62611985206604,
      "step": 15929
    },
    {
      "epoch": 1.5191078186035157e-05,
      "step": 15929,
      "training_step_time": 17.126993656158447
    },
    {
      "epoch": 1.5192031860351562e-05,
      "grad_norm": 1.2279162406921387,
      "learning_rate": 1e-05,
      "loss": 0.06,
      "step": 15930
    },
    {
      "epoch": 1.5192031860351562e-05,
      "model_forward_time": 5.629147052764893,
      "step": 15930
    },
    {
      "dynamics_loss_avg": 0.059218593128025535,
      "epoch": 1.5192031860351562e-05,
      "step": 15930
    },
    {
      "action_loss_avg": 0.007948754332028329,
      "epoch": 1.5192031860351562e-05,
      "step": 15930
    },
    {
      "epoch": 1.5192031860351562e-05,
      "step": 15930,
      "training_step_time": 17.122122049331665
    },
    {
      "epoch": 1.5192985534667968e-05,
      "model_forward_time": 5.954357862472534,
      "step": 15931
    },
    {
      "epoch": 1.5192985534667968e-05,
      "step": 15931,
      "training_step_time": 17.40911340713501
    },
    {
      "epoch": 1.5193939208984375e-05,
      "model_forward_time": 5.601269483566284,
      "step": 15932
    },
    {
      "epoch": 1.5193939208984375e-05,
      "step": 15932,
      "training_step_time": 17.082499980926514
    },
    {
      "epoch": 1.5194892883300781e-05,
      "model_forward_time": 5.618825912475586,
      "step": 15933
    },
    {
      "epoch": 1.5194892883300781e-05,
      "step": 15933,
      "training_step_time": 17.09873628616333
    },
    {
      "epoch": 1.5195846557617187e-05,
      "model_forward_time": 5.605891466140747,
      "step": 15934
    },
    {
      "epoch": 1.5195846557617187e-05,
      "step": 15934,
      "training_step_time": 17.059269666671753
    },
    {
      "epoch": 1.5196800231933594e-05,
      "model_forward_time": 5.578399658203125,
      "step": 15935
    },
    {
      "epoch": 1.5196800231933594e-05,
      "step": 15935,
      "training_step_time": 16.993521690368652
    },
    {
      "epoch": 1.519775390625e-05,
      "model_forward_time": 5.973915338516235,
      "step": 15936
    },
    {
      "epoch": 1.519775390625e-05,
      "step": 15936,
      "training_step_time": 17.405866146087646
    },
    {
      "epoch": 1.5198707580566406e-05,
      "model_forward_time": 5.6005425453186035,
      "step": 15937
    },
    {
      "epoch": 1.5198707580566406e-05,
      "step": 15937,
      "training_step_time": 17.093717336654663
    },
    {
      "epoch": 1.5199661254882813e-05,
      "model_forward_time": 5.612833738327026,
      "step": 15938
    },
    {
      "epoch": 1.5199661254882813e-05,
      "step": 15938,
      "training_step_time": 17.13485813140869
    },
    {
      "epoch": 1.5200614929199219e-05,
      "model_forward_time": 5.632197618484497,
      "step": 15939
    },
    {
      "epoch": 1.5200614929199219e-05,
      "step": 15939,
      "training_step_time": 17.16155767440796
    },
    {
      "epoch": 1.5201568603515625e-05,
      "grad_norm": 0.6333701610565186,
      "learning_rate": 1e-05,
      "loss": 0.0528,
      "step": 15940
    },
    {
      "epoch": 1.5201568603515625e-05,
      "model_forward_time": 5.626679182052612,
      "step": 15940
    },
    {
      "dynamics_loss_avg": 0.04038671776652336,
      "epoch": 1.5201568603515625e-05,
      "step": 15940
    },
    {
      "action_loss_avg": 0.005524366372264921,
      "epoch": 1.5201568603515625e-05,
      "step": 15940
    },
    {
      "epoch": 1.5201568603515625e-05,
      "step": 15940,
      "training_step_time": 17.152382135391235
    },
    {
      "epoch": 1.5202522277832032e-05,
      "model_forward_time": 5.630603551864624,
      "step": 15941
    },
    {
      "epoch": 1.5202522277832032e-05,
      "step": 15941,
      "training_step_time": 17.135555744171143
    },
    {
      "epoch": 1.5203475952148438e-05,
      "model_forward_time": 5.967421531677246,
      "step": 15942
    },
    {
      "epoch": 1.5203475952148438e-05,
      "step": 15942,
      "training_step_time": 17.473841905593872
    },
    {
      "epoch": 1.5204429626464843e-05,
      "model_forward_time": 5.618070602416992,
      "step": 15943
    },
    {
      "epoch": 1.5204429626464843e-05,
      "step": 15943,
      "training_step_time": 17.0892813205719
    },
    {
      "epoch": 1.520538330078125e-05,
      "model_forward_time": 5.583540916442871,
      "step": 15944
    },
    {
      "epoch": 1.520538330078125e-05,
      "step": 15944,
      "training_step_time": 16.99945569038391
    },
    {
      "epoch": 1.5206336975097657e-05,
      "model_forward_time": 5.577882528305054,
      "step": 15945
    },
    {
      "epoch": 1.5206336975097657e-05,
      "step": 15945,
      "training_step_time": 17.003997564315796
    },
    {
      "epoch": 1.5207290649414062e-05,
      "model_forward_time": 5.5616583824157715,
      "step": 15946
    },
    {
      "epoch": 1.5207290649414062e-05,
      "step": 15946,
      "training_step_time": 16.9926655292511
    },
    {
      "epoch": 1.5208244323730468e-05,
      "model_forward_time": 5.574867010116577,
      "step": 15947
    },
    {
      "epoch": 1.5208244323730468e-05,
      "step": 15947,
      "training_step_time": 17.3619327545166
    },
    {
      "epoch": 1.5209197998046875e-05,
      "model_forward_time": 5.55781888961792,
      "step": 15948
    },
    {
      "epoch": 1.5209197998046875e-05,
      "step": 15948,
      "training_step_time": 16.969882011413574
    },
    {
      "epoch": 1.5210151672363281e-05,
      "model_forward_time": 5.573760986328125,
      "step": 15949
    },
    {
      "epoch": 1.5210151672363281e-05,
      "step": 15949,
      "training_step_time": 17.01989436149597
    },
    {
      "epoch": 1.5211105346679687e-05,
      "grad_norm": 0.18101848661899567,
      "learning_rate": 1e-05,
      "loss": 0.0587,
      "step": 15950
    },
    {
      "epoch": 1.5211105346679687e-05,
      "model_forward_time": 5.567290306091309,
      "step": 15950
    },
    {
      "dynamics_loss_avg": 0.05381610011681914,
      "epoch": 1.5211105346679687e-05,
      "step": 15950
    },
    {
      "action_loss_avg": 0.007152102631516754,
      "epoch": 1.5211105346679687e-05,
      "step": 15950
    },
    {
      "epoch": 1.5211105346679687e-05,
      "step": 15950,
      "training_step_time": 17.000961780548096
    },
    {
      "epoch": 1.5212059020996094e-05,
      "model_forward_time": 5.578760385513306,
      "step": 15951
    },
    {
      "epoch": 1.5212059020996094e-05,
      "step": 15951,
      "training_step_time": 16.999974727630615
    },
    {
      "epoch": 1.52130126953125e-05,
      "model_forward_time": 5.586786985397339,
      "step": 15952
    },
    {
      "epoch": 1.52130126953125e-05,
      "step": 15952,
      "training_step_time": 17.061721801757812
    },
    {
      "epoch": 1.5213966369628906e-05,
      "model_forward_time": 5.962801218032837,
      "step": 15953
    },
    {
      "epoch": 1.5213966369628906e-05,
      "step": 15953,
      "training_step_time": 17.452329635620117
    },
    {
      "epoch": 1.5214920043945313e-05,
      "model_forward_time": 5.627087593078613,
      "step": 15954
    },
    {
      "epoch": 1.5214920043945313e-05,
      "step": 15954,
      "training_step_time": 17.16221046447754
    },
    {
      "epoch": 1.5215873718261719e-05,
      "model_forward_time": 5.626771926879883,
      "step": 15955
    },
    {
      "epoch": 1.5215873718261719e-05,
      "step": 15955,
      "training_step_time": 17.15069842338562
    },
    {
      "epoch": 1.5216827392578125e-05,
      "model_forward_time": 5.628310441970825,
      "step": 15956
    },
    {
      "epoch": 1.5216827392578125e-05,
      "step": 15956,
      "training_step_time": 17.145239114761353
    },
    {
      "epoch": 1.5217781066894532e-05,
      "model_forward_time": 5.612682580947876,
      "step": 15957
    },
    {
      "epoch": 1.5217781066894532e-05,
      "step": 15957,
      "training_step_time": 17.10216760635376
    },
    {
      "epoch": 1.5218734741210938e-05,
      "model_forward_time": 5.995184659957886,
      "step": 15958
    },
    {
      "epoch": 1.5218734741210938e-05,
      "step": 15958,
      "training_step_time": 17.47778820991516
    },
    {
      "epoch": 1.5219688415527343e-05,
      "model_forward_time": 5.617654085159302,
      "step": 15959
    },
    {
      "epoch": 1.5219688415527343e-05,
      "step": 15959,
      "training_step_time": 17.09437894821167
    },
    {
      "epoch": 1.522064208984375e-05,
      "grad_norm": 0.3813408315181732,
      "learning_rate": 1e-05,
      "loss": 0.0563,
      "step": 15960
    },
    {
      "epoch": 1.522064208984375e-05,
      "model_forward_time": 5.598514080047607,
      "step": 15960
    },
    {
      "dynamics_loss_avg": 0.05122501906007528,
      "epoch": 1.522064208984375e-05,
      "step": 15960
    },
    {
      "action_loss_avg": 0.00661345170228742,
      "epoch": 1.522064208984375e-05,
      "step": 15960
    },
    {
      "epoch": 1.522064208984375e-05,
      "step": 15960,
      "training_step_time": 17.067511558532715
    },
    {
      "epoch": 1.5221595764160157e-05,
      "model_forward_time": 5.607715368270874,
      "step": 15961
    },
    {
      "epoch": 1.5221595764160157e-05,
      "step": 15961,
      "training_step_time": 17.06605052947998
    },
    {
      "epoch": 1.5222549438476562e-05,
      "model_forward_time": 5.593943119049072,
      "step": 15962
    },
    {
      "epoch": 1.5222549438476562e-05,
      "step": 15962,
      "training_step_time": 17.00486445426941
    },
    {
      "epoch": 1.5223503112792968e-05,
      "model_forward_time": 5.592847585678101,
      "step": 15963
    },
    {
      "epoch": 1.5223503112792968e-05,
      "step": 15963,
      "training_step_time": 17.02396035194397
    },
    {
      "epoch": 1.5224456787109375e-05,
      "model_forward_time": 5.946470022201538,
      "step": 15964
    },
    {
      "epoch": 1.5224456787109375e-05,
      "step": 15964,
      "training_step_time": 17.387789249420166
    },
    {
      "epoch": 1.5225410461425781e-05,
      "model_forward_time": 5.5975728034973145,
      "step": 15965
    },
    {
      "epoch": 1.5225410461425781e-05,
      "step": 15965,
      "training_step_time": 17.070786237716675
    },
    {
      "epoch": 1.5226364135742187e-05,
      "model_forward_time": 5.602859258651733,
      "step": 15966
    },
    {
      "epoch": 1.5226364135742187e-05,
      "step": 15966,
      "training_step_time": 17.07965850830078
    },
    {
      "epoch": 1.5227317810058594e-05,
      "model_forward_time": 5.6100475788116455,
      "step": 15967
    },
    {
      "epoch": 1.5227317810058594e-05,
      "step": 15967,
      "training_step_time": 17.101150274276733
    },
    {
      "epoch": 1.5228271484375e-05,
      "model_forward_time": 5.618443250656128,
      "step": 15968
    },
    {
      "epoch": 1.5228271484375e-05,
      "step": 15968,
      "training_step_time": 17.1102614402771
    },
    {
      "epoch": 1.5229225158691406e-05,
      "model_forward_time": 6.004481315612793,
      "step": 15969
    },
    {
      "epoch": 1.5229225158691406e-05,
      "step": 15969,
      "training_step_time": 17.480934381484985
    },
    {
      "epoch": 1.5230178833007813e-05,
      "grad_norm": 0.7394438982009888,
      "learning_rate": 1e-05,
      "loss": 0.0543,
      "step": 15970
    },
    {
      "epoch": 1.5230178833007813e-05,
      "model_forward_time": 5.607284069061279,
      "step": 15970
    },
    {
      "dynamics_loss_avg": 0.04813610725104809,
      "epoch": 1.5230178833007813e-05,
      "step": 15970
    },
    {
      "action_loss_avg": 0.005364691070280969,
      "epoch": 1.5230178833007813e-05,
      "step": 15970
    },
    {
      "epoch": 1.5230178833007813e-05,
      "step": 15970,
      "training_step_time": 17.088403463363647
    },
    {
      "epoch": 1.5231132507324219e-05,
      "model_forward_time": 5.602613687515259,
      "step": 15971
    },
    {
      "epoch": 1.5231132507324219e-05,
      "step": 15971,
      "training_step_time": 17.07801842689514
    },
    {
      "epoch": 1.5232086181640625e-05,
      "model_forward_time": 5.614465713500977,
      "step": 15972
    },
    {
      "epoch": 1.5232086181640625e-05,
      "step": 15972,
      "training_step_time": 17.071937322616577
    },
    {
      "epoch": 1.5233039855957032e-05,
      "model_forward_time": 5.658725023269653,
      "step": 15973
    },
    {
      "epoch": 1.5233039855957032e-05,
      "step": 15973,
      "training_step_time": 17.165725708007812
    },
    {
      "epoch": 1.5233993530273438e-05,
      "model_forward_time": 5.6082541942596436,
      "step": 15974
    },
    {
      "epoch": 1.5233993530273438e-05,
      "step": 15974,
      "training_step_time": 17.053654432296753
    },
    {
      "epoch": 1.5234947204589843e-05,
      "model_forward_time": 5.962500095367432,
      "step": 15975
    },
    {
      "epoch": 1.5234947204589843e-05,
      "step": 15975,
      "training_step_time": 17.414376974105835
    },
    {
      "epoch": 1.523590087890625e-05,
      "model_forward_time": 5.598876714706421,
      "step": 15976
    },
    {
      "epoch": 1.523590087890625e-05,
      "step": 15976,
      "training_step_time": 17.061001300811768
    },
    {
      "epoch": 1.5236854553222657e-05,
      "model_forward_time": 5.629964828491211,
      "step": 15977
    },
    {
      "epoch": 1.5236854553222657e-05,
      "step": 15977,
      "training_step_time": 17.085699558258057
    },
    {
      "epoch": 1.5237808227539062e-05,
      "model_forward_time": 5.600634336471558,
      "step": 15978
    },
    {
      "epoch": 1.5237808227539062e-05,
      "step": 15978,
      "training_step_time": 17.09111976623535
    },
    {
      "epoch": 1.5238761901855468e-05,
      "model_forward_time": 5.607036828994751,
      "step": 15979
    },
    {
      "epoch": 1.5238761901855468e-05,
      "step": 15979,
      "training_step_time": 17.104130744934082
    },
    {
      "epoch": 1.5239715576171875e-05,
      "grad_norm": 0.4555678367614746,
      "learning_rate": 1e-05,
      "loss": 0.0556,
      "step": 15980
    },
    {
      "epoch": 1.5239715576171875e-05,
      "model_forward_time": 5.618226766586304,
      "step": 15980
    },
    {
      "dynamics_loss_avg": 0.051563076116144656,
      "epoch": 1.5239715576171875e-05,
      "step": 15980
    },
    {
      "action_loss_avg": 0.006262505881022662,
      "epoch": 1.5239715576171875e-05,
      "step": 15980
    },
    {
      "epoch": 1.5239715576171875e-05,
      "step": 15980,
      "training_step_time": 17.497621297836304
    },
    {
      "epoch": 1.5240669250488281e-05,
      "model_forward_time": 5.624639272689819,
      "step": 15981
    },
    {
      "epoch": 1.5240669250488281e-05,
      "step": 15981,
      "training_step_time": 17.135197639465332
    },
    {
      "epoch": 1.5241622924804687e-05,
      "model_forward_time": 5.611068487167358,
      "step": 15982
    },
    {
      "epoch": 1.5241622924804687e-05,
      "step": 15982,
      "training_step_time": 17.11672067642212
    },
    {
      "epoch": 1.5242576599121094e-05,
      "model_forward_time": 5.631456613540649,
      "step": 15983
    },
    {
      "epoch": 1.5242576599121094e-05,
      "step": 15983,
      "training_step_time": 17.11721134185791
    },
    {
      "epoch": 1.52435302734375e-05,
      "model_forward_time": 5.614839553833008,
      "step": 15984
    },
    {
      "epoch": 1.52435302734375e-05,
      "step": 15984,
      "training_step_time": 17.091246843338013
    },
    {
      "epoch": 1.5244483947753906e-05,
      "model_forward_time": 5.579219818115234,
      "step": 15985
    },
    {
      "epoch": 1.5244483947753906e-05,
      "step": 15985,
      "training_step_time": 17.00244903564453
    },
    {
      "epoch": 1.5245437622070313e-05,
      "model_forward_time": 5.936811447143555,
      "step": 15986
    },
    {
      "epoch": 1.5245437622070313e-05,
      "step": 15986,
      "training_step_time": 17.351165294647217
    },
    {
      "epoch": 1.5246391296386719e-05,
      "model_forward_time": 5.562958002090454,
      "step": 15987
    },
    {
      "epoch": 1.5246391296386719e-05,
      "step": 15987,
      "training_step_time": 16.992830276489258
    },
    {
      "epoch": 1.5247344970703125e-05,
      "model_forward_time": 5.560618877410889,
      "step": 15988
    },
    {
      "epoch": 1.5247344970703125e-05,
      "step": 15988,
      "training_step_time": 16.942444801330566
    },
    {
      "epoch": 1.5248298645019532e-05,
      "model_forward_time": 5.569902658462524,
      "step": 15989
    },
    {
      "epoch": 1.5248298645019532e-05,
      "step": 15989,
      "training_step_time": 16.988081216812134
    },
    {
      "epoch": 1.5249252319335938e-05,
      "grad_norm": 1.417849063873291,
      "learning_rate": 1e-05,
      "loss": 0.0736,
      "step": 15990
    },
    {
      "epoch": 1.5249252319335938e-05,
      "model_forward_time": 5.583322048187256,
      "step": 15990
    },
    {
      "dynamics_loss_avg": 0.06243513002991676,
      "epoch": 1.5249252319335938e-05,
      "step": 15990
    },
    {
      "action_loss_avg": 0.00896844504168257,
      "epoch": 1.5249252319335938e-05,
      "step": 15990
    },
    {
      "epoch": 1.5249252319335938e-05,
      "step": 15990,
      "training_step_time": 16.998184204101562
    },
    {
      "epoch": 1.5250205993652343e-05,
      "model_forward_time": 5.974529266357422,
      "step": 15991
    },
    {
      "epoch": 1.5250205993652343e-05,
      "step": 15991,
      "training_step_time": 17.420803785324097
    },
    {
      "epoch": 1.525115966796875e-05,
      "model_forward_time": 5.577736139297485,
      "step": 15992
    },
    {
      "epoch": 1.525115966796875e-05,
      "step": 15992,
      "training_step_time": 17.00566554069519
    },
    {
      "epoch": 1.5252113342285157e-05,
      "model_forward_time": 5.58882737159729,
      "step": 15993
    },
    {
      "epoch": 1.5252113342285157e-05,
      "step": 15993,
      "training_step_time": 17.04277801513672
    },
    {
      "epoch": 1.5253067016601562e-05,
      "model_forward_time": 5.620861053466797,
      "step": 15994
    },
    {
      "epoch": 1.5253067016601562e-05,
      "step": 15994,
      "training_step_time": 17.10111689567566
    },
    {
      "epoch": 1.5254020690917968e-05,
      "model_forward_time": 5.606622695922852,
      "step": 15995
    },
    {
      "epoch": 1.5254020690917968e-05,
      "step": 15995,
      "training_step_time": 17.089862823486328
    },
    {
      "epoch": 1.5254974365234375e-05,
      "model_forward_time": 5.604408502578735,
      "step": 15996
    },
    {
      "epoch": 1.5254974365234375e-05,
      "step": 15996,
      "training_step_time": 17.090820789337158
    },
    {
      "epoch": 1.5255928039550781e-05,
      "model_forward_time": 5.957001686096191,
      "step": 15997
    },
    {
      "epoch": 1.5255928039550781e-05,
      "step": 15997,
      "training_step_time": 17.436484813690186
    },
    {
      "epoch": 1.5256881713867187e-05,
      "model_forward_time": 5.6076719760894775,
      "step": 15998
    },
    {
      "epoch": 1.5256881713867187e-05,
      "step": 15998,
      "training_step_time": 17.091097354888916
    },
    {
      "epoch": 1.5257835388183594e-05,
      "model_forward_time": 5.6184892654418945,
      "step": 15999
    },
    {
      "epoch": 1.5257835388183594e-05,
      "step": 15999,
      "training_step_time": 17.10429048538208
    },
    {
      "epoch": 1.52587890625e-05,
      "grad_norm": 0.4101746678352356,
      "learning_rate": 1e-05,
      "loss": 0.0619,
      "step": 16000
    },
    {
      "epoch": 1.52587890625e-05,
      "model_forward_time": 5.650777339935303,
      "step": 16000
    },
    {
      "dynamics_loss_avg": 0.05141676906496286,
      "epoch": 1.52587890625e-05,
      "step": 16000
    },
    {
      "action_loss_avg": 0.007387003838084638,
      "epoch": 1.52587890625e-05,
      "step": 16000
    },
    {
      "epoch": 1.52587890625e-05,
      "step": 16000,
      "training_step_time": 17.084710121154785
    },
    {
      "epoch": 1.5259742736816407e-05,
      "model_forward_time": 5.6146814823150635,
      "step": 16001
    },
    {
      "epoch": 1.5259742736816407e-05,
      "step": 16001,
      "training_step_time": 17.032753705978394
    },
    {
      "epoch": 1.526069641113281e-05,
      "model_forward_time": 5.612706661224365,
      "step": 16002
    },
    {
      "epoch": 1.526069641113281e-05,
      "step": 16002,
      "training_step_time": 17.4149911403656
    },
    {
      "epoch": 1.526165008544922e-05,
      "model_forward_time": 5.627608299255371,
      "step": 16003
    },
    {
      "epoch": 1.526165008544922e-05,
      "step": 16003,
      "training_step_time": 17.056793928146362
    },
    {
      "epoch": 1.5262603759765626e-05,
      "model_forward_time": 5.623847961425781,
      "step": 16004
    },
    {
      "epoch": 1.5262603759765626e-05,
      "step": 16004,
      "training_step_time": 17.092586040496826
    },
    {
      "epoch": 1.526355743408203e-05,
      "model_forward_time": 5.648274660110474,
      "step": 16005
    },
    {
      "epoch": 1.526355743408203e-05,
      "step": 16005,
      "training_step_time": 17.170000791549683
    },
    {
      "epoch": 1.5264511108398438e-05,
      "model_forward_time": 5.662480592727661,
      "step": 16006
    },
    {
      "epoch": 1.5264511108398438e-05,
      "step": 16006,
      "training_step_time": 17.165300846099854
    },
    {
      "epoch": 1.5265464782714845e-05,
      "model_forward_time": 5.6799867153167725,
      "step": 16007
    },
    {
      "epoch": 1.5265464782714845e-05,
      "step": 16007,
      "training_step_time": 17.207226037979126
    },
    {
      "epoch": 1.526641845703125e-05,
      "model_forward_time": 6.035302400588989,
      "step": 16008
    },
    {
      "epoch": 1.526641845703125e-05,
      "step": 16008,
      "training_step_time": 17.54390025138855
    },
    {
      "epoch": 1.5267372131347657e-05,
      "model_forward_time": 5.658969402313232,
      "step": 16009
    },
    {
      "epoch": 1.5267372131347657e-05,
      "step": 16009,
      "training_step_time": 17.236873149871826
    },
    {
      "epoch": 1.5268325805664064e-05,
      "grad_norm": 0.5660987496376038,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 16010
    },
    {
      "epoch": 1.5268325805664064e-05,
      "model_forward_time": 5.630300760269165,
      "step": 16010
    },
    {
      "dynamics_loss_avg": 0.0717781014740467,
      "epoch": 1.5268325805664064e-05,
      "step": 16010
    },
    {
      "action_loss_avg": 0.009702760959044099,
      "epoch": 1.5268325805664064e-05,
      "step": 16010
    },
    {
      "epoch": 1.5268325805664064e-05,
      "step": 16010,
      "training_step_time": 17.14773917198181
    },
    {
      "epoch": 1.5269279479980468e-05,
      "model_forward_time": 5.61691427230835,
      "step": 16011
    },
    {
      "epoch": 1.5269279479980468e-05,
      "step": 16011,
      "training_step_time": 17.10315251350403
    },
    {
      "epoch": 1.5270233154296875e-05,
      "model_forward_time": 4.277093410491943,
      "step": 16012
    },
    {
      "epoch": 1.5270233154296875e-05,
      "step": 16012,
      "training_step_time": 12.800449848175049
    },
    {
      "epoch": 1.5271186828613283e-05,
      "model_forward_time": 5.611650466918945,
      "step": 16013
    },
    {
      "epoch": 1.5271186828613283e-05,
      "step": 16013,
      "training_step_time": 17.076377391815186
    },
    {
      "epoch": 1.5272140502929687e-05,
      "model_forward_time": 5.963207721710205,
      "step": 16014
    },
    {
      "epoch": 1.5272140502929687e-05,
      "step": 16014,
      "training_step_time": 17.400209188461304
    },
    {
      "epoch": 1.5273094177246094e-05,
      "model_forward_time": 5.604682207107544,
      "step": 16015
    },
    {
      "epoch": 1.5273094177246094e-05,
      "step": 16015,
      "training_step_time": 17.099107027053833
    },
    {
      "epoch": 1.52740478515625e-05,
      "model_forward_time": 5.611396312713623,
      "step": 16016
    },
    {
      "epoch": 1.52740478515625e-05,
      "step": 16016,
      "training_step_time": 17.117873430252075
    },
    {
      "epoch": 1.5275001525878906e-05,
      "model_forward_time": 5.607240438461304,
      "step": 16017
    },
    {
      "epoch": 1.5275001525878906e-05,
      "step": 16017,
      "training_step_time": 17.081982135772705
    },
    {
      "epoch": 1.5275955200195313e-05,
      "model_forward_time": 5.608796834945679,
      "step": 16018
    },
    {
      "epoch": 1.5275955200195313e-05,
      "step": 16018,
      "training_step_time": 17.097500562667847
    },
    {
      "epoch": 1.5276908874511717e-05,
      "model_forward_time": 5.988508939743042,
      "step": 16019
    },
    {
      "epoch": 1.5276908874511717e-05,
      "step": 16019,
      "training_step_time": 17.467741012573242
    },
    {
      "epoch": 1.5277862548828125e-05,
      "grad_norm": 0.5447949171066284,
      "learning_rate": 1e-05,
      "loss": 0.0781,
      "step": 16020
    },
    {
      "epoch": 1.5277862548828125e-05,
      "model_forward_time": 5.607424736022949,
      "step": 16020
    },
    {
      "dynamics_loss_avg": 0.05965086128562689,
      "epoch": 1.5277862548828125e-05,
      "step": 16020
    },
    {
      "action_loss_avg": 0.006230924522969872,
      "epoch": 1.5277862548828125e-05,
      "step": 16020
    },
    {
      "epoch": 1.5277862548828125e-05,
      "step": 16020,
      "training_step_time": 17.106024265289307
    },
    {
      "epoch": 1.5278816223144532e-05,
      "model_forward_time": 5.611423969268799,
      "step": 16021
    },
    {
      "epoch": 1.5278816223144532e-05,
      "step": 16021,
      "training_step_time": 17.085405111312866
    },
    {
      "epoch": 1.5279769897460936e-05,
      "model_forward_time": 5.593329668045044,
      "step": 16022
    },
    {
      "epoch": 1.5279769897460936e-05,
      "step": 16022,
      "training_step_time": 17.022318363189697
    },
    {
      "epoch": 1.5280723571777343e-05,
      "model_forward_time": 5.58756422996521,
      "step": 16023
    },
    {
      "epoch": 1.5280723571777343e-05,
      "step": 16023,
      "training_step_time": 17.00438952445984
    },
    {
      "epoch": 1.528167724609375e-05,
      "model_forward_time": 5.94744086265564,
      "step": 16024
    },
    {
      "epoch": 1.528167724609375e-05,
      "step": 16024,
      "training_step_time": 17.39595675468445
    },
    {
      "epoch": 1.5282630920410155e-05,
      "model_forward_time": 5.591002464294434,
      "step": 16025
    },
    {
      "epoch": 1.5282630920410155e-05,
      "step": 16025,
      "training_step_time": 17.043259382247925
    },
    {
      "epoch": 1.5283584594726562e-05,
      "model_forward_time": 5.5849690437316895,
      "step": 16026
    },
    {
      "epoch": 1.5283584594726562e-05,
      "step": 16026,
      "training_step_time": 17.016857624053955
    },
    {
      "epoch": 1.528453826904297e-05,
      "model_forward_time": 5.5960776805877686,
      "step": 16027
    },
    {
      "epoch": 1.528453826904297e-05,
      "step": 16027,
      "training_step_time": 17.066869974136353
    },
    {
      "epoch": 1.5285491943359374e-05,
      "model_forward_time": 5.601304292678833,
      "step": 16028
    },
    {
      "epoch": 1.5285491943359374e-05,
      "step": 16028,
      "training_step_time": 17.06879758834839
    },
    {
      "epoch": 1.528644561767578e-05,
      "model_forward_time": 5.607215404510498,
      "step": 16029
    },
    {
      "epoch": 1.528644561767578e-05,
      "step": 16029,
      "training_step_time": 17.06697988510132
    },
    {
      "epoch": 1.528739929199219e-05,
      "grad_norm": 0.6275255680084229,
      "learning_rate": 1e-05,
      "loss": 0.0496,
      "step": 16030
    },
    {
      "epoch": 1.528739929199219e-05,
      "model_forward_time": 5.966918230056763,
      "step": 16030
    },
    {
      "dynamics_loss_avg": 0.04690781328827143,
      "epoch": 1.528739929199219e-05,
      "step": 16030
    },
    {
      "action_loss_avg": 0.005720063461922109,
      "epoch": 1.528739929199219e-05,
      "step": 16030
    },
    {
      "epoch": 1.528739929199219e-05,
      "step": 16030,
      "training_step_time": 17.405954837799072
    },
    {
      "epoch": 1.5288352966308593e-05,
      "model_forward_time": 5.578933954238892,
      "step": 16031
    },
    {
      "epoch": 1.5288352966308593e-05,
      "step": 16031,
      "training_step_time": 16.988235473632812
    },
    {
      "epoch": 1.5289306640625e-05,
      "model_forward_time": 5.580345869064331,
      "step": 16032
    },
    {
      "epoch": 1.5289306640625e-05,
      "step": 16032,
      "training_step_time": 17.028462171554565
    },
    {
      "epoch": 1.5290260314941407e-05,
      "model_forward_time": 5.561741828918457,
      "step": 16033
    },
    {
      "epoch": 1.5290260314941407e-05,
      "step": 16033,
      "training_step_time": 16.96885371208191
    },
    {
      "epoch": 1.529121398925781e-05,
      "model_forward_time": 5.59199595451355,
      "step": 16034
    },
    {
      "epoch": 1.529121398925781e-05,
      "step": 16034,
      "training_step_time": 17.046441555023193
    },
    {
      "epoch": 1.529216766357422e-05,
      "model_forward_time": 5.584352016448975,
      "step": 16035
    },
    {
      "epoch": 1.529216766357422e-05,
      "step": 16035,
      "training_step_time": 17.38273787498474
    },
    {
      "epoch": 1.5293121337890626e-05,
      "model_forward_time": 5.601307392120361,
      "step": 16036
    },
    {
      "epoch": 1.5293121337890626e-05,
      "step": 16036,
      "training_step_time": 17.060014486312866
    },
    {
      "epoch": 1.529407501220703e-05,
      "model_forward_time": 5.6021199226379395,
      "step": 16037
    },
    {
      "epoch": 1.529407501220703e-05,
      "step": 16037,
      "training_step_time": 17.05021619796753
    },
    {
      "epoch": 1.5295028686523438e-05,
      "model_forward_time": 5.59215784072876,
      "step": 16038
    },
    {
      "epoch": 1.5295028686523438e-05,
      "step": 16038,
      "training_step_time": 17.029160499572754
    },
    {
      "epoch": 1.5295982360839845e-05,
      "model_forward_time": 5.58424186706543,
      "step": 16039
    },
    {
      "epoch": 1.5295982360839845e-05,
      "step": 16039,
      "training_step_time": 17.009033203125
    },
    {
      "epoch": 1.529693603515625e-05,
      "grad_norm": 0.7850489616394043,
      "learning_rate": 1e-05,
      "loss": 0.0704,
      "step": 16040
    },
    {
      "epoch": 1.529693603515625e-05,
      "model_forward_time": 5.5848939418792725,
      "step": 16040
    },
    {
      "dynamics_loss_avg": 0.06738989911973477,
      "epoch": 1.529693603515625e-05,
      "step": 16040
    },
    {
      "action_loss_avg": 0.009971944964490831,
      "epoch": 1.529693603515625e-05,
      "step": 16040
    },
    {
      "epoch": 1.529693603515625e-05,
      "step": 16040,
      "training_step_time": 17.040162563323975
    },
    {
      "epoch": 1.5297889709472657e-05,
      "model_forward_time": 5.968091011047363,
      "step": 16041
    },
    {
      "epoch": 1.5297889709472657e-05,
      "step": 16041,
      "training_step_time": 17.453224182128906
    },
    {
      "epoch": 1.5298843383789064e-05,
      "model_forward_time": 5.616544485092163,
      "step": 16042
    },
    {
      "epoch": 1.5298843383789064e-05,
      "step": 16042,
      "training_step_time": 17.138577699661255
    },
    {
      "epoch": 1.5299797058105468e-05,
      "model_forward_time": 5.620204448699951,
      "step": 16043
    },
    {
      "epoch": 1.5299797058105468e-05,
      "step": 16043,
      "training_step_time": 17.122311115264893
    },
    {
      "epoch": 1.5300750732421875e-05,
      "model_forward_time": 5.610654830932617,
      "step": 16044
    },
    {
      "epoch": 1.5300750732421875e-05,
      "step": 16044,
      "training_step_time": 17.083697080612183
    },
    {
      "epoch": 1.5301704406738283e-05,
      "model_forward_time": 5.616113901138306,
      "step": 16045
    },
    {
      "epoch": 1.5301704406738283e-05,
      "step": 16045,
      "training_step_time": 17.101701736450195
    },
    {
      "epoch": 1.5302658081054687e-05,
      "model_forward_time": 5.603696346282959,
      "step": 16046
    },
    {
      "epoch": 1.5302658081054687e-05,
      "step": 16046,
      "training_step_time": 17.453184366226196
    },
    {
      "epoch": 1.5303611755371094e-05,
      "model_forward_time": 5.603767156600952,
      "step": 16047
    },
    {
      "epoch": 1.5303611755371094e-05,
      "step": 16047,
      "training_step_time": 17.05410099029541
    },
    {
      "epoch": 1.53045654296875e-05,
      "model_forward_time": 5.591101408004761,
      "step": 16048
    },
    {
      "epoch": 1.53045654296875e-05,
      "step": 16048,
      "training_step_time": 17.04155969619751
    },
    {
      "epoch": 1.5305519104003906e-05,
      "model_forward_time": 5.580252647399902,
      "step": 16049
    },
    {
      "epoch": 1.5305519104003906e-05,
      "step": 16049,
      "training_step_time": 16.982468128204346
    },
    {
      "epoch": 1.5306472778320313e-05,
      "grad_norm": 1.3028684854507446,
      "learning_rate": 1e-05,
      "loss": 0.0685,
      "step": 16050
    },
    {
      "epoch": 1.5306472778320313e-05,
      "model_forward_time": 5.588868618011475,
      "step": 16050
    },
    {
      "dynamics_loss_avg": 0.056949305534362796,
      "epoch": 1.5306472778320313e-05,
      "step": 16050
    },
    {
      "action_loss_avg": 0.008817951381206512,
      "epoch": 1.5306472778320313e-05,
      "step": 16050
    },
    {
      "epoch": 1.5306472778320313e-05,
      "step": 16050,
      "training_step_time": 17.024229526519775
    },
    {
      "epoch": 1.5307426452636717e-05,
      "model_forward_time": 5.600738286972046,
      "step": 16051
    },
    {
      "epoch": 1.5307426452636717e-05,
      "step": 16051,
      "training_step_time": 17.060991287231445
    },
    {
      "epoch": 1.5308380126953125e-05,
      "model_forward_time": 5.974818706512451,
      "step": 16052
    },
    {
      "epoch": 1.5308380126953125e-05,
      "step": 16052,
      "training_step_time": 17.446809768676758
    },
    {
      "epoch": 1.5309333801269532e-05,
      "model_forward_time": 5.611917972564697,
      "step": 16053
    },
    {
      "epoch": 1.5309333801269532e-05,
      "step": 16053,
      "training_step_time": 17.09375238418579
    },
    {
      "epoch": 1.5310287475585936e-05,
      "model_forward_time": 5.60071325302124,
      "step": 16054
    },
    {
      "epoch": 1.5310287475585936e-05,
      "step": 16054,
      "training_step_time": 17.113792419433594
    },
    {
      "epoch": 1.5311241149902343e-05,
      "model_forward_time": 5.6095240116119385,
      "step": 16055
    },
    {
      "epoch": 1.5311241149902343e-05,
      "step": 16055,
      "training_step_time": 17.094938278198242
    },
    {
      "epoch": 1.531219482421875e-05,
      "model_forward_time": 5.6046411991119385,
      "step": 16056
    },
    {
      "epoch": 1.531219482421875e-05,
      "step": 16056,
      "training_step_time": 17.082584381103516
    },
    {
      "epoch": 1.5313148498535155e-05,
      "model_forward_time": 5.953430891036987,
      "step": 16057
    },
    {
      "epoch": 1.5313148498535155e-05,
      "step": 16057,
      "training_step_time": 17.402842044830322
    },
    {
      "epoch": 1.5314102172851562e-05,
      "model_forward_time": 5.605492115020752,
      "step": 16058
    },
    {
      "epoch": 1.5314102172851562e-05,
      "step": 16058,
      "training_step_time": 17.068451642990112
    },
    {
      "epoch": 1.531505584716797e-05,
      "model_forward_time": 5.605503082275391,
      "step": 16059
    },
    {
      "epoch": 1.531505584716797e-05,
      "step": 16059,
      "training_step_time": 17.074527740478516
    },
    {
      "epoch": 1.5316009521484374e-05,
      "grad_norm": 1.1114667654037476,
      "learning_rate": 1e-05,
      "loss": 0.0871,
      "step": 16060
    },
    {
      "epoch": 1.5316009521484374e-05,
      "model_forward_time": 5.6022608280181885,
      "step": 16060
    },
    {
      "dynamics_loss_avg": 0.07217588108032942,
      "epoch": 1.5316009521484374e-05,
      "step": 16060
    },
    {
      "action_loss_avg": 0.00717895213747397,
      "epoch": 1.5316009521484374e-05,
      "step": 16060
    },
    {
      "epoch": 1.5316009521484374e-05,
      "step": 16060,
      "training_step_time": 17.073511362075806
    },
    {
      "epoch": 1.531696319580078e-05,
      "model_forward_time": 5.595382213592529,
      "step": 16061
    },
    {
      "epoch": 1.531696319580078e-05,
      "step": 16061,
      "training_step_time": 17.04291534423828
    },
    {
      "epoch": 1.531791687011719e-05,
      "model_forward_time": 5.5994486808776855,
      "step": 16062
    },
    {
      "epoch": 1.531791687011719e-05,
      "step": 16062,
      "training_step_time": 17.047369241714478
    },
    {
      "epoch": 1.5318870544433593e-05,
      "model_forward_time": 5.984233856201172,
      "step": 16063
    },
    {
      "epoch": 1.5318870544433593e-05,
      "step": 16063,
      "training_step_time": 17.455043077468872
    },
    {
      "epoch": 1.531982421875e-05,
      "model_forward_time": 5.5999815464019775,
      "step": 16064
    },
    {
      "epoch": 1.531982421875e-05,
      "step": 16064,
      "training_step_time": 17.09364104270935
    },
    {
      "epoch": 1.5320777893066407e-05,
      "model_forward_time": 5.625146865844727,
      "step": 16065
    },
    {
      "epoch": 1.5320777893066407e-05,
      "step": 16065,
      "training_step_time": 17.10444927215576
    },
    {
      "epoch": 1.532173156738281e-05,
      "model_forward_time": 5.600206136703491,
      "step": 16066
    },
    {
      "epoch": 1.532173156738281e-05,
      "step": 16066,
      "training_step_time": 17.038838863372803
    },
    {
      "epoch": 1.532268524169922e-05,
      "model_forward_time": 5.576049327850342,
      "step": 16067
    },
    {
      "epoch": 1.532268524169922e-05,
      "step": 16067,
      "training_step_time": 16.99374485015869
    },
    {
      "epoch": 1.5323638916015626e-05,
      "model_forward_time": 5.5812294483184814,
      "step": 16068
    },
    {
      "epoch": 1.5323638916015626e-05,
      "step": 16068,
      "training_step_time": 17.35534381866455
    },
    {
      "epoch": 1.532459259033203e-05,
      "model_forward_time": 5.580537557601929,
      "step": 16069
    },
    {
      "epoch": 1.532459259033203e-05,
      "step": 16069,
      "training_step_time": 17.016973733901978
    },
    {
      "epoch": 1.5325546264648438e-05,
      "grad_norm": 0.9531896710395813,
      "learning_rate": 1e-05,
      "loss": 0.0539,
      "step": 16070
    },
    {
      "epoch": 1.5325546264648438e-05,
      "model_forward_time": 5.603334903717041,
      "step": 16070
    },
    {
      "dynamics_loss_avg": 0.05671958103775978,
      "epoch": 1.5325546264648438e-05,
      "step": 16070
    },
    {
      "action_loss_avg": 0.0056162959546782075,
      "epoch": 1.5325546264648438e-05,
      "step": 16070
    },
    {
      "epoch": 1.5325546264648438e-05,
      "step": 16070,
      "training_step_time": 17.044456958770752
    },
    {
      "epoch": 1.5326499938964845e-05,
      "model_forward_time": 5.604103088378906,
      "step": 16071
    },
    {
      "epoch": 1.5326499938964845e-05,
      "step": 16071,
      "training_step_time": 17.08063268661499
    },
    {
      "epoch": 1.532745361328125e-05,
      "model_forward_time": 5.600593090057373,
      "step": 16072
    },
    {
      "epoch": 1.532745361328125e-05,
      "step": 16072,
      "training_step_time": 17.070205211639404
    },
    {
      "epoch": 1.5328407287597657e-05,
      "model_forward_time": 5.602754354476929,
      "step": 16073
    },
    {
      "epoch": 1.5328407287597657e-05,
      "step": 16073,
      "training_step_time": 17.053613662719727
    },
    {
      "epoch": 1.5329360961914064e-05,
      "model_forward_time": 5.979569435119629,
      "step": 16074
    },
    {
      "epoch": 1.5329360961914064e-05,
      "step": 16074,
      "training_step_time": 17.430880784988403
    },
    {
      "epoch": 1.5330314636230468e-05,
      "model_forward_time": 5.564004182815552,
      "step": 16075
    },
    {
      "epoch": 1.5330314636230468e-05,
      "step": 16075,
      "training_step_time": 16.98900818824768
    },
    {
      "epoch": 1.5331268310546875e-05,
      "model_forward_time": 5.5658886432647705,
      "step": 16076
    },
    {
      "epoch": 1.5331268310546875e-05,
      "step": 16076,
      "training_step_time": 16.981526374816895
    },
    {
      "epoch": 1.5332221984863283e-05,
      "model_forward_time": 5.577630281448364,
      "step": 16077
    },
    {
      "epoch": 1.5332221984863283e-05,
      "step": 16077,
      "training_step_time": 16.981503009796143
    },
    {
      "epoch": 1.5333175659179687e-05,
      "model_forward_time": 5.568890333175659,
      "step": 16078
    },
    {
      "epoch": 1.5333175659179687e-05,
      "step": 16078,
      "training_step_time": 17.01086139678955
    },
    {
      "epoch": 1.5334129333496094e-05,
      "model_forward_time": 5.567305088043213,
      "step": 16079
    },
    {
      "epoch": 1.5334129333496094e-05,
      "step": 16079,
      "training_step_time": 17.349507808685303
    },
    {
      "epoch": 1.53350830078125e-05,
      "grad_norm": 0.5506647229194641,
      "learning_rate": 1e-05,
      "loss": 0.067,
      "step": 16080
    },
    {
      "epoch": 1.53350830078125e-05,
      "model_forward_time": 5.574225425720215,
      "step": 16080
    },
    {
      "dynamics_loss_avg": 0.05839763041585684,
      "epoch": 1.53350830078125e-05,
      "step": 16080
    },
    {
      "action_loss_avg": 0.007846434647217394,
      "epoch": 1.53350830078125e-05,
      "step": 16080
    },
    {
      "epoch": 1.53350830078125e-05,
      "step": 16080,
      "training_step_time": 17.01874566078186
    },
    {
      "epoch": 1.5336036682128906e-05,
      "model_forward_time": 5.592961311340332,
      "step": 16081
    },
    {
      "epoch": 1.5336036682128906e-05,
      "step": 16081,
      "training_step_time": 17.040210962295532
    },
    {
      "epoch": 1.5336990356445313e-05,
      "model_forward_time": 5.603456258773804,
      "step": 16082
    },
    {
      "epoch": 1.5336990356445313e-05,
      "step": 16082,
      "training_step_time": 17.0673508644104
    },
    {
      "epoch": 1.5337944030761717e-05,
      "model_forward_time": 5.594231128692627,
      "step": 16083
    },
    {
      "epoch": 1.5337944030761717e-05,
      "step": 16083,
      "training_step_time": 17.060233116149902
    },
    {
      "epoch": 1.5338897705078125e-05,
      "model_forward_time": 5.5993897914886475,
      "step": 16084
    },
    {
      "epoch": 1.5338897705078125e-05,
      "step": 16084,
      "training_step_time": 17.03956389427185
    },
    {
      "epoch": 1.5339851379394532e-05,
      "model_forward_time": 5.970548152923584,
      "step": 16085
    },
    {
      "epoch": 1.5339851379394532e-05,
      "step": 16085,
      "training_step_time": 17.399446725845337
    },
    {
      "epoch": 1.5340805053710936e-05,
      "model_forward_time": 5.595591068267822,
      "step": 16086
    },
    {
      "epoch": 1.5340805053710936e-05,
      "step": 16086,
      "training_step_time": 17.066197872161865
    },
    {
      "epoch": 1.5341758728027344e-05,
      "model_forward_time": 5.593595504760742,
      "step": 16087
    },
    {
      "epoch": 1.5341758728027344e-05,
      "step": 16087,
      "training_step_time": 17.03415274620056
    },
    {
      "epoch": 1.534271240234375e-05,
      "model_forward_time": 5.593462228775024,
      "step": 16088
    },
    {
      "epoch": 1.534271240234375e-05,
      "step": 16088,
      "training_step_time": 17.03866195678711
    },
    {
      "epoch": 1.5343666076660155e-05,
      "model_forward_time": 5.598138093948364,
      "step": 16089
    },
    {
      "epoch": 1.5343666076660155e-05,
      "step": 16089,
      "training_step_time": 17.024988889694214
    },
    {
      "epoch": 1.5344619750976562e-05,
      "grad_norm": 0.5328387022018433,
      "learning_rate": 1e-05,
      "loss": 0.0673,
      "step": 16090
    },
    {
      "epoch": 1.5344619750976562e-05,
      "model_forward_time": 5.978491306304932,
      "step": 16090
    },
    {
      "dynamics_loss_avg": 0.05554605219513178,
      "epoch": 1.5344619750976562e-05,
      "step": 16090
    },
    {
      "action_loss_avg": 0.008907217113301159,
      "epoch": 1.5344619750976562e-05,
      "step": 16090
    },
    {
      "epoch": 1.5344619750976562e-05,
      "step": 16090,
      "training_step_time": 17.45016837120056
    },
    {
      "epoch": 1.534557342529297e-05,
      "model_forward_time": 5.599957466125488,
      "step": 16091
    },
    {
      "epoch": 1.534557342529297e-05,
      "step": 16091,
      "training_step_time": 17.068065643310547
    },
    {
      "epoch": 1.5346527099609374e-05,
      "model_forward_time": 5.608262062072754,
      "step": 16092
    },
    {
      "epoch": 1.5346527099609374e-05,
      "step": 16092,
      "training_step_time": 17.08123016357422
    },
    {
      "epoch": 1.534748077392578e-05,
      "model_forward_time": 5.5990049839019775,
      "step": 16093
    },
    {
      "epoch": 1.534748077392578e-05,
      "step": 16093,
      "training_step_time": 17.03467297554016
    },
    {
      "epoch": 1.534843444824219e-05,
      "model_forward_time": 5.598150730133057,
      "step": 16094
    },
    {
      "epoch": 1.534843444824219e-05,
      "step": 16094,
      "training_step_time": 17.050211191177368
    },
    {
      "epoch": 1.5349388122558593e-05,
      "model_forward_time": 5.6020989418029785,
      "step": 16095
    },
    {
      "epoch": 1.5349388122558593e-05,
      "step": 16095,
      "training_step_time": 17.06391143798828
    },
    {
      "epoch": 1.5350341796875e-05,
      "model_forward_time": 5.970860242843628,
      "step": 16096
    },
    {
      "epoch": 1.5350341796875e-05,
      "step": 16096,
      "training_step_time": 17.421985864639282
    },
    {
      "epoch": 1.5351295471191407e-05,
      "model_forward_time": 5.606291055679321,
      "step": 16097
    },
    {
      "epoch": 1.5351295471191407e-05,
      "step": 16097,
      "training_step_time": 17.056289196014404
    },
    {
      "epoch": 1.535224914550781e-05,
      "model_forward_time": 5.597936391830444,
      "step": 16098
    },
    {
      "epoch": 1.535224914550781e-05,
      "step": 16098,
      "training_step_time": 17.04529619216919
    },
    {
      "epoch": 1.535320281982422e-05,
      "model_forward_time": 5.582329988479614,
      "step": 16099
    },
    {
      "epoch": 1.535320281982422e-05,
      "step": 16099,
      "training_step_time": 17.031229257583618
    },
    {
      "epoch": 1.5354156494140626e-05,
      "grad_norm": 0.43601682782173157,
      "learning_rate": 1e-05,
      "loss": 0.0732,
      "step": 16100
    },
    {
      "epoch": 1.5354156494140626e-05,
      "model_forward_time": 5.5894455909729,
      "step": 16100
    },
    {
      "dynamics_loss_avg": 0.06622926779091358,
      "epoch": 1.5354156494140626e-05,
      "step": 16100
    },
    {
      "action_loss_avg": 0.009801185759715735,
      "epoch": 1.5354156494140626e-05,
      "step": 16100
    },
    {
      "epoch": 1.5354156494140626e-05,
      "step": 16100,
      "training_step_time": 17.018232345581055
    },
    {
      "epoch": 1.535511016845703e-05,
      "model_forward_time": 5.587304592132568,
      "step": 16101
    },
    {
      "epoch": 1.535511016845703e-05,
      "step": 16101,
      "training_step_time": 17.40791344642639
    },
    {
      "epoch": 1.5356063842773438e-05,
      "model_forward_time": 5.594653129577637,
      "step": 16102
    },
    {
      "epoch": 1.5356063842773438e-05,
      "step": 16102,
      "training_step_time": 17.06113290786743
    },
    {
      "epoch": 1.5357017517089845e-05,
      "model_forward_time": 5.60464334487915,
      "step": 16103
    },
    {
      "epoch": 1.5357017517089845e-05,
      "step": 16103,
      "training_step_time": 17.0548734664917
    },
    {
      "epoch": 1.535797119140625e-05,
      "model_forward_time": 5.602900266647339,
      "step": 16104
    },
    {
      "epoch": 1.535797119140625e-05,
      "step": 16104,
      "training_step_time": 17.056782484054565
    },
    {
      "epoch": 1.5358924865722657e-05,
      "model_forward_time": 5.592711687088013,
      "step": 16105
    },
    {
      "epoch": 1.5358924865722657e-05,
      "step": 16105,
      "training_step_time": 17.027581214904785
    },
    {
      "epoch": 1.5359878540039064e-05,
      "model_forward_time": 5.5980000495910645,
      "step": 16106
    },
    {
      "epoch": 1.5359878540039064e-05,
      "step": 16106,
      "training_step_time": 17.040239810943604
    },
    {
      "epoch": 1.5360832214355468e-05,
      "model_forward_time": 5.975761651992798,
      "step": 16107
    },
    {
      "epoch": 1.5360832214355468e-05,
      "step": 16107,
      "training_step_time": 17.452778816223145
    },
    {
      "epoch": 1.5361785888671876e-05,
      "model_forward_time": 5.6392388343811035,
      "step": 16108
    },
    {
      "epoch": 1.5361785888671876e-05,
      "step": 16108,
      "training_step_time": 17.15491271018982
    },
    {
      "epoch": 1.5362739562988283e-05,
      "model_forward_time": 5.625669002532959,
      "step": 16109
    },
    {
      "epoch": 1.5362739562988283e-05,
      "step": 16109,
      "training_step_time": 17.1245756149292
    },
    {
      "epoch": 1.5363693237304687e-05,
      "grad_norm": 0.6688580513000488,
      "learning_rate": 1e-05,
      "loss": 0.0682,
      "step": 16110
    },
    {
      "epoch": 1.5363693237304687e-05,
      "model_forward_time": 5.605692386627197,
      "step": 16110
    },
    {
      "dynamics_loss_avg": 0.06276954710483551,
      "epoch": 1.5363693237304687e-05,
      "step": 16110
    },
    {
      "action_loss_avg": 0.007457206677645445,
      "epoch": 1.5363693237304687e-05,
      "step": 16110
    },
    {
      "epoch": 1.5363693237304687e-05,
      "step": 16110,
      "training_step_time": 17.093042612075806
    },
    {
      "epoch": 1.5364646911621094e-05,
      "model_forward_time": 5.6014628410339355,
      "step": 16111
    },
    {
      "epoch": 1.5364646911621094e-05,
      "step": 16111,
      "training_step_time": 17.093291997909546
    },
    {
      "epoch": 1.53656005859375e-05,
      "model_forward_time": 5.601781845092773,
      "step": 16112
    },
    {
      "epoch": 1.53656005859375e-05,
      "step": 16112,
      "training_step_time": 17.41917061805725
    },
    {
      "epoch": 1.5366554260253906e-05,
      "model_forward_time": 5.630255222320557,
      "step": 16113
    },
    {
      "epoch": 1.5366554260253906e-05,
      "step": 16113,
      "training_step_time": 17.111139059066772
    },
    {
      "epoch": 1.5367507934570313e-05,
      "model_forward_time": 5.607783555984497,
      "step": 16114
    },
    {
      "epoch": 1.5367507934570313e-05,
      "step": 16114,
      "training_step_time": 17.08505940437317
    },
    {
      "epoch": 1.5368461608886717e-05,
      "model_forward_time": 5.6076014041900635,
      "step": 16115
    },
    {
      "epoch": 1.5368461608886717e-05,
      "step": 16115,
      "training_step_time": 17.07672429084778
    },
    {
      "epoch": 1.5369415283203125e-05,
      "model_forward_time": 5.597952127456665,
      "step": 16116
    },
    {
      "epoch": 1.5369415283203125e-05,
      "step": 16116,
      "training_step_time": 17.07011890411377
    },
    {
      "epoch": 1.5370368957519532e-05,
      "model_forward_time": 5.606798410415649,
      "step": 16117
    },
    {
      "epoch": 1.5370368957519532e-05,
      "step": 16117,
      "training_step_time": 17.086684942245483
    },
    {
      "epoch": 1.5371322631835936e-05,
      "model_forward_time": 5.9893577098846436,
      "step": 16118
    },
    {
      "epoch": 1.5371322631835936e-05,
      "step": 16118,
      "training_step_time": 17.461714506149292
    },
    {
      "epoch": 1.5372276306152344e-05,
      "model_forward_time": 5.600164175033569,
      "step": 16119
    },
    {
      "epoch": 1.5372276306152344e-05,
      "step": 16119,
      "training_step_time": 17.065854787826538
    },
    {
      "epoch": 1.537322998046875e-05,
      "grad_norm": 0.2376091629266739,
      "learning_rate": 1e-05,
      "loss": 0.0586,
      "step": 16120
    },
    {
      "epoch": 1.537322998046875e-05,
      "model_forward_time": 5.598785877227783,
      "step": 16120
    },
    {
      "dynamics_loss_avg": 0.04457848761230707,
      "epoch": 1.537322998046875e-05,
      "step": 16120
    },
    {
      "action_loss_avg": 0.0054004306206479665,
      "epoch": 1.537322998046875e-05,
      "step": 16120
    },
    {
      "epoch": 1.537322998046875e-05,
      "step": 16120,
      "training_step_time": 17.046880960464478
    },
    {
      "epoch": 1.5374183654785155e-05,
      "model_forward_time": 5.590210437774658,
      "step": 16121
    },
    {
      "epoch": 1.5374183654785155e-05,
      "step": 16121,
      "training_step_time": 17.02917718887329
    },
    {
      "epoch": 1.5375137329101562e-05,
      "model_forward_time": 5.583653450012207,
      "step": 16122
    },
    {
      "epoch": 1.5375137329101562e-05,
      "step": 16122,
      "training_step_time": 17.06053900718689
    },
    {
      "epoch": 1.537609100341797e-05,
      "model_forward_time": 5.964563369750977,
      "step": 16123
    },
    {
      "epoch": 1.537609100341797e-05,
      "step": 16123,
      "training_step_time": 17.443844318389893
    },
    {
      "epoch": 1.5377044677734374e-05,
      "model_forward_time": 5.615599155426025,
      "step": 16124
    },
    {
      "epoch": 1.5377044677734374e-05,
      "step": 16124,
      "training_step_time": 17.12679100036621
    },
    {
      "epoch": 1.537799835205078e-05,
      "model_forward_time": 5.630211114883423,
      "step": 16125
    },
    {
      "epoch": 1.537799835205078e-05,
      "step": 16125,
      "training_step_time": 17.14348340034485
    },
    {
      "epoch": 1.537895202636719e-05,
      "model_forward_time": 5.610234498977661,
      "step": 16126
    },
    {
      "epoch": 1.537895202636719e-05,
      "step": 16126,
      "training_step_time": 17.124738216400146
    },
    {
      "epoch": 1.5379905700683593e-05,
      "model_forward_time": 5.628762245178223,
      "step": 16127
    },
    {
      "epoch": 1.5379905700683593e-05,
      "step": 16127,
      "training_step_time": 17.105175495147705
    },
    {
      "epoch": 1.5380859375e-05,
      "model_forward_time": 5.591297626495361,
      "step": 16128
    },
    {
      "epoch": 1.5380859375e-05,
      "step": 16128,
      "training_step_time": 17.005383729934692
    },
    {
      "epoch": 1.5381813049316408e-05,
      "model_forward_time": 5.960478782653809,
      "step": 16129
    },
    {
      "epoch": 1.5381813049316408e-05,
      "step": 16129,
      "training_step_time": 17.38724136352539
    },
    {
      "epoch": 1.538276672363281e-05,
      "grad_norm": 1.5207908153533936,
      "learning_rate": 1e-05,
      "loss": 0.0631,
      "step": 16130
    },
    {
      "epoch": 1.538276672363281e-05,
      "model_forward_time": 5.571814775466919,
      "step": 16130
    },
    {
      "dynamics_loss_avg": 0.05849075093865395,
      "epoch": 1.538276672363281e-05,
      "step": 16130
    },
    {
      "action_loss_avg": 0.0075279532466083765,
      "epoch": 1.538276672363281e-05,
      "step": 16130
    },
    {
      "epoch": 1.538276672363281e-05,
      "step": 16130,
      "training_step_time": 16.979943990707397
    },
    {
      "epoch": 1.538372039794922e-05,
      "model_forward_time": 5.569528102874756,
      "step": 16131
    },
    {
      "epoch": 1.538372039794922e-05,
      "step": 16131,
      "training_step_time": 16.998103380203247
    },
    {
      "epoch": 1.5384674072265626e-05,
      "model_forward_time": 5.575513601303101,
      "step": 16132
    },
    {
      "epoch": 1.5384674072265626e-05,
      "step": 16132,
      "training_step_time": 16.988768339157104
    },
    {
      "epoch": 1.538562774658203e-05,
      "model_forward_time": 5.5889410972595215,
      "step": 16133
    },
    {
      "epoch": 1.538562774658203e-05,
      "step": 16133,
      "training_step_time": 16.996383905410767
    },
    {
      "epoch": 1.5386581420898438e-05,
      "model_forward_time": 5.581072092056274,
      "step": 16134
    },
    {
      "epoch": 1.5386581420898438e-05,
      "step": 16134,
      "training_step_time": 17.384208917617798
    },
    {
      "epoch": 1.5387535095214845e-05,
      "model_forward_time": 5.5771801471710205,
      "step": 16135
    },
    {
      "epoch": 1.5387535095214845e-05,
      "step": 16135,
      "training_step_time": 17.01904320716858
    },
    {
      "epoch": 1.538848876953125e-05,
      "model_forward_time": 5.588435649871826,
      "step": 16136
    },
    {
      "epoch": 1.538848876953125e-05,
      "step": 16136,
      "training_step_time": 17.047086000442505
    },
    {
      "epoch": 1.5389442443847657e-05,
      "model_forward_time": 5.606299638748169,
      "step": 16137
    },
    {
      "epoch": 1.5389442443847657e-05,
      "step": 16137,
      "training_step_time": 17.110066652297974
    },
    {
      "epoch": 1.5390396118164064e-05,
      "model_forward_time": 5.6312336921691895,
      "step": 16138
    },
    {
      "epoch": 1.5390396118164064e-05,
      "step": 16138,
      "training_step_time": 17.16110348701477
    },
    {
      "epoch": 1.5391349792480468e-05,
      "model_forward_time": 5.628403902053833,
      "step": 16139
    },
    {
      "epoch": 1.5391349792480468e-05,
      "step": 16139,
      "training_step_time": 17.13651704788208
    },
    {
      "epoch": 1.5392303466796876e-05,
      "grad_norm": 0.6702704429626465,
      "learning_rate": 1e-05,
      "loss": 0.0627,
      "step": 16140
    },
    {
      "epoch": 1.5392303466796876e-05,
      "model_forward_time": 6.000788688659668,
      "step": 16140
    },
    {
      "dynamics_loss_avg": 0.0611370524391532,
      "epoch": 1.5392303466796876e-05,
      "step": 16140
    },
    {
      "action_loss_avg": 0.00811098339036107,
      "epoch": 1.5392303466796876e-05,
      "step": 16140
    },
    {
      "epoch": 1.5392303466796876e-05,
      "step": 16140,
      "training_step_time": 17.453697681427002
    },
    {
      "epoch": 1.539325714111328e-05,
      "model_forward_time": 5.581961154937744,
      "step": 16141
    },
    {
      "epoch": 1.539325714111328e-05,
      "step": 16141,
      "training_step_time": 16.99820113182068
    },
    {
      "epoch": 1.5394210815429687e-05,
      "model_forward_time": 5.559991836547852,
      "step": 16142
    },
    {
      "epoch": 1.5394210815429687e-05,
      "step": 16142,
      "training_step_time": 16.979095458984375
    },
    {
      "epoch": 1.5395164489746094e-05,
      "model_forward_time": 5.565246105194092,
      "step": 16143
    },
    {
      "epoch": 1.5395164489746094e-05,
      "step": 16143,
      "training_step_time": 16.983675479888916
    },
    {
      "epoch": 1.53961181640625e-05,
      "model_forward_time": 5.558337926864624,
      "step": 16144
    },
    {
      "epoch": 1.53961181640625e-05,
      "step": 16144,
      "training_step_time": 16.99338698387146
    },
    {
      "epoch": 1.5397071838378906e-05,
      "model_forward_time": 5.56161642074585,
      "step": 16145
    },
    {
      "epoch": 1.5397071838378906e-05,
      "step": 16145,
      "training_step_time": 17.336379051208496
    },
    {
      "epoch": 1.5398025512695313e-05,
      "model_forward_time": 5.561929702758789,
      "step": 16146
    },
    {
      "epoch": 1.5398025512695313e-05,
      "step": 16146,
      "training_step_time": 16.995304346084595
    },
    {
      "epoch": 1.5398979187011717e-05,
      "model_forward_time": 5.563726902008057,
      "step": 16147
    },
    {
      "epoch": 1.5398979187011717e-05,
      "step": 16147,
      "training_step_time": 16.985196113586426
    },
    {
      "epoch": 1.5399932861328125e-05,
      "model_forward_time": 5.569494247436523,
      "step": 16148
    },
    {
      "epoch": 1.5399932861328125e-05,
      "step": 16148,
      "training_step_time": 16.97356915473938
    },
    {
      "epoch": 1.5400886535644532e-05,
      "model_forward_time": 4.2476112842559814,
      "step": 16149
    },
    {
      "epoch": 1.5400886535644532e-05,
      "step": 16149,
      "training_step_time": 12.74353289604187
    },
    {
      "epoch": 1.5401840209960936e-05,
      "grad_norm": 1.8606669902801514,
      "learning_rate": 1e-05,
      "loss": 0.0691,
      "step": 16150
    },
    {
      "epoch": 1.5401840209960936e-05,
      "model_forward_time": 5.5942161083221436,
      "step": 16150
    },
    {
      "dynamics_loss_avg": 0.05200228877365589,
      "epoch": 1.5401840209960936e-05,
      "step": 16150
    },
    {
      "action_loss_avg": 0.009099694225005806,
      "epoch": 1.5401840209960936e-05,
      "step": 16150
    },
    {
      "epoch": 1.5401840209960936e-05,
      "step": 16150,
      "training_step_time": 17.031505346298218
    },
    {
      "epoch": 1.5402793884277344e-05,
      "model_forward_time": 5.9603447914123535,
      "step": 16151
    },
    {
      "epoch": 1.5402793884277344e-05,
      "step": 16151,
      "training_step_time": 17.402023553848267
    },
    {
      "epoch": 1.540374755859375e-05,
      "model_forward_time": 5.599698781967163,
      "step": 16152
    },
    {
      "epoch": 1.540374755859375e-05,
      "step": 16152,
      "training_step_time": 17.074827671051025
    },
    {
      "epoch": 1.5404701232910155e-05,
      "model_forward_time": 5.601003885269165,
      "step": 16153
    },
    {
      "epoch": 1.5404701232910155e-05,
      "step": 16153,
      "training_step_time": 17.09259557723999
    },
    {
      "epoch": 1.5405654907226562e-05,
      "model_forward_time": 5.611539602279663,
      "step": 16154
    },
    {
      "epoch": 1.5405654907226562e-05,
      "step": 16154,
      "training_step_time": 17.095098733901978
    },
    {
      "epoch": 1.540660858154297e-05,
      "model_forward_time": 5.593509674072266,
      "step": 16155
    },
    {
      "epoch": 1.540660858154297e-05,
      "step": 16155,
      "training_step_time": 17.009413957595825
    },
    {
      "epoch": 1.5407562255859374e-05,
      "model_forward_time": 5.567354440689087,
      "step": 16156
    },
    {
      "epoch": 1.5407562255859374e-05,
      "step": 16156,
      "training_step_time": 17.33485245704651
    },
    {
      "epoch": 1.540851593017578e-05,
      "model_forward_time": 5.5644800662994385,
      "step": 16157
    },
    {
      "epoch": 1.540851593017578e-05,
      "step": 16157,
      "training_step_time": 16.9872465133667
    },
    {
      "epoch": 1.540946960449219e-05,
      "model_forward_time": 5.5627641677856445,
      "step": 16158
    },
    {
      "epoch": 1.540946960449219e-05,
      "step": 16158,
      "training_step_time": 16.99497699737549
    },
    {
      "epoch": 1.5410423278808593e-05,
      "model_forward_time": 5.574325799942017,
      "step": 16159
    },
    {
      "epoch": 1.5410423278808593e-05,
      "step": 16159,
      "training_step_time": 16.99206519126892
    },
    {
      "epoch": 1.5411376953125e-05,
      "grad_norm": 0.8178063631057739,
      "learning_rate": 1e-05,
      "loss": 0.059,
      "step": 16160
    },
    {
      "epoch": 1.5411376953125e-05,
      "model_forward_time": 5.573068618774414,
      "step": 16160
    },
    {
      "dynamics_loss_avg": 0.05500415116548538,
      "epoch": 1.5411376953125e-05,
      "step": 16160
    },
    {
      "action_loss_avg": 0.005831143748946488,
      "epoch": 1.5411376953125e-05,
      "step": 16160
    },
    {
      "epoch": 1.5411376953125e-05,
      "step": 16160,
      "training_step_time": 16.98505735397339
    },
    {
      "epoch": 1.5412330627441408e-05,
      "model_forward_time": 5.5782790184021,
      "step": 16161
    },
    {
      "epoch": 1.5412330627441408e-05,
      "step": 16161,
      "training_step_time": 17.017963886260986
    },
    {
      "epoch": 1.541328430175781e-05,
      "model_forward_time": 5.968709707260132,
      "step": 16162
    },
    {
      "epoch": 1.541328430175781e-05,
      "step": 16162,
      "training_step_time": 17.421488523483276
    },
    {
      "epoch": 1.541423797607422e-05,
      "model_forward_time": 5.614329814910889,
      "step": 16163
    },
    {
      "epoch": 1.541423797607422e-05,
      "step": 16163,
      "training_step_time": 17.142332077026367
    },
    {
      "epoch": 1.5415191650390626e-05,
      "model_forward_time": 5.623166084289551,
      "step": 16164
    },
    {
      "epoch": 1.5415191650390626e-05,
      "step": 16164,
      "training_step_time": 17.131104946136475
    },
    {
      "epoch": 1.541614532470703e-05,
      "model_forward_time": 5.608536243438721,
      "step": 16165
    },
    {
      "epoch": 1.541614532470703e-05,
      "step": 16165,
      "training_step_time": 17.084941387176514
    },
    {
      "epoch": 1.5417098999023438e-05,
      "model_forward_time": 5.603130340576172,
      "step": 16166
    },
    {
      "epoch": 1.5417098999023438e-05,
      "step": 16166,
      "training_step_time": 17.048352241516113
    },
    {
      "epoch": 1.5418052673339845e-05,
      "model_forward_time": 5.96426248550415,
      "step": 16167
    },
    {
      "epoch": 1.5418052673339845e-05,
      "step": 16167,
      "training_step_time": 17.43086314201355
    },
    {
      "epoch": 1.541900634765625e-05,
      "model_forward_time": 5.604654788970947,
      "step": 16168
    },
    {
      "epoch": 1.541900634765625e-05,
      "step": 16168,
      "training_step_time": 17.06806445121765
    },
    {
      "epoch": 1.5419960021972657e-05,
      "model_forward_time": 5.600878477096558,
      "step": 16169
    },
    {
      "epoch": 1.5419960021972657e-05,
      "step": 16169,
      "training_step_time": 17.062663078308105
    },
    {
      "epoch": 1.5420913696289064e-05,
      "grad_norm": 0.4286801218986511,
      "learning_rate": 1e-05,
      "loss": 0.0557,
      "step": 16170
    },
    {
      "epoch": 1.5420913696289064e-05,
      "model_forward_time": 5.598847150802612,
      "step": 16170
    },
    {
      "dynamics_loss_avg": 0.04996808934956789,
      "epoch": 1.5420913696289064e-05,
      "step": 16170
    },
    {
      "action_loss_avg": 0.006194401974789798,
      "epoch": 1.5420913696289064e-05,
      "step": 16170
    },
    {
      "epoch": 1.5420913696289064e-05,
      "step": 16170,
      "training_step_time": 17.03568172454834
    },
    {
      "epoch": 1.5421867370605468e-05,
      "model_forward_time": 5.580689191818237,
      "step": 16171
    },
    {
      "epoch": 1.5421867370605468e-05,
      "step": 16171,
      "training_step_time": 16.99530577659607
    },
    {
      "epoch": 1.5422821044921876e-05,
      "model_forward_time": 5.582629442214966,
      "step": 16172
    },
    {
      "epoch": 1.5422821044921876e-05,
      "step": 16172,
      "training_step_time": 17.021689414978027
    },
    {
      "epoch": 1.542377471923828e-05,
      "model_forward_time": 5.975345611572266,
      "step": 16173
    },
    {
      "epoch": 1.542377471923828e-05,
      "step": 16173,
      "training_step_time": 17.45716404914856
    },
    {
      "epoch": 1.5424728393554687e-05,
      "model_forward_time": 5.623175859451294,
      "step": 16174
    },
    {
      "epoch": 1.5424728393554687e-05,
      "step": 16174,
      "training_step_time": 17.14230513572693
    },
    {
      "epoch": 1.5425682067871094e-05,
      "model_forward_time": 5.616506338119507,
      "step": 16175
    },
    {
      "epoch": 1.5425682067871094e-05,
      "step": 16175,
      "training_step_time": 17.08907127380371
    },
    {
      "epoch": 1.54266357421875e-05,
      "model_forward_time": 5.588809251785278,
      "step": 16176
    },
    {
      "epoch": 1.54266357421875e-05,
      "step": 16176,
      "training_step_time": 16.988903999328613
    },
    {
      "epoch": 1.5427589416503906e-05,
      "model_forward_time": 5.559796094894409,
      "step": 16177
    },
    {
      "epoch": 1.5427589416503906e-05,
      "step": 16177,
      "training_step_time": 16.98260498046875
    },
    {
      "epoch": 1.5428543090820313e-05,
      "model_forward_time": 5.909898519515991,
      "step": 16178
    },
    {
      "epoch": 1.5428543090820313e-05,
      "step": 16178,
      "training_step_time": 17.321852207183838
    },
    {
      "epoch": 1.5429496765136717e-05,
      "model_forward_time": 5.555412292480469,
      "step": 16179
    },
    {
      "epoch": 1.5429496765136717e-05,
      "step": 16179,
      "training_step_time": 16.965782642364502
    },
    {
      "epoch": 1.5430450439453125e-05,
      "grad_norm": 1.153180718421936,
      "learning_rate": 1e-05,
      "loss": 0.0594,
      "step": 16180
    },
    {
      "epoch": 1.5430450439453125e-05,
      "model_forward_time": 5.562469005584717,
      "step": 16180
    },
    {
      "dynamics_loss_avg": 0.05052513293921947,
      "epoch": 1.5430450439453125e-05,
      "step": 16180
    },
    {
      "action_loss_avg": 0.006729578087106347,
      "epoch": 1.5430450439453125e-05,
      "step": 16180
    },
    {
      "epoch": 1.5430450439453125e-05,
      "step": 16180,
      "training_step_time": 16.99230146408081
    },
    {
      "epoch": 1.5431404113769532e-05,
      "model_forward_time": 5.566580772399902,
      "step": 16181
    },
    {
      "epoch": 1.5431404113769532e-05,
      "step": 16181,
      "training_step_time": 16.999948740005493
    },
    {
      "epoch": 1.5432357788085936e-05,
      "model_forward_time": 5.580665826797485,
      "step": 16182
    },
    {
      "epoch": 1.5432357788085936e-05,
      "step": 16182,
      "training_step_time": 16.9917471408844
    },
    {
      "epoch": 1.5433311462402344e-05,
      "model_forward_time": 5.581574440002441,
      "step": 16183
    },
    {
      "epoch": 1.5433311462402344e-05,
      "step": 16183,
      "training_step_time": 17.02469825744629
    },
    {
      "epoch": 1.543426513671875e-05,
      "model_forward_time": 5.963935852050781,
      "step": 16184
    },
    {
      "epoch": 1.543426513671875e-05,
      "step": 16184,
      "training_step_time": 17.432656049728394
    },
    {
      "epoch": 1.5435218811035155e-05,
      "model_forward_time": 5.605753660202026,
      "step": 16185
    },
    {
      "epoch": 1.5435218811035155e-05,
      "step": 16185,
      "training_step_time": 17.082987070083618
    },
    {
      "epoch": 1.5436172485351562e-05,
      "model_forward_time": 5.604224920272827,
      "step": 16186
    },
    {
      "epoch": 1.5436172485351562e-05,
      "step": 16186,
      "training_step_time": 17.077436447143555
    },
    {
      "epoch": 1.543712615966797e-05,
      "model_forward_time": 5.596593856811523,
      "step": 16187
    },
    {
      "epoch": 1.543712615966797e-05,
      "step": 16187,
      "training_step_time": 17.06137776374817
    },
    {
      "epoch": 1.5438079833984374e-05,
      "model_forward_time": 5.594730377197266,
      "step": 16188
    },
    {
      "epoch": 1.5438079833984374e-05,
      "step": 16188,
      "training_step_time": 17.050872802734375
    },
    {
      "epoch": 1.543903350830078e-05,
      "model_forward_time": 5.957809209823608,
      "step": 16189
    },
    {
      "epoch": 1.543903350830078e-05,
      "step": 16189,
      "training_step_time": 17.464770793914795
    },
    {
      "epoch": 1.543998718261719e-05,
      "grad_norm": 1.216444969177246,
      "learning_rate": 1e-05,
      "loss": 0.0655,
      "step": 16190
    },
    {
      "epoch": 1.543998718261719e-05,
      "model_forward_time": 5.6265177726745605,
      "step": 16190
    },
    {
      "dynamics_loss_avg": 0.05751091036945581,
      "epoch": 1.543998718261719e-05,
      "step": 16190
    },
    {
      "action_loss_avg": 0.007068974059075117,
      "epoch": 1.543998718261719e-05,
      "step": 16190
    },
    {
      "epoch": 1.543998718261719e-05,
      "step": 16190,
      "training_step_time": 17.149243593215942
    },
    {
      "epoch": 1.5440940856933593e-05,
      "model_forward_time": 5.621135711669922,
      "step": 16191
    },
    {
      "epoch": 1.5440940856933593e-05,
      "step": 16191,
      "training_step_time": 17.130988121032715
    },
    {
      "epoch": 1.544189453125e-05,
      "model_forward_time": 5.605517148971558,
      "step": 16192
    },
    {
      "epoch": 1.544189453125e-05,
      "step": 16192,
      "training_step_time": 17.0856716632843
    },
    {
      "epoch": 1.5442848205566408e-05,
      "model_forward_time": 5.618505239486694,
      "step": 16193
    },
    {
      "epoch": 1.5442848205566408e-05,
      "step": 16193,
      "training_step_time": 17.1040301322937
    },
    {
      "epoch": 1.544380187988281e-05,
      "model_forward_time": 5.614435911178589,
      "step": 16194
    },
    {
      "epoch": 1.544380187988281e-05,
      "step": 16194,
      "training_step_time": 17.488810300827026
    },
    {
      "epoch": 1.544475555419922e-05,
      "model_forward_time": 5.614158391952515,
      "step": 16195
    },
    {
      "epoch": 1.544475555419922e-05,
      "step": 16195,
      "training_step_time": 17.097293615341187
    },
    {
      "epoch": 1.5445709228515626e-05,
      "model_forward_time": 5.607333660125732,
      "step": 16196
    },
    {
      "epoch": 1.5445709228515626e-05,
      "step": 16196,
      "training_step_time": 17.090954303741455
    },
    {
      "epoch": 1.544666290283203e-05,
      "model_forward_time": 5.60192084312439,
      "step": 16197
    },
    {
      "epoch": 1.544666290283203e-05,
      "step": 16197,
      "training_step_time": 17.076289415359497
    },
    {
      "epoch": 1.5447616577148438e-05,
      "model_forward_time": 5.598229885101318,
      "step": 16198
    },
    {
      "epoch": 1.5447616577148438e-05,
      "step": 16198,
      "training_step_time": 17.050992250442505
    },
    {
      "epoch": 1.5448570251464845e-05,
      "model_forward_time": 5.589370489120483,
      "step": 16199
    },
    {
      "epoch": 1.5448570251464845e-05,
      "step": 16199,
      "training_step_time": 17.019872426986694
    },
    {
      "epoch": 1.544952392578125e-05,
      "grad_norm": 0.6010753512382507,
      "learning_rate": 1e-05,
      "loss": 0.0574,
      "step": 16200
    },
    {
      "epoch": 1.544952392578125e-05,
      "model_forward_time": 5.963749170303345,
      "step": 16200
    },
    {
      "dynamics_loss_avg": 0.04863460026681423,
      "epoch": 1.544952392578125e-05,
      "step": 16200
    },
    {
      "action_loss_avg": 0.008463734108954669,
      "epoch": 1.544952392578125e-05,
      "step": 16200
    },
    {
      "epoch": 1.544952392578125e-05,
      "step": 16200,
      "training_step_time": 17.411582708358765
    },
    {
      "epoch": 1.5450477600097657e-05,
      "model_forward_time": 5.5942769050598145,
      "step": 16201
    },
    {
      "epoch": 1.5450477600097657e-05,
      "step": 16201,
      "training_step_time": 17.077195167541504
    },
    {
      "epoch": 1.5451431274414064e-05,
      "model_forward_time": 5.617365598678589,
      "step": 16202
    },
    {
      "epoch": 1.5451431274414064e-05,
      "step": 16202,
      "training_step_time": 17.136247396469116
    },
    {
      "epoch": 1.5452384948730468e-05,
      "model_forward_time": 5.6209447383880615,
      "step": 16203
    },
    {
      "epoch": 1.5452384948730468e-05,
      "step": 16203,
      "training_step_time": 17.136879205703735
    },
    {
      "epoch": 1.5453338623046876e-05,
      "model_forward_time": 5.6212098598480225,
      "step": 16204
    },
    {
      "epoch": 1.5453338623046876e-05,
      "step": 16204,
      "training_step_time": 17.137564182281494
    },
    {
      "epoch": 1.545429229736328e-05,
      "model_forward_time": 5.6086955070495605,
      "step": 16205
    },
    {
      "epoch": 1.545429229736328e-05,
      "step": 16205,
      "training_step_time": 17.444757223129272
    },
    {
      "epoch": 1.5455245971679687e-05,
      "model_forward_time": 5.589998483657837,
      "step": 16206
    },
    {
      "epoch": 1.5455245971679687e-05,
      "step": 16206,
      "training_step_time": 17.037328481674194
    },
    {
      "epoch": 1.5456199645996094e-05,
      "model_forward_time": 5.594862937927246,
      "step": 16207
    },
    {
      "epoch": 1.5456199645996094e-05,
      "step": 16207,
      "training_step_time": 17.03070616722107
    },
    {
      "epoch": 1.54571533203125e-05,
      "model_forward_time": 5.592771768569946,
      "step": 16208
    },
    {
      "epoch": 1.54571533203125e-05,
      "step": 16208,
      "training_step_time": 17.046815633773804
    },
    {
      "epoch": 1.5458106994628906e-05,
      "model_forward_time": 5.596236705780029,
      "step": 16209
    },
    {
      "epoch": 1.5458106994628906e-05,
      "step": 16209,
      "training_step_time": 17.037106037139893
    },
    {
      "epoch": 1.5459060668945313e-05,
      "grad_norm": 0.9361350536346436,
      "learning_rate": 1e-05,
      "loss": 0.0465,
      "step": 16210
    },
    {
      "epoch": 1.5459060668945313e-05,
      "model_forward_time": 5.589877367019653,
      "step": 16210
    },
    {
      "dynamics_loss_avg": 0.048284895904362204,
      "epoch": 1.5459060668945313e-05,
      "step": 16210
    },
    {
      "action_loss_avg": 0.005752514163032174,
      "epoch": 1.5459060668945313e-05,
      "step": 16210
    },
    {
      "epoch": 1.5459060668945313e-05,
      "step": 16210,
      "training_step_time": 17.028839588165283
    },
    {
      "epoch": 1.5460014343261717e-05,
      "model_forward_time": 5.944994211196899,
      "step": 16211
    },
    {
      "epoch": 1.5460014343261717e-05,
      "step": 16211,
      "training_step_time": 17.39094090461731
    },
    {
      "epoch": 1.5460968017578125e-05,
      "model_forward_time": 5.604103088378906,
      "step": 16212
    },
    {
      "epoch": 1.5460968017578125e-05,
      "step": 16212,
      "training_step_time": 17.086359977722168
    },
    {
      "epoch": 1.5461921691894532e-05,
      "model_forward_time": 5.612915277481079,
      "step": 16213
    },
    {
      "epoch": 1.5461921691894532e-05,
      "step": 16213,
      "training_step_time": 17.095791578292847
    },
    {
      "epoch": 1.5462875366210936e-05,
      "model_forward_time": 5.60283350944519,
      "step": 16214
    },
    {
      "epoch": 1.5462875366210936e-05,
      "step": 16214,
      "training_step_time": 17.106128454208374
    },
    {
      "epoch": 1.5463829040527344e-05,
      "model_forward_time": 5.626567125320435,
      "step": 16215
    },
    {
      "epoch": 1.5463829040527344e-05,
      "step": 16215,
      "training_step_time": 17.124260425567627
    },
    {
      "epoch": 1.546478271484375e-05,
      "model_forward_time": 5.999161243438721,
      "step": 16216
    },
    {
      "epoch": 1.546478271484375e-05,
      "step": 16216,
      "training_step_time": 17.494715690612793
    },
    {
      "epoch": 1.5465736389160155e-05,
      "model_forward_time": 5.594195604324341,
      "step": 16217
    },
    {
      "epoch": 1.5465736389160155e-05,
      "step": 16217,
      "training_step_time": 17.022559642791748
    },
    {
      "epoch": 1.5466690063476562e-05,
      "model_forward_time": 5.5856592655181885,
      "step": 16218
    },
    {
      "epoch": 1.5466690063476562e-05,
      "step": 16218,
      "training_step_time": 17.006483793258667
    },
    {
      "epoch": 1.546764373779297e-05,
      "model_forward_time": 5.574419975280762,
      "step": 16219
    },
    {
      "epoch": 1.546764373779297e-05,
      "step": 16219,
      "training_step_time": 17.0267972946167
    },
    {
      "epoch": 1.5468597412109374e-05,
      "grad_norm": 0.4748166799545288,
      "learning_rate": 1e-05,
      "loss": 0.0684,
      "step": 16220
    },
    {
      "epoch": 1.5468597412109374e-05,
      "model_forward_time": 5.577284336090088,
      "step": 16220
    },
    {
      "dynamics_loss_avg": 0.06329848766326904,
      "epoch": 1.5468597412109374e-05,
      "step": 16220
    },
    {
      "action_loss_avg": 0.007899284083396196,
      "epoch": 1.5468597412109374e-05,
      "step": 16220
    },
    {
      "epoch": 1.5468597412109374e-05,
      "step": 16220,
      "training_step_time": 16.9907124042511
    },
    {
      "epoch": 1.546955108642578e-05,
      "model_forward_time": 5.5771565437316895,
      "step": 16221
    },
    {
      "epoch": 1.546955108642578e-05,
      "step": 16221,
      "training_step_time": 16.993702173233032
    },
    {
      "epoch": 1.547050476074219e-05,
      "model_forward_time": 5.938875436782837,
      "step": 16222
    },
    {
      "epoch": 1.547050476074219e-05,
      "step": 16222,
      "training_step_time": 17.365591287612915
    },
    {
      "epoch": 1.5471458435058593e-05,
      "model_forward_time": 5.578466415405273,
      "step": 16223
    },
    {
      "epoch": 1.5471458435058593e-05,
      "step": 16223,
      "training_step_time": 17.02478528022766
    },
    {
      "epoch": 1.5472412109375e-05,
      "model_forward_time": 5.593538522720337,
      "step": 16224
    },
    {
      "epoch": 1.5472412109375e-05,
      "step": 16224,
      "training_step_time": 17.05353569984436
    },
    {
      "epoch": 1.5473365783691408e-05,
      "model_forward_time": 5.600637197494507,
      "step": 16225
    },
    {
      "epoch": 1.5473365783691408e-05,
      "step": 16225,
      "training_step_time": 17.093210697174072
    },
    {
      "epoch": 1.547431945800781e-05,
      "model_forward_time": 5.612565040588379,
      "step": 16226
    },
    {
      "epoch": 1.547431945800781e-05,
      "step": 16226,
      "training_step_time": 17.120516538619995
    },
    {
      "epoch": 1.547527313232422e-05,
      "model_forward_time": 6.000427722930908,
      "step": 16227
    },
    {
      "epoch": 1.547527313232422e-05,
      "step": 16227,
      "training_step_time": 17.504531621932983
    },
    {
      "epoch": 1.5476226806640626e-05,
      "model_forward_time": 5.604884147644043,
      "step": 16228
    },
    {
      "epoch": 1.5476226806640626e-05,
      "step": 16228,
      "training_step_time": 17.087473392486572
    },
    {
      "epoch": 1.547718048095703e-05,
      "model_forward_time": 5.611412286758423,
      "step": 16229
    },
    {
      "epoch": 1.547718048095703e-05,
      "step": 16229,
      "training_step_time": 17.09240198135376
    },
    {
      "epoch": 1.5478134155273438e-05,
      "grad_norm": 0.7218204736709595,
      "learning_rate": 1e-05,
      "loss": 0.0677,
      "step": 16230
    },
    {
      "epoch": 1.5478134155273438e-05,
      "model_forward_time": 5.605771780014038,
      "step": 16230
    },
    {
      "dynamics_loss_avg": 0.05479388236999512,
      "epoch": 1.5478134155273438e-05,
      "step": 16230
    },
    {
      "action_loss_avg": 0.006368475896306336,
      "epoch": 1.5478134155273438e-05,
      "step": 16230
    },
    {
      "epoch": 1.5478134155273438e-05,
      "step": 16230,
      "training_step_time": 17.092756748199463
    },
    {
      "epoch": 1.5479087829589845e-05,
      "model_forward_time": 5.613475561141968,
      "step": 16231
    },
    {
      "epoch": 1.5479087829589845e-05,
      "step": 16231,
      "training_step_time": 17.1019344329834
    },
    {
      "epoch": 1.548004150390625e-05,
      "model_forward_time": 5.613551139831543,
      "step": 16232
    },
    {
      "epoch": 1.548004150390625e-05,
      "step": 16232,
      "training_step_time": 17.08051371574402
    },
    {
      "epoch": 1.5480995178222657e-05,
      "model_forward_time": 5.955289125442505,
      "step": 16233
    },
    {
      "epoch": 1.5480995178222657e-05,
      "step": 16233,
      "training_step_time": 17.41182780265808
    },
    {
      "epoch": 1.5481948852539064e-05,
      "model_forward_time": 5.6035661697387695,
      "step": 16234
    },
    {
      "epoch": 1.5481948852539064e-05,
      "step": 16234,
      "training_step_time": 17.036494731903076
    },
    {
      "epoch": 1.5482902526855468e-05,
      "model_forward_time": 5.592638969421387,
      "step": 16235
    },
    {
      "epoch": 1.5482902526855468e-05,
      "step": 16235,
      "training_step_time": 17.035247564315796
    },
    {
      "epoch": 1.5483856201171876e-05,
      "model_forward_time": 5.591902017593384,
      "step": 16236
    },
    {
      "epoch": 1.5483856201171876e-05,
      "step": 16236,
      "training_step_time": 17.036396026611328
    },
    {
      "epoch": 1.548480987548828e-05,
      "model_forward_time": 5.603589057922363,
      "step": 16237
    },
    {
      "epoch": 1.548480987548828e-05,
      "step": 16237,
      "training_step_time": 17.081599950790405
    },
    {
      "epoch": 1.5485763549804687e-05,
      "model_forward_time": 5.986187696456909,
      "step": 16238
    },
    {
      "epoch": 1.5485763549804687e-05,
      "step": 16238,
      "training_step_time": 17.49698281288147
    },
    {
      "epoch": 1.5486717224121094e-05,
      "model_forward_time": 5.628220796585083,
      "step": 16239
    },
    {
      "epoch": 1.5486717224121094e-05,
      "step": 16239,
      "training_step_time": 17.13276767730713
    },
    {
      "epoch": 1.54876708984375e-05,
      "grad_norm": 1.2357054948806763,
      "learning_rate": 1e-05,
      "loss": 0.056,
      "step": 16240
    },
    {
      "epoch": 1.54876708984375e-05,
      "model_forward_time": 5.614001512527466,
      "step": 16240
    },
    {
      "dynamics_loss_avg": 0.0499729061499238,
      "epoch": 1.54876708984375e-05,
      "step": 16240
    },
    {
      "action_loss_avg": 0.006966012774500996,
      "epoch": 1.54876708984375e-05,
      "step": 16240
    },
    {
      "epoch": 1.54876708984375e-05,
      "step": 16240,
      "training_step_time": 17.077350616455078
    },
    {
      "epoch": 1.5488624572753906e-05,
      "model_forward_time": 5.59599757194519,
      "step": 16241
    },
    {
      "epoch": 1.5488624572753906e-05,
      "step": 16241,
      "training_step_time": 17.034278869628906
    },
    {
      "epoch": 1.5489578247070313e-05,
      "model_forward_time": 5.586757659912109,
      "step": 16242
    },
    {
      "epoch": 1.5489578247070313e-05,
      "step": 16242,
      "training_step_time": 17.037517070770264
    },
    {
      "epoch": 1.5490531921386717e-05,
      "model_forward_time": 5.963155269622803,
      "step": 16243
    },
    {
      "epoch": 1.5490531921386717e-05,
      "step": 16243,
      "training_step_time": 17.42780637741089
    },
    {
      "epoch": 1.5491485595703125e-05,
      "model_forward_time": 5.612142562866211,
      "step": 16244
    },
    {
      "epoch": 1.5491485595703125e-05,
      "step": 16244,
      "training_step_time": 17.09186029434204
    },
    {
      "epoch": 1.5492439270019532e-05,
      "model_forward_time": 5.6205198764801025,
      "step": 16245
    },
    {
      "epoch": 1.5492439270019532e-05,
      "step": 16245,
      "training_step_time": 17.127056121826172
    },
    {
      "epoch": 1.5493392944335936e-05,
      "model_forward_time": 5.621664762496948,
      "step": 16246
    },
    {
      "epoch": 1.5493392944335936e-05,
      "step": 16246,
      "training_step_time": 17.147716999053955
    },
    {
      "epoch": 1.5494346618652344e-05,
      "model_forward_time": 5.62825345993042,
      "step": 16247
    },
    {
      "epoch": 1.5494346618652344e-05,
      "step": 16247,
      "training_step_time": 17.14127826690674
    },
    {
      "epoch": 1.549530029296875e-05,
      "model_forward_time": 5.614049911499023,
      "step": 16248
    },
    {
      "epoch": 1.549530029296875e-05,
      "step": 16248,
      "training_step_time": 17.109055042266846
    },
    {
      "epoch": 1.5496253967285155e-05,
      "model_forward_time": 5.996322870254517,
      "step": 16249
    },
    {
      "epoch": 1.5496253967285155e-05,
      "step": 16249,
      "training_step_time": 17.471688985824585
    },
    {
      "epoch": 1.5497207641601562e-05,
      "grad_norm": 1.0880954265594482,
      "learning_rate": 1e-05,
      "loss": 0.0813,
      "step": 16250
    },
    {
      "epoch": 1.5497207641601562e-05,
      "model_forward_time": 5.599413633346558,
      "step": 16250
    },
    {
      "dynamics_loss_avg": 0.06940657505765557,
      "epoch": 1.5497207641601562e-05,
      "step": 16250
    },
    {
      "action_loss_avg": 0.010313617123756558,
      "epoch": 1.5497207641601562e-05,
      "step": 16250
    },
    {
      "epoch": 1.5497207641601562e-05,
      "step": 16250,
      "training_step_time": 17.05400824546814
    },
    {
      "epoch": 1.549816131591797e-05,
      "model_forward_time": 5.585345029830933,
      "step": 16251
    },
    {
      "epoch": 1.549816131591797e-05,
      "step": 16251,
      "training_step_time": 16.998196840286255
    },
    {
      "epoch": 1.5499114990234374e-05,
      "model_forward_time": 5.589262962341309,
      "step": 16252
    },
    {
      "epoch": 1.5499114990234374e-05,
      "step": 16252,
      "training_step_time": 17.027602434158325
    },
    {
      "epoch": 1.550006866455078e-05,
      "model_forward_time": 5.604278326034546,
      "step": 16253
    },
    {
      "epoch": 1.550006866455078e-05,
      "step": 16253,
      "training_step_time": 17.061317682266235
    },
    {
      "epoch": 1.550102233886719e-05,
      "model_forward_time": 5.974607706069946,
      "step": 16254
    },
    {
      "epoch": 1.550102233886719e-05,
      "step": 16254,
      "training_step_time": 17.452389001846313
    },
    {
      "epoch": 1.5501976013183593e-05,
      "model_forward_time": 5.6217710971832275,
      "step": 16255
    },
    {
      "epoch": 1.5501976013183593e-05,
      "step": 16255,
      "training_step_time": 17.136728048324585
    },
    {
      "epoch": 1.55029296875e-05,
      "model_forward_time": 5.62479305267334,
      "step": 16256
    },
    {
      "epoch": 1.55029296875e-05,
      "step": 16256,
      "training_step_time": 17.12539029121399
    },
    {
      "epoch": 1.5503883361816408e-05,
      "model_forward_time": 5.610291242599487,
      "step": 16257
    },
    {
      "epoch": 1.5503883361816408e-05,
      "step": 16257,
      "training_step_time": 17.10709547996521
    },
    {
      "epoch": 1.550483703613281e-05,
      "model_forward_time": 5.603923082351685,
      "step": 16258
    },
    {
      "epoch": 1.550483703613281e-05,
      "step": 16258,
      "training_step_time": 17.089093446731567
    },
    {
      "epoch": 1.550579071044922e-05,
      "model_forward_time": 5.606956243515015,
      "step": 16259
    },
    {
      "epoch": 1.550579071044922e-05,
      "step": 16259,
      "training_step_time": 17.072609424591064
    },
    {
      "epoch": 1.5506744384765626e-05,
      "grad_norm": 0.6193937659263611,
      "learning_rate": 1e-05,
      "loss": 0.0628,
      "step": 16260
    },
    {
      "epoch": 1.5506744384765626e-05,
      "model_forward_time": 5.973653316497803,
      "step": 16260
    },
    {
      "dynamics_loss_avg": 0.05389710795134306,
      "epoch": 1.5506744384765626e-05,
      "step": 16260
    },
    {
      "action_loss_avg": 0.006960705388337374,
      "epoch": 1.5506744384765626e-05,
      "step": 16260
    },
    {
      "epoch": 1.5506744384765626e-05,
      "step": 16260,
      "training_step_time": 17.448408365249634
    },
    {
      "epoch": 1.550769805908203e-05,
      "model_forward_time": 5.620716094970703,
      "step": 16261
    },
    {
      "epoch": 1.550769805908203e-05,
      "step": 16261,
      "training_step_time": 17.13778519630432
    },
    {
      "epoch": 1.5508651733398438e-05,
      "model_forward_time": 5.625765323638916,
      "step": 16262
    },
    {
      "epoch": 1.5508651733398438e-05,
      "step": 16262,
      "training_step_time": 17.123567581176758
    },
    {
      "epoch": 1.5509605407714845e-05,
      "model_forward_time": 5.599571228027344,
      "step": 16263
    },
    {
      "epoch": 1.5509605407714845e-05,
      "step": 16263,
      "training_step_time": 17.056885719299316
    },
    {
      "epoch": 1.551055908203125e-05,
      "model_forward_time": 5.590420246124268,
      "step": 16264
    },
    {
      "epoch": 1.551055908203125e-05,
      "step": 16264,
      "training_step_time": 17.012237548828125
    },
    {
      "epoch": 1.5511512756347657e-05,
      "model_forward_time": 5.590298175811768,
      "step": 16265
    },
    {
      "epoch": 1.5511512756347657e-05,
      "step": 16265,
      "training_step_time": 17.39873242378235
    },
    {
      "epoch": 1.551246643066406e-05,
      "model_forward_time": 5.583471775054932,
      "step": 16266
    },
    {
      "epoch": 1.551246643066406e-05,
      "step": 16266,
      "training_step_time": 17.02822780609131
    },
    {
      "epoch": 1.5513420104980468e-05,
      "model_forward_time": 5.5984203815460205,
      "step": 16267
    },
    {
      "epoch": 1.5513420104980468e-05,
      "step": 16267,
      "training_step_time": 17.073652982711792
    },
    {
      "epoch": 1.5514373779296876e-05,
      "model_forward_time": 5.5999157428741455,
      "step": 16268
    },
    {
      "epoch": 1.5514373779296876e-05,
      "step": 16268,
      "training_step_time": 17.074565649032593
    },
    {
      "epoch": 1.551532745361328e-05,
      "model_forward_time": 5.60428786277771,
      "step": 16269
    },
    {
      "epoch": 1.551532745361328e-05,
      "step": 16269,
      "training_step_time": 17.080933809280396
    },
    {
      "epoch": 1.5516281127929687e-05,
      "grad_norm": 0.4847153127193451,
      "learning_rate": 1e-05,
      "loss": 0.0561,
      "step": 16270
    },
    {
      "epoch": 1.5516281127929687e-05,
      "model_forward_time": 5.611431121826172,
      "step": 16270
    },
    {
      "dynamics_loss_avg": 0.052153496723622086,
      "epoch": 1.5516281127929687e-05,
      "step": 16270
    },
    {
      "action_loss_avg": 0.005637715896591544,
      "epoch": 1.5516281127929687e-05,
      "step": 16270
    },
    {
      "epoch": 1.5516281127929687e-05,
      "step": 16270,
      "training_step_time": 17.087948083877563
    },
    {
      "epoch": 1.5517234802246095e-05,
      "model_forward_time": 5.987950325012207,
      "step": 16271
    },
    {
      "epoch": 1.5517234802246095e-05,
      "step": 16271,
      "training_step_time": 17.471547603607178
    },
    {
      "epoch": 1.55181884765625e-05,
      "model_forward_time": 5.611643075942993,
      "step": 16272
    },
    {
      "epoch": 1.55181884765625e-05,
      "step": 16272,
      "training_step_time": 17.085580825805664
    },
    {
      "epoch": 1.5519142150878906e-05,
      "model_forward_time": 5.60719895362854,
      "step": 16273
    },
    {
      "epoch": 1.5519142150878906e-05,
      "step": 16273,
      "training_step_time": 17.08119225502014
    },
    {
      "epoch": 1.5520095825195313e-05,
      "model_forward_time": 5.599534273147583,
      "step": 16274
    },
    {
      "epoch": 1.5520095825195313e-05,
      "step": 16274,
      "training_step_time": 17.08263874053955
    },
    {
      "epoch": 1.5521049499511717e-05,
      "model_forward_time": 5.603361368179321,
      "step": 16275
    },
    {
      "epoch": 1.5521049499511717e-05,
      "step": 16275,
      "training_step_time": 17.091394424438477
    },
    {
      "epoch": 1.5522003173828125e-05,
      "model_forward_time": 5.974754810333252,
      "step": 16276
    },
    {
      "epoch": 1.5522003173828125e-05,
      "step": 16276,
      "training_step_time": 17.479800701141357
    },
    {
      "epoch": 1.5522956848144532e-05,
      "model_forward_time": 5.64141058921814,
      "step": 16277
    },
    {
      "epoch": 1.5522956848144532e-05,
      "step": 16277,
      "training_step_time": 17.153968572616577
    },
    {
      "epoch": 1.5523910522460936e-05,
      "model_forward_time": 5.624318361282349,
      "step": 16278
    },
    {
      "epoch": 1.5523910522460936e-05,
      "step": 16278,
      "training_step_time": 17.12994909286499
    },
    {
      "epoch": 1.5524864196777344e-05,
      "model_forward_time": 5.621563673019409,
      "step": 16279
    },
    {
      "epoch": 1.5524864196777344e-05,
      "step": 16279,
      "training_step_time": 17.116754055023193
    },
    {
      "epoch": 1.552581787109375e-05,
      "grad_norm": 1.1168855428695679,
      "learning_rate": 1e-05,
      "loss": 0.0637,
      "step": 16280
    },
    {
      "epoch": 1.552581787109375e-05,
      "model_forward_time": 5.613076210021973,
      "step": 16280
    },
    {
      "dynamics_loss_avg": 0.05259674247354269,
      "epoch": 1.552581787109375e-05,
      "step": 16280
    },
    {
      "action_loss_avg": 0.007797381584532559,
      "epoch": 1.552581787109375e-05,
      "step": 16280
    },
    {
      "epoch": 1.552581787109375e-05,
      "step": 16280,
      "training_step_time": 17.093873500823975
    },
    {
      "epoch": 1.5526771545410155e-05,
      "model_forward_time": 5.584982872009277,
      "step": 16281
    },
    {
      "epoch": 1.5526771545410155e-05,
      "step": 16281,
      "training_step_time": 17.005406141281128
    },
    {
      "epoch": 1.5527725219726563e-05,
      "model_forward_time": 5.961395502090454,
      "step": 16282
    },
    {
      "epoch": 1.5527725219726563e-05,
      "step": 16282,
      "training_step_time": 17.368077993392944
    },
    {
      "epoch": 1.552867889404297e-05,
      "model_forward_time": 5.563342332839966,
      "step": 16283
    },
    {
      "epoch": 1.552867889404297e-05,
      "step": 16283,
      "training_step_time": 16.995019674301147
    },
    {
      "epoch": 1.5529632568359374e-05,
      "model_forward_time": 5.570069789886475,
      "step": 16284
    },
    {
      "epoch": 1.5529632568359374e-05,
      "step": 16284,
      "training_step_time": 16.992931842803955
    },
    {
      "epoch": 1.553058624267578e-05,
      "model_forward_time": 5.574443101882935,
      "step": 16285
    },
    {
      "epoch": 1.553058624267578e-05,
      "step": 16285,
      "training_step_time": 16.983067512512207
    },
    {
      "epoch": 1.553153991699219e-05,
      "model_forward_time": 5.571732997894287,
      "step": 16286
    },
    {
      "epoch": 1.553153991699219e-05,
      "step": 16286,
      "training_step_time": 16.98772406578064
    },
    {
      "epoch": 1.5532493591308593e-05,
      "model_forward_time": 5.961137294769287,
      "step": 16287
    },
    {
      "epoch": 1.5532493591308593e-05,
      "step": 16287,
      "training_step_time": 17.401505708694458
    },
    {
      "epoch": 1.5533447265625e-05,
      "model_forward_time": 5.578885793685913,
      "step": 16288
    },
    {
      "epoch": 1.5533447265625e-05,
      "step": 16288,
      "training_step_time": 17.002274990081787
    },
    {
      "epoch": 1.5534400939941408e-05,
      "model_forward_time": 5.589589595794678,
      "step": 16289
    },
    {
      "epoch": 1.5534400939941408e-05,
      "step": 16289,
      "training_step_time": 17.05058526992798
    },
    {
      "epoch": 1.553535461425781e-05,
      "grad_norm": 0.3742382228374481,
      "learning_rate": 1e-05,
      "loss": 0.06,
      "step": 16290
    },
    {
      "epoch": 1.553535461425781e-05,
      "model_forward_time": 5.602422714233398,
      "step": 16290
    },
    {
      "dynamics_loss_avg": 0.056323380768299104,
      "epoch": 1.553535461425781e-05,
      "step": 16290
    },
    {
      "action_loss_avg": 0.0076181198935955765,
      "epoch": 1.553535461425781e-05,
      "step": 16290
    },
    {
      "epoch": 1.553535461425781e-05,
      "step": 16290,
      "training_step_time": 17.068607091903687
    },
    {
      "epoch": 1.553630828857422e-05,
      "model_forward_time": 5.6040754318237305,
      "step": 16291
    },
    {
      "epoch": 1.553630828857422e-05,
      "step": 16291,
      "training_step_time": 17.065534830093384
    },
    {
      "epoch": 1.5537261962890627e-05,
      "model_forward_time": 5.589620590209961,
      "step": 16292
    },
    {
      "epoch": 1.5537261962890627e-05,
      "step": 16292,
      "training_step_time": 17.053760051727295
    },
    {
      "epoch": 1.553821563720703e-05,
      "model_forward_time": 5.958568096160889,
      "step": 16293
    },
    {
      "epoch": 1.553821563720703e-05,
      "step": 16293,
      "training_step_time": 17.39937925338745
    },
    {
      "epoch": 1.5539169311523438e-05,
      "model_forward_time": 5.592316150665283,
      "step": 16294
    },
    {
      "epoch": 1.5539169311523438e-05,
      "step": 16294,
      "training_step_time": 17.015665531158447
    },
    {
      "epoch": 1.5540122985839845e-05,
      "model_forward_time": 5.591397285461426,
      "step": 16295
    },
    {
      "epoch": 1.5540122985839845e-05,
      "step": 16295,
      "training_step_time": 17.033819437026978
    },
    {
      "epoch": 1.554107666015625e-05,
      "model_forward_time": 5.600534677505493,
      "step": 16296
    },
    {
      "epoch": 1.554107666015625e-05,
      "step": 16296,
      "training_step_time": 17.056439876556396
    },
    {
      "epoch": 1.5542030334472657e-05,
      "model_forward_time": 5.599050283432007,
      "step": 16297
    },
    {
      "epoch": 1.5542030334472657e-05,
      "step": 16297,
      "training_step_time": 17.05540680885315
    },
    {
      "epoch": 1.554298400878906e-05,
      "model_forward_time": 5.603351831436157,
      "step": 16298
    },
    {
      "epoch": 1.554298400878906e-05,
      "step": 16298,
      "training_step_time": 17.431593656539917
    },
    {
      "epoch": 1.5543937683105468e-05,
      "model_forward_time": 5.608789443969727,
      "step": 16299
    },
    {
      "epoch": 1.5543937683105468e-05,
      "step": 16299,
      "training_step_time": 17.102027893066406
    },
    {
      "epoch": 1.5544891357421876e-05,
      "grad_norm": 1.159103274345398,
      "learning_rate": 1e-05,
      "loss": 0.0615,
      "step": 16300
    },
    {
      "epoch": 1.5544891357421876e-05,
      "model_forward_time": 5.623456001281738,
      "step": 16300
    },
    {
      "dynamics_loss_avg": 0.05313318893313408,
      "epoch": 1.5544891357421876e-05,
      "step": 16300
    },
    {
      "action_loss_avg": 0.005028453515842557,
      "epoch": 1.5544891357421876e-05,
      "step": 16300
    },
    {
      "epoch": 1.5544891357421876e-05,
      "step": 16300,
      "training_step_time": 17.10621929168701
    },
    {
      "epoch": 1.554584503173828e-05,
      "model_forward_time": 5.597519159317017,
      "step": 16301
    },
    {
      "epoch": 1.554584503173828e-05,
      "step": 16301,
      "training_step_time": 17.039607286453247
    },
    {
      "epoch": 1.5546798706054687e-05,
      "model_forward_time": 5.581906795501709,
      "step": 16302
    },
    {
      "epoch": 1.5546798706054687e-05,
      "step": 16302,
      "training_step_time": 17.01381826400757
    },
    {
      "epoch": 1.5547752380371095e-05,
      "model_forward_time": 5.574746370315552,
      "step": 16303
    },
    {
      "epoch": 1.5547752380371095e-05,
      "step": 16303,
      "training_step_time": 17.010480880737305
    },
    {
      "epoch": 1.55487060546875e-05,
      "model_forward_time": 5.962724685668945,
      "step": 16304
    },
    {
      "epoch": 1.55487060546875e-05,
      "step": 16304,
      "training_step_time": 17.371325969696045
    },
    {
      "epoch": 1.5549659729003906e-05,
      "model_forward_time": 5.59283447265625,
      "step": 16305
    },
    {
      "epoch": 1.5549659729003906e-05,
      "step": 16305,
      "training_step_time": 17.022797107696533
    },
    {
      "epoch": 1.5550613403320313e-05,
      "model_forward_time": 5.6056764125823975,
      "step": 16306
    },
    {
      "epoch": 1.5550613403320313e-05,
      "step": 16306,
      "training_step_time": 17.050137042999268
    },
    {
      "epoch": 1.5551567077636717e-05,
      "model_forward_time": 5.599806308746338,
      "step": 16307
    },
    {
      "epoch": 1.5551567077636717e-05,
      "step": 16307,
      "training_step_time": 17.075257539749146
    },
    {
      "epoch": 1.5552520751953125e-05,
      "model_forward_time": 5.612935304641724,
      "step": 16308
    },
    {
      "epoch": 1.5552520751953125e-05,
      "step": 16308,
      "training_step_time": 17.081272840499878
    },
    {
      "epoch": 1.5553474426269532e-05,
      "model_forward_time": 5.9615607261657715,
      "step": 16309
    },
    {
      "epoch": 1.5553474426269532e-05,
      "step": 16309,
      "training_step_time": 17.393831729888916
    },
    {
      "epoch": 1.5554428100585936e-05,
      "grad_norm": 0.5360390543937683,
      "learning_rate": 1e-05,
      "loss": 0.0563,
      "step": 16310
    },
    {
      "epoch": 1.5554428100585936e-05,
      "model_forward_time": 5.5922462940216064,
      "step": 16310
    },
    {
      "dynamics_loss_avg": 0.04977750238031149,
      "epoch": 1.5554428100585936e-05,
      "step": 16310
    },
    {
      "action_loss_avg": 0.007838620035909117,
      "epoch": 1.5554428100585936e-05,
      "step": 16310
    },
    {
      "epoch": 1.5554428100585936e-05,
      "step": 16310,
      "training_step_time": 17.03932523727417
    },
    {
      "epoch": 1.5555381774902344e-05,
      "model_forward_time": 5.5896522998809814,
      "step": 16311
    },
    {
      "epoch": 1.5555381774902344e-05,
      "step": 16311,
      "training_step_time": 17.017192363739014
    },
    {
      "epoch": 1.555633544921875e-05,
      "model_forward_time": 5.598469018936157,
      "step": 16312
    },
    {
      "epoch": 1.555633544921875e-05,
      "step": 16312,
      "training_step_time": 17.070158004760742
    },
    {
      "epoch": 1.5557289123535155e-05,
      "model_forward_time": 5.601838827133179,
      "step": 16313
    },
    {
      "epoch": 1.5557289123535155e-05,
      "step": 16313,
      "training_step_time": 17.08323884010315
    },
    {
      "epoch": 1.5558242797851563e-05,
      "model_forward_time": 5.6215009689331055,
      "step": 16314
    },
    {
      "epoch": 1.5558242797851563e-05,
      "step": 16314,
      "training_step_time": 17.132170915603638
    },
    {
      "epoch": 1.555919647216797e-05,
      "model_forward_time": 5.995417594909668,
      "step": 16315
    },
    {
      "epoch": 1.555919647216797e-05,
      "step": 16315,
      "training_step_time": 17.466333150863647
    },
    {
      "epoch": 1.5560150146484374e-05,
      "model_forward_time": 5.5944435596466064,
      "step": 16316
    },
    {
      "epoch": 1.5560150146484374e-05,
      "step": 16316,
      "training_step_time": 17.009748458862305
    },
    {
      "epoch": 1.556110382080078e-05,
      "model_forward_time": 5.577044725418091,
      "step": 16317
    },
    {
      "epoch": 1.556110382080078e-05,
      "step": 16317,
      "training_step_time": 17.005767583847046
    },
    {
      "epoch": 1.556205749511719e-05,
      "model_forward_time": 5.576760530471802,
      "step": 16318
    },
    {
      "epoch": 1.556205749511719e-05,
      "step": 16318,
      "training_step_time": 17.03705930709839
    },
    {
      "epoch": 1.5563011169433593e-05,
      "model_forward_time": 5.579740047454834,
      "step": 16319
    },
    {
      "epoch": 1.5563011169433593e-05,
      "step": 16319,
      "training_step_time": 16.99955701828003
    },
    {
      "epoch": 1.556396484375e-05,
      "grad_norm": 0.801007091999054,
      "learning_rate": 1e-05,
      "loss": 0.0528,
      "step": 16320
    },
    {
      "epoch": 1.556396484375e-05,
      "model_forward_time": 5.948410511016846,
      "step": 16320
    },
    {
      "dynamics_loss_avg": 0.046646959520876406,
      "epoch": 1.556396484375e-05,
      "step": 16320
    },
    {
      "action_loss_avg": 0.007073142821900547,
      "epoch": 1.556396484375e-05,
      "step": 16320
    },
    {
      "epoch": 1.556396484375e-05,
      "step": 16320,
      "training_step_time": 17.402122974395752
    },
    {
      "epoch": 1.5564918518066408e-05,
      "model_forward_time": 5.605602979660034,
      "step": 16321
    },
    {
      "epoch": 1.5564918518066408e-05,
      "step": 16321,
      "training_step_time": 17.07603096961975
    },
    {
      "epoch": 1.5565872192382812e-05,
      "model_forward_time": 5.6033124923706055,
      "step": 16322
    },
    {
      "epoch": 1.5565872192382812e-05,
      "step": 16322,
      "training_step_time": 17.104543447494507
    },
    {
      "epoch": 1.556682586669922e-05,
      "model_forward_time": 5.612797498703003,
      "step": 16323
    },
    {
      "epoch": 1.556682586669922e-05,
      "step": 16323,
      "training_step_time": 17.10841703414917
    },
    {
      "epoch": 1.5567779541015627e-05,
      "model_forward_time": 5.612327575683594,
      "step": 16324
    },
    {
      "epoch": 1.5567779541015627e-05,
      "step": 16324,
      "training_step_time": 17.08890151977539
    },
    {
      "epoch": 1.556873321533203e-05,
      "model_forward_time": 5.604380130767822,
      "step": 16325
    },
    {
      "epoch": 1.556873321533203e-05,
      "step": 16325,
      "training_step_time": 17.063209772109985
    },
    {
      "epoch": 1.5569686889648438e-05,
      "model_forward_time": 5.9698216915130615,
      "step": 16326
    },
    {
      "epoch": 1.5569686889648438e-05,
      "step": 16326,
      "training_step_time": 17.41633653640747
    },
    {
      "epoch": 1.5570640563964845e-05,
      "model_forward_time": 5.586725473403931,
      "step": 16327
    },
    {
      "epoch": 1.5570640563964845e-05,
      "step": 16327,
      "training_step_time": 17.04395341873169
    },
    {
      "epoch": 1.557159423828125e-05,
      "model_forward_time": 5.5898919105529785,
      "step": 16328
    },
    {
      "epoch": 1.557159423828125e-05,
      "step": 16328,
      "training_step_time": 17.050042152404785
    },
    {
      "epoch": 1.5572547912597657e-05,
      "model_forward_time": 5.601932525634766,
      "step": 16329
    },
    {
      "epoch": 1.5572547912597657e-05,
      "step": 16329,
      "training_step_time": 17.06864881515503
    },
    {
      "epoch": 1.557350158691406e-05,
      "grad_norm": 1.228531837463379,
      "learning_rate": 1e-05,
      "loss": 0.0745,
      "step": 16330
    },
    {
      "epoch": 1.557350158691406e-05,
      "model_forward_time": 5.614884376525879,
      "step": 16330
    },
    {
      "dynamics_loss_avg": 0.06498380620032548,
      "epoch": 1.557350158691406e-05,
      "step": 16330
    },
    {
      "action_loss_avg": 0.007411759113892913,
      "epoch": 1.557350158691406e-05,
      "step": 16330
    },
    {
      "epoch": 1.557350158691406e-05,
      "step": 16330,
      "training_step_time": 17.114819765090942
    },
    {
      "epoch": 1.5574455261230468e-05,
      "model_forward_time": 5.623366355895996,
      "step": 16331
    },
    {
      "epoch": 1.5574455261230468e-05,
      "step": 16331,
      "training_step_time": 17.476942539215088
    },
    {
      "epoch": 1.5575408935546876e-05,
      "model_forward_time": 5.609618425369263,
      "step": 16332
    },
    {
      "epoch": 1.5575408935546876e-05,
      "step": 16332,
      "training_step_time": 17.10839080810547
    },
    {
      "epoch": 1.557636260986328e-05,
      "model_forward_time": 5.608803749084473,
      "step": 16333
    },
    {
      "epoch": 1.557636260986328e-05,
      "step": 16333,
      "training_step_time": 17.100326538085938
    },
    {
      "epoch": 1.5577316284179687e-05,
      "model_forward_time": 5.608601093292236,
      "step": 16334
    },
    {
      "epoch": 1.5577316284179687e-05,
      "step": 16334,
      "training_step_time": 17.09304118156433
    },
    {
      "epoch": 1.5578269958496095e-05,
      "model_forward_time": 5.606621265411377,
      "step": 16335
    },
    {
      "epoch": 1.5578269958496095e-05,
      "step": 16335,
      "training_step_time": 17.076037883758545
    },
    {
      "epoch": 1.55792236328125e-05,
      "model_forward_time": 5.595031261444092,
      "step": 16336
    },
    {
      "epoch": 1.55792236328125e-05,
      "step": 16336,
      "training_step_time": 17.06655526161194
    },
    {
      "epoch": 1.5580177307128906e-05,
      "model_forward_time": 5.968888998031616,
      "step": 16337
    },
    {
      "epoch": 1.5580177307128906e-05,
      "step": 16337,
      "training_step_time": 17.408331632614136
    },
    {
      "epoch": 1.5581130981445313e-05,
      "model_forward_time": 5.593734264373779,
      "step": 16338
    },
    {
      "epoch": 1.5581130981445313e-05,
      "step": 16338,
      "training_step_time": 17.05807137489319
    },
    {
      "epoch": 1.5582084655761717e-05,
      "model_forward_time": 5.6009438037872314,
      "step": 16339
    },
    {
      "epoch": 1.5582084655761717e-05,
      "step": 16339,
      "training_step_time": 17.06637167930603
    },
    {
      "epoch": 1.5583038330078125e-05,
      "grad_norm": 1.1131232976913452,
      "learning_rate": 1e-05,
      "loss": 0.0673,
      "step": 16340
    },
    {
      "epoch": 1.5583038330078125e-05,
      "model_forward_time": 5.600529432296753,
      "step": 16340
    },
    {
      "dynamics_loss_avg": 0.06606418788433074,
      "epoch": 1.5583038330078125e-05,
      "step": 16340
    },
    {
      "action_loss_avg": 0.007643967680633068,
      "epoch": 1.5583038330078125e-05,
      "step": 16340
    },
    {
      "epoch": 1.5583038330078125e-05,
      "step": 16340,
      "training_step_time": 17.070558786392212
    },
    {
      "epoch": 1.5583992004394532e-05,
      "model_forward_time": 5.602500677108765,
      "step": 16341
    },
    {
      "epoch": 1.5583992004394532e-05,
      "step": 16341,
      "training_step_time": 17.063176155090332
    },
    {
      "epoch": 1.5584945678710936e-05,
      "model_forward_time": 5.953224182128906,
      "step": 16342
    },
    {
      "epoch": 1.5584945678710936e-05,
      "step": 16342,
      "training_step_time": 17.42814612388611
    },
    {
      "epoch": 1.5585899353027344e-05,
      "model_forward_time": 5.602914094924927,
      "step": 16343
    },
    {
      "epoch": 1.5585899353027344e-05,
      "step": 16343,
      "training_step_time": 17.07540988922119
    },
    {
      "epoch": 1.558685302734375e-05,
      "model_forward_time": 5.594516277313232,
      "step": 16344
    },
    {
      "epoch": 1.558685302734375e-05,
      "step": 16344,
      "training_step_time": 17.029454946517944
    },
    {
      "epoch": 1.5587806701660155e-05,
      "model_forward_time": 5.584148645401001,
      "step": 16345
    },
    {
      "epoch": 1.5587806701660155e-05,
      "step": 16345,
      "training_step_time": 17.025911808013916
    },
    {
      "epoch": 1.5588760375976563e-05,
      "model_forward_time": 5.589675426483154,
      "step": 16346
    },
    {
      "epoch": 1.5588760375976563e-05,
      "step": 16346,
      "training_step_time": 17.056729078292847
    },
    {
      "epoch": 1.558971405029297e-05,
      "model_forward_time": 5.620288848876953,
      "step": 16347
    },
    {
      "epoch": 1.558971405029297e-05,
      "step": 16347,
      "training_step_time": 17.111889839172363
    },
    {
      "epoch": 1.5590667724609374e-05,
      "model_forward_time": 5.993362903594971,
      "step": 16348
    },
    {
      "epoch": 1.5590667724609374e-05,
      "step": 16348,
      "training_step_time": 17.471187114715576
    },
    {
      "epoch": 1.559162139892578e-05,
      "model_forward_time": 5.602940082550049,
      "step": 16349
    },
    {
      "epoch": 1.559162139892578e-05,
      "step": 16349,
      "training_step_time": 17.08742928504944
    },
    {
      "epoch": 1.559257507324219e-05,
      "grad_norm": 1.982204794883728,
      "learning_rate": 1e-05,
      "loss": 0.0553,
      "step": 16350
    },
    {
      "epoch": 1.559257507324219e-05,
      "model_forward_time": 5.6109631061553955,
      "step": 16350
    },
    {
      "dynamics_loss_avg": 0.04587250333279371,
      "epoch": 1.559257507324219e-05,
      "step": 16350
    },
    {
      "action_loss_avg": 0.007302110153250396,
      "epoch": 1.559257507324219e-05,
      "step": 16350
    },
    {
      "epoch": 1.559257507324219e-05,
      "step": 16350,
      "training_step_time": 17.096763849258423
    },
    {
      "epoch": 1.5593528747558593e-05,
      "model_forward_time": 5.601559400558472,
      "step": 16351
    },
    {
      "epoch": 1.5593528747558593e-05,
      "step": 16351,
      "training_step_time": 17.071882009506226
    },
    {
      "epoch": 1.5594482421875e-05,
      "model_forward_time": 5.595956563949585,
      "step": 16352
    },
    {
      "epoch": 1.5594482421875e-05,
      "step": 16352,
      "training_step_time": 17.050241947174072
    },
    {
      "epoch": 1.5595436096191408e-05,
      "model_forward_time": 5.9599995613098145,
      "step": 16353
    },
    {
      "epoch": 1.5595436096191408e-05,
      "step": 16353,
      "training_step_time": 17.407389163970947
    },
    {
      "epoch": 1.5596389770507812e-05,
      "model_forward_time": 5.595952272415161,
      "step": 16354
    },
    {
      "epoch": 1.5596389770507812e-05,
      "step": 16354,
      "training_step_time": 17.025068998336792
    },
    {
      "epoch": 1.559734344482422e-05,
      "model_forward_time": 5.589955568313599,
      "step": 16355
    },
    {
      "epoch": 1.559734344482422e-05,
      "step": 16355,
      "training_step_time": 17.010443687438965
    },
    {
      "epoch": 1.5598297119140627e-05,
      "model_forward_time": 5.5930116176605225,
      "step": 16356
    },
    {
      "epoch": 1.5598297119140627e-05,
      "step": 16356,
      "training_step_time": 17.054627895355225
    },
    {
      "epoch": 1.559925079345703e-05,
      "model_forward_time": 5.606923341751099,
      "step": 16357
    },
    {
      "epoch": 1.559925079345703e-05,
      "step": 16357,
      "training_step_time": 17.11184048652649
    },
    {
      "epoch": 1.5600204467773438e-05,
      "model_forward_time": 5.6335601806640625,
      "step": 16358
    },
    {
      "epoch": 1.5600204467773438e-05,
      "step": 16358,
      "training_step_time": 17.164661645889282
    },
    {
      "epoch": 1.5601158142089845e-05,
      "model_forward_time": 5.991226434707642,
      "step": 16359
    },
    {
      "epoch": 1.5601158142089845e-05,
      "step": 16359,
      "training_step_time": 17.481985092163086
    },
    {
      "epoch": 1.560211181640625e-05,
      "grad_norm": 0.5590582489967346,
      "learning_rate": 1e-05,
      "loss": 0.0557,
      "step": 16360
    },
    {
      "epoch": 1.560211181640625e-05,
      "model_forward_time": 5.612277984619141,
      "step": 16360
    },
    {
      "dynamics_loss_avg": 0.04721880927681923,
      "epoch": 1.560211181640625e-05,
      "step": 16360
    },
    {
      "action_loss_avg": 0.005539607023820281,
      "epoch": 1.560211181640625e-05,
      "step": 16360
    },
    {
      "epoch": 1.560211181640625e-05,
      "step": 16360,
      "training_step_time": 17.098591566085815
    },
    {
      "epoch": 1.5603065490722657e-05,
      "model_forward_time": 5.6091039180755615,
      "step": 16361
    },
    {
      "epoch": 1.5603065490722657e-05,
      "step": 16361,
      "training_step_time": 17.099697589874268
    },
    {
      "epoch": 1.560401916503906e-05,
      "model_forward_time": 5.611222743988037,
      "step": 16362
    },
    {
      "epoch": 1.560401916503906e-05,
      "step": 16362,
      "training_step_time": 17.102596521377563
    },
    {
      "epoch": 1.560497283935547e-05,
      "model_forward_time": 5.607889890670776,
      "step": 16363
    },
    {
      "epoch": 1.560497283935547e-05,
      "step": 16363,
      "training_step_time": 17.07486653327942
    },
    {
      "epoch": 1.5605926513671876e-05,
      "model_forward_time": 5.59667706489563,
      "step": 16364
    },
    {
      "epoch": 1.5605926513671876e-05,
      "step": 16364,
      "training_step_time": 17.438586235046387
    },
    {
      "epoch": 1.560688018798828e-05,
      "model_forward_time": 5.617845773696899,
      "step": 16365
    },
    {
      "epoch": 1.560688018798828e-05,
      "step": 16365,
      "training_step_time": 17.112462520599365
    },
    {
      "epoch": 1.5607833862304687e-05,
      "model_forward_time": 5.6184797286987305,
      "step": 16366
    },
    {
      "epoch": 1.5607833862304687e-05,
      "step": 16366,
      "training_step_time": 17.119566440582275
    },
    {
      "epoch": 1.5608787536621095e-05,
      "model_forward_time": 5.609952211380005,
      "step": 16367
    },
    {
      "epoch": 1.5608787536621095e-05,
      "step": 16367,
      "training_step_time": 17.063219785690308
    },
    {
      "epoch": 1.56097412109375e-05,
      "model_forward_time": 5.598550796508789,
      "step": 16368
    },
    {
      "epoch": 1.56097412109375e-05,
      "step": 16368,
      "training_step_time": 17.042537212371826
    },
    {
      "epoch": 1.5610694885253906e-05,
      "model_forward_time": 5.597312688827515,
      "step": 16369
    },
    {
      "epoch": 1.5610694885253906e-05,
      "step": 16369,
      "training_step_time": 17.03427481651306
    },
    {
      "epoch": 1.5611648559570313e-05,
      "grad_norm": 0.40111804008483887,
      "learning_rate": 1e-05,
      "loss": 0.0657,
      "step": 16370
    },
    {
      "epoch": 1.5611648559570313e-05,
      "model_forward_time": 5.96013069152832,
      "step": 16370
    },
    {
      "dynamics_loss_avg": 0.060663672350347045,
      "epoch": 1.5611648559570313e-05,
      "step": 16370
    },
    {
      "action_loss_avg": 0.007353840791620314,
      "epoch": 1.5611648559570313e-05,
      "step": 16370
    },
    {
      "epoch": 1.5611648559570313e-05,
      "step": 16370,
      "training_step_time": 17.406942129135132
    },
    {
      "epoch": 1.5612602233886717e-05,
      "model_forward_time": 5.578475475311279,
      "step": 16371
    },
    {
      "epoch": 1.5612602233886717e-05,
      "step": 16371,
      "training_step_time": 17.017521142959595
    },
    {
      "epoch": 1.5613555908203125e-05,
      "model_forward_time": 5.594096899032593,
      "step": 16372
    },
    {
      "epoch": 1.5613555908203125e-05,
      "step": 16372,
      "training_step_time": 17.03891158103943
    },
    {
      "epoch": 1.5614509582519532e-05,
      "model_forward_time": 5.603314161300659,
      "step": 16373
    },
    {
      "epoch": 1.5614509582519532e-05,
      "step": 16373,
      "training_step_time": 17.07315754890442
    },
    {
      "epoch": 1.5615463256835936e-05,
      "model_forward_time": 5.611144542694092,
      "step": 16374
    },
    {
      "epoch": 1.5615463256835936e-05,
      "step": 16374,
      "training_step_time": 17.095155477523804
    },
    {
      "epoch": 1.5616416931152344e-05,
      "model_forward_time": 5.964743137359619,
      "step": 16375
    },
    {
      "epoch": 1.5616416931152344e-05,
      "step": 16375,
      "training_step_time": 17.415719747543335
    },
    {
      "epoch": 1.561737060546875e-05,
      "model_forward_time": 5.596895456314087,
      "step": 16376
    },
    {
      "epoch": 1.561737060546875e-05,
      "step": 16376,
      "training_step_time": 17.056702613830566
    },
    {
      "epoch": 1.5618324279785155e-05,
      "model_forward_time": 5.596808195114136,
      "step": 16377
    },
    {
      "epoch": 1.5618324279785155e-05,
      "step": 16377,
      "training_step_time": 17.049398183822632
    },
    {
      "epoch": 1.5619277954101563e-05,
      "model_forward_time": 5.592188119888306,
      "step": 16378
    },
    {
      "epoch": 1.5619277954101563e-05,
      "step": 16378,
      "training_step_time": 17.022908926010132
    },
    {
      "epoch": 1.562023162841797e-05,
      "model_forward_time": 5.585428476333618,
      "step": 16379
    },
    {
      "epoch": 1.562023162841797e-05,
      "step": 16379,
      "training_step_time": 17.040443658828735
    },
    {
      "epoch": 1.5621185302734374e-05,
      "grad_norm": 0.7040996551513672,
      "learning_rate": 1e-05,
      "loss": 0.0529,
      "step": 16380
    },
    {
      "epoch": 1.5621185302734374e-05,
      "model_forward_time": 5.592466831207275,
      "step": 16380
    },
    {
      "dynamics_loss_avg": 0.041844668611884114,
      "epoch": 1.5621185302734374e-05,
      "step": 16380
    },
    {
      "action_loss_avg": 0.005622033739928156,
      "epoch": 1.5621185302734374e-05,
      "step": 16380
    },
    {
      "epoch": 1.5621185302734374e-05,
      "step": 16380,
      "training_step_time": 17.082969665527344
    },
    {
      "epoch": 1.562213897705078e-05,
      "model_forward_time": 5.995754718780518,
      "step": 16381
    },
    {
      "epoch": 1.562213897705078e-05,
      "step": 16381,
      "training_step_time": 17.491031169891357
    },
    {
      "epoch": 1.562309265136719e-05,
      "model_forward_time": 5.60057806968689,
      "step": 16382
    },
    {
      "epoch": 1.562309265136719e-05,
      "step": 16382,
      "training_step_time": 17.04554009437561
    },
    {
      "epoch": 1.5624046325683593e-05,
      "model_forward_time": 5.590912580490112,
      "step": 16383
    },
    {
      "epoch": 1.5624046325683593e-05,
      "step": 16383,
      "training_step_time": 17.061330318450928
    },
    {
      "epoch": 1.5625e-05,
      "model_forward_time": 5.598189353942871,
      "step": 16384
    },
    {
      "epoch": 1.5625e-05,
      "step": 16384,
      "training_step_time": 17.031744241714478
    },
    {
      "epoch": 1.5625953674316408e-05,
      "model_forward_time": 5.5829033851623535,
      "step": 16385
    },
    {
      "epoch": 1.5625953674316408e-05,
      "step": 16385,
      "training_step_time": 17.02811908721924
    },
    {
      "epoch": 1.5626907348632812e-05,
      "model_forward_time": 5.978019952774048,
      "step": 16386
    },
    {
      "epoch": 1.5626907348632812e-05,
      "step": 16386,
      "training_step_time": 17.45653462409973
    },
    {
      "epoch": 1.562786102294922e-05,
      "model_forward_time": 5.6301445960998535,
      "step": 16387
    },
    {
      "epoch": 1.562786102294922e-05,
      "step": 16387,
      "training_step_time": 17.135513067245483
    },
    {
      "epoch": 1.5628814697265627e-05,
      "model_forward_time": 5.60089898109436,
      "step": 16388
    },
    {
      "epoch": 1.5628814697265627e-05,
      "step": 16388,
      "training_step_time": 17.05202579498291
    },
    {
      "epoch": 1.562976837158203e-05,
      "model_forward_time": 5.590557098388672,
      "step": 16389
    },
    {
      "epoch": 1.562976837158203e-05,
      "step": 16389,
      "training_step_time": 17.046172618865967
    },
    {
      "epoch": 1.5630722045898438e-05,
      "grad_norm": 0.39762118458747864,
      "learning_rate": 1e-05,
      "loss": 0.061,
      "step": 16390
    },
    {
      "epoch": 1.5630722045898438e-05,
      "model_forward_time": 5.607190847396851,
      "step": 16390
    },
    {
      "dynamics_loss_avg": 0.056903870031237605,
      "epoch": 1.5630722045898438e-05,
      "step": 16390
    },
    {
      "action_loss_avg": 0.007582237618044019,
      "epoch": 1.5630722045898438e-05,
      "step": 16390
    },
    {
      "epoch": 1.5630722045898438e-05,
      "step": 16390,
      "training_step_time": 17.075209379196167
    },
    {
      "epoch": 1.5631675720214842e-05,
      "model_forward_time": 5.604681730270386,
      "step": 16391
    },
    {
      "epoch": 1.5631675720214842e-05,
      "step": 16391,
      "training_step_time": 17.107946157455444
    },
    {
      "epoch": 1.563262939453125e-05,
      "model_forward_time": 6.000197649002075,
      "step": 16392
    },
    {
      "epoch": 1.563262939453125e-05,
      "step": 16392,
      "training_step_time": 17.493332624435425
    },
    {
      "epoch": 1.5633583068847657e-05,
      "model_forward_time": 5.597916126251221,
      "step": 16393
    },
    {
      "epoch": 1.5633583068847657e-05,
      "step": 16393,
      "training_step_time": 17.01995873451233
    },
    {
      "epoch": 1.563453674316406e-05,
      "model_forward_time": 5.580873966217041,
      "step": 16394
    },
    {
      "epoch": 1.563453674316406e-05,
      "step": 16394,
      "training_step_time": 17.015331029891968
    },
    {
      "epoch": 1.563549041748047e-05,
      "model_forward_time": 5.597460508346558,
      "step": 16395
    },
    {
      "epoch": 1.563549041748047e-05,
      "step": 16395,
      "training_step_time": 17.00451350212097
    },
    {
      "epoch": 1.5636444091796876e-05,
      "model_forward_time": 5.576292514801025,
      "step": 16396
    },
    {
      "epoch": 1.5636444091796876e-05,
      "step": 16396,
      "training_step_time": 17.014344453811646
    },
    {
      "epoch": 1.563739776611328e-05,
      "model_forward_time": 5.59419846534729,
      "step": 16397
    },
    {
      "epoch": 1.563739776611328e-05,
      "step": 16397,
      "training_step_time": 17.41565728187561
    },
    {
      "epoch": 1.5638351440429687e-05,
      "model_forward_time": 5.617157220840454,
      "step": 16398
    },
    {
      "epoch": 1.5638351440429687e-05,
      "step": 16398,
      "training_step_time": 17.122957229614258
    },
    {
      "epoch": 1.5639305114746095e-05,
      "model_forward_time": 5.625587224960327,
      "step": 16399
    },
    {
      "epoch": 1.5639305114746095e-05,
      "step": 16399,
      "training_step_time": 17.137750387191772
    },
    {
      "epoch": 1.56402587890625e-05,
      "grad_norm": 0.6849731206893921,
      "learning_rate": 1e-05,
      "loss": 0.0573,
      "step": 16400
    },
    {
      "epoch": 1.56402587890625e-05,
      "model_forward_time": 5.610821485519409,
      "step": 16400
    },
    {
      "dynamics_loss_avg": 0.05128373708575964,
      "epoch": 1.56402587890625e-05,
      "step": 16400
    },
    {
      "action_loss_avg": 0.006857803009916097,
      "epoch": 1.56402587890625e-05,
      "step": 16400
    },
    {
      "epoch": 1.56402587890625e-05,
      "step": 16400,
      "training_step_time": 17.095004320144653
    },
    {
      "epoch": 1.5641212463378906e-05,
      "model_forward_time": 5.5980212688446045,
      "step": 16401
    },
    {
      "epoch": 1.5641212463378906e-05,
      "step": 16401,
      "training_step_time": 17.05855107307434
    },
    {
      "epoch": 1.5642166137695313e-05,
      "model_forward_time": 5.604973554611206,
      "step": 16402
    },
    {
      "epoch": 1.5642166137695313e-05,
      "step": 16402,
      "training_step_time": 17.069881916046143
    },
    {
      "epoch": 1.5643119812011718e-05,
      "model_forward_time": 5.967012643814087,
      "step": 16403
    },
    {
      "epoch": 1.5643119812011718e-05,
      "step": 16403,
      "training_step_time": 17.396653175354004
    },
    {
      "epoch": 1.5644073486328125e-05,
      "model_forward_time": 5.589613676071167,
      "step": 16404
    },
    {
      "epoch": 1.5644073486328125e-05,
      "step": 16404,
      "training_step_time": 17.037522792816162
    },
    {
      "epoch": 1.5645027160644532e-05,
      "model_forward_time": 5.606484413146973,
      "step": 16405
    },
    {
      "epoch": 1.5645027160644532e-05,
      "step": 16405,
      "training_step_time": 17.11399483680725
    },
    {
      "epoch": 1.5645980834960936e-05,
      "model_forward_time": 5.637471914291382,
      "step": 16406
    },
    {
      "epoch": 1.5645980834960936e-05,
      "step": 16406,
      "training_step_time": 17.11908507347107
    },
    {
      "epoch": 1.5646934509277344e-05,
      "model_forward_time": 5.594780683517456,
      "step": 16407
    },
    {
      "epoch": 1.5646934509277344e-05,
      "step": 16407,
      "training_step_time": 17.00913643836975
    },
    {
      "epoch": 1.564788818359375e-05,
      "model_forward_time": 5.9255640506744385,
      "step": 16408
    },
    {
      "epoch": 1.564788818359375e-05,
      "step": 16408,
      "training_step_time": 17.345502614974976
    },
    {
      "epoch": 1.5648841857910155e-05,
      "model_forward_time": 5.567407131195068,
      "step": 16409
    },
    {
      "epoch": 1.5648841857910155e-05,
      "step": 16409,
      "training_step_time": 16.999651432037354
    },
    {
      "epoch": 1.5649795532226563e-05,
      "grad_norm": 0.6081926822662354,
      "learning_rate": 1e-05,
      "loss": 0.0534,
      "step": 16410
    },
    {
      "epoch": 1.5649795532226563e-05,
      "model_forward_time": 5.575419902801514,
      "step": 16410
    },
    {
      "dynamics_loss_avg": 0.046062207967042926,
      "epoch": 1.5649795532226563e-05,
      "step": 16410
    },
    {
      "action_loss_avg": 0.004437781847082078,
      "epoch": 1.5649795532226563e-05,
      "step": 16410
    },
    {
      "epoch": 1.5649795532226563e-05,
      "step": 16410,
      "training_step_time": 16.994327545166016
    },
    {
      "epoch": 1.565074920654297e-05,
      "model_forward_time": 5.574734210968018,
      "step": 16411
    },
    {
      "epoch": 1.565074920654297e-05,
      "step": 16411,
      "training_step_time": 16.991995811462402
    },
    {
      "epoch": 1.5651702880859374e-05,
      "model_forward_time": 5.584152698516846,
      "step": 16412
    },
    {
      "epoch": 1.5651702880859374e-05,
      "step": 16412,
      "training_step_time": 17.004573583602905
    },
    {
      "epoch": 1.565265655517578e-05,
      "model_forward_time": 5.583830833435059,
      "step": 16413
    },
    {
      "epoch": 1.565265655517578e-05,
      "step": 16413,
      "training_step_time": 17.032463788986206
    },
    {
      "epoch": 1.565361022949219e-05,
      "model_forward_time": 5.978156328201294,
      "step": 16414
    },
    {
      "epoch": 1.565361022949219e-05,
      "step": 16414,
      "training_step_time": 17.436861753463745
    },
    {
      "epoch": 1.5654563903808593e-05,
      "model_forward_time": 5.594694375991821,
      "step": 16415
    },
    {
      "epoch": 1.5654563903808593e-05,
      "step": 16415,
      "training_step_time": 17.07808828353882
    },
    {
      "epoch": 1.5655517578125e-05,
      "model_forward_time": 5.609043121337891,
      "step": 16416
    },
    {
      "epoch": 1.5655517578125e-05,
      "step": 16416,
      "training_step_time": 17.086355924606323
    },
    {
      "epoch": 1.5656471252441408e-05,
      "model_forward_time": 5.60656213760376,
      "step": 16417
    },
    {
      "epoch": 1.5656471252441408e-05,
      "step": 16417,
      "training_step_time": 17.087398290634155
    },
    {
      "epoch": 1.5657424926757812e-05,
      "model_forward_time": 5.602641820907593,
      "step": 16418
    },
    {
      "epoch": 1.5657424926757812e-05,
      "step": 16418,
      "training_step_time": 17.074279308319092
    },
    {
      "epoch": 1.565837860107422e-05,
      "model_forward_time": 5.9576735496521,
      "step": 16419
    },
    {
      "epoch": 1.565837860107422e-05,
      "step": 16419,
      "training_step_time": 17.42031478881836
    },
    {
      "epoch": 1.5659332275390627e-05,
      "grad_norm": 0.5833883285522461,
      "learning_rate": 1e-05,
      "loss": 0.0499,
      "step": 16420
    },
    {
      "epoch": 1.5659332275390627e-05,
      "model_forward_time": 5.598238468170166,
      "step": 16420
    },
    {
      "dynamics_loss_avg": 0.04580908138304949,
      "epoch": 1.5659332275390627e-05,
      "step": 16420
    },
    {
      "action_loss_avg": 0.004701707093045115,
      "epoch": 1.5659332275390627e-05,
      "step": 16420
    },
    {
      "epoch": 1.5659332275390627e-05,
      "step": 16420,
      "training_step_time": 17.049948692321777
    },
    {
      "epoch": 1.566028594970703e-05,
      "model_forward_time": 5.586505889892578,
      "step": 16421
    },
    {
      "epoch": 1.566028594970703e-05,
      "step": 16421,
      "training_step_time": 17.016697883605957
    },
    {
      "epoch": 1.5661239624023438e-05,
      "model_forward_time": 5.589550018310547,
      "step": 16422
    },
    {
      "epoch": 1.5661239624023438e-05,
      "step": 16422,
      "training_step_time": 17.011597156524658
    },
    {
      "epoch": 1.5662193298339842e-05,
      "model_forward_time": 5.605800628662109,
      "step": 16423
    },
    {
      "epoch": 1.5662193298339842e-05,
      "step": 16423,
      "training_step_time": 17.045202493667603
    },
    {
      "epoch": 1.566314697265625e-05,
      "model_forward_time": 5.613898038864136,
      "step": 16424
    },
    {
      "epoch": 1.566314697265625e-05,
      "step": 16424,
      "training_step_time": 17.111801147460938
    },
    {
      "epoch": 1.5664100646972657e-05,
      "model_forward_time": 6.0139853954315186,
      "step": 16425
    },
    {
      "epoch": 1.5664100646972657e-05,
      "step": 16425,
      "training_step_time": 17.51243305206299
    },
    {
      "epoch": 1.566505432128906e-05,
      "model_forward_time": 5.597965717315674,
      "step": 16426
    },
    {
      "epoch": 1.566505432128906e-05,
      "step": 16426,
      "training_step_time": 17.016746520996094
    },
    {
      "epoch": 1.566600799560547e-05,
      "model_forward_time": 5.588063955307007,
      "step": 16427
    },
    {
      "epoch": 1.566600799560547e-05,
      "step": 16427,
      "training_step_time": 17.031360149383545
    },
    {
      "epoch": 1.5666961669921876e-05,
      "model_forward_time": 5.58549165725708,
      "step": 16428
    },
    {
      "epoch": 1.5666961669921876e-05,
      "step": 16428,
      "training_step_time": 17.033697605133057
    },
    {
      "epoch": 1.566791534423828e-05,
      "model_forward_time": 5.5851030349731445,
      "step": 16429
    },
    {
      "epoch": 1.566791534423828e-05,
      "step": 16429,
      "training_step_time": 17.036662340164185
    },
    {
      "epoch": 1.5668869018554687e-05,
      "grad_norm": 0.3481377065181732,
      "learning_rate": 1e-05,
      "loss": 0.0566,
      "step": 16430
    },
    {
      "epoch": 1.5668869018554687e-05,
      "model_forward_time": 5.593847036361694,
      "step": 16430
    },
    {
      "dynamics_loss_avg": 0.058117246814072133,
      "epoch": 1.5668869018554687e-05,
      "step": 16430
    },
    {
      "action_loss_avg": 0.007331349956803024,
      "epoch": 1.5668869018554687e-05,
      "step": 16430
    },
    {
      "epoch": 1.5668869018554687e-05,
      "step": 16430,
      "training_step_time": 17.420690059661865
    },
    {
      "epoch": 1.5669822692871095e-05,
      "model_forward_time": 5.597178220748901,
      "step": 16431
    },
    {
      "epoch": 1.5669822692871095e-05,
      "step": 16431,
      "training_step_time": 17.071654081344604
    },
    {
      "epoch": 1.56707763671875e-05,
      "model_forward_time": 5.602872848510742,
      "step": 16432
    },
    {
      "epoch": 1.56707763671875e-05,
      "step": 16432,
      "training_step_time": 17.05458426475525
    },
    {
      "epoch": 1.5671730041503906e-05,
      "model_forward_time": 5.585287570953369,
      "step": 16433
    },
    {
      "epoch": 1.5671730041503906e-05,
      "step": 16433,
      "training_step_time": 17.023298740386963
    },
    {
      "epoch": 1.5672683715820314e-05,
      "model_forward_time": 5.586243391036987,
      "step": 16434
    },
    {
      "epoch": 1.5672683715820314e-05,
      "step": 16434,
      "training_step_time": 17.021827459335327
    },
    {
      "epoch": 1.5673637390136718e-05,
      "model_forward_time": 5.591073751449585,
      "step": 16435
    },
    {
      "epoch": 1.5673637390136718e-05,
      "step": 16435,
      "training_step_time": 17.033559799194336
    },
    {
      "epoch": 1.5674591064453125e-05,
      "model_forward_time": 5.985350608825684,
      "step": 16436
    },
    {
      "epoch": 1.5674591064453125e-05,
      "step": 16436,
      "training_step_time": 17.465758085250854
    },
    {
      "epoch": 1.5675544738769532e-05,
      "model_forward_time": 5.620114326477051,
      "step": 16437
    },
    {
      "epoch": 1.5675544738769532e-05,
      "step": 16437,
      "training_step_time": 17.086901903152466
    },
    {
      "epoch": 1.5676498413085936e-05,
      "model_forward_time": 5.582416296005249,
      "step": 16438
    },
    {
      "epoch": 1.5676498413085936e-05,
      "step": 16438,
      "training_step_time": 17.005326747894287
    },
    {
      "epoch": 1.5677452087402344e-05,
      "model_forward_time": 5.582278490066528,
      "step": 16439
    },
    {
      "epoch": 1.5677452087402344e-05,
      "step": 16439,
      "training_step_time": 17.02025294303894
    },
    {
      "epoch": 1.567840576171875e-05,
      "grad_norm": 0.7501411437988281,
      "learning_rate": 1e-05,
      "loss": 0.0708,
      "step": 16440
    },
    {
      "epoch": 1.567840576171875e-05,
      "model_forward_time": 5.564611434936523,
      "step": 16440
    },
    {
      "dynamics_loss_avg": 0.05784930847585201,
      "epoch": 1.567840576171875e-05,
      "step": 16440
    },
    {
      "action_loss_avg": 0.00614745948696509,
      "epoch": 1.567840576171875e-05,
      "step": 16440
    },
    {
      "epoch": 1.567840576171875e-05,
      "step": 16440,
      "training_step_time": 17.0003399848938
    },
    {
      "epoch": 1.5679359436035155e-05,
      "model_forward_time": 5.945301532745361,
      "step": 16441
    },
    {
      "epoch": 1.5679359436035155e-05,
      "step": 16441,
      "training_step_time": 17.379837036132812
    },
    {
      "epoch": 1.5680313110351563e-05,
      "model_forward_time": 5.577584981918335,
      "step": 16442
    },
    {
      "epoch": 1.5680313110351563e-05,
      "step": 16442,
      "training_step_time": 16.99931311607361
    },
    {
      "epoch": 1.568126678466797e-05,
      "model_forward_time": 5.596007347106934,
      "step": 16443
    },
    {
      "epoch": 1.568126678466797e-05,
      "step": 16443,
      "training_step_time": 17.04567527770996
    },
    {
      "epoch": 1.5682220458984374e-05,
      "model_forward_time": 5.598918437957764,
      "step": 16444
    },
    {
      "epoch": 1.5682220458984374e-05,
      "step": 16444,
      "training_step_time": 17.073801517486572
    },
    {
      "epoch": 1.568317413330078e-05,
      "model_forward_time": 5.611820697784424,
      "step": 16445
    },
    {
      "epoch": 1.568317413330078e-05,
      "step": 16445,
      "training_step_time": 17.100874185562134
    },
    {
      "epoch": 1.568412780761719e-05,
      "model_forward_time": 5.598389625549316,
      "step": 16446
    },
    {
      "epoch": 1.568412780761719e-05,
      "step": 16446,
      "training_step_time": 17.050429582595825
    },
    {
      "epoch": 1.5685081481933593e-05,
      "model_forward_time": 5.9727942943573,
      "step": 16447
    },
    {
      "epoch": 1.5685081481933593e-05,
      "step": 16447,
      "training_step_time": 17.415007829666138
    },
    {
      "epoch": 1.568603515625e-05,
      "model_forward_time": 5.58808970451355,
      "step": 16448
    },
    {
      "epoch": 1.568603515625e-05,
      "step": 16448,
      "training_step_time": 17.01897954940796
    },
    {
      "epoch": 1.5686988830566408e-05,
      "model_forward_time": 5.581765413284302,
      "step": 16449
    },
    {
      "epoch": 1.5686988830566408e-05,
      "step": 16449,
      "training_step_time": 17.004488706588745
    },
    {
      "epoch": 1.5687942504882812e-05,
      "grad_norm": 0.533717930316925,
      "learning_rate": 1e-05,
      "loss": 0.0597,
      "step": 16450
    },
    {
      "epoch": 1.5687942504882812e-05,
      "model_forward_time": 5.6334803104400635,
      "step": 16450
    },
    {
      "dynamics_loss_avg": 0.05108711887151003,
      "epoch": 1.5687942504882812e-05,
      "step": 16450
    },
    {
      "action_loss_avg": 0.005062685278244316,
      "epoch": 1.5687942504882812e-05,
      "step": 16450
    },
    {
      "epoch": 1.5687942504882812e-05,
      "step": 16450,
      "training_step_time": 17.132147789001465
    },
    {
      "epoch": 1.568889617919922e-05,
      "model_forward_time": 5.618807315826416,
      "step": 16451
    },
    {
      "epoch": 1.568889617919922e-05,
      "step": 16451,
      "training_step_time": 17.093501329421997
    },
    {
      "epoch": 1.5689849853515627e-05,
      "model_forward_time": 6.138026714324951,
      "step": 16452
    },
    {
      "epoch": 1.5689849853515627e-05,
      "step": 16452,
      "training_step_time": 17.6374409198761
    },
    {
      "epoch": 1.569080352783203e-05,
      "model_forward_time": 5.6211793422698975,
      "step": 16453
    },
    {
      "epoch": 1.569080352783203e-05,
      "step": 16453,
      "training_step_time": 17.125614166259766
    },
    {
      "epoch": 1.5691757202148438e-05,
      "model_forward_time": 5.633302211761475,
      "step": 16454
    },
    {
      "epoch": 1.5691757202148438e-05,
      "step": 16454,
      "training_step_time": 17.141454696655273
    },
    {
      "epoch": 1.5692710876464842e-05,
      "model_forward_time": 5.627349138259888,
      "step": 16455
    },
    {
      "epoch": 1.5692710876464842e-05,
      "step": 16455,
      "training_step_time": 17.155609846115112
    },
    {
      "epoch": 1.569366455078125e-05,
      "model_forward_time": 5.645547151565552,
      "step": 16456
    },
    {
      "epoch": 1.569366455078125e-05,
      "step": 16456,
      "training_step_time": 17.16216540336609
    },
    {
      "epoch": 1.5694618225097657e-05,
      "model_forward_time": 5.626127481460571,
      "step": 16457
    },
    {
      "epoch": 1.5694618225097657e-05,
      "step": 16457,
      "training_step_time": 17.12688660621643
    },
    {
      "epoch": 1.569557189941406e-05,
      "model_forward_time": 6.0760657787323,
      "step": 16458
    },
    {
      "epoch": 1.569557189941406e-05,
      "step": 16458,
      "training_step_time": 17.550288677215576
    },
    {
      "epoch": 1.569652557373047e-05,
      "model_forward_time": 5.610965728759766,
      "step": 16459
    },
    {
      "epoch": 1.569652557373047e-05,
      "step": 16459,
      "training_step_time": 17.10284733772278
    },
    {
      "epoch": 1.5697479248046876e-05,
      "grad_norm": 1.23874831199646,
      "learning_rate": 1e-05,
      "loss": 0.0596,
      "step": 16460
    },
    {
      "epoch": 1.5697479248046876e-05,
      "model_forward_time": 5.618774890899658,
      "step": 16460
    },
    {
      "dynamics_loss_avg": 0.05527047384530306,
      "epoch": 1.5697479248046876e-05,
      "step": 16460
    },
    {
      "action_loss_avg": 0.0068598776240833105,
      "epoch": 1.5697479248046876e-05,
      "step": 16460
    },
    {
      "epoch": 1.5697479248046876e-05,
      "step": 16460,
      "training_step_time": 17.089787244796753
    },
    {
      "epoch": 1.569843292236328e-05,
      "model_forward_time": 5.597095489501953,
      "step": 16461
    },
    {
      "epoch": 1.569843292236328e-05,
      "step": 16461,
      "training_step_time": 17.061192750930786
    },
    {
      "epoch": 1.5699386596679687e-05,
      "model_forward_time": 5.603197336196899,
      "step": 16462
    },
    {
      "epoch": 1.5699386596679687e-05,
      "step": 16462,
      "training_step_time": 17.06148624420166
    },
    {
      "epoch": 1.5700340270996095e-05,
      "model_forward_time": 5.611034631729126,
      "step": 16463
    },
    {
      "epoch": 1.5700340270996095e-05,
      "step": 16463,
      "training_step_time": 17.46629500389099
    },
    {
      "epoch": 1.57012939453125e-05,
      "model_forward_time": 5.603661298751831,
      "step": 16464
    },
    {
      "epoch": 1.57012939453125e-05,
      "step": 16464,
      "training_step_time": 17.07232427597046
    },
    {
      "epoch": 1.5702247619628906e-05,
      "model_forward_time": 5.60467529296875,
      "step": 16465
    },
    {
      "epoch": 1.5702247619628906e-05,
      "step": 16465,
      "training_step_time": 17.077988147735596
    },
    {
      "epoch": 1.5703201293945314e-05,
      "model_forward_time": 5.616266250610352,
      "step": 16466
    },
    {
      "epoch": 1.5703201293945314e-05,
      "step": 16466,
      "training_step_time": 17.102334022521973
    },
    {
      "epoch": 1.5704154968261718e-05,
      "model_forward_time": 5.613381862640381,
      "step": 16467
    },
    {
      "epoch": 1.5704154968261718e-05,
      "step": 16467,
      "training_step_time": 17.098971366882324
    },
    {
      "epoch": 1.5705108642578125e-05,
      "model_forward_time": 5.6254496574401855,
      "step": 16468
    },
    {
      "epoch": 1.5705108642578125e-05,
      "step": 16468,
      "training_step_time": 17.1139395236969
    },
    {
      "epoch": 1.5706062316894532e-05,
      "model_forward_time": 6.022996664047241,
      "step": 16469
    },
    {
      "epoch": 1.5706062316894532e-05,
      "step": 16469,
      "training_step_time": 17.44196581840515
    },
    {
      "epoch": 1.5707015991210936e-05,
      "grad_norm": 1.09703528881073,
      "learning_rate": 1e-05,
      "loss": 0.0527,
      "step": 16470
    },
    {
      "epoch": 1.5707015991210936e-05,
      "model_forward_time": 5.596940040588379,
      "step": 16470
    },
    {
      "dynamics_loss_avg": 0.0455431405454874,
      "epoch": 1.5707015991210936e-05,
      "step": 16470
    },
    {
      "action_loss_avg": 0.006140336918178946,
      "epoch": 1.5707015991210936e-05,
      "step": 16470
    },
    {
      "epoch": 1.5707015991210936e-05,
      "step": 16470,
      "training_step_time": 17.06037211418152
    },
    {
      "epoch": 1.5707969665527344e-05,
      "model_forward_time": 5.606897592544556,
      "step": 16471
    },
    {
      "epoch": 1.5707969665527344e-05,
      "step": 16471,
      "training_step_time": 17.094161987304688
    },
    {
      "epoch": 1.570892333984375e-05,
      "model_forward_time": 5.6500794887542725,
      "step": 16472
    },
    {
      "epoch": 1.570892333984375e-05,
      "step": 16472,
      "training_step_time": 17.1693332195282
    },
    {
      "epoch": 1.5709877014160155e-05,
      "model_forward_time": 5.627061367034912,
      "step": 16473
    },
    {
      "epoch": 1.5709877014160155e-05,
      "step": 16473,
      "training_step_time": 17.153270721435547
    },
    {
      "epoch": 1.5710830688476563e-05,
      "model_forward_time": 6.008173227310181,
      "step": 16474
    },
    {
      "epoch": 1.5710830688476563e-05,
      "step": 16474,
      "training_step_time": 17.514870166778564
    },
    {
      "epoch": 1.571178436279297e-05,
      "model_forward_time": 5.629825115203857,
      "step": 16475
    },
    {
      "epoch": 1.571178436279297e-05,
      "step": 16475,
      "training_step_time": 17.13479232788086
    },
    {
      "epoch": 1.5712738037109374e-05,
      "model_forward_time": 5.6293816566467285,
      "step": 16476
    },
    {
      "epoch": 1.5712738037109374e-05,
      "step": 16476,
      "training_step_time": 17.13800072669983
    },
    {
      "epoch": 1.571369171142578e-05,
      "model_forward_time": 5.631135702133179,
      "step": 16477
    },
    {
      "epoch": 1.571369171142578e-05,
      "step": 16477,
      "training_step_time": 17.12390971183777
    },
    {
      "epoch": 1.571464538574219e-05,
      "model_forward_time": 5.606958389282227,
      "step": 16478
    },
    {
      "epoch": 1.571464538574219e-05,
      "step": 16478,
      "training_step_time": 17.101722717285156
    },
    {
      "epoch": 1.5715599060058593e-05,
      "model_forward_time": 5.620452404022217,
      "step": 16479
    },
    {
      "epoch": 1.5715599060058593e-05,
      "step": 16479,
      "training_step_time": 17.097113609313965
    },
    {
      "epoch": 1.5716552734375e-05,
      "grad_norm": 0.462097704410553,
      "learning_rate": 1e-05,
      "loss": 0.0626,
      "step": 16480
    },
    {
      "epoch": 1.5716552734375e-05,
      "model_forward_time": 6.025133848190308,
      "step": 16480
    },
    {
      "dynamics_loss_avg": 0.05778681579977274,
      "epoch": 1.5716552734375e-05,
      "step": 16480
    },
    {
      "action_loss_avg": 0.006716978922486305,
      "epoch": 1.5716552734375e-05,
      "step": 16480
    },
    {
      "epoch": 1.5716552734375e-05,
      "step": 16480,
      "training_step_time": 17.512609243392944
    },
    {
      "epoch": 1.5717506408691408e-05,
      "model_forward_time": 5.619561195373535,
      "step": 16481
    },
    {
      "epoch": 1.5717506408691408e-05,
      "step": 16481,
      "training_step_time": 17.101155996322632
    },
    {
      "epoch": 1.5718460083007812e-05,
      "model_forward_time": 5.6215338706970215,
      "step": 16482
    },
    {
      "epoch": 1.5718460083007812e-05,
      "step": 16482,
      "training_step_time": 17.136985778808594
    },
    {
      "epoch": 1.571941375732422e-05,
      "model_forward_time": 5.628449440002441,
      "step": 16483
    },
    {
      "epoch": 1.571941375732422e-05,
      "step": 16483,
      "training_step_time": 17.140179872512817
    },
    {
      "epoch": 1.5720367431640627e-05,
      "model_forward_time": 5.6342480182647705,
      "step": 16484
    },
    {
      "epoch": 1.5720367431640627e-05,
      "step": 16484,
      "training_step_time": 17.136972427368164
    },
    {
      "epoch": 1.572132110595703e-05,
      "model_forward_time": 6.015656471252441,
      "step": 16485
    },
    {
      "epoch": 1.572132110595703e-05,
      "step": 16485,
      "training_step_time": 17.518682956695557
    },
    {
      "epoch": 1.5722274780273438e-05,
      "model_forward_time": 5.623655319213867,
      "step": 16486
    },
    {
      "epoch": 1.5722274780273438e-05,
      "step": 16486,
      "training_step_time": 17.111181259155273
    },
    {
      "epoch": 1.5723228454589842e-05,
      "model_forward_time": 5.648879051208496,
      "step": 16487
    },
    {
      "epoch": 1.5723228454589842e-05,
      "step": 16487,
      "training_step_time": 17.154168128967285
    },
    {
      "epoch": 1.572418212890625e-05,
      "model_forward_time": 5.627465724945068,
      "step": 16488
    },
    {
      "epoch": 1.572418212890625e-05,
      "step": 16488,
      "training_step_time": 17.119257926940918
    },
    {
      "epoch": 1.5725135803222657e-05,
      "model_forward_time": 5.61993932723999,
      "step": 16489
    },
    {
      "epoch": 1.5725135803222657e-05,
      "step": 16489,
      "training_step_time": 17.122657775878906
    },
    {
      "epoch": 1.572608947753906e-05,
      "grad_norm": 0.7607575058937073,
      "learning_rate": 1e-05,
      "loss": 0.0651,
      "step": 16490
    },
    {
      "epoch": 1.572608947753906e-05,
      "model_forward_time": 5.628899812698364,
      "step": 16490
    },
    {
      "dynamics_loss_avg": 0.055823277309536935,
      "epoch": 1.572608947753906e-05,
      "step": 16490
    },
    {
      "action_loss_avg": 0.007276987144723535,
      "epoch": 1.572608947753906e-05,
      "step": 16490
    },
    {
      "epoch": 1.572608947753906e-05,
      "step": 16490,
      "training_step_time": 17.531363010406494
    },
    {
      "epoch": 1.572704315185547e-05,
      "model_forward_time": 5.632575988769531,
      "step": 16491
    },
    {
      "epoch": 1.572704315185547e-05,
      "step": 16491,
      "training_step_time": 17.13422679901123
    },
    {
      "epoch": 1.5727996826171876e-05,
      "model_forward_time": 5.6022796630859375,
      "step": 16492
    },
    {
      "epoch": 1.5727996826171876e-05,
      "step": 16492,
      "training_step_time": 17.062700510025024
    },
    {
      "epoch": 1.572895050048828e-05,
      "model_forward_time": 5.618294715881348,
      "step": 16493
    },
    {
      "epoch": 1.572895050048828e-05,
      "step": 16493,
      "training_step_time": 17.07839870452881
    },
    {
      "epoch": 1.5729904174804687e-05,
      "model_forward_time": 5.599305629730225,
      "step": 16494
    },
    {
      "epoch": 1.5729904174804687e-05,
      "step": 16494,
      "training_step_time": 17.041240215301514
    },
    {
      "epoch": 1.5730857849121095e-05,
      "model_forward_time": 5.605307340621948,
      "step": 16495
    },
    {
      "epoch": 1.5730857849121095e-05,
      "step": 16495,
      "training_step_time": 17.061347007751465
    },
    {
      "epoch": 1.57318115234375e-05,
      "model_forward_time": 5.978215456008911,
      "step": 16496
    },
    {
      "epoch": 1.57318115234375e-05,
      "step": 16496,
      "training_step_time": 17.449299573898315
    },
    {
      "epoch": 1.5732765197753906e-05,
      "model_forward_time": 5.612600803375244,
      "step": 16497
    },
    {
      "epoch": 1.5732765197753906e-05,
      "step": 16497,
      "training_step_time": 17.124464750289917
    },
    {
      "epoch": 1.5733718872070314e-05,
      "model_forward_time": 5.622176647186279,
      "step": 16498
    },
    {
      "epoch": 1.5733718872070314e-05,
      "step": 16498,
      "training_step_time": 17.103066444396973
    },
    {
      "epoch": 1.5734672546386718e-05,
      "model_forward_time": 5.598194122314453,
      "step": 16499
    },
    {
      "epoch": 1.5734672546386718e-05,
      "step": 16499,
      "training_step_time": 17.011640787124634
    },
    {
      "epoch": 1.5735626220703125e-05,
      "grad_norm": 0.4465464949607849,
      "learning_rate": 1e-05,
      "loss": 0.0522,
      "step": 16500
    },
    {
      "epoch": 1.5735626220703125e-05,
      "model_forward_time": 5.579289197921753,
      "step": 16500
    },
    {
      "dynamics_loss_avg": 0.04454451855272055,
      "epoch": 1.5735626220703125e-05,
      "step": 16500
    },
    {
      "action_loss_avg": 0.006025619944557548,
      "epoch": 1.5735626220703125e-05,
      "step": 16500
    },
    {
      "epoch": 1.5735626220703125e-05,
      "step": 16500,
      "training_step_time": 17.016145706176758
    },
    {
      "epoch": 1.5736579895019532e-05,
      "model_forward_time": 5.989542484283447,
      "step": 16501
    },
    {
      "epoch": 1.5736579895019532e-05,
      "step": 16501,
      "training_step_time": 17.400241374969482
    },
    {
      "epoch": 1.5737533569335936e-05,
      "model_forward_time": 5.5796637535095215,
      "step": 16502
    },
    {
      "epoch": 1.5737533569335936e-05,
      "step": 16502,
      "training_step_time": 17.002243041992188
    },
    {
      "epoch": 1.5738487243652344e-05,
      "model_forward_time": 5.586651086807251,
      "step": 16503
    },
    {
      "epoch": 1.5738487243652344e-05,
      "step": 16503,
      "training_step_time": 17.003686666488647
    },
    {
      "epoch": 1.573944091796875e-05,
      "model_forward_time": 5.5805344581604,
      "step": 16504
    },
    {
      "epoch": 1.573944091796875e-05,
      "step": 16504,
      "training_step_time": 17.021454095840454
    },
    {
      "epoch": 1.5740394592285155e-05,
      "model_forward_time": 5.593630790710449,
      "step": 16505
    },
    {
      "epoch": 1.5740394592285155e-05,
      "step": 16505,
      "training_step_time": 17.06715989112854
    },
    {
      "epoch": 1.5741348266601563e-05,
      "model_forward_time": 5.633556842803955,
      "step": 16506
    },
    {
      "epoch": 1.5741348266601563e-05,
      "step": 16506,
      "training_step_time": 17.162650108337402
    },
    {
      "epoch": 1.574230194091797e-05,
      "model_forward_time": 6.031864881515503,
      "step": 16507
    },
    {
      "epoch": 1.574230194091797e-05,
      "step": 16507,
      "training_step_time": 17.5485942363739
    },
    {
      "epoch": 1.5743255615234374e-05,
      "model_forward_time": 5.622726678848267,
      "step": 16508
    },
    {
      "epoch": 1.5743255615234374e-05,
      "step": 16508,
      "training_step_time": 17.09697914123535
    },
    {
      "epoch": 1.574420928955078e-05,
      "model_forward_time": 5.623208045959473,
      "step": 16509
    },
    {
      "epoch": 1.574420928955078e-05,
      "step": 16509,
      "training_step_time": 17.10668969154358
    },
    {
      "epoch": 1.574516296386719e-05,
      "grad_norm": 0.7318785190582275,
      "learning_rate": 1e-05,
      "loss": 0.0678,
      "step": 16510
    },
    {
      "epoch": 1.574516296386719e-05,
      "model_forward_time": 5.605988502502441,
      "step": 16510
    },
    {
      "dynamics_loss_avg": 0.06091731451451778,
      "epoch": 1.574516296386719e-05,
      "step": 16510
    },
    {
      "action_loss_avg": 0.00818979712203145,
      "epoch": 1.574516296386719e-05,
      "step": 16510
    },
    {
      "epoch": 1.574516296386719e-05,
      "step": 16510,
      "training_step_time": 17.06144094467163
    },
    {
      "epoch": 1.5746116638183593e-05,
      "model_forward_time": 5.622243165969849,
      "step": 16511
    },
    {
      "epoch": 1.5746116638183593e-05,
      "step": 16511,
      "training_step_time": 17.090738534927368
    },
    {
      "epoch": 1.57470703125e-05,
      "model_forward_time": 6.029214382171631,
      "step": 16512
    },
    {
      "epoch": 1.57470703125e-05,
      "step": 16512,
      "training_step_time": 17.46540594100952
    },
    {
      "epoch": 1.5748023986816408e-05,
      "model_forward_time": 5.59873104095459,
      "step": 16513
    },
    {
      "epoch": 1.5748023986816408e-05,
      "step": 16513,
      "training_step_time": 17.032856225967407
    },
    {
      "epoch": 1.5748977661132812e-05,
      "model_forward_time": 5.607763767242432,
      "step": 16514
    },
    {
      "epoch": 1.5748977661132812e-05,
      "step": 16514,
      "training_step_time": 17.09815812110901
    },
    {
      "epoch": 1.574993133544922e-05,
      "model_forward_time": 5.667821645736694,
      "step": 16515
    },
    {
      "epoch": 1.574993133544922e-05,
      "step": 16515,
      "training_step_time": 17.21669912338257
    },
    {
      "epoch": 1.5750885009765623e-05,
      "model_forward_time": 5.633053302764893,
      "step": 16516
    },
    {
      "epoch": 1.5750885009765623e-05,
      "step": 16516,
      "training_step_time": 17.14912438392639
    },
    {
      "epoch": 1.575183868408203e-05,
      "model_forward_time": 5.629231214523315,
      "step": 16517
    },
    {
      "epoch": 1.575183868408203e-05,
      "step": 16517,
      "training_step_time": 17.132779598236084
    },
    {
      "epoch": 1.5752792358398438e-05,
      "model_forward_time": 5.987696886062622,
      "step": 16518
    },
    {
      "epoch": 1.5752792358398438e-05,
      "step": 16518,
      "training_step_time": 17.475799322128296
    },
    {
      "epoch": 1.5753746032714842e-05,
      "model_forward_time": 5.630068302154541,
      "step": 16519
    },
    {
      "epoch": 1.5753746032714842e-05,
      "step": 16519,
      "training_step_time": 17.136659383773804
    },
    {
      "epoch": 1.575469970703125e-05,
      "grad_norm": 0.9439754486083984,
      "learning_rate": 1e-05,
      "loss": 0.0529,
      "step": 16520
    },
    {
      "epoch": 1.575469970703125e-05,
      "model_forward_time": 5.624078273773193,
      "step": 16520
    },
    {
      "dynamics_loss_avg": 0.048594079725444314,
      "epoch": 1.575469970703125e-05,
      "step": 16520
    },
    {
      "action_loss_avg": 0.0055415764567442235,
      "epoch": 1.575469970703125e-05,
      "step": 16520
    },
    {
      "epoch": 1.575469970703125e-05,
      "step": 16520,
      "training_step_time": 17.125649452209473
    },
    {
      "epoch": 1.5755653381347657e-05,
      "model_forward_time": 5.6165900230407715,
      "step": 16521
    },
    {
      "epoch": 1.5755653381347657e-05,
      "step": 16521,
      "training_step_time": 17.096835136413574
    },
    {
      "epoch": 1.575660705566406e-05,
      "model_forward_time": 5.60755181312561,
      "step": 16522
    },
    {
      "epoch": 1.575660705566406e-05,
      "step": 16522,
      "training_step_time": 17.07571268081665
    },
    {
      "epoch": 1.575756072998047e-05,
      "model_forward_time": 6.020233631134033,
      "step": 16523
    },
    {
      "epoch": 1.575756072998047e-05,
      "step": 16523,
      "training_step_time": 17.480989456176758
    },
    {
      "epoch": 1.5758514404296876e-05,
      "model_forward_time": 5.602067708969116,
      "step": 16524
    },
    {
      "epoch": 1.5758514404296876e-05,
      "step": 16524,
      "training_step_time": 17.077614784240723
    },
    {
      "epoch": 1.575946807861328e-05,
      "model_forward_time": 5.600984811782837,
      "step": 16525
    },
    {
      "epoch": 1.575946807861328e-05,
      "step": 16525,
      "training_step_time": 17.099406480789185
    },
    {
      "epoch": 1.5760421752929687e-05,
      "model_forward_time": 5.621525287628174,
      "step": 16526
    },
    {
      "epoch": 1.5760421752929687e-05,
      "step": 16526,
      "training_step_time": 17.151053428649902
    },
    {
      "epoch": 1.5761375427246095e-05,
      "model_forward_time": 5.630467176437378,
      "step": 16527
    },
    {
      "epoch": 1.5761375427246095e-05,
      "step": 16527,
      "training_step_time": 17.158066034317017
    },
    {
      "epoch": 1.57623291015625e-05,
      "model_forward_time": 5.630361795425415,
      "step": 16528
    },
    {
      "epoch": 1.57623291015625e-05,
      "step": 16528,
      "training_step_time": 17.15082097053528
    },
    {
      "epoch": 1.5763282775878906e-05,
      "model_forward_time": 6.009432315826416,
      "step": 16529
    },
    {
      "epoch": 1.5763282775878906e-05,
      "step": 16529,
      "training_step_time": 17.510971069335938
    },
    {
      "epoch": 1.5764236450195314e-05,
      "grad_norm": 0.6652183532714844,
      "learning_rate": 1e-05,
      "loss": 0.0636,
      "step": 16530
    },
    {
      "epoch": 1.5764236450195314e-05,
      "model_forward_time": 5.617969751358032,
      "step": 16530
    },
    {
      "dynamics_loss_avg": 0.056062296219170096,
      "epoch": 1.5764236450195314e-05,
      "step": 16530
    },
    {
      "action_loss_avg": 0.007969913887791336,
      "epoch": 1.5764236450195314e-05,
      "step": 16530
    },
    {
      "epoch": 1.5764236450195314e-05,
      "step": 16530,
      "training_step_time": 17.098165273666382
    },
    {
      "epoch": 1.5765190124511718e-05,
      "model_forward_time": 5.620145559310913,
      "step": 16531
    },
    {
      "epoch": 1.5765190124511718e-05,
      "step": 16531,
      "training_step_time": 17.116375207901
    },
    {
      "epoch": 1.5766143798828125e-05,
      "model_forward_time": 5.627118349075317,
      "step": 16532
    },
    {
      "epoch": 1.5766143798828125e-05,
      "step": 16532,
      "training_step_time": 17.106704473495483
    },
    {
      "epoch": 1.5767097473144532e-05,
      "model_forward_time": 5.612746953964233,
      "step": 16533
    },
    {
      "epoch": 1.5767097473144532e-05,
      "step": 16533,
      "training_step_time": 17.0841064453125
    },
    {
      "epoch": 1.5768051147460936e-05,
      "model_forward_time": 6.015198707580566,
      "step": 16534
    },
    {
      "epoch": 1.5768051147460936e-05,
      "step": 16534,
      "training_step_time": 17.471070051193237
    },
    {
      "epoch": 1.5769004821777344e-05,
      "model_forward_time": 5.6048057079315186,
      "step": 16535
    },
    {
      "epoch": 1.5769004821777344e-05,
      "step": 16535,
      "training_step_time": 17.068138122558594
    },
    {
      "epoch": 1.576995849609375e-05,
      "model_forward_time": 5.63782525062561,
      "step": 16536
    },
    {
      "epoch": 1.576995849609375e-05,
      "step": 16536,
      "training_step_time": 17.16080069541931
    },
    {
      "epoch": 1.5770912170410155e-05,
      "model_forward_time": 5.624204874038696,
      "step": 16537
    },
    {
      "epoch": 1.5770912170410155e-05,
      "step": 16537,
      "training_step_time": 17.121989488601685
    },
    {
      "epoch": 1.5771865844726563e-05,
      "model_forward_time": 5.629007577896118,
      "step": 16538
    },
    {
      "epoch": 1.5771865844726563e-05,
      "step": 16538,
      "training_step_time": 17.108213901519775
    },
    {
      "epoch": 1.577281951904297e-05,
      "model_forward_time": 6.020367622375488,
      "step": 16539
    },
    {
      "epoch": 1.577281951904297e-05,
      "step": 16539,
      "training_step_time": 17.518089532852173
    },
    {
      "epoch": 1.5773773193359374e-05,
      "grad_norm": 1.3556573390960693,
      "learning_rate": 1e-05,
      "loss": 0.086,
      "step": 16540
    },
    {
      "epoch": 1.5773773193359374e-05,
      "model_forward_time": 5.614652156829834,
      "step": 16540
    },
    {
      "dynamics_loss_avg": 0.07736830618232489,
      "epoch": 1.5773773193359374e-05,
      "step": 16540
    },
    {
      "action_loss_avg": 0.011675431043840945,
      "epoch": 1.5773773193359374e-05,
      "step": 16540
    },
    {
      "epoch": 1.5773773193359374e-05,
      "step": 16540,
      "training_step_time": 17.111302852630615
    },
    {
      "epoch": 1.577472686767578e-05,
      "model_forward_time": 5.6179280281066895,
      "step": 16541
    },
    {
      "epoch": 1.577472686767578e-05,
      "step": 16541,
      "training_step_time": 17.091692209243774
    },
    {
      "epoch": 1.577568054199219e-05,
      "model_forward_time": 5.604743480682373,
      "step": 16542
    },
    {
      "epoch": 1.577568054199219e-05,
      "step": 16542,
      "training_step_time": 17.049527883529663
    },
    {
      "epoch": 1.5776634216308593e-05,
      "model_forward_time": 5.601353883743286,
      "step": 16543
    },
    {
      "epoch": 1.5776634216308593e-05,
      "step": 16543,
      "training_step_time": 17.05330228805542
    },
    {
      "epoch": 1.5777587890625e-05,
      "model_forward_time": 5.605177164077759,
      "step": 16544
    },
    {
      "epoch": 1.5777587890625e-05,
      "step": 16544,
      "training_step_time": 17.085840940475464
    },
    {
      "epoch": 1.5778541564941408e-05,
      "model_forward_time": 6.062511444091797,
      "step": 16545
    },
    {
      "epoch": 1.5778541564941408e-05,
      "step": 16545,
      "training_step_time": 17.557692527770996
    },
    {
      "epoch": 1.5779495239257812e-05,
      "model_forward_time": 5.621275186538696,
      "step": 16546
    },
    {
      "epoch": 1.5779495239257812e-05,
      "step": 16546,
      "training_step_time": 17.121885538101196
    },
    {
      "epoch": 1.578044891357422e-05,
      "model_forward_time": 5.630007266998291,
      "step": 16547
    },
    {
      "epoch": 1.578044891357422e-05,
      "step": 16547,
      "training_step_time": 17.152724266052246
    },
    {
      "epoch": 1.5781402587890623e-05,
      "model_forward_time": 5.629918336868286,
      "step": 16548
    },
    {
      "epoch": 1.5781402587890623e-05,
      "step": 16548,
      "training_step_time": 17.131463050842285
    },
    {
      "epoch": 1.578235626220703e-05,
      "model_forward_time": 5.62874960899353,
      "step": 16549
    },
    {
      "epoch": 1.578235626220703e-05,
      "step": 16549,
      "training_step_time": 17.13750123977661
    },
    {
      "epoch": 1.5783309936523438e-05,
      "grad_norm": 0.6988507509231567,
      "learning_rate": 1e-05,
      "loss": 0.0635,
      "step": 16550
    },
    {
      "epoch": 1.5783309936523438e-05,
      "model_forward_time": 5.6225361824035645,
      "step": 16550
    },
    {
      "dynamics_loss_avg": 0.05330050885677338,
      "epoch": 1.5783309936523438e-05,
      "step": 16550
    },
    {
      "action_loss_avg": 0.006291083968244493,
      "epoch": 1.5783309936523438e-05,
      "step": 16550
    },
    {
      "epoch": 1.5783309936523438e-05,
      "step": 16550,
      "training_step_time": 17.539064407348633
    },
    {
      "epoch": 1.5784263610839842e-05,
      "model_forward_time": 5.61676287651062,
      "step": 16551
    },
    {
      "epoch": 1.5784263610839842e-05,
      "step": 16551,
      "training_step_time": 17.107730865478516
    },
    {
      "epoch": 1.578521728515625e-05,
      "model_forward_time": 5.6057047843933105,
      "step": 16552
    },
    {
      "epoch": 1.578521728515625e-05,
      "step": 16552,
      "training_step_time": 17.09323000907898
    },
    {
      "epoch": 1.5786170959472657e-05,
      "model_forward_time": 5.60751748085022,
      "step": 16553
    },
    {
      "epoch": 1.5786170959472657e-05,
      "step": 16553,
      "training_step_time": 17.08701491355896
    },
    {
      "epoch": 1.578712463378906e-05,
      "model_forward_time": 5.602524280548096,
      "step": 16554
    },
    {
      "epoch": 1.578712463378906e-05,
      "step": 16554,
      "training_step_time": 17.082452297210693
    },
    {
      "epoch": 1.578807830810547e-05,
      "model_forward_time": 5.616121530532837,
      "step": 16555
    },
    {
      "epoch": 1.578807830810547e-05,
      "step": 16555,
      "training_step_time": 17.130804777145386
    },
    {
      "epoch": 1.5789031982421876e-05,
      "model_forward_time": 6.031532287597656,
      "step": 16556
    },
    {
      "epoch": 1.5789031982421876e-05,
      "step": 16556,
      "training_step_time": 17.540323734283447
    },
    {
      "epoch": 1.578998565673828e-05,
      "model_forward_time": 5.622015476226807,
      "step": 16557
    },
    {
      "epoch": 1.578998565673828e-05,
      "step": 16557,
      "training_step_time": 17.143213272094727
    },
    {
      "epoch": 1.5790939331054687e-05,
      "model_forward_time": 5.629096746444702,
      "step": 16558
    },
    {
      "epoch": 1.5790939331054687e-05,
      "step": 16558,
      "training_step_time": 17.13459062576294
    },
    {
      "epoch": 1.5791893005371095e-05,
      "model_forward_time": 5.63692045211792,
      "step": 16559
    },
    {
      "epoch": 1.5791893005371095e-05,
      "step": 16559,
      "training_step_time": 17.13163948059082
    },
    {
      "epoch": 1.57928466796875e-05,
      "grad_norm": 0.6022419333457947,
      "learning_rate": 1e-05,
      "loss": 0.0676,
      "step": 16560
    },
    {
      "epoch": 1.57928466796875e-05,
      "model_forward_time": 5.615494251251221,
      "step": 16560
    },
    {
      "dynamics_loss_avg": 0.059362147189676764,
      "epoch": 1.57928466796875e-05,
      "step": 16560
    },
    {
      "action_loss_avg": 0.007887380302418024,
      "epoch": 1.57928466796875e-05,
      "step": 16560
    },
    {
      "epoch": 1.57928466796875e-05,
      "step": 16560,
      "training_step_time": 17.116975784301758
    },
    {
      "epoch": 1.5793800354003906e-05,
      "model_forward_time": 5.618993759155273,
      "step": 16561
    },
    {
      "epoch": 1.5793800354003906e-05,
      "step": 16561,
      "training_step_time": 17.52949023246765
    },
    {
      "epoch": 1.5794754028320314e-05,
      "model_forward_time": 5.6152424812316895,
      "step": 16562
    },
    {
      "epoch": 1.5794754028320314e-05,
      "step": 16562,
      "training_step_time": 17.09815764427185
    },
    {
      "epoch": 1.5795707702636718e-05,
      "model_forward_time": 5.601165294647217,
      "step": 16563
    },
    {
      "epoch": 1.5795707702636718e-05,
      "step": 16563,
      "training_step_time": 17.07408618927002
    },
    {
      "epoch": 1.5796661376953125e-05,
      "model_forward_time": 5.602932453155518,
      "step": 16564
    },
    {
      "epoch": 1.5796661376953125e-05,
      "step": 16564,
      "training_step_time": 17.074498891830444
    },
    {
      "epoch": 1.5797615051269532e-05,
      "model_forward_time": 5.627500295639038,
      "step": 16565
    },
    {
      "epoch": 1.5797615051269532e-05,
      "step": 16565,
      "training_step_time": 17.14486026763916
    },
    {
      "epoch": 1.5798568725585936e-05,
      "model_forward_time": 5.6167151927948,
      "step": 16566
    },
    {
      "epoch": 1.5798568725585936e-05,
      "step": 16566,
      "training_step_time": 17.15775203704834
    },
    {
      "epoch": 1.5799522399902344e-05,
      "model_forward_time": 6.0550217628479,
      "step": 16567
    },
    {
      "epoch": 1.5799522399902344e-05,
      "step": 16567,
      "training_step_time": 17.58432412147522
    },
    {
      "epoch": 1.580047607421875e-05,
      "model_forward_time": 5.646570682525635,
      "step": 16568
    },
    {
      "epoch": 1.580047607421875e-05,
      "step": 16568,
      "training_step_time": 17.19438123703003
    },
    {
      "epoch": 1.5801429748535155e-05,
      "model_forward_time": 5.652612209320068,
      "step": 16569
    },
    {
      "epoch": 1.5801429748535155e-05,
      "step": 16569,
      "training_step_time": 17.144524812698364
    },
    {
      "epoch": 1.5802383422851563e-05,
      "grad_norm": 1.0113698244094849,
      "learning_rate": 1e-05,
      "loss": 0.0551,
      "step": 16570
    },
    {
      "epoch": 1.5802383422851563e-05,
      "model_forward_time": 5.593384742736816,
      "step": 16570
    },
    {
      "dynamics_loss_avg": 0.050034426525235175,
      "epoch": 1.5802383422851563e-05,
      "step": 16570
    },
    {
      "action_loss_avg": 0.0071834246162325146,
      "epoch": 1.5802383422851563e-05,
      "step": 16570
    },
    {
      "epoch": 1.5802383422851563e-05,
      "step": 16570,
      "training_step_time": 17.013779163360596
    },
    {
      "epoch": 1.580333709716797e-05,
      "model_forward_time": 5.6122424602508545,
      "step": 16571
    },
    {
      "epoch": 1.580333709716797e-05,
      "step": 16571,
      "training_step_time": 17.056535720825195
    },
    {
      "epoch": 1.5804290771484374e-05,
      "model_forward_time": 5.976512670516968,
      "step": 16572
    },
    {
      "epoch": 1.5804290771484374e-05,
      "step": 16572,
      "training_step_time": 17.420135736465454
    },
    {
      "epoch": 1.580524444580078e-05,
      "model_forward_time": 5.585229873657227,
      "step": 16573
    },
    {
      "epoch": 1.580524444580078e-05,
      "step": 16573,
      "training_step_time": 17.00933051109314
    },
    {
      "epoch": 1.580619812011719e-05,
      "model_forward_time": 5.603094577789307,
      "step": 16574
    },
    {
      "epoch": 1.580619812011719e-05,
      "step": 16574,
      "training_step_time": 17.085230827331543
    },
    {
      "epoch": 1.5807151794433593e-05,
      "model_forward_time": 5.601170063018799,
      "step": 16575
    },
    {
      "epoch": 1.5807151794433593e-05,
      "step": 16575,
      "training_step_time": 17.033560037612915
    },
    {
      "epoch": 1.580810546875e-05,
      "model_forward_time": 5.595770597457886,
      "step": 16576
    },
    {
      "epoch": 1.580810546875e-05,
      "step": 16576,
      "training_step_time": 17.0747492313385
    },
    {
      "epoch": 1.5809059143066408e-05,
      "model_forward_time": 5.602099657058716,
      "step": 16577
    },
    {
      "epoch": 1.5809059143066408e-05,
      "step": 16577,
      "training_step_time": 17.083688020706177
    },
    {
      "epoch": 1.5810012817382812e-05,
      "model_forward_time": 6.02225923538208,
      "step": 16578
    },
    {
      "epoch": 1.5810012817382812e-05,
      "step": 16578,
      "training_step_time": 17.525165796279907
    },
    {
      "epoch": 1.581096649169922e-05,
      "model_forward_time": 5.643955230712891,
      "step": 16579
    },
    {
      "epoch": 1.581096649169922e-05,
      "step": 16579,
      "training_step_time": 17.16472029685974
    },
    {
      "epoch": 1.5811920166015623e-05,
      "grad_norm": 0.13333232700824738,
      "learning_rate": 1e-05,
      "loss": 0.0505,
      "step": 16580
    },
    {
      "epoch": 1.5811920166015623e-05,
      "model_forward_time": 5.640831232070923,
      "step": 16580
    },
    {
      "dynamics_loss_avg": 0.04652285967022181,
      "epoch": 1.5811920166015623e-05,
      "step": 16580
    },
    {
      "action_loss_avg": 0.00553787323879078,
      "epoch": 1.5811920166015623e-05,
      "step": 16580
    },
    {
      "epoch": 1.5811920166015623e-05,
      "step": 16580,
      "training_step_time": 17.157294988632202
    },
    {
      "epoch": 1.581287384033203e-05,
      "model_forward_time": 5.619386434555054,
      "step": 16581
    },
    {
      "epoch": 1.581287384033203e-05,
      "step": 16581,
      "training_step_time": 17.082274198532104
    },
    {
      "epoch": 1.5813827514648438e-05,
      "model_forward_time": 5.594900846481323,
      "step": 16582
    },
    {
      "epoch": 1.5813827514648438e-05,
      "step": 16582,
      "training_step_time": 17.06378197669983
    },
    {
      "epoch": 1.5814781188964842e-05,
      "model_forward_time": 5.610151529312134,
      "step": 16583
    },
    {
      "epoch": 1.5814781188964842e-05,
      "step": 16583,
      "training_step_time": 17.508224725723267
    },
    {
      "epoch": 1.581573486328125e-05,
      "model_forward_time": 5.6171181201934814,
      "step": 16584
    },
    {
      "epoch": 1.581573486328125e-05,
      "step": 16584,
      "training_step_time": 17.11737084388733
    },
    {
      "epoch": 1.5816688537597657e-05,
      "model_forward_time": 5.611469745635986,
      "step": 16585
    },
    {
      "epoch": 1.5816688537597657e-05,
      "step": 16585,
      "training_step_time": 17.099893808364868
    },
    {
      "epoch": 1.581764221191406e-05,
      "model_forward_time": 5.621875047683716,
      "step": 16586
    },
    {
      "epoch": 1.581764221191406e-05,
      "step": 16586,
      "training_step_time": 17.115681409835815
    },
    {
      "epoch": 1.581859588623047e-05,
      "model_forward_time": 5.610795736312866,
      "step": 16587
    },
    {
      "epoch": 1.581859588623047e-05,
      "step": 16587,
      "training_step_time": 17.1102352142334
    },
    {
      "epoch": 1.5819549560546876e-05,
      "model_forward_time": 5.626490592956543,
      "step": 16588
    },
    {
      "epoch": 1.5819549560546876e-05,
      "step": 16588,
      "training_step_time": 17.1035635471344
    },
    {
      "epoch": 1.582050323486328e-05,
      "model_forward_time": 6.030799388885498,
      "step": 16589
    },
    {
      "epoch": 1.582050323486328e-05,
      "step": 16589,
      "training_step_time": 17.466594219207764
    },
    {
      "epoch": 1.5821456909179687e-05,
      "grad_norm": 0.6832829713821411,
      "learning_rate": 1e-05,
      "loss": 0.0661,
      "step": 16590
    },
    {
      "epoch": 1.5821456909179687e-05,
      "model_forward_time": 5.589594125747681,
      "step": 16590
    },
    {
      "dynamics_loss_avg": 0.057256138511002065,
      "epoch": 1.5821456909179687e-05,
      "step": 16590
    },
    {
      "action_loss_avg": 0.005985202151350677,
      "epoch": 1.5821456909179687e-05,
      "step": 16590
    },
    {
      "epoch": 1.5821456909179687e-05,
      "step": 16590,
      "training_step_time": 17.042224407196045
    },
    {
      "epoch": 1.5822410583496095e-05,
      "model_forward_time": 5.592936277389526,
      "step": 16591
    },
    {
      "epoch": 1.5822410583496095e-05,
      "step": 16591,
      "training_step_time": 17.02919840812683
    },
    {
      "epoch": 1.58233642578125e-05,
      "model_forward_time": 5.59258246421814,
      "step": 16592
    },
    {
      "epoch": 1.58233642578125e-05,
      "step": 16592,
      "training_step_time": 17.036646842956543
    },
    {
      "epoch": 1.5824317932128906e-05,
      "model_forward_time": 5.612107753753662,
      "step": 16593
    },
    {
      "epoch": 1.5824317932128906e-05,
      "step": 16593,
      "training_step_time": 17.047557830810547
    },
    {
      "epoch": 1.5825271606445314e-05,
      "model_forward_time": 5.608320951461792,
      "step": 16594
    },
    {
      "epoch": 1.5825271606445314e-05,
      "step": 16594,
      "training_step_time": 17.452406406402588
    },
    {
      "epoch": 1.5826225280761718e-05,
      "model_forward_time": 5.598423480987549,
      "step": 16595
    },
    {
      "epoch": 1.5826225280761718e-05,
      "step": 16595,
      "training_step_time": 16.993324518203735
    },
    {
      "epoch": 1.5827178955078125e-05,
      "model_forward_time": 5.612158298492432,
      "step": 16596
    },
    {
      "epoch": 1.5827178955078125e-05,
      "step": 16596,
      "training_step_time": 17.120239734649658
    },
    {
      "epoch": 1.5828132629394532e-05,
      "model_forward_time": 5.634332895278931,
      "step": 16597
    },
    {
      "epoch": 1.5828132629394532e-05,
      "step": 16597,
      "training_step_time": 17.157949686050415
    },
    {
      "epoch": 1.5829086303710937e-05,
      "model_forward_time": 5.631800651550293,
      "step": 16598
    },
    {
      "epoch": 1.5829086303710937e-05,
      "step": 16598,
      "training_step_time": 17.148632287979126
    },
    {
      "epoch": 1.5830039978027344e-05,
      "model_forward_time": 5.627631902694702,
      "step": 16599
    },
    {
      "epoch": 1.5830039978027344e-05,
      "step": 16599,
      "training_step_time": 17.138795375823975
    },
    {
      "epoch": 1.583099365234375e-05,
      "grad_norm": 0.45945343375205994,
      "learning_rate": 1e-05,
      "loss": 0.0573,
      "step": 16600
    },
    {
      "epoch": 1.583099365234375e-05,
      "model_forward_time": 6.015141487121582,
      "step": 16600
    },
    {
      "dynamics_loss_avg": 0.05104893045499921,
      "epoch": 1.583099365234375e-05,
      "step": 16600
    },
    {
      "action_loss_avg": 0.006757951536565087,
      "epoch": 1.583099365234375e-05,
      "step": 16600
    },
    {
      "epoch": 1.583099365234375e-05,
      "step": 16600,
      "training_step_time": 17.49993896484375
    },
    {
      "epoch": 1.5831947326660155e-05,
      "model_forward_time": 5.612837314605713,
      "step": 16601
    },
    {
      "epoch": 1.5831947326660155e-05,
      "step": 16601,
      "training_step_time": 17.104450225830078
    },
    {
      "epoch": 1.5832901000976563e-05,
      "model_forward_time": 5.605157375335693,
      "step": 16602
    },
    {
      "epoch": 1.5832901000976563e-05,
      "step": 16602,
      "training_step_time": 17.07892370223999
    },
    {
      "epoch": 1.583385467529297e-05,
      "model_forward_time": 5.603850603103638,
      "step": 16603
    },
    {
      "epoch": 1.583385467529297e-05,
      "step": 16603,
      "training_step_time": 17.094792127609253
    },
    {
      "epoch": 1.5834808349609374e-05,
      "model_forward_time": 5.609847545623779,
      "step": 16604
    },
    {
      "epoch": 1.5834808349609374e-05,
      "step": 16604,
      "training_step_time": 17.119468688964844
    },
    {
      "epoch": 1.583576202392578e-05,
      "model_forward_time": 6.041328191757202,
      "step": 16605
    },
    {
      "epoch": 1.583576202392578e-05,
      "step": 16605,
      "training_step_time": 17.539069890975952
    },
    {
      "epoch": 1.583671569824219e-05,
      "model_forward_time": 5.61877703666687,
      "step": 16606
    },
    {
      "epoch": 1.583671569824219e-05,
      "step": 16606,
      "training_step_time": 17.11875295639038
    },
    {
      "epoch": 1.5837669372558593e-05,
      "model_forward_time": 5.624323844909668,
      "step": 16607
    },
    {
      "epoch": 1.5837669372558593e-05,
      "step": 16607,
      "training_step_time": 17.136946201324463
    },
    {
      "epoch": 1.5838623046875e-05,
      "model_forward_time": 5.652595043182373,
      "step": 16608
    },
    {
      "epoch": 1.5838623046875e-05,
      "step": 16608,
      "training_step_time": 17.181288480758667
    },
    {
      "epoch": 1.5839576721191408e-05,
      "model_forward_time": 5.64124608039856,
      "step": 16609
    },
    {
      "epoch": 1.5839576721191408e-05,
      "step": 16609,
      "training_step_time": 17.13191270828247
    },
    {
      "epoch": 1.5840530395507812e-05,
      "grad_norm": 0.7827630043029785,
      "learning_rate": 1e-05,
      "loss": 0.0557,
      "step": 16610
    },
    {
      "epoch": 1.5840530395507812e-05,
      "model_forward_time": 5.611800193786621,
      "step": 16610
    },
    {
      "dynamics_loss_avg": 0.04779489804059267,
      "epoch": 1.5840530395507812e-05,
      "step": 16610
    },
    {
      "action_loss_avg": 0.005603779968805611,
      "epoch": 1.5840530395507812e-05,
      "step": 16610
    },
    {
      "epoch": 1.5840530395507812e-05,
      "step": 16610,
      "training_step_time": 17.08227801322937
    },
    {
      "epoch": 1.584148406982422e-05,
      "model_forward_time": 6.005634546279907,
      "step": 16611
    },
    {
      "epoch": 1.584148406982422e-05,
      "step": 16611,
      "training_step_time": 17.451682567596436
    },
    {
      "epoch": 1.5842437744140623e-05,
      "model_forward_time": 5.591141700744629,
      "step": 16612
    },
    {
      "epoch": 1.5842437744140623e-05,
      "step": 16612,
      "training_step_time": 17.033869743347168
    },
    {
      "epoch": 1.584339141845703e-05,
      "model_forward_time": 5.596343517303467,
      "step": 16613
    },
    {
      "epoch": 1.584339141845703e-05,
      "step": 16613,
      "training_step_time": 17.054478883743286
    },
    {
      "epoch": 1.5844345092773438e-05,
      "model_forward_time": 5.610568523406982,
      "step": 16614
    },
    {
      "epoch": 1.5844345092773438e-05,
      "step": 16614,
      "training_step_time": 17.09122347831726
    },
    {
      "epoch": 1.5845298767089842e-05,
      "model_forward_time": 5.607040166854858,
      "step": 16615
    },
    {
      "epoch": 1.5845298767089842e-05,
      "step": 16615,
      "training_step_time": 17.089905738830566
    },
    {
      "epoch": 1.584625244140625e-05,
      "model_forward_time": 5.599812269210815,
      "step": 16616
    },
    {
      "epoch": 1.584625244140625e-05,
      "step": 16616,
      "training_step_time": 17.45862603187561
    },
    {
      "epoch": 1.5847206115722657e-05,
      "model_forward_time": 5.607630968093872,
      "step": 16617
    },
    {
      "epoch": 1.5847206115722657e-05,
      "step": 16617,
      "training_step_time": 17.059739351272583
    },
    {
      "epoch": 1.584815979003906e-05,
      "model_forward_time": 5.590472936630249,
      "step": 16618
    },
    {
      "epoch": 1.584815979003906e-05,
      "step": 16618,
      "training_step_time": 17.038464784622192
    },
    {
      "epoch": 1.584911346435547e-05,
      "model_forward_time": 5.597581386566162,
      "step": 16619
    },
    {
      "epoch": 1.584911346435547e-05,
      "step": 16619,
      "training_step_time": 17.065677404403687
    },
    {
      "epoch": 1.5850067138671876e-05,
      "grad_norm": 0.47788652777671814,
      "learning_rate": 1e-05,
      "loss": 0.0519,
      "step": 16620
    },
    {
      "epoch": 1.5850067138671876e-05,
      "model_forward_time": 5.615196943283081,
      "step": 16620
    },
    {
      "dynamics_loss_avg": 0.0474936954677105,
      "epoch": 1.5850067138671876e-05,
      "step": 16620
    },
    {
      "action_loss_avg": 0.005998197675216943,
      "epoch": 1.5850067138671876e-05,
      "step": 16620
    },
    {
      "epoch": 1.5850067138671876e-05,
      "step": 16620,
      "training_step_time": 17.092572927474976
    },
    {
      "epoch": 1.585102081298828e-05,
      "model_forward_time": 5.6044323444366455,
      "step": 16621
    },
    {
      "epoch": 1.585102081298828e-05,
      "step": 16621,
      "training_step_time": 17.07501196861267
    },
    {
      "epoch": 1.5851974487304687e-05,
      "model_forward_time": 6.020897626876831,
      "step": 16622
    },
    {
      "epoch": 1.5851974487304687e-05,
      "step": 16622,
      "training_step_time": 17.521912336349487
    },
    {
      "epoch": 1.5852928161621095e-05,
      "model_forward_time": 5.634084939956665,
      "step": 16623
    },
    {
      "epoch": 1.5852928161621095e-05,
      "step": 16623,
      "training_step_time": 17.137532711029053
    },
    {
      "epoch": 1.58538818359375e-05,
      "model_forward_time": 5.611686706542969,
      "step": 16624
    },
    {
      "epoch": 1.58538818359375e-05,
      "step": 16624,
      "training_step_time": 17.072407007217407
    },
    {
      "epoch": 1.5854835510253906e-05,
      "model_forward_time": 5.597736120223999,
      "step": 16625
    },
    {
      "epoch": 1.5854835510253906e-05,
      "step": 16625,
      "training_step_time": 17.04744791984558
    },
    {
      "epoch": 1.5855789184570314e-05,
      "model_forward_time": 5.598802804946899,
      "step": 16626
    },
    {
      "epoch": 1.5855789184570314e-05,
      "step": 16626,
      "training_step_time": 17.064373016357422
    },
    {
      "epoch": 1.5856742858886718e-05,
      "model_forward_time": 5.601545095443726,
      "step": 16627
    },
    {
      "epoch": 1.5856742858886718e-05,
      "step": 16627,
      "training_step_time": 17.46763586997986
    },
    {
      "epoch": 1.5857696533203125e-05,
      "model_forward_time": 5.616441965103149,
      "step": 16628
    },
    {
      "epoch": 1.5857696533203125e-05,
      "step": 16628,
      "training_step_time": 17.099848747253418
    },
    {
      "epoch": 1.5858650207519533e-05,
      "model_forward_time": 5.645140647888184,
      "step": 16629
    },
    {
      "epoch": 1.5858650207519533e-05,
      "step": 16629,
      "training_step_time": 17.154773712158203
    },
    {
      "epoch": 1.5859603881835937e-05,
      "grad_norm": 0.22841550409793854,
      "learning_rate": 1e-05,
      "loss": 0.0472,
      "step": 16630
    },
    {
      "epoch": 1.5859603881835937e-05,
      "model_forward_time": 5.633078098297119,
      "step": 16630
    },
    {
      "dynamics_loss_avg": 0.0429855840280652,
      "epoch": 1.5859603881835937e-05,
      "step": 16630
    },
    {
      "action_loss_avg": 0.005147912609390915,
      "epoch": 1.5859603881835937e-05,
      "step": 16630
    },
    {
      "epoch": 1.5859603881835937e-05,
      "step": 16630,
      "training_step_time": 17.13737463951111
    },
    {
      "epoch": 1.5860557556152344e-05,
      "model_forward_time": 5.619074583053589,
      "step": 16631
    },
    {
      "epoch": 1.5860557556152344e-05,
      "step": 16631,
      "training_step_time": 17.09924006462097
    },
    {
      "epoch": 1.586151123046875e-05,
      "model_forward_time": 5.598516464233398,
      "step": 16632
    },
    {
      "epoch": 1.586151123046875e-05,
      "step": 16632,
      "training_step_time": 17.046247243881226
    },
    {
      "epoch": 1.5862464904785155e-05,
      "model_forward_time": 5.993001937866211,
      "step": 16633
    },
    {
      "epoch": 1.5862464904785155e-05,
      "step": 16633,
      "training_step_time": 17.42344069480896
    },
    {
      "epoch": 1.5863418579101563e-05,
      "model_forward_time": 5.58849573135376,
      "step": 16634
    },
    {
      "epoch": 1.5863418579101563e-05,
      "step": 16634,
      "training_step_time": 17.04494047164917
    },
    {
      "epoch": 1.586437225341797e-05,
      "model_forward_time": 5.593461036682129,
      "step": 16635
    },
    {
      "epoch": 1.586437225341797e-05,
      "step": 16635,
      "training_step_time": 17.033851861953735
    },
    {
      "epoch": 1.5865325927734374e-05,
      "model_forward_time": 5.586402177810669,
      "step": 16636
    },
    {
      "epoch": 1.5865325927734374e-05,
      "step": 16636,
      "training_step_time": 17.037638902664185
    },
    {
      "epoch": 1.586627960205078e-05,
      "model_forward_time": 5.586588382720947,
      "step": 16637
    },
    {
      "epoch": 1.586627960205078e-05,
      "step": 16637,
      "training_step_time": 17.028104543685913
    },
    {
      "epoch": 1.586723327636719e-05,
      "model_forward_time": 6.000323295593262,
      "step": 16638
    },
    {
      "epoch": 1.586723327636719e-05,
      "step": 16638,
      "training_step_time": 17.41902995109558
    },
    {
      "epoch": 1.5868186950683593e-05,
      "model_forward_time": 5.606330633163452,
      "step": 16639
    },
    {
      "epoch": 1.5868186950683593e-05,
      "step": 16639,
      "training_step_time": 17.0538911819458
    },
    {
      "epoch": 1.5869140625e-05,
      "grad_norm": 1.2569941282272339,
      "learning_rate": 1e-05,
      "loss": 0.0728,
      "step": 16640
    },
    {
      "epoch": 1.5869140625e-05,
      "model_forward_time": 5.613300085067749,
      "step": 16640
    },
    {
      "dynamics_loss_avg": 0.06631815154105425,
      "epoch": 1.5869140625e-05,
      "step": 16640
    },
    {
      "action_loss_avg": 0.007861044444143771,
      "epoch": 1.5869140625e-05,
      "step": 16640
    },
    {
      "epoch": 1.5869140625e-05,
      "step": 16640,
      "training_step_time": 17.09579563140869
    },
    {
      "epoch": 1.5870094299316405e-05,
      "model_forward_time": 5.630393743515015,
      "step": 16641
    },
    {
      "epoch": 1.5870094299316405e-05,
      "step": 16641,
      "training_step_time": 17.136547803878784
    },
    {
      "epoch": 1.5871047973632812e-05,
      "model_forward_time": 5.620943784713745,
      "step": 16642
    },
    {
      "epoch": 1.5871047973632812e-05,
      "step": 16642,
      "training_step_time": 17.132219791412354
    },
    {
      "epoch": 1.587200164794922e-05,
      "model_forward_time": 5.6457414627075195,
      "step": 16643
    },
    {
      "epoch": 1.587200164794922e-05,
      "step": 16643,
      "training_step_time": 17.153731107711792
    },
    {
      "epoch": 1.5872955322265623e-05,
      "model_forward_time": 6.0316503047943115,
      "step": 16644
    },
    {
      "epoch": 1.5872955322265623e-05,
      "step": 16644,
      "training_step_time": 17.549686431884766
    },
    {
      "epoch": 1.587390899658203e-05,
      "model_forward_time": 5.62221884727478,
      "step": 16645
    },
    {
      "epoch": 1.587390899658203e-05,
      "step": 16645,
      "training_step_time": 17.12570834159851
    },
    {
      "epoch": 1.5874862670898438e-05,
      "model_forward_time": 5.620188474655151,
      "step": 16646
    },
    {
      "epoch": 1.5874862670898438e-05,
      "step": 16646,
      "training_step_time": 17.130724668502808
    },
    {
      "epoch": 1.5875816345214842e-05,
      "model_forward_time": 5.623798608779907,
      "step": 16647
    },
    {
      "epoch": 1.5875816345214842e-05,
      "step": 16647,
      "training_step_time": 17.10874080657959
    },
    {
      "epoch": 1.587677001953125e-05,
      "model_forward_time": 5.6124184131622314,
      "step": 16648
    },
    {
      "epoch": 1.587677001953125e-05,
      "step": 16648,
      "training_step_time": 17.039663553237915
    },
    {
      "epoch": 1.5877723693847657e-05,
      "model_forward_time": 5.587263822555542,
      "step": 16649
    },
    {
      "epoch": 1.5877723693847657e-05,
      "step": 16649,
      "training_step_time": 17.40510392189026
    },
    {
      "epoch": 1.587867736816406e-05,
      "grad_norm": 0.6438816785812378,
      "learning_rate": 1e-05,
      "loss": 0.061,
      "step": 16650
    },
    {
      "epoch": 1.587867736816406e-05,
      "model_forward_time": 5.596511363983154,
      "step": 16650
    },
    {
      "dynamics_loss_avg": 0.05245963167399168,
      "epoch": 1.587867736816406e-05,
      "step": 16650
    },
    {
      "action_loss_avg": 0.0064169169287197295,
      "epoch": 1.587867736816406e-05,
      "step": 16650
    },
    {
      "epoch": 1.587867736816406e-05,
      "step": 16650,
      "training_step_time": 17.018917322158813
    },
    {
      "epoch": 1.587963104248047e-05,
      "model_forward_time": 5.588754177093506,
      "step": 16651
    },
    {
      "epoch": 1.587963104248047e-05,
      "step": 16651,
      "training_step_time": 17.01985239982605
    },
    {
      "epoch": 1.5880584716796876e-05,
      "model_forward_time": 5.600613832473755,
      "step": 16652
    },
    {
      "epoch": 1.5880584716796876e-05,
      "step": 16652,
      "training_step_time": 17.046189546585083
    },
    {
      "epoch": 1.588153839111328e-05,
      "model_forward_time": 5.603819847106934,
      "step": 16653
    },
    {
      "epoch": 1.588153839111328e-05,
      "step": 16653,
      "training_step_time": 17.03545904159546
    },
    {
      "epoch": 1.5882492065429687e-05,
      "model_forward_time": 5.607747316360474,
      "step": 16654
    },
    {
      "epoch": 1.5882492065429687e-05,
      "step": 16654,
      "training_step_time": 17.092786073684692
    },
    {
      "epoch": 1.5883445739746095e-05,
      "model_forward_time": 6.018229961395264,
      "step": 16655
    },
    {
      "epoch": 1.5883445739746095e-05,
      "step": 16655,
      "training_step_time": 17.504730939865112
    },
    {
      "epoch": 1.58843994140625e-05,
      "model_forward_time": 5.6260645389556885,
      "step": 16656
    },
    {
      "epoch": 1.58843994140625e-05,
      "step": 16656,
      "training_step_time": 17.14319157600403
    },
    {
      "epoch": 1.5885353088378906e-05,
      "model_forward_time": 5.627845048904419,
      "step": 16657
    },
    {
      "epoch": 1.5885353088378906e-05,
      "step": 16657,
      "training_step_time": 17.15593981742859
    },
    {
      "epoch": 1.5886306762695314e-05,
      "model_forward_time": 5.643901348114014,
      "step": 16658
    },
    {
      "epoch": 1.5886306762695314e-05,
      "step": 16658,
      "training_step_time": 17.159762144088745
    },
    {
      "epoch": 1.5887260437011718e-05,
      "model_forward_time": 5.623119354248047,
      "step": 16659
    },
    {
      "epoch": 1.5887260437011718e-05,
      "step": 16659,
      "training_step_time": 17.15068292617798
    },
    {
      "epoch": 1.5888214111328125e-05,
      "grad_norm": 0.4604300260543823,
      "learning_rate": 1e-05,
      "loss": 0.067,
      "step": 16660
    },
    {
      "epoch": 1.5888214111328125e-05,
      "model_forward_time": 5.632852077484131,
      "step": 16660
    },
    {
      "dynamics_loss_avg": 0.06089330725371837,
      "epoch": 1.5888214111328125e-05,
      "step": 16660
    },
    {
      "action_loss_avg": 0.005901751574128866,
      "epoch": 1.5888214111328125e-05,
      "step": 16660
    },
    {
      "epoch": 1.5888214111328125e-05,
      "step": 16660,
      "training_step_time": 17.55499243736267
    },
    {
      "epoch": 1.5889167785644533e-05,
      "model_forward_time": 5.617176055908203,
      "step": 16661
    },
    {
      "epoch": 1.5889167785644533e-05,
      "step": 16661,
      "training_step_time": 17.106475114822388
    },
    {
      "epoch": 1.5890121459960937e-05,
      "model_forward_time": 5.610485792160034,
      "step": 16662
    },
    {
      "epoch": 1.5890121459960937e-05,
      "step": 16662,
      "training_step_time": 17.081727504730225
    },
    {
      "epoch": 1.5891075134277344e-05,
      "model_forward_time": 5.602137804031372,
      "step": 16663
    },
    {
      "epoch": 1.5891075134277344e-05,
      "step": 16663,
      "training_step_time": 17.064029693603516
    },
    {
      "epoch": 1.589202880859375e-05,
      "model_forward_time": 5.6082603931427,
      "step": 16664
    },
    {
      "epoch": 1.589202880859375e-05,
      "step": 16664,
      "training_step_time": 17.10612177848816
    },
    {
      "epoch": 1.5892982482910155e-05,
      "model_forward_time": 5.621121168136597,
      "step": 16665
    },
    {
      "epoch": 1.5892982482910155e-05,
      "step": 16665,
      "training_step_time": 17.12287139892578
    },
    {
      "epoch": 1.5893936157226563e-05,
      "model_forward_time": 6.044703245162964,
      "step": 16666
    },
    {
      "epoch": 1.5893936157226563e-05,
      "step": 16666,
      "training_step_time": 17.55311346054077
    },
    {
      "epoch": 1.589488983154297e-05,
      "model_forward_time": 5.625016689300537,
      "step": 16667
    },
    {
      "epoch": 1.589488983154297e-05,
      "step": 16667,
      "training_step_time": 17.14779043197632
    },
    {
      "epoch": 1.5895843505859374e-05,
      "model_forward_time": 5.6255505084991455,
      "step": 16668
    },
    {
      "epoch": 1.5895843505859374e-05,
      "step": 16668,
      "training_step_time": 17.122055292129517
    },
    {
      "epoch": 1.5896797180175782e-05,
      "model_forward_time": 5.607207536697388,
      "step": 16669
    },
    {
      "epoch": 1.5896797180175782e-05,
      "step": 16669,
      "training_step_time": 17.036690711975098
    },
    {
      "epoch": 1.589775085449219e-05,
      "grad_norm": 1.1100794076919556,
      "learning_rate": 1e-05,
      "loss": 0.0636,
      "step": 16670
    },
    {
      "epoch": 1.589775085449219e-05,
      "model_forward_time": 5.609282970428467,
      "step": 16670
    },
    {
      "dynamics_loss_avg": 0.05763517022132873,
      "epoch": 1.589775085449219e-05,
      "step": 16670
    },
    {
      "action_loss_avg": 0.007017078995704651,
      "epoch": 1.589775085449219e-05,
      "step": 16670
    },
    {
      "epoch": 1.589775085449219e-05,
      "step": 16670,
      "training_step_time": 17.056156158447266
    },
    {
      "epoch": 1.5898704528808593e-05,
      "model_forward_time": 5.982468128204346,
      "step": 16671
    },
    {
      "epoch": 1.5898704528808593e-05,
      "step": 16671,
      "training_step_time": 17.40151810646057
    },
    {
      "epoch": 1.5899658203125e-05,
      "model_forward_time": 5.592374086380005,
      "step": 16672
    },
    {
      "epoch": 1.5899658203125e-05,
      "step": 16672,
      "training_step_time": 17.03412961959839
    },
    {
      "epoch": 1.5900611877441405e-05,
      "model_forward_time": 5.587717533111572,
      "step": 16673
    },
    {
      "epoch": 1.5900611877441405e-05,
      "step": 16673,
      "training_step_time": 17.036788940429688
    },
    {
      "epoch": 1.5901565551757812e-05,
      "model_forward_time": 5.591156005859375,
      "step": 16674
    },
    {
      "epoch": 1.5901565551757812e-05,
      "step": 16674,
      "training_step_time": 17.035045385360718
    },
    {
      "epoch": 1.590251922607422e-05,
      "model_forward_time": 5.586169958114624,
      "step": 16675
    },
    {
      "epoch": 1.590251922607422e-05,
      "step": 16675,
      "training_step_time": 17.01612162590027
    },
    {
      "epoch": 1.5903472900390623e-05,
      "model_forward_time": 5.5951995849609375,
      "step": 16676
    },
    {
      "epoch": 1.5903472900390623e-05,
      "step": 16676,
      "training_step_time": 17.046229362487793
    },
    {
      "epoch": 1.590442657470703e-05,
      "model_forward_time": 6.004646062850952,
      "step": 16677
    },
    {
      "epoch": 1.590442657470703e-05,
      "step": 16677,
      "training_step_time": 17.438568115234375
    },
    {
      "epoch": 1.5905380249023438e-05,
      "model_forward_time": 5.609130620956421,
      "step": 16678
    },
    {
      "epoch": 1.5905380249023438e-05,
      "step": 16678,
      "training_step_time": 17.092536449432373
    },
    {
      "epoch": 1.5906333923339842e-05,
      "model_forward_time": 5.624348402023315,
      "step": 16679
    },
    {
      "epoch": 1.5906333923339842e-05,
      "step": 16679,
      "training_step_time": 17.104405879974365
    },
    {
      "epoch": 1.590728759765625e-05,
      "grad_norm": 1.171701192855835,
      "learning_rate": 1e-05,
      "loss": 0.0645,
      "step": 16680
    },
    {
      "epoch": 1.590728759765625e-05,
      "model_forward_time": 5.6145875453948975,
      "step": 16680
    },
    {
      "dynamics_loss_avg": 0.05509230792522431,
      "epoch": 1.590728759765625e-05,
      "step": 16680
    },
    {
      "action_loss_avg": 0.007465772400610149,
      "epoch": 1.590728759765625e-05,
      "step": 16680
    },
    {
      "epoch": 1.590728759765625e-05,
      "step": 16680,
      "training_step_time": 17.107911109924316
    },
    {
      "epoch": 1.5908241271972657e-05,
      "model_forward_time": 5.631456613540649,
      "step": 16681
    },
    {
      "epoch": 1.5908241271972657e-05,
      "step": 16681,
      "training_step_time": 17.135613203048706
    },
    {
      "epoch": 1.590919494628906e-05,
      "model_forward_time": 5.638603210449219,
      "step": 16682
    },
    {
      "epoch": 1.590919494628906e-05,
      "step": 16682,
      "training_step_time": 17.555920124053955
    },
    {
      "epoch": 1.591014862060547e-05,
      "model_forward_time": 5.613795757293701,
      "step": 16683
    },
    {
      "epoch": 1.591014862060547e-05,
      "step": 16683,
      "training_step_time": 17.0649094581604
    },
    {
      "epoch": 1.5911102294921876e-05,
      "model_forward_time": 5.590030193328857,
      "step": 16684
    },
    {
      "epoch": 1.5911102294921876e-05,
      "step": 16684,
      "training_step_time": 17.01237201690674
    },
    {
      "epoch": 1.591205596923828e-05,
      "model_forward_time": 5.580325365066528,
      "step": 16685
    },
    {
      "epoch": 1.591205596923828e-05,
      "step": 16685,
      "training_step_time": 17.024900913238525
    },
    {
      "epoch": 1.5913009643554687e-05,
      "model_forward_time": 5.579338550567627,
      "step": 16686
    },
    {
      "epoch": 1.5913009643554687e-05,
      "step": 16686,
      "training_step_time": 17.019479990005493
    },
    {
      "epoch": 1.5913963317871095e-05,
      "model_forward_time": 5.591909885406494,
      "step": 16687
    },
    {
      "epoch": 1.5913963317871095e-05,
      "step": 16687,
      "training_step_time": 17.025957345962524
    },
    {
      "epoch": 1.59149169921875e-05,
      "model_forward_time": 6.014674663543701,
      "step": 16688
    },
    {
      "epoch": 1.59149169921875e-05,
      "step": 16688,
      "training_step_time": 17.479483366012573
    },
    {
      "epoch": 1.5915870666503906e-05,
      "model_forward_time": 5.6242170333862305,
      "step": 16689
    },
    {
      "epoch": 1.5915870666503906e-05,
      "step": 16689,
      "training_step_time": 17.118101358413696
    },
    {
      "epoch": 1.5916824340820314e-05,
      "grad_norm": 0.37591007351875305,
      "learning_rate": 1e-05,
      "loss": 0.0555,
      "step": 16690
    },
    {
      "epoch": 1.5916824340820314e-05,
      "model_forward_time": 5.610899209976196,
      "step": 16690
    },
    {
      "dynamics_loss_avg": 0.05607317369431257,
      "epoch": 1.5916824340820314e-05,
      "step": 16690
    },
    {
      "action_loss_avg": 0.006700972153339535,
      "epoch": 1.5916824340820314e-05,
      "step": 16690
    },
    {
      "epoch": 1.5916824340820314e-05,
      "step": 16690,
      "training_step_time": 17.11674666404724
    },
    {
      "epoch": 1.5917778015136718e-05,
      "model_forward_time": 5.623587608337402,
      "step": 16691
    },
    {
      "epoch": 1.5917778015136718e-05,
      "step": 16691,
      "training_step_time": 17.1381196975708
    },
    {
      "epoch": 1.5918731689453125e-05,
      "model_forward_time": 5.635828018188477,
      "step": 16692
    },
    {
      "epoch": 1.5918731689453125e-05,
      "step": 16692,
      "training_step_time": 17.157054901123047
    },
    {
      "epoch": 1.5919685363769533e-05,
      "model_forward_time": 5.624852180480957,
      "step": 16693
    },
    {
      "epoch": 1.5919685363769533e-05,
      "step": 16693,
      "training_step_time": 17.53008508682251
    },
    {
      "epoch": 1.5920639038085937e-05,
      "model_forward_time": 5.630331039428711,
      "step": 16694
    },
    {
      "epoch": 1.5920639038085937e-05,
      "step": 16694,
      "training_step_time": 17.15982747077942
    },
    {
      "epoch": 1.5921592712402344e-05,
      "model_forward_time": 5.629826068878174,
      "step": 16695
    },
    {
      "epoch": 1.5921592712402344e-05,
      "step": 16695,
      "training_step_time": 17.142170429229736
    },
    {
      "epoch": 1.592254638671875e-05,
      "model_forward_time": 5.61869740486145,
      "step": 16696
    },
    {
      "epoch": 1.592254638671875e-05,
      "step": 16696,
      "training_step_time": 17.09543800354004
    },
    {
      "epoch": 1.5923500061035155e-05,
      "model_forward_time": 5.604919672012329,
      "step": 16697
    },
    {
      "epoch": 1.5923500061035155e-05,
      "step": 16697,
      "training_step_time": 17.05042314529419
    },
    {
      "epoch": 1.5924453735351563e-05,
      "model_forward_time": 5.5827696323394775,
      "step": 16698
    },
    {
      "epoch": 1.5924453735351563e-05,
      "step": 16698,
      "training_step_time": 17.01336932182312
    },
    {
      "epoch": 1.592540740966797e-05,
      "model_forward_time": 6.0337982177734375,
      "step": 16699
    },
    {
      "epoch": 1.592540740966797e-05,
      "step": 16699,
      "training_step_time": 17.43983793258667
    },
    {
      "epoch": 1.5926361083984374e-05,
      "grad_norm": 0.5073205828666687,
      "learning_rate": 1e-05,
      "loss": 0.0646,
      "step": 16700
    },
    {
      "epoch": 1.5926361083984374e-05,
      "model_forward_time": 5.5986878871917725,
      "step": 16700
    },
    {
      "dynamics_loss_avg": 0.053705517761409284,
      "epoch": 1.5926361083984374e-05,
      "step": 16700
    },
    {
      "action_loss_avg": 0.00593709961976856,
      "epoch": 1.5926361083984374e-05,
      "step": 16700
    },
    {
      "epoch": 1.5926361083984374e-05,
      "step": 16700,
      "training_step_time": 17.048024654388428
    },
    {
      "epoch": 1.5927314758300782e-05,
      "model_forward_time": 5.626164197921753,
      "step": 16701
    },
    {
      "epoch": 1.5927314758300782e-05,
      "step": 16701,
      "training_step_time": 17.096094846725464
    },
    {
      "epoch": 1.592826843261719e-05,
      "model_forward_time": 5.605711460113525,
      "step": 16702
    },
    {
      "epoch": 1.592826843261719e-05,
      "step": 16702,
      "training_step_time": 17.102999687194824
    },
    {
      "epoch": 1.5929222106933593e-05,
      "model_forward_time": 5.617496490478516,
      "step": 16703
    },
    {
      "epoch": 1.5929222106933593e-05,
      "step": 16703,
      "training_step_time": 17.102444171905518
    },
    {
      "epoch": 1.593017578125e-05,
      "model_forward_time": 6.025500297546387,
      "step": 16704
    },
    {
      "epoch": 1.593017578125e-05,
      "step": 16704,
      "training_step_time": 17.468010663986206
    },
    {
      "epoch": 1.5931129455566405e-05,
      "model_forward_time": 4.27280592918396,
      "step": 16705
    },
    {
      "epoch": 1.5931129455566405e-05,
      "step": 16705,
      "training_step_time": 12.80322790145874
    },
    {
      "epoch": 1.5932083129882812e-05,
      "model_forward_time": 5.600059747695923,
      "step": 16706
    },
    {
      "epoch": 1.5932083129882812e-05,
      "step": 16706,
      "training_step_time": 17.04758310317993
    },
    {
      "epoch": 1.593303680419922e-05,
      "model_forward_time": 5.612250328063965,
      "step": 16707
    },
    {
      "epoch": 1.593303680419922e-05,
      "step": 16707,
      "training_step_time": 17.099838256835938
    },
    {
      "epoch": 1.5933990478515623e-05,
      "model_forward_time": 5.616485118865967,
      "step": 16708
    },
    {
      "epoch": 1.5933990478515623e-05,
      "step": 16708,
      "training_step_time": 17.11287546157837
    },
    {
      "epoch": 1.593494415283203e-05,
      "model_forward_time": 5.623380184173584,
      "step": 16709
    },
    {
      "epoch": 1.593494415283203e-05,
      "step": 16709,
      "training_step_time": 17.104841709136963
    },
    {
      "epoch": 1.593589782714844e-05,
      "grad_norm": 1.0905989408493042,
      "learning_rate": 1e-05,
      "loss": 0.053,
      "step": 16710
    },
    {
      "epoch": 1.593589782714844e-05,
      "model_forward_time": 6.032201290130615,
      "step": 16710
    },
    {
      "dynamics_loss_avg": 0.04466673508286476,
      "epoch": 1.593589782714844e-05,
      "step": 16710
    },
    {
      "action_loss_avg": 0.005826652119867504,
      "epoch": 1.593589782714844e-05,
      "step": 16710
    },
    {
      "epoch": 1.593589782714844e-05,
      "step": 16710,
      "training_step_time": 17.4822359085083
    },
    {
      "epoch": 1.5936851501464842e-05,
      "model_forward_time": 5.615950345993042,
      "step": 16711
    },
    {
      "epoch": 1.5936851501464842e-05,
      "step": 16711,
      "training_step_time": 17.07628297805786
    },
    {
      "epoch": 1.593780517578125e-05,
      "model_forward_time": 5.607919454574585,
      "step": 16712
    },
    {
      "epoch": 1.593780517578125e-05,
      "step": 16712,
      "training_step_time": 17.059823036193848
    },
    {
      "epoch": 1.5938758850097657e-05,
      "model_forward_time": 5.604112863540649,
      "step": 16713
    },
    {
      "epoch": 1.5938758850097657e-05,
      "step": 16713,
      "training_step_time": 17.074123859405518
    },
    {
      "epoch": 1.593971252441406e-05,
      "model_forward_time": 5.645040273666382,
      "step": 16714
    },
    {
      "epoch": 1.593971252441406e-05,
      "step": 16714,
      "training_step_time": 17.15505623817444
    },
    {
      "epoch": 1.594066619873047e-05,
      "model_forward_time": 5.640059232711792,
      "step": 16715
    },
    {
      "epoch": 1.594066619873047e-05,
      "step": 16715,
      "training_step_time": 17.14117407798767
    },
    {
      "epoch": 1.5941619873046876e-05,
      "model_forward_time": 6.049293041229248,
      "step": 16716
    },
    {
      "epoch": 1.5941619873046876e-05,
      "step": 16716,
      "training_step_time": 17.51265549659729
    },
    {
      "epoch": 1.594257354736328e-05,
      "model_forward_time": 5.615044593811035,
      "step": 16717
    },
    {
      "epoch": 1.594257354736328e-05,
      "step": 16717,
      "training_step_time": 17.12021231651306
    },
    {
      "epoch": 1.5943527221679687e-05,
      "model_forward_time": 5.639153003692627,
      "step": 16718
    },
    {
      "epoch": 1.5943527221679687e-05,
      "step": 16718,
      "training_step_time": 17.135399341583252
    },
    {
      "epoch": 1.5944480895996095e-05,
      "model_forward_time": 5.608725070953369,
      "step": 16719
    },
    {
      "epoch": 1.5944480895996095e-05,
      "step": 16719,
      "training_step_time": 17.055036783218384
    },
    {
      "epoch": 1.59454345703125e-05,
      "grad_norm": 1.3221566677093506,
      "learning_rate": 1e-05,
      "loss": 0.0662,
      "step": 16720
    },
    {
      "epoch": 1.59454345703125e-05,
      "model_forward_time": 5.580429553985596,
      "step": 16720
    },
    {
      "dynamics_loss_avg": 0.06057907193899155,
      "epoch": 1.59454345703125e-05,
      "step": 16720
    },
    {
      "action_loss_avg": 0.006145516596734524,
      "epoch": 1.59454345703125e-05,
      "step": 16720
    },
    {
      "epoch": 1.59454345703125e-05,
      "step": 16720,
      "training_step_time": 17.025817155838013
    },
    {
      "epoch": 1.5946388244628906e-05,
      "model_forward_time": 6.003875017166138,
      "step": 16721
    },
    {
      "epoch": 1.5946388244628906e-05,
      "step": 16721,
      "training_step_time": 17.420548677444458
    },
    {
      "epoch": 1.5947341918945314e-05,
      "model_forward_time": 5.586015939712524,
      "step": 16722
    },
    {
      "epoch": 1.5947341918945314e-05,
      "step": 16722,
      "training_step_time": 17.002299547195435
    },
    {
      "epoch": 1.5948295593261718e-05,
      "model_forward_time": 5.577163457870483,
      "step": 16723
    },
    {
      "epoch": 1.5948295593261718e-05,
      "step": 16723,
      "training_step_time": 17.042022705078125
    },
    {
      "epoch": 1.5949249267578125e-05,
      "model_forward_time": 5.599465608596802,
      "step": 16724
    },
    {
      "epoch": 1.5949249267578125e-05,
      "step": 16724,
      "training_step_time": 17.073968172073364
    },
    {
      "epoch": 1.5950202941894533e-05,
      "model_forward_time": 5.617124795913696,
      "step": 16725
    },
    {
      "epoch": 1.5950202941894533e-05,
      "step": 16725,
      "training_step_time": 17.1053204536438
    },
    {
      "epoch": 1.5951156616210937e-05,
      "model_forward_time": 5.610957145690918,
      "step": 16726
    },
    {
      "epoch": 1.5951156616210937e-05,
      "step": 16726,
      "training_step_time": 17.489333868026733
    },
    {
      "epoch": 1.5952110290527344e-05,
      "model_forward_time": 5.614563703536987,
      "step": 16727
    },
    {
      "epoch": 1.5952110290527344e-05,
      "step": 16727,
      "training_step_time": 17.107943773269653
    },
    {
      "epoch": 1.595306396484375e-05,
      "model_forward_time": 5.634502410888672,
      "step": 16728
    },
    {
      "epoch": 1.595306396484375e-05,
      "step": 16728,
      "training_step_time": 17.123998165130615
    },
    {
      "epoch": 1.5954017639160155e-05,
      "model_forward_time": 5.608014106750488,
      "step": 16729
    },
    {
      "epoch": 1.5954017639160155e-05,
      "step": 16729,
      "training_step_time": 17.10670304298401
    },
    {
      "epoch": 1.5954971313476563e-05,
      "grad_norm": 1.0292062759399414,
      "learning_rate": 1e-05,
      "loss": 0.0625,
      "step": 16730
    },
    {
      "epoch": 1.5954971313476563e-05,
      "model_forward_time": 5.608551263809204,
      "step": 16730
    },
    {
      "dynamics_loss_avg": 0.056102003902196884,
      "epoch": 1.5954971313476563e-05,
      "step": 16730
    },
    {
      "action_loss_avg": 0.008827257354278117,
      "epoch": 1.5954971313476563e-05,
      "step": 16730
    },
    {
      "epoch": 1.5954971313476563e-05,
      "step": 16730,
      "training_step_time": 17.096325397491455
    },
    {
      "epoch": 1.595592498779297e-05,
      "model_forward_time": 5.609509706497192,
      "step": 16731
    },
    {
      "epoch": 1.595592498779297e-05,
      "step": 16731,
      "training_step_time": 17.09172224998474
    },
    {
      "epoch": 1.5956878662109374e-05,
      "model_forward_time": 5.975576400756836,
      "step": 16732
    },
    {
      "epoch": 1.5956878662109374e-05,
      "step": 16732,
      "training_step_time": 17.462334871292114
    },
    {
      "epoch": 1.5957832336425782e-05,
      "model_forward_time": 5.603941917419434,
      "step": 16733
    },
    {
      "epoch": 1.5957832336425782e-05,
      "step": 16733,
      "training_step_time": 17.067927360534668
    },
    {
      "epoch": 1.595878601074219e-05,
      "model_forward_time": 5.604293346405029,
      "step": 16734
    },
    {
      "epoch": 1.595878601074219e-05,
      "step": 16734,
      "training_step_time": 17.058012008666992
    },
    {
      "epoch": 1.5959739685058593e-05,
      "model_forward_time": 5.586668491363525,
      "step": 16735
    },
    {
      "epoch": 1.5959739685058593e-05,
      "step": 16735,
      "training_step_time": 17.04005527496338
    },
    {
      "epoch": 1.5960693359375e-05,
      "model_forward_time": 5.581288576126099,
      "step": 16736
    },
    {
      "epoch": 1.5960693359375e-05,
      "step": 16736,
      "training_step_time": 17.02515411376953
    },
    {
      "epoch": 1.5961647033691405e-05,
      "model_forward_time": 5.953550577163696,
      "step": 16737
    },
    {
      "epoch": 1.5961647033691405e-05,
      "step": 16737,
      "training_step_time": 17.42084240913391
    },
    {
      "epoch": 1.5962600708007812e-05,
      "model_forward_time": 5.609786033630371,
      "step": 16738
    },
    {
      "epoch": 1.5962600708007812e-05,
      "step": 16738,
      "training_step_time": 17.11627769470215
    },
    {
      "epoch": 1.596355438232422e-05,
      "model_forward_time": 5.621267080307007,
      "step": 16739
    },
    {
      "epoch": 1.596355438232422e-05,
      "step": 16739,
      "training_step_time": 17.136876344680786
    },
    {
      "epoch": 1.5964508056640624e-05,
      "grad_norm": 0.36650073528289795,
      "learning_rate": 1e-05,
      "loss": 0.0578,
      "step": 16740
    },
    {
      "epoch": 1.5964508056640624e-05,
      "model_forward_time": 5.625500440597534,
      "step": 16740
    },
    {
      "dynamics_loss_avg": 0.05068935323506594,
      "epoch": 1.5964508056640624e-05,
      "step": 16740
    },
    {
      "action_loss_avg": 0.007412097114138305,
      "epoch": 1.5964508056640624e-05,
      "step": 16740
    },
    {
      "epoch": 1.5964508056640624e-05,
      "step": 16740,
      "training_step_time": 17.12801241874695
    },
    {
      "epoch": 1.596546173095703e-05,
      "model_forward_time": 5.617077112197876,
      "step": 16741
    },
    {
      "epoch": 1.596546173095703e-05,
      "step": 16741,
      "training_step_time": 17.12255859375
    },
    {
      "epoch": 1.596641540527344e-05,
      "model_forward_time": 5.6142356395721436,
      "step": 16742
    },
    {
      "epoch": 1.596641540527344e-05,
      "step": 16742,
      "training_step_time": 17.100882291793823
    },
    {
      "epoch": 1.5967369079589842e-05,
      "model_forward_time": 5.981608152389526,
      "step": 16743
    },
    {
      "epoch": 1.5967369079589842e-05,
      "step": 16743,
      "training_step_time": 17.455771923065186
    },
    {
      "epoch": 1.596832275390625e-05,
      "model_forward_time": 5.60672926902771,
      "step": 16744
    },
    {
      "epoch": 1.596832275390625e-05,
      "step": 16744,
      "training_step_time": 17.07641839981079
    },
    {
      "epoch": 1.5969276428222657e-05,
      "model_forward_time": 5.595674514770508,
      "step": 16745
    },
    {
      "epoch": 1.5969276428222657e-05,
      "step": 16745,
      "training_step_time": 17.06164860725403
    },
    {
      "epoch": 1.597023010253906e-05,
      "model_forward_time": 5.607136487960815,
      "step": 16746
    },
    {
      "epoch": 1.597023010253906e-05,
      "step": 16746,
      "training_step_time": 17.105961322784424
    },
    {
      "epoch": 1.597118377685547e-05,
      "model_forward_time": 5.624036550521851,
      "step": 16747
    },
    {
      "epoch": 1.597118377685547e-05,
      "step": 16747,
      "training_step_time": 17.153515100479126
    },
    {
      "epoch": 1.5972137451171876e-05,
      "model_forward_time": 5.625045299530029,
      "step": 16748
    },
    {
      "epoch": 1.5972137451171876e-05,
      "step": 16748,
      "training_step_time": 17.492114305496216
    },
    {
      "epoch": 1.597309112548828e-05,
      "model_forward_time": 5.59865403175354,
      "step": 16749
    },
    {
      "epoch": 1.597309112548828e-05,
      "step": 16749,
      "training_step_time": 17.030815362930298
    },
    {
      "epoch": 1.5974044799804688e-05,
      "grad_norm": 0.5688976645469666,
      "learning_rate": 1e-05,
      "loss": 0.0499,
      "step": 16750
    },
    {
      "epoch": 1.5974044799804688e-05,
      "model_forward_time": 5.577745676040649,
      "step": 16750
    },
    {
      "dynamics_loss_avg": 0.045789674110710624,
      "epoch": 1.5974044799804688e-05,
      "step": 16750
    },
    {
      "action_loss_avg": 0.006121513806283474,
      "epoch": 1.5974044799804688e-05,
      "step": 16750
    },
    {
      "epoch": 1.5974044799804688e-05,
      "step": 16750,
      "training_step_time": 17.010653972625732
    },
    {
      "epoch": 1.5974998474121095e-05,
      "model_forward_time": 5.575616836547852,
      "step": 16751
    },
    {
      "epoch": 1.5974998474121095e-05,
      "step": 16751,
      "training_step_time": 16.991406202316284
    },
    {
      "epoch": 1.59759521484375e-05,
      "model_forward_time": 5.566985845565796,
      "step": 16752
    },
    {
      "epoch": 1.59759521484375e-05,
      "step": 16752,
      "training_step_time": 16.976298570632935
    },
    {
      "epoch": 1.5976905822753906e-05,
      "model_forward_time": 5.590147256851196,
      "step": 16753
    },
    {
      "epoch": 1.5976905822753906e-05,
      "step": 16753,
      "training_step_time": 17.007977724075317
    },
    {
      "epoch": 1.5977859497070314e-05,
      "model_forward_time": 5.960562229156494,
      "step": 16754
    },
    {
      "epoch": 1.5977859497070314e-05,
      "step": 16754,
      "training_step_time": 17.402673482894897
    },
    {
      "epoch": 1.5978813171386718e-05,
      "model_forward_time": 5.59541654586792,
      "step": 16755
    },
    {
      "epoch": 1.5978813171386718e-05,
      "step": 16755,
      "training_step_time": 17.077202320098877
    },
    {
      "epoch": 1.5979766845703125e-05,
      "model_forward_time": 5.621312856674194,
      "step": 16756
    },
    {
      "epoch": 1.5979766845703125e-05,
      "step": 16756,
      "training_step_time": 17.13456439971924
    },
    {
      "epoch": 1.5980720520019533e-05,
      "model_forward_time": 5.616338491439819,
      "step": 16757
    },
    {
      "epoch": 1.5980720520019533e-05,
      "step": 16757,
      "training_step_time": 17.115203380584717
    },
    {
      "epoch": 1.5981674194335937e-05,
      "model_forward_time": 5.597960710525513,
      "step": 16758
    },
    {
      "epoch": 1.5981674194335937e-05,
      "step": 16758,
      "training_step_time": 17.067946195602417
    },
    {
      "epoch": 1.5982627868652344e-05,
      "model_forward_time": 5.599546670913696,
      "step": 16759
    },
    {
      "epoch": 1.5982627868652344e-05,
      "step": 16759,
      "training_step_time": 17.44838786125183
    },
    {
      "epoch": 1.598358154296875e-05,
      "grad_norm": 1.1390507221221924,
      "learning_rate": 1e-05,
      "loss": 0.062,
      "step": 16760
    },
    {
      "epoch": 1.598358154296875e-05,
      "model_forward_time": 5.5969154834747314,
      "step": 16760
    },
    {
      "dynamics_loss_avg": 0.05459336377680302,
      "epoch": 1.598358154296875e-05,
      "step": 16760
    },
    {
      "action_loss_avg": 0.005832824134267866,
      "epoch": 1.598358154296875e-05,
      "step": 16760
    },
    {
      "epoch": 1.598358154296875e-05,
      "step": 16760,
      "training_step_time": 17.068997859954834
    },
    {
      "epoch": 1.5984535217285156e-05,
      "model_forward_time": 5.602786540985107,
      "step": 16761
    },
    {
      "epoch": 1.5984535217285156e-05,
      "step": 16761,
      "training_step_time": 17.06508779525757
    },
    {
      "epoch": 1.5985488891601563e-05,
      "model_forward_time": 5.598001956939697,
      "step": 16762
    },
    {
      "epoch": 1.5985488891601563e-05,
      "step": 16762,
      "training_step_time": 17.039849281311035
    },
    {
      "epoch": 1.598644256591797e-05,
      "model_forward_time": 5.595663547515869,
      "step": 16763
    },
    {
      "epoch": 1.598644256591797e-05,
      "step": 16763,
      "training_step_time": 17.03045415878296
    },
    {
      "epoch": 1.5987396240234374e-05,
      "model_forward_time": 5.577848196029663,
      "step": 16764
    },
    {
      "epoch": 1.5987396240234374e-05,
      "step": 16764,
      "training_step_time": 16.997875213623047
    },
    {
      "epoch": 1.5988349914550782e-05,
      "model_forward_time": 5.954692125320435,
      "step": 16765
    },
    {
      "epoch": 1.5988349914550782e-05,
      "step": 16765,
      "training_step_time": 17.38915228843689
    },
    {
      "epoch": 1.5989303588867186e-05,
      "model_forward_time": 5.595818281173706,
      "step": 16766
    },
    {
      "epoch": 1.5989303588867186e-05,
      "step": 16766,
      "training_step_time": 17.05523157119751
    },
    {
      "epoch": 1.5990257263183593e-05,
      "model_forward_time": 5.5987913608551025,
      "step": 16767
    },
    {
      "epoch": 1.5990257263183593e-05,
      "step": 16767,
      "training_step_time": 17.071704626083374
    },
    {
      "epoch": 1.59912109375e-05,
      "model_forward_time": 5.60650110244751,
      "step": 16768
    },
    {
      "epoch": 1.59912109375e-05,
      "step": 16768,
      "training_step_time": 17.086002111434937
    },
    {
      "epoch": 1.5992164611816405e-05,
      "model_forward_time": 5.595007658004761,
      "step": 16769
    },
    {
      "epoch": 1.5992164611816405e-05,
      "step": 16769,
      "training_step_time": 17.047539710998535
    },
    {
      "epoch": 1.5993118286132812e-05,
      "grad_norm": 0.39049139618873596,
      "learning_rate": 1e-05,
      "loss": 0.0737,
      "step": 16770
    },
    {
      "epoch": 1.5993118286132812e-05,
      "model_forward_time": 5.9577906131744385,
      "step": 16770
    },
    {
      "dynamics_loss_avg": 0.06271252166479827,
      "epoch": 1.5993118286132812e-05,
      "step": 16770
    },
    {
      "action_loss_avg": 0.009789271280169487,
      "epoch": 1.5993118286132812e-05,
      "step": 16770
    },
    {
      "epoch": 1.5993118286132812e-05,
      "step": 16770,
      "training_step_time": 17.399065732955933
    },
    {
      "epoch": 1.599407196044922e-05,
      "model_forward_time": 5.583177328109741,
      "step": 16771
    },
    {
      "epoch": 1.599407196044922e-05,
      "step": 16771,
      "training_step_time": 17.012385845184326
    },
    {
      "epoch": 1.5995025634765624e-05,
      "model_forward_time": 5.591930389404297,
      "step": 16772
    },
    {
      "epoch": 1.5995025634765624e-05,
      "step": 16772,
      "training_step_time": 17.041118621826172
    },
    {
      "epoch": 1.599597930908203e-05,
      "model_forward_time": 5.592340707778931,
      "step": 16773
    },
    {
      "epoch": 1.599597930908203e-05,
      "step": 16773,
      "training_step_time": 17.038211345672607
    },
    {
      "epoch": 1.599693298339844e-05,
      "model_forward_time": 5.602294206619263,
      "step": 16774
    },
    {
      "epoch": 1.599693298339844e-05,
      "step": 16774,
      "training_step_time": 17.080907583236694
    },
    {
      "epoch": 1.5997886657714842e-05,
      "model_forward_time": 5.608996868133545,
      "step": 16775
    },
    {
      "epoch": 1.5997886657714842e-05,
      "step": 16775,
      "training_step_time": 17.08923888206482
    },
    {
      "epoch": 1.599884033203125e-05,
      "model_forward_time": 5.959103584289551,
      "step": 16776
    },
    {
      "epoch": 1.599884033203125e-05,
      "step": 16776,
      "training_step_time": 17.437838554382324
    },
    {
      "epoch": 1.5999794006347657e-05,
      "model_forward_time": 5.61482310295105,
      "step": 16777
    },
    {
      "epoch": 1.5999794006347657e-05,
      "step": 16777,
      "training_step_time": 17.103450536727905
    },
    {
      "epoch": 1.600074768066406e-05,
      "model_forward_time": 5.606510639190674,
      "step": 16778
    },
    {
      "epoch": 1.600074768066406e-05,
      "step": 16778,
      "training_step_time": 17.09848403930664
    },
    {
      "epoch": 1.600170135498047e-05,
      "model_forward_time": 5.59855580329895,
      "step": 16779
    },
    {
      "epoch": 1.600170135498047e-05,
      "step": 16779,
      "training_step_time": 17.073206663131714
    },
    {
      "epoch": 1.6002655029296876e-05,
      "grad_norm": 0.5601235628128052,
      "learning_rate": 1e-05,
      "loss": 0.0539,
      "step": 16780
    },
    {
      "epoch": 1.6002655029296876e-05,
      "model_forward_time": 5.598092079162598,
      "step": 16780
    },
    {
      "dynamics_loss_avg": 0.05103644505143166,
      "epoch": 1.6002655029296876e-05,
      "step": 16780
    },
    {
      "action_loss_avg": 0.005869690258987248,
      "epoch": 1.6002655029296876e-05,
      "step": 16780
    },
    {
      "epoch": 1.6002655029296876e-05,
      "step": 16780,
      "training_step_time": 17.02855944633484
    },
    {
      "epoch": 1.600360870361328e-05,
      "model_forward_time": 5.587128400802612,
      "step": 16781
    },
    {
      "epoch": 1.600360870361328e-05,
      "step": 16781,
      "training_step_time": 17.397264003753662
    },
    {
      "epoch": 1.6004562377929688e-05,
      "model_forward_time": 5.598980665206909,
      "step": 16782
    },
    {
      "epoch": 1.6004562377929688e-05,
      "step": 16782,
      "training_step_time": 17.1070499420166
    },
    {
      "epoch": 1.6005516052246095e-05,
      "model_forward_time": 5.621838331222534,
      "step": 16783
    },
    {
      "epoch": 1.6005516052246095e-05,
      "step": 16783,
      "training_step_time": 17.146220207214355
    },
    {
      "epoch": 1.60064697265625e-05,
      "model_forward_time": 5.623152732849121,
      "step": 16784
    },
    {
      "epoch": 1.60064697265625e-05,
      "step": 16784,
      "training_step_time": 17.112415075302124
    },
    {
      "epoch": 1.6007423400878906e-05,
      "model_forward_time": 5.596675395965576,
      "step": 16785
    },
    {
      "epoch": 1.6007423400878906e-05,
      "step": 16785,
      "training_step_time": 17.045785427093506
    },
    {
      "epoch": 1.6008377075195314e-05,
      "model_forward_time": 5.5895304679870605,
      "step": 16786
    },
    {
      "epoch": 1.6008377075195314e-05,
      "step": 16786,
      "training_step_time": 17.001259565353394
    },
    {
      "epoch": 1.6009330749511718e-05,
      "model_forward_time": 5.966026782989502,
      "step": 16787
    },
    {
      "epoch": 1.6009330749511718e-05,
      "step": 16787,
      "training_step_time": 17.38813591003418
    },
    {
      "epoch": 1.6010284423828125e-05,
      "model_forward_time": 5.5890138149261475,
      "step": 16788
    },
    {
      "epoch": 1.6010284423828125e-05,
      "step": 16788,
      "training_step_time": 17.038947582244873
    },
    {
      "epoch": 1.6011238098144533e-05,
      "model_forward_time": 5.587433099746704,
      "step": 16789
    },
    {
      "epoch": 1.6011238098144533e-05,
      "step": 16789,
      "training_step_time": 17.050106048583984
    },
    {
      "epoch": 1.6012191772460937e-05,
      "grad_norm": 0.6926265954971313,
      "learning_rate": 1e-05,
      "loss": 0.0675,
      "step": 16790
    },
    {
      "epoch": 1.6012191772460937e-05,
      "model_forward_time": 5.604650974273682,
      "step": 16790
    },
    {
      "dynamics_loss_avg": 0.05384945496916771,
      "epoch": 1.6012191772460937e-05,
      "step": 16790
    },
    {
      "action_loss_avg": 0.009095814730972051,
      "epoch": 1.6012191772460937e-05,
      "step": 16790
    },
    {
      "epoch": 1.6012191772460937e-05,
      "step": 16790,
      "training_step_time": 17.118143320083618
    },
    {
      "epoch": 1.6013145446777344e-05,
      "model_forward_time": 5.623287200927734,
      "step": 16791
    },
    {
      "epoch": 1.6013145446777344e-05,
      "step": 16791,
      "training_step_time": 17.148584842681885
    },
    {
      "epoch": 1.601409912109375e-05,
      "model_forward_time": 5.626379728317261,
      "step": 16792
    },
    {
      "epoch": 1.601409912109375e-05,
      "step": 16792,
      "training_step_time": 17.518158197402954
    },
    {
      "epoch": 1.6015052795410156e-05,
      "model_forward_time": 5.621742248535156,
      "step": 16793
    },
    {
      "epoch": 1.6015052795410156e-05,
      "step": 16793,
      "training_step_time": 17.116891384124756
    },
    {
      "epoch": 1.6016006469726563e-05,
      "model_forward_time": 5.600407600402832,
      "step": 16794
    },
    {
      "epoch": 1.6016006469726563e-05,
      "step": 16794,
      "training_step_time": 17.09376358985901
    },
    {
      "epoch": 1.601696014404297e-05,
      "model_forward_time": 5.611039638519287,
      "step": 16795
    },
    {
      "epoch": 1.601696014404297e-05,
      "step": 16795,
      "training_step_time": 17.087361335754395
    },
    {
      "epoch": 1.6017913818359374e-05,
      "model_forward_time": 5.595822095870972,
      "step": 16796
    },
    {
      "epoch": 1.6017913818359374e-05,
      "step": 16796,
      "training_step_time": 17.052881717681885
    },
    {
      "epoch": 1.6018867492675782e-05,
      "model_forward_time": 5.589320659637451,
      "step": 16797
    },
    {
      "epoch": 1.6018867492675782e-05,
      "step": 16797,
      "training_step_time": 17.03439474105835
    },
    {
      "epoch": 1.6019821166992186e-05,
      "model_forward_time": 5.963966131210327,
      "step": 16798
    },
    {
      "epoch": 1.6019821166992186e-05,
      "step": 16798,
      "training_step_time": 17.402915716171265
    },
    {
      "epoch": 1.6020774841308593e-05,
      "model_forward_time": 5.608614683151245,
      "step": 16799
    },
    {
      "epoch": 1.6020774841308593e-05,
      "step": 16799,
      "training_step_time": 17.117895364761353
    },
    {
      "epoch": 1.6021728515625e-05,
      "grad_norm": 0.432791531085968,
      "learning_rate": 1e-05,
      "loss": 0.0522,
      "step": 16800
    },
    {
      "epoch": 1.6021728515625e-05,
      "model_forward_time": 5.636232614517212,
      "step": 16800
    },
    {
      "dynamics_loss_avg": 0.045798512548208235,
      "epoch": 1.6021728515625e-05,
      "step": 16800
    },
    {
      "action_loss_avg": 0.005527409282512963,
      "epoch": 1.6021728515625e-05,
      "step": 16800
    },
    {
      "epoch": 1.6021728515625e-05,
      "step": 16800,
      "training_step_time": 17.164310216903687
    },
    {
      "epoch": 1.6022682189941405e-05,
      "model_forward_time": 5.6211347579956055,
      "step": 16801
    },
    {
      "epoch": 1.6022682189941405e-05,
      "step": 16801,
      "training_step_time": 17.11954140663147
    },
    {
      "epoch": 1.6023635864257812e-05,
      "model_forward_time": 5.581181287765503,
      "step": 16802
    },
    {
      "epoch": 1.6023635864257812e-05,
      "step": 16802,
      "training_step_time": 16.994929313659668
    },
    {
      "epoch": 1.602458953857422e-05,
      "model_forward_time": 5.943054914474487,
      "step": 16803
    },
    {
      "epoch": 1.602458953857422e-05,
      "step": 16803,
      "training_step_time": 17.380951642990112
    },
    {
      "epoch": 1.6025543212890624e-05,
      "model_forward_time": 5.566680669784546,
      "step": 16804
    },
    {
      "epoch": 1.6025543212890624e-05,
      "step": 16804,
      "training_step_time": 16.992482662200928
    },
    {
      "epoch": 1.602649688720703e-05,
      "model_forward_time": 5.5711829662323,
      "step": 16805
    },
    {
      "epoch": 1.602649688720703e-05,
      "step": 16805,
      "training_step_time": 16.985312700271606
    },
    {
      "epoch": 1.602745056152344e-05,
      "model_forward_time": 5.571108341217041,
      "step": 16806
    },
    {
      "epoch": 1.602745056152344e-05,
      "step": 16806,
      "training_step_time": 16.980968713760376
    },
    {
      "epoch": 1.6028404235839842e-05,
      "model_forward_time": 5.568155765533447,
      "step": 16807
    },
    {
      "epoch": 1.6028404235839842e-05,
      "step": 16807,
      "training_step_time": 17.015008449554443
    },
    {
      "epoch": 1.602935791015625e-05,
      "model_forward_time": 5.575655698776245,
      "step": 16808
    },
    {
      "epoch": 1.602935791015625e-05,
      "step": 16808,
      "training_step_time": 16.990752696990967
    },
    {
      "epoch": 1.6030311584472657e-05,
      "model_forward_time": 5.973854303359985,
      "step": 16809
    },
    {
      "epoch": 1.6030311584472657e-05,
      "step": 16809,
      "training_step_time": 17.43823504447937
    },
    {
      "epoch": 1.603126525878906e-05,
      "grad_norm": 0.4061823785305023,
      "learning_rate": 1e-05,
      "loss": 0.0612,
      "step": 16810
    },
    {
      "epoch": 1.603126525878906e-05,
      "model_forward_time": 5.602237939834595,
      "step": 16810
    },
    {
      "dynamics_loss_avg": 0.058172987587749955,
      "epoch": 1.603126525878906e-05,
      "step": 16810
    },
    {
      "action_loss_avg": 0.007152598351240158,
      "epoch": 1.603126525878906e-05,
      "step": 16810
    },
    {
      "epoch": 1.603126525878906e-05,
      "step": 16810,
      "training_step_time": 17.064297914505005
    },
    {
      "epoch": 1.603221893310547e-05,
      "model_forward_time": 5.636922121047974,
      "step": 16811
    },
    {
      "epoch": 1.603221893310547e-05,
      "step": 16811,
      "training_step_time": 17.109678983688354
    },
    {
      "epoch": 1.6033172607421876e-05,
      "model_forward_time": 5.602340936660767,
      "step": 16812
    },
    {
      "epoch": 1.6033172607421876e-05,
      "step": 16812,
      "training_step_time": 17.09040117263794
    },
    {
      "epoch": 1.603412628173828e-05,
      "model_forward_time": 5.6038665771484375,
      "step": 16813
    },
    {
      "epoch": 1.603412628173828e-05,
      "step": 16813,
      "training_step_time": 17.100080728530884
    },
    {
      "epoch": 1.6035079956054688e-05,
      "model_forward_time": 5.608299016952515,
      "step": 16814
    },
    {
      "epoch": 1.6035079956054688e-05,
      "step": 16814,
      "training_step_time": 17.472877025604248
    },
    {
      "epoch": 1.6036033630371095e-05,
      "model_forward_time": 5.62231707572937,
      "step": 16815
    },
    {
      "epoch": 1.6036033630371095e-05,
      "step": 16815,
      "training_step_time": 17.130613565444946
    },
    {
      "epoch": 1.60369873046875e-05,
      "model_forward_time": 5.618121147155762,
      "step": 16816
    },
    {
      "epoch": 1.60369873046875e-05,
      "step": 16816,
      "training_step_time": 17.102540493011475
    },
    {
      "epoch": 1.6037940979003906e-05,
      "model_forward_time": 5.601746320724487,
      "step": 16817
    },
    {
      "epoch": 1.6037940979003906e-05,
      "step": 16817,
      "training_step_time": 17.01586627960205
    },
    {
      "epoch": 1.6038894653320314e-05,
      "model_forward_time": 5.5758888721466064,
      "step": 16818
    },
    {
      "epoch": 1.6038894653320314e-05,
      "step": 16818,
      "training_step_time": 17.01232624053955
    },
    {
      "epoch": 1.6039848327636718e-05,
      "model_forward_time": 5.569867134094238,
      "step": 16819
    },
    {
      "epoch": 1.6039848327636718e-05,
      "step": 16819,
      "training_step_time": 17.00581431388855
    },
    {
      "epoch": 1.6040802001953125e-05,
      "grad_norm": 0.15046407282352448,
      "learning_rate": 1e-05,
      "loss": 0.0555,
      "step": 16820
    },
    {
      "epoch": 1.6040802001953125e-05,
      "model_forward_time": 5.952289581298828,
      "step": 16820
    },
    {
      "dynamics_loss_avg": 0.04404920507222414,
      "epoch": 1.6040802001953125e-05,
      "step": 16820
    },
    {
      "action_loss_avg": 0.00595573412720114,
      "epoch": 1.6040802001953125e-05,
      "step": 16820
    },
    {
      "epoch": 1.6040802001953125e-05,
      "step": 16820,
      "training_step_time": 17.356672286987305
    },
    {
      "epoch": 1.6041755676269533e-05,
      "model_forward_time": 5.592545509338379,
      "step": 16821
    },
    {
      "epoch": 1.6041755676269533e-05,
      "step": 16821,
      "training_step_time": 17.05207586288452
    },
    {
      "epoch": 1.6042709350585937e-05,
      "model_forward_time": 5.5953614711761475,
      "step": 16822
    },
    {
      "epoch": 1.6042709350585937e-05,
      "step": 16822,
      "training_step_time": 17.053200006484985
    },
    {
      "epoch": 1.6043663024902344e-05,
      "model_forward_time": 5.61547589302063,
      "step": 16823
    },
    {
      "epoch": 1.6043663024902344e-05,
      "step": 16823,
      "training_step_time": 17.138933181762695
    },
    {
      "epoch": 1.604461669921875e-05,
      "model_forward_time": 5.616105794906616,
      "step": 16824
    },
    {
      "epoch": 1.604461669921875e-05,
      "step": 16824,
      "training_step_time": 17.12858295440674
    },
    {
      "epoch": 1.6045570373535156e-05,
      "model_forward_time": 5.603638410568237,
      "step": 16825
    },
    {
      "epoch": 1.6045570373535156e-05,
      "step": 16825,
      "training_step_time": 17.414501905441284
    },
    {
      "epoch": 1.6046524047851563e-05,
      "model_forward_time": 5.584672212600708,
      "step": 16826
    },
    {
      "epoch": 1.6046524047851563e-05,
      "step": 16826,
      "training_step_time": 17.02047610282898
    },
    {
      "epoch": 1.604747772216797e-05,
      "model_forward_time": 5.581918239593506,
      "step": 16827
    },
    {
      "epoch": 1.604747772216797e-05,
      "step": 16827,
      "training_step_time": 16.99968910217285
    },
    {
      "epoch": 1.6048431396484374e-05,
      "model_forward_time": 5.598481178283691,
      "step": 16828
    },
    {
      "epoch": 1.6048431396484374e-05,
      "step": 16828,
      "training_step_time": 17.035786628723145
    },
    {
      "epoch": 1.6049385070800782e-05,
      "model_forward_time": 5.60474419593811,
      "step": 16829
    },
    {
      "epoch": 1.6049385070800782e-05,
      "step": 16829,
      "training_step_time": 17.061063528060913
    },
    {
      "epoch": 1.6050338745117186e-05,
      "grad_norm": 0.6386767625808716,
      "learning_rate": 1e-05,
      "loss": 0.061,
      "step": 16830
    },
    {
      "epoch": 1.6050338745117186e-05,
      "model_forward_time": 5.590680360794067,
      "step": 16830
    },
    {
      "dynamics_loss_avg": 0.05498638860881329,
      "epoch": 1.6050338745117186e-05,
      "step": 16830
    },
    {
      "action_loss_avg": 0.00654718023724854,
      "epoch": 1.6050338745117186e-05,
      "step": 16830
    },
    {
      "epoch": 1.6050338745117186e-05,
      "step": 16830,
      "training_step_time": 17.028958559036255
    },
    {
      "epoch": 1.6051292419433593e-05,
      "model_forward_time": 5.977390289306641,
      "step": 16831
    },
    {
      "epoch": 1.6051292419433593e-05,
      "step": 16831,
      "training_step_time": 17.44259214401245
    },
    {
      "epoch": 1.605224609375e-05,
      "model_forward_time": 5.603285312652588,
      "step": 16832
    },
    {
      "epoch": 1.605224609375e-05,
      "step": 16832,
      "training_step_time": 17.049274444580078
    },
    {
      "epoch": 1.6053199768066405e-05,
      "model_forward_time": 5.595063209533691,
      "step": 16833
    },
    {
      "epoch": 1.6053199768066405e-05,
      "step": 16833,
      "training_step_time": 17.044037580490112
    },
    {
      "epoch": 1.6054153442382812e-05,
      "model_forward_time": 5.588444709777832,
      "step": 16834
    },
    {
      "epoch": 1.6054153442382812e-05,
      "step": 16834,
      "training_step_time": 17.04805588722229
    },
    {
      "epoch": 1.605510711669922e-05,
      "model_forward_time": 5.593536615371704,
      "step": 16835
    },
    {
      "epoch": 1.605510711669922e-05,
      "step": 16835,
      "training_step_time": 17.03832173347473
    },
    {
      "epoch": 1.6056060791015624e-05,
      "model_forward_time": 5.941478729248047,
      "step": 16836
    },
    {
      "epoch": 1.6056060791015624e-05,
      "step": 16836,
      "training_step_time": 17.3595552444458
    },
    {
      "epoch": 1.605701446533203e-05,
      "model_forward_time": 5.593140125274658,
      "step": 16837
    },
    {
      "epoch": 1.605701446533203e-05,
      "step": 16837,
      "training_step_time": 17.032660961151123
    },
    {
      "epoch": 1.605796813964844e-05,
      "model_forward_time": 5.5911009311676025,
      "step": 16838
    },
    {
      "epoch": 1.605796813964844e-05,
      "step": 16838,
      "training_step_time": 17.052552938461304
    },
    {
      "epoch": 1.6058921813964842e-05,
      "model_forward_time": 5.600286960601807,
      "step": 16839
    },
    {
      "epoch": 1.6058921813964842e-05,
      "step": 16839,
      "training_step_time": 17.09918785095215
    },
    {
      "epoch": 1.605987548828125e-05,
      "grad_norm": 1.3304656744003296,
      "learning_rate": 1e-05,
      "loss": 0.0663,
      "step": 16840
    },
    {
      "epoch": 1.605987548828125e-05,
      "model_forward_time": 5.61972188949585,
      "step": 16840
    },
    {
      "dynamics_loss_avg": 0.06170953176915646,
      "epoch": 1.605987548828125e-05,
      "step": 16840
    },
    {
      "action_loss_avg": 0.008624239009805024,
      "epoch": 1.605987548828125e-05,
      "step": 16840
    },
    {
      "epoch": 1.605987548828125e-05,
      "step": 16840,
      "training_step_time": 17.13617777824402
    },
    {
      "epoch": 1.6060829162597657e-05,
      "model_forward_time": 5.619028329849243,
      "step": 16841
    },
    {
      "epoch": 1.6060829162597657e-05,
      "step": 16841,
      "training_step_time": 17.13239860534668
    },
    {
      "epoch": 1.606178283691406e-05,
      "model_forward_time": 5.995274543762207,
      "step": 16842
    },
    {
      "epoch": 1.606178283691406e-05,
      "step": 16842,
      "training_step_time": 17.50872826576233
    },
    {
      "epoch": 1.606273651123047e-05,
      "model_forward_time": 5.6107306480407715,
      "step": 16843
    },
    {
      "epoch": 1.606273651123047e-05,
      "step": 16843,
      "training_step_time": 17.116214513778687
    },
    {
      "epoch": 1.6063690185546876e-05,
      "model_forward_time": 5.616969108581543,
      "step": 16844
    },
    {
      "epoch": 1.6063690185546876e-05,
      "step": 16844,
      "training_step_time": 17.124961614608765
    },
    {
      "epoch": 1.606464385986328e-05,
      "model_forward_time": 5.61211633682251,
      "step": 16845
    },
    {
      "epoch": 1.606464385986328e-05,
      "step": 16845,
      "training_step_time": 17.107547283172607
    },
    {
      "epoch": 1.6065597534179688e-05,
      "model_forward_time": 5.621044397354126,
      "step": 16846
    },
    {
      "epoch": 1.6065597534179688e-05,
      "step": 16846,
      "training_step_time": 17.126801013946533
    },
    {
      "epoch": 1.6066551208496095e-05,
      "model_forward_time": 5.615447521209717,
      "step": 16847
    },
    {
      "epoch": 1.6066551208496095e-05,
      "step": 16847,
      "training_step_time": 17.50023365020752
    },
    {
      "epoch": 1.60675048828125e-05,
      "model_forward_time": 5.621042490005493,
      "step": 16848
    },
    {
      "epoch": 1.60675048828125e-05,
      "step": 16848,
      "training_step_time": 17.129066944122314
    },
    {
      "epoch": 1.6068458557128906e-05,
      "model_forward_time": 5.617311000823975,
      "step": 16849
    },
    {
      "epoch": 1.6068458557128906e-05,
      "step": 16849,
      "training_step_time": 17.07949423789978
    },
    {
      "epoch": 1.6069412231445314e-05,
      "grad_norm": 0.11758927255868912,
      "learning_rate": 1e-05,
      "loss": 0.0571,
      "step": 16850
    },
    {
      "epoch": 1.6069412231445314e-05,
      "model_forward_time": 5.580110311508179,
      "step": 16850
    },
    {
      "dynamics_loss_avg": 0.051973081286996604,
      "epoch": 1.6069412231445314e-05,
      "step": 16850
    },
    {
      "action_loss_avg": 0.007123690517619252,
      "epoch": 1.6069412231445314e-05,
      "step": 16850
    },
    {
      "epoch": 1.6069412231445314e-05,
      "step": 16850,
      "training_step_time": 16.995880365371704
    },
    {
      "epoch": 1.6070365905761718e-05,
      "model_forward_time": 5.568271160125732,
      "step": 16851
    },
    {
      "epoch": 1.6070365905761718e-05,
      "step": 16851,
      "training_step_time": 16.992271184921265
    },
    {
      "epoch": 1.6071319580078125e-05,
      "model_forward_time": 5.578855991363525,
      "step": 16852
    },
    {
      "epoch": 1.6071319580078125e-05,
      "step": 16852,
      "training_step_time": 17.003172636032104
    },
    {
      "epoch": 1.6072273254394533e-05,
      "model_forward_time": 5.935711860656738,
      "step": 16853
    },
    {
      "epoch": 1.6072273254394533e-05,
      "step": 16853,
      "training_step_time": 17.374797582626343
    },
    {
      "epoch": 1.6073226928710937e-05,
      "model_forward_time": 5.568058252334595,
      "step": 16854
    },
    {
      "epoch": 1.6073226928710937e-05,
      "step": 16854,
      "training_step_time": 16.98211693763733
    },
    {
      "epoch": 1.6074180603027344e-05,
      "model_forward_time": 5.57671594619751,
      "step": 16855
    },
    {
      "epoch": 1.6074180603027344e-05,
      "step": 16855,
      "training_step_time": 17.011320114135742
    },
    {
      "epoch": 1.607513427734375e-05,
      "model_forward_time": 5.586772441864014,
      "step": 16856
    },
    {
      "epoch": 1.607513427734375e-05,
      "step": 16856,
      "training_step_time": 17.021584272384644
    },
    {
      "epoch": 1.6076087951660156e-05,
      "model_forward_time": 5.594339370727539,
      "step": 16857
    },
    {
      "epoch": 1.6076087951660156e-05,
      "step": 16857,
      "training_step_time": 17.053537607192993
    },
    {
      "epoch": 1.6077041625976563e-05,
      "model_forward_time": 5.599215984344482,
      "step": 16858
    },
    {
      "epoch": 1.6077041625976563e-05,
      "step": 16858,
      "training_step_time": 17.41594386100769
    },
    {
      "epoch": 1.607799530029297e-05,
      "model_forward_time": 5.594589948654175,
      "step": 16859
    },
    {
      "epoch": 1.607799530029297e-05,
      "step": 16859,
      "training_step_time": 17.050071239471436
    },
    {
      "epoch": 1.6078948974609374e-05,
      "grad_norm": 1.3701777458190918,
      "learning_rate": 1e-05,
      "loss": 0.0508,
      "step": 16860
    },
    {
      "epoch": 1.6078948974609374e-05,
      "model_forward_time": 5.591440677642822,
      "step": 16860
    },
    {
      "dynamics_loss_avg": 0.042713087797164914,
      "epoch": 1.6078948974609374e-05,
      "step": 16860
    },
    {
      "action_loss_avg": 0.004321478516794741,
      "epoch": 1.6078948974609374e-05,
      "step": 16860
    },
    {
      "epoch": 1.6078948974609374e-05,
      "step": 16860,
      "training_step_time": 17.038353443145752
    },
    {
      "epoch": 1.6079902648925782e-05,
      "model_forward_time": 5.584634304046631,
      "step": 16861
    },
    {
      "epoch": 1.6079902648925782e-05,
      "step": 16861,
      "training_step_time": 17.026882648468018
    },
    {
      "epoch": 1.6080856323242186e-05,
      "model_forward_time": 5.59458065032959,
      "step": 16862
    },
    {
      "epoch": 1.6080856323242186e-05,
      "step": 16862,
      "training_step_time": 17.040404796600342
    },
    {
      "epoch": 1.6081809997558593e-05,
      "model_forward_time": 5.594277858734131,
      "step": 16863
    },
    {
      "epoch": 1.6081809997558593e-05,
      "step": 16863,
      "training_step_time": 17.0661301612854
    },
    {
      "epoch": 1.6082763671875e-05,
      "model_forward_time": 5.996786594390869,
      "step": 16864
    },
    {
      "epoch": 1.6082763671875e-05,
      "step": 16864,
      "training_step_time": 17.45885133743286
    },
    {
      "epoch": 1.6083717346191405e-05,
      "model_forward_time": 5.594507694244385,
      "step": 16865
    },
    {
      "epoch": 1.6083717346191405e-05,
      "step": 16865,
      "training_step_time": 17.02332067489624
    },
    {
      "epoch": 1.6084671020507812e-05,
      "model_forward_time": 5.587597131729126,
      "step": 16866
    },
    {
      "epoch": 1.6084671020507812e-05,
      "step": 16866,
      "training_step_time": 17.008506059646606
    },
    {
      "epoch": 1.608562469482422e-05,
      "model_forward_time": 5.593235969543457,
      "step": 16867
    },
    {
      "epoch": 1.608562469482422e-05,
      "step": 16867,
      "training_step_time": 17.024012088775635
    },
    {
      "epoch": 1.6086578369140624e-05,
      "model_forward_time": 5.584976434707642,
      "step": 16868
    },
    {
      "epoch": 1.6086578369140624e-05,
      "step": 16868,
      "training_step_time": 17.036546230316162
    },
    {
      "epoch": 1.608753204345703e-05,
      "model_forward_time": 5.961994886398315,
      "step": 16869
    },
    {
      "epoch": 1.608753204345703e-05,
      "step": 16869,
      "training_step_time": 17.450647115707397
    },
    {
      "epoch": 1.608848571777344e-05,
      "grad_norm": 0.5433655977249146,
      "learning_rate": 1e-05,
      "loss": 0.0543,
      "step": 16870
    },
    {
      "epoch": 1.608848571777344e-05,
      "model_forward_time": 5.617400884628296,
      "step": 16870
    },
    {
      "dynamics_loss_avg": 0.04754965454339981,
      "epoch": 1.608848571777344e-05,
      "step": 16870
    },
    {
      "action_loss_avg": 0.006003241869620979,
      "epoch": 1.608848571777344e-05,
      "step": 16870
    },
    {
      "epoch": 1.608848571777344e-05,
      "step": 16870,
      "training_step_time": 17.135431051254272
    },
    {
      "epoch": 1.6089439392089842e-05,
      "model_forward_time": 5.6298182010650635,
      "step": 16871
    },
    {
      "epoch": 1.6089439392089842e-05,
      "step": 16871,
      "training_step_time": 17.154592990875244
    },
    {
      "epoch": 1.609039306640625e-05,
      "model_forward_time": 5.622026205062866,
      "step": 16872
    },
    {
      "epoch": 1.609039306640625e-05,
      "step": 16872,
      "training_step_time": 17.108261823654175
    },
    {
      "epoch": 1.6091346740722657e-05,
      "model_forward_time": 5.600876092910767,
      "step": 16873
    },
    {
      "epoch": 1.6091346740722657e-05,
      "step": 16873,
      "training_step_time": 17.049548149108887
    },
    {
      "epoch": 1.609230041503906e-05,
      "model_forward_time": 5.593824148178101,
      "step": 16874
    },
    {
      "epoch": 1.609230041503906e-05,
      "step": 16874,
      "training_step_time": 17.02722930908203
    },
    {
      "epoch": 1.609325408935547e-05,
      "model_forward_time": 5.970905542373657,
      "step": 16875
    },
    {
      "epoch": 1.609325408935547e-05,
      "step": 16875,
      "training_step_time": 17.407515048980713
    },
    {
      "epoch": 1.6094207763671876e-05,
      "model_forward_time": 5.601821422576904,
      "step": 16876
    },
    {
      "epoch": 1.6094207763671876e-05,
      "step": 16876,
      "training_step_time": 17.080586910247803
    },
    {
      "epoch": 1.609516143798828e-05,
      "model_forward_time": 5.607672929763794,
      "step": 16877
    },
    {
      "epoch": 1.609516143798828e-05,
      "step": 16877,
      "training_step_time": 17.07937002182007
    },
    {
      "epoch": 1.6096115112304688e-05,
      "model_forward_time": 5.602923393249512,
      "step": 16878
    },
    {
      "epoch": 1.6096115112304688e-05,
      "step": 16878,
      "training_step_time": 17.05768585205078
    },
    {
      "epoch": 1.6097068786621095e-05,
      "model_forward_time": 5.602649450302124,
      "step": 16879
    },
    {
      "epoch": 1.6097068786621095e-05,
      "step": 16879,
      "training_step_time": 17.073466777801514
    },
    {
      "epoch": 1.60980224609375e-05,
      "grad_norm": 0.3713197112083435,
      "learning_rate": 1e-05,
      "loss": 0.056,
      "step": 16880
    },
    {
      "epoch": 1.60980224609375e-05,
      "model_forward_time": 5.613576889038086,
      "step": 16880
    },
    {
      "dynamics_loss_avg": 0.0546520845964551,
      "epoch": 1.60980224609375e-05,
      "step": 16880
    },
    {
      "action_loss_avg": 0.007028297847136855,
      "epoch": 1.60980224609375e-05,
      "step": 16880
    },
    {
      "epoch": 1.60980224609375e-05,
      "step": 16880,
      "training_step_time": 17.455352306365967
    },
    {
      "epoch": 1.6098976135253906e-05,
      "model_forward_time": 5.601603269577026,
      "step": 16881
    },
    {
      "epoch": 1.6098976135253906e-05,
      "step": 16881,
      "training_step_time": 17.092275381088257
    },
    {
      "epoch": 1.6099929809570314e-05,
      "model_forward_time": 5.60743522644043,
      "step": 16882
    },
    {
      "epoch": 1.6099929809570314e-05,
      "step": 16882,
      "training_step_time": 17.086358308792114
    },
    {
      "epoch": 1.6100883483886718e-05,
      "model_forward_time": 5.605700254440308,
      "step": 16883
    },
    {
      "epoch": 1.6100883483886718e-05,
      "step": 16883,
      "training_step_time": 17.063910722732544
    },
    {
      "epoch": 1.6101837158203125e-05,
      "model_forward_time": 5.586091041564941,
      "step": 16884
    },
    {
      "epoch": 1.6101837158203125e-05,
      "step": 16884,
      "training_step_time": 17.033005237579346
    },
    {
      "epoch": 1.6102790832519533e-05,
      "model_forward_time": 5.599872827529907,
      "step": 16885
    },
    {
      "epoch": 1.6102790832519533e-05,
      "step": 16885,
      "training_step_time": 17.069552421569824
    },
    {
      "epoch": 1.6103744506835937e-05,
      "model_forward_time": 5.99957275390625,
      "step": 16886
    },
    {
      "epoch": 1.6103744506835937e-05,
      "step": 16886,
      "training_step_time": 17.454039096832275
    },
    {
      "epoch": 1.6104698181152344e-05,
      "model_forward_time": 5.608907222747803,
      "step": 16887
    },
    {
      "epoch": 1.6104698181152344e-05,
      "step": 16887,
      "training_step_time": 17.083760738372803
    },
    {
      "epoch": 1.610565185546875e-05,
      "model_forward_time": 5.606935739517212,
      "step": 16888
    },
    {
      "epoch": 1.610565185546875e-05,
      "step": 16888,
      "training_step_time": 17.089080095291138
    },
    {
      "epoch": 1.6106605529785156e-05,
      "model_forward_time": 5.607849597930908,
      "step": 16889
    },
    {
      "epoch": 1.6106605529785156e-05,
      "step": 16889,
      "training_step_time": 17.10696530342102
    },
    {
      "epoch": 1.6107559204101563e-05,
      "grad_norm": 0.21756847202777863,
      "learning_rate": 1e-05,
      "loss": 0.0589,
      "step": 16890
    },
    {
      "epoch": 1.6107559204101563e-05,
      "model_forward_time": 5.605716705322266,
      "step": 16890
    },
    {
      "dynamics_loss_avg": 0.04749102853238583,
      "epoch": 1.6107559204101563e-05,
      "step": 16890
    },
    {
      "action_loss_avg": 0.006158207473345101,
      "epoch": 1.6107559204101563e-05,
      "step": 16890
    },
    {
      "epoch": 1.6107559204101563e-05,
      "step": 16890,
      "training_step_time": 17.09916377067566
    },
    {
      "epoch": 1.6108512878417967e-05,
      "model_forward_time": 5.610507249832153,
      "step": 16891
    },
    {
      "epoch": 1.6108512878417967e-05,
      "step": 16891,
      "training_step_time": 17.47997736930847
    },
    {
      "epoch": 1.6109466552734374e-05,
      "model_forward_time": 5.627285480499268,
      "step": 16892
    },
    {
      "epoch": 1.6109466552734374e-05,
      "step": 16892,
      "training_step_time": 17.12399697303772
    },
    {
      "epoch": 1.6110420227050782e-05,
      "model_forward_time": 5.6092369556427,
      "step": 16893
    },
    {
      "epoch": 1.6110420227050782e-05,
      "step": 16893,
      "training_step_time": 17.10531997680664
    },
    {
      "epoch": 1.6111373901367186e-05,
      "model_forward_time": 5.638296365737915,
      "step": 16894
    },
    {
      "epoch": 1.6111373901367186e-05,
      "step": 16894,
      "training_step_time": 17.14713478088379
    },
    {
      "epoch": 1.6112327575683593e-05,
      "model_forward_time": 5.611093282699585,
      "step": 16895
    },
    {
      "epoch": 1.6112327575683593e-05,
      "step": 16895,
      "training_step_time": 17.09614849090576
    },
    {
      "epoch": 1.611328125e-05,
      "model_forward_time": 5.607630968093872,
      "step": 16896
    },
    {
      "epoch": 1.611328125e-05,
      "step": 16896,
      "training_step_time": 17.085591316223145
    },
    {
      "epoch": 1.6114234924316405e-05,
      "model_forward_time": 5.98736047744751,
      "step": 16897
    },
    {
      "epoch": 1.6114234924316405e-05,
      "step": 16897,
      "training_step_time": 17.46012282371521
    },
    {
      "epoch": 1.6115188598632812e-05,
      "model_forward_time": 5.612742900848389,
      "step": 16898
    },
    {
      "epoch": 1.6115188598632812e-05,
      "step": 16898,
      "training_step_time": 17.094736099243164
    },
    {
      "epoch": 1.611614227294922e-05,
      "model_forward_time": 5.6233274936676025,
      "step": 16899
    },
    {
      "epoch": 1.611614227294922e-05,
      "step": 16899,
      "training_step_time": 17.130056142807007
    },
    {
      "epoch": 1.6117095947265624e-05,
      "grad_norm": 0.11585192382335663,
      "learning_rate": 1e-05,
      "loss": 0.0608,
      "step": 16900
    },
    {
      "epoch": 1.6117095947265624e-05,
      "model_forward_time": 5.627526760101318,
      "step": 16900
    },
    {
      "dynamics_loss_avg": 0.05707852346822619,
      "epoch": 1.6117095947265624e-05,
      "step": 16900
    },
    {
      "action_loss_avg": 0.006541909312363714,
      "epoch": 1.6117095947265624e-05,
      "step": 16900
    },
    {
      "epoch": 1.6117095947265624e-05,
      "step": 16900,
      "training_step_time": 17.126198291778564
    },
    {
      "epoch": 1.611804962158203e-05,
      "model_forward_time": 5.597094535827637,
      "step": 16901
    },
    {
      "epoch": 1.611804962158203e-05,
      "step": 16901,
      "training_step_time": 17.0136775970459
    },
    {
      "epoch": 1.611900329589844e-05,
      "model_forward_time": 5.947361946105957,
      "step": 16902
    },
    {
      "epoch": 1.611900329589844e-05,
      "step": 16902,
      "training_step_time": 17.39141845703125
    },
    {
      "epoch": 1.6119956970214843e-05,
      "model_forward_time": 5.572713375091553,
      "step": 16903
    },
    {
      "epoch": 1.6119956970214843e-05,
      "step": 16903,
      "training_step_time": 16.985123872756958
    },
    {
      "epoch": 1.612091064453125e-05,
      "model_forward_time": 5.57468843460083,
      "step": 16904
    },
    {
      "epoch": 1.612091064453125e-05,
      "step": 16904,
      "training_step_time": 16.98800301551819
    },
    {
      "epoch": 1.6121864318847657e-05,
      "model_forward_time": 5.58536696434021,
      "step": 16905
    },
    {
      "epoch": 1.6121864318847657e-05,
      "step": 16905,
      "training_step_time": 17.014243364334106
    },
    {
      "epoch": 1.612281799316406e-05,
      "model_forward_time": 5.592786073684692,
      "step": 16906
    },
    {
      "epoch": 1.612281799316406e-05,
      "step": 16906,
      "training_step_time": 17.056797742843628
    },
    {
      "epoch": 1.612377166748047e-05,
      "model_forward_time": 5.603843450546265,
      "step": 16907
    },
    {
      "epoch": 1.612377166748047e-05,
      "step": 16907,
      "training_step_time": 17.08140540122986
    },
    {
      "epoch": 1.6124725341796876e-05,
      "model_forward_time": 5.993437051773071,
      "step": 16908
    },
    {
      "epoch": 1.6124725341796876e-05,
      "step": 16908,
      "training_step_time": 17.47883677482605
    },
    {
      "epoch": 1.612567901611328e-05,
      "model_forward_time": 5.602078676223755,
      "step": 16909
    },
    {
      "epoch": 1.612567901611328e-05,
      "step": 16909,
      "training_step_time": 17.091928958892822
    },
    {
      "epoch": 1.6126632690429688e-05,
      "grad_norm": 0.8393827676773071,
      "learning_rate": 1e-05,
      "loss": 0.0741,
      "step": 16910
    },
    {
      "epoch": 1.6126632690429688e-05,
      "model_forward_time": 5.6074769496917725,
      "step": 16910
    },
    {
      "dynamics_loss_avg": 0.0642453171312809,
      "epoch": 1.6126632690429688e-05,
      "step": 16910
    },
    {
      "action_loss_avg": 0.009837294858880342,
      "epoch": 1.6126632690429688e-05,
      "step": 16910
    },
    {
      "epoch": 1.6126632690429688e-05,
      "step": 16910,
      "training_step_time": 17.113187789916992
    },
    {
      "epoch": 1.6127586364746095e-05,
      "model_forward_time": 5.608505010604858,
      "step": 16911
    },
    {
      "epoch": 1.6127586364746095e-05,
      "step": 16911,
      "training_step_time": 17.09380316734314
    },
    {
      "epoch": 1.61285400390625e-05,
      "model_forward_time": 5.604777097702026,
      "step": 16912
    },
    {
      "epoch": 1.61285400390625e-05,
      "step": 16912,
      "training_step_time": 17.067421197891235
    },
    {
      "epoch": 1.6129493713378906e-05,
      "model_forward_time": 5.590501546859741,
      "step": 16913
    },
    {
      "epoch": 1.6129493713378906e-05,
      "step": 16913,
      "training_step_time": 17.405897855758667
    },
    {
      "epoch": 1.6130447387695314e-05,
      "model_forward_time": 5.596231937408447,
      "step": 16914
    },
    {
      "epoch": 1.6130447387695314e-05,
      "step": 16914,
      "training_step_time": 17.06840229034424
    },
    {
      "epoch": 1.6131401062011718e-05,
      "model_forward_time": 5.623544931411743,
      "step": 16915
    },
    {
      "epoch": 1.6131401062011718e-05,
      "step": 16915,
      "training_step_time": 17.119226932525635
    },
    {
      "epoch": 1.6132354736328125e-05,
      "model_forward_time": 5.614354372024536,
      "step": 16916
    },
    {
      "epoch": 1.6132354736328125e-05,
      "step": 16916,
      "training_step_time": 17.137025594711304
    },
    {
      "epoch": 1.6133308410644533e-05,
      "model_forward_time": 5.625096321105957,
      "step": 16917
    },
    {
      "epoch": 1.6133308410644533e-05,
      "step": 16917,
      "training_step_time": 17.11888027191162
    },
    {
      "epoch": 1.6134262084960937e-05,
      "model_forward_time": 5.59514594078064,
      "step": 16918
    },
    {
      "epoch": 1.6134262084960937e-05,
      "step": 16918,
      "training_step_time": 17.023970127105713
    },
    {
      "epoch": 1.6135215759277344e-05,
      "model_forward_time": 4.652219295501709,
      "step": 16919
    },
    {
      "epoch": 1.6135215759277344e-05,
      "step": 16919,
      "training_step_time": 13.145575761795044
    },
    {
      "epoch": 1.613616943359375e-05,
      "grad_norm": 1.5869184732437134,
      "learning_rate": 1e-05,
      "loss": 0.0742,
      "step": 16920
    },
    {
      "epoch": 1.613616943359375e-05,
      "model_forward_time": 5.5711164474487305,
      "step": 16920
    },
    {
      "dynamics_loss_avg": 0.063633949495852,
      "epoch": 1.613616943359375e-05,
      "step": 16920
    },
    {
      "action_loss_avg": 0.009822536935098469,
      "epoch": 1.613616943359375e-05,
      "step": 16920
    },
    {
      "epoch": 1.613616943359375e-05,
      "step": 16920,
      "training_step_time": 16.974802017211914
    },
    {
      "epoch": 1.6137123107910156e-05,
      "model_forward_time": 5.5947675704956055,
      "step": 16921
    },
    {
      "epoch": 1.6137123107910156e-05,
      "step": 16921,
      "training_step_time": 17.03927206993103
    },
    {
      "epoch": 1.6138076782226563e-05,
      "model_forward_time": 5.584824562072754,
      "step": 16922
    },
    {
      "epoch": 1.6138076782226563e-05,
      "step": 16922,
      "training_step_time": 17.01679563522339
    },
    {
      "epoch": 1.6139030456542967e-05,
      "model_forward_time": 5.593507528305054,
      "step": 16923
    },
    {
      "epoch": 1.6139030456542967e-05,
      "step": 16923,
      "training_step_time": 17.054835081100464
    },
    {
      "epoch": 1.6139984130859375e-05,
      "model_forward_time": 5.600270748138428,
      "step": 16924
    },
    {
      "epoch": 1.6139984130859375e-05,
      "step": 16924,
      "training_step_time": 17.075594425201416
    },
    {
      "epoch": 1.6140937805175782e-05,
      "model_forward_time": 5.944987058639526,
      "step": 16925
    },
    {
      "epoch": 1.6140937805175782e-05,
      "step": 16925,
      "training_step_time": 17.417996406555176
    },
    {
      "epoch": 1.6141891479492186e-05,
      "model_forward_time": 5.601264238357544,
      "step": 16926
    },
    {
      "epoch": 1.6141891479492186e-05,
      "step": 16926,
      "training_step_time": 17.069150924682617
    },
    {
      "epoch": 1.6142845153808593e-05,
      "model_forward_time": 5.5946044921875,
      "step": 16927
    },
    {
      "epoch": 1.6142845153808593e-05,
      "step": 16927,
      "training_step_time": 17.04337453842163
    },
    {
      "epoch": 1.6143798828125e-05,
      "model_forward_time": 5.597365856170654,
      "step": 16928
    },
    {
      "epoch": 1.6143798828125e-05,
      "step": 16928,
      "training_step_time": 17.04323148727417
    },
    {
      "epoch": 1.6144752502441405e-05,
      "model_forward_time": 5.603200912475586,
      "step": 16929
    },
    {
      "epoch": 1.6144752502441405e-05,
      "step": 16929,
      "training_step_time": 17.068138599395752
    },
    {
      "epoch": 1.6145706176757812e-05,
      "grad_norm": 0.5009944438934326,
      "learning_rate": 1e-05,
      "loss": 0.0679,
      "step": 16930
    },
    {
      "epoch": 1.6145706176757812e-05,
      "model_forward_time": 5.978485345840454,
      "step": 16930
    },
    {
      "dynamics_loss_avg": 0.05802580919116736,
      "epoch": 1.6145706176757812e-05,
      "step": 16930
    },
    {
      "action_loss_avg": 0.006670446030329913,
      "epoch": 1.6145706176757812e-05,
      "step": 16930
    },
    {
      "epoch": 1.6145706176757812e-05,
      "step": 16930,
      "training_step_time": 17.449052333831787
    },
    {
      "epoch": 1.614665985107422e-05,
      "model_forward_time": 5.598632335662842,
      "step": 16931
    },
    {
      "epoch": 1.614665985107422e-05,
      "step": 16931,
      "training_step_time": 17.069080114364624
    },
    {
      "epoch": 1.6147613525390624e-05,
      "model_forward_time": 5.596548795700073,
      "step": 16932
    },
    {
      "epoch": 1.6147613525390624e-05,
      "step": 16932,
      "training_step_time": 17.04548454284668
    },
    {
      "epoch": 1.614856719970703e-05,
      "model_forward_time": 5.599619150161743,
      "step": 16933
    },
    {
      "epoch": 1.614856719970703e-05,
      "step": 16933,
      "training_step_time": 17.05746626853943
    },
    {
      "epoch": 1.614952087402344e-05,
      "model_forward_time": 5.5950000286102295,
      "step": 16934
    },
    {
      "epoch": 1.614952087402344e-05,
      "step": 16934,
      "training_step_time": 17.053740739822388
    },
    {
      "epoch": 1.6150474548339843e-05,
      "model_forward_time": 5.958244323730469,
      "step": 16935
    },
    {
      "epoch": 1.6150474548339843e-05,
      "step": 16935,
      "training_step_time": 17.429625988006592
    },
    {
      "epoch": 1.615142822265625e-05,
      "model_forward_time": 5.613843202590942,
      "step": 16936
    },
    {
      "epoch": 1.615142822265625e-05,
      "step": 16936,
      "training_step_time": 17.094831228256226
    },
    {
      "epoch": 1.6152381896972657e-05,
      "model_forward_time": 5.611071825027466,
      "step": 16937
    },
    {
      "epoch": 1.6152381896972657e-05,
      "step": 16937,
      "training_step_time": 17.072851181030273
    },
    {
      "epoch": 1.615333557128906e-05,
      "model_forward_time": 5.618352890014648,
      "step": 16938
    },
    {
      "epoch": 1.615333557128906e-05,
      "step": 16938,
      "training_step_time": 17.09475541114807
    },
    {
      "epoch": 1.615428924560547e-05,
      "model_forward_time": 5.593206167221069,
      "step": 16939
    },
    {
      "epoch": 1.615428924560547e-05,
      "step": 16939,
      "training_step_time": 17.029086351394653
    },
    {
      "epoch": 1.6155242919921876e-05,
      "grad_norm": 0.616018533706665,
      "learning_rate": 1e-05,
      "loss": 0.0804,
      "step": 16940
    },
    {
      "epoch": 1.6155242919921876e-05,
      "model_forward_time": 5.588009595870972,
      "step": 16940
    },
    {
      "dynamics_loss_avg": 0.07149860467761755,
      "epoch": 1.6155242919921876e-05,
      "step": 16940
    },
    {
      "action_loss_avg": 0.007997590827289968,
      "epoch": 1.6155242919921876e-05,
      "step": 16940
    },
    {
      "epoch": 1.6155242919921876e-05,
      "step": 16940,
      "training_step_time": 17.01891326904297
    },
    {
      "epoch": 1.615619659423828e-05,
      "model_forward_time": 5.950963735580444,
      "step": 16941
    },
    {
      "epoch": 1.615619659423828e-05,
      "step": 16941,
      "training_step_time": 17.37429690361023
    },
    {
      "epoch": 1.6157150268554688e-05,
      "model_forward_time": 5.5779948234558105,
      "step": 16942
    },
    {
      "epoch": 1.6157150268554688e-05,
      "step": 16942,
      "training_step_time": 16.98568320274353
    },
    {
      "epoch": 1.6158103942871095e-05,
      "model_forward_time": 5.5879127979278564,
      "step": 16943
    },
    {
      "epoch": 1.6158103942871095e-05,
      "step": 16943,
      "training_step_time": 17.033565282821655
    },
    {
      "epoch": 1.61590576171875e-05,
      "model_forward_time": 5.596873760223389,
      "step": 16944
    },
    {
      "epoch": 1.61590576171875e-05,
      "step": 16944,
      "training_step_time": 17.05220937728882
    },
    {
      "epoch": 1.6160011291503907e-05,
      "model_forward_time": 5.613886833190918,
      "step": 16945
    },
    {
      "epoch": 1.6160011291503907e-05,
      "step": 16945,
      "training_step_time": 17.115004539489746
    },
    {
      "epoch": 1.6160964965820314e-05,
      "model_forward_time": 5.631903886795044,
      "step": 16946
    },
    {
      "epoch": 1.6160964965820314e-05,
      "step": 16946,
      "training_step_time": 17.511760711669922
    },
    {
      "epoch": 1.6161918640136718e-05,
      "model_forward_time": 5.609458923339844,
      "step": 16947
    },
    {
      "epoch": 1.6161918640136718e-05,
      "step": 16947,
      "training_step_time": 17.11646032333374
    },
    {
      "epoch": 1.6162872314453125e-05,
      "model_forward_time": 5.610529899597168,
      "step": 16948
    },
    {
      "epoch": 1.6162872314453125e-05,
      "step": 16948,
      "training_step_time": 17.065547943115234
    },
    {
      "epoch": 1.6163825988769533e-05,
      "model_forward_time": 5.615763425827026,
      "step": 16949
    },
    {
      "epoch": 1.6163825988769533e-05,
      "step": 16949,
      "training_step_time": 17.0888192653656
    },
    {
      "epoch": 1.6164779663085937e-05,
      "grad_norm": 0.5597243905067444,
      "learning_rate": 1e-05,
      "loss": 0.0624,
      "step": 16950
    },
    {
      "epoch": 1.6164779663085937e-05,
      "model_forward_time": 5.604177951812744,
      "step": 16950
    },
    {
      "dynamics_loss_avg": 0.05572748649865389,
      "epoch": 1.6164779663085937e-05,
      "step": 16950
    },
    {
      "action_loss_avg": 0.0076060697436332704,
      "epoch": 1.6164779663085937e-05,
      "step": 16950
    },
    {
      "epoch": 1.6164779663085937e-05,
      "step": 16950,
      "training_step_time": 17.080164909362793
    },
    {
      "epoch": 1.6165733337402344e-05,
      "model_forward_time": 5.615915536880493,
      "step": 16951
    },
    {
      "epoch": 1.6165733337402344e-05,
      "step": 16951,
      "training_step_time": 17.096357583999634
    },
    {
      "epoch": 1.616668701171875e-05,
      "model_forward_time": 5.9887635707855225,
      "step": 16952
    },
    {
      "epoch": 1.616668701171875e-05,
      "step": 16952,
      "training_step_time": 17.427177906036377
    },
    {
      "epoch": 1.6167640686035156e-05,
      "model_forward_time": 5.5939154624938965,
      "step": 16953
    },
    {
      "epoch": 1.6167640686035156e-05,
      "step": 16953,
      "training_step_time": 17.05180048942566
    },
    {
      "epoch": 1.6168594360351563e-05,
      "model_forward_time": 5.586028099060059,
      "step": 16954
    },
    {
      "epoch": 1.6168594360351563e-05,
      "step": 16954,
      "training_step_time": 17.052520275115967
    },
    {
      "epoch": 1.6169548034667967e-05,
      "model_forward_time": 5.612091064453125,
      "step": 16955
    },
    {
      "epoch": 1.6169548034667967e-05,
      "step": 16955,
      "training_step_time": 17.089550971984863
    },
    {
      "epoch": 1.6170501708984375e-05,
      "model_forward_time": 5.6084794998168945,
      "step": 16956
    },
    {
      "epoch": 1.6170501708984375e-05,
      "step": 16956,
      "training_step_time": 17.09672498703003
    },
    {
      "epoch": 1.6171455383300782e-05,
      "model_forward_time": 5.60707426071167,
      "step": 16957
    },
    {
      "epoch": 1.6171455383300782e-05,
      "step": 16957,
      "training_step_time": 17.468535661697388
    },
    {
      "epoch": 1.6172409057617186e-05,
      "model_forward_time": 5.615124464035034,
      "step": 16958
    },
    {
      "epoch": 1.6172409057617186e-05,
      "step": 16958,
      "training_step_time": 17.102952003479004
    },
    {
      "epoch": 1.6173362731933593e-05,
      "model_forward_time": 5.6066319942474365,
      "step": 16959
    },
    {
      "epoch": 1.6173362731933593e-05,
      "step": 16959,
      "training_step_time": 17.10569930076599
    },
    {
      "epoch": 1.617431640625e-05,
      "grad_norm": 1.4429216384887695,
      "learning_rate": 1e-05,
      "loss": 0.0611,
      "step": 16960
    },
    {
      "epoch": 1.617431640625e-05,
      "model_forward_time": 5.616161584854126,
      "step": 16960
    },
    {
      "dynamics_loss_avg": 0.06156924515962601,
      "epoch": 1.617431640625e-05,
      "step": 16960
    },
    {
      "action_loss_avg": 0.007359704491682351,
      "epoch": 1.617431640625e-05,
      "step": 16960
    },
    {
      "epoch": 1.617431640625e-05,
      "step": 16960,
      "training_step_time": 17.116878271102905
    },
    {
      "epoch": 1.6175270080566405e-05,
      "model_forward_time": 5.6113386154174805,
      "step": 16961
    },
    {
      "epoch": 1.6175270080566405e-05,
      "step": 16961,
      "training_step_time": 17.073001861572266
    },
    {
      "epoch": 1.6176223754882812e-05,
      "model_forward_time": 5.601579904556274,
      "step": 16962
    },
    {
      "epoch": 1.6176223754882812e-05,
      "step": 16962,
      "training_step_time": 17.042089223861694
    },
    {
      "epoch": 1.617717742919922e-05,
      "model_forward_time": 5.963543176651001,
      "step": 16963
    },
    {
      "epoch": 1.617717742919922e-05,
      "step": 16963,
      "training_step_time": 17.429607391357422
    },
    {
      "epoch": 1.6178131103515624e-05,
      "model_forward_time": 5.6154327392578125,
      "step": 16964
    },
    {
      "epoch": 1.6178131103515624e-05,
      "step": 16964,
      "training_step_time": 17.111644744873047
    },
    {
      "epoch": 1.617908477783203e-05,
      "model_forward_time": 5.605800151824951,
      "step": 16965
    },
    {
      "epoch": 1.617908477783203e-05,
      "step": 16965,
      "training_step_time": 17.111387014389038
    },
    {
      "epoch": 1.618003845214844e-05,
      "model_forward_time": 5.618746519088745,
      "step": 16966
    },
    {
      "epoch": 1.618003845214844e-05,
      "step": 16966,
      "training_step_time": 17.11849856376648
    },
    {
      "epoch": 1.6180992126464843e-05,
      "model_forward_time": 5.620251655578613,
      "step": 16967
    },
    {
      "epoch": 1.6180992126464843e-05,
      "step": 16967,
      "training_step_time": 17.115610361099243
    },
    {
      "epoch": 1.618194580078125e-05,
      "model_forward_time": 5.961174249649048,
      "step": 16968
    },
    {
      "epoch": 1.618194580078125e-05,
      "step": 16968,
      "training_step_time": 17.443460941314697
    },
    {
      "epoch": 1.6182899475097657e-05,
      "model_forward_time": 5.613499402999878,
      "step": 16969
    },
    {
      "epoch": 1.6182899475097657e-05,
      "step": 16969,
      "training_step_time": 17.094098329544067
    },
    {
      "epoch": 1.618385314941406e-05,
      "grad_norm": 0.6214340329170227,
      "learning_rate": 1e-05,
      "loss": 0.0608,
      "step": 16970
    },
    {
      "epoch": 1.618385314941406e-05,
      "model_forward_time": 5.60633397102356,
      "step": 16970
    },
    {
      "dynamics_loss_avg": 0.05016189888119697,
      "epoch": 1.618385314941406e-05,
      "step": 16970
    },
    {
      "action_loss_avg": 0.007194693572819233,
      "epoch": 1.618385314941406e-05,
      "step": 16970
    },
    {
      "epoch": 1.618385314941406e-05,
      "step": 16970,
      "training_step_time": 17.110264539718628
    },
    {
      "epoch": 1.618480682373047e-05,
      "model_forward_time": 5.609661817550659,
      "step": 16971
    },
    {
      "epoch": 1.618480682373047e-05,
      "step": 16971,
      "training_step_time": 17.105939388275146
    },
    {
      "epoch": 1.6185760498046876e-05,
      "model_forward_time": 5.600070238113403,
      "step": 16972
    },
    {
      "epoch": 1.6185760498046876e-05,
      "step": 16972,
      "training_step_time": 17.087249279022217
    },
    {
      "epoch": 1.618671417236328e-05,
      "model_forward_time": 5.601504802703857,
      "step": 16973
    },
    {
      "epoch": 1.618671417236328e-05,
      "step": 16973,
      "training_step_time": 17.092263221740723
    },
    {
      "epoch": 1.6187667846679688e-05,
      "model_forward_time": 6.000344753265381,
      "step": 16974
    },
    {
      "epoch": 1.6187667846679688e-05,
      "step": 16974,
      "training_step_time": 17.485419750213623
    },
    {
      "epoch": 1.6188621520996095e-05,
      "model_forward_time": 5.604907274246216,
      "step": 16975
    },
    {
      "epoch": 1.6188621520996095e-05,
      "step": 16975,
      "training_step_time": 17.07563614845276
    },
    {
      "epoch": 1.61895751953125e-05,
      "model_forward_time": 5.597836017608643,
      "step": 16976
    },
    {
      "epoch": 1.61895751953125e-05,
      "step": 16976,
      "training_step_time": 17.063945055007935
    },
    {
      "epoch": 1.6190528869628907e-05,
      "model_forward_time": 5.6042351722717285,
      "step": 16977
    },
    {
      "epoch": 1.6190528869628907e-05,
      "step": 16977,
      "training_step_time": 17.056560039520264
    },
    {
      "epoch": 1.6191482543945314e-05,
      "model_forward_time": 5.605785131454468,
      "step": 16978
    },
    {
      "epoch": 1.6191482543945314e-05,
      "step": 16978,
      "training_step_time": 17.0981867313385
    },
    {
      "epoch": 1.6192436218261718e-05,
      "model_forward_time": 5.6196982860565186,
      "step": 16979
    },
    {
      "epoch": 1.6192436218261718e-05,
      "step": 16979,
      "training_step_time": 17.508190393447876
    },
    {
      "epoch": 1.6193389892578125e-05,
      "grad_norm": 0.596656084060669,
      "learning_rate": 1e-05,
      "loss": 0.0687,
      "step": 16980
    },
    {
      "epoch": 1.6193389892578125e-05,
      "model_forward_time": 5.633896589279175,
      "step": 16980
    },
    {
      "dynamics_loss_avg": 0.060347037576138976,
      "epoch": 1.6193389892578125e-05,
      "step": 16980
    },
    {
      "action_loss_avg": 0.006698667514137923,
      "epoch": 1.6193389892578125e-05,
      "step": 16980
    },
    {
      "epoch": 1.6193389892578125e-05,
      "step": 16980,
      "training_step_time": 17.160717964172363
    },
    {
      "epoch": 1.6194343566894533e-05,
      "model_forward_time": 5.621621608734131,
      "step": 16981
    },
    {
      "epoch": 1.6194343566894533e-05,
      "step": 16981,
      "training_step_time": 17.118711709976196
    },
    {
      "epoch": 1.6195297241210937e-05,
      "model_forward_time": 5.615461587905884,
      "step": 16982
    },
    {
      "epoch": 1.6195297241210937e-05,
      "step": 16982,
      "training_step_time": 17.097527742385864
    },
    {
      "epoch": 1.6196250915527344e-05,
      "model_forward_time": 5.601709365844727,
      "step": 16983
    },
    {
      "epoch": 1.6196250915527344e-05,
      "step": 16983,
      "training_step_time": 17.063125610351562
    },
    {
      "epoch": 1.619720458984375e-05,
      "model_forward_time": 5.59023380279541,
      "step": 16984
    },
    {
      "epoch": 1.619720458984375e-05,
      "step": 16984,
      "training_step_time": 17.022657871246338
    },
    {
      "epoch": 1.6198158264160156e-05,
      "model_forward_time": 5.94732141494751,
      "step": 16985
    },
    {
      "epoch": 1.6198158264160156e-05,
      "step": 16985,
      "training_step_time": 17.35415506362915
    },
    {
      "epoch": 1.6199111938476563e-05,
      "model_forward_time": 5.5769126415252686,
      "step": 16986
    },
    {
      "epoch": 1.6199111938476563e-05,
      "step": 16986,
      "training_step_time": 17.006778955459595
    },
    {
      "epoch": 1.6200065612792967e-05,
      "model_forward_time": 5.5860583782196045,
      "step": 16987
    },
    {
      "epoch": 1.6200065612792967e-05,
      "step": 16987,
      "training_step_time": 17.006507635116577
    },
    {
      "epoch": 1.6201019287109375e-05,
      "model_forward_time": 5.597289085388184,
      "step": 16988
    },
    {
      "epoch": 1.6201019287109375e-05,
      "step": 16988,
      "training_step_time": 17.06334924697876
    },
    {
      "epoch": 1.6201972961425782e-05,
      "model_forward_time": 5.607841491699219,
      "step": 16989
    },
    {
      "epoch": 1.6201972961425782e-05,
      "step": 16989,
      "training_step_time": 17.114439249038696
    },
    {
      "epoch": 1.6202926635742186e-05,
      "grad_norm": 0.3751372992992401,
      "learning_rate": 1e-05,
      "loss": 0.0584,
      "step": 16990
    },
    {
      "epoch": 1.6202926635742186e-05,
      "model_forward_time": 5.623027324676514,
      "step": 16990
    },
    {
      "dynamics_loss_avg": 0.04976275768131018,
      "epoch": 1.6202926635742186e-05,
      "step": 16990
    },
    {
      "action_loss_avg": 0.005287918006069958,
      "epoch": 1.6202926635742186e-05,
      "step": 16990
    },
    {
      "epoch": 1.6202926635742186e-05,
      "step": 16990,
      "training_step_time": 17.510758876800537
    },
    {
      "epoch": 1.6203880310058593e-05,
      "model_forward_time": 5.619587182998657,
      "step": 16991
    },
    {
      "epoch": 1.6203880310058593e-05,
      "step": 16991,
      "training_step_time": 17.142711877822876
    },
    {
      "epoch": 1.6204833984375e-05,
      "model_forward_time": 5.623829364776611,
      "step": 16992
    },
    {
      "epoch": 1.6204833984375e-05,
      "step": 16992,
      "training_step_time": 17.128674268722534
    },
    {
      "epoch": 1.6205787658691405e-05,
      "model_forward_time": 5.620015859603882,
      "step": 16993
    },
    {
      "epoch": 1.6205787658691405e-05,
      "step": 16993,
      "training_step_time": 17.12018871307373
    },
    {
      "epoch": 1.6206741333007812e-05,
      "model_forward_time": 5.612215518951416,
      "step": 16994
    },
    {
      "epoch": 1.6206741333007812e-05,
      "step": 16994,
      "training_step_time": 17.108299732208252
    },
    {
      "epoch": 1.620769500732422e-05,
      "model_forward_time": 5.608876705169678,
      "step": 16995
    },
    {
      "epoch": 1.620769500732422e-05,
      "step": 16995,
      "training_step_time": 17.094876527786255
    },
    {
      "epoch": 1.6208648681640624e-05,
      "model_forward_time": 5.977972030639648,
      "step": 16996
    },
    {
      "epoch": 1.6208648681640624e-05,
      "step": 16996,
      "training_step_time": 17.461629629135132
    },
    {
      "epoch": 1.620960235595703e-05,
      "model_forward_time": 5.610260009765625,
      "step": 16997
    },
    {
      "epoch": 1.620960235595703e-05,
      "step": 16997,
      "training_step_time": 17.09536862373352
    },
    {
      "epoch": 1.621055603027344e-05,
      "model_forward_time": 5.6072838306427,
      "step": 16998
    },
    {
      "epoch": 1.621055603027344e-05,
      "step": 16998,
      "training_step_time": 17.081655263900757
    },
    {
      "epoch": 1.6211509704589843e-05,
      "model_forward_time": 5.6008336544036865,
      "step": 16999
    },
    {
      "epoch": 1.6211509704589843e-05,
      "step": 16999,
      "training_step_time": 17.075456857681274
    },
    {
      "epoch": 1.621246337890625e-05,
      "grad_norm": 0.625505805015564,
      "learning_rate": 1e-05,
      "loss": 0.0831,
      "step": 17000
    },
    {
      "epoch": 1.621246337890625e-05,
      "model_forward_time": 5.653736114501953,
      "step": 17000
    },
    {
      "dynamics_loss_avg": 0.08140199594199657,
      "epoch": 1.621246337890625e-05,
      "step": 17000
    },
    {
      "action_loss_avg": 0.009648923436179757,
      "epoch": 1.621246337890625e-05,
      "step": 17000
    },
    {
      "epoch": 1.621246337890625e-05,
      "step": 17000,
      "training_step_time": 17.148372411727905
    },
    {
      "epoch": 1.6213417053222657e-05,
      "model_forward_time": 6.206869840621948,
      "step": 17001
    },
    {
      "epoch": 1.6213417053222657e-05,
      "step": 17001,
      "training_step_time": 17.73893976211548
    },
    {
      "epoch": 1.621437072753906e-05,
      "model_forward_time": 5.6740593910217285,
      "step": 17002
    },
    {
      "epoch": 1.621437072753906e-05,
      "step": 17002,
      "training_step_time": 17.203874826431274
    },
    {
      "epoch": 1.621532440185547e-05,
      "model_forward_time": 5.673355579376221,
      "step": 17003
    },
    {
      "epoch": 1.621532440185547e-05,
      "step": 17003,
      "training_step_time": 17.162911891937256
    },
    {
      "epoch": 1.6216278076171876e-05,
      "model_forward_time": 5.602005481719971,
      "step": 17004
    },
    {
      "epoch": 1.6216278076171876e-05,
      "step": 17004,
      "training_step_time": 17.05360198020935
    },
    {
      "epoch": 1.621723175048828e-05,
      "model_forward_time": 5.573426961898804,
      "step": 17005
    },
    {
      "epoch": 1.621723175048828e-05,
      "step": 17005,
      "training_step_time": 17.017438650131226
    },
    {
      "epoch": 1.6218185424804688e-05,
      "model_forward_time": 5.570700645446777,
      "step": 17006
    },
    {
      "epoch": 1.6218185424804688e-05,
      "step": 17006,
      "training_step_time": 16.98307156562805
    },
    {
      "epoch": 1.6219139099121095e-05,
      "model_forward_time": 5.985269784927368,
      "step": 17007
    },
    {
      "epoch": 1.6219139099121095e-05,
      "step": 17007,
      "training_step_time": 17.40639090538025
    },
    {
      "epoch": 1.62200927734375e-05,
      "model_forward_time": 5.599201679229736,
      "step": 17008
    },
    {
      "epoch": 1.62200927734375e-05,
      "step": 17008,
      "training_step_time": 17.05155110359192
    },
    {
      "epoch": 1.6221046447753907e-05,
      "model_forward_time": 5.583278179168701,
      "step": 17009
    },
    {
      "epoch": 1.6221046447753907e-05,
      "step": 17009,
      "training_step_time": 17.026806831359863
    },
    {
      "epoch": 1.6222000122070314e-05,
      "grad_norm": 0.46058031916618347,
      "learning_rate": 1e-05,
      "loss": 0.0827,
      "step": 17010
    },
    {
      "epoch": 1.6222000122070314e-05,
      "model_forward_time": 5.594229459762573,
      "step": 17010
    },
    {
      "dynamics_loss_avg": 0.06481392178684472,
      "epoch": 1.6222000122070314e-05,
      "step": 17010
    },
    {
      "action_loss_avg": 0.009826666209846735,
      "epoch": 1.6222000122070314e-05,
      "step": 17010
    },
    {
      "epoch": 1.6222000122070314e-05,
      "step": 17010,
      "training_step_time": 17.05984091758728
    },
    {
      "epoch": 1.6222953796386718e-05,
      "model_forward_time": 5.6036436557769775,
      "step": 17011
    },
    {
      "epoch": 1.6222953796386718e-05,
      "step": 17011,
      "training_step_time": 17.077255249023438
    },
    {
      "epoch": 1.6223907470703125e-05,
      "model_forward_time": 5.9443113803863525,
      "step": 17012
    },
    {
      "epoch": 1.6223907470703125e-05,
      "step": 17012,
      "training_step_time": 17.408087730407715
    },
    {
      "epoch": 1.6224861145019533e-05,
      "model_forward_time": 5.596752405166626,
      "step": 17013
    },
    {
      "epoch": 1.6224861145019533e-05,
      "step": 17013,
      "training_step_time": 17.058785438537598
    },
    {
      "epoch": 1.6225814819335937e-05,
      "model_forward_time": 5.594079971313477,
      "step": 17014
    },
    {
      "epoch": 1.6225814819335937e-05,
      "step": 17014,
      "training_step_time": 17.027320861816406
    },
    {
      "epoch": 1.6226768493652344e-05,
      "model_forward_time": 5.60400390625,
      "step": 17015
    },
    {
      "epoch": 1.6226768493652344e-05,
      "step": 17015,
      "training_step_time": 17.046356201171875
    },
    {
      "epoch": 1.622772216796875e-05,
      "model_forward_time": 5.59548282623291,
      "step": 17016
    },
    {
      "epoch": 1.622772216796875e-05,
      "step": 17016,
      "training_step_time": 17.062935829162598
    },
    {
      "epoch": 1.6228675842285156e-05,
      "model_forward_time": 5.6102070808410645,
      "step": 17017
    },
    {
      "epoch": 1.6228675842285156e-05,
      "step": 17017,
      "training_step_time": 17.443723917007446
    },
    {
      "epoch": 1.6229629516601563e-05,
      "model_forward_time": 5.614772319793701,
      "step": 17018
    },
    {
      "epoch": 1.6229629516601563e-05,
      "step": 17018,
      "training_step_time": 17.10499668121338
    },
    {
      "epoch": 1.6230583190917967e-05,
      "model_forward_time": 5.591671705245972,
      "step": 17019
    },
    {
      "epoch": 1.6230583190917967e-05,
      "step": 17019,
      "training_step_time": 17.022108554840088
    },
    {
      "epoch": 1.6231536865234375e-05,
      "grad_norm": 0.679376482963562,
      "learning_rate": 1e-05,
      "loss": 0.0742,
      "step": 17020
    },
    {
      "epoch": 1.6231536865234375e-05,
      "model_forward_time": 5.5697479248046875,
      "step": 17020
    },
    {
      "dynamics_loss_avg": 0.07079453207552433,
      "epoch": 1.6231536865234375e-05,
      "step": 17020
    },
    {
      "action_loss_avg": 0.009101480874232949,
      "epoch": 1.6231536865234375e-05,
      "step": 17020
    },
    {
      "epoch": 1.6231536865234375e-05,
      "step": 17020,
      "training_step_time": 17.004064321517944
    },
    {
      "epoch": 1.6232490539550782e-05,
      "model_forward_time": 5.5850584506988525,
      "step": 17021
    },
    {
      "epoch": 1.6232490539550782e-05,
      "step": 17021,
      "training_step_time": 17.01094126701355
    },
    {
      "epoch": 1.6233444213867186e-05,
      "model_forward_time": 5.5980589389801025,
      "step": 17022
    },
    {
      "epoch": 1.6233444213867186e-05,
      "step": 17022,
      "training_step_time": 17.009836673736572
    },
    {
      "epoch": 1.6234397888183593e-05,
      "model_forward_time": 5.94043755531311,
      "step": 17023
    },
    {
      "epoch": 1.6234397888183593e-05,
      "step": 17023,
      "training_step_time": 17.392083644866943
    },
    {
      "epoch": 1.62353515625e-05,
      "model_forward_time": 5.610626697540283,
      "step": 17024
    },
    {
      "epoch": 1.62353515625e-05,
      "step": 17024,
      "training_step_time": 17.0845148563385
    },
    {
      "epoch": 1.6236305236816405e-05,
      "model_forward_time": 5.619042158126831,
      "step": 17025
    },
    {
      "epoch": 1.6236305236816405e-05,
      "step": 17025,
      "training_step_time": 17.1328763961792
    },
    {
      "epoch": 1.6237258911132812e-05,
      "model_forward_time": 5.614732027053833,
      "step": 17026
    },
    {
      "epoch": 1.6237258911132812e-05,
      "step": 17026,
      "training_step_time": 17.128289937973022
    },
    {
      "epoch": 1.623821258544922e-05,
      "model_forward_time": 5.6320765018463135,
      "step": 17027
    },
    {
      "epoch": 1.623821258544922e-05,
      "step": 17027,
      "training_step_time": 17.129820108413696
    },
    {
      "epoch": 1.6239166259765624e-05,
      "model_forward_time": 5.994223594665527,
      "step": 17028
    },
    {
      "epoch": 1.6239166259765624e-05,
      "step": 17028,
      "training_step_time": 17.486076593399048
    },
    {
      "epoch": 1.624011993408203e-05,
      "model_forward_time": 5.609771013259888,
      "step": 17029
    },
    {
      "epoch": 1.624011993408203e-05,
      "step": 17029,
      "training_step_time": 17.107530117034912
    },
    {
      "epoch": 1.624107360839844e-05,
      "grad_norm": 0.6933956146240234,
      "learning_rate": 1e-05,
      "loss": 0.0631,
      "step": 17030
    },
    {
      "epoch": 1.624107360839844e-05,
      "model_forward_time": 5.617400884628296,
      "step": 17030
    },
    {
      "dynamics_loss_avg": 0.05958500150591135,
      "epoch": 1.624107360839844e-05,
      "step": 17030
    },
    {
      "action_loss_avg": 0.009469294198788702,
      "epoch": 1.624107360839844e-05,
      "step": 17030
    },
    {
      "epoch": 1.624107360839844e-05,
      "step": 17030,
      "training_step_time": 17.114548206329346
    },
    {
      "epoch": 1.6242027282714843e-05,
      "model_forward_time": 5.63014554977417,
      "step": 17031
    },
    {
      "epoch": 1.6242027282714843e-05,
      "step": 17031,
      "training_step_time": 17.11125659942627
    },
    {
      "epoch": 1.624298095703125e-05,
      "model_forward_time": 5.605196952819824,
      "step": 17032
    },
    {
      "epoch": 1.624298095703125e-05,
      "step": 17032,
      "training_step_time": 17.08033776283264
    },
    {
      "epoch": 1.6243934631347657e-05,
      "model_forward_time": 5.60103702545166,
      "step": 17033
    },
    {
      "epoch": 1.6243934631347657e-05,
      "step": 17033,
      "training_step_time": 17.06864094734192
    },
    {
      "epoch": 1.624488830566406e-05,
      "model_forward_time": 5.936059474945068,
      "step": 17034
    },
    {
      "epoch": 1.624488830566406e-05,
      "step": 17034,
      "training_step_time": 17.408464908599854
    },
    {
      "epoch": 1.624584197998047e-05,
      "model_forward_time": 5.612348318099976,
      "step": 17035
    },
    {
      "epoch": 1.624584197998047e-05,
      "step": 17035,
      "training_step_time": 17.089282274246216
    },
    {
      "epoch": 1.6246795654296876e-05,
      "model_forward_time": 5.6092209815979,
      "step": 17036
    },
    {
      "epoch": 1.6246795654296876e-05,
      "step": 17036,
      "training_step_time": 17.08875036239624
    },
    {
      "epoch": 1.624774932861328e-05,
      "model_forward_time": 5.618859529495239,
      "step": 17037
    },
    {
      "epoch": 1.624774932861328e-05,
      "step": 17037,
      "training_step_time": 17.108607053756714
    },
    {
      "epoch": 1.6248703002929688e-05,
      "model_forward_time": 5.606216192245483,
      "step": 17038
    },
    {
      "epoch": 1.6248703002929688e-05,
      "step": 17038,
      "training_step_time": 17.105436086654663
    },
    {
      "epoch": 1.6249656677246095e-05,
      "model_forward_time": 5.976691007614136,
      "step": 17039
    },
    {
      "epoch": 1.6249656677246095e-05,
      "step": 17039,
      "training_step_time": 17.439851999282837
    },
    {
      "epoch": 1.62506103515625e-05,
      "grad_norm": 0.43789833784103394,
      "learning_rate": 1e-05,
      "loss": 0.0637,
      "step": 17040
    },
    {
      "epoch": 1.62506103515625e-05,
      "model_forward_time": 5.583998441696167,
      "step": 17040
    },
    {
      "dynamics_loss_avg": 0.04766669236123562,
      "epoch": 1.62506103515625e-05,
      "step": 17040
    },
    {
      "action_loss_avg": 0.005556982487905771,
      "epoch": 1.62506103515625e-05,
      "step": 17040
    },
    {
      "epoch": 1.62506103515625e-05,
      "step": 17040,
      "training_step_time": 16.99808692932129
    },
    {
      "epoch": 1.6251564025878907e-05,
      "model_forward_time": 5.589459657669067,
      "step": 17041
    },
    {
      "epoch": 1.6251564025878907e-05,
      "step": 17041,
      "training_step_time": 16.989232301712036
    },
    {
      "epoch": 1.6252517700195314e-05,
      "model_forward_time": 5.573955774307251,
      "step": 17042
    },
    {
      "epoch": 1.6252517700195314e-05,
      "step": 17042,
      "training_step_time": 17.012105226516724
    },
    {
      "epoch": 1.6253471374511718e-05,
      "model_forward_time": 5.586514711380005,
      "step": 17043
    },
    {
      "epoch": 1.6253471374511718e-05,
      "step": 17043,
      "training_step_time": 17.020206689834595
    },
    {
      "epoch": 1.6254425048828125e-05,
      "model_forward_time": 5.5962910652160645,
      "step": 17044
    },
    {
      "epoch": 1.6254425048828125e-05,
      "step": 17044,
      "training_step_time": 17.414586782455444
    },
    {
      "epoch": 1.6255378723144533e-05,
      "model_forward_time": 5.613608360290527,
      "step": 17045
    },
    {
      "epoch": 1.6255378723144533e-05,
      "step": 17045,
      "training_step_time": 17.10673689842224
    },
    {
      "epoch": 1.6256332397460937e-05,
      "model_forward_time": 5.6092369556427,
      "step": 17046
    },
    {
      "epoch": 1.6256332397460937e-05,
      "step": 17046,
      "training_step_time": 17.108535528182983
    },
    {
      "epoch": 1.6257286071777344e-05,
      "model_forward_time": 5.61968469619751,
      "step": 17047
    },
    {
      "epoch": 1.6257286071777344e-05,
      "step": 17047,
      "training_step_time": 17.123390436172485
    },
    {
      "epoch": 1.625823974609375e-05,
      "model_forward_time": 5.625014781951904,
      "step": 17048
    },
    {
      "epoch": 1.625823974609375e-05,
      "step": 17048,
      "training_step_time": 17.107756853103638
    },
    {
      "epoch": 1.6259193420410156e-05,
      "model_forward_time": 5.594017744064331,
      "step": 17049
    },
    {
      "epoch": 1.6259193420410156e-05,
      "step": 17049,
      "training_step_time": 17.048084259033203
    },
    {
      "epoch": 1.6260147094726563e-05,
      "grad_norm": 0.6233298182487488,
      "learning_rate": 1e-05,
      "loss": 0.067,
      "step": 17050
    },
    {
      "epoch": 1.6260147094726563e-05,
      "model_forward_time": 5.949970245361328,
      "step": 17050
    },
    {
      "dynamics_loss_avg": 0.06055549494922161,
      "epoch": 1.6260147094726563e-05,
      "step": 17050
    },
    {
      "action_loss_avg": 0.007337454543448985,
      "epoch": 1.6260147094726563e-05,
      "step": 17050
    },
    {
      "epoch": 1.6260147094726563e-05,
      "step": 17050,
      "training_step_time": 17.38851237297058
    },
    {
      "epoch": 1.6261100769042967e-05,
      "model_forward_time": 5.579871416091919,
      "step": 17051
    },
    {
      "epoch": 1.6261100769042967e-05,
      "step": 17051,
      "training_step_time": 17.01534080505371
    },
    {
      "epoch": 1.6262054443359375e-05,
      "model_forward_time": 5.588300466537476,
      "step": 17052
    },
    {
      "epoch": 1.6262054443359375e-05,
      "step": 17052,
      "training_step_time": 17.033423900604248
    },
    {
      "epoch": 1.6263008117675782e-05,
      "model_forward_time": 5.609285831451416,
      "step": 17053
    },
    {
      "epoch": 1.6263008117675782e-05,
      "step": 17053,
      "training_step_time": 17.094514846801758
    },
    {
      "epoch": 1.6263961791992186e-05,
      "model_forward_time": 5.611628532409668,
      "step": 17054
    },
    {
      "epoch": 1.6263961791992186e-05,
      "step": 17054,
      "training_step_time": 17.092556476593018
    },
    {
      "epoch": 1.6264915466308593e-05,
      "model_forward_time": 5.9534690380096436,
      "step": 17055
    },
    {
      "epoch": 1.6264915466308593e-05,
      "step": 17055,
      "training_step_time": 17.429867267608643
    },
    {
      "epoch": 1.6265869140625e-05,
      "model_forward_time": 5.603452682495117,
      "step": 17056
    },
    {
      "epoch": 1.6265869140625e-05,
      "step": 17056,
      "training_step_time": 17.098127841949463
    },
    {
      "epoch": 1.6266822814941405e-05,
      "model_forward_time": 5.608454465866089,
      "step": 17057
    },
    {
      "epoch": 1.6266822814941405e-05,
      "step": 17057,
      "training_step_time": 17.106356143951416
    },
    {
      "epoch": 1.6267776489257812e-05,
      "model_forward_time": 5.62994909286499,
      "step": 17058
    },
    {
      "epoch": 1.6267776489257812e-05,
      "step": 17058,
      "training_step_time": 17.12281847000122
    },
    {
      "epoch": 1.626873016357422e-05,
      "model_forward_time": 5.6129150390625,
      "step": 17059
    },
    {
      "epoch": 1.626873016357422e-05,
      "step": 17059,
      "training_step_time": 17.110441207885742
    },
    {
      "epoch": 1.6269683837890624e-05,
      "grad_norm": 1.6472867727279663,
      "learning_rate": 1e-05,
      "loss": 0.0623,
      "step": 17060
    },
    {
      "epoch": 1.6269683837890624e-05,
      "model_forward_time": 5.613221645355225,
      "step": 17060
    },
    {
      "dynamics_loss_avg": 0.055671573616564274,
      "epoch": 1.6269683837890624e-05,
      "step": 17060
    },
    {
      "action_loss_avg": 0.007602247036993503,
      "epoch": 1.6269683837890624e-05,
      "step": 17060
    },
    {
      "epoch": 1.6269683837890624e-05,
      "step": 17060,
      "training_step_time": 17.114993810653687
    },
    {
      "epoch": 1.627063751220703e-05,
      "model_forward_time": 5.9853034019470215,
      "step": 17061
    },
    {
      "epoch": 1.627063751220703e-05,
      "step": 17061,
      "training_step_time": 17.475713968276978
    },
    {
      "epoch": 1.627159118652344e-05,
      "model_forward_time": 5.620325803756714,
      "step": 17062
    },
    {
      "epoch": 1.627159118652344e-05,
      "step": 17062,
      "training_step_time": 17.11130142211914
    },
    {
      "epoch": 1.6272544860839843e-05,
      "model_forward_time": 5.614198446273804,
      "step": 17063
    },
    {
      "epoch": 1.6272544860839843e-05,
      "step": 17063,
      "training_step_time": 17.106740474700928
    },
    {
      "epoch": 1.627349853515625e-05,
      "model_forward_time": 5.608905553817749,
      "step": 17064
    },
    {
      "epoch": 1.627349853515625e-05,
      "step": 17064,
      "training_step_time": 17.095059156417847
    },
    {
      "epoch": 1.6274452209472657e-05,
      "model_forward_time": 5.652888774871826,
      "step": 17065
    },
    {
      "epoch": 1.6274452209472657e-05,
      "step": 17065,
      "training_step_time": 17.14177703857422
    },
    {
      "epoch": 1.627540588378906e-05,
      "model_forward_time": 5.60706090927124,
      "step": 17066
    },
    {
      "epoch": 1.627540588378906e-05,
      "step": 17066,
      "training_step_time": 17.440547704696655
    },
    {
      "epoch": 1.627635955810547e-05,
      "model_forward_time": 5.608332395553589,
      "step": 17067
    },
    {
      "epoch": 1.627635955810547e-05,
      "step": 17067,
      "training_step_time": 17.100337505340576
    },
    {
      "epoch": 1.6277313232421876e-05,
      "model_forward_time": 5.611557960510254,
      "step": 17068
    },
    {
      "epoch": 1.6277313232421876e-05,
      "step": 17068,
      "training_step_time": 17.09463596343994
    },
    {
      "epoch": 1.627826690673828e-05,
      "model_forward_time": 5.605449676513672,
      "step": 17069
    },
    {
      "epoch": 1.627826690673828e-05,
      "step": 17069,
      "training_step_time": 17.08824110031128
    },
    {
      "epoch": 1.6279220581054688e-05,
      "grad_norm": 0.9131708741188049,
      "learning_rate": 1e-05,
      "loss": 0.0647,
      "step": 17070
    },
    {
      "epoch": 1.6279220581054688e-05,
      "model_forward_time": 5.605013370513916,
      "step": 17070
    },
    {
      "dynamics_loss_avg": 0.05580399315804243,
      "epoch": 1.6279220581054688e-05,
      "step": 17070
    },
    {
      "action_loss_avg": 0.005777103092987091,
      "epoch": 1.6279220581054688e-05,
      "step": 17070
    },
    {
      "epoch": 1.6279220581054688e-05,
      "step": 17070,
      "training_step_time": 17.066256999969482
    },
    {
      "epoch": 1.6280174255371095e-05,
      "model_forward_time": 5.598214149475098,
      "step": 17071
    },
    {
      "epoch": 1.6280174255371095e-05,
      "step": 17071,
      "training_step_time": 17.059321880340576
    },
    {
      "epoch": 1.62811279296875e-05,
      "model_forward_time": 5.9639716148376465,
      "step": 17072
    },
    {
      "epoch": 1.62811279296875e-05,
      "step": 17072,
      "training_step_time": 17.400395393371582
    },
    {
      "epoch": 1.6282081604003907e-05,
      "model_forward_time": 5.594304323196411,
      "step": 17073
    },
    {
      "epoch": 1.6282081604003907e-05,
      "step": 17073,
      "training_step_time": 17.037293434143066
    },
    {
      "epoch": 1.6283035278320314e-05,
      "model_forward_time": 5.594162464141846,
      "step": 17074
    },
    {
      "epoch": 1.6283035278320314e-05,
      "step": 17074,
      "training_step_time": 17.04326820373535
    },
    {
      "epoch": 1.6283988952636718e-05,
      "model_forward_time": 5.601520538330078,
      "step": 17075
    },
    {
      "epoch": 1.6283988952636718e-05,
      "step": 17075,
      "training_step_time": 17.08006525039673
    },
    {
      "epoch": 1.6284942626953125e-05,
      "model_forward_time": 5.61075234413147,
      "step": 17076
    },
    {
      "epoch": 1.6284942626953125e-05,
      "step": 17076,
      "training_step_time": 17.062552452087402
    },
    {
      "epoch": 1.6285896301269533e-05,
      "model_forward_time": 5.9492411613464355,
      "step": 17077
    },
    {
      "epoch": 1.6285896301269533e-05,
      "step": 17077,
      "training_step_time": 17.387993812561035
    },
    {
      "epoch": 1.6286849975585937e-05,
      "model_forward_time": 5.5958311557769775,
      "step": 17078
    },
    {
      "epoch": 1.6286849975585937e-05,
      "step": 17078,
      "training_step_time": 17.02003526687622
    },
    {
      "epoch": 1.6287803649902344e-05,
      "model_forward_time": 5.592235565185547,
      "step": 17079
    },
    {
      "epoch": 1.6287803649902344e-05,
      "step": 17079,
      "training_step_time": 17.020101070404053
    },
    {
      "epoch": 1.628875732421875e-05,
      "grad_norm": 1.200708270072937,
      "learning_rate": 1e-05,
      "loss": 0.0697,
      "step": 17080
    },
    {
      "epoch": 1.628875732421875e-05,
      "model_forward_time": 5.596524953842163,
      "step": 17080
    },
    {
      "dynamics_loss_avg": 0.06402148120105267,
      "epoch": 1.628875732421875e-05,
      "step": 17080
    },
    {
      "action_loss_avg": 0.007154773897491395,
      "epoch": 1.628875732421875e-05,
      "step": 17080
    },
    {
      "epoch": 1.628875732421875e-05,
      "step": 17080,
      "training_step_time": 17.059576272964478
    },
    {
      "epoch": 1.6289710998535156e-05,
      "model_forward_time": 5.599081516265869,
      "step": 17081
    },
    {
      "epoch": 1.6289710998535156e-05,
      "step": 17081,
      "training_step_time": 17.074016094207764
    },
    {
      "epoch": 1.6290664672851563e-05,
      "model_forward_time": 5.613651275634766,
      "step": 17082
    },
    {
      "epoch": 1.6290664672851563e-05,
      "step": 17082,
      "training_step_time": 17.12956690788269
    },
    {
      "epoch": 1.6291618347167967e-05,
      "model_forward_time": 5.9778642654418945,
      "step": 17083
    },
    {
      "epoch": 1.6291618347167967e-05,
      "step": 17083,
      "training_step_time": 17.488675832748413
    },
    {
      "epoch": 1.6292572021484375e-05,
      "model_forward_time": 5.6238555908203125,
      "step": 17084
    },
    {
      "epoch": 1.6292572021484375e-05,
      "step": 17084,
      "training_step_time": 17.121383666992188
    },
    {
      "epoch": 1.6293525695800782e-05,
      "model_forward_time": 5.617344379425049,
      "step": 17085
    },
    {
      "epoch": 1.6293525695800782e-05,
      "step": 17085,
      "training_step_time": 17.130303621292114
    },
    {
      "epoch": 1.6294479370117186e-05,
      "model_forward_time": 5.614004135131836,
      "step": 17086
    },
    {
      "epoch": 1.6294479370117186e-05,
      "step": 17086,
      "training_step_time": 17.05195140838623
    },
    {
      "epoch": 1.6295433044433594e-05,
      "model_forward_time": 5.583400726318359,
      "step": 17087
    },
    {
      "epoch": 1.6295433044433594e-05,
      "step": 17087,
      "training_step_time": 16.997360229492188
    },
    {
      "epoch": 1.629638671875e-05,
      "model_forward_time": 5.938634395599365,
      "step": 17088
    },
    {
      "epoch": 1.629638671875e-05,
      "step": 17088,
      "training_step_time": 17.365692615509033
    },
    {
      "epoch": 1.6297340393066405e-05,
      "model_forward_time": 5.561732530593872,
      "step": 17089
    },
    {
      "epoch": 1.6297340393066405e-05,
      "step": 17089,
      "training_step_time": 16.986613988876343
    },
    {
      "epoch": 1.6298294067382812e-05,
      "grad_norm": 0.444108247756958,
      "learning_rate": 1e-05,
      "loss": 0.0597,
      "step": 17090
    },
    {
      "epoch": 1.6298294067382812e-05,
      "model_forward_time": 5.568735361099243,
      "step": 17090
    },
    {
      "dynamics_loss_avg": 0.052367056161165236,
      "epoch": 1.6298294067382812e-05,
      "step": 17090
    },
    {
      "action_loss_avg": 0.006901201768778264,
      "epoch": 1.6298294067382812e-05,
      "step": 17090
    },
    {
      "epoch": 1.6298294067382812e-05,
      "step": 17090,
      "training_step_time": 16.985690355300903
    },
    {
      "epoch": 1.629924774169922e-05,
      "model_forward_time": 5.573302745819092,
      "step": 17091
    },
    {
      "epoch": 1.629924774169922e-05,
      "step": 17091,
      "training_step_time": 17.00407600402832
    },
    {
      "epoch": 1.6300201416015624e-05,
      "model_forward_time": 5.576167583465576,
      "step": 17092
    },
    {
      "epoch": 1.6300201416015624e-05,
      "step": 17092,
      "training_step_time": 17.0210919380188
    },
    {
      "epoch": 1.630115509033203e-05,
      "model_forward_time": 5.583967924118042,
      "step": 17093
    },
    {
      "epoch": 1.630115509033203e-05,
      "step": 17093,
      "training_step_time": 17.00673508644104
    },
    {
      "epoch": 1.630210876464844e-05,
      "model_forward_time": 5.952481508255005,
      "step": 17094
    },
    {
      "epoch": 1.630210876464844e-05,
      "step": 17094,
      "training_step_time": 17.444875240325928
    },
    {
      "epoch": 1.6303062438964843e-05,
      "model_forward_time": 5.6232218742370605,
      "step": 17095
    },
    {
      "epoch": 1.6303062438964843e-05,
      "step": 17095,
      "training_step_time": 17.137483835220337
    },
    {
      "epoch": 1.630401611328125e-05,
      "model_forward_time": 5.617974281311035,
      "step": 17096
    },
    {
      "epoch": 1.630401611328125e-05,
      "step": 17096,
      "training_step_time": 17.083187580108643
    },
    {
      "epoch": 1.6304969787597657e-05,
      "model_forward_time": 5.587479829788208,
      "step": 17097
    },
    {
      "epoch": 1.6304969787597657e-05,
      "step": 17097,
      "training_step_time": 16.99709725379944
    },
    {
      "epoch": 1.630592346191406e-05,
      "model_forward_time": 5.557170391082764,
      "step": 17098
    },
    {
      "epoch": 1.630592346191406e-05,
      "step": 17098,
      "training_step_time": 16.9625346660614
    },
    {
      "epoch": 1.630687713623047e-05,
      "model_forward_time": 5.565215110778809,
      "step": 17099
    },
    {
      "epoch": 1.630687713623047e-05,
      "step": 17099,
      "training_step_time": 17.33810520172119
    },
    {
      "epoch": 1.6307830810546876e-05,
      "grad_norm": 0.8321472406387329,
      "learning_rate": 1e-05,
      "loss": 0.0572,
      "step": 17100
    },
    {
      "epoch": 1.6307830810546876e-05,
      "model_forward_time": 5.5577287673950195,
      "step": 17100
    },
    {
      "dynamics_loss_avg": 0.048584737442433835,
      "epoch": 1.6307830810546876e-05,
      "step": 17100
    },
    {
      "action_loss_avg": 0.006929949950426817,
      "epoch": 1.6307830810546876e-05,
      "step": 17100
    },
    {
      "epoch": 1.6307830810546876e-05,
      "step": 17100,
      "training_step_time": 16.995344638824463
    },
    {
      "epoch": 1.630878448486328e-05,
      "model_forward_time": 5.566562652587891,
      "step": 17101
    },
    {
      "epoch": 1.630878448486328e-05,
      "step": 17101,
      "training_step_time": 16.98906922340393
    },
    {
      "epoch": 1.6309738159179688e-05,
      "model_forward_time": 5.572730779647827,
      "step": 17102
    },
    {
      "epoch": 1.6309738159179688e-05,
      "step": 17102,
      "training_step_time": 16.991658926010132
    },
    {
      "epoch": 1.6310691833496095e-05,
      "model_forward_time": 5.594027996063232,
      "step": 17103
    },
    {
      "epoch": 1.6310691833496095e-05,
      "step": 17103,
      "training_step_time": 17.04110026359558
    },
    {
      "epoch": 1.63116455078125e-05,
      "model_forward_time": 5.60076117515564,
      "step": 17104
    },
    {
      "epoch": 1.63116455078125e-05,
      "step": 17104,
      "training_step_time": 17.08598017692566
    },
    {
      "epoch": 1.6312599182128907e-05,
      "model_forward_time": 5.971699237823486,
      "step": 17105
    },
    {
      "epoch": 1.6312599182128907e-05,
      "step": 17105,
      "training_step_time": 17.455846071243286
    },
    {
      "epoch": 1.6313552856445314e-05,
      "model_forward_time": 5.610142230987549,
      "step": 17106
    },
    {
      "epoch": 1.6313552856445314e-05,
      "step": 17106,
      "training_step_time": 17.09141230583191
    },
    {
      "epoch": 1.6314506530761718e-05,
      "model_forward_time": 5.612411260604858,
      "step": 17107
    },
    {
      "epoch": 1.6314506530761718e-05,
      "step": 17107,
      "training_step_time": 17.041439056396484
    },
    {
      "epoch": 1.6315460205078126e-05,
      "model_forward_time": 5.603534460067749,
      "step": 17108
    },
    {
      "epoch": 1.6315460205078126e-05,
      "step": 17108,
      "training_step_time": 17.060579299926758
    },
    {
      "epoch": 1.6316413879394533e-05,
      "model_forward_time": 5.578372478485107,
      "step": 17109
    },
    {
      "epoch": 1.6316413879394533e-05,
      "step": 17109,
      "training_step_time": 17.002567768096924
    },
    {
      "epoch": 1.6317367553710937e-05,
      "grad_norm": 0.4606156349182129,
      "learning_rate": 1e-05,
      "loss": 0.0533,
      "step": 17110
    },
    {
      "epoch": 1.6317367553710937e-05,
      "model_forward_time": 5.927708864212036,
      "step": 17110
    },
    {
      "dynamics_loss_avg": 0.05007568057626486,
      "epoch": 1.6317367553710937e-05,
      "step": 17110
    },
    {
      "action_loss_avg": 0.0070399862946942445,
      "epoch": 1.6317367553710937e-05,
      "step": 17110
    },
    {
      "epoch": 1.6317367553710937e-05,
      "step": 17110,
      "training_step_time": 17.339389324188232
    },
    {
      "epoch": 1.6318321228027344e-05,
      "model_forward_time": 5.584648609161377,
      "step": 17111
    },
    {
      "epoch": 1.6318321228027344e-05,
      "step": 17111,
      "training_step_time": 17.02486276626587
    },
    {
      "epoch": 1.631927490234375e-05,
      "model_forward_time": 5.589321851730347,
      "step": 17112
    },
    {
      "epoch": 1.631927490234375e-05,
      "step": 17112,
      "training_step_time": 17.056498050689697
    },
    {
      "epoch": 1.6320228576660156e-05,
      "model_forward_time": 5.608174562454224,
      "step": 17113
    },
    {
      "epoch": 1.6320228576660156e-05,
      "step": 17113,
      "training_step_time": 17.103057146072388
    },
    {
      "epoch": 1.6321182250976563e-05,
      "model_forward_time": 5.612060308456421,
      "step": 17114
    },
    {
      "epoch": 1.6321182250976563e-05,
      "step": 17114,
      "training_step_time": 17.08919644355774
    },
    {
      "epoch": 1.6322135925292967e-05,
      "model_forward_time": 5.6109020709991455,
      "step": 17115
    },
    {
      "epoch": 1.6322135925292967e-05,
      "step": 17115,
      "training_step_time": 17.068993091583252
    },
    {
      "epoch": 1.6323089599609375e-05,
      "model_forward_time": 5.963346242904663,
      "step": 17116
    },
    {
      "epoch": 1.6323089599609375e-05,
      "step": 17116,
      "training_step_time": 17.404273986816406
    },
    {
      "epoch": 1.6324043273925782e-05,
      "model_forward_time": 5.590825319290161,
      "step": 17117
    },
    {
      "epoch": 1.6324043273925782e-05,
      "step": 17117,
      "training_step_time": 17.051632404327393
    },
    {
      "epoch": 1.6324996948242186e-05,
      "model_forward_time": 5.60895848274231,
      "step": 17118
    },
    {
      "epoch": 1.6324996948242186e-05,
      "step": 17118,
      "training_step_time": 17.096171140670776
    },
    {
      "epoch": 1.6325950622558594e-05,
      "model_forward_time": 5.619137763977051,
      "step": 17119
    },
    {
      "epoch": 1.6325950622558594e-05,
      "step": 17119,
      "training_step_time": 17.122538089752197
    },
    {
      "epoch": 1.6326904296875e-05,
      "grad_norm": 0.6407698392868042,
      "learning_rate": 1e-05,
      "loss": 0.0605,
      "step": 17120
    },
    {
      "epoch": 1.6326904296875e-05,
      "model_forward_time": 5.624600648880005,
      "step": 17120
    },
    {
      "dynamics_loss_avg": 0.054786070622503756,
      "epoch": 1.6326904296875e-05,
      "step": 17120
    },
    {
      "action_loss_avg": 0.006620552809908986,
      "epoch": 1.6326904296875e-05,
      "step": 17120
    },
    {
      "epoch": 1.6326904296875e-05,
      "step": 17120,
      "training_step_time": 17.117289304733276
    },
    {
      "epoch": 1.6327857971191405e-05,
      "model_forward_time": 5.96526837348938,
      "step": 17121
    },
    {
      "epoch": 1.6327857971191405e-05,
      "step": 17121,
      "training_step_time": 17.45551824569702
    },
    {
      "epoch": 1.6328811645507812e-05,
      "model_forward_time": 5.608732461929321,
      "step": 17122
    },
    {
      "epoch": 1.6328811645507812e-05,
      "step": 17122,
      "training_step_time": 17.105077266693115
    },
    {
      "epoch": 1.632976531982422e-05,
      "model_forward_time": 5.617326498031616,
      "step": 17123
    },
    {
      "epoch": 1.632976531982422e-05,
      "step": 17123,
      "training_step_time": 17.099584579467773
    },
    {
      "epoch": 1.6330718994140624e-05,
      "model_forward_time": 5.611291885375977,
      "step": 17124
    },
    {
      "epoch": 1.6330718994140624e-05,
      "step": 17124,
      "training_step_time": 17.069767713546753
    },
    {
      "epoch": 1.633167266845703e-05,
      "model_forward_time": 5.599329233169556,
      "step": 17125
    },
    {
      "epoch": 1.633167266845703e-05,
      "step": 17125,
      "training_step_time": 17.06092596054077
    },
    {
      "epoch": 1.633262634277344e-05,
      "model_forward_time": 5.598457098007202,
      "step": 17126
    },
    {
      "epoch": 1.633262634277344e-05,
      "step": 17126,
      "training_step_time": 17.085378646850586
    },
    {
      "epoch": 1.6333580017089843e-05,
      "model_forward_time": 5.986480236053467,
      "step": 17127
    },
    {
      "epoch": 1.6333580017089843e-05,
      "step": 17127,
      "training_step_time": 17.49948263168335
    },
    {
      "epoch": 1.633453369140625e-05,
      "model_forward_time": 5.627194404602051,
      "step": 17128
    },
    {
      "epoch": 1.633453369140625e-05,
      "step": 17128,
      "training_step_time": 17.153788328170776
    },
    {
      "epoch": 1.6335487365722658e-05,
      "model_forward_time": 5.632227420806885,
      "step": 17129
    },
    {
      "epoch": 1.6335487365722658e-05,
      "step": 17129,
      "training_step_time": 17.144282817840576
    },
    {
      "epoch": 1.633644104003906e-05,
      "grad_norm": 0.5215113759040833,
      "learning_rate": 1e-05,
      "loss": 0.0632,
      "step": 17130
    },
    {
      "epoch": 1.633644104003906e-05,
      "model_forward_time": 5.621696710586548,
      "step": 17130
    },
    {
      "dynamics_loss_avg": 0.052190444432199,
      "epoch": 1.633644104003906e-05,
      "step": 17130
    },
    {
      "action_loss_avg": 0.007525046868249774,
      "epoch": 1.633644104003906e-05,
      "step": 17130
    },
    {
      "epoch": 1.633644104003906e-05,
      "step": 17130,
      "training_step_time": 17.134673595428467
    },
    {
      "epoch": 1.633739471435547e-05,
      "model_forward_time": 5.623055934906006,
      "step": 17131
    },
    {
      "epoch": 1.633739471435547e-05,
      "step": 17131,
      "training_step_time": 17.122123956680298
    },
    {
      "epoch": 1.6338348388671876e-05,
      "model_forward_time": 5.617542266845703,
      "step": 17132
    },
    {
      "epoch": 1.6338348388671876e-05,
      "step": 17132,
      "training_step_time": 17.486789226531982
    },
    {
      "epoch": 1.633930206298828e-05,
      "model_forward_time": 5.611008405685425,
      "step": 17133
    },
    {
      "epoch": 1.633930206298828e-05,
      "step": 17133,
      "training_step_time": 17.096622467041016
    },
    {
      "epoch": 1.6340255737304688e-05,
      "model_forward_time": 5.610350131988525,
      "step": 17134
    },
    {
      "epoch": 1.6340255737304688e-05,
      "step": 17134,
      "training_step_time": 17.086369276046753
    },
    {
      "epoch": 1.6341209411621095e-05,
      "model_forward_time": 5.605683326721191,
      "step": 17135
    },
    {
      "epoch": 1.6341209411621095e-05,
      "step": 17135,
      "training_step_time": 17.097532749176025
    },
    {
      "epoch": 1.63421630859375e-05,
      "model_forward_time": 5.610376596450806,
      "step": 17136
    },
    {
      "epoch": 1.63421630859375e-05,
      "step": 17136,
      "training_step_time": 17.089815616607666
    },
    {
      "epoch": 1.6343116760253907e-05,
      "model_forward_time": 5.616074562072754,
      "step": 17137
    },
    {
      "epoch": 1.6343116760253907e-05,
      "step": 17137,
      "training_step_time": 17.09376835823059
    },
    {
      "epoch": 1.6344070434570314e-05,
      "model_forward_time": 5.98175573348999,
      "step": 17138
    },
    {
      "epoch": 1.6344070434570314e-05,
      "step": 17138,
      "training_step_time": 17.461777448654175
    },
    {
      "epoch": 1.6345024108886718e-05,
      "model_forward_time": 5.608141660690308,
      "step": 17139
    },
    {
      "epoch": 1.6345024108886718e-05,
      "step": 17139,
      "training_step_time": 17.089031219482422
    },
    {
      "epoch": 1.6345977783203126e-05,
      "grad_norm": 0.5955199003219604,
      "learning_rate": 1e-05,
      "loss": 0.058,
      "step": 17140
    },
    {
      "epoch": 1.6345977783203126e-05,
      "model_forward_time": 5.6085944175720215,
      "step": 17140
    },
    {
      "dynamics_loss_avg": 0.05255811610259116,
      "epoch": 1.6345977783203126e-05,
      "step": 17140
    },
    {
      "action_loss_avg": 0.006693149509374052,
      "epoch": 1.6345977783203126e-05,
      "step": 17140
    },
    {
      "epoch": 1.6345977783203126e-05,
      "step": 17140,
      "training_step_time": 17.098220109939575
    },
    {
      "epoch": 1.634693145751953e-05,
      "model_forward_time": 5.613078832626343,
      "step": 17141
    },
    {
      "epoch": 1.634693145751953e-05,
      "step": 17141,
      "training_step_time": 17.09056329727173
    },
    {
      "epoch": 1.6347885131835937e-05,
      "model_forward_time": 5.6049041748046875,
      "step": 17142
    },
    {
      "epoch": 1.6347885131835937e-05,
      "step": 17142,
      "training_step_time": 17.07546830177307
    },
    {
      "epoch": 1.6348838806152344e-05,
      "model_forward_time": 5.9515955448150635,
      "step": 17143
    },
    {
      "epoch": 1.6348838806152344e-05,
      "step": 17143,
      "training_step_time": 17.427366495132446
    },
    {
      "epoch": 1.634979248046875e-05,
      "model_forward_time": 5.603367805480957,
      "step": 17144
    },
    {
      "epoch": 1.634979248046875e-05,
      "step": 17144,
      "training_step_time": 17.073235034942627
    },
    {
      "epoch": 1.6350746154785156e-05,
      "model_forward_time": 5.601384401321411,
      "step": 17145
    },
    {
      "epoch": 1.6350746154785156e-05,
      "step": 17145,
      "training_step_time": 17.035306930541992
    },
    {
      "epoch": 1.6351699829101563e-05,
      "model_forward_time": 5.588622808456421,
      "step": 17146
    },
    {
      "epoch": 1.6351699829101563e-05,
      "step": 17146,
      "training_step_time": 17.016042470932007
    },
    {
      "epoch": 1.6352653503417967e-05,
      "model_forward_time": 5.587819337844849,
      "step": 17147
    },
    {
      "epoch": 1.6352653503417967e-05,
      "step": 17147,
      "training_step_time": 17.02448081970215
    },
    {
      "epoch": 1.6353607177734375e-05,
      "model_forward_time": 5.594465494155884,
      "step": 17148
    },
    {
      "epoch": 1.6353607177734375e-05,
      "step": 17148,
      "training_step_time": 17.078895807266235
    },
    {
      "epoch": 1.6354560852050782e-05,
      "model_forward_time": 5.9808030128479,
      "step": 17149
    },
    {
      "epoch": 1.6354560852050782e-05,
      "step": 17149,
      "training_step_time": 17.486073970794678
    },
    {
      "epoch": 1.6355514526367186e-05,
      "grad_norm": 0.6411664485931396,
      "learning_rate": 1e-05,
      "loss": 0.0638,
      "step": 17150
    },
    {
      "epoch": 1.6355514526367186e-05,
      "model_forward_time": 5.624750137329102,
      "step": 17150
    },
    {
      "dynamics_loss_avg": 0.060613111779093745,
      "epoch": 1.6355514526367186e-05,
      "step": 17150
    },
    {
      "action_loss_avg": 0.008810977777466178,
      "epoch": 1.6355514526367186e-05,
      "step": 17150
    },
    {
      "epoch": 1.6355514526367186e-05,
      "step": 17150,
      "training_step_time": 17.129725217819214
    },
    {
      "epoch": 1.6356468200683594e-05,
      "model_forward_time": 5.622231721878052,
      "step": 17151
    },
    {
      "epoch": 1.6356468200683594e-05,
      "step": 17151,
      "training_step_time": 17.11171007156372
    },
    {
      "epoch": 1.6357421875e-05,
      "model_forward_time": 5.602347373962402,
      "step": 17152
    },
    {
      "epoch": 1.6357421875e-05,
      "step": 17152,
      "training_step_time": 17.085604429244995
    },
    {
      "epoch": 1.6358375549316405e-05,
      "model_forward_time": 5.607835531234741,
      "step": 17153
    },
    {
      "epoch": 1.6358375549316405e-05,
      "step": 17153,
      "training_step_time": 17.073771238327026
    },
    {
      "epoch": 1.6359329223632812e-05,
      "model_forward_time": 5.960262298583984,
      "step": 17154
    },
    {
      "epoch": 1.6359329223632812e-05,
      "step": 17154,
      "training_step_time": 17.44174027442932
    },
    {
      "epoch": 1.636028289794922e-05,
      "model_forward_time": 5.614278316497803,
      "step": 17155
    },
    {
      "epoch": 1.636028289794922e-05,
      "step": 17155,
      "training_step_time": 17.09464693069458
    },
    {
      "epoch": 1.6361236572265624e-05,
      "model_forward_time": 4.281542062759399,
      "step": 17156
    },
    {
      "epoch": 1.6361236572265624e-05,
      "step": 17156,
      "training_step_time": 12.80478286743164
    },
    {
      "epoch": 1.636219024658203e-05,
      "model_forward_time": 5.611962080001831,
      "step": 17157
    },
    {
      "epoch": 1.636219024658203e-05,
      "step": 17157,
      "training_step_time": 17.089752435684204
    },
    {
      "epoch": 1.636314392089844e-05,
      "model_forward_time": 5.606198072433472,
      "step": 17158
    },
    {
      "epoch": 1.636314392089844e-05,
      "step": 17158,
      "training_step_time": 17.079545497894287
    },
    {
      "epoch": 1.6364097595214843e-05,
      "model_forward_time": 5.6031787395477295,
      "step": 17159
    },
    {
      "epoch": 1.6364097595214843e-05,
      "step": 17159,
      "training_step_time": 17.070306062698364
    },
    {
      "epoch": 1.636505126953125e-05,
      "grad_norm": 0.26806458830833435,
      "learning_rate": 1e-05,
      "loss": 0.0752,
      "step": 17160
    },
    {
      "epoch": 1.636505126953125e-05,
      "model_forward_time": 5.974402904510498,
      "step": 17160
    },
    {
      "dynamics_loss_avg": 0.0587849635630846,
      "epoch": 1.636505126953125e-05,
      "step": 17160
    },
    {
      "action_loss_avg": 0.008677196293137967,
      "epoch": 1.636505126953125e-05,
      "step": 17160
    },
    {
      "epoch": 1.636505126953125e-05,
      "step": 17160,
      "training_step_time": 17.414910316467285
    },
    {
      "epoch": 1.6366004943847658e-05,
      "model_forward_time": 5.606861352920532,
      "step": 17161
    },
    {
      "epoch": 1.6366004943847658e-05,
      "step": 17161,
      "training_step_time": 17.09582257270813
    },
    {
      "epoch": 1.636695861816406e-05,
      "model_forward_time": 5.628474235534668,
      "step": 17162
    },
    {
      "epoch": 1.636695861816406e-05,
      "step": 17162,
      "training_step_time": 17.144084930419922
    },
    {
      "epoch": 1.636791229248047e-05,
      "model_forward_time": 5.62755823135376,
      "step": 17163
    },
    {
      "epoch": 1.636791229248047e-05,
      "step": 17163,
      "training_step_time": 17.116206407546997
    },
    {
      "epoch": 1.6368865966796876e-05,
      "model_forward_time": 5.593421697616577,
      "step": 17164
    },
    {
      "epoch": 1.6368865966796876e-05,
      "step": 17164,
      "training_step_time": 17.024350881576538
    },
    {
      "epoch": 1.636981964111328e-05,
      "model_forward_time": 5.577581882476807,
      "step": 17165
    },
    {
      "epoch": 1.636981964111328e-05,
      "step": 17165,
      "training_step_time": 17.017656564712524
    },
    {
      "epoch": 1.6370773315429688e-05,
      "model_forward_time": 5.922927379608154,
      "step": 17166
    },
    {
      "epoch": 1.6370773315429688e-05,
      "step": 17166,
      "training_step_time": 17.32639193534851
    },
    {
      "epoch": 1.6371726989746095e-05,
      "model_forward_time": 5.5785651206970215,
      "step": 17167
    },
    {
      "epoch": 1.6371726989746095e-05,
      "step": 17167,
      "training_step_time": 16.985902786254883
    },
    {
      "epoch": 1.63726806640625e-05,
      "model_forward_time": 5.571712017059326,
      "step": 17168
    },
    {
      "epoch": 1.63726806640625e-05,
      "step": 17168,
      "training_step_time": 17.004962921142578
    },
    {
      "epoch": 1.6373634338378907e-05,
      "model_forward_time": 5.575436115264893,
      "step": 17169
    },
    {
      "epoch": 1.6373634338378907e-05,
      "step": 17169,
      "training_step_time": 17.01771092414856
    },
    {
      "epoch": 1.6374588012695314e-05,
      "grad_norm": 0.6526834964752197,
      "learning_rate": 1e-05,
      "loss": 0.0575,
      "step": 17170
    },
    {
      "epoch": 1.6374588012695314e-05,
      "model_forward_time": 5.586166620254517,
      "step": 17170
    },
    {
      "dynamics_loss_avg": 0.05376209001988173,
      "epoch": 1.6374588012695314e-05,
      "step": 17170
    },
    {
      "action_loss_avg": 0.0069011036772280935,
      "epoch": 1.6374588012695314e-05,
      "step": 17170
    },
    {
      "epoch": 1.6374588012695314e-05,
      "step": 17170,
      "training_step_time": 17.027788639068604
    },
    {
      "epoch": 1.6375541687011718e-05,
      "model_forward_time": 5.957556486129761,
      "step": 17171
    },
    {
      "epoch": 1.6375541687011718e-05,
      "step": 17171,
      "training_step_time": 17.385054111480713
    },
    {
      "epoch": 1.6376495361328126e-05,
      "model_forward_time": 5.598859786987305,
      "step": 17172
    },
    {
      "epoch": 1.6376495361328126e-05,
      "step": 17172,
      "training_step_time": 17.041542053222656
    },
    {
      "epoch": 1.637744903564453e-05,
      "model_forward_time": 5.606404066085815,
      "step": 17173
    },
    {
      "epoch": 1.637744903564453e-05,
      "step": 17173,
      "training_step_time": 17.098220109939575
    },
    {
      "epoch": 1.6378402709960937e-05,
      "model_forward_time": 5.617507219314575,
      "step": 17174
    },
    {
      "epoch": 1.6378402709960937e-05,
      "step": 17174,
      "training_step_time": 17.13892889022827
    },
    {
      "epoch": 1.6379356384277344e-05,
      "model_forward_time": 5.6222639083862305,
      "step": 17175
    },
    {
      "epoch": 1.6379356384277344e-05,
      "step": 17175,
      "training_step_time": 17.14064383506775
    },
    {
      "epoch": 1.638031005859375e-05,
      "model_forward_time": 5.964895248413086,
      "step": 17176
    },
    {
      "epoch": 1.638031005859375e-05,
      "step": 17176,
      "training_step_time": 17.46171259880066
    },
    {
      "epoch": 1.6381263732910156e-05,
      "model_forward_time": 5.617461681365967,
      "step": 17177
    },
    {
      "epoch": 1.6381263732910156e-05,
      "step": 17177,
      "training_step_time": 17.117775201797485
    },
    {
      "epoch": 1.6382217407226563e-05,
      "model_forward_time": 5.610857725143433,
      "step": 17178
    },
    {
      "epoch": 1.6382217407226563e-05,
      "step": 17178,
      "training_step_time": 17.115423440933228
    },
    {
      "epoch": 1.6383171081542967e-05,
      "model_forward_time": 5.621944904327393,
      "step": 17179
    },
    {
      "epoch": 1.6383171081542967e-05,
      "step": 17179,
      "training_step_time": 17.12688422203064
    },
    {
      "epoch": 1.6384124755859375e-05,
      "grad_norm": 0.48248639702796936,
      "learning_rate": 1e-05,
      "loss": 0.0642,
      "step": 17180
    },
    {
      "epoch": 1.6384124755859375e-05,
      "model_forward_time": 5.622856140136719,
      "step": 17180
    },
    {
      "dynamics_loss_avg": 0.05872237626463175,
      "epoch": 1.6384124755859375e-05,
      "step": 17180
    },
    {
      "action_loss_avg": 0.005799782718531788,
      "epoch": 1.6384124755859375e-05,
      "step": 17180
    },
    {
      "epoch": 1.6384124755859375e-05,
      "step": 17180,
      "training_step_time": 17.12707829475403
    },
    {
      "epoch": 1.6385078430175782e-05,
      "model_forward_time": 5.613245487213135,
      "step": 17181
    },
    {
      "epoch": 1.6385078430175782e-05,
      "step": 17181,
      "training_step_time": 17.11272382736206
    },
    {
      "epoch": 1.6386032104492186e-05,
      "model_forward_time": 5.968230247497559,
      "step": 17182
    },
    {
      "epoch": 1.6386032104492186e-05,
      "step": 17182,
      "training_step_time": 17.449471712112427
    },
    {
      "epoch": 1.6386985778808594e-05,
      "model_forward_time": 5.609196662902832,
      "step": 17183
    },
    {
      "epoch": 1.6386985778808594e-05,
      "step": 17183,
      "training_step_time": 17.09990167617798
    },
    {
      "epoch": 1.6387939453125e-05,
      "model_forward_time": 5.617514610290527,
      "step": 17184
    },
    {
      "epoch": 1.6387939453125e-05,
      "step": 17184,
      "training_step_time": 17.119323015213013
    },
    {
      "epoch": 1.6388893127441405e-05,
      "model_forward_time": 5.615645885467529,
      "step": 17185
    },
    {
      "epoch": 1.6388893127441405e-05,
      "step": 17185,
      "training_step_time": 17.124746799468994
    },
    {
      "epoch": 1.6389846801757812e-05,
      "model_forward_time": 5.611373424530029,
      "step": 17186
    },
    {
      "epoch": 1.6389846801757812e-05,
      "step": 17186,
      "training_step_time": 17.07448935508728
    },
    {
      "epoch": 1.639080047607422e-05,
      "model_forward_time": 5.966122388839722,
      "step": 17187
    },
    {
      "epoch": 1.639080047607422e-05,
      "step": 17187,
      "training_step_time": 17.405228853225708
    },
    {
      "epoch": 1.6391754150390624e-05,
      "model_forward_time": 5.575309753417969,
      "step": 17188
    },
    {
      "epoch": 1.6391754150390624e-05,
      "step": 17188,
      "training_step_time": 16.99451756477356
    },
    {
      "epoch": 1.639270782470703e-05,
      "model_forward_time": 5.572895288467407,
      "step": 17189
    },
    {
      "epoch": 1.639270782470703e-05,
      "step": 17189,
      "training_step_time": 16.972280502319336
    },
    {
      "epoch": 1.639366149902344e-05,
      "grad_norm": 0.34595218300819397,
      "learning_rate": 1e-05,
      "loss": 0.0506,
      "step": 17190
    },
    {
      "epoch": 1.639366149902344e-05,
      "model_forward_time": 5.572563409805298,
      "step": 17190
    },
    {
      "dynamics_loss_avg": 0.046474283561110497,
      "epoch": 1.639366149902344e-05,
      "step": 17190
    },
    {
      "action_loss_avg": 0.005767782393377274,
      "epoch": 1.639366149902344e-05,
      "step": 17190
    },
    {
      "epoch": 1.639366149902344e-05,
      "step": 17190,
      "training_step_time": 16.98036479949951
    },
    {
      "epoch": 1.6394615173339843e-05,
      "model_forward_time": 5.582936763763428,
      "step": 17191
    },
    {
      "epoch": 1.6394615173339843e-05,
      "step": 17191,
      "training_step_time": 17.021523237228394
    },
    {
      "epoch": 1.639556884765625e-05,
      "model_forward_time": 5.575498342514038,
      "step": 17192
    },
    {
      "epoch": 1.639556884765625e-05,
      "step": 17192,
      "training_step_time": 17.384591579437256
    },
    {
      "epoch": 1.6396522521972658e-05,
      "model_forward_time": 5.5926878452301025,
      "step": 17193
    },
    {
      "epoch": 1.6396522521972658e-05,
      "step": 17193,
      "training_step_time": 17.029228687286377
    },
    {
      "epoch": 1.639747619628906e-05,
      "model_forward_time": 5.5895140171051025,
      "step": 17194
    },
    {
      "epoch": 1.639747619628906e-05,
      "step": 17194,
      "training_step_time": 17.02547526359558
    },
    {
      "epoch": 1.639842987060547e-05,
      "model_forward_time": 5.602851390838623,
      "step": 17195
    },
    {
      "epoch": 1.639842987060547e-05,
      "step": 17195,
      "training_step_time": 17.046024322509766
    },
    {
      "epoch": 1.6399383544921876e-05,
      "model_forward_time": 5.607500791549683,
      "step": 17196
    },
    {
      "epoch": 1.6399383544921876e-05,
      "step": 17196,
      "training_step_time": 17.07184886932373
    },
    {
      "epoch": 1.640033721923828e-05,
      "model_forward_time": 5.629287004470825,
      "step": 17197
    },
    {
      "epoch": 1.640033721923828e-05,
      "step": 17197,
      "training_step_time": 17.11159658432007
    },
    {
      "epoch": 1.6401290893554688e-05,
      "model_forward_time": 5.938351154327393,
      "step": 17198
    },
    {
      "epoch": 1.6401290893554688e-05,
      "step": 17198,
      "training_step_time": 17.400266647338867
    },
    {
      "epoch": 1.6402244567871095e-05,
      "model_forward_time": 5.595734119415283,
      "step": 17199
    },
    {
      "epoch": 1.6402244567871095e-05,
      "step": 17199,
      "training_step_time": 17.036859035491943
    },
    {
      "epoch": 1.64031982421875e-05,
      "grad_norm": 1.0966085195541382,
      "learning_rate": 1e-05,
      "loss": 0.0772,
      "step": 17200
    },
    {
      "epoch": 1.64031982421875e-05,
      "model_forward_time": 5.5980987548828125,
      "step": 17200
    },
    {
      "dynamics_loss_avg": 0.07042775563895702,
      "epoch": 1.64031982421875e-05,
      "step": 17200
    },
    {
      "action_loss_avg": 0.007213346706703305,
      "epoch": 1.64031982421875e-05,
      "step": 17200
    },
    {
      "epoch": 1.64031982421875e-05,
      "step": 17200,
      "training_step_time": 17.051759481430054
    },
    {
      "epoch": 1.6404151916503907e-05,
      "model_forward_time": 5.5981950759887695,
      "step": 17201
    },
    {
      "epoch": 1.6404151916503907e-05,
      "step": 17201,
      "training_step_time": 17.07077670097351
    },
    {
      "epoch": 1.6405105590820314e-05,
      "model_forward_time": 5.655154466629028,
      "step": 17202
    },
    {
      "epoch": 1.6405105590820314e-05,
      "step": 17202,
      "training_step_time": 17.1106858253479
    },
    {
      "epoch": 1.6406059265136718e-05,
      "model_forward_time": 5.992072820663452,
      "step": 17203
    },
    {
      "epoch": 1.6406059265136718e-05,
      "step": 17203,
      "training_step_time": 17.453457355499268
    },
    {
      "epoch": 1.6407012939453126e-05,
      "model_forward_time": 5.6048264503479,
      "step": 17204
    },
    {
      "epoch": 1.6407012939453126e-05,
      "step": 17204,
      "training_step_time": 17.069835901260376
    },
    {
      "epoch": 1.640796661376953e-05,
      "model_forward_time": 5.600424528121948,
      "step": 17205
    },
    {
      "epoch": 1.640796661376953e-05,
      "step": 17205,
      "training_step_time": 17.061968326568604
    },
    {
      "epoch": 1.6408920288085937e-05,
      "model_forward_time": 5.598117113113403,
      "step": 17206
    },
    {
      "epoch": 1.6408920288085937e-05,
      "step": 17206,
      "training_step_time": 17.033169507980347
    },
    {
      "epoch": 1.6409873962402344e-05,
      "model_forward_time": 5.59114933013916,
      "step": 17207
    },
    {
      "epoch": 1.6409873962402344e-05,
      "step": 17207,
      "training_step_time": 17.033680200576782
    },
    {
      "epoch": 1.641082763671875e-05,
      "model_forward_time": 5.599641799926758,
      "step": 17208
    },
    {
      "epoch": 1.641082763671875e-05,
      "step": 17208,
      "training_step_time": 17.071685552597046
    },
    {
      "epoch": 1.6411781311035156e-05,
      "model_forward_time": 5.973018169403076,
      "step": 17209
    },
    {
      "epoch": 1.6411781311035156e-05,
      "step": 17209,
      "training_step_time": 17.459693908691406
    },
    {
      "epoch": 1.6412734985351563e-05,
      "grad_norm": 0.5344746112823486,
      "learning_rate": 1e-05,
      "loss": 0.0803,
      "step": 17210
    },
    {
      "epoch": 1.6412734985351563e-05,
      "model_forward_time": 5.617931127548218,
      "step": 17210
    },
    {
      "dynamics_loss_avg": 0.06859823875129223,
      "epoch": 1.6412734985351563e-05,
      "step": 17210
    },
    {
      "action_loss_avg": 0.0076539185363799335,
      "epoch": 1.6412734985351563e-05,
      "step": 17210
    },
    {
      "epoch": 1.6412734985351563e-05,
      "step": 17210,
      "training_step_time": 17.116149425506592
    },
    {
      "epoch": 1.6413688659667967e-05,
      "model_forward_time": 5.618096828460693,
      "step": 17211
    },
    {
      "epoch": 1.6413688659667967e-05,
      "step": 17211,
      "training_step_time": 17.083845376968384
    },
    {
      "epoch": 1.6414642333984375e-05,
      "model_forward_time": 5.593416690826416,
      "step": 17212
    },
    {
      "epoch": 1.6414642333984375e-05,
      "step": 17212,
      "training_step_time": 17.01012659072876
    },
    {
      "epoch": 1.6415596008300782e-05,
      "model_forward_time": 5.58193039894104,
      "step": 17213
    },
    {
      "epoch": 1.6415596008300782e-05,
      "step": 17213,
      "training_step_time": 16.996483087539673
    },
    {
      "epoch": 1.6416549682617186e-05,
      "model_forward_time": 5.9622886180877686,
      "step": 17214
    },
    {
      "epoch": 1.6416549682617186e-05,
      "step": 17214,
      "training_step_time": 17.39992117881775
    },
    {
      "epoch": 1.6417503356933594e-05,
      "model_forward_time": 5.581362962722778,
      "step": 17215
    },
    {
      "epoch": 1.6417503356933594e-05,
      "step": 17215,
      "training_step_time": 16.999045610427856
    },
    {
      "epoch": 1.641845703125e-05,
      "model_forward_time": 5.592335939407349,
      "step": 17216
    },
    {
      "epoch": 1.641845703125e-05,
      "step": 17216,
      "training_step_time": 17.011110067367554
    },
    {
      "epoch": 1.6419410705566405e-05,
      "model_forward_time": 5.606438875198364,
      "step": 17217
    },
    {
      "epoch": 1.6419410705566405e-05,
      "step": 17217,
      "training_step_time": 17.076283931732178
    },
    {
      "epoch": 1.6420364379882812e-05,
      "model_forward_time": 5.617923736572266,
      "step": 17218
    },
    {
      "epoch": 1.6420364379882812e-05,
      "step": 17218,
      "training_step_time": 17.098430633544922
    },
    {
      "epoch": 1.642131805419922e-05,
      "model_forward_time": 5.629416465759277,
      "step": 17219
    },
    {
      "epoch": 1.642131805419922e-05,
      "step": 17219,
      "training_step_time": 17.1112220287323
    },
    {
      "epoch": 1.6422271728515624e-05,
      "grad_norm": 0.9699080586433411,
      "learning_rate": 1e-05,
      "loss": 0.082,
      "step": 17220
    },
    {
      "epoch": 1.6422271728515624e-05,
      "model_forward_time": 5.981949329376221,
      "step": 17220
    },
    {
      "dynamics_loss_avg": 0.07237728834152221,
      "epoch": 1.6422271728515624e-05,
      "step": 17220
    },
    {
      "action_loss_avg": 0.01141577959060669,
      "epoch": 1.6422271728515624e-05,
      "step": 17220
    },
    {
      "epoch": 1.6422271728515624e-05,
      "step": 17220,
      "training_step_time": 17.494647979736328
    },
    {
      "epoch": 1.642322540283203e-05,
      "model_forward_time": 5.622472524642944,
      "step": 17221
    },
    {
      "epoch": 1.642322540283203e-05,
      "step": 17221,
      "training_step_time": 17.12259531021118
    },
    {
      "epoch": 1.642417907714844e-05,
      "model_forward_time": 5.614055156707764,
      "step": 17222
    },
    {
      "epoch": 1.642417907714844e-05,
      "step": 17222,
      "training_step_time": 17.116416215896606
    },
    {
      "epoch": 1.6425132751464843e-05,
      "model_forward_time": 5.623872518539429,
      "step": 17223
    },
    {
      "epoch": 1.6425132751464843e-05,
      "step": 17223,
      "training_step_time": 17.119621992111206
    },
    {
      "epoch": 1.642608642578125e-05,
      "model_forward_time": 5.616363525390625,
      "step": 17224
    },
    {
      "epoch": 1.642608642578125e-05,
      "step": 17224,
      "training_step_time": 17.111644506454468
    },
    {
      "epoch": 1.6427040100097658e-05,
      "model_forward_time": 5.618627548217773,
      "step": 17225
    },
    {
      "epoch": 1.6427040100097658e-05,
      "step": 17225,
      "training_step_time": 17.442204475402832
    },
    {
      "epoch": 1.642799377441406e-05,
      "model_forward_time": 5.605328559875488,
      "step": 17226
    },
    {
      "epoch": 1.642799377441406e-05,
      "step": 17226,
      "training_step_time": 17.088489294052124
    },
    {
      "epoch": 1.642894744873047e-05,
      "model_forward_time": 5.606016159057617,
      "step": 17227
    },
    {
      "epoch": 1.642894744873047e-05,
      "step": 17227,
      "training_step_time": 17.057987928390503
    },
    {
      "epoch": 1.6429901123046876e-05,
      "model_forward_time": 5.594475030899048,
      "step": 17228
    },
    {
      "epoch": 1.6429901123046876e-05,
      "step": 17228,
      "training_step_time": 17.053394317626953
    },
    {
      "epoch": 1.643085479736328e-05,
      "model_forward_time": 5.597377777099609,
      "step": 17229
    },
    {
      "epoch": 1.643085479736328e-05,
      "step": 17229,
      "training_step_time": 17.044296979904175
    },
    {
      "epoch": 1.6431808471679688e-05,
      "grad_norm": 0.33688855171203613,
      "learning_rate": 1e-05,
      "loss": 0.0571,
      "step": 17230
    },
    {
      "epoch": 1.6431808471679688e-05,
      "model_forward_time": 5.607959747314453,
      "step": 17230
    },
    {
      "dynamics_loss_avg": 0.051613268069922925,
      "epoch": 1.6431808471679688e-05,
      "step": 17230
    },
    {
      "action_loss_avg": 0.005344954819884151,
      "epoch": 1.6431808471679688e-05,
      "step": 17230
    },
    {
      "epoch": 1.6431808471679688e-05,
      "step": 17230,
      "training_step_time": 17.07268977165222
    },
    {
      "epoch": 1.6432762145996095e-05,
      "model_forward_time": 5.94438624382019,
      "step": 17231
    },
    {
      "epoch": 1.6432762145996095e-05,
      "step": 17231,
      "training_step_time": 17.400339126586914
    },
    {
      "epoch": 1.64337158203125e-05,
      "model_forward_time": 5.595994234085083,
      "step": 17232
    },
    {
      "epoch": 1.64337158203125e-05,
      "step": 17232,
      "training_step_time": 17.05211305618286
    },
    {
      "epoch": 1.6434669494628907e-05,
      "model_forward_time": 5.604726314544678,
      "step": 17233
    },
    {
      "epoch": 1.6434669494628907e-05,
      "step": 17233,
      "training_step_time": 17.084511041641235
    },
    {
      "epoch": 1.6435623168945314e-05,
      "model_forward_time": 5.619264602661133,
      "step": 17234
    },
    {
      "epoch": 1.6435623168945314e-05,
      "step": 17234,
      "training_step_time": 17.113396406173706
    },
    {
      "epoch": 1.6436576843261718e-05,
      "model_forward_time": 5.621349334716797,
      "step": 17235
    },
    {
      "epoch": 1.6436576843261718e-05,
      "step": 17235,
      "training_step_time": 17.108234405517578
    },
    {
      "epoch": 1.6437530517578126e-05,
      "model_forward_time": 5.97919225692749,
      "step": 17236
    },
    {
      "epoch": 1.6437530517578126e-05,
      "step": 17236,
      "training_step_time": 17.491286516189575
    },
    {
      "epoch": 1.643848419189453e-05,
      "model_forward_time": 5.621766090393066,
      "step": 17237
    },
    {
      "epoch": 1.643848419189453e-05,
      "step": 17237,
      "training_step_time": 17.080981731414795
    },
    {
      "epoch": 1.6439437866210937e-05,
      "model_forward_time": 5.587249517440796,
      "step": 17238
    },
    {
      "epoch": 1.6439437866210937e-05,
      "step": 17238,
      "training_step_time": 17.00983214378357
    },
    {
      "epoch": 1.6440391540527344e-05,
      "model_forward_time": 5.580992937088013,
      "step": 17239
    },
    {
      "epoch": 1.6440391540527344e-05,
      "step": 17239,
      "training_step_time": 16.99258589744568
    },
    {
      "epoch": 1.644134521484375e-05,
      "grad_norm": 1.3259055614471436,
      "learning_rate": 1e-05,
      "loss": 0.0652,
      "step": 17240
    },
    {
      "epoch": 1.644134521484375e-05,
      "model_forward_time": 5.575656414031982,
      "step": 17240
    },
    {
      "dynamics_loss_avg": 0.06322942860424519,
      "epoch": 1.644134521484375e-05,
      "step": 17240
    },
    {
      "action_loss_avg": 0.009109463053755462,
      "epoch": 1.644134521484375e-05,
      "step": 17240
    },
    {
      "epoch": 1.644134521484375e-05,
      "step": 17240,
      "training_step_time": 17.011090993881226
    },
    {
      "epoch": 1.6442298889160156e-05,
      "model_forward_time": 5.575012922286987,
      "step": 17241
    },
    {
      "epoch": 1.6442298889160156e-05,
      "step": 17241,
      "training_step_time": 17.01730513572693
    },
    {
      "epoch": 1.6443252563476563e-05,
      "model_forward_time": 5.927616357803345,
      "step": 17242
    },
    {
      "epoch": 1.6443252563476563e-05,
      "step": 17242,
      "training_step_time": 17.35891842842102
    },
    {
      "epoch": 1.6444206237792967e-05,
      "model_forward_time": 5.585111618041992,
      "step": 17243
    },
    {
      "epoch": 1.6444206237792967e-05,
      "step": 17243,
      "training_step_time": 17.01185894012451
    },
    {
      "epoch": 1.6445159912109375e-05,
      "model_forward_time": 5.5951433181762695,
      "step": 17244
    },
    {
      "epoch": 1.6445159912109375e-05,
      "step": 17244,
      "training_step_time": 17.050265789031982
    },
    {
      "epoch": 1.6446113586425782e-05,
      "model_forward_time": 5.602088689804077,
      "step": 17245
    },
    {
      "epoch": 1.6446113586425782e-05,
      "step": 17245,
      "training_step_time": 17.1095027923584
    },
    {
      "epoch": 1.6447067260742186e-05,
      "model_forward_time": 5.6217992305755615,
      "step": 17246
    },
    {
      "epoch": 1.6447067260742186e-05,
      "step": 17246,
      "training_step_time": 17.129202365875244
    },
    {
      "epoch": 1.6448020935058594e-05,
      "model_forward_time": 5.9980878829956055,
      "step": 17247
    },
    {
      "epoch": 1.6448020935058594e-05,
      "step": 17247,
      "training_step_time": 17.499089241027832
    },
    {
      "epoch": 1.6448974609375e-05,
      "model_forward_time": 5.623147010803223,
      "step": 17248
    },
    {
      "epoch": 1.6448974609375e-05,
      "step": 17248,
      "training_step_time": 17.138988494873047
    },
    {
      "epoch": 1.6449928283691405e-05,
      "model_forward_time": 5.609087228775024,
      "step": 17249
    },
    {
      "epoch": 1.6449928283691405e-05,
      "step": 17249,
      "training_step_time": 17.085689544677734
    },
    {
      "epoch": 1.6450881958007812e-05,
      "grad_norm": 0.3656979501247406,
      "learning_rate": 1e-05,
      "loss": 0.0663,
      "step": 17250
    },
    {
      "epoch": 1.6450881958007812e-05,
      "model_forward_time": 5.582621335983276,
      "step": 17250
    },
    {
      "dynamics_loss_avg": 0.052901233360171315,
      "epoch": 1.6450881958007812e-05,
      "step": 17250
    },
    {
      "action_loss_avg": 0.0046846976387314495,
      "epoch": 1.6450881958007812e-05,
      "step": 17250
    },
    {
      "epoch": 1.6450881958007812e-05,
      "step": 17250,
      "training_step_time": 17.00899314880371
    },
    {
      "epoch": 1.645183563232422e-05,
      "model_forward_time": 5.580135345458984,
      "step": 17251
    },
    {
      "epoch": 1.645183563232422e-05,
      "step": 17251,
      "training_step_time": 17.00002646446228
    },
    {
      "epoch": 1.6452789306640624e-05,
      "model_forward_time": 5.5676915645599365,
      "step": 17252
    },
    {
      "epoch": 1.6452789306640624e-05,
      "step": 17252,
      "training_step_time": 17.01633858680725
    },
    {
      "epoch": 1.645374298095703e-05,
      "model_forward_time": 5.92672061920166,
      "step": 17253
    },
    {
      "epoch": 1.645374298095703e-05,
      "step": 17253,
      "training_step_time": 17.346126079559326
    },
    {
      "epoch": 1.645469665527344e-05,
      "model_forward_time": 5.589647531509399,
      "step": 17254
    },
    {
      "epoch": 1.645469665527344e-05,
      "step": 17254,
      "training_step_time": 17.01848077774048
    },
    {
      "epoch": 1.6455650329589843e-05,
      "model_forward_time": 5.5804197788238525,
      "step": 17255
    },
    {
      "epoch": 1.6455650329589843e-05,
      "step": 17255,
      "training_step_time": 17.002962112426758
    },
    {
      "epoch": 1.645660400390625e-05,
      "model_forward_time": 5.58128547668457,
      "step": 17256
    },
    {
      "epoch": 1.645660400390625e-05,
      "step": 17256,
      "training_step_time": 17.018977403640747
    },
    {
      "epoch": 1.6457557678222658e-05,
      "model_forward_time": 5.574870586395264,
      "step": 17257
    },
    {
      "epoch": 1.6457557678222658e-05,
      "step": 17257,
      "training_step_time": 17.005735397338867
    },
    {
      "epoch": 1.645851135253906e-05,
      "model_forward_time": 5.589310169219971,
      "step": 17258
    },
    {
      "epoch": 1.645851135253906e-05,
      "step": 17258,
      "training_step_time": 17.38028860092163
    },
    {
      "epoch": 1.645946502685547e-05,
      "model_forward_time": 5.598311424255371,
      "step": 17259
    },
    {
      "epoch": 1.645946502685547e-05,
      "step": 17259,
      "training_step_time": 17.064687967300415
    },
    {
      "epoch": 1.6460418701171876e-05,
      "grad_norm": 1.1474894285202026,
      "learning_rate": 1e-05,
      "loss": 0.0727,
      "step": 17260
    },
    {
      "epoch": 1.6460418701171876e-05,
      "model_forward_time": 5.601492166519165,
      "step": 17260
    },
    {
      "dynamics_loss_avg": 0.0668411610648036,
      "epoch": 1.6460418701171876e-05,
      "step": 17260
    },
    {
      "action_loss_avg": 0.010211060079745949,
      "epoch": 1.6460418701171876e-05,
      "step": 17260
    },
    {
      "epoch": 1.6460418701171876e-05,
      "step": 17260,
      "training_step_time": 17.06318163871765
    },
    {
      "epoch": 1.646137237548828e-05,
      "model_forward_time": 5.5972900390625,
      "step": 17261
    },
    {
      "epoch": 1.646137237548828e-05,
      "step": 17261,
      "training_step_time": 17.060813426971436
    },
    {
      "epoch": 1.6462326049804688e-05,
      "model_forward_time": 5.594033241271973,
      "step": 17262
    },
    {
      "epoch": 1.6462326049804688e-05,
      "step": 17262,
      "training_step_time": 17.035074710845947
    },
    {
      "epoch": 1.6463279724121095e-05,
      "model_forward_time": 5.595099449157715,
      "step": 17263
    },
    {
      "epoch": 1.6463279724121095e-05,
      "step": 17263,
      "training_step_time": 17.025946855545044
    },
    {
      "epoch": 1.64642333984375e-05,
      "model_forward_time": 5.92038106918335,
      "step": 17264
    },
    {
      "epoch": 1.64642333984375e-05,
      "step": 17264,
      "training_step_time": 17.335710525512695
    },
    {
      "epoch": 1.6465187072753907e-05,
      "model_forward_time": 5.5856897830963135,
      "step": 17265
    },
    {
      "epoch": 1.6465187072753907e-05,
      "step": 17265,
      "training_step_time": 17.0205078125
    },
    {
      "epoch": 1.646614074707031e-05,
      "model_forward_time": 5.597827434539795,
      "step": 17266
    },
    {
      "epoch": 1.646614074707031e-05,
      "step": 17266,
      "training_step_time": 17.069374561309814
    },
    {
      "epoch": 1.6467094421386718e-05,
      "model_forward_time": 5.611571311950684,
      "step": 17267
    },
    {
      "epoch": 1.6467094421386718e-05,
      "step": 17267,
      "training_step_time": 17.11553430557251
    },
    {
      "epoch": 1.6468048095703126e-05,
      "model_forward_time": 5.627624750137329,
      "step": 17268
    },
    {
      "epoch": 1.6468048095703126e-05,
      "step": 17268,
      "training_step_time": 17.152904987335205
    },
    {
      "epoch": 1.646900177001953e-05,
      "model_forward_time": 5.9962756633758545,
      "step": 17269
    },
    {
      "epoch": 1.646900177001953e-05,
      "step": 17269,
      "training_step_time": 17.502658367156982
    },
    {
      "epoch": 1.6469955444335937e-05,
      "grad_norm": 0.565189003944397,
      "learning_rate": 1e-05,
      "loss": 0.0573,
      "step": 17270
    },
    {
      "epoch": 1.6469955444335937e-05,
      "model_forward_time": 5.6499810218811035,
      "step": 17270
    },
    {
      "dynamics_loss_avg": 0.05331577956676483,
      "epoch": 1.6469955444335937e-05,
      "step": 17270
    },
    {
      "action_loss_avg": 0.006565751414746046,
      "epoch": 1.6469955444335937e-05,
      "step": 17270
    },
    {
      "epoch": 1.6469955444335937e-05,
      "step": 17270,
      "training_step_time": 17.16789197921753
    },
    {
      "epoch": 1.6470909118652345e-05,
      "model_forward_time": 5.613694906234741,
      "step": 17271
    },
    {
      "epoch": 1.6470909118652345e-05,
      "step": 17271,
      "training_step_time": 17.10117745399475
    },
    {
      "epoch": 1.647186279296875e-05,
      "model_forward_time": 5.605823278427124,
      "step": 17272
    },
    {
      "epoch": 1.647186279296875e-05,
      "step": 17272,
      "training_step_time": 17.080820083618164
    },
    {
      "epoch": 1.6472816467285156e-05,
      "model_forward_time": 5.596524953842163,
      "step": 17273
    },
    {
      "epoch": 1.6472816467285156e-05,
      "step": 17273,
      "training_step_time": 17.06792902946472
    },
    {
      "epoch": 1.6473770141601563e-05,
      "model_forward_time": 5.597348928451538,
      "step": 17274
    },
    {
      "epoch": 1.6473770141601563e-05,
      "step": 17274,
      "training_step_time": 17.05364227294922
    },
    {
      "epoch": 1.6474723815917967e-05,
      "model_forward_time": 5.968709945678711,
      "step": 17275
    },
    {
      "epoch": 1.6474723815917967e-05,
      "step": 17275,
      "training_step_time": 17.430811405181885
    },
    {
      "epoch": 1.6475677490234375e-05,
      "model_forward_time": 5.603129863739014,
      "step": 17276
    },
    {
      "epoch": 1.6475677490234375e-05,
      "step": 17276,
      "training_step_time": 17.09183120727539
    },
    {
      "epoch": 1.6476631164550782e-05,
      "model_forward_time": 5.608179807662964,
      "step": 17277
    },
    {
      "epoch": 1.6476631164550782e-05,
      "step": 17277,
      "training_step_time": 17.110206365585327
    },
    {
      "epoch": 1.6477584838867186e-05,
      "model_forward_time": 5.609832286834717,
      "step": 17278
    },
    {
      "epoch": 1.6477584838867186e-05,
      "step": 17278,
      "training_step_time": 17.106970071792603
    },
    {
      "epoch": 1.6478538513183594e-05,
      "model_forward_time": 5.61785101890564,
      "step": 17279
    },
    {
      "epoch": 1.6478538513183594e-05,
      "step": 17279,
      "training_step_time": 17.09127640724182
    },
    {
      "epoch": 1.64794921875e-05,
      "grad_norm": 0.8769864439964294,
      "learning_rate": 1e-05,
      "loss": 0.0863,
      "step": 17280
    },
    {
      "epoch": 1.64794921875e-05,
      "model_forward_time": 5.97560715675354,
      "step": 17280
    },
    {
      "dynamics_loss_avg": 0.07634860575199127,
      "epoch": 1.64794921875e-05,
      "step": 17280
    },
    {
      "action_loss_avg": 0.009953700285404921,
      "epoch": 1.64794921875e-05,
      "step": 17280
    },
    {
      "epoch": 1.64794921875e-05,
      "step": 17280,
      "training_step_time": 17.41174864768982
    },
    {
      "epoch": 1.6480445861816405e-05,
      "model_forward_time": 5.620287179946899,
      "step": 17281
    },
    {
      "epoch": 1.6480445861816405e-05,
      "step": 17281,
      "training_step_time": 17.07757329940796
    },
    {
      "epoch": 1.6481399536132813e-05,
      "model_forward_time": 5.617146015167236,
      "step": 17282
    },
    {
      "epoch": 1.6481399536132813e-05,
      "step": 17282,
      "training_step_time": 17.122512817382812
    },
    {
      "epoch": 1.648235321044922e-05,
      "model_forward_time": 5.630964279174805,
      "step": 17283
    },
    {
      "epoch": 1.648235321044922e-05,
      "step": 17283,
      "training_step_time": 17.132319688796997
    },
    {
      "epoch": 1.6483306884765624e-05,
      "model_forward_time": 5.588307619094849,
      "step": 17284
    },
    {
      "epoch": 1.6483306884765624e-05,
      "step": 17284,
      "training_step_time": 17.014554977416992
    },
    {
      "epoch": 1.648426055908203e-05,
      "model_forward_time": 5.577330112457275,
      "step": 17285
    },
    {
      "epoch": 1.648426055908203e-05,
      "step": 17285,
      "training_step_time": 16.99645757675171
    },
    {
      "epoch": 1.648521423339844e-05,
      "model_forward_time": 5.906115531921387,
      "step": 17286
    },
    {
      "epoch": 1.648521423339844e-05,
      "step": 17286,
      "training_step_time": 17.33841633796692
    },
    {
      "epoch": 1.6486167907714843e-05,
      "model_forward_time": 5.57005763053894,
      "step": 17287
    },
    {
      "epoch": 1.6486167907714843e-05,
      "step": 17287,
      "training_step_time": 16.99640679359436
    },
    {
      "epoch": 1.648712158203125e-05,
      "model_forward_time": 5.580235719680786,
      "step": 17288
    },
    {
      "epoch": 1.648712158203125e-05,
      "step": 17288,
      "training_step_time": 16.991502046585083
    },
    {
      "epoch": 1.6488075256347658e-05,
      "model_forward_time": 5.582435846328735,
      "step": 17289
    },
    {
      "epoch": 1.6488075256347658e-05,
      "step": 17289,
      "training_step_time": 16.999398469924927
    },
    {
      "epoch": 1.648902893066406e-05,
      "grad_norm": 0.7223962545394897,
      "learning_rate": 1e-05,
      "loss": 0.0612,
      "step": 17290
    },
    {
      "epoch": 1.648902893066406e-05,
      "model_forward_time": 5.578796148300171,
      "step": 17290
    },
    {
      "dynamics_loss_avg": 0.0485270569100976,
      "epoch": 1.648902893066406e-05,
      "step": 17290
    },
    {
      "action_loss_avg": 0.005577080184593797,
      "epoch": 1.648902893066406e-05,
      "step": 17290
    },
    {
      "epoch": 1.648902893066406e-05,
      "step": 17290,
      "training_step_time": 17.023207187652588
    },
    {
      "epoch": 1.648998260498047e-05,
      "model_forward_time": 5.588555574417114,
      "step": 17291
    },
    {
      "epoch": 1.648998260498047e-05,
      "step": 17291,
      "training_step_time": 17.379509449005127
    },
    {
      "epoch": 1.6490936279296877e-05,
      "model_forward_time": 5.5926666259765625,
      "step": 17292
    },
    {
      "epoch": 1.6490936279296877e-05,
      "step": 17292,
      "training_step_time": 17.047334909439087
    },
    {
      "epoch": 1.649188995361328e-05,
      "model_forward_time": 5.60463809967041,
      "step": 17293
    },
    {
      "epoch": 1.649188995361328e-05,
      "step": 17293,
      "training_step_time": 17.105631351470947
    },
    {
      "epoch": 1.6492843627929688e-05,
      "model_forward_time": 5.617215633392334,
      "step": 17294
    },
    {
      "epoch": 1.6492843627929688e-05,
      "step": 17294,
      "training_step_time": 17.10581350326538
    },
    {
      "epoch": 1.6493797302246095e-05,
      "model_forward_time": 5.596274375915527,
      "step": 17295
    },
    {
      "epoch": 1.6493797302246095e-05,
      "step": 17295,
      "training_step_time": 17.046253442764282
    },
    {
      "epoch": 1.64947509765625e-05,
      "model_forward_time": 5.590391397476196,
      "step": 17296
    },
    {
      "epoch": 1.64947509765625e-05,
      "step": 17296,
      "training_step_time": 17.03031587600708
    },
    {
      "epoch": 1.6495704650878907e-05,
      "model_forward_time": 5.927365779876709,
      "step": 17297
    },
    {
      "epoch": 1.6495704650878907e-05,
      "step": 17297,
      "training_step_time": 17.37252378463745
    },
    {
      "epoch": 1.649665832519531e-05,
      "model_forward_time": 5.589831829071045,
      "step": 17298
    },
    {
      "epoch": 1.649665832519531e-05,
      "step": 17298,
      "training_step_time": 17.049384593963623
    },
    {
      "epoch": 1.6497611999511718e-05,
      "model_forward_time": 5.607262134552002,
      "step": 17299
    },
    {
      "epoch": 1.6497611999511718e-05,
      "step": 17299,
      "training_step_time": 17.129695177078247
    },
    {
      "epoch": 1.6498565673828126e-05,
      "grad_norm": 0.3174983561038971,
      "learning_rate": 1e-05,
      "loss": 0.0571,
      "step": 17300
    },
    {
      "epoch": 1.6498565673828126e-05,
      "model_forward_time": 5.626604795455933,
      "step": 17300
    },
    {
      "dynamics_loss_avg": 0.05343838185071945,
      "epoch": 1.6498565673828126e-05,
      "step": 17300
    },
    {
      "action_loss_avg": 0.005379115021787584,
      "epoch": 1.6498565673828126e-05,
      "step": 17300
    },
    {
      "epoch": 1.6498565673828126e-05,
      "step": 17300,
      "training_step_time": 17.147268533706665
    },
    {
      "epoch": 1.649951934814453e-05,
      "model_forward_time": 5.628011703491211,
      "step": 17301
    },
    {
      "epoch": 1.649951934814453e-05,
      "step": 17301,
      "training_step_time": 17.13849925994873
    },
    {
      "epoch": 1.6500473022460937e-05,
      "model_forward_time": 5.988989353179932,
      "step": 17302
    },
    {
      "epoch": 1.6500473022460937e-05,
      "step": 17302,
      "training_step_time": 17.474534273147583
    },
    {
      "epoch": 1.6501426696777345e-05,
      "model_forward_time": 5.605828285217285,
      "step": 17303
    },
    {
      "epoch": 1.6501426696777345e-05,
      "step": 17303,
      "training_step_time": 17.06683874130249
    },
    {
      "epoch": 1.650238037109375e-05,
      "model_forward_time": 5.598932981491089,
      "step": 17304
    },
    {
      "epoch": 1.650238037109375e-05,
      "step": 17304,
      "training_step_time": 17.041872262954712
    },
    {
      "epoch": 1.6503334045410156e-05,
      "model_forward_time": 5.5856146812438965,
      "step": 17305
    },
    {
      "epoch": 1.6503334045410156e-05,
      "step": 17305,
      "training_step_time": 17.022289514541626
    },
    {
      "epoch": 1.6504287719726563e-05,
      "model_forward_time": 5.59574294090271,
      "step": 17306
    },
    {
      "epoch": 1.6504287719726563e-05,
      "step": 17306,
      "training_step_time": 17.023550033569336
    },
    {
      "epoch": 1.6505241394042967e-05,
      "model_forward_time": 5.5970635414123535,
      "step": 17307
    },
    {
      "epoch": 1.6505241394042967e-05,
      "step": 17307,
      "training_step_time": 17.03844952583313
    },
    {
      "epoch": 1.6506195068359375e-05,
      "model_forward_time": 5.939627647399902,
      "step": 17308
    },
    {
      "epoch": 1.6506195068359375e-05,
      "step": 17308,
      "training_step_time": 17.40927767753601
    },
    {
      "epoch": 1.6507148742675782e-05,
      "model_forward_time": 5.6123223304748535,
      "step": 17309
    },
    {
      "epoch": 1.6507148742675782e-05,
      "step": 17309,
      "training_step_time": 17.121516942977905
    },
    {
      "epoch": 1.6508102416992186e-05,
      "grad_norm": 0.470389723777771,
      "learning_rate": 1e-05,
      "loss": 0.0529,
      "step": 17310
    },
    {
      "epoch": 1.6508102416992186e-05,
      "model_forward_time": 5.6212029457092285,
      "step": 17310
    },
    {
      "dynamics_loss_avg": 0.049604779481887816,
      "epoch": 1.6508102416992186e-05,
      "step": 17310
    },
    {
      "action_loss_avg": 0.00648725291248411,
      "epoch": 1.6508102416992186e-05,
      "step": 17310
    },
    {
      "epoch": 1.6508102416992186e-05,
      "step": 17310,
      "training_step_time": 17.107364654541016
    },
    {
      "epoch": 1.6509056091308594e-05,
      "model_forward_time": 5.587709426879883,
      "step": 17311
    },
    {
      "epoch": 1.6509056091308594e-05,
      "step": 17311,
      "training_step_time": 17.022358655929565
    },
    {
      "epoch": 1.6510009765625e-05,
      "model_forward_time": 5.576815605163574,
      "step": 17312
    },
    {
      "epoch": 1.6510009765625e-05,
      "step": 17312,
      "training_step_time": 17.007144927978516
    },
    {
      "epoch": 1.6510963439941405e-05,
      "model_forward_time": 5.956680059432983,
      "step": 17313
    },
    {
      "epoch": 1.6510963439941405e-05,
      "step": 17313,
      "training_step_time": 17.39906907081604
    },
    {
      "epoch": 1.6511917114257813e-05,
      "model_forward_time": 5.573546409606934,
      "step": 17314
    },
    {
      "epoch": 1.6511917114257813e-05,
      "step": 17314,
      "training_step_time": 16.986533403396606
    },
    {
      "epoch": 1.651287078857422e-05,
      "model_forward_time": 5.576818943023682,
      "step": 17315
    },
    {
      "epoch": 1.651287078857422e-05,
      "step": 17315,
      "training_step_time": 16.98763155937195
    },
    {
      "epoch": 1.6513824462890624e-05,
      "model_forward_time": 5.581501722335815,
      "step": 17316
    },
    {
      "epoch": 1.6513824462890624e-05,
      "step": 17316,
      "training_step_time": 17.025015115737915
    },
    {
      "epoch": 1.651477813720703e-05,
      "model_forward_time": 5.586032867431641,
      "step": 17317
    },
    {
      "epoch": 1.651477813720703e-05,
      "step": 17317,
      "training_step_time": 17.02007007598877
    },
    {
      "epoch": 1.651573181152344e-05,
      "model_forward_time": 5.585531234741211,
      "step": 17318
    },
    {
      "epoch": 1.651573181152344e-05,
      "step": 17318,
      "training_step_time": 17.049524545669556
    },
    {
      "epoch": 1.6516685485839843e-05,
      "model_forward_time": 5.966846466064453,
      "step": 17319
    },
    {
      "epoch": 1.6516685485839843e-05,
      "step": 17319,
      "training_step_time": 17.459924459457397
    },
    {
      "epoch": 1.651763916015625e-05,
      "grad_norm": 0.7838605642318726,
      "learning_rate": 1e-05,
      "loss": 0.073,
      "step": 17320
    },
    {
      "epoch": 1.651763916015625e-05,
      "model_forward_time": 5.616055965423584,
      "step": 17320
    },
    {
      "dynamics_loss_avg": 0.06454147025942802,
      "epoch": 1.651763916015625e-05,
      "step": 17320
    },
    {
      "action_loss_avg": 0.008636952913366257,
      "epoch": 1.651763916015625e-05,
      "step": 17320
    },
    {
      "epoch": 1.651763916015625e-05,
      "step": 17320,
      "training_step_time": 17.095312356948853
    },
    {
      "epoch": 1.6518592834472658e-05,
      "model_forward_time": 5.605900049209595,
      "step": 17321
    },
    {
      "epoch": 1.6518592834472658e-05,
      "step": 17321,
      "training_step_time": 17.077007293701172
    },
    {
      "epoch": 1.6519546508789062e-05,
      "model_forward_time": 5.594118595123291,
      "step": 17322
    },
    {
      "epoch": 1.6519546508789062e-05,
      "step": 17322,
      "training_step_time": 17.046921491622925
    },
    {
      "epoch": 1.652050018310547e-05,
      "model_forward_time": 5.609842538833618,
      "step": 17323
    },
    {
      "epoch": 1.652050018310547e-05,
      "step": 17323,
      "training_step_time": 17.096319913864136
    },
    {
      "epoch": 1.6521453857421877e-05,
      "model_forward_time": 5.603382110595703,
      "step": 17324
    },
    {
      "epoch": 1.6521453857421877e-05,
      "step": 17324,
      "training_step_time": 17.4535174369812
    },
    {
      "epoch": 1.652240753173828e-05,
      "model_forward_time": 5.58782696723938,
      "step": 17325
    },
    {
      "epoch": 1.652240753173828e-05,
      "step": 17325,
      "training_step_time": 16.998128414154053
    },
    {
      "epoch": 1.6523361206054688e-05,
      "model_forward_time": 5.576491594314575,
      "step": 17326
    },
    {
      "epoch": 1.6523361206054688e-05,
      "step": 17326,
      "training_step_time": 16.99901509284973
    },
    {
      "epoch": 1.6524314880371095e-05,
      "model_forward_time": 5.565672159194946,
      "step": 17327
    },
    {
      "epoch": 1.6524314880371095e-05,
      "step": 17327,
      "training_step_time": 16.981472969055176
    },
    {
      "epoch": 1.65252685546875e-05,
      "model_forward_time": 4.2416832447052,
      "step": 17328
    },
    {
      "epoch": 1.65252685546875e-05,
      "step": 17328,
      "training_step_time": 12.713164329528809
    },
    {
      "epoch": 1.6526222229003907e-05,
      "model_forward_time": 5.573663711547852,
      "step": 17329
    },
    {
      "epoch": 1.6526222229003907e-05,
      "step": 17329,
      "training_step_time": 16.986965894699097
    },
    {
      "epoch": 1.652717590332031e-05,
      "grad_norm": 0.6871009469032288,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 17330
    },
    {
      "epoch": 1.652717590332031e-05,
      "model_forward_time": 5.924695014953613,
      "step": 17330
    },
    {
      "dynamics_loss_avg": 0.05853283945471048,
      "epoch": 1.652717590332031e-05,
      "step": 17330
    },
    {
      "action_loss_avg": 0.006893410929478705,
      "epoch": 1.652717590332031e-05,
      "step": 17330
    },
    {
      "epoch": 1.652717590332031e-05,
      "step": 17330,
      "training_step_time": 17.338430643081665
    },
    {
      "epoch": 1.6528129577636718e-05,
      "model_forward_time": 5.587471008300781,
      "step": 17331
    },
    {
      "epoch": 1.6528129577636718e-05,
      "step": 17331,
      "training_step_time": 17.012362718582153
    },
    {
      "epoch": 1.6529083251953126e-05,
      "model_forward_time": 5.591823101043701,
      "step": 17332
    },
    {
      "epoch": 1.6529083251953126e-05,
      "step": 17332,
      "training_step_time": 17.032812356948853
    },
    {
      "epoch": 1.653003692626953e-05,
      "model_forward_time": 5.600749254226685,
      "step": 17333
    },
    {
      "epoch": 1.653003692626953e-05,
      "step": 17333,
      "training_step_time": 17.053882837295532
    },
    {
      "epoch": 1.6530990600585937e-05,
      "model_forward_time": 5.598984241485596,
      "step": 17334
    },
    {
      "epoch": 1.6530990600585937e-05,
      "step": 17334,
      "training_step_time": 17.073899507522583
    },
    {
      "epoch": 1.6531944274902345e-05,
      "model_forward_time": 5.608478784561157,
      "step": 17335
    },
    {
      "epoch": 1.6531944274902345e-05,
      "step": 17335,
      "training_step_time": 17.45790958404541
    },
    {
      "epoch": 1.653289794921875e-05,
      "model_forward_time": 5.618235349655151,
      "step": 17336
    },
    {
      "epoch": 1.653289794921875e-05,
      "step": 17336,
      "training_step_time": 17.083321809768677
    },
    {
      "epoch": 1.6533851623535156e-05,
      "model_forward_time": 5.596665620803833,
      "step": 17337
    },
    {
      "epoch": 1.6533851623535156e-05,
      "step": 17337,
      "training_step_time": 17.038994789123535
    },
    {
      "epoch": 1.6534805297851563e-05,
      "model_forward_time": 5.601659059524536,
      "step": 17338
    },
    {
      "epoch": 1.6534805297851563e-05,
      "step": 17338,
      "training_step_time": 17.066707134246826
    },
    {
      "epoch": 1.6535758972167967e-05,
      "model_forward_time": 5.593278408050537,
      "step": 17339
    },
    {
      "epoch": 1.6535758972167967e-05,
      "step": 17339,
      "training_step_time": 17.046467065811157
    },
    {
      "epoch": 1.6536712646484375e-05,
      "grad_norm": 0.9101909399032593,
      "learning_rate": 1e-05,
      "loss": 0.0622,
      "step": 17340
    },
    {
      "epoch": 1.6536712646484375e-05,
      "model_forward_time": 5.5820770263671875,
      "step": 17340
    },
    {
      "dynamics_loss_avg": 0.05373159721493721,
      "epoch": 1.6536712646484375e-05,
      "step": 17340
    },
    {
      "action_loss_avg": 0.006250936980359256,
      "epoch": 1.6536712646484375e-05,
      "step": 17340
    },
    {
      "epoch": 1.6536712646484375e-05,
      "step": 17340,
      "training_step_time": 17.034236669540405
    },
    {
      "epoch": 1.6537666320800782e-05,
      "model_forward_time": 5.948503494262695,
      "step": 17341
    },
    {
      "epoch": 1.6537666320800782e-05,
      "step": 17341,
      "training_step_time": 17.426607608795166
    },
    {
      "epoch": 1.6538619995117186e-05,
      "model_forward_time": 5.608319282531738,
      "step": 17342
    },
    {
      "epoch": 1.6538619995117186e-05,
      "step": 17342,
      "training_step_time": 17.074056148529053
    },
    {
      "epoch": 1.6539573669433594e-05,
      "model_forward_time": 5.604394435882568,
      "step": 17343
    },
    {
      "epoch": 1.6539573669433594e-05,
      "step": 17343,
      "training_step_time": 17.064802646636963
    },
    {
      "epoch": 1.654052734375e-05,
      "model_forward_time": 5.592195272445679,
      "step": 17344
    },
    {
      "epoch": 1.654052734375e-05,
      "step": 17344,
      "training_step_time": 17.05412006378174
    },
    {
      "epoch": 1.6541481018066405e-05,
      "model_forward_time": 5.601129770278931,
      "step": 17345
    },
    {
      "epoch": 1.6541481018066405e-05,
      "step": 17345,
      "training_step_time": 17.081653118133545
    },
    {
      "epoch": 1.6542434692382813e-05,
      "model_forward_time": 5.969753265380859,
      "step": 17346
    },
    {
      "epoch": 1.6542434692382813e-05,
      "step": 17346,
      "training_step_time": 17.462770223617554
    },
    {
      "epoch": 1.654338836669922e-05,
      "model_forward_time": 5.614162921905518,
      "step": 17347
    },
    {
      "epoch": 1.654338836669922e-05,
      "step": 17347,
      "training_step_time": 17.095375299453735
    },
    {
      "epoch": 1.6544342041015624e-05,
      "model_forward_time": 5.619317293167114,
      "step": 17348
    },
    {
      "epoch": 1.6544342041015624e-05,
      "step": 17348,
      "training_step_time": 17.112662315368652
    },
    {
      "epoch": 1.654529571533203e-05,
      "model_forward_time": 5.612734317779541,
      "step": 17349
    },
    {
      "epoch": 1.654529571533203e-05,
      "step": 17349,
      "training_step_time": 17.106935262680054
    },
    {
      "epoch": 1.654624938964844e-05,
      "grad_norm": 1.0125383138656616,
      "learning_rate": 1e-05,
      "loss": 0.0491,
      "step": 17350
    },
    {
      "epoch": 1.654624938964844e-05,
      "model_forward_time": 5.615189552307129,
      "step": 17350
    },
    {
      "dynamics_loss_avg": 0.04288924615830183,
      "epoch": 1.654624938964844e-05,
      "step": 17350
    },
    {
      "action_loss_avg": 0.004664916358888149,
      "epoch": 1.654624938964844e-05,
      "step": 17350
    },
    {
      "epoch": 1.654624938964844e-05,
      "step": 17350,
      "training_step_time": 17.104374647140503
    },
    {
      "epoch": 1.6547203063964843e-05,
      "model_forward_time": 5.609552621841431,
      "step": 17351
    },
    {
      "epoch": 1.6547203063964843e-05,
      "step": 17351,
      "training_step_time": 17.06465768814087
    },
    {
      "epoch": 1.654815673828125e-05,
      "model_forward_time": 5.954441547393799,
      "step": 17352
    },
    {
      "epoch": 1.654815673828125e-05,
      "step": 17352,
      "training_step_time": 17.418292760849
    },
    {
      "epoch": 1.6549110412597658e-05,
      "model_forward_time": 5.599269390106201,
      "step": 17353
    },
    {
      "epoch": 1.6549110412597658e-05,
      "step": 17353,
      "training_step_time": 17.06409215927124
    },
    {
      "epoch": 1.6550064086914062e-05,
      "model_forward_time": 5.596273422241211,
      "step": 17354
    },
    {
      "epoch": 1.6550064086914062e-05,
      "step": 17354,
      "training_step_time": 17.047184467315674
    },
    {
      "epoch": 1.655101776123047e-05,
      "model_forward_time": 5.5821356773376465,
      "step": 17355
    },
    {
      "epoch": 1.655101776123047e-05,
      "step": 17355,
      "training_step_time": 17.041249990463257
    },
    {
      "epoch": 1.6551971435546877e-05,
      "model_forward_time": 5.602213382720947,
      "step": 17356
    },
    {
      "epoch": 1.6551971435546877e-05,
      "step": 17356,
      "training_step_time": 17.099396467208862
    },
    {
      "epoch": 1.655292510986328e-05,
      "model_forward_time": 5.9932169914245605,
      "step": 17357
    },
    {
      "epoch": 1.655292510986328e-05,
      "step": 17357,
      "training_step_time": 17.511576890945435
    },
    {
      "epoch": 1.6553878784179688e-05,
      "model_forward_time": 5.605800151824951,
      "step": 17358
    },
    {
      "epoch": 1.6553878784179688e-05,
      "step": 17358,
      "training_step_time": 17.062584161758423
    },
    {
      "epoch": 1.6554832458496095e-05,
      "model_forward_time": 5.5763092041015625,
      "step": 17359
    },
    {
      "epoch": 1.6554832458496095e-05,
      "step": 17359,
      "training_step_time": 17.01817798614502
    },
    {
      "epoch": 1.65557861328125e-05,
      "grad_norm": 1.3015031814575195,
      "learning_rate": 1e-05,
      "loss": 0.0577,
      "step": 17360
    },
    {
      "epoch": 1.65557861328125e-05,
      "model_forward_time": 5.573925733566284,
      "step": 17360
    },
    {
      "dynamics_loss_avg": 0.053397082909941676,
      "epoch": 1.65557861328125e-05,
      "step": 17360
    },
    {
      "action_loss_avg": 0.006276413938030601,
      "epoch": 1.65557861328125e-05,
      "step": 17360
    },
    {
      "epoch": 1.65557861328125e-05,
      "step": 17360,
      "training_step_time": 16.9943208694458
    },
    {
      "epoch": 1.6556739807128907e-05,
      "model_forward_time": 5.575188875198364,
      "step": 17361
    },
    {
      "epoch": 1.6556739807128907e-05,
      "step": 17361,
      "training_step_time": 16.993370532989502
    },
    {
      "epoch": 1.655769348144531e-05,
      "model_forward_time": 5.572882175445557,
      "step": 17362
    },
    {
      "epoch": 1.655769348144531e-05,
      "step": 17362,
      "training_step_time": 17.01358985900879
    },
    {
      "epoch": 1.655864715576172e-05,
      "model_forward_time": 5.9133219718933105,
      "step": 17363
    },
    {
      "epoch": 1.655864715576172e-05,
      "step": 17363,
      "training_step_time": 17.315810918807983
    },
    {
      "epoch": 1.6559600830078126e-05,
      "model_forward_time": 5.593274831771851,
      "step": 17364
    },
    {
      "epoch": 1.6559600830078126e-05,
      "step": 17364,
      "training_step_time": 17.024842262268066
    },
    {
      "epoch": 1.656055450439453e-05,
      "model_forward_time": 5.587422132492065,
      "step": 17365
    },
    {
      "epoch": 1.656055450439453e-05,
      "step": 17365,
      "training_step_time": 17.03396773338318
    },
    {
      "epoch": 1.6561508178710937e-05,
      "model_forward_time": 5.59136176109314,
      "step": 17366
    },
    {
      "epoch": 1.6561508178710937e-05,
      "step": 17366,
      "training_step_time": 17.06638765335083
    },
    {
      "epoch": 1.6562461853027345e-05,
      "model_forward_time": 5.624233245849609,
      "step": 17367
    },
    {
      "epoch": 1.6562461853027345e-05,
      "step": 17367,
      "training_step_time": 17.138261318206787
    },
    {
      "epoch": 1.656341552734375e-05,
      "model_forward_time": 5.99249529838562,
      "step": 17368
    },
    {
      "epoch": 1.656341552734375e-05,
      "step": 17368,
      "training_step_time": 17.48976755142212
    },
    {
      "epoch": 1.6564369201660156e-05,
      "model_forward_time": 5.596765756607056,
      "step": 17369
    },
    {
      "epoch": 1.6564369201660156e-05,
      "step": 17369,
      "training_step_time": 17.02417826652527
    },
    {
      "epoch": 1.6565322875976563e-05,
      "grad_norm": 0.6702868938446045,
      "learning_rate": 1e-05,
      "loss": 0.0533,
      "step": 17370
    },
    {
      "epoch": 1.6565322875976563e-05,
      "model_forward_time": 5.5734639167785645,
      "step": 17370
    },
    {
      "dynamics_loss_avg": 0.053191110119223596,
      "epoch": 1.6565322875976563e-05,
      "step": 17370
    },
    {
      "action_loss_avg": 0.005829702550545335,
      "epoch": 1.6565322875976563e-05,
      "step": 17370
    },
    {
      "epoch": 1.6565322875976563e-05,
      "step": 17370,
      "training_step_time": 17.01940369606018
    },
    {
      "epoch": 1.6566276550292967e-05,
      "model_forward_time": 5.57059383392334,
      "step": 17371
    },
    {
      "epoch": 1.6566276550292967e-05,
      "step": 17371,
      "training_step_time": 16.97913384437561
    },
    {
      "epoch": 1.6567230224609375e-05,
      "model_forward_time": 5.57826566696167,
      "step": 17372
    },
    {
      "epoch": 1.6567230224609375e-05,
      "step": 17372,
      "training_step_time": 16.989144802093506
    },
    {
      "epoch": 1.6568183898925782e-05,
      "model_forward_time": 5.93166708946228,
      "step": 17373
    },
    {
      "epoch": 1.6568183898925782e-05,
      "step": 17373,
      "training_step_time": 17.34927201271057
    },
    {
      "epoch": 1.6569137573242186e-05,
      "model_forward_time": 5.573627710342407,
      "step": 17374
    },
    {
      "epoch": 1.6569137573242186e-05,
      "step": 17374,
      "training_step_time": 17.01430344581604
    },
    {
      "epoch": 1.6570091247558594e-05,
      "model_forward_time": 5.57938551902771,
      "step": 17375
    },
    {
      "epoch": 1.6570091247558594e-05,
      "step": 17375,
      "training_step_time": 16.99746346473694
    },
    {
      "epoch": 1.6571044921875e-05,
      "model_forward_time": 5.594849348068237,
      "step": 17376
    },
    {
      "epoch": 1.6571044921875e-05,
      "step": 17376,
      "training_step_time": 17.086791276931763
    },
    {
      "epoch": 1.6571998596191405e-05,
      "model_forward_time": 5.620779037475586,
      "step": 17377
    },
    {
      "epoch": 1.6571998596191405e-05,
      "step": 17377,
      "training_step_time": 17.139956951141357
    },
    {
      "epoch": 1.6572952270507813e-05,
      "model_forward_time": 5.625481843948364,
      "step": 17378
    },
    {
      "epoch": 1.6572952270507813e-05,
      "step": 17378,
      "training_step_time": 17.133113384246826
    },
    {
      "epoch": 1.657390594482422e-05,
      "model_forward_time": 5.988267421722412,
      "step": 17379
    },
    {
      "epoch": 1.657390594482422e-05,
      "step": 17379,
      "training_step_time": 17.47071862220764
    },
    {
      "epoch": 1.6574859619140624e-05,
      "grad_norm": 0.4596448540687561,
      "learning_rate": 1e-05,
      "loss": 0.0655,
      "step": 17380
    },
    {
      "epoch": 1.6574859619140624e-05,
      "model_forward_time": 5.60256028175354,
      "step": 17380
    },
    {
      "dynamics_loss_avg": 0.0515330633148551,
      "epoch": 1.6574859619140624e-05,
      "step": 17380
    },
    {
      "action_loss_avg": 0.008149612043052912,
      "epoch": 1.6574859619140624e-05,
      "step": 17380
    },
    {
      "epoch": 1.6574859619140624e-05,
      "step": 17380,
      "training_step_time": 17.088735103607178
    },
    {
      "epoch": 1.657581329345703e-05,
      "model_forward_time": 5.600954532623291,
      "step": 17381
    },
    {
      "epoch": 1.657581329345703e-05,
      "step": 17381,
      "training_step_time": 17.066696166992188
    },
    {
      "epoch": 1.657676696777344e-05,
      "model_forward_time": 5.606810092926025,
      "step": 17382
    },
    {
      "epoch": 1.657676696777344e-05,
      "step": 17382,
      "training_step_time": 17.063401699066162
    },
    {
      "epoch": 1.6577720642089843e-05,
      "model_forward_time": 5.592923641204834,
      "step": 17383
    },
    {
      "epoch": 1.6577720642089843e-05,
      "step": 17383,
      "training_step_time": 17.038435459136963
    },
    {
      "epoch": 1.657867431640625e-05,
      "model_forward_time": 5.593001127243042,
      "step": 17384
    },
    {
      "epoch": 1.657867431640625e-05,
      "step": 17384,
      "training_step_time": 17.412709951400757
    },
    {
      "epoch": 1.6579627990722658e-05,
      "model_forward_time": 5.608152866363525,
      "step": 17385
    },
    {
      "epoch": 1.6579627990722658e-05,
      "step": 17385,
      "training_step_time": 17.09218192100525
    },
    {
      "epoch": 1.6580581665039062e-05,
      "model_forward_time": 5.608063697814941,
      "step": 17386
    },
    {
      "epoch": 1.6580581665039062e-05,
      "step": 17386,
      "training_step_time": 17.09415555000305
    },
    {
      "epoch": 1.658153533935547e-05,
      "model_forward_time": 5.611254930496216,
      "step": 17387
    },
    {
      "epoch": 1.658153533935547e-05,
      "step": 17387,
      "training_step_time": 17.111679792404175
    },
    {
      "epoch": 1.6582489013671877e-05,
      "model_forward_time": 5.599125623703003,
      "step": 17388
    },
    {
      "epoch": 1.6582489013671877e-05,
      "step": 17388,
      "training_step_time": 17.08293318748474
    },
    {
      "epoch": 1.658344268798828e-05,
      "model_forward_time": 5.609635353088379,
      "step": 17389
    },
    {
      "epoch": 1.658344268798828e-05,
      "step": 17389,
      "training_step_time": 17.09643530845642
    },
    {
      "epoch": 1.6584396362304688e-05,
      "grad_norm": 1.1932919025421143,
      "learning_rate": 1e-05,
      "loss": 0.0661,
      "step": 17390
    },
    {
      "epoch": 1.6584396362304688e-05,
      "model_forward_time": 5.98819899559021,
      "step": 17390
    },
    {
      "dynamics_loss_avg": 0.06356833660975099,
      "epoch": 1.6584396362304688e-05,
      "step": 17390
    },
    {
      "action_loss_avg": 0.008366634126286954,
      "epoch": 1.6584396362304688e-05,
      "step": 17390
    },
    {
      "epoch": 1.6584396362304688e-05,
      "step": 17390,
      "training_step_time": 17.488965034484863
    },
    {
      "epoch": 1.6585350036621092e-05,
      "model_forward_time": 5.611514091491699,
      "step": 17391
    },
    {
      "epoch": 1.6585350036621092e-05,
      "step": 17391,
      "training_step_time": 17.118146181106567
    },
    {
      "epoch": 1.65863037109375e-05,
      "model_forward_time": 5.608428001403809,
      "step": 17392
    },
    {
      "epoch": 1.65863037109375e-05,
      "step": 17392,
      "training_step_time": 17.082606077194214
    },
    {
      "epoch": 1.6587257385253907e-05,
      "model_forward_time": 5.596668004989624,
      "step": 17393
    },
    {
      "epoch": 1.6587257385253907e-05,
      "step": 17393,
      "training_step_time": 17.036113739013672
    },
    {
      "epoch": 1.658821105957031e-05,
      "model_forward_time": 5.598072290420532,
      "step": 17394
    },
    {
      "epoch": 1.658821105957031e-05,
      "step": 17394,
      "training_step_time": 17.028571844100952
    },
    {
      "epoch": 1.658916473388672e-05,
      "model_forward_time": 5.583175420761108,
      "step": 17395
    },
    {
      "epoch": 1.658916473388672e-05,
      "step": 17395,
      "training_step_time": 17.379034280776978
    },
    {
      "epoch": 1.6590118408203126e-05,
      "model_forward_time": 5.594388961791992,
      "step": 17396
    },
    {
      "epoch": 1.6590118408203126e-05,
      "step": 17396,
      "training_step_time": 17.053993463516235
    },
    {
      "epoch": 1.659107208251953e-05,
      "model_forward_time": 5.611846923828125,
      "step": 17397
    },
    {
      "epoch": 1.659107208251953e-05,
      "step": 17397,
      "training_step_time": 17.079246044158936
    },
    {
      "epoch": 1.6592025756835937e-05,
      "model_forward_time": 5.61397385597229,
      "step": 17398
    },
    {
      "epoch": 1.6592025756835937e-05,
      "step": 17398,
      "training_step_time": 17.101873636245728
    },
    {
      "epoch": 1.6592979431152345e-05,
      "model_forward_time": 5.610119104385376,
      "step": 17399
    },
    {
      "epoch": 1.6592979431152345e-05,
      "step": 17399,
      "training_step_time": 17.108999013900757
    },
    {
      "epoch": 1.659393310546875e-05,
      "grad_norm": 0.8863207697868347,
      "learning_rate": 1e-05,
      "loss": 0.0543,
      "step": 17400
    },
    {
      "epoch": 1.659393310546875e-05,
      "model_forward_time": 5.625499248504639,
      "step": 17400
    },
    {
      "dynamics_loss_avg": 0.04047118220478296,
      "epoch": 1.659393310546875e-05,
      "step": 17400
    },
    {
      "action_loss_avg": 0.005223744246177375,
      "epoch": 1.659393310546875e-05,
      "step": 17400
    },
    {
      "epoch": 1.659393310546875e-05,
      "step": 17400,
      "training_step_time": 17.108153581619263
    },
    {
      "epoch": 1.6594886779785156e-05,
      "model_forward_time": 5.964742660522461,
      "step": 17401
    },
    {
      "epoch": 1.6594886779785156e-05,
      "step": 17401,
      "training_step_time": 17.41508436203003
    },
    {
      "epoch": 1.6595840454101563e-05,
      "model_forward_time": 5.58536434173584,
      "step": 17402
    },
    {
      "epoch": 1.6595840454101563e-05,
      "step": 17402,
      "training_step_time": 17.00946855545044
    },
    {
      "epoch": 1.6596794128417968e-05,
      "model_forward_time": 5.5746684074401855,
      "step": 17403
    },
    {
      "epoch": 1.6596794128417968e-05,
      "step": 17403,
      "training_step_time": 16.995478868484497
    },
    {
      "epoch": 1.6597747802734375e-05,
      "model_forward_time": 5.5734333992004395,
      "step": 17404
    },
    {
      "epoch": 1.6597747802734375e-05,
      "step": 17404,
      "training_step_time": 17.013657331466675
    },
    {
      "epoch": 1.6598701477050782e-05,
      "model_forward_time": 5.596316576004028,
      "step": 17405
    },
    {
      "epoch": 1.6598701477050782e-05,
      "step": 17405,
      "training_step_time": 17.020078897476196
    },
    {
      "epoch": 1.6599655151367186e-05,
      "model_forward_time": 5.942913770675659,
      "step": 17406
    },
    {
      "epoch": 1.6599655151367186e-05,
      "step": 17406,
      "training_step_time": 17.358533143997192
    },
    {
      "epoch": 1.6600608825683594e-05,
      "model_forward_time": 5.5891273021698,
      "step": 17407
    },
    {
      "epoch": 1.6600608825683594e-05,
      "step": 17407,
      "training_step_time": 17.03101873397827
    },
    {
      "epoch": 1.66015625e-05,
      "model_forward_time": 5.615394115447998,
      "step": 17408
    },
    {
      "epoch": 1.66015625e-05,
      "step": 17408,
      "training_step_time": 17.080605030059814
    },
    {
      "epoch": 1.6602516174316405e-05,
      "model_forward_time": 5.6048877239227295,
      "step": 17409
    },
    {
      "epoch": 1.6602516174316405e-05,
      "step": 17409,
      "training_step_time": 17.083065032958984
    },
    {
      "epoch": 1.6603469848632813e-05,
      "grad_norm": 0.6523148417472839,
      "learning_rate": 1e-05,
      "loss": 0.0766,
      "step": 17410
    },
    {
      "epoch": 1.6603469848632813e-05,
      "model_forward_time": 5.611475706100464,
      "step": 17410
    },
    {
      "dynamics_loss_avg": 0.06817924845963716,
      "epoch": 1.6603469848632813e-05,
      "step": 17410
    },
    {
      "action_loss_avg": 0.007932622719090432,
      "epoch": 1.6603469848632813e-05,
      "step": 17410
    },
    {
      "epoch": 1.6603469848632813e-05,
      "step": 17410,
      "training_step_time": 17.113362312316895
    },
    {
      "epoch": 1.660442352294922e-05,
      "model_forward_time": 5.6237030029296875,
      "step": 17411
    },
    {
      "epoch": 1.660442352294922e-05,
      "step": 17411,
      "training_step_time": 17.11596179008484
    },
    {
      "epoch": 1.6605377197265624e-05,
      "model_forward_time": 5.983506679534912,
      "step": 17412
    },
    {
      "epoch": 1.6605377197265624e-05,
      "step": 17412,
      "training_step_time": 17.491386890411377
    },
    {
      "epoch": 1.660633087158203e-05,
      "model_forward_time": 5.623136281967163,
      "step": 17413
    },
    {
      "epoch": 1.660633087158203e-05,
      "step": 17413,
      "training_step_time": 17.103888034820557
    },
    {
      "epoch": 1.660728454589844e-05,
      "model_forward_time": 5.61008358001709,
      "step": 17414
    },
    {
      "epoch": 1.660728454589844e-05,
      "step": 17414,
      "training_step_time": 17.07070541381836
    },
    {
      "epoch": 1.6608238220214843e-05,
      "model_forward_time": 5.598499774932861,
      "step": 17415
    },
    {
      "epoch": 1.6608238220214843e-05,
      "step": 17415,
      "training_step_time": 17.037028789520264
    },
    {
      "epoch": 1.660919189453125e-05,
      "model_forward_time": 5.597341775894165,
      "step": 17416
    },
    {
      "epoch": 1.660919189453125e-05,
      "step": 17416,
      "training_step_time": 17.029701709747314
    },
    {
      "epoch": 1.6610145568847658e-05,
      "model_forward_time": 5.595165729522705,
      "step": 17417
    },
    {
      "epoch": 1.6610145568847658e-05,
      "step": 17417,
      "training_step_time": 17.406283140182495
    },
    {
      "epoch": 1.6611099243164062e-05,
      "model_forward_time": 5.615138530731201,
      "step": 17418
    },
    {
      "epoch": 1.6611099243164062e-05,
      "step": 17418,
      "training_step_time": 17.093941688537598
    },
    {
      "epoch": 1.661205291748047e-05,
      "model_forward_time": 5.613431215286255,
      "step": 17419
    },
    {
      "epoch": 1.661205291748047e-05,
      "step": 17419,
      "training_step_time": 17.114595651626587
    },
    {
      "epoch": 1.6613006591796877e-05,
      "grad_norm": 1.2806817293167114,
      "learning_rate": 1e-05,
      "loss": 0.0884,
      "step": 17420
    },
    {
      "epoch": 1.6613006591796877e-05,
      "model_forward_time": 5.618958473205566,
      "step": 17420
    },
    {
      "dynamics_loss_avg": 0.07739691976457834,
      "epoch": 1.6613006591796877e-05,
      "step": 17420
    },
    {
      "action_loss_avg": 0.010843066941015422,
      "epoch": 1.6613006591796877e-05,
      "step": 17420
    },
    {
      "epoch": 1.6613006591796877e-05,
      "step": 17420,
      "training_step_time": 17.1017804145813
    },
    {
      "epoch": 1.661396026611328e-05,
      "model_forward_time": 5.588885545730591,
      "step": 17421
    },
    {
      "epoch": 1.661396026611328e-05,
      "step": 17421,
      "training_step_time": 16.990959882736206
    },
    {
      "epoch": 1.6614913940429688e-05,
      "model_forward_time": 5.584233045578003,
      "step": 17422
    },
    {
      "epoch": 1.6614913940429688e-05,
      "step": 17422,
      "training_step_time": 17.022653102874756
    },
    {
      "epoch": 1.6615867614746092e-05,
      "model_forward_time": 5.9545958042144775,
      "step": 17423
    },
    {
      "epoch": 1.6615867614746092e-05,
      "step": 17423,
      "training_step_time": 17.392006158828735
    },
    {
      "epoch": 1.66168212890625e-05,
      "model_forward_time": 5.571566820144653,
      "step": 17424
    },
    {
      "epoch": 1.66168212890625e-05,
      "step": 17424,
      "training_step_time": 16.989947080612183
    },
    {
      "epoch": 1.6617774963378907e-05,
      "model_forward_time": 5.594315052032471,
      "step": 17425
    },
    {
      "epoch": 1.6617774963378907e-05,
      "step": 17425,
      "training_step_time": 17.021055936813354
    },
    {
      "epoch": 1.661872863769531e-05,
      "model_forward_time": 5.592710971832275,
      "step": 17426
    },
    {
      "epoch": 1.661872863769531e-05,
      "step": 17426,
      "training_step_time": 17.027820110321045
    },
    {
      "epoch": 1.661968231201172e-05,
      "model_forward_time": 5.590286493301392,
      "step": 17427
    },
    {
      "epoch": 1.661968231201172e-05,
      "step": 17427,
      "training_step_time": 17.023451566696167
    },
    {
      "epoch": 1.6620635986328126e-05,
      "model_forward_time": 5.600764513015747,
      "step": 17428
    },
    {
      "epoch": 1.6620635986328126e-05,
      "step": 17428,
      "training_step_time": 17.423804759979248
    },
    {
      "epoch": 1.662158966064453e-05,
      "model_forward_time": 5.60357141494751,
      "step": 17429
    },
    {
      "epoch": 1.662158966064453e-05,
      "step": 17429,
      "training_step_time": 17.09462022781372
    },
    {
      "epoch": 1.6622543334960937e-05,
      "grad_norm": 0.4954008162021637,
      "learning_rate": 1e-05,
      "loss": 0.0581,
      "step": 17430
    },
    {
      "epoch": 1.6622543334960937e-05,
      "model_forward_time": 5.623703718185425,
      "step": 17430
    },
    {
      "dynamics_loss_avg": 0.05896155945956707,
      "epoch": 1.6622543334960937e-05,
      "step": 17430
    },
    {
      "action_loss_avg": 0.005322405300103128,
      "epoch": 1.6622543334960937e-05,
      "step": 17430
    },
    {
      "epoch": 1.6622543334960937e-05,
      "step": 17430,
      "training_step_time": 17.13175678253174
    },
    {
      "epoch": 1.6623497009277345e-05,
      "model_forward_time": 5.622272491455078,
      "step": 17431
    },
    {
      "epoch": 1.6623497009277345e-05,
      "step": 17431,
      "training_step_time": 17.137032985687256
    },
    {
      "epoch": 1.662445068359375e-05,
      "model_forward_time": 5.623554706573486,
      "step": 17432
    },
    {
      "epoch": 1.662445068359375e-05,
      "step": 17432,
      "training_step_time": 17.118534803390503
    },
    {
      "epoch": 1.6625404357910156e-05,
      "model_forward_time": 5.5951385498046875,
      "step": 17433
    },
    {
      "epoch": 1.6625404357910156e-05,
      "step": 17433,
      "training_step_time": 17.025259494781494
    },
    {
      "epoch": 1.6626358032226564e-05,
      "model_forward_time": 5.946075439453125,
      "step": 17434
    },
    {
      "epoch": 1.6626358032226564e-05,
      "step": 17434,
      "training_step_time": 17.37745499610901
    },
    {
      "epoch": 1.6627311706542968e-05,
      "model_forward_time": 5.566589117050171,
      "step": 17435
    },
    {
      "epoch": 1.6627311706542968e-05,
      "step": 17435,
      "training_step_time": 17.003385305404663
    },
    {
      "epoch": 1.6628265380859375e-05,
      "model_forward_time": 5.578292608261108,
      "step": 17436
    },
    {
      "epoch": 1.6628265380859375e-05,
      "step": 17436,
      "training_step_time": 17.008291244506836
    },
    {
      "epoch": 1.6629219055175782e-05,
      "model_forward_time": 5.575047731399536,
      "step": 17437
    },
    {
      "epoch": 1.6629219055175782e-05,
      "step": 17437,
      "training_step_time": 16.998605489730835
    },
    {
      "epoch": 1.6630172729492186e-05,
      "model_forward_time": 5.574515104293823,
      "step": 17438
    },
    {
      "epoch": 1.6630172729492186e-05,
      "step": 17438,
      "training_step_time": 17.016380786895752
    },
    {
      "epoch": 1.6631126403808594e-05,
      "model_forward_time": 5.9303789138793945,
      "step": 17439
    },
    {
      "epoch": 1.6631126403808594e-05,
      "step": 17439,
      "training_step_time": 17.343340396881104
    },
    {
      "epoch": 1.6632080078125e-05,
      "grad_norm": 0.2875250577926636,
      "learning_rate": 1e-05,
      "loss": 0.074,
      "step": 17440
    },
    {
      "epoch": 1.6632080078125e-05,
      "model_forward_time": 5.590537786483765,
      "step": 17440
    },
    {
      "dynamics_loss_avg": 0.0620901906862855,
      "epoch": 1.6632080078125e-05,
      "step": 17440
    },
    {
      "action_loss_avg": 0.008047021413221955,
      "epoch": 1.6632080078125e-05,
      "step": 17440
    },
    {
      "epoch": 1.6632080078125e-05,
      "step": 17440,
      "training_step_time": 17.045727968215942
    },
    {
      "epoch": 1.6633033752441405e-05,
      "model_forward_time": 5.602693796157837,
      "step": 17441
    },
    {
      "epoch": 1.6633033752441405e-05,
      "step": 17441,
      "training_step_time": 17.094488620758057
    },
    {
      "epoch": 1.6633987426757813e-05,
      "model_forward_time": 5.614358901977539,
      "step": 17442
    },
    {
      "epoch": 1.6633987426757813e-05,
      "step": 17442,
      "training_step_time": 17.139053344726562
    },
    {
      "epoch": 1.663494110107422e-05,
      "model_forward_time": 5.623889446258545,
      "step": 17443
    },
    {
      "epoch": 1.663494110107422e-05,
      "step": 17443,
      "training_step_time": 17.12465763092041
    },
    {
      "epoch": 1.6635894775390624e-05,
      "model_forward_time": 5.610324859619141,
      "step": 17444
    },
    {
      "epoch": 1.6635894775390624e-05,
      "step": 17444,
      "training_step_time": 17.08307933807373
    },
    {
      "epoch": 1.663684844970703e-05,
      "model_forward_time": 5.953168869018555,
      "step": 17445
    },
    {
      "epoch": 1.663684844970703e-05,
      "step": 17445,
      "training_step_time": 17.372116565704346
    },
    {
      "epoch": 1.663780212402344e-05,
      "model_forward_time": 5.573947906494141,
      "step": 17446
    },
    {
      "epoch": 1.663780212402344e-05,
      "step": 17446,
      "training_step_time": 17.0097758769989
    },
    {
      "epoch": 1.6638755798339843e-05,
      "model_forward_time": 5.583039045333862,
      "step": 17447
    },
    {
      "epoch": 1.6638755798339843e-05,
      "step": 17447,
      "training_step_time": 17.008116483688354
    },
    {
      "epoch": 1.663970947265625e-05,
      "model_forward_time": 5.577626466751099,
      "step": 17448
    },
    {
      "epoch": 1.663970947265625e-05,
      "step": 17448,
      "training_step_time": 16.998358726501465
    },
    {
      "epoch": 1.6640663146972658e-05,
      "model_forward_time": 5.576574802398682,
      "step": 17449
    },
    {
      "epoch": 1.6640663146972658e-05,
      "step": 17449,
      "training_step_time": 16.99344491958618
    },
    {
      "epoch": 1.6641616821289062e-05,
      "grad_norm": 0.4012658894062042,
      "learning_rate": 1e-05,
      "loss": 0.0499,
      "step": 17450
    },
    {
      "epoch": 1.6641616821289062e-05,
      "model_forward_time": 5.581706523895264,
      "step": 17450
    },
    {
      "dynamics_loss_avg": 0.04875445105135441,
      "epoch": 1.6641616821289062e-05,
      "step": 17450
    },
    {
      "action_loss_avg": 0.004030740540474653,
      "epoch": 1.6641616821289062e-05,
      "step": 17450
    },
    {
      "epoch": 1.6641616821289062e-05,
      "step": 17450,
      "training_step_time": 17.394221305847168
    },
    {
      "epoch": 1.664257049560547e-05,
      "model_forward_time": 5.585989952087402,
      "step": 17451
    },
    {
      "epoch": 1.664257049560547e-05,
      "step": 17451,
      "training_step_time": 17.023860931396484
    },
    {
      "epoch": 1.6643524169921877e-05,
      "model_forward_time": 5.593053817749023,
      "step": 17452
    },
    {
      "epoch": 1.6643524169921877e-05,
      "step": 17452,
      "training_step_time": 17.034229040145874
    },
    {
      "epoch": 1.664447784423828e-05,
      "model_forward_time": 5.590675115585327,
      "step": 17453
    },
    {
      "epoch": 1.664447784423828e-05,
      "step": 17453,
      "training_step_time": 17.038024425506592
    },
    {
      "epoch": 1.6645431518554688e-05,
      "model_forward_time": 5.600211143493652,
      "step": 17454
    },
    {
      "epoch": 1.6645431518554688e-05,
      "step": 17454,
      "training_step_time": 17.06680655479431
    },
    {
      "epoch": 1.6646385192871092e-05,
      "model_forward_time": 5.610701084136963,
      "step": 17455
    },
    {
      "epoch": 1.6646385192871092e-05,
      "step": 17455,
      "training_step_time": 17.09981107711792
    },
    {
      "epoch": 1.66473388671875e-05,
      "model_forward_time": 5.971047878265381,
      "step": 17456
    },
    {
      "epoch": 1.66473388671875e-05,
      "step": 17456,
      "training_step_time": 17.447022199630737
    },
    {
      "epoch": 1.6648292541503907e-05,
      "model_forward_time": 5.602436542510986,
      "step": 17457
    },
    {
      "epoch": 1.6648292541503907e-05,
      "step": 17457,
      "training_step_time": 17.048338890075684
    },
    {
      "epoch": 1.664924621582031e-05,
      "model_forward_time": 5.591569185256958,
      "step": 17458
    },
    {
      "epoch": 1.664924621582031e-05,
      "step": 17458,
      "training_step_time": 17.041152715682983
    },
    {
      "epoch": 1.665019989013672e-05,
      "model_forward_time": 5.595273733139038,
      "step": 17459
    },
    {
      "epoch": 1.665019989013672e-05,
      "step": 17459,
      "training_step_time": 17.043579578399658
    },
    {
      "epoch": 1.6651153564453126e-05,
      "grad_norm": 0.8008778691291809,
      "learning_rate": 1e-05,
      "loss": 0.0743,
      "step": 17460
    },
    {
      "epoch": 1.6651153564453126e-05,
      "model_forward_time": 5.601905822753906,
      "step": 17460
    },
    {
      "dynamics_loss_avg": 0.06417166423052549,
      "epoch": 1.6651153564453126e-05,
      "step": 17460
    },
    {
      "action_loss_avg": 0.006734399497509003,
      "epoch": 1.6651153564453126e-05,
      "step": 17460
    },
    {
      "epoch": 1.6651153564453126e-05,
      "step": 17460,
      "training_step_time": 17.07929563522339
    },
    {
      "epoch": 1.665210723876953e-05,
      "model_forward_time": 5.605901718139648,
      "step": 17461
    },
    {
      "epoch": 1.665210723876953e-05,
      "step": 17461,
      "training_step_time": 17.45388627052307
    },
    {
      "epoch": 1.6653060913085937e-05,
      "model_forward_time": 5.623347043991089,
      "step": 17462
    },
    {
      "epoch": 1.6653060913085937e-05,
      "step": 17462,
      "training_step_time": 17.118265628814697
    },
    {
      "epoch": 1.6654014587402345e-05,
      "model_forward_time": 5.61443305015564,
      "step": 17463
    },
    {
      "epoch": 1.6654014587402345e-05,
      "step": 17463,
      "training_step_time": 17.105838298797607
    },
    {
      "epoch": 1.665496826171875e-05,
      "model_forward_time": 5.6093573570251465,
      "step": 17464
    },
    {
      "epoch": 1.665496826171875e-05,
      "step": 17464,
      "training_step_time": 17.077606201171875
    },
    {
      "epoch": 1.6655921936035156e-05,
      "model_forward_time": 5.593177795410156,
      "step": 17465
    },
    {
      "epoch": 1.6655921936035156e-05,
      "step": 17465,
      "training_step_time": 17.05757164955139
    },
    {
      "epoch": 1.6656875610351564e-05,
      "model_forward_time": 5.595905542373657,
      "step": 17466
    },
    {
      "epoch": 1.6656875610351564e-05,
      "step": 17466,
      "training_step_time": 17.023125171661377
    },
    {
      "epoch": 1.6657829284667968e-05,
      "model_forward_time": 5.972201347351074,
      "step": 17467
    },
    {
      "epoch": 1.6657829284667968e-05,
      "step": 17467,
      "training_step_time": 17.408190727233887
    },
    {
      "epoch": 1.6658782958984375e-05,
      "model_forward_time": 5.596127986907959,
      "step": 17468
    },
    {
      "epoch": 1.6658782958984375e-05,
      "step": 17468,
      "training_step_time": 17.065712213516235
    },
    {
      "epoch": 1.6659736633300782e-05,
      "model_forward_time": 5.5996153354644775,
      "step": 17469
    },
    {
      "epoch": 1.6659736633300782e-05,
      "step": 17469,
      "training_step_time": 17.03748345375061
    },
    {
      "epoch": 1.6660690307617186e-05,
      "grad_norm": 0.711544930934906,
      "learning_rate": 1e-05,
      "loss": 0.0614,
      "step": 17470
    },
    {
      "epoch": 1.6660690307617186e-05,
      "model_forward_time": 5.602067232131958,
      "step": 17470
    },
    {
      "dynamics_loss_avg": 0.05655103735625744,
      "epoch": 1.6660690307617186e-05,
      "step": 17470
    },
    {
      "action_loss_avg": 0.00705723853316158,
      "epoch": 1.6660690307617186e-05,
      "step": 17470
    },
    {
      "epoch": 1.6660690307617186e-05,
      "step": 17470,
      "training_step_time": 17.079001903533936
    },
    {
      "epoch": 1.6661643981933594e-05,
      "model_forward_time": 5.6098549365997314,
      "step": 17471
    },
    {
      "epoch": 1.6661643981933594e-05,
      "step": 17471,
      "training_step_time": 17.09236741065979
    },
    {
      "epoch": 1.666259765625e-05,
      "model_forward_time": 5.977948188781738,
      "step": 17472
    },
    {
      "epoch": 1.666259765625e-05,
      "step": 17472,
      "training_step_time": 17.463684797286987
    },
    {
      "epoch": 1.6663551330566405e-05,
      "model_forward_time": 5.61211895942688,
      "step": 17473
    },
    {
      "epoch": 1.6663551330566405e-05,
      "step": 17473,
      "training_step_time": 17.080286502838135
    },
    {
      "epoch": 1.6664505004882813e-05,
      "model_forward_time": 5.604692697525024,
      "step": 17474
    },
    {
      "epoch": 1.6664505004882813e-05,
      "step": 17474,
      "training_step_time": 17.077667713165283
    },
    {
      "epoch": 1.666545867919922e-05,
      "model_forward_time": 5.5969812870025635,
      "step": 17475
    },
    {
      "epoch": 1.666545867919922e-05,
      "step": 17475,
      "training_step_time": 17.05523133277893
    },
    {
      "epoch": 1.6666412353515624e-05,
      "model_forward_time": 5.592831373214722,
      "step": 17476
    },
    {
      "epoch": 1.6666412353515624e-05,
      "step": 17476,
      "training_step_time": 17.040287256240845
    },
    {
      "epoch": 1.666736602783203e-05,
      "model_forward_time": 5.5926878452301025,
      "step": 17477
    },
    {
      "epoch": 1.666736602783203e-05,
      "step": 17477,
      "training_step_time": 17.04706048965454
    },
    {
      "epoch": 1.666831970214844e-05,
      "model_forward_time": 5.970172882080078,
      "step": 17478
    },
    {
      "epoch": 1.666831970214844e-05,
      "step": 17478,
      "training_step_time": 17.418697595596313
    },
    {
      "epoch": 1.6669273376464843e-05,
      "model_forward_time": 5.597252368927002,
      "step": 17479
    },
    {
      "epoch": 1.6669273376464843e-05,
      "step": 17479,
      "training_step_time": 17.040616512298584
    },
    {
      "epoch": 1.667022705078125e-05,
      "grad_norm": 0.5601953268051147,
      "learning_rate": 1e-05,
      "loss": 0.0572,
      "step": 17480
    },
    {
      "epoch": 1.667022705078125e-05,
      "model_forward_time": 5.598536014556885,
      "step": 17480
    },
    {
      "dynamics_loss_avg": 0.0527075532823801,
      "epoch": 1.667022705078125e-05,
      "step": 17480
    },
    {
      "action_loss_avg": 0.006050954828970135,
      "epoch": 1.667022705078125e-05,
      "step": 17480
    },
    {
      "epoch": 1.667022705078125e-05,
      "step": 17480,
      "training_step_time": 17.03992748260498
    },
    {
      "epoch": 1.6671180725097658e-05,
      "model_forward_time": 5.6178977489471436,
      "step": 17481
    },
    {
      "epoch": 1.6671180725097658e-05,
      "step": 17481,
      "training_step_time": 17.100504636764526
    },
    {
      "epoch": 1.6672134399414062e-05,
      "model_forward_time": 5.604254961013794,
      "step": 17482
    },
    {
      "epoch": 1.6672134399414062e-05,
      "step": 17482,
      "training_step_time": 17.086821794509888
    },
    {
      "epoch": 1.667308807373047e-05,
      "model_forward_time": 5.6033525466918945,
      "step": 17483
    },
    {
      "epoch": 1.667308807373047e-05,
      "step": 17483,
      "training_step_time": 17.441871881484985
    },
    {
      "epoch": 1.6674041748046877e-05,
      "model_forward_time": 5.601452589035034,
      "step": 17484
    },
    {
      "epoch": 1.6674041748046877e-05,
      "step": 17484,
      "training_step_time": 17.047781229019165
    },
    {
      "epoch": 1.667499542236328e-05,
      "model_forward_time": 5.588689804077148,
      "step": 17485
    },
    {
      "epoch": 1.667499542236328e-05,
      "step": 17485,
      "training_step_time": 17.010955810546875
    },
    {
      "epoch": 1.6675949096679688e-05,
      "model_forward_time": 5.591259002685547,
      "step": 17486
    },
    {
      "epoch": 1.6675949096679688e-05,
      "step": 17486,
      "training_step_time": 17.03112506866455
    },
    {
      "epoch": 1.6676902770996092e-05,
      "model_forward_time": 5.599188327789307,
      "step": 17487
    },
    {
      "epoch": 1.6676902770996092e-05,
      "step": 17487,
      "training_step_time": 17.06759214401245
    },
    {
      "epoch": 1.66778564453125e-05,
      "model_forward_time": 5.596314907073975,
      "step": 17488
    },
    {
      "epoch": 1.66778564453125e-05,
      "step": 17488,
      "training_step_time": 17.075488805770874
    },
    {
      "epoch": 1.6678810119628907e-05,
      "model_forward_time": 5.980557203292847,
      "step": 17489
    },
    {
      "epoch": 1.6678810119628907e-05,
      "step": 17489,
      "training_step_time": 17.475054264068604
    },
    {
      "epoch": 1.667976379394531e-05,
      "grad_norm": 0.16178929805755615,
      "learning_rate": 1e-05,
      "loss": 0.0594,
      "step": 17490
    },
    {
      "epoch": 1.667976379394531e-05,
      "model_forward_time": 5.607008695602417,
      "step": 17490
    },
    {
      "dynamics_loss_avg": 0.047143211122602224,
      "epoch": 1.667976379394531e-05,
      "step": 17490
    },
    {
      "action_loss_avg": 0.006765138334594667,
      "epoch": 1.667976379394531e-05,
      "step": 17490
    },
    {
      "epoch": 1.667976379394531e-05,
      "step": 17490,
      "training_step_time": 17.121729135513306
    },
    {
      "epoch": 1.668071746826172e-05,
      "model_forward_time": 5.6191864013671875,
      "step": 17491
    },
    {
      "epoch": 1.668071746826172e-05,
      "step": 17491,
      "training_step_time": 17.135364294052124
    },
    {
      "epoch": 1.6681671142578126e-05,
      "model_forward_time": 5.623770236968994,
      "step": 17492
    },
    {
      "epoch": 1.6681671142578126e-05,
      "step": 17492,
      "training_step_time": 17.10973572731018
    },
    {
      "epoch": 1.668262481689453e-05,
      "model_forward_time": 5.598990201950073,
      "step": 17493
    },
    {
      "epoch": 1.668262481689453e-05,
      "step": 17493,
      "training_step_time": 17.039284706115723
    },
    {
      "epoch": 1.6683578491210937e-05,
      "model_forward_time": 5.5832085609436035,
      "step": 17494
    },
    {
      "epoch": 1.6683578491210937e-05,
      "step": 17494,
      "training_step_time": 17.38196635246277
    },
    {
      "epoch": 1.6684532165527345e-05,
      "model_forward_time": 5.590961456298828,
      "step": 17495
    },
    {
      "epoch": 1.6684532165527345e-05,
      "step": 17495,
      "training_step_time": 17.028759479522705
    },
    {
      "epoch": 1.668548583984375e-05,
      "model_forward_time": 5.597463369369507,
      "step": 17496
    },
    {
      "epoch": 1.668548583984375e-05,
      "step": 17496,
      "training_step_time": 17.056419372558594
    },
    {
      "epoch": 1.6686439514160156e-05,
      "model_forward_time": 5.597222805023193,
      "step": 17497
    },
    {
      "epoch": 1.6686439514160156e-05,
      "step": 17497,
      "training_step_time": 17.090425968170166
    },
    {
      "epoch": 1.6687393188476564e-05,
      "model_forward_time": 5.62049674987793,
      "step": 17498
    },
    {
      "epoch": 1.6687393188476564e-05,
      "step": 17498,
      "training_step_time": 17.134120225906372
    },
    {
      "epoch": 1.6688346862792968e-05,
      "model_forward_time": 5.624555587768555,
      "step": 17499
    },
    {
      "epoch": 1.6688346862792968e-05,
      "step": 17499,
      "training_step_time": 17.11409902572632
    },
    {
      "epoch": 1.6689300537109375e-05,
      "grad_norm": 0.594373881816864,
      "learning_rate": 1e-05,
      "loss": 0.0535,
      "step": 17500
    },
    {
      "epoch": 1.6689300537109375e-05,
      "model_forward_time": 6.0053277015686035,
      "step": 17500
    },
    {
      "dynamics_loss_avg": 0.05254535712301731,
      "epoch": 1.6689300537109375e-05,
      "step": 17500
    },
    {
      "action_loss_avg": 0.006373455096036196,
      "epoch": 1.6689300537109375e-05,
      "step": 17500
    },
    {
      "epoch": 1.6689300537109375e-05,
      "step": 17500,
      "training_step_time": 17.500072240829468
    },
    {
      "epoch": 1.6690254211425782e-05,
      "model_forward_time": 5.600364923477173,
      "step": 17501
    },
    {
      "epoch": 1.6690254211425782e-05,
      "step": 17501,
      "training_step_time": 17.076056957244873
    },
    {
      "epoch": 1.6691207885742186e-05,
      "model_forward_time": 5.615957975387573,
      "step": 17502
    },
    {
      "epoch": 1.6691207885742186e-05,
      "step": 17502,
      "training_step_time": 17.09854745864868
    },
    {
      "epoch": 1.6692161560058594e-05,
      "model_forward_time": 5.597511053085327,
      "step": 17503
    },
    {
      "epoch": 1.6692161560058594e-05,
      "step": 17503,
      "training_step_time": 17.063417434692383
    },
    {
      "epoch": 1.6693115234375e-05,
      "model_forward_time": 5.605202674865723,
      "step": 17504
    },
    {
      "epoch": 1.6693115234375e-05,
      "step": 17504,
      "training_step_time": 17.067913055419922
    },
    {
      "epoch": 1.6694068908691405e-05,
      "model_forward_time": 5.953948259353638,
      "step": 17505
    },
    {
      "epoch": 1.6694068908691405e-05,
      "step": 17505,
      "training_step_time": 17.43204092979431
    },
    {
      "epoch": 1.6695022583007813e-05,
      "model_forward_time": 5.610848903656006,
      "step": 17506
    },
    {
      "epoch": 1.6695022583007813e-05,
      "step": 17506,
      "training_step_time": 17.11776828765869
    },
    {
      "epoch": 1.669597625732422e-05,
      "model_forward_time": 5.621518611907959,
      "step": 17507
    },
    {
      "epoch": 1.669597625732422e-05,
      "step": 17507,
      "training_step_time": 17.12762761116028
    },
    {
      "epoch": 1.6696929931640624e-05,
      "model_forward_time": 5.624823808670044,
      "step": 17508
    },
    {
      "epoch": 1.6696929931640624e-05,
      "step": 17508,
      "training_step_time": 17.127297163009644
    },
    {
      "epoch": 1.669788360595703e-05,
      "model_forward_time": 5.620157480239868,
      "step": 17509
    },
    {
      "epoch": 1.669788360595703e-05,
      "step": 17509,
      "training_step_time": 17.123257160186768
    },
    {
      "epoch": 1.669883728027344e-05,
      "grad_norm": 0.9074530601501465,
      "learning_rate": 1e-05,
      "loss": 0.0642,
      "step": 17510
    },
    {
      "epoch": 1.669883728027344e-05,
      "model_forward_time": 5.616243124008179,
      "step": 17510
    },
    {
      "dynamics_loss_avg": 0.05481460206210613,
      "epoch": 1.669883728027344e-05,
      "step": 17510
    },
    {
      "action_loss_avg": 0.007635411177761853,
      "epoch": 1.669883728027344e-05,
      "step": 17510
    },
    {
      "epoch": 1.669883728027344e-05,
      "step": 17510,
      "training_step_time": 17.117859840393066
    },
    {
      "epoch": 1.6699790954589843e-05,
      "model_forward_time": 5.98021125793457,
      "step": 17511
    },
    {
      "epoch": 1.6699790954589843e-05,
      "step": 17511,
      "training_step_time": 17.46177649497986
    },
    {
      "epoch": 1.670074462890625e-05,
      "model_forward_time": 5.639171600341797,
      "step": 17512
    },
    {
      "epoch": 1.670074462890625e-05,
      "step": 17512,
      "training_step_time": 17.070691347122192
    },
    {
      "epoch": 1.6701698303222658e-05,
      "model_forward_time": 5.590045690536499,
      "step": 17513
    },
    {
      "epoch": 1.6701698303222658e-05,
      "step": 17513,
      "training_step_time": 17.005268573760986
    },
    {
      "epoch": 1.6702651977539062e-05,
      "model_forward_time": 5.572944402694702,
      "step": 17514
    },
    {
      "epoch": 1.6702651977539062e-05,
      "step": 17514,
      "training_step_time": 17.010645627975464
    },
    {
      "epoch": 1.670360565185547e-05,
      "model_forward_time": 5.567731618881226,
      "step": 17515
    },
    {
      "epoch": 1.670360565185547e-05,
      "step": 17515,
      "training_step_time": 16.98076581954956
    },
    {
      "epoch": 1.6704559326171873e-05,
      "model_forward_time": 5.572568416595459,
      "step": 17516
    },
    {
      "epoch": 1.6704559326171873e-05,
      "step": 17516,
      "training_step_time": 17.33745551109314
    },
    {
      "epoch": 1.670551300048828e-05,
      "model_forward_time": 5.576871395111084,
      "step": 17517
    },
    {
      "epoch": 1.670551300048828e-05,
      "step": 17517,
      "training_step_time": 16.99823236465454
    },
    {
      "epoch": 1.6706466674804688e-05,
      "model_forward_time": 5.583128213882446,
      "step": 17518
    },
    {
      "epoch": 1.6706466674804688e-05,
      "step": 17518,
      "training_step_time": 17.01752781867981
    },
    {
      "epoch": 1.6707420349121092e-05,
      "model_forward_time": 5.586701154708862,
      "step": 17519
    },
    {
      "epoch": 1.6707420349121092e-05,
      "step": 17519,
      "training_step_time": 16.99503254890442
    },
    {
      "epoch": 1.67083740234375e-05,
      "grad_norm": 0.7325407862663269,
      "learning_rate": 1e-05,
      "loss": 0.0524,
      "step": 17520
    },
    {
      "epoch": 1.67083740234375e-05,
      "model_forward_time": 5.586907863616943,
      "step": 17520
    },
    {
      "dynamics_loss_avg": 0.050456498935818675,
      "epoch": 1.67083740234375e-05,
      "step": 17520
    },
    {
      "action_loss_avg": 0.006268381734844297,
      "epoch": 1.67083740234375e-05,
      "step": 17520
    },
    {
      "epoch": 1.67083740234375e-05,
      "step": 17520,
      "training_step_time": 17.055822372436523
    },
    {
      "epoch": 1.6709327697753907e-05,
      "model_forward_time": 5.599405527114868,
      "step": 17521
    },
    {
      "epoch": 1.6709327697753907e-05,
      "step": 17521,
      "training_step_time": 17.098528385162354
    },
    {
      "epoch": 1.671028137207031e-05,
      "model_forward_time": 5.980315208435059,
      "step": 17522
    },
    {
      "epoch": 1.671028137207031e-05,
      "step": 17522,
      "training_step_time": 17.4994854927063
    },
    {
      "epoch": 1.671123504638672e-05,
      "model_forward_time": 5.634028673171997,
      "step": 17523
    },
    {
      "epoch": 1.671123504638672e-05,
      "step": 17523,
      "training_step_time": 17.153738021850586
    },
    {
      "epoch": 1.6712188720703126e-05,
      "model_forward_time": 5.632539749145508,
      "step": 17524
    },
    {
      "epoch": 1.6712188720703126e-05,
      "step": 17524,
      "training_step_time": 17.15541958808899
    },
    {
      "epoch": 1.671314239501953e-05,
      "model_forward_time": 5.627299070358276,
      "step": 17525
    },
    {
      "epoch": 1.671314239501953e-05,
      "step": 17525,
      "training_step_time": 17.13520646095276
    },
    {
      "epoch": 1.6714096069335937e-05,
      "model_forward_time": 5.610743045806885,
      "step": 17526
    },
    {
      "epoch": 1.6714096069335937e-05,
      "step": 17526,
      "training_step_time": 17.102105617523193
    },
    {
      "epoch": 1.6715049743652345e-05,
      "model_forward_time": 5.6121673583984375,
      "step": 17527
    },
    {
      "epoch": 1.6715049743652345e-05,
      "step": 17527,
      "training_step_time": 17.450594663619995
    },
    {
      "epoch": 1.671600341796875e-05,
      "model_forward_time": 5.586101293563843,
      "step": 17528
    },
    {
      "epoch": 1.671600341796875e-05,
      "step": 17528,
      "training_step_time": 17.001576900482178
    },
    {
      "epoch": 1.6716957092285156e-05,
      "model_forward_time": 5.570929765701294,
      "step": 17529
    },
    {
      "epoch": 1.6716957092285156e-05,
      "step": 17529,
      "training_step_time": 16.992637872695923
    },
    {
      "epoch": 1.6717910766601564e-05,
      "grad_norm": 0.7323329448699951,
      "learning_rate": 1e-05,
      "loss": 0.0679,
      "step": 17530
    },
    {
      "epoch": 1.6717910766601564e-05,
      "model_forward_time": 5.566635370254517,
      "step": 17530
    },
    {
      "dynamics_loss_avg": 0.0542677229270339,
      "epoch": 1.6717910766601564e-05,
      "step": 17530
    },
    {
      "action_loss_avg": 0.006761189526878297,
      "epoch": 1.6717910766601564e-05,
      "step": 17530
    },
    {
      "epoch": 1.6717910766601564e-05,
      "step": 17530,
      "training_step_time": 16.99082064628601
    },
    {
      "epoch": 1.6718864440917968e-05,
      "model_forward_time": 5.57686972618103,
      "step": 17531
    },
    {
      "epoch": 1.6718864440917968e-05,
      "step": 17531,
      "training_step_time": 16.980066776275635
    },
    {
      "epoch": 1.6719818115234375e-05,
      "model_forward_time": 5.574687242507935,
      "step": 17532
    },
    {
      "epoch": 1.6719818115234375e-05,
      "step": 17532,
      "training_step_time": 16.98543620109558
    },
    {
      "epoch": 1.6720771789550782e-05,
      "model_forward_time": 5.949970006942749,
      "step": 17533
    },
    {
      "epoch": 1.6720771789550782e-05,
      "step": 17533,
      "training_step_time": 17.387492656707764
    },
    {
      "epoch": 1.6721725463867186e-05,
      "model_forward_time": 5.5646281242370605,
      "step": 17534
    },
    {
      "epoch": 1.6721725463867186e-05,
      "step": 17534,
      "training_step_time": 16.981460332870483
    },
    {
      "epoch": 1.6722679138183594e-05,
      "model_forward_time": 5.600029945373535,
      "step": 17535
    },
    {
      "epoch": 1.6722679138183594e-05,
      "step": 17535,
      "training_step_time": 17.052066326141357
    },
    {
      "epoch": 1.67236328125e-05,
      "model_forward_time": 5.599845886230469,
      "step": 17536
    },
    {
      "epoch": 1.67236328125e-05,
      "step": 17536,
      "training_step_time": 17.067286729812622
    },
    {
      "epoch": 1.6724586486816405e-05,
      "model_forward_time": 5.605023384094238,
      "step": 17537
    },
    {
      "epoch": 1.6724586486816405e-05,
      "step": 17537,
      "training_step_time": 17.088618516921997
    },
    {
      "epoch": 1.6725540161132813e-05,
      "model_forward_time": 5.961144208908081,
      "step": 17538
    },
    {
      "epoch": 1.6725540161132813e-05,
      "step": 17538,
      "training_step_time": 17.416671752929688
    },
    {
      "epoch": 1.672649383544922e-05,
      "model_forward_time": 5.590729236602783,
      "step": 17539
    },
    {
      "epoch": 1.672649383544922e-05,
      "step": 17539,
      "training_step_time": 17.010975122451782
    },
    {
      "epoch": 1.6727447509765624e-05,
      "grad_norm": 0.36810818314552307,
      "learning_rate": 1e-05,
      "loss": 0.0468,
      "step": 17540
    },
    {
      "epoch": 1.6727447509765624e-05,
      "model_forward_time": 5.584194183349609,
      "step": 17540
    },
    {
      "dynamics_loss_avg": 0.04258902836591005,
      "epoch": 1.6727447509765624e-05,
      "step": 17540
    },
    {
      "action_loss_avg": 0.0056335028726607565,
      "epoch": 1.6727447509765624e-05,
      "step": 17540
    },
    {
      "epoch": 1.6727447509765624e-05,
      "step": 17540,
      "training_step_time": 16.993210792541504
    },
    {
      "epoch": 1.672840118408203e-05,
      "model_forward_time": 5.58194375038147,
      "step": 17541
    },
    {
      "epoch": 1.672840118408203e-05,
      "step": 17541,
      "training_step_time": 17.023720026016235
    },
    {
      "epoch": 1.672935485839844e-05,
      "model_forward_time": 5.569125175476074,
      "step": 17542
    },
    {
      "epoch": 1.672935485839844e-05,
      "step": 17542,
      "training_step_time": 16.976620197296143
    },
    {
      "epoch": 1.6730308532714843e-05,
      "model_forward_time": 5.571039199829102,
      "step": 17543
    },
    {
      "epoch": 1.6730308532714843e-05,
      "step": 17543,
      "training_step_time": 16.996148109436035
    },
    {
      "epoch": 1.673126220703125e-05,
      "model_forward_time": 5.945976257324219,
      "step": 17544
    },
    {
      "epoch": 1.673126220703125e-05,
      "step": 17544,
      "training_step_time": 17.368509769439697
    },
    {
      "epoch": 1.6732215881347658e-05,
      "model_forward_time": 5.577314853668213,
      "step": 17545
    },
    {
      "epoch": 1.6732215881347658e-05,
      "step": 17545,
      "training_step_time": 17.02380633354187
    },
    {
      "epoch": 1.6733169555664062e-05,
      "model_forward_time": 5.576418399810791,
      "step": 17546
    },
    {
      "epoch": 1.6733169555664062e-05,
      "step": 17546,
      "training_step_time": 17.00258994102478
    },
    {
      "epoch": 1.673412322998047e-05,
      "model_forward_time": 5.5921666622161865,
      "step": 17547
    },
    {
      "epoch": 1.673412322998047e-05,
      "step": 17547,
      "training_step_time": 17.05454158782959
    },
    {
      "epoch": 1.6735076904296873e-05,
      "model_forward_time": 5.610820531845093,
      "step": 17548
    },
    {
      "epoch": 1.6735076904296873e-05,
      "step": 17548,
      "training_step_time": 17.093870162963867
    },
    {
      "epoch": 1.673603057861328e-05,
      "model_forward_time": 5.6050660610198975,
      "step": 17549
    },
    {
      "epoch": 1.673603057861328e-05,
      "step": 17549,
      "training_step_time": 17.42496156692505
    },
    {
      "epoch": 1.6736984252929688e-05,
      "grad_norm": 0.708463728427887,
      "learning_rate": 1e-05,
      "loss": 0.0503,
      "step": 17550
    },
    {
      "epoch": 1.6736984252929688e-05,
      "model_forward_time": 5.60254430770874,
      "step": 17550
    },
    {
      "dynamics_loss_avg": 0.04492178149521351,
      "epoch": 1.6736984252929688e-05,
      "step": 17550
    },
    {
      "action_loss_avg": 0.004867982992436737,
      "epoch": 1.6736984252929688e-05,
      "step": 17550
    },
    {
      "epoch": 1.6736984252929688e-05,
      "step": 17550,
      "training_step_time": 17.074490547180176
    },
    {
      "epoch": 1.6737937927246092e-05,
      "model_forward_time": 5.594703912734985,
      "step": 17551
    },
    {
      "epoch": 1.6737937927246092e-05,
      "step": 17551,
      "training_step_time": 17.046363353729248
    },
    {
      "epoch": 1.67388916015625e-05,
      "model_forward_time": 5.5968663692474365,
      "step": 17552
    },
    {
      "epoch": 1.67388916015625e-05,
      "step": 17552,
      "training_step_time": 17.051278829574585
    },
    {
      "epoch": 1.6739845275878907e-05,
      "model_forward_time": 5.598955869674683,
      "step": 17553
    },
    {
      "epoch": 1.6739845275878907e-05,
      "step": 17553,
      "training_step_time": 17.06875991821289
    },
    {
      "epoch": 1.674079895019531e-05,
      "model_forward_time": 5.593400001525879,
      "step": 17554
    },
    {
      "epoch": 1.674079895019531e-05,
      "step": 17554,
      "training_step_time": 17.056010961532593
    },
    {
      "epoch": 1.674175262451172e-05,
      "model_forward_time": 5.9747068881988525,
      "step": 17555
    },
    {
      "epoch": 1.674175262451172e-05,
      "step": 17555,
      "training_step_time": 17.45384430885315
    },
    {
      "epoch": 1.6742706298828126e-05,
      "model_forward_time": 5.6102135181427,
      "step": 17556
    },
    {
      "epoch": 1.6742706298828126e-05,
      "step": 17556,
      "training_step_time": 17.12268352508545
    },
    {
      "epoch": 1.674365997314453e-05,
      "model_forward_time": 5.628392457962036,
      "step": 17557
    },
    {
      "epoch": 1.674365997314453e-05,
      "step": 17557,
      "training_step_time": 17.126152992248535
    },
    {
      "epoch": 1.6744613647460937e-05,
      "model_forward_time": 5.609356641769409,
      "step": 17558
    },
    {
      "epoch": 1.6744613647460937e-05,
      "step": 17558,
      "training_step_time": 17.06890344619751
    },
    {
      "epoch": 1.6745567321777345e-05,
      "model_forward_time": 5.596959352493286,
      "step": 17559
    },
    {
      "epoch": 1.6745567321777345e-05,
      "step": 17559,
      "training_step_time": 17.029331922531128
    },
    {
      "epoch": 1.674652099609375e-05,
      "grad_norm": 0.7403408288955688,
      "learning_rate": 1e-05,
      "loss": 0.0635,
      "step": 17560
    },
    {
      "epoch": 1.674652099609375e-05,
      "model_forward_time": 5.580107927322388,
      "step": 17560
    },
    {
      "dynamics_loss_avg": 0.06292346660047769,
      "epoch": 1.674652099609375e-05,
      "step": 17560
    },
    {
      "action_loss_avg": 0.007672527362592519,
      "epoch": 1.674652099609375e-05,
      "step": 17560
    },
    {
      "epoch": 1.674652099609375e-05,
      "step": 17560,
      "training_step_time": 17.38779592514038
    },
    {
      "epoch": 1.6747474670410156e-05,
      "model_forward_time": 5.5906455516815186,
      "step": 17561
    },
    {
      "epoch": 1.6747474670410156e-05,
      "step": 17561,
      "training_step_time": 17.057780742645264
    },
    {
      "epoch": 1.6748428344726564e-05,
      "model_forward_time": 5.603031396865845,
      "step": 17562
    },
    {
      "epoch": 1.6748428344726564e-05,
      "step": 17562,
      "training_step_time": 17.106980323791504
    },
    {
      "epoch": 1.6749382019042968e-05,
      "model_forward_time": 5.616191625595093,
      "step": 17563
    },
    {
      "epoch": 1.6749382019042968e-05,
      "step": 17563,
      "training_step_time": 17.12298583984375
    },
    {
      "epoch": 1.6750335693359375e-05,
      "model_forward_time": 5.614117860794067,
      "step": 17564
    },
    {
      "epoch": 1.6750335693359375e-05,
      "step": 17564,
      "training_step_time": 17.105984687805176
    },
    {
      "epoch": 1.6751289367675782e-05,
      "model_forward_time": 5.591078758239746,
      "step": 17565
    },
    {
      "epoch": 1.6751289367675782e-05,
      "step": 17565,
      "training_step_time": 17.04590678215027
    },
    {
      "epoch": 1.6752243041992186e-05,
      "model_forward_time": 5.954526424407959,
      "step": 17566
    },
    {
      "epoch": 1.6752243041992186e-05,
      "step": 17566,
      "training_step_time": 17.40739154815674
    },
    {
      "epoch": 1.6753196716308594e-05,
      "model_forward_time": 5.594411373138428,
      "step": 17567
    },
    {
      "epoch": 1.6753196716308594e-05,
      "step": 17567,
      "training_step_time": 17.033514499664307
    },
    {
      "epoch": 1.6754150390625e-05,
      "model_forward_time": 5.595542907714844,
      "step": 17568
    },
    {
      "epoch": 1.6754150390625e-05,
      "step": 17568,
      "training_step_time": 17.074397802352905
    },
    {
      "epoch": 1.6755104064941405e-05,
      "model_forward_time": 5.61461877822876,
      "step": 17569
    },
    {
      "epoch": 1.6755104064941405e-05,
      "step": 17569,
      "training_step_time": 17.113905668258667
    },
    {
      "epoch": 1.6756057739257813e-05,
      "grad_norm": 0.892742395401001,
      "learning_rate": 1e-05,
      "loss": 0.0756,
      "step": 17570
    },
    {
      "epoch": 1.6756057739257813e-05,
      "model_forward_time": 5.596044540405273,
      "step": 17570
    },
    {
      "dynamics_loss_avg": 0.06806452739983797,
      "epoch": 1.6756057739257813e-05,
      "step": 17570
    },
    {
      "action_loss_avg": 0.008925928547978402,
      "epoch": 1.6756057739257813e-05,
      "step": 17570
    },
    {
      "epoch": 1.6756057739257813e-05,
      "step": 17570,
      "training_step_time": 17.04339623451233
    },
    {
      "epoch": 1.675701141357422e-05,
      "model_forward_time": 5.94222354888916,
      "step": 17571
    },
    {
      "epoch": 1.675701141357422e-05,
      "step": 17571,
      "training_step_time": 17.374292373657227
    },
    {
      "epoch": 1.6757965087890624e-05,
      "model_forward_time": 5.587001323699951,
      "step": 17572
    },
    {
      "epoch": 1.6757965087890624e-05,
      "step": 17572,
      "training_step_time": 17.033008575439453
    },
    {
      "epoch": 1.675891876220703e-05,
      "model_forward_time": 5.605295181274414,
      "step": 17573
    },
    {
      "epoch": 1.675891876220703e-05,
      "step": 17573,
      "training_step_time": 17.081017017364502
    },
    {
      "epoch": 1.675987243652344e-05,
      "model_forward_time": 5.617079019546509,
      "step": 17574
    },
    {
      "epoch": 1.675987243652344e-05,
      "step": 17574,
      "training_step_time": 17.12679934501648
    },
    {
      "epoch": 1.6760826110839843e-05,
      "model_forward_time": 5.623096942901611,
      "step": 17575
    },
    {
      "epoch": 1.6760826110839843e-05,
      "step": 17575,
      "training_step_time": 17.10630226135254
    },
    {
      "epoch": 1.676177978515625e-05,
      "model_forward_time": 5.6302125453948975,
      "step": 17576
    },
    {
      "epoch": 1.676177978515625e-05,
      "step": 17576,
      "training_step_time": 17.122811317443848
    },
    {
      "epoch": 1.6762733459472658e-05,
      "model_forward_time": 5.98431658744812,
      "step": 17577
    },
    {
      "epoch": 1.6762733459472658e-05,
      "step": 17577,
      "training_step_time": 17.479254245758057
    },
    {
      "epoch": 1.6763687133789062e-05,
      "model_forward_time": 5.621062994003296,
      "step": 17578
    },
    {
      "epoch": 1.6763687133789062e-05,
      "step": 17578,
      "training_step_time": 17.098533391952515
    },
    {
      "epoch": 1.676464080810547e-05,
      "model_forward_time": 5.5883262157440186,
      "step": 17579
    },
    {
      "epoch": 1.676464080810547e-05,
      "step": 17579,
      "training_step_time": 17.034289121627808
    },
    {
      "epoch": 1.6765594482421873e-05,
      "grad_norm": 0.4891323447227478,
      "learning_rate": 1e-05,
      "loss": 0.0749,
      "step": 17580
    },
    {
      "epoch": 1.6765594482421873e-05,
      "model_forward_time": 5.566643714904785,
      "step": 17580
    },
    {
      "dynamics_loss_avg": 0.059110558778047564,
      "epoch": 1.6765594482421873e-05,
      "step": 17580
    },
    {
      "action_loss_avg": 0.006637514871545136,
      "epoch": 1.6765594482421873e-05,
      "step": 17580
    },
    {
      "epoch": 1.6765594482421873e-05,
      "step": 17580,
      "training_step_time": 16.976458311080933
    },
    {
      "epoch": 1.676654815673828e-05,
      "model_forward_time": 5.578611850738525,
      "step": 17581
    },
    {
      "epoch": 1.676654815673828e-05,
      "step": 17581,
      "training_step_time": 16.990222215652466
    },
    {
      "epoch": 1.6767501831054688e-05,
      "model_forward_time": 5.581696033477783,
      "step": 17582
    },
    {
      "epoch": 1.6767501831054688e-05,
      "step": 17582,
      "training_step_time": 17.37728476524353
    },
    {
      "epoch": 1.6768455505371092e-05,
      "model_forward_time": 5.56961464881897,
      "step": 17583
    },
    {
      "epoch": 1.6768455505371092e-05,
      "step": 17583,
      "training_step_time": 17.011744737625122
    },
    {
      "epoch": 1.67694091796875e-05,
      "model_forward_time": 5.568999290466309,
      "step": 17584
    },
    {
      "epoch": 1.67694091796875e-05,
      "step": 17584,
      "training_step_time": 16.97484064102173
    },
    {
      "epoch": 1.6770362854003907e-05,
      "model_forward_time": 5.587749242782593,
      "step": 17585
    },
    {
      "epoch": 1.6770362854003907e-05,
      "step": 17585,
      "training_step_time": 17.01417374610901
    },
    {
      "epoch": 1.677131652832031e-05,
      "model_forward_time": 5.595466613769531,
      "step": 17586
    },
    {
      "epoch": 1.677131652832031e-05,
      "step": 17586,
      "training_step_time": 17.052214860916138
    },
    {
      "epoch": 1.677227020263672e-05,
      "model_forward_time": 5.59418511390686,
      "step": 17587
    },
    {
      "epoch": 1.677227020263672e-05,
      "step": 17587,
      "training_step_time": 17.04918670654297
    },
    {
      "epoch": 1.6773223876953126e-05,
      "model_forward_time": 5.981666564941406,
      "step": 17588
    },
    {
      "epoch": 1.6773223876953126e-05,
      "step": 17588,
      "training_step_time": 17.44282865524292
    },
    {
      "epoch": 1.677417755126953e-05,
      "model_forward_time": 5.603655815124512,
      "step": 17589
    },
    {
      "epoch": 1.677417755126953e-05,
      "step": 17589,
      "training_step_time": 17.058201551437378
    },
    {
      "epoch": 1.6775131225585937e-05,
      "grad_norm": 0.9344317317008972,
      "learning_rate": 1e-05,
      "loss": 0.0569,
      "step": 17590
    },
    {
      "epoch": 1.6775131225585937e-05,
      "model_forward_time": 5.588639974594116,
      "step": 17590
    },
    {
      "dynamics_loss_avg": 0.05186610575765371,
      "epoch": 1.6775131225585937e-05,
      "step": 17590
    },
    {
      "action_loss_avg": 0.0058278801618143914,
      "epoch": 1.6775131225585937e-05,
      "step": 17590
    },
    {
      "epoch": 1.6775131225585937e-05,
      "step": 17590,
      "training_step_time": 17.018478631973267
    },
    {
      "epoch": 1.6776084899902345e-05,
      "model_forward_time": 5.600294351577759,
      "step": 17591
    },
    {
      "epoch": 1.6776084899902345e-05,
      "step": 17591,
      "training_step_time": 17.089139938354492
    },
    {
      "epoch": 1.677703857421875e-05,
      "model_forward_time": 5.621161222457886,
      "step": 17592
    },
    {
      "epoch": 1.677703857421875e-05,
      "step": 17592,
      "training_step_time": 17.14010238647461
    },
    {
      "epoch": 1.6777992248535156e-05,
      "model_forward_time": 5.627342939376831,
      "step": 17593
    },
    {
      "epoch": 1.6777992248535156e-05,
      "step": 17593,
      "training_step_time": 17.484378814697266
    },
    {
      "epoch": 1.6778945922851564e-05,
      "model_forward_time": 5.59307336807251,
      "step": 17594
    },
    {
      "epoch": 1.6778945922851564e-05,
      "step": 17594,
      "training_step_time": 17.03690791130066
    },
    {
      "epoch": 1.6779899597167968e-05,
      "model_forward_time": 5.565901756286621,
      "step": 17595
    },
    {
      "epoch": 1.6779899597167968e-05,
      "step": 17595,
      "training_step_time": 16.976313829421997
    },
    {
      "epoch": 1.6780853271484375e-05,
      "model_forward_time": 5.550188064575195,
      "step": 17596
    },
    {
      "epoch": 1.6780853271484375e-05,
      "step": 17596,
      "training_step_time": 16.962820529937744
    },
    {
      "epoch": 1.6781806945800782e-05,
      "model_forward_time": 5.564326047897339,
      "step": 17597
    },
    {
      "epoch": 1.6781806945800782e-05,
      "step": 17597,
      "training_step_time": 16.971041679382324
    },
    {
      "epoch": 1.6782760620117187e-05,
      "model_forward_time": 5.559908866882324,
      "step": 17598
    },
    {
      "epoch": 1.6782760620117187e-05,
      "step": 17598,
      "training_step_time": 16.98650598526001
    },
    {
      "epoch": 1.6783714294433594e-05,
      "model_forward_time": 5.968299865722656,
      "step": 17599
    },
    {
      "epoch": 1.6783714294433594e-05,
      "step": 17599,
      "training_step_time": 17.37405490875244
    },
    {
      "epoch": 1.678466796875e-05,
      "grad_norm": 0.6705747246742249,
      "learning_rate": 1e-05,
      "loss": 0.0569,
      "step": 17600
    },
    {
      "epoch": 1.678466796875e-05,
      "model_forward_time": 5.57559061050415,
      "step": 17600
    },
    {
      "dynamics_loss_avg": 0.05366972535848617,
      "epoch": 1.678466796875e-05,
      "step": 17600
    },
    {
      "action_loss_avg": 0.006324432557448745,
      "epoch": 1.678466796875e-05,
      "step": 17600
    },
    {
      "epoch": 1.678466796875e-05,
      "step": 17600,
      "training_step_time": 16.98573064804077
    },
    {
      "epoch": 1.6785621643066405e-05,
      "model_forward_time": 5.585320234298706,
      "step": 17601
    },
    {
      "epoch": 1.6785621643066405e-05,
      "step": 17601,
      "training_step_time": 17.012551307678223
    },
    {
      "epoch": 1.6786575317382813e-05,
      "model_forward_time": 5.592871427536011,
      "step": 17602
    },
    {
      "epoch": 1.6786575317382813e-05,
      "step": 17602,
      "training_step_time": 17.046849727630615
    },
    {
      "epoch": 1.678752899169922e-05,
      "model_forward_time": 5.6000657081604,
      "step": 17603
    },
    {
      "epoch": 1.678752899169922e-05,
      "step": 17603,
      "training_step_time": 17.098674297332764
    },
    {
      "epoch": 1.6788482666015624e-05,
      "model_forward_time": 5.957782030105591,
      "step": 17604
    },
    {
      "epoch": 1.6788482666015624e-05,
      "step": 17604,
      "training_step_time": 17.462894439697266
    },
    {
      "epoch": 1.678943634033203e-05,
      "model_forward_time": 5.628041982650757,
      "step": 17605
    },
    {
      "epoch": 1.678943634033203e-05,
      "step": 17605,
      "training_step_time": 17.140528202056885
    },
    {
      "epoch": 1.679039001464844e-05,
      "model_forward_time": 5.624258995056152,
      "step": 17606
    },
    {
      "epoch": 1.679039001464844e-05,
      "step": 17606,
      "training_step_time": 17.1288161277771
    },
    {
      "epoch": 1.6791343688964843e-05,
      "model_forward_time": 5.611999988555908,
      "step": 17607
    },
    {
      "epoch": 1.6791343688964843e-05,
      "step": 17607,
      "training_step_time": 17.11738610267639
    },
    {
      "epoch": 1.679229736328125e-05,
      "model_forward_time": 5.6151041984558105,
      "step": 17608
    },
    {
      "epoch": 1.679229736328125e-05,
      "step": 17608,
      "training_step_time": 17.122301816940308
    },
    {
      "epoch": 1.6793251037597658e-05,
      "model_forward_time": 5.617350339889526,
      "step": 17609
    },
    {
      "epoch": 1.6793251037597658e-05,
      "step": 17609,
      "training_step_time": 17.103869915008545
    },
    {
      "epoch": 1.6794204711914062e-05,
      "grad_norm": 1.1344304084777832,
      "learning_rate": 1e-05,
      "loss": 0.0532,
      "step": 17610
    },
    {
      "epoch": 1.6794204711914062e-05,
      "model_forward_time": 5.984925985336304,
      "step": 17610
    },
    {
      "dynamics_loss_avg": 0.04895994830876589,
      "epoch": 1.6794204711914062e-05,
      "step": 17610
    },
    {
      "action_loss_avg": 0.005284629703965038,
      "epoch": 1.6794204711914062e-05,
      "step": 17610
    },
    {
      "epoch": 1.6794204711914062e-05,
      "step": 17610,
      "training_step_time": 17.47331428527832
    },
    {
      "epoch": 1.679515838623047e-05,
      "model_forward_time": 5.614747524261475,
      "step": 17611
    },
    {
      "epoch": 1.679515838623047e-05,
      "step": 17611,
      "training_step_time": 17.087716102600098
    },
    {
      "epoch": 1.6796112060546873e-05,
      "model_forward_time": 5.590101718902588,
      "step": 17612
    },
    {
      "epoch": 1.6796112060546873e-05,
      "step": 17612,
      "training_step_time": 17.02085828781128
    },
    {
      "epoch": 1.679706573486328e-05,
      "model_forward_time": 5.587074279785156,
      "step": 17613
    },
    {
      "epoch": 1.679706573486328e-05,
      "step": 17613,
      "training_step_time": 17.0210919380188
    },
    {
      "epoch": 1.6798019409179688e-05,
      "model_forward_time": 5.58953857421875,
      "step": 17614
    },
    {
      "epoch": 1.6798019409179688e-05,
      "step": 17614,
      "training_step_time": 17.04970955848694
    },
    {
      "epoch": 1.6798973083496092e-05,
      "model_forward_time": 5.614464282989502,
      "step": 17615
    },
    {
      "epoch": 1.6798973083496092e-05,
      "step": 17615,
      "training_step_time": 17.4298095703125
    },
    {
      "epoch": 1.67999267578125e-05,
      "model_forward_time": 5.599276065826416,
      "step": 17616
    },
    {
      "epoch": 1.67999267578125e-05,
      "step": 17616,
      "training_step_time": 17.048180103302002
    },
    {
      "epoch": 1.6800880432128907e-05,
      "model_forward_time": 5.585416316986084,
      "step": 17617
    },
    {
      "epoch": 1.6800880432128907e-05,
      "step": 17617,
      "training_step_time": 17.03144335746765
    },
    {
      "epoch": 1.680183410644531e-05,
      "model_forward_time": 5.584919214248657,
      "step": 17618
    },
    {
      "epoch": 1.680183410644531e-05,
      "step": 17618,
      "training_step_time": 17.025638103485107
    },
    {
      "epoch": 1.680278778076172e-05,
      "model_forward_time": 5.599183797836304,
      "step": 17619
    },
    {
      "epoch": 1.680278778076172e-05,
      "step": 17619,
      "training_step_time": 17.074833393096924
    },
    {
      "epoch": 1.6803741455078126e-05,
      "grad_norm": 0.6189662218093872,
      "learning_rate": 1e-05,
      "loss": 0.0723,
      "step": 17620
    },
    {
      "epoch": 1.6803741455078126e-05,
      "model_forward_time": 5.615575313568115,
      "step": 17620
    },
    {
      "dynamics_loss_avg": 0.06263361014425754,
      "epoch": 1.6803741455078126e-05,
      "step": 17620
    },
    {
      "action_loss_avg": 0.007534947851672769,
      "epoch": 1.6803741455078126e-05,
      "step": 17620
    },
    {
      "epoch": 1.6803741455078126e-05,
      "step": 17620,
      "training_step_time": 17.118471384048462
    },
    {
      "epoch": 1.680469512939453e-05,
      "model_forward_time": 5.98224663734436,
      "step": 17621
    },
    {
      "epoch": 1.680469512939453e-05,
      "step": 17621,
      "training_step_time": 17.460402727127075
    },
    {
      "epoch": 1.6805648803710937e-05,
      "model_forward_time": 5.597466230392456,
      "step": 17622
    },
    {
      "epoch": 1.6805648803710937e-05,
      "step": 17622,
      "training_step_time": 17.03774356842041
    },
    {
      "epoch": 1.6806602478027345e-05,
      "model_forward_time": 5.5787575244903564,
      "step": 17623
    },
    {
      "epoch": 1.6806602478027345e-05,
      "step": 17623,
      "training_step_time": 17.00019669532776
    },
    {
      "epoch": 1.680755615234375e-05,
      "model_forward_time": 5.592023849487305,
      "step": 17624
    },
    {
      "epoch": 1.680755615234375e-05,
      "step": 17624,
      "training_step_time": 17.021817922592163
    },
    {
      "epoch": 1.6808509826660156e-05,
      "model_forward_time": 5.576614856719971,
      "step": 17625
    },
    {
      "epoch": 1.6808509826660156e-05,
      "step": 17625,
      "training_step_time": 17.01513695716858
    },
    {
      "epoch": 1.6809463500976564e-05,
      "model_forward_time": 5.579288721084595,
      "step": 17626
    },
    {
      "epoch": 1.6809463500976564e-05,
      "step": 17626,
      "training_step_time": 17.35979652404785
    },
    {
      "epoch": 1.6810417175292968e-05,
      "model_forward_time": 5.581104040145874,
      "step": 17627
    },
    {
      "epoch": 1.6810417175292968e-05,
      "step": 17627,
      "training_step_time": 16.999242782592773
    },
    {
      "epoch": 1.6811370849609375e-05,
      "model_forward_time": 5.588700771331787,
      "step": 17628
    },
    {
      "epoch": 1.6811370849609375e-05,
      "step": 17628,
      "training_step_time": 17.021333932876587
    },
    {
      "epoch": 1.6812324523925783e-05,
      "model_forward_time": 5.584641933441162,
      "step": 17629
    },
    {
      "epoch": 1.6812324523925783e-05,
      "step": 17629,
      "training_step_time": 17.028167963027954
    },
    {
      "epoch": 1.6813278198242187e-05,
      "grad_norm": 1.0015586614608765,
      "learning_rate": 1e-05,
      "loss": 0.0569,
      "step": 17630
    },
    {
      "epoch": 1.6813278198242187e-05,
      "model_forward_time": 5.608060121536255,
      "step": 17630
    },
    {
      "dynamics_loss_avg": 0.052662986330688,
      "epoch": 1.6813278198242187e-05,
      "step": 17630
    },
    {
      "action_loss_avg": 0.005603782017715275,
      "epoch": 1.6813278198242187e-05,
      "step": 17630
    },
    {
      "epoch": 1.6813278198242187e-05,
      "step": 17630,
      "training_step_time": 17.084410429000854
    },
    {
      "epoch": 1.6814231872558594e-05,
      "model_forward_time": 5.610802412033081,
      "step": 17631
    },
    {
      "epoch": 1.6814231872558594e-05,
      "step": 17631,
      "training_step_time": 17.11284327507019
    },
    {
      "epoch": 1.6815185546875e-05,
      "model_forward_time": 5.980871915817261,
      "step": 17632
    },
    {
      "epoch": 1.6815185546875e-05,
      "step": 17632,
      "training_step_time": 17.479583740234375
    },
    {
      "epoch": 1.6816139221191405e-05,
      "model_forward_time": 5.617877721786499,
      "step": 17633
    },
    {
      "epoch": 1.6816139221191405e-05,
      "step": 17633,
      "training_step_time": 17.105653762817383
    },
    {
      "epoch": 1.6817092895507813e-05,
      "model_forward_time": 5.606294631958008,
      "step": 17634
    },
    {
      "epoch": 1.6817092895507813e-05,
      "step": 17634,
      "training_step_time": 17.09282159805298
    },
    {
      "epoch": 1.681804656982422e-05,
      "model_forward_time": 5.608902454376221,
      "step": 17635
    },
    {
      "epoch": 1.681804656982422e-05,
      "step": 17635,
      "training_step_time": 17.11194658279419
    },
    {
      "epoch": 1.6819000244140624e-05,
      "model_forward_time": 5.6094069480896,
      "step": 17636
    },
    {
      "epoch": 1.6819000244140624e-05,
      "step": 17636,
      "training_step_time": 17.10277533531189
    },
    {
      "epoch": 1.681995391845703e-05,
      "model_forward_time": 5.962338447570801,
      "step": 17637
    },
    {
      "epoch": 1.681995391845703e-05,
      "step": 17637,
      "training_step_time": 17.441314458847046
    },
    {
      "epoch": 1.682090759277344e-05,
      "model_forward_time": 5.6058573722839355,
      "step": 17638
    },
    {
      "epoch": 1.682090759277344e-05,
      "step": 17638,
      "training_step_time": 17.07899808883667
    },
    {
      "epoch": 1.6821861267089843e-05,
      "model_forward_time": 5.6164093017578125,
      "step": 17639
    },
    {
      "epoch": 1.6821861267089843e-05,
      "step": 17639,
      "training_step_time": 17.103973865509033
    },
    {
      "epoch": 1.682281494140625e-05,
      "grad_norm": 0.3824000954627991,
      "learning_rate": 1e-05,
      "loss": 0.0682,
      "step": 17640
    },
    {
      "epoch": 1.682281494140625e-05,
      "model_forward_time": 5.620196104049683,
      "step": 17640
    },
    {
      "dynamics_loss_avg": 0.05679293368011713,
      "epoch": 1.682281494140625e-05,
      "step": 17640
    },
    {
      "action_loss_avg": 0.007056395756080747,
      "epoch": 1.682281494140625e-05,
      "step": 17640
    },
    {
      "epoch": 1.682281494140625e-05,
      "step": 17640,
      "training_step_time": 17.112765312194824
    },
    {
      "epoch": 1.6823768615722655e-05,
      "model_forward_time": 5.612529277801514,
      "step": 17641
    },
    {
      "epoch": 1.6823768615722655e-05,
      "step": 17641,
      "training_step_time": 17.103352785110474
    },
    {
      "epoch": 1.6824722290039062e-05,
      "model_forward_time": 5.616309404373169,
      "step": 17642
    },
    {
      "epoch": 1.6824722290039062e-05,
      "step": 17642,
      "training_step_time": 17.097423553466797
    },
    {
      "epoch": 1.682567596435547e-05,
      "model_forward_time": 5.969573020935059,
      "step": 17643
    },
    {
      "epoch": 1.682567596435547e-05,
      "step": 17643,
      "training_step_time": 17.461551189422607
    },
    {
      "epoch": 1.6826629638671873e-05,
      "model_forward_time": 5.6166698932647705,
      "step": 17644
    },
    {
      "epoch": 1.6826629638671873e-05,
      "step": 17644,
      "training_step_time": 17.103591680526733
    },
    {
      "epoch": 1.682758331298828e-05,
      "model_forward_time": 5.607688903808594,
      "step": 17645
    },
    {
      "epoch": 1.682758331298828e-05,
      "step": 17645,
      "training_step_time": 17.101086854934692
    },
    {
      "epoch": 1.6828536987304688e-05,
      "model_forward_time": 5.60525918006897,
      "step": 17646
    },
    {
      "epoch": 1.6828536987304688e-05,
      "step": 17646,
      "training_step_time": 17.07102084159851
    },
    {
      "epoch": 1.6829490661621092e-05,
      "model_forward_time": 5.600948810577393,
      "step": 17647
    },
    {
      "epoch": 1.6829490661621092e-05,
      "step": 17647,
      "training_step_time": 17.05264449119568
    },
    {
      "epoch": 1.68304443359375e-05,
      "model_forward_time": 5.5909743309021,
      "step": 17648
    },
    {
      "epoch": 1.68304443359375e-05,
      "step": 17648,
      "training_step_time": 17.416703701019287
    },
    {
      "epoch": 1.6831398010253907e-05,
      "model_forward_time": 5.61404275894165,
      "step": 17649
    },
    {
      "epoch": 1.6831398010253907e-05,
      "step": 17649,
      "training_step_time": 17.103599548339844
    },
    {
      "epoch": 1.683235168457031e-05,
      "grad_norm": 0.8364387154579163,
      "learning_rate": 1e-05,
      "loss": 0.0724,
      "step": 17650
    },
    {
      "epoch": 1.683235168457031e-05,
      "model_forward_time": 5.6201252937316895,
      "step": 17650
    },
    {
      "dynamics_loss_avg": 0.07051872126758099,
      "epoch": 1.683235168457031e-05,
      "step": 17650
    },
    {
      "action_loss_avg": 0.00730848666280508,
      "epoch": 1.683235168457031e-05,
      "step": 17650
    },
    {
      "epoch": 1.683235168457031e-05,
      "step": 17650,
      "training_step_time": 17.090320348739624
    },
    {
      "epoch": 1.683330535888672e-05,
      "model_forward_time": 5.592357873916626,
      "step": 17651
    },
    {
      "epoch": 1.683330535888672e-05,
      "step": 17651,
      "training_step_time": 17.039077758789062
    },
    {
      "epoch": 1.6834259033203126e-05,
      "model_forward_time": 5.607860565185547,
      "step": 17652
    },
    {
      "epoch": 1.6834259033203126e-05,
      "step": 17652,
      "training_step_time": 17.09716010093689
    },
    {
      "epoch": 1.683521270751953e-05,
      "model_forward_time": 5.604917764663696,
      "step": 17653
    },
    {
      "epoch": 1.683521270751953e-05,
      "step": 17653,
      "training_step_time": 17.08815097808838
    },
    {
      "epoch": 1.6836166381835937e-05,
      "model_forward_time": 5.977219104766846,
      "step": 17654
    },
    {
      "epoch": 1.6836166381835937e-05,
      "step": 17654,
      "training_step_time": 17.46420407295227
    },
    {
      "epoch": 1.6837120056152345e-05,
      "model_forward_time": 5.609468221664429,
      "step": 17655
    },
    {
      "epoch": 1.6837120056152345e-05,
      "step": 17655,
      "training_step_time": 17.09745693206787
    },
    {
      "epoch": 1.683807373046875e-05,
      "model_forward_time": 5.610880136489868,
      "step": 17656
    },
    {
      "epoch": 1.683807373046875e-05,
      "step": 17656,
      "training_step_time": 17.085947513580322
    },
    {
      "epoch": 1.6839027404785156e-05,
      "model_forward_time": 5.605063199996948,
      "step": 17657
    },
    {
      "epoch": 1.6839027404785156e-05,
      "step": 17657,
      "training_step_time": 17.085686445236206
    },
    {
      "epoch": 1.6839981079101564e-05,
      "model_forward_time": 5.613195896148682,
      "step": 17658
    },
    {
      "epoch": 1.6839981079101564e-05,
      "step": 17658,
      "training_step_time": 17.06788682937622
    },
    {
      "epoch": 1.6840934753417968e-05,
      "model_forward_time": 5.6078572273254395,
      "step": 17659
    },
    {
      "epoch": 1.6840934753417968e-05,
      "step": 17659,
      "training_step_time": 17.40796446800232
    },
    {
      "epoch": 1.6841888427734375e-05,
      "grad_norm": 0.7556524872779846,
      "learning_rate": 1e-05,
      "loss": 0.0714,
      "step": 17660
    },
    {
      "epoch": 1.6841888427734375e-05,
      "model_forward_time": 5.609104156494141,
      "step": 17660
    },
    {
      "dynamics_loss_avg": 0.059872161410748956,
      "epoch": 1.6841888427734375e-05,
      "step": 17660
    },
    {
      "action_loss_avg": 0.007163659506477416,
      "epoch": 1.6841888427734375e-05,
      "step": 17660
    },
    {
      "epoch": 1.6841888427734375e-05,
      "step": 17660,
      "training_step_time": 17.081048488616943
    },
    {
      "epoch": 1.6842842102050783e-05,
      "model_forward_time": 5.602754354476929,
      "step": 17661
    },
    {
      "epoch": 1.6842842102050783e-05,
      "step": 17661,
      "training_step_time": 17.054237604141235
    },
    {
      "epoch": 1.6843795776367187e-05,
      "model_forward_time": 5.593895673751831,
      "step": 17662
    },
    {
      "epoch": 1.6843795776367187e-05,
      "step": 17662,
      "training_step_time": 17.027520656585693
    },
    {
      "epoch": 1.6844749450683594e-05,
      "model_forward_time": 5.572346210479736,
      "step": 17663
    },
    {
      "epoch": 1.6844749450683594e-05,
      "step": 17663,
      "training_step_time": 17.013216257095337
    },
    {
      "epoch": 1.6845703125e-05,
      "model_forward_time": 5.590380668640137,
      "step": 17664
    },
    {
      "epoch": 1.6845703125e-05,
      "step": 17664,
      "training_step_time": 17.03945779800415
    },
    {
      "epoch": 1.6846656799316405e-05,
      "model_forward_time": 5.956941604614258,
      "step": 17665
    },
    {
      "epoch": 1.6846656799316405e-05,
      "step": 17665,
      "training_step_time": 17.417109966278076
    },
    {
      "epoch": 1.6847610473632813e-05,
      "model_forward_time": 5.592468023300171,
      "step": 17666
    },
    {
      "epoch": 1.6847610473632813e-05,
      "step": 17666,
      "training_step_time": 17.060017585754395
    },
    {
      "epoch": 1.684856414794922e-05,
      "model_forward_time": 5.611823797225952,
      "step": 17667
    },
    {
      "epoch": 1.684856414794922e-05,
      "step": 17667,
      "training_step_time": 17.134321689605713
    },
    {
      "epoch": 1.6849517822265624e-05,
      "model_forward_time": 5.619930744171143,
      "step": 17668
    },
    {
      "epoch": 1.6849517822265624e-05,
      "step": 17668,
      "training_step_time": 17.105348110198975
    },
    {
      "epoch": 1.6850471496582032e-05,
      "model_forward_time": 5.595051527023315,
      "step": 17669
    },
    {
      "epoch": 1.6850471496582032e-05,
      "step": 17669,
      "training_step_time": 17.04175066947937
    },
    {
      "epoch": 1.685142517089844e-05,
      "grad_norm": 1.006801724433899,
      "learning_rate": 1e-05,
      "loss": 0.061,
      "step": 17670
    },
    {
      "epoch": 1.685142517089844e-05,
      "model_forward_time": 5.934236764907837,
      "step": 17670
    },
    {
      "dynamics_loss_avg": 0.05843634819611907,
      "epoch": 1.685142517089844e-05,
      "step": 17670
    },
    {
      "action_loss_avg": 0.006811014551203698,
      "epoch": 1.685142517089844e-05,
      "step": 17670
    },
    {
      "epoch": 1.685142517089844e-05,
      "step": 17670,
      "training_step_time": 17.378069639205933
    },
    {
      "epoch": 1.6852378845214843e-05,
      "model_forward_time": 5.57907509803772,
      "step": 17671
    },
    {
      "epoch": 1.6852378845214843e-05,
      "step": 17671,
      "training_step_time": 17.03172016143799
    },
    {
      "epoch": 1.685333251953125e-05,
      "model_forward_time": 5.627485036849976,
      "step": 17672
    },
    {
      "epoch": 1.685333251953125e-05,
      "step": 17672,
      "training_step_time": 17.12275981903076
    },
    {
      "epoch": 1.6854286193847655e-05,
      "model_forward_time": 5.615220308303833,
      "step": 17673
    },
    {
      "epoch": 1.6854286193847655e-05,
      "step": 17673,
      "training_step_time": 17.109230995178223
    },
    {
      "epoch": 1.6855239868164062e-05,
      "model_forward_time": 5.610775947570801,
      "step": 17674
    },
    {
      "epoch": 1.6855239868164062e-05,
      "step": 17674,
      "training_step_time": 17.053717136383057
    },
    {
      "epoch": 1.685619354248047e-05,
      "model_forward_time": 5.592861652374268,
      "step": 17675
    },
    {
      "epoch": 1.685619354248047e-05,
      "step": 17675,
      "training_step_time": 17.00122880935669
    },
    {
      "epoch": 1.6857147216796873e-05,
      "model_forward_time": 5.939838647842407,
      "step": 17676
    },
    {
      "epoch": 1.6857147216796873e-05,
      "step": 17676,
      "training_step_time": 17.34618878364563
    },
    {
      "epoch": 1.685810089111328e-05,
      "model_forward_time": 5.5734946727752686,
      "step": 17677
    },
    {
      "epoch": 1.685810089111328e-05,
      "step": 17677,
      "training_step_time": 16.99512553215027
    },
    {
      "epoch": 1.6859054565429688e-05,
      "model_forward_time": 5.582479953765869,
      "step": 17678
    },
    {
      "epoch": 1.6859054565429688e-05,
      "step": 17678,
      "training_step_time": 17.018640756607056
    },
    {
      "epoch": 1.6860008239746092e-05,
      "model_forward_time": 5.586219310760498,
      "step": 17679
    },
    {
      "epoch": 1.6860008239746092e-05,
      "step": 17679,
      "training_step_time": 17.0507550239563
    },
    {
      "epoch": 1.68609619140625e-05,
      "grad_norm": 0.45667850971221924,
      "learning_rate": 1e-05,
      "loss": 0.0581,
      "step": 17680
    },
    {
      "epoch": 1.68609619140625e-05,
      "model_forward_time": 5.614767789840698,
      "step": 17680
    },
    {
      "dynamics_loss_avg": 0.05153397414833307,
      "epoch": 1.68609619140625e-05,
      "step": 17680
    },
    {
      "action_loss_avg": 0.006480715924408287,
      "epoch": 1.68609619140625e-05,
      "step": 17680
    },
    {
      "epoch": 1.68609619140625e-05,
      "step": 17680,
      "training_step_time": 17.089576721191406
    },
    {
      "epoch": 1.6861915588378907e-05,
      "model_forward_time": 5.605113983154297,
      "step": 17681
    },
    {
      "epoch": 1.6861915588378907e-05,
      "step": 17681,
      "training_step_time": 17.44284200668335
    },
    {
      "epoch": 1.686286926269531e-05,
      "model_forward_time": 5.583732843399048,
      "step": 17682
    },
    {
      "epoch": 1.686286926269531e-05,
      "step": 17682,
      "training_step_time": 17.025290489196777
    },
    {
      "epoch": 1.686382293701172e-05,
      "model_forward_time": 5.594069004058838,
      "step": 17683
    },
    {
      "epoch": 1.686382293701172e-05,
      "step": 17683,
      "training_step_time": 17.05148673057556
    },
    {
      "epoch": 1.6864776611328126e-05,
      "model_forward_time": 5.610444784164429,
      "step": 17684
    },
    {
      "epoch": 1.6864776611328126e-05,
      "step": 17684,
      "training_step_time": 17.08813500404358
    },
    {
      "epoch": 1.686573028564453e-05,
      "model_forward_time": 4.279480695724487,
      "step": 17685
    },
    {
      "epoch": 1.686573028564453e-05,
      "step": 17685,
      "training_step_time": 12.81679391860962
    },
    {
      "epoch": 1.6866683959960937e-05,
      "model_forward_time": 5.609731674194336,
      "step": 17686
    },
    {
      "epoch": 1.6866683959960937e-05,
      "step": 17686,
      "training_step_time": 17.109748363494873
    },
    {
      "epoch": 1.6867637634277345e-05,
      "model_forward_time": 5.980368614196777,
      "step": 17687
    },
    {
      "epoch": 1.6867637634277345e-05,
      "step": 17687,
      "training_step_time": 17.48764705657959
    },
    {
      "epoch": 1.686859130859375e-05,
      "model_forward_time": 5.61918044090271,
      "step": 17688
    },
    {
      "epoch": 1.686859130859375e-05,
      "step": 17688,
      "training_step_time": 17.13820505142212
    },
    {
      "epoch": 1.6869544982910156e-05,
      "model_forward_time": 5.620956659317017,
      "step": 17689
    },
    {
      "epoch": 1.6869544982910156e-05,
      "step": 17689,
      "training_step_time": 17.10112762451172
    },
    {
      "epoch": 1.6870498657226564e-05,
      "grad_norm": 1.15486478805542,
      "learning_rate": 1e-05,
      "loss": 0.0478,
      "step": 17690
    },
    {
      "epoch": 1.6870498657226564e-05,
      "model_forward_time": 5.590827941894531,
      "step": 17690
    },
    {
      "dynamics_loss_avg": 0.037556810304522514,
      "epoch": 1.6870498657226564e-05,
      "step": 17690
    },
    {
      "action_loss_avg": 0.004153269855305552,
      "epoch": 1.6870498657226564e-05,
      "step": 17690
    },
    {
      "epoch": 1.6870498657226564e-05,
      "step": 17690,
      "training_step_time": 17.035027742385864
    },
    {
      "epoch": 1.6871452331542968e-05,
      "model_forward_time": 5.5980260372161865,
      "step": 17691
    },
    {
      "epoch": 1.6871452331542968e-05,
      "step": 17691,
      "training_step_time": 17.046773195266724
    },
    {
      "epoch": 1.6872406005859375e-05,
      "model_forward_time": 5.595123529434204,
      "step": 17692
    },
    {
      "epoch": 1.6872406005859375e-05,
      "step": 17692,
      "training_step_time": 17.07840061187744
    },
    {
      "epoch": 1.6873359680175783e-05,
      "model_forward_time": 5.97257399559021,
      "step": 17693
    },
    {
      "epoch": 1.6873359680175783e-05,
      "step": 17693,
      "training_step_time": 17.4746675491333
    },
    {
      "epoch": 1.6874313354492187e-05,
      "model_forward_time": 5.625258207321167,
      "step": 17694
    },
    {
      "epoch": 1.6874313354492187e-05,
      "step": 17694,
      "training_step_time": 17.11136507987976
    },
    {
      "epoch": 1.6875267028808594e-05,
      "model_forward_time": 5.59674596786499,
      "step": 17695
    },
    {
      "epoch": 1.6875267028808594e-05,
      "step": 17695,
      "training_step_time": 17.006449699401855
    },
    {
      "epoch": 1.6876220703125e-05,
      "model_forward_time": 5.579343557357788,
      "step": 17696
    },
    {
      "epoch": 1.6876220703125e-05,
      "step": 17696,
      "training_step_time": 16.99971079826355
    },
    {
      "epoch": 1.6877174377441405e-05,
      "model_forward_time": 5.568158388137817,
      "step": 17697
    },
    {
      "epoch": 1.6877174377441405e-05,
      "step": 17697,
      "training_step_time": 17.0107843875885
    },
    {
      "epoch": 1.6878128051757813e-05,
      "model_forward_time": 5.9161248207092285,
      "step": 17698
    },
    {
      "epoch": 1.6878128051757813e-05,
      "step": 17698,
      "training_step_time": 17.327187299728394
    },
    {
      "epoch": 1.687908172607422e-05,
      "model_forward_time": 5.581327438354492,
      "step": 17699
    },
    {
      "epoch": 1.687908172607422e-05,
      "step": 17699,
      "training_step_time": 17.014267444610596
    },
    {
      "epoch": 1.6880035400390624e-05,
      "grad_norm": 0.39467859268188477,
      "learning_rate": 1e-05,
      "loss": 0.0514,
      "step": 17700
    },
    {
      "epoch": 1.6880035400390624e-05,
      "model_forward_time": 5.593932151794434,
      "step": 17700
    },
    {
      "dynamics_loss_avg": 0.04789717346429825,
      "epoch": 1.6880035400390624e-05,
      "step": 17700
    },
    {
      "action_loss_avg": 0.005369073757901788,
      "epoch": 1.6880035400390624e-05,
      "step": 17700
    },
    {
      "epoch": 1.6880035400390624e-05,
      "step": 17700,
      "training_step_time": 17.008824348449707
    },
    {
      "epoch": 1.6880989074707032e-05,
      "model_forward_time": 5.592271089553833,
      "step": 17701
    },
    {
      "epoch": 1.6880989074707032e-05,
      "step": 17701,
      "training_step_time": 17.03819179534912
    },
    {
      "epoch": 1.688194274902344e-05,
      "model_forward_time": 5.591235637664795,
      "step": 17702
    },
    {
      "epoch": 1.688194274902344e-05,
      "step": 17702,
      "training_step_time": 17.03520894050598
    },
    {
      "epoch": 1.6882896423339843e-05,
      "model_forward_time": 5.972157955169678,
      "step": 17703
    },
    {
      "epoch": 1.6882896423339843e-05,
      "step": 17703,
      "training_step_time": 17.444639444351196
    },
    {
      "epoch": 1.688385009765625e-05,
      "model_forward_time": 5.607743501663208,
      "step": 17704
    },
    {
      "epoch": 1.688385009765625e-05,
      "step": 17704,
      "training_step_time": 17.099263668060303
    },
    {
      "epoch": 1.6884803771972655e-05,
      "model_forward_time": 5.611390590667725,
      "step": 17705
    },
    {
      "epoch": 1.6884803771972655e-05,
      "step": 17705,
      "training_step_time": 17.118974447250366
    },
    {
      "epoch": 1.6885757446289062e-05,
      "model_forward_time": 5.615755558013916,
      "step": 17706
    },
    {
      "epoch": 1.6885757446289062e-05,
      "step": 17706,
      "training_step_time": 17.104618549346924
    },
    {
      "epoch": 1.688671112060547e-05,
      "model_forward_time": 5.601479530334473,
      "step": 17707
    },
    {
      "epoch": 1.688671112060547e-05,
      "step": 17707,
      "training_step_time": 17.074910163879395
    },
    {
      "epoch": 1.6887664794921873e-05,
      "model_forward_time": 5.601231098175049,
      "step": 17708
    },
    {
      "epoch": 1.6887664794921873e-05,
      "step": 17708,
      "training_step_time": 17.058858394622803
    },
    {
      "epoch": 1.688861846923828e-05,
      "model_forward_time": 5.947245836257935,
      "step": 17709
    },
    {
      "epoch": 1.688861846923828e-05,
      "step": 17709,
      "training_step_time": 17.38979411125183
    },
    {
      "epoch": 1.688957214355469e-05,
      "grad_norm": 0.5873257517814636,
      "learning_rate": 1e-05,
      "loss": 0.0579,
      "step": 17710
    },
    {
      "epoch": 1.688957214355469e-05,
      "model_forward_time": 5.595887184143066,
      "step": 17710
    },
    {
      "dynamics_loss_avg": 0.054619443416595456,
      "epoch": 1.688957214355469e-05,
      "step": 17710
    },
    {
      "action_loss_avg": 0.006419938779436052,
      "epoch": 1.688957214355469e-05,
      "step": 17710
    },
    {
      "epoch": 1.688957214355469e-05,
      "step": 17710,
      "training_step_time": 17.030121326446533
    },
    {
      "epoch": 1.6890525817871092e-05,
      "model_forward_time": 5.593390941619873,
      "step": 17711
    },
    {
      "epoch": 1.6890525817871092e-05,
      "step": 17711,
      "training_step_time": 17.071324586868286
    },
    {
      "epoch": 1.68914794921875e-05,
      "model_forward_time": 5.622695446014404,
      "step": 17712
    },
    {
      "epoch": 1.68914794921875e-05,
      "step": 17712,
      "training_step_time": 17.128243923187256
    },
    {
      "epoch": 1.6892433166503907e-05,
      "model_forward_time": 5.617478609085083,
      "step": 17713
    },
    {
      "epoch": 1.6892433166503907e-05,
      "step": 17713,
      "training_step_time": 17.119080066680908
    },
    {
      "epoch": 1.689338684082031e-05,
      "model_forward_time": 5.607433795928955,
      "step": 17714
    },
    {
      "epoch": 1.689338684082031e-05,
      "step": 17714,
      "training_step_time": 17.460123538970947
    },
    {
      "epoch": 1.689434051513672e-05,
      "model_forward_time": 5.606600522994995,
      "step": 17715
    },
    {
      "epoch": 1.689434051513672e-05,
      "step": 17715,
      "training_step_time": 17.085212230682373
    },
    {
      "epoch": 1.6895294189453126e-05,
      "model_forward_time": 5.591837644577026,
      "step": 17716
    },
    {
      "epoch": 1.6895294189453126e-05,
      "step": 17716,
      "training_step_time": 17.028998613357544
    },
    {
      "epoch": 1.689624786376953e-05,
      "model_forward_time": 5.580106019973755,
      "step": 17717
    },
    {
      "epoch": 1.689624786376953e-05,
      "step": 17717,
      "training_step_time": 16.98388361930847
    },
    {
      "epoch": 1.6897201538085937e-05,
      "model_forward_time": 5.579905986785889,
      "step": 17718
    },
    {
      "epoch": 1.6897201538085937e-05,
      "step": 17718,
      "training_step_time": 17.02135133743286
    },
    {
      "epoch": 1.6898155212402345e-05,
      "model_forward_time": 5.5899574756622314,
      "step": 17719
    },
    {
      "epoch": 1.6898155212402345e-05,
      "step": 17719,
      "training_step_time": 17.067309856414795
    },
    {
      "epoch": 1.689910888671875e-05,
      "grad_norm": 0.8568190336227417,
      "learning_rate": 1e-05,
      "loss": 0.0535,
      "step": 17720
    },
    {
      "epoch": 1.689910888671875e-05,
      "model_forward_time": 5.960630178451538,
      "step": 17720
    },
    {
      "dynamics_loss_avg": 0.04368378408253193,
      "epoch": 1.689910888671875e-05,
      "step": 17720
    },
    {
      "action_loss_avg": 0.0042746385675854984,
      "epoch": 1.689910888671875e-05,
      "step": 17720
    },
    {
      "epoch": 1.689910888671875e-05,
      "step": 17720,
      "training_step_time": 17.473546266555786
    },
    {
      "epoch": 1.6900062561035156e-05,
      "model_forward_time": 5.63100790977478,
      "step": 17721
    },
    {
      "epoch": 1.6900062561035156e-05,
      "step": 17721,
      "training_step_time": 17.139167547225952
    },
    {
      "epoch": 1.6901016235351564e-05,
      "model_forward_time": 5.627741098403931,
      "step": 17722
    },
    {
      "epoch": 1.6901016235351564e-05,
      "step": 17722,
      "training_step_time": 17.137983798980713
    },
    {
      "epoch": 1.6901969909667968e-05,
      "model_forward_time": 5.6224164962768555,
      "step": 17723
    },
    {
      "epoch": 1.6901969909667968e-05,
      "step": 17723,
      "training_step_time": 17.12630033493042
    },
    {
      "epoch": 1.6902923583984375e-05,
      "model_forward_time": 5.611415386199951,
      "step": 17724
    },
    {
      "epoch": 1.6902923583984375e-05,
      "step": 17724,
      "training_step_time": 17.104204654693604
    },
    {
      "epoch": 1.6903877258300783e-05,
      "model_forward_time": 5.998553037643433,
      "step": 17725
    },
    {
      "epoch": 1.6903877258300783e-05,
      "step": 17725,
      "training_step_time": 17.47828435897827
    },
    {
      "epoch": 1.6904830932617187e-05,
      "model_forward_time": 5.6011669635772705,
      "step": 17726
    },
    {
      "epoch": 1.6904830932617187e-05,
      "step": 17726,
      "training_step_time": 17.044976949691772
    },
    {
      "epoch": 1.6905784606933594e-05,
      "model_forward_time": 5.592364072799683,
      "step": 17727
    },
    {
      "epoch": 1.6905784606933594e-05,
      "step": 17727,
      "training_step_time": 17.041513919830322
    },
    {
      "epoch": 1.690673828125e-05,
      "model_forward_time": 5.595968961715698,
      "step": 17728
    },
    {
      "epoch": 1.690673828125e-05,
      "step": 17728,
      "training_step_time": 17.07355833053589
    },
    {
      "epoch": 1.6907691955566405e-05,
      "model_forward_time": 5.614724636077881,
      "step": 17729
    },
    {
      "epoch": 1.6907691955566405e-05,
      "step": 17729,
      "training_step_time": 17.118833303451538
    },
    {
      "epoch": 1.6908645629882813e-05,
      "grad_norm": 0.7749056220054626,
      "learning_rate": 1e-05,
      "loss": 0.0596,
      "step": 17730
    },
    {
      "epoch": 1.6908645629882813e-05,
      "model_forward_time": 5.6225035190582275,
      "step": 17730
    },
    {
      "dynamics_loss_avg": 0.05321809165179729,
      "epoch": 1.6908645629882813e-05,
      "step": 17730
    },
    {
      "action_loss_avg": 0.008377246838063002,
      "epoch": 1.6908645629882813e-05,
      "step": 17730
    },
    {
      "epoch": 1.6908645629882813e-05,
      "step": 17730,
      "training_step_time": 17.13494324684143
    },
    {
      "epoch": 1.690959930419922e-05,
      "model_forward_time": 5.962696552276611,
      "step": 17731
    },
    {
      "epoch": 1.690959930419922e-05,
      "step": 17731,
      "training_step_time": 17.46492099761963
    },
    {
      "epoch": 1.6910552978515624e-05,
      "model_forward_time": 5.6058104038238525,
      "step": 17732
    },
    {
      "epoch": 1.6910552978515624e-05,
      "step": 17732,
      "training_step_time": 17.088316440582275
    },
    {
      "epoch": 1.6911506652832032e-05,
      "model_forward_time": 5.593555927276611,
      "step": 17733
    },
    {
      "epoch": 1.6911506652832032e-05,
      "step": 17733,
      "training_step_time": 17.04296350479126
    },
    {
      "epoch": 1.691246032714844e-05,
      "model_forward_time": 5.593882322311401,
      "step": 17734
    },
    {
      "epoch": 1.691246032714844e-05,
      "step": 17734,
      "training_step_time": 17.042857885360718
    },
    {
      "epoch": 1.6913414001464843e-05,
      "model_forward_time": 5.584099769592285,
      "step": 17735
    },
    {
      "epoch": 1.6913414001464843e-05,
      "step": 17735,
      "training_step_time": 17.019803047180176
    },
    {
      "epoch": 1.691436767578125e-05,
      "model_forward_time": 5.95819616317749,
      "step": 17736
    },
    {
      "epoch": 1.691436767578125e-05,
      "step": 17736,
      "training_step_time": 17.418436288833618
    },
    {
      "epoch": 1.6915321350097655e-05,
      "model_forward_time": 5.604830741882324,
      "step": 17737
    },
    {
      "epoch": 1.6915321350097655e-05,
      "step": 17737,
      "training_step_time": 17.079642295837402
    },
    {
      "epoch": 1.6916275024414062e-05,
      "model_forward_time": 5.59832501411438,
      "step": 17738
    },
    {
      "epoch": 1.6916275024414062e-05,
      "step": 17738,
      "training_step_time": 17.07079815864563
    },
    {
      "epoch": 1.691722869873047e-05,
      "model_forward_time": 5.599745988845825,
      "step": 17739
    },
    {
      "epoch": 1.691722869873047e-05,
      "step": 17739,
      "training_step_time": 17.073652744293213
    },
    {
      "epoch": 1.6918182373046874e-05,
      "grad_norm": 0.6774731278419495,
      "learning_rate": 1e-05,
      "loss": 0.0521,
      "step": 17740
    },
    {
      "epoch": 1.6918182373046874e-05,
      "model_forward_time": 5.605961799621582,
      "step": 17740
    },
    {
      "dynamics_loss_avg": 0.052132022194564344,
      "epoch": 1.6918182373046874e-05,
      "step": 17740
    },
    {
      "action_loss_avg": 0.006052128737792373,
      "epoch": 1.6918182373046874e-05,
      "step": 17740
    },
    {
      "epoch": 1.6918182373046874e-05,
      "step": 17740,
      "training_step_time": 17.073736667633057
    },
    {
      "epoch": 1.691913604736328e-05,
      "model_forward_time": 5.602656364440918,
      "step": 17741
    },
    {
      "epoch": 1.691913604736328e-05,
      "step": 17741,
      "training_step_time": 17.070509433746338
    },
    {
      "epoch": 1.692008972167969e-05,
      "model_forward_time": 5.955226182937622,
      "step": 17742
    },
    {
      "epoch": 1.692008972167969e-05,
      "step": 17742,
      "training_step_time": 17.41996169090271
    },
    {
      "epoch": 1.6921043395996092e-05,
      "model_forward_time": 5.608953952789307,
      "step": 17743
    },
    {
      "epoch": 1.6921043395996092e-05,
      "step": 17743,
      "training_step_time": 17.09640407562256
    },
    {
      "epoch": 1.69219970703125e-05,
      "model_forward_time": 5.606760740280151,
      "step": 17744
    },
    {
      "epoch": 1.69219970703125e-05,
      "step": 17744,
      "training_step_time": 17.080322265625
    },
    {
      "epoch": 1.6922950744628907e-05,
      "model_forward_time": 5.603694677352905,
      "step": 17745
    },
    {
      "epoch": 1.6922950744628907e-05,
      "step": 17745,
      "training_step_time": 17.079689741134644
    },
    {
      "epoch": 1.692390441894531e-05,
      "model_forward_time": 5.597650527954102,
      "step": 17746
    },
    {
      "epoch": 1.692390441894531e-05,
      "step": 17746,
      "training_step_time": 17.06726884841919
    },
    {
      "epoch": 1.692485809326172e-05,
      "model_forward_time": 5.615217924118042,
      "step": 17747
    },
    {
      "epoch": 1.692485809326172e-05,
      "step": 17747,
      "training_step_time": 17.449267625808716
    },
    {
      "epoch": 1.6925811767578126e-05,
      "model_forward_time": 5.598604679107666,
      "step": 17748
    },
    {
      "epoch": 1.6925811767578126e-05,
      "step": 17748,
      "training_step_time": 17.066096305847168
    },
    {
      "epoch": 1.692676544189453e-05,
      "model_forward_time": 5.596701622009277,
      "step": 17749
    },
    {
      "epoch": 1.692676544189453e-05,
      "step": 17749,
      "training_step_time": 17.05973744392395
    },
    {
      "epoch": 1.6927719116210938e-05,
      "grad_norm": 1.1227587461471558,
      "learning_rate": 1e-05,
      "loss": 0.0875,
      "step": 17750
    },
    {
      "epoch": 1.6927719116210938e-05,
      "model_forward_time": 5.604515552520752,
      "step": 17750
    },
    {
      "dynamics_loss_avg": 0.07634703405201435,
      "epoch": 1.6927719116210938e-05,
      "step": 17750
    },
    {
      "action_loss_avg": 0.00911276212427765,
      "epoch": 1.6927719116210938e-05,
      "step": 17750
    },
    {
      "epoch": 1.6927719116210938e-05,
      "step": 17750,
      "training_step_time": 17.072926998138428
    },
    {
      "epoch": 1.6928672790527345e-05,
      "model_forward_time": 5.600692987442017,
      "step": 17751
    },
    {
      "epoch": 1.6928672790527345e-05,
      "step": 17751,
      "training_step_time": 17.089027643203735
    },
    {
      "epoch": 1.692962646484375e-05,
      "model_forward_time": 5.5986328125,
      "step": 17752
    },
    {
      "epoch": 1.692962646484375e-05,
      "step": 17752,
      "training_step_time": 17.09695267677307
    },
    {
      "epoch": 1.6930580139160156e-05,
      "model_forward_time": 5.986117839813232,
      "step": 17753
    },
    {
      "epoch": 1.6930580139160156e-05,
      "step": 17753,
      "training_step_time": 17.449334621429443
    },
    {
      "epoch": 1.6931533813476564e-05,
      "model_forward_time": 5.583404064178467,
      "step": 17754
    },
    {
      "epoch": 1.6931533813476564e-05,
      "step": 17754,
      "training_step_time": 17.035598754882812
    },
    {
      "epoch": 1.6932487487792968e-05,
      "model_forward_time": 5.5926737785339355,
      "step": 17755
    },
    {
      "epoch": 1.6932487487792968e-05,
      "step": 17755,
      "training_step_time": 17.020336627960205
    },
    {
      "epoch": 1.6933441162109375e-05,
      "model_forward_time": 5.585038423538208,
      "step": 17756
    },
    {
      "epoch": 1.6933441162109375e-05,
      "step": 17756,
      "training_step_time": 17.009135246276855
    },
    {
      "epoch": 1.6934394836425783e-05,
      "model_forward_time": 5.5803022384643555,
      "step": 17757
    },
    {
      "epoch": 1.6934394836425783e-05,
      "step": 17757,
      "training_step_time": 17.01335620880127
    },
    {
      "epoch": 1.6935348510742187e-05,
      "model_forward_time": 5.949351787567139,
      "step": 17758
    },
    {
      "epoch": 1.6935348510742187e-05,
      "step": 17758,
      "training_step_time": 17.40402102470398
    },
    {
      "epoch": 1.6936302185058594e-05,
      "model_forward_time": 5.585624694824219,
      "step": 17759
    },
    {
      "epoch": 1.6936302185058594e-05,
      "step": 17759,
      "training_step_time": 17.028146266937256
    },
    {
      "epoch": 1.6937255859375e-05,
      "grad_norm": 1.5482906103134155,
      "learning_rate": 1e-05,
      "loss": 0.0683,
      "step": 17760
    },
    {
      "epoch": 1.6937255859375e-05,
      "model_forward_time": 5.5965070724487305,
      "step": 17760
    },
    {
      "dynamics_loss_avg": 0.056351514905691145,
      "epoch": 1.6937255859375e-05,
      "step": 17760
    },
    {
      "action_loss_avg": 0.006669022573623807,
      "epoch": 1.6937255859375e-05,
      "step": 17760
    },
    {
      "epoch": 1.6937255859375e-05,
      "step": 17760,
      "training_step_time": 17.057074069976807
    },
    {
      "epoch": 1.6938209533691406e-05,
      "model_forward_time": 5.621949672698975,
      "step": 17761
    },
    {
      "epoch": 1.6938209533691406e-05,
      "step": 17761,
      "training_step_time": 17.092933416366577
    },
    {
      "epoch": 1.6939163208007813e-05,
      "model_forward_time": 5.608914613723755,
      "step": 17762
    },
    {
      "epoch": 1.6939163208007813e-05,
      "step": 17762,
      "training_step_time": 17.100286722183228
    },
    {
      "epoch": 1.694011688232422e-05,
      "model_forward_time": 5.625155687332153,
      "step": 17763
    },
    {
      "epoch": 1.694011688232422e-05,
      "step": 17763,
      "training_step_time": 17.116605043411255
    },
    {
      "epoch": 1.6941070556640624e-05,
      "model_forward_time": 5.95543098449707,
      "step": 17764
    },
    {
      "epoch": 1.6941070556640624e-05,
      "step": 17764,
      "training_step_time": 17.456197261810303
    },
    {
      "epoch": 1.6942024230957032e-05,
      "model_forward_time": 5.65196681022644,
      "step": 17765
    },
    {
      "epoch": 1.6942024230957032e-05,
      "step": 17765,
      "training_step_time": 17.180585861206055
    },
    {
      "epoch": 1.6942977905273436e-05,
      "model_forward_time": 5.6252899169921875,
      "step": 17766
    },
    {
      "epoch": 1.6942977905273436e-05,
      "step": 17766,
      "training_step_time": 17.09832239151001
    },
    {
      "epoch": 1.6943931579589843e-05,
      "model_forward_time": 5.59395694732666,
      "step": 17767
    },
    {
      "epoch": 1.6943931579589843e-05,
      "step": 17767,
      "training_step_time": 17.021428108215332
    },
    {
      "epoch": 1.694488525390625e-05,
      "model_forward_time": 5.584245204925537,
      "step": 17768
    },
    {
      "epoch": 1.694488525390625e-05,
      "step": 17768,
      "training_step_time": 17.012263774871826
    },
    {
      "epoch": 1.6945838928222655e-05,
      "model_forward_time": 5.948829174041748,
      "step": 17769
    },
    {
      "epoch": 1.6945838928222655e-05,
      "step": 17769,
      "training_step_time": 17.3845477104187
    },
    {
      "epoch": 1.6946792602539062e-05,
      "grad_norm": 0.9671210646629333,
      "learning_rate": 1e-05,
      "loss": 0.0555,
      "step": 17770
    },
    {
      "epoch": 1.6946792602539062e-05,
      "model_forward_time": 5.580264568328857,
      "step": 17770
    },
    {
      "dynamics_loss_avg": 0.05230111451819539,
      "epoch": 1.6946792602539062e-05,
      "step": 17770
    },
    {
      "action_loss_avg": 0.007968537067063153,
      "epoch": 1.6946792602539062e-05,
      "step": 17770
    },
    {
      "epoch": 1.6946792602539062e-05,
      "step": 17770,
      "training_step_time": 17.000569581985474
    },
    {
      "epoch": 1.694774627685547e-05,
      "model_forward_time": 5.586941719055176,
      "step": 17771
    },
    {
      "epoch": 1.694774627685547e-05,
      "step": 17771,
      "training_step_time": 17.027079343795776
    },
    {
      "epoch": 1.6948699951171874e-05,
      "model_forward_time": 5.591175079345703,
      "step": 17772
    },
    {
      "epoch": 1.6948699951171874e-05,
      "step": 17772,
      "training_step_time": 17.03754734992981
    },
    {
      "epoch": 1.694965362548828e-05,
      "model_forward_time": 5.5906877517700195,
      "step": 17773
    },
    {
      "epoch": 1.694965362548828e-05,
      "step": 17773,
      "training_step_time": 17.07370924949646
    },
    {
      "epoch": 1.695060729980469e-05,
      "model_forward_time": 5.626177072525024,
      "step": 17774
    },
    {
      "epoch": 1.695060729980469e-05,
      "step": 17774,
      "training_step_time": 17.15393900871277
    },
    {
      "epoch": 1.6951560974121092e-05,
      "model_forward_time": 5.982227325439453,
      "step": 17775
    },
    {
      "epoch": 1.6951560974121092e-05,
      "step": 17775,
      "training_step_time": 17.491421937942505
    },
    {
      "epoch": 1.69525146484375e-05,
      "model_forward_time": 5.610383033752441,
      "step": 17776
    },
    {
      "epoch": 1.69525146484375e-05,
      "step": 17776,
      "training_step_time": 17.111312866210938
    },
    {
      "epoch": 1.6953468322753907e-05,
      "model_forward_time": 5.613595247268677,
      "step": 17777
    },
    {
      "epoch": 1.6953468322753907e-05,
      "step": 17777,
      "training_step_time": 17.09577703475952
    },
    {
      "epoch": 1.695442199707031e-05,
      "model_forward_time": 5.60754656791687,
      "step": 17778
    },
    {
      "epoch": 1.695442199707031e-05,
      "step": 17778,
      "training_step_time": 17.09919762611389
    },
    {
      "epoch": 1.695537567138672e-05,
      "model_forward_time": 5.601808071136475,
      "step": 17779
    },
    {
      "epoch": 1.695537567138672e-05,
      "step": 17779,
      "training_step_time": 17.06334686279297
    },
    {
      "epoch": 1.6956329345703126e-05,
      "grad_norm": 0.48432496190071106,
      "learning_rate": 1e-05,
      "loss": 0.063,
      "step": 17780
    },
    {
      "epoch": 1.6956329345703126e-05,
      "model_forward_time": 5.594651699066162,
      "step": 17780
    },
    {
      "dynamics_loss_avg": 0.0538899059407413,
      "epoch": 1.6956329345703126e-05,
      "step": 17780
    },
    {
      "action_loss_avg": 0.005268890433944762,
      "epoch": 1.6956329345703126e-05,
      "step": 17780
    },
    {
      "epoch": 1.6956329345703126e-05,
      "step": 17780,
      "training_step_time": 17.392789363861084
    },
    {
      "epoch": 1.695728302001953e-05,
      "model_forward_time": 5.584123373031616,
      "step": 17781
    },
    {
      "epoch": 1.695728302001953e-05,
      "step": 17781,
      "training_step_time": 17.015183210372925
    },
    {
      "epoch": 1.6958236694335938e-05,
      "model_forward_time": 5.5817718505859375,
      "step": 17782
    },
    {
      "epoch": 1.6958236694335938e-05,
      "step": 17782,
      "training_step_time": 17.03752303123474
    },
    {
      "epoch": 1.6959190368652345e-05,
      "model_forward_time": 5.597773551940918,
      "step": 17783
    },
    {
      "epoch": 1.6959190368652345e-05,
      "step": 17783,
      "training_step_time": 17.068642139434814
    },
    {
      "epoch": 1.696014404296875e-05,
      "model_forward_time": 5.602331638336182,
      "step": 17784
    },
    {
      "epoch": 1.696014404296875e-05,
      "step": 17784,
      "training_step_time": 17.07756805419922
    },
    {
      "epoch": 1.6961097717285156e-05,
      "model_forward_time": 5.600744724273682,
      "step": 17785
    },
    {
      "epoch": 1.6961097717285156e-05,
      "step": 17785,
      "training_step_time": 17.075393199920654
    },
    {
      "epoch": 1.6962051391601564e-05,
      "model_forward_time": 5.961472272872925,
      "step": 17786
    },
    {
      "epoch": 1.6962051391601564e-05,
      "step": 17786,
      "training_step_time": 17.428208112716675
    },
    {
      "epoch": 1.6963005065917968e-05,
      "model_forward_time": 5.599758148193359,
      "step": 17787
    },
    {
      "epoch": 1.6963005065917968e-05,
      "step": 17787,
      "training_step_time": 17.07822847366333
    },
    {
      "epoch": 1.6963958740234375e-05,
      "model_forward_time": 5.604979038238525,
      "step": 17788
    },
    {
      "epoch": 1.6963958740234375e-05,
      "step": 17788,
      "training_step_time": 17.069111585617065
    },
    {
      "epoch": 1.6964912414550783e-05,
      "model_forward_time": 5.621492862701416,
      "step": 17789
    },
    {
      "epoch": 1.6964912414550783e-05,
      "step": 17789,
      "training_step_time": 17.12335753440857
    },
    {
      "epoch": 1.6965866088867187e-05,
      "grad_norm": 0.6267495155334473,
      "learning_rate": 1e-05,
      "loss": 0.0424,
      "step": 17790
    },
    {
      "epoch": 1.6965866088867187e-05,
      "model_forward_time": 5.625542402267456,
      "step": 17790
    },
    {
      "dynamics_loss_avg": 0.04311159998178482,
      "epoch": 1.6965866088867187e-05,
      "step": 17790
    },
    {
      "action_loss_avg": 0.0038580505060963333,
      "epoch": 1.6965866088867187e-05,
      "step": 17790
    },
    {
      "epoch": 1.6965866088867187e-05,
      "step": 17790,
      "training_step_time": 17.142788410186768
    },
    {
      "epoch": 1.6966819763183594e-05,
      "model_forward_time": 5.997556209564209,
      "step": 17791
    },
    {
      "epoch": 1.6966819763183594e-05,
      "step": 17791,
      "training_step_time": 17.504980325698853
    },
    {
      "epoch": 1.69677734375e-05,
      "model_forward_time": 5.633938312530518,
      "step": 17792
    },
    {
      "epoch": 1.69677734375e-05,
      "step": 17792,
      "training_step_time": 17.157138109207153
    },
    {
      "epoch": 1.6968727111816406e-05,
      "model_forward_time": 5.616844654083252,
      "step": 17793
    },
    {
      "epoch": 1.6968727111816406e-05,
      "step": 17793,
      "training_step_time": 17.108227252960205
    },
    {
      "epoch": 1.6969680786132813e-05,
      "model_forward_time": 5.60605001449585,
      "step": 17794
    },
    {
      "epoch": 1.6969680786132813e-05,
      "step": 17794,
      "training_step_time": 17.098212003707886
    },
    {
      "epoch": 1.697063446044922e-05,
      "model_forward_time": 5.601657152175903,
      "step": 17795
    },
    {
      "epoch": 1.697063446044922e-05,
      "step": 17795,
      "training_step_time": 17.072949409484863
    },
    {
      "epoch": 1.6971588134765624e-05,
      "model_forward_time": 5.600902557373047,
      "step": 17796
    },
    {
      "epoch": 1.6971588134765624e-05,
      "step": 17796,
      "training_step_time": 17.070727348327637
    },
    {
      "epoch": 1.6972541809082032e-05,
      "model_forward_time": 5.934379577636719,
      "step": 17797
    },
    {
      "epoch": 1.6972541809082032e-05,
      "step": 17797,
      "training_step_time": 17.3800687789917
    },
    {
      "epoch": 1.6973495483398436e-05,
      "model_forward_time": 5.591169118881226,
      "step": 17798
    },
    {
      "epoch": 1.6973495483398436e-05,
      "step": 17798,
      "training_step_time": 17.052736282348633
    },
    {
      "epoch": 1.6974449157714843e-05,
      "model_forward_time": 5.603440523147583,
      "step": 17799
    },
    {
      "epoch": 1.6974449157714843e-05,
      "step": 17799,
      "training_step_time": 17.077455043792725
    },
    {
      "epoch": 1.697540283203125e-05,
      "grad_norm": 0.8666419386863708,
      "learning_rate": 1e-05,
      "loss": 0.0622,
      "step": 17800
    },
    {
      "epoch": 1.697540283203125e-05,
      "model_forward_time": 5.61378812789917,
      "step": 17800
    },
    {
      "dynamics_loss_avg": 0.05368848014622927,
      "epoch": 1.697540283203125e-05,
      "step": 17800
    },
    {
      "action_loss_avg": 0.0057698852731846275,
      "epoch": 1.697540283203125e-05,
      "step": 17800
    },
    {
      "epoch": 1.697540283203125e-05,
      "step": 17800,
      "training_step_time": 17.103123426437378
    },
    {
      "epoch": 1.6976356506347655e-05,
      "model_forward_time": 5.612750291824341,
      "step": 17801
    },
    {
      "epoch": 1.6976356506347655e-05,
      "step": 17801,
      "training_step_time": 17.12875771522522
    },
    {
      "epoch": 1.6977310180664062e-05,
      "model_forward_time": 5.9937615394592285,
      "step": 17802
    },
    {
      "epoch": 1.6977310180664062e-05,
      "step": 17802,
      "training_step_time": 17.466963291168213
    },
    {
      "epoch": 1.697826385498047e-05,
      "model_forward_time": 5.59225869178772,
      "step": 17803
    },
    {
      "epoch": 1.697826385498047e-05,
      "step": 17803,
      "training_step_time": 17.036954641342163
    },
    {
      "epoch": 1.6979217529296874e-05,
      "model_forward_time": 5.583655834197998,
      "step": 17804
    },
    {
      "epoch": 1.6979217529296874e-05,
      "step": 17804,
      "training_step_time": 17.045546770095825
    },
    {
      "epoch": 1.698017120361328e-05,
      "model_forward_time": 5.596346616744995,
      "step": 17805
    },
    {
      "epoch": 1.698017120361328e-05,
      "step": 17805,
      "training_step_time": 17.078797817230225
    },
    {
      "epoch": 1.698112487792969e-05,
      "model_forward_time": 5.613209009170532,
      "step": 17806
    },
    {
      "epoch": 1.698112487792969e-05,
      "step": 17806,
      "training_step_time": 17.102694511413574
    },
    {
      "epoch": 1.6982078552246092e-05,
      "model_forward_time": 5.616479158401489,
      "step": 17807
    },
    {
      "epoch": 1.6982078552246092e-05,
      "step": 17807,
      "training_step_time": 17.101309061050415
    },
    {
      "epoch": 1.69830322265625e-05,
      "model_forward_time": 5.9773149490356445,
      "step": 17808
    },
    {
      "epoch": 1.69830322265625e-05,
      "step": 17808,
      "training_step_time": 17.462393760681152
    },
    {
      "epoch": 1.6983985900878907e-05,
      "model_forward_time": 5.61501145362854,
      "step": 17809
    },
    {
      "epoch": 1.6983985900878907e-05,
      "step": 17809,
      "training_step_time": 17.106158018112183
    },
    {
      "epoch": 1.698493957519531e-05,
      "grad_norm": 0.5950670838356018,
      "learning_rate": 1e-05,
      "loss": 0.0558,
      "step": 17810
    },
    {
      "epoch": 1.698493957519531e-05,
      "model_forward_time": 5.611159563064575,
      "step": 17810
    },
    {
      "dynamics_loss_avg": 0.05635298769921064,
      "epoch": 1.698493957519531e-05,
      "step": 17810
    },
    {
      "action_loss_avg": 0.005724845110671594,
      "epoch": 1.698493957519531e-05,
      "step": 17810
    },
    {
      "epoch": 1.698493957519531e-05,
      "step": 17810,
      "training_step_time": 17.10314965248108
    },
    {
      "epoch": 1.698589324951172e-05,
      "model_forward_time": 5.611708879470825,
      "step": 17811
    },
    {
      "epoch": 1.698589324951172e-05,
      "step": 17811,
      "training_step_time": 17.11322283744812
    },
    {
      "epoch": 1.6986846923828126e-05,
      "model_forward_time": 5.612777948379517,
      "step": 17812
    },
    {
      "epoch": 1.6986846923828126e-05,
      "step": 17812,
      "training_step_time": 17.09556531906128
    },
    {
      "epoch": 1.698780059814453e-05,
      "model_forward_time": 5.594555616378784,
      "step": 17813
    },
    {
      "epoch": 1.698780059814453e-05,
      "step": 17813,
      "training_step_time": 17.41572403907776
    },
    {
      "epoch": 1.6988754272460938e-05,
      "model_forward_time": 5.600180625915527,
      "step": 17814
    },
    {
      "epoch": 1.6988754272460938e-05,
      "step": 17814,
      "training_step_time": 17.018015146255493
    },
    {
      "epoch": 1.6989707946777345e-05,
      "model_forward_time": 5.577713966369629,
      "step": 17815
    },
    {
      "epoch": 1.6989707946777345e-05,
      "step": 17815,
      "training_step_time": 17.027896881103516
    },
    {
      "epoch": 1.699066162109375e-05,
      "model_forward_time": 5.580834865570068,
      "step": 17816
    },
    {
      "epoch": 1.699066162109375e-05,
      "step": 17816,
      "training_step_time": 17.019174337387085
    },
    {
      "epoch": 1.6991615295410156e-05,
      "model_forward_time": 5.600171327590942,
      "step": 17817
    },
    {
      "epoch": 1.6991615295410156e-05,
      "step": 17817,
      "training_step_time": 17.07575273513794
    },
    {
      "epoch": 1.6992568969726564e-05,
      "model_forward_time": 5.594431638717651,
      "step": 17818
    },
    {
      "epoch": 1.6992568969726564e-05,
      "step": 17818,
      "training_step_time": 17.078826665878296
    },
    {
      "epoch": 1.6993522644042968e-05,
      "model_forward_time": 5.965819597244263,
      "step": 17819
    },
    {
      "epoch": 1.6993522644042968e-05,
      "step": 17819,
      "training_step_time": 17.447583198547363
    },
    {
      "epoch": 1.6994476318359375e-05,
      "grad_norm": 0.6125932931900024,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 17820
    },
    {
      "epoch": 1.6994476318359375e-05,
      "model_forward_time": 5.611250400543213,
      "step": 17820
    },
    {
      "dynamics_loss_avg": 0.06516670845448971,
      "epoch": 1.6994476318359375e-05,
      "step": 17820
    },
    {
      "action_loss_avg": 0.007895458769053221,
      "epoch": 1.6994476318359375e-05,
      "step": 17820
    },
    {
      "epoch": 1.6994476318359375e-05,
      "step": 17820,
      "training_step_time": 17.06664752960205
    },
    {
      "epoch": 1.6995429992675783e-05,
      "model_forward_time": 5.582054138183594,
      "step": 17821
    },
    {
      "epoch": 1.6995429992675783e-05,
      "step": 17821,
      "training_step_time": 17.00222110748291
    },
    {
      "epoch": 1.6996383666992187e-05,
      "model_forward_time": 5.591118574142456,
      "step": 17822
    },
    {
      "epoch": 1.6996383666992187e-05,
      "step": 17822,
      "training_step_time": 17.0288405418396
    },
    {
      "epoch": 1.6997337341308594e-05,
      "model_forward_time": 5.583332061767578,
      "step": 17823
    },
    {
      "epoch": 1.6997337341308594e-05,
      "step": 17823,
      "training_step_time": 16.999696493148804
    },
    {
      "epoch": 1.6998291015625e-05,
      "model_forward_time": 5.985727787017822,
      "step": 17824
    },
    {
      "epoch": 1.6998291015625e-05,
      "step": 17824,
      "training_step_time": 17.41259741783142
    },
    {
      "epoch": 1.6999244689941406e-05,
      "model_forward_time": 5.597380876541138,
      "step": 17825
    },
    {
      "epoch": 1.6999244689941406e-05,
      "step": 17825,
      "training_step_time": 17.035735845565796
    },
    {
      "epoch": 1.7000198364257813e-05,
      "model_forward_time": 5.59623908996582,
      "step": 17826
    },
    {
      "epoch": 1.7000198364257813e-05,
      "step": 17826,
      "training_step_time": 17.057049036026
    },
    {
      "epoch": 1.700115203857422e-05,
      "model_forward_time": 5.608554363250732,
      "step": 17827
    },
    {
      "epoch": 1.700115203857422e-05,
      "step": 17827,
      "training_step_time": 17.091663599014282
    },
    {
      "epoch": 1.7002105712890624e-05,
      "model_forward_time": 5.606320381164551,
      "step": 17828
    },
    {
      "epoch": 1.7002105712890624e-05,
      "step": 17828,
      "training_step_time": 17.102699756622314
    },
    {
      "epoch": 1.7003059387207032e-05,
      "model_forward_time": 5.613084077835083,
      "step": 17829
    },
    {
      "epoch": 1.7003059387207032e-05,
      "step": 17829,
      "training_step_time": 17.135590076446533
    },
    {
      "epoch": 1.7004013061523436e-05,
      "grad_norm": 0.3156641721725464,
      "learning_rate": 1e-05,
      "loss": 0.0579,
      "step": 17830
    },
    {
      "epoch": 1.7004013061523436e-05,
      "model_forward_time": 5.962950229644775,
      "step": 17830
    },
    {
      "dynamics_loss_avg": 0.05237738490104675,
      "epoch": 1.7004013061523436e-05,
      "step": 17830
    },
    {
      "action_loss_avg": 0.007562096195761115,
      "epoch": 1.7004013061523436e-05,
      "step": 17830
    },
    {
      "epoch": 1.7004013061523436e-05,
      "step": 17830,
      "training_step_time": 17.43968439102173
    },
    {
      "epoch": 1.7004966735839843e-05,
      "model_forward_time": 5.6080098152160645,
      "step": 17831
    },
    {
      "epoch": 1.7004966735839843e-05,
      "step": 17831,
      "training_step_time": 17.058600187301636
    },
    {
      "epoch": 1.700592041015625e-05,
      "model_forward_time": 5.593560218811035,
      "step": 17832
    },
    {
      "epoch": 1.700592041015625e-05,
      "step": 17832,
      "training_step_time": 17.032798767089844
    },
    {
      "epoch": 1.7006874084472655e-05,
      "model_forward_time": 5.597662925720215,
      "step": 17833
    },
    {
      "epoch": 1.7006874084472655e-05,
      "step": 17833,
      "training_step_time": 17.039960622787476
    },
    {
      "epoch": 1.7007827758789062e-05,
      "model_forward_time": 5.596781969070435,
      "step": 17834
    },
    {
      "epoch": 1.7007827758789062e-05,
      "step": 17834,
      "training_step_time": 17.043285846710205
    },
    {
      "epoch": 1.700878143310547e-05,
      "model_forward_time": 5.979291915893555,
      "step": 17835
    },
    {
      "epoch": 1.700878143310547e-05,
      "step": 17835,
      "training_step_time": 17.468929290771484
    },
    {
      "epoch": 1.7009735107421874e-05,
      "model_forward_time": 5.614116191864014,
      "step": 17836
    },
    {
      "epoch": 1.7009735107421874e-05,
      "step": 17836,
      "training_step_time": 17.11268401145935
    },
    {
      "epoch": 1.701068878173828e-05,
      "model_forward_time": 5.6061975955963135,
      "step": 17837
    },
    {
      "epoch": 1.701068878173828e-05,
      "step": 17837,
      "training_step_time": 17.08920407295227
    },
    {
      "epoch": 1.701164245605469e-05,
      "model_forward_time": 5.612396478652954,
      "step": 17838
    },
    {
      "epoch": 1.701164245605469e-05,
      "step": 17838,
      "training_step_time": 17.08265781402588
    },
    {
      "epoch": 1.7012596130371092e-05,
      "model_forward_time": 5.600158929824829,
      "step": 17839
    },
    {
      "epoch": 1.7012596130371092e-05,
      "step": 17839,
      "training_step_time": 17.056018829345703
    },
    {
      "epoch": 1.70135498046875e-05,
      "grad_norm": 0.34099653363227844,
      "learning_rate": 1e-05,
      "loss": 0.0635,
      "step": 17840
    },
    {
      "epoch": 1.70135498046875e-05,
      "model_forward_time": 5.632735967636108,
      "step": 17840
    },
    {
      "dynamics_loss_avg": 0.05416972674429417,
      "epoch": 1.70135498046875e-05,
      "step": 17840
    },
    {
      "action_loss_avg": 0.006023526866920292,
      "epoch": 1.70135498046875e-05,
      "step": 17840
    },
    {
      "epoch": 1.70135498046875e-05,
      "step": 17840,
      "training_step_time": 17.05540418624878
    },
    {
      "epoch": 1.7014503479003907e-05,
      "model_forward_time": 6.1292805671691895,
      "step": 17841
    },
    {
      "epoch": 1.7014503479003907e-05,
      "step": 17841,
      "training_step_time": 17.603187561035156
    },
    {
      "epoch": 1.701545715332031e-05,
      "model_forward_time": 5.615587949752808,
      "step": 17842
    },
    {
      "epoch": 1.701545715332031e-05,
      "step": 17842,
      "training_step_time": 17.08629035949707
    },
    {
      "epoch": 1.701641082763672e-05,
      "model_forward_time": 5.634121417999268,
      "step": 17843
    },
    {
      "epoch": 1.701641082763672e-05,
      "step": 17843,
      "training_step_time": 17.10642981529236
    },
    {
      "epoch": 1.7017364501953126e-05,
      "model_forward_time": 5.630557298660278,
      "step": 17844
    },
    {
      "epoch": 1.7017364501953126e-05,
      "step": 17844,
      "training_step_time": 17.092427968978882
    },
    {
      "epoch": 1.701831817626953e-05,
      "model_forward_time": 5.643202304840088,
      "step": 17845
    },
    {
      "epoch": 1.701831817626953e-05,
      "step": 17845,
      "training_step_time": 17.093056678771973
    },
    {
      "epoch": 1.7019271850585938e-05,
      "model_forward_time": 5.594379425048828,
      "step": 17846
    },
    {
      "epoch": 1.7019271850585938e-05,
      "step": 17846,
      "training_step_time": 17.485165119171143
    },
    {
      "epoch": 1.7020225524902345e-05,
      "model_forward_time": 5.5964577198028564,
      "step": 17847
    },
    {
      "epoch": 1.7020225524902345e-05,
      "step": 17847,
      "training_step_time": 17.065239429473877
    },
    {
      "epoch": 1.702117919921875e-05,
      "model_forward_time": 5.604763031005859,
      "step": 17848
    },
    {
      "epoch": 1.702117919921875e-05,
      "step": 17848,
      "training_step_time": 17.084762811660767
    },
    {
      "epoch": 1.7022132873535156e-05,
      "model_forward_time": 5.611975431442261,
      "step": 17849
    },
    {
      "epoch": 1.7022132873535156e-05,
      "step": 17849,
      "training_step_time": 17.049464225769043
    },
    {
      "epoch": 1.7023086547851564e-05,
      "grad_norm": 0.6481676697731018,
      "learning_rate": 1e-05,
      "loss": 0.0757,
      "step": 17850
    },
    {
      "epoch": 1.7023086547851564e-05,
      "model_forward_time": 5.601322650909424,
      "step": 17850
    },
    {
      "dynamics_loss_avg": 0.06702056750655175,
      "epoch": 1.7023086547851564e-05,
      "step": 17850
    },
    {
      "action_loss_avg": 0.007022279710508883,
      "epoch": 1.7023086547851564e-05,
      "step": 17850
    },
    {
      "epoch": 1.7023086547851564e-05,
      "step": 17850,
      "training_step_time": 17.06499934196472
    },
    {
      "epoch": 1.7024040222167968e-05,
      "model_forward_time": 5.593333959579468,
      "step": 17851
    },
    {
      "epoch": 1.7024040222167968e-05,
      "step": 17851,
      "training_step_time": 17.032607555389404
    },
    {
      "epoch": 1.7024993896484375e-05,
      "model_forward_time": 6.00324273109436,
      "step": 17852
    },
    {
      "epoch": 1.7024993896484375e-05,
      "step": 17852,
      "training_step_time": 17.4399631023407
    },
    {
      "epoch": 1.7025947570800783e-05,
      "model_forward_time": 5.593269348144531,
      "step": 17853
    },
    {
      "epoch": 1.7025947570800783e-05,
      "step": 17853,
      "training_step_time": 17.041401147842407
    },
    {
      "epoch": 1.7026901245117187e-05,
      "model_forward_time": 5.602963447570801,
      "step": 17854
    },
    {
      "epoch": 1.7026901245117187e-05,
      "step": 17854,
      "training_step_time": 17.071767807006836
    },
    {
      "epoch": 1.7027854919433594e-05,
      "model_forward_time": 5.602112293243408,
      "step": 17855
    },
    {
      "epoch": 1.7027854919433594e-05,
      "step": 17855,
      "training_step_time": 17.074862003326416
    },
    {
      "epoch": 1.702880859375e-05,
      "model_forward_time": 5.616497993469238,
      "step": 17856
    },
    {
      "epoch": 1.702880859375e-05,
      "step": 17856,
      "training_step_time": 17.090174198150635
    },
    {
      "epoch": 1.7029762268066406e-05,
      "model_forward_time": 5.96028733253479,
      "step": 17857
    },
    {
      "epoch": 1.7029762268066406e-05,
      "step": 17857,
      "training_step_time": 17.38343381881714
    },
    {
      "epoch": 1.7030715942382813e-05,
      "model_forward_time": 5.5851781368255615,
      "step": 17858
    },
    {
      "epoch": 1.7030715942382813e-05,
      "step": 17858,
      "training_step_time": 17.01733684539795
    },
    {
      "epoch": 1.703166961669922e-05,
      "model_forward_time": 5.594995498657227,
      "step": 17859
    },
    {
      "epoch": 1.703166961669922e-05,
      "step": 17859,
      "training_step_time": 17.04211926460266
    },
    {
      "epoch": 1.7032623291015624e-05,
      "grad_norm": 0.8160116076469421,
      "learning_rate": 1e-05,
      "loss": 0.0578,
      "step": 17860
    },
    {
      "epoch": 1.7032623291015624e-05,
      "model_forward_time": 5.593003749847412,
      "step": 17860
    },
    {
      "dynamics_loss_avg": 0.050409177877008916,
      "epoch": 1.7032623291015624e-05,
      "step": 17860
    },
    {
      "action_loss_avg": 0.006934247410390526,
      "epoch": 1.7032623291015624e-05,
      "step": 17860
    },
    {
      "epoch": 1.7032623291015624e-05,
      "step": 17860,
      "training_step_time": 17.03725290298462
    },
    {
      "epoch": 1.7033576965332032e-05,
      "model_forward_time": 5.601006746292114,
      "step": 17861
    },
    {
      "epoch": 1.7033576965332032e-05,
      "step": 17861,
      "training_step_time": 17.07561182975769
    },
    {
      "epoch": 1.7034530639648436e-05,
      "model_forward_time": 5.603999137878418,
      "step": 17862
    },
    {
      "epoch": 1.7034530639648436e-05,
      "step": 17862,
      "training_step_time": 17.095219612121582
    },
    {
      "epoch": 1.7035484313964843e-05,
      "model_forward_time": 5.952146053314209,
      "step": 17863
    },
    {
      "epoch": 1.7035484313964843e-05,
      "step": 17863,
      "training_step_time": 17.44196629524231
    },
    {
      "epoch": 1.703643798828125e-05,
      "model_forward_time": 5.608265161514282,
      "step": 17864
    },
    {
      "epoch": 1.703643798828125e-05,
      "step": 17864,
      "training_step_time": 17.086307764053345
    },
    {
      "epoch": 1.7037391662597655e-05,
      "model_forward_time": 5.634122848510742,
      "step": 17865
    },
    {
      "epoch": 1.7037391662597655e-05,
      "step": 17865,
      "training_step_time": 17.109182596206665
    },
    {
      "epoch": 1.7038345336914062e-05,
      "model_forward_time": 5.623907804489136,
      "step": 17866
    },
    {
      "epoch": 1.7038345336914062e-05,
      "step": 17866,
      "training_step_time": 17.092786073684692
    },
    {
      "epoch": 1.703929901123047e-05,
      "model_forward_time": 5.599640607833862,
      "step": 17867
    },
    {
      "epoch": 1.703929901123047e-05,
      "step": 17867,
      "training_step_time": 17.07162642478943
    },
    {
      "epoch": 1.7040252685546874e-05,
      "model_forward_time": 6.000899314880371,
      "step": 17868
    },
    {
      "epoch": 1.7040252685546874e-05,
      "step": 17868,
      "training_step_time": 17.44637632369995
    },
    {
      "epoch": 1.704120635986328e-05,
      "model_forward_time": 4.278692007064819,
      "step": 17869
    },
    {
      "epoch": 1.704120635986328e-05,
      "step": 17869,
      "training_step_time": 12.827659845352173
    },
    {
      "epoch": 1.704216003417969e-05,
      "grad_norm": 0.38673484325408936,
      "learning_rate": 1e-05,
      "loss": 0.0672,
      "step": 17870
    },
    {
      "epoch": 1.704216003417969e-05,
      "model_forward_time": 5.613831520080566,
      "step": 17870
    },
    {
      "dynamics_loss_avg": 0.06644363570958375,
      "epoch": 1.704216003417969e-05,
      "step": 17870
    },
    {
      "action_loss_avg": 0.0077119907829910515,
      "epoch": 1.704216003417969e-05,
      "step": 17870
    },
    {
      "epoch": 1.704216003417969e-05,
      "step": 17870,
      "training_step_time": 17.123398542404175
    },
    {
      "epoch": 1.7043113708496092e-05,
      "model_forward_time": 5.622525691986084,
      "step": 17871
    },
    {
      "epoch": 1.7043113708496092e-05,
      "step": 17871,
      "training_step_time": 17.128292083740234
    },
    {
      "epoch": 1.70440673828125e-05,
      "model_forward_time": 5.601989984512329,
      "step": 17872
    },
    {
      "epoch": 1.70440673828125e-05,
      "step": 17872,
      "training_step_time": 17.038010597229004
    },
    {
      "epoch": 1.7045021057128907e-05,
      "model_forward_time": 5.58855128288269,
      "step": 17873
    },
    {
      "epoch": 1.7045021057128907e-05,
      "step": 17873,
      "training_step_time": 17.046551942825317
    },
    {
      "epoch": 1.704597473144531e-05,
      "model_forward_time": 5.951519012451172,
      "step": 17874
    },
    {
      "epoch": 1.704597473144531e-05,
      "step": 17874,
      "training_step_time": 17.385831117630005
    },
    {
      "epoch": 1.704692840576172e-05,
      "model_forward_time": 5.5897746086120605,
      "step": 17875
    },
    {
      "epoch": 1.704692840576172e-05,
      "step": 17875,
      "training_step_time": 17.048815965652466
    },
    {
      "epoch": 1.7047882080078126e-05,
      "model_forward_time": 5.612118721008301,
      "step": 17876
    },
    {
      "epoch": 1.7047882080078126e-05,
      "step": 17876,
      "training_step_time": 17.10036611557007
    },
    {
      "epoch": 1.704883575439453e-05,
      "model_forward_time": 5.615202903747559,
      "step": 17877
    },
    {
      "epoch": 1.704883575439453e-05,
      "step": 17877,
      "training_step_time": 17.110563039779663
    },
    {
      "epoch": 1.7049789428710938e-05,
      "model_forward_time": 5.611953258514404,
      "step": 17878
    },
    {
      "epoch": 1.7049789428710938e-05,
      "step": 17878,
      "training_step_time": 17.108494758605957
    },
    {
      "epoch": 1.7050743103027345e-05,
      "model_forward_time": 5.613946437835693,
      "step": 17879
    },
    {
      "epoch": 1.7050743103027345e-05,
      "step": 17879,
      "training_step_time": 17.099128484725952
    },
    {
      "epoch": 1.705169677734375e-05,
      "grad_norm": 0.8551828265190125,
      "learning_rate": 1e-05,
      "loss": 0.0627,
      "step": 17880
    },
    {
      "epoch": 1.705169677734375e-05,
      "model_forward_time": 5.9959166049957275,
      "step": 17880
    },
    {
      "dynamics_loss_avg": 0.05558780021965504,
      "epoch": 1.705169677734375e-05,
      "step": 17880
    },
    {
      "action_loss_avg": 0.005583720421418548,
      "epoch": 1.705169677734375e-05,
      "step": 17880
    },
    {
      "epoch": 1.705169677734375e-05,
      "step": 17880,
      "training_step_time": 17.490253925323486
    },
    {
      "epoch": 1.7052650451660156e-05,
      "model_forward_time": 5.615511894226074,
      "step": 17881
    },
    {
      "epoch": 1.7052650451660156e-05,
      "step": 17881,
      "training_step_time": 17.12225604057312
    },
    {
      "epoch": 1.7053604125976564e-05,
      "model_forward_time": 5.611703157424927,
      "step": 17882
    },
    {
      "epoch": 1.7053604125976564e-05,
      "step": 17882,
      "training_step_time": 17.10607647895813
    },
    {
      "epoch": 1.7054557800292968e-05,
      "model_forward_time": 5.61655592918396,
      "step": 17883
    },
    {
      "epoch": 1.7054557800292968e-05,
      "step": 17883,
      "training_step_time": 17.10715103149414
    },
    {
      "epoch": 1.7055511474609375e-05,
      "model_forward_time": 5.610163927078247,
      "step": 17884
    },
    {
      "epoch": 1.7055511474609375e-05,
      "step": 17884,
      "training_step_time": 17.099508047103882
    },
    {
      "epoch": 1.7056465148925783e-05,
      "model_forward_time": 5.962225675582886,
      "step": 17885
    },
    {
      "epoch": 1.7056465148925783e-05,
      "step": 17885,
      "training_step_time": 17.430594444274902
    },
    {
      "epoch": 1.7057418823242187e-05,
      "model_forward_time": 5.590224027633667,
      "step": 17886
    },
    {
      "epoch": 1.7057418823242187e-05,
      "step": 17886,
      "training_step_time": 16.999114751815796
    },
    {
      "epoch": 1.7058372497558594e-05,
      "model_forward_time": 5.5800440311431885,
      "step": 17887
    },
    {
      "epoch": 1.7058372497558594e-05,
      "step": 17887,
      "training_step_time": 17.01609492301941
    },
    {
      "epoch": 1.7059326171875e-05,
      "model_forward_time": 5.575382947921753,
      "step": 17888
    },
    {
      "epoch": 1.7059326171875e-05,
      "step": 17888,
      "training_step_time": 16.993468046188354
    },
    {
      "epoch": 1.7060279846191406e-05,
      "model_forward_time": 5.580323696136475,
      "step": 17889
    },
    {
      "epoch": 1.7060279846191406e-05,
      "step": 17889,
      "training_step_time": 17.01154351234436
    },
    {
      "epoch": 1.7061233520507813e-05,
      "grad_norm": 0.6863687038421631,
      "learning_rate": 1e-05,
      "loss": 0.0724,
      "step": 17890
    },
    {
      "epoch": 1.7061233520507813e-05,
      "model_forward_time": 5.9671149253845215,
      "step": 17890
    },
    {
      "dynamics_loss_avg": 0.06399406995624304,
      "epoch": 1.7061233520507813e-05,
      "step": 17890
    },
    {
      "action_loss_avg": 0.007766081765294075,
      "epoch": 1.7061233520507813e-05,
      "step": 17890
    },
    {
      "epoch": 1.7061233520507813e-05,
      "step": 17890,
      "training_step_time": 17.408915519714355
    },
    {
      "epoch": 1.7062187194824217e-05,
      "model_forward_time": 5.602431774139404,
      "step": 17891
    },
    {
      "epoch": 1.7062187194824217e-05,
      "step": 17891,
      "training_step_time": 17.057837963104248
    },
    {
      "epoch": 1.7063140869140624e-05,
      "model_forward_time": 5.59426474571228,
      "step": 17892
    },
    {
      "epoch": 1.7063140869140624e-05,
      "step": 17892,
      "training_step_time": 17.081480979919434
    },
    {
      "epoch": 1.7064094543457032e-05,
      "model_forward_time": 5.623190641403198,
      "step": 17893
    },
    {
      "epoch": 1.7064094543457032e-05,
      "step": 17893,
      "training_step_time": 17.14876365661621
    },
    {
      "epoch": 1.7065048217773436e-05,
      "model_forward_time": 5.619861841201782,
      "step": 17894
    },
    {
      "epoch": 1.7065048217773436e-05,
      "step": 17894,
      "training_step_time": 17.11831498146057
    },
    {
      "epoch": 1.7066001892089843e-05,
      "model_forward_time": 5.616084575653076,
      "step": 17895
    },
    {
      "epoch": 1.7066001892089843e-05,
      "step": 17895,
      "training_step_time": 17.07655382156372
    },
    {
      "epoch": 1.706695556640625e-05,
      "model_forward_time": 5.957545280456543,
      "step": 17896
    },
    {
      "epoch": 1.706695556640625e-05,
      "step": 17896,
      "training_step_time": 17.422468662261963
    },
    {
      "epoch": 1.7067909240722655e-05,
      "model_forward_time": 5.598475456237793,
      "step": 17897
    },
    {
      "epoch": 1.7067909240722655e-05,
      "step": 17897,
      "training_step_time": 17.040414333343506
    },
    {
      "epoch": 1.7068862915039062e-05,
      "model_forward_time": 5.593630313873291,
      "step": 17898
    },
    {
      "epoch": 1.7068862915039062e-05,
      "step": 17898,
      "training_step_time": 17.048051118850708
    },
    {
      "epoch": 1.706981658935547e-05,
      "model_forward_time": 5.602203130722046,
      "step": 17899
    },
    {
      "epoch": 1.706981658935547e-05,
      "step": 17899,
      "training_step_time": 17.07395648956299
    },
    {
      "epoch": 1.7070770263671874e-05,
      "grad_norm": 0.7714143991470337,
      "learning_rate": 1e-05,
      "loss": 0.0622,
      "step": 17900
    },
    {
      "epoch": 1.7070770263671874e-05,
      "model_forward_time": 5.5979838371276855,
      "step": 17900
    },
    {
      "dynamics_loss_avg": 0.06011543460190296,
      "epoch": 1.7070770263671874e-05,
      "step": 17900
    },
    {
      "action_loss_avg": 0.006878153490833938,
      "epoch": 1.7070770263671874e-05,
      "step": 17900
    },
    {
      "epoch": 1.7070770263671874e-05,
      "step": 17900,
      "training_step_time": 17.080756425857544
    },
    {
      "epoch": 1.707172393798828e-05,
      "model_forward_time": 5.988125324249268,
      "step": 17901
    },
    {
      "epoch": 1.707172393798828e-05,
      "step": 17901,
      "training_step_time": 17.469866037368774
    },
    {
      "epoch": 1.707267761230469e-05,
      "model_forward_time": 5.603135347366333,
      "step": 17902
    },
    {
      "epoch": 1.707267761230469e-05,
      "step": 17902,
      "training_step_time": 17.076661825180054
    },
    {
      "epoch": 1.7073631286621093e-05,
      "model_forward_time": 5.600984334945679,
      "step": 17903
    },
    {
      "epoch": 1.7073631286621093e-05,
      "step": 17903,
      "training_step_time": 17.046783447265625
    },
    {
      "epoch": 1.70745849609375e-05,
      "model_forward_time": 5.600562334060669,
      "step": 17904
    },
    {
      "epoch": 1.70745849609375e-05,
      "step": 17904,
      "training_step_time": 17.035760641098022
    },
    {
      "epoch": 1.7075538635253907e-05,
      "model_forward_time": 5.595669507980347,
      "step": 17905
    },
    {
      "epoch": 1.7075538635253907e-05,
      "step": 17905,
      "training_step_time": 17.04206395149231
    },
    {
      "epoch": 1.707649230957031e-05,
      "model_forward_time": 5.5968756675720215,
      "step": 17906
    },
    {
      "epoch": 1.707649230957031e-05,
      "step": 17906,
      "training_step_time": 17.034002542495728
    },
    {
      "epoch": 1.707744598388672e-05,
      "model_forward_time": 5.946950435638428,
      "step": 17907
    },
    {
      "epoch": 1.707744598388672e-05,
      "step": 17907,
      "training_step_time": 17.41742753982544
    },
    {
      "epoch": 1.7078399658203126e-05,
      "model_forward_time": 5.608012914657593,
      "step": 17908
    },
    {
      "epoch": 1.7078399658203126e-05,
      "step": 17908,
      "training_step_time": 17.10458016395569
    },
    {
      "epoch": 1.707935333251953e-05,
      "model_forward_time": 5.619957208633423,
      "step": 17909
    },
    {
      "epoch": 1.707935333251953e-05,
      "step": 17909,
      "training_step_time": 17.13159155845642
    },
    {
      "epoch": 1.7080307006835938e-05,
      "grad_norm": 0.8822205662727356,
      "learning_rate": 1e-05,
      "loss": 0.0685,
      "step": 17910
    },
    {
      "epoch": 1.7080307006835938e-05,
      "model_forward_time": 5.619627952575684,
      "step": 17910
    },
    {
      "dynamics_loss_avg": 0.05347755830734968,
      "epoch": 1.7080307006835938e-05,
      "step": 17910
    },
    {
      "action_loss_avg": 0.005800424120388925,
      "epoch": 1.7080307006835938e-05,
      "step": 17910
    },
    {
      "epoch": 1.7080307006835938e-05,
      "step": 17910,
      "training_step_time": 17.125895738601685
    },
    {
      "epoch": 1.7081260681152345e-05,
      "model_forward_time": 5.603935718536377,
      "step": 17911
    },
    {
      "epoch": 1.7081260681152345e-05,
      "step": 17911,
      "training_step_time": 17.0919086933136
    },
    {
      "epoch": 1.708221435546875e-05,
      "model_forward_time": 5.624643325805664,
      "step": 17912
    },
    {
      "epoch": 1.708221435546875e-05,
      "step": 17912,
      "training_step_time": 17.466017723083496
    },
    {
      "epoch": 1.7083168029785156e-05,
      "model_forward_time": 5.606416940689087,
      "step": 17913
    },
    {
      "epoch": 1.7083168029785156e-05,
      "step": 17913,
      "training_step_time": 17.08249306678772
    },
    {
      "epoch": 1.7084121704101564e-05,
      "model_forward_time": 5.601512432098389,
      "step": 17914
    },
    {
      "epoch": 1.7084121704101564e-05,
      "step": 17914,
      "training_step_time": 17.079118728637695
    },
    {
      "epoch": 1.7085075378417968e-05,
      "model_forward_time": 5.6075849533081055,
      "step": 17915
    },
    {
      "epoch": 1.7085075378417968e-05,
      "step": 17915,
      "training_step_time": 17.086363792419434
    },
    {
      "epoch": 1.7086029052734375e-05,
      "model_forward_time": 5.600164175033569,
      "step": 17916
    },
    {
      "epoch": 1.7086029052734375e-05,
      "step": 17916,
      "training_step_time": 17.084811210632324
    },
    {
      "epoch": 1.7086982727050783e-05,
      "model_forward_time": 5.611130237579346,
      "step": 17917
    },
    {
      "epoch": 1.7086982727050783e-05,
      "step": 17917,
      "training_step_time": 17.10843586921692
    },
    {
      "epoch": 1.7087936401367187e-05,
      "model_forward_time": 5.970670700073242,
      "step": 17918
    },
    {
      "epoch": 1.7087936401367187e-05,
      "step": 17918,
      "training_step_time": 17.47318387031555
    },
    {
      "epoch": 1.7088890075683594e-05,
      "model_forward_time": 5.616587162017822,
      "step": 17919
    },
    {
      "epoch": 1.7088890075683594e-05,
      "step": 17919,
      "training_step_time": 17.128722429275513
    },
    {
      "epoch": 1.708984375e-05,
      "grad_norm": 0.9101788401603699,
      "learning_rate": 1e-05,
      "loss": 0.0721,
      "step": 17920
    },
    {
      "epoch": 1.708984375e-05,
      "model_forward_time": 5.625575304031372,
      "step": 17920
    },
    {
      "dynamics_loss_avg": 0.06780891343951226,
      "epoch": 1.708984375e-05,
      "step": 17920
    },
    {
      "action_loss_avg": 0.010040980135090648,
      "epoch": 1.708984375e-05,
      "step": 17920
    },
    {
      "epoch": 1.708984375e-05,
      "step": 17920,
      "training_step_time": 17.106666564941406
    },
    {
      "epoch": 1.7090797424316406e-05,
      "model_forward_time": 5.615647077560425,
      "step": 17921
    },
    {
      "epoch": 1.7090797424316406e-05,
      "step": 17921,
      "training_step_time": 17.113961696624756
    },
    {
      "epoch": 1.7091751098632813e-05,
      "model_forward_time": 5.61176323890686,
      "step": 17922
    },
    {
      "epoch": 1.7091751098632813e-05,
      "step": 17922,
      "training_step_time": 17.102638959884644
    },
    {
      "epoch": 1.7092704772949217e-05,
      "model_forward_time": 6.014086723327637,
      "step": 17923
    },
    {
      "epoch": 1.7092704772949217e-05,
      "step": 17923,
      "training_step_time": 17.52533984184265
    },
    {
      "epoch": 1.7093658447265625e-05,
      "model_forward_time": 5.6208109855651855,
      "step": 17924
    },
    {
      "epoch": 1.7093658447265625e-05,
      "step": 17924,
      "training_step_time": 17.12367820739746
    },
    {
      "epoch": 1.7094612121582032e-05,
      "model_forward_time": 5.613328456878662,
      "step": 17925
    },
    {
      "epoch": 1.7094612121582032e-05,
      "step": 17925,
      "training_step_time": 17.10800313949585
    },
    {
      "epoch": 1.7095565795898436e-05,
      "model_forward_time": 5.625321388244629,
      "step": 17926
    },
    {
      "epoch": 1.7095565795898436e-05,
      "step": 17926,
      "training_step_time": 17.11086106300354
    },
    {
      "epoch": 1.7096519470214843e-05,
      "model_forward_time": 5.605968952178955,
      "step": 17927
    },
    {
      "epoch": 1.7096519470214843e-05,
      "step": 17927,
      "training_step_time": 17.091614246368408
    },
    {
      "epoch": 1.709747314453125e-05,
      "model_forward_time": 5.604615926742554,
      "step": 17928
    },
    {
      "epoch": 1.709747314453125e-05,
      "step": 17928,
      "training_step_time": 17.08716130256653
    },
    {
      "epoch": 1.7098426818847655e-05,
      "model_forward_time": 5.949259996414185,
      "step": 17929
    },
    {
      "epoch": 1.7098426818847655e-05,
      "step": 17929,
      "training_step_time": 17.429176330566406
    },
    {
      "epoch": 1.7099380493164062e-05,
      "grad_norm": 0.8542922139167786,
      "learning_rate": 1e-05,
      "loss": 0.046,
      "step": 17930
    },
    {
      "epoch": 1.7099380493164062e-05,
      "model_forward_time": 5.589804649353027,
      "step": 17930
    },
    {
      "dynamics_loss_avg": 0.040445287339389326,
      "epoch": 1.7099380493164062e-05,
      "step": 17930
    },
    {
      "action_loss_avg": 0.004879426653496921,
      "epoch": 1.7099380493164062e-05,
      "step": 17930
    },
    {
      "epoch": 1.7099380493164062e-05,
      "step": 17930,
      "training_step_time": 17.03302264213562
    },
    {
      "epoch": 1.710033416748047e-05,
      "model_forward_time": 5.595245838165283,
      "step": 17931
    },
    {
      "epoch": 1.710033416748047e-05,
      "step": 17931,
      "training_step_time": 17.0467586517334
    },
    {
      "epoch": 1.7101287841796874e-05,
      "model_forward_time": 5.591883182525635,
      "step": 17932
    },
    {
      "epoch": 1.7101287841796874e-05,
      "step": 17932,
      "training_step_time": 17.053930044174194
    },
    {
      "epoch": 1.710224151611328e-05,
      "model_forward_time": 5.607046127319336,
      "step": 17933
    },
    {
      "epoch": 1.710224151611328e-05,
      "step": 17933,
      "training_step_time": 17.0936279296875
    },
    {
      "epoch": 1.710319519042969e-05,
      "model_forward_time": 5.984631061553955,
      "step": 17934
    },
    {
      "epoch": 1.710319519042969e-05,
      "step": 17934,
      "training_step_time": 17.467732429504395
    },
    {
      "epoch": 1.7104148864746093e-05,
      "model_forward_time": 5.616274356842041,
      "step": 17935
    },
    {
      "epoch": 1.7104148864746093e-05,
      "step": 17935,
      "training_step_time": 17.093936443328857
    },
    {
      "epoch": 1.71051025390625e-05,
      "model_forward_time": 5.607192039489746,
      "step": 17936
    },
    {
      "epoch": 1.71051025390625e-05,
      "step": 17936,
      "training_step_time": 17.031288146972656
    },
    {
      "epoch": 1.7106056213378907e-05,
      "model_forward_time": 5.58169412612915,
      "step": 17937
    },
    {
      "epoch": 1.7106056213378907e-05,
      "step": 17937,
      "training_step_time": 17.02403211593628
    },
    {
      "epoch": 1.710700988769531e-05,
      "model_forward_time": 5.586551904678345,
      "step": 17938
    },
    {
      "epoch": 1.710700988769531e-05,
      "step": 17938,
      "training_step_time": 17.02723217010498
    },
    {
      "epoch": 1.710796356201172e-05,
      "model_forward_time": 5.598127126693726,
      "step": 17939
    },
    {
      "epoch": 1.710796356201172e-05,
      "step": 17939,
      "training_step_time": 17.054648637771606
    },
    {
      "epoch": 1.7108917236328126e-05,
      "grad_norm": 0.936244010925293,
      "learning_rate": 1e-05,
      "loss": 0.0576,
      "step": 17940
    },
    {
      "epoch": 1.7108917236328126e-05,
      "model_forward_time": 5.95584774017334,
      "step": 17940
    },
    {
      "dynamics_loss_avg": 0.04742201743647456,
      "epoch": 1.7108917236328126e-05,
      "step": 17940
    },
    {
      "action_loss_avg": 0.006857285808655434,
      "epoch": 1.7108917236328126e-05,
      "step": 17940
    },
    {
      "epoch": 1.7108917236328126e-05,
      "step": 17940,
      "training_step_time": 17.44790244102478
    },
    {
      "epoch": 1.710987091064453e-05,
      "model_forward_time": 5.625994443893433,
      "step": 17941
    },
    {
      "epoch": 1.710987091064453e-05,
      "step": 17941,
      "training_step_time": 17.13021469116211
    },
    {
      "epoch": 1.7110824584960938e-05,
      "model_forward_time": 5.618737697601318,
      "step": 17942
    },
    {
      "epoch": 1.7110824584960938e-05,
      "step": 17942,
      "training_step_time": 17.130717277526855
    },
    {
      "epoch": 1.7111778259277345e-05,
      "model_forward_time": 5.621910095214844,
      "step": 17943
    },
    {
      "epoch": 1.7111778259277345e-05,
      "step": 17943,
      "training_step_time": 17.096670150756836
    },
    {
      "epoch": 1.711273193359375e-05,
      "model_forward_time": 5.594975709915161,
      "step": 17944
    },
    {
      "epoch": 1.711273193359375e-05,
      "step": 17944,
      "training_step_time": 17.046732187271118
    },
    {
      "epoch": 1.7113685607910157e-05,
      "model_forward_time": 5.588693380355835,
      "step": 17945
    },
    {
      "epoch": 1.7113685607910157e-05,
      "step": 17945,
      "training_step_time": 17.43863344192505
    },
    {
      "epoch": 1.7114639282226564e-05,
      "model_forward_time": 5.6007843017578125,
      "step": 17946
    },
    {
      "epoch": 1.7114639282226564e-05,
      "step": 17946,
      "training_step_time": 17.106072664260864
    },
    {
      "epoch": 1.7115592956542968e-05,
      "model_forward_time": 5.629377365112305,
      "step": 17947
    },
    {
      "epoch": 1.7115592956542968e-05,
      "step": 17947,
      "training_step_time": 17.143617391586304
    },
    {
      "epoch": 1.7116546630859375e-05,
      "model_forward_time": 5.626664638519287,
      "step": 17948
    },
    {
      "epoch": 1.7116546630859375e-05,
      "step": 17948,
      "training_step_time": 17.11666512489319
    },
    {
      "epoch": 1.7117500305175783e-05,
      "model_forward_time": 5.622073650360107,
      "step": 17949
    },
    {
      "epoch": 1.7117500305175783e-05,
      "step": 17949,
      "training_step_time": 17.107739686965942
    },
    {
      "epoch": 1.7118453979492187e-05,
      "grad_norm": 0.5891167521476746,
      "learning_rate": 1e-05,
      "loss": 0.0489,
      "step": 17950
    },
    {
      "epoch": 1.7118453979492187e-05,
      "model_forward_time": 5.607934951782227,
      "step": 17950
    },
    {
      "dynamics_loss_avg": 0.047899135015904905,
      "epoch": 1.7118453979492187e-05,
      "step": 17950
    },
    {
      "action_loss_avg": 0.006000267318449914,
      "epoch": 1.7118453979492187e-05,
      "step": 17950
    },
    {
      "epoch": 1.7118453979492187e-05,
      "step": 17950,
      "training_step_time": 17.090656757354736
    },
    {
      "epoch": 1.7119407653808594e-05,
      "model_forward_time": 5.953498125076294,
      "step": 17951
    },
    {
      "epoch": 1.7119407653808594e-05,
      "step": 17951,
      "training_step_time": 17.430532455444336
    },
    {
      "epoch": 1.7120361328125e-05,
      "model_forward_time": 5.608415842056274,
      "step": 17952
    },
    {
      "epoch": 1.7120361328125e-05,
      "step": 17952,
      "training_step_time": 17.097800970077515
    },
    {
      "epoch": 1.7121315002441406e-05,
      "model_forward_time": 5.6183483600616455,
      "step": 17953
    },
    {
      "epoch": 1.7121315002441406e-05,
      "step": 17953,
      "training_step_time": 17.125271558761597
    },
    {
      "epoch": 1.7122268676757813e-05,
      "model_forward_time": 5.609561443328857,
      "step": 17954
    },
    {
      "epoch": 1.7122268676757813e-05,
      "step": 17954,
      "training_step_time": 17.090203285217285
    },
    {
      "epoch": 1.7123222351074217e-05,
      "model_forward_time": 5.5922136306762695,
      "step": 17955
    },
    {
      "epoch": 1.7123222351074217e-05,
      "step": 17955,
      "training_step_time": 17.037109375
    },
    {
      "epoch": 1.7124176025390625e-05,
      "model_forward_time": 5.96035361289978,
      "step": 17956
    },
    {
      "epoch": 1.7124176025390625e-05,
      "step": 17956,
      "training_step_time": 17.37435483932495
    },
    {
      "epoch": 1.7125129699707032e-05,
      "model_forward_time": 5.581778049468994,
      "step": 17957
    },
    {
      "epoch": 1.7125129699707032e-05,
      "step": 17957,
      "training_step_time": 16.997504472732544
    },
    {
      "epoch": 1.7126083374023436e-05,
      "model_forward_time": 5.573704957962036,
      "step": 17958
    },
    {
      "epoch": 1.7126083374023436e-05,
      "step": 17958,
      "training_step_time": 16.975775957107544
    },
    {
      "epoch": 1.7127037048339843e-05,
      "model_forward_time": 5.579972505569458,
      "step": 17959
    },
    {
      "epoch": 1.7127037048339843e-05,
      "step": 17959,
      "training_step_time": 17.0196373462677
    },
    {
      "epoch": 1.712799072265625e-05,
      "grad_norm": 0.6081519722938538,
      "learning_rate": 1e-05,
      "loss": 0.0791,
      "step": 17960
    },
    {
      "epoch": 1.712799072265625e-05,
      "model_forward_time": 5.582143068313599,
      "step": 17960
    },
    {
      "dynamics_loss_avg": 0.06729152649641038,
      "epoch": 1.712799072265625e-05,
      "step": 17960
    },
    {
      "action_loss_avg": 0.00877434597350657,
      "epoch": 1.712799072265625e-05,
      "step": 17960
    },
    {
      "epoch": 1.712799072265625e-05,
      "step": 17960,
      "training_step_time": 17.02111506462097
    },
    {
      "epoch": 1.7128944396972655e-05,
      "model_forward_time": 5.589926242828369,
      "step": 17961
    },
    {
      "epoch": 1.7128944396972655e-05,
      "step": 17961,
      "training_step_time": 17.02621364593506
    },
    {
      "epoch": 1.7129898071289062e-05,
      "model_forward_time": 5.935143947601318,
      "step": 17962
    },
    {
      "epoch": 1.7129898071289062e-05,
      "step": 17962,
      "training_step_time": 17.403422355651855
    },
    {
      "epoch": 1.713085174560547e-05,
      "model_forward_time": 5.614632606506348,
      "step": 17963
    },
    {
      "epoch": 1.713085174560547e-05,
      "step": 17963,
      "training_step_time": 17.14507293701172
    },
    {
      "epoch": 1.7131805419921874e-05,
      "model_forward_time": 5.624427080154419,
      "step": 17964
    },
    {
      "epoch": 1.7131805419921874e-05,
      "step": 17964,
      "training_step_time": 17.14908266067505
    },
    {
      "epoch": 1.713275909423828e-05,
      "model_forward_time": 5.617657899856567,
      "step": 17965
    },
    {
      "epoch": 1.713275909423828e-05,
      "step": 17965,
      "training_step_time": 17.109494924545288
    },
    {
      "epoch": 1.713371276855469e-05,
      "model_forward_time": 5.61442494392395,
      "step": 17966
    },
    {
      "epoch": 1.713371276855469e-05,
      "step": 17966,
      "training_step_time": 17.092145204544067
    },
    {
      "epoch": 1.7134666442871093e-05,
      "model_forward_time": 5.992890119552612,
      "step": 17967
    },
    {
      "epoch": 1.7134666442871093e-05,
      "step": 17967,
      "training_step_time": 17.46202325820923
    },
    {
      "epoch": 1.71356201171875e-05,
      "model_forward_time": 5.60156774520874,
      "step": 17968
    },
    {
      "epoch": 1.71356201171875e-05,
      "step": 17968,
      "training_step_time": 17.052849292755127
    },
    {
      "epoch": 1.7136573791503907e-05,
      "model_forward_time": 5.594149827957153,
      "step": 17969
    },
    {
      "epoch": 1.7136573791503907e-05,
      "step": 17969,
      "training_step_time": 17.045215606689453
    },
    {
      "epoch": 1.713752746582031e-05,
      "grad_norm": 0.8019595146179199,
      "learning_rate": 1e-05,
      "loss": 0.0658,
      "step": 17970
    },
    {
      "epoch": 1.713752746582031e-05,
      "model_forward_time": 5.590525150299072,
      "step": 17970
    },
    {
      "dynamics_loss_avg": 0.057826668582856657,
      "epoch": 1.713752746582031e-05,
      "step": 17970
    },
    {
      "action_loss_avg": 0.007731660921126604,
      "epoch": 1.713752746582031e-05,
      "step": 17970
    },
    {
      "epoch": 1.713752746582031e-05,
      "step": 17970,
      "training_step_time": 17.038220167160034
    },
    {
      "epoch": 1.713848114013672e-05,
      "model_forward_time": 5.583976984024048,
      "step": 17971
    },
    {
      "epoch": 1.713848114013672e-05,
      "step": 17971,
      "training_step_time": 17.032508373260498
    },
    {
      "epoch": 1.7139434814453126e-05,
      "model_forward_time": 5.599630355834961,
      "step": 17972
    },
    {
      "epoch": 1.7139434814453126e-05,
      "step": 17972,
      "training_step_time": 17.095417737960815
    },
    {
      "epoch": 1.714038848876953e-05,
      "model_forward_time": 5.967190980911255,
      "step": 17973
    },
    {
      "epoch": 1.714038848876953e-05,
      "step": 17973,
      "training_step_time": 17.444647789001465
    },
    {
      "epoch": 1.7141342163085938e-05,
      "model_forward_time": 5.592903137207031,
      "step": 17974
    },
    {
      "epoch": 1.7141342163085938e-05,
      "step": 17974,
      "training_step_time": 17.02403712272644
    },
    {
      "epoch": 1.7142295837402345e-05,
      "model_forward_time": 5.577894449234009,
      "step": 17975
    },
    {
      "epoch": 1.7142295837402345e-05,
      "step": 17975,
      "training_step_time": 17.015268087387085
    },
    {
      "epoch": 1.714324951171875e-05,
      "model_forward_time": 5.586324691772461,
      "step": 17976
    },
    {
      "epoch": 1.714324951171875e-05,
      "step": 17976,
      "training_step_time": 16.995145082473755
    },
    {
      "epoch": 1.7144203186035157e-05,
      "model_forward_time": 5.576923370361328,
      "step": 17977
    },
    {
      "epoch": 1.7144203186035157e-05,
      "step": 17977,
      "training_step_time": 16.985007286071777
    },
    {
      "epoch": 1.7145156860351564e-05,
      "model_forward_time": 5.581644535064697,
      "step": 17978
    },
    {
      "epoch": 1.7145156860351564e-05,
      "step": 17978,
      "training_step_time": 17.412561893463135
    },
    {
      "epoch": 1.7146110534667968e-05,
      "model_forward_time": 5.591425657272339,
      "step": 17979
    },
    {
      "epoch": 1.7146110534667968e-05,
      "step": 17979,
      "training_step_time": 17.036415338516235
    },
    {
      "epoch": 1.7147064208984375e-05,
      "grad_norm": 1.2133649587631226,
      "learning_rate": 1e-05,
      "loss": 0.0644,
      "step": 17980
    },
    {
      "epoch": 1.7147064208984375e-05,
      "model_forward_time": 5.586061716079712,
      "step": 17980
    },
    {
      "dynamics_loss_avg": 0.056734966672956945,
      "epoch": 1.7147064208984375e-05,
      "step": 17980
    },
    {
      "action_loss_avg": 0.0057507965480908755,
      "epoch": 1.7147064208984375e-05,
      "step": 17980
    },
    {
      "epoch": 1.7147064208984375e-05,
      "step": 17980,
      "training_step_time": 17.060057163238525
    },
    {
      "epoch": 1.7148017883300783e-05,
      "model_forward_time": 5.606934547424316,
      "step": 17981
    },
    {
      "epoch": 1.7148017883300783e-05,
      "step": 17981,
      "training_step_time": 17.098186254501343
    },
    {
      "epoch": 1.7148971557617187e-05,
      "model_forward_time": 5.6102001667022705,
      "step": 17982
    },
    {
      "epoch": 1.7148971557617187e-05,
      "step": 17982,
      "training_step_time": 17.084506511688232
    },
    {
      "epoch": 1.7149925231933594e-05,
      "model_forward_time": 5.616316080093384,
      "step": 17983
    },
    {
      "epoch": 1.7149925231933594e-05,
      "step": 17983,
      "training_step_time": 17.082334518432617
    },
    {
      "epoch": 1.715087890625e-05,
      "model_forward_time": 5.961693525314331,
      "step": 17984
    },
    {
      "epoch": 1.715087890625e-05,
      "step": 17984,
      "training_step_time": 17.426430463790894
    },
    {
      "epoch": 1.7151832580566406e-05,
      "model_forward_time": 5.618447542190552,
      "step": 17985
    },
    {
      "epoch": 1.7151832580566406e-05,
      "step": 17985,
      "training_step_time": 17.073880672454834
    },
    {
      "epoch": 1.7152786254882813e-05,
      "model_forward_time": 5.5911781787872314,
      "step": 17986
    },
    {
      "epoch": 1.7152786254882813e-05,
      "step": 17986,
      "training_step_time": 17.02998638153076
    },
    {
      "epoch": 1.7153739929199217e-05,
      "model_forward_time": 5.590656042098999,
      "step": 17987
    },
    {
      "epoch": 1.7153739929199217e-05,
      "step": 17987,
      "training_step_time": 17.038317918777466
    },
    {
      "epoch": 1.7154693603515625e-05,
      "model_forward_time": 5.595017194747925,
      "step": 17988
    },
    {
      "epoch": 1.7154693603515625e-05,
      "step": 17988,
      "training_step_time": 17.043013095855713
    },
    {
      "epoch": 1.7155647277832032e-05,
      "model_forward_time": 5.98540997505188,
      "step": 17989
    },
    {
      "epoch": 1.7155647277832032e-05,
      "step": 17989,
      "training_step_time": 17.438543796539307
    },
    {
      "epoch": 1.7156600952148436e-05,
      "grad_norm": 0.4996352195739746,
      "learning_rate": 1e-05,
      "loss": 0.0553,
      "step": 17990
    },
    {
      "epoch": 1.7156600952148436e-05,
      "model_forward_time": 5.616842031478882,
      "step": 17990
    },
    {
      "dynamics_loss_avg": 0.04872352834790945,
      "epoch": 1.7156600952148436e-05,
      "step": 17990
    },
    {
      "action_loss_avg": 0.004779899574350566,
      "epoch": 1.7156600952148436e-05,
      "step": 17990
    },
    {
      "epoch": 1.7156600952148436e-05,
      "step": 17990,
      "training_step_time": 17.07113003730774
    },
    {
      "epoch": 1.7157554626464843e-05,
      "model_forward_time": 5.604950428009033,
      "step": 17991
    },
    {
      "epoch": 1.7157554626464843e-05,
      "step": 17991,
      "training_step_time": 17.074645042419434
    },
    {
      "epoch": 1.715850830078125e-05,
      "model_forward_time": 5.604825258255005,
      "step": 17992
    },
    {
      "epoch": 1.715850830078125e-05,
      "step": 17992,
      "training_step_time": 17.083720922470093
    },
    {
      "epoch": 1.7159461975097655e-05,
      "model_forward_time": 5.61118745803833,
      "step": 17993
    },
    {
      "epoch": 1.7159461975097655e-05,
      "step": 17993,
      "training_step_time": 17.065009593963623
    },
    {
      "epoch": 1.7160415649414062e-05,
      "model_forward_time": 5.5903050899505615,
      "step": 17994
    },
    {
      "epoch": 1.7160415649414062e-05,
      "step": 17994,
      "training_step_time": 17.04567050933838
    },
    {
      "epoch": 1.716136932373047e-05,
      "model_forward_time": 5.950076103210449,
      "step": 17995
    },
    {
      "epoch": 1.716136932373047e-05,
      "step": 17995,
      "training_step_time": 17.420401334762573
    },
    {
      "epoch": 1.7162322998046874e-05,
      "model_forward_time": 5.606775999069214,
      "step": 17996
    },
    {
      "epoch": 1.7162322998046874e-05,
      "step": 17996,
      "training_step_time": 17.080856561660767
    },
    {
      "epoch": 1.716327667236328e-05,
      "model_forward_time": 5.599074602127075,
      "step": 17997
    },
    {
      "epoch": 1.716327667236328e-05,
      "step": 17997,
      "training_step_time": 17.031877517700195
    },
    {
      "epoch": 1.716423034667969e-05,
      "model_forward_time": 5.590655088424683,
      "step": 17998
    },
    {
      "epoch": 1.716423034667969e-05,
      "step": 17998,
      "training_step_time": 17.026657342910767
    },
    {
      "epoch": 1.7165184020996093e-05,
      "model_forward_time": 5.584717512130737,
      "step": 17999
    },
    {
      "epoch": 1.7165184020996093e-05,
      "step": 17999,
      "training_step_time": 17.025327920913696
    },
    {
      "epoch": 1.71661376953125e-05,
      "grad_norm": 0.6964331269264221,
      "learning_rate": 1e-05,
      "loss": 0.0563,
      "step": 18000
    },
    {
      "epoch": 1.71661376953125e-05,
      "model_forward_time": 5.685965538024902,
      "step": 18000
    },
    {
      "dynamics_loss_avg": 0.05960814654827118,
      "epoch": 1.71661376953125e-05,
      "step": 18000
    },
    {
      "action_loss_avg": 0.00939991557970643,
      "epoch": 1.71661376953125e-05,
      "step": 18000
    },
    {
      "epoch": 1.71661376953125e-05,
      "step": 18000,
      "training_step_time": 17.569825887680054
    },
    {
      "epoch": 1.7167091369628907e-05,
      "model_forward_time": 5.701998949050903,
      "step": 18001
    },
    {
      "epoch": 1.7167091369628907e-05,
      "step": 18001,
      "training_step_time": 17.235087156295776
    },
    {
      "epoch": 1.716804504394531e-05,
      "model_forward_time": 5.667503118515015,
      "step": 18002
    },
    {
      "epoch": 1.716804504394531e-05,
      "step": 18002,
      "training_step_time": 17.174914836883545
    },
    {
      "epoch": 1.716899871826172e-05,
      "model_forward_time": 5.615735054016113,
      "step": 18003
    },
    {
      "epoch": 1.716899871826172e-05,
      "step": 18003,
      "training_step_time": 17.05722975730896
    },
    {
      "epoch": 1.7169952392578126e-05,
      "model_forward_time": 5.584823131561279,
      "step": 18004
    },
    {
      "epoch": 1.7169952392578126e-05,
      "step": 18004,
      "training_step_time": 17.01594829559326
    },
    {
      "epoch": 1.717090606689453e-05,
      "model_forward_time": 5.57302451133728,
      "step": 18005
    },
    {
      "epoch": 1.717090606689453e-05,
      "step": 18005,
      "training_step_time": 17.011221647262573
    },
    {
      "epoch": 1.7171859741210938e-05,
      "model_forward_time": 5.952520847320557,
      "step": 18006
    },
    {
      "epoch": 1.7171859741210938e-05,
      "step": 18006,
      "training_step_time": 17.3587863445282
    },
    {
      "epoch": 1.7172813415527345e-05,
      "model_forward_time": 5.5783162117004395,
      "step": 18007
    },
    {
      "epoch": 1.7172813415527345e-05,
      "step": 18007,
      "training_step_time": 16.996817588806152
    },
    {
      "epoch": 1.717376708984375e-05,
      "model_forward_time": 5.579243183135986,
      "step": 18008
    },
    {
      "epoch": 1.717376708984375e-05,
      "step": 18008,
      "training_step_time": 17.02387261390686
    },
    {
      "epoch": 1.7174720764160157e-05,
      "model_forward_time": 5.5832273960113525,
      "step": 18009
    },
    {
      "epoch": 1.7174720764160157e-05,
      "step": 18009,
      "training_step_time": 16.999234914779663
    },
    {
      "epoch": 1.7175674438476564e-05,
      "grad_norm": 0.7364857792854309,
      "learning_rate": 1e-05,
      "loss": 0.0718,
      "step": 18010
    },
    {
      "epoch": 1.7175674438476564e-05,
      "model_forward_time": 5.583737134933472,
      "step": 18010
    },
    {
      "dynamics_loss_avg": 0.05156713109463453,
      "epoch": 1.7175674438476564e-05,
      "step": 18010
    },
    {
      "action_loss_avg": 0.010465849144384264,
      "epoch": 1.7175674438476564e-05,
      "step": 18010
    },
    {
      "epoch": 1.7175674438476564e-05,
      "step": 18010,
      "training_step_time": 17.020297050476074
    },
    {
      "epoch": 1.7176628112792968e-05,
      "model_forward_time": 5.945762872695923,
      "step": 18011
    },
    {
      "epoch": 1.7176628112792968e-05,
      "step": 18011,
      "training_step_time": 17.422419548034668
    },
    {
      "epoch": 1.7177581787109375e-05,
      "model_forward_time": 5.622429370880127,
      "step": 18012
    },
    {
      "epoch": 1.7177581787109375e-05,
      "step": 18012,
      "training_step_time": 17.13849711418152
    },
    {
      "epoch": 1.7178535461425783e-05,
      "model_forward_time": 5.6294028759002686,
      "step": 18013
    },
    {
      "epoch": 1.7178535461425783e-05,
      "step": 18013,
      "training_step_time": 17.13579559326172
    },
    {
      "epoch": 1.7179489135742187e-05,
      "model_forward_time": 5.611940145492554,
      "step": 18014
    },
    {
      "epoch": 1.7179489135742187e-05,
      "step": 18014,
      "training_step_time": 17.10003685951233
    },
    {
      "epoch": 1.7180442810058594e-05,
      "model_forward_time": 5.609470844268799,
      "step": 18015
    },
    {
      "epoch": 1.7180442810058594e-05,
      "step": 18015,
      "training_step_time": 17.081268548965454
    },
    {
      "epoch": 1.7181396484375e-05,
      "model_forward_time": 5.592951536178589,
      "step": 18016
    },
    {
      "epoch": 1.7181396484375e-05,
      "step": 18016,
      "training_step_time": 17.02577829360962
    },
    {
      "epoch": 1.7182350158691406e-05,
      "model_forward_time": 5.946047067642212,
      "step": 18017
    },
    {
      "epoch": 1.7182350158691406e-05,
      "step": 18017,
      "training_step_time": 17.373234748840332
    },
    {
      "epoch": 1.7183303833007813e-05,
      "model_forward_time": 5.620862245559692,
      "step": 18018
    },
    {
      "epoch": 1.7183303833007813e-05,
      "step": 18018,
      "training_step_time": 17.063430786132812
    },
    {
      "epoch": 1.7184257507324217e-05,
      "model_forward_time": 5.598874092102051,
      "step": 18019
    },
    {
      "epoch": 1.7184257507324217e-05,
      "step": 18019,
      "training_step_time": 17.07066035270691
    },
    {
      "epoch": 1.7185211181640625e-05,
      "grad_norm": 0.683759331703186,
      "learning_rate": 1e-05,
      "loss": 0.0555,
      "step": 18020
    },
    {
      "epoch": 1.7185211181640625e-05,
      "model_forward_time": 5.605032920837402,
      "step": 18020
    },
    {
      "dynamics_loss_avg": 0.046002712659537794,
      "epoch": 1.7185211181640625e-05,
      "step": 18020
    },
    {
      "action_loss_avg": 0.008641445823013783,
      "epoch": 1.7185211181640625e-05,
      "step": 18020
    },
    {
      "epoch": 1.7185211181640625e-05,
      "step": 18020,
      "training_step_time": 17.08479642868042
    },
    {
      "epoch": 1.7186164855957032e-05,
      "model_forward_time": 5.604803562164307,
      "step": 18021
    },
    {
      "epoch": 1.7186164855957032e-05,
      "step": 18021,
      "training_step_time": 17.090861797332764
    },
    {
      "epoch": 1.7187118530273436e-05,
      "model_forward_time": 5.950350284576416,
      "step": 18022
    },
    {
      "epoch": 1.7187118530273436e-05,
      "step": 18022,
      "training_step_time": 17.445775985717773
    },
    {
      "epoch": 1.7188072204589843e-05,
      "model_forward_time": 5.615810871124268,
      "step": 18023
    },
    {
      "epoch": 1.7188072204589843e-05,
      "step": 18023,
      "training_step_time": 17.103935480117798
    },
    {
      "epoch": 1.718902587890625e-05,
      "model_forward_time": 5.617746353149414,
      "step": 18024
    },
    {
      "epoch": 1.718902587890625e-05,
      "step": 18024,
      "training_step_time": 17.095486402511597
    },
    {
      "epoch": 1.7189979553222655e-05,
      "model_forward_time": 5.6066014766693115,
      "step": 18025
    },
    {
      "epoch": 1.7189979553222655e-05,
      "step": 18025,
      "training_step_time": 17.09088444709778
    },
    {
      "epoch": 1.7190933227539062e-05,
      "model_forward_time": 5.613500595092773,
      "step": 18026
    },
    {
      "epoch": 1.7190933227539062e-05,
      "step": 18026,
      "training_step_time": 17.071726322174072
    },
    {
      "epoch": 1.719188690185547e-05,
      "model_forward_time": 5.5954270362854,
      "step": 18027
    },
    {
      "epoch": 1.719188690185547e-05,
      "step": 18027,
      "training_step_time": 17.400169134140015
    },
    {
      "epoch": 1.7192840576171874e-05,
      "model_forward_time": 5.581521987915039,
      "step": 18028
    },
    {
      "epoch": 1.7192840576171874e-05,
      "step": 18028,
      "training_step_time": 17.007628202438354
    },
    {
      "epoch": 1.719379425048828e-05,
      "model_forward_time": 5.595709800720215,
      "step": 18029
    },
    {
      "epoch": 1.719379425048828e-05,
      "step": 18029,
      "training_step_time": 17.0582115650177
    },
    {
      "epoch": 1.719474792480469e-05,
      "grad_norm": 0.28868383169174194,
      "learning_rate": 1e-05,
      "loss": 0.0646,
      "step": 18030
    },
    {
      "epoch": 1.719474792480469e-05,
      "model_forward_time": 5.5927734375,
      "step": 18030
    },
    {
      "dynamics_loss_avg": 0.057124117575585845,
      "epoch": 1.719474792480469e-05,
      "step": 18030
    },
    {
      "action_loss_avg": 0.009558760561048985,
      "epoch": 1.719474792480469e-05,
      "step": 18030
    },
    {
      "epoch": 1.719474792480469e-05,
      "step": 18030,
      "training_step_time": 17.073696851730347
    },
    {
      "epoch": 1.7195701599121093e-05,
      "model_forward_time": 5.6104419231414795,
      "step": 18031
    },
    {
      "epoch": 1.7195701599121093e-05,
      "step": 18031,
      "training_step_time": 17.057223558425903
    },
    {
      "epoch": 1.71966552734375e-05,
      "model_forward_time": 5.591727256774902,
      "step": 18032
    },
    {
      "epoch": 1.71966552734375e-05,
      "step": 18032,
      "training_step_time": 17.021591663360596
    },
    {
      "epoch": 1.7197608947753907e-05,
      "model_forward_time": 5.9106433391571045,
      "step": 18033
    },
    {
      "epoch": 1.7197608947753907e-05,
      "step": 18033,
      "training_step_time": 17.320517539978027
    },
    {
      "epoch": 1.719856262207031e-05,
      "model_forward_time": 5.58050012588501,
      "step": 18034
    },
    {
      "epoch": 1.719856262207031e-05,
      "step": 18034,
      "training_step_time": 16.984889030456543
    },
    {
      "epoch": 1.719951629638672e-05,
      "model_forward_time": 5.571204662322998,
      "step": 18035
    },
    {
      "epoch": 1.719951629638672e-05,
      "step": 18035,
      "training_step_time": 17.005884408950806
    },
    {
      "epoch": 1.7200469970703126e-05,
      "model_forward_time": 5.5815324783325195,
      "step": 18036
    },
    {
      "epoch": 1.7200469970703126e-05,
      "step": 18036,
      "training_step_time": 16.997631311416626
    },
    {
      "epoch": 1.720142364501953e-05,
      "model_forward_time": 5.587950944900513,
      "step": 18037
    },
    {
      "epoch": 1.720142364501953e-05,
      "step": 18037,
      "training_step_time": 17.00606632232666
    },
    {
      "epoch": 1.7202377319335938e-05,
      "model_forward_time": 5.592547416687012,
      "step": 18038
    },
    {
      "epoch": 1.7202377319335938e-05,
      "step": 18038,
      "training_step_time": 17.416950225830078
    },
    {
      "epoch": 1.7203330993652345e-05,
      "model_forward_time": 5.619590520858765,
      "step": 18039
    },
    {
      "epoch": 1.7203330993652345e-05,
      "step": 18039,
      "training_step_time": 17.144761562347412
    },
    {
      "epoch": 1.720428466796875e-05,
      "grad_norm": 0.5367464423179626,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 18040
    },
    {
      "epoch": 1.720428466796875e-05,
      "model_forward_time": 5.629188299179077,
      "step": 18040
    },
    {
      "dynamics_loss_avg": 0.055952754244208334,
      "epoch": 1.720428466796875e-05,
      "step": 18040
    },
    {
      "action_loss_avg": 0.008682744018733502,
      "epoch": 1.720428466796875e-05,
      "step": 18040
    },
    {
      "epoch": 1.720428466796875e-05,
      "step": 18040,
      "training_step_time": 17.141397714614868
    },
    {
      "epoch": 1.7205238342285157e-05,
      "model_forward_time": 5.625136613845825,
      "step": 18041
    },
    {
      "epoch": 1.7205238342285157e-05,
      "step": 18041,
      "training_step_time": 17.122173309326172
    },
    {
      "epoch": 1.7206192016601564e-05,
      "model_forward_time": 5.6037304401397705,
      "step": 18042
    },
    {
      "epoch": 1.7206192016601564e-05,
      "step": 18042,
      "training_step_time": 17.075466871261597
    },
    {
      "epoch": 1.7207145690917968e-05,
      "model_forward_time": 5.609127998352051,
      "step": 18043
    },
    {
      "epoch": 1.7207145690917968e-05,
      "step": 18043,
      "training_step_time": 17.089682579040527
    },
    {
      "epoch": 1.7208099365234375e-05,
      "model_forward_time": 5.959068536758423,
      "step": 18044
    },
    {
      "epoch": 1.7208099365234375e-05,
      "step": 18044,
      "training_step_time": 17.425552129745483
    },
    {
      "epoch": 1.7209053039550783e-05,
      "model_forward_time": 5.605702638626099,
      "step": 18045
    },
    {
      "epoch": 1.7209053039550783e-05,
      "step": 18045,
      "training_step_time": 17.0532009601593
    },
    {
      "epoch": 1.7210006713867187e-05,
      "model_forward_time": 5.598486423492432,
      "step": 18046
    },
    {
      "epoch": 1.7210006713867187e-05,
      "step": 18046,
      "training_step_time": 17.027555227279663
    },
    {
      "epoch": 1.7210960388183594e-05,
      "model_forward_time": 5.579960584640503,
      "step": 18047
    },
    {
      "epoch": 1.7210960388183594e-05,
      "step": 18047,
      "training_step_time": 17.021915435791016
    },
    {
      "epoch": 1.72119140625e-05,
      "model_forward_time": 5.594403982162476,
      "step": 18048
    },
    {
      "epoch": 1.72119140625e-05,
      "step": 18048,
      "training_step_time": 17.077228546142578
    },
    {
      "epoch": 1.7212867736816406e-05,
      "model_forward_time": 6.001197099685669,
      "step": 18049
    },
    {
      "epoch": 1.7212867736816406e-05,
      "step": 18049,
      "training_step_time": 17.522189140319824
    },
    {
      "epoch": 1.7213821411132813e-05,
      "grad_norm": 0.8500454425811768,
      "learning_rate": 1e-05,
      "loss": 0.0513,
      "step": 18050
    },
    {
      "epoch": 1.7213821411132813e-05,
      "model_forward_time": 5.632970333099365,
      "step": 18050
    },
    {
      "dynamics_loss_avg": 0.04464910142123699,
      "epoch": 1.7213821411132813e-05,
      "step": 18050
    },
    {
      "action_loss_avg": 0.009386748936958612,
      "epoch": 1.7213821411132813e-05,
      "step": 18050
    },
    {
      "epoch": 1.7213821411132813e-05,
      "step": 18050,
      "training_step_time": 17.143351316452026
    },
    {
      "epoch": 1.7214775085449217e-05,
      "model_forward_time": 5.615966558456421,
      "step": 18051
    },
    {
      "epoch": 1.7214775085449217e-05,
      "step": 18051,
      "training_step_time": 17.107576370239258
    },
    {
      "epoch": 1.7215728759765625e-05,
      "model_forward_time": 5.613805532455444,
      "step": 18052
    },
    {
      "epoch": 1.7215728759765625e-05,
      "step": 18052,
      "training_step_time": 17.095070838928223
    },
    {
      "epoch": 1.7216682434082032e-05,
      "model_forward_time": 5.596737384796143,
      "step": 18053
    },
    {
      "epoch": 1.7216682434082032e-05,
      "step": 18053,
      "training_step_time": 17.051946878433228
    },
    {
      "epoch": 1.7217636108398436e-05,
      "model_forward_time": 5.582814693450928,
      "step": 18054
    },
    {
      "epoch": 1.7217636108398436e-05,
      "step": 18054,
      "training_step_time": 17.017388105392456
    },
    {
      "epoch": 1.7218589782714843e-05,
      "model_forward_time": 5.959207534790039,
      "step": 18055
    },
    {
      "epoch": 1.7218589782714843e-05,
      "step": 18055,
      "training_step_time": 17.36505436897278
    },
    {
      "epoch": 1.721954345703125e-05,
      "model_forward_time": 5.5953333377838135,
      "step": 18056
    },
    {
      "epoch": 1.721954345703125e-05,
      "step": 18056,
      "training_step_time": 17.06650471687317
    },
    {
      "epoch": 1.7220497131347655e-05,
      "model_forward_time": 5.595121383666992,
      "step": 18057
    },
    {
      "epoch": 1.7220497131347655e-05,
      "step": 18057,
      "training_step_time": 17.081506490707397
    },
    {
      "epoch": 1.7221450805664062e-05,
      "model_forward_time": 5.6069207191467285,
      "step": 18058
    },
    {
      "epoch": 1.7221450805664062e-05,
      "step": 18058,
      "training_step_time": 17.110204458236694
    },
    {
      "epoch": 1.722240447998047e-05,
      "model_forward_time": 5.612008094787598,
      "step": 18059
    },
    {
      "epoch": 1.722240447998047e-05,
      "step": 18059,
      "training_step_time": 17.105060338974
    },
    {
      "epoch": 1.7223358154296874e-05,
      "grad_norm": 1.107278823852539,
      "learning_rate": 1e-05,
      "loss": 0.0521,
      "step": 18060
    },
    {
      "epoch": 1.7223358154296874e-05,
      "model_forward_time": 5.985870361328125,
      "step": 18060
    },
    {
      "dynamics_loss_avg": 0.04150378508493304,
      "epoch": 1.7223358154296874e-05,
      "step": 18060
    },
    {
      "action_loss_avg": 0.008236527396366,
      "epoch": 1.7223358154296874e-05,
      "step": 18060
    },
    {
      "epoch": 1.7223358154296874e-05,
      "step": 18060,
      "training_step_time": 17.462137937545776
    },
    {
      "epoch": 1.722431182861328e-05,
      "model_forward_time": 5.626326322555542,
      "step": 18061
    },
    {
      "epoch": 1.722431182861328e-05,
      "step": 18061,
      "training_step_time": 17.107612133026123
    },
    {
      "epoch": 1.722526550292969e-05,
      "model_forward_time": 5.619807481765747,
      "step": 18062
    },
    {
      "epoch": 1.722526550292969e-05,
      "step": 18062,
      "training_step_time": 17.131025552749634
    },
    {
      "epoch": 1.7226219177246093e-05,
      "model_forward_time": 5.620055437088013,
      "step": 18063
    },
    {
      "epoch": 1.7226219177246093e-05,
      "step": 18063,
      "training_step_time": 17.124890089035034
    },
    {
      "epoch": 1.72271728515625e-05,
      "model_forward_time": 5.6282734870910645,
      "step": 18064
    },
    {
      "epoch": 1.72271728515625e-05,
      "step": 18064,
      "training_step_time": 17.15029740333557
    },
    {
      "epoch": 1.7228126525878907e-05,
      "model_forward_time": 5.618149757385254,
      "step": 18065
    },
    {
      "epoch": 1.7228126525878907e-05,
      "step": 18065,
      "training_step_time": 17.10848879814148
    },
    {
      "epoch": 1.722908020019531e-05,
      "model_forward_time": 6.10571026802063,
      "step": 18066
    },
    {
      "epoch": 1.722908020019531e-05,
      "step": 18066,
      "training_step_time": 17.628025770187378
    },
    {
      "epoch": 1.723003387451172e-05,
      "model_forward_time": 5.6160805225372314,
      "step": 18067
    },
    {
      "epoch": 1.723003387451172e-05,
      "step": 18067,
      "training_step_time": 17.049028396606445
    },
    {
      "epoch": 1.7230987548828126e-05,
      "model_forward_time": 5.604259490966797,
      "step": 18068
    },
    {
      "epoch": 1.7230987548828126e-05,
      "step": 18068,
      "training_step_time": 17.041166067123413
    },
    {
      "epoch": 1.723194122314453e-05,
      "model_forward_time": 5.594924688339233,
      "step": 18069
    },
    {
      "epoch": 1.723194122314453e-05,
      "step": 18069,
      "training_step_time": 17.043647050857544
    },
    {
      "epoch": 1.7232894897460938e-05,
      "grad_norm": 0.43033209443092346,
      "learning_rate": 1e-05,
      "loss": 0.0585,
      "step": 18070
    },
    {
      "epoch": 1.7232894897460938e-05,
      "model_forward_time": 5.577086925506592,
      "step": 18070
    },
    {
      "dynamics_loss_avg": 0.04648443069308996,
      "epoch": 1.7232894897460938e-05,
      "step": 18070
    },
    {
      "action_loss_avg": 0.01103203217498958,
      "epoch": 1.7232894897460938e-05,
      "step": 18070
    },
    {
      "epoch": 1.7232894897460938e-05,
      "step": 18070,
      "training_step_time": 17.015289068222046
    },
    {
      "epoch": 1.7233848571777345e-05,
      "model_forward_time": 6.020468473434448,
      "step": 18071
    },
    {
      "epoch": 1.7233848571777345e-05,
      "step": 18071,
      "training_step_time": 17.441220998764038
    },
    {
      "epoch": 1.723480224609375e-05,
      "model_forward_time": 5.585776090621948,
      "step": 18072
    },
    {
      "epoch": 1.723480224609375e-05,
      "step": 18072,
      "training_step_time": 17.001614809036255
    },
    {
      "epoch": 1.7235755920410157e-05,
      "model_forward_time": 5.5936150550842285,
      "step": 18073
    },
    {
      "epoch": 1.7235755920410157e-05,
      "step": 18073,
      "training_step_time": 17.066099882125854
    },
    {
      "epoch": 1.7236709594726564e-05,
      "model_forward_time": 5.602332830429077,
      "step": 18074
    },
    {
      "epoch": 1.7236709594726564e-05,
      "step": 18074,
      "training_step_time": 17.110518217086792
    },
    {
      "epoch": 1.7237663269042968e-05,
      "model_forward_time": 5.62202787399292,
      "step": 18075
    },
    {
      "epoch": 1.7237663269042968e-05,
      "step": 18075,
      "training_step_time": 17.12994647026062
    },
    {
      "epoch": 1.7238616943359375e-05,
      "model_forward_time": 5.984078407287598,
      "step": 18076
    },
    {
      "epoch": 1.7238616943359375e-05,
      "step": 18076,
      "training_step_time": 17.472556352615356
    },
    {
      "epoch": 1.7239570617675783e-05,
      "model_forward_time": 5.615901470184326,
      "step": 18077
    },
    {
      "epoch": 1.7239570617675783e-05,
      "step": 18077,
      "training_step_time": 17.10243797302246
    },
    {
      "epoch": 1.7240524291992187e-05,
      "model_forward_time": 5.607877016067505,
      "step": 18078
    },
    {
      "epoch": 1.7240524291992187e-05,
      "step": 18078,
      "training_step_time": 17.076743125915527
    },
    {
      "epoch": 1.7241477966308594e-05,
      "model_forward_time": 5.612590074539185,
      "step": 18079
    },
    {
      "epoch": 1.7241477966308594e-05,
      "step": 18079,
      "training_step_time": 17.080758333206177
    },
    {
      "epoch": 1.7242431640625e-05,
      "grad_norm": 0.6871597170829773,
      "learning_rate": 1e-05,
      "loss": 0.0809,
      "step": 18080
    },
    {
      "epoch": 1.7242431640625e-05,
      "model_forward_time": 5.590773582458496,
      "step": 18080
    },
    {
      "dynamics_loss_avg": 0.06961126178503037,
      "epoch": 1.7242431640625e-05,
      "step": 18080
    },
    {
      "action_loss_avg": 0.014671651530079544,
      "epoch": 1.7242431640625e-05,
      "step": 18080
    },
    {
      "epoch": 1.7242431640625e-05,
      "step": 18080,
      "training_step_time": 17.014841556549072
    },
    {
      "epoch": 1.7243385314941406e-05,
      "model_forward_time": 5.576862812042236,
      "step": 18081
    },
    {
      "epoch": 1.7243385314941406e-05,
      "step": 18081,
      "training_step_time": 17.01280164718628
    },
    {
      "epoch": 1.7244338989257813e-05,
      "model_forward_time": 5.933286666870117,
      "step": 18082
    },
    {
      "epoch": 1.7244338989257813e-05,
      "step": 18082,
      "training_step_time": 17.337395191192627
    },
    {
      "epoch": 1.7245292663574217e-05,
      "model_forward_time": 5.564884662628174,
      "step": 18083
    },
    {
      "epoch": 1.7245292663574217e-05,
      "step": 18083,
      "training_step_time": 16.98597526550293
    },
    {
      "epoch": 1.7246246337890625e-05,
      "model_forward_time": 5.568331241607666,
      "step": 18084
    },
    {
      "epoch": 1.7246246337890625e-05,
      "step": 18084,
      "training_step_time": 16.996201515197754
    },
    {
      "epoch": 1.7247200012207032e-05,
      "model_forward_time": 5.579395055770874,
      "step": 18085
    },
    {
      "epoch": 1.7247200012207032e-05,
      "step": 18085,
      "training_step_time": 17.029637098312378
    },
    {
      "epoch": 1.7248153686523436e-05,
      "model_forward_time": 5.602953195571899,
      "step": 18086
    },
    {
      "epoch": 1.7248153686523436e-05,
      "step": 18086,
      "training_step_time": 17.09047031402588
    },
    {
      "epoch": 1.7249107360839844e-05,
      "model_forward_time": 5.98266339302063,
      "step": 18087
    },
    {
      "epoch": 1.7249107360839844e-05,
      "step": 18087,
      "training_step_time": 17.479721307754517
    },
    {
      "epoch": 1.725006103515625e-05,
      "model_forward_time": 5.601319789886475,
      "step": 18088
    },
    {
      "epoch": 1.725006103515625e-05,
      "step": 18088,
      "training_step_time": 17.063791275024414
    },
    {
      "epoch": 1.7251014709472655e-05,
      "model_forward_time": 5.592181444168091,
      "step": 18089
    },
    {
      "epoch": 1.7251014709472655e-05,
      "step": 18089,
      "training_step_time": 17.04147219657898
    },
    {
      "epoch": 1.7251968383789062e-05,
      "grad_norm": 0.6064000129699707,
      "learning_rate": 1e-05,
      "loss": 0.071,
      "step": 18090
    },
    {
      "epoch": 1.7251968383789062e-05,
      "model_forward_time": 5.601778030395508,
      "step": 18090
    },
    {
      "dynamics_loss_avg": 0.053405980579555035,
      "epoch": 1.7251968383789062e-05,
      "step": 18090
    },
    {
      "action_loss_avg": 0.012865761015564203,
      "epoch": 1.7251968383789062e-05,
      "step": 18090
    },
    {
      "epoch": 1.7251968383789062e-05,
      "step": 18090,
      "training_step_time": 17.08659267425537
    },
    {
      "epoch": 1.725292205810547e-05,
      "model_forward_time": 5.606736898422241,
      "step": 18091
    },
    {
      "epoch": 1.725292205810547e-05,
      "step": 18091,
      "training_step_time": 17.10594129562378
    },
    {
      "epoch": 1.7253875732421874e-05,
      "model_forward_time": 5.608253479003906,
      "step": 18092
    },
    {
      "epoch": 1.7253875732421874e-05,
      "step": 18092,
      "training_step_time": 17.078201293945312
    },
    {
      "epoch": 1.725482940673828e-05,
      "model_forward_time": 5.955559492111206,
      "step": 18093
    },
    {
      "epoch": 1.725482940673828e-05,
      "step": 18093,
      "training_step_time": 17.378494024276733
    },
    {
      "epoch": 1.725578308105469e-05,
      "model_forward_time": 5.564672946929932,
      "step": 18094
    },
    {
      "epoch": 1.725578308105469e-05,
      "step": 18094,
      "training_step_time": 16.980384349822998
    },
    {
      "epoch": 1.7256736755371093e-05,
      "model_forward_time": 5.56587553024292,
      "step": 18095
    },
    {
      "epoch": 1.7256736755371093e-05,
      "step": 18095,
      "training_step_time": 16.987412214279175
    },
    {
      "epoch": 1.72576904296875e-05,
      "model_forward_time": 5.559398412704468,
      "step": 18096
    },
    {
      "epoch": 1.72576904296875e-05,
      "step": 18096,
      "training_step_time": 16.98285675048828
    },
    {
      "epoch": 1.7258644104003907e-05,
      "model_forward_time": 5.563265800476074,
      "step": 18097
    },
    {
      "epoch": 1.7258644104003907e-05,
      "step": 18097,
      "training_step_time": 16.979036569595337
    },
    {
      "epoch": 1.725959777832031e-05,
      "model_forward_time": 5.573469638824463,
      "step": 18098
    },
    {
      "epoch": 1.725959777832031e-05,
      "step": 18098,
      "training_step_time": 17.36200189590454
    },
    {
      "epoch": 1.726055145263672e-05,
      "model_forward_time": 5.5773961544036865,
      "step": 18099
    },
    {
      "epoch": 1.726055145263672e-05,
      "step": 18099,
      "training_step_time": 16.995791912078857
    },
    {
      "epoch": 1.7261505126953126e-05,
      "grad_norm": 1.9683692455291748,
      "learning_rate": 1e-05,
      "loss": 0.0664,
      "step": 18100
    },
    {
      "epoch": 1.7261505126953126e-05,
      "model_forward_time": 5.586933135986328,
      "step": 18100
    },
    {
      "dynamics_loss_avg": 0.058092514611780645,
      "epoch": 1.7261505126953126e-05,
      "step": 18100
    },
    {
      "action_loss_avg": 0.01096786477137357,
      "epoch": 1.7261505126953126e-05,
      "step": 18100
    },
    {
      "epoch": 1.7261505126953126e-05,
      "step": 18100,
      "training_step_time": 17.026960134506226
    },
    {
      "epoch": 1.726245880126953e-05,
      "model_forward_time": 5.601339101791382,
      "step": 18101
    },
    {
      "epoch": 1.726245880126953e-05,
      "step": 18101,
      "training_step_time": 17.080297470092773
    },
    {
      "epoch": 1.7263412475585938e-05,
      "model_forward_time": 5.6070475578308105,
      "step": 18102
    },
    {
      "epoch": 1.7263412475585938e-05,
      "step": 18102,
      "training_step_time": 17.096028804779053
    },
    {
      "epoch": 1.7264366149902345e-05,
      "model_forward_time": 5.605201959609985,
      "step": 18103
    },
    {
      "epoch": 1.7264366149902345e-05,
      "step": 18103,
      "training_step_time": 17.096238613128662
    },
    {
      "epoch": 1.726531982421875e-05,
      "model_forward_time": 5.978660821914673,
      "step": 18104
    },
    {
      "epoch": 1.726531982421875e-05,
      "step": 18104,
      "training_step_time": 17.42543387413025
    },
    {
      "epoch": 1.7266273498535157e-05,
      "model_forward_time": 5.612424850463867,
      "step": 18105
    },
    {
      "epoch": 1.7266273498535157e-05,
      "step": 18105,
      "training_step_time": 17.054424285888672
    },
    {
      "epoch": 1.7267227172851564e-05,
      "model_forward_time": 5.587288856506348,
      "step": 18106
    },
    {
      "epoch": 1.7267227172851564e-05,
      "step": 18106,
      "training_step_time": 17.03771424293518
    },
    {
      "epoch": 1.7268180847167968e-05,
      "model_forward_time": 5.603210687637329,
      "step": 18107
    },
    {
      "epoch": 1.7268180847167968e-05,
      "step": 18107,
      "training_step_time": 17.071600675582886
    },
    {
      "epoch": 1.7269134521484376e-05,
      "model_forward_time": 5.621132135391235,
      "step": 18108
    },
    {
      "epoch": 1.7269134521484376e-05,
      "step": 18108,
      "training_step_time": 17.102394580841064
    },
    {
      "epoch": 1.7270088195800783e-05,
      "model_forward_time": 5.969032287597656,
      "step": 18109
    },
    {
      "epoch": 1.7270088195800783e-05,
      "step": 18109,
      "training_step_time": 17.452868938446045
    },
    {
      "epoch": 1.7271041870117187e-05,
      "grad_norm": 1.1358884572982788,
      "learning_rate": 1e-05,
      "loss": 0.0658,
      "step": 18110
    },
    {
      "epoch": 1.7271041870117187e-05,
      "model_forward_time": 5.60814642906189,
      "step": 18110
    },
    {
      "dynamics_loss_avg": 0.05397389475256205,
      "epoch": 1.7271041870117187e-05,
      "step": 18110
    },
    {
      "action_loss_avg": 0.009487550426274537,
      "epoch": 1.7271041870117187e-05,
      "step": 18110
    },
    {
      "epoch": 1.7271041870117187e-05,
      "step": 18110,
      "training_step_time": 17.092292070388794
    },
    {
      "epoch": 1.7271995544433594e-05,
      "model_forward_time": 5.608114004135132,
      "step": 18111
    },
    {
      "epoch": 1.7271995544433594e-05,
      "step": 18111,
      "training_step_time": 17.071911573410034
    },
    {
      "epoch": 1.727294921875e-05,
      "model_forward_time": 5.594202756881714,
      "step": 18112
    },
    {
      "epoch": 1.727294921875e-05,
      "step": 18112,
      "training_step_time": 17.047010898590088
    },
    {
      "epoch": 1.7273902893066406e-05,
      "model_forward_time": 5.597939968109131,
      "step": 18113
    },
    {
      "epoch": 1.7273902893066406e-05,
      "step": 18113,
      "training_step_time": 17.033003091812134
    },
    {
      "epoch": 1.7274856567382813e-05,
      "model_forward_time": 5.595411062240601,
      "step": 18114
    },
    {
      "epoch": 1.7274856567382813e-05,
      "step": 18114,
      "training_step_time": 17.03917908668518
    },
    {
      "epoch": 1.7275810241699217e-05,
      "model_forward_time": 5.959270477294922,
      "step": 18115
    },
    {
      "epoch": 1.7275810241699217e-05,
      "step": 18115,
      "training_step_time": 17.433826446533203
    },
    {
      "epoch": 1.7276763916015625e-05,
      "model_forward_time": 5.629201889038086,
      "step": 18116
    },
    {
      "epoch": 1.7276763916015625e-05,
      "step": 18116,
      "training_step_time": 17.135878801345825
    },
    {
      "epoch": 1.7277717590332032e-05,
      "model_forward_time": 5.608198881149292,
      "step": 18117
    },
    {
      "epoch": 1.7277717590332032e-05,
      "step": 18117,
      "training_step_time": 17.07746958732605
    },
    {
      "epoch": 1.7278671264648436e-05,
      "model_forward_time": 5.594635248184204,
      "step": 18118
    },
    {
      "epoch": 1.7278671264648436e-05,
      "step": 18118,
      "training_step_time": 17.027238845825195
    },
    {
      "epoch": 1.7279624938964844e-05,
      "model_forward_time": 5.579763650894165,
      "step": 18119
    },
    {
      "epoch": 1.7279624938964844e-05,
      "step": 18119,
      "training_step_time": 17.0300395488739
    },
    {
      "epoch": 1.728057861328125e-05,
      "grad_norm": 0.843673050403595,
      "learning_rate": 1e-05,
      "loss": 0.0759,
      "step": 18120
    },
    {
      "epoch": 1.728057861328125e-05,
      "model_forward_time": 5.936766147613525,
      "step": 18120
    },
    {
      "dynamics_loss_avg": 0.06741673843935132,
      "epoch": 1.728057861328125e-05,
      "step": 18120
    },
    {
      "action_loss_avg": 0.014348170859739185,
      "epoch": 1.728057861328125e-05,
      "step": 18120
    },
    {
      "epoch": 1.728057861328125e-05,
      "step": 18120,
      "training_step_time": 17.384113550186157
    },
    {
      "epoch": 1.7281532287597655e-05,
      "model_forward_time": 5.606265544891357,
      "step": 18121
    },
    {
      "epoch": 1.7281532287597655e-05,
      "step": 18121,
      "training_step_time": 17.113182067871094
    },
    {
      "epoch": 1.7282485961914062e-05,
      "model_forward_time": 5.636510848999023,
      "step": 18122
    },
    {
      "epoch": 1.7282485961914062e-05,
      "step": 18122,
      "training_step_time": 17.145487070083618
    },
    {
      "epoch": 1.728343963623047e-05,
      "model_forward_time": 5.622642755508423,
      "step": 18123
    },
    {
      "epoch": 1.728343963623047e-05,
      "step": 18123,
      "training_step_time": 17.092076778411865
    },
    {
      "epoch": 1.7284393310546874e-05,
      "model_forward_time": 5.593004941940308,
      "step": 18124
    },
    {
      "epoch": 1.7284393310546874e-05,
      "step": 18124,
      "training_step_time": 17.027536392211914
    },
    {
      "epoch": 1.728534698486328e-05,
      "model_forward_time": 5.588084697723389,
      "step": 18125
    },
    {
      "epoch": 1.728534698486328e-05,
      "step": 18125,
      "training_step_time": 17.007248640060425
    },
    {
      "epoch": 1.728630065917969e-05,
      "model_forward_time": 5.969748497009277,
      "step": 18126
    },
    {
      "epoch": 1.728630065917969e-05,
      "step": 18126,
      "training_step_time": 17.40722632408142
    },
    {
      "epoch": 1.7287254333496093e-05,
      "model_forward_time": 5.587397336959839,
      "step": 18127
    },
    {
      "epoch": 1.7287254333496093e-05,
      "step": 18127,
      "training_step_time": 17.032472133636475
    },
    {
      "epoch": 1.72882080078125e-05,
      "model_forward_time": 5.594022274017334,
      "step": 18128
    },
    {
      "epoch": 1.72882080078125e-05,
      "step": 18128,
      "training_step_time": 17.041661977767944
    },
    {
      "epoch": 1.7289161682128908e-05,
      "model_forward_time": 5.591639757156372,
      "step": 18129
    },
    {
      "epoch": 1.7289161682128908e-05,
      "step": 18129,
      "training_step_time": 17.051215410232544
    },
    {
      "epoch": 1.729011535644531e-05,
      "grad_norm": 0.7516305446624756,
      "learning_rate": 1e-05,
      "loss": 0.0637,
      "step": 18130
    },
    {
      "epoch": 1.729011535644531e-05,
      "model_forward_time": 5.606892824172974,
      "step": 18130
    },
    {
      "dynamics_loss_avg": 0.048765840195119384,
      "epoch": 1.729011535644531e-05,
      "step": 18130
    },
    {
      "action_loss_avg": 0.010214072023518383,
      "epoch": 1.729011535644531e-05,
      "step": 18130
    },
    {
      "epoch": 1.729011535644531e-05,
      "step": 18130,
      "training_step_time": 17.08763575553894
    },
    {
      "epoch": 1.729106903076172e-05,
      "model_forward_time": 5.607612371444702,
      "step": 18131
    },
    {
      "epoch": 1.729106903076172e-05,
      "step": 18131,
      "training_step_time": 17.487587690353394
    },
    {
      "epoch": 1.7292022705078126e-05,
      "model_forward_time": 5.609015703201294,
      "step": 18132
    },
    {
      "epoch": 1.7292022705078126e-05,
      "step": 18132,
      "training_step_time": 17.115161180496216
    },
    {
      "epoch": 1.729297637939453e-05,
      "model_forward_time": 5.6180784702301025,
      "step": 18133
    },
    {
      "epoch": 1.729297637939453e-05,
      "step": 18133,
      "training_step_time": 17.100436210632324
    },
    {
      "epoch": 1.7293930053710938e-05,
      "model_forward_time": 5.606581687927246,
      "step": 18134
    },
    {
      "epoch": 1.7293930053710938e-05,
      "step": 18134,
      "training_step_time": 17.0588276386261
    },
    {
      "epoch": 1.7294883728027345e-05,
      "model_forward_time": 5.573921203613281,
      "step": 18135
    },
    {
      "epoch": 1.7294883728027345e-05,
      "step": 18135,
      "training_step_time": 16.98094367980957
    },
    {
      "epoch": 1.729583740234375e-05,
      "model_forward_time": 5.585126638412476,
      "step": 18136
    },
    {
      "epoch": 1.729583740234375e-05,
      "step": 18136,
      "training_step_time": 17.016117095947266
    },
    {
      "epoch": 1.7296791076660157e-05,
      "model_forward_time": 5.974026203155518,
      "step": 18137
    },
    {
      "epoch": 1.7296791076660157e-05,
      "step": 18137,
      "training_step_time": 17.40375804901123
    },
    {
      "epoch": 1.7297744750976564e-05,
      "model_forward_time": 5.599717617034912,
      "step": 18138
    },
    {
      "epoch": 1.7297744750976564e-05,
      "step": 18138,
      "training_step_time": 17.079110622406006
    },
    {
      "epoch": 1.7298698425292968e-05,
      "model_forward_time": 5.610761642456055,
      "step": 18139
    },
    {
      "epoch": 1.7298698425292968e-05,
      "step": 18139,
      "training_step_time": 17.10947275161743
    },
    {
      "epoch": 1.7299652099609376e-05,
      "grad_norm": 1.115317702293396,
      "learning_rate": 1e-05,
      "loss": 0.0713,
      "step": 18140
    },
    {
      "epoch": 1.7299652099609376e-05,
      "model_forward_time": 5.613430500030518,
      "step": 18140
    },
    {
      "dynamics_loss_avg": 0.06216522343456745,
      "epoch": 1.7299652099609376e-05,
      "step": 18140
    },
    {
      "action_loss_avg": 0.010029552155174315,
      "epoch": 1.7299652099609376e-05,
      "step": 18140
    },
    {
      "epoch": 1.7299652099609376e-05,
      "step": 18140,
      "training_step_time": 17.114309072494507
    },
    {
      "epoch": 1.730060577392578e-05,
      "model_forward_time": 5.6200666427612305,
      "step": 18141
    },
    {
      "epoch": 1.730060577392578e-05,
      "step": 18141,
      "training_step_time": 17.125200033187866
    },
    {
      "epoch": 1.7301559448242187e-05,
      "model_forward_time": 5.98993444442749,
      "step": 18142
    },
    {
      "epoch": 1.7301559448242187e-05,
      "step": 18142,
      "training_step_time": 17.464503049850464
    },
    {
      "epoch": 1.7302513122558594e-05,
      "model_forward_time": 5.598439693450928,
      "step": 18143
    },
    {
      "epoch": 1.7302513122558594e-05,
      "step": 18143,
      "training_step_time": 17.04102873802185
    },
    {
      "epoch": 1.7303466796875e-05,
      "model_forward_time": 5.568787097930908,
      "step": 18144
    },
    {
      "epoch": 1.7303466796875e-05,
      "step": 18144,
      "training_step_time": 16.979305028915405
    },
    {
      "epoch": 1.7304420471191406e-05,
      "model_forward_time": 5.574047327041626,
      "step": 18145
    },
    {
      "epoch": 1.7304420471191406e-05,
      "step": 18145,
      "training_step_time": 17.018957138061523
    },
    {
      "epoch": 1.7305374145507813e-05,
      "model_forward_time": 5.585446834564209,
      "step": 18146
    },
    {
      "epoch": 1.7305374145507813e-05,
      "step": 18146,
      "training_step_time": 17.015174388885498
    },
    {
      "epoch": 1.7306327819824217e-05,
      "model_forward_time": 5.586019277572632,
      "step": 18147
    },
    {
      "epoch": 1.7306327819824217e-05,
      "step": 18147,
      "training_step_time": 17.0083224773407
    },
    {
      "epoch": 1.7307281494140625e-05,
      "model_forward_time": 5.962314128875732,
      "step": 18148
    },
    {
      "epoch": 1.7307281494140625e-05,
      "step": 18148,
      "training_step_time": 17.445950031280518
    },
    {
      "epoch": 1.7308235168457032e-05,
      "model_forward_time": 5.618582725524902,
      "step": 18149
    },
    {
      "epoch": 1.7308235168457032e-05,
      "step": 18149,
      "training_step_time": 17.12837815284729
    },
    {
      "epoch": 1.7309188842773436e-05,
      "grad_norm": 1.4209091663360596,
      "learning_rate": 1e-05,
      "loss": 0.0685,
      "step": 18150
    },
    {
      "epoch": 1.7309188842773436e-05,
      "model_forward_time": 5.610977411270142,
      "step": 18150
    },
    {
      "dynamics_loss_avg": 0.05999354459345341,
      "epoch": 1.7309188842773436e-05,
      "step": 18150
    },
    {
      "action_loss_avg": 0.01451512067578733,
      "epoch": 1.7309188842773436e-05,
      "step": 18150
    },
    {
      "epoch": 1.7309188842773436e-05,
      "step": 18150,
      "training_step_time": 17.09070658683777
    },
    {
      "epoch": 1.7310142517089844e-05,
      "model_forward_time": 5.603379964828491,
      "step": 18151
    },
    {
      "epoch": 1.7310142517089844e-05,
      "step": 18151,
      "training_step_time": 17.059290647506714
    },
    {
      "epoch": 1.731109619140625e-05,
      "model_forward_time": 5.603953123092651,
      "step": 18152
    },
    {
      "epoch": 1.731109619140625e-05,
      "step": 18152,
      "training_step_time": 17.06011700630188
    },
    {
      "epoch": 1.7312049865722655e-05,
      "model_forward_time": 5.96923828125,
      "step": 18153
    },
    {
      "epoch": 1.7312049865722655e-05,
      "step": 18153,
      "training_step_time": 17.43029761314392
    },
    {
      "epoch": 1.7313003540039062e-05,
      "model_forward_time": 5.582486152648926,
      "step": 18154
    },
    {
      "epoch": 1.7313003540039062e-05,
      "step": 18154,
      "training_step_time": 17.02322554588318
    },
    {
      "epoch": 1.731395721435547e-05,
      "model_forward_time": 5.597196578979492,
      "step": 18155
    },
    {
      "epoch": 1.731395721435547e-05,
      "step": 18155,
      "training_step_time": 17.067131757736206
    },
    {
      "epoch": 1.7314910888671874e-05,
      "model_forward_time": 5.603883743286133,
      "step": 18156
    },
    {
      "epoch": 1.7314910888671874e-05,
      "step": 18156,
      "training_step_time": 17.107702493667603
    },
    {
      "epoch": 1.731586456298828e-05,
      "model_forward_time": 5.616135597229004,
      "step": 18157
    },
    {
      "epoch": 1.731586456298828e-05,
      "step": 18157,
      "training_step_time": 17.130674123764038
    },
    {
      "epoch": 1.731681823730469e-05,
      "model_forward_time": 5.627404689788818,
      "step": 18158
    },
    {
      "epoch": 1.731681823730469e-05,
      "step": 18158,
      "training_step_time": 17.110177516937256
    },
    {
      "epoch": 1.7317771911621093e-05,
      "model_forward_time": 5.9691619873046875,
      "step": 18159
    },
    {
      "epoch": 1.7317771911621093e-05,
      "step": 18159,
      "training_step_time": 17.43937587738037
    },
    {
      "epoch": 1.73187255859375e-05,
      "grad_norm": 1.0587652921676636,
      "learning_rate": 1e-05,
      "loss": 0.0577,
      "step": 18160
    },
    {
      "epoch": 1.73187255859375e-05,
      "model_forward_time": 5.596429109573364,
      "step": 18160
    },
    {
      "dynamics_loss_avg": 0.04664606600999832,
      "epoch": 1.73187255859375e-05,
      "step": 18160
    },
    {
      "action_loss_avg": 0.008627733017783612,
      "epoch": 1.73187255859375e-05,
      "step": 18160
    },
    {
      "epoch": 1.73187255859375e-05,
      "step": 18160,
      "training_step_time": 17.044241428375244
    },
    {
      "epoch": 1.7319679260253908e-05,
      "model_forward_time": 5.594278812408447,
      "step": 18161
    },
    {
      "epoch": 1.7319679260253908e-05,
      "step": 18161,
      "training_step_time": 17.05669140815735
    },
    {
      "epoch": 1.732063293457031e-05,
      "model_forward_time": 5.595194578170776,
      "step": 18162
    },
    {
      "epoch": 1.732063293457031e-05,
      "step": 18162,
      "training_step_time": 17.067426681518555
    },
    {
      "epoch": 1.732158660888672e-05,
      "model_forward_time": 5.600521802902222,
      "step": 18163
    },
    {
      "epoch": 1.732158660888672e-05,
      "step": 18163,
      "training_step_time": 17.066654682159424
    },
    {
      "epoch": 1.7322540283203126e-05,
      "model_forward_time": 5.616116285324097,
      "step": 18164
    },
    {
      "epoch": 1.7322540283203126e-05,
      "step": 18164,
      "training_step_time": 17.47339963912964
    },
    {
      "epoch": 1.732349395751953e-05,
      "model_forward_time": 5.616164922714233,
      "step": 18165
    },
    {
      "epoch": 1.732349395751953e-05,
      "step": 18165,
      "training_step_time": 17.102787017822266
    },
    {
      "epoch": 1.7324447631835938e-05,
      "model_forward_time": 5.619672060012817,
      "step": 18166
    },
    {
      "epoch": 1.7324447631835938e-05,
      "step": 18166,
      "training_step_time": 17.121618270874023
    },
    {
      "epoch": 1.7325401306152345e-05,
      "model_forward_time": 5.602864027023315,
      "step": 18167
    },
    {
      "epoch": 1.7325401306152345e-05,
      "step": 18167,
      "training_step_time": 17.07882833480835
    },
    {
      "epoch": 1.732635498046875e-05,
      "model_forward_time": 5.589665651321411,
      "step": 18168
    },
    {
      "epoch": 1.732635498046875e-05,
      "step": 18168,
      "training_step_time": 17.046431303024292
    },
    {
      "epoch": 1.7327308654785157e-05,
      "model_forward_time": 5.574790716171265,
      "step": 18169
    },
    {
      "epoch": 1.7327308654785157e-05,
      "step": 18169,
      "training_step_time": 16.98042607307434
    },
    {
      "epoch": 1.7328262329101564e-05,
      "grad_norm": 0.5830829739570618,
      "learning_rate": 1e-05,
      "loss": 0.0621,
      "step": 18170
    },
    {
      "epoch": 1.7328262329101564e-05,
      "model_forward_time": 5.9439637660980225,
      "step": 18170
    },
    {
      "dynamics_loss_avg": 0.0505774324759841,
      "epoch": 1.7328262329101564e-05,
      "step": 18170
    },
    {
      "action_loss_avg": 0.008061509276740254,
      "epoch": 1.7328262329101564e-05,
      "step": 18170
    },
    {
      "epoch": 1.7328262329101564e-05,
      "step": 18170,
      "training_step_time": 17.365789651870728
    },
    {
      "epoch": 1.7329216003417968e-05,
      "model_forward_time": 5.563386678695679,
      "step": 18171
    },
    {
      "epoch": 1.7329216003417968e-05,
      "step": 18171,
      "training_step_time": 16.974201679229736
    },
    {
      "epoch": 1.7330169677734376e-05,
      "model_forward_time": 5.565361261367798,
      "step": 18172
    },
    {
      "epoch": 1.7330169677734376e-05,
      "step": 18172,
      "training_step_time": 16.992586135864258
    },
    {
      "epoch": 1.733112335205078e-05,
      "model_forward_time": 5.56803297996521,
      "step": 18173
    },
    {
      "epoch": 1.733112335205078e-05,
      "step": 18173,
      "training_step_time": 16.98923969268799
    },
    {
      "epoch": 1.7332077026367187e-05,
      "model_forward_time": 5.583171844482422,
      "step": 18174
    },
    {
      "epoch": 1.7332077026367187e-05,
      "step": 18174,
      "training_step_time": 17.00420045852661
    },
    {
      "epoch": 1.7333030700683594e-05,
      "model_forward_time": 5.9427244663238525,
      "step": 18175
    },
    {
      "epoch": 1.7333030700683594e-05,
      "step": 18175,
      "training_step_time": 17.344051122665405
    },
    {
      "epoch": 1.7333984375e-05,
      "model_forward_time": 5.591225624084473,
      "step": 18176
    },
    {
      "epoch": 1.7333984375e-05,
      "step": 18176,
      "training_step_time": 17.059958934783936
    },
    {
      "epoch": 1.7334938049316406e-05,
      "model_forward_time": 5.605221509933472,
      "step": 18177
    },
    {
      "epoch": 1.7334938049316406e-05,
      "step": 18177,
      "training_step_time": 17.10986089706421
    },
    {
      "epoch": 1.7335891723632813e-05,
      "model_forward_time": 5.627525329589844,
      "step": 18178
    },
    {
      "epoch": 1.7335891723632813e-05,
      "step": 18178,
      "training_step_time": 17.141384840011597
    },
    {
      "epoch": 1.7336845397949217e-05,
      "model_forward_time": 5.62428879737854,
      "step": 18179
    },
    {
      "epoch": 1.7336845397949217e-05,
      "step": 18179,
      "training_step_time": 17.120722770690918
    },
    {
      "epoch": 1.7337799072265625e-05,
      "grad_norm": 0.5500028133392334,
      "learning_rate": 1e-05,
      "loss": 0.0663,
      "step": 18180
    },
    {
      "epoch": 1.7337799072265625e-05,
      "model_forward_time": 5.630192995071411,
      "step": 18180
    },
    {
      "dynamics_loss_avg": 0.054653209447860715,
      "epoch": 1.7337799072265625e-05,
      "step": 18180
    },
    {
      "action_loss_avg": 0.009327889489941298,
      "epoch": 1.7337799072265625e-05,
      "step": 18180
    },
    {
      "epoch": 1.7337799072265625e-05,
      "step": 18180,
      "training_step_time": 17.11929202079773
    },
    {
      "epoch": 1.7338752746582032e-05,
      "model_forward_time": 5.976552724838257,
      "step": 18181
    },
    {
      "epoch": 1.7338752746582032e-05,
      "step": 18181,
      "training_step_time": 17.459049701690674
    },
    {
      "epoch": 1.7339706420898436e-05,
      "model_forward_time": 5.596059799194336,
      "step": 18182
    },
    {
      "epoch": 1.7339706420898436e-05,
      "step": 18182,
      "training_step_time": 17.06819486618042
    },
    {
      "epoch": 1.7340660095214844e-05,
      "model_forward_time": 5.59471583366394,
      "step": 18183
    },
    {
      "epoch": 1.7340660095214844e-05,
      "step": 18183,
      "training_step_time": 17.034186124801636
    },
    {
      "epoch": 1.734161376953125e-05,
      "model_forward_time": 5.595522880554199,
      "step": 18184
    },
    {
      "epoch": 1.734161376953125e-05,
      "step": 18184,
      "training_step_time": 17.057018756866455
    },
    {
      "epoch": 1.7342567443847655e-05,
      "model_forward_time": 5.595882415771484,
      "step": 18185
    },
    {
      "epoch": 1.7342567443847655e-05,
      "step": 18185,
      "training_step_time": 17.069377422332764
    },
    {
      "epoch": 1.7343521118164062e-05,
      "model_forward_time": 5.963196277618408,
      "step": 18186
    },
    {
      "epoch": 1.7343521118164062e-05,
      "step": 18186,
      "training_step_time": 17.448119640350342
    },
    {
      "epoch": 1.734447479248047e-05,
      "model_forward_time": 5.617450714111328,
      "step": 18187
    },
    {
      "epoch": 1.734447479248047e-05,
      "step": 18187,
      "training_step_time": 17.105664491653442
    },
    {
      "epoch": 1.7345428466796874e-05,
      "model_forward_time": 5.614492177963257,
      "step": 18188
    },
    {
      "epoch": 1.7345428466796874e-05,
      "step": 18188,
      "training_step_time": 17.118667125701904
    },
    {
      "epoch": 1.734638214111328e-05,
      "model_forward_time": 5.606210947036743,
      "step": 18189
    },
    {
      "epoch": 1.734638214111328e-05,
      "step": 18189,
      "training_step_time": 17.101641416549683
    },
    {
      "epoch": 1.734733581542969e-05,
      "grad_norm": 0.18175649642944336,
      "learning_rate": 1e-05,
      "loss": 0.051,
      "step": 18190
    },
    {
      "epoch": 1.734733581542969e-05,
      "model_forward_time": 4.280230522155762,
      "step": 18190
    },
    {
      "dynamics_loss_avg": 0.04697425803169608,
      "epoch": 1.734733581542969e-05,
      "step": 18190
    },
    {
      "action_loss_avg": 0.010875748586840928,
      "epoch": 1.734733581542969e-05,
      "step": 18190
    },
    {
      "epoch": 1.734733581542969e-05,
      "step": 18190,
      "training_step_time": 12.817445039749146
    },
    {
      "epoch": 1.7348289489746093e-05,
      "model_forward_time": 5.614351034164429,
      "step": 18191
    },
    {
      "epoch": 1.7348289489746093e-05,
      "step": 18191,
      "training_step_time": 17.087105751037598
    },
    {
      "epoch": 1.73492431640625e-05,
      "model_forward_time": 5.984731912612915,
      "step": 18192
    },
    {
      "epoch": 1.73492431640625e-05,
      "step": 18192,
      "training_step_time": 17.448126316070557
    },
    {
      "epoch": 1.7350196838378908e-05,
      "model_forward_time": 5.5924859046936035,
      "step": 18193
    },
    {
      "epoch": 1.7350196838378908e-05,
      "step": 18193,
      "training_step_time": 17.046717643737793
    },
    {
      "epoch": 1.735115051269531e-05,
      "model_forward_time": 5.602618932723999,
      "step": 18194
    },
    {
      "epoch": 1.735115051269531e-05,
      "step": 18194,
      "training_step_time": 17.08274507522583
    },
    {
      "epoch": 1.735210418701172e-05,
      "model_forward_time": 5.629049062728882,
      "step": 18195
    },
    {
      "epoch": 1.735210418701172e-05,
      "step": 18195,
      "training_step_time": 17.108532428741455
    },
    {
      "epoch": 1.7353057861328126e-05,
      "model_forward_time": 4.296202659606934,
      "step": 18196
    },
    {
      "epoch": 1.7353057861328126e-05,
      "step": 18196,
      "training_step_time": 12.83315920829773
    },
    {
      "epoch": 1.735401153564453e-05,
      "model_forward_time": 5.616601228713989,
      "step": 18197
    },
    {
      "epoch": 1.735401153564453e-05,
      "step": 18197,
      "training_step_time": 17.111093282699585
    },
    {
      "epoch": 1.7354965209960938e-05,
      "model_forward_time": 5.978351593017578,
      "step": 18198
    },
    {
      "epoch": 1.7354965209960938e-05,
      "step": 18198,
      "training_step_time": 17.471876621246338
    },
    {
      "epoch": 1.7355918884277345e-05,
      "model_forward_time": 5.601499557495117,
      "step": 18199
    },
    {
      "epoch": 1.7355918884277345e-05,
      "step": 18199,
      "training_step_time": 17.043792486190796
    },
    {
      "epoch": 1.735687255859375e-05,
      "grad_norm": 1.0178927183151245,
      "learning_rate": 1e-05,
      "loss": 0.0813,
      "step": 18200
    },
    {
      "epoch": 1.735687255859375e-05,
      "model_forward_time": 5.5971598625183105,
      "step": 18200
    },
    {
      "dynamics_loss_avg": 0.06400282848626375,
      "epoch": 1.735687255859375e-05,
      "step": 18200
    },
    {
      "action_loss_avg": 0.010956057999283075,
      "epoch": 1.735687255859375e-05,
      "step": 18200
    },
    {
      "epoch": 1.735687255859375e-05,
      "step": 18200,
      "training_step_time": 17.040499448776245
    },
    {
      "epoch": 1.7357826232910157e-05,
      "model_forward_time": 5.5975847244262695,
      "step": 18201
    },
    {
      "epoch": 1.7357826232910157e-05,
      "step": 18201,
      "training_step_time": 17.03629732131958
    },
    {
      "epoch": 1.7358779907226564e-05,
      "model_forward_time": 5.594872951507568,
      "step": 18202
    },
    {
      "epoch": 1.7358779907226564e-05,
      "step": 18202,
      "training_step_time": 17.02372932434082
    },
    {
      "epoch": 1.7359733581542968e-05,
      "model_forward_time": 5.980571508407593,
      "step": 18203
    },
    {
      "epoch": 1.7359733581542968e-05,
      "step": 18203,
      "training_step_time": 17.44086265563965
    },
    {
      "epoch": 1.7360687255859376e-05,
      "model_forward_time": 5.60227632522583,
      "step": 18204
    },
    {
      "epoch": 1.7360687255859376e-05,
      "step": 18204,
      "training_step_time": 17.07691979408264
    },
    {
      "epoch": 1.736164093017578e-05,
      "model_forward_time": 5.6091532707214355,
      "step": 18205
    },
    {
      "epoch": 1.736164093017578e-05,
      "step": 18205,
      "training_step_time": 17.118857383728027
    },
    {
      "epoch": 1.7362594604492187e-05,
      "model_forward_time": 5.614004373550415,
      "step": 18206
    },
    {
      "epoch": 1.7362594604492187e-05,
      "step": 18206,
      "training_step_time": 17.083001136779785
    },
    {
      "epoch": 1.7363548278808594e-05,
      "model_forward_time": 5.591081142425537,
      "step": 18207
    },
    {
      "epoch": 1.7363548278808594e-05,
      "step": 18207,
      "training_step_time": 17.018694162368774
    },
    {
      "epoch": 1.7364501953125e-05,
      "model_forward_time": 5.588310718536377,
      "step": 18208
    },
    {
      "epoch": 1.7364501953125e-05,
      "step": 18208,
      "training_step_time": 17.37342119216919
    },
    {
      "epoch": 1.7365455627441406e-05,
      "model_forward_time": 5.570939064025879,
      "step": 18209
    },
    {
      "epoch": 1.7365455627441406e-05,
      "step": 18209,
      "training_step_time": 16.976038455963135
    },
    {
      "epoch": 1.7366409301757813e-05,
      "grad_norm": 0.9948380589485168,
      "learning_rate": 1e-05,
      "loss": 0.0637,
      "step": 18210
    },
    {
      "epoch": 1.7366409301757813e-05,
      "model_forward_time": 5.579728126525879,
      "step": 18210
    },
    {
      "dynamics_loss_avg": 0.05660352595150471,
      "epoch": 1.7366409301757813e-05,
      "step": 18210
    },
    {
      "action_loss_avg": 0.008860974851995707,
      "epoch": 1.7366409301757813e-05,
      "step": 18210
    },
    {
      "epoch": 1.7366409301757813e-05,
      "step": 18210,
      "training_step_time": 17.01888656616211
    },
    {
      "epoch": 1.7367362976074217e-05,
      "model_forward_time": 5.571308851242065,
      "step": 18211
    },
    {
      "epoch": 1.7367362976074217e-05,
      "step": 18211,
      "training_step_time": 17.002881288528442
    },
    {
      "epoch": 1.7368316650390625e-05,
      "model_forward_time": 5.5828857421875,
      "step": 18212
    },
    {
      "epoch": 1.7368316650390625e-05,
      "step": 18212,
      "training_step_time": 17.040916442871094
    },
    {
      "epoch": 1.7369270324707032e-05,
      "model_forward_time": 5.596989154815674,
      "step": 18213
    },
    {
      "epoch": 1.7369270324707032e-05,
      "step": 18213,
      "training_step_time": 17.066731452941895
    },
    {
      "epoch": 1.7370223999023436e-05,
      "model_forward_time": 5.971063137054443,
      "step": 18214
    },
    {
      "epoch": 1.7370223999023436e-05,
      "step": 18214,
      "training_step_time": 17.42198610305786
    },
    {
      "epoch": 1.7371177673339844e-05,
      "model_forward_time": 5.603362321853638,
      "step": 18215
    },
    {
      "epoch": 1.7371177673339844e-05,
      "step": 18215,
      "training_step_time": 17.073991537094116
    },
    {
      "epoch": 1.737213134765625e-05,
      "model_forward_time": 5.608509063720703,
      "step": 18216
    },
    {
      "epoch": 1.737213134765625e-05,
      "step": 18216,
      "training_step_time": 17.09852433204651
    },
    {
      "epoch": 1.7373085021972655e-05,
      "model_forward_time": 5.609833478927612,
      "step": 18217
    },
    {
      "epoch": 1.7373085021972655e-05,
      "step": 18217,
      "training_step_time": 17.094629526138306
    },
    {
      "epoch": 1.7374038696289062e-05,
      "model_forward_time": 5.614559888839722,
      "step": 18218
    },
    {
      "epoch": 1.7374038696289062e-05,
      "step": 18218,
      "training_step_time": 17.11025619506836
    },
    {
      "epoch": 1.737499237060547e-05,
      "model_forward_time": 5.985239267349243,
      "step": 18219
    },
    {
      "epoch": 1.737499237060547e-05,
      "step": 18219,
      "training_step_time": 17.470093488693237
    },
    {
      "epoch": 1.7375946044921874e-05,
      "grad_norm": 0.4018633961677551,
      "learning_rate": 1e-05,
      "loss": 0.0563,
      "step": 18220
    },
    {
      "epoch": 1.7375946044921874e-05,
      "model_forward_time": 5.610836982727051,
      "step": 18220
    },
    {
      "dynamics_loss_avg": 0.04784888345748186,
      "epoch": 1.7375946044921874e-05,
      "step": 18220
    },
    {
      "action_loss_avg": 0.0066751097911037505,
      "epoch": 1.7375946044921874e-05,
      "step": 18220
    },
    {
      "epoch": 1.7375946044921874e-05,
      "step": 18220,
      "training_step_time": 17.033690214157104
    },
    {
      "epoch": 1.737689971923828e-05,
      "model_forward_time": 5.613264560699463,
      "step": 18221
    },
    {
      "epoch": 1.737689971923828e-05,
      "step": 18221,
      "training_step_time": 17.09681487083435
    },
    {
      "epoch": 1.737785339355469e-05,
      "model_forward_time": 5.6080543994903564,
      "step": 18222
    },
    {
      "epoch": 1.737785339355469e-05,
      "step": 18222,
      "training_step_time": 17.086429834365845
    },
    {
      "epoch": 1.7378807067871093e-05,
      "model_forward_time": 5.604948043823242,
      "step": 18223
    },
    {
      "epoch": 1.7378807067871093e-05,
      "step": 18223,
      "training_step_time": 17.077834844589233
    },
    {
      "epoch": 1.73797607421875e-05,
      "model_forward_time": 5.601680517196655,
      "step": 18224
    },
    {
      "epoch": 1.73797607421875e-05,
      "step": 18224,
      "training_step_time": 17.079172134399414
    },
    {
      "epoch": 1.7380714416503908e-05,
      "model_forward_time": 5.972856283187866,
      "step": 18225
    },
    {
      "epoch": 1.7380714416503908e-05,
      "step": 18225,
      "training_step_time": 17.457688808441162
    },
    {
      "epoch": 1.738166809082031e-05,
      "model_forward_time": 5.60141921043396,
      "step": 18226
    },
    {
      "epoch": 1.738166809082031e-05,
      "step": 18226,
      "training_step_time": 17.070695400238037
    },
    {
      "epoch": 1.738262176513672e-05,
      "model_forward_time": 5.603817462921143,
      "step": 18227
    },
    {
      "epoch": 1.738262176513672e-05,
      "step": 18227,
      "training_step_time": 17.065874814987183
    },
    {
      "epoch": 1.7383575439453126e-05,
      "model_forward_time": 5.6099231243133545,
      "step": 18228
    },
    {
      "epoch": 1.7383575439453126e-05,
      "step": 18228,
      "training_step_time": 17.089951515197754
    },
    {
      "epoch": 1.738452911376953e-05,
      "model_forward_time": 5.59951639175415,
      "step": 18229
    },
    {
      "epoch": 1.738452911376953e-05,
      "step": 18229,
      "training_step_time": 17.047642469406128
    },
    {
      "epoch": 1.7385482788085938e-05,
      "grad_norm": 0.3751193583011627,
      "learning_rate": 1e-05,
      "loss": 0.0623,
      "step": 18230
    },
    {
      "epoch": 1.7385482788085938e-05,
      "model_forward_time": 5.588685989379883,
      "step": 18230
    },
    {
      "dynamics_loss_avg": 0.053414597548544406,
      "epoch": 1.7385482788085938e-05,
      "step": 18230
    },
    {
      "action_loss_avg": 0.007781674771104008,
      "epoch": 1.7385482788085938e-05,
      "step": 18230
    },
    {
      "epoch": 1.7385482788085938e-05,
      "step": 18230,
      "training_step_time": 17.41214346885681
    },
    {
      "epoch": 1.7386436462402345e-05,
      "model_forward_time": 5.593628406524658,
      "step": 18231
    },
    {
      "epoch": 1.7386436462402345e-05,
      "step": 18231,
      "training_step_time": 17.03861665725708
    },
    {
      "epoch": 1.738739013671875e-05,
      "model_forward_time": 5.5935280323028564,
      "step": 18232
    },
    {
      "epoch": 1.738739013671875e-05,
      "step": 18232,
      "training_step_time": 17.063832759857178
    },
    {
      "epoch": 1.7388343811035157e-05,
      "model_forward_time": 5.6050848960876465,
      "step": 18233
    },
    {
      "epoch": 1.7388343811035157e-05,
      "step": 18233,
      "training_step_time": 17.08021068572998
    },
    {
      "epoch": 1.7389297485351564e-05,
      "model_forward_time": 5.603769302368164,
      "step": 18234
    },
    {
      "epoch": 1.7389297485351564e-05,
      "step": 18234,
      "training_step_time": 17.055158138275146
    },
    {
      "epoch": 1.7390251159667968e-05,
      "model_forward_time": 5.596799612045288,
      "step": 18235
    },
    {
      "epoch": 1.7390251159667968e-05,
      "step": 18235,
      "training_step_time": 17.05733323097229
    },
    {
      "epoch": 1.7391204833984376e-05,
      "model_forward_time": 5.977234363555908,
      "step": 18236
    },
    {
      "epoch": 1.7391204833984376e-05,
      "step": 18236,
      "training_step_time": 17.464585542678833
    },
    {
      "epoch": 1.739215850830078e-05,
      "model_forward_time": 5.620755910873413,
      "step": 18237
    },
    {
      "epoch": 1.739215850830078e-05,
      "step": 18237,
      "training_step_time": 17.12234354019165
    },
    {
      "epoch": 1.7393112182617187e-05,
      "model_forward_time": 5.614061594009399,
      "step": 18238
    },
    {
      "epoch": 1.7393112182617187e-05,
      "step": 18238,
      "training_step_time": 17.12744379043579
    },
    {
      "epoch": 1.7394065856933594e-05,
      "model_forward_time": 5.621855974197388,
      "step": 18239
    },
    {
      "epoch": 1.7394065856933594e-05,
      "step": 18239,
      "training_step_time": 17.107069969177246
    },
    {
      "epoch": 1.739501953125e-05,
      "grad_norm": 0.8856035470962524,
      "learning_rate": 1e-05,
      "loss": 0.0514,
      "step": 18240
    },
    {
      "epoch": 1.739501953125e-05,
      "model_forward_time": 5.59863805770874,
      "step": 18240
    },
    {
      "dynamics_loss_avg": 0.05236416161060333,
      "epoch": 1.739501953125e-05,
      "step": 18240
    },
    {
      "action_loss_avg": 0.007501278491690755,
      "epoch": 1.739501953125e-05,
      "step": 18240
    },
    {
      "epoch": 1.739501953125e-05,
      "step": 18240,
      "training_step_time": 17.016948223114014
    },
    {
      "epoch": 1.7395973205566406e-05,
      "model_forward_time": 5.949141979217529,
      "step": 18241
    },
    {
      "epoch": 1.7395973205566406e-05,
      "step": 18241,
      "training_step_time": 17.39928102493286
    },
    {
      "epoch": 1.7396926879882813e-05,
      "model_forward_time": 5.580874443054199,
      "step": 18242
    },
    {
      "epoch": 1.7396926879882813e-05,
      "step": 18242,
      "training_step_time": 16.99675488471985
    },
    {
      "epoch": 1.7397880554199217e-05,
      "model_forward_time": 5.591124534606934,
      "step": 18243
    },
    {
      "epoch": 1.7397880554199217e-05,
      "step": 18243,
      "training_step_time": 16.99396324157715
    },
    {
      "epoch": 1.7398834228515625e-05,
      "model_forward_time": 5.596094846725464,
      "step": 18244
    },
    {
      "epoch": 1.7398834228515625e-05,
      "step": 18244,
      "training_step_time": 17.02830672264099
    },
    {
      "epoch": 1.7399787902832032e-05,
      "model_forward_time": 5.591969013214111,
      "step": 18245
    },
    {
      "epoch": 1.7399787902832032e-05,
      "step": 18245,
      "training_step_time": 17.035473823547363
    },
    {
      "epoch": 1.7400741577148436e-05,
      "model_forward_time": 5.5990142822265625,
      "step": 18246
    },
    {
      "epoch": 1.7400741577148436e-05,
      "step": 18246,
      "training_step_time": 17.076223850250244
    },
    {
      "epoch": 1.7401695251464844e-05,
      "model_forward_time": 5.985464096069336,
      "step": 18247
    },
    {
      "epoch": 1.7401695251464844e-05,
      "step": 18247,
      "training_step_time": 17.477466583251953
    },
    {
      "epoch": 1.740264892578125e-05,
      "model_forward_time": 5.614548206329346,
      "step": 18248
    },
    {
      "epoch": 1.740264892578125e-05,
      "step": 18248,
      "training_step_time": 17.12102484703064
    },
    {
      "epoch": 1.7403602600097655e-05,
      "model_forward_time": 5.6222569942474365,
      "step": 18249
    },
    {
      "epoch": 1.7403602600097655e-05,
      "step": 18249,
      "training_step_time": 17.12770414352417
    },
    {
      "epoch": 1.7404556274414062e-05,
      "grad_norm": 0.44859588146209717,
      "learning_rate": 1e-05,
      "loss": 0.0619,
      "step": 18250
    },
    {
      "epoch": 1.7404556274414062e-05,
      "model_forward_time": 5.6102375984191895,
      "step": 18250
    },
    {
      "dynamics_loss_avg": 0.051115952990949155,
      "epoch": 1.7404556274414062e-05,
      "step": 18250
    },
    {
      "action_loss_avg": 0.008414413663558662,
      "epoch": 1.7404556274414062e-05,
      "step": 18250
    },
    {
      "epoch": 1.7404556274414062e-05,
      "step": 18250,
      "training_step_time": 17.114487171173096
    },
    {
      "epoch": 1.740550994873047e-05,
      "model_forward_time": 5.606392860412598,
      "step": 18251
    },
    {
      "epoch": 1.740550994873047e-05,
      "step": 18251,
      "training_step_time": 17.090246200561523
    },
    {
      "epoch": 1.7406463623046874e-05,
      "model_forward_time": 5.944985628128052,
      "step": 18252
    },
    {
      "epoch": 1.7406463623046874e-05,
      "step": 18252,
      "training_step_time": 17.388391256332397
    },
    {
      "epoch": 1.740741729736328e-05,
      "model_forward_time": 5.588914632797241,
      "step": 18253
    },
    {
      "epoch": 1.740741729736328e-05,
      "step": 18253,
      "training_step_time": 16.997305870056152
    },
    {
      "epoch": 1.740837097167969e-05,
      "model_forward_time": 5.5774147510528564,
      "step": 18254
    },
    {
      "epoch": 1.740837097167969e-05,
      "step": 18254,
      "training_step_time": 16.99243712425232
    },
    {
      "epoch": 1.7409324645996093e-05,
      "model_forward_time": 5.584194898605347,
      "step": 18255
    },
    {
      "epoch": 1.7409324645996093e-05,
      "step": 18255,
      "training_step_time": 16.995165586471558
    },
    {
      "epoch": 1.74102783203125e-05,
      "model_forward_time": 5.581832647323608,
      "step": 18256
    },
    {
      "epoch": 1.74102783203125e-05,
      "step": 18256,
      "training_step_time": 17.021310329437256
    },
    {
      "epoch": 1.7411231994628908e-05,
      "model_forward_time": 5.586501598358154,
      "step": 18257
    },
    {
      "epoch": 1.7411231994628908e-05,
      "step": 18257,
      "training_step_time": 17.025257349014282
    },
    {
      "epoch": 1.741218566894531e-05,
      "model_forward_time": 5.9652886390686035,
      "step": 18258
    },
    {
      "epoch": 1.741218566894531e-05,
      "step": 18258,
      "training_step_time": 17.399129629135132
    },
    {
      "epoch": 1.741313934326172e-05,
      "model_forward_time": 5.592216968536377,
      "step": 18259
    },
    {
      "epoch": 1.741313934326172e-05,
      "step": 18259,
      "training_step_time": 17.064645051956177
    },
    {
      "epoch": 1.7414093017578126e-05,
      "grad_norm": 0.8102697730064392,
      "learning_rate": 1e-05,
      "loss": 0.0556,
      "step": 18260
    },
    {
      "epoch": 1.7414093017578126e-05,
      "model_forward_time": 5.606722593307495,
      "step": 18260
    },
    {
      "dynamics_loss_avg": 0.04627414792776108,
      "epoch": 1.7414093017578126e-05,
      "step": 18260
    },
    {
      "action_loss_avg": 0.008825248572975397,
      "epoch": 1.7414093017578126e-05,
      "step": 18260
    },
    {
      "epoch": 1.7414093017578126e-05,
      "step": 18260,
      "training_step_time": 17.085651636123657
    },
    {
      "epoch": 1.741504669189453e-05,
      "model_forward_time": 5.608874320983887,
      "step": 18261
    },
    {
      "epoch": 1.741504669189453e-05,
      "step": 18261,
      "training_step_time": 17.091667890548706
    },
    {
      "epoch": 1.7416000366210938e-05,
      "model_forward_time": 5.613359451293945,
      "step": 18262
    },
    {
      "epoch": 1.7416000366210938e-05,
      "step": 18262,
      "training_step_time": 17.108726024627686
    },
    {
      "epoch": 1.7416954040527345e-05,
      "model_forward_time": 5.6075358390808105,
      "step": 18263
    },
    {
      "epoch": 1.7416954040527345e-05,
      "step": 18263,
      "training_step_time": 17.460797548294067
    },
    {
      "epoch": 1.741790771484375e-05,
      "model_forward_time": 5.611315011978149,
      "step": 18264
    },
    {
      "epoch": 1.741790771484375e-05,
      "step": 18264,
      "training_step_time": 17.10909628868103
    },
    {
      "epoch": 1.7418861389160157e-05,
      "model_forward_time": 5.61428165435791,
      "step": 18265
    },
    {
      "epoch": 1.7418861389160157e-05,
      "step": 18265,
      "training_step_time": 17.10410714149475
    },
    {
      "epoch": 1.741981506347656e-05,
      "model_forward_time": 5.60578989982605,
      "step": 18266
    },
    {
      "epoch": 1.741981506347656e-05,
      "step": 18266,
      "training_step_time": 17.096506357192993
    },
    {
      "epoch": 1.7420768737792968e-05,
      "model_forward_time": 5.601157903671265,
      "step": 18267
    },
    {
      "epoch": 1.7420768737792968e-05,
      "step": 18267,
      "training_step_time": 17.0947265625
    },
    {
      "epoch": 1.7421722412109376e-05,
      "model_forward_time": 5.614018201828003,
      "step": 18268
    },
    {
      "epoch": 1.7421722412109376e-05,
      "step": 18268,
      "training_step_time": 17.10565710067749
    },
    {
      "epoch": 1.742267608642578e-05,
      "model_forward_time": 5.977971792221069,
      "step": 18269
    },
    {
      "epoch": 1.742267608642578e-05,
      "step": 18269,
      "training_step_time": 17.46353006362915
    },
    {
      "epoch": 1.7423629760742187e-05,
      "grad_norm": 0.7964903116226196,
      "learning_rate": 1e-05,
      "loss": 0.0757,
      "step": 18270
    },
    {
      "epoch": 1.7423629760742187e-05,
      "model_forward_time": 5.607838153839111,
      "step": 18270
    },
    {
      "dynamics_loss_avg": 0.06485770158469677,
      "epoch": 1.7423629760742187e-05,
      "step": 18270
    },
    {
      "action_loss_avg": 0.0139120452105999,
      "epoch": 1.7423629760742187e-05,
      "step": 18270
    },
    {
      "epoch": 1.7423629760742187e-05,
      "step": 18270,
      "training_step_time": 17.095210313796997
    },
    {
      "epoch": 1.7424583435058595e-05,
      "model_forward_time": 5.607827186584473,
      "step": 18271
    },
    {
      "epoch": 1.7424583435058595e-05,
      "step": 18271,
      "training_step_time": 17.092073917388916
    },
    {
      "epoch": 1.7425537109375e-05,
      "model_forward_time": 5.61289381980896,
      "step": 18272
    },
    {
      "epoch": 1.7425537109375e-05,
      "step": 18272,
      "training_step_time": 17.07835841178894
    },
    {
      "epoch": 1.7426490783691406e-05,
      "model_forward_time": 5.5957746505737305,
      "step": 18273
    },
    {
      "epoch": 1.7426490783691406e-05,
      "step": 18273,
      "training_step_time": 17.038212299346924
    },
    {
      "epoch": 1.7427444458007813e-05,
      "model_forward_time": 5.94851279258728,
      "step": 18274
    },
    {
      "epoch": 1.7427444458007813e-05,
      "step": 18274,
      "training_step_time": 17.360138416290283
    },
    {
      "epoch": 1.7428398132324217e-05,
      "model_forward_time": 5.595151662826538,
      "step": 18275
    },
    {
      "epoch": 1.7428398132324217e-05,
      "step": 18275,
      "training_step_time": 17.033831357955933
    },
    {
      "epoch": 1.7429351806640625e-05,
      "model_forward_time": 5.596246719360352,
      "step": 18276
    },
    {
      "epoch": 1.7429351806640625e-05,
      "step": 18276,
      "training_step_time": 17.075267553329468
    },
    {
      "epoch": 1.7430305480957032e-05,
      "model_forward_time": 5.610767364501953,
      "step": 18277
    },
    {
      "epoch": 1.7430305480957032e-05,
      "step": 18277,
      "training_step_time": 17.109014987945557
    },
    {
      "epoch": 1.7431259155273436e-05,
      "model_forward_time": 5.628412961959839,
      "step": 18278
    },
    {
      "epoch": 1.7431259155273436e-05,
      "step": 18278,
      "training_step_time": 17.13793182373047
    },
    {
      "epoch": 1.7432212829589844e-05,
      "model_forward_time": 5.6109607219696045,
      "step": 18279
    },
    {
      "epoch": 1.7432212829589844e-05,
      "step": 18279,
      "training_step_time": 17.108287811279297
    },
    {
      "epoch": 1.743316650390625e-05,
      "grad_norm": 0.7311543226242065,
      "learning_rate": 1e-05,
      "loss": 0.0653,
      "step": 18280
    },
    {
      "epoch": 1.743316650390625e-05,
      "model_forward_time": 5.993025541305542,
      "step": 18280
    },
    {
      "dynamics_loss_avg": 0.05546565186232329,
      "epoch": 1.743316650390625e-05,
      "step": 18280
    },
    {
      "action_loss_avg": 0.007207623054273427,
      "epoch": 1.743316650390625e-05,
      "step": 18280
    },
    {
      "epoch": 1.743316650390625e-05,
      "step": 18280,
      "training_step_time": 17.501638650894165
    },
    {
      "epoch": 1.7434120178222655e-05,
      "model_forward_time": 5.618307828903198,
      "step": 18281
    },
    {
      "epoch": 1.7434120178222655e-05,
      "step": 18281,
      "training_step_time": 17.10373091697693
    },
    {
      "epoch": 1.7435073852539063e-05,
      "model_forward_time": 5.621346473693848,
      "step": 18282
    },
    {
      "epoch": 1.7435073852539063e-05,
      "step": 18282,
      "training_step_time": 17.087367296218872
    },
    {
      "epoch": 1.743602752685547e-05,
      "model_forward_time": 5.5992591381073,
      "step": 18283
    },
    {
      "epoch": 1.743602752685547e-05,
      "step": 18283,
      "training_step_time": 17.061311721801758
    },
    {
      "epoch": 1.7436981201171874e-05,
      "model_forward_time": 5.5930986404418945,
      "step": 18284
    },
    {
      "epoch": 1.7436981201171874e-05,
      "step": 18284,
      "training_step_time": 17.057721614837646
    },
    {
      "epoch": 1.743793487548828e-05,
      "model_forward_time": 5.96451210975647,
      "step": 18285
    },
    {
      "epoch": 1.743793487548828e-05,
      "step": 18285,
      "training_step_time": 17.445104122161865
    },
    {
      "epoch": 1.743888854980469e-05,
      "model_forward_time": 5.606560707092285,
      "step": 18286
    },
    {
      "epoch": 1.743888854980469e-05,
      "step": 18286,
      "training_step_time": 17.083815574645996
    },
    {
      "epoch": 1.7439842224121093e-05,
      "model_forward_time": 5.608347177505493,
      "step": 18287
    },
    {
      "epoch": 1.7439842224121093e-05,
      "step": 18287,
      "training_step_time": 17.0765483379364
    },
    {
      "epoch": 1.74407958984375e-05,
      "model_forward_time": 5.599464416503906,
      "step": 18288
    },
    {
      "epoch": 1.74407958984375e-05,
      "step": 18288,
      "training_step_time": 17.075295448303223
    },
    {
      "epoch": 1.7441749572753908e-05,
      "model_forward_time": 5.601640701293945,
      "step": 18289
    },
    {
      "epoch": 1.7441749572753908e-05,
      "step": 18289,
      "training_step_time": 17.06939172744751
    },
    {
      "epoch": 1.744270324707031e-05,
      "grad_norm": 0.5167295336723328,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 18290
    },
    {
      "epoch": 1.744270324707031e-05,
      "model_forward_time": 5.604059219360352,
      "step": 18290
    },
    {
      "dynamics_loss_avg": 0.05764134004712105,
      "epoch": 1.744270324707031e-05,
      "step": 18290
    },
    {
      "action_loss_avg": 0.009406044241040945,
      "epoch": 1.744270324707031e-05,
      "step": 18290
    },
    {
      "epoch": 1.744270324707031e-05,
      "step": 18290,
      "training_step_time": 17.047117948532104
    },
    {
      "epoch": 1.744365692138672e-05,
      "model_forward_time": 5.9481751918792725,
      "step": 18291
    },
    {
      "epoch": 1.744365692138672e-05,
      "step": 18291,
      "training_step_time": 17.406253576278687
    },
    {
      "epoch": 1.7444610595703127e-05,
      "model_forward_time": 5.594433307647705,
      "step": 18292
    },
    {
      "epoch": 1.7444610595703127e-05,
      "step": 18292,
      "training_step_time": 17.0631902217865
    },
    {
      "epoch": 1.744556427001953e-05,
      "model_forward_time": 5.601001501083374,
      "step": 18293
    },
    {
      "epoch": 1.744556427001953e-05,
      "step": 18293,
      "training_step_time": 17.080179929733276
    },
    {
      "epoch": 1.7446517944335938e-05,
      "model_forward_time": 5.599637508392334,
      "step": 18294
    },
    {
      "epoch": 1.7446517944335938e-05,
      "step": 18294,
      "training_step_time": 17.048496961593628
    },
    {
      "epoch": 1.7447471618652345e-05,
      "model_forward_time": 5.59786057472229,
      "step": 18295
    },
    {
      "epoch": 1.7447471618652345e-05,
      "step": 18295,
      "training_step_time": 17.028123140335083
    },
    {
      "epoch": 1.744842529296875e-05,
      "model_forward_time": 5.587031364440918,
      "step": 18296
    },
    {
      "epoch": 1.744842529296875e-05,
      "step": 18296,
      "training_step_time": 17.426160097122192
    },
    {
      "epoch": 1.7449378967285157e-05,
      "model_forward_time": 5.603070974349976,
      "step": 18297
    },
    {
      "epoch": 1.7449378967285157e-05,
      "step": 18297,
      "training_step_time": 17.102668523788452
    },
    {
      "epoch": 1.745033264160156e-05,
      "model_forward_time": 5.6225244998931885,
      "step": 18298
    },
    {
      "epoch": 1.745033264160156e-05,
      "step": 18298,
      "training_step_time": 17.095134735107422
    },
    {
      "epoch": 1.7451286315917968e-05,
      "model_forward_time": 5.592474460601807,
      "step": 18299
    },
    {
      "epoch": 1.7451286315917968e-05,
      "step": 18299,
      "training_step_time": 17.01643204689026
    },
    {
      "epoch": 1.7452239990234376e-05,
      "grad_norm": 1.8879696130752563,
      "learning_rate": 1e-05,
      "loss": 0.0678,
      "step": 18300
    },
    {
      "epoch": 1.7452239990234376e-05,
      "model_forward_time": 5.577677965164185,
      "step": 18300
    },
    {
      "dynamics_loss_avg": 0.05750154294073582,
      "epoch": 1.7452239990234376e-05,
      "step": 18300
    },
    {
      "action_loss_avg": 0.007377148524392396,
      "epoch": 1.7452239990234376e-05,
      "step": 18300
    },
    {
      "epoch": 1.7452239990234376e-05,
      "step": 18300,
      "training_step_time": 16.98004388809204
    },
    {
      "epoch": 1.745319366455078e-05,
      "model_forward_time": 5.582156658172607,
      "step": 18301
    },
    {
      "epoch": 1.745319366455078e-05,
      "step": 18301,
      "training_step_time": 17.014447927474976
    },
    {
      "epoch": 1.7454147338867187e-05,
      "model_forward_time": 5.9512693881988525,
      "step": 18302
    },
    {
      "epoch": 1.7454147338867187e-05,
      "step": 18302,
      "training_step_time": 17.37752413749695
    },
    {
      "epoch": 1.7455101013183595e-05,
      "model_forward_time": 5.59467887878418,
      "step": 18303
    },
    {
      "epoch": 1.7455101013183595e-05,
      "step": 18303,
      "training_step_time": 17.029229164123535
    },
    {
      "epoch": 1.74560546875e-05,
      "model_forward_time": 5.593170404434204,
      "step": 18304
    },
    {
      "epoch": 1.74560546875e-05,
      "step": 18304,
      "training_step_time": 17.055776834487915
    },
    {
      "epoch": 1.7457008361816406e-05,
      "model_forward_time": 5.61181640625,
      "step": 18305
    },
    {
      "epoch": 1.7457008361816406e-05,
      "step": 18305,
      "training_step_time": 17.07517409324646
    },
    {
      "epoch": 1.7457962036132813e-05,
      "model_forward_time": 5.595331907272339,
      "step": 18306
    },
    {
      "epoch": 1.7457962036132813e-05,
      "step": 18306,
      "training_step_time": 17.067305088043213
    },
    {
      "epoch": 1.7458915710449217e-05,
      "model_forward_time": 5.970810413360596,
      "step": 18307
    },
    {
      "epoch": 1.7458915710449217e-05,
      "step": 18307,
      "training_step_time": 17.426310539245605
    },
    {
      "epoch": 1.7459869384765625e-05,
      "model_forward_time": 5.600165367126465,
      "step": 18308
    },
    {
      "epoch": 1.7459869384765625e-05,
      "step": 18308,
      "training_step_time": 17.08127784729004
    },
    {
      "epoch": 1.7460823059082032e-05,
      "model_forward_time": 5.605509996414185,
      "step": 18309
    },
    {
      "epoch": 1.7460823059082032e-05,
      "step": 18309,
      "training_step_time": 17.06209945678711
    },
    {
      "epoch": 1.7461776733398436e-05,
      "grad_norm": 0.6254032254219055,
      "learning_rate": 1e-05,
      "loss": 0.0624,
      "step": 18310
    },
    {
      "epoch": 1.7461776733398436e-05,
      "model_forward_time": 5.601581573486328,
      "step": 18310
    },
    {
      "dynamics_loss_avg": 0.055097569897770884,
      "epoch": 1.7461776733398436e-05,
      "step": 18310
    },
    {
      "action_loss_avg": 0.008658134401775897,
      "epoch": 1.7461776733398436e-05,
      "step": 18310
    },
    {
      "epoch": 1.7461776733398436e-05,
      "step": 18310,
      "training_step_time": 17.07926034927368
    },
    {
      "epoch": 1.7462730407714844e-05,
      "model_forward_time": 5.600189447402954,
      "step": 18311
    },
    {
      "epoch": 1.7462730407714844e-05,
      "step": 18311,
      "training_step_time": 17.066633224487305
    },
    {
      "epoch": 1.746368408203125e-05,
      "model_forward_time": 5.608476161956787,
      "step": 18312
    },
    {
      "epoch": 1.746368408203125e-05,
      "step": 18312,
      "training_step_time": 17.08185052871704
    },
    {
      "epoch": 1.7464637756347655e-05,
      "model_forward_time": 5.969650506973267,
      "step": 18313
    },
    {
      "epoch": 1.7464637756347655e-05,
      "step": 18313,
      "training_step_time": 17.45100426673889
    },
    {
      "epoch": 1.7465591430664063e-05,
      "model_forward_time": 5.599413871765137,
      "step": 18314
    },
    {
      "epoch": 1.7465591430664063e-05,
      "step": 18314,
      "training_step_time": 17.080769062042236
    },
    {
      "epoch": 1.746654510498047e-05,
      "model_forward_time": 5.614452362060547,
      "step": 18315
    },
    {
      "epoch": 1.746654510498047e-05,
      "step": 18315,
      "training_step_time": 17.097636461257935
    },
    {
      "epoch": 1.7467498779296874e-05,
      "model_forward_time": 5.610575437545776,
      "step": 18316
    },
    {
      "epoch": 1.7467498779296874e-05,
      "step": 18316,
      "training_step_time": 17.09351873397827
    },
    {
      "epoch": 1.746845245361328e-05,
      "model_forward_time": 5.602535247802734,
      "step": 18317
    },
    {
      "epoch": 1.746845245361328e-05,
      "step": 18317,
      "training_step_time": 17.051767349243164
    },
    {
      "epoch": 1.746940612792969e-05,
      "model_forward_time": 5.93748140335083,
      "step": 18318
    },
    {
      "epoch": 1.746940612792969e-05,
      "step": 18318,
      "training_step_time": 17.3700168132782
    },
    {
      "epoch": 1.7470359802246093e-05,
      "model_forward_time": 5.60113525390625,
      "step": 18319
    },
    {
      "epoch": 1.7470359802246093e-05,
      "step": 18319,
      "training_step_time": 17.081618309020996
    },
    {
      "epoch": 1.74713134765625e-05,
      "grad_norm": 0.8315067887306213,
      "learning_rate": 1e-05,
      "loss": 0.0859,
      "step": 18320
    },
    {
      "epoch": 1.74713134765625e-05,
      "model_forward_time": 5.6016526222229,
      "step": 18320
    },
    {
      "dynamics_loss_avg": 0.07355886902660132,
      "epoch": 1.74713134765625e-05,
      "step": 18320
    },
    {
      "action_loss_avg": 0.009985424799378962,
      "epoch": 1.74713134765625e-05,
      "step": 18320
    },
    {
      "epoch": 1.74713134765625e-05,
      "step": 18320,
      "training_step_time": 17.071041345596313
    },
    {
      "epoch": 1.7472267150878908e-05,
      "model_forward_time": 5.598459959030151,
      "step": 18321
    },
    {
      "epoch": 1.7472267150878908e-05,
      "step": 18321,
      "training_step_time": 17.05580186843872
    },
    {
      "epoch": 1.7473220825195312e-05,
      "model_forward_time": 5.600922346115112,
      "step": 18322
    },
    {
      "epoch": 1.7473220825195312e-05,
      "step": 18322,
      "training_step_time": 17.055343627929688
    },
    {
      "epoch": 1.747417449951172e-05,
      "model_forward_time": 5.591599464416504,
      "step": 18323
    },
    {
      "epoch": 1.747417449951172e-05,
      "step": 18323,
      "training_step_time": 17.049180507659912
    },
    {
      "epoch": 1.7475128173828127e-05,
      "model_forward_time": 5.975638151168823,
      "step": 18324
    },
    {
      "epoch": 1.7475128173828127e-05,
      "step": 18324,
      "training_step_time": 17.46062159538269
    },
    {
      "epoch": 1.747608184814453e-05,
      "model_forward_time": 5.607295751571655,
      "step": 18325
    },
    {
      "epoch": 1.747608184814453e-05,
      "step": 18325,
      "training_step_time": 17.077657461166382
    },
    {
      "epoch": 1.7477035522460938e-05,
      "model_forward_time": 5.5954790115356445,
      "step": 18326
    },
    {
      "epoch": 1.7477035522460938e-05,
      "step": 18326,
      "training_step_time": 17.038797616958618
    },
    {
      "epoch": 1.7477989196777345e-05,
      "model_forward_time": 5.585050821304321,
      "step": 18327
    },
    {
      "epoch": 1.7477989196777345e-05,
      "step": 18327,
      "training_step_time": 16.998020887374878
    },
    {
      "epoch": 1.747894287109375e-05,
      "model_forward_time": 5.581892967224121,
      "step": 18328
    },
    {
      "epoch": 1.747894287109375e-05,
      "step": 18328,
      "training_step_time": 17.027711153030396
    },
    {
      "epoch": 1.7479896545410157e-05,
      "model_forward_time": 5.581915855407715,
      "step": 18329
    },
    {
      "epoch": 1.7479896545410157e-05,
      "step": 18329,
      "training_step_time": 17.426408767700195
    },
    {
      "epoch": 1.748085021972656e-05,
      "grad_norm": 0.9982749223709106,
      "learning_rate": 1e-05,
      "loss": 0.067,
      "step": 18330
    },
    {
      "epoch": 1.748085021972656e-05,
      "model_forward_time": 5.611338376998901,
      "step": 18330
    },
    {
      "dynamics_loss_avg": 0.06518178656697274,
      "epoch": 1.748085021972656e-05,
      "step": 18330
    },
    {
      "action_loss_avg": 0.009977480117231607,
      "epoch": 1.748085021972656e-05,
      "step": 18330
    },
    {
      "epoch": 1.748085021972656e-05,
      "step": 18330,
      "training_step_time": 17.111481428146362
    },
    {
      "epoch": 1.7481803894042968e-05,
      "model_forward_time": 5.61946702003479,
      "step": 18331
    },
    {
      "epoch": 1.7481803894042968e-05,
      "step": 18331,
      "training_step_time": 17.14104461669922
    },
    {
      "epoch": 1.7482757568359376e-05,
      "model_forward_time": 5.622058153152466,
      "step": 18332
    },
    {
      "epoch": 1.7482757568359376e-05,
      "step": 18332,
      "training_step_time": 17.08202862739563
    },
    {
      "epoch": 1.748371124267578e-05,
      "model_forward_time": 5.582632303237915,
      "step": 18333
    },
    {
      "epoch": 1.748371124267578e-05,
      "step": 18333,
      "training_step_time": 16.995888471603394
    },
    {
      "epoch": 1.7484664916992187e-05,
      "model_forward_time": 5.566412687301636,
      "step": 18334
    },
    {
      "epoch": 1.7484664916992187e-05,
      "step": 18334,
      "training_step_time": 16.98400092124939
    },
    {
      "epoch": 1.7485618591308595e-05,
      "model_forward_time": 5.935231685638428,
      "step": 18335
    },
    {
      "epoch": 1.7485618591308595e-05,
      "step": 18335,
      "training_step_time": 17.355090856552124
    },
    {
      "epoch": 1.7486572265625e-05,
      "model_forward_time": 5.57233738899231,
      "step": 18336
    },
    {
      "epoch": 1.7486572265625e-05,
      "step": 18336,
      "training_step_time": 17.012388944625854
    },
    {
      "epoch": 1.7487525939941406e-05,
      "model_forward_time": 5.587603569030762,
      "step": 18337
    },
    {
      "epoch": 1.7487525939941406e-05,
      "step": 18337,
      "training_step_time": 17.004698038101196
    },
    {
      "epoch": 1.7488479614257813e-05,
      "model_forward_time": 5.5893027782440186,
      "step": 18338
    },
    {
      "epoch": 1.7488479614257813e-05,
      "step": 18338,
      "training_step_time": 17.0050528049469
    },
    {
      "epoch": 1.7489433288574217e-05,
      "model_forward_time": 5.583854913711548,
      "step": 18339
    },
    {
      "epoch": 1.7489433288574217e-05,
      "step": 18339,
      "training_step_time": 17.0237557888031
    },
    {
      "epoch": 1.7490386962890625e-05,
      "grad_norm": 0.7538793087005615,
      "learning_rate": 1e-05,
      "loss": 0.08,
      "step": 18340
    },
    {
      "epoch": 1.7490386962890625e-05,
      "model_forward_time": 5.932218074798584,
      "step": 18340
    },
    {
      "dynamics_loss_avg": 0.06667485050857067,
      "epoch": 1.7490386962890625e-05,
      "step": 18340
    },
    {
      "action_loss_avg": 0.012976239761337638,
      "epoch": 1.7490386962890625e-05,
      "step": 18340
    },
    {
      "epoch": 1.7490386962890625e-05,
      "step": 18340,
      "training_step_time": 17.36532211303711
    },
    {
      "epoch": 1.7491340637207032e-05,
      "model_forward_time": 5.597346544265747,
      "step": 18341
    },
    {
      "epoch": 1.7491340637207032e-05,
      "step": 18341,
      "training_step_time": 17.084352731704712
    },
    {
      "epoch": 1.7492294311523436e-05,
      "model_forward_time": 5.610976696014404,
      "step": 18342
    },
    {
      "epoch": 1.7492294311523436e-05,
      "step": 18342,
      "training_step_time": 17.12866497039795
    },
    {
      "epoch": 1.7493247985839844e-05,
      "model_forward_time": 5.633939027786255,
      "step": 18343
    },
    {
      "epoch": 1.7493247985839844e-05,
      "step": 18343,
      "training_step_time": 17.137356758117676
    },
    {
      "epoch": 1.749420166015625e-05,
      "model_forward_time": 5.60186505317688,
      "step": 18344
    },
    {
      "epoch": 1.749420166015625e-05,
      "step": 18344,
      "training_step_time": 17.04754877090454
    },
    {
      "epoch": 1.7495155334472655e-05,
      "model_forward_time": 5.585494756698608,
      "step": 18345
    },
    {
      "epoch": 1.7495155334472655e-05,
      "step": 18345,
      "training_step_time": 17.018268823623657
    },
    {
      "epoch": 1.7496109008789063e-05,
      "model_forward_time": 5.9479382038116455,
      "step": 18346
    },
    {
      "epoch": 1.7496109008789063e-05,
      "step": 18346,
      "training_step_time": 17.396121740341187
    },
    {
      "epoch": 1.749706268310547e-05,
      "model_forward_time": 5.610749006271362,
      "step": 18347
    },
    {
      "epoch": 1.749706268310547e-05,
      "step": 18347,
      "training_step_time": 17.105512619018555
    },
    {
      "epoch": 1.7498016357421874e-05,
      "model_forward_time": 5.626817226409912,
      "step": 18348
    },
    {
      "epoch": 1.7498016357421874e-05,
      "step": 18348,
      "training_step_time": 17.118850231170654
    },
    {
      "epoch": 1.749897003173828e-05,
      "model_forward_time": 5.598664283752441,
      "step": 18349
    },
    {
      "epoch": 1.749897003173828e-05,
      "step": 18349,
      "training_step_time": 17.017348527908325
    },
    {
      "epoch": 1.749992370605469e-05,
      "grad_norm": 0.9527962803840637,
      "learning_rate": 1e-05,
      "loss": 0.0626,
      "step": 18350
    },
    {
      "epoch": 1.749992370605469e-05,
      "model_forward_time": 5.58106255531311,
      "step": 18350
    },
    {
      "dynamics_loss_avg": 0.05086446944624186,
      "epoch": 1.749992370605469e-05,
      "step": 18350
    },
    {
      "action_loss_avg": 0.008237228950019927,
      "epoch": 1.749992370605469e-05,
      "step": 18350
    },
    {
      "epoch": 1.749992370605469e-05,
      "step": 18350,
      "training_step_time": 17.009525299072266
    },
    {
      "epoch": 1.7500877380371093e-05,
      "model_forward_time": 5.945642948150635,
      "step": 18351
    },
    {
      "epoch": 1.7500877380371093e-05,
      "step": 18351,
      "training_step_time": 17.384388208389282
    },
    {
      "epoch": 1.75018310546875e-05,
      "model_forward_time": 5.587883949279785,
      "step": 18352
    },
    {
      "epoch": 1.75018310546875e-05,
      "step": 18352,
      "training_step_time": 17.065565586090088
    },
    {
      "epoch": 1.7502784729003908e-05,
      "model_forward_time": 5.626679420471191,
      "step": 18353
    },
    {
      "epoch": 1.7502784729003908e-05,
      "step": 18353,
      "training_step_time": 17.157781839370728
    },
    {
      "epoch": 1.7503738403320312e-05,
      "model_forward_time": 5.628750562667847,
      "step": 18354
    },
    {
      "epoch": 1.7503738403320312e-05,
      "step": 18354,
      "training_step_time": 17.135936975479126
    },
    {
      "epoch": 1.750469207763672e-05,
      "model_forward_time": 5.620427370071411,
      "step": 18355
    },
    {
      "epoch": 1.750469207763672e-05,
      "step": 18355,
      "training_step_time": 17.081545114517212
    },
    {
      "epoch": 1.7505645751953127e-05,
      "model_forward_time": 5.583893299102783,
      "step": 18356
    },
    {
      "epoch": 1.7505645751953127e-05,
      "step": 18356,
      "training_step_time": 16.994965314865112
    },
    {
      "epoch": 1.750659942626953e-05,
      "model_forward_time": 5.927873134613037,
      "step": 18357
    },
    {
      "epoch": 1.750659942626953e-05,
      "step": 18357,
      "training_step_time": 17.34281349182129
    },
    {
      "epoch": 1.7507553100585938e-05,
      "model_forward_time": 5.558145999908447,
      "step": 18358
    },
    {
      "epoch": 1.7507553100585938e-05,
      "step": 18358,
      "training_step_time": 16.973445653915405
    },
    {
      "epoch": 1.7508506774902345e-05,
      "model_forward_time": 5.556807994842529,
      "step": 18359
    },
    {
      "epoch": 1.7508506774902345e-05,
      "step": 18359,
      "training_step_time": 16.993209838867188
    },
    {
      "epoch": 1.750946044921875e-05,
      "grad_norm": 0.3478822708129883,
      "learning_rate": 1e-05,
      "loss": 0.0565,
      "step": 18360
    },
    {
      "epoch": 1.750946044921875e-05,
      "model_forward_time": 5.577819108963013,
      "step": 18360
    },
    {
      "dynamics_loss_avg": 0.043813216127455236,
      "epoch": 1.750946044921875e-05,
      "step": 18360
    },
    {
      "action_loss_avg": 0.005942667671479285,
      "epoch": 1.750946044921875e-05,
      "step": 18360
    },
    {
      "epoch": 1.750946044921875e-05,
      "step": 18360,
      "training_step_time": 16.993244171142578
    },
    {
      "epoch": 1.7510414123535157e-05,
      "model_forward_time": 5.573200225830078,
      "step": 18361
    },
    {
      "epoch": 1.7510414123535157e-05,
      "step": 18361,
      "training_step_time": 16.98587727546692
    },
    {
      "epoch": 1.751136779785156e-05,
      "model_forward_time": 5.581458806991577,
      "step": 18362
    },
    {
      "epoch": 1.751136779785156e-05,
      "step": 18362,
      "training_step_time": 17.40414524078369
    },
    {
      "epoch": 1.751232147216797e-05,
      "model_forward_time": 5.5847296714782715,
      "step": 18363
    },
    {
      "epoch": 1.751232147216797e-05,
      "step": 18363,
      "training_step_time": 17.026615858078003
    },
    {
      "epoch": 1.7513275146484376e-05,
      "model_forward_time": 5.5939414501190186,
      "step": 18364
    },
    {
      "epoch": 1.7513275146484376e-05,
      "step": 18364,
      "training_step_time": 17.077443599700928
    },
    {
      "epoch": 1.751422882080078e-05,
      "model_forward_time": 5.639077663421631,
      "step": 18365
    },
    {
      "epoch": 1.751422882080078e-05,
      "step": 18365,
      "training_step_time": 17.151371240615845
    },
    {
      "epoch": 1.7515182495117187e-05,
      "model_forward_time": 5.622054815292358,
      "step": 18366
    },
    {
      "epoch": 1.7515182495117187e-05,
      "step": 18366,
      "training_step_time": 17.129035711288452
    },
    {
      "epoch": 1.7516136169433595e-05,
      "model_forward_time": 5.623057842254639,
      "step": 18367
    },
    {
      "epoch": 1.7516136169433595e-05,
      "step": 18367,
      "training_step_time": 17.107420444488525
    },
    {
      "epoch": 1.751708984375e-05,
      "model_forward_time": 5.9550461769104,
      "step": 18368
    },
    {
      "epoch": 1.751708984375e-05,
      "step": 18368,
      "training_step_time": 17.35847806930542
    },
    {
      "epoch": 1.7518043518066406e-05,
      "model_forward_time": 5.57329535484314,
      "step": 18369
    },
    {
      "epoch": 1.7518043518066406e-05,
      "step": 18369,
      "training_step_time": 16.993494749069214
    },
    {
      "epoch": 1.7518997192382813e-05,
      "grad_norm": 0.8641538619995117,
      "learning_rate": 1e-05,
      "loss": 0.058,
      "step": 18370
    },
    {
      "epoch": 1.7518997192382813e-05,
      "model_forward_time": 5.564465045928955,
      "step": 18370
    },
    {
      "dynamics_loss_avg": 0.0555860249325633,
      "epoch": 1.7518997192382813e-05,
      "step": 18370
    },
    {
      "action_loss_avg": 0.007125843968242407,
      "epoch": 1.7518997192382813e-05,
      "step": 18370
    },
    {
      "epoch": 1.7518997192382813e-05,
      "step": 18370,
      "training_step_time": 17.00008797645569
    },
    {
      "epoch": 1.7519950866699217e-05,
      "model_forward_time": 5.575458526611328,
      "step": 18371
    },
    {
      "epoch": 1.7519950866699217e-05,
      "step": 18371,
      "training_step_time": 16.99912738800049
    },
    {
      "epoch": 1.7520904541015625e-05,
      "model_forward_time": 5.586504220962524,
      "step": 18372
    },
    {
      "epoch": 1.7520904541015625e-05,
      "step": 18372,
      "training_step_time": 17.020142316818237
    },
    {
      "epoch": 1.7521858215332032e-05,
      "model_forward_time": 5.9396138191223145,
      "step": 18373
    },
    {
      "epoch": 1.7521858215332032e-05,
      "step": 18373,
      "training_step_time": 17.411052703857422
    },
    {
      "epoch": 1.7522811889648436e-05,
      "model_forward_time": 5.604234457015991,
      "step": 18374
    },
    {
      "epoch": 1.7522811889648436e-05,
      "step": 18374,
      "training_step_time": 17.063766956329346
    },
    {
      "epoch": 1.7523765563964844e-05,
      "model_forward_time": 5.5987021923065186,
      "step": 18375
    },
    {
      "epoch": 1.7523765563964844e-05,
      "step": 18375,
      "training_step_time": 17.067031621932983
    },
    {
      "epoch": 1.752471923828125e-05,
      "model_forward_time": 5.609199285507202,
      "step": 18376
    },
    {
      "epoch": 1.752471923828125e-05,
      "step": 18376,
      "training_step_time": 17.055319786071777
    },
    {
      "epoch": 1.7525672912597655e-05,
      "model_forward_time": 5.595284938812256,
      "step": 18377
    },
    {
      "epoch": 1.7525672912597655e-05,
      "step": 18377,
      "training_step_time": 17.04490351676941
    },
    {
      "epoch": 1.7526626586914063e-05,
      "model_forward_time": 5.592199087142944,
      "step": 18378
    },
    {
      "epoch": 1.7526626586914063e-05,
      "step": 18378,
      "training_step_time": 17.042783737182617
    },
    {
      "epoch": 1.752758026123047e-05,
      "model_forward_time": 5.986630916595459,
      "step": 18379
    },
    {
      "epoch": 1.752758026123047e-05,
      "step": 18379,
      "training_step_time": 17.444252490997314
    },
    {
      "epoch": 1.7528533935546874e-05,
      "grad_norm": 0.8492880463600159,
      "learning_rate": 1e-05,
      "loss": 0.0569,
      "step": 18380
    },
    {
      "epoch": 1.7528533935546874e-05,
      "model_forward_time": 5.6001622676849365,
      "step": 18380
    },
    {
      "dynamics_loss_avg": 0.048451508209109304,
      "epoch": 1.7528533935546874e-05,
      "step": 18380
    },
    {
      "action_loss_avg": 0.007128298305906356,
      "epoch": 1.7528533935546874e-05,
      "step": 18380
    },
    {
      "epoch": 1.7528533935546874e-05,
      "step": 18380,
      "training_step_time": 17.056151628494263
    },
    {
      "epoch": 1.752948760986328e-05,
      "model_forward_time": 5.596216678619385,
      "step": 18381
    },
    {
      "epoch": 1.752948760986328e-05,
      "step": 18381,
      "training_step_time": 17.036849975585938
    },
    {
      "epoch": 1.753044128417969e-05,
      "model_forward_time": 5.589566707611084,
      "step": 18382
    },
    {
      "epoch": 1.753044128417969e-05,
      "step": 18382,
      "training_step_time": 17.048367261886597
    },
    {
      "epoch": 1.7531394958496093e-05,
      "model_forward_time": 5.602117538452148,
      "step": 18383
    },
    {
      "epoch": 1.7531394958496093e-05,
      "step": 18383,
      "training_step_time": 17.083938598632812
    },
    {
      "epoch": 1.75323486328125e-05,
      "model_forward_time": 5.966775894165039,
      "step": 18384
    },
    {
      "epoch": 1.75323486328125e-05,
      "step": 18384,
      "training_step_time": 17.470158576965332
    },
    {
      "epoch": 1.7533302307128908e-05,
      "model_forward_time": 5.62071418762207,
      "step": 18385
    },
    {
      "epoch": 1.7533302307128908e-05,
      "step": 18385,
      "training_step_time": 17.141009092330933
    },
    {
      "epoch": 1.7534255981445312e-05,
      "model_forward_time": 5.62808084487915,
      "step": 18386
    },
    {
      "epoch": 1.7534255981445312e-05,
      "step": 18386,
      "training_step_time": 17.100945949554443
    },
    {
      "epoch": 1.753520965576172e-05,
      "model_forward_time": 5.592683792114258,
      "step": 18387
    },
    {
      "epoch": 1.753520965576172e-05,
      "step": 18387,
      "training_step_time": 17.01269245147705
    },
    {
      "epoch": 1.7536163330078127e-05,
      "model_forward_time": 5.56621789932251,
      "step": 18388
    },
    {
      "epoch": 1.7536163330078127e-05,
      "step": 18388,
      "training_step_time": 16.97601819038391
    },
    {
      "epoch": 1.753711700439453e-05,
      "model_forward_time": 5.5601959228515625,
      "step": 18389
    },
    {
      "epoch": 1.753711700439453e-05,
      "step": 18389,
      "training_step_time": 16.976608276367188
    },
    {
      "epoch": 1.7538070678710938e-05,
      "grad_norm": 0.9515056014060974,
      "learning_rate": 1e-05,
      "loss": 0.0603,
      "step": 18390
    },
    {
      "epoch": 1.7538070678710938e-05,
      "model_forward_time": 5.927573204040527,
      "step": 18390
    },
    {
      "dynamics_loss_avg": 0.05334401959553361,
      "epoch": 1.7538070678710938e-05,
      "step": 18390
    },
    {
      "action_loss_avg": 0.007876446889713406,
      "epoch": 1.7538070678710938e-05,
      "step": 18390
    },
    {
      "epoch": 1.7538070678710938e-05,
      "step": 18390,
      "training_step_time": 17.34669327735901
    },
    {
      "epoch": 1.7539024353027342e-05,
      "model_forward_time": 5.575404405593872,
      "step": 18391
    },
    {
      "epoch": 1.7539024353027342e-05,
      "step": 18391,
      "training_step_time": 16.994723320007324
    },
    {
      "epoch": 1.753997802734375e-05,
      "model_forward_time": 5.579405307769775,
      "step": 18392
    },
    {
      "epoch": 1.753997802734375e-05,
      "step": 18392,
      "training_step_time": 17.00075650215149
    },
    {
      "epoch": 1.7540931701660157e-05,
      "model_forward_time": 5.580320119857788,
      "step": 18393
    },
    {
      "epoch": 1.7540931701660157e-05,
      "step": 18393,
      "training_step_time": 17.0291485786438
    },
    {
      "epoch": 1.754188537597656e-05,
      "model_forward_time": 5.576746702194214,
      "step": 18394
    },
    {
      "epoch": 1.754188537597656e-05,
      "step": 18394,
      "training_step_time": 17.006868600845337
    },
    {
      "epoch": 1.754283905029297e-05,
      "model_forward_time": 5.594789981842041,
      "step": 18395
    },
    {
      "epoch": 1.754283905029297e-05,
      "step": 18395,
      "training_step_time": 17.436440467834473
    },
    {
      "epoch": 1.7543792724609376e-05,
      "model_forward_time": 5.616389513015747,
      "step": 18396
    },
    {
      "epoch": 1.7543792724609376e-05,
      "step": 18396,
      "training_step_time": 17.11547613143921
    },
    {
      "epoch": 1.754474639892578e-05,
      "model_forward_time": 5.610785961151123,
      "step": 18397
    },
    {
      "epoch": 1.754474639892578e-05,
      "step": 18397,
      "training_step_time": 17.098938703536987
    },
    {
      "epoch": 1.7545700073242187e-05,
      "model_forward_time": 5.607875108718872,
      "step": 18398
    },
    {
      "epoch": 1.7545700073242187e-05,
      "step": 18398,
      "training_step_time": 17.08928918838501
    },
    {
      "epoch": 1.7546653747558595e-05,
      "model_forward_time": 5.598608732223511,
      "step": 18399
    },
    {
      "epoch": 1.7546653747558595e-05,
      "step": 18399,
      "training_step_time": 17.076566457748413
    },
    {
      "epoch": 1.7547607421875e-05,
      "grad_norm": 0.5751804113388062,
      "learning_rate": 1e-05,
      "loss": 0.0659,
      "step": 18400
    },
    {
      "epoch": 1.7547607421875e-05,
      "model_forward_time": 5.61884617805481,
      "step": 18400
    },
    {
      "dynamics_loss_avg": 0.06247686892747879,
      "epoch": 1.7547607421875e-05,
      "step": 18400
    },
    {
      "action_loss_avg": 0.00935711667407304,
      "epoch": 1.7547607421875e-05,
      "step": 18400
    },
    {
      "epoch": 1.7547607421875e-05,
      "step": 18400,
      "training_step_time": 17.08370852470398
    },
    {
      "epoch": 1.7548561096191406e-05,
      "model_forward_time": 5.968640089035034,
      "step": 18401
    },
    {
      "epoch": 1.7548561096191406e-05,
      "step": 18401,
      "training_step_time": 17.438013315200806
    },
    {
      "epoch": 1.7549514770507813e-05,
      "model_forward_time": 5.603769302368164,
      "step": 18402
    },
    {
      "epoch": 1.7549514770507813e-05,
      "step": 18402,
      "training_step_time": 17.075562953948975
    },
    {
      "epoch": 1.7550468444824218e-05,
      "model_forward_time": 5.590576171875,
      "step": 18403
    },
    {
      "epoch": 1.7550468444824218e-05,
      "step": 18403,
      "training_step_time": 17.039778232574463
    },
    {
      "epoch": 1.7551422119140625e-05,
      "model_forward_time": 5.600383281707764,
      "step": 18404
    },
    {
      "epoch": 1.7551422119140625e-05,
      "step": 18404,
      "training_step_time": 17.057385206222534
    },
    {
      "epoch": 1.7552375793457032e-05,
      "model_forward_time": 5.607300519943237,
      "step": 18405
    },
    {
      "epoch": 1.7552375793457032e-05,
      "step": 18405,
      "training_step_time": 17.09501814842224
    },
    {
      "epoch": 1.7553329467773436e-05,
      "model_forward_time": 5.954039812088013,
      "step": 18406
    },
    {
      "epoch": 1.7553329467773436e-05,
      "step": 18406,
      "training_step_time": 17.42663049697876
    },
    {
      "epoch": 1.7554283142089844e-05,
      "model_forward_time": 5.599504709243774,
      "step": 18407
    },
    {
      "epoch": 1.7554283142089844e-05,
      "step": 18407,
      "training_step_time": 17.055135488510132
    },
    {
      "epoch": 1.755523681640625e-05,
      "model_forward_time": 5.603612184524536,
      "step": 18408
    },
    {
      "epoch": 1.755523681640625e-05,
      "step": 18408,
      "training_step_time": 17.047921657562256
    },
    {
      "epoch": 1.7556190490722655e-05,
      "model_forward_time": 5.58270525932312,
      "step": 18409
    },
    {
      "epoch": 1.7556190490722655e-05,
      "step": 18409,
      "training_step_time": 17.04799747467041
    },
    {
      "epoch": 1.7557144165039063e-05,
      "grad_norm": 0.24428775906562805,
      "learning_rate": 1e-05,
      "loss": 0.0564,
      "step": 18410
    },
    {
      "epoch": 1.7557144165039063e-05,
      "model_forward_time": 5.6197779178619385,
      "step": 18410
    },
    {
      "dynamics_loss_avg": 0.04171811724081635,
      "epoch": 1.7557144165039063e-05,
      "step": 18410
    },
    {
      "action_loss_avg": 0.00646368587622419,
      "epoch": 1.7557144165039063e-05,
      "step": 18410
    },
    {
      "epoch": 1.7557144165039063e-05,
      "step": 18410,
      "training_step_time": 17.11596441268921
    },
    {
      "epoch": 1.755809783935547e-05,
      "model_forward_time": 5.6149280071258545,
      "step": 18411
    },
    {
      "epoch": 1.755809783935547e-05,
      "step": 18411,
      "training_step_time": 17.08405303955078
    },
    {
      "epoch": 1.7559051513671874e-05,
      "model_forward_time": 5.958737134933472,
      "step": 18412
    },
    {
      "epoch": 1.7559051513671874e-05,
      "step": 18412,
      "training_step_time": 17.358286380767822
    },
    {
      "epoch": 1.756000518798828e-05,
      "model_forward_time": 5.593175411224365,
      "step": 18413
    },
    {
      "epoch": 1.756000518798828e-05,
      "step": 18413,
      "training_step_time": 17.046329259872437
    },
    {
      "epoch": 1.756095886230469e-05,
      "model_forward_time": 5.5999062061309814,
      "step": 18414
    },
    {
      "epoch": 1.756095886230469e-05,
      "step": 18414,
      "training_step_time": 17.09058165550232
    },
    {
      "epoch": 1.7561912536621093e-05,
      "model_forward_time": 5.6227593421936035,
      "step": 18415
    },
    {
      "epoch": 1.7561912536621093e-05,
      "step": 18415,
      "training_step_time": 17.12946844100952
    },
    {
      "epoch": 1.75628662109375e-05,
      "model_forward_time": 5.631271600723267,
      "step": 18416
    },
    {
      "epoch": 1.75628662109375e-05,
      "step": 18416,
      "training_step_time": 17.129544258117676
    },
    {
      "epoch": 1.7563819885253908e-05,
      "model_forward_time": 5.955286264419556,
      "step": 18417
    },
    {
      "epoch": 1.7563819885253908e-05,
      "step": 18417,
      "training_step_time": 17.41891860961914
    },
    {
      "epoch": 1.7564773559570312e-05,
      "model_forward_time": 5.59820294380188,
      "step": 18418
    },
    {
      "epoch": 1.7564773559570312e-05,
      "step": 18418,
      "training_step_time": 17.03436517715454
    },
    {
      "epoch": 1.756572723388672e-05,
      "model_forward_time": 5.609144449234009,
      "step": 18419
    },
    {
      "epoch": 1.756572723388672e-05,
      "step": 18419,
      "training_step_time": 17.041329383850098
    },
    {
      "epoch": 1.7566680908203127e-05,
      "grad_norm": 0.5658620595932007,
      "learning_rate": 1e-05,
      "loss": 0.0653,
      "step": 18420
    },
    {
      "epoch": 1.7566680908203127e-05,
      "model_forward_time": 5.600549697875977,
      "step": 18420
    },
    {
      "dynamics_loss_avg": 0.059095315635204315,
      "epoch": 1.7566680908203127e-05,
      "step": 18420
    },
    {
      "action_loss_avg": 0.007438961416482925,
      "epoch": 1.7566680908203127e-05,
      "step": 18420
    },
    {
      "epoch": 1.7566680908203127e-05,
      "step": 18420,
      "training_step_time": 17.07829213142395
    },
    {
      "epoch": 1.756763458251953e-05,
      "model_forward_time": 5.626765012741089,
      "step": 18421
    },
    {
      "epoch": 1.756763458251953e-05,
      "step": 18421,
      "training_step_time": 17.10853600502014
    },
    {
      "epoch": 1.7568588256835938e-05,
      "model_forward_time": 5.596782207489014,
      "step": 18422
    },
    {
      "epoch": 1.7568588256835938e-05,
      "step": 18422,
      "training_step_time": 17.07732629776001
    },
    {
      "epoch": 1.7569541931152342e-05,
      "model_forward_time": 6.142910480499268,
      "step": 18423
    },
    {
      "epoch": 1.7569541931152342e-05,
      "step": 18423,
      "training_step_time": 17.650195360183716
    },
    {
      "epoch": 1.757049560546875e-05,
      "model_forward_time": 5.639669179916382,
      "step": 18424
    },
    {
      "epoch": 1.757049560546875e-05,
      "step": 18424,
      "training_step_time": 17.152685165405273
    },
    {
      "epoch": 1.7571449279785157e-05,
      "model_forward_time": 5.639347314834595,
      "step": 18425
    },
    {
      "epoch": 1.7571449279785157e-05,
      "step": 18425,
      "training_step_time": 17.153589963912964
    },
    {
      "epoch": 1.757240295410156e-05,
      "model_forward_time": 5.621633052825928,
      "step": 18426
    },
    {
      "epoch": 1.757240295410156e-05,
      "step": 18426,
      "training_step_time": 17.135332107543945
    },
    {
      "epoch": 1.757335662841797e-05,
      "model_forward_time": 5.626313209533691,
      "step": 18427
    },
    {
      "epoch": 1.757335662841797e-05,
      "step": 18427,
      "training_step_time": 17.097676753997803
    },
    {
      "epoch": 1.7574310302734376e-05,
      "model_forward_time": 5.6109161376953125,
      "step": 18428
    },
    {
      "epoch": 1.7574310302734376e-05,
      "step": 18428,
      "training_step_time": 17.47565531730652
    },
    {
      "epoch": 1.757526397705078e-05,
      "model_forward_time": 5.598792552947998,
      "step": 18429
    },
    {
      "epoch": 1.757526397705078e-05,
      "step": 18429,
      "training_step_time": 17.0699143409729
    },
    {
      "epoch": 1.7576217651367187e-05,
      "grad_norm": 1.339253306388855,
      "learning_rate": 1e-05,
      "loss": 0.0655,
      "step": 18430
    },
    {
      "epoch": 1.7576217651367187e-05,
      "model_forward_time": 5.593021869659424,
      "step": 18430
    },
    {
      "dynamics_loss_avg": 0.06273077912628651,
      "epoch": 1.7576217651367187e-05,
      "step": 18430
    },
    {
      "action_loss_avg": 0.010045121749863028,
      "epoch": 1.7576217651367187e-05,
      "step": 18430
    },
    {
      "epoch": 1.7576217651367187e-05,
      "step": 18430,
      "training_step_time": 17.042257070541382
    },
    {
      "epoch": 1.7577171325683595e-05,
      "model_forward_time": 5.584532022476196,
      "step": 18431
    },
    {
      "epoch": 1.7577171325683595e-05,
      "step": 18431,
      "training_step_time": 17.0311758518219
    },
    {
      "epoch": 1.7578125e-05,
      "model_forward_time": 5.5790910720825195,
      "step": 18432
    },
    {
      "epoch": 1.7578125e-05,
      "step": 18432,
      "training_step_time": 17.044231176376343
    },
    {
      "epoch": 1.7579078674316406e-05,
      "model_forward_time": 5.600306987762451,
      "step": 18433
    },
    {
      "epoch": 1.7579078674316406e-05,
      "step": 18433,
      "training_step_time": 17.094046115875244
    },
    {
      "epoch": 1.7580032348632814e-05,
      "model_forward_time": 6.052363634109497,
      "step": 18434
    },
    {
      "epoch": 1.7580032348632814e-05,
      "step": 18434,
      "training_step_time": 17.554668188095093
    },
    {
      "epoch": 1.7580986022949218e-05,
      "model_forward_time": 5.623144865036011,
      "step": 18435
    },
    {
      "epoch": 1.7580986022949218e-05,
      "step": 18435,
      "training_step_time": 17.10678195953369
    },
    {
      "epoch": 1.7581939697265625e-05,
      "model_forward_time": 5.601159334182739,
      "step": 18436
    },
    {
      "epoch": 1.7581939697265625e-05,
      "step": 18436,
      "training_step_time": 17.05198645591736
    },
    {
      "epoch": 1.7582893371582032e-05,
      "model_forward_time": 5.587968349456787,
      "step": 18437
    },
    {
      "epoch": 1.7582893371582032e-05,
      "step": 18437,
      "training_step_time": 17.005383491516113
    },
    {
      "epoch": 1.7583847045898436e-05,
      "model_forward_time": 5.585332870483398,
      "step": 18438
    },
    {
      "epoch": 1.7583847045898436e-05,
      "step": 18438,
      "training_step_time": 17.021929264068604
    },
    {
      "epoch": 1.7584800720214844e-05,
      "model_forward_time": 5.955783843994141,
      "step": 18439
    },
    {
      "epoch": 1.7584800720214844e-05,
      "step": 18439,
      "training_step_time": 17.37578511238098
    },
    {
      "epoch": 1.758575439453125e-05,
      "grad_norm": 0.879616379737854,
      "learning_rate": 1e-05,
      "loss": 0.0669,
      "step": 18440
    },
    {
      "epoch": 1.758575439453125e-05,
      "model_forward_time": 5.5711822509765625,
      "step": 18440
    },
    {
      "dynamics_loss_avg": 0.05365505665540695,
      "epoch": 1.758575439453125e-05,
      "step": 18440
    },
    {
      "action_loss_avg": 0.009064975194633007,
      "epoch": 1.758575439453125e-05,
      "step": 18440
    },
    {
      "epoch": 1.758575439453125e-05,
      "step": 18440,
      "training_step_time": 16.974154472351074
    },
    {
      "epoch": 1.7586708068847655e-05,
      "model_forward_time": 5.587605714797974,
      "step": 18441
    },
    {
      "epoch": 1.7586708068847655e-05,
      "step": 18441,
      "training_step_time": 17.04226064682007
    },
    {
      "epoch": 1.7587661743164063e-05,
      "model_forward_time": 5.603795289993286,
      "step": 18442
    },
    {
      "epoch": 1.7587661743164063e-05,
      "step": 18442,
      "training_step_time": 17.08368420600891
    },
    {
      "epoch": 1.758861541748047e-05,
      "model_forward_time": 5.614336967468262,
      "step": 18443
    },
    {
      "epoch": 1.758861541748047e-05,
      "step": 18443,
      "training_step_time": 17.1441547870636
    },
    {
      "epoch": 1.7589569091796874e-05,
      "model_forward_time": 5.6206159591674805,
      "step": 18444
    },
    {
      "epoch": 1.7589569091796874e-05,
      "step": 18444,
      "training_step_time": 17.120070934295654
    },
    {
      "epoch": 1.759052276611328e-05,
      "model_forward_time": 5.97263765335083,
      "step": 18445
    },
    {
      "epoch": 1.759052276611328e-05,
      "step": 18445,
      "training_step_time": 17.448469400405884
    },
    {
      "epoch": 1.759147644042969e-05,
      "model_forward_time": 5.601890802383423,
      "step": 18446
    },
    {
      "epoch": 1.759147644042969e-05,
      "step": 18446,
      "training_step_time": 17.055371046066284
    },
    {
      "epoch": 1.7592430114746093e-05,
      "model_forward_time": 5.601348161697388,
      "step": 18447
    },
    {
      "epoch": 1.7592430114746093e-05,
      "step": 18447,
      "training_step_time": 17.07446050643921
    },
    {
      "epoch": 1.75933837890625e-05,
      "model_forward_time": 5.605676174163818,
      "step": 18448
    },
    {
      "epoch": 1.75933837890625e-05,
      "step": 18448,
      "training_step_time": 17.091686964035034
    },
    {
      "epoch": 1.7594337463378908e-05,
      "model_forward_time": 5.6072962284088135,
      "step": 18449
    },
    {
      "epoch": 1.7594337463378908e-05,
      "step": 18449,
      "training_step_time": 17.07711958885193
    },
    {
      "epoch": 1.7595291137695312e-05,
      "grad_norm": 0.482273131608963,
      "learning_rate": 1e-05,
      "loss": 0.0614,
      "step": 18450
    },
    {
      "epoch": 1.7595291137695312e-05,
      "model_forward_time": 5.958032608032227,
      "step": 18450
    },
    {
      "dynamics_loss_avg": 0.05806705616414547,
      "epoch": 1.7595291137695312e-05,
      "step": 18450
    },
    {
      "action_loss_avg": 0.009525599633343517,
      "epoch": 1.7595291137695312e-05,
      "step": 18450
    },
    {
      "epoch": 1.7595291137695312e-05,
      "step": 18450,
      "training_step_time": 17.409733057022095
    },
    {
      "epoch": 1.759624481201172e-05,
      "model_forward_time": 5.585843563079834,
      "step": 18451
    },
    {
      "epoch": 1.759624481201172e-05,
      "step": 18451,
      "training_step_time": 17.04024910926819
    },
    {
      "epoch": 1.7597198486328127e-05,
      "model_forward_time": 5.5973961353302,
      "step": 18452
    },
    {
      "epoch": 1.7597198486328127e-05,
      "step": 18452,
      "training_step_time": 17.08787178993225
    },
    {
      "epoch": 1.759815216064453e-05,
      "model_forward_time": 5.6190619468688965,
      "step": 18453
    },
    {
      "epoch": 1.759815216064453e-05,
      "step": 18453,
      "training_step_time": 17.14476752281189
    },
    {
      "epoch": 1.7599105834960938e-05,
      "model_forward_time": 5.622036933898926,
      "step": 18454
    },
    {
      "epoch": 1.7599105834960938e-05,
      "step": 18454,
      "training_step_time": 17.1351215839386
    },
    {
      "epoch": 1.7600059509277342e-05,
      "model_forward_time": 5.599269866943359,
      "step": 18455
    },
    {
      "epoch": 1.7600059509277342e-05,
      "step": 18455,
      "training_step_time": 17.05902624130249
    },
    {
      "epoch": 1.760101318359375e-05,
      "model_forward_time": 5.98012638092041,
      "step": 18456
    },
    {
      "epoch": 1.760101318359375e-05,
      "step": 18456,
      "training_step_time": 17.39579463005066
    },
    {
      "epoch": 1.7601966857910157e-05,
      "model_forward_time": 5.588356256484985,
      "step": 18457
    },
    {
      "epoch": 1.7601966857910157e-05,
      "step": 18457,
      "training_step_time": 17.027410984039307
    },
    {
      "epoch": 1.760292053222656e-05,
      "model_forward_time": 5.574485540390015,
      "step": 18458
    },
    {
      "epoch": 1.760292053222656e-05,
      "step": 18458,
      "training_step_time": 17.00560998916626
    },
    {
      "epoch": 1.760387420654297e-05,
      "model_forward_time": 5.579087257385254,
      "step": 18459
    },
    {
      "epoch": 1.760387420654297e-05,
      "step": 18459,
      "training_step_time": 17.00698757171631
    },
    {
      "epoch": 1.7604827880859376e-05,
      "grad_norm": 1.576757550239563,
      "learning_rate": 1e-05,
      "loss": 0.0812,
      "step": 18460
    },
    {
      "epoch": 1.7604827880859376e-05,
      "model_forward_time": 5.598486661911011,
      "step": 18460
    },
    {
      "dynamics_loss_avg": 0.05760294534265995,
      "epoch": 1.7604827880859376e-05,
      "step": 18460
    },
    {
      "action_loss_avg": 0.011292298696935177,
      "epoch": 1.7604827880859376e-05,
      "step": 18460
    },
    {
      "epoch": 1.7604827880859376e-05,
      "step": 18460,
      "training_step_time": 17.06860065460205
    },
    {
      "epoch": 1.760578155517578e-05,
      "model_forward_time": 5.60478949546814,
      "step": 18461
    },
    {
      "epoch": 1.760578155517578e-05,
      "step": 18461,
      "training_step_time": 17.44821286201477
    },
    {
      "epoch": 1.7606735229492187e-05,
      "model_forward_time": 5.615879774093628,
      "step": 18462
    },
    {
      "epoch": 1.7606735229492187e-05,
      "step": 18462,
      "training_step_time": 17.08181929588318
    },
    {
      "epoch": 1.7607688903808595e-05,
      "model_forward_time": 5.596770524978638,
      "step": 18463
    },
    {
      "epoch": 1.7607688903808595e-05,
      "step": 18463,
      "training_step_time": 17.03987169265747
    },
    {
      "epoch": 1.7608642578125e-05,
      "model_forward_time": 5.581624507904053,
      "step": 18464
    },
    {
      "epoch": 1.7608642578125e-05,
      "step": 18464,
      "training_step_time": 17.016149044036865
    },
    {
      "epoch": 1.7609596252441406e-05,
      "model_forward_time": 5.582441806793213,
      "step": 18465
    },
    {
      "epoch": 1.7609596252441406e-05,
      "step": 18465,
      "training_step_time": 17.0277578830719
    },
    {
      "epoch": 1.7610549926757814e-05,
      "model_forward_time": 5.586510181427002,
      "step": 18466
    },
    {
      "epoch": 1.7610549926757814e-05,
      "step": 18466,
      "training_step_time": 17.048726797103882
    },
    {
      "epoch": 1.7611503601074218e-05,
      "model_forward_time": 5.970317363739014,
      "step": 18467
    },
    {
      "epoch": 1.7611503601074218e-05,
      "step": 18467,
      "training_step_time": 17.428060293197632
    },
    {
      "epoch": 1.7612457275390625e-05,
      "model_forward_time": 5.599515438079834,
      "step": 18468
    },
    {
      "epoch": 1.7612457275390625e-05,
      "step": 18468,
      "training_step_time": 17.060973644256592
    },
    {
      "epoch": 1.7613410949707032e-05,
      "model_forward_time": 5.605438470840454,
      "step": 18469
    },
    {
      "epoch": 1.7613410949707032e-05,
      "step": 18469,
      "training_step_time": 17.070176601409912
    },
    {
      "epoch": 1.7614364624023436e-05,
      "grad_norm": 0.5916491746902466,
      "learning_rate": 1e-05,
      "loss": 0.0675,
      "step": 18470
    },
    {
      "epoch": 1.7614364624023436e-05,
      "model_forward_time": 5.604162693023682,
      "step": 18470
    },
    {
      "dynamics_loss_avg": 0.05630544368177652,
      "epoch": 1.7614364624023436e-05,
      "step": 18470
    },
    {
      "action_loss_avg": 0.011401062598451972,
      "epoch": 1.7614364624023436e-05,
      "step": 18470
    },
    {
      "epoch": 1.7614364624023436e-05,
      "step": 18470,
      "training_step_time": 17.049317836761475
    },
    {
      "epoch": 1.7615318298339844e-05,
      "model_forward_time": 5.582821846008301,
      "step": 18471
    },
    {
      "epoch": 1.7615318298339844e-05,
      "step": 18471,
      "training_step_time": 17.015844345092773
    },
    {
      "epoch": 1.761627197265625e-05,
      "model_forward_time": 5.939723491668701,
      "step": 18472
    },
    {
      "epoch": 1.761627197265625e-05,
      "step": 18472,
      "training_step_time": 17.381584882736206
    },
    {
      "epoch": 1.7617225646972655e-05,
      "model_forward_time": 5.606104135513306,
      "step": 18473
    },
    {
      "epoch": 1.7617225646972655e-05,
      "step": 18473,
      "training_step_time": 17.07002592086792
    },
    {
      "epoch": 1.7618179321289063e-05,
      "model_forward_time": 5.619176149368286,
      "step": 18474
    },
    {
      "epoch": 1.7618179321289063e-05,
      "step": 18474,
      "training_step_time": 17.102339029312134
    },
    {
      "epoch": 1.761913299560547e-05,
      "model_forward_time": 5.5991597175598145,
      "step": 18475
    },
    {
      "epoch": 1.761913299560547e-05,
      "step": 18475,
      "training_step_time": 17.068843841552734
    },
    {
      "epoch": 1.7620086669921874e-05,
      "model_forward_time": 5.601672172546387,
      "step": 18476
    },
    {
      "epoch": 1.7620086669921874e-05,
      "step": 18476,
      "training_step_time": 17.05616784095764
    },
    {
      "epoch": 1.762104034423828e-05,
      "model_forward_time": 5.598398923873901,
      "step": 18477
    },
    {
      "epoch": 1.762104034423828e-05,
      "step": 18477,
      "training_step_time": 17.047219276428223
    },
    {
      "epoch": 1.762199401855469e-05,
      "model_forward_time": 5.974538326263428,
      "step": 18478
    },
    {
      "epoch": 1.762199401855469e-05,
      "step": 18478,
      "training_step_time": 17.386322736740112
    },
    {
      "epoch": 1.7622947692871093e-05,
      "model_forward_time": 5.58767294883728,
      "step": 18479
    },
    {
      "epoch": 1.7622947692871093e-05,
      "step": 18479,
      "training_step_time": 17.014626264572144
    },
    {
      "epoch": 1.76239013671875e-05,
      "grad_norm": 0.8752352595329285,
      "learning_rate": 1e-05,
      "loss": 0.0752,
      "step": 18480
    },
    {
      "epoch": 1.76239013671875e-05,
      "model_forward_time": 5.597101449966431,
      "step": 18480
    },
    {
      "dynamics_loss_avg": 0.06896825134754181,
      "epoch": 1.76239013671875e-05,
      "step": 18480
    },
    {
      "action_loss_avg": 0.01113652833737433,
      "epoch": 1.76239013671875e-05,
      "step": 18480
    },
    {
      "epoch": 1.76239013671875e-05,
      "step": 18480,
      "training_step_time": 17.04536485671997
    },
    {
      "epoch": 1.7624855041503908e-05,
      "model_forward_time": 5.613509654998779,
      "step": 18481
    },
    {
      "epoch": 1.7624855041503908e-05,
      "step": 18481,
      "training_step_time": 17.111093044281006
    },
    {
      "epoch": 1.7625808715820312e-05,
      "model_forward_time": 5.61536431312561,
      "step": 18482
    },
    {
      "epoch": 1.7625808715820312e-05,
      "step": 18482,
      "training_step_time": 17.11354899406433
    },
    {
      "epoch": 1.762676239013672e-05,
      "model_forward_time": 5.9822986125946045,
      "step": 18483
    },
    {
      "epoch": 1.762676239013672e-05,
      "step": 18483,
      "training_step_time": 17.44642448425293
    },
    {
      "epoch": 1.7627716064453127e-05,
      "model_forward_time": 5.603148460388184,
      "step": 18484
    },
    {
      "epoch": 1.7627716064453127e-05,
      "step": 18484,
      "training_step_time": 17.079331398010254
    },
    {
      "epoch": 1.762866973876953e-05,
      "model_forward_time": 5.601583480834961,
      "step": 18485
    },
    {
      "epoch": 1.762866973876953e-05,
      "step": 18485,
      "training_step_time": 17.079723834991455
    },
    {
      "epoch": 1.7629623413085938e-05,
      "model_forward_time": 5.60528039932251,
      "step": 18486
    },
    {
      "epoch": 1.7629623413085938e-05,
      "step": 18486,
      "training_step_time": 17.08474040031433
    },
    {
      "epoch": 1.7630577087402342e-05,
      "model_forward_time": 5.6084349155426025,
      "step": 18487
    },
    {
      "epoch": 1.7630577087402342e-05,
      "step": 18487,
      "training_step_time": 17.081897735595703
    },
    {
      "epoch": 1.763153076171875e-05,
      "model_forward_time": 5.601250410079956,
      "step": 18488
    },
    {
      "epoch": 1.763153076171875e-05,
      "step": 18488,
      "training_step_time": 17.073275089263916
    },
    {
      "epoch": 1.7632484436035157e-05,
      "model_forward_time": 5.970179796218872,
      "step": 18489
    },
    {
      "epoch": 1.7632484436035157e-05,
      "step": 18489,
      "training_step_time": 17.42958950996399
    },
    {
      "epoch": 1.763343811035156e-05,
      "grad_norm": 0.9196719527244568,
      "learning_rate": 1e-05,
      "loss": 0.0579,
      "step": 18490
    },
    {
      "epoch": 1.763343811035156e-05,
      "model_forward_time": 5.607706546783447,
      "step": 18490
    },
    {
      "dynamics_loss_avg": 0.046630514599382875,
      "epoch": 1.763343811035156e-05,
      "step": 18490
    },
    {
      "action_loss_avg": 0.006759127357508987,
      "epoch": 1.763343811035156e-05,
      "step": 18490
    },
    {
      "epoch": 1.763343811035156e-05,
      "step": 18490,
      "training_step_time": 17.101462364196777
    },
    {
      "epoch": 1.763439178466797e-05,
      "model_forward_time": 5.607877254486084,
      "step": 18491
    },
    {
      "epoch": 1.763439178466797e-05,
      "step": 18491,
      "training_step_time": 17.071357011795044
    },
    {
      "epoch": 1.7635345458984376e-05,
      "model_forward_time": 5.602585792541504,
      "step": 18492
    },
    {
      "epoch": 1.7635345458984376e-05,
      "step": 18492,
      "training_step_time": 17.061697244644165
    },
    {
      "epoch": 1.763629913330078e-05,
      "model_forward_time": 5.59814453125,
      "step": 18493
    },
    {
      "epoch": 1.763629913330078e-05,
      "step": 18493,
      "training_step_time": 17.044930934906006
    },
    {
      "epoch": 1.7637252807617187e-05,
      "model_forward_time": 5.590249300003052,
      "step": 18494
    },
    {
      "epoch": 1.7637252807617187e-05,
      "step": 18494,
      "training_step_time": 17.374145030975342
    },
    {
      "epoch": 1.7638206481933595e-05,
      "model_forward_time": 5.589947700500488,
      "step": 18495
    },
    {
      "epoch": 1.7638206481933595e-05,
      "step": 18495,
      "training_step_time": 17.02981448173523
    },
    {
      "epoch": 1.763916015625e-05,
      "model_forward_time": 5.595785856246948,
      "step": 18496
    },
    {
      "epoch": 1.763916015625e-05,
      "step": 18496,
      "training_step_time": 17.058377265930176
    },
    {
      "epoch": 1.7640113830566406e-05,
      "model_forward_time": 5.605581283569336,
      "step": 18497
    },
    {
      "epoch": 1.7640113830566406e-05,
      "step": 18497,
      "training_step_time": 17.102153539657593
    },
    {
      "epoch": 1.7641067504882814e-05,
      "model_forward_time": 5.611797094345093,
      "step": 18498
    },
    {
      "epoch": 1.7641067504882814e-05,
      "step": 18498,
      "training_step_time": 17.080714464187622
    },
    {
      "epoch": 1.7642021179199218e-05,
      "model_forward_time": 5.594733715057373,
      "step": 18499
    },
    {
      "epoch": 1.7642021179199218e-05,
      "step": 18499,
      "training_step_time": 17.021780014038086
    },
    {
      "epoch": 1.7642974853515625e-05,
      "grad_norm": 0.7608544230461121,
      "learning_rate": 1e-05,
      "loss": 0.0565,
      "step": 18500
    },
    {
      "epoch": 1.7642974853515625e-05,
      "model_forward_time": 5.963860750198364,
      "step": 18500
    },
    {
      "dynamics_loss_avg": 0.05203962260857224,
      "epoch": 1.7642974853515625e-05,
      "step": 18500
    },
    {
      "action_loss_avg": 0.008194393455050885,
      "epoch": 1.7642974853515625e-05,
      "step": 18500
    },
    {
      "epoch": 1.7642974853515625e-05,
      "step": 18500,
      "training_step_time": 17.398417472839355
    },
    {
      "epoch": 1.7643928527832032e-05,
      "model_forward_time": 5.573090553283691,
      "step": 18501
    },
    {
      "epoch": 1.7643928527832032e-05,
      "step": 18501,
      "training_step_time": 16.993546724319458
    },
    {
      "epoch": 1.7644882202148436e-05,
      "model_forward_time": 5.596089124679565,
      "step": 18502
    },
    {
      "epoch": 1.7644882202148436e-05,
      "step": 18502,
      "training_step_time": 17.05959939956665
    },
    {
      "epoch": 1.7645835876464844e-05,
      "model_forward_time": 5.620060205459595,
      "step": 18503
    },
    {
      "epoch": 1.7645835876464844e-05,
      "step": 18503,
      "training_step_time": 17.127237796783447
    },
    {
      "epoch": 1.764678955078125e-05,
      "model_forward_time": 5.62823486328125,
      "step": 18504
    },
    {
      "epoch": 1.764678955078125e-05,
      "step": 18504,
      "training_step_time": 17.147396326065063
    },
    {
      "epoch": 1.7647743225097655e-05,
      "model_forward_time": 5.971547603607178,
      "step": 18505
    },
    {
      "epoch": 1.7647743225097655e-05,
      "step": 18505,
      "training_step_time": 17.456106185913086
    },
    {
      "epoch": 1.7648696899414063e-05,
      "model_forward_time": 5.613858222961426,
      "step": 18506
    },
    {
      "epoch": 1.7648696899414063e-05,
      "step": 18506,
      "training_step_time": 17.073343992233276
    },
    {
      "epoch": 1.764965057373047e-05,
      "model_forward_time": 5.607782602310181,
      "step": 18507
    },
    {
      "epoch": 1.764965057373047e-05,
      "step": 18507,
      "training_step_time": 17.075089931488037
    },
    {
      "epoch": 1.7650604248046874e-05,
      "model_forward_time": 5.610163927078247,
      "step": 18508
    },
    {
      "epoch": 1.7650604248046874e-05,
      "step": 18508,
      "training_step_time": 17.088698148727417
    },
    {
      "epoch": 1.765155792236328e-05,
      "model_forward_time": 5.614550828933716,
      "step": 18509
    },
    {
      "epoch": 1.765155792236328e-05,
      "step": 18509,
      "training_step_time": 17.09304189682007
    },
    {
      "epoch": 1.765251159667969e-05,
      "grad_norm": 0.5367981791496277,
      "learning_rate": 1e-05,
      "loss": 0.0574,
      "step": 18510
    },
    {
      "epoch": 1.765251159667969e-05,
      "model_forward_time": 5.598072052001953,
      "step": 18510
    },
    {
      "dynamics_loss_avg": 0.04990284964442253,
      "epoch": 1.765251159667969e-05,
      "step": 18510
    },
    {
      "action_loss_avg": 0.00673582199960947,
      "epoch": 1.765251159667969e-05,
      "step": 18510
    },
    {
      "epoch": 1.765251159667969e-05,
      "step": 18510,
      "training_step_time": 17.05888795852661
    },
    {
      "epoch": 1.7653465270996093e-05,
      "model_forward_time": 5.973440170288086,
      "step": 18511
    },
    {
      "epoch": 1.7653465270996093e-05,
      "step": 18511,
      "training_step_time": 17.42518639564514
    },
    {
      "epoch": 1.76544189453125e-05,
      "model_forward_time": 5.597442865371704,
      "step": 18512
    },
    {
      "epoch": 1.76544189453125e-05,
      "step": 18512,
      "training_step_time": 17.06200337409973
    },
    {
      "epoch": 1.7655372619628908e-05,
      "model_forward_time": 5.602246046066284,
      "step": 18513
    },
    {
      "epoch": 1.7655372619628908e-05,
      "step": 18513,
      "training_step_time": 17.076236963272095
    },
    {
      "epoch": 1.7656326293945312e-05,
      "model_forward_time": 5.599554538726807,
      "step": 18514
    },
    {
      "epoch": 1.7656326293945312e-05,
      "step": 18514,
      "training_step_time": 17.082544326782227
    },
    {
      "epoch": 1.765727996826172e-05,
      "model_forward_time": 5.612617015838623,
      "step": 18515
    },
    {
      "epoch": 1.765727996826172e-05,
      "step": 18515,
      "training_step_time": 17.119077444076538
    },
    {
      "epoch": 1.7658233642578123e-05,
      "model_forward_time": 5.959827899932861,
      "step": 18516
    },
    {
      "epoch": 1.7658233642578123e-05,
      "step": 18516,
      "training_step_time": 17.462578535079956
    },
    {
      "epoch": 1.765918731689453e-05,
      "model_forward_time": 5.620572566986084,
      "step": 18517
    },
    {
      "epoch": 1.765918731689453e-05,
      "step": 18517,
      "training_step_time": 17.1174476146698
    },
    {
      "epoch": 1.7660140991210938e-05,
      "model_forward_time": 5.616094589233398,
      "step": 18518
    },
    {
      "epoch": 1.7660140991210938e-05,
      "step": 18518,
      "training_step_time": 17.09260892868042
    },
    {
      "epoch": 1.7661094665527342e-05,
      "model_forward_time": 5.590609073638916,
      "step": 18519
    },
    {
      "epoch": 1.7661094665527342e-05,
      "step": 18519,
      "training_step_time": 17.01765537261963
    },
    {
      "epoch": 1.766204833984375e-05,
      "grad_norm": 0.58846116065979,
      "learning_rate": 1e-05,
      "loss": 0.0536,
      "step": 18520
    },
    {
      "epoch": 1.766204833984375e-05,
      "model_forward_time": 5.591230630874634,
      "step": 18520
    },
    {
      "dynamics_loss_avg": 0.04365994930267334,
      "epoch": 1.766204833984375e-05,
      "step": 18520
    },
    {
      "action_loss_avg": 0.007106393878348172,
      "epoch": 1.766204833984375e-05,
      "step": 18520
    },
    {
      "epoch": 1.766204833984375e-05,
      "step": 18520,
      "training_step_time": 17.018386602401733
    },
    {
      "epoch": 1.7663002014160157e-05,
      "model_forward_time": 5.580966234207153,
      "step": 18521
    },
    {
      "epoch": 1.7663002014160157e-05,
      "step": 18521,
      "training_step_time": 17.37281346321106
    },
    {
      "epoch": 1.766395568847656e-05,
      "model_forward_time": 5.585642337799072,
      "step": 18522
    },
    {
      "epoch": 1.766395568847656e-05,
      "step": 18522,
      "training_step_time": 17.01133918762207
    },
    {
      "epoch": 1.766490936279297e-05,
      "model_forward_time": 5.587123870849609,
      "step": 18523
    },
    {
      "epoch": 1.766490936279297e-05,
      "step": 18523,
      "training_step_time": 17.0314724445343
    },
    {
      "epoch": 1.7665863037109376e-05,
      "model_forward_time": 5.604572534561157,
      "step": 18524
    },
    {
      "epoch": 1.7665863037109376e-05,
      "step": 18524,
      "training_step_time": 17.069929122924805
    },
    {
      "epoch": 1.766681671142578e-05,
      "model_forward_time": 5.597252368927002,
      "step": 18525
    },
    {
      "epoch": 1.766681671142578e-05,
      "step": 18525,
      "training_step_time": 17.05698251724243
    },
    {
      "epoch": 1.7667770385742187e-05,
      "model_forward_time": 5.6053242683410645,
      "step": 18526
    },
    {
      "epoch": 1.7667770385742187e-05,
      "step": 18526,
      "training_step_time": 17.056918382644653
    },
    {
      "epoch": 1.7668724060058595e-05,
      "model_forward_time": 5.936909437179565,
      "step": 18527
    },
    {
      "epoch": 1.7668724060058595e-05,
      "step": 18527,
      "training_step_time": 17.38420057296753
    },
    {
      "epoch": 1.7669677734375e-05,
      "model_forward_time": 5.602370977401733,
      "step": 18528
    },
    {
      "epoch": 1.7669677734375e-05,
      "step": 18528,
      "training_step_time": 17.088613748550415
    },
    {
      "epoch": 1.7670631408691406e-05,
      "model_forward_time": 5.636154890060425,
      "step": 18529
    },
    {
      "epoch": 1.7670631408691406e-05,
      "step": 18529,
      "training_step_time": 17.145731210708618
    },
    {
      "epoch": 1.7671585083007814e-05,
      "grad_norm": 0.8135231733322144,
      "learning_rate": 1e-05,
      "loss": 0.0642,
      "step": 18530
    },
    {
      "epoch": 1.7671585083007814e-05,
      "model_forward_time": 5.619652271270752,
      "step": 18530
    },
    {
      "dynamics_loss_avg": 0.05820388495922089,
      "epoch": 1.7671585083007814e-05,
      "step": 18530
    },
    {
      "action_loss_avg": 0.008808327186852694,
      "epoch": 1.7671585083007814e-05,
      "step": 18530
    },
    {
      "epoch": 1.7671585083007814e-05,
      "step": 18530,
      "training_step_time": 17.126160860061646
    },
    {
      "epoch": 1.7672538757324218e-05,
      "model_forward_time": 5.606132984161377,
      "step": 18531
    },
    {
      "epoch": 1.7672538757324218e-05,
      "step": 18531,
      "training_step_time": 17.087463855743408
    },
    {
      "epoch": 1.7673492431640625e-05,
      "model_forward_time": 5.976756572723389,
      "step": 18532
    },
    {
      "epoch": 1.7673492431640625e-05,
      "step": 18532,
      "training_step_time": 17.45264720916748
    },
    {
      "epoch": 1.7674446105957032e-05,
      "model_forward_time": 5.610291004180908,
      "step": 18533
    },
    {
      "epoch": 1.7674446105957032e-05,
      "step": 18533,
      "training_step_time": 17.064708709716797
    },
    {
      "epoch": 1.7675399780273436e-05,
      "model_forward_time": 5.592091083526611,
      "step": 18534
    },
    {
      "epoch": 1.7675399780273436e-05,
      "step": 18534,
      "training_step_time": 17.05130386352539
    },
    {
      "epoch": 1.7676353454589844e-05,
      "model_forward_time": 5.59487771987915,
      "step": 18535
    },
    {
      "epoch": 1.7676353454589844e-05,
      "step": 18535,
      "training_step_time": 17.057296752929688
    },
    {
      "epoch": 1.767730712890625e-05,
      "model_forward_time": 5.594412088394165,
      "step": 18536
    },
    {
      "epoch": 1.767730712890625e-05,
      "step": 18536,
      "training_step_time": 17.05121088027954
    },
    {
      "epoch": 1.7678260803222655e-05,
      "model_forward_time": 5.599553108215332,
      "step": 18537
    },
    {
      "epoch": 1.7678260803222655e-05,
      "step": 18537,
      "training_step_time": 17.06461262702942
    },
    {
      "epoch": 1.7679214477539063e-05,
      "model_forward_time": 5.963730812072754,
      "step": 18538
    },
    {
      "epoch": 1.7679214477539063e-05,
      "step": 18538,
      "training_step_time": 17.45521068572998
    },
    {
      "epoch": 1.768016815185547e-05,
      "model_forward_time": 5.6180970668792725,
      "step": 18539
    },
    {
      "epoch": 1.768016815185547e-05,
      "step": 18539,
      "training_step_time": 17.084245681762695
    },
    {
      "epoch": 1.7681121826171874e-05,
      "grad_norm": 0.3427368402481079,
      "learning_rate": 1e-05,
      "loss": 0.0478,
      "step": 18540
    },
    {
      "epoch": 1.7681121826171874e-05,
      "model_forward_time": 5.584790468215942,
      "step": 18540
    },
    {
      "dynamics_loss_avg": 0.04842870710417628,
      "epoch": 1.7681121826171874e-05,
      "step": 18540
    },
    {
      "action_loss_avg": 0.00798988749156706,
      "epoch": 1.7681121826171874e-05,
      "step": 18540
    },
    {
      "epoch": 1.7681121826171874e-05,
      "step": 18540,
      "training_step_time": 16.991256952285767
    },
    {
      "epoch": 1.768207550048828e-05,
      "model_forward_time": 5.579937219619751,
      "step": 18541
    },
    {
      "epoch": 1.768207550048828e-05,
      "step": 18541,
      "training_step_time": 17.02291226387024
    },
    {
      "epoch": 1.768302917480469e-05,
      "model_forward_time": 5.589902400970459,
      "step": 18542
    },
    {
      "epoch": 1.768302917480469e-05,
      "step": 18542,
      "training_step_time": 17.013527870178223
    },
    {
      "epoch": 1.7683982849121093e-05,
      "model_forward_time": 5.966546297073364,
      "step": 18543
    },
    {
      "epoch": 1.7683982849121093e-05,
      "step": 18543,
      "training_step_time": 17.390701055526733
    },
    {
      "epoch": 1.76849365234375e-05,
      "model_forward_time": 5.585566759109497,
      "step": 18544
    },
    {
      "epoch": 1.76849365234375e-05,
      "step": 18544,
      "training_step_time": 17.004583835601807
    },
    {
      "epoch": 1.7685890197753908e-05,
      "model_forward_time": 5.585108041763306,
      "step": 18545
    },
    {
      "epoch": 1.7685890197753908e-05,
      "step": 18545,
      "training_step_time": 17.031970024108887
    },
    {
      "epoch": 1.7686843872070312e-05,
      "model_forward_time": 5.59833025932312,
      "step": 18546
    },
    {
      "epoch": 1.7686843872070312e-05,
      "step": 18546,
      "training_step_time": 17.072819709777832
    },
    {
      "epoch": 1.768779754638672e-05,
      "model_forward_time": 5.625964403152466,
      "step": 18547
    },
    {
      "epoch": 1.768779754638672e-05,
      "step": 18547,
      "training_step_time": 17.15482473373413
    },
    {
      "epoch": 1.7688751220703123e-05,
      "model_forward_time": 5.657698631286621,
      "step": 18548
    },
    {
      "epoch": 1.7688751220703123e-05,
      "step": 18548,
      "training_step_time": 17.17056632041931
    },
    {
      "epoch": 1.768970489501953e-05,
      "model_forward_time": 5.959839582443237,
      "step": 18549
    },
    {
      "epoch": 1.768970489501953e-05,
      "step": 18549,
      "training_step_time": 17.42630624771118
    },
    {
      "epoch": 1.7690658569335938e-05,
      "grad_norm": 0.777244508266449,
      "learning_rate": 1e-05,
      "loss": 0.0653,
      "step": 18550
    },
    {
      "epoch": 1.7690658569335938e-05,
      "model_forward_time": 5.606329441070557,
      "step": 18550
    },
    {
      "dynamics_loss_avg": 0.04656455256044865,
      "epoch": 1.7690658569335938e-05,
      "step": 18550
    },
    {
      "action_loss_avg": 0.007435631938278675,
      "epoch": 1.7690658569335938e-05,
      "step": 18550
    },
    {
      "epoch": 1.7690658569335938e-05,
      "step": 18550,
      "training_step_time": 17.07488203048706
    },
    {
      "epoch": 1.7691612243652342e-05,
      "model_forward_time": 5.599849224090576,
      "step": 18551
    },
    {
      "epoch": 1.7691612243652342e-05,
      "step": 18551,
      "training_step_time": 17.05679702758789
    },
    {
      "epoch": 1.769256591796875e-05,
      "model_forward_time": 5.6038453578948975,
      "step": 18552
    },
    {
      "epoch": 1.769256591796875e-05,
      "step": 18552,
      "training_step_time": 17.08261275291443
    },
    {
      "epoch": 1.7693519592285157e-05,
      "model_forward_time": 5.596834659576416,
      "step": 18553
    },
    {
      "epoch": 1.7693519592285157e-05,
      "step": 18553,
      "training_step_time": 17.074246883392334
    },
    {
      "epoch": 1.769447326660156e-05,
      "model_forward_time": 5.976778268814087,
      "step": 18554
    },
    {
      "epoch": 1.769447326660156e-05,
      "step": 18554,
      "training_step_time": 17.427741050720215
    },
    {
      "epoch": 1.769542694091797e-05,
      "model_forward_time": 5.593685150146484,
      "step": 18555
    },
    {
      "epoch": 1.769542694091797e-05,
      "step": 18555,
      "training_step_time": 17.050368070602417
    },
    {
      "epoch": 1.7696380615234376e-05,
      "model_forward_time": 5.602244853973389,
      "step": 18556
    },
    {
      "epoch": 1.7696380615234376e-05,
      "step": 18556,
      "training_step_time": 17.078779935836792
    },
    {
      "epoch": 1.769733428955078e-05,
      "model_forward_time": 5.604072332382202,
      "step": 18557
    },
    {
      "epoch": 1.769733428955078e-05,
      "step": 18557,
      "training_step_time": 17.08008861541748
    },
    {
      "epoch": 1.7698287963867187e-05,
      "model_forward_time": 5.595934867858887,
      "step": 18558
    },
    {
      "epoch": 1.7698287963867187e-05,
      "step": 18558,
      "training_step_time": 17.05772829055786
    },
    {
      "epoch": 1.7699241638183595e-05,
      "model_forward_time": 5.5933239459991455,
      "step": 18559
    },
    {
      "epoch": 1.7699241638183595e-05,
      "step": 18559,
      "training_step_time": 17.052220821380615
    },
    {
      "epoch": 1.77001953125e-05,
      "grad_norm": 1.6050834655761719,
      "learning_rate": 1e-05,
      "loss": 0.0683,
      "step": 18560
    },
    {
      "epoch": 1.77001953125e-05,
      "model_forward_time": 5.928118944168091,
      "step": 18560
    },
    {
      "dynamics_loss_avg": 0.05971235111355781,
      "epoch": 1.77001953125e-05,
      "step": 18560
    },
    {
      "action_loss_avg": 0.012272768176626415,
      "epoch": 1.77001953125e-05,
      "step": 18560
    },
    {
      "epoch": 1.77001953125e-05,
      "step": 18560,
      "training_step_time": 17.360271692276
    },
    {
      "epoch": 1.7701148986816406e-05,
      "model_forward_time": 5.592997312545776,
      "step": 18561
    },
    {
      "epoch": 1.7701148986816406e-05,
      "step": 18561,
      "training_step_time": 17.04873299598694
    },
    {
      "epoch": 1.7702102661132814e-05,
      "model_forward_time": 5.613133668899536,
      "step": 18562
    },
    {
      "epoch": 1.7702102661132814e-05,
      "step": 18562,
      "training_step_time": 17.088313579559326
    },
    {
      "epoch": 1.7703056335449218e-05,
      "model_forward_time": 5.595612287521362,
      "step": 18563
    },
    {
      "epoch": 1.7703056335449218e-05,
      "step": 18563,
      "training_step_time": 17.048308849334717
    },
    {
      "epoch": 1.7704010009765625e-05,
      "model_forward_time": 5.589205265045166,
      "step": 18564
    },
    {
      "epoch": 1.7704010009765625e-05,
      "step": 18564,
      "training_step_time": 17.042656183242798
    },
    {
      "epoch": 1.7704963684082032e-05,
      "model_forward_time": 5.965934753417969,
      "step": 18565
    },
    {
      "epoch": 1.7704963684082032e-05,
      "step": 18565,
      "training_step_time": 17.424438953399658
    },
    {
      "epoch": 1.7705917358398436e-05,
      "model_forward_time": 5.6318933963775635,
      "step": 18566
    },
    {
      "epoch": 1.7705917358398436e-05,
      "step": 18566,
      "training_step_time": 17.173075914382935
    },
    {
      "epoch": 1.7706871032714844e-05,
      "model_forward_time": 5.631208896636963,
      "step": 18567
    },
    {
      "epoch": 1.7706871032714844e-05,
      "step": 18567,
      "training_step_time": 17.108938694000244
    },
    {
      "epoch": 1.770782470703125e-05,
      "model_forward_time": 5.587709426879883,
      "step": 18568
    },
    {
      "epoch": 1.770782470703125e-05,
      "step": 18568,
      "training_step_time": 17.026671409606934
    },
    {
      "epoch": 1.7708778381347655e-05,
      "model_forward_time": 5.564366579055786,
      "step": 18569
    },
    {
      "epoch": 1.7708778381347655e-05,
      "step": 18569,
      "training_step_time": 16.97716212272644
    },
    {
      "epoch": 1.7709732055664063e-05,
      "grad_norm": 1.3475075960159302,
      "learning_rate": 1e-05,
      "loss": 0.0695,
      "step": 18570
    },
    {
      "epoch": 1.7709732055664063e-05,
      "model_forward_time": 5.928918838500977,
      "step": 18570
    },
    {
      "dynamics_loss_avg": 0.05633057346567512,
      "epoch": 1.7709732055664063e-05,
      "step": 18570
    },
    {
      "action_loss_avg": 0.008108165126759558,
      "epoch": 1.7709732055664063e-05,
      "step": 18570
    },
    {
      "epoch": 1.7709732055664063e-05,
      "step": 18570,
      "training_step_time": 17.336917400360107
    },
    {
      "epoch": 1.771068572998047e-05,
      "model_forward_time": 5.570041179656982,
      "step": 18571
    },
    {
      "epoch": 1.771068572998047e-05,
      "step": 18571,
      "training_step_time": 17.008042335510254
    },
    {
      "epoch": 1.7711639404296874e-05,
      "model_forward_time": 5.579983234405518,
      "step": 18572
    },
    {
      "epoch": 1.7711639404296874e-05,
      "step": 18572,
      "training_step_time": 17.009440422058105
    },
    {
      "epoch": 1.771259307861328e-05,
      "model_forward_time": 5.571739673614502,
      "step": 18573
    },
    {
      "epoch": 1.771259307861328e-05,
      "step": 18573,
      "training_step_time": 16.983564853668213
    },
    {
      "epoch": 1.771354675292969e-05,
      "model_forward_time": 5.579087495803833,
      "step": 18574
    },
    {
      "epoch": 1.771354675292969e-05,
      "step": 18574,
      "training_step_time": 17.000205039978027
    },
    {
      "epoch": 1.7714500427246093e-05,
      "model_forward_time": 5.588818550109863,
      "step": 18575
    },
    {
      "epoch": 1.7714500427246093e-05,
      "step": 18575,
      "training_step_time": 17.027925968170166
    },
    {
      "epoch": 1.77154541015625e-05,
      "model_forward_time": 5.9556498527526855,
      "step": 18576
    },
    {
      "epoch": 1.77154541015625e-05,
      "step": 18576,
      "training_step_time": 17.390700340270996
    },
    {
      "epoch": 1.7716407775878908e-05,
      "model_forward_time": 5.5980284214019775,
      "step": 18577
    },
    {
      "epoch": 1.7716407775878908e-05,
      "step": 18577,
      "training_step_time": 17.06962537765503
    },
    {
      "epoch": 1.7717361450195312e-05,
      "model_forward_time": 5.613975286483765,
      "step": 18578
    },
    {
      "epoch": 1.7717361450195312e-05,
      "step": 18578,
      "training_step_time": 17.098289251327515
    },
    {
      "epoch": 1.771831512451172e-05,
      "model_forward_time": 5.610997676849365,
      "step": 18579
    },
    {
      "epoch": 1.771831512451172e-05,
      "step": 18579,
      "training_step_time": 17.10328459739685
    },
    {
      "epoch": 1.7719268798828123e-05,
      "grad_norm": 0.5094254612922668,
      "learning_rate": 1e-05,
      "loss": 0.0457,
      "step": 18580
    },
    {
      "epoch": 1.7719268798828123e-05,
      "model_forward_time": 5.608805894851685,
      "step": 18580
    },
    {
      "dynamics_loss_avg": 0.04622126426547766,
      "epoch": 1.7719268798828123e-05,
      "step": 18580
    },
    {
      "action_loss_avg": 0.005295203975401818,
      "epoch": 1.7719268798828123e-05,
      "step": 18580
    },
    {
      "epoch": 1.7719268798828123e-05,
      "step": 18580,
      "training_step_time": 17.10483479499817
    },
    {
      "epoch": 1.772022247314453e-05,
      "model_forward_time": 5.603334426879883,
      "step": 18581
    },
    {
      "epoch": 1.772022247314453e-05,
      "step": 18581,
      "training_step_time": 17.45651650428772
    },
    {
      "epoch": 1.7721176147460938e-05,
      "model_forward_time": 5.608339309692383,
      "step": 18582
    },
    {
      "epoch": 1.7721176147460938e-05,
      "step": 18582,
      "training_step_time": 17.08848261833191
    },
    {
      "epoch": 1.7722129821777342e-05,
      "model_forward_time": 5.612908363342285,
      "step": 18583
    },
    {
      "epoch": 1.7722129821777342e-05,
      "step": 18583,
      "training_step_time": 17.08767557144165
    },
    {
      "epoch": 1.772308349609375e-05,
      "model_forward_time": 5.612911701202393,
      "step": 18584
    },
    {
      "epoch": 1.772308349609375e-05,
      "step": 18584,
      "training_step_time": 17.095255374908447
    },
    {
      "epoch": 1.7724037170410157e-05,
      "model_forward_time": 5.604256868362427,
      "step": 18585
    },
    {
      "epoch": 1.7724037170410157e-05,
      "step": 18585,
      "training_step_time": 17.098707675933838
    },
    {
      "epoch": 1.772499084472656e-05,
      "model_forward_time": 5.607349395751953,
      "step": 18586
    },
    {
      "epoch": 1.772499084472656e-05,
      "step": 18586,
      "training_step_time": 17.093505144119263
    },
    {
      "epoch": 1.772594451904297e-05,
      "model_forward_time": 5.97384786605835,
      "step": 18587
    },
    {
      "epoch": 1.772594451904297e-05,
      "step": 18587,
      "training_step_time": 17.47007131576538
    },
    {
      "epoch": 1.7726898193359376e-05,
      "model_forward_time": 5.6147496700286865,
      "step": 18588
    },
    {
      "epoch": 1.7726898193359376e-05,
      "step": 18588,
      "training_step_time": 17.10580849647522
    },
    {
      "epoch": 1.772785186767578e-05,
      "model_forward_time": 5.621795654296875,
      "step": 18589
    },
    {
      "epoch": 1.772785186767578e-05,
      "step": 18589,
      "training_step_time": 17.110406160354614
    },
    {
      "epoch": 1.7728805541992187e-05,
      "grad_norm": 0.44012337923049927,
      "learning_rate": 1e-05,
      "loss": 0.0748,
      "step": 18590
    },
    {
      "epoch": 1.7728805541992187e-05,
      "model_forward_time": 5.599359512329102,
      "step": 18590
    },
    {
      "dynamics_loss_avg": 0.06051985509693623,
      "epoch": 1.7728805541992187e-05,
      "step": 18590
    },
    {
      "action_loss_avg": 0.010776507621631026,
      "epoch": 1.7728805541992187e-05,
      "step": 18590
    },
    {
      "epoch": 1.7728805541992187e-05,
      "step": 18590,
      "training_step_time": 17.053688287734985
    },
    {
      "epoch": 1.7729759216308595e-05,
      "model_forward_time": 5.5966246128082275,
      "step": 18591
    },
    {
      "epoch": 1.7729759216308595e-05,
      "step": 18591,
      "training_step_time": 17.033628463745117
    },
    {
      "epoch": 1.7730712890625e-05,
      "model_forward_time": 5.587934732437134,
      "step": 18592
    },
    {
      "epoch": 1.7730712890625e-05,
      "step": 18592,
      "training_step_time": 17.38211679458618
    },
    {
      "epoch": 1.7731666564941406e-05,
      "model_forward_time": 5.5904083251953125,
      "step": 18593
    },
    {
      "epoch": 1.7731666564941406e-05,
      "step": 18593,
      "training_step_time": 17.05341601371765
    },
    {
      "epoch": 1.7732620239257814e-05,
      "model_forward_time": 5.603877305984497,
      "step": 18594
    },
    {
      "epoch": 1.7732620239257814e-05,
      "step": 18594,
      "training_step_time": 17.03848171234131
    },
    {
      "epoch": 1.7733573913574218e-05,
      "model_forward_time": 5.605409622192383,
      "step": 18595
    },
    {
      "epoch": 1.7733573913574218e-05,
      "step": 18595,
      "training_step_time": 17.070518016815186
    },
    {
      "epoch": 1.7734527587890625e-05,
      "model_forward_time": 5.590416193008423,
      "step": 18596
    },
    {
      "epoch": 1.7734527587890625e-05,
      "step": 18596,
      "training_step_time": 17.0456223487854
    },
    {
      "epoch": 1.7735481262207032e-05,
      "model_forward_time": 5.588515520095825,
      "step": 18597
    },
    {
      "epoch": 1.7735481262207032e-05,
      "step": 18597,
      "training_step_time": 17.03797221183777
    },
    {
      "epoch": 1.7736434936523437e-05,
      "model_forward_time": 5.967503786087036,
      "step": 18598
    },
    {
      "epoch": 1.7736434936523437e-05,
      "step": 18598,
      "training_step_time": 17.429471492767334
    },
    {
      "epoch": 1.7737388610839844e-05,
      "model_forward_time": 5.611482620239258,
      "step": 18599
    },
    {
      "epoch": 1.7737388610839844e-05,
      "step": 18599,
      "training_step_time": 17.12182354927063
    },
    {
      "epoch": 1.773834228515625e-05,
      "grad_norm": 1.5678596496582031,
      "learning_rate": 1e-05,
      "loss": 0.0591,
      "step": 18600
    },
    {
      "epoch": 1.773834228515625e-05,
      "model_forward_time": 5.618606805801392,
      "step": 18600
    },
    {
      "dynamics_loss_avg": 0.055526625365018845,
      "epoch": 1.773834228515625e-05,
      "step": 18600
    },
    {
      "action_loss_avg": 0.009237812319770455,
      "epoch": 1.773834228515625e-05,
      "step": 18600
    },
    {
      "epoch": 1.773834228515625e-05,
      "step": 18600,
      "training_step_time": 17.122459650039673
    },
    {
      "epoch": 1.7739295959472655e-05,
      "model_forward_time": 5.619723558425903,
      "step": 18601
    },
    {
      "epoch": 1.7739295959472655e-05,
      "step": 18601,
      "training_step_time": 17.104537963867188
    },
    {
      "epoch": 1.7740249633789063e-05,
      "model_forward_time": 5.5981385707855225,
      "step": 18602
    },
    {
      "epoch": 1.7740249633789063e-05,
      "step": 18602,
      "training_step_time": 17.04444122314453
    },
    {
      "epoch": 1.774120330810547e-05,
      "model_forward_time": 5.934853553771973,
      "step": 18603
    },
    {
      "epoch": 1.774120330810547e-05,
      "step": 18603,
      "training_step_time": 17.339820384979248
    },
    {
      "epoch": 1.7742156982421874e-05,
      "model_forward_time": 5.600417137145996,
      "step": 18604
    },
    {
      "epoch": 1.7742156982421874e-05,
      "step": 18604,
      "training_step_time": 17.078760623931885
    },
    {
      "epoch": 1.774311065673828e-05,
      "model_forward_time": 5.6215903759002686,
      "step": 18605
    },
    {
      "epoch": 1.774311065673828e-05,
      "step": 18605,
      "training_step_time": 17.119108200073242
    },
    {
      "epoch": 1.774406433105469e-05,
      "model_forward_time": 5.619488477706909,
      "step": 18606
    },
    {
      "epoch": 1.774406433105469e-05,
      "step": 18606,
      "training_step_time": 17.1209499835968
    },
    {
      "epoch": 1.7745018005371093e-05,
      "model_forward_time": 5.623982906341553,
      "step": 18607
    },
    {
      "epoch": 1.7745018005371093e-05,
      "step": 18607,
      "training_step_time": 17.13852047920227
    },
    {
      "epoch": 1.77459716796875e-05,
      "model_forward_time": 5.620215177536011,
      "step": 18608
    },
    {
      "epoch": 1.77459716796875e-05,
      "step": 18608,
      "training_step_time": 17.11651086807251
    },
    {
      "epoch": 1.7746925354003908e-05,
      "model_forward_time": 5.993813514709473,
      "step": 18609
    },
    {
      "epoch": 1.7746925354003908e-05,
      "step": 18609,
      "training_step_time": 17.44004201889038
    },
    {
      "epoch": 1.7747879028320312e-05,
      "grad_norm": 1.4088935852050781,
      "learning_rate": 1e-05,
      "loss": 0.081,
      "step": 18610
    },
    {
      "epoch": 1.7747879028320312e-05,
      "model_forward_time": 5.613216161727905,
      "step": 18610
    },
    {
      "dynamics_loss_avg": 0.06327667124569417,
      "epoch": 1.7747879028320312e-05,
      "step": 18610
    },
    {
      "action_loss_avg": 0.012283325009047986,
      "epoch": 1.7747879028320312e-05,
      "step": 18610
    },
    {
      "epoch": 1.7747879028320312e-05,
      "step": 18610,
      "training_step_time": 17.094401597976685
    },
    {
      "epoch": 1.774883270263672e-05,
      "model_forward_time": 5.6299028396606445,
      "step": 18611
    },
    {
      "epoch": 1.774883270263672e-05,
      "step": 18611,
      "training_step_time": 17.11478018760681
    },
    {
      "epoch": 1.7749786376953123e-05,
      "model_forward_time": 5.6406145095825195,
      "step": 18612
    },
    {
      "epoch": 1.7749786376953123e-05,
      "step": 18612,
      "training_step_time": 17.178130626678467
    },
    {
      "epoch": 1.775074005126953e-05,
      "model_forward_time": 5.662876844406128,
      "step": 18613
    },
    {
      "epoch": 1.775074005126953e-05,
      "step": 18613,
      "training_step_time": 17.168575286865234
    },
    {
      "epoch": 1.7751693725585938e-05,
      "model_forward_time": 5.610445499420166,
      "step": 18614
    },
    {
      "epoch": 1.7751693725585938e-05,
      "step": 18614,
      "training_step_time": 17.44062089920044
    },
    {
      "epoch": 1.7752647399902342e-05,
      "model_forward_time": 5.592200040817261,
      "step": 18615
    },
    {
      "epoch": 1.7752647399902342e-05,
      "step": 18615,
      "training_step_time": 17.029249906539917
    },
    {
      "epoch": 1.775360107421875e-05,
      "model_forward_time": 5.582433462142944,
      "step": 18616
    },
    {
      "epoch": 1.775360107421875e-05,
      "step": 18616,
      "training_step_time": 17.03614068031311
    },
    {
      "epoch": 1.7754554748535157e-05,
      "model_forward_time": 5.582992076873779,
      "step": 18617
    },
    {
      "epoch": 1.7754554748535157e-05,
      "step": 18617,
      "training_step_time": 17.023727416992188
    },
    {
      "epoch": 1.775550842285156e-05,
      "model_forward_time": 5.585620164871216,
      "step": 18618
    },
    {
      "epoch": 1.775550842285156e-05,
      "step": 18618,
      "training_step_time": 17.04552984237671
    },
    {
      "epoch": 1.775646209716797e-05,
      "model_forward_time": 5.617987871170044,
      "step": 18619
    },
    {
      "epoch": 1.775646209716797e-05,
      "step": 18619,
      "training_step_time": 17.126619815826416
    },
    {
      "epoch": 1.7757415771484376e-05,
      "grad_norm": 1.2234039306640625,
      "learning_rate": 1e-05,
      "loss": 0.0573,
      "step": 18620
    },
    {
      "epoch": 1.7757415771484376e-05,
      "model_forward_time": 6.0067949295043945,
      "step": 18620
    },
    {
      "dynamics_loss_avg": 0.05008042436093092,
      "epoch": 1.7757415771484376e-05,
      "step": 18620
    },
    {
      "action_loss_avg": 0.009103656536899507,
      "epoch": 1.7757415771484376e-05,
      "step": 18620
    },
    {
      "epoch": 1.7757415771484376e-05,
      "step": 18620,
      "training_step_time": 17.48226571083069
    },
    {
      "epoch": 1.775836944580078e-05,
      "model_forward_time": 5.590410232543945,
      "step": 18621
    },
    {
      "epoch": 1.775836944580078e-05,
      "step": 18621,
      "training_step_time": 17.039364337921143
    },
    {
      "epoch": 1.7759323120117187e-05,
      "model_forward_time": 5.57779860496521,
      "step": 18622
    },
    {
      "epoch": 1.7759323120117187e-05,
      "step": 18622,
      "training_step_time": 17.008143424987793
    },
    {
      "epoch": 1.7760276794433595e-05,
      "model_forward_time": 5.581552267074585,
      "step": 18623
    },
    {
      "epoch": 1.7760276794433595e-05,
      "step": 18623,
      "training_step_time": 16.998316764831543
    },
    {
      "epoch": 1.776123046875e-05,
      "model_forward_time": 5.5877416133880615,
      "step": 18624
    },
    {
      "epoch": 1.776123046875e-05,
      "step": 18624,
      "training_step_time": 17.02126121520996
    },
    {
      "epoch": 1.7762184143066406e-05,
      "model_forward_time": 5.594607591629028,
      "step": 18625
    },
    {
      "epoch": 1.7762184143066406e-05,
      "step": 18625,
      "training_step_time": 17.393022298812866
    },
    {
      "epoch": 1.7763137817382814e-05,
      "model_forward_time": 5.588882923126221,
      "step": 18626
    },
    {
      "epoch": 1.7763137817382814e-05,
      "step": 18626,
      "training_step_time": 17.061991214752197
    },
    {
      "epoch": 1.7764091491699218e-05,
      "model_forward_time": 5.609157562255859,
      "step": 18627
    },
    {
      "epoch": 1.7764091491699218e-05,
      "step": 18627,
      "training_step_time": 17.094078302383423
    },
    {
      "epoch": 1.7765045166015625e-05,
      "model_forward_time": 5.612858057022095,
      "step": 18628
    },
    {
      "epoch": 1.7765045166015625e-05,
      "step": 18628,
      "training_step_time": 17.111286401748657
    },
    {
      "epoch": 1.7765998840332033e-05,
      "model_forward_time": 5.611824035644531,
      "step": 18629
    },
    {
      "epoch": 1.7765998840332033e-05,
      "step": 18629,
      "training_step_time": 17.09057116508484
    },
    {
      "epoch": 1.7766952514648437e-05,
      "grad_norm": 1.6659152507781982,
      "learning_rate": 1e-05,
      "loss": 0.0661,
      "step": 18630
    },
    {
      "epoch": 1.7766952514648437e-05,
      "model_forward_time": 5.620750904083252,
      "step": 18630
    },
    {
      "dynamics_loss_avg": 0.05371282696723938,
      "epoch": 1.7766952514648437e-05,
      "step": 18630
    },
    {
      "action_loss_avg": 0.008936705812811852,
      "epoch": 1.7766952514648437e-05,
      "step": 18630
    },
    {
      "epoch": 1.7766952514648437e-05,
      "step": 18630,
      "training_step_time": 17.097949266433716
    },
    {
      "epoch": 1.7767906188964844e-05,
      "model_forward_time": 5.969594717025757,
      "step": 18631
    },
    {
      "epoch": 1.7767906188964844e-05,
      "step": 18631,
      "training_step_time": 17.395379304885864
    },
    {
      "epoch": 1.776885986328125e-05,
      "model_forward_time": 5.583779335021973,
      "step": 18632
    },
    {
      "epoch": 1.776885986328125e-05,
      "step": 18632,
      "training_step_time": 17.021467447280884
    },
    {
      "epoch": 1.7769813537597655e-05,
      "model_forward_time": 5.599489450454712,
      "step": 18633
    },
    {
      "epoch": 1.7769813537597655e-05,
      "step": 18633,
      "training_step_time": 17.058768272399902
    },
    {
      "epoch": 1.7770767211914063e-05,
      "model_forward_time": 5.6134419441223145,
      "step": 18634
    },
    {
      "epoch": 1.7770767211914063e-05,
      "step": 18634,
      "training_step_time": 17.11079168319702
    },
    {
      "epoch": 1.777172088623047e-05,
      "model_forward_time": 5.616831302642822,
      "step": 18635
    },
    {
      "epoch": 1.777172088623047e-05,
      "step": 18635,
      "training_step_time": 17.108160734176636
    },
    {
      "epoch": 1.7772674560546874e-05,
      "model_forward_time": 5.978816747665405,
      "step": 18636
    },
    {
      "epoch": 1.7772674560546874e-05,
      "step": 18636,
      "training_step_time": 17.472352027893066
    },
    {
      "epoch": 1.777362823486328e-05,
      "model_forward_time": 5.611570358276367,
      "step": 18637
    },
    {
      "epoch": 1.777362823486328e-05,
      "step": 18637,
      "training_step_time": 17.08812975883484
    },
    {
      "epoch": 1.777458190917969e-05,
      "model_forward_time": 5.5979766845703125,
      "step": 18638
    },
    {
      "epoch": 1.777458190917969e-05,
      "step": 18638,
      "training_step_time": 17.03089475631714
    },
    {
      "epoch": 1.7775535583496093e-05,
      "model_forward_time": 5.5841851234436035,
      "step": 18639
    },
    {
      "epoch": 1.7775535583496093e-05,
      "step": 18639,
      "training_step_time": 17.024518489837646
    },
    {
      "epoch": 1.77764892578125e-05,
      "grad_norm": 0.6118424534797668,
      "learning_rate": 1e-05,
      "loss": 0.059,
      "step": 18640
    },
    {
      "epoch": 1.77764892578125e-05,
      "model_forward_time": 5.597967147827148,
      "step": 18640
    },
    {
      "dynamics_loss_avg": 0.05761330332607031,
      "epoch": 1.77764892578125e-05,
      "step": 18640
    },
    {
      "action_loss_avg": 0.008591814432293177,
      "epoch": 1.77764892578125e-05,
      "step": 18640
    },
    {
      "epoch": 1.77764892578125e-05,
      "step": 18640,
      "training_step_time": 17.07033348083496
    },
    {
      "epoch": 1.7777442932128905e-05,
      "model_forward_time": 5.605924129486084,
      "step": 18641
    },
    {
      "epoch": 1.7777442932128905e-05,
      "step": 18641,
      "training_step_time": 17.094649076461792
    },
    {
      "epoch": 1.7778396606445312e-05,
      "model_forward_time": 5.972503662109375,
      "step": 18642
    },
    {
      "epoch": 1.7778396606445312e-05,
      "step": 18642,
      "training_step_time": 17.4671311378479
    },
    {
      "epoch": 1.777935028076172e-05,
      "model_forward_time": 5.607405424118042,
      "step": 18643
    },
    {
      "epoch": 1.777935028076172e-05,
      "step": 18643,
      "training_step_time": 17.08565926551819
    },
    {
      "epoch": 1.7780303955078123e-05,
      "model_forward_time": 5.613019943237305,
      "step": 18644
    },
    {
      "epoch": 1.7780303955078123e-05,
      "step": 18644,
      "training_step_time": 17.100213766098022
    },
    {
      "epoch": 1.778125762939453e-05,
      "model_forward_time": 5.583504915237427,
      "step": 18645
    },
    {
      "epoch": 1.778125762939453e-05,
      "step": 18645,
      "training_step_time": 17.000235557556152
    },
    {
      "epoch": 1.7782211303710938e-05,
      "model_forward_time": 5.589555025100708,
      "step": 18646
    },
    {
      "epoch": 1.7782211303710938e-05,
      "step": 18646,
      "training_step_time": 17.04061007499695
    },
    {
      "epoch": 1.7783164978027342e-05,
      "model_forward_time": 5.601074934005737,
      "step": 18647
    },
    {
      "epoch": 1.7783164978027342e-05,
      "step": 18647,
      "training_step_time": 17.410281658172607
    },
    {
      "epoch": 1.778411865234375e-05,
      "model_forward_time": 5.612085819244385,
      "step": 18648
    },
    {
      "epoch": 1.778411865234375e-05,
      "step": 18648,
      "training_step_time": 17.088857889175415
    },
    {
      "epoch": 1.7785072326660157e-05,
      "model_forward_time": 5.610250234603882,
      "step": 18649
    },
    {
      "epoch": 1.7785072326660157e-05,
      "step": 18649,
      "training_step_time": 17.109914541244507
    },
    {
      "epoch": 1.778602600097656e-05,
      "grad_norm": 1.4517357349395752,
      "learning_rate": 1e-05,
      "loss": 0.0807,
      "step": 18650
    },
    {
      "epoch": 1.778602600097656e-05,
      "model_forward_time": 5.607599973678589,
      "step": 18650
    },
    {
      "dynamics_loss_avg": 0.06330985650420189,
      "epoch": 1.778602600097656e-05,
      "step": 18650
    },
    {
      "action_loss_avg": 0.015706623531877996,
      "epoch": 1.778602600097656e-05,
      "step": 18650
    },
    {
      "epoch": 1.778602600097656e-05,
      "step": 18650,
      "training_step_time": 17.095131635665894
    },
    {
      "epoch": 1.778697967529297e-05,
      "model_forward_time": 5.612550973892212,
      "step": 18651
    },
    {
      "epoch": 1.778697967529297e-05,
      "step": 18651,
      "training_step_time": 17.090541124343872
    },
    {
      "epoch": 1.7787933349609376e-05,
      "model_forward_time": 5.605632066726685,
      "step": 18652
    },
    {
      "epoch": 1.7787933349609376e-05,
      "step": 18652,
      "training_step_time": 17.07518172264099
    },
    {
      "epoch": 1.778888702392578e-05,
      "model_forward_time": 5.966694593429565,
      "step": 18653
    },
    {
      "epoch": 1.778888702392578e-05,
      "step": 18653,
      "training_step_time": 17.417186737060547
    },
    {
      "epoch": 1.7789840698242187e-05,
      "model_forward_time": 5.603981256484985,
      "step": 18654
    },
    {
      "epoch": 1.7789840698242187e-05,
      "step": 18654,
      "training_step_time": 17.080443382263184
    },
    {
      "epoch": 1.7790794372558595e-05,
      "model_forward_time": 4.272220134735107,
      "step": 18655
    },
    {
      "epoch": 1.7790794372558595e-05,
      "step": 18655,
      "training_step_time": 12.807369709014893
    },
    {
      "epoch": 1.7791748046875e-05,
      "model_forward_time": 5.615958213806152,
      "step": 18656
    },
    {
      "epoch": 1.7791748046875e-05,
      "step": 18656,
      "training_step_time": 17.11454153060913
    },
    {
      "epoch": 1.7792701721191406e-05,
      "model_forward_time": 5.622886896133423,
      "step": 18657
    },
    {
      "epoch": 1.7792701721191406e-05,
      "step": 18657,
      "training_step_time": 17.12492275238037
    },
    {
      "epoch": 1.7793655395507814e-05,
      "model_forward_time": 5.622178077697754,
      "step": 18658
    },
    {
      "epoch": 1.7793655395507814e-05,
      "step": 18658,
      "training_step_time": 17.129355430603027
    },
    {
      "epoch": 1.7794609069824218e-05,
      "model_forward_time": 5.955333948135376,
      "step": 18659
    },
    {
      "epoch": 1.7794609069824218e-05,
      "step": 18659,
      "training_step_time": 17.442049264907837
    },
    {
      "epoch": 1.7795562744140625e-05,
      "grad_norm": 1.0327779054641724,
      "learning_rate": 1e-05,
      "loss": 0.07,
      "step": 18660
    },
    {
      "epoch": 1.7795562744140625e-05,
      "model_forward_time": 5.621716499328613,
      "step": 18660
    },
    {
      "dynamics_loss_avg": 0.05744220893830061,
      "epoch": 1.7795562744140625e-05,
      "step": 18660
    },
    {
      "action_loss_avg": 0.008422284014523029,
      "epoch": 1.7795562744140625e-05,
      "step": 18660
    },
    {
      "epoch": 1.7795562744140625e-05,
      "step": 18660,
      "training_step_time": 17.08808207511902
    },
    {
      "epoch": 1.7796516418457033e-05,
      "model_forward_time": 5.5869786739349365,
      "step": 18661
    },
    {
      "epoch": 1.7796516418457033e-05,
      "step": 18661,
      "training_step_time": 17.00058102607727
    },
    {
      "epoch": 1.7797470092773437e-05,
      "model_forward_time": 5.569619655609131,
      "step": 18662
    },
    {
      "epoch": 1.7797470092773437e-05,
      "step": 18662,
      "training_step_time": 16.98554754257202
    },
    {
      "epoch": 1.7798423767089844e-05,
      "model_forward_time": 5.569063901901245,
      "step": 18663
    },
    {
      "epoch": 1.7798423767089844e-05,
      "step": 18663,
      "training_step_time": 17.001158714294434
    },
    {
      "epoch": 1.779937744140625e-05,
      "model_forward_time": 5.932341575622559,
      "step": 18664
    },
    {
      "epoch": 1.779937744140625e-05,
      "step": 18664,
      "training_step_time": 17.350011348724365
    },
    {
      "epoch": 1.7800331115722655e-05,
      "model_forward_time": 5.577716588973999,
      "step": 18665
    },
    {
      "epoch": 1.7800331115722655e-05,
      "step": 18665,
      "training_step_time": 16.985076904296875
    },
    {
      "epoch": 1.7801284790039063e-05,
      "model_forward_time": 5.577510118484497,
      "step": 18666
    },
    {
      "epoch": 1.7801284790039063e-05,
      "step": 18666,
      "training_step_time": 16.999528408050537
    },
    {
      "epoch": 1.780223846435547e-05,
      "model_forward_time": 5.591878414154053,
      "step": 18667
    },
    {
      "epoch": 1.780223846435547e-05,
      "step": 18667,
      "training_step_time": 17.03944706916809
    },
    {
      "epoch": 1.7803192138671874e-05,
      "model_forward_time": 5.62391209602356,
      "step": 18668
    },
    {
      "epoch": 1.7803192138671874e-05,
      "step": 18668,
      "training_step_time": 17.106775283813477
    },
    {
      "epoch": 1.7804145812988282e-05,
      "model_forward_time": 5.96282696723938,
      "step": 18669
    },
    {
      "epoch": 1.7804145812988282e-05,
      "step": 18669,
      "training_step_time": 17.439316987991333
    },
    {
      "epoch": 1.780509948730469e-05,
      "grad_norm": 0.9165120124816895,
      "learning_rate": 1e-05,
      "loss": 0.0534,
      "step": 18670
    },
    {
      "epoch": 1.780509948730469e-05,
      "model_forward_time": 5.611001968383789,
      "step": 18670
    },
    {
      "dynamics_loss_avg": 0.051642170362174514,
      "epoch": 1.780509948730469e-05,
      "step": 18670
    },
    {
      "action_loss_avg": 0.00943714592140168,
      "epoch": 1.780509948730469e-05,
      "step": 18670
    },
    {
      "epoch": 1.780509948730469e-05,
      "step": 18670,
      "training_step_time": 17.078666925430298
    },
    {
      "epoch": 1.7806053161621093e-05,
      "model_forward_time": 5.589867353439331,
      "step": 18671
    },
    {
      "epoch": 1.7806053161621093e-05,
      "step": 18671,
      "training_step_time": 17.035556316375732
    },
    {
      "epoch": 1.78070068359375e-05,
      "model_forward_time": 5.565428733825684,
      "step": 18672
    },
    {
      "epoch": 1.78070068359375e-05,
      "step": 18672,
      "training_step_time": 16.96884036064148
    },
    {
      "epoch": 1.7807960510253905e-05,
      "model_forward_time": 5.563053607940674,
      "step": 18673
    },
    {
      "epoch": 1.7807960510253905e-05,
      "step": 18673,
      "training_step_time": 16.983827829360962
    },
    {
      "epoch": 1.7808914184570312e-05,
      "model_forward_time": 5.564340829849243,
      "step": 18674
    },
    {
      "epoch": 1.7808914184570312e-05,
      "step": 18674,
      "training_step_time": 17.00366759300232
    },
    {
      "epoch": 1.780986785888672e-05,
      "model_forward_time": 5.97300386428833,
      "step": 18675
    },
    {
      "epoch": 1.780986785888672e-05,
      "step": 18675,
      "training_step_time": 17.399909496307373
    },
    {
      "epoch": 1.7810821533203123e-05,
      "model_forward_time": 5.5727808475494385,
      "step": 18676
    },
    {
      "epoch": 1.7810821533203123e-05,
      "step": 18676,
      "training_step_time": 16.98848843574524
    },
    {
      "epoch": 1.781177520751953e-05,
      "model_forward_time": 5.580568313598633,
      "step": 18677
    },
    {
      "epoch": 1.781177520751953e-05,
      "step": 18677,
      "training_step_time": 17.01385998725891
    },
    {
      "epoch": 1.7812728881835938e-05,
      "model_forward_time": 5.584205627441406,
      "step": 18678
    },
    {
      "epoch": 1.7812728881835938e-05,
      "step": 18678,
      "training_step_time": 17.053704261779785
    },
    {
      "epoch": 1.7813682556152342e-05,
      "model_forward_time": 5.622669219970703,
      "step": 18679
    },
    {
      "epoch": 1.7813682556152342e-05,
      "step": 18679,
      "training_step_time": 17.13370418548584
    },
    {
      "epoch": 1.781463623046875e-05,
      "grad_norm": 0.5707622766494751,
      "learning_rate": 1e-05,
      "loss": 0.061,
      "step": 18680
    },
    {
      "epoch": 1.781463623046875e-05,
      "model_forward_time": 5.617361545562744,
      "step": 18680
    },
    {
      "dynamics_loss_avg": 0.051970289275050166,
      "epoch": 1.781463623046875e-05,
      "step": 18680
    },
    {
      "action_loss_avg": 0.008035707473754882,
      "epoch": 1.781463623046875e-05,
      "step": 18680
    },
    {
      "epoch": 1.781463623046875e-05,
      "step": 18680,
      "training_step_time": 17.477224826812744
    },
    {
      "epoch": 1.7815589904785157e-05,
      "model_forward_time": 5.613912343978882,
      "step": 18681
    },
    {
      "epoch": 1.7815589904785157e-05,
      "step": 18681,
      "training_step_time": 17.085015773773193
    },
    {
      "epoch": 1.781654357910156e-05,
      "model_forward_time": 5.611001014709473,
      "step": 18682
    },
    {
      "epoch": 1.781654357910156e-05,
      "step": 18682,
      "training_step_time": 17.047139167785645
    },
    {
      "epoch": 1.781749725341797e-05,
      "model_forward_time": 5.593621730804443,
      "step": 18683
    },
    {
      "epoch": 1.781749725341797e-05,
      "step": 18683,
      "training_step_time": 17.039394855499268
    },
    {
      "epoch": 1.7818450927734376e-05,
      "model_forward_time": 5.580714702606201,
      "step": 18684
    },
    {
      "epoch": 1.7818450927734376e-05,
      "step": 18684,
      "training_step_time": 16.996981620788574
    },
    {
      "epoch": 1.781940460205078e-05,
      "model_forward_time": 5.59468412399292,
      "step": 18685
    },
    {
      "epoch": 1.781940460205078e-05,
      "step": 18685,
      "training_step_time": 17.056212902069092
    },
    {
      "epoch": 1.7820358276367187e-05,
      "model_forward_time": 5.988374471664429,
      "step": 18686
    },
    {
      "epoch": 1.7820358276367187e-05,
      "step": 18686,
      "training_step_time": 17.475085258483887
    },
    {
      "epoch": 1.7821311950683595e-05,
      "model_forward_time": 5.620173215866089,
      "step": 18687
    },
    {
      "epoch": 1.7821311950683595e-05,
      "step": 18687,
      "training_step_time": 17.141029119491577
    },
    {
      "epoch": 1.7822265625e-05,
      "model_forward_time": 5.625184535980225,
      "step": 18688
    },
    {
      "epoch": 1.7822265625e-05,
      "step": 18688,
      "training_step_time": 17.126396894454956
    },
    {
      "epoch": 1.7823219299316406e-05,
      "model_forward_time": 5.611704349517822,
      "step": 18689
    },
    {
      "epoch": 1.7823219299316406e-05,
      "step": 18689,
      "training_step_time": 17.100530862808228
    },
    {
      "epoch": 1.7824172973632814e-05,
      "grad_norm": 0.27892208099365234,
      "learning_rate": 1e-05,
      "loss": 0.0593,
      "step": 18690
    },
    {
      "epoch": 1.7824172973632814e-05,
      "model_forward_time": 5.610932350158691,
      "step": 18690
    },
    {
      "dynamics_loss_avg": 0.04940241500735283,
      "epoch": 1.7824172973632814e-05,
      "step": 18690
    },
    {
      "action_loss_avg": 0.007602395873982459,
      "epoch": 1.7824172973632814e-05,
      "step": 18690
    },
    {
      "epoch": 1.7824172973632814e-05,
      "step": 18690,
      "training_step_time": 17.09501004219055
    },
    {
      "epoch": 1.7825126647949218e-05,
      "model_forward_time": 5.606743335723877,
      "step": 18691
    },
    {
      "epoch": 1.7825126647949218e-05,
      "step": 18691,
      "training_step_time": 17.459322929382324
    },
    {
      "epoch": 1.7826080322265625e-05,
      "model_forward_time": 5.606244087219238,
      "step": 18692
    },
    {
      "epoch": 1.7826080322265625e-05,
      "step": 18692,
      "training_step_time": 17.076067447662354
    },
    {
      "epoch": 1.7827033996582033e-05,
      "model_forward_time": 5.615752696990967,
      "step": 18693
    },
    {
      "epoch": 1.7827033996582033e-05,
      "step": 18693,
      "training_step_time": 17.099446773529053
    },
    {
      "epoch": 1.7827987670898437e-05,
      "model_forward_time": 5.607020854949951,
      "step": 18694
    },
    {
      "epoch": 1.7827987670898437e-05,
      "step": 18694,
      "training_step_time": 17.089859008789062
    },
    {
      "epoch": 1.7828941345214844e-05,
      "model_forward_time": 5.606866121292114,
      "step": 18695
    },
    {
      "epoch": 1.7828941345214844e-05,
      "step": 18695,
      "training_step_time": 17.102031469345093
    },
    {
      "epoch": 1.782989501953125e-05,
      "model_forward_time": 5.611273765563965,
      "step": 18696
    },
    {
      "epoch": 1.782989501953125e-05,
      "step": 18696,
      "training_step_time": 17.115501165390015
    },
    {
      "epoch": 1.7830848693847655e-05,
      "model_forward_time": 5.980147361755371,
      "step": 18697
    },
    {
      "epoch": 1.7830848693847655e-05,
      "step": 18697,
      "training_step_time": 17.45129156112671
    },
    {
      "epoch": 1.7831802368164063e-05,
      "model_forward_time": 5.595058441162109,
      "step": 18698
    },
    {
      "epoch": 1.7831802368164063e-05,
      "step": 18698,
      "training_step_time": 17.018750190734863
    },
    {
      "epoch": 1.783275604248047e-05,
      "model_forward_time": 5.587366342544556,
      "step": 18699
    },
    {
      "epoch": 1.783275604248047e-05,
      "step": 18699,
      "training_step_time": 16.996992588043213
    },
    {
      "epoch": 1.7833709716796874e-05,
      "grad_norm": 1.1367167234420776,
      "learning_rate": 1e-05,
      "loss": 0.069,
      "step": 18700
    },
    {
      "epoch": 1.7833709716796874e-05,
      "model_forward_time": 5.58867883682251,
      "step": 18700
    },
    {
      "dynamics_loss_avg": 0.056648983433842656,
      "epoch": 1.7833709716796874e-05,
      "step": 18700
    },
    {
      "action_loss_avg": 0.009029583679512143,
      "epoch": 1.7833709716796874e-05,
      "step": 18700
    },
    {
      "epoch": 1.7833709716796874e-05,
      "step": 18700,
      "training_step_time": 17.030598402023315
    },
    {
      "epoch": 1.7834663391113282e-05,
      "model_forward_time": 5.587266683578491,
      "step": 18701
    },
    {
      "epoch": 1.7834663391113282e-05,
      "step": 18701,
      "training_step_time": 17.019877672195435
    },
    {
      "epoch": 1.783561706542969e-05,
      "model_forward_time": 5.955785751342773,
      "step": 18702
    },
    {
      "epoch": 1.783561706542969e-05,
      "step": 18702,
      "training_step_time": 17.39951467514038
    },
    {
      "epoch": 1.7836570739746093e-05,
      "model_forward_time": 5.596831560134888,
      "step": 18703
    },
    {
      "epoch": 1.7836570739746093e-05,
      "step": 18703,
      "training_step_time": 17.062294483184814
    },
    {
      "epoch": 1.78375244140625e-05,
      "model_forward_time": 5.595258474349976,
      "step": 18704
    },
    {
      "epoch": 1.78375244140625e-05,
      "step": 18704,
      "training_step_time": 17.04664158821106
    },
    {
      "epoch": 1.7838478088378905e-05,
      "model_forward_time": 5.596613645553589,
      "step": 18705
    },
    {
      "epoch": 1.7838478088378905e-05,
      "step": 18705,
      "training_step_time": 17.040842294692993
    },
    {
      "epoch": 1.7839431762695312e-05,
      "model_forward_time": 5.595812082290649,
      "step": 18706
    },
    {
      "epoch": 1.7839431762695312e-05,
      "step": 18706,
      "training_step_time": 17.05302119255066
    },
    {
      "epoch": 1.784038543701172e-05,
      "model_forward_time": 5.593866586685181,
      "step": 18707
    },
    {
      "epoch": 1.784038543701172e-05,
      "step": 18707,
      "training_step_time": 17.03629755973816
    },
    {
      "epoch": 1.7841339111328123e-05,
      "model_forward_time": 5.966407537460327,
      "step": 18708
    },
    {
      "epoch": 1.7841339111328123e-05,
      "step": 18708,
      "training_step_time": 17.40278434753418
    },
    {
      "epoch": 1.784229278564453e-05,
      "model_forward_time": 5.589421033859253,
      "step": 18709
    },
    {
      "epoch": 1.784229278564453e-05,
      "step": 18709,
      "training_step_time": 17.025458574295044
    },
    {
      "epoch": 1.784324645996094e-05,
      "grad_norm": 0.41928860545158386,
      "learning_rate": 1e-05,
      "loss": 0.0607,
      "step": 18710
    },
    {
      "epoch": 1.784324645996094e-05,
      "model_forward_time": 5.609389543533325,
      "step": 18710
    },
    {
      "dynamics_loss_avg": 0.053032280690968034,
      "epoch": 1.784324645996094e-05,
      "step": 18710
    },
    {
      "action_loss_avg": 0.006235465232748538,
      "epoch": 1.784324645996094e-05,
      "step": 18710
    },
    {
      "epoch": 1.784324645996094e-05,
      "step": 18710,
      "training_step_time": 17.07940435409546
    },
    {
      "epoch": 1.7844200134277342e-05,
      "model_forward_time": 5.597113132476807,
      "step": 18711
    },
    {
      "epoch": 1.7844200134277342e-05,
      "step": 18711,
      "training_step_time": 17.07707190513611
    },
    {
      "epoch": 1.784515380859375e-05,
      "model_forward_time": 5.610387086868286,
      "step": 18712
    },
    {
      "epoch": 1.784515380859375e-05,
      "step": 18712,
      "training_step_time": 17.10080313682556
    },
    {
      "epoch": 1.7846107482910157e-05,
      "model_forward_time": 5.605955600738525,
      "step": 18713
    },
    {
      "epoch": 1.7846107482910157e-05,
      "step": 18713,
      "training_step_time": 17.46688222885132
    },
    {
      "epoch": 1.784706115722656e-05,
      "model_forward_time": 5.617056846618652,
      "step": 18714
    },
    {
      "epoch": 1.784706115722656e-05,
      "step": 18714,
      "training_step_time": 17.108644247055054
    },
    {
      "epoch": 1.784801483154297e-05,
      "model_forward_time": 5.5924553871154785,
      "step": 18715
    },
    {
      "epoch": 1.784801483154297e-05,
      "step": 18715,
      "training_step_time": 17.021594285964966
    },
    {
      "epoch": 1.7848968505859376e-05,
      "model_forward_time": 5.570750951766968,
      "step": 18716
    },
    {
      "epoch": 1.7848968505859376e-05,
      "step": 18716,
      "training_step_time": 16.99480700492859
    },
    {
      "epoch": 1.784992218017578e-05,
      "model_forward_time": 5.557063817977905,
      "step": 18717
    },
    {
      "epoch": 1.784992218017578e-05,
      "step": 18717,
      "training_step_time": 16.97842311859131
    },
    {
      "epoch": 1.7850875854492187e-05,
      "model_forward_time": 5.558920621871948,
      "step": 18718
    },
    {
      "epoch": 1.7850875854492187e-05,
      "step": 18718,
      "training_step_time": 16.964345693588257
    },
    {
      "epoch": 1.7851829528808595e-05,
      "model_forward_time": 5.935457468032837,
      "step": 18719
    },
    {
      "epoch": 1.7851829528808595e-05,
      "step": 18719,
      "training_step_time": 17.352274894714355
    },
    {
      "epoch": 1.7852783203125e-05,
      "grad_norm": 0.4938778579235077,
      "learning_rate": 1e-05,
      "loss": 0.0528,
      "step": 18720
    },
    {
      "epoch": 1.7852783203125e-05,
      "model_forward_time": 5.566345691680908,
      "step": 18720
    },
    {
      "dynamics_loss_avg": 0.055639248434454204,
      "epoch": 1.7852783203125e-05,
      "step": 18720
    },
    {
      "action_loss_avg": 0.005430119927041233,
      "epoch": 1.7852783203125e-05,
      "step": 18720
    },
    {
      "epoch": 1.7852783203125e-05,
      "step": 18720,
      "training_step_time": 16.997233390808105
    },
    {
      "epoch": 1.7853736877441406e-05,
      "model_forward_time": 5.570332288742065,
      "step": 18721
    },
    {
      "epoch": 1.7853736877441406e-05,
      "step": 18721,
      "training_step_time": 16.971168994903564
    },
    {
      "epoch": 1.7854690551757814e-05,
      "model_forward_time": 5.5895867347717285,
      "step": 18722
    },
    {
      "epoch": 1.7854690551757814e-05,
      "step": 18722,
      "training_step_time": 17.037196159362793
    },
    {
      "epoch": 1.7855644226074218e-05,
      "model_forward_time": 5.596771717071533,
      "step": 18723
    },
    {
      "epoch": 1.7855644226074218e-05,
      "step": 18723,
      "training_step_time": 17.082133769989014
    },
    {
      "epoch": 1.7856597900390625e-05,
      "model_forward_time": 5.60826301574707,
      "step": 18724
    },
    {
      "epoch": 1.7856597900390625e-05,
      "step": 18724,
      "training_step_time": 17.477777242660522
    },
    {
      "epoch": 1.7857551574707033e-05,
      "model_forward_time": 5.617349624633789,
      "step": 18725
    },
    {
      "epoch": 1.7857551574707033e-05,
      "step": 18725,
      "training_step_time": 17.11758065223694
    },
    {
      "epoch": 1.7858505249023437e-05,
      "model_forward_time": 5.610625267028809,
      "step": 18726
    },
    {
      "epoch": 1.7858505249023437e-05,
      "step": 18726,
      "training_step_time": 17.084654331207275
    },
    {
      "epoch": 1.7859458923339844e-05,
      "model_forward_time": 5.601020097732544,
      "step": 18727
    },
    {
      "epoch": 1.7859458923339844e-05,
      "step": 18727,
      "training_step_time": 17.07474112510681
    },
    {
      "epoch": 1.786041259765625e-05,
      "model_forward_time": 5.6062304973602295,
      "step": 18728
    },
    {
      "epoch": 1.786041259765625e-05,
      "step": 18728,
      "training_step_time": 17.079014539718628
    },
    {
      "epoch": 1.7861366271972655e-05,
      "model_forward_time": 5.60215950012207,
      "step": 18729
    },
    {
      "epoch": 1.7861366271972655e-05,
      "step": 18729,
      "training_step_time": 17.047778367996216
    },
    {
      "epoch": 1.7862319946289063e-05,
      "grad_norm": 1.0799648761749268,
      "learning_rate": 1e-05,
      "loss": 0.0651,
      "step": 18730
    },
    {
      "epoch": 1.7862319946289063e-05,
      "model_forward_time": 5.959431409835815,
      "step": 18730
    },
    {
      "dynamics_loss_avg": 0.049359765090048316,
      "epoch": 1.7862319946289063e-05,
      "step": 18730
    },
    {
      "action_loss_avg": 0.00693159990478307,
      "epoch": 1.7862319946289063e-05,
      "step": 18730
    },
    {
      "epoch": 1.7862319946289063e-05,
      "step": 18730,
      "training_step_time": 17.42568802833557
    },
    {
      "epoch": 1.786327362060547e-05,
      "model_forward_time": 5.608997583389282,
      "step": 18731
    },
    {
      "epoch": 1.786327362060547e-05,
      "step": 18731,
      "training_step_time": 17.07029891014099
    },
    {
      "epoch": 1.7864227294921874e-05,
      "model_forward_time": 5.602967262268066,
      "step": 18732
    },
    {
      "epoch": 1.7864227294921874e-05,
      "step": 18732,
      "training_step_time": 17.069506645202637
    },
    {
      "epoch": 1.7865180969238282e-05,
      "model_forward_time": 5.589367866516113,
      "step": 18733
    },
    {
      "epoch": 1.7865180969238282e-05,
      "step": 18733,
      "training_step_time": 17.01005482673645
    },
    {
      "epoch": 1.786613464355469e-05,
      "model_forward_time": 5.586863279342651,
      "step": 18734
    },
    {
      "epoch": 1.786613464355469e-05,
      "step": 18734,
      "training_step_time": 16.996291875839233
    },
    {
      "epoch": 1.7867088317871093e-05,
      "model_forward_time": 5.922630786895752,
      "step": 18735
    },
    {
      "epoch": 1.7867088317871093e-05,
      "step": 18735,
      "training_step_time": 17.355044841766357
    },
    {
      "epoch": 1.78680419921875e-05,
      "model_forward_time": 5.598863124847412,
      "step": 18736
    },
    {
      "epoch": 1.78680419921875e-05,
      "step": 18736,
      "training_step_time": 17.091212034225464
    },
    {
      "epoch": 1.7868995666503905e-05,
      "model_forward_time": 5.6120829582214355,
      "step": 18737
    },
    {
      "epoch": 1.7868995666503905e-05,
      "step": 18737,
      "training_step_time": 17.09423518180847
    },
    {
      "epoch": 1.7869949340820312e-05,
      "model_forward_time": 5.594118356704712,
      "step": 18738
    },
    {
      "epoch": 1.7869949340820312e-05,
      "step": 18738,
      "training_step_time": 17.026834964752197
    },
    {
      "epoch": 1.787090301513672e-05,
      "model_forward_time": 5.573688268661499,
      "step": 18739
    },
    {
      "epoch": 1.787090301513672e-05,
      "step": 18739,
      "training_step_time": 17.015047073364258
    },
    {
      "epoch": 1.7871856689453124e-05,
      "grad_norm": 0.8746214509010315,
      "learning_rate": 1e-05,
      "loss": 0.0571,
      "step": 18740
    },
    {
      "epoch": 1.7871856689453124e-05,
      "model_forward_time": 5.571944952011108,
      "step": 18740
    },
    {
      "dynamics_loss_avg": 0.05239988341927528,
      "epoch": 1.7871856689453124e-05,
      "step": 18740
    },
    {
      "action_loss_avg": 0.007310394267551601,
      "epoch": 1.7871856689453124e-05,
      "step": 18740
    },
    {
      "epoch": 1.7871856689453124e-05,
      "step": 18740,
      "training_step_time": 16.985886812210083
    },
    {
      "epoch": 1.787281036376953e-05,
      "model_forward_time": 5.951238632202148,
      "step": 18741
    },
    {
      "epoch": 1.787281036376953e-05,
      "step": 18741,
      "training_step_time": 17.365065097808838
    },
    {
      "epoch": 1.787376403808594e-05,
      "model_forward_time": 5.578962564468384,
      "step": 18742
    },
    {
      "epoch": 1.787376403808594e-05,
      "step": 18742,
      "training_step_time": 16.994465827941895
    },
    {
      "epoch": 1.7874717712402342e-05,
      "model_forward_time": 5.581048250198364,
      "step": 18743
    },
    {
      "epoch": 1.7874717712402342e-05,
      "step": 18743,
      "training_step_time": 17.02338171005249
    },
    {
      "epoch": 1.787567138671875e-05,
      "model_forward_time": 5.592111110687256,
      "step": 18744
    },
    {
      "epoch": 1.787567138671875e-05,
      "step": 18744,
      "training_step_time": 17.06137442588806
    },
    {
      "epoch": 1.7876625061035157e-05,
      "model_forward_time": 5.604578495025635,
      "step": 18745
    },
    {
      "epoch": 1.7876625061035157e-05,
      "step": 18745,
      "training_step_time": 17.116075038909912
    },
    {
      "epoch": 1.787757873535156e-05,
      "model_forward_time": 5.628206729888916,
      "step": 18746
    },
    {
      "epoch": 1.787757873535156e-05,
      "step": 18746,
      "training_step_time": 17.506277322769165
    },
    {
      "epoch": 1.787853240966797e-05,
      "model_forward_time": 5.614270210266113,
      "step": 18747
    },
    {
      "epoch": 1.787853240966797e-05,
      "step": 18747,
      "training_step_time": 17.11901092529297
    },
    {
      "epoch": 1.7879486083984376e-05,
      "model_forward_time": 5.608456134796143,
      "step": 18748
    },
    {
      "epoch": 1.7879486083984376e-05,
      "step": 18748,
      "training_step_time": 17.0861496925354
    },
    {
      "epoch": 1.788043975830078e-05,
      "model_forward_time": 5.598456144332886,
      "step": 18749
    },
    {
      "epoch": 1.788043975830078e-05,
      "step": 18749,
      "training_step_time": 17.06317448616028
    },
    {
      "epoch": 1.7881393432617188e-05,
      "grad_norm": 0.5399321913719177,
      "learning_rate": 1e-05,
      "loss": 0.0629,
      "step": 18750
    },
    {
      "epoch": 1.7881393432617188e-05,
      "model_forward_time": 5.59716534614563,
      "step": 18750
    },
    {
      "dynamics_loss_avg": 0.055195832438766954,
      "epoch": 1.7881393432617188e-05,
      "step": 18750
    },
    {
      "action_loss_avg": 0.007744654023554176,
      "epoch": 1.7881393432617188e-05,
      "step": 18750
    },
    {
      "epoch": 1.7881393432617188e-05,
      "step": 18750,
      "training_step_time": 17.049088954925537
    },
    {
      "epoch": 1.7882347106933595e-05,
      "model_forward_time": 5.598924160003662,
      "step": 18751
    },
    {
      "epoch": 1.7882347106933595e-05,
      "step": 18751,
      "training_step_time": 17.048347234725952
    },
    {
      "epoch": 1.788330078125e-05,
      "model_forward_time": 5.968605279922485,
      "step": 18752
    },
    {
      "epoch": 1.788330078125e-05,
      "step": 18752,
      "training_step_time": 17.435712337493896
    },
    {
      "epoch": 1.7884254455566406e-05,
      "model_forward_time": 5.5994553565979,
      "step": 18753
    },
    {
      "epoch": 1.7884254455566406e-05,
      "step": 18753,
      "training_step_time": 17.069765090942383
    },
    {
      "epoch": 1.7885208129882814e-05,
      "model_forward_time": 5.608052730560303,
      "step": 18754
    },
    {
      "epoch": 1.7885208129882814e-05,
      "step": 18754,
      "training_step_time": 17.09162402153015
    },
    {
      "epoch": 1.7886161804199218e-05,
      "model_forward_time": 5.605113744735718,
      "step": 18755
    },
    {
      "epoch": 1.7886161804199218e-05,
      "step": 18755,
      "training_step_time": 17.099434852600098
    },
    {
      "epoch": 1.7887115478515625e-05,
      "model_forward_time": 5.640536308288574,
      "step": 18756
    },
    {
      "epoch": 1.7887115478515625e-05,
      "step": 18756,
      "training_step_time": 17.140028715133667
    },
    {
      "epoch": 1.7888069152832033e-05,
      "model_forward_time": 5.61547589302063,
      "step": 18757
    },
    {
      "epoch": 1.7888069152832033e-05,
      "step": 18757,
      "training_step_time": 17.48084330558777
    },
    {
      "epoch": 1.7889022827148437e-05,
      "model_forward_time": 5.612082481384277,
      "step": 18758
    },
    {
      "epoch": 1.7889022827148437e-05,
      "step": 18758,
      "training_step_time": 17.066386461257935
    },
    {
      "epoch": 1.7889976501464844e-05,
      "model_forward_time": 5.593116521835327,
      "step": 18759
    },
    {
      "epoch": 1.7889976501464844e-05,
      "step": 18759,
      "training_step_time": 17.0393545627594
    },
    {
      "epoch": 1.789093017578125e-05,
      "grad_norm": 0.6132007837295532,
      "learning_rate": 1e-05,
      "loss": 0.0582,
      "step": 18760
    },
    {
      "epoch": 1.789093017578125e-05,
      "model_forward_time": 5.5855841636657715,
      "step": 18760
    },
    {
      "dynamics_loss_avg": 0.05308215543627739,
      "epoch": 1.789093017578125e-05,
      "step": 18760
    },
    {
      "action_loss_avg": 0.008554511319380254,
      "epoch": 1.789093017578125e-05,
      "step": 18760
    },
    {
      "epoch": 1.789093017578125e-05,
      "step": 18760,
      "training_step_time": 17.016728162765503
    },
    {
      "epoch": 1.7891883850097656e-05,
      "model_forward_time": 5.582535982131958,
      "step": 18761
    },
    {
      "epoch": 1.7891883850097656e-05,
      "step": 18761,
      "training_step_time": 17.028691291809082
    },
    {
      "epoch": 1.7892837524414063e-05,
      "model_forward_time": 5.598986625671387,
      "step": 18762
    },
    {
      "epoch": 1.7892837524414063e-05,
      "step": 18762,
      "training_step_time": 17.06973624229431
    },
    {
      "epoch": 1.789379119873047e-05,
      "model_forward_time": 5.980716943740845,
      "step": 18763
    },
    {
      "epoch": 1.789379119873047e-05,
      "step": 18763,
      "training_step_time": 17.44597864151001
    },
    {
      "epoch": 1.7894744873046874e-05,
      "model_forward_time": 5.596397161483765,
      "step": 18764
    },
    {
      "epoch": 1.7894744873046874e-05,
      "step": 18764,
      "training_step_time": 17.0312979221344
    },
    {
      "epoch": 1.7895698547363282e-05,
      "model_forward_time": 5.589378833770752,
      "step": 18765
    },
    {
      "epoch": 1.7895698547363282e-05,
      "step": 18765,
      "training_step_time": 17.01932168006897
    },
    {
      "epoch": 1.7896652221679686e-05,
      "model_forward_time": 5.577006816864014,
      "step": 18766
    },
    {
      "epoch": 1.7896652221679686e-05,
      "step": 18766,
      "training_step_time": 17.02884554862976
    },
    {
      "epoch": 1.7897605895996093e-05,
      "model_forward_time": 5.588860034942627,
      "step": 18767
    },
    {
      "epoch": 1.7897605895996093e-05,
      "step": 18767,
      "training_step_time": 17.01871395111084
    },
    {
      "epoch": 1.78985595703125e-05,
      "model_forward_time": 5.9541404247283936,
      "step": 18768
    },
    {
      "epoch": 1.78985595703125e-05,
      "step": 18768,
      "training_step_time": 17.39494800567627
    },
    {
      "epoch": 1.7899513244628905e-05,
      "model_forward_time": 5.601235866546631,
      "step": 18769
    },
    {
      "epoch": 1.7899513244628905e-05,
      "step": 18769,
      "training_step_time": 17.05548405647278
    },
    {
      "epoch": 1.7900466918945312e-05,
      "grad_norm": 0.6393249034881592,
      "learning_rate": 1e-05,
      "loss": 0.0843,
      "step": 18770
    },
    {
      "epoch": 1.7900466918945312e-05,
      "model_forward_time": 5.606436729431152,
      "step": 18770
    },
    {
      "dynamics_loss_avg": 0.07052158452570438,
      "epoch": 1.7900466918945312e-05,
      "step": 18770
    },
    {
      "action_loss_avg": 0.012755450303666293,
      "epoch": 1.7900466918945312e-05,
      "step": 18770
    },
    {
      "epoch": 1.7900466918945312e-05,
      "step": 18770,
      "training_step_time": 17.072340726852417
    },
    {
      "epoch": 1.790142059326172e-05,
      "model_forward_time": 5.6027209758758545,
      "step": 18771
    },
    {
      "epoch": 1.790142059326172e-05,
      "step": 18771,
      "training_step_time": 17.04683232307434
    },
    {
      "epoch": 1.7902374267578124e-05,
      "model_forward_time": 5.602037668228149,
      "step": 18772
    },
    {
      "epoch": 1.7902374267578124e-05,
      "step": 18772,
      "training_step_time": 17.02958369255066
    },
    {
      "epoch": 1.790332794189453e-05,
      "model_forward_time": 5.594651460647583,
      "step": 18773
    },
    {
      "epoch": 1.790332794189453e-05,
      "step": 18773,
      "training_step_time": 17.030022621154785
    },
    {
      "epoch": 1.790428161621094e-05,
      "model_forward_time": 5.970500469207764,
      "step": 18774
    },
    {
      "epoch": 1.790428161621094e-05,
      "step": 18774,
      "training_step_time": 17.439749002456665
    },
    {
      "epoch": 1.7905235290527342e-05,
      "model_forward_time": 5.607372760772705,
      "step": 18775
    },
    {
      "epoch": 1.7905235290527342e-05,
      "step": 18775,
      "training_step_time": 17.09459090232849
    },
    {
      "epoch": 1.790618896484375e-05,
      "model_forward_time": 5.601216554641724,
      "step": 18776
    },
    {
      "epoch": 1.790618896484375e-05,
      "step": 18776,
      "training_step_time": 17.08129382133484
    },
    {
      "epoch": 1.7907142639160157e-05,
      "model_forward_time": 5.61271595954895,
      "step": 18777
    },
    {
      "epoch": 1.7907142639160157e-05,
      "step": 18777,
      "training_step_time": 17.055542945861816
    },
    {
      "epoch": 1.790809631347656e-05,
      "model_forward_time": 5.582524299621582,
      "step": 18778
    },
    {
      "epoch": 1.790809631347656e-05,
      "step": 18778,
      "training_step_time": 17.018549919128418
    },
    {
      "epoch": 1.790904998779297e-05,
      "model_forward_time": 5.586199045181274,
      "step": 18779
    },
    {
      "epoch": 1.790904998779297e-05,
      "step": 18779,
      "training_step_time": 17.37749218940735
    },
    {
      "epoch": 1.7910003662109376e-05,
      "grad_norm": 0.6830182075500488,
      "learning_rate": 1e-05,
      "loss": 0.0542,
      "step": 18780
    },
    {
      "epoch": 1.7910003662109376e-05,
      "model_forward_time": 5.604839324951172,
      "step": 18780
    },
    {
      "dynamics_loss_avg": 0.04536104467697442,
      "epoch": 1.7910003662109376e-05,
      "step": 18780
    },
    {
      "action_loss_avg": 0.0074760219431482255,
      "epoch": 1.7910003662109376e-05,
      "step": 18780
    },
    {
      "epoch": 1.7910003662109376e-05,
      "step": 18780,
      "training_step_time": 17.10174560546875
    },
    {
      "epoch": 1.791095733642578e-05,
      "model_forward_time": 5.621435642242432,
      "step": 18781
    },
    {
      "epoch": 1.791095733642578e-05,
      "step": 18781,
      "training_step_time": 17.142372846603394
    },
    {
      "epoch": 1.7911911010742188e-05,
      "model_forward_time": 5.621858596801758,
      "step": 18782
    },
    {
      "epoch": 1.7911911010742188e-05,
      "step": 18782,
      "training_step_time": 17.117138385772705
    },
    {
      "epoch": 1.7912864685058595e-05,
      "model_forward_time": 5.625268936157227,
      "step": 18783
    },
    {
      "epoch": 1.7912864685058595e-05,
      "step": 18783,
      "training_step_time": 17.068777561187744
    },
    {
      "epoch": 1.7913818359375e-05,
      "model_forward_time": 5.613404989242554,
      "step": 18784
    },
    {
      "epoch": 1.7913818359375e-05,
      "step": 18784,
      "training_step_time": 17.067935705184937
    },
    {
      "epoch": 1.7914772033691406e-05,
      "model_forward_time": 6.099970102310181,
      "step": 18785
    },
    {
      "epoch": 1.7914772033691406e-05,
      "step": 18785,
      "training_step_time": 17.566166877746582
    },
    {
      "epoch": 1.7915725708007814e-05,
      "model_forward_time": 5.597636461257935,
      "step": 18786
    },
    {
      "epoch": 1.7915725708007814e-05,
      "step": 18786,
      "training_step_time": 17.080965280532837
    },
    {
      "epoch": 1.7916679382324218e-05,
      "model_forward_time": 5.621660232543945,
      "step": 18787
    },
    {
      "epoch": 1.7916679382324218e-05,
      "step": 18787,
      "training_step_time": 17.110329151153564
    },
    {
      "epoch": 1.7917633056640625e-05,
      "model_forward_time": 5.60393762588501,
      "step": 18788
    },
    {
      "epoch": 1.7917633056640625e-05,
      "step": 18788,
      "training_step_time": 17.05040168762207
    },
    {
      "epoch": 1.7918586730957033e-05,
      "model_forward_time": 5.590140342712402,
      "step": 18789
    },
    {
      "epoch": 1.7918586730957033e-05,
      "step": 18789,
      "training_step_time": 17.03326392173767
    },
    {
      "epoch": 1.7919540405273437e-05,
      "grad_norm": 0.9734488725662231,
      "learning_rate": 1e-05,
      "loss": 0.0546,
      "step": 18790
    },
    {
      "epoch": 1.7919540405273437e-05,
      "model_forward_time": 5.596591472625732,
      "step": 18790
    },
    {
      "dynamics_loss_avg": 0.0466469407081604,
      "epoch": 1.7919540405273437e-05,
      "step": 18790
    },
    {
      "action_loss_avg": 0.005919326434377581,
      "epoch": 1.7919540405273437e-05,
      "step": 18790
    },
    {
      "epoch": 1.7919540405273437e-05,
      "step": 18790,
      "training_step_time": 17.512935876846313
    },
    {
      "epoch": 1.7920494079589844e-05,
      "model_forward_time": 5.613913059234619,
      "step": 18791
    },
    {
      "epoch": 1.7920494079589844e-05,
      "step": 18791,
      "training_step_time": 17.11475944519043
    },
    {
      "epoch": 1.792144775390625e-05,
      "model_forward_time": 5.609496593475342,
      "step": 18792
    },
    {
      "epoch": 1.792144775390625e-05,
      "step": 18792,
      "training_step_time": 17.094067811965942
    },
    {
      "epoch": 1.7922401428222656e-05,
      "model_forward_time": 5.616403818130493,
      "step": 18793
    },
    {
      "epoch": 1.7922401428222656e-05,
      "step": 18793,
      "training_step_time": 17.11871075630188
    },
    {
      "epoch": 1.7923355102539063e-05,
      "model_forward_time": 5.617106676101685,
      "step": 18794
    },
    {
      "epoch": 1.7923355102539063e-05,
      "step": 18794,
      "training_step_time": 17.093524932861328
    },
    {
      "epoch": 1.792430877685547e-05,
      "model_forward_time": 5.580808877944946,
      "step": 18795
    },
    {
      "epoch": 1.792430877685547e-05,
      "step": 18795,
      "training_step_time": 17.002056121826172
    },
    {
      "epoch": 1.7925262451171874e-05,
      "model_forward_time": 5.9505228996276855,
      "step": 18796
    },
    {
      "epoch": 1.7925262451171874e-05,
      "step": 18796,
      "training_step_time": 17.388811826705933
    },
    {
      "epoch": 1.7926216125488282e-05,
      "model_forward_time": 5.558887720108032,
      "step": 18797
    },
    {
      "epoch": 1.7926216125488282e-05,
      "step": 18797,
      "training_step_time": 16.977952480316162
    },
    {
      "epoch": 1.7927169799804686e-05,
      "model_forward_time": 5.567501783370972,
      "step": 18798
    },
    {
      "epoch": 1.7927169799804686e-05,
      "step": 18798,
      "training_step_time": 16.986908435821533
    },
    {
      "epoch": 1.7928123474121093e-05,
      "model_forward_time": 5.58324670791626,
      "step": 18799
    },
    {
      "epoch": 1.7928123474121093e-05,
      "step": 18799,
      "training_step_time": 17.01003646850586
    },
    {
      "epoch": 1.79290771484375e-05,
      "grad_norm": 0.9817795157432556,
      "learning_rate": 1e-05,
      "loss": 0.0558,
      "step": 18800
    },
    {
      "epoch": 1.79290771484375e-05,
      "model_forward_time": 5.566304922103882,
      "step": 18800
    },
    {
      "dynamics_loss_avg": 0.04949358385056257,
      "epoch": 1.79290771484375e-05,
      "step": 18800
    },
    {
      "action_loss_avg": 0.005940331867896021,
      "epoch": 1.79290771484375e-05,
      "step": 18800
    },
    {
      "epoch": 1.79290771484375e-05,
      "step": 18800,
      "training_step_time": 17.003082990646362
    },
    {
      "epoch": 1.7930030822753905e-05,
      "model_forward_time": 5.94847846031189,
      "step": 18801
    },
    {
      "epoch": 1.7930030822753905e-05,
      "step": 18801,
      "training_step_time": 17.36898136138916
    },
    {
      "epoch": 1.7930984497070312e-05,
      "model_forward_time": 5.6041100025177,
      "step": 18802
    },
    {
      "epoch": 1.7930984497070312e-05,
      "step": 18802,
      "training_step_time": 17.03235650062561
    },
    {
      "epoch": 1.793193817138672e-05,
      "model_forward_time": 5.633365631103516,
      "step": 18803
    },
    {
      "epoch": 1.793193817138672e-05,
      "step": 18803,
      "training_step_time": 17.149383783340454
    },
    {
      "epoch": 1.7932891845703124e-05,
      "model_forward_time": 5.630038738250732,
      "step": 18804
    },
    {
      "epoch": 1.7932891845703124e-05,
      "step": 18804,
      "training_step_time": 17.166480779647827
    },
    {
      "epoch": 1.793384552001953e-05,
      "model_forward_time": 5.65288782119751,
      "step": 18805
    },
    {
      "epoch": 1.793384552001953e-05,
      "step": 18805,
      "training_step_time": 17.152417182922363
    },
    {
      "epoch": 1.793479919433594e-05,
      "model_forward_time": 5.610364198684692,
      "step": 18806
    },
    {
      "epoch": 1.793479919433594e-05,
      "step": 18806,
      "training_step_time": 17.088966846466064
    },
    {
      "epoch": 1.7935752868652342e-05,
      "model_forward_time": 5.99216890335083,
      "step": 18807
    },
    {
      "epoch": 1.7935752868652342e-05,
      "step": 18807,
      "training_step_time": 17.49015522003174
    },
    {
      "epoch": 1.793670654296875e-05,
      "model_forward_time": 5.6225972175598145,
      "step": 18808
    },
    {
      "epoch": 1.793670654296875e-05,
      "step": 18808,
      "training_step_time": 17.135140895843506
    },
    {
      "epoch": 1.7937660217285157e-05,
      "model_forward_time": 5.614781618118286,
      "step": 18809
    },
    {
      "epoch": 1.7937660217285157e-05,
      "step": 18809,
      "training_step_time": 17.108226776123047
    },
    {
      "epoch": 1.793861389160156e-05,
      "grad_norm": 0.5670498013496399,
      "learning_rate": 1e-05,
      "loss": 0.0723,
      "step": 18810
    },
    {
      "epoch": 1.793861389160156e-05,
      "model_forward_time": 5.615328311920166,
      "step": 18810
    },
    {
      "dynamics_loss_avg": 0.06403475292026997,
      "epoch": 1.793861389160156e-05,
      "step": 18810
    },
    {
      "action_loss_avg": 0.009044753527268768,
      "epoch": 1.793861389160156e-05,
      "step": 18810
    },
    {
      "epoch": 1.793861389160156e-05,
      "step": 18810,
      "training_step_time": 17.080482006072998
    },
    {
      "epoch": 1.793956756591797e-05,
      "model_forward_time": 5.603445291519165,
      "step": 18811
    },
    {
      "epoch": 1.793956756591797e-05,
      "step": 18811,
      "training_step_time": 17.04739737510681
    },
    {
      "epoch": 1.7940521240234376e-05,
      "model_forward_time": 5.602026462554932,
      "step": 18812
    },
    {
      "epoch": 1.7940521240234376e-05,
      "step": 18812,
      "training_step_time": 17.431314945220947
    },
    {
      "epoch": 1.794147491455078e-05,
      "model_forward_time": 5.602662086486816,
      "step": 18813
    },
    {
      "epoch": 1.794147491455078e-05,
      "step": 18813,
      "training_step_time": 17.078941822052002
    },
    {
      "epoch": 1.7942428588867188e-05,
      "model_forward_time": 5.608839750289917,
      "step": 18814
    },
    {
      "epoch": 1.7942428588867188e-05,
      "step": 18814,
      "training_step_time": 17.099764347076416
    },
    {
      "epoch": 1.7943382263183595e-05,
      "model_forward_time": 5.604702472686768,
      "step": 18815
    },
    {
      "epoch": 1.7943382263183595e-05,
      "step": 18815,
      "training_step_time": 17.086918354034424
    },
    {
      "epoch": 1.79443359375e-05,
      "model_forward_time": 5.619063138961792,
      "step": 18816
    },
    {
      "epoch": 1.79443359375e-05,
      "step": 18816,
      "training_step_time": 17.109044313430786
    },
    {
      "epoch": 1.7945289611816406e-05,
      "model_forward_time": 5.625155687332153,
      "step": 18817
    },
    {
      "epoch": 1.7945289611816406e-05,
      "step": 18817,
      "training_step_time": 17.12367296218872
    },
    {
      "epoch": 1.7946243286132814e-05,
      "model_forward_time": 5.982337713241577,
      "step": 18818
    },
    {
      "epoch": 1.7946243286132814e-05,
      "step": 18818,
      "training_step_time": 17.453673839569092
    },
    {
      "epoch": 1.7947196960449218e-05,
      "model_forward_time": 5.577479839324951,
      "step": 18819
    },
    {
      "epoch": 1.7947196960449218e-05,
      "step": 18819,
      "training_step_time": 17.01712656021118
    },
    {
      "epoch": 1.7948150634765625e-05,
      "grad_norm": 0.27579692006111145,
      "learning_rate": 1e-05,
      "loss": 0.0598,
      "step": 18820
    },
    {
      "epoch": 1.7948150634765625e-05,
      "model_forward_time": 5.562445878982544,
      "step": 18820
    },
    {
      "dynamics_loss_avg": 0.05144582167267799,
      "epoch": 1.7948150634765625e-05,
      "step": 18820
    },
    {
      "action_loss_avg": 0.0071699652587994935,
      "epoch": 1.7948150634765625e-05,
      "step": 18820
    },
    {
      "epoch": 1.7948150634765625e-05,
      "step": 18820,
      "training_step_time": 16.97309708595276
    },
    {
      "epoch": 1.7949104309082033e-05,
      "model_forward_time": 5.5587968826293945,
      "step": 18821
    },
    {
      "epoch": 1.7949104309082033e-05,
      "step": 18821,
      "training_step_time": 16.962859392166138
    },
    {
      "epoch": 1.7950057983398437e-05,
      "model_forward_time": 5.562779188156128,
      "step": 18822
    },
    {
      "epoch": 1.7950057983398437e-05,
      "step": 18822,
      "training_step_time": 16.984018325805664
    },
    {
      "epoch": 1.7951011657714844e-05,
      "model_forward_time": 5.557879686355591,
      "step": 18823
    },
    {
      "epoch": 1.7951011657714844e-05,
      "step": 18823,
      "training_step_time": 17.36290717124939
    },
    {
      "epoch": 1.795196533203125e-05,
      "model_forward_time": 5.557860612869263,
      "step": 18824
    },
    {
      "epoch": 1.795196533203125e-05,
      "step": 18824,
      "training_step_time": 16.980602264404297
    },
    {
      "epoch": 1.7952919006347656e-05,
      "model_forward_time": 5.570718050003052,
      "step": 18825
    },
    {
      "epoch": 1.7952919006347656e-05,
      "step": 18825,
      "training_step_time": 16.991464853286743
    },
    {
      "epoch": 1.7953872680664063e-05,
      "model_forward_time": 5.575756311416626,
      "step": 18826
    },
    {
      "epoch": 1.7953872680664063e-05,
      "step": 18826,
      "training_step_time": 17.012123584747314
    },
    {
      "epoch": 1.795482635498047e-05,
      "model_forward_time": 5.588546991348267,
      "step": 18827
    },
    {
      "epoch": 1.795482635498047e-05,
      "step": 18827,
      "training_step_time": 17.058529138565063
    },
    {
      "epoch": 1.7955780029296874e-05,
      "model_forward_time": 5.598778963088989,
      "step": 18828
    },
    {
      "epoch": 1.7955780029296874e-05,
      "step": 18828,
      "training_step_time": 17.07037901878357
    },
    {
      "epoch": 1.7956733703613282e-05,
      "model_forward_time": 5.968900203704834,
      "step": 18829
    },
    {
      "epoch": 1.7956733703613282e-05,
      "step": 18829,
      "training_step_time": 17.42914342880249
    },
    {
      "epoch": 1.7957687377929686e-05,
      "grad_norm": 1.4679720401763916,
      "learning_rate": 1e-05,
      "loss": 0.0578,
      "step": 18830
    },
    {
      "epoch": 1.7957687377929686e-05,
      "model_forward_time": 5.5978171825408936,
      "step": 18830
    },
    {
      "dynamics_loss_avg": 0.048210224881768224,
      "epoch": 1.7957687377929686e-05,
      "step": 18830
    },
    {
      "action_loss_avg": 0.008794064796529711,
      "epoch": 1.7957687377929686e-05,
      "step": 18830
    },
    {
      "epoch": 1.7957687377929686e-05,
      "step": 18830,
      "training_step_time": 17.039973497390747
    },
    {
      "epoch": 1.7958641052246093e-05,
      "model_forward_time": 5.597505331039429,
      "step": 18831
    },
    {
      "epoch": 1.7958641052246093e-05,
      "step": 18831,
      "training_step_time": 17.04660677909851
    },
    {
      "epoch": 1.79595947265625e-05,
      "model_forward_time": 5.590023994445801,
      "step": 18832
    },
    {
      "epoch": 1.79595947265625e-05,
      "step": 18832,
      "training_step_time": 17.033650398254395
    },
    {
      "epoch": 1.7960548400878905e-05,
      "model_forward_time": 5.595731735229492,
      "step": 18833
    },
    {
      "epoch": 1.7960548400878905e-05,
      "step": 18833,
      "training_step_time": 17.074270725250244
    },
    {
      "epoch": 1.7961502075195312e-05,
      "model_forward_time": 5.978866338729858,
      "step": 18834
    },
    {
      "epoch": 1.7961502075195312e-05,
      "step": 18834,
      "training_step_time": 17.481117010116577
    },
    {
      "epoch": 1.796245574951172e-05,
      "model_forward_time": 5.625794172286987,
      "step": 18835
    },
    {
      "epoch": 1.796245574951172e-05,
      "step": 18835,
      "training_step_time": 17.12862229347229
    },
    {
      "epoch": 1.7963409423828124e-05,
      "model_forward_time": 5.601308584213257,
      "step": 18836
    },
    {
      "epoch": 1.7963409423828124e-05,
      "step": 18836,
      "training_step_time": 17.051350116729736
    },
    {
      "epoch": 1.796436309814453e-05,
      "model_forward_time": 5.581468105316162,
      "step": 18837
    },
    {
      "epoch": 1.796436309814453e-05,
      "step": 18837,
      "training_step_time": 17.02726125717163
    },
    {
      "epoch": 1.796531677246094e-05,
      "model_forward_time": 5.587201356887817,
      "step": 18838
    },
    {
      "epoch": 1.796531677246094e-05,
      "step": 18838,
      "training_step_time": 17.032545804977417
    },
    {
      "epoch": 1.7966270446777342e-05,
      "model_forward_time": 5.600964307785034,
      "step": 18839
    },
    {
      "epoch": 1.7966270446777342e-05,
      "step": 18839,
      "training_step_time": 17.06647539138794
    },
    {
      "epoch": 1.796722412109375e-05,
      "grad_norm": 1.079580307006836,
      "learning_rate": 1e-05,
      "loss": 0.0646,
      "step": 18840
    },
    {
      "epoch": 1.796722412109375e-05,
      "model_forward_time": 5.971250534057617,
      "step": 18840
    },
    {
      "dynamics_loss_avg": 0.05445356387645006,
      "epoch": 1.796722412109375e-05,
      "step": 18840
    },
    {
      "action_loss_avg": 0.009804339986294508,
      "epoch": 1.796722412109375e-05,
      "step": 18840
    },
    {
      "epoch": 1.796722412109375e-05,
      "step": 18840,
      "training_step_time": 17.456870317459106
    },
    {
      "epoch": 1.7968177795410157e-05,
      "model_forward_time": 5.609148263931274,
      "step": 18841
    },
    {
      "epoch": 1.7968177795410157e-05,
      "step": 18841,
      "training_step_time": 17.101840496063232
    },
    {
      "epoch": 1.796913146972656e-05,
      "model_forward_time": 5.614917755126953,
      "step": 18842
    },
    {
      "epoch": 1.796913146972656e-05,
      "step": 18842,
      "training_step_time": 17.086406707763672
    },
    {
      "epoch": 1.797008514404297e-05,
      "model_forward_time": 5.589390277862549,
      "step": 18843
    },
    {
      "epoch": 1.797008514404297e-05,
      "step": 18843,
      "training_step_time": 17.027905464172363
    },
    {
      "epoch": 1.7971038818359376e-05,
      "model_forward_time": 5.587124586105347,
      "step": 18844
    },
    {
      "epoch": 1.7971038818359376e-05,
      "step": 18844,
      "training_step_time": 17.007322072982788
    },
    {
      "epoch": 1.797199249267578e-05,
      "model_forward_time": 5.570591926574707,
      "step": 18845
    },
    {
      "epoch": 1.797199249267578e-05,
      "step": 18845,
      "training_step_time": 17.352879285812378
    },
    {
      "epoch": 1.7972946166992188e-05,
      "model_forward_time": 5.574596643447876,
      "step": 18846
    },
    {
      "epoch": 1.7972946166992188e-05,
      "step": 18846,
      "training_step_time": 17.020352602005005
    },
    {
      "epoch": 1.7973899841308595e-05,
      "model_forward_time": 5.569087266921997,
      "step": 18847
    },
    {
      "epoch": 1.7973899841308595e-05,
      "step": 18847,
      "training_step_time": 16.977768182754517
    },
    {
      "epoch": 1.7974853515625e-05,
      "model_forward_time": 5.592172145843506,
      "step": 18848
    },
    {
      "epoch": 1.7974853515625e-05,
      "step": 18848,
      "training_step_time": 17.009024381637573
    },
    {
      "epoch": 1.7975807189941406e-05,
      "model_forward_time": 5.5884621143341064,
      "step": 18849
    },
    {
      "epoch": 1.7975807189941406e-05,
      "step": 18849,
      "training_step_time": 17.030303478240967
    },
    {
      "epoch": 1.7976760864257814e-05,
      "grad_norm": 0.5745117664337158,
      "learning_rate": 1e-05,
      "loss": 0.0469,
      "step": 18850
    },
    {
      "epoch": 1.7976760864257814e-05,
      "model_forward_time": 5.5964109897613525,
      "step": 18850
    },
    {
      "dynamics_loss_avg": 0.039574124943464996,
      "epoch": 1.7976760864257814e-05,
      "step": 18850
    },
    {
      "action_loss_avg": 0.008114184881560505,
      "epoch": 1.7976760864257814e-05,
      "step": 18850
    },
    {
      "epoch": 1.7976760864257814e-05,
      "step": 18850,
      "training_step_time": 17.07558822631836
    },
    {
      "epoch": 1.7977714538574218e-05,
      "model_forward_time": 5.985229730606079,
      "step": 18851
    },
    {
      "epoch": 1.7977714538574218e-05,
      "step": 18851,
      "training_step_time": 17.49005913734436
    },
    {
      "epoch": 1.7978668212890625e-05,
      "model_forward_time": 5.6266090869903564,
      "step": 18852
    },
    {
      "epoch": 1.7978668212890625e-05,
      "step": 18852,
      "training_step_time": 17.129095315933228
    },
    {
      "epoch": 1.7979621887207033e-05,
      "model_forward_time": 5.619763135910034,
      "step": 18853
    },
    {
      "epoch": 1.7979621887207033e-05,
      "step": 18853,
      "training_step_time": 17.1190025806427
    },
    {
      "epoch": 1.7980575561523437e-05,
      "model_forward_time": 5.622769832611084,
      "step": 18854
    },
    {
      "epoch": 1.7980575561523437e-05,
      "step": 18854,
      "training_step_time": 17.117628812789917
    },
    {
      "epoch": 1.7981529235839844e-05,
      "model_forward_time": 5.61051082611084,
      "step": 18855
    },
    {
      "epoch": 1.7981529235839844e-05,
      "step": 18855,
      "training_step_time": 17.08588695526123
    },
    {
      "epoch": 1.798248291015625e-05,
      "model_forward_time": 5.591696500778198,
      "step": 18856
    },
    {
      "epoch": 1.798248291015625e-05,
      "step": 18856,
      "training_step_time": 17.393853425979614
    },
    {
      "epoch": 1.7983436584472656e-05,
      "model_forward_time": 5.579568862915039,
      "step": 18857
    },
    {
      "epoch": 1.7983436584472656e-05,
      "step": 18857,
      "training_step_time": 17.02559471130371
    },
    {
      "epoch": 1.7984390258789063e-05,
      "model_forward_time": 5.584521055221558,
      "step": 18858
    },
    {
      "epoch": 1.7984390258789063e-05,
      "step": 18858,
      "training_step_time": 16.997413873672485
    },
    {
      "epoch": 1.798534393310547e-05,
      "model_forward_time": 5.607060194015503,
      "step": 18859
    },
    {
      "epoch": 1.798534393310547e-05,
      "step": 18859,
      "training_step_time": 17.07191228866577
    },
    {
      "epoch": 1.7986297607421874e-05,
      "grad_norm": 0.11930209398269653,
      "learning_rate": 1e-05,
      "loss": 0.0387,
      "step": 18860
    },
    {
      "epoch": 1.7986297607421874e-05,
      "model_forward_time": 5.604344606399536,
      "step": 18860
    },
    {
      "dynamics_loss_avg": 0.03320023436099291,
      "epoch": 1.7986297607421874e-05,
      "step": 18860
    },
    {
      "action_loss_avg": 0.0049613064038567245,
      "epoch": 1.7986297607421874e-05,
      "step": 18860
    },
    {
      "epoch": 1.7986297607421874e-05,
      "step": 18860,
      "training_step_time": 17.09867238998413
    },
    {
      "epoch": 1.7987251281738282e-05,
      "model_forward_time": 5.621329307556152,
      "step": 18861
    },
    {
      "epoch": 1.7987251281738282e-05,
      "step": 18861,
      "training_step_time": 17.130853176116943
    },
    {
      "epoch": 1.7988204956054686e-05,
      "model_forward_time": 6.169471263885498,
      "step": 18862
    },
    {
      "epoch": 1.7988204956054686e-05,
      "step": 18862,
      "training_step_time": 17.654744148254395
    },
    {
      "epoch": 1.7989158630371093e-05,
      "model_forward_time": 5.583899736404419,
      "step": 18863
    },
    {
      "epoch": 1.7989158630371093e-05,
      "step": 18863,
      "training_step_time": 16.993435382843018
    },
    {
      "epoch": 1.79901123046875e-05,
      "model_forward_time": 5.581555128097534,
      "step": 18864
    },
    {
      "epoch": 1.79901123046875e-05,
      "step": 18864,
      "training_step_time": 17.00539779663086
    },
    {
      "epoch": 1.7991065979003905e-05,
      "model_forward_time": 5.568813800811768,
      "step": 18865
    },
    {
      "epoch": 1.7991065979003905e-05,
      "step": 18865,
      "training_step_time": 17.013055324554443
    },
    {
      "epoch": 1.7992019653320312e-05,
      "model_forward_time": 5.564419507980347,
      "step": 18866
    },
    {
      "epoch": 1.7992019653320312e-05,
      "step": 18866,
      "training_step_time": 16.982420682907104
    },
    {
      "epoch": 1.799297332763672e-05,
      "model_forward_time": 5.956207990646362,
      "step": 18867
    },
    {
      "epoch": 1.799297332763672e-05,
      "step": 18867,
      "training_step_time": 17.372037649154663
    },
    {
      "epoch": 1.7993927001953124e-05,
      "model_forward_time": 5.566206932067871,
      "step": 18868
    },
    {
      "epoch": 1.7993927001953124e-05,
      "step": 18868,
      "training_step_time": 16.99750518798828
    },
    {
      "epoch": 1.799488067626953e-05,
      "model_forward_time": 5.578845024108887,
      "step": 18869
    },
    {
      "epoch": 1.799488067626953e-05,
      "step": 18869,
      "training_step_time": 17.02025032043457
    },
    {
      "epoch": 1.799583435058594e-05,
      "grad_norm": 0.4056434631347656,
      "learning_rate": 1e-05,
      "loss": 0.0661,
      "step": 18870
    },
    {
      "epoch": 1.799583435058594e-05,
      "model_forward_time": 5.594969749450684,
      "step": 18870
    },
    {
      "dynamics_loss_avg": 0.05955532807856798,
      "epoch": 1.799583435058594e-05,
      "step": 18870
    },
    {
      "action_loss_avg": 0.01103369880001992,
      "epoch": 1.799583435058594e-05,
      "step": 18870
    },
    {
      "epoch": 1.799583435058594e-05,
      "step": 18870,
      "training_step_time": 17.060215711593628
    },
    {
      "epoch": 1.7996788024902342e-05,
      "model_forward_time": 5.6065826416015625,
      "step": 18871
    },
    {
      "epoch": 1.7996788024902342e-05,
      "step": 18871,
      "training_step_time": 17.08194136619568
    },
    {
      "epoch": 1.799774169921875e-05,
      "model_forward_time": 5.617443561553955,
      "step": 18872
    },
    {
      "epoch": 1.799774169921875e-05,
      "step": 18872,
      "training_step_time": 17.11425757408142
    },
    {
      "epoch": 1.7998695373535157e-05,
      "model_forward_time": 6.055744171142578,
      "step": 18873
    },
    {
      "epoch": 1.7998695373535157e-05,
      "step": 18873,
      "training_step_time": 17.5201256275177
    },
    {
      "epoch": 1.799964904785156e-05,
      "model_forward_time": 5.58424711227417,
      "step": 18874
    },
    {
      "epoch": 1.799964904785156e-05,
      "step": 18874,
      "training_step_time": 17.01375389099121
    },
    {
      "epoch": 1.800060272216797e-05,
      "model_forward_time": 5.593586206436157,
      "step": 18875
    },
    {
      "epoch": 1.800060272216797e-05,
      "step": 18875,
      "training_step_time": 17.0189471244812
    },
    {
      "epoch": 1.8001556396484376e-05,
      "model_forward_time": 5.58129096031189,
      "step": 18876
    },
    {
      "epoch": 1.8001556396484376e-05,
      "step": 18876,
      "training_step_time": 17.01871395111084
    },
    {
      "epoch": 1.800251007080078e-05,
      "model_forward_time": 5.582570791244507,
      "step": 18877
    },
    {
      "epoch": 1.800251007080078e-05,
      "step": 18877,
      "training_step_time": 17.02294921875
    },
    {
      "epoch": 1.8003463745117188e-05,
      "model_forward_time": 5.581453561782837,
      "step": 18878
    },
    {
      "epoch": 1.8003463745117188e-05,
      "step": 18878,
      "training_step_time": 17.40501117706299
    },
    {
      "epoch": 1.8004417419433595e-05,
      "model_forward_time": 5.617902994155884,
      "step": 18879
    },
    {
      "epoch": 1.8004417419433595e-05,
      "step": 18879,
      "training_step_time": 17.144174575805664
    },
    {
      "epoch": 1.800537109375e-05,
      "grad_norm": 0.340163916349411,
      "learning_rate": 1e-05,
      "loss": 0.0556,
      "step": 18880
    },
    {
      "epoch": 1.800537109375e-05,
      "model_forward_time": 5.615495681762695,
      "step": 18880
    },
    {
      "dynamics_loss_avg": 0.04623440708965063,
      "epoch": 1.800537109375e-05,
      "step": 18880
    },
    {
      "action_loss_avg": 0.00561836005654186,
      "epoch": 1.800537109375e-05,
      "step": 18880
    },
    {
      "epoch": 1.800537109375e-05,
      "step": 18880,
      "training_step_time": 17.08834409713745
    },
    {
      "epoch": 1.8006324768066406e-05,
      "model_forward_time": 5.596382141113281,
      "step": 18881
    },
    {
      "epoch": 1.8006324768066406e-05,
      "step": 18881,
      "training_step_time": 17.0211181640625
    },
    {
      "epoch": 1.8007278442382814e-05,
      "model_forward_time": 5.5810253620147705,
      "step": 18882
    },
    {
      "epoch": 1.8007278442382814e-05,
      "step": 18882,
      "training_step_time": 17.002819299697876
    },
    {
      "epoch": 1.8008232116699218e-05,
      "model_forward_time": 5.574039697647095,
      "step": 18883
    },
    {
      "epoch": 1.8008232116699218e-05,
      "step": 18883,
      "training_step_time": 17.005199432373047
    },
    {
      "epoch": 1.8009185791015625e-05,
      "model_forward_time": 5.9642014503479,
      "step": 18884
    },
    {
      "epoch": 1.8009185791015625e-05,
      "step": 18884,
      "training_step_time": 17.40564751625061
    },
    {
      "epoch": 1.8010139465332033e-05,
      "model_forward_time": 5.57094407081604,
      "step": 18885
    },
    {
      "epoch": 1.8010139465332033e-05,
      "step": 18885,
      "training_step_time": 16.987204790115356
    },
    {
      "epoch": 1.8011093139648437e-05,
      "model_forward_time": 5.588629722595215,
      "step": 18886
    },
    {
      "epoch": 1.8011093139648437e-05,
      "step": 18886,
      "training_step_time": 17.034355401992798
    },
    {
      "epoch": 1.8012046813964844e-05,
      "model_forward_time": 5.598948955535889,
      "step": 18887
    },
    {
      "epoch": 1.8012046813964844e-05,
      "step": 18887,
      "training_step_time": 17.046396255493164
    },
    {
      "epoch": 1.801300048828125e-05,
      "model_forward_time": 5.607864618301392,
      "step": 18888
    },
    {
      "epoch": 1.801300048828125e-05,
      "step": 18888,
      "training_step_time": 17.08675765991211
    },
    {
      "epoch": 1.8013954162597656e-05,
      "model_forward_time": 5.608259677886963,
      "step": 18889
    },
    {
      "epoch": 1.8013954162597656e-05,
      "step": 18889,
      "training_step_time": 17.465867280960083
    },
    {
      "epoch": 1.8014907836914063e-05,
      "grad_norm": 0.3691265881061554,
      "learning_rate": 1e-05,
      "loss": 0.0638,
      "step": 18890
    },
    {
      "epoch": 1.8014907836914063e-05,
      "model_forward_time": 5.607639312744141,
      "step": 18890
    },
    {
      "dynamics_loss_avg": 0.05818251650780439,
      "epoch": 1.8014907836914063e-05,
      "step": 18890
    },
    {
      "action_loss_avg": 0.006988315924536437,
      "epoch": 1.8014907836914063e-05,
      "step": 18890
    },
    {
      "epoch": 1.8014907836914063e-05,
      "step": 18890,
      "training_step_time": 17.1061794757843
    },
    {
      "epoch": 1.8015861511230467e-05,
      "model_forward_time": 5.614837884902954,
      "step": 18891
    },
    {
      "epoch": 1.8015861511230467e-05,
      "step": 18891,
      "training_step_time": 17.116891384124756
    },
    {
      "epoch": 1.8016815185546874e-05,
      "model_forward_time": 5.609538555145264,
      "step": 18892
    },
    {
      "epoch": 1.8016815185546874e-05,
      "step": 18892,
      "training_step_time": 17.08348250389099
    },
    {
      "epoch": 1.8017768859863282e-05,
      "model_forward_time": 5.6000471115112305,
      "step": 18893
    },
    {
      "epoch": 1.8017768859863282e-05,
      "step": 18893,
      "training_step_time": 17.033525943756104
    },
    {
      "epoch": 1.8018722534179686e-05,
      "model_forward_time": 5.572251558303833,
      "step": 18894
    },
    {
      "epoch": 1.8018722534179686e-05,
      "step": 18894,
      "training_step_time": 16.976263999938965
    },
    {
      "epoch": 1.8019676208496093e-05,
      "model_forward_time": 5.944574594497681,
      "step": 18895
    },
    {
      "epoch": 1.8019676208496093e-05,
      "step": 18895,
      "training_step_time": 17.3834228515625
    },
    {
      "epoch": 1.80206298828125e-05,
      "model_forward_time": 5.566216230392456,
      "step": 18896
    },
    {
      "epoch": 1.80206298828125e-05,
      "step": 18896,
      "training_step_time": 16.98833966255188
    },
    {
      "epoch": 1.8021583557128905e-05,
      "model_forward_time": 5.573888301849365,
      "step": 18897
    },
    {
      "epoch": 1.8021583557128905e-05,
      "step": 18897,
      "training_step_time": 16.97861123085022
    },
    {
      "epoch": 1.8022537231445312e-05,
      "model_forward_time": 5.573539972305298,
      "step": 18898
    },
    {
      "epoch": 1.8022537231445312e-05,
      "step": 18898,
      "training_step_time": 17.006235599517822
    },
    {
      "epoch": 1.802349090576172e-05,
      "model_forward_time": 5.585721731185913,
      "step": 18899
    },
    {
      "epoch": 1.802349090576172e-05,
      "step": 18899,
      "training_step_time": 17.034966945648193
    },
    {
      "epoch": 1.8024444580078124e-05,
      "grad_norm": 1.0779186487197876,
      "learning_rate": 1e-05,
      "loss": 0.0663,
      "step": 18900
    },
    {
      "epoch": 1.8024444580078124e-05,
      "model_forward_time": 5.952600717544556,
      "step": 18900
    },
    {
      "dynamics_loss_avg": 0.055928238481283185,
      "epoch": 1.8024444580078124e-05,
      "step": 18900
    },
    {
      "action_loss_avg": 0.00732168722897768,
      "epoch": 1.8024444580078124e-05,
      "step": 18900
    },
    {
      "epoch": 1.8024444580078124e-05,
      "step": 18900,
      "training_step_time": 17.383557081222534
    },
    {
      "epoch": 1.802539825439453e-05,
      "model_forward_time": 5.599143743515015,
      "step": 18901
    },
    {
      "epoch": 1.802539825439453e-05,
      "step": 18901,
      "training_step_time": 17.094279527664185
    },
    {
      "epoch": 1.802635192871094e-05,
      "model_forward_time": 5.61357045173645,
      "step": 18902
    },
    {
      "epoch": 1.802635192871094e-05,
      "step": 18902,
      "training_step_time": 17.140723943710327
    },
    {
      "epoch": 1.8027305603027343e-05,
      "model_forward_time": 5.611963748931885,
      "step": 18903
    },
    {
      "epoch": 1.8027305603027343e-05,
      "step": 18903,
      "training_step_time": 17.111194849014282
    },
    {
      "epoch": 1.802825927734375e-05,
      "model_forward_time": 5.611805200576782,
      "step": 18904
    },
    {
      "epoch": 1.802825927734375e-05,
      "step": 18904,
      "training_step_time": 17.095786809921265
    },
    {
      "epoch": 1.8029212951660157e-05,
      "model_forward_time": 5.608400821685791,
      "step": 18905
    },
    {
      "epoch": 1.8029212951660157e-05,
      "step": 18905,
      "training_step_time": 17.07576060295105
    },
    {
      "epoch": 1.803016662597656e-05,
      "model_forward_time": 5.972726106643677,
      "step": 18906
    },
    {
      "epoch": 1.803016662597656e-05,
      "step": 18906,
      "training_step_time": 17.438490867614746
    },
    {
      "epoch": 1.803112030029297e-05,
      "model_forward_time": 5.593203544616699,
      "step": 18907
    },
    {
      "epoch": 1.803112030029297e-05,
      "step": 18907,
      "training_step_time": 17.046064615249634
    },
    {
      "epoch": 1.8032073974609376e-05,
      "model_forward_time": 5.628913164138794,
      "step": 18908
    },
    {
      "epoch": 1.8032073974609376e-05,
      "step": 18908,
      "training_step_time": 17.067307710647583
    },
    {
      "epoch": 1.803302764892578e-05,
      "model_forward_time": 5.5971550941467285,
      "step": 18909
    },
    {
      "epoch": 1.803302764892578e-05,
      "step": 18909,
      "training_step_time": 17.037757635116577
    },
    {
      "epoch": 1.8033981323242188e-05,
      "grad_norm": 0.9211951494216919,
      "learning_rate": 1e-05,
      "loss": 0.0621,
      "step": 18910
    },
    {
      "epoch": 1.8033981323242188e-05,
      "model_forward_time": 5.587718486785889,
      "step": 18910
    },
    {
      "dynamics_loss_avg": 0.05963444635272026,
      "epoch": 1.8033981323242188e-05,
      "step": 18910
    },
    {
      "action_loss_avg": 0.00967821681406349,
      "epoch": 1.8033981323242188e-05,
      "step": 18910
    },
    {
      "epoch": 1.8033981323242188e-05,
      "step": 18910,
      "training_step_time": 17.062402486801147
    },
    {
      "epoch": 1.8034934997558595e-05,
      "model_forward_time": 5.616273403167725,
      "step": 18911
    },
    {
      "epoch": 1.8034934997558595e-05,
      "step": 18911,
      "training_step_time": 17.47012495994568
    },
    {
      "epoch": 1.8035888671875e-05,
      "model_forward_time": 5.607740640640259,
      "step": 18912
    },
    {
      "epoch": 1.8035888671875e-05,
      "step": 18912,
      "training_step_time": 17.08912682533264
    },
    {
      "epoch": 1.8036842346191406e-05,
      "model_forward_time": 5.597838401794434,
      "step": 18913
    },
    {
      "epoch": 1.8036842346191406e-05,
      "step": 18913,
      "training_step_time": 17.058993339538574
    },
    {
      "epoch": 1.8037796020507814e-05,
      "model_forward_time": 5.593739032745361,
      "step": 18914
    },
    {
      "epoch": 1.8037796020507814e-05,
      "step": 18914,
      "training_step_time": 17.036977767944336
    },
    {
      "epoch": 1.8038749694824218e-05,
      "model_forward_time": 5.600341081619263,
      "step": 18915
    },
    {
      "epoch": 1.8038749694824218e-05,
      "step": 18915,
      "training_step_time": 17.041199445724487
    },
    {
      "epoch": 1.8039703369140625e-05,
      "model_forward_time": 5.587564945220947,
      "step": 18916
    },
    {
      "epoch": 1.8039703369140625e-05,
      "step": 18916,
      "training_step_time": 17.026720762252808
    },
    {
      "epoch": 1.8040657043457033e-05,
      "model_forward_time": 5.962884426116943,
      "step": 18917
    },
    {
      "epoch": 1.8040657043457033e-05,
      "step": 18917,
      "training_step_time": 17.407712697982788
    },
    {
      "epoch": 1.8041610717773437e-05,
      "model_forward_time": 5.59392523765564,
      "step": 18918
    },
    {
      "epoch": 1.8041610717773437e-05,
      "step": 18918,
      "training_step_time": 17.065917491912842
    },
    {
      "epoch": 1.8042564392089844e-05,
      "model_forward_time": 5.603877067565918,
      "step": 18919
    },
    {
      "epoch": 1.8042564392089844e-05,
      "step": 18919,
      "training_step_time": 17.0778591632843
    },
    {
      "epoch": 1.804351806640625e-05,
      "grad_norm": 0.7625292539596558,
      "learning_rate": 1e-05,
      "loss": 0.058,
      "step": 18920
    },
    {
      "epoch": 1.804351806640625e-05,
      "model_forward_time": 5.601584434509277,
      "step": 18920
    },
    {
      "dynamics_loss_avg": 0.05563501063734293,
      "epoch": 1.804351806640625e-05,
      "step": 18920
    },
    {
      "action_loss_avg": 0.00672402058262378,
      "epoch": 1.804351806640625e-05,
      "step": 18920
    },
    {
      "epoch": 1.804351806640625e-05,
      "step": 18920,
      "training_step_time": 17.074917554855347
    },
    {
      "epoch": 1.8044471740722656e-05,
      "model_forward_time": 5.6054582595825195,
      "step": 18921
    },
    {
      "epoch": 1.8044471740722656e-05,
      "step": 18921,
      "training_step_time": 17.056787252426147
    },
    {
      "epoch": 1.8045425415039063e-05,
      "model_forward_time": 5.5901782512664795,
      "step": 18922
    },
    {
      "epoch": 1.8045425415039063e-05,
      "step": 18922,
      "training_step_time": 17.38818049430847
    },
    {
      "epoch": 1.8046379089355467e-05,
      "model_forward_time": 5.593608617782593,
      "step": 18923
    },
    {
      "epoch": 1.8046379089355467e-05,
      "step": 18923,
      "training_step_time": 17.036390781402588
    },
    {
      "epoch": 1.8047332763671875e-05,
      "model_forward_time": 5.590407848358154,
      "step": 18924
    },
    {
      "epoch": 1.8047332763671875e-05,
      "step": 18924,
      "training_step_time": 17.053727626800537
    },
    {
      "epoch": 1.8048286437988282e-05,
      "model_forward_time": 5.598409175872803,
      "step": 18925
    },
    {
      "epoch": 1.8048286437988282e-05,
      "step": 18925,
      "training_step_time": 17.08290672302246
    },
    {
      "epoch": 1.8049240112304686e-05,
      "model_forward_time": 5.617347955703735,
      "step": 18926
    },
    {
      "epoch": 1.8049240112304686e-05,
      "step": 18926,
      "training_step_time": 17.111709594726562
    },
    {
      "epoch": 1.8050193786621093e-05,
      "model_forward_time": 5.60631537437439,
      "step": 18927
    },
    {
      "epoch": 1.8050193786621093e-05,
      "step": 18927,
      "training_step_time": 17.091376304626465
    },
    {
      "epoch": 1.80511474609375e-05,
      "model_forward_time": 5.9771199226379395,
      "step": 18928
    },
    {
      "epoch": 1.80511474609375e-05,
      "step": 18928,
      "training_step_time": 17.44791269302368
    },
    {
      "epoch": 1.8052101135253905e-05,
      "model_forward_time": 5.5991504192352295,
      "step": 18929
    },
    {
      "epoch": 1.8052101135253905e-05,
      "step": 18929,
      "training_step_time": 17.02416229248047
    },
    {
      "epoch": 1.8053054809570312e-05,
      "grad_norm": 0.5783951282501221,
      "learning_rate": 1e-05,
      "loss": 0.073,
      "step": 18930
    },
    {
      "epoch": 1.8053054809570312e-05,
      "model_forward_time": 5.581691741943359,
      "step": 18930
    },
    {
      "dynamics_loss_avg": 0.053376142866909504,
      "epoch": 1.8053054809570312e-05,
      "step": 18930
    },
    {
      "action_loss_avg": 0.009441052004694939,
      "epoch": 1.8053054809570312e-05,
      "step": 18930
    },
    {
      "epoch": 1.8053054809570312e-05,
      "step": 18930,
      "training_step_time": 17.021697998046875
    },
    {
      "epoch": 1.805400848388672e-05,
      "model_forward_time": 5.595530033111572,
      "step": 18931
    },
    {
      "epoch": 1.805400848388672e-05,
      "step": 18931,
      "training_step_time": 17.046791076660156
    },
    {
      "epoch": 1.8054962158203124e-05,
      "model_forward_time": 5.598147392272949,
      "step": 18932
    },
    {
      "epoch": 1.8054962158203124e-05,
      "step": 18932,
      "training_step_time": 17.06686782836914
    },
    {
      "epoch": 1.805591583251953e-05,
      "model_forward_time": 5.964697360992432,
      "step": 18933
    },
    {
      "epoch": 1.805591583251953e-05,
      "step": 18933,
      "training_step_time": 17.433884143829346
    },
    {
      "epoch": 1.805686950683594e-05,
      "model_forward_time": 5.598355293273926,
      "step": 18934
    },
    {
      "epoch": 1.805686950683594e-05,
      "step": 18934,
      "training_step_time": 17.065329790115356
    },
    {
      "epoch": 1.8057823181152343e-05,
      "model_forward_time": 5.612078428268433,
      "step": 18935
    },
    {
      "epoch": 1.8057823181152343e-05,
      "step": 18935,
      "training_step_time": 17.045132398605347
    },
    {
      "epoch": 1.805877685546875e-05,
      "model_forward_time": 5.591371059417725,
      "step": 18936
    },
    {
      "epoch": 1.805877685546875e-05,
      "step": 18936,
      "training_step_time": 17.0289409160614
    },
    {
      "epoch": 1.8059730529785157e-05,
      "model_forward_time": 5.588599681854248,
      "step": 18937
    },
    {
      "epoch": 1.8059730529785157e-05,
      "step": 18937,
      "training_step_time": 17.043514728546143
    },
    {
      "epoch": 1.806068420410156e-05,
      "model_forward_time": 5.603758811950684,
      "step": 18938
    },
    {
      "epoch": 1.806068420410156e-05,
      "step": 18938,
      "training_step_time": 17.10046625137329
    },
    {
      "epoch": 1.806163787841797e-05,
      "model_forward_time": 5.986863374710083,
      "step": 18939
    },
    {
      "epoch": 1.806163787841797e-05,
      "step": 18939,
      "training_step_time": 17.459195852279663
    },
    {
      "epoch": 1.8062591552734376e-05,
      "grad_norm": 0.5750161409378052,
      "learning_rate": 1e-05,
      "loss": 0.0529,
      "step": 18940
    },
    {
      "epoch": 1.8062591552734376e-05,
      "model_forward_time": 5.60512638092041,
      "step": 18940
    },
    {
      "dynamics_loss_avg": 0.049092472763732076,
      "epoch": 1.8062591552734376e-05,
      "step": 18940
    },
    {
      "action_loss_avg": 0.0075914143526460975,
      "epoch": 1.8062591552734376e-05,
      "step": 18940
    },
    {
      "epoch": 1.8062591552734376e-05,
      "step": 18940,
      "training_step_time": 17.053553819656372
    },
    {
      "epoch": 1.806354522705078e-05,
      "model_forward_time": 5.586306571960449,
      "step": 18941
    },
    {
      "epoch": 1.806354522705078e-05,
      "step": 18941,
      "training_step_time": 17.049840927124023
    },
    {
      "epoch": 1.8064498901367188e-05,
      "model_forward_time": 5.606362342834473,
      "step": 18942
    },
    {
      "epoch": 1.8064498901367188e-05,
      "step": 18942,
      "training_step_time": 17.058874368667603
    },
    {
      "epoch": 1.8065452575683595e-05,
      "model_forward_time": 5.5963544845581055,
      "step": 18943
    },
    {
      "epoch": 1.8065452575683595e-05,
      "step": 18943,
      "training_step_time": 17.06741452217102
    },
    {
      "epoch": 1.806640625e-05,
      "model_forward_time": 5.592266082763672,
      "step": 18944
    },
    {
      "epoch": 1.806640625e-05,
      "step": 18944,
      "training_step_time": 17.43605065345764
    },
    {
      "epoch": 1.8067359924316407e-05,
      "model_forward_time": 5.620008945465088,
      "step": 18945
    },
    {
      "epoch": 1.8067359924316407e-05,
      "step": 18945,
      "training_step_time": 17.132808923721313
    },
    {
      "epoch": 1.8068313598632814e-05,
      "model_forward_time": 5.6250622272491455,
      "step": 18946
    },
    {
      "epoch": 1.8068313598632814e-05,
      "step": 18946,
      "training_step_time": 17.122339725494385
    },
    {
      "epoch": 1.8069267272949218e-05,
      "model_forward_time": 5.6129796504974365,
      "step": 18947
    },
    {
      "epoch": 1.8069267272949218e-05,
      "step": 18947,
      "training_step_time": 17.102120637893677
    },
    {
      "epoch": 1.8070220947265625e-05,
      "model_forward_time": 5.60788631439209,
      "step": 18948
    },
    {
      "epoch": 1.8070220947265625e-05,
      "step": 18948,
      "training_step_time": 17.105581998825073
    },
    {
      "epoch": 1.8071174621582033e-05,
      "model_forward_time": 5.626511812210083,
      "step": 18949
    },
    {
      "epoch": 1.8071174621582033e-05,
      "step": 18949,
      "training_step_time": 17.113340139389038
    },
    {
      "epoch": 1.8072128295898437e-05,
      "grad_norm": 0.9702653288841248,
      "learning_rate": 1e-05,
      "loss": 0.054,
      "step": 18950
    },
    {
      "epoch": 1.8072128295898437e-05,
      "model_forward_time": 5.971383571624756,
      "step": 18950
    },
    {
      "dynamics_loss_avg": 0.04731006734073162,
      "epoch": 1.8072128295898437e-05,
      "step": 18950
    },
    {
      "action_loss_avg": 0.005595537670888007,
      "epoch": 1.8072128295898437e-05,
      "step": 18950
    },
    {
      "epoch": 1.8072128295898437e-05,
      "step": 18950,
      "training_step_time": 17.4411518573761
    },
    {
      "epoch": 1.8073081970214844e-05,
      "model_forward_time": 5.5959694385528564,
      "step": 18951
    },
    {
      "epoch": 1.8073081970214844e-05,
      "step": 18951,
      "training_step_time": 17.036914587020874
    },
    {
      "epoch": 1.807403564453125e-05,
      "model_forward_time": 5.582316160202026,
      "step": 18952
    },
    {
      "epoch": 1.807403564453125e-05,
      "step": 18952,
      "training_step_time": 17.02344036102295
    },
    {
      "epoch": 1.8074989318847656e-05,
      "model_forward_time": 5.5937628746032715,
      "step": 18953
    },
    {
      "epoch": 1.8074989318847656e-05,
      "step": 18953,
      "training_step_time": 17.05699920654297
    },
    {
      "epoch": 1.8075942993164063e-05,
      "model_forward_time": 5.601761817932129,
      "step": 18954
    },
    {
      "epoch": 1.8075942993164063e-05,
      "step": 18954,
      "training_step_time": 17.074559688568115
    },
    {
      "epoch": 1.8076896667480467e-05,
      "model_forward_time": 5.607525587081909,
      "step": 18955
    },
    {
      "epoch": 1.8076896667480467e-05,
      "step": 18955,
      "training_step_time": 17.44779133796692
    },
    {
      "epoch": 1.8077850341796875e-05,
      "model_forward_time": 5.600733995437622,
      "step": 18956
    },
    {
      "epoch": 1.8077850341796875e-05,
      "step": 18956,
      "training_step_time": 17.0548837184906
    },
    {
      "epoch": 1.8078804016113282e-05,
      "model_forward_time": 5.578235387802124,
      "step": 18957
    },
    {
      "epoch": 1.8078804016113282e-05,
      "step": 18957,
      "training_step_time": 17.016855478286743
    },
    {
      "epoch": 1.8079757690429686e-05,
      "model_forward_time": 5.585173845291138,
      "step": 18958
    },
    {
      "epoch": 1.8079757690429686e-05,
      "step": 18958,
      "training_step_time": 17.005786657333374
    },
    {
      "epoch": 1.8080711364746093e-05,
      "model_forward_time": 5.592926263809204,
      "step": 18959
    },
    {
      "epoch": 1.8080711364746093e-05,
      "step": 18959,
      "training_step_time": 17.033246755599976
    },
    {
      "epoch": 1.80816650390625e-05,
      "grad_norm": 0.602293074131012,
      "learning_rate": 1e-05,
      "loss": 0.0592,
      "step": 18960
    },
    {
      "epoch": 1.80816650390625e-05,
      "model_forward_time": 5.607003450393677,
      "step": 18960
    },
    {
      "dynamics_loss_avg": 0.05443634558469057,
      "epoch": 1.80816650390625e-05,
      "step": 18960
    },
    {
      "action_loss_avg": 0.007549610373098403,
      "epoch": 1.80816650390625e-05,
      "step": 18960
    },
    {
      "epoch": 1.80816650390625e-05,
      "step": 18960,
      "training_step_time": 17.081827640533447
    },
    {
      "epoch": 1.8082618713378905e-05,
      "model_forward_time": 5.966863393783569,
      "step": 18961
    },
    {
      "epoch": 1.8082618713378905e-05,
      "step": 18961,
      "training_step_time": 17.46314811706543
    },
    {
      "epoch": 1.8083572387695312e-05,
      "model_forward_time": 5.601664066314697,
      "step": 18962
    },
    {
      "epoch": 1.8083572387695312e-05,
      "step": 18962,
      "training_step_time": 17.07443070411682
    },
    {
      "epoch": 1.808452606201172e-05,
      "model_forward_time": 5.6057868003845215,
      "step": 18963
    },
    {
      "epoch": 1.808452606201172e-05,
      "step": 18963,
      "training_step_time": 17.081847429275513
    },
    {
      "epoch": 1.8085479736328124e-05,
      "model_forward_time": 5.599816560745239,
      "step": 18964
    },
    {
      "epoch": 1.8085479736328124e-05,
      "step": 18964,
      "training_step_time": 17.083885431289673
    },
    {
      "epoch": 1.808643341064453e-05,
      "model_forward_time": 5.609015941619873,
      "step": 18965
    },
    {
      "epoch": 1.808643341064453e-05,
      "step": 18965,
      "training_step_time": 17.092954397201538
    },
    {
      "epoch": 1.808738708496094e-05,
      "model_forward_time": 5.968223810195923,
      "step": 18966
    },
    {
      "epoch": 1.808738708496094e-05,
      "step": 18966,
      "training_step_time": 17.442278385162354
    },
    {
      "epoch": 1.8088340759277343e-05,
      "model_forward_time": 5.599529981613159,
      "step": 18967
    },
    {
      "epoch": 1.8088340759277343e-05,
      "step": 18967,
      "training_step_time": 17.062222242355347
    },
    {
      "epoch": 1.808929443359375e-05,
      "model_forward_time": 5.593486070632935,
      "step": 18968
    },
    {
      "epoch": 1.808929443359375e-05,
      "step": 18968,
      "training_step_time": 17.058405876159668
    },
    {
      "epoch": 1.8090248107910157e-05,
      "model_forward_time": 5.59683632850647,
      "step": 18969
    },
    {
      "epoch": 1.8090248107910157e-05,
      "step": 18969,
      "training_step_time": 17.03393316268921
    },
    {
      "epoch": 1.809120178222656e-05,
      "grad_norm": 1.5098425149917603,
      "learning_rate": 1e-05,
      "loss": 0.0608,
      "step": 18970
    },
    {
      "epoch": 1.809120178222656e-05,
      "model_forward_time": 5.594627380371094,
      "step": 18970
    },
    {
      "dynamics_loss_avg": 0.04737369157373905,
      "epoch": 1.809120178222656e-05,
      "step": 18970
    },
    {
      "action_loss_avg": 0.007272489019669592,
      "epoch": 1.809120178222656e-05,
      "step": 18970
    },
    {
      "epoch": 1.809120178222656e-05,
      "step": 18970,
      "training_step_time": 17.05656409263611
    },
    {
      "epoch": 1.809215545654297e-05,
      "model_forward_time": 5.60029673576355,
      "step": 18971
    },
    {
      "epoch": 1.809215545654297e-05,
      "step": 18971,
      "training_step_time": 17.063252925872803
    },
    {
      "epoch": 1.8093109130859376e-05,
      "model_forward_time": 5.970387935638428,
      "step": 18972
    },
    {
      "epoch": 1.8093109130859376e-05,
      "step": 18972,
      "training_step_time": 17.45201349258423
    },
    {
      "epoch": 1.809406280517578e-05,
      "model_forward_time": 5.610354900360107,
      "step": 18973
    },
    {
      "epoch": 1.809406280517578e-05,
      "step": 18973,
      "training_step_time": 17.082778692245483
    },
    {
      "epoch": 1.8095016479492188e-05,
      "model_forward_time": 5.603872299194336,
      "step": 18974
    },
    {
      "epoch": 1.8095016479492188e-05,
      "step": 18974,
      "training_step_time": 17.065433263778687
    },
    {
      "epoch": 1.8095970153808595e-05,
      "model_forward_time": 5.596411466598511,
      "step": 18975
    },
    {
      "epoch": 1.8095970153808595e-05,
      "step": 18975,
      "training_step_time": 17.065124034881592
    },
    {
      "epoch": 1.8096923828125e-05,
      "model_forward_time": 5.6099631786346436,
      "step": 18976
    },
    {
      "epoch": 1.8096923828125e-05,
      "step": 18976,
      "training_step_time": 17.115934133529663
    },
    {
      "epoch": 1.8097877502441407e-05,
      "model_forward_time": 5.61396598815918,
      "step": 18977
    },
    {
      "epoch": 1.8097877502441407e-05,
      "step": 18977,
      "training_step_time": 17.463536739349365
    },
    {
      "epoch": 1.8098831176757814e-05,
      "model_forward_time": 5.612297773361206,
      "step": 18978
    },
    {
      "epoch": 1.8098831176757814e-05,
      "step": 18978,
      "training_step_time": 17.110105514526367
    },
    {
      "epoch": 1.8099784851074218e-05,
      "model_forward_time": 5.6064534187316895,
      "step": 18979
    },
    {
      "epoch": 1.8099784851074218e-05,
      "step": 18979,
      "training_step_time": 17.081660747528076
    },
    {
      "epoch": 1.8100738525390625e-05,
      "grad_norm": 1.1976244449615479,
      "learning_rate": 1e-05,
      "loss": 0.0575,
      "step": 18980
    },
    {
      "epoch": 1.8100738525390625e-05,
      "model_forward_time": 5.60254168510437,
      "step": 18980
    },
    {
      "dynamics_loss_avg": 0.057230615988373756,
      "epoch": 1.8100738525390625e-05,
      "step": 18980
    },
    {
      "action_loss_avg": 0.007232385594397784,
      "epoch": 1.8100738525390625e-05,
      "step": 18980
    },
    {
      "epoch": 1.8100738525390625e-05,
      "step": 18980,
      "training_step_time": 17.089147567749023
    },
    {
      "epoch": 1.8101692199707033e-05,
      "model_forward_time": 5.604572057723999,
      "step": 18981
    },
    {
      "epoch": 1.8101692199707033e-05,
      "step": 18981,
      "training_step_time": 17.094804763793945
    },
    {
      "epoch": 1.8102645874023437e-05,
      "model_forward_time": 5.612168788909912,
      "step": 18982
    },
    {
      "epoch": 1.8102645874023437e-05,
      "step": 18982,
      "training_step_time": 17.07469367980957
    },
    {
      "epoch": 1.8103599548339844e-05,
      "model_forward_time": 5.957639455795288,
      "step": 18983
    },
    {
      "epoch": 1.8103599548339844e-05,
      "step": 18983,
      "training_step_time": 17.435741424560547
    },
    {
      "epoch": 1.810455322265625e-05,
      "model_forward_time": 5.595154523849487,
      "step": 18984
    },
    {
      "epoch": 1.810455322265625e-05,
      "step": 18984,
      "training_step_time": 17.07576084136963
    },
    {
      "epoch": 1.8105506896972656e-05,
      "model_forward_time": 5.612755537033081,
      "step": 18985
    },
    {
      "epoch": 1.8105506896972656e-05,
      "step": 18985,
      "training_step_time": 17.108989477157593
    },
    {
      "epoch": 1.8106460571289063e-05,
      "model_forward_time": 5.613384485244751,
      "step": 18986
    },
    {
      "epoch": 1.8106460571289063e-05,
      "step": 18986,
      "training_step_time": 17.10273838043213
    },
    {
      "epoch": 1.8107414245605467e-05,
      "model_forward_time": 5.626630067825317,
      "step": 18987
    },
    {
      "epoch": 1.8107414245605467e-05,
      "step": 18987,
      "training_step_time": 17.12671971321106
    },
    {
      "epoch": 1.8108367919921875e-05,
      "model_forward_time": 5.606903076171875,
      "step": 18988
    },
    {
      "epoch": 1.8108367919921875e-05,
      "step": 18988,
      "training_step_time": 17.45757508277893
    },
    {
      "epoch": 1.8109321594238282e-05,
      "model_forward_time": 5.595464468002319,
      "step": 18989
    },
    {
      "epoch": 1.8109321594238282e-05,
      "step": 18989,
      "training_step_time": 17.073904275894165
    },
    {
      "epoch": 1.8110275268554686e-05,
      "grad_norm": 0.7562255859375,
      "learning_rate": 1e-05,
      "loss": 0.0502,
      "step": 18990
    },
    {
      "epoch": 1.8110275268554686e-05,
      "model_forward_time": 5.5979530811309814,
      "step": 18990
    },
    {
      "dynamics_loss_avg": 0.03929139096289873,
      "epoch": 1.8110275268554686e-05,
      "step": 18990
    },
    {
      "action_loss_avg": 0.004931006609695032,
      "epoch": 1.8110275268554686e-05,
      "step": 18990
    },
    {
      "epoch": 1.8110275268554686e-05,
      "step": 18990,
      "training_step_time": 17.035522937774658
    },
    {
      "epoch": 1.8111228942871093e-05,
      "model_forward_time": 5.590111970901489,
      "step": 18991
    },
    {
      "epoch": 1.8111228942871093e-05,
      "step": 18991,
      "training_step_time": 17.028071403503418
    },
    {
      "epoch": 1.81121826171875e-05,
      "model_forward_time": 5.584380626678467,
      "step": 18992
    },
    {
      "epoch": 1.81121826171875e-05,
      "step": 18992,
      "training_step_time": 17.017090797424316
    },
    {
      "epoch": 1.8113136291503905e-05,
      "model_forward_time": 5.5784289836883545,
      "step": 18993
    },
    {
      "epoch": 1.8113136291503905e-05,
      "step": 18993,
      "training_step_time": 17.01807713508606
    },
    {
      "epoch": 1.8114089965820312e-05,
      "model_forward_time": 5.9426915645599365,
      "step": 18994
    },
    {
      "epoch": 1.8114089965820312e-05,
      "step": 18994,
      "training_step_time": 17.38107204437256
    },
    {
      "epoch": 1.811504364013672e-05,
      "model_forward_time": 5.597618103027344,
      "step": 18995
    },
    {
      "epoch": 1.811504364013672e-05,
      "step": 18995,
      "training_step_time": 17.04002022743225
    },
    {
      "epoch": 1.8115997314453124e-05,
      "model_forward_time": 5.598269462585449,
      "step": 18996
    },
    {
      "epoch": 1.8115997314453124e-05,
      "step": 18996,
      "training_step_time": 17.065924406051636
    },
    {
      "epoch": 1.811695098876953e-05,
      "model_forward_time": 5.607330560684204,
      "step": 18997
    },
    {
      "epoch": 1.811695098876953e-05,
      "step": 18997,
      "training_step_time": 17.091991424560547
    },
    {
      "epoch": 1.811790466308594e-05,
      "model_forward_time": 5.613754034042358,
      "step": 18998
    },
    {
      "epoch": 1.811790466308594e-05,
      "step": 18998,
      "training_step_time": 17.116589546203613
    },
    {
      "epoch": 1.8118858337402343e-05,
      "model_forward_time": 5.963186264038086,
      "step": 18999
    },
    {
      "epoch": 1.8118858337402343e-05,
      "step": 18999,
      "training_step_time": 17.441661596298218
    },
    {
      "epoch": 1.811981201171875e-05,
      "grad_norm": 0.3568035960197449,
      "learning_rate": 1e-05,
      "loss": 0.0454,
      "step": 19000
    },
    {
      "epoch": 1.811981201171875e-05,
      "model_forward_time": 5.6399030685424805,
      "step": 19000
    },
    {
      "dynamics_loss_avg": 0.041907825693488124,
      "epoch": 1.811981201171875e-05,
      "step": 19000
    },
    {
      "action_loss_avg": 0.005610989243723452,
      "epoch": 1.811981201171875e-05,
      "step": 19000
    },
    {
      "epoch": 1.811981201171875e-05,
      "step": 19000,
      "training_step_time": 17.095757484436035
    },
    {
      "epoch": 1.8120765686035157e-05,
      "model_forward_time": 5.633358001708984,
      "step": 19001
    },
    {
      "epoch": 1.8120765686035157e-05,
      "step": 19001,
      "training_step_time": 17.097893476486206
    },
    {
      "epoch": 1.812171936035156e-05,
      "model_forward_time": 5.6111133098602295,
      "step": 19002
    },
    {
      "epoch": 1.812171936035156e-05,
      "step": 19002,
      "training_step_time": 17.057491779327393
    },
    {
      "epoch": 1.812267303466797e-05,
      "model_forward_time": 5.606328248977661,
      "step": 19003
    },
    {
      "epoch": 1.812267303466797e-05,
      "step": 19003,
      "training_step_time": 17.106351375579834
    },
    {
      "epoch": 1.8123626708984376e-05,
      "model_forward_time": 5.6396238803863525,
      "step": 19004
    },
    {
      "epoch": 1.8123626708984376e-05,
      "step": 19004,
      "training_step_time": 17.18079710006714
    },
    {
      "epoch": 1.812458038330078e-05,
      "model_forward_time": 6.001994371414185,
      "step": 19005
    },
    {
      "epoch": 1.812458038330078e-05,
      "step": 19005,
      "training_step_time": 17.50730299949646
    },
    {
      "epoch": 1.8125534057617188e-05,
      "model_forward_time": 5.619349241256714,
      "step": 19006
    },
    {
      "epoch": 1.8125534057617188e-05,
      "step": 19006,
      "training_step_time": 17.114378690719604
    },
    {
      "epoch": 1.8126487731933595e-05,
      "model_forward_time": 5.62197732925415,
      "step": 19007
    },
    {
      "epoch": 1.8126487731933595e-05,
      "step": 19007,
      "training_step_time": 17.09702444076538
    },
    {
      "epoch": 1.812744140625e-05,
      "model_forward_time": 5.6086907386779785,
      "step": 19008
    },
    {
      "epoch": 1.812744140625e-05,
      "step": 19008,
      "training_step_time": 17.080255270004272
    },
    {
      "epoch": 1.8128395080566407e-05,
      "model_forward_time": 5.604424715042114,
      "step": 19009
    },
    {
      "epoch": 1.8128395080566407e-05,
      "step": 19009,
      "training_step_time": 17.05852770805359
    },
    {
      "epoch": 1.8129348754882814e-05,
      "grad_norm": 0.1733420044183731,
      "learning_rate": 1e-05,
      "loss": 0.0745,
      "step": 19010
    },
    {
      "epoch": 1.8129348754882814e-05,
      "model_forward_time": 5.948780059814453,
      "step": 19010
    },
    {
      "dynamics_loss_avg": 0.06663775406777858,
      "epoch": 1.8129348754882814e-05,
      "step": 19010
    },
    {
      "action_loss_avg": 0.010928758594673127,
      "epoch": 1.8129348754882814e-05,
      "step": 19010
    },
    {
      "epoch": 1.8129348754882814e-05,
      "step": 19010,
      "training_step_time": 17.376948833465576
    },
    {
      "epoch": 1.8130302429199218e-05,
      "model_forward_time": 5.5888800621032715,
      "step": 19011
    },
    {
      "epoch": 1.8130302429199218e-05,
      "step": 19011,
      "training_step_time": 17.027117013931274
    },
    {
      "epoch": 1.8131256103515625e-05,
      "model_forward_time": 5.592980146408081,
      "step": 19012
    },
    {
      "epoch": 1.8131256103515625e-05,
      "step": 19012,
      "training_step_time": 17.028535842895508
    },
    {
      "epoch": 1.8132209777832033e-05,
      "model_forward_time": 5.60273814201355,
      "step": 19013
    },
    {
      "epoch": 1.8132209777832033e-05,
      "step": 19013,
      "training_step_time": 17.05297040939331
    },
    {
      "epoch": 1.8133163452148437e-05,
      "model_forward_time": 5.617656946182251,
      "step": 19014
    },
    {
      "epoch": 1.8133163452148437e-05,
      "step": 19014,
      "training_step_time": 17.07280969619751
    },
    {
      "epoch": 1.8134117126464844e-05,
      "model_forward_time": 5.594711780548096,
      "step": 19015
    },
    {
      "epoch": 1.8134117126464844e-05,
      "step": 19015,
      "training_step_time": 17.017641305923462
    },
    {
      "epoch": 1.813507080078125e-05,
      "model_forward_time": 5.965308666229248,
      "step": 19016
    },
    {
      "epoch": 1.813507080078125e-05,
      "step": 19016,
      "training_step_time": 17.41326355934143
    },
    {
      "epoch": 1.8136024475097656e-05,
      "model_forward_time": 5.612023115158081,
      "step": 19017
    },
    {
      "epoch": 1.8136024475097656e-05,
      "step": 19017,
      "training_step_time": 17.10706901550293
    },
    {
      "epoch": 1.8136978149414063e-05,
      "model_forward_time": 5.625091791152954,
      "step": 19018
    },
    {
      "epoch": 1.8136978149414063e-05,
      "step": 19018,
      "training_step_time": 17.101799964904785
    },
    {
      "epoch": 1.8137931823730467e-05,
      "model_forward_time": 5.596646547317505,
      "step": 19019
    },
    {
      "epoch": 1.8137931823730467e-05,
      "step": 19019,
      "training_step_time": 17.0458345413208
    },
    {
      "epoch": 1.8138885498046875e-05,
      "grad_norm": 0.9008811116218567,
      "learning_rate": 1e-05,
      "loss": 0.0662,
      "step": 19020
    },
    {
      "epoch": 1.8138885498046875e-05,
      "model_forward_time": 5.607724189758301,
      "step": 19020
    },
    {
      "dynamics_loss_avg": 0.05538500547409057,
      "epoch": 1.8138885498046875e-05,
      "step": 19020
    },
    {
      "action_loss_avg": 0.006112225051037967,
      "epoch": 1.8138885498046875e-05,
      "step": 19020
    },
    {
      "epoch": 1.8138885498046875e-05,
      "step": 19020,
      "training_step_time": 17.099612712860107
    },
    {
      "epoch": 1.8139839172363282e-05,
      "model_forward_time": 6.078866481781006,
      "step": 19021
    },
    {
      "epoch": 1.8139839172363282e-05,
      "step": 19021,
      "training_step_time": 17.54246711730957
    },
    {
      "epoch": 1.8140792846679686e-05,
      "model_forward_time": 5.611832141876221,
      "step": 19022
    },
    {
      "epoch": 1.8140792846679686e-05,
      "step": 19022,
      "training_step_time": 17.08194637298584
    },
    {
      "epoch": 1.8141746520996093e-05,
      "model_forward_time": 5.61273717880249,
      "step": 19023
    },
    {
      "epoch": 1.8141746520996093e-05,
      "step": 19023,
      "training_step_time": 17.11326551437378
    },
    {
      "epoch": 1.81427001953125e-05,
      "model_forward_time": 5.62491774559021,
      "step": 19024
    },
    {
      "epoch": 1.81427001953125e-05,
      "step": 19024,
      "training_step_time": 17.13057589530945
    },
    {
      "epoch": 1.8143653869628905e-05,
      "model_forward_time": 5.619864225387573,
      "step": 19025
    },
    {
      "epoch": 1.8143653869628905e-05,
      "step": 19025,
      "training_step_time": 17.118367671966553
    },
    {
      "epoch": 1.8144607543945312e-05,
      "model_forward_time": 5.6249754428863525,
      "step": 19026
    },
    {
      "epoch": 1.8144607543945312e-05,
      "step": 19026,
      "training_step_time": 17.55494451522827
    },
    {
      "epoch": 1.814556121826172e-05,
      "model_forward_time": 5.6174867153167725,
      "step": 19027
    },
    {
      "epoch": 1.814556121826172e-05,
      "step": 19027,
      "training_step_time": 17.11469578742981
    },
    {
      "epoch": 1.8146514892578124e-05,
      "model_forward_time": 5.613790988922119,
      "step": 19028
    },
    {
      "epoch": 1.8146514892578124e-05,
      "step": 19028,
      "training_step_time": 17.102823734283447
    },
    {
      "epoch": 1.814746856689453e-05,
      "model_forward_time": 5.616852521896362,
      "step": 19029
    },
    {
      "epoch": 1.814746856689453e-05,
      "step": 19029,
      "training_step_time": 17.123513221740723
    },
    {
      "epoch": 1.814842224121094e-05,
      "grad_norm": 0.9077795147895813,
      "learning_rate": 1e-05,
      "loss": 0.0611,
      "step": 19030
    },
    {
      "epoch": 1.814842224121094e-05,
      "model_forward_time": 5.617060899734497,
      "step": 19030
    },
    {
      "dynamics_loss_avg": 0.05389365656301379,
      "epoch": 1.814842224121094e-05,
      "step": 19030
    },
    {
      "action_loss_avg": 0.00936904245754704,
      "epoch": 1.814842224121094e-05,
      "step": 19030
    },
    {
      "epoch": 1.814842224121094e-05,
      "step": 19030,
      "training_step_time": 17.11455273628235
    },
    {
      "epoch": 1.8149375915527343e-05,
      "model_forward_time": 5.604157447814941,
      "step": 19031
    },
    {
      "epoch": 1.8149375915527343e-05,
      "step": 19031,
      "training_step_time": 17.073781967163086
    },
    {
      "epoch": 1.815032958984375e-05,
      "model_forward_time": 5.95328426361084,
      "step": 19032
    },
    {
      "epoch": 1.815032958984375e-05,
      "step": 19032,
      "training_step_time": 17.405660152435303
    },
    {
      "epoch": 1.8151283264160157e-05,
      "model_forward_time": 5.584420919418335,
      "step": 19033
    },
    {
      "epoch": 1.8151283264160157e-05,
      "step": 19033,
      "training_step_time": 16.993571519851685
    },
    {
      "epoch": 1.815223693847656e-05,
      "model_forward_time": 5.587416887283325,
      "step": 19034
    },
    {
      "epoch": 1.815223693847656e-05,
      "step": 19034,
      "training_step_time": 17.050039768218994
    },
    {
      "epoch": 1.815319061279297e-05,
      "model_forward_time": 5.600149631500244,
      "step": 19035
    },
    {
      "epoch": 1.815319061279297e-05,
      "step": 19035,
      "training_step_time": 17.067391872406006
    },
    {
      "epoch": 1.8154144287109376e-05,
      "model_forward_time": 5.613231897354126,
      "step": 19036
    },
    {
      "epoch": 1.8154144287109376e-05,
      "step": 19036,
      "training_step_time": 17.100883960723877
    },
    {
      "epoch": 1.815509796142578e-05,
      "model_forward_time": 5.6149420738220215,
      "step": 19037
    },
    {
      "epoch": 1.815509796142578e-05,
      "step": 19037,
      "training_step_time": 17.474863290786743
    },
    {
      "epoch": 1.8156051635742188e-05,
      "model_forward_time": 5.6084983348846436,
      "step": 19038
    },
    {
      "epoch": 1.8156051635742188e-05,
      "step": 19038,
      "training_step_time": 17.09609317779541
    },
    {
      "epoch": 1.8157005310058595e-05,
      "model_forward_time": 5.614152193069458,
      "step": 19039
    },
    {
      "epoch": 1.8157005310058595e-05,
      "step": 19039,
      "training_step_time": 17.102365255355835
    },
    {
      "epoch": 1.8157958984375e-05,
      "grad_norm": 0.4598018229007721,
      "learning_rate": 1e-05,
      "loss": 0.0672,
      "step": 19040
    },
    {
      "epoch": 1.8157958984375e-05,
      "model_forward_time": 5.624378204345703,
      "step": 19040
    },
    {
      "dynamics_loss_avg": 0.057589917071163656,
      "epoch": 1.8157958984375e-05,
      "step": 19040
    },
    {
      "action_loss_avg": 0.005994255188852549,
      "epoch": 1.8157958984375e-05,
      "step": 19040
    },
    {
      "epoch": 1.8157958984375e-05,
      "step": 19040,
      "training_step_time": 17.095194578170776
    },
    {
      "epoch": 1.8158912658691407e-05,
      "model_forward_time": 5.61218523979187,
      "step": 19041
    },
    {
      "epoch": 1.8158912658691407e-05,
      "step": 19041,
      "training_step_time": 17.113874435424805
    },
    {
      "epoch": 1.8159866333007814e-05,
      "model_forward_time": 5.60462212562561,
      "step": 19042
    },
    {
      "epoch": 1.8159866333007814e-05,
      "step": 19042,
      "training_step_time": 17.105119228363037
    },
    {
      "epoch": 1.8160820007324218e-05,
      "model_forward_time": 5.976291179656982,
      "step": 19043
    },
    {
      "epoch": 1.8160820007324218e-05,
      "step": 19043,
      "training_step_time": 17.470011949539185
    },
    {
      "epoch": 1.8161773681640625e-05,
      "model_forward_time": 5.609043598175049,
      "step": 19044
    },
    {
      "epoch": 1.8161773681640625e-05,
      "step": 19044,
      "training_step_time": 17.06979274749756
    },
    {
      "epoch": 1.8162727355957033e-05,
      "model_forward_time": 5.596457242965698,
      "step": 19045
    },
    {
      "epoch": 1.8162727355957033e-05,
      "step": 19045,
      "training_step_time": 17.05860948562622
    },
    {
      "epoch": 1.8163681030273437e-05,
      "model_forward_time": 5.5998313426971436,
      "step": 19046
    },
    {
      "epoch": 1.8163681030273437e-05,
      "step": 19046,
      "training_step_time": 17.06708860397339
    },
    {
      "epoch": 1.8164634704589844e-05,
      "model_forward_time": 5.599953651428223,
      "step": 19047
    },
    {
      "epoch": 1.8164634704589844e-05,
      "step": 19047,
      "training_step_time": 17.06037926673889
    },
    {
      "epoch": 1.816558837890625e-05,
      "model_forward_time": 5.980980157852173,
      "step": 19048
    },
    {
      "epoch": 1.816558837890625e-05,
      "step": 19048,
      "training_step_time": 17.421783924102783
    },
    {
      "epoch": 1.8166542053222656e-05,
      "model_forward_time": 5.5938544273376465,
      "step": 19049
    },
    {
      "epoch": 1.8166542053222656e-05,
      "step": 19049,
      "training_step_time": 17.03575611114502
    },
    {
      "epoch": 1.8167495727539063e-05,
      "grad_norm": 0.3252081573009491,
      "learning_rate": 1e-05,
      "loss": 0.0749,
      "step": 19050
    },
    {
      "epoch": 1.8167495727539063e-05,
      "model_forward_time": 5.582319498062134,
      "step": 19050
    },
    {
      "dynamics_loss_avg": 0.07593790143728256,
      "epoch": 1.8167495727539063e-05,
      "step": 19050
    },
    {
      "action_loss_avg": 0.009073770744726062,
      "epoch": 1.8167495727539063e-05,
      "step": 19050
    },
    {
      "epoch": 1.8167495727539063e-05,
      "step": 19050,
      "training_step_time": 17.02288031578064
    },
    {
      "epoch": 1.8168449401855467e-05,
      "model_forward_time": 5.6045262813568115,
      "step": 19051
    },
    {
      "epoch": 1.8168449401855467e-05,
      "step": 19051,
      "training_step_time": 17.07767915725708
    },
    {
      "epoch": 1.8169403076171875e-05,
      "model_forward_time": 5.600322484970093,
      "step": 19052
    },
    {
      "epoch": 1.8169403076171875e-05,
      "step": 19052,
      "training_step_time": 17.065977811813354
    },
    {
      "epoch": 1.8170356750488282e-05,
      "model_forward_time": 5.60532021522522,
      "step": 19053
    },
    {
      "epoch": 1.8170356750488282e-05,
      "step": 19053,
      "training_step_time": 17.061421394348145
    },
    {
      "epoch": 1.8171310424804686e-05,
      "model_forward_time": 5.935175180435181,
      "step": 19054
    },
    {
      "epoch": 1.8171310424804686e-05,
      "step": 19054,
      "training_step_time": 17.388603687286377
    },
    {
      "epoch": 1.8172264099121093e-05,
      "model_forward_time": 5.611902236938477,
      "step": 19055
    },
    {
      "epoch": 1.8172264099121093e-05,
      "step": 19055,
      "training_step_time": 17.094151735305786
    },
    {
      "epoch": 1.81732177734375e-05,
      "model_forward_time": 5.615411043167114,
      "step": 19056
    },
    {
      "epoch": 1.81732177734375e-05,
      "step": 19056,
      "training_step_time": 17.117197513580322
    },
    {
      "epoch": 1.8174171447753905e-05,
      "model_forward_time": 5.6276466846466064,
      "step": 19057
    },
    {
      "epoch": 1.8174171447753905e-05,
      "step": 19057,
      "training_step_time": 17.115074634552002
    },
    {
      "epoch": 1.8175125122070312e-05,
      "model_forward_time": 5.602304458618164,
      "step": 19058
    },
    {
      "epoch": 1.8175125122070312e-05,
      "step": 19058,
      "training_step_time": 17.049739360809326
    },
    {
      "epoch": 1.817607879638672e-05,
      "model_forward_time": 5.952275514602661,
      "step": 19059
    },
    {
      "epoch": 1.817607879638672e-05,
      "step": 19059,
      "training_step_time": 17.376131057739258
    },
    {
      "epoch": 1.8177032470703124e-05,
      "grad_norm": 0.5024186372756958,
      "learning_rate": 1e-05,
      "loss": 0.0759,
      "step": 19060
    },
    {
      "epoch": 1.8177032470703124e-05,
      "model_forward_time": 5.58719277381897,
      "step": 19060
    },
    {
      "dynamics_loss_avg": 0.05997497346252203,
      "epoch": 1.8177032470703124e-05,
      "step": 19060
    },
    {
      "action_loss_avg": 0.006580296589527279,
      "epoch": 1.8177032470703124e-05,
      "step": 19060
    },
    {
      "epoch": 1.8177032470703124e-05,
      "step": 19060,
      "training_step_time": 17.01002287864685
    },
    {
      "epoch": 1.817798614501953e-05,
      "model_forward_time": 5.57648777961731,
      "step": 19061
    },
    {
      "epoch": 1.817798614501953e-05,
      "step": 19061,
      "training_step_time": 17.006877899169922
    },
    {
      "epoch": 1.817893981933594e-05,
      "model_forward_time": 5.577316045761108,
      "step": 19062
    },
    {
      "epoch": 1.817893981933594e-05,
      "step": 19062,
      "training_step_time": 17.010437965393066
    },
    {
      "epoch": 1.8179893493652343e-05,
      "model_forward_time": 5.5982091426849365,
      "step": 19063
    },
    {
      "epoch": 1.8179893493652343e-05,
      "step": 19063,
      "training_step_time": 17.027965784072876
    },
    {
      "epoch": 1.818084716796875e-05,
      "model_forward_time": 5.592686653137207,
      "step": 19064
    },
    {
      "epoch": 1.818084716796875e-05,
      "step": 19064,
      "training_step_time": 17.029953956604004
    },
    {
      "epoch": 1.8181800842285157e-05,
      "model_forward_time": 5.945750951766968,
      "step": 19065
    },
    {
      "epoch": 1.8181800842285157e-05,
      "step": 19065,
      "training_step_time": 17.431174516677856
    },
    {
      "epoch": 1.818275451660156e-05,
      "model_forward_time": 5.625708341598511,
      "step": 19066
    },
    {
      "epoch": 1.818275451660156e-05,
      "step": 19066,
      "training_step_time": 17.153603076934814
    },
    {
      "epoch": 1.818370819091797e-05,
      "model_forward_time": 5.637147665023804,
      "step": 19067
    },
    {
      "epoch": 1.818370819091797e-05,
      "step": 19067,
      "training_step_time": 17.144994020462036
    },
    {
      "epoch": 1.8184661865234376e-05,
      "model_forward_time": 5.62109899520874,
      "step": 19068
    },
    {
      "epoch": 1.8184661865234376e-05,
      "step": 19068,
      "training_step_time": 17.109195709228516
    },
    {
      "epoch": 1.818561553955078e-05,
      "model_forward_time": 5.606573820114136,
      "step": 19069
    },
    {
      "epoch": 1.818561553955078e-05,
      "step": 19069,
      "training_step_time": 17.072991371154785
    },
    {
      "epoch": 1.8186569213867188e-05,
      "grad_norm": 0.40778812766075134,
      "learning_rate": 1e-05,
      "loss": 0.0663,
      "step": 19070
    },
    {
      "epoch": 1.8186569213867188e-05,
      "model_forward_time": 5.972099781036377,
      "step": 19070
    },
    {
      "dynamics_loss_avg": 0.06175114084035158,
      "epoch": 1.8186569213867188e-05,
      "step": 19070
    },
    {
      "action_loss_avg": 0.0083184449467808,
      "epoch": 1.8186569213867188e-05,
      "step": 19070
    },
    {
      "epoch": 1.8186569213867188e-05,
      "step": 19070,
      "training_step_time": 17.42903423309326
    },
    {
      "epoch": 1.8187522888183595e-05,
      "model_forward_time": 5.602650165557861,
      "step": 19071
    },
    {
      "epoch": 1.8187522888183595e-05,
      "step": 19071,
      "training_step_time": 17.102731466293335
    },
    {
      "epoch": 1.81884765625e-05,
      "model_forward_time": 5.614902019500732,
      "step": 19072
    },
    {
      "epoch": 1.81884765625e-05,
      "step": 19072,
      "training_step_time": 17.123081922531128
    },
    {
      "epoch": 1.8189430236816407e-05,
      "model_forward_time": 5.631164789199829,
      "step": 19073
    },
    {
      "epoch": 1.8189430236816407e-05,
      "step": 19073,
      "training_step_time": 17.139785289764404
    },
    {
      "epoch": 1.8190383911132814e-05,
      "model_forward_time": 5.615872144699097,
      "step": 19074
    },
    {
      "epoch": 1.8190383911132814e-05,
      "step": 19074,
      "training_step_time": 17.121166944503784
    },
    {
      "epoch": 1.8191337585449218e-05,
      "model_forward_time": 5.961146831512451,
      "step": 19075
    },
    {
      "epoch": 1.8191337585449218e-05,
      "step": 19075,
      "training_step_time": 17.4524667263031
    },
    {
      "epoch": 1.8192291259765625e-05,
      "model_forward_time": 5.606543302536011,
      "step": 19076
    },
    {
      "epoch": 1.8192291259765625e-05,
      "step": 19076,
      "training_step_time": 17.061880588531494
    },
    {
      "epoch": 1.8193244934082033e-05,
      "model_forward_time": 5.599399089813232,
      "step": 19077
    },
    {
      "epoch": 1.8193244934082033e-05,
      "step": 19077,
      "training_step_time": 17.04204750061035
    },
    {
      "epoch": 1.8194198608398437e-05,
      "model_forward_time": 5.587643146514893,
      "step": 19078
    },
    {
      "epoch": 1.8194198608398437e-05,
      "step": 19078,
      "training_step_time": 17.036516666412354
    },
    {
      "epoch": 1.8195152282714844e-05,
      "model_forward_time": 5.605462074279785,
      "step": 19079
    },
    {
      "epoch": 1.8195152282714844e-05,
      "step": 19079,
      "training_step_time": 17.099777221679688
    },
    {
      "epoch": 1.819610595703125e-05,
      "grad_norm": 0.36962398886680603,
      "learning_rate": 1e-05,
      "loss": 0.0571,
      "step": 19080
    },
    {
      "epoch": 1.819610595703125e-05,
      "model_forward_time": 5.6153645515441895,
      "step": 19080
    },
    {
      "dynamics_loss_avg": 0.04904253724962473,
      "epoch": 1.819610595703125e-05,
      "step": 19080
    },
    {
      "action_loss_avg": 0.007037076784763485,
      "epoch": 1.819610595703125e-05,
      "step": 19080
    },
    {
      "epoch": 1.819610595703125e-05,
      "step": 19080,
      "training_step_time": 17.117065906524658
    },
    {
      "epoch": 1.8197059631347656e-05,
      "model_forward_time": 5.990274429321289,
      "step": 19081
    },
    {
      "epoch": 1.8197059631347656e-05,
      "step": 19081,
      "training_step_time": 17.489526510238647
    },
    {
      "epoch": 1.8198013305664063e-05,
      "model_forward_time": 5.622162103652954,
      "step": 19082
    },
    {
      "epoch": 1.8198013305664063e-05,
      "step": 19082,
      "training_step_time": 17.123204469680786
    },
    {
      "epoch": 1.8198966979980467e-05,
      "model_forward_time": 5.597257375717163,
      "step": 19083
    },
    {
      "epoch": 1.8198966979980467e-05,
      "step": 19083,
      "training_step_time": 17.065124988555908
    },
    {
      "epoch": 1.8199920654296875e-05,
      "model_forward_time": 5.592688798904419,
      "step": 19084
    },
    {
      "epoch": 1.8199920654296875e-05,
      "step": 19084,
      "training_step_time": 17.05629062652588
    },
    {
      "epoch": 1.8200874328613282e-05,
      "model_forward_time": 5.602755308151245,
      "step": 19085
    },
    {
      "epoch": 1.8200874328613282e-05,
      "step": 19085,
      "training_step_time": 17.076447248458862
    },
    {
      "epoch": 1.8201828002929686e-05,
      "model_forward_time": 5.9789814949035645,
      "step": 19086
    },
    {
      "epoch": 1.8201828002929686e-05,
      "step": 19086,
      "training_step_time": 17.48330807685852
    },
    {
      "epoch": 1.8202781677246094e-05,
      "model_forward_time": 5.627685070037842,
      "step": 19087
    },
    {
      "epoch": 1.8202781677246094e-05,
      "step": 19087,
      "training_step_time": 17.131836652755737
    },
    {
      "epoch": 1.82037353515625e-05,
      "model_forward_time": 5.610641241073608,
      "step": 19088
    },
    {
      "epoch": 1.82037353515625e-05,
      "step": 19088,
      "training_step_time": 17.08954095840454
    },
    {
      "epoch": 1.8204689025878905e-05,
      "model_forward_time": 5.6075334548950195,
      "step": 19089
    },
    {
      "epoch": 1.8204689025878905e-05,
      "step": 19089,
      "training_step_time": 17.066580295562744
    },
    {
      "epoch": 1.8205642700195312e-05,
      "grad_norm": 1.1538501977920532,
      "learning_rate": 1e-05,
      "loss": 0.0772,
      "step": 19090
    },
    {
      "epoch": 1.8205642700195312e-05,
      "model_forward_time": 5.588405609130859,
      "step": 19090
    },
    {
      "dynamics_loss_avg": 0.06782484725117684,
      "epoch": 1.8205642700195312e-05,
      "step": 19090
    },
    {
      "action_loss_avg": 0.008270867029204964,
      "epoch": 1.8205642700195312e-05,
      "step": 19090
    },
    {
      "epoch": 1.8205642700195312e-05,
      "step": 19090,
      "training_step_time": 17.033844232559204
    },
    {
      "epoch": 1.820659637451172e-05,
      "model_forward_time": 5.5962700843811035,
      "step": 19091
    },
    {
      "epoch": 1.820659637451172e-05,
      "step": 19091,
      "training_step_time": 17.07215428352356
    },
    {
      "epoch": 1.8207550048828124e-05,
      "model_forward_time": 5.968130350112915,
      "step": 19092
    },
    {
      "epoch": 1.8207550048828124e-05,
      "step": 19092,
      "training_step_time": 17.445068359375
    },
    {
      "epoch": 1.820850372314453e-05,
      "model_forward_time": 5.60697603225708,
      "step": 19093
    },
    {
      "epoch": 1.820850372314453e-05,
      "step": 19093,
      "training_step_time": 17.08184504508972
    },
    {
      "epoch": 1.820945739746094e-05,
      "model_forward_time": 5.610827207565308,
      "step": 19094
    },
    {
      "epoch": 1.820945739746094e-05,
      "step": 19094,
      "training_step_time": 17.109965801239014
    },
    {
      "epoch": 1.8210411071777343e-05,
      "model_forward_time": 5.615460395812988,
      "step": 19095
    },
    {
      "epoch": 1.8210411071777343e-05,
      "step": 19095,
      "training_step_time": 17.105635166168213
    },
    {
      "epoch": 1.821136474609375e-05,
      "model_forward_time": 5.60079288482666,
      "step": 19096
    },
    {
      "epoch": 1.821136474609375e-05,
      "step": 19096,
      "training_step_time": 17.046738386154175
    },
    {
      "epoch": 1.8212318420410157e-05,
      "model_forward_time": 5.575061082839966,
      "step": 19097
    },
    {
      "epoch": 1.8212318420410157e-05,
      "step": 19097,
      "training_step_time": 17.382951021194458
    },
    {
      "epoch": 1.821327209472656e-05,
      "model_forward_time": 5.587461709976196,
      "step": 19098
    },
    {
      "epoch": 1.821327209472656e-05,
      "step": 19098,
      "training_step_time": 17.02995276451111
    },
    {
      "epoch": 1.821422576904297e-05,
      "model_forward_time": 5.600499391555786,
      "step": 19099
    },
    {
      "epoch": 1.821422576904297e-05,
      "step": 19099,
      "training_step_time": 17.065218687057495
    },
    {
      "epoch": 1.8215179443359376e-05,
      "grad_norm": 1.0050824880599976,
      "learning_rate": 1e-05,
      "loss": 0.0663,
      "step": 19100
    },
    {
      "epoch": 1.8215179443359376e-05,
      "model_forward_time": 5.6019980907440186,
      "step": 19100
    },
    {
      "dynamics_loss_avg": 0.06382782049477101,
      "epoch": 1.8215179443359376e-05,
      "step": 19100
    },
    {
      "action_loss_avg": 0.007235502498224378,
      "epoch": 1.8215179443359376e-05,
      "step": 19100
    },
    {
      "epoch": 1.8215179443359376e-05,
      "step": 19100,
      "training_step_time": 17.065900325775146
    },
    {
      "epoch": 1.821613311767578e-05,
      "model_forward_time": 5.601256847381592,
      "step": 19101
    },
    {
      "epoch": 1.821613311767578e-05,
      "step": 19101,
      "training_step_time": 17.073427438735962
    },
    {
      "epoch": 1.8217086791992188e-05,
      "model_forward_time": 5.595370531082153,
      "step": 19102
    },
    {
      "epoch": 1.8217086791992188e-05,
      "step": 19102,
      "training_step_time": 17.074562311172485
    },
    {
      "epoch": 1.8218040466308595e-05,
      "model_forward_time": 5.973126173019409,
      "step": 19103
    },
    {
      "epoch": 1.8218040466308595e-05,
      "step": 19103,
      "training_step_time": 17.419574975967407
    },
    {
      "epoch": 1.8218994140625e-05,
      "model_forward_time": 5.580847501754761,
      "step": 19104
    },
    {
      "epoch": 1.8218994140625e-05,
      "step": 19104,
      "training_step_time": 17.016737937927246
    },
    {
      "epoch": 1.8219947814941407e-05,
      "model_forward_time": 5.5739967823028564,
      "step": 19105
    },
    {
      "epoch": 1.8219947814941407e-05,
      "step": 19105,
      "training_step_time": 16.982434034347534
    },
    {
      "epoch": 1.8220901489257814e-05,
      "model_forward_time": 5.567883729934692,
      "step": 19106
    },
    {
      "epoch": 1.8220901489257814e-05,
      "step": 19106,
      "training_step_time": 16.977460145950317
    },
    {
      "epoch": 1.8221855163574218e-05,
      "model_forward_time": 5.5678136348724365,
      "step": 19107
    },
    {
      "epoch": 1.8221855163574218e-05,
      "step": 19107,
      "training_step_time": 16.99053692817688
    },
    {
      "epoch": 1.8222808837890626e-05,
      "model_forward_time": 5.934700012207031,
      "step": 19108
    },
    {
      "epoch": 1.8222808837890626e-05,
      "step": 19108,
      "training_step_time": 17.37103009223938
    },
    {
      "epoch": 1.8223762512207033e-05,
      "model_forward_time": 5.5723302364349365,
      "step": 19109
    },
    {
      "epoch": 1.8223762512207033e-05,
      "step": 19109,
      "training_step_time": 17.005632638931274
    },
    {
      "epoch": 1.8224716186523437e-05,
      "grad_norm": 0.5761492252349854,
      "learning_rate": 1e-05,
      "loss": 0.0863,
      "step": 19110
    },
    {
      "epoch": 1.8224716186523437e-05,
      "model_forward_time": 5.581260442733765,
      "step": 19110
    },
    {
      "dynamics_loss_avg": 0.07426044829189778,
      "epoch": 1.8224716186523437e-05,
      "step": 19110
    },
    {
      "action_loss_avg": 0.011031583114527167,
      "epoch": 1.8224716186523437e-05,
      "step": 19110
    },
    {
      "epoch": 1.8224716186523437e-05,
      "step": 19110,
      "training_step_time": 17.001961946487427
    },
    {
      "epoch": 1.8225669860839844e-05,
      "model_forward_time": 5.591236591339111,
      "step": 19111
    },
    {
      "epoch": 1.8225669860839844e-05,
      "step": 19111,
      "training_step_time": 17.033764600753784
    },
    {
      "epoch": 1.822662353515625e-05,
      "model_forward_time": 5.605713605880737,
      "step": 19112
    },
    {
      "epoch": 1.822662353515625e-05,
      "step": 19112,
      "training_step_time": 17.094631910324097
    },
    {
      "epoch": 1.8227577209472656e-05,
      "model_forward_time": 5.629693269729614,
      "step": 19113
    },
    {
      "epoch": 1.8227577209472656e-05,
      "step": 19113,
      "training_step_time": 17.118868827819824
    },
    {
      "epoch": 1.8228530883789063e-05,
      "model_forward_time": 5.968936920166016,
      "step": 19114
    },
    {
      "epoch": 1.8228530883789063e-05,
      "step": 19114,
      "training_step_time": 17.416361331939697
    },
    {
      "epoch": 1.8229484558105467e-05,
      "model_forward_time": 5.585216760635376,
      "step": 19115
    },
    {
      "epoch": 1.8229484558105467e-05,
      "step": 19115,
      "training_step_time": 17.03089714050293
    },
    {
      "epoch": 1.8230438232421875e-05,
      "model_forward_time": 5.5819926261901855,
      "step": 19116
    },
    {
      "epoch": 1.8230438232421875e-05,
      "step": 19116,
      "training_step_time": 17.021964073181152
    },
    {
      "epoch": 1.8231391906738282e-05,
      "model_forward_time": 5.594539642333984,
      "step": 19117
    },
    {
      "epoch": 1.8231391906738282e-05,
      "step": 19117,
      "training_step_time": 17.054312705993652
    },
    {
      "epoch": 1.8232345581054686e-05,
      "model_forward_time": 5.600382328033447,
      "step": 19118
    },
    {
      "epoch": 1.8232345581054686e-05,
      "step": 19118,
      "training_step_time": 17.079962015151978
    },
    {
      "epoch": 1.8233299255371094e-05,
      "model_forward_time": 5.972796440124512,
      "step": 19119
    },
    {
      "epoch": 1.8233299255371094e-05,
      "step": 19119,
      "training_step_time": 17.47929859161377
    },
    {
      "epoch": 1.82342529296875e-05,
      "grad_norm": 0.948319137096405,
      "learning_rate": 1e-05,
      "loss": 0.0617,
      "step": 19120
    },
    {
      "epoch": 1.82342529296875e-05,
      "model_forward_time": 5.625523567199707,
      "step": 19120
    },
    {
      "dynamics_loss_avg": 0.052341210283339025,
      "epoch": 1.82342529296875e-05,
      "step": 19120
    },
    {
      "action_loss_avg": 0.006636216747574508,
      "epoch": 1.82342529296875e-05,
      "step": 19120
    },
    {
      "epoch": 1.82342529296875e-05,
      "step": 19120,
      "training_step_time": 17.116165161132812
    },
    {
      "epoch": 1.8235206604003905e-05,
      "model_forward_time": 5.597804307937622,
      "step": 19121
    },
    {
      "epoch": 1.8235206604003905e-05,
      "step": 19121,
      "training_step_time": 17.010509967803955
    },
    {
      "epoch": 1.8236160278320312e-05,
      "model_forward_time": 5.589908123016357,
      "step": 19122
    },
    {
      "epoch": 1.8236160278320312e-05,
      "step": 19122,
      "training_step_time": 17.0074679851532
    },
    {
      "epoch": 1.823711395263672e-05,
      "model_forward_time": 5.571779012680054,
      "step": 19123
    },
    {
      "epoch": 1.823711395263672e-05,
      "step": 19123,
      "training_step_time": 17.006513118743896
    },
    {
      "epoch": 1.8238067626953124e-05,
      "model_forward_time": 5.578413724899292,
      "step": 19124
    },
    {
      "epoch": 1.8238067626953124e-05,
      "step": 19124,
      "training_step_time": 17.002683639526367
    },
    {
      "epoch": 1.823902130126953e-05,
      "model_forward_time": 5.949464321136475,
      "step": 19125
    },
    {
      "epoch": 1.823902130126953e-05,
      "step": 19125,
      "training_step_time": 17.37497067451477
    },
    {
      "epoch": 1.823997497558594e-05,
      "model_forward_time": 5.595770359039307,
      "step": 19126
    },
    {
      "epoch": 1.823997497558594e-05,
      "step": 19126,
      "training_step_time": 17.04319143295288
    },
    {
      "epoch": 1.8240928649902343e-05,
      "model_forward_time": 5.599822044372559,
      "step": 19127
    },
    {
      "epoch": 1.8240928649902343e-05,
      "step": 19127,
      "training_step_time": 17.0653076171875
    },
    {
      "epoch": 1.824188232421875e-05,
      "model_forward_time": 5.610192775726318,
      "step": 19128
    },
    {
      "epoch": 1.824188232421875e-05,
      "step": 19128,
      "training_step_time": 17.09517192840576
    },
    {
      "epoch": 1.8242835998535158e-05,
      "model_forward_time": 5.600085020065308,
      "step": 19129
    },
    {
      "epoch": 1.8242835998535158e-05,
      "step": 19129,
      "training_step_time": 17.060986280441284
    },
    {
      "epoch": 1.824378967285156e-05,
      "grad_norm": 0.49061280488967896,
      "learning_rate": 1e-05,
      "loss": 0.0778,
      "step": 19130
    },
    {
      "epoch": 1.824378967285156e-05,
      "model_forward_time": 5.590546607971191,
      "step": 19130
    },
    {
      "dynamics_loss_avg": 0.06767781693488359,
      "epoch": 1.824378967285156e-05,
      "step": 19130
    },
    {
      "action_loss_avg": 0.006797176948748529,
      "epoch": 1.824378967285156e-05,
      "step": 19130
    },
    {
      "epoch": 1.824378967285156e-05,
      "step": 19130,
      "training_step_time": 17.389342546463013
    },
    {
      "epoch": 1.824474334716797e-05,
      "model_forward_time": 5.584869861602783,
      "step": 19131
    },
    {
      "epoch": 1.824474334716797e-05,
      "step": 19131,
      "training_step_time": 17.005425691604614
    },
    {
      "epoch": 1.8245697021484376e-05,
      "model_forward_time": 5.598398923873901,
      "step": 19132
    },
    {
      "epoch": 1.8245697021484376e-05,
      "step": 19132,
      "training_step_time": 17.06774067878723
    },
    {
      "epoch": 1.824665069580078e-05,
      "model_forward_time": 5.611907243728638,
      "step": 19133
    },
    {
      "epoch": 1.824665069580078e-05,
      "step": 19133,
      "training_step_time": 17.106558561325073
    },
    {
      "epoch": 1.8247604370117188e-05,
      "model_forward_time": 5.60526442527771,
      "step": 19134
    },
    {
      "epoch": 1.8247604370117188e-05,
      "step": 19134,
      "training_step_time": 17.059449911117554
    },
    {
      "epoch": 1.8248558044433595e-05,
      "model_forward_time": 5.604442358016968,
      "step": 19135
    },
    {
      "epoch": 1.8248558044433595e-05,
      "step": 19135,
      "training_step_time": 17.06227397918701
    },
    {
      "epoch": 1.824951171875e-05,
      "model_forward_time": 5.9657933712005615,
      "step": 19136
    },
    {
      "epoch": 1.824951171875e-05,
      "step": 19136,
      "training_step_time": 17.41600799560547
    },
    {
      "epoch": 1.8250465393066407e-05,
      "model_forward_time": 5.589853763580322,
      "step": 19137
    },
    {
      "epoch": 1.8250465393066407e-05,
      "step": 19137,
      "training_step_time": 17.047574996948242
    },
    {
      "epoch": 1.8251419067382814e-05,
      "model_forward_time": 5.607328653335571,
      "step": 19138
    },
    {
      "epoch": 1.8251419067382814e-05,
      "step": 19138,
      "training_step_time": 17.081900596618652
    },
    {
      "epoch": 1.8252372741699218e-05,
      "model_forward_time": 5.612169504165649,
      "step": 19139
    },
    {
      "epoch": 1.8252372741699218e-05,
      "step": 19139,
      "training_step_time": 17.118879795074463
    },
    {
      "epoch": 1.8253326416015626e-05,
      "grad_norm": 0.5770246386528015,
      "learning_rate": 1e-05,
      "loss": 0.0526,
      "step": 19140
    },
    {
      "epoch": 1.8253326416015626e-05,
      "model_forward_time": 5.6209142208099365,
      "step": 19140
    },
    {
      "dynamics_loss_avg": 0.04729246515780687,
      "epoch": 1.8253326416015626e-05,
      "step": 19140
    },
    {
      "action_loss_avg": 0.005282024736516178,
      "epoch": 1.8253326416015626e-05,
      "step": 19140
    },
    {
      "epoch": 1.8253326416015626e-05,
      "step": 19140,
      "training_step_time": 17.144707441329956
    },
    {
      "epoch": 1.825428009033203e-05,
      "model_forward_time": 5.971640110015869,
      "step": 19141
    },
    {
      "epoch": 1.825428009033203e-05,
      "step": 19141,
      "training_step_time": 17.466615438461304
    },
    {
      "epoch": 1.8255233764648437e-05,
      "model_forward_time": 5.614234209060669,
      "step": 19142
    },
    {
      "epoch": 1.8255233764648437e-05,
      "step": 19142,
      "training_step_time": 17.098559856414795
    },
    {
      "epoch": 1.8256187438964844e-05,
      "model_forward_time": 5.601953506469727,
      "step": 19143
    },
    {
      "epoch": 1.8256187438964844e-05,
      "step": 19143,
      "training_step_time": 17.041179895401
    },
    {
      "epoch": 1.825714111328125e-05,
      "model_forward_time": 5.593241214752197,
      "step": 19144
    },
    {
      "epoch": 1.825714111328125e-05,
      "step": 19144,
      "training_step_time": 17.041791439056396
    },
    {
      "epoch": 1.8258094787597656e-05,
      "model_forward_time": 5.599274635314941,
      "step": 19145
    },
    {
      "epoch": 1.8258094787597656e-05,
      "step": 19145,
      "training_step_time": 17.065855026245117
    },
    {
      "epoch": 1.8259048461914063e-05,
      "model_forward_time": 5.6021928787231445,
      "step": 19146
    },
    {
      "epoch": 1.8259048461914063e-05,
      "step": 19146,
      "training_step_time": 17.059577226638794
    },
    {
      "epoch": 1.8260002136230467e-05,
      "model_forward_time": 5.968355894088745,
      "step": 19147
    },
    {
      "epoch": 1.8260002136230467e-05,
      "step": 19147,
      "training_step_time": 17.4067223072052
    },
    {
      "epoch": 1.8260955810546875e-05,
      "model_forward_time": 5.5954484939575195,
      "step": 19148
    },
    {
      "epoch": 1.8260955810546875e-05,
      "step": 19148,
      "training_step_time": 17.0069637298584
    },
    {
      "epoch": 1.8261909484863282e-05,
      "model_forward_time": 5.618108034133911,
      "step": 19149
    },
    {
      "epoch": 1.8261909484863282e-05,
      "step": 19149,
      "training_step_time": 17.103121280670166
    },
    {
      "epoch": 1.8262863159179686e-05,
      "grad_norm": 0.5350869297981262,
      "learning_rate": 1e-05,
      "loss": 0.0512,
      "step": 19150
    },
    {
      "epoch": 1.8262863159179686e-05,
      "model_forward_time": 5.6101179122924805,
      "step": 19150
    },
    {
      "dynamics_loss_avg": 0.047915426827967164,
      "epoch": 1.8262863159179686e-05,
      "step": 19150
    },
    {
      "action_loss_avg": 0.006448726821690798,
      "epoch": 1.8262863159179686e-05,
      "step": 19150
    },
    {
      "epoch": 1.8262863159179686e-05,
      "step": 19150,
      "training_step_time": 17.119267225265503
    },
    {
      "epoch": 1.8263816833496094e-05,
      "model_forward_time": 5.613058567047119,
      "step": 19151
    },
    {
      "epoch": 1.8263816833496094e-05,
      "step": 19151,
      "training_step_time": 17.097962617874146
    },
    {
      "epoch": 1.82647705078125e-05,
      "model_forward_time": 5.954984426498413,
      "step": 19152
    },
    {
      "epoch": 1.82647705078125e-05,
      "step": 19152,
      "training_step_time": 17.37916588783264
    },
    {
      "epoch": 1.8265724182128905e-05,
      "model_forward_time": 5.583751678466797,
      "step": 19153
    },
    {
      "epoch": 1.8265724182128905e-05,
      "step": 19153,
      "training_step_time": 17.02360224723816
    },
    {
      "epoch": 1.8266677856445312e-05,
      "model_forward_time": 5.570149660110474,
      "step": 19154
    },
    {
      "epoch": 1.8266677856445312e-05,
      "step": 19154,
      "training_step_time": 16.99756097793579
    },
    {
      "epoch": 1.826763153076172e-05,
      "model_forward_time": 5.577784061431885,
      "step": 19155
    },
    {
      "epoch": 1.826763153076172e-05,
      "step": 19155,
      "training_step_time": 16.99432945251465
    },
    {
      "epoch": 1.8268585205078124e-05,
      "model_forward_time": 5.575926065444946,
      "step": 19156
    },
    {
      "epoch": 1.8268585205078124e-05,
      "step": 19156,
      "training_step_time": 17.022393226623535
    },
    {
      "epoch": 1.826953887939453e-05,
      "model_forward_time": 5.591160297393799,
      "step": 19157
    },
    {
      "epoch": 1.826953887939453e-05,
      "step": 19157,
      "training_step_time": 17.0343074798584
    },
    {
      "epoch": 1.827049255371094e-05,
      "model_forward_time": 5.9590795040130615,
      "step": 19158
    },
    {
      "epoch": 1.827049255371094e-05,
      "step": 19158,
      "training_step_time": 17.418216228485107
    },
    {
      "epoch": 1.8271446228027343e-05,
      "model_forward_time": 5.604559659957886,
      "step": 19159
    },
    {
      "epoch": 1.8271446228027343e-05,
      "step": 19159,
      "training_step_time": 17.053768634796143
    },
    {
      "epoch": 1.827239990234375e-05,
      "grad_norm": 1.0710052251815796,
      "learning_rate": 1e-05,
      "loss": 0.0643,
      "step": 19160
    },
    {
      "epoch": 1.827239990234375e-05,
      "model_forward_time": 5.599184274673462,
      "step": 19160
    },
    {
      "dynamics_loss_avg": 0.06001141462475061,
      "epoch": 1.827239990234375e-05,
      "step": 19160
    },
    {
      "action_loss_avg": 0.00741311782039702,
      "epoch": 1.827239990234375e-05,
      "step": 19160
    },
    {
      "epoch": 1.827239990234375e-05,
      "step": 19160,
      "training_step_time": 17.057180166244507
    },
    {
      "epoch": 1.8273353576660158e-05,
      "model_forward_time": 5.607028484344482,
      "step": 19161
    },
    {
      "epoch": 1.8273353576660158e-05,
      "step": 19161,
      "training_step_time": 17.08187198638916
    },
    {
      "epoch": 1.827430725097656e-05,
      "model_forward_time": 5.601577520370483,
      "step": 19162
    },
    {
      "epoch": 1.827430725097656e-05,
      "step": 19162,
      "training_step_time": 17.080170154571533
    },
    {
      "epoch": 1.827526092529297e-05,
      "model_forward_time": 5.606210231781006,
      "step": 19163
    },
    {
      "epoch": 1.827526092529297e-05,
      "step": 19163,
      "training_step_time": 17.454339265823364
    },
    {
      "epoch": 1.8276214599609376e-05,
      "model_forward_time": 5.609398365020752,
      "step": 19164
    },
    {
      "epoch": 1.8276214599609376e-05,
      "step": 19164,
      "training_step_time": 17.090848684310913
    },
    {
      "epoch": 1.827716827392578e-05,
      "model_forward_time": 5.61418080329895,
      "step": 19165
    },
    {
      "epoch": 1.827716827392578e-05,
      "step": 19165,
      "training_step_time": 17.10105872154236
    },
    {
      "epoch": 1.8278121948242188e-05,
      "model_forward_time": 5.599305629730225,
      "step": 19166
    },
    {
      "epoch": 1.8278121948242188e-05,
      "step": 19166,
      "training_step_time": 17.070136547088623
    },
    {
      "epoch": 1.8279075622558595e-05,
      "model_forward_time": 5.601420640945435,
      "step": 19167
    },
    {
      "epoch": 1.8279075622558595e-05,
      "step": 19167,
      "training_step_time": 17.057297945022583
    },
    {
      "epoch": 1.8280029296875e-05,
      "model_forward_time": 5.599149227142334,
      "step": 19168
    },
    {
      "epoch": 1.8280029296875e-05,
      "step": 19168,
      "training_step_time": 17.036136865615845
    },
    {
      "epoch": 1.8280982971191407e-05,
      "model_forward_time": 5.9481165409088135,
      "step": 19169
    },
    {
      "epoch": 1.8280982971191407e-05,
      "step": 19169,
      "training_step_time": 17.369781017303467
    },
    {
      "epoch": 1.8281936645507814e-05,
      "grad_norm": 1.1493678092956543,
      "learning_rate": 1e-05,
      "loss": 0.0726,
      "step": 19170
    },
    {
      "epoch": 1.8281936645507814e-05,
      "model_forward_time": 5.593585729598999,
      "step": 19170
    },
    {
      "dynamics_loss_avg": 0.06504270248115063,
      "epoch": 1.8281936645507814e-05,
      "step": 19170
    },
    {
      "action_loss_avg": 0.007776431948877871,
      "epoch": 1.8281936645507814e-05,
      "step": 19170
    },
    {
      "epoch": 1.8281936645507814e-05,
      "step": 19170,
      "training_step_time": 17.059655904769897
    },
    {
      "epoch": 1.8282890319824218e-05,
      "model_forward_time": 5.6019651889801025,
      "step": 19171
    },
    {
      "epoch": 1.8282890319824218e-05,
      "step": 19171,
      "training_step_time": 17.04927706718445
    },
    {
      "epoch": 1.8283843994140626e-05,
      "model_forward_time": 5.601311206817627,
      "step": 19172
    },
    {
      "epoch": 1.8283843994140626e-05,
      "step": 19172,
      "training_step_time": 17.04018235206604
    },
    {
      "epoch": 1.828479766845703e-05,
      "model_forward_time": 5.589677333831787,
      "step": 19173
    },
    {
      "epoch": 1.828479766845703e-05,
      "step": 19173,
      "training_step_time": 17.039726495742798
    },
    {
      "epoch": 1.8285751342773437e-05,
      "model_forward_time": 5.938265800476074,
      "step": 19174
    },
    {
      "epoch": 1.8285751342773437e-05,
      "step": 19174,
      "training_step_time": 17.389827728271484
    },
    {
      "epoch": 1.8286705017089844e-05,
      "model_forward_time": 5.61494255065918,
      "step": 19175
    },
    {
      "epoch": 1.8286705017089844e-05,
      "step": 19175,
      "training_step_time": 17.100886821746826
    },
    {
      "epoch": 1.828765869140625e-05,
      "model_forward_time": 5.646918058395386,
      "step": 19176
    },
    {
      "epoch": 1.828765869140625e-05,
      "step": 19176,
      "training_step_time": 17.117722272872925
    },
    {
      "epoch": 1.8288612365722656e-05,
      "model_forward_time": 5.586862564086914,
      "step": 19177
    },
    {
      "epoch": 1.8288612365722656e-05,
      "step": 19177,
      "training_step_time": 17.02217960357666
    },
    {
      "epoch": 1.8289566040039063e-05,
      "model_forward_time": 5.576230525970459,
      "step": 19178
    },
    {
      "epoch": 1.8289566040039063e-05,
      "step": 19178,
      "training_step_time": 16.985262393951416
    },
    {
      "epoch": 1.8290519714355467e-05,
      "model_forward_time": 5.577930688858032,
      "step": 19179
    },
    {
      "epoch": 1.8290519714355467e-05,
      "step": 19179,
      "training_step_time": 17.004658699035645
    },
    {
      "epoch": 1.8291473388671875e-05,
      "grad_norm": 0.3673461079597473,
      "learning_rate": 1e-05,
      "loss": 0.0735,
      "step": 19180
    },
    {
      "epoch": 1.8291473388671875e-05,
      "model_forward_time": 5.989947319030762,
      "step": 19180
    },
    {
      "dynamics_loss_avg": 0.06399994306266307,
      "epoch": 1.8291473388671875e-05,
      "step": 19180
    },
    {
      "action_loss_avg": 0.007776530040428042,
      "epoch": 1.8291473388671875e-05,
      "step": 19180
    },
    {
      "epoch": 1.8291473388671875e-05,
      "step": 19180,
      "training_step_time": 17.433638095855713
    },
    {
      "epoch": 1.8292427062988282e-05,
      "model_forward_time": 5.571812391281128,
      "step": 19181
    },
    {
      "epoch": 1.8292427062988282e-05,
      "step": 19181,
      "training_step_time": 16.997151136398315
    },
    {
      "epoch": 1.8293380737304686e-05,
      "model_forward_time": 5.610050439834595,
      "step": 19182
    },
    {
      "epoch": 1.8293380737304686e-05,
      "step": 19182,
      "training_step_time": 17.065013885498047
    },
    {
      "epoch": 1.8294334411621094e-05,
      "model_forward_time": 5.60240912437439,
      "step": 19183
    },
    {
      "epoch": 1.8294334411621094e-05,
      "step": 19183,
      "training_step_time": 17.099710702896118
    },
    {
      "epoch": 1.82952880859375e-05,
      "model_forward_time": 5.624162673950195,
      "step": 19184
    },
    {
      "epoch": 1.82952880859375e-05,
      "step": 19184,
      "training_step_time": 17.109341859817505
    },
    {
      "epoch": 1.8296241760253905e-05,
      "model_forward_time": 5.960317134857178,
      "step": 19185
    },
    {
      "epoch": 1.8296241760253905e-05,
      "step": 19185,
      "training_step_time": 17.412561416625977
    },
    {
      "epoch": 1.8297195434570312e-05,
      "model_forward_time": 5.601299047470093,
      "step": 19186
    },
    {
      "epoch": 1.8297195434570312e-05,
      "step": 19186,
      "training_step_time": 17.080566883087158
    },
    {
      "epoch": 1.829814910888672e-05,
      "model_forward_time": 5.617207765579224,
      "step": 19187
    },
    {
      "epoch": 1.829814910888672e-05,
      "step": 19187,
      "training_step_time": 17.122928619384766
    },
    {
      "epoch": 1.8299102783203124e-05,
      "model_forward_time": 5.626372337341309,
      "step": 19188
    },
    {
      "epoch": 1.8299102783203124e-05,
      "step": 19188,
      "training_step_time": 17.114821434020996
    },
    {
      "epoch": 1.830005645751953e-05,
      "model_forward_time": 5.596472978591919,
      "step": 19189
    },
    {
      "epoch": 1.830005645751953e-05,
      "step": 19189,
      "training_step_time": 17.01470184326172
    },
    {
      "epoch": 1.830101013183594e-05,
      "grad_norm": 1.083755612373352,
      "learning_rate": 1e-05,
      "loss": 0.044,
      "step": 19190
    },
    {
      "epoch": 1.830101013183594e-05,
      "model_forward_time": 5.5876429080963135,
      "step": 19190
    },
    {
      "dynamics_loss_avg": 0.039608131349086764,
      "epoch": 1.830101013183594e-05,
      "step": 19190
    },
    {
      "action_loss_avg": 0.004404283117037266,
      "epoch": 1.830101013183594e-05,
      "step": 19190
    },
    {
      "epoch": 1.830101013183594e-05,
      "step": 19190,
      "training_step_time": 17.020591259002686
    },
    {
      "epoch": 1.8301963806152343e-05,
      "model_forward_time": 5.958991527557373,
      "step": 19191
    },
    {
      "epoch": 1.8301963806152343e-05,
      "step": 19191,
      "training_step_time": 17.393893480300903
    },
    {
      "epoch": 1.830291748046875e-05,
      "model_forward_time": 5.574887752532959,
      "step": 19192
    },
    {
      "epoch": 1.830291748046875e-05,
      "step": 19192,
      "training_step_time": 17.024955987930298
    },
    {
      "epoch": 1.8303871154785158e-05,
      "model_forward_time": 5.593426942825317,
      "step": 19193
    },
    {
      "epoch": 1.8303871154785158e-05,
      "step": 19193,
      "training_step_time": 17.046947479248047
    },
    {
      "epoch": 1.830482482910156e-05,
      "model_forward_time": 5.600112676620483,
      "step": 19194
    },
    {
      "epoch": 1.830482482910156e-05,
      "step": 19194,
      "training_step_time": 17.067620754241943
    },
    {
      "epoch": 1.830577850341797e-05,
      "model_forward_time": 5.601122856140137,
      "step": 19195
    },
    {
      "epoch": 1.830577850341797e-05,
      "step": 19195,
      "training_step_time": 17.083096265792847
    },
    {
      "epoch": 1.8306732177734376e-05,
      "model_forward_time": 5.613084316253662,
      "step": 19196
    },
    {
      "epoch": 1.8306732177734376e-05,
      "step": 19196,
      "training_step_time": 17.477816343307495
    },
    {
      "epoch": 1.830768585205078e-05,
      "model_forward_time": 5.613114595413208,
      "step": 19197
    },
    {
      "epoch": 1.830768585205078e-05,
      "step": 19197,
      "training_step_time": 17.116045475006104
    },
    {
      "epoch": 1.8308639526367188e-05,
      "model_forward_time": 5.619204759597778,
      "step": 19198
    },
    {
      "epoch": 1.8308639526367188e-05,
      "step": 19198,
      "training_step_time": 17.11145806312561
    },
    {
      "epoch": 1.8309593200683595e-05,
      "model_forward_time": 5.614377975463867,
      "step": 19199
    },
    {
      "epoch": 1.8309593200683595e-05,
      "step": 19199,
      "training_step_time": 17.07041621208191
    },
    {
      "epoch": 1.8310546875e-05,
      "grad_norm": 0.7077670693397522,
      "learning_rate": 1e-05,
      "loss": 0.0649,
      "step": 19200
    },
    {
      "epoch": 1.8310546875e-05,
      "model_forward_time": 5.588066101074219,
      "step": 19200
    },
    {
      "dynamics_loss_avg": 0.05478376094251871,
      "epoch": 1.8310546875e-05,
      "step": 19200
    },
    {
      "action_loss_avg": 0.005876126140356064,
      "epoch": 1.8310546875e-05,
      "step": 19200
    },
    {
      "epoch": 1.8310546875e-05,
      "step": 19200,
      "training_step_time": 17.00221276283264
    },
    {
      "epoch": 1.8311500549316407e-05,
      "model_forward_time": 5.580733776092529,
      "step": 19201
    },
    {
      "epoch": 1.8311500549316407e-05,
      "step": 19201,
      "training_step_time": 16.99518585205078
    },
    {
      "epoch": 1.8312454223632814e-05,
      "model_forward_time": 5.933701276779175,
      "step": 19202
    },
    {
      "epoch": 1.8312454223632814e-05,
      "step": 19202,
      "training_step_time": 17.369629621505737
    },
    {
      "epoch": 1.8313407897949218e-05,
      "model_forward_time": 5.575200319290161,
      "step": 19203
    },
    {
      "epoch": 1.8313407897949218e-05,
      "step": 19203,
      "training_step_time": 17.018866300582886
    },
    {
      "epoch": 1.8314361572265626e-05,
      "model_forward_time": 5.592977285385132,
      "step": 19204
    },
    {
      "epoch": 1.8314361572265626e-05,
      "step": 19204,
      "training_step_time": 17.021749258041382
    },
    {
      "epoch": 1.831531524658203e-05,
      "model_forward_time": 5.586207151412964,
      "step": 19205
    },
    {
      "epoch": 1.831531524658203e-05,
      "step": 19205,
      "training_step_time": 17.04101800918579
    },
    {
      "epoch": 1.8316268920898437e-05,
      "model_forward_time": 5.6020238399505615,
      "step": 19206
    },
    {
      "epoch": 1.8316268920898437e-05,
      "step": 19206,
      "training_step_time": 17.077524662017822
    },
    {
      "epoch": 1.8317222595214844e-05,
      "model_forward_time": 5.956702470779419,
      "step": 19207
    },
    {
      "epoch": 1.8317222595214844e-05,
      "step": 19207,
      "training_step_time": 17.418659925460815
    },
    {
      "epoch": 1.831817626953125e-05,
      "model_forward_time": 5.619158506393433,
      "step": 19208
    },
    {
      "epoch": 1.831817626953125e-05,
      "step": 19208,
      "training_step_time": 17.08345365524292
    },
    {
      "epoch": 1.8319129943847656e-05,
      "model_forward_time": 5.596003532409668,
      "step": 19209
    },
    {
      "epoch": 1.8319129943847656e-05,
      "step": 19209,
      "training_step_time": 17.016817808151245
    },
    {
      "epoch": 1.8320083618164063e-05,
      "grad_norm": 0.848582923412323,
      "learning_rate": 1e-05,
      "loss": 0.0539,
      "step": 19210
    },
    {
      "epoch": 1.8320083618164063e-05,
      "model_forward_time": 5.598158836364746,
      "step": 19210
    },
    {
      "dynamics_loss_avg": 0.049391722120344636,
      "epoch": 1.8320083618164063e-05,
      "step": 19210
    },
    {
      "action_loss_avg": 0.0053913884330540895,
      "epoch": 1.8320083618164063e-05,
      "step": 19210
    },
    {
      "epoch": 1.8320083618164063e-05,
      "step": 19210,
      "training_step_time": 17.045480012893677
    },
    {
      "epoch": 1.8321037292480467e-05,
      "model_forward_time": 5.603117942810059,
      "step": 19211
    },
    {
      "epoch": 1.8321037292480467e-05,
      "step": 19211,
      "training_step_time": 17.07143259048462
    },
    {
      "epoch": 1.8321990966796875e-05,
      "model_forward_time": 5.602729797363281,
      "step": 19212
    },
    {
      "epoch": 1.8321990966796875e-05,
      "step": 19212,
      "training_step_time": 17.059123039245605
    },
    {
      "epoch": 1.8322944641113282e-05,
      "model_forward_time": 5.979355812072754,
      "step": 19213
    },
    {
      "epoch": 1.8322944641113282e-05,
      "step": 19213,
      "training_step_time": 17.456789255142212
    },
    {
      "epoch": 1.8323898315429686e-05,
      "model_forward_time": 5.617049932479858,
      "step": 19214
    },
    {
      "epoch": 1.8323898315429686e-05,
      "step": 19214,
      "training_step_time": 17.032986402511597
    },
    {
      "epoch": 1.8324851989746094e-05,
      "model_forward_time": 5.622176885604858,
      "step": 19215
    },
    {
      "epoch": 1.8324851989746094e-05,
      "step": 19215,
      "training_step_time": 17.11318016052246
    },
    {
      "epoch": 1.83258056640625e-05,
      "model_forward_time": 5.610422372817993,
      "step": 19216
    },
    {
      "epoch": 1.83258056640625e-05,
      "step": 19216,
      "training_step_time": 17.086606979370117
    },
    {
      "epoch": 1.8326759338378905e-05,
      "model_forward_time": 5.609402894973755,
      "step": 19217
    },
    {
      "epoch": 1.8326759338378905e-05,
      "step": 19217,
      "training_step_time": 17.059166431427002
    },
    {
      "epoch": 1.8327713012695312e-05,
      "model_forward_time": 5.941725492477417,
      "step": 19218
    },
    {
      "epoch": 1.8327713012695312e-05,
      "step": 19218,
      "training_step_time": 17.400508642196655
    },
    {
      "epoch": 1.832866668701172e-05,
      "model_forward_time": 5.588107585906982,
      "step": 19219
    },
    {
      "epoch": 1.832866668701172e-05,
      "step": 19219,
      "training_step_time": 17.039351224899292
    },
    {
      "epoch": 1.8329620361328124e-05,
      "grad_norm": 1.0739481449127197,
      "learning_rate": 1e-05,
      "loss": 0.0582,
      "step": 19220
    },
    {
      "epoch": 1.8329620361328124e-05,
      "model_forward_time": 5.590361595153809,
      "step": 19220
    },
    {
      "dynamics_loss_avg": 0.050389119796454905,
      "epoch": 1.8329620361328124e-05,
      "step": 19220
    },
    {
      "action_loss_avg": 0.006162727158516646,
      "epoch": 1.8329620361328124e-05,
      "step": 19220
    },
    {
      "epoch": 1.8329620361328124e-05,
      "step": 19220,
      "training_step_time": 17.044145107269287
    },
    {
      "epoch": 1.833057403564453e-05,
      "model_forward_time": 5.606544256210327,
      "step": 19221
    },
    {
      "epoch": 1.833057403564453e-05,
      "step": 19221,
      "training_step_time": 17.077077627182007
    },
    {
      "epoch": 1.833152770996094e-05,
      "model_forward_time": 5.61331844329834,
      "step": 19222
    },
    {
      "epoch": 1.833152770996094e-05,
      "step": 19222,
      "training_step_time": 17.066434621810913
    },
    {
      "epoch": 1.8332481384277343e-05,
      "model_forward_time": 5.608747482299805,
      "step": 19223
    },
    {
      "epoch": 1.8332481384277343e-05,
      "step": 19223,
      "training_step_time": 17.049089670181274
    },
    {
      "epoch": 1.833343505859375e-05,
      "model_forward_time": 5.961162805557251,
      "step": 19224
    },
    {
      "epoch": 1.833343505859375e-05,
      "step": 19224,
      "training_step_time": 17.40461778640747
    },
    {
      "epoch": 1.8334388732910158e-05,
      "model_forward_time": 5.595941543579102,
      "step": 19225
    },
    {
      "epoch": 1.8334388732910158e-05,
      "step": 19225,
      "training_step_time": 17.044149160385132
    },
    {
      "epoch": 1.833534240722656e-05,
      "model_forward_time": 5.581298589706421,
      "step": 19226
    },
    {
      "epoch": 1.833534240722656e-05,
      "step": 19226,
      "training_step_time": 17.033382415771484
    },
    {
      "epoch": 1.833629608154297e-05,
      "model_forward_time": 5.594663381576538,
      "step": 19227
    },
    {
      "epoch": 1.833629608154297e-05,
      "step": 19227,
      "training_step_time": 17.055655002593994
    },
    {
      "epoch": 1.8337249755859376e-05,
      "model_forward_time": 5.598820924758911,
      "step": 19228
    },
    {
      "epoch": 1.8337249755859376e-05,
      "step": 19228,
      "training_step_time": 17.07506513595581
    },
    {
      "epoch": 1.833820343017578e-05,
      "model_forward_time": 5.610947132110596,
      "step": 19229
    },
    {
      "epoch": 1.833820343017578e-05,
      "step": 19229,
      "training_step_time": 17.45893383026123
    },
    {
      "epoch": 1.8339157104492188e-05,
      "grad_norm": 0.8614616990089417,
      "learning_rate": 1e-05,
      "loss": 0.0608,
      "step": 19230
    },
    {
      "epoch": 1.8339157104492188e-05,
      "model_forward_time": 5.615564584732056,
      "step": 19230
    },
    {
      "dynamics_loss_avg": 0.05522062536329031,
      "epoch": 1.8339157104492188e-05,
      "step": 19230
    },
    {
      "action_loss_avg": 0.00656951303826645,
      "epoch": 1.8339157104492188e-05,
      "step": 19230
    },
    {
      "epoch": 1.8339157104492188e-05,
      "step": 19230,
      "training_step_time": 17.11005187034607
    },
    {
      "epoch": 1.8340110778808595e-05,
      "model_forward_time": 5.620702028274536,
      "step": 19231
    },
    {
      "epoch": 1.8340110778808595e-05,
      "step": 19231,
      "training_step_time": 17.11430335044861
    },
    {
      "epoch": 1.8341064453125e-05,
      "model_forward_time": 5.615351438522339,
      "step": 19232
    },
    {
      "epoch": 1.8341064453125e-05,
      "step": 19232,
      "training_step_time": 17.100044012069702
    },
    {
      "epoch": 1.8342018127441407e-05,
      "model_forward_time": 5.608039617538452,
      "step": 19233
    },
    {
      "epoch": 1.8342018127441407e-05,
      "step": 19233,
      "training_step_time": 17.06998586654663
    },
    {
      "epoch": 1.8342971801757814e-05,
      "model_forward_time": 5.582297325134277,
      "step": 19234
    },
    {
      "epoch": 1.8342971801757814e-05,
      "step": 19234,
      "training_step_time": 17.005313634872437
    },
    {
      "epoch": 1.8343925476074218e-05,
      "model_forward_time": 5.946208715438843,
      "step": 19235
    },
    {
      "epoch": 1.8343925476074218e-05,
      "step": 19235,
      "training_step_time": 17.355405569076538
    },
    {
      "epoch": 1.8344879150390626e-05,
      "model_forward_time": 5.586049795150757,
      "step": 19236
    },
    {
      "epoch": 1.8344879150390626e-05,
      "step": 19236,
      "training_step_time": 17.015549898147583
    },
    {
      "epoch": 1.834583282470703e-05,
      "model_forward_time": 5.573341608047485,
      "step": 19237
    },
    {
      "epoch": 1.834583282470703e-05,
      "step": 19237,
      "training_step_time": 17.024770975112915
    },
    {
      "epoch": 1.8346786499023437e-05,
      "model_forward_time": 5.5871195793151855,
      "step": 19238
    },
    {
      "epoch": 1.8346786499023437e-05,
      "step": 19238,
      "training_step_time": 17.046096563339233
    },
    {
      "epoch": 1.8347740173339844e-05,
      "model_forward_time": 5.611537933349609,
      "step": 19239
    },
    {
      "epoch": 1.8347740173339844e-05,
      "step": 19239,
      "training_step_time": 17.088154315948486
    },
    {
      "epoch": 1.834869384765625e-05,
      "grad_norm": 0.7579537630081177,
      "learning_rate": 1e-05,
      "loss": 0.0499,
      "step": 19240
    },
    {
      "epoch": 1.834869384765625e-05,
      "model_forward_time": 5.982555627822876,
      "step": 19240
    },
    {
      "dynamics_loss_avg": 0.04544783271849155,
      "epoch": 1.834869384765625e-05,
      "step": 19240
    },
    {
      "action_loss_avg": 0.0058408842538483444,
      "epoch": 1.834869384765625e-05,
      "step": 19240
    },
    {
      "epoch": 1.834869384765625e-05,
      "step": 19240,
      "training_step_time": 17.45698070526123
    },
    {
      "epoch": 1.8349647521972656e-05,
      "model_forward_time": 5.619884729385376,
      "step": 19241
    },
    {
      "epoch": 1.8349647521972656e-05,
      "step": 19241,
      "training_step_time": 17.09982442855835
    },
    {
      "epoch": 1.8350601196289063e-05,
      "model_forward_time": 5.60582709312439,
      "step": 19242
    },
    {
      "epoch": 1.8350601196289063e-05,
      "step": 19242,
      "training_step_time": 17.085189819335938
    },
    {
      "epoch": 1.8351554870605467e-05,
      "model_forward_time": 5.607458591461182,
      "step": 19243
    },
    {
      "epoch": 1.8351554870605467e-05,
      "step": 19243,
      "training_step_time": 17.05859899520874
    },
    {
      "epoch": 1.8352508544921875e-05,
      "model_forward_time": 5.601925849914551,
      "step": 19244
    },
    {
      "epoch": 1.8352508544921875e-05,
      "step": 19244,
      "training_step_time": 17.039719104766846
    },
    {
      "epoch": 1.8353462219238282e-05,
      "model_forward_time": 5.592886209487915,
      "step": 19245
    },
    {
      "epoch": 1.8353462219238282e-05,
      "step": 19245,
      "training_step_time": 17.036322116851807
    },
    {
      "epoch": 1.8354415893554686e-05,
      "model_forward_time": 5.977011680603027,
      "step": 19246
    },
    {
      "epoch": 1.8354415893554686e-05,
      "step": 19246,
      "training_step_time": 17.447308778762817
    },
    {
      "epoch": 1.8355369567871094e-05,
      "model_forward_time": 5.617282390594482,
      "step": 19247
    },
    {
      "epoch": 1.8355369567871094e-05,
      "step": 19247,
      "training_step_time": 17.094472408294678
    },
    {
      "epoch": 1.83563232421875e-05,
      "model_forward_time": 5.6024534702301025,
      "step": 19248
    },
    {
      "epoch": 1.83563232421875e-05,
      "step": 19248,
      "training_step_time": 17.09168791770935
    },
    {
      "epoch": 1.8357276916503905e-05,
      "model_forward_time": 5.6271045207977295,
      "step": 19249
    },
    {
      "epoch": 1.8357276916503905e-05,
      "step": 19249,
      "training_step_time": 17.1396267414093
    },
    {
      "epoch": 1.8358230590820312e-05,
      "grad_norm": 0.49290597438812256,
      "learning_rate": 1e-05,
      "loss": 0.0422,
      "step": 19250
    },
    {
      "epoch": 1.8358230590820312e-05,
      "model_forward_time": 5.618704319000244,
      "step": 19250
    },
    {
      "dynamics_loss_avg": 0.037886359915137294,
      "epoch": 1.8358230590820312e-05,
      "step": 19250
    },
    {
      "action_loss_avg": 0.0036124961450695993,
      "epoch": 1.8358230590820312e-05,
      "step": 19250
    },
    {
      "epoch": 1.8358230590820312e-05,
      "step": 19250,
      "training_step_time": 17.108954906463623
    },
    {
      "epoch": 1.835918426513672e-05,
      "model_forward_time": 5.961158752441406,
      "step": 19251
    },
    {
      "epoch": 1.835918426513672e-05,
      "step": 19251,
      "training_step_time": 17.407378435134888
    },
    {
      "epoch": 1.8360137939453124e-05,
      "model_forward_time": 5.595320701599121,
      "step": 19252
    },
    {
      "epoch": 1.8360137939453124e-05,
      "step": 19252,
      "training_step_time": 17.03137445449829
    },
    {
      "epoch": 1.836109161376953e-05,
      "model_forward_time": 5.599867582321167,
      "step": 19253
    },
    {
      "epoch": 1.836109161376953e-05,
      "step": 19253,
      "training_step_time": 17.063465356826782
    },
    {
      "epoch": 1.836204528808594e-05,
      "model_forward_time": 5.602320909500122,
      "step": 19254
    },
    {
      "epoch": 1.836204528808594e-05,
      "step": 19254,
      "training_step_time": 17.071309089660645
    },
    {
      "epoch": 1.8362998962402343e-05,
      "model_forward_time": 5.592692613601685,
      "step": 19255
    },
    {
      "epoch": 1.8362998962402343e-05,
      "step": 19255,
      "training_step_time": 17.03218388557434
    },
    {
      "epoch": 1.836395263671875e-05,
      "model_forward_time": 5.584891319274902,
      "step": 19256
    },
    {
      "epoch": 1.836395263671875e-05,
      "step": 19256,
      "training_step_time": 17.037779331207275
    },
    {
      "epoch": 1.8364906311035158e-05,
      "model_forward_time": 5.962275981903076,
      "step": 19257
    },
    {
      "epoch": 1.8364906311035158e-05,
      "step": 19257,
      "training_step_time": 17.421515941619873
    },
    {
      "epoch": 1.836585998535156e-05,
      "model_forward_time": 5.600932359695435,
      "step": 19258
    },
    {
      "epoch": 1.836585998535156e-05,
      "step": 19258,
      "training_step_time": 17.105769395828247
    },
    {
      "epoch": 1.836681365966797e-05,
      "model_forward_time": 5.630488872528076,
      "step": 19259
    },
    {
      "epoch": 1.836681365966797e-05,
      "step": 19259,
      "training_step_time": 17.133493185043335
    },
    {
      "epoch": 1.8367767333984376e-05,
      "grad_norm": 0.7219009399414062,
      "learning_rate": 1e-05,
      "loss": 0.0437,
      "step": 19260
    },
    {
      "epoch": 1.8367767333984376e-05,
      "model_forward_time": 5.6084794998168945,
      "step": 19260
    },
    {
      "dynamics_loss_avg": 0.04806007072329521,
      "epoch": 1.8367767333984376e-05,
      "step": 19260
    },
    {
      "action_loss_avg": 0.005112786253448576,
      "epoch": 1.8367767333984376e-05,
      "step": 19260
    },
    {
      "epoch": 1.8367767333984376e-05,
      "step": 19260,
      "training_step_time": 17.067968368530273
    },
    {
      "epoch": 1.836872100830078e-05,
      "model_forward_time": 5.5889246463775635,
      "step": 19261
    },
    {
      "epoch": 1.836872100830078e-05,
      "step": 19261,
      "training_step_time": 17.02611780166626
    },
    {
      "epoch": 1.8369674682617188e-05,
      "model_forward_time": 5.581604957580566,
      "step": 19262
    },
    {
      "epoch": 1.8369674682617188e-05,
      "step": 19262,
      "training_step_time": 17.371756315231323
    },
    {
      "epoch": 1.8370628356933595e-05,
      "model_forward_time": 5.588724374771118,
      "step": 19263
    },
    {
      "epoch": 1.8370628356933595e-05,
      "step": 19263,
      "training_step_time": 17.018941640853882
    },
    {
      "epoch": 1.837158203125e-05,
      "model_forward_time": 5.595512390136719,
      "step": 19264
    },
    {
      "epoch": 1.837158203125e-05,
      "step": 19264,
      "training_step_time": 17.066197872161865
    },
    {
      "epoch": 1.8372535705566407e-05,
      "model_forward_time": 5.6095592975616455,
      "step": 19265
    },
    {
      "epoch": 1.8372535705566407e-05,
      "step": 19265,
      "training_step_time": 17.081918239593506
    },
    {
      "epoch": 1.837348937988281e-05,
      "model_forward_time": 5.615322589874268,
      "step": 19266
    },
    {
      "epoch": 1.837348937988281e-05,
      "step": 19266,
      "training_step_time": 17.102582216262817
    },
    {
      "epoch": 1.8374443054199218e-05,
      "model_forward_time": 5.60880184173584,
      "step": 19267
    },
    {
      "epoch": 1.8374443054199218e-05,
      "step": 19267,
      "training_step_time": 17.100815057754517
    },
    {
      "epoch": 1.8375396728515626e-05,
      "model_forward_time": 4.64414119720459,
      "step": 19268
    },
    {
      "epoch": 1.8375396728515626e-05,
      "step": 19268,
      "training_step_time": 13.184435844421387
    },
    {
      "epoch": 1.837635040283203e-05,
      "model_forward_time": 5.606629848480225,
      "step": 19269
    },
    {
      "epoch": 1.837635040283203e-05,
      "step": 19269,
      "training_step_time": 17.08569836616516
    },
    {
      "epoch": 1.8377304077148437e-05,
      "grad_norm": 0.9196962118148804,
      "learning_rate": 1e-05,
      "loss": 0.0575,
      "step": 19270
    },
    {
      "epoch": 1.8377304077148437e-05,
      "model_forward_time": 5.60555100440979,
      "step": 19270
    },
    {
      "dynamics_loss_avg": 0.05040062367916107,
      "epoch": 1.8377304077148437e-05,
      "step": 19270
    },
    {
      "action_loss_avg": 0.00612306222319603,
      "epoch": 1.8377304077148437e-05,
      "step": 19270
    },
    {
      "epoch": 1.8377304077148437e-05,
      "step": 19270,
      "training_step_time": 17.107597589492798
    },
    {
      "epoch": 1.8378257751464845e-05,
      "model_forward_time": 5.621567487716675,
      "step": 19271
    },
    {
      "epoch": 1.8378257751464845e-05,
      "step": 19271,
      "training_step_time": 17.142120122909546
    },
    {
      "epoch": 1.837921142578125e-05,
      "model_forward_time": 5.625030040740967,
      "step": 19272
    },
    {
      "epoch": 1.837921142578125e-05,
      "step": 19272,
      "training_step_time": 17.127882957458496
    },
    {
      "epoch": 1.8380165100097656e-05,
      "model_forward_time": 5.982345104217529,
      "step": 19273
    },
    {
      "epoch": 1.8380165100097656e-05,
      "step": 19273,
      "training_step_time": 17.441929578781128
    },
    {
      "epoch": 1.8381118774414063e-05,
      "model_forward_time": 5.5935704708099365,
      "step": 19274
    },
    {
      "epoch": 1.8381118774414063e-05,
      "step": 19274,
      "training_step_time": 17.00056791305542
    },
    {
      "epoch": 1.8382072448730467e-05,
      "model_forward_time": 5.570083856582642,
      "step": 19275
    },
    {
      "epoch": 1.8382072448730467e-05,
      "step": 19275,
      "training_step_time": 17.00383996963501
    },
    {
      "epoch": 1.8383026123046875e-05,
      "model_forward_time": 5.563250541687012,
      "step": 19276
    },
    {
      "epoch": 1.8383026123046875e-05,
      "step": 19276,
      "training_step_time": 16.985631465911865
    },
    {
      "epoch": 1.8383979797363282e-05,
      "model_forward_time": 5.564754486083984,
      "step": 19277
    },
    {
      "epoch": 1.8383979797363282e-05,
      "step": 19277,
      "training_step_time": 16.98111391067505
    },
    {
      "epoch": 1.8384933471679686e-05,
      "model_forward_time": 5.568916082382202,
      "step": 19278
    },
    {
      "epoch": 1.8384933471679686e-05,
      "step": 19278,
      "training_step_time": 16.99605917930603
    },
    {
      "epoch": 1.8385887145996094e-05,
      "model_forward_time": 5.985609292984009,
      "step": 19279
    },
    {
      "epoch": 1.8385887145996094e-05,
      "step": 19279,
      "training_step_time": 17.437103986740112
    },
    {
      "epoch": 1.83868408203125e-05,
      "grad_norm": 0.43162694573402405,
      "learning_rate": 1e-05,
      "loss": 0.0573,
      "step": 19280
    },
    {
      "epoch": 1.83868408203125e-05,
      "model_forward_time": 5.5873401165008545,
      "step": 19280
    },
    {
      "dynamics_loss_avg": 0.04354754090309143,
      "epoch": 1.83868408203125e-05,
      "step": 19280
    },
    {
      "action_loss_avg": 0.005317490291781723,
      "epoch": 1.83868408203125e-05,
      "step": 19280
    },
    {
      "epoch": 1.83868408203125e-05,
      "step": 19280,
      "training_step_time": 17.0616774559021
    },
    {
      "epoch": 1.8387794494628905e-05,
      "model_forward_time": 5.618495941162109,
      "step": 19281
    },
    {
      "epoch": 1.8387794494628905e-05,
      "step": 19281,
      "training_step_time": 17.134114742279053
    },
    {
      "epoch": 1.8388748168945313e-05,
      "model_forward_time": 5.636880159378052,
      "step": 19282
    },
    {
      "epoch": 1.8388748168945313e-05,
      "step": 19282,
      "training_step_time": 17.163346529006958
    },
    {
      "epoch": 1.838970184326172e-05,
      "model_forward_time": 5.621133327484131,
      "step": 19283
    },
    {
      "epoch": 1.838970184326172e-05,
      "step": 19283,
      "training_step_time": 17.106801748275757
    },
    {
      "epoch": 1.8390655517578124e-05,
      "model_forward_time": 5.96444296836853,
      "step": 19284
    },
    {
      "epoch": 1.8390655517578124e-05,
      "step": 19284,
      "training_step_time": 17.439852237701416
    },
    {
      "epoch": 1.839160919189453e-05,
      "model_forward_time": 5.600194215774536,
      "step": 19285
    },
    {
      "epoch": 1.839160919189453e-05,
      "step": 19285,
      "training_step_time": 17.066453456878662
    },
    {
      "epoch": 1.839256286621094e-05,
      "model_forward_time": 5.606576681137085,
      "step": 19286
    },
    {
      "epoch": 1.839256286621094e-05,
      "step": 19286,
      "training_step_time": 17.086463689804077
    },
    {
      "epoch": 1.8393516540527343e-05,
      "model_forward_time": 5.612898826599121,
      "step": 19287
    },
    {
      "epoch": 1.8393516540527343e-05,
      "step": 19287,
      "training_step_time": 17.096341848373413
    },
    {
      "epoch": 1.839447021484375e-05,
      "model_forward_time": 5.6080992221832275,
      "step": 19288
    },
    {
      "epoch": 1.839447021484375e-05,
      "step": 19288,
      "training_step_time": 17.08436679840088
    },
    {
      "epoch": 1.8395423889160158e-05,
      "model_forward_time": 5.596612453460693,
      "step": 19289
    },
    {
      "epoch": 1.8395423889160158e-05,
      "step": 19289,
      "training_step_time": 17.050519227981567
    },
    {
      "epoch": 1.839637756347656e-05,
      "grad_norm": 0.292254239320755,
      "learning_rate": 1e-05,
      "loss": 0.0505,
      "step": 19290
    },
    {
      "epoch": 1.839637756347656e-05,
      "model_forward_time": 5.944789886474609,
      "step": 19290
    },
    {
      "dynamics_loss_avg": 0.052017856016755105,
      "epoch": 1.839637756347656e-05,
      "step": 19290
    },
    {
      "action_loss_avg": 0.004769365349784493,
      "epoch": 1.839637756347656e-05,
      "step": 19290
    },
    {
      "epoch": 1.839637756347656e-05,
      "step": 19290,
      "training_step_time": 17.39157199859619
    },
    {
      "epoch": 1.839733123779297e-05,
      "model_forward_time": 5.584336519241333,
      "step": 19291
    },
    {
      "epoch": 1.839733123779297e-05,
      "step": 19291,
      "training_step_time": 17.04086923599243
    },
    {
      "epoch": 1.8398284912109377e-05,
      "model_forward_time": 5.60197901725769,
      "step": 19292
    },
    {
      "epoch": 1.8398284912109377e-05,
      "step": 19292,
      "training_step_time": 17.096991062164307
    },
    {
      "epoch": 1.839923858642578e-05,
      "model_forward_time": 5.61092734336853,
      "step": 19293
    },
    {
      "epoch": 1.839923858642578e-05,
      "step": 19293,
      "training_step_time": 17.083969354629517
    },
    {
      "epoch": 1.8400192260742188e-05,
      "model_forward_time": 5.615461111068726,
      "step": 19294
    },
    {
      "epoch": 1.8400192260742188e-05,
      "step": 19294,
      "training_step_time": 17.10457134246826
    },
    {
      "epoch": 1.8401145935058595e-05,
      "model_forward_time": 5.606564521789551,
      "step": 19295
    },
    {
      "epoch": 1.8401145935058595e-05,
      "step": 19295,
      "training_step_time": 17.44194722175598
    },
    {
      "epoch": 1.8402099609375e-05,
      "model_forward_time": 5.598819017410278,
      "step": 19296
    },
    {
      "epoch": 1.8402099609375e-05,
      "step": 19296,
      "training_step_time": 17.063069343566895
    },
    {
      "epoch": 1.8403053283691407e-05,
      "model_forward_time": 5.598438024520874,
      "step": 19297
    },
    {
      "epoch": 1.8403053283691407e-05,
      "step": 19297,
      "training_step_time": 17.063093423843384
    },
    {
      "epoch": 1.840400695800781e-05,
      "model_forward_time": 5.6065833568573,
      "step": 19298
    },
    {
      "epoch": 1.840400695800781e-05,
      "step": 19298,
      "training_step_time": 17.095147848129272
    },
    {
      "epoch": 1.8404960632324218e-05,
      "model_forward_time": 5.612333536148071,
      "step": 19299
    },
    {
      "epoch": 1.8404960632324218e-05,
      "step": 19299,
      "training_step_time": 17.09698796272278
    },
    {
      "epoch": 1.8405914306640626e-05,
      "grad_norm": 0.7378087639808655,
      "learning_rate": 1e-05,
      "loss": 0.0731,
      "step": 19300
    },
    {
      "epoch": 1.8405914306640626e-05,
      "model_forward_time": 5.620651960372925,
      "step": 19300
    },
    {
      "dynamics_loss_avg": 0.06296798028051853,
      "epoch": 1.8405914306640626e-05,
      "step": 19300
    },
    {
      "action_loss_avg": 0.00769906872883439,
      "epoch": 1.8405914306640626e-05,
      "step": 19300
    },
    {
      "epoch": 1.8405914306640626e-05,
      "step": 19300,
      "training_step_time": 17.092195510864258
    },
    {
      "epoch": 1.840686798095703e-05,
      "model_forward_time": 5.98037576675415,
      "step": 19301
    },
    {
      "epoch": 1.840686798095703e-05,
      "step": 19301,
      "training_step_time": 17.443326473236084
    },
    {
      "epoch": 1.8407821655273437e-05,
      "model_forward_time": 5.584434270858765,
      "step": 19302
    },
    {
      "epoch": 1.8407821655273437e-05,
      "step": 19302,
      "training_step_time": 17.03351926803589
    },
    {
      "epoch": 1.8408775329589845e-05,
      "model_forward_time": 5.583942174911499,
      "step": 19303
    },
    {
      "epoch": 1.8408775329589845e-05,
      "step": 19303,
      "training_step_time": 17.02561330795288
    },
    {
      "epoch": 1.840972900390625e-05,
      "model_forward_time": 5.592109441757202,
      "step": 19304
    },
    {
      "epoch": 1.840972900390625e-05,
      "step": 19304,
      "training_step_time": 17.0631742477417
    },
    {
      "epoch": 1.8410682678222656e-05,
      "model_forward_time": 5.605591535568237,
      "step": 19305
    },
    {
      "epoch": 1.8410682678222656e-05,
      "step": 19305,
      "training_step_time": 17.088536262512207
    },
    {
      "epoch": 1.8411636352539063e-05,
      "model_forward_time": 5.941925048828125,
      "step": 19306
    },
    {
      "epoch": 1.8411636352539063e-05,
      "step": 19306,
      "training_step_time": 17.41433048248291
    },
    {
      "epoch": 1.8412590026855467e-05,
      "model_forward_time": 5.600581169128418,
      "step": 19307
    },
    {
      "epoch": 1.8412590026855467e-05,
      "step": 19307,
      "training_step_time": 17.085657358169556
    },
    {
      "epoch": 1.8413543701171875e-05,
      "model_forward_time": 5.60742974281311,
      "step": 19308
    },
    {
      "epoch": 1.8413543701171875e-05,
      "step": 19308,
      "training_step_time": 17.092965364456177
    },
    {
      "epoch": 1.8414497375488282e-05,
      "model_forward_time": 5.6158447265625,
      "step": 19309
    },
    {
      "epoch": 1.8414497375488282e-05,
      "step": 19309,
      "training_step_time": 17.09355902671814
    },
    {
      "epoch": 1.8415451049804686e-05,
      "grad_norm": 0.43276846408843994,
      "learning_rate": 1e-05,
      "loss": 0.0733,
      "step": 19310
    },
    {
      "epoch": 1.8415451049804686e-05,
      "model_forward_time": 5.616560220718384,
      "step": 19310
    },
    {
      "dynamics_loss_avg": 0.06262869201600552,
      "epoch": 1.8415451049804686e-05,
      "step": 19310
    },
    {
      "action_loss_avg": 0.0068397146416828035,
      "epoch": 1.8415451049804686e-05,
      "step": 19310
    },
    {
      "epoch": 1.8415451049804686e-05,
      "step": 19310,
      "training_step_time": 17.107959747314453
    },
    {
      "epoch": 1.8416404724121094e-05,
      "model_forward_time": 5.6115944385528564,
      "step": 19311
    },
    {
      "epoch": 1.8416404724121094e-05,
      "step": 19311,
      "training_step_time": 17.104297637939453
    },
    {
      "epoch": 1.84173583984375e-05,
      "model_forward_time": 5.968539476394653,
      "step": 19312
    },
    {
      "epoch": 1.84173583984375e-05,
      "step": 19312,
      "training_step_time": 17.45517611503601
    },
    {
      "epoch": 1.8418312072753905e-05,
      "model_forward_time": 5.617607593536377,
      "step": 19313
    },
    {
      "epoch": 1.8418312072753905e-05,
      "step": 19313,
      "training_step_time": 17.121309280395508
    },
    {
      "epoch": 1.8419265747070313e-05,
      "model_forward_time": 5.615917682647705,
      "step": 19314
    },
    {
      "epoch": 1.8419265747070313e-05,
      "step": 19314,
      "training_step_time": 17.091707229614258
    },
    {
      "epoch": 1.842021942138672e-05,
      "model_forward_time": 5.588314533233643,
      "step": 19315
    },
    {
      "epoch": 1.842021942138672e-05,
      "step": 19315,
      "training_step_time": 17.002525329589844
    },
    {
      "epoch": 1.8421173095703124e-05,
      "model_forward_time": 5.569623231887817,
      "step": 19316
    },
    {
      "epoch": 1.8421173095703124e-05,
      "step": 19316,
      "training_step_time": 16.99368405342102
    },
    {
      "epoch": 1.842212677001953e-05,
      "model_forward_time": 5.934555768966675,
      "step": 19317
    },
    {
      "epoch": 1.842212677001953e-05,
      "step": 19317,
      "training_step_time": 17.367771863937378
    },
    {
      "epoch": 1.842308044433594e-05,
      "model_forward_time": 5.5586512088775635,
      "step": 19318
    },
    {
      "epoch": 1.842308044433594e-05,
      "step": 19318,
      "training_step_time": 16.982174158096313
    },
    {
      "epoch": 1.8424034118652343e-05,
      "model_forward_time": 5.574508905410767,
      "step": 19319
    },
    {
      "epoch": 1.8424034118652343e-05,
      "step": 19319,
      "training_step_time": 16.97723364830017
    },
    {
      "epoch": 1.842498779296875e-05,
      "grad_norm": 0.7157536745071411,
      "learning_rate": 1e-05,
      "loss": 0.0558,
      "step": 19320
    },
    {
      "epoch": 1.842498779296875e-05,
      "model_forward_time": 5.574702262878418,
      "step": 19320
    },
    {
      "dynamics_loss_avg": 0.05206209234893322,
      "epoch": 1.842498779296875e-05,
      "step": 19320
    },
    {
      "action_loss_avg": 0.0045440543442964556,
      "epoch": 1.842498779296875e-05,
      "step": 19320
    },
    {
      "epoch": 1.842498779296875e-05,
      "step": 19320,
      "training_step_time": 17.01051115989685
    },
    {
      "epoch": 1.8425941467285158e-05,
      "model_forward_time": 5.5857391357421875,
      "step": 19321
    },
    {
      "epoch": 1.8425941467285158e-05,
      "step": 19321,
      "training_step_time": 17.04883861541748
    },
    {
      "epoch": 1.8426895141601562e-05,
      "model_forward_time": 5.597091436386108,
      "step": 19322
    },
    {
      "epoch": 1.8426895141601562e-05,
      "step": 19322,
      "training_step_time": 17.07342553138733
    },
    {
      "epoch": 1.842784881591797e-05,
      "model_forward_time": 5.98920464515686,
      "step": 19323
    },
    {
      "epoch": 1.842784881591797e-05,
      "step": 19323,
      "training_step_time": 17.467645406723022
    },
    {
      "epoch": 1.8428802490234377e-05,
      "model_forward_time": 5.606672525405884,
      "step": 19324
    },
    {
      "epoch": 1.8428802490234377e-05,
      "step": 19324,
      "training_step_time": 17.075517892837524
    },
    {
      "epoch": 1.842975616455078e-05,
      "model_forward_time": 5.595586776733398,
      "step": 19325
    },
    {
      "epoch": 1.842975616455078e-05,
      "step": 19325,
      "training_step_time": 17.050994157791138
    },
    {
      "epoch": 1.8430709838867188e-05,
      "model_forward_time": 5.595325708389282,
      "step": 19326
    },
    {
      "epoch": 1.8430709838867188e-05,
      "step": 19326,
      "training_step_time": 17.03804087638855
    },
    {
      "epoch": 1.8431663513183595e-05,
      "model_forward_time": 5.60062050819397,
      "step": 19327
    },
    {
      "epoch": 1.8431663513183595e-05,
      "step": 19327,
      "training_step_time": 17.068570852279663
    },
    {
      "epoch": 1.84326171875e-05,
      "model_forward_time": 5.59641695022583,
      "step": 19328
    },
    {
      "epoch": 1.84326171875e-05,
      "step": 19328,
      "training_step_time": 17.409387350082397
    },
    {
      "epoch": 1.8433570861816407e-05,
      "model_forward_time": 5.58702278137207,
      "step": 19329
    },
    {
      "epoch": 1.8433570861816407e-05,
      "step": 19329,
      "training_step_time": 17.009491443634033
    },
    {
      "epoch": 1.843452453613281e-05,
      "grad_norm": 0.3949299454689026,
      "learning_rate": 1e-05,
      "loss": 0.0543,
      "step": 19330
    },
    {
      "epoch": 1.843452453613281e-05,
      "model_forward_time": 5.582826375961304,
      "step": 19330
    },
    {
      "dynamics_loss_avg": 0.045774086471647026,
      "epoch": 1.843452453613281e-05,
      "step": 19330
    },
    {
      "action_loss_avg": 0.005618020088877529,
      "epoch": 1.843452453613281e-05,
      "step": 19330
    },
    {
      "epoch": 1.843452453613281e-05,
      "step": 19330,
      "training_step_time": 17.054498434066772
    },
    {
      "epoch": 1.8435478210449218e-05,
      "model_forward_time": 5.6146228313446045,
      "step": 19331
    },
    {
      "epoch": 1.8435478210449218e-05,
      "step": 19331,
      "training_step_time": 17.11169457435608
    },
    {
      "epoch": 1.8436431884765626e-05,
      "model_forward_time": 5.6092588901519775,
      "step": 19332
    },
    {
      "epoch": 1.8436431884765626e-05,
      "step": 19332,
      "training_step_time": 17.11575484275818
    },
    {
      "epoch": 1.843738555908203e-05,
      "model_forward_time": 5.620983839035034,
      "step": 19333
    },
    {
      "epoch": 1.843738555908203e-05,
      "step": 19333,
      "training_step_time": 17.101221323013306
    },
    {
      "epoch": 1.8438339233398437e-05,
      "model_forward_time": 5.963625907897949,
      "step": 19334
    },
    {
      "epoch": 1.8438339233398437e-05,
      "step": 19334,
      "training_step_time": 17.39421558380127
    },
    {
      "epoch": 1.8439292907714845e-05,
      "model_forward_time": 5.584445953369141,
      "step": 19335
    },
    {
      "epoch": 1.8439292907714845e-05,
      "step": 19335,
      "training_step_time": 17.021428108215332
    },
    {
      "epoch": 1.844024658203125e-05,
      "model_forward_time": 5.596805810928345,
      "step": 19336
    },
    {
      "epoch": 1.844024658203125e-05,
      "step": 19336,
      "training_step_time": 17.049360752105713
    },
    {
      "epoch": 1.8441200256347656e-05,
      "model_forward_time": 5.600770711898804,
      "step": 19337
    },
    {
      "epoch": 1.8441200256347656e-05,
      "step": 19337,
      "training_step_time": 17.014219760894775
    },
    {
      "epoch": 1.8442153930664063e-05,
      "model_forward_time": 5.602648973464966,
      "step": 19338
    },
    {
      "epoch": 1.8442153930664063e-05,
      "step": 19338,
      "training_step_time": 17.067570209503174
    },
    {
      "epoch": 1.8443107604980467e-05,
      "model_forward_time": 5.943387031555176,
      "step": 19339
    },
    {
      "epoch": 1.8443107604980467e-05,
      "step": 19339,
      "training_step_time": 17.37205219268799
    },
    {
      "epoch": 1.8444061279296875e-05,
      "grad_norm": 0.7556493282318115,
      "learning_rate": 1e-05,
      "loss": 0.0466,
      "step": 19340
    },
    {
      "epoch": 1.8444061279296875e-05,
      "model_forward_time": 5.595545053482056,
      "step": 19340
    },
    {
      "dynamics_loss_avg": 0.04534101597964764,
      "epoch": 1.8444061279296875e-05,
      "step": 19340
    },
    {
      "action_loss_avg": 0.005408455338329077,
      "epoch": 1.8444061279296875e-05,
      "step": 19340
    },
    {
      "epoch": 1.8444061279296875e-05,
      "step": 19340,
      "training_step_time": 17.102323293685913
    },
    {
      "epoch": 1.8445014953613282e-05,
      "model_forward_time": 5.618946552276611,
      "step": 19341
    },
    {
      "epoch": 1.8445014953613282e-05,
      "step": 19341,
      "training_step_time": 17.137017488479614
    },
    {
      "epoch": 1.8445968627929686e-05,
      "model_forward_time": 5.632267475128174,
      "step": 19342
    },
    {
      "epoch": 1.8445968627929686e-05,
      "step": 19342,
      "training_step_time": 17.13881254196167
    },
    {
      "epoch": 1.8446922302246094e-05,
      "model_forward_time": 5.607375621795654,
      "step": 19343
    },
    {
      "epoch": 1.8446922302246094e-05,
      "step": 19343,
      "training_step_time": 17.04361581802368
    },
    {
      "epoch": 1.84478759765625e-05,
      "model_forward_time": 5.588882684707642,
      "step": 19344
    },
    {
      "epoch": 1.84478759765625e-05,
      "step": 19344,
      "training_step_time": 17.003617763519287
    },
    {
      "epoch": 1.8448829650878905e-05,
      "model_forward_time": 5.9441001415252686,
      "step": 19345
    },
    {
      "epoch": 1.8448829650878905e-05,
      "step": 19345,
      "training_step_time": 17.349685668945312
    },
    {
      "epoch": 1.8449783325195313e-05,
      "model_forward_time": 5.572591304779053,
      "step": 19346
    },
    {
      "epoch": 1.8449783325195313e-05,
      "step": 19346,
      "training_step_time": 16.990201950073242
    },
    {
      "epoch": 1.845073699951172e-05,
      "model_forward_time": 5.5886125564575195,
      "step": 19347
    },
    {
      "epoch": 1.845073699951172e-05,
      "step": 19347,
      "training_step_time": 17.02770709991455
    },
    {
      "epoch": 1.8451690673828124e-05,
      "model_forward_time": 5.5967466831207275,
      "step": 19348
    },
    {
      "epoch": 1.8451690673828124e-05,
      "step": 19348,
      "training_step_time": 17.057600498199463
    },
    {
      "epoch": 1.845264434814453e-05,
      "model_forward_time": 5.607208251953125,
      "step": 19349
    },
    {
      "epoch": 1.845264434814453e-05,
      "step": 19349,
      "training_step_time": 17.081864595413208
    },
    {
      "epoch": 1.845359802246094e-05,
      "grad_norm": 0.8133560419082642,
      "learning_rate": 1e-05,
      "loss": 0.0664,
      "step": 19350
    },
    {
      "epoch": 1.845359802246094e-05,
      "model_forward_time": 5.956873893737793,
      "step": 19350
    },
    {
      "dynamics_loss_avg": 0.058284064568579194,
      "epoch": 1.845359802246094e-05,
      "step": 19350
    },
    {
      "action_loss_avg": 0.005719173257239163,
      "epoch": 1.845359802246094e-05,
      "step": 19350
    },
    {
      "epoch": 1.845359802246094e-05,
      "step": 19350,
      "training_step_time": 17.429771661758423
    },
    {
      "epoch": 1.8454551696777343e-05,
      "model_forward_time": 5.607685565948486,
      "step": 19351
    },
    {
      "epoch": 1.8454551696777343e-05,
      "step": 19351,
      "training_step_time": 17.086241483688354
    },
    {
      "epoch": 1.845550537109375e-05,
      "model_forward_time": 5.611404895782471,
      "step": 19352
    },
    {
      "epoch": 1.845550537109375e-05,
      "step": 19352,
      "training_step_time": 17.091618537902832
    },
    {
      "epoch": 1.8456459045410158e-05,
      "model_forward_time": 5.6102752685546875,
      "step": 19353
    },
    {
      "epoch": 1.8456459045410158e-05,
      "step": 19353,
      "training_step_time": 17.10077214241028
    },
    {
      "epoch": 1.8457412719726562e-05,
      "model_forward_time": 5.616916656494141,
      "step": 19354
    },
    {
      "epoch": 1.8457412719726562e-05,
      "step": 19354,
      "training_step_time": 17.094353437423706
    },
    {
      "epoch": 1.845836639404297e-05,
      "model_forward_time": 5.597084999084473,
      "step": 19355
    },
    {
      "epoch": 1.845836639404297e-05,
      "step": 19355,
      "training_step_time": 17.083349466323853
    },
    {
      "epoch": 1.8459320068359377e-05,
      "model_forward_time": 5.98232364654541,
      "step": 19356
    },
    {
      "epoch": 1.8459320068359377e-05,
      "step": 19356,
      "training_step_time": 17.457662343978882
    },
    {
      "epoch": 1.846027374267578e-05,
      "model_forward_time": 5.6053383350372314,
      "step": 19357
    },
    {
      "epoch": 1.846027374267578e-05,
      "step": 19357,
      "training_step_time": 17.091878652572632
    },
    {
      "epoch": 1.8461227416992188e-05,
      "model_forward_time": 5.601488828659058,
      "step": 19358
    },
    {
      "epoch": 1.8461227416992188e-05,
      "step": 19358,
      "training_step_time": 17.077007055282593
    },
    {
      "epoch": 1.8462181091308595e-05,
      "model_forward_time": 5.606604337692261,
      "step": 19359
    },
    {
      "epoch": 1.8462181091308595e-05,
      "step": 19359,
      "training_step_time": 17.09660267829895
    },
    {
      "epoch": 1.8463134765625e-05,
      "grad_norm": 0.39479371905326843,
      "learning_rate": 1e-05,
      "loss": 0.0582,
      "step": 19360
    },
    {
      "epoch": 1.8463134765625e-05,
      "model_forward_time": 5.607280015945435,
      "step": 19360
    },
    {
      "dynamics_loss_avg": 0.05180923910811543,
      "epoch": 1.8463134765625e-05,
      "step": 19360
    },
    {
      "action_loss_avg": 0.005463624210096896,
      "epoch": 1.8463134765625e-05,
      "step": 19360
    },
    {
      "epoch": 1.8463134765625e-05,
      "step": 19360,
      "training_step_time": 17.088214635849
    },
    {
      "epoch": 1.8464088439941407e-05,
      "model_forward_time": 5.6010565757751465,
      "step": 19361
    },
    {
      "epoch": 1.8464088439941407e-05,
      "step": 19361,
      "training_step_time": 17.432532787322998
    },
    {
      "epoch": 1.846504211425781e-05,
      "model_forward_time": 5.59613299369812,
      "step": 19362
    },
    {
      "epoch": 1.846504211425781e-05,
      "step": 19362,
      "training_step_time": 17.048521041870117
    },
    {
      "epoch": 1.846599578857422e-05,
      "model_forward_time": 5.585855007171631,
      "step": 19363
    },
    {
      "epoch": 1.846599578857422e-05,
      "step": 19363,
      "training_step_time": 17.03471088409424
    },
    {
      "epoch": 1.8466949462890626e-05,
      "model_forward_time": 5.606848239898682,
      "step": 19364
    },
    {
      "epoch": 1.8466949462890626e-05,
      "step": 19364,
      "training_step_time": 17.102609157562256
    },
    {
      "epoch": 1.846790313720703e-05,
      "model_forward_time": 5.627223491668701,
      "step": 19365
    },
    {
      "epoch": 1.846790313720703e-05,
      "step": 19365,
      "training_step_time": 17.140982389450073
    },
    {
      "epoch": 1.8468856811523437e-05,
      "model_forward_time": 5.618385076522827,
      "step": 19366
    },
    {
      "epoch": 1.8468856811523437e-05,
      "step": 19366,
      "training_step_time": 17.130560874938965
    },
    {
      "epoch": 1.8469810485839845e-05,
      "model_forward_time": 5.988381385803223,
      "step": 19367
    },
    {
      "epoch": 1.8469810485839845e-05,
      "step": 19367,
      "training_step_time": 17.451579332351685
    },
    {
      "epoch": 1.847076416015625e-05,
      "model_forward_time": 5.588889122009277,
      "step": 19368
    },
    {
      "epoch": 1.847076416015625e-05,
      "step": 19368,
      "training_step_time": 17.00919222831726
    },
    {
      "epoch": 1.8471717834472656e-05,
      "model_forward_time": 5.564992189407349,
      "step": 19369
    },
    {
      "epoch": 1.8471717834472656e-05,
      "step": 19369,
      "training_step_time": 16.98703622817993
    },
    {
      "epoch": 1.8472671508789063e-05,
      "grad_norm": 0.41700494289398193,
      "learning_rate": 1e-05,
      "loss": 0.061,
      "step": 19370
    },
    {
      "epoch": 1.8472671508789063e-05,
      "model_forward_time": 5.557748079299927,
      "step": 19370
    },
    {
      "dynamics_loss_avg": 0.057760413736104965,
      "epoch": 1.8472671508789063e-05,
      "step": 19370
    },
    {
      "action_loss_avg": 0.00808412090409547,
      "epoch": 1.8472671508789063e-05,
      "step": 19370
    },
    {
      "epoch": 1.8472671508789063e-05,
      "step": 19370,
      "training_step_time": 16.96937656402588
    },
    {
      "epoch": 1.8473625183105467e-05,
      "model_forward_time": 5.549480438232422,
      "step": 19371
    },
    {
      "epoch": 1.8473625183105467e-05,
      "step": 19371,
      "training_step_time": 16.96527862548828
    },
    {
      "epoch": 1.8474578857421875e-05,
      "model_forward_time": 5.932991027832031,
      "step": 19372
    },
    {
      "epoch": 1.8474578857421875e-05,
      "step": 19372,
      "training_step_time": 17.35473942756653
    },
    {
      "epoch": 1.8475532531738282e-05,
      "model_forward_time": 5.5703513622283936,
      "step": 19373
    },
    {
      "epoch": 1.8475532531738282e-05,
      "step": 19373,
      "training_step_time": 17.007699728012085
    },
    {
      "epoch": 1.8476486206054686e-05,
      "model_forward_time": 5.572040796279907,
      "step": 19374
    },
    {
      "epoch": 1.8476486206054686e-05,
      "step": 19374,
      "training_step_time": 17.000693798065186
    },
    {
      "epoch": 1.8477439880371094e-05,
      "model_forward_time": 5.58183741569519,
      "step": 19375
    },
    {
      "epoch": 1.8477439880371094e-05,
      "step": 19375,
      "training_step_time": 17.009705543518066
    },
    {
      "epoch": 1.84783935546875e-05,
      "model_forward_time": 5.586092948913574,
      "step": 19376
    },
    {
      "epoch": 1.84783935546875e-05,
      "step": 19376,
      "training_step_time": 17.05869436264038
    },
    {
      "epoch": 1.8479347229003905e-05,
      "model_forward_time": 5.613475322723389,
      "step": 19377
    },
    {
      "epoch": 1.8479347229003905e-05,
      "step": 19377,
      "training_step_time": 17.134490489959717
    },
    {
      "epoch": 1.8480300903320313e-05,
      "model_forward_time": 5.9825310707092285,
      "step": 19378
    },
    {
      "epoch": 1.8480300903320313e-05,
      "step": 19378,
      "training_step_time": 17.44499683380127
    },
    {
      "epoch": 1.848125457763672e-05,
      "model_forward_time": 5.638695001602173,
      "step": 19379
    },
    {
      "epoch": 1.848125457763672e-05,
      "step": 19379,
      "training_step_time": 17.07299828529358
    },
    {
      "epoch": 1.8482208251953124e-05,
      "grad_norm": 0.36918121576309204,
      "learning_rate": 1e-05,
      "loss": 0.0622,
      "step": 19380
    },
    {
      "epoch": 1.8482208251953124e-05,
      "model_forward_time": 5.604042053222656,
      "step": 19380
    },
    {
      "dynamics_loss_avg": 0.050535543169826266,
      "epoch": 1.8482208251953124e-05,
      "step": 19380
    },
    {
      "action_loss_avg": 0.006596559099853039,
      "epoch": 1.8482208251953124e-05,
      "step": 19380
    },
    {
      "epoch": 1.8482208251953124e-05,
      "step": 19380,
      "training_step_time": 17.03243327140808
    },
    {
      "epoch": 1.848316192626953e-05,
      "model_forward_time": 5.59043025970459,
      "step": 19381
    },
    {
      "epoch": 1.848316192626953e-05,
      "step": 19381,
      "training_step_time": 17.038851261138916
    },
    {
      "epoch": 1.848411560058594e-05,
      "model_forward_time": 5.611344814300537,
      "step": 19382
    },
    {
      "epoch": 1.848411560058594e-05,
      "step": 19382,
      "training_step_time": 17.110318660736084
    },
    {
      "epoch": 1.8485069274902343e-05,
      "model_forward_time": 5.996795892715454,
      "step": 19383
    },
    {
      "epoch": 1.8485069274902343e-05,
      "step": 19383,
      "training_step_time": 17.49873924255371
    },
    {
      "epoch": 1.848602294921875e-05,
      "model_forward_time": 5.608538627624512,
      "step": 19384
    },
    {
      "epoch": 1.848602294921875e-05,
      "step": 19384,
      "training_step_time": 17.083739280700684
    },
    {
      "epoch": 1.8486976623535158e-05,
      "model_forward_time": 5.590635776519775,
      "step": 19385
    },
    {
      "epoch": 1.8486976623535158e-05,
      "step": 19385,
      "training_step_time": 17.033304929733276
    },
    {
      "epoch": 1.8487930297851562e-05,
      "model_forward_time": 5.5809876918792725,
      "step": 19386
    },
    {
      "epoch": 1.8487930297851562e-05,
      "step": 19386,
      "training_step_time": 17.00490713119507
    },
    {
      "epoch": 1.848888397216797e-05,
      "model_forward_time": 5.580157995223999,
      "step": 19387
    },
    {
      "epoch": 1.848888397216797e-05,
      "step": 19387,
      "training_step_time": 17.020430088043213
    },
    {
      "epoch": 1.8489837646484377e-05,
      "model_forward_time": 5.596546411514282,
      "step": 19388
    },
    {
      "epoch": 1.8489837646484377e-05,
      "step": 19388,
      "training_step_time": 17.048636436462402
    },
    {
      "epoch": 1.849079132080078e-05,
      "model_forward_time": 5.975816965103149,
      "step": 19389
    },
    {
      "epoch": 1.849079132080078e-05,
      "step": 19389,
      "training_step_time": 17.463066577911377
    },
    {
      "epoch": 1.8491744995117188e-05,
      "grad_norm": 0.7183869481086731,
      "learning_rate": 1e-05,
      "loss": 0.0545,
      "step": 19390
    },
    {
      "epoch": 1.8491744995117188e-05,
      "model_forward_time": 5.617244005203247,
      "step": 19390
    },
    {
      "dynamics_loss_avg": 0.050701281800866124,
      "epoch": 1.8491744995117188e-05,
      "step": 19390
    },
    {
      "action_loss_avg": 0.006181014818139374,
      "epoch": 1.8491744995117188e-05,
      "step": 19390
    },
    {
      "epoch": 1.8491744995117188e-05,
      "step": 19390,
      "training_step_time": 17.132794857025146
    },
    {
      "epoch": 1.8492698669433592e-05,
      "model_forward_time": 5.62523889541626,
      "step": 19391
    },
    {
      "epoch": 1.8492698669433592e-05,
      "step": 19391,
      "training_step_time": 17.109156131744385
    },
    {
      "epoch": 1.849365234375e-05,
      "model_forward_time": 5.5957350730896,
      "step": 19392
    },
    {
      "epoch": 1.849365234375e-05,
      "step": 19392,
      "training_step_time": 17.037522315979004
    },
    {
      "epoch": 1.8494606018066407e-05,
      "model_forward_time": 5.563077211380005,
      "step": 19393
    },
    {
      "epoch": 1.8494606018066407e-05,
      "step": 19393,
      "training_step_time": 16.999594926834106
    },
    {
      "epoch": 1.849555969238281e-05,
      "model_forward_time": 5.570111274719238,
      "step": 19394
    },
    {
      "epoch": 1.849555969238281e-05,
      "step": 19394,
      "training_step_time": 17.337250471115112
    },
    {
      "epoch": 1.849651336669922e-05,
      "model_forward_time": 5.569319009780884,
      "step": 19395
    },
    {
      "epoch": 1.849651336669922e-05,
      "step": 19395,
      "training_step_time": 16.98399782180786
    },
    {
      "epoch": 1.8497467041015626e-05,
      "model_forward_time": 5.573681831359863,
      "step": 19396
    },
    {
      "epoch": 1.8497467041015626e-05,
      "step": 19396,
      "training_step_time": 16.999674558639526
    },
    {
      "epoch": 1.849842071533203e-05,
      "model_forward_time": 5.5825817584991455,
      "step": 19397
    },
    {
      "epoch": 1.849842071533203e-05,
      "step": 19397,
      "training_step_time": 17.01708197593689
    },
    {
      "epoch": 1.8499374389648437e-05,
      "model_forward_time": 5.574682235717773,
      "step": 19398
    },
    {
      "epoch": 1.8499374389648437e-05,
      "step": 19398,
      "training_step_time": 17.001391172409058
    },
    {
      "epoch": 1.8500328063964845e-05,
      "model_forward_time": 5.593038320541382,
      "step": 19399
    },
    {
      "epoch": 1.8500328063964845e-05,
      "step": 19399,
      "training_step_time": 17.019981861114502
    },
    {
      "epoch": 1.850128173828125e-05,
      "grad_norm": 0.6112749576568604,
      "learning_rate": 1e-05,
      "loss": 0.0565,
      "step": 19400
    },
    {
      "epoch": 1.850128173828125e-05,
      "model_forward_time": 5.98399806022644,
      "step": 19400
    },
    {
      "dynamics_loss_avg": 0.050550188310444354,
      "epoch": 1.850128173828125e-05,
      "step": 19400
    },
    {
      "action_loss_avg": 0.0055902045452967284,
      "epoch": 1.850128173828125e-05,
      "step": 19400
    },
    {
      "epoch": 1.850128173828125e-05,
      "step": 19400,
      "training_step_time": 17.417115926742554
    },
    {
      "epoch": 1.8502235412597656e-05,
      "model_forward_time": 5.5935938358306885,
      "step": 19401
    },
    {
      "epoch": 1.8502235412597656e-05,
      "step": 19401,
      "training_step_time": 17.083568334579468
    },
    {
      "epoch": 1.8503189086914063e-05,
      "model_forward_time": 5.610542297363281,
      "step": 19402
    },
    {
      "epoch": 1.8503189086914063e-05,
      "step": 19402,
      "training_step_time": 17.0881245136261
    },
    {
      "epoch": 1.8504142761230468e-05,
      "model_forward_time": 5.606477737426758,
      "step": 19403
    },
    {
      "epoch": 1.8504142761230468e-05,
      "step": 19403,
      "training_step_time": 17.09141969680786
    },
    {
      "epoch": 1.8505096435546875e-05,
      "model_forward_time": 5.60334324836731,
      "step": 19404
    },
    {
      "epoch": 1.8505096435546875e-05,
      "step": 19404,
      "training_step_time": 17.07306432723999
    },
    {
      "epoch": 1.8506050109863282e-05,
      "model_forward_time": 5.974243402481079,
      "step": 19405
    },
    {
      "epoch": 1.8506050109863282e-05,
      "step": 19405,
      "training_step_time": 17.44106912612915
    },
    {
      "epoch": 1.8507003784179686e-05,
      "model_forward_time": 5.602820873260498,
      "step": 19406
    },
    {
      "epoch": 1.8507003784179686e-05,
      "step": 19406,
      "training_step_time": 17.05189871788025
    },
    {
      "epoch": 1.8507957458496094e-05,
      "model_forward_time": 5.588228702545166,
      "step": 19407
    },
    {
      "epoch": 1.8507957458496094e-05,
      "step": 19407,
      "training_step_time": 17.050135612487793
    },
    {
      "epoch": 1.85089111328125e-05,
      "model_forward_time": 5.596521377563477,
      "step": 19408
    },
    {
      "epoch": 1.85089111328125e-05,
      "step": 19408,
      "training_step_time": 17.051050662994385
    },
    {
      "epoch": 1.8509864807128905e-05,
      "model_forward_time": 5.594470500946045,
      "step": 19409
    },
    {
      "epoch": 1.8509864807128905e-05,
      "step": 19409,
      "training_step_time": 17.024741411209106
    },
    {
      "epoch": 1.8510818481445313e-05,
      "grad_norm": 0.545348584651947,
      "learning_rate": 1e-05,
      "loss": 0.0622,
      "step": 19410
    },
    {
      "epoch": 1.8510818481445313e-05,
      "model_forward_time": 5.590911388397217,
      "step": 19410
    },
    {
      "dynamics_loss_avg": 0.05763031849637627,
      "epoch": 1.8510818481445313e-05,
      "step": 19410
    },
    {
      "action_loss_avg": 0.006675566354533658,
      "epoch": 1.8510818481445313e-05,
      "step": 19410
    },
    {
      "epoch": 1.8510818481445313e-05,
      "step": 19410,
      "training_step_time": 17.02127766609192
    },
    {
      "epoch": 1.851177215576172e-05,
      "model_forward_time": 5.957263708114624,
      "step": 19411
    },
    {
      "epoch": 1.851177215576172e-05,
      "step": 19411,
      "training_step_time": 17.398831844329834
    },
    {
      "epoch": 1.8512725830078124e-05,
      "model_forward_time": 5.5944504737854,
      "step": 19412
    },
    {
      "epoch": 1.8512725830078124e-05,
      "step": 19412,
      "training_step_time": 17.053789615631104
    },
    {
      "epoch": 1.851367950439453e-05,
      "model_forward_time": 5.610502481460571,
      "step": 19413
    },
    {
      "epoch": 1.851367950439453e-05,
      "step": 19413,
      "training_step_time": 17.076292991638184
    },
    {
      "epoch": 1.851463317871094e-05,
      "model_forward_time": 5.604367733001709,
      "step": 19414
    },
    {
      "epoch": 1.851463317871094e-05,
      "step": 19414,
      "training_step_time": 17.050347328186035
    },
    {
      "epoch": 1.8515586853027343e-05,
      "model_forward_time": 5.596442222595215,
      "step": 19415
    },
    {
      "epoch": 1.8515586853027343e-05,
      "step": 19415,
      "training_step_time": 17.042691946029663
    },
    {
      "epoch": 1.851654052734375e-05,
      "model_forward_time": 5.941242456436157,
      "step": 19416
    },
    {
      "epoch": 1.851654052734375e-05,
      "step": 19416,
      "training_step_time": 17.369494438171387
    },
    {
      "epoch": 1.8517494201660158e-05,
      "model_forward_time": 5.5975916385650635,
      "step": 19417
    },
    {
      "epoch": 1.8517494201660158e-05,
      "step": 19417,
      "training_step_time": 17.07458996772766
    },
    {
      "epoch": 1.8518447875976562e-05,
      "model_forward_time": 5.613959550857544,
      "step": 19418
    },
    {
      "epoch": 1.8518447875976562e-05,
      "step": 19418,
      "training_step_time": 17.116354942321777
    },
    {
      "epoch": 1.851940155029297e-05,
      "model_forward_time": 5.623521089553833,
      "step": 19419
    },
    {
      "epoch": 1.851940155029297e-05,
      "step": 19419,
      "training_step_time": 17.132839918136597
    },
    {
      "epoch": 1.8520355224609377e-05,
      "grad_norm": 0.5420309901237488,
      "learning_rate": 1e-05,
      "loss": 0.0564,
      "step": 19420
    },
    {
      "epoch": 1.8520355224609377e-05,
      "model_forward_time": 5.6266398429870605,
      "step": 19420
    },
    {
      "dynamics_loss_avg": 0.05405927244573831,
      "epoch": 1.8520355224609377e-05,
      "step": 19420
    },
    {
      "action_loss_avg": 0.0063286247896030545,
      "epoch": 1.8520355224609377e-05,
      "step": 19420
    },
    {
      "epoch": 1.8520355224609377e-05,
      "step": 19420,
      "training_step_time": 17.115201473236084
    },
    {
      "epoch": 1.852130889892578e-05,
      "model_forward_time": 5.6101086139678955,
      "step": 19421
    },
    {
      "epoch": 1.852130889892578e-05,
      "step": 19421,
      "training_step_time": 17.103419065475464
    },
    {
      "epoch": 1.8522262573242188e-05,
      "model_forward_time": 5.996746778488159,
      "step": 19422
    },
    {
      "epoch": 1.8522262573242188e-05,
      "step": 19422,
      "training_step_time": 17.47913122177124
    },
    {
      "epoch": 1.8523216247558592e-05,
      "model_forward_time": 5.626389026641846,
      "step": 19423
    },
    {
      "epoch": 1.8523216247558592e-05,
      "step": 19423,
      "training_step_time": 17.07476043701172
    },
    {
      "epoch": 1.8524169921875e-05,
      "model_forward_time": 5.592823266983032,
      "step": 19424
    },
    {
      "epoch": 1.8524169921875e-05,
      "step": 19424,
      "training_step_time": 17.053507566452026
    },
    {
      "epoch": 1.8525123596191407e-05,
      "model_forward_time": 5.603338241577148,
      "step": 19425
    },
    {
      "epoch": 1.8525123596191407e-05,
      "step": 19425,
      "training_step_time": 17.069496870040894
    },
    {
      "epoch": 1.852607727050781e-05,
      "model_forward_time": 5.611999273300171,
      "step": 19426
    },
    {
      "epoch": 1.852607727050781e-05,
      "step": 19426,
      "training_step_time": 17.114033937454224
    },
    {
      "epoch": 1.852703094482422e-05,
      "model_forward_time": 5.620075941085815,
      "step": 19427
    },
    {
      "epoch": 1.852703094482422e-05,
      "step": 19427,
      "training_step_time": 17.4597065448761
    },
    {
      "epoch": 1.8527984619140626e-05,
      "model_forward_time": 5.612674236297607,
      "step": 19428
    },
    {
      "epoch": 1.8527984619140626e-05,
      "step": 19428,
      "training_step_time": 17.090394735336304
    },
    {
      "epoch": 1.852893829345703e-05,
      "model_forward_time": 5.592902898788452,
      "step": 19429
    },
    {
      "epoch": 1.852893829345703e-05,
      "step": 19429,
      "training_step_time": 17.011534929275513
    },
    {
      "epoch": 1.8529891967773437e-05,
      "grad_norm": 0.8961918354034424,
      "learning_rate": 1e-05,
      "loss": 0.074,
      "step": 19430
    },
    {
      "epoch": 1.8529891967773437e-05,
      "model_forward_time": 5.573206424713135,
      "step": 19430
    },
    {
      "dynamics_loss_avg": 0.05968859363347292,
      "epoch": 1.8529891967773437e-05,
      "step": 19430
    },
    {
      "action_loss_avg": 0.00614389069378376,
      "epoch": 1.8529891967773437e-05,
      "step": 19430
    },
    {
      "epoch": 1.8529891967773437e-05,
      "step": 19430,
      "training_step_time": 17.0031316280365
    },
    {
      "epoch": 1.8530845642089845e-05,
      "model_forward_time": 5.5611348152160645,
      "step": 19431
    },
    {
      "epoch": 1.8530845642089845e-05,
      "step": 19431,
      "training_step_time": 16.993546724319458
    },
    {
      "epoch": 1.853179931640625e-05,
      "model_forward_time": 5.565796136856079,
      "step": 19432
    },
    {
      "epoch": 1.853179931640625e-05,
      "step": 19432,
      "training_step_time": 16.970882654190063
    },
    {
      "epoch": 1.8532752990722656e-05,
      "model_forward_time": 5.9337334632873535,
      "step": 19433
    },
    {
      "epoch": 1.8532752990722656e-05,
      "step": 19433,
      "training_step_time": 17.350960969924927
    },
    {
      "epoch": 1.8533706665039064e-05,
      "model_forward_time": 5.569300413131714,
      "step": 19434
    },
    {
      "epoch": 1.8533706665039064e-05,
      "step": 19434,
      "training_step_time": 16.99530792236328
    },
    {
      "epoch": 1.8534660339355468e-05,
      "model_forward_time": 5.575428485870361,
      "step": 19435
    },
    {
      "epoch": 1.8534660339355468e-05,
      "step": 19435,
      "training_step_time": 17.01505994796753
    },
    {
      "epoch": 1.8535614013671875e-05,
      "model_forward_time": 5.591374158859253,
      "step": 19436
    },
    {
      "epoch": 1.8535614013671875e-05,
      "step": 19436,
      "training_step_time": 17.05251979827881
    },
    {
      "epoch": 1.8536567687988282e-05,
      "model_forward_time": 5.608381748199463,
      "step": 19437
    },
    {
      "epoch": 1.8536567687988282e-05,
      "step": 19437,
      "training_step_time": 17.07798957824707
    },
    {
      "epoch": 1.8537521362304686e-05,
      "model_forward_time": 5.971186399459839,
      "step": 19438
    },
    {
      "epoch": 1.8537521362304686e-05,
      "step": 19438,
      "training_step_time": 17.450486421585083
    },
    {
      "epoch": 1.8538475036621094e-05,
      "model_forward_time": 5.610811710357666,
      "step": 19439
    },
    {
      "epoch": 1.8538475036621094e-05,
      "step": 19439,
      "training_step_time": 17.126106023788452
    },
    {
      "epoch": 1.85394287109375e-05,
      "grad_norm": 0.2666831314563751,
      "learning_rate": 1e-05,
      "loss": 0.0582,
      "step": 19440
    },
    {
      "epoch": 1.85394287109375e-05,
      "model_forward_time": 5.634557723999023,
      "step": 19440
    },
    {
      "dynamics_loss_avg": 0.05808425843715668,
      "epoch": 1.85394287109375e-05,
      "step": 19440
    },
    {
      "action_loss_avg": 0.005908606527373195,
      "epoch": 1.85394287109375e-05,
      "step": 19440
    },
    {
      "epoch": 1.85394287109375e-05,
      "step": 19440,
      "training_step_time": 17.130394458770752
    },
    {
      "epoch": 1.8540382385253905e-05,
      "model_forward_time": 5.603945732116699,
      "step": 19441
    },
    {
      "epoch": 1.8540382385253905e-05,
      "step": 19441,
      "training_step_time": 17.073488235473633
    },
    {
      "epoch": 1.8541336059570313e-05,
      "model_forward_time": 5.59903621673584,
      "step": 19442
    },
    {
      "epoch": 1.8541336059570313e-05,
      "step": 19442,
      "training_step_time": 17.043780088424683
    },
    {
      "epoch": 1.854228973388672e-05,
      "model_forward_time": 5.585327386856079,
      "step": 19443
    },
    {
      "epoch": 1.854228973388672e-05,
      "step": 19443,
      "training_step_time": 17.01713252067566
    },
    {
      "epoch": 1.8543243408203124e-05,
      "model_forward_time": 5.956331968307495,
      "step": 19444
    },
    {
      "epoch": 1.8543243408203124e-05,
      "step": 19444,
      "training_step_time": 17.411755323410034
    },
    {
      "epoch": 1.854419708251953e-05,
      "model_forward_time": 5.593202829360962,
      "step": 19445
    },
    {
      "epoch": 1.854419708251953e-05,
      "step": 19445,
      "training_step_time": 17.062915086746216
    },
    {
      "epoch": 1.854515075683594e-05,
      "model_forward_time": 5.610397815704346,
      "step": 19446
    },
    {
      "epoch": 1.854515075683594e-05,
      "step": 19446,
      "training_step_time": 17.10058569908142
    },
    {
      "epoch": 1.8546104431152343e-05,
      "model_forward_time": 5.605445384979248,
      "step": 19447
    },
    {
      "epoch": 1.8546104431152343e-05,
      "step": 19447,
      "training_step_time": 17.0952570438385
    },
    {
      "epoch": 1.854705810546875e-05,
      "model_forward_time": 5.616550445556641,
      "step": 19448
    },
    {
      "epoch": 1.854705810546875e-05,
      "step": 19448,
      "training_step_time": 17.111265897750854
    },
    {
      "epoch": 1.8548011779785158e-05,
      "model_forward_time": 5.964805364608765,
      "step": 19449
    },
    {
      "epoch": 1.8548011779785158e-05,
      "step": 19449,
      "training_step_time": 17.4563889503479
    },
    {
      "epoch": 1.8548965454101562e-05,
      "grad_norm": 0.2581881880760193,
      "learning_rate": 1e-05,
      "loss": 0.0462,
      "step": 19450
    },
    {
      "epoch": 1.8548965454101562e-05,
      "model_forward_time": 5.62172269821167,
      "step": 19450
    },
    {
      "dynamics_loss_avg": 0.037125824484974146,
      "epoch": 1.8548965454101562e-05,
      "step": 19450
    },
    {
      "action_loss_avg": 0.003987629734911025,
      "epoch": 1.8548965454101562e-05,
      "step": 19450
    },
    {
      "epoch": 1.8548965454101562e-05,
      "step": 19450,
      "training_step_time": 17.128114461898804
    },
    {
      "epoch": 1.854991912841797e-05,
      "model_forward_time": 5.618885517120361,
      "step": 19451
    },
    {
      "epoch": 1.854991912841797e-05,
      "step": 19451,
      "training_step_time": 17.111802339553833
    },
    {
      "epoch": 1.8550872802734377e-05,
      "model_forward_time": 5.624780893325806,
      "step": 19452
    },
    {
      "epoch": 1.8550872802734377e-05,
      "step": 19452,
      "training_step_time": 17.127334356307983
    },
    {
      "epoch": 1.855182647705078e-05,
      "model_forward_time": 5.6270318031311035,
      "step": 19453
    },
    {
      "epoch": 1.855182647705078e-05,
      "step": 19453,
      "training_step_time": 17.104649305343628
    },
    {
      "epoch": 1.8552780151367188e-05,
      "model_forward_time": 5.615582704544067,
      "step": 19454
    },
    {
      "epoch": 1.8552780151367188e-05,
      "step": 19454,
      "training_step_time": 17.116056203842163
    },
    {
      "epoch": 1.8553733825683592e-05,
      "model_forward_time": 5.980425596237183,
      "step": 19455
    },
    {
      "epoch": 1.8553733825683592e-05,
      "step": 19455,
      "training_step_time": 17.469606399536133
    },
    {
      "epoch": 1.85546875e-05,
      "model_forward_time": 5.607583999633789,
      "step": 19456
    },
    {
      "epoch": 1.85546875e-05,
      "step": 19456,
      "training_step_time": 17.099741220474243
    },
    {
      "epoch": 1.8555641174316407e-05,
      "model_forward_time": 5.60952615737915,
      "step": 19457
    },
    {
      "epoch": 1.8555641174316407e-05,
      "step": 19457,
      "training_step_time": 17.095810413360596
    },
    {
      "epoch": 1.855659484863281e-05,
      "model_forward_time": 5.6117894649505615,
      "step": 19458
    },
    {
      "epoch": 1.855659484863281e-05,
      "step": 19458,
      "training_step_time": 17.107808351516724
    },
    {
      "epoch": 1.855754852294922e-05,
      "model_forward_time": 5.628969430923462,
      "step": 19459
    },
    {
      "epoch": 1.855754852294922e-05,
      "step": 19459,
      "training_step_time": 17.119049310684204
    },
    {
      "epoch": 1.8558502197265626e-05,
      "grad_norm": 0.6580091118812561,
      "learning_rate": 1e-05,
      "loss": 0.0719,
      "step": 19460
    },
    {
      "epoch": 1.8558502197265626e-05,
      "model_forward_time": 5.615553140640259,
      "step": 19460
    },
    {
      "dynamics_loss_avg": 0.06783192120492458,
      "epoch": 1.8558502197265626e-05,
      "step": 19460
    },
    {
      "action_loss_avg": 0.008395749586634338,
      "epoch": 1.8558502197265626e-05,
      "step": 19460
    },
    {
      "epoch": 1.8558502197265626e-05,
      "step": 19460,
      "training_step_time": 17.444247722625732
    },
    {
      "epoch": 1.855945587158203e-05,
      "model_forward_time": 5.608783721923828,
      "step": 19461
    },
    {
      "epoch": 1.855945587158203e-05,
      "step": 19461,
      "training_step_time": 17.056054830551147
    },
    {
      "epoch": 1.8560409545898437e-05,
      "model_forward_time": 5.583616256713867,
      "step": 19462
    },
    {
      "epoch": 1.8560409545898437e-05,
      "step": 19462,
      "training_step_time": 17.00861954689026
    },
    {
      "epoch": 1.8561363220214845e-05,
      "model_forward_time": 5.59209680557251,
      "step": 19463
    },
    {
      "epoch": 1.8561363220214845e-05,
      "step": 19463,
      "training_step_time": 17.00304412841797
    },
    {
      "epoch": 1.856231689453125e-05,
      "model_forward_time": 5.592166900634766,
      "step": 19464
    },
    {
      "epoch": 1.856231689453125e-05,
      "step": 19464,
      "training_step_time": 17.034261226654053
    },
    {
      "epoch": 1.8563270568847656e-05,
      "model_forward_time": 5.594570636749268,
      "step": 19465
    },
    {
      "epoch": 1.8563270568847656e-05,
      "step": 19465,
      "training_step_time": 17.073992490768433
    },
    {
      "epoch": 1.8564224243164064e-05,
      "model_forward_time": 5.964184284210205,
      "step": 19466
    },
    {
      "epoch": 1.8564224243164064e-05,
      "step": 19466,
      "training_step_time": 17.45094084739685
    },
    {
      "epoch": 1.8565177917480468e-05,
      "model_forward_time": 5.603658437728882,
      "step": 19467
    },
    {
      "epoch": 1.8565177917480468e-05,
      "step": 19467,
      "training_step_time": 17.078247785568237
    },
    {
      "epoch": 1.8566131591796875e-05,
      "model_forward_time": 5.599138975143433,
      "step": 19468
    },
    {
      "epoch": 1.8566131591796875e-05,
      "step": 19468,
      "training_step_time": 17.05131244659424
    },
    {
      "epoch": 1.8567085266113282e-05,
      "model_forward_time": 5.590084791183472,
      "step": 19469
    },
    {
      "epoch": 1.8567085266113282e-05,
      "step": 19469,
      "training_step_time": 17.027356147766113
    },
    {
      "epoch": 1.8568038940429686e-05,
      "grad_norm": 1.0387330055236816,
      "learning_rate": 1e-05,
      "loss": 0.0609,
      "step": 19470
    },
    {
      "epoch": 1.8568038940429686e-05,
      "model_forward_time": 5.59406304359436,
      "step": 19470
    },
    {
      "dynamics_loss_avg": 0.0534167792648077,
      "epoch": 1.8568038940429686e-05,
      "step": 19470
    },
    {
      "action_loss_avg": 0.005096117570064962,
      "epoch": 1.8568038940429686e-05,
      "step": 19470
    },
    {
      "epoch": 1.8568038940429686e-05,
      "step": 19470,
      "training_step_time": 17.064147472381592
    },
    {
      "epoch": 1.8568992614746094e-05,
      "model_forward_time": 5.967570543289185,
      "step": 19471
    },
    {
      "epoch": 1.8568992614746094e-05,
      "step": 19471,
      "training_step_time": 17.43720269203186
    },
    {
      "epoch": 1.85699462890625e-05,
      "model_forward_time": 5.605137586593628,
      "step": 19472
    },
    {
      "epoch": 1.85699462890625e-05,
      "step": 19472,
      "training_step_time": 17.07303762435913
    },
    {
      "epoch": 1.8570899963378905e-05,
      "model_forward_time": 5.595284938812256,
      "step": 19473
    },
    {
      "epoch": 1.8570899963378905e-05,
      "step": 19473,
      "training_step_time": 17.06376600265503
    },
    {
      "epoch": 1.8571853637695313e-05,
      "model_forward_time": 5.607409715652466,
      "step": 19474
    },
    {
      "epoch": 1.8571853637695313e-05,
      "step": 19474,
      "training_step_time": 17.090245723724365
    },
    {
      "epoch": 1.857280731201172e-05,
      "model_forward_time": 5.604140996932983,
      "step": 19475
    },
    {
      "epoch": 1.857280731201172e-05,
      "step": 19475,
      "training_step_time": 17.087303400039673
    },
    {
      "epoch": 1.8573760986328124e-05,
      "model_forward_time": 5.606146335601807,
      "step": 19476
    },
    {
      "epoch": 1.8573760986328124e-05,
      "step": 19476,
      "training_step_time": 17.08147120475769
    },
    {
      "epoch": 1.857471466064453e-05,
      "model_forward_time": 5.96653938293457,
      "step": 19477
    },
    {
      "epoch": 1.857471466064453e-05,
      "step": 19477,
      "training_step_time": 17.455402374267578
    },
    {
      "epoch": 1.857566833496094e-05,
      "model_forward_time": 5.613556146621704,
      "step": 19478
    },
    {
      "epoch": 1.857566833496094e-05,
      "step": 19478,
      "training_step_time": 17.09850573539734
    },
    {
      "epoch": 1.8576622009277343e-05,
      "model_forward_time": 5.614304065704346,
      "step": 19479
    },
    {
      "epoch": 1.8576622009277343e-05,
      "step": 19479,
      "training_step_time": 17.099778413772583
    },
    {
      "epoch": 1.857757568359375e-05,
      "grad_norm": 0.8362931609153748,
      "learning_rate": 1e-05,
      "loss": 0.0659,
      "step": 19480
    },
    {
      "epoch": 1.857757568359375e-05,
      "model_forward_time": 5.607855796813965,
      "step": 19480
    },
    {
      "dynamics_loss_avg": 0.06148519366979599,
      "epoch": 1.857757568359375e-05,
      "step": 19480
    },
    {
      "action_loss_avg": 0.007683234545402229,
      "epoch": 1.857757568359375e-05,
      "step": 19480
    },
    {
      "epoch": 1.857757568359375e-05,
      "step": 19480,
      "training_step_time": 17.070867776870728
    },
    {
      "epoch": 1.8578529357910158e-05,
      "model_forward_time": 5.607560873031616,
      "step": 19481
    },
    {
      "epoch": 1.8578529357910158e-05,
      "step": 19481,
      "training_step_time": 17.08125352859497
    },
    {
      "epoch": 1.8579483032226562e-05,
      "model_forward_time": 5.954987287521362,
      "step": 19482
    },
    {
      "epoch": 1.8579483032226562e-05,
      "step": 19482,
      "training_step_time": 17.3831524848938
    },
    {
      "epoch": 1.858043670654297e-05,
      "model_forward_time": 5.584185361862183,
      "step": 19483
    },
    {
      "epoch": 1.858043670654297e-05,
      "step": 19483,
      "training_step_time": 17.013627529144287
    },
    {
      "epoch": 1.8581390380859377e-05,
      "model_forward_time": 5.57688307762146,
      "step": 19484
    },
    {
      "epoch": 1.8581390380859377e-05,
      "step": 19484,
      "training_step_time": 17.014318227767944
    },
    {
      "epoch": 1.858234405517578e-05,
      "model_forward_time": 5.581246376037598,
      "step": 19485
    },
    {
      "epoch": 1.858234405517578e-05,
      "step": 19485,
      "training_step_time": 17.00792407989502
    },
    {
      "epoch": 1.8583297729492188e-05,
      "model_forward_time": 5.591247797012329,
      "step": 19486
    },
    {
      "epoch": 1.8583297729492188e-05,
      "step": 19486,
      "training_step_time": 17.050858974456787
    },
    {
      "epoch": 1.8584251403808592e-05,
      "model_forward_time": 5.6064581871032715,
      "step": 19487
    },
    {
      "epoch": 1.8584251403808592e-05,
      "step": 19487,
      "training_step_time": 17.075613737106323
    },
    {
      "epoch": 1.8585205078125e-05,
      "model_forward_time": 5.977323055267334,
      "step": 19488
    },
    {
      "epoch": 1.8585205078125e-05,
      "step": 19488,
      "training_step_time": 17.449455738067627
    },
    {
      "epoch": 1.8586158752441407e-05,
      "model_forward_time": 5.609811544418335,
      "step": 19489
    },
    {
      "epoch": 1.8586158752441407e-05,
      "step": 19489,
      "training_step_time": 17.104223012924194
    },
    {
      "epoch": 1.858711242675781e-05,
      "grad_norm": 0.7372378706932068,
      "learning_rate": 1e-05,
      "loss": 0.0555,
      "step": 19490
    },
    {
      "epoch": 1.858711242675781e-05,
      "model_forward_time": 5.619074821472168,
      "step": 19490
    },
    {
      "dynamics_loss_avg": 0.05326757626608014,
      "epoch": 1.858711242675781e-05,
      "step": 19490
    },
    {
      "action_loss_avg": 0.005097727640531957,
      "epoch": 1.858711242675781e-05,
      "step": 19490
    },
    {
      "epoch": 1.858711242675781e-05,
      "step": 19490,
      "training_step_time": 17.12450623512268
    },
    {
      "epoch": 1.858806610107422e-05,
      "model_forward_time": 5.6213696002960205,
      "step": 19491
    },
    {
      "epoch": 1.858806610107422e-05,
      "step": 19491,
      "training_step_time": 17.102749824523926
    },
    {
      "epoch": 1.8589019775390626e-05,
      "model_forward_time": 5.6048126220703125,
      "step": 19492
    },
    {
      "epoch": 1.8589019775390626e-05,
      "step": 19492,
      "training_step_time": 17.050599813461304
    },
    {
      "epoch": 1.858997344970703e-05,
      "model_forward_time": 5.584509372711182,
      "step": 19493
    },
    {
      "epoch": 1.858997344970703e-05,
      "step": 19493,
      "training_step_time": 17.381216526031494
    },
    {
      "epoch": 1.8590927124023437e-05,
      "model_forward_time": 5.582524299621582,
      "step": 19494
    },
    {
      "epoch": 1.8590927124023437e-05,
      "step": 19494,
      "training_step_time": 16.999292850494385
    },
    {
      "epoch": 1.8591880798339845e-05,
      "model_forward_time": 5.579107999801636,
      "step": 19495
    },
    {
      "epoch": 1.8591880798339845e-05,
      "step": 19495,
      "training_step_time": 17.013927936553955
    },
    {
      "epoch": 1.859283447265625e-05,
      "model_forward_time": 5.604662895202637,
      "step": 19496
    },
    {
      "epoch": 1.859283447265625e-05,
      "step": 19496,
      "training_step_time": 17.102625131607056
    },
    {
      "epoch": 1.8593788146972656e-05,
      "model_forward_time": 5.615062713623047,
      "step": 19497
    },
    {
      "epoch": 1.8593788146972656e-05,
      "step": 19497,
      "training_step_time": 17.116612434387207
    },
    {
      "epoch": 1.8594741821289064e-05,
      "model_forward_time": 5.625004768371582,
      "step": 19498
    },
    {
      "epoch": 1.8594741821289064e-05,
      "step": 19498,
      "training_step_time": 17.12017011642456
    },
    {
      "epoch": 1.8595695495605468e-05,
      "model_forward_time": 5.993676424026489,
      "step": 19499
    },
    {
      "epoch": 1.8595695495605468e-05,
      "step": 19499,
      "training_step_time": 17.470473766326904
    },
    {
      "epoch": 1.8596649169921875e-05,
      "grad_norm": 0.5246766805648804,
      "learning_rate": 1e-05,
      "loss": 0.0643,
      "step": 19500
    },
    {
      "epoch": 1.8596649169921875e-05,
      "model_forward_time": 5.604679584503174,
      "step": 19500
    },
    {
      "dynamics_loss_avg": 0.05559019707143307,
      "epoch": 1.8596649169921875e-05,
      "step": 19500
    },
    {
      "action_loss_avg": 0.00490223771193996,
      "epoch": 1.8596649169921875e-05,
      "step": 19500
    },
    {
      "epoch": 1.8596649169921875e-05,
      "step": 19500,
      "training_step_time": 17.064842224121094
    },
    {
      "epoch": 1.8597602844238282e-05,
      "model_forward_time": 5.599181652069092,
      "step": 19501
    },
    {
      "epoch": 1.8597602844238282e-05,
      "step": 19501,
      "training_step_time": 17.02867031097412
    },
    {
      "epoch": 1.8598556518554686e-05,
      "model_forward_time": 5.592132329940796,
      "step": 19502
    },
    {
      "epoch": 1.8598556518554686e-05,
      "step": 19502,
      "training_step_time": 17.042465448379517
    },
    {
      "epoch": 1.8599510192871094e-05,
      "model_forward_time": 5.605178356170654,
      "step": 19503
    },
    {
      "epoch": 1.8599510192871094e-05,
      "step": 19503,
      "training_step_time": 17.08498764038086
    },
    {
      "epoch": 1.86004638671875e-05,
      "model_forward_time": 5.965663909912109,
      "step": 19504
    },
    {
      "epoch": 1.86004638671875e-05,
      "step": 19504,
      "training_step_time": 17.453753232955933
    },
    {
      "epoch": 1.8601417541503905e-05,
      "model_forward_time": 5.618221282958984,
      "step": 19505
    },
    {
      "epoch": 1.8601417541503905e-05,
      "step": 19505,
      "training_step_time": 17.132464170455933
    },
    {
      "epoch": 1.8602371215820313e-05,
      "model_forward_time": 5.6153342723846436,
      "step": 19506
    },
    {
      "epoch": 1.8602371215820313e-05,
      "step": 19506,
      "training_step_time": 17.09609317779541
    },
    {
      "epoch": 1.860332489013672e-05,
      "model_forward_time": 5.597178220748901,
      "step": 19507
    },
    {
      "epoch": 1.860332489013672e-05,
      "step": 19507,
      "training_step_time": 17.088109016418457
    },
    {
      "epoch": 1.8604278564453124e-05,
      "model_forward_time": 5.598977327346802,
      "step": 19508
    },
    {
      "epoch": 1.8604278564453124e-05,
      "step": 19508,
      "training_step_time": 17.086162567138672
    },
    {
      "epoch": 1.860523223876953e-05,
      "model_forward_time": 5.610254526138306,
      "step": 19509
    },
    {
      "epoch": 1.860523223876953e-05,
      "step": 19509,
      "training_step_time": 17.118911027908325
    },
    {
      "epoch": 1.860618591308594e-05,
      "grad_norm": 1.0153990983963013,
      "learning_rate": 1e-05,
      "loss": 0.0641,
      "step": 19510
    },
    {
      "epoch": 1.860618591308594e-05,
      "model_forward_time": 5.973203659057617,
      "step": 19510
    },
    {
      "dynamics_loss_avg": 0.06037454828619957,
      "epoch": 1.860618591308594e-05,
      "step": 19510
    },
    {
      "action_loss_avg": 0.007843414042145013,
      "epoch": 1.860618591308594e-05,
      "step": 19510
    },
    {
      "epoch": 1.860618591308594e-05,
      "step": 19510,
      "training_step_time": 17.439150094985962
    },
    {
      "epoch": 1.8607139587402343e-05,
      "model_forward_time": 5.590712308883667,
      "step": 19511
    },
    {
      "epoch": 1.8607139587402343e-05,
      "step": 19511,
      "training_step_time": 17.03170084953308
    },
    {
      "epoch": 1.860809326171875e-05,
      "model_forward_time": 5.585521459579468,
      "step": 19512
    },
    {
      "epoch": 1.860809326171875e-05,
      "step": 19512,
      "training_step_time": 16.994304656982422
    },
    {
      "epoch": 1.8609046936035158e-05,
      "model_forward_time": 5.5901734828948975,
      "step": 19513
    },
    {
      "epoch": 1.8609046936035158e-05,
      "step": 19513,
      "training_step_time": 17.048539400100708
    },
    {
      "epoch": 1.8610000610351562e-05,
      "model_forward_time": 5.604839086532593,
      "step": 19514
    },
    {
      "epoch": 1.8610000610351562e-05,
      "step": 19514,
      "training_step_time": 17.07984709739685
    },
    {
      "epoch": 1.861095428466797e-05,
      "model_forward_time": 5.956087112426758,
      "step": 19515
    },
    {
      "epoch": 1.861095428466797e-05,
      "step": 19515,
      "training_step_time": 17.441014766693115
    },
    {
      "epoch": 1.8611907958984373e-05,
      "model_forward_time": 5.60587739944458,
      "step": 19516
    },
    {
      "epoch": 1.8611907958984373e-05,
      "step": 19516,
      "training_step_time": 17.082128524780273
    },
    {
      "epoch": 1.861286163330078e-05,
      "model_forward_time": 5.600754499435425,
      "step": 19517
    },
    {
      "epoch": 1.861286163330078e-05,
      "step": 19517,
      "training_step_time": 17.075923442840576
    },
    {
      "epoch": 1.8613815307617188e-05,
      "model_forward_time": 5.612107992172241,
      "step": 19518
    },
    {
      "epoch": 1.8613815307617188e-05,
      "step": 19518,
      "training_step_time": 17.072449445724487
    },
    {
      "epoch": 1.8614768981933592e-05,
      "model_forward_time": 5.601732015609741,
      "step": 19519
    },
    {
      "epoch": 1.8614768981933592e-05,
      "step": 19519,
      "training_step_time": 17.057944774627686
    },
    {
      "epoch": 1.861572265625e-05,
      "grad_norm": 0.6844568848609924,
      "learning_rate": 1e-05,
      "loss": 0.0564,
      "step": 19520
    },
    {
      "epoch": 1.861572265625e-05,
      "model_forward_time": 5.603205680847168,
      "step": 19520
    },
    {
      "dynamics_loss_avg": 0.044725185818970205,
      "epoch": 1.861572265625e-05,
      "step": 19520
    },
    {
      "action_loss_avg": 0.0037645438220351934,
      "epoch": 1.861572265625e-05,
      "step": 19520
    },
    {
      "epoch": 1.861572265625e-05,
      "step": 19520,
      "training_step_time": 17.464651823043823
    },
    {
      "epoch": 1.8616676330566407e-05,
      "model_forward_time": 5.6028876304626465,
      "step": 19521
    },
    {
      "epoch": 1.8616676330566407e-05,
      "step": 19521,
      "training_step_time": 17.074742794036865
    },
    {
      "epoch": 1.861763000488281e-05,
      "model_forward_time": 5.596674203872681,
      "step": 19522
    },
    {
      "epoch": 1.861763000488281e-05,
      "step": 19522,
      "training_step_time": 17.046640396118164
    },
    {
      "epoch": 1.861858367919922e-05,
      "model_forward_time": 5.597029447555542,
      "step": 19523
    },
    {
      "epoch": 1.861858367919922e-05,
      "step": 19523,
      "training_step_time": 17.02289128303528
    },
    {
      "epoch": 1.8619537353515626e-05,
      "model_forward_time": 5.581350803375244,
      "step": 19524
    },
    {
      "epoch": 1.8619537353515626e-05,
      "step": 19524,
      "training_step_time": 16.987341165542603
    },
    {
      "epoch": 1.862049102783203e-05,
      "model_forward_time": 5.593786001205444,
      "step": 19525
    },
    {
      "epoch": 1.862049102783203e-05,
      "step": 19525,
      "training_step_time": 17.03900647163391
    },
    {
      "epoch": 1.8621444702148437e-05,
      "model_forward_time": 5.9494099617004395,
      "step": 19526
    },
    {
      "epoch": 1.8621444702148437e-05,
      "step": 19526,
      "training_step_time": 17.423978805541992
    },
    {
      "epoch": 1.8622398376464845e-05,
      "model_forward_time": 5.6089186668396,
      "step": 19527
    },
    {
      "epoch": 1.8622398376464845e-05,
      "step": 19527,
      "training_step_time": 17.097915649414062
    },
    {
      "epoch": 1.862335205078125e-05,
      "model_forward_time": 5.609848976135254,
      "step": 19528
    },
    {
      "epoch": 1.862335205078125e-05,
      "step": 19528,
      "training_step_time": 17.083231210708618
    },
    {
      "epoch": 1.8624305725097656e-05,
      "model_forward_time": 5.5933942794799805,
      "step": 19529
    },
    {
      "epoch": 1.8624305725097656e-05,
      "step": 19529,
      "training_step_time": 17.047924518585205
    },
    {
      "epoch": 1.8625259399414064e-05,
      "grad_norm": 0.6746785640716553,
      "learning_rate": 1e-05,
      "loss": 0.0745,
      "step": 19530
    },
    {
      "epoch": 1.8625259399414064e-05,
      "model_forward_time": 5.5923871994018555,
      "step": 19530
    },
    {
      "dynamics_loss_avg": 0.07038977667689324,
      "epoch": 1.8625259399414064e-05,
      "step": 19530
    },
    {
      "action_loss_avg": 0.007258723792620003,
      "epoch": 1.8625259399414064e-05,
      "step": 19530
    },
    {
      "epoch": 1.8625259399414064e-05,
      "step": 19530,
      "training_step_time": 17.035281658172607
    },
    {
      "epoch": 1.8626213073730468e-05,
      "model_forward_time": 5.970798015594482,
      "step": 19531
    },
    {
      "epoch": 1.8626213073730468e-05,
      "step": 19531,
      "training_step_time": 17.417428493499756
    },
    {
      "epoch": 1.8627166748046875e-05,
      "model_forward_time": 5.6168646812438965,
      "step": 19532
    },
    {
      "epoch": 1.8627166748046875e-05,
      "step": 19532,
      "training_step_time": 17.111095190048218
    },
    {
      "epoch": 1.8628120422363282e-05,
      "model_forward_time": 5.6086156368255615,
      "step": 19533
    },
    {
      "epoch": 1.8628120422363282e-05,
      "step": 19533,
      "training_step_time": 17.108110427856445
    },
    {
      "epoch": 1.8629074096679686e-05,
      "model_forward_time": 5.62163782119751,
      "step": 19534
    },
    {
      "epoch": 1.8629074096679686e-05,
      "step": 19534,
      "training_step_time": 17.11087942123413
    },
    {
      "epoch": 1.8630027770996094e-05,
      "model_forward_time": 5.600536346435547,
      "step": 19535
    },
    {
      "epoch": 1.8630027770996094e-05,
      "step": 19535,
      "training_step_time": 17.018733263015747
    },
    {
      "epoch": 1.86309814453125e-05,
      "model_forward_time": 5.591920375823975,
      "step": 19536
    },
    {
      "epoch": 1.86309814453125e-05,
      "step": 19536,
      "training_step_time": 17.000346183776855
    },
    {
      "epoch": 1.8631935119628905e-05,
      "model_forward_time": 5.921227693557739,
      "step": 19537
    },
    {
      "epoch": 1.8631935119628905e-05,
      "step": 19537,
      "training_step_time": 17.372572898864746
    },
    {
      "epoch": 1.8632888793945313e-05,
      "model_forward_time": 5.582146167755127,
      "step": 19538
    },
    {
      "epoch": 1.8632888793945313e-05,
      "step": 19538,
      "training_step_time": 17.0111985206604
    },
    {
      "epoch": 1.863384246826172e-05,
      "model_forward_time": 5.576029300689697,
      "step": 19539
    },
    {
      "epoch": 1.863384246826172e-05,
      "step": 19539,
      "training_step_time": 17.0030734539032
    },
    {
      "epoch": 1.8634796142578124e-05,
      "grad_norm": 0.7744590640068054,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 19540
    },
    {
      "epoch": 1.8634796142578124e-05,
      "model_forward_time": 5.58527684211731,
      "step": 19540
    },
    {
      "dynamics_loss_avg": 0.058506864681839944,
      "epoch": 1.8634796142578124e-05,
      "step": 19540
    },
    {
      "action_loss_avg": 0.006238724151626229,
      "epoch": 1.8634796142578124e-05,
      "step": 19540
    },
    {
      "epoch": 1.8634796142578124e-05,
      "step": 19540,
      "training_step_time": 17.03658413887024
    },
    {
      "epoch": 1.863574981689453e-05,
      "model_forward_time": 5.5900421142578125,
      "step": 19541
    },
    {
      "epoch": 1.863574981689453e-05,
      "step": 19541,
      "training_step_time": 17.022627592086792
    },
    {
      "epoch": 1.863670349121094e-05,
      "model_forward_time": 5.977936744689941,
      "step": 19542
    },
    {
      "epoch": 1.863670349121094e-05,
      "step": 19542,
      "training_step_time": 17.42440438270569
    },
    {
      "epoch": 1.8637657165527343e-05,
      "model_forward_time": 5.600999116897583,
      "step": 19543
    },
    {
      "epoch": 1.8637657165527343e-05,
      "step": 19543,
      "training_step_time": 17.117210626602173
    },
    {
      "epoch": 1.863861083984375e-05,
      "model_forward_time": 5.623805999755859,
      "step": 19544
    },
    {
      "epoch": 1.863861083984375e-05,
      "step": 19544,
      "training_step_time": 17.121281623840332
    },
    {
      "epoch": 1.8639564514160158e-05,
      "model_forward_time": 5.6303722858428955,
      "step": 19545
    },
    {
      "epoch": 1.8639564514160158e-05,
      "step": 19545,
      "training_step_time": 17.138195037841797
    },
    {
      "epoch": 1.8640518188476562e-05,
      "model_forward_time": 5.6412975788116455,
      "step": 19546
    },
    {
      "epoch": 1.8640518188476562e-05,
      "step": 19546,
      "training_step_time": 17.13871455192566
    },
    {
      "epoch": 1.864147186279297e-05,
      "model_forward_time": 5.600379705429077,
      "step": 19547
    },
    {
      "epoch": 1.864147186279297e-05,
      "step": 19547,
      "training_step_time": 17.05256485939026
    },
    {
      "epoch": 1.8642425537109373e-05,
      "model_forward_time": 5.93656849861145,
      "step": 19548
    },
    {
      "epoch": 1.8642425537109373e-05,
      "step": 19548,
      "training_step_time": 17.37356734275818
    },
    {
      "epoch": 1.864337921142578e-05,
      "model_forward_time": 5.588202714920044,
      "step": 19549
    },
    {
      "epoch": 1.864337921142578e-05,
      "step": 19549,
      "training_step_time": 17.033096075057983
    },
    {
      "epoch": 1.8644332885742188e-05,
      "grad_norm": 0.26788678765296936,
      "learning_rate": 1e-05,
      "loss": 0.0573,
      "step": 19550
    },
    {
      "epoch": 1.8644332885742188e-05,
      "model_forward_time": 5.591999053955078,
      "step": 19550
    },
    {
      "dynamics_loss_avg": 0.05855425074696541,
      "epoch": 1.8644332885742188e-05,
      "step": 19550
    },
    {
      "action_loss_avg": 0.00659585886169225,
      "epoch": 1.8644332885742188e-05,
      "step": 19550
    },
    {
      "epoch": 1.8644332885742188e-05,
      "step": 19550,
      "training_step_time": 17.01790738105774
    },
    {
      "epoch": 1.8645286560058592e-05,
      "model_forward_time": 5.589272975921631,
      "step": 19551
    },
    {
      "epoch": 1.8645286560058592e-05,
      "step": 19551,
      "training_step_time": 17.0492525100708
    },
    {
      "epoch": 1.8646240234375e-05,
      "model_forward_time": 5.6074934005737305,
      "step": 19552
    },
    {
      "epoch": 1.8646240234375e-05,
      "step": 19552,
      "training_step_time": 17.09159207344055
    },
    {
      "epoch": 1.8647193908691407e-05,
      "model_forward_time": 5.99418044090271,
      "step": 19553
    },
    {
      "epoch": 1.8647193908691407e-05,
      "step": 19553,
      "training_step_time": 17.500028133392334
    },
    {
      "epoch": 1.864814758300781e-05,
      "model_forward_time": 5.623795986175537,
      "step": 19554
    },
    {
      "epoch": 1.864814758300781e-05,
      "step": 19554,
      "training_step_time": 17.13096022605896
    },
    {
      "epoch": 1.864910125732422e-05,
      "model_forward_time": 5.622971296310425,
      "step": 19555
    },
    {
      "epoch": 1.864910125732422e-05,
      "step": 19555,
      "training_step_time": 17.12120509147644
    },
    {
      "epoch": 1.8650054931640626e-05,
      "model_forward_time": 5.625220060348511,
      "step": 19556
    },
    {
      "epoch": 1.8650054931640626e-05,
      "step": 19556,
      "training_step_time": 17.119791269302368
    },
    {
      "epoch": 1.865100860595703e-05,
      "model_forward_time": 5.62734842300415,
      "step": 19557
    },
    {
      "epoch": 1.865100860595703e-05,
      "step": 19557,
      "training_step_time": 17.107818841934204
    },
    {
      "epoch": 1.8651962280273437e-05,
      "model_forward_time": 5.61566686630249,
      "step": 19558
    },
    {
      "epoch": 1.8651962280273437e-05,
      "step": 19558,
      "training_step_time": 17.102243900299072
    },
    {
      "epoch": 1.8652915954589845e-05,
      "model_forward_time": 5.962320566177368,
      "step": 19559
    },
    {
      "epoch": 1.8652915954589845e-05,
      "step": 19559,
      "training_step_time": 17.43800377845764
    },
    {
      "epoch": 1.865386962890625e-05,
      "grad_norm": 0.14694221317768097,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 19560
    },
    {
      "epoch": 1.865386962890625e-05,
      "model_forward_time": 5.597384929656982,
      "step": 19560
    },
    {
      "dynamics_loss_avg": 0.05713026048615575,
      "epoch": 1.865386962890625e-05,
      "step": 19560
    },
    {
      "action_loss_avg": 0.005984221317339689,
      "epoch": 1.865386962890625e-05,
      "step": 19560
    },
    {
      "epoch": 1.865386962890625e-05,
      "step": 19560,
      "training_step_time": 17.061134815216064
    },
    {
      "epoch": 1.8654823303222656e-05,
      "model_forward_time": 5.597370386123657,
      "step": 19561
    },
    {
      "epoch": 1.8654823303222656e-05,
      "step": 19561,
      "training_step_time": 17.05172300338745
    },
    {
      "epoch": 1.8655776977539064e-05,
      "model_forward_time": 5.601497650146484,
      "step": 19562
    },
    {
      "epoch": 1.8655776977539064e-05,
      "step": 19562,
      "training_step_time": 17.048712968826294
    },
    {
      "epoch": 1.8656730651855468e-05,
      "model_forward_time": 5.596263647079468,
      "step": 19563
    },
    {
      "epoch": 1.8656730651855468e-05,
      "step": 19563,
      "training_step_time": 17.037294149398804
    },
    {
      "epoch": 1.8657684326171875e-05,
      "model_forward_time": 5.982037305831909,
      "step": 19564
    },
    {
      "epoch": 1.8657684326171875e-05,
      "step": 19564,
      "training_step_time": 17.444350957870483
    },
    {
      "epoch": 1.8658638000488282e-05,
      "model_forward_time": 5.595341205596924,
      "step": 19565
    },
    {
      "epoch": 1.8658638000488282e-05,
      "step": 19565,
      "training_step_time": 17.055773735046387
    },
    {
      "epoch": 1.8659591674804686e-05,
      "model_forward_time": 5.594202518463135,
      "step": 19566
    },
    {
      "epoch": 1.8659591674804686e-05,
      "step": 19566,
      "training_step_time": 17.063745260238647
    },
    {
      "epoch": 1.8660545349121094e-05,
      "model_forward_time": 5.6110594272613525,
      "step": 19567
    },
    {
      "epoch": 1.8660545349121094e-05,
      "step": 19567,
      "training_step_time": 17.11372685432434
    },
    {
      "epoch": 1.86614990234375e-05,
      "model_forward_time": 5.633632659912109,
      "step": 19568
    },
    {
      "epoch": 1.86614990234375e-05,
      "step": 19568,
      "training_step_time": 17.131595849990845
    },
    {
      "epoch": 1.8662452697753905e-05,
      "model_forward_time": 5.9857659339904785,
      "step": 19569
    },
    {
      "epoch": 1.8662452697753905e-05,
      "step": 19569,
      "training_step_time": 17.423532962799072
    },
    {
      "epoch": 1.8663406372070313e-05,
      "grad_norm": 0.674785852432251,
      "learning_rate": 1e-05,
      "loss": 0.0581,
      "step": 19570
    },
    {
      "epoch": 1.8663406372070313e-05,
      "model_forward_time": 5.594651222229004,
      "step": 19570
    },
    {
      "dynamics_loss_avg": 0.05504291485995054,
      "epoch": 1.8663406372070313e-05,
      "step": 19570
    },
    {
      "action_loss_avg": 0.0064625863684341315,
      "epoch": 1.8663406372070313e-05,
      "step": 19570
    },
    {
      "epoch": 1.8663406372070313e-05,
      "step": 19570,
      "training_step_time": 17.041603326797485
    },
    {
      "epoch": 1.866436004638672e-05,
      "model_forward_time": 5.5939929485321045,
      "step": 19571
    },
    {
      "epoch": 1.866436004638672e-05,
      "step": 19571,
      "training_step_time": 17.05224299430847
    },
    {
      "epoch": 1.8665313720703124e-05,
      "model_forward_time": 5.606825351715088,
      "step": 19572
    },
    {
      "epoch": 1.8665313720703124e-05,
      "step": 19572,
      "training_step_time": 17.106419801712036
    },
    {
      "epoch": 1.866626739501953e-05,
      "model_forward_time": 5.613845109939575,
      "step": 19573
    },
    {
      "epoch": 1.866626739501953e-05,
      "step": 19573,
      "training_step_time": 17.12229561805725
    },
    {
      "epoch": 1.866722106933594e-05,
      "model_forward_time": 5.611888647079468,
      "step": 19574
    },
    {
      "epoch": 1.866722106933594e-05,
      "step": 19574,
      "training_step_time": 17.095511198043823
    },
    {
      "epoch": 1.8668174743652343e-05,
      "model_forward_time": 5.985288143157959,
      "step": 19575
    },
    {
      "epoch": 1.8668174743652343e-05,
      "step": 19575,
      "training_step_time": 17.4760582447052
    },
    {
      "epoch": 1.866912841796875e-05,
      "model_forward_time": 5.644299030303955,
      "step": 19576
    },
    {
      "epoch": 1.866912841796875e-05,
      "step": 19576,
      "training_step_time": 17.114548206329346
    },
    {
      "epoch": 1.8670082092285158e-05,
      "model_forward_time": 5.588691473007202,
      "step": 19577
    },
    {
      "epoch": 1.8670082092285158e-05,
      "step": 19577,
      "training_step_time": 17.00290322303772
    },
    {
      "epoch": 1.8671035766601562e-05,
      "model_forward_time": 5.577104568481445,
      "step": 19578
    },
    {
      "epoch": 1.8671035766601562e-05,
      "step": 19578,
      "training_step_time": 17.013418674468994
    },
    {
      "epoch": 1.867198944091797e-05,
      "model_forward_time": 5.587892532348633,
      "step": 19579
    },
    {
      "epoch": 1.867198944091797e-05,
      "step": 19579,
      "training_step_time": 17.02359414100647
    },
    {
      "epoch": 1.8672943115234373e-05,
      "grad_norm": 0.7693548798561096,
      "learning_rate": 1e-05,
      "loss": 0.0757,
      "step": 19580
    },
    {
      "epoch": 1.8672943115234373e-05,
      "model_forward_time": 5.579159736633301,
      "step": 19580
    },
    {
      "dynamics_loss_avg": 0.06951058395206929,
      "epoch": 1.8672943115234373e-05,
      "step": 19580
    },
    {
      "action_loss_avg": 0.006103807711042464,
      "epoch": 1.8672943115234373e-05,
      "step": 19580
    },
    {
      "epoch": 1.8672943115234373e-05,
      "step": 19580,
      "training_step_time": 17.367979288101196
    },
    {
      "epoch": 1.867389678955078e-05,
      "model_forward_time": 5.5775251388549805,
      "step": 19581
    },
    {
      "epoch": 1.867389678955078e-05,
      "step": 19581,
      "training_step_time": 17.013967990875244
    },
    {
      "epoch": 1.8674850463867188e-05,
      "model_forward_time": 4.2527055740356445,
      "step": 19582
    },
    {
      "epoch": 1.8674850463867188e-05,
      "step": 19582,
      "training_step_time": 12.745574712753296
    },
    {
      "epoch": 1.8675804138183592e-05,
      "model_forward_time": 5.590315818786621,
      "step": 19583
    },
    {
      "epoch": 1.8675804138183592e-05,
      "step": 19583,
      "training_step_time": 17.054124116897583
    },
    {
      "epoch": 1.86767578125e-05,
      "model_forward_time": 5.597217798233032,
      "step": 19584
    },
    {
      "epoch": 1.86767578125e-05,
      "step": 19584,
      "training_step_time": 17.093844652175903
    },
    {
      "epoch": 1.8677711486816407e-05,
      "model_forward_time": 5.62530517578125,
      "step": 19585
    },
    {
      "epoch": 1.8677711486816407e-05,
      "step": 19585,
      "training_step_time": 17.13893413543701
    },
    {
      "epoch": 1.867866516113281e-05,
      "model_forward_time": 5.993783950805664,
      "step": 19586
    },
    {
      "epoch": 1.867866516113281e-05,
      "step": 19586,
      "training_step_time": 17.483977556228638
    },
    {
      "epoch": 1.867961883544922e-05,
      "model_forward_time": 5.597110271453857,
      "step": 19587
    },
    {
      "epoch": 1.867961883544922e-05,
      "step": 19587,
      "training_step_time": 17.035113096237183
    },
    {
      "epoch": 1.8680572509765626e-05,
      "model_forward_time": 5.590328931808472,
      "step": 19588
    },
    {
      "epoch": 1.8680572509765626e-05,
      "step": 19588,
      "training_step_time": 17.009342670440674
    },
    {
      "epoch": 1.868152618408203e-05,
      "model_forward_time": 5.576263189315796,
      "step": 19589
    },
    {
      "epoch": 1.868152618408203e-05,
      "step": 19589,
      "training_step_time": 16.997544527053833
    },
    {
      "epoch": 1.8682479858398437e-05,
      "grad_norm": 0.6322414875030518,
      "learning_rate": 1e-05,
      "loss": 0.0644,
      "step": 19590
    },
    {
      "epoch": 1.8682479858398437e-05,
      "model_forward_time": 5.578941822052002,
      "step": 19590
    },
    {
      "dynamics_loss_avg": 0.05661038029938936,
      "epoch": 1.8682479858398437e-05,
      "step": 19590
    },
    {
      "action_loss_avg": 0.007458959228824824,
      "epoch": 1.8682479858398437e-05,
      "step": 19590
    },
    {
      "epoch": 1.8682479858398437e-05,
      "step": 19590,
      "training_step_time": 17.010966300964355
    },
    {
      "epoch": 1.8683433532714845e-05,
      "model_forward_time": 5.5812458992004395,
      "step": 19591
    },
    {
      "epoch": 1.8683433532714845e-05,
      "step": 19591,
      "training_step_time": 17.394461393356323
    },
    {
      "epoch": 1.868438720703125e-05,
      "model_forward_time": 5.595474720001221,
      "step": 19592
    },
    {
      "epoch": 1.868438720703125e-05,
      "step": 19592,
      "training_step_time": 17.075485706329346
    },
    {
      "epoch": 1.8685340881347656e-05,
      "model_forward_time": 5.597800254821777,
      "step": 19593
    },
    {
      "epoch": 1.8685340881347656e-05,
      "step": 19593,
      "training_step_time": 17.065301179885864
    },
    {
      "epoch": 1.8686294555664064e-05,
      "model_forward_time": 5.595317125320435,
      "step": 19594
    },
    {
      "epoch": 1.8686294555664064e-05,
      "step": 19594,
      "training_step_time": 17.08200168609619
    },
    {
      "epoch": 1.8687248229980468e-05,
      "model_forward_time": 5.599009990692139,
      "step": 19595
    },
    {
      "epoch": 1.8687248229980468e-05,
      "step": 19595,
      "training_step_time": 17.06400203704834
    },
    {
      "epoch": 1.8688201904296875e-05,
      "model_forward_time": 5.593146562576294,
      "step": 19596
    },
    {
      "epoch": 1.8688201904296875e-05,
      "step": 19596,
      "training_step_time": 17.066248893737793
    },
    {
      "epoch": 1.8689155578613282e-05,
      "model_forward_time": 5.973951578140259,
      "step": 19597
    },
    {
      "epoch": 1.8689155578613282e-05,
      "step": 19597,
      "training_step_time": 17.435282468795776
    },
    {
      "epoch": 1.8690109252929687e-05,
      "model_forward_time": 5.600157976150513,
      "step": 19598
    },
    {
      "epoch": 1.8690109252929687e-05,
      "step": 19598,
      "training_step_time": 17.066144943237305
    },
    {
      "epoch": 1.8691062927246094e-05,
      "model_forward_time": 5.604796886444092,
      "step": 19599
    },
    {
      "epoch": 1.8691062927246094e-05,
      "step": 19599,
      "training_step_time": 17.109516620635986
    },
    {
      "epoch": 1.86920166015625e-05,
      "grad_norm": 0.6492798328399658,
      "learning_rate": 1e-05,
      "loss": 0.0555,
      "step": 19600
    },
    {
      "epoch": 1.86920166015625e-05,
      "model_forward_time": 5.612232446670532,
      "step": 19600
    },
    {
      "dynamics_loss_avg": 0.05472494270652532,
      "epoch": 1.86920166015625e-05,
      "step": 19600
    },
    {
      "action_loss_avg": 0.007612813147716224,
      "epoch": 1.86920166015625e-05,
      "step": 19600
    },
    {
      "epoch": 1.86920166015625e-05,
      "step": 19600,
      "training_step_time": 17.12399983406067
    },
    {
      "epoch": 1.8692970275878905e-05,
      "model_forward_time": 5.6215057373046875,
      "step": 19601
    },
    {
      "epoch": 1.8692970275878905e-05,
      "step": 19601,
      "training_step_time": 17.119937419891357
    },
    {
      "epoch": 1.8693923950195313e-05,
      "model_forward_time": 5.971877813339233,
      "step": 19602
    },
    {
      "epoch": 1.8693923950195313e-05,
      "step": 19602,
      "training_step_time": 17.46617817878723
    },
    {
      "epoch": 1.869487762451172e-05,
      "model_forward_time": 5.611673831939697,
      "step": 19603
    },
    {
      "epoch": 1.869487762451172e-05,
      "step": 19603,
      "training_step_time": 17.10051155090332
    },
    {
      "epoch": 1.8695831298828124e-05,
      "model_forward_time": 5.606762647628784,
      "step": 19604
    },
    {
      "epoch": 1.8695831298828124e-05,
      "step": 19604,
      "training_step_time": 17.072832584381104
    },
    {
      "epoch": 1.869678497314453e-05,
      "model_forward_time": 5.591927766799927,
      "step": 19605
    },
    {
      "epoch": 1.869678497314453e-05,
      "step": 19605,
      "training_step_time": 17.03788471221924
    },
    {
      "epoch": 1.869773864746094e-05,
      "model_forward_time": 5.578913688659668,
      "step": 19606
    },
    {
      "epoch": 1.869773864746094e-05,
      "step": 19606,
      "training_step_time": 17.001961946487427
    },
    {
      "epoch": 1.8698692321777343e-05,
      "model_forward_time": 5.568248748779297,
      "step": 19607
    },
    {
      "epoch": 1.8698692321777343e-05,
      "step": 19607,
      "training_step_time": 16.99527668952942
    },
    {
      "epoch": 1.869964599609375e-05,
      "model_forward_time": 5.954099178314209,
      "step": 19608
    },
    {
      "epoch": 1.869964599609375e-05,
      "step": 19608,
      "training_step_time": 17.363914012908936
    },
    {
      "epoch": 1.8700599670410158e-05,
      "model_forward_time": 5.584577560424805,
      "step": 19609
    },
    {
      "epoch": 1.8700599670410158e-05,
      "step": 19609,
      "training_step_time": 17.021129846572876
    },
    {
      "epoch": 1.8701553344726562e-05,
      "grad_norm": 0.7400174736976624,
      "learning_rate": 1e-05,
      "loss": 0.075,
      "step": 19610
    },
    {
      "epoch": 1.8701553344726562e-05,
      "model_forward_time": 5.575048446655273,
      "step": 19610
    },
    {
      "dynamics_loss_avg": 0.0635134655982256,
      "epoch": 1.8701553344726562e-05,
      "step": 19610
    },
    {
      "action_loss_avg": 0.006598041066899896,
      "epoch": 1.8701553344726562e-05,
      "step": 19610
    },
    {
      "epoch": 1.8701553344726562e-05,
      "step": 19610,
      "training_step_time": 16.99122405052185
    },
    {
      "epoch": 1.870250701904297e-05,
      "model_forward_time": 5.584858655929565,
      "step": 19611
    },
    {
      "epoch": 1.870250701904297e-05,
      "step": 19611,
      "training_step_time": 17.003974437713623
    },
    {
      "epoch": 1.8703460693359373e-05,
      "model_forward_time": 5.5882463455200195,
      "step": 19612
    },
    {
      "epoch": 1.8703460693359373e-05,
      "step": 19612,
      "training_step_time": 17.050878047943115
    },
    {
      "epoch": 1.870441436767578e-05,
      "model_forward_time": 5.607597351074219,
      "step": 19613
    },
    {
      "epoch": 1.870441436767578e-05,
      "step": 19613,
      "training_step_time": 17.441420316696167
    },
    {
      "epoch": 1.8705368041992188e-05,
      "model_forward_time": 5.624626874923706,
      "step": 19614
    },
    {
      "epoch": 1.8705368041992188e-05,
      "step": 19614,
      "training_step_time": 17.124309301376343
    },
    {
      "epoch": 1.8706321716308592e-05,
      "model_forward_time": 5.61968207359314,
      "step": 19615
    },
    {
      "epoch": 1.8706321716308592e-05,
      "step": 19615,
      "training_step_time": 17.08539319038391
    },
    {
      "epoch": 1.8707275390625e-05,
      "model_forward_time": 5.591132879257202,
      "step": 19616
    },
    {
      "epoch": 1.8707275390625e-05,
      "step": 19616,
      "training_step_time": 17.003885984420776
    },
    {
      "epoch": 1.8708229064941407e-05,
      "model_forward_time": 5.572094440460205,
      "step": 19617
    },
    {
      "epoch": 1.8708229064941407e-05,
      "step": 19617,
      "training_step_time": 17.012446880340576
    },
    {
      "epoch": 1.870918273925781e-05,
      "model_forward_time": 5.57379412651062,
      "step": 19618
    },
    {
      "epoch": 1.870918273925781e-05,
      "step": 19618,
      "training_step_time": 16.987573862075806
    },
    {
      "epoch": 1.871013641357422e-05,
      "model_forward_time": 5.94291615486145,
      "step": 19619
    },
    {
      "epoch": 1.871013641357422e-05,
      "step": 19619,
      "training_step_time": 17.35353112220764
    },
    {
      "epoch": 1.8711090087890626e-05,
      "grad_norm": 0.4543435573577881,
      "learning_rate": 1e-05,
      "loss": 0.0476,
      "step": 19620
    },
    {
      "epoch": 1.8711090087890626e-05,
      "model_forward_time": 5.583690881729126,
      "step": 19620
    },
    {
      "dynamics_loss_avg": 0.04561823708936572,
      "epoch": 1.8711090087890626e-05,
      "step": 19620
    },
    {
      "action_loss_avg": 0.005159281485248357,
      "epoch": 1.8711090087890626e-05,
      "step": 19620
    },
    {
      "epoch": 1.8711090087890626e-05,
      "step": 19620,
      "training_step_time": 17.03984522819519
    },
    {
      "epoch": 1.871204376220703e-05,
      "model_forward_time": 5.575226068496704,
      "step": 19621
    },
    {
      "epoch": 1.871204376220703e-05,
      "step": 19621,
      "training_step_time": 17.003040075302124
    },
    {
      "epoch": 1.8712997436523437e-05,
      "model_forward_time": 5.585415601730347,
      "step": 19622
    },
    {
      "epoch": 1.8712997436523437e-05,
      "step": 19622,
      "training_step_time": 17.015453100204468
    },
    {
      "epoch": 1.8713951110839845e-05,
      "model_forward_time": 5.585105895996094,
      "step": 19623
    },
    {
      "epoch": 1.8713951110839845e-05,
      "step": 19623,
      "training_step_time": 17.017509698867798
    },
    {
      "epoch": 1.871490478515625e-05,
      "model_forward_time": 5.598156929016113,
      "step": 19624
    },
    {
      "epoch": 1.871490478515625e-05,
      "step": 19624,
      "training_step_time": 17.40196919441223
    },
    {
      "epoch": 1.8715858459472656e-05,
      "model_forward_time": 5.5985400676727295,
      "step": 19625
    },
    {
      "epoch": 1.8715858459472656e-05,
      "step": 19625,
      "training_step_time": 17.069820642471313
    },
    {
      "epoch": 1.8716812133789064e-05,
      "model_forward_time": 5.601010084152222,
      "step": 19626
    },
    {
      "epoch": 1.8716812133789064e-05,
      "step": 19626,
      "training_step_time": 17.065945625305176
    },
    {
      "epoch": 1.8717765808105468e-05,
      "model_forward_time": 5.597103834152222,
      "step": 19627
    },
    {
      "epoch": 1.8717765808105468e-05,
      "step": 19627,
      "training_step_time": 17.052469491958618
    },
    {
      "epoch": 1.8718719482421875e-05,
      "model_forward_time": 5.594146966934204,
      "step": 19628
    },
    {
      "epoch": 1.8718719482421875e-05,
      "step": 19628,
      "training_step_time": 17.039385080337524
    },
    {
      "epoch": 1.8719673156738283e-05,
      "model_forward_time": 5.587929725646973,
      "step": 19629
    },
    {
      "epoch": 1.8719673156738283e-05,
      "step": 19629,
      "training_step_time": 17.011390686035156
    },
    {
      "epoch": 1.8720626831054687e-05,
      "grad_norm": 0.7864540815353394,
      "learning_rate": 1e-05,
      "loss": 0.0843,
      "step": 19630
    },
    {
      "epoch": 1.8720626831054687e-05,
      "model_forward_time": 5.959860801696777,
      "step": 19630
    },
    {
      "dynamics_loss_avg": 0.07076169215142727,
      "epoch": 1.8720626831054687e-05,
      "step": 19630
    },
    {
      "action_loss_avg": 0.007509033638052642,
      "epoch": 1.8720626831054687e-05,
      "step": 19630
    },
    {
      "epoch": 1.8720626831054687e-05,
      "step": 19630,
      "training_step_time": 17.398070573806763
    },
    {
      "epoch": 1.8721580505371094e-05,
      "model_forward_time": 5.59188437461853,
      "step": 19631
    },
    {
      "epoch": 1.8721580505371094e-05,
      "step": 19631,
      "training_step_time": 17.044681310653687
    },
    {
      "epoch": 1.87225341796875e-05,
      "model_forward_time": 5.602737665176392,
      "step": 19632
    },
    {
      "epoch": 1.87225341796875e-05,
      "step": 19632,
      "training_step_time": 17.064873456954956
    },
    {
      "epoch": 1.8723487854003905e-05,
      "model_forward_time": 5.603663444519043,
      "step": 19633
    },
    {
      "epoch": 1.8723487854003905e-05,
      "step": 19633,
      "training_step_time": 17.05955743789673
    },
    {
      "epoch": 1.8724441528320313e-05,
      "model_forward_time": 5.59889817237854,
      "step": 19634
    },
    {
      "epoch": 1.8724441528320313e-05,
      "step": 19634,
      "training_step_time": 17.072633028030396
    },
    {
      "epoch": 1.872539520263672e-05,
      "model_forward_time": 5.956697702407837,
      "step": 19635
    },
    {
      "epoch": 1.872539520263672e-05,
      "step": 19635,
      "training_step_time": 17.44115924835205
    },
    {
      "epoch": 1.8726348876953124e-05,
      "model_forward_time": 5.607535123825073,
      "step": 19636
    },
    {
      "epoch": 1.8726348876953124e-05,
      "step": 19636,
      "training_step_time": 17.09777522087097
    },
    {
      "epoch": 1.872730255126953e-05,
      "model_forward_time": 5.606104135513306,
      "step": 19637
    },
    {
      "epoch": 1.872730255126953e-05,
      "step": 19637,
      "training_step_time": 17.08510684967041
    },
    {
      "epoch": 1.872825622558594e-05,
      "model_forward_time": 5.60614013671875,
      "step": 19638
    },
    {
      "epoch": 1.872825622558594e-05,
      "step": 19638,
      "training_step_time": 17.077622413635254
    },
    {
      "epoch": 1.8729209899902343e-05,
      "model_forward_time": 5.604536533355713,
      "step": 19639
    },
    {
      "epoch": 1.8729209899902343e-05,
      "step": 19639,
      "training_step_time": 17.08677339553833
    },
    {
      "epoch": 1.873016357421875e-05,
      "grad_norm": 0.930075466632843,
      "learning_rate": 1e-05,
      "loss": 0.0804,
      "step": 19640
    },
    {
      "epoch": 1.873016357421875e-05,
      "model_forward_time": 5.608604907989502,
      "step": 19640
    },
    {
      "dynamics_loss_avg": 0.0762038068845868,
      "epoch": 1.873016357421875e-05,
      "step": 19640
    },
    {
      "action_loss_avg": 0.008592436113394797,
      "epoch": 1.873016357421875e-05,
      "step": 19640
    },
    {
      "epoch": 1.873016357421875e-05,
      "step": 19640,
      "training_step_time": 17.063692092895508
    },
    {
      "epoch": 1.8731117248535155e-05,
      "model_forward_time": 6.005491733551025,
      "step": 19641
    },
    {
      "epoch": 1.8731117248535155e-05,
      "step": 19641,
      "training_step_time": 17.461220264434814
    },
    {
      "epoch": 1.8732070922851562e-05,
      "model_forward_time": 5.594542026519775,
      "step": 19642
    },
    {
      "epoch": 1.8732070922851562e-05,
      "step": 19642,
      "training_step_time": 17.05408501625061
    },
    {
      "epoch": 1.873302459716797e-05,
      "model_forward_time": 5.60683012008667,
      "step": 19643
    },
    {
      "epoch": 1.873302459716797e-05,
      "step": 19643,
      "training_step_time": 17.0752215385437
    },
    {
      "epoch": 1.8733978271484373e-05,
      "model_forward_time": 5.617132663726807,
      "step": 19644
    },
    {
      "epoch": 1.8733978271484373e-05,
      "step": 19644,
      "training_step_time": 17.120789051055908
    },
    {
      "epoch": 1.873493194580078e-05,
      "model_forward_time": 5.618723392486572,
      "step": 19645
    },
    {
      "epoch": 1.873493194580078e-05,
      "step": 19645,
      "training_step_time": 17.097732067108154
    },
    {
      "epoch": 1.8735885620117188e-05,
      "model_forward_time": 5.593578338623047,
      "step": 19646
    },
    {
      "epoch": 1.8735885620117188e-05,
      "step": 19646,
      "training_step_time": 17.389591455459595
    },
    {
      "epoch": 1.8736839294433592e-05,
      "model_forward_time": 5.5643556118011475,
      "step": 19647
    },
    {
      "epoch": 1.8736839294433592e-05,
      "step": 19647,
      "training_step_time": 16.998773097991943
    },
    {
      "epoch": 1.873779296875e-05,
      "model_forward_time": 5.566103935241699,
      "step": 19648
    },
    {
      "epoch": 1.873779296875e-05,
      "step": 19648,
      "training_step_time": 16.976070880889893
    },
    {
      "epoch": 1.8738746643066407e-05,
      "model_forward_time": 5.5825817584991455,
      "step": 19649
    },
    {
      "epoch": 1.8738746643066407e-05,
      "step": 19649,
      "training_step_time": 16.98877716064453
    },
    {
      "epoch": 1.873970031738281e-05,
      "grad_norm": 0.6409307718276978,
      "learning_rate": 1e-05,
      "loss": 0.0713,
      "step": 19650
    },
    {
      "epoch": 1.873970031738281e-05,
      "model_forward_time": 5.574359893798828,
      "step": 19650
    },
    {
      "dynamics_loss_avg": 0.05912658609449863,
      "epoch": 1.873970031738281e-05,
      "step": 19650
    },
    {
      "action_loss_avg": 0.007438250887207687,
      "epoch": 1.873970031738281e-05,
      "step": 19650
    },
    {
      "epoch": 1.873970031738281e-05,
      "step": 19650,
      "training_step_time": 16.99160146713257
    },
    {
      "epoch": 1.874065399169922e-05,
      "model_forward_time": 5.575401067733765,
      "step": 19651
    },
    {
      "epoch": 1.874065399169922e-05,
      "step": 19651,
      "training_step_time": 17.019011974334717
    },
    {
      "epoch": 1.8741607666015626e-05,
      "model_forward_time": 5.969166040420532,
      "step": 19652
    },
    {
      "epoch": 1.8741607666015626e-05,
      "step": 19652,
      "training_step_time": 17.38602638244629
    },
    {
      "epoch": 1.874256134033203e-05,
      "model_forward_time": 5.593141794204712,
      "step": 19653
    },
    {
      "epoch": 1.874256134033203e-05,
      "step": 19653,
      "training_step_time": 17.035263299942017
    },
    {
      "epoch": 1.8743515014648437e-05,
      "model_forward_time": 5.5968897342681885,
      "step": 19654
    },
    {
      "epoch": 1.8743515014648437e-05,
      "step": 19654,
      "training_step_time": 17.054410696029663
    },
    {
      "epoch": 1.8744468688964845e-05,
      "model_forward_time": 5.5965845584869385,
      "step": 19655
    },
    {
      "epoch": 1.8744468688964845e-05,
      "step": 19655,
      "training_step_time": 17.038012266159058
    },
    {
      "epoch": 1.874542236328125e-05,
      "model_forward_time": 5.592106342315674,
      "step": 19656
    },
    {
      "epoch": 1.874542236328125e-05,
      "step": 19656,
      "training_step_time": 17.0323166847229
    },
    {
      "epoch": 1.8746376037597656e-05,
      "model_forward_time": 5.593508243560791,
      "step": 19657
    },
    {
      "epoch": 1.8746376037597656e-05,
      "step": 19657,
      "training_step_time": 17.389805555343628
    },
    {
      "epoch": 1.8747329711914064e-05,
      "model_forward_time": 5.577653884887695,
      "step": 19658
    },
    {
      "epoch": 1.8747329711914064e-05,
      "step": 19658,
      "training_step_time": 17.00802755355835
    },
    {
      "epoch": 1.8748283386230468e-05,
      "model_forward_time": 5.593204975128174,
      "step": 19659
    },
    {
      "epoch": 1.8748283386230468e-05,
      "step": 19659,
      "training_step_time": 17.073840856552124
    },
    {
      "epoch": 1.8749237060546875e-05,
      "grad_norm": 0.4912360608577728,
      "learning_rate": 1e-05,
      "loss": 0.0497,
      "step": 19660
    },
    {
      "epoch": 1.8749237060546875e-05,
      "model_forward_time": 5.614401817321777,
      "step": 19660
    },
    {
      "dynamics_loss_avg": 0.04620526731014252,
      "epoch": 1.8749237060546875e-05,
      "step": 19660
    },
    {
      "action_loss_avg": 0.0060183258261531595,
      "epoch": 1.8749237060546875e-05,
      "step": 19660
    },
    {
      "epoch": 1.8749237060546875e-05,
      "step": 19660,
      "training_step_time": 17.129037857055664
    },
    {
      "epoch": 1.8750190734863283e-05,
      "model_forward_time": 5.621880531311035,
      "step": 19661
    },
    {
      "epoch": 1.8750190734863283e-05,
      "step": 19661,
      "training_step_time": 17.114238262176514
    },
    {
      "epoch": 1.8751144409179687e-05,
      "model_forward_time": 5.602667570114136,
      "step": 19662
    },
    {
      "epoch": 1.8751144409179687e-05,
      "step": 19662,
      "training_step_time": 17.027501106262207
    },
    {
      "epoch": 1.8752098083496094e-05,
      "model_forward_time": 5.9359047412872314,
      "step": 19663
    },
    {
      "epoch": 1.8752098083496094e-05,
      "step": 19663,
      "training_step_time": 17.36604142189026
    },
    {
      "epoch": 1.87530517578125e-05,
      "model_forward_time": 5.5801451206207275,
      "step": 19664
    },
    {
      "epoch": 1.87530517578125e-05,
      "step": 19664,
      "training_step_time": 16.995048761367798
    },
    {
      "epoch": 1.8754005432128905e-05,
      "model_forward_time": 5.582788944244385,
      "step": 19665
    },
    {
      "epoch": 1.8754005432128905e-05,
      "step": 19665,
      "training_step_time": 17.019096851348877
    },
    {
      "epoch": 1.8754959106445313e-05,
      "model_forward_time": 5.599541425704956,
      "step": 19666
    },
    {
      "epoch": 1.8754959106445313e-05,
      "step": 19666,
      "training_step_time": 17.062522888183594
    },
    {
      "epoch": 1.875591278076172e-05,
      "model_forward_time": 5.61008358001709,
      "step": 19667
    },
    {
      "epoch": 1.875591278076172e-05,
      "step": 19667,
      "training_step_time": 17.116628646850586
    },
    {
      "epoch": 1.8756866455078124e-05,
      "model_forward_time": 5.986501932144165,
      "step": 19668
    },
    {
      "epoch": 1.8756866455078124e-05,
      "step": 19668,
      "training_step_time": 17.47095561027527
    },
    {
      "epoch": 1.8757820129394532e-05,
      "model_forward_time": 5.594275236129761,
      "step": 19669
    },
    {
      "epoch": 1.8757820129394532e-05,
      "step": 19669,
      "training_step_time": 17.048540592193604
    },
    {
      "epoch": 1.875877380371094e-05,
      "grad_norm": 0.32411056756973267,
      "learning_rate": 1e-05,
      "loss": 0.0682,
      "step": 19670
    },
    {
      "epoch": 1.875877380371094e-05,
      "model_forward_time": 5.587817192077637,
      "step": 19670
    },
    {
      "dynamics_loss_avg": 0.0574025945737958,
      "epoch": 1.875877380371094e-05,
      "step": 19670
    },
    {
      "action_loss_avg": 0.0064404257689602675,
      "epoch": 1.875877380371094e-05,
      "step": 19670
    },
    {
      "epoch": 1.875877380371094e-05,
      "step": 19670,
      "training_step_time": 17.023654222488403
    },
    {
      "epoch": 1.8759727478027343e-05,
      "model_forward_time": 5.596152067184448,
      "step": 19671
    },
    {
      "epoch": 1.8759727478027343e-05,
      "step": 19671,
      "training_step_time": 17.049380779266357
    },
    {
      "epoch": 1.876068115234375e-05,
      "model_forward_time": 5.603709697723389,
      "step": 19672
    },
    {
      "epoch": 1.876068115234375e-05,
      "step": 19672,
      "training_step_time": 17.091398239135742
    },
    {
      "epoch": 1.8761634826660155e-05,
      "model_forward_time": 5.607993125915527,
      "step": 19673
    },
    {
      "epoch": 1.8761634826660155e-05,
      "step": 19673,
      "training_step_time": 17.087488412857056
    },
    {
      "epoch": 1.8762588500976562e-05,
      "model_forward_time": 5.996369123458862,
      "step": 19674
    },
    {
      "epoch": 1.8762588500976562e-05,
      "step": 19674,
      "training_step_time": 17.454375505447388
    },
    {
      "epoch": 1.876354217529297e-05,
      "model_forward_time": 5.608790159225464,
      "step": 19675
    },
    {
      "epoch": 1.876354217529297e-05,
      "step": 19675,
      "training_step_time": 17.090739965438843
    },
    {
      "epoch": 1.8764495849609373e-05,
      "model_forward_time": 5.608675241470337,
      "step": 19676
    },
    {
      "epoch": 1.8764495849609373e-05,
      "step": 19676,
      "training_step_time": 17.093834400177002
    },
    {
      "epoch": 1.876544952392578e-05,
      "model_forward_time": 5.610677480697632,
      "step": 19677
    },
    {
      "epoch": 1.876544952392578e-05,
      "step": 19677,
      "training_step_time": 17.10616421699524
    },
    {
      "epoch": 1.8766403198242188e-05,
      "model_forward_time": 5.626918315887451,
      "step": 19678
    },
    {
      "epoch": 1.8766403198242188e-05,
      "step": 19678,
      "training_step_time": 17.126965045928955
    },
    {
      "epoch": 1.8767356872558592e-05,
      "model_forward_time": 5.614237070083618,
      "step": 19679
    },
    {
      "epoch": 1.8767356872558592e-05,
      "step": 19679,
      "training_step_time": 17.483454704284668
    },
    {
      "epoch": 1.8768310546875e-05,
      "grad_norm": 0.808788537979126,
      "learning_rate": 1e-05,
      "loss": 0.0667,
      "step": 19680
    },
    {
      "epoch": 1.8768310546875e-05,
      "model_forward_time": 5.6148247718811035,
      "step": 19680
    },
    {
      "dynamics_loss_avg": 0.06682192720472813,
      "epoch": 1.8768310546875e-05,
      "step": 19680
    },
    {
      "action_loss_avg": 0.007569676032289863,
      "epoch": 1.8768310546875e-05,
      "step": 19680
    },
    {
      "epoch": 1.8768310546875e-05,
      "step": 19680,
      "training_step_time": 17.108745336532593
    },
    {
      "epoch": 1.8769264221191407e-05,
      "model_forward_time": 5.605837106704712,
      "step": 19681
    },
    {
      "epoch": 1.8769264221191407e-05,
      "step": 19681,
      "training_step_time": 17.088639974594116
    },
    {
      "epoch": 1.877021789550781e-05,
      "model_forward_time": 5.617837905883789,
      "step": 19682
    },
    {
      "epoch": 1.877021789550781e-05,
      "step": 19682,
      "training_step_time": 17.098983764648438
    },
    {
      "epoch": 1.877117156982422e-05,
      "model_forward_time": 5.603177070617676,
      "step": 19683
    },
    {
      "epoch": 1.877117156982422e-05,
      "step": 19683,
      "training_step_time": 17.047736167907715
    },
    {
      "epoch": 1.8772125244140626e-05,
      "model_forward_time": 5.601483345031738,
      "step": 19684
    },
    {
      "epoch": 1.8772125244140626e-05,
      "step": 19684,
      "training_step_time": 17.03151535987854
    },
    {
      "epoch": 1.877307891845703e-05,
      "model_forward_time": 5.957630395889282,
      "step": 19685
    },
    {
      "epoch": 1.877307891845703e-05,
      "step": 19685,
      "training_step_time": 17.379902839660645
    },
    {
      "epoch": 1.8774032592773437e-05,
      "model_forward_time": 5.589420795440674,
      "step": 19686
    },
    {
      "epoch": 1.8774032592773437e-05,
      "step": 19686,
      "training_step_time": 17.0509135723114
    },
    {
      "epoch": 1.8774986267089845e-05,
      "model_forward_time": 5.598432302474976,
      "step": 19687
    },
    {
      "epoch": 1.8774986267089845e-05,
      "step": 19687,
      "training_step_time": 17.05334782600403
    },
    {
      "epoch": 1.877593994140625e-05,
      "model_forward_time": 5.598924875259399,
      "step": 19688
    },
    {
      "epoch": 1.877593994140625e-05,
      "step": 19688,
      "training_step_time": 17.052706480026245
    },
    {
      "epoch": 1.8776893615722656e-05,
      "model_forward_time": 5.593887567520142,
      "step": 19689
    },
    {
      "epoch": 1.8776893615722656e-05,
      "step": 19689,
      "training_step_time": 17.04667019844055
    },
    {
      "epoch": 1.8777847290039064e-05,
      "grad_norm": 0.7119278311729431,
      "learning_rate": 1e-05,
      "loss": 0.0718,
      "step": 19690
    },
    {
      "epoch": 1.8777847290039064e-05,
      "model_forward_time": 5.613327503204346,
      "step": 19690
    },
    {
      "dynamics_loss_avg": 0.058721472322940824,
      "epoch": 1.8777847290039064e-05,
      "step": 19690
    },
    {
      "action_loss_avg": 0.007856030354741961,
      "epoch": 1.8777847290039064e-05,
      "step": 19690
    },
    {
      "epoch": 1.8777847290039064e-05,
      "step": 19690,
      "training_step_time": 17.493136882781982
    },
    {
      "epoch": 1.8778800964355468e-05,
      "model_forward_time": 5.6236138343811035,
      "step": 19691
    },
    {
      "epoch": 1.8778800964355468e-05,
      "step": 19691,
      "training_step_time": 17.12860369682312
    },
    {
      "epoch": 1.8779754638671875e-05,
      "model_forward_time": 5.611479997634888,
      "step": 19692
    },
    {
      "epoch": 1.8779754638671875e-05,
      "step": 19692,
      "training_step_time": 17.093899965286255
    },
    {
      "epoch": 1.8780708312988283e-05,
      "model_forward_time": 5.610019683837891,
      "step": 19693
    },
    {
      "epoch": 1.8780708312988283e-05,
      "step": 19693,
      "training_step_time": 17.08277654647827
    },
    {
      "epoch": 1.8781661987304687e-05,
      "model_forward_time": 5.6033313274383545,
      "step": 19694
    },
    {
      "epoch": 1.8781661987304687e-05,
      "step": 19694,
      "training_step_time": 17.085681915283203
    },
    {
      "epoch": 1.8782615661621094e-05,
      "model_forward_time": 5.611890554428101,
      "step": 19695
    },
    {
      "epoch": 1.8782615661621094e-05,
      "step": 19695,
      "training_step_time": 17.092811107635498
    },
    {
      "epoch": 1.87835693359375e-05,
      "model_forward_time": 5.9766364097595215,
      "step": 19696
    },
    {
      "epoch": 1.87835693359375e-05,
      "step": 19696,
      "training_step_time": 17.45371651649475
    },
    {
      "epoch": 1.8784523010253905e-05,
      "model_forward_time": 5.616817951202393,
      "step": 19697
    },
    {
      "epoch": 1.8784523010253905e-05,
      "step": 19697,
      "training_step_time": 17.084596633911133
    },
    {
      "epoch": 1.8785476684570313e-05,
      "model_forward_time": 5.5966761112213135,
      "step": 19698
    },
    {
      "epoch": 1.8785476684570313e-05,
      "step": 19698,
      "training_step_time": 17.026100158691406
    },
    {
      "epoch": 1.878643035888672e-05,
      "model_forward_time": 5.592183351516724,
      "step": 19699
    },
    {
      "epoch": 1.878643035888672e-05,
      "step": 19699,
      "training_step_time": 17.03451681137085
    },
    {
      "epoch": 1.8787384033203124e-05,
      "grad_norm": 0.5825884938240051,
      "learning_rate": 1e-05,
      "loss": 0.0587,
      "step": 19700
    },
    {
      "epoch": 1.8787384033203124e-05,
      "model_forward_time": 5.5870983600616455,
      "step": 19700
    },
    {
      "dynamics_loss_avg": 0.05565945971757173,
      "epoch": 1.8787384033203124e-05,
      "step": 19700
    },
    {
      "action_loss_avg": 0.006007410469464958,
      "epoch": 1.8787384033203124e-05,
      "step": 19700
    },
    {
      "epoch": 1.8787384033203124e-05,
      "step": 19700,
      "training_step_time": 17.028873682022095
    },
    {
      "epoch": 1.8788337707519532e-05,
      "model_forward_time": 5.95511269569397,
      "step": 19701
    },
    {
      "epoch": 1.8788337707519532e-05,
      "step": 19701,
      "training_step_time": 17.40830945968628
    },
    {
      "epoch": 1.878929138183594e-05,
      "model_forward_time": 5.594717025756836,
      "step": 19702
    },
    {
      "epoch": 1.878929138183594e-05,
      "step": 19702,
      "training_step_time": 17.073672771453857
    },
    {
      "epoch": 1.8790245056152343e-05,
      "model_forward_time": 5.6352057456970215,
      "step": 19703
    },
    {
      "epoch": 1.8790245056152343e-05,
      "step": 19703,
      "training_step_time": 17.0846848487854
    },
    {
      "epoch": 1.879119873046875e-05,
      "model_forward_time": 5.6011693477630615,
      "step": 19704
    },
    {
      "epoch": 1.879119873046875e-05,
      "step": 19704,
      "training_step_time": 17.037606477737427
    },
    {
      "epoch": 1.8792152404785155e-05,
      "model_forward_time": 5.58944845199585,
      "step": 19705
    },
    {
      "epoch": 1.8792152404785155e-05,
      "step": 19705,
      "training_step_time": 17.050394773483276
    },
    {
      "epoch": 1.8793106079101562e-05,
      "model_forward_time": 5.597985744476318,
      "step": 19706
    },
    {
      "epoch": 1.8793106079101562e-05,
      "step": 19706,
      "training_step_time": 17.064325094223022
    },
    {
      "epoch": 1.879405975341797e-05,
      "model_forward_time": 5.966515064239502,
      "step": 19707
    },
    {
      "epoch": 1.879405975341797e-05,
      "step": 19707,
      "training_step_time": 17.45129108428955
    },
    {
      "epoch": 1.8795013427734373e-05,
      "model_forward_time": 5.623702526092529,
      "step": 19708
    },
    {
      "epoch": 1.8795013427734373e-05,
      "step": 19708,
      "training_step_time": 17.11743426322937
    },
    {
      "epoch": 1.879596710205078e-05,
      "model_forward_time": 5.613619327545166,
      "step": 19709
    },
    {
      "epoch": 1.879596710205078e-05,
      "step": 19709,
      "training_step_time": 17.118237018585205
    },
    {
      "epoch": 1.879692077636719e-05,
      "grad_norm": 0.6181737184524536,
      "learning_rate": 1e-05,
      "loss": 0.06,
      "step": 19710
    },
    {
      "epoch": 1.879692077636719e-05,
      "model_forward_time": 5.633222341537476,
      "step": 19710
    },
    {
      "dynamics_loss_avg": 0.04838133230805397,
      "epoch": 1.879692077636719e-05,
      "step": 19710
    },
    {
      "action_loss_avg": 0.006613183859735727,
      "epoch": 1.879692077636719e-05,
      "step": 19710
    },
    {
      "epoch": 1.879692077636719e-05,
      "step": 19710,
      "training_step_time": 17.054471492767334
    },
    {
      "epoch": 1.8797874450683592e-05,
      "model_forward_time": 5.58708381652832,
      "step": 19711
    },
    {
      "epoch": 1.8797874450683592e-05,
      "step": 19711,
      "training_step_time": 17.018820762634277
    },
    {
      "epoch": 1.8798828125e-05,
      "model_forward_time": 5.583714485168457,
      "step": 19712
    },
    {
      "epoch": 1.8798828125e-05,
      "step": 19712,
      "training_step_time": 17.382781505584717
    },
    {
      "epoch": 1.8799781799316407e-05,
      "model_forward_time": 5.5607218742370605,
      "step": 19713
    },
    {
      "epoch": 1.8799781799316407e-05,
      "step": 19713,
      "training_step_time": 16.971269130706787
    },
    {
      "epoch": 1.880073547363281e-05,
      "model_forward_time": 5.565214157104492,
      "step": 19714
    },
    {
      "epoch": 1.880073547363281e-05,
      "step": 19714,
      "training_step_time": 16.977956533432007
    },
    {
      "epoch": 1.880168914794922e-05,
      "model_forward_time": 5.561913728713989,
      "step": 19715
    },
    {
      "epoch": 1.880168914794922e-05,
      "step": 19715,
      "training_step_time": 16.998277187347412
    },
    {
      "epoch": 1.8802642822265626e-05,
      "model_forward_time": 5.56390380859375,
      "step": 19716
    },
    {
      "epoch": 1.8802642822265626e-05,
      "step": 19716,
      "training_step_time": 17.004326105117798
    },
    {
      "epoch": 1.880359649658203e-05,
      "model_forward_time": 5.575156927108765,
      "step": 19717
    },
    {
      "epoch": 1.880359649658203e-05,
      "step": 19717,
      "training_step_time": 16.995404958724976
    },
    {
      "epoch": 1.8804550170898437e-05,
      "model_forward_time": 5.9542396068573,
      "step": 19718
    },
    {
      "epoch": 1.8804550170898437e-05,
      "step": 19718,
      "training_step_time": 17.377827644348145
    },
    {
      "epoch": 1.8805503845214845e-05,
      "model_forward_time": 5.5906147956848145,
      "step": 19719
    },
    {
      "epoch": 1.8805503845214845e-05,
      "step": 19719,
      "training_step_time": 17.02130627632141
    },
    {
      "epoch": 1.880645751953125e-05,
      "grad_norm": 0.7814587354660034,
      "learning_rate": 1e-05,
      "loss": 0.057,
      "step": 19720
    },
    {
      "epoch": 1.880645751953125e-05,
      "model_forward_time": 5.603729248046875,
      "step": 19720
    },
    {
      "dynamics_loss_avg": 0.05417847223579884,
      "epoch": 1.880645751953125e-05,
      "step": 19720
    },
    {
      "action_loss_avg": 0.007233875896781683,
      "epoch": 1.880645751953125e-05,
      "step": 19720
    },
    {
      "epoch": 1.880645751953125e-05,
      "step": 19720,
      "training_step_time": 17.087267875671387
    },
    {
      "epoch": 1.8807411193847656e-05,
      "model_forward_time": 5.610779523849487,
      "step": 19721
    },
    {
      "epoch": 1.8807411193847656e-05,
      "step": 19721,
      "training_step_time": 17.111217975616455
    },
    {
      "epoch": 1.8808364868164064e-05,
      "model_forward_time": 5.622777223587036,
      "step": 19722
    },
    {
      "epoch": 1.8808364868164064e-05,
      "step": 19722,
      "training_step_time": 17.133429765701294
    },
    {
      "epoch": 1.8809318542480468e-05,
      "model_forward_time": 5.620647668838501,
      "step": 19723
    },
    {
      "epoch": 1.8809318542480468e-05,
      "step": 19723,
      "training_step_time": 17.44681477546692
    },
    {
      "epoch": 1.8810272216796875e-05,
      "model_forward_time": 5.581702470779419,
      "step": 19724
    },
    {
      "epoch": 1.8810272216796875e-05,
      "step": 19724,
      "training_step_time": 17.01535940170288
    },
    {
      "epoch": 1.8811225891113283e-05,
      "model_forward_time": 5.592363595962524,
      "step": 19725
    },
    {
      "epoch": 1.8811225891113283e-05,
      "step": 19725,
      "training_step_time": 17.022452116012573
    },
    {
      "epoch": 1.8812179565429687e-05,
      "model_forward_time": 5.59043550491333,
      "step": 19726
    },
    {
      "epoch": 1.8812179565429687e-05,
      "step": 19726,
      "training_step_time": 17.035298109054565
    },
    {
      "epoch": 1.8813133239746094e-05,
      "model_forward_time": 5.59496283531189,
      "step": 19727
    },
    {
      "epoch": 1.8813133239746094e-05,
      "step": 19727,
      "training_step_time": 17.029573917388916
    },
    {
      "epoch": 1.88140869140625e-05,
      "model_forward_time": 5.5881054401397705,
      "step": 19728
    },
    {
      "epoch": 1.88140869140625e-05,
      "step": 19728,
      "training_step_time": 17.03878092765808
    },
    {
      "epoch": 1.8815040588378905e-05,
      "model_forward_time": 5.9630491733551025,
      "step": 19729
    },
    {
      "epoch": 1.8815040588378905e-05,
      "step": 19729,
      "training_step_time": 17.408485174179077
    },
    {
      "epoch": 1.8815994262695313e-05,
      "grad_norm": 0.6044677495956421,
      "learning_rate": 1e-05,
      "loss": 0.0673,
      "step": 19730
    },
    {
      "epoch": 1.8815994262695313e-05,
      "model_forward_time": 5.590088367462158,
      "step": 19730
    },
    {
      "dynamics_loss_avg": 0.05807335376739502,
      "epoch": 1.8815994262695313e-05,
      "step": 19730
    },
    {
      "action_loss_avg": 0.007173487520776689,
      "epoch": 1.8815994262695313e-05,
      "step": 19730
    },
    {
      "epoch": 1.8815994262695313e-05,
      "step": 19730,
      "training_step_time": 17.03573989868164
    },
    {
      "epoch": 1.881694793701172e-05,
      "model_forward_time": 5.611703872680664,
      "step": 19731
    },
    {
      "epoch": 1.881694793701172e-05,
      "step": 19731,
      "training_step_time": 17.051034927368164
    },
    {
      "epoch": 1.8817901611328124e-05,
      "model_forward_time": 5.592300653457642,
      "step": 19732
    },
    {
      "epoch": 1.8817901611328124e-05,
      "step": 19732,
      "training_step_time": 17.0294246673584
    },
    {
      "epoch": 1.8818855285644532e-05,
      "model_forward_time": 5.598092794418335,
      "step": 19733
    },
    {
      "epoch": 1.8818855285644532e-05,
      "step": 19733,
      "training_step_time": 17.055208206176758
    },
    {
      "epoch": 1.881980895996094e-05,
      "model_forward_time": 5.94230842590332,
      "step": 19734
    },
    {
      "epoch": 1.881980895996094e-05,
      "step": 19734,
      "training_step_time": 17.39461326599121
    },
    {
      "epoch": 1.8820762634277343e-05,
      "model_forward_time": 5.648881912231445,
      "step": 19735
    },
    {
      "epoch": 1.8820762634277343e-05,
      "step": 19735,
      "training_step_time": 17.149943828582764
    },
    {
      "epoch": 1.882171630859375e-05,
      "model_forward_time": 5.626501798629761,
      "step": 19736
    },
    {
      "epoch": 1.882171630859375e-05,
      "step": 19736,
      "training_step_time": 17.13519525527954
    },
    {
      "epoch": 1.8822669982910155e-05,
      "model_forward_time": 5.6017045974731445,
      "step": 19737
    },
    {
      "epoch": 1.8822669982910155e-05,
      "step": 19737,
      "training_step_time": 17.05817484855652
    },
    {
      "epoch": 1.8823623657226562e-05,
      "model_forward_time": 5.594568490982056,
      "step": 19738
    },
    {
      "epoch": 1.8823623657226562e-05,
      "step": 19738,
      "training_step_time": 17.037197589874268
    },
    {
      "epoch": 1.882457733154297e-05,
      "model_forward_time": 5.581931114196777,
      "step": 19739
    },
    {
      "epoch": 1.882457733154297e-05,
      "step": 19739,
      "training_step_time": 16.99383854866028
    },
    {
      "epoch": 1.8825531005859374e-05,
      "grad_norm": 0.4185228645801544,
      "learning_rate": 1e-05,
      "loss": 0.0503,
      "step": 19740
    },
    {
      "epoch": 1.8825531005859374e-05,
      "model_forward_time": 5.965665340423584,
      "step": 19740
    },
    {
      "dynamics_loss_avg": 0.045465904660522936,
      "epoch": 1.8825531005859374e-05,
      "step": 19740
    },
    {
      "action_loss_avg": 0.006175387173425406,
      "epoch": 1.8825531005859374e-05,
      "step": 19740
    },
    {
      "epoch": 1.8825531005859374e-05,
      "step": 19740,
      "training_step_time": 17.426247119903564
    },
    {
      "epoch": 1.882648468017578e-05,
      "model_forward_time": 5.605383634567261,
      "step": 19741
    },
    {
      "epoch": 1.882648468017578e-05,
      "step": 19741,
      "training_step_time": 17.07367491722107
    },
    {
      "epoch": 1.882743835449219e-05,
      "model_forward_time": 5.602485656738281,
      "step": 19742
    },
    {
      "epoch": 1.882743835449219e-05,
      "step": 19742,
      "training_step_time": 17.065836191177368
    },
    {
      "epoch": 1.8828392028808592e-05,
      "model_forward_time": 5.59386682510376,
      "step": 19743
    },
    {
      "epoch": 1.8828392028808592e-05,
      "step": 19743,
      "training_step_time": 17.041468381881714
    },
    {
      "epoch": 1.8829345703125e-05,
      "model_forward_time": 5.609927654266357,
      "step": 19744
    },
    {
      "epoch": 1.8829345703125e-05,
      "step": 19744,
      "training_step_time": 17.095938444137573
    },
    {
      "epoch": 1.8830299377441407e-05,
      "model_forward_time": 5.603963375091553,
      "step": 19745
    },
    {
      "epoch": 1.8830299377441407e-05,
      "step": 19745,
      "training_step_time": 17.46849799156189
    },
    {
      "epoch": 1.883125305175781e-05,
      "model_forward_time": 5.6087329387664795,
      "step": 19746
    },
    {
      "epoch": 1.883125305175781e-05,
      "step": 19746,
      "training_step_time": 17.09857988357544
    },
    {
      "epoch": 1.883220672607422e-05,
      "model_forward_time": 5.605301856994629,
      "step": 19747
    },
    {
      "epoch": 1.883220672607422e-05,
      "step": 19747,
      "training_step_time": 17.07257890701294
    },
    {
      "epoch": 1.8833160400390626e-05,
      "model_forward_time": 5.611022472381592,
      "step": 19748
    },
    {
      "epoch": 1.8833160400390626e-05,
      "step": 19748,
      "training_step_time": 17.085314989089966
    },
    {
      "epoch": 1.883411407470703e-05,
      "model_forward_time": 5.628219366073608,
      "step": 19749
    },
    {
      "epoch": 1.883411407470703e-05,
      "step": 19749,
      "training_step_time": 17.12928819656372
    },
    {
      "epoch": 1.8835067749023438e-05,
      "grad_norm": 0.8091188073158264,
      "learning_rate": 1e-05,
      "loss": 0.063,
      "step": 19750
    },
    {
      "epoch": 1.8835067749023438e-05,
      "model_forward_time": 5.607279300689697,
      "step": 19750
    },
    {
      "dynamics_loss_avg": 0.05543552823364735,
      "epoch": 1.8835067749023438e-05,
      "step": 19750
    },
    {
      "action_loss_avg": 0.006862503150478006,
      "epoch": 1.8835067749023438e-05,
      "step": 19750
    },
    {
      "epoch": 1.8835067749023438e-05,
      "step": 19750,
      "training_step_time": 17.086040496826172
    },
    {
      "epoch": 1.8836021423339845e-05,
      "model_forward_time": 5.980409383773804,
      "step": 19751
    },
    {
      "epoch": 1.8836021423339845e-05,
      "step": 19751,
      "training_step_time": 17.472679376602173
    },
    {
      "epoch": 1.883697509765625e-05,
      "model_forward_time": 5.598460674285889,
      "step": 19752
    },
    {
      "epoch": 1.883697509765625e-05,
      "step": 19752,
      "training_step_time": 17.062264442443848
    },
    {
      "epoch": 1.8837928771972656e-05,
      "model_forward_time": 5.5952208042144775,
      "step": 19753
    },
    {
      "epoch": 1.8837928771972656e-05,
      "step": 19753,
      "training_step_time": 17.02647638320923
    },
    {
      "epoch": 1.8838882446289064e-05,
      "model_forward_time": 5.5857789516448975,
      "step": 19754
    },
    {
      "epoch": 1.8838882446289064e-05,
      "step": 19754,
      "training_step_time": 17.026209831237793
    },
    {
      "epoch": 1.8839836120605468e-05,
      "model_forward_time": 5.605592489242554,
      "step": 19755
    },
    {
      "epoch": 1.8839836120605468e-05,
      "step": 19755,
      "training_step_time": 17.082807302474976
    },
    {
      "epoch": 1.8840789794921875e-05,
      "model_forward_time": 5.604137897491455,
      "step": 19756
    },
    {
      "epoch": 1.8840789794921875e-05,
      "step": 19756,
      "training_step_time": 17.43071222305298
    },
    {
      "epoch": 1.8841743469238283e-05,
      "model_forward_time": 4.273440599441528,
      "step": 19757
    },
    {
      "epoch": 1.8841743469238283e-05,
      "step": 19757,
      "training_step_time": 12.791337013244629
    },
    {
      "epoch": 1.8842697143554687e-05,
      "model_forward_time": 5.587486982345581,
      "step": 19758
    },
    {
      "epoch": 1.8842697143554687e-05,
      "step": 19758,
      "training_step_time": 17.041547060012817
    },
    {
      "epoch": 1.8843650817871094e-05,
      "model_forward_time": 5.5969460010528564,
      "step": 19759
    },
    {
      "epoch": 1.8843650817871094e-05,
      "step": 19759,
      "training_step_time": 17.0842502117157
    },
    {
      "epoch": 1.88446044921875e-05,
      "grad_norm": 1.1441575288772583,
      "learning_rate": 1e-05,
      "loss": 0.0638,
      "step": 19760
    },
    {
      "epoch": 1.88446044921875e-05,
      "model_forward_time": 5.624404191970825,
      "step": 19760
    },
    {
      "dynamics_loss_avg": 0.05488210376352072,
      "epoch": 1.88446044921875e-05,
      "step": 19760
    },
    {
      "action_loss_avg": 0.006979115842841566,
      "epoch": 1.88446044921875e-05,
      "step": 19760
    },
    {
      "epoch": 1.88446044921875e-05,
      "step": 19760,
      "training_step_time": 17.130471229553223
    },
    {
      "epoch": 1.8845558166503906e-05,
      "model_forward_time": 5.623656511306763,
      "step": 19761
    },
    {
      "epoch": 1.8845558166503906e-05,
      "step": 19761,
      "training_step_time": 17.130184173583984
    },
    {
      "epoch": 1.8846511840820313e-05,
      "model_forward_time": 5.993112087249756,
      "step": 19762
    },
    {
      "epoch": 1.8846511840820313e-05,
      "step": 19762,
      "training_step_time": 17.47492265701294
    },
    {
      "epoch": 1.884746551513672e-05,
      "model_forward_time": 5.6014814376831055,
      "step": 19763
    },
    {
      "epoch": 1.884746551513672e-05,
      "step": 19763,
      "training_step_time": 17.03872323036194
    },
    {
      "epoch": 1.8848419189453124e-05,
      "model_forward_time": 5.58431077003479,
      "step": 19764
    },
    {
      "epoch": 1.8848419189453124e-05,
      "step": 19764,
      "training_step_time": 17.018174171447754
    },
    {
      "epoch": 1.8849372863769532e-05,
      "model_forward_time": 5.5699849128723145,
      "step": 19765
    },
    {
      "epoch": 1.8849372863769532e-05,
      "step": 19765,
      "training_step_time": 17.01851463317871
    },
    {
      "epoch": 1.8850326538085936e-05,
      "model_forward_time": 5.575418472290039,
      "step": 19766
    },
    {
      "epoch": 1.8850326538085936e-05,
      "step": 19766,
      "training_step_time": 16.98057508468628
    },
    {
      "epoch": 1.8851280212402343e-05,
      "model_forward_time": 5.583778381347656,
      "step": 19767
    },
    {
      "epoch": 1.8851280212402343e-05,
      "step": 19767,
      "training_step_time": 17.376654624938965
    },
    {
      "epoch": 1.885223388671875e-05,
      "model_forward_time": 5.581318378448486,
      "step": 19768
    },
    {
      "epoch": 1.885223388671875e-05,
      "step": 19768,
      "training_step_time": 17.025830268859863
    },
    {
      "epoch": 1.8853187561035155e-05,
      "model_forward_time": 5.606318235397339,
      "step": 19769
    },
    {
      "epoch": 1.8853187561035155e-05,
      "step": 19769,
      "training_step_time": 17.088406324386597
    },
    {
      "epoch": 1.8854141235351562e-05,
      "grad_norm": 1.1752898693084717,
      "learning_rate": 1e-05,
      "loss": 0.0597,
      "step": 19770
    },
    {
      "epoch": 1.8854141235351562e-05,
      "model_forward_time": 5.604583263397217,
      "step": 19770
    },
    {
      "dynamics_loss_avg": 0.057441108487546445,
      "epoch": 1.8854141235351562e-05,
      "step": 19770
    },
    {
      "action_loss_avg": 0.005826911050826311,
      "epoch": 1.8854141235351562e-05,
      "step": 19770
    },
    {
      "epoch": 1.8854141235351562e-05,
      "step": 19770,
      "training_step_time": 17.094674348831177
    },
    {
      "epoch": 1.885509490966797e-05,
      "model_forward_time": 5.608643054962158,
      "step": 19771
    },
    {
      "epoch": 1.885509490966797e-05,
      "step": 19771,
      "training_step_time": 17.112011909484863
    },
    {
      "epoch": 1.8856048583984374e-05,
      "model_forward_time": 5.607423543930054,
      "step": 19772
    },
    {
      "epoch": 1.8856048583984374e-05,
      "step": 19772,
      "training_step_time": 17.090718030929565
    },
    {
      "epoch": 1.885700225830078e-05,
      "model_forward_time": 5.989735126495361,
      "step": 19773
    },
    {
      "epoch": 1.885700225830078e-05,
      "step": 19773,
      "training_step_time": 17.483649015426636
    },
    {
      "epoch": 1.885795593261719e-05,
      "model_forward_time": 5.611408948898315,
      "step": 19774
    },
    {
      "epoch": 1.885795593261719e-05,
      "step": 19774,
      "training_step_time": 17.10599160194397
    },
    {
      "epoch": 1.8858909606933592e-05,
      "model_forward_time": 5.613756895065308,
      "step": 19775
    },
    {
      "epoch": 1.8858909606933592e-05,
      "step": 19775,
      "training_step_time": 17.12235689163208
    },
    {
      "epoch": 1.885986328125e-05,
      "model_forward_time": 5.623135089874268,
      "step": 19776
    },
    {
      "epoch": 1.885986328125e-05,
      "step": 19776,
      "training_step_time": 17.07341766357422
    },
    {
      "epoch": 1.8860816955566407e-05,
      "model_forward_time": 5.611728191375732,
      "step": 19777
    },
    {
      "epoch": 1.8860816955566407e-05,
      "step": 19777,
      "training_step_time": 17.047847509384155
    },
    {
      "epoch": 1.886177062988281e-05,
      "model_forward_time": 5.929966688156128,
      "step": 19778
    },
    {
      "epoch": 1.886177062988281e-05,
      "step": 19778,
      "training_step_time": 17.339146375656128
    },
    {
      "epoch": 1.886272430419922e-05,
      "model_forward_time": 5.5843117237091064,
      "step": 19779
    },
    {
      "epoch": 1.886272430419922e-05,
      "step": 19779,
      "training_step_time": 17.00800132751465
    },
    {
      "epoch": 1.8863677978515626e-05,
      "grad_norm": 0.9446530342102051,
      "learning_rate": 1e-05,
      "loss": 0.0663,
      "step": 19780
    },
    {
      "epoch": 1.8863677978515626e-05,
      "model_forward_time": 5.571002721786499,
      "step": 19780
    },
    {
      "dynamics_loss_avg": 0.05906989574432373,
      "epoch": 1.8863677978515626e-05,
      "step": 19780
    },
    {
      "action_loss_avg": 0.006793086300604045,
      "epoch": 1.8863677978515626e-05,
      "step": 19780
    },
    {
      "epoch": 1.8863677978515626e-05,
      "step": 19780,
      "training_step_time": 17.014251470565796
    },
    {
      "epoch": 1.886463165283203e-05,
      "model_forward_time": 5.5818562507629395,
      "step": 19781
    },
    {
      "epoch": 1.886463165283203e-05,
      "step": 19781,
      "training_step_time": 17.005999326705933
    },
    {
      "epoch": 1.8865585327148438e-05,
      "model_forward_time": 5.5823585987091064,
      "step": 19782
    },
    {
      "epoch": 1.8865585327148438e-05,
      "step": 19782,
      "training_step_time": 17.032371997833252
    },
    {
      "epoch": 1.8866539001464845e-05,
      "model_forward_time": 5.612165212631226,
      "step": 19783
    },
    {
      "epoch": 1.8866539001464845e-05,
      "step": 19783,
      "training_step_time": 17.102293014526367
    },
    {
      "epoch": 1.886749267578125e-05,
      "model_forward_time": 6.0036187171936035,
      "step": 19784
    },
    {
      "epoch": 1.886749267578125e-05,
      "step": 19784,
      "training_step_time": 17.511668920516968
    },
    {
      "epoch": 1.8868446350097656e-05,
      "model_forward_time": 5.620909214019775,
      "step": 19785
    },
    {
      "epoch": 1.8868446350097656e-05,
      "step": 19785,
      "training_step_time": 17.14097285270691
    },
    {
      "epoch": 1.8869400024414064e-05,
      "model_forward_time": 5.622316122055054,
      "step": 19786
    },
    {
      "epoch": 1.8869400024414064e-05,
      "step": 19786,
      "training_step_time": 17.132272481918335
    },
    {
      "epoch": 1.8870353698730468e-05,
      "model_forward_time": 5.627563714981079,
      "step": 19787
    },
    {
      "epoch": 1.8870353698730468e-05,
      "step": 19787,
      "training_step_time": 17.104448556900024
    },
    {
      "epoch": 1.8871307373046875e-05,
      "model_forward_time": 5.596336841583252,
      "step": 19788
    },
    {
      "epoch": 1.8871307373046875e-05,
      "step": 19788,
      "training_step_time": 17.044920444488525
    },
    {
      "epoch": 1.8872261047363283e-05,
      "model_forward_time": 5.958230257034302,
      "step": 19789
    },
    {
      "epoch": 1.8872261047363283e-05,
      "step": 19789,
      "training_step_time": 17.390974283218384
    },
    {
      "epoch": 1.8873214721679687e-05,
      "grad_norm": 0.42851001024246216,
      "learning_rate": 1e-05,
      "loss": 0.0628,
      "step": 19790
    },
    {
      "epoch": 1.8873214721679687e-05,
      "model_forward_time": 5.579190731048584,
      "step": 19790
    },
    {
      "dynamics_loss_avg": 0.056533248908817765,
      "epoch": 1.8873214721679687e-05,
      "step": 19790
    },
    {
      "action_loss_avg": 0.005913308379240334,
      "epoch": 1.8873214721679687e-05,
      "step": 19790
    },
    {
      "epoch": 1.8873214721679687e-05,
      "step": 19790,
      "training_step_time": 17.01899003982544
    },
    {
      "epoch": 1.8874168395996094e-05,
      "model_forward_time": 5.580843925476074,
      "step": 19791
    },
    {
      "epoch": 1.8874168395996094e-05,
      "step": 19791,
      "training_step_time": 16.996872663497925
    },
    {
      "epoch": 1.88751220703125e-05,
      "model_forward_time": 5.596555948257446,
      "step": 19792
    },
    {
      "epoch": 1.88751220703125e-05,
      "step": 19792,
      "training_step_time": 17.062551498413086
    },
    {
      "epoch": 1.8876075744628906e-05,
      "model_forward_time": 5.628519058227539,
      "step": 19793
    },
    {
      "epoch": 1.8876075744628906e-05,
      "step": 19793,
      "training_step_time": 17.13358759880066
    },
    {
      "epoch": 1.8877029418945313e-05,
      "model_forward_time": 5.619427680969238,
      "step": 19794
    },
    {
      "epoch": 1.8877029418945313e-05,
      "step": 19794,
      "training_step_time": 17.107929944992065
    },
    {
      "epoch": 1.887798309326172e-05,
      "model_forward_time": 5.966762542724609,
      "step": 19795
    },
    {
      "epoch": 1.887798309326172e-05,
      "step": 19795,
      "training_step_time": 17.40728497505188
    },
    {
      "epoch": 1.8878936767578124e-05,
      "model_forward_time": 5.5818071365356445,
      "step": 19796
    },
    {
      "epoch": 1.8878936767578124e-05,
      "step": 19796,
      "training_step_time": 17.017041444778442
    },
    {
      "epoch": 1.8879890441894532e-05,
      "model_forward_time": 5.5870680809021,
      "step": 19797
    },
    {
      "epoch": 1.8879890441894532e-05,
      "step": 19797,
      "training_step_time": 17.002001523971558
    },
    {
      "epoch": 1.8880844116210936e-05,
      "model_forward_time": 5.5867156982421875,
      "step": 19798
    },
    {
      "epoch": 1.8880844116210936e-05,
      "step": 19798,
      "training_step_time": 17.024336099624634
    },
    {
      "epoch": 1.8881797790527343e-05,
      "model_forward_time": 5.5858681201934814,
      "step": 19799
    },
    {
      "epoch": 1.8881797790527343e-05,
      "step": 19799,
      "training_step_time": 17.020925760269165
    },
    {
      "epoch": 1.888275146484375e-05,
      "grad_norm": 0.3645497262477875,
      "learning_rate": 1e-05,
      "loss": 0.0709,
      "step": 19800
    },
    {
      "epoch": 1.888275146484375e-05,
      "model_forward_time": 5.940246343612671,
      "step": 19800
    },
    {
      "dynamics_loss_avg": 0.06409555952996016,
      "epoch": 1.888275146484375e-05,
      "step": 19800
    },
    {
      "action_loss_avg": 0.006995142309460789,
      "epoch": 1.888275146484375e-05,
      "step": 19800
    },
    {
      "epoch": 1.888275146484375e-05,
      "step": 19800,
      "training_step_time": 17.39855432510376
    },
    {
      "epoch": 1.8883705139160155e-05,
      "model_forward_time": 5.593040943145752,
      "step": 19801
    },
    {
      "epoch": 1.8883705139160155e-05,
      "step": 19801,
      "training_step_time": 17.05822443962097
    },
    {
      "epoch": 1.8884658813476562e-05,
      "model_forward_time": 5.6068925857543945,
      "step": 19802
    },
    {
      "epoch": 1.8884658813476562e-05,
      "step": 19802,
      "training_step_time": 17.09268069267273
    },
    {
      "epoch": 1.888561248779297e-05,
      "model_forward_time": 5.614426136016846,
      "step": 19803
    },
    {
      "epoch": 1.888561248779297e-05,
      "step": 19803,
      "training_step_time": 17.10270094871521
    },
    {
      "epoch": 1.8886566162109374e-05,
      "model_forward_time": 5.613529920578003,
      "step": 19804
    },
    {
      "epoch": 1.8886566162109374e-05,
      "step": 19804,
      "training_step_time": 17.117487907409668
    },
    {
      "epoch": 1.888751983642578e-05,
      "model_forward_time": 5.613861322402954,
      "step": 19805
    },
    {
      "epoch": 1.888751983642578e-05,
      "step": 19805,
      "training_step_time": 17.497536182403564
    },
    {
      "epoch": 1.888847351074219e-05,
      "model_forward_time": 5.622570276260376,
      "step": 19806
    },
    {
      "epoch": 1.888847351074219e-05,
      "step": 19806,
      "training_step_time": 17.10612988471985
    },
    {
      "epoch": 1.8889427185058592e-05,
      "model_forward_time": 5.609332084655762,
      "step": 19807
    },
    {
      "epoch": 1.8889427185058592e-05,
      "step": 19807,
      "training_step_time": 17.058866262435913
    },
    {
      "epoch": 1.8890380859375e-05,
      "model_forward_time": 5.597202301025391,
      "step": 19808
    },
    {
      "epoch": 1.8890380859375e-05,
      "step": 19808,
      "training_step_time": 17.050982236862183
    },
    {
      "epoch": 1.8891334533691407e-05,
      "model_forward_time": 5.591611623764038,
      "step": 19809
    },
    {
      "epoch": 1.8891334533691407e-05,
      "step": 19809,
      "training_step_time": 17.025085926055908
    },
    {
      "epoch": 1.889228820800781e-05,
      "grad_norm": 0.30895087122917175,
      "learning_rate": 1e-05,
      "loss": 0.0547,
      "step": 19810
    },
    {
      "epoch": 1.889228820800781e-05,
      "model_forward_time": 5.58331036567688,
      "step": 19810
    },
    {
      "dynamics_loss_avg": 0.047496641613543035,
      "epoch": 1.889228820800781e-05,
      "step": 19810
    },
    {
      "action_loss_avg": 0.0044348008232191205,
      "epoch": 1.889228820800781e-05,
      "step": 19810
    },
    {
      "epoch": 1.889228820800781e-05,
      "step": 19810,
      "training_step_time": 17.039104461669922
    },
    {
      "epoch": 1.889324188232422e-05,
      "model_forward_time": 5.949276685714722,
      "step": 19811
    },
    {
      "epoch": 1.889324188232422e-05,
      "step": 19811,
      "training_step_time": 17.440345764160156
    },
    {
      "epoch": 1.8894195556640626e-05,
      "model_forward_time": 5.645933628082275,
      "step": 19812
    },
    {
      "epoch": 1.8894195556640626e-05,
      "step": 19812,
      "training_step_time": 17.175978660583496
    },
    {
      "epoch": 1.889514923095703e-05,
      "model_forward_time": 5.603943109512329,
      "step": 19813
    },
    {
      "epoch": 1.889514923095703e-05,
      "step": 19813,
      "training_step_time": 17.065971851348877
    },
    {
      "epoch": 1.8896102905273438e-05,
      "model_forward_time": 5.589734792709351,
      "step": 19814
    },
    {
      "epoch": 1.8896102905273438e-05,
      "step": 19814,
      "training_step_time": 17.02043104171753
    },
    {
      "epoch": 1.8897056579589845e-05,
      "model_forward_time": 5.5812294483184814,
      "step": 19815
    },
    {
      "epoch": 1.8897056579589845e-05,
      "step": 19815,
      "training_step_time": 16.996530055999756
    },
    {
      "epoch": 1.889801025390625e-05,
      "model_forward_time": 5.585430383682251,
      "step": 19816
    },
    {
      "epoch": 1.889801025390625e-05,
      "step": 19816,
      "training_step_time": 17.35607075691223
    },
    {
      "epoch": 1.8898963928222656e-05,
      "model_forward_time": 5.591110467910767,
      "step": 19817
    },
    {
      "epoch": 1.8898963928222656e-05,
      "step": 19817,
      "training_step_time": 17.01956272125244
    },
    {
      "epoch": 1.8899917602539064e-05,
      "model_forward_time": 5.583837985992432,
      "step": 19818
    },
    {
      "epoch": 1.8899917602539064e-05,
      "step": 19818,
      "training_step_time": 17.017131090164185
    },
    {
      "epoch": 1.8900871276855468e-05,
      "model_forward_time": 5.578709840774536,
      "step": 19819
    },
    {
      "epoch": 1.8900871276855468e-05,
      "step": 19819,
      "training_step_time": 17.0445077419281
    },
    {
      "epoch": 1.8901824951171875e-05,
      "grad_norm": 0.6310526728630066,
      "learning_rate": 1e-05,
      "loss": 0.0655,
      "step": 19820
    },
    {
      "epoch": 1.8901824951171875e-05,
      "model_forward_time": 5.621297121047974,
      "step": 19820
    },
    {
      "dynamics_loss_avg": 0.05995897985994816,
      "epoch": 1.8901824951171875e-05,
      "step": 19820
    },
    {
      "action_loss_avg": 0.006086431001313031,
      "epoch": 1.8901824951171875e-05,
      "step": 19820
    },
    {
      "epoch": 1.8901824951171875e-05,
      "step": 19820,
      "training_step_time": 17.125970125198364
    },
    {
      "epoch": 1.8902778625488283e-05,
      "model_forward_time": 5.612437009811401,
      "step": 19821
    },
    {
      "epoch": 1.8902778625488283e-05,
      "step": 19821,
      "training_step_time": 17.095444679260254
    },
    {
      "epoch": 1.8903732299804687e-05,
      "model_forward_time": 5.9832258224487305,
      "step": 19822
    },
    {
      "epoch": 1.8903732299804687e-05,
      "step": 19822,
      "training_step_time": 17.466262817382812
    },
    {
      "epoch": 1.8904685974121094e-05,
      "model_forward_time": 5.609826326370239,
      "step": 19823
    },
    {
      "epoch": 1.8904685974121094e-05,
      "step": 19823,
      "training_step_time": 17.084566593170166
    },
    {
      "epoch": 1.89056396484375e-05,
      "model_forward_time": 5.630291938781738,
      "step": 19824
    },
    {
      "epoch": 1.89056396484375e-05,
      "step": 19824,
      "training_step_time": 17.117628812789917
    },
    {
      "epoch": 1.8906593322753906e-05,
      "model_forward_time": 5.608688116073608,
      "step": 19825
    },
    {
      "epoch": 1.8906593322753906e-05,
      "step": 19825,
      "training_step_time": 17.1106595993042
    },
    {
      "epoch": 1.8907546997070313e-05,
      "model_forward_time": 5.609999656677246,
      "step": 19826
    },
    {
      "epoch": 1.8907546997070313e-05,
      "step": 19826,
      "training_step_time": 17.135173797607422
    },
    {
      "epoch": 1.890850067138672e-05,
      "model_forward_time": 5.9980692863464355,
      "step": 19827
    },
    {
      "epoch": 1.890850067138672e-05,
      "step": 19827,
      "training_step_time": 17.491649866104126
    },
    {
      "epoch": 1.8909454345703124e-05,
      "model_forward_time": 5.6104724407196045,
      "step": 19828
    },
    {
      "epoch": 1.8909454345703124e-05,
      "step": 19828,
      "training_step_time": 17.108993768692017
    },
    {
      "epoch": 1.8910408020019532e-05,
      "model_forward_time": 5.6174376010894775,
      "step": 19829
    },
    {
      "epoch": 1.8910408020019532e-05,
      "step": 19829,
      "training_step_time": 17.11020040512085
    },
    {
      "epoch": 1.8911361694335936e-05,
      "grad_norm": 0.6499208211898804,
      "learning_rate": 1e-05,
      "loss": 0.0611,
      "step": 19830
    },
    {
      "epoch": 1.8911361694335936e-05,
      "model_forward_time": 5.6138389110565186,
      "step": 19830
    },
    {
      "dynamics_loss_avg": 0.05551557019352913,
      "epoch": 1.8911361694335936e-05,
      "step": 19830
    },
    {
      "action_loss_avg": 0.005561695899814367,
      "epoch": 1.8911361694335936e-05,
      "step": 19830
    },
    {
      "epoch": 1.8911361694335936e-05,
      "step": 19830,
      "training_step_time": 17.106698274612427
    },
    {
      "epoch": 1.8912315368652343e-05,
      "model_forward_time": 5.617720603942871,
      "step": 19831
    },
    {
      "epoch": 1.8912315368652343e-05,
      "step": 19831,
      "training_step_time": 17.108097076416016
    },
    {
      "epoch": 1.891326904296875e-05,
      "model_forward_time": 5.611681699752808,
      "step": 19832
    },
    {
      "epoch": 1.891326904296875e-05,
      "step": 19832,
      "training_step_time": 17.10615372657776
    },
    {
      "epoch": 1.8914222717285155e-05,
      "model_forward_time": 5.96304726600647,
      "step": 19833
    },
    {
      "epoch": 1.8914222717285155e-05,
      "step": 19833,
      "training_step_time": 17.437751293182373
    },
    {
      "epoch": 1.8915176391601562e-05,
      "model_forward_time": 5.603100061416626,
      "step": 19834
    },
    {
      "epoch": 1.8915176391601562e-05,
      "step": 19834,
      "training_step_time": 17.075735569000244
    },
    {
      "epoch": 1.891613006591797e-05,
      "model_forward_time": 5.591486930847168,
      "step": 19835
    },
    {
      "epoch": 1.891613006591797e-05,
      "step": 19835,
      "training_step_time": 17.05604887008667
    },
    {
      "epoch": 1.8917083740234374e-05,
      "model_forward_time": 5.615691423416138,
      "step": 19836
    },
    {
      "epoch": 1.8917083740234374e-05,
      "step": 19836,
      "training_step_time": 17.100627422332764
    },
    {
      "epoch": 1.891803741455078e-05,
      "model_forward_time": 5.6140296459198,
      "step": 19837
    },
    {
      "epoch": 1.891803741455078e-05,
      "step": 19837,
      "training_step_time": 17.1042320728302
    },
    {
      "epoch": 1.891899108886719e-05,
      "model_forward_time": 6.003227472305298,
      "step": 19838
    },
    {
      "epoch": 1.891899108886719e-05,
      "step": 19838,
      "training_step_time": 17.502244472503662
    },
    {
      "epoch": 1.8919944763183592e-05,
      "model_forward_time": 5.61283540725708,
      "step": 19839
    },
    {
      "epoch": 1.8919944763183592e-05,
      "step": 19839,
      "training_step_time": 17.097683429718018
    },
    {
      "epoch": 1.89208984375e-05,
      "grad_norm": 0.7240055203437805,
      "learning_rate": 1e-05,
      "loss": 0.056,
      "step": 19840
    },
    {
      "epoch": 1.89208984375e-05,
      "model_forward_time": 5.616266965866089,
      "step": 19840
    },
    {
      "dynamics_loss_avg": 0.049300067871809,
      "epoch": 1.89208984375e-05,
      "step": 19840
    },
    {
      "action_loss_avg": 0.005768560478463769,
      "epoch": 1.89208984375e-05,
      "step": 19840
    },
    {
      "epoch": 1.89208984375e-05,
      "step": 19840,
      "training_step_time": 17.051215648651123
    },
    {
      "epoch": 1.8921852111816407e-05,
      "model_forward_time": 5.583122968673706,
      "step": 19841
    },
    {
      "epoch": 1.8921852111816407e-05,
      "step": 19841,
      "training_step_time": 17.03503131866455
    },
    {
      "epoch": 1.892280578613281e-05,
      "model_forward_time": 5.590006113052368,
      "step": 19842
    },
    {
      "epoch": 1.892280578613281e-05,
      "step": 19842,
      "training_step_time": 17.04431986808777
    },
    {
      "epoch": 1.892375946044922e-05,
      "model_forward_time": 5.620851993560791,
      "step": 19843
    },
    {
      "epoch": 1.892375946044922e-05,
      "step": 19843,
      "training_step_time": 17.1073637008667
    },
    {
      "epoch": 1.8924713134765626e-05,
      "model_forward_time": 5.976010799407959,
      "step": 19844
    },
    {
      "epoch": 1.8924713134765626e-05,
      "step": 19844,
      "training_step_time": 17.424396991729736
    },
    {
      "epoch": 1.892566680908203e-05,
      "model_forward_time": 5.594639539718628,
      "step": 19845
    },
    {
      "epoch": 1.892566680908203e-05,
      "step": 19845,
      "training_step_time": 17.0471510887146
    },
    {
      "epoch": 1.8926620483398438e-05,
      "model_forward_time": 5.586709260940552,
      "step": 19846
    },
    {
      "epoch": 1.8926620483398438e-05,
      "step": 19846,
      "training_step_time": 17.039361000061035
    },
    {
      "epoch": 1.8927574157714845e-05,
      "model_forward_time": 5.608312368392944,
      "step": 19847
    },
    {
      "epoch": 1.8927574157714845e-05,
      "step": 19847,
      "training_step_time": 17.098798513412476
    },
    {
      "epoch": 1.892852783203125e-05,
      "model_forward_time": 5.597489356994629,
      "step": 19848
    },
    {
      "epoch": 1.892852783203125e-05,
      "step": 19848,
      "training_step_time": 17.03886604309082
    },
    {
      "epoch": 1.8929481506347656e-05,
      "model_forward_time": 5.972703456878662,
      "step": 19849
    },
    {
      "epoch": 1.8929481506347656e-05,
      "step": 19849,
      "training_step_time": 17.412622451782227
    },
    {
      "epoch": 1.8930435180664064e-05,
      "grad_norm": 0.6506568789482117,
      "learning_rate": 1e-05,
      "loss": 0.0653,
      "step": 19850
    },
    {
      "epoch": 1.8930435180664064e-05,
      "model_forward_time": 5.580154657363892,
      "step": 19850
    },
    {
      "dynamics_loss_avg": 0.06183671988546848,
      "epoch": 1.8930435180664064e-05,
      "step": 19850
    },
    {
      "action_loss_avg": 0.005377590516582131,
      "epoch": 1.8930435180664064e-05,
      "step": 19850
    },
    {
      "epoch": 1.8930435180664064e-05,
      "step": 19850,
      "training_step_time": 17.02873659133911
    },
    {
      "epoch": 1.8931388854980468e-05,
      "model_forward_time": 5.596262216567993,
      "step": 19851
    },
    {
      "epoch": 1.8931388854980468e-05,
      "step": 19851,
      "training_step_time": 17.07896089553833
    },
    {
      "epoch": 1.8932342529296875e-05,
      "model_forward_time": 5.623522043228149,
      "step": 19852
    },
    {
      "epoch": 1.8932342529296875e-05,
      "step": 19852,
      "training_step_time": 17.123147010803223
    },
    {
      "epoch": 1.8933296203613283e-05,
      "model_forward_time": 5.624356269836426,
      "step": 19853
    },
    {
      "epoch": 1.8933296203613283e-05,
      "step": 19853,
      "training_step_time": 17.129298210144043
    },
    {
      "epoch": 1.8934249877929687e-05,
      "model_forward_time": 5.962666273117065,
      "step": 19854
    },
    {
      "epoch": 1.8934249877929687e-05,
      "step": 19854,
      "training_step_time": 17.42270255088806
    },
    {
      "epoch": 1.8935203552246094e-05,
      "model_forward_time": 5.593708515167236,
      "step": 19855
    },
    {
      "epoch": 1.8935203552246094e-05,
      "step": 19855,
      "training_step_time": 17.038381099700928
    },
    {
      "epoch": 1.89361572265625e-05,
      "model_forward_time": 5.599292755126953,
      "step": 19856
    },
    {
      "epoch": 1.89361572265625e-05,
      "step": 19856,
      "training_step_time": 17.06377911567688
    },
    {
      "epoch": 1.8937110900878906e-05,
      "model_forward_time": 5.612046957015991,
      "step": 19857
    },
    {
      "epoch": 1.8937110900878906e-05,
      "step": 19857,
      "training_step_time": 17.093457221984863
    },
    {
      "epoch": 1.8938064575195313e-05,
      "model_forward_time": 5.609995365142822,
      "step": 19858
    },
    {
      "epoch": 1.8938064575195313e-05,
      "step": 19858,
      "training_step_time": 17.07472825050354
    },
    {
      "epoch": 1.893901824951172e-05,
      "model_forward_time": 5.599534273147583,
      "step": 19859
    },
    {
      "epoch": 1.893901824951172e-05,
      "step": 19859,
      "training_step_time": 17.052316188812256
    },
    {
      "epoch": 1.8939971923828124e-05,
      "grad_norm": 0.808055579662323,
      "learning_rate": 1e-05,
      "loss": 0.0631,
      "step": 19860
    },
    {
      "epoch": 1.8939971923828124e-05,
      "model_forward_time": 5.973430633544922,
      "step": 19860
    },
    {
      "dynamics_loss_avg": 0.05752781201153993,
      "epoch": 1.8939971923828124e-05,
      "step": 19860
    },
    {
      "action_loss_avg": 0.0073010374559089545,
      "epoch": 1.8939971923828124e-05,
      "step": 19860
    },
    {
      "epoch": 1.8939971923828124e-05,
      "step": 19860,
      "training_step_time": 17.436266899108887
    },
    {
      "epoch": 1.8940925598144532e-05,
      "model_forward_time": 5.605544805526733,
      "step": 19861
    },
    {
      "epoch": 1.8940925598144532e-05,
      "step": 19861,
      "training_step_time": 17.09098243713379
    },
    {
      "epoch": 1.8941879272460936e-05,
      "model_forward_time": 5.620687484741211,
      "step": 19862
    },
    {
      "epoch": 1.8941879272460936e-05,
      "step": 19862,
      "training_step_time": 17.107449769973755
    },
    {
      "epoch": 1.8942832946777343e-05,
      "model_forward_time": 5.601805210113525,
      "step": 19863
    },
    {
      "epoch": 1.8942832946777343e-05,
      "step": 19863,
      "training_step_time": 17.07047700881958
    },
    {
      "epoch": 1.894378662109375e-05,
      "model_forward_time": 5.603105545043945,
      "step": 19864
    },
    {
      "epoch": 1.894378662109375e-05,
      "step": 19864,
      "training_step_time": 17.040748357772827
    },
    {
      "epoch": 1.8944740295410155e-05,
      "model_forward_time": 5.949650526046753,
      "step": 19865
    },
    {
      "epoch": 1.8944740295410155e-05,
      "step": 19865,
      "training_step_time": 17.384225368499756
    },
    {
      "epoch": 1.8945693969726562e-05,
      "model_forward_time": 5.579283952713013,
      "step": 19866
    },
    {
      "epoch": 1.8945693969726562e-05,
      "step": 19866,
      "training_step_time": 17.014262199401855
    },
    {
      "epoch": 1.894664764404297e-05,
      "model_forward_time": 5.5944297313690186,
      "step": 19867
    },
    {
      "epoch": 1.894664764404297e-05,
      "step": 19867,
      "training_step_time": 17.060633659362793
    },
    {
      "epoch": 1.8947601318359374e-05,
      "model_forward_time": 5.613460540771484,
      "step": 19868
    },
    {
      "epoch": 1.8947601318359374e-05,
      "step": 19868,
      "training_step_time": 17.10884690284729
    },
    {
      "epoch": 1.894855499267578e-05,
      "model_forward_time": 5.626060724258423,
      "step": 19869
    },
    {
      "epoch": 1.894855499267578e-05,
      "step": 19869,
      "training_step_time": 17.06324005126953
    },
    {
      "epoch": 1.894950866699219e-05,
      "grad_norm": 0.7301456332206726,
      "learning_rate": 1e-05,
      "loss": 0.0586,
      "step": 19870
    },
    {
      "epoch": 1.894950866699219e-05,
      "model_forward_time": 5.604569435119629,
      "step": 19870
    },
    {
      "dynamics_loss_avg": 0.04990321062505245,
      "epoch": 1.894950866699219e-05,
      "step": 19870
    },
    {
      "action_loss_avg": 0.005174371344037354,
      "epoch": 1.894950866699219e-05,
      "step": 19870
    },
    {
      "epoch": 1.894950866699219e-05,
      "step": 19870,
      "training_step_time": 17.069376707077026
    },
    {
      "epoch": 1.8950462341308592e-05,
      "model_forward_time": 5.971370458602905,
      "step": 19871
    },
    {
      "epoch": 1.8950462341308592e-05,
      "step": 19871,
      "training_step_time": 17.43265461921692
    },
    {
      "epoch": 1.8951416015625e-05,
      "model_forward_time": 5.590330123901367,
      "step": 19872
    },
    {
      "epoch": 1.8951416015625e-05,
      "step": 19872,
      "training_step_time": 17.05935764312744
    },
    {
      "epoch": 1.8952369689941407e-05,
      "model_forward_time": 5.601944208145142,
      "step": 19873
    },
    {
      "epoch": 1.8952369689941407e-05,
      "step": 19873,
      "training_step_time": 17.049644470214844
    },
    {
      "epoch": 1.895332336425781e-05,
      "model_forward_time": 5.598891019821167,
      "step": 19874
    },
    {
      "epoch": 1.895332336425781e-05,
      "step": 19874,
      "training_step_time": 17.049546480178833
    },
    {
      "epoch": 1.895427703857422e-05,
      "model_forward_time": 5.597593307495117,
      "step": 19875
    },
    {
      "epoch": 1.895427703857422e-05,
      "step": 19875,
      "training_step_time": 17.07716464996338
    },
    {
      "epoch": 1.8955230712890626e-05,
      "model_forward_time": 5.5980544090271,
      "step": 19876
    },
    {
      "epoch": 1.8955230712890626e-05,
      "step": 19876,
      "training_step_time": 17.442288160324097
    },
    {
      "epoch": 1.895618438720703e-05,
      "model_forward_time": 5.615983247756958,
      "step": 19877
    },
    {
      "epoch": 1.895618438720703e-05,
      "step": 19877,
      "training_step_time": 17.103798151016235
    },
    {
      "epoch": 1.8957138061523438e-05,
      "model_forward_time": 5.610077619552612,
      "step": 19878
    },
    {
      "epoch": 1.8957138061523438e-05,
      "step": 19878,
      "training_step_time": 17.10439682006836
    },
    {
      "epoch": 1.8958091735839845e-05,
      "model_forward_time": 5.612754583358765,
      "step": 19879
    },
    {
      "epoch": 1.8958091735839845e-05,
      "step": 19879,
      "training_step_time": 17.101567029953003
    },
    {
      "epoch": 1.895904541015625e-05,
      "grad_norm": 0.280175119638443,
      "learning_rate": 1e-05,
      "loss": 0.0456,
      "step": 19880
    },
    {
      "epoch": 1.895904541015625e-05,
      "model_forward_time": 5.6191112995147705,
      "step": 19880
    },
    {
      "dynamics_loss_avg": 0.04748816527426243,
      "epoch": 1.895904541015625e-05,
      "step": 19880
    },
    {
      "action_loss_avg": 0.004086286597885192,
      "epoch": 1.895904541015625e-05,
      "step": 19880
    },
    {
      "epoch": 1.895904541015625e-05,
      "step": 19880,
      "training_step_time": 17.10786247253418
    },
    {
      "epoch": 1.8959999084472656e-05,
      "model_forward_time": 5.616092205047607,
      "step": 19881
    },
    {
      "epoch": 1.8959999084472656e-05,
      "step": 19881,
      "training_step_time": 17.112385272979736
    },
    {
      "epoch": 1.8960952758789064e-05,
      "model_forward_time": 5.974233388900757,
      "step": 19882
    },
    {
      "epoch": 1.8960952758789064e-05,
      "step": 19882,
      "training_step_time": 17.46606945991516
    },
    {
      "epoch": 1.8961906433105468e-05,
      "model_forward_time": 5.617157220840454,
      "step": 19883
    },
    {
      "epoch": 1.8961906433105468e-05,
      "step": 19883,
      "training_step_time": 17.104706525802612
    },
    {
      "epoch": 1.8962860107421875e-05,
      "model_forward_time": 5.616567134857178,
      "step": 19884
    },
    {
      "epoch": 1.8962860107421875e-05,
      "step": 19884,
      "training_step_time": 17.089281797409058
    },
    {
      "epoch": 1.8963813781738283e-05,
      "model_forward_time": 5.600361108779907,
      "step": 19885
    },
    {
      "epoch": 1.8963813781738283e-05,
      "step": 19885,
      "training_step_time": 17.078871726989746
    },
    {
      "epoch": 1.8964767456054687e-05,
      "model_forward_time": 5.6135571002960205,
      "step": 19886
    },
    {
      "epoch": 1.8964767456054687e-05,
      "step": 19886,
      "training_step_time": 17.080739498138428
    },
    {
      "epoch": 1.8965721130371094e-05,
      "model_forward_time": 5.964312314987183,
      "step": 19887
    },
    {
      "epoch": 1.8965721130371094e-05,
      "step": 19887,
      "training_step_time": 17.442618131637573
    },
    {
      "epoch": 1.89666748046875e-05,
      "model_forward_time": 5.606794595718384,
      "step": 19888
    },
    {
      "epoch": 1.89666748046875e-05,
      "step": 19888,
      "training_step_time": 17.080870866775513
    },
    {
      "epoch": 1.8967628479003906e-05,
      "model_forward_time": 5.613733530044556,
      "step": 19889
    },
    {
      "epoch": 1.8967628479003906e-05,
      "step": 19889,
      "training_step_time": 17.07950258255005
    },
    {
      "epoch": 1.8968582153320313e-05,
      "grad_norm": 1.3015919923782349,
      "learning_rate": 1e-05,
      "loss": 0.0678,
      "step": 19890
    },
    {
      "epoch": 1.8968582153320313e-05,
      "model_forward_time": 5.587917804718018,
      "step": 19890
    },
    {
      "dynamics_loss_avg": 0.060679486207664016,
      "epoch": 1.8968582153320313e-05,
      "step": 19890
    },
    {
      "action_loss_avg": 0.005613178806379437,
      "epoch": 1.8968582153320313e-05,
      "step": 19890
    },
    {
      "epoch": 1.8968582153320313e-05,
      "step": 19890,
      "training_step_time": 17.02357530593872
    },
    {
      "epoch": 1.8969535827636717e-05,
      "model_forward_time": 5.5827038288116455,
      "step": 19891
    },
    {
      "epoch": 1.8969535827636717e-05,
      "step": 19891,
      "training_step_time": 17.03561544418335
    },
    {
      "epoch": 1.8970489501953124e-05,
      "model_forward_time": 5.594794988632202,
      "step": 19892
    },
    {
      "epoch": 1.8970489501953124e-05,
      "step": 19892,
      "training_step_time": 17.087779998779297
    },
    {
      "epoch": 1.8971443176269532e-05,
      "model_forward_time": 5.98387336730957,
      "step": 19893
    },
    {
      "epoch": 1.8971443176269532e-05,
      "step": 19893,
      "training_step_time": 17.461844205856323
    },
    {
      "epoch": 1.8972396850585936e-05,
      "model_forward_time": 5.615942001342773,
      "step": 19894
    },
    {
      "epoch": 1.8972396850585936e-05,
      "step": 19894,
      "training_step_time": 17.09629249572754
    },
    {
      "epoch": 1.8973350524902343e-05,
      "model_forward_time": 5.609638929367065,
      "step": 19895
    },
    {
      "epoch": 1.8973350524902343e-05,
      "step": 19895,
      "training_step_time": 17.10975456237793
    },
    {
      "epoch": 1.897430419921875e-05,
      "model_forward_time": 5.614999294281006,
      "step": 19896
    },
    {
      "epoch": 1.897430419921875e-05,
      "step": 19896,
      "training_step_time": 17.099671602249146
    },
    {
      "epoch": 1.8975257873535155e-05,
      "model_forward_time": 5.604267120361328,
      "step": 19897
    },
    {
      "epoch": 1.8975257873535155e-05,
      "step": 19897,
      "training_step_time": 17.0496723651886
    },
    {
      "epoch": 1.8976211547851562e-05,
      "model_forward_time": 5.945190906524658,
      "step": 19898
    },
    {
      "epoch": 1.8976211547851562e-05,
      "step": 19898,
      "training_step_time": 17.39277458190918
    },
    {
      "epoch": 1.897716522216797e-05,
      "model_forward_time": 5.5801310539245605,
      "step": 19899
    },
    {
      "epoch": 1.897716522216797e-05,
      "step": 19899,
      "training_step_time": 16.998279094696045
    },
    {
      "epoch": 1.8978118896484374e-05,
      "grad_norm": 0.5530292391777039,
      "learning_rate": 1e-05,
      "loss": 0.0665,
      "step": 19900
    },
    {
      "epoch": 1.8978118896484374e-05,
      "model_forward_time": 5.572465181350708,
      "step": 19900
    },
    {
      "dynamics_loss_avg": 0.057788952998816964,
      "epoch": 1.8978118896484374e-05,
      "step": 19900
    },
    {
      "action_loss_avg": 0.005383079522289336,
      "epoch": 1.8978118896484374e-05,
      "step": 19900
    },
    {
      "epoch": 1.8978118896484374e-05,
      "step": 19900,
      "training_step_time": 16.984558582305908
    },
    {
      "epoch": 1.897907257080078e-05,
      "model_forward_time": 5.589653015136719,
      "step": 19901
    },
    {
      "epoch": 1.897907257080078e-05,
      "step": 19901,
      "training_step_time": 17.03523850440979
    },
    {
      "epoch": 1.898002624511719e-05,
      "model_forward_time": 5.582422494888306,
      "step": 19902
    },
    {
      "epoch": 1.898002624511719e-05,
      "step": 19902,
      "training_step_time": 17.038658142089844
    },
    {
      "epoch": 1.8980979919433593e-05,
      "model_forward_time": 5.60304594039917,
      "step": 19903
    },
    {
      "epoch": 1.8980979919433593e-05,
      "step": 19903,
      "training_step_time": 17.077403783798218
    },
    {
      "epoch": 1.898193359375e-05,
      "model_forward_time": 5.986827850341797,
      "step": 19904
    },
    {
      "epoch": 1.898193359375e-05,
      "step": 19904,
      "training_step_time": 17.477969884872437
    },
    {
      "epoch": 1.8982887268066407e-05,
      "model_forward_time": 5.624872446060181,
      "step": 19905
    },
    {
      "epoch": 1.8982887268066407e-05,
      "step": 19905,
      "training_step_time": 17.127155303955078
    },
    {
      "epoch": 1.898384094238281e-05,
      "model_forward_time": 5.626423120498657,
      "step": 19906
    },
    {
      "epoch": 1.898384094238281e-05,
      "step": 19906,
      "training_step_time": 17.12902307510376
    },
    {
      "epoch": 1.898479461669922e-05,
      "model_forward_time": 5.614253044128418,
      "step": 19907
    },
    {
      "epoch": 1.898479461669922e-05,
      "step": 19907,
      "training_step_time": 17.099791288375854
    },
    {
      "epoch": 1.8985748291015626e-05,
      "model_forward_time": 5.593337774276733,
      "step": 19908
    },
    {
      "epoch": 1.8985748291015626e-05,
      "step": 19908,
      "training_step_time": 17.01051926612854
    },
    {
      "epoch": 1.898670196533203e-05,
      "model_forward_time": 5.580465316772461,
      "step": 19909
    },
    {
      "epoch": 1.898670196533203e-05,
      "step": 19909,
      "training_step_time": 17.405941486358643
    },
    {
      "epoch": 1.8987655639648438e-05,
      "grad_norm": 0.7720972299575806,
      "learning_rate": 1e-05,
      "loss": 0.0576,
      "step": 19910
    },
    {
      "epoch": 1.8987655639648438e-05,
      "model_forward_time": 5.5869200229644775,
      "step": 19910
    },
    {
      "dynamics_loss_avg": 0.05787777323275804,
      "epoch": 1.8987655639648438e-05,
      "step": 19910
    },
    {
      "action_loss_avg": 0.007981119072064757,
      "epoch": 1.8987655639648438e-05,
      "step": 19910
    },
    {
      "epoch": 1.8987655639648438e-05,
      "step": 19910,
      "training_step_time": 16.997695446014404
    },
    {
      "epoch": 1.8988609313964845e-05,
      "model_forward_time": 5.584326505661011,
      "step": 19911
    },
    {
      "epoch": 1.8988609313964845e-05,
      "step": 19911,
      "training_step_time": 17.014824867248535
    },
    {
      "epoch": 1.898956298828125e-05,
      "model_forward_time": 5.587197303771973,
      "step": 19912
    },
    {
      "epoch": 1.898956298828125e-05,
      "step": 19912,
      "training_step_time": 17.019328832626343
    },
    {
      "epoch": 1.8990516662597656e-05,
      "model_forward_time": 5.587615728378296,
      "step": 19913
    },
    {
      "epoch": 1.8990516662597656e-05,
      "step": 19913,
      "training_step_time": 17.02563762664795
    },
    {
      "epoch": 1.8991470336914064e-05,
      "model_forward_time": 5.588958024978638,
      "step": 19914
    },
    {
      "epoch": 1.8991470336914064e-05,
      "step": 19914,
      "training_step_time": 17.045227527618408
    },
    {
      "epoch": 1.8992424011230468e-05,
      "model_forward_time": 5.9859631061553955,
      "step": 19915
    },
    {
      "epoch": 1.8992424011230468e-05,
      "step": 19915,
      "training_step_time": 17.495429277420044
    },
    {
      "epoch": 1.8993377685546875e-05,
      "model_forward_time": 5.626633167266846,
      "step": 19916
    },
    {
      "epoch": 1.8993377685546875e-05,
      "step": 19916,
      "training_step_time": 17.138601541519165
    },
    {
      "epoch": 1.8994331359863283e-05,
      "model_forward_time": 5.614750623703003,
      "step": 19917
    },
    {
      "epoch": 1.8994331359863283e-05,
      "step": 19917,
      "training_step_time": 17.08712935447693
    },
    {
      "epoch": 1.8995285034179687e-05,
      "model_forward_time": 5.593444108963013,
      "step": 19918
    },
    {
      "epoch": 1.8995285034179687e-05,
      "step": 19918,
      "training_step_time": 17.0366792678833
    },
    {
      "epoch": 1.8996238708496094e-05,
      "model_forward_time": 5.582040309906006,
      "step": 19919
    },
    {
      "epoch": 1.8996238708496094e-05,
      "step": 19919,
      "training_step_time": 17.012457847595215
    },
    {
      "epoch": 1.89971923828125e-05,
      "grad_norm": 0.40720129013061523,
      "learning_rate": 1e-05,
      "loss": 0.0596,
      "step": 19920
    },
    {
      "epoch": 1.89971923828125e-05,
      "model_forward_time": 5.940980672836304,
      "step": 19920
    },
    {
      "dynamics_loss_avg": 0.04886638596653938,
      "epoch": 1.89971923828125e-05,
      "step": 19920
    },
    {
      "action_loss_avg": 0.0048173199174925685,
      "epoch": 1.89971923828125e-05,
      "step": 19920
    },
    {
      "epoch": 1.89971923828125e-05,
      "step": 19920,
      "training_step_time": 17.36725640296936
    },
    {
      "epoch": 1.8998146057128906e-05,
      "model_forward_time": 5.594444513320923,
      "step": 19921
    },
    {
      "epoch": 1.8998146057128906e-05,
      "step": 19921,
      "training_step_time": 17.057790994644165
    },
    {
      "epoch": 1.8999099731445313e-05,
      "model_forward_time": 5.614691734313965,
      "step": 19922
    },
    {
      "epoch": 1.8999099731445313e-05,
      "step": 19922,
      "training_step_time": 17.13056969642639
    },
    {
      "epoch": 1.9000053405761717e-05,
      "model_forward_time": 5.613262176513672,
      "step": 19923
    },
    {
      "epoch": 1.9000053405761717e-05,
      "step": 19923,
      "training_step_time": 17.124528646469116
    },
    {
      "epoch": 1.9001007080078125e-05,
      "model_forward_time": 5.630138635635376,
      "step": 19924
    },
    {
      "epoch": 1.9001007080078125e-05,
      "step": 19924,
      "training_step_time": 17.150182485580444
    },
    {
      "epoch": 1.9001960754394532e-05,
      "model_forward_time": 5.621767997741699,
      "step": 19925
    },
    {
      "epoch": 1.9001960754394532e-05,
      "step": 19925,
      "training_step_time": 17.123302221298218
    },
    {
      "epoch": 1.9002914428710936e-05,
      "model_forward_time": 5.97260046005249,
      "step": 19926
    },
    {
      "epoch": 1.9002914428710936e-05,
      "step": 19926,
      "training_step_time": 17.448182821273804
    },
    {
      "epoch": 1.9003868103027343e-05,
      "model_forward_time": 5.598325252532959,
      "step": 19927
    },
    {
      "epoch": 1.9003868103027343e-05,
      "step": 19927,
      "training_step_time": 17.04054355621338
    },
    {
      "epoch": 1.900482177734375e-05,
      "model_forward_time": 5.590163707733154,
      "step": 19928
    },
    {
      "epoch": 1.900482177734375e-05,
      "step": 19928,
      "training_step_time": 17.02457904815674
    },
    {
      "epoch": 1.9005775451660155e-05,
      "model_forward_time": 5.576078414916992,
      "step": 19929
    },
    {
      "epoch": 1.9005775451660155e-05,
      "step": 19929,
      "training_step_time": 16.994091033935547
    },
    {
      "epoch": 1.9006729125976562e-05,
      "grad_norm": 0.8966622948646545,
      "learning_rate": 1e-05,
      "loss": 0.0704,
      "step": 19930
    },
    {
      "epoch": 1.9006729125976562e-05,
      "model_forward_time": 5.585069179534912,
      "step": 19930
    },
    {
      "dynamics_loss_avg": 0.0616899037733674,
      "epoch": 1.9006729125976562e-05,
      "step": 19930
    },
    {
      "action_loss_avg": 0.0062312059570103885,
      "epoch": 1.9006729125976562e-05,
      "step": 19930
    },
    {
      "epoch": 1.9006729125976562e-05,
      "step": 19930,
      "training_step_time": 17.016321659088135
    },
    {
      "epoch": 1.900768280029297e-05,
      "model_forward_time": 5.95734977722168,
      "step": 19931
    },
    {
      "epoch": 1.900768280029297e-05,
      "step": 19931,
      "training_step_time": 17.38991117477417
    },
    {
      "epoch": 1.9008636474609374e-05,
      "model_forward_time": 5.602734804153442,
      "step": 19932
    },
    {
      "epoch": 1.9008636474609374e-05,
      "step": 19932,
      "training_step_time": 17.080724954605103
    },
    {
      "epoch": 1.900959014892578e-05,
      "model_forward_time": 5.616403579711914,
      "step": 19933
    },
    {
      "epoch": 1.900959014892578e-05,
      "step": 19933,
      "training_step_time": 17.133050680160522
    },
    {
      "epoch": 1.901054382324219e-05,
      "model_forward_time": 5.626773357391357,
      "step": 19934
    },
    {
      "epoch": 1.901054382324219e-05,
      "step": 19934,
      "training_step_time": 17.1488139629364
    },
    {
      "epoch": 1.9011497497558593e-05,
      "model_forward_time": 5.623743057250977,
      "step": 19935
    },
    {
      "epoch": 1.9011497497558593e-05,
      "step": 19935,
      "training_step_time": 17.140883207321167
    },
    {
      "epoch": 1.9012451171875e-05,
      "model_forward_time": 5.613036632537842,
      "step": 19936
    },
    {
      "epoch": 1.9012451171875e-05,
      "step": 19936,
      "training_step_time": 17.123167276382446
    },
    {
      "epoch": 1.9013404846191407e-05,
      "model_forward_time": 5.973261117935181,
      "step": 19937
    },
    {
      "epoch": 1.9013404846191407e-05,
      "step": 19937,
      "training_step_time": 17.452073097229004
    },
    {
      "epoch": 1.901435852050781e-05,
      "model_forward_time": 5.593331336975098,
      "step": 19938
    },
    {
      "epoch": 1.901435852050781e-05,
      "step": 19938,
      "training_step_time": 17.016713857650757
    },
    {
      "epoch": 1.901531219482422e-05,
      "model_forward_time": 5.59751558303833,
      "step": 19939
    },
    {
      "epoch": 1.901531219482422e-05,
      "step": 19939,
      "training_step_time": 17.037476301193237
    },
    {
      "epoch": 1.9016265869140626e-05,
      "grad_norm": 0.42692437767982483,
      "learning_rate": 1e-05,
      "loss": 0.0565,
      "step": 19940
    },
    {
      "epoch": 1.9016265869140626e-05,
      "model_forward_time": 5.5965797901153564,
      "step": 19940
    },
    {
      "dynamics_loss_avg": 0.05000109560787678,
      "epoch": 1.9016265869140626e-05,
      "step": 19940
    },
    {
      "action_loss_avg": 0.006109288637526333,
      "epoch": 1.9016265869140626e-05,
      "step": 19940
    },
    {
      "epoch": 1.9016265869140626e-05,
      "step": 19940,
      "training_step_time": 17.04986548423767
    },
    {
      "epoch": 1.901721954345703e-05,
      "model_forward_time": 5.589290380477905,
      "step": 19941
    },
    {
      "epoch": 1.901721954345703e-05,
      "step": 19941,
      "training_step_time": 17.01106905937195
    },
    {
      "epoch": 1.9018173217773438e-05,
      "model_forward_time": 5.5786216259002686,
      "step": 19942
    },
    {
      "epoch": 1.9018173217773438e-05,
      "step": 19942,
      "training_step_time": 17.370943069458008
    },
    {
      "epoch": 1.9019126892089845e-05,
      "model_forward_time": 5.581815242767334,
      "step": 19943
    },
    {
      "epoch": 1.9019126892089845e-05,
      "step": 19943,
      "training_step_time": 17.022899866104126
    },
    {
      "epoch": 1.902008056640625e-05,
      "model_forward_time": 5.588114023208618,
      "step": 19944
    },
    {
      "epoch": 1.902008056640625e-05,
      "step": 19944,
      "training_step_time": 17.024428367614746
    },
    {
      "epoch": 1.9021034240722657e-05,
      "model_forward_time": 5.5866758823394775,
      "step": 19945
    },
    {
      "epoch": 1.9021034240722657e-05,
      "step": 19945,
      "training_step_time": 17.016554355621338
    },
    {
      "epoch": 1.9021987915039064e-05,
      "model_forward_time": 5.594576358795166,
      "step": 19946
    },
    {
      "epoch": 1.9021987915039064e-05,
      "step": 19946,
      "training_step_time": 17.05255889892578
    },
    {
      "epoch": 1.9022941589355468e-05,
      "model_forward_time": 5.629241704940796,
      "step": 19947
    },
    {
      "epoch": 1.9022941589355468e-05,
      "step": 19947,
      "training_step_time": 17.087830066680908
    },
    {
      "epoch": 1.9023895263671875e-05,
      "model_forward_time": 5.959913015365601,
      "step": 19948
    },
    {
      "epoch": 1.9023895263671875e-05,
      "step": 19948,
      "training_step_time": 17.42866611480713
    },
    {
      "epoch": 1.9024848937988283e-05,
      "model_forward_time": 5.630922794342041,
      "step": 19949
    },
    {
      "epoch": 1.9024848937988283e-05,
      "step": 19949,
      "training_step_time": 17.144431114196777
    },
    {
      "epoch": 1.9025802612304687e-05,
      "grad_norm": 0.4205995202064514,
      "learning_rate": 1e-05,
      "loss": 0.0506,
      "step": 19950
    },
    {
      "epoch": 1.9025802612304687e-05,
      "model_forward_time": 5.656772613525391,
      "step": 19950
    },
    {
      "dynamics_loss_avg": 0.048086545616388324,
      "epoch": 1.9025802612304687e-05,
      "step": 19950
    },
    {
      "action_loss_avg": 0.003627922246232629,
      "epoch": 1.9025802612304687e-05,
      "step": 19950
    },
    {
      "epoch": 1.9025802612304687e-05,
      "step": 19950,
      "training_step_time": 17.194732666015625
    },
    {
      "epoch": 1.9026756286621094e-05,
      "model_forward_time": 5.640477895736694,
      "step": 19951
    },
    {
      "epoch": 1.9026756286621094e-05,
      "step": 19951,
      "training_step_time": 17.132654428482056
    },
    {
      "epoch": 1.90277099609375e-05,
      "model_forward_time": 5.588404417037964,
      "step": 19952
    },
    {
      "epoch": 1.90277099609375e-05,
      "step": 19952,
      "training_step_time": 17.02764630317688
    },
    {
      "epoch": 1.9028663635253906e-05,
      "model_forward_time": 5.928006649017334,
      "step": 19953
    },
    {
      "epoch": 1.9028663635253906e-05,
      "step": 19953,
      "training_step_time": 17.344531297683716
    },
    {
      "epoch": 1.9029617309570313e-05,
      "model_forward_time": 5.564598560333252,
      "step": 19954
    },
    {
      "epoch": 1.9029617309570313e-05,
      "step": 19954,
      "training_step_time": 16.984294652938843
    },
    {
      "epoch": 1.9030570983886717e-05,
      "model_forward_time": 5.574383735656738,
      "step": 19955
    },
    {
      "epoch": 1.9030570983886717e-05,
      "step": 19955,
      "training_step_time": 17.01376986503601
    },
    {
      "epoch": 1.9031524658203125e-05,
      "model_forward_time": 5.5660154819488525,
      "step": 19956
    },
    {
      "epoch": 1.9031524658203125e-05,
      "step": 19956,
      "training_step_time": 16.985326528549194
    },
    {
      "epoch": 1.9032478332519532e-05,
      "model_forward_time": 5.5722362995147705,
      "step": 19957
    },
    {
      "epoch": 1.9032478332519532e-05,
      "step": 19957,
      "training_step_time": 16.992021799087524
    },
    {
      "epoch": 1.9033432006835936e-05,
      "model_forward_time": 5.5755980014801025,
      "step": 19958
    },
    {
      "epoch": 1.9033432006835936e-05,
      "step": 19958,
      "training_step_time": 17.0091609954834
    },
    {
      "epoch": 1.9034385681152343e-05,
      "model_forward_time": 5.9496543407440186,
      "step": 19959
    },
    {
      "epoch": 1.9034385681152343e-05,
      "step": 19959,
      "training_step_time": 17.413881540298462
    },
    {
      "epoch": 1.903533935546875e-05,
      "grad_norm": 0.8595303297042847,
      "learning_rate": 1e-05,
      "loss": 0.0699,
      "step": 19960
    },
    {
      "epoch": 1.903533935546875e-05,
      "model_forward_time": 5.58990740776062,
      "step": 19960
    },
    {
      "dynamics_loss_avg": 0.059906115382909776,
      "epoch": 1.903533935546875e-05,
      "step": 19960
    },
    {
      "action_loss_avg": 0.006839184835553169,
      "epoch": 1.903533935546875e-05,
      "step": 19960
    },
    {
      "epoch": 1.903533935546875e-05,
      "step": 19960,
      "training_step_time": 17.02595067024231
    },
    {
      "epoch": 1.9036293029785155e-05,
      "model_forward_time": 5.594958305358887,
      "step": 19961
    },
    {
      "epoch": 1.9036293029785155e-05,
      "step": 19961,
      "training_step_time": 17.07315444946289
    },
    {
      "epoch": 1.9037246704101562e-05,
      "model_forward_time": 5.611072063446045,
      "step": 19962
    },
    {
      "epoch": 1.9037246704101562e-05,
      "step": 19962,
      "training_step_time": 17.12048029899597
    },
    {
      "epoch": 1.903820037841797e-05,
      "model_forward_time": 5.620030879974365,
      "step": 19963
    },
    {
      "epoch": 1.903820037841797e-05,
      "step": 19963,
      "training_step_time": 17.122708082199097
    },
    {
      "epoch": 1.9039154052734374e-05,
      "model_forward_time": 5.956334352493286,
      "step": 19964
    },
    {
      "epoch": 1.9039154052734374e-05,
      "step": 19964,
      "training_step_time": 17.446216583251953
    },
    {
      "epoch": 1.904010772705078e-05,
      "model_forward_time": 5.607141971588135,
      "step": 19965
    },
    {
      "epoch": 1.904010772705078e-05,
      "step": 19965,
      "training_step_time": 17.083536386489868
    },
    {
      "epoch": 1.904106140136719e-05,
      "model_forward_time": 5.604982852935791,
      "step": 19966
    },
    {
      "epoch": 1.904106140136719e-05,
      "step": 19966,
      "training_step_time": 17.06209659576416
    },
    {
      "epoch": 1.9042015075683593e-05,
      "model_forward_time": 5.59152364730835,
      "step": 19967
    },
    {
      "epoch": 1.9042015075683593e-05,
      "step": 19967,
      "training_step_time": 17.03973364830017
    },
    {
      "epoch": 1.904296875e-05,
      "model_forward_time": 5.615106105804443,
      "step": 19968
    },
    {
      "epoch": 1.904296875e-05,
      "step": 19968,
      "training_step_time": 17.061367511749268
    },
    {
      "epoch": 1.9043922424316407e-05,
      "model_forward_time": 5.597358226776123,
      "step": 19969
    },
    {
      "epoch": 1.9043922424316407e-05,
      "step": 19969,
      "training_step_time": 17.05111813545227
    },
    {
      "epoch": 1.904487609863281e-05,
      "grad_norm": 0.3935020864009857,
      "learning_rate": 1e-05,
      "loss": 0.0772,
      "step": 19970
    },
    {
      "epoch": 1.904487609863281e-05,
      "model_forward_time": 5.971563816070557,
      "step": 19970
    },
    {
      "dynamics_loss_avg": 0.07156510166823864,
      "epoch": 1.904487609863281e-05,
      "step": 19970
    },
    {
      "action_loss_avg": 0.008012305712327361,
      "epoch": 1.904487609863281e-05,
      "step": 19970
    },
    {
      "epoch": 1.904487609863281e-05,
      "step": 19970,
      "training_step_time": 17.44084668159485
    },
    {
      "epoch": 1.904582977294922e-05,
      "model_forward_time": 5.609319686889648,
      "step": 19971
    },
    {
      "epoch": 1.904582977294922e-05,
      "step": 19971,
      "training_step_time": 17.08082604408264
    },
    {
      "epoch": 1.9046783447265626e-05,
      "model_forward_time": 5.621173620223999,
      "step": 19972
    },
    {
      "epoch": 1.9046783447265626e-05,
      "step": 19972,
      "training_step_time": 17.101112365722656
    },
    {
      "epoch": 1.904773712158203e-05,
      "model_forward_time": 5.635283470153809,
      "step": 19973
    },
    {
      "epoch": 1.904773712158203e-05,
      "step": 19973,
      "training_step_time": 17.109678506851196
    },
    {
      "epoch": 1.9048690795898438e-05,
      "model_forward_time": 5.596802473068237,
      "step": 19974
    },
    {
      "epoch": 1.9048690795898438e-05,
      "step": 19974,
      "training_step_time": 17.060104370117188
    },
    {
      "epoch": 1.9049644470214845e-05,
      "model_forward_time": 5.595323085784912,
      "step": 19975
    },
    {
      "epoch": 1.9049644470214845e-05,
      "step": 19975,
      "training_step_time": 17.392874717712402
    },
    {
      "epoch": 1.905059814453125e-05,
      "model_forward_time": 5.594080209732056,
      "step": 19976
    },
    {
      "epoch": 1.905059814453125e-05,
      "step": 19976,
      "training_step_time": 17.052428007125854
    },
    {
      "epoch": 1.9051551818847657e-05,
      "model_forward_time": 5.600027322769165,
      "step": 19977
    },
    {
      "epoch": 1.9051551818847657e-05,
      "step": 19977,
      "training_step_time": 17.06452703475952
    },
    {
      "epoch": 1.9052505493164064e-05,
      "model_forward_time": 5.5972819328308105,
      "step": 19978
    },
    {
      "epoch": 1.9052505493164064e-05,
      "step": 19978,
      "training_step_time": 17.06278133392334
    },
    {
      "epoch": 1.9053459167480468e-05,
      "model_forward_time": 5.610001087188721,
      "step": 19979
    },
    {
      "epoch": 1.9053459167480468e-05,
      "step": 19979,
      "training_step_time": 17.085305213928223
    },
    {
      "epoch": 1.9054412841796875e-05,
      "grad_norm": 0.4275819659233093,
      "learning_rate": 1e-05,
      "loss": 0.0493,
      "step": 19980
    },
    {
      "epoch": 1.9054412841796875e-05,
      "model_forward_time": 5.5976269245147705,
      "step": 19980
    },
    {
      "dynamics_loss_avg": 0.04831195566803217,
      "epoch": 1.9054412841796875e-05,
      "step": 19980
    },
    {
      "action_loss_avg": 0.004750172805506736,
      "epoch": 1.9054412841796875e-05,
      "step": 19980
    },
    {
      "epoch": 1.9054412841796875e-05,
      "step": 19980,
      "training_step_time": 17.05609631538391
    },
    {
      "epoch": 1.9055366516113283e-05,
      "model_forward_time": 5.965717077255249,
      "step": 19981
    },
    {
      "epoch": 1.9055366516113283e-05,
      "step": 19981,
      "training_step_time": 17.42141842842102
    },
    {
      "epoch": 1.9056320190429687e-05,
      "model_forward_time": 5.597121000289917,
      "step": 19982
    },
    {
      "epoch": 1.9056320190429687e-05,
      "step": 19982,
      "training_step_time": 17.045883417129517
    },
    {
      "epoch": 1.9057273864746094e-05,
      "model_forward_time": 5.605069875717163,
      "step": 19983
    },
    {
      "epoch": 1.9057273864746094e-05,
      "step": 19983,
      "training_step_time": 17.065404891967773
    },
    {
      "epoch": 1.90582275390625e-05,
      "model_forward_time": 5.5933308601379395,
      "step": 19984
    },
    {
      "epoch": 1.90582275390625e-05,
      "step": 19984,
      "training_step_time": 17.05171036720276
    },
    {
      "epoch": 1.9059181213378906e-05,
      "model_forward_time": 5.6018290519714355,
      "step": 19985
    },
    {
      "epoch": 1.9059181213378906e-05,
      "step": 19985,
      "training_step_time": 17.053908586502075
    },
    {
      "epoch": 1.9060134887695313e-05,
      "model_forward_time": 5.9742112159729,
      "step": 19986
    },
    {
      "epoch": 1.9060134887695313e-05,
      "step": 19986,
      "training_step_time": 17.455932140350342
    },
    {
      "epoch": 1.9061088562011717e-05,
      "model_forward_time": 5.609427452087402,
      "step": 19987
    },
    {
      "epoch": 1.9061088562011717e-05,
      "step": 19987,
      "training_step_time": 17.09789204597473
    },
    {
      "epoch": 1.9062042236328125e-05,
      "model_forward_time": 5.608953237533569,
      "step": 19988
    },
    {
      "epoch": 1.9062042236328125e-05,
      "step": 19988,
      "training_step_time": 17.093680143356323
    },
    {
      "epoch": 1.9062995910644532e-05,
      "model_forward_time": 5.613590717315674,
      "step": 19989
    },
    {
      "epoch": 1.9062995910644532e-05,
      "step": 19989,
      "training_step_time": 17.096948385238647
    },
    {
      "epoch": 1.9063949584960936e-05,
      "grad_norm": 0.7801159620285034,
      "learning_rate": 1e-05,
      "loss": 0.0556,
      "step": 19990
    },
    {
      "epoch": 1.9063949584960936e-05,
      "model_forward_time": 5.5999791622161865,
      "step": 19990
    },
    {
      "dynamics_loss_avg": 0.047181186638772486,
      "epoch": 1.9063949584960936e-05,
      "step": 19990
    },
    {
      "action_loss_avg": 0.004648121097125113,
      "epoch": 1.9063949584960936e-05,
      "step": 19990
    },
    {
      "epoch": 1.9063949584960936e-05,
      "step": 19990,
      "training_step_time": 17.078779935836792
    },
    {
      "epoch": 1.9064903259277343e-05,
      "model_forward_time": 5.60402774810791,
      "step": 19991
    },
    {
      "epoch": 1.9064903259277343e-05,
      "step": 19991,
      "training_step_time": 17.086593627929688
    },
    {
      "epoch": 1.906585693359375e-05,
      "model_forward_time": 5.961849212646484,
      "step": 19992
    },
    {
      "epoch": 1.906585693359375e-05,
      "step": 19992,
      "training_step_time": 17.416653633117676
    },
    {
      "epoch": 1.9066810607910155e-05,
      "model_forward_time": 5.581122875213623,
      "step": 19993
    },
    {
      "epoch": 1.9066810607910155e-05,
      "step": 19993,
      "training_step_time": 17.02212381362915
    },
    {
      "epoch": 1.9067764282226562e-05,
      "model_forward_time": 5.584828615188599,
      "step": 19994
    },
    {
      "epoch": 1.9067764282226562e-05,
      "step": 19994,
      "training_step_time": 17.016674757003784
    },
    {
      "epoch": 1.906871795654297e-05,
      "model_forward_time": 5.602505445480347,
      "step": 19995
    },
    {
      "epoch": 1.906871795654297e-05,
      "step": 19995,
      "training_step_time": 17.08000111579895
    },
    {
      "epoch": 1.9069671630859374e-05,
      "model_forward_time": 5.610419750213623,
      "step": 19996
    },
    {
      "epoch": 1.9069671630859374e-05,
      "step": 19996,
      "training_step_time": 17.088921070098877
    },
    {
      "epoch": 1.907062530517578e-05,
      "model_forward_time": 5.970927476882935,
      "step": 19997
    },
    {
      "epoch": 1.907062530517578e-05,
      "step": 19997,
      "training_step_time": 17.478672742843628
    },
    {
      "epoch": 1.907157897949219e-05,
      "model_forward_time": 5.6133811473846436,
      "step": 19998
    },
    {
      "epoch": 1.907157897949219e-05,
      "step": 19998,
      "training_step_time": 17.103032112121582
    },
    {
      "epoch": 1.9072532653808593e-05,
      "model_forward_time": 5.612900257110596,
      "step": 19999
    },
    {
      "epoch": 1.9072532653808593e-05,
      "step": 19999,
      "training_step_time": 17.112627029418945
    },
    {
      "epoch": 1.9073486328125e-05,
      "grad_norm": 0.6798304319381714,
      "learning_rate": 1e-05,
      "loss": 0.055,
      "step": 20000
    },
    {
      "epoch": 1.9073486328125e-05,
      "step": 20000,
      "total_flos": 0.0,
      "train_loss": 0.07172271664887667,
      "train_runtime": 355283.9381,
      "train_samples_per_second": 0.056,
      "train_steps_per_second": 0.056
    }
  ],
  "logging_steps": 10,
  "max_steps": 20000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 10000,
  "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": 1,
  "trial_name": null,
  "trial_params": null
}